--- title: 02d Flow Datasets keywords: fastai sidebar: home_sidebar summary: "Here be the tailored datasets for FRED! In addition to storing points, we must store a graph, and intelligently sample from the graph so that batches preserve 'flow neighborhoods'" description: "Here be the tailored datasets for FRED! In addition to storing points, we must store a graph, and intelligently sample from the graph so that batches preserve 'flow neighborhoods'" nb_path: "02d Flow Datasets.ipynb" ---
{% raw %}
{% endraw %}

As previously listed, this is our preprocessing pipeline.

  1. Turn data to torch tensors
  2. Build graph from data, using flashlight kernel
    1. Estimate best parameters for kernel selection
  3. Take diffusion map on data (or any ‘best guess embedding’) we want to penalize against. (Can be done on CPU or GPU). Would likely use established methods, like PHATE, tSNE, node2vec.
    1. Visualize embedding
  4. Find flow neighborhoods of data (possibly at multiple scales)
  5. Break data into batches (both affinity matrix of batches and points in batches).
  6. Return Dataloader with minibatch X and subgraph A of each batch.

0-3 are handled by separate functions. This notebook calls upon these functions, and implements the machinery for 4 and 5

Two Types of Data

FRED can operate on both directed graphs (assumed to be sampled from a manifold with flow, ala the "Manifold with Flow" hypothesis) and pointclouds with associated velocity vectors (like RNA velocity single cell data). Each type is processed differently, and requires a different dataset class.

{% raw %}

class ManifoldWithVectorField[source]

ManifoldWithVectorField(X, velocities, labels, sigma='automatic', prior_embedding='diffusion map', t_dmap=1, dmap_coords_to_use=2, n_neighbors=5, minibatch_size=100, nbhd_strategy='flow neighbors') :: Dataset

Dataset object to be used with FRED for pointcloud and velocity input data. Takes np.arrays for X (points) and velocities (velocity vectors per point). For each item retrieved, returns a neighborhood around that point (based on local euclidean neighbors) containing local affinities

{% endraw %} {% raw %}
{% endraw %}

Testing

{% raw %}
from FRED.datasets import directed_swiss_roll_delayed
X, flow, labels = directed_swiss_roll_delayed(num_nodes=1000)
{% endraw %} {% raw %}
ds = ManifoldWithVectorField(X, flow, labels)
Set sigma =  2.3770864
{% endraw %} {% raw %}
for data in ds:
    print(data)
{'P': tensor([[4.0173e-02, 2.0302e-01, 1.4001e-01,  ..., 3.0227e-06, 7.4944e-05,
         9.0509e-05],
        [5.7301e-03, 4.8314e-02, 1.7639e-01,  ..., 3.6155e-06, 1.1521e-04,
         1.3691e-04],
        [3.7528e-03, 5.3394e-03, 5.2184e-02,  ..., 5.4380e-06, 2.1628e-04,
         2.7195e-04],
        ...,
        [4.0596e-05, 3.4174e-05, 3.4960e-05,  ..., 5.3405e-01, 4.7545e-05,
         5.5604e-05],
        [9.8737e-07, 9.0580e-07, 8.7754e-07,  ..., 1.1897e-05, 4.3833e-03,
         1.0262e-05],
        [6.1747e-06, 5.9572e-06, 6.7358e-06,  ..., 3.8245e-04, 1.7551e-01,
         3.5869e-02]]), 'X': tensor([[  0.1206,  -4.7363,   8.7103],
        [  0.6952,  -4.8060,   8.8976],
        [  1.7228,  -4.7575,   8.3396],
        [  0.9074,  -4.8139,   8.1674],
        [  2.1120,  -4.6818,   8.6144],
        [  2.1504,  -4.6726,   8.7908],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -0.0410,   7.8591,   9.3457],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.3516,   0.6425,   0.3425],
        [  3.2701,   6.6338,   9.9209],
        [  6.1685,  12.1979,  -6.2473],
        [  8.3791,  10.5409,   5.5590],
        [  4.5954,  13.0099,  -4.4439],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -8.5900,   2.9767,   7.4811],
        [  5.6672,  -1.8631,   8.1020],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.0547,   7.2957,  -1.9343],
        [ -4.0868,   7.2981,   8.1874],
        [ 10.2712,   8.3707,  -4.9743],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.0961,   1.6529,   3.1553],
        [  6.1989,  -9.7591,   0.8289],
        [  9.2811,  -7.4336,   5.1771],
        [  2.5318,  13.7231,   6.3692],
        [  8.5093,  -8.1764,  -3.2381],
        [  6.2535,  -0.1709,  -9.4102],
        [  5.0190,   4.9705,  -7.9752],
        [  1.5383, -11.0274,  -4.9666],
        [ -3.2486,   7.5925,  -1.3719],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.1575,  12.2046,   3.3257],
        [  2.2373,  -4.6506,  -1.7334],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.3460,   1.4252,   1.2291],
        [ 12.5717,   1.9266,   7.4903],
        [ -6.3306,   5.9272,   5.6935],
        [  4.3523, -10.5232,   8.2138],
        [ -9.4501,  -1.7133,   2.2980],
        [ -7.0120,   5.2825,   7.7762],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.4671,  -0.5553,   0.1117],
        [  6.0073,   3.0986,   6.7937],
        [ -7.3446,   4.9113,  -5.9356],
        [ 12.2311,  -2.0445,   1.3545],
        [ -6.1914,  -8.2997,   8.5380],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.1960,  -3.2964,   1.6921],
        [  4.5800,   5.5034,  -3.9432],
        [  6.3587,   1.1596,  -3.6629],
        [  6.2905,   1.9366,   2.1169],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.9526,   3.2525,  -4.3821],
        [  5.1861,   4.7361,  -3.7312],
        [ -5.0943,  -9.1682,   0.5073],
        [  2.7590,  13.6621,   9.4496],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.1890,   4.7318,   0.9515],
        [ -5.7347,  -8.6908,   9.2878],
        [ -7.4807,   4.7463,   2.0616],
        [  2.1379,  -4.6757,   6.4981],
        [  6.0345,   3.0175,  -1.4424],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.4572,   7.1085,   7.7376],
        [ -7.5720,   4.6310,   2.1717],
        [  5.6093,  -1.9705,   6.4807],
        [ 11.7162,   5.6728,  -8.9372],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.0039,   7.3173,  -1.5885],
        [ -1.6334, -10.7207,   7.1397],
        [  6.3497,   1.3680,  -1.9076],
        [  6.3606,   0.9204,   3.2525],
        [ 12.6000,   1.3839,  -3.2992],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.1372,   2.6726,  -6.7885],
        [  0.6933,   7.7336,   2.1179],
        [  6.3200,   0.2690,   1.5900],
        [ 11.7418,   5.6070,  -1.4192],
        [  3.3839,  -4.1966,   8.3574],
        [ -1.5408,   7.8978,   5.3470],
        [  9.5203,   9.3483,   5.8781],
        [  5.8979,   3.3958,  -3.8713],
        [  6.0073,   3.0986,   6.7937],
        [ -9.3384,  -2.6161,  -4.7630],
        [  0.9659,  -4.8144,  -0.9594],
        [  5.3860,  -2.3418,   1.0826],
        [  4.1348,   5.9514,  -0.8856],
        [  6.1222,  -0.7261,   8.8204],
        [ -9.0961,   1.6529,   3.1553],
        [  1.5653,  13.9377,   4.1773],
        [  5.7151,  -1.7700,   3.1542],
        [ -1.6275, -10.7221,   3.2376],
        [  1.8719,   7.3713,  -3.7465],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -6.9838,  -7.4965,   0.9796],
        [ -0.2586,   7.8825,  -9.9840],
        [  9.0051,   9.9234,   5.4411],
        [ 11.1469,   6.9237,   1.6160]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0183, 0.0397,  ..., 1.0319, 0.4221, 0.4737],
        [0.0183, 0.0000, 0.0250,  ..., 1.0408, 0.4039, 0.4658],
        [0.0397, 0.0250, 0.0000,  ..., 1.0334, 0.3858, 0.4422],
        ...,
        [1.0319, 1.0408, 1.0334,  ..., 0.0000, 1.2644, 0.8532],
        [0.4221, 0.4039, 0.3858,  ..., 1.2644, 0.0000, 0.4443],
        [0.4737, 0.4658, 0.4422,  ..., 0.8532, 0.4443, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.73784222,  4.85603886,  5.05981914,  4.89869205,  5.13616224,
        5.14371005, 10.06437598, 10.32715715,  7.85919687,  9.50654967,
       10.07238463,  6.38399789,  7.39599986, 13.66896739, 13.46553623,
       13.79762415, 12.05300943,  9.09119351,  5.96556248, 12.9089559 ,
        7.57946459,  8.36445758, 13.25017211,  8.94395628,  9.24501829,
       11.56146873, 11.89105259, 13.95472331, 11.80092578,  6.25586297,
        7.06372172, 11.1341798 ,  8.25828409, 12.10942412, 13.66991027,
        5.16077764, 12.33167382,  6.50409372, 12.71843462,  8.67227386,
       11.38774149,  9.60412735,  8.77913428, 11.62158554,  9.48337063,
        6.7593917 ,  8.8353674 , 12.40074901, 10.35464916,  5.53255882,
       10.06437598,  9.60359158,  9.76896904,  7.15990676,  6.46356806,
        6.58184474, 12.1494023 ,  6.78325151,  7.02326515, 10.48840476,
       13.93790235, 12.32408516,  7.02253235, 10.412299  ,  8.85939626,
        5.14124612,  6.74686071,  6.10774544,  7.52117132,  8.87587457,
        5.94535529, 13.01728313, 10.31613269,  7.58668144, 10.84437705,
        6.49538841,  6.42689705, 12.67576792, 13.04217487, 13.01187484,
        6.69389431,  7.76457095,  6.32572323, 13.01187484,  5.39098669,
        8.04664886, 13.3426541 ,  6.80559669,  6.7593917 ,  9.69792513,
        4.91038884,  5.87305005,  7.24677664,  6.16512856,  9.24501829,
       14.02532758,  5.98285319, 10.84493732,  7.60529892, 12.33167382,
        5.91854397,  9.60080639, 10.24557047,  7.88677538, 13.40024224,
       13.12218013])}
{'P': tensor([[4.9490e-02, 1.5156e-01, 1.3517e-01,  ..., 2.1036e-03, 6.0857e-05,
         3.0897e-05],
        [3.4654e-03, 5.6177e-02, 2.6353e-02,  ..., 6.2701e-04, 4.8607e-05,
         2.8119e-05],
        [4.9679e-03, 5.4702e-02, 7.0618e-02,  ..., 1.0231e-03, 7.1857e-05,
         4.5569e-05],
        ...,
        [6.7935e-04, 1.7123e-03, 2.2919e-03,  ..., 1.5085e-02, 1.3171e-06,
         1.0505e-06],
        [5.6071e-04, 2.1301e-04, 2.8238e-04,  ..., 3.6791e-04, 2.9155e-02,
         5.6511e-04],
        [1.2404e-04, 5.3041e-05, 7.6161e-05,  ..., 8.7878e-05, 2.9782e-02,
         2.8419e-02]]), 'X': tensor([[  0.1532,  -4.7422,  -7.0145],
        [  2.0899,  -4.6870,  -7.4260],
        [  1.9503,  -4.7172,  -6.2313],
        [  0.8700,  -4.8132,  -7.8042],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.2990,  -4.2415,  -7.4857],
        [ -6.8618,  -7.6322,   3.7027],
        [ -7.5401,   4.6718,   9.0033],
        [  5.9003,  12.3550,   2.1917],
        [ -0.8675, -10.8815,   6.8561],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.1795,  -3.3623,   8.5259],
        [  3.8182,  -3.9347,  -1.7961],
        [ -5.2752,  -9.0409,  -8.2568],
        [  9.5261,  -7.1675,   1.4206],
        [  0.9445,   7.6736,   0.5095],
        [ -0.0410,   7.8591,   9.3457],
        [  5.4732,  -2.2041,  -4.3830],
        [ -8.8154,   2.4611,   2.5198],
        [ -8.3900,   3.3727,   3.1650],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 12.6035,   0.7494,   7.1536],
        [  6.2365,  -0.2578,  -6.0212],
        [ -8.6801,   2.7809,  -0.8443],
        [  0.5600, -11.0321,  -0.8660],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.6523,   3.9443,  -2.9449],
        [  0.9074,  -4.8139,   8.1674],
        [ 12.2435,   4.0076,   8.1257],
        [  6.0305,   3.0296,   0.1375],
        [  0.6142,  14.0802,   0.5407],
        [  6.2419,  -0.2313,   4.2440],
        [  0.3715,  -4.7756,   0.2312],
        [ 12.4901,  -0.7052,   1.7315],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.0642,  -3.7599,  -3.4565],
        [ -3.9508,  -9.8515,   1.7923],
        [ -7.5603,   4.6459,   8.2002],
        [  6.9693,  -9.3200,  -8.3822],
        [  3.9663,   6.1018,  -5.4104],
        [ -7.3470,  -7.0615,   5.0066],
        [  6.3606,   0.9204,   3.2525],
        [  6.1730,  -0.5375,  -8.8861],
        [ -7.4808,   4.7462,   2.0266],
        [ -4.2040,   7.2486,   2.9875],
        [  3.1892, -10.8220,  -6.8409],
        [ -9.0574,   1.7801,   8.8986],
        [ -6.3306,   5.9272,   5.6935],
        [ -9.3705,   0.4208,  -3.5815],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.8318,   3.5567,  -4.7692],
        [  8.3791,  10.5409,   5.5590],
        [  7.9141,  -8.6656,   9.6948],
        [ -5.6792,  -8.7352,   4.0318],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.9337,  12.8566,  -3.6582],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.2036,   2.4030,  -1.2500],
        [ -0.3432,   7.8900,   7.1805],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.0955,   2.8210,  -4.3321],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -8.5607,  -5.0938,  -5.4531],
        [  9.0424,  -7.6779,  -7.5286],
        [  6.1685,  12.1979,  -6.2473],
        [  2.7558,  -4.4844,   0.0636],
        [  0.3724,   7.7974,  -6.8722],
        [  2.7685, -10.8982,  -6.8928],
        [  9.3560,   9.5391,   9.4943],
        [  5.0506,   4.9277,  -9.5094],
        [ -2.5056,   7.7720,   5.5878],
        [ -7.8280,   4.2848,  -7.1955],
        [ -6.8466,  -7.6488,   1.9648],
        [  4.5325,  -3.3643,   6.7699],
        [  6.3413,   0.4883,  -6.8416],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.7418,   5.6070,  -1.4192],
        [ -0.8885,   7.9160,   0.0571],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.2576,   2.1359,  -3.3398],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.9695,  -2.9407,  -6.3498],
        [  2.1771, -10.9776,  -3.2007],
        [  1.7218,  -4.7576,  -3.3617],
        [  1.7659,   7.4127,   4.7871],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.3167,   6.6022,   6.3554],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.3107,  -2.7697,  -3.2919],
        [  0.2480, -11.0153,  -3.0183],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.3490,  -4.6199,  -4.2489],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.9883,   3.1536,   6.2320],
        [ -8.0534,  -6.0429,   2.4583],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.1756, -10.8247,   6.8882],
        [  0.2480, -11.0153,  -3.0183],
        [ -8.8058,   2.4850,  -1.7425],
        [ -7.5720,   4.6310,   2.1717]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0558, 0.0430,  ..., 0.5594, 0.7104, 0.7581],
        [0.0558, 0.0000, 0.0434,  ..., 0.6129, 0.7656, 0.8109],
        [0.0430, 0.0434, 0.0000,  ..., 0.5960, 0.7433, 0.7742],
        ...,
        [0.5594, 0.6129, 0.5960,  ..., 0.0000, 0.1784, 0.4564],
        [0.7104, 0.7656, 0.7433,  ..., 0.1784, 0.0000, 0.3600],
        [0.7581, 0.8109, 0.7742,  ..., 0.4564, 0.3600, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.74468406,  5.13182506,  5.10444097,  4.89121805,  5.34250225,
        5.37343173, 10.26327895,  8.87007495, 13.69162795, 10.91601655,
        6.80559669,  9.77588801,  5.48276038, 10.46739073, 11.92133129,
        7.73151309,  7.85919687,  5.90035324,  9.15253344,  9.04255655,
        9.47369628, 12.62576026,  6.24187044,  9.11473391, 11.04629471,
        5.69913817,  4.91652556,  6.89245352,  4.89869205, 12.88270018,
        6.74872981, 14.09357547,  6.24614905,  4.79003361, 12.50996828,
       10.74409834,  5.5366698 , 10.61417181,  8.8737556 , 11.63765273,
        7.27759233, 10.19036081,  6.42689705,  6.19633139,  8.859416  ,
        8.37954853, 11.28215429,  9.23071698,  8.67227386,  9.37989631,
       12.30695098,  9.85384623,  6.83083863, 13.46553623, 11.73567759,
       10.41909211, 12.17901033, 13.77075174, 11.59512194,  6.652742  ,
        7.89745107, 11.54672755,  6.7166297 ,  9.5014146 ,  9.96152788,
       11.86239961, 13.66896739,  5.26342203,  7.80625981, 11.24434374,
       13.3614598 ,  7.05627461,  8.16584537,  8.92395381, 10.26545792,
        5.64464807,  6.36004393,  8.50268322, 12.19805394, 13.01187484,
        7.96575782, 12.22503754,  6.61211224, 12.22276278, 12.32545637,
       11.1913598 ,  5.05963217,  7.62011734,  5.19065439,  7.38844703,
       12.4013153 ,  9.75075273,  9.71392179, 11.018084  ,  5.35774573,
        5.18277858,  9.64157461,  6.76791483, 10.06850677,  9.54154096,
        9.11473391, 12.52372629, 11.28093507, 11.018084  ,  9.14973319,
        8.87587457])}
{'P': tensor([[5.0548e-02, 1.7541e-01, 1.2957e-01,  ..., 4.8312e-04, 1.0174e-04,
         8.5882e-05],
        [5.1541e-03, 8.0280e-02, 2.2370e-02,  ..., 1.3540e-03, 9.9172e-05,
         9.6078e-05],
        [3.3401e-03, 4.8566e-02, 3.4404e-02,  ..., 4.4810e-04, 4.6866e-05,
         4.5534e-05],
        ...,
        [4.2873e-04, 3.0895e-04, 3.0374e-04,  ..., 1.1806e-01, 1.5168e-03,
         2.9359e-03],
        [2.7011e-03, 1.0885e-03, 1.3625e-03,  ..., 1.1705e-05, 3.5439e-02,
         4.2109e-04],
        [6.0415e-04, 2.4273e-04, 3.2145e-04,  ..., 5.6731e-06, 1.2760e-01,
         1.4571e-02]]), 'X': tensor([[  0.1677,  -4.7447,  -6.1529],
        [  1.9503,  -4.7172,  -6.2313],
        [  1.5246,  -4.7838,  -5.2095],
        [  2.3373,  -4.6232,  -5.2246],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.4356,  -4.5942,  -5.0808],
        [ -6.8466,  -7.6488,   1.9648],
        [  3.3943,   6.5483,   1.0790],
        [ -1.0407,   7.9166,  -6.6021],
        [  9.9750,  -6.6338,  -1.2440],
        [ -0.7599, -10.8995,  -6.0366],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.3118,   4.5452,   4.2114],
        [ -3.3812, -10.1235,   7.4384],
        [  1.9204,   7.3518,   8.4781],
        [  6.1752,   2.5242,   3.5874],
        [  0.6952,  -4.8060,   8.8976],
        [  5.4242, -10.1261,   3.9595],
        [  3.2647, -10.8066,  -0.3768],
        [  4.3459,  -3.5327,   0.3667],
        [  3.8182,  -3.9347,  -1.7961],
        [  2.6310, -10.9195,   3.8054],
        [  2.1120,  -4.6818,   8.6144],
        [  6.9284,  11.7061,  -0.3732],
        [  0.1931,  -4.7491,   7.4385],
        [  3.6469,   6.3624,  -1.8414],
        [  4.4915,   5.5990,   7.0405],
        [  7.7794,  11.0629,   8.9069],
        [ 12.5763,   0.1333,  -9.2650],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.3076,   0.7852,  -6.7010],
        [ -7.3554,   4.8985,  -0.6665],
        [ 11.7779,   5.5127,  -2.0699],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -7.4733,   4.7555,   6.2490],
        [  7.0515,  11.6194,   8.4289],
        [  2.5620,   7.0549,   8.4825],
        [  6.3550,   0.7129,   0.4917],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.1519, -10.2387,   9.8191],
        [  8.0381,  -8.5690,  -8.9523],
        [ -9.2974,   0.8371,   2.4852],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.6792,  -8.7352,   4.0318],
        [  8.8474,  10.0868,  -8.8644],
        [  1.7815,  -4.7481,   4.4289],
        [  4.6231,   5.4556,   8.4973],
        [  5.0190,   4.9705,  -7.9752],
        [  6.2242,  -0.3171,   7.4978],
        [  2.5128, -10.9365,   9.9645],
        [  2.6970,   6.9826,   9.7350],
        [  6.3569,   0.7591,   6.5011],
        [ -1.6220,   7.8917,   0.0315],
        [  6.3122,  -9.6994,  -6.3282],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.6801,   2.7809,  -0.8443],
        [  4.5340,  -3.3629,   8.9652],
        [  2.4149,  -4.6005,  -4.3339],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.7218,  -4.7576,  -3.3617],
        [  9.4554,   9.4245,   9.8800],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.4887,  -5.2448,  -0.6585],
        [  7.3592,  11.3937,   6.2407],
        [  1.3767, -11.0341,   9.9415],
        [ -0.9715, -10.8630,   0.3555],
        [  2.5974,   7.0363,  -7.7810],
        [ -8.0965,  -5.9712,   4.3466],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.9942,  -3.9990,   6.0448],
        [ -3.9508,  -9.8515,   1.7923],
        [  0.2571,  -4.7594,   1.5618],
        [  2.2659,  13.7894,   6.5436],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.5650,   2.0132,   4.6272],
        [ -2.0840,   7.8415,  -7.8657],
        [  2.6321,  -4.5296,  -2.5159],
        [ -0.0410,   7.8591,   9.3457],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.6260,  -1.9401,   3.0259],
        [  6.3550,   0.7129,   0.4917],
        [  1.3339,   7.5627,  -0.9034],
        [  6.3534,   1.2990,  -7.2942],
        [ -6.3066,  -8.1936,   9.5822],
        [ 11.8951,  -3.1570,   7.0742],
        [ -4.0633,  -9.7927,  -3.0825],
        [  9.4516,  -7.2501,  -1.8868],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.3373,  13.7721,  -0.2442],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -7.6525,  -6.6534,   5.1721],
        [  5.9912,   3.1453,  -3.9989],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -0.8675, -10.8815,   6.8561],
        [ 12.2692,  -1.8882,   3.9253],
        [ -2.1382,   7.8339,   7.3524],
        [ -6.8618,  -7.6322,   3.7027],
        [ -8.5607,  -5.0938,  -5.4531],
        [  1.8320,   7.3871,   5.1452],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -9.1459,   1.4780,  -4.2403]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.5881e-02, 3.7716e-02,  ..., 1.2206e-01, 8.3431e-01,
         7.3372e-01],
        [4.5881e-02, 0.0000e+00, 4.0549e-02,  ..., 1.4067e-01, 8.7801e-01,
         7.7852e-01],
        [3.7716e-02, 4.0549e-02, 0.0000e+00,  ..., 1.5827e-01, 8.6403e-01,
         7.6121e-01],
        ...,
        [1.2206e-01, 1.4067e-01, 1.5827e-01,  ..., 0.0000e+00, 7.7538e-01,
         6.8654e-01],
        [8.3431e-01, 8.7801e-01, 8.6403e-01,  ..., 7.7538e-01, 0.0000e+00,
         1.2469e-01],
        [7.3372e-01, 7.7852e-01, 7.6121e-01,  ..., 6.8654e-01, 1.2469e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.74770899,  5.10444097,  5.0209005 ,  5.18047638,  5.1224079 ,
        5.19986797, 10.26545792,  7.37576154,  7.98469166, 11.97949273,
       10.92597196, 10.23946247, 12.49317563,  6.99097013, 10.67322527,
        7.5984809 ,  6.67121748,  4.85603886, 11.48734978, 11.28895489,
        5.60064319,  5.48276038, 11.23201619,  5.13616224, 13.60275211,
        4.75303127,  7.33350642,  7.17789666, 13.52430176, 12.5769659 ,
        9.55720651,  9.50913849,  9.34061938,  8.83725081, 13.00413829,
        9.93553424,  8.8580671 , 13.59170809,  7.50564223,  6.3949043 ,
       12.64235359, 11.46176259, 11.74901778,  9.3349858 , 10.51437947,
       10.41909211, 13.41713661,  5.0713364 ,  7.15100303,  7.06372172,
        6.2322769 , 11.22142195,  7.48538905,  6.4020404 ,  8.05668609,
       11.57249984,  8.37954853, 12.46513646,  9.11473391,  5.64500512,
        5.19577146, 10.1979669 ,  5.05963217, 13.35013374, 10.05558893,
        9.97822473, 13.56367857, 11.11969815, 10.90638066,  7.50035355,
       10.06023665,  7.85919687,  9.84314712, 10.61417181,  4.7663526 ,
       13.97430191,  5.39450305,  9.37989631, 12.72524615,  8.1137452 ,
        5.23878371,  7.85919687,  9.54154096,  5.95111804,  6.3949043 ,
        7.67939546,  6.48486171, 10.33958639, 12.30695098, 10.60226614,
       11.9120321 , 12.25806559, 13.96905756, 10.37698435, 10.14044489,
        6.76661595,  9.96152788, 10.91601655, 12.41366754,  8.12043983,
       10.26327895,  9.96152788,  7.61088125,  7.48311002,  9.50654967,
        9.26455795])}
{'P': tensor([[4.1139e-02, 1.3687e-01, 1.2619e-01,  ..., 6.0020e-03, 1.8490e-02,
         2.1408e-04],
        [7.0207e-03, 5.2976e-02, 2.4365e-01,  ..., 9.0038e-03, 3.7164e-02,
         4.5747e-04],
        [5.8969e-03, 7.7068e-03, 9.7803e-02,  ..., 1.4939e-02, 1.2320e-01,
         1.4346e-03],
        ...,
        [2.4410e-04, 4.0001e-04, 9.5164e-04,  ..., 5.2651e-02, 1.9158e-02,
         7.0563e-03],
        [4.7157e-04, 5.5180e-04, 7.3069e-04,  ..., 2.6250e-04, 3.7837e-02,
         2.8409e-03],
        [3.4229e-06, 3.9795e-06, 4.7913e-06,  ..., 6.4743e-07, 2.9750e-05,
         1.4386e-02]]), 'X': tensor([[ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0296, 0.0534,  ..., 0.2545, 0.1171, 0.4922],
        [0.0296, 0.0000, 0.0345,  ..., 0.2816, 0.0938, 0.4629],
        [0.0534, 0.0345, 0.0000,  ..., 0.3069, 0.0640, 0.4562],
        ...,
        [0.2545, 0.2816, 0.3069,  ..., 0.0000, 0.3708, 0.7183],
        [0.1171, 0.0938, 0.0640,  ..., 0.3708, 0.0000, 0.4121],
        [0.4922, 0.4629, 0.4562,  ..., 0.7183, 0.4121, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.75303127,  4.91336994,  5.14607014,  5.08372149,  5.05981914,
        4.89869205,  9.41970031,  6.28961988, 11.04313641, 14.02058621,
       13.54774355, 14.05633441, 10.9036226 ,  7.02326515,  6.32572323,
        8.47371868,  9.49511524,  9.85330646,  5.5366698 , 13.45124917,
        7.19513806, 11.4890861 , 11.85743425, 10.57302546,  9.27474386,
        8.41171369,  9.12220271, 10.08001894,  9.41970031, 13.89264822,
        7.54127425, 13.44593519,  9.09661212,  9.50913849, 13.37121923,
        9.78286733,  5.14124612, 13.77075174,  5.14999799,  7.57946459,
        5.18047638, 12.32545637, 13.103709  ,  7.24326992,  9.19837792,
        8.36995953,  6.81084724,  7.17789666,  5.26067019,  7.72668467,
       11.42420508, 13.05148826, 12.17901033,  6.29821673,  8.25828409,
        6.84796202, 10.42667093, 11.72749923,  8.92549752,  8.10886021,
        6.55800267,  9.36256096, 12.1494023 , 11.97238624, 12.52372629,
        9.41605684,  8.29203356, 12.10942412,  9.34061938,  9.17016857,
       11.46176259, 13.96746816,  7.66377554, 13.32557133,  6.28961988,
        7.67939546, 11.37939695,  9.12220271,  4.89869205,  4.978704  ,
        9.34127013,  5.96556248,  5.48276038,  9.23178204, 10.71162681,
        6.21275016,  8.11227668,  5.70405525,  5.96556248,  8.29640613,
        6.29821673,  5.63824971,  7.16058587, 12.41366754, 11.04313641,
        8.10886021, 14.10796466,  6.58184474, 13.08414689,  7.89745107,
        7.16747878,  7.27759233,  9.58347456, 11.72802491,  5.94535529,
       12.85032836])}
{'P': tensor([[4.5437e-02, 1.4281e-01, 1.3215e-01,  ..., 9.0338e-05, 6.6545e-05,
         8.6647e-05],
        [2.8329e-03, 4.6383e-02, 1.9638e-01,  ..., 4.9344e-05, 5.6796e-05,
         2.3786e-04],
        [4.1540e-03, 1.3257e-02, 7.8348e-02,  ..., 7.5195e-05, 9.8392e-05,
         4.4135e-04],
        ...,
        [9.3749e-04, 4.4585e-04, 3.6956e-04,  ..., 1.3271e-02, 1.1109e-05,
         4.0201e-07],
        [3.7755e-03, 1.4582e-03, 1.3246e-03,  ..., 2.7324e-02, 2.3464e-01,
         3.6434e-06],
        [1.6237e-05, 1.6898e-05, 1.6771e-05,  ..., 1.5203e-05, 1.1664e-04,
         9.5553e-02]]), 'X': tensor([[  0.2571,  -4.7594,   1.5618],
        [  2.1824,  -4.6647,   1.1985],
        [  2.4697,  -4.5836,   1.4593],
        [  1.8721,  -4.7322,   0.5524],
        [  2.8890,  -4.4316,   1.1600],
        [  1.7000,  -4.7609,   2.7935],
        [  6.2535,  -0.1709,  -9.4102],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.6314,  -1.9302,  -9.6543],
        [ -3.3116, -10.1539,   7.5919],
        [  2.7590,  13.6621,   9.4496],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.6719,  -6.6260,   7.1468],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -2.6069, -10.4298,  -6.1995],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.2895,   0.0405,   5.7652],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.3569,   0.7591,   6.5011],
        [  5.9548,  -1.2273,  -2.6520],
        [ 10.1798,   8.5002,  -5.8893],
        [ -8.4824,  -5.2575,  -0.0795],
        [  9.6662,   9.1725,   8.1056],
        [  3.4401,  13.4540,   4.6576],
        [  1.6232,  -4.7717,  -2.5655],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -6.4368,   5.8356,   3.3014],
        [ -0.8569,   7.9156,  -6.5557],
        [ -3.5029,   7.5138,   8.1853],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.3087,   0.7792,  -7.5587],
        [  6.0054,  -1.0902,   8.3079],
        [  2.5318,  13.7231,   6.3692],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.2435,   4.0076,   8.1257],
        [ 12.2589,  -1.9312,   9.8657],
        [  2.6613,   7.0021,   4.5783],
        [  8.6420,  -8.0580,   6.5781],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -3.9456,  -9.8542,   0.7466],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.0608,   3.9362,  -5.9567],
        [ -7.4733,   4.7555,   6.2490],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -3.1204, -10.2345,   9.2288],
        [ 11.6634,   5.8050,   9.1590],
        [  3.3633,  13.4794,   0.5193],
        [  3.2159, -10.8166,  -1.0406],
        [  6.1752,   2.5242,   3.5874],
        [ -0.8569,   7.9156,  -6.5557],
        [  2.6970,   6.9826,   9.7350],
        [  5.7218,  -9.9934,   6.5764],
        [  7.1441,  11.5529,   2.3232],
        [ -4.7334,   6.9982,  -0.4392],
        [ -4.2040,   7.2486,   2.9875],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.5083,  -4.1177,   0.2389],
        [ -8.6801,   2.7809,  -0.8443],
        [  0.8572,  14.0501,  -8.3364],
        [  6.3542,   0.6934,  -3.3840],
        [  6.9433,  -9.3362,  -0.3421],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.1752,   2.5242,   3.5874],
        [  1.8766,   7.3694,   1.1887],
        [  2.4697,  -4.5836,   1.4593],
        [ 11.2939,   6.6325,  -5.8311],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.4932,   7.0903,  -5.4214],
        [ 11.0438,   7.1180,  -5.6991],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.1472,  -2.6801,   8.1194],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.1440, -10.9811,   8.5281],
        [ -5.9938,  -8.4745,   3.9735],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -1.6334, -10.7207,   7.1397],
        [  2.6310, -10.9195,   3.8054],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -0.8885,   7.9160,   0.0571],
        [  0.3724,   7.7974,  -6.8722],
        [ 12.1379,   4.4125,  -3.8833],
        [  1.9241,   7.3503,   7.9004],
        [ -2.9497, -10.3028,  -7.3479],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.3535,  -5.5118,   9.8399],
        [ 11.1469,   6.9237,   1.6160],
        [  5.9912,   3.1453,  -3.9989],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -5.6651,   6.4380,   1.3660],
        [  5.7151,  -1.7700,   3.1542],
        [  5.4167, -10.1293,   6.2156],
        [  0.9392,  -4.8143,  -3.4649],
        [  2.7590,  13.6621,   9.4496],
        [  3.6154,  -4.0640,  -7.9698],
        [  8.1528,  10.7452,   9.1493],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.4336,  -2.2676,   5.6933],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.7802,  -1.6363,  -6.2782],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -7.8126,   4.3066,   5.1791],
        [ 10.2712,   8.3707,  -4.9743]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[5.3633e-05, 5.3386e-02, 6.0314e-02,  ..., 9.1107e-01, 6.0764e-01,
         6.4391e-01],
        [5.3386e-02, 0.0000e+00, 2.1976e-02,  ..., 9.5367e-01, 6.5981e-01,
         6.0051e-01],
        [6.0314e-02, 2.1976e-02, 0.0000e+00,  ..., 9.6895e-01, 6.5853e-01,
         6.1167e-01],
        ...,
        [9.1107e-01, 9.5367e-01, 9.6895e-01,  ..., 0.0000e+00, 7.0727e-01,
         1.2656e+00],
        [6.0764e-01, 6.5981e-01, 6.5853e-01,  ..., 7.0727e-01, 2.1453e-04,
         1.2298e+00],
        [6.4391e-01, 6.0051e-01, 6.1167e-01,  ..., 1.2656e+00, 1.2298e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.7663526 ,  5.14999799,  5.20659892,  5.08910491,  5.29009836,
        5.05534609,  6.25586297,  8.92395381,  5.95297459, 10.68031351,
       13.93790235,  8.95133961, 10.1371552 ,  9.4997987 , 10.7506423 ,
        9.69629   ,  6.28961988,  8.62631801,  6.4020404 ,  6.07992623,
       13.26209545,  9.97964013, 13.32557133, 13.88683436,  5.04028538,
       10.06407839,  8.68832398,  7.96181288,  8.29022197,  5.07306274,
        9.15976746,  9.34127013,  6.10360073, 13.95472331, 13.99048094,
       12.88270018, 12.41012046,  7.49076899, 11.8159254 , 10.39942497,
       10.61472636,  6.27905292,  8.97052115,  8.8580671 , 10.23946247,
       10.69963954, 13.02818702, 13.89264822, 11.2845608 ,  6.67121748,
        7.96181288,  7.48538905, 11.51557166, 13.58333857,  8.4486771 ,
        8.37954853,  6.83515201, 12.22276278,  9.11473391, 14.07623266,
        6.39187285, 11.63502679,  9.86089579, 10.74409834,  6.67121748,
        7.60463349,  5.20659892, 13.09737053, 12.00995898,  7.5158489 ,
       13.13888841, 12.3489324 ,  5.80312949,  9.96152788, 13.01728313,
       11.18839169, 10.37998078, 10.53368644, 10.84437705, 11.23201619,
        9.50913849,  7.96575782,  7.80625981, 12.91505111,  7.59795445,
       10.7167285 ,  5.69913817, 10.00802627, 13.12218013,  6.76661595,
        9.47369628,  8.57560275,  5.98285319, 11.48663767,  4.90505566,
       13.93790235,  5.43943791, 13.48809595, 10.67322527, 10.06850677,
        5.88782913,  5.53175275,  6.00732193,  9.75401449,  8.92098752,
       13.25017211])}
{'P': tensor([[2.6155e-02, 1.3065e-01, 1.2302e-01,  ..., 1.6239e-03, 2.0246e-05,
         7.1272e-05],
        [2.7184e-03, 2.4938e-02, 1.2312e-01,  ..., 2.1770e-03, 1.6521e-05,
         5.0208e-05],
        [3.8798e-03, 6.3237e-03, 4.4816e-02,  ..., 4.5190e-03, 2.5959e-05,
         7.8817e-05],
        ...,
        [1.4605e-05, 1.8826e-05, 2.1792e-05,  ..., 1.2621e-02, 4.0825e-07,
         3.7609e-07],
        [1.3335e-03, 1.1184e-03, 9.1274e-04,  ..., 1.9905e-05, 1.0743e-01,
         3.7758e-03],
        [9.7739e-03, 7.2242e-03, 6.2573e-03,  ..., 6.6676e-05, 1.9303e-04,
         7.1108e-02]]), 'X': tensor([[  0.2732,  -4.7619,  -1.1577],
        [  0.9659,  -4.8144,  -0.9594],
        [  1.3403,  -4.8011,  -1.1698],
        [  1.3103,  -4.8032,  -0.9300],
        [  1.4749,  -4.7892,  -0.8089],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.9284,  11.7061,  -0.3732],
        [  2.3490,  -4.6199,  -4.2489],
        [  1.7228,  -4.7575,   8.3396],
        [ -0.0254, -10.9932,   3.2200],
        [ 10.5955,   7.8824,  -3.7006],
        [  2.3373,  13.7721,  -0.2442],
        [  0.6363, -11.0348,   6.1414],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.2242,  -0.3171,   7.4978],
        [  2.6614,  -4.5192,  -9.3641],
        [ -5.1398,  -9.1367,   6.2074],
        [ 10.7566,   7.6212,  -2.3279],
        [ -3.5173, -10.0623,   4.8280],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.3569,   0.7591,   6.5011],
        [  5.9912,   3.1453,  -3.9989],
        [  9.4516,  -7.2501,  -1.8868],
        [ -2.0229, -10.6163,   8.6930],
        [  2.8890,  -4.4316,   1.1600],
        [ -8.3539,  -5.5111,   0.3373],
        [  3.1756, -10.8247,   6.8882],
        [  0.1532,  -4.7422,  -7.0145],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4716,  -0.6673,   7.6068],
        [  5.8275,  -1.5334,   2.0124],
        [ 11.1469,   6.9237,   1.6160],
        [  4.8432,   5.1968,   1.5006],
        [  1.8447,  -4.7372,   8.1235],
        [  5.8902,  -1.3887,   4.0388],
        [ -3.4986,   7.5152,   9.0712],
        [ -8.8058,   2.4850,  -1.7425],
        [ -0.2586,   7.8825,  -9.9840],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.7295,  13.6703,  -8.9328],
        [  2.0899,  -4.6870,  -7.4260],
        [ -3.3999, -10.1152,  -1.2088],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.6165,  -3.8715,   6.4594],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.1644, -11.0394,   7.8511],
        [  0.9967,  -4.8144,  -9.7370],
        [ -9.3384,  -2.6161,  -4.7630],
        [  5.5424,  -2.0885,   7.2487],
        [  1.3403,  -4.8011,  -1.1698],
        [ -1.6275, -10.7221,   3.2376],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.2435,   7.8191,  -6.4709],
        [  5.5424,  -2.0885,   7.2487],
        [ -7.9667,   4.0814,  -0.6397],
        [  2.1440, -10.9811,   8.5281],
        [  6.3542,   0.6934,  -3.3840],
        [  5.5626,  -2.0536,   8.4433],
        [ -5.9938,  -8.4745,   3.9735],
        [  5.9883,   3.1536,   6.2320],
        [ -3.1204, -10.2345,   9.2288],
        [ -8.0750,  -6.0072,  -5.8267],
        [  1.3670,  -4.7990,   3.2603],
        [  0.4119,  14.1019,  -4.1870],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.5636, -10.9293,   9.6519],
        [  4.9152,  12.8653,  -1.4459],
        [  2.0899,  -4.6870,  -7.4260],
        [ -9.1178,   1.5784,   2.1410],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.9314,  -9.3435,  -1.2729],
        [  9.3836,   9.5075,  -5.9278],
        [ -5.9940,   6.1985,   5.9566],
        [  1.9503,  -4.7172,  -6.2313],
        [  2.6451,  -4.5250,  -0.5565],
        [  3.2990,  -4.2415,  -7.4857],
        [ -3.9700,   7.3453,  -4.1486],
        [ -9.4753,  -0.8012,  -5.2050],
        [ 12.6000,   1.3839,  -3.2992],
        [  9.7328,   9.0902,  -0.8917],
        [ 11.3800,  -4.3919,   1.4066],
        [ -3.2053, -10.1993,  -3.8736],
        [ -7.6734,   4.4980,  -5.4248],
        [ -5.9225,  -8.5355,   2.4778],
        [  2.8602,   6.8905,   5.8779],
        [  4.2612, -10.5514,  -6.6268],
        [  6.2895,   0.0405,   5.7652],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.2979,   9.6048,   7.3138],
        [  3.5965,  -4.0754,  -3.5035],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -5.6423,   6.4537,  -1.2091],
        [  8.2154,  -8.4262,   8.0171],
        [ -4.8751,  -9.3148,   4.0406],
        [  3.3744,   6.5623,  -6.3636],
        [ -1.3178, -10.7939,   0.4236],
        [  5.0348,  -2.8229,   3.5436],
        [  2.6451,  -4.5250,  -0.5565],
        [  3.3943,   6.5483,   1.0790],
        [ 10.6283,  -5.7253,   7.4878],
        [  3.3094,  13.4969,  -3.6507],
        [ 12.0582,  -2.6640,   0.7132],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.4716,  -0.6673,   7.6068],
        [ -8.9379,  -4.1655,  -2.8114]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0210, 0.0303,  ..., 0.1228, 0.9558, 0.8731],
        [0.0210, 0.0000, 0.0162,  ..., 0.1027, 0.9626, 0.8928],
        [0.0303, 0.0162, 0.0000,  ..., 0.1040, 0.9788, 0.9029],
        ...,
        [0.1228, 0.1027, 0.1040,  ..., 0.0000, 0.9695, 0.9753],
        [0.9558, 0.9626, 0.9788,  ..., 0.9695, 0.0000, 0.7971],
        [0.8731, 0.8928, 0.9029,  ..., 0.9753, 0.7971, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.76969364,  4.91038884,  4.98461983,  4.978704  ,  5.01114719,
        5.16077764, 13.60275211,  5.18277858,  5.05981914, 10.99326577,
       13.20597932, 13.96905756, 11.05317597,  9.97964013,  6.2322769 ,
        5.24461068, 10.48315436, 13.18278606, 10.65929968,  8.12580976,
        6.4020404 ,  6.76661595, 11.9120321 , 10.80728788,  5.29009836,
       10.00794468, 11.28093507,  4.74468406,  9.5014146 ,  9.49511524,
        6.02588517, 13.12218013,  7.10378919,  5.08372149,  6.05165248,
        8.28968067,  9.14973319,  7.88677538, 10.1371552 , 13.94008948,
        5.13182506, 10.67132524, 12.1460931 , 12.24467041,  8.30125201,
       11.10066126,  4.91652556,  9.69792513,  5.92281142,  4.98461983,
       10.84493732, 10.30972193,  7.822854  ,  5.92281142,  8.95133961,
       11.18839169,  6.39187285,  5.92952538, 10.37998078,  6.76791483,
       10.69963954, 10.06437598,  4.98989906, 14.10796466,  9.51074763,
       11.22596994, 13.77222752,  5.13182506,  9.25336371,  8.94395628,
       11.63383221, 13.35832725,  8.62260327,  5.10444097,  5.24137909,
        5.37343173,  8.34949383,  9.50913849, 12.67576792, 13.31764181,
       12.19805394, 10.6910768 ,  8.89458273, 10.38897563,  7.46053281,
       11.37939695,  6.28961988,  5.26067019, 13.36800073,  5.43544626,
       13.09737053,  9.61942445,  8.5724042 , 11.76830981, 10.51340829,
        7.37902766, 10.87409005,  5.77219077,  5.24137909,  7.37576154,
       12.07225469, 13.89671314, 12.3489324 , 12.22276278,  9.49511524,
        9.86089579])}
{'P': tensor([[3.6242e-02, 1.4240e-01, 1.2014e-01,  ..., 2.9021e-04, 5.3404e-04,
         1.0244e-03],
        [4.3833e-03, 4.3809e-02, 1.6438e-01,  ..., 2.9197e-04, 9.7738e-04,
         6.5585e-04],
        [3.8341e-03, 6.9352e-03, 6.4020e-02,  ..., 2.4113e-04, 1.8453e-03,
         4.7452e-04],
        ...,
        [7.4994e-05, 1.3752e-04, 1.5583e-04,  ..., 1.4940e-02, 2.7160e-05,
         1.0034e-04],
        [5.9013e-04, 5.4135e-04, 4.8083e-04,  ..., 2.2849e-05, 1.9000e-01,
         4.5732e-05],
        [5.5302e-03, 1.0389e-02, 1.1478e-02,  ..., 3.1505e-02, 1.2328e-04,
         7.5807e-02]]), 'X': tensor([[ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0385, 0.0500,  ..., 0.8705, 0.2687, 0.6897],
        [0.0385, 0.0000, 0.0394,  ..., 0.8618, 0.2377, 0.7022],
        [0.0500, 0.0394, 0.0000,  ..., 0.9012, 0.2687, 0.7364],
        ...,
        [0.8705, 0.8618, 0.9012,  ..., 0.0000, 0.6943, 0.4590],
        [0.2687, 0.2377, 0.2687,  ..., 0.6943, 0.0000, 0.6656],
        [0.6897, 0.7022, 0.7364,  ..., 0.4590, 0.6656, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.78002665,  4.98461983,  5.16077764,  4.978704  ,  5.01114719,
        4.91038884,  5.41348121,  6.07992623,  7.88677538, 14.02532758,
       11.81423161, 12.59840193,  9.94173708, 12.10942412,  7.39004226,
        8.49161154, 13.76047999,  9.47369628, 10.92597196, 12.05395056,
       12.1460931 , 11.00508681, 10.68416401,  9.17595899,  5.70405525,
        7.27759233, 11.018084  , 13.35927527,  6.53250937,  7.72668467,
        9.25336371, 12.19805394,  6.76056625, 13.72594022,  8.57560275,
        5.79543038,  6.34059021, 11.27160035, 10.06286074,  6.80370254,
       10.06437598,  5.53255882, 11.62158554,  5.57341623, 11.73567759,
       12.30858677,  6.76056625,  9.27474386,  6.83083863,  4.76969364,
       12.00995898, 13.36800073, 11.77124373,  9.41244865, 12.50996828,
        6.16512856, 12.41366754,  9.95552626, 10.1979669 , 12.05395056,
       11.24434374,  6.34059021,  7.89745107,  7.52117132, 13.36092624,
       13.744387  ,  5.91854397,  9.37989631,  4.78258092, 10.48315436,
        5.57341623, 11.35345345,  8.95983586,  6.81084724,  5.84533345,
        7.27340834, 12.67576792,  8.89301516, 12.05300943,  9.54154096,
        9.14973319,  6.61386436,  7.02253235, 14.03505624, 12.13568598,
       14.03618649,  5.43943791, 13.97430191, 13.62896593,  9.12220271,
       12.00995898,  5.10446653, 13.25017211, 13.59170809, 10.31613269,
       10.33712258,  5.24461068, 11.9182956 ,  5.21280156,  8.04718433,
        6.2322769 ,  9.55593374,  8.67227386, 11.18839169,  7.23276582,
       10.9036226 ])}
{'P': tensor([[2.7571e-02, 1.1520e-01, 9.8828e-02,  ..., 1.9918e-03, 1.6148e-05,
         1.6323e-03],
        [2.1113e-03, 2.4743e-02, 1.5261e-01,  ..., 1.8285e-03, 1.5262e-05,
         1.8420e-03],
        [1.8649e-03, 2.9298e-03, 2.8372e-02,  ..., 1.7884e-03, 1.9544e-05,
         2.5192e-03],
        ...,
        [1.0614e-03, 2.3539e-03, 3.3078e-03,  ..., 2.2024e-01, 8.9246e-06,
         2.4672e-03],
        [2.5898e-05, 1.7380e-05, 1.5861e-05,  ..., 3.8340e-07, 3.1055e-02,
         3.0026e-05],
        [3.5881e-04, 4.1535e-04, 4.7673e-04,  ..., 3.6151e-05, 1.7305e-03,
         7.6985e-02]]), 'X': tensor([[ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0340, 0.0435,  ..., 0.2909, 0.7982, 0.4091],
        [0.0340, 0.0000, 0.0121,  ..., 0.2810, 0.8221, 0.4172],
        [0.0435, 0.0121, 0.0000,  ..., 0.2873, 0.8224, 0.4117],
        ...,
        [0.2909, 0.2810, 0.2873,  ..., 0.0000, 1.0714, 0.6981],
        [0.7982, 0.8221, 0.8224,  ..., 1.0714, 0.0000, 0.5019],
        [0.4091, 0.4172, 0.4117,  ..., 0.6981, 0.5019, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.78258092,  5.0209005 ,  5.1224079 ,  5.18047638,  5.19986797,
        5.19577146, 13.95361098,  6.76661595,  8.29203356,  7.18354885,
        6.81084724,  9.62238828,  7.5092834 ,  7.88677538,  6.02789634,
        9.43737523,  8.04483495, 10.87409005, 12.0495521 , 10.06850677,
        9.05843201,  8.50268322, 13.45124917,  5.91677188, 12.3828957 ,
        8.29203356, 13.89264822, 12.52372629,  4.7663526 ,  5.22903117,
        5.30158497, 11.57249984,  8.86082124, 10.51340829, 13.12218013,
        5.26067019,  5.80860165, 12.49317563, 12.91143491, 13.59480906,
       10.08001894,  6.42439344,  9.25336371,  7.24326992,  5.5513072 ,
       11.54071859, 10.24557047, 14.05633441,  5.22903117, 11.63502679,
       10.67132524, 13.59673406, 10.31886066, 11.77124373,  5.5671389 ,
        4.89869205,  9.38232121,  8.94395628,  6.53250937,  5.70591986,
       12.3344536 ,  5.0209005 , 10.46739073,  8.23525433, 10.30972193,
        6.81084724,  5.10446653, 10.27572673,  9.63742669,  5.9510489 ,
        7.78980299,  9.6282763 ,  9.88310614, 11.9702762 , 13.72594022,
        9.94173708,  8.06990483, 11.2845608 , 13.54774355,  8.19703512,
       10.61472636, 12.17451685,  8.53014915,  9.60412735,  8.53014915,
        5.18277858,  7.09887557,  6.35029274, 14.02526505, 12.95702647,
        9.71392179, 11.48434767, 13.62896593, 10.76154256, 13.27994304,
       10.87409005,  5.40689332,  5.18047638,  9.32094055,  8.53014915,
        5.39098669, 12.26997165,  5.79562691, 11.64436684,  8.11227668,
        6.2134961 ])}
{'P': tensor([[4.6873e-02, 1.6858e-01, 1.2597e-01,  ..., 4.4040e-04, 7.0507e-05,
         3.9049e-04],
        [4.1523e-03, 5.9202e-02, 4.5859e-02,  ..., 1.2991e-03, 5.4595e-05,
         7.0713e-04],
        [3.5495e-03, 2.2936e-02, 4.8894e-02,  ..., 8.5013e-04, 2.8770e-05,
         3.4807e-04],
        ...,
        [6.0428e-06, 8.6229e-06, 7.7038e-06,  ..., 1.8858e-02, 2.3783e-07,
         1.0543e-04],
        [1.7449e-04, 2.8518e-04, 1.7618e-04,  ..., 2.0247e-04, 6.4012e-02,
         3.9434e-03],
        [1.0700e-04, 1.5944e-04, 1.1079e-04,  ..., 4.8010e-03, 4.6319e-05,
         7.3089e-02]]), 'X': tensor([[  0.3493,  -4.7727,  -6.5682],
        [  1.9503,  -4.7172,  -6.2313],
        [  2.0899,  -4.6870,  -7.4260],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.4009,  -4.1874,  -5.9751],
        [  3.2990,  -4.2415,  -7.4857],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.0373,   7.1299,  -8.5895],
        [ -6.3489,   5.9117,   6.5161],
        [  8.6271,  -8.0715,  -7.4813],
        [ -7.6719,  -6.6260,   7.1468],
        [ 11.1469,   6.9237,   1.6160],
        [ 12.4592,  -0.9168,  -6.7571],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.0899,  -4.6870,  -7.4260],
        [  4.3554,   5.7393,  -9.0766],
        [  1.0093,   7.6567,   2.3526],
        [  0.2480, -11.0153,  -3.0183],
        [  6.0955,   2.8210,  -4.3321],
        [  6.7792,  11.8084,   1.6251],
        [ -9.3444,   0.5821,   6.9101],
        [  0.7275,  14.0667,  -0.9815],
        [  5.4336,  -2.2676,   5.6933],
        [  6.1685,  12.1979,  -6.2473],
        [  9.5261,  -7.1675,   1.4206],
        [  6.3583,   1.1729,   8.7835],
        [  0.9036,   7.6840,   9.9237],
        [  0.3724,   7.7974,  -6.8722],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.9193,  -1.3176,   5.1739],
        [  5.0233,   4.9647,  -3.1788],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.6672,  -1.8631,   8.1020],
        [ -8.8058,   2.4850,  -1.7425],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.1480,  -4.3165,  -7.8002],
        [ -9.2345,  -3.1345,   3.9715],
        [ 11.1420,   6.9331,  -2.4779],
        [  8.6420,  -8.0580,   6.5781],
        [  5.0233,   4.9647,  -3.1788],
        [  5.9026,   3.3837,  -0.6573],
        [  1.7000,  -4.7609,   2.7935],
        [ -8.6426,  -4.9136,   9.4941],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.3539,  -5.5111,   0.3373],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.8082,  -6.8396,   4.9615],
        [  0.2480, -11.0153,  -3.0183],
        [  6.2975,   0.0947,   5.1791],
        [ -6.2200,  -8.2737,   5.5079],
        [ -3.3999, -10.1152,  -1.2088],
        [  3.7916,  -3.9524,  -7.9547],
        [ -0.5385,   7.9037,   6.3830],
        [ 12.5789,   0.1722,   1.6470],
        [  5.9912,   3.1453,  -3.9989],
        [  5.6093,  -1.9705,   6.4807],
        [ -8.7544,   2.6096,   3.3146],
        [  8.8474,  10.0868,  -8.8644],
        [ 11.9641,  -2.9568,   7.7211],
        [  4.9957,  -2.8705,   6.5978],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.4731,  -0.7120,  -6.0271],
        [  9.4925,   9.3811,   8.9842],
        [  5.8318,   3.5567,  -4.7692],
        [ -5.6423,   6.4537,  -1.2091],
        [  2.2076,  13.8032,  -1.5099],
        [ -4.0409,   7.3169,  -3.5779],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.9314,  -9.3435,  -1.2729],
        [  5.3118,   4.5452,   4.2114],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.6112,   2.9319,  -7.9943],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -5.9940,   6.1985,   5.9566],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.0243,   7.3086,   7.5432],
        [ -5.3382,   6.6533,   7.8054],
        [  5.4867,  12.5816,   2.9549],
        [  5.0348,  -2.8229,   3.5436],
        [  3.1892, -10.8220,  -6.8409],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.2435,   4.0076,   8.1257],
        [  9.1409,  -7.5789,   1.5814],
        [  5.9526,   3.2525,  -4.3821],
        [  5.6022,  -1.9833,   8.3081],
        [ -3.5520,   7.4976,   9.1522],
        [  5.9548,  -1.2273,  -2.6520],
        [  4.4633,   5.6287,  -0.7754],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -6.7992,   5.4991,   9.9186],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.3616,  -2.4749,   0.1793],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.5626,  -2.0536,   8.4433],
        [ -1.5451,   7.8975,   7.6104],
        [ -9.3444,   0.5821,   6.9101],
        [  6.0467,  -9.8369,  -6.5816],
        [  9.6662,   9.1725,   8.1056],
        [  4.0587, -10.6112,   6.2273],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.5770,   1.8512,  -0.6578],
        [ -1.3940, -10.7772,   6.7835],
        [  5.9193,  -1.3176,   5.1739]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0363, 0.0603,  ..., 0.3629, 1.1321, 0.7279],
        [0.0363, 0.0000, 0.0434,  ..., 0.3358, 1.1499, 0.7177],
        [0.0603, 0.0434, 0.0000,  ..., 0.3630, 1.1902, 0.7578],
        ...,
        [0.3629, 0.3358, 0.3630,  ..., 0.0000, 1.1186, 0.4635],
        [1.1321, 1.1499, 1.1902,  ..., 1.1186, 0.0000, 0.8210],
        [0.7279, 0.7177, 0.7578,  ..., 0.4635, 0.8210, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.78544875,  5.10444097,  5.13182506,  5.18047638,  5.39450305,
        5.37343173, 12.86648771, 13.1399181 ,  8.67502348, 11.81423161,
       10.1371552 , 13.12218013, 12.49291768, 11.80092578,  5.13182506,
        7.20482224,  7.72291946, 11.018084  ,  6.7166297 , 13.61600536,
        9.36256096, 14.08549802,  5.88782913, 13.66896739, 11.92133129,
        6.46560181,  7.73692436,  7.80625981, 12.25806559,  6.06416957,
        7.06272256, 12.19805394,  5.96556248,  9.14973319, 11.77124373,
        5.34250225,  9.75200848, 13.12299145, 11.8159254 ,  7.06272256,
        6.80370254,  5.05534609,  9.94173708, 10.24557047, 10.00794468,
        5.18277858, 11.95743558, 11.018084  ,  6.29821673, 10.35093485,
       10.67132524,  5.47702766,  7.92200342, 12.58006038,  6.76661595,
        5.94535529,  9.13507321, 13.41713661, 12.32408516,  5.76165934,
        9.78286733,  7.8575648 ,  9.4997987 , 13.34587027,  6.83083863,
        8.5724042 , 13.97857591,  8.35856487, 12.64235359, 11.63383221,
        6.99097013,  8.83725081,  9.09661212, 10.61047596,  8.62260327,
        8.30125201,  7.58378655,  8.53014915, 13.72594022,  5.77219077,
       11.28215429, 11.62770332, 12.88270018, 11.87412439,  6.78325151,
        5.94293185,  8.29640613,  6.07992623,  7.18354885,  5.94535529,
        9.69792513,  8.74470218,  9.09963894,  9.68323686, 12.18045978,
        5.92952538,  8.04718433,  9.36256096, 11.54672755, 13.32557133,
       11.36089355,  6.81961996, 13.13205357, 12.7125079 , 10.86693708,
        6.06416957])}
{'P': tensor([[3.6870e-02, 1.6588e-01, 1.5047e-01,  ..., 1.5028e-02, 2.0389e-04,
         1.5676e-03],
        [6.6955e-03, 5.5964e-02, 2.9553e-01,  ..., 2.5051e-02, 3.4562e-04,
         2.4942e-03],
        [6.7472e-03, 8.9721e-03, 9.1224e-02,  ..., 2.9332e-02, 7.3501e-04,
         3.2224e-03],
        ...,
        [6.0508e-03, 8.9925e-03, 1.3318e-02,  ..., 1.1940e-01, 4.3703e-04,
         7.4951e-03],
        [1.1399e-04, 9.5688e-05, 8.9223e-05,  ..., 3.6407e-05, 5.6238e-02,
         1.7399e-06],
        [6.8129e-05, 1.0775e-04, 1.6428e-04,  ..., 7.3611e-04, 7.4268e-06,
         1.4534e-02]]), 'X': tensor([[  0.3715,  -4.7756,   0.2312],
        [  1.0391,  -4.8141,  -0.0768],
        [  1.7903,  -4.7467,   0.1168],
        [  0.9945,  -4.8144,  -0.1897],
        [  0.9478,  -4.8144,   0.6371],
        [  1.8721,  -4.7322,   0.5524],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.3305,   1.6119,   1.1538],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.8407,  11.7666,   3.0045],
        [  1.3705,  13.9723,   7.9266],
        [ -1.0012, -10.8576,   2.2867],
        [ -7.0948,   5.1940,   9.8171],
        [  1.1139,   7.6281,  -2.5680],
        [  6.0890,  -0.8382,   2.9166],
        [ -8.0965,  -5.9712,   4.3466],
        [ -6.3489,   5.9117,   6.5161],
        [  2.3373,  -4.6232,  -5.2246],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.4735,  -1.2620,  -3.6742],
        [  1.9503,  -4.7172,  -6.2313],
        [  4.1537,   5.9340,  -6.0423],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.4167, -10.1293,   6.2156],
        [ 12.1751,  -2.2592,  -5.7751],
        [  3.9591, -10.6391,  -6.2557],
        [ -5.3736,   6.6310,  -1.5174],
        [  3.2647, -10.8066,  -0.3768],
        [ -1.1885, -10.8211,   0.0865],
        [ -3.3812, -10.1235,   7.4384],
        [  3.2159, -10.8166,  -1.0406],
        [ -8.9606,  -4.0994,   1.7886],
        [  3.1271, -10.8343,   2.6442],
        [  5.5158,  -2.1337,   3.5246],
        [ -6.1914,  -8.2997,   8.5380],
        [  1.7815,  -4.7481,   4.4289],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 12.4877,  -0.7222,   8.8347],
        [ -1.0407,   7.9166,  -6.6021],
        [  8.6420,  -8.0580,   6.5781],
        [  5.1292,   4.8183,  -5.7997],
        [ -9.0604,   1.7707,   1.1066],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.8318,   3.5567,  -4.7692],
        [ -3.5173, -10.0623,   4.8280],
        [  7.1441,  11.5529,   2.3232],
        [  2.6078, -10.9229,   6.2869],
        [ -6.0195,   6.1789,  -2.9496],
        [  3.0719, -10.8449,  -4.1673],
        [  0.6952,  -4.8060,   8.8976],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -7.4602,   4.7718,  -4.6277],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.5123,  -0.5339,  -1.0728],
        [  6.2804,   2.0019,   0.8607],
        [  7.8425,  -8.7202,   6.4530],
        [  0.8572,  14.0501,  -8.3364],
        [  1.4489,   7.5256,   4.4529],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.1741,  -0.5329,   3.1913],
        [ 11.9019,  -3.1379,  -3.4534],
        [  3.1892, -10.8220,  -6.8409],
        [  6.0955,   2.8210,  -4.3321],
        [  7.5064,  -8.9651,  -0.0994],
        [  7.0515,  11.6194,   8.4289],
        [  4.4915,   5.5990,   7.0405],
        [  3.3474,  -4.2162,   5.2455],
        [ -9.4378,  -1.8606,  -2.5668],
        [  3.1480,  -4.3165,  -7.8002],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.0233,   4.9647,  -3.1788],
        [  5.3804,  -2.3504,   7.2744],
        [  0.9036,   7.6840,   9.9237],
        [  2.3261,  13.7749,   5.9121],
        [  2.0243,   7.3086,   7.5432],
        [ -1.0482, -10.8487,  -5.9066],
        [  6.0890,  -0.8382,   2.9166],
        [ -3.5520,   7.4976,   9.1522],
        [ -4.3963,   7.1629,   4.5263],
        [  5.6260,  -1.9401,   3.0259],
        [ -3.1204, -10.2345,   9.2288],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.4296,  -1.9458,   4.0833],
        [ 11.7418,   5.6070,  -1.4192],
        [ -2.8162,   7.7059,  -9.6885],
        [ 10.9916,   7.2133,  -0.9179],
        [ -6.7992,   5.4991,   9.9186],
        [ -0.5385,   7.9037,   6.3830],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.1706,   1.3863,   8.4032],
        [  6.3303,   1.6145,  -8.4083],
        [  6.1066,  -0.7799,  -4.1746],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.1890,   4.7318,   0.9515],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.1588,   4.3364,  -8.0867],
        [  3.9515,  -3.8425,   6.1729],
        [  2.7420,  -4.4896,  -3.7139],
        [  2.8474,   6.8979,   6.6457],
        [  6.3122,  -9.6994,  -6.3282]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0244, 0.0389,  ..., 0.2395, 0.4926, 0.5106],
        [0.0244, 0.0000, 0.0227,  ..., 0.2207, 0.5075, 0.5055],
        [0.0389, 0.0227, 0.0000,  ..., 0.2295, 0.4966, 0.5249],
        ...,
        [0.2395, 0.2207, 0.2295,  ..., 0.0000, 0.7259, 0.3578],
        [0.4926, 0.5075, 0.4966,  ..., 0.7259, 0.0000, 0.9829],
        [0.5106, 0.5055, 0.5249,  ..., 0.3578, 0.9829, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.79003361,  4.92497514,  5.07306274,  4.9160962 ,  4.90676242,
        5.08910491,  8.05242591,  6.53250937, 11.66453059, 13.6105595 ,
       14.0393949 , 10.9036226 ,  8.79284466,  7.70897675,  6.1463896 ,
       10.06023665,  8.67502348,  5.18047638, 11.9120321 ,  9.58072286,
        9.55720651,  5.10444097,  7.24326992,  6.02789634, 11.48663767,
       12.3828957 , 11.35182589,  8.53501777, 11.28895489, 10.88618114,
       10.67322527, 11.2845608 ,  9.85384623, 11.27656488,  5.91408504,
       10.35464916,  5.0713364 ,  9.85330646, 12.5085995 ,  7.98469166,
       11.8159254 ,  7.03733775,  9.23178204,  5.63824971,  6.83083863,
       10.65929968, 13.58333857, 11.22993218,  8.62631801, 11.27160035,
        4.85603886, 10.51437947, 12.53345445,  8.85572292,  9.43737523,
       12.12448912, 11.99633819, 12.52372629,  6.59176018, 11.72802491,
       14.07623266,  7.66377554, 10.412299  ,  6.19707944, 12.30858677,
       11.28215429,  6.7166297 , 11.69264243, 13.59170809,  7.17789666,
        5.38342077,  9.61942445,  5.34250225,  7.83486943, 12.4217523 ,
       12.32545637,  7.06272256,  5.87132718,  7.73692436, 13.96987806,
        7.58378655, 10.8992524 ,  6.1463896 ,  8.29640613,  8.40445872,
        5.95111804, 10.69963954,  9.66977802,  9.6282763 , 13.01187484,
        8.20436764, 13.1471313 ,  8.74470218,  7.92200342,  9.91668182,
        9.27474386,  6.53290663,  6.1561646 ,  8.25828409,  7.02253235,
       11.57249984, 12.9089559 ,  5.51177501,  5.26067019,  7.46248947,
       11.57249984])}
{'P': tensor([[3.2264e-02, 1.7397e-01, 1.4959e-01,  ..., 1.7207e-04, 6.1786e-05,
         5.7976e-02],
        [3.7342e-03, 3.5068e-02, 2.2657e-01,  ..., 2.5912e-04, 7.3289e-05,
         3.5311e-02],
        [3.7525e-03, 4.6587e-03, 4.4712e-02,  ..., 4.2431e-04, 1.0335e-04,
         2.4941e-02],
        ...,
        [1.1426e-06, 1.1391e-06, 1.1648e-06,  ..., 4.2828e-03, 1.3567e-05,
         1.2121e-06],
        [3.1332e-05, 3.0345e-05, 3.0938e-05,  ..., 3.3154e-04, 7.5311e-02,
         4.8097e-05],
        [1.2911e-02, 2.7062e-02, 5.2023e-02,  ..., 1.0390e-03, 4.1746e-04,
         1.0084e-01]]), 'X': tensor([[  0.5850,  -4.7982,   9.6639],
        [  1.2273,  -4.8081,   9.8129],
        [  1.6954,  -4.7616,   9.8198],
        [  2.0870,  -4.6877,   9.6823],
        [  1.9505,  -4.7171,   9.9868],
        [  2.2644,  -4.6434,   9.7508],
        [  1.5662,  13.9375,   4.5142],
        [  9.3607,   9.5337,  -4.5412],
        [ 12.1940,  -2.1881,   7.1499],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.4716,  -0.6673,   7.6068],
        [  5.9003,  12.3550,   2.1917],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.8058,   2.4850,  -1.7425],
        [  9.2692,   9.6370,   7.0746],
        [  6.3087,   1.8053,  -1.9481],
        [ 11.6544,   5.8273,  -3.1375],
        [ -8.8934,   2.2597,   3.5831],
        [  2.3928,   7.1404,  -1.7035],
        [ -7.5603,   4.6459,   8.2002],
        [ -0.9715, -10.8630,   0.3555],
        [  2.0729,   7.2878,   7.0880],
        [  9.0424,  -7.6779,  -7.5286],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.3305,   1.6119,   1.1538],
        [  6.2905,   1.9366,   2.1169],
        [  5.8979,   3.3958,  -3.8713],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.3453,  -1.5462,   7.0212],
        [  8.6420,  -8.0580,   6.5781],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 12.5154,   2.5072,  -9.3274],
        [  3.3069,   6.6089,   8.7377],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.7544,  -3.1425,  -1.5315],
        [  6.0310,  12.2795,  -6.4528],
        [ -1.5408,   7.8978,   5.3470],
        [  5.1196,  -2.7160,   4.2705],
        [ -1.5451,   7.8975,   7.6104],
        [  2.2437,  -4.6489,   6.1430],
        [  4.1348,   5.9514,  -0.8856],
        [ -4.0409,   7.3169,  -3.5779],
        [ -7.6525,  -6.6534,   5.1721],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.2036,   2.4030,  -1.2500],
        [  2.4932,   7.0903,  -5.4214],
        [  6.2770,  -0.0379,   2.4195],
        [ 10.9875,   7.2207,  -8.8549],
        [ -4.7334,   6.9982,  -0.4392],
        [ -3.3116, -10.1539,   7.5919],
        [  1.7228,  -4.7575,   8.3396],
        [  8.6420,  -8.0580,   6.5781],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.1372,   2.6726,  -6.7885],
        [ -3.1671,   7.6158,   6.0765],
        [  5.7218,  -9.9934,   6.5764],
        [  9.2692,   9.6370,   7.0746],
        [  6.2902,   0.0451,   3.7116],
        [  9.5203,   9.3483,   5.8781],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.1231,  -3.5739,   8.4708],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.6603,   2.8250,   5.9181],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.5859,  -5.0393,   0.5476],
        [  1.3103,  -4.8032,  -0.9300],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0515,  11.6194,   8.4289],
        [  0.6952,  -4.8060,   8.8976],
        [ -8.3539,  -5.5111,   0.3373],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.8836,   3.4315,  -4.0327],
        [ -8.8934,   2.2597,   3.5831],
        [ 10.5678,   7.9261,  -0.6100],
        [  4.5954,  13.0099,  -4.4439],
        [ -8.8363,   2.4086,  -5.6296],
        [  1.9312,  13.8649,   8.3563],
        [  6.0047,   3.1062,  -6.7914],
        [ 12.0772,  -2.6014,  -7.1832],
        [  0.9074,  -4.8139,   8.1674],
        [ -8.9033,   2.2329,  -8.8912],
        [ 10.9162,  -5.2599,  -7.0812],
        [  9.2979,   9.6048,   7.3138],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.1625,  -4.6697,   7.0658],
        [  5.0348,  -2.8229,   3.5436],
        [ -1.0012, -10.8576,   2.2867],
        [  3.3744,   6.5623,  -6.3636],
        [  3.7121,   6.3117,   0.6738],
        [ -9.3810,  -2.3455,  -9.5194],
        [  0.9809,  -4.8145,   6.8706],
        [ -4.7389,   6.9953,   3.7638],
        [  5.9883,   3.1536,   6.2320],
        [ -1.2454,   7.9127,  -0.5715],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.3373,  -4.6232,  -5.2246],
        [ -7.0948,   5.1940,   9.8171],
        [ -8.6801,   2.7809,  -0.8443],
        [  0.2435,   7.8191,  -6.4709],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.0534,  -6.0429,   2.4583],
        [ 10.9931,   7.2107,   7.7933],
        [  5.8979,   3.3958,  -3.8713],
        [  1.7228,  -4.7575,   8.3396]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0170, 0.0271,  ..., 0.4378, 0.7932, 0.0344],
        [0.0170, 0.0000, 0.0103,  ..., 0.4210, 0.7975, 0.0337],
        [0.0271, 0.0103, 0.0000,  ..., 0.4116, 0.7977, 0.0347],
        ...,
        [0.4378, 0.4210, 0.4116,  ..., 0.0000, 1.0440, 0.4342],
        [0.7932, 0.7975, 0.7977,  ..., 1.0440, 0.0000, 0.7639],
        [0.0344, 0.0337, 0.0347,  ..., 0.4342, 0.7639, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.83371341,  4.96231532,  5.05444288,  5.13125513,  5.10446653,
        5.16611782, 14.02526505, 13.36092624, 12.38881635,  8.09704052,
        9.41970031,  9.49511524, 13.69162795,  9.61942445,  9.14973319,
       13.37121923,  6.5618968 , 13.03002338,  9.17595899,  7.53063885,
        8.8737556 , 10.90638066,  7.57685908, 11.86239961,  5.53255882,
        6.53250937,  6.58184474,  6.80559669, 12.64235359, 12.44177641,
       11.8159254 , 10.5575601 , 12.76408562,  7.39004226, 12.9089559 ,
        5.69913817, 13.68063098,  8.04664886,  5.79543038,  8.04718433,
        5.16205087,  7.24677664,  8.35856487, 10.14044489,  8.43626114,
       10.06850677,  6.652742  ,  7.5158489 ,  6.27715105, 13.14777433,
        8.4486771 , 10.68031351,  5.05981914, 11.8159254 ,  5.5513072 ,
        6.69389431,  8.24808832, 11.51557166, 13.37121923,  6.29034958,
       13.3426541 ,  9.69629   ,  9.79814997, 10.95593729,  9.10945678,
        8.37954853,  9.95552626,  4.978704  , 13.45642379, 13.59170809,
        4.85603886, 10.00794468, 12.0495521 ,  6.81117802,  9.17595899,
       13.20988996, 13.79762415,  9.15866607, 13.99877413,  6.76056625,
       12.35421461,  4.89869205,  9.17905491, 12.11735334, 13.36800073,
        9.45995955,  5.14607014,  5.77219077, 10.9036226 ,  7.37902766,
        7.32233733,  9.66977802,  4.91336994,  8.44940661,  6.76791483,
        8.01009083, 12.84056604,  5.18047638,  8.79284466,  9.11473391,
        7.822854  , 10.1979669 , 10.06850677, 13.14690137,  6.80559669,
        5.05981914])}
{'P': tensor([[4.2708e-02, 1.7638e-01, 1.7151e-01,  ..., 2.5791e-03, 6.5249e-05,
         5.5852e-05],
        [8.0962e-03, 1.1471e-01, 6.6846e-02,  ..., 1.3222e-02, 1.4912e-04,
         1.9319e-04],
        [7.5599e-03, 1.5160e-01, 1.0285e-01,  ..., 1.1894e-02, 1.3479e-04,
         1.8591e-04],
        ...,
        [2.6082e-05, 4.6537e-05, 4.5987e-05,  ..., 2.4389e-02, 3.5521e-06,
         7.5149e-05],
        [4.9731e-03, 2.4104e-03, 2.4771e-03,  ..., 2.2884e-05, 2.1771e-01,
         2.1294e-05],
        [2.7422e-05, 3.3884e-05, 3.5809e-05,  ..., 6.9226e-05, 2.2829e-05,
         9.1953e-02]]), 'X': tensor([[  0.6952,  -4.8060,   8.8976],
        [  2.1504,  -4.6726,   8.7908],
        [  2.1120,  -4.6818,   8.6144],
        [  1.7228,  -4.7575,   8.3396],
        [  1.8447,  -4.7372,   8.1235],
        [  2.0870,  -4.6877,   9.6823],
        [  8.3791,  10.5409,   5.5590],
        [ -7.5603,   4.6459,   8.2002],
        [  1.1350,  14.0104,   0.6321],
        [ -9.4366,  -0.1189,  -2.6736],
        [  4.9152,  12.8653,  -1.4459],
        [ -7.6293,  -6.6859,  -5.5161],
        [  5.5294,  -2.1106,   9.0626],
        [  9.5203,   9.3483,   5.8781],
        [  9.3624,   9.5318,   0.2779],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.4501,  -1.7133,   2.2980],
        [ -1.9600,   7.8577,   6.9121],
        [ 11.1085,  -4.9204,   0.0736],
        [ 10.9931,   7.2107,   7.7933],
        [  5.4242, -10.1261,   3.9595],
        [  1.1350,  14.0104,   0.6321],
        [  1.8721,  -4.7322,   0.5524],
        [  1.7301,   7.4262,   5.0522],
        [  2.1625,  -4.6697,   7.0658],
        [  2.3889,  -4.6082,  -8.7291],
        [  0.9945,  -4.8144,  -0.1897],
        [ -2.8162,   7.7059,  -9.6885],
        [ -1.1885, -10.8211,   0.0865],
        [ -8.7544,   2.6096,   3.3146],
        [  6.0073,   3.0986,   6.7937],
        [ 11.1469,   6.9237,   1.6160],
        [  2.4149,  -4.6005,  -4.3339],
        [ 11.5083,  -4.1177,   0.2389],
        [  1.6308,  13.9254,  -2.0984],
        [ -4.7847,  -9.3729,  -3.7741],
        [  8.3718,  -8.2953,   8.4666],
        [  1.4749,  -4.7892,  -0.8089],
        [ -0.2586,   7.8825,  -9.9840],
        [ 12.5224,  -0.4497,   4.8448],
        [  1.8721,  -4.7322,   0.5524],
        [  8.7131,  10.2216,   7.7915],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.3583,   1.1729,   8.7835],
        [  6.1579,  -0.5961,  -6.5595],
        [  6.0788,  -0.8710,  -6.8719],
        [ -5.3179,  -9.0101,   9.4472],
        [  1.9312,  13.8649,   8.3563],
        [ -5.1369,   6.7755,   8.3836],
        [ -8.5900,   2.9767,   7.4811],
        [ -1.9646,   7.8571,   3.3817],
        [  9.0051,   9.9234,   5.4411],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.6095,  -1.9702,  -7.8233],
        [  0.9036,   7.6840,   9.9237],
        [ -8.8540,  -4.3971,  -5.2144],
        [  9.0051,   9.9234,   5.4411],
        [ -4.4520,   7.1370,  -8.7867],
        [  2.2076,  13.8032,  -1.5099],
        [ -6.5295,  -7.9785,  -7.4429],
        [  6.1737,   2.5305,   3.9049],
        [  6.1741,  -0.5329,   3.1913],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.0310,  12.2795,  -6.4528],
        [  2.8890,  -4.4316,   1.1600],
        [  0.1287, -11.0065,   6.5757],
        [  5.5424,  -2.0885,   7.2487],
        [ -4.0868,   7.2981,   8.1874],
        [ -5.6423,   6.4537,  -1.2091],
        [ -8.0611,  -6.0302,   4.6962],
        [ 10.2712,   8.3707,  -4.9743],
        [  5.4426, -10.1181,  -5.1268],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.0012, -10.8576,   2.2867],
        [  5.0506,   4.9277,  -9.5094],
        [  4.5767,   5.5071,   3.9069],
        [  0.1460,   7.8340,  -9.4614],
        [  2.3373,  -4.6232,  -5.2246],
        [  8.0381,  -8.5690,  -8.9523],
        [ -9.4524,  -1.6813,  -6.6862],
        [ 11.1085,  -4.9204,   0.0736],
        [ 10.6283,  -5.7253,   7.4878],
        [ 12.1504,   4.3674,  -7.2998],
        [ -3.4100,   7.5436,  -7.6980],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.3413,   0.4883,  -6.8416],
        [  7.4132,  11.3526,   0.3322],
        [ -6.3489,   5.9117,   6.5161],
        [  2.9461,  -4.4077,  -0.4831],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.3804,  -2.3504,   7.2744],
        [  5.7218,  -9.9934,   6.5764],
        [ -0.8675, -10.8815,   6.8561],
        [ 12.4877,  -0.7222,   8.8347],
        [ -2.8162,   7.7059,  -9.6885],
        [ 12.4996,   2.6335,  -6.3166],
        [ -4.6393,   7.0461,   1.1624],
        [  6.0047,   3.1062,  -6.7914],
        [  4.0457,  -3.7737,  -0.4704],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.5900,   2.9767,   7.4811],
        [  6.3413,   0.4883,  -6.8416]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0351, 0.0339,  ..., 0.1420, 0.6568, 0.8989],
        [0.0351, 0.0000, 0.0032,  ..., 0.1422, 0.6917, 0.8920],
        [0.0339, 0.0032, 0.0000,  ..., 0.1450, 0.6901, 0.8894],
        ...,
        [0.1420, 0.1422, 0.1450,  ..., 0.0000, 0.7026, 1.0342],
        [0.6568, 0.6917, 0.6901,  ..., 0.7026, 0.0000, 1.1677],
        [0.8989, 0.8920, 0.8894,  ..., 1.0342, 1.1677, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.85603886,  5.14371005,  5.13616224,  5.05981914,  5.08372149,
        5.13125513, 13.46553623,  8.8737556 , 14.05633441,  9.43737523,
       13.77222752, 10.14436912,  5.91854397, 13.3426541 , 13.3607379 ,
        8.36995953,  9.60412735,  8.09842704, 12.1494023 , 13.14690137,
       11.48734978, 14.05633441,  5.08910491,  7.62509582,  5.14607014,
        5.19065439,  4.9160962 ,  8.20436764, 10.88618114,  9.13507321,
        6.7593917 , 13.12218013,  5.19577146, 12.22276278, 14.02058621,
       10.52357479, 11.785566  ,  5.01114719,  7.88677538, 12.53047489,
        5.08910491, 13.43127131,  9.75401449,  6.46560181,  6.18668463,
        6.14086582, 10.46238054, 13.99877413,  8.50268322,  9.09119351,
        8.09899874, 13.40024224,  8.8353674 ,  5.94541803,  7.73692436,
        9.88571713, 13.40024224,  8.41171369, 13.97857591, 10.30972193,
        6.67218687,  6.19707944, 13.02818702,  9.70108409, 13.68063098,
        5.29009836, 11.00726483,  5.92281142,  8.36445758,  8.5724042 ,
       10.06703358, 13.25017211, 11.4890861 ,  6.61211224, 10.9036226 ,
        7.05627461,  7.16058587,  7.83534987,  5.18047638, 11.74901778,
        9.60080639, 12.1494023 , 12.07225469, 12.91143491,  8.27853076,
       10.74409834,  7.09887557,  9.3109884 ,  9.69629   ,  6.36004393,
       13.55868684,  8.67502348,  5.30158497, 10.53368644,  5.87132718,
       11.51557166, 10.91601655, 12.5085995 ,  8.20436764, 12.77402351,
        8.43626114,  6.76056625,  5.53255882, 12.30695098,  9.09119351,
        6.36004393])}
{'P': tensor([[3.7416e-02, 1.8518e-01, 1.2699e-01,  ..., 4.9646e-04, 9.8419e-05,
         5.7941e-03],
        [4.4448e-03, 4.8060e-02, 2.5580e-01,  ..., 3.3077e-04, 1.4985e-04,
         7.6435e-03],
        [5.4304e-03, 7.4079e-03, 9.4483e-02,  ..., 3.4055e-04, 3.9474e-04,
         1.2240e-02],
        ...,
        [9.3747e-03, 1.0535e-02, 9.7082e-03,  ..., 3.8515e-02, 1.3044e-05,
         1.6315e-02],
        [1.3613e-04, 1.2243e-04, 1.2228e-04,  ..., 2.3720e-05, 1.9966e-01,
         1.6078e-05],
        [2.4429e-04, 4.4265e-04, 9.8609e-04,  ..., 5.8476e-06, 1.8395e-04,
         4.2799e-02]]), 'X': tensor([[  0.7145,  -4.8071,  -2.3766],
        [  1.6232,  -4.7717,  -2.5655],
        [  2.6321,  -4.5296,  -2.5159],
        [  2.2373,  -4.6506,  -1.7334],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.7420,  -4.4896,  -3.7139],
        [ -5.9938,  -8.4745,   3.9735],
        [  2.7558,  -4.4844,   0.0636],
        [  2.4356,  -4.5942,  -5.0808],
        [  6.9433,  -9.3362,  -0.3421],
        [ -3.6381,   7.4683,   7.3544],
        [  6.3205,   1.7068,  -3.8619],
        [  5.3804,  -2.3504,   7.2744],
        [  1.5246,  -4.7838,  -5.2095],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.6229,   2.9068,   2.4429],
        [ 12.5123,  -0.5339,  -1.0728],
        [  4.5325,  -3.3643,   6.7699],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.9788,  -4.8145,  -3.8661],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 10.3251,   8.2928,  -3.0006],
        [  0.1460,   7.8340,  -9.4614],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.9526,   3.2525,  -4.3821],
        [  2.2437,  -4.6489,   6.1430],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.5250, -11.0306,   5.5358],
        [ -7.4887,   4.7364,   8.9074],
        [ -6.7992,   5.4991,   9.9186],
        [  6.2022,   2.4092,   3.0349],
        [ -1.9600,   7.8577,   6.9121],
        [  3.6469,   6.3624,  -1.8414],
        [  2.0039,   7.3173,  -1.5885],
        [  9.3424,  -7.3685,  -3.6807],
        [  9.0004,  -7.7195,   4.6945],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.0330,  -6.0765,   6.7945],
        [  9.6662,   9.1725,   8.1056],
        [ 12.4996,   2.6335,  -6.3166],
        [ -1.8788, -10.6567,   3.7068],
        [ 12.5123,  -0.5339,  -1.0728],
        [  0.7275,  14.0667,  -0.9815],
        [  0.9809,  -4.8145,   6.8706],
        [  6.3589,   0.8239,  -8.4230],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.0277,   4.9589,   7.0175],
        [  9.6816,   9.1536,  -5.8543],
        [  1.3339,   7.5627,  -0.9034],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.2535,  -0.1709,  -9.4102],
        [  4.1637,  13.1898,  -1.9189],
        [  6.3550,   0.7129,   0.4917],
        [  6.3515,   0.6418,  -6.5823],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.4773,  -1.0024,   5.2599],
        [  3.1480,  -4.3165,  -7.8002],
        [ -1.0012, -10.8576,   2.2867],
        [ -4.3963,   7.1629,   4.5263],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.2939,  -2.4784,  -4.5840],
        [  9.3836,   9.5075,  -5.9278],
        [  6.2242,  -0.3171,   7.4978],
        [  5.8318,   3.5567,  -4.7692],
        [  4.0078,  -3.8018,  -6.2168],
        [ -8.1531,  -5.8748,   2.6934],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.3411,  -2.6004,  -2.7062],
        [  7.2340,  -9.1505,  -5.8321],
        [  5.2939,  -2.4784,  -4.5840],
        [  4.9152,  12.8653,  -1.4459],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.2974,   0.8371,   2.4852],
        [ -0.4341, -10.9473,   7.4509],
        [ 12.6035,   0.7494,   7.1536],
        [  5.0506,   4.9277,  -9.5094],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.0608,   3.9362,  -5.9567],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.9587,  12.3216,   9.1695],
        [  1.7228,  -4.7575,   8.3396],
        [  4.4290,  -3.4596,   1.4181],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -8.8154,   2.4611,   2.5198],
        [ 11.5935,   5.9740,  -4.3952],
        [  4.5503,   5.5359,   7.4735],
        [ -2.0840,   7.8415,  -7.8657],
        [  7.3592,  11.3937,   6.2407],
        [  5.9587,  12.3216,   9.1695],
        [  6.2036,   2.4030,  -1.2500],
        [ 11.7349,  -3.5838,   2.1644],
        [  9.8082,  -6.8396,   4.9615],
        [  1.6422,   7.4586,  -6.8056],
        [ -8.1204,  -5.9308,   4.0761],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.0788,  -0.8710,  -6.8719],
        [ -3.3999, -10.1152,  -1.2088],
        [  4.4039,   5.6903,   8.7366],
        [  8.0877,  -8.5297,  -1.1050]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0271, 0.0482,  ..., 0.6975, 0.7043, 0.1556],
        [0.0271, 0.0000, 0.0232,  ..., 0.7226, 0.7158, 0.1826],
        [0.0482, 0.0232, 0.0000,  ..., 0.7453, 0.7134, 0.2032],
        ...,
        [0.6975, 0.7226, 0.7453,  ..., 0.0000, 0.9746, 0.5508],
        [0.7043, 0.7158, 0.7134,  ..., 0.9746, 0.0000, 0.6695],
        [0.1556, 0.1826, 0.2032,  ..., 0.5508, 0.6695, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.85995002,  5.04028538,  5.23878371,  5.16077764,  5.05963217,
        5.26067019, 10.37998078,  5.26342203,  5.19986797, 11.63502679,
        8.30729352,  6.54693987,  5.87132718,  5.0209005 ,  9.17595899,
        9.09963894, 12.52372629,  5.64464807,  8.83341608,  4.91296205,
        9.5642993 ,  9.85330646, 13.2430377 ,  7.83534987, 10.14436912,
        9.99211663,  6.78325151,  5.16205087, 10.00410625, 11.04313641,
        8.86082124,  8.74470218,  6.65368551,  8.09842704,  7.33350642,
        7.58668144, 11.89855267, 11.85743425,  8.04718433, 10.07238463,
       13.32557133, 12.77402351, 10.82106505, 12.52372629, 14.08549802,
        4.91336994,  6.41202986, 11.04629471,  9.5642993 ,  8.92395381,
        7.06169391, 13.32374056,  7.67939546,  9.49511524,  6.25586297,
       13.83139024,  6.3949043 ,  6.383897  , 10.67132524,  5.21280156,
        9.53015376,  5.34250225, 10.9036226 ,  8.40445872, 10.30972193,
        8.85572292,  5.84533345, 13.35832725,  6.2322769 ,  6.83083863,
        5.52416368, 10.04916678,  8.22728399,  9.69629   , 11.66453059,
        5.84533345, 13.77222752,  6.1463896 ,  9.3349858 , 10.95593729,
       12.62576026,  7.05627461,  4.9160962 ,  8.97052115, 13.04217487,
       13.68672855,  5.05981914,  5.62005629, 12.07818871,  9.15253344,
       13.04217487,  7.16598273,  8.1137452 , 13.56367857, 13.68672855,
        6.652742  , 12.26997165, 11.95743558,  7.63726134, 10.05558893,
       13.77075174,  9.51074763,  6.14086582, 10.67132524,  7.19534146,
       11.75438006])}
{'P': tensor([[3.4966e-02, 1.5373e-01, 1.4425e-01,  ..., 3.2408e-05, 1.2777e-05,
         2.5514e-03],
        [4.7490e-03, 4.5671e-02, 4.2208e-02,  ..., 6.3268e-05, 2.4212e-05,
         5.1766e-03],
        [4.2504e-03, 2.3317e-02, 4.2325e-02,  ..., 5.4981e-05, 1.9910e-05,
         4.0952e-03],
        ...,
        [3.8809e-06, 3.4886e-06, 3.1773e-06,  ..., 2.3155e-02, 9.8034e-02,
         2.0215e-05],
        [1.2497e-06, 1.1095e-06, 9.5944e-07,  ..., 8.9816e-05, 1.1406e-02,
         4.6876e-06],
        [3.0059e-04, 3.3210e-04, 3.0845e-04,  ..., 1.1806e-02, 4.5885e-03,
         6.7579e-02]]), 'X': tensor([[  0.7685,  -4.8098,  -2.9310],
        [  1.6232,  -4.7717,  -2.5655],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.6321,  -4.5296,  -2.5159],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.7420,  -4.4896,  -3.7139],
        [ -9.0574,   1.7801,   8.8986],
        [  6.0140,  -1.0661,  -2.5040],
        [  0.2480, -11.0153,  -3.0183],
        [  2.2076,  13.8032,  -1.5099],
        [  6.9284,  11.7061,  -0.3732],
        [ 12.5650,   2.0132,   4.6272],
        [ -3.9508,  -9.8515,   1.7923],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.2096,   5.8815,   1.9230],
        [  6.2630,   2.1057,  -5.2136],
        [ 12.2435,   4.0076,   8.1257],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.8934,   2.2597,   3.5831],
        [ 11.9808,   4.9340,  -3.3715],
        [  5.4336,  -2.2676,   5.6933],
        [  6.2038,   2.4020,  -3.1692],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.0467,  -9.8369,  -6.5816],
        [  3.6154,  -4.0640,  -7.9698],
        [ -3.2053, -10.1993,  -3.8736],
        [  1.1994,   7.6035,   2.4295],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -8.5149,  -5.1905,   1.5239],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.7485,  -9.9811,  -3.2355],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -9.4505,  -1.7081,  -0.7039],
        [  2.0729,   7.2878,   7.0880],
        [ -5.1974,   6.7396,   3.3484],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.2076,  13.8032,  -1.5099],
        [  4.2612, -10.5514,  -6.6268],
        [ -5.3736,   6.6310,  -1.5174],
        [ -7.8359,   4.2735,   7.2943],
        [ -9.4196,   0.0478,   9.4140],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.5375,   7.0676,  -4.3490],
        [  1.4306,  13.9620,  -0.2075],
        [  1.0000,   7.6591,  -4.5011],
        [  4.7156,  -3.1831,  -8.0224],
        [  6.2770,  -0.0379,   2.4195],
        [  4.7739,  -3.1218,  -4.1188],
        [  2.5318,  13.7231,   6.3692],
        [ 10.6673,  -5.6649,  -4.1646],
        [  1.8320,   7.3871,   5.1452],
        [  4.7739,  -3.1218,  -4.1188],
        [ -4.0868,   7.2981,   8.1874],
        [ -9.4766,  -1.1116,  -2.1706],
        [  0.1206,  -4.7363,   8.7103],
        [ -6.3489,   5.9117,   6.5161],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.3928,   7.1404,  -1.7035],
        [  9.7328,   9.0902,  -0.8917],
        [  6.2902,   0.0451,   3.7116],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.2366,  -4.2732,   5.1532],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.8407,  11.7666,   3.0045],
        [ -7.4807,   4.7463,   2.0616],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.8363,   2.4086,  -5.6296],
        [ 11.4235,   6.3594,   5.1663],
        [  0.9074,  -4.8139,   8.1674],
        [  1.7903,  -4.7467,   0.1168],
        [ 12.2218,   4.0953,  -8.2072],
        [  2.6310, -10.9195,   3.8054],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.6260,  -1.9401,   3.0259],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.3943,   6.5483,   1.0790],
        [ 11.5568,   6.0602,   9.3244],
        [ -8.7409,  -4.6837,  -6.5840],
        [  5.4179,  -2.2925,  -3.4641],
        [ -5.1377,   6.7751,  -0.4438],
        [  5.1667,  -2.6543,   1.2879],
        [  6.0788,  -0.8710,  -6.8719],
        [ -3.5029,   7.5138,   8.1853],
        [  4.8432,   5.1968,   1.5006],
        [  6.3200,   0.2690,   1.5900],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.2939,  -2.4784,  -4.5840],
        [  9.1409,  -7.5789,   1.5814],
        [ 12.3357,   3.6001,   0.7903],
        [  2.9410,   6.8428,   5.4458],
        [ -3.6381,   7.4683,   7.3544],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -8.9033,   2.2329,  -8.8912],
        [  4.2096,   5.8815,   1.9230],
        [  5.4429,  -2.2529,   9.7306],
        [  6.3604,   0.9042,  -6.5901],
        [ 12.5919,   0.4035,  -5.8460],
        [  9.2692,   9.6370,   7.0746],
        [ -8.6801,   2.7809,  -0.8443],
        [ 10.8218,  -5.4177,   6.8573],
        [  5.9003,  12.3550,   2.1917],
        [  3.9022,  13.2906,   5.5788],
        [  6.2804,   2.0019,   0.8607]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9251e-02, 3.6641e-02,  ..., 6.1047e-01, 9.4942e-01,
         2.9711e-01],
        [2.9251e-02, 0.0000e+00, 4.6217e-02,  ..., 5.8139e-01, 9.2160e-01,
         2.7002e-01],
        [3.6641e-02, 4.6217e-02, 0.0000e+00,  ..., 6.1436e-01, 9.6021e-01,
         3.1121e-01],
        ...,
        [6.1047e-01, 5.8139e-01, 6.1436e-01,  ..., 0.0000e+00, 3.7275e-01,
         3.3256e-01],
        [9.4942e-01, 9.2160e-01, 9.6021e-01,  ..., 3.7275e-01, 0.0000e+00,
         6.5304e-01],
        [2.9711e-01, 2.7002e-01, 3.1121e-01,  ..., 3.3256e-01, 6.5304e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.87083348,  5.04028538,  5.05963217,  5.23878371,  5.05774295,
        5.26067019,  9.23071698,  6.10774544, 11.018084  , 13.97857591,
       13.60275211, 12.72524615, 10.61417181,  5.68942572,  7.23276582,
        6.60752373, 12.88270018,  6.07992623,  5.05989547,  9.17595899,
       12.95702647,  5.88782913,  6.65259775,  5.39450305, 11.54672755,
        5.43943791, 10.6910768 ,  7.69752262,  9.23178204, 12.35421461,
        9.97220799,  9.62238828, 11.51811603,  9.85330646,  9.60359158,
        7.57685908,  8.51088913,  9.48969978, 13.97857591, 11.37939695,
        8.53501777,  8.92549752,  9.41970031,  8.30125201,  7.5092834 ,
       14.03505624,  7.72415048,  5.68942572,  6.27715105,  5.70405525,
       13.95472331, 12.07818871,  7.61088125,  5.70405525,  8.36445758,
        9.54154096,  4.73784222,  8.67502348, 11.9182956 ,  7.53063885,
       13.31764181,  6.29034958,  5.9510489 ,  5.36061974,  5.18047638,
       13.6105595 ,  8.85939626,  9.85330646,  9.15866607, 13.0743399 ,
        4.89869205,  5.07306274, 12.889695  , 11.23201619,  9.17905491,
        5.95111804,  8.49161154,  7.37576154, 13.0493368 ,  9.91668182,
        5.8828986 ,  8.50278896,  5.80860165,  6.14086582,  8.29022197,
        7.10378919,  6.32572323, 12.07225469,  5.84533345, 11.87412439,
       12.85032836,  7.44805901,  8.30729352,  8.41148722, 12.18045978,
        9.17905491,  7.23276582,  5.89074104,  6.42439344, 12.59840193,
       13.37121923,  9.11473391, 12.10222204, 13.69162795, 13.85158417,
        6.59176018])}
{'P': tensor([[4.0719e-02, 1.4612e-01, 1.4179e-01,  ..., 2.5653e-05, 4.0719e-02,
         2.5221e-02],
        [3.2787e-03, 5.4186e-02, 1.4797e-02,  ..., 1.7532e-05, 3.2787e-03,
         1.1918e-02],
        [3.8973e-03, 1.4807e-01, 5.5937e-02,  ..., 1.7340e-05, 3.8973e-03,
         1.2390e-02],
        ...,
        [1.4451e-05, 1.8769e-05, 1.5938e-05,  ..., 1.3389e-02, 1.4451e-05,
         6.9013e-05],
        [4.0719e-02, 1.4612e-01, 1.4179e-01,  ..., 2.5653e-05, 4.0719e-02,
         2.5221e-02],
        [6.7276e-03, 2.2247e-02, 1.6910e-02,  ..., 7.3763e-05, 6.7276e-03,
         7.1488e-02]]), 'X': tensor([[  0.7958,  -4.8109,  -8.2248],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.4920,  -4.1366,  -8.3955],
        [  9.4516,  -7.2501,  -1.8868],
        [ -5.2752,  -9.0409,  -8.2568],
        [  1.8721,  -4.7322,   0.5524],
        [ -9.1178,   1.5784,   2.1410],
        [ -8.9788,   2.0208,  -3.2867],
        [ -5.1398,  -9.1367,   6.2074],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.3800,  -4.3919,   1.4066],
        [ -8.8934,   2.2597,   3.5831],
        [  1.3670,  -4.7990,   3.2603],
        [  0.5850,  -4.7982,   9.6639],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.3118,   6.5957,  -4.5667],
        [  1.4150,  13.9647,  -4.0915],
        [  6.0047,   3.1062,  -6.7914],
        [ -1.9714,   7.8563,  -1.8084],
        [ -1.6906,   7.8860,   7.7142],
        [ 12.0582,  -2.6640,   0.7132],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.0766,  -6.0046,  -3.1873],
        [  0.1047, -11.0046,   0.5188],
        [ -0.8569,   7.9156,  -6.5557],
        [ -8.8363,   2.4086,  -5.6296],
        [ -5.8389,  -8.6056,  -3.5098],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.4283,  -1.9585,  -0.9129],
        [  0.5600, -11.0321,  -0.8660],
        [  0.2571,  -4.7594,   1.5618],
        [  7.8375,  -8.7240,  -9.8913],
        [  3.1271, -10.8343,   2.6442],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.6248,   3.9984,   8.4933],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.6022,  -1.9833,   8.3081],
        [ -8.0829,  -5.9940,   5.0564],
        [  2.2659,  13.7894,   6.5436],
        [  5.3924, -10.1396,  -4.0863],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.6414, -10.0303,   7.8525],
        [  4.4915,   5.5990,   7.0405],
        [  0.4996,   7.7738,  -4.4711],
        [  2.6310, -10.9195,   3.8054],
        [  3.9663,   6.1018,  -5.4104],
        [  5.7845,   3.6652,   0.9175],
        [ -6.2668,   5.9808,   2.5614],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.1231,  -3.5739,   8.4708],
        [  5.1861,   4.7361,  -3.7312],
        [  0.9788,  -4.8145,  -3.8661],
        [  9.2692,   9.6370,   7.0746],
        [  6.2576,   2.1359,  -3.3398],
        [  1.7122,  -4.7591,  -3.6734],
        [  0.6142,  14.0802,   0.5407],
        [ 10.4758,  -5.9539,   1.3667],
        [  2.3200,   7.1756,   1.1938],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.0031,  12.2958,  -6.5992],
        [  3.2226,  -4.2802,  -4.0428],
        [  4.9531,   5.0576,   5.6794],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.1203,  -2.7151,   6.2110],
        [ -7.0948,   5.1940,   9.8171],
        [  0.9478,  -4.8144,   0.6371],
        [  5.2608,  12.6977,   4.9166],
        [  0.1252, -11.0062,   7.2564],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.6319,  11.9066,   3.5165],
        [  7.7750,  11.0665,   2.8364],
        [ -1.6275, -10.7221,   3.2376],
        [  3.1480,  -4.3165,  -7.8002],
        [ -1.9600,   7.8577,   6.9121],
        [ -5.2752,  -9.0409,  -8.2568],
        [  2.3490,  -4.6199,  -4.2489],
        [ -7.0948,   5.1940,   9.8171],
        [  0.1931,  -4.7491,   7.4385],
        [  5.6095,  -1.9702,  -7.8233],
        [ -8.6426,  -4.9136,   9.4941],
        [ -5.6423,   6.4537,  -1.2091],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3200,   0.2690,   1.5900],
        [  2.0870,  -4.6877,   9.6823],
        [  6.2883,   0.0325,   5.2692],
        [ 12.3453,  -1.5462,   7.0212],
        [  9.2811,  -7.4336,   5.1771],
        [  5.6523,   3.9443,  -2.9449],
        [  2.5813,  -4.5471,   5.3597],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.3246,   1.6696,  -0.1544],
        [  6.0031,  12.2958,  -6.5992],
        [ 12.4996,   2.6335,  -6.3166],
        [  6.9693,  -9.3200,  -8.3822],
        [ -9.0604,   1.7707,   1.1066],
        [  2.8602,   6.8905,   5.8779],
        [ -0.4341, -10.9473,   7.4509],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.3373,  -4.6232,  -5.2246]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0425, 0.0418,  ..., 1.1794, 0.0000, 0.0903],
        [0.0425, 0.0000, 0.0047,  ..., 1.2121, 0.0425, 0.1049],
        [0.0418, 0.0047, 0.0000,  ..., 1.2143, 0.0418, 0.1084],
        ...,
        [1.1794, 1.2121, 1.2143,  ..., 0.0000, 1.1794, 1.1138],
        [0.0000, 0.0425, 0.0418,  ..., 1.1794, 0.0000, 0.0903],
        [0.0903, 0.1049, 0.1084,  ..., 1.1138, 0.0903, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.8763135 ,  5.22903117,  5.19065439,  5.13182506,  5.34250225,
        5.41348121, 11.9120321 , 10.46739073,  5.08910491,  9.25336371,
        9.20340031, 10.48315436,  5.1224079 , 12.19805394,  9.17595899,
        4.98989906,  4.83371341,  9.85330646, 13.09426142, 14.03618649,
        6.76056625,  8.09984231,  8.06517143, 12.3489324 ,  5.35774573,
       10.06407839, 11.00508681,  7.96181288,  9.15866607, 10.39942497,
       11.97238624,  9.62958973, 11.04629471,  4.7663526 , 11.72749923,
       11.27656488, 12.17901033,  6.90115901,  9.69629   ,  5.94293185,
       10.06286074, 13.97430191, 11.48434767, 12.84056604, 11.50791859,
        7.17789666,  7.78980299, 11.23201619,  7.27759233,  6.84796202,
        8.66273171,  9.11473391, 13.59673406,  9.79814997,  7.02326515,
        4.91296205, 13.37121923,  6.61211224,  5.05774295, 14.09357547,
       12.0495521 ,  7.54127425, 12.4013153 , 12.30695098,  9.86089579,
       13.6829843 ,  5.35774573,  7.07902036,  5.70405525,  5.79562691,
        8.79284466,  4.90676242, 13.744387  , 11.00694688, 12.7125079 ,
       13.62896593, 13.5247215 , 10.84493732,  5.34250225,  8.09842704,
       10.46739073,  5.18277858,  8.79284466,  4.75303127,  5.94541803,
        9.94173708,  8.5724042 , 10.06850677,  7.60529892,  6.32572323,
        5.13125513,  6.28835101, 12.44177641, 11.89105259,  6.89245352,
        5.2287058 ,  9.69629   ,  6.54127431, 13.6829843 , 12.77402351,
       11.63765273,  9.23178204,  7.46053281, 10.95593729,  4.8763135 ,
        5.18047638])}
{'P': tensor([[4.7225e-02, 1.9552e-01, 1.5269e-01,  ..., 1.2493e-03, 6.7049e-04,
         5.5476e-04],
        [4.5859e-03, 5.5897e-02, 9.5018e-02,  ..., 2.3746e-03, 1.5458e-03,
         8.5573e-04],
        [5.7941e-03, 2.0394e-02, 8.7621e-02,  ..., 2.7771e-03, 3.2401e-03,
         8.7399e-04],
        ...,
        [3.2664e-04, 4.2917e-04, 3.7481e-04,  ..., 1.2413e-01, 2.3264e-03,
         3.6769e-03],
        [6.0273e-06, 7.2048e-06, 8.0965e-06,  ..., 8.0558e-05, 1.6111e-02,
         7.7813e-06],
        [9.9378e-05, 1.4693e-04, 1.1881e-04,  ..., 4.2176e-02, 3.3590e-04,
         4.4381e-02]]), 'X': tensor([[ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0243, 0.0470,  ..., 0.4308, 0.3900, 0.6335],
        [0.0243, 0.0000, 0.0301,  ..., 0.4233, 0.3714, 0.6276],
        [0.0470, 0.0301, 0.0000,  ..., 0.4449, 0.3432, 0.6503],
        ...,
        [0.4308, 0.4233, 0.4449,  ..., 0.0000, 0.6700, 0.2069],
        [0.3900, 0.3714, 0.3432,  ..., 0.6700, 0.0000, 0.8680],
        [0.6335, 0.6276, 0.6503,  ..., 0.2069, 0.8680, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.89121805,  5.13182506,  5.22903117,  5.34250225,  5.19065439,
        5.37343173, 12.04924159, 10.14044489, 13.52430176,  4.98989906,
       11.9182956 ,  5.18047638, 10.00410625, 13.1399181 , 11.11038551,
        6.97106912,  9.26455795,  6.28835101, 10.8992524 , 10.57302546,
       10.65929968,  5.14999799,  8.89458273, 10.06286074,  6.27715105,
       10.67322527, 13.6922425 ,  6.49538841,  9.5642993 ,  9.76896904,
        5.0209005 ,  8.8353674 ,  9.94173708, 11.28895489,  7.50917551,
        5.05963217, 11.27160035,  7.46248947, 11.56146873, 11.31276109,
        7.48538905,  8.04718433,  5.70591986,  8.08320235,  9.88310614,
        8.91551231, 10.61047596, 11.75618513, 13.95472331,  6.12886024,
       10.66631784, 11.83570132, 13.77222752,  7.02496223,  5.82345867,
        8.1137452 , 10.01804223,  6.81961996,  5.43544626,  5.95297459,
       11.89855267,  4.89869205,  7.19534146, 12.53047489, 10.52587243,
        9.13507321,  9.95552626,  9.6282763 , 11.05317597,  9.51074763,
        8.85572292, 11.00812996, 13.20597932, 12.84056604,  6.07992623,
       13.43127131, 10.7506423 ,  4.74468406, 11.74901778, 10.52587243,
       10.61472636,  5.82345867, 11.63383221,  7.06272256,  6.05113217,
        9.562174  ,  9.26455795,  7.62509582, 10.65929968,  6.05165248,
       13.85158417, 13.14690137,  4.91038884, 11.05672136, 11.8159254 ,
        5.2287058 ,  7.15990676, 13.54774355,  9.69792513, 11.86239961,
       10.14436912,  6.19707944,  6.67121748,  6.3949043 , 12.889695  ,
        5.91408504])}
{'P': tensor([[3.0614e-02, 1.5962e-01, 1.5961e-01,  ..., 4.9599e-06, 1.3402e-05,
         1.1477e-04],
        [4.2125e-03, 4.4352e-02, 1.1289e-01,  ..., 6.4292e-06, 2.5295e-05,
         1.9053e-04],
        [5.4162e-03, 1.9866e-02, 6.2348e-02,  ..., 9.5461e-06, 3.9801e-05,
         2.7619e-04],
        ...,
        [6.8482e-05, 5.0180e-05, 5.1924e-05,  ..., 8.7937e-02, 2.8789e-06,
         3.3573e-05],
        [2.9059e-06, 2.7481e-06, 2.8876e-06,  ..., 3.4482e-04, 5.7406e-02,
         5.6249e-04],
        [4.9410e-04, 3.7303e-04, 3.6098e-04,  ..., 8.6823e-04, 9.1768e-06,
         1.0731e-01]]), 'X': tensor([[  0.9074,  -4.8139,   8.1674],
        [  1.7228,  -4.7575,   8.3396],
        [  1.8447,  -4.7372,   8.1235],
        [  2.1120,  -4.6818,   8.6144],
        [  2.1504,  -4.6726,   8.7908],
        [  2.9260,  -4.4162,   8.3808],
        [  0.6363, -11.0348,   6.1414],
        [ -7.3470,  -7.0615,   5.0066],
        [ -3.1671,   7.6158,   6.0765],
        [  3.7298,  13.3537,  -9.4653],
        [  5.7845,   3.6652,   0.9175],
        [  4.9452, -10.3187,  -7.8621],
        [  6.2790,  -0.0259,  -5.4116],
        [ -9.3464,   0.5708,  -1.6397],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.1588,   4.3364,  -8.0867],
        [ -0.5385,   7.9037,   6.3830],
        [ -1.9600,   7.8577,   6.9121],
        [  1.3103,  -4.8032,  -0.9300],
        [ 12.5832,   1.7543,  -7.6134],
        [ -9.4773,  -1.0024,   5.2599],
        [  9.1820,   9.7334,  -2.7932],
        [ -1.3178, -10.7939,   0.4236],
        [ -8.4824,  -5.2575,  -0.0795],
        [  1.1139,   7.6281,  -2.5680],
        [  0.1041, -11.0045,   2.4311],
        [  0.1252, -11.0062,   7.2564],
        [ -3.1204, -10.2345,   9.2288],
        [ -5.6169,  -8.7844,   8.9057],
        [ -9.1624,  -3.4286,   2.0647],
        [  3.3633,  13.4794,   0.5193],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.5859,  -5.0393,   0.5476],
        [  3.7916,  -3.9524,  -7.9547],
        [ 11.7726,  -3.4876,  -6.7402],
        [  6.3031,   1.8477,  -7.1706],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.3403,  -4.8011,  -1.1698],
        [  6.1222,  -0.7261,   8.8204],
        [  4.0642,  -3.7599,  -3.4565],
        [ 12.2589,  -1.9312,   9.8657],
        [ -1.7009,   7.8851,  -8.2437],
        [ -6.6998,   5.5954,   0.0823],
        [ -0.5385,   7.9037,   6.3830],
        [ 12.3453,  -1.5462,   7.0212],
        [  3.3474,  -4.2162,   5.2455],
        [ 12.6038,   0.7665,  -3.4439],
        [  1.0391,  -4.8141,  -0.0768],
        [  1.8766,   7.3694,   1.1887],
        [ -9.4645,  -1.4880,  -2.0226],
        [  1.1994,   7.6035,   2.4295],
        [ -1.6906,   7.8860,   7.7142],
        [ -4.2040,   7.2486,   2.9875],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.4150,  13.9647,  -4.0915],
        [  2.4932,   7.0903,  -5.4214],
        [ -9.0574,   1.7801,   8.8986],
        [ 11.5198,  -4.0923,   5.5088],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 12.4691,   2.8525,  -5.5110],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.5900,   2.9767,   7.4811],
        [ 12.4877,  -0.7222,   8.8347],
        [ -1.5875,   7.8944,  -9.4286],
        [  8.5093,  -8.1764,  -3.2381],
        [ -7.6719,  -6.6260,   7.1468],
        [  0.3038,   7.8092,   0.4164],
        [ 11.6165,  -3.8715,   6.4594],
        [  5.3804,  -2.3504,   7.2744],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -8.8400,   2.3991,   1.3457],
        [  0.6396,  14.0772,  -6.3786],
        [ 12.4691,   2.8525,  -5.5110],
        [  1.7218,  -4.7576,  -3.3617],
        [  5.1196,  -2.7160,   4.2705],
        [  1.2273,  -4.8081,   9.8129],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.3767, -11.0341,   9.9415],
        [ -2.0445,   7.8469,  -0.5857],
        [  0.5250, -11.0306,   5.5358],
        [ 10.0619,   8.6627,  -9.2884],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.6310, -10.9195,   3.8054],
        [  0.9809,   7.6642,  -1.4217],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.7726,  -3.4876,  -6.7402],
        [  1.7228,  -4.7575,   8.3396],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.5900,   2.9767,   7.4811],
        [ -5.2752,  -9.0409,  -8.2568],
        [  4.7544,  -3.1425,  -1.5315],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.5832,   1.7543,  -7.6134],
        [  5.1793,   4.7461,   0.3899],
        [ -8.0534,  -6.0429,   2.4583],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.4501,  -1.7133,   2.2980],
        [  1.1274,  14.0116,   8.1753],
        [  4.4633,   5.6287,  -0.7754],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.0608,   3.9362,  -5.9567],
        [  9.3607,   9.5337,  -4.5412],
        [ -2.1382,   7.8339,   7.3524]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0216, 0.0235,  ..., 1.1268, 0.9632, 0.2489],
        [0.0216, 0.0000, 0.0080,  ..., 1.1440, 0.9616, 0.2693],
        [0.0235, 0.0080, 0.0000,  ..., 1.1396, 0.9536, 0.2724],
        ...,
        [1.1268, 1.1440, 1.1396,  ..., 0.0000, 1.0135, 1.0181],
        [0.9632, 0.9616, 0.9536,  ..., 1.0135, 0.0000, 1.0942],
        [0.2489, 0.2693, 0.2724,  ..., 1.0181, 1.0942, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.89869205,  5.05981914,  5.08372149,  5.13616224,  5.14371005,
        5.29755487, 11.05317597, 10.19036081,  8.24808832, 13.86480175,
        6.84796202, 11.4424816 ,  6.27905292,  9.36378438,  5.91677188,
       12.9089559 ,  7.92200342,  8.09842704,  4.978704  , 12.70489563,
        9.53015376, 13.38091121, 10.87409005,  9.97964013,  7.70897675,
       11.0050379 , 11.00694688, 10.69963954, 10.42667093,  9.78285424,
       13.89264822, 11.83075005,  9.95552626,  5.47702766, 12.27835996,
        6.56833864, 12.0495521 ,  6.27905292,  4.98461983,  6.16512856,
        5.5366698 , 12.41012046,  8.06643487,  8.72896381,  7.92200342,
       12.44177641,  5.38342077, 12.62711252,  4.92497514,  7.60463349,
        9.58072286,  7.69752262,  8.06517143,  8.37954853, 10.91805683,
       14.03618649,  7.5158489 ,  9.23071698, 12.22503754, 10.30972193,
        9.78147542, 12.79126785, 12.27993098,  9.09119351, 12.5085995 ,
        8.05242591, 11.80092578, 10.1371552 ,  7.81509631, 12.24467041,
        5.87132718,  9.562174  ,  9.15976746, 14.09176259, 12.79126785,
        5.05963217,  5.79543038,  4.96231532, 13.12299145, 11.11969815,
        8.10886021, 11.04313641, 13.27717982, 10.46238054, 11.23201619,
        7.72668467,  4.91336994, 12.91143491, 12.27835996,  5.05981914,
       10.01804223,  9.09119351, 10.46739073,  5.69913817,  8.43626114,
       12.70489563,  7.02496223, 10.06850677,  9.37296102,  9.60412735,
       14.05687815,  7.18354885,  4.9160962 ,  8.97052115, 13.36092624,
        8.12043983])}
{'P': tensor([[2.9096e-02, 1.5897e-01, 1.5005e-01,  ..., 9.9448e-03, 1.2852e-03,
         2.9561e-05],
        [4.7918e-03, 5.0952e-02, 4.1866e-02,  ..., 1.2292e-02, 3.3226e-03,
         6.3392e-05],
        [4.2903e-03, 4.0687e-02, 4.3463e-02,  ..., 9.2139e-03, 2.8843e-03,
         4.8743e-05],
        ...,
        [1.6735e-02, 2.8123e-02, 2.4618e-02,  ..., 1.2671e-01, 4.8714e-03,
         4.5852e-04],
        [3.8117e-05, 4.8090e-05, 4.8355e-05,  ..., 3.8819e-05, 5.0169e-02,
         1.2435e-04],
        [2.1969e-05, 2.0358e-05, 1.8677e-05,  ..., 4.8579e-05, 2.1036e-06,
         4.9633e-02]]), 'X': tensor([[  0.9392,  -4.8143,  -3.4649],
        [  1.7218,  -4.7576,  -3.3617],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.7420,  -4.4896,  -3.7139],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.7976,  -4.4682,  -4.2020],
        [ -1.5261,   7.8988,   9.3768],
        [ -8.5900,   2.9767,   7.4811],
        [  6.3515,   0.6418,  -6.5823],
        [  6.9693,  -9.3200,  -8.3822],
        [  7.1697,  11.5343,  -0.7276],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.4867,  12.5816,   2.9549],
        [ -9.3810,  -2.3455,  -9.5194],
        [  0.4119,  14.1019,  -4.1870],
        [ -7.9298,   4.1367,  -3.9773],
        [  8.2422,  -8.4041,  -1.4962],
        [  2.3261,  13.7749,   5.9121],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.3413,   0.4883,  -6.8416],
        [  6.2883,   0.0325,   5.2692],
        [  3.1892, -10.8220,  -6.8409],
        [ -2.1382,   7.8339,   7.3524],
        [ -6.3489,   5.9117,   6.5161],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.5154,   2.5072,  -9.3274],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.1120,  -4.6818,   8.6144],
        [  5.8323,  -1.5228,   5.1132],
        [ -9.1231,  -3.5739,   8.4708],
        [ -0.1626, -10.9796,   6.4629],
        [  0.1382, -11.0073,   1.0026],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.1861,   4.7361,  -3.7312],
        [  6.2022,   2.4092,   3.0349],
        [ -5.1369,   6.7755,   8.3836],
        [ 12.6035,   0.7494,   7.1536],
        [  2.0039,   7.3173,  -1.5885],
        [  1.6954,  -4.7616,   9.8198],
        [  0.9392,  -4.8143,  -3.4649],
        [  4.1637,  13.1898,  -1.9189],
        [ 12.5763,   0.1333,  -9.2650],
        [  1.6422,   7.4586,  -6.8056],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.3076,   0.7852,  -6.7010],
        [ -5.1398,  -9.1367,   6.2074],
        [  5.3118,   4.5452,   4.2114],
        [  0.6396,  14.0772,  -6.3786],
        [  9.6662,   9.1725,   8.1056],
        [  3.9591, -10.6391,  -6.2557],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.0591,   6.0202,  -1.6339],
        [  3.6154,  -4.0640,  -7.9698],
        [ 10.0567,   8.6697,   4.6329],
        [ -8.6229,   2.9068,   2.4429],
        [  3.1756, -10.8247,   6.8882],
        [  4.9337,  12.8566,  -3.6582],
        [  0.7275,  14.0667,  -0.9815],
        [  6.0955,   2.8210,  -4.3321],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -9.3739,   0.3982,  -4.2564],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.8407,  11.7666,   3.0045],
        [  3.1756, -10.8247,   6.8882],
        [  0.1931,  -4.7491,   7.4385],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.4426, -10.1181,  -5.1268],
        [  5.7784,  -1.6401,   9.5587],
        [  4.0421,  -3.7765,  -8.3631],
        [  1.2728, -11.0372,   9.8121],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -0.5709,   7.9055,   7.6089],
        [  9.2692,   9.6370,   7.0746],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.4572,   7.1085,   7.7376],
        [  6.1755,   2.5233,  -2.0042],
        [  1.5653,  13.9377,   4.1773],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.7792,  11.8084,   1.6251],
        [  5.0348,  -2.8229,   3.5436],
        [  4.4915,   5.5990,   7.0405],
        [ 10.5955,   7.8824,  -3.7006],
        [ 12.3357,   3.6001,   0.7903],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.3305,   1.6119,   1.1538],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.6703,   6.3443,  -5.4401],
        [ -3.5903,   7.4847,  -8.9112],
        [ -7.4808,   4.7462,   2.0266],
        [  0.5850,  -4.7982,   9.6639],
        [ -3.9700,   7.3453,  -4.1486],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -1.5261,   7.8988,   9.3768],
        [ -7.0253,  -7.4493,  -1.5978],
        [  1.9503,  -4.7172,  -6.2313],
        [  5.3860,  -2.3418,   1.0826],
        [  6.1755,   2.5233,  -2.0042],
        [  7.0642,  11.6104,  -9.8316],
        [ -7.0825,   5.2073,  -5.0497],
        [  1.7903,  -4.7467,   0.1168],
        [ 12.4597,  -0.9136,  -4.8763],
        [  2.1477,   7.2550,   8.9688]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.0014e-02, 2.3758e-02,  ..., 2.1307e-01, 2.7203e-01,
         7.8162e-01],
        [2.0014e-02, 0.0000e+00, 1.5741e-02,  ..., 2.0904e-01, 2.6003e-01,
         7.8051e-01],
        [2.3758e-02, 1.5741e-02, 0.0000e+00,  ..., 2.2478e-01, 2.4870e-01,
         7.9618e-01],
        ...,
        [2.1307e-01, 2.0904e-01, 2.2478e-01,  ..., 0.0000e+00, 4.3530e-01,
         5.7328e-01],
        [2.7203e-01, 2.6003e-01, 2.4870e-01,  ..., 4.3530e-01, 0.0000e+00,
         9.9932e-01],
        [7.8162e-01, 7.8051e-01, 7.9618e-01,  ..., 5.7328e-01, 9.9932e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.90505566,  5.05963217,  5.05774295,  5.26067019,  5.18277858,
        5.27178363,  8.04483495,  9.09119351,  6.383897  , 11.63765273,
       13.58101794,  9.62958973, 13.72594022,  9.66977802, 14.10796466,
        8.94395628, 11.77124373, 13.96987806, 10.61047596,  6.36004393,
        6.28835101, 11.28215429,  8.12043983,  8.67502348,  5.8828986 ,
        6.2134961 , 12.76408562,  5.05963217,  5.13616224,  6.02779201,
        9.79814997, 10.98076873, 11.00812996, 10.0674887 ,  7.02326515,
        6.65368551,  8.50268322, 12.62576026,  7.58668144,  5.05444288,
        4.90505566, 13.83139024, 12.5769659 ,  7.63726134,  6.14086582,
        9.19837792,  9.34061938, 10.48315436,  6.99097013, 14.09176259,
       13.32557133, 11.35182589,  5.68942572,  7.26074925,  5.43943791,
       13.2778378 ,  9.09963894, 11.28093507, 13.77075174, 14.08549802,
        6.7166297 ,  9.54154096,  9.38232121, 10.48315436, 13.6105595 ,
       11.28093507,  4.75303127,  5.19986797, 11.4890861 ,  6.00663314,
        5.53175275, 11.11038551,  9.93553424,  7.92607342, 13.37121923,
        7.83486943, 12.7125079 ,  7.52117132,  6.67108662, 14.02532758,
        8.41171369, 13.61600536,  5.77219077,  7.17789666, 13.20597932,
       12.85032836,  6.19633139,  6.53250937, 13.00413829,  7.32950385,
        8.30125201,  8.859416  ,  4.83371341,  8.34949383, 12.10942412,
        9.50654967,  8.04483495, 10.23946247,  5.10444097,  5.87305005,
        6.67108662, 13.59056613,  8.7907923 ,  5.07306274, 12.49317563,
        7.56617037])}
{'P': tensor([[3.0540e-02, 1.5936e-01, 1.1533e-01,  ..., 5.6956e-03, 1.5641e-04,
         3.1098e-03],
        [3.4026e-03, 3.8889e-02, 2.1786e-02,  ..., 9.3278e-03, 3.2299e-04,
         5.2624e-03],
        [4.6187e-03, 4.9422e-02, 4.0208e-02,  ..., 7.8463e-03, 3.4442e-04,
         4.4923e-03],
        ...,
        [5.2460e-04, 6.5133e-04, 5.9835e-04,  ..., 5.7799e-02, 9.7253e-04,
         3.1863e-01],
        [3.3404e-05, 3.5900e-05, 3.7132e-05,  ..., 9.1371e-05, 1.5064e-01,
         1.5658e-04],
        [1.1655e-03, 1.3582e-03, 1.2468e-03,  ..., 1.0246e-02, 2.8839e-03,
         2.0408e-01]]), 'X': tensor([[  0.9478,  -4.8144,   0.6371],
        [  1.8721,  -4.7322,   0.5524],
        [  1.7903,  -4.7467,   0.1168],
        [  2.1824,  -4.6647,   1.1985],
        [  2.7558,  -4.4844,   0.0636],
        [  2.8890,  -4.4316,   1.1600],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -3.5173,   7.5091,  -8.9585],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -2.0229, -10.6163,   8.6930],
        [ -5.6169,  -8.7844,   8.9057],
        [ -9.3384,  -2.6161,  -4.7630],
        [  8.6420,  -8.0580,   6.5781],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.2883,   0.0325,   5.2692],
        [ -7.0120,   5.2825,   7.7762],
        [  1.7232,  -4.7574,   3.1270],
        [  0.2480, -11.0153,  -3.0183],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.4203,  -2.0340,   2.5856],
        [  5.6928,  12.4710,   3.6001],
        [ 12.5770,   1.8512,  -0.6578],
        [  8.0381,  -8.5690,  -8.9523],
        [ -6.2668,   5.9808,   2.5614],
        [ -7.3446,   4.9113,  -5.9356],
        [  4.0587, -10.6112,   6.2273],
        [  3.2366,  -4.2732,   5.1532],
        [  4.8051, -10.3703,   0.5268],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.7122,  -4.7591,  -3.6734],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.4604,  -4.1545,   1.8091],
        [ -8.8363,   2.4086,  -5.6296],
        [  2.3200,   7.1756,   1.1938],
        [  6.9693,  -9.3200,  -8.3822],
        [  0.6933,   7.7336,   2.1179],
        [  1.7659,   7.4127,   4.7871],
        [ -6.7992,   5.4991,   9.9186],
        [ -7.8280,   4.2848,  -7.1955],
        [  0.6470, -11.0352,  -1.1931],
        [ -0.8675, -10.8815,   6.8561],
        [  6.3236,   0.3011,  -4.6668],
        [  2.9461,  -4.4077,  -0.4831],
        [ -1.0482, -10.8487,  -5.9066],
        [  5.6248,   3.9984,   8.4933],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.2301,  -3.1537,  -2.7477],
        [  5.9003,  12.3550,   2.1917],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.1531,  -5.8748,   2.6934],
        [ -9.0574,   1.7801,   8.8986],
        [  6.3246,   1.6696,  -0.1544],
        [  6.2770,  -0.0379,   2.4195],
        [  4.9957,  -2.8705,   6.5978],
        [ -8.8154,   2.4611,   2.5198],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.8323,  -1.5228,   5.1132],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.9470,  13.6085,   2.9809],
        [  6.8704,  -9.3809,   7.5769],
        [ 11.6019,   5.9540,  -5.1715],
        [ -2.5683, -10.4433,  -2.9182],
        [ -3.9456,  -9.8542,   0.7466],
        [  3.6469,   6.3624,  -1.8414],
        [ -1.9487,   7.8590,  -3.3330],
        [ -8.9624,  -4.0944,  -4.9622],
        [  0.5250, -11.0306,   5.5358],
        [  1.3705,  13.9723,   7.9266],
        [  5.6258,  -1.9404,  -6.0715],
        [  7.7750,  11.0665,   2.8364],
        [  4.5491,  -3.3485,   6.8412],
        [  1.2728, -11.0372,   9.8121],
        [  5.4336,  -2.2676,   5.6933],
        [ -4.7642,  -9.3860,  -5.1133],
        [  0.9392,  -4.8143,  -3.4649],
        [  4.1297,  -3.7098,  -3.9838],
        [  5.6414, -10.0303,   7.8525],
        [  2.2076,  13.8032,  -1.5099],
        [  5.7151,  -1.7700,   3.1542],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -5.6423,   6.4537,  -1.2091],
        [  2.2076,  13.8032,  -1.5099],
        [  6.2576,   2.1359,  -3.3398],
        [  4.3554,   5.7393,  -9.0766],
        [  5.8902,  -1.3887,   4.0388],
        [  2.5829,  -4.5466,  -8.4015],
        [ -8.9282,   2.1648,   8.4894],
        [  1.7218,  -4.7576,  -3.3617],
        [ -0.4341, -10.9473,   7.4509],
        [ -9.4739,  -1.2497,   0.5483],
        [  8.0877,  -8.5297,  -1.1050],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.8890,  -4.4316,   1.1600],
        [  2.5974,   7.0363,  -7.7810],
        [ -5.9938,  -8.4745,   3.9735],
        [ 10.9717,  -5.1646,  -6.9039],
        [  5.9193,  -1.3176,   5.1739],
        [ 11.3118,   6.5957,  -4.5667],
        [  6.2895,   0.0405,   5.7652]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0236, 0.0369,  ..., 0.3814, 0.5551, 0.4180],
        [0.0236, 0.0000, 0.0283,  ..., 0.3762, 0.5380, 0.4133],
        [0.0369, 0.0283, 0.0000,  ..., 0.4040, 0.5185, 0.4412],
        ...,
        [0.3814, 0.3762, 0.4040,  ..., 0.0000, 0.8085, 0.0376],
        [0.5551, 0.5380, 0.5185,  ..., 0.8085, 0.0000, 0.8431],
        [0.4180, 0.4133, 0.4412,  ..., 0.0376, 0.8431, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.90676242,  5.08910491,  5.07306274,  5.14999799,  5.26342203,
        5.29009836, 12.53345445,  8.29203356,  9.96152788, 10.80728788,
       10.42667093,  9.69792513, 11.8159254 ,  8.12580976,  6.28835101,
        8.77913428,  5.05989547, 11.018084  ,  6.14086582,  9.09502385,
        9.63742669, 13.70893492, 12.7125079 , 11.74901778,  8.66273171,
        8.8353674 , 11.36089355,  5.36061974, 11.42947632,  8.06990483,
        5.05774295,  9.55593374,  9.36093562,  5.40689332,  9.15866607,
        7.54127425, 11.63765273,  7.76457095,  7.62011734,  8.74470218,
        8.92395381, 11.05413545, 10.91601655,  6.33075682,  5.30158497,
       10.8992524 ,  6.90115901, 11.59512194,  9.75401449, 13.69162795,
        7.89745107,  9.37989631, 10.04916678,  9.23071698,  6.54127431,
        6.27715105,  5.76165934,  9.15253344,  9.84314712,  6.02779201,
        9.97822473, 12.35625694, 13.00413829, 13.92390689, 11.62770332,
       13.04051744, 10.75443106, 10.61472636,  7.33350642,  8.09704052,
        9.85330646, 11.04313641, 14.0393949 ,  5.9510489 , 13.5247215 ,
        5.64865173, 11.11038551,  5.88782913, 10.52587243,  4.90505566,
        5.5513072 , 11.50791859, 13.97857591,  5.98285319,  7.48311002,
        9.27474386,  9.62958973,  8.5724042 , 13.97857591,  6.61211224,
        7.20482224,  6.05165248,  5.22903117,  9.18690018,  5.05963217,
       10.95593729,  9.55593374, 11.75438006,  9.55720651,  5.29009836,
        7.50035355, 10.37998078, 12.12641969,  6.06416957, 13.09426142,
        6.28961988])}
{'P': tensor([[3.2071e-02, 2.1116e-01, 1.8195e-01,  ..., 3.1270e-05, 4.7480e-05,
         2.2987e-03],
        [5.3272e-03, 4.6921e-02, 2.2681e-01,  ..., 5.3273e-05, 6.3804e-05,
         3.9218e-03],
        [6.7508e-03, 1.0429e-02, 5.9894e-02,  ..., 7.5226e-05, 7.8056e-05,
         5.4023e-03],
        ...,
        [1.6002e-06, 1.5897e-06, 1.6207e-06,  ..., 1.7829e-02, 3.1239e-06,
         5.2319e-05],
        [1.2746e-03, 1.0680e-03, 9.7220e-04,  ..., 2.6034e-07, 3.1112e-02,
         1.7184e-05],
        [3.8545e-04, 3.8576e-04, 3.8706e-04,  ..., 1.3211e-03, 2.5201e-04,
         8.7071e-02]]), 'X': tensor([[  0.9659,  -4.8144,  -0.9594],
        [  1.3103,  -4.8032,  -0.9300],
        [  1.4749,  -4.7892,  -0.8089],
        [  1.3403,  -4.8011,  -1.1698],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.6451,  -4.5250,  -0.5565],
        [ 10.9931,   7.2107,   7.7933],
        [ -3.2736, -10.1703,   3.8712],
        [ 12.5789,   0.1722,   1.6470],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.4572,   7.1085,   7.7376],
        [ -5.0550,   6.8231,  -9.2502],
        [  5.8617,   3.4853,  -9.8080],
        [  2.5663,   7.0526,  -6.8769],
        [  0.6952,  -4.8060,   8.8976],
        [  3.9770, -10.6341,   8.0783],
        [  6.2247,  -0.3150,  -9.3657],
        [  7.1821,  -9.1845,   9.8548],
        [ -5.6792,  -8.7352,   4.0318],
        [  7.5309,  11.2617,   7.2656],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.5258,  -0.4200,  -9.3493],
        [  2.0729,   7.2878,   7.0880],
        [  6.3549,   0.7096,   6.9198],
        [ -8.3653,   3.4184,   9.3454],
        [  6.3516,   0.6425,   0.3425],
        [ -6.3489,   5.9117,   6.5161],
        [ -7.6719,  -6.6260,   7.1468],
        [  3.9770, -10.6341,   8.0783],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.3653,   3.4184,   9.3454],
        [  3.3943,   6.5483,   1.0790],
        [ 12.4996,   2.6335,  -6.3166],
        [  2.3200,   7.1756,   1.1938],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.1787,  -2.6383,  -4.2533],
        [  2.1625,  -4.6697,   7.0658],
        [  9.0051,   9.9234,   5.4411],
        [ -2.5056,   7.7720,   5.5878],
        [  2.0243,   7.3086,   7.5432],
        [  2.0729,   7.2878,   7.0880],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.4297,  -5.3637,  -9.3962],
        [  0.9074,  -4.8139,   8.1674],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.3539,  -5.5111,   0.3373],
        [ -0.8885,   7.9160,   0.0571],
        [ -8.5859,  -5.0393,   0.5476],
        [  4.4290,  -3.4596,   1.4181],
        [ -7.9925,  -6.1423,   2.8990],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.8674,  12.8875,   6.7622],
        [  6.6319,  11.9066,   3.5165],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.2647, -10.8066,  -0.3768],
        [ -8.0829,  -5.9940,   5.0564],
        [  1.7122,  -4.7591,  -3.6734],
        [ -7.4807,   4.7463,   2.0616],
        [ -3.9858,  -9.8334,  -3.7323],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.3384,  -2.6161,  -4.7630],
        [  5.7784,  -1.6401,   9.5587],
        [ -7.8280,   4.2848,  -7.1955],
        [  9.2811,  -7.4336,   5.1771],
        [  6.0310,  12.2795,  -6.4528],
        [  6.1989,  -9.7591,   0.8289],
        [  5.8902,  -1.3887,   4.0388],
        [  7.9403,  -8.6455,  -8.7433],
        [  3.2226,  -4.2802,  -4.0428],
        [  1.7218,  -4.7576,  -3.3617],
        [  4.9152,  12.8653,  -1.4459],
        [  3.6469,   6.3624,  -1.8414],
        [  9.8082,  -6.8396,   4.9615],
        [  1.7000,  -4.7609,   2.7935],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.6022,  -1.9833,   8.3081],
        [  2.5974,   7.0363,  -7.7810],
        [ -5.1974,   6.7396,   3.3484],
        [ 10.9931,   7.2107,   7.7933],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.3384,  -2.6161,  -4.7630],
        [  2.3261,  13.7749,   5.9121],
        [ -0.6323,   7.9085,  -5.9544],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.9655,   3.2172,  -7.5953],
        [  0.9036,   7.6840,   9.9237],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.5428,  -9.5728,  -2.9897],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.8934,   2.2597,   3.5831],
        [ -4.8751,  -9.3148,   4.0406],
        [ -8.7735,   2.5640,  -6.3313],
        [  2.6613,   7.0021,   4.5783],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.2182,  13.8007,   5.1889],
        [  9.1620,  -7.5573,  -9.9882],
        [ -8.1531,  -5.8748,   2.6934],
        [  4.1537,   5.9340,  -6.0423],
        [  0.6363, -11.0348,   6.1414],
        [  9.4925,   9.3811,   8.9842],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.9026,   3.3837,  -0.6573]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0113, 0.0195,  ..., 0.9945, 0.8017, 0.1152],
        [0.0113, 0.0000, 0.0093,  ..., 0.9865, 0.8129, 0.1039],
        [0.0195, 0.0093, 0.0000,  ..., 0.9773, 0.8197, 0.0961],
        ...,
        [0.9945, 0.9865, 0.9773,  ..., 0.0000, 1.6300, 0.9069],
        [0.8017, 0.8129, 0.8197,  ..., 1.6300, 0.0000, 0.9148],
        [0.1152, 0.1039, 0.0961,  ..., 0.9069, 0.9148, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.91038884,  4.978704  ,  5.01114719,  4.98461983,  5.21280156,
        5.24137909, 13.14690137, 10.68416401, 12.58006038, 13.12299145,
        7.52117132,  8.49161154,  6.81961996,  7.50500064,  4.85603886,
       11.35345345,  6.23262289, 11.65923396, 10.41909211, 13.54774355,
        8.37954853, 12.53285633,  7.57685908,  6.39438044,  9.0368458 ,
        6.38399789,  8.67502348, 10.1371552 , 11.35345345, 10.06286074,
        9.0368458 ,  7.37576154, 12.77402351,  7.54127425,  6.23262289,
        5.81199422,  5.14607014, 13.40024224,  8.16584537,  7.58378655,
        7.57685908, 10.09979747,  9.55720651,  9.77777346,  9.99145096,
        4.89869205,  7.8575648 , 10.00794468,  7.96575782,  9.95552626,
        5.62005629, 10.08001894, 11.75618513, 13.77604293, 13.62896593,
       12.50996828,  9.20340031,  9.36093562, 11.28895489, 10.06286074,
        5.05774295,  8.85939626, 10.61047596,  7.822854  ,  9.69792513,
        6.00663314,  8.92395381, 11.89105259, 13.68063098, 11.56146873,
        6.05165248, 11.73848065,  5.35774573,  5.05963217, 13.77222752,
        7.33350642, 11.95743558,  5.05534609,  9.94173708,  5.94293185,
        7.50035355,  8.51088913, 13.14690137, 10.80728788,  9.69792513,
       13.96987806,  7.9337647 , 10.51340829,  6.77777606,  7.73692436,
        5.39450305, 11.59512194, 13.25336001,  9.17595899, 10.51340829,
        9.14044984,  7.49076899, 10.8207297 , 13.97779839, 11.87665501,
       10.04916678,  7.24326992, 11.05317597, 13.34587027,  9.43737523,
        6.80370254])}
{'P': tensor([[3.4039e-02, 1.7971e-01, 1.3590e-01,  ..., 1.3130e-02, 3.3499e-03,
         1.5837e-03],
        [4.2519e-03, 4.2558e-02, 1.4355e-01,  ..., 1.0786e-02, 5.1608e-03,
         2.7170e-03],
        [3.7667e-03, 7.1804e-03, 5.0337e-02,  ..., 1.1009e-02, 9.3912e-03,
         4.6134e-03],
        ...,
        [1.5294e-02, 2.2494e-02, 4.2599e-02,  ..., 1.3623e-01, 4.9473e-02,
         1.2992e-02],
        [7.6554e-04, 8.3074e-04, 1.0100e-03,  ..., 1.1906e-03, 1.4435e-01,
         2.6635e-03],
        [1.8027e-05, 2.2730e-05, 3.0306e-05,  ..., 2.8704e-05, 4.1800e-03,
         2.1424e-02]]), 'X': tensor([[  0.9788,  -4.8145,  -3.8661],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.7420,  -4.4896,  -3.7139],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.9526,   3.2525,  -4.3821],
        [ -7.6734,   4.4980,  -5.4248],
        [  4.2096,   5.8815,   1.9230],
        [ -7.7842,   4.3466,  -1.7548],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.0330,  -6.0765,   6.7945],
        [  0.4996,   7.7738,  -4.4711],
        [ -7.5720,   4.6310,   2.1717],
        [  4.1999,  -3.6543,  -4.1998],
        [ -6.9838,  -7.4965,   0.9796],
        [ -3.2053, -10.1993,  -3.8736],
        [  6.0140,  -1.0661,  -2.5040],
        [ -8.0750,  -6.0072,  -5.8267],
        [  8.3718,  -8.2953,   8.4666],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.9641,  -2.9568,   7.7211],
        [ -5.0943,  -9.1682,   0.5073],
        [  7.1697,  11.5343,  -0.7276],
        [ 12.3025,  -1.7440,  -6.6632],
        [  4.1999,  -3.6543,  -4.1998],
        [  1.5653,  13.9377,   4.1773],
        [  0.1532,  -4.7422,  -7.0145],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -3.3116, -10.1539,   7.5919],
        [ 11.2517,  -4.6495,   2.9800],
        [ -5.0943,  -9.1682,   0.5073],
        [  6.0031,  12.2958,  -6.5992],
        [  4.7482, -10.3907,   5.9149],
        [ 12.3453,  -1.5462,   7.0212],
        [ -9.3107,  -2.7697,  -3.2919],
        [  4.1140,  -3.7220,  -0.1273],
        [  0.3724,   7.7974,  -6.8722],
        [  4.7482, -10.3907,   5.9149],
        [ 10.9162,  -5.2599,  -7.0812],
        [  1.3767, -11.0341,   9.9415],
        [  7.9141,  -8.6656,   9.6948],
        [ -6.4844,  -8.0230,   8.5209],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3497,   1.3680,  -1.9076],
        [ 12.4691,   2.8525,  -5.5110],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.5010,  -4.5736,  -0.8989],
        [ 11.1085,  -4.9204,   0.0736],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 10.1503,   8.5414,  -7.2681],
        [ -6.7643,  -7.7372,  -9.6663],
        [  1.8447,  -4.7372,   8.1235],
        [ 11.1420,   6.9331,  -2.4779],
        [ -2.2188,   7.8218,   8.6064],
        [  4.7482, -10.3907,   5.9149],
        [  3.0719, -10.8449,  -4.1673],
        [  2.2644,  -4.6434,   9.7508],
        [ -8.8058,   2.4850,  -1.7425],
        [ 12.4877,  -0.7222,   8.8347],
        [  3.4604,  -4.1545,   1.8091],
        [  5.5424,  -2.0885,   7.2487],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.7685, -10.8982,  -6.8928],
        [  5.4867,  12.5816,   2.9549],
        [ -8.8835,  -4.3178,   9.3080],
        [ -4.7389,   6.9953,   3.7638],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.3339,   7.5627,  -0.9034],
        [ -8.9178,   2.1936,   7.5388],
        [ -0.8675, -10.8815,   6.8561],
        [  3.9663,   6.1018,  -5.4104],
        [  1.7232,  -4.7574,   3.1270],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.1504,   4.3674,  -7.2998],
        [ -7.5603,   4.6459,   8.2002],
        [ -1.6334, -10.7207,   7.1397],
        [ -4.8751,  -9.3148,   4.0406],
        [ -1.6220,   7.8917,   0.0315],
        [  8.8091,  -7.9037,   0.3631],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.6470, -11.0352,  -1.1931],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.7544,   2.6096,   3.3146],
        [  6.3301,   0.3638,  -7.5685],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.3800,  -4.3919,   1.4066],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.4996,   7.7738,  -4.4711],
        [  4.1537,   5.9340,  -6.0423],
        [ 12.5650,   2.0132,   4.6272],
        [  2.8602,   6.8905,   5.8779],
        [  9.9217,  -6.7006,  -5.6728],
        [  1.3339,   7.5627,  -0.9034],
        [  5.6314,  -1.9302,  -9.6543],
        [  4.1348,   5.9514,  -0.8856],
        [  5.0617,  12.7957,   8.3544],
        [  9.6816,   9.1536,  -5.8543],
        [ -7.9298,   4.1367,  -3.9773],
        [  2.0899,  -4.6870,  -7.4260],
        [  6.3505,   0.6235,  -8.5849],
        [ 12.3025,  -1.7440,  -6.6632]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0187, 0.0379,  ..., 0.1427, 0.2161, 0.3289],
        [0.0187, 0.0000, 0.0295,  ..., 0.1444, 0.2119, 0.3249],
        [0.0379, 0.0295, 0.0000,  ..., 0.1172, 0.1824, 0.2954],
        ...,
        [0.1427, 0.1444, 0.1172,  ..., 0.0000, 0.0899, 0.1950],
        [0.2161, 0.2119, 0.1824,  ..., 0.0899, 0.0000, 0.1131],
        [0.3289, 0.3249, 0.2954,  ..., 0.1950, 0.1131, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.91296205,  5.05774295,  5.18277858,  5.19577146,  5.05963217,
        5.26067019, 13.04217487,  6.78325151,  8.89458273,  7.23276582,
        8.91551231, 10.36844813, 10.07238463,  7.78980299,  8.87587457,
        5.5671389 , 10.24557047, 10.6910768 ,  6.10774544, 10.06437598,
       11.785566  , 12.10942412, 12.32408516, 10.48840476, 13.58101794,
       12.42554808,  5.5671389 , 14.02532758,  4.74468406,  9.96391045,
        9.61942445, 10.23946247, 10.68031351, 12.17451685, 10.48840476,
       13.6829843 , 11.42420508, 12.44177641,  9.71392179,  5.54776934,
        7.80625981, 11.42420508, 12.11735334, 11.11969815, 11.73567759,
       10.31584102,  6.48486171,  6.49538841, 12.79126785, 11.75807252,
        5.21280156, 12.1494023 , 10.39942497, 13.26590071, 10.27716434,
        5.08372149, 13.12299145,  8.1303945 , 11.42420508, 11.27160035,
        5.16611782,  9.14973319, 12.5085995 ,  5.40689332,  5.92281142,
        9.71392179, 10.92597196, 11.24434374, 13.72594022,  9.87720102,
        8.44940661, 12.32408516,  7.67939546,  9.18358562, 10.91601655,
        7.27759233,  5.05989547, 11.62770332, 12.91143491,  8.8737556 ,
       10.84437705, 10.51340829,  8.05668609, 11.83509372, 10.33712258,
       11.05413545,  9.94173708,  9.13507321,  6.34059021,  9.96152788,
       12.19805394, 10.30972193,  7.78980299,  7.24326992, 12.72524615,
        7.46053281, 11.97238624,  7.67939546,  5.95297459,  7.24677664,
       13.76047999, 13.32374056,  8.94395628,  5.13182506,  6.38105015,
       12.42554808])}
{'P': tensor([[4.8515e-02, 2.2313e-01, 2.0633e-01,  ..., 9.5481e-05, 1.1603e-04,
         1.5897e-04],
        [6.7642e-03, 8.5841e-02, 6.1495e-02,  ..., 1.7675e-04, 2.2463e-04,
         3.4363e-04],
        [6.5997e-03, 6.7992e-02, 7.8568e-02,  ..., 1.6363e-04, 2.6153e-04,
         3.3283e-04],
        ...,
        [4.6256e-04, 2.7467e-04, 2.8189e-04,  ..., 5.4227e-02, 6.7818e-06,
         3.5406e-04],
        [1.3734e-04, 1.5085e-04, 1.8185e-04,  ..., 2.1765e-04, 2.8355e-01,
         1.5233e-03],
        [3.6505e-04, 2.4730e-04, 2.6370e-04,  ..., 2.1635e-01, 3.5600e-05,
         9.9862e-02]]), 'X': tensor([[  0.9809,  -4.8145,   6.8706],
        [  2.1625,  -4.6697,   7.0658],
        [  2.1379,  -4.6757,   6.4981],
        [  2.2437,  -4.6489,   6.1430],
        [  3.9515,  -3.8425,   6.1729],
        [  2.9260,  -4.4162,   8.3808],
        [  2.5636, -10.9293,   9.6519],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.9663,   6.1018,  -5.4104],
        [ -6.3489,   5.9117,   6.5161],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.0534,  -6.0429,   2.4583],
        [  4.6992,   5.3690,   3.0935],
        [  1.3705,  13.9723,   7.9266],
        [  5.4336,  -2.2676,   5.6933],
        [  5.7784,  -1.6401,   9.5587],
        [  0.1532,  -4.7422,  -7.0145],
        [ 12.5154,   2.5072,  -9.3274],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.3373,  13.7721,  -0.2442],
        [  5.6022,  -1.9833,   8.3081],
        [  1.2728, -11.0372,   9.8121],
        [  0.3724,   7.7974,  -6.8722],
        [  1.4749,  -4.7892,  -0.8089],
        [  3.5965,  -4.0754,  -3.5035],
        [  8.5093,  -8.1764,  -3.2381],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 11.2847,  -4.5848,  -4.7503],
        [  3.8182,  -3.9347,  -1.7961],
        [ -9.2508,   1.0572,   2.9978],
        [  6.9284,  11.7061,  -0.3732],
        [  5.1787,  -2.6383,  -4.2533],
        [  9.6614,   9.1784,  -1.2087],
        [  5.9026,   3.3837,  -0.6573],
        [ -0.5739,   7.9056,   0.6382],
        [ 11.0275,  -5.0666,   8.4154],
        [ -0.2586,   7.8825,  -9.9840],
        [  1.4489,   7.5256,   4.4529],
        [  0.2480, -11.0153,  -3.0183],
        [  0.7145,  -4.8071,  -2.3766],
        [ -7.4887,   4.7364,   8.9074],
        [ -6.3066,  -8.1936,   9.5822],
        [ -5.9940,   6.1985,   5.9566],
        [ -1.6334, -10.7207,   7.1397],
        [  2.9470,  13.6085,   2.9809],
        [ -0.8885,   7.9160,   0.0571],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -4.0868,   7.2981,   8.1874],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.2707,   0.9661,   4.4863],
        [  3.1480,  -4.3165,  -7.8002],
        [ 11.8951,  -3.1570,   7.0742],
        [ -9.4296,  -1.9458,   4.0833],
        [ -8.4297,  -5.3637,  -9.3962],
        [  8.1043,  -8.5164,  -7.9290],
        [  7.9816,  10.8938,  -0.5378],
        [  6.0310,  12.2795,  -6.4528],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.5650,   2.0132,   4.6272],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.2555,   2.1474,   4.1685],
        [  0.6757, -11.0361,  -1.3047],
        [ -9.1960,  -3.2964,   1.6921],
        [ -9.2707,   0.9661,   4.4863],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.1579,  -0.5961,  -6.5595],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -7.6293,  -6.6859,  -5.5161],
        [  0.3724,   7.7974,  -6.8722],
        [ -2.9497, -10.3028,  -7.3479],
        [ -2.1533, -10.5778,   1.0112],
        [  9.7328,   9.0902,  -0.8917],
        [ 12.5650,   2.0132,   4.6272],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4645,  -1.4880,  -2.0226],
        [  0.6363, -11.0348,   6.1414],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.5428,  -9.5728,  -2.9897],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -1.9714,   7.8563,  -1.8084],
        [  9.5261,  -7.1675,   1.4206],
        [  1.7232,  -4.7574,   3.1270],
        [  3.3943,   6.5483,   1.0790],
        [ -5.1398,  -9.1367,   6.2074],
        [  0.8572,  14.0501,  -8.3364],
        [  2.2182,  13.8007,   5.1889],
        [ -8.0611,  -6.0302,   4.6962],
        [ 11.1469,   6.9237,   1.6160],
        [  0.9074,  -4.8139,   8.1674],
        [  6.0560,  -0.9421,  -5.0334],
        [ -8.0608,   3.9362,  -5.9567],
        [  5.4867,  12.5816,   2.9549],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -0.5385,   7.9037,   6.3830],
        [  0.3231,  -4.7691,  -1.6120],
        [  5.1787,  -2.6383,  -4.2533],
        [  3.9913,  -3.8139,  -6.8311],
        [  4.3523, -10.5232,   8.2138],
        [ -5.9940,   6.1985,   5.9566],
        [  6.3301,   0.3638,  -7.5685],
        [ -1.9646,   7.8571,   3.3817]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0345, 0.0316,  ..., 0.4315, 0.8503, 0.3125],
        [0.0345, 0.0000, 0.0249,  ..., 0.4644, 0.8585, 0.3460],
        [0.0316, 0.0249, 0.0000,  ..., 0.4603, 0.8343, 0.3300],
        ...,
        [0.4315, 0.4644, 0.4603,  ..., 0.0000, 0.9900, 0.2413],
        [0.8503, 0.8585, 0.8343,  ..., 0.9900, 0.0000, 0.7576],
        [0.3125, 0.3460, 0.3300,  ..., 0.2413, 0.7576, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.91336994,  5.14607014,  5.14124612,  5.16205087,  5.51177501,
        5.29755487, 11.22596994,  6.80370254,  9.36093562,  7.27759233,
        8.67502348, 12.35625694, 13.99048094, 12.50996828, 10.06850677,
        7.13501264, 14.0393949 ,  5.88782913,  6.00663314,  4.74468406,
       12.76408562, 10.41909211, 13.96905756,  5.94293185, 11.11038551,
        7.80625981,  5.01114719,  5.43544626, 11.80092578,  9.93553424,
       12.18045978,  5.48276038,  9.3109884 , 13.60275211,  5.81199422,
       13.32613445,  6.80370254,  7.92644746, 12.13568598,  7.88677538,
        7.66377554, 11.018084  ,  4.85995002,  8.86082124, 10.33958639,
        8.62260327, 10.84437705, 13.92390689,  7.96575782,  9.7712321 ,
        8.36445758, 10.30972193,  9.32094055,  5.34250225, 12.30695098,
        9.6282763 ,  9.99145096, 11.75618513, 13.50484539, 13.68063098,
        5.60064319, 12.72524615,  5.53175275,  6.61386436, 11.05672136,
        9.76896904,  9.32094055,  8.91551231,  6.18668463,  8.89458273,
       12.35421461, 10.14436912,  7.80625981, 10.7167285 , 10.794755  ,
       13.31764181, 12.72524615,  8.04483495,  9.58072286, 11.05317597,
        9.14973319, 11.59512194, 10.09979747,  8.09984231, 11.92133129,
        5.05989547,  7.37576154, 10.48315436, 14.07623266, 13.97779839,
       10.06703358, 13.12218013,  4.89869205,  6.12886024,  8.97052115,
       13.72594022,  7.58668144,  9.97964013,  7.92200342,  4.78002665,
        5.81199422,  5.5205189 , 11.38774149,  8.62260327,  6.34059021,
        8.09899874])}
{'P': tensor([[2.7556e-02, 1.3080e-01, 1.0088e-01,  ..., 7.9681e-04, 6.6536e-04,
         1.9721e-05],
        [2.9975e-03, 2.9222e-02, 9.9012e-02,  ..., 1.1857e-03, 8.7830e-04,
         2.7589e-05],
        [2.4781e-03, 4.4545e-03, 3.8419e-02,  ..., 1.5073e-03, 8.8009e-04,
         5.5918e-05],
        ...,
        [9.0190e-06, 1.3017e-05, 1.7628e-05,  ..., 1.0191e-02, 4.2339e-04,
         2.8851e-05],
        [6.0156e-04, 7.9655e-04, 8.1053e-04,  ..., 1.0133e-02, 1.5945e-01,
         1.4645e-04],
        [4.6666e-05, 3.8107e-05, 3.4397e-05,  ..., 9.7187e-07, 7.9846e-06,
         2.4294e-01]]), 'X': tensor([[ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0285, 0.0424,  ..., 0.6590, 0.5872, 0.4768],
        [0.0285, 0.0000, 0.0410,  ..., 0.6386, 0.5637, 0.4742],
        [0.0424, 0.0410, 0.0000,  ..., 0.6752, 0.5968, 0.4361],
        ...,
        [0.6590, 0.6386, 0.6752,  ..., 0.0000, 0.1271, 1.0172],
        [0.5872, 0.5637, 0.5968,  ..., 0.1271, 0.0000, 0.9078],
        [0.4768, 0.4742, 0.4361,  ..., 1.0172, 0.9078, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.9160962 ,  5.07306274,  5.24137909,  5.26342203,  5.30158497,
        5.21280156,  5.26067019, 10.53368644, 12.18998195, 12.27991991,
        7.58378655,  9.76896904,  9.19837792, 10.37998078, 13.06498554,
        5.69913817, 11.51557166,  8.89301516, 12.33167382,  5.1224079 ,
       10.01804223,  6.36004393,  9.17016857,  9.09661212,  6.83515201,
        9.23178204, 13.56367857, 10.09979747,  6.19633139, 13.34587027,
       11.11038551, 12.52875793, 11.37939695,  8.43626114,  5.38342077,
        8.01009083, 12.91505111,  9.41970031, 13.14777433,  4.91296205,
        5.26342203,  7.13501264, 11.83509372,  6.58184474,  7.83534987,
       12.4980845 , 13.45124917, 10.41909211, 12.00995898, 10.51340829,
       11.83570132,  9.69629   ,  9.68324553,  7.96181288, 11.00694688,
        5.19577146,  7.66377554,  5.63824971,  7.63726134, 12.5085995 ,
        7.72291946, 11.64436684,  5.90035324, 10.30972193,  8.37954853,
        7.4544866 ,  8.62162372,  6.83515201,  6.58184474,  6.38105015,
       12.11735334, 13.14777433,  5.46244988, 10.41909211, 10.01804223,
        9.50913849, 10.91601655,  9.11473391,  8.35856487,  7.34250948,
       12.9089559 ,  5.26067019,  6.34059021,  4.73784222, 13.66991027,
        5.20659892, 10.24557047,  6.35029274, 13.89974693,  7.78980299,
        5.63824971, 12.3344536 ,  9.48969978,  7.34250948,  8.87587457,
        6.38968357, 10.51340829,  5.70591986,  6.33517349,  5.21280156,
       13.91198191,  6.65259775,  6.652742  , 12.24467041,  5.91222346,
       13.89671314])}
{'P': tensor([[5.0715e-02, 1.8434e-01, 1.3122e-01,  ..., 3.4208e-05, 1.6132e-05,
         3.7751e-03],
        [3.3708e-03, 5.2584e-02, 1.6164e-02,  ..., 2.2097e-05, 1.2588e-05,
         8.8332e-03],
        [2.9711e-03, 5.9713e-02, 3.2984e-02,  ..., 1.7214e-05, 9.6801e-06,
         5.4114e-03],
        ...,
        [3.0886e-04, 2.2449e-04, 3.0549e-04,  ..., 2.8566e-02, 1.5563e-04,
         1.6477e-05],
        [1.4554e-04, 9.7800e-05, 1.3328e-04,  ..., 2.8353e-01, 3.9631e-02,
         1.1498e-05],
        [5.6659e-04, 6.6398e-04, 6.6674e-04,  ..., 5.1836e-05, 7.3590e-05,
         1.3005e-01]]), 'X': tensor([[  0.9967,  -4.8144,  -9.7370],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.4140,  -4.1803,  -8.8224],
        [  2.5829,  -4.5466,  -8.4015],
        [  3.7992,  -3.9474,  -8.9019],
        [ -2.0229, -10.6163,   8.6930],
        [ -5.6792,  -8.7352,   4.0318],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.4867,  12.5816,   2.9549],
        [  0.6142,  14.0802,   0.5407],
        [ -9.2373,  -3.1225,  -8.9763],
        [  7.5784,  -8.9141,   5.8410],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.0054,  -1.0902,   8.3079],
        [ -4.2040,   7.2486,   2.9875],
        [  2.1120,  -4.6818,   8.6144],
        [  3.3839,  -4.1966,   8.3574],
        [  0.6470, -11.0352,  -1.1931],
        [ -2.6485, -10.4151,   0.8766],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.4691,   2.8525,  -5.5110],
        [ -2.9497, -10.3028,  -7.3479],
        [ 11.1469,   6.9237,   1.6160],
        [ -4.7389,   6.9953,   3.7638],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -7.9667,   4.0814,  -0.6397],
        [ -5.0550,   6.8231,  -9.2502],
        [ -9.4351,  -1.8891,   3.5642],
        [  3.1064,  13.5607,   6.7605],
        [ -4.2040,   7.2486,   2.9875],
        [ -1.0012, -10.8576,   2.2867],
        [ -6.6895,   5.6051,   2.7899],
        [  6.2804,   2.0019,   0.8607],
        [ 10.4737,  -5.9570,  -1.2108],
        [  2.0937,  13.8293,  -2.3533],
        [  8.8091,  -7.9037,   0.3631],
        [ -5.9225,  -8.5355,   2.4778],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -1.6906,   7.8860,   7.7142],
        [ -8.0829,  -5.9940,   5.0564],
        [  2.2076,  13.8032,  -1.5099],
        [  9.4554,   9.4245,   9.8800],
        [  8.3718,  -8.2953,   8.4666],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.5506,  -5.1154,   6.3423],
        [  0.1041, -11.0045,   2.4311],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.9526,   3.2525,  -4.3821],
        [ 11.0438,   7.1180,  -5.6991],
        [  1.8766,   7.3694,   1.1887],
        [  0.4119,  14.1019,  -4.1870],
        [ 10.1798,   8.5002,  -5.8893],
        [  5.4426, -10.1181,  -5.1268],
        [ -6.0844,  -8.3955,   0.7705],
        [ 11.0373,   7.1299,  -8.5895],
        [ -5.2752,  -9.0409,  -8.2568],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.6451,  -4.5250,  -0.5565],
        [  9.9756,  -6.6330,  -3.3429],
        [  6.3583,   1.1729,   8.7835],
        [  6.0073,   3.0986,   6.7937],
        [ 12.0582,  -2.6640,   0.7132],
        [  9.1820,   9.7334,  -2.7932],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.3087,   0.7792,  -7.5587],
        [ 11.9122,   5.1397,  -1.5498],
        [  1.1006,   7.6318,  -2.4877],
        [  5.1472,  -2.6801,   8.1194],
        [  6.2036,   2.4030,  -1.2500],
        [  3.4920,  -4.1366,  -8.3955],
        [ 10.6283,  -5.7253,   7.4878],
        [ 12.1940,  -2.1881,   7.1499],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.1989,  -9.7591,   0.8289],
        [ 12.4597,  -0.9136,  -4.8763],
        [  9.3753,   9.5171,   3.9131],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -8.0829,  -5.9940,   5.0564],
        [ -1.2454,   7.9127,  -0.5715],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.1178,   1.5784,   2.1410],
        [ 11.4495,   6.3026,   6.8257],
        [ -0.4341, -10.9473,   7.4509],
        [ -5.0943,  -9.1682,   0.5073],
        [  3.9895,   6.0816,  -6.5235],
        [ -7.4733,   4.7555,   6.2490],
        [  6.1741,  -0.5329,   3.1913],
        [  6.2022,   2.4092,   3.0349],
        [ -5.1377,   6.7751,  -0.4438],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.0608,   3.9362,  -5.9567],
        [ -2.5056,   7.7720,   5.5878],
        [  5.7151,  -1.7700,   3.1542],
        [  3.6154,  -4.0640,  -7.9698],
        [ -8.5859,  -5.0393,   0.5476],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.1372,   2.6726,  -6.7885]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0347, 0.0244,  ..., 0.9912, 1.0194, 0.0811],
        [0.0347, 0.0000, 0.0123,  ..., 1.0229, 1.0486, 0.0520],
        [0.0244, 0.0123, 0.0000,  ..., 1.0106, 1.0364, 0.0575],
        ...,
        [0.9912, 1.0229, 1.0106,  ..., 0.0000, 0.1571, 1.0449],
        [1.0194, 1.0486, 1.0364,  ..., 0.1571, 0.0000, 1.0633],
        [0.0811, 0.0520, 0.0575,  ..., 1.0449, 1.0633, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.91652556,  5.24461068,  5.19065439,  5.39723241,  5.22903117,
        5.47866221, 10.80728788, 10.41909211, 12.10942412, 13.72594022,
       14.09357547,  9.75075273, 11.70016217, 11.72749923,  6.10360073,
        8.37954853,  5.13616224,  5.39098669, 11.05413545, 10.74655346,
       12.00995898, 12.27991991,  6.38968357, 12.79126785, 10.7167285 ,
       13.12218013,  8.44940661, 10.53368644,  8.95133961,  8.49161154,
        9.62238828, 13.91198191,  8.37954853, 10.9036226 ,  8.7273569 ,
        6.59176018, 12.04924159, 13.98691426, 11.83509372, 10.38897563,
       11.92133129, 10.00410625,  8.06517143, 10.06286074, 13.97857591,
       13.35013374, 11.785566  , 10.60226614, 12.1494023 ,  9.96391045,
       11.0050379 ,  5.41348121,  6.78325151, 13.13888841,  7.60463349,
       14.10796466, 13.26209545, 11.4890861 , 10.36844813, 13.1399181 ,
       10.46739073,  4.78258092,  5.24137909, 11.97957425,  6.46560181,
        6.7593917 , 12.3489324 , 13.38091121, 12.26997165,  6.83083863,
        9.34127013, 12.97370371,  7.71075226,  5.80312949,  6.652742  ,
        5.41348121, 12.07225469, 12.38881635, 10.52357479, 10.00802627,
       11.56146873, 12.49317563, 13.35927527,  9.43737523, 10.06286074,
        8.01009083, 12.58006038,  9.25336371, 13.06957425, 10.95593729,
       10.48840476,  7.27340834,  8.8580671 ,  6.19707944,  6.65368551,
        8.50278896,  5.91854397,  9.69792513, 10.30972193,  8.97052115,
        8.16584537,  5.98285319,  5.43943791,  9.95552626,  9.60412735,
        6.69389431])}
{'P': tensor([[3.3666e-02, 1.7778e-01, 1.2690e-01,  ..., 5.9610e-05, 1.0420e-03,
         1.4264e-02],
        [4.1480e-03, 4.2175e-02, 2.3086e-01,  ..., 5.4856e-05, 1.6026e-03,
         2.3420e-02],
        [5.2639e-03, 6.8234e-03, 8.6772e-02,  ..., 7.1714e-05, 3.5692e-03,
         5.2916e-02],
        ...,
        [5.7439e-04, 6.1515e-04, 5.3728e-04,  ..., 2.9073e-02, 6.3116e-04,
         6.7864e-04],
        [9.9700e-04, 1.2422e-03, 1.5856e-03,  ..., 1.3466e-04, 2.6397e-01,
         1.2120e-02],
        [9.9785e-04, 1.3174e-03, 2.0156e-03,  ..., 2.8289e-05, 1.0348e-02,
         5.3260e-02]]), 'X': tensor([[  1.0391,  -4.8141,  -0.0768],
        [  1.7903,  -4.7467,   0.1168],
        [  2.7558,  -4.4844,   0.0636],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.8721,  -4.7322,   0.5524],
        [  2.9461,  -4.4077,  -0.4831],
        [ -3.1671,   7.6158,   6.0765],
        [ -3.3812, -10.1235,   7.4384],
        [  2.0729,   7.2878,   7.0880],
        [ 11.6544,   5.8273,  -3.1375],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.6363, -11.0348,   6.1414],
        [  5.9026,   3.3837,  -0.6573],
        [ -2.1817,   7.8274,  -4.6298],
        [  5.8318,   3.5567,  -4.7692],
        [  3.9515,  -3.8425,   6.1729],
        [  9.5203,   9.3483,   5.8781],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.1666,   3.7653,   3.6075],
        [  1.1274,  14.0116,   8.1753],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -7.4807,   4.7463,   2.0616],
        [  5.7963,  -1.6018,   1.8856],
        [  8.1528,  10.7452,   9.1493],
        [ -6.2668,   5.9808,   2.5614],
        [  4.1297,  -3.7098,  -3.9838],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.4887,  -5.2448,  -0.6585],
        [  2.1771, -10.9776,  -3.2007],
        [  5.1793,   4.7461,   0.3899],
        [  7.1697,  11.5343,  -0.7276],
        [ -2.5161,   7.7699,  -9.3531],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.2849, -10.9655,  -7.6542],
        [  8.0877,  -8.5297,  -1.1050],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -7.6650,   4.5092,   0.2520],
        [ -1.1885, -10.8211,   0.0865],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.3354,  -4.7708,  -4.9125],
        [  9.5261,  -7.1675,   1.4206],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.7802,  -1.6363,  -6.2782],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.1813,  -3.3550,   1.8827],
        [  3.6817,   6.3354,   6.1233],
        [  5.4429,  -2.2529,   9.7306],
        [  6.0047,   3.1062,  -6.7914],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.2911,  -2.4824,  -3.6877],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -8.0766,  -6.0046,  -3.1873],
        [  5.0348,  -2.8229,   3.5436],
        [ -8.0087,   4.0173,   9.1091],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.3246,   1.6696,  -0.1544],
        [ 12.3357,   3.6001,   0.7903],
        [  6.0305,   3.0296,   0.1375],
        [  5.8323,  -1.5228,   5.1132],
        [  3.7992,  -3.9474,  -8.9019],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.0560,  -0.9421,  -5.0334],
        [ -4.7334,   6.9982,  -0.4392],
        [  5.9526,   3.2525,  -4.3821],
        [ -3.0644,   7.6439,  -1.4322],
        [  5.5158,  -2.1337,   3.5246],
        [  2.3200,   7.1756,   1.1938],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -7.0948,   5.1940,   9.8171],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.9526,   3.2525,  -4.3821],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.6426,  -4.9136,   9.4941],
        [ -3.5903,   7.4847,  -8.9112],
        [ 10.6283,  -5.7253,   7.4878],
        [ 10.1798,   8.5002,  -5.8893],
        [ -2.0229, -10.6163,   8.6930],
        [  7.1821,  -9.1845,   9.8548],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.8447,  -4.7372,   8.1235],
        [  4.4915,   5.5990,   7.0405],
        [  5.7784,  -1.6401,   9.5587],
        [  8.9284,  10.0036,  -6.9210],
        [ 10.5057,  -5.9099,  -5.1599],
        [  7.5784,  -8.9141,   5.8410],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.1519, -10.2387,   9.8191],
        [ -7.2515,   5.0195,   4.6434],
        [  3.3633,  13.4794,   0.5193],
        [ -1.6334, -10.7207,   7.1397],
        [  1.9204,   7.3518,   8.4781],
        [ 10.5678,   7.9261,  -0.6100],
        [  2.0039,   7.3173,  -1.5885],
        [  8.5716,  10.3596,   6.8818],
        [  5.4242, -10.1261,   3.9595],
        [  0.3715,  -4.7756,   0.2312],
        [ -4.2040,   7.2486,   2.9875],
        [  6.2576,   2.1359,  -3.3398],
        [ -5.3179,  -9.0101,   9.4472],
        [  5.8889,  -1.3916,   8.8993],
        [  5.0348,  -2.8229,   3.5436]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0227, 0.0432,  ..., 0.9838, 0.5760, 0.2896],
        [0.0227, 0.0000, 0.0220,  ..., 0.9902, 0.5595, 0.2716],
        [0.0432, 0.0220, 0.0000,  ..., 1.0064, 0.5540, 0.2645],
        ...,
        [0.9838, 0.9902, 1.0064,  ..., 0.0000, 0.8877, 0.9235],
        [0.5760, 0.5595, 0.5540,  ..., 0.8877, 0.0000, 0.2901],
        [0.2896, 0.2716, 0.2645,  ..., 0.9235, 0.2901, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.92497514,  5.07306274,  5.26342203,  5.24137909,  5.08910491,
        5.30158497,  8.24808832, 10.67322527,  7.57685908, 13.03002338,
        6.14086582, 11.05317597,  6.80370254,  8.12580976,  6.83083863,
        5.51177501, 13.3426541 , 10.52357479,  8.99276564, 14.05687815,
        5.47702766, 12.49291768,  8.85939626,  6.01356004, 13.48809595,
        8.66273171,  5.5513072 , 13.94008948,  9.97822473, 11.1913598 ,
        7.02496223, 13.58101794,  8.16714836, 12.7125079 , 11.20100795,
       11.75438006,  8.22728399,  9.97964013,  8.89301516, 10.88618114,
        8.62631801,  4.78258092, 11.92133129,  7.26074925,  9.97220799,
        6.00732193,  5.21280156,  9.7751189 ,  7.32754882,  5.89074104,
        6.76056625,  9.62238828,  5.84450127, 12.1460931 , 10.06407839,
        5.77219077,  8.95983586, 12.9089559 ,  6.54127431, 12.85032836,
        6.74872981,  6.02779201,  5.47866221, 12.3489324 ,  6.12886024,
        8.4486771 ,  6.78325151,  8.23525433,  5.91408504,  7.54127425,
        5.91677188, 10.06407839,  8.79284466,  4.98461983,  6.78325151,
        5.43943791, 12.62576026,  9.94173708,  8.30125201, 12.07225469,
       13.26209545, 10.80728788, 11.65923396,  9.38232121,  5.08372149,
        7.17789666,  6.00663314, 13.40850328, 12.05395056, 11.70016217,
       12.32545637, 11.46176259,  8.81930525, 13.89264822, 10.84437705,
        7.5984809 , 13.20988996,  7.58668144, 13.44593519, 11.48734978,
        4.79003361,  8.37954853,  6.61211224, 10.46238054,  6.05113217,
        5.77219077])}
{'P': tensor([[3.3052e-02, 2.1354e-01, 1.8085e-01,  ..., 1.4614e-05, 1.7055e-05,
         2.2866e-03],
        [3.9020e-03, 3.7450e-02, 1.9142e-01,  ..., 1.7685e-05, 1.8834e-05,
         3.1515e-03],
        [6.1280e-03, 9.5079e-03, 6.2983e-02,  ..., 2.8733e-05, 3.0283e-05,
         5.6960e-03],
        ...,
        [1.3650e-05, 1.4175e-05, 1.3698e-05,  ..., 1.2088e-01, 2.9969e-04,
         3.4253e-05],
        [1.8860e-04, 1.5791e-04, 1.3716e-04,  ..., 2.2334e-05, 7.0699e-02,
         3.7178e-06],
        [3.0870e-04, 3.7146e-04, 4.1064e-04,  ..., 4.8666e-04, 4.0440e-05,
         2.5057e-01]]), 'X': tensor([[  1.2273,  -4.8081,   9.8129],
        [  1.6954,  -4.7616,   9.8198],
        [  1.9505,  -4.7171,   9.9868],
        [  2.0870,  -4.6877,   9.6823],
        [  2.2644,  -4.6434,   9.7508],
        [  2.3376,  -4.6231,   9.8436],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.8363,   2.4086,  -5.6296],
        [  2.9410,   6.8428,   5.4458],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.3048,   0.1469,  -9.1691],
        [  6.9433,  -9.3362,  -0.3421],
        [ 12.5154,   2.5072,  -9.3274],
        [  9.9217,  -6.7006,  -5.6728],
        [ 11.9122,   5.1397,  -1.5498],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.5428,  -9.5728,  -2.9897],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -8.6603,   2.8250,   5.9181],
        [ -8.5900,   2.9767,   7.4811],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 10.5057,  -5.9099,  -5.1599],
        [  0.9788,  -4.8145,  -3.8661],
        [  7.7794,  11.0629,   8.9069],
        [  3.0876,   6.7527,  -5.3540],
        [  2.7295,  13.6703,  -8.9328],
        [  4.9957,  -2.8705,   6.5978],
        [  0.7275,  14.0667,  -0.9815],
        [  2.7420,  -4.4896,  -3.7139],
        [  8.5716,  10.3596,   6.8818],
        [  4.7156,  -3.1831,  -8.0224],
        [  2.8890,  -4.4316,   1.1600],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.6112,   2.9319,  -7.9943],
        [  0.9788,  -4.8145,  -3.8661],
        [ -1.9487,   7.8590,  -3.3330],
        [  2.5128, -10.9365,   9.9645],
        [ -9.3616,  -2.4749,   0.1793],
        [ 10.0567,   8.6697,   4.6329],
        [  1.1006,   7.6318,  -2.4877],
        [  4.0421,  -3.7765,  -8.3631],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.3616,  -2.4750,   3.7765],
        [  6.3205,   1.7068,  -3.8619],
        [  5.6414, -10.0303,   7.8525],
        [  1.0391,  -4.8141,  -0.0768],
        [ -2.0840,   7.8415,  -7.8657],
        [ 12.4996,   2.6335,  -6.3166],
        [  1.6408,   7.4591,  -4.2294],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.3205,   1.7068,  -3.8619],
        [ -7.6293,  -6.6859,  -5.5161],
        [  3.2701,   6.6338,   9.9209],
        [ -1.3940, -10.7772,   6.7835],
        [  4.1637,  13.1898,  -1.9189],
        [ -5.1377,   6.7751,  -0.4438],
        [ -3.1204, -10.2345,   9.2288],
        [  9.6662,   9.1725,   8.1056],
        [  1.7500,   7.4187,   4.4237],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -2.6485, -10.4151,   0.8766],
        [  4.4915,   5.5990,   7.0405],
        [  6.3542,   0.6934,  -3.3840],
        [ -7.5401,   4.6718,   9.0033],
        [  4.3377,   5.7570,   1.3712],
        [  6.3589,   0.8239,  -8.4230],
        [  2.9470,  13.6085,   2.9809],
        [  1.9241,   7.3503,   7.9004],
        [  5.2199,   4.6861,  -1.9304],
        [  6.3516,   0.6425,   0.3425],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.1519, -10.2387,   9.8191],
        [  1.8320,   7.3871,   5.1452],
        [  5.3858,   4.4260,  -9.6769],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.7482, -10.3907,   5.9149],
        [ 12.5832,   1.7543,  -7.6134],
        [ -1.3178, -10.7939,   0.4236],
        [  3.9085,   6.1512,  -2.2560],
        [  1.1350,  14.0104,   0.6321],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -3.9456,  -9.8542,   0.7466],
        [ -1.2454,   7.9127,  -0.5715],
        [  3.2366,  -4.2732,   5.1532],
        [  5.7151,  -1.7700,   3.1542],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.4671,  -0.5553,   0.1117],
        [  4.1537,   5.9340,  -6.0423],
        [ -9.3616,  -2.4749,   0.1793],
        [  9.4925,   9.3811,   8.9842],
        [  1.7000,  -4.7609,   2.7935],
        [  2.8890,  -4.4316,   1.1600],
        [  3.6249, -10.7255,   5.6605],
        [  4.9337,  12.8566,  -3.6582],
        [ -6.6998,   5.5954,   0.0823],
        [ -7.6650,   4.5092,   0.2520],
        [ -2.6485, -10.4151,   0.8766],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.0424,  -7.6779,  -7.5286],
        [  6.3303,   1.6145,  -8.4083],
        [ -7.6650,   4.5092,   0.2520],
        [ 11.8951,  -3.1570,   7.0742]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0103, 0.0167,  ..., 0.9648, 0.8223, 0.1183],
        [0.0103, 0.0000, 0.0069,  ..., 0.9648, 0.8305, 0.1168],
        [0.0167, 0.0069, 0.0000,  ..., 0.9617, 0.8334, 0.1194],
        ...,
        [0.9648, 0.9648, 0.9617,  ..., 0.0000, 0.7996, 1.0810],
        [0.8223, 0.8305, 0.8334,  ..., 0.7996, 0.0000, 0.9152],
        [0.1183, 0.1168, 0.1194,  ..., 1.0810, 0.9152, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.96231532,  5.05444288,  5.10446653,  5.13125513,  5.16611782,
        5.1805345 ,  7.53063885,  9.15866607,  7.44805901, 13.26590071,
        6.30647746, 11.63502679, 12.76408562, 11.97238624, 12.97370371,
        4.78002665, 11.59512194, 12.11735334,  9.10945678,  9.09119351,
       10.5575601 , 12.05395056,  4.91296205, 13.52430176,  7.42512621,
       13.94008948,  5.76165934, 14.08549802,  5.26067019, 13.44593519,
        5.68942572,  5.29009836, 11.97957425,  9.97822473, 10.86693708,
        9.09661212,  4.91296205,  8.09704052, 11.22142195,  9.68323686,
       13.2778378 ,  7.71075226,  5.53175275, 10.33958639,  5.24137909,
        9.68324553,  6.54693987, 11.50791859,  4.92497514,  8.1137452 ,
       12.77402351,  7.6374603 , 10.794755  ,  9.37296102,  6.54693987,
       10.14436912,  7.39599986, 10.86693708, 13.83139024,  8.50278896,
       10.69963954, 13.32557133,  7.62232307, 10.33712258, 10.74655346,
        7.17789666,  6.39187285,  8.87007495,  7.20826219,  6.41202986,
       13.92390689,  7.59795445,  7.01475376,  6.38399789,  8.92395381,
       11.46176259,  7.61088125,  6.97106912, 11.51811603, 11.42420508,
       12.70489563, 10.87409005,  7.28793663, 14.05633441, 12.18045978,
       10.61472636,  8.01009083,  5.36061974,  5.98285319,  5.56809335,
        9.48337063,  7.24326992,  9.68323686, 13.34587027,  5.05534609,
        5.29009836, 11.32149259, 13.77075174,  8.72896381,  8.89301516,
       10.74655346,  5.18277858, 11.86239961,  6.53290663,  8.89301516,
       12.30695098])}
{'P': tensor([[3.4801e-02, 1.6852e-01, 1.6822e-01,  ..., 1.1723e-03, 4.1645e-05,
         3.6970e-03],
        [5.1747e-03, 6.9536e-02, 5.5156e-03,  ..., 1.1105e-03, 1.4242e-04,
         1.2744e-02],
        [7.0793e-03, 2.1146e-01, 4.0658e-02,  ..., 1.2851e-03, 5.1327e-05,
         4.5646e-03],
        ...,
        [1.7461e-03, 3.5213e-03, 2.0032e-03,  ..., 4.3335e-02, 5.0105e-06,
         4.8821e-04],
        [4.0518e-06, 3.5518e-06, 3.9067e-06,  ..., 1.1690e-07, 2.7060e-02,
         2.9752e-05],
        [5.9241e-04, 6.5954e-04, 5.9388e-04,  ..., 2.4308e-05, 1.9579e-02,
         1.2161e-01]]), 'X': tensor([[  1.3103,  -4.8032,  -0.9300],
        [  2.5010,  -4.5736,  -0.8989],
        [  1.4749,  -4.7892,  -0.8089],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.1685,  12.1979,  -6.2473],
        [ -1.6086,   7.8928,   4.3385],
        [ -7.8359,   4.2735,   7.2943],
        [ -9.3444,   0.5821,   6.9101],
        [  5.1519, -10.2387,   9.8191],
        [ -9.4296,  -1.9458,   4.0833],
        [  2.2076,  13.8032,  -1.5099],
        [  7.7794,  11.0629,   8.9069],
        [  4.9452, -10.3187,  -7.8621],
        [  1.3705,  13.9723,   7.9266],
        [ 12.4013,  -1.2597,  -7.6474],
        [  9.2979,   9.6048,   7.3138],
        [  2.5128, -10.9365,   9.9645],
        [  0.6933,   7.7336,   2.1179],
        [  5.9548,  -1.2273,  -2.6520],
        [ -3.3116, -10.1539,   7.5919],
        [ -4.0868,   7.2981,   8.1874],
        [  5.8325,  -1.5222,  -2.1341],
        [ -5.1377,   6.7751,  -0.4438],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.4572,   7.1085,   7.7376],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.0574,   1.7801,   8.8986],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.2038,   2.4020,  -3.1692],
        [ -8.1905,   3.7253,   4.1175],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.6396,  14.0772,  -6.3786],
        [  4.4915,   5.5990,   7.0405],
        [  5.9655,   3.2172,  -7.5953],
        [ 10.9599,  -5.1849,  -3.5526],
        [  4.9337,  12.8566,  -3.6582],
        [  2.4572,   7.1085,   7.7376],
        [ -5.6174,   6.4708,  -1.7339],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.1795,  -3.3623,   8.5259],
        [ 11.3800,  -4.3919,   1.4066],
        [ 10.9717,  -5.1646,  -6.9039],
        [  7.8425,  -8.7202,   6.4530],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.1504,   4.3674,  -7.2998],
        [  5.3924, -10.1396,  -4.0863],
        [  3.9515,  -3.8425,   6.1729],
        [  6.0305,   3.0296,   0.1375],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.6672,  -1.8631,   8.1020],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.2345,  -3.1345,   3.9715],
        [  2.6614,  -4.5192,  -9.3641],
        [  0.7145,  -4.8071,  -2.3766],
        [ -9.4716,  -0.6673,   7.6068],
        [ -2.3317,   7.8034,  -3.6274],
        [  7.7794,  11.0629,   8.9069],
        [  1.5653,  13.9377,   4.1773],
        [  3.9770, -10.6341,   8.0783],
        [  3.5939,   6.4029,  -1.8357],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.8093,  -9.4178,   8.4244],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.3602,  -2.3808,   7.2098],
        [  2.6321,  -4.5296,  -2.5159],
        [  2.1824,  -4.6647,   1.1985],
        [  2.1440, -10.9811,   8.5281],
        [ -9.3076,   0.7852,  -6.7010],
        [ -7.4808,   4.7462,   2.0266],
        [ 12.4691,   2.8525,  -5.5110],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.9284,  11.7061,  -0.3732],
        [  6.7565,  11.8237,  -4.9882],
        [  2.1504,  -4.6726,   8.7908],
        [  7.0515,  11.6194,   8.4289],
        [ -2.8162,   7.7059,  -9.6885],
        [  9.1820,   9.7334,  -2.7932],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.0890,  -0.8382,   2.9166],
        [  1.9241,   7.3503,   7.9004],
        [  1.1139,   7.6281,  -2.5680],
        [  0.1047, -11.0046,   0.5188],
        [  2.2437,  -4.6489,   6.1430],
        [ -6.7992,   5.4991,   9.9186],
        [ -4.7334,   6.9982,  -0.4392],
        [  2.6614,  -4.5192,  -9.3641],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.7218,  -9.9934,   6.5764],
        [  5.0617,  12.7957,   8.3544],
        [ -2.5161,   7.7699,  -9.3531],
        [  9.9756,  -6.6330,  -3.3429],
        [  2.6310, -10.9195,   3.8054],
        [  5.6928,  12.4710,   3.6001],
        [  4.2041,  -3.6509,  -4.9488],
        [ -1.6275, -10.7221,   3.2376],
        [ -3.2736, -10.1703,   3.8712],
        [ -3.4986,   7.5152,   9.0712],
        [  4.1999,  -3.6543,  -4.1998],
        [  2.6310, -10.9195,   3.8054],
        [  6.1685,  12.1979,  -6.2473],
        [  6.2038,   2.4020,  -3.1692]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0282, 0.0093,  ..., 0.6113, 0.5429, 0.1782],
        [0.0282, 0.0000, 0.0222,  ..., 0.6286, 0.5294, 0.1682],
        [0.0093, 0.0222, 0.0000,  ..., 0.6105, 0.5449, 0.1812],
        ...,
        [0.6113, 0.6286, 0.6105,  ..., 0.0000, 1.1521, 0.7867],
        [0.5429, 0.5294, 0.5449,  ..., 1.1521, 0.0000, 0.3655],
        [0.1782, 0.1682, 0.1812,  ..., 0.7867, 0.3655, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.978704  ,  5.21280156,  5.01114719,  5.24137909,  5.30158497,
        5.16077764, 13.66896739,  8.0550286 ,  8.92549752,  9.36256096,
       11.46176259,  9.6282763 , 13.97857591, 13.52430176, 11.4424816 ,
       14.0393949 , 12.46513646, 13.36800073, 11.22142195,  7.76457095,
        6.07992623, 10.68031351,  8.36445758,  6.02789634,  8.50278896,
       12.07818871,  7.52117132, 11.97238624,  9.23071698, 11.69264243,
        6.65259775,  8.99791316, 10.30972193, 14.09176259,  7.17789666,
        6.77777606, 12.12448912, 13.77075174,  7.52117132,  8.56889738,
        7.27340834,  9.77588801, 12.19805394, 12.12641969, 11.72802491,
       12.59840193, 12.91143491, 11.48434767,  5.51177501,  6.74872981,
        8.89458273,  5.96556248,  9.49511524,  9.75200848,  5.24461068,
        4.85995002,  9.49511524,  8.14433542, 13.52430176, 14.02532758,
       11.35345345,  7.34250948, 12.4980845 , 12.26997165, 11.62158554,
       11.73848065,  5.86519525,  5.23878371,  5.14999799, 11.18839169,
        9.34061938,  8.859416  , 12.79126785, 10.27716434, 13.60275211,
       13.61801526,  5.14371005, 13.59170809,  8.20436764, 13.38091121,
       12.18045978,  5.19065439,  5.34250225,  6.1463896 ,  7.59795445,
        7.70897675, 11.00508681,  5.16205087,  8.74470218,  8.4486771 ,
        5.24461068, 12.26997165, 11.51557166, 13.76047999,  8.16714836,
       11.97957425, 11.23201619, 13.70893492,  5.56809335, 10.84493732,
       10.68416401,  8.28968067,  5.5671389 , 11.23201619, 13.66896739,
        6.65259775])}
{'P': tensor([[3.5492e-02, 1.6345e-01, 1.3317e-01,  ..., 2.2365e-03, 3.2360e-05,
         8.7063e-03],
        [3.5293e-03, 4.5118e-02, 1.5956e-02,  ..., 2.1740e-03, 7.2949e-05,
         3.8396e-03],
        [6.0355e-03, 7.5228e-02, 5.5715e-02,  ..., 4.1540e-03, 6.3508e-05,
         5.1063e-03],
        ...,
        [4.3058e-03, 5.7590e-03, 7.0628e-03,  ..., 2.5978e-01, 4.9514e-05,
         5.8712e-03],
        [4.0747e-06, 4.3789e-06, 3.6718e-06,  ..., 7.3790e-07, 5.3301e-02,
         2.8944e-06],
        [5.5671e-02, 7.9880e-02, 1.4285e-01,  ..., 9.7922e-03, 2.9150e-05,
         6.4855e-02]]), 'X': tensor([[  1.3403,  -4.8011,  -1.1698],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.2373,  -4.6506,  -1.7334],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.9461,  -4.4077,  -0.4831],
        [  3.8182,  -3.9347,  -1.7961],
        [  9.3424,  -7.3685,  -3.6807],
        [ -4.4503,   7.1378,   3.3751],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.4366,  -0.1189,  -2.6736],
        [  8.7131,  10.2216,   7.7915],
        [ -9.4645,  -1.4880,  -2.0226],
        [  2.3261,   7.1726,   6.2522],
        [  5.7784,  -1.6401,   9.5587],
        [  1.7228,  -4.7575,   8.3396],
        [  7.8982,  -8.6778,   0.0239],
        [ 10.3251,   8.2928,  -3.0006],
        [ 12.5763,   0.1333,  -9.2650],
        [ -6.3066,  -8.1936,   9.5822],
        [ -6.7992,   5.4991,   9.9186],
        [  1.7815,  -4.7481,   4.4289],
        [  0.9809,   7.6642,  -1.4217],
        [ -1.6086,   7.8928,   4.3385],
        [  1.7815,  -4.7481,   4.4289],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.3087,   0.7792,  -7.5587],
        [  3.6249, -10.7255,   5.6605],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -9.3616,  -2.4749,   0.1793],
        [ -8.8745,   2.3097,   3.5600],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -5.5525,   6.5146,  -4.4984],
        [ -7.5603,   4.6459,   8.2002],
        [ -9.4296,  -1.9458,   4.0833],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -5.6792,  -8.7352,   4.0318],
        [  6.3551,   0.7149,   1.3344],
        [  4.5340,  -3.3629,   8.9652],
        [ 11.2209,  -4.7092,   2.6117],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.5966,   5.9666,   6.0083],
        [ -0.0410,   7.8591,   9.3457],
        [  2.5813,  -4.5471,   5.3597],
        [  9.4925,   9.3811,   8.9842],
        [  5.7151,  -1.7700,   3.1542],
        [  2.6614,  -4.5192,  -9.3641],
        [ -3.9700,   7.3453,  -4.1486],
        [ -8.6050,   2.9451,  -0.7012],
        [  5.1861,   4.7361,  -3.7312],
        [ -1.6334, -10.7207,   7.1397],
        [ -8.8154,   2.4611,   2.5198],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.4671,  -0.5553,   0.1117],
        [  1.6408,   7.4591,  -4.2294],
        [ 11.2767,  -4.6006,   0.7868],
        [ -3.9700,   7.3453,  -4.1486],
        [ -9.4739,  -1.2497,   0.5483],
        [ -1.6906,   7.8860,   7.7142],
        [ -5.9872,   6.2037,  -9.4477],
        [ -2.2188,   7.8218,   8.6064],
        [  3.3069,   6.6089,   8.7377],
        [  6.0345,   3.0175,  -1.4424],
        [  2.2849, -10.9655,  -7.6542],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.9260,  -4.4162,   8.3808],
        [  4.2041,  -3.6509,  -4.9488],
        [  2.9461,  -4.4077,  -0.4831],
        [ -6.2668,   5.9808,   2.5614],
        [  5.5424,  -2.0885,   7.2487],
        [  4.8653,   5.1694,  -4.0823],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.1504,   4.3674,  -7.2998],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -6.3489,   5.9117,   6.5161],
        [ -6.2668,   5.9808,   2.5614],
        [ 11.5568,   6.0602,   9.3244],
        [  5.3860,  -2.3418,   1.0826],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.7295,  13.6703,  -8.9328],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.5620,   7.0549,   8.4825],
        [ -1.8823, -10.6558,  -3.8952],
        [ -6.1899,  -8.3011,  -7.9269],
        [  4.3459,  -3.5327,   0.3667],
        [  3.3744,   6.5623,  -6.3636],
        [  1.1274,  14.0116,   8.1753],
        [  4.4049,   5.6892,  -4.4394],
        [  2.5318,  13.7231,   6.3692],
        [  9.0652,   9.8597,  -3.5949],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.4235,   6.3594,   5.1663],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.3539,  -5.5111,   0.3373],
        [  4.1999,  -3.6543,  -4.1998],
        [ -7.6525,  -6.6534,   5.1721],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.1771, -10.9776,  -3.2007],
        [  8.3718,  -8.2953,   8.4666],
        [ -1.9600,   7.8577,   6.9121],
        [ -9.4501,  -1.7133,   2.2980],
        [  3.7992,  -3.9474,  -8.9019],
        [  9.3560,   9.5391,   9.4943],
        [  0.7685,  -4.8098,  -2.9310]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0326, 0.0394,  ..., 0.3367, 1.0088, 0.1073],
        [0.0326, 0.0000, 0.0522,  ..., 0.3494, 0.9794, 0.1315],
        [0.0394, 0.0522, 0.0000,  ..., 0.2994, 1.0277, 0.0812],
        ...,
        [0.3367, 0.3494, 0.2994,  ..., 0.0000, 1.2708, 0.2410],
        [1.0088, 0.9794, 1.0277,  ..., 1.2708, 0.0000, 1.1088],
        [0.1073, 0.1315, 0.0812,  ..., 0.2410, 1.1088, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.98461983,  5.21280156,  5.16077764,  5.24137909,  5.30158497,
        5.48276038, 11.89855267,  8.41148722, 13.27717982,  9.41605684,
        9.43737523, 13.43127131,  9.58072286,  7.5403822 ,  6.00663314,
        5.05981914, 11.73397309, 13.2430377 , 12.5769659 , 10.33958639,
        8.74470218,  5.0713364 ,  7.72668467,  8.0550286 ,  5.0713364 ,
       11.04629471,  9.34127013, 11.32149259, 10.5575601 ,  9.68323686,
        9.17016857, 12.12448912, 12.35625694,  8.55982105,  8.8737556 ,
        9.6282763 , 10.09979747, 10.41909211,  6.39520397,  5.64500512,
       12.16901615, 13.0743399 , 13.04156075,  7.85919687,  5.2287058 ,
       13.34587027,  5.98285319,  5.24461068,  8.34949383,  9.09502385,
        7.02326515, 10.84437705,  9.15253344,  4.89869205,  9.48337063,
        7.6374603 , 12.17901033,  8.34949383,  9.55593374,  8.06517143,
        8.62162372,  8.1303945 ,  7.39004226,  6.74686071, 11.20100795,
        8.25828409,  5.29755487,  5.56809335,  5.30158497,  8.66273171,
        5.92281142,  7.09887557, 12.50996828, 12.91143491,  9.54154096,
        8.67502348,  8.66273171, 13.0493368 ,  5.87305005, 11.75807252,
       13.94008948, 13.04217487,  7.50564223, 10.8207297 , 10.35485331,
        5.60064319,  7.37902766, 14.05687815,  7.19513806, 13.95472331,
       13.39372426, 10.24557047,  6.00732193, 13.0743399 , 11.97238624,
       10.00794468,  5.5671389 , 10.14044489,  9.79814997, 11.1913598 ,
       11.785566  ,  8.09842704,  9.60412735,  5.47866221, 13.3614598 ,
        4.87083348])}
{'P': tensor([[4.2230e-02, 2.5071e-01, 1.1096e-01,  ..., 2.9959e-02, 1.1741e-05,
         1.0381e-04],
        [7.4180e-03, 6.0759e-02, 4.5597e-02,  ..., 4.1414e-02, 2.0006e-05,
         1.5406e-04],
        [1.3842e-02, 5.9322e-02, 5.8981e-02,  ..., 3.3276e-02, 2.1136e-05,
         1.4802e-04],
        ...,
        [1.5574e-03, 1.8203e-03, 1.7499e-03,  ..., 4.2877e-02, 1.4258e-05,
         1.6224e-04],
        [2.5880e-06, 2.5909e-06, 2.7943e-06,  ..., 1.2762e-06, 6.3828e-02,
         2.7284e-04],
        [2.5554e-04, 2.2084e-04, 2.2091e-04,  ..., 8.1496e-05, 1.7068e-06,
         4.5134e-02]]), 'X': tensor([[  1.3670,  -4.7990,   3.2603],
        [  1.7232,  -4.7574,   3.1270],
        [  1.7000,  -4.7609,   2.7935],
        [  3.4604,  -4.1545,   1.8091],
        [  3.2366,  -4.2732,   5.1532],
        [  3.3474,  -4.2162,   5.2455],
        [  5.8318,   3.5567,  -4.7692],
        [ 11.3367,  -4.4806,   0.4643],
        [ -5.9872,   6.2037,  -9.4477],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.1472,  -2.6801,   8.1194],
        [  6.3550,   0.7129,   0.4917],
        [ 12.2592,   3.9425,   2.0383],
        [ -5.0943,  -9.1682,   0.5073],
        [ -7.5720,   4.6310,   2.1717],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.5240,  -2.1198,   6.2055],
        [ -4.2271,   7.2387,  -1.8316],
        [ 12.1940,  -2.1881,   7.1499],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.2790,  -0.0259,  -5.4116],
        [  8.1043,  -8.5164,  -7.9290],
        [ -8.3717,  -5.4769,  -9.2228],
        [  3.9085,   6.1512,  -2.2560],
        [  2.5620,   7.0549,   8.4825],
        [  3.9770, -10.6341,   8.0783],
        [ 11.4703,  -4.2006,   7.4086],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 12.1940,  -2.1881,   7.1499],
        [  0.5850,  -4.7982,   9.6639],
        [  0.6952,  -4.8060,   8.8976],
        [ -0.0410,   7.8591,   9.3457],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.1472,  -2.6801,   8.1194],
        [  5.6248,   3.9984,   8.4933],
        [  6.3515,   0.6418,  -6.5823],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.1749,  -3.3803,   6.4629],
        [  4.9957,  -2.8705,   6.5978],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.4735,  -1.2620,  -3.6742],
        [  7.0664,  -9.2590,   8.0879],
        [  8.7715,  -7.9390,   5.2900],
        [ -3.0004,   7.6607,  -7.2640],
        [  8.9284,  10.0036,  -6.9210],
        [ -6.3489,   5.9117,   6.5161],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.1929,   2.4501,  -6.9352],
        [  6.3542,   0.6934,  -3.3840],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.4366,  -0.1189,  -2.6736],
        [  1.7122,  -4.7591,  -3.6734],
        [  9.3624,   9.5318,   0.2779],
        [ 12.0582,  -2.6640,   0.7132],
        [ -1.6275, -10.7221,   3.2376],
        [  6.0345,   3.0175,  -1.4424],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 11.9808,   4.9340,  -3.3715],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 12.2592,   3.9425,   2.0383],
        [ -8.6675,  -4.8569,  -9.1769],
        [  9.5261,  -7.1675,   1.4206],
        [  3.2647, -10.8066,  -0.3768],
        [  2.4932,   7.0903,  -5.4214],
        [  6.3087,   1.8053,  -1.9481],
        [ -5.6174,   6.4708,  -1.7339],
        [ -6.2668,   5.9808,   2.5614],
        [ -6.7992,   5.4991,   9.9186],
        [  9.3607,   9.5337,  -4.5412],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.9913,  -3.8139,  -6.8311],
        [  3.6249, -10.7255,   5.6605],
        [ -3.6381,   7.4683,   7.3544],
        [  3.3943,   6.5483,   1.0790],
        [  9.6587,   9.1817,  -0.8212],
        [  5.3860,  -2.3418,   1.0826],
        [  3.3839,  -4.1966,   8.3574],
        [  9.0652,   9.8597,  -3.5949],
        [  4.2339,  -3.6268,   6.3125],
        [  0.9478,  -4.8144,   0.6371],
        [  6.2555,   2.1474,   4.1685],
        [ -1.0482, -10.8487,  -5.9066],
        [ -9.4766,  -1.1116,  -2.1706],
        [  9.7328,   9.0902,  -0.8917],
        [  0.8572,  14.0501,  -8.3364],
        [  0.1382, -11.0073,   1.0026],
        [  6.3122,  -9.6994,  -6.3282],
        [  5.8201,   3.5841,  -8.0609],
        [  4.5325,  -3.3643,   6.7699],
        [ -8.1666,   3.7653,   3.6075],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.0587, -10.6112,   6.2273],
        [  8.5716,  10.3596,   6.8818],
        [  3.3069,   6.6089,   8.7377],
        [  9.4516,  -7.2501,  -1.8868],
        [ 12.5258,  -0.4200,  -9.3493],
        [  3.4604,  -4.1545,   1.8091],
        [  4.2339,  -3.6268,   6.3125],
        [  7.0642,  11.6104,  -9.8316],
        [ -4.7389,   6.9953,   3.7638]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4573e-02, 3.7608e-02,  ..., 2.4252e-01, 9.0705e-01,
         3.7153e-01],
        [1.4573e-02, 1.0727e-04, 2.8628e-02,  ..., 2.4628e-01, 8.9491e-01,
         3.8525e-01],
        [3.7608e-02, 2.8628e-02, 0.0000e+00,  ..., 2.7446e-01, 8.6995e-01,
         3.9026e-01],
        ...,
        [2.4252e-01, 2.4628e-01, 2.7446e-01,  ..., 0.0000e+00, 1.0960e+00,
         4.6698e-01],
        [9.0705e-01, 8.9491e-01, 8.6995e-01,  ..., 1.0960e+00, 0.0000e+00,
         1.1675e+00],
        [3.7153e-01, 3.8525e-01, 3.9026e-01,  ..., 4.6698e-01, 1.1675e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 4.98989906,  5.05989547,  5.05534609,  5.40689332,  5.36061974,
        5.38342077,  6.83083863, 12.18998195,  8.62162372,  5.56809335,
        5.80312949,  6.3949043 , 12.87752052, 10.48840476,  8.87587457,
        9.9353921 ,  5.91677188,  8.38252937, 12.38881635,  6.19707944,
        9.5014146 ,  5.79391825,  9.79814997,  6.27905292, 11.75618513,
       10.00410625,  7.28793663,  7.50564223, 11.35345345, 12.21532282,
        9.69792513, 12.38881635,  4.83371341,  4.85603886,  7.85919687,
       10.48721382,  6.27905292,  5.80312949,  6.90115901,  6.383897  ,
        9.76896904,  9.09661212,  9.77777346,  5.76165934, 11.83509372,
        9.55720651, 11.64746882, 11.83075005,  8.22728399, 13.40850328,
        8.67502348,  9.61942445,  6.65991877,  6.39187285, 11.59512194,
        9.43737523,  5.05774295, 13.3607379 , 12.3489324 , 10.84493732,
        6.74686071,  7.53884104,  9.43112957, 12.95702647, 12.4980845 ,
       12.87752052,  9.93553424, 11.92133129, 11.28895489,  7.5158489 ,
        6.5618968 ,  8.56889738,  8.66273171,  8.74470218, 13.36092624,
        5.18047638,  5.5205189 , 11.32149259,  8.30729352,  7.37576154,
       13.32645552,  5.87305005,  5.39098669, 13.39372426,  5.5748624 ,
        4.90676242,  6.61386436, 10.8992524 ,  9.54154096, 13.31764181,
       14.07623266, 11.00812996, 11.57249984,  6.83515201,  5.64464807,
        8.99276564,  9.60412735, 11.36089355, 13.44593519,  7.39004226,
       11.9120321 , 12.53285633,  5.40689332,  5.5748624 , 13.59056613,
        8.44940661])}
{'P': tensor([[3.6620e-02, 1.9046e-01, 1.7102e-01,  ..., 4.7724e-05, 5.3048e-03,
         5.1642e-04],
        [3.9666e-03, 5.0007e-02, 1.0264e-01,  ..., 5.3886e-05, 1.1698e-02,
         1.0290e-03],
        [3.3340e-03, 1.5526e-02, 4.3675e-02,  ..., 4.5298e-05, 1.1507e-02,
         8.5079e-04],
        ...,
        [2.8506e-03, 1.6860e-03, 1.5203e-03,  ..., 9.1224e-02, 6.1302e-05,
         2.9897e-05],
        [4.0587e-04, 4.5273e-04, 4.6830e-04,  ..., 1.2742e-04, 7.4103e-02,
         3.6516e-03],
        [1.6960e-04, 1.7275e-04, 1.6270e-04,  ..., 3.5287e-04, 7.2271e-04,
         9.7421e-02]]), 'X': tensor([[  1.4749,  -4.7892,  -0.8089],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.7558,  -4.4844,   0.0636],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.1426,   3.8048,   9.3542],
        [  4.8432,   5.1968,   1.5006],
        [ -9.3076,   0.7852,  -6.7010],
        [ -9.4505,  -1.7081,  -0.7039],
        [  4.4290,  -3.4596,   1.4181],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.1519, -10.2387,   9.8191],
        [  4.5491,  -3.3485,   6.8412],
        [ -8.1666,   3.7653,   3.6075],
        [  5.8318,   3.5567,  -4.7692],
        [ -7.9925,  -6.1423,   2.8990],
        [ -0.5709,   7.9055,   7.6089],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.3587,   1.1596,  -3.6629],
        [ -7.5603,   4.6459,   8.2002],
        [  2.5974,   7.0363,  -7.7810],
        [  6.9284,  11.7061,  -0.3732],
        [ 10.9394,   7.3069,  -8.9365],
        [ -7.6734,   4.4980,  -5.4248],
        [  3.7121,   6.3117,   0.6738],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.7420,  -4.4896,  -3.7139],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.5318,  13.7231,   6.3692],
        [  6.2902,   0.0451,   3.7116],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.0955,   2.8210,  -4.3321],
        [  9.2979,   9.6048,   7.3138],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.2770,  -0.0379,   2.4195],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.9026,   3.3837,  -0.6573],
        [ -0.3432,   7.8900,   7.1805],
        [ 11.5456,   6.0861,   8.1101],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.6093,  -1.9705,   6.4807],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.2480, -11.0153,  -3.0183],
        [  9.6614,   9.1784,  -1.2087],
        [ -2.3317,   7.8034,  -3.6274],
        [ -2.0840,   7.8415,  -7.8657],
        [ -4.4503,   7.1378,   3.3751],
        [  5.1196,  -2.7160,   4.2705],
        [ 12.3357,   3.6001,   0.7903],
        [  6.3587,   1.1596,  -3.6629],
        [  4.9452, -10.3187,  -7.8621],
        [  0.1206,  -4.7363,   8.7103],
        [  7.0171,  11.6439,  -5.0178],
        [ 11.0864,   7.0387,  -0.1238],
        [  5.0190,   4.9705,  -7.9752],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3549,   0.7096,   6.9198],
        [ -7.4602,   4.7718,  -4.6277],
        [ 12.6035,   0.7494,   7.1536],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.5832,   1.7543,  -7.6134],
        [ -5.0132,  -9.2234,  -8.3531],
        [  5.7963,  -1.6018,   1.8856],
        [  1.0000,   7.6591,  -4.5011],
        [ 10.4758,  -5.9539,   1.3667],
        [  1.5653,  13.9377,   4.1773],
        [ -7.6650,   4.5092,   0.2520],
        [  0.9445,   7.6736,   0.5095],
        [  6.8093,  -9.4178,   8.4244],
        [  2.1504,  -4.6726,   8.7908],
        [  5.0233,   4.9647,  -3.1788],
        [  1.1350,  14.0104,   0.6321],
        [  4.7813,  -3.1139,   8.2691],
        [  1.7228,  -4.7575,   8.3396],
        [ -6.7992,   5.4991,   9.9186],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -5.0550,   6.8231,  -9.2502],
        [  0.1206,  -4.7363,   8.7103],
        [ -4.7389,   6.9953,   3.7638],
        [  4.4049,   5.6892,  -4.4394],
        [ 12.4837,   2.7510,   0.4449],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.5267,  -0.4125,  -8.7331],
        [  2.3889,  -4.6082,  -8.7291],
        [  8.7131,  10.2216,   7.7915],
        [  6.3413,   0.4883,  -6.8416],
        [  6.3583,   1.1729,   8.7835],
        [ -9.3331,  -2.6465,  -8.2028],
        [  3.4140,  -4.1803,  -8.8224],
        [ 11.5568,   6.0602,   9.3244],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.3119,   1.7797,   9.9186],
        [  6.3551,   0.7149,   1.3344],
        [  1.0391,  -4.8141,  -0.0768],
        [  0.5600, -11.0321,  -0.8660],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -2.5056,   7.7720,   5.5878],
        [  5.1196,  -2.7160,   4.2705],
        [  6.1741,  -0.5329,   3.1913],
        [  5.6258,  -1.9404,  -6.0715],
        [ -9.4366,  -0.1189,  -2.6736],
        [  6.2804,   2.0019,   0.8607],
        [  5.0190,   4.9705,  -7.9752]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.2234e-02, 3.1081e-02,  ..., 8.1975e-01, 1.7921e-01,
         3.4645e-01],
        [2.2234e-02, 0.0000e+00, 2.0374e-02,  ..., 8.4116e-01, 1.6807e-01,
         3.4223e-01],
        [3.1081e-02, 2.0374e-02, 0.0000e+00,  ..., 8.4802e-01, 1.4957e-01,
         3.6213e-01],
        ...,
        [8.1975e-01, 8.4116e-01, 8.4802e-01,  ..., 0.0000e+00, 9.6234e-01,
         8.7249e-01],
        [1.7921e-01, 1.6807e-01, 1.4957e-01,  ..., 9.6234e-01, 1.0727e-04,
         4.8832e-01],
        [3.4645e-01, 3.4223e-01, 3.6213e-01,  ..., 8.7249e-01, 4.8832e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.01114719,  5.21280156,  5.24137909,  5.30158497,  5.26342203,
        5.53255882,  8.98765644,  7.10378919,  9.34061938,  9.60359158,
        5.62005629, 10.46238054,  5.1224079 , 11.46176259,  5.64865173,
        8.99276564,  6.83083863, 10.08001894,  7.92607342,  5.13125513,
        9.7712321 ,  6.46356806,  8.8737556 ,  7.50035355, 13.60275211,
       13.15525803,  8.89458273,  7.32233733,  5.1224079 ,  5.26067019,
       10.53368644, 13.95472331,  6.29034958,  9.76896904,  6.7166297 ,
       13.36800073, 13.1471313 ,  6.27715105, 12.1494023 ,  6.80370254,
        7.89745107, 13.05148826, 11.81423161,  5.94535529,  6.27905292,
       11.018084  , 13.32613445,  8.14433542,  8.1137452 ,  8.41148722,
        5.79543038, 12.85032836,  6.46356806, 11.4424816 ,  4.73784222,
       13.59480906, 13.13205357,  7.06372172, 12.87752052,  6.39438044,
        8.85572292, 12.62576026,  5.18047638, 12.70489563, 10.4977132 ,
        6.01356004,  7.72415048, 12.0495521 , 14.02532758,  8.89301516,
        7.73151309, 11.62158554,  5.14371005,  7.06272256, 14.05633441,
        5.70591986,  5.05981914,  8.74470218, 12.05395056,  8.49161154,
        4.73784222,  8.44940661,  7.19513806, 12.78327037, 12.30858677,
       12.53345445,  5.19065439, 13.43127131,  6.36004393,  6.46560181,
        9.70108409,  5.39723241, 13.0493368 , 12.27991991,  6.55800267,
        6.39520397,  4.92497514, 11.04629471, 12.27835996,  8.16584537,
        5.79543038,  6.19707944,  5.9510489 ,  9.43737523,  6.59176018,
        7.06372172])}
{'P': tensor([[3.1571e-02, 1.9473e-01, 1.8163e-01,  ..., 1.0695e-04, 1.1722e-04,
         2.2265e-04],
        [3.4257e-03, 3.3175e-02, 2.0589e-01,  ..., 1.2077e-04, 1.3525e-04,
         1.6520e-04],
        [3.9753e-03, 5.6064e-03, 4.5923e-02,  ..., 1.5924e-04, 2.0325e-04,
         1.9198e-04],
        ...,
        [1.6967e-04, 1.9498e-04, 1.9763e-04,  ..., 1.8338e-01, 4.4366e-05,
         6.1967e-06],
        [9.4712e-04, 8.0191e-04, 7.4727e-04,  ..., 1.6115e-05, 3.1149e-01,
         3.9464e-04],
        [2.2930e-02, 2.3018e-02, 2.3711e-02,  ..., 5.6047e-05, 2.4934e-05,
         1.4052e-01]]), 'X': tensor([[  1.5246,  -4.7838,  -5.2095],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.3373,  -4.6232,  -5.2246],
        [  2.4356,  -4.5942,  -5.0808],
        [  3.7237,  -3.9966,  -5.0376],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.4117,   0.1160,   8.2574],
        [ 12.4837,   2.7510,   0.4449],
        [  0.6933,   7.7336,   2.1179],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.9531,   5.0576,   5.6794],
        [ -9.1231,  -3.5739,   8.4708],
        [  4.9957,  -2.8705,   6.5978],
        [ -5.6169,  -8.7844,   8.9057],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.1625,  -4.6697,   7.0658],
        [ -8.4887,  -5.2448,  -0.6585],
        [  9.2811,  -7.4336,   5.1771],
        [ -7.3446,   4.9113,  -5.9356],
        [  4.1348,   5.9514,  -0.8856],
        [  6.2790,  -0.0259,  -5.4116],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -0.7599, -10.8995,  -6.0366],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.5966,   5.9666,   6.0083],
        [  4.4049,   5.6892,  -4.4394],
        [  4.7813,  -3.1139,   8.2691],
        [ -7.5720,   4.6310,   2.1717],
        [  6.2247,  -0.3150,  -9.3657],
        [ -5.9940,   6.1985,   5.9566],
        [  4.2275,  -3.6320,  -0.2714],
        [ 12.1588,   4.3364,  -8.0867],
        [  7.7750,  11.0665,   2.8364],
        [  0.6142,  14.0802,   0.5407],
        [ -6.0175,  -8.4541,  -9.8408],
        [  1.3705,  13.9723,   7.9266],
        [ -8.7409,  -4.6837,  -6.5840],
        [  6.2036,   2.4030,  -1.2500],
        [ 11.7418,   5.6070,  -1.4192],
        [ 12.4691,   2.8525,  -5.5110],
        [  4.0587, -10.6112,   6.2273],
        [ -7.9298,   4.1367,  -3.9773],
        [  4.7482, -10.3907,   5.9149],
        [ -9.1795,  -3.3623,   8.5259],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.2041,  -3.6509,  -4.9488],
        [  3.9085,   6.1512,  -2.2560],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.0042,  -2.8350,   2.6864],
        [  6.3119,   1.7797,   9.9186],
        [  0.2480, -11.0153,  -3.0183],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.9003,  12.3550,   2.1917],
        [ -4.6393,   7.0461,   1.1624],
        [  2.5318,  13.7231,   6.3692],
        [  5.6093,  -1.9705,   6.4807],
        [  5.2199,   4.6861,  -1.9304],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -6.3489,   5.9117,   6.5161],
        [ -8.4573,   3.2448,   5.4370],
        [ -0.2586,   7.8825,  -9.9840],
        [ -7.8849,  -6.3115,  -4.7479],
        [  3.3943,   6.5483,   1.0790],
        [  9.3753,   9.5171,   3.9131],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.8359,   4.2735,   7.2943],
        [ -8.0829,  -5.9940,   5.0564],
        [  3.4140,  -4.1803,  -8.8224],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -6.6998,   5.5954,   0.0823],
        [ -1.7290,   7.8825,  -3.3831],
        [  6.2555,   2.1474,   4.1685],
        [  4.4290,  -3.4596,   1.4181],
        [ -9.3331,  -2.6465,  -8.2028],
        [  0.6933,   7.7336,   2.1179],
        [ 10.7566,   7.6212,  -2.3279],
        [  2.3261,  13.7749,   5.9121],
        [  5.3602,  -2.3808,   7.2098],
        [  9.6587,   9.1817,  -0.8212],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.3062,  -5.6008,   0.0287],
        [  0.6470, -11.0352,  -1.1931],
        [  3.7992,  -3.9474,  -8.9019],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4671,  -0.5553,   0.1117],
        [ -7.4733,   4.7555,   6.2490],
        [ 11.7795,  -3.4697,   3.5624],
        [ -9.4773,  -1.0024,   5.2599],
        [  1.1006,   7.6318,  -2.4877],
        [  9.4554,   9.4245,   9.8800],
        [ -4.3963,   7.1629,   4.5263],
        [ -7.9667,   4.0814,  -0.6397],
        [  6.3124,   0.2057,  -0.4593],
        [  9.9756,  -6.6330,  -3.3429],
        [  2.5813,  -4.5471,   5.3597],
        [  9.9750,  -6.6338,  -1.2440],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.8835,  -4.3178,   9.3080],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.5966,   5.9666,   6.0083],
        [  5.5294,  -2.1106,   9.0626],
        [ -1.8367,   7.8718,  -8.6523],
        [ -3.0008, -10.2827,  -8.9480]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0121, 0.0210,  ..., 0.8383, 0.3044, 0.7411],
        [0.0121, 0.0000, 0.0101,  ..., 0.8338, 0.3158, 0.7531],
        [0.0210, 0.0101, 0.0000,  ..., 0.8366, 0.3214, 0.7605],
        ...,
        [0.8383, 0.8338, 0.8366,  ..., 0.0000, 1.0680, 1.3558],
        [0.3044, 0.3158, 0.3214,  ..., 1.0680, 0.0000, 0.4526],
        [0.7411, 0.7531, 0.7605,  ..., 1.3558, 0.4526, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.0209005 ,  5.1224079 ,  5.18047638,  5.19986797,  5.46244988,
        5.39450305,  9.41244865, 12.78327037,  7.76457095,  5.48276038,
        7.07902036,  9.79814997,  5.76165934, 10.42667093, 12.46513646,
        9.53015376,  5.14607014,  9.97822473, 11.89105259,  8.8353674 ,
        7.24677664,  6.27905292, 10.54658597, 10.92597196, 10.61472636,
       12.26997165, 13.04156075,  7.19513806,  5.70591986,  8.87587457,
        6.23262289,  8.62260327,  5.57341623, 12.9089559 , 13.5247215 ,
       14.09357547, 10.37698435, 14.0393949 ,  9.91668182,  6.652742  ,
       13.01187484, 12.79126785, 11.36089355,  8.94395628, 11.42420508,
        9.77588801, 10.31584102,  5.56809335,  7.28793663, 11.62770332,
       12.3344536 ,  6.55800267, 11.018084  ,  5.79391825, 13.69162795,
        8.43626114, 13.95472331,  5.94535529,  7.01475376, 10.23946247,
        8.67502348,  9.05843201,  7.88677538, 10.09979747,  7.37576154,
       13.35927527, 11.9182956 ,  8.92549752, 10.06286074,  5.39723241,
        9.69792513,  8.72896381,  8.06990483,  6.61386436,  5.62005629,
        9.70108409,  7.76457095, 13.18278606, 13.96987806,  5.86519525,
       13.32645552,  9.97822473, 10.95593729, 10.01804223, 11.05413545,
        5.47866221, 13.25336001,  9.48337063,  8.8580671 , 12.27991991,
        9.53015376,  7.71075226, 13.35013374,  8.40445872,  8.95133961,
        6.31576264, 11.97957425,  5.2287058 , 11.97949273, 10.33958639,
        9.87720102,  6.01356004, 13.04156075,  5.91854397,  8.08320235,
       10.71162681])}
{'P': tensor([[3.9254e-02, 2.0894e-01, 1.0394e-01,  ..., 6.2576e-03, 3.4731e-03,
         5.4324e-03],
        [4.3049e-03, 5.4906e-02, 2.0116e-01,  ..., 1.4397e-02, 7.7768e-03,
         1.0743e-02],
        [5.5343e-03, 8.9887e-03, 1.1728e-01,  ..., 5.3947e-02, 2.7796e-02,
         4.4314e-02],
        ...,
        [3.5522e-04, 3.9642e-04, 4.8604e-04,  ..., 6.3922e-02, 3.9985e-01,
         1.8881e-03],
        [4.5419e-04, 4.8086e-04, 5.4546e-04,  ..., 4.7177e-03, 1.0567e-01,
         1.3856e-03],
        [2.7651e-04, 3.5178e-04, 5.7142e-04,  ..., 7.8800e-03, 6.5745e-03,
         4.1329e-02]]), 'X': tensor([[  1.6232,  -4.7717,  -2.5655],
        [  2.6321,  -4.5296,  -2.5159],
        [  3.8182,  -3.9347,  -1.7961],
        [  3.5965,  -4.0754,  -3.5035],
        [  4.0642,  -3.7599,  -3.4565],
        [  2.2373,  -4.6506,  -1.7334],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.2373,  -4.6506,  -1.7334],
        [ -4.6393,   7.0461,   1.1624],
        [ -6.8618,  -7.6322,   3.7027],
        [  2.8474,   6.8979,   6.6457],
        [  6.3305,   1.6119,   1.1538],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.8400,   2.3991,   1.3457],
        [  8.2422,  -8.4041,  -1.4962],
        [ -4.6942,  -9.4299,  -9.1846],
        [  9.1620,  -7.5573,  -9.9882],
        [ 11.6634,   5.8050,   9.1590],
        [  8.9284,  10.0036,  -6.9210],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.2644,  -4.6434,   9.7508],
        [  4.5340,  -3.3629,   8.9652],
        [  1.7903,  -4.7467,   0.1168],
        [  4.0642,  -3.7599,  -3.4565],
        [  4.4915,   5.5990,   7.0405],
        [ 12.5267,  -0.4125,  -8.7331],
        [  4.6231,   5.4556,   8.4973],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 12.5154,   2.5072,  -9.3274],
        [ -1.8367,   7.8718,  -8.6523],
        [  3.1480,  -4.3165,  -7.8002],
        [ -5.1974,   6.7396,   3.3484],
        [ -1.5408,   7.8978,   5.3470],
        [  4.6992,   5.3690,   3.0935],
        [ -1.7290,   7.8825,  -3.3831],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.9026,   3.3837,  -0.6573],
        [  2.0039,   7.3173,  -1.5885],
        [  4.9531,   5.0576,   5.6794],
        [ 11.0438,   7.1180,  -5.6991],
        [ -1.5261,   7.8988,   9.3768],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.3118,   4.5452,   4.2114],
        [  4.8432,   5.1968,   1.5006],
        [  2.0729,   7.2878,   7.0880],
        [  5.0348,  -2.8229,   3.5436],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -1.0012, -10.8576,   2.2867],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.2905,   1.9366,   2.1169],
        [ -9.4501,  -1.7133,   2.2980],
        [  3.4604,  -4.1545,   1.8091],
        [  7.7794,  11.0629,   8.9069],
        [ -1.6275, -10.7221,   3.2376],
        [ -8.1426,   3.8048,   9.3542],
        [ -7.5603,   4.6459,   8.2002],
        [  6.8407,  11.7666,   3.0045],
        [ -8.9282,   2.1648,   8.4894],
        [ 11.5456,   6.0861,   8.1101],
        [  8.7131,  10.2216,   7.7915],
        [  3.3744,   6.5623,  -6.3636],
        [ -8.6112,   2.9319,  -7.9943],
        [ 12.5770,   1.8512,  -0.6578],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.1579,  -0.5961,  -6.5595],
        [  8.5716,  10.3596,   6.8818],
        [ -0.3432,   7.8900,   7.1805],
        [  6.3360,   0.4258,  -9.2509],
        [  6.0047,   3.1062,  -6.7914],
        [  6.2022,   2.4092,   3.0349],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.7735,   2.5640,  -6.3313],
        [  4.0591,   6.0202,  -1.6339],
        [  6.2026,  12.1774,  -1.5887],
        [  7.0642,  11.6104,  -9.8316],
        [  3.3069,   6.6089,   8.7377],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.6357,   5.4415,   7.9067],
        [ -2.6485, -10.4151,   0.8766],
        [ -3.4986,   7.5152,   9.0712],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.2592,   3.9425,   2.0383],
        [ -9.0961,   1.6529,   3.1553],
        [  8.7715,  -7.9390,   5.2900],
        [  1.8721,  -4.7322,   0.5524],
        [  4.2339,  -3.6268,   6.3125],
        [  6.3266,   0.3292,   9.3502],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.9470,  13.6085,   2.9809],
        [ -3.4986,   7.5152,   9.0712],
        [  6.2247,  -0.3150,  -9.3657],
        [  1.6232,  -4.7717,  -2.5655],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.1706,   1.3863,   8.4032],
        [  1.3339,   7.5627,  -0.9034],
        [  5.6672,  -1.8631,   8.1020],
        [  6.2576,   2.1359,  -3.3398],
        [  5.8836,   3.4315,  -4.0327],
        [  6.2770,  -0.0379,   2.4195]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0232, 0.0671,  ..., 0.1117, 0.1213, 0.3665],
        [0.0232, 0.0000, 0.0517,  ..., 0.0935, 0.1082, 0.3578],
        [0.0671, 0.0517, 0.0000,  ..., 0.1145, 0.1402, 0.3074],
        ...,
        [0.1117, 0.0935, 0.1145,  ..., 0.0000, 0.0348, 0.4026],
        [0.1213, 0.1082, 0.1402,  ..., 0.0348, 0.0000, 0.4356],
        [0.3665, 0.3578, 0.3074,  ..., 0.4026, 0.4356, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.04028538,  5.23878371,  5.48276038,  5.43544626,  5.5366698 ,
        5.16077764, 11.75807252,  5.16077764,  8.43626114, 10.26327895,
        7.46248947,  6.53250937,  6.49538841,  9.15976746, 11.77124373,
       10.53368644, 11.87665501, 13.02818702, 13.40850328, 10.31886066,
        5.16611782,  5.64500512,  5.07306274,  5.5366698 ,  7.17789666,
       12.53345445,  7.15100303,  9.34127013,  9.75401449, 12.76408562,
        8.08320235,  5.34250225,  8.51088913,  8.04664886,  7.13501264,
        8.06990483,  9.64157461,  6.80370254,  7.58668144,  7.07902036,
       13.13888841,  8.04483495, 12.10942412,  9.62958973,  6.99097013,
        7.10378919,  7.57685908,  5.77219077, 10.52587243,  9.75075273,
        9.70108409, 10.9036226 , 13.13888841,  6.58184474,  9.60412735,
        5.40689332, 13.52430176, 10.84493732,  8.98765644,  8.8737556 ,
       13.6105595 ,  9.18690018, 13.05148826, 13.43127131,  7.37902766,
        9.09661212, 12.7125079 , 10.67322527, 10.06703358,  6.18668463,
       13.44593519,  7.89745107,  6.35029274,  6.76056625,  6.65368551,
        9.14973319,  9.14044984,  7.26074925, 13.6660671 , 13.59056613,
        7.39004226, 13.25017211,  7.14837514, 10.74655346,  8.28968067,
        7.822854  ,  9.34127013, 12.87752052,  9.24501829, 11.83075005,
        5.08910491,  5.5748624 ,  6.33517349, 12.95702647, 13.92390689,
        8.28968067,  6.23262289,  5.04028538, 10.52587243,  9.09661212,
        9.27474386,  7.67939546,  5.96556248,  6.61211224,  6.81117802,
        6.27715105])}
{'P': tensor([[3.0457e-02, 1.9282e-01, 1.8996e-01,  ..., 7.8688e-05, 1.6774e-03,
         1.0839e-05],
        [7.0016e-03, 7.2532e-02, 2.6685e-01,  ..., 1.9792e-04, 4.7152e-03,
         2.4996e-05],
        [9.2908e-03, 2.4757e-02, 1.0075e-01,  ..., 2.6495e-04, 6.3999e-03,
         3.3813e-05],
        ...,
        [3.1281e-05, 3.9818e-05, 3.9841e-05,  ..., 9.7179e-02, 7.4984e-03,
         8.2504e-06],
        [2.5824e-04, 2.9629e-04, 2.9531e-04,  ..., 2.1679e-04, 5.9094e-02,
         1.3445e-04],
        [7.2636e-05, 6.0038e-05, 5.6837e-05,  ..., 5.8916e-06, 2.9024e-05,
         3.2194e-02]]), 'X': tensor([[ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0154, 0.0180,  ..., 0.9148, 0.3744, 0.9735],
        [0.0154, 0.0000, 0.0026,  ..., 0.9108, 0.3678, 0.9815],
        [0.0180, 0.0026, 0.0000,  ..., 0.9109, 0.3675, 0.9835],
        ...,
        [0.9148, 0.9108, 0.9109,  ..., 0.0000, 0.5511, 0.7499],
        [0.3744, 0.3678, 0.3675,  ..., 0.5511, 0.0000, 0.8327],
        [0.9735, 0.9815, 0.9835,  ..., 0.7499, 0.8327, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.05444288,  5.16611782,  5.1805345 ,  5.13125513,  5.10446653,
        5.39098669, 10.99326577, 13.00380624,  9.25336371,  8.83341608,
        6.19707944,  7.03733775,  6.383897  , 12.26997165,  8.05242591,
        9.34061938,  8.50268322, 11.63724792, 11.37939695,  9.36093562,
       10.8207297 , 13.06498554, 11.27656488, 11.2845608 ,  5.94535529,
       11.77124373, 13.06498554,  4.75303127, 13.31764181, 14.10045797,
        8.50278896, 11.77124373, 11.89105259,  4.91652556, 11.64746882,
        4.92497514,  8.05242591,  7.20826219, 10.26327895, 13.09737053,
       13.43127131,  6.56833864,  8.92549752, 10.8207297 ,  9.62238828,
       13.36800073,  7.73692436,  8.86082124,  6.00663314, 12.11735334,
        6.28835101,  4.978704  , 11.80092578, 12.52875793,  9.25336371,
        7.6374603 ,  6.53250937, 10.88618114, 13.02818702, 11.9182956 ,
       12.22503754, 13.96905756,  5.39098669, 11.785566  , 10.31584102,
        6.58184474,  5.92952538, 11.95743558,  9.88310614, 10.19036081,
       12.13568598,  6.99097013,  7.37576154,  5.29009836, 12.79126785,
       12.27993098, 13.72594022, 10.06407839,  5.24461068, 10.39942497,
       11.85743425,  5.56809335,  8.27853076,  9.12220271,  7.16598273,
       12.07225469,  7.72291946,  5.47702766, 11.05413545,  9.64157461,
        6.81084724, 13.39372426,  9.54154096,  7.50564223, 11.38774149,
        7.54620849, 12.9089559 , 11.48734978,  8.36995953,  5.46244988,
        8.94395628, 11.63765273,  5.56809335, 12.07818871,  6.27715105,
        9.11473391])}
{'P': tensor([[5.9651e-02, 1.6118e-01, 1.0118e-01,  ..., 1.4343e-03, 2.4735e-04,
         3.7091e-03],
        [3.3886e-03, 5.3865e-02, 2.7433e-01,  ..., 2.1164e-03, 1.0507e-04,
         2.6219e-03],
        [2.9236e-03, 5.5428e-03, 8.1117e-02,  ..., 4.4917e-03, 9.3745e-05,
         2.6268e-03],
        ...,
        [3.1603e-04, 2.7621e-04, 2.8716e-04,  ..., 1.2510e-01, 5.7106e-06,
         7.6734e-05],
        [9.1088e-04, 1.7179e-03, 1.8000e-03,  ..., 1.0482e-05, 4.0572e-02,
         1.0533e-02],
        [6.3668e-03, 1.7299e-02, 2.6656e-02,  ..., 5.9650e-04, 1.3352e-03,
         1.5215e-01]]), 'X': tensor([[  1.7000,  -4.7609,   2.7935],
        [  3.4604,  -4.1545,   1.8091],
        [  4.4290,  -3.4596,   1.4181],
        [  2.4697,  -4.5836,   1.4593],
        [  2.8890,  -4.4316,   1.1600],
        [  5.0348,  -2.8229,   3.5436],
        [  5.2911,  -2.4824,  -3.6877],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.7162,   5.6728,  -8.9372],
        [ 11.2571,   6.7072,   8.3693],
        [ 10.1503,   8.5414,  -7.2681],
        [  9.0051,   9.9234,   5.4411],
        [  6.3497,   1.3680,  -1.9076],
        [ -7.0120,   5.2825,   7.7762],
        [ -1.8367,   7.8718,  -8.6523],
        [  0.9074,  -4.8139,   8.1674],
        [  5.3118,   4.5452,   4.2114],
        [ -2.4956, -10.4682,   3.5632],
        [  2.9410,   6.8428,   5.4458],
        [ -5.0550,   6.8231,  -9.2502],
        [ 11.7796,  -3.4696,   9.0088],
        [  9.5203,   9.3483,   5.8781],
        [ -3.9858,  -9.8334,  -3.7323],
        [  4.3523, -10.5232,   8.2138],
        [  2.8994,   6.8675,  -9.5987],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.4235,   6.3594,   5.1663],
        [  0.6470, -11.0352,  -1.1931],
        [  1.3670,  -4.7990,   3.2603],
        [  6.3516,   0.6425,   0.3425],
        [ 12.5224,  -0.4497,   4.8448],
        [  5.9843,  -9.8679,  -6.1449],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.7792,  11.8084,   1.6251],
        [  7.9141,  -8.6656,   9.6948],
        [ -9.0604,   1.7707,   1.1066],
        [  1.0000,   7.6591,  -4.5011],
        [  5.1861,   4.7361,  -3.7312],
        [  6.3516,   0.6425,   0.3425],
        [  5.0348,  -2.8229,   3.5436],
        [  0.5250, -11.0306,   5.5358],
        [  9.0051,   9.9234,   5.4411],
        [ -9.4378,  -1.8606,  -2.5668],
        [  0.1532,  -4.7422,  -7.0145],
        [ -9.4203,  -2.0340,   2.5856],
        [  0.1041, -11.0045,   2.4311],
        [  6.2883,   0.0325,   5.2692],
        [ 10.9717,  -5.1646,  -6.9039],
        [  2.6613,   7.0021,   4.5783],
        [  1.7228,  -4.7575,   8.3396],
        [  6.1973,  -0.4372,   6.5720],
        [  1.0000,   7.6591,  -4.5011],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.7845,   3.6652,   0.9175],
        [  4.8653,   5.1694,  -4.0823],
        [ -4.6393,   7.0461,   1.1624],
        [  4.3554,   5.7393,  -9.0766],
        [ 12.0582,  -2.6640,   0.7132],
        [ -0.8675, -10.8815,   6.8561],
        [  2.2076,  13.8032,  -1.5099],
        [  4.3459,  -3.5327,   0.3667],
        [ -5.9938,  -8.4745,   3.9735],
        [  2.4639, -10.9431,  -7.3718],
        [ -2.5161,   7.7699,  -9.3531],
        [ 12.3609,   3.4769,  -8.5762],
        [  4.5057,  -3.3894,  -0.8160],
        [ -3.3812, -10.1235,   7.4384],
        [  7.1697,  11.5343,  -0.7276],
        [  4.6063,  13.0051,  -3.1456],
        [  3.2692,  13.5098,  -4.9032],
        [  3.9895,   6.0816,  -6.5235],
        [ -8.3062,  -5.6008,   0.0287],
        [  3.4401,  13.4540,   4.6576],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.3767, -11.0341,   9.9415],
        [  0.9659,  -4.8144,  -0.9594],
        [ -1.0012, -10.8576,   2.2867],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.6396,  14.0772,  -6.3786],
        [ -5.3382,   6.6533,   7.8054],
        [ -7.9298,   4.1367,  -3.9773],
        [ -6.4368,   5.8356,   3.3014],
        [ -0.0282,   7.8575,  -1.1746],
        [ -6.8618,  -7.6322,   3.7027],
        [  0.2435,   7.8191,  -6.4709],
        [  6.3211,   1.7017,  -0.2164],
        [ -8.0330,  -6.0765,   6.7945],
        [  7.9141,  -8.6656,   9.6948],
        [ -5.1974,   6.7396,   3.3484],
        [ 11.1085,  -4.9204,   0.0736],
        [ -9.0961,   1.6529,   3.1553],
        [  6.2902,   0.0451,   3.7116],
        [  9.2811,  -7.4336,   5.1771],
        [ -2.6069, -10.4298,  -6.1995],
        [  8.0877,  -8.5297,  -1.1050],
        [ 10.1503,   8.5414,  -7.2681],
        [ -9.4157,   0.0821,  -5.0025],
        [ 12.2435,   4.0076,   8.1257],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.1440, -10.9811,   8.5281],
        [  4.4915,   5.5990,   7.0405],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.3490,  -4.6199,  -4.2489]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.5146e-02, 1.0338e-01,  ..., 3.0921e-01, 8.9024e-01,
         4.4038e-01],
        [7.5146e-02, 1.0727e-04, 2.8488e-02,  ..., 3.7567e-01, 8.6948e-01,
         3.7493e-01],
        [1.0338e-01, 2.8488e-02, 0.0000e+00,  ..., 3.9960e-01, 8.6763e-01,
         3.5396e-01],
        ...,
        [3.0921e-01, 3.7567e-01, 3.9960e-01,  ..., 0.0000e+00, 1.1426e+00,
         7.4915e-01],
        [8.9024e-01, 8.6948e-01, 8.6763e-01,  ..., 1.1426e+00, 0.0000e+00,
         6.5513e-01],
        [4.4038e-01, 3.7493e-01, 3.5396e-01,  ..., 7.4915e-01, 6.5513e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.05534609,  5.40689332,  5.62005629,  5.20659892,  5.29009836,
        5.77219077,  5.84450127,  9.69629   ,  8.85572292, 13.01728313,
       13.103709  , 13.26590071, 13.40024224,  6.49538841,  8.77913428,
        8.08320235,  4.89869205,  6.99097013, 10.76154256,  7.44805901,
        8.49161154, 12.27993098, 13.3426541 , 10.61047596, 11.38774149,
        7.4544866 ,  9.60080639,  8.08320235, 13.0743399 , 11.05413545,
        4.98989906,  6.38399789, 12.53047489, 11.54071859, 10.05558893,
        9.94285511, 13.61600536, 11.73567759,  9.23178204,  7.72415048,
        7.02326515,  6.38399789,  5.77219077, 11.04313641, 13.40024224,
        9.61942445,  4.74468406,  9.63742669, 11.0050379 ,  6.28835101,
       12.12641969,  7.49076899,  5.05981914,  6.21275016,  7.72415048,
        9.37989631,  6.84796202,  7.09887557,  8.43626114,  7.20482224,
       12.3489324 , 10.91601655, 13.97857591,  5.60064319, 10.37998078,
       11.21703654,  8.16714836, 12.84056604,  5.63824971, 10.67322527,
       13.58101794, 13.79676654, 13.89974693,  7.27340834, 10.01804223,
       13.88683436, 12.17451685, 11.11969815,  4.91038884, 10.9036226 ,
       12.25806559, 14.09176259,  8.53014915,  8.94395628,  8.68832398,
        7.8575648 , 10.26327895,  7.822854  ,  6.54615887, 10.07238463,
       11.73567759,  8.51088913, 12.1494023 ,  9.24501829,  6.29034958,
       11.89105259, 10.7506423 , 11.75438006, 13.26590071,  9.41605684,
       12.88270018,  9.55593374, 11.18839169,  7.17789666, 10.92597196,
        5.18277858])}
{'P': tensor([[2.9991e-02, 1.5924e-01, 1.2115e-01,  ..., 8.8483e-05, 2.0752e-05,
         1.7092e-03],
        [3.4812e-03, 4.5295e-02, 4.7941e-02,  ..., 2.2684e-04, 4.8404e-05,
         3.6233e-03],
        [3.9277e-03, 2.8177e-02, 4.5107e-02,  ..., 2.2099e-04, 5.2610e-05,
         3.2207e-03],
        ...,
        [2.9520e-06, 2.8622e-06, 2.7739e-06,  ..., 1.6104e-02, 5.4707e-03,
         4.4499e-07],
        [2.2883e-05, 1.9027e-05, 1.9769e-05,  ..., 8.3816e-05, 1.6308e-01,
         7.2900e-07],
        [1.4078e-05, 2.2424e-05, 2.1897e-05,  ..., 2.7964e-03, 5.0886e-05,
         1.2059e-02]]), 'X': tensor([[  1.7122,  -4.7591,  -3.6734],
        [  2.7420,  -4.4896,  -3.7139],
        [  2.7976,  -4.4682,  -4.2020],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.5965,  -4.0754,  -3.5035],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.2199,   4.6861,  -1.9304],
        [  2.2076,  13.8032,  -1.5099],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.7558,  -4.4844,   0.0636],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.5789,   0.1722,   1.6470],
        [  4.5954,  13.0099,  -4.4439],
        [  4.4049,   5.6892,  -4.4394],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.4242, -10.1261,   3.9595],
        [  5.0617,  12.7957,   8.3544],
        [  2.3373,  -4.6232,  -5.2246],
        [  7.1441,  11.5529,   2.3232],
        [  6.0345,   3.0175,  -1.4424],
        [  2.4697,  -4.5836,   1.4593],
        [  2.5829,  -4.5466,  -8.4015],
        [  0.3354,  -4.7708,  -4.9125],
        [ -4.6942,  -9.4299,  -9.1846],
        [  0.1252, -11.0062,   7.2564],
        [ -6.4002,  -8.1048,  -0.9108],
        [  4.3377,   5.7570,   1.3712],
        [  4.6063,  13.0051,  -3.1456],
        [  5.1203,  -2.7151,   6.2110],
        [  3.2809,  13.5061,  -4.0199],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.0611,  -6.0302,   4.6962],
        [ -5.6423,   6.4537,  -1.2091],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.3413,   0.4883,  -6.8416],
        [ 12.5650,   2.0132,   4.6272],
        [ 11.6801,  -3.7196,  -0.9181],
        [  3.7237,  -3.9966,  -5.0376],
        [ -6.4464,   5.8272,   7.5038],
        [  0.9967,  -4.8144,  -9.7370],
        [ 11.2847,  -4.5848,  -4.7503],
        [  1.6308,  13.9254,  -2.0984],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.7976,  -4.4682,  -4.2020],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.2707,   0.9661,   4.4863],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.9624,  -4.0944,  -4.9622],
        [  5.8275,  -1.5334,   2.0124],
        [  0.6757, -11.0361,  -1.3047],
        [  3.4604,  -4.1545,   1.8091],
        [ -2.0840,   7.8415,  -7.8657],
        [  6.2242,  -0.3171,   7.4978],
        [ -9.1459,   1.4780,  -4.2403],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.3569,   0.7591,   6.5011],
        [  3.8111, -10.6787,  -4.6539],
        [ -2.5683, -10.4433,  -2.9182],
        [  7.7794,  11.0629,   8.9069],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.4117,   0.1160,   8.2574],
        [  6.2905,   1.9366,   2.1169],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -1.2454,   7.9127,  -0.5715],
        [ 11.5456,   6.0861,   8.1101],
        [ -7.9925,  -6.1423,   2.8990],
        [  6.2975,   0.0947,   5.1791],
        [ 12.5199,  -0.4711,   3.0436],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.8323,  -1.5228,   5.1132],
        [ -9.2687,  -2.9794,   0.6436],
        [  1.6408,   7.4591,  -4.2294],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.2883,   0.0325,   5.2692],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.0310,  12.2795,  -6.4528],
        [  3.3943,   6.5483,   1.0790],
        [  5.4426, -10.1181,  -5.1268],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.3357,   3.6001,   0.7903],
        [ 11.8951,  -3.1570,   7.0742],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.2975,   0.0947,   5.1791],
        [  3.7298,  13.3537,  -9.4653],
        [ -7.6650,   4.5092,   0.2520],
        [ -9.3384,  -2.6161,  -4.7630],
        [  6.3587,   1.1596,  -3.6629],
        [ 11.4743,   6.2478,  -2.1100],
        [ 12.2592,   3.9425,   2.0383],
        [  2.1824,  -4.6647,   1.1985],
        [  8.4691,  10.4571,  -5.0650],
        [ 11.9936,  -2.8677,   6.0614],
        [ -9.1623,  -3.4288,  -4.9961],
        [  0.6952,  -4.8060,   8.8976],
        [  8.7715,  -7.9390,   5.2900],
        [  2.5469,  13.7192,  -9.6935],
        [  5.9843,  -9.8679,  -6.1449],
        [ 11.9641,  -2.9568,   7.7211],
        [ -9.3604,   0.4855,  -3.0158],
        [  9.8082,  -6.8396,   4.9615],
        [  6.2419,  -0.2313,   4.2440],
        [  7.9816,  10.8938,  -0.5378],
        [  2.7295,  13.6703,  -8.9328],
        [ 12.0582,  -2.6640,   0.7132]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0236, 0.0372,  ..., 0.4013, 0.4455, 0.2879],
        [0.0236, 0.0000, 0.0226,  ..., 0.3815, 0.4298, 0.2848],
        [0.0372, 0.0226, 0.0000,  ..., 0.3884, 0.4092, 0.3058],
        ...,
        [0.4013, 0.3815, 0.3884,  ..., 0.0000, 0.5387, 0.2999],
        [0.4455, 0.4298, 0.4092,  ..., 0.5387, 0.0000, 0.6602],
        [0.2879, 0.2848, 0.3058,  ..., 0.2999, 0.6602, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.05774295,  5.26067019,  5.27178363,  5.35774573,  5.43544626,
        5.18277858,  7.01475376, 13.97857591, 13.25336001,  5.26342203,
        5.64464807, 12.58006038, 13.79762415,  7.19513806,  6.12886024,
       11.48734978, 13.76047999,  5.18047638, 13.58333857,  6.74686071,
        5.20659892,  5.22903117,  4.78258092, 10.53368644, 11.00694688,
       10.32715715,  7.20826219, 13.79676654,  5.79562691, 13.89886469,
        6.27905292, 10.06703358,  8.5724042 ,  8.05242591,  6.36004393,
       12.72524615, 12.25806559,  5.46244988,  8.68979052,  4.91652556,
       12.18045978, 14.02058621, 11.63502679,  5.27178363, 11.1341798 ,
        9.32094055,  8.49161154,  9.85330646,  6.02588517, 11.05672136,
        5.40689332,  8.1137452 ,  6.2322769 ,  9.26455795, 10.53368644,
        6.4020404 , 11.33837396, 10.75443106, 13.52430176, 13.38091121,
        9.41244865,  6.58184474,  9.58072286,  8.01009083, 13.05148826,
       10.08001894,  6.29821673, 12.52875793, 12.12641969,  6.18668463,
        6.02779201,  9.73579738,  7.6374603 , 10.33712258,  6.28835101,
        9.50654967, 13.68063098,  7.37576154, 11.4890861 , 12.86648771,
       12.85032836, 12.30695098,  8.16584537, 12.18045978,  6.29821673,
       13.86480175,  8.89301516,  9.69792513,  6.46356806, 13.06498554,
       12.87752052,  5.14999799, 13.45642379, 12.33167382,  9.78286733,
        4.85603886, 11.83075005, 13.95361098, 11.54071859, 12.32408516,
        9.37296102, 11.95743558,  6.24614905, 13.50484539, 13.94008948,
       12.3489324 ])}
{'P': tensor([[3.0936e-02, 1.4472e-01, 1.1146e-01,  ..., 3.0862e-03, 1.6998e-03,
         2.3926e-05],
        [2.9482e-03, 3.5153e-02, 2.0055e-01,  ..., 5.8034e-03, 1.1634e-03,
         4.0875e-05],
        [2.4773e-03, 3.7135e-03, 4.8023e-02,  ..., 1.2526e-02, 1.1893e-03,
         8.5873e-05],
        ...,
        [6.3638e-04, 6.8139e-04, 7.5695e-04,  ..., 1.4294e-01, 3.1139e-04,
         1.8674e-02],
        [6.7116e-03, 7.4184e-03, 9.5594e-03,  ..., 2.2433e-03, 9.7715e-02,
         4.8230e-05],
        [7.3245e-06, 5.7980e-06, 5.0902e-06,  ..., 1.8961e-05, 6.0962e-06,
         3.9414e-02]]), 'X': tensor([[ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0305, 0.0427,  ..., 0.0868, 0.4362, 0.2241],
        [0.0305, 0.0000, 0.0179,  ..., 0.0583, 0.4558, 0.2057],
        [0.0427, 0.0179, 0.0000,  ..., 0.0443, 0.4508, 0.1878],
        ...,
        [0.0868, 0.0583, 0.0443,  ..., 0.0000, 0.4746, 0.1581],
        [0.4362, 0.4558, 0.4508,  ..., 0.4746, 0.0000, 0.4497],
        [0.2241, 0.2057, 0.1878,  ..., 0.1581, 0.4497, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.05963217,  5.26067019,  5.43544626,  5.35774573,  5.27178363,
        5.5366698 ,  7.49076899,  9.14973319,  8.1303945 , 13.69162795,
       13.56367857,  4.87083348,  9.37989631, 10.63038565,  6.18668463,
       11.67875855,  7.76457095, 11.72749923,  7.20826219, 11.52733581,
       12.52372629,  7.13501264,  4.98989906,  9.60080639,  9.3109884 ,
        7.53884104, 11.36089355, 13.03002338,  7.27340834,  5.20659892,
        5.63824971,  9.60080639,  9.09661212,  5.68942572,  9.49511524,
        6.81084724, 11.31276109, 11.97949273,  5.60064319, 11.22142195,
       13.46553623,  5.26067019, 11.05317597, 11.97238624, 13.36800073,
       11.42947632, 13.04156075, 11.4088043 , 13.1399181 ,  9.77777346,
        5.95111804,  8.40445872,  7.50564223, 13.18278606,  4.74468406,
       11.59512194,  7.57685908,  8.22728399,  5.24461068,  6.76056625,
       14.02532758, 11.33837396, 10.412299  ,  5.95111804,  6.76791483,
        9.0368458 , 14.08549802, 13.95472331, 11.51811603, 11.23201619,
        9.3109884 , 11.05413545, 13.45642379,  8.7907923 ,  4.78258092,
        5.94909709,  5.70405525,  9.09661212, 13.89974693, 12.19805394,
       10.7506423 , 10.01804223,  8.68979052, 13.03002338,  9.97822473,
       12.59840193, 13.01187484,  5.16077764, 12.4217523 ,  9.15253344,
       10.04916678,  7.60529892,  5.43943791, 11.9702762 ,  8.56889738,
       11.69264243,  5.64464807,  6.02789634,  6.49538841,  8.04483495,
        8.68832398,  6.29821673, 10.1979669 ,  6.80559669,  4.98989906,
       14.02058621])}
{'P': tensor([[2.9520e-02, 1.4557e-01, 1.3571e-01,  ..., 3.9405e-04, 2.7038e-05,
         1.9535e-04],
        [3.3035e-03, 4.6689e-02, 4.0463e-03,  ..., 7.9491e-04, 3.0453e-05,
         2.8986e-04],
        [6.2237e-03, 2.3477e-01, 4.2936e-02,  ..., 5.5925e-04, 3.3597e-05,
         2.5558e-04],
        ...,
        [9.7208e-05, 1.1748e-04, 9.5352e-05,  ..., 6.1892e-02, 4.4690e-05,
         1.0491e-04],
        [6.1846e-04, 3.6446e-04, 4.8979e-04,  ..., 2.8985e-05, 2.5820e-02,
         3.9054e-05],
        [7.3433e-06, 1.2618e-05, 8.0428e-06,  ..., 2.9944e-03, 2.6559e-07,
         9.0749e-03]]), 'X': tensor([[  1.7228,  -4.7575,   8.3396],
        [  2.9260,  -4.4162,   8.3808],
        [  2.1120,  -4.6818,   8.6144],
        [  3.3839,  -4.1966,   8.3574],
        [  2.1504,  -4.6726,   8.7908],
        [  1.8447,  -4.7372,   8.1235],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.9515,  -3.8425,   6.1729],
        [ -9.3419,   0.5972,  -1.2841],
        [ -6.6998,   5.5954,   0.0823],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.4703,  -4.2006,   7.4086],
        [  8.2154,  -8.4262,   8.0171],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.3542,   0.6934,  -3.3840],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.3103,  -4.8032,  -0.9300],
        [  8.7715,  -7.9390,   5.2900],
        [  5.8201,   3.5841,  -8.0609],
        [  5.8979,   3.3958,  -3.8713],
        [ -8.4267,  -5.3697,   2.4172],
        [  7.8982,  -8.6778,   0.0239],
        [  5.6928,  12.4710,   3.6001],
        [  9.3753,   9.5171,   3.9131],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 11.5456,   6.0861,   8.1101],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.0574,   1.7801,   8.8986],
        [  2.2437,  -4.6489,   6.1430],
        [ -1.9600,   7.8577,   6.9121],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3606,   0.9204,   3.2525],
        [  2.0937,  13.8293,  -2.3533],
        [  6.8407,  11.7666,   3.0045],
        [  4.8051, -10.3703,   0.5268],
        [ -9.2301,  -3.1537,  -2.7477],
        [  5.7151,  -1.7700,   3.1542],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.4430,  -0.1897,   4.8383],
        [  5.7218,  -9.9934,   6.5764],
        [  6.0054,  -1.0902,   8.3079],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.3303,   1.6145,  -8.4083],
        [  6.3551,   0.7149,   1.3344],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.4925,   9.3811,   8.9842],
        [  6.3266,   0.3292,   9.3502],
        [ -7.2515,   5.0195,   4.6434],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.6970,   6.9826,   9.7350],
        [  4.0078,  -3.8018,  -6.2168],
        [ -2.0721,   7.8432,   6.6380],
        [ -9.4716,  -0.6673,   7.6068],
        [ -7.5720,   4.6310,   2.1717],
        [  5.3118,   4.5452,   4.2114],
        [ -6.9838,  -7.4965,   0.9796],
        [  6.2883,   0.0325,   5.2692],
        [  4.2096,   5.8815,   1.9230],
        [ -6.4368,   5.8356,   3.3014],
        [ -8.0330,  -6.0765,   6.7945],
        [  5.1292,   4.8183,  -5.7997],
        [  7.8425,  -8.7202,   6.4530],
        [ 11.2571,   6.7072,   8.3693],
        [  1.6232,  -4.7717,  -2.5655],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.9410,   6.8428,   5.4458],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -2.1533, -10.5778,   1.0112],
        [ 11.7726,  -3.4876,  -6.7402],
        [  5.4732,  -2.2041,  -4.3830],
        [  2.0899,  -4.6870,  -7.4260],
        [  6.3301,   0.3638,  -7.5685],
        [ 12.2435,   4.0076,   8.1257],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.3384,  -2.6161,  -4.7630],
        [  5.9548,  -1.2273,  -2.6520],
        [  9.9217,  -6.7006,  -5.6728],
        [ -3.1671,   7.6158,   6.0765],
        [ -1.9600,   7.8577,   6.9121],
        [  1.8721,  -4.7322,   0.5524],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.1931,  -4.7491,   7.4385],
        [  5.0233,   4.9647,  -3.1788],
        [ -9.2974,   0.8371,   2.4852],
        [ 10.6673,  -5.6649,  -4.1646],
        [  1.3339,   7.5627,  -0.9034],
        [  9.6614,   9.1784,  -1.2087],
        [ -3.4986,   7.5152,   9.0712],
        [  5.8201,   3.5841,  -8.0609],
        [  5.0233,   4.9647,  -3.1788],
        [ 12.4901,  -0.7052,   1.7315],
        [  3.3839,  -4.1966,   8.3574],
        [  2.7295,  13.6703,  -8.9328],
        [ -5.3328,  -8.9992,   7.3440],
        [ 12.1940,  -2.1881,   7.1499],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.1813,  -3.3550,   1.8827],
        [  9.9756,  -6.6330,  -3.3429]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0293, 0.0131,  ..., 0.6484, 0.9528, 0.8251],
        [0.0293, 0.0000, 0.0167,  ..., 0.6546, 0.9820, 0.8365],
        [0.0131, 0.0167, 0.0000,  ..., 0.6534, 0.9659, 0.8323],
        ...,
        [0.6484, 0.6546, 0.6534,  ..., 0.0000, 0.9816, 0.2264],
        [0.9528, 0.9820, 0.9659,  ..., 0.9816, 0.0000, 0.9187],
        [0.8251, 0.8365, 0.8323,  ..., 0.2264, 0.9187, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.05981914,  5.29755487,  5.13616224,  5.39098669,  5.14371005,
        5.08372149,  9.36093562,  5.51177501,  9.36093562,  8.72896381,
       11.63502679, 12.1494023 , 12.21532282, 11.76830981, 10.35485331,
        6.39187285, 10.32715715,  4.978704  , 11.83075005,  6.83515201,
        6.80559669,  9.99211663, 11.73397309, 13.70893492, 13.35927527,
        6.80370254,  9.5642993 , 13.05148826,  9.9353921 ,  9.50913849,
        9.23071698,  5.16205087,  8.09842704, 12.10942412,  9.36093562,
        6.30647746,  6.42689705, 13.98691426, 13.6105595 , 11.42947632,
        9.75401449,  5.98285319,  4.87083348, 13.09426142,  9.44486609,
       11.51557166,  6.10360073,  9.69629   ,  6.53290663,  6.39520397,
       10.06286074, 13.34587027,  6.33517349,  8.81930525, 10.35485331,
        9.75075273,  7.48538905,  5.52416368,  8.11227668,  9.49511524,
        8.87587457,  6.99097013, 10.24557047,  6.28835101,  7.23276582,
        8.68832398, 10.07238463,  7.03733775, 11.72802491, 13.103709  ,
        5.04028538,  5.13616224,  9.48337063,  7.44805901,  9.47369628,
       10.794755  , 12.27835996,  5.90035324,  5.13182506,  6.34059021,
       12.88270018,  7.5092834 ,  9.69792513,  6.07992623, 11.97238624,
        8.24808832,  8.09842704,  5.08910491, 10.5575601 ,  4.75303127,
        7.06272256,  9.3349858 , 12.07818871,  7.67939546, 13.32613445,
        8.28968067,  6.83515201,  7.06272256, 12.50996828,  5.39098669,
       13.94008948, 10.4606265 , 12.38881635,  6.49538841,  9.7751189 ,
       11.97957425])}
{'P': tensor([[6.6252e-02, 1.4428e-01, 1.0480e-01,  ..., 1.3491e-03, 2.0910e-04,
         2.9257e-03],
        [3.3406e-03, 4.8810e-02, 8.9574e-02,  ..., 1.4002e-03, 3.9502e-04,
         5.7089e-03],
        [1.3362e-03, 3.0652e-03, 5.4288e-02,  ..., 4.0491e-04, 2.5233e-04,
         3.2441e-03],
        ...,
        [2.6793e-05, 1.0018e-04, 1.7755e-04,  ..., 1.5438e-02, 4.0964e-02,
         6.0659e-03],
        [2.2561e-06, 5.6619e-06, 8.4427e-06,  ..., 5.3015e-06, 1.3532e-02,
         6.0574e-04],
        [3.6904e-04, 7.5103e-04, 9.5489e-04,  ..., 3.2096e-04, 3.2581e-03,
         1.0014e-01]]), 'X': tensor([[  1.7232,  -4.7574,   3.1270],
        [  3.4604,  -4.1545,   1.8091],
        [  5.0348,  -2.8229,   3.5436],
        [  4.4290,  -3.4596,   1.4181],
        [  5.1196,  -2.7160,   4.2705],
        [  5.2188,  -2.5839,   2.1833],
        [ -4.8751,  -9.3148,   4.0406],
        [  6.3236,   0.3011,  -4.6668],
        [ 11.7795,  -3.4697,   3.5624],
        [  3.1271, -10.8343,   2.6442],
        [ -9.1433,  -3.5003,  -3.0209],
        [  8.5093,  -8.1764,  -3.2381],
        [  7.4132,  11.3526,   0.3322],
        [  6.1741,  -0.5329,   3.1913],
        [  9.6614,   9.1784,  -1.2087],
        [  1.1350,  14.0104,   0.6321],
        [  5.5240,  -2.1198,   6.2055],
        [ 11.9936,  -2.8677,   6.0614],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 12.2692,  -1.8882,   3.9253],
        [  3.6249, -10.7255,   5.6605],
        [ 12.2692,  -1.8882,   3.9253],
        [ -1.1885, -10.8211,   0.0865],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.5284, -10.7484,   5.3347],
        [  4.5503,   5.5359,   7.4735],
        [ -7.4887,   4.7364,   8.9074],
        [ 11.3691,   6.4760,   1.0787],
        [  6.2804,   2.0019,   0.8607],
        [ -3.5173,   7.5091,  -8.9585],
        [ 12.3357,   3.6001,   0.7903],
        [  5.6022,  -1.9833,   8.3081],
        [ -7.6525,  -6.6534,   5.1721],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.4877,  -0.7222,   8.8347],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.2419,  -0.2313,   4.2440],
        [  5.5101,  -2.1433,   9.5442],
        [ -7.6650,   4.5092,   0.2520],
        [  6.3119,   1.7797,   9.9186],
        [ -0.1626, -10.9796,   6.4629],
        [  6.3266,   0.3292,   9.3502],
        [  5.4426, -10.1181,  -5.1268],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.8675, -10.8815,   6.8561],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.2218,   4.0953,  -8.2072],
        [ -3.7964,  -9.9294,   7.7147],
        [  7.3724,  -9.0577,  -8.9843],
        [ -2.6069, -10.4298,  -6.1995],
        [  1.7228,  -4.7575,   8.3396],
        [  4.1348,   5.9514,  -0.8856],
        [ 11.6634,   5.8050,   9.1590],
        [  6.9284,  11.7061,  -0.3732],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.1666,   3.7653,   3.6075],
        [ -9.1795,  -3.3623,   8.5259],
        [ -8.3653,   3.4184,   9.3454],
        [ -0.6323,   7.9085,  -5.9544],
        [  1.2728, -11.0372,   9.8121],
        [  0.9788,  -4.8145,  -3.8661],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -6.8618,  -7.6322,   3.7027],
        [ -8.8934,   2.2597,   3.5831],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.8674,  12.8875,   6.7622],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 12.2921,  -1.7902,  -4.0870],
        [  3.8182,  -3.9347,  -1.7961],
        [ -7.6719,  -6.6260,   7.1468],
        [  0.5250, -11.0306,   5.5358],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.6022,  -1.9833,   8.3081],
        [  6.0310,  12.2795,  -6.4528],
        [ -9.3464,   0.5708,  -1.6397],
        [  6.3587,   1.1596,  -3.6629],
        [ -1.0012, -10.8576,   2.2867],
        [ 11.2767,  -4.6006,   0.7868],
        [ -1.6275, -10.7221,   3.2376],
        [ -6.4368,   5.8356,   3.3014],
        [  6.3303,   1.6145,  -8.4083],
        [ -3.5903,   7.4847,  -8.9112],
        [ 11.3367,  -4.4806,   0.4643],
        [ -8.3062,  -5.6008,   0.0287],
        [ 12.4013,  -1.2597,  -7.6474],
        [  3.1064,  13.5607,   6.7605],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.4351,  -1.8891,   3.5642],
        [  3.0719, -10.8449,  -4.1673],
        [  1.8721,  -4.7322,   0.5524],
        [ -9.1178,   1.5784,   2.1410],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.8154,   2.4611,   2.5198],
        [  6.1984,  -0.4327,   1.5438],
        [  3.6154,  -4.0640,  -7.9698],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.1292,   4.8183,  -5.7997],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.4824,  -5.2575,  -0.0795],
        [  3.7992,  -3.9474,  -8.9019],
        [  9.9217,  -6.7006,  -5.6728],
        [ 12.5763,   0.1333,  -9.2650],
        [  6.3236,   0.3011,  -4.6668]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 9.9973e-02, 8.1903e-02,  ..., 6.6941e-01, 8.5746e-01,
         4.9837e-01],
        [9.9973e-02, 1.0727e-04, 1.3917e-01,  ..., 5.7787e-01, 7.5749e-01,
         3.9975e-01],
        [8.1903e-02, 1.3917e-01, 0.0000e+00,  ..., 7.1538e-01, 8.7394e-01,
         5.2921e-01],
        ...,
        [6.6941e-01, 5.7787e-01, 7.1538e-01,  ..., 0.0000e+00, 3.4873e-01,
         2.1816e-01],
        [8.5746e-01, 7.5749e-01, 8.7394e-01,  ..., 3.4873e-01, 0.0000e+00,
         3.7310e-01],
        [4.9837e-01, 3.9975e-01, 5.2921e-01,  ..., 2.1816e-01, 3.7310e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.05989547,  5.40689332,  5.77219077,  5.62005629,  5.79543038,
        5.82345867, 10.51340829,  6.33075682, 12.27991991, 11.27656488,
        9.79039371, 11.80092578, 13.55868684,  6.19707944, 13.32613445,
       14.05633441,  5.91677188, 12.33167382, 10.27716434, 12.41366754,
       11.32149259, 12.41366754, 10.88618114, 10.8207297 , 11.31276109,
        7.16598273,  8.86082124, 13.08414689,  6.59176018,  8.29203356,
       12.85032836,  5.94293185, 10.14044489, 13.77222752, 12.5085995 ,
        5.5748624 ,  9.36093562,  6.24614905,  5.91222346,  8.89301516,
        6.55800267, 10.98076873,  6.33517349, 11.4890861 ,  7.822854  ,
       10.91601655, 12.18045978,  6.06416957, 12.889695  , 10.63038565,
       11.67875855, 10.7506423 ,  5.05981914,  7.24677664, 13.02818702,
       13.60275211, 10.7167285 ,  8.99276564,  9.77588801,  9.0368458 ,
        7.9337647 , 11.11038551,  4.91296205, 10.61047596, 10.26327895,
        9.17595899, 13.6660671 ,  9.38232121, 13.77604293,  9.50654967,
       12.4217523 ,  5.48276038, 10.1371552 , 11.04313641, 12.84056604,
        5.94293185, 13.68063098,  9.36378438,  6.46356806, 10.9036226 ,
       12.17901033, 10.84493732,  8.68832398,  6.53290663,  8.30125201,
       12.18998195, 10.01804223, 12.46513646, 13.91198191,  6.53290663,
        9.62238828, 11.27160035,  5.08910491,  9.25336371,  9.77777346,
        9.15253344,  6.2134961 ,  5.43943791,  5.79391825,  7.03733775,
       10.95593729,  9.97964013,  5.47866221, 11.97238624, 12.5769659 ,
        6.33075682])}
{'P': tensor([[4.1803e-02, 1.4078e-01, 1.3020e-01,  ..., 2.1624e-02, 5.2964e-05,
         5.0086e-05],
        [2.9181e-03, 4.0850e-02, 2.2819e-01,  ..., 1.8369e-02, 4.3851e-05,
         3.5068e-05],
        [3.4602e-03, 7.9925e-03, 5.0868e-02,  ..., 2.1693e-02, 5.4092e-05,
         4.2870e-05],
        ...,
        [1.9179e-03, 3.3138e-03, 3.4712e-03,  ..., 9.9382e-02, 1.1947e-04,
         3.6436e-05],
        [7.8986e-04, 3.5652e-04, 3.3415e-04,  ..., 1.0333e-04, 4.3205e-02,
         2.6949e-02],
        [5.4138e-04, 2.9576e-04, 2.8185e-04,  ..., 3.6723e-05, 4.3399e-05,
         1.3112e-02]]), 'X': tensor([[  1.7815,  -4.7481,   4.4289],
        [  3.2366,  -4.2732,   5.1532],
        [  3.3474,  -4.2162,   5.2455],
        [  2.5813,  -4.5471,   5.3597],
        [  3.9515,  -3.8425,   6.1729],
        [  5.1196,  -2.7160,   4.2705],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.4721,  -1.3097,  -7.4377],
        [  4.7482, -10.3907,   5.9149],
        [ -1.7290,   7.8825,  -3.3831],
        [ 11.0864,   7.0387,  -0.1238],
        [ -3.4488, -10.0934,   2.5195],
        [  3.6469,   6.3624,  -1.8414],
        [ 11.2517,  -4.6495,   2.9800],
        [ 10.0619,   8.6627,  -9.2884],
        [ -0.0410,   7.8591,   9.3457],
        [ -3.6381,   7.4683,   7.3544],
        [  5.5294,  -2.1106,   9.0626],
        [ -8.1666,   3.7653,   3.6075],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.2026,  12.1774,  -1.5887],
        [  2.1440, -10.9811,   8.5281],
        [  6.9314,  -9.3435,  -1.2729],
        [ -6.0844,  -8.3955,   0.7705],
        [  6.9693,  -9.3200,  -8.3822],
        [  0.1460,   7.8340,  -9.4614],
        [ -6.4620,  -8.0449,   6.2719],
        [ -7.0253,  -7.4493,  -1.5978],
        [  5.6523,   3.9443,  -2.9449],
        [ -1.9487,   7.8590,  -3.3330],
        [ -1.0407,   7.9166,  -6.6021],
        [  5.1203,  -2.7151,   6.2110],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.2555,   2.1474,   4.1685],
        [  5.9003,  12.3550,   2.1917],
        [ 11.1469,   6.9237,   1.6160],
        [  0.7685,  -4.8098,  -2.9310],
        [  0.5250, -11.0306,   5.5358],
        [  2.2860,   7.1916,  -6.1011],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.1667,  -2.6543,   1.2879],
        [  6.0073,   3.0986,   6.7937],
        [  2.9461,  -4.4077,  -0.4831],
        [  9.2692,   9.6370,   7.0746],
        [ 10.3251,   8.2928,  -3.0006],
        [ -0.5385,   7.9037,   6.3830],
        [  1.6422,   7.4586,  -6.8056],
        [  4.3554,   5.7393,  -9.0766],
        [  6.1989,  -9.7591,   0.8289],
        [ 10.3251,   8.2928,  -3.0006],
        [ -8.9636,   2.0648,   8.1946],
        [ -4.5778,  -9.5013,  -5.0667],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.5813,  -4.5471,   5.3597],
        [  4.3554,   5.7393,  -9.0766],
        [ -9.1231,  -3.5739,   8.4708],
        [ -1.7009,   7.8851,  -8.2437],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.5240,  -2.1198,   6.2055],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.5663,   7.0526,  -6.8769],
        [ -1.3178, -10.7939,   0.4236],
        [ -9.4430,  -0.1897,   4.8383],
        [  4.1537,   5.9340,  -6.0423],
        [  3.9895,   6.0816,  -6.5235],
        [  2.5318,  13.7231,   6.3692],
        [ -9.1178,   1.5784,   2.1410],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -8.3717,  -5.4769,  -9.2228],
        [  6.2883,   0.0325,   5.2692],
        [ 12.4837,   2.7510,   0.4449],
        [ -2.1533, -10.5778,   1.0112],
        [ -8.0965,  -5.9712,   4.3466],
        [  1.6408,   7.4591,  -4.2294],
        [ -8.3653,   3.4184,   9.3454],
        [ -6.4620,  -8.0449,   6.2719],
        [ -3.5520,   7.4976,   9.1522],
        [  1.7232,  -4.7574,   3.1270],
        [  6.2975,   0.0947,   5.1791],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.8363,   2.4086,  -5.6296],
        [ -7.4807,   4.7463,   2.0616],
        [  2.2849, -10.9655,  -7.6542],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.5620,   7.0549,   8.4825],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.8602,   6.8905,   5.8779],
        [  2.0870,  -4.6877,   9.6823],
        [  9.8082,  -6.8396,   4.9615],
        [ -9.4203,  -2.0340,   2.5856],
        [ -3.0008, -10.2827,  -8.9480],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.4867,  12.5816,   2.9549],
        [ -2.2188,   7.8218,   8.6064],
        [  9.2811,  -7.4336,   5.1771],
        [ -6.2200,  -8.2737,   5.5079],
        [  4.2096,   5.8815,   1.9230],
        [ -0.5739,   7.9056,   0.6382],
        [  6.2555,   2.1474,   4.1685],
        [  3.3839,  -4.1966,   8.3574],
        [  5.1890,   4.7318,   0.9515],
        [  6.3550,   0.7129,   0.4917],
        [ -8.3653,   3.4184,   9.3454],
        [  5.3860,  -2.3418,   1.0826],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.1795,  -3.3623,   8.5259]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.8977e-02, 7.6589e-02,  ..., 2.5491e-01, 7.3243e-01,
         8.8230e-01],
        [6.8977e-02, 1.5170e-04, 7.6174e-03,  ..., 3.0583e-01, 7.7866e-01,
         9.0472e-01],
        [7.6589e-02, 7.6174e-03, 0.0000e+00,  ..., 3.1203e-01, 7.8379e-01,
         9.0724e-01],
        ...,
        [2.5491e-01, 3.0583e-01, 3.1203e-01,  ..., 1.0727e-04, 8.0549e-01,
         1.0354e+00],
        [7.3243e-01, 7.7866e-01, 7.8379e-01,  ..., 8.0549e-01, 2.1453e-04,
         3.5707e-01],
        [8.8230e-01, 9.0472e-01, 9.0724e-01,  ..., 1.0354e+00, 3.5707e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.0713364 ,  5.36061974,  5.38342077,  5.2287058 ,  5.51177501,
        5.79543038,  6.21275016,  9.562174  , 11.42420508,  8.06990483,
       13.13205357, 10.66631784,  7.33350642, 12.17451685, 13.27717982,
        7.85919687,  8.30729352,  5.91854397,  8.99276564, 12.22503754,
       13.6660671 , 11.18839169, 11.63383221, 10.36844813, 11.63765273,
        7.83534987, 10.31886066, 10.23946247,  6.89245352,  8.09704052,
        7.98469166,  5.79562691,  9.87720102,  6.61386436, 13.69162795,
       13.12218013,  4.87083348, 11.04313641,  7.54620849,  5.24137909,
        5.80860165,  6.7593917 ,  5.30158497, 13.37121923, 13.2430377 ,
        7.92200342,  7.63726134,  7.20482224, 11.56146873, 13.2430377 ,
        9.19837792, 10.54658597,  5.21280156,  5.2287058 ,  7.20482224,
        9.79814997,  8.06643487,  5.21280156,  5.91677188, 10.5575601 ,
        7.50500064, 10.87409005,  9.44486609,  7.24326992,  7.27340834,
       13.95472331,  9.25336371,  9.60080639, 10.00410625,  6.28835101,
       12.78327037, 10.794755  , 10.06023665,  7.6374603 ,  9.0368458 ,
       10.31886066,  8.29640613,  5.05989547,  6.29821673,  6.00663314,
        9.15866607,  8.85939626, 11.20100795,  6.19633139,  7.50564223,
       12.17901033,  7.46053281,  5.13125513, 11.95743558,  9.63742669,
       10.71162681, 11.80092578, 13.72594022,  8.1303945 , 11.89105259,
       10.35093485,  7.23276582,  7.92644746,  6.61386436,  5.39098669,
        7.02253235,  6.3949043 ,  9.0368458 ,  5.87305005,  9.24501829,
        9.77588801])}
{'P': tensor([[3.3803e-02, 1.8504e-01, 1.2895e-01,  ..., 3.1914e-05, 9.5628e-03,
         6.3144e-04],
        [3.1588e-03, 4.0170e-02, 6.7665e-02,  ..., 5.8261e-05, 3.6576e-03,
         3.5833e-04],
        [3.6961e-03, 1.5888e-02, 4.3740e-02,  ..., 7.3154e-05, 2.9731e-03,
         3.2124e-04],
        ...,
        [2.7477e-05, 2.3434e-05, 2.3704e-05,  ..., 1.7517e-01, 3.2840e-05,
         1.7781e-06],
        [4.6900e-02, 9.2518e-02, 1.3638e-01,  ..., 3.6539e-05, 4.1805e-02,
         8.8326e-04],
        [3.2089e-03, 4.5090e-03, 4.0576e-03,  ..., 1.0060e-06, 1.8121e-03,
         2.2901e-02]]), 'X': tensor([[  1.7903,  -4.7467,   0.1168],
        [  2.7558,  -4.4844,   0.0636],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.1140,  -3.7220,  -0.1273],
        [  4.0457,  -3.7737,  -0.4704],
        [ -7.5603,   4.6459,   8.2002],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.2630,   2.1057,  -5.2136],
        [  2.6614,  -4.5192,  -9.3641],
        [ -7.0825,   5.2073,  -5.0497],
        [  6.3515,   0.6418,  -6.5823],
        [  4.3459,  -3.5327,   0.3667],
        [  5.4336,  -2.2676,   5.6933],
        [ -0.0254, -10.9932,   3.2200],
        [ 12.5919,   0.4035,  -5.8460],
        [  4.3377,   5.7570,   1.3712],
        [ -8.9178,   2.1936,   7.5388],
        [ -7.6650,   4.5092,   0.2520],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.3373,  -4.6232,  -5.2246],
        [ -4.4503,   7.1378,   3.3751],
        [  2.7685, -10.8982,  -6.8928],
        [  2.6310, -10.9195,   3.8054],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.2790,  -0.0259,  -5.4116],
        [  4.4290,  -3.4596,   1.4181],
        [  1.7500,   7.4187,   4.4237],
        [ -6.4620,  -8.0449,   6.2719],
        [  3.1892, -10.8220,  -6.8409],
        [ -1.5261,   7.8988,   9.3768],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.3419,   0.5972,  -1.2841],
        [ -5.9938,  -8.4745,   3.9735],
        [  5.7845,   3.6652,   0.9175],
        [  9.1409,  -7.5789,   1.5814],
        [ -7.3554,   4.8985,  -0.6665],
        [ -6.7992,   5.4991,   9.9186],
        [  0.5960, -11.0335,   3.1144],
        [  6.3246,   1.6696,  -0.1544],
        [ -1.2454,   7.9127,  -0.5715],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.5428,  -9.5728,  -2.9897],
        [ 11.2767,  -4.6006,   0.7868],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -2.2188,   7.8218,   8.6064],
        [ -8.7544,   2.6096,   3.3146],
        [  2.0449,  13.8402,  -7.1642],
        [  4.5340,  -3.3629,   8.9652],
        [  2.6970,   6.9826,   9.7350],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.6050,   2.9451,  -0.7012],
        [ -0.1626, -10.9796,   6.4629],
        [ -0.0254, -10.9932,   3.2200],
        [ 12.0582,  -2.6640,   0.7132],
        [ -9.2974,   0.8371,   2.4852],
        [  5.2608,  -2.5256,  -9.1446],
        [  7.1821,  -9.1845,   9.8548],
        [  4.1348,   5.9514,  -0.8856],
        [  7.8982,  -8.6778,   0.0239],
        [  4.5503,   5.5359,   7.4735],
        [ -3.6381,   7.4683,   7.3544],
        [ -1.8823, -10.6558,  -3.8952],
        [ -3.1204, -10.2345,   9.2288],
        [ -2.1817,   7.8274,  -4.6298],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.4520,   7.1370,  -8.7867],
        [ -5.9225,  -8.5355,   2.4778],
        [  4.8432,   5.1968,   1.5006],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -6.0195,   6.1789,  -2.9496],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -8.7544,   2.6096,   3.3146],
        [  1.6232,  -4.7717,  -2.5655],
        [  9.0652,   9.8597,  -3.5949],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.3460,   1.4252,   1.2291],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.2188,  -2.5839,   2.1833],
        [  9.3836,   9.5075,  -5.9278],
        [ 11.5083,  -4.1177,   0.2389],
        [ -5.9940,   6.1985,   5.9566],
        [  6.2975,   0.0947,   5.1791],
        [ -4.2271,   7.2387,  -1.8316],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.7228,  -4.7575,   8.3396],
        [  3.3744,   6.5623,  -6.3636],
        [  3.2226,  -4.2802,  -4.0428],
        [ -1.3178, -10.7939,   0.4236],
        [  8.1528,  10.7452,   9.1493],
        [  5.6260,  -1.9401,   3.0259],
        [ -8.0087,   4.0173,   9.1091],
        [  2.2182,  13.8007,   5.1889],
        [  5.0277,   4.9589,   7.0175],
        [ -7.9298,   4.1367,  -3.9773],
        [  3.3094,  13.4969,  -3.6507],
        [  1.3403,  -4.8011,  -1.1698],
        [  0.5960, -11.0335,   3.1144]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0220, 0.0407,  ..., 0.4742, 0.0776, 0.6467],
        [0.0220, 0.0000, 0.0329,  ..., 0.4593, 0.0846, 0.6659],
        [0.0407, 0.0329, 0.0000,  ..., 0.4339, 0.0582, 0.6840],
        ...,
        [0.4742, 0.4593, 0.4339,  ..., 0.0000, 0.4337, 1.1092],
        [0.0776, 0.0846, 0.0582,  ..., 0.4337, 0.0000, 0.6757],
        [0.6467, 0.6659, 0.6840,  ..., 1.1092, 0.6757, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.07306274,  5.26342203,  5.30158497,  5.24137909,  5.54776934,
        5.53255882,  8.8737556 , 12.35625694, 12.79126785,  6.60752373,
        5.24461068,  8.7907923 ,  6.383897  ,  5.60064319,  5.88782913,
       10.99326577, 12.59840193,  7.20826219,  9.18358562,  8.89301516,
       13.01728313,  5.18047638,  8.41148722, 11.24434374, 11.23201619,
       10.1371552 ,  6.27905292,  5.62005629,  7.62232307, 10.31886066,
       11.28215429,  8.04483495,  6.06416957,  9.36093562, 10.37998078,
        6.84796202, 11.87412439,  8.83725081,  8.74470218, 11.04953767,
        6.54127431,  8.01009083,  8.30125201, 11.59512194, 12.17901033,
        8.49161154,  5.43544626,  9.47369628, 12.30858677,  8.1303945 ,
        9.13507321, 13.99048094,  5.64500512,  7.48538905,  9.37989631,
       11.1913598 ,  9.09502385, 10.98076873, 10.99326577, 12.3489324 ,
        9.3349858 ,  5.83560009, 11.65923396,  7.24677664, 11.73397309,
        7.16598273,  8.30729352, 10.8207297 , 10.69963954,  8.12580976,
       10.60226614,  8.41171369, 10.38897563,  7.10378919,  4.91336994,
       12.4013153 ,  8.62631801,  8.53014915,  9.24501829,  9.62958973,
        9.13507321,  5.04028538, 13.39372426,  7.88677538,  6.50409372,
        9.7712321 ,  5.82345867, 13.35832725, 12.22276278,  8.62260327,
        6.29821673,  8.38252937, 12.32408516,  5.05981914,  7.37902766,
        5.35774573, 10.87409005, 13.48809595,  5.95111804,  8.95983586,
       13.97779839,  7.06169391,  8.94395628, 13.89671314,  4.98461983,
       11.04953767])}
{'P': tensor([[4.5431e-02, 2.2306e-01, 1.8841e-01,  ..., 4.2677e-05, 1.2598e-04,
         4.9181e-05],
        [3.2446e-03, 4.2169e-02, 3.0886e-01,  ..., 2.8313e-05, 1.4635e-04,
         7.3721e-05],
        [3.0766e-03, 4.5451e-03, 5.1057e-02,  ..., 3.0853e-05, 2.0586e-04,
         1.1325e-04],
        ...,
        [3.8647e-04, 2.2495e-04, 1.8120e-04,  ..., 1.4991e-02, 3.7707e-06,
         1.9310e-07],
        [5.1420e-05, 4.3090e-05, 4.1974e-05,  ..., 3.9119e-04, 6.4690e-02,
         9.3962e-04],
        [4.7566e-05, 3.9252e-05, 3.7211e-05,  ..., 2.9851e-04, 2.7506e-03,
         2.7052e-01]]), 'X': tensor([[  1.8447,  -4.7372,   8.1235],
        [  2.9260,  -4.4162,   8.3808],
        [  3.3839,  -4.1966,   8.3574],
        [  2.1120,  -4.6818,   8.6144],
        [  4.5340,  -3.3629,   8.9652],
        [  4.7813,  -3.1139,   8.2691],
        [  9.4554,   9.4245,   9.8800],
        [  6.2770,  -0.0379,   2.4195],
        [  6.2630,   2.1057,  -5.2136],
        [  5.5626,  -2.0536,   8.4433],
        [ 11.3800,  -4.3919,   1.4066],
        [ -1.0012, -10.8576,   2.2867],
        [ -2.0445,   7.8469,  -0.5857],
        [  4.3377,   5.7570,   1.3712],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.3444,   0.5821,   6.9101],
        [  6.2419,  -0.2313,   4.2440],
        [  5.7218,  -9.9934,   6.5764],
        [  6.6319,  11.9066,   3.5165],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.5149,  -5.1905,   1.5239],
        [ -1.9487,   7.8590,  -3.3330],
        [  4.5954,  13.0099,  -4.4439],
        [  4.1297,  -3.7098,  -3.9838],
        [  3.4401,  13.4540,   4.6576],
        [  2.5382,   7.0672,   4.5097],
        [  2.0243,   7.3086,   7.5432],
        [ -6.0844,  -8.3955,   0.7705],
        [  9.9756,  -6.6330,  -3.3429],
        [  6.0890,  -0.8382,   2.9166],
        [ 11.2571,   6.7072,   8.3693],
        [  6.3505,   0.6235,  -8.5849],
        [  7.7794,  11.0629,   8.9069],
        [  9.3753,   9.5171,   3.9131],
        [  4.6992,   5.3690,   3.0935],
        [  6.0073,   3.0986,   6.7937],
        [  5.3860,  -2.3418,   1.0826],
        [  5.7845,   3.6652,   0.9175],
        [ 10.1503,   8.5414,  -7.2681],
        [ -2.0445,   7.8469,  -0.5857],
        [ -7.8126,   4.3066,   5.1791],
        [  2.3200,   7.1756,   1.1938],
        [  2.0937,  13.8293,  -2.3533],
        [ -4.1296,   7.2803,  -2.7194],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.2849, -10.9655,  -7.6542],
        [  6.2975,   0.0947,   5.1791],
        [ -8.1666,   3.7653,   3.6075],
        [  4.4039,   5.6903,   8.7366],
        [  0.1532,  -4.7422,  -7.0145],
        [  1.0093,   7.6567,   2.3526],
        [  2.5010,  -4.5736,  -0.8989],
        [  9.6614,   9.1784,  -1.2087],
        [ 12.6000,   1.3839,  -3.2992],
        [ -3.5029,   7.5138,   8.1853],
        [  5.8902,  -1.3887,   4.0388],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.2975,   0.0947,   5.1791],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.0961,   1.6529,   3.1553],
        [ -7.8126,   4.3066,   5.1791],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 12.2921,  -1.7902,  -4.0870],
        [  5.1519, -10.2387,   9.8191],
        [  1.8721,  -4.7322,   0.5524],
        [ -0.8675, -10.8815,   6.8561],
        [  9.5203,   9.3483,   5.8781],
        [  5.1142,  -2.7230,  -4.1966],
        [  8.3718,  -8.2953,   8.4666],
        [ -6.1914,  -8.2997,   8.5380],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.1426,   3.8048,   9.3542],
        [ -7.3470,  -7.0615,   5.0066],
        [  2.6613,   7.0021,   4.5783],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.2589,  -1.9312,   9.8657],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.3367,   7.1676,  -9.5263],
        [  2.5469,  13.7192,  -9.6935],
        [  4.8432,   5.1968,   1.5006],
        [ -7.7842,   4.3466,  -1.7548],
        [  0.1677,  -4.7447,  -6.1529],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.4691,   2.8525,  -5.5110],
        [ -9.0961,   1.6529,   3.1553],
        [  9.3560,   9.5391,   9.4943],
        [  5.0277,   4.9589,   7.0175],
        [ -2.7572,   7.7194,   5.9580],
        [ -9.1813,  -3.3550,   1.8827],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -2.0229, -10.6163,   8.6930],
        [  6.0310,  12.2795,  -6.4528],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.9470,  13.6085,   2.9809],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.7784,  -1.6401,   9.5587],
        [  2.6613,   7.0021,   4.5783],
        [  5.2199,   4.6861,  -1.9304],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.0039,   7.3173,  -1.5885],
        [  3.1064,  13.5607,   6.7605]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0299, 0.0374,  ..., 0.9478, 0.6221, 0.4389],
        [0.0299, 0.0000, 0.0082,  ..., 0.9772, 0.6419, 0.4091],
        [0.0374, 0.0082, 0.0000,  ..., 0.9851, 0.6443, 0.4017],
        ...,
        [0.9478, 0.9772, 0.9851,  ..., 0.0000, 0.8337, 1.3819],
        [0.6221, 0.6419, 0.6443,  ..., 0.8337, 0.0000, 0.9660],
        [0.4389, 0.4091, 0.4017,  ..., 1.3819, 0.9660, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.08372149,  5.29755487,  5.39098669,  5.13616224,  5.64500512,
        5.70591986, 13.35013374,  6.27715105,  6.60752373,  5.92952538,
       12.19805394, 10.9036226 ,  8.10886021,  7.20826219, 10.37698435,
       12.04924159,  9.36256096,  6.24614905, 11.51557166, 13.62896593,
        5.5513072 ,  9.97220799,  8.09704052, 13.79762415,  5.5513072 ,
       13.88683436,  7.50917551,  7.58378655, 10.36844813, 11.97957425,
        6.1463896 , 13.103709  ,  6.38105015, 13.52430176, 13.35927527,
        7.13501264,  6.7593917 ,  5.87305005,  6.84796202, 13.26590071,
        8.10886021,  8.92098752,  7.54127425, 13.98691426,  8.36995953,
       13.03002338, 11.20100795,  6.29821673,  8.99276564,  7.19534146,
        4.74468406,  7.72291946,  5.21280156, 13.32613445, 12.67576792,
        8.29022197,  6.05165248,  9.5642993 ,  6.29821673,  7.48311002,
        9.24501829,  8.92098752,  7.98469166,  9.78147542, 12.4217523 ,
       11.46176259,  5.08910491, 10.91601655, 13.3426541 ,  5.79391825,
       11.785566  , 10.35464916, 12.30695098,  9.85384623,  8.98765644,
       10.19036081,  7.49076899,  8.68832398, 12.41012046, 12.25806559,
        5.95297459,  7.53884104, 13.95361098,  7.10378919,  8.91551231,
        4.74770899,  4.91336994, 12.79126785,  9.24501829, 13.3614598 ,
        7.06169391,  8.19703512,  9.7751189 , 10.54658597, 10.80728788,
       13.68063098,  9.14973319, 13.92390689, 12.76408562,  6.00663314,
        7.49076899,  7.01475376,  5.94535529,  9.78285424,  7.58668144,
       13.91198191])}
{'P': tensor([[3.8172e-02, 1.6146e-01, 1.4519e-01,  ..., 8.6523e-03, 1.7260e-03,
         4.3008e-05],
        [4.1834e-03, 4.2440e-02, 3.5356e-02,  ..., 5.3621e-03, 2.5144e-03,
         5.1590e-05],
        [5.0202e-03, 2.5279e-02, 5.3394e-02,  ..., 5.0234e-03, 4.2096e-03,
         5.6302e-05],
        ...,
        [2.3520e-02, 5.2118e-02, 2.8393e-02,  ..., 7.4103e-02, 1.4254e-03,
         1.1267e-04],
        [3.5871e-04, 3.4922e-04, 3.9112e-04,  ..., 2.2933e-04, 1.1610e-01,
         8.3884e-04],
        [1.9887e-03, 1.4684e-03, 1.1609e-03,  ..., 3.2939e-03, 1.0904e-04,
         3.7373e-01]]), 'X': tensor([[  1.8721,  -4.7322,   0.5524],
        [  2.7558,  -4.4844,   0.0636],
        [  2.8890,  -4.4316,   1.1600],
        [  4.1140,  -3.7220,  -0.1273],
        [  4.3459,  -3.5327,   0.3667],
        [  2.9461,  -4.4077,  -0.4831],
        [  8.2154,  -8.4262,   8.0171],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.9470,  13.6085,   2.9809],
        [ -8.9606,  -4.0994,   1.7886],
        [  9.4516,  -7.2501,  -1.8868],
        [ 12.4013,  -1.2597,  -7.6474],
        [  2.7590,  13.6621,   9.4496],
        [  6.2247,  -0.3150,  -9.3657],
        [ -1.8367,   7.8718,  -8.6523],
        [ -4.8665,  -9.3204,  -5.2564],
        [  4.5954,  13.0099,  -4.4439],
        [  4.5800,   5.5034,  -3.9432],
        [ -8.5506,  -5.1154,   6.3423],
        [  9.9756,  -6.6330,  -3.3429],
        [  6.1579,  -0.5961,  -6.5595],
        [ -9.2707,   0.9661,   4.4863],
        [  4.4633,   5.6287,  -0.7754],
        [ -6.3306,   5.9272,   5.6935],
        [  6.3266,   0.3292,   9.3502],
        [  3.2647, -10.8066,  -0.3768],
        [ 10.9717,  -5.1646,  -6.9039],
        [  2.7976,  -4.4682,  -4.2020],
        [ -6.7992,   5.4991,   9.9186],
        [  6.0345,   3.0175,  -1.4424],
        [  0.2571,  -4.7594,   1.5618],
        [ -1.9646,   7.8571,   3.3817],
        [  6.3550,   0.7129,   0.4917],
        [ 12.2435,   4.0076,   8.1257],
        [  1.4749,  -4.7892,  -0.8089],
        [  3.9085,   6.1512,  -2.2560],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.3373,  -4.6232,  -5.2246],
        [ -2.0445,   7.8469,  -0.5857],
        [ 10.9875,   7.2207,  -8.8549],
        [ -0.8569,   7.9156,  -6.5557],
        [  1.3670,  -4.7990,   3.2603],
        [  5.1203,  -2.7151,   6.2110],
        [ 12.2589,  -1.9312,   9.8657],
        [  2.7121,   6.9743,  -8.4202],
        [  0.5250, -11.0306,   5.5358],
        [ -9.3616,  -2.4750,   3.7765],
        [  1.9241,   7.3503,   7.9004],
        [  5.8201,   3.5841,  -8.0609],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.5340,  -3.3629,   8.9652],
        [ -7.0948,   5.1940,   9.8171],
        [ -9.3810,  -2.3455,  -9.5194],
        [  5.5101,  -2.1433,   9.5442],
        [ 12.5154,   2.5072,  -9.3274],
        [ -8.1426,   3.8048,   9.3542],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.2902,   0.0451,   3.7116],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.2576,   2.1359,  -3.3398],
        [ -8.5149,  -5.1905,   1.5239],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.1372,   2.6726,  -6.7885],
        [  9.8082,  -6.8396,   4.9615],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.3587,   1.1596,  -3.6629],
        [ 11.6019,   5.9540,  -5.1715],
        [ -0.5385,   7.9037,   6.3830],
        [  5.9843,  -9.8679,  -6.1449],
        [  4.5491,  -3.3485,   6.8412],
        [  9.5018,  -7.1945,  -1.1892],
        [  6.6319,  11.9066,   3.5165],
        [ -9.2508,   1.0572,   2.9978],
        [  4.1537,   5.9340,  -6.0423],
        [ -5.9225,  -8.5355,   2.4778],
        [ -5.9938,  -8.4745,   3.9735],
        [  1.9505,  -4.7171,   9.9868],
        [ -5.1974,   6.7396,   3.3484],
        [  3.1480,  -4.3165,  -7.8002],
        [ -0.2586,   7.8825,  -9.9840],
        [ -3.5173,   7.5091,  -8.9585],
        [ -2.1533, -10.5778,   1.0112],
        [ -8.6681,  -4.8556,   9.2518],
        [  1.4306,  13.9620,  -0.2075],
        [  1.7500,   7.4187,   4.4237],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.3589,   0.8239,  -8.4230],
        [ -9.3616,  -2.4750,   3.7765],
        [  5.1203,  -2.7151,   6.2110],
        [ -2.1382,   7.8339,   7.3524],
        [  4.0457,  -3.7737,  -0.4704],
        [  0.9945,  -4.8144,  -0.1897],
        [  6.3516,   0.6425,   0.3425],
        [  3.4401,  13.4540,   4.6576],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.2659,  13.7894,   6.5436],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.4159,  -2.0739,   9.6841],
        [  4.4290,  -3.4596,   1.4181],
        [  6.2555,   2.1474,   4.1685],
        [  6.0310,  12.2795,  -6.4528],
        [  9.3624,   9.5318,   0.2779],
        [  1.6232,  -4.7717,  -2.5655],
        [  5.3118,   4.5452,   4.2114],
        [ -7.4602,   4.7718,  -4.6277]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0331, 0.0547,  ..., 0.1913, 0.3238, 0.6782],
        [0.0331, 0.0000, 0.0751,  ..., 0.1658, 0.3477, 0.6817],
        [0.0547, 0.0751, 0.0000,  ..., 0.2408, 0.2726, 0.7261],
        ...,
        [0.1913, 0.1658, 0.2408,  ..., 0.0000, 0.5134, 0.6004],
        [0.3238, 0.3477, 0.2726,  ..., 0.5134, 0.0000, 0.9248],
        [0.6782, 0.6817, 0.7261,  ..., 0.6004, 0.9248, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.08910491,  5.26342203,  5.29009836,  5.54776934,  5.60064319,
        5.30158497, 11.76830981,  9.68324553, 13.92390689,  9.85384623,
       11.9120321 , 12.46513646, 13.93790235,  6.23262289,  8.08320235,
       10.51437947, 13.79762415,  7.15990676,  9.96391045, 11.97957425,
        6.18668463,  9.32094055,  7.18354885,  8.67227386,  6.33517349,
       11.28895489, 12.12641969,  5.27178363,  8.74470218,  6.74686071,
        4.7663526 ,  8.09899874,  6.3949043 , 12.88270018,  5.01114719,
        7.28793663,  8.25828409,  5.18047638,  8.10886021, 13.14777433,
        7.96181288,  4.98989906,  5.79562691, 12.41012046,  7.48311002,
       11.04313641,  9.68324553,  7.59795445,  6.83515201,  7.88677538,
       11.51811603,  5.64500512,  8.79284466,  9.66977802,  5.91222346,
       12.76408562,  8.98765644,  5.1224079 ,  6.29034958,  8.83725081,
        6.61211224,  9.97220799, 12.9089559 ,  6.69389431, 11.95743558,
       11.72749923,  6.46356806, 13.04051744,  7.92200342, 11.54071859,
        5.64865173, 11.9182956 , 13.62896593,  9.3109884 ,  7.24326992,
       10.38897563, 10.37998078,  5.10446653,  8.51088913,  5.34250225,
        7.88677538,  8.29203356, 10.794755  ,  9.9353921 , 14.03505624,
        7.62232307, 10.1371552 ,  6.41202986,  9.68324553,  5.79562691,
        8.12043983,  5.53255882,  4.9160962 ,  6.38399789, 13.88683436,
       12.10222204, 13.97430191,  6.77777606,  9.64157461,  5.62005629,
        6.61386436, 13.68063098, 13.3607379 ,  5.04028538,  6.99097013,
        8.85572292])}
{'P': tensor([[3.4887e-02, 1.4992e-01, 1.1573e-01,  ..., 4.4778e-05, 3.1567e-03,
         1.0693e-04],
        [2.3156e-03, 3.7141e-02, 2.2864e-01,  ..., 3.1815e-05, 6.0037e-03,
         1.2819e-04],
        [2.5384e-03, 4.8053e-03, 5.6006e-02,  ..., 4.0766e-05, 1.0815e-02,
         1.6735e-04],
        ...,
        [6.7298e-03, 3.3439e-03, 2.2714e-03,  ..., 1.2926e-01, 5.0691e-05,
         1.4142e-04],
        [2.7904e-05, 5.6249e-05, 8.4762e-05,  ..., 6.4600e-07, 2.1929e-02,
         1.5173e-05],
        [5.2071e-05, 7.7862e-05, 8.2264e-05,  ..., 9.9126e-06, 8.6497e-05,
         5.9033e-02]]), 'X': tensor([[  1.9503,  -4.7172,  -6.2313],
        [  3.4009,  -4.1874,  -5.9751],
        [  4.0078,  -3.8018,  -6.2168],
        [  3.9913,  -3.8139,  -6.8311],
        [  3.7237,  -3.9966,  -5.0376],
        [  3.2990,  -4.2415,  -7.4857],
        [  4.5814, -10.4485,   2.5496],
        [ -8.8363,   2.4086,  -5.6296],
        [  6.3119,   1.7797,   9.9186],
        [  5.0506,   4.9277,  -9.5094],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3246,   1.6696,  -0.1544],
        [  6.3516,   0.6425,   0.3425],
        [ -9.4773,  -1.0024,   5.2599],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.0449,  13.8402,  -7.1642],
        [  8.4691,  10.4571,  -5.0650],
        [  6.0345,   3.0175,  -1.4424],
        [  5.0277,   4.9589,   7.0175],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.6258,  -1.9404,  -6.0715],
        [  4.5800,   5.5034,  -3.9432],
        [  9.0424,  -7.6779,  -7.5286],
        [  6.2895,   0.0405,   5.7652],
        [ 12.5224,  -0.4497,   4.8448],
        [  2.6078, -10.9229,   6.2869],
        [ -8.9788,   2.0208,  -3.2867],
        [ 11.5966,   5.9666,   6.0083],
        [ -0.1626, -10.9796,   6.4629],
        [ 11.2571,   6.7072,   8.3693],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.3542,   0.6934,  -3.3840],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.6063,  13.0051,  -3.1456],
        [ 12.2692,  -1.8882,   3.9253],
        [ -2.0840,   7.8415,  -7.8657],
        [  8.8091,  -7.9037,   0.3631],
        [  5.6672,  -1.8631,   8.1020],
        [ -2.9497, -10.3028,  -7.3479],
        [  2.5813,  -4.5471,   5.3597],
        [  2.3589,  13.7668,  -3.6501],
        [  4.4290,  -3.4596,   1.4181],
        [  0.2732,  -4.7619,  -1.1577],
        [ -9.4117,   0.1160,   8.2574],
        [  4.7544,  -3.1425,  -1.5315],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -2.0229, -10.6163,   8.6930],
        [ -3.9858,  -9.8334,  -3.7323],
        [  4.2096,   5.8815,   1.9230],
        [  1.4306,  13.9620,  -0.2075],
        [  5.5626,  -2.0536,   8.4433],
        [  1.4489,   7.5256,   4.4529],
        [  5.9026,   3.3837,  -0.6573],
        [  0.9967,  -4.8144,  -9.7370],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.3246,   1.6696,  -0.1544],
        [  6.3360,   0.4258,  -9.2509],
        [ 10.9931,   7.2107,   7.7933],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.5636, -10.9293,   9.6519],
        [  4.5491,  -3.3485,   6.8412],
        [  5.2608,  -2.5256,  -9.1446],
        [  8.3718,  -8.2953,   8.4666],
        [  3.7992,  -3.9474,  -8.9019],
        [  4.5430,   5.5438,  -3.8669],
        [ -3.6381,   7.4683,   7.3544],
        [  3.9515,  -3.8425,   6.1729],
        [  2.2182,  13.8007,   5.1889],
        [ 10.5678,   7.9261,  -0.6100],
        [ -9.4739,  -1.2497,   0.5483],
        [  0.9445,   7.6736,   0.5095],
        [ 12.5770,   1.8512,  -0.6578],
        [ -8.0087,   4.0173,   9.1091],
        [  9.3560,   9.5391,   9.4943],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.7845,   3.6652,   0.9175],
        [  4.6992,   5.3690,   3.0935],
        [  9.7328,   9.0902,  -0.8917],
        [ -4.0868,   7.2981,   8.1874],
        [  3.9913,  -3.8139,  -6.8311],
        [ -9.2707,   0.9661,   4.4863],
        [  6.1222,  -0.7261,   8.8204],
        [ -8.0611,  -6.0302,   4.6962],
        [ -1.6906,   7.8860,   7.7142],
        [ -6.3489,   5.9117,   6.5161],
        [  3.7992,  -3.9474,  -8.9019],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.0560,  -0.9421,  -5.0334],
        [  3.5284, -10.7484,   5.3347],
        [  0.6470, -11.0352,  -1.1931],
        [  2.0419,  -4.6978,  -5.0983],
        [  1.8766,   7.3694,   1.1887],
        [ -1.5261,   7.8988,   9.3768],
        [ -1.6334, -10.7207,   7.1397],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2895,   0.0405,   5.7652],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.9022,  13.2906,   5.5788],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.6022,  -1.9833,   8.3081]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0314, 0.0460,  ..., 0.9254, 0.2334, 0.8569],
        [0.0314, 0.0000, 0.0189,  ..., 0.9549, 0.2218, 0.8457],
        [0.0460, 0.0189, 0.0000,  ..., 0.9714, 0.2045, 0.8552],
        ...,
        [0.9254, 0.9549, 0.9714,  ..., 0.0000, 1.1043, 1.2766],
        [0.2334, 0.2218, 0.2045,  ..., 1.1043, 0.0000, 1.0302],
        [0.8569, 0.8457, 0.8552,  ..., 1.2766, 1.0302, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.10444097,  5.39450305,  5.52416368,  5.5205189 ,  5.46244988,
        5.37343173, 11.4088043 ,  9.15866607,  6.55800267,  7.05627461,
        6.27905292,  6.54127431,  6.38399789,  9.53015376, 12.7125079 ,
        5.8828986 , 12.17901033, 13.99048094, 13.45642379,  6.74686071,
        7.06169391, 10.01804223,  5.9510489 ,  7.15990676, 11.86239961,
        6.28961988, 12.53047489, 11.22993218,  9.20340031, 13.04156075,
       10.98076873, 13.103709  ,  4.76969364,  6.39187285,  9.88310614,
       12.9089559 , 13.79676654, 12.41366754,  8.1137452 , 11.83509372,
        5.96556248, 10.7167285 ,  5.2287058 , 13.96746816,  5.62005629,
        4.76969364,  9.41244865,  5.69913817, 12.4217523 , 10.80728788,
       10.61047596,  7.23276582, 14.03505624,  5.92952538,  7.66377554,
        6.80370254,  4.91652556, 12.30695098,  6.54127431,  6.35029274,
       13.14690137,  6.19633139, 11.22596994,  5.64865173,  5.83560009,
       11.785566  ,  5.47866221,  7.16747878,  8.30729352,  5.51177501,
       13.97779839, 13.20988996,  9.55593374,  7.73151309, 12.7125079 ,
        8.95983586, 13.3614598 ,  9.36093562,  6.84796202,  7.13501264,
       13.31764181,  8.36445758,  5.5205189 ,  9.32094055,  6.16512856,
       10.06703358,  8.06517143,  8.67502348,  5.47866221, 12.77402351,
        9.37989631,  6.12886024, 11.31276109, 11.05413545,  5.1224079 ,
        7.60463349,  8.04483495, 10.84437705,  6.33075682,  6.28961988,
       11.89855267, 13.85158417,  5.07306274,  9.86089579, 12.42554808,
        5.94293185])}
{'P': tensor([[4.7158e-02, 2.9070e-01, 2.2045e-01,  ..., 3.3676e-04, 7.6972e-05,
         1.1162e-03],
        [1.1567e-02, 1.0182e-01, 2.5018e-02,  ..., 8.3235e-04, 1.7725e-04,
         2.8231e-03],
        [1.1284e-02, 2.7163e-01, 7.3833e-02,  ..., 6.1874e-04, 1.3355e-04,
         2.0844e-03],
        ...,
        [2.8789e-05, 3.1809e-05, 3.2323e-05,  ..., 2.8368e-02, 5.6972e-05,
         9.6695e-03],
        [4.2839e-06, 5.1936e-06, 5.2027e-06,  ..., 4.2488e-03, 1.6444e-02,
         1.5735e-03],
        [2.0643e-04, 2.2675e-04, 2.2905e-04,  ..., 2.0663e-02, 1.2300e-04,
         9.3321e-02]]), 'X': tensor([[  1.9505,  -4.7171,   9.9868],
        [  2.3376,  -4.6231,   9.8436],
        [  2.2644,  -4.6434,   9.7508],
        [  2.0870,  -4.6877,   9.6823],
        [  4.5340,  -3.3629,   8.9652],
        [  3.3839,  -4.1966,   8.3574],
        [  8.3791,  10.5409,   5.5590],
        [  7.7794,  11.0629,   8.9069],
        [ -7.8359,   4.2735,   7.2943],
        [ -3.4488, -10.0934,   2.5195],
        [ -6.2668,   5.9808,   2.5614],
        [ 11.7162,   5.6728,  -8.9372],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.3087,   1.8053,  -1.9481],
        [  0.9809,   7.6642,  -1.4217],
        [ 11.6165,  -3.8715,   6.4594],
        [  3.0876,   6.7527,  -5.3540],
        [  4.8653,   5.1694,  -4.0823],
        [ -0.5739,   7.9056,   0.6382],
        [  6.3516,   0.6425,   0.3425],
        [  5.1793,   4.7461,   0.3899],
        [ 12.5996,   1.3971,  -1.0694],
        [ -9.4671,  -0.5553,   0.1117],
        [  0.1460,   7.8340,  -9.4614],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.8704,  -9.3809,   7.5769],
        [ -3.3812, -10.1235,   7.4384],
        [  8.5198,  10.4091,  -8.6486],
        [  5.6248,   3.9984,   8.4933],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.9282,   2.1648,   8.4894],
        [  2.6078, -10.9229,   6.2869],
        [ 10.9162,  -5.2599,  -7.0812],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.9241,   7.3503,   7.9004],
        [  0.9392,  -4.8143,  -3.4649],
        [ -6.0844,  -8.3955,   0.7705],
        [ 11.9019,  -3.1379,  -3.4534],
        [  2.0547,   7.2957,  -1.9343],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.7218,  -9.9934,   6.5764],
        [  2.6970,   6.9826,   9.7350],
        [ -6.3489,   5.9117,   6.5161],
        [ -2.1382,   7.8339,   7.3524],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -9.4735,  -1.2620,  -3.6742],
        [  0.3231,  -4.7691,  -1.6120],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.9282,   2.1648,   8.4894],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.5383, -11.0274,  -4.9666],
        [  5.3118,   4.5452,   4.2114],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.3087,   1.8053,  -1.9481],
        [ -6.3489,   5.9117,   6.5161],
        [  3.1271, -10.8343,   2.6442],
        [ -1.5408,   7.8978,   5.3470],
        [ 12.6035,   0.7494,   7.1536],
        [ -1.1885, -10.8211,   0.0865],
        [  5.6022,  -1.9833,   8.3081],
        [  4.1637,  13.1898,  -1.9189],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -9.4748,  -0.7765,  -5.6695],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -8.6050,   2.9451,  -0.7012],
        [  0.9788,  -4.8145,  -3.8661],
        [  1.9241,   7.3503,   7.9004],
        [ 10.8218,  -5.4177,   6.8573],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 11.3118,   6.5957,  -4.5667],
        [  4.5800,   5.5034,  -3.9432],
        [ -8.9636,   2.0648,   8.1946],
        [  9.3753,   9.5171,   3.9131],
        [  3.2701,   6.6338,   9.9209],
        [ -9.1459,   1.4780,  -4.2403],
        [ -0.8885,   7.9160,   0.0571],
        [ -1.0407,   7.9166,  -6.6021],
        [  9.6587,   9.1817,  -0.8212],
        [  6.2026,  12.1774,  -1.5887],
        [  9.9750,  -6.6338,  -1.2440],
        [ -7.7842,   4.3466,  -1.7548],
        [  8.1528,  10.7452,   9.1493],
        [ 11.7418,   5.6070,  -1.4192],
        [ 12.2692,  -1.8882,   3.9253],
        [  4.5057,  -3.3894,  -0.8160],
        [  4.7813,  -3.1139,   8.2691],
        [ -2.1817,   7.8274,  -4.6298],
        [  3.3744,   6.5623,  -6.3636],
        [  6.7792,  11.8084,   1.6251],
        [  2.5813,  -4.5471,   5.3597],
        [ -3.5903,   7.4847,  -8.9112],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.0642,  -3.7599,  -3.4565],
        [  9.5261,  -7.1675,   1.4206],
        [  7.5309,  11.2617,   7.2656],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.1771, -10.9776,  -3.2007],
        [  2.4356,  -4.5942,  -5.0808],
        [  9.9058,  -6.7203,   5.6928],
        [  6.3497,   1.3680,  -1.9076],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.3305,   1.6119,   1.1538]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0112, 0.0086,  ..., 0.6788, 0.9808, 0.4515],
        [0.0112, 0.0000, 0.0026,  ..., 0.6763, 0.9793, 0.4486],
        [0.0086, 0.0026, 0.0000,  ..., 0.6764, 0.9792, 0.4488],
        ...,
        [0.6788, 0.6763, 0.6764,  ..., 0.0000, 0.3121, 0.2286],
        [0.9808, 0.9793, 0.9792,  ..., 0.3121, 0.0000, 0.5374],
        [0.4515, 0.4486, 0.4488,  ..., 0.2286, 0.5374, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.10446653,  5.1805345 ,  5.16611782,  5.13125513,  5.64500512,
        5.39098669, 13.46553623, 13.52430176,  8.92549752, 10.66631784,
        8.66273171, 13.01728313,  4.85995002,  4.91652556,  6.5618968 ,
        7.72668467, 12.24467041,  7.42512621,  7.09887557,  7.92644746,
        6.38399789,  7.02496223, 12.67680439,  9.48337063,  7.83534987,
        7.9337647 , 11.62770332, 10.67322527, 13.45124917,  6.90115901,
        9.26455795,  9.18690018, 11.22993218, 12.11735334,  5.04028538,
       12.32545637,  7.59795445,  4.90505566, 10.36844813, 12.30858677,
        7.57946459, 11.81423161, 11.51557166,  7.48538905,  8.67502348,
        8.12043983, 10.51437947,  9.55720651,  4.78002665, 13.26590071,
        9.18690018, 13.04217487, 11.1341798 ,  6.99097013,  5.63824971,
        6.5618968 ,  8.67502348, 11.27656488,  8.04664886, 12.62576026,
       10.88618114,  5.94293185, 13.83139024, 12.12448912,  9.50654967,
        5.68942572, 11.48434767,  9.55720651,  9.09502385,  4.91296205,
        7.59795445, 12.10222204, 10.14436912, 13.09426142,  7.15990676,
        9.19837792, 13.35927527,  7.39599986,  9.26455795,  7.96575782,
        7.98469166, 13.32645552, 13.6660671 , 11.97949273,  8.91551231,
       13.48809595, 13.01187484, 12.41366754,  5.63824971,  5.70591986,
        8.12580976,  7.37902766, 13.61600536,  5.2287058 ,  8.30125201,
       11.66453059,  5.5366698 , 11.92133129, 13.54774355,  9.78286733,
       11.1913598 ,  5.19986797, 11.9702762 ,  6.49538841, 12.05395056,
        6.53250937])}
{'P': tensor([[2.8142e-02, 1.9909e-01, 1.7466e-01,  ..., 1.1194e-04, 1.2138e-02,
         1.1526e-05],
        [4.0385e-03, 4.0254e-02, 1.0767e-02,  ..., 1.7771e-04, 2.1319e-02,
         1.9308e-05],
        [4.4173e-03, 1.1660e-01, 3.6183e-02,  ..., 1.5151e-04, 1.8054e-02,
         1.6160e-05],
        ...,
        [2.1508e-04, 1.9723e-04, 1.9701e-04,  ..., 1.2068e-01, 1.5038e-04,
         6.7217e-04],
        [9.4481e-04, 1.0165e-03, 9.9608e-04,  ..., 5.0358e-03, 1.0479e-01,
         9.2344e-04],
        [5.5284e-07, 5.2323e-07, 5.1700e-07,  ..., 1.7982e-03, 9.3830e-07,
         7.9835e-03]]), 'X': tensor([[  2.0419,  -4.6978,  -5.0983],
        [  2.4356,  -4.5942,  -5.0808],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.7237,  -3.9966,  -5.0376],
        [  4.2041,  -3.6509,  -4.9488],
        [  3.4009,  -4.1874,  -5.9751],
        [  3.2647, -10.8066,  -0.3768],
        [  3.9515,  -3.8425,   6.1729],
        [  5.8930,  12.3592,  -8.9201],
        [ 12.1379,   4.4125,  -3.8833],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.2182,  13.8007,   5.1889],
        [  6.3505,   0.6235,  -8.5849],
        [  1.3767, -11.0341,   9.9415],
        [  8.7131,  10.2216,   7.7915],
        [  0.1287, -11.0065,   6.5757],
        [  6.3413,   0.4883,  -6.8416],
        [ -4.7642,  -9.3860,  -5.1133],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.5224,  -0.4497,   4.8448],
        [ -5.1369,   6.7755,   8.3836],
        [  5.6258,  -1.9404,  -6.0715],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.0965,  -5.9712,   4.3466],
        [  2.4697,  -4.5836,   1.4593],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.3583,   1.1729,   8.7835],
        [ -3.9700,   7.3453,  -4.1486],
        [ 10.9875,   7.2207,  -8.8549],
        [  3.6249, -10.7255,   5.6605],
        [  7.7794,  11.0629,   8.9069],
        [ -3.9456,  -9.8542,   0.7466],
        [  2.5128, -10.9365,   9.9645],
        [  6.3515,   0.6418,  -6.5823],
        [  9.2811,  -7.4336,   5.1771],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.8363,   2.4086,  -5.6296],
        [  0.6952,  -4.8060,   8.8976],
        [  7.0515,  11.6194,   8.4289],
        [ -9.3444,   0.5821,   6.9101],
        [ -5.3382,   6.6533,   7.8054],
        [  8.1528,  10.7452,   9.1493],
        [  5.2188,  -2.5839,   2.1833],
        [ 10.0995,  -6.4739,   2.0974],
        [  5.6260,  -1.9401,   3.0259],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.6662,   9.1725,   8.1056],
        [  5.4242, -10.1261,   3.9595],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.4157,   0.0821,  -5.0025],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.9883,   3.1536,   6.2320],
        [  7.0515,  11.6194,   8.4289],
        [ -6.2668,   5.9808,   2.5614],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.2974,   0.8371,   2.4852],
        [ -6.4368,   5.8356,   3.3014],
        [ -2.1817,   7.8274,  -4.6298],
        [ -5.9938,  -8.4745,   3.9735],
        [  7.5784,  -8.9141,   5.8410],
        [ 11.3800,  -4.3919,   1.4066],
        [  7.0642,  11.6104,  -9.8316],
        [  4.0078,  -3.8018,  -6.2168],
        [ 10.0400,   8.6922,  -6.2847],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.2480, -11.0153,  -3.0183],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.1204,  -5.9308,   4.0761],
        [  2.2076,  13.8032,  -1.5099],
        [ 12.4901,  -0.7052,   1.7315],
        [  2.5620,   7.0549,   8.4825],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.4732,  -2.2041,  -4.3830],
        [ -1.9487,   7.8590,  -3.3330],
        [  5.9843,  -9.8679,  -6.1449],
        [  4.3377,   5.7570,   1.3712],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 12.4592,  -0.9168,  -6.7571],
        [  0.8700,  -4.8132,  -7.8042],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.8930,  12.3592,  -8.9201],
        [  9.0051,   9.9234,   5.4411],
        [ -3.4488, -10.0934,   2.5195],
        [ -1.6334, -10.7207,   7.1397],
        [ -0.9715, -10.8630,   0.3555],
        [  0.4996,   7.7738,  -4.4711],
        [ -9.0604,   1.7707,   1.1066],
        [  8.9284,  10.0036,  -6.9210],
        [  4.9337,  12.8566,  -3.6582],
        [  4.1348,   5.9514,  -0.8856],
        [ -2.2508, -10.5479,  -2.7235],
        [  3.9663,   6.1018,  -5.4104],
        [ -4.9217,   6.8979,  -1.7667],
        [  1.3103,  -4.8032,  -0.9300],
        [  4.5767,   5.5071,   3.9069],
        [  5.4426, -10.1181,  -5.1268],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.4426, -10.1181,  -5.1268],
        [ -0.5739,   7.9056,   0.6382],
        [  6.3236,   0.3011,  -4.6668],
        [  3.4401,  13.4540,   4.6576]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1691e-02, 1.0086e-02,  ..., 3.8976e-01, 8.7131e-02,
         9.4414e-01],
        [1.1691e-02, 0.0000e+00, 5.5902e-03,  ..., 3.9471e-01, 7.5464e-02,
         9.3829e-01],
        [1.0086e-02, 5.5902e-03, 0.0000e+00,  ..., 3.9784e-01, 7.8755e-02,
         9.4388e-01],
        ...,
        [3.8976e-01, 3.9471e-01, 3.9784e-01,  ..., 4.7405e-06, 4.2860e-01,
         7.8576e-01],
        [8.7131e-02, 7.5464e-02, 7.8755e-02,  ..., 4.2860e-01, 0.0000e+00,
         8.9863e-01],
        [9.4414e-01, 9.3829e-01, 9.4388e-01,  ..., 7.8576e-01, 8.9863e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.1224079 ,  5.19986797,  5.18047638,  5.46244988,  5.56809335,
        5.39450305, 11.28895489,  5.51177501, 13.6922425 , 12.91505111,
       12.12448912,  9.75401449, 13.97779839,  6.38105015, 11.11969815,
       13.43127131, 11.00726483,  6.36004393, 10.52587243, 11.85743425,
       12.53047489,  8.50268322,  5.9510489 , 12.0495521 , 10.06023665,
        5.20659892, 10.23946247,  6.46560181,  8.34949383, 13.14777433,
       11.32149259, 13.52430176, 10.61472636, 11.22142195,  6.383897  ,
       11.89105259, 11.05413545,  9.26455795,  9.15866607,  4.85603886,
       13.59170809,  9.36256096,  8.53014915, 13.48809595,  5.82345867,
       11.99633819,  5.95111804, 10.06286074, 13.32557133, 11.48734978,
       10.68031351,  9.41605684, 12.5085995 ,  6.76791483, 13.59170809,
        8.66273171,  6.69389431,  9.55720651,  9.3349858 ,  8.68832398,
        8.12580976, 10.37998078, 11.70016217, 12.19805394, 13.59056613,
        5.52416368, 13.27994304, 13.27994304,  9.94285511, 11.018084  ,
        7.09887557, 10.05558893, 13.97857591, 12.50996828,  7.50564223,
        5.84533345,  5.90035324,  8.09704052, 11.54071859,  7.20826219,
       12.18045978, 12.49291768,  4.89121805, 11.67875855,  5.26067019,
       13.6922425 , 13.40024224, 10.66631784, 10.84437705, 10.90638066,
        7.78980299,  9.23178204, 13.40850328, 13.77075174,  7.24677664,
       10.78533838,  7.27759233,  8.47371868,  4.978704  ,  7.16058587,
       11.4890861 , 12.91505111, 11.4890861 ,  7.92644746,  6.33075682,
       13.88683436])}
{'P': tensor([[4.2530e-02, 2.8675e-01, 2.2771e-01,  ..., 7.2189e-04, 4.2956e-06,
         5.7148e-05],
        [8.5630e-03, 6.8044e-02, 2.5034e-01,  ..., 1.2217e-03, 7.0946e-06,
         9.8041e-05],
        [1.3422e-02, 2.2823e-02, 9.2884e-02,  ..., 1.6928e-03, 9.5841e-06,
         1.3635e-04],
        ...,
        [2.7909e-04, 2.7174e-04, 2.6736e-04,  ..., 1.1500e-01, 2.4202e-05,
         4.9043e-03],
        [8.0857e-08, 8.0193e-08, 7.8476e-08,  ..., 1.3439e-05, 7.5041e-03,
         1.8973e-04],
        [1.3969e-06, 1.3609e-06, 1.3370e-06,  ..., 7.6191e-04, 3.0770e-06,
         1.1343e-02]]), 'X': tensor([[  2.0870,  -4.6877,   9.6823],
        [  2.2644,  -4.6434,   9.7508],
        [  2.3376,  -4.6231,   9.8436],
        [  4.5340,  -3.3629,   8.9652],
        [  3.3839,  -4.1966,   8.3574],
        [  2.9260,  -4.4162,   8.3808],
        [ -8.0087,   4.0173,   9.1091],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.6035,   0.7494,   7.1536],
        [  9.0652,   9.8597,  -3.5949],
        [ -2.6735, -10.4062,  -1.5911],
        [  0.9036,   7.6840,   9.9237],
        [  4.5814, -10.4485,   2.5496],
        [ -4.0868,   7.2981,   8.1874],
        [ -1.9600,   7.8577,   6.9121],
        [  2.3367,   7.1676,  -9.5263],
        [  5.0348,  -2.8229,   3.5436],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.1519, -10.2387,   9.8191],
        [  2.0899,  -4.6870,  -7.4260],
        [ -1.2454,   7.9127,  -0.5715],
        [  6.3606,   0.9204,   3.2525],
        [  6.2975,   0.0947,   5.1791],
        [ 10.0995,  -6.4739,   2.0974],
        [  5.0506,   4.9277,  -9.5094],
        [  1.7122,  -4.7591,  -3.6734],
        [ -8.8934,   2.2597,   3.5831],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.0054,  -1.0902,   8.3079],
        [  9.3753,   9.5171,   3.9131],
        [  6.3587,   1.1596,  -3.6629],
        [ 10.8670,  -5.3428,  -2.8637],
        [  0.1677,  -4.7447,  -6.1529],
        [  7.1441,  11.5529,   2.3232],
        [  0.9967,  -4.8144,  -9.7370],
        [ 12.0582,  -2.6640,   0.7132],
        [ -6.7745,  -7.7264,   8.4684],
        [  0.1931,  -4.7491,   7.4385],
        [ -8.8400,   2.3991,   1.3457],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.8930,  12.3592,  -8.9201],
        [ -8.0588,  -6.0341,  -2.3006],
        [  6.1984,  -0.4327,   1.5438],
        [  5.0506,   4.9277,  -9.5094],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.3858,   4.4260,  -9.6769],
        [  5.6414, -10.0303,   7.8525],
        [ -3.4986,   7.5152,   9.0712],
        [  2.7558,  -4.4844,   0.0636],
        [ -3.0644,   7.6439,  -1.4322],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.7685, -10.8982,  -6.8928],
        [  2.7558,  -4.4844,   0.0636],
        [  1.4150,  13.9647,  -4.0915],
        [  2.5382,   7.0672,   4.5097],
        [ -9.1960,  -3.2964,   1.6921],
        [  4.6231,   5.4556,   8.4973],
        [  9.1409,  -7.5789,   1.5814],
        [  6.0467,  -9.8369,  -6.5816],
        [ -9.4296,  -1.9458,   4.0833],
        [  2.4697,  -4.5836,   1.4593],
        [  5.3858,   4.4260,  -9.6769],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.1905,   3.7253,   4.1175],
        [  5.3924, -10.1396,  -4.0863],
        [  7.5784,  -8.9141,   5.8410],
        [  6.3413,   0.4883,  -6.8416],
        [ 10.9717,  -5.1646,  -6.9039],
        [  2.1120,  -4.6818,   8.6144],
        [ -4.0409,   7.3169,  -3.5779],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.9942,  -3.9990,   6.0448],
        [  3.3094,  13.4969,  -3.6507],
        [ -0.1626, -10.9796,   6.4629],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 11.1420,   6.9331,  -2.4779],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.9470,  13.6085,   2.9809],
        [  2.3200,   7.1756,   1.1938],
        [ 12.6038,   0.7665,  -3.4439],
        [  7.8375,  -8.7240,  -9.8913],
        [  5.8930,  12.3592,  -8.9201],
        [ -5.6174,   6.4708,  -1.7339],
        [  1.8766,   7.3694,   1.1887],
        [  4.7156,  -3.1831,  -8.0224],
        [  8.5198,  10.4091,  -8.6486],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.4495,   6.3026,   6.8257],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.4697,  -4.5836,   1.4593],
        [  5.1472,  -2.6801,   8.1194],
        [  7.3592,  11.3937,   6.2407],
        [ -3.5173, -10.0623,   4.8280],
        [ 10.8218,  -5.4177,   6.8573],
        [  0.5960, -11.0335,   3.1144],
        [  1.5246,  -4.7838,  -5.2095],
        [  2.8474,   6.8979,   6.6457],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.6928,  12.4710,   3.6001]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0059, 0.0085,  ..., 0.1190, 1.2112, 0.3532],
        [0.0059, 0.0000, 0.0026,  ..., 0.1241, 1.2093, 0.3473],
        [0.0085, 0.0026, 0.0000,  ..., 0.1264, 1.2086, 0.3447],
        ...,
        [0.1190, 0.1241, 0.1264,  ..., 0.0000, 1.1898, 0.4505],
        [1.2112, 1.2093, 1.2086,  ..., 1.1898, 0.0000, 1.0957],
        [0.3532, 0.3473, 0.3447,  ..., 0.4505, 1.0957, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.13125513,  5.16611782,  5.1805345 ,  5.64500512,  5.39098669,
        5.29755487,  8.95983586,  4.91296205,  5.81199422, 12.62576026,
       13.39372426, 10.74409834,  7.73692436, 11.4088043 ,  8.36445758,
        8.09842704,  7.53884104,  5.77219077, 14.09176259, 10.00802627,
       11.46176259,  5.13182506,  8.01009083,  6.42689705,  6.29821673,
       11.99633819,  7.05627461,  5.05774295,  9.17595899, 10.74409834,
        6.10360073, 13.35927527,  6.46356806, 12.10942412,  4.74770899,
       13.58333857,  4.91652556, 12.3489324 , 10.27572673,  4.75303127,
        9.15976746, 12.71843462, 12.3489324 , 13.6922425 , 10.0674887 ,
        6.2134961 ,  7.05627461,  4.91038884,  9.97822473,  6.97106912,
       11.50791859,  8.28968067,  5.26342203,  8.23525433, 11.75807252,
       11.24434374,  5.26342203, 14.03618649,  7.50917551,  9.76896904,
        7.15100303, 11.87412439, 11.54672755,  9.6282763 ,  5.20659892,
        6.97106912, 10.52357479, 13.25336001,  8.99791316, 11.48434767,
       11.70016217,  6.36004393, 12.12641969,  5.13616224,  8.35856487,
       13.13205357,  9.84314712, 13.89671314, 10.98076873,  9.55720651,
       13.12299145,  9.53015376, 13.96905756, 12.3828957 , 13.92390689,
        7.54127425, 12.62711252, 11.72749923, 13.6922425 ,  8.56889738,
        7.60463349,  5.68942572, 13.45124917,  8.92098752, 13.06957425,
        9.78286733,  5.20659892,  5.80312949, 13.56367857, 10.65929968,
       12.10222204, 11.04953767,  5.0209005 ,  7.46248947, 13.27717982,
       13.70893492])}
{'P': tensor([[2.5972e-02, 1.2982e-01, 1.2158e-01,  ..., 6.4751e-04, 5.3847e-05,
         1.3056e-05],
        [2.6722e-03, 3.9542e-02, 1.1949e-02,  ..., 1.6497e-03, 1.4910e-04,
         1.1943e-05],
        [2.1640e-03, 6.7116e-02, 2.7384e-02,  ..., 1.0437e-03, 8.9171e-05,
         7.9801e-06],
        ...,
        [2.0745e-04, 1.9038e-04, 1.8942e-04,  ..., 7.3963e-02, 9.2528e-04,
         3.5650e-06],
        [4.5656e-06, 4.6235e-06, 4.3937e-06,  ..., 2.1070e-03, 3.6534e-02,
         3.3179e-07],
        [5.3207e-05, 5.6225e-05, 4.9091e-05,  ..., 2.7741e-06, 1.9975e-06,
         2.2213e-02]]), 'X': tensor([[ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0270, 0.0280,  ..., 0.0478, 0.4545, 1.2053],
        [0.0270, 0.0000, 0.0109,  ..., 0.0747, 0.4362, 1.2230],
        [0.0280, 0.0109, 0.0000,  ..., 0.0752, 0.4451, 1.2304],
        ...,
        [0.0478, 0.0747, 0.0752,  ..., 0.0000, 0.4848, 1.1704],
        [0.4545, 0.4362, 0.4451,  ..., 0.4848, 0.0000, 1.2468],
        [1.2053, 1.2230, 1.2304,  ..., 1.1704, 1.2468, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.13182506,  5.37343173,  5.34250225,  5.43943791,  5.47702766,
        5.5205189 ,  8.92098752, 11.00508681, 10.57302546,  5.56809335,
        5.47702766,  6.58184474, 13.88683436,  8.06643487,  6.35029274,
       11.69264243, 11.33837396,  9.75200848, 11.85743425, 13.77604293,
        5.34250225,  9.43737523,  9.17016857,  9.27474386,  8.56889738,
        5.70591986,  6.65991877,  7.28793663, 13.91198191,  4.76969364,
        8.31625171, 10.52587243,  8.04664886,  5.01114719, 13.44593519,
        9.99211663, 12.05300943,  8.41148722,  5.19577146,  7.50035355,
        8.06990483,  6.80370254,  6.06416957, 11.22596994, 11.64746882,
        6.41202986, 13.89671314,  4.76969364, 13.69162795,  9.17016857,
        5.91854397,  8.1303945 , 12.79126785,  5.89074104, 10.04916678,
        8.99276564, 13.68672855,  6.19707944, 11.10066126, 11.05413545,
       10.5575601 , 13.66051884, 11.07637531,  7.53063885, 13.25017211,
        7.24677664,  9.25336371, 13.6922425 , 11.66453059,  8.8737556 ,
        8.1137452 , 11.21703654, 10.48721382,  5.46244988,  6.67121748,
        5.53175275,  8.92549752, 10.33958639,  9.91668182,  5.05981914,
        8.90405823, 13.39372426,  6.80559669,  7.39004226,  8.06643487,
        4.74468406,  6.34059021, 11.00508681,  8.50268322, 11.46176259,
        8.30125201,  8.05668609, 12.17451685,  8.1137452 ,  7.39004226,
       12.91143491,  8.11227668,  5.54776934,  7.72291946, 12.3344536 ,
       11.22596994,  5.70405525,  5.19065439,  7.37902766, 13.50484539,
       10.84437705])}
{'P': tensor([[3.9175e-02, 2.1421e-01, 1.8213e-01,  ..., 4.0703e-04, 1.1146e-05,
         4.6520e-05],
        [3.3235e-03, 3.8349e-02, 2.8088e-01,  ..., 5.1989e-04, 1.0062e-05,
         6.7043e-05],
        [3.1406e-03, 4.1695e-03, 4.6837e-02,  ..., 7.5793e-04, 1.1616e-05,
         1.0389e-04],
        ...,
        [4.5083e-04, 3.8509e-04, 3.5906e-04,  ..., 1.7209e-01, 3.9700e-04,
         4.4644e-03],
        [6.6369e-05, 5.2955e-05, 4.4406e-05,  ..., 1.9428e-06, 2.2958e-02,
         2.4766e-07],
        [1.2583e-05, 1.1121e-05, 1.0542e-05,  ..., 7.9867e-03, 1.2243e-04,
         7.6640e-02]]), 'X': tensor([[  2.1120,  -4.6818,   8.6144],
        [  2.9260,  -4.4162,   8.3808],
        [  3.3839,  -4.1966,   8.3574],
        [  4.5340,  -3.3629,   8.9652],
        [  4.7813,  -3.1139,   8.2691],
        [  5.1472,  -2.6801,   8.1194],
        [ -5.6423,   6.4537,  -1.2091],
        [  5.4429,  -2.2529,   9.7306],
        [ 12.5770,   1.8512,  -0.6578],
        [  0.3493,  -4.7727,  -6.5682],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.4597,  -0.9136,  -4.8763],
        [  3.1892, -10.8220,  -6.8409],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 11.2209,  -4.7092,   2.6117],
        [ -5.1974,   6.7396,   3.3484],
        [  6.9284,  11.7061,  -0.3732],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -7.4733,   4.7555,   6.2490],
        [  4.4633,   5.6287,  -0.7754],
        [  2.2182,  13.8007,   5.1889],
        [  4.5325,  -3.3643,   6.7699],
        [ 11.1420,   6.9331,  -2.4779],
        [  5.2188,  -2.5839,   2.1833],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.4336,  -2.2676,   5.6933],
        [ -5.1369,   6.7755,   8.3836],
        [  2.1504,  -4.6726,   8.7908],
        [  2.4572,   7.1085,   7.7376],
        [  5.9548,  -1.2273,  -2.6520],
        [  0.3715,  -4.7756,   0.2312],
        [  5.8201,   3.5841,  -8.0609],
        [ -4.2040,   7.2486,   2.9875],
        [  5.0348,  -2.8229,   3.5436],
        [  2.2849, -10.9655,  -7.6542],
        [ -7.0948,   5.1940,   9.8171],
        [ -9.4731,  -0.7120,  -6.0271],
        [  9.4925,   9.3811,   8.9842],
        [  2.0729,   7.2878,   7.0880],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.3246,   1.6696,  -0.1544],
        [  5.6523,   3.9443,  -2.9449],
        [  6.0788,  -0.8710,  -6.8719],
        [ -2.5056,   7.7720,   5.5878],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.4624,  -0.4633,  -7.3918],
        [  1.3339,   7.5627,  -0.9034],
        [ -7.5401,   4.6718,   9.0033],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.1085,  -4.9204,   0.0736],
        [  0.6757, -11.0361,  -1.3047],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.2707,   0.9661,   4.4863],
        [  5.1142,  -2.7230,  -4.1966],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.4501,  -1.7133,   2.2980],
        [ -0.5739,   7.9056,   0.6382],
        [  9.1620,  -7.5573,  -9.9882],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.3367,   7.1676,  -9.5263],
        [  3.3167,   6.6022,   6.3554],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 11.9641,  -2.9568,   7.7211],
        [ 10.6673,  -5.6649,  -4.1646],
        [  3.7992,  -3.9474,  -8.9019],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -4.5778,  -9.5013,  -5.0667],
        [  1.6954,  -4.7616,   9.8198],
        [ -4.2040,   7.2486,   2.9875],
        [  5.8450,  -9.9356,   3.0767],
        [  2.4149,  -4.6005,  -4.3339],
        [ -5.6792,  -8.7352,   4.0318],
        [  0.2732,  -4.7619,  -1.1577],
        [ -7.6293,  -6.6859,  -5.5161],
        [  0.1252, -11.0062,   7.2564],
        [  9.1820,   9.7334,  -2.7932],
        [ -8.5506,  -5.1154,   6.3423],
        [ 12.0042,  -2.8350,   2.6864],
        [ 11.6634,   5.8050,   9.1590],
        [  2.3261,   7.1726,   6.2522],
        [  5.6258,  -1.9404,  -6.0715],
        [  9.1620,  -7.5573,  -9.9882],
        [  2.2849, -10.9655,  -7.6542],
        [  0.5250, -11.0306,   5.5358],
        [ -9.1231,  -3.5739,   8.4708],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.6603,   2.8250,   5.9181],
        [ 12.0845,  -2.5772,  -4.6073],
        [  4.2339,  -3.6268,   6.3125],
        [ -4.4503,   7.1378,   3.3751],
        [  6.0955,   2.8210,  -4.3321],
        [  6.3505,   0.6235,  -8.5849],
        [ -3.9508,  -9.8515,   1.7923],
        [  0.6933,   7.7336,   2.1179],
        [  6.7792,  11.8084,   1.6251],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.2419,  -0.2313,   4.2440],
        [  6.3505,   0.6235,  -8.5849],
        [  6.0955,   2.8210,  -4.3321],
        [  0.9036,   7.6840,   9.9237],
        [ -9.4735,  -0.7274,  -2.6749],
        [  3.1064,  13.5607,   6.7605]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0167, 0.0249,  ..., 0.2220, 1.1676, 0.4243],
        [0.0167, 0.0000, 0.0082,  ..., 0.2377, 1.1816, 0.4091],
        [0.0249, 0.0082, 0.0000,  ..., 0.2455, 1.1884, 0.4017],
        ...,
        [0.2220, 0.2377, 0.2455,  ..., 0.0000, 1.0508, 0.6056],
        [1.1676, 1.1816, 1.1884,  ..., 1.0508, 0.0000, 1.5869],
        [0.4243, 0.4091, 0.4017,  ..., 0.6056, 1.5869, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.13616224,  5.29755487,  5.39098669,  5.64500512,  5.70591986,
        5.80312949,  8.5724042 ,  5.89074104, 12.7125079 ,  4.78544875,
       10.794755  , 12.49317563, 11.28215429, 12.52372629, 12.16901615,
        8.51088913, 13.60275211,  9.58072286,  8.8580671 ,  7.18354885,
       13.97779839,  5.64464807, 13.12299145,  5.82345867, 10.61472636,
        5.88782913,  8.50268322,  5.14371005,  7.52117132,  6.07992623,
        4.79003361,  6.83515201,  8.37954853,  5.77219077, 11.20100795,
        8.79284466,  9.4997987 , 13.34587027,  7.57685908, 13.01728313,
        6.54127431,  6.89245352,  6.14086582,  8.16584537, 13.2430377 ,
        9.47369628,  7.67939546,  8.87007495,  7.02326515, 12.1494023 ,
       11.05672136, 10.86693708,  9.50654967,  9.55593374,  9.32094055,
        5.79391825,  8.94395628,  9.60412735,  7.92644746, 11.87665501,
        5.35774573,  7.53884104,  7.38844703,  5.83560009, 10.0674887 ,
        9.5642993 , 12.32408516, 12.07818871,  5.47866221,  9.97822473,
       10.54658597,  5.05444288,  8.37954853, 11.52733581,  5.19577146,
       10.41909211,  4.76969364, 10.14436912, 11.00694688, 13.38091121,
        9.96391045, 12.3344536 , 13.02818702,  7.5403822 ,  5.9510489 ,
       11.87665501, 11.20100795, 11.04313641,  9.79814997, 12.27993098,
        9.10945678, 12.35625694,  5.5748624 ,  8.41148722,  6.7166297 ,
        6.38105015, 10.61417181,  7.76457095, 13.61600536, 12.35421461,
        6.24614905,  6.38105015,  6.7166297 ,  7.73692436,  9.5014146 ,
       13.91198191])}
{'P': tensor([[3.4857e-02, 1.2587e-01, 1.1183e-01,  ..., 1.2587e-01, 1.0582e-03,
         1.9975e-03],
        [1.4257e-03, 2.8220e-02, 2.0324e-01,  ..., 2.8220e-02, 1.9972e-03,
         1.9378e-03],
        [1.5218e-03, 3.5545e-03, 3.5294e-02,  ..., 3.5545e-03, 2.7332e-03,
         2.1735e-03],
        ...,
        [1.4257e-03, 2.8220e-02, 2.0324e-01,  ..., 2.8220e-02, 1.9972e-03,
         1.9378e-03],
        [1.7676e-04, 2.1236e-04, 2.1816e-04,  ..., 2.1236e-04, 7.4636e-02,
         3.7221e-05],
        [7.1701e-05, 2.4045e-04, 2.9637e-04,  ..., 2.4045e-04, 6.0514e-03,
         3.5611e-02]]), 'X': tensor([[ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0331, 0.0386,  ..., 0.0331, 0.3581, 0.3924],
        [0.0331, 0.0000, 0.0088,  ..., 0.0000, 0.3533, 0.3910],
        [0.0386, 0.0088, 0.0000,  ..., 0.0088, 0.3603, 0.3985],
        ...,
        [0.0331, 0.0000, 0.0088,  ..., 0.0000, 0.3533, 0.3910],
        [0.3581, 0.3533, 0.3603,  ..., 0.3533, 0.0000, 0.0525],
        [0.3924, 0.3910, 0.3985,  ..., 0.3910, 0.0525, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.14124612,  5.51177501,  5.5748624 ,  5.64464807,  5.64865173,
        5.38342077,  8.57560275, 14.02532758,  8.34949383, 11.32149259,
        5.13125513, 13.94008948, 11.83570132, 10.51437947,  5.81481044,
       11.38774149,  6.60752373,  5.98285319,  8.16714836, 13.45124917,
       14.10045797, 11.63765273,  5.87132718, 10.04916678,  7.53063885,
        9.9353921 ,  9.75075273,  5.16611782,  9.99211663,  9.55593374,
       11.54071859, 13.2430377 ,  7.96575782, 13.58333857, 12.78327037,
       14.03618649, 10.27572673,  5.90035324, 10.90638066,  6.28961988,
        8.62631801,  5.05981914, 12.42554808,  7.39004226, 11.48734978,
        7.54620849, 11.83075005,  7.5092834 , 11.04629471, 12.38881635,
       11.97957425,  7.24677664, 12.4013153 , 10.27572673, 10.82106505,
        6.24614905,  5.21280156,  7.15990676,  6.65368551,  9.79039371,
       13.09426142, 10.61472636,  9.20340031, 14.05687815,  7.53063885,
       11.67875855,  7.83534987,  7.70897675,  6.19707944, 13.46553623,
       13.56367857, 12.27835996,  6.3949043 , 13.46553623,  5.39450305,
        9.17905491,  8.41148722,  8.5724042 ,  7.76457095,  9.54154096,
       10.51437947,  9.45995955,  6.07992623,  6.06416957,  9.99211663,
       11.38774149, 10.90638066,  4.92497514, 13.66896739,  9.91668182,
       11.1341798 , 12.3828957 , 11.37939695, 10.99326577,  6.2322769 ,
        7.96575782,  5.77219077, 11.38774149, 10.36844813,  5.24137909,
       10.92597196,  5.82345867, 11.59512194,  5.51177501,  7.02253235,
       12.17901033])}
{'P': tensor([[3.7839e-02, 1.7603e-01, 1.6405e-01,  ..., 9.6600e-05, 4.4838e-05,
         9.5659e-06],
        [3.1178e-03, 3.1296e-02, 2.2922e-01,  ..., 1.1150e-04, 3.3189e-05,
         1.1683e-05],
        [3.3345e-03, 4.1492e-03, 4.6609e-02,  ..., 1.9295e-04, 4.7025e-05,
         2.0110e-05],
        ...,
        [2.5330e-05, 2.5257e-05, 2.4270e-05,  ..., 3.3301e-02, 2.6439e-04,
         9.6159e-06],
        [4.2847e-03, 2.8452e-03, 2.2113e-03,  ..., 3.5452e-05, 1.9170e-01,
         6.9739e-07],
        [4.2453e-07, 5.4497e-07, 6.0672e-07,  ..., 9.9657e-04, 3.8528e-07,
         2.0198e-02]]), 'X': tensor([[  2.1504,  -4.6726,   8.7908],
        [  2.9260,  -4.4162,   8.3808],
        [  3.3839,  -4.1966,   8.3574],
        [  4.5340,  -3.3629,   8.9652],
        [  4.7813,  -3.1139,   8.2691],
        [  5.1472,  -2.6801,   8.1194],
        [  5.8979,   3.3958,  -3.8713],
        [  2.0419,  -4.6978,  -5.0983],
        [  1.3705,  13.9723,   7.9266],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.5859,  -5.0393,   0.5476],
        [ -1.6220,   7.8917,   0.0315],
        [ -1.2454,   7.9127,  -0.5715],
        [ -7.9925,  -6.1423,   2.8990],
        [  9.3836,   9.5075,  -5.9278],
        [ 11.9019,  -3.1379,  -3.4534],
        [  5.5424,  -2.0885,   7.2487],
        [ -3.4488, -10.0934,   2.5195],
        [  1.1644, -11.0394,   7.8511],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.8400,   2.3991,   1.3457],
        [  7.9141,  -8.6656,   9.6948],
        [  8.4691,  10.4571,  -5.0650],
        [  2.7590,  13.6621,   9.4496],
        [  0.6757, -11.0361,  -1.3047],
        [  6.2555,   2.1474,   4.1685],
        [ -0.1626, -10.9796,   6.4629],
        [  5.8450,  -9.9356,   3.0767],
        [  2.0870,  -4.6877,   9.6823],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.3839,  -4.1966,   8.3574],
        [  3.6249, -10.7255,   5.6605],
        [ 10.2470,   8.4054,  -0.1177],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.6414, -10.0303,   7.8525],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -9.4501,  -1.7133,   2.2980],
        [ -7.6525,  -6.6534,   5.1721],
        [ -3.2486,   7.5925,  -1.3719],
        [  1.7815,  -4.7481,   4.4289],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.1271, -10.8343,   2.6442],
        [  1.0000,   7.6591,  -4.5011],
        [ -3.6381,   7.4683,   7.3544],
        [  9.1620,  -7.5573,  -9.9882],
        [  1.9204,   7.3518,   8.4781],
        [  2.6078, -10.9229,   6.2869],
        [  2.5010,  -4.5736,  -0.8989],
        [  3.9022,  13.2906,   5.5788],
        [ -9.4203,  -2.0340,   2.5856],
        [ -1.8823, -10.6558,  -3.8952],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.7298,  13.3537,  -9.4653],
        [ -0.5385,   7.9037,   6.3830],
        [  2.6321,  -4.5296,  -2.5159],
        [  9.3836,   9.5075,  -5.9278],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.6970,   6.9826,   9.7350],
        [  3.9591, -10.6391,  -6.2557],
        [  1.8721,  -4.7322,   0.5524],
        [  0.7958,  -4.8109,  -8.2248],
        [  5.0348,  -2.8229,   3.5436],
        [  3.0876,   6.7527,  -5.3540],
        [ -1.9600,   7.8577,   6.9121],
        [ 10.8218,  -5.4177,   6.8573],
        [ -1.2454,   7.9127,  -0.5715],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.1813,  -3.3550,   1.8827],
        [  8.7715,  -7.9390,   5.2900],
        [ -3.4488, -10.0934,   2.5195],
        [  6.3266,   0.3292,   9.3502],
        [  2.5128, -10.9365,   9.9645],
        [  9.6587,   9.1817,  -0.8212],
        [  6.3589,   0.8239,  -8.4230],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.3497,   1.3680,  -1.9076],
        [ 12.5258,  -0.4200,  -9.3493],
        [  2.5813,  -4.5471,   5.3597],
        [ 12.0772,  -2.6014,  -7.1832],
        [  7.1441,  11.5529,   2.3232],
        [  7.0515,  11.6194,   8.4289],
        [  5.1793,   4.7461,   0.3899],
        [  8.5198,  10.4091,  -8.6486],
        [  4.8674,  12.8875,   6.7622],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.9659,  -4.8144,  -0.9594],
        [ -5.1398,  -9.1367,   6.2074],
        [  1.4489,   7.5256,   4.4529],
        [  6.3460,   1.4252,   1.2291],
        [  5.9193,  -1.3176,   5.1739],
        [  1.7301,   7.4262,   5.0522],
        [  4.2275,  -3.6320,  -0.2714],
        [  4.1348,   5.9514,  -0.8856],
        [  9.2811,  -7.4336,   5.1771],
        [  4.3523, -10.5232,   8.2138],
        [  3.2990,  -4.2415,  -7.4857],
        [ -1.7290,   7.8825,  -3.3831],
        [ 11.4703,  -4.2006,   7.4086],
        [ -3.9456,  -9.8542,   0.7466],
        [  1.3339,   7.5627,  -0.9034],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.7162,   5.6728,  -8.9372]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4941e-02, 2.3106e-02,  ..., 6.0703e-01, 8.1022e-01,
         1.1695e+00],
        [1.4941e-02, 0.0000e+00, 8.2155e-03,  ..., 6.1008e-01, 8.2517e-01,
         1.1649e+00],
        [2.3106e-02, 8.2155e-03, 0.0000e+00,  ..., 6.1302e-01, 8.3331e-01,
         1.1636e+00],
        ...,
        [6.0703e-01, 6.1008e-01, 6.1302e-01,  ..., 0.0000e+00, 9.1204e-01,
         7.0777e-01],
        [8.1022e-01, 8.2517e-01, 8.3331e-01,  ..., 9.1204e-01, 3.0339e-04,
         1.6176e+00],
        [1.1695e+00, 1.1649e+00, 1.1636e+00,  ..., 7.0777e-01, 1.6176e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.14371005,  5.29755487,  5.39098669,  5.64500512,  5.70591986,
        5.80312949,  6.80559669,  5.1224079 , 14.0393949 ,  6.2322769 ,
        9.95552626,  8.05668609,  8.01009083, 10.08001894, 13.35832725,
       12.30858677,  5.92281142, 10.66631784, 11.10066126,  7.26074925,
        9.15976746, 11.73567759, 13.45642379, 13.93790235, 11.05672136,
        6.61386436, 10.98076873, 11.52733581,  5.13125513, 13.00413829,
        5.39098669, 11.32149259, 13.25336001,  5.34250225,  6.10774544,
       11.50791859, 12.07818871,  9.60412735, 10.14044489,  8.25828409,
        5.0713364 ,  4.76969364, 11.27656488,  7.72415048,  8.30729352,
       11.87665501,  7.5984809 , 11.22993218,  5.21280156, 13.85158417,
        9.63742669, 10.8207297 , 10.32715715, 12.07225469, 13.09737053,
       13.86480175,  7.92200342,  5.23878371, 13.35832725,  6.12886024,
        7.48538905, 11.35182589,  5.08910491,  4.8763135 ,  5.77219077,
        7.42512621,  8.09842704, 12.10222204,  8.01009083,  5.39450305,
        9.7751189 , 11.83075005, 10.66631784,  6.33517349, 11.22142195,
       13.32645552,  6.41202986,  9.60412735,  6.49538841, 12.53285633,
        5.2287058 , 12.35421461, 13.58333857, 13.59170809,  7.02496223,
       13.45124917, 13.77604293,  6.00732193, 12.25806559,  4.91038884,
       10.48315436,  7.66377554,  6.50409372,  6.06416957,  7.62509582,
        5.57341623,  7.24677664, 11.89105259, 11.38774149,  5.37343173,
        8.06990483, 12.21532282, 10.61472636,  7.67939546,  9.41244865,
       13.01728313])}
{'P': tensor([[4.3075e-02, 1.2454e-01, 1.2091e-01,  ..., 1.7883e-05, 7.4336e-04,
         4.6224e-06],
        [1.8071e-03, 3.2265e-02, 2.3237e-01,  ..., 8.2833e-06, 1.2193e-03,
         2.5917e-06],
        [1.8741e-03, 4.1217e-03, 4.0926e-02,  ..., 8.9032e-06, 1.7981e-03,
         2.8935e-06],
        ...,
        [1.1266e-04, 1.5399e-04, 1.3903e-04,  ..., 4.2194e-02, 3.0821e-06,
         1.0506e-04],
        [3.6479e-04, 3.0959e-04, 3.0671e-04,  ..., 9.0226e-06, 1.3703e-01,
         1.1369e-05],
        [9.6883e-05, 1.0012e-04, 8.7444e-05,  ..., 5.7421e-01, 3.0446e-06,
         1.0878e-01]]), 'X': tensor([[  2.1625,  -4.6697,   7.0658],
        [  3.9515,  -3.8425,   6.1729],
        [  4.2339,  -3.6268,   6.3125],
        [  4.5325,  -3.3643,   6.7699],
        [  4.5491,  -3.3485,   6.8412],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.3419,   0.5972,  -1.2841],
        [  2.3928,   7.1404,  -1.7035],
        [  2.3373,  13.7721,  -0.2442],
        [  0.6933,   7.7336,   2.1179],
        [ -9.4351,  -1.8891,   3.5642],
        [ 12.5154,   2.5072,  -9.3274],
        [  1.5653,  13.9377,   4.1773],
        [ -8.0608,   3.9362,  -5.9567],
        [  3.2226,  -4.2802,  -4.0428],
        [  5.1886,  -2.6250,   6.4677],
        [ -5.6792,  -8.7352,   4.0318],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.3031,   1.8477,  -7.1706],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.1196,  -2.7160,   4.2705],
        [  7.4132,  11.3526,   0.3322],
        [  2.6970,   6.9826,   9.7350],
        [  2.4572,   7.1085,   7.7376],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.3505,   0.6235,  -8.5849],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.1480,  -4.3165,  -7.8002],
        [ -8.1531,  -5.8748,   2.6934],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.9531,   5.0576,   5.6794],
        [  3.2226,  -4.2802,  -4.0428],
        [  1.7218,  -4.7576,  -3.3617],
        [  5.5294,  -2.1106,   9.0626],
        [  0.9967,  -4.8144,  -9.7370],
        [ -2.1817,   7.8274,  -4.6298],
        [ -1.3940, -10.7772,   6.7835],
        [  1.9204,   7.3518,   8.4781],
        [  4.5814, -10.4485,   2.5496],
        [ 12.5258,  -0.4200,  -9.3493],
        [  2.1504,  -4.6726,   8.7908],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.0305,   3.0296,   0.1375],
        [ -3.1671,   7.6158,   6.0765],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.1372,   2.6726,  -6.7885],
        [  4.5814, -10.4485,   2.5496],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.1477,   7.2550,   8.9688],
        [  5.0617,  12.7957,   8.3544],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.3549,   0.7096,   6.9198],
        [  4.6231,   5.4556,   8.4973],
        [  2.1625,  -4.6697,   7.0658],
        [  4.0591,   6.0202,  -1.6339],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.8407,  11.7666,   3.0045],
        [ -1.8367,   7.8718,  -8.6523],
        [  5.8979,   3.3958,  -3.8713],
        [ -3.4986,   7.5152,   9.0712],
        [  6.3124,   0.2057,  -0.4593],
        [  5.9526,   3.2525,  -4.3821],
        [  2.8994,   6.8675,  -9.5987],
        [ -7.5603,   4.6459,   8.2002],
        [  5.7151,  -1.7700,   3.1542],
        [  6.9284,  11.7061,  -0.3732],
        [ -5.6792,  -8.7352,   4.0318],
        [ -9.1624,  -3.4286,   2.0647],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.0961,   1.6529,   3.1553],
        [ -0.7599, -10.8995,  -6.0366],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.0140,  -1.0661,  -2.5040],
        [  9.9750,  -6.6338,  -1.2440],
        [  5.8979,   3.3958,  -3.8713],
        [ 11.9808,   4.9340,  -3.3715],
        [  0.3724,   7.7974,  -6.8722],
        [ 12.1588,   4.3364,  -8.0867],
        [  9.3753,   9.5171,   3.9131],
        [  6.3550,   0.7129,   0.4917],
        [  2.3889,  -4.6082,  -8.7291],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.1041, -11.0045,   2.4311],
        [ -2.6485, -10.4151,   0.8766],
        [  5.5424,  -2.0885,   7.2487],
        [ -9.3444,   0.5821,   6.9101],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.5935,   5.9740,  -4.3952],
        [  8.5198,  10.4091,  -8.6486],
        [  4.9452, -10.3187,  -7.8621],
        [ 12.4996,   2.6335,  -6.3166],
        [  9.9217,  -6.7006,  -5.6728],
        [  3.6249, -10.7255,   5.6605],
        [  5.6202,  -1.9508,   0.7471],
        [  0.6952,  -4.8060,   8.8976],
        [ -8.6603,   2.8250,   5.9181],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.0054,  -1.0902,   8.3079],
        [ -2.9497, -10.3028,  -7.3479],
        [  2.3261,   7.1726,   6.2522],
        [ -6.7643,  -7.7372,  -9.6663]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0411, 0.0414,  ..., 1.2256, 0.1232, 1.4153],
        [0.0411, 0.0000, 0.0088,  ..., 1.2202, 0.1553, 1.4132],
        [0.0414, 0.0088, 0.0000,  ..., 1.2290, 0.1598, 1.4220],
        ...,
        [1.2256, 1.2202, 1.2290,  ..., 0.0000, 1.1577, 0.2186],
        [0.1232, 0.1553, 0.1598,  ..., 1.1577, 0.0000, 1.3384],
        [1.4153, 1.4132, 1.4220,  ..., 0.2186, 1.3384, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.14607014,  5.51177501,  5.5748624 ,  5.64464807,  5.64865173,
        5.39098669,  9.36093562,  7.53063885, 13.96905756,  7.76457095,
        9.62238828, 12.76408562, 14.02532758,  8.97052115,  5.35774573,
        5.81481044, 10.41909211, 13.20597932,  6.23262289,  6.56833864,
        4.78258092,  5.79543038, 13.55868684,  7.48538905,  7.52117132,
        9.7712321 ,  9.7712321 , 10.412299  ,  6.38105015, 10.65929968,
        9.36093562,  5.34250225, 10.04916678, 11.51811603,  7.07902036,
        5.35774573,  5.05963217,  5.91854397,  4.91652556,  8.12580976,
       10.86693708,  7.5984809 , 11.4088043 , 12.53285633,  5.14371005,
        9.86089579,  6.74872981,  8.24808832,  8.36995953,  6.69389431,
       11.4088043 ,  8.97052115,  7.56617037, 13.76047999, 10.06703358,
        6.39438044,  7.15100303,  5.14607014,  7.26074925, 11.9120321 ,
       13.6105595 ,  8.08320235,  6.80559669,  8.28968067,  6.31576264,
        6.78325151,  7.4544866 ,  8.8737556 ,  5.98285319, 13.60275211,
       10.41909211,  9.78285424,  9.14973319,  9.62958973,  9.24501829,
       10.92597196, 11.69264243,  6.10774544, 11.97949273,  6.80559669,
       12.95702647,  7.80625981, 12.9089559 , 13.35927527,  6.3949043 ,
        5.19065439, 11.81423161, 11.0050379 , 10.74655346,  5.92281142,
        9.36256096, 12.91505111, 13.04217487, 13.45124917, 11.4424816 ,
       12.77402351, 11.97238624, 11.32149259,  5.94909709,  4.85603886,
        9.10945678, 12.40074901,  6.10360073, 10.7167285 ,  7.5403822 ,
       10.27716434])}
{'P': tensor([[3.2438e-02, 2.0322e-01, 1.3734e-01,  ..., 6.2972e-05, 7.3982e-03,
         7.2540e-05],
        [3.8133e-03, 4.4712e-02, 5.9003e-03,  ..., 9.8056e-05, 1.0183e-02,
         1.0893e-04],
        [6.3931e-03, 1.8221e-01, 3.7783e-02,  ..., 8.0534e-05, 1.0020e-02,
         8.7146e-05],
        ...,
        [1.3466e-04, 1.0766e-04, 1.2865e-04,  ..., 6.6765e-02, 9.3859e-05,
         1.5593e-02],
        [2.5948e-03, 3.4596e-03, 3.0573e-03,  ..., 1.1290e-03, 1.2905e-01,
         3.8872e-04],
        [3.7520e-04, 2.8232e-04, 3.3393e-04,  ..., 1.5248e-03, 8.3280e-05,
         7.8015e-02]]), 'X': tensor([[ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0179, 0.0220,  ..., 0.5524, 0.3960, 0.4416],
        [0.0179, 0.0000, 0.0196,  ..., 0.5603, 0.3899, 0.4589],
        [0.0220, 0.0196, 0.0000,  ..., 0.5409, 0.3744, 0.4489],
        ...,
        [0.5524, 0.5603, 0.5409,  ..., 0.0000, 0.4000, 0.4357],
        [0.3960, 0.3899, 0.3744,  ..., 0.4000, 0.0000, 0.6273],
        [0.4416, 0.4589, 0.4489,  ..., 0.4357, 0.6273, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.14999799,  5.29009836,  5.20659892,  5.40689332,  5.62005629,
        5.60064319,  9.5642993 ,  5.16077764,  7.50035355, 14.02532758,
        8.06990483,  9.94285511,  6.25586297, 11.89105259,  7.8575648 ,
       10.41909211, 11.62158554,  6.76661595,  9.78285424,  5.38342077,
        9.12220271, 10.91805683, 11.27656488,  8.09842704,  5.77219077,
        6.77777606,  9.97220799,  4.96231532,  5.04028538,  8.09842704,
       11.54071859,  5.53175275,  5.18277858,  9.32094055,  6.24187044,
        9.48337063, 12.91143491,  8.57560275,  9.05843201,  8.51928531,
       11.51811603, 13.77604293,  7.48311002,  6.07992623, 13.13205357,
       14.09357547,  8.98765644,  5.8828986 ,  9.50654967, 10.92597196,
        6.3949043 , 13.103709  ,  9.79039371, 12.53047489, 12.25806559,
       12.27993098, 13.32374056, 13.2430377 ,  6.00663314,  5.41348121,
        6.67121748, 12.18998195,  6.07992623, 11.20100795,  4.75303127,
        6.2322769 ,  8.68979052, 11.64436684, 12.9089559 ,  9.17905491,
       13.48809595,  9.5642993 , 11.66453059, 11.72802491,  8.56889738,
        7.54127425,  8.25828409, 11.85743425, 11.785566  ,  9.53015376,
       13.95361098, 10.1371552 ,  8.31625171, 12.88270018, 10.71162681,
        5.57341623, 12.27991991, 13.88683436, 13.79762415,  7.50035355,
        4.91296205,  5.91222346,  9.45995955,  9.43112957, 13.41713661,
        7.96181288, 11.83509372,  5.05534609, 10.06703358, 10.66631784,
        7.37902766, 13.3426541 ,  5.54776934,  8.30729352,  5.81481044,
        8.57560275])}
{'P': tensor([[4.5483e-02, 1.9168e-01, 1.1408e-01,  ..., 8.5260e-05, 5.6381e-05,
         2.9627e-05],
        [3.0958e-03, 5.6242e-02, 3.1459e-01,  ..., 2.4218e-04, 4.2981e-05,
         1.8926e-05],
        [3.7566e-03, 6.9626e-03, 1.1192e-01,  ..., 9.3674e-04, 6.6323e-05,
         2.7538e-05],
        ...,
        [8.6391e-07, 8.2668e-07, 9.3390e-07,  ..., 6.2708e-03, 7.8914e-07,
         2.4731e-07],
        [1.5243e-03, 6.8220e-04, 3.6141e-04,  ..., 2.8528e-07, 2.8354e-02,
         1.4711e-04],
        [3.7715e-03, 2.6676e-03, 2.0726e-03,  ..., 8.4733e-06, 1.6612e-04,
         3.7600e-01]]), 'X': tensor([[ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0342, 0.0562,  ..., 0.6914, 0.9726, 1.0849],
        [0.0342, 0.0000, 0.0267,  ..., 0.6762, 1.0035, 1.1130],
        [0.0562, 0.0267, 0.0000,  ..., 0.6511, 1.0285, 1.1179],
        ...,
        [0.6914, 0.6762, 0.6511,  ..., 0.0000, 1.5820, 1.2057],
        [0.9726, 1.0035, 1.0285,  ..., 1.5820, 0.0000, 1.0494],
        [1.0849, 1.1130, 1.1179,  ..., 1.2057, 1.0494, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.16077764,  5.48276038,  5.69913817,  5.63824971,  5.53255882,
        5.57341623,  9.36256096, 13.70893492, 10.31584102,  6.49538841,
       10.06437598, 10.91805683, 13.36092624,  6.65991877,  7.50917551,
        9.79814997,  6.32572323,  7.05627461,  4.89869205,  9.60080639,
        9.41605684,  8.66273171, 12.77402351,  9.85330646,  9.75401449,
        6.99097013, 12.91143491, 12.53345445,  5.39098669,  6.60752373,
       13.89886469,  7.8575648 ,  4.91336994,  5.64500512,  5.53255882,
        7.54127425,  8.7907923 , 12.17901033,  8.16714836,  7.6374603 ,
        9.41605684,  9.87720102,  8.85939626, 11.87665501,  9.26455795,
        5.05963217,  5.27178363, 14.10045797,  8.31625171,  7.32950385,
       12.46513646, 13.96905756,  9.48969978, 12.86648771, 13.31764181,
       11.1341798 ,  7.80625981, 11.70016217,  5.89074104,  6.50409372,
        9.14973319, 12.62576026, 11.83509372,  4.7663526 ,  7.60529892,
        7.8575648 , 12.88270018,  9.17905491,  8.62162372,  8.8580671 ,
        5.84533345,  7.822854  ,  9.09963894,  8.55982105,  6.54127431,
        7.54620849, 13.14690137, 12.50996828, 10.27572673,  5.29009836,
        8.29022197,  5.08910491,  9.27474386, 13.12299145, 11.73848065,
        5.91222346,  5.43943791, 10.84493732,  5.19986797,  7.07902036,
        7.19513806,  9.63742669,  7.02253235, 10.08001894,  9.86089579,
       13.20988996,  5.13182506,  8.51928531,  8.50278896, 13.01728313,
       11.81423161,  8.97052115, 13.36092624, 13.66991027,  9.96152788,
       10.412299  ])}
{'P': tensor([[4.4230e-02, 1.7522e-01, 1.4994e-01,  ..., 3.7670e-05, 1.9856e-04,
         7.6495e-04],
        [2.2667e-03, 4.3999e-02, 3.1688e-01,  ..., 5.7442e-05, 3.2718e-04,
         2.0922e-03],
        [2.7520e-03, 6.2054e-03, 6.1616e-02,  ..., 8.3871e-05, 5.0485e-04,
         3.5479e-03],
        ...,
        [1.4404e-04, 1.1622e-04, 1.0952e-04,  ..., 4.0788e-01, 1.1977e-02,
         2.2370e-05],
        [4.7148e-04, 3.0362e-04, 2.8241e-04,  ..., 1.2424e-03, 1.7847e-01,
         2.0698e-05],
        [2.5359e-06, 3.8259e-06, 4.2756e-06,  ..., 2.3551e-05, 3.5269e-04,
         7.6186e-03]]), 'X': tensor([[  2.2437,  -4.6489,   6.1430],
        [  3.9515,  -3.8425,   6.1729],
        [  4.2339,  -3.6268,   6.3125],
        [  3.3474,  -4.2162,   5.2455],
        [  4.5325,  -3.3643,   6.7699],
        [  3.2366,  -4.2732,   5.1532],
        [  2.5382,   7.0672,   4.5097],
        [  6.3589,   0.8239,  -8.4230],
        [  3.6249, -10.7255,   5.6605],
        [  7.8425,  -8.7202,   6.4530],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -5.3382,   6.6533,   7.8054],
        [  5.3118,   4.5452,   4.2114],
        [ 12.4013,  -1.2597,  -7.6474],
        [ 12.1379,   4.4125,  -3.8833],
        [ 10.9916,   7.2133,  -0.9179],
        [  0.7145,  -4.8071,  -2.3766],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.2242,  -0.3171,   7.4978],
        [ -1.9600,   7.8577,   6.9121],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.1940,  -2.1881,   7.1499],
        [  6.3460,   1.4252,   1.2291],
        [  2.6310, -10.9195,   3.8054],
        [  7.0171,  11.6439,  -5.0178],
        [ -2.1382,   7.8339,   7.3524],
        [ -9.1706,   1.3863,   8.4032],
        [  6.2365,  -0.2578,  -6.0212],
        [  2.4639, -10.9431,  -7.3718],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.0845,  -2.5772,  -4.6073],
        [  3.3633,  13.4794,   0.5193],
        [ 12.2692,  -1.8882,   3.9253],
        [  3.8111, -10.6787,  -4.6539],
        [ -2.6069, -10.4298,  -6.1995],
        [  7.0642,  11.6104,  -9.8316],
        [ 11.3691,   6.4760,   1.0787],
        [  6.3542,   0.6934,  -3.3840],
        [  1.5383, -11.0274,  -4.9666],
        [  5.7151,  -1.7700,   3.1542],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.7418,   5.6070,  -1.4192],
        [  2.5375,   7.0676,  -4.3490],
        [ -3.0008, -10.2827,  -8.9480],
        [ -1.6086,   7.8928,   4.3385],
        [  5.6202,  -1.9508,   0.7471],
        [  2.5974,   7.0363,  -7.7810],
        [  4.9337,  12.8566,  -3.6582],
        [ -4.9217,   6.8979,  -1.7667],
        [  7.5064,  -8.9651,  -0.0994],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.4541,  -0.3327,   8.8246],
        [  3.5965,  -4.0754,  -3.5035],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.4242, -10.1261,   3.9595],
        [ -5.6792,  -8.7352,   4.0318],
        [  5.0617,  12.7957,   8.3544],
        [  5.6022,  -1.9833,   8.3081],
        [  1.6408,   7.4591,  -4.2294],
        [ -2.5056,   7.7720,   5.5878],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.5424,  -2.0885,   7.2487],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.7845,   3.6652,   0.9175],
        [  5.7218,  -9.9934,   6.5764],
        [ -2.0445,   7.8469,  -0.5857],
        [ -1.6086,   7.8928,   4.3385],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.6485, -10.4151,   0.8766],
        [ 10.5057,  -5.9099,  -5.1599],
        [  4.7156,  -3.1831,  -8.0224],
        [  6.0310,  12.2795,  -6.4528],
        [  5.7784,  -1.6401,   9.5587],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.4366,  -0.1189,  -2.6736],
        [  4.4633,   5.6287,  -0.7754],
        [  3.6703,   6.3443,  -5.4401],
        [  5.8617,   3.4853,  -9.8080],
        [  0.3231,  -4.7691,  -1.6120],
        [ -5.0943,  -9.1682,   0.5073],
        [ -5.1398,  -9.1367,   6.2074],
        [  1.9503,  -4.7172,  -6.2313],
        [  6.9693,  -9.3200,  -8.3822],
        [  4.8051, -10.3703,   0.5268],
        [ 12.4013,  -1.2597,  -7.6474],
        [ 12.0042,  -2.8350,   2.6864],
        [ -8.0534,  -6.0429,   2.4583],
        [ 12.4597,  -0.9136,  -4.8763],
        [  0.1931,  -4.7491,   7.4385],
        [  2.1824,  -4.6647,   1.1985],
        [  3.2701,   6.6338,   9.9209],
        [ 11.2571,   6.7072,   8.3693],
        [ -3.2736, -10.1703,   3.8712],
        [ 11.5568,   6.0602,   9.3244],
        [  6.3301,   0.3638,  -7.5685],
        [  4.1537,   5.9340,  -6.0423],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.3569,   0.7591,   6.5011],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.4430,  -0.1897,   4.8383],
        [  0.2435,   7.8191,  -6.4709],
        [ -1.9646,   7.8571,   3.3817],
        [ 11.5456,   6.0861,   8.1101]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0340, 0.0420,  ..., 0.8331, 0.3229, 0.4576],
        [0.0340, 0.0000, 0.0088,  ..., 0.8519, 0.3567, 0.4263],
        [0.0420, 0.0088, 0.0000,  ..., 0.8598, 0.3649, 0.4175],
        ...,
        [0.8331, 0.8519, 0.8598,  ..., 0.0000, 0.6779, 1.2360],
        [0.3229, 0.3567, 0.3649,  ..., 0.6779, 0.0000, 0.7731],
        [0.4576, 0.4263, 0.4175,  ..., 1.2360, 0.7731, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.16205087,  5.51177501,  5.5748624 ,  5.38342077,  5.64464807,
        5.36061974,  7.50917551,  6.41202986, 11.32149259, 11.72802491,
        9.96152788,  8.53014915,  6.99097013, 12.46513646, 12.91505111,
       13.1471313 ,  4.85995002,  9.37989631,  6.2322769 ,  8.09842704,
        4.8763135 ,  9.79039371, 13.04051744, 12.38881635,  6.50409372,
       11.23201619, 13.59480906,  8.12043983,  9.27474386,  6.24187044,
       11.21703654,  9.37989631, 12.35625694, 13.89264822, 12.41366754,
       11.33837396, 10.7506423 , 13.59056613, 13.08414689,  6.39187285,
       11.1341798 ,  5.98285319,  5.35774573, 13.01187484,  7.5092834 ,
       10.71162681,  8.0550286 ,  5.94909709,  7.50035355, 13.77075174,
        8.47371868, 11.69264243,  9.49511524,  9.45995955,  5.43544626,
        9.97220799,  9.97964013, 11.48734978, 10.41909211, 13.76047999,
        5.94293185,  7.6374603 ,  8.16584537,  9.562174  ,  5.92281142,
        4.98461983,  6.84796202, 11.51557166,  8.10886021,  8.0550286 ,
        9.34127013, 10.74655346, 12.05395056,  5.68942572, 13.68063098,
        6.00663314,  7.37902766,  9.43737523,  7.18354885,  7.32950385,
        6.81961996,  4.78002665, 10.48840476, 10.48315436,  5.10444097,
       11.63765273, 11.42947632, 12.46513646, 12.3344536 , 10.06850677,
       12.49317563,  4.75303127,  5.14999799,  7.39599986, 13.103709  ,
       10.68416401, 13.0493368 ,  6.34059021,  7.24326992, 10.74409834,
        6.4020404 ,  9.13507321,  9.44486609,  7.822854  ,  8.09899874,
       13.05148826])}
{'P': tensor([[5.6721e-02, 2.0868e-01, 1.4406e-01,  ..., 4.5714e-05, 1.9704e-04,
         1.0861e-03],
        [1.7790e-02, 7.2400e-02, 1.8324e-01,  ..., 5.7302e-05, 2.5182e-04,
         1.4177e-03],
        [2.9304e-03, 3.0451e-03, 7.3492e-02,  ..., 1.4289e-04, 6.0647e-04,
         4.2267e-03],
        ...,
        [2.3412e-05, 2.2716e-05, 3.2586e-05,  ..., 1.2426e-01, 4.7559e-03,
         4.9937e-04],
        [3.2087e-04, 3.1040e-04, 2.1662e-04,  ..., 3.1264e-04, 1.7114e-01,
         1.5037e-03],
        [1.3244e-04, 1.3076e-04, 1.1670e-04,  ..., 9.7014e-05, 3.9025e-02,
         5.0715e-02]]), 'X': tensor([[  2.2644,  -4.6434,   9.7508],
        [  2.3376,  -4.6231,   9.8436],
        [  4.5340,  -3.3629,   8.9652],
        [  3.3839,  -4.1966,   8.3574],
        [  4.7813,  -3.1139,   8.2691],
        [  5.4429,  -2.2529,   9.7306],
        [  5.1886,  -2.6250,   6.4677],
        [  5.1292,   4.8183,  -5.7997],
        [ 11.6634,   5.8050,   9.1590],
        [  5.6260,  -1.9401,   3.0259],
        [  7.7794,  11.0629,   8.9069],
        [  5.7963,  -1.6018,   1.8856],
        [  5.6260,  -1.9401,   3.0259],
        [  3.5939,   6.4029,  -1.8357],
        [ -9.3411,  -2.6004,  -2.7062],
        [  4.5767,   5.5071,   3.9069],
        [ 10.0400,   8.6922,  -6.2847],
        [  8.3791,  10.5409,   5.5590],
        [  7.2340,  -9.1505,  -5.8321],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.1196,  -2.7160,   4.2705],
        [ -1.0407,   7.9166,  -6.6021],
        [  5.6414, -10.0303,   7.8525],
        [  1.7232,  -4.7574,   3.1270],
        [ -6.6998,   5.5954,   0.0823],
        [ -7.3446,   4.9113,  -5.9356],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.8450,  -9.9356,   3.0767],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.6614,  -4.5192,  -9.3641],
        [ -7.0253,  -7.4493,  -1.5978],
        [  1.9204,   7.3518,   8.4781],
        [  1.1644, -11.0394,   7.8511],
        [ 10.1798,   8.5002,  -5.8893],
        [ -3.5173,   7.5091,  -8.9585],
        [  5.5240,  -2.1198,   6.2055],
        [ -5.7347,  -8.6908,   9.2878],
        [  8.1217,  -8.5024,  -0.6149],
        [ -5.6651,   6.4380,   1.3660],
        [ -7.5401,   4.6718,   9.0033],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.9693,  -9.3200,  -8.3822],
        [ -9.2345,  -3.1345,   3.9715],
        [  7.7794,  11.0629,   8.9069],
        [  4.5057,  -3.3894,  -0.8160],
        [ -8.9033,   2.2329,  -8.8912],
        [  4.9957,  -2.8705,   6.5978],
        [  5.6095,  -1.9702,  -7.8233],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.1624,  -3.4286,   2.0647],
        [  6.3246,   1.6696,  -0.1544],
        [  5.6523,   3.9443,  -2.9449],
        [  6.9284,  11.7061,  -0.3732],
        [  5.4242, -10.1261,   3.9595],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.7558,  -4.4844,   0.0636],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -3.0008, -10.2827,  -8.9480],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.2608,  12.6977,   4.9166],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.2902,   0.0451,   3.7116],
        [ -0.8455, -10.8853,  -6.7400],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.8836,   3.4315,  -4.0327],
        [  2.0937,  13.8293,  -2.3533],
        [  6.3542,   0.6934,  -3.3840],
        [  6.2365,  -0.2578,  -6.0212],
        [ -2.0229, -10.6163,   8.6930],
        [  5.8902,  -1.3887,   4.0388],
        [  5.8450,  -9.9356,   3.0767],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.7963,  -1.6018,   1.8856],
        [  4.5491,  -3.3485,   6.8412],
        [  8.5198,  10.4091,  -8.6486],
        [  5.1861,   4.7361,  -3.7312],
        [ -8.0611,  -6.0302,   4.6962],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.0506,   4.9277,  -9.5094],
        [  6.2883,   0.0325,   5.2692],
        [  5.6202,  -1.9508,   0.7471],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.1667,  -2.6543,   1.2879],
        [  6.0788,  -0.8710,  -6.8719],
        [  9.9756,  -6.6330,  -3.3429],
        [ 10.7566,   7.6212,  -2.3279],
        [ -9.2373,  -3.1225,  -8.9763],
        [  4.4633,   5.6287,  -0.7754],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.1579,  -0.5961,  -6.5595],
        [ -1.0012, -10.8576,   2.2867],
        [  5.6414, -10.0303,   7.8525],
        [  0.6396,  14.0772,  -6.3786],
        [ -9.3444,   0.5821,   6.9101],
        [  0.3724,   7.7974,  -6.8722],
        [ -7.8126,   4.3066,   5.1791],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.5240,  -2.1198,   6.2055],
        [  1.8447,  -4.7372,   8.1235],
        [  3.9663,   6.1018,  -5.4104],
        [ -1.6086,   7.8928,   4.3385],
        [  2.5620,   7.0549,   8.4825]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0026, 0.0339,  ..., 0.8695, 0.3451, 0.1323],
        [0.0026, 0.0000, 0.0314,  ..., 0.8697, 0.3473, 0.1349],
        [0.0339, 0.0314, 0.0000,  ..., 0.8644, 0.3694, 0.1657],
        ...,
        [0.8695, 0.8697, 0.8644,  ..., 0.0000, 0.7210, 0.8760],
        [0.3451, 0.3473, 0.3694,  ..., 0.7210, 0.0000, 0.2481],
        [0.1323, 0.1349, 0.1657,  ..., 0.8760, 0.2481, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.16611782,  5.1805345 ,  5.64500512,  5.39098669,  5.70591986,
        5.89074104,  5.81481044,  7.03733775, 13.02818702,  5.95111804,
       13.52430176,  6.01356004,  5.95111804,  7.34250948,  9.69629   ,
        7.16058587, 13.27994304, 13.46553623, 11.66453059, 13.04217487,
        5.79543038,  7.98469166, 11.50791859,  5.05989547,  8.72896381,
        8.8353674 , 10.24557047, 11.52733581,  8.62329466,  5.24461068,
       10.23946247,  7.5984809 , 11.10066126, 13.26209545,  8.29203356,
        5.91677188, 10.412299  , 11.75807252,  8.57560275,  8.87007495,
       12.889695  , 11.63765273,  9.75200848, 13.52430176,  5.63824971,
        9.17905491,  5.76165934,  5.94541803, 10.33712258,  9.78285424,
        6.54127431,  6.89245352, 13.60275211, 11.48734978, 12.84056604,
        5.26342203, 12.42554808, 10.71162681,  8.89458273, 12.5085995 ,
       13.744387  ,  9.58072286,  6.29034958, 10.91805683,  5.19986797,
        6.81117802, 13.98691426,  6.39187285,  6.24187044, 10.80728788,
        6.05165248, 11.52733581,  5.70405525,  6.61211224,  9.53015376,
        6.01356004,  5.64865173, 13.45124917,  7.02326515, 10.06703358,
        9.36093562,  7.05627461,  6.28835101,  5.94909709, 12.67576792,
        5.80860165,  6.14086582, 11.97957425, 13.18278606,  9.75075273,
        7.18354885,  9.88310614,  9.79039371,  6.18668463, 10.9036226 ,
       11.50791859, 14.09176259,  9.36256096,  7.80625981,  8.92098752,
       12.49317563,  5.91677188,  5.08372149,  7.27759233,  8.0550286 ,
        7.50564223])}
{'P': tensor([[4.3893e-02, 1.9993e-01, 1.5594e-01,  ..., 5.6438e-05, 6.5305e-03,
         5.9565e-04],
        [4.0809e-03, 6.7503e-02, 5.0009e-01,  ..., 5.9716e-05, 8.9675e-03,
         1.6388e-03],
        [7.0574e-03, 1.2203e-02, 1.4958e-01,  ..., 1.1898e-04, 1.6154e-02,
         4.6468e-03],
        ...,
        [2.1910e-03, 1.0044e-03, 7.4127e-04,  ..., 3.8402e-02, 7.4620e-05,
         9.8092e-06],
        [6.6943e-05, 1.9070e-04, 3.0180e-04,  ..., 4.3672e-07, 1.3548e-02,
         1.1610e-04],
        [6.2222e-05, 5.4724e-05, 5.3830e-05,  ..., 1.3915e-04, 2.6935e-06,
         2.7912e-02]]), 'X': tensor([[  2.3373,  -4.6232,  -5.2246],
        [  3.7237,  -3.9966,  -5.0376],
        [  4.2041,  -3.6509,  -4.9488],
        [  3.4009,  -4.1874,  -5.9751],
        [  2.4356,  -4.5942,  -5.0808],
        [  4.0078,  -3.8018,  -6.2168],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.2914,   3.8037,   6.0978],
        [  5.8617,   3.4853,  -9.8080],
        [  1.3103,  -4.8032,  -0.9300],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.2974,   0.8371,   2.4852],
        [ -5.3736,   6.6310,  -1.5174],
        [  1.0391,  -4.8141,  -0.0768],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.1795,  -3.3623,   8.5259],
        [  3.3943,   6.5483,   1.0790],
        [  7.3592,  11.3937,   6.2407],
        [ 11.5456,   6.0861,   8.1101],
        [  7.0171,  11.6439,  -5.0178],
        [ -9.2373,  -3.1225,  -8.9763],
        [  1.1006,   7.6318,  -2.4877],
        [ -2.0229, -10.6163,   8.6930],
        [  2.3367,   7.1676,  -9.5263],
        [  0.9036,   7.6840,   9.9237],
        [ 10.3251,   8.2928,  -3.0006],
        [  3.3943,   6.5483,   1.0790],
        [  8.5198,  10.4091,  -8.6486],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -7.8126,   4.3066,   5.1791],
        [  7.9816,  10.8938,  -0.5378],
        [  5.6095,  -1.9702,  -7.8233],
        [  8.1528,  10.7452,   9.1493],
        [  2.7558,  -4.4844,   0.0636],
        [  0.6142,  14.0802,   0.5407],
        [ 12.2592,   3.9425,   2.0383],
        [ -1.3178, -10.7939,   0.4236],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.7643,  -7.7372,  -9.6663],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.3444,   0.5821,   6.9101],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.3705,  13.9723,   7.9266],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.3535,   0.6792,  -7.5101],
        [ -7.0120,   5.2825,   7.7762],
        [  1.4749,  -4.7892,  -0.8089],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.8994,   6.8675,  -9.5987],
        [  2.3376,  -4.6231,   9.8436],
        [  0.9445,   7.6736,   0.5095],
        [ -4.0409,   7.3169,  -3.5779],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 12.5770,   1.8512,  -0.6578],
        [  9.9756,  -6.6330,  -3.3429],
        [  4.9452, -10.3187,  -7.8621],
        [ -8.9282,   2.1648,   8.4894],
        [  2.1440, -10.9811,   8.5281],
        [  2.3376,  -4.6231,   9.8436],
        [  0.9809,  -4.8145,   6.8706],
        [ -1.6334, -10.7207,   7.1397],
        [  7.5930,  -8.9037,   7.2323],
        [  2.2849, -10.9655,  -7.6542],
        [ -6.2200,  -8.2737,   5.5079],
        [  5.7784,  -1.6401,   9.5587],
        [  2.6970,   6.9826,   9.7350],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.5149,  -5.1905,   1.5239],
        [  9.9756,  -6.6330,  -3.3429],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.3464,   0.5708,  -1.6397],
        [  4.0362,  13.2397,  -7.0030],
        [ -1.9646,   7.8571,   3.3817],
        [  3.9895,   6.0816,  -6.5235],
        [  5.9655,   3.2172,  -7.5953],
        [ -1.2454,   7.9127,  -0.5715],
        [  9.9217,  -6.7006,  -5.6728],
        [ -4.3963,   7.1629,   4.5263],
        [ 11.3800,  -4.3919,   1.4066],
        [  4.5503,   5.5359,   7.4735],
        [  3.9022,  13.2906,   5.5788],
        [  3.3943,   6.5483,   1.0790],
        [  8.3791,  10.5409,   5.5590],
        [ -7.4602,   4.7718,  -4.6277],
        [  1.3403,  -4.8011,  -1.1698],
        [ -6.3489,   5.9117,   6.5161],
        [ -1.5875,   7.8944,  -9.4286],
        [ -0.6323,   7.9085,  -5.9544],
        [  3.4401,  13.4540,   4.6576],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.5653,  13.9377,   4.1773],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.3589,   0.8239,  -8.4230],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.3367,   7.1676,  -9.5263],
        [ -7.0120,   5.2825,   7.7762],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 10.4994,  -5.9193,  -7.5153],
        [  2.0547,   7.2957,  -1.9343]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0283, 0.0385,  ..., 0.9439, 0.2464, 0.1882],
        [0.0283, 0.0000, 0.0102,  ..., 0.9721, 0.2542, 0.1965],
        [0.0385, 0.0102, 0.0000,  ..., 0.9823, 0.2568, 0.2012],
        ...,
        [0.9439, 0.9721, 0.9823,  ..., 0.0000, 0.9485, 0.9001],
        [0.2464, 0.2542, 0.2568,  ..., 0.9485, 0.0000, 0.4241],
        [0.1882, 0.1965, 0.2012,  ..., 0.9001, 0.4241, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.18047638,  5.46244988,  5.56809335,  5.39450305,  5.19986797,
        5.52416368, 12.62711252, 12.86648771,  6.81961996,  4.978704  ,
       10.01804223,  9.3349858 ,  8.53501777,  4.92497514,  7.71075226,
        9.77588801,  7.37576154, 13.56367857, 13.05148826, 13.59480906,
        9.75075273,  7.71075226, 10.80728788,  7.53884104,  7.73692436,
       13.2430377 ,  7.37576154, 13.45124917, 12.32545637,  8.92098752,
       13.50484539,  5.94541803, 13.48809595,  5.26342203, 14.09357547,
       12.87752052, 10.87409005, 11.83075005,  9.11473391, 10.27716434,
        7.27340834,  9.36256096,  9.88571713, 14.0393949 , 11.75618513,
        6.38968357,  8.77913428,  5.01114719, 10.71162681,  7.4544866 ,
        5.1805345 ,  7.73151309,  8.35856487, 12.27835996, 12.7125079 ,
       11.97957425, 11.4424816 ,  9.18690018, 11.18839169,  5.1805345 ,
        4.91336994, 10.84437705, 11.70168755, 11.20100795, 10.35093485,
        6.00663314,  7.48538905,  8.01009083,  9.97220799, 11.97957425,
        6.01356004,  9.36378438, 13.84126197,  8.09899874,  7.27340834,
        6.77777606,  8.01009083, 11.97238624,  8.40445872, 12.19805394,
        7.16598273, 13.85158417,  7.37576154, 13.46553623,  8.85572292,
        4.98461983,  8.67502348,  8.05242591,  7.9337647 , 13.88683436,
       14.10796466, 12.32408516, 14.02532758, 12.07818871,  5.13616224,
        9.61942445, 12.4013153 ,  6.41202986, 11.9120321 ,  7.53884104,
        8.77913428, 11.81423161,  9.17595899,  9.85330646, 12.05300943,
        7.57946459])}
{'P': tensor([[7.6195e-02, 1.9284e-01, 1.2935e-01,  ..., 1.1820e-05, 1.7966e-03,
         2.8240e-02],
        [2.2907e-03, 5.5286e-02, 3.5975e-03,  ..., 6.3543e-06, 1.1708e-03,
         6.6447e-02],
        [6.1758e-03, 2.1125e-01, 5.0435e-02,  ..., 8.8204e-06, 1.9300e-03,
         4.7533e-02],
        ...,
        [1.0940e-04, 7.2529e-05, 1.4056e-04,  ..., 6.3596e-02, 1.2356e-04,
         4.6032e-05],
        [2.9743e-05, 1.5553e-04, 7.7528e-05,  ..., 3.7417e-07, 1.4697e-02,
         1.7958e-03],
        [5.1232e-04, 1.2279e-03, 7.3810e-04,  ..., 7.5497e-06, 2.5642e-04,
         4.7642e-02]]), 'X': tensor([[  2.3376,  -4.6231,   9.8436],
        [  4.5340,  -3.3629,   8.9652],
        [  3.3839,  -4.1966,   8.3574],
        [  4.7813,  -3.1139,   8.2691],
        [  5.4429,  -2.2529,   9.7306],
        [  5.5101,  -2.1433,   9.5442],
        [ -1.3178, -10.7939,   0.4236],
        [  4.0362,  13.2397,  -7.0030],
        [  4.2275,  -3.6320,  -0.2714],
        [  9.6587,   9.1817,  -0.8212],
        [ 11.7794,   5.5086,  -9.6929],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3569,   0.7591,   6.5011],
        [  5.9655,   3.2172,  -7.5953],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.7275,  14.0667,  -0.9815],
        [  6.3606,   0.9204,   3.2525],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.7685, -10.8982,  -6.8928],
        [  1.6422,   7.4586,  -6.8056],
        [ -3.5173, -10.0623,   4.8280],
        [ -6.3306,   5.9272,   5.6935],
        [  6.9693,  -9.3200,  -8.3822],
        [ -3.5903,   7.4847,  -8.9112],
        [  0.5250, -11.0306,   5.5358],
        [ -4.0868,   7.2981,   8.1874],
        [  3.1271, -10.8343,   2.6442],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.9033,   2.2329,  -8.8912],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.3122,  -9.6994,  -6.3282],
        [ -1.6906,   7.8860,   7.7142],
        [ 11.4743,   6.2478,  -2.1100],
        [ -5.9987,   6.1949,  -6.2114],
        [ 11.5083,  -4.1177,   0.2389],
        [  3.8111, -10.6787,  -4.6539],
        [ -1.7009,   7.8851,  -8.2437],
        [  4.8674,  12.8875,   6.7622],
        [  2.7976,  -4.4682,  -4.2020],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.9284,  11.7061,  -0.3732],
        [ -8.3653,   3.4184,   9.3454],
        [  4.1537,   5.9340,  -6.0423],
        [  1.7122,  -4.7591,  -3.6734],
        [ -9.4430,  -0.1897,   4.8383],
        [  2.3490,  -4.6199,  -4.2489],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.0788,  -0.8710,  -6.8719],
        [ 11.7779,   5.5127,  -2.0699],
        [ -7.0120,   5.2825,   7.7762],
        [ -9.4753,  -0.8012,  -5.2050],
        [  7.9141,  -8.6656,   9.6948],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.4401,  13.4540,   4.6576],
        [  3.3943,   6.5483,   1.0790],
        [ -2.6485, -10.4151,   0.8766],
        [ 11.7779,   5.5127,  -2.0699],
        [  1.9312,  13.8649,   8.3563],
        [  5.1203,  -2.7151,   6.2110],
        [  6.3301,   0.3638,  -7.5685],
        [ -1.2454,   7.9127,  -0.5715],
        [  8.3718,  -8.2953,   8.4666],
        [  1.5246,  -4.7838,  -5.2095],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.2242,  -0.3171,   7.4978],
        [ -0.0254, -10.9932,   3.2200],
        [  3.9085,   6.1512,  -2.2560],
        [ -9.4117,   0.1160,   8.2574],
        [  8.7715,  -7.9390,   5.2900],
        [ -1.6086,   7.8928,   4.3385],
        [ -2.2188,   7.8218,   8.6064],
        [ 11.3118,   6.5957,  -4.5667],
        [ -7.0948,   5.1940,   9.8171],
        [ -8.8934,   2.2597,   3.5831],
        [  3.3167,   6.6022,   6.3554],
        [ -7.5401,   4.6718,   9.0033],
        [ -7.4733,   4.7555,   6.2490],
        [  1.8766,   7.3694,   1.1887],
        [  1.6954,  -4.7616,   9.8198],
        [  0.8940, -11.0402,  -0.9783],
        [ 11.1469,   6.9237,   1.6160],
        [  0.2480, -11.0153,  -3.0183],
        [  8.1528,  10.7452,   9.1493],
        [ 12.4690,  -0.8528,  -9.3374],
        [  4.5814, -10.4485,   2.5496],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.2644,  -4.6434,   9.7508],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.3516,   0.6425,   0.3425],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.7544,   2.6096,   3.3146],
        [  0.1206,  -4.7363,   8.7103],
        [ -4.5778,  -9.5013,  -5.0667],
        [  2.6613,   7.0021,   4.5783],
        [  3.4401,  13.4540,   4.6576],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.0719, -10.8449,  -4.1673],
        [  6.3569,   0.7591,   6.5011],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.1973,  -0.4372,   6.5720]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0314, 0.0244,  ..., 1.3222, 0.4545, 0.0984],
        [0.0314, 0.0000, 0.0222,  ..., 1.3396, 0.4553, 0.0794],
        [0.0244, 0.0222, 0.0000,  ..., 1.3175, 0.4375, 0.0740],
        ...,
        [1.3222, 1.3396, 1.3175,  ..., 0.0000, 0.9773, 1.3056],
        [0.4545, 0.4553, 0.4375,  ..., 0.9773, 0.0000, 0.3904],
        [0.0984, 0.0794, 0.0740,  ..., 1.3056, 0.3904, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.1805345 ,  5.64500512,  5.39098669,  5.70591986,  5.89074104,
        5.91222346, 10.87409005, 13.84126197,  5.57341623, 13.32645552,
       13.00380624, 12.3489324 ,  6.4020404 ,  6.77777606,  8.62631801,
       14.08549802,  6.42689705, 11.63724792, 11.24434374,  7.63726134,
       10.65929968,  8.67227386, 11.63765273,  8.30125201, 11.04313641,
        8.36445758, 11.27656488,  7.58668144, 10.00794468, 10.00794468,
        9.17905491,  9.97964013, 11.57249984,  8.06517143, 13.06498554,
        8.62329466, 12.22276278, 11.33837396,  8.06643487, 13.77604293,
        5.27178363,  9.55720651, 10.54658597,  7.96181288, 13.60275211,
        9.0368458 ,  7.24326992,  5.05774295,  9.44486609,  5.18277858,
        7.96181288,  6.14086582, 13.00413829,  8.77913428,  9.50913849,
       11.73567759, 10.53368644, 13.88683436,  7.37576154, 10.74655346,
       13.00413829, 13.99877413,  5.79562691,  6.34059021,  8.01009083,
       11.785566  ,  5.0209005 , 10.63038565,  6.2322769 , 10.99326577,
        7.28793663,  9.41244865, 11.83075005,  8.0550286 ,  8.1303945 ,
       13.09426142,  8.79284466,  9.17595899,  7.38844703,  8.87007495,
        8.8580671 ,  7.60463349,  5.05444288, 11.07637531, 13.12218013,
       11.018084  , 13.48809595, 12.4980845 , 11.4088043 , 13.00380624,
        5.8828986 ,  5.16611782, 10.4977132 ,  6.38399789, 13.13205357,
        9.13507321,  4.73784222, 10.54658597,  7.49076899, 13.88683436,
       13.09737053, 11.27160035,  6.4020404 ,  9.94285511, 12.0495521 ,
        6.21275016])}
{'P': tensor([[2.5481e-02, 1.7856e-01, 1.4162e-01,  ..., 6.1587e-05, 3.4473e-04,
         1.1213e-03],
        [3.0639e-03, 3.2040e-02, 2.0789e-01,  ..., 7.2582e-05, 3.5989e-04,
         1.1539e-03],
        [2.7625e-03, 3.5094e-03, 3.4425e-02,  ..., 7.6109e-05, 3.2561e-04,
         9.5083e-04],
        ...,
        [1.6476e-04, 1.7823e-04, 2.0089e-04,  ..., 5.5796e-02, 2.1060e-03,
         1.2258e-05],
        [9.0997e-04, 1.1715e-03, 1.5397e-03,  ..., 1.2274e-03, 7.4594e-02,
         4.0055e-04],
        [8.2444e-04, 1.1131e-03, 1.4340e-03,  ..., 5.6910e-06, 8.0469e-05,
         5.3024e-02]]), 'X': tensor([[ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0114, 0.0201,  ..., 0.8074, 0.7078, 0.3217],
        [0.0114, 0.0000, 0.0096,  ..., 0.8074, 0.7136, 0.3248],
        [0.0201, 0.0096, 0.0000,  ..., 0.8023, 0.7132, 0.3322],
        ...,
        [0.8074, 0.8074, 0.8023,  ..., 0.0000, 0.4093, 1.1220],
        [0.7078, 0.7136, 0.7132,  ..., 0.4093, 0.0000, 0.9634],
        [0.3217, 0.3248, 0.3322,  ..., 1.1220, 0.9634, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.18277858,  5.27178363,  5.35774573,  5.5513072 ,  5.5671389 ,
        5.19577146,  9.78147542,  8.25828409,  9.17595899, 11.81423161,
       11.59512194, 10.46739073, 13.59673406, 10.68031351,  7.16058587,
       10.6910768 ,  8.89458273,  4.92497514,  5.91222346,  8.92098752,
       13.00413829,  4.85995002,  5.5513072 , 13.43127131, 10.00410625,
       13.00380624, 14.09176259, 11.76830981, 13.72594022,  9.55720651,
        9.26455795,  5.14371005, 10.48315436,  5.1224079 ,  9.36256096,
       13.06957425,  5.14607014,  6.32572323, 12.53345445,  6.77777606,
        6.29034958, 12.13568598, 13.98691426, 10.52587243, 12.30858677,
        8.05242591, 10.5575601 ,  7.96181288,  7.73692436,  6.81084724,
       12.95702647,  9.17016857,  8.36995953, 13.01728313,  6.78325151,
       13.35832725, 10.91805683, 13.15525803, 13.76047999,  4.90676242,
       10.31584102,  5.16611782,  5.41348121,  9.48969978,  8.95133961,
       11.97957425,  5.80860165, 11.07637531,  9.43112957,  5.10446653,
        5.5671389 ,  5.16077764, 13.45642379,  9.99145096, 12.11735334,
       11.76830981, 13.54774355, 12.4980845 , 10.9036226 , 13.27994304,
        9.71392179, 13.97779839, 10.51340829,  5.26067019, 11.62770332,
        5.70591986,  6.78325151,  6.14086582,  9.36378438, 10.06703358,
        7.50564223, 12.25806559,  7.02496223, 11.07637531,  5.24137909,
        8.92395381, 12.26997165,  7.72668467, 12.59969936,  6.76661595,
        4.978704  , 11.11969815,  7.37902766,  5.05444288, 11.48734978,
       11.64436684])}
{'P': tensor([[3.5421e-02, 1.9381e-01, 1.7096e-01,  ..., 3.8927e-05, 6.6253e-06,
         1.0388e-04],
        [3.2030e-03, 4.4772e-02, 5.8907e-02,  ..., 5.0037e-05, 9.7344e-06,
         1.3793e-04],
        [3.5733e-03, 2.5891e-02, 4.9169e-02,  ..., 5.8154e-05, 1.2548e-05,
         1.8381e-04],
        ...,
        [5.9902e-04, 3.7371e-04, 3.8345e-04,  ..., 1.1401e-01, 7.9707e-05,
         3.5356e-05],
        [4.2162e-06, 3.4229e-06, 3.8587e-06,  ..., 6.6395e-04, 3.2241e-02,
         4.1029e-05],
        [1.0994e-04, 1.3000e-04, 1.5371e-04,  ..., 1.9270e-04, 8.1504e-03,
         1.7788e-01]]), 'X': tensor([[  2.3889,  -4.6082,  -8.7291],
        [  3.4140,  -4.1803,  -8.8224],
        [  3.4920,  -4.1366,  -8.3955],
        [  3.7992,  -3.9474,  -8.9019],
        [  4.0421,  -3.7765,  -8.3631],
        [  3.6154,  -4.0640,  -7.9698],
        [  0.5600, -11.0321,  -0.8660],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.8058,   2.4850,  -1.7425],
        [  0.3354,  -4.7708,  -4.9125],
        [  0.4996,   7.7738,  -4.4711],
        [  4.0642,  -3.7599,  -3.4565],
        [  2.9410,   6.8428,   5.4458],
        [  4.5800,   5.5034,  -3.9432],
        [ -3.4488, -10.0934,   2.5195],
        [ -8.3900,   3.3727,   3.1650],
        [  6.8704,  -9.3809,   7.5769],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.3200,   7.1756,   1.1938],
        [  4.3377,   5.7570,   1.3712],
        [  2.4932,   7.0903,  -5.4214],
        [  5.9843,  -9.8679,  -6.1449],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -8.0611,  -6.0302,   4.6962],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.3589,   0.8239,  -8.4230],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.3360,   0.4258,  -9.2509],
        [  3.2701,   6.6338,   9.9209],
        [ -7.6525,  -6.6534,   5.1721],
        [ 10.4737,  -5.9570,  -1.2108],
        [  3.7237,  -3.9966,  -5.0376],
        [  7.5784,  -8.9141,   5.8410],
        [  1.1350,  14.0104,   0.6321],
        [  2.3367,   7.1676,  -9.5263],
        [  0.8572,  14.0501,  -8.3364],
        [  5.6093,  -1.9705,   6.4807],
        [  0.6363, -11.0348,   6.1414],
        [  0.1677,  -4.7447,  -6.1529],
        [ -7.9667,   4.0814,  -0.6397],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.9193,  -1.3176,   5.1739],
        [  7.2340,  -9.1505,  -5.8321],
        [  1.1274,  14.0116,   8.1753],
        [  5.7963,  -1.6018,   1.8856],
        [ -6.7745,  -7.7264,   8.4684],
        [  6.2535,  -0.1709,  -9.4102],
        [  4.5954,  13.0099,  -4.4439],
        [  4.1999,  -3.6543,  -4.1998],
        [  2.2849, -10.9655,  -7.6542],
        [ -7.4807,   4.7463,   2.0616],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.3261,  13.7749,   5.9121],
        [  1.8320,   7.3871,   5.1452],
        [  8.8474,  10.0868,  -8.8644],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -6.0195,   6.1789,  -2.9496],
        [  5.2911,  -2.4824,  -3.6877],
        [ -5.9940,   6.1985,   5.9566],
        [  6.1741,  -0.5329,   3.1913],
        [  3.3744,   6.5623,  -6.3636],
        [  9.0051,   9.9234,   5.4411],
        [ -7.0948,   5.1940,   9.8171],
        [  4.5503,   5.5359,   7.4735],
        [ 11.4235,   6.3594,   5.1663],
        [  6.3236,   0.3011,  -4.6668],
        [  1.6954,  -4.7616,   9.8198],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.2517,  -4.6495,   2.9800],
        [ -5.1046,  -9.1610,   2.3418],
        [ -5.3382,   6.6533,   7.8054],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.5101,  -2.1433,   9.5442],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.1504,  -4.6726,   8.7908],
        [  6.1372,   2.6726,  -6.7885],
        [  6.3604,   0.9042,  -6.5901],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.3303,   1.6145,  -8.4083],
        [  5.0190,   4.9705,  -7.9752],
        [ -1.0482, -10.8487,  -5.9066],
        [ 10.4737,  -5.9570,  -1.2108],
        [  0.9788,  -4.8145,  -3.8661],
        [ -1.7009,   7.8851,  -8.2437],
        [ -5.0132,  -9.2234,  -8.3531],
        [  1.4489,   7.5256,   4.4529],
        [  4.4633,   5.6287,  -0.7754],
        [  1.1644, -11.0394,   7.8511],
        [  0.1252, -11.0062,   7.2564],
        [ -3.6381,   7.4683,   7.3544],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.4572,   7.1085,   7.7376],
        [ -8.8934,   2.2597,   3.5831],
        [ -3.3999, -10.1152,  -1.2088],
        [  8.5716,  10.3596,   6.8818],
        [ 11.6165,  -3.8715,   6.4594],
        [  5.3858,   4.4260,  -9.6769],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3246,   1.6696,  -0.1544],
        [  2.6321,  -4.5296,  -2.5159],
        [ -2.0840,   7.8415,  -7.8657],
        [  6.3211,   1.7017,  -0.2164],
        [ -7.9298,   4.1367,  -3.9773],
        [ -2.1382,   7.8339,   7.3524],
        [  6.2895,   0.0405,   5.7652]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.3774e-02, 2.2453e-02,  ..., 6.8670e-01, 9.7159e-01,
         8.2221e-01],
        [2.3774e-02, 2.1453e-04, 7.2084e-03,  ..., 7.1047e-01, 9.8233e-01,
         8.2474e-01],
        [2.2453e-02, 7.2084e-03, 0.0000e+00,  ..., 7.0819e-01, 9.7520e-01,
         8.1770e-01],
        ...,
        [6.8670e-01, 7.1047e-01, 7.0819e-01,  ..., 0.0000e+00, 9.1461e-01,
         1.0266e+00],
        [9.7159e-01, 9.8233e-01, 9.7520e-01,  ..., 9.1461e-01, 0.0000e+00,
         3.5782e-01],
        [8.2221e-01, 8.2474e-01, 8.1770e-01,  ..., 1.0266e+00, 3.5782e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.19065439,  5.39723241,  5.41348121,  5.47866221,  5.53175275,
        5.43943791, 11.04629471, 13.89974693,  9.88571713,  9.14973319,
        4.78258092,  7.78980299,  5.5366698 ,  7.44805901,  7.15990676,
       10.66631784,  9.04255655, 11.62770332, 10.31613269,  7.54127425,
        7.20826219,  7.5158489 , 11.54071859, 10.06437598, 10.06703358,
       13.15525803,  6.41202986, 12.91143491,  6.35029274,  7.39599986,
       10.14044489, 12.04924159,  5.46244988, 11.70016217, 14.05633441,
        7.53884104, 14.07623266,  5.94535529, 11.05317597,  4.74770899,
        8.95133961,  8.30125201,  6.06416957, 11.66453059, 14.05687815,
        6.01356004, 10.27572673,  6.25586297, 13.79762415,  5.5671389 ,
       11.20100795,  8.85939626,  9.55593374, 13.96987806,  7.61088125,
       13.41713661,  9.78147542,  8.62631801,  5.84450127,  8.62260327,
        6.19707944,  7.37902766, 13.40024224,  8.79284466,  7.16598273,
       13.0743399 ,  6.33075682,  5.05444288, 12.1460931 , 12.17451685,
       10.48721382,  8.53014915,  5.18277858,  5.91222346, 12.05395056,
        5.14371005,  6.69389431,  6.42439344, 12.00995898,  6.53290663,
        7.06372172, 10.8992524 , 12.04924159,  4.91296205,  8.06643487,
       10.4977132 ,  7.66377554,  7.18354885, 11.10066126, 11.00694688,
        8.30729352, 12.19805394,  7.52117132,  9.17595899, 10.67132524,
       13.44593519, 12.24467041,  6.97106912,  9.09963894,  6.54127431,
        5.23878371,  8.1137452 ,  6.54615887,  8.94395628,  8.12043983,
        6.28961988])}
{'P': tensor([[2.7759e-02, 1.8112e-01, 1.4826e-01,  ..., 8.6217e-05, 1.2500e-03,
         1.6801e-02],
        [3.1957e-03, 2.9836e-02, 1.9359e-01,  ..., 9.9050e-05, 1.4115e-03,
         1.1663e-02],
        [3.7403e-03, 4.4030e-03, 4.3190e-02,  ..., 1.5577e-04, 2.1323e-03,
         9.2326e-03],
        ...,
        [2.5016e-04, 2.2059e-04, 1.9262e-04,  ..., 1.2319e-01, 2.1568e-05,
         1.7481e-04],
        [1.5540e-03, 1.8350e-03, 2.2419e-03,  ..., 1.2445e-04, 1.6287e-01,
         2.8648e-03],
        [2.4242e-02, 4.2121e-02, 7.9615e-02,  ..., 1.6174e-04, 7.3765e-03,
         6.8556e-02]]), 'X': tensor([[ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0090, 0.0185,  ..., 0.3771, 0.5708, 0.0926],
        [0.0090, 0.0000, 0.0096,  ..., 0.3857, 0.5667, 0.0900],
        [0.0185, 0.0096, 0.0000,  ..., 0.3952, 0.5610, 0.0864],
        ...,
        [0.3771, 0.3857, 0.3952,  ..., 0.0000, 0.8776, 0.4388],
        [0.5708, 0.5667, 0.5610,  ..., 0.8776, 0.0000, 0.4792],
        [0.0926, 0.0900, 0.0864,  ..., 0.4388, 0.4792, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.19577146,  5.27178363,  5.35774573,  5.5671389 ,  5.5513072 ,
        5.46244988,  7.18354885, 10.68031351, 12.5085995 ,  4.91296205,
        6.32572323, 11.51811603,  7.57946459,  6.23262289,  7.8575648 ,
        8.79284466,  9.7751189 ,  5.08910491, 12.95702647,  8.36445758,
       12.04924159, 10.26327895, 10.88618114, 10.95593729, 12.05395056,
        5.35774573,  7.19513806, 13.50484539,  7.15990676, 12.41012046,
        5.96556248,  7.44805901,  5.79543038, 12.97370371,  7.48538905,
       11.63724792, 13.66991027,  9.18690018,  4.78002665,  6.83515201,
       13.08414689,  6.81084724, 13.44593519, 10.05558893,  8.19703512,
        6.90115901, 13.744387  ,  8.41148722,  5.20659892,  8.12043983,
        8.09899874,  8.09899874, 11.1341798 , 11.28895489,  4.78544875,
        5.0209005 ,  7.46248947, 11.04313641,  6.53290663,  8.31625171,
       13.01728313,  7.13501264, 11.4424816 ,  9.14044984,  9.49511524,
        7.72668467,  9.36256096,  6.54693987,  6.38105015, 13.1399181 ,
        9.48969978,  5.84533345,  5.19065439,  5.38342077,  9.9353921 ,
        4.79003361, 12.1460931 ,  9.3109884 , 13.54774355,  6.77777606,
       12.05395056,  5.21280156, 13.72594022, 13.39372426,  5.70591986,
        9.41970031,  8.62162372, 12.12448912, 13.36092624, 12.30858677,
        7.83486943, 10.00410625, 10.26545792,  8.66273171,  6.76056625,
       11.785566  ,  4.7663526 , 13.04051744,  7.23276582,  8.56889738,
       10.412299  , 12.70489563,  7.85919687,  8.16714836,  5.79543038,
        5.23878371])}
{'P': tensor([[3.5144e-02, 1.7161e-01, 1.3391e-01,  ..., 6.0420e-04, 5.0143e-05,
         3.9666e-04],
        [2.2830e-03, 3.6836e-02, 2.7290e-01,  ..., 1.2915e-03, 3.3825e-05,
         6.8655e-04],
        [4.3764e-03, 7.3731e-03, 9.0378e-02,  ..., 4.1195e-03, 7.1966e-05,
         2.1321e-03],
        ...,
        [2.3633e-06, 3.2616e-06, 3.8756e-06,  ..., 7.2204e-03, 2.1945e-07,
         6.0616e-04],
        [2.5124e-02, 1.2665e-02, 9.4647e-03,  ..., 2.2518e-05, 3.0997e-01,
         7.5078e-05],
        [9.0743e-05, 7.5681e-05, 7.2724e-05,  ..., 1.2467e-05, 1.5820e-04,
         3.9383e-02]]), 'X': tensor([[ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0254, 0.0356,  ..., 0.4639, 0.9169, 0.1869],
        [0.0254, 0.0000, 0.0102,  ..., 0.4460, 0.9423, 0.2054],
        [0.0356, 0.0102, 0.0000,  ..., 0.4397, 0.9524, 0.2128],
        ...,
        [0.4639, 0.4460, 0.4397,  ..., 0.0000, 1.3016, 0.6507],
        [0.9169, 0.9423, 0.9524,  ..., 1.3016, 0.0000, 0.7903],
        [0.1869, 0.2054, 0.2128,  ..., 0.6507, 0.7903, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.19986797,  5.46244988,  5.56809335,  5.5671389 ,  5.39450305,
        5.5513072 ,  5.08910491,  9.14044984, 10.66631784, 10.26327895,
        5.83560009,  7.16598273,  9.95552626,  8.24808832,  7.15990676,
        9.05843201, 11.83509372,  9.85330646,  6.06416957, 12.86648771,
       11.51557166,  8.83341608,  9.34127013,  5.43943791,  8.25828409,
        8.41171369,  7.96181288, 10.67322527, 12.35421461, 12.67680439,
        7.15100303, 10.05558893, 12.59840193,  9.19837792, 10.51340829,
       12.3828957 ,  6.42439344,  7.58378655,  8.68979052,  7.61088125,
        6.54693987, 13.36092624,  6.46356806, 11.83509372,  7.14837514,
        5.56809335,  6.34059021,  7.85919687, 11.64436684, 12.7125079 ,
       11.62770332, 11.37939695,  9.78286733, 13.744387  , 12.72524615,
        7.39004226,  7.58668144, 13.39372426,  8.09984231, 13.04156075,
       13.14777433,  5.94535529, 11.22142195,  9.43737523, 13.89264822,
        6.19707944,  9.77777346, 13.45642379,  9.97220799,  7.48311002,
        8.31625171,  6.80370254, 13.32374056, 13.93790235,  8.04483495,
        6.28835101, 11.31276109,  6.23262289,  5.5205189 ,  9.09119351,
        8.31625171, 11.11038551,  9.61942445, 13.103709  ,  5.63824971,
       12.5085995 ,  6.60752373,  9.36378438,  8.5724042 , 12.72524615,
        8.04664886, 13.94008948,  9.05843201,  8.34949383,  5.05534609,
        9.50913849, 11.73848065,  6.58184474,  4.78002665,  6.83515201,
        8.06990483,  9.64157461,  7.37902766, 13.01728313, 10.06407839,
        7.70897675])}
{'P': tensor([[3.0389e-02, 1.5031e-01, 1.4656e-01,  ..., 1.5323e-04, 6.8171e-05,
         4.0755e-04],
        [3.8599e-03, 4.9609e-02, 7.2290e-03,  ..., 2.9757e-04, 7.0220e-05,
         4.3189e-04],
        [5.1246e-03, 1.2412e-01, 3.8833e-02,  ..., 2.4612e-04, 7.0621e-05,
         5.2253e-04],
        ...,
        [5.2412e-05, 4.9274e-05, 5.6712e-05,  ..., 7.1724e-02, 9.6164e-06,
         1.3223e-05],
        [2.4875e-02, 1.6364e-02, 2.0535e-02,  ..., 2.1835e-05, 1.4339e-01,
         1.6471e-03],
        [3.1345e-04, 4.8169e-04, 4.3273e-04,  ..., 1.6096e-03, 2.7077e-06,
         7.4955e-02]]), 'X': tensor([[  2.4697,  -4.5836,   1.4593],
        [  3.4604,  -4.1545,   1.8091],
        [  2.8890,  -4.4316,   1.1600],
        [  4.4290,  -3.4596,   1.4181],
        [  4.3459,  -3.5327,   0.3667],
        [  5.1667,  -2.6543,   1.2879],
        [  0.2435,   7.8191,  -6.4709],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4756,  -0.8166,  -9.6021],
        [  5.1196,  -2.7160,   4.2705],
        [  4.3377,   5.7570,   1.3712],
        [  4.7739,  -3.1218,  -4.1188],
        [ -8.6426,  -4.9136,   9.4941],
        [  9.4554,   9.4245,   9.8800],
        [ -6.3066,  -8.1936,   9.5822],
        [  0.9809,   7.6642,  -1.4217],
        [  6.1372,   2.6726,  -6.7885],
        [  2.1440, -10.9811,   8.5281],
        [ 11.7162,   5.6728,  -8.9372],
        [  0.4996,   7.7738,  -4.4711],
        [ 11.1469,   6.9237,   1.6160],
        [ -9.2687,  -2.9794,   0.6436],
        [  4.4049,   5.6892,  -4.4394],
        [ -1.6906,   7.8860,   7.7142],
        [  4.7813,  -3.1139,   8.2691],
        [ -6.2668,   5.9808,   2.5614],
        [  6.0788,  -0.8710,  -6.8719],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.2373,  -4.6506,  -1.7334],
        [ -9.4721,  -1.3097,  -7.4377],
        [  6.2036,   2.4030,  -1.2500],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.9809,   7.6642,  -1.4217],
        [  3.6154,  -4.0640,  -7.9698],
        [  2.5813,  -4.5471,   5.3597],
        [ 11.3691,   6.4760,   1.0787],
        [  6.3550,   0.7129,   0.4917],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.6675,  -4.8569,  -9.1769],
        [  4.7739,  -3.1218,  -4.1188],
        [  9.6587,   9.1817,  -0.8212],
        [ 10.9916,   7.2133,  -0.9179],
        [  1.4489,   7.5256,   4.4529],
        [  6.2036,   2.4030,  -1.2500],
        [  2.0243,   7.3086,   7.5432],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 12.4837,   2.7510,   0.4449],
        [  1.7301,   7.4262,   5.0522],
        [  3.3839,  -4.1966,   8.3574],
        [  2.0870,  -4.6877,   9.6823],
        [ -6.2668,   5.9808,   2.5614],
        [ -8.9636,   2.0648,   8.1946],
        [  1.6408,   7.4591,  -4.2294],
        [  0.6952,  -4.8060,   8.8976],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 11.7796,  -3.4696,   9.0088],
        [ 11.2517,  -4.6495,   2.9800],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.4867,  12.5816,   2.9549],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -4.6942,  -9.4299,  -9.1846],
        [  9.3424,  -7.3685,  -3.6807],
        [ 12.2914,   3.8037,   6.0978],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.1292,   4.8183,  -5.7997],
        [  2.8890,  -4.4316,   1.1600],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.1066,  -0.7799,  -4.1746],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.8719,   7.3713,  -3.7465],
        [  4.8653,   5.1694,  -4.0823],
        [  1.3670,  -4.7990,   3.2603],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.6970,   6.9826,   9.7350],
        [  6.3236,   0.3011,  -4.6668],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.9659,  -4.8144,  -0.9594],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.5057,  -3.3894,  -0.8160],
        [  4.1537,   5.9340,  -6.0423],
        [  6.2022,   2.4092,   3.0349],
        [  3.9895,   6.0816,  -6.5235],
        [  2.8994,   6.8675,  -9.5987],
        [  2.7976,  -4.4682,  -4.2020],
        [ -1.6334, -10.7207,   7.1397],
        [ -9.4117,   0.1160,   8.2574],
        [  3.9515,  -3.8425,   6.1729],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.3087,   1.8053,  -1.9481],
        [  2.4356,  -4.5942,  -5.0808],
        [ -8.9178,   2.1936,   7.5388],
        [  6.0054,  -1.0902,   8.3079],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.3384,  -2.6161,  -4.7630],
        [  0.6757, -11.0361,  -1.3047],
        [  4.5814, -10.4485,   2.5496],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -8.0766,  -6.0046,  -3.1873],
        [  4.1140,  -3.7220,  -0.1273],
        [  4.3554,   5.7393,  -9.0766],
        [ -5.9225,  -8.5355,   2.4778],
        [  8.1043,  -8.5164,  -7.9290]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.9413e-02, 1.9586e-02,  ..., 5.2596e-01, 8.5131e-01,
         6.2647e-01],
        [3.9413e-02, 1.0727e-04, 5.1236e-02,  ..., 5.5860e-01, 8.7162e-01,
         6.6218e-01],
        [1.9586e-02, 5.1236e-02, 0.0000e+00,  ..., 5.0880e-01, 8.6231e-01,
         6.1110e-01],
        ...,
        [5.2596e-01, 5.5860e-01, 5.0880e-01,  ..., 0.0000e+00, 1.0418e+00,
         1.3828e-01],
        [8.5131e-01, 8.7162e-01, 8.6231e-01,  ..., 1.0418e+00, 0.0000e+00,
         1.0202e+00],
        [6.2647e-01, 6.6218e-01, 6.1110e-01,  ..., 1.3828e-01, 1.0202e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.20659892,  5.40689332,  5.29009836,  5.62005629,  5.60064319,
        5.80860165,  7.822854  ,  8.09704052,  9.51074763,  5.79543038,
        7.20826219,  5.70405525,  9.94173708, 13.35013374, 10.33958639,
        7.72668467,  6.69389431, 11.18839169, 13.01728313,  7.78980299,
       13.12218013,  9.73579738,  7.19513806,  8.06517143,  5.70591986,
        8.66273171,  6.14086582, 12.07818871,  5.16077764,  9.562174  ,
        6.652742  , 10.24557047,  7.72668467,  5.43943791,  5.2287058 ,
       13.08414689,  6.3949043 ,  9.3349858 ,  9.93553424,  5.70405525,
       13.32645552, 13.1471313 ,  7.66377554,  6.652742  ,  7.58378655,
       12.12448912, 12.78327037,  7.62509582,  5.39098669,  5.13125513,
        8.66273171,  9.19837792,  7.6374603 ,  4.85603886,  9.78286733,
       12.35625694, 12.27993098, 12.17451685,  6.10774544, 13.72594022,
       12.22503754,  6.80370254,  9.62958973, 10.53368644, 11.89855267,
       12.86648771, 12.00995898,  7.03733775,  5.29009836,  9.50913849,
        6.1561646 ,  9.66977802,  7.60529892,  7.09887557,  4.98989906,
        5.41348121,  7.48538905,  6.33075682, 10.24557047,  4.91038884,
       13.14777433,  5.63824971,  7.24326992,  6.65368551,  7.27340834,
        7.4544866 ,  5.27178363, 10.84437705,  9.41244865,  5.51177501,
        9.96391045,  6.5618968 ,  5.19986797,  9.18358562,  6.10360073,
       11.99633819,  9.34127013,  9.69792513, 11.05672136, 11.4088043 ,
       10.14436912, 10.06407839,  5.54776934,  7.20482224, 10.38897563,
       11.75618513])}
{'P': tensor([[0.0350, 0.1382, 0.1372,  ..., 0.0027, 0.0026, 0.0017],
        [0.0028, 0.0506, 0.0032,  ..., 0.0016, 0.0017, 0.0041],
        [0.0048, 0.1729, 0.0319,  ..., 0.0019, 0.0018, 0.0021],
        ...,
        [0.1230, 0.0747, 0.0935,  ..., 0.0675, 0.0345, 0.0012],
        [0.1158, 0.0715, 0.1197,  ..., 0.0111, 0.0361, 0.0015],
        [0.0005, 0.0008, 0.0006,  ..., 0.0003, 0.0004, 0.1271]]), 'X': tensor([[  2.5010,  -4.5736,  -0.8989],
        [  4.0457,  -3.7737,  -0.4704],
        [  2.9461,  -4.4077,  -0.4831],
        [  4.2275,  -3.6320,  -0.2714],
        [  4.5057,  -3.3894,  -0.8160],
        [  4.1140,  -3.7220,  -0.1273],
        [ 11.3367,  -4.4806,   0.4643],
        [  2.1824,  -4.6647,   1.1985],
        [  2.2644,  -4.6434,   9.7508],
        [  2.6310, -10.9195,   3.8054],
        [  6.2036,   2.4030,  -1.2500],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.1196,  -2.7160,   4.2705],
        [ -1.8788, -10.6567,   3.7068],
        [  6.6319,  11.9066,   3.5165],
        [  3.8111, -10.6787,  -4.6539],
        [ -5.1377,   6.7751,  -0.4438],
        [  3.3943,   6.5483,   1.0790],
        [  6.3604,   0.9042,  -6.5901],
        [  2.9260,  -4.4162,   8.3808],
        [  7.8982,  -8.6778,   0.0239],
        [ -4.4520,   7.1370,  -8.7867],
        [ -2.7572,   7.7194,   5.9580],
        [ -7.2986,  -7.1225,  -3.0814],
        [  0.9036,   7.6840,   9.9237],
        [ 11.5966,   5.9666,   6.0083],
        [  6.2247,  -0.3150,  -9.3657],
        [ -6.4368,   5.8356,   3.3014],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.4296,  -1.9458,   4.0833],
        [  9.3424,  -7.3685,  -3.6807],
        [  4.7813,  -3.1139,   8.2691],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.7218,  -9.9934,   6.5764],
        [ -1.9487,   7.8590,  -3.3330],
        [  4.9152,  12.8653,  -1.4459],
        [ -0.5385,   7.9037,   6.3830],
        [ 12.2921,  -1.7902,  -4.0870],
        [  7.7794,  11.0629,   8.9069],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.6059,   0.9597,  -6.3264],
        [  3.3167,   6.6022,   6.3554],
        [  1.7903,  -4.7467,   0.1168],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.7590,  13.6621,   9.4496],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -7.3333,   4.9246,  -7.9212],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.6165,  -3.8715,   6.4594],
        [ -9.3076,   0.7852,  -6.7010],
        [  7.2340,  -9.1505,  -5.8321],
        [  9.1409,  -7.5789,   1.5814],
        [  5.4732,  -2.2041,  -4.3830],
        [  3.9022,  13.2906,   5.5788],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.0348,  -2.8229,   3.5436],
        [  7.0515,  11.6194,   8.4289],
        [  2.2437,  -4.6489,   6.1430],
        [  5.9193,  -1.3176,   5.1739],
        [  9.4925,   9.3811,   8.9842],
        [  6.3534,   1.2990,  -7.2942],
        [ 10.2470,   8.4054,  -0.1177],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.5382,   7.0672,   4.5097],
        [ -7.3333,   4.9246,  -7.9212],
        [ -9.4196,   0.0478,   9.4140],
        [  7.8425,  -8.7202,   6.4530],
        [  5.3118,   4.5452,   4.2114],
        [  0.7275,  14.0667,  -0.9815],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.6414, -10.0303,   7.8525],
        [ 12.1504,   4.3674,  -7.2998],
        [  7.8375,  -8.7240,  -9.8913],
        [  1.1994,   7.6035,   2.4295],
        [  1.1350,  14.0104,   0.6321],
        [ -4.2040,   7.2486,   2.9875],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -3.4100,   7.5436,  -7.6980],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.4877,  -0.7222,   8.8347],
        [ -5.6169,  -8.7844,   8.9057],
        [  5.3924, -10.1396,  -4.0863],
        [ -2.6485, -10.4151,   0.8766],
        [  5.3118,   4.5452,   4.2114],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.4179,  -2.2925,  -3.4641],
        [  3.6703,   6.3443,  -5.4401],
        [  1.8719,   7.3713,  -3.7465],
        [  5.3118,   4.5452,   4.2114],
        [  5.5158,  -2.1337,   3.5246],
        [ -5.1046,  -9.1610,   2.3418],
        [ -0.7599, -10.8995,  -6.0366],
        [  9.3607,   9.5337,  -4.5412],
        [  4.9957,  -2.8705,   6.5978],
        [  5.9843,  -9.8679,  -6.1449],
        [ -7.0825,   5.2073,  -5.0497],
        [  9.5018,  -7.1945,  -1.1892],
        [ -3.3116, -10.1539,   7.5919],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.9945,  -4.8144,  -0.1897],
        [  6.3549,   0.7096,   6.9198]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0445, 0.0277,  ..., 0.1019, 0.0521, 0.5533],
        [0.0445, 0.0000, 0.0223,  ..., 0.1437, 0.0688, 0.5180],
        [0.0277, 0.0223, 0.0000,  ..., 0.1296, 0.0481, 0.5262],
        ...,
        [0.1019, 0.1437, 0.1296,  ..., 0.0000, 0.1317, 0.6532],
        [0.0521, 0.0688, 0.0481,  ..., 0.1317, 0.0000, 0.5245],
        [0.5533, 0.5180, 0.5262,  ..., 0.6532, 0.5245, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.21280156,  5.53255882,  5.30158497,  5.57341623,  5.63824971,
        5.54776934, 12.18998195,  5.14999799,  5.16611782, 11.23201619,
        6.652742  , 12.32408516,  5.79543038, 10.82106505, 13.62896593,
       11.33837396,  8.50278896,  7.37576154,  6.42439344,  5.29755487,
       11.73397309,  8.41171369,  8.19703512, 10.1979669 ,  7.73692436,
       13.04156075,  6.23262289,  8.68832398,  5.53255882,  9.6282763 ,
       11.89855267,  5.70591986, 12.4217523 ,  8.30125201, 11.51557166,
        8.09704052, 13.77222752,  7.92200342, 12.4217523 , 13.52430176,
        8.50278896, 12.64235359,  7.38844703,  5.07306274,  8.35856487,
       13.93790235, 12.12448912,  8.83341608, 13.79762415, 12.24467041,
        9.34061938, 11.66453059, 11.87412439,  5.90035324, 13.85158417,
        9.75075273,  9.11473391,  5.77219077, 13.59170809,  5.16205087,
        6.06416957, 13.34587027,  6.48486171, 13.25336001, 11.51811603,
        5.94535529,  9.69792513,  8.56889738,  7.50917551,  8.83341608,
        9.41970031, 11.72802491,  6.99097013, 14.08549802,  9.53015376,
       11.50791859, 12.91143491, 11.72749923,  7.69752262, 14.05633441,
        8.37954853, 10.23946247,  8.27853076, 13.77222752, 12.5085995 ,
       10.42667093, 11.48434767, 10.74655346,  6.99097013,  7.9337647 ,
        5.8828986 ,  7.32950385,  7.60529892,  6.99097013,  5.91408504,
       10.48721382, 10.92597196, 13.36092624,  5.76165934, 11.54071859,
        8.7907923 , 11.9182956 , 10.68031351,  4.85995002,  4.9160962 ,
        6.39438044])}
{'P': tensor([[2.8759e-02, 1.7611e-01, 1.7152e-01,  ..., 1.8679e-02, 1.3521e-05,
         4.9558e-05],
        [4.2971e-03, 5.3451e-02, 8.3984e-03,  ..., 5.3106e-02, 1.9963e-05,
         1.0056e-04],
        [3.1733e-03, 1.9698e-01, 3.5263e-02,  ..., 3.2852e-02, 1.3918e-05,
         6.5805e-05],
        ...,
        [2.6781e-03, 3.2228e-03, 3.1293e-03,  ..., 2.4748e-01, 4.2423e-05,
         1.6809e-03],
        [6.9039e-04, 5.5968e-04, 6.0112e-04,  ..., 7.5970e-05, 7.3595e-02,
         1.4647e-05],
        [2.0380e-04, 1.4882e-04, 1.5698e-04,  ..., 3.8414e-05, 1.5363e-03,
         4.5309e-02]]), 'X': tensor([[  2.5813,  -4.5471,   5.3597],
        [  3.3474,  -4.2162,   5.2455],
        [  3.2366,  -4.2732,   5.1532],
        [  3.9515,  -3.8425,   6.1729],
        [  4.2339,  -3.6268,   6.3125],
        [  5.1203,  -2.7151,   6.2110],
        [ -7.9667,   4.0814,  -0.6397],
        [  7.0664,  -9.2590,   8.0879],
        [  5.9026,   3.3837,  -0.6573],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.8670,  -5.3428,  -2.8637],
        [  7.0664,  -9.2590,   8.0879],
        [  0.5250, -11.0306,   5.5358],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.5382,   7.0672,   4.5097],
        [  4.0587, -10.6112,   6.2273],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.5318,  13.7231,   6.3692],
        [ 11.0373,   7.1299,  -8.5895],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.6650,   4.5092,   0.2520],
        [ 12.2692,  -1.8882,   3.9253],
        [  5.8836,   3.4315,  -4.0327],
        [  6.2576,   2.1359,  -3.3398],
        [ -0.8569,   7.9156,  -6.5557],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.7845,   3.6652,   0.9175],
        [  0.6470, -11.0352,  -1.1931],
        [ -8.8154,   2.4611,   2.5198],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.2365,  -0.2578,  -6.0212],
        [  2.0729,   7.2878,   7.0880],
        [  8.8474,  10.0868,  -8.8644],
        [  7.0171,  11.6439,  -5.0178],
        [ 11.5083,  -4.1177,   0.2389],
        [  1.4749,  -4.7892,  -0.8089],
        [ -2.7572,   7.7194,   5.9580],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.6523,   3.9443,  -2.9449],
        [  0.1532,  -4.7422,  -7.0145],
        [  4.5340,  -3.3629,   8.9652],
        [ -9.4735,  -1.2620,  -3.6742],
        [  1.7815,  -4.7481,   4.4289],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -2.2188,   7.8218,   8.6064],
        [  5.1519, -10.2387,   9.8191],
        [ -5.1369,   6.7755,   8.3836],
        [  0.7685,  -4.8098,  -2.9310],
        [  4.7739,  -3.1218,  -4.1188],
        [  1.6408,   7.4591,  -4.2294],
        [ 11.5083,  -4.1177,   0.2389],
        [  9.2811,  -7.4336,   5.1771],
        [  5.4426, -10.1181,  -5.1268],
        [ -9.4157,   0.0821,  -5.0025],
        [  0.2435,   7.8191,  -6.4709],
        [  3.1480,  -4.3165,  -7.8002],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.0961,   1.6529,   3.1553],
        [ 11.7796,  -3.4696,   9.0088],
        [ -2.0840,   7.8415,  -7.8657],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.3653,   3.4184,   9.3454],
        [  0.8700,  -4.8132,  -7.8042],
        [  6.0890,  -0.8382,   2.9166],
        [  5.3858,   4.4260,  -9.6769],
        [  6.3515,   0.6418,  -6.5823],
        [  0.1382, -11.0073,   1.0026],
        [  4.2041,  -3.6509,  -4.9488],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.1531,  -5.8748,   2.6934],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -1.3178, -10.7939,   0.4236],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.9282,   2.1648,   8.4894],
        [  0.6933,   7.7336,   2.1179],
        [  5.6928,  12.4710,   3.6001],
        [  3.3474,  -4.2162,   5.2455],
        [ -1.8788, -10.6567,   3.7068],
        [  9.8082,  -6.8396,   4.9615],
        [ -3.3999, -10.1152,  -1.2088],
        [  5.0348,  -2.8229,   3.5436],
        [ -2.8162,   7.7059,  -9.6885],
        [  5.7151,  -1.7700,   3.1542],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.3554,   4.8985,  -0.6665],
        [  1.9241,   7.3503,   7.9004],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.4356,  -4.5942,  -5.0808],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -3.5029,   7.5138,   8.1853],
        [ -1.1885, -10.8211,   0.0865],
        [  6.0890,  -0.8382,   2.9166],
        [ -7.0948,   5.1940,   9.8171],
        [  6.9284,  11.7061,  -0.3732],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.3413,   0.4883,  -6.8416],
        [  5.0233,   4.9647,  -3.1788],
        [ -3.2053, -10.1993,  -3.8736],
        [  6.2883,   0.0325,   5.2692],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -6.6895,   5.6051,   2.7899]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.6686e-02, 1.6154e-02,  ..., 6.5423e-02, 1.2087e+00,
         5.5570e-01],
        [1.6686e-02, 0.0000e+00, 7.6174e-03,  ..., 4.9542e-02, 1.2186e+00,
         5.7000e-01],
        [1.6154e-02, 7.6174e-03, 1.5170e-04,  ..., 5.4551e-02, 1.2114e+00,
         5.6437e-01],
        ...,
        [6.5423e-02, 4.9542e-02, 5.4551e-02,  ..., 0.0000e+00, 1.2607e+00,
         6.1885e-01],
        [1.2087e+00, 1.2186e+00, 1.2114e+00,  ..., 1.2607e+00, 0.0000e+00,
         7.2579e-01],
        [5.5570e-01, 5.7000e-01, 5.6437e-01,  ..., 6.1885e-01, 7.2579e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.2287058 ,  5.38342077,  5.36061974,  5.51177501,  5.5748624 ,
        5.79562691,  8.95133961, 11.64746882,  6.80370254,  6.00663314,
       12.10942412, 11.64746882, 11.04313641,  9.14973319,  7.50917551,
       11.36089355,  5.1224079 , 13.95472331, 13.1399181 ,  4.91652556,
       10.0674887 ,  8.89301516, 12.41366754,  6.81117802,  6.61211224,
        7.96181288,  9.84314712,  6.84796202, 11.05413545,  9.15253344,
       12.16901615,  6.24187044,  7.57685908, 13.41713661, 13.59480906,
       12.22276278,  5.01114719,  8.19703512,  9.48337063,  6.89245352,
        4.74468406,  5.64500512,  9.55720651,  5.0713364 ,  7.37902766,
        9.66977802,  8.1303945 , 11.46176259,  8.50268322,  4.87083348,
        5.70405525,  7.6374603 , 12.22276278, 11.89105259, 11.4890861 ,
        9.41605684,  7.822854  ,  5.34250225,  9.96152788,  9.24501829,
       12.27993098,  8.1137452 ,  4.91652556,  9.0368458 ,  4.89121805,
        6.1463896 ,  6.97106912,  6.383897  , 11.00812996,  5.56809335,
       12.44177641, 10.04916678,  9.61942445, 12.4217523 , 10.06437598,
       10.06407839, 10.87409005, 10.8992524 ,  9.18690018,  7.76457095,
       13.70893492,  5.38342077, 10.82106505, 11.95743558, 10.67132524,
        5.77219077,  8.20436764,  5.98285319, 10.0674887 ,  8.83725081,
        7.59795445, 13.00413829,  5.19986797, 12.1460931 ,  8.29022197,
       10.88618114,  6.1463896 ,  8.79284466, 13.60275211, 13.1471313 ,
        6.36004393,  7.06272256, 10.6910768 ,  6.28835101, 10.09979747,
        8.7273569 ])}
{'P': tensor([[3.2291e-02, 1.8971e-01, 1.5435e-01,  ..., 2.7818e-05, 4.2929e-05,
         8.3940e-04],
        [3.5790e-03, 4.8658e-02, 2.5622e-02,  ..., 4.6302e-05, 6.6616e-05,
         8.9114e-04],
        [5.4323e-03, 7.9315e-02, 6.0283e-02,  ..., 4.7344e-05, 7.9699e-05,
         9.9286e-04],
        ...,
        [3.6599e-05, 4.2406e-05, 3.5844e-05,  ..., 1.8156e-01, 5.7659e-05,
         3.0929e-04],
        [5.1451e-04, 3.3503e-04, 3.3517e-04,  ..., 6.0839e-06, 7.7261e-02,
         7.7444e-04],
        [8.3422e-03, 9.4969e-03, 7.6895e-03,  ..., 1.8599e-03, 1.5005e-04,
         1.3203e-01]]), 'X': tensor([[ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.0062e-02, 2.2741e-02,  ..., 9.3700e-01, 6.9867e-01,
         3.3280e-01],
        [2.0062e-02, 0.0000e+00, 7.2084e-03,  ..., 9.3691e-01, 7.1871e-01,
         3.3907e-01],
        [2.2741e-02, 7.2084e-03, 2.1453e-04,  ..., 9.4395e-01, 7.1994e-01,
         3.4622e-01],
        ...,
        [9.3700e-01, 9.3691e-01, 9.4395e-01,  ..., 0.0000e+00, 1.2063e+00,
         6.2747e-01],
        [6.9867e-01, 7.1871e-01, 7.1994e-01,  ..., 1.2063e+00, 0.0000e+00,
         6.9999e-01],
        [3.3280e-01, 3.3907e-01, 3.4622e-01,  ..., 6.2747e-01, 6.9999e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.22903117,  5.41348121,  5.39723241,  5.43943791,  5.53175275,
        5.47702766,  6.10774544,  9.0368458 ,  8.86082124,  9.97822473,
        4.79003361,  6.35029274,  5.10444097, 13.76047999,  9.09119351,
        6.67121748,  4.83371341,  6.19707944,  8.36995953,  7.01475376,
        6.06416957, 11.27160035, 11.21703654, 11.8159254 , 13.97857591,
        9.64157461,  9.20340031, 11.22142195,  6.19633139,  6.61211224,
       12.27993098,  6.05165248,  8.31625171,  6.46560181,  7.72668467,
        6.54693987, 11.99633819,  7.37902766, 12.12641969, 11.37939695,
        9.562174  , 10.32715715, 10.80728788, 12.13568598, 13.32645552,
       10.0674887 , 12.86648771, 12.05395056, 13.12299145,  4.79003361,
       13.62896593, 11.4890861 ,  5.91222346,  6.31576264,  9.60080639,
        5.95111804,  6.02588517,  5.0209005 , 11.24434374,  5.10446653,
        8.16584537,  8.1303945 ,  6.53290663, 13.94008948, 10.57302546,
        8.7907923 , 10.06437598,  5.39450305, 11.54071859, 11.54071859,
       10.88618114,  7.50500064,  8.81930525,  7.52117132,  9.78147542,
        9.41244865, 11.89855267, 10.08001894, 13.1399181 ,  4.96231532,
        6.30647746,  6.383897  ,  6.06416957,  9.97822473,  9.10945678,
       10.68416401, 10.06023665, 13.95472331,  8.36445758,  5.19577146,
       12.26997165, 13.66896739,  4.98989906,  7.19534146,  9.24501829,
        9.97964013, 13.96746816,  4.7663526 , 13.15525803,  5.5513072 ,
       12.10222204,  9.23071698,  6.61211224,  6.46560181,  8.97052115,
        4.91038884])}
{'P': tensor([[4.5839e-02, 1.6794e-01, 1.3999e-01,  ..., 9.4071e-05, 4.3055e-03,
         3.4109e-02],
        [4.9056e-03, 6.4005e-02, 4.7312e-02,  ..., 2.3767e-04, 4.6494e-03,
         7.5869e-03],
        [5.6961e-03, 2.7630e-02, 8.4487e-02,  ..., 1.9211e-04, 9.2252e-03,
         5.0158e-03],
        ...,
        [3.2191e-05, 3.6240e-05, 2.3493e-05,  ..., 6.8438e-02, 1.1550e-06,
         2.3089e-05],
        [3.0100e-04, 6.2841e-04, 9.9350e-04,  ..., 4.3255e-05, 7.9896e-02,
         3.6381e-04],
        [1.8555e-02, 4.7567e-02, 1.7642e-01,  ..., 5.3951e-05, 4.8642e-03,
         3.8011e-02]]), 'X': tensor([[  2.6321,  -4.5296,  -2.5159],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.5965,  -4.0754,  -3.5035],
        [  4.7544,  -3.1425,  -1.5315],
        [  4.1297,  -3.7098,  -3.9838],
        [ -0.1626, -10.9796,   6.4629],
        [ -7.9925,  -6.1423,   2.8990],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.6314,  -1.9302,  -9.6543],
        [ -8.5859,  -5.0393,   0.5476],
        [ -7.7842,   4.3466,  -1.7548],
        [ -0.5739,   7.9056,   0.6382],
        [  2.0039,   7.3173,  -1.5885],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 12.2435,   4.0076,   8.1257],
        [  1.6422,   7.4586,  -6.8056],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -7.4887,   4.7364,   8.9074],
        [  1.6954,  -4.7616,   9.8198],
        [  8.5093,  -8.1764,  -3.2381],
        [  0.3231,  -4.7691,  -1.6120],
        [ -6.4464,   5.8272,   7.5038],
        [  6.3413,   0.4883,  -6.8416],
        [  9.4554,   9.4245,   9.8800],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.3705,   0.4208,  -3.5815],
        [  0.6933,   7.7336,   2.1179],
        [  8.0877,  -8.5297,  -1.1050],
        [ -5.9987,   6.1949,  -6.2114],
        [ 12.5996,   1.3971,  -1.0694],
        [ -7.3554,   4.8985,  -0.6665],
        [ -1.0012, -10.8576,   2.2867],
        [ -1.5261,   7.8988,   9.3768],
        [  3.0719, -10.8449,  -4.1673],
        [  3.9022,  13.2906,   5.5788],
        [  2.1477,   7.2550,   8.9688],
        [  3.0719, -10.8449,  -4.1673],
        [  3.6469,   6.3624,  -1.8414],
        [  2.2076,  13.8032,  -1.5099],
        [  2.1120,  -4.6818,   8.6144],
        [  9.6662,   9.1725,   8.1056],
        [  2.4149,  -4.6005,  -4.3339],
        [ -3.3116, -10.1539,   7.5919],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.4735,  -0.7274,  -2.6749],
        [  7.8425,  -8.7202,   6.4530],
        [ 12.3357,   3.6001,   0.7903],
        [  5.0348,  -2.8229,   3.5436],
        [  6.0073,   3.0986,   6.7937],
        [  2.0449,  13.8402,  -7.1642],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.9284,  11.7061,  -0.3732],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.3357,   3.6001,   0.7903],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.0054,  -1.0902,   8.3079],
        [  6.0467,  -9.8369,  -6.5816],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.1497,   7.8334,  -5.4412],
        [  1.7122,  -4.7591,  -3.6734],
        [ -3.9456,  -9.8542,   0.7466],
        [  0.6470, -11.0352,  -1.1931],
        [  5.8325,  -1.5222,  -2.1341],
        [  2.5128, -10.9365,   9.9645],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -1.5408,   7.8978,   5.3470],
        [  6.3534,   1.2990,  -7.2942],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -1.5261,   7.8988,   9.3768],
        [ -3.7964,  -9.9294,   7.7147],
        [ 11.2517,  -4.6495,   2.9800],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.4297,  -5.3637,  -9.3962],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.5789,   0.1722,   1.6470],
        [ 11.7795,  -3.4697,   3.5624],
        [ -8.1531,  -5.8748,   2.6934],
        [  7.3592,  11.3937,   6.2407],
        [ -9.4309,  -0.0599,  -4.0830],
        [  5.9526,   3.2525,  -4.3821],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.9282,   2.1648,   8.4894],
        [ 10.5955,   7.8824,  -3.7006],
        [  9.2692,   9.6370,   7.0746],
        [  2.2849, -10.9655,  -7.6542],
        [ 12.5154,   2.5072,  -9.3274],
        [  0.8940, -11.0402,  -0.9783],
        [ -7.4887,   4.7364,   8.9074],
        [ -9.4430,  -0.1897,   4.8383],
        [  0.3038,   7.8092,   0.4164],
        [  3.9895,   6.0816,  -6.5235],
        [  3.9663,   6.1018,  -5.4104],
        [ -7.6719,  -6.6260,   7.1468],
        [  5.3118,   4.5452,   4.2114],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.7121,   6.9743,  -8.4202],
        [ -8.4824,  -5.2575,  -0.0795],
        [  2.0729,   7.2878,   7.0880],
        [  2.1477,   7.2550,   8.9688],
        [ 10.9717,  -5.1646,  -6.9039],
        [  2.7420,  -4.4896,  -3.7139]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.1720e-02, 6.8033e-02,  ..., 7.3517e-01, 3.5875e-01,
         6.7537e-02],
        [5.1720e-02, 0.0000e+00, 1.0618e-01,  ..., 6.9379e-01, 4.0340e-01,
         1.1484e-01],
        [6.8033e-02, 1.0618e-01, 0.0000e+00,  ..., 7.9943e-01, 2.9750e-01,
         2.8274e-02],
        ...,
        [7.3517e-01, 6.9379e-01, 7.9943e-01,  ..., 2.1453e-04, 1.0935e+00,
         8.0255e-01],
        [3.5875e-01, 4.0340e-01, 2.9750e-01,  ..., 1.0935e+00, 0.0000e+00,
         2.9122e-01],
        [6.7537e-02, 1.1484e-01, 2.8274e-02,  ..., 8.0255e-01, 2.9122e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.23878371,  5.48276038,  5.5366698 ,  5.43544626,  5.69913817,
        5.5513072 , 10.98076873, 10.08001894, 13.0743399 , 12.7125079 ,
        5.95297459,  9.95552626,  8.91551231,  7.92644746,  7.58668144,
       10.14436912, 12.88270018,  7.63726134,  5.56809335,  9.5014146 ,
        8.86082124,  5.05444288, 11.80092578,  4.78002665,  8.68979052,
        6.36004393, 13.35013374,  6.49538841,  9.37989631,  7.76457095,
       11.75438006,  8.62329466, 12.67680439,  8.83725081, 10.9036226 ,
        8.04483495, 11.27160035, 13.85158417,  7.56617037, 11.27160035,
        7.33350642, 13.97857591,  5.13616224, 13.32557133,  5.19577146,
       10.68031351,  6.33075682,  9.5014146 , 11.72802491, 12.85032836,
        5.77219077,  6.7593917 , 13.99048094,  9.51074763, 13.60275211,
        9.34127013, 12.85032836, 11.57249984,  6.10360073, 11.54672755,
       10.24557047,  7.83486943,  5.05774295, 10.61472636, 11.05413545,
        6.02789634, 11.22142195, 10.60226614,  8.04664886,  6.48486171,
       11.9182956 , 10.1979669 ,  8.04483495, 10.63038565, 12.17451685,
        9.12220271, 10.06286074,  9.99145096, 12.30858677, 12.58006038,
       12.27991991, 10.04916678, 13.56367857,  9.43112957,  6.78325151,
        9.13507321,  9.18690018, 13.20597932, 13.37121923, 11.20100795,
       12.76408562, 11.07637531,  8.86082124,  9.44486609,  7.81509631,
        7.27340834,  7.27759233, 10.1371552 ,  6.99097013, 10.08001894,
        7.48311002,  9.97964013,  7.57685908,  7.56617037, 12.12641969,
        5.26067019])}
{'P': tensor([[3.3890e-02, 2.4649e-01, 1.5661e-01,  ..., 3.2857e-03, 2.3728e-05,
         3.4450e-03],
        [4.1562e-03, 3.9944e-02, 2.1675e-01,  ..., 4.0646e-03, 3.2298e-05,
         4.3148e-03],
        [3.1247e-03, 3.4900e-03, 5.5103e-02,  ..., 5.3267e-03, 7.7539e-05,
         7.9436e-03],
        ...,
        [1.8883e-03, 2.1433e-03, 3.4201e-03,  ..., 1.6026e-01, 1.5459e-03,
         9.9042e-04],
        [2.5538e-05, 2.4622e-05, 2.1968e-05,  ..., 3.6296e-05, 2.6983e-01,
         1.4358e-05],
        [7.6129e-04, 8.0730e-04, 1.1289e-03,  ..., 2.8941e-04, 3.4433e-04,
         1.4140e-01]]), 'X': tensor([[ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0096, 0.0315,  ..., 0.5029, 0.8157, 0.3298],
        [0.0096, 0.0000, 0.0223,  ..., 0.4962, 0.8075, 0.3328],
        [0.0315, 0.0223, 0.0000,  ..., 0.4881, 0.7943, 0.3325],
        ...,
        [0.5029, 0.4962, 0.4881,  ..., 0.0000, 0.3492, 0.8100],
        [0.8157, 0.8075, 0.7943,  ..., 0.3492, 0.0000, 1.0853],
        [0.3298, 0.3328, 0.3325,  ..., 0.8100, 1.0853, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.24137909,  5.30158497,  5.53255882,  5.54776934,  5.57341623,
        5.63824971, 13.59480906, 10.65929968, 14.03618649,  5.13616224,
       13.12218013,  5.86519525,  8.29022197,  6.383897  ,  5.64464807,
       11.42947632,  5.10444097, 12.04924159,  4.91336994,  5.51177501,
        6.81961996,  5.16077764, 10.61417181,  5.92281142, 13.14777433,
        8.09704052, 10.51437947,  5.87132718, 12.32408516,  9.66977802,
        8.09899874,  9.09661212,  6.30647746, 12.05395056,  8.4486771 ,
       13.08414689,  7.23276582, 13.00413829,  5.16205087, 11.87665501,
        9.41244865, 11.73567759, 10.27572673,  7.09887557, 11.04953767,
        4.91038884, 10.66631784,  9.54154096, 10.61472636,  5.86519525,
        7.62011734,  9.84314712, 10.65929968,  8.99276564, 12.35421461,
        5.05774295,  5.5748624 ,  8.31625171, 10.4606265 , 10.04916678,
       12.7125079 ,  6.41202986, 10.8992524 , 13.59480906, 10.82106505,
        6.19707944, 12.11735334, 10.84437705,  7.73692436,  8.92549752,
        5.70405525,  5.79391825, 10.05558893,  7.50917551,  6.16512856,
        9.43112957,  5.39723241, 10.37698435,  7.20826219,  6.65368551,
        5.83560009, 13.95472331, 13.32374056,  7.5403822 ,  7.39004226,
       13.00413829, 13.95472331,  8.36445758, 10.04916678,  7.73151309,
        6.46356806, 12.0495521 , 14.03505624,  7.32950385, 12.67576792,
        7.70897675,  5.13616224, 13.59480906, 13.6660671 , 10.30972193,
       11.87412439,  7.92607342,  5.05444288,  5.94535529, 13.97430191,
        6.42439344])}
{'P': tensor([[4.1441e-02, 1.8562e-01, 1.7116e-01,  ..., 1.1184e-04, 2.5096e-05,
         4.0419e-04],
        [6.0879e-03, 8.5248e-02, 7.6872e-03,  ..., 2.8889e-04, 9.4989e-05,
         1.7263e-03],
        [4.4340e-03, 3.1727e-01, 4.2027e-02,  ..., 1.3000e-04, 3.9074e-05,
         6.7399e-04],
        ...,
        [1.4415e-04, 9.4731e-05, 1.0873e-04,  ..., 4.0069e-02, 1.4509e-05,
         2.7919e-06],
        [1.4101e-05, 1.2743e-05, 1.3466e-05,  ..., 1.6089e-02, 1.5308e-01,
         1.2182e-05],
        [9.0374e-06, 1.1031e-05, 1.0278e-05,  ..., 1.2086e-03, 5.7274e-02,
         3.8715e-02]]), 'X': tensor([[  2.6614,  -4.5192,  -9.3641],
        [  3.7992,  -3.9474,  -8.9019],
        [  3.4140,  -4.1803,  -8.8224],
        [  4.0421,  -3.7765,  -8.3631],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.0506,   4.9277,  -9.5094],
        [  5.8450,  -9.9356,   3.0767],
        [ 11.2939,   6.6325,  -5.8311],
        [ -3.1671,   7.6158,   6.0765],
        [ -8.0965,  -5.9712,   4.3466],
        [  4.0362,  13.2397,  -7.0030],
        [ -0.0410,   7.8591,   9.3457],
        [ 10.7566,   7.6212,  -2.3279],
        [ -9.4196,   0.0478,   9.4140],
        [  5.4867,  12.5816,   2.9549],
        [  2.3490,  -4.6199,  -4.2489],
        [ -3.5903,   7.4847,  -8.9112],
        [  0.3715,  -4.7756,   0.2312],
        [  6.0305,   3.0296,   0.1375],
        [  3.1892, -10.8220,  -6.8409],
        [ -8.4267,  -5.3697,   2.4172],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.0608,   3.9362,  -5.9567],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.8745,   2.3097,   3.5600],
        [ 12.6038,   0.7665,  -3.4439],
        [  6.3542,   0.6934,  -3.3840],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.0348,  -2.8229,   3.5436],
        [ -1.0012, -10.8576,   2.2867],
        [ -5.9872,   6.2037,  -9.4477],
        [  1.5662,  13.9375,   4.5142],
        [  3.9591, -10.6391,  -6.2557],
        [  6.2365,  -0.2578,  -6.0212],
        [  7.5309,  11.2617,   7.2656],
        [  0.3231,  -4.7691,  -1.6120],
        [  3.5939,   6.4029,  -1.8357],
        [  9.3607,   9.5337,  -4.5412],
        [  1.4489,   7.5256,   4.4529],
        [ -2.1817,   7.8274,  -4.6298],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.9655,   3.2172,  -7.5953],
        [ -3.5520,   7.4976,   9.1522],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.3373,  -4.6232,  -5.2246],
        [ -8.3535,  -5.5118,   9.8399],
        [  0.6363, -11.0348,   6.1414],
        [  8.0877,  -8.5297,  -1.1050],
        [  5.4426, -10.1181,  -5.1268],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.5224,  -0.4497,   4.8448],
        [ -9.0604,   1.7707,   1.1066],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -8.8835,  -4.3178,   9.3080],
        [ 12.5717,   1.9266,   7.4903],
        [  7.9816,  10.8938,  -0.5378],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.3549,   0.7096,   6.9198],
        [  3.3943,   6.5483,   1.0790],
        [  6.1737,   2.5305,   3.9049],
        [  6.6319,  11.9066,   3.5165],
        [  1.8447,  -4.7372,   8.1235],
        [  3.6154,  -4.0640,  -7.9698],
        [ -9.4430,  -0.1897,   4.8383],
        [  1.4150,  13.9647,  -4.0915],
        [  0.6142,  14.0802,   0.5407],
        [  6.0890,  -0.8382,   2.9166],
        [  5.0277,   4.9589,   7.0175],
        [  2.6614,  -4.5192,  -9.3641],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -3.4986,   7.5152,   9.0712],
        [  0.3493,  -4.7727,  -6.5682],
        [ -4.0409,   7.3169,  -3.5779],
        [ -5.6169,  -8.7844,   8.9057],
        [ 11.7726,  -3.4876,  -6.7402],
        [  4.4049,   5.6892,  -4.4394],
        [  2.6078, -10.9229,   6.2869],
        [ -9.0574,   1.7801,   8.8986],
        [  6.0073,   3.0986,   6.7937],
        [ 12.5199,  -0.4711,   3.0436],
        [  5.9193,  -1.3176,   5.1739],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.0348,  -2.8229,   3.5436],
        [ -7.0948,   5.1940,   9.8171],
        [ 11.6165,  -3.8715,   6.4594],
        [  1.2273,  -4.8081,   9.8129],
        [  9.8082,  -6.8396,   4.9615],
        [ 12.5650,   2.0132,   4.6272],
        [ 12.5832,   1.7543,  -7.6134],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.2036,   2.4030,  -1.2500],
        [  2.8602,   6.8905,   5.8779],
        [ -0.6323,   7.9085,  -5.9544],
        [ -9.4430,  -0.1897,   4.8383],
        [ -1.5875,   7.8944,  -9.4286],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -5.6792,  -8.7352,   4.0318],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.3384,  -2.6161,  -4.7630],
        [  3.9663,   6.1018,  -5.4104],
        [ -5.0550,   6.8231,  -9.2502],
        [  1.6308,  13.9254,  -2.0984],
        [ 10.0400,   8.6922,  -6.2847]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.3001e-02, 1.2869e-02,  ..., 4.8312e-01, 2.8691e-01,
         3.2154e-01],
        [2.3001e-02, 0.0000e+00, 1.0335e-02,  ..., 5.0589e-01, 2.7395e-01,
         2.9960e-01],
        [1.2869e-02, 1.0335e-02, 2.1453e-04,  ..., 4.9598e-01, 2.7830e-01,
         3.0980e-01],
        ...,
        [4.8312e-01, 5.0589e-01, 4.9598e-01,  ..., 2.1453e-04, 7.1405e-01,
         7.8625e-01],
        [2.8691e-01, 2.7395e-01, 2.7830e-01,  ..., 7.1405e-01, 0.0000e+00,
         3.5287e-01],
        [3.2154e-01, 2.9960e-01, 3.0980e-01,  ..., 7.8625e-01, 3.5287e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.24461068,  5.47866221,  5.39723241,  5.53175275,  5.41348121,
        5.83560009,  7.05627461, 11.52733581, 13.09737053,  8.24808832,
       10.06023665, 13.84126197,  7.85919687, 13.18278606,  9.41970031,
       13.72594022,  5.18277858,  8.30125201,  4.79003361,  6.74872981,
       11.28215429,  9.99211663,  5.38342077,  8.97052115, 11.76830981,
        9.17016857, 12.62711252,  6.39187285, 12.07225469,  5.77219077,
       10.9036226 ,  8.62162372, 14.02526505, 11.35182589,  6.24187044,
       13.54774355,  4.78002665,  7.34250948, 13.36092624,  7.66377554,
        8.12580976,  5.1224079 ,  6.77777606,  8.29640613, 11.9120321 ,
        5.18047638, 10.00802627, 11.05317597, 11.75438006, 11.4890861 ,
        5.47702766, 12.53047489,  9.23178204, 10.31613269,  9.87720102,
       12.71843462, 13.50484539,  9.36093562,  6.39438044,  7.37576154,
        6.67218687, 13.62896593,  5.08372149,  5.43943791,  9.44486609,
       14.03618649, 14.09357547,  6.1463896 ,  7.06169391,  5.24461068,
        9.85330646,  8.28968067,  4.78544875,  8.35856487, 10.42667093,
       12.27835996,  7.19513806, 11.22993218,  9.23071698,  6.7593917 ,
       12.52875793,  6.06416957,  9.87720102,  6.1561646 ,  5.77219077,
        8.79284466, 12.24467041,  4.96231532, 11.95743558, 12.72524615,
       12.70489563,  8.85572292, 12.27991991,  6.652742  ,  7.46053281,
        7.9337647 ,  9.44486609,  8.05242591,  9.5014146 , 10.41909211,
       13.08414689,  9.69792513,  7.27759233,  8.49161154, 14.02058621,
       13.27994304])}
{'P': tensor([[2.3106e-02, 1.3182e-01, 1.1405e-01,  ..., 1.3264e-04, 1.4504e-03,
         1.9163e-03],
        [2.4998e-03, 3.2327e-02, 6.0708e-03,  ..., 1.2182e-04, 1.0307e-03,
         4.2291e-03],
        [3.2819e-03, 8.1425e-02, 2.7267e-02,  ..., 1.0975e-04, 1.2743e-03,
         2.7498e-03],
        ...,
        [2.5883e-04, 3.7990e-04, 2.7141e-04,  ..., 1.5941e-02, 3.2645e-04,
         6.9939e-05],
        [4.2461e-03, 8.2274e-03, 6.7028e-03,  ..., 1.6369e-04, 8.3837e-02,
         1.3788e-03],
        [2.7368e-04, 2.9277e-04, 2.7647e-04,  ..., 9.6248e-06, 1.4533e-05,
         7.7539e-02]]), 'X': tensor([[  2.7420,  -4.4896,  -3.7139],
        [  3.5965,  -4.0754,  -3.5035],
        [  3.2226,  -4.2802,  -4.0428],
        [  4.0642,  -3.7599,  -3.4565],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.1999,  -3.6543,  -4.1998],
        [ -2.4956, -10.4682,   3.5632],
        [ 11.6544,   5.8273,  -3.1375],
        [ 12.5650,   2.0132,   4.6272],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.6202,  -1.9508,   0.7471],
        [ -9.4501,  -1.7133,   2.2980],
        [  5.6248,   3.9984,   8.4933],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.8979,   3.3958,  -3.8713],
        [  9.2692,   9.6370,   7.0746],
        [ -6.4844,  -8.0230,   8.5209],
        [  6.9653,  -9.3225,  -3.4727],
        [  9.3624,   9.5318,   0.2779],
        [  5.6260,  -1.9401,   3.0259],
        [  9.3607,   9.5337,  -4.5412],
        [  6.8704,  -9.3809,   7.5769],
        [  4.6063,  13.0051,  -3.1456],
        [  5.6202,  -1.9508,   0.7471],
        [ -0.3432,   7.8900,   7.1805],
        [ 11.5568,   6.0602,   9.3244],
        [ -1.3940, -10.7772,   6.7835],
        [ -6.6895,   5.6051,   2.7899],
        [  2.0449,  13.8402,  -7.1642],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.4932,   7.0903,  -5.4214],
        [ -8.0588,  -6.0341,  -2.3006],
        [ 10.8218,  -5.4177,   6.8573],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.9655,   3.2172,  -7.5953],
        [  5.8450,  -9.9356,   3.0767],
        [  1.3767, -11.0341,   9.9415],
        [ 10.4737,  -5.9570,  -1.2108],
        [  6.0305,   3.0296,   0.1375],
        [ -8.6426,  -4.9136,   9.4941],
        [  6.0031,  12.2958,  -6.5992],
        [  5.8325,  -1.5222,  -2.1341],
        [ -2.6485, -10.4151,   0.8766],
        [  2.3376,  -4.6231,   9.8436],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.3373,  13.7721,  -0.2442],
        [  0.3493,  -4.7727,  -6.5682],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -6.9838,  -7.4965,   0.9796],
        [ -1.5261,   7.8988,   9.3768],
        [ 11.0891,  -4.9558,  -2.3178],
        [  3.5939,   6.4029,  -1.8357],
        [  6.8093,  -9.4178,   8.4244],
        [ 10.9875,   7.2207,  -8.8549],
        [  0.5960, -11.0335,   3.1144],
        [  3.5284, -10.7484,   5.3347],
        [  2.7976,  -4.4682,  -4.2020],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -9.0961,   1.6529,   3.1553],
        [  3.5965,  -4.0754,  -3.5035],
        [  4.9452, -10.3187,  -7.8621],
        [ -9.0961,   1.6529,   3.1553],
        [ -1.0012, -10.8576,   2.2867],
        [  0.3038,   7.8092,   0.4164],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -1.6220,   7.8917,   0.0315],
        [ -4.8751,  -9.3148,   4.0406],
        [ -6.0195,   6.1789,  -2.9496],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -0.5385,   7.9037,   6.3830],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.3587,   1.1596,  -3.6629],
        [  1.1994,   7.6035,   2.4295],
        [  7.5930,  -8.9037,   7.2323],
        [ -9.1231,  -3.5739,   8.4708],
        [  4.5491,  -3.3485,   6.8412],
        [  9.0652,   9.8597,  -3.5949],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -8.1666,   3.7653,   3.6075],
        [ -8.5149,  -5.1905,   1.5239],
        [  9.6816,   9.1536,  -5.8543],
        [  4.1999,  -3.6543,  -4.1998],
        [  0.3231,  -4.7691,  -1.6120],
        [  2.6614,  -4.5192,  -9.3641],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.7544,   2.6096,   3.3146],
        [  6.3542,   0.6934,  -3.3840],
        [ 12.2435,   4.0076,   8.1257],
        [  5.5626,  -2.0536,   8.4433],
        [ 12.2311,  -2.0445,   1.3545],
        [ 10.9931,   7.2107,   7.7933],
        [ 12.2435,   4.0076,   8.1257],
        [ -0.8455, -10.8853,  -6.7400],
        [  7.0358,  -9.2784,  -9.8479],
        [  3.9913,  -3.8139,  -6.8311],
        [  5.7218,  -9.9934,   6.5764],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.2675,  12.1379,  -8.9720],
        [ -5.3179,  -9.0101,   9.4472],
        [  7.3592,  11.3937,   6.2407],
        [ 11.9936,  -2.8677,   6.0614],
        [  4.7544,  -3.1425,  -1.5315],
        [  3.5284, -10.7484,   5.3347],
        [  6.3122,  -9.6994,  -6.3282],
        [  5.2199,   4.6861,  -1.9304]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0179, 0.0210,  ..., 0.8748, 0.3578, 0.1178],
        [0.0179, 0.0000, 0.0240,  ..., 0.8787, 0.3745, 0.1082],
        [0.0210, 0.0240, 0.0000,  ..., 0.8958, 0.3558, 0.1321],
        ...,
        [0.8748, 0.8787, 0.8958,  ..., 0.0000, 0.9996, 0.8013],
        [0.3578, 0.3745, 0.3558,  ..., 0.9996, 0.0000, 0.4620],
        [0.1178, 0.1082, 0.1321,  ..., 0.8013, 0.4620, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.26067019,  5.43544626,  5.35774573,  5.5366698 ,  5.5513072 ,
        5.5671389 , 10.76154256, 13.03002338, 12.72524615, 10.52357479,
        5.94909709,  9.60412735,  6.90115901,  9.11473391,  6.80559669,
       13.37121923, 10.31584102, 11.63724792, 13.3607379 ,  5.95111804,
       13.36092624, 11.62770332, 13.79676654,  5.94909709,  7.89745107,
       13.0493368 , 10.86693708,  8.7273569 , 13.99048094,  9.68324553,
        7.5158489 , 10.0674887 , 12.10222204, 12.22503754,  6.77777606,
       11.52733581, 11.11969815, 12.04924159,  6.74872981,  9.94173708,
       13.6829843 ,  6.02789634, 10.74655346,  5.1805345 , 13.01728313,
       13.96905756,  4.78544875, 10.35485331, 10.24557047,  8.04483495,
       12.1460931 ,  7.34250948, 11.62158554, 13.14777433, 11.04953767,
       11.31276109,  5.27178363, 10.57302546,  9.24501829,  5.43544626,
       11.4424816 ,  9.24501829, 10.9036226 ,  7.81509631, 10.99326577,
       10.06407839,  8.05668609, 10.51340829,  8.62631801, 12.53345445,
        7.92200342, 12.35421461,  6.46356806,  7.69752262, 11.70168755,
        9.79814997,  5.64865173, 13.39372426,  9.55720651,  8.99276564,
        9.97220799, 13.32374056,  5.5671389 ,  4.78002665,  5.24461068,
       13.1399181 ,  9.13507321,  6.39187285, 12.88270018,  5.92952538,
       12.40074901, 13.14690137, 12.88270018, 10.91805683, 11.64436684,
        5.5205189 , 11.51557166,  6.02789634, 13.66051884, 10.46238054,
       13.56367857, 12.33167382,  5.69913817, 11.31276109, 11.57249984,
        7.01475376])}
{'P': tensor([[3.7600e-02, 1.7468e-01, 1.5910e-01,  ..., 1.9694e-03, 8.5343e-03,
         1.1654e-04],
        [3.0313e-03, 5.2730e-02, 2.4754e-01,  ..., 3.0309e-03, 2.2632e-02,
         2.3435e-04],
        [4.1864e-03, 1.3661e-02, 7.6160e-02,  ..., 4.1411e-03, 3.6285e-02,
         3.4328e-04],
        ...,
        [2.1742e-04, 4.9006e-04, 5.2062e-04,  ..., 1.3309e-01, 2.7216e-03,
         5.1400e-04],
        [2.1971e-04, 3.2737e-04, 3.4505e-04,  ..., 6.6808e-05, 3.1472e-02,
         3.3638e-04],
        [1.3188e-04, 9.1503e-05, 8.7324e-05,  ..., 5.3225e-06, 3.8852e-05,
         7.7023e-02]]), 'X': tensor([[  2.7558,  -4.4844,   0.0636],
        [  4.1140,  -3.7220,  -0.1273],
        [  4.2275,  -3.6320,  -0.2714],
        [  4.0457,  -3.7737,  -0.4704],
        [  4.3459,  -3.5327,   0.3667],
        [  4.5057,  -3.3894,  -0.8160],
        [ 12.4691,   2.8525,  -5.5110],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4721,  -1.3097,  -7.4377],
        [  0.6396,  14.0772,  -6.3786],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.8979,   3.3958,  -3.8713],
        [  1.8719,   7.3713,  -3.7465],
        [  8.6271,  -8.0715,  -7.4813],
        [  6.3031,   1.8477,  -7.1706],
        [  5.7784,  -1.6401,   9.5587],
        [  5.9912,   3.1453,  -3.9989],
        [  4.0362,  13.2397,  -7.0030],
        [ -8.9033,   2.2329,  -8.8912],
        [ 10.5678,   7.9261,  -0.6100],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.0233,   4.9647,  -3.1788],
        [  5.6523,   3.9443,  -2.9449],
        [  9.4516,  -7.2501,  -1.8868],
        [ -1.0482, -10.8487,  -5.9066],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.3717,  -5.4769,  -9.2228],
        [  5.8902,  -1.3887,   4.0388],
        [ -7.0948,   5.1940,   9.8171],
        [  6.3236,   0.3011,  -4.6668],
        [  9.0652,   9.8597,  -3.5949],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.8540,  -4.3971,  -5.2144],
        [  0.6952,  -4.8060,   8.8976],
        [  5.1292,   4.8183,  -5.7997],
        [  9.9217,  -6.7006,  -5.6728],
        [  1.3670,  -4.7990,   3.2603],
        [  5.4732,  -2.2041,  -4.3830],
        [  2.4356,  -4.5942,  -5.0808],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -5.1369,   6.7755,   8.3836],
        [  6.0073,   3.0986,   6.7937],
        [ -8.9379,  -4.1655,  -2.8114],
        [  7.2340,  -9.1505,  -5.8321],
        [  0.9809,  -4.8145,   6.8706],
        [ -1.9600,   7.8577,   6.9121],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.2571,  -4.7594,   1.5618],
        [  1.4150,  13.9647,  -4.0915],
        [  0.1931,  -4.7491,   7.4385],
        [  5.8201,   3.5841,  -8.0609],
        [ -2.2188,   7.8218,   8.6064],
        [ -5.1377,   6.7751,  -0.4438],
        [  1.3767, -11.0341,   9.9415],
        [  0.9809,   7.6642,  -1.4217],
        [  6.3535,   0.6792,  -7.5101],
        [ -5.1974,   6.7396,   3.3484],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.2036,   2.4030,  -1.2500],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.6050,   2.9451,  -0.7012],
        [  9.3624,   9.5318,   0.2779],
        [  1.7232,  -4.7574,   3.1270],
        [ 10.9916,   7.2133,  -0.9179],
        [ -4.7847,  -9.3729,  -3.7741],
        [  9.7328,   9.0902,  -0.8917],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.1504,   4.3674,  -7.2998],
        [  2.6613,   7.0021,   4.5783],
        [  4.8653,   5.1694,  -4.0823],
        [  6.3587,   1.1596,  -3.6629],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.9956,  11.6590,  -7.6876],
        [  3.9591, -10.6391,  -6.2557],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.4495,   6.3026,   6.8257],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.0890,  -0.8382,   2.9166],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.1504,  -4.6726,   8.7908],
        [  9.0129,  -7.7072,   9.1400],
        [ -1.6906,   7.8860,   7.7142],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.1222,  -0.7261,   8.8204],
        [  0.6933,   7.7336,   2.1179],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -2.0721,   7.8432,   6.6380],
        [  0.6396,  14.0772,  -6.3786],
        [  5.6672,  -1.8631,   8.1020],
        [ -6.3489,   5.9117,   6.5161],
        [ -1.5408,   7.8978,   5.3470],
        [  3.0876,   6.7527,  -5.3540],
        [  0.4996,   7.7738,  -4.4711],
        [  1.2728, -11.0372,   9.8121],
        [  4.2339,  -3.6268,   6.3125],
        [ -8.6675,  -4.8569,  -9.1769],
        [  6.3246,   1.6696,  -0.1544],
        [  3.5939,   6.4029,  -1.8357],
        [  0.5850,  -4.7982,   9.6639],
        [ 11.7162,   5.6728,  -8.9372],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.3587,   1.1596,  -3.6629],
        [ -2.5056,   7.7720,   5.5878]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0282, 0.0346,  ..., 0.6175, 0.2447, 0.5169],
        [0.0282, 0.0000, 0.0097,  ..., 0.6200, 0.2325, 0.5356],
        [0.0346, 0.0097, 0.0000,  ..., 0.6286, 0.2231, 0.5452],
        ...,
        [0.6175, 0.6200, 0.6286,  ..., 0.0000, 0.8458, 0.3685],
        [0.2447, 0.2325, 0.2231,  ..., 0.8458, 0.0000, 0.7582],
        [0.5169, 0.5356, 0.5452,  ..., 0.3685, 0.7582, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.26342203,  5.54776934,  5.57341623,  5.53255882,  5.60064319,
        5.63824971, 12.79126785,  9.5014146 ,  9.562174  , 14.09176259,
       10.51340829,  6.80559669,  7.60529892, 11.81423161,  6.56833864,
        6.00663314,  6.76661595, 13.84126197,  9.17905491, 13.20988996,
        9.7751189 ,  7.06272256,  6.89245352, 11.9120321 , 10.8992524 ,
       11.95743558, 10.00410625,  6.05165248,  8.79284466,  6.33075682,
       13.39372426,  9.7712321 ,  9.88571713,  4.85603886,  7.03733775,
       11.97238624,  4.98989906,  5.90035324,  5.19986797, 12.35625694,
        8.50268322,  6.7593917 ,  9.86089579, 11.66453059,  4.91336994,
        8.09842704, 11.81423161,  4.7663526 , 14.03618649,  4.75303127,
        6.83515201,  8.1303945 ,  8.50278896, 11.11969815,  7.72668467,
        6.38968357,  8.51088913,  5.18277858,  6.652742  , 10.48840476,
        9.09502385, 13.3607379 ,  5.05989547, 13.1471313 , 10.52357479,
       13.31764181, 12.33167382, 12.91143491,  7.49076899,  7.09887557,
        6.46356806, 10.48315436, 13.59673406, 11.35182589, 13.18278606,
       13.06957425, 12.64235359,  6.1463896 ,  5.84450127,  5.14371005,
       11.85890664,  8.06517143,  9.78286733,  8.85572292, 11.51811603,
        6.16512856,  7.76457095,  4.89869205,  9.79039371,  8.11227668,
       14.09176259,  5.96556248,  8.67502348,  8.04664886,  7.42512621,
        7.78980299, 11.11038551,  5.5748624 ,  9.93553424,  6.54127431,
        7.34250948,  4.83371341, 13.01728313, 12.33167382,  6.46356806,
        8.16584537])}
{'P': tensor([[3.2924e-02, 2.1362e-01, 1.5393e-01,  ..., 5.6440e-04, 4.0330e-03,
         6.0671e-04],
        [3.8697e-03, 3.7959e-02, 2.2760e-01,  ..., 7.8547e-04, 4.3446e-03,
         8.5893e-04],
        [6.5439e-03, 7.6107e-03, 1.1257e-01,  ..., 3.5697e-03, 8.3139e-03,
         3.8802e-03],
        ...,
        [1.4374e-04, 1.3868e-04, 1.3040e-04,  ..., 6.4899e-02, 7.7342e-06,
         3.5943e-03],
        [4.7210e-04, 6.7928e-04, 1.6048e-03,  ..., 3.8031e-04, 4.2333e-02,
         7.7411e-04],
        [1.1633e-04, 1.1706e-04, 1.1873e-04,  ..., 8.3686e-02, 9.5739e-06,
         6.3178e-02]]), 'X': tensor([[  2.7976,  -4.4682,  -4.2020],
        [  3.2226,  -4.2802,  -4.0428],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.1999,  -3.6543,  -4.1998],
        [  4.0642,  -3.7599,  -3.4565],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.9956,  11.6590,  -7.6876],
        [ 11.7349,  -3.5838,   2.1644],
        [ 10.3251,   8.2928,  -3.0006],
        [  1.7228,  -4.7575,   8.3396],
        [  5.7218,  -9.9934,   6.5764],
        [ -6.7992,   5.4991,   9.9186],
        [ -0.8569,   7.9156,  -6.5557],
        [ -5.1974,   6.7396,   3.3484],
        [  0.6470, -11.0352,  -1.1931],
        [  7.7750,  11.0665,   2.8364],
        [  0.1252, -11.0062,   7.2564],
        [  4.5503,   5.5359,   7.4735],
        [ -5.0943,  -9.1682,   0.5073],
        [ -1.6906,   7.8860,   7.7142],
        [  3.1271, -10.8343,   2.6442],
        [  9.5018,  -7.1945,  -1.1892],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.0449,  13.8402,  -7.1642],
        [  6.0345,   3.0175,  -1.4424],
        [ -8.8154,   2.4611,   2.5198],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.6928,  12.4710,   3.6001],
        [ -1.9714,   7.8563,  -1.8084],
        [  4.6063,  13.0051,  -3.1456],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.5789,   0.1722,   1.6470],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.3246,   1.6696,  -0.1544],
        [  0.9445,   7.6736,   0.5095],
        [ -3.0004,   7.6607,  -7.2640],
        [  4.6357,   5.4415,   7.9067],
        [  0.8940, -11.0402,  -0.9783],
        [ -9.2707,   0.9661,   4.4863],
        [ -7.3554,   4.8985,  -0.6665],
        [  4.9452, -10.3187,  -7.8621],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.3589,   0.8239,  -8.4230],
        [  3.2809,  13.5061,  -4.0199],
        [  5.1196,  -2.7160,   4.2705],
        [ -7.3446,   4.9113,  -5.9356],
        [  6.9693,  -9.3200,  -8.3822],
        [  1.5662,  13.9375,   4.5142],
        [  6.3569,   0.7591,   6.5011],
        [  6.1579,  -0.5961,  -6.5595],
        [ -4.2271,   7.2387,  -1.8316],
        [  8.8474,  10.0868,  -8.8644],
        [  2.8994,   6.8675,  -9.5987],
        [ -7.2515,   5.0195,   4.6434],
        [  2.1824,  -4.6647,   1.1985],
        [ -5.0550,   6.8231,  -9.2502],
        [ -3.6381,   7.4683,   7.3544],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.4378,  -1.8606,  -2.5668],
        [  4.8432,   5.1968,   1.5006],
        [ -8.1204,  -5.9308,   4.0761],
        [ -2.1382,   7.8339,   7.3524],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.0588,  -6.0341,  -2.3006],
        [  0.1931,  -4.7491,   7.4385],
        [  2.5620,   7.0549,   8.4825],
        [ -8.0608,   3.9362,  -5.9567],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.4159,  -2.0739,   9.6841],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -2.1533, -10.5778,   1.0112],
        [  7.0515,  11.6194,   8.4289],
        [  4.5767,   5.5071,   3.9069],
        [  6.0955,   2.8210,  -4.3321],
        [  9.0129,  -7.7072,   9.1400],
        [  5.9883,   3.1536,   6.2320],
        [ 10.2712,   8.3707,  -4.9743],
        [ 10.2470,   8.4054,  -0.1177],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.3633,  13.4794,   0.5193],
        [ -9.4501,  -1.7133,   2.2980],
        [ 12.4691,   2.8525,  -5.5110],
        [  7.4132,  11.3526,   0.3322],
        [  3.3943,   6.5483,   1.0790],
        [ 12.5763,   0.1333,  -9.2650],
        [  9.0129,  -7.7072,   9.1400],
        [ -5.1377,   6.7751,  -0.4438],
        [ -1.8367,   7.8718,  -8.6523],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.4196,   0.0478,   9.4140],
        [ 12.1588,   4.3364,  -8.0867],
        [ -2.5161,   7.7699,  -9.3531],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.2849, -10.9655,  -7.6542],
        [  1.3767, -11.0341,   9.9415],
        [ 12.5258,  -0.4200,  -9.3493],
        [  3.9591, -10.6391,  -6.2557],
        [  5.1203,  -2.7151,   6.2110],
        [  2.0039,   7.3173,  -1.5885],
        [  9.5018,  -7.1945,  -1.1892],
        [  3.3943,   6.5483,   1.0790]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0096, 0.0264,  ..., 0.1747, 0.2067, 0.3471],
        [0.0096, 0.0000, 0.0176,  ..., 0.1746, 0.2088, 0.3432],
        [0.0264, 0.0176, 0.0000,  ..., 0.1841, 0.2211, 0.3455],
        ...,
        [0.1747, 0.1746, 0.1841,  ..., 0.0000, 0.0541, 0.1977],
        [0.2067, 0.2088, 0.2211,  ..., 0.0541, 0.0000, 0.2164],
        [0.3471, 0.3432, 0.3455,  ..., 0.1977, 0.2164, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.27178363,  5.35774573,  5.5513072 ,  5.5671389 ,  5.5366698 ,
        5.56809335, 13.59673406, 12.26997165, 13.2430377 ,  5.05981914,
       11.51557166,  8.74470218,  7.96181288,  8.51088913, 11.05413545,
       13.5247215 , 11.00694688,  7.16598273, 10.48840476,  8.06517143,
       11.27656488, 11.9182956 ,  8.20436764, 13.99048094,  6.74686071,
        9.15253344,  9.562174  , 13.70893492,  8.09984231, 13.79676654,
        9.84314712, 12.58006038, 10.46739073,  9.5642993 ,  6.54127431,
        7.73151309,  8.22728399,  7.14837514, 11.07637531,  9.32094055,
        8.83725081, 11.4424816 , 10.00794468,  6.41202986, 13.89886469,
        5.79543038,  8.8353674 , 11.63765273, 14.02526505,  6.4020404 ,
        6.18668463,  8.38252937, 13.41713661,  7.4544866 ,  8.81930525,
        5.14999799,  8.49161154,  8.30729352,  5.1224079 ,  9.50654967,
        9.91668182,  9.61942445,  7.10378919, 10.05558893,  8.12043983,
       10.61472636, 10.0674887 ,  4.75303127,  7.50564223,  8.97052115,
        5.43544626,  9.64157461, 10.37698435, 10.794755  , 13.59170809,
        7.16058587,  6.7166297 , 11.85890664,  6.76791483, 13.25017211,
       13.25336001, 12.35625694,  8.37954853, 12.52372629, 13.89264822,
        9.60412735, 12.79126785, 13.55868684,  7.37576154, 12.5769659 ,
       11.85890664,  8.50278896,  8.08320235,  9.96152788,  9.41970031,
       12.9089559 ,  8.16714836,  8.25828409, 11.20100795, 11.11969815,
       12.53285633, 11.35182589,  5.79562691,  7.58668144, 11.9182956 ,
        7.37576154])}
{'P': tensor([[3.7014e-02, 1.5458e-01, 1.2598e-01,  ..., 2.7642e-03, 3.3304e-04,
         5.4266e-03],
        [2.5450e-03, 4.9955e-02, 2.5110e-02,  ..., 4.6203e-03, 8.1881e-04,
         2.5800e-03],
        [3.4741e-03, 4.7280e-02, 5.7839e-02,  ..., 3.8691e-03, 9.9344e-04,
         3.3608e-03],
        ...,
        [4.1060e-04, 1.2339e-03, 9.7711e-04,  ..., 2.0318e-01, 1.0623e-03,
         1.8229e-04],
        [1.9814e-04, 1.5075e-04, 1.6121e-04,  ..., 5.6319e-06, 9.1829e-02,
         8.9622e-06],
        [6.9447e-04, 3.2141e-03, 3.0591e-03,  ..., 3.6903e-02, 7.6122e-05,
         3.4079e-02]]), 'X': tensor([[  2.8890,  -4.4316,   1.1600],
        [  4.4290,  -3.4596,   1.4181],
        [  4.3459,  -3.5327,   0.3667],
        [  3.4604,  -4.1545,   1.8091],
        [  5.1667,  -2.6543,   1.2879],
        [  4.1140,  -3.7220,  -0.1273],
        [  2.3261,   7.1726,   6.2522],
        [  6.2247,  -0.3150,  -9.3657],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -8.4297,  -5.3637,  -9.3962],
        [  6.3569,   0.7591,   6.5011],
        [ -9.1795,  -3.3623,   8.5259],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.8093,  -9.4178,   8.4244],
        [  5.1142,  -2.7230,  -4.1966],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.4572,   7.1085,   7.7376],
        [  7.7750,  11.0665,   2.8364],
        [  6.3589,   0.8239,  -8.4230],
        [  6.2038,   2.4020,  -3.1692],
        [ -0.0282,   7.8575,  -1.1746],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.1624,  -3.4286,   2.0647],
        [ -8.8058,   2.4850,  -1.7425],
        [ 12.5789,   0.1722,   1.6470],
        [  6.7565,  11.8237,  -4.9882],
        [  7.0642,  11.6104,  -9.8316],
        [  6.3534,   1.2990,  -7.2942],
        [  6.0305,   3.0296,   0.1375],
        [ -5.1046,  -9.1610,   2.3418],
        [ -8.1531,  -5.8748,   2.6934],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -5.3736,   6.6310,  -1.5174],
        [ -2.7572,   7.7194,   5.9580],
        [  3.5965,  -4.0754,  -3.5035],
        [  2.3928,   7.1404,  -1.7035],
        [  6.0140,  -1.0661,  -2.5040],
        [  4.2041,  -3.6509,  -4.9488],
        [ -0.8675, -10.8815,   6.8561],
        [  2.0729,   7.2878,   7.0880],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -8.5859,  -5.0393,   0.5476],
        [ -3.4100,   7.5436,  -7.6980],
        [ -8.0965,  -5.9712,   4.3466],
        [  8.6420,  -8.0580,   6.5781],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.0275,  -5.0666,   8.4154],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -3.3116, -10.1539,   7.5919],
        [ -6.2200,  -8.2737,   5.5079],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.5900,   2.9767,   7.4811],
        [ -1.9600,   7.8577,   6.9121],
        [  6.2022,   2.4092,   3.0349],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.0617,  12.7957,   8.3544],
        [ -7.4887,   4.7364,   8.9074],
        [  2.4697,  -4.5836,   1.4593],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.6603,   2.8250,   5.9181],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.2974,   0.8371,   2.4852],
        [  8.7715,  -7.9390,   5.2900],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.0876,   6.7527,  -5.3540],
        [ -8.0829,  -5.9940,   5.0564],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.0965,  -5.9712,   4.3466],
        [ -8.5149,  -5.1905,   1.5239],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.7963,  -1.6018,   1.8856],
        [  7.4132,  11.3526,   0.3322],
        [ -5.1377,   6.7751,  -0.4438],
        [ -1.5875,   7.8944,  -9.4286],
        [  4.8051, -10.3703,   0.5268],
        [ -3.3812, -10.1235,   7.4384],
        [ -3.9508,  -9.8515,   1.7923],
        [ -8.9788,   2.0208,  -3.2867],
        [  7.2340,  -9.1505,  -5.8321],
        [ 10.4994,  -5.9193,  -7.5153],
        [  0.5600, -11.0321,  -0.8660],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.2647, -10.8066,  -0.3768],
        [ -2.5056,   7.7720,   5.5878],
        [  2.2437,  -4.6489,   6.1430],
        [  6.3505,   0.6235,  -8.5849],
        [  5.9193,  -1.3176,   5.1739],
        [  5.7218,  -9.9934,   6.5764],
        [  4.9152,  12.8653,  -1.4459],
        [  6.3516,   0.6425,   0.3425],
        [  3.7298,  13.3537,  -9.4653],
        [ -5.0943,  -9.1682,   0.5073],
        [  4.5057,  -3.3894,  -0.8160],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.3211,   1.7017,  -0.2164],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.3122,  -9.6994,  -6.3282],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.0591,   6.0202,  -1.6339],
        [ 11.9936,  -2.8677,   6.0614],
        [ -0.1963,   7.8765,  -1.1371],
        [  8.8091,  -7.9037,   0.3631]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0419, 0.0518,  ..., 0.5425, 0.2298, 0.1686],
        [0.0419, 0.0000, 0.0702,  ..., 0.5140, 0.2717, 0.2079],
        [0.0518, 0.0702, 0.0000,  ..., 0.5842, 0.2280, 0.1892],
        ...,
        [0.5425, 0.5140, 0.5842,  ..., 0.0000, 0.7267, 0.6262],
        [0.2298, 0.2717, 0.2280,  ..., 0.7267, 0.0000, 0.1028],
        [0.1686, 0.2079, 0.1892,  ..., 0.6262, 0.1028, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.29009836,  5.62005629,  5.60064319,  5.40689332,  5.80860165,
        5.54776934,  7.5403822 ,  6.23262289, 10.23946247,  9.99145096,
        6.4020404 ,  9.77588801, 12.79126785, 11.62158554,  5.79391825,
       12.04924159, 12.95702647,  7.52117132, 13.5247215 ,  6.41202986,
        6.65259775,  7.8575648 ,  8.09704052,  9.78285424,  9.14973319,
       12.58006038, 13.61801526, 13.59056613,  6.48486171,  6.74872981,
       10.48721382, 10.04916678,  9.75401449,  8.53501777,  8.19703512,
        5.43544626,  7.53063885,  6.10774544,  5.56809335, 10.91601655,
        7.57685908, 12.05395056,  9.95552626,  8.27853076, 10.06023665,
       11.8159254 ,  5.22903117, 12.13568598,  9.88571713, 10.68031351,
       10.35093485, 10.27716434, 10.52357479,  9.09119351,  8.09842704,
        6.65368551, 12.59969936,  5.16077764, 13.76047999,  8.86082124,
        5.20659892, 10.99326577,  9.10945678,  8.53014915,  9.3349858 ,
       11.83075005,  5.53255882,  7.42512621, 10.06286074, 13.99048094,
       10.06023665,  9.97220799,  6.10774544,  6.01356004, 13.55868684,
        8.50278896,  8.05242591, 11.42947632, 10.67322527, 10.61417181,
        9.20340031, 11.66453059, 12.05300943, 11.04629471,  4.76969364,
       11.28895489,  8.16584537,  5.16205087,  6.38105015,  6.06416957,
       11.51557166, 13.77222752,  6.38399789, 13.86480175, 10.48840476,
        5.63824971,  8.8353674 ,  5.8828986 ,  6.54615887, 11.74901778,
       11.57249984, 13.01728313,  7.26074925, 12.33167382,  7.8788965 ,
       11.83509372])}
{'P': tensor([[4.1019e-02, 3.0043e-01, 1.4857e-01,  ..., 3.5241e-05, 7.0244e-06,
         3.3721e-04],
        [4.3885e-03, 4.9298e-02, 2.0649e-01,  ..., 3.6496e-05, 8.6571e-06,
         4.3891e-04],
        [2.9088e-03, 3.7166e-03, 5.7116e-02,  ..., 2.7771e-05, 9.4099e-06,
         4.3963e-04],
        ...,
        [1.9656e-03, 1.5983e-03, 7.4164e-04,  ..., 6.4451e-02, 1.8532e-05,
         8.6362e-05],
        [2.8040e-05, 2.4351e-05, 1.3817e-05,  ..., 3.5641e-03, 6.7124e-02,
         5.0185e-05],
        [9.9647e-05, 1.1042e-04, 1.2294e-04,  ..., 2.8935e-05, 5.4847e-04,
         8.3653e-02]]), 'X': tensor([[  2.9260,  -4.4162,   8.3808],
        [  3.3839,  -4.1966,   8.3574],
        [  4.5340,  -3.3629,   8.9652],
        [  4.7813,  -3.1139,   8.2691],
        [  5.1472,  -2.6801,   8.1194],
        [  5.5626,  -2.0536,   8.4433],
        [ -8.0766,  -6.0046,  -3.1873],
        [  1.6954,  -4.7616,   9.8198],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.5927,   0.4199,  -9.4997],
        [  3.0719, -10.8449,  -4.1673],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.5763,   0.1333,  -9.2650],
        [ -5.3179,  -9.0101,   9.4472],
        [ -9.2707,   0.9661,   4.4863],
        [  5.8201,   3.5841,  -8.0609],
        [  4.9452, -10.3187,  -7.8621],
        [  2.1379,  -4.6757,   6.4981],
        [  5.9912,   3.1453,  -3.9989],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.3715,  -4.7756,   0.2312],
        [  9.2811,  -7.4336,   5.1771],
        [  8.4691,  10.4571,  -5.0650],
        [  4.5430,   5.5438,  -3.8669],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -2.0840,   7.8415,  -7.8657],
        [  2.8474,   6.8979,   6.6457],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.5813,  -4.5471,   5.3597],
        [  1.9312,  13.8649,   8.3563],
        [ -7.3333,   4.9246,  -7.9212],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -0.8885,   7.9160,   0.0571],
        [ 11.7349,  -3.5838,   2.1644],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.0961,   1.6529,   3.1553],
        [ -2.1817,   7.8274,  -4.6298],
        [ 12.6000,   1.3839,  -3.2992],
        [  2.3373,  -4.6232,  -5.2246],
        [  2.5318,  13.7231,   6.3692],
        [ 10.0619,   8.6627,  -9.2884],
        [  7.8375,  -8.7240,  -9.8913],
        [ -7.3554,   4.8985,  -0.6665],
        [  0.5175,  14.0910,   4.6540],
        [  5.8617,   3.4853,  -9.8080],
        [  2.1379,  -4.6757,   6.4981],
        [ -0.8569,   7.9156,  -6.5557],
        [  9.1620,  -7.5573,  -9.9882],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.8631,  -4.3728,  -1.1746],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.8572,  14.0501,  -8.3364],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.6451,  -4.5250,  -0.5565],
        [ -5.1369,   6.7755,   8.3836],
        [  6.2022,   2.4092,   3.0349],
        [  7.0171,  11.6439,  -5.0178],
        [ -3.1671,   7.6158,   6.0765],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.0042,  -2.8350,   2.6864],
        [  5.1196,  -2.7160,   4.2705],
        [  5.5101,  -2.1433,   9.5442],
        [  0.4119,  14.1019,  -4.1870],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.1706,   1.3863,   8.4032],
        [ -4.1296,   7.2803,  -2.7194],
        [ 11.1085,  -4.9204,   0.0736],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.9461,  -4.4077,  -0.4831],
        [  5.9883,   3.1536,   6.2320],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.6310, -10.9195,   3.8054],
        [  3.3943,   6.5483,   1.0790],
        [  8.6271,  -8.0715,  -7.4813],
        [ -4.8751,  -9.3148,   4.0406],
        [ -3.5173,   7.5091,  -8.9585],
        [  2.8890,  -4.4316,   1.1600],
        [  5.9548,  -1.2273,  -2.6520],
        [ -8.5149,  -5.1905,   1.5239],
        [ -1.9646,   7.8571,   3.3817],
        [  5.4242, -10.1261,   3.9595],
        [ -6.6998,   5.5954,   0.0823],
        [ -1.9646,   7.8571,   3.3817],
        [ 12.2218,   4.0953,  -8.2072],
        [  9.4554,   9.4245,   9.8800],
        [  1.5383, -11.0274,  -4.9666],
        [ -2.4956, -10.4682,   3.5632],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -4.8665,  -9.3204,  -5.2564],
        [  4.7544,  -3.1425,  -1.5315],
        [ -5.6651,   6.4380,   1.3660],
        [  6.3122,  -9.6994,  -6.3282],
        [  4.4915,   5.5990,   7.0405],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -8.6112,   2.9319,  -7.9943],
        [ -3.3812, -10.1235,   7.4384],
        [  3.6817,   6.3354,   6.1233],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.9942,  -3.9990,   6.0448],
        [  4.6992,   5.3690,   3.0935],
        [ -8.1531,  -5.8748,   2.6934],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.3587,   1.1596,  -3.6629]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0082, 0.0295,  ..., 0.9780, 1.1219, 0.7616],
        [0.0082, 0.0000, 0.0222,  ..., 0.9860, 1.1274, 0.7621],
        [0.0295, 0.0222, 0.0000,  ..., 1.0071, 1.1488, 0.7751],
        ...,
        [0.9780, 0.9860, 1.0071,  ..., 0.0000, 0.6282, 1.0533],
        [1.1219, 1.1274, 1.1488,  ..., 0.6282, 0.0000, 0.7198],
        [0.7616, 0.7621, 0.7751,  ..., 1.0533, 0.7198, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.29755487,  5.39098669,  5.64500512,  5.70591986,  5.80312949,
        5.92952538, 10.06407839,  5.05444288,  4.78544875, 12.59969936,
       11.27160035, 12.35421461, 12.5769659 , 10.46238054,  9.32094055,
        6.83515201, 11.4424816 ,  5.14124612,  6.76661595,  4.91296205,
        4.79003361, 11.89105259, 13.45642379,  7.16747878,  9.54154096,
        8.1137452 ,  7.46248947, 10.35485331,  5.2287058 , 13.99877413,
        8.83341608,  9.43112957,  7.96575782, 12.26997165, 10.31613269,
       13.05148826,  9.24501829,  8.12580976, 12.67576792,  5.18047638,
       13.95472331, 13.27717982, 11.72749923,  8.83725081, 14.10045797,
        6.81961996,  5.14124612,  7.96181288, 11.87665501,  5.62005629,
        9.88310614, 13.45642379, 12.12641969,  9.95552626, 14.07623266,
       10.67132524,  5.24137909,  8.50268322,  6.65368551, 13.59480906,
        8.24808832, 11.62770332, 12.3344536 ,  5.79543038,  5.91222346,
       14.10796466,  7.89745107,  9.27474386,  8.36995953, 12.1494023 ,
       12.3828957 ,  5.30158497,  6.76791483, 12.07818871, 11.23201619,
        7.37576154, 11.81423161, 10.51340829,  8.29203356,  5.29009836,
        6.07992623,  9.97220799,  8.09899874, 11.48734978,  8.72896381,
        8.09899874, 12.889695  , 13.35013374, 11.1341798 , 10.76154256,
       12.49317563, 10.51437947,  5.69913817,  8.57560275, 11.57249984,
        7.17789666,  9.43112957,  9.09661212, 10.67322527,  7.32754882,
        9.13507321,  9.84314712,  7.13501264, 10.04916678,  8.89458273,
        6.46356806])}
{'P': tensor([[3.0925e-02, 1.6781e-01, 1.4860e-01,  ..., 3.2595e-03, 3.7596e-04,
         8.5848e-05],
        [2.4411e-03, 3.8542e-02, 1.9316e-01,  ..., 2.4064e-03, 7.3205e-04,
         1.3660e-04],
        [4.0966e-03, 1.0856e-02, 7.0296e-02,  ..., 4.2170e-03, 1.3842e-03,
         2.6248e-04],
        ...,
        [8.2830e-03, 1.3316e-02, 1.5749e-02,  ..., 1.0089e-01, 2.7459e-04,
         1.8709e-04],
        [9.3854e-05, 8.7988e-05, 8.5260e-05,  ..., 2.8665e-05, 5.9267e-02,
         1.2338e-02],
        [2.2675e-04, 1.5168e-04, 1.4213e-04,  ..., 1.2878e-04, 1.2843e-04,
         6.9343e-02]]), 'X': tensor([[ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.2349e-02, 3.4340e-02,  ..., 4.1637e-01, 3.3006e-01,
         4.0476e-01],
        [2.2349e-02, 0.0000e+00, 1.5930e-02,  ..., 4.1022e-01, 3.4069e-01,
         4.2655e-01],
        [3.4340e-02, 1.5930e-02, 0.0000e+00,  ..., 3.9553e-01, 3.5651e-01,
         4.3332e-01],
        ...,
        [4.1637e-01, 4.1022e-01, 3.9553e-01,  ..., 1.5170e-04, 7.4404e-01,
         6.0149e-01],
        [3.3006e-01, 3.4069e-01, 3.5651e-01,  ..., 7.4404e-01, 0.0000e+00,
         4.5905e-01],
        [4.0476e-01, 4.2655e-01, 4.3332e-01,  ..., 6.0149e-01, 4.5905e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.30158497,  5.53255882,  5.57341623,  5.54776934,  5.63824971,
        5.60064319, 13.744387  ,  5.63824971,  5.0209005 , 13.6829843 ,
        8.7907923 ,  5.16077764, 11.63383221,  8.95983586,  8.31625171,
        6.97106912,  6.50409372,  7.48311002, 11.72749923,  7.20482224,
        5.29009836, 12.4980845 ,  8.89301516,  8.01009083,  7.5092834 ,
       11.27160035, 10.00410625,  5.51177501,  4.73784222,  6.53290663,
        8.7907923 , 12.77402351, 13.40024224, 13.48809595,  7.34250948,
        7.96181288,  8.92098752, 11.00726483,  6.80370254,  5.87132718,
       13.36092624, 13.84126197,  5.18277858, 10.37998078,  4.74770899,
       10.14044489,  4.74468406,  9.15976746,  9.41970031, 13.89886469,
       11.83075005,  9.95552626,  5.16077764, 13.18278606,  4.91296205,
        7.48311002, 12.17901033, 12.18045978,  7.78980299,  7.57685908,
        9.32094055, 12.52875793, 10.80728788, 12.22276278,  7.62232307,
        9.78286733, 11.22142195,  6.61386436, 12.7125079 ,  8.28968067,
        8.19703512,  5.14124612,  8.12043983,  6.652742  , 13.01187484,
       10.61047596,  7.76457095,  5.89074104,  9.562174  , 14.03618649,
        6.27715105,  8.30125201, 11.05672136, 10.69963954, 13.46553623,
        8.25828409, 13.2778378 , 13.27994304,  8.7907923 ,  6.89245352,
       10.74655346,  5.98285319, 10.1371552 ,  6.42439344, 13.01187484,
       10.39942497,  7.8788965 ,  8.37954853,  7.53884104,  7.92607342,
        9.97822473, 10.82106505,  8.20436764,  5.36061974,  7.54620849,
        8.50278896])}
{'P': tensor([[2.5284e-02, 1.6687e-01, 1.6365e-01,  ..., 1.0876e-03, 6.7107e-05,
         1.3390e-03],
        [3.0611e-03, 3.2297e-02, 2.8127e-01,  ..., 1.0920e-03, 9.4163e-05,
         1.8019e-03],
        [3.9413e-03, 4.6387e-03, 4.8168e-02,  ..., 1.5259e-03, 1.4892e-04,
         2.8265e-03],
        ...,
        [5.5546e-03, 5.9524e-03, 6.3068e-03,  ..., 1.1303e-01, 7.5560e-04,
         4.3349e-02],
        [8.8016e-05, 7.7391e-05, 7.5062e-05,  ..., 2.1119e-04, 8.8640e-02,
         1.4204e-04],
        [3.9150e-04, 4.2317e-04, 4.4579e-04,  ..., 8.0072e-04, 1.4254e-02,
         1.0952e-01]]), 'X': tensor([[  3.1480,  -4.3165,  -7.8002],
        [  3.6154,  -4.0640,  -7.9698],
        [  3.7916,  -3.9524,  -7.9547],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.7156,  -3.1831,  -8.0224],
        [  3.9913,  -3.8139,  -6.8311],
        [ -2.5056,   7.7720,   5.5878],
        [  9.1820,   9.7334,  -2.7932],
        [  7.2340,  -9.1505,  -5.8321],
        [  8.9284,  10.0036,  -6.9210],
        [ -0.0282,   7.8575,  -1.1746],
        [ -1.0407,   7.9166,  -6.6021],
        [  6.7792,  11.8084,   1.6251],
        [  0.9809,   7.6642,  -1.4217],
        [  8.5198,  10.4091,  -8.6486],
        [  6.1730,  -0.5375,  -8.8861],
        [  3.2647, -10.8066,  -0.3768],
        [  4.5325,  -3.3643,   6.7699],
        [ 11.8951,  -3.1570,   7.0742],
        [  5.3804,  -2.3504,   7.2744],
        [  6.1755,   2.5233,  -2.0042],
        [ -0.5385,   7.9037,   6.3830],
        [  5.2939,  -2.4784,  -4.5840],
        [  4.8051, -10.3703,   0.5268],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -3.4986,   7.5152,   9.0712],
        [ 11.7779,   5.5127,  -2.0699],
        [  9.4925,   9.3811,   8.9842],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.3076,   0.7852,  -6.7010],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.1579,  -0.5961,  -6.5595],
        [  6.3606,   0.9204,   3.2525],
        [ -5.1046,  -9.1610,   2.3418],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.8951,  -3.1570,   7.0742],
        [ 10.1983,  -6.3430,  -0.3200],
        [  4.9452, -10.3187,  -7.8621],
        [  9.1620,  -7.5573,  -9.9882],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.2026,  12.1774,  -1.5887],
        [ -6.2200,  -8.2737,   5.5079],
        [ 11.7796,  -3.4696,   9.0088],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -5.2752,  -9.0409,  -8.2568],
        [  9.2811,  -7.4336,   5.1771],
        [ -2.8162,   7.7059,  -9.6885],
        [ 12.1504,   4.3674,  -7.2998],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.8275,  -1.5334,   2.0124],
        [  4.2339,  -3.6268,   6.3125],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.3122,  -9.6994,  -6.3282],
        [  3.4604,  -4.1545,   1.8091],
        [  0.2435,   7.8191,  -6.4709],
        [  6.2038,   2.4020,  -3.1692],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.5267,  -0.4125,  -8.7331],
        [ 11.1420,   6.9331,  -2.4779],
        [ 10.1798,   8.5002,  -5.8893],
        [ -0.8569,   7.9156,  -6.5557],
        [ 11.9641,  -2.9568,   7.7211],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.8400,   2.3991,   1.3457],
        [ -5.6651,   6.4380,   1.3660],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.3900,   3.3727,   3.1650],
        [  1.6954,  -4.7616,   9.8198],
        [ 11.2209,  -4.7092,   2.6117],
        [  7.1697,  11.5343,  -0.7276],
        [ -6.2200,  -8.2737,   5.5079],
        [ -9.4753,  -0.8012,  -5.2050],
        [ 12.4691,   2.8525,  -5.5110],
        [  3.4604,  -4.1545,   1.8091],
        [ -5.5525,   6.5146,  -4.4984],
        [  1.0000,   7.6591,  -4.5011],
        [ -5.1369,   6.7755,   8.3836],
        [ 12.6000,   1.3839,  -3.2992],
        [ -6.6998,   5.5954,   0.0823],
        [ -1.0407,   7.9166,  -6.6021],
        [ 11.4703,  -4.2006,   7.4086],
        [  1.7815,  -4.7481,   4.4289],
        [  5.4429,  -2.2529,   9.7306],
        [ -8.8934,   2.2597,   3.5831],
        [  8.1043,  -8.5164,  -7.9290],
        [  1.2728, -11.0372,   9.8121],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.1178,   1.5784,   2.1410],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.4633,   5.6287,  -0.7754],
        [  5.6095,  -1.9702,  -7.8233],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.0890,  -0.8382,   2.9166],
        [  9.2979,   9.6048,   7.3138],
        [  3.6154,  -4.0640,  -7.9698],
        [  6.3087,   1.8053,  -1.9481],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 11.3367,  -4.4806,   0.4643],
        [  2.9461,  -4.4077,  -0.4831],
        [ -1.6220,   7.8917,   0.0315],
        [  6.3211,   1.7017,  -0.2164]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0138, 0.0193,  ..., 0.3483, 0.4726, 0.3869],
        [0.0138, 0.0000, 0.0057,  ..., 0.3556, 0.4846, 0.3917],
        [0.0193, 0.0057, 0.0000,  ..., 0.3566, 0.4881, 0.3916],
        ...,
        [0.3483, 0.3556, 0.3566,  ..., 0.0000, 0.2376, 0.0831],
        [0.4726, 0.4846, 0.4881,  ..., 0.2376, 0.0000, 0.3075],
        [0.3869, 0.3917, 0.3916,  ..., 0.0831, 0.3075, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.34250225,  5.43943791,  5.47702766,  5.53175275,  5.68942572,
        5.5205189 ,  8.16584537, 13.38091121, 11.66453059, 13.40850328,
        7.8575648 ,  7.98469166, 13.61600536,  7.72668467, 13.45124917,
        6.19633139, 11.28895489,  5.64464807, 12.30695098,  5.87132718,
        6.67108662,  7.92200342,  5.84533345, 11.42947632, 12.30858677,
        8.28968067, 13.00413829, 13.34587027,  9.58072286,  9.34061938,
        5.1224079 ,  6.18668463,  6.42689705, 10.48721382, 10.31613269,
       12.30695098, 12.00995898, 11.4424816 , 11.87665501, 12.1460931 ,
        8.94395628, 13.6660671 , 10.35093485, 12.27993098,  9.70108409,
       10.46739073, 11.89105259,  8.20436764, 12.91143491,  9.3109884 ,
        9.84314712,  6.02588517,  5.5748624 , 12.84056604, 11.57249984,
        5.40689332,  7.822854  ,  6.65259775, 12.42554808, 12.49317563,
       12.53345445, 13.12299145, 13.26209545,  7.96181288, 12.32408516,
        9.96391045,  9.15976746,  8.57560275, 10.24557047,  9.04255655,
        5.05444288, 12.16901615, 13.58101794, 10.35093485,  9.50913849,
       12.79126785,  5.40689332,  8.55982105,  7.72415048,  8.50268322,
       12.67576792,  8.72896381,  7.98469166, 12.21532282,  5.0713364 ,
        5.89074104,  9.17595899, 11.75618513, 11.11038551,  5.79391825,
        9.43737523,  9.25336371, 10.37998078,  7.18354885,  5.94541803,
       13.36092624,  9.5642993 ,  6.1463896 , 13.36800073,  5.43943791,
        6.5618968 ,  9.71392179, 12.18998195,  5.30158497,  8.05668609,
        6.54615887])}
{'P': tensor([[2.5471e-02, 1.5272e-01, 1.4432e-01,  ..., 4.6420e-06, 3.0002e-03,
         5.1281e-03],
        [2.2347e-03, 3.3055e-02, 7.2261e-02,  ..., 5.1072e-06, 2.9986e-03,
         6.0865e-03],
        [2.2452e-03, 1.2805e-02, 3.4078e-02,  ..., 4.9304e-06, 3.1476e-03,
         6.9428e-03],
        ...,
        [2.1853e-04, 1.5633e-04, 1.4121e-04,  ..., 1.0680e-01, 1.1028e-05,
         1.1025e-05],
        [2.3070e-04, 5.0335e-04, 5.5641e-04,  ..., 1.3764e-07, 2.7579e-02,
         3.8486e-03],
        [1.4573e-03, 2.2559e-03, 2.4805e-03,  ..., 1.9213e-06, 1.9262e-03,
         7.2165e-02]]), 'X': tensor([[  3.2226,  -4.2802,  -4.0428],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.1999,  -3.6543,  -4.1998],
        [  4.0642,  -3.7599,  -3.4565],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.1771, -10.9776,  -3.2007],
        [  2.2644,  -4.6434,   9.7508],
        [  1.0000,   7.6591,  -4.5011],
        [ 11.2767,  -4.6006,   0.7868],
        [  3.1892, -10.8220,  -6.8409],
        [ 12.3453,  -1.5462,   7.0212],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.1706,   1.3863,   8.4032],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -1.5408,   7.8978,   5.3470],
        [ -2.7572,   7.7194,   5.9580],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.2608,  12.6977,   4.9166],
        [  9.1409,  -7.5789,   1.5814],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.0233,   4.9647,  -3.1788],
        [ 12.4691,   2.8525,  -5.5110],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.9942,  -3.9990,   6.0448],
        [  6.6319,  11.9066,   3.5165],
        [  4.5503,   5.5359,   7.4735],
        [  3.4140,  -4.1803,  -8.8224],
        [ -7.9925,  -6.1423,   2.8990],
        [  3.5965,  -4.0754,  -3.5035],
        [ -1.5875,   7.8944,  -9.4286],
        [ -2.4956, -10.4682,   3.5632],
        [  9.2692,   9.6370,   7.0746],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.2860,   7.1916,  -6.1011],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -7.6525,  -6.6534,   5.1721],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.7796,  -3.4696,   9.0088],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.7239,   4.4298,   3.1216],
        [  6.2804,   2.0019,   0.8607],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.3515,   0.6418,  -6.5823],
        [  7.0171,  11.6439,  -5.0178],
        [ 12.5717,   1.9266,   7.4903],
        [  9.0129,  -7.7072,   9.1400],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.3767, -11.0341,   9.9415],
        [  2.2437,  -4.6489,   6.1430],
        [  2.0243,   7.3086,   7.5432],
        [  1.9241,   7.3503,   7.9004],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.1472,  -2.6801,   8.1194],
        [  9.5018,  -7.1945,  -1.1892],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.0611,  -6.0302,   4.6962],
        [  5.9843,  -9.8679,  -6.1449],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -6.4002,  -8.1048,  -0.9108],
        [  8.9284,  10.0036,  -6.9210],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.2199,   4.6861,  -1.9304],
        [  3.7237,  -3.9966,  -5.0376],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.0560,  -0.9421,  -5.0334],
        [  1.6422,   7.4586,  -6.8056],
        [  4.4039,   5.6903,   8.7366],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.2608,  -2.5256,  -9.1446],
        [ 12.0042,  -2.8350,   2.6864],
        [  3.2990,  -4.2415,  -7.4857],
        [ -3.9508,  -9.8515,   1.7923],
        [ -5.1046,  -9.1610,   2.3418],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3360,   0.4258,  -9.2509],
        [  1.8719,   7.3713,  -3.7465],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 11.1420,   6.9331,  -2.4779],
        [ -6.4464,   5.8272,   7.5038],
        [  4.2096,   5.8815,   1.9230],
        [ 11.7349,  -3.5838,   2.1644],
        [  4.8674,  12.8875,   6.7622],
        [  6.2247,  -0.3150,  -9.3657],
        [ 11.2517,  -4.6495,   2.9800],
        [ -7.0120,   5.2825,   7.7762],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.6019,   5.9540,  -5.1715],
        [  5.6672,  -1.8631,   8.1020],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -8.9942,  -3.9990,   6.0448],
        [ 11.1469,   6.9237,   1.6160],
        [ -9.1231,  -3.5739,   8.4708],
        [ 10.4994,  -5.9193,  -7.5153],
        [  5.6314,  -1.9302,  -9.6543]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0176, 0.0216,  ..., 1.1980, 0.2905, 0.1872],
        [0.0176, 0.0000, 0.0092,  ..., 1.2103, 0.2927, 0.1815],
        [0.0216, 0.0092, 0.0000,  ..., 1.2181, 0.2842, 0.1723],
        ...,
        [1.1980, 1.2103, 1.2181,  ..., 0.0000, 1.4048, 1.3732],
        [0.2905, 0.2927, 0.2842,  ..., 1.4048, 0.0000, 0.1503],
        [0.1872, 0.1815, 0.1723,  ..., 1.3732, 0.1503, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.35774573,  5.5513072 ,  5.5671389 ,  5.5366698 ,  5.70405525,
        5.79391825, 11.1913598 ,  5.16611782,  7.72415048, 12.17901033,
       11.28215429, 12.44177641, 11.70168755, 13.03002338,  9.41244865,
        9.27474386, 10.30972193,  8.04664886,  8.19703512,  8.8353674 ,
       13.744387  , 11.87412439, 11.8159254 ,  9.85330646,  9.84314712,
        7.06272256, 12.79126785, 13.41713661,  9.84314712, 13.62896593,
        7.16598273,  5.39723241, 10.08001894,  5.43544626,  8.05242591,
       10.76154256, 13.37121923,  5.84450127,  7.54620849, 12.4013153 ,
       10.14044489, 12.10942412, 12.27993098,  8.83341608,  8.90405823,
        6.59176018, 11.83075005,  9.17595899,  9.41605684,  6.383897  ,
       13.59480906, 12.71843462, 11.85890664, 12.07818871, 12.32545637,
       11.11969815,  5.16205087,  7.58378655,  7.59795445, 13.41713661,
        9.11473391,  5.80312949, 11.9182956 ,  6.1561646 ,  5.91677188,
       10.06703358, 11.54071859, 12.10942412, 10.32715715, 13.40850328,
        5.53175275,  7.01475376,  5.46244988, 10.37698435,  6.12886024,
        7.63726134,  7.19534146, 12.10942412,  5.83560009, 12.3344536 ,
        5.37343173, 10.61417181, 10.48721382, 11.33837396,  6.35029274,
        7.60529892,  9.69792513, 13.12299145,  8.68979052,  7.23276582,
       12.26997165, 13.77604293,  6.23262289, 12.17451685,  8.77913428,
        8.04718433, 13.04051744,  5.96556248, 10.99326577,  9.69792513,
        9.78147542,  9.84314712, 13.12218013,  9.79814997, 12.05300943,
        5.95297459])}
{'P': tensor([[4.7244e-02, 2.6391e-01, 1.3181e-01,  ..., 1.6801e-04, 6.4042e-03,
         3.5987e-05],
        [9.6213e-03, 6.1234e-02, 1.7111e-01,  ..., 2.0134e-04, 8.3347e-03,
         4.5408e-05],
        [4.7293e-03, 5.0495e-03, 1.1742e-01,  ..., 1.8821e-04, 4.4784e-02,
         8.2646e-05],
        ...,
        [1.1137e-04, 1.1267e-04, 3.2282e-04,  ..., 2.3559e-02, 1.0561e-03,
         8.2692e-07],
        [3.1118e-04, 3.1544e-04, 7.0408e-04,  ..., 2.3380e-05, 5.5927e-02,
         7.9334e-05],
        [1.7109e-03, 1.5884e-03, 6.3127e-04,  ..., 4.1984e-04, 1.4091e-04,
         1.5028e-01]]), 'X': tensor([[ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 7.6174e-03, 7.4665e-02,  ..., 9.0270e-01, 3.7365e-01,
         9.2234e-01],
        [7.6174e-03, 0.0000e+00, 7.9366e-02,  ..., 9.1012e-01, 3.7969e-01,
         9.2836e-01],
        [7.4665e-02, 7.9366e-02, 0.0000e+00,  ..., 8.4800e-01, 3.0269e-01,
         9.2802e-01],
        ...,
        [9.0270e-01, 9.1012e-01, 8.4800e-01,  ..., 0.0000e+00, 5.8577e-01,
         7.8289e-01],
        [3.7365e-01, 3.7969e-01, 3.0269e-01,  ..., 5.8577e-01, 0.0000e+00,
         9.0621e-01],
        [9.2234e-01, 9.2836e-01, 9.2802e-01,  ..., 7.8289e-01, 9.0621e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.36061974,  5.38342077,  5.79543038,  5.79562691,  5.88782913,
        5.5748624 , 13.66991027, 10.53368644,  5.21280156, 11.51811603,
        8.08320235, 13.55868684, 13.05148826,  5.29009836,  9.69629   ,
        7.06169391,  7.48538905, 11.73397309,  5.69913817, 10.54658597,
        9.27474386,  7.92644746,  5.84533345,  6.81084724,  6.02588517,
        9.43737523, 13.48809595,  5.90035324, 13.08414689,  9.36378438,
        8.67227386, 13.35013374,  4.85995002, 14.03505624,  9.63742669,
       14.10045797, 13.1399181 , 11.86239961,  8.68832398, 13.68672855,
       11.92133129, 12.1494023 , 13.04051744,  6.65991877, 13.35832725,
       12.5769659 ,  9.562174  ,  5.41348121,  8.35856487, 11.51557166,
        9.62238828,  5.08910491,  8.04483495, 13.20988996,  6.83515201,
       11.62158554, 13.1399181 , 10.30972193,  7.05627461,  7.59795445,
        5.18277858,  7.50035355,  8.74470218,  8.5724042 ,  9.88571713,
        9.96391045,  8.66273171,  8.85572292, 11.27160035,  6.33517349,
       13.13205357,  4.74468406, 12.21532282, 10.53368644,  4.9160962 ,
        6.89245352,  5.13616224,  6.76661595, 10.67322527, 12.24467041,
        9.36093562,  9.77588801,  4.78258092,  8.40445872,  6.61211224,
       11.1913598 ,  4.76969364,  7.16747878,  7.53063885, 11.63383221,
       13.61600536, 13.59673406, 11.00726483,  8.81930525, 12.78327037,
        6.74686071, 14.02526505,  5.53255882,  8.67502348,  6.21275016,
       11.75618513,  4.91652556, 12.77402351, 11.57249984,  6.54127431,
        9.55593374])}
{'P': tensor([[3.6063e-02, 1.4938e-01, 1.4410e-01,  ..., 1.6937e-05, 1.7204e-06,
         2.9547e-05],
        [5.9626e-03, 4.6000e-02, 2.7501e-01,  ..., 1.9583e-05, 1.7777e-06,
         3.5896e-05],
        [7.2959e-03, 8.3818e-03, 1.3802e-01,  ..., 6.0913e-05, 4.7407e-06,
         1.2027e-04],
        ...,
        [4.5222e-05, 3.2524e-05, 2.1049e-05,  ..., 2.1140e-02, 2.4088e-03,
         4.9470e-04],
        [1.1162e-05, 8.0376e-06, 5.6172e-06,  ..., 1.1286e-04, 3.7431e-02,
         3.4046e-05],
        [8.6701e-05, 6.3960e-05, 4.2146e-05,  ..., 1.4994e-01, 1.4640e-03,
         3.5330e-02]]), 'X': tensor([[  3.2990,  -4.2415,  -7.4857],
        [  3.7916,  -3.9524,  -7.9547],
        [  4.7156,  -3.1831,  -8.0224],
        [  3.9913,  -3.8139,  -6.8311],
        [  3.6154,  -4.0640,  -7.9698],
        [  4.0421,  -3.7765,  -8.3631],
        [  9.2811,  -7.4336,   5.1771],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.3900,   3.3727,   3.1650],
        [  2.9410,   6.8428,   5.4458],
        [ -8.4267,  -5.3697,   2.4172],
        [  7.9403,  -8.6455,  -8.7433],
        [  2.4356,  -4.5942,  -5.0808],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.0604,   1.7707,   1.1066],
        [  5.3858,   4.4260,  -9.6769],
        [ -9.4748,  -0.7765,  -5.6695],
        [  9.2692,   9.6370,   7.0746],
        [ 12.3357,   3.6001,   0.7903],
        [  3.6469,   6.3624,  -1.8414],
        [ -0.5739,   7.9056,   0.6382],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.1625,  -4.6697,   7.0658],
        [ -0.5385,   7.9037,   6.3830],
        [  3.4401,  13.4540,   4.6576],
        [  1.4749,  -4.7892,  -0.8089],
        [ -3.9508,  -9.8515,   1.7923],
        [  9.3624,   9.5318,   0.2779],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.7237,  -3.9966,  -5.0376],
        [  1.1350,  14.0104,   0.6321],
        [  9.1820,   9.7334,  -2.7932],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.8823, -10.6558,  -3.8952],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.3550,   0.7129,   0.4917],
        [ -9.2974,   0.8371,   2.4852],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.1575,  12.2046,   3.3257],
        [ -1.6334, -10.7207,   7.1397],
        [  3.5284, -10.7484,   5.3347],
        [  3.8111, -10.6787,  -4.6539],
        [  6.2247,  -0.3150,  -9.3657],
        [ -1.9487,   7.8590,  -3.3330],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.3889,  -4.6082,  -8.7291],
        [ -5.6169,  -8.7844,   8.9057],
        [ -1.8367,   7.8718,  -8.6523],
        [  2.5829,  -4.5466,  -8.4015],
        [  9.4554,   9.4245,   9.8800],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.1222,  -0.7261,   8.8204],
        [  2.3376,  -4.6231,   9.8436],
        [ -5.8389,  -8.6056,  -3.5098],
        [  4.1637,  13.1898,  -1.9189],
        [ 11.7418,   5.6070,  -1.4192],
        [ -3.9456,  -9.8542,   0.7466],
        [ -4.2271,   7.2387,  -1.8316],
        [  2.7685, -10.8982,  -6.8928],
        [  4.4633,   5.6287,  -0.7754],
        [  5.9655,   3.2172,  -7.5953],
        [  6.0054,  -1.0902,   8.3079],
        [ 12.2921,  -1.7902,  -4.0870],
        [  4.7813,  -3.1139,   8.2691],
        [ -1.9600,   7.8577,   6.9121],
        [ -3.4488, -10.0934,   2.5195],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.3653,   3.4184,   9.3454],
        [ -6.1914,  -8.2997,   8.5380],
        [  4.5814, -10.4485,   2.5496],
        [ -2.2188,   7.8218,   8.6064],
        [  4.0457,  -3.7737,  -0.4704],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.9693,  -9.3200,  -8.3822],
        [ -7.9667,   4.0814,  -0.6397],
        [ -8.8835,  -4.3178,   9.3080],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.0348,  -2.8229,   3.5436],
        [ -9.4430,  -0.1897,   4.8383],
        [ 12.6035,   0.7494,   7.1536],
        [  4.5340,  -3.3629,   8.9652],
        [ -9.4721,  -1.3097,  -7.4377],
        [  2.3589,  13.7668,  -3.6501],
        [  8.5716,  10.3596,   6.8818],
        [  1.8320,   7.3871,   5.1452],
        [  6.3497,   1.3680,  -1.9076],
        [  5.1142,  -2.7230,  -4.1966],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 11.7794,   5.5086,  -9.6929],
        [  4.0457,  -3.7737,  -0.4704],
        [  4.5491,  -3.3485,   6.8412],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 12.2218,   4.0953,  -8.2072],
        [ -9.1623,  -3.4288,  -4.9961],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.3604,   0.4855,  -3.0158],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 10.9931,   7.2107,   7.7933],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.4159,  -2.0739,   9.6841],
        [ -7.9667,   4.0814,  -0.6397]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0236, 0.0396,  ..., 0.8829, 1.3061, 0.7213],
        [0.0236, 0.0000, 0.0172,  ..., 0.9058, 1.3297, 0.7433],
        [0.0396, 0.0172, 0.0000,  ..., 0.9225, 1.3447, 0.7604],
        ...,
        [0.8829, 0.9058, 0.9225,  ..., 0.0000, 0.5145, 0.1945],
        [1.3061, 1.3297, 1.3447,  ..., 0.5145, 0.0000, 0.7053],
        [0.7213, 0.7433, 0.7604,  ..., 0.1945, 0.7053, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.37343173,  5.47702766,  5.68942572,  5.5205189 ,  5.43943791,
        5.53175275, 11.89105259, 13.50484539,  9.04255655,  7.44805901,
        9.99211663, 11.73848065,  5.19986797,  9.37296102,  9.23178204,
        6.97106912,  9.50654967, 13.37121923, 12.85032836,  7.33350642,
        7.92644746, 11.72749923,  5.14607014,  7.92200342, 13.88683436,
        5.01114719, 10.61417181, 13.3607379 ,  5.35774573,  5.46244988,
       14.05633441, 13.38091121, 13.08414689,  9.51074763,  9.53015376,
       10.8207297 ,  9.43112957,  6.3949043 ,  9.3349858 ,  5.68942572,
        9.7712321 ,  7.88677538, 13.66991027, 10.84437705, 11.31276109,
       11.33837396,  6.23262289,  8.09704052, 11.97949273, 10.00410625,
        5.19065439, 10.42667093,  8.08320235,  5.22903117, 13.35013374,
       12.12641969,  6.16512856,  5.1805345 , 10.39942497, 13.83139024,
       13.01187484, 10.61472636,  8.38252937, 11.24434374,  7.18354885,
        6.77777606,  6.10360073, 12.4217523 ,  5.70591986,  8.09842704,
       10.66631784,  9.09119351,  9.0368458 , 10.35464916, 11.4088043 ,
        8.1303945 ,  5.53255882,  5.5671389 , 11.63765273,  8.95133961,
        9.87720102, 12.07225469,  5.77219077,  9.44486609, 12.62576026,
        5.64500512,  9.562174  , 13.96746816, 13.44593519,  7.61088125,
        6.49538841,  5.79391825, 10.33712258, 13.00380624,  5.53255882,
        5.64865173,  9.91668182, 12.889695  ,  9.78286733, 11.75618513,
        9.37296102, 12.4217523 , 13.14690137,  9.15976746,  9.64157461,
        8.95133961])}
{'P': tensor([[4.7245e-02, 1.3259e-01, 1.3202e-01,  ..., 2.1201e-03, 8.4233e-04,
         3.9432e-03],
        [2.9494e-03, 6.8729e-02, 3.0338e-02,  ..., 2.7410e-03, 1.2664e-03,
         1.5981e-03],
        [3.5052e-03, 3.6083e-02, 8.1510e-02,  ..., 6.4035e-03, 3.5759e-03,
         1.9323e-03],
        ...,
        [7.9293e-05, 2.2906e-04, 3.2606e-04,  ..., 6.8482e-02, 1.9737e-02,
         5.1896e-05],
        [1.1327e-04, 1.3796e-04, 2.3076e-04,  ..., 2.3898e-04, 7.2153e-02,
         1.5201e-04],
        [1.0987e-01, 5.7021e-02, 1.1904e-01,  ..., 3.7569e-03, 2.1616e-03,
         9.0703e-02]]), 'X': tensor([[  3.3474,  -4.2162,   5.2455],
        [  5.1203,  -2.7151,   6.2110],
        [  5.1196,  -2.7160,   4.2705],
        [  5.4336,  -2.2676,   5.6933],
        [  4.2339,  -3.6268,   6.3125],
        [  5.1886,  -2.6250,   6.4677],
        [ -8.4267,  -5.3697,   2.4172],
        [  1.9204,   7.3518,   8.4781],
        [  1.7815,  -4.7481,   4.4289],
        [  4.0362,  13.2397,  -7.0030],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.1813,  -3.3550,   1.8827],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -3.1671,   7.6158,   6.0765],
        [  9.0004,  -7.7195,   4.6945],
        [ -8.8540,  -4.3971,  -5.2144],
        [  4.0421,  -3.7765,  -8.3631],
        [ -9.4351,  -1.8891,   3.5642],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -6.3306,   5.9272,   5.6935],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -4.7389,   6.9953,   3.7638],
        [ 11.9936,  -2.8677,   6.0614],
        [  4.2339,  -3.6268,   6.3125],
        [  1.8719,   7.3713,  -3.7465],
        [  2.2373,  -4.6506,  -1.7334],
        [ -7.6293,  -6.6859,  -5.5161],
        [  4.6357,   5.4415,   7.9067],
        [  5.9526,   3.2525,  -4.3821],
        [  2.0243,   7.3086,   7.5432],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -1.5451,   7.8975,   7.6104],
        [  1.4150,  13.9647,  -4.0915],
        [  4.0362,  13.2397,  -7.0030],
        [ -8.0588,  -6.0341,  -2.3006],
        [  9.2979,   9.6048,   7.3138],
        [ 11.0891,  -4.9558,  -2.3178],
        [  2.5663,   7.0526,  -6.8769],
        [  1.1350,  14.0104,   0.6321],
        [ -9.4541,  -0.3327,   8.8246],
        [  4.9452, -10.3187,  -7.8621],
        [  7.2340,  -9.1505,  -5.8321],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.4267,  -5.3697,   2.4172],
        [  2.3367,   7.1676,  -9.5263],
        [ -0.0282,   7.8575,  -1.1746],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.4887,  -5.2448,  -0.6585],
        [  9.3836,   9.5075,  -5.9278],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.2975,   0.0947,   5.1791],
        [  2.4932,   7.0903,  -5.4214],
        [  2.2644,  -4.6434,   9.7508],
        [  0.4119,  14.1019,  -4.1870],
        [  5.8617,   3.4853,  -9.8080],
        [  6.1579,  -0.5961,  -6.5595],
        [  9.6816,   9.1536,  -5.8543],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -0.5709,   7.9055,   7.6089],
        [ 10.2470,   8.4054,  -0.1177],
        [  6.3413,   0.4883,  -6.8416],
        [  6.3604,   0.9042,  -6.5901],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.0195,   6.1789,  -2.9496],
        [ 11.5935,   5.9740,  -4.3952],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.4739,  -1.2497,   0.5483],
        [ 12.2589,  -1.9312,   9.8657],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.3236,   0.3011,  -4.6668],
        [ -2.6735, -10.4062,  -1.5911],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.3604,   0.9042,  -6.5901],
        [  6.0560,  -0.9421,  -5.0334],
        [ -7.4808,   4.7462,   2.0266],
        [  4.8051, -10.3703,   0.5268],
        [ 10.9394,   7.3069,  -8.9365],
        [ -4.3963,   7.1629,   4.5263],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.8407,  11.7666,   3.0045],
        [ -7.3446,   4.9113,  -5.9356],
        [ 12.3453,  -1.5462,   7.0212],
        [  7.8375,  -8.7240,  -9.8913],
        [ -5.9938,  -8.4745,   3.9735],
        [  7.9816,  10.8938,  -0.5378],
        [  6.2419,  -0.2313,   4.2440],
        [  7.0171,  11.6439,  -5.0178],
        [  6.3460,   1.4252,   1.2291],
        [ 10.1983,  -6.3430,  -0.3200],
        [  1.3103,  -4.8032,  -0.9300],
        [  2.5375,   7.0676,  -4.3490],
        [  9.4554,   9.4245,   9.8800],
        [ 12.6059,   0.9597,  -6.3264],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.4766,  -1.1116,  -2.1706],
        [  6.3122,  -9.6994,  -6.3282],
        [  7.5930,  -8.9037,   7.2323],
        [  3.6703,   6.3443,  -5.4401],
        [ 12.5123,  -0.5339,  -1.0728],
        [  5.9526,   3.2525,  -4.3821],
        [  1.7232,  -4.7574,   3.1270]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.4973e-02, 7.9366e-02,  ..., 4.9071e-01, 6.5386e-01,
         1.7865e-01],
        [6.4973e-02, 0.0000e+00, 1.3182e-01,  ..., 5.4068e-01, 7.0943e-01,
         2.4234e-01],
        [7.9366e-02, 1.3182e-01, 0.0000e+00,  ..., 4.1190e-01, 5.7785e-01,
         1.2240e-01],
        ...,
        [4.9071e-01, 5.4068e-01, 4.1190e-01,  ..., 0.0000e+00, 1.8790e-01,
         3.4492e-01],
        [6.5386e-01, 7.0943e-01, 5.7785e-01,  ..., 1.8790e-01, 0.0000e+00,
         4.8882e-01],
        [1.7865e-01, 2.4234e-01, 1.2240e-01,  ..., 3.4492e-01, 4.8882e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.38342077,  5.79562691,  5.79543038,  5.88782913,  5.5748624 ,
        5.81481044,  9.99211663,  7.5984809 ,  5.0713364 , 13.84126197,
       11.70016217,  9.7751189 , 12.07818871,  8.24808832, 11.85743425,
        9.88571713,  5.53175275,  9.62238828,  5.05981914,  9.50913849,
        8.67227386,  9.97822473,  8.44940661, 12.33167382,  5.5748624 ,
        7.60529892,  5.16077764, 10.14436912,  7.14837514,  6.78325151,
        7.58378655, 10.54658597,  8.04718433, 14.03618649, 13.84126197,
       10.0674887 , 13.36800073, 12.1460931 ,  7.50500064, 14.05633441,
        9.45995955, 11.4424816 , 11.66453059,  8.92098752, 13.06957425,
        9.99211663,  7.53884104,  7.8575648 ,  5.14999799,  9.97822473,
       13.35832725,  8.29203356,  6.29821673,  7.5158489 ,  5.16611782,
       14.10796466,  6.81961996,  6.18668463, 13.32374056, 10.52357479,
        7.92607342, 13.25336001,  6.36004393,  6.42439344, 12.30695098,
       10.01804223,  8.62631801, 13.04217487, 10.86693708,  9.55593374,
       12.41012046,  6.25586297,  6.33075682, 10.74409834, 12.05300943,
        9.95552626,  6.42439344,  6.12886024,  8.859416  , 11.42947632,
       13.15525803,  8.40445872, 10.33712258, 13.6105595 ,  8.8353674 ,
       12.44177641, 11.72749923, 10.37998078, 13.50484539,  6.24614905,
       13.59480906,  6.50409372, 12.00995898,  4.978704  ,  7.5092834 ,
       13.35013374, 12.64235359,  4.91652556,  5.79391825,  9.54154096,
       11.57249984, 11.70168755,  7.32950385, 12.52372629,  6.78325151,
        5.05989547])}
{'P': tensor([[4.8166e-02, 2.2024e-01, 2.0175e-01,  ..., 1.3606e-03, 2.1914e-03,
         7.4654e-04],
        [2.0669e-03, 4.1901e-02, 1.0477e-02,  ..., 2.9158e-04, 4.1955e-03,
         7.8300e-04],
        [3.7661e-03, 1.2013e-01, 5.7877e-02,  ..., 5.3385e-04, 4.4673e-03,
         8.2276e-04],
        ...,
        [5.4399e-03, 1.0911e-02, 1.1001e-02,  ..., 3.5327e-02, 1.5415e-04,
         9.2142e-04],
        [1.4357e-04, 1.5367e-04, 1.4360e-04,  ..., 7.6314e-06, 5.6311e-02,
         1.6014e-05],
        [2.0380e-05, 4.7529e-05, 3.3709e-05,  ..., 1.2347e-06, 1.4587e-03,
         3.0098e-02]]), 'X': tensor([[  3.3839,  -4.1966,   8.3574],
        [  4.7813,  -3.1139,   8.2691],
        [  4.5340,  -3.3629,   8.9652],
        [  5.1472,  -2.6801,   8.1194],
        [  5.5626,  -2.0536,   8.4433],
        [  5.6022,  -1.9833,   8.3081],
        [ -9.3087,   0.7792,  -7.5587],
        [ 10.2470,   8.4054,  -0.1177],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.1706,   1.3863,   8.4032],
        [  3.3474,  -4.2162,   5.2455],
        [  9.4554,   9.4245,   9.8800],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -5.2752,  -9.0409,  -8.2568],
        [  0.1532,  -4.7422,  -7.0145],
        [  2.6613,   7.0021,   4.5783],
        [ -8.1905,   3.7253,   4.1175],
        [ 10.4758,  -5.9539,   1.3667],
        [  1.4306,  13.9620,  -0.2075],
        [  2.0729,   7.2878,   7.0880],
        [ 11.2209,  -4.7092,   2.6117],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.9003,  12.3550,   2.1917],
        [ -1.5261,   7.8988,   9.3768],
        [ -7.3554,   4.8985,  -0.6665],
        [  4.3459,  -3.5327,   0.3667],
        [  7.5930,  -8.9037,   7.2323],
        [  2.6078, -10.9229,   6.2869],
        [ -4.3963,   7.1629,   4.5263],
        [  2.6613,   7.0021,   4.5783],
        [ -9.2301,  -3.1537,  -2.7477],
        [  0.3354,  -4.7708,  -4.9125],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.0772,  -2.6014,  -7.1832],
        [  4.1999,  -3.6543,  -4.1998],
        [  1.3767, -11.0341,   9.9415],
        [ -3.0004,   7.6607,  -7.2640],
        [ -7.6734,   4.4980,  -5.4248],
        [ -3.0004,   7.6607,  -7.2640],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -2.5683, -10.4433,  -2.9182],
        [  4.0457,  -3.7737,  -0.4704],
        [ -4.4503,   7.1378,   3.3751],
        [  4.6063,  13.0051,  -3.1456],
        [  2.1504,  -4.6726,   8.7908],
        [ -7.5603,   4.6459,   8.2002],
        [ -4.3359,  -9.6431,  -9.0616],
        [  4.3377,   5.7570,   1.3712],
        [ 12.5650,   2.0132,   4.6272],
        [ 12.4597,  -0.9136,  -4.8763],
        [  8.2154,  -8.4262,   8.0171],
        [  8.7715,  -7.9390,   5.2900],
        [  3.2647, -10.8066,  -0.3768],
        [  9.6614,   9.1784,  -1.2087],
        [  0.1382, -11.0073,   1.0026],
        [ -6.4002,  -8.1048,  -0.9108],
        [  0.6933,   7.7336,   2.1179],
        [  2.1824,  -4.6647,   1.1985],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -3.1204, -10.2345,   9.2288],
        [ -5.7347,  -8.6908,   9.2878],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.8318,   3.5567,  -4.7692],
        [  2.5974,   7.0363,  -7.7810],
        [  8.6271,  -8.0715,  -7.4813],
        [ -9.4159,  -2.0739,   9.6841],
        [ 10.7566,   7.6212,  -2.3279],
        [  3.9085,   6.1512,  -2.2560],
        [  7.5309,  11.2617,   7.2656],
        [ 12.5717,   1.9266,   7.4903],
        [ -5.0943,  -9.1682,   0.5073],
        [ -7.2515,   5.0195,   4.6434],
        [  9.8082,  -6.8396,   4.9615],
        [ -3.9700,   7.3453,  -4.1486],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.3331,  -2.6465,  -8.2028],
        [  1.1274,  14.0116,   8.1753],
        [ -8.0330,  -6.0765,   6.7945],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.3038,   7.8092,   0.4164],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -4.7334,   6.9982,  -0.4392],
        [ -7.4807,   4.7463,   2.0616],
        [ -6.2668,   5.9808,   2.5614],
        [  3.9022,  13.2906,   5.5788],
        [  2.6310, -10.9195,   3.8054],
        [  9.9750,  -6.6338,  -1.2440],
        [ 12.5919,   0.4035,  -5.8460],
        [ -3.0008, -10.2827,  -8.9480],
        [ -0.6323,   7.9085,  -5.9544],
        [ -9.4351,  -1.8891,   3.5642],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.1379,  -4.6757,   6.4981],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.4524,  -1.6813,  -6.6862],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.1755,   2.5233,  -2.0042],
        [  3.3094,  13.4969,  -3.6507],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.2895,   0.0405,   5.7652],
        [  9.3560,   9.5391,   9.4943],
        [  5.1519, -10.2387,   9.8191],
        [  3.6817,   6.3354,   6.1233],
        [ 12.5123,  -0.5339,  -1.0728]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.9000e-02, 2.2221e-02,  ..., 4.3195e-01, 1.0470e-01,
         6.2847e-01],
        [1.9000e-02, 2.1453e-04, 1.4627e-02,  ..., 4.4999e-01, 1.1942e-01,
         6.2503e-01],
        [2.2221e-02, 1.4627e-02, 0.0000e+00,  ..., 4.4370e-01, 1.2691e-01,
         6.3966e-01],
        ...,
        [4.3195e-01, 4.4999e-01, 4.4370e-01,  ..., 0.0000e+00, 3.9220e-01,
         9.2340e-01],
        [1.0470e-01, 1.1942e-01, 1.2691e-01,  ..., 3.9220e-01, 0.0000e+00,
         5.8128e-01],
        [6.2847e-01, 6.2503e-01, 6.3966e-01,  ..., 9.2340e-01, 5.8128e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.39098669,  5.70591986,  5.64500512,  5.80312949,  5.92952538,
        5.94293185,  9.34127013, 13.25336001, 13.08414689,  9.27474386,
        5.38342077, 13.35013374,  9.78286733, 10.46739073,  4.74468406,
        7.49076899,  8.99791316, 12.0495521 , 14.03505624,  7.57685908,
       12.16901615, 10.01804223, 13.69162795,  8.04483495,  8.83725081,
        5.60064319, 11.70168755, 11.22993218,  8.40445872,  7.49076899,
        9.75401449,  4.78258092, 12.35421461, 12.35421461,  5.5671389 ,
       11.11969815,  8.22728399,  8.89458273,  8.22728399, 12.3828957 ,
       10.75443106,  5.53255882,  8.41148722, 13.79676654,  5.14371005,
        8.8737556 , 10.57302546,  7.20826219, 12.72524615, 12.49317563,
       11.76830981, 11.83075005, 11.28895489, 13.32613445, 11.00812996,
       10.32715715,  7.76457095,  5.14999799, 11.33837396, 12.4013153 ,
       10.69963954, 10.412299  , 13.01187484,  6.83083863,  7.50035355,
       11.81423161,  9.64157461, 13.18278606,  7.28793663, 13.54774355,
       12.71843462, 10.48840476,  8.81930525, 11.95743558,  8.34949383,
        9.64157461,  9.70108409, 14.05687815, 10.07238463, 12.0495521 ,
        7.81509631,  9.93553424,  8.4486771 ,  8.85939626,  8.66273171,
       13.85158417, 11.23201619, 11.97949273, 12.59840193, 10.71162681,
        7.9337647 ,  9.62238828, 12.3828957 ,  5.14124612, 10.00802627,
        9.3109884 ,  9.60080639, 12.05395056,  6.67108662, 13.89671314,
       11.74901778,  6.28961988, 13.3614598 , 11.46176259,  7.32754882,
       12.52372629])}
{'P': tensor([[4.1512e-02, 2.5555e-01, 1.0843e-01,  ..., 1.1245e-04, 2.1316e-04,
         9.0868e-05],
        [5.5052e-03, 6.4163e-02, 4.5915e-02,  ..., 1.9273e-04, 4.0442e-04,
         1.1185e-04],
        [1.0684e-02, 5.8918e-02, 7.0712e-02,  ..., 1.6905e-04, 4.5011e-04,
         9.6358e-05],
        ...,
        [2.9486e-05, 2.7705e-05, 2.3427e-05,  ..., 6.8220e-02, 5.5095e-04,
         1.5878e-04],
        [1.0879e-04, 9.2628e-05, 9.4230e-05,  ..., 3.2942e-05, 4.9022e-02,
         6.9110e-06],
        [7.7659e-04, 7.6725e-04, 5.8725e-04,  ..., 7.6065e-03, 6.1809e-05,
         3.0171e-01]]), 'X': tensor([[  3.4009,  -4.1874,  -5.9751],
        [  4.0078,  -3.8018,  -6.2168],
        [  3.9913,  -3.8139,  -6.8311],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.3535,   0.6792,  -7.5101],
        [ -1.6334, -10.7207,   7.1397],
        [  5.4167, -10.1293,   6.2156],
        [ -3.5029,   7.5138,   8.1853],
        [  6.3305,   1.6119,   1.1538],
        [ -3.1671,   7.6158,   6.0765],
        [ -0.8455, -10.8853,  -6.7400],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.9193,  -1.3176,   5.1739],
        [  1.7232,  -4.7574,   3.1270],
        [ -6.3489,   5.9117,   6.5161],
        [  1.0000,   7.6591,  -4.5011],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -3.5520,   7.4976,   9.1522],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.5813,  -4.5471,   5.3597],
        [  1.9241,   7.3503,   7.9004],
        [  8.0877,  -8.5297,  -1.1050],
        [  3.6469,   6.3624,  -1.8414],
        [  6.1752,   2.5242,   3.5874],
        [  2.2076,  13.8032,  -1.5099],
        [ -5.3179,  -9.0101,   9.4472],
        [  5.9912,   3.1453,  -3.9989],
        [  4.6231,   5.4556,   8.4973],
        [  2.0547,   7.2957,  -1.9343],
        [  1.9204,   7.3518,   8.4781],
        [ -6.4620,  -8.0449,   6.2719],
        [ -9.4645,  -1.4880,  -2.0226],
        [  0.5600, -11.0321,  -0.8660],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.6202,  -1.9508,   0.7471],
        [ -0.5709,   7.9055,   7.6089],
        [ -9.3616,  -2.4749,   0.1793],
        [  3.0719, -10.8449,  -4.1673],
        [ 12.1504,   4.3674,  -7.2998],
        [  4.6357,   5.4415,   7.9067],
        [  5.2911,  -2.4824,  -3.6877],
        [ -5.6651,   6.4380,   1.3660],
        [ -3.3999, -10.1152,  -1.2088],
        [ -7.4733,   4.7555,   6.2490],
        [  4.9337,  12.8566,  -3.6582],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.0937,  13.8293,  -2.3533],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.9636,   2.0648,   8.1946],
        [  3.7298,  13.3537,  -9.4653],
        [  2.6321,  -4.5296,  -2.5159],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.4267,  -5.3697,   2.4172],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.5600, -11.0321,  -0.8660],
        [ 11.4703,  -4.2006,   7.4086],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.7794,   5.5086,  -9.6929],
        [  6.3119,   1.7797,   9.9186],
        [  1.7122,  -4.7591,  -3.6734],
        [  3.3069,   6.6089,   8.7377],
        [  2.5382,   7.0672,   4.5097],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.5128, -10.9365,   9.9645],
        [ 11.1420,   6.9331,  -2.4779],
        [  5.1861,   4.7361,  -3.7312],
        [  5.8201,   3.5841,  -8.0609],
        [  6.1372,   2.6726,  -6.7885],
        [  1.0000,   7.6591,  -4.5011],
        [  5.9843,  -9.8679,  -6.1449],
        [ -5.6792,  -8.7352,   4.0318],
        [ -9.3331,  -2.6465,  -8.2028],
        [  2.8474,   6.8979,   6.6457],
        [ 12.5927,   0.4199,  -9.4997],
        [  4.0591,   6.0202,  -1.6339],
        [ -7.3470,  -7.0615,   5.0066],
        [  3.7916,  -3.9524,  -7.9547],
        [  1.9312,  13.8649,   8.3563],
        [  2.2076,  13.8032,  -1.5099],
        [  4.7482, -10.3907,   5.9149],
        [ -3.4100,   7.5436,  -7.6980],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.3236,   0.3011,  -4.6668],
        [ -4.7334,   6.9982,  -0.4392],
        [ -8.3717,  -5.4769,  -9.2228],
        [  1.1274,  14.0116,   8.1753],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.1749,  -3.3803,   6.4629],
        [ -3.4100,   7.5436,  -7.6980],
        [  0.1041, -11.0045,   2.4311],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.9337,  12.8566,  -3.6582],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.6603,   2.8250,   5.9181],
        [ -8.1905,   3.7253,   4.1175],
        [  2.8602,   6.8905,   5.8779],
        [ -3.0004,   7.6607,  -7.2640],
        [  2.9260,  -4.4162,   8.3808]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0189, 0.0379,  ..., 0.7780, 0.3587, 0.8469],
        [0.0189, 0.0000, 0.0221,  ..., 0.7906, 0.3713, 0.8572],
        [0.0379, 0.0221, 0.0000,  ..., 0.8126, 0.3710, 0.8788],
        ...,
        [0.7780, 0.7906, 0.8126,  ..., 0.0000, 0.9050, 0.1383],
        [0.3587, 0.3713, 0.3710,  ..., 0.9050, 0.0000, 1.0129],
        [0.8469, 0.8572, 0.8788,  ..., 0.1383, 1.0129, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.39450305,  5.52416368,  5.5205189 ,  5.56809335,  5.9510489 ,
        5.84533345,  6.38968357, 10.84437705, 11.48663767,  8.29022197,
        6.53250937,  8.24808832, 10.91805683,  9.9353921 ,  6.06416957,
        5.05989547,  8.67502348,  7.72415048, 12.18045978,  8.29640613,
        9.78147542,  5.2287058 ,  7.59795445, 11.75438006,  7.33350642,
        6.67121748, 13.97857591, 10.46238054,  6.76661595,  7.15100303,
        7.57946459,  7.5984809 , 10.31886066,  9.58072286, 11.04629471,
       10.92597196,  5.94909709,  7.92607342,  9.68323686, 11.27160035,
       12.91143491,  7.14837514,  5.84450127,  8.57560275, 10.67132524,
        8.8580671 , 13.77075174, 10.53368644, 13.98691426, 13.27994304,
        9.19837792, 13.86480175,  5.23878371,  8.22728399,  6.02588517,
        9.95552626,  9.99211663,  7.96181288, 11.04629471, 12.21532282,
        6.83515201, 13.00380624,  6.55800267,  5.05774295,  7.39004226,
        7.50917551,  8.37954853,  9.9353921 , 11.22142195, 13.12299145,
        7.02326515,  6.83515201,  6.69389431,  7.72415048, 11.54071859,
       10.41909211,  9.70108409,  7.46248947, 12.59969936,  7.26074925,
       10.19036081,  5.47702766, 13.99877413, 13.97857591, 11.42420508,
        8.27853076, 12.12641969,  6.33075682,  8.4486771 , 10.00410625,
       14.05687815, 13.27717982,  5.83560009,  9.34127013,  9.77777346,
        8.27853076, 11.0050379 ,  5.35774573, 13.01728313, 13.77075174,
       13.1399181 ,  9.10945678,  8.99791316,  7.46053281,  8.22728399,
        5.29755487])}
{'P': tensor([[3.3918e-02, 2.5606e-01, 1.6024e-01,  ..., 8.2902e-05, 1.1016e-02,
         4.8317e-04],
        [4.1987e-03, 4.6563e-02, 1.0795e-01,  ..., 1.2278e-04, 1.3471e-02,
         7.9310e-04],
        [4.5478e-03, 1.1547e-02, 4.4923e-02,  ..., 1.2349e-04, 1.6378e-02,
         1.0002e-03],
        ...,
        [1.2571e-03, 1.0807e-03, 9.6961e-04,  ..., 4.0825e-01, 4.4087e-04,
         1.2677e-05],
        [2.4742e-03, 3.0532e-03, 4.0449e-03,  ..., 1.6156e-04, 6.0444e-02,
         6.0874e-03],
        [6.2731e-06, 6.9531e-06, 8.0883e-06,  ..., 8.7644e-05, 1.9815e-05,
         2.2684e-02]]), 'X': tensor([[  3.4140,  -4.1803,  -8.8224],
        [  3.7992,  -3.9474,  -8.9019],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.6314,  -1.9302,  -9.6543],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.1459,   1.4780,  -4.2403],
        [ 11.4495,   6.3026,   6.8257],
        [  7.1441,  11.5529,   2.3232],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.6220,   7.8917,   0.0315],
        [  3.2990,  -4.2415,  -7.4857],
        [ -2.0840,   7.8415,  -7.8657],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.7122,  -4.7591,  -3.6734],
        [ 12.5199,  -0.4711,   3.0436],
        [  0.1532,  -4.7422,  -7.0145],
        [  4.9452, -10.3187,  -7.8621],
        [  4.0591,   6.0202,  -1.6339],
        [ -3.4488, -10.0934,   2.5195],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.2905,   1.9366,   2.1169],
        [  6.3266,   0.3292,   9.3502],
        [  6.2022,   2.4092,   3.0349],
        [  6.2535,  -0.1709,  -9.4102],
        [  4.8432,   5.1968,   1.5006],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.3360,   0.4258,  -9.2509],
        [  1.6232,  -4.7717,  -2.5655],
        [ -7.2515,   5.0195,   4.6434],
        [ -7.0948,   5.1940,   9.8171],
        [ -0.0410,   7.8591,   9.3457],
        [  1.4150,  13.9647,  -4.0915],
        [  5.7963,  -1.6018,   1.8856],
        [ -0.9715, -10.8630,   0.3555],
        [ -3.3812, -10.1235,   7.4384],
        [  9.5261,  -7.1675,   1.4206],
        [ 12.4690,  -0.8528,  -9.3374],
        [  4.9531,   5.0576,   5.6794],
        [ -0.9715, -10.8630,   0.3555],
        [  6.2038,   2.4020,  -3.1692],
        [  1.7000,  -4.7609,   2.7935],
        [  6.9314,  -9.3435,  -1.2729],
        [  8.4691,  10.4571,  -5.0650],
        [  3.3474,  -4.2162,   5.2455],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.2692,  13.5098,  -4.9032],
        [ -3.2736, -10.1703,   3.8712],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.0870,  -4.6877,   9.6823],
        [  4.5340,  -3.3629,   8.9652],
        [  6.3087,   1.8053,  -1.9481],
        [ -8.1204,  -5.9308,   4.0761],
        [  0.1931,  -4.7491,   7.4385],
        [ 11.7779,   5.5127,  -2.0699],
        [ -1.7009,   7.8851,  -8.2437],
        [  5.9003,  12.3550,   2.1917],
        [ 12.4690,  -0.8528,  -9.3374],
        [  2.3261,   7.1726,   6.2522],
        [ 12.5199,  -0.4711,   3.0436],
        [  5.8450,  -9.9356,   3.0767],
        [  2.0729,   7.2878,   7.0880],
        [  0.6363, -11.0348,   6.1414],
        [  3.6249, -10.7255,   5.6605],
        [  5.1196,  -2.7160,   4.2705],
        [  0.1931,  -4.7491,   7.4385],
        [ -5.1974,   6.7396,   3.3484],
        [  5.1787,  -2.6383,  -4.2533],
        [ -7.4808,   4.7462,   2.0266],
        [  0.1382, -11.0073,   1.0026],
        [  5.1292,   4.8183,  -5.7997],
        [  3.0876,   6.7527,  -5.3540],
        [  3.8182,  -3.9347,  -1.7961],
        [ -7.3554,   4.8985,  -0.6665],
        [  3.3943,   6.5483,   1.0790],
        [  8.6420,  -8.0580,   6.5781],
        [ -5.1974,   6.7396,   3.3484],
        [  1.8721,  -4.7322,   0.5524],
        [  6.2675,  12.1379,  -8.9720],
        [  7.1697,  11.5343,  -0.7276],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.2911,  -2.4824,  -3.6877],
        [  8.8474,  10.0868,  -8.8644],
        [  4.9337,  12.8566,  -3.6582],
        [ -7.9667,   4.0814,  -0.6397],
        [  5.8902,  -1.3887,   4.0388],
        [  6.9284,  11.7061,  -0.3732],
        [ -9.3464,   0.5708,  -1.6397],
        [ -6.3252,  -8.1760,  -1.0891],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.8318,   3.5567,  -4.7692],
        [ 11.0891,  -4.9558,  -2.3178],
        [  6.1066,  -0.7799,  -4.1746],
        [  9.2811,  -7.4336,   5.1771],
        [ 12.5224,  -0.4497,   4.8448],
        [ -8.9178,   2.1936,   7.5388],
        [ 11.6019,   5.9540,  -5.1715],
        [  2.1379,  -4.6757,   6.4981],
        [  0.9036,   7.6840,   9.9237],
        [ -7.9667,   4.0814,  -0.6397],
        [ -5.9872,   6.2037,  -9.4477],
        [  5.1787,  -2.6383,  -4.2533],
        [ 11.4743,   6.2478,  -2.1100]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 1.0335e-02, 1.2871e-02,  ..., 5.7765e-01, 1.3989e-01,
         3.4432e-01],
        [1.0335e-02, 0.0000e+00, 5.9006e-03,  ..., 5.8765e-01, 1.4097e-01,
         3.3763e-01],
        [1.2871e-02, 5.9006e-03, 0.0000e+00,  ..., 5.9026e-01, 1.3542e-01,
         3.3244e-01],
        ...,
        [5.7765e-01, 5.8765e-01, 5.9026e-01,  ..., 0.0000e+00, 6.1960e-01,
         8.8396e-01],
        [1.3989e-01, 1.4097e-01, 1.3542e-01,  ..., 6.1960e-01, 0.0000e+00,
         2.6436e-01],
        [3.4432e-01, 3.3763e-01, 3.3244e-01,  ..., 8.8396e-01, 2.6436e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.39723241,  5.47866221,  5.53175275,  5.68942572,  5.83560009,
        5.95297459, 10.61472636,  9.91668182,  9.26455795, 13.06957425,
       13.58333857, 10.76154256,  8.05668609,  5.37343173,  8.1137452 ,
       12.59840193,  5.05774295, 12.52875793,  4.74468406, 11.4424816 ,
        7.26074925, 10.66631784,  6.25586297,  6.58184474,  6.33517349,
        6.65368551,  6.25586297,  7.10378919,  8.16714836,  6.35029274,
        5.04028538,  8.81930525,  8.79284466,  7.85919687, 14.03618649,
        6.01356004, 10.90638066, 10.67322527, 11.92133129, 12.4980845 ,
        7.07902036, 10.90638066,  6.65259775,  5.05534609, 11.63383221,
       13.45642379,  5.38342077,  8.05242591, 13.89974693, 10.68416401,
       10.32715715,  5.13125513,  5.64500512,  6.5618968 , 10.05558893,
        4.75303127, 13.00413829,  8.06643487, 13.69162795, 12.4980845 ,
        7.5403822 , 12.52875793, 11.52733581,  7.57685908, 11.05317597,
       11.32149259,  5.79543038,  4.75303127,  8.51088913,  5.81199422,
        8.859416  , 11.00812996,  7.03733775,  7.42512621,  5.48276038,
        8.83725081,  7.37576154, 11.8159254 ,  8.51088913,  5.08910491,
       13.66051884, 13.58101794,  5.80312949,  9.37989631,  5.84450127,
       13.41713661, 13.77075174,  8.95133961,  6.05165248, 13.60275211,
        9.36378438, 10.33712258,  6.00732193,  6.83083863, 12.1460931 ,
        6.1561646 , 11.89105259, 12.53047489,  9.18358562, 13.04051744,
        5.14124612,  7.73692436,  8.95133961,  8.62162372,  5.81199422,
       13.06498554])}
{'P': tensor([[4.2805e-02, 2.1801e-01, 1.4805e-01,  ..., 1.9581e-03, 5.2236e-04,
         1.7485e-04],
        [2.6128e-03, 3.8237e-02, 1.5305e-01,  ..., 1.0035e-03, 3.4905e-04,
         2.0349e-04],
        [2.4673e-03, 4.8685e-03, 6.3594e-02,  ..., 6.4814e-04, 2.6224e-04,
         5.7364e-04],
        ...,
        [3.2387e-04, 7.9767e-04, 1.1665e-03,  ..., 1.4615e-02, 5.3790e-03,
         3.6610e-06],
        [2.9786e-04, 6.1728e-04, 7.5825e-04,  ..., 3.8264e-03, 5.4822e-02,
         5.2792e-06],
        [8.7028e-05, 6.6617e-05, 6.4427e-05,  ..., 2.3599e-06, 9.0286e-07,
         6.2045e-02]]), 'X': tensor([[ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 2.8488e-02, 4.7742e-02,  ..., 4.4741e-01, 6.5934e-01,
         4.9263e-01],
        [2.8488e-02, 0.0000e+00, 5.6285e-02,  ..., 4.3463e-01, 6.3990e-01,
         5.2103e-01],
        [4.7742e-02, 5.6285e-02, 0.0000e+00,  ..., 4.9054e-01, 6.9538e-01,
         4.8987e-01],
        ...,
        [4.4741e-01, 4.3463e-01, 4.9054e-01,  ..., 0.0000e+00, 2.5848e-01,
         8.2537e-01],
        [6.5934e-01, 6.3990e-01, 6.9538e-01,  ..., 2.5848e-01, 0.0000e+00,
         1.0785e+00],
        [4.9263e-01, 5.2103e-01, 4.8987e-01,  ..., 8.2537e-01, 1.0785e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.40689332,  5.62005629,  5.82345867,  5.80860165,  5.87305005,
        6.01356004,  9.97220799, 11.83570132,  8.90405823,  9.86089579,
       11.83509372,  9.75401449,  7.50917551,  9.79814997, 13.86480175,
       13.98691426,  9.44486609,  7.72668467,  9.50654967, 11.63502679,
       14.02058621,  9.69792513,  7.5092834 , 13.97430191,  5.88782913,
        7.46053281, 13.61801526,  9.3109884 , 11.785566  ,  8.53501777,
       13.13888841, 12.95702647,  8.87007495,  7.70897675,  8.09842704,
        7.48311002, 10.8992524 , 12.87752052, 11.4424816 ,  8.16714836,
       12.19805394, 14.02058621, 12.04924159,  6.53250937,  9.58072286,
       13.37121923,  8.8353674 ,  9.77588801, 10.7506423 , 10.07238463,
        7.83534987,  5.05774295,  9.62958973,  7.07902036,  9.18690018,
       10.35485331, 13.32645552,  9.3349858 ,  6.05165248,  9.14973319,
        6.39438044, 12.05395056,  5.56809335,  9.48337063, 13.96987806,
       12.59840193,  7.58378655, 12.85032836,  9.24501829,  6.10774544,
        8.10886021, 11.63502679,  5.88782913,  9.91668182, 12.77402351,
        5.89074104,  4.91336994,  9.85330646,  4.83371341, 11.99633819,
        9.14044984,  8.36445758, 10.06407839,  7.822854  ,  4.74770899,
       10.27572673,  7.32233733, 12.30858677,  5.35774573,  4.87083348,
        9.58347456, 13.35013374, 11.07637531, 10.54658597, 10.1979669 ,
        9.0368458 ,  7.66377554,  8.55982105, 13.58333857,  9.49511524,
        8.99791316, 12.1460931 , 12.30695098, 11.80092578, 11.86239961,
        8.12043983])}
{'P': tensor([[3.3059e-02, 2.3966e-01, 1.5441e-01,  ..., 3.3992e-05, 1.0565e-03,
         7.6205e-04],
        [3.9580e-03, 4.9927e-02, 2.9063e-01,  ..., 3.8592e-05, 1.4009e-03,
         1.1126e-03],
        [4.4330e-03, 6.0712e-03, 7.9266e-02,  ..., 4.5181e-05, 2.0052e-03,
         1.7005e-03],
        ...,
        [3.3849e-04, 3.5494e-04, 3.9833e-04,  ..., 3.6098e-02, 7.8413e-03,
         7.2155e-03],
        [2.1844e-03, 2.6744e-03, 3.9288e-03,  ..., 3.2132e-04, 1.1705e-01,
         2.1567e-02],
        [1.7377e-05, 2.4882e-05, 4.5014e-05,  ..., 5.8988e-07, 1.0377e-04,
         8.0867e-03]]), 'X': tensor([[ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0135, 0.0219,  ..., 0.9445, 0.4233, 0.4048],
        [0.0135, 0.0000, 0.0095,  ..., 0.9555, 0.4267, 0.4087],
        [0.0219, 0.0095, 0.0000,  ..., 0.9585, 0.4233, 0.4056],
        ...,
        [0.9445, 0.9555, 0.9585,  ..., 0.0000, 0.6856, 0.6793],
        [0.4233, 0.4267, 0.4233,  ..., 0.6856, 0.0000, 0.0248],
        [0.4048, 0.4087, 0.4056,  ..., 0.6793, 0.0248, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.41348121,  5.53175275,  5.68942572,  5.47702766,  5.83560009,
        5.47866221,  6.54615887,  6.21275016, 13.20597932,  5.91222346,
       13.89264822, 11.51557166, 10.39942497, 10.04916678,  6.29821673,
        5.91408504,  7.39004226, 10.52587243,  5.1805345 ,  8.1303945 ,
       11.32149259, 11.57249984,  5.5366698 ,  9.20340031, 12.3489324 ,
        9.78285424, 13.25336001,  9.17905491,  6.27905292, 13.00413829,
       13.48809595,  9.7712321 ,  6.67121748, 13.35927527, 11.04953767,
        4.91038884, 12.40074901,  9.97220799,  4.78544875,  4.78002665,
       13.83139024,  6.1463896 ,  5.24137909, 13.97430191, 13.01728313,
        8.68832398,  7.27340834, 11.67875855,  8.30125201, 13.39372426,
       11.80092578,  9.58347456, 14.03618649,  9.562174  , 11.52733581,
        6.7166297 ,  5.83560009,  5.64464807,  7.62232307,  8.38252937,
       10.48315436, 13.96746816,  9.69792513,  5.70405525,  6.34059021,
        6.10360073,  7.17789666,  6.81084724,  5.24461068,  5.13182506,
        6.81084724,  4.978704  ,  7.39004226, 13.26590071,  5.87132718,
       12.3344536 , 13.89974693, 10.33958639, 11.9120321 ,  5.5205189 ,
        5.86519525,  8.08320235,  9.48969978,  8.99791316, 10.00802627,
        6.41202986,  4.85995002,  5.5748624 ,  8.83725081, 13.14690137,
       10.57302546, 11.63765273,  8.04483495, 12.77402351, 12.52875793,
       13.6105595 , 11.63383221,  7.73692436,  7.88677538,  9.76896904,
       13.99877413, 10.7506423 , 11.00726483, 11.04953767,  5.60064319,
       12.22276278])}
{'P': tensor([[2.7335e-02, 2.0635e-01, 1.2046e-01,  ..., 7.8493e-04, 9.7058e-03,
         6.5112e-04],
        [2.9338e-03, 3.5662e-02, 4.0616e-02,  ..., 1.3705e-03, 1.5317e-02,
         8.3174e-04],
        [3.3385e-03, 1.6235e-02, 2.8971e-02,  ..., 1.2186e-03, 1.0080e-02,
         5.2632e-04],
        ...,
        [2.9727e-06, 3.3973e-06, 3.5410e-06,  ..., 8.1091e-03, 2.4158e-05,
         1.8279e-06],
        [1.3513e-03, 1.6078e-03, 1.5709e-03,  ..., 9.6857e-03, 1.3257e-01,
         2.8664e-03],
        [1.0755e-03, 1.2842e-03, 1.0860e-03,  ..., 9.0877e-04, 2.6377e-02,
         1.6174e-01]]), 'X': tensor([[  3.5965,  -4.0754,  -3.5035],
        [  4.0642,  -3.7599,  -3.4565],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.1142,  -2.7230,  -4.1966],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.2026,  12.1774,  -1.5887],
        [  2.5636, -10.9293,   9.6519],
        [  2.9260,  -4.4162,   8.3808],
        [  2.2373,  -4.6506,  -1.7334],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.4604,  -4.1545,   1.8091],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.0233,   4.9647,  -3.1788],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.2419,  -0.2313,   4.2440],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 11.3800,  -4.3919,   1.4066],
        [ -7.4602,   4.7718,  -4.6277],
        [  0.4119,  14.1019,  -4.1870],
        [  5.1787,  -2.6383,  -4.2533],
        [  3.2366,  -4.2732,   5.1532],
        [  1.7659,   7.4127,   4.7871],
        [ -9.0574,   1.7801,   8.8986],
        [  8.0877,  -8.5297,  -1.1050],
        [  1.4306,  13.9620,  -0.2075],
        [  2.5375,   7.0676,  -4.3490],
        [  5.5158,  -2.1337,   3.5246],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -3.5029,   7.5138,   8.1853],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.0582,  -2.6640,   0.7132],
        [  9.9058,  -6.7203,   5.6928],
        [ -6.7992,   5.4991,   9.9186],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.8540,  -4.3971,  -5.2144],
        [  8.1528,  10.7452,   9.1493],
        [  6.0047,   3.1062,  -6.7914],
        [  6.3587,   1.1596,  -3.6629],
        [  6.0310,  12.2795,  -6.4528],
        [  1.1350,  14.0104,   0.6321],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 11.7418,   5.6070,  -1.4192],
        [  0.9478,  -4.8144,   0.6371],
        [  4.5767,   5.5071,   3.9069],
        [ -2.8162,   7.7059,  -9.6885],
        [ -7.2515,   5.0195,   4.6434],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.9693,  -9.3200,  -8.3822],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.0031,  12.2958,  -6.5992],
        [  0.1041, -11.0045,   2.4311],
        [ -9.4739,  -1.2497,   0.5483],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.0965,  -5.9712,   4.3466],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.7544,   2.6096,   3.3146],
        [  7.5064,  -8.9651,  -0.0994],
        [  2.6078, -10.9229,   6.2869],
        [  1.0000,   7.6591,  -4.5011],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -5.1974,   6.7396,   3.3484],
        [  6.3516,   0.6425,   0.3425],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.3062,  -5.6008,   0.0287],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.4867,  12.5816,   2.9549],
        [ 12.2435,   4.0076,   8.1257],
        [ -9.4501,  -1.7133,   2.2980],
        [  3.4401,  13.4540,   4.6576],
        [  6.3497,   1.3680,  -1.9076],
        [  5.4167, -10.1293,   6.2156],
        [ -0.8569,   7.9156,  -6.5557],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -9.4203,  -2.0340,   2.5856],
        [ -0.8885,   7.9160,   0.0571],
        [  1.5653,  13.9377,   4.1773],
        [ -3.1204, -10.2345,   9.2288],
        [ -6.6998,   5.5954,   0.0823],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.2437,  -4.6489,   6.1430],
        [  2.9461,  -4.4077,  -0.4831],
        [ 11.1469,   6.9237,   1.6160],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.2770,  -0.0379,   2.4195],
        [ 12.2311,  -2.0445,   1.3545],
        [  3.3069,   6.6089,   8.7377],
        [  2.5375,   7.0676,  -4.3490],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.4196,   0.0478,   9.4140],
        [  6.3031,   1.8477,  -7.1706],
        [  7.7794,  11.0629,   8.9069],
        [  4.6231,   5.4556,   8.4973],
        [  6.2770,  -0.0379,   2.4195],
        [ -9.3076,   0.7852,  -6.7010],
        [ 10.6283,  -5.7253,   7.4878],
        [  8.8474,  10.0868,  -8.8644],
        [  1.9204,   7.3518,   8.4781],
        [ -8.7544,   2.6096,   3.3146],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.3516,   0.6425,   0.3425],
        [  5.6093,  -1.9705,   6.4807]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0105, 0.0261,  ..., 0.3994, 0.2679, 0.6817],
        [0.0105, 0.0000, 0.0246,  ..., 0.3924, 0.2642, 0.6790],
        [0.0261, 0.0246, 0.0000,  ..., 0.3755, 0.2880, 0.7032],
        ...,
        [0.3994, 0.3924, 0.3755,  ..., 0.0000, 0.5679, 0.9545],
        [0.2679, 0.2642, 0.2880,  ..., 0.5679, 0.0000, 0.4166],
        [0.6817, 0.6790, 0.7032,  ..., 0.9545, 0.4166, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.43544626,  5.5366698 ,  5.5513072 ,  5.70405525,  5.84450127,
        5.79391825, 12.53047489, 13.6660671 , 11.22596994,  5.29755487,
        5.16077764, 12.19805394,  5.40689332,  9.9353921 ,  7.06272256,
        8.91551231,  6.24614905, 12.49291768, 12.19805394,  8.85572292,
       14.10796466,  5.81199422,  5.36061974,  7.62011734,  9.23071698,
       11.75438006, 14.03505624,  7.5092834 ,  5.91408504,  9.4997987 ,
        9.69629   ,  8.29022197,  5.81199422, 12.3489324 , 11.9702762 ,
        8.74470218,  8.8737556 ,  9.88571713, 13.48809595,  6.76056625,
        6.46356806, 13.68063098, 14.05633441,  9.61942445, 13.01187484,
        4.90676242,  7.16058587,  8.20436764,  8.81930525,  5.43544626,
       11.63765273,  8.4486771 , 13.6829843 , 11.0050379 ,  9.55593374,
       13.27994304, 10.06023665,  7.85919687,  9.13507321,  9.13507321,
       11.69264243, 11.22993218,  7.72415048, 10.74655346, 10.23946247,
        8.51088913,  6.38399789, 10.61472636, 10.01804223,  8.22728399,
       13.72594022, 12.88270018,  9.60412735, 13.88683436,  6.49538841,
       11.48663767,  7.96181288, 12.49291768,  9.63742669,  7.96575782,
       14.02532758, 10.69963954,  8.72896381,  8.89458273,  5.16205087,
        5.30158497, 13.12218013, 10.06286074,  6.27715105, 12.40074901,
        7.39004226,  7.5092834 ,  5.13125513,  9.41970031,  6.56833864,
       13.52430176,  7.15100303,  6.27715105,  9.34061938, 12.07225469,
       13.41713661,  7.5984809 ,  9.13507321, 13.13888841,  6.38399789,
        5.94535529])}
{'P': tensor([[3.3692e-02, 2.9341e-01, 1.8264e-01,  ..., 4.2948e-04, 1.3670e-03,
         9.3449e-05],
        [4.6609e-03, 4.8399e-02, 2.8935e-01,  ..., 6.9014e-04, 1.8086e-03,
         1.3092e-04],
        [3.9403e-03, 4.2496e-03, 6.9978e-02,  ..., 1.8535e-03, 1.5490e-03,
         1.4921e-04],
        ...,
        [1.0116e-05, 1.0433e-05, 1.3091e-05,  ..., 4.4456e-02, 9.2855e-06,
         4.1294e-06],
        [1.2309e-02, 1.2958e-02, 1.4772e-02,  ..., 6.7969e-04, 1.1017e-01,
         5.9766e-03],
        [2.2713e-04, 2.3673e-04, 2.7561e-04,  ..., 2.7191e-04, 1.8365e-03,
         9.3150e-02]]), 'X': tensor([[  3.6154,  -4.0640,  -7.9698],
        [  3.7916,  -3.9524,  -7.9547],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.3860,  -2.3418,   1.0826],
        [  1.7218,  -4.7576,  -3.3617],
        [  3.9663,   6.1018,  -5.4104],
        [  6.2247,  -0.3150,  -9.3657],
        [  2.5318,  13.7231,   6.3692],
        [ 11.7162,   5.6728,  -8.9372],
        [  7.5784,  -8.9141,   5.8410],
        [ 10.9931,   7.2107,   7.7933],
        [  2.2659,  13.7894,   6.5436],
        [ -0.8675, -10.8815,   6.8561],
        [  8.2422,  -8.4041,  -1.4962],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.4758,  -5.9539,   1.3667],
        [ -0.3432,   7.8900,   7.1805],
        [  1.7815,  -4.7481,   4.4289],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.2883,   0.0325,   5.2692],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.2883,   0.0325,   5.2692],
        [ -3.9508,  -9.8515,   1.7923],
        [ -9.4721,  -1.3097,  -7.4377],
        [  2.2659,  13.7894,   6.5436],
        [  7.8375,  -8.7240,  -9.8913],
        [  3.8111, -10.6787,  -4.6539],
        [  9.9756,  -6.6330,  -3.3429],
        [ -7.4602,   4.7718,  -4.6277],
        [ 12.5650,   2.0132,   4.6272],
        [  8.6420,  -8.0580,   6.5781],
        [  3.9515,  -3.8425,   6.1729],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.0054,  -1.0902,   8.3079],
        [  6.3246,   1.6696,  -0.1544],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.4901,  -0.7052,   1.7315],
        [  9.9750,  -6.6338,  -1.2440],
        [ -6.2200,  -8.2737,   5.5079],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.6260,  -1.9401,   3.0259],
        [ -8.3535,  -5.5118,   9.8399],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.3231,  -4.7691,  -1.6120],
        [  2.1771, -10.9776,  -3.2007],
        [  6.0305,   3.0296,   0.1375],
        [  6.2555,   2.1474,   4.1685],
        [  5.1142,  -2.7230,  -4.1966],
        [  3.1064,  13.5607,   6.7605],
        [ 11.4235,   6.3594,   5.1663],
        [ -9.4157,   0.0821,  -5.0025],
        [  0.4996,   7.7738,  -4.4711],
        [  3.1064,  13.5607,   6.7605],
        [  6.3122,  -9.6994,  -6.3282],
        [ -6.3252,  -8.1760,  -1.0891],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.8572,  14.0501,  -8.3364],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.3589,  13.7668,  -3.6501],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.4167, -10.1293,   6.2156],
        [  5.6523,   3.9443,  -2.9449],
        [  7.0642,  11.6104,  -9.8316],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -1.6275, -10.7221,   3.2376],
        [ -7.4602,   4.7718,  -4.6277],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.6422,   7.4586,  -6.8056],
        [  1.7301,   7.4262,   5.0522],
        [  4.4290,  -3.4596,   1.4181],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.2687,  -2.9794,   0.6436],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -0.5739,   7.9056,   0.6382],
        [  3.7237,  -3.9966,  -5.0376],
        [ 11.9019,  -3.1379,  -3.4534],
        [  5.7485,  -9.9811,  -3.2355],
        [  7.7794,  11.0629,   8.9069],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.8400,   2.3991,   1.3457],
        [  9.2979,   9.6048,   7.3138],
        [  2.8602,   6.8905,   5.8779],
        [  6.9314,  -9.3435,  -1.2729],
        [  8.1528,  10.7452,   9.1493],
        [  3.6249, -10.7255,   5.6605],
        [  6.1737,   2.5305,   3.9049],
        [ -8.0087,   4.0173,   9.1091],
        [  9.2692,   9.6370,   7.0746],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.2419,  -0.2313,   4.2440],
        [  0.3231,  -4.7691,  -1.6120],
        [  9.9756,  -6.6330,  -3.3429],
        [  4.3523, -10.5232,   8.2138],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.4267,  -5.3697,   2.4172],
        [ 12.1504,   4.3674,  -7.2998],
        [  7.0664,  -9.2590,   8.0879],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 10.2712,   8.3707,  -4.9743],
        [  5.9883,   3.1536,   6.2320],
        [  9.3607,   9.5337,  -4.5412],
        [  2.2373,  -4.6506,  -1.7334],
        [  3.9515,  -3.8425,   6.1729]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0057, 0.0227,  ..., 0.2949, 0.2817, 0.8317],
        [0.0057, 0.0000, 0.0172,  ..., 0.2892, 0.2833, 0.8319],
        [0.0227, 0.0172, 0.0000,  ..., 0.2730, 0.2928, 0.8375],
        ...,
        [0.2949, 0.2892, 0.2730,  ..., 0.0000, 0.4501, 0.8793],
        [0.2817, 0.2833, 0.2928,  ..., 0.4501, 0.0000, 0.5613],
        [0.8317, 0.8319, 0.8375,  ..., 0.8793, 0.5613, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.43943791,  5.47702766,  5.68942572,  5.53175275,  5.94541803,
        5.83560009,  5.87305005,  5.05963217,  7.27759233,  6.23262289,
       13.95472331, 13.01728313, 11.70016217, 13.14690137, 13.97430191,
       10.91601655, 11.77124373, 10.38897563, 12.0495521 ,  7.89745107,
        5.0713364 , 13.13888841,  6.28835101, 13.18278606,  6.28835101,
       10.61417181,  9.562174  , 13.97430191, 11.72749923, 11.33837396,
       11.97957425,  8.85572292, 12.72524615, 11.8159254 ,  5.51177501,
       12.91143491,  6.10360073,  6.54127431, 10.794755  , 12.50996828,
       11.97949273, 10.35093485, 12.67576792,  5.95111804, 10.00802627,
        5.19986797,  4.78002665, 11.1913598 ,  6.74872981,  6.61386436,
        5.79391825, 13.91198191, 13.0743399 ,  9.41605684,  7.78980299,
       13.91198191, 11.57249984, 10.33712258, 11.81423161, 14.07623266,
       13.13205357, 13.96746816,  6.12886024, 11.48663767,  6.89245352,
       13.59056613,  9.66977802, 10.84493732,  8.85572292, 10.91805683,
        7.63726134,  7.62509582,  5.62005629, 13.77222752,  9.73579738,
       12.18045978,  7.92644746,  5.46244988, 12.30858677, 11.51811603,
       13.52430176,  9.7712321 ,  9.15976746, 13.36800073,  7.46053281,
       11.63383221, 13.48809595, 11.32149259,  6.67218687,  8.95983586,
       13.37121923, 10.06437598,  6.24614905,  4.78002665, 11.97957425,
       11.38774149,  7.26074925,  9.99211663, 12.91143491, 11.64746882,
       10.27716434, 13.25017211,  6.76791483, 13.36092624,  5.16077764,
        5.51177501])}
{'P': tensor([[3.4417e-02, 2.5497e-01, 1.2770e-01,  ..., 5.2108e-05, 1.2392e-02,
         3.3185e-02],
        [3.2285e-03, 3.9574e-02, 1.9443e-01,  ..., 7.1605e-05, 1.8418e-02,
         3.5780e-02],
        [2.9229e-03, 3.6231e-03, 7.0852e-02,  ..., 2.2376e-04, 5.9454e-02,
         2.7613e-02],
        ...,
        [1.5774e-05, 1.5244e-05, 1.5419e-05,  ..., 4.0079e-02, 1.3248e-05,
         6.5740e-06],
        [6.8693e-04, 7.7090e-04, 1.1571e-03,  ..., 4.8377e-04, 1.0218e-01,
         1.3681e-03],
        [3.6572e-03, 5.0750e-03, 1.5413e-02,  ..., 1.1441e-04, 1.5645e-01,
         9.6009e-02]]), 'X': tensor([[  3.7237,  -3.9966,  -5.0376],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.1787,  -2.6383,  -4.2533],
        [  5.1142,  -2.7230,  -4.1966],
        [  4.7739,  -3.1218,  -4.1188],
        [ -4.4503,   7.1378,   3.3751],
        [  2.1771, -10.9776,  -3.2007],
        [  4.6063,  13.0051,  -3.1456],
        [  1.4306,  13.9620,  -0.2075],
        [ -9.4378,  -1.8606,  -2.5668],
        [  7.1441,  11.5529,   2.3232],
        [  5.9843,  -9.8679,  -6.1449],
        [  8.5198,  10.4091,  -8.6486],
        [ -4.6393,   7.0461,   1.1624],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.4824,  -5.2575,  -0.0795],
        [  7.9816,  10.8938,  -0.5378],
        [  2.1440, -10.9811,   8.5281],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.1624,  -3.4286,   2.0647],
        [ -8.0608,   3.9362,  -5.9567],
        [ 10.0619,   8.6627,  -9.2884],
        [  9.9750,  -6.6338,  -1.2440],
        [ 10.9916,   7.2133,  -0.9179],
        [  1.7122,  -4.7591,  -3.6734],
        [  7.8425,  -8.7202,   6.4530],
        [  9.3624,   9.5318,   0.2779],
        [  6.3087,   1.8053,  -1.9481],
        [  9.3607,   9.5337,  -4.5412],
        [  7.0664,  -9.2590,   8.0879],
        [  0.6757, -11.0361,  -1.3047],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.6992,   5.3690,   3.0935],
        [  5.6260,  -1.9401,   3.0259],
        [  4.3523, -10.5232,   8.2138],
        [  0.1677,  -4.7447,  -6.1529],
        [  8.4691,  10.4571,  -5.0650],
        [ -9.4624,  -0.4633,  -7.3918],
        [  5.8325,  -1.5222,  -2.1341],
        [ -8.5900,   2.9767,   7.4811],
        [ -1.1885, -10.8211,   0.0865],
        [  6.3505,   0.6235,  -8.5849],
        [  1.7228,  -4.7575,   8.3396],
        [  8.7131,  10.2216,   7.7915],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.0955,   2.8210,  -4.3321],
        [  5.3858,   4.4260,  -9.6769],
        [  3.6817,   6.3354,   6.1233],
        [  4.6357,   5.4415,   7.9067],
        [ -1.8367,   7.8718,  -8.6523],
        [ -9.4714,  -1.3301,  -3.1998],
        [  1.3103,  -4.8032,  -0.9300],
        [  4.4633,   5.6287,  -0.7754],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.1231,  -3.5739,   8.4708],
        [ -2.0721,   7.8432,   6.6380],
        [  5.5424,  -2.0885,   7.2487],
        [ -9.0574,   1.7801,   8.8986],
        [ -7.6719,  -6.6260,   7.1468],
        [ -6.0195,   6.1789,  -2.9496],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.6112,   2.9319,  -7.9943],
        [  5.3602,  -2.3808,   7.2098],
        [  1.3339,   7.5627,  -0.9034],
        [ -7.4808,   4.7462,   2.0266],
        [  6.0560,  -0.9421,  -5.0334],
        [ -1.6275, -10.7221,   3.2376],
        [  2.1477,   7.2550,   8.9688],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.7590,  13.6621,   9.4496],
        [ -0.6323,   7.9085,  -5.9544],
        [ -1.6275, -10.7221,   3.2376],
        [ 11.5456,   6.0861,   8.1101],
        [  6.2419,  -0.2313,   4.2440],
        [  2.3261,  13.7749,   5.9121],
        [  3.6469,   6.3624,  -1.8414],
        [  0.6952,  -4.8060,   8.8976],
        [  6.0031,  12.2958,  -6.5992],
        [  2.6613,   7.0021,   4.5783],
        [ 12.2435,   4.0076,   8.1257],
        [ -7.5603,   4.6459,   8.2002],
        [  9.1820,   9.7334,  -2.7932],
        [  4.1537,   5.9340,  -6.0423],
        [ -1.5408,   7.8978,   5.3470],
        [  9.7328,   9.0902,  -0.8917],
        [  0.9809,   7.6642,  -1.4217],
        [  4.1348,   5.9514,  -0.8856],
        [ -8.9178,   2.1936,   7.5388],
        [ 10.2712,   8.3707,  -4.9743],
        [ -9.4157,   0.0821,  -5.0025],
        [ 10.6283,  -5.7253,   7.4878],
        [  9.3624,   9.5318,   0.2779],
        [ 12.0042,  -2.8350,   2.6864],
        [  2.5636, -10.9293,   9.6519],
        [  1.9503,  -4.7172,  -6.2313],
        [  3.7121,   6.3117,   0.6738],
        [  2.1120,  -4.6818,   8.6144],
        [ -8.0534,  -6.0429,   2.4583],
        [ -1.6220,   7.8917,   0.0315],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.1206,  -4.7363,   8.7103],
        [ -0.5385,   7.9037,   6.3830],
        [  6.1929,   2.4501,  -6.9352],
        [  5.6095,  -1.9702,  -7.8233]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0102, 0.0317,  ..., 0.8043, 0.1026, 0.1148],
        [0.0102, 0.0000, 0.0219,  ..., 0.8058, 0.0990, 0.1117],
        [0.0317, 0.0219, 0.0000,  ..., 0.8020, 0.1022, 0.1158],
        ...,
        [0.8043, 0.8058, 0.8020,  ..., 0.0000, 0.9038, 0.9171],
        [0.1026, 0.0990, 0.1022,  ..., 0.9038, 0.0000, 0.0138],
        [0.1148, 0.1117, 0.1158,  ..., 0.9171, 0.0138, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.46244988,  5.56809335,  5.84533345,  5.81199422,  5.79391825,
        5.70405525,  8.41148722, 11.1913598 , 13.79676654, 14.03505624,
        9.61942445, 13.58333857, 11.54071859, 13.45124917,  8.43626114,
        9.50654967,  9.97964013, 13.50484539, 11.18839169, 10.80728788,
        9.78285424,  8.97052115, 13.27717982, 11.97949273, 13.1471313 ,
        5.05774295, 11.72802491, 13.3607379 ,  6.5618968 , 13.36092624,
       11.64746882, 11.05672136, 13.00413829,  7.13501264,  5.95111804,
       11.38774149,  4.74770899, 13.45642379,  9.47369628,  6.02789634,
        9.09119351, 10.88618114,  6.38105015,  5.05981914, 13.43127131,
       13.15525803,  6.7166297 ,  6.97106912,  7.32754882,  7.14837514,
        8.08320235,  9.5642993 ,  4.978704  ,  7.18354885,  9.62958973,
        9.79814997,  8.11227668,  5.92281142,  9.23071698, 10.1371552 ,
        8.62631801, 10.53368644,  6.652742  ,  9.09661212,  5.86519525,
        7.67939546,  8.859416  ,  6.12886024, 10.84493732,  7.56617037,
        7.8788965 , 13.93790235,  7.9337647 , 10.84493732, 13.05148826,
        6.24614905, 13.96987806,  7.33350642,  4.85603886, 13.6829843 ,
        7.49076899, 12.88270018,  8.8737556 , 13.38091121,  7.24326992,
        8.04664886, 13.31764181,  7.72668467,  7.24677664,  9.18358562,
       13.25017211,  9.41605684, 12.07225469, 13.3607379 , 12.3344536 ,
       11.22596994,  5.10444097,  7.32233733,  5.13616224, 10.06850677,
        8.05668609, 12.19805394,  4.73784222,  7.92200342,  6.65991877,
        5.94541803])}
{'P': tensor([[3.8094e-02, 2.2775e-01, 1.2199e-01,  ..., 3.1904e-05, 6.3596e-05,
         2.7227e-06],
        [4.0866e-03, 6.7295e-02, 3.8014e-01,  ..., 5.1035e-05, 8.5500e-05,
         3.4445e-06],
        [5.6156e-03, 8.5406e-03, 1.7329e-01,  ..., 1.2171e-04, 1.7101e-04,
         6.8353e-06],
        ...,
        [1.3571e-04, 1.6250e-04, 2.2484e-04,  ..., 3.5585e-01, 5.7425e-03,
         1.8941e-04],
        [4.6584e-04, 4.9278e-04, 5.5471e-04,  ..., 8.7771e-02, 1.6327e-01,
         4.4256e-04],
        [3.2596e-05, 2.5155e-05, 1.8737e-05,  ..., 9.8517e-04, 7.6146e-03,
         4.3650e-02]]), 'X': tensor([[ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0172, 0.0269,  ..., 0.9381, 0.8373, 1.3192],
        [0.0172, 0.0000, 0.0114,  ..., 0.9429, 0.8435, 1.3340],
        [0.0269, 0.0114, 0.0000,  ..., 0.9383, 0.8398, 1.3375],
        ...,
        [0.9381, 0.9429, 0.9383,  ..., 0.0000, 0.1270, 0.9006],
        [0.8373, 0.8435, 0.8398,  ..., 0.1270, 0.0000, 0.8375],
        [1.3192, 1.3340, 1.3375,  ..., 0.9006, 0.8375, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.47702766,  5.68942572,  5.94541803,  5.53175275,  5.83560009,
        5.95297459,  8.09842704, 13.97430191, 10.6910768 , 14.05633441,
       12.46513646,  6.60752373,  7.05627461,  5.39723241,  8.62260327,
        9.19837792, 14.09357547,  7.46053281,  8.92549752,  6.19707944,
       11.69264243, 11.48734978,  6.58184474, 13.32613445,  7.18354885,
        8.1303945 ,  9.15866607,  8.98765644,  5.92952538,  5.05963217,
        6.67218687,  9.78285424,  9.55720651, 12.52372629,  5.14371005,
       10.91601655,  8.57560275,  7.73692436, 12.58006038, 11.23201619,
       13.93790235,  7.20826219,  7.03733775,  8.05668609, 10.31613269,
        5.14124612, 12.45418236,  7.17789666, 13.97857591,  8.68979052,
       10.51340829, 10.74409834, 12.05300943, 13.08414689,  5.07306274,
       13.00380624, 13.00380624,  5.23878371, 10.06703358,  4.73784222,
       10.48721382,  8.27853076,  4.73784222, 13.89671314,  9.27474386,
        5.13125513,  4.91038884,  9.32094055, 11.97949273, 13.86480175,
        7.60529892,  7.88677538, 11.23201619,  8.92549752,  6.81084724,
        9.32094055,  9.54154096, 12.58006038,  8.05668609, 13.77604293,
       10.14436912, 11.85890664, 11.63765273, 10.04916678, 12.30858677,
        7.54620849,  8.09842704, 13.77075174, 11.4890861 , 10.09979747,
        8.68832398,  8.05242591, 12.17451685, 10.67322527,  9.4997987 ,
        5.68942572, 11.83075005, 10.5575601 , 12.05395056, 10.90638066,
        6.54693987,  7.96181288,  7.02496223,  6.00663314,  5.14124612,
       10.42667093])}
{'P': tensor([[4.1893e-02, 1.7414e-01, 1.4289e-01,  ..., 4.6680e-05, 4.7372e-06,
         9.7956e-06],
        [3.4437e-03, 8.0559e-02, 1.0519e-02,  ..., 2.2080e-04, 1.1638e-05,
         2.6428e-05],
        [4.8745e-03, 1.3530e-01, 5.7871e-02,  ..., 1.2938e-04, 1.0423e-05,
         2.3576e-05],
        ...,
        [2.6984e-05, 2.5138e-05, 2.8210e-05,  ..., 3.2417e-01, 1.0759e-02,
         7.4741e-03],
        [1.5206e-05, 1.0033e-05, 1.6087e-05,  ..., 3.2851e-05, 1.5817e-01,
         1.0273e-03],
        [5.6048e-06, 4.4909e-06, 6.5989e-06,  ..., 1.2547e-04, 3.3297e-01,
         6.2427e-02]]), 'X': tensor([[  3.7992,  -3.9474,  -8.9019],
        [  5.2608,  -2.5256,  -9.1446],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.6095,  -1.9702,  -7.8233],
        [  4.0421,  -3.7765,  -8.3631],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.2895,   0.0405,   5.7652],
        [  4.9337,  12.8566,  -3.6582],
        [  4.5325,  -3.3643,   6.7699],
        [  5.7963,  -1.6018,   1.8856],
        [  4.6357,   5.4415,   7.9067],
        [  5.9883,   3.1536,   6.2320],
        [  7.8982,  -8.6778,   0.0239],
        [  2.3376,  -4.6231,   9.8436],
        [  3.0876,   6.7527,  -5.3540],
        [ -7.9298,   4.1367,  -3.9773],
        [  0.6363, -11.0348,   6.1414],
        [  4.2612, -10.5514,  -6.6268],
        [  3.1756, -10.8247,   6.8882],
        [ -8.1666,   3.7653,   3.6075],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 12.0042,  -2.8350,   2.6864],
        [  3.9913,  -3.8139,  -6.8311],
        [  1.8320,   7.3871,   5.1452],
        [ 12.6059,   0.9597,  -6.3264],
        [ 10.5955,   7.8824,  -3.7006],
        [ -1.7290,   7.8825,  -3.3831],
        [  4.9957,  -2.8705,   6.5978],
        [  7.8375,  -8.7240,  -9.8913],
        [ 12.5996,   1.3971,  -1.0694],
        [  3.4920,  -4.1366,  -8.3955],
        [ 10.7566,   7.6212,  -2.3279],
        [  0.3354,  -4.7708,  -4.9125],
        [ -3.2736, -10.1703,   3.8712],
        [  5.2188,  -2.5839,   2.1833],
        [  1.3403,  -4.8011,  -1.1698],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.1519, -10.2387,   9.8191],
        [ 12.2328,  -2.0376,  -1.1234],
        [  7.7794,  11.0629,   8.9069],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.0039,   7.3173,  -1.5885],
        [  7.5064,  -8.9651,  -0.0994],
        [ -8.1666,   3.7653,   3.6075],
        [  2.3261,   7.1726,   6.2522],
        [ -9.3616,  -2.4749,   0.1793],
        [  4.2275,  -3.6320,  -0.2714],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.3266,   0.3292,   9.3502],
        [  0.5600, -11.0321,  -0.8660],
        [  6.3535,   0.6792,  -7.5101],
        [  3.7298,  13.3537,  -9.4653],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.5636, -10.9293,   9.6519],
        [  5.1203,  -2.7151,   6.2110],
        [ -6.6998,   5.5954,   0.0823],
        [ 10.9931,   7.2107,   7.7933],
        [  5.7218,  -9.9934,   6.5764],
        [  9.3424,  -7.3685,  -3.6807],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.3515,   0.6418,  -6.5823],
        [  5.6022,  -1.9833,   8.3081],
        [  0.5250, -11.0306,   5.5358],
        [ -9.3384,  -2.6161,  -4.7630],
        [  0.9392,  -4.8143,  -3.4649],
        [ -9.4630,  -1.5145,   0.2983],
        [ 11.9019,  -3.1379,  -3.4534],
        [  5.8902,  -1.3887,   4.0388],
        [ -3.5520,   7.4976,   9.1522],
        [  5.9026,   3.3837,  -0.6573],
        [ 11.9808,   4.9340,  -3.3715],
        [ -7.6650,   4.5092,   0.2520],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.3419,   0.5972,  -1.2841],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -1.7290,   7.8825,  -3.3831],
        [ -5.9940,   6.1985,   5.9566],
        [  4.6357,   5.4415,   7.9067],
        [ -0.3432,   7.8900,   7.1805],
        [  0.4996,   7.7738,  -4.4711],
        [  4.6357,   5.4415,   7.9067],
        [  4.7156,  -3.1831,  -8.0224],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.4501,  -1.7133,   2.2980],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.0175,  -8.4541,  -9.8408],
        [  5.1203,  -2.7151,   6.2110],
        [ 11.0275,  -5.0666,   8.4154],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.3124,   0.2057,  -0.4593],
        [ -9.4773,  -1.0024,   5.2599],
        [ 11.9122,   5.1397,  -1.5498],
        [ 10.2470,   8.4054,  -0.1177],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.0042,  -2.8350,   2.6864],
        [ -2.4956, -10.4682,   3.5632],
        [  3.7916,  -3.9524,  -7.9547],
        [  9.7328,   9.0902,  -0.8917],
        [ 12.0042,  -2.8350,   2.6864],
        [ -1.6086,   7.8928,   4.3385],
        [ -6.5295,  -7.9785,  -7.4429],
        [  1.6308,  13.9254,  -2.0984],
        [ -5.3382,   6.6533,   7.8054],
        [ -1.5451,   7.8975,   7.6104]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0252, 0.0142,  ..., 0.2740, 1.0590, 0.9839],
        [0.0252, 0.0000, 0.0229,  ..., 0.2708, 1.0774, 0.9992],
        [0.0142, 0.0229, 0.0000,  ..., 0.2603, 1.0548, 0.9775],
        ...,
        [0.2740, 0.2708, 0.2603,  ..., 0.0000, 0.9345, 0.8194],
        [1.0590, 1.0774, 1.0548,  ..., 0.9345, 0.0000, 0.1889],
        [0.9839, 0.9992, 0.9775,  ..., 0.8194, 0.1889, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.47866221,  5.83560009,  5.68942572,  5.95297459,  5.94541803,
        5.53175275, 10.32715715,  8.20436764,  6.28961988, 13.77075174,
        5.64464807,  6.01356004,  7.14837514,  6.76791483, 11.73397309,
        5.1805345 ,  7.42512621,  8.94395628, 11.05317597, 11.37939695,
       11.28093507,  8.99276564, 12.3828957 , 12.3344536 ,  5.5205189 ,
        7.61088125, 12.64235359, 13.20597932,  8.06990483,  5.76165934,
       11.72749923, 12.67680439,  5.41348121, 13.18278606,  4.78258092,
       10.68416401,  5.82345867,  4.98461983,  9.71392179, 11.46176259,
       12.4013153 , 13.52430176,  4.8763135 ,  7.58668144, 11.69264243,
        8.99276564,  7.5403822 ,  9.68323686,  5.57341623, 11.75618513,
        6.33517349, 11.04629471,  6.38968357, 13.86480175, 11.75807252,
       11.22596994,  5.79562691,  8.72896381, 13.14690137, 11.51557166,
       11.89855267,  9.7751189 ,  6.383897  ,  5.94293185, 11.04313641,
        9.69792513,  4.90505566,  9.58347456, 12.30858677,  6.05165248,
        8.29640613,  6.80370254, 12.95702647,  8.89301516, 12.0495521 ,
        9.36093562, 12.32545637,  8.06990483,  8.62260327,  7.14837514,
        7.89745107,  7.78980299,  7.14837514,  5.68942572, 10.06286074,
        9.60412735,  9.11473391, 10.37698435,  5.79562691, 12.13568598,
        9.37989631,  6.31576264,  9.53015376, 12.97370371, 13.25336001,
       12.59840193, 12.3344536 , 10.76154256,  5.47702766, 13.31764181,
       12.3344536 ,  8.0550286 , 10.30972193, 14.02058621,  8.53014915,
        8.04718433])}
{'P': tensor([[4.9658e-02, 2.7776e-01, 1.3198e-01,  ..., 3.0542e-05, 5.2016e-03,
         4.7194e-05],
        [3.8365e-03, 6.1669e-02, 1.5708e-02,  ..., 4.7768e-05, 6.5162e-03,
         8.3232e-05],
        [8.9268e-03, 1.4412e-01, 7.1763e-02,  ..., 6.1650e-05, 5.2560e-03,
         1.0830e-04],
        ...,
        [5.8742e-05, 4.4212e-05, 5.6904e-05,  ..., 6.6356e-02, 3.5435e-06,
         3.9039e-04],
        [1.4399e-03, 2.0098e-03, 1.5004e-03,  ..., 4.7257e-05, 2.0250e-01,
         6.7740e-05],
        [2.2112e-05, 1.9206e-05, 2.3940e-05,  ..., 2.0779e-01, 2.2434e-06,
         4.7889e-02]]), 'X': tensor([[ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0267, 0.0679,  ..., 0.8532, 0.2995, 0.7772],
        [0.0267, 0.0000, 0.0502,  ..., 0.8541, 0.3112, 0.7728],
        [0.0679, 0.0502, 0.0000,  ..., 0.8131, 0.3614, 0.7267],
        ...,
        [0.8532, 0.8541, 0.8131,  ..., 0.0000, 1.1318, 0.1817],
        [0.2995, 0.3112, 0.3614,  ..., 1.1318, 0.0000, 1.0709],
        [0.7772, 0.7728, 0.7267,  ..., 0.1817, 1.0709, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.48276038,  5.69913817,  5.63824971,  6.02789634,  6.07992623,
        6.10774544,  6.5618968 , 13.39372426,  9.23178204, 12.5769659 ,
       12.26997165, 13.04156075,  7.37902766,  5.39450305, 13.03002338,
       13.13205357,  9.68323686, 11.83509372,  5.39723241, 10.00794468,
       12.40074901,  9.24501829, 10.1371552 ,  7.07902036,  8.67502348,
       13.6922425 ,  7.24326992,  8.92098752,  8.79284466,  4.85995002,
        7.66377554,  8.51088913, 13.59170809, 12.50996828,  8.67502348,
       10.87409005, 13.58101794,  6.28835101, 12.62576026, 10.04916678,
       12.07818871,  6.05165248, 13.99048094,  7.92200342, 11.10066126,
        8.1303945 ,  9.3349858 ,  7.46053281,  8.04718433,  9.36093562,
        8.95133961,  5.92281142,  5.0713364 ,  8.95983586, 13.89974693,
       10.7506423 , 14.02532758,  9.60359158, 11.20100795,  9.77777346,
        5.14124612,  9.79814997,  5.14999799,  9.62958973, 12.25806559,
        8.4486771 ,  9.70108409, 11.74901778, 10.48315436,  7.48311002,
       13.09737053,  5.07306274, 14.05633441,  5.05774295, 12.10942412,
        7.19513806,  8.29203356,  5.14124612, 10.31886066,  6.81117802,
       10.27716434,  9.99145096, 13.79762415, 11.83570132,  7.66377554,
        5.56809335, 12.40074901,  8.7273569 , 10.4977132 , 13.3607379 ,
        5.10446653,  9.24501829,  7.02253235, 11.63502679, 12.42554808,
        5.83560009,  8.55982105, 11.65923396,  7.23276582, 11.87665501,
        5.86519525, 11.48663767,  7.73692436,  8.77913428,  6.38105015,
        8.28968067])}
{'P': tensor([[2.7892e-02, 2.0088e-01, 1.4286e-01,  ..., 2.0414e-04, 5.3555e-05,
         2.9377e-04],
        [3.1934e-03, 3.1709e-02, 1.9116e-01,  ..., 2.5047e-04, 6.3610e-05,
         3.4712e-04],
        [4.5988e-03, 5.2394e-03, 9.2391e-02,  ..., 1.1457e-03, 2.4152e-04,
         1.4298e-03],
        ...,
        [1.6419e-04, 1.5685e-04, 1.5670e-04,  ..., 1.7108e-01, 5.5529e-02,
         1.4289e-03],
        [8.0646e-05, 7.0784e-05, 4.9249e-05,  ..., 5.4740e-05, 3.4232e-02,
         1.7251e-05],
        [1.4324e-04, 1.4387e-04, 1.7782e-04,  ..., 2.0395e-01, 3.6383e-03,
         1.5077e-01]]), 'X': tensor([[  3.9515,  -3.8425,   6.1729],
        [  4.2339,  -3.6268,   6.3125],
        [  5.1203,  -2.7151,   6.2110],
        [  4.9957,  -2.8705,   6.5978],
        [  5.1886,  -2.6250,   6.4677],
        [  4.5325,  -3.3643,   6.7699],
        [ -8.8934,   2.2597,   3.5831],
        [ 11.2517,  -4.6495,   2.9800],
        [ -9.0604,   1.7707,   1.1066],
        [ -0.0410,   7.8591,   9.3457],
        [ 10.9394,   7.3069,  -8.9365],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.2849, -10.9655,  -7.6542],
        [  4.0587, -10.6112,   6.2273],
        [ -9.0604,   1.7707,   1.1066],
        [ -8.1905,   3.7253,   4.1175],
        [  1.3103,  -4.8032,  -0.9300],
        [ 11.0373,   7.1299,  -8.5895],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -1.6086,   7.8928,   4.3385],
        [  3.3474,  -4.2162,   5.2455],
        [  4.1297,  -3.7098,  -3.9838],
        [  0.5600, -11.0321,  -0.8660],
        [ -3.4986,   7.5152,   9.0712],
        [  5.6672,  -1.8631,   8.1020],
        [  0.1931,  -4.7491,   7.4385],
        [  3.5939,   6.4029,  -1.8357],
        [  9.3753,   9.5171,   3.9131],
        [ -1.8788, -10.6567,   3.7068],
        [ -3.1671,   7.6158,   6.0765],
        [  2.5010,  -4.5736,  -0.8989],
        [ -3.1204, -10.2345,   9.2288],
        [ -6.0844,  -8.3955,   0.7705],
        [  2.5974,   7.0363,  -7.7810],
        [ -9.1459,   1.4780,  -4.2403],
        [ -3.5520,   7.4976,   9.1522],
        [  9.6662,   9.1725,   8.1056],
        [ -6.4002,  -8.1048,  -0.9108],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.6681,  -4.8556,   9.2518],
        [ 12.5123,  -0.5339,  -1.0728],
        [  2.4149,  -4.6005,  -4.3339],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.2209,  -4.7092,   2.6117],
        [ 12.5267,  -0.4125,  -8.7331],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.4541,  -0.3327,   8.8246],
        [ -7.7842,   4.3466,  -1.7548],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.2804,   2.0019,   0.8607],
        [  4.0587, -10.6112,   6.2273],
        [ 12.0582,  -2.6640,   0.7132],
        [ -0.5385,   7.9037,   6.3830],
        [ 11.0864,   7.0387,  -0.1238],
        [ -4.8751,  -9.3148,   4.0406],
        [  6.1737,   2.5305,   3.9049],
        [  8.5198,  10.4091,  -8.6486],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4430,  -0.1897,   4.8383],
        [  9.9058,  -6.7203,   5.6928],
        [  1.4489,   7.5256,   4.4529],
        [  0.2732,  -4.7619,  -1.1577],
        [  1.8320,   7.3871,   5.1452],
        [  5.1472,  -2.6801,   8.1194],
        [  9.6662,   9.1725,   8.1056],
        [ -9.0574,   1.7801,   8.8986],
        [  2.1120,  -4.6818,   8.6144],
        [  7.3592,  11.3937,   6.2407],
        [ 10.4994,  -5.9193,  -7.5153],
        [  0.7145,  -4.8071,  -2.3766],
        [  1.6232,  -4.7717,  -2.5655],
        [  0.3715,  -4.7756,   0.2312],
        [  2.2644,  -4.6434,   9.7508],
        [ -3.5173, -10.0623,   4.8280],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -6.2668,   5.9808,   2.5614],
        [  9.6587,   9.1817,  -0.8212],
        [ -6.4620,  -8.0449,   6.2719],
        [ 10.0567,   8.6697,   4.6329],
        [ 11.5198,  -4.0923,   5.5088],
        [ -9.3087,   0.7792,  -7.5587],
        [  9.2811,  -7.4336,   5.1771],
        [ -4.1296,   7.2803,  -2.7194],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -6.7745,  -7.7264,   8.4684],
        [  4.6063,  13.0051,  -3.1456],
        [  0.2480, -11.0153,  -3.0183],
        [  2.5010,  -4.5736,  -0.8989],
        [  6.6319,  11.9066,   3.5165],
        [  8.1528,  10.7452,   9.1493],
        [ -9.4671,  -0.5553,   0.1117],
        [  9.5018,  -7.1945,  -1.1892],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -8.9942,  -3.9990,   6.0448],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.4735,  -1.2620,  -3.6742],
        [  5.7963,  -1.6018,   1.8856],
        [ -0.0410,   7.8591,   9.3457],
        [  9.9058,  -6.7203,   5.6928],
        [  5.6672,  -1.8631,   8.1020],
        [  2.0937,  13.8293,  -2.3533],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.9809,   7.6642,  -1.4217],
        [ -6.6895,   5.6051,   2.7899],
        [  4.8653,   5.1694,  -4.0823]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0088, 0.0200,  ..., 0.5647, 0.6010, 0.7009],
        [0.0088, 0.0000, 0.0167,  ..., 0.5728, 0.6087, 0.7077],
        [0.0200, 0.0167, 0.0000,  ..., 0.5690, 0.6187, 0.6994],
        ...,
        [0.5647, 0.5728, 0.5690,  ..., 0.0000, 0.5038, 0.2255],
        [0.6010, 0.6087, 0.6187,  ..., 0.5038, 0.0000, 0.7282],
        [0.7009, 0.7077, 0.6994,  ..., 0.2255, 0.7282, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.51177501,  5.5748624 ,  5.79562691,  5.76165934,  5.81481044,
        5.64464807,  9.17595899, 12.17451685,  9.23178204,  7.85919687,
       13.15525803, 11.80092578, 11.20100795, 11.36089355,  9.23178204,
        8.99791316,  4.978704  , 13.1399181 , 10.78533838, 12.12448912,
        8.0550286 ,  5.38342077,  5.5513072 , 11.04629471,  8.28968067,
        5.96556248,  4.75303127,  7.34250948, 13.35927527, 10.82106505,
        8.24808832,  5.21280156, 10.69963954, 10.36844813,  7.50035355,
        9.26455795,  8.29640613, 13.32557133, 10.32715715,  6.02588517,
        9.9353921 , 12.52372629,  5.19577146, 13.02818702, 12.16901615,
       12.53345445,  6.19707944,  9.45995955,  8.91551231, 12.9089559 ,
        6.59176018, 11.36089355, 12.3489324 ,  7.92200342, 13.13205357,
       10.51340829,  6.67218687, 13.45124917,  9.41244865,  9.44486609,
       11.9702762 ,  7.66377554,  4.76969364,  7.61088125,  5.80312949,
       13.32557133,  9.23071698,  5.13616224, 13.56367857, 12.05300943,
        4.85995002,  5.04028538,  4.79003361,  5.16611782, 10.65929968,
        9.97964013,  8.66273171, 13.32645552, 10.31886066, 13.2778378 ,
       12.22503754,  9.34127013, 11.89105259,  8.36995953, 12.46513646,
       10.27572673, 13.79676654, 11.018084  ,  5.21280156, 13.62896593,
       13.48809595,  9.48337063, 11.9182956 , 10.53368644,  9.84314712,
        9.562174  ,  9.55720651,  6.01356004,  7.85919687, 11.9702762 ,
        5.96556248, 13.98691426, 12.0495521 ,  7.72668467,  8.7273569 ,
        7.09887557])}
{'P': tensor([[4.6127e-02, 1.4210e-01, 1.3373e-01,  ..., 5.9512e-02, 5.5248e-02,
         6.7589e-04],
        [1.3656e-03, 3.7487e-02, 2.6431e-01,  ..., 7.3059e-03, 1.3474e-01,
         1.5231e-03],
        [1.4201e-03, 4.6775e-03, 4.6129e-02,  ..., 5.9896e-03, 2.0535e-01,
         1.8880e-03],
        ...,
        [4.1689e-03, 8.2693e-02, 9.3460e-02,  ..., 6.2781e-02, 8.9911e-02,
         3.6762e-03],
        [1.3759e-03, 3.0590e-03, 3.3944e-03,  ..., 2.7369e-03, 1.0191e-01,
         5.0206e-03],
        [1.4653e-04, 2.3839e-04, 2.4437e-04,  ..., 2.7980e-04, 3.7403e-04,
         1.2753e-01]]), 'X': tensor([[  3.9913,  -3.8139,  -6.8311],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.6095,  -1.9702,  -7.8233],
        [  4.7156,  -3.1831,  -8.0224],
        [  6.0788,  -0.8710,  -6.8719],
        [  9.9058,  -6.7203,   5.6928],
        [  3.1756, -10.8247,   6.8882],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.4645,  -1.4880,  -2.0226],
        [  3.1064,  13.5607,   6.7605],
        [ -6.1914,  -8.2997,   8.5380],
        [  2.2076,  13.8032,  -1.5099],
        [  6.3119,   1.7797,   9.9186],
        [ -1.3178, -10.7939,   0.4236],
        [  5.5101,  -2.1433,   9.5442],
        [  6.8407,  11.7666,   3.0045],
        [  1.5662,  13.9375,   4.5142],
        [  4.2339,  -3.6268,   6.3125],
        [  0.1041, -11.0045,   2.4311],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.8930,  12.3592,  -8.9201],
        [ 11.7418,   5.6070,  -1.4192],
        [  7.7750,  11.0665,   2.8364],
        [ -9.4366,  -0.1189,  -2.6736],
        [  1.7301,   7.4262,   5.0522],
        [  6.3305,   1.6119,   1.1538],
        [  1.3705,  13.9723,   7.9266],
        [  5.0617,  12.7957,   8.3544],
        [ 10.5678,   7.9261,  -0.6100],
        [  2.2659,  13.7894,   6.5436],
        [  5.9526,   3.2525,  -4.3821],
        [  0.7145,  -4.8071,  -2.3766],
        [  7.9816,  10.8938,  -0.5378],
        [ 11.9695,  -2.9407,  -6.3498],
        [  6.3515,   0.6418,  -6.5823],
        [  2.5318,  13.7231,   6.3692],
        [  2.5469,  13.7192,  -9.6935],
        [  6.3236,   0.3011,  -4.6668],
        [  4.1999,  -3.6543,  -4.1998],
        [  9.7328,   9.0902,  -0.8917],
        [  2.0937,  13.8293,  -2.3533],
        [  6.1973,  -0.4372,   6.5720],
        [ -2.0840,   7.8415,  -7.8657],
        [  2.7685, -10.8982,  -6.8928],
        [ -8.7735,   2.5640,  -6.3313],
        [ -8.1531,  -5.8748,   2.6934],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.4671,  -0.5553,   0.1117],
        [ -8.0611,  -6.0302,   4.6962],
        [  1.3103,  -4.8032,  -0.9300],
        [ -2.1382,   7.8339,   7.3524],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4697,  -0.6157,  -6.8472],
        [  4.3377,   5.7570,   1.3712],
        [  4.3459,  -3.5327,   0.3667],
        [  1.7500,   7.4187,   4.4237],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.3167,   6.6022,   6.3554],
        [  3.1756, -10.8247,   6.8882],
        [ -5.1974,   6.7396,   3.3484],
        [  9.0051,   9.9234,   5.4411],
        [ -8.7735,   2.5640,  -6.3313],
        [  9.0424,  -7.6779,  -7.5286],
        [  9.2979,   9.6048,   7.3138],
        [ -9.2974,   0.8371,   2.4852],
        [  6.5428,  -9.5728,  -2.9897],
        [  2.4356,  -4.5942,  -5.0808],
        [  7.5784,  -8.9141,   5.8410],
        [  3.0719, -10.8449,  -4.1673],
        [  5.3858,   4.4260,  -9.6769],
        [  9.7328,   9.0902,  -0.8917],
        [  3.0719, -10.8449,  -4.1673],
        [ -7.5603,   4.6459,   8.2002],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.8979,   3.3958,  -3.8713],
        [  1.9505,  -4.7171,   9.9868],
        [ 10.4737,  -5.9570,  -1.2108],
        [  6.9653,  -9.3225,  -3.4727],
        [ 12.0845,  -2.5772,  -4.6073],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.4336,  -2.2676,   5.6933],
        [ -2.4956, -10.4682,   3.5632],
        [ -4.7389,   6.9953,   3.7638],
        [ 11.5456,   6.0861,   8.1101],
        [  6.1973,  -0.4372,   6.5720],
        [ -5.9938,  -8.4745,   3.9735],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 10.9717,  -5.1646,  -6.9039],
        [  0.3493,  -4.7727,  -6.5682],
        [ 11.9695,  -2.9407,  -6.3498],
        [  7.8982,  -8.6778,   0.0239],
        [  6.3583,   1.1729,   8.7835],
        [ -7.4733,   4.7555,   6.2490],
        [ 12.2435,   4.0076,   8.1257],
        [  6.3551,   0.7149,   1.3344],
        [ -5.9225,  -8.5355,   2.4778],
        [  6.9693,  -9.3200,  -8.3822],
        [  3.4009,  -4.1874,  -5.9751],
        [  1.3767, -11.0341,   9.9415],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.3413,   0.4883,  -6.8416],
        [  4.2096,   5.8815,   1.9230]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0275, 0.0281,  ..., 0.0771, 0.0396, 0.5105],
        [0.0275, 0.0000, 0.0089,  ..., 0.0620, 0.0307, 0.4981],
        [0.0281, 0.0089, 0.0000,  ..., 0.0705, 0.0218, 0.5065],
        ...,
        [0.0771, 0.0620, 0.0705,  ..., 0.0000, 0.0916, 0.4362],
        [0.0396, 0.0307, 0.0218,  ..., 0.0916, 0.0000, 0.5269],
        [0.5105, 0.4981, 0.5065,  ..., 0.4362, 0.5269, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.5205189 ,  5.9510489 ,  6.00732193,  5.94541803,  5.68942572,
        6.14086582, 11.9702762 , 11.28093507, 13.6660671 ,  9.58072286,
       13.91198191, 10.35464916, 13.97857591,  6.55800267, 10.87409005,
        5.91222346, 13.6105595 , 14.02526505,  5.5748624 , 11.0050379 ,
        8.62260327,  9.68323686, 13.6922425 , 13.01187484, 13.5247215 ,
        9.43737523,  7.62509582,  6.53250937, 14.0393949 , 13.76047999,
       13.20988996, 13.97430191,  6.78325151,  4.85995002, 13.50484539,
       12.32545637,  6.383897  , 13.95472331, 13.95361098,  6.33075682,
        5.5671389 , 13.31764181, 13.98691426,  6.21275016,  8.1137452 ,
       11.24434374,  9.14044984, 10.04916678,  9.99145096, 10.33712258,
        9.48337063, 10.06703358,  4.978704  ,  8.12043983, 13.79676654,
        9.27474386,  9.48969978,  7.20826219,  5.60064319,  7.62232307,
        5.34250225,  7.38844703, 11.28093507,  8.51088913, 13.40024224,
        9.14044984, 11.86239961, 13.36800073,  9.3349858 , 11.59512194,
        5.19986797, 11.70016217, 11.27160035,  6.97106912, 13.31764181,
       11.27160035,  8.8737556 ,  9.69629   ,  6.80559669,  5.10446653,
       12.04924159, 11.63724792, 12.35625694,  4.91296205,  5.88782913,
       10.76154256,  8.44940661, 13.05148826,  6.21275016, 10.37998078,
       12.1460931 , 12.12641969,  4.78544875, 12.32545637, 11.73397309,
        6.46560181,  8.8580671 , 12.88270018,  6.39520397, 10.38897563,
       11.63765273,  5.39450305, 11.11969815,  5.84533345,  6.36004393,
        7.23276582])}
{'P': tensor([[4.4453e-02, 1.5962e-01, 1.3895e-01,  ..., 3.5137e-03, 2.3593e-03,
         1.6010e-04],
        [1.6350e-03, 4.6932e-02, 3.3090e-01,  ..., 6.0788e-04, 7.8746e-03,
         4.4551e-04],
        [1.9442e-03, 6.4937e-03, 6.4041e-02,  ..., 6.2434e-04, 1.2506e-02,
         6.2623e-04],
        ...,
        [2.7606e-03, 1.5494e-02, 1.6935e-02,  ..., 7.2470e-02, 1.0751e-03,
         1.0880e-03],
        [1.8806e-04, 2.1827e-04, 2.2676e-04,  ..., 1.0852e-05, 7.6427e-02,
         1.7318e-04],
        [3.4762e-06, 6.6335e-06, 7.1428e-06,  ..., 3.7772e-07, 8.8865e-04,
         3.3026e-02]]), 'X': tensor([[  4.0078,  -3.8018,  -6.2168],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.2939,  -2.4784,  -4.5840],
        [  0.9445,   7.6736,   0.5095],
        [ -8.4573,   3.2448,   5.4370],
        [ -2.0229, -10.6163,   8.6930],
        [ -5.6651,   6.4380,   1.3660],
        [  6.2895,   0.0405,   5.7652],
        [  2.6078, -10.9229,   6.2869],
        [  1.5662,  13.9375,   4.5142],
        [ -5.8389,  -8.6056,  -3.5098],
        [  3.7121,   6.3117,   0.6738],
        [  2.2659,  13.7894,   6.5436],
        [  5.1886,  -2.6250,   6.4677],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.2804,   2.0019,   0.8607],
        [ 11.3118,   6.5957,  -4.5667],
        [  4.2612, -10.5514,  -6.6268],
        [  4.6063,  13.0051,  -3.1456],
        [  8.3791,  10.5409,   5.5590],
        [  4.7544,  -3.1425,  -1.5315],
        [ 11.0373,   7.1299,  -8.5895],
        [  8.8091,  -7.9037,   0.3631],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.6414, -10.0303,   7.8525],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3516,   0.6425,   0.3425],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.3551,   0.7149,   1.3344],
        [ -1.6086,   7.8928,   4.3385],
        [  7.9403,  -8.6455,  -8.7433],
        [ -0.4341, -10.9473,   7.4509],
        [  3.2701,   6.6338,   9.9209],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.3858,   4.4260,  -9.6769],
        [  1.7000,  -4.7609,   2.7935],
        [ -8.6381,  -4.9238,  -4.4077],
        [  2.3490,  -4.6199,  -4.2489],
        [ -1.7290,   7.8825,  -3.3831],
        [ -1.6334, -10.7207,   7.1397],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.3087,   0.7792,  -7.5587],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.2939,  -2.4784,  -4.5840],
        [ -6.3306,   5.9272,   5.6935],
        [ -8.3900,   3.3727,   3.1650],
        [ -9.2974,   0.8371,   2.4852],
        [  2.4639, -10.9431,  -7.3718],
        [  1.6954,  -4.7616,   9.8198],
        [ -5.9225,  -8.5355,   2.4778],
        [  4.5057,  -3.3894,  -0.8160],
        [  0.3231,  -4.7691,  -1.6120],
        [  3.6817,   6.3354,   6.1233],
        [  2.5469,  13.7192,  -9.6935],
        [ -3.0644,   7.6439,  -1.4322],
        [  0.9392,  -4.8143,  -3.4649],
        [  9.3560,   9.5391,   9.4943],
        [  5.8323,  -1.5228,   5.1132],
        [  5.0277,   4.9589,   7.0175],
        [  3.4604,  -4.1545,   1.8091],
        [  1.0000,   7.6591,  -4.5011],
        [ -7.2515,   5.0195,   4.6434],
        [ -9.3810,  -2.3455,  -9.5194],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.0788,  -0.8710,  -6.8719],
        [ 11.2767,  -4.6006,   0.7868],
        [  1.3403,  -4.8011,  -1.1698],
        [  3.6469,   6.3624,  -1.8414],
        [  4.5767,   5.5071,   3.9069],
        [ -9.4739,  -1.2497,   0.5483],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.6603,   2.8250,   5.9181],
        [  4.1999,  -3.6543,  -4.1998],
        [ 11.9808,   4.9340,  -3.3715],
        [ -9.4501,  -1.7133,   2.2980],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.5954,  13.0099,  -4.4439],
        [ 10.6283,  -5.7253,   7.4878],
        [  1.9505,  -4.7171,   9.9868],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -5.9938,  -8.4745,   3.9735],
        [ -7.5401,   4.6718,   9.0033],
        [ -0.5739,   7.9056,   0.6382],
        [  3.3744,   6.5623,  -6.3636],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.1625,  -4.6697,   7.0658],
        [ -5.2589,   6.7024,  -5.5809],
        [  4.0457,  -3.7737,  -0.4704],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.9526,   3.2525,  -4.3821],
        [ -8.1905,   3.7253,   4.1175],
        [ -1.1885, -10.8211,   0.0865],
        [ -8.0750,  -6.0072,  -5.8267],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.5954,  13.0099,  -4.4439],
        [  5.1886,  -2.6250,   6.4677],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.0876,   6.7527,  -5.3540],
        [  9.3753,   9.5171,   3.9131]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0276, 0.0341,  ..., 0.1563, 0.0830, 0.8449],
        [0.0276, 0.0000, 0.0089,  ..., 0.1836, 0.1104, 0.8407],
        [0.0341, 0.0089, 0.0000,  ..., 0.1887, 0.1160, 0.8465],
        ...,
        [0.1563, 0.1836, 0.1887,  ..., 0.0000, 0.0737, 0.9072],
        [0.0830, 0.1104, 0.1160,  ..., 0.0737, 0.0000, 0.8707],
        [0.8449, 0.8407, 0.8465,  ..., 0.9072, 0.8707, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.52416368,  5.9510489 ,  6.00732193,  6.14086582,  5.94541803,
        5.84533345,  7.73151309,  9.05843201, 10.80728788,  8.57560275,
        6.28961988, 11.22993218, 14.02526505, 10.39942497,  7.32233733,
       13.97430191,  5.81481044,  4.87083348, 12.16901615,  6.59176018,
       13.09426142, 11.37939695, 13.79676654, 13.46553623,  5.69913817,
       13.1399181 , 11.83509372, 12.00995898, 11.50791859, 11.62158554,
        6.38399789,  5.53175275,  6.39520397,  8.0550286 , 11.73848065,
       10.95593729,  7.39599986,  5.18277858,  6.97106912,  5.05534609,
        9.94285511,  5.18277858,  8.06990483, 10.84437705, 11.66453059,
        9.6282763 ,  9.34127013,  9.88571713,  7.88677538,  5.84533345,
        8.67227386,  9.04255655,  9.3349858 , 11.21703654,  5.05444288,
       10.38897563,  5.63824971,  4.78002665,  7.32754882, 13.95361098,
        8.23525433,  4.90505566, 13.3614598 ,  6.02779201,  7.06169391,
        5.40689332,  7.72415048,  8.81930525,  9.66977802,  5.70405525,
        6.14086582, 12.17901033,  4.98461983,  7.33350642,  7.16058587,
        9.55593374, 13.08414689,  9.10945678,  5.5671389 , 12.95702647,
        9.60412735, 13.01728313, 13.79762415, 12.07225469,  5.10446653,
       12.52372629, 10.37998078,  8.87007495,  7.92644746,  7.37902766,
       10.52587243,  8.8353674 ,  5.14607014,  8.51928531,  5.53255882,
       12.33167382,  6.78325151,  8.99791316, 10.88618114, 10.06437598,
       12.05300943, 13.79762415,  5.81481044, 11.89855267,  7.42512621,
       13.35927527])}
{'P': tensor([[4.3942e-02, 2.5578e-01, 1.6430e-01,  ..., 6.5392e-05, 6.3236e-04,
         9.5995e-04],
        [3.9082e-03, 5.1026e-02, 1.1930e-01,  ..., 8.0304e-05, 1.1061e-03,
         1.8336e-03],
        [2.8753e-03, 7.0207e-03, 5.3768e-02,  ..., 4.6443e-05, 1.1497e-03,
         2.1870e-03],
        ...,
        [7.2990e-05, 1.1572e-04, 1.0596e-04,  ..., 2.3371e-01, 1.8952e-03,
         1.1007e-02],
        [1.0707e-04, 1.1280e-04, 9.7382e-05,  ..., 6.6098e-06, 8.3576e-02,
         1.1166e-04],
        [3.1683e-06, 4.1378e-06, 4.9262e-06,  ..., 4.8532e-07, 4.9637e-03,
         1.0368e-02]]), 'X': tensor([[  4.0421,  -3.7765,  -8.3631],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.6314,  -1.9302,  -9.6543],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.3560,   9.5391,   9.4943],
        [  5.0617,  12.7957,   8.3544],
        [ -3.2736, -10.1703,   3.8712],
        [  6.3589,   0.8239,  -8.4230],
        [ -6.7992,   5.4991,   9.9186],
        [  1.0391,  -4.8141,  -0.0768],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.5620,   7.0549,   8.4825],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.5430,   5.5438,  -3.8669],
        [ 11.3800,  -4.3919,   1.4066],
        [  4.9957,  -2.8705,   6.5978],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.0039,   7.3173,  -1.5885],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.3606,   0.9204,   3.2525],
        [  2.3261,  13.7749,   5.9121],
        [ -5.3382,   6.6533,   7.8054],
        [  6.3542,   0.6934,  -3.3840],
        [ -9.0574,   1.7801,   8.8986],
        [ -2.5056,   7.7720,   5.5878],
        [ -9.3705,   0.4208,  -3.5815],
        [ -0.1963,   7.8765,  -1.1371],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.2022,   2.4092,   3.0349],
        [ -5.1398,  -9.1367,   6.2074],
        [ -9.4697,  -0.6157,  -6.8472],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.1890,   4.7318,   0.9515],
        [  6.3266,   0.3292,   9.3502],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.5813,  -4.5471,   5.3597],
        [  5.1890,   4.7318,   0.9515],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.4242, -10.1261,   3.9595],
        [ -2.2188,   7.8218,   8.6064],
        [  0.1047, -11.0046,   0.5188],
        [  0.7275,  14.0667,  -0.9815],
        [  6.8093,  -9.4178,   8.4244],
        [  9.6662,   9.1725,   8.1056],
        [  5.0348,  -2.8229,   3.5436],
        [  4.7739,  -3.1218,  -4.1188],
        [ 10.0995,  -6.4739,   2.0974],
        [ 11.4743,   6.2478,  -2.1100],
        [  4.1140,  -3.7220,  -0.1273],
        [  9.0652,   9.8597,  -3.5949],
        [  0.6396,  14.0772,  -6.3786],
        [  3.9895,   6.0816,  -6.5235],
        [  2.5128, -10.9365,   9.9645],
        [ -8.5149,  -5.1905,   1.5239],
        [  3.6703,   6.3443,  -5.4401],
        [ -7.4807,   4.7463,   2.0616],
        [  2.2373,  -4.6506,  -1.7334],
        [ -4.0868,   7.2981,   8.1874],
        [  2.8890,  -4.4316,   1.1600],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.2242,  -0.3171,   7.4978],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.8450,  -9.9356,   3.0767],
        [ -7.5401,   4.6718,   9.0033],
        [  5.9587,  12.3216,   9.1695],
        [  2.0243,   7.3086,   7.5432],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.6019,   5.9540,  -5.1715],
        [ -8.1426,   3.8048,   9.3542],
        [  1.8447,  -4.7372,   8.1235],
        [ -2.9497, -10.3028,  -7.3479],
        [  6.3604,   0.9042,  -6.5901],
        [  5.0617,  12.7957,   8.3544],
        [ -6.4844,  -8.0230,   8.5209],
        [  0.3493,  -4.7727,  -6.5682],
        [ -7.9925,  -6.1423,   2.8990],
        [  4.5503,   5.5359,   7.4735],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -0.0254, -10.9932,   3.2200],
        [  5.7151,  -1.7700,   3.1542],
        [  4.2275,  -3.6320,  -0.2714],
        [ -0.8455, -10.8853,  -6.7400],
        [  0.8940, -11.0402,  -0.9783],
        [  0.1532,  -4.7422,  -7.0145],
        [  2.1440, -10.9811,   8.5281],
        [  0.6933,   7.7336,   2.1179],
        [ -8.9624,  -4.0944,  -4.9622],
        [  5.3118,   4.5452,   4.2114],
        [  2.5469,  13.7192,  -9.6935],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -7.7239,   4.4298,   3.1216],
        [  8.7715,  -7.9390,   5.2900],
        [ -6.0195,   6.1789,  -2.9496],
        [ -4.4779,  -9.5609,  -2.4505],
        [  6.9653,  -9.3225,  -3.4727],
        [  9.0004,  -7.7195,   4.6945],
        [  6.2022,   2.4092,   3.0349],
        [  2.3367,   7.1676,  -9.5263],
        [ 11.8951,  -3.1570,   7.0742],
        [  2.0547,   7.2957,  -1.9343],
        [ 10.5955,   7.8824,  -3.7006]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0095, 0.0262,  ..., 1.0681, 0.3091, 0.2762],
        [0.0095, 0.0000, 0.0229,  ..., 1.0643, 0.3094, 0.2668],
        [0.0262, 0.0229, 0.0000,  ..., 1.0835, 0.3321, 0.2603],
        ...,
        [1.0681, 1.0643, 1.0835,  ..., 0.0000, 0.7993, 1.0316],
        [0.3091, 0.3094, 0.3321,  ..., 0.7993, 0.0000, 0.4493],
        [0.2762, 0.2668, 0.2603,  ..., 1.0316, 0.4493, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.53175275,  5.68942572,  5.83560009,  5.94541803,  5.95297459,
        6.19633139,  5.26067019, 13.3614598 , 13.76047999, 10.68416401,
        6.41202986,  8.74470218,  4.92497514, 10.5575601 ,  7.50564223,
       13.09737053,  7.16747878, 12.19805394,  5.76165934, 12.59969936,
        9.43737523,  7.58668144, 10.27716434,  6.42689705, 13.96987806,
        8.53014915,  6.39187285,  9.23071698,  8.16584537,  9.37989631,
        7.8788965 , 12.18045978,  9.44486609,  6.65368551, 10.48315436,
        9.48969978, 12.16901615,  7.02253235,  6.33517349, 12.22276278,
        5.2287058 ,  7.02253235, 11.86239961, 11.48734978,  8.1303945 ,
       11.00508681, 14.08549802, 11.62158554, 13.32557133,  5.77219077,
        5.70405525, 11.99633819, 13.06498554,  5.54776934, 13.39372426,
       14.09176259,  7.27340834, 11.22142195,  9.97220799,  7.32950385,
        8.85939626,  5.16077764,  8.36445758,  5.29009836, 12.12641969,
        6.2322769 , 10.61472636, 11.52733581,  8.87007495, 13.68672855,
        7.58378655,  6.81961996, 13.04051744,  8.98765644,  5.08372149,
       10.7167285 ,  6.42439344, 13.76047999, 10.31584102,  4.78544875,
       10.08001894,  7.16598273, 12.46513646, 10.99326577,  5.98285319,
        5.57341623, 10.91805683, 11.07637531,  4.74468406, 11.18839169,
        7.76457095,  9.85330646,  6.99097013, 13.95361098, 12.49291768,
        8.90405823, 11.83075005,  8.62631801, 10.5575601 , 11.63724792,
       11.85743425,  6.65368551,  7.53884104, 12.30695098,  7.57946459,
       13.20597932])}
{'P': tensor([[3.2742e-02, 1.8099e-01, 1.6409e-01,  ..., 5.8151e-03, 3.3234e-03,
         1.7997e-03],
        [5.6815e-03, 5.6214e-02, 1.1594e-02,  ..., 8.2159e-03, 3.9293e-03,
         4.1180e-03],
        [6.6814e-03, 1.2059e-01, 4.3263e-02,  ..., 7.3285e-03, 3.7809e-03,
         2.4439e-03],
        ...,
        [7.9699e-04, 1.2890e-03, 9.6150e-04,  ..., 5.9287e-02, 1.7076e-04,
         5.8221e-02],
        [6.9264e-04, 1.1787e-03, 8.3841e-04,  ..., 4.3460e-01, 1.6394e-02,
         1.0653e-02],
        [2.3190e-05, 3.1072e-05, 2.4975e-05,  ..., 8.0996e-06, 3.0153e-06,
         2.7104e-02]]), 'X': tensor([[ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0245, 0.0159,  ..., 0.0558, 0.1708, 0.4666],
        [0.0245, 0.0000, 0.0360,  ..., 0.0775, 0.1753, 0.4420],
        [0.0159, 0.0360, 0.0000,  ..., 0.0577, 0.1831, 0.4759],
        ...,
        [0.0558, 0.0775, 0.0577,  ..., 0.0000, 0.1418, 0.5140],
        [0.1708, 0.1753, 0.1831,  ..., 0.1418, 0.0000, 0.5135],
        [0.4666, 0.4420, 0.4759,  ..., 0.5140, 0.5135, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.53255882,  5.63824971,  5.57341623,  5.69913817,  5.94909709,
        5.87305005, 13.91198191,  7.78980299,  9.24501829, 10.27572673,
        5.60064319, 11.57249984,  5.23878371, 12.1494023 ,  4.7663526 ,
       10.07238463, 13.04156075,  7.83534987, 10.27572673, 13.45642379,
        6.49538841,  7.03733775,  6.42689705,  8.11227668, 10.98076873,
        7.09887557,  5.84450127, 12.889695  ,  9.48969978,  5.70405525,
        6.27715105,  8.68832398,  5.68942572,  5.30158497, 13.61801526,
       13.59170809, 13.25017211,  6.89245352, 11.05413545, 11.07637531,
        5.14124612, 10.48840476,  8.06990483, 11.4424816 ,  6.81961996,
       11.24434374,  4.90676242,  7.17789666,  5.43544626,  9.20340031,
       11.85890664, 13.36800073,  7.54620849, 11.63724792,  7.67939546,
        8.1137452 ,  9.38232121, 13.01187484,  4.73784222,  9.09963894,
        8.25828409, 14.03618649,  6.90115901, 11.07637531, 13.55868684,
       11.20100795,  8.41148722,  7.10378919,  7.32233733,  8.49161154,
        9.99145096, 13.60275211, 10.99326577, 10.35464916,  7.78980299,
       11.64746882, 13.04156075,  9.3109884 , 10.61047596, 13.20988996,
        5.23878371, 12.7125079 ,  8.31625171, 11.83075005,  9.34061938,
        9.45995955,  9.17905491, 13.55868684,  6.5618968 ,  6.27715105,
        5.94909709, 11.83570132,  8.09842704,  6.65991877,  9.94285511,
        9.75075273,  7.33350642, 10.57302546,  6.46356806,  9.48337063,
       11.20100795,  4.8763135 ,  6.23262289, 12.1494023 , 11.9182956 ,
       12.79126785])}
{'P': tensor([[3.5234e-02, 1.6730e-01, 1.5623e-01,  ..., 5.2790e-04, 1.9236e-04,
         1.3118e-05],
        [3.5523e-03, 7.5807e-02, 3.7650e-01,  ..., 2.2062e-03, 1.0773e-03,
         2.5671e-05],
        [4.8989e-03, 1.7928e-02, 1.1652e-01,  ..., 3.8195e-03, 1.8745e-03,
         4.1419e-05],
        ...,
        [3.1756e-04, 2.7783e-04, 2.8040e-04,  ..., 1.7975e-01, 8.5702e-04,
         3.5750e-03],
        [4.5202e-06, 5.1720e-06, 5.3392e-06,  ..., 1.1625e-02, 2.9698e-02,
         2.8090e-05],
        [1.2328e-04, 5.8822e-05, 5.7020e-05,  ..., 1.1181e-05, 4.5353e-07,
         4.2849e-02]]), 'X': tensor([[  4.0642,  -3.7599,  -3.4565],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.1787,  -2.6383,  -4.2533],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.8617,   3.4853,  -9.8080],
        [  7.8375,  -8.7240,  -9.8913],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.1041, -11.0045,   2.4311],
        [  3.9022,  13.2906,   5.5788],
        [ -9.4773,  -1.0024,   5.2599],
        [  3.9591, -10.6391,  -6.2557],
        [  6.3087,   1.8053,  -1.9481],
        [  1.5662,  13.9375,   4.5142],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.1931,  -4.7491,   7.4385],
        [  5.6022,  -1.9833,   8.3081],
        [  4.2275,  -3.6320,  -0.2714],
        [  6.2975,   0.0947,   5.1791],
        [  1.5246,  -4.7838,  -5.2095],
        [ -2.2188,   7.8218,   8.6064],
        [  1.3767, -11.0341,   9.9415],
        [  7.9403,  -8.6455,  -8.7433],
        [  1.9503,  -4.7172,  -6.2313],
        [ -6.3306,   5.9272,   5.6935],
        [  1.2728, -11.0372,   9.8121],
        [  0.1382, -11.0073,   1.0026],
        [ -9.3739,   0.3982,  -4.2564],
        [ -7.5603,   4.6459,   8.2002],
        [  4.2612, -10.5514,  -6.6268],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.6231,   5.4556,   8.4973],
        [  2.6614,  -4.5192,  -9.3641],
        [  0.6396,  14.0772,  -6.3786],
        [ -5.3736,   6.6310,  -1.5174],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -2.5056,   7.7720,   5.5878],
        [  0.6396,  14.0772,  -6.3786],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.4639, -10.9431,  -7.3718],
        [  5.9193,  -1.3176,   5.1739],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.5960, -11.0335,   3.1144],
        [  2.1504,  -4.6726,   8.7908],
        [  1.1139,   7.6281,  -2.5680],
        [  1.7815,  -4.7481,   4.4289],
        [  4.2041,  -3.6509,  -4.9488],
        [  3.4140,  -4.1803,  -8.8224],
        [  3.7992,  -3.9474,  -8.9019],
        [ -2.6485, -10.4151,   0.8766],
        [  6.3551,   0.7149,   1.3344],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.6933,   7.7336,   2.1179],
        [  5.7802,  -1.6363,  -6.2782],
        [ -9.1178,   1.5784,   2.1410],
        [ -7.2986,  -7.1225,  -3.0814],
        [  6.3497,   1.3680,  -1.9076],
        [  4.5491,  -3.3485,   6.8412],
        [ -7.4733,   4.7555,   6.2490],
        [  2.6614,  -4.5192,  -9.3641],
        [  0.9036,   7.6840,   9.9237],
        [ -8.1204,  -5.9308,   4.0761],
        [ -0.8675, -10.8815,   6.8561],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.8474,   6.8979,   6.6457],
        [ 12.4690,  -0.8528,  -9.3374],
        [  6.2675,  12.1379,  -8.9720],
        [  6.3124,   0.2057,  -0.4593],
        [  6.0054,  -1.0902,   8.3079],
        [ -6.0844,  -8.3955,   0.7705],
        [ -9.4296,  -1.9458,   4.0833],
        [  0.1460,   7.8340,  -9.4614],
        [  3.4604,  -4.1545,   1.8091],
        [  9.4925,   9.3811,   8.9842],
        [  6.9956,  11.6590,  -7.6876],
        [ 11.7162,   5.6728,  -8.9372],
        [ -8.6681,  -4.8556,   9.2518],
        [  7.9141,  -8.6656,   9.6948],
        [  0.3231,  -4.7691,  -1.6120],
        [  7.1441,  11.5529,   2.3232],
        [ -9.2508,   1.0572,   2.9978],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.1426,   3.8048,   9.3542],
        [ -3.1671,   7.6158,   6.0765],
        [ -4.7389,   6.9953,   3.7638],
        [ -9.4766,  -1.1116,  -2.1706],
        [  2.7121,   6.9743,  -8.4202],
        [ -8.1204,  -5.9308,   4.0761],
        [  1.6408,   7.4591,  -4.2294],
        [ -8.5900,   2.9767,   7.4811],
        [  2.1379,  -4.6757,   6.4981],
        [  2.8994,   6.8675,  -9.5987],
        [  9.4925,   9.3811,   8.9842],
        [  6.2022,   2.4092,   3.0349],
        [ -1.5875,   7.8944,  -9.4286],
        [  9.3836,   9.5075,  -5.9278],
        [ -8.0750,  -6.0072,  -5.8267],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.6817,   6.3354,   6.1233],
        [ -0.0282,   7.8575,  -1.1746],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.4773,  -1.0024,   5.2599]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.7692e-02, 2.8266e-02,  ..., 2.3279e-01, 3.1149e-01,
         1.0307e+00],
        [2.7692e-02, 0.0000e+00, 1.1995e-02,  ..., 2.5962e-01, 2.8386e-01,
         1.0577e+00],
        [2.8266e-02, 1.1995e-02, 1.5170e-04,  ..., 2.5512e-01, 2.8530e-01,
         1.0531e+00],
        ...,
        [2.3279e-01, 2.5962e-01, 2.5512e-01,  ..., 0.0000e+00, 5.3712e-01,
         7.9807e-01],
        [3.1149e-01, 2.8386e-01, 2.8530e-01,  ..., 5.3712e-01, 0.0000e+00,
         1.3317e+00],
        [1.0307e+00, 1.0577e+00, 1.0531e+00,  ..., 7.9807e-01, 1.3317e+00,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.5366698 ,  5.84450127,  5.8828986 ,  5.70405525,  5.79391825,
        5.81199422, 10.52357479,  6.81961996, 11.72749923,  4.91296205,
       11.0050379 , 13.85158417,  9.53015376, 11.35182589,  6.5618968 ,
       14.02526505, 12.42554808,  4.75303127,  5.94293185,  5.57341623,
        6.29821673,  5.0209005 ,  8.1303945 , 11.11969815, 11.73848065,
        5.10444097,  8.67227386, 11.11038551, 11.00812996,  9.38232121,
        8.8737556 , 11.37939695,  5.68942572,  7.15100303,  5.24461068,
       14.09176259,  8.53501777, 10.61047596, 10.61047596,  8.16584537,
       14.09176259,  5.21280156, 11.21703654,  6.06416957,  5.27178363,
       11.04953767,  5.14371005,  7.70897675,  5.0713364 ,  5.56809335,
        5.39723241,  5.47866221, 10.74655346,  6.39520397,  9.95552626,
        7.76457095,  6.00732193,  9.25336371, 10.1979669 ,  6.49538841,
        5.64865173,  8.8580671 ,  5.24461068,  7.73692436, 10.05558893,
       10.91601655, 10.0674887 ,  7.46248947, 12.4980845 , 13.66051884,
        6.31576264,  6.10360073, 10.36844813,  9.6282763 ,  7.83534987,
        5.40689332, 13.34587027, 13.59673406, 13.01728313,  9.9353921 ,
       11.73567759,  4.78002665, 13.58333857,  9.3109884 , 11.97949273,
        8.98765644,  8.24808832,  8.44940661,  9.54154096,  7.48311002,
       10.05558893,  7.6374603 ,  9.09119351,  5.14124612,  7.4544866 ,
       13.34587027,  6.65368551,  8.05242591, 13.35832725, 10.06437598,
        4.85995002,  6.25586297,  7.32754882,  7.8575648 , 13.77075174,
        9.53015376])}
{'P': tensor([[3.7353e-02, 1.7535e-01, 1.1661e-01,  ..., 3.9173e-04, 1.9134e-05,
         5.3932e-05],
        [1.0618e-02, 5.9197e-02, 1.7981e-01,  ..., 6.8662e-04, 3.0127e-05,
         7.9663e-05],
        [4.4286e-03, 4.8485e-03, 1.1178e-01,  ..., 1.8583e-03, 3.7663e-05,
         7.1928e-05],
        ...,
        [1.5702e-05, 1.6707e-05, 2.9545e-05,  ..., 7.0066e-02, 5.0645e-06,
         4.2479e-07],
        [2.2740e-03, 2.2056e-03, 6.7778e-04,  ..., 1.0606e-05, 2.1598e-01,
         1.9224e-02],
        [3.4514e-03, 3.1889e-03, 1.3466e-03,  ..., 3.0529e-06, 1.4554e-05,
         2.7424e-02]]), 'X': tensor([[ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0097, 0.0706,  ..., 0.6678, 1.0454, 0.8262],
        [0.0097, 0.0000, 0.0763,  ..., 0.6582, 1.0454, 0.8326],
        [0.0706, 0.0763, 0.0000,  ..., 0.7043, 1.1052, 0.8480],
        ...,
        [0.6678, 0.6582, 0.7043,  ..., 0.0000, 1.2838, 1.3851],
        [1.0454, 1.0454, 1.1052,  ..., 1.2838, 0.0000, 0.6899],
        [0.8262, 0.8326, 0.8480,  ..., 1.3851, 0.6899, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.54776934,  5.57341623,  5.94909709,  5.63824971,  5.87305005,
        5.60064319,  5.21280156, 10.19036081, 11.33837396,  8.16584537,
        6.1561646 ,  8.4486771 ,  5.05989547, 12.45418236,  8.37954853,
        9.58347456,  9.96152788, 13.27994304,  8.66273171, 10.68031351,
        4.91038884,  5.94293185,  5.52416368,  7.27340834, 13.56367857,
        8.41148722, 12.18045978, 13.69162795,  8.11227668,  5.39723241,
        7.20482224, 11.54071859,  8.86082124, 13.93790235,  7.52117132,
        9.58072286, 10.33958639, 10.4606265 , 13.01187484, 10.1371552 ,
        8.94395628, 10.37998078, 12.17901033,  6.81084724,  5.83560009,
        7.06372172, 11.73848065,  5.96556248,  5.48276038,  7.8575648 ,
        9.7751189 , 13.04156075, 14.08549802,  4.83371341, 12.86648771,
        9.44486609,  5.0209005 ,  5.79562691, 10.82106505,  5.77219077,
        9.62238828,  6.50409372,  4.83371341,  7.17789666, 13.1399181 ,
       13.36800073,  9.66977802,  8.06990483, 14.09357547,  5.39723241,
        9.05843201,  9.62958973,  8.83341608,  8.62329466, 11.9182956 ,
       13.05148826, 13.95472331,  8.31625171,  9.04255655,  4.83371341,
        9.34127013,  7.27759233,  9.34061938, 13.1471313 , 10.82106505,
       13.5247215 ,  5.86519525,  5.91408504,  6.42439344,  8.11227668,
        7.37576154,  5.57341623,  8.44940661, 13.1399181 ,  6.84796202,
        7.58378655,  9.69629   ,  8.79284466, 11.81423161,  7.39004226,
        6.18668463, 12.76408562,  7.49076899, 13.1399181 ,  9.91668182,
       10.66631784])}
{'P': tensor([[2.8821e-02, 1.9762e-01, 1.5971e-01,  ..., 2.3321e-05, 3.0732e-03,
         1.9537e-05],
        [1.8688e-03, 2.6926e-02, 2.2691e-01,  ..., 3.0628e-05, 1.8309e-03,
         1.6644e-05],
        [2.1554e-03, 2.9719e-03, 3.9116e-02,  ..., 5.4759e-05, 1.8795e-03,
         2.3347e-05],
        ...,
        [5.9468e-06, 5.7745e-06, 5.8068e-06,  ..., 1.1299e-01, 4.5463e-07,
         2.1380e-04],
        [9.1186e-04, 1.8103e-03, 2.7024e-03,  ..., 6.6025e-05, 4.4421e-02,
         1.6602e-06],
        [2.6637e-04, 1.7400e-04, 1.3582e-04,  ..., 3.5511e-07, 3.0668e-05,
         2.7255e-02]]), 'X': tensor([[  4.1297,  -3.7098,  -3.9838],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.1787,  -2.6383,  -4.2533],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.2770,  -0.0379,   2.4195],
        [  6.2790,  -0.0259,  -5.4116],
        [  2.3261,   7.1726,   6.2522],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.7295,  13.6703,  -8.9328],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.3031,   1.8477,  -7.1706],
        [  6.3505,   0.6235,  -8.5849],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.3943,   6.5483,   1.0790],
        [ 12.4996,   2.6335,  -6.3166],
        [ -5.5525,   6.5146,  -4.4984],
        [ -8.3539,  -5.5111,   0.3373],
        [  1.7301,   7.4262,   5.0522],
        [  0.8700,  -4.8132,  -7.8042],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.2435,   4.0076,   8.1257],
        [  6.2419,  -0.2313,   4.2440],
        [ 12.5650,   2.0132,   4.6272],
        [  2.8474,   6.8979,   6.6457],
        [ -2.2508, -10.5479,  -2.7235],
        [ -4.2040,   7.2486,   2.9875],
        [  2.3376,  -4.6231,   9.8436],
        [  7.1821,  -9.1845,   9.8548],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.0766,  -6.0046,  -3.1873],
        [  2.5382,   7.0672,   4.5097],
        [  6.3087,   1.8053,  -1.9481],
        [  7.7750,  11.0665,   2.8364],
        [  5.3118,   4.5452,   4.2114],
        [ -9.4430,  -0.1897,   4.8383],
        [  5.2608,  12.6977,   4.9166],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.3616,  -2.4749,   0.1793],
        [ -4.2040,   7.2486,   2.9875],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.9624,  -4.0944,  -4.9622],
        [  0.1382, -11.0073,   1.0026],
        [  1.9503,  -4.7172,  -6.2313],
        [ 11.1469,   6.9237,   1.6160],
        [  6.2247,  -0.3150,  -9.3657],
        [ 11.2571,   6.7072,   8.3693],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.4242, -10.1261,   3.9595],
        [  1.5653,  13.9377,   4.1773],
        [  6.2770,  -0.0379,   2.4195],
        [  2.9470,  13.6085,   2.9809],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.0506,   4.9277,  -9.5094],
        [  2.6321,  -4.5296,  -2.5159],
        [  0.9036,   7.6840,   9.9237],
        [ -6.7745,  -7.7264,   8.4684],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.0087,   4.0173,   9.1091],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.3167,   6.6022,   6.3554],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -2.3317,   7.8034,  -3.6274],
        [ -2.5056,   7.7720,   5.5878],
        [  3.3744,   6.5623,  -6.3636],
        [  0.2732,  -4.7619,  -1.1577],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -2.5056,   7.7720,   5.5878],
        [  6.2883,   0.0325,   5.2692],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -1.9714,   7.8563,  -1.8084],
        [ -3.1204, -10.2345,   9.2288],
        [ -9.4645,  -1.4880,  -2.0226],
        [  2.2373,  -4.6506,  -1.7334],
        [ 11.1469,   6.9237,   1.6160],
        [  2.2373,  -4.6506,  -1.7334],
        [  1.7659,   7.4127,   4.7871],
        [ -8.8400,   2.3991,   1.3457],
        [  9.0004,  -7.7195,   4.6945],
        [ 10.9931,   7.2107,   7.7933],
        [ -6.4002,  -8.1048,  -0.9108],
        [  0.1931,  -4.7491,   7.4385],
        [  1.8721,  -4.7322,   0.5524],
        [ 10.5678,   7.9261,  -0.6100],
        [  3.3474,  -4.2162,   5.2455],
        [  5.5240,  -2.1198,   6.2055],
        [  9.4516,  -7.2501,  -1.8868],
        [  5.2188,  -2.5839,   2.1833],
        [  2.4932,   7.0903,  -5.4214],
        [  3.1892, -10.8220,  -6.8409],
        [  7.3592,  11.3937,   6.2407],
        [  6.3587,   1.1596,  -3.6629],
        [  6.3413,   0.4883,  -6.8416],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.6321,  -4.5296,  -2.5159],
        [  0.9659,  -4.8144,  -0.9594],
        [  3.5965,  -4.0754,  -3.5035],
        [ -7.8280,   4.2848,  -7.1955],
        [  2.3261,  13.7749,   5.9121],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.3616,  -2.4749,   0.1793]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0156, 0.0236,  ..., 0.9225, 0.1702, 0.9379],
        [0.0156, 0.0000, 0.0080,  ..., 0.9269, 0.1846, 0.9529],
        [0.0236, 0.0080, 0.0000,  ..., 0.9280, 0.1916, 0.9608],
        ...,
        [0.9225, 0.9269, 0.9280,  ..., 0.0000, 0.8258, 1.3253],
        [0.1702, 0.1846, 0.1916,  ..., 0.8258, 0.0000, 0.8148],
        [0.9379, 0.9529, 0.9608,  ..., 1.3253, 0.8148, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.5513072 ,  5.70405525,  5.79391825,  5.81199422,  5.84450127,
        5.84533345,  6.27715105,  6.27905292,  7.5403822 , 13.14777433,
       13.94008948,  6.06416957, 12.35421461,  6.56833864,  6.38105015,
        8.05242591,  7.37576154, 12.77402351,  8.55982105, 10.00794468,
        7.62509582,  4.89121805, 12.91143491, 12.88270018,  6.24614905,
       12.72524615,  7.46248947, 10.78533838,  8.37954853,  5.1805345 ,
       11.65923396,  9.88310614, 10.06407839,  7.50917551,  6.5618968 ,
       13.5247215 ,  6.99097013,  9.44486609, 13.744387  , 13.77075174,
        9.68323686,  8.37954853,  6.00663314,  9.85330646, 11.00812996,
        5.10444097, 13.12218013,  6.23262289, 13.103709  , 12.33167382,
       11.48734978, 14.02532758,  6.27715105, 13.92390689, 11.70016217,
        9.77777346,  7.05627461,  5.23878371,  7.73692436, 10.27572673,
        8.36995953,  9.23071698,  8.95983586, 13.00413829,  7.38844703,
       10.60226614,  8.14433542,  8.16584537,  7.37902766,  4.76969364,
        9.78147542,  8.16584537,  6.28835101, 12.52372629, 12.27835996,
        8.09984231, 10.69963954,  9.58072286,  5.16077764, 13.12218013,
        5.16077764,  7.62011734,  9.15976746, 11.85743425, 13.14690137,
       10.32715715,  4.75303127,  5.08910491, 13.20988996,  5.38342077,
        5.91677188, 11.9120321 ,  5.82345867,  7.5158489 , 11.28215429,
       13.56367857,  6.46356806,  6.36004393, 10.33958639,  5.23878371,
        4.91038884,  5.43544626,  8.92395381, 13.96987806, 12.04924159,
        9.68323686])}
{'P': tensor([[2.8120e-02, 2.0472e-01, 1.6821e-01,  ..., 8.5764e-05, 3.9086e-05,
         4.0824e-02],
        [1.9928e-03, 2.8135e-02, 2.3709e-01,  ..., 6.3426e-05, 4.8891e-05,
         5.9502e-02],
        [2.3665e-03, 3.2247e-03, 4.2444e-02,  ..., 7.3756e-05, 8.2832e-05,
         1.1697e-01],
        ...,
        [1.0256e-04, 1.3670e-04, 1.5236e-04,  ..., 1.8037e-02, 1.3509e-05,
         1.6103e-04],
        [3.2764e-05, 3.0665e-05, 2.8823e-05,  ..., 9.9577e-06, 8.3772e-02,
         2.5610e-05],
        [1.5851e-03, 1.9884e-03, 2.3364e-03,  ..., 3.4991e-05, 4.3292e-04,
         1.0262e-01]]), 'X': tensor([[  4.1999,  -3.6543,  -4.1998],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.1787,  -2.6383,  -4.2533],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.2911,  -2.4824,  -3.6877],
        [  9.9058,  -6.7203,   5.6928],
        [  2.3373,  -4.6232,  -5.2246],
        [ -5.3382,   6.6533,   7.8054],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.4887,   4.7364,   8.9074],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.9178,   2.1936,   7.5388],
        [ -6.0175,  -8.4541,  -9.8408],
        [  1.5246,  -4.7838,  -5.2095],
        [ 10.0619,   8.6627,  -9.2884],
        [  2.2182,  13.8007,   5.1889],
        [ 11.7349,  -3.5838,   2.1644],
        [  3.9515,  -3.8425,   6.1729],
        [  6.1752,   2.5242,   3.5874],
        [ -2.4956, -10.4682,   3.5632],
        [ -3.0644,   7.6439,  -1.4322],
        [ -7.3446,   4.9113,  -5.9356],
        [ -9.2687,  -2.9794,   0.6436],
        [ 10.2470,   8.4054,  -0.1177],
        [  3.2990,  -4.2415,  -7.4857],
        [  7.2340,  -9.1505,  -5.8321],
        [ -5.2589,   6.7024,  -5.5809],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.3460,   1.4252,   1.2291],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.0233,   4.9647,  -3.1788],
        [ 11.5083,  -4.1177,   0.2389],
        [  6.2022,   2.4092,   3.0349],
        [ 11.7349,  -3.5838,   2.1644],
        [  9.0051,   9.9234,   5.4411],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 12.4877,  -0.7222,   8.8347],
        [  4.9531,   5.0576,   5.6794],
        [  3.9770, -10.6341,   8.0783],
        [  5.8201,   3.5841,  -8.0609],
        [  5.7218,  -9.9934,   6.5764],
        [  3.5965,  -4.0754,  -3.5035],
        [ 12.5919,   0.4035,  -5.8460],
        [ -9.4714,  -1.3301,  -3.1998],
        [  8.4691,  10.4571,  -5.0650],
        [  9.9058,  -6.7203,   5.6928],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -3.9508,  -9.8515,   1.7923],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.3107,  -2.7697,  -3.2919],
        [  7.7794,  11.0629,   8.9069],
        [  4.5325,  -3.3643,   6.7699],
        [  4.0587, -10.6112,   6.2273],
        [  5.9912,   3.1453,  -3.9989],
        [ -9.3107,  -2.7697,  -3.2919],
        [  0.1497,   7.8334,  -5.4412],
        [  4.8674,  12.8875,   6.7622],
        [  1.9241,   7.3503,   7.9004],
        [  6.2905,   1.9366,   2.1169],
        [  6.3569,   0.7591,   6.5011],
        [  2.6614,  -4.5192,  -9.3641],
        [ -8.1426,   3.8048,   9.3542],
        [  1.3705,  13.9723,   7.9266],
        [  2.3589,  13.7668,  -3.6501],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.3924, -10.1396,  -4.0863],
        [ -3.2736, -10.1703,   3.8712],
        [ -7.4807,   4.7463,   2.0616],
        [ -3.5173,   7.5091,  -8.9585],
        [ 10.1983,  -6.3430,  -0.3200],
        [  9.3836,   9.5075,  -5.9278],
        [ -0.0254, -10.9932,   3.2200],
        [  7.5064,  -8.9651,  -0.0994],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.6672,  -1.8631,   8.1020],
        [ -8.0965,  -5.9712,   4.3466],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.2515,   5.0195,   4.6434],
        [  1.7228,  -4.7575,   8.3396],
        [  3.5284, -10.7484,   5.3347],
        [  2.0937,  13.8293,  -2.3533],
        [  7.7750,  11.0665,   2.8364],
        [ -1.0012, -10.8576,   2.2867],
        [  6.2895,   0.0405,   5.7652],
        [  5.5424,  -2.0885,   7.2487],
        [  0.9945,  -4.8144,  -0.1897],
        [  0.2732,  -4.7619,  -1.1577],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.1741,  -0.5329,   3.1913],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3303,   1.6145,  -8.4083],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.1579,  -0.5961,  -6.5595],
        [  0.9967,  -4.8144,  -9.7370],
        [ -5.6174,   6.4708,  -1.7339],
        [ -3.3812, -10.1235,   7.4384],
        [  9.5261,  -7.1675,   1.4206],
        [ 11.5198,  -4.0923,   5.5088],
        [  2.1440, -10.9811,   8.5281],
        [ 11.5456,   6.0861,   8.1101],
        [  7.5930,  -8.9037,   7.2323],
        [ -1.9600,   7.8577,   6.9121],
        [  6.3236,   0.3011,  -4.6668]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0106, 0.0181,  ..., 0.9298, 0.8164, 0.0468],
        [0.0106, 0.0000, 0.0080,  ..., 0.9331, 0.8198, 0.0373],
        [0.0181, 0.0080, 0.0000,  ..., 0.9387, 0.8255, 0.0294],
        ...,
        [0.9298, 0.9331, 0.9387,  ..., 0.0000, 0.1139, 0.9613],
        [0.8164, 0.8198, 0.8255,  ..., 0.1139, 0.0000, 0.8484],
        [0.0468, 0.0373, 0.0294,  ..., 0.9613, 0.8484, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.5671389 ,  5.70405525,  5.79391825,  5.81199422,  5.84533345,
        5.84450127, 11.9702762 ,  5.18047638,  8.53014915, 12.85032836,
        8.86082124, 13.89974693,  9.18358562, 10.37698435,  5.0209005 ,
       13.27717982, 13.97779839, 12.26997165,  5.51177501,  6.67121748,
       10.76154256,  8.23525433,  8.8353674 ,  9.73579738, 13.25336001,
        5.37343173, 11.66453059,  8.51928531,  9.76896904,  6.50409372,
        9.26455795,  7.06272256, 12.22276278,  6.65368551, 12.26997165,
       13.40024224,  9.69629   , 12.5085995 ,  7.07902036, 11.35345345,
        6.83515201, 11.51557166,  5.43544626, 12.59840193,  9.5642993 ,
       13.45642379, 11.9702762 , 10.06407839, 12.42554808, 10.61417181,
       13.27717982,  9.71392179, 13.52430176,  5.64464807, 11.36089355,
        6.76661595,  9.71392179,  7.83486943, 13.77604293,  7.59795445,
        6.58184474,  6.4020404 ,  5.24461068,  8.98765644, 14.0393949 ,
       13.96746816,  4.90505566, 11.48434767, 10.68416401,  8.85939626,
        8.29203356, 12.00995898, 13.35832725, 10.99326577, 11.69264243,
       12.0495521 ,  5.96556248, 10.06023665,  5.94909709, 10.0674887 ,
        8.81930525,  5.05981914, 11.31276109, 13.98691426, 13.5247215 ,
       10.9036226 ,  6.28961988,  5.92281142,  4.9160962 ,  4.76969364,
       11.51811603,  6.19707944,  9.09963894,  6.53290663, 12.50996828,
        6.18668463,  4.91652556,  8.56889738, 10.67322527, 11.92133129,
       12.22503754, 11.18839169, 13.05148826, 11.70168755,  8.09842704,
        6.33075682])}
{'P': tensor([[2.7616e-02, 1.3568e-01, 1.1621e-01,  ..., 1.2699e-05, 2.7616e-03,
         1.8709e-05],
        [1.8142e-03, 3.5477e-02, 8.7552e-03,  ..., 1.4769e-05, 3.1631e-03,
         2.0220e-05],
        [1.9920e-03, 9.1900e-02, 3.1357e-02,  ..., 1.3848e-05, 2.5943e-03,
         1.8626e-05],
        ...,
        [6.0161e-04, 3.2671e-04, 3.7985e-04,  ..., 9.0813e-02, 6.9452e-06,
         7.2643e-02],
        [2.1165e-04, 5.5446e-04, 4.6162e-04,  ..., 1.0633e-06, 7.8444e-02,
         1.5189e-06],
        [1.5600e-03, 7.5753e-04, 8.6360e-04,  ..., 7.2470e-03, 1.8600e-05,
         1.0022e-01]]), 'X': tensor([[ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0219, 0.0280,  ..., 1.0160, 0.3616, 0.9468],
        [0.0219, 0.0000, 0.0134,  ..., 1.0290, 0.3601, 0.9642],
        [0.0280, 0.0134, 0.0000,  ..., 1.0203, 0.3731, 0.9587],
        ...,
        [1.0160, 1.0290, 1.0203,  ..., 0.0000, 1.3401, 0.2850],
        [0.3616, 0.3601, 0.3731,  ..., 1.3401, 0.0000, 1.2234],
        [0.9468, 0.9642, 0.9587,  ..., 0.2850, 1.2234, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.56809335,  5.84533345,  5.81199422,  5.90035324,  5.79391825,
        5.70405525,  8.27853076, 10.1371552 , 10.92597196,  8.0550286 ,
       11.73848065, 12.4980845 ,  5.84450127,  8.98765644, 11.89105259,
        6.652742  , 12.07225469,  6.81084724, 10.31584102,  6.78325151,
        7.24326992,  8.12580976,  9.10945678, 13.08414689, 12.41012046,
        7.37576154,  5.52416368, 12.35421461, 10.36844813,  5.29009836,
        6.76056625, 13.6829843 ,  6.12886024,  9.99145096,  5.08372149,
        6.74872981,  6.33075682,  9.7751189 ,  6.31576264,  7.50500064,
        6.02789634,  8.56889738,  6.53250937,  9.77777346,  6.29821673,
       13.15525803,  5.07306274,  8.859416  , 12.35421461, 13.25017211,
       12.38881635,  5.18277858,  8.74470218,  5.21280156,  7.57946459,
       11.9702762 ,  7.24326992,  9.7712321 ,  8.97052115, 13.6829843 ,
        8.10886021,  8.37954853,  6.74686071,  8.40445872,  6.76661595,
       13.94008948,  9.62238828,  8.55982105,  7.53884104,  5.48276038,
       11.27656488, 11.05672136, 13.0493368 ,  6.83515201,  9.37296102,
       11.54672755, 13.04156075, 12.33167382, 13.04217487, 13.32557133,
        7.54127425,  5.14607014,  6.34059021, 10.35485331,  5.18047638,
       13.77604293,  5.19577146, 11.21703654,  5.94541803,  8.04718433,
        8.86082124,  7.5403822 ,  8.30729352, 12.18998195,  6.7166297 ,
       10.7167285 , 12.32545637,  8.41148722,  6.7593917 , 10.48721382,
        6.10360073,  7.17789666, 12.62576026,  9.78285424, 12.4980845 ,
        9.88310614])}
{'P': tensor([[5.3631e-02, 1.6290e-01, 1.4950e-01,  ..., 1.4718e-03, 3.8017e-03,
         7.1106e-02],
        [5.1592e-03, 1.1894e-01, 7.1730e-03,  ..., 5.1212e-03, 1.1777e-02,
         5.8175e-02],
        [1.3596e-02, 1.4524e-01, 6.5919e-02,  ..., 2.4071e-03, 6.3572e-03,
         1.2703e-01],
        ...,
        [1.0058e-04, 1.0526e-04, 1.1102e-04,  ..., 6.1882e-02, 1.1889e-03,
         1.8948e-04],
        [1.1876e-04, 1.5430e-04, 1.3471e-04,  ..., 2.6239e-01, 4.2630e-02,
         2.8960e-04],
        [1.9049e-03, 7.5378e-03, 2.1677e-03,  ..., 1.1241e-02, 3.0803e-02,
         6.4000e-02]]), 'X': tensor([[  4.2275,  -3.6320,  -0.2714],
        [  5.6202,  -1.9508,   0.7471],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.3860,  -2.3418,   1.0826],
        [  5.1667,  -2.6543,   1.2879],
        [  5.8325,  -1.5222,  -2.1341],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -6.4464,   5.8272,   7.5038],
        [ -0.6323,   7.9085,  -5.9544],
        [ -7.8359,   4.2735,   7.2943],
        [  0.1287, -11.0065,   6.5757],
        [  1.1350,  14.0104,   0.6321],
        [  6.3122,  -9.6994,  -6.3282],
        [ -3.3812, -10.1235,   7.4384],
        [ 11.4703,  -4.2006,   7.4086],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.0031,  12.2958,  -6.5992],
        [  5.6022,  -1.9833,   8.3081],
        [  6.8407,  11.7666,   3.0045],
        [  8.7131,  10.2216,   7.7915],
        [ -5.9872,   6.2037,  -9.4477],
        [  8.5198,  10.4091,  -8.6486],
        [  6.2883,   0.0325,   5.2692],
        [  0.1047, -11.0046,   0.5188],
        [ -1.9646,   7.8571,   3.3817],
        [ 11.0373,   7.1299,  -8.5895],
        [  1.3670,  -4.7990,   3.2603],
        [  2.1440, -10.9811,   8.5281],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.4748,  -0.7765,  -5.6695],
        [  3.2701,   6.6338,   9.9209],
        [  9.8082,  -6.8396,   4.9615],
        [  6.3587,   1.1596,  -3.6629],
        [  1.6954,  -4.7616,   9.8198],
        [ -7.5720,   4.6310,   2.1717],
        [  6.3583,   1.1729,   8.7835],
        [  0.7145,  -4.8071,  -2.3766],
        [  9.5261,  -7.1675,   1.4206],
        [  5.6258,  -1.9404,  -6.0715],
        [  4.0591,   6.0202,  -1.6339],
        [  1.3339,   7.5627,  -0.9034],
        [  4.9531,   5.0576,   5.6794],
        [  3.3167,   6.6022,   6.3554],
        [  1.0093,   7.6567,   2.3526],
        [  5.1793,   4.7461,   0.3899],
        [  6.9284,  11.7061,  -0.3732],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.7784,  -1.6401,   9.5587],
        [  0.1497,   7.8334,  -5.4412],
        [ -2.0721,   7.8432,   6.6380],
        [  1.6408,   7.4591,  -4.2294],
        [ -6.4844,  -8.0230,   8.5209],
        [  6.3303,   1.6145,  -8.4083],
        [  6.2576,   2.1359,  -3.3398],
        [  2.1625,  -4.6697,   7.0658],
        [  3.3943,   6.5483,   1.0790],
        [  9.4925,   9.3811,   8.9842],
        [ -9.4501,  -1.7133,   2.2980],
        [  0.9445,   7.6736,   0.5095],
        [  2.2373,  -4.6506,  -1.7334],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 12.3453,  -1.5462,   7.0212],
        [ -1.6906,   7.8860,   7.7142],
        [  2.1477,   7.2550,   8.9688],
        [ -1.8788, -10.6567,   3.7068],
        [ -6.0195,   6.1789,  -2.9496],
        [  8.7715,  -7.9390,   5.2900],
        [  2.1504,  -4.6726,   8.7908],
        [  9.1409,  -7.5789,   1.5814],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.3554,   5.7393,  -9.0766],
        [  6.0073,   3.0986,   6.7937],
        [  6.3587,   1.1596,  -3.6629],
        [  2.6310, -10.9195,   3.8054],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.4837,   2.7510,   0.4449],
        [  9.1820,   9.7334,  -2.7932],
        [  7.8982,  -8.6778,   0.0239],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.1575,  12.2046,   3.3257],
        [  0.7145,  -4.8071,  -2.3766],
        [  7.7794,  11.0629,   8.9069],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.6703,   6.3443,  -5.4401],
        [  0.3231,  -4.7691,  -1.6120],
        [ -7.5720,   4.6310,   2.1717],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -6.4844,  -8.0230,   8.5209],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.9967,  -4.8144,  -9.7370],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.2555,   2.1474,   4.1685],
        [ -8.6050,   2.9451,  -0.7012],
        [  2.1379,  -4.6757,   6.4981],
        [ -1.6086,   7.8928,   4.3385],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.2419,  -0.2313,   4.2440],
        [  5.8617,   3.4853,  -9.8080],
        [  6.3266,   0.3292,   9.3502],
        [ -8.9788,   2.0208,  -3.2867],
        [  3.3744,   6.5623,  -6.3636],
        [  5.1292,   4.8183,  -5.7997],
        [  6.0140,  -1.0661,  -2.5040]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0763, 0.0360,  ..., 0.3510, 0.3192, 0.1567],
        [0.0763, 0.0000, 0.1077,  ..., 0.4251, 0.3889, 0.2202],
        [0.0360, 0.1077, 0.0000,  ..., 0.3175, 0.2836, 0.1208],
        ...,
        [0.3510, 0.4251, 0.3175,  ..., 0.0000, 0.0743, 0.2204],
        [0.3192, 0.3889, 0.2836,  ..., 0.0743, 0.0000, 0.1716],
        [0.1567, 0.2202, 0.1208,  ..., 0.2204, 0.1716, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.57341623,  5.94909709,  5.63824971,  5.87305005,  5.80860165,
        6.02789634, 12.4980845 ,  8.68979052,  7.9337647 ,  8.92549752,
       11.00726483, 14.05633441, 11.57249984, 10.67322527, 12.21532282,
       11.74901778, 13.6829843 ,  5.94293185, 13.6105595 , 13.43127131,
        8.62162372, 13.45124917,  6.28835101, 11.00508681,  8.09899874,
       13.1399181 ,  4.98989906, 11.18839169, 10.5575601 , 13.20597932,
        9.50654967,  7.39599986, 11.95743558,  6.46356806,  5.05444288,
        8.87587457,  6.46560181,  4.85995002, 11.92133129,  5.9510489 ,
        7.26074925,  7.67939546,  7.07902036,  7.38844703,  7.72291946,
        7.02496223, 13.60275211,  9.95552626,  6.00663314,  7.83486943,
        8.11227668,  7.6374603 , 10.31584102,  6.53290663,  6.61211224,
        5.14607014,  7.37576154, 13.34587027,  9.60412735,  7.73151309,
        5.16077764,  9.88310614, 12.44177641,  8.06517143,  7.56617037,
       10.82106505,  8.62631801, 11.83075005,  5.14371005, 11.87412439,
       11.66453059,  7.20482224,  6.7593917 ,  6.46356806, 11.23201619,
       12.59969936, 12.78327037, 13.38091121, 11.73397309, 10.63038565,
       13.66991027,  4.85995002, 13.52430176, 10.6910768 ,  7.32950385,
        4.78002665,  8.87587457, 10.30972193, 10.06437598, 10.31584102,
        9.95552626,  4.91652556,  9.45995955,  6.61386436,  9.09502385,
        5.14124612,  8.0550286 ,  7.98469166,  9.45995955,  6.24614905,
        6.81961996,  6.33517349,  9.20340031,  7.37902766,  7.03733775,
        6.10774544])}
{'P': tensor([[2.8205e-02, 1.7004e-01, 1.6925e-01,  ..., 1.0068e-05, 2.6180e-05,
         2.4282e-05],
        [1.8263e-03, 3.2205e-02, 8.9591e-03,  ..., 1.1940e-05, 3.9223e-05,
         1.8618e-05],
        [1.9362e-03, 7.3740e-02, 2.9583e-02,  ..., 1.0039e-05, 2.9896e-05,
         1.6014e-05],
        ...,
        [7.8774e-05, 5.0897e-05, 4.9662e-05,  ..., 4.5491e-02, 1.1080e-06,
         4.2134e-05],
        [1.3953e-05, 2.1047e-05, 1.7629e-05,  ..., 4.9691e-05, 2.4483e-01,
         7.8355e-06],
        [4.4603e-05, 6.4483e-05, 5.2525e-05,  ..., 1.3320e-06, 4.6150e-03,
         3.4624e-02]]), 'X': tensor([[  4.2339,  -3.6268,   6.3125],
        [  5.1203,  -2.7151,   6.2110],
        [  4.9957,  -2.8705,   6.5978],
        [  5.1886,  -2.6250,   6.4677],
        [  5.5240,  -2.1198,   6.2055],
        [  5.6093,  -1.9705,   6.4807],
        [ 12.2692,  -1.8882,   3.9253],
        [ -9.3087,   0.7792,  -7.5587],
        [  2.6310, -10.9195,   3.8054],
        [ -3.1204, -10.2345,   9.2288],
        [ -3.5029,   7.5138,   8.1853],
        [  3.6703,   6.3443,  -5.4401],
        [ -8.9178,   2.1936,   7.5388],
        [  0.9659,  -4.8144,  -0.9594],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -0.0282,   7.8575,  -1.1746],
        [  1.6954,  -4.7616,   9.8198],
        [ -0.7599, -10.8995,  -6.0366],
        [ -7.3446,   4.9113,  -5.9356],
        [ -3.4488, -10.0934,   2.5195],
        [ 12.4592,  -0.9168,  -6.7571],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.8367,   7.8718,  -8.6523],
        [  5.8201,   3.5841,  -8.0609],
        [  2.8602,   6.8905,   5.8779],
        [  6.9653,  -9.3225,  -3.4727],
        [ 12.1588,   4.3364,  -8.0867],
        [  0.6757, -11.0361,  -1.3047],
        [  5.2911,  -2.4824,  -3.6877],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.7485,  -9.9811,  -3.2355],
        [ -2.7572,   7.7194,   5.9580],
        [  1.4749,  -4.7892,  -0.8089],
        [  9.0424,  -7.6779,  -7.5286],
        [ 11.5083,  -4.1177,   0.2389],
        [ -5.1046,  -9.1610,   2.3418],
        [  2.2659,  13.7894,   6.5436],
        [  6.0890,  -0.8382,   2.9166],
        [  0.1460,   7.8340,  -9.4614],
        [ -0.7599, -10.8995,  -6.0366],
        [  9.0051,   9.9234,   5.4411],
        [  5.2188,  -2.5839,   2.1833],
        [ -1.1885, -10.8211,   0.0865],
        [  9.0652,   9.8597,  -3.5949],
        [ -9.4773,  -1.0024,   5.2599],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.6970,   6.9826,   9.7350],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.0611,  -6.0302,   4.6962],
        [  0.1252, -11.0062,   7.2564],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.1064,  13.5607,   6.7605],
        [  6.2036,   2.4030,  -1.2500],
        [ 10.0995,  -6.4739,   2.0974],
        [  5.4426, -10.1181,  -5.1268],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.5123,  -0.5339,  -1.0728],
        [  2.4572,   7.1085,   7.7376],
        [ -2.1533, -10.5778,   1.0112],
        [ -8.3900,   3.3727,   3.1650],
        [  0.1252, -11.0062,   7.2564],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.3200,   7.1756,   1.1938],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -8.5506,  -5.1154,   6.3423],
        [ -1.0482, -10.8487,  -5.9066],
        [  0.2732,  -4.7619,  -1.1577],
        [  1.3705,  13.9723,   7.9266],
        [ -7.4808,   4.7462,   2.0266],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -7.4602,   4.7718,  -4.6277],
        [ -5.1369,   6.7755,   8.3836],
        [ -3.9700,   7.3453,  -4.1486],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.2902,   0.0451,   3.7116],
        [  7.7750,  11.0665,   2.8364],
        [  8.2422,  -8.4041,  -1.4962],
        [  2.7558,  -4.4844,   0.0636],
        [ 11.7349,  -3.5838,   2.1644],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.8994,   6.8675,  -9.5987],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.3305,   1.6119,   1.1538],
        [  7.1821,  -9.1845,   9.8548],
        [ -9.4748,  -0.7765,  -5.6695],
        [  2.7976,  -4.4682,  -4.2020],
        [ -4.5778,  -9.5013,  -5.0667],
        [  4.7482, -10.3907,   5.9149],
        [ 11.9936,  -2.8677,   6.0614],
        [ -9.2707,   0.9661,   4.4863],
        [  5.1787,  -2.6383,  -4.2533],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.2076,  13.8032,  -1.5099],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.1824,  -4.6647,   1.1985],
        [ -9.3604,   0.4855,  -3.0158],
        [ 11.7794,   5.5086,  -9.6929],
        [  7.9403,  -8.6455,  -8.7433]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0167, 0.0185,  ..., 1.1164, 1.0899, 1.0034],
        [0.0167, 0.0000, 0.0215,  ..., 1.1219, 1.0775, 0.9967],
        [0.0185, 0.0215, 0.0000,  ..., 1.1348, 1.0984, 1.0178],
        ...,
        [1.1164, 1.1219, 1.1348,  ..., 0.0000, 1.2173, 0.7811],
        [1.0899, 1.0775, 1.0984,  ..., 1.2173, 0.0000, 0.4456],
        [1.0034, 0.9967, 1.0178,  ..., 0.7811, 0.4456, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.5748624 ,  5.79562691,  5.76165934,  5.81481044,  5.91677188,
        5.94535529, 12.41366754,  9.34127013, 11.23201619, 10.69963954,
        8.29022197,  7.32950385,  9.18358562,  4.91038884,  7.53884104,
        9.75075273,  7.8575648 ,  5.05444288, 10.92597196,  8.8353674 ,
       10.66631784, 12.49291768,  4.75303127,  8.08320235,  6.83515201,
        7.46053281, 11.63724792, 12.9089559 , 11.05672136,  5.84450127,
       10.794755  , 12.91505111, 11.51811603,  8.19703512,  5.01114719,
       11.86239961, 12.22276278, 10.48721382, 13.97430191,  6.1463896 ,
        7.83534987, 10.92597196, 13.40024224,  5.82345867, 10.88618114,
       13.39372426,  9.53015376,  9.96391045,  7.48538905,  6.1561646 ,
        5.91677188, 10.06703358, 11.00694688, 10.52587243, 13.91198191,
        6.652742  , 11.99633819, 11.4890861 ,  6.14086582, 12.52372629,
        7.52117132, 10.794755  ,  9.04255655, 11.00694688, 12.22276278,
        7.54127425, 12.04924159,  9.96391045, 10.8992524 ,  4.76969364,
       14.0393949 ,  8.859416  , 10.57302546,  8.85572292,  8.50268322,
        8.34949383,  7.09887557,  9.94285511,  6.29034958, 13.5247215 ,
       11.77124373,  5.26342203, 12.26997165, 12.42554808,  9.60080639,
       10.37998078,  9.68324553,  9.75401449,  7.4544866 , 10.63038565,
        6.53250937, 11.65923396,  9.50654967,  5.27178363, 10.54658597,
       11.42420508, 12.33167382,  9.32094055,  5.81199422, 10.24557047,
       13.97857591, 11.9120321 ,  5.14999799,  9.37296102, 13.00380624,
       11.73848065])}
{'P': tensor([[3.5535e-02, 1.5095e-01, 1.4626e-01,  ..., 2.7128e-05, 5.4771e-04,
         4.6178e-03],
        [2.1542e-03, 5.2073e-02, 5.7060e-03,  ..., 3.0718e-05, 3.7269e-04,
         2.4533e-03],
        [1.8423e-03, 1.8513e-01, 3.2660e-02,  ..., 2.0133e-05, 2.3995e-04,
         2.2266e-03],
        ...,
        [8.0207e-04, 2.9293e-04, 3.6873e-04,  ..., 3.4999e-02, 1.8569e-04,
         6.6755e-05],
        [1.7814e-03, 1.9384e-03, 1.6026e-03,  ..., 2.0674e-05, 6.6358e-02,
         4.3018e-04],
        [3.9488e-04, 2.2002e-03, 1.5848e-03,  ..., 8.1938e-07, 2.8018e-05,
         2.7475e-02]]), 'X': tensor([[ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.8251e-02, 5.2304e-02,  ..., 9.1603e-01, 4.3050e-01,
         2.5319e-01],
        [3.8251e-02, 0.0000e+00, 2.2031e-02,  ..., 9.4276e-01, 4.5479e-01,
         2.3428e-01],
        [5.2304e-02, 2.2031e-02, 1.0727e-04,  ..., 9.3657e-01, 4.7624e-01,
         2.1227e-01],
        ...,
        [9.1603e-01, 9.4276e-01, 9.3657e-01,  ..., 0.0000e+00, 1.0715e+00,
         8.9358e-01],
        [4.3050e-01, 4.5479e-01, 4.7624e-01,  ..., 1.0715e+00, 2.1453e-04,
         6.8330e-01],
        [2.5319e-01, 2.3428e-01, 2.1227e-01,  ..., 8.9358e-01, 6.8330e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.60064319,  5.94909709,  5.87305005,  5.80860165,  6.01356004,
        6.02588517,  9.17016857, 10.14436912,  6.58184474,  6.76056625,
        6.81084724,  4.79003361, 12.87752052, 12.3828957 ,  6.50409372,
       10.35093485,  5.43544626,  5.76165934,  6.35029274, 12.91505111,
        8.83341608,  9.41605684, 11.67875855,  7.53063885, 12.12641969,
        5.30158497, 10.27572673,  9.70108409, 10.74409834,  9.94173708,
       11.18839169, 13.68672855,  6.39520397,  9.36378438,  5.68942572,
       10.92597196, 12.67680439, 11.35182589,  8.23525433, 11.28093507,
        8.89301516, 12.35625694, 10.04916678,  5.91854397,  8.68832398,
       13.18278606,  5.51177501, 10.06850677, 11.85743425,  6.02779201,
       11.75438006, 10.06703358,  5.08372149, 13.09426142,  9.66977802,
        5.14371005,  7.37576154, 13.20988996, 11.50791859,  6.1463896 ,
       11.64746882, 12.27993098, 10.61417181,  4.92497514, 13.3614598 ,
       11.63765273, 11.87412439,  6.02779201,  7.83486943, 12.3489324 ,
       10.92597196,  7.80625981, 13.36092624, 13.59170809,  8.62329466,
        6.39187285, 12.41012046,  9.53015376,  9.84314712,  7.50564223,
        5.84450127,  6.01356004,  9.78285424,  9.99145096,  8.31625171,
        6.07992623, 12.9089559 , 14.03505624, 11.07637531,  7.54620849,
        5.47866221,  9.91668182,  9.19837792,  7.70897675, 12.53047489,
       13.46553623,  7.16598273, 10.37698435, 14.07623266,  8.27853076,
       10.90638066,  9.38232121,  8.50268322,  9.97220799,  5.39723241,
       12.17451685])}
{'P': tensor([[2.8770e-02, 1.8811e-01, 1.5250e-01,  ..., 1.2958e-03, 4.1517e-02,
         2.4627e-04],
        [1.8789e-03, 3.0993e-02, 2.1778e-01,  ..., 1.5984e-03, 7.7605e-02,
         4.8106e-04],
        [2.1345e-03, 3.9809e-03, 4.0285e-02,  ..., 2.3023e-03, 1.2065e-01,
         7.8300e-04],
        ...,
        [7.3607e-04, 1.0551e-03, 1.2655e-03,  ..., 1.6424e-01, 3.6054e-03,
         1.0042e-02],
        [1.6765e-03, 2.2892e-03, 2.5834e-03,  ..., 3.6498e-03, 1.1585e-01,
         1.0694e-02],
        [1.1261e-06, 1.3004e-06, 1.4037e-06,  ..., 3.6582e-06, 3.5516e-06,
         6.8945e-03]]), 'X': tensor([[  4.4290,  -3.4596,   1.4181],
        [  5.1667,  -2.6543,   1.2879],
        [  5.3860,  -2.3418,   1.0826],
        [  5.2188,  -2.5839,   2.1833],
        [  5.6202,  -1.9508,   0.7471],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.8930,  12.3592,  -8.9201],
        [  6.3549,   0.7096,   6.9198],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.8940, -11.0402,  -0.9783],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.1231,  -3.5739,   8.4708],
        [ -2.3317,   7.8034,  -3.6274],
        [ -9.0574,   1.7801,   8.8986],
        [ -6.8466,  -7.6488,   1.9648],
        [ -1.9600,   7.8577,   6.9121],
        [ -0.8455, -10.8853,  -6.7400],
        [  4.5325,  -3.3643,   6.7699],
        [  5.9548,  -1.2273,  -2.6520],
        [ -4.1296,   7.2803,  -2.7194],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.2022,   2.4092,   3.0349],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -3.5173, -10.0623,   4.8280],
        [ -6.7745,  -7.7264,   8.4684],
        [  4.7813,  -3.1139,   8.2691],
        [  0.9445,   7.6736,   0.5095],
        [  6.2419,  -0.2313,   4.2440],
        [  3.9515,  -3.8425,   6.1729],
        [  4.5491,  -3.3485,   6.8412],
        [ 12.4691,   2.8525,  -5.5110],
        [ -2.0721,   7.8432,   6.6380],
        [ -1.2454,   7.9127,  -0.5715],
        [ -0.0282,   7.8575,  -1.1746],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -1.3940, -10.7772,   6.7835],
        [ 10.9931,   7.2107,   7.7933],
        [  0.6396,  14.0772,  -6.3786],
        [  7.8375,  -8.7240,  -9.8913],
        [  3.9770, -10.6341,   8.0783],
        [ 12.4837,   2.7510,   0.4449],
        [  2.8474,   6.8979,   6.6457],
        [ -0.8569,   7.9156,  -6.5557],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.9788,   2.0208,  -3.2867],
        [  1.4489,   7.5256,   4.4529],
        [ -9.0574,   1.7801,   8.8986],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.7544,  -3.1425,  -1.5315],
        [  2.0937,  13.8293,  -2.3533],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.4773,  -1.0024,   5.2599],
        [  7.0642,  11.6104,  -9.8316],
        [  5.9548,  -1.2273,  -2.6520],
        [  7.7750,  11.0665,   2.8364],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.3087,   1.8053,  -1.9481],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.6817,   6.3354,   6.1233],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.4242, -10.1261,   3.9595],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.2860,   7.1916,  -6.1011],
        [  6.3200,   0.2690,   1.5900],
        [  1.7903,  -4.7467,   0.1168],
        [  6.1741,  -0.5329,   3.1913],
        [  1.8766,   7.3694,   1.1887],
        [ -3.6381,   7.4683,   7.3544],
        [  3.5965,  -4.0754,  -3.5035],
        [  1.8447,  -4.7372,   8.1235],
        [ 11.7162,   5.6728,  -8.9372],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.0087,   4.0173,   9.1091],
        [  1.7000,  -4.7609,   2.7935],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.0574,   1.7801,   8.8986],
        [  1.1994,   7.6035,   2.4295],
        [ 10.8218,  -5.4177,   6.8573],
        [  3.3839,  -4.1966,   8.3574],
        [  7.1697,  11.5343,  -0.7276],
        [  1.1006,   7.6318,  -2.4877],
        [  6.3122,  -9.6994,  -6.3282],
        [  4.2612, -10.5514,  -6.6268],
        [ -4.5778,  -9.5013,  -5.0667],
        [  8.2154,  -8.4262,   8.0171],
        [  7.2340,  -9.1505,  -5.8321],
        [  9.0652,   9.8597,  -3.5949],
        [  0.8700,  -4.8132,  -7.8042],
        [ 12.6038,   0.7665,  -3.4439],
        [  0.6952,  -4.8060,   8.8976],
        [  4.9531,   5.0576,   5.6794],
        [  6.3303,   1.6145,  -8.4083],
        [ 12.2914,   3.8037,   6.0978],
        [ -2.0840,   7.8415,  -7.8657],
        [  2.0937,  13.8293,  -2.3533],
        [  8.8474,  10.0868,  -8.8644],
        [  2.6078, -10.9229,   6.2869],
        [  2.2182,  13.8007,   5.1889],
        [ -5.3382,   6.6533,   7.8054],
        [  2.5469,  13.7192,  -9.6935],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3551,   0.7149,   1.3344],
        [  6.2026,  12.1774,  -1.5887]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.8454e-02, 3.1211e-02,  ..., 4.8177e-01, 4.6066e-02,
         3.7672e-01],
        [1.8454e-02, 0.0000e+00, 1.4064e-02,  ..., 4.6955e-01, 3.7916e-02,
         3.5831e-01],
        [3.1211e-02, 1.4064e-02, 1.0727e-04,  ..., 4.5608e-01, 4.4462e-02,
         3.4608e-01],
        ...,
        [4.8177e-01, 4.6955e-01, 4.5608e-01,  ..., 0.0000e+00, 4.9235e-01,
         3.3620e-01],
        [4.6066e-02, 3.7916e-02, 4.4462e-02,  ..., 4.9235e-01, 1.0727e-04,
         3.5346e-01],
        [3.7672e-01, 3.5831e-01, 3.4608e-01,  ..., 3.3620e-01, 3.5346e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.62005629,  5.80860165,  5.87305005,  5.82345867,  5.94909709,
        6.01356004,  9.15866607, 13.6922425 ,  6.39438044, 10.33712258,
       11.07637531, 13.97857591,  9.79814997,  8.14433542,  9.23071698,
       10.26545792,  8.09842704, 10.91805683,  5.64464807,  6.07992623,
        8.36995953,  4.91652556,  6.65368551, 10.52357479, 10.65929968,
       10.27572673,  5.70591986,  7.73151309,  6.24614905,  5.51177501,
        5.64865173, 12.79126785,  8.11227668,  8.01009083,  7.8575648 ,
       12.49291768, 10.86693708, 13.14690137, 14.09176259, 11.72749923,
       11.35345345, 12.78327037,  7.46248947,  7.96181288, 13.1399181 ,
        9.20340031,  7.66377554,  9.23071698, 13.00413829,  5.69913817,
       13.98691426, 11.48434767,  9.53015376, 13.59056613,  6.07992623,
       13.5247215 , 12.67576792,  6.5618968 ,  8.05242591,  7.32754882,
        5.18277858, 11.48734978,  5.35774573,  9.9353921 ,  7.54620849,
        6.32572323,  5.07306274,  6.19707944,  7.60463349,  8.30729352,
        5.43544626,  5.08372149, 13.01728313, 10.52357479,  8.95983586,
        5.05534609,  7.822854  ,  9.23071698,  7.69752262, 12.10222204,
        5.39098669, 13.58101794,  7.71075226, 11.57249984, 11.37939695,
       10.54658597, 11.76830981, 11.66453059, 13.39372426,  4.89121805,
       12.62711252,  4.85603886,  7.07902036,  6.53290663, 12.86648771,
        8.1137452 , 13.98691426, 13.41713661, 11.22993218, 13.97779839,
        8.53014915, 13.95361098, 13.1399181 ,  6.48486171,  6.39520397,
       13.6660671 ])}
{'P': tensor([[4.6801e-02, 1.2020e-01, 1.0312e-01,  ..., 4.0231e-05, 6.7984e-03,
         8.2144e-04],
        [2.1590e-03, 5.1239e-02, 9.9866e-03,  ..., 2.5163e-05, 1.5581e-03,
         5.2090e-04],
        [3.4025e-03, 2.4799e-02, 5.6421e-02,  ..., 2.3847e-05, 3.1431e-03,
         2.2328e-03],
        ...,
        [2.0404e-03, 8.6842e-04, 6.4919e-04,  ..., 3.7412e-02, 4.4199e-04,
         1.0582e-05],
        [8.2359e-04, 4.5012e-03, 4.2674e-03,  ..., 1.2610e-06, 4.5558e-02,
         1.4654e-03],
        [2.3278e-04, 2.7393e-04, 6.1713e-04,  ..., 4.1060e-06, 1.1236e-04,
         1.0339e-01]]), 'X': tensor([[  4.5057,  -3.3894,  -0.8160],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.6202,  -1.9508,   0.7471],
        [  4.7544,  -3.1425,  -1.5315],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.3124,   0.2057,  -0.4593],
        [  6.3301,   0.3638,  -7.5685],
        [ 11.4495,   6.3026,   6.8257],
        [  4.6231,   5.4556,   8.4973],
        [  0.9036,   7.6840,   9.9237],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.7735,   2.5640,  -6.3313],
        [  1.9312,  13.8649,   8.3563],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.1222,  -0.7261,   8.8204],
        [  6.0073,   3.0986,   6.7937],
        [ -8.5859,  -5.0393,   0.5476],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.0305,   3.0296,   0.1375],
        [  4.1637,  13.1898,  -1.9189],
        [ 12.4597,  -0.9136,  -4.8763],
        [  9.0004,  -7.7195,   4.6945],
        [  6.3119,   1.7797,   9.9186],
        [  2.1379,  -4.6757,   6.4981],
        [  5.7802,  -1.6363,  -6.2782],
        [ -5.3736,   6.6310,  -1.5174],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -3.4100,   7.5436,  -7.6980],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.1892, -10.8220,  -6.8409],
        [  4.5057,  -3.3894,  -0.8160],
        [ -5.1974,   6.7396,   3.3484],
        [  5.3860,  -2.3418,   1.0826],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.1139,   7.6281,  -2.5680],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.1984,  -0.4327,   1.5438],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.2480, -11.0153,  -3.0183],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.6703,   6.3443,  -5.4401],
        [  1.7122,  -4.7591,  -3.6734],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.3569,   0.7591,   6.5011],
        [ -8.3539,  -5.5111,   0.3373],
        [  1.1350,  14.0104,   0.6321],
        [ -7.6525,  -6.6534,   5.1721],
        [ -8.8835,  -4.3178,   9.3080],
        [  5.2911,  -2.4824,  -3.6877],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.8979,   3.3958,  -3.8713],
        [  6.3301,   0.3638,  -7.5685],
        [ -2.0445,   7.8469,  -0.5857],
        [  3.0876,   6.7527,  -5.3540],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.1623,  -3.4288,  -4.9961],
        [  3.6154,  -4.0640,  -7.9698],
        [ -8.8934,   2.2597,   3.5831],
        [ -6.7745,  -7.7264,   8.4684],
        [  6.9314,  -9.3435,  -1.2729],
        [  4.9531,   5.0576,   5.6794],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.4149,  -4.6005,  -4.3339],
        [  0.9392,  -4.8143,  -3.4649],
        [ -8.1426,   3.8048,   9.3542],
        [ -3.1204, -10.2345,   9.2288],
        [  5.4242, -10.1261,   3.9595],
        [ 12.4837,   2.7510,   0.4449],
        [ 10.6283,  -5.7253,   7.4878],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -6.7992,   5.4991,   9.9186],
        [  4.4915,   5.5990,   7.0405],
        [  5.1793,   4.7461,   0.3899],
        [ 12.4691,   2.8525,  -5.5110],
        [  1.3670,  -4.7990,   3.2603],
        [  5.8323,  -1.5228,   5.1132],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.2905,   1.9366,   2.1169],
        [  6.0031,  12.2958,  -6.5992],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3119,   1.7797,   9.9186],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.0611,  -6.0302,   4.6962],
        [  3.3744,   6.5623,  -6.3636],
        [  5.9655,   3.2172,  -7.5953],
        [  3.3839,  -4.1966,   8.3574],
        [  1.4489,   7.5256,   4.4529],
        [  5.8902,  -1.3887,   4.0388],
        [ 12.3357,   3.6001,   0.7903],
        [ -9.1624,  -3.4286,   2.0647],
        [ -2.0445,   7.8469,  -0.5857],
        [  9.4925,   9.3811,   8.9842],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.0729,   7.2878,   7.0880],
        [ -0.8675, -10.8815,   6.8561],
        [  0.5850,  -4.7982,   9.6639],
        [  8.2154,  -8.4262,   8.0171],
        [  0.1252, -11.0062,   7.2564],
        [  6.3542,   0.6934,  -3.3840],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 11.2767,  -4.6006,   0.7868],
        [  6.3119,   1.7797,   9.9186]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0951, 0.1077,  ..., 0.9201, 0.1146, 0.6112],
        [0.0951, 0.0000, 0.1950,  ..., 0.9183, 0.2097, 0.7004],
        [0.1077, 0.1950, 0.0000,  ..., 0.9872, 0.0636, 0.5055],
        ...,
        [0.9201, 0.9183, 0.9872,  ..., 0.0000, 0.9330, 1.3089],
        [0.1146, 0.2097, 0.0636,  ..., 0.9330, 0.0000, 0.5078],
        [0.6112, 0.7004, 0.5055,  ..., 1.3089, 0.5078, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.63824971,  6.02789634,  5.94909709,  5.69913817,  6.10774544,
        6.31576264,  6.34059021, 13.06957425,  7.15100303,  7.73692436,
        8.43626114,  9.14044984, 13.99877413,  5.5205189 ,  6.16512856,
        6.7593917 ,  9.95552626,  9.75401449, 10.52587243,  6.74872981,
       13.83139024, 12.49317563, 11.85743425,  6.55800267,  5.14124612,
        6.00732193,  8.53501777,  9.50654967,  8.27853076,  9.88310614,
       11.28215429,  5.63824971,  8.51088913,  5.87305005, 10.04916678,
        7.70897675, 13.20597932,  6.2134961 ,  5.5366698 , 11.018084  ,
       11.89855267,  7.32950385,  5.05774295, 12.85032836,  8.85572292,
        6.4020404 , 10.00794468, 14.05633441, 10.14044489,  9.87720102,
        5.84450127,  4.91296205,  6.80559669,  6.34059021,  8.10886021,
        7.42512621,  7.89745107,  9.78286733,  5.43943791,  9.17595899,
       10.27572673, 11.63383221,  7.07902036, 11.63502679,  5.19577146,
        4.90505566,  8.98765644, 10.69963954, 11.48734978, 12.78327037,
       12.07225469, 10.65929968,  9.55720651,  9.79039371,  8.74470218,
        7.17789666,  7.02496223, 12.79126785,  4.98989906,  6.02779201,
       13.27994304,  6.58184474, 13.6829843 , 10.8207297 ,  6.55800267,
       10.0674887 , 10.06703358,  7.37902766,  6.77777606,  5.39098669,
        7.66377554,  6.05165248, 12.85032836,  9.78285424,  8.10886021,
       13.34587027,  8.14433542,  7.57685908, 10.91601655,  4.83371341,
       11.76830981, 11.00694688,  6.39187285, 10.52587243, 12.17901033,
        6.55800267])}
{'P': tensor([[2.5897e-02, 1.8990e-01, 1.6101e-01,  ..., 3.1087e-02, 3.9396e-04,
         4.1866e-05],
        [2.3909e-03, 3.1559e-02, 2.5050e-01,  ..., 2.4406e-02, 6.0538e-04,
         4.7534e-05],
        [3.1220e-03, 4.3901e-03, 4.6317e-02,  ..., 2.6456e-02, 1.0062e-03,
         6.8623e-05],
        ...,
        [1.5869e-02, 2.8250e-02, 3.7188e-02,  ..., 2.0036e-01, 1.6157e-03,
         9.5757e-05],
        [9.3338e-05, 1.0858e-04, 1.1867e-04,  ..., 6.6137e-05, 7.8834e-02,
         2.1121e-04],
        [6.1154e-04, 7.0133e-04, 7.6003e-04,  ..., 2.5124e-04, 1.1919e-02,
         2.2607e-01]]), 'X': tensor([[ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0109, 0.0195,  ..., 0.0733, 0.6750, 0.8533],
        [0.0109, 0.0000, 0.0089,  ..., 0.0785, 0.6677, 0.8458],
        [0.0195, 0.0089, 0.0000,  ..., 0.0857, 0.6599, 0.8379],
        ...,
        [0.0733, 0.0785, 0.0857,  ..., 0.0000, 0.7453, 0.9232],
        [0.6750, 0.6677, 0.6599,  ..., 0.7453, 0.0000, 0.1787],
        [0.8533, 0.8458, 0.8379,  ..., 0.9232, 0.1787, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.64464807,  5.76165934,  5.81481044,  5.86519525,  5.87132718,
        5.94535529, 13.00413829,  5.16077764,  8.37954853,  6.29034958,
        4.90676242,  9.58072286,  9.99145096, 10.8992524 ,  9.76896904,
       13.72594022, 13.85158417,  8.89301516,  6.01356004,  5.10444097,
        7.50035355, 11.63502679, 12.58006038,  5.20659892, 11.46176259,
        9.96152788,  9.12220271,  9.48969978,  7.96575782,  6.1463896 ,
        9.48337063, 10.4606265 , 10.794755  , 11.63765273,  8.06643487,
       11.67875855,  8.8737556 ,  8.86082124,  8.89301516, 11.83509372,
        5.64464807,  8.31625171, 12.18998195,  6.02588517, 14.0393949 ,
        5.70405525,  8.51088913, 12.3828957 , 10.33712258, 10.74409834,
        8.89301516, 10.87409005,  5.39098669,  8.50268322,  7.15990676,
        4.87083348,  9.12220271,  8.22728399,  5.05534609, 10.23946247,
        6.21275016, 11.785566  ,  6.39187285,  5.63824971, 12.42554808,
        7.37576154, 13.34587027,  6.02588517,  9.69629   ,  7.57946459,
       13.26590071,  8.90405823,  8.10886021, 10.4606265 ,  8.44940661,
        8.40445872,  7.72291946,  9.78285424, 12.97370371, 10.00410625,
       12.17451685, 10.19036081, 13.26209545, 10.7506423 , 13.103709  ,
        8.04718433, 13.27717982, 11.1341798 , 13.14777433,  5.79543038,
       13.66051884,  9.76896904, 13.744387  ,  4.91038884,  9.48969978,
        4.91296205, 12.21532282,  5.04028538,  5.56809335,  9.97220799,
        5.8828986 , 11.54071859,  5.1224079 ,  5.91854397,  7.06272256,
        5.34250225])}
{'P': tensor([[2.8602e-02, 1.5575e-01, 1.3224e-01,  ..., 8.0489e-05, 4.9643e-05,
         1.5928e-05],
        [3.5225e-03, 7.3879e-02, 7.3755e-02,  ..., 1.3592e-04, 1.6984e-04,
         2.9918e-05],
        [2.5840e-03, 3.0762e-02, 5.1862e-02,  ..., 1.1585e-04, 1.2417e-04,
         2.6464e-05],
        ...,
        [5.7829e-04, 5.6101e-04, 7.3786e-04,  ..., 7.2359e-02, 2.7814e-05,
         1.5879e-02],
        [3.1791e-04, 1.8899e-04, 1.9010e-04,  ..., 2.0864e-04, 1.2264e-01,
         2.4270e-05],
        [9.5371e-05, 1.0329e-04, 1.3579e-04,  ..., 6.0566e-03, 1.2023e-05,
         1.2203e-01]]), 'X': tensor([[  4.5340,  -3.3629,   8.9652],
        [  5.5294,  -2.1106,   9.0626],
        [  5.5626,  -2.0536,   8.4433],
        [  5.5101,  -2.1433,   9.5442],
        [  5.6022,  -1.9833,   8.3081],
        [  5.4429,  -2.2529,   9.7306],
        [ 11.5966,   5.9666,   6.0083],
        [ -0.8675, -10.8815,   6.8561],
        [ 12.2589,  -1.9312,   9.8657],
        [ -5.6651,   6.4380,   1.3660],
        [  9.6662,   9.1725,   8.1056],
        [  0.1532,  -4.7422,  -7.0145],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -5.6792,  -8.7352,   4.0318],
        [  0.2480, -11.0153,  -3.0183],
        [ -8.1666,   3.7653,   3.6075],
        [  9.3560,   9.5391,   9.4943],
        [ -0.5709,   7.9055,   7.6089],
        [  4.8653,   5.1694,  -4.0823],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.1420,   6.9331,  -2.4779],
        [  9.6587,   9.1817,  -0.8212],
        [  3.9022,  13.2906,   5.5788],
        [ -8.5506,  -5.1154,   6.3423],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -7.0120,   5.2825,   7.7762],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -5.9940,   6.1985,   5.9566],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -5.6792,  -8.7352,   4.0318],
        [  9.6614,   9.1784,  -1.2087],
        [  1.4489,   7.5256,   4.4529],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.3551,   0.7149,   1.3344],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.3515,   0.6418,  -6.5823],
        [  5.8836,   3.4315,  -4.0327],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.4753,  -0.8012,  -5.2050],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.3048,   0.1469,  -9.1691],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.8934,   2.2597,   3.5831],
        [  6.1752,   2.5242,   3.5874],
        [  5.5101,  -2.1433,   9.5442],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.7976,  -4.4682,  -4.2020],
        [ -9.2345,  -3.1345,   3.9715],
        [ -1.5408,   7.8978,   5.3470],
        [ -9.2373,  -3.1225,  -8.9763],
        [  0.9074,  -4.8139,   8.1674],
        [  6.3211,   1.7017,  -0.2164],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -0.0254, -10.9932,   3.2200],
        [  1.9204,   7.3518,   8.4781],
        [ -8.1905,   3.7253,   4.1175],
        [  6.2022,   2.4092,   3.0349],
        [  6.1372,   2.6726,  -6.7885],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.2975,   0.0947,   5.1791],
        [ -7.2515,   5.0195,   4.6434],
        [  4.5800,   5.5034,  -3.9432],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.2535,  -0.1709,  -9.4102],
        [  5.7218,  -9.9934,   6.5764],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.1066,  -0.7799,  -4.1746],
        [ -6.1914,  -8.2997,   8.5380],
        [ -8.1204,  -5.9308,   4.0761],
        [ -6.2200,  -8.2737,   5.5079],
        [  6.3549,   0.7096,   6.9198],
        [  2.4697,  -4.5836,   1.4593],
        [ -6.4368,   5.8356,   3.3014],
        [  8.2154,  -8.4262,   8.0171],
        [  6.3542,   0.6934,  -3.3840],
        [ -5.6169,  -8.7844,   8.9057],
        [ -9.4624,  -0.4633,  -7.3918],
        [  2.1440, -10.9811,   8.5281],
        [  6.3497,   1.3680,  -1.9076],
        [  5.4732,  -2.2041,  -4.3830],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.1813,  -3.3550,   1.8827],
        [ -7.8359,   4.2735,   7.2943],
        [  1.7122,  -4.7591,  -3.6734],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.3211,   1.7017,  -0.2164],
        [ -9.4721,  -1.3097,  -7.4377],
        [  4.3523, -10.5232,   8.2138],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.2576,   2.1359,  -3.3398],
        [  6.1737,   2.5305,   3.9049],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.0054,  -1.0902,   8.3079],
        [  9.2692,   9.6370,   7.0746],
        [ -5.9938,  -8.4745,   3.9735],
        [  7.2340,  -9.1505,  -5.8321],
        [ 11.5456,   6.0861,   8.1101],
        [ -6.8466,  -7.6488,   1.9648],
        [ -7.4807,   4.7463,   2.0616],
        [  1.7218,  -4.7576,  -3.3617],
        [ -7.8359,   4.2735,   7.2943],
        [  3.4920,  -4.1366,  -8.3955]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0167, 0.0210,  ..., 0.7600, 0.6670, 0.9365],
        [0.0167, 0.0000, 0.0121,  ..., 0.7607, 0.6836, 0.9354],
        [0.0210, 0.0121, 0.0000,  ..., 0.7486, 0.6827, 0.9233],
        ...,
        [0.7600, 0.7607, 0.7486,  ..., 0.0000, 0.9315, 0.1949],
        [0.6670, 0.6836, 0.6827,  ..., 0.9315, 0.0000, 1.1193],
        [0.9365, 0.9354, 0.9233,  ..., 0.1949, 1.1193, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.64500512,  5.91854397,  5.92952538,  5.91222346,  5.94293185,
        5.89074104, 13.04156075, 10.91601655, 12.41012046,  8.57560275,
       13.32557133,  4.74468406, 10.09979747, 10.41909211, 11.018084  ,
        8.99276564, 13.3614598 ,  7.92607342,  7.09887557, 12.1460931 ,
       13.12299145, 13.32645552, 13.85158417,  9.96391045, 12.27835996,
        8.77913428, 13.61801526, 12.25806559,  8.62260327, 12.12448912,
       10.41909211, 13.32613445,  7.66377554, 13.18278606,  6.39520397,
       10.1371552 ,  6.383897  ,  6.81117802,  8.5724042 ,  9.50913849,
       12.84056604,  6.30647746,  7.8575648 ,  9.17595899,  6.67121748,
        5.91222346,  5.41348121,  5.27178363,  9.75200848,  8.04664886,
        9.75075273,  4.89869205,  6.54615887,  9.84314712, 12.52372629,
       10.99326577,  7.5984809 ,  8.99791316,  6.65368551,  6.69389431,
       12.10942412,  6.29821673,  8.81930525,  7.15990676,  9.78147542,
        6.25586297, 11.51557166,  5.19577146,  6.1561646 , 10.35464916,
       10.05558893, 10.35093485,  6.39438044,  5.20659892,  8.68832398,
       11.76830981,  6.39187285, 10.42667093,  9.47369628, 11.18839169,
        6.49538841,  5.90035324, 12.26997165, 12.13568598,  6.83083863,
        9.7751189 ,  8.92549752,  5.05774295, 11.99633819,  6.54615887,
        9.562174  , 11.38774149,  5.84450127,  6.61211224,  6.67218687,
       12.16901615,  6.10360073, 13.37121923, 10.37998078, 11.66453059,
       13.05148826, 10.26545792,  8.85939626,  5.05963217,  8.92549752,
        5.41348121])}
{'P': tensor([[2.1398e-02, 1.4570e-01, 1.2585e-01,  ..., 1.0381e-01, 1.0591e-05,
         2.8103e-06],
        [2.4705e-03, 3.0265e-02, 2.4022e-01,  ..., 2.1102e-01, 1.4670e-05,
         4.7665e-06],
        [3.5396e-03, 4.7152e-03, 4.9747e-02,  ..., 4.0850e-01, 2.4106e-05,
         8.6229e-06],
        ...,
        [7.2121e-03, 8.8934e-03, 9.8625e-03,  ..., 1.5587e-01, 7.3785e-05,
         3.1087e-05],
        [9.1906e-05, 7.2358e-05, 6.4735e-05,  ..., 4.6399e-05, 1.5603e-02,
         4.4512e-06],
        [1.7942e-05, 1.7209e-05, 1.7009e-05,  ..., 1.5116e-05, 4.9040e-03,
         1.8776e-01]]), 'X': tensor([[  4.5491,  -3.3485,   6.8412],
        [  4.9957,  -2.8705,   6.5978],
        [  5.1886,  -2.6250,   6.4677],
        [  5.3602,  -2.3808,   7.2098],
        [  5.3804,  -2.3504,   7.2744],
        [  5.5424,  -2.0885,   7.2487],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 12.4691,   2.8525,  -5.5110],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.8745,   2.3097,   3.5600],
        [  6.2036,   2.4030,  -1.2500],
        [  2.3261,   7.1726,   6.2522],
        [  1.7000,  -4.7609,   2.7935],
        [  6.3087,   1.8053,  -1.9481],
        [  1.4306,  13.9620,  -0.2075],
        [ -2.1533, -10.5778,   1.0112],
        [  6.7792,  11.8084,   1.6251],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.0047,   3.1062,  -6.7914],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.7156,  -3.1831,  -8.0224],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.7659,   7.4127,   4.7871],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.2847,  -4.5848,  -4.7503],
        [  4.9957,  -2.8705,   6.5978],
        [ -3.1204, -10.2345,   9.2288],
        [  4.5325,  -3.3643,   6.7699],
        [  2.3367,   7.1676,  -9.5263],
        [  0.1460,   7.8340,  -9.4614],
        [  6.7792,  11.8084,   1.6251],
        [ -5.0550,   6.8231,  -9.2502],
        [ 12.5199,  -0.4711,   3.0436],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.0750,  -6.0072,  -5.8267],
        [  1.7218,  -4.7576,  -3.3617],
        [ -8.4573,   3.2448,   5.4370],
        [  6.2630,   2.1057,  -5.2136],
        [ -0.9715, -10.8630,   0.3555],
        [ 11.6634,   5.8050,   9.1590],
        [ -6.9838,  -7.4965,   0.9796],
        [  6.2675,  12.1379,  -8.9720],
        [ -2.5056,   7.7720,   5.5878],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.8930,  12.3592,  -8.9201],
        [  4.2339,  -3.6268,   6.3125],
        [  9.3836,   9.5075,  -5.9278],
        [ -9.1459,   1.4780,  -4.2403],
        [ 10.9931,   7.2107,   7.7933],
        [  6.0345,   3.0175,  -1.4424],
        [  5.1472,  -2.6801,   8.1194],
        [  6.3583,   1.1729,   8.7835],
        [  8.6420,  -8.0580,   6.5781],
        [  0.3493,  -4.7727,  -6.5682],
        [  6.9693,  -9.3200,  -8.3822],
        [ 11.6165,  -3.8715,   6.4594],
        [  5.3858,   4.4260,  -9.6769],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -8.1666,   3.7653,   3.6075],
        [ 12.5650,   2.0132,   4.6272],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.3444,   0.5821,   6.9101],
        [ 11.5966,   5.9666,   6.0083],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.9531,   5.0576,   5.6794],
        [  9.4516,  -7.2501,  -1.8868],
        [ -8.8058,   2.4850,  -1.7425],
        [  9.5203,   9.3483,   5.8781],
        [  6.3124,   0.2057,  -0.4593],
        [  4.8051, -10.3703,   0.5268],
        [ 11.9808,   4.9340,  -3.3715],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.3301,   0.3638,  -7.5685],
        [ -7.5401,   4.6718,   9.0033],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.4157,   0.0821,  -5.0025],
        [ -4.7334,   6.9982,  -0.4392],
        [  2.6078, -10.9229,   6.2869],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.9282,   2.1648,   8.4894],
        [ -1.6334, -10.7207,   7.1397],
        [ 12.6059,   0.9597,  -6.3264],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.6381,  -4.9238,  -4.4077],
        [  3.5939,   6.4029,  -1.8357],
        [ -0.8885,   7.9160,   0.0571],
        [ -4.1296,   7.2803,  -2.7194],
        [ 12.2692,  -1.8882,   3.9253],
        [  1.8320,   7.3871,   5.1452],
        [  3.6469,   6.3624,  -1.8414],
        [  1.7000,  -4.7609,   2.7935],
        [  2.5974,   7.0363,  -7.7810],
        [ 12.3357,   3.6001,   0.7903],
        [ -6.2668,   5.9808,   2.5614],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.0243,   7.3086,   7.5432],
        [  9.6614,   9.1784,  -1.2087],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.2687,  -2.9794,   0.6436],
        [ -5.9872,   6.2037,  -9.4477]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0127, 0.0215,  ..., 0.0282, 1.0222, 1.1257],
        [0.0127, 0.0000, 0.0089,  ..., 0.0158, 1.0243, 1.1202],
        [0.0215, 0.0089, 0.0000,  ..., 0.0090, 1.0245, 1.1153],
        ...,
        [0.0282, 0.0158, 0.0090,  ..., 0.0000, 1.0320, 1.1187],
        [1.0222, 1.0243, 1.0245,  ..., 1.0320, 0.0000, 0.6453],
        [1.1257, 1.1202, 1.1153,  ..., 1.1187, 0.6453, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.64865173,  5.76165934,  5.81481044,  5.86519525,  5.87132718,
        5.92281142, 12.53285633, 12.79126785,  9.95552626,  9.17016857,
        6.652742  ,  7.5403822 ,  5.05534609,  6.5618968 , 14.03505624,
       10.794755  , 13.61600536,  5.57341623,  9.50654967,  9.5014146 ,
        6.76056625,  5.1224079 ,  5.53175275,  5.68942572,  9.85330646,
       12.79126785,  5.9510489 ,  7.62011734,  8.08320235, 12.18045978,
        5.76165934, 10.69963954,  5.64464807,  7.53884104,  7.83534987,
       13.61600536,  8.49161154, 12.52875793, 13.12299145, 10.06437598,
        5.05963217,  9.05843201,  6.60752373, 10.90638066, 13.02818702,
       10.24557047, 13.66051884,  8.16584537,  5.1224079 , 13.6922425 ,
        5.5748624 , 13.35832725,  9.26455795, 13.14690137,  6.74686071,
        5.80312949,  6.46560181, 11.8159254 ,  4.78544875, 11.63765273,
       12.24467041,  6.97106912,  9.34127013, 12.4980845 ,  8.99276564,
       12.72524615,  5.0209005 ,  9.36256096, 13.04156075,  8.5724042 ,
        7.07902036, 11.9120321 ,  9.14973319, 13.3426541 ,  6.31576264,
       11.42947632, 12.95702647, 10.32715715,  6.34059021,  8.87007495,
        7.89745107,  9.41605684,  8.4486771 , 11.22993218,  9.95552626,
        9.18690018, 10.84437705, 12.64235359, 10.61472636,  9.94285511,
        7.34250948,  7.96575782,  8.36995953, 12.41366754,  7.61088125,
        7.33350642,  5.05534609,  7.50035355, 12.85032836,  8.66273171,
        9.73579738,  7.58378655, 13.32613445,  5.94535529,  9.73579738,
        8.62162372])}
{'P': tensor([[3.9482e-02, 2.2303e-01, 1.0418e-01,  ..., 6.9887e-02, 2.8290e-05,
         3.5021e-04],
        [2.1318e-03, 4.3256e-02, 1.5402e-01,  ..., 1.4705e-01, 3.8985e-05,
         2.5509e-04],
        [1.5740e-03, 3.3978e-03, 5.8384e-02,  ..., 4.3558e-01, 5.7354e-05,
         1.6267e-04],
        ...,
        [1.7801e-03, 2.9547e-03, 5.1323e-03,  ..., 1.0620e-01, 1.4656e-04,
         2.3968e-04],
        [1.3276e-04, 8.5340e-05, 4.6980e-05,  ..., 3.9026e-05, 9.2638e-02,
         9.8140e-04],
        [3.2768e-03, 3.6923e-03, 2.0607e-03,  ..., 2.0230e-03, 1.2609e-04,
         9.9819e-02]]), 'X': tensor([[ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0114, 0.0301,  ..., 0.0261, 0.8222, 0.4703],
        [0.0114, 0.0000, 0.0270,  ..., 0.0222, 0.8303, 0.4680],
        [0.0301, 0.0270, 0.0000,  ..., 0.0048, 0.8508, 0.4941],
        ...,
        [0.0261, 0.0222, 0.0048,  ..., 0.0000, 0.8477, 0.4893],
        [0.8222, 0.8303, 0.8508,  ..., 0.8477, 0.0000, 0.6914],
        [0.4703, 0.4680, 0.4941,  ..., 0.4893, 0.6914, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.68942572,  5.94541803,  6.19633139,  6.14086582,  5.83560009,
        6.18668463,  9.04255655, 13.48809595, 12.10222204, 10.32715715,
        6.7593917 , 10.88618114,  6.83083863, 11.62158554,  7.73151309,
       11.11969815,  8.95133961,  6.33517349, 11.97238624, 11.70016217,
       13.59480906,  6.38399789,  9.96391045,  5.05989547, 12.04924159,
       10.0674887 , 10.46238054, 11.2845608 ,  8.4486771 , 12.35625694,
        9.0368458 ,  5.62005629, 10.48315436,  8.68979052, 13.89974693,
        8.68979052, 14.05633441,  5.27178363,  8.51928531, 13.35013374,
       11.63765273, 10.84437705, 13.56367857,  8.67502348,  6.48486171,
        5.81199422, 12.76408562,  5.64464807,  6.54127431,  8.53014915,
       11.42420508,  9.54154096,  5.60064319, 13.43127131,  7.98469166,
       13.09737053,  9.70108409,  5.83560009,  6.33517349, 13.54774355,
        7.48538905, 13.1471313 ,  4.78258092, 10.4977132 , 10.68416401,
       13.02818702, 11.54672755,  8.23525433,  7.58668144, 13.66896739,
        7.03733775,  8.50268322,  5.46244988, 12.52372629, 11.76830981,
        6.56833864,  6.39187285,  8.95133961,  9.3109884 ,  6.39438044,
       13.43127131,  9.84314712, 13.15525803, 14.10045797, 11.83570132,
       10.74409834, 12.21532282, 10.06437598, 11.00694688, 10.67322527,
       11.51557166,  7.5092834 ,  5.22903117, 13.58333857,  9.99211663,
       10.68031351,  7.72668467,  7.50917551, 12.77402351, 10.78533838,
       10.61047596,  8.97052115, 13.25017211,  6.38105015,  9.09502385,
        5.14999799])}
{'P': tensor([[3.6818e-02, 1.5434e-01, 1.1324e-01,  ..., 3.5008e-05, 1.3808e-03,
         1.9361e-02],
        [1.4465e-03, 3.3666e-02, 1.9398e-01,  ..., 6.4124e-05, 6.0176e-04,
         1.0068e-02],
        [1.7104e-03, 4.9179e-03, 4.7528e-02,  ..., 1.0051e-04, 7.1435e-04,
         9.8001e-03],
        ...,
        [1.4935e-05, 1.4774e-05, 1.4673e-05,  ..., 2.9875e-01, 2.1578e-05,
         3.4949e-05],
        [1.0216e-01, 4.1892e-02, 3.1796e-02,  ..., 1.4227e-05, 3.8525e-02,
         1.3208e-02],
        [1.9186e-03, 4.8870e-03, 5.8574e-03,  ..., 9.9085e-04, 8.1890e-04,
         8.3631e-02]]), 'X': tensor([[  4.7544,  -3.1425,  -1.5315],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.3124,   0.2057,  -0.4593],
        [  6.3497,   1.3680,  -1.9076],
        [  9.0424,  -7.6779,  -7.5286],
        [ 11.9808,   4.9340,  -3.3715],
        [  4.0078,  -3.8018,  -6.2168],
        [  2.1379,  -4.6757,   6.4981],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -9.3444,   0.5821,   6.9101],
        [  2.3376,  -4.6231,   9.8436],
        [ -6.3066,  -8.1936,   9.5822],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.0190,   4.9705,  -7.9752],
        [ 12.3453,  -1.5462,   7.0212],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.5240,  -2.1198,   6.2055],
        [ -6.4844,  -8.0230,   8.5209],
        [  1.0000,   7.6591,  -4.5011],
        [ 12.5789,   0.1722,   1.6470],
        [  4.4039,   5.6903,   8.7366],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.1575,  12.2046,   3.3257],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.3804,  -2.3504,   7.2744],
        [  5.4732,  -2.2041,  -4.3830],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.6523,   3.9443,  -2.9449],
        [  6.3205,   1.7068,  -3.8619],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -0.8885,   7.9160,   0.0571],
        [  6.2770,  -0.0379,   2.4195],
        [ -8.5859,  -5.0393,   0.5476],
        [  1.8447,  -4.7372,   8.1235],
        [ -5.2589,   6.7024,  -5.5809],
        [  0.3715,  -4.7756,   0.2312],
        [  9.4925,   9.3811,   8.9842],
        [  6.3550,   0.7129,   0.4917],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.8674,  12.8875,   6.7622],
        [  2.9461,  -4.4077,  -0.4831],
        [ -4.0868,   7.2981,   8.1874],
        [ 11.7349,  -3.5838,   2.1644],
        [  3.1064,  13.5607,   6.7605],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.5057,  -3.3894,  -0.8160],
        [  9.5203,   9.3483,   5.8781],
        [ -2.1382,   7.8339,   7.3524],
        [ 12.5919,   0.4035,  -5.8460],
        [ -4.2271,   7.2387,  -1.8316],
        [ -8.4267,  -5.3697,   2.4172],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -1.9600,   7.8577,   6.9121],
        [  3.2226,  -4.2802,  -4.0428],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -1.7009,   7.8851,  -8.2437],
        [  0.6396,  14.0772,  -6.3786],
        [  4.2096,   5.8815,   1.9230],
        [  1.1350,  14.0104,   0.6321],
        [ -3.4488, -10.0934,   2.5195],
        [  3.2159, -10.8166,  -1.0406],
        [  5.8318,   3.5567,  -4.7692],
        [  4.4915,   5.5990,   7.0405],
        [  3.3069,   6.6089,   8.7377],
        [ -8.9282,   2.1648,   8.4894],
        [  6.3583,   1.1729,   8.7835],
        [ -7.8126,   4.3066,   5.1791],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.3124,   0.2057,  -0.4593],
        [  6.1741,  -0.5329,   3.1913],
        [ -3.9700,   7.3453,  -4.1486],
        [  3.4009,  -4.1874,  -5.9751],
        [ -6.1914,  -8.2997,   8.5380],
        [  2.2373,  -4.6506,  -1.7334],
        [  1.3705,  13.9723,   7.9266],
        [ -2.1533, -10.5778,   1.0112],
        [  5.6095,  -1.9702,  -7.8233],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -7.8359,   4.2735,   7.2943],
        [  7.8982,  -8.6778,   0.0239],
        [  5.8323,  -1.5228,   5.1132],
        [  9.3753,   9.5171,   3.9131],
        [ -6.0175,  -8.4541,  -9.8408],
        [  2.9470,  13.6085,   2.9809],
        [  6.2036,   2.4030,  -1.2500],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.3118,   6.5957,  -4.5667],
        [  6.3551,   0.7149,   1.3344],
        [  2.7121,   6.9743,  -8.4202],
        [  1.8447,  -4.7372,   8.1235],
        [  3.4604,  -4.1545,   1.8091],
        [  0.6952,  -4.8060,   8.8976],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.4633,   5.6287,  -0.7754],
        [  0.9967,  -4.8144,  -9.7370],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -0.0410,   7.8591,   9.3457],
        [  6.3413,   0.4883,  -6.8416],
        [  1.1274,  14.0116,   8.1753],
        [  2.5010,  -4.5736,  -0.8989],
        [  6.2770,  -0.0379,   2.4195]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0468, 0.0719,  ..., 0.8504, 0.0652, 0.2877],
        [0.0468, 0.0000, 0.0258,  ..., 0.8864, 0.1099, 0.3261],
        [0.0719, 0.0258, 0.0000,  ..., 0.9103, 0.1330, 0.3509],
        ...,
        [0.8504, 0.8864, 0.9103,  ..., 0.0000, 0.8230, 0.5634],
        [0.0652, 0.1099, 0.1330,  ..., 0.8230, 0.0000, 0.2612],
        [0.2877, 0.3261, 0.3509,  ..., 0.5634, 0.2612, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.69913817,  6.02789634,  6.10774544,  6.07992623,  6.31576264,
        6.49538841, 11.86239961, 12.95702647,  5.52416368,  5.14124612,
       10.53368644,  9.36256096,  5.1805345 , 10.33958639, 12.12448912,
       11.51811603,  7.06372172, 12.44177641, 11.63724792,  5.91677188,
       10.31584102,  7.72415048, 12.58006038,  7.19534146, 12.9089559 ,
       13.66991027, 12.07225469,  5.87132718,  5.90035324,  8.30125201,
        6.89245352,  6.54693987,  9.88310614,  7.96575782,  6.27715105,
        9.95552626,  5.08372149,  8.51928531,  4.79003361, 13.34587027,
        6.3949043 , 12.3489324 ,  5.8828986 , 13.77604293,  5.30158497,
        8.36445758, 12.26997165, 13.91198191, 13.09737053,  5.63824971,
       13.3426541 ,  8.12043983, 12.59840193,  8.38252937,  9.99211663,
       12.05300943,  8.09842704,  5.35774573,  9.50654967,  8.06643487,
       14.09176259,  7.23276582, 14.05633441, 10.66631784, 11.2845608 ,
        6.83083863,  7.17789666,  7.39004226,  9.18690018,  6.46560181,
        8.92098752,  5.94541803,  6.31576264,  6.19707944,  8.34949383,
        5.39450305, 10.35464916,  5.16077764, 14.0393949 , 10.794755  ,
        5.94541803,  9.70108409,  9.51074763,  8.92549752, 11.73397309,
        6.02779201, 13.35927527, 10.37698435, 13.92390689,  6.652742  ,
        8.8353674 , 13.09426142,  6.39520397,  7.48311002,  5.08372149,
        5.40689332,  4.85603886,  5.48276038,  7.18354885,  4.91652556,
       10.30972193,  7.85919687,  6.36004393, 14.05687815,  5.21280156,
        6.27715105])}
{'P': tensor([[2.0195e-02, 1.7019e-01, 1.5834e-01,  ..., 1.0717e-03, 3.3830e-05,
         6.8488e-04],
        [2.0735e-03, 2.7291e-02, 2.1686e-01,  ..., 9.9071e-04, 4.9830e-05,
         8.7148e-04],
        [2.7314e-03, 4.1887e-03, 3.6837e-02,  ..., 1.2255e-03, 6.8691e-05,
         1.1369e-03],
        ...,
        [1.1303e-03, 1.6534e-03, 1.7624e-03,  ..., 1.9968e-02, 1.8366e-06,
         1.7200e-03],
        [1.5129e-04, 1.2711e-04, 1.2311e-04,  ..., 1.0092e-05, 6.2806e-02,
         1.4081e-05],
        [4.4017e-04, 5.0443e-04, 5.1165e-04,  ..., 1.7275e-04, 4.7430e-05,
         6.8371e-02]]), 'X': tensor([[  4.7739,  -3.1218,  -4.1188],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.1787,  -2.6383,  -4.2533],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.1372,   2.6726,  -6.7885],
        [ -5.9938,  -8.4745,   3.9735],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.3705,   0.4208,  -3.5815],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.0424,  -7.6779,  -7.5286],
        [  3.5284, -10.7484,   5.3347],
        [ 10.0995,  -6.4739,   2.0974],
        [ -8.5607,  -5.0938,  -5.4531],
        [  6.3266,   0.3292,   9.3502],
        [  8.5716,  10.3596,   6.8818],
        [  0.1532,  -4.7422,  -7.0145],
        [  9.7328,   9.0902,  -0.8917],
        [ -7.2515,   5.0195,   4.6434],
        [  1.5662,  13.9375,   4.5142],
        [ -6.4368,   5.8356,   3.3014],
        [  4.7156,  -3.1831,  -8.0224],
        [  2.5375,   7.0676,  -4.3490],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.3464,   0.5708,  -1.6397],
        [  0.8700,  -4.8132,  -7.8042],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -4.8665,  -9.3204,  -5.2564],
        [  5.7218,  -9.9934,   6.5764],
        [ -9.4203,  -2.0340,   2.5856],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.4242, -10.1261,   3.9595],
        [  0.1047, -11.0046,   0.5188],
        [  0.1460,   7.8340,  -9.4614],
        [ -8.7544,   2.6096,   3.3146],
        [  6.2905,   1.9366,   2.1169],
        [ 10.9916,   7.2133,  -0.9179],
        [  4.8432,   5.1968,   1.5006],
        [ -3.9700,   7.3453,  -4.1486],
        [ -3.2736, -10.1703,   3.8712],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.8474,   6.8979,   6.6457],
        [ 12.3759,  -1.3943,   6.4659],
        [  4.9337,  12.8566,  -3.6582],
        [  6.1989,  -9.7591,   0.8289],
        [  5.8275,  -1.5334,   2.0124],
        [  5.0277,   4.9589,   7.0175],
        [  6.2905,   1.9366,   2.1169],
        [  6.3515,   0.6418,  -6.5823],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.3118,   4.5452,   4.2114],
        [  5.8836,   3.4315,  -4.0327],
        [  1.7301,   7.4262,   5.0522],
        [  3.2701,   6.6338,   9.9209],
        [ -7.7842,   4.3466,  -1.7548],
        [ -8.3900,   3.3727,   3.1650],
        [  9.1820,   9.7334,  -2.7932],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.0277,   4.9589,   7.0175],
        [ -3.0008, -10.2827,  -8.9480],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.8318,   3.5567,  -4.7692],
        [  5.2939,  -2.4784,  -4.5840],
        [ -5.7347,  -8.6908,   9.2878],
        [ -2.8162,   7.7059,  -9.6885],
        [ -8.7078,   2.7182,   1.8146],
        [ -1.0407,   7.9166,  -6.6021],
        [  6.1737,   2.5305,   3.9049],
        [  2.4932,   7.0903,  -5.4214],
        [  2.2076,  13.8032,  -1.5099],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 11.9808,   4.9340,  -3.3715],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 11.1469,   6.9237,   1.6160],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -5.9987,   6.1949,  -6.2114],
        [ -2.5683, -10.4433,  -2.9182],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.9548,  -1.2273,  -2.6520],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.8325,  -1.5222,  -2.1341],
        [ -7.5401,   4.6718,   9.0033],
        [  5.5626,  -2.0536,   8.4433],
        [ -9.1795,  -3.3623,   8.5259],
        [ 11.7794,   5.5086,  -9.6929],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.4986,   7.5152,   9.0712],
        [  3.5939,   6.4029,  -1.8357],
        [  5.6248,   3.9984,   8.4933],
        [ -2.1382,   7.8339,   7.3524],
        [  6.6319,  11.9066,   3.5165],
        [  2.6613,   7.0021,   4.5783],
        [ 11.3800,  -4.3919,   1.4066],
        [ -7.0948,   5.1940,   9.8171],
        [ -0.0282,   7.8575,  -1.1746],
        [  3.3167,   6.6022,   6.3554],
        [ -6.3066,  -8.1936,   9.5822],
        [  3.3839,  -4.1966,   8.3574],
        [  3.1064,  13.5607,   6.7605],
        [ -8.9624,  -4.0944,  -4.9622],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.0608,   3.9362,  -5.9567],
        [  6.2883,   0.0325,   5.2692]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0080, 0.0116,  ..., 0.2086, 0.7290, 0.6488],
        [0.0080, 0.0000, 0.0036,  ..., 0.2161, 0.7355, 0.6517],
        [0.0116, 0.0036, 0.0000,  ..., 0.2198, 0.7378, 0.6539],
        ...,
        [0.2086, 0.2161, 0.2198,  ..., 0.0000, 0.6377, 0.5332],
        [0.7290, 0.7355, 0.7378,  ..., 0.6377, 0.0000, 1.0917],
        [0.6488, 0.6517, 0.6539,  ..., 0.5332, 1.0917, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.70405525,  5.79391825,  5.81199422,  5.90035324,  5.84450127,
        5.84533345,  6.69389431, 10.37998078,  5.24137909,  9.37989631,
        9.68324553, 11.86239961, 11.31276109, 11.99633819,  9.96152788,
        6.33517349, 13.44593519,  4.74468406, 13.31764181,  8.81930525,
       14.02526505,  8.68832398,  5.68942572,  7.5092834 , 13.25336001,
        9.43737523,  6.78325151,  9.36378438,  4.89121805, 10.06407839,
       10.51437947, 11.51557166,  9.63742669,  5.48276038, 11.48734978,
       11.00508681,  7.83534987,  9.13507321,  6.58184474, 13.1471313 ,
        7.10378919,  8.34949383, 10.68416401,  5.24461068,  7.46248947,
       12.45418236, 13.77075174, 11.56146873,  6.02588517,  7.06169391,
        6.58184474,  6.383897  , 10.92597196,  6.99097013,  6.81117802,
        7.62509582,  7.39599986,  8.91551231,  9.04255655, 13.38091121,
       12.77402351,  7.06169391, 10.71162681, 11.75618513,  6.83083863,
        5.84533345, 10.412299  ,  8.20436764,  9.12220271,  7.98469166,
        6.67218687,  7.5158489 , 13.97857591, 12.25806559, 12.95702647,
        9.562174  , 13.12218013,  9.43737523,  8.62329466, 10.75443106,
        5.56809335,  6.07992623, 12.41012046,  6.02789634,  8.87007495,
        5.92952538,  9.77588801, 13.00380624,  8.28968067,  8.28968067,
        7.34250948,  6.90115901,  8.12043983, 13.62896593,  7.49076899,
       12.19805394,  8.79284466,  7.8575648 ,  7.38844703, 10.33958639,
        5.39098669, 13.91198191,  9.85330646, 11.97949273,  8.97052115,
        6.28835101])}
{'P': tensor([[2.2707e-02, 1.7753e-01, 1.3970e-01,  ..., 7.2283e-05, 5.4566e-06,
         6.2255e-05],
        [1.8444e-03, 2.3629e-02, 1.5567e-01,  ..., 8.7434e-05, 7.1936e-06,
         6.6617e-05],
        [2.3458e-03, 3.3646e-03, 4.4132e-02,  ..., 1.9581e-04, 1.4999e-05,
         1.0039e-04],
        ...,
        [9.5559e-05, 8.3523e-05, 6.9249e-05,  ..., 5.6921e-02, 1.2274e-06,
         9.9369e-06],
        [7.0061e-06, 7.8404e-06, 8.1115e-06,  ..., 1.0963e-03, 4.8624e-01,
         1.4998e-04],
        [1.9950e-04, 2.3805e-04, 2.4424e-04,  ..., 1.4932e-04, 9.0135e-03,
         1.8067e-01]]), 'X': tensor([[  4.7813,  -3.1139,   8.2691],
        [  5.1472,  -2.6801,   8.1194],
        [  5.5626,  -2.0536,   8.4433],
        [  5.6022,  -1.9833,   8.3081],
        [  5.6672,  -1.8631,   8.1020],
        [  6.0054,  -1.0902,   8.3079],
        [ -9.0574,   1.7801,   8.8986],
        [ -7.4808,   4.7462,   2.0266],
        [  2.1477,   7.2550,   8.9688],
        [  2.5375,   7.0676,  -4.3490],
        [  3.6703,   6.3443,  -5.4401],
        [  5.2939,  -2.4784,  -4.5840],
        [  1.9503,  -4.7172,  -6.2313],
        [  3.6249, -10.7255,   5.6605],
        [ -7.4808,   4.7462,   2.0266],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.2895,   0.0405,   5.7652],
        [ -7.8359,   4.2735,   7.2943],
        [  5.3924, -10.1396,  -4.0863],
        [  7.0664,  -9.2590,   8.0879],
        [  3.2366,  -4.2732,   5.1532],
        [ 10.4758,  -5.9539,   1.3667],
        [ 11.6634,   5.8050,   9.1590],
        [  3.0719, -10.8449,  -4.1673],
        [ -5.3179,  -9.0101,   9.4472],
        [  6.2895,   0.0405,   5.7652],
        [ -5.1046,  -9.1610,   2.3418],
        [  0.6363, -11.0348,   6.1414],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.7590,  13.6621,   9.4496],
        [  5.4336,  -2.2676,   5.6933],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.9433,  -9.3362,  -0.3421],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.4179,  -2.2925,  -3.4641],
        [  1.3103,  -4.8032,  -0.9300],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 11.0275,  -5.0666,   8.4154],
        [  1.9505,  -4.7171,   9.9868],
        [ -7.2986,  -7.1225,  -3.0814],
        [  3.5939,   6.4029,  -1.8357],
        [  5.8617,   3.4853,  -9.8080],
        [  9.7328,   9.0902,  -0.8917],
        [  4.0587, -10.6112,   6.2273],
        [ -6.4822,  -8.0252,  -3.8983],
        [  7.0664,  -9.2590,   8.0879],
        [  3.5939,   6.4029,  -1.8357],
        [  4.1140,  -3.7220,  -0.1273],
        [  2.2437,  -4.6489,   6.1430],
        [ 12.5224,  -0.4497,   4.8448],
        [  4.9337,  12.8566,  -3.6582],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.8407,  11.7666,   3.0045],
        [ -6.4464,   5.8272,   7.5038],
        [  6.2675,  12.1379,  -8.9720],
        [  8.4691,  10.4571,  -5.0650],
        [  0.1287, -11.0065,   6.5757],
        [ -5.3179,  -9.0101,   9.4472],
        [  7.0642,  11.6104,  -9.8316],
        [  3.1271, -10.8343,   2.6442],
        [ -6.4844,  -8.0230,   8.5209],
        [ 11.4495,   6.3026,   6.8257],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.5662,  13.9375,   4.5142],
        [  5.6928,  12.4710,   3.6001],
        [ 12.1504,   4.3674,  -7.2998],
        [  5.1142,  -2.7230,  -4.1966],
        [  9.5203,   9.3483,   5.8781],
        [ -9.1706,   1.3863,   8.4032],
        [ -1.9714,   7.8563,  -1.8084],
        [ -0.5739,   7.9056,   0.6382],
        [  9.9058,  -6.7203,   5.6928],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.4505,  -1.7081,  -0.7039],
        [  5.3804,  -2.3504,   7.2744],
        [ -5.9225,  -8.5355,   2.4778],
        [ 11.2571,   6.7072,   8.3693],
        [  5.6672,  -1.8631,   8.1020],
        [  2.9461,  -4.4077,  -0.4831],
        [  8.0381,  -8.5690,  -8.9523],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.6229,   2.9068,   2.4429],
        [  2.7590,  13.6621,   9.4496],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -5.5525,   6.5146,  -4.4984],
        [  1.7228,  -4.7575,   8.3396],
        [ -3.1671,   7.6158,   6.0765],
        [  0.1287, -11.0065,   6.5757],
        [  8.2422,  -8.4041,  -1.4962],
        [  1.1274,  14.0116,   8.1753],
        [ 10.9931,   7.2107,   7.7933],
        [ -4.1296,   7.2803,  -2.7194],
        [  5.8836,   3.4315,  -4.0327],
        [ -8.1204,  -5.9308,   4.0761],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.2589,  -1.9312,   9.8657],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.9022,  13.2906,   5.5788],
        [ 11.6544,   5.8273,  -3.1375],
        [ -7.6293,  -6.6859,  -5.5161],
        [  7.0358,  -9.2784,  -9.8479],
        [ -5.9940,   6.1985,   5.9566],
        [  5.8930,  12.3592,  -8.9201],
        [  6.1579,  -0.5961,  -6.5595]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 7.5985e-03, 1.4838e-02,  ..., 5.1959e-01, 1.1667e+00,
         8.8064e-01],
        [7.5985e-03, 0.0000e+00, 9.9924e-03,  ..., 5.2522e-01, 1.1611e+00,
         8.7629e-01],
        [1.4838e-02, 9.9924e-03, 0.0000e+00,  ..., 5.3438e-01, 1.1644e+00,
         8.8148e-01],
        ...,
        [5.1959e-01, 5.2522e-01, 5.3438e-01,  ..., 0.0000e+00, 1.3143e+00,
         9.6296e-01],
        [1.1667e+00, 1.1611e+00, 1.1644e+00,  ..., 1.3143e+00, 0.0000e+00,
         3.5571e-01],
        [8.8064e-01, 8.7629e-01, 8.8148e-01,  ..., 9.6296e-01, 3.5571e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.70591986,  5.80312949,  5.92952538,  5.94293185,  5.96556248,
        6.10360073,  9.23071698,  8.859416  ,  7.56617037,  7.5092834 ,
        7.32950385,  5.84533345,  5.10444097, 11.32149259,  8.859416  ,
       10.27716434,  6.28961988,  8.92549752, 11.48434767, 11.64746882,
        5.36061974, 12.0495521 , 13.02818702, 11.27160035, 10.46238054,
        6.28961988, 10.48721382, 11.05317597,  8.14433542, 13.93790235,
        5.88782913,  4.8763135 , 11.63502679, 12.10942412,  5.8828986 ,
        4.978704  , 10.09979747, 12.13568598,  5.10446653, 10.1979669 ,
        7.34250948,  6.81961996, 13.31764181, 11.36089355, 10.31613269,
       11.64746882,  7.34250948,  5.54776934,  5.16205087, 12.53047489,
       13.77075174, 12.46513646, 13.6105595 ,  8.68979052, 13.66051884,
       13.45642379, 11.00726483, 10.46238054, 13.59056613, 11.27656488,
       10.31584102, 13.06957425, 12.32408516, 14.02526505, 13.70893492,
       12.91143491,  5.79391825, 13.3426541 ,  9.27474386,  8.09984231,
        7.92644746, 11.9702762 , 11.04629471,  9.60359158,  5.87132718,
       10.38897563, 13.103709  ,  5.96556248,  5.30158497, 11.74901778,
        9.87720102,  9.93553424,  9.09963894, 13.93790235, 12.35421461,
        9.50913849,  8.55982105,  5.05981914,  8.24808832, 11.00726483,
       11.77124373, 14.05687815, 13.14690137,  8.36995953,  6.81117802,
       10.05558893, 12.889695  , 12.41012046, 10.53368644, 13.85158417,
       13.03002338, 10.14436912, 11.64436684,  8.62260327, 13.6922425 ,
        6.18668463])}
{'P': tensor([[2.3480e-02, 1.8637e-01, 1.6371e-01,  ..., 3.9093e-05, 2.0384e-05,
         3.3117e-06],
        [2.7698e-03, 2.9223e-02, 2.3996e-01,  ..., 5.0555e-05, 2.2607e-05,
         3.8794e-06],
        [3.5874e-03, 3.9783e-03, 6.2872e-02,  ..., 1.0132e-04, 3.4387e-05,
         6.6679e-06],
        ...,
        [1.6935e-04, 1.8946e-04, 2.1938e-04,  ..., 2.3691e-01, 1.1289e-04,
         1.0060e-04],
        [5.1716e-03, 5.3324e-03, 4.8098e-03,  ..., 9.9228e-03, 2.5821e-01,
         4.9000e-04],
        [5.3770e-04, 5.2955e-04, 4.3687e-04,  ..., 5.0683e-03, 4.3917e-01,
         2.2028e-01]]), 'X': tensor([[  4.9957,  -2.8705,   6.5978],
        [  5.1886,  -2.6250,   6.4677],
        [  5.6093,  -1.9705,   6.4807],
        [  5.5240,  -2.1198,   6.2055],
        [  5.5424,  -2.0885,   7.2487],
        [  5.3602,  -2.3808,   7.2098],
        [ -5.0132,  -9.2234,  -8.3531],
        [  4.4915,   5.5990,   7.0405],
        [  3.2692,  13.5098,  -4.9032],
        [ -2.0445,   7.8469,  -0.5857],
        [ -1.1885, -10.8211,   0.0865],
        [  9.2811,  -7.4336,   5.1771],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -5.6169,  -8.7844,   8.9057],
        [  1.9241,   7.3503,   7.9004],
        [  1.4489,   7.5256,   4.4529],
        [ -8.0829,  -5.9940,   5.0564],
        [ -7.0825,   5.2073,  -5.0497],
        [  7.1821,  -9.1845,   9.8548],
        [ -7.3446,   4.9113,  -5.9356],
        [  1.2728, -11.0372,   9.8121],
        [  9.8082,  -6.8396,   4.9615],
        [  6.1752,   2.5242,   3.5874],
        [ -9.4753,  -0.8012,  -5.2050],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.6059,   0.9597,  -6.3264],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.0961,   1.6529,   3.1553],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.7544,   2.6096,   3.3146],
        [ -7.5603,   4.6459,   8.2002],
        [  3.1756, -10.8247,   6.8882],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -4.0409,   7.3169,  -3.5779],
        [  3.9770, -10.6341,   8.0783],
        [  6.1575,  12.2046,   3.3257],
        [  5.7963,  -1.6018,   1.8856],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.6229,   2.9068,   2.4429],
        [  2.6613,   7.0021,   4.5783],
        [  7.5930,  -8.9037,   7.2323],
        [ -3.3116, -10.1539,   7.5919],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -0.3432,   7.8900,   7.1805],
        [  2.0729,   7.2878,   7.0880],
        [ -7.6650,   4.5092,   0.2520],
        [  0.8940, -11.0402,  -0.9783],
        [ -9.3384,  -2.6161,  -4.7630],
        [  7.7794,  11.0629,   8.9069],
        [ -9.3087,   0.7792,  -7.5587],
        [ -6.3489,   5.9117,   6.5161],
        [ -1.0012, -10.8576,   2.2867],
        [ -8.8745,   2.3097,   3.5600],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.3739,   0.3982,  -4.2564],
        [ -4.6393,   7.0461,   1.1624],
        [  5.8201,   3.5841,  -8.0609],
        [ -7.8280,   4.2848,  -7.1955],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.3490,  -4.6199,  -4.2489],
        [  3.3069,   6.6089,   8.7377],
        [ -5.9872,   6.2037,  -9.4477],
        [  9.1409,  -7.5789,   1.5814],
        [ -8.0766,  -6.0046,  -3.1873],
        [  3.1480,  -4.3165,  -7.8002],
        [ -2.4956, -10.4682,   3.5632],
        [ -7.3470,  -7.0615,   5.0066],
        [  3.6817,   6.3354,   6.1233],
        [  5.4179,  -2.2925,  -3.4641],
        [  9.5203,   9.3483,   5.8781],
        [ -0.5385,   7.9037,   6.3830],
        [ -1.7009,   7.8851,  -8.2437],
        [  3.3744,   6.5623,  -6.3636],
        [  6.2895,   0.0405,   5.7652],
        [  2.0870,  -4.6877,   9.6823],
        [ -2.0721,   7.8432,   6.6380],
        [  1.8320,   7.3871,   5.1452],
        [  6.2242,  -0.3171,   7.4978],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.4501,  -1.7133,   2.2980],
        [ 11.9122,   5.1397,  -1.5498],
        [  9.6587,   9.1817,  -0.8212],
        [  5.4167, -10.1293,   6.2156],
        [  6.1752,   2.5242,   3.5874],
        [  0.9478,  -4.8144,   0.6371],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.9003,  12.3550,   2.1917],
        [  1.7000,  -4.7609,   2.7935],
        [  0.1931,  -4.7491,   7.4385],
        [  6.1730,  -0.5375,  -8.8861],
        [ -9.4501,  -1.7133,   2.2980],
        [ -7.3333,   4.9246,  -7.9212],
        [  1.6954,  -4.7616,   9.8198],
        [  9.2979,   9.6048,   7.3138],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -0.4341, -10.9473,   7.4509],
        [  5.6022,  -1.9833,   8.3081],
        [  9.0652,   9.8597,  -3.5949],
        [ -3.6381,   7.4683,   7.3544],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.2247,  -0.3150,  -9.3657],
        [  2.1771, -10.9776,  -3.2007],
        [ -2.9497, -10.3028,  -7.3479]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0089, 0.0158,  ..., 0.8853, 0.9247, 1.2472],
        [0.0089, 0.0000, 0.0090,  ..., 0.8771, 0.9205, 1.2433],
        [0.0158, 0.0090, 0.0000,  ..., 0.8753, 0.9246, 1.2476],
        ...,
        [0.8853, 0.8771, 0.8753,  ..., 0.0000, 0.6024, 0.8114],
        [0.9247, 0.9205, 0.9246,  ..., 0.6024, 0.0000, 0.3240],
        [1.2472, 1.2433, 1.2476,  ..., 0.8114, 0.3240, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.76165934,  5.81481044,  5.94535529,  5.91677188,  5.92281142,
        5.86519525, 10.4977132 ,  7.17789666, 13.89974693,  8.10886021,
       10.88618114, 11.89105259, 10.60226614, 10.42667093,  7.59795445,
        7.66377554, 10.06286074,  8.7907923 , 11.65923396,  8.8353674 ,
       11.11038551, 11.95743558,  6.67121748,  9.50913849, 12.97370371,
        6.06416957, 12.64235359, 10.33712258,  9.24501829, 10.00802627,
        6.652742  ,  9.13507321,  8.8737556 , 11.28093507, 10.46739073,
        8.35856487, 11.35345345, 13.66991027,  6.01356004,  8.92098752,
        9.09963894,  7.49076899, 11.70168755, 10.68031351, 10.60226614,
        7.89745107,  7.57685908,  8.89301516, 11.07637531,  9.69792513,
       13.52430176,  9.34127013,  8.67502348, 10.9036226 ,  9.17016857,
       11.64746882,  9.38232121,  8.43626114,  6.83515201,  8.92395381,
        9.85330646,  5.18277858,  7.39004226,  8.62162372, 11.87412439,
       10.06407839,  5.34250225, 10.76154256, 10.19036081,  7.32754882,
        5.8828986 , 13.3426541 ,  7.92200342,  8.06643487,  7.37902766,
        6.28961988,  5.13125513,  8.11227668,  7.61088125,  6.2322769 ,
        7.72668467,  9.60412735, 12.97370371, 13.32645552, 11.48663767,
        6.67121748,  4.90676242,  8.25828409, 13.69162795,  5.05534609,
        4.75303127,  6.19633139,  9.60412735,  8.83341608,  5.05444288,
       13.36800073,  5.01114719,  9.51074763, 10.95593729,  5.94293185,
       13.39372426,  8.30729352, 10.06437598,  6.23262289, 11.1913598 ,
       10.7167285 ])}
{'P': tensor([[2.4393e-02, 1.9182e-01, 1.4210e-01,  ..., 2.5842e-05, 1.2390e-05,
         7.7801e-05],
        [2.0079e-03, 3.2029e-02, 1.5951e-01,  ..., 2.3548e-05, 1.2628e-05,
         1.3154e-04],
        [2.8741e-03, 6.3538e-03, 5.0519e-02,  ..., 3.2975e-05, 1.8777e-05,
         2.6449e-04],
        ...,
        [3.7623e-03, 3.0381e-03, 2.9825e-03,  ..., 6.5452e-02, 1.2619e-04,
         1.5772e-05],
        [5.7110e-04, 4.1658e-04, 3.8913e-04,  ..., 3.1579e-01, 2.5807e-02,
         4.6121e-06],
        [5.6952e-05, 5.5635e-05, 6.0254e-05,  ..., 4.5503e-05, 1.5365e-04,
         1.2483e-01]]), 'X': tensor([[  5.0348,  -2.8229,   3.5436],
        [  5.5158,  -2.1337,   3.5246],
        [  5.7151,  -1.7700,   3.1542],
        [  5.6260,  -1.9401,   3.0259],
        [  5.8902,  -1.3887,   4.0388],
        [  6.0890,  -0.8382,   2.9166],
        [ 10.1983,  -6.3430,  -0.3200],
        [  0.1931,  -4.7491,   7.4385],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.6019,   5.9540,  -5.1715],
        [  9.3624,   9.5318,   0.2779],
        [  0.4119,  14.1019,  -4.1870],
        [  3.3633,  13.4794,   0.5193],
        [  6.3587,   1.1596,  -3.6629],
        [  1.3705,  13.9723,   7.9266],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.0611,  -6.0302,   4.6962],
        [ -8.6603,   2.8250,   5.9181],
        [  5.8902,  -1.3887,   4.0388],
        [  6.2022,   2.4092,   3.0349],
        [  3.3943,   6.5483,   1.0790],
        [  6.3534,   1.2990,  -7.2942],
        [  2.1379,  -4.6757,   6.4981],
        [ -1.1885, -10.8211,   0.0865],
        [  6.3535,   0.6792,  -7.5101],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -0.8569,   7.9156,  -6.5557],
        [ -8.9282,   2.1648,   8.4894],
        [  3.4140,  -4.1803,  -8.8224],
        [ -7.9298,   4.1367,  -3.9773],
        [ -7.9667,   4.0814,  -0.6397],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -3.4986,   7.5152,   9.0712],
        [  2.9470,  13.6085,   2.9809],
        [ -2.6735, -10.4062,  -1.5911],
        [  1.7903,  -4.7467,   0.1168],
        [  2.7558,  -4.4844,   0.0636],
        [  0.5175,  14.0910,   4.6540],
        [ -8.9282,   2.1648,   8.4894],
        [  4.6357,   5.4415,   7.9067],
        [  3.2647, -10.8066,  -0.3768],
        [  2.7976,  -4.4682,  -4.2020],
        [  3.9895,   6.0816,  -6.5235],
        [  7.2340,  -9.1505,  -5.8321],
        [  8.9284,  10.0036,  -6.9210],
        [ -9.1459,   1.4780,  -4.2403],
        [  7.3592,  11.3937,   6.2407],
        [  5.0233,   4.9647,  -3.1788],
        [  9.2979,   9.6048,   7.3138],
        [ 10.0619,   8.6627,  -9.2884],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.3804,  -2.3504,   7.2744],
        [ -1.3940, -10.7772,   6.7835],
        [ -3.5173,   7.5091,  -8.9585],
        [  0.2435,   7.8191,  -6.4709],
        [  4.0587, -10.6112,   6.2273],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.2687,  -2.9794,   0.6436],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.6202,  -1.9508,   0.7471],
        [  4.5503,   5.5359,   7.4735],
        [  2.0243,   7.3086,   7.5432],
        [ -9.3604,   0.4855,  -3.0158],
        [ -7.4602,   4.7718,  -4.6277],
        [  2.5636, -10.9293,   9.6519],
        [  9.3624,   9.5318,   0.2779],
        [ -0.0282,   7.8575,  -1.1746],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.4430,  -0.1897,   4.8383],
        [ -6.8466,  -7.6488,   1.9648],
        [ -6.4620,  -8.0449,   6.2719],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.5620,   7.0549,   8.4825],
        [ -0.9715, -10.8630,   0.3555],
        [ -2.1533, -10.5778,   1.0112],
        [  5.6022,  -1.9833,   8.3081],
        [  0.6470, -11.0352,  -1.1931],
        [  7.3724,  -9.0577,  -8.9843],
        [  4.8432,   5.1968,   1.5006],
        [  1.4150,  13.9647,  -4.0915],
        [  6.2365,  -0.2578,  -6.0212],
        [  5.8450,  -9.9356,   3.0767],
        [  1.9505,  -4.7171,   9.9868],
        [ -8.1666,   3.7653,   3.6075],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.0305,   3.0296,   0.1375],
        [ -8.3062,  -5.6008,   0.0287],
        [  0.1041, -11.0045,   2.4311],
        [  7.5309,  11.2617,   7.2656],
        [  0.1532,  -4.7422,  -7.0145],
        [  8.6420,  -8.0580,   6.5781],
        [ -3.1204, -10.2345,   9.2288],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.3569,   0.7591,   6.5011],
        [  3.6703,   6.3443,  -5.4401],
        [  8.5716,  10.3596,   6.8818],
        [ -7.4733,   4.7555,   6.2490],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.6112,   2.9319,  -7.9943],
        [ -2.5056,   7.7720,   5.5878],
        [ -6.0175,  -8.4541,  -9.8408],
        [  0.2480, -11.0153,  -3.0183],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -1.0407,   7.9166,  -6.6021]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0106, 0.0298,  ..., 0.8415, 1.0264, 0.7037],
        [0.0106, 0.0000, 0.0273,  ..., 0.8504, 1.0357, 0.7088],
        [0.0298, 0.0273, 0.0000,  ..., 0.8368, 1.0245, 0.6859],
        ...,
        [0.8415, 0.8504, 0.8368,  ..., 0.0000, 0.2069, 0.3904],
        [1.0264, 1.0357, 1.0245,  ..., 0.2069, 0.0000, 0.5829],
        [0.7037, 0.7088, 0.6859,  ..., 0.3904, 0.5829, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.77219077,  5.91408504,  5.98285319,  5.95111804,  6.05165248,
        6.1463896 , 12.00995898,  4.75303127, 12.91143491, 13.04051744,
       13.3607379 , 14.10796466, 13.89264822,  6.46356806, 14.0393949 ,
        9.7712321 , 10.67322527, 10.06703358,  9.10945678,  6.05165248,
        6.65368551,  7.37576154,  6.48486171,  5.14124612, 10.88618114,
        6.38968357,  5.53255882,  9.47369628,  7.96181288,  9.18690018,
        5.39723241,  8.94395628,  8.95133961,  9.43737523,  8.28968067,
       13.92390689, 10.74409834,  5.07306274,  5.26342203, 14.10045797,
        9.18690018,  7.14837514, 11.28895489,  5.27178363,  7.27340834,
       11.66453059, 13.40850328,  9.26455795, 13.56367857,  7.06272256,
       13.36800073, 13.27717982,  5.0209005 ,  5.87132718, 10.86693708,
        8.29203356,  7.822854  , 11.36089355,  6.10774544,  9.73579738,
       12.07225469,  5.94909709,  7.16598273,  7.58378655,  9.37296102,
        8.85572292, 11.22596994, 13.3607379 ,  7.8575648 ,  6.01356004,
        9.44486609, 10.26545792, 10.31886066, 10.0674887 ,  7.50564223,
       10.90638066, 10.794755  ,  5.94293185, 11.05413545, 11.67875855,
        7.10378919, 14.03618649,  6.24187044, 11.52733581,  5.10446653,
        8.99276564,  8.08320235,  6.74872981, 10.01804223, 11.0050379 ,
       13.54774355,  4.74468406, 11.8159254 , 10.69963954,  5.5513072 ,
        6.4020404 ,  7.32950385, 13.44593519,  8.8580671 , 10.92597196,
        9.09661212,  8.16584537, 10.37698435, 11.018084  , 10.61047596,
        7.98469166])}
{'P': tensor([[2.5865e-02, 2.0553e-01, 1.9992e-01,  ..., 3.5533e-05, 3.8262e-03,
         1.7132e-04],
        [3.5886e-03, 3.1560e-02, 2.6496e-01,  ..., 4.3806e-05, 4.6671e-03,
         1.9939e-04],
        [3.0397e-03, 3.0893e-03, 4.0871e-02,  ..., 6.1586e-05, 6.0326e-03,
         2.1355e-04],
        ...,
        [1.8969e-04, 1.8246e-04, 1.5546e-04,  ..., 1.6972e-01, 6.9871e-06,
         1.0428e-05],
        [1.4919e-04, 1.5848e-04, 2.1905e-04,  ..., 3.3855e-05, 4.6608e-02,
         7.2074e-05],
        [2.2243e-05, 2.2956e-05, 2.8003e-05,  ..., 2.1042e-06, 2.3377e-03,
         9.4879e-03]]), 'X': tensor([[  5.1142,  -2.7230,  -4.1966],
        [  5.1787,  -2.6383,  -4.2533],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.3604,   0.9042,  -6.5901],
        [  1.6954,  -4.7616,   9.8198],
        [ -9.4756,  -0.8166,  -9.6021],
        [  1.6308,  13.9254,  -2.0984],
        [  2.7558,  -4.4844,   0.0636],
        [ 11.6634,   5.8050,   9.1590],
        [ -3.3999, -10.1152,  -1.2088],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.8318,   3.5567,  -4.7692],
        [  3.4401,  13.4540,   4.6576],
        [  5.6672,  -1.8631,   8.1020],
        [ 11.0438,   7.1180,  -5.6991],
        [  1.1006,   7.6318,  -2.4877],
        [  4.8051, -10.3703,   0.5268],
        [ -3.5520,   7.4976,   9.1522],
        [  0.2732,  -4.7619,  -1.1577],
        [ -6.3306,   5.9272,   5.6935],
        [  2.3200,   7.1756,   1.1938],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.0093,   7.6567,   2.3526],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -1.6086,   7.8928,   4.3385],
        [ -0.8455, -10.8853,  -6.7400],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.1624,  -3.4286,   2.0647],
        [ 11.0864,   7.0387,  -0.1238],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -1.9487,   7.8590,  -3.3330],
        [ -8.0965,  -5.9712,   4.3466],
        [ 10.4994,  -5.9193,  -7.5153],
        [  8.2154,  -8.4262,   8.0171],
        [ -1.2454,   7.9127,  -0.5715],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -2.1533, -10.5778,   1.0112],
        [  5.1886,  -2.6250,   6.4677],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.3550,   0.7129,   0.4917],
        [  2.8890,  -4.4316,   1.1600],
        [  4.3459,  -3.5327,   0.3667],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.5650,   2.0132,   4.6272],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.0330,  -6.0765,   6.7945],
        [  7.8375,  -8.7240,  -9.8913],
        [ -6.2200,  -8.2737,   5.5079],
        [  3.8111, -10.6787,  -4.6539],
        [  9.0424,  -7.6779,  -7.5286],
        [  3.2366,  -4.2732,   5.1532],
        [  9.5018,  -7.1945,  -1.1892],
        [ 12.4901,  -0.7052,   1.7315],
        [  2.8602,   6.8905,   5.8779],
        [  5.6672,  -1.8631,   8.1020],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.3804,  -2.3504,   7.2744],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -2.0840,   7.8415,  -7.8657],
        [ -3.4488, -10.0934,   2.5195],
        [  8.1043,  -8.5164,  -7.9290],
        [  7.9141,  -8.6656,   9.6948],
        [  6.2790,  -0.0259,  -5.4116],
        [ 12.5154,   2.5072,  -9.3274],
        [  8.2422,  -8.4041,  -1.4962],
        [ -9.3739,   0.3982,  -4.2564],
        [ -6.2668,   5.9808,   2.5614],
        [  5.1667,  -2.6543,   1.2879],
        [ -8.0608,   3.9362,  -5.9567],
        [ -5.0550,   6.8231,  -9.2502],
        [ 11.2517,  -4.6495,   2.9800],
        [ 11.9936,  -2.8677,   6.0614],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.4150,  13.9647,  -4.0915],
        [  7.5930,  -8.9037,   7.2323],
        [ 10.8218,  -5.4177,   6.8573],
        [ 10.0995,  -6.4739,   2.0974],
        [ -5.9225,  -8.5355,   2.4778],
        [  6.2630,   2.1057,  -5.2136],
        [  6.0345,   3.0175,  -1.4424],
        [  1.9312,  13.8649,   8.3563],
        [  4.2096,   5.8815,   1.9230],
        [  0.9809,   7.6642,  -1.4217],
        [  6.0054,  -1.0902,   8.3079],
        [  1.4489,   7.5256,   4.4529],
        [  2.4149,  -4.6005,  -4.3339],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.8447,  -4.7372,   8.1235],
        [  6.3413,   0.4883,  -6.8416],
        [  3.1271, -10.8343,   2.6442],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.0534,  -6.0429,   2.4583],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.5506,  -5.1154,   6.3423],
        [  4.1999,  -3.6543,  -4.1998],
        [  0.1497,   7.8334,  -5.4412],
        [  5.3858,   4.4260,  -9.6769],
        [ -4.0409,   7.3169,  -3.5779],
        [  4.6357,   5.4415,   7.9067],
        [ -2.1382,   7.8339,   7.3524],
        [ -7.4808,   4.7462,   2.0266],
        [ 12.5996,   1.3971,  -1.0694],
        [ 11.8951,  -3.1570,   7.0742]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.6486e-03, 1.3226e-02,  ..., 7.6001e-01, 2.1662e-01,
         9.3035e-01],
        [3.6486e-03, 0.0000e+00, 9.5816e-03,  ..., 7.6365e-01, 2.1659e-01,
         9.3274e-01],
        [1.3226e-02, 9.5816e-03, 0.0000e+00,  ..., 7.7322e-01, 2.1639e-01,
         9.3872e-01],
        ...,
        [7.6001e-01, 7.6365e-01, 7.7322e-01,  ..., 0.0000e+00, 8.1113e-01,
         7.8428e-01],
        [2.1662e-01, 2.1659e-01, 2.1639e-01,  ..., 8.1113e-01, 1.5170e-04,
         7.8190e-01],
        [9.3035e-01, 9.3274e-01, 9.3872e-01,  ..., 7.8428e-01, 7.8190e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.79391825,  5.81199422,  5.90035324,  6.1561646 ,  6.12886024,
        5.84533345,  6.42439344,  5.05444288,  9.51074763, 14.02058621,
        5.26342203, 13.02818702, 10.67132524, 10.06850677,  6.83083863,
       13.88683436,  5.96556248, 13.13888841,  7.71075226, 11.42947632,
        8.29640613,  4.76969364,  8.67227386,  7.54127425, 12.3828957 ,
        7.72291946, 10.30972193,  8.0550286 , 10.91805683, 10.06286074,
        9.78285424, 13.13205357, 12.12641969,  8.09704052, 10.06023665,
       12.05300943, 11.76830981,  8.01009083,  9.75401449, 10.794755  ,
        5.81481044,  5.27178363,  6.3949043 ,  5.29009836,  5.60064319,
        8.68832398, 12.72524615,  6.652742  , 10.07238463, 11.72749923,
       10.35093485, 11.33837396, 11.86239961,  5.36061974, 11.9182956 ,
       12.50996828,  7.46053281,  5.96556248,  9.78285424,  5.87132718,
       10.4977132 ,  8.1137452 , 10.66631784, 11.75618513, 11.73567759,
        6.27905292, 12.76408562, 11.77124373,  9.38232121,  8.66273171,
        5.80860165,  8.97052115,  8.49161154, 12.17451685, 12.33167382,
        9.17905491, 14.03618649, 11.70168755, 12.10222204, 11.99633819,
       10.38897563,  6.60752373,  6.74686071, 13.99877413,  7.23276582,
        7.72668467,  6.10360073,  7.66377554,  5.19577146,  9.75401449,
        5.08372149,  6.36004393, 11.27656488,  6.02588517, 10.06850677,
       12.44177641,  9.96391045,  5.5671389 ,  7.83486943,  6.97106912,
        8.35856487,  7.14837514,  8.12043983,  8.859416  , 12.67680439,
       12.30695098])}
{'P': tensor([[3.6654e-02, 2.1220e-01, 1.4839e-01,  ..., 9.4692e-05, 4.5955e-04,
         4.5985e-05],
        [3.6139e-03, 7.7225e-02, 3.4915e-02,  ..., 3.9382e-04, 5.9952e-04,
         5.1484e-05],
        [3.2407e-03, 4.5750e-02, 5.1502e-02,  ..., 1.7432e-04, 4.7357e-04,
         3.2357e-05],
        ...,
        [2.7579e-06, 3.9207e-06, 3.0709e-06,  ..., 3.9556e-02, 1.1542e-06,
         3.6264e-07],
        [1.5414e-02, 9.3883e-03, 1.3900e-02,  ..., 5.8836e-06, 6.8911e-02,
         5.8073e-04],
        [6.8995e-03, 4.4407e-03, 3.5167e-03,  ..., 8.5433e-06, 5.7805e-04,
         7.5580e-02]]), 'X': tensor([[  5.1196,  -2.7160,   4.2705],
        [  5.8902,  -1.3887,   4.0388],
        [  5.8323,  -1.5228,   5.1132],
        [  5.9193,  -1.3176,   5.1739],
        [  6.2419,  -0.2313,   4.2440],
        [  5.5158,  -2.1337,   3.5246],
        [  8.9284,  10.0036,  -6.9210],
        [ 11.4495,   6.3026,   6.8257],
        [  2.6614,  -4.5192,  -9.3641],
        [  5.9526,   3.2525,  -4.3821],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.1504,  -4.6726,   8.7908],
        [  9.4554,   9.4245,   9.8800],
        [  5.2608,  12.6977,   4.9166],
        [ 11.1085,  -4.9204,   0.0736],
        [  6.1066,  -0.7799,  -4.1746],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.7237,  -3.9966,  -5.0376],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.3691,   6.4760,   1.0787],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.7565,  11.8237,  -4.9882],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -2.0445,   7.8469,  -0.5857],
        [ -8.9606,  -4.0994,   1.7886],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.0961,   1.6529,   3.1553],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -5.6651,   6.4380,   1.3660],
        [ -1.6086,   7.8928,   4.3385],
        [ -0.2586,   7.8825,  -9.9840],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.6154,  -4.0640,  -7.9698],
        [ -0.0410,   7.8591,   9.3457],
        [ -2.1533, -10.5778,   1.0112],
        [  5.1472,  -2.6801,   8.1194],
        [  1.7659,   7.4127,   4.7871],
        [ -9.3411,  -2.6004,  -2.7062],
        [  1.6408,   7.4591,  -4.2294],
        [  0.4119,  14.1019,  -4.1870],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -5.1369,   6.7755,   8.3836],
        [  0.3231,  -4.7691,  -1.6120],
        [  5.9003,  12.3550,   2.1917],
        [  4.3554,   5.7393,  -9.0766],
        [  1.7228,  -4.7575,   8.3396],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -1.9600,   7.8577,   6.9121],
        [ -8.7409,  -4.6837,  -6.5840],
        [  3.9770, -10.6341,   8.0783],
        [  5.0233,   4.9647,  -3.1788],
        [  2.1771, -10.9776,  -3.2007],
        [  4.1348,   5.9514,  -0.8856],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.5382,   7.0672,   4.5097],
        [  2.2659,  13.7894,   6.5436],
        [ -8.0608,   3.9362,  -5.9567],
        [ -1.8788, -10.6567,   3.7068],
        [  2.6451,  -4.5250,  -0.5565],
        [  9.3836,   9.5075,  -5.9278],
        [ 12.2592,   3.9425,   2.0383],
        [  9.9217,  -6.7006,  -5.6728],
        [ -2.0721,   7.8432,   6.6380],
        [ -2.5056,   7.7720,   5.5878],
        [  0.1460,   7.8340,  -9.4614],
        [  4.8653,   5.1694,  -4.0823],
        [  3.7298,  13.3537,  -9.4653],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.6078, -10.9229,   6.2869],
        [  1.3103,  -4.8032,  -0.9300],
        [ 10.5678,   7.9261,  -0.6100],
        [  3.9085,   6.1512,  -2.2560],
        [ -6.2200,  -8.2737,   5.5079],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -3.9508,  -9.8515,   1.7923],
        [ -8.4267,  -5.3697,   2.4172],
        [ -3.2053, -10.1993,  -3.8736],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.7845,   3.6652,   0.9175],
        [  3.6249, -10.7255,   5.6605],
        [  6.0467,  -9.8369,  -6.5816],
        [  2.1504,  -4.6726,   8.7908],
        [ -9.1459,   1.4780,  -4.2403],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.8930,  12.3592,  -8.9201],
        [  5.7151,  -1.7700,   3.1542],
        [ -6.9838,  -7.4965,   0.9796],
        [  6.3122,  -9.6994,  -6.3282],
        [  3.5939,   6.4029,  -1.8357],
        [  6.3589,   0.8239,  -8.4230],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.0330,  -6.0765,   6.7945],
        [  9.1409,  -7.5789,   1.5814],
        [  5.9548,  -1.2273,  -2.6520],
        [ -8.3535,  -5.5118,   9.8399],
        [ -3.1204, -10.2345,   9.2288],
        [ -2.1382,   7.8339,   7.3524],
        [  0.1931,  -4.7491,   7.4385],
        [ -8.0330,  -6.0765,   6.7945],
        [  4.9957,  -2.8705,   6.5978],
        [  6.0310,  12.2795,  -6.4528],
        [  0.1206,  -4.7363,   8.7103],
        [  0.5600, -11.0321,  -0.8660]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0207, 0.0720,  ..., 0.8342, 0.2323, 0.7706],
        [0.0207, 0.0000, 0.0835,  ..., 0.8163, 0.2510, 0.7768],
        [0.0720, 0.0835, 0.0000,  ..., 0.8954, 0.1794, 0.8242],
        ...,
        [0.8342, 0.8163, 0.8954,  ..., 0.0000, 1.0661, 0.9973],
        [0.2323, 0.2510, 0.1794,  ..., 1.0661, 0.0000, 0.8426],
        [0.7706, 0.7768, 0.8242,  ..., 0.9973, 0.8426, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.79543038,  6.05165248,  6.02779201,  6.06416957,  6.24614905,
        5.91408504, 13.40850328, 13.06957425,  5.24461068,  6.78325151,
        8.01009083,  5.14371005, 13.35013374, 13.744387  , 12.1494023 ,
        6.1561646 , 10.53368644,  5.46244988, 12.32408516, 13.08414689,
       11.99633819, 13.61801526,  9.43112957,  8.10886021,  9.85384623,
       13.32645552,  9.24501829, 10.52587243,  8.57560275,  8.0550286 ,
        7.88677538, 13.00413829,  5.43943791,  7.85919687, 10.794755  ,
        5.80312949,  7.62011734,  9.69629   ,  7.6374603 , 14.10796466,
        8.50278896, 12.35421461,  8.50268322,  4.78002665, 13.69162795,
        7.20482224,  5.05981914, 10.53368644,  8.09842704,  9.91668182,
       11.35345345,  7.06272256, 11.1913598 ,  7.24677664, 12.24467041,
        7.50917551, 13.97430191,  8.97052115, 10.82106505,  5.24137909,
       13.35832725, 12.87752052, 11.97238624,  8.11227668,  8.16584537,
        7.83534987,  7.09887557, 13.86480175,  5.34250225, 11.22993218,
        4.978704  , 13.20988996,  7.28793663, 10.35093485,  9.54154096,
       10.61417181,  9.99211663, 10.6910768 , 11.57249984,  6.12886024,
        6.84796202, 11.32149259, 11.54672755,  5.14371005,  9.26455795,
        4.78258092, 13.6922425 ,  5.98285319, 10.24557047, 11.57249984,
        7.34250948,  6.41202986,  9.58072286, 10.95593729, 10.07238463,
       11.87412439,  6.07992623, 10.00802627, 10.69963954,  8.12043983,
        4.75303127, 10.07238463,  5.76165934, 13.68063098,  4.73784222,
       11.04629471])}
{'P': tensor([[2.1917e-02, 1.8281e-01, 1.5126e-01,  ..., 1.3536e-04, 1.1104e-05,
         7.1405e-06],
        [2.7384e-03, 4.1834e-02, 1.3435e-01,  ..., 2.6697e-04, 1.8000e-05,
         1.3598e-05],
        [3.2949e-03, 1.1791e-02, 4.9757e-02,  ..., 2.8134e-04, 2.0023e-05,
         1.5377e-05],
        ...,
        [1.6079e-04, 1.8985e-04, 1.7975e-04,  ..., 1.4891e-01, 3.0984e-05,
         1.7028e-04],
        [1.4271e-03, 1.0299e-03, 8.9875e-04,  ..., 8.4755e-05, 7.2681e-02,
         7.0708e-05],
        [2.5534e-04, 1.9475e-04, 1.7133e-04,  ..., 8.8183e-05, 1.5980e-01,
         1.3790e-01]]), 'X': tensor([[  5.1203,  -2.7151,   6.2110],
        [  5.5240,  -2.1198,   6.2055],
        [  5.6093,  -1.9705,   6.4807],
        [  5.4336,  -2.2676,   5.6933],
        [  6.1973,  -0.4372,   6.5720],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.2435,   4.0076,   8.1257],
        [ -3.0644,   7.6439,  -1.4322],
        [  5.3858,   4.4260,  -9.6769],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.1798,   8.5002,  -5.8893],
        [  0.3715,  -4.7756,   0.2312],
        [  3.2701,   6.6338,   9.9209],
        [ -0.3432,   7.8900,   7.1805],
        [  5.5240,  -2.1198,   6.2055],
        [  4.8674,  12.8875,   6.7622],
        [ -4.0409,   7.3169,  -3.5779],
        [  3.9022,  13.2906,   5.5788],
        [  6.2242,  -0.3171,   7.4978],
        [ -7.4807,   4.7463,   2.0616],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.6078, -10.9229,   6.2869],
        [  5.9587,  12.3216,   9.1695],
        [ -8.5859,  -5.0393,   0.5476],
        [ 12.4592,  -0.9168,  -6.7571],
        [  4.0078,  -3.8018,  -6.2168],
        [ -9.1749,  -3.3803,   6.4629],
        [ -6.8466,  -7.6488,   1.9648],
        [  8.2154,  -8.4262,   8.0171],
        [  5.8450,  -9.9356,   3.0767],
        [ -0.8569,   7.9156,  -6.5557],
        [  3.7237,  -3.9966,  -5.0376],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -5.2589,   6.7024,  -5.5809],
        [  9.3836,   9.5075,  -5.9278],
        [  6.3211,   1.7017,  -0.2164],
        [ -6.2668,   5.9808,   2.5614],
        [  5.5101,  -2.1433,   9.5442],
        [  4.1348,   5.9514,  -0.8856],
        [ -3.5029,   7.5138,   8.1853],
        [  2.5128, -10.9365,   9.9645],
        [  5.2608,  12.6977,   4.9166],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.5663,   7.0526,  -6.8769],
        [ 11.4703,  -4.2006,   7.4086],
        [ -0.8675, -10.8815,   6.8561],
        [ -2.0229, -10.6163,   8.6930],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.1706,   1.3863,   8.4032],
        [ -2.5161,   7.7699,  -9.3531],
        [ -4.4520,   7.1370,  -8.7867],
        [  0.6952,  -4.8060,   8.8976],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -1.3178, -10.7939,   0.4236],
        [  5.0348,  -2.8229,   3.5436],
        [  7.0171,  11.6439,  -5.0178],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.1440, -10.9811,   8.5281],
        [ -7.9298,   4.1367,  -3.9773],
        [ 12.6038,   0.7665,  -3.4439],
        [  1.0000,   7.6591,  -4.5011],
        [  9.9217,  -6.7006,  -5.6728],
        [  6.3048,   0.1469,  -9.1691],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.9587,  12.3216,   9.1695],
        [ 12.4837,   2.7510,   0.4449],
        [  2.5318,  13.7231,   6.3692],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.3246,   1.6696,  -0.1544],
        [  6.2975,   0.0947,   5.1791],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.2939,  -2.4784,  -4.5840],
        [ 12.6000,   1.3839,  -3.2992],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.3246,   1.6696,  -0.1544],
        [  6.2535,  -0.1709,  -9.4102],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.2435,   4.0076,   8.1257],
        [ -5.6792,  -8.7352,   4.0318],
        [  4.9531,   5.0576,   5.6794],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.1771, -10.9776,  -3.2007],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.4179,  -2.2925,  -3.4641],
        [ -6.4620,  -8.0449,   6.2719],
        [  1.8766,   7.3694,   1.1887],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -7.8359,   4.2735,   7.2943],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.5198,  -4.0923,   5.5088],
        [ -1.6220,   7.8917,   0.0315],
        [ -8.8058,   2.4850,  -1.7425],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.1905,   3.7253,   4.1175],
        [  4.0591,   6.0202,  -1.6339],
        [  6.9956,  11.6590,  -7.6876],
        [  0.7685,  -4.8098,  -2.9310],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.7845,   3.6652,   0.9175],
        [  3.5284, -10.7484,   5.3347],
        [  6.1929,   2.4501,  -6.9352],
        [ -6.0844,  -8.3955,   0.7705],
        [ -9.4378,  -1.8606,  -2.5668]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0091, 0.0168,  ..., 0.8168, 0.9985, 1.1744],
        [0.0091, 0.0000, 0.0135,  ..., 0.8153, 1.0063, 1.1809],
        [0.0168, 0.0135, 0.0000,  ..., 0.8285, 1.0141, 1.1910],
        ...,
        [0.8168, 0.8153, 0.8285,  ..., 0.0000, 1.0302, 0.9806],
        [0.9985, 1.0063, 1.0141,  ..., 1.0302, 0.0000, 0.2978],
        [1.1744, 1.1809, 1.1910,  ..., 0.9806, 0.2978, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.79562691,  5.91677188,  5.94535529,  5.88782913,  6.21275016,
        6.06416957, 12.88270018,  8.23525433,  6.97106912,  6.00663314,
       13.26209545,  4.79003361,  7.39599986,  7.89745107,  5.91677188,
       13.77604293,  8.35856487, 13.85158417,  6.2322769 ,  8.85939626,
        5.21280156, 11.22993218, 13.68672855,  9.95552626, 12.49291768,
        5.52416368,  9.77777346, 10.26545792, 11.76830981, 11.52733581,
        7.96181288,  5.46244988, 12.4013153 ,  9.5014146 ,  8.51928531,
       13.35832725,  6.54615887,  8.66273171,  5.91222346,  7.24677664,
        8.29022197, 11.22142195, 13.744387  , 12.76408562,  7.50500064,
       12.21532282, 10.91601655, 10.80728788, 12.97370371,  9.75075273,
        9.27474386,  8.16714836,  8.41171369,  4.85603886, 12.53285633,
       10.87409005,  5.77219077, 13.59480906,  8.62329466, 11.18839169,
        8.94395628, 12.62711252,  7.72415048, 11.97238624,  6.30647746,
       11.63724792, 13.68672855, 12.78327037, 13.95472331,  9.37296102,
        6.54127431,  6.29821673,  8.47371868,  5.84533345, 12.67576792,
        9.45995955,  6.54127431,  6.25586297,  4.83371341, 12.88270018,
       10.41909211,  7.07902036, 10.41909211, 11.1913598 , 12.16901615,
        5.8828986 , 10.31886066,  7.60463349,  9.43112957,  8.92549752,
       12.07225469, 12.22503754,  8.05668609,  9.14973319,  8.89458273,
        6.05113217,  8.99791316,  7.26074925, 13.59673406,  4.87083348,
        8.47371868,  6.84796202, 11.31276109,  6.65991877, 10.36844813,
        9.61942445])}
{'P': tensor([[2.4353e-02, 1.8436e-01, 1.8182e-01,  ..., 3.7015e-05, 8.3957e-04,
         1.3189e-05],
        [2.8321e-03, 4.8432e-02, 1.1147e-02,  ..., 1.0849e-04, 1.7842e-03,
         1.7592e-05],
        [2.4589e-03, 1.3282e-01, 3.7637e-02,  ..., 7.4842e-05, 1.2531e-03,
         1.3749e-05],
        ...,
        [3.0785e-07, 3.5862e-07, 3.3566e-07,  ..., 6.2304e-03, 4.7044e-06,
         6.0919e-08],
        [1.7626e-04, 2.3459e-04, 2.1200e-04,  ..., 7.4570e-03, 4.9857e-02,
         1.4806e-05],
        [1.2231e-03, 1.1128e-03, 1.0438e-03,  ..., 2.0154e-05, 2.7240e-03,
         1.1568e-01]]), 'X': tensor([[  5.1472,  -2.6801,   8.1194],
        [  5.6672,  -1.8631,   8.1020],
        [  5.6022,  -1.9833,   8.3081],
        [  5.5626,  -2.0536,   8.4433],
        [  6.0054,  -1.0902,   8.3079],
        [  5.8889,  -1.3916,   8.8993],
        [  6.3124,   0.2057,  -0.4593],
        [  3.3094,  13.4969,  -3.6507],
        [ -1.3178, -10.7939,   0.4236],
        [  2.5829,  -4.5466,  -8.4015],
        [  5.2188,  -2.5839,   2.1833],
        [  4.4633,   5.6287,  -0.7754],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.4753,  -0.8012,  -5.2050],
        [  1.3103,  -4.8032,  -0.9300],
        [ -7.4733,   4.7555,   6.2490],
        [ -6.7745,  -7.7264,   8.4684],
        [  8.6420,  -8.0580,   6.5781],
        [  0.3038,   7.8092,   0.4164],
        [  6.3048,   0.1469,  -9.1691],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.5919,   0.4035,  -5.8460],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.2339,  -3.6268,   6.3125],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.1666,   3.7653,   3.6075],
        [  6.7792,  11.8084,   1.6251],
        [  6.3534,   1.2990,  -7.2942],
        [  2.6614,  -4.5192,  -9.3641],
        [ -7.8280,   4.2848,  -7.1955],
        [  8.7715,  -7.9390,   5.2900],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.1120,  -4.6818,   8.6144],
        [ -1.5875,   7.8944,  -9.4286],
        [ -5.9938,  -8.4745,   3.9735],
        [  7.0171,  11.6439,  -5.0178],
        [ 11.5198,  -4.0923,   5.5088],
        [  0.9478,  -4.8144,   0.6371],
        [ 11.0275,  -5.0666,   8.4154],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.9883,   3.1536,   6.2320],
        [  8.9284,  10.0036,  -6.9210],
        [ 11.1469,   6.9237,   1.6160],
        [  4.8674,  12.8875,   6.7622],
        [ -7.8126,   4.3066,   5.1791],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -5.1369,   6.7755,   8.3836],
        [ -7.4887,   4.7364,   8.9074],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.3124,   0.2057,  -0.4593],
        [  4.1537,   5.9340,  -6.0423],
        [  0.9036,   7.6840,   9.9237],
        [  5.9026,   3.3837,  -0.6573],
        [  5.9193,  -1.3176,   5.1739],
        [  0.9074,  -4.8139,   8.1674],
        [ 11.5198,  -4.0923,   5.5088],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3460,   1.4252,   1.2291],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.4645,  -1.4880,  -2.0226],
        [  7.1441,  11.5529,   2.3232],
        [  1.5662,  13.9375,   4.5142],
        [ 11.5083,  -4.1177,   0.2389],
        [  0.6470, -11.0352,  -1.1931],
        [  9.0051,   9.9234,   5.4411],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.3604,   0.9042,  -6.5901],
        [  5.1203,  -2.7151,   6.2110],
        [  6.2905,   1.9366,   2.1169],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.2508,   1.0572,   2.9978],
        [  5.0233,   4.9647,  -3.1788],
        [  5.9587,  12.3216,   9.1695],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.6319,  11.9066,   3.5165],
        [ -9.2301,  -3.1537,  -2.7477],
        [  5.7845,   3.6652,   0.9175],
        [  4.5814, -10.4485,   2.5496],
        [  6.7792,  11.8084,   1.6251],
        [  2.0870,  -4.6877,   9.6823],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.5240,  -2.1198,   6.2055],
        [  5.6260,  -1.9401,   3.0259],
        [  5.8450,  -9.9356,   3.0767],
        [  1.7000,  -4.7609,   2.7935],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.3261,   7.1726,   6.2522],
        [  6.0047,   3.1062,  -6.7914],
        [  0.6470, -11.0352,  -1.1931],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.4671,  -0.5553,   0.1117],
        [  1.5383, -11.0274,  -4.9666],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3589,   0.8239,  -8.4230],
        [ 10.8218,  -5.4177,   6.8573],
        [  4.3459,  -3.5327,   0.3667],
        [  4.9337,  12.8566,  -3.6582],
        [  6.2036,   2.4030,  -1.2500],
        [  2.1771, -10.9776,  -3.2007]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0116, 0.0097,  ..., 0.8810, 0.5949, 0.9774],
        [0.0116, 0.0000, 0.0065,  ..., 0.8735, 0.5905, 0.9797],
        [0.0097, 0.0065, 0.0000,  ..., 0.8799, 0.5968, 0.9843],
        ...,
        [0.8810, 0.8735, 0.8799,  ..., 0.0000, 0.3604, 0.8143],
        [0.5949, 0.5905, 0.5968,  ..., 0.3604, 0.0000, 0.5898],
        [0.9774, 0.9797, 0.9843,  ..., 0.8143, 0.5898, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.80312949,  5.96556248,  5.94293185,  5.92952538,  6.10360073,
        6.05113217,  6.31576264, 13.89671314, 10.87409005,  5.22903117,
        5.82345867,  7.18354885,  9.69629   , 13.2430377 ,  9.50913849,
        4.978704  ,  8.8580671 , 10.27572673, 11.8159254 ,  7.81509631,
        6.30647746, 13.04156075, 12.59840193,  9.75401449,  5.5748624 ,
        9.14973319,  8.99276564, 13.61600536,  6.48486171,  5.24461068,
        8.92395381, 11.83075005,  7.53063885,  8.97052115,  5.13616224,
        8.05242591, 10.37998078, 13.59480906, 12.22503754,  4.90676242,
       12.13568598,  7.83486943, 13.01187484,  6.76791483, 13.40850328,
       13.12218013, 13.77604293,  8.92098752, 12.4217523 ,  8.50268322,
        8.86082124,  9.79814997,  6.31576264,  7.24326992,  7.73692436,
        6.80370254,  6.06416957,  4.89869205, 12.22503754,  5.5366698 ,
        6.50409372,  7.9337647 , 12.0495521 ,  7.83534987,  9.58072286,
       13.58333857, 14.02526505, 12.22276278, 11.05413545, 13.40024224,
       13.01728313,  6.42439344,  5.79562691,  6.58184474,  9.68324553,
       11.9120321 ,  5.40689332,  9.3109884 ,  7.06272256, 13.68672855,
        8.16714836, 13.62896593,  9.75401449,  6.84796202, 11.4088043 ,
       13.61600536,  5.13125513,  5.48276038,  5.91677188,  5.95111804,
       11.52733581,  5.05534609, 12.24467041,  7.5403822 ,  6.76056625,
       11.05413545, 12.3344536 ,  9.48337063, 11.1341798 ,  6.16512856,
        6.41202986, 12.10222204,  5.60064319, 13.77075174,  6.652742  ,
       11.1913598 ])}
{'P': tensor([[2.5781e-02, 1.8116e-01, 1.3072e-01,  ..., 1.1638e-04, 3.3568e-04,
         6.4844e-05],
        [3.3998e-03, 3.4404e-02, 1.9502e-01,  ..., 1.2410e-04, 5.4109e-04,
         6.8407e-05],
        [4.7030e-03, 6.0437e-03, 5.5154e-02,  ..., 1.5454e-04, 1.1240e-03,
         8.3224e-05],
        ...,
        [4.9262e-03, 5.4226e-03, 5.9901e-03,  ..., 3.1664e-02, 8.1113e-05,
         2.0742e-04],
        [7.0964e-05, 7.6696e-05, 8.6364e-05,  ..., 9.0893e-06, 8.0424e-02,
         2.4788e-06],
        [9.0180e-04, 8.1130e-04, 6.7893e-04,  ..., 3.8689e-04, 1.3189e-06,
         5.7131e-03]]), 'X': tensor([[  5.1667,  -2.6543,   1.2879],
        [  5.3860,  -2.3418,   1.0826],
        [  5.6202,  -1.9508,   0.7471],
        [  5.7963,  -1.6018,   1.8856],
        [  5.8275,  -1.5334,   2.0124],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.5717,   1.9266,   7.4903],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.4242, -10.1261,   3.9595],
        [ -3.7964,  -9.9294,   7.7147],
        [ -1.1885, -10.8211,   0.0865],
        [ -5.3382,   6.6533,   7.8054],
        [ -4.0633,  -9.7927,  -3.0825],
        [  8.5716,  10.3596,   6.8818],
        [  3.4604,  -4.1545,   1.8091],
        [  3.6154,  -4.0640,  -7.9698],
        [ -1.0407,   7.9166,  -6.6021],
        [ 10.2712,   8.3707,  -4.9743],
        [  2.4639, -10.9431,  -7.3718],
        [  5.4167, -10.1293,   6.2156],
        [  8.5198,  10.4091,  -8.6486],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.9260,  -4.4162,   8.3808],
        [ -5.9987,   6.1949,  -6.2114],
        [  1.7815,  -4.7481,   4.4289],
        [  2.6614,  -4.5192,  -9.3641],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.9026,   3.3837,  -0.6573],
        [ -5.5525,   6.5146,  -4.4984],
        [  4.9337,  12.8566,  -3.6582],
        [  5.1292,   4.8183,  -5.7997],
        [  9.3624,   9.5318,   0.2779],
        [ -8.1905,   3.7253,   4.1175],
        [  6.3413,   0.4883,  -6.8416],
        [  5.2188,  -2.5839,   2.1833],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.9655,   3.2172,  -7.5953],
        [ -8.6801,   2.7809,  -0.8443],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.1886,  -2.6250,   6.4677],
        [ -2.6069, -10.4298,  -6.1995],
        [  5.9003,  12.3550,   2.1917],
        [  3.3744,   6.5623,  -6.3636],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.4167, -10.1293,   6.2156],
        [  5.0617,  12.7957,   8.3544],
        [ 11.7779,   5.5127,  -2.0699],
        [ -7.5401,   4.6718,   9.0033],
        [ -3.3116, -10.1539,   7.5919],
        [ -7.8849,  -6.3115,  -4.7479],
        [  5.5158,  -2.1337,   3.5246],
        [  5.1292,   4.8183,  -5.7997],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.5620,   7.0549,   8.4825],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.7796,  -3.4696,   9.0088],
        [  2.5663,   7.0526,  -6.8769],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.0345,   3.0175,  -1.4424],
        [  5.8275,  -1.5334,   2.0124],
        [  6.2036,   2.4030,  -1.2500],
        [  1.1350,  14.0104,   0.6321],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3515,   0.6418,  -6.5823],
        [ -1.8367,   7.8718,  -8.6523],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.3301,   0.3638,  -7.5685],
        [  2.5469,  13.7192,  -9.6935],
        [  6.2535,  -0.1709,  -9.4102],
        [  9.4516,  -7.2501,  -1.8868],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.4167, -10.1293,   6.2156],
        [  5.0617,  12.7957,   8.3544],
        [  4.1637,  13.1898,  -1.9189],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 11.2571,   6.7072,   8.3693],
        [  6.3542,   0.6934,  -3.3840],
        [  5.7218,  -9.9934,   6.5764],
        [ 12.4996,   2.6335,  -6.3166],
        [  4.9957,  -2.8705,   6.5978],
        [  3.9913,  -3.8139,  -6.8311],
        [ 10.1983,  -6.3430,  -0.3200],
        [  0.8572,  14.0501,  -8.3364],
        [  5.9526,   3.2525,  -4.3821],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.1519, -10.2387,   9.8191],
        [  2.5382,   7.0672,   4.5097],
        [  6.1752,   2.5242,   3.5874],
        [  6.3549,   0.7096,   6.9198],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.6614,  -4.5192,  -9.3641],
        [  4.1297,  -3.7098,  -3.9838],
        [ 11.3691,   6.4760,   1.0787],
        [ -0.4341, -10.9473,   7.4509],
        [ 11.5456,   6.0861,   8.1101],
        [  6.3087,   1.8053,  -1.9481],
        [  0.6363, -11.0348,   6.1414],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.8154,   2.4611,   2.5198],
        [  5.7485,  -9.9811,  -3.2355],
        [  2.5974,   7.0363,  -7.7810],
        [  3.6249, -10.7255,   5.6605]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4064e-02, 3.5989e-02,  ..., 5.0615e-01, 5.1729e-01,
         6.9361e-01],
        [1.4064e-02, 1.0727e-04, 2.2031e-02,  ..., 5.0122e-01, 5.0623e-01,
         7.0594e-01],
        [3.5989e-02, 2.2031e-02, 0.0000e+00,  ..., 4.9102e-01, 4.8722e-01,
         7.2357e-01],
        ...,
        [5.0615e-01, 5.0122e-01, 4.9102e-01,  ..., 0.0000e+00, 2.7416e-01,
         8.0180e-01],
        [5.1729e-01, 5.0623e-01, 4.8722e-01,  ..., 2.7416e-01, 0.0000e+00,
         1.0163e+00],
        [6.9361e-01, 7.0594e-01, 7.2357e-01,  ..., 8.0180e-01, 1.0163e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.80860165,  5.87305005,  5.94909709,  6.01356004,  6.02588517,
        6.2134961 , 12.71843462,  9.7751189 , 11.48734978, 10.63038565,
       10.88618114,  8.53014915, 10.60226614, 13.44593519,  5.40689332,
        5.43943791,  7.98469166, 13.25017211, 11.21703654, 11.48663767,
       13.45124917,  9.97220799,  5.29755487,  8.62329466,  5.0713364 ,
        5.24461068, 10.23946247,  8.50278896, 12.53285633,  6.80370254,
        8.55982105, 13.77075174,  7.03733775, 13.3607379 ,  8.99791316,
        6.36004393,  5.82345867, 12.10942412,  6.77777606,  9.11473391,
       10.48840476,  5.81481044, 10.7506423 , 13.69162795,  7.37902766,
        5.26067019, 11.48663767, 13.76047999, 13.00413829,  8.87007495,
       10.68031351, 10.09979747,  5.91408504,  7.03733775, 13.27994304,
        7.50564223, 12.19805394, 12.27993098,  7.50500064,  9.17595899,
        9.99211663,  6.74686071,  6.02588517,  6.652742  , 14.05633441,
        6.30647746,  6.383897  ,  8.08320235, 12.889695  ,  6.34059021,
       13.95361098,  6.25586297, 11.9120321 , 12.35625694,  8.47371868,
       11.48663767, 13.76047999, 13.83139024, 10.35485331, 13.103709  ,
        6.39187285, 11.51557166, 12.77402351,  5.76165934,  5.5205189 ,
       12.00995898, 14.07623266,  6.78325151,  8.27853076, 11.46176259,
        7.50917551,  6.67121748,  6.39438044, 13.12299145,  5.24461068,
        5.5513072 , 13.08414689, 10.95593729, 13.05148826,  6.5618968 ,
       11.05317597, 14.09176259,  9.15253344, 11.51811603,  7.50035355,
       11.32149259])}
{'P': tensor([[2.6027e-02, 2.1851e-01, 1.2494e-01,  ..., 3.9612e-04, 1.0052e-03,
         6.8061e-04],
        [2.7264e-03, 3.6070e-02, 2.1759e-01,  ..., 4.7147e-04, 9.0935e-04,
         8.0015e-04],
        [1.5640e-03, 1.9053e-03, 4.3160e-02,  ..., 3.7165e-04, 3.0035e-04,
         6.9015e-04],
        ...,
        [5.6006e-02, 4.7171e-02, 2.4663e-02,  ..., 1.0830e-01, 1.1427e-02,
         1.1006e-03],
        [6.4655e-03, 9.4872e-03, 1.8352e-02,  ..., 1.1882e-04, 5.4638e-02,
         5.9403e-04],
        [1.8980e-03, 2.1056e-03, 3.4738e-03,  ..., 2.4433e-04, 3.1210e-04,
         1.4149e-01]]), 'X': tensor([[  5.1787,  -2.6383,  -4.2533],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.3236,   0.3011,  -4.6668],
        [ -1.3940, -10.7772,   6.7835],
        [  6.3550,   0.7129,   0.4917],
        [  3.4140,  -4.1803,  -8.8224],
        [  7.0358,  -9.2784,  -9.8479],
        [  6.2038,   2.4020,  -3.1692],
        [  2.6451,  -4.5250,  -0.5565],
        [ 10.5955,   7.8824,  -3.7006],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.3759,  -1.3943,   6.4659],
        [  9.3836,   9.5075,  -5.9278],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.0140,  -1.0661,  -2.5040],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.2911,  -2.4824,  -3.6877],
        [  3.3744,   6.5623,  -6.3636],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.8400,   2.3991,   1.3457],
        [  4.4633,   5.6287,  -0.7754],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.2435,   4.0076,   8.1257],
        [ -0.1963,   7.8765,  -1.1371],
        [  6.1973,  -0.4372,   6.5720],
        [ -7.6525,  -6.6534,   5.1721],
        [ 12.5154,   2.5072,  -9.3274],
        [ 11.5456,   6.0861,   8.1101],
        [ 12.5832,   1.7543,  -7.6134],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -6.8466,  -7.6488,   1.9648],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 12.2692,  -1.8882,   3.9253],
        [  0.9967,  -4.8144,  -9.7370],
        [ -2.6485, -10.4151,   0.8766],
        [  6.3534,   1.2990,  -7.2942],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.9085,   6.1512,  -2.2560],
        [  9.9750,  -6.6338,  -1.2440],
        [ -9.4309,  -0.0599,  -4.0830],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.6970,   6.9826,   9.7350],
        [  0.1041, -11.0045,   2.4311],
        [ -5.3328,  -8.9992,   7.3440],
        [  3.3094,  13.4969,  -3.6507],
        [ 11.5456,   6.0861,   8.1101],
        [  5.9912,   3.1453,  -3.9989],
        [ 12.5832,   1.7543,  -7.6134],
        [  6.3122,  -9.6994,  -6.3282],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.3124,   0.2057,  -0.4593],
        [  6.3211,   1.7017,  -0.2164],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.1905,   3.7253,   4.1175],
        [ -9.1907,  -3.3180,  -5.5943],
        [  1.8719,   7.3713,  -3.7465],
        [ -8.6801,   2.7809,  -0.8443],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.2974,   0.8371,   2.4852],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.2038,   2.4020,  -3.1692],
        [  0.9074,  -4.8139,   8.1674],
        [ -8.0829,  -5.9940,   5.0564],
        [  0.9074,  -4.8139,   8.1674],
        [  6.1730,  -0.5375,  -8.8861],
        [  5.6260,  -1.9401,   3.0259],
        [  9.2979,   9.6048,   7.3138],
        [  1.0000,   7.6591,  -4.5011],
        [ 11.6801,  -3.7196,  -0.9181],
        [  7.3724,  -9.0577,  -8.9843],
        [  3.3167,   6.6022,   6.3554],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 11.0373,   7.1299,  -8.5895],
        [  2.5382,   7.0672,   4.5097],
        [ -7.9298,   4.1367,  -3.9773],
        [  3.9895,   6.0816,  -6.5235],
        [  1.4489,   7.5256,   4.4529],
        [ -2.0721,   7.8432,   6.6380],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.1994,   7.6035,   2.4295],
        [ -8.9178,   2.1936,   7.5388],
        [  5.6672,  -1.8631,   8.1020],
        [ -0.5385,   7.9037,   6.3830],
        [  0.5250, -11.0306,   5.5358],
        [ -9.1178,   1.5784,   2.1410],
        [  5.1787,  -2.6383,  -4.2533],
        [  0.4996,   7.7738,  -4.4711],
        [  6.3505,   0.6235,  -8.5849],
        [  4.9452, -10.3187,  -7.8621],
        [  0.7275,  14.0667,  -0.9815],
        [ -8.6229,   2.9068,   2.4429],
        [  2.6451,  -4.5250,  -0.5565],
        [  6.0305,   3.0296,   0.1375],
        [ -9.4203,  -2.0340,   2.5856],
        [ -8.8363,   2.4086,  -5.6296],
        [  7.0642,  11.6104,  -9.8316],
        [  0.1677,  -4.7447,  -6.1529],
        [  9.9217,  -6.7006,  -5.6728],
        [  5.8323,  -1.5228,   5.1132]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0096, 0.0191,  ..., 0.1162, 0.2168, 0.6424],
        [0.0096, 0.0000, 0.0164,  ..., 0.1205, 0.2155, 0.6479],
        [0.0191, 0.0164, 0.0000,  ..., 0.1351, 0.2318, 0.6347],
        ...,
        [0.1162, 0.1205, 0.1351,  ..., 0.0000, 0.1354, 0.7093],
        [0.2168, 0.2155, 0.2318,  ..., 0.1354, 0.0000, 0.8429],
        [0.6424, 0.6479, 0.6347,  ..., 0.7093, 0.8429, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.81199422,  5.90035324,  6.1561646 ,  6.12886024,  5.84533345,
        6.33075682, 10.86693708,  6.3949043 ,  5.39723241, 11.64436684,
        6.65259775,  5.24137909, 13.20597932,  5.81199422, 12.45418236,
       13.35832725,  7.71075226,  9.53015376,  6.10774544, 10.31613269,
        5.84450127,  7.37902766,  8.41148722,  9.85384623,  9.15976746,
        7.18354885, 11.05672136, 12.88270018,  7.8788965 ,  6.21275016,
       10.14044489, 12.76408562, 13.05148826, 12.70489563,  9.62958973,
       10.26545792,  7.9337647 , 12.10942412, 12.41366754,  4.91652556,
       10.74655346,  6.48486171, 12.91143491,  7.28793663, 11.97949273,
        9.43112957,  5.22903117,  7.48538905, 11.0050379 , 10.4606265 ,
       13.89671314, 13.05148826,  6.76661595, 12.70489563, 11.57249984,
       13.26590071,  6.31576264,  6.54615887, 11.1913598 ,  8.99791316,
        9.7712321 ,  7.60529892,  9.11473391, 11.64746882,  9.3349858 ,
        8.36995953,  9.70108409,  6.65259775,  4.89869205, 10.06286074,
        4.89869205,  6.19633139,  5.95111804, 13.36800073,  7.72415048,
       12.25806559, 11.67875855,  7.38844703,  9.61942445, 13.1399181 ,
        7.50917551,  8.94395628,  7.27340834,  7.66377554,  8.11227668,
       12.04924159,  7.69752262,  9.18358562,  5.96556248,  7.92200342,
       11.04313641,  9.25336371,  5.81199422,  7.78980299,  6.38105015,
       11.4424816 , 14.08549802,  9.09963894,  5.24137909,  6.74872981,
        9.63742669,  9.15866607, 13.59056613,  4.74770899, 11.97238624,
        6.02779201])}
{'P': tensor([[2.4068e-02, 1.9763e-01, 1.6948e-01,  ..., 2.2418e-03, 4.0551e-03,
         8.5493e-06],
        [3.2998e-03, 5.2149e-02, 1.2358e-02,  ..., 3.4576e-03, 7.5631e-03,
         1.8826e-05],
        [3.6334e-03, 1.4326e-01, 4.4607e-02,  ..., 3.4448e-03, 6.4275e-03,
         1.6852e-05],
        ...,
        [9.7871e-04, 1.7865e-03, 1.6368e-03,  ..., 6.4400e-02, 2.3283e-02,
         2.7568e-06],
        [5.2567e-04, 9.5330e-04, 8.1286e-04,  ..., 7.8670e-04, 6.8557e-02,
         1.5593e-06],
        [1.3035e-04, 9.7627e-05, 1.1052e-04,  ..., 1.6068e-05, 3.7535e-06,
         8.1156e-02]]), 'X': tensor([[  5.1886,  -2.6250,   6.4677],
        [  5.6093,  -1.9705,   6.4807],
        [  5.5240,  -2.1198,   6.2055],
        [  6.1973,  -0.4372,   6.5720],
        [  6.2895,   0.0405,   5.7652],
        [  5.5424,  -2.0885,   7.2487],
        [ -2.1817,   7.8274,  -4.6298],
        [  1.9505,  -4.7171,   9.9868],
        [  4.5800,   5.5034,  -3.9432],
        [ -1.8788, -10.6567,   3.7068],
        [  6.3542,   0.6934,  -3.3840],
        [  1.3339,   7.5627,  -0.9034],
        [  8.2422,  -8.4041,  -1.4962],
        [  6.3413,   0.4883,  -6.8416],
        [  1.5662,  13.9375,   4.5142],
        [ -9.4630,  -1.5145,   0.2983],
        [ -9.1795,  -3.3623,   8.5259],
        [ 11.0275,  -5.0666,   8.4154],
        [  2.1379,  -4.6757,   6.4981],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.4179,  -2.2925,  -3.4641],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.0588,  -6.0341,  -2.3006],
        [  9.6816,   9.1536,  -5.8543],
        [  2.9260,  -4.4162,   8.3808],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.1472,  -2.6801,   8.1194],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.3211,   1.7017,  -0.2164],
        [  0.1460,   7.8340,  -9.4614],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.7420,  -4.4896,  -3.7139],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.3602,  -2.3808,   7.2098],
        [  9.6662,   9.1725,   8.1056],
        [ -1.9646,   7.8571,   3.3817],
        [  6.2365,  -0.2578,  -6.0212],
        [  5.0348,  -2.8229,   3.5436],
        [  2.5469,  13.7192,  -9.6935],
        [ -3.5173,   7.5091,  -8.9585],
        [  0.3724,   7.7974,  -6.8722],
        [ -9.0961,   1.6529,   3.1553],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.1813,  -3.3550,   1.8827],
        [  0.6363, -11.0348,   6.1414],
        [  3.9770, -10.6341,   8.0783],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.0845,  -2.5772,  -4.6073],
        [  1.1139,   7.6281,  -2.5680],
        [ -1.6275, -10.7221,   3.2376],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.0348,  -2.8229,   3.5436],
        [ -1.6086,   7.8928,   4.3385],
        [ 12.2592,   3.9425,   2.0383],
        [ 11.9936,  -2.8677,   6.0614],
        [ -7.4887,   4.7364,   8.9074],
        [  6.9314,  -9.3435,  -1.2729],
        [  9.5261,  -7.1675,   1.4206],
        [  5.1292,   4.8183,  -5.7997],
        [  2.5620,   7.0549,   8.4825],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.2508,   1.0572,   2.9978],
        [  5.6314,  -1.9302,  -9.6543],
        [ -5.0550,   6.8231,  -9.2502],
        [  2.8994,   6.8675,  -9.5987],
        [ 11.0891,  -4.9558,  -2.3178],
        [  1.8721,  -4.7322,   0.5524],
        [ -2.0840,   7.8415,  -7.8657],
        [  2.0547,   7.2957,  -1.9343],
        [ 11.5568,   6.0602,   9.3244],
        [  4.3377,   5.7570,   1.3712],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.3453,  -1.5462,   7.0212],
        [ -1.8367,   7.8718,  -8.6523],
        [  2.1504,  -4.6726,   8.7908],
        [ -6.2668,   5.9808,   2.5614],
        [ -8.8835,  -4.3178,   9.3080],
        [ -5.1369,   6.7755,   8.3836],
        [  4.2612, -10.5514,  -6.6268],
        [  3.5939,   6.4029,  -1.8357],
        [  5.9526,   3.2525,  -4.3821],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.6050,   2.9451,  -0.7012],
        [  9.7328,   9.0902,  -0.8917],
        [ -9.0574,   1.7801,   8.8986],
        [ -3.4488, -10.0934,   2.5195],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.9587,  12.3216,   9.1695],
        [  5.5158,  -2.1337,   3.5246],
        [  3.3094,  13.4969,  -3.6507],
        [ 10.6283,  -5.7253,   7.4878],
        [  8.8474,  10.0868,  -8.8644],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -8.9282,   2.1648,   8.4894],
        [ -8.5607,  -5.0938,  -5.4531],
        [  0.7275,  14.0667,  -0.9815],
        [  5.7218,  -9.9934,   6.5764],
        [ -2.1817,   7.8274,  -4.6298],
        [  5.6248,   3.9984,   8.4933],
        [  3.6249, -10.7255,   5.6605],
        [  3.2159, -10.8166,  -1.0406],
        [  6.3569,   0.7591,   6.5011],
        [ 12.0042,  -2.8350,   2.6864],
        [ 12.4877,  -0.7222,   8.8347],
        [ -9.4766,  -1.1116,  -2.1706]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0090, 0.0159,  ..., 0.2226, 0.2886, 1.1412],
        [0.0090, 0.0000, 0.0135,  ..., 0.2213, 0.2869, 1.1475],
        [0.0159, 0.0135, 0.0000,  ..., 0.2080, 0.3003, 1.1376],
        ...,
        [0.2226, 0.2213, 0.2080,  ..., 0.0000, 0.5023, 1.0262],
        [0.2886, 0.2869, 0.3003,  ..., 0.5023, 0.0000, 1.3939],
        [1.1412, 1.1475, 1.1376,  ..., 1.0262, 1.3939, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.81481044,  5.94535529,  5.91677188,  6.21275016,  6.28961988,
        5.92281142,  8.12580976,  5.10446653,  7.15990676, 10.82106505,
        6.39187285,  7.67939546, 11.77124373,  6.36004393, 14.02526505,
        9.58347456,  9.77588801, 12.13568598,  5.14124612,  6.27905292,
        5.8828986 ,  9.75200848, 10.0674887 , 13.32374056,  5.29755487,
       12.77402351,  5.80312949,  9.14973319,  6.54615887,  7.83534987,
       11.48434767,  9.53015376,  5.26067019,  5.18047638,  5.86519525,
       13.32557133,  8.09899874,  6.24187044,  5.77219077, 13.95361098,
        8.29203356,  7.80625981,  9.24501829,  8.97052115,  9.7751189 ,
       11.05317597, 11.35345345,  4.83371341, 12.35625694,  7.70897675,
       10.84493732, 10.00802627,  5.77219077,  8.0550286 , 12.87752052,
       12.33167382,  8.86082124, 11.63383221, 11.92133129,  7.03733775,
        7.50564223, 12.77402351,  9.3109884 ,  5.95297459,  8.49161154,
        7.4544866 , 12.1460931 ,  5.08910491,  8.1137452 ,  7.57946459,
       13.0493368 ,  7.20826219,  5.91677188, 12.44177641,  8.08320235,
        5.14371005,  8.66273171,  9.87720102,  8.50268322, 11.37939695,
        7.34250948,  6.78325151, 11.72749923,  9.09502385, 13.31764181,
        9.23071698, 10.66631784, 12.3489324 , 13.68672855,  5.91408504,
       13.89671314, 12.07225469, 13.41713661,  9.50913849,  9.18690018,
        9.96152788, 14.08549802, 11.51557166,  8.12580976,  6.90115901,
       11.32149259, 11.2845608 ,  6.4020404 , 12.3344536 , 12.5085995 ,
        9.54154096])}
{'P': tensor([[3.2010e-02, 2.1507e-01, 2.0743e-01,  ..., 2.4470e-05, 2.2677e-03,
         2.3766e-03],
        [2.2705e-03, 4.2797e-02, 1.1241e-02,  ..., 3.3456e-05, 4.1197e-03,
         2.1306e-03],
        [2.2001e-03, 1.2914e-01, 3.8397e-02,  ..., 2.9750e-05, 3.4392e-03,
         2.0706e-03],
        ...,
        [1.0274e-04, 6.0128e-05, 6.1651e-05,  ..., 1.9027e-02, 5.9496e-06,
         3.3798e-05],
        [2.1596e-04, 2.7002e-04, 2.5977e-04,  ..., 2.2997e-04, 8.4400e-02,
         4.9722e-05],
        [6.5133e-03, 1.0947e-02, 1.1316e-02,  ..., 4.6331e-05, 6.8362e-04,
         1.2293e-01]]), 'X': tensor([[ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0178, 0.0240,  ..., 0.8887, 0.3958, 0.4190],
        [0.0178, 0.0000, 0.0097,  ..., 0.9041, 0.4091, 0.4079],
        [0.0240, 0.0097, 0.0000,  ..., 0.9040, 0.4182, 0.3983],
        ...,
        [0.8887, 0.9041, 0.9040,  ..., 0.0000, 0.8578, 1.0415],
        [0.3958, 0.4091, 0.4182,  ..., 0.8578, 0.0000, 0.8127],
        [0.4190, 0.4079, 0.3983,  ..., 1.0415, 0.8127, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.82345867,  6.02588517,  6.01356004,  6.1463896 ,  6.2134961 ,
        6.27715105,  5.91677188, 14.10796466,  7.54127425,  9.68324553,
        7.50917551, 13.1471313 , 12.77402351,  7.48311002,  9.49511524,
        4.78002665, 10.75443106,  5.70591986, 10.91601655,  8.24808832,
        4.78002665, 13.96746816, 12.3344536 ,  7.92200342, 12.70489563,
       13.103709  , 12.22276278, 10.91805683, 11.11969815, 11.9120321 ,
       10.75443106, 11.83570132,  7.48538905,  8.99791316,  8.5724042 ,
       11.85743425,  7.26074925, 12.95702647, 13.40850328, 10.26545792,
        7.38844703,  9.87720102,  5.19986797,  8.29022197,  6.84796202,
        9.77588801,  5.5513072 ,  5.79562691,  6.49538841,  7.18354885,
       10.24557047, 11.51811603,  5.64464807, 12.17901033, 13.58101794,
       13.32557133,  8.28968067,  9.43737523, 13.12299145,  5.21280156,
        9.5642993 , 10.66631784, 13.55868684, 13.2778378 ,  8.57560275,
        4.90676242, 12.10942412, 12.49291768, 11.11969815, 13.93790235,
       13.40024224,  8.50278896, 10.06437598,  6.19633139,  6.18668463,
       13.3614598 , 13.60275211,  6.65368551, 11.28093507,  6.34059021,
       14.10796466, 10.5575601 ,  4.89121805, 12.19805394, 10.75443106,
        5.91677188, 13.03002338,  5.5513072 ,  5.39450305,  9.58072286,
       11.63502679,  7.88677538,  5.87132718, 10.60226614, 10.41909211,
        8.66273171, 11.62158554,  7.33350642,  8.25828409,  9.78285424,
        4.91336994, 11.4890861 , 13.58101794,  9.62238828,  7.15100303,
        5.90035324])}
{'P': tensor([[3.1854e-02, 1.5926e-01, 1.4434e-01,  ..., 1.3195e-05, 3.1848e-05,
         2.0822e-04],
        [2.9913e-03, 3.1016e-02, 1.4161e-01,  ..., 8.3234e-06, 3.1885e-05,
         2.6961e-04],
        [1.1786e-03, 1.8154e-03, 3.4074e-02,  ..., 6.6454e-06, 6.1719e-05,
         7.2044e-04],
        ...,
        [1.2903e-04, 1.1633e-04, 1.8222e-04,  ..., 8.5602e-02, 7.9895e-06,
         1.8359e-05],
        [3.8424e-05, 2.9762e-05, 2.7427e-05,  ..., 1.1747e-05, 7.1010e-02,
         7.9362e-06],
        [1.7612e-06, 1.8229e-06, 2.5156e-06,  ..., 3.6388e-08, 1.3249e-03,
         1.4640e-02]]), 'X': tensor([[ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0110, 0.0096,  ..., 1.0550, 0.7196, 0.2012],
        [0.0110, 0.0000, 0.0069,  ..., 1.0648, 0.7304, 0.1938],
        [0.0096, 0.0069, 0.0000,  ..., 1.0590, 0.7280, 0.1917],
        ...,
        [1.0550, 1.0648, 1.0590,  ..., 0.0000, 0.6415, 1.1276],
        [0.7196, 0.7304, 0.7280,  ..., 0.6415, 0.0000, 0.8880],
        [0.2012, 0.1938, 0.1917,  ..., 1.1276, 0.8880, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.83560009,  5.95297459,  6.19633139,  6.23262289,  6.25586297,
        6.30647746,  4.76969364, 11.59512194,  7.52117132,  9.41605684,
        5.43943791,  5.68942572,  4.98461983, 10.1979669 , 13.59673406,
       12.4980845 ,  4.74770899, 10.31613269, 13.27994304, 10.00410625,
        6.25586297,  8.23525433,  8.12580976,  5.1805345 , 13.40024224,
       10.41909211, 13.68672855,  6.54693987,  9.34061938, 13.94008948,
       13.96746816,  6.61386436, 13.06498554,  7.32950385,  5.07306274,
       12.5085995 ,  9.562174  ,  8.29203356, 13.3614598 , 14.09176259,
        7.46053281,  9.24501829, 13.20597932, 12.62711252,  8.09984231,
        8.09704052, 13.6922425 , 10.66631784, 12.85032836,  7.57685908,
        8.19703512, 12.32408516, 12.27835996, 13.3607379 ,  4.89121805,
        6.28961988,  9.11473391, 10.78533838,  9.41605684, 13.60275211,
        8.31625171, 13.37121923,  8.04483495,  6.41202986,  6.4020404 ,
       12.59840193,  9.99211663,  5.62005629,  5.05444288, 10.67322527,
       11.48434767,  6.69389431, 12.71843462, 13.1399181 , 13.02818702,
        7.6374603 ,  5.79543038, 13.59170809, 11.4890861 ,  5.64865173,
        9.94173708,  7.06272256,  5.18047638,  7.53884104,  5.92952538,
       11.1341798 , 10.60226614, 13.77075174, 13.61801526,  7.73692436,
        6.33075682,  5.5671389 ,  7.89745107,  7.73151309,  5.92281142,
       11.11038551, 11.28215429, 12.97370371,  7.06169391,  5.95111804,
       11.70168755,  8.72896381,  9.75401449, 11.72802491,  8.72896381,
       13.89974693])}
{'P': tensor([[3.2124e-02, 1.5954e-01, 1.5296e-01,  ..., 1.6816e-05, 8.3093e-04,
         4.9812e-04],
        [5.3482e-03, 3.4758e-02, 1.5804e-01,  ..., 1.7636e-05, 1.0251e-03,
         4.4049e-04],
        [1.3752e-03, 1.6230e-03, 3.3766e-02,  ..., 1.1968e-05, 1.6967e-03,
         1.3863e-04],
        ...,
        [3.6478e-04, 3.4053e-04, 1.3699e-04,  ..., 1.7407e-02, 3.8564e-06,
         2.4161e-04],
        [2.1166e-04, 2.1627e-04, 2.1608e-04,  ..., 2.9288e-04, 1.5072e-01,
         8.6203e-06],
        [3.0529e-02, 3.1891e-02, 6.1774e-02,  ..., 8.9691e-06, 3.3969e-04,
         2.5420e-01]]), 'X': tensor([[  5.2911,  -2.4824,  -3.6877],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.3542,   0.6934,  -3.3840],
        [ -4.9217,   6.8979,  -1.7667],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.3550,   0.7129,   0.4917],
        [  4.3523, -10.5232,   8.2138],
        [ -7.0253,  -7.4493,  -1.5978],
        [  8.7131,  10.2216,   7.7915],
        [  6.3549,   0.7096,   6.9198],
        [ -8.7544,   2.6096,   3.3146],
        [  3.1756, -10.8247,   6.8882],
        [ 12.6035,   0.7494,   7.1536],
        [ 10.0567,   8.6697,   4.6329],
        [  5.7963,  -1.6018,   1.8856],
        [ -3.1204, -10.2345,   9.2288],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.9505,  -4.7171,   9.9868],
        [  4.0421,  -3.7765,  -8.3631],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.1497,   7.8334,  -5.4412],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.7232,  -4.7574,   3.1270],
        [ -6.0844,  -8.3955,   0.7705],
        [  3.2809,  13.5061,  -4.0199],
        [ -5.9987,   6.1949,  -6.2114],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.4697,  -0.6157,  -6.8472],
        [ 12.3609,   3.4769,  -8.5762],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.2517,  -4.6495,   2.9800],
        [  2.3589,  13.7668,  -3.6501],
        [ -8.0087,   4.0173,   9.1091],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.8318,   3.5567,  -4.7692],
        [  7.0664,  -9.2590,   8.0879],
        [  5.0233,   4.9647,  -3.1788],
        [  6.2576,   2.1359,  -3.3398],
        [  2.5974,   7.0363,  -7.7810],
        [ -5.3179,  -9.0101,   9.4472],
        [  5.2199,   4.6861,  -1.9304],
        [ 12.3453,  -1.5462,   7.0212],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.3236,   0.3011,  -4.6668],
        [  5.0233,   4.9647,  -3.1788],
        [  4.5954,  13.0099,  -4.4439],
        [  4.1537,   5.9340,  -6.0423],
        [  5.9883,   3.1536,   6.2320],
        [  4.5430,   5.5438,  -3.8669],
        [ -3.4986,   7.5152,   9.0712],
        [  0.9074,  -4.8139,   8.1674],
        [  0.8572,  14.0501,  -8.3364],
        [ -5.0943,  -9.1682,   0.5073],
        [  2.9410,   6.8428,   5.4458],
        [  2.5974,   7.0363,  -7.7810],
        [  0.3493,  -4.7727,  -6.5682],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.1178,   1.5784,   2.1410],
        [  2.5620,   7.0549,   8.4825],
        [ -5.3179,  -9.0101,   9.4472],
        [ -9.1706,   1.3863,   8.4032],
        [  8.1217,  -8.5024,  -0.6149],
        [ -8.0829,  -5.9940,   5.0564],
        [ 10.8218,  -5.4177,   6.8573],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -3.5903,   7.4847,  -8.9112],
        [ -3.5173,   7.5091,  -8.9585],
        [ -8.6050,   2.9451,  -0.7012],
        [  9.6662,   9.1725,   8.1056],
        [  2.1824,  -4.6647,   1.1985],
        [  5.4167, -10.1293,   6.2156],
        [  2.2644,  -4.6434,   9.7508],
        [ -6.6998,   5.5954,   0.0823],
        [  5.4167, -10.1293,   6.2156],
        [ -3.2736, -10.1703,   3.8712],
        [  3.9515,  -3.8425,   6.1729],
        [ -6.6998,   5.5954,   0.0823],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.2975,   0.0947,   5.1791],
        [ -1.6275, -10.7221,   3.2376],
        [ -9.2974,   0.8371,   2.4852],
        [  2.4697,  -4.5836,   1.4593],
        [  6.0955,   2.8210,  -4.3321],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.4716,  -0.6673,   7.6068],
        [  9.6816,   9.1536,  -5.8543],
        [  3.5939,   6.4029,  -1.8357],
        [  8.5198,  10.4091,  -8.6486],
        [ -3.5520,   7.4976,   9.1522],
        [  6.1755,   2.5233,  -2.0042],
        [  0.9036,   7.6840,   9.9237],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.0345,   3.0175,  -1.4424],
        [  1.0093,   7.6567,   2.3526],
        [  0.5960, -11.0335,   3.1144],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.9003,  12.3550,   2.1917],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.3038,   7.8092,   0.4164],
        [  9.0424,  -7.6779,  -7.5286]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1995e-02, 2.7510e-02,  ..., 9.5459e-01, 3.4034e-01,
         3.2620e-01],
        [1.1995e-02, 1.5170e-04, 3.6018e-02,  ..., 9.5355e-01, 3.3305e-01,
         3.3797e-01],
        [2.7510e-02, 3.6018e-02, 0.0000e+00,  ..., 9.7814e-01, 3.6785e-01,
         3.1538e-01],
        ...,
        [9.5459e-01, 9.5355e-01, 9.7814e-01,  ..., 0.0000e+00, 6.9372e-01,
         9.7506e-01],
        [3.4034e-01, 3.3305e-01, 3.6785e-01,  ..., 6.9372e-01, 0.0000e+00,
         5.6743e-01],
        [3.2620e-01, 3.3797e-01, 3.1538e-01,  ..., 9.7506e-01, 5.6743e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.84450127,  5.8828986 ,  6.1561646 ,  6.07992623,  6.10774544,
        6.39187285,  8.47371868, 10.06850677,  6.3949043 , 11.38774149,
       10.23946247, 13.43127131,  6.39438044,  9.13507321, 11.28093507,
       12.62576026, 13.2778378 ,  6.01356004, 10.69963954,  8.30125201,
        5.10446653,  5.53175275, 10.7167285 ,  7.83486943, 10.27572673,
        5.05989547, 10.36844813, 13.89886469,  8.62329466, 13.86480175,
        9.48969978, 12.84056604,  9.51074763, 12.17451685, 13.96746816,
        8.95983586,  5.21280156,  6.83083863, 11.64746882,  7.06272256,
        6.61211224,  7.50035355, 10.46238054,  7.01475376, 12.44177641,
        5.0209005 ,  9.78147542,  6.33075682,  7.06272256, 13.79762415,
        7.24326992,  6.76791483,  7.16747878,  8.28968067,  4.89869205,
       14.07623266, 10.48840476,  7.44805901,  7.50035355,  4.78544875,
       13.32645552,  9.25336371,  7.50564223, 10.46238054,  9.27474386,
       11.75807252, 10.06286074, 12.10222204, 10.30972193,  8.30125201,
        8.29203356,  9.09502385, 13.32557133,  5.14999799, 11.48663767,
        5.16611782,  8.72896381, 11.48663767, 10.68416401,  5.51177501,
        8.72896381, 10.52587243,  6.29821673, 10.84493732,  9.3349858 ,
        5.20659892,  6.7166297 ,  9.09963894,  9.49511524, 13.32374056,
        7.34250948, 13.45124917,  8.29640613,  6.67108662,  7.73692436,
       10.09979747, 10.54658597, 10.52357479,  6.74686071,  7.72291946,
       11.04953767,  9.48337063, 13.69162795, 10.36844813,  7.81509631,
       11.86239961])}
{'P': tensor([[3.0699e-02, 2.0806e-01, 1.5705e-01,  ..., 8.3841e-04, 8.3433e-04,
         1.6648e-05],
        [3.9393e-03, 3.6896e-02, 1.8273e-01,  ..., 1.0726e-03, 8.4372e-04,
         2.2330e-05],
        [1.6416e-03, 2.0719e-03, 3.7322e-02,  ..., 6.4835e-04, 5.1238e-04,
         2.5353e-05],
        ...,
        [6.5021e-04, 7.7130e-04, 9.2289e-04,  ..., 9.0542e-02, 7.8072e-05,
         4.7615e-04],
        [5.5608e-02, 4.8915e-02, 5.0997e-02,  ..., 4.9783e-04, 1.1212e-01,
         1.5403e-05],
        [6.6571e-05, 6.4826e-05, 4.2165e-05,  ..., 1.4956e-04, 5.6947e-05,
         1.6685e-01]]), 'X': tensor([[  5.2939,  -2.4784,  -4.5840],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.1685,  12.1979,  -6.2473],
        [  4.1999,  -3.6543,  -4.1998],
        [ -1.5408,   7.8978,   5.3470],
        [  6.2895,   0.0405,   5.7652],
        [ 10.4758,  -5.9539,   1.3667],
        [  3.8182,  -3.9347,  -1.7961],
        [ -7.5720,   4.6310,   2.1717],
        [ -2.5683, -10.4433,  -2.9182],
        [ -5.3328,  -8.9992,   7.3440],
        [  5.2199,   4.6861,  -1.9304],
        [  5.4429,  -2.2529,   9.7306],
        [  5.7485,  -9.9811,  -3.2355],
        [  2.2373,  -4.6506,  -1.7334],
        [ -5.6174,   6.4708,  -1.7339],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.2975,   0.0947,   5.1791],
        [  5.0617,  12.7957,   8.3544],
        [  1.7659,   7.4127,   4.7871],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.1960,  -3.2964,   1.6921],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.3539,  -5.5111,   0.3373],
        [ -9.3739,   0.3982,  -4.2564],
        [ -9.4541,  -0.3327,   8.8246],
        [  4.7482, -10.3907,   5.9149],
        [ 12.4996,   2.6335,  -6.3166],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.0890,  -0.8382,   2.9166],
        [ -1.9487,   7.8590,  -3.3330],
        [ -2.0229, -10.6163,   8.6930],
        [  2.2849, -10.9655,  -7.6542],
        [  0.3038,   7.8092,   0.4164],
        [  1.5653,  13.9377,   4.1773],
        [ 12.5832,   1.7543,  -7.6134],
        [  6.2555,   2.1474,   4.1685],
        [  5.6095,  -1.9702,  -7.8233],
        [  9.4554,   9.4245,   9.8800],
        [  6.3031,   1.8477,  -7.1706],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.3231,  -4.7691,  -1.6120],
        [  1.9505,  -4.7171,   9.9868],
        [ -8.8540,  -4.3971,  -5.2144],
        [  0.5960, -11.0335,   3.1144],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.8154,   2.4611,   2.5198],
        [ -3.5520,   7.4976,   9.1522],
        [  6.3200,   0.2690,   1.5900],
        [ 11.9641,  -2.9568,   7.7211],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.0891,  -4.9558,  -2.3178],
        [  6.2975,   0.0947,   5.1791],
        [  2.5663,   7.0526,  -6.8769],
        [  6.3211,   1.7017,  -0.2164],
        [  2.0039,   7.3173,  -1.5885],
        [ -7.6525,  -6.6534,   5.1721],
        [ -5.6423,   6.4537,  -1.2091],
        [ -3.3812, -10.1235,   7.4384],
        [  5.2939,  -2.4784,  -4.5840],
        [ -8.0608,   3.9362,  -5.9567],
        [  6.2038,   2.4020,  -3.1692],
        [ -0.6323,   7.9085,  -5.9544],
        [  0.6396,  14.0772,  -6.3786],
        [ -6.8466,  -7.6488,   1.9648],
        [  1.8766,   7.3694,   1.1887],
        [ -8.7409,  -4.6837,  -6.5840],
        [  5.1472,  -2.6801,   8.1194],
        [ -2.1533, -10.5778,   1.0112],
        [ -4.7389,   6.9953,   3.7638],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.4837,   2.7510,   0.4449],
        [ 11.0373,   7.1299,  -8.5895],
        [  3.7121,   6.3117,   0.6738],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.9478,  -4.8144,   0.6371],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -6.8466,  -7.6488,   1.9648],
        [  9.2979,   9.6048,   7.3138],
        [  5.8902,  -1.3887,   4.0388],
        [  2.9410,   6.8428,   5.4458],
        [  5.1292,   4.8183,  -5.7997],
        [  5.5294,  -2.1106,   9.0626],
        [ -6.9838,  -7.4965,   0.9796],
        [ -4.8665,  -9.3204,  -5.2564],
        [  1.7228,  -4.7575,   8.3396],
        [ -1.9600,   7.8577,   6.9121],
        [ 10.2712,   8.3707,  -4.9743],
        [  6.0955,   2.8210,  -4.3321],
        [  1.6422,   7.4586,  -6.8056],
        [ -3.3812, -10.1235,   7.4384],
        [  6.3266,   0.3292,   9.3502],
        [  2.5469,  13.7192,  -9.6935],
        [ -2.6069, -10.4298,  -6.1995],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.2975,   0.0947,   5.1791],
        [  2.5829,  -4.5466,  -8.4015],
        [ -7.4733,   4.7555,   6.2490]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0084, 0.0230,  ..., 0.6607, 0.1225, 0.9622],
        [0.0084, 0.0000, 0.0265,  ..., 0.6533, 0.1306, 0.9594],
        [0.0230, 0.0265, 0.0000,  ..., 0.6767, 0.1122, 0.9851],
        ...,
        [0.6607, 0.6533, 0.6767,  ..., 0.0000, 0.7810, 0.6550],
        [0.1225, 0.1306, 0.1122,  ..., 0.7810, 0.0000, 1.0377],
        [0.9622, 0.9594, 0.9851,  ..., 0.6550, 1.0377, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.84533345,  5.90035324,  6.12886024,  6.1561646 ,  6.33075682,
        6.27905292, 13.66896739,  5.5671389 ,  8.04664886,  6.28961988,
       12.0495521 ,  5.48276038,  8.87587457, 10.75443106, 10.4606265 ,
        7.01475376,  5.89074104, 11.51811603,  5.16077764,  8.56889738,
        9.88571713,  6.29821673, 13.76047999,  7.62011734,  9.14973319,
        9.76896904,  5.38342077, 10.00794468,  9.38232121,  9.45995955,
       11.42420508, 12.77402351,  5.39450305,  6.1463896 ,  8.09704052,
       10.80728788, 11.20100795,  7.81509631, 14.02532758, 12.70489563,
        6.61386436,  5.94541803, 13.35013374,  6.56833864,  5.26067019,
        4.78002665,  5.10446653,  9.88571713, 11.04953767,  5.54776934,
        9.15253344,  8.29640613,  6.32572323, 12.32408516, 10.30972193,
       12.1460931 , 12.91505111, 12.1460931 ,  6.29821673,  7.50500064,
        6.54615887,  7.58668144, 10.14044489,  8.5724042 , 10.67322527,
        5.84533345,  8.97052115,  6.65259775,  7.9337647 , 14.09176259,
       10.26545792,  7.60463349,  9.91668182,  5.80312949, 10.794755  ,
        8.44940661,  6.25586297, 12.78327037, 13.1399181 ,  7.32233733,
        5.19577146, 12.41012046,  4.90676242, 10.14436912, 10.26545792,
       13.36800073,  6.05165248,  7.44805901,  7.03733775,  5.91854397,
       10.24557047, 10.51437947,  5.05981914,  8.09842704, 13.25017211,
        6.7166297 ,  7.63726134, 10.67322527,  6.33517349, 13.95361098,
       10.7506423 , 12.12448912,  8.91551231,  6.29821673,  5.22903117,
        8.8580671 ])}
{'P': tensor([[2.9161e-02, 2.8575e-01, 1.3039e-01,  ..., 1.3021e-05, 1.5873e-05,
         2.9984e-05],
        [3.9373e-03, 5.1645e-02, 2.7299e-01,  ..., 2.6042e-05, 2.6916e-05,
         5.1999e-05],
        [6.2854e-03, 7.0857e-03, 1.8750e-01,  ..., 1.7916e-04, 8.5681e-05,
         1.4045e-04],
        ...,
        [4.4456e-06, 4.4829e-06, 5.2397e-06,  ..., 1.9272e-01, 4.9514e-05,
         2.0615e-06],
        [5.5035e-04, 4.6383e-04, 1.8694e-04,  ..., 2.7507e-07, 4.9816e-02,
         2.1041e-06],
        [1.6796e-06, 1.8800e-06, 3.4713e-06,  ..., 7.5267e-04, 8.4795e-08,
         1.1464e-02]]), 'X': tensor([[  5.3602,  -2.3808,   7.2098],
        [  5.5424,  -2.0885,   7.2487],
        [  6.2242,  -0.3171,   7.4978],
        [  6.1973,  -0.4372,   6.5720],
        [  6.0054,  -1.0902,   8.3079],
        [  5.3804,  -2.3504,   7.2744],
        [  3.4140,  -4.1803,  -8.8224],
        [ -4.0409,   7.3169,  -3.5779],
        [  5.9843,  -9.8679,  -6.1449],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.3587,   1.1596,  -3.6629],
        [ -6.6895,   5.6051,   2.7899],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.3717,  -5.4769,  -9.2228],
        [  6.3550,   0.7129,   0.4917],
        [  2.3376,  -4.6231,   9.8436],
        [  5.3804,  -2.3504,   7.2744],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.3554,   4.8985,  -0.6665],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.0560,  -0.9421,  -5.0334],
        [  7.9816,  10.8938,  -0.5378],
        [  5.6258,  -1.9404,  -6.0715],
        [ -3.2486,   7.5925,  -1.3719],
        [  1.6308,  13.9254,  -2.0984],
        [ -3.3999, -10.1152,  -1.2088],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.1771, -10.9776,  -3.2007],
        [ -5.2589,   6.7024,  -5.5809],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.5173,   7.5091,  -8.9585],
        [ -5.5525,   6.5146,  -4.4984],
        [  0.7958,  -4.8109,  -8.2248],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -9.1459,   1.4780,  -4.2403],
        [  0.5850,  -4.7982,   9.6639],
        [ 11.1420,   6.9331,  -2.4779],
        [  5.3858,   4.4260,  -9.6769],
        [ -7.3554,   4.8985,  -0.6665],
        [  3.2809,  13.5061,  -4.0199],
        [  4.6357,   5.4415,   7.9067],
        [  0.1041, -11.0045,   2.4311],
        [  8.1043,  -8.5164,  -7.9290],
        [  1.1644, -11.0394,   7.8511],
        [  2.1379,  -4.6757,   6.4981],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.1579,  -0.5961,  -6.5595],
        [ 10.7566,   7.6212,  -2.3279],
        [  2.7976,  -4.4682,  -4.2020],
        [  4.3523, -10.5232,   8.2138],
        [  1.2273,  -4.8081,   9.8129],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.2199,   4.6861,  -1.9304],
        [  2.2373,  -4.6506,  -1.7334],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.0054,  -1.0902,   8.3079],
        [ -3.5903,   7.4847,  -8.9112],
        [ 10.0400,   8.6922,  -6.2847],
        [  5.6314,  -1.9302,  -9.6543],
        [ -7.6650,   4.5092,   0.2520],
        [  2.8474,   6.8979,   6.6457],
        [ -7.5720,   4.6310,   2.1717],
        [ -7.6650,   4.5092,   0.2520],
        [  0.5600, -11.0321,  -0.8660],
        [ -8.7735,   2.5640,  -6.3313],
        [  0.1287, -11.0065,   6.5757],
        [ -8.5506,  -5.1154,   6.3423],
        [  4.3377,   5.7570,   1.3712],
        [ 11.4495,   6.3026,   6.8257],
        [ -9.4697,  -0.6157,  -6.8472],
        [  6.7565,  11.8237,  -4.9882],
        [  4.3554,   5.7393,  -9.0766],
        [ -9.2707,   0.9661,   4.4863],
        [ 11.8951,  -3.1570,   7.0742],
        [  4.4915,   5.5990,   7.0405],
        [  1.5662,  13.9375,   4.5142],
        [ -5.1046,  -9.1610,   2.3418],
        [  4.5954,  13.0099,  -4.4439],
        [  4.6992,   5.3690,   3.0935],
        [ -6.1914,  -8.2997,   8.5380],
        [  3.7237,  -3.9966,  -5.0376],
        [  6.1929,   2.4501,  -6.9352],
        [  2.5469,  13.7192,  -9.6935],
        [  8.5716,  10.3596,   6.8818],
        [  7.1441,  11.5529,   2.3232],
        [ -7.4733,   4.7555,   6.2490],
        [ -4.4520,   7.1370,  -8.7867],
        [  3.8111, -10.6787,  -4.6539],
        [ -8.6801,   2.7809,  -0.8443],
        [  1.2273,  -4.8081,   9.8129],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.7275,  14.0667,  -0.9815],
        [  5.7485,  -9.9811,  -3.2355],
        [  3.5939,   6.4029,  -1.8357],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.4739,  -1.2497,   0.5483],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 11.0891,  -4.9558,  -2.3178],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.1204,  -5.9308,   4.0761],
        [ 12.5154,   2.5072,  -9.3274]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 6.4216e-03, 2.2524e-02,  ..., 9.3325e-01, 9.5462e-01,
         1.1076e+00],
        [6.4216e-03, 0.0000e+00, 1.7371e-02,  ..., 9.3474e-01, 9.6103e-01,
         1.1077e+00],
        [2.2524e-02, 1.7371e-02, 0.0000e+00,  ..., 9.4921e-01, 9.7325e-01,
         1.1199e+00],
        ...,
        [9.3325e-01, 9.3474e-01, 9.4921e-01,  ..., 0.0000e+00, 1.2133e+00,
         2.7393e-01],
        [9.5462e-01, 9.6103e-01, 9.7325e-01,  ..., 1.2133e+00, 3.0339e-04,
         1.4861e+00],
        [1.1076e+00, 1.1077e+00, 1.1199e+00,  ..., 2.7393e-01, 1.4861e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.86519525,  5.92281142,  6.2322769 ,  6.21275016,  6.10360073,
        5.87132718,  5.39723241,  8.35856487, 11.54071859, 13.01728313,
        6.46356806,  8.7273569 ,  9.12220271, 10.00410625,  6.3949043 ,
        5.1805345 ,  5.87132718, 12.25806559, 10.7167285 , 10.06437598,
        8.83725081, 11.28895489, 10.33958639,  9.79814997,  6.12886024,
       13.50484539,  5.9510489 ,  8.25828409, 14.02058621, 10.67132524,
       12.35625694, 11.1913598 ,  8.51928531, 10.48721382,  8.29203356,
        8.55982105,  4.8763135 , 10.06437598,  9.26455795,  4.83371341,
       13.12299145,  6.97106912,  8.83725081, 13.89886469,  7.14837514,
       11.0050379 , 11.75618513, 11.10066126,  5.14124612, 13.1399181 ,
        6.18668463, 13.18278606,  5.27178363, 11.38774149,  4.96231532,
       10.31613269, 12.67680439,  7.01475376,  5.16077764,  5.18277858,
        6.10360073,  8.30125201, 13.27994304,  5.95297459,  8.89301516,
        7.46248947,  8.87587457,  8.89301516, 11.04629471,  9.14044984,
       11.00726483,  9.96391045,  7.20826219, 13.06957425,  9.48969978,
       13.61801526,  7.20482224,  9.32094055, 12.30695098,  7.17789666,
       14.02526505, 10.48721382, 13.79762415,  7.13501264, 10.35464916,
        5.46244988,  6.65991877, 13.95361098, 13.44593519, 13.58333857,
        8.8580671 ,  8.41171369, 11.33837396,  9.11473391,  4.96231532,
       10.00410625, 14.08549802, 11.51811603,  7.34250948, 11.9120321 ,
        9.55593374, 10.27716434, 12.1460931 , 14.09176259, 10.05558893,
       12.76408562])}
{'P': tensor([[3.1399e-02, 3.1526e-01, 1.4444e-01,  ..., 8.6875e-06, 4.3339e-04,
         1.2423e-05],
        [3.2744e-03, 4.2636e-02, 2.2537e-01,  ..., 1.0640e-05, 5.2839e-04,
         1.8109e-05],
        [2.4417e-03, 2.7312e-03, 7.2273e-02,  ..., 8.2378e-06, 4.3756e-04,
         4.3886e-05],
        ...,
        [6.0057e-05, 6.3309e-05, 6.3313e-05,  ..., 4.8783e-02, 3.3128e-04,
         7.4242e-06],
        [4.9455e-03, 5.0397e-03, 3.9837e-03,  ..., 6.9523e-04, 5.5529e-02,
         7.1608e-05],
        [9.4190e-06, 9.0388e-06, 6.7789e-06,  ..., 1.1807e-05, 1.2088e-04,
         5.4342e-02]]), 'X': tensor([[  5.3804,  -2.3504,   7.2744],
        [  5.5424,  -2.0885,   7.2487],
        [  6.2242,  -0.3171,   7.4978],
        [  6.1973,  -0.4372,   6.5720],
        [  6.0054,  -1.0902,   8.3079],
        [  6.3549,   0.7096,   6.9198],
        [  7.1441,  11.5529,   2.3232],
        [ -2.6735, -10.4062,  -1.5911],
        [  2.0449,  13.8402,  -7.1642],
        [  0.6470, -11.0352,  -1.1931],
        [ 12.4837,   2.7510,   0.4449],
        [  2.6078, -10.9229,   6.2869],
        [ -0.2586,   7.8825,  -9.9840],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.9587,  12.3216,   9.1695],
        [ -1.6906,   7.8860,   7.7142],
        [ -1.8788, -10.6567,   3.7068],
        [  5.0506,   4.9277,  -9.5094],
        [ -5.3382,   6.6533,   7.8054],
        [  6.2022,   2.4092,   3.0349],
        [ -9.3616,  -2.4749,   0.1793],
        [  2.0899,  -4.6870,  -7.4260],
        [ -6.3489,   5.9117,   6.5161],
        [ 11.6801,  -3.7196,  -0.9181],
        [  4.2275,  -3.6320,  -0.2714],
        [ -6.4464,   5.8272,   7.5038],
        [  9.0129,  -7.7072,   9.1400],
        [  6.3583,   1.1729,   8.7835],
        [  5.7845,   3.6652,   0.9175],
        [ -0.3432,   7.8900,   7.1805],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.6451,  -4.5250,  -0.5565],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.4697,  -4.5836,   1.4593],
        [ -7.3333,   4.9246,  -7.9212],
        [  9.4925,   9.3811,   8.9842],
        [  4.6063,  13.0051,  -3.1456],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -8.8745,   2.3097,   3.5600],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.1271, -10.8343,   2.6442],
        [ -7.6734,   4.4980,  -5.4248],
        [ 10.9875,   7.2207,  -8.8549],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.3122,  -9.6994,  -6.3282],
        [  0.9478,  -4.8144,   0.6371],
        [  8.6271,  -8.0715,  -7.4813],
        [  2.7685, -10.8982,  -6.8928],
        [  0.5175,  14.0910,   4.6540],
        [  2.0449,  13.8402,  -7.1642],
        [ -5.3382,   6.6533,   7.8054],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.3413,   0.4883,  -6.8416],
        [ -3.9456,  -9.8542,   0.7466],
        [ -5.1974,   6.7396,   3.3484],
        [  4.3554,   5.7393,  -9.0766],
        [  1.4489,   7.5256,   4.4529],
        [  8.2422,  -8.4041,  -1.4962],
        [ -4.3963,   7.1629,   4.5263],
        [  2.1824,  -4.6647,   1.1985],
        [ -0.5385,   7.9037,   6.3830],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -3.0644,   7.6439,  -1.4322],
        [  1.8766,   7.3694,   1.1887],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.5620,   7.0549,   8.4825],
        [ -8.5900,   2.9767,   7.4811],
        [  5.2188,  -2.5839,   2.1833],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.2096,   5.8815,   1.9230],
        [  6.2975,   0.0947,   5.1791],
        [  5.2608,  -2.5256,  -9.1446],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.5974,   7.0363,  -7.7810],
        [ -5.6423,   6.4537,  -1.2091],
        [ -0.5739,   7.9056,   0.6382],
        [  8.2422,  -8.4041,  -1.4962],
        [ -7.5720,   4.6310,   2.1717],
        [  6.3360,   0.4258,  -9.2509],
        [  6.9653,  -9.3225,  -3.4727],
        [  4.1348,   5.9514,  -0.8856],
        [ -2.1382,   7.8339,   7.3524],
        [  4.9957,  -2.8705,   6.5978],
        [  5.6095,  -1.9702,  -7.8233],
        [  7.9141,  -8.6656,   9.6948],
        [  5.7845,   3.6652,   0.9175],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.3236,   0.3011,  -4.6668],
        [ -6.3306,   5.9272,   5.6935],
        [  6.3542,   0.6934,  -3.3840],
        [  4.5057,  -3.3894,  -0.8160],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.8890,  -4.4316,   1.1600],
        [  5.7802,  -1.6363,  -6.2782],
        [ 10.4758,  -5.9539,   1.3667],
        [  1.1350,  14.0104,   0.6321],
        [ 12.4877,  -0.7222,   8.8347],
        [  1.1274,  14.0116,   8.1753],
        [ -9.0574,   1.7801,   8.8986],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.7558,  -4.4844,   0.0636],
        [ -5.9987,   6.1949,  -6.2114]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0052, 0.0201,  ..., 1.0600, 0.4979, 1.0504],
        [0.0052, 0.0000, 0.0174,  ..., 1.0602, 0.4979, 1.0524],
        [0.0201, 0.0174, 0.0000,  ..., 1.0763, 0.5139, 1.0698],
        ...,
        [1.0600, 1.0602, 1.0763,  ..., 0.0000, 0.5627, 0.3860],
        [0.4979, 0.4979, 0.5139,  ..., 0.5627, 0.0000, 0.6231],
        [1.0504, 1.0524, 1.0698,  ..., 0.3860, 0.6231, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.87132718,  5.92281142,  6.2322769 ,  6.21275016,  6.10360073,
        6.39438044, 13.58333857, 10.74409834, 13.99048094, 11.05413545,
       12.78327037, 11.22993218,  7.88677538,  9.11473391, 13.68672855,
        8.06517143, 10.82106505,  7.05627461,  8.53014915,  6.65368551,
        9.68323686,  5.13182506,  8.67502348, 12.25806559,  5.57341623,
        8.68979052, 11.85890664,  6.46560181,  6.84796202,  7.89745107,
       10.46238054,  5.24137909, 12.22276278,  5.20659892,  8.83341608,
       13.34587027, 13.79676654, 13.26590071,  9.97964013,  9.17016857,
        9.66977802, 12.91143491, 11.27656488,  8.89458273, 13.14777433,
       10.06850677, 11.57249984,  4.90676242, 11.81423161, 11.24434374,
       14.10045797, 13.99048094,  8.53014915, 13.79762415, 12.32408516,
        6.36004393, 10.61472636,  8.51088913,  7.20482224,  7.66377554,
       11.77124373,  8.40445872,  5.14999799,  7.92200342, 10.54658597,
        8.23525433,  7.60463349, 10.23946247,  9.49511524,  7.50564223,
        9.09119351,  5.82345867,  9.60412735,  7.23276582,  6.29821673,
        5.83560009, 10.31584102,  7.50035355,  8.5724042 ,  7.92644746,
       11.77124373,  8.87587457,  6.35029274, 11.63724792,  7.24677664,
        8.12043983,  5.76165934,  5.94541803, 11.73567759,  6.84796202,
        5.63824971,  6.33075682,  8.67227386,  6.39187285,  5.63824971,
       12.18045978,  5.29009836,  6.00732193, 12.0495521 , 14.05633441,
       12.5085995 , 14.05687815,  9.23071698, 11.67875855,  5.26342203,
        8.62329466])}
{'P': tensor([[2.6579e-02, 1.5066e-01, 1.2124e-01,  ..., 1.2656e-03, 9.1765e-02,
         2.2612e-05],
        [3.6105e-03, 3.2949e-02, 1.4544e-01,  ..., 1.2140e-03, 1.2221e-01,
         3.0166e-05],
        [9.5217e-04, 1.2837e-03, 2.5968e-02,  ..., 2.4697e-04, 2.5764e-01,
         2.7280e-05],
        ...,
        [3.6928e-03, 5.5701e-03, 1.1503e-02,  ..., 7.1214e-02, 1.4175e-02,
         4.2708e-06],
        [1.2456e-03, 1.5480e-03, 2.5481e-03,  ..., 2.5794e-04, 4.6083e-02,
         5.4938e-05],
        [4.8091e-05, 4.2039e-05, 2.2073e-05,  ..., 6.0905e-06, 1.7803e-05,
         1.9909e-02]]), 'X': tensor([[ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 2.2031e-02, 4.5201e-02,  ..., 2.8647e-01, 5.5504e-02,
         9.2824e-01],
        [2.2031e-02, 0.0000e+00, 6.3838e-02,  ..., 2.6616e-01, 7.4218e-02,
         9.2545e-01],
        [4.5201e-02, 6.3838e-02, 0.0000e+00,  ..., 3.2994e-01, 1.0400e-02,
         9.6178e-01],
        ...,
        [2.8647e-01, 2.6616e-01, 3.2994e-01,  ..., 0.0000e+00, 3.4028e-01,
         8.2389e-01],
        [5.5504e-02, 7.4218e-02, 1.0400e-02,  ..., 3.4028e-01, 0.0000e+00,
         9.6852e-01],
        [9.2824e-01, 9.2545e-01, 9.6178e-01,  ..., 8.2389e-01, 9.6852e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.87305005,  5.94909709,  6.2134961 ,  6.01356004,  6.32572323,
        6.02588517,  6.10774544, 11.80092578,  9.25336371, 12.5769659 ,
        9.77588801, 13.744387  , 10.95593729,  9.41970031,  6.652742  ,
       12.4980845 ,  5.19577146, 11.48434767, 10.68031351, 13.46553623,
        5.1224079 , 11.97238624,  6.50409372, 13.14777433, 11.76830981,
       10.39942497,  7.80625981,  9.5014146 ,  4.74770899,  6.1561646 ,
        7.56617037, 10.98076873,  6.81084724,  7.60463349,  7.02253235,
       11.28215429, 13.54774355,  4.75303127,  8.34949383,  6.90115901,
        5.91677188,  5.79562691,  5.1805345 ,  9.75401449, 12.50996828,
        8.35856487,  6.67108662, 11.59512194, 12.58006038,  6.4020404 ,
       10.06437598,  5.16077764,  7.98469166, 10.51437947,  9.7751189 ,
        7.85919687,  6.67121748, 10.1979669 ,  7.50917551, 14.03505624,
       13.58101794,  7.20482224, 12.27993098,  9.60080639,  6.46560181,
        5.79391825,  6.31576264,  6.76791483,  9.41970031,  8.62162372,
       11.97957425,  8.99276564, 11.51811603,  5.41348121,  6.67108662,
       10.86693708,  8.11227668, 13.20597932,  6.02789634, 10.6910768 ,
       11.81423161,  9.41970031, 10.06407839, 14.02058621,  5.39723241,
        5.0209005 ,  4.9160962 ,  6.4020404 , 11.72749923,  8.29203356,
        5.2287058 ,  8.8353674 ,  8.87587457,  9.62238828, 12.27991991,
       11.4424816 , 13.26590071,  5.53175275, 11.63502679, 10.67132524,
        5.5366698 ,  9.7751189 , 10.06437598, 12.1460931 ,  6.32572323,
        9.37296102])}
{'P': tensor([[3.2381e-02, 1.4723e-01, 1.3587e-01,  ..., 1.2075e-05, 3.7146e-04,
         1.6216e-03],
        [2.2128e-03, 4.6037e-02, 1.1095e-02,  ..., 7.7153e-06, 9.2915e-04,
         4.5131e-03],
        [2.8128e-03, 4.3123e-02, 3.9743e-02,  ..., 1.1274e-05, 7.5421e-04,
         2.2265e-03],
        ...,
        [2.3796e-04, 1.3776e-04, 2.7529e-04,  ..., 3.0899e-02, 2.3474e-06,
         3.3501e-06],
        [8.9067e-05, 7.5215e-05, 8.2832e-05,  ..., 1.0258e-05, 6.9260e-02,
         5.2282e-06],
        [4.3529e-05, 1.0081e-04, 6.6687e-05,  ..., 1.0811e-07, 1.4705e-03,
         7.1125e-02]]), 'X': tensor([[  5.4179,  -2.2925,  -3.4641],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.3542,   0.6934,  -3.3840],
        [  6.3236,   0.3011,  -4.6668],
        [  0.7685,  -4.8098,  -2.9310],
        [  6.1984,  -0.4327,   1.5438],
        [ 10.1798,   8.5002,  -5.8893],
        [ -9.4671,  -0.5553,   0.1117],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.3490,  -4.6199,  -4.2489],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -0.7599, -10.8995,  -6.0366],
        [  4.5430,   5.5438,  -3.8669],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.5083,  -4.1177,   0.2389],
        [  1.9505,  -4.7171,   9.9868],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.9178,   2.1936,   7.5388],
        [ -7.0948,   5.1940,   9.8171],
        [  1.4150,  13.9647,  -4.0915],
        [  2.2076,  13.8032,  -1.5099],
        [  6.9314,  -9.3435,  -1.2729],
        [  2.1504,  -4.6726,   8.7908],
        [  4.2339,  -3.6268,   6.3125],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.1685,  12.1979,  -6.2473],
        [  4.7544,  -3.1425,  -1.5315],
        [ -7.8280,   4.2848,  -7.1955],
        [  6.0305,   3.0296,   0.1375],
        [ -0.7599, -10.8995,  -6.0366],
        [  8.2154,  -8.4262,   8.0171],
        [  1.5383, -11.0274,  -4.9666],
        [  1.0093,   7.6567,   2.3526],
        [  4.6992,   5.3690,   3.0935],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.0729,   7.2878,   7.0880],
        [ -9.3087,   0.7792,  -7.5587],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4756,  -0.8166,  -9.6021],
        [  7.9403,  -8.6455,  -8.7433],
        [  4.3377,   5.7570,   1.3712],
        [ -3.5029,   7.5138,   8.1853],
        [  9.9756,  -6.6330,  -3.3429],
        [  1.7903,  -4.7467,   0.1168],
        [  1.9204,   7.3518,   8.4781],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.2373,  -4.6506,  -1.7334],
        [ 12.1379,   4.4125,  -3.8833],
        [ -7.0948,   5.1940,   9.8171],
        [  0.1931,  -4.7491,   7.4385],
        [ 11.9936,  -2.8677,   6.0614],
        [  2.4639, -10.9431,  -7.3718],
        [  4.0587, -10.6112,   6.2273],
        [  2.4697,  -4.5836,   1.4593],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.0345,   3.0175,  -1.4424],
        [ -9.4541,  -0.3327,   8.8246],
        [  1.3670,  -4.7990,   3.2603],
        [  2.1477,   7.2550,   8.9688],
        [  5.8979,   3.3958,  -3.8713],
        [  1.8721,  -4.7322,   0.5524],
        [ -4.0868,   7.2981,   8.1874],
        [ 11.7162,   5.6728,  -8.9372],
        [ -3.1204, -10.2345,   9.2288],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.7121,   6.3117,   0.6738],
        [  9.2811,  -7.4336,   5.1771],
        [ 11.2517,  -4.6495,   2.9800],
        [  7.9816,  10.8938,  -0.5378],
        [  4.7482, -10.3907,   5.9149],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.4426, -10.1181,  -5.1268],
        [  5.2911,  -2.4824,  -3.6877],
        [ -6.7643,  -7.7372,  -9.6663],
        [  4.9531,   5.0576,   5.6794],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.4716,  -0.6673,   7.6068],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.1178,   1.5784,   2.1410],
        [ -4.0868,   7.2981,   8.1874],
        [  2.0449,  13.8402,  -7.1642],
        [ -6.2668,   5.9808,   2.5614],
        [ -0.0282,   7.8575,  -1.1746],
        [  9.1409,  -7.5789,   1.5814],
        [  2.0039,   7.3173,  -1.5885],
        [ -1.9600,   7.8577,   6.9121],
        [  5.4429,  -2.2529,   9.7306],
        [ -5.9938,  -8.4745,   3.9735],
        [  3.8111, -10.6787,  -4.6539],
        [  3.1064,  13.5607,   6.7605],
        [  8.5716,  10.3596,   6.8818],
        [ -9.1907,  -3.3180,  -5.5943],
        [  9.9058,  -6.7203,   5.6928],
        [  7.5930,  -8.9037,   7.2323],
        [  6.2036,   2.4030,  -1.2500],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.8890,  -4.4316,   1.1600],
        [  5.6022,  -1.9833,   8.3081],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -4.3963,   7.1629,   4.5263],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.1440, -10.9811,   8.5281],
        [ -3.0644,   7.6439,  -1.4322],
        [ 10.9394,   7.3069,  -8.9365]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 3.6018e-02, 4.6723e-02,  ..., 1.0326e+00, 3.4533e-01,
         4.8751e-01],
        [3.6018e-02, 0.0000e+00, 7.8923e-02,  ..., 1.0682e+00, 3.7247e-01,
         4.5264e-01],
        [4.6723e-02, 7.8923e-02, 0.0000e+00,  ..., 9.9909e-01, 3.3896e-01,
         5.1926e-01],
        ...,
        [1.0326e+00, 1.0682e+00, 9.9909e-01,  ..., 0.0000e+00, 7.7528e-01,
         1.5184e+00],
        [3.4533e-01, 3.7247e-01, 3.3896e-01,  ..., 7.7528e-01, 0.0000e+00,
         8.0690e-01],
        [4.8751e-01, 4.5264e-01, 5.1926e-01,  ..., 1.5184e+00, 8.0690e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.8828986 ,  6.1561646 ,  6.07992623,  6.10774544,  6.39187285,
        6.33075682,  4.87083348,  6.2134961 , 13.26209545,  9.48337063,
        8.56889738,  5.18277858,  9.66977802, 10.92597196,  7.16747878,
       11.63502679, 12.22276278,  5.10446653, 13.13205357,  9.18358562,
        8.79284466, 14.03618649, 13.97857591, 11.63383221,  5.14371005,
        5.5748624 ,  5.8828986 , 13.66896739,  5.69913817,  8.92395381,
        6.74872981, 10.92597196, 11.76830981, 11.1341798 ,  7.72291946,
        7.13501264, 10.67132524,  7.57685908,  9.34127013,  5.57341623,
        9.51074763, 11.73848065,  7.20826219,  8.29022197, 11.97957425,
        5.07306274,  7.5984809 ,  5.79391825,  5.16077764, 12.91505111,
        8.79284466,  4.75303127, 12.33167382, 11.21703654, 11.36089355,
        5.20659892,  9.11473391,  6.74686071,  9.45995955,  4.98989906,
        7.56617037,  6.80559669,  5.08910491,  8.36445758, 13.01728313,
       10.69963954,  5.18047638,  7.32233733, 11.89105259, 12.17451685,
       13.50484539, 11.42420508, 11.80092578, 11.4890861 ,  5.84450127,
       10.27716434,  7.07902036, 12.35625694,  9.49511524, 13.2430377 ,
        9.25336371,  8.36445758, 13.99048094,  8.66273171,  7.8575648 ,
       11.87412439,  7.58668144,  8.09842704,  5.89074104, 10.37998078,
       11.33837396, 13.91198191, 13.44593519,  9.7712321 , 11.9702762 ,
       11.70168755,  6.652742  ,  6.1561646 ,  5.29009836,  5.94293185,
        9.75075273,  8.40445872,  4.8763135 , 11.18839169,  8.23525433,
       13.15525803])}
{'P': tensor([[3.5130e-02, 1.9641e-01, 1.7423e-01,  ..., 1.5662e-03, 3.3047e-04,
         2.0528e-05],
        [3.0860e-03, 4.6829e-02, 3.7754e-03,  ..., 3.0006e-03, 3.7851e-04,
         2.5107e-05],
        [2.8199e-03, 3.3828e-01, 3.3124e-02,  ..., 1.9141e-03, 2.9388e-04,
         1.8058e-05],
        ...,
        [2.9923e-06, 4.1831e-06, 3.8252e-06,  ..., 7.5620e-03, 3.8948e-07,
         4.7170e-07],
        [2.4385e-03, 3.5124e-03, 3.5401e-03,  ..., 1.3413e-04, 1.0929e-01,
         4.0836e-05],
        [1.0306e-03, 6.2386e-04, 6.9570e-04,  ..., 2.4981e-06, 6.6037e-04,
         8.4941e-02]]), 'X': tensor([[  5.4336,  -2.2676,   5.6933],
        [  5.9193,  -1.3176,   5.1739],
        [  5.8323,  -1.5228,   5.1132],
        [  6.2895,   0.0405,   5.7652],
        [  6.2883,   0.0325,   5.2692],
        [  6.1973,  -0.4372,   6.5720],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.1795,  -3.3623,   8.5259],
        [ -4.5778,  -9.5013,  -5.0667],
        [  6.0467,  -9.8369,  -6.5816],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.9659,  -4.8144,  -0.9594],
        [ -9.3087,   0.7792,  -7.5587],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -0.1963,   7.8765,  -1.1371],
        [ -7.8849,  -6.3115,  -4.7479],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.8766,   7.3694,   1.1887],
        [  1.9505,  -4.7171,   9.9868],
        [ -8.5607,  -5.0938,  -5.4531],
        [  3.9591, -10.6391,  -6.2557],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.2038,   2.4020,  -3.1692],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.3303,   1.6145,  -8.4083],
        [  6.9433,  -9.3362,  -0.3421],
        [  0.7958,  -4.8109,  -8.2248],
        [ 10.1983,  -6.3430,  -0.3200],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.6426,  -4.9136,   9.4941],
        [  7.5784,  -8.9141,   5.8410],
        [ -2.0840,   7.8415,  -7.8657],
        [ -3.5173, -10.0623,   4.8280],
        [  0.5250, -11.0306,   5.5358],
        [ -1.1885, -10.8211,   0.0865],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.0876,   6.7527,  -5.3540],
        [  2.9470,  13.6085,   2.9809],
        [  5.0506,   4.9277,  -9.5094],
        [  6.3122,  -9.6994,  -6.3282],
        [  7.0515,  11.6194,   8.4289],
        [  6.3087,   1.8053,  -1.9481],
        [  2.5663,   7.0526,  -6.8769],
        [ -8.3535,  -5.5118,   9.8399],
        [  9.4516,  -7.2501,  -1.8868],
        [  5.3858,   4.4260,  -9.6769],
        [ 12.1504,   4.3674,  -7.2998],
        [ -3.7964,  -9.9294,   7.7147],
        [ -8.7735,   2.5640,  -6.3313],
        [  4.1348,   5.9514,  -0.8856],
        [  2.0547,   7.2957,  -1.9343],
        [  7.5930,  -8.9037,   7.2323],
        [  1.8320,   7.3871,   5.1452],
        [ -9.3419,   0.5972,  -1.2841],
        [  4.9957,  -2.8705,   6.5978],
        [  6.1752,   2.5242,   3.5874],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.5456,   6.0861,   8.1101],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.3670,  -4.7990,   3.2603],
        [ 11.5456,   6.0861,   8.1101],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -6.6998,   5.5954,   0.0823],
        [ -0.7599, -10.8995,  -6.0366],
        [ 10.0995,  -6.4739,   2.0974],
        [ -0.8455, -10.8853,  -6.7400],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -8.1531,  -5.8748,   2.6934],
        [  0.1677,  -4.7447,  -6.1529],
        [ -0.0254, -10.9932,   3.2200],
        [  6.3549,   0.7096,   6.9198],
        [ -0.2586,   7.8825,  -9.9840],
        [  0.9392,  -4.8143,  -3.4649],
        [  4.7482, -10.3907,   5.9149],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.3505,   0.6235,  -8.5849],
        [  3.2701,   6.6338,   9.9209],
        [  3.7992,  -3.9474,  -8.9019],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -9.3087,   0.7792,  -7.5587],
        [ -8.1905,   3.7253,   4.1175],
        [  2.0899,  -4.6870,  -7.4260],
        [  9.3424,  -7.3685,  -3.6807],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.4378,  -1.8606,  -2.5668],
        [  4.0642,  -3.7599,  -3.4565],
        [  7.5784,  -8.9141,   5.8410],
        [  5.0617,  12.7957,   8.3544],
        [  2.5318,  13.7231,   6.3692],
        [  7.0515,  11.6194,   8.4289],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.1771, -10.9776,  -3.2007],
        [  6.3505,   0.6235,  -8.5849],
        [  6.2026,  12.1774,  -1.5887],
        [ 11.5083,  -4.1177,   0.2389],
        [ 10.0400,   8.6922,  -6.2847],
        [  9.0129,  -7.7072,   9.1400],
        [ -4.2271,   7.2387,  -1.8316],
        [  8.3718,  -8.2953,   8.4666],
        [  9.3560,   9.5391,   9.4943],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.0965,  -5.9712,   4.3466]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0268, 0.0305,  ..., 0.5379, 0.6676, 0.9386],
        [0.0268, 0.0000, 0.0058,  ..., 0.5510, 0.6432, 0.9462],
        [0.0305, 0.0058, 0.0000,  ..., 0.5568, 0.6382, 0.9428],
        ...,
        [0.5379, 0.5510, 0.5568,  ..., 0.0000, 1.1402, 1.3772],
        [0.6676, 0.6432, 0.6382,  ..., 1.1402, 0.0000, 1.0747],
        [0.9386, 0.9462, 0.9428,  ..., 1.3772, 1.0747, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.88782913,  6.06416957,  6.02779201,  6.28961988,  6.28835101,
        6.21275016, 12.26997165,  9.77588801, 10.54658597, 11.54672755,
        4.91038884,  4.91038884,  9.34127013, 12.10942412,  7.8788965 ,
       10.09979747,  5.9510489 ,  7.60463349,  5.10446653,  9.96152788,
       11.35182589,  9.62958973,  9.75200848,  6.65259775,  9.96152788,
        8.12580976,  6.53290663, 11.63502679,  4.8763135 , 12.00995898,
        4.85603886, 13.01187484,  9.61942445,  9.94173708, 11.70016217,
        8.1137452 , 10.65929968, 11.04313641, 10.88618114,  5.19065439,
        7.42512621, 13.92390689,  7.05627461, 11.57249984, 13.59170809,
        6.5618968 ,  7.50500064, 10.00802627, 11.9120321 ,  6.97106912,
       12.91143491, 10.63038565,  9.14044984,  7.24677664,  7.57946459,
       11.70168755,  7.61088125,  9.36093562,  5.76165934,  6.67121748,
        5.22903117, 13.05148826,  6.27905292,  4.98989906, 13.05148826,
       10.09979747,  8.72896381, 10.92597196, 11.99633819, 10.91805683,
       10.27716434, 10.04916678,  4.74770899, 10.99326577,  6.39438044,
        7.88677538,  4.90505566, 11.42420508, 12.12448912,  6.38105015,
        7.39599986,  5.47866221, 12.32545637,  9.34127013,  8.99791316,
        5.13182506, 11.89855267,  5.01114719,  9.61942445,  5.5366698 ,
       11.70016217, 13.76047999, 13.95472331, 13.59170809, 12.9089559 ,
       11.1913598 ,  6.38105015, 13.6660671 , 12.22276278, 13.27994304,
       11.85890664,  8.38252937, 11.785566  , 13.3614598 ,  5.5671389 ,
       10.06023665])}
{'P': tensor([[2.6846e-02, 2.1696e-01, 1.0618e-01,  ..., 1.0315e-04, 3.7770e-05,
         2.4619e-03],
        [3.0973e-03, 4.5856e-02, 2.0420e-01,  ..., 2.2873e-04, 5.8547e-05,
         3.7847e-03],
        [3.2823e-03, 5.1604e-03, 6.4365e-02,  ..., 5.5396e-04, 8.4742e-05,
         5.1032e-03],
        ...,
        [3.3671e-05, 3.5628e-05, 4.4628e-05,  ..., 8.2851e-02, 1.7183e-04,
         1.1476e-04],
        [3.1903e-04, 3.4930e-04, 4.7819e-04,  ..., 6.3878e-04, 7.8865e-02,
         6.9756e-03],
        [2.7452e-03, 3.8985e-03, 7.9062e-03,  ..., 4.1409e-03, 7.0848e-04,
         9.0124e-02]]), 'X': tensor([[ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0086, 0.0147,  ..., 0.6787, 0.8065, 0.3340],
        [0.0086, 0.0000, 0.0082,  ..., 0.6826, 0.8080, 0.3350],
        [0.0147, 0.0082, 0.0000,  ..., 0.6784, 0.8020, 0.3286],
        ...,
        [0.6787, 0.6826, 0.6784,  ..., 0.0000, 0.2486, 0.3851],
        [0.8065, 0.8080, 0.8020,  ..., 0.2486, 0.0000, 0.4741],
        [0.3340, 0.3350, 0.3286,  ..., 0.3851, 0.4741, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.89074104,  6.00663314,  6.16512856,  6.05113217,  5.91222346,
        6.33517349,  6.49538841, 10.26327895, 11.4890861 , 13.58101794,
       13.48809595,  6.2322769 ,  7.15100303,  8.29022197, 13.35013374,
       11.56146873,  7.13501264,  5.8828986 , 13.97857591,  9.41970031,
        9.36093562, 13.66051884, 13.06957425, 12.27835996, 13.32557133,
       10.74409834, 10.48840476, 14.02058621,  8.04483495,  9.77777346,
       12.22276278,  9.36378438, 12.12641969, 12.87752052,  8.91551231,
        6.21275016,  8.31625171,  8.57560275,  5.5205189 ,  8.89301516,
       11.83509372,  8.09899874,  5.14371005, 11.83075005, 10.24557047,
        5.80312949, 11.28895489,  6.29034958, 12.53285633, 12.45418236,
       13.68672855,  7.83486943,  5.20659892,  5.08910491, 10.27716434,
        5.91854397, 10.78533838,  9.38232121,  4.96231532, 13.59170809,
        9.3109884 , 12.25806559, 10.14044489,  6.06416957,  7.16058587,
       11.11969815,  5.18047638,  6.05165248, 13.96905756,  9.50654967,
        6.05113217, 12.18998195,  8.55982105,  8.12580976, 13.99877413,
       13.3426541 , 10.46739073, 14.10045797,  5.48276038,  8.29640613,
       11.63383221, 11.77124373, 13.6922425 ,  5.5671389 ,  5.92952538,
        8.77913428,  7.50917551,  5.39450305, 13.3607379 , 11.21703654,
        5.62005629, 13.01728313, 13.52430176, 14.07623266,  9.50654967,
        6.4020404 ,  5.96556248, 13.32374056,  6.90115901,  8.62162372,
       10.7506423 ,  8.67227386, 12.38881635,  7.8788965 ,  5.27178363,
        6.1463896 ])}
{'P': tensor([[3.3264e-02, 2.0067e-01, 1.6474e-01,  ..., 6.6868e-04, 2.8772e-04,
         5.8575e-05],
        [1.7686e-03, 4.0064e-02, 1.6898e-02,  ..., 1.5052e-03, 7.0640e-04,
         3.3978e-05],
        [1.8147e-03, 3.7104e-02, 3.2689e-02,  ..., 8.7940e-04, 4.1656e-04,
         2.7876e-05],
        ...,
        [1.7479e-04, 2.0395e-04, 1.7396e-04,  ..., 1.6002e-01, 8.0851e-04,
         2.6335e-05],
        [2.1920e-06, 3.3989e-06, 2.8409e-06,  ..., 7.7310e-03, 1.5923e-02,
         1.2020e-07],
        [5.2151e-03, 3.4808e-03, 3.0062e-03,  ..., 2.7378e-05, 1.3659e-05,
         3.3358e-02]]), 'X': tensor([[ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0164, 0.0265,  ..., 0.5075, 0.8008, 0.5796],
        [0.0164, 0.0000, 0.0368,  ..., 0.5004, 0.7850, 0.5840],
        [0.0265, 0.0368, 0.0000,  ..., 0.5340, 0.8182, 0.5997],
        ...,
        [0.5075, 0.5004, 0.5340,  ..., 0.0000, 0.6092, 0.3920],
        [0.8008, 0.7850, 0.8182,  ..., 0.6092, 0.0000, 0.9978],
        [0.5796, 0.5840, 0.5997,  ..., 0.3920, 0.9978, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.90035324,  6.1561646 ,  6.12886024,  6.33075682,  6.27905292,
        6.39187285,  9.68323686, 12.05395056, 13.31764181, 13.94008948,
        8.06990483,  8.31625171, 10.7506423 ,  5.56809335, 12.0495521 ,
        8.8353674 ,  5.76165934,  7.96181288,  5.13125513,  7.46248947,
       12.24467041, 13.95361098,  6.48486171,  6.36004393,  7.48311002,
       10.65929968, 11.67875855, 12.53285633, 13.09737053,  9.36378438,
       11.62158554,  9.19837792,  9.94285511,  5.64464807,  8.30125201,
       13.00413829, 12.87752052, 13.92390689, 13.18278606,  6.77777606,
        5.92281142,  5.95297459, 10.48721382, 11.22596994,  5.04028538,
        9.25336371,  5.91222346, 11.63502679, 11.00726483, 13.14777433,
       14.03505624, 12.30858677,  7.92200342,  9.99145096,  7.49076899,
        8.04718433,  5.80860165, 13.79676654,  7.27759233,  5.29755487,
       10.78533838,  6.39438044, 12.58006038,  7.44805901,  7.78980299,
        6.33517349, 12.11735334,  5.08910491, 10.5575601 ,  9.34061938,
       13.3426541 ,  9.14044984, 13.25017211,  6.05113217, 10.63038565,
        6.83083863,  9.7712321 , 13.04051744,  4.76969364,  6.23262289,
        7.72291946,  9.5642993 , 10.412299  ,  6.34059021, 11.04313641,
       13.60275211,  9.6282763 , 10.69963954,  4.74770899,  8.29022197,
       11.11969815, 13.34587027,  9.96391045,  9.27474386, 12.59969936,
        9.45995955,  7.46053281, 11.83075005, 13.52430176, 11.86239961,
        6.80559669,  5.69913817,  5.14607014,  7.72291946, 13.62896593,
       11.28895489])}
{'P': tensor([[3.3563e-02, 3.1176e-01, 1.5549e-01,  ..., 2.9195e-05, 4.7480e-04,
         4.4179e-05],
        [3.7982e-03, 5.7015e-02, 2.5390e-01,  ..., 5.4320e-05, 1.0502e-03,
         8.2879e-05],
        [4.3208e-03, 6.7363e-03, 8.4022e-02,  ..., 1.1044e-04, 2.8512e-03,
         1.8858e-04],
        ...,
        [3.0883e-05, 2.6115e-05, 2.2849e-05,  ..., 3.1335e-02, 9.9668e-07,
         5.2496e-07],
        [8.9409e-06, 1.0212e-05, 1.4552e-05,  ..., 5.2131e-04, 5.0510e-02,
         4.4482e-05],
        [2.1490e-06, 2.5574e-06, 4.4111e-06,  ..., 3.1576e-05, 4.8764e-03,
         3.2458e-02]]), 'X': tensor([[  5.5101,  -2.1433,   9.5442],
        [  5.7784,  -1.6401,   9.5587],
        [  6.1222,  -0.7261,   8.8204],
        [  5.8889,  -1.3916,   8.8993],
        [  6.3266,   0.3292,   9.3502],
        [  6.0054,  -1.0902,   8.3079],
        [ 11.4235,   6.3594,   5.1663],
        [  4.7544,  -3.1425,  -1.5315],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.9433,  -9.3362,  -0.3421],
        [ -6.5295,  -7.9785,  -7.4429],
        [  3.0876,   6.7527,  -5.3540],
        [  4.7813,  -3.1139,   8.2691],
        [  2.4572,   7.1085,   7.7376],
        [ 10.3251,   8.2928,  -3.0006],
        [  8.8474,  10.0868,  -8.8644],
        [ -9.3444,   0.5821,   6.9101],
        [ -5.5525,   6.5146,  -4.4984],
        [ -9.0961,   1.6529,   3.1553],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.0588,  -6.0341,  -2.3006],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.4573,   3.2448,   5.4370],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.4735,  -0.7274,  -2.6749],
        [  1.7815,  -4.7481,   4.4289],
        [  0.6142,  14.0802,   0.5407],
        [  1.8719,   7.3713,  -3.7465],
        [ -2.5161,   7.7699,  -9.3531],
        [  4.0421,  -3.7765,  -8.3631],
        [  2.3373,  13.7721,  -0.2442],
        [ -5.3179,  -9.0101,   9.4472],
        [  6.6319,  11.9066,   3.5165],
        [ -2.2188,   7.8218,   8.6064],
        [  2.6310, -10.9195,   3.8054],
        [  0.6470, -11.0352,  -1.1931],
        [  6.1066,  -0.7799,  -4.1746],
        [ 12.5650,   2.0132,   4.6272],
        [  8.2154,  -8.4262,   8.0171],
        [ -9.2687,  -2.9794,   0.6436],
        [  7.8425,  -8.7202,   6.4530],
        [  2.4932,   7.0903,  -5.4214],
        [  6.3534,   1.2990,  -7.2942],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.3604,   0.9042,  -6.5901],
        [  5.5626,  -2.0536,   8.4433],
        [  6.9433,  -9.3362,  -0.3421],
        [  0.2480, -11.0153,  -3.0183],
        [  4.0587, -10.6112,   6.2273],
        [  6.0560,  -0.9421,  -5.0334],
        [ 11.5198,  -4.0923,   5.5088],
        [ 12.1504,   4.3674,  -7.2998],
        [  7.2340,  -9.1505,  -5.8321],
        [  2.9260,  -4.4162,   8.3808],
        [ 10.9931,   7.2107,   7.7933],
        [ -8.6229,   2.9068,   2.4429],
        [  8.5198,  10.4091,  -8.6486],
        [  1.9241,   7.3503,   7.9004],
        [ 11.2517,  -4.6495,   2.9800],
        [ -6.1914,  -8.2997,   8.5380],
        [  7.9403,  -8.6455,  -8.7433],
        [  2.0729,   7.2878,   7.0880],
        [  2.6613,   7.0021,   4.5783],
        [ 10.9394,   7.3069,  -8.9365],
        [  5.9193,  -1.3176,   5.1739],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.3122,  -9.6994,  -6.3282],
        [ -0.1963,   7.8765,  -1.1371],
        [ -6.0175,  -8.4541,  -9.8408],
        [  7.9816,  10.8938,  -0.5378],
        [  2.2373,  -4.6506,  -1.7334],
        [ -7.2515,   5.0195,   4.6434],
        [ -6.3306,   5.9272,   5.6935],
        [ -6.3252,  -8.1760,  -1.0891],
        [  4.5954,  13.0099,  -4.4439],
        [  9.2811,  -7.4336,   5.1771],
        [  8.6271,  -8.0715,  -7.4813],
        [ -1.2454,   7.9127,  -0.5715],
        [  4.5503,   5.5359,   7.4735],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.3604,   0.4855,  -3.0158],
        [  8.0381,  -8.5690,  -8.9523],
        [  4.3459,  -3.5327,   0.3667],
        [  0.8940, -11.0402,  -0.9783],
        [  2.7685, -10.8982,  -6.8928],
        [  4.9531,   5.0576,   5.6794],
        [  6.3569,   0.7591,   6.5011],
        [ -8.0965,  -5.9712,   4.3466],
        [  4.7482, -10.3907,   5.9149],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.9452, -10.3187,  -7.8621],
        [  6.8407,  11.7666,   3.0045],
        [ -1.6334, -10.7207,   7.1397],
        [  5.4426, -10.1181,  -5.1268],
        [ 11.5966,   5.9666,   6.0083],
        [  6.3413,   0.4883,  -6.8416],
        [ 12.2311,  -2.0445,   1.3545],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -8.6229,   2.9068,   2.4429],
        [  6.6319,  11.9066,   3.5165],
        [ 10.1798,   8.5002,  -5.8893]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0069, 0.0126,  ..., 0.8381, 0.3055, 1.0517],
        [0.0069, 0.0000, 0.0082,  ..., 0.8448, 0.2995, 1.0517],
        [0.0126, 0.0082, 0.0000,  ..., 0.8465, 0.2930, 1.0441],
        ...,
        [0.8381, 0.8448, 0.8465,  ..., 0.0000, 1.0883, 1.2160],
        [0.3055, 0.2995, 0.2930,  ..., 1.0883, 0.0000, 0.9380],
        [1.0517, 1.0517, 1.0441,  ..., 1.2160, 0.9380, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.91222346,  6.00663314,  6.16512856,  6.05113217,  6.33517349,
        6.10360073, 13.0743399 ,  5.69913817,  9.37296102, 11.63502679,
       10.30972193,  7.42512621,  5.70591986,  7.52117132, 13.2430377 ,
       13.41713661,  9.36256096,  8.55982105,  9.24501829,  8.19703512,
       10.0674887 , 11.62158554,  9.05843201,  9.60359158,  9.5014146 ,
        5.0713364 , 14.09357547,  7.60529892,  8.16714836,  5.53175275,
       13.96905756, 10.46238054, 13.62896593,  8.1303945 , 11.23201619,
       11.05413545,  6.1561646 , 12.72524615, 11.76830981,  9.73579738,
       11.72802491,  7.5158489 ,  6.48486171,  9.78147542,  6.42439344,
        5.92952538, 11.63502679, 11.018084  , 11.36089355,  6.12886024,
       12.22503754, 12.91143491, 11.66453059,  5.29755487, 13.14690137,
        9.09963894, 13.45124917,  7.59795445, 12.17451685, 10.35464916,
       11.73848065,  7.57685908,  7.49076899, 13.15525803,  6.06416957,
        9.20340031, 11.57249984,  7.8788965 , 10.37698435, 13.50484539,
        5.16077764,  8.81930525,  8.67227386, 10.33712258, 13.79762415,
       11.89105259, 11.81423161,  8.01009083,  7.16598273,  7.9337647 ,
       12.11735334,  9.97964013,  9.6282763 ,  9.37296102, 11.74901778,
        5.60064319, 11.07637531, 11.24434374,  7.07902036,  6.4020404 ,
       10.06023665, 11.42420508, 12.0495521 ,  9.50913849, 10.63038565,
       11.4424816 , 13.6105595 , 10.84437705, 11.4890861 , 13.04156075,
        6.36004393, 12.40074901, 12.49291768,  9.09963894, 13.62896593,
       13.26209545])}
{'P': tensor([[2.9283e-02, 1.6046e-01, 1.5959e-01,  ..., 7.1843e-03, 3.5038e-04,
         1.2912e-05],
        [3.1880e-03, 7.6970e-02, 7.2085e-03,  ..., 2.1074e-02, 5.5853e-04,
         3.6787e-05],
        [4.0066e-03, 1.8768e-01, 4.9787e-02,  ..., 9.4667e-03, 3.8066e-04,
         2.0319e-05],
        ...,
        [8.7712e-04, 2.0533e-03, 1.1230e-03,  ..., 6.5111e-02, 4.9790e-04,
         6.2216e-05],
        [7.4439e-03, 9.4216e-03, 6.2236e-03,  ..., 5.6683e-02, 1.6514e-01,
         1.3662e-04],
        [2.2268e-04, 1.1798e-04, 1.4398e-04,  ..., 8.1630e-05, 1.3336e-03,
         7.2322e-02]]), 'X': tensor([[  5.5158,  -2.1337,   3.5246],
        [  6.1741,  -0.5329,   3.1913],
        [  5.8902,  -1.3887,   4.0388],
        [  6.0890,  -0.8382,   2.9166],
        [  5.7151,  -1.7700,   3.1542],
        [  6.2902,   0.0451,   3.7116],
        [  3.9770, -10.6341,   8.0783],
        [  3.6249, -10.7255,   5.6605],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.3739,   0.3982,  -4.2564],
        [  5.8889,  -1.3916,   8.8993],
        [  9.2692,   9.6370,   7.0746],
        [ -8.4297,  -5.3637,  -9.3962],
        [ 12.5717,   1.9266,   7.4903],
        [  2.2076,  13.8032,  -1.5099],
        [ 12.3357,   3.6001,   0.7903],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.0506,   4.9277,  -9.5094],
        [  5.1203,  -2.7151,   6.2110],
        [  1.0391,  -4.8141,  -0.0768],
        [ -3.5520,   7.4976,   9.1522],
        [ 11.4703,  -4.2006,   7.4086],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 11.1420,   6.9331,  -2.4779],
        [  7.8425,  -8.7202,   6.4530],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.1206,  -4.7363,   8.7103],
        [ 11.0438,   7.1180,  -5.6991],
        [  0.3493,  -4.7727,  -6.5682],
        [ -6.3252,  -8.1760,  -1.0891],
        [  9.4925,   9.3811,   8.9842],
        [ -2.2188,   7.8218,   8.6064],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.2480, -11.0153,  -3.0183],
        [  2.9410,   6.8428,   5.4458],
        [  5.8617,   3.4853,  -9.8080],
        [ -0.8455, -10.8853,  -6.7400],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -1.5408,   7.8978,   5.3470],
        [  8.7715,  -7.9390,   5.2900],
        [  0.9036,   7.6840,   9.9237],
        [  3.5965,  -4.0754,  -3.5035],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.5829,  -4.5466,  -8.4015],
        [  4.2096,   5.8815,   1.9230],
        [ -4.0868,   7.2981,   8.1874],
        [  2.3589,  13.7668,  -3.6501],
        [ -5.8389,  -8.6056,  -3.5098],
        [  0.3493,  -4.7727,  -6.5682],
        [  1.5662,  13.9375,   4.5142],
        [  1.8766,   7.3694,   1.1887],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.4159,  -2.0739,   9.6841],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.4877,  -0.7222,   8.8347],
        [ -4.6393,   7.0461,   1.1624],
        [ -5.6169,  -8.7844,   8.9057],
        [  2.6310, -10.9195,   3.8054],
        [  9.4925,   9.3811,   8.9842],
        [  3.6249, -10.7255,   5.6605],
        [ -5.7347,  -8.6908,   9.2878],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.5128, -10.9365,   9.9645],
        [  5.6523,   3.9443,  -2.9449],
        [  2.4932,   7.0903,  -5.4214],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.3551,   0.7149,   1.3344],
        [  3.1892, -10.8220,  -6.8409],
        [  0.1206,  -4.7363,   8.7103],
        [ 11.5198,  -4.0923,   5.5088],
        [  4.5800,   5.5034,  -3.9432],
        [  0.6396,  14.0772,  -6.3786],
        [  5.1292,   4.8183,  -5.7997],
        [  5.1886,  -2.6250,   6.4677],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.2036,   2.4030,  -1.2500],
        [  6.3569,   0.7591,   6.5011],
        [  5.5626,  -2.0536,   8.4433],
        [ 12.2592,   3.9425,   2.0383],
        [ -5.3328,  -8.9992,   7.3440],
        [ 12.1379,   4.4125,  -3.8833],
        [ -8.9636,   2.0648,   8.1946],
        [ -1.5408,   7.8978,   5.3470],
        [  2.9410,   6.8428,   5.4458],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.3373,  -4.6232,  -5.2246],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.0534,  -6.0429,   2.4583],
        [ 11.5456,   6.0861,   8.1101],
        [  3.1756, -10.8247,   6.8882],
        [ -3.2736, -10.1703,   3.8712],
        [  1.1644, -11.0394,   7.8511],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.4203,  -2.0340,   2.5856],
        [ -5.1398,  -9.1367,   6.2074],
        [ -1.0012, -10.8576,   2.2867],
        [ -0.5739,   7.9056,   0.6382],
        [ 11.4235,   6.3594,   5.1663],
        [  6.3087,   1.8053,  -1.9481],
        [  3.7237,  -3.9966,  -5.0376],
        [ -9.2301,  -3.1537,  -2.7477]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.2195e-02, 4.2190e-02,  ..., 3.8291e-01, 5.4299e-01,
         1.1031e+00],
        [2.2195e-02, 1.5170e-04, 5.8786e-02,  ..., 3.6533e-01, 5.2649e-01,
         1.1072e+00],
        [4.2190e-02, 5.8786e-02, 0.0000e+00,  ..., 4.2393e-01, 5.8463e-01,
         1.1312e+00],
        ...,
        [3.8291e-01, 3.6533e-01, 4.2393e-01,  ..., 0.0000e+00, 1.6415e-01,
         9.9259e-01],
        [5.4299e-01, 5.2649e-01, 5.8463e-01,  ..., 1.6415e-01, 0.0000e+00,
         9.4434e-01],
        [1.1031e+00, 1.1072e+00, 1.1312e+00,  ..., 9.9259e-01, 9.4434e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.91408504,  6.19707944,  6.05165248,  6.1463896 ,  5.98285319,
        6.29034958, 11.35345345, 11.32149259, 11.83509372,  9.38232121,
        6.05113217, 13.37121923,  9.99145096, 12.71843462, 13.97857591,
       12.85032836, 12.41012046,  7.05627461,  5.79562691,  4.92497514,
        8.29640613, 12.21532282, 10.52587243, 13.12299145, 11.72802491,
        5.52416368,  4.73784222, 13.13888841,  4.78544875, 10.33712258,
       13.34587027,  8.1303945 , 12.19805394, 11.018084  ,  7.44805901,
        6.81961996, 10.91805683,  6.06416957, 12.35625694,  8.04664886,
       11.83075005,  7.73692436,  5.43544626, 10.1371552 ,  5.22903117,
        7.23276582,  8.36445758, 13.96746816, 10.39942497,  4.78544875,
       14.02526505,  7.60463349,  9.43112957,  9.09963894,  9.64157461,
       11.81423161, 12.5085995 ,  8.43626114, 10.42667093, 11.23201619,
       13.34587027, 11.32149259, 10.412299  ,  9.36093562,  4.8763135 ,
       11.22142195,  6.89245352,  7.5158489 , 12.4980845 ,  9.64157461,
        6.39520397, 11.28215429,  4.73784222, 12.22503754,  7.15990676,
       14.09176259,  7.03733775,  5.81481044,  6.19633139,  6.652742  ,
        6.4020404 ,  5.92952538, 12.87752052, 10.4606265 , 12.91505111,
        9.19837792,  8.04664886,  7.44805901, 12.9089559 ,  5.18047638,
       10.36844813, 10.06850677, 13.05148826, 11.28093507, 10.68416401,
       11.10066126, 12.4217523 ,  5.13125513,  9.63742669, 10.48315436,
       10.9036226 ,  7.92644746, 13.0743399 ,  6.5618968 ,  5.46244988,
        9.75401449])}
{'P': tensor([[3.3686e-02, 1.7745e-01, 1.4531e-01,  ..., 6.4991e-06, 7.0431e-04,
         3.5376e-04],
        [1.7390e-03, 4.3406e-02, 1.1151e-01,  ..., 8.1267e-06, 5.0289e-04,
         2.3701e-04],
        [2.0747e-03, 1.1234e-02, 6.3765e-02,  ..., 1.4910e-05, 6.4170e-04,
         3.9745e-04],
        ...,
        [3.0500e-04, 1.5039e-04, 1.5822e-04,  ..., 3.3241e-01, 1.4221e-03,
         2.3403e-03],
        [1.2367e-01, 5.5687e-02, 4.5315e-02,  ..., 2.7208e-05, 8.4387e-02,
         1.7002e-03],
        [7.6747e-04, 8.6576e-04, 1.3183e-03,  ..., 3.8354e-05, 2.6837e-04,
         5.2071e-02]]), 'X': tensor([[  5.5240,  -2.1198,   6.2055],
        [  6.1973,  -0.4372,   6.5720],
        [  6.2895,   0.0405,   5.7652],
        [  6.2883,   0.0325,   5.2692],
        [  6.2975,   0.0947,   5.1791],
        [  6.3569,   0.7591,   6.5011],
        [  6.2975,   0.0947,   5.1791],
        [  6.1222,  -0.7261,   8.8204],
        [  5.9883,   3.1536,   6.2320],
        [  6.3031,   1.8477,  -7.1706],
        [ 10.9916,   7.2133,  -0.9179],
        [ -1.7009,   7.8851,  -8.2437],
        [ -5.7347,  -8.6908,   9.2878],
        [  1.1274,  14.0116,   8.1753],
        [ 11.1469,   6.9237,   1.6160],
        [  5.8979,   3.3958,  -3.8713],
        [  0.5600, -11.0321,  -0.8660],
        [  4.7544,  -3.1425,  -1.5315],
        [  9.0051,   9.9234,   5.4411],
        [  2.9461,  -4.4077,  -0.4831],
        [ -8.0330,  -6.0765,   6.7945],
        [ -2.5683, -10.4433,  -2.9182],
        [ -3.9508,  -9.8515,   1.7923],
        [  4.7739,  -3.1218,  -4.1188],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.4697,  -0.6157,  -6.8472],
        [  6.3497,   1.3680,  -1.9076],
        [  5.2939,  -2.4784,  -4.5840],
        [  7.8375,  -8.7240,  -9.8913],
        [ -1.9646,   7.8571,   3.3817],
        [  6.0467,  -9.8369,  -6.5816],
        [  4.3377,   5.7570,   1.3712],
        [  6.2630,   2.1057,  -5.2136],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.4378,  -1.8606,  -2.5668],
        [  7.1697,  11.5343,  -0.7276],
        [  2.0937,  13.8293,  -2.3533],
        [  7.3592,  11.3937,   6.2407],
        [  5.0617,  12.7957,   8.3544],
        [  2.5318,  13.7231,   6.3692],
        [  3.3839,  -4.1966,   8.3574],
        [  2.5974,   7.0363,  -7.7810],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.3587,   1.1596,  -3.6629],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.4630,  -1.5145,   0.2983],
        [ -6.6998,   5.5954,   0.0823],
        [ 10.9875,   7.2207,  -8.8549],
        [  7.7750,  11.0665,   2.8364],
        [  5.7218,  -9.9934,   6.5764],
        [ -0.0410,   7.8591,   9.3457],
        [  6.3211,   1.7017,  -0.2164],
        [ -5.7347,  -8.6908,   9.2878],
        [ -1.9600,   7.8577,   6.9121],
        [ -3.6381,   7.4683,   7.3544],
        [  6.3550,   0.7129,   0.4917],
        [ 11.2767,  -4.6006,   0.7868],
        [ -8.4267,  -5.3697,   2.4172],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.7295,  13.6703,  -8.9328],
        [  6.1989,  -9.7591,   0.8289],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3583,   1.1729,   8.7835],
        [  0.6470, -11.0352,  -1.1931],
        [ -4.9217,   6.8979,  -1.7667],
        [ -7.3554,   4.8985,  -0.6665],
        [ -4.4520,   7.1370,  -8.7867],
        [  3.4604,  -4.1545,   1.8091],
        [ 12.6059,   0.9597,  -6.3264],
        [ -0.2586,   7.8825,  -9.9840],
        [  3.3744,   6.5623,  -6.3636],
        [ -5.6651,   6.4380,   1.3660],
        [ -3.5520,   7.4976,   9.1522],
        [  9.7328,   9.0902,  -0.8917],
        [  4.8674,  12.8875,   6.7622],
        [  6.3551,   0.7149,   1.3344],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.7078,   2.7182,   1.8146],
        [  1.8719,   7.3713,  -3.7465],
        [  2.0419,  -4.6978,  -5.0983],
        [  9.6614,   9.1784,  -1.2087],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.4309,  -0.0599,  -4.0830],
        [  1.9204,   7.3518,   8.4781],
        [  3.9663,   6.1018,  -5.4104],
        [  0.9788,  -4.8145,  -3.8661],
        [  6.0310,  12.2795,  -6.4528],
        [  2.5010,  -4.5736,  -0.8989],
        [  6.3246,   1.6696,  -0.1544],
        [  9.1820,   9.7334,  -2.7932],
        [  0.9036,   7.6840,   9.9237],
        [  0.2435,   7.8191,  -6.4709],
        [ -7.7239,   4.4298,   3.1216],
        [  6.3535,   0.6792,  -7.5101],
        [ -9.4430,  -0.1897,   4.8383],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.3251,   8.2928,  -3.0006],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.2366,  -4.2732,   5.1532],
        [  6.7565,  11.8237,  -4.9882],
        [  2.5620,   7.0549,   8.4825],
        [  4.0457,  -3.7737,  -0.4704],
        [  0.3038,   7.8092,   0.4164],
        [ -9.3384,  -2.6161,  -4.7630],
        [  2.2437,  -4.6489,   6.1430],
        [  5.1142,  -2.7230,  -4.1966]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0272, 0.0202,  ..., 1.2903, 0.0612, 0.7007],
        [0.0272, 0.0000, 0.0430,  ..., 1.3150, 0.0795, 0.7256],
        [0.0202, 0.0430, 0.0000,  ..., 1.2852, 0.0698, 0.6827],
        ...,
        [1.2903, 1.3150, 1.2852,  ..., 0.0000, 1.2387, 0.9767],
        [0.0612, 0.0795, 0.0698,  ..., 1.2387, 0.0000, 0.6927],
        [0.7007, 0.7256, 0.6827,  ..., 0.9767, 0.6927, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.91677188,  6.21275016,  6.28961988,  6.28835101,  6.29821673,
        6.4020404 ,  6.29821673,  6.16512856,  6.76791483,  6.56833864,
       13.1471313 ,  8.06643487, 10.412299  , 14.05687815, 13.12218013,
        6.80559669, 11.04629471,  5.69913817, 13.40024224,  5.30158497,
       10.07238463, 10.75443106, 10.61417181,  5.70405525, 11.64746882,
        9.48969978,  6.49538841,  5.84533345, 11.72749923,  8.09899874,
       11.54672755,  7.20826219,  6.60752373, 12.19805394,  9.61942445,
       13.58101794, 13.98691426, 13.56367857, 13.76047999, 13.95472331,
        5.39098669,  7.50035355,  4.76969364,  6.46356806,  9.69629   ,
        9.58347456,  8.72896381, 13.14777433, 13.5247215 , 11.51557166,
        7.85919687,  6.54615887, 10.412299  ,  8.09842704,  8.30729352,
        6.3949043 , 12.17901033,  9.99211663, 10.4606265 , 13.94008948,
       11.56146873,  6.12886024,  6.46560181, 11.05413545,  8.47371868,
        8.83725081,  8.41171369,  5.40689332, 12.64235359,  7.88677538,
        7.37902766,  8.57560275,  8.29640613, 13.31764181, 13.77604293,
        6.39520397,  9.3109884 ,  9.12220271,  7.60529892,  5.1224079 ,
       13.32613445,  7.37902766,  9.43112957,  7.5984809 ,  7.27759233,
        4.91296205, 13.68063098,  5.21280156,  6.54127431, 13.38091121,
        7.73692436,  7.822854  ,  8.90405823,  6.38968357,  9.44486609,
        6.80559669, 13.2430377 , 11.89855267,  5.36061974, 13.61801526,
        7.50564223,  5.53255882,  7.81509631,  9.69792513,  5.16205087,
        5.79391825])}
{'P': tensor([[3.2495e-02, 2.6088e-01, 1.9947e-01,  ..., 1.6840e-04, 6.6295e-06,
         9.3024e-06],
        [2.8315e-03, 4.5330e-02, 4.1627e-01,  ..., 1.3794e-04, 1.0204e-05,
         1.1506e-05],
        [3.7169e-03, 4.9359e-03, 8.2343e-02,  ..., 1.6505e-04, 2.0332e-05,
         1.9115e-05],
        ...,
        [1.0736e-02, 1.0609e-02, 9.4397e-03,  ..., 2.9246e-02, 5.4612e-07,
         2.1316e-06],
        [2.7484e-05, 2.2452e-05, 1.8704e-05,  ..., 3.2611e-05, 5.4485e-02,
         3.0635e-02],
        [3.0898e-04, 2.3174e-04, 1.7595e-04,  ..., 1.9502e-03, 2.4959e-05,
         4.4093e-02]]), 'X': tensor([[  5.5294,  -2.1106,   9.0626],
        [  5.8889,  -1.3916,   8.8993],
        [  6.1222,  -0.7261,   8.8204],
        [  5.7784,  -1.6401,   9.5587],
        [  6.0054,  -1.0902,   8.3079],
        [  6.3266,   0.3292,   9.3502],
        [ -3.5903,   7.4847,  -8.9112],
        [ -5.3382,   6.6533,   7.8054],
        [  9.2811,  -7.4336,   5.1771],
        [  5.0506,   4.9277,  -9.5094],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.3413,   0.4883,  -6.8416],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.0190,   4.9705,  -7.9752],
        [  4.3377,   5.7570,   1.3712],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.0829,  -5.9940,   5.0564],
        [  3.8111, -10.6787,  -4.6539],
        [  3.2701,   6.6338,   9.9209],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.2990,  -4.2415,  -7.4857],
        [ 10.1503,   8.5414,  -7.2681],
        [ 12.0772,  -2.6014,  -7.1832],
        [  0.1931,  -4.7491,   7.4385],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.5382,   7.0672,   4.5097],
        [  2.2182,  13.8007,   5.1889],
        [  2.3200,   7.1756,   1.1938],
        [ 12.6038,   0.7665,  -3.4439],
        [ -7.5720,   4.6310,   2.1717],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.1204,  -5.9308,   4.0761],
        [  1.9204,   7.3518,   8.4781],
        [ 11.9641,  -2.9568,   7.7211],
        [  3.3167,   6.6022,   6.3554],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -0.9715, -10.8630,   0.3555],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.7485,  -9.9811,  -3.2355],
        [  3.7237,  -3.9966,  -5.0376],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.4117,   0.1160,   8.2574],
        [  5.6248,   3.9984,   8.4933],
        [  6.2026,  12.1774,  -1.5887],
        [ -1.8367,   7.8718,  -8.6523],
        [  8.5716,  10.3596,   6.8818],
        [ -8.5149,  -5.1905,   1.5239],
        [ -5.9987,   6.1949,  -6.2114],
        [  4.4633,   5.6287,  -0.7754],
        [  8.6420,  -8.0580,   6.5781],
        [  2.5375,   7.0676,  -4.3490],
        [ 10.9916,   7.2133,  -0.9179],
        [ 11.4495,   6.3026,   6.8257],
        [  4.0587, -10.6112,   6.2273],
        [  6.2576,   2.1359,  -3.3398],
        [ -7.0825,   5.2073,  -5.0497],
        [ -1.8367,   7.8718,  -8.6523],
        [ -1.6086,   7.8928,   4.3385],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -8.3535,  -5.5118,   9.8399],
        [ -3.4488, -10.0934,   2.5195],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.4464,   5.8272,   7.5038],
        [  5.6258,  -1.9404,  -6.0715],
        [ -4.4779,  -9.5609,  -2.4505],
        [  9.4925,   9.3811,   8.9842],
        [  2.7590,  13.6621,   9.4496],
        [ -9.4296,  -1.9458,   4.0833],
        [  6.2038,   2.4020,  -3.1692],
        [ -7.9925,  -6.1423,   2.8990],
        [ -0.8455, -10.8853,  -6.7400],
        [ 10.5678,   7.9261,  -0.6100],
        [ -7.6650,   4.5092,   0.2520],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.5900,   2.9767,   7.4811],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.0948,   5.1940,   9.8171],
        [ 11.7779,   5.5127,  -2.0699],
        [ -7.3554,   4.8985,  -0.6665],
        [ -9.4541,  -0.3327,   8.8246],
        [ -5.1377,   6.7751,  -0.4438],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.2770,  -0.0379,   2.4195],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -7.4733,   4.7555,   6.2490],
        [  1.7301,   7.4262,   5.0522],
        [  3.9770, -10.6341,   8.0783],
        [  7.3724,  -9.0577,  -8.9843],
        [  0.9788,  -4.8145,  -3.8661],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.3616,  -2.4750,   3.7765],
        [ 12.6000,   1.3839,  -3.2992],
        [  9.0424,  -7.6779,  -7.5286],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.4716,  -0.6673,   7.6068],
        [ 11.0275,  -5.0666,   8.4154],
        [  4.6231,   5.4556,   8.4973],
        [  3.2159, -10.8166,  -1.0406],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -6.0844,  -8.3955,   0.7705]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0094, 0.0131,  ..., 0.4134, 1.2011, 1.0549],
        [0.0094, 0.0000, 0.0037,  ..., 0.4216, 1.2094, 1.0639],
        [0.0131, 0.0037, 0.0000,  ..., 0.4247, 1.2128, 1.0676],
        ...,
        [0.4134, 0.4216, 0.4247,  ..., 0.0000, 1.0388, 0.8148],
        [1.2011, 1.2094, 1.2128,  ..., 1.0388, 0.0000, 0.2893],
        [1.0549, 1.0639, 1.0676,  ..., 0.8148, 0.2893, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.91854397,  6.05113217,  6.16512856,  6.00663314,  6.10360073,
        6.33517349,  8.30125201,  8.53014915, 11.89105259,  7.05627461,
       10.46739073,  6.36004393,  5.56809335,  7.06372172,  7.20826219,
       13.1399181 , 10.06286074, 11.33837396,  7.39599986,  4.76969364,
        5.37343173, 13.26590071, 12.35421461,  4.75303127, 10.412299  ,
        7.50917551, 13.97779839,  7.54127425, 12.62711252,  8.87587457,
        8.22728399, 10.05558893,  7.5984809 , 12.32408516,  7.38844703,
       10.37698435, 10.90638066,  4.90505566, 11.51811603,  5.46244988,
       10.19036081,  9.41244865,  6.90115901, 13.6660671 ,  8.08320235,
       13.44593519,  9.97220799,  8.62329466,  7.18354885, 11.8159254 ,
        7.5092834 , 13.1471313 , 13.06957425, 11.36089355,  6.61211224,
        8.7907923 ,  8.08320235,  8.0550286 , 10.09979747, 10.00802627,
       10.66631784, 13.96905756, 12.71843462, 10.01804223,  8.68979052,
        5.9510489 , 10.5575601 , 13.34587027, 13.93790235,  9.6282763 ,
        6.65259775, 10.08001894, 10.91805683, 13.20988996,  8.89301516,
        8.1137452 ,  9.09119351,  9.43112957, 11.59512194,  9.7712321 ,
        8.79284466, 13.00413829,  8.83725081,  9.45995955,  8.50278896,
        6.10774544,  6.27715105, 10.30972193,  8.8580671 ,  7.62509582,
       11.35345345, 11.67875855,  4.91296205,  9.60412735,  9.68324553,
       12.67576792, 11.86239961,  9.88571713,  8.94395628,  9.49511524,
       12.13568598,  7.15100303, 11.2845608 , 11.62158554,  9.43737523,
       10.36844813])}
{'P': tensor([[3.2812e-02, 1.7344e-01, 1.5128e-01,  ..., 2.4282e-05, 9.7622e-05,
         1.9176e-04],
        [2.4471e-03, 6.4756e-02, 3.1099e-02,  ..., 1.9714e-05, 3.0963e-04,
         2.1103e-04],
        [1.8666e-03, 4.0429e-02, 4.2602e-02,  ..., 1.6849e-05, 2.6896e-04,
         1.7903e-04],
        ...,
        [7.7451e-04, 6.2716e-04, 9.2675e-04,  ..., 6.9005e-02, 3.3999e-05,
         1.5705e-03],
        [1.7274e-05, 1.6549e-05, 2.0288e-05,  ..., 7.6713e-06, 4.6015e-02,
         1.6347e-04],
        [4.0507e-03, 2.8298e-03, 4.2479e-03,  ..., 8.4009e-03, 3.6757e-04,
         8.9602e-02]]), 'X': tensor([[  5.5424,  -2.0885,   7.2487],
        [  6.2242,  -0.3171,   7.4978],
        [  6.1973,  -0.4372,   6.5720],
        [  6.3549,   0.7096,   6.9198],
        [  6.0054,  -1.0902,   8.3079],
        [  6.3569,   0.7591,   6.5011],
        [  5.9655,   3.2172,  -7.5953],
        [  8.1528,  10.7452,   9.1493],
        [  5.5294,  -2.1106,   9.0626],
        [ -8.9942,  -3.9990,   6.0448],
        [ -7.6719,  -6.6260,   7.1468],
        [  1.7500,   7.4187,   4.4237],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.7232,  -4.7574,   3.1270],
        [  0.3231,  -4.7691,  -1.6120],
        [  3.8182,  -3.9347,  -1.7961],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.8675, -10.8815,   6.8561],
        [  3.3167,   6.6022,   6.3554],
        [ -7.8359,   4.2735,   7.2943],
        [ -1.0012, -10.8576,   2.2867],
        [ 12.5224,  -0.4497,   4.8448],
        [  0.1931,  -4.7491,   7.4385],
        [ -8.8934,   2.2597,   3.5831],
        [ -7.4887,   4.7364,   8.9074],
        [ -2.6069, -10.4298,  -6.1995],
        [ -4.6393,   7.0461,   1.1624],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.9587,  12.3216,   9.1695],
        [  5.7845,   3.6652,   0.9175],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.6603,   2.8250,   5.9181],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.2038,   2.4020,  -3.1692],
        [  1.6232,  -4.7717,  -2.5655],
        [  2.3261,   7.1726,   6.2522],
        [  0.9967,  -4.8144,  -9.7370],
        [ -7.8359,   4.2735,   7.2943],
        [  6.2790,  -0.0259,  -5.4116],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -1.5408,   7.8978,   5.3470],
        [  5.3118,   4.5452,   4.2114],
        [  3.6154,  -4.0640,  -7.9698],
        [ -9.4196,   0.0478,   9.4140],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.3810,  -2.3455,  -9.5194],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -0.5739,   7.9056,   0.6382],
        [  3.3474,  -4.2162,   5.2455],
        [ -1.5408,   7.8978,   5.3470],
        [  9.1620,  -7.5573,  -9.9882],
        [  3.9663,   6.1018,  -5.4104],
        [  5.4426, -10.1181,  -5.1268],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.8201,   3.5841,  -8.0609],
        [ -7.8359,   4.2735,   7.2943],
        [  1.2728, -11.0372,   9.8121],
        [  6.1752,   2.5242,   3.5874],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -8.0608,   3.9362,  -5.9567],
        [  8.4691,  10.4571,  -5.0650],
        [ -2.8162,   7.7059,  -9.6885],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 12.4877,  -0.7222,   8.8347],
        [  7.5309,  11.2617,   7.2656],
        [  5.6093,  -1.9705,   6.4807],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.2339,  -3.6268,   6.3125],
        [  0.6933,   7.7336,   2.1179],
        [  6.3211,   1.7017,  -0.2164],
        [  6.2902,   0.0451,   3.7116],
        [  0.5175,  14.0910,   4.6540],
        [  3.9663,   6.1018,  -5.4104],
        [  5.4242, -10.1261,   3.9595],
        [ -5.9938,  -8.4745,   3.9735],
        [  0.6757, -11.0361,  -1.3047],
        [  0.5175,  14.0910,   4.6540],
        [ -9.3464,   0.5708,  -1.6397],
        [ -8.8154,   2.4611,   2.5198],
        [  6.1973,  -0.4372,   6.5720],
        [ -3.2736, -10.1703,   3.8712],
        [  2.2373,  -4.6506,  -1.7334],
        [ -8.4297,  -5.3637,  -9.3962],
        [  9.6587,   9.1817,  -0.8212],
        [  3.4009,  -4.1874,  -5.9751],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.5829,  -4.5466,  -8.4015],
        [ 12.2692,  -1.8882,   3.9253],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 12.5770,   1.8512,  -0.6578],
        [ -9.1178,   1.5784,   2.1410],
        [ -2.3317,   7.8034,  -3.6274],
        [ -5.0550,   6.8231,  -9.2502],
        [  4.6063,  13.0051,  -3.1456],
        [ -5.1398,  -9.1367,   6.2074],
        [ -3.1671,   7.6158,   6.0765],
        [  2.4639, -10.9431,  -7.3718],
        [  5.3924, -10.1396,  -4.0863],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.3403,  -4.8011,  -1.1698]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0174, 0.0232,  ..., 0.8962, 0.7674, 0.5808],
        [0.0174, 0.0000, 0.0367,  ..., 0.9134, 0.7845, 0.5970],
        [0.0232, 0.0367, 0.0000,  ..., 0.8797, 0.7498, 0.5607],
        ...,
        [0.8962, 0.9134, 0.8797,  ..., 0.0000, 0.1427, 0.3667],
        [0.7674, 0.7845, 0.7498,  ..., 0.1427, 0.0000, 0.2240],
        [0.5808, 0.5970, 0.5607,  ..., 0.3667, 0.2240, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.92281142,  6.2322769 ,  6.21275016,  6.39438044,  6.10360073,
        6.4020404 ,  6.77777606, 13.48809595,  5.91854397,  9.84314712,
       10.1371552 ,  7.62232307, 11.8159254 ,  9.17905491,  5.05989547,
        4.78002665,  5.48276038,  7.822854  , 10.91601655,  7.38844703,
        8.92549752, 10.9036226 , 12.53047489,  4.75303127,  9.17595899,
        8.86082124, 10.7506423 ,  8.43626114, 11.51811603, 13.68672855,
        6.84796202,  9.85384623,  9.10945678, 10.06407839,  6.65259775,
        5.04028538,  7.5403822 ,  4.91652556,  8.92549752,  6.27905292,
       10.30972193,  8.04664886,  6.99097013,  5.43943791,  9.41970031,
       12.64235359,  9.77588801,  9.66977802,  4.91038884,  9.96152788,
        7.92644746,  5.38342077,  8.04664886, 11.87665501,  7.27759233,
       11.4890861 , 12.59840193,  6.83515201,  8.92549752, 11.11038551,
        6.67121748, 10.37698435,  8.97052115, 13.45642379,  8.20436764,
        9.50654967, 12.5085995 , 13.54774355,  5.94535529,  7.27340834,
        9.79039371, 13.25017211,  5.5748624 ,  7.76457095,  6.54615887,
        6.29034958, 14.10045797,  7.27759233, 11.48734978, 10.37998078,
       11.05672136, 14.10045797,  9.36378438,  9.15253344,  6.21275016,
       10.68416401,  5.16077764,  9.99145096, 13.32645552,  5.39450305,
       13.14777433,  5.22903117, 12.41366754,  6.00663314, 12.10942412,
       12.7125079 ,  9.25336371,  8.14433542,  8.49161154, 13.79676654,
       10.48315436,  8.24808832, 11.21703654, 11.48434767,  8.06990483,
        4.98461983])}
{'P': tensor([[3.2921e-02, 2.4750e-01, 1.9604e-01,  ..., 2.4157e-04, 4.3097e-05,
         3.9679e-03],
        [3.2857e-03, 6.0717e-02, 2.1261e-01,  ..., 6.9758e-04, 6.5567e-05,
         4.2843e-03],
        [3.7927e-03, 1.3184e-02, 7.9060e-02,  ..., 8.7463e-04, 6.5923e-05,
         3.8954e-03],
        ...,
        [1.8676e-06, 2.5508e-06, 2.6343e-06,  ..., 1.4310e-02, 2.2373e-06,
         3.3298e-06],
        [4.2390e-04, 4.3489e-04, 3.3979e-04,  ..., 2.1446e-04, 8.8576e-02,
         3.9250e-03],
        [1.6259e-02, 2.7758e-02, 2.4138e-02,  ..., 2.8697e-03, 8.3092e-04,
         1.2297e-01]]), 'X': tensor([[  5.5626,  -2.0536,   8.4433],
        [  6.0054,  -1.0902,   8.3079],
        [  6.1222,  -0.7261,   8.8204],
        [  5.8889,  -1.3916,   8.8993],
        [  6.2242,  -0.3171,   7.4978],
        [  6.3266,   0.3292,   9.3502],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.9655,   3.2172,  -7.5953],
        [  7.0642,  11.6104,  -9.8316],
        [ -6.6895,   5.6051,   2.7899],
        [  1.9503,  -4.7172,  -6.2313],
        [ -7.4808,   4.7462,   2.0266],
        [  7.7750,  11.0665,   2.8364],
        [  3.7916,  -3.9524,  -7.9547],
        [ 10.5057,  -5.9099,  -5.1599],
        [  8.2422,  -8.4041,  -1.4962],
        [  4.7482, -10.3907,   5.9149],
        [  3.1064,  13.5607,   6.7605],
        [ -5.1046,  -9.1610,   2.3418],
        [  2.3373,  13.7721,  -0.2442],
        [  3.3069,   6.6089,   8.7377],
        [  1.3403,  -4.8011,  -1.1698],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.3460,   1.4252,   1.2291],
        [ -5.6174,   6.4708,  -1.7339],
        [  1.5246,  -4.7838,  -5.2095],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.0788,  -0.8710,  -6.8719],
        [ -7.0948,   5.1940,   9.8171],
        [ -5.0132,  -9.2234,  -8.3531],
        [  2.1824,  -4.6647,   1.1985],
        [ 11.7794,   5.5086,  -9.6929],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -8.6050,   2.9451,  -0.7012],
        [  8.1043,  -8.5164,  -7.9290],
        [ -7.5603,   4.6459,   8.2002],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -3.3116, -10.1539,   7.5919],
        [  2.3367,   7.1676,  -9.5263],
        [ -4.8665,  -9.3204,  -5.2564],
        [  6.3031,   1.8477,  -7.1706],
        [  2.8890,  -4.4316,   1.1600],
        [  5.8836,   3.4315,  -4.0327],
        [ 11.7795,  -3.4697,   3.5624],
        [ -5.9987,   6.1949,  -6.2114],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -8.9178,   2.1936,   7.5388],
        [  6.1752,   2.5242,   3.5874],
        [  5.2188,  -2.5839,   2.1833],
        [  1.3339,   7.5627,  -0.9034],
        [ -9.4430,  -0.1897,   4.8383],
        [  4.5767,   5.5071,   3.9069],
        [ -1.0407,   7.9166,  -6.6021],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.3205,   1.7068,  -3.8619],
        [  5.7218,  -9.9934,   6.5764],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3604,   0.9042,  -6.5901],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.3889,  -4.6082,  -8.7291],
        [  7.5784,  -8.9141,   5.8410],
        [  6.7565,  11.8237,  -4.9882],
        [  0.1041, -11.0045,   2.4311],
        [  3.5284, -10.7484,   5.3347],
        [ -2.6069, -10.4298,  -6.1995],
        [ 12.6059,   0.9597,  -6.3264],
        [ -7.3470,  -7.0615,   5.0066],
        [ -7.2515,   5.0195,   4.6434],
        [ -2.0229, -10.6163,   8.6930],
        [  6.0031,  12.2958,  -6.5992],
        [ -3.5173, -10.0623,   4.8280],
        [  5.4179,  -2.2925,  -3.4641],
        [ -0.3432,   7.8900,   7.1805],
        [ 12.5832,   1.7543,  -7.6134],
        [ -9.1749,  -3.3803,   6.4629],
        [  6.9693,  -9.3200,  -8.3822],
        [  6.3534,   1.2990,  -7.2942],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.9883,   3.1536,   6.2320],
        [ 12.2914,   3.8037,   6.0978],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.8058,   2.4850,  -1.7425],
        [ -2.5683, -10.4433,  -2.9182],
        [ -5.3382,   6.6533,   7.8054],
        [ 11.9695,  -2.9407,  -6.3498],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.2974,   0.8371,   2.4852],
        [ 10.0619,   8.6627,  -9.2884],
        [ -6.1914,  -8.2997,   8.5380],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.3360,   0.4258,  -9.2509],
        [  2.6321,  -4.5296,  -2.5159],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -8.8745,   2.3097,   3.5600],
        [ -6.6895,   5.6051,   2.7899],
        [ -4.4503,   7.1378,   3.3751],
        [  4.4915,   5.5990,   7.0405],
        [  4.5503,   5.5359,   7.4735],
        [  1.9312,  13.8649,   8.3563],
        [  3.3633,  13.4794,   0.5193],
        [  7.1697,  11.5343,  -0.7276],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.5158,  -2.1337,   3.5246]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.8123e-03, 2.0081e-02,  ..., 6.5758e-01, 8.2172e-01,
         2.7309e-01],
        [9.8123e-03, 2.1453e-04, 1.6262e-02,  ..., 6.5320e-01, 8.2205e-01,
         2.7311e-01],
        [2.0081e-02, 1.6262e-02, 0.0000e+00,  ..., 6.6679e-01, 8.3825e-01,
         2.8927e-01],
        ...,
        [6.5758e-01, 6.5320e-01, 6.6679e-01,  ..., 2.1453e-04, 3.9686e-01,
         4.2905e-01],
        [8.2172e-01, 8.2205e-01, 8.3825e-01,  ..., 3.9686e-01, 0.0000e+00,
         5.4902e-01],
        [2.7309e-01, 2.7311e-01, 2.8927e-01,  ..., 4.2905e-01, 5.4902e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.92952538,  6.10360073,  6.16512856,  6.05113217,  6.2322769 ,
        6.33517349, 10.00802627,  6.77777606, 13.59056613,  8.7273569 ,
        5.10444097,  8.859416  , 13.5247215 ,  5.47702766, 12.05395056,
       11.77124373, 11.42420508, 13.91198191, 10.48721382, 13.96905756,
        7.39004226,  4.98461983, 12.79126785,  6.50409372,  8.56889738,
        5.0209005 ,  5.5513072 ,  6.14086582,  8.79284466, 10.4977132 ,
        5.14999799, 13.00380624,  9.78286733,  9.09502385, 11.75618513,
        8.8737556 , 12.05395056, 10.68031351,  7.53884104, 10.51437947,
        6.56833864,  5.29009836,  6.81117802, 12.27991991,  8.62329466,
       10.00410625,  9.18358562,  6.67121748,  5.82345867,  7.67939546,
        9.44486609,  7.16058587,  7.98469166, 10.37698435,  6.54693987,
       11.51557166, 10.06286074,  6.60752373,  6.42439344, 13.13205357,
       11.63765273,  5.19065439, 11.70016217, 13.61801526, 11.0050379 ,
       11.31276109, 10.7506423 , 12.64235359, 10.19036081,  8.81930525,
       10.80728788, 13.6829843 , 10.65929968,  5.8828986 ,  7.89745107,
       12.70489563,  9.77777346, 11.63765273,  6.48486171,  5.94541803,
        6.76791483, 12.86648771, 10.48840476,  9.14973319, 10.75443106,
        8.53014915, 12.32545637, 11.04629471,  9.3349858 , 13.27717982,
       10.35464916, 12.35421461, 10.48315436,  6.35029274,  5.23878371,
       10.51437947,  9.17016857,  8.7273569 ,  8.41148722,  7.17789666,
        7.16598273, 13.99877413, 13.89264822, 13.58101794,  5.1224079 ,
        5.91408504])}
{'P': tensor([[2.7861e-02, 2.2281e-01, 1.5637e-01,  ..., 2.4397e-06, 2.2604e-05,
         3.8267e-04],
        [4.4046e-03, 8.0334e-02, 2.8130e-01,  ..., 8.0376e-06, 7.5220e-05,
         7.8110e-04],
        [9.4318e-03, 2.9993e-02, 1.7986e-01,  ..., 1.6234e-05, 1.6557e-04,
         1.3104e-03],
        ...,
        [7.6229e-06, 6.2278e-06, 4.9642e-06,  ..., 8.4922e-02, 1.9636e-03,
         1.4475e-04],
        [4.6452e-05, 3.4115e-05, 2.8131e-05,  ..., 2.7949e-04, 3.5590e-02,
         1.8270e-04],
        [3.1004e-03, 3.4140e-03, 2.7251e-03,  ..., 4.0597e-05, 6.8969e-05,
         6.6525e-02]]), 'X': tensor([[  5.6022,  -1.9833,   8.3081],
        [  6.0054,  -1.0902,   8.3079],
        [  6.1222,  -0.7261,   8.8204],
        [  5.8889,  -1.3916,   8.8993],
        [  6.2242,  -0.3171,   7.4978],
        [  5.6672,  -1.8631,   8.1020],
        [  2.6613,   7.0021,   4.5783],
        [ -7.5401,   4.6718,   9.0033],
        [ -6.4844,  -8.0230,   8.5209],
        [ -5.1046,  -9.1610,   2.3418],
        [  5.8836,   3.4315,  -4.0327],
        [  7.0642,  11.6104,  -9.8316],
        [  6.2905,   1.9366,   2.1169],
        [ 11.3691,   6.4760,   1.0787],
        [  3.2701,   6.6338,   9.9209],
        [  5.6260,  -1.9401,   3.0259],
        [ 11.3367,  -4.4806,   0.4643],
        [  1.3403,  -4.8011,  -1.1698],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -6.8618,  -7.6322,   3.7027],
        [  0.2732,  -4.7619,  -1.1577],
        [  5.6093,  -1.9705,   6.4807],
        [  2.2437,  -4.6489,   6.1430],
        [  1.1994,   7.6035,   2.4295],
        [ -6.1914,  -8.2997,   8.5380],
        [  3.3094,  13.4969,  -3.6507],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.2687,  -2.9794,   0.6436],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.6414, -10.0303,   7.8525],
        [ -8.4267,  -5.3697,   2.4172],
        [  8.5198,  10.4091,  -8.6486],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.0750,  -6.0072,  -5.8267],
        [  1.3705,  13.9723,   7.9266],
        [  1.4306,  13.9620,  -0.2075],
        [  7.8425,  -8.7202,   6.4530],
        [  1.6408,   7.4591,  -4.2294],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -7.8359,   4.2735,   7.2943],
        [  3.4920,  -4.1366,  -8.3955],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.7420,  -4.4896,  -3.7139],
        [  1.7815,  -4.7481,   4.4289],
        [  5.1472,  -2.6801,   8.1194],
        [  7.5784,  -8.9141,   5.8410],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.2914,   3.8037,   6.0978],
        [  5.9193,  -1.3176,   5.1739],
        [  4.5057,  -3.3894,  -0.8160],
        [ -5.3328,  -8.9992,   7.3440],
        [  5.9912,   3.1453,  -3.9989],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.3860,  -2.3418,   1.0826],
        [  2.3367,   7.1676,  -9.5263],
        [  8.2154,  -8.4262,   8.0171],
        [  3.1892, -10.8220,  -6.8409],
        [  4.5767,   5.5071,   3.9069],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -5.3736,   6.6310,  -1.5174],
        [  6.3211,   1.7017,  -0.2164],
        [  5.5158,  -2.1337,   3.5246],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.5469,  13.7192,  -9.6935],
        [  1.5653,  13.9377,   4.1773],
        [  6.0890,  -0.8382,   2.9166],
        [ -8.7735,   2.5640,  -6.3313],
        [ 10.2712,   8.3707,  -4.9743],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.9756,  -6.6330,  -3.3429],
        [ 11.2209,  -4.7092,   2.6117],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.7916,  -3.9524,  -7.9547],
        [  0.9036,   7.6840,   9.9237],
        [  5.4429,  -2.2529,   9.7306],
        [  2.8602,   6.8905,   5.8779],
        [  6.2895,   0.0405,   5.7652],
        [ -9.3604,   0.4855,  -3.0158],
        [ -8.6681,  -4.8556,   9.2518],
        [  0.6363, -11.0348,   6.1414],
        [ -3.5903,   7.4847,  -8.9112],
        [ -8.9942,  -3.9990,   6.0448],
        [  9.2979,   9.6048,   7.3138],
        [ -7.4807,   4.7463,   2.0616],
        [  1.3670,  -4.7990,   3.2603],
        [  6.0305,   3.0296,   0.1375],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -4.8665,  -9.3204,  -5.2564],
        [  4.4290,  -3.4596,   1.4181],
        [ 11.8951,  -3.1570,   7.0742],
        [ -6.2668,   5.9808,   2.5614],
        [ -2.8162,   7.7059,  -9.6885],
        [  5.8889,  -1.3916,   8.8993],
        [  5.1196,  -2.7160,   4.2705],
        [  6.3587,   1.1596,  -3.6629],
        [ 12.5789,   0.1722,   1.6470],
        [ -7.0948,   5.1940,   9.8171],
        [ -8.9788,   2.0208,  -3.2867],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.1178,   1.5784,   2.1410],
        [  4.1140,  -3.7220,  -0.1273]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.1874e-03, 2.2287e-02,  ..., 1.2990e+00, 9.0307e-01,
         5.3551e-01],
        [9.1874e-03, 2.1453e-04, 1.6262e-02,  ..., 1.3067e+00, 9.1221e-01,
         5.3931e-01],
        [2.2287e-02, 1.6262e-02, 0.0000e+00,  ..., 1.3211e+00, 9.2181e-01,
         5.5551e-01],
        ...,
        [1.2990e+00, 1.3067e+00, 1.3211e+00,  ..., 0.0000e+00, 6.4064e-01,
         8.9846e-01],
        [9.0307e-01, 9.1221e-01, 9.2181e-01,  ..., 6.4064e-01, 2.1453e-04,
         7.8551e-01],
        [5.3551e-01, 5.3931e-01, 5.5551e-01,  ..., 8.9846e-01, 7.8551e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.94293185,  6.10360073,  6.16512856,  6.05113217,  6.2322769 ,
        5.96556248,  7.49076899,  8.87007495, 10.31584102, 10.48721382,
        6.81117802, 13.59056613,  6.58184474, 13.08414689,  7.39599986,
        5.95111804, 12.18998195,  4.98461983,  9.85330646, 10.26327895,
        4.76969364,  5.94535529,  5.16205087,  7.69752262, 10.35464916,
       13.89671314, 12.77402351,  9.73579738,  4.978704  , 11.50791859,
        9.99211663, 13.45124917,  4.91038884, 10.06437598, 14.0393949 ,
       14.03505624, 11.72802491,  7.6374603 ,  9.58072286,  8.92549752,
        5.41348121, 11.63724792,  5.26067019,  5.0713364 ,  5.80312949,
       11.70016217,  8.01009083,  5.27178363, 12.86648771,  6.06416957,
        5.63824971, 10.4606265 ,  6.76661595,  9.66977802, 10.5575601 ,
       10.00794468,  5.87305005,  7.53884104, 11.76830981, 11.28215429,
        7.16058587,  9.09119351, 10.06407839,  8.53501777,  6.54615887,
        5.91408504, 11.80092578, 13.95361098, 14.02532758,  6.1463896 ,
        9.14044984, 13.25017211, 11.9182956 , 11.97957425, 12.16901615,
        9.14044984,  5.47702766,  7.73692436,  5.89074104,  7.46053281,
        6.28961988,  9.37296102,  9.9353921 , 11.05317597,  8.30125201,
        9.84314712, 13.36800073,  8.85939626,  4.98989906,  6.74872981,
        9.60412735,  9.5642993 , 10.51437947,  5.62005629, 12.30695098,
        8.66273171,  8.20436764,  6.05113217,  5.79543038,  6.46356806,
       12.58006038,  8.79284466,  9.20340031,  9.09661212,  9.25336371,
        5.54776934])}
{'P': tensor([[3.3726e-02, 2.0492e-01, 1.4058e-01,  ..., 1.9641e-04, 1.0661e-04,
         9.2223e-06],
        [1.5577e-03, 3.7764e-02, 9.7020e-02,  ..., 3.6783e-04, 1.6043e-04,
         7.2034e-06],
        [2.4034e-03, 1.1624e-02, 6.5976e-02,  ..., 9.6572e-04, 4.3010e-04,
         1.3380e-05],
        ...,
        [7.1726e-05, 6.9909e-05, 8.2456e-05,  ..., 1.1866e-01, 8.2079e-04,
         2.6173e-04],
        [3.8442e-05, 4.6915e-05, 6.4535e-05,  ..., 5.4424e-02, 1.1990e-01,
         5.6687e-05],
        [1.7227e-04, 8.7907e-05, 8.9164e-05,  ..., 3.5153e-06, 7.2398e-06,
         2.1067e-02]]), 'X': tensor([[  5.6093,  -1.9705,   6.4807],
        [  6.1973,  -0.4372,   6.5720],
        [  6.2895,   0.0405,   5.7652],
        [  6.2242,  -0.3171,   7.4978],
        [  6.3569,   0.7591,   6.5011],
        [  6.3549,   0.7096,   6.9198],
        [ 10.0619,   8.6627,  -9.2884],
        [  2.0937,  13.8293,  -2.3533],
        [  3.7237,  -3.9966,  -5.0376],
        [  7.5784,  -8.9141,   5.8410],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.3062,  -5.6008,   0.0287],
        [  3.5939,   6.4029,  -1.8357],
        [  3.1064,  13.5607,   6.7605],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.3376,  -4.6231,   9.8436],
        [ -7.4733,   4.7555,   6.2490],
        [ 12.5224,  -0.4497,   4.8448],
        [  5.3804,  -2.3504,   7.2744],
        [ -8.8745,   2.3097,   3.5600],
        [  1.1350,  14.0104,   0.6321],
        [  5.1196,  -2.7160,   4.2705],
        [ -9.4296,  -1.9458,   4.0833],
        [  0.5850,  -4.7982,   9.6639],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -9.1795,  -3.3623,   8.5259],
        [  3.2159, -10.8166,  -1.0406],
        [  4.0362,  13.2397,  -7.0030],
        [ -3.3116, -10.1539,   7.5919],
        [  2.4697,  -4.5836,   1.4593],
        [  4.8432,   5.1968,   1.5006],
        [ 11.4495,   6.3026,   6.8257],
        [ -1.3940, -10.7772,   6.7835],
        [ -3.1204, -10.2345,   9.2288],
        [ -9.1749,  -3.3803,   6.4629],
        [  4.9957,  -2.8705,   6.5978],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.6050,   2.9451,  -0.7012],
        [  4.0642,  -3.7599,  -3.4565],
        [ -9.4117,   0.1160,   8.2574],
        [  2.1771, -10.9776,  -3.2007],
        [  3.3839,  -4.1966,   8.3574],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.2345,  -3.1345,   3.9715],
        [  3.3839,  -4.1966,   8.3574],
        [  6.0467,  -9.8369,  -6.5816],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.7239,   4.4298,   3.1216],
        [  5.1203,  -2.7151,   6.2110],
        [ -2.2188,   7.8218,   8.6064],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -2.6735, -10.4062,  -1.5911],
        [  3.6469,   6.3624,  -1.8414],
        [  6.1737,   2.5305,   3.9049],
        [  5.0277,   4.9589,   7.0175],
        [ -4.2040,   7.2486,   2.9875],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.1426,   3.8048,   9.3542],
        [  6.7792,  11.8084,   1.6251],
        [  4.5814, -10.4485,   2.5496],
        [ 11.3118,   6.5957,  -4.5667],
        [  2.5636, -10.9293,   9.6519],
        [  2.4572,   7.1085,   7.7376],
        [ -9.4714,  -1.3301,  -3.1998],
        [  1.8719,   7.3713,  -3.7465],
        [ 12.5224,  -0.4497,   4.8448],
        [ -8.0087,   4.0173,   9.1091],
        [ -3.9700,   7.3453,  -4.1486],
        [  9.3836,   9.5075,  -5.9278],
        [ 11.5966,   5.9666,   6.0083],
        [  5.4867,  12.5816,   2.9549],
        [ 12.5650,   2.0132,   4.6272],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.3460,   1.4252,   1.2291],
        [  9.3560,   9.5391,   9.4943],
        [ -8.6675,  -4.8569,  -9.1769],
        [  3.3839,  -4.1966,   8.3574],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.5382,   7.0672,   4.5097],
        [  0.1287, -11.0065,   6.5757],
        [  4.4049,   5.6892,  -4.4394],
        [  3.4140,  -4.1803,  -8.8224],
        [ -8.9636,   2.0648,   8.1946],
        [ -1.8823, -10.6558,  -3.8952],
        [  5.3860,  -2.3418,   1.0826],
        [  0.1497,   7.8334,  -5.4412],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -9.3076,   0.7852,  -6.7010],
        [ -0.8675, -10.8815,   6.8561],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.8093,  -9.4178,   8.4244],
        [ -7.8280,   4.2848,  -7.1955],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.3062,  -5.6008,   0.0287],
        [  3.6703,   6.3443,  -5.4401],
        [ 12.2589,  -1.9312,   9.8657],
        [  7.1697,  11.5343,  -0.7276],
        [  3.6154,  -4.0640,  -7.9698],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.2883,   0.0325,   5.2692],
        [  4.6063,  13.0051,  -3.1456],
        [ -1.9714,   7.8563,  -1.8084],
        [  2.5663,   7.0526,  -6.8769],
        [ -4.4779,  -9.5609,  -2.4505]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0143, 0.0320,  ..., 0.6488, 0.8503, 1.1035],
        [0.0143, 0.0000, 0.0430,  ..., 0.6629, 0.8630, 1.1173],
        [0.0320, 0.0430, 0.0000,  ..., 0.6241, 0.8204, 1.0891],
        ...,
        [0.6488, 0.6629, 0.6241,  ..., 0.0000, 0.3073, 0.5761],
        [0.8503, 0.8630, 0.8204,  ..., 0.3073, 0.0000, 0.7406],
        [1.1035, 1.1173, 1.0891,  ..., 0.5761, 0.7406, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.94535529,  6.21275016,  6.28961988,  6.2322769 ,  6.4020404 ,
        6.39438044, 13.27717982, 13.98691426,  5.46244988, 11.70016217,
        8.83725081, 10.01804223,  7.34250948, 13.91198191,  9.48337063,
        5.1805345 ,  8.8580671 , 12.53047489,  5.87132718,  9.17016857,
       14.05633441,  5.79543038,  9.6282763 ,  4.83371341, 10.52357479,
        9.77588801, 11.2845608 , 13.84126197, 10.68031351,  5.20659892,
        7.10378919, 13.06957425, 10.86693708, 10.69963954,  9.77777346,
        5.76165934, 11.95743558,  9.09502385,  5.5366698 ,  9.41244865,
       11.1913598 ,  5.39098669,  5.16611782,  9.75200848,  5.39098669,
       11.54672755, 10.06437598,  8.90405823,  5.79562691,  8.1303945 ,
        9.562174  , 10.74409834,  7.33350642,  6.67218687,  7.06169391,
        8.37954853, 10.35485331, 12.0495521 ,  8.98765644, 13.61600536,
       11.4088043 , 13.09426142, 11.22596994,  7.52117132,  9.5642993 ,
        7.60529892, 12.53047489,  8.95983586,  8.34949383, 13.35832725,
       13.04156075, 13.72594022, 12.72524615,  5.84533345,  6.50409372,
       13.3614598 ,  9.93553424,  5.39098669,  5.35774573,  7.50917551,
       11.00726483,  7.19513806,  5.39723241,  9.19837792, 10.8207297 ,
        5.87305005,  7.83486943, 10.32715715, 10.39942497,  9.34061938,
       10.91601655,  5.22903117, 11.62158554,  8.92395381,  9.47369628,
       10.01804223,  7.32950385, 12.41012046, 13.58101794,  5.43943791,
       12.32408516,  6.28835101, 13.79676654,  8.09984231,  7.50500064,
       10.5575601 ])}
{'P': tensor([[3.9648e-02, 1.6947e-01, 1.4766e-01,  ..., 6.5259e-03, 1.3538e-04,
         4.6988e-04],
        [1.4472e-03, 4.8845e-02, 5.7472e-01,  ..., 1.5052e-03, 3.9555e-04,
         1.5128e-03],
        [3.1205e-03, 7.7470e-03, 1.1948e-01,  ..., 2.7813e-03, 1.0656e-03,
         4.2211e-03],
        ...,
        [4.4104e-04, 3.7298e-03, 4.6648e-03,  ..., 2.2751e-02, 3.9605e-04,
         8.2656e-05],
        [4.5740e-07, 9.8955e-07, 1.1251e-06,  ..., 7.2647e-08, 5.6567e-03,
         5.1421e-05],
        [4.7409e-04, 4.0752e-04, 4.0716e-04,  ..., 2.2772e-05, 3.3499e-05,
         3.7646e-01]]), 'X': tensor([[  5.6095,  -1.9702,  -7.8233],
        [  6.3301,   0.3638,  -7.5685],
        [  6.3535,   0.6792,  -7.5101],
        [  6.0788,  -0.8710,  -6.8719],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.3505,   0.6235,  -8.5849],
        [ -8.8631,  -4.3728,  -1.1746],
        [  1.2273,  -4.8081,   9.8129],
        [  5.4867,  12.5816,   2.9549],
        [  6.2790,  -0.0259,  -5.4116],
        [ -0.9715, -10.8630,   0.3555],
        [  6.7565,  11.8237,  -4.9882],
        [  3.5939,   6.4029,  -1.8357],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 10.0619,   8.6627,  -9.2884],
        [  2.6614,  -4.5192,  -9.3641],
        [  8.1043,  -8.5164,  -7.9290],
        [ -8.4267,  -5.3697,   2.4172],
        [ -6.4844,  -8.0230,   8.5209],
        [  1.3705,  13.9723,   7.9266],
        [  0.8572,  14.0501,  -8.3364],
        [  5.1292,   4.8183,  -5.7997],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.1206,  -4.7363,   8.7103],
        [  7.3592,  11.3937,   6.2407],
        [  8.5198,  10.4091,  -8.6486],
        [ -8.8154,   2.4611,   2.5198],
        [  6.3200,   0.2690,   1.5900],
        [ -8.3539,  -5.5111,   0.3373],
        [ -3.9508,  -9.8515,   1.7923],
        [ -5.2589,   6.7024,  -5.5809],
        [ -1.2454,   7.9127,  -0.5715],
        [ -5.9938,  -8.4745,   3.9735],
        [ -7.6525,  -6.6534,   5.1721],
        [  3.5965,  -4.0754,  -3.5035],
        [  9.8082,  -6.8396,   4.9615],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.3236,   0.3011,  -4.6668],
        [  2.0449,  13.8402,  -7.1642],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.5927,   0.4199,  -9.4997],
        [  5.1787,  -2.6383,  -4.2533],
        [ -9.0574,   1.7801,   8.8986],
        [  9.1820,   9.7334,  -2.7932],
        [ -2.7572,   7.7194,   5.9580],
        [  5.0506,   4.9277,  -9.5094],
        [  6.9433,  -9.3362,  -0.3421],
        [ -4.7389,   6.9953,   3.7638],
        [ 10.9931,   7.2107,   7.7933],
        [  1.1139,   7.6281,  -2.5680],
        [ -1.9600,   7.8577,   6.9121],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.5974,   7.0363,  -7.7810],
        [ -8.3062,  -5.6008,   0.0287],
        [  6.2770,  -0.0379,   2.4195],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.1658,  -3.4155,  -2.8880],
        [  0.6363, -11.0348,   6.1414],
        [ 10.1798,   8.5002,  -5.8893],
        [  4.7544,  -3.1425,  -1.5315],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.9503,  -4.7172,  -6.2313],
        [ -9.4196,   0.0478,   9.4140],
        [ 12.1588,   4.3364,  -8.0867],
        [  0.6470, -11.0352,  -1.1931],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.6093,  -1.9705,   6.4807],
        [ -1.5408,   7.8978,   5.3470],
        [  8.1528,  10.7452,   9.1493],
        [  6.0031,  12.2958,  -6.5992],
        [  3.1064,  13.5607,   6.7605],
        [  3.5284, -10.7484,   5.3347],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.8275,  -1.5334,   2.0124],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.2328,  -2.0376,  -1.1234],
        [  2.9410,   6.8428,   5.4458],
        [  5.3804,  -2.3504,   7.2744],
        [  2.3889,  -4.6082,  -8.7291],
        [  4.5057,  -3.3894,  -0.8160],
        [ 12.2921,  -1.7902,  -4.0870],
        [  6.8093,  -9.4178,   8.4244],
        [  2.9470,  13.6085,   2.9809],
        [  6.9284,  11.7061,  -0.3732],
        [  0.1252, -11.0062,   7.2564],
        [ -7.4808,   4.7462,   2.0266],
        [  5.3804,  -2.3504,   7.2744],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.2576,   2.1359,  -3.3398],
        [ 12.2589,  -1.9312,   9.8657],
        [  6.9433,  -9.3362,  -0.3421],
        [  6.3460,   1.4252,   1.2291],
        [  9.0129,  -7.7072,   9.1400],
        [  3.3094,  13.4969,  -3.6507],
        [  7.9403,  -8.6455,  -8.7433],
        [  9.5261,  -7.1675,   1.4206],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.1793,   4.7461,   0.3899],
        [  6.9653,  -9.3225,  -3.4727],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.6319,  11.9066,   3.5165],
        [ -3.5173,   7.5091,  -8.9585]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0010e-02, 1.0440e-02,  ..., 1.7850e-01, 9.0693e-01,
         4.2479e-01],
        [1.0010e-02, 0.0000e+00, 4.2906e-04,  ..., 1.7866e-01, 8.9850e-01,
         4.3393e-01],
        [1.0440e-02, 4.2906e-04, 0.0000e+00,  ..., 1.7870e-01, 8.9813e-01,
         4.3431e-01],
        ...,
        [1.7850e-01, 1.7866e-01, 1.7870e-01,  ..., 0.0000e+00, 1.0124e+00,
         5.2610e-01],
        [9.0693e-01, 8.9850e-01, 8.9813e-01,  ..., 1.0124e+00, 0.0000e+00,
         1.1930e+00],
        [4.2479e-01, 4.3393e-01, 4.3431e-01,  ..., 5.2610e-01, 1.1930e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.94541803,  6.34059021,  6.38968357,  6.14086582,  6.19633139,
        6.38105015,  9.88310614,  4.96231532, 13.72594022,  6.27905292,
       10.90638066, 13.61801526,  7.34250948,  8.04718433, 12.1460931 ,
       13.27717982,  5.24461068, 11.75618513,  9.99211663, 10.31584102,
       14.0393949 , 14.07623266,  7.03733775, 10.7167285 ,  4.73784222,
       13.56367857, 13.45124917,  9.15253344,  6.32572323, 10.00794468,
       10.61417181,  8.51928531,  8.01009083, 10.37998078, 10.14044489,
        5.43544626, 11.95743558, 10.74409834,  6.33075682, 13.99048094,
       11.33837396, 12.59969936,  5.81199422,  9.23071698, 13.38091121,
        8.19703512,  7.05627461, 11.63502679,  8.44940661, 13.14690137,
        7.70897675,  8.09842704,  8.95983586,  9.53015376,  7.50035355,
       10.01804223,  6.27715105, 12.11735334,  9.78147542, 11.05317597,
       13.26209545,  5.69913817, 10.32715715,  5.10444097,  9.41970031,
       12.9089559 , 11.05413545, 13.00380624,  5.94535529,  8.04664886,
       13.48809595, 13.6829843 , 13.91198191, 11.31276109,  9.562174  ,
        6.02588517, 11.33837396, 12.4013153 ,  7.44805901,  5.87132718,
        5.19065439,  5.63824971, 12.4217523 , 11.62158554, 13.92390689,
       13.60275211, 11.00694688,  8.859416  ,  5.87132718, 11.72749923,
       10.05558893,  9.14973319,  6.61211224, 12.41012046, 11.63502679,
        6.50409372, 11.85890664, 13.89671314, 11.73848065, 11.92133129,
        9.7751189 ,  7.02496223, 11.63724792, 12.46513646, 13.62896593,
        8.29203356])}
{'P': tensor([[3.5993e-02, 1.5888e-01, 1.3533e-01,  ..., 1.6456e-05, 1.3711e-03,
         1.6727e-05],
        [1.3652e-03, 2.7617e-02, 8.0287e-02,  ..., 1.1479e-05, 9.4878e-04,
         1.4652e-05],
        [1.5559e-03, 5.0996e-03, 5.7908e-02,  ..., 2.0910e-05, 3.7956e-03,
         2.8875e-05],
        ...,
        [6.9889e-04, 3.6094e-04, 1.9146e-04,  ..., 6.7605e-02, 7.5163e-06,
         2.2872e-04],
        [2.6479e-04, 3.3084e-04, 6.3267e-04,  ..., 1.3884e-05, 1.1975e-01,
         2.0693e-05],
        [1.4132e-04, 8.2449e-05, 4.6317e-05,  ..., 3.2646e-01, 2.4899e-06,
         4.3295e-02]]), 'X': tensor([[ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.3838e-02, 2.1635e-02,  ..., 9.8736e-01, 4.5668e-01,
         9.9458e-01],
        [6.3838e-02, 0.0000e+00, 7.7669e-02,  ..., 9.8316e-01, 5.1981e-01,
         9.8504e-01],
        [2.1635e-02, 7.7669e-02, 0.0000e+00,  ..., 1.0067e+00, 4.4258e-01,
         1.0146e+00],
        ...,
        [9.8736e-01, 9.8316e-01, 1.0067e+00,  ..., 0.0000e+00, 1.1885e+00,
         8.3359e-02],
        [4.5668e-01, 5.1981e-01, 4.4258e-01,  ..., 1.1885e+00, 0.0000e+00,
         1.2252e+00],
        [9.9458e-01, 9.8504e-01, 1.0146e+00,  ..., 8.3359e-02, 1.2252e+00,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.94909709,  6.2134961 ,  6.38399789,  6.3949043 ,  6.32572323,
        6.39520397,  4.91652556, 11.05317597,  6.65259775, 13.79762415,
        9.58072286, 11.54071859, 13.55868684,  5.13616224, 11.4424816 ,
        5.88782913,  6.58184474,  8.99276564,  9.6282763 , 12.59969936,
        5.5366698 ,  9.77588801,  6.81084724, 13.59170809, 10.9036226 ,
       13.27994304,  8.24808832,  5.88782913,  9.95552626,  5.04028538,
        6.4020404 , 12.32545637,  4.85995002,  6.2322769 ,  7.80625981,
        6.02779201, 10.84437705, 13.09737053,  5.13616224,  8.92395381,
        4.9160962 ,  9.49511524,  9.20340031,  5.04028538,  5.5513072 ,
       12.3344536 , 11.00726483, 13.14690137,  7.06272256,  5.64865173,
        8.8353674 ,  7.02326515,  9.14973319,  5.9510489 , 12.78327037,
        9.41970031, 10.14044489, 12.10942412,  8.14433542,  7.48538905,
        6.23262289, 12.18045978, 10.69963954, 11.00726483,  9.23071698,
        9.36093562,  7.5092834 ,  7.8575648 ,  6.00663314,  9.36093562,
       12.24467041, 13.02818702, 11.67875855,  9.09119351,  7.17789666,
        8.83725081, 12.32408516,  5.37343173,  9.55720651,  7.23276582,
        5.19065439,  9.0368458 , 11.54071859,  5.84533345, 12.24467041,
        6.5618968 , 12.17451685,  5.9510489 , 10.04916678,  6.81961996,
        6.30647746, 12.35421461,  9.75401449, 11.11038551,  9.25336371,
        6.33517349, 12.10222204,  8.92549752,  5.08910491,  7.37902766,
       10.91601655,  5.19577146, 11.22596994, 10.19036081,  7.06372172,
        9.77777346])}
{'P': tensor([[2.7088e-02, 1.9099e-01, 1.5570e-01,  ..., 4.4950e-04, 2.5566e-04,
         2.0154e-06],
        [2.9864e-03, 2.9452e-02, 1.9029e-01,  ..., 5.4574e-04, 2.3525e-04,
         2.0570e-06],
        [2.1970e-03, 2.5502e-03, 5.6976e-02,  ..., 1.9669e-03, 3.0642e-04,
         4.3620e-06],
        ...,
        [5.2561e-05, 5.0778e-05, 4.9954e-05,  ..., 4.2017e-02, 9.2396e-05,
         1.4291e-04],
        [1.3120e-02, 1.1650e-02, 1.0920e-02,  ..., 4.9865e-04, 1.5711e-01,
         1.1221e-04],
        [4.4361e-05, 3.7335e-05, 2.6042e-05,  ..., 1.0534e-05, 2.1311e-03,
         1.2896e-01]]), 'X': tensor([[  5.6258,  -1.9404,  -6.0715],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.1579,  -0.5961,  -6.5595],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.8602,   6.8905,   5.8779],
        [  0.7275,  14.0667,  -0.9815],
        [ 12.5224,  -0.4497,   4.8448],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.2311,  -2.0445,   1.3545],
        [ -8.8363,   2.4086,  -5.6296],
        [ -7.6650,   4.5092,   0.2520],
        [ -5.2589,   6.7024,  -5.5809],
        [  7.9141,  -8.6656,   9.6948],
        [  8.8091,  -7.9037,   0.3631],
        [  4.5954,  13.0099,  -4.4439],
        [  3.6703,   6.3443,  -5.4401],
        [  3.7121,   6.3117,   0.6738],
        [  5.8323,  -1.5228,   5.1132],
        [  6.2555,   2.1474,   4.1685],
        [  6.0955,   2.8210,  -4.3321],
        [ -9.3419,   0.5972,  -1.2841],
        [  7.5309,  11.2617,   7.2656],
        [  5.3924, -10.1396,  -4.0863],
        [  3.6703,   6.3443,  -5.4401],
        [  3.5965,  -4.0754,  -3.5035],
        [  5.4429,  -2.2529,   9.7306],
        [  6.3031,   1.8477,  -7.1706],
        [ -9.4671,  -0.5553,   0.1117],
        [  0.5850,  -4.7982,   9.6639],
        [  1.9312,  13.8649,   8.3563],
        [  2.0243,   7.3086,   7.5432],
        [ -6.0195,   6.1789,  -2.9496],
        [ 11.4235,   6.3594,   5.1663],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 10.9916,   7.2133,  -0.9179],
        [  4.7156,  -3.1831,  -8.0224],
        [ -1.2454,   7.9127,  -0.5715],
        [ 11.9122,   5.1397,  -1.5498],
        [  1.8766,   7.3694,   1.1887],
        [ 11.4703,  -4.2006,   7.4086],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.6249, -10.7255,   5.6605],
        [  4.5430,   5.5438,  -3.8669],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.3715,  -4.7756,   0.2312],
        [ -9.3739,   0.3982,  -4.2564],
        [  3.7121,   6.3117,   0.6738],
        [  2.1477,   7.2550,   8.9688],
        [ -3.2053, -10.1993,  -3.8736],
        [  8.8474,  10.0868,  -8.8644],
        [  6.1755,   2.5233,  -2.0042],
        [  0.5250, -11.0306,   5.5358],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.5128, -10.9365,   9.9645],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.9945,  -4.8144,  -0.1897],
        [ 11.6019,   5.9540,  -5.1715],
        [ 10.5678,   7.9261,  -0.6100],
        [ -7.4887,   4.7364,   8.9074],
        [  0.4119,  14.1019,  -4.1870],
        [ -9.2508,   1.0572,   2.9978],
        [  2.9461,  -4.4077,  -0.4831],
        [  6.3534,   1.2990,  -7.2942],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.4573,   3.2448,   5.4370],
        [ -9.2707,   0.9661,   4.4863],
        [ -3.5173,   7.5091,  -8.9585],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -7.7239,   4.4298,   3.1216],
        [ -2.6485, -10.4151,   0.8766],
        [  0.2435,   7.8191,  -6.4709],
        [ -3.4986,   7.5152,   9.0712],
        [  2.0937,  13.8293,  -2.3533],
        [  9.9058,  -6.7203,   5.6928],
        [  6.1984,  -0.4327,   1.5438],
        [  5.8617,   3.4853,  -9.8080],
        [  6.3606,   0.9204,   3.2525],
        [  7.2340,  -9.1505,  -5.8321],
        [ -8.1531,  -5.8748,   2.6934],
        [  9.9750,  -6.6338,  -1.2440],
        [  3.7916,  -3.9524,  -7.9547],
        [  2.3200,   7.1756,   1.1938],
        [  1.1350,  14.0104,   0.6321],
        [  1.8719,   7.3713,  -3.7465],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.4541,  -0.3327,   8.8246],
        [  9.4554,   9.4245,   9.8800],
        [ -7.6734,   4.4980,  -5.4248],
        [ -8.3900,   3.3727,   3.1650],
        [ -5.3736,   6.6310,  -1.5174],
        [  6.1575,  12.2046,   3.3257],
        [  9.3624,   9.5318,   0.2779],
        [ -2.4956, -10.4682,   3.5632],
        [  0.1047, -11.0046,   0.5188],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.2701,   6.6338,   9.9209],
        [  3.9913,  -3.8139,  -6.8311],
        [  7.2340,  -9.1505,  -5.8321],
        [ -6.3066,  -8.1936,   9.5822],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.1824,  -4.6647,   1.1985],
        [ -9.1231,  -3.5739,   8.4708]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.9321e-03, 1.1276e-02,  ..., 3.4404e-01, 4.1420e-01,
         1.2993e+00],
        [8.9321e-03, 0.0000e+00, 1.2590e-02,  ..., 3.5055e-01, 4.2274e-01,
         1.3079e+00],
        [1.1276e-02, 1.2590e-02, 0.0000e+00,  ..., 3.5361e-01, 4.1371e-01,
         1.3050e+00],
        ...,
        [3.4404e-01, 3.5055e-01, 3.5361e-01,  ..., 1.0727e-04, 3.8901e-01,
         1.0070e+00],
        [4.1420e-01, 4.2274e-01, 4.1371e-01,  ..., 3.8901e-01, 0.0000e+00,
         9.8002e-01],
        [1.2993e+00, 1.3079e+00, 1.3050e+00,  ..., 1.0070e+00, 9.8002e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.9510489 ,  6.00732193,  6.24187044,  6.18668463,  6.27905292,
        6.14086582,  7.46053281, 14.08549802, 12.53047489,  5.05444288,
       12.40074901,  9.15866607,  8.89301516,  8.51928531, 11.73567759,
       11.83509372, 13.79762415,  7.32950385,  7.32233733,  6.02779201,
        6.61386436,  6.7166297 ,  9.36093562, 13.54774355, 11.48434767,
        7.32950385,  5.43544626,  5.89074104,  6.56833864,  9.48337063,
        4.83371341, 13.99877413,  7.58378655,  8.62631801, 13.0743399 ,
        9.4997987 , 13.1471313 ,  5.68942572,  8.01009083, 12.97370371,
        7.60463349, 12.21532282, 12.19805394, 11.32149259,  7.16747878,
        8.62631801,  4.79003361,  9.38232121,  7.32233733,  7.56617037,
       10.6910768 , 13.41713661,  6.67108662, 11.04313641,  9.73579738,
       11.22142195,  9.09502385,  9.64157461,  4.9160962 , 13.04051744,
       13.20988996,  8.86082124, 14.10796466,  9.3109884 ,  5.30158497,
        6.48486171, 12.17451685,  5.83560009,  9.05843201,  9.32094055,
        8.29203356,  9.94285511,  8.90405823, 10.74655346,  7.822854  ,
        8.28968067, 13.98691426, 11.9702762 ,  6.2134961 ,  6.81961996,
        6.42689705, 11.66453059, 10.04916678, 11.97949273,  5.47702766,
        7.54127425, 14.05633441,  7.60529892,  7.822854  ,  9.45995955,
       13.35013374,  8.89458273,  9.04255655,  8.53501777, 13.66991027,
       13.3607379 , 10.76154256, 11.00508681, 10.6910768 ,  7.39599986,
        5.5205189 , 11.66453059, 10.33958639,  8.14433542,  5.14999799,
        9.79814997])}
{'P': tensor([[2.3920e-02, 1.7374e-01, 1.7000e-01,  ..., 3.7656e-04, 3.2206e-04,
         3.2524e-05],
        [1.5885e-03, 3.1673e-02, 1.7865e-03,  ..., 7.8278e-04, 7.9705e-04,
         3.4572e-05],
        [3.1399e-03, 2.0168e-01, 2.7101e-02,  ..., 4.6757e-04, 4.0915e-04,
         3.3840e-05],
        ...,
        [1.0962e-04, 1.0453e-04, 1.0969e-04,  ..., 9.4057e-02, 1.5164e-04,
         1.4777e-05],
        [2.3632e-05, 2.9235e-05, 2.4464e-05,  ..., 9.8557e-02, 1.5158e-01,
         3.7210e-06],
        [1.0476e-03, 9.8849e-04, 9.7745e-04,  ..., 2.4153e-05, 1.2718e-05,
         1.6767e-01]]), 'X': tensor([[  5.6260,  -1.9401,   3.0259],
        [  6.0890,  -0.8382,   2.9166],
        [  5.7151,  -1.7700,   3.1542],
        [  6.1741,  -0.5329,   3.1913],
        [  6.2770,  -0.0379,   2.4195],
        [  6.2902,   0.0451,   3.7116],
        [  9.0129,  -7.7072,   9.1400],
        [ 10.4758,  -5.9539,   1.3667],
        [  1.9503,  -4.7172,  -6.2313],
        [ 11.3691,   6.4760,   1.0787],
        [  1.7301,   7.4262,   5.0522],
        [ -6.8466,  -7.6488,   1.9648],
        [  2.4356,  -4.5942,  -5.0808],
        [ -4.6393,   7.0461,   1.1624],
        [  6.0305,   3.0296,   0.1375],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.8474,   6.8979,   6.6457],
        [  3.5939,   6.4029,  -1.8357],
        [  0.7685,  -4.8098,  -2.9310],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.1222,  -0.7261,   8.8204],
        [  5.5424,  -2.0885,   7.2487],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.2905,   1.9366,   2.1169],
        [  6.8407,  11.7666,   3.0045],
        [  6.2555,   2.1474,   4.1685],
        [ 12.6059,   0.9597,  -6.3264],
        [ -6.4844,  -8.0230,   8.5209],
        [ -0.2586,   7.8825,  -9.9840],
        [ -3.5173, -10.0623,   4.8280],
        [ -6.3306,   5.9272,   5.6935],
        [ -0.2586,   7.8825,  -9.9840],
        [  0.9809,   7.6642,  -1.4217],
        [  6.2975,   0.0947,   5.1791],
        [ -3.9456,  -9.8542,   0.7466],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.1426,   3.8048,   9.3542],
        [  8.5093,  -8.1764,  -3.2381],
        [  6.3589,   0.8239,  -8.4230],
        [  6.0047,   3.1062,  -6.7914],
        [ -5.0550,   6.8231,  -9.2502],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.3211,   1.7017,  -0.2164],
        [  1.7218,  -4.7576,  -3.3617],
        [ -1.9714,   7.8563,  -1.8084],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -0.8455, -10.8853,  -6.7400],
        [  9.0652,   9.8597,  -3.5949],
        [  6.2036,   2.4030,  -1.2500],
        [ -9.4296,  -1.9458,   4.0833],
        [  5.7218,  -9.9934,   6.5764],
        [ -0.4341, -10.9473,   7.4509],
        [  2.0899,  -4.6870,  -7.4260],
        [  8.5716,  10.3596,   6.8818],
        [ -3.7964,  -9.9294,   7.7147],
        [  3.6154,  -4.0640,  -7.9698],
        [  6.7565,  11.8237,  -4.9882],
        [ 12.2592,   3.9425,   2.0383],
        [  3.9022,  13.2906,   5.5788],
        [  0.1206,  -4.7363,   8.7103],
        [  2.7420,  -4.4896,  -3.7139],
        [ 11.4235,   6.3594,   5.1663],
        [ -2.1817,   7.8274,  -4.6298],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.8994,   6.8675,  -9.5987],
        [ -1.9646,   7.8571,   3.3817],
        [  6.9956,  11.6590,  -7.6876],
        [  5.6248,   3.9984,   8.4933],
        [  9.0424,  -7.6779,  -7.5286],
        [  0.7275,  14.0667,  -0.9815],
        [ -6.7745,  -7.7264,   8.4684],
        [ -5.8389,  -8.6056,  -3.5098],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.1624,  -3.4286,   2.0647],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.1755,   2.5233,  -2.0042],
        [  0.7275,  14.0667,  -0.9815],
        [  6.3534,   1.2990,  -7.2942],
        [  9.5018,  -7.1945,  -1.1892],
        [ 11.3118,   6.5957,  -4.5667],
        [  0.6933,   7.7336,   2.1179],
        [ -4.0409,   7.3169,  -3.5779],
        [  1.6422,   7.4586,  -6.8056],
        [ -5.9938,  -8.4745,   3.9735],
        [ -8.7735,   2.5640,  -6.3313],
        [  1.6308,  13.9254,  -2.0984],
        [ 12.4877,  -0.7222,   8.8347],
        [ -4.4503,   7.1378,   3.3751],
        [ -2.3317,   7.8034,  -3.6274],
        [  0.1047, -11.0046,   0.5188],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.9284,  11.7061,  -0.3732],
        [  2.6451,  -4.5250,  -0.5565],
        [ -7.5401,   4.6718,   9.0033],
        [  4.5767,   5.5071,   3.9069],
        [  3.8111, -10.6787,  -4.6539],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -4.6393,   7.0461,   1.1624],
        [  2.6613,   7.0021,   4.5783],
        [  5.0277,   4.9589,   7.0175],
        [  6.3569,   0.7591,   6.5011],
        [ -7.0948,   5.1940,   9.8171],
        [ -2.0721,   7.8432,   6.6380],
        [  3.9022,  13.2906,   5.5788],
        [  0.9967,  -4.8144,  -9.7370]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0109, 0.0113,  ..., 0.4223, 0.5437, 0.6237],
        [0.0109, 0.0000, 0.0151,  ..., 0.4321, 0.5397, 0.6222],
        [0.0113, 0.0151, 0.0000,  ..., 0.4182, 0.5329, 0.6348],
        ...,
        [0.4223, 0.4321, 0.4182,  ..., 0.0000, 0.6598, 0.9349],
        [0.5437, 0.5397, 0.5329,  ..., 0.6598, 0.0000, 1.1258],
        [0.6237, 0.6222, 0.6348,  ..., 0.9349, 1.1258, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.95111804,  6.1463896 ,  5.98285319,  6.19707944,  6.27715105,
        6.29034958, 11.85890664, 12.0495521 ,  5.10444097, 13.08414689,
        7.62509582, 10.26545792,  5.19986797,  8.43626114,  6.74872981,
       11.72749923,  7.46248947,  7.34250948,  4.87083348,  6.19633139,
        6.16512856,  5.92281142,  6.23262289,  6.58184474, 13.6105595 ,
        6.61386436, 12.64235359, 10.31584102,  7.88677538, 10.65929968,
        8.67227386,  7.88677538,  7.72668467,  6.29821673, 10.61472636,
        7.85919687,  8.98765644, 11.80092578,  6.41202986,  6.76056625,
        8.49161154,  8.49161154,  6.54615887,  5.05963217,  8.09984231,
       10.53368644, 10.91805683, 13.39372426,  6.652742  ,  9.6282763 ,
       11.51557166, 10.95593729,  5.13182506, 13.44593519, 10.63038565,
        5.43943791, 13.61801526, 12.87752052, 13.85158417,  4.73784222,
        5.26067019, 13.0743399 ,  8.12580976, 12.22276278,  7.4544866 ,
        8.09899874, 13.59673406,  6.90115901, 11.86239961, 14.08549802,
       10.27572673, 10.39942497,  7.27340834,  9.78285424,  9.50913849,
        9.68323686,  6.67108662, 14.08549802,  6.48486171, 11.9182956 ,
       13.09426142,  7.76457095,  8.35856487,  7.63726134, 10.37998078,
        9.14044984, 14.02058621, 12.5085995 ,  8.41148722,  8.14433542,
       11.00508681,  9.20340031, 13.60275211,  5.24137909,  8.87007495,
        7.16058587, 11.33837396,  9.69792513,  8.43626114,  7.49076899,
        7.06169391,  6.4020404 ,  8.79284466,  8.11227668, 13.85158417,
        4.91652556])}
{'P': tensor([[2.7457e-02, 1.5569e-01, 1.4900e-01,  ..., 6.4105e-06, 3.2045e-02,
         4.6657e-06],
        [1.0327e-03, 2.5417e-02, 2.8513e-01,  ..., 4.0499e-06, 4.3782e-02,
         3.1772e-06],
        [1.4642e-03, 3.0323e-03, 3.8708e-02,  ..., 5.9276e-06, 6.3415e-02,
         4.6746e-06],
        ...,
        [1.4831e-04, 9.0277e-05, 8.4311e-05,  ..., 2.1558e-02, 6.9211e-05,
         2.4902e-04],
        [1.2266e-03, 2.9071e-03, 3.2340e-03,  ..., 1.0981e-05, 6.6988e-02,
         9.7319e-06],
        [7.6512e-05, 4.6345e-05, 4.3310e-05,  ..., 4.0673e-01, 3.6849e-05,
         1.9455e-02]]), 'X': tensor([[  5.6314,  -1.9302,  -9.6543],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3360,   0.4258,  -9.2509],
        [ -3.3812, -10.1235,   7.4384],
        [ 11.7779,   5.5127,  -2.0699],
        [ -2.1817,   7.8274,  -4.6298],
        [  1.6408,   7.4591,  -4.2294],
        [  4.0642,  -3.7599,  -3.4565],
        [  5.7963,  -1.6018,   1.8856],
        [ -1.5875,   7.8944,  -9.4286],
        [  2.5829,  -4.5466,  -8.4015],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.7845,   3.6652,   0.9175],
        [  5.7802,  -1.6363,  -6.2782],
        [ -3.4488, -10.0934,   2.5195],
        [  0.2480, -11.0153,  -3.0183],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.2939,  -2.4784,  -4.5840],
        [  4.6231,   5.4556,   8.4973],
        [  1.3767, -11.0341,   9.9415],
        [ -7.3470,  -7.0615,   5.0066],
        [  5.0506,   4.9277,  -9.5094],
        [ 11.5966,   5.9666,   6.0083],
        [  2.6451,  -4.5250,  -0.5565],
        [ -5.2589,   6.7024,  -5.5809],
        [ 11.1085,  -4.9204,   0.0736],
        [ -1.3178, -10.7939,   0.4236],
        [ -1.9600,   7.8577,   6.9121],
        [ 12.0772,  -2.6014,  -7.1832],
        [  9.6587,   9.1817,  -0.8212],
        [  1.5246,  -4.7838,  -5.2095],
        [ 11.7162,   5.6728,  -8.9372],
        [ -9.4117,   0.1160,   8.2574],
        [ -7.2515,   5.0195,   4.6434],
        [  4.1140,  -3.7220,  -0.1273],
        [  6.3497,   1.3680,  -1.9076],
        [ -6.4844,  -8.0230,   8.5209],
        [  1.7903,  -4.7467,   0.1168],
        [  2.2860,   7.1916,  -6.1011],
        [  9.0652,   9.8597,  -3.5949],
        [  5.9843,  -9.8679,  -6.1449],
        [  6.2905,   1.9366,   2.1169],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.1519, -10.2387,   9.8191],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.9655,   3.2172,  -7.5953],
        [  6.1730,  -0.5375,  -8.8861],
        [  4.4049,   5.6892,  -4.4394],
        [  5.1196,  -2.7160,   4.2705],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3305,   1.6119,   1.1538],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.1667,  -2.6543,   1.2879],
        [ -6.3252,  -8.1760,  -1.0891],
        [  2.7420,  -4.4896,  -3.7139],
        [  4.8674,  12.8875,   6.7622],
        [ -0.8455, -10.8853,  -6.7400],
        [ -6.2668,   5.9808,   2.5614],
        [  5.7802,  -1.6363,  -6.2782],
        [  9.3624,   9.5318,   0.2779],
        [ -5.1046,  -9.1610,   2.3418],
        [  4.6231,   5.4556,   8.4973],
        [  7.5309,  11.2617,   7.2656],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.4149,  -4.6005,  -4.3339],
        [ 11.9641,  -2.9568,   7.7211],
        [  4.5767,   5.5071,   3.9069],
        [ -5.3382,   6.6533,   7.8054],
        [  6.9284,  11.7061,  -0.3732],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.5240,  -2.1198,   6.2055],
        [  6.7565,  11.8237,  -4.9882],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.1749,  -3.3803,   6.4629],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.3589,  13.7668,  -3.6501],
        [ -2.5161,   7.7699,  -9.3531],
        [ -8.3900,   3.3727,   3.1650],
        [  4.0078,  -3.8018,  -6.2168],
        [ -2.9497, -10.3028,  -7.3479],
        [  2.3261,  13.7749,   5.9121],
        [ 11.0864,   7.0387,  -0.1238],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.7796,  -3.4696,   9.0088],
        [  1.0000,   7.6591,  -4.5011],
        [ -8.9624,  -4.0944,  -4.9622],
        [  5.6202,  -1.9508,   0.7471],
        [  0.8572,  14.0501,  -8.3364],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.8889,  -1.3916,   8.8993],
        [ -1.6086,   7.8928,   4.3385],
        [ -8.8835,  -4.3178,   9.3080],
        [ -7.6525,  -6.6534,   5.1721],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.6319,  11.9066,   3.5165],
        [  9.2979,   9.6048,   7.3138],
        [  6.3301,   0.3638,  -7.5685],
        [  8.2154,  -8.4262,   8.0171],
        [  0.9659,  -4.8144,  -0.9594],
        [ -2.6485, -10.4151,   0.8766],
        [  6.3031,   1.8477,  -7.1706],
        [ -3.9508,  -9.8515,   1.7923]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.9085e-03, 5.7983e-03,  ..., 9.8382e-01, 3.9460e-02,
         1.0484e+00],
        [4.9085e-03, 0.0000e+00, 8.8454e-04,  ..., 9.8872e-01, 4.2410e-02,
         1.0533e+00],
        [5.7983e-03, 8.8454e-04, 0.0000e+00,  ..., 9.8961e-01, 4.2887e-02,
         1.0542e+00],
        ...,
        [9.8382e-01, 9.8872e-01, 9.8961e-01,  ..., 0.0000e+00, 9.6069e-01,
         8.5450e-02],
        [3.9460e-02, 4.2410e-02, 4.2887e-02,  ..., 9.6069e-01, 0.0000e+00,
         1.0235e+00],
        [1.0484e+00, 1.0533e+00, 1.0542e+00,  ..., 8.5450e-02, 1.0235e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.95297459,  6.23262289,  6.25586297,  6.19633139,  6.30647746,
        6.35029274, 10.67322527, 13.00413829,  8.12580976,  7.6374603 ,
        5.5366698 ,  6.01356004,  8.05242591,  5.22903117, 12.76408562,
        6.84796202,  6.00732193, 10.66631784, 11.018084  , 12.17901033,
        5.84533345,  7.15100303, 11.11969815, 10.19036081,  7.05627461,
       13.04156075,  5.24137909,  8.51928531, 12.1494023 , 10.87409005,
        8.09842704, 12.35421461, 13.32645552,  5.0209005 , 13.01728313,
        9.41244865,  8.81930525,  5.54776934,  6.49538841, 10.31584102,
        5.07306274,  7.54620849, 13.39372426, 11.54071859,  6.58184474,
        9.70108409, 11.46176259,  5.23878371,  6.77777606,  6.19633139,
        7.19513806,  5.79543038, 10.5575601 , 12.3489324 ,  6.53250937,
       12.67680439,  5.80860165, 10.33712258,  5.26067019, 13.77604293,
       10.91805683,  8.66273171,  6.00732193, 13.3607379 , 10.48721382,
        7.15100303, 13.54774355,  5.30158497,  5.19577146, 12.32408516,
        7.16058587,  8.53014915, 13.60275211, 10.92597196,  5.91677188,
       13.61801526,  5.83560009,  9.77777346,  9.45995955, 13.96746816,
        8.16714836,  9.04255655,  5.52416368, 10.7167285 , 13.96987806,
       13.13205357,  8.8353674 , 12.27993098,  7.72415048,  9.85330646,
        5.94909709, 14.07623266,  4.91652556,  6.05113217,  8.0550286 ,
        9.87720102, 10.14044489, 10.60226614, 13.62896593, 13.36800073,
        6.34059021, 11.76830981,  4.91038884, 10.74655346,  6.56833864,
       10.61417181])}
{'P': tensor([[2.5010e-02, 1.9657e-01, 1.2612e-01,  ..., 1.7425e-04, 2.9119e-05,
         2.4321e-05],
        [2.0978e-03, 3.4237e-02, 1.2815e-01,  ..., 1.3513e-04, 4.6086e-05,
         3.7709e-05],
        [3.1202e-03, 7.0226e-03, 6.8644e-02,  ..., 1.8946e-04, 1.3411e-04,
         1.2470e-04],
        ...,
        [1.2663e-03, 1.5007e-03, 2.3387e-03,  ..., 2.3276e-02, 2.8768e-06,
         2.2459e-05],
        [4.6342e-05, 3.7034e-05, 3.6772e-05,  ..., 1.4076e-05, 7.0163e-02,
         9.2036e-05],
        [6.8335e-06, 6.5873e-06, 8.6956e-06,  ..., 2.5436e-06, 6.8950e-03,
         4.8765e-02]]), 'X': tensor([[  5.6672,  -1.8631,   8.1020],
        [  6.0054,  -1.0902,   8.3079],
        [  6.2242,  -0.3171,   7.4978],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3266,   0.3292,   9.3502],
        [  6.3583,   1.1729,   8.7835],
        [ -8.3900,   3.3727,   3.1650],
        [  2.4932,   7.0903,  -5.4214],
        [ -9.0961,   1.6529,   3.1553],
        [  2.7121,   6.9743,  -8.4202],
        [ -5.1369,   6.7755,   8.3836],
        [  2.5382,   7.0672,   4.5097],
        [  6.3569,   0.7591,   6.5011],
        [ -0.8455, -10.8853,  -6.7400],
        [  4.7813,  -3.1139,   8.2691],
        [ -2.9497, -10.3028,  -7.3479],
        [ 12.5154,   2.5072,  -9.3274],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.2790,  -0.0259,  -5.4116],
        [ -0.8885,   7.9160,   0.0571],
        [  2.5663,   7.0526,  -6.8769],
        [ -9.4753,  -0.8012,  -5.2050],
        [  2.1504,  -4.6726,   8.7908],
        [  6.0310,  12.2795,  -6.4528],
        [  2.3373,  13.7721,  -0.2442],
        [  5.1890,   4.7318,   0.9515],
        [  5.1861,   4.7361,  -3.7312],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.7845,   3.6652,   0.9175],
        [  3.2366,  -4.2732,   5.1532],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.9074,  -4.8139,   8.1674],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.1196,  -2.7160,   4.2705],
        [  4.1297,  -3.7098,  -3.9838],
        [  1.0093,   7.6567,   2.3526],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.3384,  -2.6161,  -4.7630],
        [  0.9967,  -4.8144,  -9.7370],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.3606,   0.9204,   3.2525],
        [ 11.1420,   6.9331,  -2.4779],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3048,   0.1469,  -9.1691],
        [  6.1372,   2.6726,  -6.7885],
        [  2.0039,   7.3173,  -1.5885],
        [  0.5175,  14.0910,   4.6540],
        [  9.2811,  -7.4336,   5.1771],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -7.4733,   4.7555,   6.2490],
        [  6.2675,  12.1379,  -8.9720],
        [  0.9445,   7.6736,   0.5095],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.0574,   1.7801,   8.8986],
        [ -7.2515,   5.0195,   4.6434],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.2022,   2.4092,   3.0349],
        [ -8.8745,   2.3097,   3.5600],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.1741,  -0.5329,   3.1913],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.1459,   1.4780,  -4.2403],
        [  2.0729,   7.2878,   7.0880],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -8.6675,  -4.8569,  -9.1769],
        [  5.3118,   4.5452,   4.2114],
        [  9.3753,   9.5171,   3.9131],
        [ -8.8745,   2.3097,   3.5600],
        [  2.1379,  -4.6757,   6.4981],
        [  2.5318,  13.7231,   6.3692],
        [ -8.8934,   2.2597,   3.5831],
        [ 11.1469,   6.9237,   1.6160],
        [ -0.2586,   7.8825,  -9.9840],
        [ -6.3306,   5.9272,   5.6935],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.1206,  -4.7363,   8.7103],
        [  3.9663,   6.1018,  -5.4104],
        [  6.1222,  -0.7261,   8.8204],
        [  0.9809,   7.6642,  -1.4217],
        [ -8.9624,  -4.0944,  -4.9622],
        [  4.5503,   5.5359,   7.4735],
        [  5.7845,   3.6652,   0.9175],
        [  6.1372,   2.6726,  -6.7885],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.3167,   6.6022,   6.3554],
        [ -8.8934,   2.2597,   3.5831],
        [ 12.5789,   0.1722,   1.6470],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.9967,  -4.8144,  -9.7370],
        [  7.2340,  -9.1505,  -5.8321],
        [  5.1787,  -2.6383,  -4.2533],
        [ -1.0482, -10.8487,  -5.9066],
        [ -2.8162,   7.7059,  -9.6885],
        [  1.9503,  -4.7172,  -6.2313],
        [  9.7328,   9.0902,  -0.8917],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.8323,  -1.5228,   5.1132],
        [ 11.0373,   7.1299,  -8.5895],
        [  8.5198,  10.4091,  -8.6486],
        [ 11.1085,  -4.9204,   0.0736],
        [ -3.4100,   7.5436,  -7.6980],
        [  9.5261,  -7.1675,   1.4206],
        [ -7.6650,   4.5092,   0.2520],
        [ -0.8569,   7.9156,  -6.5557]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1894e-02, 1.7338e-02,  ..., 4.5055e-01, 8.5900e-01,
         9.5189e-01],
        [1.1894e-02, 2.1453e-04, 2.5664e-02,  ..., 4.6212e-01, 8.7060e-01,
         9.6287e-01],
        [1.7338e-02, 2.5664e-02, 0.0000e+00,  ..., 4.3842e-01, 8.5356e-01,
         9.3785e-01],
        ...,
        [4.5055e-01, 4.6212e-01, 4.3842e-01,  ..., 0.0000e+00, 4.9723e-01,
         5.1233e-01],
        [8.5900e-01, 8.7060e-01, 8.5356e-01,  ..., 4.9723e-01, 0.0000e+00,
         5.6043e-01],
        [9.5189e-01, 9.6287e-01, 9.3785e-01,  ..., 5.1233e-01, 5.6043e-01,
         1.5170e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.96556248,  6.10360073,  6.2322769 ,  6.16512856,  6.33517349,
        6.46560181,  9.04255655,  7.5158489 ,  9.24501829,  7.48311002,
        8.50268322,  7.50917551,  6.4020404 , 10.91805683,  5.70591986,
       10.7167285 , 12.76408562,  8.22728399,  6.27905292,  7.96575782,
        7.50500064,  9.50913849,  5.14371005, 13.68063098, 13.96905756,
        7.02253235,  7.02326515,  8.16714836,  6.84796202,  5.36061974,
       10.46238054,  4.89869205, 12.50996828,  5.79543038,  5.5513072 ,
        7.72291946, 12.3828957 ,  9.69792513,  4.91652556, 10.23946247,
        6.42689705, 13.12299145,  6.16512856,  6.30647746,  6.69389431,
        7.58668144, 14.10045797, 11.89105259,  6.1561646 , 12.27835996,
        8.8580671 , 13.66051884,  7.73151309,  4.78544875,  9.23071698,
        8.81930525, 11.9120321 ,  6.65368551,  9.17016857, 12.27993098,
        6.19707944,  7.71075226,  9.26455795,  7.57685908, 12.18045978,
        9.93553424,  6.99097013, 13.35927527,  9.17016857,  5.14124612,
       13.95472331,  9.17595899, 13.12218013,  7.88677538,  8.67227386,
       10.30972193,  4.73784222,  7.27759233,  6.16512856,  7.72668467,
        9.85330646,  7.16598273,  6.84796202,  6.69389431, 12.91143491,
        7.38844703,  9.17595899, 12.58006038,  5.18277858,  4.91652556,
       11.66453059,  5.81199422, 10.8992524 ,  8.20436764,  5.10444097,
       13.31764181, 12.21532282,  9.9353921 ,  6.02779201, 13.1399181 ,
       13.45124917, 12.1494023 ,  8.27853076, 11.92133129,  8.89301516,
        7.96181288])}
{'P': tensor([[2.1002e-02, 1.5629e-01, 1.4862e-01,  ..., 1.1722e-04, 1.1515e-02,
         9.5231e-04],
        [1.2885e-03, 2.2844e-02, 1.3271e-01,  ..., 1.8816e-04, 1.3785e-02,
         1.6598e-03],
        [1.5639e-03, 4.0641e-03, 3.3601e-02,  ..., 2.7827e-04, 2.4464e-02,
         2.6981e-03],
        ...,
        [1.6038e-04, 1.7391e-04, 1.6702e-04,  ..., 3.2276e-01, 1.1138e-04,
         9.6007e-03],
        [1.6206e-03, 2.3578e-03, 2.7549e-03,  ..., 7.4817e-04, 1.4872e-01,
         1.7698e-02],
        [7.0696e-04, 7.5411e-04, 7.6059e-04,  ..., 3.9105e-02, 1.0099e-03,
         4.4526e-01]]), 'X': tensor([[  5.7151,  -1.7700,   3.1542],
        [  6.0890,  -0.8382,   2.9166],
        [  6.1741,  -0.5329,   3.1913],
        [  6.2902,   0.0451,   3.7116],
        [  6.2770,  -0.0379,   2.4195],
        [  6.3606,   0.9204,   3.2525],
        [ -9.1658,  -3.4155,  -2.8880],
        [  3.2366,  -4.2732,   5.1532],
        [  5.2939,  -2.4784,  -4.5840],
        [ 12.5789,   0.1722,   1.6470],
        [ -2.5683, -10.4433,  -2.9182],
        [  6.0073,   3.0986,   6.7937],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 10.6673,  -5.6649,  -4.1646],
        [  5.1472,  -2.6801,   8.1194],
        [ -5.3328,  -8.9992,   7.3440],
        [ -7.6734,   4.4980,  -5.4248],
        [  0.5960, -11.0335,   3.1144],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.1787,  -2.6383,  -4.2533],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.5154,   2.5072,  -9.3274],
        [ 10.9875,   7.2207,  -8.8549],
        [  9.2692,   9.6370,   7.0746],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.1737,   2.5305,   3.9049],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4309,  -0.0599,  -4.0830],
        [  0.9478,  -4.8144,   0.6371],
        [  6.8407,  11.7666,   3.0045],
        [  2.9260,  -4.4162,   8.3808],
        [  2.8890,  -4.4316,   1.1600],
        [  0.1931,  -4.7491,   7.4385],
        [  6.3119,   1.7797,   9.9186],
        [  0.8700,  -4.8132,  -7.8042],
        [  9.6587,   9.1817,  -0.8212],
        [  0.9478,  -4.8144,   0.6371],
        [ 10.9931,   7.2107,   7.7933],
        [  2.1440, -10.9811,   8.5281],
        [ 12.4837,   2.7510,   0.4449],
        [  4.0587, -10.6112,   6.2273],
        [  2.8890,  -4.4316,   1.1600],
        [  6.0047,   3.1062,  -6.7914],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.3539,  -5.5111,   0.3373],
        [  0.5600, -11.0321,  -0.8660],
        [  2.5382,   7.0672,   4.5097],
        [ -2.0721,   7.8432,   6.6380],
        [ 10.5678,   7.9261,  -0.6100],
        [ -4.8751,  -9.3148,   4.0406],
        [  4.9152,  12.8653,  -1.4459],
        [  2.7420,  -4.4896,  -3.7139],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.1741,  -0.5329,   3.1913],
        [  9.4516,  -7.2501,  -1.8868],
        [ -6.8466,  -7.6488,   1.9648],
        [ -1.5875,   7.8944,  -9.4286],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.6310, -10.9195,   3.8054],
        [  8.2154,  -8.4262,   8.0171],
        [  6.2770,  -0.0379,   2.4195],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.1907,  -3.3180,  -5.5943],
        [  2.0419,  -4.6978,  -5.0983],
        [  0.3715,  -4.7756,   0.2312],
        [  6.2790,  -0.0259,  -5.4116],
        [  2.5010,  -4.5736,  -0.8989],
        [ -1.6275, -10.7221,   3.2376],
        [  5.8979,   3.3958,  -3.8713],
        [  4.9957,  -2.8705,   6.5978],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.5149,  -5.1905,   1.5239],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.0093,   7.6567,   2.3526],
        [  5.4429,  -2.2529,   9.7306],
        [  5.4867,  12.5816,   2.9549],
        [  3.9085,   6.1512,  -2.2560],
        [  1.3767, -11.0341,   9.9415],
        [  1.8719,   7.3713,  -3.7465],
        [ -3.9456,  -9.8542,   0.7466],
        [ -0.5709,   7.9055,   7.6089],
        [  0.6952,  -4.8060,   8.8976],
        [  7.3592,  11.3937,   6.2407],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.6414, -10.0303,   7.8525],
        [  6.9956,  11.6590,  -7.6876],
        [  6.3246,   1.6696,  -0.1544],
        [ -1.5451,   7.8975,   7.6104],
        [  5.1890,   4.7318,   0.9515],
        [  4.2339,  -3.6268,   6.3125],
        [ -3.9858,  -9.8334,  -3.7323],
        [  1.4150,  13.9647,  -4.0915],
        [  5.6202,  -1.9508,   0.7471],
        [  3.3943,   6.5483,   1.0790],
        [  5.7485,  -9.9811,  -3.2355],
        [ -6.3066,  -8.1936,   9.5822],
        [  6.2038,   2.4020,  -3.1692],
        [ 12.2692,  -1.8882,   3.9253],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.0073,   3.0986,   6.7937],
        [  0.3038,   7.8092,   0.4164]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.5115e-02, 1.3283e-02,  ..., 6.7423e-01, 2.5622e-01,
         3.0421e-01],
        [1.5115e-02, 0.0000e+00, 2.2960e-02,  ..., 6.6468e-01, 2.6997e-01,
         3.0095e-01],
        [1.3283e-02, 2.2960e-02, 1.5170e-04,  ..., 6.8657e-01, 2.4737e-01,
         3.1747e-01],
        ...,
        [6.7423e-01, 6.6468e-01, 6.8657e-01,  ..., 1.5170e-04, 9.1259e-01,
         4.2160e-01],
        [2.5622e-01, 2.6997e-01, 2.4737e-01,  ..., 9.1259e-01, 0.0000e+00,
         5.0628e-01],
        [3.0421e-01, 3.0095e-01, 3.1747e-01,  ..., 4.2160e-01, 5.0628e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 5.98285319,  6.1463896 ,  6.19707944,  6.29034958,  6.27715105,
        6.42689705,  9.78147542,  5.36061974,  5.84533345, 12.58006038,
       10.75443106,  6.7593917 ,  9.78286733, 12.07818871,  5.80312949,
       10.4606265 ,  8.89458273, 11.04953767, 10.51340829,  5.81199422,
       13.77222752, 12.76408562, 13.14777433, 13.37121923, 10.60226614,
       10.51437947, 10.39942497,  6.67218687, 11.9182956 , 13.36092624,
        9.43112957,  4.90676242, 13.6105595 ,  5.29755487,  5.29009836,
        4.75303127,  6.55800267,  4.89121805, 13.32645552,  4.90676242,
       13.14690137, 11.18839169, 12.78327037, 11.36089355,  5.29009836,
        6.76056625, 10.8992524 , 10.00794468, 11.04629471,  7.50917551,
        8.11227668, 13.20988996, 10.51340829, 13.77222752,  5.26067019,
       12.00995898,  9.64157461,  6.19707944, 11.9120321 , 10.26545792,
        8.05242591,  9.73579738, 11.23201619, 11.76830981,  6.27715105,
        9.69792513,  9.7712321 ,  5.1224079 ,  4.79003361,  6.27905292,
        5.21280156, 10.84493732,  6.80559669,  5.76165934,  5.62005629,
        9.97220799,  7.26074925, 10.04916678,  7.72291946,  5.89074104,
       13.72594022,  7.28793663, 11.11969815,  7.60529892, 10.61472636,
        7.92607342,  4.85603886, 13.56367857,  9.20340031, 11.50791859,
       13.59673406,  6.54127431,  8.04718433,  7.02253235,  5.5748624 ,
       10.61047596, 14.03618649,  5.94909709,  7.37576154, 11.51811603,
       10.33958639,  6.65259775, 12.41366754,  7.96181288,  6.7593917 ,
        7.81509631])}
{'P': tensor([[3.3776e-02, 1.7577e-01, 1.5041e-01,  ..., 5.1574e-05, 6.5446e-03,
         1.0183e-05],
        [1.7628e-03, 5.1683e-02, 3.0674e-03,  ..., 1.3497e-04, 9.9799e-03,
         2.6674e-05],
        [2.6917e-03, 2.0638e-01, 3.3574e-02,  ..., 7.1113e-05, 9.8120e-03,
         1.6549e-05],
        ...,
        [3.9749e-05, 2.5748e-05, 3.4279e-05,  ..., 3.9923e-02, 2.5850e-05,
         3.0687e-05],
        [6.5124e-04, 1.6799e-03, 1.1175e-03,  ..., 4.8160e-04, 5.8218e-02,
         4.6548e-04],
        [1.5568e-05, 1.6629e-05, 1.9109e-05,  ..., 5.1988e-03, 9.3115e-05,
         2.4733e-01]]), 'X': tensor([[  5.7784,  -1.6401,   9.5587],
        [  6.3266,   0.3292,   9.3502],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3583,   1.1729,   8.7835],
        [  6.3119,   1.7797,   9.9186],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -9.2974,   0.8371,   2.4852],
        [  6.1755,   2.5233,  -2.0042],
        [ -4.8665,  -9.3204,  -5.2564],
        [  0.9945,  -4.8144,  -0.1897],
        [  2.3490,  -4.6199,  -4.2489],
        [ -4.6942,  -9.4299,  -9.1846],
        [  1.6408,   7.4591,  -4.2294],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.2242,  -0.3171,   7.4978],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.1737,   2.5305,   3.9049],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.3453,  -1.5462,   7.0212],
        [  2.2182,  13.8007,   5.1889],
        [  1.5246,  -4.7838,  -5.2095],
        [  4.5503,   5.5359,   7.4735],
        [  0.1047, -11.0046,   0.5188],
        [ -9.4157,   0.0821,  -5.0025],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.5382,   7.0672,   4.5097],
        [  0.8940, -11.0402,  -0.9783],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.1178,   1.5784,   2.1410],
        [  5.3118,   4.5452,   4.2114],
        [  4.0362,  13.2397,  -7.0030],
        [ -8.7409,  -4.6837,  -6.5840],
        [  9.6587,   9.1817,  -0.8212],
        [  2.1771, -10.9776,  -3.2007],
        [  6.2022,   2.4092,   3.0349],
        [  0.1041, -11.0045,   2.4311],
        [ 11.6165,  -3.8715,   6.4594],
        [  3.9022,  13.2906,   5.5788],
        [ -1.6086,   7.8928,   4.3385],
        [ -0.0254, -10.9932,   3.2200],
        [  1.6308,  13.9254,  -2.0984],
        [  5.8930,  12.3592,  -8.9201],
        [  2.0449,  13.8402,  -7.1642],
        [ -6.4368,   5.8356,   3.3014],
        [  2.8474,   6.8979,   6.6457],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.2905,   1.9366,   2.1169],
        [  2.3200,   7.1756,   1.1938],
        [  5.3804,  -2.3504,   7.2744],
        [  6.3266,   0.3292,   9.3502],
        [ -9.4671,  -0.5553,   0.1117],
        [ -1.6220,   7.8917,   0.0315],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.0719, -10.8449,  -4.1673],
        [  5.1890,   4.7318,   0.9515],
        [  1.7232,  -4.7574,   3.1270],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3460,   1.4252,   1.2291],
        [  2.9470,  13.6085,   2.9809],
        [ -9.4430,  -0.1897,   4.8383],
        [  2.6970,   6.9826,   9.7350],
        [ -9.4697,  -0.6157,  -6.8472],
        [  7.7750,  11.0665,   2.8364],
        [  5.8836,   3.4315,  -4.0327],
        [  6.3569,   0.7591,   6.5011],
        [  4.0078,  -3.8018,  -6.2168],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.4572,   7.1085,   7.7376],
        [ -7.9667,   4.0814,  -0.6397],
        [  1.3767, -11.0341,   9.9415],
        [ -9.4203,  -2.0340,   2.5856],
        [  2.7295,  13.6703,  -8.9328],
        [ -4.7642,  -9.3860,  -5.1133],
        [  9.6614,   9.1784,  -1.2087],
        [ -2.0445,   7.8469,  -0.5857],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.6523,   3.9443,  -2.9449],
        [ -6.3306,   5.9272,   5.6935],
        [  4.9452, -10.3187,  -7.8621],
        [  3.2647, -10.8066,  -0.3768],
        [  1.1006,   7.6318,  -2.4877],
        [  3.6469,   6.3624,  -1.8414],
        [  2.4572,   7.1085,   7.7376],
        [  6.3542,   0.6934,  -3.3840],
        [ -0.4341, -10.9473,   7.4509],
        [  3.9770, -10.6341,   8.0783],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3551,   0.7149,   1.3344],
        [  6.1973,  -0.4372,   6.5720],
        [ 11.3118,   6.5957,  -4.5667],
        [ -4.5778,  -9.5013,  -5.0667],
        [  7.9141,  -8.6656,   9.6948],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.3411,  -2.6004,  -2.7062],
        [  2.7685, -10.8982,  -6.8928],
        [ -6.2200,  -8.2737,   5.5079],
        [ 12.5123,  -0.5339,  -1.0728],
        [  1.6422,   7.4586,  -6.8056],
        [ -5.3328,  -8.9992,   7.3440],
        [ 12.1588,   4.3364,  -8.0867],
        [ -8.1905,   3.7253,   4.1175],
        [  6.1752,   2.5242,   3.5874],
        [ -3.0004,   7.6607,  -7.2640]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0035, 0.0082,  ..., 0.7697, 0.2573, 1.0523],
        [0.0035, 0.0000, 0.0095,  ..., 0.7729, 0.2602, 1.0556],
        [0.0082, 0.0095, 0.0000,  ..., 0.7722, 0.2513, 1.0481],
        ...,
        [0.7697, 0.7729, 0.7722,  ..., 0.0000, 0.7030, 0.7856],
        [0.2573, 0.2602, 0.2513,  ..., 0.7030, 0.0000, 0.8104],
        [1.0523, 1.0556, 1.0481,  ..., 0.7856, 0.8104, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.00663314,  6.33517349,  6.16512856,  6.46560181,  6.55800267,
        6.05113217,  9.93553424,  9.3349858 ,  6.67108662, 10.51437947,
        4.9160962 ,  5.18277858, 10.53368644,  7.6374603 ,  5.5205189 ,
        6.2322769 , 10.4977132 ,  6.67218687, 12.86648771, 12.44177641,
       13.97779839,  5.0209005 ,  7.16598273, 11.00508681,  9.41605684,
        8.23525433,  7.50917551, 11.07637531, 10.33958639,  9.25336371,
        6.99097013, 13.84126197,  9.91668182, 13.32645552, 11.1913598 ,
        6.65368551, 11.0050379 , 12.24467041, 13.85158417,  8.0550286 ,
       10.99326577, 14.02058621, 13.6922425 , 13.99048094,  8.68832398,
        7.46248947, 12.12641969, 10.27716434,  6.58184474,  7.54127425,
        5.87132718,  6.33517349,  9.48337063,  8.05668609, 13.25017211,
       11.27160035,  7.02253235,  5.05989547,  6.12886024,  6.50409372,
       13.92390689,  9.44486609,  7.48538905,  9.48969978, 13.5247215 ,
        6.81117802,  6.4020404 ,  5.52416368, 10.06407839, 11.63765273,
        7.52117132,  8.95133961, 11.11969815,  9.63742669, 13.94008948,
       10.52587243, 13.32613445,  8.10886021,  8.85572292,  6.89245352,
        8.67227386, 11.4424816 , 11.28895489,  7.71075226,  7.33350642,
        7.52117132,  6.39187285, 10.95593729, 11.35345345, 13.36092624,
        6.39520397,  6.21275016, 13.09426142, 10.54658597, 11.73567759,
       10.06286074,  9.69629   , 11.24434374, 10.35093485, 12.52372629,
        7.63726134, 10.4606265 , 12.9089559 ,  8.99791316,  6.67121748,
        8.22728399])}
{'P': tensor([[2.1982e-02, 1.5645e-01, 1.4203e-01,  ..., 2.4303e-04, 2.4313e-06,
         8.2827e-04],
        [1.5507e-03, 2.8844e-02, 8.9211e-02,  ..., 2.2765e-04, 2.8733e-06,
         9.1183e-04],
        [1.8386e-03, 7.7470e-03, 4.1077e-02,  ..., 3.1229e-04, 4.6613e-06,
         1.6237e-03],
        ...,
        [2.5445e-02, 1.6262e-02, 1.6698e-02,  ..., 8.6694e-02, 8.1172e-05,
         3.1222e-03],
        [9.9333e-05, 6.2750e-05, 6.5344e-05,  ..., 2.2737e-03, 1.3593e-01,
         1.4732e-04],
        [4.3517e-04, 5.6818e-04, 7.6143e-04,  ..., 2.6025e-04, 5.2765e-05,
         8.4753e-02]]), 'X': tensor([[ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0129, 0.0126,  ..., 0.1991, 1.2429, 0.5936],
        [0.0129, 0.0000, 0.0210,  ..., 0.2119, 1.2555, 0.6039],
        [0.0126, 0.0210, 0.0000,  ..., 0.1950, 1.2408, 0.5829],
        ...,
        [0.1991, 0.2119, 0.1950,  ..., 0.0000, 1.0478, 0.4535],
        [1.2429, 1.2555, 1.2408,  ..., 1.0478, 0.0000, 0.9585],
        [0.5936, 0.6039, 0.5829,  ..., 0.4535, 0.9585, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.00732193,  6.18668463,  6.24187044,  6.14086582,  6.383897  ,
        6.36004393,  9.32094055,  6.24187044,  9.55593374, 10.9036226 ,
        6.59176018, 11.00694688,  6.10360073,  8.22728399,  5.53255882,
        4.75303127,  5.57341623,  9.97822473, 11.1341798 , 13.93790235,
        5.54776934,  8.83341608, 13.89671314,  7.50564223, 13.09426142,
        6.41202986, 10.90638066,  9.38232121, 13.79762415, 12.32545637,
       10.71162681, 12.53047489, 11.70168755,  8.05242591,  9.79039371,
        5.79391825, 12.18045978, 13.54774355,  7.58668144, 11.77124373,
       13.03002338, 13.744387  , 11.24434374,  7.62232307,  8.66273171,
        9.64157461,  6.80370254,  8.20436764, 10.27716434,  9.17905491,
       11.83570132, 11.97238624,  5.79391825, 10.36844813, 12.77402351,
        9.26455795, 10.48840476, 12.79126785, 12.22503754, 11.74901778,
        7.37576154,  9.09119351,  9.37296102,  6.99097013,  8.859416  ,
        5.16077764,  7.76457095,  5.13182506, 11.63724792, 11.70168755,
       13.58101794,  7.70897675,  6.81117802, 11.04953767, 12.00995898,
        6.02588517, 12.05395056,  6.53290663,  4.78258092,  9.17016857,
        6.61211224,  7.6374603 ,  6.7166297 , 12.91143491, 10.01804223,
        9.64157461, 14.10045797,  8.62260327,  9.05843201,  6.61211224,
        6.49538841,  7.32950385,  7.5403822 ,  5.47866221,  9.7712321 ,
       10.61047596,  7.46248947, 12.27993098, 13.66896739,  9.51074763,
        8.04664886, 12.71843462, 10.63038565,  4.87083348,  9.96391045,
        6.65368551])}
{'P': tensor([[2.6157e-02, 1.7385e-01, 1.4035e-01,  ..., 4.8137e-03, 4.1958e-06,
         3.7414e-05],
        [1.2221e-03, 2.3901e-02, 4.7062e-02,  ..., 2.4102e-03, 4.0327e-06,
         1.8580e-05],
        [2.7599e-03, 1.4222e-02, 6.3007e-02,  ..., 3.8377e-03, 8.8029e-06,
         4.4328e-05],
        ...,
        [1.1547e-02, 2.6635e-02, 1.9967e-02,  ..., 9.8183e-02, 3.5662e-05,
         6.1949e-05],
        [1.4644e-04, 1.0312e-04, 7.1144e-05,  ..., 3.7210e-04, 1.3492e-01,
         2.0534e-04],
        [2.4075e-03, 1.5396e-03, 1.6337e-03,  ..., 4.7310e-04, 2.9232e-07,
         1.7624e-02]]), 'X': tensor([[  5.7963,  -1.6018,   1.8856],
        [  6.1984,  -0.4327,   1.5438],
        [  6.2770,  -0.0379,   2.4195],
        [  6.3200,   0.2690,   1.5900],
        [  6.3551,   0.7149,   1.3344],
        [  5.8275,  -1.5334,   2.0124],
        [ 11.1469,   6.9237,   1.6160],
        [  6.3266,   0.3292,   9.3502],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.4739,  -1.2497,   0.5483],
        [ -4.2040,   7.2486,   2.9875],
        [  4.0642,  -3.7599,  -3.4565],
        [  5.8979,   3.3958,  -3.8713],
        [ -8.0750,  -6.0072,  -5.8267],
        [ 12.5763,   0.1333,  -9.2650],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.0054,  -1.0902,   8.3079],
        [  1.5246,  -4.7838,  -5.2095],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.1989,  -9.7591,   0.8289],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.3411,  -2.6004,  -2.7062],
        [  0.9392,  -4.8143,  -3.4649],
        [ -8.5900,   2.9767,   7.4811],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.6022,  -1.9833,   8.3081],
        [  1.0000,   7.6591,  -4.5011],
        [ -7.9298,   4.1367,  -3.9773],
        [ -2.6069, -10.4298,  -6.1995],
        [  9.5203,   9.3483,   5.8781],
        [  2.2182,  13.8007,   5.1889],
        [ -7.6293,  -6.6859,  -5.5161],
        [  6.3200,   0.2690,   1.5900],
        [ 11.2517,  -4.6495,   2.9800],
        [ -9.3076,   0.7852,  -6.7010],
        [ 10.2712,   8.3707,  -4.9743],
        [  1.1994,   7.6035,   2.4295],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.5650,   2.0132,   4.6272],
        [ -1.9600,   7.8577,   6.9121],
        [  9.3607,   9.5337,  -4.5412],
        [ -8.8540,  -4.3971,  -5.2144],
        [  4.1140,  -3.7220,  -0.1273],
        [ -3.5520,   7.4976,   9.1522],
        [  1.4489,   7.5256,   4.4529],
        [  1.1006,   7.6318,  -2.4877],
        [  7.0171,  11.6439,  -5.0178],
        [  4.4039,   5.6903,   8.7366],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.6672,  -1.8631,   8.1020],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.3705,  13.9723,   7.9266],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.1890,   4.7318,   0.9515],
        [  4.6231,   5.4556,   8.4973],
        [ -3.0008, -10.2827,  -8.9480],
        [  9.3836,   9.5075,  -5.9278],
        [ -0.3432,   7.8900,   7.1805],
        [  7.0642,  11.6104,  -9.8316],
        [  1.4150,  13.9647,  -4.0915],
        [ -8.3653,   3.4184,   9.3454],
        [ 10.1798,   8.5002,  -5.8893],
        [ -7.5603,   4.6459,   8.2002],
        [  5.9526,   3.2525,  -4.3821],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.3589,   0.8239,  -8.4230],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.3305,   1.6119,   1.1538],
        [ -8.5900,   2.9767,   7.4811],
        [ -1.5451,   7.8975,   7.6104],
        [  5.9026,   3.3837,  -0.6573],
        [ -6.4002,  -8.1048,  -0.9108],
        [  3.7992,  -3.9474,  -8.9019],
        [ -2.0445,   7.8469,  -0.5857],
        [ -9.4309,  -0.0599,  -4.0830],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.5325,  -3.3643,   6.7699],
        [  0.9809,   7.6642,  -1.4217],
        [  5.7218,  -9.9934,   6.5764],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.9433,  -9.3362,  -0.3421],
        [  1.0391,  -4.8141,  -0.0768],
        [  3.3094,  13.4969,  -3.6507],
        [  3.2159, -10.8166,  -1.0406],
        [  5.3860,  -2.3418,   1.0826],
        [ -9.1960,  -3.2964,   1.6921],
        [ -0.0254, -10.9932,   3.2200],
        [  3.2647, -10.8066,  -0.3768],
        [  0.9392,  -4.8143,  -3.4649],
        [  3.2647, -10.8066,  -0.3768],
        [  2.9470,  13.6085,   2.9809],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.6426,  -4.9136,   9.4941],
        [  2.3889,  -4.6082,  -8.7291],
        [  0.6363, -11.0348,   6.1414],
        [  6.7565,  11.8237,  -4.9882],
        [  3.8111, -10.6787,  -4.6539],
        [ 11.0891,  -4.9558,  -2.3178],
        [  4.9337,  12.8566,  -3.6582],
        [  5.9548,  -1.2273,  -2.6520],
        [ -8.6675,  -4.8569,  -9.1769],
        [  4.0587, -10.6112,   6.2273]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0180, 0.0545,  ..., 0.3067, 1.2524, 0.6958],
        [0.0180, 0.0000, 0.0689,  ..., 0.2910, 1.2507, 0.7124],
        [0.0545, 0.0689, 0.0000,  ..., 0.3599, 1.2933, 0.6740],
        ...,
        [0.3067, 0.2910, 0.3599,  ..., 0.0000, 1.1079, 0.9201],
        [1.2524, 1.2507, 1.2933,  ..., 1.1079, 0.0000, 1.2478],
        [0.6958, 0.7124, 0.6740,  ..., 0.9201, 1.2478, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.01356004,  6.2134961 ,  6.27715105,  6.32572323,  6.39520397,
        6.02588517, 13.12218013,  6.33517349,  4.75303127,  7.98469166,
        9.55593374,  8.37954853,  5.5366698 ,  6.80559669, 10.06437598,
       12.5769659 ,  9.94285511,  6.10360073,  5.0209005 , 10.1371552 ,
       11.56146873,  5.94535529,  9.69629   ,  4.90505566,  9.09119351,
        7.9337647 ,  5.94293185,  7.72415048,  8.94395628, 10.7506423 ,
       13.3426541 , 13.97779839, 10.14436912,  6.32572323, 12.17451685,
        9.34061938, 13.25017211,  7.69752262, 12.889695  , 12.72524615,
        8.09842704, 13.36092624,  9.88571713,  5.54776934,  8.29640613,
        7.66377554,  7.71075226, 13.59480906,  7.19534146, 12.76408562,
        5.96556248,  9.38232121, 14.0393949 ,  5.94541803,  7.02253235,
        7.15100303, 10.71162681, 13.35832725,  7.89745107, 13.59056613,
       14.03618649,  9.0368458 , 13.26209545,  8.8737556 ,  6.78325151,
        5.5205189 ,  6.41202986, 10.4606265 ,  5.16611782,  9.50654967,
        6.53250937,  9.09119351,  8.04718433,  6.80370254, 10.32715715,
        5.47866221,  8.10886021,  9.43112957,  5.35774573,  5.41348121,
        5.64464807,  7.72668467, 11.51557166, 13.01728313, 11.63502679,
        4.92497514, 13.89671314, 11.2845608 ,  5.87305005,  9.76896904,
       10.99326577, 11.28895489,  4.90505566, 11.28895489, 13.92390689,
       10.06850677,  9.94173708,  5.19065439, 11.05317597, 13.61801526,
       11.33837396, 12.1460931 , 13.77075174,  6.07992623,  9.93553424,
       11.36089355])}
{'P': tensor([[3.1043e-02, 1.9133e-01, 1.8272e-01,  ..., 2.4014e-04, 2.0789e-03,
         6.5808e-05],
        [1.6396e-03, 2.8989e-02, 5.7081e-02,  ..., 3.6723e-04, 2.6781e-03,
         7.9434e-05],
        [2.6426e-03, 1.3651e-02, 6.0474e-02,  ..., 7.6064e-04, 7.6931e-03,
         1.9042e-04],
        ...,
        [1.7177e-04, 1.6487e-04, 1.4710e-04,  ..., 1.9092e-01, 1.2507e-05,
         6.5781e-02],
        [1.1076e-05, 1.7439e-05, 2.2562e-05,  ..., 3.4772e-04, 1.9970e-02,
         7.4849e-05],
        [5.1184e-05, 3.7047e-05, 3.4677e-05,  ..., 9.9039e-05, 1.0082e-06,
         3.5259e-02]]), 'X': tensor([[  5.8275,  -1.5334,   2.0124],
        [  6.1984,  -0.4327,   1.5438],
        [  6.2770,  -0.0379,   2.4195],
        [  6.3200,   0.2690,   1.5900],
        [  6.3551,   0.7149,   1.3344],
        [  6.3460,   1.4252,   1.2291],
        [  0.2435,   7.8191,  -6.4709],
        [  3.2692,  13.5098,  -4.9032],
        [ 12.1379,   4.4125,  -3.8833],
        [  4.3377,   5.7570,   1.3712],
        [ -5.2589,   6.7024,  -5.5809],
        [ -5.8389,  -8.6056,  -3.5098],
        [  1.7500,   7.4187,   4.4237],
        [  7.8982,  -8.6778,   0.0239],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.6634,   5.8050,   9.1590],
        [  2.3376,  -4.6231,   9.8436],
        [  0.9809,   7.6642,  -1.4217],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.2974,   0.8371,   2.4852],
        [  6.2975,   0.0947,   5.1791],
        [  4.1348,   5.9514,  -0.8856],
        [  5.4867,  12.5816,   2.9549],
        [ -8.9282,   2.1648,   8.4894],
        [  5.2911,  -2.4824,  -3.6877],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.8201,   3.5841,  -8.0609],
        [  1.3767, -11.0341,   9.9415],
        [ -8.0588,  -6.0341,  -2.3006],
        [  0.5850,  -4.7982,   9.6639],
        [ -0.8885,   7.9160,   0.0571],
        [  0.8940, -11.0402,  -0.9783],
        [  6.3122,  -9.6994,  -6.3282],
        [  3.9663,   6.1018,  -5.4104],
        [ -5.1369,   6.7755,   8.3836],
        [ -9.3384,  -2.6161,  -4.7630],
        [  6.2242,  -0.3171,   7.4978],
        [  6.1579,  -0.5961,  -6.5595],
        [  0.6396,  14.0772,  -6.3786],
        [ -9.3616,  -2.4750,   3.7765],
        [ -5.7347,  -8.6908,   9.2878],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.7485,  -9.9811,  -3.2355],
        [  2.2644,  -4.6434,   9.7508],
        [  9.8082,  -6.8396,   4.9615],
        [  6.1755,   2.5233,  -2.0042],
        [  2.9260,  -4.4162,   8.3808],
        [  1.4306,  13.9620,  -0.2075],
        [ -7.5401,   4.6718,   9.0033],
        [  5.0348,  -2.8229,   3.5436],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.4732,  -2.2041,  -4.3830],
        [  8.0877,  -8.5297,  -1.1050],
        [ 12.2921,  -1.7902,  -4.0870],
        [  6.6319,  11.9066,   3.5165],
        [  5.2188,  -2.5839,   2.1833],
        [  6.1575,  12.2046,   3.3257],
        [ -9.3087,   0.7792,  -7.5587],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.3118,   6.5957,  -4.5667],
        [  8.7715,  -7.9390,   5.2900],
        [  9.1620,  -7.5573,  -9.9882],
        [  0.1931,  -4.7491,   7.4385],
        [  9.5018,  -7.1945,  -1.1892],
        [ 10.9394,   7.3069,  -8.9365],
        [  9.1820,   9.7334,  -2.7932],
        [  0.1460,   7.8340,  -9.4614],
        [  3.1480,  -4.3165,  -7.8002],
        [ -8.6229,   2.9068,   2.4429],
        [  5.2199,   4.6861,  -1.9304],
        [  2.6613,   7.0021,   4.5783],
        [  5.3804,  -2.3504,   7.2744],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.9452, -10.3187,  -7.8621],
        [ -9.0604,   1.7707,   1.1066],
        [ -3.4100,   7.5436,  -7.6980],
        [  9.3624,   9.5318,   0.2779],
        [  2.6321,  -4.5296,  -2.5159],
        [ -1.5451,   7.8975,   7.6104],
        [  9.5261,  -7.1675,   1.4206],
        [  2.9260,  -4.4162,   8.3808],
        [  5.5158,  -2.1337,   3.5246],
        [  4.4633,   5.6287,  -0.7754],
        [  5.3118,   4.5452,   4.2114],
        [  5.9193,  -1.3176,   5.1739],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.9770, -10.6341,   8.0783],
        [  0.5175,  14.0910,   4.6540],
        [ -5.6651,   6.4380,   1.3660],
        [  3.5284, -10.7484,   5.3347],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.7162,   5.6728,  -8.9372],
        [ -6.4464,   5.8272,   7.5038],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.6672,  -1.8631,   8.1020],
        [  1.0093,   7.6567,   2.3526],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3497,   1.3680,  -1.9076],
        [  6.9433,  -9.3362,  -0.3421],
        [ -8.1531,  -5.8748,   2.6934],
        [ -4.1296,   7.2803,  -2.7194],
        [  9.5203,   9.3483,   5.8781],
        [ -8.3900,   3.3727,   3.1650]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0263, 0.0448,  ..., 0.5221, 0.5958, 0.7198],
        [0.0263, 0.0000, 0.0689,  ..., 0.5125, 0.6138, 0.7307],
        [0.0448, 0.0689, 0.0000,  ..., 0.5583, 0.5533, 0.7260],
        ...,
        [0.5221, 0.5125, 0.5583,  ..., 0.0000, 1.1014, 0.5197],
        [0.5958, 0.6138, 0.5533,  ..., 1.1014, 0.0000, 1.1058],
        [0.7198, 0.7307, 0.7260,  ..., 0.5197, 1.1058, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.02588517,  6.2134961 ,  6.27715105,  6.32572323,  6.39520397,
        6.50409372,  7.822854  , 13.89974693, 12.91505111,  7.20826219,
        8.51928531, 10.39942497,  7.62232307, 11.73397309, 12.12641969,
        6.1561646 , 13.02818702,  5.1805345 ,  7.72668467, 12.58006038,
        9.3349858 ,  6.29821673,  7.24677664, 13.72594022,  9.18690018,
        5.84450127,  8.1137452 ,  6.83515201, 11.11969815, 10.0674887 ,
        4.83371341,  7.96575782, 11.07637531, 11.57249984,  7.27759233,
        8.50268322,  9.69792513,  6.2322769 ,  6.18668463, 14.09176259,
        9.68324553, 10.412299  ,  5.60064319, 12.5085995 , 11.51811603,
        5.16611782, 11.95743558,  6.67108662,  5.29755487, 14.03505624,
        8.87007495,  5.77219077, 10.53368644,  5.90035324, 11.75438006,
       12.4217523 , 13.62896593,  5.82345867, 13.66991027,  9.34127013,
       13.15525803, 13.09426142, 11.83075005, 11.87665501,  4.75303127,
       11.9182956 , 13.15525803, 13.38091121,  7.83534987,  5.34250225,
        9.09963894,  7.01475376,  7.49076899,  5.87132718, 10.4606265 ,
       11.4424816 ,  9.23178204,  8.27853076, 13.3607379 ,  5.23878371,
        8.04718433, 11.92133129,  5.29755487,  5.91408504,  7.18354885,
        6.99097013,  6.06416957,  6.25586297, 11.35345345, 14.10045797,
        8.57560275, 11.31276109, 13.18278606, 13.01728313,  8.68979052,
        5.05963217,  5.1224079 ,  5.96556248,  7.72291946, 12.27993098,
        6.49538841, 11.63502679, 10.04916678,  8.36995953, 13.3426541 ,
        9.04255655])}
{'P': tensor([[2.3348e-02, 2.3845e-01, 1.4672e-01,  ..., 3.1823e-04, 1.5458e-04,
         7.4320e-05],
        [2.3157e-03, 2.8724e-02, 2.0233e-01,  ..., 3.9022e-04, 2.0585e-04,
         8.6930e-05],
        [3.7870e-03, 4.0842e-03, 9.3649e-02,  ..., 1.2278e-03, 1.1147e-03,
         2.2324e-04],
        ...,
        [2.8463e-06, 3.1053e-06, 5.9809e-06,  ..., 4.9075e-03, 5.2791e-06,
         2.6145e-05],
        [1.1340e-04, 1.1017e-04, 9.4827e-05,  ..., 1.5676e-06, 1.0679e-01,
         5.1122e-06],
        [1.1078e-04, 1.1349e-04, 1.4532e-04,  ..., 9.0712e-03, 1.9026e-05,
         7.0347e-02]]), 'X': tensor([[  5.8323,  -1.5228,   5.1132],
        [  5.9193,  -1.3176,   5.1739],
        [  6.2883,   0.0325,   5.2692],
        [  6.2975,   0.0947,   5.1791],
        [  6.2895,   0.0405,   5.7652],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.4756,  -0.8166,  -9.6021],
        [  0.3231,  -4.7691,  -1.6120],
        [ -8.0608,   3.9362,  -5.9567],
        [  8.5198,  10.4091,  -8.6486],
        [ -4.9217,   6.8979,  -1.7667],
        [ -1.5261,   7.8988,   9.3768],
        [ 12.5717,   1.9266,   7.4903],
        [ -4.4520,   7.1370,  -8.7867],
        [  9.1820,   9.7334,  -2.7932],
        [  0.8572,  14.0501,  -8.3364],
        [  5.7151,  -1.7700,   3.1542],
        [  0.3038,   7.8092,   0.4164],
        [ 11.7418,   5.6070,  -1.4192],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.4366,  -0.1189,  -2.6736],
        [  3.1480,  -4.3165,  -7.8002],
        [ -0.6323,   7.9085,  -5.9544],
        [ -8.3539,  -5.5111,   0.3373],
        [  7.1697,  11.5343,  -0.7276],
        [  9.5261,  -7.1675,   1.4206],
        [  5.8617,   3.4853,  -9.8080],
        [  2.7590,  13.6621,   9.4496],
        [ -5.9938,  -8.4745,   3.9735],
        [  3.3744,   6.5623,  -6.3636],
        [  3.6154,  -4.0640,  -7.9698],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.9433,  -9.3362,  -0.3421],
        [  7.8425,  -8.7202,   6.4530],
        [ 10.2470,   8.4054,  -0.1177],
        [  1.0093,   7.6567,   2.3526],
        [ -7.6293,  -6.6859,  -5.5161],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.0864,   7.0387,  -0.1238],
        [  9.3560,   9.5391,   9.4943],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3550,   0.7129,   0.4917],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.6614,  -4.5192,  -9.3641],
        [  1.8719,   7.3713,  -3.7465],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.8400,   2.3991,   1.3457],
        [  4.6357,   5.4415,   7.9067],
        [  8.4691,  10.4571,  -5.0650],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.3236,   0.3011,  -4.6668],
        [  0.4996,   7.7738,  -4.4711],
        [ 12.6059,   0.9597,  -6.3264],
        [ -7.5720,   4.6310,   2.1717],
        [  5.1142,  -2.7230,  -4.1966],
        [ 11.7796,  -3.4696,   9.0088],
        [  8.5093,  -8.1764,  -3.2381],
        [ 11.0438,   7.1180,  -5.6991],
        [ 10.9916,   7.2133,  -0.9179],
        [ -7.5401,   4.6718,   9.0033],
        [  6.1989,  -9.7591,   0.8289],
        [ 12.2218,   4.0953,  -8.2072],
        [  4.5340,  -3.3629,   8.9652],
        [ 12.4996,   2.6335,  -6.3166],
        [  4.4633,   5.6287,  -0.7754],
        [  1.9503,  -4.7172,  -6.2313],
        [ -8.4267,  -5.3697,   2.4172],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.8275,  -1.5334,   2.0124],
        [  6.0140,  -1.0661,  -2.5040],
        [ -1.3940, -10.7772,   6.7835],
        [  4.8653,   5.1694,  -4.0823],
        [  5.1203,  -2.7151,   6.2110],
        [  2.9461,  -4.4077,  -0.4831],
        [ 12.5789,   0.1722,   1.6470],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -8.4824,  -5.2575,  -0.0795],
        [  1.7301,   7.4262,   5.0522],
        [  7.3724,  -9.0577,  -8.9843],
        [  9.3607,   9.5337,  -4.5412],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.0373,   7.1299,  -8.5895],
        [  7.5784,  -8.9141,   5.8410],
        [  2.4932,   7.0903,  -5.4214],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.2883,   0.0325,   5.2692],
        [  2.3261,  13.7749,   5.9121],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.9655,   3.2172,  -7.5953],
        [  9.5018,  -7.1945,  -1.1892],
        [ 12.1940,  -2.1881,   7.1499],
        [ -9.3604,   0.4855,  -3.0158],
        [  2.3928,   7.1404,  -1.7035],
        [  9.1620,  -7.5573,  -9.9882],
        [  2.5636, -10.9293,   9.6519],
        [ -5.0550,   6.8231,  -9.2502],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.1579,  -0.5961,  -6.5595],
        [ 11.6019,   5.9540,  -5.1715],
        [ -5.3382,   6.6533,   7.8054],
        [  6.3360,   0.4258,  -9.2509]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0058, 0.0129,  ..., 0.8423, 0.5131, 0.8007],
        [0.0058, 0.0000, 0.0071,  ..., 0.8460, 0.5146, 0.8054],
        [0.0129, 0.0071, 0.0000,  ..., 0.8514, 0.5155, 0.8118],
        ...,
        [0.8423, 0.8460, 0.8514,  ..., 0.0000, 1.2130, 0.2170],
        [0.5131, 0.5146, 0.5155,  ..., 1.2130, 0.0000, 1.0908],
        [0.8007, 0.8054, 0.8118,  ..., 0.2170, 1.0908, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.02779201,  6.06416957,  6.28835101,  6.29821673,  6.28961988,
        6.24614905,  9.51074763,  4.78002665,  8.97052115, 13.45124917,
        8.47371868,  8.04483495, 12.71843462,  8.41171369, 13.38091121,
       14.07623266,  5.98285319,  7.81509631, 13.01187484, 10.33958639,
        9.43737523,  5.34250225,  7.9337647 , 10.00794468, 13.58101794,
       11.92133129,  6.81961996, 13.93790235, 10.37998078,  7.37902766,
        5.43943791,  5.27178363, 11.63502679, 11.72802491, 13.25336001,
        7.72291946, 10.14436912,  5.60064319, 13.13205357, 13.3614598 ,
        5.0209005 ,  6.02588517,  6.3949043 , 11.63765273,  5.24461068,
        7.60529892,  9.96391045,  9.15976746,  7.14837514, 13.45642379,
        5.26067019,  6.33075682,  7.78980299, 12.64235359,  8.87587457,
        5.79391825, 12.27993098, 11.80092578, 13.13888841, 13.1471313 ,
        8.87007495, 11.56146873, 12.889695  ,  5.64500512, 12.77402351,
        7.18354885,  5.10444097,  9.99211663,  9.48337063,  6.02588517,
        6.10774544, 10.86693708,  7.09887557,  5.79562691,  5.30158497,
       12.58006038, 12.72524615,  9.54154096,  9.97964013,  7.62509582,
       11.67875855, 13.36092624,  6.83515201, 13.1399181 , 11.70016217,
        7.5158489 ,  6.80559669, 13.26590071,  6.12886024,  6.28835101,
       13.96987806, 10.52357479, 13.27717982,  6.77777606, 11.9182956 ,
       12.38881635,  9.37296102,  7.53063885, 11.87665501, 11.22596994,
        8.49161154,  8.22728399,  6.18668463, 13.04051744,  8.53014915,
        6.35029274])}
{'P': tensor([[3.4632e-02, 1.9955e-01, 1.2478e-01,  ..., 2.0056e-04, 2.2927e-05,
         4.2471e-04],
        [4.8006e-03, 4.6395e-02, 1.9414e-01,  ..., 2.8139e-04, 3.1177e-05,
         4.3012e-04],
        [1.3026e-03, 1.5921e-03, 5.5913e-02,  ..., 9.1610e-04, 4.6180e-05,
         3.0363e-04],
        ...,
        [1.8140e-05, 1.6754e-05, 2.1488e-05,  ..., 3.3765e-02, 2.4563e-04,
         4.4820e-05],
        [2.6606e-04, 2.1630e-04, 8.1307e-05,  ..., 7.7131e-06, 5.9573e-02,
         5.4972e-05],
        [3.4256e-03, 3.2406e-03, 6.0245e-03,  ..., 4.7389e-03, 4.0433e-05,
         2.3239e-01]]), 'X': tensor([[  5.8325,  -1.5222,  -2.1341],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.3497,   1.3680,  -1.9076],
        [  6.3542,   0.6934,  -3.3840],
        [  6.3087,   1.8053,  -1.9481],
        [  5.9548,  -1.2273,  -2.6520],
        [ -5.1369,   6.7755,   8.3836],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.1178,   1.5784,   2.1410],
        [  0.3715,  -4.7756,   0.2312],
        [ -9.4697,  -0.6157,  -6.8472],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.3124,   0.2057,  -0.4593],
        [  3.3167,   6.6022,   6.3554],
        [  4.4290,  -3.4596,   1.4181],
        [  0.3231,  -4.7691,  -1.6120],
        [ 10.1503,   8.5414,  -7.2681],
        [ 11.3118,   6.5957,  -4.5667],
        [  4.9531,   5.0576,   5.6794],
        [  1.5246,  -4.7838,  -5.2095],
        [ -3.1671,   7.6158,   6.0765],
        [ -6.3066,  -8.1936,   9.5822],
        [  5.6928,  12.4710,   3.6001],
        [  4.5325,  -3.3643,   6.7699],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.4773,  -1.0024,   5.2599],
        [ -8.7735,   2.5640,  -6.3313],
        [ -9.4430,  -0.1897,   4.8383],
        [  0.1497,   7.8334,  -5.4412],
        [  3.9663,   6.1018,  -5.4104],
        [  6.3200,   0.2690,   1.5900],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.6229,   2.9068,   2.4429],
        [  0.9809,  -4.8145,   6.8706],
        [  8.4691,  10.4571,  -5.0650],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.9152,  12.8653,  -1.4459],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -8.0829,  -5.9940,   5.0564],
        [ -5.0943,  -9.1682,   0.5073],
        [ -6.3306,   5.9272,   5.6935],
        [  2.1824,  -4.6647,   1.1985],
        [  3.3474,  -4.2162,   5.2455],
        [  6.3048,   0.1469,  -9.1691],
        [  2.5636, -10.9293,   9.6519],
        [ 10.1798,   8.5002,  -5.8893],
        [ 12.5267,  -0.4125,  -8.7331],
        [  1.7500,   7.4187,   4.4237],
        [ -9.4296,  -1.9458,   4.0833],
        [  6.0788,  -0.8710,  -6.8719],
        [  6.3236,   0.3011,  -4.6668],
        [ -8.7544,   2.6096,   3.3146],
        [  6.3301,   0.3638,  -7.5685],
        [  1.9312,  13.8649,   8.3563],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.0961,   1.6529,   3.1553],
        [  5.7802,  -1.6363,  -6.2782],
        [  1.8447,  -4.7372,   8.1235],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -0.1626, -10.9796,   6.4629],
        [  9.0652,   9.8597,  -3.5949],
        [  3.1064,  13.5607,   6.7605],
        [ -7.0120,   5.2825,   7.7762],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.4426, -10.1181,  -5.1268],
        [  3.9770, -10.6341,   8.0783],
        [  3.6154,  -4.0640,  -7.9698],
        [  3.3167,   6.6022,   6.3554],
        [  5.8323,  -1.5228,   5.1132],
        [  6.2902,   0.0451,   3.7116],
        [ -6.3489,   5.9117,   6.5161],
        [  4.2275,  -3.6320,  -0.2714],
        [ 12.4877,  -0.7222,   8.8347],
        [  0.9445,   7.6736,   0.5095],
        [ -8.6381,  -4.9238,  -4.4077],
        [  7.7750,  11.0665,   2.8364],
        [  6.3048,   0.1469,  -9.1691],
        [ 12.2328,  -2.0376,  -1.1234],
        [  4.0421,  -3.7765,  -8.3631],
        [ -9.3087,   0.7792,  -7.5587],
        [ -3.5903,   7.4847,  -8.9112],
        [ -7.6734,   4.4980,  -5.4248],
        [  8.1217,  -8.5024,  -0.6149],
        [ -3.0644,   7.6439,  -1.4322],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3303,   1.6145,  -8.4083],
        [  1.6422,   7.4586,  -6.8056],
        [ 12.5154,   2.5072,  -9.3274],
        [  9.5203,   9.3483,   5.8781],
        [ -0.7599, -10.8995,  -6.0366],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.1231,  -3.5739,   8.4708],
        [  1.7815,  -4.7481,   4.4289],
        [  6.2022,   2.4092,   3.0349],
        [  5.9843,  -9.8679,  -6.1449],
        [  6.3550,   0.7129,   0.4917],
        [  9.5018,  -7.1945,  -1.1892],
        [ -1.0407,   7.9166,  -6.6021],
        [ -2.0721,   7.8432,   6.6380],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.6672,  -1.8631,   8.1020]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0258, 0.0184,  ..., 0.7116, 0.9877, 0.6606],
        [0.0258, 0.0000, 0.0408,  ..., 0.7367, 0.9909, 0.6854],
        [0.0184, 0.0408, 0.0000,  ..., 0.7022, 0.9996, 0.6447],
        ...,
        [0.7116, 0.7367, 0.7022,  ..., 0.0000, 1.0137, 0.3381],
        [0.9877, 0.9909, 0.9996,  ..., 1.0137, 0.0000, 1.2744],
        [0.6606, 0.6854, 0.6447,  ..., 0.3381, 1.2744, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.02789634,  6.10774544,  6.49538841,  6.39187285,  6.5618968 ,
        6.07992623,  8.50268322, 11.72802491,  9.25336371,  4.79003361,
        9.48969978, 12.7125079 ,  6.31576264,  7.38844703,  5.62005629,
        4.78002665, 13.26590071, 13.09426142,  7.07902036,  5.0209005 ,
        8.24808832, 10.33958639, 13.70893492,  5.64464807,  6.02588517,
        9.53015376,  9.14044984,  9.44486609,  7.83486943,  7.27759233,
        6.32572323,  8.22728399,  9.09963894,  4.91336994, 13.45642379,
        5.53175275, 13.77222752, 10.14436912, 10.06286074, 10.48840476,
        8.67227386,  5.14999799,  5.38342077,  6.30647746, 11.22596994,
       13.26209545, 12.53345445,  7.62232307,  9.6282763 ,  6.14086582,
        6.33075682,  9.13507321,  6.34059021, 13.99877413,  9.96391045,
        6.12886024,  6.38968357, 12.62576026,  9.24501829,  6.00732193,
        5.08372149, 12.4217523 , 10.98076873, 13.39372426, 13.91198191,
        8.77913428,  4.91652556, 11.4890861 , 11.35345345,  5.43943791,
        7.38844703,  6.02779201,  6.29034958,  8.67502348,  5.57341623,
       12.5085995 ,  7.73151309,  9.94285511, 13.5247215 ,  6.30647746,
       12.4013153 ,  5.53175275,  9.34127013,  8.30125201,  8.89458273,
       11.75807252,  8.23525433, 12.10942412, 12.27993098,  6.53290663,
        7.63726134, 12.76408562, 13.3426541 , 10.92597196, 12.12641969,
        9.64157461,  9.79814997,  5.0713364 ,  6.65368551, 11.54071859,
        6.3949043 , 11.9182956 ,  7.98469166,  8.11227668,  9.79039371,
        5.96556248])}
{'P': tensor([[3.3400e-02, 3.0671e-01, 1.8215e-01,  ..., 2.9708e-04, 3.5413e-06,
         5.7269e-03],
        [2.8122e-03, 4.6915e-02, 2.8839e-01,  ..., 5.1449e-04, 4.4619e-06,
         2.9764e-03],
        [2.3001e-03, 3.4773e-03, 5.8589e-02,  ..., 6.5871e-04, 4.2304e-06,
         2.1618e-03],
        ...,
        [5.2166e-05, 5.8558e-05, 6.1465e-05,  ..., 8.5217e-02, 6.0191e-05,
         5.9549e-05],
        [5.5953e-05, 4.5325e-05, 2.6321e-05,  ..., 1.0635e-05, 3.1151e-02,
         8.7322e-05],
        [8.3731e-02, 1.5273e-01, 9.6674e-02,  ..., 3.3381e-04, 4.5542e-06,
         3.1995e-02]]), 'X': tensor([[  5.8889,  -1.3916,   8.8993],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3266,   0.3292,   9.3502],
        [  6.3583,   1.1729,   8.7835],
        [  6.3119,   1.7797,   9.9186],
        [  6.0054,  -1.0902,   8.3079],
        [  5.9587,  12.3216,   9.1695],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.0078,  -3.8018,  -6.2168],
        [  5.9587,  12.3216,   9.1695],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.4430,  -0.1897,   4.8383],
        [ -1.6086,   7.8928,   4.3385],
        [  6.2770,  -0.0379,   2.4195],
        [ -1.8367,   7.8718,  -8.6523],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.4309,  -0.0599,  -4.0830],
        [  7.1697,  11.5343,  -0.7276],
        [ 11.2571,   6.7072,   8.3693],
        [ -9.4351,  -1.8891,   3.5642],
        [  8.1528,  10.7452,   9.1493],
        [ -2.5683, -10.4433,  -2.9182],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.9808,   4.9340,  -3.3715],
        [ -0.7599, -10.8995,  -6.0366],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.3551,   0.7149,   1.3344],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.5295,  -7.9785,  -7.4429],
        [  1.9241,   7.3503,   7.9004],
        [  0.6142,  14.0802,   0.5407],
        [  3.3839,  -4.1966,   8.3574],
        [  9.4925,   9.3811,   8.9842],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -9.4309,  -0.0599,  -4.0830],
        [  4.7813,  -3.1139,   8.2691],
        [ -2.1382,   7.8339,   7.3524],
        [  4.8051, -10.3703,   0.5268],
        [ 12.4592,  -0.9168,  -6.7571],
        [  2.9461,  -4.4077,  -0.4831],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.4996,   2.6335,  -6.3166],
        [ 11.6165,  -3.8715,   6.4594],
        [ 12.3609,   3.4769,  -8.5762],
        [ -8.8540,  -4.3971,  -5.2144],
        [  8.2154,  -8.4262,   8.0171],
        [  6.3535,   0.6792,  -7.5101],
        [  9.6587,   9.1817,  -0.8212],
        [  1.1139,   7.6281,  -2.5680],
        [  5.4167, -10.1293,   6.2156],
        [  0.7685,  -4.8098,  -2.9310],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.5678,   7.9261,  -0.6100],
        [  6.2630,   2.1057,  -5.2136],
        [ -8.3900,   3.3727,   3.1650],
        [ -7.3333,   4.9246,  -7.9212],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.0617,  12.7957,   8.3544],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.0047,   3.1062,  -6.7914],
        [ 12.5650,   2.0132,   4.6272],
        [ -8.1905,   3.7253,   4.1175],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.9452, -10.3187,  -7.8621],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -5.7347,  -8.6908,   9.2878],
        [  4.6231,   5.4556,   8.4973],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.3589,   0.8239,  -8.4230],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.8400,   2.3991,   1.3457],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.3211,   1.7017,  -0.2164],
        [ -6.0175,  -8.4541,  -9.8408],
        [  9.0652,   9.8597,  -3.5949],
        [ 11.3118,   6.5957,  -4.5667],
        [  3.4401,  13.4540,   4.6576],
        [ -7.4602,   4.7718,  -4.6277],
        [  3.9085,   6.1512,  -2.2560],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.0031,  12.2958,  -6.5992],
        [  6.3200,   0.2690,   1.5900],
        [  6.3549,   0.7096,   6.9198],
        [  1.3767, -11.0341,   9.9415],
        [  5.4867,  12.5816,   2.9549],
        [  4.1999,  -3.6543,  -4.1998],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -6.7643,  -7.7372,  -9.6663],
        [  9.2811,  -7.4336,   5.1771],
        [  6.2419,  -0.2313,   4.2440],
        [  5.6672,  -1.8631,   8.1020],
        [ -6.8466,  -7.6488,   1.9648],
        [ -7.8359,   4.2735,   7.2943],
        [  3.1064,  13.5607,   6.7605],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -7.6734,   4.4980,  -5.4248],
        [  4.3554,   5.7393,  -9.0766],
        [  1.3103,  -4.8032,  -0.9300],
        [  4.5800,   5.5034,  -3.9432],
        [  2.2437,  -4.6489,   6.1430],
        [ -1.8788, -10.6567,   3.7068],
        [  2.0547,   7.2957,  -1.9343],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.6672,  -1.8631,   8.1020]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0037, 0.0103,  ..., 0.6865, 1.2646, 0.0251],
        [0.0037, 0.0000, 0.0095,  ..., 0.6885, 1.2679, 0.0274],
        [0.0103, 0.0095, 0.0000,  ..., 0.6968, 1.2724, 0.0353],
        ...,
        [0.6865, 0.6885, 0.6968,  ..., 0.0000, 0.7820, 0.6615],
        [1.2646, 1.2679, 1.2724,  ..., 0.7820, 0.0000, 1.2429],
        [0.0251, 0.0274, 0.0353,  ..., 0.6615, 1.2429, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.05113217,  6.16512856,  6.33517349,  6.46560181,  6.55800267,
        6.10360073, 13.68672855,  5.18047638,  5.52416368, 13.68672855,
       12.11735334,  9.44486609,  8.0550286 ,  6.27715105,  8.08320235,
        9.50913849,  9.43112957, 13.58101794, 13.103709  ,  9.62238828,
       13.48809595, 10.75443106,  8.50268322, 12.95702647, 10.92597196,
       13.01187484,  6.39520397,  9.11473391, 10.30972193,  7.59795445,
       14.09357547,  5.39098669, 13.34587027, 13.79762415, 12.46513646,
        9.43112957,  5.70591986,  8.12043983, 11.42947632, 12.49291768,
        5.30158497, 12.44177641, 12.77402351, 12.24467041, 12.84056604,
        9.88571713, 11.76830981,  6.38968357, 13.32645552,  7.70897675,
       11.48663767,  4.87083348,  6.00663314, 13.20988996,  6.60752373,
        9.04255655,  8.83341608,  5.63824971, 13.76047999,  5.84450127,
        6.76056625, 12.72524615,  8.99791316,  9.97964013, 11.4424816 ,
        7.09887557,  9.50654967, 10.412299  ,  7.15100303,  5.26067019,
        6.41202986, 13.6660671 ,  9.15976746, 10.39942497,  6.54615887,
       10.37698435, 13.39372426, 13.09426142, 13.88683436,  8.85572292,
        7.28793663, 12.4013153 , 13.6829843 ,  6.32572323,  6.39438044,
       11.11969815, 13.72594022,  5.5671389 , 12.4217523 , 10.27716434,
       11.89105259,  6.24614905,  5.96556248, 10.26545792,  8.92549752,
       13.91198191,  9.58072286,  8.89458273,  7.20482224,  4.978704  ,
        7.15990676,  5.16205087, 10.82106505,  7.57946459, 10.52357479,
        5.96556248])}
{'P': tensor([[2.1575e-02, 1.5764e-01, 1.3629e-01,  ..., 1.0889e-05, 2.7334e-04,
         5.0890e-02],
        [1.9893e-03, 4.0807e-02, 1.0663e-01,  ..., 2.0068e-05, 2.0140e-04,
         1.7090e-01],
        [1.8045e-03, 9.4786e-03, 4.1092e-02,  ..., 2.1243e-05, 1.6934e-04,
         2.0076e-01],
        ...,
        [2.3317e-04, 1.3414e-04, 1.2588e-04,  ..., 4.7105e-02, 1.8708e-04,
         4.8124e-05],
        [1.1221e-02, 1.5134e-02, 1.7015e-02,  ..., 1.5739e-06, 4.6460e-02,
         1.0987e-02],
        [2.0802e-03, 3.0789e-03, 3.3618e-03,  ..., 8.5925e-05, 1.2707e-04,
         1.3410e-01]]), 'X': tensor([[ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.1943e-02, 1.6711e-02,  ..., 9.9403e-01, 1.9296e-01,
         2.1988e-02],
        [2.1943e-02, 0.0000e+00, 3.6387e-02,  ..., 1.0058e+00, 2.1152e-01,
         1.2565e-02],
        [1.6711e-02, 3.6387e-02, 1.5170e-04,  ..., 9.9642e-01, 1.8896e-01,
         3.8637e-02],
        ...,
        [9.9403e-01, 1.0058e+00, 9.9642e-01,  ..., 0.0000e+00, 8.2153e-01,
         9.9382e-01],
        [1.9296e-01, 2.1152e-01, 1.8896e-01,  ..., 8.2153e-01, 0.0000e+00,
         2.0251e-01],
        [2.1988e-02, 1.2565e-02, 3.8637e-02,  ..., 9.9382e-01, 2.0251e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.05165248,  6.24614905,  6.29034958,  6.42689705,  6.19707944,
        6.29821673,  5.51177501, 13.58101794,  5.80312949,  5.39723241,
        6.49538841, 11.73567759,  6.3949043 , 12.21532282,  6.54127431,
        7.72291946, 13.09737053, 11.89855267,  9.79039371, 11.97957425,
       11.99633819, 10.68031351,  5.47702766, 11.28895489, 11.73397309,
        9.11473391, 13.2430377 ,  8.51088913, 10.31886066, 13.43127131,
        5.5748624 , 13.01187484,  6.81961996, 12.1494023 ,  7.20826219,
        6.24614905, 10.90638066, 11.018084  , 13.20597932, 13.31764181,
        4.74770899,  5.94293185,  5.39723241,  8.22728399, 10.48315436,
        5.29755487,  5.77219077,  7.822854  ,  8.31625171,  8.94395628,
        7.32754882,  5.47866221,  5.23878371,  5.16077764, 12.86648771,
       14.05633441, 11.36089355, 14.09357547,  6.61386436, 10.23946247,
       11.4890861 , 10.7506423 , 13.58101794,  6.39187285,  5.37343173,
       11.1913598 , 13.0743399 , 10.4606265 ,  5.91408504, 10.37698435,
        6.14086582, 12.62576026, 10.4977132 , 14.10796466,  5.79562691,
       13.00380624,  7.96575782, 12.5769659 ,  6.41202986,  8.06990483,
       13.50484539, 13.97779839, 11.69264243, 13.89974693,  9.95552626,
        5.05963217,  5.13616224, 13.94008948,  7.39004226, 10.30972193,
        9.66977802, 12.41012046, 12.97370371,  5.68942572,  9.10945678,
        5.35774573, 10.14044489,  9.26455795,  8.0550286 , 13.54774355,
        6.27715105,  6.1463896 , 10.33712258,  9.95552626, 11.99633819,
        6.67218687])}
{'P': tensor([[3.1779e-02, 2.2385e-01, 2.1976e-01,  ..., 1.1220e-04, 1.2968e-03,
         3.0461e-05],
        [2.9907e-03, 6.8574e-02, 2.9233e-01,  ..., 3.6633e-04, 1.3585e-03,
         8.6184e-05],
        [4.2223e-03, 2.1365e-02, 9.9900e-02,  ..., 5.5893e-04, 1.9093e-03,
         1.2953e-04],
        ...,
        [2.5702e-05, 2.3673e-05, 2.4067e-05,  ..., 5.4874e-02, 1.4243e-05,
         7.3877e-02],
        [6.7169e-02, 7.1088e-02, 6.7428e-02,  ..., 7.7800e-05, 1.2342e-01,
         4.0104e-05],
        [5.7414e-05, 3.9721e-05, 3.9752e-05,  ..., 1.4043e-04, 5.3456e-05,
         7.0045e-02]]), 'X': tensor([[  5.9193,  -1.3176,   5.1739],
        [  6.2883,   0.0325,   5.2692],
        [  6.2975,   0.0947,   5.1791],
        [  6.2895,   0.0405,   5.7652],
        [  6.2419,  -0.2313,   4.2440],
        [  6.3569,   0.7591,   6.5011],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.3858,   4.4260,  -9.6769],
        [ -2.2508, -10.5479,  -2.7235],
        [  2.7295,  13.6703,  -8.9328],
        [  3.3943,   6.5483,   1.0790],
        [ -6.9838,  -7.4965,   0.9796],
        [ -6.0175,  -8.4541,  -9.8408],
        [  1.8766,   7.3694,   1.1887],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -6.5295,  -7.9785,  -7.4429],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.6523,   3.9443,  -2.9449],
        [  7.5064,  -8.9651,  -0.0994],
        [ -2.0229, -10.6163,   8.6930],
        [ 12.1588,   4.3364,  -8.0867],
        [  7.7750,  11.0665,   2.8364],
        [  6.3542,   0.6934,  -3.3840],
        [  3.9515,  -3.8425,   6.1729],
        [  3.1756, -10.8247,   6.8882],
        [ 12.5267,  -0.4125,  -8.7331],
        [  3.3943,   6.5483,   1.0790],
        [  6.8093,  -9.4178,   8.4244],
        [ 12.2218,   4.0953,  -8.2072],
        [ -4.3359,  -9.6431,  -9.0616],
        [  6.0305,   3.0296,   0.1375],
        [ -6.0844,  -8.3955,   0.7705],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.9913,  -3.8139,  -6.8311],
        [ -4.1296,   7.2803,  -2.7194],
        [ -1.7290,   7.8825,  -3.3831],
        [  0.4996,   7.7738,  -4.4711],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 10.0619,   8.6627,  -9.2884],
        [ -3.2736, -10.1703,   3.8712],
        [  6.3236,   0.3011,  -4.6668],
        [  8.1528,  10.7452,   9.1493],
        [  1.6308,  13.9254,  -2.0984],
        [ 12.4691,   2.8525,  -5.5110],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.9531,   5.0576,   5.6794],
        [ 12.3609,   3.4769,  -8.5762],
        [ -1.6086,   7.8928,   4.3385],
        [ -7.6650,   4.5092,   0.2520],
        [ -8.1426,   3.8048,   9.3542],
        [  2.0243,   7.3086,   7.5432],
        [  4.2339,  -3.6268,   6.3125],
        [  3.2366,  -4.2732,   5.1532],
        [ 11.5935,   5.9740,  -4.3952],
        [ -0.5709,   7.9055,   7.6089],
        [  9.0004,  -7.7195,   4.6945],
        [  2.2659,  13.7894,   6.5436],
        [  2.3373,  13.7721,  -0.2442],
        [ -5.5525,   6.5146,  -4.4984],
        [ -1.6334, -10.7207,   7.1397],
        [  1.1139,   7.6281,  -2.5680],
        [ -5.1377,   6.7751,  -0.4438],
        [ -7.3333,   4.9246,  -7.9212],
        [ -6.3489,   5.9117,   6.5161],
        [  4.4915,   5.5990,   7.0405],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.4501,  -1.7133,   2.2980],
        [  0.2435,   7.8191,  -6.4709],
        [ 11.9019,  -3.1379,  -3.4534],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.2608,  -2.5256,  -9.1446],
        [ -7.8126,   4.3066,   5.1791],
        [  9.4925,   9.3811,   8.9842],
        [  5.8450,  -9.9356,   3.0767],
        [  7.7794,  11.0629,   8.9069],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.3889,  -4.6082,  -8.7291],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.4633,   5.6287,  -0.7754],
        [  8.4691,  10.4571,  -5.0650],
        [  6.3542,   0.6934,  -3.3840],
        [  6.3515,   0.6418,  -6.5823],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.2435,   4.0076,   8.1257],
        [  6.3266,   0.3292,   9.3502],
        [ -9.4505,  -1.7081,  -0.7039],
        [  4.3554,   5.7393,  -9.0766],
        [  4.6357,   5.4415,   7.9067],
        [  7.0171,  11.6439,  -5.0178],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.4753,  -0.8012,  -5.2050],
        [  7.3724,  -9.0577,  -8.9843],
        [  9.5261,  -7.1675,   1.4206],
        [  2.0899,  -4.6870,  -7.4260],
        [  6.0560,  -0.9421,  -5.0334],
        [ -8.7735,   2.5640,  -6.3313],
        [  1.6308,  13.9254,  -2.0984],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.4501,  -1.7133,   2.2980],
        [ -0.8569,   7.9156,  -6.5557],
        [ -4.0409,   7.3169,  -3.5779],
        [  4.5340,  -3.3629,   8.9652],
        [ -8.8058,   2.4850,  -1.7425]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0071, 0.0030,  ..., 0.7605, 0.1574, 0.9710],
        [0.0071, 0.0000, 0.0060,  ..., 0.7676, 0.1513, 0.9772],
        [0.0030, 0.0060, 0.0000,  ..., 0.7626, 0.1572, 0.9737],
        ...,
        [0.7605, 0.7676, 0.7626,  ..., 0.0000, 0.8881, 0.3871],
        [0.1574, 0.1513, 0.1572,  ..., 0.8881, 0.0000, 1.0563],
        [0.9710, 0.9772, 0.9737,  ..., 0.3871, 1.0563, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.06416957,  6.28835101,  6.29821673,  6.28961988,  6.24614905,
        6.4020404 , 11.63724792,  6.97106912, 10.78533838, 13.94008948,
        7.37576154, 10.24557047, 10.37698435,  7.60463349, 12.4217523 ,
       10.30972193,  6.14086582,  6.89245352, 11.69264243, 10.80728788,
       12.9089559 , 13.5247215 ,  6.39187285,  5.51177501, 11.28093507,
       12.53345445,  7.37576154, 11.62158554, 12.889695  , 10.57302546,
        6.74872981, 10.36844813, 12.27993098,  5.5205189 ,  8.36995953,
        8.06990483,  7.78980299,  4.8763135 , 12.4013153 , 13.27717982,
       10.68416401,  6.33075682, 13.48809595, 14.02058621, 12.79126785,
       13.25336001,  5.24137909,  7.07902036, 12.84056604,  8.0550286 ,
        8.89301516,  8.98765644,  7.58378655,  5.5748624 ,  5.36061974,
       13.04217487,  7.92607342, 11.85743425, 13.97430191, 13.96905756,
        8.55982105, 10.84437705,  7.70897675,  8.50278896,  8.83341608,
        8.67502348,  7.17789666,  7.53063885,  9.60412735,  7.822854  ,
       12.30858677,  5.1224079 ,  5.83560009,  8.92098752, 13.34587027,
       11.52733581, 13.52430176,  9.91668182,  5.19065439, 13.25017211,
        7.18354885, 13.45642379,  6.39187285,  6.383897  , 13.96905756,
       12.88270018,  6.33517349,  9.60359158,  7.20482224,  7.14837514,
       13.59480906, 10.76154256,  9.6282763 ,  9.50913849, 11.67875855,
       11.92133129,  5.13182506,  6.12886024,  9.14044984, 14.02058621,
       10.67322527,  9.60412735,  7.96181288,  8.35856487,  5.64500512,
        9.14973319])}
{'P': tensor([[2.7523e-02, 1.7008e-01, 1.2527e-01,  ..., 3.1662e-05, 4.5186e-05,
         2.3849e-05],
        [4.1869e-03, 3.1311e-02, 1.3566e-01,  ..., 3.3473e-05, 5.1816e-05,
         2.7595e-05],
        [1.1901e-03, 1.3570e-03, 3.1539e-02,  ..., 1.6264e-05, 2.6195e-05,
         3.4504e-05],
        ...,
        [6.2903e-03, 5.9813e-03, 2.4880e-03,  ..., 3.5549e-02, 2.5875e-04,
         2.6785e-06],
        [1.8462e-03, 1.9557e-03, 1.1503e-03,  ..., 8.9910e-04, 2.3547e-01,
         5.8939e-05],
        [1.2701e-04, 1.1899e-04, 6.1134e-05,  ..., 1.0987e-03, 1.2644e-04,
         4.8007e-02]]), 'X': tensor([[  5.9548,  -1.2273,  -2.6520],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.3542,   0.6934,  -3.3840],
        [  6.3497,   1.3680,  -1.9076],
        [  6.3587,   1.1596,  -3.6629],
        [  6.3087,   1.8053,  -1.9481],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.2209,  -4.7092,   2.6117],
        [  3.6817,   6.3354,   6.1233],
        [ -9.4309,  -0.0599,  -4.0830],
        [  5.9526,   3.2525,  -4.3821],
        [ -7.6719,  -6.6260,   7.1468],
        [ -2.2508, -10.5479,  -2.7235],
        [ -7.7842,   4.3466,  -1.7548],
        [ -7.3333,   4.9246,  -7.9212],
        [  4.4915,   5.5990,   7.0405],
        [ 10.1983,  -6.3430,  -0.3200],
        [  1.5653,  13.9377,   4.1773],
        [ -9.4351,  -1.8891,   3.5642],
        [  0.1460,   7.8340,  -9.4614],
        [ 11.3691,   6.4760,   1.0787],
        [  6.3087,   1.8053,  -1.9481],
        [  5.1793,   4.7461,   0.3899],
        [  6.3583,   1.1729,   8.7835],
        [  7.9403,  -8.6455,  -8.7433],
        [ -1.5451,   7.8975,   7.6104],
        [  2.2860,   7.1916,  -6.1011],
        [  6.8407,  11.7666,   3.0045],
        [  0.9809,  -4.8145,   6.8706],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.2974,   0.8371,   2.4852],
        [  0.1252, -11.0062,   7.2564],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.3602,  -2.3808,   7.2098],
        [ 12.3357,   3.6001,   0.7903],
        [ -9.3087,   0.7792,  -7.5587],
        [ -3.3116, -10.1539,   7.5919],
        [  2.2644,  -4.6434,   9.7508],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.4691,   2.8525,  -5.5110],
        [ 10.0619,   8.6627,  -9.2884],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.3411,  -2.6004,  -2.7062],
        [  1.5653,  13.9377,   4.1773],
        [  1.7232,  -4.7574,   3.1270],
        [ -9.4430,  -0.1897,   4.8383],
        [  9.2692,   9.6370,   7.0746],
        [  5.9843,  -9.8679,  -6.1449],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.5650,   2.0132,   4.6272],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.6970,   6.9826,   9.7350],
        [  0.1677,  -4.7447,  -6.1529],
        [ -8.5506,  -5.1154,   6.3423],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.0039,   7.3173,  -1.5885],
        [  5.0617,  12.7957,   8.3544],
        [  6.3497,   1.3680,  -1.9076],
        [ 11.4495,   6.3026,   6.8257],
        [  2.3928,   7.1404,  -1.7035],
        [  5.6202,  -1.9508,   0.7471],
        [ -3.4100,   7.5436,  -7.6980],
        [  0.6363, -11.0348,   6.1414],
        [ -7.5603,   4.6459,   8.2002],
        [  2.0729,   7.2878,   7.0880],
        [  5.9883,   3.1536,   6.2320],
        [  5.3858,   4.4260,  -9.6769],
        [  0.7685,  -4.8098,  -2.9310],
        [ -8.4573,   3.2448,   5.4370],
        [  1.1274,  14.0116,   8.1753],
        [ -9.3739,   0.3982,  -4.2564],
        [  2.2849, -10.9655,  -7.6542],
        [ -9.3076,   0.7852,  -6.7010],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4159,  -2.0739,   9.6841],
        [ 11.7418,   5.6070,  -1.4192],
        [ 10.5678,   7.9261,  -0.6100],
        [  2.5010,  -4.5736,  -0.8989],
        [  1.7228,  -4.7575,   8.3396],
        [ -8.6603,   2.8250,   5.9181],
        [ -5.3382,   6.6533,   7.8054],
        [ 12.3357,   3.6001,   0.7903],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.4592,  -0.9168,  -6.7571],
        [  1.8721,  -4.7322,   0.5524],
        [ -9.4735,  -1.2620,  -3.6742],
        [  4.5430,   5.5438,  -3.8669],
        [  2.5663,   7.0526,  -6.8769],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.2707,   0.9661,   4.4863],
        [ -6.3489,   5.9117,   6.5161],
        [  2.5375,   7.0676,  -4.3490],
        [  6.3589,   0.8239,  -8.4230],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -9.4309,  -0.0599,  -4.0830],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.9912,   3.1453,  -3.9989],
        [ 11.0864,   7.0387,  -0.1238],
        [  0.9445,   7.6736,   0.5095],
        [  5.8617,   3.4853,  -9.8080],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.3490,  -4.6199,  -4.2489],
        [  3.2159, -10.8166,  -1.0406],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.4645,  -1.4880,  -2.0226]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0092, 0.0434,  ..., 0.5401, 0.7256, 0.9390],
        [0.0092, 0.0000, 0.0518,  ..., 0.5407, 0.7168, 0.9411],
        [0.0434, 0.0518, 0.0000,  ..., 0.5573, 0.7684, 0.9488],
        ...,
        [0.5401, 0.5407, 0.5573,  ..., 0.0000, 0.8169, 0.4164],
        [0.7256, 0.7168, 0.7684,  ..., 0.8169, 0.0000, 1.1719],
        [0.9390, 0.9411, 0.9488,  ..., 0.4164, 1.1719, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.07992623,  6.10774544,  6.39187285,  6.49538841,  6.46356806,
        6.5618968 , 11.63502679, 12.16901615,  7.32754882,  9.43112957,
        6.78325151, 10.1371552 , 10.78533838,  8.91551231,  8.83341608,
        7.17789666, 12.00995898, 14.02532758,  9.62238828,  7.83534987,
       13.08414689,  6.5618968 ,  7.02496223,  6.46560181, 11.73848065,
        8.04718433,  7.54620849, 13.6105595 ,  4.91336994, 13.77222752,
        9.3349858 , 11.00694688,  9.75200848,  5.86519525, 12.85032836,
        9.34127013, 10.68031351,  5.16611782,  8.89458273, 12.79126785,
       13.27717982,  9.96391045,  9.69629   , 14.02532758,  5.05989547,
        9.44486609, 13.37121923, 11.54071859, 12.87752052, 12.72524615,
       10.1371552 ,  7.48538905,  4.74770899,  9.96391045, 12.10942412,
        7.58668144, 13.76047999,  6.49538841, 13.06957425,  7.53063885,
        5.94909709,  8.27853076, 11.05317597,  8.8737556 ,  7.57685908,
        6.76791483,  6.97106912,  4.87083348,  9.05843201, 14.05687815,
        9.38232121, 11.20100795,  9.34061938,  7.13501264,  9.64157461,
       13.01187484, 13.20988996,  5.21280156,  5.05981914,  9.10945678,
        8.53014915, 12.85032836,  4.83371341, 12.49291768,  5.08910491,
        9.55720651,  7.16747878,  7.50500064,  9.20340031,  9.32094055,
        8.67502348,  7.5092834 ,  6.41202986,  8.62260327,  9.54154096,
        9.43112957,  5.19986797,  6.76661595, 13.13205357,  7.73151309,
        6.81961996,  9.14973319,  5.18277858, 11.2845608 ,  5.16611782,
        9.58072286])}
{'P': tensor([[3.8387e-02, 1.7713e-01, 1.4551e-01,  ..., 2.2292e-05, 1.3780e-04,
         1.6765e-05],
        [6.9659e-03, 2.3008e-01, 1.6555e-02,  ..., 1.4578e-04, 1.5917e-03,
         5.3739e-05],
        [5.3951e-03, 5.4472e-01, 9.2215e-02,  ..., 4.1963e-05, 4.2492e-04,
         2.0518e-05],
        ...,
        [7.6578e-06, 9.1202e-06, 6.5424e-06,  ..., 8.9829e-02, 9.0565e-04,
         2.0513e-04],
        [2.5434e-05, 1.8656e-05, 1.8253e-05,  ..., 1.2688e-05, 3.5127e-02,
         2.1932e-05],
        [2.0845e-04, 1.6196e-04, 1.3104e-04,  ..., 5.0091e-03, 8.3770e-05,
         1.7020e-01]]), 'X': tensor([[  6.0054,  -1.0902,   8.3079],
        [  6.3583,   1.1729,   8.7835],
        [  6.3266,   0.3292,   9.3502],
        [  6.2242,  -0.3171,   7.4978],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3549,   0.7096,   6.9198],
        [  6.3551,   0.7149,   1.3344],
        [ -6.2668,   5.9808,   2.5614],
        [  1.1274,  14.0116,   8.1753],
        [ 12.4877,  -0.7222,   8.8347],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.1206,  -4.7363,   8.7103],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.3464,   0.5708,  -1.6397],
        [  9.5018,  -7.1945,  -1.1892],
        [ -1.6334, -10.7207,   7.1397],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.9883,   3.1536,   6.2320],
        [  1.7659,   7.4127,   4.7871],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.6093,  -1.9705,   6.4807],
        [ 12.0582,  -2.6640,   0.7132],
        [ -1.9646,   7.8571,   3.3817],
        [ -1.5875,   7.8944,  -9.4286],
        [ -8.8154,   2.4611,   2.5198],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.3107,  -2.7697,  -3.2919],
        [  2.2860,   7.1916,  -6.1011],
        [  9.7328,   9.0902,  -0.8917],
        [  5.9655,   3.2172,  -7.5953],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.3411,  -2.6004,  -2.7062],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.0534,  -6.0429,   2.4583],
        [  0.6363, -11.0348,   6.1414],
        [ 12.5789,   0.1722,   1.6470],
        [  4.0457,  -3.7737,  -0.4704],
        [ 12.1588,   4.3364,  -8.0867],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.3542,   0.6934,  -3.3840],
        [  9.5018,  -7.1945,  -1.1892],
        [ 11.1469,   6.9237,   1.6160],
        [ -5.1377,   6.7751,  -0.4438],
        [  3.6703,   6.3443,  -5.4401],
        [ -9.1623,  -3.4288,  -4.9961],
        [  0.7145,  -4.8071,  -2.3766],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.6319,  11.9066,   3.5165],
        [ -1.0482, -10.8487,  -5.9066],
        [  6.3087,   1.8053,  -1.9481],
        [ -4.3963,   7.1629,   4.5263],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -5.9940,   6.1985,   5.9566],
        [ -0.9715, -10.8630,   0.3555],
        [  0.3354,  -4.7708,  -4.9125],
        [ -1.5875,   7.8944,  -9.4286],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.0608,   3.9362,  -5.9567],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.8154,   2.4611,   2.5198],
        [  7.1697,  11.5343,  -0.7276],
        [ -8.3062,  -5.6008,   0.0287],
        [ 10.1503,   8.5414,  -7.2681],
        [  5.1519, -10.2387,   9.8191],
        [ -1.9600,   7.8577,   6.9121],
        [  5.4426, -10.1181,  -5.1268],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.1666,   3.7653,   3.6075],
        [  0.6363, -11.0348,   6.1414],
        [ 12.2921,  -1.7902,  -4.0870],
        [  6.1685,  12.1979,  -6.2473],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.1064,  13.5607,   6.7605],
        [  6.3266,   0.3292,   9.3502],
        [  1.7500,   7.4187,   4.4237],
        [  9.9750,  -6.6338,  -1.2440],
        [  4.0078,  -3.8018,  -6.2168],
        [  1.1274,  14.0116,   8.1753],
        [  2.3376,  -4.6231,   9.8436],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -9.4671,  -0.5553,   0.1117],
        [ 12.1751,  -2.2592,  -5.7751],
        [  0.6142,  14.0802,   0.5407],
        [  3.2226,  -4.2802,  -4.0428],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.3453,  -1.5462,   7.0212],
        [  1.7500,   7.4187,   4.4237],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.3339,   7.5627,  -0.9034],
        [  5.1890,   4.7318,   0.9515],
        [  3.7298,  13.3537,  -9.4653],
        [  8.0381,  -8.5690,  -8.9523],
        [  4.2612, -10.5514,  -6.6268],
        [ 10.2470,   8.4054,  -0.1177],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3303,   1.6145,  -8.4083],
        [ 10.5057,  -5.9099,  -5.1599],
        [  5.5294,  -2.1106,   9.0626],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.3367,   7.1676,  -9.5263],
        [ -6.2668,   5.9808,   2.5614],
        [  3.4920,  -4.1366,  -8.3955]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 1.5857e-02, 2.5121e-02,  ..., 1.0149e+00, 6.5320e-01,
         9.2295e-01],
        [1.5857e-02, 0.0000e+00, 1.3417e-02,  ..., 1.0273e+00, 6.5414e-01,
         9.3691e-01],
        [2.5121e-02, 1.3417e-02, 0.0000e+00,  ..., 1.0396e+00, 6.6643e-01,
         9.4807e-01],
        ...,
        [1.0149e+00, 1.0273e+00, 1.0396e+00,  ..., 0.0000e+00, 7.6100e-01,
         1.9458e-01],
        [6.5320e-01, 6.5414e-01, 6.6643e-01,  ..., 7.6100e-01, 0.0000e+00,
         8.0781e-01],
        [9.2295e-01, 9.3691e-01, 9.4807e-01,  ..., 1.9458e-01, 8.0781e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.10360073,  6.46560181,  6.33517349,  6.2322769 ,  6.16512856,
        6.39438044,  6.39520397,  8.66273171, 14.05687815, 12.5085995 ,
        8.83341608,  4.73784222,  6.80559669,  9.36378438, 11.9182956 ,
       10.84437705,  5.21280156,  5.9510489 ,  6.76791483,  7.62011734,
        9.50654967, 12.12448912,  9.17905491,  5.94535529, 12.3489324 ,
        8.09899874,  8.05242591,  9.15253344, 10.37998078,  9.71392179,
        7.54620849, 13.31764181,  6.77777606,  9.10945678,  9.69629   ,
       11.21703654, 10.06850677, 11.05317597, 12.58006038,  5.53255882,
       12.9089559 ,  9.53015376,  6.39187285, 11.9182956 , 13.12218013,
        8.50278896,  7.32950385,  9.78286733,  4.85995002,  9.99211663,
       13.62896593, 10.8992524 ,  6.5618968 ,  8.40445872, 10.52587243,
        8.62260327, 10.90638066,  4.78258092,  8.05242591,  8.09984231,
        8.97052115,  7.78980299,  9.15253344, 13.58101794, 10.01804223,
       13.26590071, 11.46176259,  8.09842704, 11.4890861 ,  4.91038884,
        8.99276564, 11.05317597, 12.4217523 , 13.66896739, 12.19805394,
       13.91198191,  6.33517349,  7.62232307, 11.97949273,  5.52416368,
       14.05687815,  5.1805345 , 12.53285633,  9.48337063, 12.3828957 ,
       14.09357547,  5.35774573,  9.37989631, 12.44177641,  7.62232307,
        5.19577146,  7.67939546,  7.02253235, 13.86480175, 11.74901778,
       11.37939695, 13.25336001, 10.8207297 ,  6.53290663, 12.05395056,
        5.91854397, 12.27835996, 10.61417181,  7.53884104,  8.66273171,
        5.41348121])}
{'P': tensor([[2.7200e-02, 1.1785e-01, 1.1382e-01,  ..., 1.2810e-05, 2.1375e-04,
         8.7859e-03],
        [1.0594e-03, 2.4621e-02, 3.7810e-02,  ..., 1.0176e-05, 1.1574e-04,
         8.7474e-03],
        [6.2406e-04, 3.6421e-03, 2.1917e-02,  ..., 8.5683e-06, 1.0957e-04,
         1.8470e-02],
        ...,
        [2.5411e-03, 1.0960e-03, 7.6022e-04,  ..., 2.1340e-01, 1.9265e-02,
         1.1246e-04],
        [1.6337e-02, 7.0969e-03, 8.2162e-03,  ..., 4.7193e-04, 1.0375e-01,
         2.6285e-03],
        [2.7533e-04, 4.5565e-04, 6.6208e-04,  ..., 2.0740e-05, 1.9484e-04,
         4.9989e-02]]), 'X': tensor([[ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.1834e-02, 4.0785e-02,  ..., 9.7502e-01, 2.8548e-01,
         2.9805e-01],
        [5.1834e-02, 0.0000e+00, 9.1943e-02,  ..., 9.7529e-01, 3.3470e-01,
         3.4988e-01],
        [4.0785e-02, 9.1943e-02, 0.0000e+00,  ..., 9.8648e-01, 2.5261e-01,
         2.5876e-01],
        ...,
        [9.7502e-01, 9.7529e-01, 9.8648e-01,  ..., 0.0000e+00, 9.2438e-01,
         1.0231e+00],
        [2.8548e-01, 3.3470e-01, 2.5261e-01,  ..., 9.2438e-01, 5.3633e-05,
         9.8766e-02],
        [2.9805e-01, 3.4988e-01, 2.5876e-01,  ..., 1.0231e+00, 9.8766e-02,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.10774544,  6.39187285,  6.49538841,  6.5618968 ,  6.46356806,
        6.61211224,  7.24326992, 14.09357547,  6.67108662,  6.29821673,
        5.77219077, 11.97957425, 13.98691426, 13.86480175,  8.24808832,
        5.53255882,  6.48486171, 13.32645552,  5.05989547, 10.91601655,
       11.63383221,  8.38252937,  9.99145096,  6.36004393, 12.78327037,
        6.46356806, 13.96905756,  5.19986797,  4.7663526 , 13.96987806,
        6.21275016, 12.97370371,  7.89745107,  5.43943791, 13.13205357,
       13.38091121, 10.98076873,  6.50409372, 12.53285633,  9.94173708,
        5.05981914,  7.53884104, 13.85158417, 12.27993098,  4.74770899,
       14.10796466,  7.72291946, 13.2778378 ,  8.04483495, 13.89974693,
       12.889695  , 12.05300943, 11.20100795, 13.85158417,  9.85330646,
        7.57946459, 13.1471313 ,  7.54127425,  8.7273569 ,  9.75401449,
        6.21275016,  6.31576264, 11.11969815,  6.1561646 ,  7.54127425,
        6.05113217,  6.5618968 , 11.00694688,  7.39004226,  8.53501777,
        9.17016857, 11.87412439, 13.26590071, 10.06850677, 10.06850677,
       13.99877413,  5.24137909,  9.14973319,  8.06990483,  5.88782913,
        5.56809335,  8.31625171, 11.73848065,  5.19065439, 11.2845608 ,
        6.99097013,  8.8737556 , 11.51557166, 13.32557133, 11.23201619,
        7.02253235,  5.94293185, 13.44593519,  7.96575782,  9.94285511,
        7.822854  ,  6.55800267, 11.24434374,  5.04028538,  7.46248947,
        9.73579738,  6.24187044,  7.5158489 , 10.31613269,  4.7663526 ,
        7.10378919])}
{'P': tensor([[2.6404e-02, 1.8691e-01, 1.7949e-01,  ..., 2.0354e-03, 9.6440e-04,
         1.3825e-05],
        [2.1764e-03, 3.6217e-02, 7.4275e-02,  ..., 4.4762e-03, 8.2855e-04,
         1.8681e-05],
        [1.5892e-03, 8.0820e-03, 3.2893e-02,  ..., 4.3416e-03, 8.8215e-04,
         1.6794e-05],
        ...,
        [9.5401e-05, 1.0500e-04, 1.0594e-04,  ..., 5.4281e-02, 3.2991e-05,
         2.8560e-04],
        [2.1247e-03, 2.3841e-03, 3.6580e-03,  ..., 6.3712e-04, 6.2381e-02,
         1.4333e-05],
        [1.7539e-04, 1.1461e-04, 9.7299e-05,  ..., 6.4373e-06, 5.3191e-05,
         2.8564e-02]]), 'X': tensor([[  6.0560,  -0.9421,  -5.0334],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3205,   1.7068,  -3.8619],
        [  6.3587,   1.1596,  -3.6629],
        [  3.9515,  -3.8425,   6.1729],
        [  9.0004,  -7.7195,   4.6945],
        [  6.0467,  -9.8369,  -6.5816],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.5900,   2.9767,   7.4811],
        [  1.1274,  14.0116,   8.1753],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.3604,   0.9042,  -6.5901],
        [ -8.3653,   3.4184,   9.3454],
        [ 12.3357,   3.6001,   0.7903],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -0.1626, -10.9796,   6.4629],
        [  4.0591,   6.0202,  -1.6339],
        [  0.7145,  -4.8071,  -2.3766],
        [ -4.8665,  -9.3204,  -5.2564],
        [  6.3606,   0.9204,   3.2525],
        [  8.5093,  -8.1764,  -3.2381],
        [ -2.2188,   7.8218,   8.6064],
        [  6.2036,   2.4030,  -1.2500],
        [ -6.4822,  -8.0252,  -3.8983],
        [  1.8447,  -4.7372,   8.1235],
        [  1.9312,  13.8649,   8.3563],
        [ -1.7009,   7.8851,  -8.2437],
        [  3.4604,  -4.1545,   1.8091],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -0.0282,   7.8575,  -1.1746],
        [  7.1441,  11.5529,   2.3232],
        [ -1.8367,   7.8718,  -8.6523],
        [  3.7121,   6.3117,   0.6738],
        [ 11.0438,   7.1180,  -5.6991],
        [  4.2339,  -3.6268,   6.3125],
        [  5.0233,   4.9647,  -3.1788],
        [ -7.7239,   4.4298,   3.1216],
        [  2.5636, -10.9293,   9.6519],
        [  4.1537,   5.9340,  -6.0423],
        [ 11.0275,  -5.0666,   8.4154],
        [  3.1892, -10.8220,  -6.8409],
        [  4.4290,  -3.4596,   1.4181],
        [ -0.6323,   7.9085,  -5.9544],
        [  8.0381,  -8.5690,  -8.9523],
        [ 11.2571,   6.7072,   8.3693],
        [  3.6249, -10.7255,   5.6605],
        [  6.3246,   1.6696,  -0.1544],
        [ 10.0400,   8.6922,  -6.2847],
        [ -9.4731,  -0.7120,  -6.0271],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.5626,  -2.0536,   8.4433],
        [  0.8572,  14.0501,  -8.3364],
        [ -7.6650,   4.5092,   0.2520],
        [  0.3715,  -4.7756,   0.2312],
        [  1.0093,   7.6567,   2.3526],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.1041, -11.0045,   2.4311],
        [  8.5716,  10.3596,   6.8818],
        [ -4.2271,   7.2387,  -1.8316],
        [ -5.1046,  -9.1610,   2.3418],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -0.5739,   7.9056,   0.6382],
        [  2.6613,   7.0021,   4.5783],
        [  3.1271, -10.8343,   2.6442],
        [  4.2096,   5.8815,   1.9230],
        [  4.9531,   5.0576,   5.6794],
        [  0.7958,  -4.8109,  -8.2248],
        [  1.7218,  -4.7576,  -3.3617],
        [  6.2038,   2.4020,  -3.1692],
        [  8.2422,  -8.4041,  -1.4962],
        [  0.1382, -11.0073,   1.0026],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -9.3419,   0.5972,  -1.2841],
        [  2.8474,   6.8979,   6.6457],
        [  2.3261,  13.7749,   5.9121],
        [  6.0047,   3.1062,  -6.7914],
        [  0.9036,   7.6840,   9.9237],
        [  6.6319,  11.9066,   3.5165],
        [ 11.3691,   6.4760,   1.0787],
        [ -6.4822,  -8.0252,  -3.8983],
        [  3.6703,   6.3443,  -5.4401],
        [ -9.4645,  -1.4880,  -2.0226],
        [  4.4290,  -3.4596,   1.4181],
        [  2.5813,  -4.5471,   5.3597],
        [  5.7963,  -1.6018,   1.8856],
        [ -4.2271,   7.2387,  -1.8316],
        [  3.5965,  -4.0754,  -3.5035],
        [  3.7237,  -3.9966,  -5.0376],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.1667,  -2.6543,   1.2879],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.6523,   3.9443,  -2.9449],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.1737,   2.5305,   3.9049],
        [  6.3122,  -9.6994,  -6.3282],
        [ -3.5173, -10.0623,   4.8280],
        [  0.3724,   7.7974,  -6.8722],
        [  6.0890,  -0.8382,   2.9166],
        [ -8.0750,  -6.0072,  -5.8267]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0148, 0.0176,  ..., 0.2533, 0.5065, 1.0138],
        [0.0148, 0.0000, 0.0312,  ..., 0.2526, 0.5204, 1.0171],
        [0.0176, 0.0312, 0.0000,  ..., 0.2639, 0.4892, 1.0196],
        ...,
        [0.2533, 0.2526, 0.2639,  ..., 0.0000, 0.6553, 0.7749],
        [0.5065, 0.5204, 0.4892,  ..., 0.6553, 0.0000, 1.1960],
        [1.0138, 1.0171, 1.0196,  ..., 0.7749, 1.1960, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.12886024,  6.27905292,  6.33075682,  6.60752373,  6.54693987,
        6.46356806,  5.51177501, 11.85743425, 11.54672755,  8.19703512,
        9.09119351, 14.05687815, 13.27994304,  6.42439344,  9.0368458 ,
       12.85032836,  9.97964013, 10.98076873,  7.26074925,  4.85995002,
       10.51437947,  6.42689705, 11.80092578,  8.1303945 ,  6.652742  ,
       10.31613269,  5.08372149, 13.99877413,  8.06643487,  5.40689332,
       13.94008948, 10.06407839,  7.8575648 , 13.58333857,  8.08320235,
        7.32233733, 13.13888841,  5.5748624 ,  7.06272256,  8.90405823,
       11.22596994,  7.24326992, 12.13568598, 11.28215429,  5.62005629,
        7.9337647 , 11.74901778, 13.103709  , 11.32149259,  6.54127431,
       13.27994304,  9.4997987 ,  6.1561646 ,  5.92952538, 14.07623266,
        8.89301516,  4.79003361,  7.72291946, 10.1371552 , 12.41012046,
       11.0050379 , 13.44593519,  8.38252937, 10.48721382, 12.91143491,
       13.1399181 ,  8.95983586,  9.43737523,  7.92644746,  7.49076899,
       11.27656488,  7.23276582,  7.07902036,  4.8763135 ,  5.05963217,
        6.65259775, 11.77124373, 11.00812996, 10.54658597,  9.36093562,
        7.46248947, 13.96987806,  6.76056625,  7.73692436, 13.62896593,
       13.08414689, 10.31613269,  7.32950385,  9.58072286,  5.62005629,
        5.2287058 ,  6.01356004,  8.38252937,  5.43544626,  5.46244988,
       12.97370371,  5.80860165,  9.62958973,  6.89245352, 10.06023665,
        6.67218687, 11.57249984, 10.65929968,  7.80625981,  6.1463896 ,
       10.06437598])}
{'P': tensor([[2.4499e-02, 1.8051e-01, 1.6039e-01,  ..., 5.2727e-06, 4.9350e-05,
         5.0438e-06],
        [2.1523e-03, 5.0847e-02, 1.7292e-01,  ..., 1.2588e-05, 1.1474e-04,
         1.0304e-05],
        [2.0296e-03, 1.1305e-02, 4.9548e-02,  ..., 1.3279e-05, 1.2631e-04,
         1.0578e-05],
        ...,
        [2.8119e-05, 1.8658e-05, 1.9014e-05,  ..., 5.3833e-02, 1.6654e-05,
         3.3269e-01],
        [2.5988e-05, 3.4639e-05, 3.9174e-05,  ..., 2.6701e-04, 8.6273e-02,
         6.2827e-05],
        [1.5602e-04, 9.3029e-05, 9.3381e-05,  ..., 3.1518e-04, 2.8437e-05,
         9.4791e-02]]), 'X': tensor([[ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0038, 0.0090,  ..., 1.1311, 0.9181, 1.1099],
        [0.0038, 0.0000, 0.0074,  ..., 1.1335, 0.9177, 1.1126],
        [0.0090, 0.0074, 0.0000,  ..., 1.1289, 0.9103, 1.1087],
        ...,
        [1.1311, 1.1335, 1.1289,  ..., 0.0000, 0.8372, 0.1302],
        [0.9181, 0.9177, 0.9103,  ..., 0.8372, 0.0000, 0.9270],
        [1.1099, 1.1126, 1.1087,  ..., 0.1302, 0.9270, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.14086582,  6.36004393,  6.383897  ,  6.42439344,  6.38968357,
        6.34059021,  7.37902766, 13.61801526,  5.10446653, 11.24434374,
        5.91854397, 12.71843462,  8.27853076,  7.33350642,  9.562174  ,
       12.25806559,  6.34059021, 10.6910768 ,  9.58072286,  5.94541803,
        7.16058587,  8.38252937, 12.97370371, 10.9036226 , 12.21532282,
        7.61088125, 12.32408516,  5.41348121,  7.06169391, 12.62711252,
        7.62011734,  9.18690018,  6.12886024,  8.98765644,  5.5205189 ,
        6.99097013, 13.44593519,  4.73784222, 10.09979747,  9.62958973,
        6.99097013,  6.24614905,  6.59176018,  6.06416957,  6.97106912,
        8.14433542, 10.33958639,  9.60080639,  7.62509582, 10.9036226 ,
        7.69752262,  7.58668144,  8.50268322, 11.018084  ,  6.81084724,
       12.00995898,  6.28835101, 12.12641969, 11.35345345,  4.91336994,
        7.5984809 ,  6.65368551,  8.30125201,  9.77777346,  7.17789666,
       13.01728313,  4.79003361, 13.72594022, 12.889695  , 12.35625694,
       12.41012046,  8.16714836,  8.05668609, 11.07637531, 12.27835996,
       13.54774355, 10.98076873,  6.42689705,  8.10886021, 10.46739073,
        6.06416957, 10.69963954,  8.62329466, 11.73567759, 11.54071859,
        5.14999799, 12.05300943, 10.31886066,  6.81961996, 11.37939695,
        8.41148722,  5.18047638,  9.36256096, 11.72749923,  8.1137452 ,
        8.62162372, 10.19036081, 11.785566  , 13.58333857, 13.88683436,
        8.14433542,  8.41148722, 11.51811603,  9.6282763 ,  7.19534146,
       10.06850677])}
{'P': tensor([[2.9195e-02, 1.7639e-01, 1.7289e-01,  ..., 3.2106e-04, 2.8192e-05,
         3.7599e-04],
        [3.5174e-03, 4.7691e-02, 2.0421e-01,  ..., 4.1896e-04, 3.3162e-05,
         4.4262e-04],
        [3.1619e-03, 6.8786e-03, 8.5924e-02,  ..., 6.0555e-04, 4.2603e-05,
         3.6287e-04],
        ...,
        [5.8431e-02, 3.9976e-02, 2.7609e-02,  ..., 1.1573e-01, 8.2228e-03,
         2.1827e-03],
        [1.7160e-02, 1.1959e-02, 7.4919e-03,  ..., 7.4208e-03, 1.0720e-01,
         1.0552e-02],
        [2.2384e-04, 4.0430e-04, 4.2735e-04,  ..., 6.7784e-06, 7.2201e-07,
         1.0768e-02]]), 'X': tensor([[  6.0890,  -0.8382,   2.9166],
        [  6.2770,  -0.0379,   2.4195],
        [  6.3606,   0.9204,   3.2525],
        [  6.1741,  -0.5329,   3.1913],
        [  6.2902,   0.0451,   3.7116],
        [  6.2905,   1.9366,   2.1169],
        [ -7.0948,   5.1940,   9.8171],
        [ 10.0400,   8.6922,  -6.2847],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.0345,   3.0175,  -1.4424],
        [  2.1379,  -4.6757,   6.4981],
        [  1.8721,  -4.7322,   0.5524],
        [  6.3305,   1.6119,   1.1538],
        [  2.1504,  -4.6726,   8.7908],
        [  8.9284,  10.0036,  -6.9210],
        [  9.2692,   9.6370,   7.0746],
        [  0.5600, -11.0321,  -0.8660],
        [ 11.7349,  -3.5838,   2.1644],
        [ -3.6381,   7.4683,   7.3544],
        [ -6.7643,  -7.7372,  -9.6663],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 11.2767,  -4.6006,   0.7868],
        [  3.7992,  -3.9474,  -8.9019],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -2.3317,   7.8034,  -3.6274],
        [  4.2612, -10.5514,  -6.6268],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.2517,  -4.6495,   2.9800],
        [  4.2041,  -3.6509,  -4.9488],
        [  3.1892, -10.8220,  -6.8409],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.0582,  -2.6640,   0.7132],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.5829,  -4.5466,  -8.4015],
        [ -3.4986,   7.5152,   9.0712],
        [  4.8432,   5.1968,   1.5006],
        [ -8.5607,  -5.0938,  -5.4531],
        [  7.3724,  -9.0577,  -8.9843],
        [  8.7131,  10.2216,   7.7915],
        [  6.1730,  -0.5375,  -8.8861],
        [  4.4633,   5.6287,  -0.7754],
        [ -8.8745,   2.3097,   3.5600],
        [  3.6703,   6.3443,  -5.4401],
        [  2.1625,  -4.6697,   7.0658],
        [  4.9531,   5.0576,   5.6794],
        [  4.4039,   5.6903,   8.7366],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.6112,   2.9319,  -7.9943],
        [  1.9204,   7.3518,   8.4781],
        [  6.3301,   0.3638,  -7.5685],
        [  3.7298,  13.3537,  -9.4653],
        [ 12.3759,  -1.3943,   6.4659],
        [ -1.6906,   7.8860,   7.7142],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.8835,  -4.3178,   9.3080],
        [ 11.0864,   7.0387,  -0.1238],
        [ -5.3179,  -9.0101,   9.4472],
        [  1.2728, -11.0372,   9.8121],
        [  2.2182,  13.8007,   5.1889],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.8400,   2.3991,   1.3457],
        [ -3.1204, -10.2345,   9.2288],
        [  6.0890,  -0.8382,   2.9166],
        [  1.1994,   7.6035,   2.4295],
        [  3.2159, -10.8166,  -1.0406],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 11.3367,  -4.4806,   0.4643],
        [  5.6258,  -1.9404,  -6.0715],
        [  0.1287, -11.0065,   6.5757],
        [  0.6470, -11.0352,  -1.1931],
        [  5.8323,  -1.5228,   5.1132],
        [  7.7794,  11.0629,   8.9069],
        [ -5.0132,  -9.2234,  -8.3531],
        [  0.6363, -11.0348,   6.1414],
        [  0.9967,  -4.8144,  -9.7370],
        [  1.7301,   7.4262,   5.0522],
        [ -9.1231,  -3.5739,   8.4708],
        [ 12.2589,  -1.9312,   9.8657],
        [ 11.3691,   6.4760,   1.0787],
        [ -6.7745,  -7.7264,   8.4684],
        [ -8.8400,   2.3991,   1.3457],
        [  6.2630,   2.1057,  -5.2136],
        [  0.4996,   7.7738,  -4.4711],
        [  5.3924, -10.1396,  -4.0863],
        [  6.3413,   0.4883,  -6.8416],
        [  1.6422,   7.4586,  -6.8056],
        [ -4.7334,   6.9982,  -0.4392],
        [  5.8889,  -1.3916,   8.8993],
        [  9.0652,   9.8597,  -3.5949],
        [ -8.3539,  -5.5111,   0.3373],
        [ -7.4807,   4.7463,   2.0616],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.7735,   2.5640,  -6.3313],
        [  5.6414, -10.0303,   7.8525],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.0608,   3.9362,  -5.9567],
        [  5.3804,  -2.3504,   7.2744],
        [  1.3670,  -4.7990,   3.2603],
        [  3.1271, -10.8343,   2.6442],
        [ 12.0845,  -2.5772,  -4.6073]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0315, 0.0455,  ..., 0.1057, 0.5700, 0.5963],
        [0.0315, 0.0000, 0.0767,  ..., 0.1143, 0.5730, 0.5648],
        [0.0455, 0.0767, 0.0000,  ..., 0.1169, 0.5767, 0.6412],
        ...,
        [0.1057, 0.1143, 0.1169,  ..., 0.0000, 0.4646, 0.6181],
        [0.5700, 0.5730, 0.5767,  ..., 0.4646, 0.0000, 0.8475],
        [0.5963, 0.5648, 0.6412,  ..., 0.6181, 0.8475, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.1463896 ,  6.27715105,  6.42689705,  6.19707944,  6.29034958,
        6.58184474,  8.79284466, 13.27994304, 13.00413829,  6.74686071,
        5.14124612,  5.08910491,  6.53250937,  5.14371005, 13.40850328,
       13.37121923, 11.04629471, 12.26997165,  8.30729352, 10.27716434,
        5.56809335,  9.78147542, 12.17901033,  5.47866221, 10.0674887 ,
        8.14433542, 11.37939695,  9.43737523,  5.24137909,  5.23878371,
       12.17451685,  5.56809335, 11.28215429,  6.69389431,  9.61942445,
       12.3489324 ,  8.06990483,  5.22903117,  8.28968067,  7.10378919,
        9.96152788, 11.67875855, 13.43127131,  6.19633139,  7.18354885,
        9.17016857,  7.32950385,  5.14607014,  7.07902036,  7.19534146,
        4.91038884,  9.09661212,  7.5984809 ,  6.34059021, 13.86480175,
       12.45418236,  8.06517143, 12.41366754, 13.66896739,  9.5642993 ,
        9.87720102, 13.13205357, 10.46238054, 11.11038551, 13.97779839,
       12.62576026,  9.15976746, 10.69963954,  6.1463896 ,  7.69752262,
       11.2845608 , 12.49317563, 12.18998195,  5.9510489 , 11.00726483,
       11.05413545,  6.02779201, 13.52430176, 10.4977132 , 11.05317597,
        4.91652556,  7.62509582,  9.79814997, 12.41012046, 13.08414689,
       10.27572673,  9.15976746,  6.60752373,  7.78980299, 11.48434767,
        6.36004393,  7.63726134,  8.4486771 ,  6.05113217, 13.39372426,
       10.00794468,  8.85939626,  8.41148722,  9.14044984, 11.50791859,
        5.53255882,  8.97052115,  5.87132718,  4.98989906, 11.27656488,
       12.35625694])}
{'P': tensor([[2.0670e-02, 1.3158e-01, 1.0817e-01,  ..., 5.6187e-04, 2.6061e-04,
         9.3506e-02],
        [1.5452e-03, 2.7163e-02, 8.7451e-02,  ..., 3.0976e-04, 5.7186e-04,
         1.3828e-01],
        [6.4384e-04, 1.8864e-03, 1.8460e-02,  ..., 1.0097e-04, 4.4486e-04,
         1.8321e-01],
        ...,
        [1.7360e-02, 3.5885e-02, 3.7619e-02,  ..., 1.4211e-01, 2.6387e-04,
         4.3118e-02],
        [3.2938e-05, 3.3832e-05, 3.0860e-05,  ..., 2.3962e-06, 3.7820e-02,
         3.2486e-05],
        [8.3855e-04, 1.5830e-03, 1.9418e-03,  ..., 1.1758e-04, 9.8660e-04,
         3.1489e-02]]), 'X': tensor([[  6.1066,  -0.7799,  -4.1746],
        [  6.3236,   0.3011,  -4.6668],
        [  6.3587,   1.1596,  -3.6629],
        [  6.3542,   0.6934,  -3.3840],
        [  6.3205,   1.7068,  -3.8619],
        [  6.2576,   2.1359,  -3.3398],
        [  0.1497,   7.8334,  -5.4412],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.2036,   2.4030,  -1.2500],
        [  2.6321,  -4.5296,  -2.5159],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.2644,  -4.6434,   9.7508],
        [ -6.6998,   5.5954,   0.0823],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.2373,  -3.1225,  -8.9763],
        [  6.1730,  -0.5375,  -8.8861],
        [ -2.9497, -10.3028,  -7.3479],
        [ -9.0961,   1.6529,   3.1553],
        [ -5.1974,   6.7396,   3.3484],
        [  5.6928,  12.4710,   3.6001],
        [  3.4401,  13.4540,   4.6576],
        [  4.5800,   5.5034,  -3.9432],
        [  8.5716,  10.3596,   6.8818],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.3048,   0.1469,  -9.1691],
        [  5.1292,   4.8183,  -5.7997],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -2.1382,   7.8339,   7.3524],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.4505,  -1.7081,  -0.7039],
        [  0.9478,  -4.8144,   0.6371],
        [  4.0078,  -3.8018,  -6.2168],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.7590,  13.6621,   9.4496],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.0330,  -6.0765,   6.7945],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -7.3333,   4.9246,  -7.9212],
        [  6.2419,  -0.2313,   4.2440],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.4203,  -2.0340,   2.5856],
        [ -7.0948,   5.1940,   9.8171],
        [  1.1994,   7.6035,   2.4295],
        [ -9.4773,  -1.0024,   5.2599],
        [  9.3836,   9.5075,  -5.9278],
        [  4.2275,  -3.6320,  -0.2714],
        [  1.0391,  -4.8141,  -0.0768],
        [  4.5800,   5.5034,  -3.9432],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.6357,   5.4415,   7.9067],
        [  3.9895,   6.0816,  -6.5235],
        [  2.5382,   7.0672,   4.5097],
        [  6.7565,  11.8237,  -4.9882],
        [ -1.9487,   7.8590,  -3.3330],
        [ -7.8126,   4.3066,   5.1791],
        [ 12.2592,   3.9425,   2.0383],
        [  1.0093,   7.6567,   2.3526],
        [ -3.7964,  -9.9294,   7.7147],
        [  3.5965,  -4.0754,  -3.5035],
        [ -5.9872,   6.2037,  -9.4477],
        [ -4.9217,   6.8979,  -1.7667],
        [ -2.8162,   7.7059,  -9.6885],
        [  5.9026,   3.3837,  -0.6573],
        [  5.4167, -10.1293,   6.2156],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.1960,  -3.2964,   1.6921],
        [  2.5813,  -4.5471,   5.3597],
        [ -6.8466,  -7.6488,   1.9648],
        [ -5.1974,   6.7396,   3.3484],
        [ -8.4297,  -5.3637,  -9.3962],
        [  6.3542,   0.6934,  -3.3840],
        [ -5.9872,   6.2037,  -9.4477],
        [  0.6470, -11.0352,  -1.1931],
        [  6.3549,   0.7096,   6.9198],
        [  9.0004,  -7.7195,   4.6945],
        [  0.5960, -11.0335,   3.1144],
        [ -8.8631,  -4.3728,  -1.1746],
        [  2.9260,  -4.4162,   8.3808],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.0039,   7.3173,  -1.5885],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.9314,  -9.3435,  -1.2729],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.0608,   3.9362,  -5.9567],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3497,   1.3680,  -1.9076],
        [  6.2038,   2.4020,  -3.1692],
        [  4.5767,   5.5071,   3.9069],
        [ -8.5149,  -5.1905,   1.5239],
        [  1.6408,   7.4591,  -4.2294],
        [  8.2422,  -8.4041,  -1.4962],
        [  4.2612, -10.5514,  -6.6268],
        [ -9.3616,  -2.4749,   0.1793],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.1824,  -4.6647,   1.1985],
        [ 11.7795,  -3.4697,   3.5624],
        [ -9.4283,  -1.9585,  -0.9129],
        [  1.9503,  -4.7172,  -6.2313],
        [  0.9945,  -4.8144,  -0.1897],
        [  1.4306,  13.9620,  -0.2075],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.3205,   1.7068,  -3.8619]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0201, 0.0215,  ..., 0.2621, 0.4677, 0.0110],
        [0.0201, 0.0000, 0.0408,  ..., 0.2423, 0.4628, 0.0304],
        [0.0215, 0.0408, 0.0000,  ..., 0.2831, 0.4815, 0.0105],
        ...,
        [0.2621, 0.2423, 0.2831,  ..., 0.0000, 0.4213, 0.2727],
        [0.4677, 0.4628, 0.4815,  ..., 0.4213, 0.0000, 0.4752],
        [0.0110, 0.0304, 0.0105,  ..., 0.2727, 0.4752, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.1561646 ,  6.33075682,  6.46356806,  6.39187285,  6.54693987,
        6.61211224,  7.83486943, 11.9120321 ,  6.652742  ,  5.23878371,
       11.74901778,  5.16611782,  8.72896381,  9.23071698,  9.75075273,
        6.19633139, 10.7167285 ,  9.24501829,  8.51088913, 13.70893492,
       13.88683436,  7.15990676, 13.44593519,  9.99211663,  6.30647746,
        7.03733775, 10.0674887 ,  8.12043983,  6.78325151,  9.60359158,
        4.90676242,  5.52416368,  8.20436764, 13.93790235, 11.97238624,
       10.07238463,  9.51074763,  8.83341608,  6.24614905,  8.36995953,
        9.63742669,  8.79284466,  7.69752262,  9.53015376, 13.35832725,
        5.57341623,  4.92497514,  7.15990676,  5.8828986 ,  7.14837514,
        7.27340834,  7.50917551, 13.61801526,  8.09704052,  8.92098752,
       12.87752052,  7.72291946, 10.63038565,  5.43544626,  8.62162372,
        8.47371868,  8.20436764,  6.80370254, 11.48663767, 10.92597196,
        8.95983586,  9.76896904,  5.2287058 , 10.26545792,  8.51088913,
        9.99145096,  6.39187285,  8.62162372, 11.05413545,  6.39438044,
       11.85743425, 11.04953767,  9.88310614,  5.29755487,  5.53255882,
        7.37902766,  9.58347456,  7.58668144,  5.9510489 , 11.63383221,
        9.47369628,  8.97052115,  6.30647746,  6.49538841,  6.65259775,
        7.16058587,  9.97220799,  7.6374603 , 11.77124373, 11.37939695,
        9.68323686, 12.7125079 ,  5.14999799, 12.27991991,  9.62958973,
        5.10444097,  4.9160962 , 14.03505624, 12.27835996,  8.29203356,
        6.54693987])}
{'P': tensor([[4.3300e-02, 2.6617e-01, 2.5879e-01,  ..., 1.3171e-06, 3.7535e-06,
         3.0878e-04],
        [2.9214e-03, 4.9222e-02, 2.9076e-01,  ..., 1.3447e-06, 3.8510e-06,
         2.4273e-04],
        [2.4597e-03, 5.2938e-03, 7.3572e-02,  ..., 2.5656e-06, 7.1041e-06,
         4.0176e-04],
        ...,
        [8.2680e-06, 5.2827e-06, 5.1835e-06,  ..., 1.0838e-01, 5.1625e-02,
         9.0779e-05],
        [7.8932e-06, 4.8688e-06, 4.5072e-06,  ..., 1.6285e-03, 2.5245e-02,
         3.3638e-05],
        [1.5526e-04, 1.6464e-04, 2.5198e-04,  ..., 3.9946e-05, 4.4880e-05,
         5.7995e-02]]), 'X': tensor([[  6.1222,  -0.7261,   8.8204],
        [  6.3266,   0.3292,   9.3502],
        [  6.3583,   1.1729,   8.7835],
        [  6.3119,   1.7797,   9.9186],
        [  6.3549,   0.7096,   6.9198],
        [  5.6248,   3.9984,   8.4933],
        [  5.8930,  12.3592,  -8.9201],
        [ 10.9916,   7.2133,  -0.9179],
        [ 11.3800,  -4.3919,   1.4066],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.4009,  -4.1874,  -5.9751],
        [  4.4039,   5.6903,   8.7366],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.4541,  -0.3327,   8.8246],
        [  9.5018,  -7.1945,  -1.1892],
        [ 11.4495,   6.3026,   6.8257],
        [  3.6154,  -4.0640,  -7.9698],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -1.9600,   7.8577,   6.9121],
        [ -3.4488, -10.0934,   2.5195],
        [  7.8982,  -8.6778,   0.0239],
        [  4.5491,  -3.3485,   6.8412],
        [ -2.4956, -10.4682,   3.5632],
        [  2.5636, -10.9293,   9.6519],
        [  5.3602,  -2.3808,   7.2098],
        [  3.3943,   6.5483,   1.0790],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.2790,  -0.0259,  -5.4116],
        [ -9.3444,   0.5821,   6.9101],
        [ -7.5720,   4.6310,   2.1717],
        [ -6.3306,   5.9272,   5.6935],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.4159,  -2.0739,   9.6841],
        [ -3.3812, -10.1235,   7.4384],
        [  5.8889,  -1.3916,   8.8993],
        [  9.9756,  -6.6330,  -3.3429],
        [ 11.7418,   5.6070,  -1.4192],
        [  3.5939,   6.4029,  -1.8357],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.7409,  -4.6837,  -6.5840],
        [  1.5246,  -4.7838,  -5.2095],
        [ 12.5199,  -0.4711,   3.0436],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.6587,   9.1817,  -0.8212],
        [  6.2770,  -0.0379,   2.4195],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.0611,  -6.0302,   4.6962],
        [  4.0421,  -3.7765,  -8.3631],
        [ -7.0253,  -7.4493,  -1.5978],
        [  5.5158,  -2.1337,   3.5246],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.1625,  -4.6697,   7.0658],
        [ 11.9936,  -2.8677,   6.0614],
        [ -2.5056,   7.7720,   5.5878],
        [  1.4306,  13.9620,  -0.2075],
        [  4.4039,   5.6903,   8.7366],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.9282,   2.1648,   8.4894],
        [  6.8407,  11.7666,   3.0045],
        [  6.3301,   0.3638,  -7.5685],
        [  2.7558,  -4.4844,   0.0636],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.1006,   7.6318,  -2.4877],
        [ 12.0772,  -2.6014,  -7.1832],
        [  3.2990,  -4.2415,  -7.4857],
        [  9.2979,   9.6048,   7.3138],
        [  0.9809,  -4.8145,   6.8706],
        [  6.3246,   1.6696,  -0.1544],
        [  1.1274,  14.0116,   8.1753],
        [  4.5800,   5.5034,  -3.9432],
        [  5.7802,  -1.6363,  -6.2782],
        [  3.8111, -10.6787,  -4.6539],
        [  8.1043,  -8.5164,  -7.9290],
        [ -6.4002,  -8.1048,  -0.9108],
        [  4.0457,  -3.7737,  -0.4704],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.5128, -10.9365,   9.9645],
        [  0.9478,  -4.8144,   0.6371],
        [ -3.4100,   7.5436,  -7.6980],
        [ -7.6719,  -6.6260,   7.1468],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -4.6942,  -9.4299,  -9.1846],
        [  7.9403,  -8.6455,  -8.7433],
        [  6.3460,   1.4252,   1.2291],
        [ -8.3900,   3.3727,   3.1650],
        [  9.3836,   9.5075,  -5.9278],
        [  4.1537,   5.9340,  -6.0423],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.2275,  -3.6320,  -0.2714],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.1755,   2.5233,  -2.0042],
        [  2.3928,   7.1404,  -1.7035],
        [ -1.5875,   7.8944,  -9.4286],
        [  2.8474,   6.8979,   6.6457],
        [ -9.4739,  -1.2497,   0.5483],
        [  5.7151,  -1.7700,   3.1542],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.0729,   7.2878,   7.0880],
        [  4.5814, -10.4485,   2.5496],
        [ -9.4739,  -1.2497,   0.5483],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.3205,   1.7068,  -3.8619]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0095, 0.0105,  ..., 1.5244, 1.3969, 0.7873],
        [0.0095, 0.0000, 0.0134,  ..., 1.5294, 1.4013, 0.7964],
        [0.0105, 0.0134, 0.0000,  ..., 1.5161, 1.3881, 0.7854],
        ...,
        [1.5244, 1.5294, 1.5161,  ..., 0.0000, 0.1664, 1.0762],
        [1.3969, 1.4013, 1.3881,  ..., 0.1664, 0.0000, 1.0233],
        [0.7873, 0.7964, 0.7854,  ..., 1.0762, 1.0233, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.16512856,  6.33517349,  6.46560181,  6.55800267,  6.39438044,
        6.90115901, 13.6922425 , 13.1471313 , 12.19805394,  6.10774544,
        5.39450305,  7.19534146,  4.78002665,  9.45995955, 11.9182956 ,
       13.06957425,  5.43943791,  9.96152788,  8.09842704, 10.66631784,
       11.73397309,  5.64865173, 10.76154256, 11.22596994,  5.86519525,
        7.37576154, 10.61047596,  6.27905292,  9.36256096,  8.87587457,
        8.67227386,  6.78325151,  9.23071698,  9.64157461, 10.67322527,
        6.05113217, 11.97957425, 13.01187484,  7.34250948, 12.44177641,
        9.91668182,  5.0209005 , 12.52875793,  9.68324553, 13.32645552,
        6.27715105,  8.8737556 , 10.06703358,  5.53175275, 10.23946247,
        5.91408504, 12.24467041,  5.14607014, 12.33167382,  8.16584537,
       14.03505624,  7.19534146, 12.21532282,  9.18690018, 13.6105595 ,
        6.34059021,  5.26342203, 12.04924159,  7.71075226, 12.35421461,
        5.37343173, 13.36800073,  4.91336994,  6.54127431, 14.05687815,
        7.15990676,  6.00732193, 11.33837396, 11.75618513, 10.32715715,
        5.53255882, 10.5575601 , 11.22142195,  4.90676242,  8.27853076,
       10.1371552 , 12.25806559, 10.53368644, 11.73848065,  6.50409372,
        9.04255655, 13.35832725,  7.24326992,  8.8353674 , 13.00413829,
        5.57341623, 11.74901778,  6.67108662,  7.53063885,  8.05242591,
        7.46248947,  9.55593374,  5.98285319,  5.35774573,  7.57685908,
       11.4088043 ,  9.55593374,  9.87720102,  9.75075273,  9.85330646,
        6.54693987])}
{'P': tensor([[2.2105e-02, 1.7527e-01, 1.6950e-01,  ..., 1.9779e-05, 1.5048e-05,
         3.1230e-04],
        [1.0842e-03, 2.4675e-02, 5.6297e-03,  ..., 2.7269e-05, 1.1156e-05,
         6.8498e-04],
        [1.1942e-03, 8.0944e-02, 2.3802e-02,  ..., 2.3007e-05, 1.1168e-05,
         6.1918e-04],
        ...,
        [1.1187e-05, 1.2958e-05, 1.1673e-05,  ..., 1.0203e-01, 2.9640e-06,
         3.4148e-05],
        [6.8866e-03, 3.7564e-03, 3.9658e-03,  ..., 4.7423e-06, 7.5870e-02,
         3.0184e-06],
        [7.3316e-05, 9.2638e-05, 9.0219e-05,  ..., 1.4218e-04, 7.8502e-06,
         5.5416e-01]]), 'X': tensor([[  6.1579,  -0.5961,  -6.5595],
        [  6.3515,   0.6418,  -6.5823],
        [  6.3413,   0.4883,  -6.8416],
        [  6.3604,   0.9042,  -6.5901],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3031,   1.8477,  -7.1706],
        [  8.4691,  10.4571,  -5.0650],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.2644,  -4.6434,   9.7508],
        [  5.8201,   3.5841,  -8.0609],
        [  2.2182,  13.8007,   5.1889],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -0.6323,   7.9085,  -5.9544],
        [ -0.5709,   7.9055,   7.6089],
        [  2.2659,  13.7894,   6.5436],
        [ 11.1085,  -4.9204,   0.0736],
        [  6.0031,  12.2958,  -6.5992],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.3860,  -2.3418,   1.0826],
        [  3.3633,  13.4794,   0.5193],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.6112,   2.9319,  -7.9943],
        [ -1.6275, -10.7221,   3.2376],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.2373,  -3.1225,  -8.9763],
        [  9.9058,  -6.7203,   5.6928],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.4117,   0.1160,   8.2574],
        [ -3.6381,   7.4683,   7.3544],
        [ -9.4524,  -1.6813,  -6.6862],
        [  7.3592,  11.3937,   6.2407],
        [ -1.9600,   7.8577,   6.9121],
        [  6.3460,   1.4252,   1.2291],
        [  5.0233,   4.9647,  -3.1788],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.8275,  -1.5334,   2.0124],
        [  6.2630,   2.1057,  -5.2136],
        [  4.0587, -10.6112,   6.2273],
        [  3.1480,  -4.3165,  -7.8002],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.0039,   7.3173,  -1.5885],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.1477,   7.2550,   8.9688],
        [ -3.7964,  -9.9294,   7.7147],
        [ -4.8751,  -9.3148,   4.0406],
        [  4.4039,   5.6903,   8.7366],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.2437,  -4.6489,   6.1430],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3360,   0.4258,  -9.2509],
        [  8.8474,  10.0868,  -8.8644],
        [ -1.8788, -10.6567,   3.7068],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.4932,   7.0903,  -5.4214],
        [  0.5600, -11.0321,  -0.8660],
        [ 11.3367,  -4.4806,   0.4643],
        [  2.2659,  13.7894,   6.5436],
        [ -1.5451,   7.8975,   7.6104],
        [  5.1861,   4.7361,  -3.7312],
        [  1.1644, -11.0394,   7.8511],
        [ -3.4488, -10.0934,   2.5195],
        [  5.8979,   3.3958,  -3.8713],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.3331,  -2.6465,  -8.2028],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.2974,   0.8371,   2.4852],
        [  6.2555,   2.1474,   4.1685],
        [ -2.6069, -10.4298,  -6.1995],
        [  9.2692,   9.6370,   7.0746],
        [  5.8201,   3.5841,  -8.0609],
        [  3.9663,   6.1018,  -5.4104],
        [  6.0305,   3.0296,   0.1375],
        [ -1.0407,   7.9166,  -6.6021],
        [ 10.9916,   7.2133,  -0.9179],
        [ -6.4368,   5.8356,   3.3014],
        [  4.3554,   5.7393,  -9.0766],
        [ 11.5083,  -4.1177,   0.2389],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.4235,   6.3594,   5.1663],
        [  0.2480, -11.0153,  -3.0183],
        [  0.9809,   7.6642,  -1.4217],
        [  7.7750,  11.0665,   2.8364],
        [ -9.1433,  -3.5003,  -3.0209],
        [  3.6817,   6.3354,   6.1233],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.3200,   0.2690,   1.5900],
        [ -9.1960,  -3.2964,   1.6921],
        [ 12.4013,  -1.2597,  -7.6474],
        [  2.5010,  -4.5736,  -0.8989],
        [  1.0391,  -4.8141,  -0.0768],
        [ 11.5966,   5.9666,   6.0083],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.2373,  -4.6506,  -1.7334],
        [ -3.0008, -10.2827,  -8.9480],
        [ -8.4267,  -5.3697,   2.4172],
        [  0.2732,  -4.7619,  -1.1577],
        [  8.9284,  10.0036,  -6.9210],
        [ -3.3812, -10.1235,   7.4384],
        [  2.2849, -10.9655,  -7.6542],
        [  0.9809,  -4.8145,   6.8706],
        [ -9.4366,  -0.1189,  -2.6736],
        [  0.9036,   7.6840,   9.9237],
        [  1.5383, -11.0274,  -4.9666],
        [  0.8572,  14.0501,  -8.3364]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0031, 0.0092,  ..., 0.9677, 0.6163, 0.1986],
        [0.0031, 0.0000, 0.0074,  ..., 0.9697, 0.6191, 0.1970],
        [0.0092, 0.0074, 0.0000,  ..., 0.9769, 0.6190, 0.1896],
        ...,
        [0.9677, 0.9697, 0.9769,  ..., 0.0000, 1.0367, 1.1653],
        [0.6163, 0.6191, 0.6190,  ..., 1.0367, 0.0000, 0.6652],
        [0.1986, 0.1970, 0.1896,  ..., 1.1653, 0.6652, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.18668463,  6.383897  ,  6.36004393,  6.42439344,  6.48486171,
        6.56833864, 13.45642379,  8.83341608,  5.16611782,  6.83515201,
       13.97779839, 10.31613269,  7.9337647 ,  7.92607342, 13.97430191,
       12.1494023 , 13.6829843 ,  5.18277858,  5.87305005, 13.89264822,
        8.86082124,  9.09661212, 10.84493732, 12.35421461,  9.75075273,
       11.9702762 ,  6.24614905,  9.41244865,  8.30729352,  9.60080639,
       13.56367857,  8.09842704,  6.50409372,  7.06272256,  4.78258092,
       11.59512194,  6.02588517,  6.60752373, 11.36089355,  5.34250225,
       12.10942412,  7.58668144,  6.02789634,  5.79391825,  7.56617037,
       10.63038565, 10.51340829,  7.19534146, 12.17901033,  5.16205087,
        9.09963894,  6.35029274, 13.41713661, 10.82106505,  9.53015376,
        7.5158489 , 11.04629471, 12.18998195, 13.97430191,  8.04718433,
        7.02326515, 11.10066126, 10.66631784,  6.80559669, 13.6660671 ,
        9.70108409, 11.04629471,  9.3349858 ,  6.61386436, 10.7506423 ,
       13.37121923,  6.83515201,  7.27759233,  6.74872981,  7.98469166,
       13.1471313 ,  8.68832398,  7.20482224, 12.22276278, 11.70168755,
       13.0743399 , 11.018084  ,  7.72668467, 13.5247215 ,  9.79039371,
        7.32754882, 12.84056604,  6.32572323,  9.76896904, 12.46513646,
        5.21280156,  4.92497514, 13.04156075, 10.61417181,  5.16077764,
       10.71162681,  9.99211663,  4.76969364, 13.40850328, 10.67322527,
       11.20100795,  4.91336994,  9.43737523,  7.73692436, 11.1341798 ,
       14.07623266])}
{'P': tensor([[2.6254e-02, 2.1136e-01, 1.9587e-01,  ..., 1.2203e-05, 1.4342e-05,
         1.8194e-04],
        [2.4133e-03, 3.6585e-02, 1.4142e-01,  ..., 1.7215e-05, 2.0169e-05,
         1.5110e-04],
        [2.3663e-03, 6.7093e-03, 4.8963e-02,  ..., 2.6296e-05, 2.7376e-05,
         1.5697e-04],
        ...,
        [4.2934e-05, 3.1739e-05, 2.9476e-05,  ..., 2.7151e-02, 6.3629e-03,
         3.4786e-05],
        [1.1796e-03, 8.6855e-04, 7.0470e-04,  ..., 3.3128e-03, 2.6499e-01,
         6.3914e-04],
        [2.8805e-03, 3.0943e-03, 4.0122e-03,  ..., 7.5259e-07, 5.6201e-07,
         2.1412e-02]]), 'X': tensor([[  6.1730,  -0.5375,  -8.8861],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3505,   0.6235,  -8.5849],
        [  6.3360,   0.4258,  -9.2509],
        [  6.3589,   0.8239,  -8.4230],
        [  6.3303,   1.6145,  -8.4083],
        [  6.0560,  -0.9421,  -5.0334],
        [  1.4306,  13.9620,  -0.2075],
        [ -6.6895,   5.6051,   2.7899],
        [  2.1824,  -4.6647,   1.1985],
        [  5.8979,   3.3958,  -3.8713],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.9204,   7.3518,   8.4781],
        [  5.8323,  -1.5228,   5.1132],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.0845,  -2.5772,  -4.6073],
        [  1.3403,  -4.8011,  -1.1698],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.4808,   4.7462,   2.0266],
        [  3.7121,   6.3117,   0.6738],
        [ -4.7847,  -9.3729,  -3.7741],
        [  3.4401,  13.4540,   4.6576],
        [  6.2630,   2.1057,  -5.2136],
        [  5.6260,  -1.9401,   3.0259],
        [  3.4009,  -4.1874,  -5.9751],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.5158,  -2.1337,   3.5246],
        [ -7.3554,   4.8985,  -0.6665],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4753,  -0.8012,  -5.2050],
        [  8.1217,  -8.5024,  -0.6149],
        [ 10.4758,  -5.9539,   1.3667],
        [  2.9470,  13.6085,   2.9809],
        [ -7.5401,   4.6718,   9.0033],
        [  5.1667,  -2.6543,   1.2879],
        [ -7.3470,  -7.0615,   5.0066],
        [ -8.0087,   4.0173,   9.1091],
        [  1.0000,   7.6591,  -4.5011],
        [  0.5960, -11.0335,   3.1144],
        [ -6.9838,  -7.4965,   0.9796],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.3804,  -2.3504,   7.2744],
        [  2.7976,  -4.4682,  -4.2020],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.0310,  12.2795,  -6.4528],
        [ -7.3470,  -7.0615,   5.0066],
        [  3.2647, -10.8066,  -0.3768],
        [  2.1504,  -4.6726,   8.7908],
        [  9.3607,   9.5337,  -4.5412],
        [ 11.2571,   6.7072,   8.3693],
        [  6.0345,   3.0175,  -1.4424],
        [  0.9478,  -4.8144,   0.6371],
        [  9.0424,  -7.6779,  -7.5286],
        [ -1.5451,   7.8975,   7.6104],
        [  3.3094,  13.4969,  -3.6507],
        [  5.7151,  -1.7700,   3.1542],
        [  3.1892, -10.8220,  -6.8409],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.4572,   7.1085,   7.7376],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.2275,  -3.6320,  -0.2714],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.7544,   2.6096,   3.3146],
        [  6.3569,   0.7591,   6.5011],
        [  6.3205,   1.7068,  -3.8619],
        [  5.4732,  -2.2041,  -4.3830],
        [  1.5653,  13.9377,   4.1773],
        [ -9.4501,  -1.7133,   2.2980],
        [ -1.0407,   7.9166,  -6.6021],
        [ -4.0633,  -9.7927,  -3.0825],
        [  2.6451,  -4.5250,  -0.5565],
        [ -7.6734,   4.4980,  -5.4248],
        [  1.7000,  -4.7609,   2.7935],
        [ -9.1231,  -3.5739,   8.4708],
        [ -4.4779,  -9.5609,  -2.4505],
        [  7.5309,  11.2617,   7.2656],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.1297,  -3.7098,  -3.9838],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.2339,  -3.6268,   6.3125],
        [ -4.7334,   6.9982,  -0.4392],
        [ -6.0844,  -8.3955,   0.7705],
        [  8.8091,  -7.9037,   0.3631],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.4150,  13.9647,  -4.0915],
        [  6.1685,  12.1979,  -6.2473],
        [  1.7000,  -4.7609,   2.7935],
        [  1.1006,   7.6318,  -2.4877],
        [ -8.8835,  -4.3178,   9.3080],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -9.2974,   0.8371,   2.4852],
        [  3.5284, -10.7484,   5.3347],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.2182,  13.8007,   5.1889],
        [  7.0358,  -9.2784,  -9.8479],
        [  6.3551,   0.7149,   1.3344],
        [ -8.8363,   2.4086,  -5.6296],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 10.6673,  -5.6649,  -4.1646]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0079, 0.0048,  ..., 1.0209, 1.0516, 0.1126],
        [0.0079, 0.0000, 0.0127,  ..., 1.0253, 1.0536, 0.1189],
        [0.0048, 0.0127, 0.0000,  ..., 1.0188, 1.0509, 0.1092],
        ...,
        [1.0209, 1.0253, 1.0188,  ..., 0.0000, 0.3267, 0.9147],
        [1.0516, 1.0536, 1.0509,  ..., 0.3267, 0.0000, 0.9631],
        [0.1126, 0.1189, 0.1092,  ..., 0.9147, 0.9631, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.19633139,  6.30647746,  6.38105015,  6.35029274,  6.41202986,
        6.53290663,  6.12886024, 14.03505624,  8.7273569 ,  5.14999799,
        6.80559669, 10.27572673,  7.5984809 ,  6.02779201,  8.83725081,
       12.64235359, 11.57249984, 12.35625694,  4.98461983,  8.92549752,
        8.859416  ,  7.32233733, 10.52357479, 13.88683436,  6.60752373,
        5.95111804,  5.39450305,  8.34949383,  6.14086582,  5.91408504,
        8.83725081, 11.97957425, 10.0674887 ,  9.50913849, 11.75807252,
       12.0495521 , 13.92390689,  8.87007495,  5.80860165, 10.19036081,
        8.95983586,  7.72415048, 11.04953767, 10.24557047,  9.47369628,
        9.15866607,  5.87132718,  5.27178363,  9.96391045,  9.49511524,
       13.68063098, 10.19036081, 11.28895489,  5.14371005, 13.36092624,
       13.103709  ,  6.74686071,  4.90676242, 11.86239961,  8.04718433,
       13.89671314,  5.98285319, 11.28215429,  8.53501777,  7.52117132,
       13.09737053,  5.57341623, 10.86693708,  9.13507321,  6.4020404 ,
        6.54693987,  5.90035324, 14.02532758,  9.60412735,  7.98469166,
       10.60226614,  5.24137909,  8.89458273,  5.05534609,  9.79814997,
       10.5575601 , 13.54774355,  4.91296205,  5.5513072 , 10.63038565,
        5.5748624 ,  8.4486771 , 10.36844813, 11.83509372, 10.35093485,
       14.03618649, 13.66896739,  5.05534609,  7.71075226,  9.87720102,
       10.09979747,  9.3349858 , 11.31276109, 10.32715715, 13.97779839,
       11.64436684,  6.39520397,  9.15866607,  9.75401449, 10.00410625,
       12.07818871])}
{'P': tensor([[3.2687e-02, 2.3806e-01, 1.6379e-01,  ..., 1.2906e-03, 9.4083e-05,
         7.6030e-04],
        [1.6237e-03, 4.0615e-02, 2.5886e-03,  ..., 2.0465e-03, 1.7664e-04,
         5.0040e-04],
        [4.1048e-03, 2.7166e-01, 3.9857e-02,  ..., 1.4161e-03, 1.1479e-04,
         6.2073e-04],
        ...,
        [7.0586e-05, 1.0166e-04, 7.2221e-05,  ..., 3.8040e-02, 2.5232e-02,
         1.1400e-04],
        [3.3878e-05, 3.4204e-05, 2.9966e-05,  ..., 3.0440e-04, 1.1372e-01,
         1.2819e-04],
        [9.4213e-03, 1.0746e-02, 7.9511e-03,  ..., 1.4300e-02, 7.1423e-04,
         9.4004e-02]]), 'X': tensor([[  6.1741,  -0.5329,   3.1913],
        [  6.3606,   0.9204,   3.2525],
        [  6.2902,   0.0451,   3.7116],
        [  6.2022,   2.4092,   3.0349],
        [  6.1752,   2.5242,   3.5874],
        [  6.2905,   1.9366,   2.1169],
        [ -5.7347,  -8.6908,   9.2878],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -2.1533, -10.5778,   1.0112],
        [  6.3604,   0.9042,  -6.5901],
        [ 10.3251,   8.2928,  -3.0006],
        [  5.6523,   3.9443,  -2.9449],
        [  2.6451,  -4.5250,  -0.5565],
        [  6.2630,   2.1057,  -5.2136],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.3464,   0.5708,  -1.6397],
        [ -1.7290,   7.8825,  -3.3831],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.4572,   7.1085,   7.7376],
        [  1.8766,   7.3694,   1.1887],
        [  2.4932,   7.0903,  -5.4214],
        [ -8.6229,   2.9068,   2.4429],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.2687,  -2.9794,   0.6436],
        [ -9.3616,  -2.4749,   0.1793],
        [ 10.9599,  -5.1849,  -3.5526],
        [  3.7237,  -3.9966,  -5.0376],
        [ -1.8367,   7.8718,  -8.6523],
        [ -9.1706,   1.3863,   8.4032],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.6260,  -1.9401,   3.0259],
        [ 11.6019,   5.9540,  -5.1715],
        [ -1.9600,   7.8577,   6.9121],
        [  5.9193,  -1.3176,   5.1739],
        [ 11.9122,   5.1397,  -1.5498],
        [  1.3339,   7.5627,  -0.9034],
        [  1.7301,   7.4262,   5.0522],
        [  5.8275,  -1.5334,   2.0124],
        [ 11.7349,  -3.5838,   2.1644],
        [ -3.5029,   7.5138,   8.1853],
        [  2.5636, -10.9293,   9.6519],
        [ -5.1974,   6.7396,   3.3484],
        [  2.2182,  13.8007,   5.1889],
        [  1.7301,   7.4262,   5.0522],
        [ 12.0772,  -2.6014,  -7.1832],
        [  0.2480, -11.0153,  -3.0183],
        [  2.3261,   7.1726,   6.2522],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.8093,  -9.4178,   8.4244],
        [ 10.9875,   7.2207,  -8.8549],
        [  0.8940, -11.0402,  -0.9783],
        [  3.1271, -10.8343,   2.6442],
        [ 11.4495,   6.3026,   6.8257],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.3301,   0.3638,  -7.5685],
        [  6.1372,   2.6726,  -6.7885],
        [ -7.3446,   4.9113,  -5.9356],
        [ -9.3616,  -2.4750,   3.7765],
        [  4.7813,  -3.1139,   8.2691],
        [  5.3804,  -2.3504,   7.2744],
        [ -7.9925,  -6.1423,   2.8990],
        [  0.3724,   7.7974,  -6.8722],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.4915,   5.5990,   7.0405],
        [  0.5250, -11.0306,   5.5358],
        [  7.1821,  -9.1845,   9.8548],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.4150,  13.9647,  -4.0915],
        [ -0.9715, -10.8630,   0.3555],
        [  0.9967,  -4.8144,  -9.7370],
        [  3.9022,  13.2906,   5.5788],
        [  2.3376,  -4.6231,   9.8436],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.5607,  -5.0938,  -5.4531],
        [  0.1677,  -4.7447,  -6.1529],
        [ -8.9033,   2.2329,  -8.8912],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.1824,  -4.6647,   1.1985],
        [  5.3858,   4.4260,  -9.6769],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.7078,   2.7182,   1.8146],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.2804,   2.0019,   0.8607],
        [ -9.3107,  -2.7697,  -3.2919],
        [  1.6422,   7.4586,  -6.8056],
        [  5.3602,  -2.3808,   7.2098],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -7.0120,   5.2825,   7.7762],
        [ 11.4703,  -4.2006,   7.4086],
        [  7.8982,  -8.6778,   0.0239],
        [ -1.5261,   7.8988,   9.3768],
        [  2.5636, -10.9293,   9.6519],
        [ -7.7239,   4.4298,   3.1216],
        [  0.3715,  -4.7756,   0.2312],
        [ -3.4100,   7.5436,  -7.6980],
        [ 12.5123,  -0.5339,  -1.0728],
        [  9.0652,   9.8597,  -3.5949],
        [  6.7792,  11.8084,   1.6251],
        [ 11.4495,   6.3026,   6.8257],
        [  4.3554,   5.7393,  -9.0766],
        [ -6.1914,  -8.2997,   8.5380],
        [  2.5382,   7.0672,   4.5097],
        [  3.6703,   6.3443,  -5.4401],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.2911,  -2.4824,  -3.6877]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 2.2552e-02, 4.3360e-02,  ..., 5.6820e-01, 7.9883e-01,
         4.7228e-01],
        [2.2552e-02, 0.0000e+00, 2.0890e-02,  ..., 5.9073e-01, 8.1933e-01,
         4.9477e-01],
        [4.3360e-02, 2.0890e-02, 1.5170e-04,  ..., 6.1134e-01, 8.3724e-01,
         5.1563e-01],
        ...,
        [5.6820e-01, 5.9073e-01, 6.1134e-01,  ..., 0.0000e+00, 3.4919e-01,
         1.1388e-01],
        [7.9883e-01, 8.1933e-01, 8.3724e-01,  ..., 3.4919e-01, 0.0000e+00,
         4.5027e-01],
        [4.7228e-01, 4.9477e-01, 5.1563e-01,  ..., 1.1388e-01, 4.5027e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.19707944,  6.42689705,  6.29034958,  6.65368551,  6.67121748,
        6.58184474, 10.412299  , 10.39942497, 10.794755  ,  6.42439344,
       13.2430377 ,  6.89245352,  5.24137909,  6.60752373,  6.24614905,
        9.36378438,  8.06990483, 13.04217487,  7.52117132,  7.60463349,
        7.5158489 ,  9.09963894,  8.97052115,  9.73579738,  9.68323686,
       12.12448912,  5.46244988,  8.08320235,  9.27474386, 11.51811603,
        5.95111804, 13.04051744,  8.09842704,  6.06416957, 12.97370371,
        7.67939546,  7.62509582,  6.02588517, 12.26997165,  8.29022197,
       11.22596994,  8.51088913, 13.97779839,  7.62509582, 12.35421461,
       11.018084  ,  7.5403822 , 12.12448912, 11.62158554, 13.14777433,
       11.07637531, 11.27656488, 13.06957425,  5.90035324,  6.34059021,
        6.69389431,  8.8353674 ,  9.68324553,  5.70591986,  5.87132718,
       10.08001894,  7.80625981, 10.31584102,  7.17789666, 11.04313641,
       11.65923396, 12.17451685, 14.03618649, 10.90638066,  4.91652556,
       13.85158417,  5.1805345 , 12.0495521 ,  9.96152788,  4.74770899,
        9.17905491,  9.51074763,  5.14999799,  6.97106912,  5.35774573,
        9.12220271, 12.30858677, 10.46739073,  6.59176018,  9.71392179,
        7.63726134,  5.86519525,  9.88571713,  8.77913428, 12.21532282,
       11.73397309,  8.04483495, 11.22596994,  8.90405823,  4.79003361,
        8.27853076, 12.52372629, 13.39372426, 13.61600536, 13.06957425,
        7.20482224, 10.35464916,  7.50917551,  7.32950385,  8.27853076,
        5.84450127])}
{'P': tensor([[3.5544e-02, 2.8894e-01, 2.5980e-01,  ..., 2.2264e-03, 5.5000e-06,
         3.3011e-04],
        [2.7619e-03, 6.2092e-02, 3.8030e-02,  ..., 1.7881e-03, 8.7147e-06,
         6.8583e-04],
        [3.1561e-03, 7.3031e-02, 6.3778e-02,  ..., 1.8020e-03, 8.2642e-06,
         5.9246e-04],
        ...,
        [1.4873e-01, 1.4468e-01, 1.8291e-01,  ..., 7.9808e-02, 1.0963e-05,
         2.4478e-04],
        [5.5777e-05, 3.5600e-05, 3.2437e-05,  ..., 5.9126e-05, 2.5297e-02,
         1.8502e-06],
        [2.5060e-06, 4.2778e-06, 3.7794e-06,  ..., 8.8751e-07, 2.6539e-07,
         8.7831e-03]]), 'X': tensor([[  6.1973,  -0.4372,   6.5720],
        [  6.3569,   0.7591,   6.5011],
        [  6.3549,   0.7096,   6.9198],
        [  6.0073,   3.0986,   6.7937],
        [  5.9883,   3.1536,   6.2320],
        [  6.2895,   0.0405,   5.7652],
        [  9.3424,  -7.3685,  -3.6807],
        [  6.3551,   0.7149,   1.3344],
        [ -8.1204,  -5.9308,   4.0761],
        [ -9.3616,  -2.4750,   3.7765],
        [  5.9003,  12.3550,   2.1917],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.6928,  12.4710,   3.6001],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 11.0275,  -5.0666,   8.4154],
        [ -3.4488, -10.0934,   2.5195],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -0.7599, -10.8995,  -6.0366],
        [  4.3377,   5.7570,   1.3712],
        [  4.5814, -10.4485,   2.5496],
        [  8.5093,  -8.1764,  -3.2381],
        [ 12.4690,  -0.8528,  -9.3374],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.1041, -11.0045,   2.4311],
        [ -4.2040,   7.2486,   2.9875],
        [  4.3523, -10.5232,   8.2138],
        [  6.0031,  12.2958,  -6.5992],
        [ -4.3963,   7.1629,   4.5263],
        [ -6.4464,   5.8272,   7.5038],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.7078,   2.7182,   1.8146],
        [ 12.3453,  -1.5462,   7.0212],
        [  3.3633,  13.4794,   0.5193],
        [ -6.4368,   5.8356,   3.3014],
        [  6.1755,   2.5233,  -2.0042],
        [  9.1409,  -7.5789,   1.5814],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.8051, -10.3703,   0.5268],
        [  7.1697,  11.5343,  -0.7276],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.7565,  11.8237,  -4.9882],
        [ 12.5919,   0.4035,  -5.8460],
        [ -7.6719,  -6.6260,   7.1468],
        [  0.1382, -11.0073,   1.0026],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4721,  -1.3097,  -7.4377],
        [  1.7218,  -4.7576,  -3.3617],
        [ 12.5789,   0.1722,   1.6470],
        [  6.3301,   0.3638,  -7.5685],
        [  6.3534,   1.2990,  -7.2942],
        [  1.6954,  -4.7616,   9.8198],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.9074,  -4.8139,   8.1674],
        [  5.1861,   4.7361,  -3.7312],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.7802,  -1.6363,  -6.2782],
        [ -4.4503,   7.1378,   3.3751],
        [  9.3607,   9.5337,  -4.5412],
        [  7.9141,  -8.6656,   9.6948],
        [  4.5954,  13.0099,  -4.4439],
        [  9.9756,  -6.6330,  -3.3429],
        [  5.8450,  -9.9356,   3.0767],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.9526,   3.2525,  -4.3821],
        [  5.2939,  -2.4784,  -4.5840],
        [ -5.6423,   6.4537,  -1.2091],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.6613,   7.0021,   4.5783],
        [  1.7232,  -4.7574,   3.1270],
        [  8.1217,  -8.5024,  -0.6149],
        [  3.6249, -10.7255,   5.6605],
        [  3.4009,  -4.1874,  -5.9751],
        [ 11.2209,  -4.7092,   2.6117],
        [ -0.8675, -10.8815,   6.8561],
        [  3.3069,   6.6089,   8.7377],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.3211,   1.7017,  -0.2164],
        [  1.1644, -11.0394,   7.8511],
        [  2.5974,   7.0363,  -7.7810],
        [  2.7685, -10.8982,  -6.8928],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.8719,   7.3713,  -3.7465],
        [  5.5626,  -2.0536,   8.4433],
        [  2.8474,   6.8979,   6.6457],
        [ -7.3333,   4.9246,  -7.9212],
        [  8.7131,  10.2216,   7.7915],
        [  8.1217,  -8.5024,  -0.6149],
        [  0.3493,  -4.7727,  -6.5682],
        [  4.0591,   6.0202,  -1.6339],
        [  6.2036,   2.4030,  -1.2500],
        [  6.3515,   0.6418,  -6.5823],
        [  5.8201,   3.5841,  -8.0609],
        [  6.1372,   2.6726,  -6.7885],
        [ -3.1204, -10.2345,   9.2288],
        [  8.8474,  10.0868,  -8.8644],
        [ -3.6381,   7.4683,   7.3544],
        [  5.7963,  -1.6018,   1.8856],
        [  6.2022,   2.4092,   3.0349],
        [  6.3236,   0.3011,  -4.6668],
        [  5.5626,  -2.0536,   8.4433],
        [ -5.8389,  -8.6056,  -3.5098],
        [  9.3607,   9.5337,  -4.5412]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0027, 0.0180,  ..., 0.0631, 1.2329, 0.8864],
        [0.0027, 0.0000, 0.0184,  ..., 0.0634, 1.2306, 0.8868],
        [0.0180, 0.0184, 0.0000,  ..., 0.0452, 1.2429, 0.9040],
        ...,
        [0.0631, 0.0634, 0.0452,  ..., 0.0000, 1.2696, 0.9482],
        [1.2329, 1.2306, 1.2429,  ..., 1.2696, 0.0000, 1.2202],
        [0.8864, 0.8868, 0.9040,  ..., 0.9482, 1.2202, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.21275016,  6.4020404 ,  6.39438044,  6.7593917 ,  6.76791483,
        6.28961988, 11.89855267,  6.39520397, 10.05558893,  9.68324553,
       13.69162795, 10.60226614, 13.70893492,  5.35774573,  9.09661212,
        9.48969978,  9.91668182, 12.13568598, 10.66631784,  9.69792513,
       10.92597196,  7.20826219, 11.4088043 , 11.80092578, 12.4980845 ,
        5.57341623, 11.0050379 ,  8.37954853, 11.38774149, 13.6829843 ,
        8.40445872,  8.68979052,  4.7663526 ,  9.12220271, 12.44177641,
       13.89264822,  8.68832398,  6.67108662, 11.87412439,  9.38232121,
       11.42947632, 13.58101794,  5.35774573, 13.61801526, 12.59840193,
       10.1371552 , 11.00812996,  9.41244865,  9.562174  ,  5.05963217,
       12.58006038,  6.34059021,  6.48486171,  5.05444288,  9.50654967,
        4.89869205,  7.02326515,  9.62958973,  6.00732193,  8.41148722,
       13.36092624, 11.73567759, 13.79762415, 11.97957425, 11.52733581,
        9.9353921 ,  6.78325151,  5.84533345,  8.5724042 ,  9.94285511,
       12.05395056,  7.49076899,  5.05989547, 11.75807252, 11.32149259,
        5.39450305, 12.16901615, 10.91601655,  7.39004226,  5.70405525,
        6.54615887, 11.10066126,  7.50035355, 11.24434374, 12.04924159,
        7.60529892,  5.92952538,  7.46248947,  8.83341608, 13.43127131,
       11.75807252,  4.78544875,  7.26074925,  6.652742  ,  6.383897  ,
        6.83515201,  6.69389431, 10.69963954, 13.41713661,  8.30729352,
        6.01356004,  6.65368551,  6.33075682,  5.92952538, 10.39942497,
       13.36092624])}
{'P': tensor([[2.2402e-02, 2.2226e-01, 1.7780e-01,  ..., 1.8257e-05, 2.3364e-03,
         2.8471e-05],
        [1.3581e-03, 2.4562e-02, 1.9597e-01,  ..., 1.4486e-05, 2.9632e-03,
         3.5691e-05],
        [1.2010e-03, 2.0848e-03, 2.5659e-02,  ..., 1.2988e-05, 2.8995e-03,
         4.1873e-05],
        ...,
        [8.5897e-04, 7.0681e-04, 6.6967e-04,  ..., 1.4165e-02, 1.1028e-05,
         7.0550e-07],
        [8.3702e-04, 1.0513e-03, 1.1612e-03,  ..., 7.2735e-06, 2.0884e-01,
         4.3101e-04],
        [2.0069e-04, 1.6166e-04, 1.4547e-04,  ..., 4.7337e-04, 2.4997e-05,
         1.5508e-01]]), 'X': tensor([[ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0400e-02, 4.6528e-03,  ..., 6.3443e-01, 4.3537e-01,
         9.0243e-01],
        [1.0400e-02, 0.0000e+00, 1.2811e-02,  ..., 6.4428e-01, 4.2635e-01,
         9.0811e-01],
        [4.6528e-03, 1.2811e-02, 1.0727e-04,  ..., 6.3429e-01, 4.3896e-01,
         9.0520e-01],
        ...,
        [6.3443e-01, 6.4428e-01, 6.3429e-01,  ..., 0.0000e+00, 9.7969e-01,
         5.6476e-01],
        [4.3537e-01, 4.2635e-01, 4.3896e-01,  ..., 9.7969e-01, 0.0000e+00,
         1.0248e+00],
        [9.0243e-01, 9.0811e-01, 9.0520e-01,  ..., 5.6476e-01, 1.0248e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.2134961 ,  6.32572323,  6.39520397,  6.50409372,  6.53250937,
        6.58184474, 10.88618114, 11.4890861 ,  6.18668463, 11.27160035,
        9.09963894,  5.5366698 ,  5.0209005 , 10.48840476, 12.1460931 ,
       13.36092624, 11.73848065, 13.91198191,  9.11473391,  7.60463349,
       11.64436684,  5.53175275, 11.22596994, 13.36800073, 10.06437598,
        7.88677538, 11.66453059, 11.83570132,  9.05843201,  9.96152788,
        5.82345867,  5.5671389 ,  5.14124612,  6.46356806,  6.00732193,
       13.03002338, 13.12218013, 13.95361098, 13.97430191,  8.09899874,
       13.83139024, 11.77124373,  7.32950385,  5.18047638,  6.76056625,
       12.33167382, 10.61417181,  6.28961988,  9.75075273, 10.8992524 ,
        9.3349858 , 12.3344536 ,  5.29755487, 11.1913598 , 13.3426541 ,
       12.30858677,  4.92497514,  7.01475376,  4.98461983, 10.75443106,
       12.76408562,  5.18047638,  8.67502348, 13.62896593, 12.24467041,
        9.62238828,  9.77588801,  9.96391045,  7.32754882, 11.785566  ,
       13.27994304,  7.19513806,  5.83560009, 11.63765273,  7.60529892,
        6.53290663,  8.62631801,  7.15100303, 10.8207297 , 13.26209545,
        6.42439344,  4.85995002,  4.96231532, 10.95593729, 10.63038565,
       11.22596994,  8.38252937,  6.67218687, 11.35345345, 13.18278606,
        5.43544626,  6.59176018, 10.1979669 , 12.26997165,  6.1561646 ,
       11.83509372,  6.25586297, 11.57249984, 13.35927527, 10.7506423 ,
       12.72524615,  6.05165248, 12.97370371, 11.4890861 ,  7.15100303,
        9.36093562])}
{'P': tensor([[4.7177e-02, 2.8081e-01, 1.7410e-01,  ..., 2.5814e-05, 1.3000e-04,
         1.0033e-04],
        [4.0320e-03, 6.5068e-02, 7.4509e-02,  ..., 4.8413e-05, 1.9067e-04,
         1.4948e-04],
        [5.0483e-03, 3.9400e-02, 6.4328e-02,  ..., 5.2517e-05, 2.2221e-04,
         1.7455e-04],
        ...,
        [2.7951e-05, 2.4000e-05, 2.5617e-05,  ..., 5.2926e-02, 3.9087e-05,
         3.3807e-05],
        [1.4761e-04, 2.2983e-04, 2.7707e-04,  ..., 8.1120e-05, 7.2014e-02,
         1.1139e-01],
        [1.4602e-04, 2.2615e-04, 2.7228e-04,  ..., 1.0306e-04, 3.5062e-02,
         9.7347e-02]]), 'X': tensor([[ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0192, 0.0374,  ..., 1.0835, 0.8636, 0.8837],
        [0.0192, 0.0000, 0.0184,  ..., 1.0693, 0.8450, 0.8650],
        [0.0374, 0.0184, 0.0000,  ..., 1.0582, 0.8266, 0.8467],
        ...,
        [1.0835, 1.0693, 1.0582,  ..., 0.0000, 0.9185, 0.9262],
        [0.8636, 0.8450, 0.8266,  ..., 0.9185, 0.0000, 0.0202],
        [0.8837, 0.8650, 0.8467,  ..., 0.9262, 0.0202, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.2322769 ,  6.39438044,  6.4020404 ,  6.46560181,  6.7593917 ,
        6.76791483,  5.35774573,  6.61211224,  5.38342077,  4.978704  ,
       13.66991027,  6.14086582,  4.74770899, 11.65923396,  8.36995953,
        7.71075226,  9.77588801,  6.00732193,  9.05843201, 14.05687815,
       13.27717982,  6.77777606, 10.48840476, 14.03618649,  5.40689332,
        8.05242591, 13.35013374,  7.10378919, 10.412299  , 12.77402351,
       13.97779839, 13.27994304,  7.67939546,  9.562174  ,  5.10444097,
       11.51811603, 12.7125079 ,  8.8580671 , 12.38881635,  4.92497514,
       13.59056613,  5.80312949,  5.39450305,  5.64865173,  9.60080639,
       11.31276109,  5.64500512,  7.10378919,  4.9160962 ,  6.07992623,
        4.79003361,  5.14124612, 11.83570132, 13.13205357, 12.16901615,
        6.05113217,  7.62232307, 14.09357547,  8.62631801, 11.00812996,
       10.09979747, 11.75438006,  8.68832398,  6.65259775,  5.83560009,
        8.05242591,  9.09502385, 10.65929968, 14.08549802, 12.86648771,
        8.16714836, 12.3344536 , 12.889695  , 11.69264243,  5.40689332,
        9.78147542, 10.78533838,  5.16077764,  4.91296205, 13.20597932,
        6.383897  , 10.33712258, 13.26209545,  5.29009836, 12.78327037,
        9.73579738,  5.21280156,  8.04483495,  7.73151309, 13.84126197,
        8.87007495, 10.0674887 ,  9.64157461, 13.12218013, 13.59170809,
       13.2430377 , 11.11969815,  5.05989547,  9.34127013,  5.40689332,
       13.3426541 ,  7.34250948, 10.95593729,  9.36093562,  6.36004393,
        6.38968357])}
{'P': tensor([[2.1294e-02, 2.3888e-01, 2.0412e-01,  ..., 5.6623e-03, 1.8580e-03,
         7.5925e-06],
        [2.3624e-03, 3.0157e-02, 2.9095e-01,  ..., 8.3530e-03, 2.4055e-03,
         1.0484e-05],
        [3.4451e-03, 3.8340e-03, 6.2483e-02,  ..., 2.2821e-02, 3.5219e-03,
         2.0390e-05],
        ...,
        [2.5883e-04, 2.6721e-04, 3.1597e-04,  ..., 6.4954e-02, 3.6781e-05,
         3.3032e-05],
        [7.0653e-04, 7.9317e-04, 1.3420e-03,  ..., 6.8463e-03, 2.5958e-02,
         4.8978e-07],
        [3.5802e-04, 3.3153e-04, 2.5456e-04,  ..., 2.0544e-05, 6.9911e-05,
         3.4519e-02]]), 'X': tensor([[  6.2247,  -0.3150,  -9.3657],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.3360,   0.4258,  -9.2509],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3505,   0.6235,  -8.5849],
        [  6.3303,   1.6145,  -8.4083],
        [ -2.6485, -10.4151,   0.8766],
        [ 12.3357,   3.6001,   0.7903],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.4426, -10.1181,  -5.1268],
        [ -3.9700,   7.3453,  -4.1486],
        [ -9.4117,   0.1160,   8.2574],
        [  6.2905,   1.9366,   2.1169],
        [ 10.9717,  -5.1646,  -6.9039],
        [  7.3592,  11.3937,   6.2407],
        [  2.6321,  -4.5296,  -2.5159],
        [  6.2555,   2.1474,   4.1685],
        [  1.5653,  13.9377,   4.1773],
        [  2.2659,  13.7894,   6.5436],
        [ -8.1204,  -5.9308,   4.0761],
        [  3.4140,  -4.1803,  -8.8224],
        [  7.0642,  11.6104,  -9.8316],
        [ 11.3118,   6.5957,  -4.5667],
        [ 10.6283,  -5.7253,   7.4878],
        [  0.1382, -11.0073,   1.0026],
        [  5.2939,  -2.4784,  -4.5840],
        [ 12.5717,   1.9266,   7.4903],
        [  4.4049,   5.6892,  -4.4394],
        [  6.3516,   0.6425,   0.3425],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.2975,   0.0947,   5.1791],
        [  4.9957,  -2.8705,   6.5978],
        [  4.9337,  12.8566,  -3.6582],
        [  1.9241,   7.3503,   7.9004],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 10.2712,   8.3707,  -4.9743],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.3339,   7.5627,  -0.9034],
        [ -1.0407,   7.9166,  -6.6021],
        [  0.5175,  14.0910,   4.6540],
        [  8.5198,  10.4091,  -8.6486],
        [ 11.7795,  -3.4697,   3.5624],
        [  0.9478,  -4.8144,   0.6371],
        [ -8.9033,   2.2329,  -8.8912],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.1041, -11.0045,   2.4311],
        [  6.1989,  -9.7591,   0.8289],
        [  6.2555,   2.1474,   4.1685],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.4697,  -0.6157,  -6.8472],
        [  8.1528,  10.7452,   9.1493],
        [  1.6422,   7.4586,  -6.8056],
        [ -5.0132,  -9.2234,  -8.3531],
        [  3.2159, -10.8166,  -1.0406],
        [ -1.7009,   7.8851,  -8.2437],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.6757, -11.0361,  -1.3047],
        [  5.8902,  -1.3887,   4.0388],
        [  5.9912,   3.1453,  -3.9989],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.8617,   3.4853,  -9.8080],
        [ -7.9298,   4.1367,  -3.9773],
        [  8.5198,  10.4091,  -8.6486],
        [  2.3200,   7.1756,   1.1938],
        [ -9.4753,  -0.8012,  -5.2050],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.1497,   7.8334,  -5.4412],
        [  5.1787,  -2.6383,  -4.2533],
        [  4.2041,  -3.6509,  -4.9488],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.5859,  -5.0393,   0.5476],
        [ 10.9931,   7.2107,   7.7933],
        [ -2.6069, -10.4298,  -6.1995],
        [  9.0051,   9.9234,   5.4411],
        [ 10.0400,   8.6922,  -6.2847],
        [ 11.3691,   6.4760,   1.0787],
        [  3.2366,  -4.2732,   5.1532],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.4721,  -1.3097,  -7.4377],
        [  3.9085,   6.1512,  -2.2560],
        [ -9.4671,  -0.5553,   0.1117],
        [  6.3534,   1.2990,  -7.2942],
        [  7.5784,  -8.9141,   5.8410],
        [  4.8051, -10.3703,   0.5268],
        [  1.7815,  -4.7481,   4.4289],
        [  2.5010,  -4.5736,  -0.8989],
        [ 12.3025,  -1.7440,  -6.6632],
        [  3.1271, -10.8343,   2.6442],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.1752,   2.5242,   3.5874],
        [  1.6232,  -4.7717,  -2.5655],
        [  5.1142,  -2.7230,  -4.1966],
        [  6.1222,  -0.7261,   8.8204],
        [  5.1196,  -2.7160,   4.2705],
        [  3.4401,  13.4540,   4.6576],
        [  6.2902,   0.0451,   3.7116],
        [  6.0054,  -1.0902,   8.3079],
        [  0.7275,  14.0667,  -0.9815],
        [  4.8051, -10.3703,   0.5268],
        [  5.0277,   4.9589,   7.0175],
        [  3.6703,   6.3443,  -5.4401],
        [ 12.6059,   0.9597,  -6.3264],
        [ -4.8665,  -9.3204,  -5.2564]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.8454e-04, 9.5942e-04,  ..., 1.5039e-01, 1.4234e-01,
         9.1395e-01],
        [8.8454e-04, 0.0000e+00, 1.6479e-03,  ..., 1.5122e-01, 1.4144e-01,
         9.1481e-01],
        [9.5942e-04, 1.6479e-03, 0.0000e+00,  ..., 1.4957e-01, 1.4299e-01,
         9.1362e-01],
        ...,
        [1.5039e-01, 1.5122e-01, 1.4957e-01,  ..., 0.0000e+00, 2.8946e-01,
         8.0436e-01],
        [1.4234e-01, 1.4144e-01, 1.4299e-01,  ..., 2.8946e-01, 0.0000e+00,
         1.0467e+00],
        [9.1395e-01, 9.1481e-01, 9.1362e-01,  ..., 8.0436e-01, 1.0467e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.23262289,  6.25586297,  6.35029274,  6.30647746,  6.38105015,
        6.53290663, 10.74655346, 12.85032836, 10.39942497,  5.84533345,
       11.4890861 ,  8.34949383,  9.41244865,  6.58184474, 12.12641969,
       13.56367857,  5.23878371,  6.61386436, 14.02532758, 13.97430191,
       10.05558893,  5.39723241, 13.59056613, 13.09426142, 12.07225469,
       11.00812996,  5.84533345, 12.71843462,  7.19513806,  6.38399789,
        6.60752373,  9.58072286,  6.29821673,  5.76165934, 13.77075174,
        7.59795445, 12.18045978, 13.25017211,  5.9510489 ,  7.67939546,
        7.98469166, 14.10045797, 13.45124917, 12.27991991,  4.90676242,
        9.17905491, 10.36844813, 11.0050379 , 11.56146873,  6.61386436,
        4.9160962 ,  9.96152788,  9.48969978, 13.48809595,  7.63726134,
       10.4977132 , 11.2845608 ,  8.06643487, 13.00380624, 11.05672136,
        6.05165248,  6.76661595,  5.27178363,  6.81961996,  8.94395628,
       13.45124917,  7.54127425,  9.50913849,  5.5366698 ,  7.83486943,
        5.81199422,  5.56809335, 11.95743558,  9.95552626, 13.14690137,
       10.7506423 , 13.40024224, 13.27994304, 13.08414689,  5.36061974,
        4.8763135 ,  9.562174  ,  7.28793663,  9.48337063,  6.48486171,
       11.70016217, 11.42947632,  5.0713364 ,  5.21280156, 12.42554808,
       11.27656488, 11.99633819,  6.67121748,  5.04028538,  5.79391825,
        6.16512856,  5.79543038, 13.88683436,  6.29034958,  6.10360073,
       14.08549802, 11.42947632,  7.06169391,  7.32950385, 12.64235359,
       10.51437947])}
{'P': tensor([[3.0627e-02, 1.8698e-01, 1.8092e-01,  ..., 2.5914e-04, 3.2663e-05,
         8.1388e-05],
        [2.9427e-03, 5.3625e-02, 3.3583e-03,  ..., 1.8658e-04, 6.5412e-05,
         7.3866e-05],
        [2.1625e-03, 4.0118e-01, 3.1336e-02,  ..., 1.3031e-04, 3.5693e-05,
         4.9641e-05],
        ...,
        [3.3655e-03, 3.6656e-03, 3.5516e-03,  ..., 1.8581e-02, 3.7205e-06,
         5.8574e-05],
        [1.7557e-05, 1.4217e-05, 1.4403e-05,  ..., 4.0872e-06, 9.0720e-02,
         2.8310e-06],
        [1.8707e-02, 1.5564e-02, 1.6700e-02,  ..., 7.0912e-02, 1.9947e-06,
         5.3643e-02]]), 'X': tensor([[ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0241, 0.0225,  ..., 0.1459, 0.9447, 0.2744],
        [0.0241, 0.0000, 0.0016,  ..., 0.1687, 0.9671, 0.2622],
        [0.0225, 0.0016, 0.0000,  ..., 0.1671, 0.9655, 0.2628],
        ...,
        [0.1459, 0.1687, 0.1671,  ..., 0.0000, 0.7991, 0.3382],
        [0.9447, 0.9671, 0.9655,  ..., 0.7991, 0.0000, 1.0299],
        [0.2744, 0.2622, 0.2628,  ..., 0.3382, 1.0299, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.24187044,  6.42439344,  6.383897  ,  6.60752373,  6.36004393,
        6.56833864, 10.9036226 ,  9.88571713,  8.41148722, 11.97949273,
        8.28968067,  7.09887557,  6.00663314, 13.43127131, 11.42420508,
       12.07225469,  9.58072286, 13.14690137, 12.30858677,  7.37902766,
        4.91336994, 11.83075005, 10.31613269, 10.23946247,  5.84450127,
       13.08414689,  7.58668144, 13.00413829,  5.08910491,  7.57685908,
        9.76896904,  7.8575648 ,  8.50268322,  5.05981914,  8.57560275,
        5.94541803, 12.7125079 ,  6.53290663,  4.87083348, 12.12448912,
        9.60359158, 10.31584102,  8.31625171, 12.42554808, 13.86480175,
        5.87132718, 11.9120321 , 13.50484539, 11.018084  , 13.31764181,
        7.72291946,  6.81084724, 10.63038565,  7.46248947,  5.86519525,
        6.99097013,  8.89458273,  8.90405823, 10.76154256,  9.97822473,
        9.64157461,  9.10945678,  7.24677664, 11.73567759,  6.50409372,
       14.02532758,  7.27759233, 13.13888841,  5.18277858,  7.53063885,
        9.88571713,  5.0209005 ,  8.01009083, 10.30972193,  6.14086582,
       14.07623266,  9.38232121,  5.64500512,  6.00732193, 11.8159254 ,
        9.41605684,  9.86089579,  9.48337063, 11.83509372, 12.52875793,
       10.52587243, 10.06286074,  7.24677664, 13.39372426,  5.0209005 ,
       10.60226614, 13.04217487, 10.90638066,  7.80625981, 13.27717982,
        9.55720651, 12.78327037,  8.24808832,  9.50913849,  8.97052115,
       13.70893492,  7.59795445,  6.76661595, 12.1460931 ,  8.62260327,
       11.81423161])}
{'P': tensor([[3.4918e-02, 1.7712e-01, 1.4623e-01,  ..., 2.6230e-05, 2.4202e-04,
         1.6105e-04],
        [1.1883e-03, 4.4647e-02, 3.6211e-01,  ..., 1.4492e-05, 1.8654e-04,
         1.6634e-04],
        [2.0652e-03, 7.4546e-03, 8.9952e-02,  ..., 2.5699e-05, 3.6282e-04,
         3.4015e-04],
        ...,
        [1.0616e-02, 3.7217e-03, 2.9573e-03,  ..., 1.0878e-01, 9.7128e-04,
         2.5189e-05],
        [1.4400e-02, 6.7040e-03, 6.2434e-03,  ..., 1.6536e-03, 9.6450e-02,
         2.1266e-04],
        [3.5571e-05, 1.0598e-04, 1.3834e-04,  ..., 1.8804e-07, 1.2578e-05,
         9.7341e-02]]), 'X': tensor([[ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0173, 0.0126,  ..., 0.6325, 0.3580, 0.9572],
        [0.0173, 0.0000, 0.0169,  ..., 0.6186, 0.3707, 0.9743],
        [0.0126, 0.0169, 0.0000,  ..., 0.6217, 0.3538, 0.9610],
        ...,
        [0.6325, 0.6186, 0.6217,  ..., 0.0000, 0.7816, 1.4915],
        [0.3580, 0.3707, 0.3538,  ..., 0.7816, 0.0000, 0.7106],
        [0.9572, 0.9743, 0.9610,  ..., 1.4915, 0.7106, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.24614905,  6.61386436,  6.67218687,  6.67121748,  6.42689705,
        6.65368551,  9.84314712,  5.35774573,  5.0713364 ,  6.84796202,
       12.21532282,  5.13182506, 12.18998195,  6.36004393, 12.05395056,
       13.52430176,  9.58072286,  5.54776934,  8.23525433,  8.09984231,
        5.24461068,  7.16747878, 12.41012046, 12.77402351,  7.16598273,
       13.3614598 , 13.3426541 ,  6.00663314,  9.562174  , 10.06286074,
        8.0550286 , 11.28895489,  8.09899874, 12.10942412, 14.03618649,
        8.35856487, 12.0495521 , 12.44177641,  5.36061974, 11.18839169,
       13.89974693,  6.54615887, 13.6105595 ,  9.96152788,  8.44940661,
       11.22596994,  6.2322769 , 10.00794468,  6.46356806, 13.05148826,
       13.25336001,  6.84796202, 13.35013374,  6.7166297 , 10.32715715,
        5.22903117, 12.07225469,  5.94909709,  9.32094055, 12.71843462,
       10.4606265 , 11.36089355, 12.42554808,  7.32754882,  8.09704052,
       13.92390689,  7.14837514,  7.42512621,  4.90505566,  5.01114719,
       11.04953767, 13.0743399 , 13.68672855,  5.47866221,  5.22903117,
        9.97964013,  9.97822473, 13.96746816, 12.77402351,  9.96391045,
        5.19986797,  5.14371005, 13.45642379,  6.81084724,  9.58347456,
        7.71075226,  6.76056625,  7.37576154, 12.889695  ,  6.12886024,
       12.41012046,  8.09984231,  6.25586297, 12.35625694,  5.10444097,
       13.32557133,  5.41348121,  6.59176018,  9.93553424,  4.9160962 ,
       11.05672136,  8.99791316,  5.39450305, 11.35345345,  4.92497514,
       13.14777433])}
{'P': tensor([[2.8595e-02, 2.7588e-01, 1.9480e-01,  ..., 1.7379e-06, 7.7607e-04,
         9.9775e-05],
        [3.6844e-03, 6.0045e-02, 4.2548e-03,  ..., 4.2393e-06, 1.8778e-03,
         1.8846e-04],
        [3.2735e-03, 3.4517e-01, 3.1279e-02,  ..., 2.1645e-06, 9.7491e-04,
         1.0685e-04],
        ...,
        [2.0993e-06, 1.9975e-06, 2.1429e-06,  ..., 5.3019e-02, 7.1681e-07,
         2.0695e-04],
        [3.8559e-06, 5.2966e-06, 4.6860e-06,  ..., 4.1560e-06, 6.1677e-03,
         1.4503e-06],
        [5.6647e-03, 5.3641e-03, 5.8705e-03,  ..., 7.1000e-05, 7.6904e-04,
         1.5434e-01]]), 'X': tensor([[  6.2535,  -0.1709,  -9.4102],
        [  6.3360,   0.4258,  -9.2509],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3303,   1.6145,  -8.4083],
        [  6.3505,   0.6235,  -8.5849],
        [  6.3589,   0.8239,  -8.4230],
        [ 11.9641,  -2.9568,   7.7211],
        [ -7.4807,   4.7463,   2.0616],
        [ 12.5154,   2.5072,  -9.3274],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.8940, -11.0402,  -0.9783],
        [  5.1292,   4.8183,  -5.7997],
        [ 12.2311,  -2.0445,   1.3545],
        [  9.3607,   9.5337,  -4.5412],
        [  4.1348,   5.9514,  -0.8856],
        [  2.4697,  -4.5836,   1.4593],
        [ 12.0845,  -2.5772,  -4.6073],
        [  8.6420,  -8.0580,   6.5781],
        [ -7.6719,  -6.6260,   7.1468],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.2804,   2.0019,   0.8607],
        [  7.1697,  11.5343,  -0.7276],
        [  1.6408,   7.4591,  -4.2294],
        [  7.5784,  -8.9141,   5.8410],
        [ -3.2053, -10.1993,  -3.8736],
        [ 12.2218,   4.0953,  -8.2072],
        [  9.4516,  -7.2501,  -1.8868],
        [  1.6232,  -4.7717,  -2.5655],
        [  8.6420,  -8.0580,   6.5781],
        [ -6.0175,  -8.4541,  -9.8408],
        [  1.4150,  13.9647,  -4.0915],
        [ -7.9298,   4.1367,  -3.9773],
        [  9.4554,   9.4245,   9.8800],
        [  2.3928,   7.1404,  -1.7035],
        [ -5.1046,  -9.1610,   2.3418],
        [  1.9312,  13.8649,   8.3563],
        [  2.6970,   6.9826,   9.7350],
        [  6.2804,   2.0019,   0.8607],
        [  5.0190,   4.9705,  -7.9752],
        [  1.9503,  -4.7172,  -6.2313],
        [ -8.8154,   2.4611,   2.5198],
        [  7.5309,  11.2617,   7.2656],
        [ 10.9916,   7.2133,  -0.9179],
        [ -2.4956, -10.4682,   3.5632],
        [ -3.2736, -10.1703,   3.8712],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.7544,   2.6096,   3.3146],
        [  5.1890,   4.7318,   0.9515],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.8951,  -3.1570,   7.0742],
        [ -9.3739,   0.3982,  -4.2564],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 10.0619,   8.6627,  -9.2884],
        [  6.1929,   2.4501,  -6.9352],
        [  3.6817,   6.3354,   6.1233],
        [  6.3119,   1.7797,   9.9186],
        [  1.5246,  -4.7838,  -5.2095],
        [  0.9809,   7.6642,  -1.4217],
        [ -8.5506,  -5.1154,   6.3423],
        [  3.0719, -10.8449,  -4.1673],
        [ -9.4196,   0.0478,   9.4140],
        [ 11.3118,   6.5957,  -4.5667],
        [  2.5974,   7.0363,  -7.7810],
        [  0.6363, -11.0348,   6.1414],
        [  6.2902,   0.0451,   3.7116],
        [  6.8407,  11.7666,   3.0045],
        [  1.7500,   7.4187,   4.4237],
        [  7.0358,  -9.2784,  -9.8479],
        [ 12.2914,   3.8037,   6.0978],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.3464,   0.5708,  -1.6397],
        [  0.3354,  -4.7708,  -4.9125],
        [ 10.2712,   8.3707,  -4.9743],
        [  7.8375,  -8.7240,  -9.8913],
        [ 11.9641,  -2.9568,   7.7211],
        [  7.9816,  10.8938,  -0.5378],
        [ -5.3736,   6.6310,  -1.5174],
        [ -4.8665,  -9.3204,  -5.2564],
        [  0.6470, -11.0352,  -1.1931],
        [  1.8320,   7.3871,   5.1452],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.3377,   5.7570,   1.3712],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 11.9936,  -2.8677,   6.0614],
        [  0.1532,  -4.7422,  -7.0145],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.5770,   1.8512,  -0.6578],
        [ -8.8745,   2.3097,   3.5600],
        [  4.8051, -10.3703,   0.5268],
        [  6.1973,  -0.4372,   6.5720],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.0039,   7.3173,  -1.5885],
        [  2.2860,   7.1916,  -6.1011],
        [ -4.7389,   6.9953,   3.7638],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.2860,   7.1916,  -6.1011],
        [  8.8474,  10.0868,  -8.8644],
        [  8.8474,  10.0868,  -8.8644],
        [  0.5175,  14.0910,   4.6540],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -8.9282,   2.1648,   8.4894],
        [  9.7328,   9.0902,  -0.8917],
        [  1.3403,  -4.8011,  -1.1698]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.6479e-03, 1.9662e-03,  ..., 1.2774e+00, 4.3450e-01,
         3.6676e-01],
        [1.6479e-03, 0.0000e+00, 1.0289e-03,  ..., 1.2758e+00, 4.3413e-01,
         3.6523e-01],
        [1.9662e-03, 1.0289e-03, 0.0000e+00,  ..., 1.2757e+00, 4.3313e-01,
         3.6482e-01],
        ...,
        [1.2774e+00, 1.2758e+00, 1.2757e+00,  ..., 0.0000e+00, 1.2135e+00,
         9.3370e-01],
        [4.3450e-01, 4.3413e-01, 4.3313e-01,  ..., 1.2135e+00, 0.0000e+00,
         3.7126e-01],
        [3.6676e-01, 3.6523e-01, 3.6482e-01,  ..., 9.3370e-01, 3.7126e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.25586297,  6.35029274,  6.30647746,  6.53290663,  6.38105015,
        6.41202986, 12.32408516,  8.85939626, 12.76408562,  9.84314712,
       12.4217523 ,  8.62631801, 11.07637531,  7.03733775, 12.40074901,
       13.36092624,  7.24677664,  5.20659892, 12.35625694, 11.8159254 ,
       10.1371552 ,  5.52416368,  6.59176018, 13.58101794,  7.6374603 ,
       11.70016217, 10.6910768 , 12.889695  , 11.9120321 ,  5.04028538,
       11.8159254 , 10.37698435, 14.03618649,  8.94395628, 13.35013374,
        7.53063885, 10.48721382, 13.99877413,  7.48538905,  6.59176018,
        7.06372172,  5.10444097,  9.15253344, 13.54774355, 13.1471313 ,
       10.76154256, 10.68416401,  6.02588517,  9.13507321,  7.02253235,
        5.35774573, 12.30695098,  9.38232121, 10.52587243, 13.27717982,
        6.65991877,  7.32754882,  6.55800267,  5.0209005 ,  7.72668467,
        9.96391045, 11.27160035,  9.41970031, 13.09426142,  7.50035355,
       11.05317597,  6.29034958, 13.6105595 ,  7.62232307, 11.64436684,
       12.86648771, 13.27717982,  9.36378438,  4.78258092, 13.25017211,
       11.72749923, 12.32408516, 13.50484539,  8.53501777, 10.51437947,
       11.05413545,  7.61088125,  9.75401449,  7.20826219, 10.51437947,
       12.33167382,  4.74468406,  9.97964013, 12.45418236, 12.7125079 ,
        9.17016857, 11.42947632,  6.21275016, 11.74901778,  7.58668144,
        7.54620849,  8.44940661, 11.63724792,  7.54620849, 13.41713661,
       13.41713661, 14.10045797, 10.06437598,  9.18690018, 13.31764181,
        4.98461983])}
{'P': tensor([[3.4870e-02, 2.0469e-01, 1.5586e-01,  ..., 2.3085e-05, 1.4306e-05,
         1.7505e-05],
        [2.4025e-03, 7.5722e-02, 1.7544e-01,  ..., 2.6617e-05, 2.8858e-05,
         1.7642e-05],
        [2.5741e-03, 1.7368e-02, 9.6230e-02,  ..., 2.8348e-05, 3.5623e-05,
         1.5305e-05],
        ...,
        [2.2552e-03, 9.2913e-04, 6.5322e-04,  ..., 2.4313e-02, 7.2262e-06,
         4.1260e-03],
        [1.4427e-04, 5.8288e-05, 4.5068e-05,  ..., 1.7060e-02, 1.8106e-02,
         2.5101e-04],
        [1.5848e-03, 1.4284e-03, 9.0414e-04,  ..., 1.6866e-04, 5.3416e-06,
         1.3645e-01]]), 'X': tensor([[  6.2770,  -0.0379,   2.4195],
        [  6.2905,   1.9366,   2.1169],
        [  6.2022,   2.4092,   3.0349],
        [  6.3606,   0.9204,   3.2525],
        [  6.3460,   1.4252,   1.2291],
        [  6.3305,   1.6119,   1.1538],
        [  1.1994,   7.6035,   2.4295],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.2770,  -0.0379,   2.4195],
        [  6.1685,  12.1979,  -6.2473],
        [  6.0310,  12.2795,  -6.4528],
        [  1.2273,  -4.8081,   9.8129],
        [  6.2535,  -0.1709,  -9.4102],
        [  9.5018,  -7.1945,  -1.1892],
        [ -1.6275, -10.7221,   3.2376],
        [  3.2701,   6.6338,   9.9209],
        [  8.3718,  -8.2953,   8.4666],
        [ -7.6650,   4.5092,   0.2520],
        [  2.0547,   7.2957,  -1.9343],
        [  4.2612, -10.5514,  -6.6268],
        [  5.5424,  -2.0885,   7.2487],
        [  2.5663,   7.0526,  -6.8769],
        [  3.1064,  13.5607,   6.7605],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.8653,   5.1694,  -4.0823],
        [  0.9809,   7.6642,  -1.4217],
        [ -8.9282,   2.1648,   8.4894],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.2895,   0.0405,   5.7652],
        [  9.6587,   9.1817,  -0.8212],
        [  6.3604,   0.9042,  -6.5901],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.5859,  -5.0393,   0.5476],
        [ -9.1706,   1.3863,   8.4032],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.4824,  -5.2575,  -0.0795],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.5996,   1.3971,  -1.0694],
        [ -4.8751,  -9.3148,   4.0406],
        [ 11.8951,  -3.1570,   7.0742],
        [  5.1519, -10.2387,   9.8191],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.4203,  -2.0340,   2.5856],
        [ -7.5603,   4.6459,   8.2002],
        [  9.6614,   9.1784,  -1.2087],
        [ 11.4495,   6.3026,   6.8257],
        [  2.1625,  -4.6697,   7.0658],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.2675,  12.1379,  -8.9720],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.0899,  -4.6870,  -7.4260],
        [  0.9478,  -4.8144,   0.6371],
        [  0.9478,  -4.8144,   0.6371],
        [  4.2096,   5.8815,   1.9230],
        [ -0.7599, -10.8995,  -6.0366],
        [  4.1537,   5.9340,  -6.0423],
        [ 12.5154,   2.5072,  -9.3274],
        [ 10.0995,  -6.4739,   2.0974],
        [ -5.0943,  -9.1682,   0.5073],
        [  9.1409,  -7.5789,   1.5814],
        [ -6.6998,   5.5954,   0.0823],
        [ -7.3554,   4.8985,  -0.6665],
        [  2.1120,  -4.6818,   8.6144],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.8325,  -1.5222,  -2.1341],
        [ 11.2517,  -4.6495,   2.9800],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.3633,  13.4794,   0.5193],
        [  9.5203,   9.3483,   5.8781],
        [ 12.0582,  -2.6640,   0.7132],
        [  0.5960, -11.0335,   3.1144],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -4.4503,   7.1378,   3.3751],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.9461,  -4.4077,  -0.4831],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.4196,   0.0478,   9.4140],
        [  6.1989,  -9.7591,   0.8289],
        [ -2.7572,   7.7194,   5.9580],
        [  3.2990,  -4.2415,  -7.4857],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.6078, -10.9229,   6.2869],
        [  2.0547,   7.2957,  -1.9343],
        [  4.8432,   5.1968,   1.5006],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -7.0253,  -7.4493,  -1.5978],
        [  4.1297,  -3.7098,  -3.9838],
        [ -3.5520,   7.4976,   9.1522],
        [ -9.3616,  -2.4749,   0.1793],
        [  0.5175,  14.0910,   4.6540],
        [  4.2275,  -3.6320,  -0.2714],
        [ -7.8849,  -6.3115,  -4.7479],
        [  7.3592,  11.3937,   6.2407],
        [  9.9058,  -6.7203,   5.6928],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.8363,   2.4086,  -5.6296],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.0844,  -8.3955,   0.7705],
        [  8.0381,  -8.5690,  -8.9523]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0100, 0.0672,  ..., 0.5984, 0.9403, 0.7455],
        [0.0100, 0.0000, 0.0748,  ..., 0.5900, 0.9335, 0.7363],
        [0.0672, 0.0748, 0.0000,  ..., 0.6191, 0.9424, 0.8091],
        ...,
        [0.5984, 0.5900, 0.6191,  ..., 0.0000, 0.3950, 0.6386],
        [0.9403, 0.9335, 0.9424,  ..., 0.3950, 0.0000, 0.9871],
        [0.7455, 0.7363, 0.8091,  ..., 0.6386, 0.9871, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.27715105,  6.58184474,  6.65368551,  6.42689705,  6.50409372,
        6.53250937,  7.69752262,  9.99211663,  6.27715105, 13.66896739,
       13.68063098,  4.96231532,  6.25586297, 11.9182956 , 10.84493732,
        7.39599986, 11.785566  ,  8.89301516,  7.57946459, 11.37939695,
        5.92281142,  7.50500064, 13.91198191, 10.0674887 ,  9.54154096,
       12.22276278,  7.09887557,  7.72668467,  9.18690018, 13.1471313 ,
        6.28961988, 13.32645552,  6.42439344,  9.15976746,  9.95552626,
        9.27474386, 10.30972193,  9.97964013,  5.04028538, 13.0743399 ,
       12.67680439, 10.51340829, 12.30695098, 11.46176259,  6.49538841,
        9.0368458 ,  9.63742669,  8.8737556 , 13.32613445, 13.06957425,
        5.14607014, 12.04924159, 13.20597932, 13.66051884,  9.20340031,
        5.13182506,  4.90676242,  4.90676242,  7.23276582, 10.92597196,
        7.24326992, 12.76408562, 11.99633819, 10.48840476, 11.87412439,
        8.72896381,  8.83725081,  5.13616224, 10.86693708, 10.0674887 ,
        6.02789634, 12.17451685, 13.09737053, 13.89264822, 13.3426541 ,
       12.3489324 , 11.04953767, 12.4980845 ,  8.41148722,  8.16714836,
        8.8353674 ,  5.30158497, 12.10222204,  9.41970031, 11.56146873,
        8.19703512,  5.37343173, 12.4217523 , 11.22993218,  7.57946459,
        7.10378919, 10.52587243, 10.23946247,  5.5513072 ,  8.29640613,
        9.68323686, 14.10045797,  5.57341623, 10.09979747, 13.56367857,
       11.9702762 ,  9.14973319,  9.15866607, 11.28895489, 10.36844813,
       11.74901778])}
{'P': tensor([[2.8028e-02, 1.5800e-01, 9.1356e-02,  ..., 2.4112e-04, 1.0857e-05,
         1.9100e-05],
        [9.2981e-04, 3.1749e-02, 1.1520e-01,  ..., 1.5217e-04, 1.4324e-05,
         3.6330e-05],
        [6.3698e-04, 2.8365e-03, 2.7029e-02,  ..., 1.2055e-04, 1.4720e-05,
         4.5259e-05],
        ...,
        [1.8378e-02, 1.0840e-02, 1.0914e-02,  ..., 9.0801e-02, 1.1914e-04,
         8.8463e-05],
        [1.3478e-04, 6.1023e-05, 5.2795e-05,  ..., 4.1219e-03, 5.7491e-02,
         8.1395e-05],
        [2.1253e-05, 1.3877e-05, 1.4275e-05,  ..., 4.6063e-04, 6.1528e-02,
         4.2279e-02]]), 'X': tensor([[  6.2790,  -0.0259,  -5.4116],
        [  6.2630,   2.1057,  -5.2136],
        [  6.0955,   2.8210,  -4.3321],
        [  6.1372,   2.6726,  -6.7885],
        [  5.9526,   3.2525,  -4.3821],
        [  5.8318,   3.5567,  -4.7692],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -7.0253,  -7.4493,  -1.5978],
        [  2.2644,  -4.6434,   9.7508],
        [  5.1787,  -2.6383,  -4.2533],
        [ -9.4714,  -1.3301,  -3.1998],
        [  5.4179,  -2.2925,  -3.4641],
        [ -9.1231,  -3.5739,   8.4708],
        [ 11.6019,   5.9540,  -5.1715],
        [  6.1730,  -0.5375,  -8.8861],
        [ -9.4157,   0.0821,  -5.0025],
        [ 11.5198,  -4.0923,   5.5088],
        [ -1.6334, -10.7207,   7.1397],
        [  4.7813,  -3.1139,   8.2691],
        [  9.6816,   9.1536,  -5.8543],
        [ -1.0482, -10.8487,  -5.9066],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -8.3062,  -5.6008,   0.0287],
        [  6.3460,   1.4252,   1.2291],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.9452, -10.3187,  -7.8621],
        [  5.7784,  -1.6401,   9.5587],
        [  0.5850,  -4.7982,   9.6639],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.3200,   7.1756,   1.1938],
        [  9.2692,   9.6370,   7.0746],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.5375,   7.0676,  -4.3490],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -6.6895,   5.6051,   2.7899],
        [  5.3858,   4.4260,  -9.6769],
        [  4.5430,   5.5438,  -3.8669],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.8902,  -1.3887,   4.0388],
        [ -5.2752,  -9.0409,  -8.2568],
        [  1.7000,  -4.7609,   2.7935],
        [ 11.1085,  -4.9204,   0.0736],
        [ -9.4351,  -1.8891,   3.5642],
        [  0.9036,   7.6840,   9.9237],
        [ -6.0175,  -8.4541,  -9.8408],
        [  3.0719, -10.8449,  -4.1673],
        [  3.9895,   6.0816,  -6.5235],
        [ -4.0409,   7.3169,  -3.5779],
        [  1.4150,  13.9647,  -4.0915],
        [  4.9152,  12.8653,  -1.4459],
        [  2.6310, -10.9195,   3.8054],
        [  6.2576,   2.1359,  -3.3398],
        [  6.3205,   1.7068,  -3.8619],
        [ -5.3328,  -8.9992,   7.3440],
        [  3.8182,  -3.9347,  -1.7961],
        [  0.8572,  14.0501,  -8.3364],
        [  6.3604,   0.9042,  -6.5901],
        [ 12.2589,  -1.9312,   9.8657],
        [ 10.0995,  -6.4739,   2.0974],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.1537,   5.9340,  -6.0423],
        [ 10.9394,   7.3069,  -8.9365],
        [ -9.4735,  -0.7274,  -2.6749],
        [  4.7482, -10.3907,   5.9149],
        [  2.7121,   6.9743,  -8.4202],
        [ 10.0995,  -6.4739,   2.0974],
        [  1.3670,  -4.7990,   3.2603],
        [  6.2038,   2.4020,  -3.1692],
        [  6.3534,   1.2990,  -7.2942],
        [ 12.2592,   3.9425,   2.0383],
        [ -9.3616,  -2.4749,   0.1793],
        [  8.5093,  -8.1764,  -3.2381],
        [  1.8766,   7.3694,   1.1887],
        [ -0.5739,   7.9056,   0.6382],
        [  5.6095,  -1.9702,  -7.8233],
        [  9.3424,  -7.3685,  -3.6807],
        [ -0.6323,   7.9085,  -5.9544],
        [ 12.6035,   0.7494,   7.1536],
        [ -4.6393,   7.0461,   1.1624],
        [ -9.3810,  -2.3455,  -9.5194],
        [  2.3200,   7.1756,   1.1938],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.1989,  -9.7591,   0.8289],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.1706,   1.3863,   8.4032],
        [  0.9478,  -4.8144,   0.6371],
        [ -1.6334, -10.7207,   7.1397],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.0345,   3.0175,  -1.4424],
        [ -9.4157,   0.0821,  -5.0025],
        [  4.2041,  -3.6509,  -4.9488],
        [ -6.3489,   5.9117,   6.5161],
        [  1.8721,  -4.7322,   0.5524],
        [ 12.5717,   1.9266,   7.4903],
        [ -5.6792,  -8.7352,   4.0318],
        [  8.6271,  -8.0715,  -7.4813],
        [  4.8051, -10.3703,   0.5268],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.2508,   1.0572,   2.9978]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.1649e-03, 4.1049e-02,  ..., 2.7922e-01, 1.0101e+00,
         9.9620e-01],
        [8.1649e-03, 0.0000e+00, 3.4651e-02,  ..., 2.7228e-01, 1.0026e+00,
         9.8811e-01],
        [4.1049e-02, 3.4651e-02, 0.0000e+00,  ..., 2.3826e-01, 9.8865e-01,
         9.6616e-01],
        ...,
        [2.7922e-01, 2.7228e-01, 2.3826e-01,  ..., 0.0000e+00, 8.7615e-01,
         7.9795e-01],
        [1.0101e+00, 1.0026e+00, 9.8866e-01,  ..., 8.7615e-01, 0.0000e+00,
         2.6796e-01],
        [9.9620e-01, 9.8811e-01, 9.6616e-01,  ..., 7.9795e-01, 2.6796e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.27905292,  6.60752373,  6.7166297 ,  6.69389431,  6.78325151,
        6.83083863, 12.25806559, 10.23946247,  5.16611782,  5.81199422,
        9.5642993 ,  5.8828986 ,  9.79814997, 13.04051744,  6.19633139,
        9.41605684, 12.22503754, 10.84437705,  5.70591986, 13.32374056,
       10.8992524 , 12.04924159, 10.01804223,  6.50409372,  5.8828986 ,
       11.4424816 ,  6.00663314,  4.83371341, 11.72749923,  7.54127425,
       13.37121923,  9.70108409, 11.63502679,  7.5092834 , 12.49291768,
        8.7273569 ,  6.97106912,  7.16747878,  5.21280156,  9.54154096,
        6.05165248, 10.46739073,  5.05534609, 12.1494023 ,  9.62238828,
        7.73692436, 10.37698435, 11.27160035,  7.27340834,  8.35856487,
       14.03618649, 13.77222752, 11.23201619,  6.61211224,  6.54693987,
       10.4606265 ,  5.48276038, 14.07623266,  6.42439344, 12.41012046,
       11.99633819, 13.09426142,  9.5014146 , 13.09737053,  7.24326992,
       13.15525803,  9.5014146 , 11.42420508,  7.48311002, 11.99633819,
        4.98989906,  6.65259775,  6.48486171, 12.87752052,  9.68323686,
       11.80092578,  7.60463349,  7.92644746,  5.94541803, 11.89855267,
        7.9337647 , 12.62576026,  8.43626114,  9.66977802,  7.54127425,
        7.96181288, 11.56146873, 12.25806559,  8.95983586,  9.27474386,
        4.90676242, 10.84437705,  9.47369628, 10.61047596,  6.74686071,
        9.41605684,  5.56809335,  8.67502348,  5.08910491, 12.71843462,
       10.41909211, 11.81423161, 11.42947632,  5.24137909, 10.01804223,
        9.3109884 ])}
{'P': tensor([[4.4833e-02, 1.9112e-01, 1.7182e-01,  ..., 1.1444e-04, 1.4756e-04,
         1.0061e-05],
        [1.1329e-02, 5.2970e-02, 2.1794e-01,  ..., 1.4133e-04, 1.8178e-04,
         1.1895e-05],
        [1.8988e-03, 1.9175e-03, 5.4104e-02,  ..., 3.7187e-04, 4.7534e-04,
         1.0093e-05],
        ...,
        [2.1574e-05, 2.1897e-05, 2.5535e-05,  ..., 6.7496e-02, 1.1750e-01,
         2.5321e-04],
        [3.1615e-05, 3.1964e-05, 3.4597e-05,  ..., 1.0678e-02, 7.4631e-02,
         3.6705e-04],
        [8.2311e-04, 8.2154e-04, 4.2962e-04,  ..., 1.1800e-05, 1.1800e-05,
         1.0701e-01]]), 'X': tensor([[  6.2883,   0.0325,   5.2692],
        [  6.2975,   0.0947,   5.1791],
        [  6.2555,   2.1474,   4.1685],
        [  5.9883,   3.1536,   6.2320],
        [  6.1737,   2.5305,   3.9049],
        [  6.1752,   2.5242,   3.5874],
        [  3.7916,  -3.9524,  -7.9547],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.4748,  -0.7765,  -5.6695],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.7349,  -3.5838,   2.1644],
        [  0.3038,   7.8092,   0.4164],
        [  0.9809,  -4.8145,   6.8706],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -8.3653,   3.4184,   9.3454],
        [  0.7275,  14.0667,  -0.9815],
        [  4.5491,  -3.3485,   6.8412],
        [ -9.4203,  -2.0340,   2.5856],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.3261,   7.1726,   6.2522],
        [ -9.1231,  -3.5739,   8.4708],
        [ -5.6174,   6.4708,  -1.7339],
        [ -1.3178, -10.7939,   0.4236],
        [  6.3087,   1.8053,  -1.9481],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.0054,  -1.0902,   8.3079],
        [  7.7794,  11.0629,   8.9069],
        [ -0.8675, -10.8815,   6.8561],
        [ -9.3419,   0.5972,  -1.2841],
        [  2.6970,   6.9826,   9.7350],
        [ 12.5763,   0.1333,  -9.2650],
        [  4.2612, -10.5514,  -6.6268],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.8704,  -9.3809,   7.5769],
        [  1.4150,  13.9647,  -4.0915],
        [  6.9284,  11.7061,  -0.3732],
        [  9.1409,  -7.5789,   1.5814],
        [  6.9433,  -9.3362,  -0.3421],
        [  3.9022,  13.2906,   5.5788],
        [  3.2809,  13.5061,  -4.0199],
        [  0.9478,  -4.8144,   0.6371],
        [  4.6063,  13.0051,  -3.1456],
        [ -4.9217,   6.8979,  -1.7667],
        [  2.3889,  -4.6082,  -8.7291],
        [ -9.4501,  -1.7133,   2.2980],
        [  5.7151,  -1.7700,   3.1542],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.4624,  -0.4633,  -7.3918],
        [  1.9312,  13.8649,   8.3563],
        [  5.6248,   3.9984,   8.4933],
        [  6.3516,   0.6425,   0.3425],
        [  3.9895,   6.0816,  -6.5235],
        [  1.2273,  -4.8081,   9.8129],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.5927,   0.4199,  -9.4997],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.0449,  13.8402,  -7.1642],
        [  3.3633,  13.4794,   0.5193],
        [ -8.9788,   2.0208,  -3.2867],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.8704,  -9.3809,   7.5769],
        [  2.3367,   7.1676,  -9.5263],
        [  5.9548,  -1.2273,  -2.6520],
        [ -1.0407,   7.9166,  -6.6021],
        [  1.6408,   7.4591,  -4.2294],
        [  6.3587,   1.1596,  -3.6629],
        [  2.1120,  -4.6818,   8.6144],
        [  4.6063,  13.0051,  -3.1456],
        [  3.9022,  13.2906,   5.5788],
        [  7.1441,  11.5529,   2.3232],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.1531,  -5.8748,   2.6934],
        [ -8.7735,   2.5640,  -6.3313],
        [  6.3583,   1.1729,   8.7835],
        [  5.0348,  -2.8229,   3.5436],
        [  0.5175,  14.0910,   4.6540],
        [  1.7228,  -4.7575,   8.3396],
        [  6.3551,   0.7149,   1.3344],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.7420,  -4.4896,  -3.7139],
        [  7.5064,  -8.9651,  -0.0994],
        [ -5.9987,   6.1949,  -6.2114],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.3542,   0.6934,  -3.3840],
        [ -2.1817,   7.8274,  -4.6298],
        [ -9.4716,  -0.6673,   7.6068],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.3384,  -2.6161,  -4.7630],
        [  5.7845,   3.6652,   0.9175],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.3246,   1.6696,  -0.1544],
        [  8.5198,  10.4091,  -8.6486],
        [ -0.1963,   7.8765,  -1.1371],
        [  7.1697,  11.5343,  -0.7276],
        [  5.6022,  -1.9833,   8.3081],
        [ -2.0445,   7.8469,  -0.5857],
        [ -5.5525,   6.5146,  -4.4984],
        [ -6.0195,   6.1789,  -2.9496],
        [ -2.2508, -10.5479,  -2.7235]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.0203e-03, 5.8526e-02,  ..., 8.8162e-01, 8.2643e-01,
         1.0116e+00],
        [6.0203e-03, 0.0000e+00, 5.2575e-02,  ..., 8.7677e-01, 8.2197e-01,
         1.0077e+00],
        [5.8527e-02, 5.2575e-02, 0.0000e+00,  ..., 8.3042e-01, 7.7847e-01,
         9.6849e-01],
        ...,
        [8.8162e-01, 8.7677e-01, 8.3042e-01,  ..., 1.5170e-04, 1.0201e-01,
         2.5671e-01],
        [8.2643e-01, 8.2197e-01, 7.7847e-01,  ..., 1.0201e-01, 0.0000e+00,
         2.2216e-01],
        [1.0116e+00, 1.0077e+00, 9.6849e-01,  ..., 2.5671e-01, 2.2216e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.28835101,  6.29821673,  6.61386436,  6.76791483,  6.67218687,
        6.67121748,  5.47702766, 10.76154256,  9.50654967,  6.00732193,
       12.26997165,  7.81509631,  4.91336994,  9.51074763,  9.0368458 ,
       14.08549802,  5.64865173,  9.63742669, 11.63724792,  7.5403822 ,
        9.79814997,  8.56889738, 10.87409005,  6.5618968 ,  5.70405525,
        6.19707944,  9.75200848,  9.86089579,  6.10360073, 13.52430176,
       10.91601655,  9.36093562,  7.48538905, 12.5769659 , 11.37939695,
        9.5642993 , 11.62770332, 14.03618649, 13.60275211, 11.87412439,
       11.63502679, 13.85158417, 13.89886469,  4.90676242, 13.79676654,
        8.47371868,  5.19065439,  9.60412735,  5.98285319,  8.62260327,
        9.47369628, 13.99877413,  6.90115901,  6.38399789,  7.27340834,
        4.96231532, 10.00794468, 12.59969936,  5.5513072 ,  6.25586297,
        5.38342077,  9.85330646, 13.99048094, 13.89264822,  9.20340031,
        5.70405525, 11.62770332,  7.53884104,  6.07992623,  7.98469166,
        7.6374603 ,  6.46356806,  5.13616224, 13.79676654, 13.85158417,
       13.58333857, 12.21532282, 10.04916678,  9.14044984,  6.46560181,
        5.77219077, 14.10045797,  5.05981914,  6.39520397,  9.68324553,
        5.26067019, 11.69264243,  8.62329466,  5.53255882,  6.39187285,
        8.12580976,  9.49511524, 12.12641969,  6.61211224,  9.69792513,
        6.84796202, 10.37698435,  6.54127431, 13.45124917,  7.8788965 ,
       13.58101794,  5.94293185,  8.10886021,  8.55982105,  8.62631801,
       10.78533838])}
{'P': tensor([[4.2132e-02, 1.7256e-01, 1.5093e-01,  ..., 1.7000e-05, 5.1938e-04,
         3.7649e-04],
        [4.9265e-03, 4.8030e-02, 2.5003e-01,  ..., 1.7778e-05, 3.7876e-04,
         2.6597e-04],
        [1.5934e-03, 2.0175e-03, 8.1661e-02,  ..., 3.0789e-05, 3.0644e-04,
         2.5363e-04],
        ...,
        [9.9858e-04, 6.7188e-04, 2.3585e-04,  ..., 1.2753e-01, 4.4574e-03,
         2.7678e-04],
        [1.2041e-02, 8.2534e-03, 6.7709e-03,  ..., 6.4416e-05, 9.1314e-02,
         3.3045e-02],
        [1.3671e-03, 1.1621e-03, 1.8500e-03,  ..., 2.5125e-05, 2.7099e-03,
         1.4825e-01]]), 'X': tensor([[  6.2895,   0.0405,   5.7652],
        [  6.3569,   0.7591,   6.5011],
        [  5.9883,   3.1536,   6.2320],
        [  6.0073,   3.0986,   6.7937],
        [  6.2555,   2.1474,   4.1685],
        [  6.1737,   2.5305,   3.9049],
        [ 12.3357,   3.6001,   0.7903],
        [  8.0877,  -8.5297,  -1.1050],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.9461,  -4.4077,  -0.4831],
        [ 10.1503,   8.5414,  -7.2681],
        [  0.5600, -11.0321,  -0.8660],
        [ 10.4758,  -5.9539,   1.3667],
        [ -7.6525,  -6.6534,   5.1721],
        [  3.9591, -10.6391,  -6.2557],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.5224,  -0.4497,   4.8448],
        [ 10.9394,   7.3069,  -8.9365],
        [  9.3836,   9.5075,  -5.9278],
        [ -8.9606,  -4.0994,   1.7886],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.3900,   3.3727,   3.1650],
        [ -8.8400,   2.3991,   1.3457],
        [ 12.5258,  -0.4200,  -9.3493],
        [  3.6249, -10.7255,   5.6605],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.1752,   2.5242,   3.5874],
        [ -1.9600,   7.8577,   6.9121],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.3464,   0.5708,  -1.6397],
        [  5.2608,  -2.5256,  -9.1446],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.2921,  -1.7902,  -4.0870],
        [  5.2939,  -2.4784,  -4.5840],
        [ 12.2589,  -1.9312,   9.8657],
        [ 11.5198,  -4.0923,   5.5088],
        [  0.3715,  -4.7756,   0.2312],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.1984,  -0.4327,   1.5438],
        [  3.3633,  13.4794,   0.5193],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.0233,   4.9647,  -3.1788],
        [ 12.4592,  -0.9168,  -6.7571],
        [  5.2608,  -2.5256,  -9.1446],
        [ 11.2939,   6.6325,  -5.8311],
        [ -5.0550,   6.8231,  -9.2502],
        [ -1.5875,   7.8944,  -9.4286],
        [  5.0277,   4.9589,   7.0175],
        [ 12.2592,   3.9425,   2.0383],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.3889,  -4.6082,  -8.7291],
        [  9.3560,   9.5391,   9.4943],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.9314,  -9.3435,  -1.2729],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 10.0567,   8.6697,   4.6329],
        [ -3.3116, -10.1539,   7.5919],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.0617,  12.7957,   8.3544],
        [  2.6614,  -4.5192,  -9.3641],
        [ 11.0373,   7.1299,  -8.5895],
        [  2.0899,  -4.6870,  -7.4260],
        [  1.6954,  -4.7616,   9.8198],
        [ -6.1899,  -8.3011,  -7.9269],
        [  5.7845,   3.6652,   0.9175],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.3453,  -1.5462,   7.0212],
        [  4.1537,   5.9340,  -6.0423],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.7078,   2.7182,   1.8146],
        [ 12.1940,  -2.1881,   7.1499],
        [  4.4039,   5.6903,   8.7366],
        [  2.2076,  13.8032,  -1.5099],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -6.3306,   5.9272,   5.6935],
        [ 10.2712,   8.3707,  -4.9743],
        [  6.1575,  12.2046,   3.3257],
        [ -5.5525,   6.5146,  -4.4984],
        [ 10.1503,   8.5414,  -7.2681],
        [  4.8674,  12.8875,   6.7622],
        [  5.8323,  -1.5228,   5.1132],
        [  4.5057,  -3.3894,  -0.8160],
        [  8.5716,  10.3596,   6.8818],
        [  4.2096,   5.8815,   1.9230],
        [ -5.6423,   6.4537,  -1.2091],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.1142,  -2.7230,  -4.1966],
        [  1.7218,  -4.7576,  -3.3617],
        [  7.8982,  -8.6778,   0.0239],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.9636,   2.0648,   8.1946],
        [  4.3554,   5.7393,  -9.0766],
        [  0.7958,  -4.8109,  -8.2248],
        [ -2.5161,   7.7699,  -9.3531],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.8275,  -1.5334,   2.0124],
        [ -4.0868,   7.2981,   8.1874],
        [  6.3305,   1.6119,   1.1538],
        [ -7.6293,  -6.6859,  -5.5161],
        [  0.9392,  -4.8143,  -3.4649],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.2790,  -0.0259,  -5.4116]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0424, 0.0460,  ..., 0.9531, 0.4767, 0.7275],
        [0.0424, 0.0000, 0.0266,  ..., 0.9594, 0.5189, 0.7699],
        [0.0460, 0.0266, 0.0000,  ..., 0.9328, 0.5127, 0.7660],
        ...,
        [0.9531, 0.9594, 0.9328,  ..., 0.0000, 0.9664, 1.1277],
        [0.4767, 0.5189, 0.5127,  ..., 0.9664, 0.0000, 0.2582],
        [0.7275, 0.7699, 0.7660,  ..., 1.1277, 0.2582, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.28961988,  6.4020404 ,  6.76791483,  6.7593917 ,  6.61386436,
        6.67218687, 12.85032836, 11.75438006,  8.97052115,  5.30158497,
       13.26590071, 11.04629471, 12.0495521 , 10.14044489, 11.35182589,
       12.00995898, 12.53047489, 13.15525803, 13.35832725,  9.85384623,
       10.63038565,  9.70108409, 13.103709  ,  9.04255655,  9.15976746,
       12.53285633, 11.32149259, 12.7125079 ,  6.67121748,  8.09842704,
        9.4997987 ,  9.36378438,  5.83560009, 12.67576792, 12.4217523 ,
        5.84533345, 12.41012046, 12.22503754,  4.79003361,  6.19633139,
        6.2134961 , 13.89264822,  5.56809335,  7.06272256, 12.49291768,
        5.83560009, 13.09737053,  8.49161154,  8.05242591,  7.06169391,
       12.87752052,  8.8353674 ,  5.19065439, 13.3614598 , 12.91505111,
       11.63383221, 12.49291768, 13.2778378 , 10.68031351, 10.23946247,
       12.10942412, 13.76047999,  5.24461068, 13.1399181 ,  5.13182506,
        5.05444288, 10.35485331,  6.84796202, 12.67680439, 12.44177641,
        7.24326992,  5.83560009,  9.12220271, 12.38881635,  7.19534146,
       13.97857591, 12.52372629,  8.67227386, 13.25017211, 13.66991027,
        8.55982105, 13.26590071, 13.77604293,  6.02779201,  5.63824971,
       13.44593519,  7.23276582,  8.5724042 ,  6.23262289,  5.79391825,
        5.05963217, 11.73397309,  6.2322769 ,  9.19837792,  7.20482224,
        4.8763135 ,  8.16714836, 12.1494023 ,  6.02588517,  8.36445758,
        6.53250937, 10.14436912,  4.90505566, 10.37998078,  5.63824971,
        6.27905292])}
{'P': tensor([[3.0507e-02, 2.0793e-01, 1.6378e-01,  ..., 2.9138e-03, 7.3913e-03,
         3.7757e-06],
        [1.9512e-03, 3.0614e-02, 1.3997e-01,  ..., 4.3134e-03, 1.0783e-02,
         3.5940e-06],
        [1.4220e-03, 2.8059e-03, 4.6667e-02,  ..., 1.0119e-02, 3.3076e-02,
         3.9194e-06],
        ...,
        [1.0684e-05, 1.5128e-05, 2.7081e-05,  ..., 2.2150e-02, 7.5042e-03,
         1.8711e-07],
        [1.3279e-04, 1.5287e-04, 2.0046e-04,  ..., 4.9736e-04, 4.9095e-02,
         5.0688e-06],
        [5.3104e-04, 4.3996e-04, 2.1145e-04,  ..., 4.8401e-06, 1.7922e-05,
         1.9978e-01]]), 'X': tensor([[  6.2902,   0.0451,   3.7116],
        [  6.3606,   0.9204,   3.2525],
        [  6.2555,   2.1474,   4.1685],
        [  6.1752,   2.5242,   3.5874],
        [  6.1737,   2.5305,   3.9049],
        [  6.2022,   2.4092,   3.0349],
        [ 11.1469,   6.9237,   1.6160],
        [  1.7122,  -4.7591,  -3.6734],
        [  1.4489,   7.5256,   4.4529],
        [ -9.3076,   0.7852,  -6.7010],
        [  3.7916,  -3.9524,  -7.9547],
        [  9.5203,   9.3483,   5.8781],
        [  3.9913,  -3.8139,  -6.8311],
        [ -1.6334, -10.7207,   7.1397],
        [  7.7750,  11.0665,   2.8364],
        [ 11.5456,   6.0861,   8.1101],
        [  0.1287, -11.0065,   6.5757],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.5267,  -0.4125,  -8.7331],
        [  3.1756, -10.8247,   6.8882],
        [ -2.0721,   7.8432,   6.6380],
        [ -8.8154,   2.4611,   2.5198],
        [ -7.8849,  -6.3115,  -4.7479],
        [  5.0617,  12.7957,   8.3544],
        [  6.3122,  -9.6994,  -6.3282],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.3464,   0.5708,  -1.6397],
        [ -7.4808,   4.7462,   2.0266],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.9526,   3.2525,  -4.3821],
        [  6.2630,   2.1057,  -5.2136],
        [  6.2675,  12.1379,  -8.9720],
        [ -7.0120,   5.2825,   7.7762],
        [  8.6271,  -8.0715,  -7.4813],
        [  8.0381,  -8.5690,  -8.9523],
        [ -8.8154,   2.4611,   2.5198],
        [ -2.4956, -10.4682,   3.5632],
        [  4.6063,  13.0051,  -3.1456],
        [  2.5375,   7.0676,  -4.3490],
        [  2.7976,  -4.4682,  -4.2020],
        [  1.8719,   7.3713,  -3.7465],
        [ 12.2692,  -1.8882,   3.9253],
        [ 12.5154,   2.5072,  -9.3274],
        [  3.4604,  -4.1545,   1.8091],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.2036,   2.4030,  -1.2500],
        [  2.5375,   7.0676,  -4.3490],
        [  0.5960, -11.0335,   3.1144],
        [  0.3231,  -4.7691,  -1.6120],
        [  0.3724,   7.7974,  -6.8722],
        [  2.6613,   7.0021,   4.5783],
        [  5.0233,   4.9647,  -3.1788],
        [  4.8432,   5.1968,   1.5006],
        [ 12.2218,   4.0953,  -8.2072],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.1989,  -9.7591,   0.8289],
        [ -1.6086,   7.8928,   4.3385],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.3444,   0.5821,   6.9101],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.4671,  -0.5553,   0.1117],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.9193,  -1.3176,   5.1739],
        [  2.6321,  -4.5296,  -2.5159],
        [  3.3633,  13.4794,   0.5193],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.3464,   0.5708,  -1.6397],
        [ 11.3118,   6.5957,  -4.5667],
        [ 12.5789,   0.1722,   1.6470],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.1196,  -2.7160,   4.2705],
        [  6.3303,   1.6145,  -8.4083],
        [  4.9152,  12.8653,  -1.4459],
        [  5.4242, -10.1261,   3.9595],
        [  6.3550,   0.7129,   0.4917],
        [  9.1409,  -7.5789,   1.5814],
        [ -4.4503,   7.1378,   3.3751],
        [ -4.4520,   7.1370,  -8.7867],
        [  5.2608,  -2.5256,  -9.1446],
        [ -6.3252,  -8.1760,  -1.0891],
        [  5.4242, -10.1261,   3.9595],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.3717,  -5.4769,  -9.2228],
        [  9.2979,   9.6048,   7.3138],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.3505,   0.6235,  -8.5849],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.9505,  -4.7171,   9.9868],
        [  8.7715,  -7.9390,   5.2900],
        [  1.5653,  13.9377,   4.1773],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.1931,  -4.7491,   7.4385],
        [  6.0310,  12.2795,  -6.4528],
        [ -5.1377,   6.7751,  -0.4438],
        [  2.5382,   7.0672,   4.5097],
        [  5.9912,   3.1453,  -3.9989],
        [ 11.2939,   6.6325,  -5.8311],
        [ -3.5173, -10.0623,   4.8280],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.6319,  11.9066,   3.5165],
        [  1.7659,   7.4127,   4.7871],
        [ -2.9497, -10.3028,  -7.3479]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 2.0890e-02, 5.2760e-02,  ..., 3.5220e-01, 2.3948e-01,
         1.1205e+00],
        [2.0890e-02, 0.0000e+00, 7.3526e-02,  ..., 3.6372e-01, 2.5010e-01,
         1.1058e+00],
        [5.2760e-02, 7.3526e-02, 0.0000e+00,  ..., 3.3364e-01, 2.1297e-01,
         1.1536e+00],
        ...,
        [3.5220e-01, 3.6372e-01, 3.3364e-01,  ..., 0.0000e+00, 5.1435e-01,
         1.4657e+00],
        [2.3948e-01, 2.5010e-01, 2.1297e-01,  ..., 5.1435e-01, 1.5170e-04,
         1.0775e+00],
        [1.1205e+00, 1.1058e+00, 1.1536e+00,  ..., 1.4657e+00, 1.0775e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.29034958,  6.42689705,  6.61386436,  6.67121748,  6.67218687,
        6.65368551, 13.12218013,  5.05774295,  7.66377554,  9.34061938,
        5.47702766, 13.3426541 ,  5.5205189 , 10.84437705, 13.5247215 ,
       13.05148826, 11.00726483,  9.34127013, 12.53345445, 11.28093507,
        8.11227668,  9.15253344, 10.09979747, 13.76047999, 11.57249984,
        9.61942445,  6.21275016,  9.36378438,  8.859416  ,  5.63824971,
        6.78325151,  6.60752373, 13.66051884,  8.77913428, 11.81423161,
       11.74901778,  9.15253344, 10.76154256, 13.79676654,  7.5092834 ,
        5.27178363,  7.60529892, 12.41366754, 12.76408562,  5.40689332,
       10.63038565,  6.652742  ,  7.5092834 , 11.04953767,  4.78002665,
        7.80625981,  7.49076899,  7.06272256,  7.10378919, 12.889695  ,
        9.75401449,  9.85384623, 11.56146873,  8.0550286 ,  5.91677188,
        9.14973319,  9.36256096,  6.49538841,  9.48337063, 10.52357479,
        6.06416957,  5.23878371, 13.89264822,  5.39098669,  9.36378438,
       13.09426142, 12.58006038,  5.94541803,  5.79543038,  6.53290663,
       13.77222752, 11.48734978,  6.3949043 , 11.87412439,  8.41148722,
        8.41171369,  5.83560009, 10.33712258, 11.48734978, 13.26590071,
       10.00410625, 13.36800073,  5.5671389 ,  6.38105015,  8.05242591,
        5.10446653, 11.83075005, 14.02532758,  8.62631801,  4.75303127,
       13.68063098,  8.50278896,  7.50917551,  6.76661595, 13.09737053,
       10.65929968, 10.09979747, 10.61047596, 13.62896593,  7.62011734,
       10.7167285 ])}
{'P': tensor([[3.7508e-02, 1.5432e-01, 1.2320e-01,  ..., 8.7857e-03, 3.5396e-05,
         2.6212e-05],
        [1.4388e-03, 4.0596e-02, 3.2926e-01,  ..., 1.7899e-02, 3.3530e-05,
         1.4675e-05],
        [2.1781e-03, 6.0047e-03, 7.2456e-02,  ..., 3.3693e-02, 5.9636e-05,
         2.3466e-05],
        ...,
        [3.3817e-04, 5.0691e-04, 5.5876e-04,  ..., 1.2078e-01, 1.2593e-05,
         2.2978e-06],
        [4.1059e-04, 5.7018e-04, 6.1317e-04,  ..., 4.8946e-05, 1.0436e-01,
         1.5994e-02],
        [5.7032e-04, 9.6624e-04, 1.0502e-03,  ..., 5.3329e-05, 1.4813e-03,
         9.0774e-02]]), 'X': tensor([[  6.2975,   0.0947,   5.1791],
        [  6.2555,   2.1474,   4.1685],
        [  6.1737,   2.5305,   3.9049],
        [  5.9883,   3.1536,   6.2320],
        [  6.1752,   2.5242,   3.5874],
        [  6.0073,   3.0986,   6.7937],
        [ -7.4733,   4.7555,   6.2490],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.9452, -10.3187,  -7.8621],
        [ -7.9667,   4.0814,  -0.6397],
        [  5.1793,   4.7461,   0.3899],
        [  0.7958,  -4.8109,  -8.2248],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.5813,  -4.5471,   5.3597],
        [ -5.1974,   6.7396,   3.3484],
        [  5.9548,  -1.2273,  -2.6520],
        [  0.9478,  -4.8144,   0.6371],
        [ 12.3759,  -1.3943,   6.4659],
        [  1.6232,  -4.7717,  -2.5655],
        [ -1.6220,   7.8917,   0.0315],
        [ -3.5029,   7.5138,   8.1853],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -0.1626, -10.9796,   6.4629],
        [  4.0362,  13.2397,  -7.0030],
        [  8.5716,  10.3596,   6.8818],
        [  2.6613,   7.0021,   4.5783],
        [  5.7845,   3.6652,   0.9175],
        [ -8.0330,  -6.0765,   6.7945],
        [  2.3261,   7.1726,   6.2522],
        [  6.3589,   0.8239,  -8.4230],
        [  2.0937,  13.8293,  -2.3533],
        [  6.0031,  12.2958,  -6.5992],
        [ 12.5789,   0.1722,   1.6470],
        [  5.0233,   4.9647,  -3.1788],
        [  6.2975,   0.0947,   5.1791],
        [  5.6248,   3.9984,   8.4933],
        [  6.3589,   0.8239,  -8.4230],
        [  6.0788,  -0.8710,  -6.8719],
        [ -9.1749,  -3.3803,   6.4629],
        [  2.0937,  13.8293,  -2.3533],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -6.7643,  -7.7372,  -9.6663],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.3087,   1.8053,  -1.9481],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.4426, -10.1181,  -5.1268],
        [  0.7145,  -4.8071,  -2.3766],
        [  5.5626,  -2.0536,   8.4433],
        [  2.4697,  -4.5836,   1.4593],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3569,   0.7591,   6.5011],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.3804,  -2.3504,   7.2744],
        [  0.1677,  -4.7447,  -6.1529],
        [  1.1350,  14.0104,   0.6321],
        [ 11.3800,  -4.3919,   1.4066],
        [  4.5503,   5.5359,   7.4735],
        [  1.7903,  -4.7467,   0.1168],
        [  5.1890,   4.7318,   0.9515],
        [ -1.9646,   7.8571,   3.3817],
        [ 12.2592,   3.9425,   2.0383],
        [ -7.6650,   4.5092,   0.2520],
        [ -1.2454,   7.9127,  -0.5715],
        [  0.9445,   7.6736,   0.5095],
        [  3.1271, -10.8343,   2.6442],
        [  5.4426, -10.1181,  -5.1268],
        [  1.7301,   7.4262,   5.0522],
        [  3.1480,  -4.3165,  -7.8002],
        [  7.5309,  11.2617,   7.2656],
        [ -2.0445,   7.8469,  -0.5857],
        [ -4.3359,  -9.6431,  -9.0616],
        [ 10.0567,   8.6697,   4.6329],
        [  1.8721,  -4.7322,   0.5524],
        [ 11.6544,   5.8273,  -3.1375],
        [ -3.1671,   7.6158,   6.0765],
        [ -9.4309,  -0.0599,  -4.0830],
        [  2.8994,   6.8675,  -9.5987],
        [ -1.7290,   7.8825,  -3.3831],
        [ -9.3419,   0.5972,  -1.2841],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.8400,   2.3991,   1.3457],
        [  6.2630,   2.1057,  -5.2136],
        [  5.0617,  12.7957,   8.3544],
        [  4.5491,  -3.3485,   6.8412],
        [ -3.5173,   7.5091,  -8.9585],
        [ -1.3178, -10.7939,   0.4236],
        [  3.9663,   6.1018,  -5.4104],
        [  6.2975,   0.0947,   5.1791],
        [  2.9260,  -4.4162,   8.3808],
        [  4.2612, -10.5514,  -6.6268],
        [  4.4633,   5.6287,  -0.7754],
        [  0.1252, -11.0062,   7.2564],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.3205,   1.7068,  -3.8619],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.0788,  -0.8710,  -6.8719],
        [  9.5261,  -7.1675,   1.4206],
        [  3.7237,  -3.9966,  -5.0376],
        [  3.6249, -10.7255,   5.6605],
        [ 10.0567,   8.6697,   4.6329],
        [  2.0243,   7.3086,   7.5432],
        [  3.7992,  -3.9474,  -8.9019],
        [ 10.4994,  -5.9193,  -7.5153]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0526, 0.0693,  ..., 0.2507, 0.7885, 0.9279],
        [0.0526, 0.0000, 0.0169,  ..., 0.2760, 0.7365, 0.8755],
        [0.0693, 0.0169, 0.0000,  ..., 0.2831, 0.7206, 0.8593],
        ...,
        [0.2507, 0.2760, 0.2831,  ..., 0.0000, 0.9460, 1.0712],
        [0.7885, 0.7365, 0.7206,  ..., 0.9460, 0.0000, 0.1484],
        [0.9279, 0.8755, 0.8593,  ..., 1.0712, 0.1484, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.29821673,  6.61386436,  6.67218687,  6.76791483,  6.67121748,
        6.7593917 ,  8.8580671 ,  5.53175275, 11.4424816 ,  8.95133961,
        7.02496223,  4.8763135 , 11.86239961,  5.2287058 ,  8.51088913,
        6.07992623,  4.90676242, 12.45418236,  5.04028538,  8.05668609,
        8.29022197,  5.91677188, 10.06407839, 10.98076873, 13.84126197,
       13.44593519,  7.49076899,  6.84796202, 10.07238463,  7.5403822 ,
        6.41202986, 13.98691426, 13.6829843 , 12.58006038,  7.06272256,
        6.29821673,  6.90115901,  6.41202986,  6.14086582,  9.77777346,
       13.98691426,  9.51074763, 10.27716434,  4.92497514,  6.5618968 ,
       12.07225469, 11.4890861 ,  4.85995002,  5.92952538,  5.20659892,
        7.96181288,  6.4020404 , 12.7125079 ,  5.87132718,  4.74770899,
       14.05633441, 12.19805394,  7.16598273,  5.07306274,  7.02253235,
        8.09899874, 12.87752052,  8.89301516,  8.01009083,  7.73151309,
       11.27656488, 11.4890861 ,  7.62509582,  5.34250225, 13.54774355,
        8.10886021, 10.57302546, 13.2778378 ,  5.08910491, 13.03002338,
        8.24808832,  9.43112957,  7.4544866 ,  8.06990483,  9.36093562,
       12.0495521 ,  9.15976746,  6.60752373, 13.76047999,  5.64865173,
        8.29203356, 10.87409005,  7.27759233,  6.29821673,  5.29755487,
       11.37939695,  7.18354885, 11.00694688, 10.30972193, 10.35485331,
       12.64235359,  6.54693987,  4.978704  ,  6.14086582, 11.92133129,
        5.46244988, 11.32149259, 13.2778378 ,  7.58378655,  5.47866221,
       12.05300943])}
{'P': tensor([[3.2256e-02, 3.5594e-01, 1.6929e-01,  ..., 1.0208e-05, 3.7169e-05,
         1.4370e-05],
        [4.4051e-03, 6.2166e-02, 2.9650e-01,  ..., 1.8082e-05, 6.5163e-05,
         2.5523e-05],
        [5.1577e-03, 6.6947e-03, 1.0880e-01,  ..., 3.4828e-05, 1.1777e-04,
         4.7691e-05],
        ...,
        [4.6816e-05, 4.6101e-05, 7.0833e-05,  ..., 7.7530e-02, 1.6407e-02,
         1.6118e-03],
        [1.5733e-04, 1.5513e-04, 2.4024e-04,  ..., 1.6255e-02, 5.8821e-02,
         1.3111e-03],
        [1.2275e-04, 1.1585e-04, 1.4650e-04,  ..., 4.1051e-02, 9.4174e-02,
         6.1070e-02]]), 'X': tensor([[  6.3048,   0.1469,  -9.1691],
        [  6.3360,   0.4258,  -9.2509],
        [  6.3303,   1.6145,  -8.4083],
        [  6.3505,   0.6235,  -8.5849],
        [  6.3589,   0.8239,  -8.4230],
        [  5.8617,   3.4853,  -9.8080],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -2.9497, -10.3028,  -7.3479],
        [  4.5800,   5.5034,  -3.9432],
        [  4.6357,   5.4415,   7.9067],
        [  3.0719, -10.8449,  -4.1673],
        [  6.3551,   0.7149,   1.3344],
        [  4.5491,  -3.3485,   6.8412],
        [ -8.3900,   3.3727,   3.1650],
        [ -0.0254, -10.9932,   3.2200],
        [  0.9392,  -4.8143,  -3.4649],
        [ -1.6086,   7.8928,   4.3385],
        [ -1.0482, -10.8487,  -5.9066],
        [  3.2647, -10.8066,  -0.3768],
        [  4.6992,   5.3690,   3.0935],
        [  2.3889,  -4.6082,  -8.7291],
        [  5.8318,   3.5567,  -4.7692],
        [  2.5813,  -4.5471,   5.3597],
        [ -5.6423,   6.4537,  -1.2091],
        [ -8.5900,   2.9767,   7.4811],
        [ -2.9497, -10.3028,  -7.3479],
        [  6.2247,  -0.3150,  -9.3657],
        [  2.4639, -10.9431,  -7.3718],
        [ -0.4341, -10.9473,   7.4509],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.9587,  12.3216,   9.1695],
        [ -6.2200,  -8.2737,   5.5079],
        [  3.9663,   6.1018,  -5.4104],
        [ -2.1533, -10.5778,   1.0112],
        [  1.2273,  -4.8081,   9.8129],
        [  2.3889,  -4.6082,  -8.7291],
        [  4.8051, -10.3703,   0.5268],
        [ -9.4430,  -0.1897,   4.8383],
        [ -5.6169,  -8.7844,   8.9057],
        [  5.8889,  -1.3916,   8.8993],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.5789,   0.1722,   1.6470],
        [ 11.9641,  -2.9568,   7.7211],
        [  4.1637,  13.1898,  -1.9189],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.8323,  -1.5228,   5.1132],
        [  0.6363, -11.0348,   6.1414],
        [  7.0642,  11.6104,  -9.8316],
        [ -8.1531,  -5.8748,   2.6934],
        [ 11.5966,   5.9666,   6.0083],
        [  3.2692,  13.5098,  -4.9032],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.2659,  13.7894,   6.5436],
        [  4.6992,   5.3690,   3.0935],
        [  5.1793,   4.7461,   0.3899],
        [  5.6928,  12.4710,   3.6001],
        [  1.6308,  13.9254,  -2.0984],
        [ -8.1426,   3.8048,   9.3542],
        [  8.5198,  10.4091,  -8.6486],
        [  6.2555,   2.1474,   4.1685],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -2.0840,   7.8415,  -7.8657],
        [  0.6363, -11.0348,   6.1414],
        [ -2.3317,   7.8034,  -3.6274],
        [ -8.9788,   2.0208,  -3.2867],
        [  0.9392,  -4.8143,  -3.4649],
        [  9.3560,   9.5391,   9.4943],
        [  9.1409,  -7.5789,   1.5814],
        [  5.2608,  12.6977,   4.9166],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.0890,  -0.8382,   2.9166],
        [ -7.4602,   4.7718,  -4.6277],
        [  7.1697,  11.5343,  -0.7276],
        [  6.3211,   1.7017,  -0.2164],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.5662,  13.9375,   4.5142],
        [  2.4639, -10.9431,  -7.3718],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -4.0409,   7.3169,  -3.5779],
        [  5.8323,  -1.5228,   5.1132],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.5325,  -3.3643,   6.7699],
        [  3.4009,  -4.1874,  -5.9751],
        [ -0.9715, -10.8630,   0.3555],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.1989,  -9.7591,   0.8289],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.7802,  -1.6363,  -6.2782],
        [  3.2701,   6.6338,   9.9209],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.8930,  12.3592,  -8.9201],
        [  3.3839,  -4.1966,   8.3574],
        [  5.5294,  -2.1106,   9.0626],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.5663,   7.0526,  -6.8769],
        [ 12.2921,  -1.7902,  -4.0870],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.5294,  -2.1106,   9.0626],
        [  5.4336,  -2.2676,   5.6933],
        [  2.2437,  -4.6489,   6.1430]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0010, 0.0143,  ..., 0.9599, 0.8301, 0.8557],
        [0.0010, 0.0000, 0.0144,  ..., 0.9606, 0.8308, 0.8563],
        [0.0143, 0.0144, 0.0000,  ..., 0.9490, 0.8190, 0.8440],
        ...,
        [0.9599, 0.9606, 0.9490,  ..., 0.0000, 0.1308, 0.1277],
        [0.8301, 0.8308, 0.8190,  ..., 0.1308, 0.0000, 0.0599],
        [0.8557, 0.8563, 0.8440,  ..., 0.1277, 0.0599, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.30647746,  6.35029274,  6.53290663,  6.38105015,  6.41202986,
        6.81961996,  9.91668182, 10.7167285 ,  7.15990676,  7.14837514,
       11.27160035,  6.39520397,  5.64865173,  9.04255655, 10.99326577,
        4.90505566,  8.0550286 , 10.8992524 , 11.28895489,  7.13501264,
        5.19065439,  6.83083863,  5.2287058 ,  8.5724042 ,  9.09119351,
       10.7167285 ,  6.23262289, 11.21703654, 10.95593729,  8.8353674 ,
       12.17901033, 13.68672855, 10.35093485,  7.27759233, 10.794755  ,
        4.96231532,  5.19065439, 11.42947632,  9.44486609, 10.42667093,
        6.05113217, 12.62711252, 12.58006038, 12.32408516, 13.83139024,
        6.00732193,  6.02779201, 11.05317597, 13.59056613, 10.04916678,
       13.04156075, 13.89974693,  9.51074763, 13.97430191,  7.13501264,
        7.02496223, 13.70893492, 14.02058621,  8.98765644, 13.45124917,
        6.61386436, 12.05395056,  8.1137452 , 11.05317597,  8.14433542,
        9.20340031,  4.90505566, 13.3614598 , 11.87412439, 13.744387  ,
        4.978704  ,  6.1463896 ,  8.85572292, 13.58101794,  6.54615887,
       12.3828957 , 14.02526505, 11.21703654,  9.7712321 ,  8.35856487,
        6.02779201,  7.09887557,  9.23071698,  9.60412735,  5.64464807,
        5.39450305, 10.90638066, 10.09979747, 12.05395056, 11.56146873,
        6.23262289,  6.00732193,  7.39599986,  4.9160962 , 13.6922425 ,
        5.39098669,  5.91854397, 13.25336001,  7.50500064, 12.4217523 ,
       11.85743425, 12.4217523 , 10.48840476,  5.91854397,  5.88782913,
        5.16205087])}
{'P': tensor([[2.0653e-02, 1.5005e-01, 1.4808e-01,  ..., 8.3368e-04, 7.7193e-06,
         1.4492e-04],
        [9.7792e-04, 2.5430e-02, 7.0131e-03,  ..., 1.5801e-03, 8.9530e-06,
         6.5123e-05],
        [9.3485e-04, 8.0636e-02, 2.3585e-02,  ..., 1.4256e-03, 8.2603e-06,
         6.1011e-05],
        ...,
        [5.4048e-06, 9.7206e-06, 9.5100e-06,  ..., 1.6022e-02, 1.1243e-06,
         2.2384e-07],
        [4.4030e-04, 2.1959e-04, 2.2116e-04,  ..., 2.0466e-06, 8.2343e-02,
         3.6680e-04],
        [1.1206e-03, 1.4953e-03, 1.4530e-03,  ..., 9.7375e-04, 4.1931e-07,
         1.6697e-02]]), 'X': tensor([[  6.3124,   0.2057,  -0.4593],
        [  6.3211,   1.7017,  -0.2164],
        [  6.3246,   1.6696,  -0.1544],
        [  6.2036,   2.4030,  -1.2500],
        [  6.0305,   3.0296,   0.1375],
        [  5.9026,   3.3837,  -0.6573],
        [ 12.5789,   0.1722,   1.6470],
        [  1.4306,  13.9620,  -0.2075],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.0054,  -1.0902,   8.3079],
        [  4.6231,   5.4556,   8.4973],
        [  6.3606,   0.9204,   3.2525],
        [  3.7121,   6.3117,   0.6738],
        [ -1.6906,   7.8860,   7.7142],
        [  6.2675,  12.1379,  -8.9720],
        [  2.6614,  -4.5192,  -9.3641],
        [  3.9895,   6.0816,  -6.5235],
        [  6.2036,   2.4030,  -1.2500],
        [ -7.4733,   4.7555,   6.2490],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 10.9394,   7.3069,  -8.9365],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.1892, -10.8220,  -6.8409],
        [  8.0877,  -8.5297,  -1.1050],
        [  2.5382,   7.0672,   4.5097],
        [  6.1929,   2.4501,  -6.9352],
        [  0.2571,  -4.7594,   1.5618],
        [ -9.3411,  -2.6004,  -2.7062],
        [  3.1756, -10.8247,   6.8882],
        [  6.0305,   3.0296,   0.1375],
        [  1.5383, -11.0274,  -4.9666],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.3367,  -4.4806,   0.4643],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.3411,  -2.6004,  -2.7062],
        [  8.1217,  -8.5024,  -0.6149],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.5662,  13.9375,   4.5142],
        [  6.1755,   2.5233,  -2.0042],
        [ 11.7796,  -3.4696,   9.0088],
        [ -7.4887,   4.7364,   8.9074],
        [  6.2804,   2.0019,   0.8607],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.7295,  13.6703,  -8.9328],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.2096,   5.8815,   1.9230],
        [ -9.2508,   1.0572,   2.9978],
        [  1.3339,   7.5627,  -0.9034],
        [  4.6357,   5.4415,   7.9067],
        [  6.3551,   0.7149,   1.3344],
        [ -9.3411,  -2.6004,  -2.7062],
        [  2.6310, -10.9195,   3.8054],
        [ -8.4267,  -5.3697,   2.4172],
        [ -7.0825,   5.2073,  -5.0497],
        [  8.2154,  -8.4262,   8.0171],
        [  0.9809,  -4.8145,   6.8706],
        [  5.1861,   4.7361,  -3.7312],
        [  5.6314,  -1.9302,  -9.6543],
        [ -8.1531,  -5.8748,   2.6934],
        [  3.3474,  -4.2162,   5.2455],
        [ -6.9838,  -7.4965,   0.9796],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.0588,  -6.0341,  -2.3006],
        [  1.2273,  -4.8081,   9.8129],
        [  4.8653,   5.1694,  -4.0823],
        [ -6.2200,  -8.2737,   5.5079],
        [ -6.6998,   5.5954,   0.0823],
        [ -7.3470,  -7.0615,   5.0066],
        [ 12.2435,   4.0076,   8.1257],
        [  6.3119,   1.7797,   9.9186],
        [  6.3534,   1.2990,  -7.2942],
        [ -5.0550,   6.8231,  -9.2502],
        [ 12.5717,   1.9266,   7.4903],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.3583,   1.1729,   8.7835],
        [  0.8572,  14.0501,  -8.3364],
        [ 11.5456,   6.0861,   8.1101],
        [ -4.4520,   7.1370,  -8.7867],
        [  3.3167,   6.6022,   6.3554],
        [  2.4697,  -4.5836,   1.4593],
        [ -3.4100,   7.5436,  -7.6980],
        [  2.3373,  13.7721,  -0.2442],
        [  6.3236,   0.3011,  -4.6668],
        [  2.0870,  -4.6877,   9.6823],
        [  0.6396,  14.0772,  -6.3786],
        [  4.4915,   5.5990,   7.0405],
        [  2.9260,  -4.4162,   8.3808],
        [ -3.1671,   7.6158,   6.0765],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.9461,  -4.4077,  -0.4831],
        [ -2.2508, -10.5479,  -2.7235],
        [  8.6420,  -8.0580,   6.5781],
        [ -0.5385,   7.9037,   6.3830],
        [  4.0457,  -3.7737,  -0.4704],
        [  2.0729,   7.2878,   7.0880],
        [  2.1824,  -4.6647,   1.1985],
        [ -9.4430,  -0.1897,   4.8383],
        [ -5.9940,   6.1985,   5.9566],
        [  6.1685,  12.1979,  -6.2473],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.7726,  -3.4876,  -6.7402]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0277, 0.0321,  ..., 0.5122, 1.0120, 0.4924],
        [0.0277, 0.0000, 0.0044,  ..., 0.5363, 1.0209, 0.5201],
        [0.0321, 0.0044, 0.0000,  ..., 0.5401, 1.0227, 0.5245],
        ...,
        [0.5122, 0.5363, 0.5401,  ..., 0.0000, 1.2262, 0.2405],
        [1.0120, 1.0209, 1.0227,  ..., 1.2262, 0.0000, 0.9999],
        [0.4924, 0.5201, 0.5245,  ..., 0.2405, 0.9999, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.31576264,  6.54615887,  6.54127431,  6.652742  ,  6.74872981,
        6.80370254, 12.58006038, 14.03505624, 12.50996828,  6.10360073,
        7.15100303,  6.42689705,  7.32233733,  8.06517143, 13.66051884,
        5.24461068,  7.27340834,  6.652742  ,  8.8580671 , 12.30858677,
       13.15525803,  5.34250225, 11.28215429, 11.75438006,  7.50917551,
        6.65991877,  4.7663526 ,  9.69629   , 11.28093507,  6.74872981,
       11.1341798 , 12.10942412, 12.18998195, 11.83509372,  9.69629   ,
       11.75807252,  9.14044984,  5.35774573, 12.32545637, 10.06850677,
       14.02526505,  6.67108662, 12.27993098,  8.86082124,  6.59176018,
        5.5748624 ,  9.55593374, 13.94008948, 12.05300943,  7.23276582,
        9.3109884 ,  7.67939546,  7.14837514,  6.39520397,  9.69629   ,
       11.23201619,  9.99211663,  8.7907923 , 11.76830981,  4.91336994,
        7.02326515,  5.95297459, 10.04916678,  5.38342077, 10.24557047,
       12.07818871, 13.08414689, 10.0674887 ,  4.96231532,  7.09887557,
       10.35093485,  8.72896381, 10.19036081, 12.88270018,  6.55800267,
        6.48486171,  8.49161154, 12.71843462,  8.91551231,  6.46560181,
       14.07623266, 13.05148826,  8.41171369,  7.38844703,  5.20659892,
        8.27853076, 13.96905756,  6.33075682,  5.13125513, 14.09176259,
        7.17789666,  5.29755487,  8.24808832, 12.16901615,  5.30158497,
       10.78533838, 11.8159254 ,  7.92200342,  5.53255882,  7.57685908,
        5.14999799,  9.44486609,  8.62260327, 13.66896739, 10.31613269,
       12.27835996])}
{'P': tensor([[2.3266e-02, 1.8563e-01, 1.7631e-01,  ..., 3.7457e-04, 4.1841e-05,
         3.8282e-04],
        [1.9181e-03, 2.3607e-02, 2.4656e-01,  ..., 3.2425e-04, 5.0675e-05,
         3.0094e-04],
        [1.4248e-03, 1.5657e-03, 2.9921e-02,  ..., 2.7897e-04, 8.0897e-05,
         2.8085e-04],
        ...,
        [4.7631e-04, 6.6782e-04, 1.0032e-03,  ..., 2.7518e-02, 1.1804e-05,
         1.1897e-05],
        [1.4374e-05, 1.3973e-05, 1.2894e-05,  ..., 1.6128e-06, 3.1350e-02,
         3.9188e-06],
        [1.3095e-02, 1.2197e-02, 1.2202e-02,  ..., 3.0208e-04, 2.4691e-05,
         1.6349e-01]]), 'X': tensor([[  6.3200,   0.2690,   1.5900],
        [  6.3551,   0.7149,   1.3344],
        [  6.3460,   1.4252,   1.2291],
        [  6.3305,   1.6119,   1.1538],
        [  6.2905,   1.9366,   2.1169],
        [  6.2804,   2.0019,   0.8607],
        [  5.1667,  -2.6543,   1.2879],
        [ -3.9508,  -9.8515,   1.7923],
        [  1.1644, -11.0394,   7.8511],
        [ 11.3691,   6.4760,   1.0787],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.0772,  -2.6014,  -7.1832],
        [  9.6587,   9.1817,  -0.8212],
        [ -1.6086,   7.8928,   4.3385],
        [  6.2576,   2.1359,  -3.3398],
        [ -3.1671,   7.6158,   6.0765],
        [ -8.7409,  -4.6837,  -6.5840],
        [  7.3592,  11.3937,   6.2407],
        [ -3.5173,   7.5091,  -8.9585],
        [  1.4489,   7.5256,   4.4529],
        [  1.2728, -11.0372,   9.8121],
        [  6.2535,  -0.1709,  -9.4102],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.3118,   4.5452,   4.2114],
        [  6.2630,   2.1057,  -5.2136],
        [ -2.5056,   7.7720,   5.5878],
        [ -8.8363,   2.4086,  -5.6296],
        [ 12.3025,  -1.7440,  -6.6632],
        [  2.4932,   7.0903,  -5.4214],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.3705,  13.9723,   7.9266],
        [  6.2883,   0.0325,   5.2692],
        [  2.1504,  -4.6726,   8.7908],
        [ -9.0961,   1.6529,   3.1553],
        [ 11.4235,   6.3594,   5.1663],
        [  2.2659,  13.7894,   6.5436],
        [ 11.7162,   5.6728,  -8.9372],
        [ -8.9942,  -3.9990,   6.0448],
        [  1.8766,   7.3694,   1.1887],
        [ -7.4808,   4.7462,   2.0266],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.9074,  -4.8139,   8.1674],
        [  6.3119,   1.7797,   9.9186],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.9461,  -4.4077,  -0.4831],
        [  9.0424,  -7.6779,  -7.5286],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.3860,  -2.3418,   1.0826],
        [  0.7958,  -4.8109,  -8.2248],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.4495,   6.3026,   6.8257],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -5.3328,  -8.9992,   7.3440],
        [ 11.2209,  -4.7092,   2.6117],
        [  9.2979,   9.6048,   7.3138],
        [  3.8111, -10.6787,  -4.6539],
        [ -9.4716,  -0.6673,   7.6068],
        [ -6.6895,   5.6051,   2.7899],
        [  6.3360,   0.4258,  -9.2509],
        [  4.4633,   5.6287,  -0.7754],
        [ 12.5224,  -0.4497,   4.8448],
        [ -5.3179,  -9.0101,   9.4472],
        [ 11.5456,   6.0861,   8.1101],
        [ -7.6650,   4.5092,   0.2520],
        [  2.7295,  13.6703,  -8.9328],
        [  4.3377,   5.7570,   1.3712],
        [ -9.4351,  -1.8891,   3.5642],
        [ -3.4986,   7.5152,   9.0712],
        [ 11.7796,  -3.4696,   9.0088],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.0233,   4.9647,  -3.1788],
        [  6.1984,  -0.4327,   1.5438],
        [  0.5250, -11.0306,   5.5358],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -4.7334,   6.9982,  -0.4392],
        [  5.6314,  -1.9302,  -9.6543],
        [  1.3767, -11.0341,   9.9415],
        [  5.1292,   4.8183,  -5.7997],
        [  5.8318,   3.5567,  -4.7692],
        [ 11.7418,   5.6070,  -1.4192],
        [  2.5620,   7.0549,   8.4825],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.3705,   0.4208,  -3.5815],
        [  4.5491,  -3.3485,   6.8412],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.9503,  -4.7172,  -6.2313],
        [  9.0129,  -7.7072,   9.1400],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.6523,   3.9443,  -2.9449],
        [ -8.8058,   2.4850,  -1.7425],
        [  4.5340,  -3.3629,   8.9652],
        [ -7.2986,  -7.1225,  -3.0814],
        [  8.2154,  -8.4262,   8.0171],
        [ -4.8751,  -9.3148,   4.0406],
        [  3.7237,  -3.9966,  -5.0376],
        [  5.4426, -10.1181,  -5.1268],
        [  2.2659,  13.7894,   6.5436],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.0617,  12.7957,   8.3544],
        [  9.0652,   9.8597,  -3.5949],
        [  0.5175,  14.0910,   4.6540],
        [  4.3377,   5.7570,   1.3712],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.5919,   0.4035,  -5.8460],
        [ -8.0608,   3.9362,  -5.9567],
        [  5.5294,  -2.1106,   9.0626]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2811e-02, 1.7231e-02,  ..., 5.9447e-01, 9.1185e-01,
         4.1540e-01],
        [1.2811e-02, 1.0727e-04, 4.4699e-03,  ..., 5.8189e-01, 9.0492e-01,
         4.2819e-01],
        [1.7231e-02, 4.4699e-03, 0.0000e+00,  ..., 5.7771e-01, 9.0192e-01,
         4.3263e-01],
        ...,
        [5.9447e-01, 5.8189e-01, 5.7771e-01,  ..., 0.0000e+00, 8.7855e-01,
         1.0030e+00],
        [9.1185e-01, 9.0492e-01, 9.0192e-01,  ..., 8.7855e-01, 0.0000e+00,
         1.2037e+00],
        [4.1540e-01, 4.2819e-01, 4.3263e-01,  ..., 1.0030e+00, 1.2037e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.32572323,  6.39520397,  6.50409372,  6.53250937,  6.58184474,
        6.59176018,  5.80860165, 10.61417181, 11.10066126, 13.08414689,
       12.84056604, 12.35421461, 13.32645552,  8.0550286 ,  6.61211224,
        8.24808832,  9.91668182, 13.56367857,  8.29203356,  7.66377554,
       11.11038551,  6.25586297, 10.74409834,  6.99097013,  6.60752373,
        8.16584537,  9.15866607, 12.42554808,  7.5158489 ,  5.9510489 ,
       14.0393949 ,  6.28835101,  5.14371005,  9.24501829, 13.0743399 ,
       13.97430191, 13.01728313,  9.84314712,  7.60463349,  8.859416  ,
        9.4997987 ,  4.89869205,  6.55800267, 12.41366754,  5.30158497,
       11.86239961,  8.27853076,  5.87305005,  4.8763135 , 11.27160035,
       13.06957425, 10.33712258, 10.4606265 , 12.16901615, 13.36800073,
       11.33837396,  9.49511524,  8.7273569 ,  6.35029274,  7.18354885,
       12.53047489, 10.46238054, 13.05148826,  8.89301516, 13.94008948,
        7.20826219,  9.62238828,  8.28968067, 12.27993098, 12.84056604,
        7.06272256,  6.2134961 , 11.04313641, 12.42554808,  8.4486771 ,
        5.95297459, 11.11969815,  7.03733775,  6.83083863, 13.01187484,
        7.50564223,  5.0209005 ,  9.37989631,  5.64865173, 10.1979669 ,
        5.10444097, 11.85890664, 10.36844813,  6.89245352,  9.14973319,
        5.64500512, 10.1979669 , 11.76830981, 10.51340829,  5.46244988,
       11.4890861 , 13.97430191,  5.18047638, 13.76047999, 13.39372426,
       14.10045797,  7.20826219,  5.91677188, 12.59840193,  8.97052115,
        5.91854397])}
{'P': tensor([[2.7240e-02, 1.5754e-01, 1.3867e-01,  ..., 1.4582e-02, 8.7992e-06,
         4.2075e-04],
        [1.6862e-03, 4.7694e-02, 1.2064e-02,  ..., 3.6008e-02, 6.5507e-06,
         4.0797e-04],
        [1.3460e-03, 3.2305e-02, 2.6774e-02,  ..., 3.2429e-02, 5.9240e-06,
         2.6008e-04],
        ...,
        [3.9736e-04, 7.1934e-04, 6.2172e-04,  ..., 6.8199e-02, 5.1812e-06,
         1.8739e-04],
        [4.2838e-04, 2.8257e-04, 5.2138e-04,  ..., 2.5183e-04, 6.4446e-02,
         2.3694e-04],
        [7.4696e-02, 3.2591e-02, 4.1090e-02,  ..., 4.1191e-03, 1.3610e-04,
         5.1452e-02]]), 'X': tensor([[  6.3236,   0.3011,  -4.6668],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3205,   1.7068,  -3.8619],
        [  6.0955,   2.8210,  -4.3321],
        [  5.9526,   3.2525,  -4.3821],
        [  5.9912,   3.1453,  -3.9989],
        [  4.1348,   5.9514,  -0.8856],
        [  5.8902,  -1.3887,   4.0388],
        [ -9.2345,  -3.1345,   3.9715],
        [  2.3200,   7.1756,   1.1938],
        [  2.4356,  -4.5942,  -5.0808],
        [  3.7916,  -3.9524,  -7.9547],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -7.4808,   4.7462,   2.0266],
        [  0.9659,  -4.8144,  -0.9594],
        [  4.2096,   5.8815,   1.9230],
        [ -4.4779,  -9.5609,  -2.4505],
        [  6.1575,  12.2046,   3.3257],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.8363,   2.4086,  -5.6296],
        [ -7.8359,   4.2735,   7.2943],
        [  5.9587,  12.3216,   9.1695],
        [ -1.0012, -10.8576,   2.2867],
        [  5.6523,   3.9443,  -2.9449],
        [  7.8375,  -8.7240,  -9.8913],
        [ -9.4283,  -1.9585,  -0.9129],
        [  2.0419,  -4.6978,  -5.0983],
        [ 10.0567,   8.6697,   4.6329],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -8.7078,   2.7182,   1.8146],
        [ 11.3367,  -4.4806,   0.4643],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 12.4996,   2.6335,  -6.3166],
        [  9.3560,   9.5391,   9.4943],
        [  6.3583,   1.1729,   8.7835],
        [  5.6258,  -1.9404,  -6.0715],
        [ 11.1469,   6.9237,   1.6160],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.7795,  -3.4697,   3.5624],
        [  5.2939,  -2.4784,  -4.5840],
        [  7.1821,  -9.1845,   9.8548],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -1.9600,   7.8577,   6.9121],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -9.1231,  -3.5739,   8.4708],
        [  8.0877,  -8.5297,  -1.1050],
        [  0.7685,  -4.8098,  -2.9310],
        [  8.6420,  -8.0580,   6.5781],
        [  2.3200,   7.1756,   1.1938],
        [ -1.5408,   7.8978,   5.3470],
        [  9.6587,   9.1817,  -0.8212],
        [ -1.9487,   7.8590,  -3.3330],
        [  1.3670,  -4.7990,   3.2603],
        [  7.8982,  -8.6778,   0.0239],
        [  2.5469,  13.7192,  -9.6935],
        [ -3.9456,  -9.8542,   0.7466],
        [  1.8719,   7.3713,  -3.7465],
        [  5.9883,   3.1536,   6.2320],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.0891,  -4.9558,  -2.3178],
        [  6.3048,   0.1469,  -9.1691],
        [  3.2366,  -4.2732,   5.1532],
        [  8.4691,  10.4571,  -5.0650],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.9916,   7.2133,  -0.9179],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -0.5709,   7.9055,   7.6089],
        [  0.9809,   7.6642,  -1.4217],
        [ -7.4887,   4.7364,   8.9074],
        [  9.6662,   9.1725,   8.1056],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.2365,  -0.2578,  -6.0212],
        [ -9.3331,  -2.6465,  -8.2028],
        [  4.0642,  -3.7599,  -3.4565],
        [ -8.9178,   2.1936,   7.5388],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -4.0409,   7.3169,  -3.5779],
        [  0.5850,  -4.7982,   9.6639],
        [  1.6308,  13.9254,  -2.0984],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.6050,   2.9451,  -0.7012],
        [ 12.5258,  -0.4200,  -9.3493],
        [  4.5340,  -3.3629,   8.9652],
        [  6.2630,   2.1057,  -5.2136],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.1196,  -2.7160,   4.2705],
        [  4.4633,   5.6287,  -0.7754],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.2159, -10.8166,  -1.0406],
        [  1.3767, -11.0341,   9.9415],
        [  6.0073,   3.0986,   6.7937],
        [  4.0591,   6.0202,  -1.6339],
        [  8.2154,  -8.4262,   8.0171],
        [  4.1999,  -3.6543,  -4.1998]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0263, 0.0304,  ..., 0.1877, 0.9785, 0.0468],
        [0.0263, 0.0000, 0.0562,  ..., 0.2097, 1.0019, 0.0584],
        [0.0304, 0.0562, 0.0000,  ..., 0.1596, 0.9488, 0.0425],
        ...,
        [0.1877, 0.2097, 0.1596,  ..., 0.0000, 0.7932, 0.1561],
        [0.9785, 1.0019, 0.9488,  ..., 0.7932, 0.0000, 0.9492],
        [0.0468, 0.0584, 0.0425,  ..., 0.1561, 0.9492, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.33075682,  6.60752373,  6.54693987,  6.7166297 ,  6.78325151,
        6.76661595,  7.24677664,  6.05165248,  9.75200848,  7.54127425,
        5.19986797,  5.47702766, 10.09979747,  8.859416  ,  4.91038884,
        7.23276582, 10.5575601 , 13.66991027, 10.27716434, 12.22503754,
        6.1561646 ,  9.15866607,  8.92549752, 13.68672855, 10.9036226 ,
        6.89245352, 11.72749923,  9.62958973,  5.1224079 , 13.2778378 ,
       10.32715715,  9.12220271, 12.18998195, 10.52357479, 12.11735334,
       12.77402351, 13.3614598 ,  6.46560181,  5.9510489 , 13.12218013,
       12.18045978,  6.01356004, 12.27991991,  5.84533345, 11.65923396,
        9.51074763,  8.09842704,  9.69629   ,  9.88571713,  9.79814997,
       11.75438006,  4.87083348, 11.8159254 ,  7.54127425,  8.04664886,
       13.32645552,  8.09704052,  4.98989906, 11.73397309, 13.95361098,
       10.61472636,  7.60529892,  6.76791483, 11.99633819, 12.91143491,
       12.1460931 ,  6.30647746,  5.36061974, 13.45642379, 13.32374056,
       13.1471313 , 10.53368644,  9.62958973,  7.92607342,  7.72668467,
        8.86082124, 13.32557133, 13.13205357,  9.88571713,  9.85384623,
        4.92497514,  6.24187044,  9.70108409,  5.5366698 ,  9.18358562,
       13.03002338,  9.60359158,  8.35856487,  4.83371341, 14.02058621,
       10.33958639,  9.09502385, 12.53285633,  5.64500512,  6.60752373,
        9.88571713,  9.26455795,  5.79543038,  7.18354885,  4.76969364,
       11.2845608 , 11.11969815,  6.7593917 ,  7.26074925, 11.76830981,
        5.5671389 ])}
{'P': tensor([[4.5463e-02, 2.8168e-01, 2.6856e-01,  ..., 2.1809e-03, 7.8684e-05,
         2.1111e-04],
        [3.5330e-03, 8.1189e-02, 1.3158e-02,  ..., 6.0222e-03, 7.9832e-05,
         2.4413e-04],
        [4.2773e-03, 1.2411e-01, 5.9444e-02,  ..., 4.5736e-03, 1.0266e-04,
         3.2184e-04],
        ...,
        [3.8753e-06, 6.4691e-06, 5.4951e-06,  ..., 1.2193e-02, 1.0796e-06,
         5.8371e-06],
        [4.2330e-04, 3.5571e-04, 5.7891e-04,  ..., 4.0843e-04, 9.4883e-02,
         1.2413e-01],
        [1.4254e-04, 1.5975e-04, 2.1808e-04,  ..., 1.8599e-03, 8.6351e-04,
         8.0025e-02]]), 'X': tensor([[  6.3266,   0.3292,   9.3502],
        [  6.3119,   1.7797,   9.9186],
        [  6.3583,   1.1729,   8.7835],
        [  5.6248,   3.9984,   8.4933],
        [  6.0073,   3.0986,   6.7937],
        [  4.6231,   5.4556,   8.4973],
        [ -8.3900,   3.3727,   3.1650],
        [ -6.2668,   5.9808,   2.5614],
        [ 11.0864,   7.0387,  -0.1238],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -1.6275, -10.7221,   3.2376],
        [  0.2480, -11.0153,  -3.0183],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.7000,  -4.7609,   2.7935],
        [  8.0877,  -8.5297,  -1.1050],
        [  0.1206,  -4.7363,   8.7103],
        [  1.8721,  -4.7322,   0.5524],
        [ -3.0008, -10.2827,  -8.9480],
        [ -5.1974,   6.7396,   3.3484],
        [  6.3460,   1.4252,   1.2291],
        [  7.1441,  11.5529,   2.3232],
        [  6.2770,  -0.0379,   2.4195],
        [  7.8425,  -8.7202,   6.4530],
        [  9.5261,  -7.1675,   1.4206],
        [ -1.6275, -10.7221,   3.2376],
        [ -3.4488, -10.0934,   2.5195],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.6321,  -4.5296,  -2.5159],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.4309,  -0.0599,  -4.0830],
        [  0.4119,  14.1019,  -4.1870],
        [  3.3839,  -4.1966,   8.3574],
        [  0.6933,   7.7336,   2.1179],
        [ 11.2517,  -4.6495,   2.9800],
        [  2.5010,  -4.5736,  -0.8989],
        [ 12.6038,   0.7665,  -3.4439],
        [ 11.6019,   5.9540,  -5.1715],
        [  5.6258,  -1.9404,  -6.0715],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.6093,  -1.9705,   6.4807],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.6035,   0.7494,   7.1536],
        [  6.3587,   1.1596,  -3.6629],
        [  2.0547,   7.2957,  -1.9343],
        [  6.2026,  12.1774,  -1.5887],
        [  0.5960, -11.0335,   3.1144],
        [  5.0348,  -2.8229,   3.5436],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.9470,  13.6085,   2.9809],
        [ -0.9715, -10.8630,   0.3555],
        [  3.4140,  -4.1803,  -8.8224],
        [ -7.5603,   4.6459,   8.2002],
        [  6.1984,  -0.4327,   1.5438],
        [  5.2911,  -2.4824,  -3.6877],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 11.6634,   5.8050,   9.1590],
        [  1.0000,   7.6591,  -4.5011],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.0890,  -0.8382,   2.9166],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.6231,   5.4556,   8.4973],
        [  8.5198,  10.4091,  -8.6486],
        [  5.1667,  -2.6543,   1.2879],
        [  5.1203,  -2.7151,   6.2110],
        [ -8.3900,   3.3727,   3.1650],
        [  1.7903,  -4.7467,   0.1168],
        [  2.5663,   7.0526,  -6.8769],
        [ 11.2939,   6.6325,  -5.8311],
        [  0.2435,   7.8191,  -6.4709],
        [  0.9659,  -4.8144,  -0.9594],
        [  4.7739,  -3.1218,  -4.1188],
        [  7.5309,  11.2617,   7.2656],
        [ 12.5258,  -0.4200,  -9.3493],
        [  9.4516,  -7.2501,  -1.8868],
        [  9.5018,  -7.1945,  -1.1892],
        [ -9.2974,   0.8371,   2.4852],
        [  1.5653,  13.9377,   4.1773],
        [  0.1287, -11.0065,   6.5757],
        [ -5.1046,  -9.1610,   2.3418],
        [ 12.1379,   4.4125,  -3.8833],
        [  7.5309,  11.2617,   7.2656],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.3523, -10.5232,   8.2138],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.4196,   0.0478,   9.4140],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.0876,   6.7527,  -5.3540],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.6112,   2.9319,  -7.9943],
        [  4.6357,   5.4415,   7.9067],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.3535,   0.6792,  -7.5101],
        [ -9.3464,   0.5708,  -1.6397],
        [ 11.6544,   5.8273,  -3.1375],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.0233,   4.9647,  -3.1788],
        [  7.8982,  -8.6778,   0.0239],
        [  5.2608,  12.6977,   4.9166],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.0955,   2.8210,  -4.3321]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0200, 0.0134,  ..., 0.3045, 0.8206, 0.8184],
        [0.0200, 0.0000, 0.0201,  ..., 0.3220, 0.8260, 0.8239],
        [0.0134, 0.0201, 0.0000,  ..., 0.3147, 0.8095, 0.8074],
        ...,
        [0.3045, 0.3220, 0.3147,  ..., 0.0000, 0.9461, 0.9422],
        [0.8206, 0.8260, 0.8095,  ..., 0.9461, 0.0000, 0.0063],
        [0.8184, 0.8239, 0.8074,  ..., 0.9422, 0.0063, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.33517349,  6.55800267,  6.46560181,  6.90115901,  6.7593917 ,
        7.15100303,  9.04255655,  8.66273171, 13.13205357, 10.31613269,
       10.84493732, 11.018084  , 12.12641969, 12.59840193,  5.05534609,
       11.75438006,  4.73784222,  5.08910491, 10.71162681,  8.51088913,
        6.50409372, 13.58333857,  6.27715105, 11.72802491, 11.92133129,
       10.84493732, 10.66631784,  5.46244988,  5.23878371, 12.64235359,
        9.43112957, 14.10796466,  5.39098669,  7.76457095, 12.17451685,
        5.21280156, 12.62711252, 13.04051744,  5.9510489 , 10.31584102,
        5.94535529,  7.6374603 , 12.62576026,  6.46356806,  7.57946459,
       13.6660671 , 11.04953767,  5.77219077, 13.00413829, 13.92390689,
       10.90638066,  5.39723241,  8.8737556 ,  6.2134961 ,  5.84450127,
        9.75075273, 13.02818702,  7.72415048,  8.20436764,  6.1463896 ,
       13.86480175, 10.74409834,  7.15100303, 13.45124917,  5.80860165,
        5.79562691,  9.04255655,  5.07306274,  7.50500064, 13.09737053,
        7.822854  ,  4.91038884,  5.70405525, 13.54774355, 12.53285633,
       11.9120321 , 11.9182956 ,  9.3349858 , 14.02532758, 11.00726483,
       10.48721382, 12.91505111, 13.54774355,  5.41348121, 11.38774149,
        5.21280156,  9.41970031,  5.05963217,  5.19065439,  7.42512621,
        8.49161154,  9.09661212,  7.14837514,  5.70405525,  9.49511524,
        6.38968357,  9.36378438, 13.03002338, 10.67322527,  9.562174  ,
        9.69629   ,  7.06272256, 11.73397309, 13.744387  ,  5.90035324,
        6.7166297 ])}
{'P': tensor([[1.8396e-02, 2.1645e-01, 1.5649e-01,  ..., 1.3822e-06, 5.0896e-05,
         2.8727e-05],
        [1.4089e-03, 2.1728e-02, 2.0573e-01,  ..., 1.6238e-06, 6.7264e-05,
         3.2333e-05],
        [1.6021e-03, 1.8818e-03, 3.0951e-02,  ..., 2.3732e-06, 1.2380e-04,
         4.3342e-05],
        ...,
        [1.6013e-05, 1.4620e-05, 1.2606e-05,  ..., 2.6099e-02, 4.0566e-06,
         5.1031e-03],
        [5.1782e-05, 5.2413e-05, 5.5519e-05,  ..., 2.3856e-03, 1.9536e-01,
         1.0575e-03],
        [1.6564e-03, 1.6123e-03, 1.5778e-03,  ..., 5.0993e-04, 3.7227e-04,
         1.4341e-01]]), 'X': tensor([[  6.3301,   0.3638,  -7.5685],
        [  6.3535,   0.6792,  -7.5101],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3031,   1.8477,  -7.1706],
        [  6.1929,   2.4501,  -6.9352],
        [  6.3303,   1.6145,  -8.4083],
        [  5.5158,  -2.1337,   3.5246],
        [ 11.9808,   4.9340,  -3.3715],
        [  0.7958,  -4.8109,  -8.2248],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.3205,   1.7068,  -3.8619],
        [  5.3924, -10.1396,  -4.0863],
        [ -7.0120,   5.2825,   7.7762],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.9788,   2.0208,  -3.2867],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.8602,   6.8905,   5.8779],
        [  3.5965,  -4.0754,  -3.5035],
        [  3.5965,  -4.0754,  -3.5035],
        [  4.5800,   5.5034,  -3.9432],
        [  4.5340,  -3.3629,   8.9652],
        [  2.3373,  13.7721,  -0.2442],
        [  6.0890,  -0.8382,   2.9166],
        [ -4.3359,  -9.6431,  -9.0616],
        [  1.9505,  -4.7171,   9.9868],
        [ -9.1706,   1.3863,   8.4032],
        [  4.1537,   5.9340,  -6.0423],
        [  5.4732,  -2.2041,  -4.3830],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.3246,   1.6696,  -0.1544],
        [  0.1382, -11.0073,   1.0026],
        [  5.0190,   4.9705,  -7.9752],
        [  3.8182,  -3.9347,  -1.7961],
        [ -9.1960,  -3.2964,   1.6921],
        [  3.4401,  13.4540,   4.6576],
        [  9.6614,   9.1784,  -1.2087],
        [ -5.1398,  -9.1367,   6.2074],
        [ 11.0864,   7.0387,  -0.1238],
        [ -1.9714,   7.8563,  -1.8084],
        [  0.5175,  14.0910,   4.6540],
        [  6.0955,   2.8210,  -4.3321],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.1795,  -3.3623,   8.5259],
        [ 10.7566,   7.6212,  -2.3279],
        [ -8.5506,  -5.1154,   6.3423],
        [ -2.0445,   7.8469,  -0.5857],
        [ -9.4773,  -1.0024,   5.2599],
        [  0.9445,   7.6736,   0.5095],
        [  3.1480,  -4.3165,  -7.8002],
        [  0.3038,   7.8092,   0.4164],
        [  6.1737,   2.5305,   3.9049],
        [  5.1787,  -2.6383,  -4.2533],
        [  6.3550,   0.7129,   0.4917],
        [  9.9756,  -6.6330,  -3.3429],
        [  2.2659,  13.7894,   6.5436],
        [ -3.9456,  -9.8542,   0.7466],
        [  6.1973,  -0.4372,   6.5720],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.2242,  -0.3171,   7.4978],
        [  1.1994,   7.6035,   2.4295],
        [ -5.6792,  -8.7352,   4.0318],
        [  5.0233,   4.9647,  -3.1788],
        [  2.6310, -10.9195,   3.8054],
        [ 11.5456,   6.0861,   8.1101],
        [ 11.2571,   6.7072,   8.3693],
        [  4.3377,   5.7570,   1.3712],
        [ -5.1377,   6.7751,  -0.4438],
        [ -0.4341, -10.9473,   7.4509],
        [  0.9445,   7.6736,   0.5095],
        [ 11.4235,   6.3594,   5.1663],
        [  2.3376,  -4.6231,   9.8436],
        [ -9.3705,   0.4208,  -3.5815],
        [  7.2340,  -9.1505,  -5.8321],
        [  0.8940, -11.0402,  -0.9783],
        [  7.9816,  10.8938,  -0.5378],
        [  6.2804,   2.0019,   0.8607],
        [ -8.0611,  -6.0302,   4.6962],
        [  1.3670,  -4.7990,   3.2603],
        [ -8.6050,   2.9451,  -0.7012],
        [  1.9204,   7.3518,   8.4781],
        [  6.2535,  -0.1709,  -9.4102],
        [  4.8653,   5.1694,  -4.0823],
        [  0.1460,   7.8340,  -9.4614],
        [  4.7482, -10.3907,   5.9149],
        [  9.1820,   9.7334,  -2.7932],
        [ 11.2939,   6.6325,  -5.8311],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.8407,  11.7666,   3.0045],
        [  5.9548,  -1.2273,  -2.6520],
        [ -6.8466,  -7.6488,   1.9648],
        [  7.8425,  -8.7202,   6.4530],
        [ 12.4901,  -0.7052,   1.7315],
        [  0.9478,  -4.8144,   0.6371],
        [ -2.0721,   7.8432,   6.6380],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.3534,   1.2990,  -7.2942],
        [  3.2990,  -4.2415,  -7.4857],
        [ -6.2200,  -8.2737,   5.5079],
        [ -4.4503,   7.1378,   3.3751],
        [  1.3670,  -4.7990,   3.2603]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.2906e-04, 2.0910e-03,  ..., 1.2273e+00, 8.0753e-01,
         6.1688e-01],
        [4.2906e-04, 0.0000e+00, 1.8702e-03,  ..., 1.2274e+00, 8.0748e-01,
         6.1666e-01],
        [2.0910e-03, 1.8702e-03, 0.0000e+00,  ..., 1.2258e+00, 8.0572e-01,
         6.1480e-01],
        ...,
        [1.2273e+00, 1.2274e+00, 1.2258e+00,  ..., 0.0000e+00, 4.8799e-01,
         8.3076e-01],
        [8.0753e-01, 8.0748e-01, 8.0572e-01,  ..., 4.8799e-01, 0.0000e+00,
         3.4422e-01],
        [6.1688e-01, 6.1666e-01, 6.1480e-01,  ..., 8.3076e-01, 3.4422e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.34059021,  6.38968357,  6.48486171,  6.56833864,  6.65991877,
        6.53290663,  5.91408504, 12.95702647,  4.8763135 ,  8.4486771 ,
        6.54693987, 11.48434767,  8.77913428,  7.28793663,  9.20340031,
       13.03002338,  7.46053281,  5.43544626,  5.43544626,  7.15990676,
        5.64500512, 13.96905756,  6.1463896 , 10.57302546,  5.10446653,
        9.27474386,  7.24326992,  5.90035324, 12.3828957 , 10.1371552 ,
        6.54127431, 11.00812996,  7.06372172,  5.48276038,  9.76896904,
       13.88683436, 13.32613445, 10.48315436, 13.13205357,  8.09984231,
       14.10045797,  6.7166297 ,  9.87720102,  9.77588801, 13.18278606,
        9.96391045,  8.10886021,  9.53015376,  7.73151309,  5.34250225,
        7.81509631,  6.67218687,  5.81199422,  6.3949043 , 11.97957425,
       13.97430191, 10.61472636,  6.21275016, 10.06437598, 10.33712258,
        6.2322769 ,  7.69752262, 10.41909211,  7.06272256, 11.23201619,
       13.05148826, 13.103709  ,  7.20826219,  8.50278896, 10.95593729,
        7.73151309, 13.0743399 ,  5.1805345 ,  9.37989631, 11.66453059,
       11.07637531, 13.50484539,  6.59176018, 10.06703358,  4.98989906,
        9.09502385,  7.5984809 ,  6.25586297,  7.09887557,  7.83534987,
       11.42420508, 13.38091121, 13.09737053,  4.76969364, 13.6105595 ,
        6.07992623, 10.26545792, 11.72802491, 12.50996828,  4.90676242,
        8.11227668,  5.5671389 ,  6.10774544,  9.62958973, 12.05395056,
       12.24467041,  6.48486171,  5.37343173, 10.35093485,  8.41148722,
        4.98989906])}
{'P': tensor([[4.5068e-02, 2.1495e-01, 1.6470e-01,  ..., 7.2134e-06, 6.7181e-02,
         9.8204e-04],
        [2.3200e-03, 3.7703e-02, 1.2792e-01,  ..., 5.3775e-06, 7.5820e-02,
         3.7496e-04],
        [8.8848e-04, 1.8181e-03, 4.5809e-02,  ..., 3.7142e-06, 5.6791e-03,
         1.0010e-04],
        ...,
        [2.5564e-04, 2.1244e-04, 6.6773e-05,  ..., 4.6054e-02, 2.0983e-04,
         1.0859e-03],
        [2.6381e-03, 6.3237e-03, 3.7540e-02,  ..., 1.0006e-05, 6.5293e-02,
         3.1328e-04],
        [1.1484e-03, 2.8109e-03, 2.7526e-03,  ..., 4.1219e-07, 5.8084e-03,
         1.5542e-02]]), 'X': tensor([[ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0144, 0.0491,  ..., 0.8448, 0.0541, 0.0729],
        [0.0144, 0.0000, 0.0461,  ..., 0.8311, 0.0404, 0.0780],
        [0.0491, 0.0461, 0.0000,  ..., 0.8151, 0.0699, 0.1219],
        ...,
        [0.8448, 0.8311, 0.8151,  ..., 0.0000, 0.8002, 0.8872],
        [0.0541, 0.0404, 0.0699,  ..., 0.8002, 0.0000, 0.0909],
        [0.0729, 0.0780, 0.1219,  ..., 0.8872, 0.0909, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.35029274,  6.53290663,  6.81961996,  6.83515201,  6.77777606,
        6.97106912,  7.4544866 , 13.3426541 ,  9.7712321 , 10.66631784,
        9.32094055,  6.67108662,  5.8828986 ,  9.60080639, 10.67322527,
        5.43544626, 13.45642379,  7.50564223,  4.92497514,  7.53063885,
       13.44593519,  9.38232121,  8.49161154, 13.01187484,  6.7593917 ,
        8.24808832,  5.91408504, 12.00995898, 13.66051884,  5.69913817,
       12.07225469,  6.1561646 ,  8.74470218,  5.27178363, 13.66896739,
        7.20482224, 10.86693708,  7.62232307,  8.53501777,  6.5618968 ,
       10.67322527, 13.18278606, 11.27656488, 12.40074901, 13.58333857,
       13.54774355, 13.91198191, 13.46553623,  9.88310614, 10.06023665,
       13.45642379, 13.72594022,  6.54615887, 12.41366754,  9.97822473,
        8.57560275,  4.91296205,  9.77777346,  8.25828409,  5.29755487,
       11.51557166, 12.35421461, 11.22993218, 11.22993218,  5.80312949,
       13.2778378 ,  7.15990676,  5.05534609, 10.33712258,  6.76791483,
        9.66977802, 12.10222204, 10.26545792, 13.76047999, 12.17451685,
       13.40024224,  5.23878371,  9.15866607,  8.36995953, 12.24467041,
        6.2134961 ,  5.1224079 ,  5.53175275,  8.19703512,  8.92395381,
        5.05774295, 12.53047489,  8.29022197, 13.61600536,  5.64500512,
       14.10796466, 12.59840193, 12.91143491, 10.5575601 , 12.76408562,
        8.31625171,  8.77913428, 13.32645552, 13.18278606, 14.02526505,
       11.83570132, 10.67132524, 12.05300943, 11.00812996,  6.69389431,
       12.49317563])}
{'P': tensor([[1.9857e-02, 1.5706e-01, 1.4803e-01,  ..., 3.8630e-03, 1.5270e-03,
         6.6225e-07],
        [2.1320e-03, 2.5390e-02, 1.5812e-01,  ..., 4.1448e-03, 2.0801e-03,
         8.8157e-07],
        [1.1942e-03, 1.8245e-03, 2.8892e-02,  ..., 2.7731e-03, 1.6355e-03,
         8.2329e-07],
        ...,
        [2.4971e-04, 3.5700e-04, 8.6129e-04,  ..., 1.7552e-02, 9.3909e-03,
         4.5438e-08],
        [2.3648e-05, 3.1894e-05, 5.3977e-05,  ..., 4.2901e-05, 8.8923e-03,
         8.8361e-08],
        [3.6772e-06, 3.5173e-06, 2.1740e-06,  ..., 4.5290e-07, 9.1319e-07,
         1.5105e-02]]), 'X': tensor([[  6.3413,   0.4883,  -6.8416],
        [  6.3604,   0.9042,  -6.5901],
        [  6.3031,   1.8477,  -7.1706],
        [  6.3534,   1.2990,  -7.2942],
        [  6.1929,   2.4501,  -6.9352],
        [  6.1372,   2.6726,  -6.7885],
        [ -0.8885,   7.9160,   0.0571],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -6.9838,  -7.4965,   0.9796],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.9526,   3.2525,  -4.3821],
        [  9.5203,   9.3483,   5.8781],
        [ -8.6426,  -4.9136,   9.4941],
        [  6.1372,   2.6726,  -6.7885],
        [  5.3858,   4.4260,  -9.6769],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -0.8885,   7.9160,   0.0571],
        [ -5.9872,   6.2037,  -9.4477],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.1741,  -0.5329,   3.1913],
        [  4.1348,   5.9514,  -0.8856],
        [  3.3839,  -4.1966,   8.3574],
        [ -8.5859,  -5.0393,   0.5476],
        [ -4.8751,  -9.3148,   4.0406],
        [  2.7685, -10.8982,  -6.8928],
        [  5.6095,  -1.9702,  -7.8233],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.9606,  -4.0994,   1.7886],
        [ -9.4296,  -1.9458,   4.0833],
        [ -1.5875,   7.8944,  -9.4286],
        [  0.7275,  14.0667,  -0.9815],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -8.0588,  -6.0341,  -2.3006],
        [  1.1350,  14.0104,   0.6321],
        [  5.1890,   4.7318,   0.9515],
        [  2.5318,  13.7231,   6.3692],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.5128, -10.9365,   9.9645],
        [  9.5018,  -7.1945,  -1.1892],
        [ -2.1533, -10.5778,   1.0112],
        [  1.4306,  13.9620,  -0.2075],
        [ 12.4691,   2.8525,  -5.5110],
        [ 11.4743,   6.2478,  -2.1100],
        [ -7.7239,   4.4298,   3.1216],
        [ -5.2752,  -9.0409,  -8.2568],
        [  5.1890,   4.7318,   0.9515],
        [  3.7992,  -3.9474,  -8.9019],
        [  9.3607,   9.5337,  -4.5412],
        [  6.1755,   2.5233,  -2.0042],
        [ 11.7418,   5.6070,  -1.4192],
        [  1.7500,   7.4187,   4.4237],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 12.5789,   0.1722,   1.6470],
        [  9.1620,  -7.5573,  -9.9882],
        [ -5.6174,   6.4708,  -1.7339],
        [  1.3403,  -4.8011,  -1.1698],
        [  3.9663,   6.1018,  -5.4104],
        [ -8.7544,   2.6096,   3.3146],
        [ -2.5056,   7.7720,   5.5878],
        [ -8.9636,   2.0648,   8.1946],
        [  9.6614,   9.1784,  -1.2087],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -3.3116, -10.1539,   7.5919],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.1477,   7.2550,   8.9688],
        [  3.7916,  -3.9524,  -7.9547],
        [ 10.1983,  -6.3430,  -0.3200],
        [  9.1620,  -7.5573,  -9.9882],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3587,   1.1596,  -3.6629],
        [ 11.7794,   5.5086,  -9.6929],
        [  3.3094,  13.4969,  -3.6507],
        [  6.2576,   2.1359,  -3.3398],
        [ 10.9916,   7.2133,  -0.9179],
        [ -2.5161,   7.7699,  -9.3531],
        [ -5.3179,  -9.0101,   9.4472],
        [  3.3633,  13.4794,   0.5193],
        [  3.3633,  13.4794,   0.5193],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -5.5525,   6.5146,  -4.4984],
        [  4.5057,  -3.3894,  -0.8160],
        [ -7.6734,   4.4980,  -5.4248],
        [ 10.9916,   7.2133,  -0.9179],
        [  5.9883,   3.1536,   6.2320],
        [  9.1620,  -7.5573,  -9.9882],
        [ -6.8618,  -7.6322,   3.7027],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.1222,  -0.7261,   8.8204],
        [  3.4401,  13.4540,   4.6576],
        [  6.3587,   1.1596,  -3.6629],
        [  1.5246,  -4.7838,  -5.2095],
        [ 11.6634,   5.8050,   9.1590],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.1504,   4.3674,  -7.2998],
        [ 10.9916,   7.2133,  -0.9179],
        [ -5.7347,  -8.6908,   9.2878]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0058, 0.0180,  ..., 0.2741, 0.3774, 1.3344],
        [0.0058, 0.0000, 0.0238,  ..., 0.2775, 0.3726, 1.3301],
        [0.0180, 0.0238, 0.0000,  ..., 0.2642, 0.3922, 1.3476],
        ...,
        [0.2741, 0.2775, 0.2642,  ..., 0.0000, 0.4654, 1.6037],
        [0.3774, 0.3726, 0.3922,  ..., 0.4654, 0.0000, 1.3121],
        [1.3344, 1.3301, 1.3476,  ..., 1.6037, 1.3121, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.36004393,  6.42439344,  6.56833864,  6.48486171,  6.65991877,
        6.69389431,  7.96575782,  9.61942445, 10.24557047,  8.25828409,
        6.78325151, 13.3426541 ,  9.94173708,  6.69389431,  6.97106912,
       10.54658597,  7.96575782,  8.62162372, 10.46739073,  6.19707944,
        7.24677664,  5.39098669,  9.95552626, 10.51340829, 11.24434374,
        5.94541803,  9.7712321 ,  9.96152788, 13.1399181 ,  6.21275016,
        9.4997987 ,  9.48969978,  5.13125513,  9.85384623,  9.6282763 ,
        8.05242591, 14.08549802,  8.04483495,  9.43112957, 10.0674887 ,
       14.05633441,  7.02253235, 13.95472331,  8.06990483, 11.22142195,
       11.9182956 , 10.794755  , 14.03505624, 12.79126785, 13.06498554,
        8.90405823, 10.46739073,  7.02253235,  5.47866221, 13.36092624,
        6.67108662, 13.01187484,  7.62232307,  9.94285511, 12.58006038,
       11.87665501,  8.56889738,  4.98461983,  7.27759233,  9.13507321,
        8.16584537,  9.19837792, 13.32613445,  9.91668182, 10.68031351,
       10.0674887 ,  7.56617037,  5.47702766, 12.00995898, 11.87665501,
       13.96905756, 12.3489324 ,  6.46356806, 13.00380624, 13.89671314,
        6.61211224, 13.1471313 ,  8.16714836, 10.46238054, 13.89264822,
       13.89264822,  9.88310614,  8.55982105,  5.63824971,  8.89458273,
       13.1471313 ,  6.76791483, 11.87665501, 10.26327895,  7.8575648 ,
        6.16512856, 13.88683436,  6.46356806,  5.0209005 , 13.02818702,
        6.02588517,  9.64157461,  6.38968357, 12.91143491, 13.1471313 ,
       10.412299  ])}
{'P': tensor([[2.9029e-02, 2.6496e-01, 2.0602e-01,  ..., 1.4488e-03, 4.6467e-06,
         2.3405e-03],
        [2.1214e-03, 4.5171e-02, 2.2082e-03,  ..., 3.5567e-03, 6.7639e-06,
         2.9423e-03],
        [3.7638e-03, 3.4881e-01, 3.3182e-02,  ..., 1.8680e-03, 5.4707e-06,
         2.7148e-03],
        ...,
        [6.3084e-05, 9.1842e-05, 6.8314e-05,  ..., 2.0570e-01, 8.5724e-06,
         1.6287e-05],
        [1.7300e-04, 1.6444e-04, 1.8072e-04,  ..., 3.6869e-05, 2.7760e-01,
         2.8552e-04],
        [3.3758e-05, 7.0296e-05, 3.9784e-05,  ..., 1.5564e-02, 4.8398e-07,
         6.3299e-03]]), 'X': tensor([[  6.3505,   0.6235,  -8.5849],
        [  6.3303,   1.6145,  -8.4083],
        [  6.3589,   0.8239,  -8.4230],
        [  5.8201,   3.5841,  -8.0609],
        [  5.9655,   3.2172,  -7.5953],
        [  5.8617,   3.4853,  -9.8080],
        [ -5.5525,   6.5146,  -4.4984],
        [  2.3261,   7.1726,   6.2522],
        [  4.3459,  -3.5327,   0.3667],
        [  4.7482, -10.3907,   5.9149],
        [ -6.6895,   5.6051,   2.7899],
        [ -6.2200,  -8.2737,   5.5079],
        [ -9.4541,  -0.3327,   8.8246],
        [  5.2199,   4.6861,  -1.9304],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.8704,  -9.3809,   7.5769],
        [  2.6078, -10.9229,   6.2869],
        [  6.1737,   2.5305,   3.9049],
        [ -9.3076,   0.7852,  -6.7010],
        [  3.1892, -10.8220,  -6.8409],
        [ -3.2736, -10.1703,   3.8712],
        [  8.1528,  10.7452,   9.1493],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.3107,  -2.7697,  -3.2919],
        [  1.7000,  -4.7609,   2.7935],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -6.4464,   5.8272,   7.5038],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.9653,  -9.3225,  -3.4727],
        [  3.3069,   6.6089,   8.7377],
        [ 10.7566,   7.6212,  -2.3279],
        [  3.7121,   6.3117,   0.6738],
        [  2.3373,  -4.6232,  -5.2246],
        [ -1.8367,   7.8718,  -8.6523],
        [ -8.9942,  -3.9990,   6.0448],
        [  3.3069,   6.6089,   8.7377],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.0039,   7.3173,  -1.5885],
        [ -0.1626, -10.9796,   6.4629],
        [  5.1196,  -2.7160,   4.2705],
        [  5.2188,  -2.5839,   2.1833],
        [  0.8940, -11.0402,  -0.9783],
        [  6.2555,   2.1474,   4.1685],
        [  5.8325,  -1.5222,  -2.1341],
        [ -1.9714,   7.8563,  -1.8084],
        [ 10.5057,  -5.9099,  -5.1599],
        [  1.8719,   7.3713,  -3.7465],
        [  4.8051, -10.3703,   0.5268],
        [ -9.1231,  -3.5739,   8.4708],
        [  3.3167,   6.6022,   6.3554],
        [ -1.6334, -10.7207,   7.1397],
        [  6.0345,   3.0175,  -1.4424],
        [ 12.0582,  -2.6640,   0.7132],
        [  4.4049,   5.6892,  -4.4394],
        [ -8.4267,  -5.3697,   2.4172],
        [  3.2990,  -4.2415,  -7.4857],
        [  6.2535,  -0.1709,  -9.4102],
        [  2.3889,  -4.6082,  -8.7291],
        [  4.5954,  13.0099,  -4.4439],
        [  5.6314,  -1.9302,  -9.6543],
        [  3.6817,   6.3354,   6.1233],
        [  5.9883,   3.1536,   6.2320],
        [ -4.7642,  -9.3860,  -5.1133],
        [  8.6420,  -8.0580,   6.5781],
        [  4.1140,  -3.7220,  -0.1273],
        [  3.3633,  13.4794,   0.5193],
        [ 11.6544,   5.8273,  -3.1375],
        [ -1.7290,   7.8825,  -3.3831],
        [  9.9756,  -6.6330,  -3.3429],
        [ -7.9925,  -6.1423,   2.8990],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -5.6651,   6.4380,   1.3660],
        [ -9.3616,  -2.4749,   0.1793],
        [ -8.3539,  -5.5111,   0.3373],
        [ -9.3384,  -2.6161,  -4.7630],
        [  1.3767, -11.0341,   9.9415],
        [  6.3535,   0.6792,  -7.5101],
        [  8.4691,  10.4571,  -5.0650],
        [ -0.5739,   7.9056,   0.6382],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -7.3333,   4.9246,  -7.9212],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.6093,  -1.9705,   6.4807],
        [ -1.5261,   7.8988,   9.3768],
        [ -3.2736, -10.1703,   3.8712],
        [  8.5716,  10.3596,   6.8818],
        [  6.1372,   2.6726,  -6.7885],
        [  5.1886,  -2.6250,   6.4677],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 12.6038,   0.7665,  -3.4439],
        [  3.2701,   6.6338,   9.9209],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.5430,   5.5438,  -3.8669],
        [ -8.9282,   2.1648,   8.4894],
        [  2.7420,  -4.4896,  -3.7139],
        [  8.9284,  10.0036,  -6.9210],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.4039,   5.6903,   8.7366],
        [ 11.0864,   7.0387,  -0.1238],
        [  4.5954,  13.0099,  -4.4439],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.6544,   5.8273,  -3.1375]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0047, 0.0023,  ..., 0.2371, 0.9451, 0.2520],
        [0.0047, 0.0000, 0.0049,  ..., 0.2418, 0.9456, 0.2563],
        [0.0023, 0.0049, 0.0000,  ..., 0.2376, 0.9428, 0.2514],
        ...,
        [0.2371, 0.2418, 0.2376,  ..., 0.0000, 0.9689, 0.1156],
        [0.9451, 0.9456, 0.9428,  ..., 0.9689, 0.0000, 0.8541],
        [0.2520, 0.2563, 0.2514,  ..., 0.1156, 0.8541, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.38105015,  6.53290663,  6.41202986,  6.83515201,  6.77777606,
        6.81961996,  8.55982105,  7.5403822 ,  5.60064319, 11.42420508,
        8.7273569 , 10.35093485,  9.45995955,  7.01475376, 12.30695098,
       11.62770332, 11.22993218,  6.67218687,  9.34061938, 11.28215429,
       10.68416401, 13.48809595,  6.80559669,  9.71392179,  5.05534609,
       10.53368644,  8.68979052,  9.53015376, 11.63724792,  7.39004226,
       13.18278606,  7.32233733,  5.18047638,  8.08320235,  9.84314712,
        7.39004226, 11.66453059,  9.58347456,  7.58668144, 10.98076873,
        5.79543038,  5.82345867, 11.07637531,  6.61386436,  6.02789634,
        8.09984231, 12.05395056,  7.60529892, 11.42947632,  9.79814997,
        7.38844703, 10.84437705,  6.74686071, 12.3489324 ,  7.19513806,
        9.99211663,  5.37343173,  6.25586297,  5.19065439, 13.79762415,
        5.95297459,  7.32754882,  6.76791483, 10.52587243, 11.8159254 ,
        5.54776934, 13.89264822, 13.03002338,  8.06990483, 11.97957425,
       10.08001894,  9.97822473,  8.57560275,  9.68323686, 10.00794468,
        9.69792513, 11.11969815,  6.38968357, 13.45642379,  7.92644746,
       12.72524615,  9.50654967,  9.62958973,  8.83341608, 11.81423161,
        5.94535529,  8.04483495, 10.68416401, 13.44593519,  6.69389431,
        5.81481044,  9.69792513, 12.62711252,  7.39599986, 13.32374056,
       13.25017211,  7.16747878,  9.18690018,  5.26067019, 13.40850328,
       13.25017211,  7.19534146, 13.13205357, 13.79762415,  4.85603886,
       13.03002338])}
{'P': tensor([[2.2155e-02, 2.8364e-01, 1.4205e-01,  ..., 1.3121e-05, 7.9100e-06,
         1.5020e-05],
        [1.7856e-03, 2.8512e-02, 2.0382e-01,  ..., 1.5925e-05, 1.0015e-05,
         2.0096e-05],
        [9.7799e-04, 1.0883e-03, 2.9715e-02,  ..., 1.0438e-05, 9.9114e-06,
         2.6653e-05],
        ...,
        [4.0834e-04, 4.1307e-04, 3.6928e-04,  ..., 2.7955e-01, 2.9594e-05,
         6.1124e-05],
        [4.0261e-04, 3.5611e-04, 1.7244e-04,  ..., 1.1714e-04, 7.2474e-02,
         6.4061e-05],
        [6.9344e-05, 6.3788e-05, 3.9216e-05,  ..., 7.0686e-05, 1.0161e-01,
         7.4518e-02]]), 'X': tensor([[ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0016, 0.0209,  ..., 0.9036, 0.9622, 0.9899],
        [0.0016, 0.0000, 0.0195,  ..., 0.9052, 0.9627, 0.9908],
        [0.0209, 0.0195, 0.0000,  ..., 0.9238, 0.9577, 0.9904],
        ...,
        [0.9036, 0.9052, 0.9238,  ..., 0.0000, 1.2834, 1.1349],
        [0.9622, 0.9627, 0.9577,  ..., 1.2834, 0.0000, 0.2289],
        [0.9899, 0.9908, 0.9904,  ..., 1.1349, 0.2289, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.383897  ,  6.42439344,  6.65991877,  6.56833864,  6.69389431,
        6.76056625,  7.24677664, 13.76047999,  5.64500512, 11.65923396,
        9.18690018, 11.21703654,  9.73579738,  9.18690018,  7.822854  ,
        9.562174  ,  4.98989906,  7.48538905, 10.38897563,  9.99145096,
        4.96231532,  4.98989906,  5.91854397, 12.07225469,  6.65259775,
        7.24326992,  7.62509582, 11.04953767,  7.58668144,  5.79391825,
        7.02496223, 11.63383221, 12.16901615,  9.91668182,  9.93553424,
       12.07225469, 12.71843462, 11.21703654,  4.98989906, 10.09979747,
       11.63383221,  7.37902766, 10.67322527,  7.73151309,  5.1805345 ,
       11.31276109, 11.57249984,  9.55593374,  7.02326515,  5.43544626,
        8.06643487,  5.56809335, 10.09979747,  8.10886021, 11.69264243,
        9.9353921 ,  7.48538905, 12.58006038,  5.26342203,  9.75200848,
        7.62011734, 11.20100795, 11.48734978,  9.10945678, 12.1494023 ,
        6.27905292, 13.3607379 ,  8.62162372, 12.76408562, 11.83570132,
        8.7273569 , 10.00802627, 11.018084  ,  8.05242591,  7.98469166,
        8.41171369,  5.18047638, 13.58333857,  4.79003361, 11.52733581,
        9.24501829,  6.1463896 , 10.99326577, 12.10222204,  7.49076899,
        5.63824971, 11.36089355, 13.72594022,  4.79003361,  8.27853076,
        8.89458273, 10.27572673,  9.14044984,  7.02253235,  7.03733775,
        5.08910491,  8.99276564, 13.96746816, 11.86239961, 12.52372629,
       12.18998195, 11.46176259, 11.05413545,  5.89074104, 10.39942497,
        9.62958973])}
{'P': tensor([[2.1989e-02, 1.5162e-01, 1.4617e-01,  ..., 2.3048e-05, 7.0232e-05,
         4.0146e-06],
        [1.8213e-03, 3.2807e-02, 4.9264e-02,  ..., 4.1988e-05, 9.1018e-05,
         5.1936e-06],
        [2.0229e-03, 1.2408e-02, 4.5801e-02,  ..., 6.1143e-05, 9.2189e-05,
         8.5609e-06],
        ...,
        [1.1833e-04, 8.6733e-05, 7.6698e-05,  ..., 8.7829e-02, 7.5357e-04,
         5.2939e-03],
        [6.8269e-03, 6.9831e-03, 4.3409e-03,  ..., 1.1109e-04, 1.9993e-01,
         1.8538e-05],
        [7.7162e-05, 4.5471e-05, 5.0132e-05,  ..., 1.4950e-05, 2.5147e-05,
         3.4267e-02]]), 'X': tensor([[ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9277e-02, 4.6106e-02,  ..., 9.3135e-01, 4.3323e-01,
         1.0508e+00],
        [2.9277e-02, 0.0000e+00, 7.5334e-02,  ..., 9.2696e-01, 4.0397e-01,
         1.0655e+00],
        [4.6106e-02, 7.5334e-02, 1.0727e-04,  ..., 9.3670e-01, 4.7908e-01,
         1.0260e+00],
        ...,
        [9.3135e-01, 9.2696e-01, 9.3670e-01,  ..., 0.0000e+00, 9.4457e-01,
         6.6994e-01],
        [4.3323e-01, 4.0397e-01, 4.7908e-01,  ..., 9.4457e-01, 0.0000e+00,
         1.3074e+00],
        [1.0508e+00, 1.0655e+00, 1.0260e+00,  ..., 6.6994e-01, 1.3074e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.38399789,  6.54127431,  6.59176018,  6.54615887,  6.74872981,
        6.53250937,  6.81084724,  5.2287058 ,  5.0209005 ,  7.72668467,
       10.91805683, 13.06957425,  7.48311002,  7.67939546,  7.5158489 ,
        5.56809335, 11.48663767, 12.91505111,  5.86519525, 10.7506423 ,
        6.05113217,  5.35774573, 13.05148826, 10.31613269, 12.7125079 ,
        9.34127013, 12.7125079 , 12.53345445,  6.50409372, 11.75807252,
       11.05672136,  6.35029274,  9.34127013, 13.01728313, 12.70489563,
        8.40445872,  8.92395381, 10.27572673,  8.38252937,  9.55720651,
       12.45418236, 10.33712258, 12.64235359,  8.92549752,  5.30158497,
       10.61047596,  7.34250948, 12.58006038,  6.24614905, 11.04953767,
        8.27853076, 11.87665501, 10.9036226 ,  9.38232121, 10.82106505,
        4.91652556,  8.43626114, 13.96905756,  5.13616224,  7.72668467,
       10.67132524, 12.70489563, 11.87412439,  7.53884104,  7.13501264,
        9.58072286, 10.52357479,  8.89458273, 13.3426541 , 13.13205357,
       11.97957425,  8.35856487, 11.35182589, 13.70893492,  5.88782913,
       11.75807252,  7.03733775,  6.76056625,  6.05165248,  9.36378438,
        9.68324553, 10.69963954, 12.67680439,  6.32572323, 11.56146873,
       12.22503754, 13.97430191,  7.02326515,  6.76791483,  7.03733775,
       11.95743558,  7.20826219,  6.35029274, 11.22596994,  7.72415048,
        5.10446653,  5.08910491, 11.66453059, 10.24557047,  9.58347456,
        6.53290663,  9.77588801,  5.91854397,  9.60359158,  5.22903117,
       10.42667093])}
{'P': tensor([[2.1202e-02, 2.0075e-01, 1.6700e-01,  ..., 3.7429e-06, 1.2472e-03,
         5.1903e-03],
        [1.2468e-03, 2.0507e-02, 2.2790e-01,  ..., 3.5683e-06, 1.3590e-03,
         4.6811e-03],
        [1.0253e-03, 1.2589e-03, 2.2501e-02,  ..., 3.7897e-06, 1.5463e-03,
         4.4726e-03],
        ...,
        [1.3129e-04, 1.3394e-04, 1.3155e-04,  ..., 1.8368e-01, 1.7541e-04,
         2.2078e-05],
        [3.1320e-05, 4.5666e-05, 6.4708e-05,  ..., 1.9807e-06, 2.5046e-02,
         1.5189e-04],
        [1.6456e-04, 2.5857e-04, 4.0088e-04,  ..., 4.8355e-07, 6.0681e-03,
         3.5043e-02]]), 'X': tensor([[  6.3535,   0.6792,  -7.5101],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3031,   1.8477,  -7.1706],
        [  6.1929,   2.4501,  -6.9352],
        [  6.1372,   2.6726,  -6.7885],
        [  5.9655,   3.2172,  -7.5953],
        [  5.3804,  -2.3504,   7.2744],
        [  1.1006,   7.6318,  -2.4877],
        [  7.1821,  -9.1845,   9.8548],
        [  6.3549,   0.7096,   6.9198],
        [  1.5662,  13.9375,   4.5142],
        [  6.3460,   1.4252,   1.2291],
        [  1.3103,  -4.8032,  -0.9300],
        [  8.6420,  -8.0580,   6.5781],
        [ 12.2435,   4.0076,   8.1257],
        [  5.2199,   4.6861,  -1.9304],
        [  2.2860,   7.1916,  -6.1011],
        [  5.9912,   3.1453,  -3.9989],
        [  7.0171,  11.6439,  -5.0178],
        [ 10.8670,  -5.3428,  -2.8637],
        [  0.9809,   7.6642,  -1.4217],
        [ -6.9838,  -7.4965,   0.9796],
        [ -7.4602,   4.7718,  -4.6277],
        [  7.7794,  11.0629,   8.9069],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.9912,   3.1453,  -3.9989],
        [ 10.4737,  -5.9570,  -1.2108],
        [  0.1460,   7.8340,  -9.4614],
        [  4.4290,  -3.4596,   1.4181],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -5.9872,   6.2037,  -9.4477],
        [  5.4179,  -2.2925,  -3.4641],
        [  5.3924, -10.1396,  -4.0863],
        [  0.3715,  -4.7756,   0.2312],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.3860,  -2.3418,   1.0826],
        [ -4.4779,  -9.5609,  -2.4505],
        [  1.0000,   7.6591,  -4.5011],
        [ -8.8540,  -4.3971,  -5.2144],
        [  2.7420,  -4.4896,  -3.7139],
        [ -2.1382,   7.8339,   7.3524],
        [  6.3124,   0.2057,  -0.4593],
        [  6.3550,   0.7129,   0.4917],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.9809,   7.6642,  -1.4217],
        [  4.4049,   5.6892,  -4.4394],
        [  5.7485,  -9.9811,  -3.2355],
        [ 12.0772,  -2.6014,  -7.1832],
        [  2.4639, -10.9431,  -7.3718],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.0467,  -9.8369,  -6.5816],
        [  4.5325,  -3.3643,   6.7699],
        [ 10.7566,   7.6212,  -2.3279],
        [ -4.5778,  -9.5013,  -5.0667],
        [  8.7715,  -7.9390,   5.2900],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.1222,  -0.7261,   8.8204],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.3804,  -2.3504,   7.2744],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.1348,   5.9514,  -0.8856],
        [  3.8111, -10.6787,  -4.6539],
        [ 11.1085,  -4.9204,   0.0736],
        [ 12.0772,  -2.6014,  -7.1832],
        [  7.0515,  11.6194,   8.4289],
        [  9.2979,   9.6048,   7.3138],
        [  2.2076,  13.8032,  -1.5099],
        [ 12.0042,  -2.8350,   2.6864],
        [ -6.1914,  -8.2997,   8.5380],
        [ -0.5385,   7.9037,   6.3830],
        [  2.9410,   6.8428,   5.4458],
        [  5.1861,   4.7361,  -3.7312],
        [  0.5250, -11.0306,   5.5358],
        [ 10.8218,  -5.4177,   6.8573],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.1929,   2.4501,  -6.9352],
        [  5.8275,  -1.5334,   2.0124],
        [  3.2692,  13.5098,  -4.9032],
        [  5.2608,  12.6977,   4.9166],
        [  1.1644, -11.0394,   7.8511],
        [  4.9152,  12.8653,  -1.4459],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.8934,   2.2597,   3.5831],
        [  6.3535,   0.6792,  -7.5101],
        [  1.7815,  -4.7481,   4.4289],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -0.9715, -10.8630,   0.3555],
        [  2.0729,   7.2878,   7.0880],
        [  4.5325,  -3.3643,   6.7699],
        [  1.7000,  -4.7609,   2.7935],
        [  6.2895,   0.0405,   5.7652],
        [ -6.8618,  -7.6322,   3.7027],
        [ -7.9925,  -6.1423,   2.8990],
        [ -0.0254, -10.9932,   3.2200],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.4766,  -1.1116,  -2.1706],
        [  1.9204,   7.3518,   8.4781],
        [  4.0457,  -3.7737,  -0.4704],
        [  9.4516,  -7.2501,  -1.8868],
        [  1.2273,  -4.8081,   9.8129],
        [  9.6587,   9.1817,  -0.8212],
        [ 11.0373,   7.1299,  -8.5895]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0019, 0.0032,  ..., 0.9382, 0.4125, 0.3356],
        [0.0019, 0.0000, 0.0020,  ..., 0.9364, 0.4113, 0.3371],
        [0.0032, 0.0020, 0.0000,  ..., 0.9360, 0.4123, 0.3388],
        ...,
        [0.9382, 0.9364, 0.9360,  ..., 0.0000, 0.6848, 1.1897],
        [0.4125, 0.4113, 0.4123,  ..., 0.6848, 0.0000, 0.5355],
        [0.3356, 0.3371, 0.3388,  ..., 1.1897, 0.5355, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.38968357,  6.48486171,  6.56833864,  6.65991877,  6.69389431,
        6.77777606,  5.87132718,  7.71075226, 11.65923396,  6.39438044,
       14.02526505,  6.50409372,  4.978704  , 11.8159254 , 12.88270018,
        7.01475376,  7.54620849,  6.76661595, 13.59480906, 12.10942412,
        7.72668467, 10.24557047,  8.85572292, 13.52430176,  9.54154096,
        6.76661595, 12.04924159,  7.83534987,  5.62005629,  7.5092834 ,
        9.50913849, 10.61047596,  8.62162372,  5.8828986 , 11.48434767,
        4.79003361,  5.94541803,  5.87305005, 10.5575601 ,  7.72415048,
        9.88571713,  5.26067019,  8.12043983,  6.31576264,  6.3949043 ,
        9.50654967,  7.72668467,  7.19513806, 11.51811603, 12.35421461,
       11.21703654,  5.56809335, 11.54672755,  5.64464807, 13.18278606,
       10.54658597, 11.83075005, 13.18278606,  6.16512856,  9.77777346,
        5.87132718,  4.91296205,  7.24677664, 11.33837396, 12.1494023 ,
       12.35421461, 13.59170809, 13.36800073, 13.97857591, 12.3344536 ,
       10.35464916,  7.92200342,  7.44805901,  7.02326515, 11.04313641,
       12.10222204,  8.29022197,  9.58072286,  6.65991877,  6.02588517,
       13.89974693, 13.744387  , 11.10066126, 13.77222752,  6.1561646 ,
        9.17595899,  6.38968357,  5.0713364 , 10.30972193, 10.90638066,
        7.57685908,  5.64464807,  5.05534609,  6.28961988, 10.26327895,
       10.08001894, 10.99326577, 13.18278606, 13.59673406,  9.54154096,
        7.5984809 ,  5.53255882, 11.9120321 ,  4.96231532, 13.32645552,
       13.1399181 ])}
{'P': tensor([[2.4871e-02, 1.9845e-01, 1.9571e-01,  ..., 1.1849e-04, 1.8389e-04,
         5.8383e-04],
        [1.6458e-03, 4.4347e-02, 2.1866e-03,  ..., 3.4943e-04, 1.1661e-04,
         1.7830e-03],
        [2.1271e-03, 2.2477e-01, 2.6437e-02,  ..., 1.4189e-04, 1.3528e-04,
         6.7612e-04],
        ...,
        [2.3857e-05, 2.0268e-05, 2.1848e-05,  ..., 3.3754e-02, 2.9644e-06,
         8.1087e-05],
        [3.1859e-02, 3.8724e-02, 3.3416e-02,  ..., 2.6189e-05, 1.1357e-01,
         3.3999e-04],
        [9.2986e-05, 1.0938e-04, 9.2737e-05,  ..., 9.9497e-02, 7.0833e-06,
         1.1248e-01]]), 'X': tensor([[ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0039, 0.0151,  ..., 0.7093, 0.1300, 0.6103],
        [0.0039, 0.0000, 0.0153,  ..., 0.7057, 0.1293, 0.6085],
        [0.0151, 0.0153, 0.0000,  ..., 0.7149, 0.1446, 0.6233],
        ...,
        [0.7093, 0.7057, 0.7149,  ..., 0.0000, 0.6408, 0.4343],
        [0.1300, 0.1293, 0.1446,  ..., 0.6408, 0.0000, 0.4868],
        [0.6103, 0.6085, 0.6233,  ..., 0.4343, 0.4868, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.39187285,  6.61211224,  6.46356806,  6.54693987,  6.65259775,
        6.76661595, 10.33712258, 11.11038551,  9.63742669, 11.00726483,
       10.41909211,  4.85603886,  8.01009083,  8.7273569 , 11.85890664,
        9.71392179,  9.37989631, 13.5247215 , 10.412299  , 13.31764181,
       10.71162681,  9.38232121,  4.91652556,  5.19065439,  7.96181288,
        9.75200848, 10.41909211, 10.68416401,  9.11473391, 12.13568598,
        7.37576154, 13.41713661, 10.78533838, 13.35927527,  8.53014915,
        9.5014146 ,  8.31625171, 10.87409005,  9.34061938, 10.91601655,
        5.16205087, 13.27717982,  9.41244865,  8.94395628, 12.33167382,
        5.16077764, 11.48734978, 10.92597196,  6.29034958, 11.89105259,
        5.94535529,  9.43112957,  5.21280156,  6.34059021,  6.27715105,
       13.14777433,  7.50035355, 11.83075005,  5.20659892,  7.17789666,
        9.71392179,  6.12886024, 10.61047596, 14.02526505,  8.49161154,
        5.87132718,  7.57685908, 12.10222204, 12.22503754, 13.5247215 ,
        7.54620849, 12.49317563,  7.05627461,  8.29203356, 10.31584102,
        8.20436764,  9.48337063, 13.0743399 ,  5.18277858,  6.36004393,
        7.62011734,  9.41970031,  7.26074925,  8.98765644,  9.88571713,
       12.41012046, 11.69264243,  7.60463349,  8.12580976,  5.19986797,
        8.29640613,  7.06372172,  6.53250937,  8.1137452 ,  8.25828409,
        7.06169391, 11.28093507,  4.78258092,  9.48337063,  6.25586297,
        5.80312949, 12.04924159, 13.15525803,  8.89301516, 12.25806559,
        8.09899874])}
{'P': tensor([[3.4263e-02, 1.8162e-01, 1.5517e-01,  ..., 1.5413e-02, 1.0080e-03,
         1.1547e-04],
        [8.4032e-04, 3.4051e-02, 3.5901e-02,  ..., 5.1547e-02, 7.2519e-04,
         2.4368e-04],
        [9.1095e-04, 2.1064e-02, 3.5716e-02,  ..., 5.7045e-02, 9.6330e-04,
         3.1949e-04],
        ...,
        [3.6951e-04, 6.5776e-04, 6.6696e-04,  ..., 1.0333e-01, 3.7683e-04,
         3.0014e-03],
        [5.6938e-04, 1.3292e-03, 1.6878e-03,  ..., 5.0225e-03, 1.0621e-01,
         1.4628e-02],
        [2.1321e-05, 2.5808e-05, 2.9447e-05,  ..., 9.7585e-05, 1.0033e-04,
         7.8609e-02]]), 'X': tensor([[ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0262, 0.0354,  ..., 0.1914, 0.6747, 0.8764],
        [0.0262, 0.0000, 0.0242,  ..., 0.1654, 0.6755, 0.8653],
        [0.0354, 0.0242, 0.0000,  ..., 0.1645, 0.6513, 0.8434],
        ...,
        [0.1914, 0.1654, 0.1645,  ..., 0.0000, 0.6830, 0.7916],
        [0.6747, 0.6755, 0.6513,  ..., 0.6830, 0.0000, 0.4051],
        [0.8764, 0.8653, 0.8434,  ..., 0.7916, 0.4051, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.39438044,  6.7593917 ,  6.76791483,  6.90115901,  7.06169391,
        7.07902036,  6.53290663, 10.32715715,  4.91652556,  4.83371341,
        5.62005629,  5.91854397, 14.03505624,  5.18277858,  8.99276564,
       12.17901033, 10.00794468, 13.61600536, 13.99877413,  6.99097013,
       10.08001894,  9.51074763, 13.69162795,  7.73151309,  8.8737556 ,
        6.42439344,  5.26342203,  9.14973319,  8.29022197, 10.74409834,
        6.83515201, 13.46553623, 12.49291768, 10.46238054, 10.06023665,
       12.67680439,  9.60359158, 13.95472331,  8.62631801, 13.61801526,
       13.14777433, 11.48434767, 10.08001894,  5.05774295, 13.26209545,
        5.54776934,  5.91677188,  6.7593917 ,  9.97964013,  8.12043983,
        5.94535529, 10.61047596, 11.77124373,  6.31576264, 12.27991991,
        6.53250937,  8.41148722,  7.20482224,  8.47371868, 11.97238624,
        7.9337647 , 13.32374056, 10.31584102, 11.28895489, 13.36092624,
        4.78002665,  9.86089579,  6.61386436,  7.67939546, 11.73397309,
       10.52357479,  6.76056625,  9.99211663,  5.46244988,  6.06416957,
       13.32557133,  5.19577146, 11.00508681,  5.87132718, 13.00380624,
       13.3607379 ,  4.74770899, 10.1371552 ,  6.18668463, 13.39372426,
        7.62011734,  7.58668144,  9.68323686, 13.02818702, 13.03002338,
       14.02532758,  8.31625171,  5.79543038, 13.99877413,  9.97964013,
       11.52733581,  7.06169391,  9.43112957,  7.5158489 ,  5.30158497,
        7.06372172, 12.12641969,  9.78147542,  7.5403822 ,  6.89245352,
        8.34949383])}
{'P': tensor([[2.5279e-02, 1.9018e-01, 1.4357e-01,  ..., 1.8653e-04, 1.5572e-03,
         3.6670e-04],
        [2.3279e-03, 5.4204e-02, 1.4684e-02,  ..., 5.9542e-04, 5.6751e-03,
         3.6214e-04],
        [2.5388e-03, 5.7204e-02, 3.8095e-02,  ..., 5.0288e-04, 3.9258e-03,
         4.1292e-04],
        ...,
        [3.1061e-05, 4.1398e-05, 4.3720e-05,  ..., 3.4684e-01, 1.0613e-03,
         3.5474e-06],
        [2.5357e-06, 4.0075e-06, 3.6419e-06,  ..., 5.7979e-03, 8.0460e-03,
         1.6429e-07],
        [1.1435e-04, 1.9463e-04, 2.4051e-04,  ..., 5.6274e-04, 1.1845e-03,
         9.4219e-03]]), 'X': tensor([[  6.3550,   0.7129,   0.4917],
        [  6.2804,   2.0019,   0.8607],
        [  6.3246,   1.6696,  -0.1544],
        [  6.3305,   1.6119,   1.1538],
        [  6.3211,   1.7017,  -0.2164],
        [  6.0305,   3.0296,   0.1375],
        [ 12.6038,   0.7665,  -3.4439],
        [  3.6154,  -4.0640,  -7.9698],
        [  1.8320,   7.3871,   5.1452],
        [  7.8375,  -8.7240,  -9.8913],
        [  0.3724,   7.7974,  -6.8722],
        [  4.8432,   5.1968,   1.5006],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.4524,  -1.6813,  -6.6862],
        [  1.1274,  14.0116,   8.1753],
        [  2.1379,  -4.6757,   6.4981],
        [ -9.4671,  -0.5553,   0.1117],
        [ -1.6220,   7.8917,   0.0315],
        [ -7.3446,   4.9113,  -5.9356],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.0345,   3.0175,  -1.4424],
        [ -2.8162,   7.7059,  -9.6885],
        [  1.9503,  -4.7172,  -6.2313],
        [ -7.3333,   4.9246,  -7.9212],
        [  6.2038,   2.4020,  -3.1692],
        [  3.6154,  -4.0640,  -7.9698],
        [ -1.6906,   7.8860,   7.7142],
        [ 11.7794,   5.5086,  -9.6929],
        [  6.3122,  -9.6994,  -6.3282],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -0.9715, -10.8630,   0.3555],
        [  1.1644, -11.0394,   7.8511],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -8.0965,  -5.9712,   4.3466],
        [ 10.0567,   8.6697,   4.6329],
        [  6.2247,  -0.3150,  -9.3657],
        [  3.1271, -10.8343,   2.6442],
        [ -9.2373,  -3.1225,  -8.9763],
        [  3.3069,   6.6089,   8.7377],
        [ -7.8359,   4.2735,   7.2943],
        [  2.5974,   7.0363,  -7.7810],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -3.6381,   7.4683,   7.3544],
        [ 12.0582,  -2.6640,   0.7132],
        [ -9.4721,  -1.3097,  -7.4377],
        [  7.5930,  -8.9037,   7.2323],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.0937,  13.8293,  -2.3533],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 11.2847,  -4.5848,  -4.7503],
        [  4.1637,  13.1898,  -1.9189],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -7.9667,   4.0814,  -0.6397],
        [  0.9074,  -4.8139,   8.1674],
        [  4.5430,   5.5438,  -3.8669],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.1973,  -0.4372,   6.5720],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.0961,   1.6529,   3.1553],
        [  9.9058,  -6.7203,   5.6928],
        [ -1.6086,   7.8928,   4.3385],
        [  3.1892, -10.8220,  -6.8409],
        [ -1.6220,   7.8917,   0.0315],
        [  1.2728, -11.0372,   9.8121],
        [  3.4140,  -4.1803,  -8.8224],
        [ 12.4901,  -0.7052,   1.7315],
        [ 10.6673,  -5.6649,  -4.1646],
        [  6.1752,   2.5242,   3.5874],
        [  0.1047, -11.0046,   0.5188],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.9284,  11.7061,  -0.3732],
        [ -8.3653,   3.4184,   9.3454],
        [  6.8704,  -9.3809,   7.5769],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.4773,  -1.0024,   5.2599],
        [ -5.7347,  -8.6908,   9.2878],
        [  5.0506,   4.9277,  -9.5094],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.3464,   0.5708,  -1.6397],
        [ -3.9700,   7.3453,  -4.1486],
        [  5.9587,  12.3216,   9.1695],
        [ 12.1504,   4.3674,  -7.2998],
        [ -0.0254, -10.9932,   3.2200],
        [ -1.8788, -10.6567,   3.7068],
        [ -8.5900,   2.9767,   7.4811],
        [  6.2419,  -0.2313,   4.2440],
        [ -1.0012, -10.8576,   2.2867],
        [  1.1006,   7.6318,  -2.4877],
        [  0.1047, -11.0046,   0.5188],
        [  6.6319,  11.9066,   3.5165],
        [  6.1989,  -9.7591,   0.8289],
        [  4.3377,   5.7570,   1.3712],
        [ -6.0175,  -8.4541,  -9.8408],
        [  2.0870,  -4.6877,   9.6823],
        [  2.4697,  -4.5836,   1.4593],
        [ -4.1296,   7.2803,  -2.7194],
        [  7.7794,  11.0629,   8.9069],
        [ -3.2736, -10.1703,   3.8712],
        [  0.8572,  14.0501,  -8.3364],
        [  4.1637,  13.1898,  -1.9189],
        [ 12.5832,   1.7543,  -7.6134]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.4435e-02, 4.0977e-02,  ..., 5.9946e-01, 2.9609e-01,
         6.2055e-01],
        [3.4435e-02, 1.0727e-04, 7.5334e-02,  ..., 6.3386e-01, 3.2040e-01,
         6.5363e-01],
        [4.0977e-02, 7.5334e-02, 0.0000e+00,  ..., 5.5890e-01, 2.7318e-01,
         5.8272e-01],
        ...,
        [5.9946e-01, 6.3386e-01, 5.5890e-01,  ..., 0.0000e+00, 4.4156e-01,
         1.4976e-01],
        [2.9609e-01, 3.2040e-01, 2.7318e-01,  ..., 4.4156e-01, 0.0000e+00,
         3.9636e-01],
        [6.2055e-01, 6.5363e-01, 5.8272e-01,  ..., 1.4976e-01, 3.9636e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.3949043 ,  6.59176018,  6.54127431,  6.53250937,  6.54615887,
        6.74872981, 12.62711252,  5.43943791,  7.61088125, 11.72749923,
        7.80625981,  7.10378919, 13.20597932,  9.60080639, 14.05687815,
        5.14124612,  9.48337063,  8.05668609,  8.8353674 , 10.06023665,
        6.74686071,  8.20436764,  5.10444097,  8.83341608,  6.65259775,
        5.43943791,  8.06517143, 13.00380624, 11.57249984,  9.99145096,
       10.90638066, 11.10066126, 12.05395056, 10.06023665, 13.2778378 ,
        6.23262289, 11.27656488,  9.75075273,  7.39004226,  8.92549752,
        7.50035355,  9.17016857,  9.88310614,  8.30729352, 12.3489324 ,
        9.562174  , 11.70168755,  6.14086582,  6.02588517,  9.41605684,
       13.98691426, 12.26997165,  9.71392179, 12.18045978, 13.83139024,
        9.61942445,  8.95133961,  4.89869205,  7.16747878,  5.83560009,
        6.21275016,  7.37902766,  9.24501829, 11.9702762 ,  8.0550286 ,
       11.28215429,  8.05668609, 11.11038551,  5.39723241, 12.50996828,
       12.07818871,  6.67121748, 11.00508681,  5.63824971, 13.60275211,
        9.0368458 , 11.62770332, 13.09737053,  7.37902766,  9.53015376,
       10.412299  ,  7.05627461,  5.0713364 ,  9.36378438,  8.34949383,
       13.68672855, 12.91143491, 10.99326577, 10.82106505,  9.09119351,
        6.24614905, 10.9036226 ,  7.71075226, 11.00508681, 13.62896593,
       11.56146873,  7.20826219, 10.37698435,  5.13125513,  5.20659892,
        8.36995953, 13.52430176, 10.68416401, 14.07623266, 13.83139024,
       12.70489563])}
{'P': tensor([[2.1399e-02, 2.2350e-01, 2.0160e-01,  ..., 6.5968e-05, 5.1615e-05,
         8.9786e-05],
        [1.5047e-03, 2.8756e-02, 3.2653e-01,  ..., 5.7025e-05, 4.5804e-05,
         9.0982e-05],
        [2.2082e-03, 3.3390e-03, 4.4936e-02,  ..., 8.0016e-05, 6.4696e-05,
         1.3328e-04],
        ...,
        [8.7100e-03, 8.4585e-03, 8.3441e-03,  ..., 2.7614e-02, 1.2676e-04,
         7.1011e-03],
        [2.0659e-03, 1.9047e-03, 1.8562e-03,  ..., 7.3500e-01, 6.8484e-03,
         1.6974e-03],
        [3.3188e-04, 4.5881e-04, 5.0636e-04,  ..., 2.5345e-05, 1.7277e-05,
         5.1652e-02]]), 'X': tensor([[  6.3551,   0.7149,   1.3344],
        [  6.3460,   1.4252,   1.2291],
        [  6.3305,   1.6119,   1.1538],
        [  6.2804,   2.0019,   0.8607],
        [  6.2905,   1.9366,   2.1169],
        [  5.7845,   3.6652,   0.9175],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.2026,  12.1774,  -1.5887],
        [ -5.6423,   6.4537,  -1.2091],
        [ 11.3367,  -4.4806,   0.4643],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.3569,   0.7591,   6.5011],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.2974,   0.8371,   2.4852],
        [  2.6310, -10.9195,   3.8054],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -7.4808,   4.7462,   2.0266],
        [  5.4867,  12.5816,   2.9549],
        [ -5.1046,  -9.1610,   2.3418],
        [  5.8318,   3.5567,  -4.7692],
        [  6.2022,   2.4092,   3.0349],
        [  5.9843,  -9.8679,  -6.1449],
        [  5.5158,  -2.1337,   3.5246],
        [  6.3534,   1.2990,  -7.2942],
        [ -6.7745,  -7.7264,   8.4684],
        [  6.2883,   0.0325,   5.2692],
        [  2.4572,   7.1085,   7.7376],
        [  5.8318,   3.5567,  -4.7692],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.0608,   3.9362,  -5.9567],
        [  4.5503,   5.5359,   7.4735],
        [ -7.8359,   4.2735,   7.2943],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.8275,  -1.5334,   2.0124],
        [ 12.4837,   2.7510,   0.4449],
        [  7.3592,  11.3937,   6.2407],
        [  6.3551,   0.7149,   1.3344],
        [  3.3167,   6.6022,   6.3554],
        [  5.4167, -10.1293,   6.2156],
        [  5.9193,  -1.3176,   5.1739],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.6314,  -1.9302,  -9.6543],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 11.9019,  -3.1379,  -3.4534],
        [  1.7903,  -4.7467,   0.1168],
        [  6.9314,  -9.3435,  -1.2729],
        [  3.2701,   6.6338,   9.9209],
        [ -8.8934,   2.2597,   3.5831],
        [ -6.0844,  -8.3955,   0.7705],
        [  7.5064,  -8.9651,  -0.0994],
        [ -3.5173,   7.5091,  -8.9585],
        [  3.2366,  -4.2732,   5.1532],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -5.1974,   6.7396,   3.3484],
        [  6.2770,  -0.0379,   2.4195],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.1752,   2.5242,   3.5874],
        [  9.1409,  -7.5789,   1.5814],
        [ -8.0766,  -6.0046,  -3.1873],
        [  1.8721,  -4.7322,   0.5524],
        [ -4.4503,   7.1378,   3.3751],
        [  0.5175,  14.0910,   4.6540],
        [  6.3031,   1.8477,  -7.1706],
        [ 12.4996,   2.6335,  -6.3166],
        [ -0.4341, -10.9473,   7.4509],
        [  3.6249, -10.7255,   5.6605],
        [  9.2811,  -7.4336,   5.1771],
        [  3.2159, -10.8166,  -1.0406],
        [ -4.4520,   7.1370,  -8.7867],
        [  7.5930,  -8.9037,   7.2323],
        [ -9.3705,   0.4208,  -3.5815],
        [  3.9770, -10.6341,   8.0783],
        [  0.3724,   7.7974,  -6.8722],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3266,   0.3292,   9.3502],
        [  5.8902,  -1.3887,   4.0388],
        [  5.6093,  -1.9705,   6.4807],
        [ -7.3333,   4.9246,  -7.9212],
        [  6.1737,   2.5305,   3.9049],
        [ -9.4714,  -1.3301,  -3.1998],
        [  9.3836,   9.5075,  -5.9278],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 11.0373,   7.1299,  -8.5895],
        [  1.6408,   7.4591,  -4.2294],
        [ -1.9646,   7.8571,   3.3817],
        [ -8.0608,   3.9362,  -5.9567],
        [  0.7275,  14.0667,  -0.9815],
        [  5.9526,   3.2525,  -4.3821],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.8745,   2.3097,   3.5600],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -6.3252,  -8.1760,  -1.0891],
        [  3.3744,   6.5623,  -6.3636],
        [  5.8617,   3.4853,  -9.8080],
        [  2.5636, -10.9293,   9.6519],
        [  5.9883,   3.1536,   6.2320],
        [  8.9284,  10.0036,  -6.9210],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.1203,  -2.7151,   6.2110],
        [ 10.4737,  -5.9570,  -1.2108],
        [  9.9750,  -6.6338,  -1.2440],
        [ 12.5927,   0.4199,  -9.4997]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 4.4699e-03, 1.0072e-02,  ..., 2.5811e-01, 2.8415e-01,
         7.3631e-01],
        [4.4699e-03, 0.0000e+00, 5.6061e-03,  ..., 2.5405e-01, 2.8033e-01,
         7.3224e-01],
        [1.0072e-02, 5.6061e-03, 0.0000e+00,  ..., 2.4885e-01, 2.7541e-01,
         7.2730e-01],
        ...,
        [2.5811e-01, 2.5405e-01, 2.4885e-01,  ..., 0.0000e+00, 4.0597e-02,
         5.9975e-01],
        [2.8415e-01, 2.8033e-01, 2.7541e-01,  ..., 4.0597e-02, 0.0000e+00,
         6.2193e-01],
        [7.3631e-01, 7.3224e-01, 7.2730e-01,  ..., 5.9975e-01, 6.2193e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.39520397,  6.50409372,  6.53250937,  6.59176018,  6.58184474,
        6.84796202,  8.34949383, 13.6660671 ,  8.5724042 , 12.18998195,
        5.94541803,  6.4020404 , 11.66453059,  9.3349858 , 11.23201619,
        9.78286733,  8.859416  , 13.72594022, 10.48721382,  6.83083863,
        6.65368551, 11.54071859,  5.91408504,  6.48486171, 10.27572673,
        6.28835101,  7.52117132,  6.83083863, 13.89974693,  8.97052115,
        7.16598273,  8.92549752, 11.75807252,  6.02588517, 12.78327037,
       13.56367857,  6.39520397,  7.38844703, 11.48663767,  6.06416957,
       11.63724792,  5.95297459,  8.97052115,  9.11473391,  9.87720102,
        9.27474386,  9.562174  , 12.30858677,  5.07306274, 11.63383221,
        7.39599986,  9.17595899, 10.36844813, 11.69264243,  8.29203356,
        5.36061974,  9.75075273,  8.51088913,  6.27715105,  6.1561646 ,
        6.67121748, 11.87412439, 10.06407839,  5.08910491,  8.41148722,
       14.10045797,  6.56833864, 12.77402351, 10.95593729, 11.32149259,
       11.89105259, 11.2845608 ,  8.41171369, 11.70168755,  9.37989631,
       11.35345345,  7.80625981, 12.87752052,  6.33517349,  6.05165248,
        5.94535529,  8.83341608,  6.67218687,  9.5642993 , 13.35832725,
       10.06407839, 13.1399181 ,  7.6374603 ,  8.09899874,  8.97052115,
       14.08549802,  6.78325151, 10.52357479,  9.17016857,  9.71392179,
       10.33712258,  7.37902766,  6.81961996, 11.22596994,  6.76791483,
       13.40850328,  4.9160962 ,  5.79562691, 12.04924159, 11.97949273,
       12.59969936])}
{'P': tensor([[4.3764e-02, 2.3225e-01, 2.2783e-01,  ..., 1.2922e-03, 3.0812e-05,
         3.9894e-05],
        [1.0590e-03, 4.1752e-02, 4.4020e-02,  ..., 4.3993e-04, 1.2462e-05,
         1.5413e-05],
        [1.1143e-03, 2.6599e-02, 4.5100e-02,  ..., 4.6516e-04, 1.3700e-05,
         1.6045e-05],
        ...,
        [1.1831e-01, 6.4359e-02, 5.3916e-02,  ..., 5.9510e-02, 1.5466e-04,
         2.7941e-04],
        [5.8634e-03, 1.9745e-03, 2.0925e-03,  ..., 6.7491e-03, 5.2882e-02,
         1.4061e-02],
        [2.3226e-03, 8.4174e-04, 7.7645e-04,  ..., 4.8947e-03, 4.9498e-04,
         1.2997e-02]]), 'X': tensor([[ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0321, 0.0266,  ..., 0.0599, 0.4514, 0.4434],
        [0.0321, 0.0000, 0.0242,  ..., 0.0465, 0.4386, 0.4120],
        [0.0266, 0.0242, 0.0000,  ..., 0.0688, 0.4260, 0.4221],
        ...,
        [0.0599, 0.0465, 0.0688,  ..., 0.0000, 0.4798, 0.4188],
        [0.4514, 0.4386, 0.4260,  ..., 0.4798, 0.0000, 0.3979],
        [0.4434, 0.4120, 0.4221,  ..., 0.4188, 0.3979, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.4020404 ,  6.7593917 ,  6.76791483,  7.06169391,  6.90115901,
        7.07902036, 10.84437705,  6.61211224, 13.66051884, 10.84493732,
       11.76830981,  9.86089579,  5.8828986 ,  5.63824971,  7.5403822 ,
        6.80559669, 11.65923396,  9.79814997,  8.81930525,  9.77777346,
        7.4544866 , 11.62158554,  9.58072286,  7.32950385, 13.66051884,
       12.10222204,  6.61386436,  5.52416368,  9.58072286, 14.0393949 ,
        7.8575648 , 10.67132524, 13.89671314,  5.5748624 , 13.96987806,
       10.06850677,  7.33350642,  7.4544866 ,  9.09502385,  5.24461068,
       13.04156075, 13.61801526, 13.98691426,  6.10360073, 11.83570132,
       13.13205357,  8.05242591,  6.77777606, 13.61600536, 13.85158417,
       13.77604293,  9.60359158, 12.49317563, 13.96905756, 11.51811603,
        9.15866607, 12.52372629,  9.15976746, 13.79762415,  9.85330646,
       12.24467041,  9.17905491, 13.09737053,  5.21280156,  7.92607342,
       11.57249984, 11.69264243, 12.44177641,  9.45995955,  9.75200848,
       10.48721382,  9.61942445, 11.9120321 ,  6.65259775, 13.59480906,
       10.99326577,  7.63726134,  7.37902766,  8.47371868,  8.1303945 ,
        8.29640613,  8.50278896, 11.35182589,  7.78980299, 11.62158554,
        7.26074925,  7.73692436, 13.36800073, 10.99326577,  5.47866221,
        8.53501777,  7.15100303,  9.13507321,  7.39599986,  9.94173708,
        8.95983586, 13.44593519,  8.30729352,  9.79814997,  9.84314712,
        6.38105015,  8.16714836,  8.22728399,  5.94293185, 11.52733581,
       11.62158554])}
{'P': tensor([[2.5804e-02, 2.7126e-01, 1.1302e-01,  ..., 3.9441e-04, 4.5108e-06,
         8.4129e-04],
        [1.4937e-03, 3.0555e-02, 1.9253e-01,  ..., 4.1352e-04, 4.1343e-06,
         1.5139e-03],
        [2.7571e-03, 3.6676e-03, 1.3123e-01,  ..., 1.0355e-03, 9.6976e-06,
         1.7466e-02],
        ...,
        [3.7799e-05, 5.7863e-05, 1.9191e-04,  ..., 2.4940e-02, 3.7750e-06,
         1.5735e-03],
        [4.4900e-05, 5.2314e-05, 7.4784e-05,  ..., 2.3313e-02, 9.7374e-02,
         1.9230e-05],
        [5.2515e-05, 6.4307e-05, 1.2886e-04,  ..., 6.6807e-06, 2.4549e-07,
         2.9373e-01]]), 'X': tensor([[  6.3589,   0.8239,  -8.4230],
        [  6.3303,   1.6145,  -8.4083],
        [  5.8201,   3.5841,  -8.0609],
        [  5.9655,   3.2172,  -7.5953],
        [  5.8617,   3.4853,  -9.8080],
        [  6.1929,   2.4501,  -6.9352],
        [  2.5813,  -4.5471,   5.3597],
        [ 10.1503,   8.5414,  -7.2681],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.2939,  -2.4784,  -4.5840],
        [ -0.8885,   7.9160,   0.0571],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.3589,   0.8239,  -8.4230],
        [  6.3515,   0.6418,  -6.5823],
        [  0.8572,  14.0501,  -8.3364],
        [ -5.3382,   6.6533,   7.8054],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.2247,  -0.3150,  -9.3657],
        [ -3.0004,   7.6607,  -7.2640],
        [  9.4554,   9.4245,   9.8800],
        [ 10.9916,   7.2133,  -0.9179],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -0.1626, -10.9796,   6.4629],
        [ 10.9394,   7.3069,  -8.9365],
        [  0.9809,   7.6642,  -1.4217],
        [  8.0877,  -8.5297,  -1.1050],
        [ 12.4996,   2.6335,  -6.3166],
        [  0.5250, -11.0306,   5.5358],
        [  2.7121,   6.9743,  -8.4202],
        [  6.1984,  -0.4327,   1.5438],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.8631,  -4.3728,  -1.1746],
        [  4.6231,   5.4556,   8.4973],
        [  9.2811,  -7.4336,   5.1771],
        [ -7.9667,   4.0814,  -0.6397],
        [  1.0000,   7.6591,  -4.5011],
        [  5.0617,  12.7957,   8.3544],
        [ -4.6942,  -9.4299,  -9.1846],
        [  1.1006,   7.6318,  -2.4877],
        [ -0.0254, -10.9932,   3.2200],
        [  6.9433,  -9.3362,  -0.3421],
        [ -5.6651,   6.4380,   1.3660],
        [ -8.6675,  -4.8569,  -9.1769],
        [  9.7328,   9.0902,  -0.8917],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.2592,   3.9425,   2.0383],
        [  2.1824,  -4.6647,   1.1985],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 11.5198,  -4.0923,   5.5088],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -5.9872,   6.2037,  -9.4477],
        [ -1.3178, -10.7939,   0.4236],
        [  2.5620,   7.0549,   8.4825],
        [  8.2154,  -8.4262,   8.0171],
        [ -6.3306,   5.9272,   5.6935],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.9178,   2.1936,   7.5388],
        [  6.0073,   3.0986,   6.7937],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.4501,  -1.7133,   2.2980],
        [  5.6258,  -1.9404,  -6.0715],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.0961,   1.6529,   3.1553],
        [  6.0788,  -0.8710,  -6.8719],
        [  9.3560,   9.5391,   9.4943],
        [  1.6954,  -4.7616,   9.8198],
        [  8.3791,  10.5409,   5.5590],
        [  0.6757, -11.0361,  -1.3047],
        [  8.8091,  -7.9037,   0.3631],
        [  5.4429,  -2.2529,   9.7306],
        [ 11.5568,   6.0602,   9.3244],
        [  6.9956,  11.6590,  -7.6876],
        [ -2.0721,   7.8432,   6.6380],
        [  1.7500,   7.4187,   4.4237],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.6970,   6.9826,   9.7350],
        [ -9.1624,  -3.4286,   2.0647],
        [  6.7792,  11.8084,   1.6251],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.1480,  -4.3165,  -7.8002],
        [ -2.9497, -10.3028,  -7.3479],
        [  5.6022,  -1.9833,   8.3081],
        [  1.8320,   7.3871,   5.1452],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.1793,   4.7461,   0.3899],
        [ -9.4157,   0.0821,  -5.0025],
        [ 11.7726,  -3.4876,  -6.7402],
        [  5.8201,   3.5841,  -8.0609],
        [  2.9260,  -4.4162,   8.3808],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.4336,  -2.2676,   5.6933],
        [  2.8602,   6.8905,   5.8779],
        [  9.1409,  -7.5789,   1.5814],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.9655,   3.2172,  -7.5953],
        [ -0.0254, -10.9932,   3.2200],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.2728, -11.0372,   9.8121],
        [ 11.1469,   6.9237,   1.6160],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.8572,  14.0501,  -8.3364]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0049, 0.0350,  ..., 0.6237, 1.1288, 0.1459],
        [0.0049, 0.0000, 0.0318,  ..., 0.6279, 1.1323, 0.1435],
        [0.0350, 0.0318, 0.0000,  ..., 0.6321, 1.1273, 0.1602],
        ...,
        [0.6237, 0.6279, 0.6321,  ..., 0.0000, 0.5549, 0.7530],
        [1.1288, 1.1323, 1.1273,  ..., 0.5549, 0.0000, 1.2704],
        [0.1459, 0.1435, 0.1602,  ..., 0.7530, 1.2704, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.41202986,  6.53290663,  6.83515201,  6.77777606,  6.81961996,
        6.65991877,  5.2287058 , 13.26590071,  5.19986797,  5.84533345,
        7.96575782, 12.07225469,  6.41202986,  6.383897  , 14.07623266,
        8.53014915,  7.8788965 , 12.52875793,  6.23262289,  8.22728399,
       13.35013374, 13.1471313 , 10.61047596, 10.98076873, 13.15525803,
        7.72668467, 11.75438006, 12.77402351, 11.04313641,  7.48311002,
        6.2134961 , 12.1494023 ,  9.88310614,  7.15100303, 11.89105259,
        8.95133961,  7.72415048, 13.76047999, 10.53368644,  7.71075226,
       10.99326577, 11.63502679,  8.57560275,  9.93553424, 13.31764181,
       10.05558893, 11.57249984, 12.87752052,  5.14999799,  5.5748624 ,
        9.58072286, 12.22503754,  9.97964013,  8.62162372, 10.87409005,
        7.50564223, 11.76830981,  8.67227386,  9.94285511,  9.18358562,
        6.7593917 , 13.77222752,  9.60412735,  5.9510489 ,  9.13507321,
        9.24501829,  6.14086582, 13.3614598 ,  5.05444288, 13.46553623,
       11.05672136, 11.83509372,  5.89074104, 13.0493368 , 13.59673406,
        8.11227668,  7.62232307, 12.59840193,  7.48538905,  9.78285424,
       13.61600536, 11.97238624,  9.14973319,  6.25586297,  5.34250225,
       10.7167285 ,  5.94293185,  7.61088125,  9.17905491,  7.02496223,
        9.41605684, 12.27835996,  6.83515201,  5.29755487, 12.67576792,
        5.88782913,  7.46053281, 11.87412439, 10.39942497,  6.77777606,
       10.99326577,  5.84450127, 11.11038551, 13.12218013, 12.41012046,
       14.07623266])}
{'P': tensor([[2.1397e-02, 1.5295e-01, 1.4760e-01,  ..., 1.6245e-04, 2.8382e-04,
         1.5277e-05],
        [9.5108e-04, 2.5968e-02, 2.4026e-01,  ..., 1.0947e-04, 3.8270e-04,
         8.4197e-06],
        [1.0905e-03, 2.9239e-03, 3.4018e-02,  ..., 1.3790e-04, 5.4006e-04,
         1.0402e-05],
        ...,
        [1.9448e-02, 1.3590e-02, 1.3895e-02,  ..., 1.8742e-01, 1.6843e-03,
         1.0109e-02],
        [4.8408e-06, 8.8379e-06, 1.0176e-05,  ..., 3.1329e-06, 1.8515e-02,
         2.5959e-07],
        [1.5915e-04, 1.6209e-04, 1.7391e-04,  ..., 1.9228e-04, 1.5715e-03,
         3.0489e-02]]), 'X': tensor([[  6.3604,   0.9042,  -6.5901],
        [  6.1929,   2.4501,  -6.9352],
        [  6.1372,   2.6726,  -6.7885],
        [  6.3031,   1.8477,  -7.1706],
        [  6.0047,   3.1062,  -6.7914],
        [  5.9655,   3.2172,  -7.5953],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4697,  -0.6157,  -6.8472],
        [  9.9217,  -6.7006,  -5.6728],
        [  0.9392,  -4.8143,  -3.4649],
        [  0.9445,   7.6736,   0.5095],
        [  1.6232,  -4.7717,  -2.5655],
        [ -4.3359,  -9.6431,  -9.0616],
        [  6.0955,   2.8210,  -4.3321],
        [ -9.1178,   1.5784,   2.1410],
        [  3.1756, -10.8247,   6.8882],
        [ 12.3357,   3.6001,   0.7903],
        [ -2.8162,   7.7059,  -9.6885],
        [ -2.2188,   7.8218,   8.6064],
        [  4.4633,   5.6287,  -0.7754],
        [ 11.0891,  -4.9558,  -2.3178],
        [  7.1697,  11.5343,  -0.7276],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.2435,   4.0076,   8.1257],
        [ -3.2736, -10.1703,   3.8712],
        [ 12.2311,  -2.0445,   1.3545],
        [ -7.8849,  -6.3115,  -4.7479],
        [  2.5010,  -4.5736,  -0.8989],
        [ -1.1885, -10.8211,   0.0865],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -8.6229,   2.9068,   2.4429],
        [ -6.4464,   5.8272,   7.5038],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.0275,  -5.0666,   8.4154],
        [ -2.2508, -10.5479,  -2.7235],
        [  9.4925,   9.3811,   8.9842],
        [  8.7131,  10.2216,   7.7915],
        [ -1.8367,   7.8718,  -8.6523],
        [  4.9152,  12.8653,  -1.4459],
        [ 11.4743,   6.2478,  -2.1100],
        [ 12.4877,  -0.7222,   8.8347],
        [ -4.4503,   7.1378,   3.3751],
        [  4.4915,   5.5990,   7.0405],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.0031,  12.2958,  -6.5992],
        [ -2.6485, -10.4151,   0.8766],
        [ -6.3489,   5.9117,   6.5161],
        [  6.1929,   2.4501,  -6.9352],
        [  3.4401,  13.4540,   4.6576],
        [  4.6063,  13.0051,  -3.1456],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 10.6283,  -5.7253,   7.4878],
        [  3.6703,   6.3443,  -5.4401],
        [ -1.6220,   7.8917,   0.0315],
        [ -2.0840,   7.8415,  -7.8657],
        [  1.1994,   7.6035,   2.4295],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.2675,  12.1379,  -8.9720],
        [  3.9663,   6.1018,  -5.4104],
        [ -6.1914,  -8.2997,   8.5380],
        [ -5.9940,   6.1985,   5.9566],
        [ 12.5832,   1.7543,  -7.6134],
        [ -7.0948,   5.1940,   9.8171],
        [  2.7558,  -4.4844,   0.0636],
        [  6.3549,   0.7096,   6.9198],
        [ 11.6544,   5.8273,  -3.1375],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.9314,  -9.3435,  -1.2729],
        [  4.0457,  -3.7737,  -0.4704],
        [  0.4996,   7.7738,  -4.4711],
        [  7.1697,  11.5343,  -0.7276],
        [  2.0899,  -4.6870,  -7.4260],
        [ -9.4296,  -1.9458,   4.0833],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 12.6038,   0.7665,  -3.4439],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.3589,   0.8239,  -8.4230],
        [  2.0729,   7.2878,   7.0880],
        [ -2.6069, -10.4298,  -6.1995],
        [  2.1440, -10.9811,   8.5281],
        [  2.7685, -10.8982,  -6.8928],
        [ -8.9942,  -3.9990,   6.0448],
        [ -6.0195,   6.1789,  -2.9496],
        [ -3.4488, -10.0934,   2.5195],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.3076,   0.7852,  -6.7010],
        [ 11.0891,  -4.9558,  -2.3178],
        [  2.1120,  -4.6818,   8.6144],
        [ -7.8280,   4.2848,  -7.1955],
        [  4.5430,   5.5438,  -3.8669],
        [ 10.9875,   7.2207,  -8.8549],
        [ -8.3539,  -5.5111,   0.3373],
        [  3.7121,   6.3117,   0.6738],
        [  9.9217,  -6.7006,  -5.6728],
        [  4.4290,  -3.4596,   1.4181],
        [  3.9663,   6.1018,  -5.4104],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.9394,   7.3069,  -8.9365],
        [  9.6587,   9.1817,  -0.8212],
        [  0.1047, -11.0046,   0.5188],
        [ -1.1885, -10.8211,   0.0865],
        [ -8.7544,   2.6096,   3.3146],
        [  1.4150,  13.9647,  -4.0915],
        [  4.3459,  -3.5327,   0.3667],
        [  6.6319,  11.9066,   3.5165],
        [ 11.6165,  -3.8715,   6.4594]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0195, 0.0171,  ..., 0.3887, 0.8744, 0.9949],
        [0.0195, 0.0000, 0.0064,  ..., 0.4063, 0.8940, 1.0128],
        [0.0171, 0.0064, 0.0000,  ..., 0.4009, 0.8908, 1.0076],
        ...,
        [0.3887, 0.4063, 0.4009,  ..., 0.0000, 0.5624, 0.6071],
        [0.8744, 0.8940, 0.8908,  ..., 0.5624, 0.0000, 0.4224],
        [0.9949, 1.0128, 1.0076,  ..., 0.6071, 0.4224, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.42439344,  6.65991877,  6.69389431,  6.56833864,  6.76056625,
        6.77777606,  8.04483495,  9.48969978, 11.97238624,  4.90505566,
        7.73151309,  5.04028538, 10.57302546,  6.7166297 ,  9.25336371,
       11.28093507, 12.85032836,  8.20436764,  8.1303945 ,  7.18354885,
       12.1460931 , 13.58101794,  5.43943791, 12.88270018, 10.68416401,
       12.40074901, 10.09979747,  5.21280156, 10.88618114, 10.37698435,
        9.09963894,  8.68979052,  8.41148722, 12.13568598, 10.78533838,
       13.34587027, 13.43127131,  8.08320235, 13.77222752, 13.06498554,
       12.5085995 ,  8.41148722,  7.17789666, 12.32408516, 13.6829843 ,
       10.74655346,  8.67502348,  6.65991877, 13.88683436, 13.79676654,
       12.32545637, 12.07225469,  7.32950385,  8.05668609,  8.1137452 ,
        7.69752262, 11.66453059, 13.66051884,  7.27759233, 10.35464916,
        8.62260327, 12.70489563,  8.79284466,  5.26342203,  6.39438044,
       13.03002338,  9.88571713, 11.63383221,  5.53255882,  7.78980299,
       13.58101794,  5.13182506,  9.6282763 ,  9.88310614, 12.62711252,
        5.83560009,  6.41202986,  7.57685908, 10.7506423 , 11.18839169,
       11.24434374,  9.84314712,  8.62631801, 10.66631784,  5.79391825,
        9.34061938, 12.1460931 ,  5.13616224,  8.92395381,  7.16747878,
       13.14777433, 10.00794468,  7.32233733, 11.97238624,  5.62005629,
        7.27759233,  6.00663314, 13.15525803, 13.32645552, 11.00508681,
       10.88618114,  9.13507321, 14.03618649,  5.60064319, 13.62896593,
       12.24467041])}
{'P': tensor([[2.8034e-02, 2.1500e-01, 1.9755e-01,  ..., 1.4125e-03, 6.9141e-06,
         2.6111e-05],
        [2.7845e-03, 7.6264e-02, 1.0733e-01,  ..., 4.8901e-03, 1.5552e-05,
         3.7974e-05],
        [2.9091e-03, 3.6256e-02, 8.2080e-02,  ..., 4.0358e-03, 1.7724e-05,
         3.7844e-05],
        ...,
        [1.1646e-04, 1.9718e-04, 1.7810e-04,  ..., 4.3581e-02, 1.8565e-06,
         5.1544e-06],
        [1.0301e-04, 5.0494e-05, 5.2859e-05,  ..., 1.2378e-06, 1.5236e-02,
         8.6642e-04],
        [3.9215e-03, 2.3183e-03, 1.9958e-03,  ..., 1.7320e-04, 2.8656e-06,
         2.5099e-02]]), 'X': tensor([[  6.3606,   0.9204,   3.2525],
        [  6.2022,   2.4092,   3.0349],
        [  6.1752,   2.5242,   3.5874],
        [  6.1737,   2.5305,   3.9049],
        [  6.2555,   2.1474,   4.1685],
        [  6.2905,   1.9366,   2.1169],
        [  2.1440, -10.9811,   8.5281],
        [ -7.5720,   4.6310,   2.1717],
        [  8.2422,  -8.4041,  -1.4962],
        [  6.6319,  11.9066,   3.5165],
        [ -5.5525,   6.5146,  -4.4984],
        [  5.1787,  -2.6383,  -4.2533],
        [ 11.1469,   6.9237,   1.6160],
        [  4.2096,   5.8815,   1.9230],
        [ 11.3691,   6.4760,   1.0787],
        [ -7.9667,   4.0814,  -0.6397],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.2974,   0.8371,   2.4852],
        [ 12.2589,  -1.9312,   9.8657],
        [  2.0039,   7.3173,  -1.5885],
        [ -2.6485, -10.4151,   0.8766],
        [  5.8902,  -1.3887,   4.0388],
        [  4.6063,  13.0051,  -3.1456],
        [  6.3551,   0.7149,   1.3344],
        [ 11.0438,   7.1180,  -5.6991],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.6232,  -4.7717,  -2.5655],
        [ -3.6381,   7.4683,   7.3544],
        [  6.3246,   1.6696,  -0.1544],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -4.0868,   7.2981,   8.1874],
        [  4.5767,   5.5071,   3.9069],
        [  2.3376,  -4.6231,   9.8436],
        [ 11.3118,   6.5957,  -4.5667],
        [  0.1931,  -4.7491,   7.4385],
        [  7.3592,  11.3937,   6.2407],
        [  7.0664,  -9.2590,   8.0879],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1756, -10.8247,   6.8882],
        [ -3.2053, -10.1993,  -3.8736],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.4932,   7.0903,  -5.4214],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.5175,  14.0910,   4.6540],
        [  4.4290,  -3.4596,   1.4181],
        [ -5.9940,   6.1985,   5.9566],
        [  3.1892, -10.8220,  -6.8409],
        [ -1.6275, -10.7221,   3.2376],
        [  9.3560,   9.5391,   9.4943],
        [ -5.3382,   6.6533,   7.8054],
        [ -1.6334, -10.7207,   7.1397],
        [  1.9204,   7.3518,   8.4781],
        [  1.1644, -11.0394,   7.8511],
        [  4.7544,  -3.1425,  -1.5315],
        [ 11.6019,   5.9540,  -5.1715],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.0467,  -9.8369,  -6.5816],
        [ -8.3539,  -5.5111,   0.3373],
        [ 11.3691,   6.4760,   1.0787],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.1905,   3.7253,   4.1175],
        [  2.2182,  13.8007,   5.1889],
        [ -9.4505,  -1.7081,  -0.7039],
        [  4.5491,  -3.3485,   6.8412],
        [  3.4140,  -4.1803,  -8.8224],
        [  9.2979,   9.6048,   7.3138],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.6451,  -4.5250,  -0.5565],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.6651,   6.4380,   1.3660],
        [ 12.5123,  -0.5339,  -1.0728],
        [  9.2692,   9.6370,   7.0746],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -4.8665,  -9.3204,  -5.2564],
        [  5.4732,  -2.2041,  -4.3830],
        [  3.9913,  -3.8139,  -6.8311],
        [ -0.5739,   7.9056,   0.6382],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.1531,  -5.8748,   2.6934],
        [ -7.5401,   4.6718,   9.0033],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.8835,  -4.3178,   9.3080],
        [ -2.0229, -10.6163,   8.6930],
        [ 12.1588,   4.3364,  -8.0867],
        [ -8.0766,  -6.0046,  -3.1873],
        [  7.5309,  11.2617,   7.2656],
        [ -0.8569,   7.9156,  -6.5557],
        [ 11.2517,  -4.6495,   2.9800],
        [  3.4140,  -4.1803,  -8.8224],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.4639, -10.9431,  -7.3718],
        [  3.2701,   6.6338,   9.9209],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 12.4901,  -0.7052,   1.7315],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.8890,  -4.4316,   1.1600],
        [  9.1409,  -7.5789,   1.5814],
        [  1.1350,  14.0104,   0.6321],
        [  3.6703,   6.3443,  -5.4401],
        [ -3.1204, -10.2345,   9.2288],
        [  7.5064,  -8.9651,  -0.0994]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0162, 0.0358,  ..., 0.5907, 0.8839, 0.4373],
        [0.0162, 0.0000, 0.0433,  ..., 0.5788, 0.8767, 0.4210],
        [0.0358, 0.0433, 0.0000,  ..., 0.6215, 0.8580, 0.4489],
        ...,
        [0.5907, 0.5788, 0.6215,  ..., 0.0000, 1.1866, 0.3917],
        [0.8839, 0.8767, 0.8580,  ..., 1.1866, 0.0000, 0.7952],
        [0.4373, 0.4210, 0.4489,  ..., 0.3917, 0.7952, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.42689705,  6.65368551,  6.67121748,  6.67218687,  6.61386436,
        6.58184474, 11.18839169,  8.87587457, 11.77124373, 13.62896593,
        8.55982105,  5.81199422, 13.12218013,  7.23276582, 13.08414689,
        8.95133961,  9.0368458 ,  9.3349858 , 12.41012046,  7.58668144,
       10.74655346,  6.05165248, 13.79676654,  6.39520397, 13.13888841,
       12.17451685,  5.04028538,  8.30729352,  6.54127431, 12.32545637,
        9.71392179, 10.35485331,  8.36445758,  7.16058587,  5.1805345 ,
       13.09426142,  4.75303127, 13.56367857, 11.64746882,  5.13182506,
       11.28093507, 10.6910768 ,  9.79039371,  6.12886024,  7.5158489 ,
        5.16077764,  6.14086582, 14.10045797,  5.62005629,  8.62260327,
       11.28215429, 10.84493732, 13.3614598 ,  8.53014915, 10.84437705,
        7.5984809 , 11.10066126,  5.69913817, 13.04051744,  5.5671389 ,
       11.54672755, 10.00794468, 13.08414689, 13.95361098,  8.99791316,
       13.97779839,  9.60359158,  5.64865173,  5.39723241, 13.36800073,
        8.35856487,  5.24137909,  8.24808832,  8.57560275, 12.52372629,
       13.37121923,  8.27853076,  9.47369628, 10.51437947,  5.90035324,
        5.5205189 ,  7.92644746,  7.78980299, 10.04916678,  8.87007495,
       10.67322527,  9.87720102, 10.80728788, 12.9089559 , 10.06407839,
       13.54774355,  7.96181288, 12.17451685,  5.39723241,  5.18277858,
       11.21703654,  7.39599986,  9.86089579, 12.50996828, 10.412299  ,
        5.29009836, 11.87412439, 14.05633441,  7.32950385, 10.69963954,
       11.69264243])}
{'P': tensor([[2.0149e-02, 1.9997e-01, 1.7130e-01,  ..., 5.4663e-03, 2.0149e-02,
         2.4704e-05],
        [1.7069e-03, 2.7680e-02, 1.1941e-01,  ..., 1.0005e-02, 1.7069e-03,
         2.6041e-05],
        [1.6441e-03, 4.3667e-03, 3.3345e-02,  ..., 1.2388e-02, 1.6441e-03,
         2.4677e-05],
        ...,
        [1.4237e-04, 1.6222e-04, 1.7544e-04,  ..., 4.8200e-02, 1.4237e-04,
         5.4286e-06],
        [2.0149e-02, 1.9997e-01, 1.7130e-01,  ..., 5.4663e-03, 2.0149e-02,
         2.4704e-05],
        [4.1017e-03, 3.7402e-03, 2.8920e-03,  ..., 1.4002e-04, 4.1017e-03,
         2.2910e-02]]), 'X': tensor([[  6.3587,   1.1596,  -3.6629],
        [  6.3205,   1.7068,  -3.8619],
        [  6.2576,   2.1359,  -3.3398],
        [  6.2038,   2.4020,  -3.1692],
        [  5.9912,   3.1453,  -3.9989],
        [  6.0955,   2.8210,  -4.3321],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.2435,   7.8191,  -6.4709],
        [  6.3031,   1.8477,  -7.1706],
        [  5.0506,   4.9277,  -9.5094],
        [  5.6260,  -1.9401,   3.0259],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.3373,  13.7721,  -0.2442],
        [ 11.1469,   6.9237,   1.6160],
        [  0.1532,  -4.7422,  -7.0145],
        [ -4.6942,  -9.4299,  -9.1846],
        [  7.0515,  11.6194,   8.4289],
        [  5.9587,  12.3216,   9.1695],
        [ -3.3999, -10.1152,  -1.2088],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.0140,  -1.0661,  -2.5040],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.8154,   2.4611,   2.5198],
        [  3.3943,   6.5483,   1.0790],
        [ -8.0611,  -6.0302,   4.6962],
        [ -2.6069, -10.4298,  -6.1995],
        [ -6.5295,  -7.9785,  -7.4429],
        [  5.1472,  -2.6801,   8.1194],
        [  5.5626,  -2.0536,   8.4433],
        [ -2.7572,   7.7194,   5.9580],
        [  9.8082,  -6.8396,   4.9615],
        [ -6.8466,  -7.6488,   1.9648],
        [  9.3560,   9.5391,   9.4943],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.3211,   1.7017,  -0.2164],
        [ 12.6035,   0.7494,   7.1536],
        [  6.2902,   0.0451,   3.7116],
        [ -5.6423,   6.4537,  -1.2091],
        [ -3.4488, -10.0934,   2.5195],
        [  1.3339,   7.5627,  -0.9034],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.1140,  -3.7220,  -0.1273],
        [ -9.4505,  -1.7081,  -0.7039],
        [  0.2732,  -4.7619,  -1.1577],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.9548,  -1.2273,  -2.6520],
        [ -8.8400,   2.3991,   1.3457],
        [  2.5620,   7.0549,   8.4825],
        [ 10.9916,   7.2133,  -0.9179],
        [ -1.9714,   7.8563,  -1.8084],
        [ -7.3470,  -7.0615,   5.0066],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.1824,  -4.6647,   1.1985],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.0467,  -9.8369,  -6.5816],
        [ -7.4807,   4.7463,   2.0616],
        [  4.1348,   5.9514,  -0.8856],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.6451,  -4.5250,  -0.5565],
        [ -5.3328,  -8.9992,   7.3440],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.9026,   3.3837,  -0.6573],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.1372,   2.6726,  -6.7885],
        [  6.3266,   0.3292,   9.3502],
        [  1.1644, -11.0394,   7.8511],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.3211,   1.7017,  -0.2164],
        [  0.2480, -11.0153,  -3.0183],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.1120,  -4.6818,   8.6144],
        [  0.1497,   7.8334,  -5.4412],
        [ -4.8665,  -9.3204,  -5.2564],
        [  5.1292,   4.8183,  -5.7997],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.4401,  13.4540,   4.6576],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.6078, -10.9229,   6.2869],
        [ 10.0619,   8.6627,  -9.2884],
        [ -1.0482, -10.8487,  -5.9066],
        [  9.6614,   9.1784,  -1.2087],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 10.2470,   8.4054,  -0.1177],
        [ 12.0582,  -2.6640,   0.7132],
        [  1.7000,  -4.7609,   2.7935],
        [ -4.8665,  -9.3204,  -5.2564],
        [  2.9470,  13.6085,   2.9809],
        [ -3.4986,   7.5152,   9.0712],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.1749,  -3.3803,   6.4629],
        [ -4.6942,  -9.4299,  -9.1846],
        [  4.0642,  -3.7599,  -3.4565],
        [  9.0129,  -7.7072,   9.1400],
        [  6.3515,   0.6418,  -6.5823],
        [  4.5057,  -3.3894,  -0.8160],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.2209,  -4.7092,   2.6117],
        [  1.5662,  13.9375,   4.5142],
        [ -9.3444,   0.5821,   6.9101],
        [  1.6422,   7.4586,  -6.8056],
        [  6.3587,   1.1596,  -3.6629],
        [  9.0424,  -7.6779,  -7.5286]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0105, 0.0153,  ..., 0.2419, 0.0000, 0.3357],
        [0.0105, 0.0000, 0.0253,  ..., 0.2345, 0.0105, 0.3259],
        [0.0153, 0.0253, 0.0000,  ..., 0.2480, 0.0153, 0.3466],
        ...,
        [0.2419, 0.2345, 0.2480,  ..., 0.0000, 0.2419, 0.1463],
        [0.0000, 0.0105, 0.0153,  ..., 0.2419, 0.0000, 0.3357],
        [0.3357, 0.3259, 0.3466,  ..., 0.1463, 0.3357, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.46356806,  6.54693987,  6.61211224,  6.65259775,  6.76661595,
        6.7166297 ,  6.27905292,  7.822854  ,  6.56833864,  7.05627461,
        5.95111804, 12.1460931 ,  9.73579738, 13.96905756, 13.12218013,
        4.74468406, 10.53368644, 13.59170809, 13.68672855, 10.67132524,
        4.76969364,  6.10774544,  7.26074925,  9.15253344,  7.37576154,
       10.06703358, 10.7506423 , 10.30972193,  5.80312949,  5.92952538,
        8.19703512, 11.95743558, 10.26545792, 13.3614598 , 13.18278606,
        6.54615887, 12.62576026,  6.29034958,  8.5724042 , 10.66631784,
        7.67939546,  5.04028538,  5.54776934,  9.60359158,  4.76969364,
       12.19805394,  6.07992623,  9.15976746,  7.50564223, 13.1471313 ,
        8.09984231, 10.19036081,  8.56889738,  5.14999799, 12.11735334,
       11.54672755,  8.85939626,  7.24677664,  5.27178363,  5.24137909,
       10.4606265 , 13.03002338,  6.80370254, 10.30972193,  4.91652556,
        6.69389431,  6.33517349, 11.10066126,  9.58072286,  6.54615887,
       11.018084  ,  9.51074763, 10.8207297 ,  5.13616224,  7.83486943,
       10.51437947,  7.03733775,  5.19065439, 13.88683436,  8.14433542,
       11.22993218, 13.27717982, 10.8992524 , 13.32613445, 10.5575601 ,
        9.85330646, 13.25336001, 12.3489324 ,  5.05534609, 10.51437947,
       13.92390689,  8.28968067, 13.09426142,  9.77777346, 10.53368644,
        5.5366698 , 11.85890664,  6.383897  ,  5.63824971,  8.04718433,
       12.16901615, 14.02526505,  9.36256096,  7.63726134,  6.46356806,
       11.86239961])}
{'P': tensor([[5.5535e-02, 2.4047e-01, 1.1595e-01,  ..., 1.1327e-04, 2.0082e-05,
         8.0184e-05],
        [9.1720e-04, 4.6870e-02, 1.4194e-03,  ..., 4.8351e-05, 1.7528e-05,
         1.2495e-04],
        [1.3450e-02, 2.7672e-01, 8.2991e-02,  ..., 9.7000e-05, 2.7821e-05,
         8.4331e-05],
        ...,
        [1.0491e-03, 1.0923e-03, 6.3684e-04,  ..., 1.2447e-01, 2.4331e-05,
         2.4067e-03],
        [4.9098e-04, 1.3720e-04, 3.4438e-04,  ..., 3.6586e-04, 8.6350e-02,
         2.7098e-06],
        [1.2182e-05, 2.1347e-05, 9.2871e-06,  ..., 1.2351e-04, 1.9629e-05,
         8.9643e-02]]), 'X': tensor([[  6.3583,   1.1729,   8.7835],
        [  5.6248,   3.9984,   8.4933],
        [  6.3119,   1.7797,   9.9186],
        [  6.0073,   3.0986,   6.7937],
        [  4.6231,   5.4556,   8.4973],
        [  4.6357,   5.4415,   7.9067],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.0608,   3.9362,  -5.9567],
        [  1.7232,  -4.7574,   3.1270],
        [  1.6308,  13.9254,  -2.0984],
        [  7.9816,  10.8938,  -0.5378],
        [  0.2435,   7.8191,  -6.4709],
        [ -5.1369,   6.7755,   8.3836],
        [ -7.0948,   5.1940,   9.8171],
        [ -1.1885, -10.8211,   0.0865],
        [  0.9809,  -4.8145,   6.8706],
        [  3.2226,  -4.2802,  -4.0428],
        [ -7.3470,  -7.0615,   5.0066],
        [  9.9750,  -6.6338,  -1.2440],
        [  5.4426, -10.1181,  -5.1268],
        [ -1.5875,   7.8944,  -9.4286],
        [ 10.3251,   8.2928,  -3.0006],
        [  6.1737,   2.5305,   3.9049],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.3542,   0.6934,  -3.3840],
        [  4.4290,  -3.4596,   1.4181],
        [  5.1793,   4.7461,   0.3899],
        [ -5.5525,   6.5146,  -4.4984],
        [  0.2480, -11.0153,  -3.0183],
        [  3.4401,  13.4540,   4.6576],
        [  1.6308,  13.9254,  -2.0984],
        [ -1.6906,   7.8860,   7.7142],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.1929,   2.4501,  -6.9352],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.2974,   0.8371,   2.4852],
        [  6.1222,  -0.7261,   8.8204],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.1989,  -9.7591,   0.8289],
        [ 12.5717,   1.9266,   7.4903],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.3200,   7.1756,   1.1938],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.3444,   0.5821,   6.9101],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.2311,  -2.0445,   1.3545],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.5158,  -2.1337,   3.5246],
        [  3.4401,  13.4540,   4.6576],
        [  4.0591,   6.0202,  -1.6339],
        [ 10.1798,   8.5002,  -5.8893],
        [  2.8994,   6.8675,  -9.5987],
        [  3.2226,  -4.2802,  -4.0428],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 11.1469,   6.9237,   1.6160],
        [ -0.4341, -10.9473,   7.4509],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.0965,  -5.9712,   4.3466],
        [ -8.8154,   2.4611,   2.5198],
        [  3.3943,   6.5483,   1.0790],
        [  5.6093,  -1.9705,   6.4807],
        [ -7.4808,   4.7462,   2.0266],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.4753,  -0.8012,  -5.2050],
        [ 11.4235,   6.3594,   5.1663],
        [  0.2732,  -4.7619,  -1.1577],
        [ 10.9916,   7.2133,  -0.9179],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.4297,  -5.3637,  -9.3962],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.1886,  -2.6250,   6.4677],
        [  8.0381,  -8.5690,  -8.9523],
        [ -5.0550,   6.8231,  -9.2502],
        [  9.4554,   9.4245,   9.8800],
        [  6.3031,   1.8477,  -7.1706],
        [ -9.4203,  -2.0340,   2.5856],
        [  3.2159, -10.8166,  -1.0406],
        [ -9.3419,   0.5972,  -1.2841],
        [  8.6271,  -8.0715,  -7.4813],
        [  3.9022,  13.2906,   5.5788],
        [  6.2022,   2.4092,   3.0349],
        [  1.8447,  -4.7372,   8.1235],
        [  3.7992,  -3.9474,  -8.9019],
        [  2.7685, -10.8982,  -6.8928],
        [  2.8474,   6.8979,   6.6457],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.3535,  -5.5118,   9.8399],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.9260,  -4.4162,   8.3808],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.2373,  -4.6506,  -1.7334],
        [  0.7145,  -4.8071,  -2.3766],
        [ 10.0567,   8.6697,   4.6329],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.5600, -11.0321,  -0.8660],
        [ -8.5506,  -5.1154,   6.3423],
        [ 10.6283,  -5.7253,   7.4878],
        [  8.8091,  -7.9037,   0.3631],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.3119,   1.7797,   9.9186],
        [  0.5600, -11.0321,  -0.8660],
        [  5.2911,  -2.4824,  -3.6877],
        [ -5.3328,  -8.9992,   7.3440],
        [ -1.0407,   7.9166,  -6.6021]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.8875e-02, 2.0101e-02,  ..., 7.7588e-01, 8.9364e-01,
         9.8366e-01],
        [4.8875e-02, 0.0000e+00, 4.2540e-02,  ..., 7.7222e-01, 8.4495e-01,
         9.6639e-01],
        [2.0101e-02, 4.2540e-02, 0.0000e+00,  ..., 7.9220e-01, 8.8233e-01,
         9.9615e-01],
        ...,
        [7.7588e-01, 7.7222e-01, 7.9220e-01,  ..., 0.0000e+00, 1.1696e+00,
         3.1914e-01],
        [8.9364e-01, 8.4495e-01, 8.8233e-01,  ..., 1.1696e+00, 3.0339e-04,
         1.1153e+00],
        [9.8366e-01, 9.6639e-01, 9.9615e-01,  ..., 3.1914e-01, 1.1153e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.46560181,  6.90115901,  6.55800267,  6.7593917 ,  7.15100303,
        7.14837514,  7.26074925,  8.97052115,  5.05989547, 14.02058621,
       13.50484539,  7.822854  ,  8.50268322,  8.79284466, 10.88618114,
        4.91336994,  5.35774573, 10.19036081, 11.97949273, 11.4890861 ,
        8.05242591, 13.2430377 ,  6.67218687,  5.18277858,  6.39187285,
        5.62005629,  7.02496223,  8.55982105, 11.018084  , 13.88683436,
       14.02058621,  8.06517143, 10.00794468,  6.65991877,  6.77777606,
        9.3349858 ,  6.16512856,  9.62238828, 11.56146873, 12.71843462,
        9.62958973,  6.19633139,  7.54127425, 11.75618513,  9.36256096,
        9.84314712, 12.40074901, 12.42554808,  8.30125201,  5.91408504,
       13.88683436,  7.26074925, 13.26209545,  7.4544866 ,  5.35774573,
       12.35421461, 13.12218013, 10.95593729,  8.04664886, 10.06023665,
        9.15253344,  7.37576154,  5.94535529,  8.859416  ,  9.41605684,
        9.50913849, 13.0743399 ,  4.76969364, 13.1471313 ,  9.77777346,
        9.99145096, 11.67875855,  5.81481044, 11.74901778,  8.49161154,
       13.35013374,  6.56833864,  9.63742669, 11.2845608 ,  9.36093562,
       11.81423161, 13.85158417,  6.65368551,  5.08372149,  5.47866221,
       11.24434374,  7.46248947, 12.50996828, 10.00802627,  7.98469166,
        5.29755487, 12.40074901,  5.16077764,  4.85995002, 13.2778378 ,
       10.5575601 , 11.04629471,  9.96391045, 12.07225469, 11.83509372,
        5.5513072 ,  6.55800267, 11.04629471,  5.84450127, 10.4606265 ,
        7.98469166])}
{'P': tensor([[1.7097e-02, 1.9000e-01, 1.3818e-01,  ..., 4.3759e-06, 3.1408e-06,
         2.9550e-06],
        [1.0144e-03, 1.8129e-02, 1.7891e-01,  ..., 4.1675e-06, 3.0326e-06,
         3.2229e-06],
        [9.1485e-04, 1.2006e-03, 2.0576e-02,  ..., 4.3155e-06, 3.1993e-06,
         3.9005e-06],
        ...,
        [3.4424e-04, 2.7607e-04, 2.1902e-04,  ..., 2.4455e-02, 3.3339e-03,
         1.2862e-05],
        [4.9628e-04, 4.0850e-04, 3.3496e-04,  ..., 7.6062e-02, 6.9089e-02,
         4.2152e-05],
        [2.8411e-05, 2.3822e-05, 2.0108e-05,  ..., 3.0862e-02, 6.1659e-02,
         2.6861e-02]]), 'X': tensor([[ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0020, 0.0096,  ..., 0.7124, 0.8160, 1.0977],
        [0.0020, 0.0000, 0.0079,  ..., 0.7105, 0.8142, 1.0959],
        [0.0096, 0.0079, 0.0000,  ..., 0.7031, 0.8064, 1.0882],
        ...,
        [0.7124, 0.7105, 0.7031,  ..., 0.0000, 0.1906, 0.4021],
        [0.8160, 0.8142, 0.8064,  ..., 0.1906, 0.0000, 0.2918],
        [1.0977, 1.0959, 1.0882,  ..., 0.4021, 0.2918, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.48486171,  6.56833864,  6.65991877,  6.69389431,  6.77777606,
        6.76056625, 13.02818702,  6.53290663, 11.28093507,  9.94173708,
        8.85939626,  6.41202986,  5.60064319,  5.53255882,  5.70405525,
        6.00732193,  8.23525433, 13.91198191,  9.17595899,  6.81084724,
        7.02326515,  9.13507321, 11.69264243,  9.97220799,  7.32233733,
        9.6282763 ,  6.12886024,  7.54127425,  9.5642993 ,  7.5092834 ,
       10.86693708,  8.05668609,  7.38844703, 11.75807252, 11.04313641,
        5.29009836,  4.74770899,  9.32094055, 10.51340829, 12.7125079 ,
       10.05558893,  7.60463349,  9.23178204,  4.89121805, 13.14777433,
        5.26342203,  5.41348121,  9.88310614,  9.76896904, 12.70489563,
        5.5205189 ,  6.69389431,  9.88310614, 12.53047489, 12.64235359,
       10.88618114, 14.05687815,  8.77913428,  5.51177501, 10.68416401,
        8.99276564, 13.66896739,  7.01475376, 10.5575601 ,  6.19707944,
       13.96905756,  7.72291946,  8.22728399,  6.27715105,  7.58378655,
       13.83139024,  7.50035355, 11.38774149,  4.8763135 ,  9.41605684,
        9.62958973, 11.46176259, 11.73848065,  8.8580671 ,  9.77777346,
       10.61047596,  6.67108662,  5.24137909,  7.70897675,  6.67218687,
       11.07637531, 13.103709  , 11.00508681, 13.32374056,  8.8353674 ,
       12.0495521 , 13.6105595 ,  6.83515201,  7.24326992,  5.94293185,
        5.0713364 , 13.84126197,  8.83725081, 10.8207297 ,  8.09842704,
       12.07225469, 10.794755  ,  7.85919687, 11.05413545, 11.00812996,
       10.26545792])}
{'P': tensor([[1.7961e-02, 2.2240e-01, 1.6615e-01,  ..., 6.9765e-04, 6.0747e-06,
         4.0324e-04],
        [1.2721e-03, 2.2829e-02, 2.5978e-01,  ..., 6.6570e-04, 7.5451e-06,
         5.3409e-04],
        [1.6418e-03, 1.9326e-03, 4.0727e-02,  ..., 7.0252e-04, 1.3219e-05,
         9.7738e-04],
        ...,
        [6.7842e-04, 9.4252e-04, 1.5495e-03,  ..., 1.6757e-02, 2.1243e-07,
         2.8800e-04],
        [1.5595e-04, 1.2685e-04, 9.0940e-05,  ..., 1.7314e-05, 2.9803e-02,
         1.0036e-06],
        [8.2799e-06, 1.0092e-05, 1.4345e-05,  ..., 6.7503e-07, 3.7534e-07,
         1.6523e-02]]), 'X': tensor([[ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0016, 0.0079,  ..., 0.3997, 0.9145, 0.9507],
        [0.0016, 0.0000, 0.0064,  ..., 0.3997, 0.9133, 0.9519],
        [0.0079, 0.0064, 0.0000,  ..., 0.4013, 0.9075, 0.9558],
        ...,
        [0.3997, 0.3997, 0.4013,  ..., 0.0000, 1.2081, 1.2432],
        [0.9145, 0.9133, 0.9075,  ..., 1.2081, 0.0000, 1.4425],
        [0.9507, 0.9519, 0.9558,  ..., 1.2432, 1.4425, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.49538841,  6.5618968 ,  6.67108662,  6.74686071,  6.652742  ,
        6.89245352, 11.9702762 ,  9.3109884 , 11.28895489,  5.23878371,
        7.70897675,  5.18047638, 10.67132524,  8.31625171, 13.92390689,
       12.62576026,  6.383897  ,  8.89301516,  7.8575648 ,  8.97052115,
        7.92200342, 13.04217487, 11.04953767, 11.75618513,  5.07306274,
       11.05317597, 10.07238463,  9.50913849, 13.14690137, 10.19036081,
       13.06957425,  8.20436764,  5.63824971,  6.58184474, 12.889695  ,
        9.78147542, 11.37939695,  6.19707944, 13.6105595 , 13.14777433,
       11.63502679,  5.07306274,  9.60080639, 10.00802627, 13.744387  ,
        4.87083348, 13.99877413, 12.13568598, 13.6922425 ,  7.6374603 ,
        9.17595899,  8.19703512, 13.84126197,  6.27905292,  7.17789666,
       13.26590071,  7.32233733, 10.33958639,  5.92952538,  7.06272256,
        6.383897  ,  5.76165934,  7.48538905, 13.93790235, 10.76154256,
       11.04629471,  7.66377554, 10.53368644,  7.37576154, 12.32408516,
        8.06990483,  6.02588517,  5.64865173,  7.6374603 ,  9.58347456,
       13.62896593,  7.10378919,  6.4020404 ,  9.88310614, 10.31613269,
        4.87083348, 12.1460931 , 10.86693708, 11.0050379 ,  6.652742  ,
        6.76791483,  9.68324553,  8.01009083,  9.12220271,  6.80559669,
        6.76056625, 10.46238054, 10.1979669 ,  5.1224079 , 10.37698435,
        8.11227668, 12.85032836,  9.66977802, 11.27160035,  9.96152788,
       11.28215429,  5.87132718,  8.22728399, 12.77402351, 10.48840476,
       13.54774355])}
{'P': tensor([[2.5572e-02, 2.9037e-01, 1.9040e-01,  ..., 1.7319e-05, 2.3854e-05,
         4.7593e-05],
        [2.6006e-03, 3.4999e-02, 2.5371e-01,  ..., 2.4545e-05, 3.3705e-05,
         5.7857e-05],
        [3.0206e-03, 3.6620e-03, 4.0107e-02,  ..., 3.0798e-05, 4.1510e-05,
         5.1272e-05],
        ...,
        [4.5340e-05, 4.2962e-05, 3.9011e-05,  ..., 4.2266e-02, 4.9291e-03,
         1.8013e-05],
        [4.2181e-05, 3.9698e-05, 3.5116e-05,  ..., 1.2794e-02, 3.3656e-02,
         2.3488e-05],
        [1.9332e-03, 1.8724e-03, 1.6436e-03,  ..., 2.4216e-07, 4.0155e-07,
         1.5247e-02]]), 'X': tensor([[ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.6061e-03, 2.4570e-02,  ..., 1.0632e+00, 9.4943e-01,
         5.5949e-01],
        [5.6061e-03, 0.0000e+00, 1.8967e-02,  ..., 1.0603e+00, 9.4802e-01,
         5.6445e-01],
        [2.4570e-02, 1.8967e-02, 1.0727e-04,  ..., 1.0511e+00, 9.4424e-01,
         5.8166e-01],
        ...,
        [1.0632e+00, 1.0603e+00, 1.0511e+00,  ..., 0.0000e+00, 3.2239e-01,
         1.2368e+00],
        [9.4943e-01, 9.4802e-01, 9.4424e-01,  ..., 3.2239e-01, 0.0000e+00,
         9.8666e-01],
        [5.5949e-01, 5.6445e-01, 5.8166e-01,  ..., 1.2368e+00, 9.8666e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.50409372,  6.53250937,  6.59176018,  6.84796202,  6.74872981,
        6.81084724,  6.99097013,  8.859416  ,  9.11473391,  5.19577146,
        7.17789666, 10.794755  , 10.52587243,  7.98469166,  9.78285424,
       12.17901033,  8.8580671 ,  8.83341608, 11.00726483, 11.27656488,
       10.98076873, 12.21532282, 11.95743558,  6.24614905, 12.1460931 ,
        9.562174  ,  9.68324553, 11.75807252,  6.42689705,  4.83371341,
       11.22142195,  5.20659892, 12.70489563, 13.99877413,  5.64464807,
       10.4977132 , 10.33958639, 10.19036081, 10.75443106,  7.92644746,
       10.24557047, 11.04953767,  5.80312949, 12.0495521 , 11.10066126,
        9.97822473,  9.55720651,  9.09963894,  5.13125513,  5.84533345,
       12.18998195,  8.8353674 , 13.20988996, 12.91143491, 13.15525803,
        5.30158497,  6.48486171, 11.46176259,  5.1805345 , 11.33837396,
        7.60463349, 13.43127131,  5.19577146,  5.10444097,  9.14044984,
       10.91805683,  5.92281142,  5.5366698 ,  7.50917551,  5.0713364 ,
        7.44805901,  4.96231532,  5.30158497, 11.018084  ,  6.54615887,
       12.3489324 ,  8.67227386,  8.11227668,  4.75303127,  5.88782913,
        5.9510489 ,  8.37954853, 10.33712258, 10.09979747,  5.08910491,
        6.24614905, 12.07225469,  5.80860165, 11.23201619,  5.18277858,
       12.4980845 ,  7.14837514, 11.4424816 , 11.85890664, 11.75807252,
        9.04255655, 11.11038551, 14.02532758,  9.18358562,  6.383897  ,
       13.99877413,  7.53063885, 13.05148826,  9.79039371,  9.73579738,
       12.10222204])}
{'P': tensor([[3.0828e-02, 2.2348e-01, 1.9253e-01,  ..., 9.5914e-04, 8.4911e-06,
         2.4010e-04],
        [3.0523e-03, 3.3429e-02, 2.5723e-01,  ..., 1.3109e-03, 8.9530e-06,
         2.1848e-04],
        [1.5530e-03, 1.7665e-03, 5.8594e-02,  ..., 3.7994e-03, 1.3656e-05,
         2.3013e-04],
        ...,
        [1.6092e-04, 1.7993e-04, 2.5288e-04,  ..., 1.7281e-01, 1.1526e-04,
         4.2591e-05],
        [6.4641e-04, 5.5577e-04, 2.4561e-04,  ..., 1.3734e-05, 7.4172e-02,
         1.6354e-03],
        [1.5458e-02, 1.2577e-02, 8.2708e-03,  ..., 7.3486e-05, 5.5489e-05,
         7.4855e-02]]), 'X': tensor([[ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.8967e-02, 2.1882e-02,  ..., 5.7730e-01, 8.5110e-01,
         3.4959e-01],
        [1.8967e-02, 1.0727e-04, 1.9116e-02,  ..., 5.5976e-01, 8.4085e-01,
         3.6812e-01],
        [2.1882e-02, 1.9116e-02, 0.0000e+00,  ..., 5.5912e-01, 8.2923e-01,
         3.5839e-01],
        ...,
        [5.7730e-01, 5.5976e-01, 5.5912e-01,  ..., 0.0000e+00, 4.9986e-01,
         8.8735e-01],
        [8.5110e-01, 8.4085e-01, 8.2923e-01,  ..., 4.9986e-01, 0.0000e+00,
         1.0276e+00],
        [3.4959e-01, 3.6812e-01, 3.5839e-01,  ..., 8.8735e-01, 1.0276e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.53250937,  6.59176018,  6.84796202,  6.74872981,  6.81084724,
        7.02253235, 12.86648771,  6.46560181,  7.72415048,  5.88782913,
       11.11038551, 12.53047489, 12.53285633,  6.99097013, 11.62158554,
       10.35485331,  7.37902766,  9.91668182,  8.92098752, 10.71162681,
        6.41202986, 13.68063098,  6.50409372, 12.00995898, 13.96905756,
       13.40024224,  4.83371341,  7.63726134,  9.3349858 ,  4.74468406,
        5.95297459,  6.38399789,  6.7166297 , 10.86693708,  5.5748624 ,
       13.56367857,  8.23525433,  6.10360073, 11.85743425,  8.09984231,
        8.25828409, 13.18278606,  5.0209005 , 11.28895489,  8.41148722,
        7.50917551,  9.20340031, 11.63383221,  9.60080639,  6.38968357,
       11.56146873, 10.30972193, 13.88683436,  9.68324553, 12.41366754,
       10.4977132 ,  8.12043983,  5.26067019,  5.23878371, 11.42420508,
       10.68416401,  7.53884104,  6.81117802,  6.46560181,  4.91296205,
       13.12299145, 13.37121923,  5.04028538, 12.07225469, 13.35013374,
       12.49317563,  7.02326515,  8.83725081, 11.4890861 ,  8.94395628,
        9.19837792, 11.87665501, 12.17901033, 13.04156075, 10.412299  ,
        5.91222346, 12.52875793,  7.53884104,  6.49538841,  5.5366698 ,
       12.91505111,  9.47369628, 12.95702647,  8.62631801, 13.20988996,
       12.53047489, 14.08549802,  5.5366698 ,  8.7907923 ,  7.03733775,
        8.05668609,  5.80860165, 12.41366754,  7.20482224, 14.03618649,
       12.77402351,  6.55800267, 13.08414689,  7.98469166, 10.78533838,
        5.5748624 ])}
{'P': tensor([[3.5364e-02, 2.2284e-01, 1.9188e-01,  ..., 1.4080e-05, 2.5092e-05,
         1.3454e-04],
        [1.3267e-03, 4.7469e-02, 5.8841e-03,  ..., 2.7535e-05, 5.6104e-05,
         1.4061e-04],
        [1.5799e-03, 1.9358e-01, 4.0100e-02,  ..., 2.3243e-05, 5.4402e-05,
         1.3309e-04],
        ...,
        [6.0076e-05, 3.4740e-05, 4.1549e-05,  ..., 8.0375e-02, 5.7529e-07,
         9.3755e-03],
        [7.7369e-06, 1.5963e-05, 1.5891e-05,  ..., 1.4102e-04, 3.1257e-01,
         2.1671e-05],
        [5.4190e-03, 3.1324e-03, 3.9778e-03,  ..., 2.2931e-04, 1.5231e-05,
         1.0193e-01]]), 'X': tensor([[  6.3303,   1.6145,  -8.4083],
        [  5.8201,   3.5841,  -8.0609],
        [  5.9655,   3.2172,  -7.5953],
        [  5.8617,   3.4853,  -9.8080],
        [  5.0190,   4.9705,  -7.9752],
        [  5.3858,   4.4260,  -9.6769],
        [  0.1382, -11.0073,   1.0026],
        [ 10.1798,   8.5002,  -5.8893],
        [ 10.9394,   7.3069,  -8.9365],
        [  3.8111, -10.6787,  -4.6539],
        [  5.1292,   4.8183,  -5.7997],
        [  1.3403,  -4.8011,  -1.1698],
        [ 12.6059,   0.9597,  -6.3264],
        [ -5.9987,   6.1949,  -6.2114],
        [  5.6928,  12.4710,   3.6001],
        [  3.9663,   6.1018,  -5.4104],
        [ -2.6069, -10.4298,  -6.1995],
        [  5.4426, -10.1181,  -5.1268],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 12.2589,  -1.9312,   9.8657],
        [ -7.2515,   5.0195,   4.6434],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.7916,  -3.9524,  -7.9547],
        [ -2.2188,   7.8218,   8.6064],
        [ -9.1795,  -3.3623,   8.5259],
        [  5.7218,  -9.9934,   6.5764],
        [ -9.4296,  -1.9458,   4.0833],
        [  6.3606,   0.9204,   3.2525],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -1.6275, -10.7221,   3.2376],
        [  6.3119,   1.7797,   9.9186],
        [  8.2422,  -8.4041,  -1.4962],
        [ 11.2517,  -4.6495,   2.9800],
        [  2.7420,  -4.4896,  -3.7139],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.9808,   4.9340,  -3.3715],
        [ -6.6895,   5.6051,   2.7899],
        [  4.7739,  -3.1218,  -4.1188],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.4716,  -0.6673,   7.6068],
        [  0.2435,   7.8191,  -6.4709],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.8363,   2.4086,  -5.6296],
        [  0.9036,   7.6840,   9.9237],
        [ 12.6000,   1.3839,  -3.2992],
        [ 10.0995,  -6.4739,   2.0974],
        [ -7.0825,   5.2073,  -5.0497],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 11.3691,   6.4760,   1.0787],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.0766,  -6.0046,  -3.1873],
        [  3.7992,  -3.9474,  -8.9019],
        [  3.7298,  13.3537,  -9.4653],
        [ -5.6423,   6.4537,  -1.2091],
        [  6.3515,   0.6418,  -6.5823],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.4306,  13.9620,  -0.2075],
        [  6.2026,  12.1774,  -1.5887],
        [  6.1752,   2.5242,   3.5874],
        [ -3.4100,   7.5436,  -7.6980],
        [  0.9809,   7.6642,  -1.4217],
        [  5.2188,  -2.5839,   2.1833],
        [  6.1685,  12.1979,  -6.2473],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.2860,   7.1916,  -6.1011],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.0305,   3.0296,   0.1375],
        [  6.2630,   2.1057,  -5.2136],
        [  6.2022,   2.4092,   3.0349],
        [ -4.9217,   6.8979,  -1.7667],
        [  3.9770, -10.6341,   8.0783],
        [ -9.4739,  -1.2497,   0.5483],
        [ -3.1204, -10.2345,   9.2288],
        [ -8.8154,   2.4611,   2.5198],
        [  6.2905,   1.9366,   2.1169],
        [  0.9392,  -4.8143,  -3.4649],
        [  1.3670,  -4.7990,   3.2603],
        [  8.3791,  10.5409,   5.5590],
        [ -2.6069, -10.4298,  -6.1995],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 12.1588,   4.3364,  -8.0867],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.5650,   2.0132,   4.6272],
        [  0.6142,  14.0802,   0.5407],
        [ -2.0445,   7.8469,  -0.5857],
        [ -7.5720,   4.6310,   2.1717],
        [ -2.6485, -10.4151,   0.8766],
        [  1.8320,   7.3871,   5.1452],
        [  8.7715,  -7.9390,   5.2900],
        [  1.6422,   7.4586,  -6.8056],
        [  0.4119,  14.1019,  -4.1870],
        [  6.1984,  -0.4327,   1.5438],
        [  6.0073,   3.0986,   6.7937],
        [  6.1575,  12.2046,   3.3257],
        [  4.2339,  -3.6268,   6.3125],
        [  6.1741,  -0.5329,   3.1913],
        [  4.9152,  12.8653,  -1.4459],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.8631,  -4.3728,  -1.1746],
        [  1.9312,  13.8649,   8.3563],
        [  0.7685,  -4.8098,  -2.9310]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0318, 0.0284,  ..., 1.0231, 1.2283, 0.2595],
        [0.0318, 0.0000, 0.0143,  ..., 0.9913, 1.2275, 0.2380],
        [0.0284, 0.0143, 0.0000,  ..., 0.9971, 1.2148, 0.2326],
        ...,
        [1.0231, 0.9913, 0.9971,  ..., 0.0000, 1.5256, 0.8465],
        [1.2283, 1.2275, 1.2148,  ..., 1.5256, 0.0000, 1.0544],
        [0.2595, 0.2380, 0.2326,  ..., 0.8465, 1.0544, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.53290663,  6.83515201,  6.77777606,  6.81961996,  7.06372172,
        6.97106912, 11.00812996, 13.26209545, 13.15525803, 11.33837396,
        7.03733775,  4.98461983, 12.64235359,  8.62329466, 13.70893492,
        7.27759233, 10.7506423 , 11.4890861 , 14.07623266,  9.85330646,
        6.1463896 ,  9.78147542, 12.41012046,  8.81930525, 10.52587243,
        5.47702766,  8.1303945 ,  9.77588801, 11.51557166,  9.6282763 ,
        6.42689705, 10.4977132 , 10.84493732,  6.55800267, 11.77124373,
       12.17451685,  5.26067019, 12.07225469, 12.95702647,  8.7273569 ,
        5.70405525,  8.36995953,  9.49511524,  7.822854  ,  8.29022197,
        9.15866607,  7.73692436, 12.67576792, 11.99633819,  8.7907923 ,
        9.69629   , 13.08414689,  7.8575648 , 10.06407839,  5.47866221,
       13.86480175,  8.5724042 ,  6.383897  ,  6.27905292, 14.03505624,
       13.6660671 ,  6.67121748,  8.27853076,  7.72668467,  5.82345867,
       13.66896739,  8.92098752,  9.49511524,  7.54620849,  9.60412735,
        6.74872981,  6.60752373,  6.65368551,  8.47371868, 11.35345345,
        9.55593374, 10.69963954,  9.15253344,  6.58184474,  4.90505566,
        4.98989906, 13.46553623, 10.7506423 ,  8.27853076,  9.55720651,
       12.9089559 , 10.794755  , 12.72524615, 14.09357547,  8.10886021,
        8.87587457, 10.74655346,  7.61088125, 11.83075005,  7.63726134,
       14.10796466,  6.2134961 ,  6.7593917 , 13.66991027,  5.5748624 ,
        6.19707944, 13.77222752, 10.7167285 ,  9.88310614, 13.99877413,
        4.87083348])}
{'P': tensor([[2.6676e-02, 2.1618e-01, 1.9208e-01,  ..., 6.2778e-03, 8.2172e-04,
         9.2285e-04],
        [9.7997e-04, 2.6267e-02, 3.2458e-01,  ..., 6.2417e-03, 4.0809e-04,
         5.3358e-04],
        [1.3293e-03, 2.4674e-03, 4.4313e-02,  ..., 8.8371e-03, 5.9686e-04,
         7.1531e-04],
        ...,
        [2.9301e-03, 6.1123e-03, 7.4504e-03,  ..., 2.2599e-01, 2.6661e-04,
         2.4233e-03],
        [1.1114e-02, 1.3469e-02, 1.5360e-02,  ..., 1.1774e-03, 1.7641e-01,
         4.0982e-04],
        [1.1187e-02, 1.8479e-02, 1.8599e-02,  ..., 2.7309e-01, 3.6106e-04,
         1.5717e-01]]), 'X': tensor([[ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0305, 0.0242,  ..., 0.4309, 0.3905, 0.4752],
        [0.0305, 0.0000, 0.0119,  ..., 0.4005, 0.4205, 0.4459],
        [0.0242, 0.0119, 0.0000,  ..., 0.4094, 0.4112, 0.4560],
        ...,
        [0.4309, 0.4005, 0.4094,  ..., 0.0000, 0.8204, 0.0891],
        [0.3905, 0.4205, 0.4112,  ..., 0.8204, 0.0000, 0.8650],
        [0.4752, 0.4459, 0.4560,  ..., 0.0891, 0.8650, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.54127431,  6.74872981,  6.81084724,  6.80370254,  6.84796202,
        7.02496223,  7.02253235,  9.19837792,  5.64865173, 13.97857591,
       10.67322527,  7.37576154, 12.91505111,  8.89301516,  8.86082124,
       13.15525803, 10.412299  , 13.48809595, 13.6105595 , 12.21532282,
       12.52372629, 10.31886066,  7.4544866 , 12.4013153 ,  9.19837792,
        7.20482224,  9.66977802, 10.33958639,  5.0713364 , 11.83509372,
       13.54774355,  9.88571713, 10.5575601 ,  9.95552626,  5.1224079 ,
       13.43127131, 11.32149259, 11.1341798 ,  5.47866221,  5.64464807,
       11.48734978,  9.11473391,  9.9353921 ,  9.34061938, 13.89671314,
        5.90035324, 11.42420508,  9.09661212, 13.04217487, 13.04051744,
        6.38105015, 10.41909211,  6.05113217,  5.48276038,  5.84533345,
       11.07637531,  6.19633139,  5.16611782,  9.49511524,  9.54154096,
       13.62896593, 10.48315436, 13.59673406,  7.72415048, 13.59056613,
        5.53175275, 10.63038565, 11.70168755, 13.66051884,  6.12886024,
       11.63502679, 10.91805683,  8.4486771 ,  4.79003361,  8.37954853,
        9.50654967, 10.06850677,  8.85939626,  6.29034958,  4.91652556,
        8.98765644, 10.06407839, 12.32545637,  8.01009083,  8.67227386,
       13.58101794,  5.1224079 ,  5.94535529,  5.20659892,  8.09842704,
        8.30729352,  4.87083348,  4.9160962 ,  9.96391045, 13.77075174,
        6.4020404 ,  8.68979052,  9.18358562,  8.56889738, 12.70489563,
        5.39450305,  5.5748624 ,  9.41970031,  7.07902036,  6.38968357,
        6.39438044])}
{'P': tensor([[2.6318e-02, 2.0880e-01, 1.8984e-01,  ..., 1.4293e-04, 9.5847e-06,
         5.2799e-03],
        [1.0146e-03, 2.6243e-02, 3.2428e-01,  ..., 1.0103e-04, 1.0575e-05,
         7.2408e-03],
        [1.2671e-03, 2.3069e-03, 4.1430e-02,  ..., 1.4910e-04, 1.7174e-05,
         1.1989e-02],
        ...,
        [1.6837e-02, 9.6530e-03, 8.6315e-03,  ..., 1.1527e-01, 1.1172e-04,
         3.2285e-04],
        [1.0150e-04, 6.1042e-05, 5.1403e-05,  ..., 1.1866e-03, 4.3776e-02,
         1.0723e-06],
        [1.6666e-05, 3.5762e-05, 4.7211e-05,  ..., 2.4220e-06, 9.8222e-07,
         1.5381e-02]]), 'X': tensor([[ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0347, 0.0278,  ..., 0.2681, 0.9856, 0.2586],
        [0.0347, 0.0000, 0.0119,  ..., 0.2993, 0.9698, 0.2744],
        [0.0278, 0.0119, 0.0000,  ..., 0.2883, 0.9659, 0.2779],
        ...,
        [0.2681, 0.2993, 0.2883,  ..., 0.0000, 1.0229, 0.3846],
        [0.9856, 0.9698, 0.9659,  ..., 1.0229, 0.0000, 1.2437],
        [0.2586, 0.2744, 0.2779,  ..., 0.3846, 1.2437, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.54615887,  6.74872981,  6.81084724,  6.80370254,  6.84796202,
        7.02496223,  9.61942445,  5.05534609, 13.59480906, 10.46238054,
       10.37698435,  6.1463896 , 10.60226614, 13.66991027,  4.978704  ,
        8.87587457,  6.67218687, 10.00410625,  5.53255882,  6.38105015,
        5.16077764, 13.96987806,  5.10444097, 11.37939695,  7.54127425,
       10.61047596, 10.80728788, 10.39942497,  9.77777346, 10.412299  ,
        9.79039371,  5.89074104, 13.79762415, 12.27835996, 11.10066126,
        9.6282763 , 10.61472636,  4.91652556,  8.85939626, 10.42667093,
        7.28793663,  8.79284466,  4.9160962 ,  9.10945678,  8.09842704,
        5.29009836, 12.42554808, 11.97238624,  9.85330646,  4.74468406,
        7.34250948,  8.36995953, 10.54658597,  5.23878371,  8.47371868,
        9.78285424, 11.97957425, 13.95472331, 12.3489324 ,  5.84533345,
        8.50268322, 11.63765273,  8.35856487,  8.20436764,  6.46560181,
       10.27572673,  9.37296102,  7.39599986, 12.25806559, 10.39942497,
       10.35485331,  6.48486171, 10.61047596, 11.70016217,  7.14837514,
        5.13182506, 13.83139024, 10.68416401,  8.79284466,  8.50278896,
        8.86082124,  6.10774544, 14.08549802,  8.53014915,  4.91296205,
        8.01009083, 10.67132524,  7.76457095,  6.10360073,  7.32233733,
        6.78325151, 13.01187484,  8.62631801,  9.7712321 , 10.78533838,
        8.1137452 , 12.0495521 , 13.00380624, 12.4217523 , 10.52357479,
        9.79039371,  6.30647746,  8.16584537,  5.18277858, 10.26327895,
       13.60275211])}
{'P': tensor([[1.9477e-02, 1.6176e-01, 1.4717e-01,  ..., 1.3934e-05, 1.1354e-05,
         2.3591e-02],
        [9.0954e-04, 2.6284e-02, 2.8966e-03,  ..., 1.0445e-05, 1.9537e-05,
         4.6603e-02],
        [1.0395e-03, 1.3436e-01, 2.2109e-02,  ..., 1.0191e-05, 1.4606e-05,
         3.0566e-02],
        ...,
        [1.3806e-02, 8.0493e-03, 9.9102e-03,  ..., 1.1680e-01, 2.3273e-06,
         1.0278e-03],
        [1.9495e-05, 1.2491e-05, 1.3033e-05,  ..., 3.7561e-05, 2.4964e-02,
         7.9481e-06],
        [1.4462e-03, 2.6249e-03, 2.2860e-03,  ..., 1.9136e-05, 4.0510e-04,
         1.5863e-01]]), 'X': tensor([[ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0117, 0.0227,  ..., 0.3473, 1.0309, 0.1596],
        [0.0117, 0.0000, 0.0161,  ..., 0.3365, 1.0244, 0.1605],
        [0.0227, 0.0161, 0.0000,  ..., 0.3261, 1.0344, 0.1763],
        ...,
        [0.3473, 0.3365, 0.3261,  ..., 0.0000, 1.0168, 0.4378],
        [1.0309, 1.0244, 1.0344,  ..., 1.0168, 0.0000, 0.9073],
        [0.1596, 0.1605, 0.1763,  ..., 0.4378, 0.9073, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.54693987,  6.76661595,  6.7166297 ,  6.80559669,  6.81117802,
        6.78325151,  9.09119351, 11.00812996, 12.10942412,  4.76969364,
       11.33837396, 10.48840476,  7.02496223, 11.48734978,  9.17905491,
        8.8737556 , 13.0493368 , 10.794755  ,  9.58347456,  9.20340031,
        6.89245352,  7.14837514, 13.59480906, 13.1471313 , 12.67576792,
        5.1805345 ,  9.48969978,  6.02789634,  7.02253235, 13.55868684,
       13.1399181 ,  9.11473391, 10.38897563, 13.59170809,  8.7907923 ,
       11.32149259,  7.06169391, 10.06286074,  5.24461068, 12.53345445,
        5.13125513, 12.22503754, 12.59840193,  9.58072286,  7.96181288,
        5.08910491,  5.64464807,  9.71392179,  5.29755487, 13.08414689,
       13.04051744,  8.53014915,  6.54693987,  4.85995002,  7.73692436,
       13.32374056,  7.05627461, 12.5085995 , 11.38774149,  9.10945678,
       13.06498554,  6.25586297, 13.12218013,  9.97220799, 10.0674887 ,
       13.32613445,  5.94535529, 11.35345345, 12.87752052,  9.37989631,
        9.12220271, 13.32645552,  8.99791316, 11.4890861 ,  6.60752373,
        6.65368551, 14.08549802,  7.15100303, 13.09737053,  9.0368458 ,
        8.22728399, 11.23201619,  9.88310614,  6.3949043 , 12.88270018,
        9.68324553, 13.59056613, 14.09357547,  4.74468406,  7.92200342,
       11.35345345,  5.86519525, 11.66453059,  6.49538841, 10.8992524 ,
       10.06286074,  6.81084724,  9.11473391, 12.7125079 ,  9.36378438,
        4.978704  ,  8.89458273,  7.8575648 , 11.75618513,  9.78285424,
        7.26074925])}
{'P': tensor([[5.5755e-02, 1.8591e-01, 1.2012e-01,  ..., 6.1735e-05, 2.9056e-04,
         9.9941e-06],
        [8.4224e-04, 2.7811e-02, 2.4089e-01,  ..., 8.3589e-05, 1.0172e-04,
         1.4379e-05],
        [6.3496e-04, 1.3155e-03, 5.0447e-02,  ..., 2.7635e-04, 1.4159e-04,
         4.0353e-05],
        ...,
        [1.3888e-05, 1.3242e-05, 1.2099e-05,  ..., 4.5853e-02, 3.9880e-04,
         3.4684e-04],
        [8.3838e-03, 5.3588e-03, 2.8077e-03,  ..., 5.9765e-05, 6.2970e-02,
         1.4754e-05],
        [4.3454e-06, 6.1499e-06, 6.5983e-06,  ..., 9.4310e-03, 8.0896e-05,
         1.1344e-01]]), 'X': tensor([[ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0425, 0.0899,  ..., 0.9531, 0.1774, 1.2431],
        [0.0425, 0.0000, 0.0509,  ..., 0.9106, 0.1494, 1.2049],
        [0.0899, 0.0509, 0.0000,  ..., 0.8689, 0.1547, 1.1790],
        ...,
        [0.9531, 0.9106, 0.8689,  ..., 0.0000, 0.8277, 0.5304],
        [0.1774, 0.1494, 0.1547,  ..., 0.8277, 0.0000, 1.0753],
        [1.2431, 1.2049, 1.1790,  ..., 0.5304, 1.0753, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.55800267,  6.90115901,  7.15100303,  7.19534146,  7.14837514,
        7.16598273,  8.7273569 ,  6.19633139, 11.83075005,  8.36995953,
       12.49291768, 11.9702762 ,  4.7663526 , 13.6105595 , 13.36092624,
        5.35774573, 13.56367857,  9.50913849, 13.59056613,  7.83534987,
        4.83371341, 11.9120321 ,  5.38342077,  6.652742  ,  5.04028538,
        9.3349858 ,  6.29821673,  6.83083863,  4.79003361,  6.74686071,
        4.73784222, 13.85158417,  6.84796202,  5.5205189 , 13.12299145,
        7.59795445, 10.63038565, 12.59840193,  7.09887557,  9.6282763 ,
        5.94535529,  7.8788965 ,  9.11473391, 12.35625694, 10.99326577,
        5.19986797,  7.20826219,  9.37989631,  6.30647746, 13.1399181 ,
        9.79039371,  6.21275016, 11.83570132, 11.48663767,  6.33517349,
        8.57560275, 11.4424816 , 13.25017211, 10.48721382, 10.14436912,
       12.40074901, 12.30858677,  6.74872981, 13.38091121,  9.61942445,
        8.85939626,  8.68832398, 10.27716434,  9.09119351, 13.744387  ,
        6.1463896 , 13.58333857,  8.8580671 , 14.0393949 , 10.06407839,
       10.412299  ,  5.88782913,  7.07902036,  7.76457095, 13.0743399 ,
       10.39942497,  8.85572292,  7.9337647 ,  6.90115901,  9.88310614,
       10.46238054,  8.50278896, 13.97430191,  5.9510489 ,  9.78147542,
       10.84493732,  5.84533345,  6.56833864,  8.04718433,  8.1137452 ,
        8.51088913,  5.92281142,  7.63726134, 13.40024224, 10.35464916,
        9.78285424, 10.4977132 ,  6.76791483,  9.23178204,  5.2287058 ,
        8.92395381])}
{'P': tensor([[2.2976e-02, 2.6145e-01, 1.6781e-01,  ..., 1.5117e-05, 1.8786e-04,
         1.4532e-04],
        [1.6840e-03, 3.5488e-02, 1.6925e-01,  ..., 2.6989e-05, 3.8084e-04,
         1.8586e-04],
        [2.1171e-03, 5.2829e-03, 4.7813e-02,  ..., 3.6990e-05, 6.2623e-04,
         2.3170e-04],
        ...,
        [8.7270e-05, 7.1841e-05, 5.7941e-05,  ..., 1.3362e-01, 1.6052e-04,
         7.4444e-04],
        [5.9190e-05, 5.5235e-05, 5.3303e-05,  ..., 2.0735e-02, 8.2769e-02,
         4.1815e-04],
        [2.6703e-02, 2.0209e-02, 1.8824e-02,  ..., 4.5140e-05, 1.5842e-04,
         1.1587e-01]]), 'X': tensor([[ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0064, 0.0423,  ..., 1.0744, 0.6812, 0.2107],
        [0.0064, 0.0000, 0.0427,  ..., 1.0680, 0.6750, 0.2103],
        [0.0423, 0.0427, 0.0000,  ..., 1.0796, 0.6724, 0.1685],
        ...,
        [1.0744, 1.0680, 1.0796,  ..., 0.0000, 0.4908, 1.1034],
        [0.6812, 0.6750, 0.6724,  ..., 0.4908, 0.0000, 0.6501],
        [0.2107, 0.2103, 0.1685,  ..., 1.1034, 0.6501, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.5618968 ,  6.67108662,  6.74686071,  6.89245352,  7.01475376,
        6.652742  , 12.85032836, 13.04156075,  5.92952538, 12.22276278,
       13.96746816,  6.83083863,  8.89301516, 10.74409834, 11.50791859,
        8.25828409,  9.23178204,  5.5748624 , 12.41012046, 13.44593519,
        6.83515201, 11.75807252, 10.82106505, 11.9182956 ,  6.50409372,
       13.04156075, 10.68031351,  7.76457095,  7.6374603 ,  7.42512621,
       12.27991991, 10.42667093,  8.01009083,  6.84796202, 10.05558893,
       12.3828957 , 11.62158554, 13.62896593, 13.27717982,  7.85919687,
        9.48969978,  9.79039371, 13.56367857,  5.05444288, 11.46176259,
        6.10774544,  9.15866607,  8.01009083,  5.94293185,  9.18690018,
       13.92390689, 11.05672136, 10.33712258,  7.62232307, 13.36800073,
       10.61472636,  7.83486943,  7.88677538, 14.05633441, 10.99326577,
        7.39599986, 12.53047489,  6.80559669,  6.33517349,  4.78002665,
        6.7166297 ,  7.60463349,  9.63742669,  6.53290663, 12.22276278,
       13.09426142,  5.5671389 , 11.1913598 ,  5.39098669, 11.83570132,
        8.98765644,  5.1224079 ,  9.3349858 ,  6.90115901,  6.38399789,
        4.74468406,  5.39450305, 10.87409005,  9.94285511,  7.60463349,
        9.50913849,  5.64500512,  5.26342203, 14.10045797, 11.31276109,
       10.14436912, 11.33837396, 10.82106505, 14.09176259,  9.63742669,
        7.34250948,  6.76791483, 13.37121923, 10.5575601 ,  8.50268322,
        5.08372149,  8.09899874,  5.94909709,  9.70108409,  8.91551231,
        5.29009836])}
{'P': tensor([[1.8422e-02, 1.8179e-01, 1.5312e-01,  ..., 2.2012e-05, 4.9351e-05,
         5.1301e-04],
        [1.1597e-03, 1.9874e-02, 1.8388e-01,  ..., 3.0245e-05, 4.9424e-05,
         4.8767e-04],
        [1.2911e-03, 2.0072e-03, 2.3353e-02,  ..., 3.9662e-05, 5.7277e-05,
         5.5439e-04],
        ...,
        [9.9198e-06, 1.2535e-05, 1.3948e-05,  ..., 2.9844e-01, 3.0949e-05,
         4.5385e-05],
        [5.4437e-03, 5.0917e-03, 5.0497e-03,  ..., 6.2338e-05, 1.6192e-01,
         3.6561e-02],
        [3.7201e-03, 4.9715e-03, 5.6769e-03,  ..., 2.1535e-03, 7.2904e-04,
         1.1437e-01]]), 'X': tensor([[  6.3031,   1.8477,  -7.1706],
        [  6.1929,   2.4501,  -6.9352],
        [  6.1372,   2.6726,  -6.7885],
        [  6.0047,   3.1062,  -6.7914],
        [  5.9655,   3.2172,  -7.5953],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.7544,   2.6096,   3.3146],
        [  5.0233,   4.9647,  -3.1788],
        [  6.2895,   0.0405,   5.7652],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.1142,  -2.7230,  -4.1966],
        [  4.3377,   5.7570,   1.3712],
        [ -0.8675, -10.8815,   6.8561],
        [ -0.0254, -10.9932,   3.2200],
        [  8.7131,  10.2216,   7.7915],
        [  2.5974,   7.0363,  -7.7810],
        [  7.3592,  11.3937,   6.2407],
        [ 12.6035,   0.7494,   7.1536],
        [  5.8323,  -1.5228,   5.1132],
        [  1.2728, -11.0372,   9.8121],
        [ 12.2311,  -2.0445,   1.3545],
        [  3.0876,   6.7527,  -5.3540],
        [  6.0047,   3.1062,  -6.7914],
        [ -6.7745,  -7.7264,   8.4684],
        [  6.3413,   0.4883,  -6.8416],
        [  9.3836,   9.5075,  -5.9278],
        [ 11.5966,   5.9666,   6.0083],
        [ -3.5173,   7.5091,  -8.9585],
        [ -8.0829,  -5.9940,   5.0564],
        [ -2.0445,   7.8469,  -0.5857],
        [  2.7295,  13.6703,  -8.9328],
        [  9.4925,   9.3811,   8.9842],
        [ -3.3999, -10.1152,  -1.2088],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -4.7389,   6.9953,   3.7638],
        [  2.3200,   7.1756,   1.1938],
        [  0.5850,  -4.7982,   9.6639],
        [  3.2366,  -4.2732,   5.1532],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.3087,   0.7792,  -7.5587],
        [ -8.5859,  -5.0393,   0.5476],
        [ -4.7389,   6.9953,   3.7638],
        [  6.2419,  -0.2313,   4.2440],
        [  9.3836,   9.5075,  -5.9278],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.4524,  -1.6813,  -6.6862],
        [  2.2860,   7.1916,  -6.1011],
        [  9.7328,   9.0902,  -0.8917],
        [  5.3602,  -2.3808,   7.2098],
        [ 11.8951,  -3.1570,   7.0742],
        [  2.5318,  13.7231,   6.3692],
        [  2.5620,   7.0549,   8.4825],
        [  4.5057,  -3.3894,  -0.8160],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -4.4503,   7.1378,   3.3751],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.5128, -10.9365,   9.9645],
        [ 10.0619,   8.6627,  -9.2884],
        [ 11.3691,   6.4760,   1.0787],
        [ -7.6525,  -6.6534,   5.1721],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.7565,  11.8237,  -4.9882],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.3076,   0.7852,  -6.7010],
        [  1.3339,   7.5627,  -0.9034],
        [  9.8082,  -6.8396,   4.9615],
        [ 11.3118,   6.5957,  -4.5667],
        [  6.3048,   0.1469,  -9.1691],
        [ -1.6906,   7.8860,   7.7142],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.1666,   3.7653,   3.6075],
        [  4.8051, -10.3703,   0.5268],
        [ -7.7842,   4.3466,  -1.7548],
        [  2.6613,   7.0021,   4.5783],
        [  6.8407,  11.7666,   3.0045],
        [  7.9403,  -8.6455,  -8.7433],
        [  3.9515,  -3.8425,   6.1729],
        [  1.9204,   7.3518,   8.4781],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.8407,  11.7666,   3.0045],
        [  2.2659,  13.7894,   6.5436],
        [ -7.9298,   4.1367,  -3.9773],
        [ 12.5832,   1.7543,  -7.6134],
        [ -0.8455, -10.8853,  -6.7400],
        [  9.6587,   9.1817,  -0.8212],
        [  5.7485,  -9.9811,  -3.2355],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.4932,   7.0903,  -5.4214],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.8407,  11.7666,   3.0045],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.3367,  -4.4806,   0.4643],
        [ -6.3306,   5.9272,   5.6935],
        [ 10.1798,   8.5002,  -5.8893],
        [ -1.6086,   7.8928,   4.3385],
        [  1.7000,  -4.7609,   2.7935],
        [ -1.9714,   7.8563,  -1.8084],
        [ -9.4624,  -0.4633,  -7.3918],
        [  1.1274,  14.0116,   8.1753],
        [  1.8721,  -4.7322,   0.5524],
        [  6.3305,   1.6119,   1.1538]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0079, 0.0143,  ..., 1.1192, 0.4194, 0.4822],
        [0.0079, 0.0000, 0.0064,  ..., 1.1145, 0.4127, 0.4768],
        [0.0143, 0.0064, 0.0000,  ..., 1.1102, 0.4069, 0.4719],
        ...,
        [1.1192, 1.1145, 1.1102,  ..., 0.0000, 0.7447, 0.6436],
        [0.4194, 0.4127, 0.4069,  ..., 0.7447, 0.0000, 0.1278],
        [0.4822, 0.4768, 0.4719,  ..., 0.6436, 0.1278, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.56833864,  6.65991877,  6.69389431,  6.76056625,  6.77777606,
        6.83515201,  9.13507321,  7.06272256,  6.28961988,  9.4997987 ,
       13.04217487,  5.79391825,  7.20826219, 10.91601655, 10.99326577,
       13.43127131,  7.50035355, 13.56367857, 12.62576026,  6.02779201,
       11.11038551, 12.40074901,  7.42512621,  6.76056625, 10.27572673,
        6.36004393, 13.35832725, 13.04156075,  8.29203356, 10.06286074,
        8.10886021, 13.94008948, 13.34587027, 10.67132524,  9.47369628,
        8.44940661,  7.54127425,  4.83371341,  5.36061974, 10.46739073,
        9.11473391,  9.34127013,  9.95552626,  8.44940661,  6.24614905,
       13.35832725, 11.72802491,  9.60080639,  7.54620849, 13.31764181,
        5.86519525, 12.30695098, 13.95472331,  7.50564223,  5.63824971,
       12.10942412,  8.41148722, 10.412299  , 11.22142195, 13.27717982,
       13.08414689, 10.14044489,  8.49161154, 13.61801526,  9.17595899,
        9.34061938,  7.67939546, 11.95743558, 13.09426142,  6.30647746,
        8.06517143,  8.87007495,  8.99276564, 11.42947632,  8.91551231,
        7.49076899, 13.6105595 , 11.73848065,  5.51177501,  7.5984809 ,
       10.91805683, 13.6105595 , 13.97430191,  8.94395628, 12.70489563,
       10.91805683, 13.32645552, 11.51811603, 12.18045978,  7.5158489 ,
       10.8207297 , 13.6105595 ,  9.34127013,  9.78286733,  8.85572292,
       13.02818702, 12.18998195,  8.67227386, 13.26209545,  8.0550286 ,
        5.05534609,  8.09984231,  9.47369628, 14.05687815,  5.08910491,
        6.53250937])}
{'P': tensor([[3.9737e-02, 1.6019e-01, 1.3436e-01,  ..., 7.2907e-06, 2.4627e-04,
         3.4876e-03],
        [9.9013e-04, 2.4651e-02, 2.6138e-01,  ..., 2.7070e-06, 1.0579e-04,
         7.5926e-04],
        [8.4376e-04, 1.4481e-03, 4.2933e-02,  ..., 4.1339e-06, 1.3741e-04,
         4.5833e-04],
        ...,
        [7.2168e-04, 2.4847e-04, 1.5172e-04,  ..., 8.7295e-02, 3.8097e-04,
         8.0925e-05],
        [1.3395e-02, 1.2563e-02, 8.2681e-03,  ..., 4.4285e-05, 8.7644e-02,
         4.9504e-03],
        [3.1582e-04, 1.7281e-03, 3.7583e-03,  ..., 8.3747e-08, 8.5133e-06,
         1.4780e-02]]), 'X': tensor([[  6.2905,   1.9366,   2.1169],
        [  5.7845,   3.6652,   0.9175],
        [  5.1890,   4.7318,   0.9515],
        [  4.8432,   5.1968,   1.5006],
        [  4.6992,   5.3690,   3.0935],
        [  5.1793,   4.7461,   0.3899],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.1469,   6.9237,   1.6160],
        [ -0.2586,   7.8825,  -9.9840],
        [ -8.6681,  -4.8556,   9.2518],
        [  8.7131,  10.2216,   7.7915],
        [  2.1477,   7.2550,   8.9688],
        [  1.8719,   7.3713,  -3.7465],
        [  4.7813,  -3.1139,   8.2691],
        [ -5.3328,  -8.9992,   7.3440],
        [  6.3087,   1.8053,  -1.9481],
        [  3.3943,   6.5483,   1.0790],
        [  9.6614,   9.1784,  -1.2087],
        [ 12.4592,  -0.9168,  -6.7571],
        [  0.6363, -11.0348,   6.1414],
        [ -7.4602,   4.7718,  -4.6277],
        [  7.8982,  -8.6778,   0.0239],
        [  1.8447,  -4.7372,   8.1235],
        [ 11.1469,   6.9237,   1.6160],
        [ -8.0588,  -6.0341,  -2.3006],
        [  6.9284,  11.7061,  -0.3732],
        [  9.9217,  -6.7006,  -5.6728],
        [ 12.2435,   4.0076,   8.1257],
        [ 11.6019,   5.9540,  -5.1715],
        [  3.1271, -10.8343,   2.6442],
        [ -5.1377,   6.7751,  -0.4438],
        [  1.7903,  -4.7467,   0.1168],
        [ -4.6393,   7.0461,   1.1624],
        [  5.5424,  -2.0885,   7.2487],
        [ -9.4748,  -0.7765,  -5.6695],
        [  4.9337,  12.8566,  -3.6582],
        [  5.1196,  -2.7160,   4.2705],
        [ 11.9641,  -2.9568,   7.7211],
        [  2.7976,  -4.4682,  -4.2020],
        [ -1.9646,   7.8571,   3.3817],
        [  2.7558,  -4.4844,   0.0636],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.8704,  -9.3809,   7.5769],
        [  4.7156,  -3.1831,  -8.0224],
        [ -5.6792,  -8.7352,   4.0318],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.6165,  -3.8715,   6.4594],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.0965,  -5.9712,   4.3466],
        [  4.9337,  12.8566,  -3.6582],
        [  2.3261,  13.7749,   5.9121],
        [  9.4516,  -7.2501,  -1.8868],
        [ 12.2592,   3.9425,   2.0383],
        [  4.5814, -10.4485,   2.5496],
        [ -7.2515,   5.0195,   4.6434],
        [  4.1348,   5.9514,  -0.8856],
        [  1.8320,   7.3871,   5.1452],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -1.5451,   7.8975,   7.6104],
        [ -1.8788, -10.6567,   3.7068],
        [ -8.5859,  -5.0393,   0.5476],
        [  9.8082,  -6.8396,   4.9615],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.9178,   2.1936,   7.5388],
        [  5.6202,  -1.9508,   0.7471],
        [ -9.3604,   0.4855,  -3.0158],
        [  1.9241,   7.3503,   7.9004],
        [ -8.5506,  -5.1154,   6.3423],
        [ -7.5720,   4.6310,   2.1717],
        [ 11.3118,   6.5957,  -4.5667],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.3411,  -2.6004,  -2.7062],
        [  0.4119,  14.1019,  -4.1870],
        [  2.0899,  -4.6870,  -7.4260],
        [  5.1787,  -2.6383,  -4.2533],
        [  3.1756, -10.8247,   6.8882],
        [  5.5424,  -2.0885,   7.2487],
        [ -0.8455, -10.8853,  -6.7400],
        [ -1.6220,   7.8917,   0.0315],
        [  5.0348,  -2.8229,   3.5436],
        [  5.8902,  -1.3887,   4.0388],
        [  2.4639, -10.9431,  -7.3718],
        [ -7.3470,  -7.0615,   5.0066],
        [ 12.0582,  -2.6640,   0.7132],
        [ 11.5456,   6.0861,   8.1101],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -5.0943,  -9.1682,   0.5073],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.8902,  -1.3887,   4.0388],
        [ -5.7347,  -8.6908,   9.2878],
        [ -8.6229,   2.9068,   2.4429],
        [  3.1480,  -4.3165,  -7.8002],
        [  7.2340,  -9.1505,  -5.8321],
        [  7.0171,  11.6439,  -5.0178],
        [  1.3767, -11.0341,   9.9415],
        [  4.0642,  -3.7599,  -3.4565],
        [ 11.7418,   5.6070,  -1.4192]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0845, 0.0923,  ..., 0.7619, 0.4011, 0.3504],
        [0.0845, 0.0000, 0.0216,  ..., 0.8077, 0.3170, 0.3012],
        [0.0923, 0.0216, 0.0000,  ..., 0.7941, 0.3145, 0.3171],
        ...,
        [0.7619, 0.8077, 0.7941,  ..., 0.0000, 1.0450, 1.1077],
        [0.4011, 0.3170, 0.3145,  ..., 1.0450, 0.0000, 0.2842],
        [0.3504, 0.3012, 0.3171,  ..., 1.1077, 0.2842, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.58184474,  6.84796202,  7.02253235,  7.10378919,  7.13501264,
        7.02496223, 12.42554808, 10.61472636, 13.12218013,  7.88677538,
        9.9353921 , 13.43127131,  7.56617037,  7.60529892,  5.70591986,
       10.4606265 ,  6.5618968 ,  7.37576154, 13.32613445, 12.49291768,
       11.05317597,  8.85572292, 11.73397309,  5.08372149, 13.12218013,
       10.0674887 , 13.60275211, 11.97238624, 12.88270018, 13.04051744,
       11.27656488,  8.50278896,  5.07306274,  8.43626114,  5.92281142,
        9.50654967, 13.77075174,  5.79543038, 12.32408516,  5.27178363,
        8.09899874,  5.26342203,  8.44940661, 12.52372629, 12.84056604,
       11.62770332,  5.68942572, 10.41909211,  5.60064319, 12.24467041,
        5.68942572,  9.3109884 ,  9.3349858 ,  8.98765644, 10.06023665,
       13.77075174, 13.96987806, 11.9120321 , 12.87752052, 11.4088043 ,
        8.81930525,  7.24677664,  7.61088125,  7.13501264,  9.51074763,
        8.04718433, 10.82106505,  9.95552626, 11.95743558, 10.65929968,
        9.79814997,  9.18358562,  5.94909709,  9.37296102,  7.59795445,
        9.96391045,  8.87587457, 13.09426142, 12.22276278,  9.69629   ,
       14.10796466,  5.13182506,  5.81199422, 11.28093507,  5.92281142,
       10.91805683,  8.05668609,  5.77219077,  6.05165248, 11.21703654,
       10.19036081, 12.3489324 , 13.05148826, 10.14436912, 10.48840476,
       12.45418236,  6.1561646 ,  6.05165248, 10.412299  ,  9.09963894,
        5.34250225, 11.66453059, 13.59480906, 11.11969815,  5.5366698 ,
       13.01187484])}
{'P': tensor([[2.7666e-02, 2.1288e-01, 1.5833e-01,  ..., 8.2292e-06, 5.0734e-03,
         3.1764e-03],
        [1.0583e-03, 3.5103e-02, 3.5363e-03,  ..., 1.0219e-05, 2.4342e-03,
         5.3327e-03],
        [1.4854e-03, 1.0373e-01, 2.4981e-02,  ..., 7.5707e-06, 3.0186e-03,
         2.4269e-03],
        ...,
        [9.3854e-05, 4.3519e-05, 6.2267e-05,  ..., 2.1997e-02, 5.5658e-06,
         3.0295e-06],
        [6.2337e-05, 2.9311e-04, 1.6651e-04,  ..., 8.8372e-08, 5.0831e-03,
         2.4727e-04],
        [5.4021e-04, 9.8519e-04, 6.6237e-04,  ..., 3.2065e-05, 1.2338e-04,
         1.6598e-01]]), 'X': tensor([[ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 1.9116e-02, 4.6445e-02,  ..., 9.2090e-01, 2.7484e-01,
         4.7828e-01],
        [1.9116e-02, 0.0000e+00, 5.5497e-02,  ..., 9.0815e-01, 2.9395e-01,
         4.6392e-01],
        [4.6445e-02, 5.5497e-02, 0.0000e+00,  ..., 8.9899e-01, 2.6253e-01,
         5.1858e-01],
        ...,
        [9.2090e-01, 9.0815e-01, 8.9899e-01,  ..., 0.0000e+00, 1.1190e+00,
         1.0248e+00],
        [2.7484e-01, 2.9395e-01, 2.6253e-01,  ..., 1.1190e+00, 0.0000e+00,
         7.1369e-01],
        [4.7828e-01, 4.6392e-01, 5.1858e-01,  ..., 1.0248e+00, 7.1369e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.59176018,  6.84796202,  6.74872981,  6.81084724,  7.02253235,
        7.02496223,  7.15100303,  9.79814997, 10.23946247, 13.85158417,
       12.49317563, 14.0393949 ,  9.27474386,  5.1805345 , 10.8992524 ,
        9.26455795,  8.30125201,  7.6374603 ,  6.99097013, 13.66896739,
        9.41605684,  8.16714836,  5.64500512,  7.44805901,  9.36378438,
       13.77604293,  8.91551231,  9.13507321, 13.32613445, 10.74409834,
        5.52416368,  6.81117802,  7.67939546,  5.91222346, 13.72594022,
       10.88618114,  5.98285319,  7.24677664,  6.02789634, 13.36800073,
       12.42554808, 12.04924159,  5.39450305,  9.41970031,  8.36995953,
        8.19703512, 14.03618649, 11.77124373, 13.70893492, 10.46238054,
       10.7506423 , 11.38774149,  9.26455795, 10.48315436, 13.14777433,
       13.54774355, 10.08001894,  5.34250225,  9.23071698,  9.77588801,
        5.91677188,  6.36004393,  6.81961996,  6.23262289,  6.81117802,
        9.97220799,  5.47702766,  7.06272256, 11.38774149, 13.83139024,
       13.27994304, 10.92597196,  9.99211663, 13.32613445,  5.37343173,
       13.86480175, 12.12641969, 12.44177641,  5.69913817,  5.16205087,
        5.1805345 , 13.40024224,  7.19513806,  6.18668463,  5.92281142,
        6.38968357, 11.9182956 , 11.72802491,  5.68942572, 12.59969936,
        9.60080639,  6.10774544,  6.38399789,  9.58072286,  9.64157461,
       11.87412439,  5.27178363,  5.94535529,  9.3109884 , 11.07637531,
       10.00410625,  7.73692436,  9.77777346, 10.5575601 , 13.1471313 ,
        7.52117132])}
{'P': tensor([[3.1790e-02, 2.3531e-01, 1.6773e-01,  ..., 5.4907e-05, 5.4716e-03,
         3.6367e-03],
        [2.5611e-03, 7.1489e-02, 9.2736e-03,  ..., 5.6286e-05, 4.4653e-03,
         1.5888e-02],
        [2.8210e-03, 2.5371e-01, 5.0417e-02,  ..., 4.6723e-05, 3.9375e-03,
         1.1705e-02],
        ...,
        [2.5785e-03, 1.7753e-03, 2.0439e-03,  ..., 7.5038e-03, 4.6486e-03,
         4.3938e-05],
        [1.6276e-04, 6.4830e-04, 4.6596e-04,  ..., 5.4853e-07, 8.5219e-03,
         7.0967e-04],
        [2.3673e-04, 3.6258e-04, 3.4429e-04,  ..., 1.0114e-05, 3.3276e-05,
         1.2075e-01]]), 'X': tensor([[ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0234, 0.0349,  ..., 0.0656, 0.3059, 0.4044],
        [0.0234, 0.0000, 0.0185,  ..., 0.0560, 0.3291, 0.3819],
        [0.0349, 0.0185, 0.0000,  ..., 0.0718, 0.3348, 0.3708],
        ...,
        [0.0656, 0.0560, 0.0718,  ..., 0.0000, 0.3519, 0.3877],
        [0.3059, 0.3291, 0.3348,  ..., 0.3519, 0.0000, 0.6986],
        [0.4044, 0.3819, 0.3708,  ..., 0.3877, 0.6986, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.60752373,  6.83083863,  6.78325151,  7.03733775,  6.81117802,
        6.7166297 , 10.05558893,  6.2322769 , 11.83509372, 13.12218013,
       13.59170809, 12.27835996,  7.73692436,  5.98285319, 11.27656488,
        7.58668144, 13.35832725,  6.42689705, 12.59969936, 13.45642379,
       12.78327037,  5.05444288, 10.36844813,  6.5618968 , 10.31886066,
       12.58006038,  7.9337647 ,  8.31625171,  9.62238828,  8.06643487,
        5.89074104, 13.89671314,  5.05981914,  6.55800267,  4.74468406,
        7.27340834,  6.24614905,  6.58184474,  6.12886024, 11.05413545,
       11.32149259, 10.4606265 ,  8.79284466, 13.58101794, 10.36844813,
        5.91854397,  8.86082124,  5.94909709,  7.62509582,  9.9353921 ,
        6.1463896 , 11.00694688,  6.81961996, 13.92390689, 13.54774355,
       13.14777433,  9.6282763 , 12.18045978, 10.48315436,  7.92607342,
        9.87720102, 11.4088043 , 12.62711252,  8.83341608, 12.30858677,
        5.96556248,  9.78286733, 10.74409834,  6.74872981, 13.66896739,
       13.83139024,  6.46560181, 13.59170809, 11.63502679,  7.92607342,
        6.46356806, 12.07225469, 12.35625694, 12.78327037,  5.1805345 ,
       10.71162681, 12.85032836, 10.07238463, 11.18839169, 10.61472636,
       12.50996828,  5.08910491,  5.1805345 ,  7.76457095, 11.28093507,
        7.37576154, 10.1979669 , 13.77075174,  6.54615887,  8.5724042 ,
        6.33075682, 13.88683436, 13.68063098,  8.87007495, 10.31886066,
       14.10045797,  5.2287058 ,  9.94173708, 12.12448912, 13.09737053,
        7.81509631])}
{'P': tensor([[2.0783e-02, 1.8774e-01, 1.5043e-01,  ..., 1.5215e-03, 9.1989e-05,
         1.0494e-05],
        [2.4350e-03, 2.8808e-02, 1.7181e-01,  ..., 2.3912e-03, 1.4059e-04,
         1.3267e-05],
        [2.8494e-03, 3.9543e-03, 6.7774e-02,  ..., 9.2737e-03, 4.5553e-04,
         2.1163e-05],
        ...,
        [6.0407e-05, 6.6011e-05, 9.8237e-05,  ..., 2.1093e-01, 9.3240e-03,
         1.1620e-06],
        [5.6883e-05, 5.4923e-05, 4.6896e-05,  ..., 4.0788e-06, 8.6604e-02,
         8.5484e-05],
        [3.0334e-03, 2.7652e-03, 1.5408e-03,  ..., 8.4509e-06, 2.0753e-06,
         3.3502e-02]]), 'X': tensor([[  6.2576,   2.1359,  -3.3398],
        [  6.2038,   2.4020,  -3.1692],
        [  5.8979,   3.3958,  -3.8713],
        [  5.6523,   3.9443,  -2.9449],
        [  5.8836,   3.4315,  -4.0327],
        [  5.9912,   3.1453,  -3.9989],
        [ -8.9178,   2.1936,   7.5388],
        [  2.5469,  13.7192,  -9.6935],
        [  6.0305,   3.0296,   0.1375],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.2188,  -2.5839,   2.1833],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.6093,  -1.9705,   6.4807],
        [ -6.4368,   5.8356,   3.3014],
        [  6.0890,  -0.8382,   2.9166],
        [  0.2435,   7.8191,  -6.4709],
        [ -2.8162,   7.7059,  -9.6885],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 11.4235,   6.3594,   5.1663],
        [ -3.0004,   7.6607,  -7.2640],
        [ -7.4808,   4.7462,   2.0266],
        [  5.5101,  -2.1433,   9.5442],
        [ -9.4524,  -1.6813,  -6.6862],
        [  1.1994,   7.6035,   2.4295],
        [  3.3943,   6.5483,   1.0790],
        [  7.2340,  -9.1505,  -5.8321],
        [ 12.4592,  -0.9168,  -6.7571],
        [  5.8836,   3.4315,  -4.0327],
        [  5.6672,  -1.8631,   8.1020],
        [  7.0664,  -9.2590,   8.0879],
        [ -3.6381,   7.4683,   7.3544],
        [  5.5424,  -2.0885,   7.2487],
        [ 12.5717,   1.9266,   7.4903],
        [  2.5469,  13.7192,  -9.6935],
        [  0.9392,  -4.8143,  -3.4649],
        [ -0.5739,   7.9056,   0.6382],
        [  6.3589,   0.8239,  -8.4230],
        [  5.8325,  -1.5222,  -2.1341],
        [  2.3261,   7.1726,   6.2522],
        [  3.3744,   6.5623,  -6.3636],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3589,   0.8239,  -8.4230],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.4203,  -2.0340,   2.5856],
        [  5.2608,  12.6977,   4.9166],
        [  4.3377,   5.7570,   1.3712],
        [  0.1047, -11.0046,   0.5188],
        [ -0.6323,   7.9085,  -5.9544],
        [ -8.6681,  -4.8556,   9.2518],
        [ 12.5717,   1.9266,   7.4903],
        [  1.7903,  -4.7467,   0.1168],
        [ -6.4844,  -8.0230,   8.5209],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.5767,   5.5071,   3.9069],
        [  0.9478,  -4.8144,   0.6371],
        [  2.3367,   7.1676,  -9.5263],
        [  6.5428,  -9.5728,  -2.9897],
        [  9.4554,   9.4245,   9.8800],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.2026,  12.1774,  -1.5887],
        [  8.5198,  10.4091,  -8.6486],
        [ -3.2053, -10.1993,  -3.8736],
        [ -4.0633,  -9.7927,  -3.0825],
        [  0.5600, -11.0321,  -0.8660],
        [  0.1252, -11.0062,   7.2564],
        [ 12.1379,   4.4125,  -3.8833],
        [ -6.4368,   5.8356,   3.3014],
        [ -9.4753,  -0.8012,  -5.2050],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.2911,  -2.4824,  -3.6877],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.7544,   2.6096,   3.3146],
        [ -0.2586,   7.8825,  -9.9840],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.1625,  -4.6697,   7.0658],
        [ -7.6719,  -6.6260,   7.1468],
        [  1.1350,  14.0104,   0.6321],
        [  2.6614,  -4.5192,  -9.3641],
        [ -8.7735,   2.5640,  -6.3313],
        [  0.5960, -11.0335,   3.1144],
        [  1.3767, -11.0341,   9.9415],
        [  7.0515,  11.6194,   8.4289],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.3604,   0.9042,  -6.5901],
        [  3.7992,  -3.9474,  -8.9019],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.0467,  -9.8369,  -6.5816],
        [  7.4132,  11.3526,   0.3322],
        [  8.5198,  10.4091,  -8.6486],
        [  2.9260,  -4.4162,   8.3808],
        [  2.7295,  13.6703,  -8.9328],
        [ -0.2586,   7.8825,  -9.9840],
        [ -1.8788, -10.6567,   3.7068],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.4117,   0.1160,   8.2574],
        [ -0.8569,   7.9156,  -6.5557],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.6314,  -1.9302,  -9.6543],
        [  7.0642,  11.6104,  -9.8316],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.9461,  -4.4077,  -0.4831],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.4150,  13.9647,  -4.0915],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.9433,  -9.3362,  -0.3421]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0106, 0.0271,  ..., 0.2527, 0.7985, 0.4083],
        [0.0106, 0.0000, 0.0356,  ..., 0.2611, 0.7946, 0.4009],
        [0.0271, 0.0356, 0.0000,  ..., 0.2503, 0.7889, 0.4095],
        ...,
        [0.2527, 0.2611, 0.2503,  ..., 0.0000, 1.0257, 0.6597],
        [0.7985, 0.7946, 0.7889,  ..., 1.0257, 0.0000, 0.4484],
        [0.4083, 0.4009, 0.4095,  ..., 0.6597, 0.4484, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.61211224,  6.65259775,  6.80559669,  6.89245352,  6.81117802,
        6.76661595,  9.18358562, 13.95361098,  6.74872981,  8.16714836,
        5.82345867,  9.58347456,  5.94535529,  8.68832398,  6.1463896 ,
        7.822854  ,  8.20436764, 10.14436912, 13.0743399 ,  8.22728399,
        8.859416  ,  5.91222346,  9.60080639,  7.69752262,  7.37576154,
       11.66453059, 12.49291768,  6.81117802,  5.96556248, 11.64746882,
        8.30729352,  5.92281142, 12.71843462, 13.95361098,  4.90505566,
        7.92644746,  6.41202986,  6.02789634,  7.5403822 ,  7.37902766,
       11.62158554,  6.41202986,  7.98469166,  9.63742669, 13.744387  ,
        7.20826219, 11.00508681,  7.9337647 ,  9.9353921 , 12.71843462,
        5.07306274, 10.31584102, 12.22276278,  7.16058587,  4.90676242,
        7.53884104, 11.59512194, 13.35013374, 12.0495521 , 13.6660671 ,
       13.45124917, 10.6910768 , 10.60226614, 11.04629471, 11.00694688,
       12.91505111,  8.68832398,  9.50913849,  4.74468406,  5.84450127,
        5.5671389 ,  9.13507321,  7.88677538,  9.4997987 ,  5.14607014,
       10.1371552 , 14.05633441,  5.24461068,  9.14044984, 11.04953767,
       11.11969815, 13.59170809, 12.67576792,  6.00732193,  6.42439344,
        5.47866221, 12.11735334, 11.54672755, 13.55868684, 13.45124917,
        5.29755487, 13.94008948,  7.88677538, 10.82106505,  8.29203356,
        9.41244865,  7.96181288, 12.17901033,  5.95297459, 13.59056613,
        8.25828409,  5.30158497,  8.06990483, 14.03618649,  9.14973319,
       11.63502679])}
{'P': tensor([[2.4522e-02, 1.9889e-01, 1.3357e-01,  ..., 1.1532e-04, 8.2080e-06,
         8.4913e-04],
        [3.2393e-03, 3.9087e-02, 2.4826e-01,  ..., 1.7304e-04, 1.2930e-05,
         1.3498e-03],
        [8.0858e-04, 9.4233e-04, 3.8510e-02,  ..., 1.0688e-04, 1.6007e-05,
         5.2626e-04],
        ...,
        [1.1018e-02, 1.0534e-02, 4.6556e-03,  ..., 9.5561e-02, 7.0863e-05,
         1.1486e-02],
        [7.1130e-05, 5.8717e-05, 2.6981e-05,  ..., 6.2670e-04, 3.4836e-02,
         9.3635e-06],
        [3.8886e-03, 4.9610e-03, 8.1735e-03,  ..., 7.8516e-04, 1.3880e-05,
         1.7604e-01]]), 'X': tensor([[ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0169, 0.0423,  ..., 0.3841, 0.9674, 0.5993],
        [0.0169, 0.0000, 0.0432,  ..., 0.3674, 0.9663, 0.5834],
        [0.0423, 0.0432, 0.0000,  ..., 0.3860, 0.9252, 0.6093],
        ...,
        [0.3841, 0.3674, 0.3860,  ..., 0.0000, 1.0623, 0.2383],
        [0.9674, 0.9663, 0.9252,  ..., 1.0623, 0.0000, 1.2684],
        [0.5993, 0.5834, 0.6093,  ..., 0.2383, 1.2684, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.61386436,  6.67218687,  6.99097013,  6.67121748,  7.16058587,
        7.13501264,  6.2322769 ,  8.87007495,  7.76457095, 10.31584102,
       10.53368644,  8.31625171, 12.17451685,  8.29203356, 11.63383221,
       10.74655346, 13.6829843 , 11.87412439,  6.5618968 , 13.26209545,
        7.5984809 , 11.1913598 , 11.83509372,  8.7907923 ,  5.26342203,
        5.94909709, 13.13888841,  6.41202986, 14.0393949 ,  9.75401449,
        5.24461068, 11.11038551, 10.1979669 ,  9.58072286, 12.10942412,
        5.43943791, 14.02058621, 11.97957425, 11.28895489, 12.3828957 ,
       13.13888841, 10.9036226 , 10.06850677,  4.7663526 , 13.59056613,
        6.42439344, 11.48434767,  5.05534609,  7.07902036, 13.89974693,
       13.13888841, 14.08549802, 11.05317597,  5.82345867,  6.65991877,
        6.652742  ,  6.39438044,  7.62509582,  6.24614905,  9.34127013,
        6.02789634, 13.66051884, 10.14436912, 13.36092624,  6.67218687,
       11.4890861 ,  9.88310614,  8.87587457,  5.43943791, 11.75438006,
        7.60463349, 13.35927527, 10.46238054, 10.51437947, 11.04953767,
        6.2134961 ,  5.39723241, 13.77604293,  5.68942572, 13.1399181 ,
        9.41605684, 10.06023665, 10.33958639,  8.29640613,  5.94535529,
       12.49291768, 13.18278606, 11.42420508,  6.49538841, 10.88618114,
        8.24808832, 13.06957425, 11.81423161,  6.54693987,  7.60463349,
        7.72415048, 11.1913598 ,  7.39599986,  8.16584537,  6.33517349,
       13.32557133,  7.06169391,  8.08320235,  5.24137909, 10.31584102,
        6.7166297 ])}
{'P': tensor([[2.0147e-02, 1.6068e-01, 1.2016e-01,  ..., 1.1245e-04, 2.6527e-04,
         3.6802e-04],
        [8.0172e-04, 2.5695e-02, 3.4268e-03,  ..., 2.6492e-04, 3.1012e-04,
         8.2704e-04],
        [1.2650e-03, 6.4198e-02, 2.1680e-02,  ..., 1.3630e-04, 1.8717e-04,
         7.0460e-04],
        ...,
        [1.4741e-05, 1.6576e-05, 1.3307e-05,  ..., 4.4866e-02, 4.2493e-05,
         5.6419e-05],
        [1.3103e-04, 2.3194e-04, 1.3789e-04,  ..., 5.0869e-03, 4.7983e-02,
         3.2547e-05],
        [3.9361e-04, 4.3178e-04, 4.5757e-04,  ..., 1.3561e-03, 3.7051e-05,
         6.3622e-01]]), 'X': tensor([[  6.2038,   2.4020,  -3.1692],
        [  5.6523,   3.9443,  -2.9449],
        [  5.8979,   3.3958,  -3.8713],
        [  5.1861,   4.7361,  -3.7312],
        [  5.0233,   4.9647,  -3.1788],
        [  5.8836,   3.4315,  -4.0327],
        [  8.3791,  10.5409,   5.5590],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.6093,  -1.9705,   6.4807],
        [ 11.7795,  -3.4697,   3.5624],
        [  2.7685, -10.8982,  -6.8928],
        [  6.2675,  12.1379,  -8.9720],
        [  2.3367,   7.1676,  -9.5263],
        [  2.3373,  13.7721,  -0.2442],
        [ -6.7992,   5.4991,   9.9186],
        [  7.7794,  11.0629,   8.9069],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.6952,  -4.8060,   8.8976],
        [ -3.0004,   7.6607,  -7.2640],
        [  4.5800,   5.5034,  -3.9432],
        [  1.3103,  -4.8032,  -0.9300],
        [ 12.2914,   3.8037,   6.0978],
        [  6.0054,  -1.0902,   8.3079],
        [  4.1637,  13.1898,  -1.9189],
        [  5.8836,   3.4315,  -4.0327],
        [  7.9141,  -8.6656,   9.6948],
        [ -9.0961,   1.6529,   3.1553],
        [ -1.3940, -10.7772,   6.7835],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.4267,  -5.3697,   2.4172],
        [  1.4489,   7.5256,   4.4529],
        [  3.4009,  -4.1874,  -5.9751],
        [  4.0591,   6.0202,  -1.6339],
        [  1.9312,  13.8649,   8.3563],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -8.4267,  -5.3697,   2.4172],
        [  0.1287, -11.0065,   6.5757],
        [  1.3403,  -4.8011,  -1.1698],
        [  0.3724,   7.7974,  -6.8722],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.6408,   7.4591,  -4.2294],
        [  0.7275,  14.0667,  -0.9815],
        [ -5.3328,  -8.9992,   7.3440],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.5770,   1.8512,  -0.6578],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.0305,   3.0296,   0.1375],
        [ -3.7964,  -9.9294,   7.7147],
        [  2.2373,  -4.6506,  -1.7334],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.0087,   4.0173,   9.1091],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.2905,   1.9366,   2.1169],
        [  6.2895,   0.0405,   5.7652],
        [ -9.3444,   0.5821,   6.9101],
        [  6.3569,   0.7591,   6.5011],
        [ -9.2508,   1.0572,   2.9978],
        [ -6.6895,   5.6051,   2.7899],
        [  1.8320,   7.3871,   5.1452],
        [  5.9548,  -1.2273,  -2.6520],
        [ 10.2712,   8.3707,  -4.9743],
        [  1.8766,   7.3694,   1.1887],
        [ 12.5763,   0.1333,  -9.2650],
        [  7.5309,  11.2617,   7.2656],
        [  5.8323,  -1.5228,   5.1132],
        [  1.4489,   7.5256,   4.4529],
        [  3.2701,   6.6338,   9.9209],
        [  1.1350,  14.0104,   0.6321],
        [  9.7328,   9.0902,  -0.8917],
        [  4.3554,   5.7393,  -9.0766],
        [  6.2247,  -0.3150,  -9.3657],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.5568,   6.0602,   9.3244],
        [  1.1350,  14.0104,   0.6321],
        [ -1.0407,   7.9166,  -6.6021],
        [  6.3549,   0.7096,   6.9198],
        [ -9.2508,   1.0572,   2.9978],
        [ 11.5456,   6.0861,   8.1101],
        [ -4.2040,   7.2486,   2.9875],
        [ 11.9936,  -2.8677,   6.0614],
        [ -3.3116, -10.1539,   7.5919],
        [  5.3860,  -2.3418,   1.0826],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.4996,   2.6335,  -6.3166],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.4378,  -1.8606,  -2.5668],
        [  4.4039,   5.6903,   8.7366],
        [  9.3560,   9.5391,   9.4943],
        [ 12.2589,  -1.9312,   9.8657],
        [  2.4932,   7.0903,  -5.4214],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.1203,  -2.7151,   6.2110],
        [  4.1999,  -3.6543,  -4.1998],
        [ -9.0604,   1.7707,   1.1066],
        [ -3.9456,  -9.8542,   0.7466],
        [ -6.8618,  -7.6322,   3.7027],
        [ -1.9714,   7.8563,  -1.8084],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.3261,   7.1726,   6.2522],
        [  0.9659,  -4.8144,  -0.9594],
        [ 12.5650,   2.0132,   4.6272],
        [ -5.9940,   6.1985,   5.9566],
        [  4.6231,   5.4556,   8.4973],
        [ -5.0550,   6.8231,  -9.2502]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.5568e-02, 3.5563e-02,  ..., 8.3604e-01, 7.5455e-01,
         5.8001e-01],
        [2.5568e-02, 0.0000e+00, 5.3289e-02,  ..., 8.1070e-01, 7.3493e-01,
         5.7346e-01],
        [3.5563e-02, 5.3289e-02, 0.0000e+00,  ..., 8.5839e-01, 7.8800e-01,
         5.5614e-01],
        ...,
        [8.3604e-01, 8.1070e-01, 8.5839e-01,  ..., 0.0000e+00, 4.4148e-01,
         9.4417e-01],
        [7.5455e-01, 7.3493e-01, 7.8800e-01,  ..., 4.4148e-01, 0.0000e+00,
         1.1181e+00],
        [5.8001e-01, 5.7346e-01, 5.5614e-01,  ..., 9.4417e-01, 1.1181e+00,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.65259775,  6.89245352,  6.80559669,  7.02326515,  7.06272256,
        6.81117802, 13.46553623, 11.75618513,  5.94535529, 12.27991991,
       11.24434374, 13.66051884,  7.53884104, 13.96905756,  8.74470218,
       13.52430176, 10.52357479,  4.85603886,  8.22728399,  7.15990676,
        4.978704  , 12.86648771,  6.10360073, 13.83139024,  6.81117802,
       11.73567759,  9.24501829, 10.86693708,  7.09887557,  9.99211663,
        7.66377554,  5.39450305,  7.26074925, 13.99877413, 10.60226614,
        9.99211663, 11.00726483,  4.98461983,  7.80625981, 10.01804223,
        9.75401449,  7.6374603 , 14.08549802, 10.4606265 , 12.00995898,
       12.7125079 ,  9.99211663,  6.74872981, 10.63038565,  5.16077764,
        9.7712321 ,  9.73579738,  8.95983586,  8.08320235,  6.58184474,
        6.28961988,  9.36256096,  6.4020404 ,  9.3109884 ,  8.7273569 ,
        7.61088125,  6.07992623, 13.25017211,  7.60463349, 12.5769659 ,
       13.54774355,  6.02779201,  7.66377554,  7.39599986, 14.05633441,
       13.31764181,  7.20482224,  6.23262289,  8.92098752, 13.0493368 ,
       14.05633441,  7.98469166,  6.39438044,  9.3109884 , 13.05148826,
        8.37954853, 12.33167382, 10.68031351,  5.87305005,  7.83486943,
       12.77402351, 13.77075174,  9.61942445,  7.19534146, 13.3614598 ,
       12.41012046,  7.5158489 ,  7.88677538,  5.79562691,  5.5671389 ,
        9.23178204, 10.61472636, 10.26327895,  8.09984231,  6.19633139,
        7.5403822 ,  4.91038884, 12.72524615,  8.62260327,  7.15100303,
        8.49161154])}
{'P': tensor([[2.6077e-02, 2.8336e-01, 1.7762e-01,  ..., 8.5098e-05, 5.1837e-04,
         1.9716e-04],
        [2.1090e-03, 4.0124e-02, 9.7057e-02,  ..., 1.1003e-04, 1.0582e-03,
         2.2321e-04],
        [2.7348e-03, 9.9829e-03, 5.0475e-02,  ..., 1.4670e-04, 1.1635e-03,
         3.1563e-04],
        ...,
        [7.2968e-03, 5.7462e-03, 6.6841e-03,  ..., 1.3275e-01, 3.6124e-05,
         2.2147e-03],
        [1.1397e-04, 1.2631e-04, 1.1629e-04,  ..., 4.2579e-05, 1.8087e-01,
         1.2251e-06],
        [8.6605e-04, 1.1065e-03, 1.7697e-03,  ..., 7.0179e-05, 6.9660e-05,
         2.2305e-01]]), 'X': tensor([[ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0159, 0.0480,  ..., 0.4644, 0.5761, 0.9034],
        [0.0159, 0.0000, 0.0616,  ..., 0.4784, 0.5612, 0.9192],
        [0.0480, 0.0616, 0.0000,  ..., 0.4168, 0.6041, 0.8634],
        ...,
        [0.4644, 0.4784, 0.4168,  ..., 0.0000, 0.9471, 0.5253],
        [0.5761, 0.5612, 0.6041,  ..., 0.9471, 0.0000, 1.4500],
        [0.9034, 0.9192, 0.8634,  ..., 0.5253, 1.4500, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.652742  ,  6.74686071,  6.80370254,  7.01475376,  7.18354885,
        6.81084724, 10.6910768 , 11.1341798 , 10.1371552 ,  9.3109884 ,
        7.39599986,  9.562174  ,  5.94909709,  7.06272256, 11.37939695,
       10.33712258,  9.61942445,  7.06372172, 13.6922425 , 13.27717982,
       11.27160035,  9.71392179, 10.76154256, 11.63383221,  5.79391825,
       11.85743425,  7.38844703, 13.09426142,  9.85384623,  8.85939626,
        7.5984809 , 10.48840476,  9.68324553, 13.18278606,  7.8788965 ,
       11.75807252, 14.02058621, 13.04217487, 11.42420508,  7.70897675,
        7.62509582,  8.92395381,  6.35029274, 12.67680439,  5.19577146,
        9.50913849,  8.1137452 , 11.83509372, 12.41366754, 11.23201619,
        8.89458273, 10.19036081,  8.7273569 , 10.80728788,  5.16205087,
        8.56889738,  4.89869205,  6.18668463, 10.7167285 ,  9.77777346,
       13.66991027,  5.81199422,  9.75401449, 11.51811603,  9.94173708,
       12.53345445,  9.10945678,  6.78325151,  4.78258092,  6.81961996,
       13.79676654, 11.10066126, 11.04629471,  5.10446653,  9.09963894,
       10.09979747,  6.2322769 ,  7.42512621,  6.1561646 , 11.42420508,
        6.02588517, 10.6910768 ,  7.5092834 ,  8.29022197, 13.04156075,
       10.92597196,  5.79562691,  6.41202986,  7.10378919,  9.78286733,
        6.00663314,  6.19707944, 12.76408562, 11.27160035,  8.24808832,
        6.7593917 , 10.41909211,  6.53290663,  7.38844703,  4.87083348,
       12.4013153 , 12.4013153 ,  9.53015376,  5.2287058 ,  8.29203356,
       13.0493368 ])}
{'P': tensor([[3.2399e-02, 1.3169e-01, 1.2984e-01,  ..., 1.2412e-04, 3.2248e-03,
         3.6479e-05],
        [8.0357e-04, 2.9016e-02, 1.1148e-01,  ..., 1.0148e-04, 1.8350e-03,
         8.0235e-06],
        [1.3670e-03, 6.4955e-03, 8.9494e-02,  ..., 2.6268e-04, 8.9810e-03,
         2.1046e-05],
        ...,
        [7.2295e-03, 4.1820e-03, 2.5262e-03,  ..., 1.6504e-01, 1.9218e-04,
         6.7065e-05],
        [3.6144e-04, 6.3083e-04, 1.3037e-03,  ..., 6.1807e-05, 1.0370e-01,
         2.4138e-05],
        [3.8613e-04, 3.1958e-04, 5.1309e-04,  ..., 2.2637e-06, 3.9330e-03,
         5.8633e-02]]), 'X': tensor([[  6.2022,   2.4092,   3.0349],
        [  5.3118,   4.5452,   4.2114],
        [  4.6992,   5.3690,   3.0935],
        [  4.5767,   5.5071,   3.9069],
        [  4.8432,   5.1968,   1.5006],
        [  4.2096,   5.8815,   1.9230],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.1892, -10.8220,  -6.8409],
        [  6.3246,   1.6696,  -0.1544],
        [  2.4356,  -4.5942,  -5.0808],
        [  6.0073,   3.0986,   6.7937],
        [  9.5203,   9.3483,   5.8781],
        [ 10.9875,   7.2207,  -8.8549],
        [ -0.0282,   7.8575,  -1.1746],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.7685, -10.8982,  -6.8928],
        [  4.5430,   5.5438,  -3.8669],
        [  3.7237,  -3.9966,  -5.0376],
        [ -1.8788, -10.6567,   3.7068],
        [ 10.4758,  -5.9539,   1.3667],
        [  9.6614,   9.1784,  -1.2087],
        [ -9.4714,  -1.3301,  -3.1998],
        [  2.2860,   7.1916,  -6.1011],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.1532,  -4.7422,  -7.0145],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.9967,  -4.8144,  -9.7370],
        [  7.0515,  11.6194,   8.4289],
        [ -8.6801,   2.7809,  -0.8443],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.2038,   2.4020,  -3.1692],
        [ -8.9606,  -4.0994,   1.7886],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 12.3357,   3.6001,   0.7903],
        [  5.6202,  -1.9508,   0.7471],
        [ 12.5717,   1.9266,   7.4903],
        [  0.9809,  -4.8145,   6.8706],
        [  3.4401,  13.4540,   4.6576],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.1658,  -3.4155,  -2.8880],
        [  5.3118,   4.5452,   4.2114],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.4501,  -1.7133,   2.2980],
        [  9.4925,   9.3811,   8.9842],
        [ -3.9700,   7.3453,  -4.1486],
        [ 11.2209,  -4.7092,   2.6117],
        [ -7.6734,   4.4980,  -5.4248],
        [ -5.9940,   6.1985,   5.9566],
        [ -2.2188,   7.8218,   8.6064],
        [  7.1441,  11.5529,   2.3232],
        [  1.7903,  -4.7467,   0.1168],
        [  0.8700,  -4.8132,  -7.8042],
        [ -2.5683, -10.4433,  -2.9182],
        [  4.5767,   5.5071,   3.9069],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.4297,  -5.3637,  -9.3962],
        [  6.1752,   2.5242,   3.5874],
        [ 11.4495,   6.3026,   6.8257],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.3305,   1.6119,   1.1538],
        [  3.7298,  13.3537,  -9.4653],
        [ -5.6651,   6.4380,   1.3660],
        [  1.3705,  13.9723,   7.9266],
        [ 12.5717,   1.9266,   7.4903],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.8407,  11.7666,   3.0045],
        [  3.0719, -10.8449,  -4.1673],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.4996,   2.6335,  -6.3166],
        [  2.5636, -10.9293,   9.6519],
        [ -8.9033,   2.2329,  -8.8912],
        [  9.0051,   9.9234,   5.4411],
        [  0.6757, -11.0361,  -1.3047],
        [ -6.3489,   5.9117,   6.5161],
        [  3.3167,   6.6022,   6.3554],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.3551,   0.7149,   1.3344],
        [  3.2226,  -4.2802,  -4.0428],
        [  9.4554,   9.4245,   9.8800],
        [  2.1824,  -4.6647,   1.1985],
        [  5.9003,  12.3550,   2.1917],
        [ -9.4735,  -1.2620,  -3.6742],
        [  6.0047,   3.1062,  -6.7914],
        [  9.1820,   9.7334,  -2.7932],
        [  1.1274,  14.0116,   8.1753],
        [ -8.8400,   2.3991,   1.3457],
        [  5.7784,  -1.6401,   9.5587],
        [  3.9022,  13.2906,   5.5788],
        [ -1.9487,   7.8590,  -3.3330],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.1140,  -3.7220,  -0.1273],
        [ -0.1626, -10.9796,   6.4629],
        [  7.0515,  11.6194,   8.4289],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -5.5525,   6.5146,  -4.4984],
        [  6.9314,  -9.3435,  -1.2729],
        [  2.6614,  -4.5192,  -9.3641],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.1931,  -4.7491,   7.4385],
        [  2.5375,   7.0676,  -4.3490],
        [ 12.5763,   0.1333,  -9.2650]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0988, 0.0660,  ..., 0.2548, 0.5369, 0.8663],
        [0.0988, 0.0000, 0.0857,  ..., 0.1562, 0.6207, 0.9644],
        [0.0660, 0.0857, 0.0000,  ..., 0.2247, 0.5370, 0.8953],
        ...,
        [0.2548, 0.1562, 0.2247,  ..., 0.0000, 0.7554, 1.1182],
        [0.5369, 0.6207, 0.5370,  ..., 0.7554, 0.0000, 0.4623],
        [0.8663, 0.9644, 0.8953,  ..., 1.1182, 0.4623, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.65368551,  6.99097013,  7.13501264,  7.16058587,  7.10378919,
        7.23276582,  5.35774573, 11.28215429,  6.54127431,  5.19986797,
        6.7593917 , 13.3426541 , 13.14777433,  7.8575648 , 12.41366754,
       11.24434374,  7.16747878,  5.46244988, 10.82106505, 12.0495521 ,
       13.32613445,  9.5642993 ,  7.54620849, 11.81423161,  4.74468406,
       12.25806559,  4.91652556, 13.59170809,  9.11473391,  8.94395628,
        6.65259775,  9.85384623, 10.39942497, 12.05300943, 12.85032836,
        5.94909709, 12.71843462,  4.91336994, 13.88683436, 10.14436912,
       12.04924159,  9.78147542,  6.99097013,  9.96152788,  9.60412735,
       13.34587027,  8.34949383, 12.16901615,  8.89458273,  8.62260327,
        8.1303945 , 13.58333857,  5.07306274,  4.89121805, 10.75443106,
        7.16058587,  9.75401449,  9.97220799,  9.99145096,  6.67121748,
       13.06957425, 11.74901778,  6.53250937, 13.86480175,  8.57560275,
       14.0393949 , 12.71843462,  8.08320235, 13.6105595 , 11.27160035,
        9.97822473, 12.77402351, 11.22596994,  9.17905491, 13.40024224,
       11.05672136,  8.67502348,  7.38844703, 10.06023665,  6.39520397,
        5.35774573, 13.35013374,  5.14999799, 13.69162795,  9.55720651,
        6.76056625, 13.38091121, 14.05687815,  9.15976746,  6.00663314,
       13.85158417,  8.09704052, 11.64746882,  9.75401449,  5.54776934,
       10.98076873, 13.59170809,  9.71392179,  9.78286733,  8.55982105,
       11.63383221,  5.24461068, 12.0495521 ,  4.75303127,  7.5092834 ,
       12.5769659 ])}
{'P': tensor([[2.5154e-02, 2.8716e-01, 2.3273e-01,  ..., 6.8899e-06, 2.6355e-06,
         2.1174e-05],
        [3.1032e-03, 6.3709e-02, 3.2091e-03,  ..., 1.6990e-05, 6.4060e-06,
         6.1226e-05],
        [2.7745e-03, 4.3761e-01, 3.2280e-02,  ..., 8.7949e-06, 3.3904e-06,
         2.8485e-05],
        ...,
        [1.8863e-04, 1.3957e-04, 1.7191e-04,  ..., 3.7938e-02, 4.8909e-03,
         2.2020e-05],
        [1.6930e-04, 1.3467e-04, 1.6104e-04,  ..., 1.5468e-03, 9.2941e-02,
         2.2361e-05],
        [4.4022e-05, 3.5925e-05, 4.1236e-05,  ..., 6.1403e-02, 3.7099e-03,
         4.4035e-02]]), 'X': tensor([[  6.1929,   2.4501,  -6.9352],
        [  6.0047,   3.1062,  -6.7914],
        [  6.1372,   2.6726,  -6.7885],
        [  5.9655,   3.2172,  -7.5953],
        [  5.1292,   4.8183,  -5.7997],
        [  5.0190,   4.9705,  -7.9752],
        [  3.7992,  -3.9474,  -8.9019],
        [ -7.2986,  -7.1225,  -3.0814],
        [  4.4039,   5.6903,   8.7366],
        [ -3.3116, -10.1539,   7.5919],
        [  5.7485,  -9.9811,  -3.2355],
        [  8.0381,  -8.5690,  -8.9523],
        [ -8.0330,  -6.0765,   6.7945],
        [ 11.4495,   6.3026,   6.8257],
        [ -9.4753,  -0.8012,  -5.2050],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.5965,  -4.0754,  -3.5035],
        [  3.9913,  -3.8139,  -6.8311],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.5813,  -4.5471,   5.3597],
        [ -9.3604,   0.4855,  -3.0158],
        [  8.6420,  -8.0580,   6.5781],
        [  2.3373,  -4.6232,  -5.2246],
        [  1.6408,   7.4591,  -4.2294],
        [  6.3031,   1.8477,  -7.1706],
        [  3.8182,  -3.9347,  -1.7961],
        [  3.3943,   6.5483,   1.0790],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.7000,  -4.7609,   2.7935],
        [  3.2701,   6.6338,   9.9209],
        [  1.1644, -11.0394,   7.8511],
        [ 12.3453,  -1.5462,   7.0212],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -9.1960,  -3.2964,   1.6921],
        [ -3.1671,   7.6158,   6.0765],
        [  5.1890,   4.7318,   0.9515],
        [ -9.2508,   1.0572,   2.9978],
        [ 11.4495,   6.3026,   6.8257],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.2712,   8.3707,  -4.9743],
        [ 11.8951,  -3.1570,   7.0742],
        [ -9.3464,   0.5708,  -1.6397],
        [ -7.2986,  -7.1225,  -3.0814],
        [  6.8093,  -9.4178,   8.4244],
        [  6.2038,   2.4020,  -3.1692],
        [  6.0345,   3.0175,  -1.4424],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.3604,   0.4855,  -3.0158],
        [  4.4290,  -3.4596,   1.4181],
        [ 12.0042,  -2.8350,   2.6864],
        [  5.3860,  -2.3418,   1.0826],
        [ -8.3717,  -5.4769,  -9.2228],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.1820,   9.7334,  -2.7932],
        [ -7.5720,   4.6310,   2.1717],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.7544,   2.6096,   3.3146],
        [  2.5636, -10.9293,   9.6519],
        [  6.7565,  11.8237,  -4.9882],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.5294,  -2.1106,   9.0626],
        [ 11.1469,   6.9237,   1.6160],
        [  5.9526,   3.2525,  -4.3821],
        [  6.2902,   0.0451,   3.7116],
        [ -2.1533, -10.5778,   1.0112],
        [ -2.5161,   7.7699,  -9.3531],
        [ 10.0619,   8.6627,  -9.2884],
        [  8.1528,  10.7452,   9.1493],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.6059,   0.9597,  -6.3264],
        [ -6.1899,  -8.3011,  -7.9269],
        [  1.3767, -11.0341,   9.9415],
        [  7.0515,  11.6194,   8.4289],
        [ -1.0482, -10.8487,  -5.9066],
        [  9.1620,  -7.5573,  -9.9882],
        [  3.4401,  13.4540,   4.6576],
        [ -4.0868,   7.2981,   8.1874],
        [ 12.1379,   4.4125,  -3.8833],
        [  1.7815,  -4.7481,   4.4289],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.3516,   0.6425,   0.3425],
        [  5.6022,  -1.9833,   8.3081],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.3705,   0.4208,  -3.5815],
        [ -9.4630,  -1.5145,   0.2983],
        [  0.1497,   7.8334,  -5.4412],
        [ -3.6381,   7.4683,   7.3544],
        [  5.0617,  12.7957,   8.3544],
        [  5.8617,   3.4853,  -9.8080],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.8325,  -1.5222,  -2.1341],
        [ -1.6275, -10.7221,   3.2376],
        [  4.2096,   5.8815,   1.9230],
        [  3.9913,  -3.8139,  -6.8311],
        [  0.5600, -11.0321,  -0.8660],
        [  1.3403,  -4.8011,  -1.1698],
        [  0.3354,  -4.7708,  -4.9125],
        [ 12.3759,  -1.3943,   6.4659],
        [  7.7794,  11.0629,   8.9069],
        [  6.3549,   0.7096,   6.9198],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.9433,  -9.3362,  -0.3421],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.6275, -10.7221,   3.2376],
        [ -8.9624,  -4.0944,  -4.9622]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0138, 0.0064,  ..., 0.8406, 1.0336, 1.0179],
        [0.0138, 0.0000, 0.0084,  ..., 0.8278, 1.0204, 1.0056],
        [0.0064, 0.0084, 0.0000,  ..., 0.8362, 1.0273, 1.0138],
        ...,
        [0.8406, 0.8278, 0.8362,  ..., 0.0000, 0.6195, 0.1915],
        [1.0336, 1.0204, 1.0273,  ..., 0.6195, 0.0000, 0.7236],
        [1.0179, 1.0056, 1.0138,  ..., 0.1915, 0.7236, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.65991877,  6.76056625,  6.69389431,  6.77777606,  7.03733775,
        7.06372172,  5.47866221, 10.1979669 ,  7.19534146, 10.68031351,
       11.51811603, 11.74901778, 10.07238463, 13.06957425,  9.50913849,
        5.13182506,  5.43544626,  5.5205189 , 12.7125079 ,  5.2287058 ,
        9.37296102, 11.8159254 ,  5.18047638,  7.6374603 ,  6.56833864,
        5.48276038,  7.37576154, 12.04924159,  5.05534609,  7.39599986,
       11.10066126, 12.44177641, 12.25806559,  9.76896904,  8.24808832,
        7.02253235,  9.3109884 , 13.06957425,  7.71075226, 13.25017211,
       12.30695098,  9.36378438, 10.1979669 , 11.62158554,  6.65259775,
        6.74686071,  9.78147542,  9.37296102,  5.62005629, 12.3344536 ,
        5.87305005, 10.00410625, 11.9182956 , 13.38091121,  8.87587457,
        5.54776934,  9.13507321, 11.22596994, 13.61801526, 10.92597196,
        5.91854397, 13.12218013,  6.78325151,  6.29034958, 10.794755  ,
        8.16714836, 13.27717982, 13.48809595,  5.16611782, 12.64235359,
       10.35485331, 11.11969815, 13.59170809, 10.8992524 , 11.87665501,
       13.88683436,  8.36445758, 12.91505111,  5.0713364 , 12.07225469,
        6.38399789,  5.94293185, 13.01187484,  9.37989631,  9.58347456,
        7.83486943,  8.30729352, 13.76047999,  6.81961996, 13.27994304,
        9.85384623,  6.02789634, 10.84493732,  7.23276582,  5.5205189 ,
       11.04629471,  4.98461983,  4.78258092, 12.45418236, 13.52430176,
        6.39438044, 12.00995898, 11.63502679, 10.6910768 , 10.84493732,
        9.85330646])}
{'P': tensor([[3.1071e-02, 1.8962e-01, 1.5898e-01,  ..., 5.3654e-05, 1.9700e-03,
         2.5018e-03],
        [1.8378e-03, 8.2080e-02, 7.2546e-03,  ..., 8.1945e-05, 1.4144e-02,
         1.6599e-02],
        [1.3251e-03, 1.1236e-01, 3.0383e-02,  ..., 2.7590e-05, 4.1971e-03,
         5.1163e-03],
        ...,
        [1.4055e-03, 1.0420e-03, 7.8178e-04,  ..., 9.4038e-02, 2.7857e-05,
         3.4361e-05],
        [7.4040e-05, 1.8964e-04, 1.3276e-04,  ..., 7.2980e-06, 2.9135e-01,
         5.3347e-04],
        [1.3455e-06, 3.7355e-06, 2.5293e-06,  ..., 1.1341e-07, 9.8274e-01,
         4.2336e-03]]), 'X': tensor([[  6.1755,   2.5233,  -2.0042],
        [  5.2199,   4.6861,  -1.9304],
        [  5.6523,   3.9443,  -2.9449],
        [  6.0345,   3.0175,  -1.4424],
        [  5.0233,   4.9647,  -3.1788],
        [  4.4633,   5.6287,  -0.7754],
        [  5.8889,  -1.3916,   8.8993],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.5318,  13.7231,   6.3692],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 11.9808,   4.9340,  -3.3715],
        [  1.9505,  -4.7171,   9.9868],
        [ 10.3251,   8.2928,  -3.0006],
        [  0.7685,  -4.8098,  -2.9310],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.4690,  -0.8528,  -9.3374],
        [  5.3924, -10.1396,  -4.0863],
        [  0.8572,  14.0501,  -8.3364],
        [  4.1537,   5.9340,  -6.0423],
        [  6.1929,   2.4501,  -6.9352],
        [  0.8572,  14.0501,  -8.3364],
        [  7.1697,  11.5343,  -0.7276],
        [ -4.3963,   7.1629,   4.5263],
        [  2.3367,   7.1676,  -9.5263],
        [ -6.6998,   5.5954,   0.0823],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.5318,  13.7231,   6.3692],
        [ -9.2707,   0.9661,   4.4863],
        [ -8.0608,   3.9362,  -5.9567],
        [  6.2902,   0.0451,   3.7116],
        [  3.6249, -10.7255,   5.6605],
        [  9.5018,  -7.1945,  -1.1892],
        [ -9.4196,   0.0478,   9.4140],
        [ -8.9942,  -3.9990,   6.0448],
        [  2.9461,  -4.4077,  -0.4831],
        [  1.9503,  -4.7172,  -6.2313],
        [ -2.0445,   7.8469,  -0.5857],
        [  7.1697,  11.5343,  -0.7276],
        [  3.9591, -10.6391,  -6.2557],
        [ -6.4464,   5.8272,   7.5038],
        [  7.0515,  11.6194,   8.4289],
        [  2.8474,   6.8979,   6.6457],
        [ -5.6169,  -8.7844,   8.9057],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.2555,   2.1474,   4.1685],
        [ 11.6019,   5.9540,  -5.1715],
        [ -0.3432,   7.8900,   7.1805],
        [  9.0652,   9.8597,  -3.5949],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.8093,  -9.4178,   8.4244],
        [ -1.8367,   7.8718,  -8.6523],
        [  1.3705,  13.9723,   7.9266],
        [ -3.2486,   7.5925,  -1.3719],
        [  6.2905,   1.9366,   2.1169],
        [  9.9756,  -6.6330,  -3.3429],
        [ 12.5919,   0.4035,  -5.8460],
        [  9.9756,  -6.6330,  -3.3429],
        [ -5.1046,  -9.1610,   2.3418],
        [ 12.2592,   3.9425,   2.0383],
        [  9.3424,  -7.3685,  -3.6807],
        [  8.0381,  -8.5690,  -8.9523],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.3444,   0.5821,   6.9101],
        [  5.6672,  -1.8631,   8.1020],
        [  3.9895,   6.0816,  -6.5235],
        [  6.3360,   0.4258,  -9.2509],
        [  2.5620,   7.0549,   8.4825],
        [ -3.3999, -10.1152,  -1.2088],
        [  1.3767, -11.0341,   9.9415],
        [  5.0233,   4.9647,  -3.1788],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -9.1231,  -3.5739,   8.4708],
        [ -3.7964,  -9.9294,   7.7147],
        [ 11.5935,   5.9740,  -4.3952],
        [  6.2883,   0.0325,   5.2692],
        [ -8.0534,  -6.0429,   2.4583],
        [  0.7145,  -4.8071,  -2.3766],
        [ -7.6525,  -6.6534,   5.1721],
        [  0.1047, -11.0046,   0.5188],
        [  0.9788,  -4.8145,  -3.8661],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -0.8675, -10.8815,   6.8561],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.6451,  -4.5250,  -0.5565],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.4505,  -1.7081,  -0.7039],
        [  1.9241,   7.3503,   7.9004],
        [ -8.6801,   2.7809,  -0.8443],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.1623,  -3.4288,  -4.9961],
        [  1.9503,  -4.7172,  -6.2313],
        [  6.6319,  11.9066,   3.5165],
        [ 11.4743,   6.2478,  -2.1100],
        [ -8.5900,   2.9767,   7.4811],
        [ -9.4430,  -0.1897,   4.8383],
        [ 11.6634,   5.8050,   9.1590],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.9433,  -9.3362,  -0.3421],
        [  9.3607,   9.5337,  -4.5412],
        [  6.2555,   2.1474,   4.1685],
        [  1.3705,  13.9723,   7.9266],
        [  8.1528,  10.7452,   9.1493],
        [  4.5340,  -3.3629,   8.9652],
        [  0.4119,  14.1019,  -4.1870],
        [  1.4150,  13.9647,  -4.0915]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0328, 0.0634,  ..., 0.6706, 0.2176, 0.3171],
        [0.0328, 0.0000, 0.0747,  ..., 0.6605, 0.2414, 0.3466],
        [0.0634, 0.0747, 0.0000,  ..., 0.7326, 0.1711, 0.2864],
        ...,
        [0.6706, 0.6605, 0.7326,  ..., 0.0000, 0.8737, 0.9271],
        [0.2176, 0.2414, 0.1711,  ..., 0.8737, 0.0000, 0.1325],
        [0.3171, 0.3466, 0.2864,  ..., 0.9271, 0.1325, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.67108662,  7.01475376,  6.89245352,  6.74686071,  7.06272256,
        7.18354885,  6.05113217,  5.41348121, 13.95472331, 10.52587243,
       12.95702647,  5.10446653, 13.2430377 ,  4.87083348,  8.06643487,
       12.4980845 , 11.48434767, 14.07623266,  7.24326992,  6.65991877,
       14.07623266, 13.58101794,  8.40445872,  7.53884104,  8.72896381,
       13.25336001, 13.95472331,  9.32094055,  8.97052115,  6.29034958,
       11.32149259, 11.9182956 ,  9.41970031,  9.84314712,  5.30158497,
        5.10444097,  8.10886021, 13.58101794, 11.35182589,  8.68979052,
       13.59170809,  7.46248947, 10.42667093,  9.86089579,  6.61386436,
       13.04051744,  7.89745107, 13.39372426, 13.20597932, 11.62158554,
        8.08320235, 14.0393949 ,  8.25828409,  6.58184474, 11.97957425,
       12.59840193, 11.97957425, 10.48721382, 12.87752052, 11.89855267,
       11.74901778,  5.0713364 ,  9.36256096,  5.96556248,  7.27340834,
        6.35029274,  7.50564223, 10.67132524, 11.11969815,  7.06272256,
       10.09979747, 12.05395056,  9.79814997, 10.63038565, 13.04217487,
        6.28835101, 10.06850677,  4.85995002, 10.14044489, 11.00508681,
        4.91296205,  9.5642993 , 10.91601655, 10.31613269,  5.24137909,
        7.98469166,  9.60359158,  7.59795445,  9.11473391,  5.05981914,
        9.78286733,  5.10444097, 13.62896593, 13.06498554,  9.09119351,
        9.44486609, 13.02818702,  8.62631801, 11.63502679, 13.36092624,
        6.61386436, 14.0393949 , 13.48809595,  5.64500512, 14.10796466,
       14.03618649])}
{'P': tensor([[3.7664e-02, 2.1422e-01, 1.5245e-01,  ..., 4.4297e-05, 8.5963e-06,
         1.5823e-05],
        [9.0227e-04, 3.5082e-02, 1.3479e-01,  ..., 6.7781e-05, 8.6794e-06,
         6.2318e-06],
        [1.0212e-03, 4.5925e-03, 6.3274e-02,  ..., 2.0805e-04, 2.1818e-05,
         9.6888e-06],
        ...,
        [2.7168e-05, 1.6885e-05, 1.7435e-05,  ..., 5.9842e-02, 8.9372e-02,
         1.0096e-04],
        [2.3206e-05, 9.6144e-06, 8.5334e-06,  ..., 1.7637e-05, 2.1727e-02,
         1.1800e-03],
        [8.8350e-04, 3.3376e-04, 2.8966e-04,  ..., 8.7605e-07, 1.1658e-06,
         1.9277e-02]]), 'X': tensor([[ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0600, 0.0715,  ..., 1.0684, 1.1464, 0.6267],
        [0.0600, 0.0000, 0.0857,  ..., 1.0643, 1.1457, 0.6613],
        [0.0715, 0.0857, 0.0000,  ..., 0.9970, 1.0755, 0.5757],
        ...,
        [1.0684, 1.0643, 0.9970,  ..., 0.0000, 0.1006, 0.6973],
        [1.1464, 1.1457, 1.0755,  ..., 0.1006, 0.0000, 0.7302],
        [0.6267, 0.6613, 0.5757,  ..., 0.6973, 0.7302, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.67121748,  6.99097013,  7.13501264,  7.16058587,  7.23276582,
        7.07902036, 12.25806559,  7.73692436, 10.30972193,  7.98469166,
       13.89264822,  7.71075226,  7.50564223,  6.61386436,  5.90035324,
        5.76165934, 10.61417181, 11.83570132, 11.89105259,  7.50035355,
       11.35345345,  6.19707944, 10.60226614,  9.85330646, 13.0493368 ,
        8.86082124,  6.46560181,  9.48337063, 12.07818871, 11.10066126,
       11.35182589,  6.41202986, 10.06703358,  5.16205087,  6.12886024,
        7.50917551,  7.50564223, 10.68031351,  5.05534609,  9.84314712,
        8.8737556 ,  7.58378655, 11.97238624,  5.29009836,  7.26074925,
       10.30972193,  9.38232121,  5.70405525, 12.59969936,  5.24461068,
        6.60752373,  6.49538841,  8.56889738,  8.67227386,  6.24614905,
       11.1341798 ,  8.98765644,  6.76661595, 13.14690137,  9.87720102,
       13.79676654,  5.30158497,  5.14999799, 11.27656488, 13.79762415,
        6.42689705, 10.1979669 , 10.60226614, 12.53285633, 13.77604293,
        5.84533345, 12.52372629,  8.68979052,  6.21275016,  7.54127425,
        9.43112957,  6.83083863, 10.68416401, 13.37121923, 10.67322527,
        5.94535529, 13.40024224,  9.91668182,  8.4486771 ,  5.5671389 ,
       12.52372629,  7.10378919, 10.53368644,  8.91551231, 11.21703654,
       10.53368644,  6.54127431, 11.10066126,  6.81117802,  8.29022197,
        7.50500064, 11.48434767, 10.33712258, 10.87409005,  4.78258092,
       11.31276109, 11.1913598 ,  5.92952538,  9.5014146 , 10.31613269,
       11.80092578])}
{'P': tensor([[4.1895e-02, 2.6610e-01, 1.6501e-01,  ..., 2.2095e-05, 1.9347e-04,
         4.6272e-04],
        [9.4230e-04, 3.8509e-02, 3.9289e-01,  ..., 3.3064e-05, 1.1680e-04,
         1.0339e-03],
        [1.1812e-03, 2.7360e-03, 7.9792e-02,  ..., 1.0648e-04, 1.9396e-04,
         3.7064e-03],
        ...,
        [1.5567e-05, 1.2381e-05, 1.3159e-05,  ..., 1.3836e-01, 2.5159e-05,
         3.7068e-04],
        [1.9831e-02, 1.0210e-02, 6.2740e-03,  ..., 8.3266e-06, 1.6076e-01,
         1.7359e-04],
        [1.7801e-05, 1.6520e-05, 1.7338e-05,  ..., 3.2805e-04, 9.5964e-05,
         2.6729e-02]]), 'X': tensor([[ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0432, 0.0715,  ..., 1.1921, 0.2114, 0.6608],
        [0.0432, 0.0000, 0.0380,  ..., 1.1839, 0.1851, 0.6291],
        [0.0715, 0.0380, 0.0000,  ..., 1.1491, 0.2038, 0.5921],
        ...,
        [1.1921, 1.1839, 1.1491,  ..., 0.0000, 1.3268, 0.7677],
        [0.2114, 0.1851, 0.2038,  ..., 1.3268, 0.0000, 0.6916],
        [0.6608, 0.6291, 0.5921,  ..., 0.7677, 0.6916, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.67218687,  6.99097013,  7.16058587,  7.13501264,  7.07902036,
        7.23276582,  7.80625981,  5.1805345 ,  4.91652556,  6.02588517,
       12.35421461,  9.53015376,  5.84533345,  5.0209005 , 13.32645552,
        6.81961996,  8.85572292, 13.89886469,  8.97052115, 12.59969936,
        5.2287058 ,  9.58347456, 10.63038565,  5.39723241, 10.54658597,
       12.26997165, 10.14436912, 13.41713661,  5.5671389 ,  5.5366698 ,
        9.60080639, 13.06957425,  7.28793663,  6.39520397, 12.27991991,
       13.61801526, 10.7167285 ,  8.41171369, 13.32557133, 12.10222204,
        9.25336371,  7.62509582, 13.76047999, 12.52875793, 11.92133129,
       10.87409005,  5.36061974,  9.15253344, 10.51437947, 10.41909211,
        6.19633139,  9.09119351, 10.33958639, 13.45124917,  7.63726134,
       11.67875855,  6.74686071, 11.83570132,  8.27853076, 13.37121923,
        5.05444288, 13.94008948, 10.37698435,  8.30729352,  6.38105015,
       12.41366754,  5.62005629, 10.1371552 , 13.04156075,  5.07306274,
       11.63502679,  5.05989547,  5.13182506,  7.24326992,  4.74770899,
       10.33958639,  7.50500064, 13.744387  ,  9.23071698, 11.75438006,
        8.85572292,  7.69752262, 13.3426541 ,  5.88782913,  6.36004393,
        7.63726134,  9.55593374,  5.81481044, 10.14436912,  6.12886024,
        5.43544626,  7.5403822 , 10.74655346,  6.3949043 ,  6.35029274,
        5.39723241,  6.55800267,  8.98765644, 11.28895489, 14.09176259,
       12.27991991, 10.54658597,  9.32094055,  9.17905491,  5.29755487,
        8.90405823])}
{'P': tensor([[2.4843e-02, 3.3679e-01, 1.1329e-01,  ..., 6.7687e-06, 1.1136e-03,
         1.0226e-05],
        [1.8661e-03, 3.7047e-02, 1.8673e-01,  ..., 9.8796e-06, 1.1820e-03,
         1.2851e-05],
        [7.8963e-04, 1.0044e-03, 3.1259e-02,  ..., 8.8639e-06, 2.2491e-04,
         6.4342e-06],
        ...,
        [4.0040e-04, 3.2508e-04, 1.5081e-04,  ..., 8.8365e-02, 8.6018e-05,
         3.2238e-02],
        [1.0264e-03, 1.4581e-03, 5.7174e-03,  ..., 2.7953e-07, 1.8652e-02,
         7.0516e-07],
        [4.0795e-04, 3.4420e-04, 1.8817e-04,  ..., 1.0370e-06, 1.2730e-03,
         7.4145e-03]]), 'X': tensor([[ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0084, 0.0558,  ..., 0.8362, 0.2235, 0.3853],
        [0.0084, 0.0000, 0.0490,  ..., 0.8278, 0.2308, 0.3788],
        [0.0558, 0.0490, 0.0000,  ..., 0.7914, 0.2550, 0.3298],
        ...,
        [0.8362, 0.8278, 0.7914,  ..., 0.0000, 1.0419, 0.6057],
        [0.2235, 0.2308, 0.2550,  ..., 1.0419, 0.0000, 0.5140],
        [0.3853, 0.3788, 0.3298,  ..., 0.6057, 0.5140, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.69389431,  6.76056625,  7.03733775,  7.06372172,  6.77777606,
        7.24326992,  5.29009836,  8.68832398,  8.56889738, 10.80728788,
       13.41713661, 11.70016217, 11.83509372, 11.70168755, 13.40850328,
       12.91505111,  6.12886024, 10.41909211, 10.06850677,  8.66273171,
        8.12580976,  6.81084724,  8.67502348,  8.27853076,  7.09887557,
        7.06169391,  8.10886021, 10.05558893,  5.35774573, 12.04924159,
       13.32613445, 12.9089559 , 10.14436912, 12.4217523 ,  9.88310614,
        7.69752262,  8.83725081,  7.46248947, 13.44593519,  9.27474386,
       11.05413545, 11.9120321 , 11.97238624, 13.37121923, 13.15525803,
        6.39438044,  4.79003361, 11.89855267, 13.77222752, 12.1460931 ,
        5.46244988,  5.91854397, 10.39942497, 10.98076873, 12.52875793,
       13.6829843 ,  8.30125201, 11.04953767,  8.85939626, 13.37121923,
        6.89245352,  5.16611782, 10.09979747,  4.98989906,  9.73579738,
        5.23878371,  9.04255655, 13.0493368 ,  5.81481044,  9.76896904,
        7.15990676, 12.17451685,  9.60359158,  5.82345867, 10.74409834,
        6.42439344, 10.4606265 , 12.67576792, 13.61801526,  9.14044984,
        6.5618968 , 12.7125079 , 13.32374056, 13.2778378 ,  5.16077764,
       11.10066126,  6.65368551,  7.63726134, 12.30858677,  7.24326992,
       13.79676654,  6.5618968 , 11.85743425,  8.01009083, 13.66051884,
       13.86480175, 12.1460931 , 12.3344536 ,  5.52416368, 13.83139024,
        7.83486943, 11.69264243,  7.67939546, 10.6910768 , 12.95702647,
       11.75438006])}
{'P': tensor([[1.7672e-02, 2.3747e-01, 1.6793e-01,  ..., 1.3082e-04, 1.1005e-02,
         2.4451e-05],
        [2.0102e-03, 3.9688e-02, 1.3094e-01,  ..., 2.8587e-04, 3.0125e-02,
         6.1788e-05],
        [3.3758e-03, 1.2658e-02, 5.8407e-02,  ..., 4.9170e-04, 5.2095e-02,
         1.0595e-04],
        ...,
        [2.4915e-05, 2.7666e-05, 3.2831e-05,  ..., 2.1949e-02, 3.3782e-04,
         8.6442e-04],
        [5.8232e-04, 6.5268e-04, 6.8823e-04,  ..., 6.2713e-04, 1.5190e-01,
         3.8828e-03],
        [2.4580e-05, 2.3429e-05, 2.4830e-05,  ..., 6.3215e-05, 4.6246e-05,
         1.2492e-01]]), 'X': tensor([[ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.8422e-03, 1.7102e-02,  ..., 8.3097e-01, 2.3062e-01,
         1.0207e+00],
        [7.8422e-03, 0.0000e+00, 1.5935e-02,  ..., 8.3140e-01, 2.2479e-01,
         1.0162e+00],
        [1.7102e-02, 1.5935e-02, 0.0000e+00,  ..., 8.1548e-01, 2.1551e-01,
         1.0041e+00],
        ...,
        [8.3097e-01, 8.3140e-01, 8.1548e-01,  ..., 0.0000e+00, 7.1143e-01,
         6.3388e-01],
        [2.3062e-01, 2.2479e-01, 2.1551e-01,  ..., 7.1143e-01, 0.0000e+00,
         7.9814e-01],
        [1.0207e+00, 1.0162e+00, 1.0041e+00,  ..., 6.3388e-01, 7.9814e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.7166297 ,  6.78325151,  6.81117802,  6.83083863,  6.80559669,
        6.76661595,  6.53250937, 10.67132524,  8.68832398,  8.92549752,
       10.00794468, 13.13888841, 12.11735334,  6.39438044, 12.49317563,
       12.3828957 ,  8.29203356,  9.43737523, 10.8207297 , 13.83139024,
       12.25806559, 10.14436912, 11.99633819,  8.0550286 ,  5.1805345 ,
        7.53884104,  5.36061974,  8.30125201, 12.3828957 , 12.58006038,
        5.19577146, 13.61801526,  7.85919687, 12.77402351, 11.73567759,
       10.5575601 ,  6.53290663, 13.97857591,  5.43544626,  5.91222346,
        9.60412735, 10.4977132 ,  5.19577146, 11.87665501,  9.18358562,
       10.91805683, 13.3607379 ,  6.54615887, 13.79762415, 10.42667093,
       10.6910768 , 13.744387  ,  7.5984809 ,  8.89301516, 10.66631784,
        5.43544626, 13.61801526, 13.05148826, 11.67875855, 12.33167382,
        6.29034958, 13.77604293,  7.92200342,  7.27759233,  7.57946459,
       14.02532758,  4.91038884,  6.46560181,  4.978704  ,  6.60752373,
        6.46356806, 14.03618649,  9.17595899, 11.32149259,  6.60752373,
       10.6910768 , 13.3607379 ,  9.79039371,  7.73151309, 13.09426142,
        5.60064319,  8.10886021, 12.3828957 ,  7.56617037, 10.01804223,
       10.95593729,  9.75075273,  9.79039371,  7.19513806,  7.56617037,
       11.83570132, 10.54658597, 13.38091121, 13.40850328, 12.67680439,
       12.12448912,  8.24808832,  9.10945678,  5.5671389 , 13.3426541 ,
        7.46053281, 10.48721382,  6.1463896 ,  7.39004226,  7.71075226,
        9.10945678])}
{'P': tensor([[2.9623e-02, 2.0558e-01, 1.2652e-01,  ..., 1.8629e-05, 1.0635e-03,
         4.6354e-06],
        [6.6353e-04, 2.3161e-02, 9.6488e-02,  ..., 7.1843e-06, 1.4197e-03,
         2.4970e-06],
        [4.5003e-04, 1.5599e-03, 2.6527e-02,  ..., 5.1363e-06, 1.0642e-03,
         3.0054e-06],
        ...,
        [7.0512e-03, 3.7645e-03, 1.7505e-03,  ..., 5.9146e-02, 1.1681e-04,
         9.1770e-06],
        [2.8884e-06, 7.5362e-06, 1.2262e-05,  ..., 4.4293e-08, 8.4156e-03,
         1.6885e-08],
        [5.1519e-04, 2.0668e-04, 1.5824e-04,  ..., 2.1053e-03, 9.8488e-07,
         7.3097e-02]]), 'X': tensor([[ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0406, 0.0667,  ..., 0.3818, 0.5341, 0.8902],
        [0.0406, 0.0000, 0.0785,  ..., 0.3449, 0.5147, 0.8964],
        [0.0667, 0.0785, 0.0000,  ..., 0.4137, 0.5929, 0.8239],
        ...,
        [0.3818, 0.3449, 0.4137,  ..., 0.0000, 0.3178, 1.1351],
        [0.5341, 0.5147, 0.5929,  ..., 0.3178, 0.0000, 1.3975],
        [0.8902, 0.8964, 0.8239,  ..., 1.1351, 1.3975, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.74686071,  7.01475376,  7.18354885,  7.26074925,  7.24677664,
        7.28793663, 13.58333857,  6.27905292, 14.07623266,  6.60752373,
        7.34250948,  9.50654967,  5.84450127, 11.56146873, 11.63765273,
        6.35029274,  6.81961996, 13.98691426,  7.06169391,  4.91336994,
        8.83725081,  5.19577146,  8.51928531,  6.76056625,  8.27853076,
        6.4020404 , 13.40024224,  7.15100303,  5.95297459,  6.77777606,
        7.5403822 , 14.10045797,  5.0209005 ,  8.40445872,  6.27905292,
       10.76154256, 11.64746882,  6.41202986,  6.76056625, 13.08414689,
       10.67322527, 11.10066126,  5.79391825,  9.50654967, 13.93790235,
       10.27572673, 11.63724792, 10.0674887 , 13.45124917,  8.99791316,
        5.05981914,  5.26067019,  5.19065439,  6.42689705, 13.14777433,
        5.52416368, 11.77124373, 10.32715715, 10.75443106, 13.66991027,
        8.29640613,  6.78325151,  7.02496223,  8.89458273,  5.19986797,
       13.08414689,  7.38844703,  7.53063885, 12.52372629,  9.68323686,
        7.71075226, 13.35832725, 11.05317597,  6.35029274,  5.70591986,
       10.76154256, 13.58333857,  4.90676242,  9.60412735, 13.05148826,
        9.20340031,  5.52416368, 13.32374056, 13.08414689, 12.49291768,
        7.62232307, 10.7506423 , 10.412299  ,  7.62011734, 11.0050379 ,
       12.30695098,  4.78002665, 12.18998195, 13.72594022,  6.27715105,
        5.56809335, 12.00995898, 11.33837396, 13.32374056,  8.62329466,
        9.09502385,  8.31625171, 10.90638066, 11.97238624, 13.94008948,
       11.22993218])}
{'P': tensor([[2.4020e-02, 2.9682e-01, 1.7780e-01,  ..., 1.1508e-04, 2.9255e-05,
         1.1842e-05],
        [1.9404e-03, 3.4849e-02, 2.9223e-01,  ..., 1.5506e-04, 3.4003e-05,
         1.8230e-05],
        [1.6951e-03, 2.0795e-03, 6.3779e-02,  ..., 2.3050e-04, 3.1552e-05,
         5.0705e-05],
        ...,
        [1.2519e-02, 1.0514e-02, 5.9225e-03,  ..., 9.9798e-02, 1.4100e-02,
         6.8135e-05],
        [9.3054e-03, 8.1495e-03, 4.6933e-03,  ..., 7.2145e-05, 2.7934e-02,
         5.9627e-07],
        [2.1918e-05, 1.9294e-05, 1.5044e-05,  ..., 9.0731e-04, 2.9030e-05,
         6.2615e-02]]), 'X': tensor([[ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0119, 0.0328,  ..., 0.0908, 0.0720, 1.0629],
        [0.0119, 0.0000, 0.0316,  ..., 0.0830, 0.0690, 1.0620],
        [0.0328, 0.0316, 0.0000,  ..., 0.0634, 0.0393, 1.0308],
        ...,
        [0.0908, 0.0830, 0.0634,  ..., 0.0000, 0.0386, 0.9940],
        [0.0720, 0.0690, 0.0393,  ..., 0.0386, 0.0000, 0.9935],
        [1.0629, 1.0620, 1.0308,  ..., 0.9940, 0.9935, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.74872981,  6.81084724,  7.02496223,  7.02253235,  6.84796202,
        7.18354885, 13.85158417,  7.38844703, 13.6660671 ,  9.85384623,
       10.794755  ,  6.77777606, 10.06437598, 10.5575601 ,  6.48486171,
        4.92497514,  8.99791316, 11.35182589, 12.1460931 , 12.13568598,
        6.41202986, 11.22142195,  6.02588517,  9.70108409,  5.43544626,
        9.60359158,  9.60412735,  5.43544626,  7.83534987, 11.63502679,
       11.37939695,  8.66273171, 13.69162795, 13.77222752, 13.66051884,
       11.27656488, 13.13205357, 11.8159254 ,  8.5724042 ,  5.82345867,
       12.70489563,  7.60529892, 11.32149259, 10.27716434, 10.8992524 ,
        7.46248947, 13.34587027, 13.50484539,  5.43943791,  7.50917551,
       12.62576026,  5.9510489 ,  5.05774295,  5.41348121,  7.81509631,
       12.35625694, 12.5769659 ,  7.38844703,  9.34127013, 14.03618649,
        9.96391045,  5.84533345, 10.91601655,  5.82345867, 12.9089559 ,
       11.81423161,  5.95297459,  5.38342077, 13.50484539, 12.87752052,
        7.48311002,  5.36061974, 13.59673406,  9.78285424,  9.64157461,
       10.04916678,  7.38844703,  8.68832398,  9.75200848,  8.04483495,
       13.6105595 ,  8.62260327,  6.02789634,  6.31576264,  7.85919687,
        8.51088913, 12.17451685, 13.00380624, 11.77124373,  7.06272256,
        6.2134961 , 11.37939695,  9.32094055,  5.26067019, 13.41713661,
       10.1371552 , 10.57302546,  8.91551231,  5.84450127, 11.11038551,
       12.50996828,  7.54127425,  6.60752373,  5.08910491, 12.17901033,
        9.77588801])}
{'P': tensor([[3.0174e-02, 2.0956e-01, 1.4831e-01,  ..., 3.3105e-05, 3.9534e-03,
         2.2644e-05],
        [6.7356e-04, 2.7710e-02, 3.7623e-01,  ..., 2.3225e-05, 9.9471e-03,
         1.4763e-05],
        [1.5184e-03, 3.2145e-03, 8.8134e-02,  ..., 6.9658e-05, 4.7525e-02,
         4.3020e-05],
        ...,
        [5.6880e-04, 3.2329e-04, 2.6434e-04,  ..., 7.5618e-02, 6.7261e-05,
         1.0456e-01],
        [1.4588e-04, 2.0210e-04, 2.3998e-04,  ..., 1.4419e-04, 1.0646e-01,
         6.8638e-05],
        [2.6879e-04, 1.6104e-04, 1.3353e-04,  ..., 6.8678e-03, 3.3296e-05,
         4.9995e-02]]), 'X': tensor([[  6.0073,   3.0986,   6.7937],
        [  5.0277,   4.9589,   7.0175],
        [  4.4915,   5.5990,   7.0405],
        [  4.5503,   5.5359,   7.4735],
        [  4.9531,   5.0576,   5.6794],
        [  4.6357,   5.4415,   7.9067],
        [  4.7813,  -3.1139,   8.2691],
        [ -0.3432,   7.8900,   7.1805],
        [ -3.9508,  -9.8515,   1.7923],
        [ -1.1885, -10.8211,   0.0865],
        [  5.6260,  -1.9401,   3.0259],
        [ -1.6220,   7.8917,   0.0315],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.4624,  -0.4633,  -7.3918],
        [  2.0449,  13.8402,  -7.1642],
        [  4.1637,  13.1898,  -1.9189],
        [  6.3542,   0.6934,  -3.3840],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 11.0891,  -4.9558,  -2.3178],
        [  5.6928,  12.4710,   3.6001],
        [  4.2612, -10.5514,  -6.6268],
        [  9.9217,  -6.7006,  -5.6728],
        [  4.5767,   5.5071,   3.9069],
        [ 11.5966,   5.9666,   6.0083],
        [  2.5636, -10.9293,   9.6519],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.3331,  -2.6465,  -8.2028],
        [  2.5663,   7.0526,  -6.8769],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.0047,   3.1062,  -6.7914],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.2860,   7.1916,  -6.1011],
        [  2.5636, -10.9293,   9.6519],
        [  1.8719,   7.3713,  -3.7465],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.7232,  -4.7574,   3.1270],
        [  0.9809,   7.6642,  -1.4217],
        [ 11.4495,   6.3026,   6.8257],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.8323,  -1.5228,   5.1132],
        [ -8.3535,  -5.5118,   9.8399],
        [ 11.5935,   5.9740,  -4.3952],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.2345,  -3.1345,   3.9715],
        [  0.3231,  -4.7691,  -1.6120],
        [  3.0719, -10.8449,  -4.1673],
        [  9.3624,   9.5318,   0.2779],
        [ -6.3066,  -8.1936,   9.5822],
        [ 12.0042,  -2.8350,   2.6864],
        [  9.2692,   9.6370,   7.0746],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.1905,   3.7253,   4.1175],
        [ -6.8618,  -7.6322,   3.7027],
        [  9.0652,   9.8597,  -3.5949],
        [  3.7121,   6.3117,   0.6738],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.1588,   4.3364,  -8.0867],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.2701,   6.6338,   9.9209],
        [  2.0899,  -4.6870,  -7.4260],
        [ -1.8823, -10.6558,  -3.8952],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 12.2328,  -2.0376,  -1.1234],
        [  4.2096,   5.8815,   1.9230],
        [  3.3943,   6.5483,   1.0790],
        [  1.7815,  -4.7481,   4.4289],
        [  2.2860,   7.1916,  -6.1011],
        [ -5.8389,  -8.6056,  -3.5098],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.6681,  -4.8556,   9.2518],
        [  6.3303,   1.6145,  -8.4083],
        [ 11.5456,   6.0861,   8.1101],
        [ -0.5385,   7.9037,   6.3830],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.1755,   2.5233,  -2.0042],
        [ -5.3382,   6.6533,   7.8054],
        [  5.5626,  -2.0536,   8.4433],
        [ 11.4743,   6.2478,  -2.1100],
        [  4.0362,  13.2397,  -7.0030],
        [  3.4009,  -4.1874,  -5.9751],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.6310, -10.9195,   3.8054],
        [  0.6363, -11.0348,   6.1414],
        [ -0.8455, -10.8853,  -6.7400],
        [ -7.3470,  -7.0615,   5.0066],
        [  2.0937,  13.8293,  -2.3533],
        [ -2.9497, -10.3028,  -7.3479],
        [ -2.0229, -10.6163,   8.6930],
        [  2.3376,  -4.6231,   9.8436],
        [  0.9967,  -4.8144,  -9.7370],
        [  7.1441,  11.5529,   2.3232],
        [  1.6422,   7.4586,  -6.8056],
        [ 12.2435,   4.0076,   8.1257],
        [  0.7145,  -4.8071,  -2.3766],
        [ -6.5295,  -7.9785,  -7.4429],
        [  8.3791,  10.5409,   5.5590],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -3.5029,   7.5138,   8.1853],
        [  8.8091,  -7.9037,   0.3631],
        [  2.6613,   7.0021,   4.5783],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.2974,   0.8371,   2.4852],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.9392,  -4.8143,  -3.4649],
        [ -3.1671,   7.6158,   6.0765],
        [  2.0419,  -4.6978,  -5.0983]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0439, 0.0709,  ..., 0.7089, 0.3700, 0.7763],
        [0.0439, 0.0000, 0.0273,  ..., 0.7137, 0.3278, 0.7834],
        [0.0709, 0.0273, 0.0000,  ..., 0.7131, 0.3007, 0.7842],
        ...,
        [0.7089, 0.7137, 0.7131,  ..., 0.0000, 0.7366, 0.0793],
        [0.3700, 0.3278, 0.3007,  ..., 0.7366, 0.0000, 0.8157],
        [0.7763, 0.7834, 0.7842,  ..., 0.0793, 0.8157, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.7593917 ,  7.06169391,  7.17789666,  7.16598273,  7.07902036,
        7.14837514,  5.70591986,  7.89745107, 10.61417181, 10.88618114,
        5.95111804,  8.05668609,  6.80370254,  9.47369628, 13.99048094,
       13.83139024,  6.39187285,  9.78286733, 12.1460931 , 13.70893492,
       11.37939695, 11.97238624,  7.16058587, 13.04156075, 11.22596994,
       12.4217523 ,  9.70108409,  7.50500064,  6.10774544,  6.76056625,
       11.86239961,  7.54620849, 11.22596994,  7.60529892,  5.19577146,
        5.05989547,  7.72668467, 13.06957425,  5.57341623,  6.02779201,
       10.00802627, 13.04217487, 11.75618513,  9.75200848,  4.78002665,
       11.27160035, 13.3607379 , 10.33958639, 12.3344536 , 13.37121923,
        9.13507321,  8.99791316, 10.26327895, 13.39372426,  7.32233733,
        5.18047638, 12.9089559 , 10.6910768 ,  7.39599986,  5.13182506,
       10.8207297 , 12.12641969, 12.4013153 ,  7.23276582,  7.37576154,
        5.0713364 ,  7.54620849, 10.39942497,  7.78980299,  9.9353921 ,
        6.53290663, 13.05148826,  7.92200342, 10.23946247,  6.67108662,
        8.53014915,  5.92952538, 13.06498554, 13.84126197,  5.39450305,
        5.46244988, 11.23201619, 11.05317597, 10.91805683, 10.19036081,
       13.98691426, 10.7167285 , 10.80728788,  5.1805345 ,  4.91652556,
       13.58333857,  7.63726134, 12.88270018,  4.85995002, 10.30972193,
       13.46553623,  9.50913849,  8.29022197, 11.83509372,  7.49076899,
        7.822854  ,  9.3349858 , 12.41012046,  4.90505566,  8.24808832,
        5.1224079 ])}
{'P': tensor([[3.0886e-02, 1.5568e-01, 1.3333e-01,  ..., 2.2804e-05, 2.8088e-04,
         1.9890e-05],
        [8.0342e-04, 2.5003e-02, 1.2846e-02,  ..., 1.6150e-05, 1.2764e-04,
         7.9259e-06],
        [1.3891e-03, 1.4220e-02, 4.5875e-02,  ..., 1.4262e-05, 1.8941e-04,
         1.2005e-05],
        ...,
        [1.0134e-03, 1.3563e-03, 5.2057e-04,  ..., 2.4636e-01, 2.7315e-03,
         9.0450e-04],
        [4.3475e-02, 3.0150e-02, 1.5163e-02,  ..., 9.0234e-04, 8.3973e-02,
         2.4817e-03],
        [1.6911e-03, 8.8916e-04, 5.6525e-04,  ..., 1.0208e-04, 2.0447e-03,
         1.2064e-02]]), 'X': tensor([[  6.0047,   3.1062,  -6.7914],
        [  5.1292,   4.8183,  -5.7997],
        [  5.0190,   4.9705,  -7.9752],
        [  4.1537,   5.9340,  -6.0423],
        [  3.9895,   6.0816,  -6.5235],
        [  3.9663,   6.1018,  -5.4104],
        [  1.7232,  -4.7574,   3.1270],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.6970,   6.9826,   9.7350],
        [  1.6954,  -4.7616,   9.8198],
        [  2.5469,  13.7192,  -9.6935],
        [ 12.5789,   0.1722,   1.6470],
        [ -0.1963,   7.8765,  -1.1371],
        [ -6.7745,  -7.7264,   8.4684],
        [  2.3490,  -4.6199,  -4.2489],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.8407,  11.7666,   3.0045],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.2790,  -0.0259,  -5.4116],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.8275,  -1.5334,   2.0124],
        [  2.5382,   7.0672,   4.5097],
        [  6.3087,   1.8053,  -1.9481],
        [  2.1477,   7.2550,   8.9688],
        [ -7.3446,   4.9113,  -5.9356],
        [ -1.8367,   7.8718,  -8.6523],
        [ -8.9636,   2.0648,   8.1946],
        [ 11.2767,  -4.6006,   0.7868],
        [ -1.9600,   7.8577,   6.9121],
        [  3.2159, -10.8166,  -1.0406],
        [  6.3589,   0.8239,  -8.4230],
        [ -7.9667,   4.0814,  -0.6397],
        [ -0.5739,   7.9056,   0.6382],
        [  4.2096,   5.8815,   1.9230],
        [  6.1755,   2.5233,  -2.0042],
        [ -3.5173, -10.0623,   4.8280],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.7976,  -4.4682,  -4.2020],
        [ -7.0825,   5.2073,  -5.0497],
        [  8.7715,  -7.9390,   5.2900],
        [ -7.6650,   4.5092,   0.2520],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.3860,  -2.3418,   1.0826],
        [ 11.9641,  -2.9568,   7.7211],
        [ -0.8675, -10.8815,   6.8561],
        [  5.6248,   3.9984,   8.4933],
        [  8.3791,  10.5409,   5.5590],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.6229,   2.9068,   2.4429],
        [ -4.6393,   7.0461,   1.1624],
        [  4.3459,  -3.5327,   0.3667],
        [ -1.6906,   7.8860,   7.7142],
        [  0.5960, -11.0335,   3.1144],
        [  5.9003,  12.3550,   2.1917],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.9433,  -9.3362,  -0.3421],
        [  3.3069,   6.6089,   8.7377],
        [  2.6613,   7.0021,   4.5783],
        [  4.4633,   5.6287,  -0.7754],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.7565,  11.8237,  -4.9882],
        [ -5.1974,   6.7396,   3.3484],
        [  1.3339,   7.5627,  -0.9034],
        [  6.3124,   0.2057,  -0.4593],
        [  2.5974,   7.0363,  -7.7810],
        [ -6.1914,  -8.2997,   8.5380],
        [ -9.2508,   1.0572,   2.9978],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.3724,   7.7974,  -6.8722],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.6675,  -4.8569,  -9.1769],
        [  7.0515,  11.6194,   8.4289],
        [ -5.9940,   6.1985,   5.9566],
        [  6.2675,  12.1379,  -8.9720],
        [ -9.4735,  -1.2620,  -3.6742],
        [  3.9022,  13.2906,   5.5788],
        [  4.7156,  -3.1831,  -8.0224],
        [ -6.4844,  -8.0230,   8.5209],
        [  3.8182,  -3.9347,  -1.7961],
        [  2.2860,   7.1916,  -6.1011],
        [ -8.8540,  -4.3971,  -5.2144],
        [  4.8653,   5.1694,  -4.0823],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.3609,   3.4769,  -8.5762],
        [  1.5662,  13.9375,   4.5142],
        [  5.0190,   4.9705,  -7.9752],
        [ 12.2435,   4.0076,   8.1257],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -8.0965,  -5.9712,   4.3466],
        [  1.3705,  13.9723,   7.9266],
        [ -1.7009,   7.8851,  -8.2437],
        [ -7.0120,   5.2825,   7.7762],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.1989,  -9.7591,   0.8289],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.3357,   3.6001,   0.7903],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.3616,  -2.4750,   3.7765],
        [  5.4167, -10.1293,   6.2156],
        [ -8.3900,   3.3727,   3.1650],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.3804,  -2.3504,   7.2744],
        [  5.4179,  -2.2925,  -3.4641],
        [  9.9756,  -6.6330,  -3.3429]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.8991e-02, 5.9634e-02,  ..., 8.5714e-01, 1.4739e-01,
         1.5622e-01],
        [4.8991e-02, 0.0000e+00, 6.8376e-02,  ..., 8.2435e-01, 1.1572e-01,
         1.1069e-01],
        [5.9634e-02, 6.8376e-02, 0.0000e+00,  ..., 8.9194e-01, 1.8409e-01,
         1.3578e-01],
        ...,
        [8.5714e-01, 8.2435e-01, 8.9194e-01,  ..., 0.0000e+00, 7.1036e-01,
         8.0328e-01],
        [1.4739e-01, 1.1572e-01, 1.8409e-01,  ..., 7.1036e-01, 1.5170e-04,
         1.5215e-01],
        [1.5622e-01, 1.1069e-01, 1.3578e-01,  ..., 8.0328e-01, 1.5215e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.76056625,  7.03733775,  7.06372172,  7.24326992,  7.27340834,
        7.27759233,  5.05989547, 10.61417181,  7.48538905,  5.05444288,
       13.95361098, 12.58006038,  7.8788965 , 10.27572673,  5.18277858,
       12.45418236, 13.6105595 ,  8.83725081,  6.27905292, 10.1979669 ,
        6.02588517,  7.50917551,  6.5618968 ,  7.56617037,  8.8353674 ,
        8.08320235,  9.19837792, 12.17901033,  8.09842704, 11.2845608 ,
        6.41202986,  8.95133961,  7.92644746,  7.23276582,  6.67108662,
       10.65929968, 10.92597196,  5.27178363,  8.7907923 , 11.83075005,
        8.89301516,  9.26455795,  5.87305005, 12.32408516, 10.91601655,
        6.90115901, 13.46553623, 13.95361098,  9.09963894,  8.43626114,
        5.60064319,  8.06517143, 11.04953767, 13.69162795,  5.05444288,
       10.06023665, 11.63502679,  7.39004226,  7.49076899,  7.18354885,
        5.81199422, 12.53047489, 13.61801526,  8.51088913,  7.67939546,
        6.31576264,  7.50035355, 10.35464916,  9.3109884 ,  6.14086582,
        7.80625981,  8.01009083,  9.93553424, 13.59170809,  8.62260327,
       13.66051884,  9.55720651, 13.85158417,  5.68942572, 10.31584102,
        5.48276038,  7.54620849,  9.88571713,  7.09887557,  8.89458273,
       12.84056604, 14.02526505,  7.06372172, 12.88270018, 12.10942412,
       10.06023665, 14.0393949 ,  8.06643487,  8.77913428, 12.4013153 ,
       11.56146873, 12.50996828, 12.85032836, 12.3344536 ,  9.68324553,
       11.48663767,  9.04255655, 12.00995898,  5.87132718,  5.8828986 ,
       11.97957425])}
{'P': tensor([[1.6346e-02, 2.3798e-01, 1.8455e-01,  ..., 1.8013e-03, 1.9867e-03,
         4.4480e-06],
        [1.3466e-03, 2.5397e-02, 1.2289e-02,  ..., 1.4679e-03, 2.7116e-03,
         6.8510e-06],
        [1.6848e-03, 4.3870e-02, 2.4406e-02,  ..., 1.8849e-03, 2.5534e-03,
         6.6490e-06],
        ...,
        [1.1591e-01, 1.8125e-01, 1.3000e-01,  ..., 1.9073e-02, 2.8492e-03,
         5.1490e-06],
        [4.6771e-04, 6.0667e-04, 5.7480e-04,  ..., 3.6643e-04, 3.2805e-02,
         3.9438e-07],
        [1.3006e-04, 1.1309e-04, 1.1656e-04,  ..., 1.4733e-04, 6.0500e-06,
         3.7431e-02]]), 'X': tensor([[ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0060, 0.0091,  ..., 0.0161, 0.4010, 0.8256],
        [0.0060, 0.0000, 0.0084,  ..., 0.0171, 0.4041, 0.8201],
        [0.0091, 0.0084, 0.0000,  ..., 0.0245, 0.4100, 0.8187],
        ...,
        [0.0161, 0.0171, 0.0245,  ..., 0.0000, 0.3874, 0.8321],
        [0.4010, 0.4041, 0.4100,  ..., 0.3874, 0.0000, 1.1767],
        [0.8256, 0.8201, 0.8187,  ..., 0.8321, 1.1767, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.76661595,  6.81117802,  6.80559669,  7.02326515,  7.09887557,
        7.06272256,  6.81084724,  9.60080639,  9.70108409,  5.14607014,
        5.87305005,  9.94285511,  9.63742669, 13.01728313,  7.53884104,
        9.17595899,  9.53015376,  9.99145096,  6.29821673,  9.36256096,
        7.8575648 ,  6.38399789,  6.74872981, 11.67875855, 12.4980845 ,
       13.00413829, 12.21532282,  8.62162372,  5.94909709, 11.32149259,
       13.58333857,  5.05989547,  6.65259775,  5.2287058 , 12.16901615,
        5.35774573,  7.6374603 , 10.08001894, 13.15525803,  7.98469166,
       12.38881635,  7.09887557, 11.69264243, 13.39372426, 12.46513646,
       11.73848065,  7.49076899, 13.97430191, 12.76408562,  6.67218687,
        7.59795445,  7.27340834,  7.10378919, 13.0493368 , 11.48734978,
       11.28215429, 11.63724792,  9.27474386,  5.81199422, 13.00380624,
       10.74655346, 12.44177641,  6.61211224,  5.91854397,  4.83371341,
        9.38232121,  7.06372172, 13.32374056, 11.1913598 ,  9.05843201,
       13.3614598 ,  8.68979052,  8.83725081,  8.87587457, 10.26545792,
       13.58101794, 12.11735334, 12.4013153 ,  7.58668144, 10.92597196,
        4.78258092,  9.49511524, 11.70016217,  5.10446653, 10.05558893,
       13.25336001, 12.22503754, 10.90638066, 10.67322527, 12.07225469,
       11.48663767,  8.47371868, 12.64235359,  6.02789634, 13.6922425 ,
       11.46176259,  9.50654967,  5.54776934, 10.80728788, 12.76408562,
        7.80625981,  7.73692436,  8.99276564,  6.7166297 , 13.26590071,
       10.67132524])}
{'P': tensor([[2.8438e-02, 1.7512e-01, 1.6142e-01,  ..., 3.2026e-05, 8.5686e-05,
         8.3770e-06],
        [7.8030e-04, 3.1544e-02, 1.3667e-02,  ..., 5.9501e-05, 7.5653e-05,
         1.8588e-05],
        [6.1609e-04, 1.6458e-02, 2.2086e-02,  ..., 4.3615e-05, 4.3486e-05,
         8.4177e-06],
        ...,
        [3.5501e-05, 2.0695e-05, 2.2645e-05,  ..., 4.9190e-02, 7.1278e-04,
         3.4034e-05],
        [7.2023e-03, 4.1779e-03, 3.0798e-03,  ..., 1.0646e-04, 1.5594e-01,
         1.0320e-04],
        [6.3198e-06, 5.5458e-06, 4.0473e-06,  ..., 1.7727e-04, 6.3577e-04,
         6.2506e-02]]), 'X': tensor([[  5.9883,   3.1536,   6.2320],
        [  4.9531,   5.0576,   5.6794],
        [  5.0277,   4.9589,   7.0175],
        [  4.4915,   5.5990,   7.0405],
        [  4.5503,   5.5359,   7.4735],
        [  3.6817,   6.3354,   6.1233],
        [  2.7420,  -4.4896,  -3.7139],
        [  3.3167,   6.6022,   6.3554],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.3464,   0.5708,  -1.6397],
        [ -4.0633,  -9.7927,  -3.0825],
        [  7.3724,  -9.0577,  -8.9843],
        [ -4.8665,  -9.3204,  -5.2564],
        [  3.1480,  -4.3165,  -7.8002],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.0619,   8.6627,  -9.2884],
        [ 12.5919,   0.4035,  -5.8460],
        [ 11.7795,  -3.4697,   3.5624],
        [  1.7301,   7.4262,   5.0522],
        [  0.9036,   7.6840,   9.9237],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.5875,   7.8944,  -9.4286],
        [ -8.8363,   2.4086,  -5.6296],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.4996,   2.6335,  -6.3166],
        [ -2.6069, -10.4298,  -6.1995],
        [ -6.1914,  -8.2997,   8.5380],
        [  0.1206,  -4.7363,   8.7103],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.4697,  -4.5836,   1.4593],
        [  6.0054,  -1.0902,   8.3079],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.1006,   7.6318,  -2.4877],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.4691,   2.8525,  -5.5110],
        [  2.3367,   7.1676,  -9.5263],
        [ -7.4807,   4.7463,   2.0616],
        [  4.5767,   5.5071,   3.9069],
        [  0.6363, -11.0348,   6.1414],
        [  6.2770,  -0.0379,   2.4195],
        [ -8.5506,  -5.1154,   6.3423],
        [  1.9503,  -4.7172,  -6.2313],
        [  7.8375,  -8.7240,  -9.8913],
        [  3.7298,  13.3537,  -9.4653],
        [ -8.6603,   2.8250,   5.9181],
        [  0.2571,  -4.7594,   1.5618],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.2770,  -0.0379,   2.4195],
        [  2.2076,  13.8032,  -1.5099],
        [  2.6613,   7.0021,   4.5783],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -1.9600,   7.8577,   6.9121],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.9260,  -4.4162,   8.3808],
        [  5.6248,   3.9984,   8.4933],
        [  0.3715,  -4.7756,   0.2312],
        [  2.0547,   7.2957,  -1.9343],
        [  5.2939,  -2.4784,  -4.5840],
        [ -3.3116, -10.1539,   7.5919],
        [ -3.2486,   7.5925,  -1.3719],
        [ -5.6169,  -8.7844,   8.9057],
        [ -1.7290,   7.8825,  -3.3831],
        [ -1.2454,   7.9127,  -0.5715],
        [  4.9152,  12.8653,  -1.4459],
        [  9.8082,  -6.8396,   4.9615],
        [  8.2154,  -8.4262,   8.0171],
        [  6.6319,  11.9066,   3.5165],
        [  3.6249, -10.7255,   5.6605],
        [  9.6662,   9.1725,   8.1056],
        [  3.0719, -10.8449,  -4.1673],
        [  4.0078,  -3.8018,  -6.2168],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.3266,   0.3292,   9.3502],
        [ -0.8569,   7.9156,  -6.5557],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.4235,   6.3594,   5.1663],
        [  5.8979,   3.3958,  -3.8713],
        [  9.7328,   9.0902,  -0.8917],
        [ -1.6220,   7.8917,   0.0315],
        [ -0.1963,   7.8765,  -1.1371],
        [  0.9788,  -4.8145,  -3.8661],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.0591,   6.0202,  -1.6339],
        [  2.6970,   6.9826,   9.7350],
        [  9.4554,   9.4245,   9.8800],
        [  2.3373,  -4.6232,  -5.2246],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -0.8455, -10.8853,  -6.7400],
        [ -1.0012, -10.8576,   2.2867],
        [  5.8889,  -1.3916,   8.8993],
        [ -2.9497, -10.3028,  -7.3479],
        [  9.6587,   9.1817,  -0.8212],
        [ 11.2517,  -4.6495,   2.9800],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.8721,  -4.7322,   0.5524],
        [ -9.4697,  -0.6157,  -6.8472]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0541, 0.0537,  ..., 0.9190, 0.4446, 1.3156],
        [0.0541, 0.0000, 0.0600,  ..., 0.8755, 0.4124, 1.2637],
        [0.0537, 0.0600, 0.0000,  ..., 0.8739, 0.4724, 1.3108],
        ...,
        [0.9190, 0.8755, 0.8739,  ..., 0.0000, 0.9940, 1.0383],
        [0.4446, 0.4124, 0.4724,  ..., 0.9940, 0.0000, 0.9819],
        [1.3156, 1.2637, 1.3108,  ..., 1.0383, 0.9819, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.76791483,  7.07902036,  7.06169391,  7.17789666,  7.16598273,
        7.32754882,  5.26067019,  7.38844703,  8.95983586,  9.75200848,
        9.36378438, 10.60226614, 11.67875855, 10.51437947,  5.34250225,
       10.38897563, 13.27717982, 12.59840193, 12.27991991,  7.62509582,
        7.73692436, 10.6910768 ,  8.05242591,  9.15866607,  4.78544875,
       12.77402351, 10.7506423 , 10.35464916,  4.73784222, 10.46238054,
        5.20659892,  6.10360073, 10.4977132 ,  6.27905292,  7.71075226,
       11.85743425, 12.91143491, 12.79126785,  7.53884104,  8.85939626,
        7.16058587, 11.05317597,  6.27715105,  9.96391045,  5.10444097,
       11.72749923, 13.86480175,  9.10945678,  4.7663526 , 12.27991991,
        6.27715105, 13.97857591,  7.49076899,  9.13507321,  9.43737523,
        8.09842704, 13.03002338,  5.29755487,  6.90115901,  4.79003361,
        7.57946459,  5.84533345, 10.68031351,  8.25828409, 10.42667093,
        8.06990483,  8.01009083, 13.77222752, 11.95743558, 11.76830981,
       13.62896593, 11.32149259, 13.32557133, 11.27160035,  5.52416368,
        9.41605684,  6.33517349,  7.96181288,  9.93553424, 12.46513646,
       13.66051884, 13.0743399 ,  6.80559669, 13.31764181,  8.05668609,
        7.8788965 ,  4.91296205, 10.06407839, 10.63038565,  7.26074925,
        7.48538905, 13.35013374,  5.18047638, 10.27716434, 10.91805683,
       10.9036226 ,  6.05113217, 10.7167285 , 13.32645552, 12.17451685,
       12.12641969,  9.24501829,  9.75075273,  9.77588801,  5.08910491,
        9.48969978])}
{'P': tensor([[3.7614e-02, 2.6373e-01, 1.8158e-01,  ..., 2.0739e-05, 4.3510e-06,
         9.5830e-06],
        [1.9022e-03, 7.3489e-02, 2.1221e-03,  ..., 5.8853e-05, 4.9702e-06,
         9.7175e-06],
        [5.0382e-03, 3.6523e-01, 4.0645e-02,  ..., 2.1044e-05, 3.8645e-06,
         8.4756e-06],
        ...,
        [2.8373e-06, 2.6386e-06, 2.4876e-06,  ..., 4.1957e-02, 2.6096e-05,
         9.0971e-06],
        [1.8677e-04, 8.0615e-05, 1.3968e-04,  ..., 2.2890e-06, 3.3402e-02,
         1.3105e-01],
        [6.6547e-04, 2.7683e-04, 5.0076e-04,  ..., 1.5732e-06, 2.6585e-04,
         2.7480e-02]]), 'X': tensor([[  5.9655,   3.2172,  -7.5953],
        [  5.0190,   4.9705,  -7.9752],
        [  5.8201,   3.5841,  -8.0609],
        [  3.9895,   6.0816,  -6.5235],
        [  4.3554,   5.7393,  -9.0766],
        [  4.1537,   5.9340,  -6.0423],
        [ -3.2736, -10.1703,   3.8712],
        [  6.3301,   0.3638,  -7.5685],
        [  9.4554,   9.4245,   9.8800],
        [ -9.1231,  -3.5739,   8.4708],
        [ -7.0120,   5.2825,   7.7762],
        [  6.2365,  -0.2578,  -6.0212],
        [  4.9452, -10.3187,  -7.8621],
        [  6.3569,   0.7591,   6.5011],
        [ -7.9667,   4.0814,  -0.6397],
        [ -8.9636,   2.0648,   8.1946],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.6248,   3.9984,   8.4933],
        [ -9.4203,  -2.0340,   2.5856],
        [  4.8674,  12.8875,   6.7622],
        [ -0.4341, -10.9473,   7.4509],
        [  6.0031,  12.2958,  -6.5992],
        [  0.4996,   7.7738,  -4.4711],
        [  5.3602,  -2.3808,   7.2098],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.1706,   1.3863,   8.4032],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -3.7964,  -9.9294,   7.7147],
        [  8.8474,  10.0868,  -8.8644],
        [ -2.8162,   7.7059,  -9.6885],
        [  4.3377,   5.7570,   1.3712],
        [  4.5491,  -3.3485,   6.8412],
        [  5.3860,  -2.3418,   1.0826],
        [  9.4554,   9.4245,   9.8800],
        [  5.9843,  -9.8679,  -6.1449],
        [ -5.3328,  -8.9992,   7.3440],
        [  9.1409,  -7.5789,   1.5814],
        [  5.6022,  -1.9833,   8.3081],
        [  6.3589,   0.8239,  -8.4230],
        [  0.9074,  -4.8139,   8.1674],
        [  2.2849, -10.9655,  -7.6542],
        [ -9.4739,  -1.2497,   0.5483],
        [ -7.9298,   4.1367,  -3.9773],
        [  1.7122,  -4.7591,  -3.6734],
        [ -1.8823, -10.6558,  -3.8952],
        [  9.5203,   9.3483,   5.8781],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.0039,   7.3173,  -1.5885],
        [  4.5057,  -3.3894,  -0.8160],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.3858,   4.4260,  -9.6769],
        [  2.4572,   7.1085,   7.7376],
        [  2.2644,  -4.6434,   9.7508],
        [  6.1222,  -0.7261,   8.8204],
        [  4.4290,  -3.4596,   1.4181],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.6414, -10.0303,   7.8525],
        [ -8.5149,  -5.1905,   1.5239],
        [ 11.5935,   5.9740,  -4.3952],
        [  7.5309,  11.2617,   7.2656],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 12.2218,   4.0953,  -8.2072],
        [  0.8572,  14.0501,  -8.3364],
        [  5.8323,  -1.5228,   5.1132],
        [ -9.1178,   1.5784,   2.1410],
        [ -3.9700,   7.3453,  -4.1486],
        [  3.9770, -10.6341,   8.0783],
        [  4.8432,   5.1968,   1.5006],
        [ 12.2589,  -1.9312,   9.8657],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.7544,  -3.1425,  -1.5315],
        [  8.4691,  10.4571,  -5.0650],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.2608,  12.6977,   4.9166],
        [ -4.9217,   6.8979,  -1.7667],
        [  7.0515,  11.6194,   8.4289],
        [  5.6093,  -1.9705,   6.4807],
        [  6.1741,  -0.5329,   3.1913],
        [  2.5382,   7.0672,   4.5097],
        [  5.0617,  12.7957,   8.3544],
        [  0.9036,   7.6840,   9.9237],
        [  1.3670,  -4.7990,   3.2603],
        [ 11.6801,  -3.7196,  -0.9181],
        [  7.5064,  -8.9651,  -0.0994],
        [ -1.6334, -10.7207,   7.1397],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.6078, -10.9229,   6.2869],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.3266,   0.3292,   9.3502],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -1.0482, -10.8487,  -5.9066],
        [ -4.6393,   7.0461,   1.1624],
        [  1.9312,  13.8649,   8.3563],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.2675,  12.1379,  -8.9720],
        [ -0.0410,   7.8591,   9.3457],
        [ -7.8359,   4.2735,   7.2943],
        [  5.8450,  -9.9356,   3.0767],
        [  7.8982,  -8.6778,   0.0239]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0485, 0.0143,  ..., 1.1258, 0.7620, 0.4852],
        [0.0485, 0.0000, 0.0378,  ..., 1.1083, 0.7496, 0.4681],
        [0.0143, 0.0378, 0.0000,  ..., 1.1293, 0.7669, 0.4885],
        ...,
        [1.1258, 1.1083, 1.1293,  ..., 0.0000, 0.3756, 0.6407],
        [0.7620, 0.7496, 0.7669,  ..., 0.3756, 0.0000, 0.2859],
        [0.4852, 0.4681, 0.4885,  ..., 0.6407, 0.2859, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.77777606,  7.06372172,  6.83515201,  7.27340834,  7.20482224,
        7.24326992, 10.68416401,  6.34059021, 13.35013374,  9.79814997,
        8.77913428,  6.24187044, 11.4424816 ,  6.4020404 ,  8.95133961,
        9.19837792,  6.02789634,  6.90115901,  9.63742669, 13.77604293,
       10.95593729, 13.6829843 ,  7.78980299,  5.86519525, 12.11735334,
        9.562174  ,  9.27474386, 10.37698435, 12.46513646,  9.47369628,
       10.63038565, 13.41713661,  8.20436764,  7.20826219,  5.64865173,
        5.87305005, 13.35013374, 11.54071859, 10.4606265 , 11.87412439,
        5.94293185,  6.41202986,  4.89869205, 11.20100795,  9.55593374,
        8.94395628,  5.05774295, 10.8207297 , 13.3426541 , 12.25806559,
        7.58668144,  5.63824971,  8.47371868,  6.97106912,  7.52117132,
        5.16611782,  6.16512856,  5.62005629, 11.63724792, 11.50791859,
        9.97220799, 13.04217487, 13.54774355, 10.53368644, 12.889695  ,
       14.07623266,  6.02779201,  9.25336371,  8.34949383, 11.35345345,
        7.10378919, 12.41012046,  5.62005629,  9.97964013,  5.69913817,
       13.45642379, 12.13568598, 13.744387  ,  8.47371868, 13.59170809,
        5.94535529,  6.19707944,  7.50917551, 13.76047999,  7.73692436,
        4.98989906, 12.25806559, 11.69264243, 10.84437705,  7.02326515,
       12.32545637, 10.31584102, 11.22993218,  9.66977802,  6.33517349,
        8.41148722,  9.94285511, 10.8992524 ,  8.43626114, 13.99877413,
        9.66977802, 13.66051884,  7.85919687,  8.92549752, 11.52733581,
       11.73397309])}
{'P': tensor([[2.5558e-02, 1.7157e-01, 1.6944e-01,  ..., 4.0836e-03, 1.7273e-06,
         3.2218e-04],
        [6.1303e-04, 1.8592e-02, 1.6648e-01,  ..., 4.7126e-03, 1.0595e-06,
         3.2398e-04],
        [4.3472e-04, 1.2170e-03, 1.8724e-02,  ..., 6.5193e-03, 9.2015e-07,
         2.6814e-04],
        ...,
        [1.0873e-04, 1.4976e-04, 1.7771e-04,  ..., 6.3155e-02, 1.1937e-06,
         3.2768e-05],
        [5.4339e-05, 3.7009e-05, 2.7256e-05,  ..., 7.7347e-07, 2.6333e-02,
         4.2999e-05],
        [1.3395e-04, 2.6730e-04, 2.8222e-04,  ..., 2.5299e-04, 1.7298e-05,
         1.1191e-01]]), 'X': tensor([[ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0341, 0.0315,  ..., 0.3663, 1.0527, 0.8138],
        [0.0341, 0.0000, 0.0195,  ..., 0.3650, 1.0189, 0.7824],
        [0.0315, 0.0195, 0.0000,  ..., 0.3472, 1.0288, 0.7968],
        ...,
        [0.3663, 0.3650, 0.3472,  ..., 0.0000, 1.1202, 0.9867],
        [1.0527, 1.0189, 1.0288,  ..., 1.1202, 0.0000, 0.3557],
        [0.8138, 0.7824, 0.7968,  ..., 0.9867, 0.3557, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.78325151,  7.02326515,  7.09887557,  7.15990676,  7.16747878,
        7.19513806,  6.76056625,  8.31625171,  7.5158489 ,  6.12886024,
        6.74872981,  4.91296205,  8.86082124,  9.14044984, 13.744387  ,
       10.412299  ,  5.68942572,  5.21280156,  7.02496223,  8.37954853,
       11.73567759,  7.49076899,  9.11473391,  6.29034958, 13.89974693,
        6.33075682, 10.90638066,  6.02588517,  5.05774295,  6.67108662,
       11.28093507, 11.83075005,  5.24461068,  5.84533345,  7.4544866 ,
        7.73151309,  7.06169391, 11.63502679,  6.41202986,  9.50654967,
        7.88677538, 13.68063098,  6.48486171, 11.89855267, 11.97238624,
        8.41171369,  9.76896904, 13.93790235,  6.4020404 ,  5.23878371,
       13.04217487, 10.68031351,  5.54776934,  6.19633139,  7.96181288,
       11.05672136, 13.61801526,  6.05113217,  8.67227386, 13.09426142,
        6.50409372,  4.9160962 ,  4.73784222,  7.56617037,  8.92098752,
        6.1561646 ,  9.34127013,  7.78980299, 13.72594022, 11.23201619,
        6.24187044,  9.5642993 ,  9.76896904, 10.4977132 , 10.46739073,
        8.20436764,  9.18690018,  9.50913849,  7.6374603 , 13.26209545,
        7.9337647 ,  9.61942445, 10.42667093,  6.69389431,  9.43737523,
        7.73151309,  6.30647746,  4.92497514, 11.63765273, 13.41713661,
        7.50917551, 13.14777433,  8.04718433, 11.51557166, 11.97238624,
       13.01728313, 12.62711252,  8.8353674 ,  8.01009083, 12.59969936,
       11.89105259, 10.66631784,  7.4544866 ,  8.06643487, 11.35345345,
        7.58378655])}
{'P': tensor([[3.1304e-02, 1.7709e-01, 1.5236e-01,  ..., 1.3858e-04, 3.8958e-06,
         1.0431e-04],
        [5.1746e-04, 2.7661e-02, 1.7782e-03,  ..., 9.6131e-05, 2.6614e-06,
         6.8850e-05],
        [1.8762e-03, 1.7335e-01, 4.3834e-02,  ..., 1.6039e-04, 5.3603e-06,
         1.0291e-04],
        ...,
        [4.1573e-03, 1.5764e-03, 2.2426e-03,  ..., 5.8003e-02, 5.0881e-04,
         1.2478e-02],
        [8.4169e-05, 2.7102e-05, 5.5655e-05,  ..., 3.1164e-04, 2.3948e-02,
         7.4939e-05],
        [4.0326e-03, 1.7077e-03, 1.8897e-03,  ..., 1.0043e-02, 1.2953e-04,
         7.9607e-02]]), 'X': tensor([[  5.9026,   3.3837,  -0.6573],
        [  4.4633,   5.6287,  -0.7754],
        [  5.1793,   4.7461,   0.3899],
        [  4.1348,   5.9514,  -0.8856],
        [  5.2199,   4.6861,  -1.9304],
        [  4.0591,   6.0202,  -1.6339],
        [ -6.4620,  -8.0449,   6.2719],
        [ -6.3066,  -8.1936,   9.5822],
        [  4.1140,  -3.7220,  -0.1273],
        [  3.1064,  13.5607,   6.7605],
        [  6.3205,   1.7068,  -3.8619],
        [  8.2154,  -8.4262,   8.0171],
        [  1.7903,  -4.7467,   0.1168],
        [  5.1519, -10.2387,   9.8191],
        [  3.1064,  13.5607,   6.7605],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.1579,  -0.5961,  -6.5595],
        [  1.3339,   7.5627,  -0.9034],
        [  9.0129,  -7.7072,   9.1400],
        [  0.8700,  -4.8132,  -7.8042],
        [  4.6063,  13.0051,  -3.1456],
        [  1.4749,  -4.7892,  -0.8089],
        [  1.8766,   7.3694,   1.1887],
        [  6.5428,  -9.5728,  -2.9897],
        [  3.9895,   6.0816,  -6.5235],
        [  7.8375,  -8.7240,  -9.8913],
        [  1.5246,  -4.7838,  -5.2095],
        [  8.2154,  -8.4262,   8.0171],
        [  6.8704,  -9.3809,   7.5769],
        [  5.6928,  12.4710,   3.6001],
        [ -0.0254, -10.9932,   3.2200],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.1820,   9.7334,  -2.7932],
        [  5.1196,  -2.7160,   4.2705],
        [  4.0642,  -3.7599,  -3.4565],
        [  1.7659,   7.4127,   4.7871],
        [  3.9022,  13.2906,   5.5788],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.3616,  -2.4750,   3.7765],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.3550,   0.7129,   0.4917],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.4336,  -2.2676,   5.6933],
        [ -9.1907,  -3.3180,  -5.5943],
        [  1.4150,  13.9647,  -4.0915],
        [  5.9026,   3.3837,  -0.6573],
        [ 10.0619,   8.6627,  -9.2884],
        [  8.6271,  -8.0715,  -7.4813],
        [  8.8091,  -7.9037,   0.3631],
        [  1.7000,  -4.7609,   2.7935],
        [ -3.6381,   7.4683,   7.3544],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.3076,   0.7852,  -6.7010],
        [ 12.4013,  -1.2597,  -7.6474],
        [  0.2571,  -4.7594,   1.5618],
        [ -4.4520,   7.1370,  -8.7867],
        [  9.6587,   9.1817,  -0.8212],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.9284,  11.7061,  -0.3732],
        [  1.9503,  -4.7172,  -6.2313],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.8318,   3.5567,  -4.7692],
        [ 12.5717,   1.9266,   7.4903],
        [  2.9470,  13.6085,   2.9809],
        [  2.2437,  -4.6489,   6.1430],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.0031,  12.2958,  -6.5992],
        [  4.3459,  -3.5327,   0.3667],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.0087,   4.0173,   9.1091],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.1624,  -3.4286,   2.0647],
        [ -1.3940, -10.7772,   6.7835],
        [ 12.2435,   4.0076,   8.1257],
        [ -1.3940, -10.7772,   6.7835],
        [  5.4336,  -2.2676,   5.6933],
        [  2.1771, -10.9776,  -3.2007],
        [ 12.4690,  -0.8528,  -9.3374],
        [  6.1222,  -0.7261,   8.8204],
        [ -8.4573,   3.2448,   5.4370],
        [  6.1755,   2.5233,  -2.0042],
        [  2.6078, -10.9229,   6.2869],
        [  7.5064,  -8.9651,  -0.0994],
        [ 12.5763,   0.1333,  -9.2650],
        [ -9.1624,  -3.4286,   2.0647],
        [ -4.7334,   6.9982,  -0.4392],
        [ -0.3432,   7.8900,   7.1805],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 11.6019,   5.9540,  -5.1715],
        [ -9.2508,   1.0572,   2.9978],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.1751,  -2.2592,  -5.7751],
        [  4.8051, -10.3703,   0.5268],
        [ -1.0407,   7.9166,  -6.6021],
        [  4.3554,   5.7393,  -9.0766],
        [ -7.0825,   5.2073,  -5.0497],
        [  5.5294,  -2.1106,   9.0626],
        [ 12.1940,  -2.1881,   7.1499],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.0547,   7.2957,  -1.9343],
        [ -2.0721,   7.8432,   6.6380],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.9659,  -4.8144,  -0.9594],
        [  1.1644, -11.0394,   7.8511],
        [  1.5246,  -4.7838,  -5.2095]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0502, 0.0763,  ..., 0.1152, 0.9029, 0.2767],
        [0.0502, 0.0000, 0.0951,  ..., 0.0650, 0.8845, 0.2497],
        [0.0763, 0.0951, 0.0000,  ..., 0.1457, 0.8364, 0.3439],
        ...,
        [0.1152, 0.0650, 0.1457,  ..., 0.0000, 0.8680, 0.2233],
        [0.9029, 0.8845, 0.8364,  ..., 0.8680, 0.0000, 1.0685],
        [0.2767, 0.2497, 0.3439,  ..., 0.2233, 1.0685, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.80370254,  7.18354885,  7.02496223,  7.24677664,  7.01475376,
        7.26074925, 10.31886066, 10.33958639,  5.54776934, 13.91198191,
        6.54693987, 11.76830981,  5.07306274, 11.46176259, 13.91198191,
        9.14973319,  6.18668463,  7.67939546, 11.85890664,  4.89121805,
       13.79676654,  5.01114719,  7.60463349, 11.59512194,  7.27340834,
       11.72749923,  5.0209005 , 11.76830981, 11.62770332, 13.70893492,
       10.99326577, 12.10222204, 13.38091121,  5.79543038,  5.5366698 ,
        7.62011734, 13.85158417,  8.09984231,  9.09661212,  9.68324553,
        8.94395628,  6.3949043 ,  9.70108409,  5.88782913,  9.7712321 ,
       14.03618649,  6.80370254, 13.27717982, 11.81423161, 11.83509372,
        5.05534609,  8.30729352,  4.8763135 ,  9.34061938, 12.46513646,
        4.7663526 ,  8.41171369, 13.32645552, 11.51811603, 13.60275211,
        5.10444097, 10.24557047,  6.83083863, 12.71843462, 13.92390689,
        5.16205087,  5.9510489 , 13.6829843 ,  5.60064319,  8.01009083,
        8.95983586,  5.68942572,  9.78285424, 10.86693708, 12.88270018,
       10.86693708,  5.88782913, 11.1913598 , 12.4980845 ,  6.16512856,
        9.05843201,  6.67108662, 11.22993218, 11.69264243, 12.5769659 ,
        9.78285424,  8.4486771 ,  7.89745107, 12.35625694, 13.04051744,
        9.3109884 ,  5.91677188, 12.3828957 , 11.42947632,  7.98469166,
        7.20482224,  8.7907923 ,  5.91854397, 12.38881635, 11.72749923,
        7.57946459,  8.11227668,  5.18277858,  4.91038884, 11.10066126,
        5.0209005 ])}
{'P': tensor([[1.9509e-02, 1.7749e-01, 1.4130e-01,  ..., 1.1797e-04, 3.0149e-06,
         1.0215e-04],
        [5.5343e-04, 1.8151e-02, 1.6252e-01,  ..., 6.9946e-05, 2.4632e-06,
         9.0374e-05],
        [6.4636e-04, 1.7265e-03, 2.6561e-02,  ..., 8.4562e-05, 3.3407e-06,
         1.0434e-04],
        ...,
        [1.7139e-02, 1.2103e-02, 9.1416e-03,  ..., 8.5734e-02, 1.3053e-04,
         4.9541e-04],
        [4.3471e-04, 2.3055e-04, 1.7144e-04,  ..., 1.8353e-02, 9.9347e-02,
         1.0515e-05],
        [9.2467e-05, 1.9345e-04, 2.2299e-04,  ..., 3.2142e-05, 2.5325e-06,
         2.8803e-01]]), 'X': tensor([[  5.8979,   3.3958,  -3.8713],
        [  5.1861,   4.7361,  -3.7312],
        [  4.8653,   5.1694,  -4.0823],
        [  5.0233,   4.9647,  -3.1788],
        [  4.5800,   5.5034,  -3.9432],
        [  4.5430,   5.5438,  -3.8669],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.5424,  -2.0885,   7.2487],
        [  1.7500,   7.4187,   4.4237],
        [  1.1994,   7.6035,   2.4295],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.9282,   2.1648,   8.4894],
        [  2.2437,  -4.6489,   6.1430],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.6050,   2.9451,  -0.7012],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.7500,   7.4187,   4.4237],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.4541,  -0.3327,   8.8246],
        [ 11.1469,   6.9237,   1.6160],
        [ 12.6000,   1.3839,  -3.2992],
        [  1.4306,  13.9620,  -0.2075],
        [  7.8425,  -8.7202,   6.4530],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.3107,  -2.7697,  -3.2919],
        [  0.6933,   7.7336,   2.1179],
        [ -9.4716,  -0.6673,   7.6068],
        [ -2.4956, -10.4682,   3.5632],
        [  9.1820,   9.7334,  -2.7932],
        [  6.2247,  -0.3150,  -9.3657],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.9410,   6.8428,   5.4458],
        [  4.4039,   5.6903,   8.7366],
        [ 11.5456,   6.0861,   8.1101],
        [ -6.3066,  -8.1936,   9.5822],
        [  6.2419,  -0.2313,   4.2440],
        [  1.4489,   7.5256,   4.4529],
        [ -9.0574,   1.7801,   8.8986],
        [ -2.5683, -10.4433,  -2.9182],
        [ -7.9925,  -6.1423,   2.8990],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.3516,   0.6425,   0.3425],
        [  6.1222,  -0.7261,   8.8204],
        [  5.2939,  -2.4784,  -4.5840],
        [  0.6396,  14.0772,  -6.3786],
        [  4.5767,   5.5071,   3.9069],
        [  6.3497,   1.3680,  -1.9076],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.4877,  -0.7222,   8.8347],
        [  2.5663,   7.0526,  -6.8769],
        [  7.9816,  10.8938,  -0.5378],
        [  2.5636, -10.9293,   9.6519],
        [ -1.6220,   7.8917,   0.0315],
        [  8.0381,  -8.5690,  -8.9523],
        [  3.2990,  -4.2415,  -7.4857],
        [  5.3118,   4.5452,   4.2114],
        [ -3.2486,   7.5925,  -1.3719],
        [  4.2612, -10.5514,  -6.6268],
        [  0.2480, -11.0153,  -3.0183],
        [  0.3231,  -4.7691,  -1.6120],
        [  3.5965,  -4.0754,  -3.5035],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.3118,   4.5452,   4.2114],
        [  2.5469,  13.7192,  -9.6935],
        [  3.9895,   6.0816,  -6.5235],
        [  6.2038,   2.4020,  -3.1692],
        [  2.0870,  -4.6877,   9.6823],
        [  3.5939,   6.4029,  -1.8357],
        [ -5.0550,   6.8231,  -9.2502],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.0766,  -6.0046,  -3.1873],
        [  1.5383, -11.0274,  -4.9666],
        [ -7.0825,   5.2073,  -5.0497],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.5650,   2.0132,   4.6272],
        [  0.4119,  14.1019,  -4.1870],
        [ -1.7290,   7.8825,  -3.3831],
        [ 11.3800,  -4.3919,   1.4066],
        [  4.5767,   5.5071,   3.9069],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.2692,  13.5098,  -4.9032],
        [  2.8602,   6.8905,   5.8779],
        [  4.8653,   5.1694,  -4.0823],
        [  9.9750,  -6.6338,  -1.2440],
        [  2.2373,  -4.6506,  -1.7334],
        [  8.5716,  10.3596,   6.8818],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.9808,   4.9340,  -3.3715],
        [  3.0719, -10.8449,  -4.1673],
        [  4.9152,  12.8653,  -1.4459],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.5503,   5.5359,   7.4735],
        [  5.0506,   4.9277,  -9.5094],
        [  5.9587,  12.3216,   9.1695],
        [ -5.9940,   6.1985,   5.9566],
        [  7.9141,  -8.6656,   9.6948],
        [  5.4429,  -2.2529,   9.7306],
        [  5.1667,  -2.6543,   1.2879],
        [  0.5960, -11.0335,   3.1144],
        [  5.9587,  12.3216,   9.1695]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0182, 0.0307,  ..., 0.3309, 0.8502, 1.1284],
        [0.0182, 0.0000, 0.0195,  ..., 0.3273, 0.8341, 1.1309],
        [0.0307, 0.0195, 0.0000,  ..., 0.3444, 0.8373, 1.1501],
        ...,
        [0.3309, 0.3273, 0.3444,  ..., 0.0000, 0.6841, 0.8271],
        [0.8502, 0.8341, 0.8373,  ..., 0.6841, 0.0000, 1.1621],
        [1.1284, 1.1309, 1.1501,  ..., 0.8271, 1.1621, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.80559669,  7.02326515,  7.09887557,  7.06272256,  7.15990676,
        7.16747878, 11.83509372,  9.48969978,  9.09502385,  9.68323686,
        5.92281142,  7.62232307,  7.69752262,  4.85995002,  4.7663526 ,
        9.18690018,  5.16205087, 12.4013153 ,  9.09502385, 10.35093485,
        7.62232307,  9.79039371,  6.1463896 ,  9.45995955, 13.12218013,
       12.67576792, 14.03505624, 11.72802491, 13.59673406,  9.71392179,
        7.76457095,  9.49511524, 10.76154256, 13.38091121,  6.23262289,
        9.4997987 ,  7.44805901,  7.19534146, 13.05148826, 10.33958639,
        6.24614905,  7.66377554,  9.23071698, 10.75443106, 10.08001894,
       10.52587243,  9.44486609,  6.38399789,  6.16512856,  5.84533345,
       14.09176259,  7.16058587,  6.49538841,  4.8763135 , 12.5085995 ,
        7.50500064, 13.50484539, 11.22596994,  8.05668609, 11.74901778,
        5.37343173,  6.99097013,  8.25828409, 11.37939695, 11.018084  ,
        4.78002665,  5.43544626,  8.16714836,  6.99097013, 13.95361098,
        7.27340834,  6.65259775,  5.13125513,  7.34250948,  8.49161154,
        8.04664886, 10.06407839, 11.1341798 ,  8.7907923 , 11.28895489,
       12.72524615, 14.10796466,  8.06990483, 12.19805394,  7.16058587,
        5.34250225, 13.89974693,  7.46053281,  7.09887557, 11.97949273,
        5.16077764, 13.44593519, 12.91505111, 12.95702647, 11.27160035,
       13.77222752, 13.09737053,  7.16598273,  7.05627461, 13.68672855,
        8.62260327, 11.73567759,  5.89074104,  5.80860165, 11.04953767,
       13.68672855])}
{'P': tensor([[2.8662e-02, 2.4035e-01, 1.5336e-01,  ..., 1.7876e-03, 6.9284e-06,
         4.3644e-04],
        [1.4665e-03, 4.4979e-02, 3.2461e-02,  ..., 3.4214e-03, 1.0240e-05,
         1.3059e-03],
        [1.9327e-03, 3.8979e-02, 4.5135e-02,  ..., 4.2023e-03, 1.0472e-05,
         1.5570e-03],
        ...,
        [5.8875e-05, 1.4683e-04, 1.5760e-04,  ..., 6.7223e-02, 9.7963e-07,
         1.4934e-02],
        [8.1510e-05, 4.1706e-05, 4.3170e-05,  ..., 7.9366e-07, 1.4096e-02,
         1.1928e-06],
        [2.6145e-05, 3.2627e-05, 3.5393e-05,  ..., 9.9317e-06, 8.1152e-05,
         5.3723e-02]]), 'X': tensor([[ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0316, 0.0673,  ..., 0.7973, 0.7568, 0.6970],
        [0.0316, 0.0000, 0.0431,  ..., 0.7905, 0.7284, 0.6657],
        [0.0673, 0.0431, 0.0000,  ..., 0.7523, 0.7212, 0.6351],
        ...,
        [0.7973, 0.7905, 0.7523,  ..., 0.0000, 1.2088, 0.8410],
        [0.7568, 0.7284, 0.7212,  ..., 1.2088, 0.0000, 0.4285],
        [0.6970, 0.6657, 0.6351,  ..., 0.8410, 0.4285, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.81084724,  7.02496223,  7.02253235,  7.18354885,  7.24677664,
        7.20826219,  8.859416  , 13.02818702,  7.58668144, 13.43127131,
        6.67108662,  5.5513072 ,  6.61211224, 10.09979747, 13.79762415,
        9.15253344,  7.05627461,  8.01009083, 13.79762415,  5.39098669,
       13.01728313, 13.88683436,  5.91854397, 11.63383221,  9.14973319,
       12.10942412,  5.01114719,  5.24137909, 13.56367857, 12.12641969,
        5.05981914, 12.44177641,  5.80860165,  8.74470218, 11.31276109,
        5.05963217,  9.87720102,  5.5748624 , 12.33167382,  5.0713364 ,
       13.92390689, 11.97238624,  8.06990483,  7.01475376, 12.5085995 ,
       12.13568598,  6.89245352,  8.27853076,  4.75303127,  5.94541803,
        8.7273569 ,  7.9337647 ,  6.383897  ,  8.12580976,  8.7907923 ,
        7.32950385, 10.04916678,  6.54693987, 10.31613269,  4.79003361,
        8.89301516,  6.2134961 ,  8.98765644, 10.4977132 , 11.75807252,
       11.0050379 , 11.63383221,  6.90115901,  7.14837514,  5.53175275,
       13.99048094,  4.91652556,  6.78325151,  8.91551231, 13.09426142,
       13.77222752, 13.25336001,  6.77777606,  8.05242591,  8.43626114,
        9.11473391,  8.92549752,  5.94535529,  5.16077764, 12.27835996,
       11.9702762 , 10.48721382, 10.38897563,  5.83560009,  6.81117802,
       13.2778378 ,  5.18277858,  4.91336994, 10.6910768 ,  5.23878371,
        8.35856487, 12.27835996,  9.60080639, 11.1341798 ,  5.90035324,
        5.37343173,  7.62011734, 12.07225469, 13.77604293, 10.9036226 ,
        8.53014915])}
{'P': tensor([[2.4675e-02, 2.1623e-01, 1.8542e-01,  ..., 6.3507e-04, 4.2401e-02,
         1.7105e-02],
        [7.5642e-04, 2.3611e-02, 2.1142e-01,  ..., 6.2066e-04, 2.8937e-02,
         1.6943e-02],
        [7.1888e-04, 1.9130e-03, 2.9431e-02,  ..., 6.0284e-04, 3.6503e-02,
         2.5744e-02],
        ...,
        [3.1810e-04, 5.7742e-04, 6.1472e-04,  ..., 9.7989e-02, 3.5136e-04,
         3.0459e-04],
        [1.3693e-03, 3.6952e-03, 5.9141e-03,  ..., 3.9739e-04, 7.4132e-02,
         3.5550e-01],
        [6.7473e-04, 1.3030e-03, 1.7786e-03,  ..., 1.7812e-04, 2.9369e-03,
         8.2766e-02]]), 'X': tensor([[ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0214, 0.0280,  ..., 0.7356, 0.1337, 0.2217],
        [0.0214, 0.0000, 0.0195,  ..., 0.7213, 0.1357, 0.2187],
        [0.0280, 0.0195, 0.0000,  ..., 0.7369, 0.1164, 0.1994],
        ...,
        [0.7356, 0.7213, 0.7369,  ..., 0.0000, 0.8407, 0.8862],
        [0.1337, 0.1357, 0.1164,  ..., 0.8407, 0.0000, 0.0951],
        [0.2217, 0.2187, 0.1994,  ..., 0.8862, 0.0951, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.81117802,  7.02326515,  7.09887557,  7.15990676,  7.16747878,
        7.06272256, 13.59170809, 12.30695098,  8.94395628,  5.57341623,
        5.05981914,  9.47369628, 10.9036226 ,  7.44805901, 10.9036226 ,
        8.50278896,  8.83725081, 11.04313641,  4.85995002, 10.57302546,
       13.77075174,  6.67121748,  7.13501264,  8.74470218, 13.46553623,
        6.2134961 , 10.06850677, 11.70016217,  4.87083348, 11.33837396,
       10.32715715,  8.16584537,  9.3349858 ,  7.9337647 , 12.53345445,
       12.16901615, 11.73567759,  7.88677538, 12.35421461, 10.69963954,
        9.43737523,  5.5366698 ,  7.06169391,  5.08372149,  9.51074763,
       10.65929968,  4.91296205,  4.87083348,  5.9510489 , 11.46176259,
       10.00794468,  8.53014915, 14.07623266,  5.35774573, 10.412299  ,
        8.50278896, 10.14436912,  4.83371341,  6.33075682, 13.6829843 ,
        7.96181288,  9.6282763 ,  5.84450127,  6.49538841,  9.63742669,
        9.7712321 ,  8.31625171, 13.40024224,  9.87720102, 11.10066126,
        5.92952538, 12.85032836, 10.33712258, 10.32715715,  6.58184474,
        4.76969364,  7.14837514,  7.69752262, 10.48840476, 12.91143491,
       13.48809595, 13.98691426, 10.06703358, 11.54672755, 13.89671314,
        4.90505566,  9.60359158,  5.41348121, 11.87412439, 10.31886066,
        9.26455795,  4.92497514,  9.94285511,  7.66377554, 13.35013374,
        4.92497514, 11.92133129,  5.1224079 ,  6.39520397, 10.37998078,
        5.83560009, 10.4977132 ,  5.39450305,  7.38844703,  7.27340834,
        7.50035355])}
{'P': tensor([[2.9866e-02, 3.2708e-01, 2.4759e-01,  ..., 1.6525e-04, 5.5334e-06,
         6.1234e-06],
        [1.3133e-03, 3.5157e-02, 4.4960e-01,  ..., 1.4089e-04, 6.3678e-06,
         6.6013e-06],
        [1.5747e-03, 2.4805e-03, 5.3728e-02,  ..., 1.8467e-04, 9.9168e-06,
         9.9727e-06],
        ...,
        [1.2878e-02, 1.1112e-02, 1.0296e-02,  ..., 9.8261e-02, 8.7556e-05,
         5.5192e-04],
        [6.5363e-05, 4.2476e-05, 3.3935e-05,  ..., 2.2321e-03, 2.2807e-02,
         3.7394e-04],
        [1.2994e-04, 1.1600e-04, 1.1190e-04,  ..., 3.5884e-03, 2.2547e-04,
         1.2601e-01]]), 'X': tensor([[  5.8617,   3.4853,  -9.8080],
        [  5.3858,   4.4260,  -9.6769],
        [  5.0506,   4.9277,  -9.5094],
        [  4.3554,   5.7393,  -9.0766],
        [  2.8994,   6.8675,  -9.5987],
        [  5.0190,   4.9705,  -7.9752],
        [  9.0004,  -7.7195,   4.6945],
        [  3.9022,  13.2906,   5.5788],
        [  8.7131,  10.2216,   7.7915],
        [ -3.3812, -10.1235,   7.4384],
        [  9.4554,   9.4245,   9.8800],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.5424,  -2.0885,   7.2487],
        [ 12.6059,   0.9597,  -6.3264],
        [  0.9945,  -4.8144,  -0.1897],
        [ -1.3940, -10.7772,   6.7835],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.2974,   0.8371,   2.4852],
        [  9.1409,  -7.5789,   1.5814],
        [  9.6662,   9.1725,   8.1056],
        [ 11.4495,   6.3026,   6.8257],
        [  6.2036,   2.4030,  -1.2500],
        [  6.2365,  -0.2578,  -6.0212],
        [  9.6662,   9.1725,   8.1056],
        [  8.1043,  -8.5164,  -7.9290],
        [  1.0093,   7.6567,   2.3526],
        [  0.6757, -11.0361,  -1.3047],
        [  2.9410,   6.8428,   5.4458],
        [  3.2366,  -4.2732,   5.1532],
        [  0.1382, -11.0073,   1.0026],
        [ -3.5173, -10.0623,   4.8280],
        [  9.2811,  -7.4336,   5.1771],
        [ 12.4691,   2.8525,  -5.5110],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.2576,   2.1359,  -3.3398],
        [  0.6952,  -4.8060,   8.8976],
        [  0.5250, -11.0306,   5.5358],
        [  5.6672,  -1.8631,   8.1020],
        [  6.3200,   0.2690,   1.5900],
        [ -5.9872,   6.2037,  -9.4477],
        [  4.5800,   5.5034,  -3.9432],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.2480, -11.0153,  -3.0183],
        [  5.2188,  -2.5839,   2.1833],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -5.1398,  -9.1367,   6.2074],
        [ -8.7544,   2.6096,   3.3146],
        [  3.3094,  13.4969,  -3.6507],
        [ 10.2712,   8.3707,  -4.9743],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.6928,  12.4710,   3.6001],
        [ 11.1469,   6.9237,   1.6160],
        [  7.5309,  11.2617,   7.2656],
        [  0.6933,   7.7336,   2.1179],
        [  3.4140,  -4.1803,  -8.8224],
        [ -0.8569,   7.9156,  -6.5557],
        [ -2.5056,   7.7720,   5.5878],
        [  4.6231,   5.4556,   8.4973],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -1.8823, -10.6558,  -3.8952],
        [  0.5600, -11.0321,  -0.8660],
        [  4.7544,  -3.1425,  -1.5315],
        [  6.3515,   0.6418,  -6.5823],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -7.9667,   4.0814,  -0.6397],
        [  9.6614,   9.1784,  -1.2087],
        [  0.8940, -11.0402,  -0.9783],
        [ -2.3317,   7.8034,  -3.6274],
        [  5.8617,   3.4853,  -9.8080],
        [  6.6319,  11.9066,   3.5165],
        [ -9.1813,  -3.3550,   1.8827],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.2687,  -2.9794,   0.6436],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.3179,  -9.0101,   9.4472],
        [  1.1350,  14.0104,   0.6321],
        [  1.6232,  -4.7717,  -2.5655],
        [  6.3119,   1.7797,   9.9186],
        [ -4.0868,   7.2981,   8.1874],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.3717,  -5.4769,  -9.2228],
        [  5.2939,  -2.4784,  -4.5840],
        [ 11.0864,   7.0387,  -0.1238],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.4505,  -1.7081,  -0.7039],
        [  9.3624,   9.5318,   0.2779],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.9531,   5.0576,   5.6794],
        [ -9.3107,  -2.7697,  -3.2919],
        [  3.7916,  -3.9524,  -7.9547],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.0305,   3.0296,   0.1375],
        [  7.0642,  11.6104,  -9.8316],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.3167,   6.6022,   6.3554],
        [  4.1297,  -3.7098,  -3.9838],
        [ -5.3382,   6.6533,   7.8054],
        [  3.1480,  -4.3165,  -7.8002],
        [ -3.5029,   7.5138,   8.1853],
        [  5.8617,   3.4853,  -9.8080],
        [ -8.7544,   2.6096,   3.3146],
        [  5.4179,  -2.2925,  -3.4641],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.2437,  -4.6489,   6.1430]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.3251e-02, 4.1585e-02,  ..., 2.1202e-01, 7.4121e-01,
         8.6660e-01],
        [2.3251e-02, 0.0000e+00, 1.8426e-02,  ..., 2.1794e-01, 7.1907e-01,
         8.6981e-01],
        [4.1585e-02, 1.8426e-02, 0.0000e+00,  ..., 2.2191e-01, 7.0096e-01,
         8.7035e-01],
        ...,
        [2.1202e-01, 2.1794e-01, 2.2191e-01,  ..., 1.5170e-04, 7.5014e-01,
         6.5546e-01],
        [7.4121e-01, 7.1907e-01, 7.0096e-01,  ..., 7.5014e-01, 0.0000e+00,
         1.0346e+00],
        [8.6660e-01, 8.6981e-01, 8.7035e-01,  ..., 6.5546e-01, 1.0346e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.81961996,  6.97106912,  7.05627461,  7.20482224,  7.4544866 ,
        7.06372172, 11.85743425, 13.85158417, 13.43127131, 10.67322527,
       13.35013374, 13.03002338,  5.92281142, 12.64235359,  4.9160962 ,
       10.86693708, 10.63038565,  9.3349858 , 11.87412439, 13.32557133,
       13.06957425,  6.652742  ,  6.24187044, 13.32557133, 11.75618513,
        7.72291946, 11.05672136,  7.44805901,  5.36061974, 11.00812996,
       10.65929968, 11.89105259, 12.79126785, 12.05300943, 10.53368644,
        6.61211224,  4.85603886, 11.04313641,  5.96556248,  6.32572323,
        8.62162372,  7.15990676, 12.19805394, 11.018084  ,  5.82345867,
       10.5575601 , 10.48315436,  9.13507321, 13.89671314, 13.25017211,
       10.01804223, 13.70893492, 13.12218013, 13.54774355,  7.76457095,
        5.39723241,  7.96181288,  8.16584537,  7.15100303, 12.27835996,
       10.8207297 , 11.04629471,  5.69913817,  6.383897  , 10.35485331,
        8.95133961, 13.32613445, 11.07637531,  8.14433542,  6.81961996,
       13.62896593,  9.7751189 ,  9.96152788,  9.73579738,  8.24808832,
       10.46238054, 14.05633441,  5.04028538,  6.55800267,  8.36445758,
        8.01009083, 10.00410625,  5.84533345, 13.13205357,  9.64157461,
        9.60359158, 13.3607379 , 13.03002338,  9.38232121,  7.07902036,
        9.71392179,  5.47702766, 10.48721382,  6.74872981, 13.59056613,
       12.52372629,  7.38844703,  5.5513072 ,  8.53014915,  5.34250225,
        8.29022197,  6.81961996,  9.13507321,  5.8828986 , 10.8207297 ,
        5.16205087])}
{'P': tensor([[2.5384e-02, 1.6257e-01, 1.4745e-01,  ..., 1.0567e-03, 1.3558e-04,
         7.5926e-06],
        [5.1091e-04, 1.7629e-02, 1.7876e-01,  ..., 2.6699e-04, 6.0335e-05,
         5.4996e-06],
        [1.7254e-03, 4.5176e-03, 8.8360e-02,  ..., 9.3497e-04, 2.7100e-04,
         2.3687e-05],
        ...,
        [4.7517e-02, 8.9480e-02, 6.9803e-02,  ..., 1.3395e-01, 8.2134e-04,
         2.0201e-04],
        [3.3261e-02, 1.4168e-02, 1.1267e-02,  ..., 1.5151e-02, 1.1743e-01,
         8.4731e-05],
        [2.6647e-04, 2.4316e-04, 1.8660e-04,  ..., 2.3222e-03, 2.8583e-04,
         1.8641e-01]]), 'X': tensor([[  5.8318,   3.5567,  -4.7692],
        [  4.8653,   5.1694,  -4.0823],
        [  4.4049,   5.6892,  -4.4394],
        [  4.5800,   5.5034,  -3.9432],
        [  4.5430,   5.5438,  -3.8669],
        [  5.1292,   4.8183,  -5.7997],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.2849, -10.9655,  -7.6542],
        [ -6.8466,  -7.6488,   1.9648],
        [  6.2247,  -0.3150,  -9.3657],
        [ -0.7599, -10.8995,  -6.0366],
        [  6.3360,   0.4258,  -9.2509],
        [  5.3804,  -2.3504,   7.2744],
        [  6.9956,  11.6590,  -7.6876],
        [  9.3424,  -7.3685,  -3.6807],
        [ -1.5408,   7.8978,   5.3470],
        [  2.1477,   7.2550,   8.9688],
        [  9.3753,   9.5171,   3.9131],
        [  6.0047,   3.1062,  -6.7914],
        [ 11.6634,   5.8050,   9.1590],
        [  6.8093,  -9.4178,   8.4244],
        [  5.6248,   3.9984,   8.4933],
        [  4.5340,  -3.3629,   8.9652],
        [ -1.8788, -10.6567,   3.7068],
        [ -2.0721,   7.8432,   6.6380],
        [  0.1041, -11.0045,   2.4311],
        [  6.0047,   3.1062,  -6.7914],
        [  7.7794,  11.0629,   8.9069],
        [ -5.9872,   6.2037,  -9.4477],
        [  5.9843,  -9.8679,  -6.1449],
        [  2.5010,  -4.5736,  -0.8989],
        [ 11.1469,   6.9237,   1.6160],
        [ -3.6381,   7.4683,   7.3544],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.2849, -10.9655,  -7.6542],
        [  0.1041, -11.0045,   2.4311],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.2608,  -2.5256,  -9.1446],
        [ 12.2435,   4.0076,   8.1257],
        [ 10.0400,   8.6922,  -6.2847],
        [ 10.2712,   8.3707,  -4.9743],
        [ -5.1398,  -9.1367,   6.2074],
        [  4.5430,   5.5438,  -3.8669],
        [  6.3031,   1.8477,  -7.1706],
        [ -5.6169,  -8.7844,   8.9057],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.7794,   5.5086,  -9.6929],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.8093,  -9.4178,   8.4244],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.2435,   7.8191,  -6.4709],
        [  4.5503,   5.5359,   7.4735],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.9883,   3.1536,   6.2320],
        [  4.3459,  -3.5327,   0.3667],
        [  6.2975,   0.0947,   5.1791],
        [  3.4920,  -4.1366,  -8.3955],
        [  8.8474,  10.0868,  -8.8644],
        [  6.3587,   1.1596,  -3.6629],
        [ -6.4822,  -8.0252,  -3.8983],
        [  0.5250, -11.0306,   5.5358],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -0.1626, -10.9796,   6.4629],
        [ -9.4721,  -1.3097,  -7.4377],
        [  6.9956,  11.6590,  -7.6876],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -6.3066,  -8.1936,   9.5822],
        [ -5.6792,  -8.7352,   4.0318],
        [ -5.6792,  -8.7352,   4.0318],
        [ -3.1671,   7.6158,   6.0765],
        [  5.5294,  -2.1106,   9.0626],
        [  4.7544,  -3.1425,  -1.5315],
        [ -1.0482, -10.8487,  -5.9066],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.2860,   7.1916,  -6.1011],
        [ -5.1398,  -9.1367,   6.2074],
        [ -8.7735,   2.5640,  -6.3313],
        [ -9.4505,  -1.7081,  -0.7039],
        [  2.7121,   6.9743,  -8.4202],
        [ -7.8280,   4.2848,  -7.1955],
        [  2.8890,  -4.4316,   1.1600],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.4203,  -2.0340,   2.5856],
        [  5.3858,   4.4260,  -9.6769],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.3087,   0.7792,  -7.5587],
        [  2.5620,   7.0549,   8.4825],
        [  4.8051, -10.3703,   0.5268],
        [  5.5240,  -2.1198,   6.2055],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 11.7794,   5.5086,  -9.6929],
        [  3.3069,   6.6089,   8.7377],
        [  9.0051,   9.9234,   5.4411],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.5083,  -4.1177,   0.2389],
        [ -1.6220,   7.8917,   0.0315],
        [ -9.4731,  -0.7120,  -6.0271],
        [  6.3246,   1.6696,  -0.1544],
        [  4.0078,  -3.8018,  -6.2168],
        [  1.2273,  -4.8081,   9.8129]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0374, 0.0505,  ..., 0.2935, 0.0461, 0.8391],
        [0.0374, 0.0000, 0.0297,  ..., 0.2674, 0.0744, 0.8095],
        [0.0505, 0.0297, 0.0000,  ..., 0.2879, 0.0685, 0.8254],
        ...,
        [0.2935, 0.2674, 0.2879,  ..., 0.0000, 0.3389, 0.5483],
        [0.0461, 0.0744, 0.0685,  ..., 0.3389, 0.0000, 0.8833],
        [0.8391, 0.8095, 0.8254,  ..., 0.5483, 0.8833, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.83083863,  7.09887557,  7.19513806,  7.15990676,  7.16747878,
        7.03733775, 11.74901778, 11.20100795, 10.26545792,  6.23262289,
       10.92597196,  6.35029274,  5.87132718, 13.59673406, 11.89855267,
        8.04664886,  7.56617037, 13.35927527,  6.76056625, 13.02818702,
       11.62158554,  6.90115901,  5.64500512, 10.82106505,  8.11227668,
       11.0050379 ,  6.76056625, 13.52430176,  8.62162372, 11.54071859,
        5.21280156, 13.12218013,  8.30729352, 12.4217523 , 11.20100795,
       11.0050379 , 12.12641969,  9.61942445,  5.83560009, 12.88270018,
       13.27994304, 13.25017211, 10.48315436,  7.16747878,  6.56833864,
       10.42667093, 12.53047489, 13.00380624, 10.61047596,  9.43737523,
       12.4013153 , 11.62158554, 12.3344536 ,  9.64157461,  7.822854  ,
        7.16598273, 11.80092578,  6.76791483,  5.60064319,  6.29821673,
        5.41348121, 13.41713661,  6.46356806, 10.31613269, 11.04313641,
       12.10942412, 10.98076873,  9.562174  , 13.59673406, 12.49317563,
       10.33958639, 10.41909211, 10.41909211,  8.24808832,  5.91854397,
        5.69913817, 10.8992524 ,  9.78147542,  7.54620849, 10.48315436,
        9.14044984,  9.60359158,  7.48311002,  8.92395381,  5.29009836,
       12.4013153 ,  9.63742669,  6.97106912, 10.65929968,  9.34127013,
        7.50564223, 11.42947632,  5.91677188, 10.37698435,  9.69629   ,
       12.35625694, 13.00380624,  7.39004226, 13.40024224,  7.83486943,
       12.22276278,  8.05668609,  9.4997987 ,  6.54127431,  5.52416368,
        4.96231532])}
{'P': tensor([[3.4795e-02, 3.1266e-01, 1.5591e-01,  ..., 7.0831e-06, 1.0221e-05,
         8.0387e-05],
        [1.6456e-03, 5.6988e-02, 2.2639e-01,  ..., 6.9457e-06, 2.3080e-05,
         2.1726e-04],
        [1.9234e-03, 6.4694e-03, 9.0654e-02,  ..., 6.9053e-06, 3.5941e-05,
         4.7384e-04],
        ...,
        [1.0597e-04, 6.5385e-05, 3.1869e-05,  ..., 7.2230e-03, 2.8531e-07,
         1.5926e-07],
        [1.1990e-05, 8.2749e-06, 5.4143e-06,  ..., 3.2348e-04, 4.1370e-02,
         1.3066e-04],
        [1.4807e-04, 1.1221e-04, 9.6489e-05,  ..., 1.4621e-04, 2.0915e-02,
         2.5438e-01]]), 'X': tensor([[ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0378, 0.0864,  ..., 0.5697, 1.0771, 0.9662],
        [0.0378, 0.0000, 0.0536,  ..., 0.5580, 1.0446, 0.9284],
        [0.0864, 0.0536, 0.0000,  ..., 0.5810, 1.0279, 0.8848],
        ...,
        [0.5697, 0.5580, 0.5810,  ..., 0.0000, 0.7360, 0.9808],
        [1.0771, 1.0446, 1.0279,  ..., 0.7360, 0.0000, 0.6169],
        [0.9662, 0.9284, 0.8848,  ..., 0.9808, 0.6169, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.83515201,  7.06372172,  7.20482224,  7.05627461,  7.27340834,
        7.48311002, 10.48840476,  6.18668463, 13.58101794, 10.31886066,
       10.48315436, 11.97238624,  9.32094055,  9.50654967,  9.17905491,
        7.02326515, 13.79762415,  5.92952538, 10.38897563,  8.85939626,
       11.11969815,  7.03733775, 11.67875855,  8.79284466,  9.15866607,
        6.05165248,  6.34059021, 13.6105595 ,  9.12220271, 13.6922425 ,
        5.13125513,  5.05774295, 13.37121923,  5.89074104, 11.92133129,
        6.42689705, 13.32645552,  7.13501264,  6.81084724, 13.77222752,
        7.32233733,  6.99097013,  6.76661595, 11.54672755,  9.63742669,
       12.18998195,  8.10886021,  9.75200848, 13.83139024, 10.65929968,
       11.72749923,  9.11473391,  9.09502385,  8.62329466, 13.05148826,
        8.06517143,  5.05534609, 11.00508681,  5.05444288, 13.89671314,
        9.37989631, 13.45642379, 12.70489563, 12.25806559,  9.48969978,
        9.7751189 ,  5.13125513, 12.78327037,  6.27905292,  7.62509582,
        8.56889738, 11.04629471, 10.46739073,  5.87305005,  7.02326515,
        9.55593374,  8.87007495,  9.5642993 ,  5.64500512, 10.7506423 ,
        8.55982105,  6.19633139, 11.05317597, 12.25806559, 10.39942497,
        8.43626114, 13.79762415,  6.77777606,  8.31625171, 13.99048094,
        5.39098669, 13.32374056,  7.85919687, 13.59673406,  7.92607342,
       10.82106505,  7.44805901,  5.0713364 ,  5.90035324,  5.84450127,
        7.19513806, 13.96905756,  5.5205189 , 11.87412439,  9.97220799,
        9.562174  ])}
{'P': tensor([[2.0603e-02, 2.1846e-01, 1.6492e-01,  ..., 3.2343e-06, 7.1564e-03,
         1.3359e-03],
        [9.3251e-04, 2.7646e-02, 2.3875e-02,  ..., 3.4912e-06, 6.0463e-03,
         2.3815e-03],
        [1.4985e-03, 2.6877e-02, 3.7241e-02,  ..., 4.3451e-06, 1.0925e-02,
         4.1237e-03],
        ...,
        [3.4478e-04, 2.0311e-04, 1.7734e-04,  ..., 4.1127e-02, 3.7723e-05,
         2.2393e-06],
        [2.9404e-03, 6.2273e-03, 7.3933e-03,  ..., 4.3616e-06, 9.0867e-02,
         7.7609e-02],
        [2.6006e-04, 3.5670e-04, 3.9357e-04,  ..., 4.7557e-06, 7.6045e-04,
         1.7771e-01]]), 'X': tensor([[ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0216, 0.0500,  ..., 0.7570, 0.3427, 0.5591],
        [0.0216, 0.0000, 0.0431,  ..., 0.7436, 0.3395, 0.5475],
        [0.0500, 0.0431, 0.0000,  ..., 0.7755, 0.2964, 0.5092],
        ...,
        [0.7570, 0.7436, 0.7755,  ..., 0.0000, 1.0138, 1.0674],
        [0.3427, 0.3395, 0.2964,  ..., 1.0138, 0.0000, 0.2785],
        [0.5591, 0.5475, 0.5092,  ..., 1.0674, 0.2785, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.84796202,  7.02253235,  7.02496223,  7.10378919,  7.20826219,
        7.23276582,  4.75303127, 10.88618114, 13.26209545, 10.23946247,
       13.32645552,  6.38399789,  5.05981914,  9.86089579, 11.66453059,
        5.14607014,  9.77777346, 12.59840193,  5.70591986,  4.91296205,
       12.07818871,  8.1303945 ,  8.38252937,  4.91652556,  5.5205189 ,
       11.18839169,  6.16512856, 10.27572673,  5.05444288, 11.8159254 ,
        9.15253344, 11.35345345, 10.27572673,  7.60463349,  8.62162372,
        9.09661212,  8.67502348, 12.30695098,  7.13501264, 10.1371552 ,
        8.62260327,  8.68979052, 10.71162681,  6.3949043 ,  4.75303127,
        7.72291946, 11.00726483, 13.13888841,  5.35774573,  9.99211663,
        7.33350642,  4.91038884,  6.83083863,  6.67218687, 12.72524615,
        6.35029274, 10.26545792, 13.89886469,  7.73151309,  5.68942572,
        8.09704052, 10.78533838, 11.83570132, 11.00812996, 10.90638066,
        9.64157461,  7.33350642, 13.35832725, 13.02818702,  7.98469166,
       11.81423161,  6.67121748, 10.09979747,  9.49511524,  4.76969364,
       12.00995898,  5.80860165,  6.33075682,  5.0713364 , 11.81423161,
       11.35345345,  5.05989547,  7.15990676,  8.87007495,  5.69913817,
       11.97238624, 11.04629471, 13.35927527, 10.26327895,  4.91336994,
       13.09426142, 13.99877413, 11.70016217, 10.61417181,  9.60412735,
        7.06272256,  9.43737523,  8.7273569 , 13.70893492,  9.20340031,
       13.89264822,  5.47702766,  5.92281142, 11.35345345,  7.15990676,
        7.98469166])}
{'P': tensor([[2.3510e-02, 2.4103e-01, 1.3055e-01,  ..., 3.5961e-05, 1.9063e-04,
         5.3562e-06],
        [1.2040e-03, 3.4492e-02, 1.9952e-01,  ..., 7.1950e-05, 1.6785e-04,
         6.6616e-06],
        [2.6877e-03, 6.7005e-03, 9.1394e-02,  ..., 2.0576e-04, 2.9553e-04,
         1.6525e-05],
        ...,
        [3.4108e-05, 2.8512e-05, 2.3896e-05,  ..., 1.2720e-01, 1.8880e-04,
         4.8341e-04],
        [2.4258e-02, 2.0485e-02, 1.3763e-02,  ..., 2.7060e-04, 2.1838e-01,
         8.2842e-05],
        [2.2723e-04, 1.2593e-04, 8.8536e-05,  ..., 6.1625e-07, 3.6028e-04,
         1.6515e-02]]), 'X': tensor([[  5.6523,   3.9443,  -2.9449],
        [  5.0233,   4.9647,  -3.1788],
        [  4.5430,   5.5438,  -3.8669],
        [  5.1861,   4.7361,  -3.7312],
        [  4.5800,   5.5034,  -3.9432],
        [  3.9085,   6.1512,  -2.2560],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.3924, -10.1396,  -4.0863],
        [  4.1537,   5.9340,  -6.0423],
        [  5.0506,   4.9277,  -9.5094],
        [ -4.4520,   7.1370,  -8.7867],
        [  7.0642,  11.6104,  -9.8316],
        [  2.9470,  13.6085,   2.9809],
        [  0.5600, -11.0321,  -0.8660],
        [  1.5383, -11.0274,  -4.9666],
        [  5.8889,  -1.3916,   8.8993],
        [  3.1271, -10.8343,   2.6442],
        [  3.2990,  -4.2415,  -7.4857],
        [ -6.6998,   5.5954,   0.0823],
        [  0.5250, -11.0306,   5.5358],
        [ -6.4464,   5.8272,   7.5038],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.1907,  -3.3180,  -5.5943],
        [  9.1620,  -7.5573,  -9.9882],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -7.6734,   4.4980,  -5.4248],
        [  4.7544,  -3.1425,  -1.5315],
        [ -7.6650,   4.5092,   0.2520],
        [  2.1504,  -4.6726,   8.7908],
        [ -8.0829,  -5.9940,   5.0564],
        [  1.8447,  -4.7372,   8.1235],
        [ -5.9987,   6.1949,  -6.2114],
        [  8.5716,  10.3596,   6.8818],
        [  5.9655,   3.2172,  -7.5953],
        [  6.3087,   1.8053,  -1.9481],
        [  0.4996,   7.7738,  -4.4711],
        [  2.4149,  -4.6005,  -4.3339],
        [  5.3602,  -2.3808,   7.2098],
        [  5.1667,  -2.6543,   1.2879],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.0891,  -4.9558,  -2.3178],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.0604,   1.7707,   1.1066],
        [  4.2612, -10.5514,  -6.6268],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.0582,  -2.6640,   0.7132],
        [  9.0129,  -7.7072,   9.1400],
        [ -5.3382,   6.6533,   7.8054],
        [ -3.0008, -10.2827,  -8.9480],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.6314,  -1.9302,  -9.6543],
        [  7.0515,  11.6194,   8.4289],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4716,  -0.6673,   7.6068],
        [ 12.5199,  -0.4711,   3.0436],
        [  2.6310, -10.9195,   3.8054],
        [  5.8318,   3.5567,  -4.7692],
        [  2.1824,  -4.6647,   1.1985],
        [  5.7151,  -1.7700,   3.1542],
        [  6.3534,   1.2990,  -7.2942],
        [  2.1379,  -4.6757,   6.4981],
        [ 10.9875,   7.2207,  -8.8549],
        [  6.3119,   1.7797,   9.9186],
        [  0.1931,  -4.7491,   7.4385],
        [ -2.4956, -10.4682,   3.5632],
        [  9.3836,   9.5075,  -5.9278],
        [  8.5093,  -8.1764,  -3.2381],
        [ 11.0373,   7.1299,  -8.5895],
        [  0.8700,  -4.8132,  -7.8042],
        [  5.1787,  -2.6383,  -4.2533],
        [  8.0381,  -8.5690,  -8.9523],
        [  9.0051,   9.9234,   5.4411],
        [ -2.0229, -10.6163,   8.6930],
        [  3.6817,   6.3354,   6.1233],
        [  2.9461,  -4.4077,  -0.4831],
        [ -4.7642,  -9.3860,  -5.1133],
        [  5.2188,  -2.5839,   2.1833],
        [  6.3048,   0.1469,  -9.1691],
        [ -8.3653,   3.4184,   9.3454],
        [  3.7121,   6.3117,   0.6738],
        [  5.9003,  12.3550,   2.1917],
        [  6.3535,   0.6792,  -7.5101],
        [  0.7145,  -4.8071,  -2.3766],
        [  3.9085,   6.1512,  -2.2560],
        [  5.9655,   3.2172,  -7.5953],
        [ -2.5161,   7.7699,  -9.3531],
        [ 11.5456,   6.0861,   8.1101],
        [  4.3523, -10.5232,   8.2138],
        [ -5.1974,   6.7396,   3.3484],
        [  8.9284,  10.0036,  -6.9210],
        [  1.0093,   7.6567,   2.3526],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.3384,  -2.6161,  -4.7630],
        [  5.6928,  12.4710,   3.6001],
        [  6.3542,   0.6934,  -3.3840],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.8889,  -1.3916,   8.8993],
        [ -6.6998,   5.5954,   0.0823],
        [ -4.0409,   7.3169,  -3.5779],
        [  1.3103,  -4.8032,  -0.9300],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.2419,  -0.2313,   4.2440],
        [  3.0719, -10.8449,  -4.1673]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.3755e-02, 6.4589e-02,  ..., 9.9385e-01, 5.0931e-01,
         5.2439e-01],
        [2.3755e-02, 0.0000e+00, 4.3697e-02,  ..., 9.8374e-01, 5.2336e-01,
         5.0152e-01],
        [6.4589e-02, 4.3697e-02, 0.0000e+00,  ..., 9.9268e-01, 5.6518e-01,
         4.7533e-01],
        ...,
        [9.9385e-01, 9.8374e-01, 9.9268e-01,  ..., 3.0339e-04, 8.6971e-01,
         7.5978e-01],
        [5.0931e-01, 5.2336e-01, 5.6518e-01,  ..., 8.6971e-01, 0.0000e+00,
         8.4621e-01],
        [5.2439e-01, 5.0152e-01, 4.7533e-01,  ..., 7.5978e-01, 8.4621e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.89245352,  7.06272256,  7.16747878,  7.02326515,  7.15990676,
        7.28793663, 10.31584102, 11.48434767,  7.24326992,  7.05627461,
        8.41171369, 13.59056613, 13.92390689, 11.04629471, 11.1341798 ,
        6.05113217, 11.27656488,  5.37343173,  8.72896381, 11.04313641,
        8.68979052,  5.24137909,  9.7712321 , 11.87665501, 12.18045978,
        8.89458273,  5.69913817,  8.89301516,  5.14371005, 10.06286074,
        5.08372149,  8.62329466, 13.44593519,  6.77777606,  6.5618968 ,
        7.78980299,  5.19577146,  5.86519525,  5.80860165,  5.24461068,
        9.58072286,  8.16584537, 12.1460931 ,  6.77777606,  9.23178204,
       11.37939695, 12.86648771, 12.3489324 , 11.85890664,  8.53014915,
       10.71162681,  9.97822473,  5.95297459, 13.59170809,  9.24501829,
        9.49511524, 12.52875793, 11.23201619,  6.83083863,  5.14999799,
        5.98285319,  6.48486171,  5.14124612, 13.14777433,  6.55800267,
        4.75303127, 10.76154256, 13.35832725, 11.80092578, 13.1399181 ,
        4.89121805,  5.81199422, 11.74901778, 13.40024224, 10.80728788,
        7.32754882,  5.30158497, 10.52587243,  5.82345867,  6.30647746,
        9.0368458 ,  7.32233733, 13.69162795,  6.38968357,  4.85995002,
        7.28793663,  6.77777606,  8.16714836, 13.05148826, 11.38774149,
        8.51088913, 13.40850328,  7.72291946,  9.69792513, 10.1979669 ,
        9.69792513, 13.70893492,  6.39187285, 12.18045978,  6.05113217,
        8.72896381,  8.35856487,  4.978704  ,  9.44486609,  6.24614905,
       11.27160035])}
{'P': tensor([[2.5383e-02, 2.1987e-01, 1.8755e-01,  ..., 6.0128e-06, 8.1066e-04,
         3.2039e-06],
        [9.4576e-04, 3.6269e-02, 3.3648e-01,  ..., 4.5601e-06, 2.3603e-04,
         3.7954e-06],
        [1.2579e-03, 4.1879e-03, 5.4417e-02,  ..., 5.9403e-06, 2.7178e-04,
         5.1195e-06],
        ...,
        [5.8059e-04, 3.4804e-04, 2.8852e-04,  ..., 3.3000e-02, 2.0667e-03,
         8.4604e-05],
        [7.4489e-03, 2.0765e-02, 2.2520e-02,  ..., 1.9932e-06, 4.7278e-02,
         1.2608e-06],
        [4.9483e-05, 3.7508e-05, 3.2406e-05,  ..., 2.8603e-03, 3.0420e-05,
         8.4999e-02]]), 'X': tensor([[  5.6248,   3.9984,   8.4933],
        [  4.6231,   5.4556,   8.4973],
        [  4.4039,   5.6903,   8.7366],
        [  4.6357,   5.4415,   7.9067],
        [  4.5503,   5.5359,   7.4735],
        [  3.3069,   6.6089,   8.7377],
        [  4.8674,  12.8875,   6.7622],
        [ -6.3306,   5.9272,   5.6935],
        [  2.8994,   6.8675,  -9.5987],
        [  5.8275,  -1.5334,   2.0124],
        [ -5.3382,   6.6533,   7.8054],
        [  6.3305,   1.6119,   1.1538],
        [  2.3589,  13.7668,  -3.6501],
        [ -7.0120,   5.2825,   7.7762],
        [  6.2022,   2.4092,   3.0349],
        [ 11.5568,   6.0602,   9.3244],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.6038,   0.7665,  -3.4439],
        [ 10.0567,   8.6697,   4.6329],
        [ -5.3179,  -9.0101,   9.4472],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.3583,   1.1729,   8.7835],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.0876,   6.7527,  -5.3540],
        [  1.1644, -11.0394,   7.8511],
        [  2.2659,  13.7894,   6.5436],
        [  3.9022,  13.2906,   5.5788],
        [  0.3724,   7.7974,  -6.8722],
        [ -9.1624,  -3.4286,   2.0647],
        [ 12.4597,  -0.9136,  -4.8763],
        [  1.1274,  14.0116,   8.1753],
        [ -3.4100,   7.5436,  -7.6980],
        [  2.9260,  -4.4162,   8.3808],
        [  2.3928,   7.1404,  -1.7035],
        [  6.3551,   0.7149,   1.3344],
        [  6.2036,   2.4030,  -1.2500],
        [  1.6232,  -4.7717,  -2.5655],
        [ -1.2454,   7.9127,  -0.5715],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 11.6165,  -3.8715,   6.4594],
        [ -7.6650,   4.5092,   0.2520],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -8.9178,   2.1936,   7.5388],
        [ -5.5525,   6.5146,  -4.4984],
        [  2.3928,   7.1404,  -1.7035],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.6319,  11.9066,   3.5165],
        [  6.2883,   0.0325,   5.2692],
        [ -9.4203,  -2.0340,   2.5856],
        [ 12.2311,  -2.0445,   1.3545],
        [  0.1931,  -4.7491,   7.4385],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.4773,  -1.0024,   5.2599],
        [  0.6933,   7.7336,   2.1179],
        [ -5.0550,   6.8231,  -9.2502],
        [  5.5294,  -2.1106,   9.0626],
        [ 11.4495,   6.3026,   6.8257],
        [  3.5284, -10.7484,   5.3347],
        [  3.1480,  -4.3165,  -7.8002],
        [ -8.5607,  -5.0938,  -5.4531],
        [  3.9591, -10.6391,  -6.2557],
        [ -5.3179,  -9.0101,   9.4472],
        [  4.0362,  13.2397,  -7.0030],
        [  2.1625,  -4.6697,   7.0658],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.0330,  -6.0765,   6.7945],
        [  1.2273,  -4.8081,   9.8129],
        [  0.4996,   7.7738,  -4.4711],
        [  5.2608,  -2.5256,  -9.1446],
        [  0.6396,  14.0772,  -6.3786],
        [  6.2804,   2.0019,   0.8607],
        [  9.9756,  -6.6330,  -3.3429],
        [ -9.0604,   1.7707,   1.1066],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.3200,   7.1756,   1.1938],
        [ -7.5401,   4.6718,   9.0033],
        [  2.6614,  -4.5192,  -9.3641],
        [  1.1006,   7.6318,  -2.4877],
        [  2.8474,   6.8979,   6.6457],
        [  1.4489,   7.5256,   4.4529],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.0042,  -2.8350,   2.6864],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.7558,  -4.4844,   0.0636],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.3943,   6.5483,   1.0790],
        [  6.3266,   0.3292,   9.3502],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.3516,   0.6425,   0.3425],
        [  4.9531,   5.0576,   5.6794],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.3464,   0.5708,  -1.6397],
        [ 10.9916,   7.2133,  -0.9179],
        [  1.1644, -11.0394,   7.8511],
        [ -9.2974,   0.8371,   2.4852],
        [  6.2365,  -0.2578,  -6.0212],
        [ 12.5832,   1.7543,  -7.6134],
        [ 12.1379,   4.4125,  -3.8833],
        [ -1.9714,   7.8563,  -1.8084],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.5456,   6.0861,   8.1101],
        [  2.3889,  -4.6082,  -8.7291]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0509, 0.0656,  ..., 0.6514, 0.3743, 0.9362],
        [0.0509, 0.0000, 0.0157,  ..., 0.6488, 0.4160, 0.9402],
        [0.0656, 0.0157, 0.0000,  ..., 0.6550, 0.4249, 0.9480],
        ...,
        [0.6514, 0.6488, 0.6550,  ..., 0.0000, 0.9449, 0.3053],
        [0.3743, 0.4160, 0.4249,  ..., 0.9449, 0.0000, 1.1879],
        [0.9362, 0.9402, 0.9480,  ..., 0.3053, 1.1879, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.90115901,  7.15100303,  7.19534146,  7.14837514,  7.16598273,
        7.39004226, 13.77604293,  8.67227386,  7.4544866 ,  6.02588517,
        8.53014915,  6.53250937, 13.96746816,  8.77913428,  6.65368551,
       13.0493368 ,  4.78544875, 12.62711252, 13.2778378 , 10.46238054,
        5.35774573,  5.34250225,  6.46560181,  5.5366698 ,  7.42512621,
       11.10066126, 13.97430191, 13.85158417,  7.80625981,  9.78285424,
       12.49317563, 14.05687815,  8.27853076,  5.29755487,  7.53063885,
        6.39520397,  6.652742  ,  5.04028538,  8.01009083,  7.13501264,
        9.62958973, 12.24467041,  8.89301516, 12.3828957 ,  9.18358562,
        8.55982105,  7.53063885, 10.74655346,  9.5642993 , 13.62896593,
        6.28835101,  9.63742669, 12.40074901,  4.75303127,  4.89121805,
        9.53015376,  7.76457095,  8.49161154,  5.91854397, 13.06957425,
       11.31276109,  5.34250225,  9.96152788, 11.35182589, 10.46238054,
       13.84126197,  5.14607014,  8.1137452 , 10.07238463,  4.96231532,
        7.78980299,  5.83560009, 14.09176259,  6.59176018, 11.97957425,
        9.23178204,  8.56889738,  7.54127425,  8.87007495,  5.24461068,
        7.71075226,  7.46248947,  7.66377554,  9.58072286, 11.52733581,
       12.3344536 , 13.01728313,  5.26342203,  5.35774573,  7.37576154,
        6.33517349,  8.41171369,  6.38399789,  7.07902036, 12.59969936,
        9.36378438, 13.1471313 , 11.10066126,  9.3349858 ,  6.24187044,
       12.70489563, 12.91505111,  8.09984231, 12.04924159, 13.05148826,
        5.19065439])}
{'P': tensor([[2.9230e-02, 3.7381e-01, 2.1873e-01,  ..., 2.7616e-05, 9.9548e-04,
         3.6370e-05],
        [1.6766e-03, 3.6316e-02, 3.4822e-01,  ..., 3.3370e-05, 1.5195e-03,
         4.0866e-05],
        [1.1081e-03, 1.4583e-03, 3.6887e-02,  ..., 3.4027e-05, 2.2924e-03,
         3.6489e-05],
        ...,
        [5.7545e-04, 5.2881e-04, 4.7020e-04,  ..., 8.2022e-02, 3.6917e-04,
         1.2235e-01],
        [6.2787e-05, 7.5335e-05, 1.1080e-04,  ..., 2.0581e-04, 6.2914e-02,
         8.8729e-05],
        [1.3751e-03, 1.3514e-03, 1.3351e-03,  ..., 2.4048e-03, 1.4093e-03,
         1.0073e-01]]), 'X': tensor([[  5.3858,   4.4260,  -9.6769],
        [  5.0506,   4.9277,  -9.5094],
        [  4.3554,   5.7393,  -9.0766],
        [  2.8994,   6.8675,  -9.5987],
        [  2.3367,   7.1676,  -9.5263],
        [  2.7121,   6.9743,  -8.4202],
        [ 10.4758,  -5.9539,   1.3667],
        [ -5.9940,   6.1985,   5.9566],
        [ 11.2939,   6.6325,  -5.8311],
        [  2.0729,   7.2878,   7.0880],
        [  9.3560,   9.5391,   9.4943],
        [ -1.1885, -10.8211,   0.0865],
        [ 12.2328,  -2.0376,  -1.1234],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.6154,  -4.0640,  -7.9698],
        [ -3.4100,   7.5436,  -7.6980],
        [ -3.4488, -10.0934,   2.5195],
        [ 11.2209,  -4.7092,   2.6117],
        [  4.3377,   5.7570,   1.3712],
        [  2.7295,  13.6703,  -8.9328],
        [ 11.5568,   6.0602,   9.3244],
        [  9.2811,  -7.4336,   5.1771],
        [  6.3460,   1.4252,   1.2291],
        [  4.4039,   5.6903,   8.7366],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.4541,  -0.3327,   8.8246],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.6634,   5.8050,   9.1590],
        [  2.0449,  13.8402,  -7.1642],
        [ -6.7992,   5.4991,   9.9186],
        [ 11.6634,   5.8050,   9.1590],
        [ -8.8934,   2.2597,   3.5831],
        [  4.3377,   5.7570,   1.3712],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.3087,   1.8053,  -1.9481],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.1730,  -0.5375,  -8.8861],
        [  7.8425,  -8.7202,   6.4530],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.2247,  -0.3150,  -9.3657],
        [ -5.3328,  -8.9992,   7.3440],
        [  0.6757, -11.0361,  -1.3047],
        [  3.2159, -10.8166,  -1.0406],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.6954,  -4.7616,   9.8198],
        [  6.0305,   3.0296,   0.1375],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.8447,  -4.7372,   8.1235],
        [ -0.4341, -10.9473,   7.4509],
        [ -5.6792,  -8.7352,   4.0318],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.5653,  13.9377,   4.1773],
        [  6.7792,  11.8084,   1.6251],
        [  9.5018,  -7.1945,  -1.1892],
        [  8.5198,  10.4091,  -8.6486],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.7792,  11.8084,   1.6251],
        [  2.6613,   7.0021,   4.5783],
        [  1.4150,  13.9647,  -4.0915],
        [  6.1737,   2.5305,   3.9049],
        [ -6.1899,  -8.3011,  -7.9269],
        [  1.6408,   7.4591,  -4.2294],
        [  5.4242, -10.1261,   3.9595],
        [  8.5093,  -8.1764,  -3.2381],
        [ -7.4733,   4.7555,   6.2490],
        [  6.1752,   2.5242,   3.5874],
        [ -2.4956, -10.4682,   3.5632],
        [ -6.4464,   5.8272,   7.5038],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.6817,   6.3354,   6.1233],
        [ -4.1296,   7.2803,  -2.7194],
        [  2.3376,  -4.6231,   9.8436],
        [  5.8617,   3.4853,  -9.8080],
        [ -5.5525,   6.5146,  -4.4984],
        [ -4.7334,   6.9982,  -0.4392],
        [  1.5383, -11.0274,  -4.9666],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.9193,  -1.3176,   5.1739],
        [ -3.3116, -10.1539,   7.5919],
        [  0.3038,   7.8092,   0.4164],
        [ -0.4341, -10.9473,   7.4509],
        [ 11.7794,   5.5086,  -9.6929],
        [  2.3376,  -4.6231,   9.8436],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.1140,  -3.7220,  -0.1273],
        [ 12.5789,   0.1722,   1.6470],
        [  1.8719,   7.3713,  -3.7465],
        [  6.0345,   3.0175,  -1.4424],
        [  5.0348,  -2.8229,   3.5436],
        [  4.7739,  -3.1218,  -4.1188],
        [  3.3839,  -4.1966,   8.3574],
        [ -3.2736, -10.1703,   3.8712],
        [  0.1532,  -4.7422,  -7.0145],
        [ -1.6086,   7.8928,   4.3385],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.1886,  -2.6250,   6.4677],
        [ 11.9122,   5.1397,  -1.5498],
        [  2.4697,  -4.5836,   1.4593],
        [ -0.8569,   7.9156,  -6.5557],
        [ -2.5056,   7.7720,   5.5878],
        [  1.9503,  -4.7172,  -6.2313],
        [  1.8721,  -4.7322,   0.5524],
        [ -2.0445,   7.8469,  -0.5857],
        [  5.1667,  -2.6543,   1.2879]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0184, 0.0481,  ..., 0.4645, 0.4797, 0.5257],
        [0.0184, 0.0000, 0.0297,  ..., 0.4634, 0.4702, 0.5274],
        [0.0481, 0.0297, 0.0000,  ..., 0.4605, 0.4537, 0.5288],
        ...,
        [0.4645, 0.4634, 0.4605,  ..., 0.0000, 0.2245, 0.0975],
        [0.4797, 0.4702, 0.4537,  ..., 0.2245, 0.0000, 0.3159],
        [0.5257, 0.5274, 0.5288,  ..., 0.0975, 0.3159, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.97106912,  7.05627461,  7.20482224,  7.4544866 ,  7.53884104,
        7.48311002, 12.0495521 ,  8.62260327, 13.09737053,  7.57685908,
       13.3614598 , 10.88618114, 12.4013153 ,  5.18047638,  5.43943791,
        8.27853076, 10.66631784, 12.16901615,  7.20826219, 13.94008948,
       13.0493368 , 11.89105259,  6.50409372,  7.19534146, 10.76154256,
        9.45995955, 10.61472636, 13.02818702, 13.99048094,  8.74470218,
       13.02818702,  9.17595899,  7.20826219, 11.83509372,  9.97964013,
        6.5618968 , 12.889695  ,  6.19633139, 11.72802491,  6.07992623,
        6.23262289, 10.4606265 , 11.05672136, 11.2845608 ,  9.64157461,
        9.75401449,  5.05444288,  6.74872981,  9.55593374,  5.08372149,
       10.95593729, 10.41909211, 12.32408516, 14.02532758, 13.61600536,
       11.9182956 , 13.45124917,  5.94541803, 13.61600536,  7.49076899,
       14.03618649,  6.67218687, 10.35485331,  7.6374603 , 11.48734978,
       11.80092578,  8.8580671 ,  6.67121748, 10.76154256,  8.68979052,
        5.35774573,  7.32754882,  8.36995953,  5.1805345 ,  6.81961996,
        8.55982105,  8.4486771 , 11.1341798 , 10.52357479,  6.06416957,
       10.68031351,  7.81509631, 10.95593729, 13.00380624,  5.1805345 ,
       13.25017211,  5.54776934, 12.58006038,  7.60529892,  6.74686071,
        5.77219077,  5.70405525,  5.39098669, 10.68416401,  4.74468406,
        8.0550286 , 10.1979669 ,  5.81481044, 12.97370371,  5.20659892,
        7.96181288,  8.16584537,  5.10444097,  5.08910491,  8.10886021,
        5.80860165])}
{'P': tensor([[2.2537e-02, 2.2993e-01, 8.9426e-02,  ..., 6.7557e-06, 2.1077e-03,
         2.5877e-05],
        [1.2856e-03, 3.7492e-02, 2.3646e-01,  ..., 7.9628e-06, 1.2774e-03,
         6.4224e-05],
        [3.4548e-04, 5.5387e-04, 2.9131e-02,  ..., 3.5560e-06, 1.2086e-04,
         9.6576e-05],
        ...,
        [4.4046e-04, 2.5284e-04, 7.0151e-05,  ..., 7.3262e-03, 2.0982e-04,
         2.5142e-07],
        [2.8913e-03, 8.1836e-03, 3.4247e-02,  ..., 1.0088e-06, 6.9765e-02,
         8.9357e-06],
        [2.8428e-05, 2.4475e-05, 1.7737e-05,  ..., 1.9138e-04, 1.3694e-06,
         9.6551e-02]]), 'X': tensor([[  5.3118,   4.5452,   4.2114],
        [  4.5767,   5.5071,   3.9069],
        [  2.6613,   7.0021,   4.5783],
        [  4.6992,   5.3690,   3.0935],
        [  2.5382,   7.0672,   4.5097],
        [  2.9410,   6.8428,   5.4458],
        [  4.2339,  -3.6268,   6.3125],
        [  6.9433,  -9.3362,  -0.3421],
        [ -9.4378,  -1.8606,  -2.5668],
        [  9.9756,  -6.6330,  -3.3429],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.2576,   2.1359,  -3.3398],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -1.3178, -10.7939,   0.4236],
        [  1.7500,   7.4187,   4.4237],
        [ -3.2486,   7.5925,  -1.3719],
        [ -9.1795,  -3.3623,   8.5259],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.0870,  -4.6877,   9.6823],
        [  2.1440, -10.9811,   8.5281],
        [  2.2849, -10.9655,  -7.6542],
        [  9.8082,  -6.8396,   4.9615],
        [ -6.7992,   5.4991,   9.9186],
        [ -9.4753,  -0.8012,  -5.2050],
        [  1.1274,  14.0116,   8.1753],
        [  5.4429,  -2.2529,   9.7306],
        [  1.3670,  -4.7990,   3.2603],
        [ -6.9838,  -7.4965,   0.9796],
        [  4.7156,  -3.1831,  -8.0224],
        [  9.4554,   9.4245,   9.8800],
        [  3.4140,  -4.1803,  -8.8224],
        [ -3.3999, -10.1152,  -1.2088],
        [  9.8082,  -6.8396,   4.9615],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -1.6220,   7.8917,   0.0315],
        [  4.9531,   5.0576,   5.6794],
        [  0.9809,   7.6642,  -1.4217],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.3419,   0.5972,  -1.2841],
        [ 10.2470,   8.4054,  -0.1177],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -5.1377,   6.7751,  -0.4438],
        [ -9.1813,  -3.3550,   1.8827],
        [ -4.2271,   7.2387,  -1.8316],
        [  4.1297,  -3.7098,  -3.9838],
        [  2.4697,  -4.5836,   1.4593],
        [  6.3048,   0.1469,  -9.1691],
        [  4.3523, -10.5232,   8.2138],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.2508,   1.0572,   2.9978],
        [  6.3048,   0.1469,  -9.1691],
        [  2.0870,  -4.6877,   9.6823],
        [  9.9217,  -6.7006,  -5.6728],
        [ -7.3333,   4.9246,  -7.9212],
        [ 12.3759,  -1.3943,   6.4659],
        [  4.5767,   5.5071,   3.9069],
        [  4.0457,  -3.7737,  -0.4704],
        [ 11.7349,  -3.5838,   2.1644],
        [ -8.8058,   2.4850,  -1.7425],
        [  4.3554,   5.7393,  -9.0766],
        [  8.1043,  -8.5164,  -7.9290],
        [ 10.2470,   8.4054,  -0.1177],
        [  4.1537,   5.9340,  -6.0423],
        [  5.9548,  -1.2273,  -2.6520],
        [  7.1441,  11.5529,   2.3232],
        [  5.8275,  -1.5334,   2.0124],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.4117,   0.1160,   8.2574],
        [  4.5800,   5.5034,  -3.9432],
        [  4.9452, -10.3187,  -7.8621],
        [  1.9204,   7.3518,   8.4781],
        [ 12.1751,  -2.2592,  -5.7751],
        [  6.3205,   1.7068,  -3.8619],
        [  5.3924, -10.1396,  -4.0863],
        [  3.7992,  -3.9474,  -8.9019],
        [  5.6202,  -1.9508,   0.7471],
        [ -7.5401,   4.6718,   9.0033],
        [  2.2644,  -4.6434,   9.7508],
        [ 10.4758,  -5.9539,   1.3667],
        [  2.0937,  13.8293,  -2.3533],
        [ -9.1749,  -3.3803,   6.4629],
        [  4.3377,   5.7570,   1.3712],
        [  5.8836,   3.4315,  -4.0327],
        [  5.8902,  -1.3887,   4.0388],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.3122,  -9.6994,  -6.3282],
        [  7.1697,  11.5343,  -0.7276],
        [  2.1824,  -4.6647,   1.1985],
        [  4.2275,  -3.6320,  -0.2714],
        [ -2.8162,   7.7059,  -9.6885],
        [  9.3624,   9.5318,   0.2779],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.6249, -10.7255,   5.6605],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.6703,   6.3443,  -5.4401],
        [ 11.5966,   5.9666,   6.0083],
        [  6.3266,   0.3292,   9.3502],
        [  5.6260,  -1.9401,   3.0259],
        [  6.1579,  -0.5961,  -6.5595],
        [  9.1409,  -7.5789,   1.5814],
        [  9.5203,   9.3483,   5.8781],
        [ -9.3411,  -2.6004,  -2.7062]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0380, 0.1102,  ..., 0.2766, 0.4623, 1.1110],
        [0.0380, 0.0000, 0.0843,  ..., 0.2437, 0.4983, 1.0736],
        [0.1102, 0.0843, 0.0000,  ..., 0.2550, 0.5284, 1.0430],
        ...,
        [0.2766, 0.2437, 0.2550,  ..., 0.0000, 0.7368, 0.8515],
        [0.4623, 0.4983, 0.5284,  ..., 0.7368, 0.0000, 1.5694],
        [1.1110, 1.0736, 1.0430,  ..., 0.8515, 1.5694, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 6.99097013,  7.16058587,  7.49076899,  7.13501264,  7.50917551,
        7.44805901,  5.5748624 , 11.63502679,  9.61942445, 11.97957425,
        5.34250225,  6.61211224, 10.5575601 , 10.87409005,  7.62232307,
        8.25828409,  9.77588801, 12.22276278,  5.13125513, 11.18839169,
       11.20100795, 11.95743558,  8.74470218,  9.50913849, 14.05687815,
        5.89074104,  4.98989906, 10.24557047,  5.68942572, 13.35013374,
        5.39723241, 10.67132524, 11.95743558, 12.52372629,  8.05668609,
        7.07902036,  7.72668467,  8.53014915,  9.43737523,  9.60080639,
        9.36093562, 13.25336001, 10.52587243,  8.50278896,  9.7751189 ,
        8.38252937,  5.5513072 ,  5.20659892,  6.30647746, 11.38774149,
        9.09502385,  9.3109884 ,  6.30647746,  5.13125513, 11.97238624,
        8.83341608, 12.45418236,  7.16058587,  5.53255882, 12.26997165,
        9.14973319,  7.20482224, 11.75618513, 13.25336001,  7.24326992,
        6.07992623, 13.58333857,  6.02588517, 12.72524615,  9.41244865,
        7.15990676, 11.4424816 ,  7.5984809 , 12.3828957 ,  6.54693987,
       11.48434767,  5.47866221,  5.94909709,  8.87007495,  5.16611782,
       12.0495521 , 13.98691426,  9.77777346,  7.20826219,  6.81117802,
        6.05165248, 10.60226614, 12.12448912, 11.57249984, 13.58101794,
        5.14999799,  5.57341623,  8.20436764, 13.3607379 ,  9.71392179,
       12.19805394, 11.32149259,  8.05242591,  7.32950385, 13.04156075,
        6.33517349,  5.95111804,  6.18668463, 11.87412439, 13.3426541 ,
        9.69629   ])}
{'P': tensor([[1.8813e-02, 1.6243e-01, 1.4952e-01,  ..., 1.7316e-05, 1.4797e-03,
         1.5305e-06],
        [5.6625e-04, 2.2101e-02, 4.2396e-02,  ..., 3.7202e-05, 2.1280e-03,
         1.5350e-06],
        [8.8265e-04, 1.1289e-02, 3.7523e-02,  ..., 5.6145e-05, 5.0045e-03,
         2.2961e-06],
        ...,
        [4.4287e-05, 4.1579e-05, 3.6336e-05,  ..., 3.2094e-01, 8.6671e-06,
         3.2305e-03],
        [1.0516e-04, 1.7090e-04, 2.0492e-04,  ..., 3.2177e-05, 6.3003e-02,
         8.5273e-07],
        [3.6502e-04, 1.9427e-04, 1.5329e-04,  ..., 9.5713e-06, 3.2328e-06,
         1.4957e-01]]), 'X': tensor([[  5.2199,   4.6861,  -1.9304],
        [  4.0591,   6.0202,  -1.6339],
        [  3.9085,   6.1512,  -2.2560],
        [  3.6469,   6.3624,  -1.8414],
        [  3.5939,   6.4029,  -1.8357],
        [  4.1348,   5.9514,  -0.8856],
        [ -2.1382,   7.8339,   7.3524],
        [ -5.1369,   6.7755,   8.3836],
        [  3.2701,   6.6338,   9.9209],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.7847,  -9.3729,  -3.7741],
        [  3.2701,   6.6338,   9.9209],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.6319,  11.9066,   3.5165],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 10.9931,   7.2107,   7.7933],
        [ -0.9715, -10.8630,   0.3555],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.6093,  -1.9705,   6.4807],
        [  8.3791,  10.5409,   5.5590],
        [ 11.7795,  -3.4697,   3.5624],
        [  4.0591,   6.0202,  -1.6339],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.4297,  -5.3637,  -9.3962],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3550,   0.7129,   0.4917],
        [  6.1989,  -9.7591,   0.8289],
        [ -8.9282,   2.1648,   8.4894],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 11.7349,  -3.5838,   2.1644],
        [ -3.6381,   7.4683,   7.3544],
        [ -3.5173,   7.5091,  -8.9585],
        [  1.3670,  -4.7990,   3.2603],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.6422,   7.4586,  -6.8056],
        [  1.2273,  -4.8081,   9.8129],
        [  6.3569,   0.7591,   6.5011],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.6928,  12.4710,   3.6001],
        [  6.0345,   3.0175,  -1.4424],
        [ -7.7842,   4.3466,  -1.7548],
        [  5.9912,   3.1453,  -3.9989],
        [ 10.5678,   7.9261,  -0.6100],
        [  3.3474,  -4.2162,   5.2455],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.9693,  -9.3200,  -8.3822],
        [  7.2340,  -9.1505,  -5.8321],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.9445,   7.6736,   0.5095],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -8.6050,   2.9451,  -0.7012],
        [  6.1066,  -0.7799,  -4.1746],
        [ -6.7992,   5.4991,   9.9186],
        [  9.2979,   9.6048,   7.3138],
        [  4.9452, -10.3187,  -7.8621],
        [  1.7000,  -4.7609,   2.7935],
        [ -7.0120,   5.2825,   7.7762],
        [ -7.4733,   4.7555,   6.2490],
        [  0.9478,  -4.8144,   0.6371],
        [  8.2422,  -8.4041,  -1.4962],
        [ -1.5875,   7.8944,  -9.4286],
        [ -2.4956, -10.4682,   3.5632],
        [ -7.7842,   4.3466,  -1.7548],
        [  5.1196,  -2.7160,   4.2705],
        [  5.8889,  -1.3916,   8.8993],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.1973,  -0.4372,   6.5720],
        [  0.1287, -11.0065,   6.5757],
        [  6.0054,  -1.0902,   8.3079],
        [ 12.4691,   2.8525,  -5.5110],
        [ -8.1666,   3.7653,   3.6075],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3550,   0.7129,   0.4917],
        [ -9.2345,  -3.1345,   3.9715],
        [ -6.2200,  -8.2737,   5.5079],
        [  5.2608,  12.6977,   4.9166],
        [  3.9515,  -3.8425,   6.1729],
        [  1.7301,   7.4262,   5.0522],
        [  2.4697,  -4.5836,   1.4593],
        [ -7.7842,   4.3466,  -1.7548],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -8.8540,  -4.3971,  -5.2144],
        [  5.9003,  12.3550,   2.1917],
        [ -5.0132,  -9.2234,  -8.3531],
        [ 12.4691,   2.8525,  -5.5110],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.3505,   0.6235,  -8.5849],
        [  6.3550,   0.7129,   0.4917],
        [ 10.0619,   8.6627,  -9.2884],
        [ 11.6165,  -3.8715,   6.4594],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.3928,   7.1404,  -1.7035],
        [  3.7992,  -3.9474,  -8.9019],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 11.6165,  -3.8715,   6.4594],
        [ 12.0582,  -2.6640,   0.7132],
        [ -9.4541,  -0.3327,   8.8246],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.9770, -10.6341,   8.0783]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.9499e-02, 5.2679e-02,  ..., 1.0834e+00, 4.7336e-01,
         9.1009e-01],
        [3.9499e-02, 0.0000e+00, 3.9060e-02,  ..., 1.0467e+00, 4.6135e-01,
         8.7974e-01],
        [5.2679e-02, 3.9060e-02, 0.0000e+00,  ..., 1.0655e+00, 4.2516e-01,
         9.0842e-01],
        ...,
        [1.0834e+00, 1.0467e+00, 1.0655e+00,  ..., 0.0000e+00, 1.1919e+00,
         3.5166e-01],
        [4.7336e-01, 4.6135e-01, 4.2516e-01,  ..., 1.1919e+00, 2.1453e-04,
         1.1545e+00],
        [9.1009e-01, 8.7974e-01, 9.0842e-01,  ..., 3.5166e-01, 1.1545e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.01475376,  7.26074925,  7.28793663,  7.33350642,  7.34250948,
        7.24677664,  8.12043983,  8.50268322,  7.39599986,  8.29022197,
       10.52357479,  7.39599986,  7.9337647 , 13.62896593,  5.84450127,
        6.00663314,  9.58072286, 13.14690137, 10.90638066,  4.90505566,
        5.94535529, 13.46553623, 12.27991991,  7.26074925,  9.61942445,
       12.52875793,  9.99145096, 13.36092624,  6.3949043 , 11.56146873,
        9.18690018,  5.16611782, 12.49317563, 12.26997165,  8.30729352,
        8.29203356,  4.98989906, 12.30858677,  9.85384623,  7.63726134,
        4.96231532,  6.4020404 , 10.86693708,  9.58072286, 13.70893492,
        6.74686071,  8.91551231,  6.76661595, 13.20988996,  5.38342077,
        5.53255882, 11.63765273, 11.66453059,  8.83341608,  7.73151309,
        9.51074763,  9.09502385,  6.1561646 ,  8.74470218, 13.36800073,
       11.4424816 ,  5.05534609,  8.77913428,  8.8580671 ,  4.90676242,
       11.77124373,  8.05242591, 10.76154256,  8.91551231,  5.79543038,
        6.05113217,  9.50913849,  6.21275016, 11.00726483,  6.10360073,
       12.79126785,  8.99276564,  6.12886024,  6.3949043 ,  9.75200848,
       10.35093485, 13.744387  ,  5.51177501,  7.62509582,  5.20659892,
        8.91551231, 12.27835996,  9.88571713, 13.69162795, 10.4977132 ,
       12.79126785,  9.93553424, 12.79126785,  6.38105015,  6.3949043 ,
       13.27717982, 12.24467041, 12.59969936,  7.53063885,  5.47866221,
        9.5014146 , 12.24467041, 12.3489324 ,  9.45995955,  8.05242591,
       11.35345345])}
{'P': tensor([[1.9109e-02, 1.7719e-01, 1.4300e-01,  ..., 4.6990e-03, 5.6151e-04,
         1.9884e-04],
        [8.6031e-04, 2.5844e-02, 1.6657e-01,  ..., 7.4289e-03, 3.7072e-04,
         1.5379e-04],
        [1.1058e-03, 3.2169e-03, 5.1822e-02,  ..., 1.1682e-02, 4.9765e-04,
         2.8648e-04],
        ...,
        [7.6713e-05, 1.8677e-04, 3.4589e-04,  ..., 5.0378e-02, 1.9774e-05,
         2.4271e-05],
        [1.1175e-01, 7.7571e-02, 8.4814e-02,  ..., 4.1238e-03, 5.5550e-02,
         1.7258e-03],
        [2.8423e-05, 4.9558e-05, 9.5834e-05,  ..., 3.4695e-04, 1.2369e-05,
         1.5649e-02]]), 'X': tensor([[  5.1890,   4.7318,   0.9515],
        [  4.3377,   5.7570,   1.3712],
        [  3.7121,   6.3117,   0.6738],
        [  3.3943,   6.5483,   1.0790],
        [  4.8432,   5.1968,   1.5006],
        [  4.2096,   5.8815,   1.9230],
        [  3.6703,   6.3443,  -5.4401],
        [  0.1252, -11.0062,   7.2564],
        [  3.4604,  -4.1545,   1.8091],
        [ 11.5568,   6.0602,   9.3244],
        [  5.8979,   3.3958,  -3.8713],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.4573,   3.2448,   5.4370],
        [  3.5965,  -4.0754,  -3.5035],
        [ 12.6000,   1.3839,  -3.2992],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.3569,   0.7591,   6.5011],
        [ -9.4351,  -1.8891,   3.5642],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.3515,   0.6418,  -6.5823],
        [ 11.5456,   6.0861,   8.1101],
        [  3.3943,   6.5483,   1.0790],
        [  4.2096,   5.8815,   1.9230],
        [ -8.7544,   2.6096,   3.3146],
        [  9.2811,  -7.4336,   5.1771],
        [  5.8979,   3.3958,  -3.8713],
        [  6.1222,  -0.7261,   8.8204],
        [ -0.6323,   7.9085,  -5.9544],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 10.1798,   8.5002,  -5.8893],
        [ -1.0407,   7.9166,  -6.6021],
        [  0.1047, -11.0046,   0.5188],
        [  1.0000,   7.6591,  -4.5011],
        [  6.1372,   2.6726,  -6.7885],
        [ 12.5789,   0.1722,   1.6470],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.3860,  -2.3418,   1.0826],
        [ 11.5198,  -4.0923,   5.5088],
        [  1.1350,  14.0104,   0.6321],
        [  6.1752,   2.5242,   3.5874],
        [  1.1274,  14.0116,   8.1753],
        [  1.9204,   7.3518,   8.4781],
        [  0.1047, -11.0046,   0.5188],
        [  1.7218,  -4.7576,  -3.3617],
        [ -9.4117,   0.1160,   8.2574],
        [  5.8930,  12.3592,  -8.9201],
        [ -9.3384,  -2.6161,  -4.7630],
        [  4.8653,   5.1694,  -4.0823],
        [ 11.3800,  -4.3919,   1.4066],
        [ 10.6283,  -5.7253,   7.4878],
        [ -9.3464,   0.5708,  -1.6397],
        [ -8.9033,   2.2329,  -8.8912],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.9659,  -4.8144,  -0.9594],
        [  2.4356,  -4.5942,  -5.0808],
        [  6.3122,  -9.6994,  -6.3282],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.4426, -10.1181,  -5.1268],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -4.0409,   7.3169,  -3.5779],
        [  5.6202,  -1.9508,   0.7471],
        [ -1.6906,   7.8860,   7.7142],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.3924, -10.1396,  -4.0863],
        [ 11.5456,   6.0861,   8.1101],
        [  1.7500,   7.4187,   4.4237],
        [  3.2809,  13.5061,  -4.0199],
        [  6.2902,   0.0451,   3.7116],
        [  3.4604,  -4.1545,   1.8091],
        [  6.1579,  -0.5961,  -6.5595],
        [  0.6757, -11.0361,  -1.3047],
        [ -3.1204, -10.2345,   9.2288],
        [  6.3551,   0.7149,   1.3344],
        [  2.2182,  13.8007,   5.1889],
        [ 11.7349,  -3.5838,   2.1644],
        [ -5.0550,   6.8231,  -9.2502],
        [  1.3103,  -4.8032,  -0.9300],
        [ -5.1398,  -9.1367,   6.2074],
        [  3.3069,   6.6089,   8.7377],
        [  0.7275,  14.0667,  -0.9815],
        [ -5.6423,   6.4537,  -1.2091],
        [ -4.4520,   7.1370,  -8.7867],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.3516,   0.6425,   0.3425],
        [ 12.5919,   0.4035,  -5.8460],
        [ -7.6650,   4.5092,   0.2520],
        [  5.1196,  -2.7160,   4.2705],
        [ -2.7572,   7.7194,   5.9580],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.7726,  -3.4876,  -6.7402],
        [  0.1206,  -4.7363,   8.7103],
        [  6.9284,  11.7061,  -0.3732],
        [ -9.3604,   0.4855,  -3.0158],
        [  1.3705,  13.9723,   7.9266],
        [  6.3211,   1.7017,  -0.2164],
        [ -7.9298,   4.1367,  -3.9773],
        [  5.1519, -10.2387,   9.8191],
        [ -2.0229, -10.6163,   8.6930],
        [  2.9470,  13.6085,   2.9809],
        [  6.2036,   2.4030,  -1.2500],
        [  7.0642,  11.6104,  -9.8316]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0420, 0.0675,  ..., 0.4329, 0.1660, 0.7606],
        [0.0420, 0.0000, 0.0564,  ..., 0.4420, 0.1969, 0.7953],
        [0.0675, 0.0564, 0.0000,  ..., 0.4954, 0.1638, 0.7620],
        ...,
        [0.4329, 0.4420, 0.4954,  ..., 0.0000, 0.5499, 1.0227],
        [0.1660, 0.1969, 0.1638,  ..., 0.5499, 0.0000, 0.5994],
        [0.7606, 0.7953, 0.7620,  ..., 1.0227, 0.5994, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.02253235,  7.20826219,  7.32233733,  7.37576154,  7.10378919,
        7.23276582,  7.32950385, 11.00694688,  5.40689332, 13.0493368 ,
        6.80559669, 12.27993098,  9.05843201,  5.43544626, 12.67576792,
        8.08320235,  6.4020404 ,  9.62238828, 13.0743399 , 12.45418236,
        6.383897  , 13.05148826,  7.37576154,  7.23276582,  9.13507321,
       11.89105259,  6.80559669,  6.16512856,  7.9337647 , 12.25806559,
       13.26209545,  7.98469166, 11.00508681,  7.72415048,  6.69389431,
       12.58006038,  5.05989547,  9.85384623,  5.01114719,  5.87305005,
       12.22503754, 14.05633441,  6.67121748, 14.05687815,  7.5984809 ,
       11.00508681,  5.05963217,  9.41244865, 13.6922425 ,  9.69792513,
        7.09887557, 12.19805394, 12.07225469,  9.36378438,  9.17905491,
        4.91038884,  4.91038884,  5.19986797, 11.57249984, 10.00802627,
       11.4890861 ,  6.01356004,  9.5642993 , 10.0674887 ,  8.35856487,
        5.94909709,  8.06517143, 10.06850677, 11.48434767, 13.05148826,
        7.62232307, 13.89886469,  6.29034958,  5.40689332,  6.18668463,
       11.05672136, 10.69963954,  6.39520397, 13.97779839, 12.26997165,
        8.49161154,  4.978704  , 10.48315436,  7.39004226, 14.08549802,
        8.5724042 ,  8.41171369,  9.14973319,  6.38399789, 12.59840193,
        8.89301516,  5.79543038,  8.19703512,  5.04028538, 12.27835996,
        4.73784222, 13.60275211,  9.37296102, 14.0393949 ,  6.54615887,
        8.94395628, 11.46176259, 10.80728788, 13.92390689,  6.652742  ,
       13.59056613])}
{'P': tensor([[1.9993e-02, 2.3985e-01, 2.3384e-01,  ..., 2.7292e-04, 2.4190e-05,
         1.8832e-06],
        [2.0868e-03, 6.0122e-02, 1.0138e-02,  ..., 3.3663e-04, 9.5303e-05,
         5.2912e-06],
        [1.6215e-03, 2.4116e-01, 4.4008e-02,  ..., 2.5786e-04, 6.8788e-05,
         3.8265e-06],
        ...,
        [2.8370e-03, 4.4516e-03, 4.3979e-03,  ..., 9.6582e-03, 5.5319e-07,
         6.2818e-08],
        [5.3049e-05, 4.1799e-05, 4.2330e-05,  ..., 6.6056e-06, 9.4361e-02,
         2.9463e-03],
        [1.8092e-04, 1.1672e-04, 1.1712e-04,  ..., 1.1489e-04, 1.6290e-05,
         1.3420e-01]]), 'X': tensor([[  5.1861,   4.7361,  -3.7312],
        [  4.5430,   5.5438,  -3.8669],
        [  4.5800,   5.5034,  -3.9432],
        [  4.8653,   5.1694,  -4.0823],
        [  4.4049,   5.6892,  -4.4394],
        [  2.5375,   7.0676,  -4.3490],
        [ -5.0550,   6.8231,  -9.2502],
        [ -0.3432,   7.8900,   7.1805],
        [  2.1625,  -4.6697,   7.0658],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.3493,  -4.7727,  -6.5682],
        [  2.0870,  -4.6877,   9.6823],
        [ 12.5154,   2.5072,  -9.3274],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.0890,  -0.8382,   2.9166],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.2076,  13.8032,  -1.5099],
        [  4.8432,   5.1968,   1.5006],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.6022,  -1.9833,   8.3081],
        [  4.9152,  12.8653,  -1.4459],
        [  2.4697,  -4.5836,   1.4593],
        [ -1.0012, -10.8576,   2.2867],
        [  2.1379,  -4.6757,   6.4981],
        [  1.3767, -11.0341,   9.9415],
        [ -9.4753,  -0.8012,  -5.2050],
        [  0.4119,  14.1019,  -4.1870],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.1886,  -2.6250,   6.4677],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.2849, -10.9655,  -7.6542],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.4732,  -2.2041,  -4.3830],
        [ -0.0410,   7.8591,   9.3457],
        [ 10.7566,   7.6212,  -2.3279],
        [  7.3592,  11.3937,   6.2407],
        [  1.1006,   7.6318,  -2.4877],
        [ -1.9714,   7.8563,  -1.8084],
        [  3.3094,  13.4969,  -3.6507],
        [  6.3516,   0.6425,   0.3425],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.1929,   2.4501,  -6.9352],
        [ 12.5832,   1.7543,  -7.6134],
        [ 11.6544,   5.8273,  -3.1375],
        [  9.6614,   9.1784,  -1.2087],
        [  5.2939,  -2.4784,  -4.5840],
        [ -8.8400,   2.3991,   1.3457],
        [  3.7992,  -3.9474,  -8.9019],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.8836,   3.4315,  -4.0327],
        [  6.1752,   2.5242,   3.5874],
        [  6.3124,   0.2057,  -0.4593],
        [ -7.8280,   4.2848,  -7.1955],
        [  0.2732,  -4.7619,  -1.1577],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -5.9225,  -8.5355,   2.4778],
        [ -9.4157,   0.0821,  -5.0025],
        [ -3.5173, -10.0623,   4.8280],
        [ -8.1426,   3.8048,   9.3542],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -8.0750,  -6.0072,  -5.8267],
        [  0.9478,  -4.8144,   0.6371],
        [  2.3261,   7.1726,   6.2522],
        [  6.0047,   3.1062,  -6.7914],
        [  5.6928,  12.4710,   3.6001],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.6470, -11.0352,  -1.1931],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.2373,  -4.6506,  -1.7334],
        [ -7.4807,   4.7463,   2.0616],
        [ -1.9714,   7.8563,  -1.8084],
        [  5.9587,  12.3216,   9.1695],
        [  4.7739,  -3.1218,  -4.1188],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.0965,  -5.9712,   4.3466],
        [  4.6357,   5.4415,   7.9067],
        [  6.0955,   2.8210,  -4.3321],
        [  5.2939,  -2.4784,  -4.5840],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.2373,  -3.1225,  -8.9763],
        [  9.3424,  -7.3685,  -3.6807],
        [ 11.5198,  -4.0923,   5.5088],
        [  4.2612, -10.5514,  -6.6268],
        [  6.8093,  -9.4178,   8.4244],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.1178,   1.5784,   2.1410],
        [  5.6260,  -1.9401,   3.0259],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.1960,  -3.2964,   1.6921],
        [ -0.5385,   7.9037,   6.3830],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.3516,   0.6425,   0.3425],
        [ 11.9641,  -2.9568,   7.7211],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.2974,   0.8371,   2.4852],
        [  4.1637,  13.1898,  -1.9189],
        [ 11.5935,   5.9740,  -4.3952],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.5250, -11.0306,   5.5358]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9402e-02, 2.6850e-02,  ..., 2.9743e-01, 9.7404e-01,
         9.7133e-01],
        [2.9402e-02, 0.0000e+00, 5.6972e-03,  ..., 3.2225e-01, 9.4464e-01,
         9.5553e-01],
        [2.6850e-02, 5.6972e-03, 0.0000e+00,  ..., 3.1720e-01, 9.4768e-01,
         9.6122e-01],
        ...,
        [2.9743e-01, 3.2225e-01, 3.1720e-01,  ..., 3.0339e-04, 1.2329e+00,
         1.2508e+00],
        [9.7404e-01, 9.4464e-01, 9.4768e-01,  ..., 1.2329e+00, 0.0000e+00,
         9.1949e-01],
        [9.7133e-01, 9.5553e-01, 9.6122e-01,  ..., 1.2508e+00, 9.1949e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.02326515,  7.16747878,  7.15990676,  7.09887557,  7.19513806,
        7.5092834 ,  8.49161154,  7.89745107,  5.14607014,  8.62329466,
        4.78544875,  5.13125513, 12.76408562,  4.78258092,  6.1463896 ,
       12.84056604, 13.97857591,  7.10378919,  9.97220799,  5.94293185,
       13.77222752,  5.20659892, 10.9036226 ,  5.14124612, 11.11969815,
        9.50913849, 14.10796466,  9.73579738,  5.81481044, 12.59969936,
       12.7125079 , 11.20100795,  4.9160962 ,  5.90035324,  7.85919687,
       13.18278606, 13.56367857,  7.71075226,  8.09984231, 13.89671314,
        6.38399789, 11.92133129, 10.00802627,  6.65991877, 12.70489563,
       13.03002338, 13.32613445,  5.84533345,  9.15976746,  5.47866221,
        9.48969978,  6.81117802,  6.67121748,  6.31576264,  8.92395381,
        4.76969364,  9.61942445, 10.38897563,  9.41605684, 10.65929968,
        8.98765644,  9.43737523, 10.06437598,  4.90676242,  7.5403822 ,
        6.76056625, 13.70893492, 10.27572673,  9.50654967, 11.05413545,
       12.59969936,  5.16077764,  8.85939626,  8.09984231, 13.68672855,
        5.70405525,  8.19703512,  9.97220799, 10.06023665,  7.14837514,
        6.7166297 ,  5.84533345, 13.20597932,  9.75075273, 11.89855267,
       12.22503754, 11.37939695, 11.62158554,  8.09899874,  9.25336371,
        5.95111804, 10.794755  , 12.50996828,  6.12886024,  9.76896904,
        7.92200342,  8.47371868, 11.54672755,  6.38399789, 12.32408516,
        9.41244865,  9.3349858 , 13.83139024, 13.04217487,  9.94285511,
       11.04313641])}
{'P': tensor([[2.7614e-02, 2.0797e-01, 1.5796e-01,  ..., 2.3394e-05, 7.2541e-05,
         9.3078e-04],
        [1.3002e-03, 6.0496e-02, 4.0547e-01,  ..., 2.0044e-05, 1.0668e-04,
         6.2141e-04],
        [2.1874e-03, 1.0580e-02, 1.1423e-01,  ..., 3.2778e-05, 2.0027e-04,
         9.8688e-04],
        ...,
        [2.7597e-03, 1.1902e-03, 1.0013e-03,  ..., 9.1901e-03, 4.1242e-05,
         3.3387e-03],
        [1.4134e-02, 8.1949e-03, 8.1151e-03,  ..., 1.6778e-02, 2.2854e-01,
         8.8602e-02],
        [7.6721e-02, 8.0379e-02, 8.6305e-02,  ..., 2.7006e-04, 5.8684e-04,
         7.6066e-02]]), 'X': tensor([[  5.1793,   4.7461,   0.3899],
        [  3.7121,   6.3117,   0.6738],
        [  3.3943,   6.5483,   1.0790],
        [  4.3377,   5.7570,   1.3712],
        [  4.1348,   5.9514,  -0.8856],
        [  4.4633,   5.6287,  -0.7754],
        [  5.7845,   3.6652,   0.9175],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.1531,  -5.8748,   2.6934],
        [ -5.6651,   6.4380,   1.3660],
        [  1.0000,   7.6591,  -4.5011],
        [  5.8930,  12.3592,  -8.9201],
        [  5.4167, -10.1293,   6.2156],
        [  6.2555,   2.1474,   4.1685],
        [ 11.9695,  -2.9407,  -6.3498],
        [  9.0004,  -7.7195,   4.6945],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.3062,  -5.6008,   0.0287],
        [ -4.2040,   7.2486,   2.9875],
        [ -7.4733,   4.7555,   6.2490],
        [  7.0642,  11.6104,  -9.8316],
        [  6.2247,  -0.3150,  -9.3657],
        [  0.1460,   7.8340,  -9.4614],
        [  6.9956,  11.6590,  -7.6876],
        [ -3.2486,   7.5925,  -1.3719],
        [  9.9058,  -6.7203,   5.6928],
        [ -5.1369,   6.7755,   8.3836],
        [ -8.0829,  -5.9940,   5.0564],
        [  4.0642,  -3.7599,  -3.4565],
        [  9.9058,  -6.7203,   5.6928],
        [  2.3490,  -4.6199,  -4.2489],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.7685, -10.8982,  -6.8928],
        [  0.4996,   7.7738,  -4.4711],
        [  5.4732,  -2.2041,  -4.3830],
        [ -8.9178,   2.1936,   7.5388],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.2790,  -0.0259,  -5.4116],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.4877,  -0.7222,   8.8347],
        [  6.0305,   3.0296,   0.1375],
        [ -8.6675,  -4.8569,  -9.1769],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.9470,  13.6085,   2.9809],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.5789,   0.1722,   1.6470],
        [  0.3715,  -4.7756,   0.2312],
        [ 12.1588,   4.3364,  -8.0867],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.8994,   6.8675,  -9.5987],
        [  2.0729,   7.2878,   7.0880],
        [  9.3836,   9.5075,  -5.9278],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.3460,   1.4252,   1.2291],
        [ 10.2470,   8.4054,  -0.1177],
        [ -3.0008, -10.2827,  -8.9480],
        [ -8.3900,   3.3727,   3.1650],
        [  5.8201,   3.5841,  -8.0609],
        [  3.4604,  -4.1545,   1.8091],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.1890,   4.7318,   0.9515],
        [  6.0310,  12.2795,  -6.4528],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 12.4013,  -1.2597,  -7.6474],
        [  2.9470,  13.6085,   2.9809],
        [ -8.3900,   3.3727,   3.1650],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.3900,   3.3727,   3.1650],
        [  6.7792,  11.8084,   1.6251],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.5650,   2.0132,   4.6272],
        [ -8.4297,  -5.3637,  -9.3962],
        [  2.7121,   6.9743,  -8.4202],
        [ -7.6734,   4.4980,  -5.4248],
        [ 11.6544,   5.8273,  -3.1375],
        [  3.5939,   6.4029,  -1.8357],
        [ 10.5955,   7.8824,  -3.7006],
        [ -3.3812, -10.1235,   7.4384],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.2905,   1.9366,   2.1169],
        [ 11.1469,   6.9237,   1.6160],
        [  5.9843,  -9.8679,  -6.1449],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -7.7842,   4.3466,  -1.7548],
        [ 10.9931,   7.2107,   7.7933],
        [  1.3705,  13.9723,   7.9266],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.7162,   5.6728,  -8.9372],
        [ -6.6998,   5.5954,   0.0823],
        [ -2.6485, -10.4151,   0.8766],
        [ -5.2589,   6.7024,  -5.5809],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.7779,   5.5127,  -2.0699],
        [  1.3339,   7.5627,  -0.9034],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.3860,  -2.3418,   1.0826],
        [ 12.2311,  -2.0445,   1.3545],
        [  4.5491,  -3.3485,   6.8412],
        [  6.1752,   2.5242,   3.5874]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.2015e-02, 8.3209e-02,  ..., 7.2227e-02, 4.2684e-01,
         2.4783e-01],
        [6.2015e-02, 0.0000e+00, 2.9329e-02,  ..., 1.1345e-01, 4.1977e-01,
         2.4880e-01],
        [8.3209e-02, 2.9329e-02, 0.0000e+00,  ..., 1.1836e-01, 3.9617e-01,
         2.2924e-01],
        ...,
        [7.2227e-02, 1.1345e-01, 1.1836e-01,  ..., 1.0727e-04, 3.7073e-01,
         1.8999e-01],
        [4.2684e-01, 4.1977e-01, 3.9617e-01,  ..., 3.7073e-01, 0.0000e+00,
         1.8075e-01],
        [2.4783e-01, 2.4880e-01, 2.2924e-01,  ..., 1.8999e-01, 1.8075e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.02496223,  7.32233733,  7.37576154,  7.20826219,  7.24677664,
        7.18354885,  6.84796202, 10.48840476, 10.04916678,  8.57560275,
        7.72415048, 13.6922425 , 11.48663767,  6.61386436, 12.32545637,
       11.85743425, 10.71162681, 13.97857591,  9.27474386, 10.01804223,
        8.37954853,  8.8580671 , 13.59056613,  6.23262289,  7.83534987,
       13.59673406,  8.25828409, 11.9702762 ,  8.50268322, 10.06286074,
        5.5366698 , 11.9702762 ,  5.18277858,  8.20436764, 11.24434374,
        7.78980299,  5.90035324,  9.18358562,  9.51074763,  6.27905292,
       12.62711252, 12.5085995 ,  6.74872981,  9.93553424, 11.86239961,
       13.92390689, 13.0743399 , 12.77402351, 12.58006038,  4.79003361,
       12.9089559 ,  9.97220799,  7.4544866 ,  7.57685908, 13.35832725,
       13.13888841,  6.50409372, 13.25336001, 10.71162681,  9.04255655,
        6.83515201,  5.40689332,  9.88571713, 10.06407839, 12.18045978,
        7.02253235, 13.68063098,  9.54154096,  9.5014146 , 12.46513646,
       13.92390689,  9.04255655,  9.43737523, 13.06957425,  9.04255655,
       13.61600536, 12.67680439, 12.72524615,  9.99145096,  7.48311002,
        8.89458273, 13.03002338,  7.34250948, 13.20597932, 10.67322527,
       13.09737053,  6.58184474, 13.12218013, 11.54071859, 10.61047596,
        8.91551231, 13.14690137, 14.0393949 ,  4.87083348, 13.01728313,
        8.72896381, 10.74655346,  8.51928531,  8.50268322, 13.00413829,
        7.67939546, 12.19805394,  5.87305005, 12.40074901,  5.64865173,
        6.67121748])}
{'P': tensor([[1.8523e-02, 1.8254e-01, 1.5761e-01,  ..., 2.0239e-05, 1.1634e-05,
         3.2400e-03],
        [7.7488e-04, 2.7017e-02, 6.2126e-02,  ..., 4.3177e-05, 2.2655e-05,
         7.5956e-03],
        [5.2692e-04, 4.8993e-03, 1.9966e-02,  ..., 3.9840e-05, 1.8528e-05,
         7.7035e-03],
        ...,
        [7.5716e-06, 6.0620e-06, 6.5824e-06,  ..., 3.5073e-02, 2.0320e-02,
         1.3788e-05],
        [5.1574e-05, 3.3472e-05, 3.2647e-05,  ..., 1.7136e-04, 7.6879e-02,
         1.3973e-05],
        [2.2783e-04, 3.2105e-04, 3.5921e-04,  ..., 1.1859e-02, 1.7456e-03,
         1.4021e-01]]), 'X': tensor([[ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0371, 0.0403,  ..., 1.0225, 0.9342, 0.3820],
        [0.0371, 0.0000, 0.0257,  ..., 0.9979, 0.8993, 0.3621],
        [0.0403, 0.0257, 0.0000,  ..., 0.9827, 0.8962, 0.3436],
        ...,
        [1.0225, 0.9979, 0.9827,  ..., 0.0000, 0.4857, 0.6456],
        [0.9342, 0.8993, 0.8962,  ..., 0.4857, 0.0000, 0.6630],
        [0.3820, 0.3621, 0.3436,  ..., 0.6456, 0.6630, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.03733775,  7.24326992,  7.27759233,  7.32950385,  7.27340834,
        7.37902766,  5.19065439, 10.01804223,  7.05627461, 10.51340829,
        9.27474386,  6.76791483,  5.64464807, 13.56367857,  7.32950385,
        9.96152788,  9.85330646, 13.68063098,  7.46248947,  5.94541803,
       11.69264243,  9.7751189 , 13.45124917, 12.86648771,  6.81084724,
       12.52875793, 12.22503754,  8.50268322, 10.65929968, 13.25017211,
        8.34949383,  9.37989631, 13.97430191, 11.46176259, 13.58101794,
        7.07902036,  6.81084724, 10.57302546, 13.45642379, 11.70016217,
        7.32233733, 11.87412439, 12.4980845 ,  7.52117132,  5.19577146,
       10.76154256, 12.64235359,  8.66273171,  6.46356806, 13.6105595 ,
        9.6282763 , 10.06437598, 12.27993098,  7.23276582, 10.71162681,
        6.46356806, 12.71843462,  6.46356806,  7.28793663,  8.29640613,
        8.74470218, 10.412299  , 12.9089559 ,  8.77913428, 11.85890664,
       13.31764181,  9.15976746,  8.41171369, 11.018084  , 13.59170809,
        8.41148722,  9.9353921 ,  6.61386436, 10.54658597, 12.32545637,
       11.20100795,  9.93553424, 13.77604293,  7.6374603 , 13.61801526,
        8.06990483,  5.96556248, 13.83139024,  9.60080639,  8.98765644,
       13.45642379, 11.46176259,  6.76056625,  6.16512856,  7.56617037,
       13.5247215 , 13.32374056,  8.51928531, 13.6922425 ,  9.23178204,
       11.74901778, 12.70489563,  5.5671389 ,  5.21280156,  5.43943791,
       12.05300943,  6.29034958,  8.8353674 ,  9.63742669, 10.1979669 ,
        8.23525433])}
{'P': tensor([[2.5777e-02, 2.4717e-01, 1.4714e-01,  ..., 6.2557e-03, 5.0669e-05,
         2.4910e-06],
        [9.1229e-04, 2.3076e-02, 1.9044e-01,  ..., 5.6174e-03, 3.8971e-05,
         2.4354e-06],
        [5.8316e-04, 9.2263e-04, 3.8122e-02,  ..., 3.8598e-03, 4.1746e-05,
         4.1276e-06],
        ...,
        [2.8598e-05, 6.3533e-05, 3.0310e-04,  ..., 1.0845e-02, 1.1957e-06,
         1.4285e-07],
        [4.1186e-03, 3.6962e-03, 1.8642e-03,  ..., 4.9804e-04, 1.1382e-01,
         1.7123e-04],
        [3.0973e-05, 2.2924e-05, 1.0883e-05,  ..., 1.2827e-06, 6.5434e-03,
         3.8906e-02]]), 'X': tensor([[ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0297, 0.1080,  ..., 0.2998, 0.3383, 0.8060],
        [0.0297, 0.0000, 0.0793,  ..., 0.3288, 0.3407, 0.7801],
        [0.1080, 0.0793, 0.0000,  ..., 0.3995, 0.3813, 0.7294],
        ...,
        [0.2998, 0.3288, 0.3995,  ..., 0.0000, 0.4923, 1.0974],
        [0.3383, 0.3407, 0.3813,  ..., 0.4923, 0.0000, 0.7203],
        [0.8060, 0.7801, 0.7294,  ..., 1.0974, 0.7203, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.05627461,  7.20482224,  7.4544866 ,  7.53884104,  7.48311002,
        7.50035355, 11.81423161, 13.6829843 , 11.64746882,  8.12580976,
       13.83139024,  4.98461983,  6.05113217,  7.8788965 ,  8.19703512,
        9.09963894,  7.98469166,  4.85603886, 13.77075174,  4.9160962 ,
       10.54658597,  9.7712321 , 10.1979669 ,  9.09502385, 11.36089355,
        9.48969978, 11.0050379 ,  7.50500064,  4.76969364, 13.09426142,
        4.83371341,  9.41244865,  4.78544875, 12.04924159,  7.67939546,
        8.859416  , 13.66051884, 14.05633441,  5.05534609,  7.96181288,
       13.95361098, 13.66051884, 12.5085995 ,  8.90405823,  8.66273171,
        4.73784222, 13.744387  , 12.13568598, 11.69264243,  7.01475376,
        8.51928531,  6.42439344,  7.15100303, 11.85890664, 14.10796466,
        8.04664886,  9.43112957,  6.81084724, 13.95472331, 11.99633819,
        7.10378919,  6.97106912, 12.33167382,  6.02789634,  5.56809335,
       13.15525803, 12.85032836,  7.4544866 , 10.90638066,  8.77913428,
        8.0550286 ,  5.60064319,  9.48969978, 12.67576792,  7.76457095,
        5.91408504, 13.04156075, 10.65929968,  8.86082124,  7.34250948,
       13.85158417,  9.34061938,  4.78258092, 13.26209545, 14.09176259,
       10.06407839,  8.66273171, 13.1399181 ,  6.18668463, 11.05317597,
       11.00726483,  9.99211663, 13.6660671 , 11.46176259,  9.60080639,
        9.61942445,  7.16598273, 13.88683436,  7.59795445,  9.09963894,
        9.32094055,  5.35774573, 11.05317597, 13.84126197,  5.69913817,
       10.88618114])}
{'P': tensor([[1.8640e-02, 2.5309e-01, 1.6143e-01,  ..., 3.3578e-03, 4.4345e-03,
         2.2808e-05],
        [1.0491e-03, 2.8764e-02, 1.3323e-02,  ..., 2.2012e-03, 7.3366e-03,
         2.2070e-05],
        [1.8284e-03, 4.9304e-02, 3.3112e-02,  ..., 2.6641e-03, 9.1016e-03,
         2.4846e-05],
        ...,
        [2.1694e-02, 4.0420e-02, 2.8285e-02,  ..., 4.3009e-02, 4.6812e-03,
         1.0390e-04],
        [3.9681e-04, 6.8132e-04, 6.5335e-04,  ..., 2.4491e-04, 2.3959e-01,
         2.3101e-06],
        [1.0094e-03, 9.9558e-04, 8.3988e-04,  ..., 2.8781e-03, 3.0457e-04,
         1.5352e-02]]), 'X': tensor([[ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0273, 0.0329,  ..., 0.1527, 0.4280, 0.5868],
        [0.0273, 0.0000, 0.0144,  ..., 0.1590, 0.4470, 0.5971],
        [0.0329, 0.0144, 0.0000,  ..., 0.1732, 0.4390, 0.6111],
        ...,
        [0.1527, 0.1590, 0.1732,  ..., 0.0000, 0.5370, 0.4391],
        [0.4280, 0.4470, 0.4390,  ..., 0.5370, 0.0000, 0.8653],
        [0.5868, 0.5971, 0.6111,  ..., 0.4391, 0.8653, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.06169391,  7.17789666,  7.16598273,  7.38844703,  7.32754882,
        7.46248947, 12.1494023 ,  9.6282763 ,  7.48538905, 13.48809595,
        5.13616224, 13.97779839,  7.06169391, 13.45124917,  7.76457095,
        9.51074763,  5.89074104, 13.05148826,  6.77777606,  9.53015376,
        8.56889738, 11.73397309,  5.70405525,  7.39599986, 10.24557047,
        5.92952538,  9.97822473,  5.84450127,  7.02496223,  9.34061938,
        7.26074925, 12.35421461, 13.06498554, 13.88683436, 10.35464916,
       13.32645552,  5.05444288,  8.87587457,  6.65368551,  5.08372149,
        6.65259775, 13.06957425, 10.95593729,  5.91408504,  6.48486171,
        7.5092834 , 13.68672855, 10.06407839,  7.62232307,  6.53290663,
        8.04483495,  9.47369628,  6.01356004,  7.66377554, 13.96746816,
        6.81084724, 10.6910768 , 13.6660671 ,  5.16611782,  9.562174  ,
       11.42420508,  8.41148722,  5.20659892,  9.94173708,  8.20436764,
        6.54693987,  6.10360073,  9.58347456, 11.64746882,  5.84450127,
       11.89855267,  8.77913428,  6.39520397,  5.69913817, 14.10045797,
        7.48538905,  7.72291946, 12.67680439, 11.00694688, 12.33167382,
       13.25017211,  6.81084724,  5.81199422, 11.73848065,  5.39098669,
        9.3349858 ,  4.78002665,  9.38232121, 12.17901033, 11.81423161,
        7.10378919,  5.46244988,  8.79284466, 13.00413829,  8.08320235,
        5.9510489 , 11.31276109,  9.09661212,  8.92549752,  9.75075273,
       12.5769659 ,  7.60463349,  9.36093562,  6.99097013, 13.93790235,
       12.7125079 ])}
{'P': tensor([[2.5212e-02, 1.5202e-01, 1.4584e-01,  ..., 6.3817e-06, 1.7800e-05,
         1.2911e-03],
        [1.0420e-03, 3.0337e-02, 3.3245e-03,  ..., 8.2447e-06, 2.7593e-05,
         3.9545e-04],
        [3.5433e-03, 9.4192e-02, 4.8330e-02,  ..., 1.4423e-05, 4.4940e-05,
         1.2052e-03],
        ...,
        [1.5854e-04, 8.0398e-05, 1.3413e-04,  ..., 2.0814e-01, 3.3926e-04,
         1.3485e-05],
        [7.2437e-06, 4.5047e-06, 6.6447e-06,  ..., 1.2909e-01, 1.9768e-02,
         4.1417e-07],
        [3.6410e-04, 1.4311e-03, 5.8640e-04,  ..., 6.2654e-08, 2.3163e-07,
         5.0834e-03]]), 'X': tensor([[  5.0233,   4.9647,  -3.1788],
        [  3.9085,   6.1512,  -2.2560],
        [  4.5430,   5.5438,  -3.8669],
        [  4.5800,   5.5034,  -3.9432],
        [  3.6469,   6.3624,  -1.8414],
        [  3.5939,   6.4029,  -1.8357],
        [  0.9036,   7.6840,   9.9237],
        [  8.9284,  10.0036,  -6.9210],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.5010,  -4.5736,  -0.8989],
        [ 11.8951,  -3.1570,   7.0742],
        [ -4.8751,  -9.3148,   4.0406],
        [  1.1006,   7.6318,  -2.4877],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.4179,  -2.2925,  -3.4641],
        [ -6.4464,   5.8272,   7.5038],
        [ 11.0275,  -5.0666,   8.4154],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.3516,   0.6425,   0.3425],
        [  6.1989,  -9.7591,   0.8289],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.2096,   5.8815,   1.9230],
        [  6.1730,  -0.5375,  -8.8861],
        [ 11.7796,  -3.4696,   9.0088],
        [  2.2659,  13.7894,   6.5436],
        [ -5.1377,   6.7751,  -0.4438],
        [  4.6231,   5.4556,   8.4973],
        [ -3.9508,  -9.8515,   1.7923],
        [ -4.4520,   7.1370,  -8.7867],
        [  1.0093,   7.6567,   2.3526],
        [ -2.6069, -10.4298,  -6.1995],
        [  6.3413,   0.4883,  -6.8416],
        [ -4.6393,   7.0461,   1.1624],
        [  2.3261,   7.1726,   6.2522],
        [ -1.5408,   7.8978,   5.3470],
        [  6.3516,   0.6425,   0.3425],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.3038,   7.8092,   0.4164],
        [ -8.3535,  -5.5118,   9.8399],
        [ -7.2515,   5.0195,   4.6434],
        [  0.9074,  -4.8139,   8.1674],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.7121,   6.9743,  -8.4202],
        [  1.1350,  14.0104,   0.6321],
        [ 11.5966,   5.9666,   6.0083],
        [  8.3718,  -8.2953,   8.4666],
        [ -6.8618,  -7.6322,   3.7027],
        [  5.3602,  -2.3808,   7.2098],
        [  5.9526,   3.2525,  -4.3821],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.3377,   5.7570,   1.3712],
        [  0.1041, -11.0045,   2.4311],
        [  5.7802,  -1.6363,  -6.2782],
        [  1.0093,   7.6567,   2.3526],
        [  0.9809,   7.6642,  -1.4217],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -7.7239,   4.4298,   3.1216],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.4867,  12.5816,   2.9549],
        [ -8.5900,   2.9767,   7.4811],
        [  5.9912,   3.1453,  -3.9989],
        [  1.6408,   7.4591,  -4.2294],
        [  5.1793,   4.7461,   0.3899],
        [  7.9816,  10.8938,  -0.5378],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -4.2040,   7.2486,   2.9875],
        [  6.3534,   1.2990,  -7.2942],
        [ -8.8058,   2.4850,  -1.7425],
        [ 12.5717,   1.9266,   7.4903],
        [  0.1287, -11.0065,   6.5757],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.3534,   1.2990,  -7.2942],
        [ -1.5408,   7.8978,   5.3470],
        [  5.8275,  -1.5334,   2.0124],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -9.4735,  -1.2620,  -3.6742],
        [  5.0617,  12.7957,   8.3544],
        [  5.6248,   3.9984,   8.4933],
        [ 10.5955,   7.8824,  -3.7006],
        [  7.8425,  -8.7202,   6.4530],
        [  5.1142,  -2.7230,  -4.1966],
        [ -6.0175,  -8.4541,  -9.8408],
        [  7.9816,  10.8938,  -0.5378],
        [  5.8902,  -1.3887,   4.0388],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.1737,   2.5305,   3.9049],
        [  9.7328,   9.0902,  -0.8917],
        [  9.6662,   9.1725,   8.1056],
        [  5.1667,  -2.6543,   1.2879],
        [  4.8653,   5.1694,  -4.0823],
        [  9.2811,  -7.4336,   5.1771],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -2.2188,   7.8218,   8.6064],
        [  1.4306,  13.9620,  -0.2075],
        [ -8.8363,   2.4086,  -5.6296],
        [  1.7659,   7.4127,   4.7871],
        [  6.2555,   2.1474,   4.1685],
        [ -1.6275, -10.7221,   3.2376],
        [  3.7916,  -3.9524,  -7.9547],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -8.3717,  -5.4769,  -9.2228],
        [  9.7328,   9.0902,  -0.8917]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0733, 0.0437,  ..., 0.9432, 1.0439, 0.3360],
        [0.0733, 0.0000, 0.1029,  ..., 0.9308, 1.0279, 0.3556],
        [0.0437, 0.1029, 0.0000,  ..., 0.9118, 1.0140, 0.3702],
        ...,
        [0.9432, 0.9308, 0.9118,  ..., 0.0000, 0.1109, 1.2790],
        [1.0439, 1.0279, 1.0140,  ..., 0.1109, 0.0000, 1.3790],
        [0.3360, 0.3556, 0.3702,  ..., 1.2790, 1.3790, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.06272256,  7.28793663,  7.16747878,  7.15990676,  7.33350642,
        7.34250948,  7.73692436, 13.40850328,  9.91668182,  5.21280156,
       12.30695098, 10.51340829,  7.71075226,  7.9337647 ,  5.8828986 ,
        8.68979052, 12.13568598, 10.48721382,  6.38399789, 11.56146873,
        9.75401449,  7.23276582,  6.19633139, 12.27993098, 13.97430191,
        8.50278896,  7.15100303, 10.61417181,  8.41171369,  7.72291946,
       10.7506423 ,  6.36004393,  8.43626114,  7.5403822 ,  8.04664886,
        6.38399789, 12.18045978, 10.30972193,  9.50913849, 10.52357479,
        7.81509631, 10.00802627,  8.81930525,  4.89869205, 11.86239961,
        7.48311002, 14.05633441, 13.04156075, 11.785566  , 10.26327895,
        5.86519525,  6.78325151,  6.83083863,  9.60412735,  7.20826219,
       11.0050379 ,  6.00732193,  7.72291946,  7.72668467, 10.27716434,
        8.90405823,  4.9160962 , 13.72594022,  9.09119351,  6.76661595,
        7.6374603 ,  7.02496223, 13.50484539,  9.62958973,  8.37954853,
        6.48486171,  9.14973319, 12.71843462, 11.00726483,  8.47371868,
        6.48486171,  8.04664886,  6.02588517, 10.32715715,  9.55720651,
       13.76047999,  6.90115901, 13.20597932, 11.72802491,  5.79391825,
       10.37698435, 13.50484539,  6.05165248,  9.85330646,  6.67218687,
       13.31764181, 13.32557133,  5.80860165,  7.09887557, 11.89105259,
       12.04924159,  8.1303945 , 14.03505624,  9.15866607,  7.62011734,
        6.61386436, 10.84493732,  5.47702766, 10.53368644, 10.00410625,
       13.31764181])}
{'P': tensor([[3.1071e-02, 1.5910e-01, 1.5608e-01,  ..., 6.9366e-06, 2.6862e-02,
         7.8210e-06],
        [1.2496e-03, 8.5986e-02, 1.2199e-01,  ..., 5.2191e-06, 2.4050e-01,
         1.1220e-05],
        [9.7086e-04, 2.9050e-02, 7.1349e-02,  ..., 5.0273e-06, 2.7203e-01,
         9.2018e-06],
        ...,
        [8.6586e-06, 1.0472e-05, 1.3895e-05,  ..., 1.1366e-02, 1.6083e-05,
         6.4246e-08],
        [9.9319e-04, 2.7395e-03, 2.6995e-03,  ..., 6.9949e-06, 2.7956e-01,
         2.7804e-05],
        [9.1115e-04, 1.8857e-04, 1.7907e-04,  ..., 2.8951e-05, 2.9928e-05,
         2.0340e-02]]), 'X': tensor([[  5.0190,   4.9705,  -7.9752],
        [  2.7121,   6.9743,  -8.4202],
        [  2.5974,   7.0363,  -7.7810],
        [  4.3554,   5.7393,  -9.0766],
        [  2.5663,   7.0526,  -6.8769],
        [  3.9895,   6.0816,  -6.5235],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 12.6038,   0.7665,  -3.4439],
        [  2.0243,   7.3086,   7.5432],
        [ -6.3306,   5.9272,   5.6935],
        [  6.3505,   0.6235,  -8.5849],
        [ -5.0943,  -9.1682,   0.5073],
        [  6.3124,   0.2057,  -0.4593],
        [  3.6469,   6.3624,  -1.8414],
        [  2.2644,  -4.6434,   9.7508],
        [  6.1372,   2.6726,  -6.7885],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.8617,   3.4853,  -9.8080],
        [  9.5203,   9.3483,   5.8781],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.3246,   1.6696,  -0.1544],
        [ -0.7599, -10.8995,  -6.0366],
        [ -3.5520,   7.4976,   9.1522],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.3107,  -2.7697,  -3.2919],
        [  6.3031,   1.8477,  -7.1706],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.5149,  -5.1905,   1.5239],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.2339,  -3.6268,   6.3125],
        [  1.9505,  -4.7171,   9.9868],
        [  6.3048,   0.1469,  -9.1691],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.2860,   7.1916,  -6.1011],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -9.1459,   1.4780,  -4.2403],
        [  9.6614,   9.1784,  -1.2087],
        [  9.3753,   9.5171,   3.9131],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.6928,  12.4710,   3.6001],
        [  3.4140,  -4.1803,  -8.8224],
        [ -1.6086,   7.8928,   4.3385],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.6414, -10.0303,   7.8525],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.3539,  -5.5111,   0.3373],
        [  9.2692,   9.6370,   7.0746],
        [  6.2883,   0.0325,   5.2692],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.2707,   0.9661,   4.4863],
        [  9.3836,   9.5075,  -5.9278],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.0047,   3.1062,  -6.7914],
        [ -9.4430,  -0.1897,   4.8383],
        [ -0.4341, -10.9473,   7.4509],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.6229,   2.9068,   2.4429],
        [  9.2692,   9.6370,   7.0746],
        [  9.3753,   9.5171,   3.9131],
        [  3.2647, -10.8066,  -0.3768],
        [  4.1348,   5.9514,  -0.8856],
        [  2.3373,  13.7721,  -0.2442],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.8835,  -4.3178,   9.3080],
        [  3.3474,  -4.2162,   5.2455],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.5154,   2.5072,  -9.3274],
        [  6.3606,   0.9204,   3.2525],
        [ 10.8218,  -5.4177,   6.8573],
        [  5.1890,   4.7318,   0.9515],
        [ -1.6334, -10.7207,   7.1397],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.3534,   1.2990,  -7.2942],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.3122,  -9.6994,  -6.3282],
        [  1.0093,   7.6567,   2.3526],
        [  4.5491,  -3.3485,   6.8412],
        [ 10.3251,   8.2928,  -3.0006],
        [ -5.1046,  -9.1610,   2.3418],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3551,   0.7149,   1.3344],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.1196,  -2.7160,   4.2705],
        [  1.6232,  -4.7717,  -2.5655],
        [  9.3560,   9.5391,   9.4943],
        [ -0.5709,   7.9055,   7.6089],
        [ 10.0400,   8.6922,  -6.2847],
        [ 11.3367,  -4.4806,   0.4643],
        [ -8.3900,   3.3727,   3.1650],
        [  5.7845,   3.6652,   0.9175],
        [ -9.2707,   0.9661,   4.4863],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.5568,   6.0602,   9.3244],
        [ -1.0407,   7.9166,  -6.6021],
        [  8.6271,  -8.0715,  -7.4813]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1225, 0.1158,  ..., 1.2313, 0.2408, 0.1805],
        [0.1225, 0.0000, 0.0190,  ..., 1.2947, 0.1257, 0.0976],
        [0.1158, 0.0190, 0.0000,  ..., 1.2759, 0.1260, 0.1165],
        ...,
        [1.2313, 1.2947, 1.2759,  ..., 0.0000, 1.3126, 1.3896],
        [0.2408, 0.1257, 0.1260,  ..., 1.3126, 0.0000, 0.1699],
        [0.1805, 0.0976, 0.1165,  ..., 1.3896, 0.1699, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.06372172,  7.48311002,  7.50035355,  7.20482224,  7.50500064,
        7.27340834, 12.05300943, 12.62711252,  7.58378655,  8.67227386,
        6.38105015, 10.48840476,  6.31576264,  7.33350642,  5.16611782,
        6.69389431,  5.63824971,  6.81961996, 13.3426541 , 10.31613269,
        8.83725081,  6.54127431, 10.92597196,  8.29640613,  6.10774544,
        5.9510489 ,  7.27340834,  9.36378438,  9.71392179,  6.56833864,
        9.88571713, 10.01804223,  9.97220799,  9.60412735,  5.5748624 ,
        5.10446653,  6.30647746,  7.9337647 ,  7.54620849, 10.46739073,
        9.26455795, 13.32613445, 13.35927527,  4.978704  , 13.70893492,
        5.39723241,  8.0550286 ,  9.38232121,  9.93553424,  5.18277858,
       11.50791859, 10.75443106,  9.70108409, 12.0495521 , 10.00794468,
       13.37121923,  6.28835101,  9.19837792,  9.32094055, 13.35832725,
        5.90035324,  6.76056625,  9.44486609, 10.95593729, 10.86693708,
        9.09963894, 13.37121923, 13.35927527, 11.28895489,  7.24677664,
       13.96905756,  5.5671389 ,  9.87720102,  5.38342077,  8.43626114,
       12.76408562,  6.42689705, 12.10222204,  7.02253235, 10.84437705,
        5.18277858,  6.48486171,  7.9337647 , 11.57249984,  7.72291946,
        5.64865173, 13.2430377 , 10.48721382,  9.17905491,  6.02588517,
        6.39520397,  9.69792513,  8.92395381,  5.79543038,  5.04028538,
       13.3614598 ,  7.92607342, 13.27994304, 12.18998195,  9.04255655,
        6.84796202,  9.32094055, 12.26997165, 13.0493368 ,  7.98469166,
       11.81423161])}
{'P': tensor([[2.2007e-02, 1.8160e-01, 1.4182e-01,  ..., 2.6891e-06, 1.2552e-04,
         4.2093e-06],
        [3.6347e-04, 1.4301e-02, 1.7657e-01,  ..., 9.3536e-07, 1.6878e-04,
         1.2283e-06],
        [3.8427e-04, 9.4423e-04, 1.7880e-02,  ..., 9.9771e-07, 2.5818e-04,
         1.2471e-06],
        ...,
        [8.2433e-05, 3.9879e-05, 3.1565e-05,  ..., 9.8130e-03, 1.1662e-07,
         1.5759e-01],
        [2.3337e-05, 2.1312e-05, 2.1316e-05,  ..., 4.2874e-06, 5.6801e-02,
         1.0778e-06],
        [3.3441e-04, 2.0806e-04, 1.7167e-04,  ..., 4.5862e-05, 4.8353e-07,
         4.1076e-02]]), 'X': tensor([[ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.1528e-02, 7.3138e-02,  ..., 8.6437e-01, 8.0432e-01,
         9.4870e-01],
        [5.1528e-02, 0.0000e+00, 2.1701e-02,  ..., 8.9241e-01, 7.5742e-01,
         9.8332e-01],
        [7.3138e-02, 2.1701e-02, 0.0000e+00,  ..., 9.0668e-01, 7.3900e-01,
         9.9994e-01],
        ...,
        [8.6437e-01, 8.9241e-01, 9.0668e-01,  ..., 0.0000e+00, 1.2587e+00,
         1.7242e-01],
        [8.0432e-01, 7.5742e-01, 7.3900e-01,  ..., 1.2587e+00, 3.0339e-04,
         1.4153e+00],
        [9.4870e-01, 9.8332e-01, 9.9994e-01,  ..., 1.7242e-01, 1.4153e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.07902036,  7.32754882,  7.38844703,  7.44805901,  7.46053281,
        7.46248947,  9.51074763,  6.34059021,  8.89301516,  8.87007495,
       10.8992524 ,  8.7907923 , 10.84437705, 13.2778378 , 13.97430191,
       12.5085995 ,  8.81930525, 12.17901033,  9.11473391, 12.889695  ,
       11.65923396,  8.859416  ,  9.34127013,  9.88571713, 12.4217523 ,
       11.73397309, 11.00508681, 11.37939695, 12.22276278, 11.1341798 ,
       13.06498554,  7.81509631,  7.92607342, 14.02526505,  7.27759233,
        6.41202986,  8.36445758, 10.794755  ,  5.64464807,  7.76457095,
        6.38968357, 13.59480906,  5.39450305,  8.31625171, 13.39372426,
       12.62576026, 10.00802627,  7.96181288,  7.70897675, 13.3614598 ,
       10.7167285 ,  5.79543038, 13.34587027, 13.59056613,  8.85939626,
        5.5671389 ,  5.19986797,  8.09842704,  5.27178363, 11.04313641,
        8.16584537, 11.64746882,  5.14999799,  8.49161154,  8.06643487,
       11.05413545,  8.36445758,  8.51928531, 12.35625694, 10.87409005,
        9.97964013,  7.57685908, 12.13568598,  7.5403822 , 12.19805394,
       13.41713661,  9.77588801, 11.83509372, 10.35485331, 11.51557166,
       13.34587027,  8.29022197, 11.51811603,  5.87305005, 11.11969815,
       12.18998195,  6.21275016, 11.38774149,  7.28793663,  9.15253344,
       10.412299  , 12.5769659 , 13.1471313 , 13.68063098,  4.78002665,
        6.38968357,  7.06372172,  5.29009836,  4.78544875,  9.54154096,
        8.99276564, 12.3828957 , 10.51340829, 12.05395056,  9.41244865,
       12.42554808])}
{'P': tensor([[1.4770e-02, 2.0620e-01, 1.7915e-01,  ..., 4.5068e-04, 2.5559e-06,
         8.3076e-04],
        [1.4081e-03, 2.8969e-02, 1.5875e-01,  ..., 5.8578e-04, 4.8376e-06,
         2.0147e-03],
        [1.9390e-03, 6.2413e-03, 3.7014e-02,  ..., 7.0942e-04, 6.1769e-06,
         2.6874e-03],
        ...,
        [1.7899e-02, 2.4626e-02, 2.5386e-02,  ..., 1.1378e-01, 1.0997e-06,
         1.2724e-03],
        [2.4557e-04, 2.0175e-04, 1.9793e-04,  ..., 1.0295e-04, 4.0264e-02,
         4.4998e-06],
        [1.2762e-04, 1.3824e-04, 1.4055e-04,  ..., 7.2076e-06, 1.0460e-04,
         1.6334e-01]]), 'X': tensor([[  4.8653,   5.1694,  -4.0823],
        [  4.5800,   5.5034,  -3.9432],
        [  4.5430,   5.5438,  -3.8669],
        [  4.4049,   5.6892,  -4.4394],
        [  2.5375,   7.0676,  -4.3490],
        [  3.0876,   6.7527,  -5.3540],
        [  1.1994,   7.6035,   2.4295],
        [ -1.5408,   7.8978,   5.3470],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.3231,  -4.7691,  -1.6120],
        [ -3.9700,   7.3453,  -4.1486],
        [  9.6614,   9.1784,  -1.2087],
        [ -5.6423,   6.4537,  -1.2091],
        [  6.3301,   0.3638,  -7.5685],
        [ 11.5456,   6.0861,   8.1101],
        [  0.1532,  -4.7422,  -7.0145],
        [  4.6231,   5.4556,   8.4973],
        [  2.7685, -10.8982,  -6.8928],
        [  0.3038,   7.8092,   0.4164],
        [  2.4932,   7.0903,  -5.4214],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.8889,  -1.3916,   8.8993],
        [  0.7958,  -4.8109,  -8.2248],
        [  4.1297,  -3.7098,  -3.9838],
        [ -6.3306,   5.9272,   5.6935],
        [  4.2612, -10.5514,  -6.6268],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.1973,  -0.4372,   6.5720],
        [  1.7815,  -4.7481,   4.4289],
        [ -5.6651,   6.4380,   1.3660],
        [  4.8432,   5.1968,   1.5006],
        [ -5.3382,   6.6533,   7.8054],
        [  3.4604,  -4.1545,   1.8091],
        [  6.2247,  -0.3150,  -9.3657],
        [  0.3038,   7.8092,   0.4164],
        [  3.5284, -10.7484,   5.3347],
        [  9.3624,   9.5318,   0.2779],
        [ -6.1899,  -8.3011,  -7.9269],
        [  4.9957,  -2.8705,   6.5978],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -5.2752,  -9.0409,  -8.2568],
        [  1.7659,   7.4127,   4.7871],
        [ 12.2921,  -1.7902,  -4.0870],
        [  3.1064,  13.5607,   6.7605],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.7218,  -9.9934,   6.5764],
        [  5.6258,  -1.9404,  -6.0715],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -1.8788, -10.6567,   3.7068],
        [  1.5383, -11.0274,  -4.9666],
        [  5.9912,   3.1453,  -3.9989],
        [  2.0243,   7.3086,   7.5432],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.6675,  -4.8569,  -9.1769],
        [  0.6396,  14.0772,  -6.3786],
        [  4.4049,   5.6892,  -4.4394],
        [ -9.2707,   0.9661,   4.4863],
        [ -9.3384,  -2.6161,  -4.7630],
        [  9.2811,  -7.4336,   5.1771],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.0845,  -2.5772,  -4.6073],
        [  1.4749,  -4.7892,  -0.8089],
        [  3.6469,   6.3624,  -1.8414],
        [  1.1350,  14.0104,   0.6321],
        [  2.0419,  -4.6978,  -5.0983],
        [ -0.4341, -10.9473,   7.4509],
        [  5.2939,  -2.4784,  -4.5840],
        [ -2.6069, -10.4298,  -6.1995],
        [ -9.1795,  -3.3623,   8.5259],
        [  9.5018,  -7.1945,  -1.1892],
        [  6.3497,   1.3680,  -1.9076],
        [  3.9913,  -3.8139,  -6.8311],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -6.9838,  -7.4965,   0.9796],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.3301,   0.3638,  -7.5685],
        [  1.3339,   7.5627,  -0.9034],
        [  2.5829,  -4.5466,  -8.4015],
        [ -0.8675, -10.8815,   6.8561],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.9260,  -4.4162,   8.3808],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -9.4351,  -1.8891,   3.5642],
        [  2.3889,  -4.6082,  -8.7291],
        [ -9.3464,   0.5708,  -1.6397],
        [  3.2701,   6.6338,   9.9209],
        [  9.6662,   9.1725,   8.1056],
        [ -5.5525,   6.5146,  -4.4984],
        [ -9.1795,  -3.3623,   8.5259],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.3550,   0.7129,   0.4917],
        [  2.3889,  -4.6082,  -8.7291],
        [ -4.9217,   6.8979,  -1.7667],
        [ -6.0844,  -8.3955,   0.7705],
        [ 11.4235,   6.3594,   5.1663],
        [  5.6672,  -1.8631,   8.1020],
        [  9.4925,   9.3811,   8.9842],
        [ -6.9838,  -7.4965,   0.9796],
        [ -1.7290,   7.8825,  -3.3831],
        [  7.0358,  -9.2784,  -9.8479],
        [  6.2419,  -0.2313,   4.2440],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.2159, -10.8166,  -1.0406],
        [ -5.6651,   6.4380,   1.3660]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0180, 0.0233,  ..., 0.3465, 0.5255, 0.6126],
        [0.0180, 0.0000, 0.0057,  ..., 0.3645, 0.5078, 0.5948],
        [0.0233, 0.0057, 0.0000,  ..., 0.3697, 0.5031, 0.5893],
        ...,
        [0.3465, 0.3645, 0.3697,  ..., 0.0000, 0.8694, 0.9555],
        [0.5255, 0.5078, 0.5031,  ..., 0.8694, 0.0000, 0.2099],
        [0.6126, 0.5948, 0.5893,  ..., 0.9555, 0.2099, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.09887557,  7.15990676,  7.16747878,  7.19513806,  7.5092834 ,
        7.42512621,  7.69752262,  8.04664886, 12.19805394,  4.78002665,
        8.34949383, 13.32613445,  8.5724042 ,  6.34059021, 13.05148826,
        4.74468406,  7.15100303, 11.24434374,  7.81509631,  7.5158489 ,
        5.53255882,  6.05113217,  4.8763135 ,  5.5513072 ,  8.67227386,
       11.37939695, 10.74409834,  6.21275016,  5.0713364 ,  8.57560275,
        7.10378919,  8.53014915,  5.40689332,  6.23262289,  7.81509631,
       11.31276109, 13.3607379 , 10.35485331,  5.76165934, 10.4977132 ,
       10.46739073,  7.62011734, 12.4217523 , 13.91198191, 12.53285633,
       11.51557166,  5.9510489 ,  9.97822473, 10.82106505, 11.1341798 ,
        6.76661595,  7.58378655, 12.30695098,  9.93553424, 14.09176259,
        7.19513806,  9.32094055,  9.69792513, 11.89105259,  5.05444288,
       12.59840193, 12.35625694,  5.01114719,  7.33350642, 14.05633441,
        5.1224079 , 10.95593729,  5.84533345, 10.7506423 ,  9.77588801,
       11.9182956 ,  6.49538841,  5.5205189 , 12.04924159, 10.24557047,
       11.59512194,  6.34059021,  7.67939546,  5.22903117, 10.91601655,
       13.12299145,  5.29755487, 10.54658597,  9.62238828,  5.19065439,
        9.36378438,  7.39599986, 13.32557133,  8.55982105,  9.77588801,
       12.4013153 ,  9.7712321 ,  6.3949043 ,  5.19065439,  8.47371868,
       10.36844813, 13.0743399 ,  5.96556248, 13.34587027, 10.24557047,
        8.06990483, 11.64436684,  6.24614905, 13.25017211, 11.2845608 ,
        8.57560275])}
{'P': tensor([[1.9094e-02, 2.6026e-01, 1.8746e-01,  ..., 7.5139e-04, 7.7000e-04,
         4.8482e-06],
        [1.0773e-03, 2.7963e-02, 5.5336e-02,  ..., 1.5578e-03, 1.4132e-03,
         7.7629e-06],
        [1.7400e-03, 1.2542e-02, 4.0432e-02,  ..., 2.3511e-03, 1.6601e-03,
         1.0752e-05],
        ...,
        [5.4587e-05, 5.7264e-05, 5.7183e-05,  ..., 8.5540e-02, 2.1743e-04,
         1.7495e-03],
        [8.0987e-05, 1.0063e-04, 9.4437e-05,  ..., 1.1713e-02, 8.0369e-02,
         1.7913e-04],
        [8.6157e-05, 6.6867e-05, 5.8569e-05,  ..., 1.1929e-05, 1.2912e-05,
         9.7791e-02]]), 'X': tensor([[  4.8432,   5.1968,   1.5006],
        [  4.3377,   5.7570,   1.3712],
        [  4.2096,   5.8815,   1.9230],
        [  3.3943,   6.5483,   1.0790],
        [  3.7121,   6.3117,   0.6738],
        [  2.3200,   7.1756,   1.1938],
        [ -1.6906,   7.8860,   7.7142],
        [  7.0642,  11.6104,  -9.8316],
        [ 12.5199,  -0.4711,   3.0436],
        [  4.5800,   5.5034,  -3.9432],
        [ 11.6165,  -3.8715,   6.4594],
        [  0.9659,  -4.8144,  -0.9594],
        [ -2.8162,   7.7059,  -9.6885],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.9003,  12.3550,   2.1917],
        [ 11.9808,   4.9340,  -3.3715],
        [ -9.4773,  -1.0024,   5.2599],
        [ -9.1795,  -3.3623,   8.5259],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.8902,  -1.3887,   4.0388],
        [  2.3490,  -4.6199,  -4.2489],
        [ 10.4737,  -5.9570,  -1.2108],
        [  2.5375,   7.0676,  -4.3490],
        [  6.3569,   0.7591,   6.5011],
        [ -9.1459,   1.4780,  -4.2403],
        [  6.2905,   1.9366,   2.1169],
        [  5.1793,   4.7461,   0.3899],
        [  1.7122,  -4.7591,  -3.6734],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.2860,   7.1916,  -6.1011],
        [  4.9531,   5.0576,   5.6794],
        [  2.7590,  13.6621,   9.4496],
        [  7.3592,  11.3937,   6.2407],
        [  3.3167,   6.6022,   6.3554],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.1274,  14.0116,   8.1753],
        [ -7.3446,   4.9113,  -5.9356],
        [  9.0051,   9.9234,   5.4411],
        [  5.7845,   3.6652,   0.9175],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.5770,   1.8512,  -0.6578],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.0449,  13.8402,  -7.1642],
        [  5.9003,  12.3550,   2.1917],
        [ -1.6275, -10.7221,   3.2376],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3583,   1.1729,   8.7835],
        [ -9.4645,  -1.4880,  -2.0226],
        [  4.8653,   5.1694,  -4.0823],
        [  1.5383, -11.0274,  -4.9666],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.7963,  -1.6018,   1.8856],
        [  8.3718,  -8.2953,   8.4666],
        [  5.4732,  -2.2041,  -4.3830],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.9526,   3.2525,  -4.3821],
        [ -5.5525,   6.5146,  -4.4984],
        [  0.1931,  -4.7491,   7.4385],
        [  6.2535,  -0.1709,  -9.4102],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.3534,   1.2990,  -7.2942],
        [ -7.5720,   4.6310,   2.1717],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.1752,   2.5242,   3.5874],
        [  0.5250, -11.0306,   5.5358],
        [  2.8602,   6.8905,   5.8779],
        [  6.1984,  -0.4327,   1.5438],
        [  5.6248,   3.9984,   8.4933],
        [  6.1973,  -0.4372,   6.5720],
        [  6.3413,   0.4883,  -6.8416],
        [ -2.0721,   7.8432,   6.6380],
        [  1.7500,   7.4187,   4.4237],
        [  2.7295,  13.6703,  -8.9328],
        [  3.2990,  -4.2415,  -7.4857],
        [  4.3459,  -3.5327,   0.3667],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.8359,   4.2735,   7.2943],
        [ -1.6275, -10.7221,   3.2376],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.0277,   4.9589,   7.0175],
        [ -2.8162,   7.7059,  -9.6885],
        [ -6.3489,   5.9117,   6.5161],
        [  4.4049,   5.6892,  -4.4394],
        [  0.3038,   7.8092,   0.4164],
        [  5.6248,   3.9984,   8.4933],
        [ -1.6275, -10.7221,   3.2376],
        [ -7.8280,   4.2848,  -7.1955],
        [  7.0171,  11.6439,  -5.0178],
        [  9.5018,  -7.1945,  -1.1892],
        [  0.1677,  -4.7447,  -6.1529],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -8.9788,   2.0208,  -3.2867],
        [ -0.5709,   7.9055,   7.6089],
        [  0.9074,  -4.8139,   8.1674],
        [ -0.0410,   7.8591,   9.3457],
        [  7.0515,  11.6194,   8.4289],
        [  6.2419,  -0.2313,   4.2440],
        [  6.7792,  11.8084,   1.6251],
        [ -6.4822,  -8.0252,  -3.8983],
        [  6.3246,   1.6696,  -0.1544],
        [  6.3236,   0.3011,  -4.6668],
        [ -7.6650,   4.5092,   0.2520],
        [ -3.0004,   7.6607,  -7.2640],
        [ -3.9858,  -9.8334,  -3.7323]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0261, 0.0443,  ..., 0.6336, 0.6430, 0.9322],
        [0.0261, 0.0000, 0.0436,  ..., 0.6092, 0.6197, 0.9061],
        [0.0443, 0.0436, 0.0000,  ..., 0.6066, 0.6504, 0.9162],
        ...,
        [0.6336, 0.6092, 0.6066,  ..., 0.0000, 0.5075, 0.3731],
        [0.6430, 0.6197, 0.6504,  ..., 0.5075, 0.0000, 0.4954],
        [0.9322, 0.9061, 0.9162,  ..., 0.3731, 0.4954, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.10378919,  7.20826219,  7.23276582,  7.37576154,  7.32233733,
        7.54127425,  8.06517143, 13.59056613, 12.52875793,  7.15990676,
       12.24467041,  4.91038884,  8.20436764, 10.48840476, 13.69162795,
       12.95702647,  9.53015376,  9.77588801,  6.02789634,  6.05165248,
        5.18277858, 12.04924159,  7.5092834 ,  6.4020404 ,  9.26455795,
        6.58184474,  7.02496223,  5.05774295,  8.89458273,  7.54620849,
        7.07902036, 13.93790235, 13.56367857,  7.38844703,  6.07992623,
       14.05687815,  8.8353674 , 13.40024224,  6.84796202, 12.50996828,
       12.7125079 ,  7.13501264,  9.55593374, 13.99048094, 13.69162795,
       10.84493732,  6.02588517,  6.46560181,  9.58072286,  7.09887557,
       11.1341798 , 12.10942412,  6.01356004, 11.785566  ,  5.90035324,
        9.68323686,  6.78325151,  8.55982105,  4.75303127,  6.25586297,
        5.19577146,  6.48486171,  8.87587457, 12.59969936,  6.67121748,
       11.04313641,  7.46053281,  6.2134961 ,  6.90115901,  6.21275016,
        6.36004393,  8.11227668,  7.62232307, 13.94008948,  5.37343173,
        5.60064319, 11.9182956 ,  8.92549752, 10.84493732,  6.18668463,
        7.06169391,  8.20436764,  8.67502348,  7.19513806,  7.81509631,
        6.90115901, 10.84493732,  8.92395381, 13.59480906, 11.9182956 ,
        4.74770899,  9.93553424,  9.99145096,  9.20340031,  7.92607342,
        4.89869205,  7.85919687, 13.59170809,  6.24614905, 13.61600536,
       10.31613269,  6.54127431,  6.33075682,  8.89301516,  8.22728399,
       10.61047596])}
{'P': tensor([[3.4103e-02, 1.3337e-01, 1.3064e-01,  ..., 2.9120e-05, 9.8061e-04,
         2.4178e-04],
        [4.4711e-04, 2.3575e-02, 1.3361e-03,  ..., 4.4142e-05, 1.6703e-04,
         7.5694e-05],
        [3.7552e-04, 2.7892e-01, 1.8014e-02,  ..., 3.2361e-05, 1.3464e-04,
         5.9888e-05],
        ...,
        [1.8884e-05, 1.0696e-05, 1.0970e-05,  ..., 3.5596e-02, 4.4348e-05,
         1.8432e-04],
        [6.1333e-02, 8.9885e-02, 9.6379e-02,  ..., 4.4294e-05, 8.0013e-02,
         1.7142e-03],
        [1.0346e-01, 3.3918e-02, 3.5957e-02,  ..., 5.1482e-05, 1.4569e-01,
         1.0696e-01]]), 'X': tensor([[ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1402, 0.1363,  ..., 0.8867, 0.2298, 0.0955],
        [0.1402, 0.0000, 0.0100,  ..., 0.7917, 0.1806, 0.1722],
        [0.1363, 0.0100, 0.0000,  ..., 0.8014, 0.1717, 0.1638],
        ...,
        [0.8867, 0.7917, 0.8014,  ..., 0.0000, 0.9379, 0.9575],
        [0.2298, 0.1806, 0.1717,  ..., 0.9379, 0.0000, 0.1683],
        [0.0955, 0.1722, 0.1638,  ..., 0.9575, 0.1683, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.13501264,  7.50917551,  7.49076899,  7.62232307,  7.69752262,
        7.66377554,  4.73784222,  7.60463349,  5.14124612,  9.5642993 ,
       10.05558893,  8.68979052, 12.07818871,  7.59795445, 12.49317563,
       11.87665501, 12.59840193, 10.66631784, 13.13888841, 11.38774149,
        4.96231532,  7.09887557,  9.41970031,  7.05627461,  7.44805901,
        7.9337647 ,  7.92607342,  5.05444288,  8.01009083, 11.73397309,
       13.96987806,  9.17016857,  7.92644746,  5.77219077,  5.64500512,
       10.41909211,  5.5748624 , 11.10066126, 13.97779839,  6.39520397,
       11.10066126,  9.60412735, 10.76154256,  9.63742669, 11.33837396,
        7.71075226,  7.83534987,  6.27905292,  6.7593917 , 12.50996828,
       12.12641969, 13.3426541 , 13.59480906,  6.19707944,  6.02588517,
        9.50913849,  5.79391825,  8.40445872, 11.67875855, 11.4424816 ,
       12.87752052,  9.5014146 ,  7.57946459,  7.58668144, 11.48734978,
       11.63765273, 13.79762415,  6.23262289,  5.91222346, 12.27991991,
       13.66991027, 10.61047596,  8.859416  , 10.4977132 ,  5.48276038,
       11.92133129, 14.02526505, 11.28093507, 13.35013374,  8.31625171,
        9.04255655, 10.794755  ,  5.62005629, 10.6910768 , 10.00410625,
       10.00794468,  5.48276038,  5.83560009, 13.13888841, 10.31584102,
        7.06272256, 12.16901615,  7.32950385,  8.11227668, 13.66051884,
        4.85603886,  7.63726134, 11.70016217,  7.33350642,  8.49161154,
        9.68324553,  5.36061974,  5.26342203, 10.14044489,  6.7593917 ,
        6.24614905])}
{'P': tensor([[2.0957e-02, 1.4003e-01, 1.3211e-01,  ..., 6.7909e-04, 1.5440e-05,
         1.6890e-05],
        [4.7941e-04, 1.6584e-02, 8.1053e-02,  ..., 2.7009e-04, 1.7394e-05,
         7.1741e-06],
        [4.1091e-04, 1.6423e-03, 2.6158e-02,  ..., 5.1447e-04, 3.9392e-05,
         1.3446e-05],
        ...,
        [2.0225e-04, 2.8128e-04, 5.8031e-04,  ..., 3.9378e-02, 3.9711e-05,
         6.4166e-04],
        [5.8287e-05, 3.5499e-05, 3.1358e-05,  ..., 1.9099e-05, 1.1194e-01,
         3.7154e-07],
        [1.0403e-05, 1.4976e-05, 3.1043e-05,  ..., 1.2885e-03, 2.4310e-06,
         8.0622e-02]]), 'X': tensor([[ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0635, 0.0849,  ..., 0.6495, 0.8715, 1.1794],
        [0.0635, 0.0000, 0.0371,  ..., 0.6767, 0.8225, 1.2244],
        [0.0849, 0.0371, 0.0000,  ..., 0.6537, 0.7895, 1.2114],
        ...,
        [0.6495, 0.6767, 0.6537,  ..., 0.0000, 0.9408, 0.6197],
        [0.8715, 0.8225, 0.7895,  ..., 0.9408, 0.0000, 1.5377],
        [1.1794, 1.2244, 1.2114,  ..., 0.6197, 1.5377, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.14837514,  7.39004226,  7.52117132,  7.50564223,  7.58378655,
        7.59795445,  9.85330646, 12.77402351,  7.50035355,  6.67218687,
        8.99791316, 11.63383221, 11.83570132,  8.37954853,  7.70897675,
       14.05633441,  7.53063885, 11.20100795, 13.13205357, 13.20597932,
        6.53290663,  5.47866221,  7.5158489 ,  9.17905491,  7.32754882,
        7.56617037,  7.02253235, 10.31613269, 11.22993218, 10.86693708,
        5.52416368,  9.99145096,  5.69913817, 12.3828957 , 12.52372629,
        8.51088913, 12.85032836,  5.29009836, 11.75618513, 10.36844813,
        6.12886024, 11.76830981,  5.95111804, 12.0495521 ,  5.87305005,
        7.66377554,  7.05627461, 13.31764181,  8.06643487, 10.53368644,
        7.50917551,  5.68942572, 13.99048094,  9.58072286, 12.49317563,
       13.91198191, 10.37698435,  6.97106912, 13.76047999,  8.67502348,
        8.87007495,  8.67502348,  9.87720102, 14.05633441, 13.36800073,
        5.43544626, 11.80092578,  9.09963894,  4.85995002,  6.24187044,
       13.84126197,  6.48486171,  6.60752373,  5.27178363,  5.83560009,
        7.66377554,  6.54693987, 13.61600536,  8.47371868, 13.02818702,
        9.27474386,  4.7663526 ,  9.34061938,  5.29755487,  5.90035324,
       11.63765273, 10.84493732,  7.28793663,  7.50564223, 11.51811603,
       13.66991027,  5.0713364 ,  6.78325151,  5.77219077,  6.19707944,
       11.70168755,  7.70897675, 13.0493368 ,  7.96575782, 12.88270018,
       11.1913598 , 11.33837396,  5.60064319,  7.57946459, 10.19036081,
       13.86480175])}
{'P': tensor([[2.1197e-02, 1.9848e-01, 1.9665e-01,  ..., 4.0066e-06, 3.3030e-05,
         1.6955e-05],
        [5.6485e-04, 2.3379e-02, 6.2200e-04,  ..., 4.2144e-06, 2.5969e-05,
         2.5537e-05],
        [1.8706e-03, 2.7410e-01, 2.4306e-02,  ..., 4.5099e-06, 3.2932e-05,
         1.9160e-05],
        ...,
        [1.0625e-04, 4.7752e-05, 9.1289e-05,  ..., 2.3810e-02, 1.8353e-06,
         2.4847e-07],
        [5.8731e-05, 7.8362e-05, 5.7537e-05,  ..., 1.6393e-06, 9.6222e-02,
         1.3510e-02],
        [2.3873e-06, 2.6395e-06, 2.2865e-06,  ..., 6.2643e-06, 3.3128e-05,
         4.1044e-02]]), 'X': tensor([[  4.6231,   5.4556,   8.4973],
        [  3.3069,   6.6089,   8.7377],
        [  4.4039,   5.6903,   8.7366],
        [  2.5620,   7.0549,   8.4825],
        [  2.4572,   7.1085,   7.7376],
        [  2.1477,   7.2550,   8.9688],
        [ 12.4996,   2.6335,  -6.3166],
        [  4.0591,   6.0202,  -1.6339],
        [ -7.8849,  -6.3115,  -4.7479],
        [  5.4167, -10.1293,   6.2156],
        [ -1.6220,   7.8917,   0.0315],
        [ 11.5966,   5.9666,   6.0083],
        [  9.3753,   9.5171,   3.9131],
        [  0.3231,  -4.7691,  -1.6120],
        [ -8.6112,   2.9319,  -7.9943],
        [  8.4691,  10.4571,  -5.0650],
        [ 11.6019,   5.9540,  -5.1715],
        [ 10.5678,   7.9261,  -0.6100],
        [ 12.4901,  -0.7052,   1.7315],
        [  7.0358,  -9.2784,  -9.8479],
        [ -5.2752,  -9.0409,  -8.2568],
        [  9.6662,   9.1725,   8.1056],
        [ -7.6734,   4.4980,  -5.4248],
        [  3.9913,  -3.8139,  -6.8311],
        [ -4.0868,   7.2981,   8.1874],
        [  5.8979,   3.3958,  -3.8713],
        [ -1.1885, -10.8211,   0.0865],
        [ -9.3087,   0.7792,  -7.5587],
        [  1.3767, -11.0341,   9.9415],
        [ 11.3367,  -4.4806,   0.4643],
        [ -2.5683, -10.4433,  -2.9182],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.9470,  13.6085,   2.9809],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.0870,  -4.6877,   9.6823],
        [ -2.3317,   7.8034,  -3.6274],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -9.1624,  -3.4286,   2.0647],
        [ 11.7418,   5.6070,  -1.4192],
        [  2.3928,   7.1404,  -1.7035],
        [  4.1999,  -3.6543,  -4.1998],
        [  5.8275,  -1.5334,   2.0124],
        [  3.9022,  13.2906,   5.5788],
        [ -3.5029,   7.5138,   8.1853],
        [  5.7218,  -9.9934,   6.5764],
        [  4.9531,   5.0576,   5.6794],
        [  5.6523,   3.9443,  -2.9449],
        [  9.7328,   9.0902,  -0.8917],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.5506,  -5.1154,   6.3423],
        [ 12.1751,  -2.2592,  -5.7751],
        [  7.9403,  -8.6455,  -8.7433],
        [ -1.9487,   7.8590,  -3.3330],
        [  3.7992,  -3.9474,  -8.9019],
        [  4.2612, -10.5514,  -6.6268],
        [ -5.1369,   6.7755,   8.3836],
        [  2.6613,   7.0021,   4.5783],
        [  6.2576,   2.1359,  -3.3398],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -6.3489,   5.9117,   6.5161],
        [ -2.6735, -10.4062,  -1.5911],
        [ -9.4501,  -1.7133,   2.2980],
        [  1.5653,  13.9377,   4.1773],
        [ 12.2218,   4.0953,  -8.2072],
        [ -1.0407,   7.9166,  -6.6021],
        [  3.9085,   6.1512,  -2.2560],
        [  6.8093,  -9.4178,   8.4244],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.3505,   0.6235,  -8.5849],
        [ -3.5173, -10.0623,   4.8280],
        [  3.3094,  13.4969,  -3.6507],
        [  4.4915,   5.5990,   7.0405],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.1795,  -3.3623,   8.5259],
        [ -7.5401,   4.6718,   9.0033],
        [  8.2154,  -8.4262,   8.0171],
        [  3.9515,  -3.8425,   6.1729],
        [ -8.5506,  -5.1154,   6.3423],
        [  1.3705,  13.9723,   7.9266],
        [ -0.7599, -10.8995,  -6.0366],
        [  0.5250, -11.0306,   5.5358],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.1940,  -2.1881,   7.1499],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.8474,   6.8979,   6.6457],
        [  0.1382, -11.0073,   1.0026],
        [ -0.6323,   7.9085,  -5.9544],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.3551,   0.7149,   1.3344],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -0.5709,   7.9055,   7.6089],
        [ 11.6165,  -3.8715,   6.4594],
        [ -3.4488, -10.0934,   2.5195],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.8889,  -1.3916,   8.8993],
        [ -6.4368,   5.8356,   3.3014],
        [  5.8323,  -1.5228,   5.1132],
        [ -0.5709,   7.9055,   7.6089],
        [ -4.7642,  -9.3860,  -5.1133],
        [  7.0358,  -9.2784,  -9.8479],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.3122,  -9.6994,  -6.3282],
        [  2.1440, -10.9811,   8.5281],
        [  2.5974,   7.0363,  -7.7810],
        [ -7.8280,   4.2848,  -7.1955]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0547, 0.0157,  ..., 0.5059, 0.9540, 1.1790],
        [0.0547, 0.0000, 0.0396,  ..., 0.4513, 0.9635, 1.1616],
        [0.0157, 0.0396, 0.0000,  ..., 0.4908, 0.9600, 1.1774],
        ...,
        [0.5059, 0.4513, 0.4908,  ..., 0.0000, 1.1281, 1.0966],
        [0.9540, 0.9635, 0.9600,  ..., 1.1281, 0.0000, 0.5629],
        [1.1790, 1.1616, 1.1774,  ..., 1.0966, 0.5629, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.15100303,  7.39004226,  7.19534146,  7.50564223,  7.52117132,
        7.56617037, 12.77402351,  7.26074925, 10.09979747, 11.48663767,
        8.05668609, 13.04156075, 13.35927527,  4.78002665,  9.09661212,
       13.45642379, 13.04051744, 13.20988996, 12.50996828, 11.64436684,
       10.46739073, 13.32557133,  8.89458273,  5.5205189 ,  8.36445758,
        6.80559669, 10.88618114,  9.34127013, 11.11969815, 12.18998195,
       10.75443106, 11.9182956 , 13.92390689, 12.95702647,  5.13125513,
        8.14433542, 10.51437947,  9.78285424, 13.01187484,  7.53063885,
        5.5671389 ,  6.02588517, 13.85158417,  8.29022197, 11.51557166,
        7.07902036,  6.89245352, 13.31764181,  6.83515201,  9.96391045,
       12.3828957 , 11.73848065,  8.09704052,  5.47866221, 11.37939695,
        8.50268322,  7.49076899,  6.61211224, 12.49291768,  8.67502348,
       10.74409834,  9.60412735, 14.02532758, 12.889695  ,  7.98469166,
        7.28793663, 11.62158554, 13.18278606,  6.38105015, 10.65929968,
       13.89671314,  7.17789666, 10.74655346,  9.77588801,  8.87007495,
       11.76830981,  5.51177501,  9.96391045, 14.0393949 , 10.92597196,
       11.04313641,  5.47702766, 12.38881635,  8.23525433,  7.46248947,
       11.00812996,  7.9337647 ,  9.69629   ,  6.39520397, 10.39942497,
        7.92607342, 12.24467041, 10.66631784, 12.00995898,  6.05113217,
        8.68832398,  6.02779201,  7.92607342, 10.52587243, 11.64436684,
        5.9510489 , 11.57249984, 11.57249984, 11.18839169,  7.50035355,
        8.92395381])}
{'P': tensor([[3.3845e-02, 2.2118e-01, 1.8547e-01,  ..., 6.1297e-05, 5.6339e-05,
         2.8657e-06],
        [6.0287e-04, 3.5248e-02, 6.2134e-04,  ..., 1.7684e-05, 1.2943e-04,
         1.3388e-06],
        [6.9421e-03, 2.6999e-01, 4.1170e-02,  ..., 7.4961e-05, 7.0009e-05,
         3.4809e-06],
        ...,
        [3.1832e-04, 5.6091e-04, 3.3492e-04,  ..., 1.0208e-01, 5.3580e-06,
         2.8072e-06],
        [5.6742e-05, 3.7540e-05, 5.6397e-05,  ..., 1.8511e-06, 1.4252e-01,
         7.7209e-05],
        [1.3163e-04, 5.1969e-05, 1.3356e-04,  ..., 4.8639e-03, 4.2937e-07,
         2.7497e-02]]), 'X': tensor([[ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0786, 0.0057,  ..., 1.1273, 0.9178, 0.9439],
        [0.0786, 0.0000, 0.0767,  ..., 1.1779, 0.8477, 0.9653],
        [0.0057, 0.0767, 0.0000,  ..., 1.1245, 0.9136, 0.9392],
        ...,
        [1.1273, 1.1779, 1.1245,  ..., 0.0000, 1.6023, 0.4823],
        [0.9178, 0.8477, 0.9136,  ..., 1.6023, 0.0000, 1.1680],
        [0.9439, 0.9653, 0.9392,  ..., 0.4823, 1.1680, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.15990676,  7.5092834 ,  7.16747878,  7.60529892,  7.6374603 ,
        7.42512621, 12.17901033,  8.30729352,  6.48486171, 13.09737053,
        9.60412735, 13.01187484,  6.02789634,  8.20436764,  9.09963894,
        5.48276038, 13.34587027,  9.32094055,  9.47369628, 12.49291768,
       12.5769659 , 10.88618114, 13.92390689, 10.8992524 ,  5.83560009,
       11.04629471,  9.27474386, 12.04924159,  9.9353921 ,  4.8763135 ,
        7.59795445, 10.61472636,  9.99145096,  6.02779201,  4.91652556,
       12.62576026,  5.92281142, 11.23201619,  4.91038884, 12.0495521 ,
       13.27717982, 12.11735334,  6.35029274, 12.12448912,  9.58072286,
       13.97430191,  7.50917551, 11.10066126, 13.58101794,  5.94293185,
        6.28835101,  5.84533345,  5.5748624 ,  4.89121805,  9.3109884 ,
       10.19036081,  8.36995953, 10.66631784, 14.08549802,  9.78286733,
       12.67680439, 10.53368644, 10.06703358, 11.24434374,  4.89121805,
        7.92200342, 13.03002338,  5.05534609, 14.03505624, 12.40074901,
        8.37954853,  5.77219077,  8.44940661,  5.26067019,  8.31625171,
        5.14371005, 12.05395056,  6.35029274,  6.24614905,  5.5748624 ,
       11.64746882,  8.47371868,  5.47866221,  7.73151309,  9.17905491,
        7.19534146, 10.48315436,  8.20436764,  9.09119351, 12.44177641,
       10.42667093, 12.13568598, 12.59969936,  6.32572323,  9.43112957,
        5.83560009, 11.42420508,  8.56889738,  5.10446653,  9.71392179,
       13.94008948, 12.27835996,  9.50654967, 13.34587027,  9.88310614,
       12.13568598])}
{'P': tensor([[2.6042e-02, 1.6425e-01, 1.6087e-01,  ..., 7.2201e-05, 9.3577e-06,
         3.4568e-06],
        [2.7789e-04, 1.4616e-02, 2.2630e-01,  ..., 8.5441e-05, 3.5533e-06,
         1.1071e-06],
        [3.2504e-04, 1.0463e-03, 1.8461e-02,  ..., 1.1505e-04, 4.5470e-06,
         1.3842e-06],
        ...,
        [9.8076e-06, 7.4976e-06, 7.5194e-06,  ..., 3.1267e-02, 2.2049e-04,
         7.8741e-05],
        [4.9890e-04, 2.3797e-04, 2.3703e-04,  ..., 1.1283e-04, 2.3509e-01,
         1.0386e-02],
        [5.3852e-04, 1.5895e-04, 1.5197e-04,  ..., 1.9270e-06, 4.5587e-04,
         4.7304e-02]]), 'X': tensor([[  4.5767,   5.5071,   3.9069],
        [  2.6613,   7.0021,   4.5783],
        [  2.5382,   7.0672,   4.5097],
        [  1.7500,   7.4187,   4.4237],
        [  1.7659,   7.4127,   4.7871],
        [  1.4489,   7.5256,   4.4529],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.7976,  -4.4682,  -4.2020],
        [  8.1217,  -8.5024,  -0.6149],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.3048,   0.1469,  -9.1691],
        [ -2.6735, -10.4062,  -1.5911],
        [ 10.0995,  -6.4739,   2.0974],
        [ -4.4503,   7.1378,   3.3751],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 11.3118,   6.5957,  -4.5667],
        [  1.6308,  13.9254,  -2.0984],
        [  3.6703,   6.3443,  -5.4401],
        [  0.7685,  -4.8098,  -2.9310],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.7121,   6.9743,  -8.4202],
        [  4.9152,  12.8653,  -1.4459],
        [  3.2647, -10.8066,  -0.3768],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -7.2515,   5.0195,   4.6434],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.1624,  -3.4286,   2.0647],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.1974,   6.7396,   3.3484],
        [  0.1497,   7.8334,  -5.4412],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.4117,   0.1160,   8.2574],
        [  1.0093,   7.6567,   2.3526],
        [  2.7590,  13.6621,   9.4496],
        [ -3.0008, -10.2827,  -8.9480],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.3464,   0.5708,  -1.6397],
        [  7.8425,  -8.7202,   6.4530],
        [ -4.7847,  -9.3729,  -3.7741],
        [  4.6357,   5.4415,   7.9067],
        [ 12.0845,  -2.5772,  -4.6073],
        [  4.5325,  -3.3643,   6.7699],
        [  5.6260,  -1.9401,   3.0259],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.4196,   0.0478,   9.4140],
        [ 12.0772,  -2.6014,  -7.1832],
        [  1.1350,  14.0104,   0.6321],
        [  2.6613,   7.0021,   4.5783],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.7802,  -1.6363,  -6.2782],
        [ 10.9931,   7.2107,   7.7933],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.5149,  -5.1905,   1.5239],
        [ 12.6035,   0.7494,   7.1536],
        [  5.4867,  12.5816,   2.9549],
        [  6.0467,  -9.8369,  -6.5816],
        [  1.7000,  -4.7609,   2.7935],
        [ -7.3470,  -7.0615,   5.0066],
        [ -8.6229,   2.9068,   2.4429],
        [  0.9659,  -4.8144,  -0.9594],
        [  6.3550,   0.7129,   0.4917],
        [  2.6078, -10.9229,   6.2869],
        [  1.6232,  -4.7717,  -2.5655],
        [ -7.8359,   4.2735,   7.2943],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -8.1204,  -5.9308,   4.0761],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -1.0407,   7.9166,  -6.6021],
        [  6.3119,   1.7797,   9.9186],
        [  5.4336,  -2.2676,   5.6933],
        [ -0.5385,   7.9037,   6.3830],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3048,   0.1469,  -9.1691],
        [ -1.3940, -10.7772,   6.7835],
        [  2.1771, -10.9776,  -3.2007],
        [  6.9433,  -9.3362,  -0.3421],
        [  5.9526,   3.2525,  -4.3821],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.4429,  -2.2529,   9.7306],
        [  4.0591,   6.0202,  -1.6339],
        [ -9.0961,   1.6529,   3.1553],
        [  1.8447,  -4.7372,   8.1235],
        [ -5.3328,  -8.9992,   7.3440],
        [  5.6022,  -1.9833,   8.3081],
        [ -9.4203,  -2.0340,   2.5856],
        [ -6.0175,  -8.4541,  -9.8408],
        [  4.2275,  -3.6320,  -0.2714],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.0305,   3.0296,   0.1375],
        [  2.0870,  -4.6877,   9.6823],
        [ -7.6734,   4.4980,  -5.4248],
        [  9.5261,  -7.1675,   1.4206],
        [  5.8325,  -1.5222,  -2.1341],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.0407,   7.9166,  -6.6021],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.8631,  -4.3728,  -1.1746],
        [  5.3602,  -2.3808,   7.2098],
        [ -9.4283,  -1.9585,  -0.9129],
        [  2.6614,  -4.5192,  -9.3641],
        [  8.5093,  -8.1764,  -3.2381]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0843, 0.0915,  ..., 0.9519, 0.7332, 0.6379],
        [0.0843, 0.0000, 0.0100,  ..., 0.9116, 0.7870, 0.6705],
        [0.0915, 0.0100, 0.0000,  ..., 0.9021, 0.7863, 0.6671],
        ...,
        [0.9519, 0.9116, 0.9021,  ..., 0.0000, 0.9584, 0.7302],
        [0.7332, 0.7870, 0.7863,  ..., 0.9584, 0.0000, 0.2325],
        [0.6379, 0.6705, 0.6671,  ..., 0.7302, 0.2325, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.16058587,  7.49076899,  7.50917551,  7.62232307,  7.62011734,
        7.66377554, 10.92597196,  5.27178363, 11.75807252,  4.8763135 ,
        9.75200848,  6.30647746, 10.74409834, 11.99633819,  8.41148722,
       10.53368644, 13.09426142, 14.02058621,  7.32950385,  4.87083348,
       11.63724792,  7.48311002, 13.77222752, 11.28895489,  9.43112957,
        8.81930525,  4.78002665,  9.78285424, 10.51437947,  8.51088913,
        7.83486943, 11.04629471,  9.41244865,  7.72291946, 13.93790235,
       10.71162681, 13.79676654,  9.36378438, 11.72802491, 10.52357479,
        7.14837514, 12.35625694,  5.64464807,  5.95111804,  9.10945678,
        9.41970031, 12.35421461, 14.05633441,  7.49076899,  9.62238828,
        6.00732193, 13.14690137,  9.27474386,  9.97220799, 12.62576026,
       13.72594022, 11.54672755,  5.05534609, 10.19036081,  9.09963894,
        4.91038884,  6.3949043 , 11.22993218,  5.04028538,  8.92549752,
       10.37698435,  8.44940661, 12.49317563, 10.05558893, 10.46739073,
        7.98469166,  6.55800267,  5.88782913,  7.92200342, 12.91505111,
       11.62158554,  6.30647746, 10.86693708, 11.1913598 , 11.63502679,
        6.78325151, 12.91505111,  5.89074104,  7.26074925,  9.24501829,
        5.08372149, 10.4606265 ,  5.94293185,  9.63742669, 10.37698435,
        5.57341623, 12.12641969,  6.74872981,  5.13125513,  8.89458273,
       11.92133129,  6.02789634, 13.25336001,  9.53015376,  7.98469166,
        8.86082124,  9.88310614,  5.86519525,  9.62958973,  5.24461068,
       11.80092578])}
{'P': tensor([[2.0512e-02, 1.3571e-01, 1.2875e-01,  ..., 1.9145e-04, 6.5724e-04,
         1.8189e-05],
        [3.7124e-04, 2.2479e-02, 4.2283e-03,  ..., 5.9723e-04, 4.3722e-04,
         3.9768e-05],
        [5.0293e-04, 4.4311e-02, 2.2466e-02,  ..., 5.5847e-04, 8.0569e-04,
         3.6920e-05],
        ...,
        [4.7067e-05, 4.6458e-05, 4.8361e-05,  ..., 1.1823e-01, 3.5746e-05,
         1.8113e-01],
        [1.8441e-04, 4.2957e-04, 5.4132e-04,  ..., 1.1236e-03, 5.2996e-02,
         7.1825e-05],
        [3.5767e-05, 2.0050e-05, 2.2729e-05,  ..., 6.5612e-05, 9.6096e-06,
         7.2363e-02]]), 'X': tensor([[  4.5503,   5.5359,   7.4735],
        [  2.4572,   7.1085,   7.7376],
        [  2.8474,   6.8979,   6.6457],
        [  2.0243,   7.3086,   7.5432],
        [  2.0729,   7.2878,   7.0880],
        [  2.5620,   7.0549,   8.4825],
        [ -3.9456,  -9.8542,   0.7466],
        [  6.2770,  -0.0379,   2.4195],
        [ 12.2218,   4.0953,  -8.2072],
        [ -7.7842,   4.3466,  -1.7548],
        [  9.6587,   9.1817,  -0.8212],
        [ 12.4691,   2.8525,  -5.5110],
        [ -1.7290,   7.8825,  -3.3831],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.4336,  -2.2676,   5.6933],
        [  6.7792,  11.8084,   1.6251],
        [ 11.4235,   6.3594,   5.1663],
        [  1.5383, -11.0274,  -4.9666],
        [  6.3551,   0.7149,   1.3344],
        [  6.2804,   2.0019,   0.8607],
        [  0.8700,  -4.8132,  -7.8042],
        [  5.6022,  -1.9833,   8.3081],
        [ -8.0608,   3.9362,  -5.9567],
        [ -3.2736, -10.1703,   3.8712],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.9433,  -9.3362,  -0.3421],
        [ -6.4368,   5.8356,   3.3014],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.9587,  12.3216,   9.1695],
        [  2.2076,  13.8032,  -1.5099],
        [  6.3587,   1.1596,  -3.6629],
        [  2.6613,   7.0021,   4.5783],
        [ -9.4731,  -0.7120,  -6.0271],
        [  4.5325,  -3.3643,   6.7699],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 12.5770,   1.8512,  -0.6578],
        [  4.4915,   5.5990,   7.0405],
        [  2.8890,  -4.4316,   1.1600],
        [ 12.2692,  -1.8882,   3.9253],
        [ 10.0995,  -6.4739,   2.0974],
        [  9.7328,   9.0902,  -0.8917],
        [  6.0140,  -1.0661,  -2.5040],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.2535,  -0.1709,  -9.4102],
        [  2.3889,  -4.6082,  -8.7291],
        [ -5.7347,  -8.6908,   9.2878],
        [  0.9788,  -4.8145,  -3.8661],
        [  2.7685, -10.8982,  -6.8928],
        [ -4.3359,  -9.6431,  -9.0616],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.2247,  -0.3150,  -9.3657],
        [ -2.0229, -10.6163,   8.6930],
        [  3.3839,  -4.1966,   8.3574],
        [  6.6319,  11.9066,   3.5165],
        [ -3.6381,   7.4683,   7.3544],
        [ -8.7409,  -4.6837,  -6.5840],
        [  0.1252, -11.0062,   7.2564],
        [  2.0729,   7.2878,   7.0880],
        [ -1.9646,   7.8571,   3.3817],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.0890,  -0.8382,   2.9166],
        [  5.9526,   3.2525,  -4.3821],
        [  7.9141,  -8.6656,   9.6948],
        [  6.2883,   0.0325,   5.2692],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 11.5966,   5.9666,   6.0083],
        [  6.8704,  -9.3809,   7.5769],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.4356,  -4.5942,  -5.0808],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.1350,  14.0104,   0.6321],
        [  6.3534,   1.2990,  -7.2942],
        [  3.9895,   6.0816,  -6.5235],
        [  2.5318,  13.7231,   6.3692],
        [  3.1756, -10.8247,   6.8882],
        [  6.1685,  12.1979,  -6.2473],
        [  2.0243,   7.3086,   7.5432],
        [  5.3858,   4.4260,  -9.6769],
        [ 11.1469,   6.9237,   1.6160],
        [  1.5246,  -4.7838,  -5.2095],
        [  1.8320,   7.3871,   5.1452],
        [  4.3554,   5.7393,  -9.0766],
        [  6.3236,   0.3011,  -4.6668],
        [  0.6933,   7.7336,   2.1179],
        [  1.0000,   7.6591,  -4.5011],
        [ -0.5739,   7.9056,   0.6382],
        [ -8.5900,   2.9767,   7.4811],
        [  5.6202,  -1.9508,   0.7471],
        [  9.6662,   9.1725,   8.1056],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.1047, -11.0046,   0.5188],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.1706,   1.3863,   8.4032],
        [ -1.9487,   7.8590,  -3.3330],
        [  0.8940, -11.0402,  -0.9783],
        [  0.1047, -11.0046,   0.5188],
        [  5.6523,   3.9443,  -2.9449],
        [  6.0140,  -1.0661,  -2.5040],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.2707,   0.9661,   4.4863],
        [  1.1139,   7.6281,  -2.5680],
        [ -7.6525,  -6.6534,   5.1721]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0866, 0.0666,  ..., 0.7673, 0.6800, 0.8694],
        [0.0866, 0.0000, 0.0457,  ..., 0.6922, 0.6913, 0.7896],
        [0.0666, 0.0457, 0.0000,  ..., 0.7008, 0.6520, 0.8038],
        ...,
        [0.7673, 0.6922, 0.7008,  ..., 0.0000, 0.7878, 0.1381],
        [0.6800, 0.6913, 0.6520,  ..., 0.7878, 0.0000, 0.9252],
        [0.8694, 0.7896, 0.8038,  ..., 0.1381, 0.9252, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.16598273,  7.52117132,  7.46248947,  7.58378655,  7.57685908,
        7.50564223, 10.61472636,  6.27715105, 12.889695  ,  8.91551231,
       13.32645552, 12.79126785,  8.06990483,  8.53014915,  9.68323686,
        5.88782913, 13.61600536, 13.0743399 , 11.1341798 ,  6.39520397,
        6.59176018,  4.89121805,  5.94293185,  8.97052115, 10.68416401,
       11.69264243, 11.63502679,  8.68832398,  9.86089579,  8.89458273,
       13.68672855, 13.97857591,  6.46356806,  7.49076899,  9.4997987 ,
        5.64464807, 10.35485331, 12.7125079 ,  7.17789666,  5.29009836,
       12.41366754, 11.99633819, 13.31764181,  6.10774544,  7.9337647 ,
        6.25586297,  5.19065439, 10.412299  ,  4.91296205, 11.24434374,
       10.57302546, 11.92133129, 10.05558893,  6.23262289, 10.80728788,
        5.39098669, 13.62896593,  8.30729352,  9.91668182, 11.00694688,
        7.57685908,  8.09899874,  5.5671389 ,  6.1463896 ,  6.78325151,
       11.73567759,  6.28835101,  9.4997987 , 13.04156075, 11.62770332,
        7.9337647 ,  5.19986797,  9.88571713, 14.05633441,  6.48486171,
        7.27340834, 13.95472331, 11.28093507, 13.66896739,  7.58378655,
        6.97106912, 13.12218013,  5.0209005 ,  7.61088125,  7.20482224,
        6.33075682,  7.76457095,  7.72415048,  7.92644746,  9.09119351,
        5.94909709, 13.32557133,  4.91296205, 11.00508681,  9.9353921 ,
        9.27474386,  8.09704052, 11.07637531, 11.00508681,  6.89245352,
        6.10774544,  4.78544875,  9.62958973,  9.32094055,  7.70897675,
       10.14044489])}
{'P': tensor([[2.3457e-02, 1.5382e-01, 1.2085e-01,  ..., 2.3625e-04, 1.3631e-03,
         1.6396e-05],
        [3.3759e-04, 1.9152e-02, 1.4603e-01,  ..., 3.0604e-04, 2.0514e-04,
         7.0898e-06],
        [6.4987e-04, 3.0177e-03, 5.1080e-02,  ..., 1.2309e-03, 3.5795e-04,
         1.9786e-05],
        ...,
        [2.6195e-05, 3.7298e-05, 5.0729e-05,  ..., 8.3459e-02, 5.7756e-06,
         1.0694e-04],
        [1.0856e-03, 5.4104e-03, 9.8252e-03,  ..., 3.6109e-03, 8.9502e-02,
         1.5211e-05],
        [7.1972e-04, 3.8537e-04, 4.1973e-04,  ..., 4.0723e-04, 1.2825e-03,
         1.0014e-01]]), 'X': tensor([[  4.5430,   5.5438,  -3.8669],
        [  2.5375,   7.0676,  -4.3490],
        [  1.8719,   7.3713,  -3.7465],
        [  1.6408,   7.4591,  -4.2294],
        [  3.0876,   6.7527,  -5.3540],
        [  2.4932,   7.0903,  -5.4214],
        [  4.5503,   5.5359,   7.4735],
        [  0.1460,   7.8340,  -9.4614],
        [ -1.0012, -10.8576,   2.2867],
        [ -8.0766,  -6.0046,  -3.1873],
        [  4.5325,  -3.3643,   6.7699],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.5340,  -3.3629,   8.9652],
        [  4.4049,   5.6892,  -4.4394],
        [  6.3569,   0.7591,   6.5011],
        [ -3.2736, -10.1703,   3.8712],
        [  7.5784,  -8.9141,   5.8410],
        [  6.2242,  -0.3171,   7.4978],
        [  4.4039,   5.6903,   8.7366],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.3261,  13.7749,   5.9121],
        [  1.0000,   7.6591,  -4.5011],
        [  1.8719,   7.3713,  -3.7465],
        [ -7.5720,   4.6310,   2.1717],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.5430,   5.5438,  -3.8669],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.6112,   2.9319,  -7.9943],
        [ 12.6038,   0.7665,  -3.4439],
        [ -6.3306,   5.9272,   5.6935],
        [ -7.6719,  -6.6260,   7.1468],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.7078,   2.7182,   1.8146],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.6232,  -4.7717,  -2.5655],
        [ -8.4267,  -5.3697,   2.4172],
        [  2.2373,  -4.6506,  -1.7334],
        [  3.9663,   6.1018,  -5.4104],
        [  1.9204,   7.3518,   8.4781],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.7218,  -9.9934,   6.5764],
        [  7.9141,  -8.6656,   9.6948],
        [  1.4749,  -4.7892,  -0.8089],
        [ -8.0087,   4.0173,   9.1091],
        [  9.3624,   9.5318,   0.2779],
        [  3.9770, -10.6341,   8.0783],
        [ -2.0445,   7.8469,  -0.5857],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.9808,   4.9340,  -3.3715],
        [  6.9653,  -9.3225,  -3.4727],
        [  8.5716,  10.3596,   6.8818],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.3413,   0.4883,  -6.8416],
        [  0.6470, -11.0352,  -1.1931],
        [ 11.4235,   6.3594,   5.1663],
        [  2.5010,  -4.5736,  -0.8989],
        [ -6.4368,   5.8356,   3.3014],
        [ 11.7349,  -3.5838,   2.1644],
        [  0.3724,   7.7974,  -6.8722],
        [  3.4140,  -4.1803,  -8.8224],
        [ -9.4157,   0.0821,  -5.0025],
        [  9.5261,  -7.1675,   1.4206],
        [  6.2975,   0.0947,   5.1791],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.0277,   4.9589,   7.0175],
        [  1.5383, -11.0274,  -4.9666],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.1178,   1.5784,   2.1410],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.8363,   2.4086,  -5.6296],
        [ -8.7409,  -4.6837,  -6.5840],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.3534,   1.2990,  -7.2942],
        [ 12.0582,  -2.6640,   0.7132],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 12.4597,  -0.9136,  -4.8763],
        [  3.1064,  13.5607,   6.7605],
        [  9.0424,  -7.6779,  -7.5286],
        [ -9.1178,   1.5784,   2.1410],
        [ 12.4877,  -0.7222,   8.8347],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -1.0407,   7.9166,  -6.6021],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.8093,  -9.4178,   8.4244],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.3118,   4.5452,   4.2114],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.1905,   3.7253,   4.1175],
        [  6.1066,  -0.7799,  -4.1746],
        [  1.3705,  13.9723,   7.9266],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.8407,  11.7666,   3.0045],
        [ -0.4341, -10.9473,   7.4509],
        [  0.6757, -11.0361,  -1.3047],
        [  8.4691,  10.4571,  -5.0650],
        [  4.7544,  -3.1425,  -1.5315],
        [  6.3534,   1.2990,  -7.2942],
        [  8.2154,  -8.4262,   8.0171],
        [ -6.3489,   5.9117,   6.5161],
        [  6.7792,  11.8084,   1.6251],
        [  5.5626,  -2.0536,   8.4433]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0767, 0.1070,  ..., 0.8736, 0.5457, 0.7747],
        [0.0767, 0.0000, 0.0486,  ..., 0.8598, 0.6109, 0.8119],
        [0.1070, 0.0486, 0.0000,  ..., 0.8137, 0.6091, 0.7864],
        ...,
        [0.8736, 0.8598, 0.8137,  ..., 0.0000, 0.8479, 0.5637],
        [0.5457, 0.6109, 0.6091,  ..., 0.8479, 0.0000, 0.4030],
        [0.7747, 0.8119, 0.7864,  ..., 0.5637, 0.4030, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.16747878,  7.5092834 ,  7.60529892,  7.6374603 ,  7.42512621,
        7.5158489 ,  7.16598273,  7.83534987, 10.9036226 , 10.06407839,
        5.64464807,  5.1224079 ,  5.64500512,  7.19513806,  6.4020404 ,
       10.68416401, 11.70016217,  6.2322769 ,  7.19534146,  6.12886024,
       13.96987806,  7.72415048,  7.60529892,  8.87587457,  9.54154096,
        7.16747878, 10.33712258,  9.09661212, 12.62711252,  8.67227386,
       10.1371552 ,  8.05242591,  5.38342077,  9.12220271, 12.59840193,
        5.04028538,  9.99211663,  5.16077764,  7.27759233,  7.5984809 ,
       12.72524615,  9.61942445, 11.51557166, 11.73567759,  5.01114719,
        8.95983586, 13.3607379 , 11.35345345,  8.10886021,  9.51074763,
       12.95702647, 11.63724792, 13.44593519,  9.58072286,  6.36004393,
       11.05413545, 13.0743399 ,  5.21280156,  8.68832398, 12.26997165,
        7.80625981,  5.39723241,  9.41605684, 11.92133129,  6.29821673,
        5.39450305,  7.06169391, 11.1341798 , 11.72802491,  9.25336371,
        8.1137452 ,  9.15866607,  9.91668182,  6.02588517, 10.00410625,
       12.10222204,  6.48486171, 12.3489324 , 10.52357479, 12.49317563,
       13.91198191, 11.86239961,  9.25336371, 12.5085995 ,  9.99145096,
        7.98469166,  8.08320235, 11.62158554,  4.90505566,  6.99097013,
        9.96391045,  8.99791316,  6.1561646 , 14.0393949 ,  9.70108409,
       12.59969936, 13.6105595 , 10.95593729, 11.05672136, 13.45642379,
        5.69913817,  6.48486171, 11.76830981,  8.67502348, 13.61600536,
        5.92952538])}
{'P': tensor([[2.0750e-02, 1.6450e-01, 1.6088e-01,  ..., 3.7982e-03, 1.4666e-06,
         3.9913e-05],
        [5.3978e-04, 2.5731e-02, 1.2756e-03,  ..., 1.2940e-02, 9.6042e-07,
         3.7414e-05],
        [5.6547e-04, 2.1648e-01, 1.8173e-02,  ..., 7.7065e-03, 8.8676e-07,
         2.9902e-05],
        ...,
        [7.6855e-05, 1.2485e-04, 1.0979e-04,  ..., 5.8002e-02, 7.5341e-07,
         9.4139e-05],
        [4.7558e-04, 4.5624e-04, 5.5831e-04,  ..., 1.1885e-04, 3.2911e-01,
         2.6950e-04],
        [5.1189e-03, 3.0317e-03, 3.9109e-03,  ..., 4.8471e-04, 1.9842e-03,
         1.3380e-01]]), 'X': tensor([[  4.4915,   5.5990,   7.0405],
        [  2.8474,   6.8979,   6.6457],
        [  3.3167,   6.6022,   6.3554],
        [  2.4572,   7.1085,   7.7376],
        [  2.0729,   7.2878,   7.0880],
        [  2.3261,   7.1726,   6.2522],
        [  0.1252, -11.0062,   7.2564],
        [  8.3718,  -8.2953,   8.4666],
        [  6.2535,  -0.1709,  -9.4102],
        [ -9.3739,   0.3982,  -4.2564],
        [  6.0310,  12.2795,  -6.4528],
        [ -8.1426,   3.8048,   9.3542],
        [  5.7485,  -9.9811,  -3.2355],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -8.3539,  -5.5111,   0.3373],
        [  1.0093,   7.6567,   2.3526],
        [ -4.0409,   7.3169,  -3.5779],
        [  6.0890,  -0.8382,   2.9166],
        [  6.2905,   1.9366,   2.1169],
        [ -6.4822,  -8.0252,  -3.8983],
        [  1.6232,  -4.7717,  -2.5655],
        [ 12.3759,  -1.3943,   6.4659],
        [ -7.7239,   4.4298,   3.1216],
        [  4.9452, -10.3187,  -7.8621],
        [ -1.9600,   7.8577,   6.9121],
        [  2.6970,   6.9826,   9.7350],
        [  8.6420,  -8.0580,   6.5781],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.5375,   7.0676,  -4.3490],
        [  1.3767, -11.0341,   9.9415],
        [ -3.0644,   7.6439,  -1.4322],
        [  0.1460,   7.8340,  -9.4614],
        [  9.0051,   9.9234,   5.4411],
        [  2.1504,  -4.6726,   8.7908],
        [  0.3715,  -4.7756,   0.2312],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.4196,   0.0478,   9.4140],
        [  1.8320,   7.3871,   5.1452],
        [  4.1140,  -3.7220,  -0.1273],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -3.6381,   7.4683,   7.3544],
        [  0.9809,  -4.8145,   6.8706],
        [  5.0233,   4.9647,  -3.1788],
        [ -9.4501,  -1.7133,   2.2980],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.0305,   3.0296,   0.1375],
        [  6.3266,   0.3292,   9.3502],
        [ -3.5029,   7.5138,   8.1853],
        [  7.5309,  11.2617,   7.2656],
        [  1.4489,   7.5256,   4.4529],
        [  0.5850,  -4.7982,   9.6639],
        [ -1.2454,   7.9127,  -0.5715],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.0039,   7.3173,  -1.5885],
        [ -6.7992,   5.4991,   9.9186],
        [  5.1861,   4.7361,  -3.7312],
        [  0.2732,  -4.7619,  -1.1577],
        [ -1.3940, -10.7772,   6.7835],
        [  5.7218,  -9.9934,   6.5764],
        [ 10.4758,  -5.9539,   1.3667],
        [  4.0591,   6.0202,  -1.6339],
        [  1.8766,   7.3694,   1.1887],
        [  2.6970,   6.9826,   9.7350],
        [ 12.2435,   4.0076,   8.1257],
        [ -5.1369,   6.7755,   8.3836],
        [  8.8091,  -7.9037,   0.3631],
        [  4.7813,  -3.1139,   8.2691],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.0582,  -2.6640,   0.7132],
        [ 10.2470,   8.4054,  -0.1177],
        [ -5.1369,   6.7755,   8.3836],
        [  3.2701,   6.6338,   9.9209],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.0345,   3.0175,  -1.4424],
        [  6.9284,  11.7061,  -0.3732],
        [  4.4290,  -3.4596,   1.4181],
        [  5.0190,   4.9705,  -7.9752],
        [ -5.6423,   6.4537,  -1.2091],
        [  8.7715,  -7.9390,   5.2900],
        [  2.5010,  -4.5736,  -0.8989],
        [  1.2273,  -4.8081,   9.8129],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.1426,   3.8048,   9.3542],
        [  5.0277,   4.9589,   7.0175],
        [  4.8674,  12.8875,   6.7622],
        [ -8.9379,  -4.1655,  -2.8114],
        [  3.2647, -10.8066,  -0.3768],
        [  1.5653,  13.9377,   4.1773],
        [  6.0955,   2.8210,  -4.3321],
        [  4.5954,  13.0099,  -4.4439],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.3200,   7.1756,   1.1938],
        [  4.5800,   5.5034,  -3.9432],
        [  6.1730,  -0.5375,  -8.8861],
        [ -8.1666,   3.7653,   3.6075],
        [  9.0004,  -7.7195,   4.6945],
        [  0.3038,   7.8092,   0.4164],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4157,   0.0821,  -5.0025],
        [ -8.8154,   2.4611,   2.5198],
        [  0.9809,   7.6642,  -1.4217],
        [  6.3515,   0.6418,  -6.5823],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.5927,   0.4199,  -9.4997],
        [  4.3459,  -3.5327,   0.3667]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0656, 0.0437,  ..., 0.4252, 1.1292, 0.4690],
        [0.0656, 0.0000, 0.0299,  ..., 0.3649, 1.1429, 0.4735],
        [0.0437, 0.0299, 0.0000,  ..., 0.3815, 1.1213, 0.4545],
        ...,
        [0.4252, 0.3649, 0.3815,  ..., 0.0000, 1.1284, 0.5050],
        [1.1292, 1.1429, 1.1213,  ..., 1.1284, 0.0000, 0.6734],
        [0.4690, 0.4735, 0.4545,  ..., 0.5050, 0.6734, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.17789666,  7.46248947,  7.38844703,  7.52117132,  7.57685908,
        7.5403822 , 11.00694688, 11.785566  ,  6.25586297,  9.38232121,
       13.68063098,  8.98765644, 11.51811603, 12.07818871, 10.00794468,
        7.72291946,  8.35856487,  6.1463896 ,  6.58184474, 10.31613269,
        5.04028538, 12.45418236,  8.90405823, 11.4424816 ,  8.09842704,
        7.48538905, 11.8159254 ,  9.73579738,  7.5092834 , 11.11969815,
        8.23525433,  7.83534987, 13.40024224,  5.14371005,  4.79003361,
        8.5724042 ,  9.41970031,  7.61088125,  5.54776934,  9.47369628,
        8.30729352,  4.91336994,  7.06272256,  9.60412735,  5.27178363,
        5.84533345,  6.74872981,  6.33517349,  8.29022197, 13.54774355,
        7.66377554,  4.83371341,  8.01009083,  4.92497514,  7.58668144,
        8.74470218,  7.02326515,  4.76969364, 10.86693708, 11.51557166,
       12.0495521 ,  7.26074925,  7.60463349,  7.48538905, 12.88270018,
        8.50268322, 11.83509372,  5.70591986, 10.00794468, 12.3489324 ,
       13.25336001,  8.50268322,  7.39599986, 10.06850677,  6.74686071,
       13.60275211,  5.62005629,  7.06372172,  8.5724042 , 11.83075005,
        5.21280156,  4.96231532, 10.92597196,  8.98765644,  7.06169391,
       13.77604293,  9.86089579, 11.28895489, 14.02532758,  6.7166297 ,
       13.79762415, 11.75807252,  7.54127425,  7.15990676,  6.19633139,
        8.99276564, 11.85743425,  7.81509631,  5.13616224,  9.41605684,
        9.15253344,  7.72668467,  6.383897  ,  8.44940661, 12.59969936,
        5.60064319])}
{'P': tensor([[2.0669e-02, 3.1482e-01, 1.1227e-01,  ..., 3.6228e-04, 7.0971e-05,
         9.6545e-05],
        [1.3165e-03, 2.9867e-02, 1.9781e-01,  ..., 4.3263e-04, 8.6591e-05,
         1.2351e-04],
        [5.1213e-04, 6.2076e-04, 2.7580e-02,  ..., 1.7896e-04, 3.7794e-05,
         7.0650e-05],
        ...,
        [1.4063e-01, 1.1549e-01, 4.1892e-02,  ..., 3.5472e-02, 4.7763e-04,
         4.5311e-04],
        [1.3934e-02, 1.2295e-02, 5.6536e-03,  ..., 2.6175e-02, 1.6539e-01,
         2.9579e-03],
        [2.9270e-02, 2.4451e-02, 1.0575e-02,  ..., 7.5092e-02, 2.4048e-03,
         6.9529e-02]]), 'X': tensor([[  4.4633,   5.6287,  -0.7754],
        [  4.1348,   5.9514,  -0.8856],
        [  2.3928,   7.1404,  -1.7035],
        [  2.0039,   7.3173,  -1.5885],
        [  3.5939,   6.4029,  -1.8357],
        [  3.6469,   6.3624,  -1.8414],
        [ -7.3470,  -7.0615,   5.0066],
        [  3.9770, -10.6341,   8.0783],
        [  4.4633,   5.6287,  -0.7754],
        [  4.1140,  -3.7220,  -0.1273],
        [  2.5375,   7.0676,  -4.3490],
        [  2.7685, -10.8982,  -6.8928],
        [ -1.8367,   7.8718,  -8.6523],
        [  5.0348,  -2.8229,   3.5436],
        [  6.3236,   0.3011,  -4.6668],
        [ -7.5401,   4.6718,   9.0033],
        [  6.7792,  11.8084,   1.6251],
        [ -8.1905,   3.7253,   4.1175],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.6321,  -4.5296,  -2.5159],
        [  6.8093,  -9.4178,   8.4244],
        [  2.4697,  -4.5836,   1.4593],
        [ -0.0254, -10.9932,   3.2200],
        [  6.2804,   2.0019,   0.8607],
        [  7.8425,  -8.7202,   6.4530],
        [ 10.0995,  -6.4739,   2.0974],
        [ -3.2736, -10.1703,   3.8712],
        [  7.3592,  11.3937,   6.2407],
        [ -0.8569,   7.9156,  -6.5557],
        [ 10.1798,   8.5002,  -5.8893],
        [ -7.0948,   5.1940,   9.8171],
        [ 11.0438,   7.1180,  -5.6991],
        [  7.4132,  11.3526,   0.3322],
        [  4.5954,  13.0099,  -4.4439],
        [ -8.0330,  -6.0765,   6.7945],
        [  9.3836,   9.5075,  -5.9278],
        [  1.3403,  -4.8011,  -1.1698],
        [ 11.5935,   5.9740,  -4.3952],
        [ -8.9379,  -4.1655,  -2.8114],
        [  5.9655,   3.2172,  -7.5953],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.4157,   0.0821,  -5.0025],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.6801,   2.7809,  -0.8443],
        [  2.4572,   7.1085,   7.7376],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.1440, -10.9811,   8.5281],
        [  4.5767,   5.5071,   3.9069],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -1.6906,   7.8860,   7.7142],
        [  6.3301,   0.3638,  -7.5685],
        [  2.5663,   7.0526,  -6.8769],
        [ -9.2707,   0.9661,   4.4863],
        [ -5.0550,   6.8231,  -9.2502],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -4.9217,   6.8979,  -1.7667],
        [  8.1528,  10.7452,   9.1493],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.4844,  -8.0230,   8.5209],
        [  9.1620,  -7.5573,  -9.9882],
        [  6.3122,  -9.6994,  -6.3282],
        [ -0.1963,   7.8765,  -1.1371],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.3587,   1.1596,  -3.6629],
        [  1.4489,   7.5256,   4.4529],
        [  6.2902,   0.0451,   3.7116],
        [  2.2182,  13.8007,   5.1889],
        [  4.5503,   5.5359,   7.4735],
        [ -7.2986,  -7.1225,  -3.0814],
        [  3.3744,   6.5623,  -6.3636],
        [ -8.4573,   3.2448,   5.4370],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.4013,  -1.2597,  -7.6474],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3236,   0.3011,  -4.6668],
        [  4.5767,   5.5071,   3.9069],
        [ -1.0012, -10.8576,   2.2867],
        [  4.1537,   5.9340,  -6.0423],
        [  2.6613,   7.0021,   4.5783],
        [ 12.5927,   0.4199,  -9.4997],
        [  1.3403,  -4.8011,  -1.1698],
        [ -8.8058,   2.4850,  -1.7425],
        [ -6.7643,  -7.7372,  -9.6663],
        [  0.5175,  14.0910,   4.6540],
        [  1.7218,  -4.7576,  -3.3617],
        [  1.3339,   7.5627,  -0.9034],
        [  4.3377,   5.7570,   1.3712],
        [  6.2675,  12.1379,  -8.9720],
        [  6.0955,   2.8210,  -4.3321],
        [  5.6093,  -1.9705,   6.4807],
        [ -7.4887,   4.7364,   8.9074],
        [  8.7131,  10.2216,   7.7915],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.4179,  -2.2925,  -3.4641],
        [  5.4167, -10.1293,   6.2156],
        [  8.3718,  -8.2953,   8.4666],
        [  2.0937,  13.8293,  -2.3533],
        [  2.7685, -10.8982,  -6.8928],
        [  6.1730,  -0.5375,  -8.8861],
        [  9.3560,   9.5391,   9.4943],
        [ -8.0965,  -5.9712,   4.3466],
        [ -8.3653,   3.4184,   9.3454],
        [  0.3493,  -4.7727,  -6.5682],
        [  6.0305,   3.0296,   0.1375],
        [  6.1973,  -0.4372,   6.5720],
        [  5.8325,  -1.5222,  -2.1341]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0194, 0.0936,  ..., 0.0989, 0.5150, 0.1122],
        [0.0194, 0.0000, 0.0753,  ..., 0.1166, 0.5270, 0.1140],
        [0.0936, 0.0753, 0.0000,  ..., 0.1919, 0.5939, 0.1230],
        ...,
        [0.0989, 0.1166, 0.1919,  ..., 0.0000, 0.4281, 0.1797],
        [0.5150, 0.5270, 0.5939,  ..., 0.4281, 0.0000, 0.6077],
        [0.1122, 0.1140, 0.1230,  ..., 0.1797, 0.6077, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.18354885,  7.24677664,  7.53063885,  7.58668144,  7.34250948,
        7.33350642, 10.19036081, 11.35345345,  7.18354885,  5.54776934,
        7.5092834 , 11.24434374,  8.08320235,  5.77219077,  6.33075682,
        8.87007495, 13.61600536,  8.99791316,  9.75075273,  5.23878371,
       11.62158554,  5.20659892, 10.99326577,  6.59176018, 11.72802491,
       11.99633819, 10.68416401, 13.56367857,  7.96181288, 13.26209545,
        8.79284466, 13.13888841, 13.55868684, 13.79762415, 10.07238463,
       13.35832725,  4.98461983, 13.04217487,  9.86089579,  6.77777606,
       13.86480175,  9.41605684, 10.48840476,  9.11473391,  7.52117132,
       12.4217523 , 11.18839169,  7.16058587, 10.39942497,  8.06517143,
        6.34059021,  7.50500064,  9.32094055,  8.49161154,  9.79039371,
        8.47371868, 13.48809595, 10.01804223, 10.31584102, 11.87665501,
       11.57249984,  7.8788965 , 12.18045978,  6.46356806,  7.66377554,
        6.29034958, 13.97779839,  7.16598273, 10.1979669 ,  7.37902766,
        9.05843201,  8.41171369, 12.46513646,  5.5366698 ,  6.33075682,
        7.16058587, 10.9036226 ,  7.24326992,  7.49076899, 12.59969936,
        4.98461983,  9.14973319, 10.27716434, 14.10045797,  5.05963217,
        7.67939546,  7.20826219, 13.66051884,  6.7166297 ,  5.94535529,
        8.86082124, 13.43127131, 10.38897563,  5.8828986 , 11.48663767,
       11.785566  , 13.98691426, 11.24434374,  6.19633139, 13.3614598 ,
       10.06023665,  9.0368458 ,  4.78544875,  6.74872981,  6.21275016,
        6.02789634])}
{'P': tensor([[2.0680e-02, 1.6010e-01, 1.3193e-01,  ..., 1.3274e-03, 4.7763e-06,
         5.4650e-06],
        [3.9261e-04, 2.1456e-02, 2.4762e-03,  ..., 3.3166e-04, 2.8577e-06,
         2.2224e-06],
        [7.1132e-04, 7.3041e-02, 2.2907e-02,  ..., 5.7358e-04, 3.3765e-06,
         2.4388e-06],
        ...,
        [7.3347e-03, 1.3599e-02, 2.2674e-02,  ..., 3.6392e-02, 8.4093e-06,
         3.6698e-06],
        [1.1168e-03, 3.4378e-04, 4.4763e-04,  ..., 7.7719e-04, 2.3390e-02,
         1.7624e-03],
        [2.8882e-04, 2.0761e-04, 1.5904e-04,  ..., 3.2757e-05, 5.1591e-06,
         2.3771e-02]]), 'X': tensor([[ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0659, 0.0659,  ..., 0.1656, 0.0842, 0.9805],
        [0.0659, 0.0000, 0.0538,  ..., 0.1797, 0.0329, 0.9931],
        [0.0659, 0.0538, 0.0000,  ..., 0.1259, 0.0395, 1.0366],
        ...,
        [0.1656, 0.1797, 0.1259,  ..., 0.0000, 0.1584, 1.1437],
        [0.0842, 0.0329, 0.0395,  ..., 0.1584, 0.0000, 1.0256],
        [0.9805, 0.9931, 1.0366,  ..., 1.1437, 1.0256, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.19513806,  7.5092834 ,  7.42512621,  7.5158489 ,  7.60529892,
        7.6374603 ,  8.81930525,  7.62011734,  5.19577146,  7.73692436,
        9.48337063,  6.53250937,  7.48311002,  7.5984809 ,  9.25336371,
        5.39723241, 12.17901033, 12.24467041, 13.27994304,  9.62238828,
       13.91198191,  9.61942445,  4.90505566,  5.35774573,  8.1303945 ,
        8.14433542, 11.87412439,  9.23178204, 11.75807252,  6.31576264,
        9.78147542, 11.54672755,  4.91296205, 12.78327037,  4.83371341,
       12.35421461, 10.61417181,  7.39599986,  5.29009836, 10.36844813,
       13.84126197,  7.4544866 ,  8.29640613,  9.38232121,  8.31625171,
        5.64464807, 12.27835996,  7.73151309, 13.59056613, 11.28093507,
        8.06643487, 12.70489563,  7.53884104,  5.94541803,  5.39450305,
        5.56809335,  7.6374603 , 12.86648771, 10.99326577,  9.34127013,
        5.52416368, 10.74409834,  5.05534609,  4.91296205, 12.85032836,
        6.28835101, 12.91143491,  7.39004226, 11.64746882,  9.43737523,
        8.16714836, 13.20597932,  9.53015376, 13.31764181, 11.4890861 ,
       13.59480906, 10.1371552 ,  7.48311002, 13.1399181 ,  5.87132718,
        7.15990676, 10.1371552 ,  9.94173708, 10.87409005,  7.62011734,
        7.61088125,  9.94173708,  8.92395381,  7.32950385,  8.05242591,
       10.31613269,  9.51074763,  8.62260327, 12.33167382, 12.58006038,
        4.73784222, 11.32149259, 13.03002338, 12.18045978,  7.06272256,
        9.9353921 ,  7.76457095,  7.50500064,  7.05627461, 11.97957425,
       12.62576026])}
{'P': tensor([[1.9036e-02, 2.1467e-01, 1.4605e-01,  ..., 4.4325e-02, 3.4929e-06,
         4.2849e-04],
        [4.6053e-04, 1.7310e-02, 2.3155e-01,  ..., 7.9338e-02, 3.8147e-06,
         6.6640e-04],
        [4.6230e-04, 8.6217e-04, 2.4694e-02,  ..., 1.5002e-01, 6.5665e-06,
         1.4174e-03],
        ...,
        [1.8718e-03, 2.9812e-03, 4.2891e-03,  ..., 2.8852e-01, 1.1984e-04,
         2.8831e-02],
        [2.9948e-05, 1.9128e-05, 1.5644e-05,  ..., 7.3160e-06, 6.5711e-02,
         8.7608e-06],
        [1.8302e-05, 2.1716e-05, 2.5831e-05,  ..., 4.3342e-05, 9.3846e-04,
         4.0504e-02]]), 'X': tensor([[ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0396, 0.0709,  ..., 0.1807, 0.9156, 0.6663],
        [0.0396, 0.0000, 0.0317,  ..., 0.1411, 0.8860, 0.6378],
        [0.0709, 0.0317, 0.0000,  ..., 0.1104, 0.8584, 0.6111],
        ...,
        [0.1807, 0.1411, 0.1104,  ..., 0.0000, 0.7869, 0.5468],
        [0.9156, 0.8860, 0.8584,  ..., 0.7869, 0.0000, 0.2507],
        [0.6663, 0.6378, 0.6111,  ..., 0.5468, 0.2507, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.19534146,  7.39004226,  7.50564223,  7.56617037,  7.48538905,
        7.5984809 , 14.03618649, 13.93790235, 10.33712258, 10.99326577,
        5.56809335, 13.6105595 , 12.07225469,  9.38232121,  5.94909709,
        5.79562691,  6.39520397,  7.92200342,  6.67108662, 11.63502679,
       11.86239961,  8.41171369,  6.02588517,  6.89245352,  6.35029274,
        7.4544866 ,  5.39723241, 12.3344536 , 11.70168755, 13.06957425,
        4.7663526 , 13.08414689, 13.68063098,  6.24187044,  4.91038884,
        4.8763135 , 13.96905756, 13.34587027,  9.97964013, 13.46553623,
        8.92098752, 11.018084  ,  8.66273171,  9.97964013,  5.05444288,
        5.60064319,  5.05774295, 13.00380624,  6.25586297,  5.37343173,
        8.8580671 , 12.45418236, 10.86693708, 11.72802491, 12.4217523 ,
        7.37576154, 12.04924159,  9.17595899, 11.27160035,  9.41244865,
        7.06372172,  7.6374603 ,  9.09119351,  5.63824971,  5.92952538,
        6.7166297 , 11.57249984, 13.86480175, 14.02526505,  6.69389431,
        9.61942445, 13.89264822,  5.80860165,  6.84796202, 10.32715715,
       12.25806559,  8.51928531, 11.24434374,  6.19633139,  5.95111804,
        5.26342203,  9.45995955,  9.15253344,  6.24187044,  7.48538905,
        8.09899874, 13.66896739,  5.43544626,  9.71392179,  7.19534146,
        5.91408504, 14.03505624, 10.04916678,  6.81084724, 12.53285633,
       11.52733581, 12.97370371,  4.78544875, 12.25806559,  7.73692436,
        4.8763135 ,  6.67218687, 11.4890861 ,  7.85919687, 10.61472636,
        8.85939626])}
{'P': tensor([[2.6158e-02, 2.1588e-01, 1.8935e-01,  ..., 2.1418e-06, 1.3623e-04,
         1.3168e-05],
        [8.3694e-04, 3.4582e-02, 3.6593e-02,  ..., 2.1469e-06, 4.9746e-05,
         1.0728e-05],
        [1.6311e-03, 2.5022e-02, 7.2556e-02,  ..., 6.5967e-06, 1.0746e-04,
         3.5095e-05],
        ...,
        [3.2092e-05, 1.9412e-05, 2.9095e-05,  ..., 1.0443e-01, 1.8135e-04,
         1.2183e-04],
        [2.9765e-03, 3.7916e-03, 6.2037e-03,  ..., 3.9247e-06, 4.5506e-02,
         2.0911e-03],
        [3.3933e-06, 5.4511e-06, 9.1547e-06,  ..., 6.9771e-06, 2.0183e-06,
         4.1510e-02]]), 'X': tensor([[ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.9333e-02, 7.9302e-02,  ..., 9.6574e-01, 4.2354e-01,
         1.2788e+00],
        [7.9333e-02, 0.0000e+00, 2.9652e-02,  ..., 9.9298e-01, 4.9858e-01,
         1.3209e+00],
        [7.9302e-02, 2.9652e-02, 0.0000e+00,  ..., 9.6393e-01, 4.8527e-01,
         1.2936e+00],
        ...,
        [9.6574e-01, 9.9298e-01, 9.6393e-01,  ..., 0.0000e+00, 7.7659e-01,
         3.9843e-01],
        [4.2354e-01, 4.9858e-01, 4.8527e-01,  ..., 7.7659e-01, 2.1453e-04,
         9.8504e-01],
        [1.2788e+00, 1.3209e+00, 1.2936e+00,  ..., 3.9843e-01, 9.8504e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.20482224,  7.4544866 ,  7.48311002,  7.53884104,  7.50035355,
        7.83534987,  5.70591986,  5.80860165,  8.8737556 , 11.57249984,
       10.60226614, 11.83570132, 12.44177641, 13.99877413,  6.3949043 ,
        9.50654967,  8.95133961, 11.66453059,  8.0550286 ,  6.2322769 ,
        9.79814997,  5.63824971, 14.02532758,  5.68942572, 12.1460931 ,
        5.5513072 , 11.0050379 ,  9.68323686, 11.62158554,  8.62329466,
        7.71075226,  9.19837792,  6.19633139,  7.72415048, 10.27572673,
        5.24137909, 12.4217523 ,  9.4997987 , 11.54672755, 12.1460931 ,
       11.83570132, 11.11969815, 12.91143491, 10.06703358,  7.50035355,
        9.6282763 , 12.58006038,  7.18354885, 12.05300943, 10.31886066,
        9.34127013,  5.81481044, 12.91143491, 14.02532758, 11.33837396,
       11.57249984,  7.37902766,  8.87587457, 13.14690137,  9.78286733,
        5.16611782,  6.4020404 ,  5.5205189 ,  6.7166297 ,  8.50268322,
       13.12218013,  9.88310614, 13.61801526,  7.05627461,  5.87132718,
       14.09176259,  5.21280156, 13.96905756,  5.16077764, 10.1371552 ,
       10.48315436,  5.96556248, 11.32149259,  8.09899874,  9.5642993 ,
       13.32374056,  7.19513806,  8.1303945 ,  5.34250225,  4.89869205,
       11.85743425,  8.92549752,  6.00732193,  9.62958973, 11.4088043 ,
       13.70893492,  9.36378438, 11.52733581,  6.42439344,  6.41202986,
       12.87752052,  8.8580671 ,  6.39438044,  7.81509631,  9.27474386,
        6.12886024,  9.78285424, 14.09176259,  5.14371005, 13.18278606,
       13.93790235])}
{'P': tensor([[2.3812e-02, 2.6620e-01, 1.7292e-01,  ..., 1.0500e-03, 6.5697e-05,
         3.2485e-04],
        [8.3188e-04, 2.6971e-02, 3.4899e-01,  ..., 1.5707e-03, 5.4863e-05,
         2.2796e-04],
        [5.1150e-04, 8.0366e-04, 2.9750e-02,  ..., 1.6941e-03, 4.2943e-05,
         1.2033e-04],
        ...,
        [6.7820e-04, 1.0704e-03, 1.6299e-03,  ..., 4.5336e-01, 5.8080e-04,
         2.4470e-04],
        [5.1370e-03, 4.5959e-03, 3.2721e-03,  ..., 3.0183e-03, 8.3871e-02,
         6.2382e-03],
        [1.9802e-03, 4.3492e-03, 7.5197e-03,  ..., 1.4725e-01, 1.3214e-04,
         3.1612e-01]]), 'X': tensor([[  4.3377,   5.7570,   1.3712],
        [  3.3943,   6.5483,   1.0790],
        [  2.3200,   7.1756,   1.1938],
        [  3.7121,   6.3117,   0.6738],
        [  1.8766,   7.3694,   1.1887],
        [  1.1994,   7.6035,   2.4295],
        [  5.2608,  12.6977,   4.9166],
        [  1.0093,   7.6567,   2.3526],
        [ -4.6393,   7.0461,   1.1624],
        [ 11.7726,  -3.4876,  -6.7402],
        [  4.0457,  -3.7737,  -0.4704],
        [ -0.5709,   7.9055,   7.6089],
        [  5.2188,  -2.5839,   2.1833],
        [  6.3303,   1.6145,  -8.4083],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.0345,   3.0175,  -1.4424],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 12.2589,  -1.9312,   9.8657],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.1637,  13.1898,  -1.9189],
        [  0.1287, -11.0065,   6.5757],
        [  5.8450,  -9.9356,   3.0767],
        [ 11.9641,  -2.9568,   7.7211],
        [ -2.1382,   7.8339,   7.3524],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.0373,   7.1299,  -8.5895],
        [ 11.0438,   7.1180,  -5.6991],
        [  3.0876,   6.7527,  -5.3540],
        [  8.7715,  -7.9390,   5.2900],
        [  9.5261,  -7.1675,   1.4206],
        [  1.7000,  -4.7609,   2.7935],
        [  4.8432,   5.1968,   1.5006],
        [ -9.3444,   0.5821,   6.9101],
        [  6.3124,   0.2057,  -0.4593],
        [  3.9085,   6.1512,  -2.2560],
        [  4.5430,   5.5438,  -3.8669],
        [ -2.1533, -10.5778,   1.0112],
        [  2.4932,   7.0903,  -5.4214],
        [  4.6992,   5.3690,   3.0935],
        [ -6.4844,  -8.0230,   8.5209],
        [ -4.2271,   7.2387,  -1.8316],
        [ -3.2736, -10.1703,   3.8712],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.7685,  -4.8098,  -2.9310],
        [ -0.8455, -10.8853,  -6.7400],
        [ -6.7745,  -7.7264,   8.4684],
        [ -0.1626, -10.9796,   6.4629],
        [  5.2911,  -2.4824,  -3.6877],
        [ -8.6603,   2.8250,   5.9181],
        [  6.0054,  -1.0902,   8.3079],
        [  5.5240,  -2.1198,   6.2055],
        [  6.2790,  -0.0259,  -5.4116],
        [  7.0515,  11.6194,   8.4289],
        [ -9.4630,  -1.5145,   0.2983],
        [  6.3516,   0.6425,   0.3425],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.3810,  -2.3455,  -9.5194],
        [  7.3592,  11.3937,   6.2407],
        [ -0.7599, -10.8995,  -6.0366],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -9.4773,  -1.0024,   5.2599],
        [ 11.1420,   6.9331,  -2.4779],
        [ -3.7964,  -9.9294,   7.7147],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.6050,   2.9451,  -0.7012],
        [  2.6451,  -4.5250,  -0.5565],
        [  7.7794,  11.0629,   8.9069],
        [ -9.3076,   0.7852,  -6.7010],
        [ 11.6801,  -3.7196,  -0.9181],
        [  3.1271, -10.8343,   2.6442],
        [  5.1787,  -2.6383,  -4.2533],
        [ -0.1626, -10.9796,   6.4629],
        [ 11.9936,  -2.8677,   6.0614],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.9883,   3.1536,   6.2320],
        [  4.9531,   5.0576,   5.6794],
        [  7.8375,  -8.7240,  -9.8913],
        [  0.9659,  -4.8144,  -0.9594],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.4766,  -1.1116,  -2.1706],
        [  9.3836,   9.5075,  -5.9278],
        [  6.3122,  -9.6994,  -6.3282],
        [  8.1217,  -8.5024,  -0.6149],
        [  3.4140,  -4.1803,  -8.8224],
        [ -5.6169,  -8.7844,   8.9057],
        [  3.7237,  -3.9966,  -5.0376],
        [  0.1206,  -4.7363,   8.7103],
        [  5.9883,   3.1536,   6.2320],
        [  6.0047,   3.1062,  -6.7914],
        [ -7.5401,   4.6718,   9.0033],
        [ -4.7642,  -9.3860,  -5.1133],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.0890,  -0.8382,   2.9166],
        [  2.2849, -10.9655,  -7.6542],
        [  5.1472,  -2.6801,   8.1194],
        [ -2.0840,   7.8415,  -7.8657],
        [  6.1579,  -0.5961,  -6.5595],
        [  6.9956,  11.6590,  -7.6876]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0441, 0.0786,  ..., 0.6170, 0.4746, 0.7407],
        [0.0441, 0.0000, 0.0376,  ..., 0.5803, 0.4604, 0.7386],
        [0.0786, 0.0376, 0.0000,  ..., 0.5665, 0.4719, 0.7586],
        ...,
        [0.6170, 0.5803, 0.5665,  ..., 0.0000, 0.3641, 0.6156],
        [0.4746, 0.4604, 0.4719,  ..., 0.3641, 0.0000, 0.3182],
        [0.7407, 0.7386, 0.7586,  ..., 0.6156, 0.3182, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.20826219,  7.37576154,  7.54127425,  7.32233733,  7.60463349,
        7.69752262, 13.744387  ,  7.72291946,  8.43626114, 12.27835996,
        5.53255882,  7.92607342,  5.82345867,  6.53290663,  4.8763135 ,
        6.74686071, 12.53285633, 12.41012046, 11.9120321 ,  9.38232121,
       13.83139024, 11.00726483, 11.52733581, 12.32408516,  8.12043983,
       12.42554808, 13.04217487, 13.1399181 , 13.13888841,  7.42512621,
       11.83075005, 11.92133129,  5.05534609,  7.10378919,  9.36256096,
        6.31576264,  7.28793663,  7.16747878, 10.794755  ,  7.5158489 ,
        7.13501264, 10.31584102,  8.38252937, 10.68416401, 11.33837396,
        6.54615887, 13.06498554,  4.87083348, 10.91805683, 10.27572673,
       10.98076873,  5.84450127,  9.10945678,  6.10360073,  5.91677188,
        6.27905292, 13.59170809,  9.58347456,  6.38399789, 12.0495521 ,
        9.66977802, 13.56367857, 10.92597196, 10.31613269,  9.69629   ,
       12.53285633,  9.53015376, 13.12299145, 10.63038565,  8.40445872,
        9.09502385,  5.24137909, 13.52430176,  9.34061938, 12.25806559,
       11.27656488,  5.81199422, 10.98076873, 12.33167382,  9.77777346,
        6.76791483,  7.07902036, 11.72749923,  4.91038884,  5.68942572,
        9.562174  ,  9.54154096, 13.35832725, 11.57249984, 11.75807252,
        5.39723241, 10.42667093,  5.46244988,  4.73784222,  6.76791483,
        6.76056625,  8.87007495, 10.52587243, 13.50484539,  9.87720102,
        6.1463896 , 11.20100795,  5.80312949,  8.1137452 ,  6.18668463,
       13.59673406])}
{'P': tensor([[2.9055e-02, 1.9185e-01, 1.7608e-01,  ..., 7.0649e-05, 1.2856e-03,
         5.7086e-05],
        [5.8007e-04, 2.8940e-02, 7.8178e-04,  ..., 4.6160e-05, 8.7034e-04,
         4.8156e-05],
        [1.3374e-03, 3.2647e-01, 2.5231e-02,  ..., 4.6590e-05, 1.2597e-03,
         4.1164e-05],
        ...,
        [4.6813e-03, 1.8057e-03, 2.6570e-03,  ..., 7.6375e-02, 1.5384e-04,
         2.3518e-03],
        [6.7865e-04, 2.2261e-03, 1.4111e-03,  ..., 1.7431e-05, 5.9769e-02,
         1.2148e-05],
        [1.0226e-03, 4.6885e-04, 6.2640e-04,  ..., 1.0974e-01, 4.4757e-05,
         9.6947e-02]]), 'X': tensor([[  4.2096,   5.8815,   1.9230],
        [  2.3200,   7.1756,   1.1938],
        [  3.3943,   6.5483,   1.0790],
        [  1.8766,   7.3694,   1.1887],
        [  1.1994,   7.6035,   2.4295],
        [  1.0093,   7.6567,   2.3526],
        [ -7.6293,  -6.6859,  -5.5161],
        [  2.5375,   7.0676,  -4.3490],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.3464,   0.5708,  -1.6397],
        [  1.9312,  13.8649,   8.3563],
        [ -7.8849,  -6.3115,  -4.7479],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.9808,   4.9340,  -3.3715],
        [ -9.3604,   0.4855,  -3.0158],
        [  4.5767,   5.5071,   3.9069],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.1178,   1.5784,   2.1410],
        [ -4.2271,   7.2387,  -1.8316],
        [  1.7301,   7.4262,   5.0522],
        [ 10.4758,  -5.9539,   1.3667],
        [ -4.4503,   7.1378,   3.3751],
        [  6.3119,   1.7797,   9.9186],
        [ -2.5161,   7.7699,  -9.3531],
        [ 12.1940,  -2.1881,   7.1499],
        [  1.8719,   7.3713,  -3.7465],
        [  5.6248,   3.9984,   8.4933],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -1.6275, -10.7221,   3.2376],
        [  2.5813,  -4.5471,   5.3597],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.3653,   3.4184,   9.3454],
        [  1.5246,  -4.7838,  -5.2095],
        [ 11.5568,   6.0602,   9.3244],
        [ 10.1798,   8.5002,  -5.8893],
        [  4.6357,   5.4415,   7.9067],
        [ -7.5401,   4.6718,   9.0033],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.2883,   0.0325,   5.2692],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.2905,   1.9366,   2.1169],
        [  9.3607,   9.5337,  -4.5412],
        [  2.5318,  13.7231,   6.3692],
        [ -3.6381,   7.4683,   7.3544],
        [  0.4996,   7.7738,  -4.4711],
        [  0.5600, -11.0321,  -0.8660],
        [  6.3200,   0.2690,   1.5900],
        [ -9.0574,   1.7801,   8.8986],
        [  0.7685,  -4.8098,  -2.9310],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.3609,   3.4769,  -8.5762],
        [ -8.1426,   3.8048,   9.3542],
        [ -3.5173, -10.0623,   4.8280],
        [ -8.5859,  -5.0393,   0.5476],
        [ -3.9456,  -9.8542,   0.7466],
        [ -1.7009,   7.8851,  -8.2437],
        [ -3.5173,   7.5091,  -8.9585],
        [  3.4009,  -4.1874,  -5.9751],
        [  1.6408,   7.4591,  -4.2294],
        [  6.3551,   0.7149,   1.3344],
        [  3.8182,  -3.9347,  -1.7961],
        [ -5.6169,  -8.7844,   8.9057],
        [ -0.5385,   7.9037,   6.3830],
        [  2.3367,   7.1676,  -9.5263],
        [ 11.3367,  -4.4806,   0.4643],
        [  0.1252, -11.0062,   7.2564],
        [  5.4179,  -2.2925,  -3.4641],
        [  0.9809,  -4.8145,   6.8706],
        [  3.3474,  -4.2162,   5.2455],
        [ 10.0567,   8.6697,   4.6329],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.9022,  13.2906,   5.5788],
        [  7.1441,  11.5529,   2.3232],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.1196,  -2.7160,   4.2705],
        [  0.1287, -11.0065,   6.5757],
        [ -7.2515,   5.0195,   4.6434],
        [  4.5814, -10.4485,   2.5496],
        [  3.3167,   6.6022,   6.3554],
        [  0.3715,  -4.7756,   0.2312],
        [  6.2905,   1.9366,   2.1169],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 10.4737,  -5.9570,  -1.2108],
        [  6.1752,   2.5242,   3.5874],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.3087,   1.8053,  -1.9481],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.0937,  13.8293,  -2.3533],
        [  6.3266,   0.3292,   9.3502],
        [ -9.4756,  -0.8166,  -9.6021],
        [  5.4336,  -2.2676,   5.6933],
        [  6.9956,  11.6590,  -7.6876],
        [  2.1504,  -4.6726,   8.7908],
        [  4.6992,   5.3690,   3.0935],
        [  6.0955,   2.8210,  -4.3321],
        [  2.2849, -10.9655,  -7.6542],
        [  4.3554,   5.7393,  -9.0766],
        [ -1.0407,   7.9166,  -6.6021],
        [  5.9883,   3.1536,   6.2320],
        [ 10.8218,  -5.4177,   6.8573],
        [  8.5198,  10.4091,  -8.6486],
        [  4.2339,  -3.6268,   6.3125],
        [  2.5663,   7.0526,  -6.8769],
        [  0.9074,  -4.8139,   8.1674]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0904, 0.0697,  ..., 0.3030, 0.5410, 0.3579],
        [0.0904, 0.0000, 0.0376,  ..., 0.3830, 0.4724, 0.4283],
        [0.0697, 0.0376, 0.0000,  ..., 0.3721, 0.4739, 0.4240],
        ...,
        [0.3030, 0.3830, 0.3721,  ..., 0.0000, 0.8430, 0.0871],
        [0.5410, 0.4724, 0.4739,  ..., 0.8430, 0.0000, 0.8978],
        [0.3579, 0.4283, 0.4240,  ..., 0.0871, 0.8978, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.23276582,  7.54127425,  7.37576154,  7.60463349,  7.69752262,
        7.72291946, 10.14436912,  7.5092834 , 10.68031351,  9.36378438,
       13.99877413, 10.09979747, 13.61801526, 12.95702647,  9.37296102,
        7.16058587, 12.64235359,  6.80370254,  9.25336371,  8.38252937,
        7.62509582, 12.0495521 ,  8.41148722,  6.55800267,  8.16714836,
       12.38881635,  7.60529892,  6.90115901, 12.04924159, 10.84493732,
        5.2287058 ,  7.53063885,  9.0368458 ,  5.0209005 , 13.0493368 ,
       13.26209545,  7.14837514,  8.87007495,  5.53175275,  6.28835101,
       10.4977132 ,  6.58184474, 13.36092624, 13.95472331,  8.30729352,
        7.78980299, 11.04629471,  6.32572323,  9.23071698,  4.87083348,
       11.05672136, 12.84056604,  8.98765644, 10.65929968,  9.95552626,
       10.61472636,  8.06643487,  8.29203356,  5.39450305,  7.6374603 ,
        6.39520397,  5.48276038, 10.42667093,  7.92200342,  7.53884104,
       12.18998195, 11.00694688,  5.8828986 ,  4.91336994,  5.38342077,
       13.2778378 , 13.25017211, 13.85158417, 13.58333857, 12.18045978,
        9.48337063,  5.79543038, 11.00726483,  8.81930525, 11.4088043 ,
        7.38844703,  4.79003361,  6.58184474, 12.25806559, 12.04924159,
        6.67121748,  5.47866221,  6.5618968 , 11.99633819, 13.98691426,
        6.33517349,  9.51074763,  5.88782913, 13.59673406,  5.14371005,
        7.13501264,  6.7166297 , 11.20100795,  7.20482224,  7.98469166,
        6.76791483, 12.10222204, 13.45124917,  5.5748624 ,  7.50500064,
        4.89869205])}
{'P': tensor([[2.0427e-02, 2.3728e-01, 1.6649e-01,  ..., 1.1265e-04, 3.4187e-04,
         2.4363e-05],
        [9.8644e-04, 2.7750e-02, 3.3627e-01,  ..., 1.1460e-04, 5.8765e-04,
         4.2577e-05],
        [1.7264e-03, 2.6218e-03, 9.3612e-02,  ..., 2.7938e-04, 3.2700e-03,
         2.1670e-04],
        ...,
        [5.2279e-02, 3.3535e-02, 1.9536e-02,  ..., 7.0297e-02, 1.2610e-04,
         3.3213e-05],
        [3.3412e-05, 3.5978e-05, 4.6477e-05,  ..., 3.9308e-05, 8.8423e-02,
         1.0662e-02],
        [4.3959e-05, 3.5113e-05, 2.9107e-05,  ..., 4.4328e-04, 3.9169e-05,
         1.1252e-01]]), 'X': tensor([[ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0374, 0.0793,  ..., 0.0858, 0.7697, 0.9128],
        [0.0374, 0.0000, 0.0451,  ..., 0.1221, 0.7560, 0.8804],
        [0.0793, 0.0451, 0.0000,  ..., 0.1652, 0.7209, 0.8353],
        ...,
        [0.0858, 0.1221, 0.1652,  ..., 0.0000, 0.8268, 0.9968],
        [0.7697, 0.7560, 0.7209,  ..., 0.8268, 0.0000, 0.5638],
        [0.9128, 0.8804, 0.8353,  ..., 0.9968, 0.5638, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.24326992,  7.37902766,  7.54620849,  7.42512621,  7.5158489 ,
        7.50500064,  9.53015376,  6.39187285, 11.04953767, 13.89671314,
        6.32572323,  9.37989631,  4.7663526 ,  8.85572292, 10.67132524,
        9.69629   , 10.7167285 ,  6.60752373, 10.66631784, 14.09357547,
        8.28968067, 13.0493368 ,  5.94541803, 13.08414689,  7.5403822 ,
        9.66977802, 10.19036081, 11.32149259, 13.3614598 ,  7.16747878,
       12.5769659 ,  6.7593917 , 14.03505624, 13.97779839, 13.72594022,
        7.16598273,  5.64500512,  6.14086582,  6.74686071,  7.92200342,
        5.91408504, 10.00802627,  6.21275016, 12.35625694,  8.47371868,
        4.91652556,  8.99276564,  8.40445872, 11.69264243,  5.05989547,
        4.76969364,  8.47371868,  8.09984231, 11.0050379 , 12.25806559,
        6.81084724, 12.62576026, 13.05148826,  7.24326992,  9.60412735,
       10.71162681, 13.20988996,  5.48276038, 11.89855267,  5.53255882,
        8.09984231, 12.1460931 ,  6.83083863,  6.53250937, 11.73567759,
        5.98285319,  9.97220799, 13.02818702, 10.84437705,  6.7593917 ,
        7.28793663,  7.62509582,  9.7712321 , 11.4424816 ,  8.24808832,
       11.8159254 , 14.10796466,  7.73692436,  5.05989547,  7.89745107,
        6.42689705,  6.02789634,  5.70591986,  8.77913428, 13.52430176,
        8.09899874,  6.05165248,  9.88310614, 12.5085995 , 13.77075174,
       11.32149259, 11.05413545,  7.96181288,  5.8828986 ,  5.18047638,
        5.88782913, 11.57249984,  4.8763135 ,  6.24187044,  8.85939626,
       10.06407839])}
{'P': tensor([[2.7910e-02, 1.8485e-01, 1.7403e-01,  ..., 3.6700e-06, 6.0361e-03,
         3.1703e-06],
        [3.7892e-04, 1.6835e-02, 2.9298e-01,  ..., 1.3670e-06, 9.7750e-04,
         1.2702e-06],
        [5.8245e-04, 1.3114e-03, 3.3115e-02,  ..., 2.5784e-06, 1.3587e-03,
         2.4382e-06],
        ...,
        [6.6415e-04, 2.1539e-04, 1.8236e-04,  ..., 5.6027e-02, 1.5411e-04,
         2.7353e-04],
        [4.0245e-03, 2.4147e-02, 2.7009e-02,  ..., 2.4581e-06, 4.1566e-02,
         2.0756e-06],
        [9.8430e-05, 3.2444e-05, 2.7634e-05,  ..., 7.1668e-01, 2.3129e-05,
         1.3864e-02]]), 'X': tensor([[ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0753, 0.0875,  ..., 0.6112, 0.2696, 0.6561],
        [0.0753, 0.0000, 0.0251,  ..., 0.6531, 0.2154, 0.6920],
        [0.0875, 0.0251, 0.0000,  ..., 0.6400, 0.2288, 0.6770],
        ...,
        [0.6112, 0.6531, 0.6400,  ..., 0.0000, 0.8683, 0.0736],
        [0.2696, 0.2154, 0.2288,  ..., 0.8683, 0.0000, 0.9058],
        [0.6561, 0.6920, 0.6770,  ..., 0.0736, 0.9058, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.24677664,  7.53063885,  7.58668144,  7.67939546,  7.57946459,
        7.72668467, 10.8992524 ,  9.58072286,  6.29034958,  6.00663314,
       11.64746882,  8.7907923 , 11.00812996, 11.31276109, 12.52372629,
       11.21703654,  7.06272256,  5.29009836, 13.37121923,  9.91668182,
       10.33958639, 11.00508681, 13.13205357,  5.5671389 ,  6.61386436,
        6.02588517, 11.72802491,  4.90505566, 12.49291768,  7.50035355,
       13.94008948, 13.40850328, 10.71162681,  9.5014146 ,  8.41171369,
        7.14837514,  7.70897675,  5.05989547,  5.08910491,  6.29821673,
        7.26074925,  5.95111804,  5.88782913, 12.5085995 , 14.07623266,
        7.37902766, 10.00802627,  5.83560009,  6.56833864, 11.54672755,
        9.86089579,  6.10360073, 10.6910768 , 14.03618649, 10.67132524,
        5.82345867,  6.27715105, 13.96905756,  5.91854397,  6.38968357,
        9.96152788, 11.97949273, 10.9036226 ,  7.20482224,  9.43737523,
        5.08372149,  9.41605684,  9.79814997,  5.80860165, 13.04217487,
        7.5158489 ,  8.24808832, 10.35464916,  8.19703512,  8.55982105,
        6.2134961 ,  5.83560009,  5.19986797,  5.91677188,  4.78258092,
       12.88270018, 12.71843462, 13.77604293,  8.62260327, 11.83570132,
        6.31576264,  9.05843201,  5.39450305, 13.66051884, 12.16901615,
       13.97430191,  5.16077764, 13.48809595, 10.7167285 ,  6.28961988,
       13.70893492,  7.59795445,  5.79543038, 14.09176259,  5.40689332,
        6.00663314, 11.00726483,  8.29640613, 11.83075005,  7.32950385,
       11.70016217])}
{'P': tensor([[1.7944e-02, 2.4549e-01, 2.3980e-01,  ..., 9.0745e-06, 4.6557e-04,
         5.9541e-05],
        [1.1864e-03, 2.7698e-02, 1.2785e-03,  ..., 1.5581e-05, 5.0901e-04,
         1.0831e-04],
        [8.3032e-04, 3.6847e-01, 1.7992e-02,  ..., 9.9281e-06, 3.4033e-04,
         6.8662e-05],
        ...,
        [6.5726e-06, 5.8227e-06, 5.8628e-06,  ..., 5.9650e-02, 2.5083e-05,
         5.0435e-04],
        [3.7796e-03, 4.0621e-03, 3.9815e-03,  ..., 6.8127e-05, 5.0882e-02,
         2.2256e-04],
        [7.8923e-06, 7.4145e-06, 7.4314e-06,  ..., 4.8070e-02, 1.4836e-05,
         3.4136e-02]]), 'X': tensor([[ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.3092e-02, 1.7693e-02,  ..., 1.1170e+00, 5.4047e-01,
         9.3790e-01],
        [2.3092e-02, 0.0000e+00, 7.2767e-03,  ..., 1.1107e+00, 5.5441e-01,
         9.2849e-01],
        [1.7693e-02, 7.2767e-03, 0.0000e+00,  ..., 1.1164e+00, 5.5415e-01,
         9.3474e-01],
        ...,
        [1.1170e+00, 1.1107e+00, 1.1164e+00,  ..., 0.0000e+00, 8.9598e-01,
         2.3435e-01],
        [5.4047e-01, 5.5441e-01, 5.5415e-01,  ..., 8.9598e-01, 0.0000e+00,
         8.2344e-01],
        [9.3790e-01, 9.2849e-01, 9.3474e-01,  ..., 2.3435e-01, 8.2344e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.26074925,  7.34250948,  7.33350642,  7.53063885,  7.58668144,
        7.57946459,  5.19986797, 12.24467041, 10.33712258, 13.70893492,
        4.75303127,  8.30729352, 12.41012046, 10.05558893, 10.26327895,
        5.80312949, 12.9089559 ,  6.2134961 ,  9.58072286,  6.27715105,
       11.92133129, 11.63502679,  8.49161154, 12.21532282, 13.89264822,
       11.63765273, 13.04217487, 10.48315436,  7.02326515,  5.81481044,
        7.38844703, 10.99326577, 10.05558893, 13.96746816,  6.49538841,
        7.5403822 , 10.61047596,  7.50564223, 11.28215429,  4.978704  ,
        9.96152788,  9.79814997,  4.76969364,  8.66273171,  9.18358562,
       11.38774149, 13.86480175, 10.66631784, 12.12641969, 13.59480906,
        5.94293185,  6.38968357,  5.88782913,  9.96152788,  8.16584537,
       11.66453059, 12.52875793, 13.86480175,  6.99097013,  9.68323686,
       10.84437705,  7.15990676, 11.70016217,  9.15866607, 12.12448912,
       11.38774149,  5.5513072 ,  5.91408504,  5.94541803, 11.76830981,
       10.63038565, 10.30972193,  9.14044984,  5.39098669,  7.16598273,
        7.32233733,  5.39450305, 10.31613269, 12.45418236,  6.77777606,
        5.94541803,  9.99145096,  8.85572292,  6.39187285,  9.68324553,
       13.20597932, 11.37939695,  8.37954853,  8.72896381,  7.15990676,
        8.97052115,  8.31625171, 13.12299145,  7.59795445, 10.31584102,
        5.5748624 , 13.5247215 ,  9.27474386,  6.19633139, 12.22276278,
        9.76896904,  5.1224079 , 13.48809595,  9.87720102,  7.07902036,
        9.53015376])}
{'P': tensor([[1.4364e-02, 2.0374e-01, 1.3724e-01,  ..., 5.0638e-04, 9.7791e-05,
         9.7307e-07],
        [7.2066e-04, 1.9984e-02, 1.9319e-01,  ..., 3.4571e-04, 7.4051e-05,
         1.1031e-06],
        [6.9957e-04, 1.2284e-03, 2.9356e-02,  ..., 2.1240e-04, 5.4329e-05,
         1.1180e-06],
        ...,
        [5.1292e-03, 9.5001e-03, 1.4815e-02,  ..., 4.0419e-02, 5.2111e-05,
         4.8694e-07],
        [5.8654e-03, 7.6793e-03, 7.4224e-03,  ..., 7.5703e-01, 2.0303e-02,
         8.0747e-07],
        [3.2467e-05, 2.7923e-05, 1.7474e-05,  ..., 1.0967e-04, 4.4551e-04,
         8.6633e-03]]), 'X': tensor([[  3.9895,   6.0816,  -6.5235],
        [  3.3744,   6.5623,  -6.3636],
        [  2.5663,   7.0526,  -6.8769],
        [  2.2860,   7.1916,  -6.1011],
        [  1.6422,   7.4586,  -6.8056],
        [  2.4932,   7.0903,  -5.4214],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.5636, -10.9293,   9.6519],
        [  1.0093,   7.6567,   2.3526],
        [  5.2608,  -2.5256,  -9.1446],
        [  3.3744,   6.5623,  -6.3636],
        [  4.5814, -10.4485,   2.5496],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.3367,   7.1676,  -9.5263],
        [  5.4429,  -2.2529,   9.7306],
        [  6.3205,   1.7068,  -3.8619],
        [ 11.7779,   5.5127,  -2.0699],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.2345,  -3.1345,   3.9715],
        [  1.7122,  -4.7591,  -3.6734],
        [  6.0054,  -1.0902,   8.3079],
        [  0.2571,  -4.7594,   1.5618],
        [  3.2990,  -4.2415,  -7.4857],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.6059,   0.9597,  -6.3264],
        [ -1.1885, -10.8211,   0.0865],
        [  1.8719,   7.3713,  -3.7465],
        [  6.1973,  -0.4372,   6.5720],
        [ 12.3453,  -1.5462,   7.0212],
        [ -1.3940, -10.7772,   6.7835],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.4290,  -3.4596,   1.4181],
        [ -4.9217,   6.8979,  -1.7667],
        [ -5.1974,   6.7396,   3.3484],
        [  2.7295,  13.6703,  -8.9328],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.3858,   4.4260,  -9.6769],
        [  9.6662,   9.1725,   8.1056],
        [  2.2849, -10.9655,  -7.6542],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.1989,  -9.7591,   0.8289],
        [  5.6202,  -1.9508,   0.7471],
        [  5.1292,   4.8183,  -5.7997],
        [  1.6308,  13.9254,  -2.0984],
        [ -0.7599, -10.8995,  -6.0366],
        [ -0.1626, -10.9796,   6.4629],
        [  3.2159, -10.8166,  -1.0406],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.1625,  -4.6697,   7.0658],
        [ -0.5385,   7.9037,   6.3830],
        [ 10.8670,  -5.3428,  -2.8637],
        [  0.3724,   7.7974,  -6.8722],
        [ -3.4100,   7.5436,  -7.6980],
        [ -7.0253,  -7.4493,  -1.5978],
        [  3.4009,  -4.1874,  -5.9751],
        [ -2.1817,   7.8274,  -4.6298],
        [  0.6142,  14.0802,   0.5407],
        [  0.8572,  14.0501,  -8.3364],
        [ -1.6220,   7.8917,   0.0315],
        [  9.3424,  -7.3685,  -3.6807],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 12.5267,  -0.4125,  -8.7331],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.9503,  -4.7172,  -6.2313],
        [ -3.3999, -10.1152,  -1.2088],
        [  0.9967,  -4.8144,  -9.7370],
        [  4.2612, -10.5514,  -6.6268],
        [  0.6757, -11.0361,  -1.3047],
        [  0.9659,  -4.8144,  -0.9594],
        [ 11.8951,  -3.1570,   7.0742],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.0604,   1.7707,   1.1066],
        [ -5.9225,  -8.5355,   2.4778],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.3812, -10.1235,   7.4384],
        [  5.5158,  -2.1337,   3.5246],
        [  2.6321,  -4.5296,  -2.5159],
        [  6.2675,  12.1379,  -8.9720],
        [  8.0877,  -8.5297,  -1.1050],
        [ -8.8400,   2.3991,   1.3457],
        [  4.5325,  -3.3643,   6.7699],
        [  5.8979,   3.3958,  -3.8713],
        [ -2.0445,   7.8469,  -0.5857],
        [  7.5064,  -8.9651,  -0.0994],
        [ 11.4743,   6.2478,  -2.1100],
        [  8.2154,  -8.4262,   8.0171],
        [ -2.1382,   7.8339,   7.3524],
        [ 11.6634,   5.8050,   9.1590],
        [  5.8836,   3.4315,  -4.0327],
        [ -6.8466,  -7.6488,   1.9648],
        [ -8.9178,   2.1936,   7.5388],
        [  5.1861,   4.7361,  -3.7312],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.9912,   3.1453,  -3.9989],
        [ 12.6035,   0.7494,   7.1536],
        [  6.2419,  -0.2313,   4.2440],
        [  3.0719, -10.8449,  -4.1673],
        [ -7.7239,   4.4298,   3.1216],
        [  4.1140,  -3.7220,  -0.1273],
        [  6.3583,   1.1729,   8.7835],
        [  7.0664,  -9.2590,   8.0879],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.6165,  -3.8715,   6.4594]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0234, 0.0667,  ..., 0.3739, 0.3426, 0.9980],
        [0.0234, 0.0000, 0.0458,  ..., 0.3973, 0.3653, 0.9960],
        [0.0667, 0.0458, 0.0000,  ..., 0.4355, 0.4088, 1.0184],
        ...,
        [0.3739, 0.3973, 0.4355,  ..., 0.0000, 0.1093, 1.1164],
        [0.3426, 0.3653, 0.4088,  ..., 0.1093, 0.0000, 1.0072],
        [0.9980, 0.9960, 1.0184,  ..., 1.1164, 1.0072, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.27340834,  7.37902766,  7.50500064,  7.54620849,  7.63726134,
        7.5158489 ,  8.56889738, 11.22596994,  7.72291946,  5.83560009,
        7.37902766, 11.4088043 ,  9.78285424,  7.53884104,  5.89074104,
        6.54693987, 13.00413829, 12.04924159,  9.75200848,  5.05774295,
        6.10360073,  4.7663526 ,  5.37343173,  9.23178204, 12.64235359,
       10.88618114,  7.60529892,  6.21275016, 12.44177641, 10.86693708,
       12.9089559 ,  5.62005629,  8.47371868,  8.51088913, 13.94008948,
       12.67576792,  6.97106912, 13.32557133, 11.20100795,  4.92497514,
       11.56146873,  5.94909709,  7.03733775, 14.02058621, 10.92597196,
       10.98076873, 11.2845608 ,  9.94173708,  5.8828986 ,  5.14607014,
        7.92200342, 12.10942412,  7.80625981,  8.27853076, 10.23946247,
        5.39450305,  8.12580976, 14.09357547, 14.07623266,  8.05668609,
       11.89855267, 10.14436912, 12.53345445,  5.22903117, 12.32545637,
        5.10444097, 10.67132524,  4.91652556, 11.37939695, 11.05672136,
        4.91038884, 12.30695098,  5.83560009,  9.23178204, 10.38897563,
        8.28968067, 10.67322527,  5.91408504,  5.23878371, 13.66051884,
       11.75438006,  9.15976746,  5.64464807,  6.80559669,  8.10886021,
       11.69264243, 13.06498554, 11.76830981,  8.12043983, 13.02818702,
        6.81117802, 10.26545792,  9.18358562,  7.02326515, 12.07225469,
        6.76661595, 12.62576026,  6.24614905, 11.27160035,  8.90405823,
        5.54776934,  6.46560181, 11.64746882, 13.45642379, 13.20597932,
       12.24467041])}
{'P': tensor([[1.4826e-02, 2.6212e-01, 1.9785e-01,  ..., 5.5120e-06, 4.0371e-05,
         1.0728e-04],
        [7.5187e-04, 1.8359e-02, 2.8660e-01,  ..., 6.3260e-06, 4.7459e-05,
         1.1281e-04],
        [7.9795e-04, 9.3767e-04, 2.6755e-02,  ..., 8.4083e-06, 6.3932e-05,
         1.2599e-04],
        ...,
        [1.4260e-04, 1.3204e-04, 1.2023e-04,  ..., 4.3060e-02, 2.0008e-04,
         4.0882e-03],
        [6.0042e-03, 5.1790e-03, 4.0068e-03,  ..., 1.0408e-03, 9.9860e-02,
         1.9995e-02],
        [2.7682e-02, 2.6558e-02, 2.5513e-02,  ..., 1.0146e-03, 9.1404e-04,
         1.4689e-01]]), 'X': tensor([[  3.9663,   6.1018,  -5.4104],
        [  3.6703,   6.3443,  -5.4401],
        [  3.0876,   6.7527,  -5.3540],
        [  2.4932,   7.0903,  -5.4214],
        [  2.2860,   7.1916,  -6.1011],
        [  2.5375,   7.0676,  -4.3490],
        [  6.1929,   2.4501,  -6.9352],
        [  6.1973,  -0.4372,   6.5720],
        [  2.3373,  13.7721,  -0.2442],
        [  3.2701,   6.6338,   9.9209],
        [ -2.2508, -10.5479,  -2.7235],
        [ 12.5123,  -0.5339,  -1.0728],
        [  6.2419,  -0.2313,   4.2440],
        [  5.4242, -10.1261,   3.9595],
        [  6.1929,   2.4501,  -6.9352],
        [ 12.5199,  -0.4711,   3.0436],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.4203,  -2.0340,   2.5856],
        [ -6.1914,  -8.2997,   8.5380],
        [ -1.1885, -10.8211,   0.0865],
        [  0.9945,  -4.8144,  -0.1897],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.4867,  12.5816,   2.9549],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.2535,  -0.1709,  -9.4102],
        [  0.6952,  -4.8060,   8.8976],
        [  5.5626,  -2.0536,   8.4433],
        [ -8.0330,  -6.0765,   6.7945],
        [ 12.1379,   4.4125,  -3.8833],
        [  2.0937,  13.8293,  -2.3533],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.4119,  14.1019,  -4.1870],
        [ -2.2508, -10.5479,  -2.7235],
        [  1.0391,  -4.8141,  -0.0768],
        [  5.8323,  -1.5228,   5.1132],
        [  1.2273,  -4.8081,   9.8129],
        [  6.2247,  -0.3150,  -9.3657],
        [  4.0591,   6.0202,  -1.6339],
        [  2.3367,   7.1676,  -9.5263],
        [  0.5960, -11.0335,   3.1144],
        [ -9.3411,  -2.6004,  -2.7062],
        [  2.5663,   7.0526,  -6.8769],
        [ -9.3616,  -2.4749,   0.1793],
        [  1.6422,   7.4586,  -6.8056],
        [  0.9659,  -4.8144,  -0.9594],
        [ -2.7572,   7.7194,   5.9580],
        [ -1.7009,   7.8851,  -8.2437],
        [  0.9945,  -4.8144,  -0.1897],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.2026,  12.1774,  -1.5887],
        [ -6.0175,  -8.4541,  -9.8408],
        [  5.5101,  -2.1433,   9.5442],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.1178,   1.5784,   2.1410],
        [ -7.4887,   4.7364,   8.9074],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.3739,   0.3982,  -4.2564],
        [ -7.5401,   4.6718,   9.0033],
        [  6.0073,   3.0986,   6.7937],
        [  5.1292,   4.8183,  -5.7997],
        [  7.9141,  -8.6656,   9.6948],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.4283,  -1.9585,  -0.9129],
        [  0.6757, -11.0361,  -1.3047],
        [  6.3569,   0.7591,   6.5011],
        [ 12.4877,  -0.7222,   8.8347],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3497,   1.3680,  -1.9076],
        [ 12.0772,  -2.6014,  -7.1832],
        [  5.7151,  -1.7700,   3.1542],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.5626,  -2.0536,   8.4433],
        [  2.4356,  -4.5942,  -5.0808],
        [ -7.8280,   4.2848,  -7.1955],
        [  6.2022,   2.4092,   3.0349],
        [ -8.9636,   2.0648,   8.1946],
        [  0.9945,  -4.8144,  -0.1897],
        [  8.2154,  -8.4262,   8.0171],
        [  2.0243,   7.3086,   7.5432],
        [ -9.4630,  -1.5145,   0.2983],
        [  9.3624,   9.5318,   0.2779],
        [  2.1379,  -4.6757,   6.4981],
        [ 11.2209,  -4.7092,   2.6117],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.3236,   0.3011,  -4.6668],
        [ 10.1503,   8.5414,  -7.2681],
        [ -9.4748,  -0.7765,  -5.6695],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.3038,   7.8092,   0.4164],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.6000,   1.3839,  -3.2992],
        [ -7.8126,   4.3066,   5.1791],
        [  3.3839,  -4.1966,   8.3574],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.1623,  -3.4288,  -4.9961],
        [  9.9217,  -6.7006,  -5.6728],
        [  8.1528,  10.7452,   9.1493],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.1472,  -2.6801,   8.1194],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.2804,   2.0019,   0.8607]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2812e-02, 3.3842e-02,  ..., 8.4628e-01, 9.7369e-02,
         4.1036e-01],
        [1.2812e-02, 0.0000e+00, 2.1299e-02,  ..., 8.5000e-01, 1.0340e-01,
         4.1531e-01],
        [3.3842e-02, 2.1299e-02, 0.0000e+00,  ..., 8.5116e-01, 1.1131e-01,
         4.1905e-01],
        ...,
        [8.4628e-01, 8.5000e-01, 8.5116e-01,  ..., 0.0000e+00, 7.5008e-01,
         4.3949e-01],
        [9.7369e-02, 1.0340e-01, 1.1131e-01,  ..., 7.5008e-01, 0.0000e+00,
         3.1316e-01],
        [4.1036e-01, 4.1531e-01, 4.1905e-01,  ..., 4.3949e-01, 3.1316e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.27759233,  7.32950385,  7.42512621,  7.5158489 ,  7.54620849,
        7.5092834 ,  6.65991877,  6.21275016, 13.96905756,  7.39599986,
       10.78533838, 12.52372629,  6.24614905, 11.48734978,  6.65991877,
       12.52875793,  9.43112957, 10.48840476,  9.09661212,  9.63742669,
       10.35464916, 10.88618114,  4.9160962 , 10.1979669 , 13.72594022,
       12.24467041,  6.25586297,  4.85603886,  5.92952538, 10.07238463,
       12.91505111, 13.98691426, 12.25806559, 14.10796466, 10.78533838,
        4.92497514,  6.02779201,  4.96231532,  6.23262289,  7.26074925,
        7.53884104, 11.04953767,  9.69629   ,  7.50500064,  9.68323686,
        7.63726134,  4.91038884,  8.19703512,  8.06643487,  4.9160962 ,
        8.49161154, 13.6660671 , 10.37698435,  5.91222346,  5.79391825,
        9.25336371,  8.86082124,  7.8575648 ,  9.38232121,  8.87007495,
        6.7593917 ,  7.03733775, 11.73567759,  8.97052115,  9.62958973,
       11.05672136,  6.4020404 , 12.5085995 ,  6.60752373,  6.49538841,
       12.35421461,  5.98285319, 10.37698435,  9.61942445,  5.92952538,
        5.19986797,  8.92395381,  6.65368551,  9.19837792,  4.9160962 ,
       11.76830981,  7.58378655,  9.58347456, 13.3607379 ,  5.14124612,
       12.16901615,  5.54776934,  9.94285511,  6.33075682, 13.26590071,
        9.50654967,  5.01114719,  7.81509631,  9.84314712, 12.67576792,
        8.92098752,  5.39098669,  8.90405823,  9.69629   ,  9.78286733,
       11.97238624, 13.48809595, 10.48840476,  5.80312949,  5.43544626,
        6.59176018])}
{'P': tensor([[1.9833e-02, 1.6791e-01, 1.6299e-01,  ..., 3.9728e-05, 6.1620e-05,
         4.0012e-03],
        [5.4068e-04, 2.2496e-02, 2.7207e-01,  ..., 3.0784e-05, 1.5012e-04,
         1.0512e-02],
        [1.0655e-03, 3.1950e-03, 5.5851e-02,  ..., 7.5379e-05, 4.0475e-04,
         3.2026e-02],
        ...,
        [6.1744e-03, 2.9687e-03, 2.8070e-03,  ..., 1.2949e-01, 1.6316e-05,
         2.9533e-04],
        [1.1450e-05, 1.1816e-05, 1.1497e-05,  ..., 3.1039e-05, 5.4915e-02,
         4.1479e-05],
        [5.6883e-05, 7.9352e-05, 8.6387e-05,  ..., 6.5060e-05, 4.3375e-03,
         4.9699e-02]]), 'X': tensor([[  3.9085,   6.1512,  -2.2560],
        [  2.3928,   7.1404,  -1.7035],
        [  2.0547,   7.2957,  -1.9343],
        [  2.0039,   7.3173,  -1.5885],
        [  3.5939,   6.4029,  -1.8357],
        [  1.1006,   7.6318,  -2.4877],
        [  2.7121,   6.9743,  -8.4202],
        [  2.4697,  -4.5836,   1.4593],
        [  5.2939,  -2.4784,  -4.5840],
        [ -3.3812, -10.1235,   7.4384],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -8.7735,   2.5640,  -6.3313],
        [  6.3360,   0.4258,  -9.2509],
        [  6.8407,  11.7666,   3.0045],
        [ -2.1533, -10.5778,   1.0112],
        [ -2.1817,   7.8274,  -4.6298],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.3025,  -1.7440,  -6.6632],
        [  9.2979,   9.6048,   7.3138],
        [ -7.2515,   5.0195,   4.6434],
        [  6.3460,   1.4252,   1.2291],
        [ -9.4117,   0.1160,   8.2574],
        [ -5.3328,  -8.9992,   7.3440],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 12.5832,   1.7543,  -7.6134],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.1787,  -2.6383,  -4.2533],
        [  6.1575,  12.2046,   3.3257],
        [  6.9956,  11.6590,  -7.6876],
        [ -5.9225,  -8.5355,   2.4778],
        [ -7.0948,   5.1940,   9.8171],
        [  2.0419,  -4.6978,  -5.0983],
        [ -8.0608,   3.9362,  -5.9567],
        [  7.3724,  -9.0577,  -8.9843],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4630,  -1.5145,   0.2983],
        [ -5.1974,   6.7396,   3.3484],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.8994,   6.8675,  -9.5987],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 12.5650,   2.0132,   4.6272],
        [  5.1861,   4.7361,  -3.7312],
        [ -2.2188,   7.8218,   8.6064],
        [ -0.4341, -10.9473,   7.4509],
        [  5.3924, -10.1396,  -4.0863],
        [ -3.5173,   7.5091,  -8.9585],
        [  9.5203,   9.3483,   5.8781],
        [ 10.9717,  -5.1646,  -6.9039],
        [  4.8653,   5.1694,  -4.0823],
        [  9.1820,   9.7334,  -2.7932],
        [ -6.3066,  -8.1936,   9.5822],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.2707,   0.9661,   4.4863],
        [  5.3804,  -2.3504,   7.2744],
        [ -7.0948,   5.1940,   9.8171],
        [ -8.1666,   3.7653,   3.6075],
        [ 11.9695,  -2.9407,  -6.3498],
        [  6.1929,   2.4501,  -6.9352],
        [  1.6308,  13.9254,  -2.0984],
        [ 11.7162,   5.6728,  -8.9372],
        [  7.0664,  -9.2590,   8.0879],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.1504,   4.3674,  -7.2998],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.4808,   4.7462,   2.0266],
        [  5.9655,   3.2172,  -7.5953],
        [  7.0642,  11.6104,  -9.8316],
        [  6.2675,  12.1379,  -8.9720],
        [ -6.3306,   5.9272,   5.6935],
        [  1.6232,  -4.7717,  -2.5655],
        [ -9.1706,   1.3863,   8.4032],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 12.0845,  -2.5772,  -4.6073],
        [  8.5198,  10.4091,  -8.6486],
        [ -3.2053, -10.1993,  -3.8736],
        [  2.3889,  -4.6082,  -8.7291],
        [  7.9141,  -8.6656,   9.6948],
        [ 12.6038,   0.7665,  -3.4439],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -6.5295,  -7.9785,  -7.4429],
        [  9.0652,   9.8597,  -3.5949],
        [ -8.9942,  -3.9990,   6.0448],
        [  2.5829,  -4.5466,  -8.4015],
        [ -7.6650,   4.5092,   0.2520],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -2.5683, -10.4433,  -2.9182],
        [  8.6420,  -8.0580,   6.5781],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.8363,   2.4086,  -5.6296],
        [  1.0000,   7.6591,  -4.5011],
        [ -5.3179,  -9.0101,   9.4472],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -7.7842,   4.3466,  -1.7548],
        [ -2.5683, -10.4433,  -2.9182],
        [  6.9284,  11.7061,  -0.3732],
        [  1.5662,  13.9375,   4.5142],
        [ -3.1204, -10.2345,   9.2288],
        [  6.3606,   0.9204,   3.2525],
        [  6.9693,  -9.3200,  -8.3822],
        [  9.4516,  -7.2501,  -1.8868],
        [ -0.7599, -10.8995,  -6.0366],
        [  1.9505,  -4.7171,   9.9868],
        [  6.0955,   2.8210,  -4.3321],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.4773,  -1.0024,   5.2599],
        [ -6.0195,   6.1789,  -2.9496]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.5265e-02, 6.9738e-02,  ..., 2.6302e-01, 9.5087e-01,
         4.7292e-01],
        [5.5265e-02, 0.0000e+00, 2.4950e-02,  ..., 3.0994e-01, 8.9571e-01,
         4.3546e-01],
        [6.9738e-02, 2.4950e-02, 0.0000e+00,  ..., 3.0938e-01, 8.8384e-01,
         4.1136e-01],
        ...,
        [2.6302e-01, 3.0994e-01, 3.0938e-01,  ..., 0.0000e+00, 1.1685e+00,
         5.8054e-01],
        [9.5087e-01, 8.9571e-01, 8.8384e-01,  ..., 1.1685e+00, 3.0339e-04,
         6.7311e-01],
        [4.7292e-01, 4.3546e-01, 4.1136e-01,  ..., 5.8054e-01, 6.7311e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.28793663,  7.53063885,  7.57946459,  7.58668144,  7.34250948,
        7.71075226,  7.48311002,  5.20659892,  5.84533345, 10.67322527,
       12.25806559,  9.14044984,  6.35029274, 13.6105595 , 10.794755  ,
        8.12580976,  4.91336994, 12.42554808, 13.36800073,  8.81930525,
        6.50409372,  9.41244865, 10.4606265 , 12.32545637, 12.70489563,
       12.76408562,  5.81199422, 13.66991027, 13.59673406, 10.38897563,
        8.79284466,  5.1224079 ,  8.97052115, 11.67875855, 13.36092624,
        9.58347456,  8.51088913, 12.38881635,  7.4544866 ,  9.5642993 ,
       12.72524615,  7.02326515,  8.1303945 , 10.95593729, 11.48434767,
        8.29203356, 13.3426541 , 12.12641969,  7.09887557, 13.38091121,
       10.33958639, 10.74655346,  9.32094055,  5.87132718,  8.79284466,
        8.99276564, 12.32545637,  6.65991877, 14.02058621, 13.01728313,
       11.64746882, 12.86648771, 12.91143491, 11.9182956 ,  8.859416  ,
        6.77777606, 13.59056613, 13.66051884,  8.67227386,  5.04028538,
        9.27474386, 10.53368644, 12.35625694, 13.45124917, 10.6910768 ,
        5.19065439, 11.73567759, 12.62711252,  9.91668182, 10.30972193,
       13.39372426,  9.84314712,  5.22903117,  8.89301516, 12.4217523 ,
       10.75443106, 11.8159254 ,  7.53063885,  9.15866607,  7.72415048,
       10.46238054,  9.78147542,  8.91551231, 10.75443106, 13.60275211,
       14.02526505, 10.69963954,  6.42689705, 11.63765273, 11.9120321 ,
       10.92597196,  5.10446653,  6.7166297 ,  5.68942572,  9.53015376,
        8.62631801])}
{'P': tensor([[2.5294e-02, 2.3106e-01, 2.0166e-01,  ..., 4.8509e-05, 3.9239e-06,
         2.2390e-06],
        [6.3153e-04, 2.4501e-02, 4.6953e-01,  ..., 5.0545e-05, 2.9486e-06,
         1.9468e-06],
        [9.2145e-04, 1.5973e-03, 4.5992e-02,  ..., 9.5618e-05, 5.2674e-06,
         3.5373e-06],
        ...,
        [1.9882e-03, 1.4259e-03, 1.2576e-03,  ..., 2.1908e-01, 7.0103e-03,
         3.3207e-02],
        [9.8630e-05, 4.2424e-05, 3.3578e-05,  ..., 2.1567e-05, 7.1137e-03,
         6.3811e-05],
        [5.3470e-05, 2.9334e-05, 2.3837e-05,  ..., 1.6925e-04, 7.4926e-04,
         1.5197e-02]]), 'X': tensor([[ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0562, 0.0783,  ..., 0.3985, 0.2116, 0.6644],
        [0.0562, 0.0000, 0.0260,  ..., 0.3764, 0.1928, 0.6198],
        [0.0783, 0.0260, 0.0000,  ..., 0.3854, 0.1738, 0.6143],
        ...,
        [0.3985, 0.3764, 0.3854,  ..., 0.0000, 0.5504, 0.3590],
        [0.2116, 0.1928, 0.1738,  ..., 0.5504, 0.0000, 0.7282],
        [0.6644, 0.6198, 0.6143,  ..., 0.3590, 0.7282, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.32233733,  7.54127425,  7.60463349,  7.37576154,  7.73151309,
        7.81509631, 11.4424816 ,  5.5513072 ,  7.5403822 ,  6.5618968 ,
       10.26327895, 13.12299145,  8.87007495,  6.83515201, 12.53285633,
        5.79391825,  9.95552626,  7.42512621, 11.018084  , 12.07818871,
       11.63383221, 11.36089355, 13.1471313 , 13.44593519,  4.78002665,
       10.54658597,  9.95552626, 11.51811603,  7.46248947, 13.00380624,
       12.71843462, 12.04924159,  8.87587457,  9.48969978,  9.86089579,
       13.6660671 ,  8.87007495, 13.26209545,  6.81084724,  5.63824971,
        5.27178363, 11.9182956 ,  9.73579738,  5.63824971, 13.95361098,
       11.4088043 ,  9.23178204,  7.18354885, 11.33837396, 12.46513646,
        8.81930525, 13.05148826, 14.09357547, 11.92133129,  9.48337063,
        6.84796202, 11.11038551, 11.4424816 ,  7.5158489 , 13.91198191,
       11.95743558,  6.2322769 ,  9.3349858 , 13.77075174,  8.12580976,
        7.62011734,  9.3109884 ,  6.89245352, 10.80728788, 10.51340829,
        8.28968067, 10.35485331, 12.1494023 ,  7.37902766, 10.95593729,
       11.76830981, 13.94008948, 12.49291768,  9.5642993 , 13.96746816,
        7.38844703, 11.92133129, 11.72749923,  9.79814997, 12.62711252,
        9.96391045, 14.10796466,  9.36256096,  9.43737523,  9.78286733,
       14.02058621, 12.26997165, 12.33167382, 13.18278606,  8.04664886,
       11.4088043 ,  7.4544866 , 14.09176259,  7.57685908, 12.10942412,
        5.36061974,  8.36995953, 10.35093485,  4.91336994, 11.75807252,
       11.62158554])}
{'P': tensor([[1.0542e-02, 1.3579e-01, 1.3016e-01,  ..., 6.8517e-07, 1.3233e-06,
         1.3192e-06],
        [1.7920e-03, 5.4514e-02, 3.9673e-03,  ..., 2.6704e-06, 8.1633e-06,
         8.4243e-06],
        [1.2098e-03, 1.7165e-01, 2.2909e-02,  ..., 1.1960e-06, 2.8986e-06,
         2.9121e-06],
        ...,
        [2.8776e-05, 2.6809e-05, 2.4368e-05,  ..., 1.1319e-02, 2.1040e-07,
         2.1460e-07],
        [3.3550e-05, 2.6355e-05, 2.7878e-05,  ..., 7.1037e-04, 1.3175e-01,
         1.7423e-03],
        [7.9547e-06, 6.4986e-06, 6.6974e-06,  ..., 1.6539e-04, 7.3524e-01,
         5.0068e-02]]), 'X': tensor([[  3.6817,   6.3354,   6.1233],
        [  2.8602,   6.8905,   5.8779],
        [  3.3167,   6.6022,   6.3554],
        [  2.3261,   7.1726,   6.2522],
        [  2.8474,   6.8979,   6.6457],
        [  2.9410,   6.8428,   5.4458],
        [  4.1348,   5.9514,  -0.8856],
        [  1.1139,   7.6281,  -2.5680],
        [  5.6202,  -1.9508,   0.7471],
        [  5.9026,   3.3837,  -0.6573],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.3048,   0.1469,  -9.1691],
        [  1.6308,  13.9254,  -2.0984],
        [  6.2675,  12.1379,  -8.9720],
        [  4.1537,   5.9340,  -6.0423],
        [  5.6414, -10.0303,   7.8525],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.3516,   0.6425,   0.3425],
        [  3.9591, -10.6391,  -6.2557],
        [  5.6093,  -1.9705,   6.4807],
        [ -1.2454,   7.9127,  -0.5715],
        [ 12.6035,   0.7494,   7.1536],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.2182,  13.8007,   5.1889],
        [ 11.5568,   6.0602,   9.3244],
        [ -9.4630,  -1.5145,   0.2983],
        [  3.5284, -10.7484,   5.3347],
        [ 12.3609,   3.4769,  -8.5762],
        [  3.3167,   6.6022,   6.3554],
        [  6.0890,  -0.8382,   2.9166],
        [  7.7750,  11.0665,   2.8364],
        [ -1.6086,   7.8928,   4.3385],
        [  7.0515,  11.6194,   8.4289],
        [ 11.7726,  -3.4876,  -6.7402],
        [  5.0190,   4.9705,  -7.9752],
        [ 11.7796,  -3.4696,   9.0088],
        [ -2.8162,   7.7059,  -9.6885],
        [ -8.1666,   3.7653,   3.6075],
        [  1.7815,  -4.7481,   4.4289],
        [ 11.2767,  -4.6006,   0.7868],
        [ -9.4735,  -0.7274,  -2.6749],
        [  7.8425,  -8.7202,   6.4530],
        [  6.9433,  -9.3362,  -0.3421],
        [  6.3122,  -9.6994,  -6.3282],
        [  3.6154,  -4.0640,  -7.9698],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.8323,  -1.5228,   5.1132],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.0788,  -0.8710,  -6.8719],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 12.2589,  -1.9312,   9.8657],
        [  3.3167,   6.6022,   6.3554],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.1624,  -3.4286,   2.0647],
        [  6.2036,   2.4030,  -1.2500],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.3606,   0.9204,   3.2525],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.2218,   4.0953,  -8.2072],
        [  5.9587,  12.3216,   9.1695],
        [ -9.1624,  -3.4286,   2.0647],
        [ -2.0229, -10.6163,   8.6930],
        [ 10.5678,   7.9261,  -0.6100],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.2630,   2.1057,  -5.2136],
        [ 11.7418,   5.6070,  -1.4192],
        [  9.0129,  -7.7072,   9.1400],
        [  5.8930,  12.3592,  -8.9201],
        [ -5.9987,   6.1949,  -6.2114],
        [  3.3167,   6.6022,   6.3554],
        [  3.5939,   6.4029,  -1.8357],
        [  3.2809,  13.5061,  -4.0199],
        [ -8.3653,   3.4184,   9.3454],
        [  2.4697,  -4.5836,   1.4593],
        [  4.9337,  12.8566,  -3.6582],
        [  6.6319,  11.9066,   3.5165],
        [  7.0642,  11.6104,  -9.8316],
        [  3.4920,  -4.1366,  -8.3955],
        [  6.1372,   2.6726,  -6.7885],
        [  1.3103,  -4.8032,  -0.9300],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.9895,   6.0816,  -6.5235],
        [  4.4633,   5.6287,  -0.7754],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.2555,   2.1474,   4.1685],
        [  4.1348,   5.9514,  -0.8856],
        [ -0.1626, -10.9796,   6.4629],
        [  6.2038,   2.4020,  -3.1692],
        [  9.2811,  -7.4336,   5.1771],
        [  2.0419,  -4.6978,  -5.0983],
        [  9.6614,   9.1784,  -1.2087],
        [  9.9058,  -6.7203,   5.6928],
        [  5.4336,  -2.2676,   5.6933],
        [ 12.2592,   3.9425,   2.0383],
        [  6.1929,   2.4501,  -6.9352],
        [  6.2026,  12.1774,  -1.5887],
        [ -3.1671,   7.6158,   6.0765],
        [ -7.3470,  -7.0615,   5.0066],
        [ -7.4808,   4.7462,   2.0266],
        [ -5.6423,   6.4537,  -1.2091],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -3.2053, -10.1993,  -3.8736],
        [ -4.5778,  -9.5013,  -5.0667]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0438, 0.0217,  ..., 1.0588, 1.0730, 1.1496],
        [0.0438, 0.0000, 0.0353,  ..., 1.0605, 1.0367, 1.1136],
        [0.0217, 0.0353, 0.0000,  ..., 1.0764, 1.0718, 1.1487],
        ...,
        [1.0588, 1.0605, 1.0764,  ..., 0.0000, 1.0294, 1.0546],
        [1.0730, 1.0367, 1.0718,  ..., 1.0294, 0.0000, 0.0785],
        [1.1496, 1.1136, 1.1487,  ..., 1.0546, 0.0785, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.32754882,  7.46053281,  7.38844703,  7.5403822 ,  7.46248947,
        7.44805901,  7.24677664,  7.70897675,  5.94909709,  6.80370254,
        5.34250225,  6.30647746, 14.02058621, 13.66051884,  7.24326992,
       11.50791859,  4.91652556,  6.38399789, 11.35182589,  5.94535529,
        8.01009083, 12.62576026, 12.24467041, 13.97779839, 13.0493368 ,
        9.58347456, 11.31276109, 12.84056604,  7.38844703,  6.1463896 ,
       13.5247215 ,  8.0550286 , 13.59170809, 12.27835996,  7.06372172,
       12.27993098,  8.20436764,  8.99276564,  5.0713364 , 12.17901033,
        9.5014146 , 11.72802491, 11.63502679, 11.57249984,  5.43943791,
       11.75807252,  6.02779201,  5.83560009,  6.14086582,  9.5642993 ,
       12.41012046,  7.38844703,  5.39098669,  9.4997987 ,  9.78285424,
        6.652742  ,  5.35774573, 12.18998195,  6.42689705, 12.91143491,
       12.889695  , 13.68672855,  9.78285424, 10.80728788, 13.20988996,
        6.25586297,  6.60752373, 13.01187484, 11.85890664, 13.6922425 ,
        8.62329466,  7.38844703,  7.34250948, 13.89886469,  9.0368458 ,
        5.20659892, 13.77075174, 13.62896593, 13.59056613,  5.41348121,
        6.69389431,  4.978704  ,  9.88310614,  7.27340834,  7.18354885,
        4.85995002,  6.61386436,  7.24677664, 10.98076873,  6.65259775,
       11.89105259,  5.1224079 , 13.32613445, 11.9702762 ,  5.88782913,
       12.87752052,  6.65991877, 13.6660671 ,  8.24808832, 10.19036081,
        8.859416  ,  8.5724042 ,  4.91336994, 12.53345445, 10.6910768 ,
       10.54658597])}
{'P': tensor([[1.5657e-02, 2.4443e-01, 1.8966e-01,  ..., 2.0514e-04, 3.6416e-04,
         1.3023e-03],
        [7.3684e-04, 2.1024e-02, 3.4840e-01,  ..., 2.0015e-04, 6.5310e-04,
         1.5849e-03],
        [1.5324e-03, 2.0303e-03, 5.9964e-02,  ..., 3.8462e-04, 2.5289e-03,
         3.6364e-03],
        ...,
        [4.3986e-05, 6.7758e-05, 9.4415e-05,  ..., 5.2391e-03, 5.7655e-06,
         2.1594e-03],
        [2.8378e-05, 2.9198e-05, 3.0841e-05,  ..., 6.3905e-07, 6.5335e-02,
         1.1903e-05],
        [1.0879e-03, 1.5316e-03, 2.0381e-03,  ..., 7.8928e-04, 7.3051e-04,
         1.0127e-01]]), 'X': tensor([[  3.6703,   6.3443,  -5.4401],
        [  3.0876,   6.7527,  -5.3540],
        [  2.4932,   7.0903,  -5.4214],
        [  2.2860,   7.1916,  -6.1011],
        [  2.5375,   7.0676,  -4.3490],
        [  1.6408,   7.4591,  -4.2294],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.4877,  -0.7222,   8.8347],
        [  6.0955,   2.8210,  -4.3321],
        [  2.7295,  13.6703,  -8.9328],
        [ -0.5739,   7.9056,   0.6382],
        [ -5.5525,   6.5146,  -4.4984],
        [  4.7813,  -3.1139,   8.2691],
        [ 11.0438,   7.1180,  -5.6991],
        [ -1.8367,   7.8718,  -8.6523],
        [ -4.7334,   6.9982,  -0.4392],
        [  4.9337,  12.8566,  -3.6582],
        [ -8.8058,   2.4850,  -1.7425],
        [  4.2096,   5.8815,   1.9230],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.6093,  -1.9705,   6.4807],
        [  8.5198,  10.4091,  -8.6486],
        [  8.5198,  10.4091,  -8.6486],
        [ 12.0042,  -2.8350,   2.6864],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.3317,   7.8034,  -3.6274],
        [ 12.6059,   0.9597,  -6.3264],
        [  1.3339,   7.5627,  -0.9034],
        [  6.2804,   2.0019,   0.8607],
        [ -8.9606,  -4.0994,   1.7886],
        [ -7.0948,   5.1940,   9.8171],
        [  0.1252, -11.0062,   7.2564],
        [  7.0515,  11.6194,   8.4289],
        [  0.9788,  -4.8145,  -3.8661],
        [ 11.4703,  -4.2006,   7.4086],
        [  5.1196,  -2.7160,   4.2705],
        [ -6.0175,  -8.4541,  -9.8408],
        [  0.1287, -11.0065,   6.5757],
        [ 12.5267,  -0.4125,  -8.7331],
        [  4.5800,   5.5034,  -3.9432],
        [  6.9956,  11.6590,  -7.6876],
        [  5.8889,  -1.3916,   8.8993],
        [  0.9809,   7.6642,  -1.4217],
        [  3.8182,  -3.9347,  -1.7961],
        [ -0.5709,   7.9055,   7.6089],
        [  3.4401,  13.4540,   4.6576],
        [ -8.6426,  -4.9136,   9.4941],
        [  3.4401,  13.4540,   4.6576],
        [  5.5626,  -2.0536,   8.4433],
        [  3.9770, -10.6341,   8.0783],
        [  2.7121,   6.9743,  -8.4202],
        [  5.1861,   4.7361,  -3.7312],
        [  6.0955,   2.8210,  -4.3321],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.3205,   1.7068,  -3.8619],
        [ -7.6525,  -6.6534,   5.1721],
        [ -8.9178,   2.1936,   7.5388],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.6078, -10.9229,   6.2869],
        [ 10.1798,   8.5002,  -5.8893],
        [  3.1480,  -4.3165,  -7.8002],
        [ -9.4196,   0.0478,   9.4140],
        [  4.3523, -10.5232,   8.2138],
        [  0.1047, -11.0046,   0.5188],
        [  3.6817,   6.3354,   6.1233],
        [ -9.4430,  -0.1897,   4.8383],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.0617,  12.7957,   8.3544],
        [  3.4401,  13.4540,   4.6576],
        [  2.0729,   7.2878,   7.0880],
        [  5.0277,   4.9589,   7.0175],
        [  5.3118,   4.5452,   4.2114],
        [  7.8425,  -8.7202,   6.4530],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.2860,   7.1916,  -6.1011],
        [ -2.0445,   7.8469,  -0.5857],
        [  3.9022,  13.2906,   5.5788],
        [ -8.7735,   2.5640,  -6.3313],
        [  9.5203,   9.3483,   5.8781],
        [  3.3094,  13.4969,  -3.6507],
        [ -1.1885, -10.8211,   0.0865],
        [  5.0506,   4.9277,  -9.5094],
        [ 12.5832,   1.7543,  -7.6134],
        [  2.0243,   7.3086,   7.5432],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.2365,  -0.2578,  -6.0212],
        [  2.3261,  13.7749,   5.9121],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.2188,  -2.5839,   2.1833],
        [  6.3211,   1.7017,  -0.2164],
        [ -8.6112,   2.9319,  -7.9943],
        [ -3.4986,   7.5152,   9.0712],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3515,   0.6418,  -6.5823],
        [ -5.3382,   6.6533,   7.8054],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.3373,  13.7721,  -0.2442],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.0604,   1.7707,   1.1066],
        [  3.9770, -10.6341,   8.0783],
        [  2.5974,   7.0363,  -7.7810],
        [  6.1575,  12.2046,   3.3257],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.1994,   7.6035,   2.4295]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0213, 0.0508,  ..., 0.8708, 0.8039, 0.5294],
        [0.0213, 0.0000, 0.0298,  ..., 0.8795, 0.7835, 0.5233],
        [0.0508, 0.0298, 0.0000,  ..., 0.8980, 0.7537, 0.5231],
        ...,
        [0.8708, 0.8795, 0.8980,  ..., 0.0000, 1.5164, 0.5842],
        [0.8039, 0.7835, 0.7537,  ..., 1.5164, 0.0000, 0.9537],
        [0.5294, 0.5233, 0.5231,  ..., 0.5842, 0.9537, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.32950385,  7.42512621,  7.5158489 ,  7.54620849,  7.5092834 ,
        7.6374603 ,  5.43943791, 12.91143491, 12.5085995 ,  6.7166297 ,
       13.94008948,  7.92644746,  8.55982105,  5.70591986, 13.13888841,
        8.08320235,  8.4486771 , 13.77075174,  9.14973319,  7.23276582,
       11.1341798 ,  9.70108409,  5.94535529, 13.45124917, 13.45124917,
       12.3344536 , 13.86480175,  8.14433542, 12.64235359,  7.67939546,
        6.59176018,  9.85384623,  8.79284466, 11.00694688, 13.59170809,
        4.91296205, 12.21532282,  5.79543038, 10.37698435, 11.00726483,
       12.53345445,  7.15990676, 13.59673406,  6.05113217,  7.72668467,
        5.48276038,  7.92607342, 13.88683436,  9.94173708, 13.88683436,
        5.92952538, 11.35345345,  7.48311002,  7.02326515,  6.7166297 ,
        5.24461068,  6.54693987, 10.14044489,  9.18358562,  9.75401449,
       11.22993218, 13.26209545,  5.34250225,  9.41970031, 11.38774149,
       11.00508681,  7.32754882,  9.44486609,  5.53175275, 13.76047999,
       13.88683436,  7.57685908,  7.06169391,  6.99097013, 11.72802491,
       12.38881635,  7.54620849,  8.10886021, 13.85158417,  9.14044984,
       13.3426541 , 13.89671314, 10.88618114,  7.05627461, 12.70489563,
        7.58378655, 13.20597932,  6.24187044, 13.96987806, 13.01187484,
        5.82345867,  6.54615887,  9.09661212,  8.28968067,  6.27905292,
        6.383897  ,  8.53014915, 10.31886066, 13.96905756,  9.34127013,
        9.23178204, 11.35345345,  7.50035355, 13.66991027,  9.17905491,
        7.69752262])}
{'P': tensor([[1.4092e-02, 2.8859e-01, 1.6270e-01,  ..., 9.6331e-06, 1.7279e-06,
         2.8341e-06],
        [9.2300e-04, 1.9997e-02, 2.3819e-01,  ..., 1.3506e-05, 2.4321e-06,
         4.0347e-06],
        [4.3588e-04, 4.4685e-04, 1.7515e-02,  ..., 9.4161e-06, 1.8712e-06,
         3.9436e-06],
        ...,
        [3.4846e-03, 3.4490e-03, 2.7196e-03,  ..., 4.1153e-01, 8.4543e-05,
         2.1857e-04],
        [1.5555e-04, 1.5066e-04, 7.7624e-05,  ..., 4.6303e-05, 3.2349e-02,
         3.2397e-05],
        [1.6110e-04, 1.5684e-04, 9.1428e-05,  ..., 2.7270e-03, 3.3801e-02,
         1.1708e-01]]), 'X': tensor([[  3.6469,   6.3624,  -1.8414],
        [  3.5939,   6.4029,  -1.8357],
        [  2.3928,   7.1404,  -1.7035],
        [  2.0547,   7.2957,  -1.9343],
        [  2.0039,   7.3173,  -1.5885],
        [  0.9809,   7.6642,  -1.4217],
        [ -8.1905,   3.7253,   4.1175],
        [  6.2576,   2.1359,  -3.3398],
        [  6.2026,  12.1774,  -1.5887],
        [ -0.0410,   7.8591,   9.3457],
        [  2.5829,  -4.5466,  -8.4015],
        [  1.5383, -11.0274,  -4.9666],
        [  3.6817,   6.3354,   6.1233],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.3464,   0.5708,  -1.6397],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 11.1420,   6.9331,  -2.4779],
        [ -1.7290,   7.8825,  -3.3831],
        [  6.3515,   0.6418,  -6.5823],
        [ -9.4366,  -0.1189,  -2.6736],
        [  1.5246,  -4.7838,  -5.2095],
        [ 11.4743,   6.2478,  -2.1100],
        [  2.3373,  -4.6232,  -5.2246],
        [ -5.3736,   6.6310,  -1.5174],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -3.9508,  -9.8515,   1.7923],
        [  5.6095,  -1.9702,  -7.8233],
        [ -6.6895,   5.6051,   2.7899],
        [  1.1644, -11.0394,   7.8511],
        [ 11.0438,   7.1180,  -5.6991],
        [ -6.0844,  -8.3955,   0.7705],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.1905,   3.7253,   4.1175],
        [  7.9403,  -8.6455,  -8.7433],
        [ -9.1178,   1.5784,   2.1410],
        [ -3.1671,   7.6158,   6.0765],
        [  6.3460,   1.4252,   1.2291],
        [ -3.5173, -10.0623,   4.8280],
        [  2.1379,  -4.6757,   6.4981],
        [  1.7301,   7.4262,   5.0522],
        [ -9.4351,  -1.8891,   3.5642],
        [ -8.8540,  -4.3971,  -5.2144],
        [  2.1824,  -4.6647,   1.1985],
        [  3.1756, -10.8247,   6.8882],
        [  5.1203,  -2.7151,   6.2110],
        [ -8.5607,  -5.0938,  -5.4531],
        [  7.3592,  11.3937,   6.2407],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -5.5525,   6.5146,  -4.4984],
        [  6.2247,  -0.3150,  -9.3657],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.7121,   6.9743,  -8.4202],
        [  3.6469,   6.3624,  -1.8414],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.7802,  -1.6363,  -6.2782],
        [ -2.6485, -10.4151,   0.8766],
        [  3.2366,  -4.2732,   5.1532],
        [  5.4867,  12.5816,   2.9549],
        [ -9.1178,   1.5784,   2.1410],
        [  5.1519, -10.2387,   9.8191],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -4.4520,   7.1370,  -8.7867],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.4267,  -5.3697,   2.4172],
        [  2.7420,  -4.4896,  -3.7139],
        [ -8.0965,  -5.9712,   4.3466],
        [  5.0233,   4.9647,  -3.1788],
        [ -4.6393,   7.0461,   1.1624],
        [  9.7328,   9.0902,  -0.8917],
        [ 11.4235,   6.3594,   5.1663],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.6816,   9.1536,  -5.8543],
        [  3.3094,  13.4969,  -3.6507],
        [ -1.9646,   7.8571,   3.3817],
        [ -7.7239,   4.4298,   3.1216],
        [  9.0051,   9.9234,   5.4411],
        [  6.1575,  12.2046,   3.3257],
        [  7.5064,  -8.9651,  -0.0994],
        [  9.3836,   9.5075,  -5.9278],
        [  3.6469,   6.3624,  -1.8414],
        [  6.0310,  12.2795,  -6.4528],
        [  1.3670,  -4.7990,   3.2603],
        [  4.6231,   5.4556,   8.4973],
        [  1.7218,  -4.7576,  -3.3617],
        [ -1.7009,   7.8851,  -8.2437],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.3535,  -5.5118,   9.8399],
        [  1.9204,   7.3518,   8.4781],
        [  2.2373,  -4.6506,  -1.7334],
        [ -8.6112,   2.9319,  -7.9943],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.6050,   2.9451,  -0.7012],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.3413,   0.4883,  -6.8416],
        [ -8.9282,   2.1648,   8.4894],
        [ -0.4341, -10.9473,   7.4509],
        [  5.5101,  -2.1433,   9.5442],
        [  6.0467,  -9.8369,  -6.5816],
        [  0.1382, -11.0073,   1.0026]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0073, 0.0419,  ..., 0.6744, 0.4488, 0.6049],
        [0.0073, 0.0000, 0.0348,  ..., 0.6754, 0.4448, 0.5980],
        [0.0419, 0.0348, 0.0000,  ..., 0.6740, 0.4332, 0.5634],
        ...,
        [0.6744, 0.6754, 0.6740,  ..., 0.0000, 1.0940, 0.8264],
        [0.4488, 0.4448, 0.4332,  ..., 1.0940, 0.0000, 0.6392],
        [0.6049, 0.5980, 0.5634,  ..., 0.8264, 0.6392, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.33350642,  7.34250948,  7.53063885,  7.57946459,  7.58668144,
        7.72668467,  8.99791316,  6.61211224, 13.6660671 ,  7.85919687,
        5.22903117, 11.1341798 ,  7.32754882,  6.83515201,  9.36378438,
       13.01187484,  9.5642993 , 13.12299145,  8.06990483,  6.383897  ,
        9.43737523,  5.0209005 , 13.06498554,  5.18047638,  8.53501777,
       12.10942412, 10.61417181,  5.94541803,  8.7273569 , 11.10066126,
       13.13888841, 10.36844813,  5.48276038,  5.70405525,  9.38232121,
        8.99791316, 11.73848065,  9.25336371,  8.24808832,  6.50409372,
       10.65929968,  5.14124612,  7.62509582,  9.62238828,  9.88571713,
        5.14999799, 11.28093507,  5.79562691,  9.96152788, 13.56367857,
        7.71075226, 12.05300943, 12.12641969,  8.55982105,  6.23262289,
       12.17901033,  7.48311002,  7.33350642, 10.35485331, 10.51340829,
        6.00732193, 10.74655346,  5.36061974, 13.72594022,  9.25336371,
       11.46176259,  9.97964013,  8.41171369, 10.99326577,  9.99211663,
        5.26067019, 10.06023665,  7.06272256,  8.43626114, 13.31764181,
       13.0743399 , 10.06286074, 13.32374056, 13.89671314,  8.09899874,
        8.90405823, 13.40024224, 13.66991027, 11.69264243, 13.35832725,
        7.33350642, 13.68063098,  4.98989906,  7.15100303,  5.05963217,
        8.06643487,  8.86082124, 10.00802627,  7.5984809 ,  5.16077764,
        9.09661212,  5.94909709,  8.97052115,  9.09502385, 12.32408516,
        6.36004393,  9.18690018, 10.95593729,  5.91222346, 11.54672755,
       11.00812996])}
{'P': tensor([[1.9321e-02, 2.3014e-01, 1.9940e-01,  ..., 9.1531e-04, 4.3177e-05,
         3.9626e-03],
        [4.5329e-04, 1.7767e-02, 2.1488e-01,  ..., 3.5507e-04, 3.6861e-05,
         1.4136e-03],
        [7.2907e-04, 1.9280e-03, 3.3702e-02,  ..., 5.2907e-04, 7.1467e-05,
         1.9422e-03],
        ...,
        [1.5461e-02, 2.5025e-02, 2.2967e-02,  ..., 5.6087e-02, 5.8055e-05,
         6.7141e-03],
        [1.9923e-03, 1.2624e-03, 1.1759e-03,  ..., 8.4173e-04, 1.1451e-01,
         1.3070e-04],
        [2.9428e-03, 1.1596e-02, 1.6653e-02,  ..., 6.1879e-04, 2.7131e-05,
         1.8603e-01]]), 'X': tensor([[ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0348, 0.0560,  ..., 0.3816, 0.2107, 0.3222],
        [0.0348, 0.0000, 0.0249,  ..., 0.3778, 0.2244, 0.3566],
        [0.0560, 0.0249, 0.0000,  ..., 0.3934, 0.2199, 0.3777],
        ...,
        [0.3816, 0.3778, 0.3934,  ..., 0.0000, 0.5902, 0.5022],
        [0.2107, 0.2244, 0.2199,  ..., 0.5902, 0.0000, 0.3507],
        [0.3222, 0.3566, 0.3777,  ..., 0.5022, 0.3507, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.34250948,  7.53063885,  7.57946459,  7.58668144,  7.72668467,
        7.71075226,  6.39438044,  6.02779201, 12.59969936,  6.97106912,
       13.36800073, 13.40024224,  6.81084724, 11.23201619,  8.83341608,
        6.10774544, 13.09426142,  7.49076899, 13.1399181 ,  8.16714836,
        6.2322769 ,  5.19577146,  8.89301516,  7.06272256, 11.89105259,
       13.25336001, 13.06498554,  7.33350642,  5.39098669, 11.64436684,
       11.11969815,  6.81117802,  4.91038884, 13.97779839,  5.13182506,
        6.53250937,  4.74468406, 10.31886066,  7.89745107, 13.3614598 ,
        6.16512856,  8.86082124, 10.95593729,  6.80559669,  5.52416368,
        7.50917551,  5.14607014,  4.83371341,  5.92281142, 10.19036081,
        5.1805345 ,  6.02789634, 12.35625694, 13.12299145, 12.50996828,
        7.59795445,  7.37902766, 12.27835996,  7.16747878,  7.6374603 ,
       13.95361098,  8.09704052,  8.41148722,  5.80860165,  6.3949043 ,
       10.36844813,  5.07306274,  7.09887557,  5.14124612,  6.46356806,
        5.64464807,  8.74470218,  8.27853076, 10.5575601 ,  5.05444288,
       11.73567759, 13.76047999, 12.30695098, 12.27991991, 13.40850328,
       11.62158554, 11.00812996,  5.05444288,  7.66377554,  6.19633139,
        5.13125513, 10.52357479, 13.89974693,  9.50913849,  7.822854  ,
       12.59969936,  7.5984809 ,  6.33517349, 11.99633819, 10.4977132 ,
       11.70016217,  8.44940661, 13.48809595, 10.4606265 ,  4.91038884,
        6.05113217, 11.22142195, 13.41713661,  7.13501264,  5.10444097,
       13.83139024])}
{'P': tensor([[1.5685e-02, 2.0296e-01, 1.6758e-01,  ..., 1.8779e-05, 1.1303e-06,
         3.2264e-06],
        [4.5819e-04, 1.6961e-02, 3.2504e-01,  ..., 2.8842e-05, 1.1280e-06,
         2.4991e-06],
        [5.3450e-04, 8.4952e-04, 2.4461e-02,  ..., 4.8533e-05, 1.6150e-06,
         3.2236e-06],
        ...,
        [2.0899e-05, 1.9413e-05, 1.9328e-05,  ..., 1.9030e-01, 2.9546e-03,
         4.0039e-05],
        [2.7246e-05, 1.6108e-05, 1.3354e-05,  ..., 1.1997e-06, 1.9571e-02,
         3.4637e-03],
        [5.5827e-03, 2.9625e-03, 2.3152e-03,  ..., 1.8567e-06, 2.4807e-05,
         1.0748e-01]]), 'X': tensor([[  3.3943,   6.5483,   1.0790],
        [  2.3200,   7.1756,   1.1938],
        [  1.8766,   7.3694,   1.1887],
        [  0.9445,   7.6736,   0.5095],
        [  0.6933,   7.7336,   2.1179],
        [  0.3038,   7.8092,   0.4164],
        [  9.5261,  -7.1675,   1.4206],
        [  9.1820,   9.7334,  -2.7932],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.3360,   0.4258,  -9.2509],
        [ -0.3432,   7.8900,   7.1805],
        [  2.3200,   7.1756,   1.1938],
        [  9.3753,   9.5171,   3.9131],
        [ 11.9019,  -3.1379,  -3.4534],
        [  4.2612, -10.5514,  -6.6268],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.8889,  -1.3916,   8.8993],
        [ -9.4739,  -1.2497,   0.5483],
        [ 11.4743,   6.2478,  -2.1100],
        [  2.1625,  -4.6697,   7.0658],
        [ -2.0445,   7.8469,  -0.5857],
        [  7.8425,  -8.7202,   6.4530],
        [  4.8674,  12.8875,   6.7622],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.4351,  -1.8891,   3.5642],
        [ -8.0330,  -6.0765,   6.7945],
        [ -7.2515,   5.0195,   4.6434],
        [  8.5198,  10.4091,  -8.6486],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -9.3705,   0.4208,  -3.5815],
        [  1.5383, -11.0274,  -4.9666],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4196,   0.0478,   9.4140],
        [  1.9505,  -4.7171,   9.9868],
        [  6.2247,  -0.3150,  -9.3657],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.6078, -10.9229,   6.2869],
        [ 10.9394,   7.3069,  -8.9365],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -5.1974,   6.7396,   3.3484],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.1755,   2.5233,  -2.0042],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.0604,   1.7707,   1.1066],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.9503,  -4.7172,  -6.2313],
        [ -9.4716,  -0.6673,   7.6068],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.3900,   3.3727,   3.1650],
        [  5.8979,   3.3958,  -3.8713],
        [  8.7131,  10.2216,   7.7915],
        [ -0.5739,   7.9056,   0.6382],
        [  4.9531,   5.0576,   5.6794],
        [  2.2849, -10.9655,  -7.6542],
        [ -5.3736,   6.6310,  -1.5174],
        [  1.1139,   7.6281,  -2.5680],
        [ 12.4690,  -0.8528,  -9.3374],
        [  9.2692,   9.6370,   7.0746],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.6363, -11.0348,   6.1414],
        [ -8.1426,   3.8048,   9.3542],
        [  1.7815,  -4.7481,   4.4289],
        [ -0.5385,   7.9037,   6.3830],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.4203,  -2.0340,   2.5856],
        [  5.4336,  -2.2676,   5.6933],
        [  2.7976,  -4.4682,  -4.2020],
        [ 11.5966,   5.9666,   6.0083],
        [  0.7275,  14.0667,  -0.9815],
        [ -3.9508,  -9.8515,   1.7923],
        [ -5.3736,   6.6310,  -1.5174],
        [  3.3167,   6.6022,   6.3554],
        [ -7.2515,   5.0195,   4.6434],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 11.9808,   4.9340,  -3.3715],
        [  1.6232,  -4.7717,  -2.5655],
        [  6.0890,  -0.8382,   2.9166],
        [ -4.3963,   7.1629,   4.5263],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.3924, -10.1396,  -4.0863],
        [  4.3459,  -3.5327,   0.3667],
        [  4.9337,  12.8566,  -3.6582],
        [ -6.5295,  -7.9785,  -7.4429],
        [  6.3534,   1.2990,  -7.2942],
        [ -1.7290,   7.8825,  -3.3831],
        [ -8.5900,   2.9767,   7.4811],
        [  6.9693,  -9.3200,  -8.3822],
        [ 10.9931,   7.2107,   7.7933],
        [  6.3211,   1.7017,  -0.2164],
        [ 12.3453,  -1.5462,   7.0212],
        [  6.3606,   0.9204,   3.2525],
        [  2.7295,  13.6703,  -8.9328],
        [  0.1287, -11.0065,   6.5757],
        [  0.9074,  -4.8139,   8.1674],
        [ -1.8367,   7.8718,  -8.6523],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.7218,  -9.9934,   6.5764],
        [  9.3753,   9.5171,   3.9131],
        [  4.6231,   5.4556,   8.4973],
        [  5.1886,  -2.6250,   6.4677],
        [  1.7122,  -4.7591,  -3.6734],
        [ -9.3810,  -2.3455,  -9.5194],
        [  4.9452, -10.3187,  -7.8621],
        [ 11.3367,  -4.4806,   0.4643]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0376, 0.0634,  ..., 1.1200, 0.6683, 0.0732],
        [0.0376, 0.0000, 0.0260,  ..., 1.0898, 0.6552, 0.0960],
        [0.0634, 0.0260, 0.0000,  ..., 1.0670, 0.6434, 0.1138],
        ...,
        [1.1200, 1.0898, 1.0670,  ..., 0.0000, 0.6566, 1.1100],
        [0.6683, 0.6552, 0.6434,  ..., 0.6566, 0.0000, 0.6220],
        [0.0732, 0.0960, 0.1138,  ..., 1.1100, 0.6220, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.37576154,  7.54127425,  7.60463349,  7.73151309,  7.76457095,
        7.81509631, 11.92133129, 13.38091121,  6.19633139,  6.35029274,
        7.89745107,  7.54127425, 13.35927527, 12.30858677, 11.37939695,
       12.0495521 ,  6.05113217,  9.55593374, 13.06498554,  5.14607014,
        8.10886021, 11.72802491, 13.77604293, 10.01804223,  9.62238828,
       10.07238463,  8.81930525, 13.45124917, 12.53285633,  9.37989631,
       11.1341798 ,  5.91854397,  9.41970031,  5.10446653,  6.23262289,
       10.8207297 , 11.22993218, 13.15525803,  9.85330646,  8.51088913,
        6.27905292, 10.06850677,  6.67108662,  8.1137452 ,  9.23178204,
        9.38232121,  5.10444097,  9.49511524, 10.5575601 ,  9.04255655,
        6.80559669, 13.43127131,  7.92644746,  7.07902036, 11.20100795,
        8.53501777,  7.70897675, 12.4980845 , 13.37121923, 13.06498554,
       11.05317597,  8.98765644,  5.0713364 ,  7.92200342, 11.97238624,
        9.63742669,  5.88782913,  5.27178363, 13.04156075, 14.08549802,
       10.61417181,  8.53501777,  7.38844703,  8.81930525, 10.52357479,
       12.95702647,  5.04028538,  6.1463896 ,  8.40445872,  5.13616224,
        9.36093562, 11.48434767,  5.60064319, 13.77075174, 10.30972193,
        6.48486171,  8.06990483,  9.09119351, 11.63765273, 13.14690137,
        6.54615887, 12.44177641,  6.42689705, 13.94008948, 11.00726483,
        4.89869205,  8.08320235,  8.92395381, 11.51557166, 13.35927527,
        7.15100303,  5.81481044,  5.05774295,  9.66977802, 11.4424816 ,
       12.18998195])}
{'P': tensor([[2.0919e-02, 2.5350e-01, 2.0223e-01,  ..., 9.6475e-06, 2.2004e-06,
         3.0107e-06],
        [1.3141e-03, 4.6920e-02, 1.0208e-02,  ..., 3.2278e-05, 3.8443e-06,
         5.9024e-06],
        [1.2506e-03, 6.7954e-02, 2.9887e-02,  ..., 1.5552e-05, 2.3533e-06,
         3.8300e-06],
        ...,
        [1.9606e-06, 1.7381e-06, 1.5463e-06,  ..., 1.7322e-02, 7.5728e-05,
         1.0061e-04],
        [6.4522e-04, 4.0209e-04, 3.7911e-04,  ..., 1.9640e-06, 6.3183e-02,
         1.2946e-04],
        [5.5125e-04, 2.9869e-04, 3.4186e-04,  ..., 1.8649e-06, 2.4597e-02,
         6.5279e-02]]), 'X': tensor([[  3.3744,   6.5623,  -6.3636],
        [  2.2860,   7.1916,  -6.1011],
        [  2.5663,   7.0526,  -6.8769],
        [  1.6422,   7.4586,  -6.8056],
        [  2.4932,   7.0903,  -5.4214],
        [  0.3724,   7.7974,  -6.8722],
        [ -8.0608,   3.9362,  -5.9567],
        [  9.0051,   9.9234,   5.4411],
        [  6.3534,   1.2990,  -7.2942],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.5240,  -2.1198,   6.2055],
        [  5.3118,   4.5452,   4.2114],
        [  5.4242, -10.1261,   3.9595],
        [  6.2790,  -0.0259,  -5.4116],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.8836,   3.4315,  -4.0327],
        [ -1.5875,   7.8944,  -9.4286],
        [ -2.4956, -10.4682,   3.5632],
        [ -8.6426,  -4.9136,   9.4941],
        [  2.1771, -10.9776,  -3.2007],
        [  2.7590,  13.6621,   9.4496],
        [  6.1372,   2.6726,  -6.7885],
        [  1.5653,  13.9377,   4.1773],
        [  2.8994,   6.8675,  -9.5987],
        [  9.3836,   9.5075,  -5.9278],
        [ -0.8885,   7.9160,   0.0571],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.3261,   7.1726,   6.2522],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.2675,  12.1379,  -8.9720],
        [  5.2939,  -2.4784,  -4.5840],
        [ -1.9487,   7.8590,  -3.3330],
        [ -7.8126,   4.3066,   5.1791],
        [ -4.5778,  -9.5013,  -5.0667],
        [  0.6470, -11.0352,  -1.1931],
        [ -2.0721,   7.8432,   6.6380],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.5900,   2.9767,   7.4811],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.8275,  -1.5334,   2.0124],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.3963,   7.1629,   4.5263],
        [ 11.7779,   5.5127,  -2.0699],
        [  1.8721,  -4.7322,   0.5524],
        [ -9.1459,   1.4780,  -4.2403],
        [ 12.6059,   0.9597,  -6.3264],
        [  9.9756,  -6.6330,  -3.3429],
        [ -7.7239,   4.4298,   3.1216],
        [  6.0890,  -0.8382,   2.9166],
        [ -7.8280,   4.2848,  -7.1955],
        [ -3.1204, -10.2345,   9.2288],
        [ -9.4159,  -2.0739,   9.6841],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.1886,  -2.6250,   6.4677],
        [  9.6587,   9.1817,  -0.8212],
        [ -8.8835,  -4.3178,   9.3080],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.1140,  -3.7220,  -0.1273],
        [  5.8323,  -1.5228,   5.1132],
        [ -7.2986,  -7.1225,  -3.0814],
        [  4.7739,  -3.1218,  -4.1188],
        [  1.8721,  -4.7322,   0.5524],
        [  0.5175,  14.0910,   4.6540],
        [  8.3791,  10.5409,   5.5590],
        [  4.5491,  -3.3485,   6.8412],
        [  2.8602,   6.8905,   5.8779],
        [ -8.3653,   3.4184,   9.3454],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.3616,  -2.4749,   0.1793],
        [ 12.3357,   3.6001,   0.7903],
        [  0.9036,   7.6840,   9.9237],
        [ -9.0961,   1.6529,   3.1553],
        [  5.8836,   3.4315,  -4.0327],
        [  9.1820,   9.7334,  -2.7932],
        [ -6.8618,  -7.6322,   3.7027],
        [  0.5175,  14.0910,   4.6540],
        [ -3.9700,   7.3453,  -4.1486],
        [  5.9526,   3.2525,  -4.3821],
        [  0.5850,  -4.7982,   9.6639],
        [ -9.0604,   1.7707,   1.1066],
        [ -9.3411,  -2.6004,  -2.7062],
        [  3.4140,  -4.1803,  -8.8224],
        [  4.7156,  -3.1831,  -8.0224],
        [  3.5284, -10.7484,   5.3347],
        [  5.1793,   4.7461,   0.3899],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.8653,   5.1694,  -4.0823],
        [  9.5261,  -7.1675,   1.4206],
        [ -7.6525,  -6.6534,   5.1721],
        [  3.8111, -10.6787,  -4.6539],
        [  8.0381,  -8.5690,  -8.9523],
        [  5.9655,   3.2172,  -7.5953],
        [  6.2535,  -0.1709,  -9.4102],
        [  2.3367,   7.1676,  -9.5263],
        [  6.1984,  -0.4327,   1.5438],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.1459,   1.4780,  -4.2403],
        [ -7.8126,   4.3066,   5.1791],
        [  9.3836,   9.5075,  -5.9278],
        [  5.9655,   3.2172,  -7.5953],
        [  4.0457,  -3.7737,  -0.4704],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.1204,  -5.9308,   4.0761],
        [ 10.4758,  -5.9539,   1.3667],
        [  7.2340,  -9.1505,  -5.8321]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.5121e-02, 4.5751e-02,  ..., 1.0666e+00, 4.6832e-01,
         1.9178e-01],
        [4.5121e-02, 0.0000e+00, 2.6070e-02,  ..., 1.0288e+00, 4.6240e-01,
         1.5636e-01],
        [4.5751e-02, 2.6070e-02, 0.0000e+00,  ..., 1.0486e+00, 4.8805e-01,
         1.4603e-01],
        ...,
        [1.0666e+00, 1.0288e+00, 1.0486e+00,  ..., 3.0339e-04, 8.2467e-01,
         1.0048e+00],
        [4.6832e-01, 4.6240e-01, 4.8805e-01,  ..., 8.2467e-01, 0.0000e+00,
         5.7273e-01],
        [1.9178e-01, 1.5636e-01, 1.4603e-01,  ..., 1.0048e+00, 5.7273e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.37902766,  7.54620849,  7.50500064,  7.63726134,  7.5158489 ,
        7.80625981,  8.97052115, 13.40024224,  6.48486171, 11.54672755,
        5.91677188,  6.99097013, 11.48734978,  6.27905292,  5.53175275,
        6.81117802,  8.05242591, 10.76154256,  9.94173708, 11.1913598 ,
       13.93790235,  6.69389431, 14.02532758,  7.4544866 , 13.35832725,
        7.96575782, 12.41366754,  7.5403822 , 10.52587243, 13.66051884,
        5.84533345,  8.09704052,  8.92098752, 10.54658597, 11.05413545,
        8.11227668,  8.12043983,  9.09119351,  9.78285424,  6.02588517,
       10.60226614,  8.40445872, 13.00413829,  5.08910491,  9.26455795,
       12.64235359, 11.97957425,  8.90405823,  6.1463896 ,  8.92395381,
       10.69963954,  9.64157461, 12.77402351,  5.81481044, 13.32645552,
        9.87720102, 10.31584102,  5.54776934,  6.02779201, 10.1979669 ,
        5.70405525,  5.08910491, 14.10045797, 13.46553623,  5.64865173,
        7.46053281,  9.0368458 ,  7.89745107,  9.68323686, 12.85032836,
        7.73692436,  9.24501829,  6.81117802, 13.38091121, 10.26327895,
       14.10045797,  8.34949383,  6.78325151,  4.83371341,  9.23178204,
        9.69629   ,  5.39723241,  5.68942572, 11.31276109,  7.02496223,
        9.55720651, 10.37998078,  7.09887557, 11.92133129, 10.14044489,
       11.33837396, 11.74901778,  6.77777606,  6.25586297,  7.53884104,
        6.2134961 ,  7.8575648 ,  9.26455795,  8.92098752, 13.35832725,
        6.77777606,  5.53255882, 13.94008948, 10.05558893, 12.0495521 ,
       11.66453059])}
{'P': tensor([[1.6197e-02, 2.1977e-01, 1.9294e-01,  ..., 2.7322e-02, 1.0920e-05,
         3.1824e-05],
        [1.7382e-03, 6.1946e-02, 3.4169e-03,  ..., 1.6357e-01, 3.2974e-05,
         1.1062e-04],
        [1.1454e-03, 2.3076e-01, 2.3105e-02,  ..., 4.2464e-02, 1.2572e-05,
         4.1378e-05],
        ...,
        [4.8597e-04, 7.4614e-04, 6.0426e-04,  ..., 8.2182e-02, 3.3955e-05,
         1.5828e-04],
        [6.5567e-04, 5.8218e-04, 5.3825e-04,  ..., 5.0560e-04, 8.4141e-02,
         5.6285e-04],
        [2.7674e-03, 1.9310e-03, 2.1666e-03,  ..., 7.1403e-04, 6.6747e-03,
         1.0066e-01]]), 'X': tensor([[  3.3167,   6.6022,   6.3554],
        [  2.3261,   7.1726,   6.2522],
        [  2.8474,   6.8979,   6.6457],
        [  2.0729,   7.2878,   7.0880],
        [  2.8602,   6.8905,   5.8779],
        [  1.8320,   7.3871,   5.1452],
        [  1.0093,   7.6567,   2.3526],
        [ 10.4758,  -5.9539,   1.3667],
        [  4.2612, -10.5514,  -6.6268],
        [  7.3724,  -9.0577,  -8.9843],
        [ -0.1626, -10.9796,   6.4629],
        [  6.1730,  -0.5375,  -8.8861],
        [ -5.1046,  -9.1610,   2.3418],
        [  4.4633,   5.6287,  -0.7754],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3550,   0.7129,   0.4917],
        [ -3.1204, -10.2345,   9.2288],
        [  9.5261,  -7.1675,   1.4206],
        [  6.7792,  11.8084,   1.6251],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.4639, -10.9431,  -7.3718],
        [  8.4691,  10.4571,  -5.0650],
        [ -8.5900,   2.9767,   7.4811],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.3384,  -2.6161,  -4.7630],
        [  4.2096,   5.8815,   1.9230],
        [  2.3367,   7.1676,  -9.5263],
        [  1.1274,  14.0116,   8.1753],
        [ -1.7290,   7.8825,  -3.3831],
        [ -9.4430,  -0.1897,   4.8383],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.9217,  -6.7006,  -5.6728],
        [ 12.2692,  -1.8882,   3.9253],
        [  5.8930,  12.3592,  -8.9201],
        [  6.6319,  11.9066,   3.5165],
        [ -1.2454,   7.9127,  -0.5715],
        [ -7.8359,   4.2735,   7.2943],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.6603,   2.8250,   5.9181],
        [ 10.9916,   7.2133,  -0.9179],
        [ -7.5401,   4.6718,   9.0033],
        [  0.3354,  -4.7708,  -4.9125],
        [ -4.0633,  -9.7927,  -3.0825],
        [  2.1440, -10.9811,   8.5281],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -2.0445,   7.8469,  -0.5857],
        [  9.3607,   9.5337,  -4.5412],
        [ 12.2435,   4.0076,   8.1257],
        [  0.4119,  14.1019,  -4.1870],
        [  5.5424,  -2.0885,   7.2487],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.1372,   2.6726,  -6.7885],
        [ -8.3900,   3.3727,   3.1650],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3516,   0.6425,   0.3425],
        [  6.1066,  -0.7799,  -4.1746],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.6675,  -4.8569,  -9.1769],
        [  5.7845,   3.6652,   0.9175],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.3419,   0.5972,  -1.2841],
        [ -4.0868,   7.2981,   8.1874],
        [ -8.1666,   3.7653,   3.6075],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.6928,  12.4710,   3.6001],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.2365,  -0.2578,  -6.0212],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.7151,  -1.7700,   3.1542],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.0829,  -5.9940,   5.0564],
        [  7.7794,  11.0629,   8.9069],
        [  3.7121,   6.3117,   0.6738],
        [ -9.4203,  -2.0340,   2.5856],
        [ -0.1626, -10.9796,   6.4629],
        [  2.8890,  -4.4316,   1.1600],
        [  0.1497,   7.8334,  -5.4412],
        [  6.2022,   2.4092,   3.0349],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.5900,   2.9767,   7.4811],
        [  5.0190,   4.9705,  -7.9752],
        [  8.6420,  -8.0580,   6.5781],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -1.0012, -10.8576,   2.2867],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.2939,  -2.4784,  -4.5840],
        [ -7.5401,   4.6718,   9.0033],
        [  4.9452, -10.3187,  -7.8621],
        [  4.8051, -10.3703,   0.5268],
        [ -9.4753,  -0.8012,  -5.2050],
        [  0.4996,   7.7738,  -4.4711],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.7121,   6.3117,   0.6738],
        [ -7.8126,   4.3066,   5.1791],
        [  9.3624,   9.5318,   0.2779],
        [  3.2990,  -4.2415,  -7.4857],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.0729,   7.2878,   7.0880],
        [ -5.1398,  -9.1367,   6.2074],
        [ -1.6086,   7.8928,   4.3385],
        [  6.2365,  -0.2578,  -6.0212],
        [  3.4604,  -4.1545,   1.8091]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.5956e-02, 2.9876e-02,  ..., 2.3602e-01, 8.1911e-01,
         3.6106e-01],
        [5.5956e-02, 0.0000e+00, 3.3469e-02,  ..., 1.8892e-01, 8.2850e-01,
         3.7377e-01],
        [2.9876e-02, 3.3469e-02, 0.0000e+00,  ..., 2.2189e-01, 8.3762e-01,
         3.8040e-01],
        ...,
        [2.3602e-01, 1.8892e-01, 2.2189e-01,  ..., 0.0000e+00, 7.6659e-01,
         3.5763e-01],
        [8.1911e-01, 8.2850e-01, 8.3762e-01,  ..., 7.6659e-01, 0.0000e+00,
         4.5818e-01],
        [3.6106e-01, 3.7377e-01, 3.8040e-01,  ..., 3.5763e-01, 4.5818e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.38844703,  7.5403822 ,  7.46248947,  7.57685908,  7.46053281,
        7.61088125,  7.72291946, 12.0495521 , 11.37939695, 11.67875855,
       10.98076873,  6.19633139, 10.48721382,  7.18354885,  7.60529892,
        6.3949043 , 10.69963954, 11.92133129, 13.61600536, 10.5575601 ,
       11.21703654, 13.45642379,  9.09119351,  9.41244865,  9.69792513,
        7.23276582,  7.53884104, 14.05687815,  8.06990483,  9.44486609,
        5.18277858, 11.97238624, 12.41366754, 13.6922425 , 13.62896593,
        8.01009083,  8.92549752,  6.00663314,  9.10945678, 13.1471313 ,
        8.87007495,  4.78258092, 10.60226614, 11.18839169,  9.60080639,
        8.10886021, 13.36092624, 12.88270018, 14.10796466,  5.92281142,
        9.75401449,  6.69389431,  9.04255655, 11.33837396,  6.38399789,
        6.1561646 ,  8.19703512,  9.93553424,  6.84796202, 11.80092578,
        9.58072286,  6.01356004,  9.36093562,  8.36445758,  8.99276564,
       12.32408516, 13.70893492,  8.20436764,  6.24187044,  9.64157461,
        5.98285319, 12.25806559,  9.12220271, 10.06286074, 13.52430176,
        7.32233733,  9.63742669, 10.98076873,  5.29009836,  7.83486943,
        6.65368551,  6.27905292,  6.652742  ,  9.09119351,  7.06372172,
       11.8159254 , 12.27835996, 10.9036226 , 12.41012046,  5.84533345,
        8.87007495, 11.4424816 , 11.42947632,  9.50913849,  7.78980299,
       12.91143491,  7.32233733,  8.92098752, 13.3607379 ,  5.37343173,
       12.7125079 ,  7.57685908, 10.48315436,  8.0550286 ,  6.24187044,
        5.40689332])}
{'P': tensor([[1.5252e-02, 2.0401e-01, 1.8452e-01,  ..., 8.9566e-07, 2.3474e-04,
         7.5976e-06],
        [6.9489e-04, 1.9903e-02, 1.3007e-01,  ..., 1.5352e-06, 2.9413e-04,
         7.1854e-06],
        [9.8925e-04, 3.2388e-03, 3.7174e-02,  ..., 2.7010e-06, 3.9875e-04,
         9.4702e-06],
        ...,
        [1.1956e-06, 1.1821e-06, 9.9207e-07,  ..., 7.4233e-02, 1.1437e-05,
         2.7260e-06],
        [2.0254e-04, 2.9625e-04, 2.8466e-04,  ..., 1.2476e-04, 4.9057e-02,
         4.2014e-05],
        [1.2418e-04, 1.5959e-04, 1.3524e-04,  ..., 1.7866e-06, 1.0781e-02,
         2.8039e-02]]), 'X': tensor([[  3.3069,   6.6089,   8.7377],
        [  2.5620,   7.0549,   8.4825],
        [  2.1477,   7.2550,   8.9688],
        [  1.9204,   7.3518,   8.4781],
        [  1.9241,   7.3503,   7.9004],
        [  0.9036,   7.6840,   9.9237],
        [  0.3493,  -4.7727,  -6.5682],
        [ -5.9872,   6.2037,  -9.4477],
        [  9.3424,  -7.3685,  -3.6807],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -5.0550,   6.8231,  -9.2502],
        [ -1.9714,   7.8563,  -1.8084],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -0.7599, -10.8995,  -6.0366],
        [ 12.5763,   0.1333,  -9.2650],
        [  4.3377,   5.7570,   1.3712],
        [ 10.0619,   8.6627,  -9.2884],
        [  0.9036,   7.6840,   9.9237],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.1644, -11.0394,   7.8511],
        [  1.4489,   7.5256,   4.4529],
        [  0.9788,  -4.8145,  -3.8661],
        [ -5.3736,   6.6310,  -1.5174],
        [  4.2096,   5.8815,   1.9230],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.7409,  -4.6837,  -6.5840],
        [  6.0467,  -9.8369,  -6.5816],
        [ 10.1503,   8.5414,  -7.2681],
        [ -0.9715, -10.8630,   0.3555],
        [ -6.2668,   5.9808,   2.5614],
        [ -9.4714,  -1.3301,  -3.1998],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.2535,  -0.1709,  -9.4102],
        [  1.8320,   7.3871,   5.1452],
        [  8.8474,  10.0868,  -8.8644],
        [  1.1994,   7.6035,   2.4295],
        [ -0.1963,   7.8765,  -1.1371],
        [  5.9026,   3.3837,  -0.6573],
        [  2.1440, -10.9811,   8.5281],
        [  2.9470,  13.6085,   2.9809],
        [  4.0457,  -3.7737,  -0.4704],
        [ -6.7992,   5.4991,   9.9186],
        [  6.9956,  11.6590,  -7.6876],
        [  5.6202,  -1.9508,   0.7471],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.0534,  -6.0429,   2.4583],
        [  4.5800,   5.5034,  -3.9432],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.2159, -10.8166,  -1.0406],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.6248,   3.9984,   8.4933],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.8154,   2.4611,   2.5198],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -6.3306,   5.9272,   5.6935],
        [  2.4572,   7.1085,   7.7376],
        [  0.6470, -11.0352,  -1.1931],
        [  4.2612, -10.5514,  -6.6268],
        [  3.7298,  13.3537,  -9.4653],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -1.7009,   7.8851,  -8.2437],
        [  8.5198,  10.4091,  -8.6486],
        [  4.3523, -10.5232,   8.2138],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.3589,   0.8239,  -8.4230],
        [  4.3554,   5.7393,  -9.0766],
        [  4.5503,   5.5359,   7.4735],
        [  6.2576,   2.1359,  -3.3398],
        [  0.3493,  -4.7727,  -6.5682],
        [  3.5284, -10.7484,   5.3347],
        [ -9.2687,  -2.9794,   0.6436],
        [  1.2273,  -4.8081,   9.8129],
        [  8.0381,  -8.5690,  -8.9523],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.7992,  -3.9474,  -8.9019],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.3413,   0.4883,  -6.8416],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.1178,   1.5784,   2.1410],
        [  3.7237,  -3.9966,  -5.0376],
        [ -4.8751,  -9.3148,   4.0406],
        [ -0.1963,   7.8765,  -1.1371],
        [  8.3718,  -8.2953,   8.4666],
        [  0.3715,  -4.7756,   0.2312],
        [  3.6469,   6.3624,  -1.8414],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -5.0943,  -9.1682,   0.5073],
        [  6.1929,   2.4501,  -6.9352],
        [  9.3424,  -7.3685,  -3.6807],
        [  5.1292,   4.8183,  -5.7997],
        [ -6.3489,   5.9117,   6.5161],
        [ 11.2767,  -4.6006,   0.7868],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -1.5451,   7.8975,   7.6104],
        [  4.9152,  12.8653,  -1.4459],
        [ -8.4297,  -5.3637,  -9.3962],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.2712,   8.3707,  -4.9743]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.1746e-02, 6.6936e-02,  ..., 1.4328e+00, 7.1143e-01,
         1.0498e+00],
        [3.1746e-02, 0.0000e+00, 3.6074e-02,  ..., 1.4113e+00, 7.0622e-01,
         1.0598e+00],
        [6.6936e-02, 3.6074e-02, 2.1453e-04,  ..., 1.3972e+00, 7.1360e-01,
         1.0822e+00],
        ...,
        [1.4328e+00, 1.4113e+00, 1.3972e+00,  ..., 0.0000e+00, 9.1600e-01,
         1.2962e+00],
        [7.1143e-01, 7.0622e-01, 7.1360e-01,  ..., 9.1600e-01, 0.0000e+00,
         5.3986e-01],
        [1.0498e+00, 1.0598e+00, 1.0822e+00,  ..., 1.2962e+00, 5.3986e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.39004226,  7.50564223,  7.56617037,  7.5984809 ,  7.59795445,
        7.73692436,  4.78544875,  8.62162372, 11.89855267,  9.75401449,
        8.49161154,  8.09984231,  9.51074763, 10.92597196, 12.5769659 ,
        7.20826219, 13.27717982,  7.73692436, 13.20988996, 11.10066126,
        7.66377554,  4.91296205,  8.53501777,  7.23276582, 13.06498554,
        6.02588517,  9.91668182, 11.54672755, 13.26590071, 10.90638066,
        8.66273171,  9.5642993 ,  5.05981914,  9.41605684,  6.25586297,
        7.61088125, 13.41713661,  7.69752262,  7.8788965 ,  6.80370254,
       11.18839169, 13.92390689,  5.53255882,  8.74470218, 13.59673406,
        5.94909709,  8.43626114, 10.06850677,  7.15990676,  8.29022197,
        9.9353921 , 11.2845608 ,  9.84314712,  6.90115901, 12.0495521 ,
        9.15253344, 12.11735334, 12.3828957 ,  9.75075273,  8.67227386,
        7.52117132, 11.05413545, 11.37939695, 13.86480175, 12.1460931 ,
       12.42554808,  8.06643487, 13.45124917, 11.38774149, 10.67322527,
       10.00794468,  6.41202986,  7.20482224,  7.16598273,  6.61211224,
        4.78544875, 11.31276109,  9.73579738,  4.96231532, 11.74901778,
        8.22728399,  5.47866221,  5.39450305,  6.36004393, 13.77222752,
        9.25336371,  5.46244988, 10.51340829,  7.8788965 , 11.785566  ,
        4.79003361,  7.33350642, 10.00410625, 10.48840476,  6.65991877,
       11.89855267,  7.03733775,  8.67502348, 12.17901033,  9.97822473,
       12.12448912,  8.04718433, 13.77222752,  9.99145096,  7.71075226,
       13.25017211])}
{'P': tensor([[2.4077e-02, 3.5603e-01, 1.8144e-01,  ..., 5.4807e-05, 3.7751e-02,
         1.6996e-05],
        [1.2777e-03, 3.3939e-02, 3.4534e-01,  ..., 6.6026e-05, 7.2846e-02,
         2.5381e-05],
        [9.6957e-04, 1.2451e-03, 6.4528e-02,  ..., 7.5571e-05, 1.8312e-01,
         4.9303e-05],
        ...,
        [9.4049e-03, 8.7462e-03, 4.9441e-03,  ..., 1.4451e-01, 4.1447e-03,
         9.3660e-04],
        [1.3604e-03, 1.7280e-03, 4.6199e-03,  ..., 2.3354e-04, 2.3172e-01,
         3.8802e-04],
        [7.5237e-05, 6.9411e-05, 4.5534e-05,  ..., 5.5069e-03, 5.2538e-05,
         4.3145e-02]]), 'X': tensor([[  3.2701,   6.6338,   9.9209],
        [  2.6970,   6.9826,   9.7350],
        [  0.9036,   7.6840,   9.9237],
        [  2.1477,   7.2550,   8.9688],
        [ -0.0410,   7.8591,   9.3457],
        [  1.9204,   7.3518,   8.4781],
        [ 12.6059,   0.9597,  -6.3264],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.1041, -11.0045,   2.4311],
        [ 10.4994,  -5.9193,  -7.5153],
        [  8.8091,  -7.9037,   0.3631],
        [  0.9945,  -4.8144,  -0.1897],
        [  0.6396,  14.0772,  -6.3786],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -1.7290,   7.8825,  -3.3831],
        [  6.0054,  -1.0902,   8.3079],
        [  0.1041, -11.0045,   2.4311],
        [  2.8602,   6.8905,   5.8779],
        [  4.7156,  -3.1831,  -8.0224],
        [ 10.6673,  -5.6649,  -4.1646],
        [  4.4290,  -3.4596,   1.4181],
        [  1.9505,  -4.7171,   9.9868],
        [  1.1644, -11.0394,   7.8511],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.9461,  -4.4077,  -0.4831],
        [ 11.5083,  -4.1177,   0.2389],
        [  6.3535,   0.6792,  -7.5101],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.3523, -10.5232,   8.2138],
        [ 12.2592,   3.9425,   2.0383],
        [ 10.9394,   7.3069,  -8.9365],
        [  0.3354,  -4.7708,  -4.9125],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.6954,  -4.7616,   9.8198],
        [  1.8719,   7.3713,  -3.7465],
        [ 11.9808,   4.9340,  -3.3715],
        [ 11.3691,   6.4760,   1.0787],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.1984,  -0.4327,   1.5438],
        [  4.5491,  -3.3485,   6.8412],
        [ -9.1459,   1.4780,  -4.2403],
        [  9.6614,   9.1784,  -1.2087],
        [  6.7565,  11.8237,  -4.9882],
        [ -6.3066,  -8.1936,   9.5822],
        [  6.3211,   1.7017,  -0.2164],
        [  2.1440, -10.9811,   8.5281],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.1999,  -3.6543,  -4.1998],
        [ -2.5056,   7.7720,   5.5878],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.4731,  -0.7120,  -6.0271],
        [  6.3205,   1.7068,  -3.8619],
        [  6.8704,  -9.3809,   7.5769],
        [ -7.9925,  -6.1423,   2.8990],
        [  1.4749,  -4.7892,  -0.8089],
        [ -8.4824,  -5.2575,  -0.0795],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.4541,  -0.3327,   8.8246],
        [  4.5814, -10.4485,   2.5496],
        [  9.0051,   9.9234,   5.4411],
        [  1.9312,  13.8649,   8.3563],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.9085,   6.1512,  -2.2560],
        [  6.3569,   0.7591,   6.5011],
        [ -9.3464,   0.5708,  -1.6397],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.7151,  -1.7700,   3.1542],
        [  1.4150,  13.9647,  -4.0915],
        [  8.6271,  -8.0715,  -7.4813],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.1178,   1.5784,   2.1410],
        [  1.0000,   7.6591,  -4.5011],
        [  6.9956,  11.6590,  -7.6876],
        [  5.8902,  -1.3887,   4.0388],
        [ -8.4887,  -5.2448,  -0.6585],
        [  6.1741,  -0.5329,   3.1913],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.3497,   1.3680,  -1.9076],
        [  6.2026,  12.1774,  -1.5887],
        [ 10.1503,   8.5414,  -7.2681],
        [ 12.4690,  -0.8528,  -9.3374],
        [  6.0073,   3.0986,   6.7937],
        [  6.1575,  12.2046,   3.3257],
        [  4.5491,  -3.3485,   6.8412],
        [ -3.3116, -10.1539,   7.5919],
        [ -7.7239,   4.4298,   3.1216],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.8474,   6.8979,   6.6457],
        [  2.2373,  -4.6506,  -1.7334],
        [ -0.8885,   7.9160,   0.0571],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.9655,   3.2172,  -7.5953],
        [  2.5010,  -4.5736,  -0.8989],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.7151,  -1.7700,   3.1542],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.0087,   4.0173,   9.1091],
        [  3.6249, -10.7255,   5.6605],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.2770,  -0.0379,   2.4195],
        [ -1.6906,   7.8860,   7.7142],
        [  0.2732,  -4.7619,  -1.1577]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0271, 0.0961,  ..., 0.4221, 0.1417, 0.6510],
        [0.0271, 0.0000, 0.0690,  ..., 0.4344, 0.1153, 0.6531],
        [0.0961, 0.0690, 0.0000,  ..., 0.4744, 0.0537, 0.6674],
        ...,
        [0.4221, 0.4344, 0.4744,  ..., 0.0000, 0.4761, 0.3034],
        [0.1417, 0.1153, 0.0537,  ..., 0.4761, 0.0000, 0.6462],
        [0.6510, 0.6531, 0.6674,  ..., 0.3034, 0.6462, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.39599986,  7.48538905,  7.73692436,  7.56617037,  7.85919687,
        7.5984809 , 12.64235359, 10.5575601 , 11.0050379 , 12.05300943,
       11.83509372,  4.9160962 , 14.09176259, 12.1460931 , 12.52372629,
        8.06990483,  6.10360073, 11.0050379 ,  7.46053281,  5.68942572,
       12.07818871,  5.62005629,  5.10446653, 11.10066126,  7.8788965 ,
        5.30158497, 12.22276278,  6.38968357,  8.5724042 , 11.38774149,
       12.87752052, 13.15525803,  4.78258092, 12.17451685,  5.05444288,
        7.60529892, 12.95702647, 13.08414689,  5.48276038,  6.2134961 ,
        5.64865173,  9.26455795, 13.32613445, 13.61801526, 10.33958639,
        6.54615887, 11.18839169,  5.48276038,  5.5671389 ,  8.16584537,
        6.25586297, 12.3828957 ,  9.4997987 ,  6.54693987, 11.62770332,
       10.08001894,  5.01114719,  9.97964013,  4.89869205,  9.45995955,
       11.4088043 , 13.40024224, 13.99877413, 10.8207297 ,  7.28793663,
        6.4020404 ,  9.36378438, 12.67680439,  5.98285319, 14.03618649,
       11.81423161,  9.47369628,  7.37902766,  9.25336371,  7.72415048,
       13.59673406,  6.05165248,  9.97822473,  6.19707944, 10.48721382,
        6.49538841, 13.6660671 , 13.26590071, 12.4980845 ,  6.7593917 ,
       13.66991027,  5.64865173, 10.68031351,  8.90405823,  5.05963217,
        7.46248947,  5.16077764,  7.96575782, 10.1979669 , 12.07225469,
        6.77777606,  5.21280156, 11.81423161,  5.98285319, 13.12299145,
        8.95983586, 11.32149259,  4.78258092,  6.27715105,  8.06517143,
        4.76969364])}
{'P': tensor([[1.9829e-02, 3.2860e-01, 1.3464e-01,  ..., 4.6149e-05, 2.3015e-06,
         8.4205e-04],
        [9.7429e-04, 2.8775e-02, 5.8359e-02,  ..., 6.7156e-05, 3.7262e-06,
         7.7032e-04],
        [2.0870e-03, 1.0312e-02, 3.8257e-02,  ..., 7.9967e-05, 4.3513e-06,
         8.0192e-04],
        ...,
        [6.0236e-04, 4.9968e-04, 4.1198e-04,  ..., 8.5416e-02, 3.7312e-04,
         1.8266e-03],
        [3.3643e-06, 2.8575e-06, 2.3173e-06,  ..., 1.9379e-03, 3.9750e-02,
         1.4221e-05],
        [3.5996e-02, 4.2798e-02, 3.0121e-02,  ..., 3.5238e-04, 2.1934e-05,
         9.5240e-02]]), 'X': tensor([[ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9794e-02, 5.5778e-02,  ..., 3.1976e-01, 1.1669e+00,
         2.8100e-01],
        [2.9794e-02, 0.0000e+00, 3.3460e-02,  ..., 3.2325e-01, 1.1514e+00,
         2.9190e-01],
        [5.5778e-02, 3.3460e-02, 0.0000e+00,  ..., 3.5402e-01, 1.1663e+00,
         3.2496e-01],
        ...,
        [3.1976e-01, 3.2325e-01, 3.5402e-01,  ..., 0.0000e+00, 9.2755e-01,
         8.5415e-02],
        [1.1669e+00, 1.1514e+00, 1.1663e+00,  ..., 9.2755e-01, 3.0339e-04,
         1.0122e+00],
        [2.8100e-01, 2.9190e-01, 3.2496e-01,  ..., 8.5415e-02, 1.0122e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.42512621,  7.5158489 ,  7.54620849,  7.72415048,  7.83486943,
        7.78980299,  6.10360073, 13.58101794,  5.5205189 , 10.78533838,
        7.14837514, 13.61801526,  4.85995002,  8.23525433,  5.29009836,
       12.11735334,  8.27853076, 13.1471313 , 12.46513646, 12.71843462,
        4.78258092, 10.80728788, 11.48663767, 10.95593729,  5.48276038,
       14.02532758, 12.00995898,  9.43737523,  8.14433542, 10.91805683,
        5.14999799,  7.50564223,  5.43943791, 11.81423161, 13.09737053,
        7.53884104,  9.97964013,  9.99145096,  8.68979052,  6.42689705,
       12.64235359,  5.8828986 ,  8.50278896,  8.14433542,  5.18047638,
        6.42689705,  8.87587457, 13.43127131,  6.18668463, 11.85890664,
        8.36445758,  6.28835101, 13.3614598 , 11.80092578,  5.24461068,
        8.16584537,  5.64500512,  6.2134961 ,  7.5984809 , 14.05687815,
       11.80092578, 13.09737053,  7.32233733, 10.35464916, 10.33712258,
        9.49511524,  5.83560009, 12.45418236,  9.43112957, 11.28215429,
       12.52875793, 12.05300943,  9.60080639, 10.66631784,  5.54776934,
        6.61386436, 13.00380624,  6.76056625,  6.67121748, 11.83570132,
       11.785566  ,  7.01475376, 11.9702762 ,  5.14124612,  8.08320235,
        6.39438044,  8.99276564,  5.19065439, 13.31764181, 13.00380624,
        8.55982105,  8.85939626,  6.27715105,  6.54615887, 11.8159254 ,
        8.30125201, 11.00508681, 12.32408516,  9.95552626, 11.83075005,
       11.75618513,  8.04483495, 11.35345345,  4.79003361, 10.4606265 ,
        7.18354885])}
{'P': tensor([[1.6783e-02, 2.0564e-01, 1.8780e-01,  ..., 1.3012e-05, 2.4745e-06,
         1.6641e-05],
        [1.0386e-03, 3.7091e-02, 3.9212e-01,  ..., 2.5212e-05, 4.0814e-06,
         5.0643e-05],
        [1.9459e-03, 5.9313e-03, 7.0750e-02,  ..., 4.8331e-05, 7.6507e-06,
         9.8960e-05],
        ...,
        [1.3222e-03, 1.0874e-03, 1.0921e-03,  ..., 1.2228e-01, 5.9179e-03,
         1.3561e-05],
        [5.7076e-04, 3.4119e-04, 3.2940e-04,  ..., 5.3930e-04, 2.2529e-02,
         3.8301e-07],
        [1.3555e-05, 1.0385e-05, 1.0121e-05,  ..., 6.3685e-05, 8.8601e-05,
         4.1855e-02]]), 'X': tensor([[  2.9410,   6.8428,   5.4458],
        [  1.8320,   7.3871,   5.1452],
        [  1.7301,   7.4262,   5.0522],
        [  1.7659,   7.4127,   4.7871],
        [  1.4489,   7.5256,   4.4529],
        [  1.7500,   7.4187,   4.4237],
        [  1.1139,   7.6281,  -2.5680],
        [  2.7121,   6.9743,  -8.4202],
        [  5.5626,  -2.0536,   8.4433],
        [ -5.9872,   6.2037,  -9.4477],
        [ -9.2345,  -3.1345,   3.9715],
        [  3.2159, -10.8166,  -1.0406],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.3119,   1.7797,   9.9186],
        [  5.5101,  -2.1433,   9.5442],
        [  2.3200,   7.1756,   1.1938],
        [ 11.1085,  -4.9204,   0.0736],
        [  1.7000,  -4.7609,   2.7935],
        [ -1.5451,   7.8975,   7.6104],
        [  3.9022,  13.2906,   5.5788],
        [ -6.6998,   5.5954,   0.0823],
        [ -8.7544,   2.6096,   3.3146],
        [  4.2612, -10.5514,  -6.6268],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.4758,  -5.9539,   1.3667],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.2096,   5.8815,   1.9230],
        [ -8.0608,   3.9362,  -5.9567],
        [  3.3094,  13.4969,  -3.6507],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 11.0275,  -5.0666,   8.4154],
        [ -0.0410,   7.8591,   9.3457],
        [ -6.0844,  -8.3955,   0.7705],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -2.9497, -10.3028,  -7.3479],
        [  2.1771, -10.9776,  -3.2007],
        [  4.8653,   5.1694,  -4.0823],
        [  5.4242, -10.1261,   3.9595],
        [  5.9843,  -9.8679,  -6.1449],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.9284,  11.7061,  -0.3732],
        [  2.3200,   7.1756,   1.1938],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -3.9508,  -9.8515,   1.7923],
        [  7.0171,  11.6439,  -5.0178],
        [  5.0277,   4.9589,   7.0175],
        [  3.9663,   6.1018,  -5.4104],
        [ -2.4956, -10.4682,   3.5632],
        [  2.3928,   7.1404,  -1.7035],
        [  2.6970,   6.9826,   9.7350],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.4429,  -2.2529,   9.7306],
        [  4.7482, -10.3907,   5.9149],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.5600, -11.0321,  -0.8660],
        [  1.7500,   7.4187,   4.4237],
        [ -5.3328,  -8.9992,   7.3440],
        [  7.5309,  11.2617,   7.2656],
        [  2.5375,   7.0676,  -4.3490],
        [ -4.9217,   6.8979,  -1.7667],
        [  3.4401,  13.4540,   4.6576],
        [  0.1047, -11.0046,   0.5188],
        [  5.9548,  -1.2273,  -2.6520],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.3357,   3.6001,   0.7903],
        [  0.9445,   7.6736,   0.5095],
        [  0.8940, -11.0402,  -0.9783],
        [ -8.6050,   2.9451,  -0.7012],
        [ -4.6393,   7.0461,   1.1624],
        [  1.1994,   7.6035,   2.4295],
        [ -8.9033,   2.2329,  -8.8912],
        [ 12.3357,   3.6001,   0.7903],
        [ -3.2053, -10.1993,  -3.8736],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -3.5029,   7.5138,   8.1853],
        [ 12.2692,  -1.8882,   3.9253],
        [ -6.4620,  -8.0449,   6.2719],
        [  0.1047, -11.0046,   0.5188],
        [  1.6308,  13.9254,  -2.0984],
        [  3.3094,  13.4969,  -3.6507],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.7239,   4.4298,   3.1216],
        [  0.1287, -11.0065,   6.5757],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.1231,  -3.5739,   8.4708],
        [  7.5784,  -8.9141,   5.8410],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.2630,   2.1057,  -5.2136],
        [  8.7131,  10.2216,   7.7915],
        [ 10.7566,   7.6212,  -2.3279],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.4733,   4.7555,   6.2490],
        [  5.6414, -10.0303,   7.8525],
        [  4.5430,   5.5438,  -3.8669],
        [  1.6422,   7.4586,  -6.8056],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.1861,   4.7361,  -3.7312],
        [ -8.7544,   2.6096,   3.3146],
        [  5.2939,  -2.4784,  -4.5840],
        [ 11.5083,  -4.1177,   0.2389],
        [ -6.7745,  -7.7264,   8.4684]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 5.2504e-02, 6.4209e-02,  ..., 7.1124e-01, 4.2430e-01,
         8.1132e-01],
        [5.2504e-02, 0.0000e+00, 1.2060e-02,  ..., 7.1730e-01, 4.3251e-01,
         7.6169e-01],
        [6.4209e-02, 1.2060e-02, 0.0000e+00,  ..., 7.1600e-01, 4.3205e-01,
         7.5185e-01],
        ...,
        [7.1124e-01, 7.1730e-01, 7.1600e-01,  ..., 0.0000e+00, 2.8708e-01,
         1.2706e+00],
        [4.2430e-01, 4.3251e-01, 4.3205e-01,  ..., 2.8708e-01, 0.0000e+00,
         1.0571e+00],
        [8.1132e-01, 7.6169e-01, 7.5185e-01,  ..., 1.2706e+00, 1.0571e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.44805901,  7.61088125,  7.62509582,  7.62011734,  7.66377554,
        7.62232307,  7.70897675,  7.48311002,  5.92952538,  8.62162372,
        9.75200848, 11.2845608 , 13.1471313 ,  6.55800267,  5.91222346,
        7.54127425, 12.1494023 ,  5.05534609,  8.04718433, 13.85158417,
        8.72896381,  9.13507321, 11.37939695, 10.09979747, 10.38897563,
       12.0495521 , 10.37998078,  7.23276582,  8.97052115, 13.89671314,
       10.60226614, 12.13568598,  7.85919687, 10.36844813,  7.27340834,
        9.71392179, 10.7167285 , 11.1913598 ,  7.09887557, 11.48734978,
       11.54071859, 12.18998195, 13.60275211,  7.54127425,  9.96152788,
       10.31613269, 10.61417181, 13.59480906,  7.06169391,  7.27759233,
       10.76154256,  7.53063885,  7.48538905,  5.84533345,  5.89074104,
       11.42420508, 13.00380624, 11.04629471,  7.62232307, 10.4606265 ,
       13.54774355,  7.5092834 ,  8.47371868, 13.88683436, 11.00508681,
        6.07992623,  8.68832398, 12.85032836,  7.73151309, 11.07637531,
        9.09502385,  8.43626114,  7.69752262,  9.17905491, 12.85032836,
       10.6910768 , 10.33712258,  8.29022197, 12.41366754, 10.31886066,
       11.00508681, 14.02058621, 13.89671314,  8.85939626,  8.90405823,
       11.00726483, 13.86480175,  9.79814997, 11.70016217,  9.85384623,
        6.25586297, 12.4013153 ,  6.60752373, 13.43127131, 13.18278606,
        8.90405823,  8.8580671 , 11.50791859,  7.16747878,  7.63726134,
       10.39942497,  7.02326515,  9.13507321,  5.84533345, 12.22276278,
       10.27572673])}
{'P': tensor([[2.5149e-02, 4.2753e-01, 1.6773e-01,  ..., 1.6553e-06, 2.9473e-05,
         4.1046e-07],
        [1.1621e-03, 3.3902e-02, 3.0777e-01,  ..., 2.0788e-06, 2.8118e-05,
         5.4496e-07],
        [3.7394e-04, 4.6700e-04, 2.9131e-02,  ..., 1.5291e-06, 7.5071e-06,
         4.7277e-07],
        ...,
        [2.7539e-05, 2.1486e-05, 8.5113e-06,  ..., 1.0904e-02, 5.0050e-04,
         6.0756e-05],
        [2.1776e-03, 2.3990e-03, 2.1577e-03,  ..., 1.0244e-06, 5.5141e-02,
         5.2472e-07],
        [2.9640e-06, 2.4641e-06, 1.2160e-06,  ..., 5.2720e-03, 8.6909e-05,
         1.6501e-02]]), 'X': tensor([[ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0352, 0.1076,  ..., 0.4144, 0.5963, 1.0697],
        [0.0352, 0.0000, 0.0724,  ..., 0.4120, 0.6247, 1.0651],
        [0.1076, 0.0724, 0.0000,  ..., 0.4200, 0.6870, 1.0629],
        ...,
        [0.4144, 0.4120, 0.4200,  ..., 0.0000, 0.5239, 0.6558],
        [0.5963, 0.6247, 0.6870,  ..., 0.5239, 0.0000, 0.9420],
        [1.0697, 1.0651, 1.0629,  ..., 0.6558, 0.9420, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.4544866 ,  7.53884104,  7.83534987,  7.88677538,  8.05242591,
        8.08320235,  8.35856487,  5.83560009, 10.69963954,  5.14124612,
        5.08910491,  8.86082124,  6.32572323,  9.53015376, 12.11735334,
       10.06023665, 12.12448912, 11.70016217,  5.16205087,  8.98765644,
       13.00413829, 13.84126197, 12.10222204, 12.41012046, 11.63502679,
       10.69963954, 10.41909211,  7.50917551,  8.62260327,  5.35774573,
       12.25806559,  9.58072286,  8.1137452 ,  9.78286733,  5.05981914,
       13.86480175,  8.56889738, 14.03618649,  9.60412735,  6.42689705,
       13.13888841, 10.66631784,  8.47371868,  8.14433542, 13.97857591,
        5.52416368, 12.97370371,  7.71075226, 10.8207297 ,  5.08372149,
        9.5014146 , 12.22503754,  5.70405525, 12.21532282,  4.79003361,
       12.88270018, 11.27160035,  9.05843201, 13.94008948, 10.54658597,
        7.17789666, 12.91505111,  5.10444097,  7.6374603 , 14.03505624,
        5.05774295, 13.66896739, 11.69264243, 13.39372426, 10.95593729,
        8.31625171,  7.27340834, 10.51340829,  9.5642993 , 11.50791859,
        9.09963894, 11.28895489, 10.51340829,  5.83560009, 12.10942412,
        5.48276038,  7.59795445, 10.9036226 , 12.21532282,  5.22903117,
        6.38968357,  6.05113217,  5.05444288,  5.84533345,  8.09704052,
        9.43737523,  5.08910491,  9.0368458 , 12.04924159,  9.69792513,
       10.78533838, 12.04924159, 10.26545792,  9.20340031,  9.0368458 ,
       13.06957425,  5.91222346, 11.28215429, 11.75807252, 13.20988996,
       11.48663767])}
{'P': tensor([[2.0268e-02, 2.1411e-01, 1.6006e-01,  ..., 2.3775e-04, 3.4746e-06,
         2.7422e-06],
        [1.7647e-03, 3.3897e-02, 8.0390e-02,  ..., 2.7384e-04, 6.2036e-06,
         4.5030e-06],
        [1.1587e-03, 4.2885e-03, 2.8653e-02,  ..., 2.7215e-04, 4.9957e-06,
         3.5816e-06],
        ...,
        [7.2366e-04, 8.9344e-04, 1.7485e-03,  ..., 1.9130e-01, 5.9712e-07,
         4.9329e-07],
        [4.3362e-05, 3.5455e-05, 2.5919e-05,  ..., 6.1397e-07, 4.0079e-02,
         3.6090e-01],
        [7.8651e-05, 6.0619e-05, 4.4450e-05,  ..., 1.7860e-06, 3.9080e-05,
         2.1588e-02]]), 'X': tensor([[  2.8602,   6.8905,   5.8779],
        [  2.3261,   7.1726,   6.2522],
        [  1.8320,   7.3871,   5.1452],
        [  1.7301,   7.4262,   5.0522],
        [  1.7659,   7.4127,   4.7871],
        [ -0.5385,   7.9037,   6.3830],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.5996,   1.3971,  -1.0694],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.0652,   9.8597,  -3.5949],
        [  2.7295,  13.6703,  -8.9328],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -8.1531,  -5.8748,   2.6934],
        [ -1.8788, -10.6567,   3.7068],
        [ -9.3419,   0.5972,  -1.2841],
        [ -1.5451,   7.8975,   7.6104],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.5663,   7.0526,  -6.8769],
        [ -5.1974,   6.7396,   3.3484],
        [ -4.3963,   7.1629,   4.5263],
        [  1.6422,   7.4586,  -6.8056],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 10.2470,   8.4054,  -0.1177],
        [  1.3339,   7.5627,  -0.9034],
        [  6.3534,   1.2990,  -7.2942],
        [  5.4167, -10.1293,   6.2156],
        [  6.3516,   0.6425,   0.3425],
        [  3.6249, -10.7255,   5.6605],
        [ -7.3333,   4.9246,  -7.9212],
        [ -9.1749,  -3.3803,   6.4629],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.3759,  -1.3943,   6.4659],
        [ 10.6283,  -5.7253,   7.4878],
        [  9.0051,   9.9234,   5.4411],
        [  4.4915,   5.5990,   7.0405],
        [ -2.5056,   7.7720,   5.5878],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3516,   0.6425,   0.3425],
        [  2.8602,   6.8905,   5.8779],
        [ -9.4671,  -0.5553,   0.1117],
        [ -4.7389,   6.9953,   3.7638],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.1372,   2.6726,  -6.7885],
        [ 12.4690,  -0.8528,  -9.3374],
        [  6.3460,   1.4252,   1.2291],
        [ -1.9600,   7.8577,   6.9121],
        [ 12.3759,  -1.3943,   6.4659],
        [  4.5430,   5.5438,  -3.8669],
        [  4.0362,  13.2397,  -7.0030],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.0449,  13.8402,  -7.1642],
        [  7.7794,  11.0629,   8.9069],
        [  5.9883,   3.1536,   6.2320],
        [ -9.4541,  -0.3327,   8.8246],
        [  7.0358,  -9.2784,  -9.8479],
        [ -4.6393,   7.0461,   1.1624],
        [ -7.6293,  -6.6859,  -5.5161],
        [  0.8940, -11.0402,  -0.9783],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.2804,   2.0019,   0.8607],
        [  5.9003,  12.3550,   2.1917],
        [  9.1409,  -7.5789,   1.5814],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.2022,   2.4092,   3.0349],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -6.1899,  -8.3011,  -7.9269],
        [  8.3718,  -8.2953,   8.4666],
        [  5.1196,  -2.7160,   4.2705],
        [ 12.2435,   4.0076,   8.1257],
        [  4.2096,   5.8815,   1.9230],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -6.3306,   5.9272,   5.6935],
        [  0.3354,  -4.7708,  -4.9125],
        [ 10.9931,   7.2107,   7.7933],
        [  1.2728, -11.0372,   9.8121],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -3.5903,   7.4847,  -8.9112],
        [ -8.8154,   2.4611,   2.5198],
        [  5.7218,  -9.9934,   6.5764],
        [  9.4925,   9.3811,   8.9842],
        [  5.2199,   4.6861,  -1.9304],
        [  2.2659,  13.7894,   6.5436],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.1204,  -5.9308,   4.0761],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.7145,  -4.8071,  -2.3766],
        [ -3.7964,  -9.9294,   7.7147],
        [  2.2860,   7.1916,  -6.1011],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.7685, -10.8982,  -6.8928],
        [ 12.4013,  -1.2597,  -7.6474],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.7792,  11.8084,   1.6251],
        [ 11.4703,  -4.2006,   7.4086],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.7565,  11.8237,  -4.9882],
        [  3.3069,   6.6089,   8.7377],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.2809,  13.5061,  -4.0199],
        [  1.2728, -11.0372,   9.8121],
        [  5.6414, -10.0303,   7.8525]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0380, 0.0578,  ..., 0.8934, 0.4927, 0.3712],
        [0.0380, 0.0000, 0.0580,  ..., 0.9273, 0.4554, 0.3334],
        [0.0578, 0.0580, 0.0000,  ..., 0.8866, 0.4674, 0.3521],
        ...,
        [0.8934, 0.9273, 0.8866,  ..., 0.0000, 1.3183, 1.2237],
        [0.4927, 0.4554, 0.4674,  ..., 1.3183, 0.0000, 0.1282],
        [0.3712, 0.3334, 0.3521,  ..., 1.2237, 0.1282, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.46053281,  7.5403822 ,  7.61088125,  7.62509582,  7.62011734,
        7.92200342, 10.794755  , 12.67680439,  9.66977802, 13.39372426,
       13.94008948, 10.31613269, 10.04916678, 10.82106505,  9.36093562,
        8.04718433,  5.27178363,  7.50500064,  8.51088913,  8.40445872,
        7.63726134,  9.75075273, 13.25336001,  7.67939546,  6.48486171,
       11.48663767,  6.38399789, 11.32149259,  8.83341608,  9.77777346,
        8.43626114, 12.45418236, 12.07225469, 13.40024224,  7.17789666,
        8.16584537,  6.02588517,  6.38399789,  7.46053281,  9.48337063,
        8.44940661, 10.1979669 ,  5.01114719,  6.69389431, 12.4980845 ,
        6.50409372,  8.09842704, 12.45418236,  7.16747878, 13.84126197,
        5.24461068, 13.99048094, 13.52430176,  6.76791483,  9.45995955,
       11.64436684,  8.43626114, 10.14436912, 11.07637531, 12.12448912,
        6.59176018, 13.69162795, 11.87412439, 10.06407839, 10.05558893,
        6.65368551, 10.1979669 , 10.35485331, 11.785566  ,  5.79543038,
       12.88270018,  7.23276582, 10.52357479,  8.67227386,  4.78258092,
       13.14690137, 11.11038551,  9.96152788,  9.60359158,  8.30125201,
        9.15253344, 11.51557166, 13.34587027,  7.01475376, 13.97430191,
        8.29022197, 10.05558893, 10.52357479,  4.85995002, 10.63038565,
        7.54620849,  5.79391825, 11.24434374, 12.46513646, 12.30695098,
       13.13888841, 13.61600536, 12.21532282,  7.9337647 , 13.61801526,
        7.39004226,  9.77588801,  6.25586297, 13.89886469, 11.11038551,
       11.50791859])}
{'P': tensor([[1.6719e-02, 1.8050e-01, 1.6698e-01,  ..., 2.7365e-03, 2.3566e-06,
         2.5570e-06],
        [1.2258e-03, 3.0016e-02, 7.6607e-03,  ..., 5.5634e-03, 3.0345e-06,
         2.9371e-06],
        [1.7658e-03, 5.9776e-02, 3.2012e-02,  ..., 7.4154e-03, 3.9247e-06,
         4.1830e-06],
        ...,
        [4.0037e-04, 4.8111e-04, 4.8148e-04,  ..., 3.0232e-01, 1.6546e-05,
         9.6327e-06],
        [6.2471e-04, 3.9456e-04, 5.5908e-04,  ..., 2.7185e-05, 2.9217e-02,
         8.7161e-02],
        [1.2365e-04, 9.0383e-05, 1.3312e-04,  ..., 2.6383e-05, 1.7967e-05,
         1.3533e-02]]), 'X': tensor([[ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0489, 0.0335,  ..., 0.5350, 0.6178, 0.9793],
        [0.0489, 0.0000, 0.0370,  ..., 0.5170, 0.6482, 1.0123],
        [0.0335, 0.0370, 0.0000,  ..., 0.5031, 0.6115, 0.9754],
        ...,
        [0.5350, 0.5170, 0.5031,  ..., 0.0000, 0.5875, 0.9001],
        [0.6178, 0.6482, 0.6115,  ..., 0.5875, 0.0000, 0.3671],
        [0.9793, 1.0123, 0.9754,  ..., 0.9001, 0.3671, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.46248947,  7.57685908,  7.5403822 ,  7.58378655,  7.89745107,
        7.92200342,  7.85919687, 13.77075174, 12.70489563,  5.91677188,
        8.62329466,  7.38844703, 14.10045797, 11.54672755, 10.06286074,
        6.48486171, 13.3607379 , 12.85032836, 11.35182589, 10.36844813,
        6.69389431,  6.76056625, 11.04313641,  5.57341623, 11.72802491,
       13.76047999, 13.96746816, 13.50484539,  7.8575648 , 11.72802491,
       10.26545792, 13.95361098, 14.02532758,  8.11227668, 11.86239961,
        7.37576154, 11.48434767,  5.69913817,  7.27759233,  7.58378655,
        8.44940661, 10.36844813, 12.95702647,  5.70405525,  8.41148722,
        6.652742  , 10.19036081,  8.11227668,  9.95552626,  6.81084724,
       10.1979669 ,  9.69629   ,  7.58378655,  5.87305005, 10.07238463,
        7.83486943, 12.26997165,  9.70108409, 14.10045797, 13.99877413,
        7.32233733,  9.99211663,  9.55720651, 12.26997165,  9.96391045,
        7.46248947, 13.1399181 , 12.58006038, 11.62770332,  5.04028538,
       12.49317563,  5.29009836, 11.48663767, 13.32374056,  6.33517349,
        8.06643487,  4.74468406,  6.10360073,  5.0713364 , 13.79676654,
        6.48486171,  5.5748624 ,  7.07902036,  6.42689705,  9.7751189 ,
        8.8580671 , 13.59480906, 13.59056613,  6.1463896 , 10.14044489,
       13.84126197,  5.95111804,  6.35029274,  8.4486771 , 11.4890861 ,
        6.39438044,  8.05668609, 11.62770332, 13.13888841, 13.5247215 ,
       10.98076873, 10.09979747, 12.49291768,  8.57560275, 12.4013153 ,
       12.79126785])}
{'P': tensor([[3.3790e-02, 1.8774e-01, 1.4250e-01,  ..., 7.7262e-06, 5.4123e-05,
         1.6857e-03],
        [3.8945e-04, 2.4818e-02, 6.2347e-03,  ..., 8.0501e-06, 3.1213e-05,
         4.4549e-03],
        [1.3267e-03, 2.9259e-02, 6.5482e-02,  ..., 2.9532e-05, 1.2197e-04,
         9.3809e-03],
        ...,
        [2.0754e-05, 8.6130e-06, 1.2853e-05,  ..., 5.9961e-02, 1.6861e-02,
         5.5350e-06],
        [1.9440e-03, 7.2098e-04, 1.2590e-03,  ..., 7.1874e-04, 1.6529e-01,
         1.2215e-04],
        [2.6957e-05, 3.6961e-05, 3.4620e-05,  ..., 8.9564e-04, 2.7204e-04,
         5.2020e-02]]), 'X': tensor([[ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1442e-01, 6.1341e-02,  ..., 6.7039e-01, 2.8003e-01,
         7.0378e-01],
        [1.1442e-01, 0.0000e+00, 9.2133e-02,  ..., 6.0345e-01, 3.6990e-01,
         5.9691e-01],
        [6.1341e-02, 9.2133e-02, 0.0000e+00,  ..., 6.0916e-01, 2.7777e-01,
         6.5395e-01],
        ...,
        [6.7039e-01, 6.0345e-01, 6.0916e-01,  ..., 2.1453e-04, 7.0311e-01,
         3.6764e-01],
        [2.8003e-01, 3.6990e-01, 2.7777e-01,  ..., 7.0311e-01, 0.0000e+00,
         8.6328e-01],
        [7.0378e-01, 5.9691e-01, 6.5395e-01,  ..., 3.6764e-01, 8.6328e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.48311002,  7.83534987,  7.80625981,  7.88677538,  8.06643487,
        7.822854  ,  8.05242591,  6.56833864,  6.02779201,  8.86082124,
       11.11969815, 11.00694688,  9.51074763,  8.01009083, 12.49317563,
       13.76047999,  9.88310614,  9.09963894,  9.15253344,  7.50500064,
        9.71392179,  9.10945678,  8.87587457,  9.23178204, 12.05300943,
       14.07623266,  7.71075226, 11.57249984,  6.19707944, 11.00694688,
       10.88618114,  6.76791483,  6.3949043 ,  8.05242591, 13.84126197,
       12.04924159, 10.09979747,  6.67218687, 12.7125079 , 10.82106505,
        5.18047638,  7.44805901, 11.83570132, 10.35093485,  5.2287058 ,
        6.28835101,  9.62238828,  8.0550286 ,  7.92644746,  7.5403822 ,
       10.61472636,  6.38399789, 10.27572673,  5.84533345,  5.16077764,
        4.89869205, 13.2778378 , 10.412299  , 11.27656488,  5.40689332,
        9.79814997,  8.87587457, 12.10942412, 10.8207297 , 13.05148826,
        9.62958973,  7.06272256, 11.04313641,  5.95297459, 12.49291768,
       13.39372426,  9.58347456, 13.59673406,  8.66273171, 10.24557047,
       12.44177641, 10.69963954,  6.74686071,  8.62631801,  4.91652556,
        6.60752373,  8.1303945 ,  7.39599986,  9.75200848,  7.5158489 ,
       11.28093507,  9.48969978, 13.43127131,  7.28793663, 10.51437947,
       10.38897563,  9.23071698, 11.48734978, 12.1494023 , 12.27993098,
       11.62770332, 12.9089559 , 13.52430176,  8.91551231,  8.04483495,
        7.61088125, 13.20597932, 10.06703358, 10.74409834,  5.23878371,
        9.17905491])}
{'P': tensor([[3.0871e-02, 3.1412e-01, 2.0530e-01,  ..., 1.9769e-06, 6.9723e-03,
         1.7323e-04],
        [9.2881e-04, 4.8138e-02, 4.8555e-01,  ..., 2.6117e-06, 3.0827e-03,
         2.6092e-04],
        [7.6364e-04, 2.2510e-03, 5.8317e-02,  ..., 3.7206e-06, 2.1876e-03,
         3.9599e-04],
        ...,
        [2.7549e-05, 1.8481e-05, 1.9386e-05,  ..., 1.0556e-01, 1.4135e-04,
         3.3243e-04],
        [1.0796e-02, 3.3211e-02, 7.2458e-02,  ..., 4.6947e-05, 1.7119e-01,
         1.0543e-02],
        [2.7277e-05, 4.3354e-05, 6.9695e-05,  ..., 1.1944e-04, 1.3951e-04,
         6.5180e-02]]), 'X': tensor([[  2.6970,   6.9826,   9.7350],
        [  0.9036,   7.6840,   9.9237],
        [ -0.0410,   7.8591,   9.3457],
        [  2.1477,   7.2550,   8.9688],
        [ -1.5261,   7.8988,   9.3768],
        [  1.9204,   7.3518,   8.4781],
        [  2.2182,  13.8007,   5.1889],
        [  5.6095,  -1.9702,  -7.8233],
        [  1.9241,   7.3503,   7.9004],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -8.7544,   2.6096,   3.3146],
        [ 11.2939,   6.6325,  -5.8311],
        [  0.6952,  -4.8060,   8.8976],
        [ -3.3116, -10.1539,   7.5919],
        [ 11.4703,  -4.2006,   7.4086],
        [  3.2990,  -4.2415,  -7.4857],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.2790,  -0.0259,  -5.4116],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -5.8389,  -8.6056,  -3.5098],
        [  9.9750,  -6.6338,  -1.2440],
        [ -6.5295,  -7.9785,  -7.4429],
        [  5.5158,  -2.1337,   3.5246],
        [ -9.3705,   0.4208,  -3.5815],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.3549,   0.7096,   6.9198],
        [  3.2692,  13.5098,  -4.9032],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.0054,  -1.0902,   8.3079],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.5813,  -4.5471,   5.3597],
        [  1.9312,  13.8649,   8.3563],
        [  6.1741,  -0.5329,   3.1913],
        [  6.2247,  -0.3150,  -9.3657],
        [ -5.6651,   6.4380,   1.3660],
        [  4.4290,  -3.4596,   1.4181],
        [ 11.1420,   6.9331,  -2.4779],
        [ -7.6293,  -6.6859,  -5.5161],
        [  9.2979,   9.6048,   7.3138],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.1348,   5.9514,  -0.8856],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -2.2508, -10.5479,  -2.7235],
        [  6.2770,  -0.0379,   2.4195],
        [  3.1064,  13.5607,   6.7605],
        [  9.3607,   9.5337,  -4.5412],
        [  9.4554,   9.4245,   9.8800],
        [  3.2366,  -4.2732,   5.1532],
        [  5.5424,  -2.0885,   7.2487],
        [ -7.6734,   4.4980,  -5.4248],
        [  9.9750,  -6.6338,  -1.2440],
        [ 11.7349,  -3.5838,   2.1644],
        [ -8.7078,   2.7182,   1.8146],
        [  4.8051, -10.3703,   0.5268],
        [  6.3535,   0.6792,  -7.5101],
        [  6.2804,   2.0019,   0.8607],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.6022,  -1.9833,   8.3081],
        [ 12.5927,   0.4199,  -9.4997],
        [ -1.5875,   7.8944,  -9.4286],
        [  5.1787,  -2.6383,  -4.2533],
        [  6.3551,   0.7149,   1.3344],
        [  1.7659,   7.4127,   4.7871],
        [  5.7784,  -1.6401,   9.5587],
        [  1.7122,  -4.7591,  -3.6734],
        [  5.8450,  -9.9356,   3.0767],
        [ 10.5955,   7.8824,  -3.7006],
        [  5.8889,  -1.3916,   8.8993],
        [  5.8450,  -9.9356,   3.0767],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -1.0482, -10.8487,  -5.9066],
        [ -9.3444,   0.5821,   6.9101],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.3515,   0.6418,  -6.5823],
        [ -7.2515,   5.0195,   4.6434],
        [  3.4401,  13.4540,   4.6576],
        [  0.2571,  -4.7594,   1.5618],
        [ -5.9872,   6.2037,  -9.4477],
        [  9.2692,   9.6370,   7.0746],
        [  2.7590,  13.6621,   9.4496],
        [  0.6470, -11.0352,  -1.1931],
        [  3.1480,  -4.3165,  -7.8002],
        [  1.5383, -11.0274,  -4.9666],
        [ 12.4592,  -0.9168,  -6.7571],
        [  1.5383, -11.0274,  -4.9666],
        [  6.3569,   0.7591,   6.5011],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.4117,   0.1160,   8.2574],
        [ -8.1426,   3.8048,   9.3542],
        [ 11.5935,   5.9740,  -4.3952],
        [ -7.5720,   4.6310,   2.1717],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.6523,   3.9443,  -2.9449],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.3604,   0.9042,  -6.5901],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -6.0195,   6.1789,  -2.9496],
        [  3.4920,  -4.1366,  -8.3955],
        [  1.7659,   7.4127,   4.7871],
        [ -3.9700,   7.3453,  -4.1486]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.8992e-02, 8.3660e-02,  ..., 9.7889e-01, 1.7380e-01,
         8.7737e-01],
        [6.8992e-02, 0.0000e+00, 1.4835e-02,  ..., 9.9833e-01, 1.8941e-01,
         8.6763e-01],
        [8.3660e-02, 1.4835e-02, 0.0000e+00,  ..., 1.0007e+00, 1.9353e-01,
         8.6381e-01],
        ...,
        [9.7889e-01, 9.9833e-01, 1.0007e+00,  ..., 0.0000e+00, 8.0990e-01,
         4.1063e-01],
        [1.7380e-01, 1.8941e-01, 1.9353e-01,  ..., 8.0990e-01, 1.5170e-04,
         7.0871e-01],
        [8.7737e-01, 8.6763e-01, 8.6381e-01,  ..., 4.1063e-01, 7.0871e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.48538905,  7.73692436,  7.85919687,  7.56617037,  8.04483495,
        7.5984809 , 13.97779839,  5.94541803,  7.59795445, 10.4977132 ,
        9.13507321, 13.09737053,  4.85603886, 10.68031351, 12.21532282,
        5.37343173, 10.06286074,  6.27905292, 10.53368644,  9.5642993 ,
       10.39942497, 11.97949273, 10.30972193,  5.91408504,  9.37989631,
        7.9337647 ,  6.39438044, 13.89974693, 10.31613269, 10.60226614,
        9.85330646,  6.10360073, 12.16901615,  5.2287058 , 13.99877413,
        6.19707944,  6.23262289,  8.57560275,  5.62005629, 13.12299145,
       10.14436912, 13.36800073, 10.00802627,  7.24677664,  9.86089579,
       10.78533838,  6.27715105, 13.91198191, 13.36092624, 13.35013374,
        5.36061974,  5.92281142,  8.89458273, 11.97949273, 12.26997165,
        9.12220271, 11.42947632,  6.38968357,  6.59176018,  5.18047638,
        5.94293185, 12.59969936,  8.05242591,  5.81199422,  6.39520397,
        7.62011734,  6.00663314,  5.05774295, 11.52733581, 13.20597932,
        6.05113217, 11.52733581,  9.48969978, 10.8992524 ,  9.36256096,
        6.49538841,  9.75401449,  6.383897  ,  8.81930525, 13.88683436,
        4.7663526 ,  8.62162372, 13.37121923, 13.93790235, 11.05413545,
        5.34250225, 11.1341798 , 12.49291768, 11.1341798 ,  6.4020404 ,
       10.33712258,  9.41244865,  8.98765644, 13.04217487,  8.87587457,
       12.9089559 , 11.54672755,  6.89245352, 11.57249984,  6.42439344,
       10.92597196, 10.00410625,  8.62631801,  5.41348121,  7.62011734,
        8.34949383])}
{'P': tensor([[1.1591e-02, 1.7947e-01, 1.6925e-01,  ..., 1.4074e-06, 5.5151e-07,
         6.0980e-05],
        [8.1801e-04, 1.4433e-02, 2.3028e-01,  ..., 1.7247e-06, 6.9387e-07,
         7.7898e-05],
        [9.9307e-04, 1.0281e-03, 3.4029e-02,  ..., 3.8184e-06, 1.6586e-06,
         1.7994e-04],
        ...,
        [2.6266e-04, 2.4439e-04, 1.6247e-04,  ..., 6.0368e-02, 5.2924e-05,
         2.3449e-06],
        [2.4209e-05, 2.3404e-05, 1.7479e-05,  ..., 2.8592e-04, 4.9352e-02,
         1.9004e-05],
        [4.9768e-04, 5.2980e-04, 6.9778e-04,  ..., 6.3933e-06, 1.1125e-04,
         5.9756e-01]]), 'X': tensor([[  2.6613,   7.0021,   4.5783],
        [  2.5382,   7.0672,   4.5097],
        [  1.7500,   7.4187,   4.4237],
        [  1.7659,   7.4127,   4.7871],
        [  1.4489,   7.5256,   4.4529],
        [  1.7301,   7.4262,   5.0522],
        [  6.3569,   0.7591,   6.5011],
        [ -9.4773,  -1.0024,   5.2599],
        [  9.4554,   9.4245,   9.8800],
        [ -8.9178,   2.1936,   7.5388],
        [ 11.5198,  -4.0923,   5.5088],
        [  3.1892, -10.8220,  -6.8409],
        [ 12.2592,   3.9425,   2.0383],
        [ -1.8823, -10.6558,  -3.8952],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.9655,   3.2172,  -7.5953],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.2199,   4.6861,  -1.9304],
        [  2.6451,  -4.5250,  -0.5565],
        [  3.2701,   6.6338,   9.9209],
        [  7.3724,  -9.0577,  -8.9843],
        [ -1.6220,   7.8917,   0.0315],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.3118,   4.5452,   4.2114],
        [ -3.0008, -10.2827,  -8.9480],
        [ -2.2188,   7.8218,   8.6064],
        [ -7.4887,   4.7364,   8.9074],
        [ -7.8849,  -6.3115,  -4.7479],
        [  6.3589,   0.8239,  -8.4230],
        [  9.3424,  -7.3685,  -3.6807],
        [  9.1620,  -7.5573,  -9.9882],
        [ 12.3609,   3.4769,  -8.5762],
        [  1.0000,   7.6591,  -4.5011],
        [  0.9392,  -4.8143,  -3.4649],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.8058,   2.4850,  -1.7425],
        [  1.8320,   7.3871,   5.1452],
        [  7.0515,  11.6194,   8.4289],
        [  0.7958,  -4.8109,  -8.2248],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.3116, -10.1539,   7.5919],
        [ -8.9606,  -4.0994,   1.7886],
        [  8.9284,  10.0036,  -6.9210],
        [  0.1287, -11.0065,   6.5757],
        [  1.0093,   7.6567,   2.3526],
        [  9.6587,   9.1817,  -0.8212],
        [  2.3261,   7.1726,   6.2522],
        [  8.7715,  -7.9390,   5.2900],
        [  4.1537,   5.9340,  -6.0423],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.5626,  -2.0536,   8.4433],
        [  3.6703,   6.3443,  -5.4401],
        [  3.7992,  -3.9474,  -8.9019],
        [  1.4306,  13.9620,  -0.2075],
        [  5.3602,  -2.3808,   7.2098],
        [  1.5653,  13.9377,   4.1773],
        [ -7.0120,   5.2825,   7.7762],
        [  5.4167, -10.1293,   6.2156],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.7544,   2.6096,   3.3146],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.6038,   0.7665,  -3.4439],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.7145,  -4.8071,  -2.3766],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.4242, -10.1261,   3.9595],
        [  1.2728, -11.0372,   9.8121],
        [  1.7000,  -4.7609,   2.7935],
        [  6.3569,   0.7591,   6.5011],
        [  2.6321,  -4.5296,  -2.5159],
        [  0.6396,  14.0772,  -6.3786],
        [  3.9663,   6.1018,  -5.4104],
        [ -9.4366,  -0.1189,  -2.6736],
        [  6.3301,   0.3638,  -7.5685],
        [  0.2732,  -4.7619,  -1.1577],
        [ -9.1907,  -3.3180,  -5.5943],
        [  9.2979,   9.6048,   7.3138],
        [  0.1287, -11.0065,   6.5757],
        [  5.6248,   3.9984,   8.4933],
        [ -5.8389,  -8.6056,  -3.5098],
        [  0.1532,  -4.7422,  -7.0145],
        [  2.3200,   7.1756,   1.1938],
        [  2.7121,   6.9743,  -8.4202],
        [  4.4915,   5.5990,   7.0405],
        [  0.9788,  -4.8145,  -3.8661],
        [  6.3606,   0.9204,   3.2525],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.3261,  13.7749,   5.9121],
        [ -4.7334,   6.9982,  -0.4392],
        [ -6.4844,  -8.0230,   8.5209],
        [  7.9816,  10.8938,  -0.5378],
        [  2.3261,   7.1726,   6.2522],
        [ -2.1533, -10.5778,   1.0112],
        [ 12.5832,   1.7543,  -7.6134],
        [  6.3534,   1.2990,  -7.2942],
        [  6.8704,  -9.3809,   7.5769],
        [  4.2612, -10.5514,  -6.6268],
        [ -1.5875,   7.8944,  -9.4286]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.9659e-03, 5.1493e-02,  ..., 3.6253e-01, 9.2383e-01,
         8.8501e-01],
        [9.9659e-03, 0.0000e+00, 4.1692e-02,  ..., 3.5733e-01, 9.1861e-01,
         8.8091e-01],
        [5.1493e-02, 4.1692e-02, 0.0000e+00,  ..., 3.3066e-01, 9.0544e-01,
         8.7291e-01],
        ...,
        [3.6253e-01, 3.5733e-01, 3.3066e-01,  ..., 0.0000e+00, 1.1310e+00,
         1.1255e+00],
        [9.2383e-01, 9.1861e-01, 9.0544e-01,  ..., 1.1310e+00, 0.0000e+00,
         1.2018e-01],
        [8.8501e-01, 8.8091e-01, 8.7291e-01,  ..., 1.1255e+00, 1.2018e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.49076899,  7.50917551,  7.62232307,  7.62011734,  7.66377554,
        7.62509582,  6.4020404 ,  9.53015376, 13.35013374,  9.18358562,
       12.22503754, 11.28215429, 12.87752052, 10.8207297 ,  9.53015376,
        6.77777606, 12.53285633,  9.95552626,  7.01475376,  5.24137909,
        7.39599986, 11.67875855,  8.05668609,  9.87720102,  9.70108409,
        9.20340031,  6.99097013, 10.71162681,  8.1303945 ,  8.86082124,
       10.09979747,  6.41202986, 11.89855267, 11.87665501, 12.84056604,
        7.72415048,  4.90505566,  9.93553424,  9.9353921 ,  9.14973319,
        7.61088125, 13.59170809,  4.8763135 ,  8.28968067, 10.68031351,
        9.85384623, 13.40850328, 11.00726483,  7.72291946, 13.32645552,
        7.5403822 , 11.83075005,  7.24326992, 13.66051884, 12.26997165,
        5.92952538,  7.32950385,  5.47866221, 14.03505624,  5.86519525,
       14.02532758,  8.77913428, 11.48663767,  5.5671389 ,  9.13507321,
        7.822854  ,  9.78286733, 12.35421461, 12.62711252,  5.69913817,
        4.85995002, 10.24557047, 11.48734978, 11.11038551,  5.05534609,
        6.4020404 ,  5.23878371, 14.09176259,  7.27759233,  9.43737523,
        6.34059021,  4.76969364,  9.7712321 , 13.36800073, 11.00726483,
        6.90115901, 10.39942497,  4.74468406,  7.54127425,  7.48311002,
        7.17789666,  4.91296205,  6.42689705,  9.14973319,  9.45995955,
       13.96987806,  8.4486771 , 10.31584102, 13.50484539,  7.5403822 ,
       10.794755  , 12.70489563,  6.48486171, 11.62770332, 11.37939695,
        8.05242591])}
{'P': tensor([[3.4662e-02, 2.2754e-01, 1.8977e-01,  ..., 3.3567e-03, 3.0791e-05,
         7.0136e-05],
        [9.2775e-04, 5.0798e-02, 1.1837e-03,  ..., 2.3519e-03, 8.3396e-05,
         2.6888e-05],
        [1.4147e-03, 3.6561e-01, 2.5778e-02,  ..., 2.5565e-03, 2.8409e-05,
         2.8546e-05],
        ...,
        [5.5362e-03, 2.6019e-02, 1.3656e-02,  ..., 1.9572e-01, 9.9539e-05,
         1.8941e-04],
        [1.1234e-05, 6.7869e-06, 8.5005e-06,  ..., 3.9146e-06, 3.3063e-02,
         2.2157e-06],
        [1.0300e-02, 1.0984e-02, 1.4099e-02,  ..., 1.9999e-02, 8.6430e-07,
         4.6868e-02]]), 'X': tensor([[  2.5974,   7.0363,  -7.7810],
        [  0.3724,   7.7974,  -6.8722],
        [  1.6422,   7.4586,  -6.8056],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.1460,   7.8340,  -9.4614],
        [ -7.7239,   4.4298,   3.1216],
        [ 11.5083,  -4.1177,   0.2389],
        [  6.1730,  -0.5375,  -8.8861],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.0955,   2.8210,  -4.3321],
        [ -8.0829,  -5.9940,   5.0564],
        [ -7.4602,   4.7718,  -4.6277],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.3124,   0.2057,  -0.4593],
        [  2.2644,  -4.6434,   9.7508],
        [  3.3094,  13.4969,  -3.6507],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.2692,  13.5098,  -4.9032],
        [  1.1644, -11.0394,   7.8511],
        [ -7.0948,   5.1940,   9.8171],
        [  0.7275,  14.0667,  -0.9815],
        [  6.1989,  -9.7591,   0.8289],
        [  2.5974,   7.0363,  -7.7810],
        [  1.5383, -11.0274,  -4.9666],
        [  4.5325,  -3.3643,   6.7699],
        [  3.2366,  -4.2732,   5.1532],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.0305,   3.0296,   0.1375],
        [ -5.9225,  -8.5355,   2.4778],
        [  2.3367,   7.1676,  -9.5263],
        [  6.8093,  -9.4178,   8.4244],
        [  5.5626,  -2.0536,   8.4433],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.3549,   0.7096,   6.9198],
        [  1.1350,  14.0104,   0.6321],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 10.3251,   8.2928,  -3.0006],
        [  4.7156,  -3.1831,  -8.0224],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.1890,   4.7318,   0.9515],
        [  4.5340,  -3.3629,   8.9652],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.0073,   3.0986,   6.7937],
        [  0.6757, -11.0361,  -1.3047],
        [  2.5128, -10.9365,   9.9645],
        [  4.5325,  -3.3643,   6.7699],
        [  6.8407,  11.7666,   3.0045],
        [ 12.0042,  -2.8350,   2.6864],
        [ -3.3116, -10.1539,   7.5919],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.1666,   3.7653,   3.6075],
        [  4.5057,  -3.3894,  -0.8160],
        [  3.6703,   6.3443,  -5.4401],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.8363,   2.4086,  -5.6296],
        [ -7.6719,  -6.6260,   7.1468],
        [  1.2728, -11.0372,   9.8121],
        [  6.3266,   0.3292,   9.3502],
        [  9.1409,  -7.5789,   1.5814],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.4157,   0.0821,  -5.0025],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.6672,  -1.8631,   8.1020],
        [  5.2911,  -2.4824,  -3.6877],
        [ 10.0567,   8.6697,   4.6329],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.2182,  13.8007,   5.1889],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.2188,  -2.5839,   2.1833],
        [  2.9260,  -4.4162,   8.3808],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.1666,   3.7653,   3.6075],
        [ -7.9925,  -6.1423,   2.8990],
        [  1.7232,  -4.7574,   3.1270],
        [ -1.3940, -10.7772,   6.7835],
        [ 10.5955,   7.8824,  -3.7006],
        [ 10.7566,   7.6212,  -2.3279],
        [  0.1532,  -4.7422,  -7.0145],
        [  6.1752,   2.5242,   3.5874],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.8890,  -4.4316,   1.1600],
        [ -0.9715, -10.8630,   0.3555],
        [  4.5430,   5.5438,  -3.8669],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -6.7992,   5.4991,   9.9186],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.3653,   3.4184,   9.3454],
        [  1.3339,   7.5627,  -0.9034],
        [ -5.6651,   6.4380,   1.3660],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.2508,   1.0572,   2.9978],
        [ 11.6544,   5.8273,  -3.1375],
        [ 10.5678,   7.9261,  -0.6100],
        [ -4.0633,  -9.7927,  -3.0825],
        [  3.5939,   6.4029,  -1.8357],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -0.0410,   7.8591,   9.3457],
        [  5.4732,  -2.2041,  -4.3830],
        [  1.3339,   7.5627,  -0.9034],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 10.3251,   8.2928,  -3.0006]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.9110e-02, 2.3366e-02,  ..., 3.4709e-01, 7.9521e-01,
         4.5243e-01],
        [5.9110e-02, 0.0000e+00, 4.6783e-02,  ..., 3.3824e-01, 7.4138e-01,
         5.0410e-01],
        [2.3366e-02, 4.6783e-02, 0.0000e+00,  ..., 3.2677e-01, 7.8808e-01,
         4.5757e-01],
        ...,
        [3.4709e-01, 3.3824e-01, 3.2677e-01,  ..., 0.0000e+00, 9.2148e-01,
         4.8118e-01],
        [7.9521e-01, 7.4138e-01, 7.8808e-01,  ..., 9.2148e-01, 0.0000e+00,
         1.2453e+00],
        [4.5243e-01, 5.0410e-01, 4.5757e-01,  ..., 4.8118e-01, 1.2453e+00,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.50035355,  7.80625981,  7.63726134,  7.822854  ,  7.96181288,
        7.83534987,  8.90405823, 12.22276278,  6.19633139,  9.66977802,
        6.7166297 , 10.06286074,  8.85572292,  7.89745107,  9.47369628,
       10.53368644,  6.31576264,  5.16611782, 13.89671314,  8.8353674 ,
       13.89974693, 11.10066126,  8.79284466, 14.08549802, 11.56146873,
        7.50035355, 11.1341798 ,  5.64464807,  5.36061974, 12.42554808,
        6.74872981, 10.38897563,  7.53884104, 11.62158554,  5.92952538,
       12.42554808,  6.39438044, 14.05633441,  9.97822473, 13.2430377 ,
        5.68942572, 10.60226614,  7.02253235,  5.64500512, 12.4013153 ,
        6.7593917 , 11.05672136, 11.22142195,  5.64464807, 13.6105595 ,
       12.3344536 , 10.68031351,  9.95552626,  8.99276564,  5.63824971,
        7.32950385, 12.1494023 ,  9.15866607, 10.1371552 , 11.11038551,
        6.33517349, 11.87412439, 13.66051884, 12.22276278, 12.26997165,
        9.41605684, 10.38897563,  5.96556248,  5.84450127, 13.2778378 ,
       11.72749923, 13.97779839, 13.09426142,  5.82345867,  5.29755487,
       12.44177641,  8.99276564, 10.08001894,  5.05989547, 10.86693708,
       13.20597932, 13.18278606,  4.74468406,  6.67121748, 10.32715715,
        5.29009836, 10.90638066,  7.16747878, 10.31613269,  8.74470218,
        6.2322769 ,  9.0368458 ,  7.67939546,  8.57560275,  6.80370254,
        9.3109884 , 13.03002338, 13.20988996, 10.60226614,  7.34250948,
        9.88571713,  7.85919687,  5.90035324,  7.67939546, 10.37698435,
       13.2430377 ])}
{'P': tensor([[1.7588e-02, 2.6622e-01, 1.5587e-01,  ..., 1.3849e-06, 8.0498e-06,
         2.1664e-06],
        [4.4296e-04, 1.5707e-02, 2.2277e-01,  ..., 1.0293e-06, 4.6806e-06,
         1.7428e-06],
        [4.7916e-04, 6.9502e-04, 2.9827e-02,  ..., 1.5985e-06, 5.4566e-06,
         3.0862e-06],
        ...,
        [3.4259e-05, 2.2684e-05, 1.2695e-05,  ..., 5.7096e-03, 2.2852e-03,
         1.1488e-05],
        [1.9734e-03, 1.3134e-03, 7.1537e-04,  ..., 1.4729e-06, 8.2770e-03,
         5.3669e-07],
        [2.3400e-04, 1.4319e-04, 7.8179e-05,  ..., 1.8246e-03, 1.2407e-02,
         3.5915e-02]]), 'X': tensor([[  2.5663,   7.0526,  -6.8769],
        [  1.6422,   7.4586,  -6.8056],
        [  0.3724,   7.7974,  -6.8722],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.8569,   7.9156,  -6.5557],
        [ -0.6323,   7.9085,  -5.9544],
        [ 11.2209,  -4.7092,   2.6117],
        [  8.8474,  10.0868,  -8.8644],
        [ -2.1382,   7.8339,   7.3524],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.2419,  -0.2313,   4.2440],
        [ -2.7572,   7.7194,   5.9580],
        [ 11.5966,   5.9666,   6.0083],
        [ -7.9667,   4.0814,  -0.6397],
        [  8.9284,  10.0036,  -6.9210],
        [ -8.1666,   3.7653,   3.6075],
        [  1.2728, -11.0372,   9.8121],
        [  0.9809,   7.6642,  -1.4217],
        [ -6.8618,  -7.6322,   3.7027],
        [  1.8447,  -4.7372,   8.1235],
        [ -8.5900,   2.9767,   7.4811],
        [  1.8320,   7.3871,   5.1452],
        [ -4.7334,   6.9982,  -0.4392],
        [ -1.9487,   7.8590,  -3.3330],
        [ -7.0948,   5.1940,   9.8171],
        [  6.3413,   0.4883,  -6.8416],
        [  6.3119,   1.7797,   9.9186],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.0506,   4.9277,  -9.5094],
        [ 10.7566,   7.6212,  -2.3279],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.3122,  -9.6994,  -6.3282],
        [  2.8890,  -4.4316,   1.1600],
        [ 10.4994,  -5.9193,  -7.5153],
        [  6.3413,   0.4883,  -6.8416],
        [  0.3724,   7.7974,  -6.8722],
        [  3.3474,  -4.2162,   5.2455],
        [  6.2365,  -0.2578,  -6.0212],
        [ 10.4758,  -5.9539,   1.3667],
        [ 11.0373,   7.1299,  -8.5895],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.5128, -10.9365,   9.9645],
        [ -3.5903,   7.4847,  -8.9112],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.1625,  -4.6697,   7.0658],
        [  1.6308,  13.9254,  -2.0984],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -5.7347,  -8.6908,   9.2878],
        [ -2.7572,   7.7194,   5.9580],
        [  5.1519, -10.2387,   9.8191],
        [ -0.5739,   7.9056,   0.6382],
        [ 11.3367,  -4.4806,   0.4643],
        [ 11.2939,   6.6325,  -5.8311],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.8323,  -1.5228,   5.1132],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.0619,   8.6627,  -9.2884],
        [ -2.2188,   7.8218,   8.6064],
        [ -2.4956, -10.4682,   3.5632],
        [ 11.3367,  -4.4806,   0.4643],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.3464,   0.5708,  -1.6397],
        [  3.9591, -10.6391,  -6.2557],
        [  4.6231,   5.4556,   8.4973],
        [  5.0190,   4.9705,  -7.9752],
        [ -0.8675, -10.8815,   6.8561],
        [  2.2849, -10.9655,  -7.6542],
        [  4.3377,   5.7570,   1.3712],
        [ -9.4366,  -0.1189,  -2.6736],
        [  9.0051,   9.9234,   5.4411],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.4039,   5.6903,   8.7366],
        [  3.1480,  -4.3165,  -7.8002],
        [ -8.5900,   2.9767,   7.4811],
        [  9.6587,   9.1817,  -0.8212],
        [  6.3606,   0.9204,   3.2525],
        [  6.2895,   0.0405,   5.7652],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 10.9916,   7.2133,  -0.9179],
        [ -9.4196,   0.0478,   9.4140],
        [  6.3505,   0.6235,  -8.5849],
        [ -5.5525,   6.5146,  -4.4984],
        [ -0.5709,   7.9055,   7.6089],
        [  4.2612, -10.5514,  -6.6268],
        [  6.8704,  -9.3809,   7.5769],
        [ -4.4503,   7.1378,   3.3751],
        [  0.8572,  14.0501,  -8.3364],
        [  5.4167, -10.1293,   6.2156],
        [  8.0381,  -8.5690,  -8.9523],
        [  5.5294,  -2.1106,   9.0626],
        [  3.9085,   6.1512,  -2.2560],
        [  4.8051, -10.3703,   0.5268],
        [  5.2199,   4.6861,  -1.9304],
        [ -9.1706,   1.3863,   8.4032],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.9916,   7.2133,  -0.9179],
        [  4.0421,  -3.7765,  -8.3631],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 10.4758,  -5.9539,   1.3667],
        [ 12.4691,   2.8525,  -5.5110],
        [  8.0381,  -8.5690,  -8.9523]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0325, 0.0793,  ..., 0.4880, 0.3419, 0.1645],
        [0.0325, 0.0000, 0.0468,  ..., 0.4970, 0.3726, 0.1613],
        [0.0793, 0.0468, 0.0000,  ..., 0.5121, 0.4174, 0.1691],
        ...,
        [0.4880, 0.4970, 0.5121,  ..., 0.0000, 0.6271, 0.6524],
        [0.3419, 0.3726, 0.4174,  ..., 0.6271, 0.0000, 0.3544],
        [0.1645, 0.1613, 0.1691,  ..., 0.6524, 0.3544, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.50500064,  7.63726134,  7.80625981,  7.822854  ,  7.96181288,
        7.9337647 , 12.16901615, 13.41713661,  8.12043983, 10.412299  ,
        6.24614905,  8.19703512, 13.04156075,  8.95133961, 13.40850328,
        8.99276564, 11.11038551,  7.72668467, 10.26327895,  5.08372149,
        9.09119351,  7.61088125,  8.4486771 ,  8.09704052,  8.79284466,
        6.36004393,  6.55800267, 11.67875855,  7.05627461, 13.18278606,
       12.67576792, 11.57249984,  5.29009836, 12.05300943,  6.36004393,
        7.80625981,  5.38342077,  6.24187044, 12.0495521 , 13.1399181 ,
        9.60359158,  8.30125201, 11.22142195,  8.30125201, 13.1399181 ,
        9.85330646,  5.14607014, 14.02058621,  8.83725081, 12.58006038,
        9.70108409, 10.412299  ,  8.19703512, 11.46176259,  7.92644746,
       12.18998195, 13.09737053, 10.1979669 ,  6.02779201, 13.45642379,
       13.27717982,  8.1303945 , 10.76154256, 12.18998195, 12.4980845 ,
        9.36378438, 11.35182589,  7.15100303,  7.06372172, 10.91601655,
       11.20100795,  7.20826219,  9.43737523, 13.40024224, 12.91505111,
       12.17901033,  7.19534146,  5.34250225,  9.09119351, 13.32645552,
        6.42689705,  6.28961988, 10.52587243, 13.1471313 ,  9.41970031,
        6.38105015,  8.55982105,  7.92607342, 11.37939695, 11.62770332,
        8.41148722, 14.07623266, 11.48663767, 11.74901778,  5.91854397,
        7.28793663, 11.42947632,  7.01475376,  9.27474386, 13.32374056,
       13.1471313 ,  5.53175275, 12.4217523 , 12.0495521 , 12.79126785,
       11.74901778])}
{'P': tensor([[1.4481e-02, 2.4999e-01, 1.1127e-01,  ..., 1.6855e-03, 9.8248e-06,
         1.1127e-01],
        [7.9234e-04, 2.4795e-02, 1.9019e-02,  ..., 3.6600e-03, 2.0605e-05,
         1.9019e-02],
        [1.4023e-03, 1.8619e-02, 2.3223e-02,  ..., 4.2416e-03, 2.3175e-05,
         2.3223e-02],
        ...,
        [7.5924e-05, 9.0456e-05, 9.5791e-05,  ..., 7.0869e-02, 6.9158e-03,
         9.5791e-05],
        [8.1261e-06, 8.0296e-06, 9.1721e-06,  ..., 7.5756e-05, 1.3981e-01,
         9.1721e-06],
        [1.4023e-03, 1.8619e-02, 2.3223e-02,  ..., 4.2416e-03, 2.3175e-05,
         2.3223e-02]]), 'X': tensor([[  2.5620,   7.0549,   8.4825],
        [  1.9204,   7.3518,   8.4781],
        [  1.9241,   7.3503,   7.9004],
        [  2.1477,   7.2550,   8.9688],
        [ -0.0410,   7.8591,   9.3457],
        [ -0.5709,   7.9055,   7.6089],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.1379,  -4.6757,   6.4981],
        [  5.7784,  -1.6401,   9.5587],
        [  6.1984,  -0.4327,   1.5438],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.3062,  -5.6008,   0.0287],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.7792,  11.8084,   1.6251],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.6363, -11.0348,   6.1414],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.0617,  12.7957,   8.3544],
        [  3.3474,  -4.2162,   5.2455],
        [ -3.0644,   7.6439,  -1.4322],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -0.8675, -10.8815,   6.8561],
        [  2.0870,  -4.6877,   9.6823],
        [  8.0381,  -8.5690,  -8.9523],
        [  3.5939,   6.4029,  -1.8357],
        [ -2.1382,   7.8339,   7.3524],
        [ 12.1940,  -2.1881,   7.1499],
        [  9.4925,   9.3811,   8.9842],
        [  5.0348,  -2.8229,   3.5436],
        [ -9.4351,  -1.8891,   3.5642],
        [ -2.1382,   7.8339,   7.3524],
        [ 10.2470,   8.4054,  -0.1177],
        [  0.2732,  -4.7619,  -1.1577],
        [ 12.2589,  -1.9312,   9.8657],
        [ -3.2053, -10.1993,  -3.8736],
        [ -3.5029,   7.5138,   8.1853],
        [  4.8674,  12.8875,   6.7622],
        [  4.6357,   5.4415,   7.9067],
        [  5.9003,  12.3550,   2.1917],
        [  5.8836,   3.4315,  -4.0327],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.1231,  -3.5739,   8.4708],
        [ -4.9217,   6.8979,  -1.7667],
        [ -9.1960,  -3.2964,   1.6921],
        [ -1.8788, -10.6567,   3.7068],
        [  6.5428,  -9.5728,  -2.9897],
        [  9.8082,  -6.8396,   4.9615],
        [  5.8318,   3.5567,  -4.7692],
        [  1.1644, -11.0394,   7.8511],
        [  6.2804,   2.0019,   0.8607],
        [  2.6613,   7.0021,   4.5783],
        [  0.2732,  -4.7619,  -1.1577],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.3497,   1.3680,  -1.9076],
        [ -2.5056,   7.7720,   5.5878],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.1440, -10.9811,   8.5281],
        [ 11.7349,  -3.5838,   2.1644],
        [ 12.2921,  -1.7902,  -4.0870],
        [  6.0047,   3.1062,  -6.7914],
        [ -8.1426,   3.8048,   9.3542],
        [  1.9503,  -4.7172,  -6.2313],
        [  5.3924, -10.1396,  -4.0863],
        [  2.6321,  -4.5296,  -2.5159],
        [ -7.9667,   4.0814,  -0.6397],
        [  3.2159, -10.8166,  -1.0406],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -4.7334,   6.9982,  -0.4392],
        [  3.8182,  -3.9347,  -1.7961],
        [  3.4140,  -4.1803,  -8.8224],
        [  7.3592,  11.3937,   6.2407],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.1787,  -2.6383,  -4.2533],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.1497,   7.8334,  -5.4412],
        [ -8.4297,  -5.3637,  -9.3962],
        [  5.8275,  -1.5334,   2.0124],
        [  2.1120,  -4.6818,   8.6144],
        [  5.1292,   4.8183,  -5.7997],
        [  3.3744,   6.5623,  -6.3636],
        [ 12.1940,  -2.1881,   7.1499],
        [ -6.3489,   5.9117,   6.5161],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -1.6906,   7.8860,   7.7142],
        [  2.3928,   7.1404,  -1.7035],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.4732,  -2.2041,  -4.3830],
        [  2.7976,  -4.4682,  -4.2020],
        [  3.0719, -10.8449,  -4.1673],
        [  2.5813,  -4.5471,   5.3597],
        [  0.3724,   7.7974,  -6.8722],
        [  5.7218,  -9.9934,   6.5764],
        [ 11.5568,   6.0602,   9.3244],
        [  5.5101,  -2.1433,   9.5442],
        [ 11.9695,  -2.9407,  -6.3498],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.3670,  -4.7990,   3.2603],
        [  2.4697,  -4.5836,   1.4593],
        [ -0.5709,   7.9055,   7.6089],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.1666,   3.7653,   3.6075],
        [ -6.2668,   5.9808,   2.5614],
        [ -9.3384,  -2.6161,  -4.7630],
        [  1.9241,   7.3503,   7.9004]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0376, 0.0357,  ..., 0.5091, 1.2423, 0.0357],
        [0.0376, 0.0000, 0.0161,  ..., 0.4819, 1.2182, 0.0161],
        [0.0357, 0.0161, 0.0000,  ..., 0.4751, 1.2099, 0.0000],
        ...,
        [0.5091, 0.4819, 0.4751,  ..., 0.0000, 0.7406, 0.4751],
        [1.2423, 1.2182, 1.2099,  ..., 0.7406, 0.0000, 1.2099],
        [0.0357, 0.0161, 0.0000,  ..., 0.4751, 1.2099, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.50564223,  7.5984809 ,  7.59795445,  7.56617037,  7.85919687,
        7.92607342, 12.17901033,  5.14124612,  6.00663314,  6.2134961 ,
       10.06286074, 10.01804223,  8.08320235, 13.61600536,  6.14086582,
       11.05317597, 12.53285633, 13.76047999,  5.38342077,  8.23525433,
       12.30858677, 10.91601655,  5.13125513, 11.74901778,  7.34250948,
        8.12043983, 12.38881635, 13.34587027,  5.77219077,  9.62238828,
        8.12043983, 13.25336001,  4.76969364, 12.41012046, 10.6910768 ,
        8.29022197, 13.77604293,  7.14837514, 13.69162795,  6.81117802,
       11.62158554,  9.79814997,  8.47371868,  9.76896904, 10.82106505,
       11.59512194, 11.95743558,  6.83083863, 11.10066126,  6.59176018,
        7.49076899,  4.76969364,  8.49161154,  6.49538841,  8.16584537,
        9.62958973, 12.3489324 , 11.18839169, 12.26997165, 12.4217523 ,
        6.76056625,  8.98765644,  5.10444097, 11.48434767,  5.23878371,
        8.95133961, 11.2845608 , 12.10942412,  8.4486771 ,  5.48276038,
        5.39723241, 13.56367857, 10.35485331,  8.89458273,  5.81199422,
        5.19986797,  7.83486943,  9.99145096,  6.02588517,  5.13616224,
        7.03733775,  7.37902766, 12.38881635,  8.67502348,  9.88310614,
        8.06517143,  7.53063885, 12.1494023 ,  5.90035324,  5.27178363,
       11.27160035,  5.2287058 ,  7.80625981, 11.51557166, 13.0493368 ,
        5.91222346, 12.32545637,  5.19577146,  4.98989906,  5.20659892,
        7.92607342, 12.52875793,  8.99276564,  8.66273171,  9.69792513,
        7.59795445])}
{'P': tensor([[1.4601e-02, 2.3296e-01, 2.0730e-01,  ..., 3.1615e-06, 4.0850e-04,
         2.9458e-02],
        [9.8655e-04, 3.2653e-02, 6.8207e-01,  ..., 5.3747e-06, 5.2916e-04,
         4.4335e-02],
        [3.2527e-03, 4.5673e-03, 1.2497e-01,  ..., 1.8771e-05, 1.6914e-03,
         1.2330e-01],
        ...,
        [8.9219e-04, 5.8252e-04, 4.9884e-04,  ..., 1.2050e-02, 2.8875e-03,
         2.5204e-04],
        [1.0522e-01, 1.0198e-01, 1.0060e-01,  ..., 3.2978e-05, 5.6807e-02,
         2.5754e-02],
        [4.4404e-03, 1.0371e-02, 1.5431e-02,  ..., 1.3080e-05, 1.0569e-03,
         1.2093e-01]]), 'X': tensor([[  2.5382,   7.0672,   4.5097],
        [  1.7500,   7.4187,   4.4237],
        [  1.4489,   7.5256,   4.4529],
        [  1.7659,   7.4127,   4.7871],
        [  1.7301,   7.4262,   5.0522],
        [  1.8320,   7.3871,   5.1452],
        [ -9.4624,  -0.4633,  -7.3918],
        [  4.7544,  -3.1425,  -1.5315],
        [  2.5128, -10.9365,   9.9645],
        [  8.8474,  10.0868,  -8.8644],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 10.9931,   7.2107,   7.7933],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.7218,  -9.9934,   6.5764],
        [  4.7156,  -3.1831,  -8.0224],
        [ -8.8154,   2.4611,   2.5198],
        [  0.9074,  -4.8139,   8.1674],
        [  6.2883,   0.0325,   5.2692],
        [  3.2692,  13.5098,  -4.9032],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.8745,   2.3097,   3.5600],
        [  5.7963,  -1.6018,   1.8856],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.6000,   1.3839,  -3.2992],
        [ -1.7290,   7.8825,  -3.3831],
        [  5.2199,   4.6861,  -1.9304],
        [  0.1047, -11.0046,   0.5188],
        [  6.3534,   1.2990,  -7.2942],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.9026,   3.3837,  -0.6573],
        [  5.5240,  -2.1198,   6.2055],
        [ -9.3739,   0.3982,  -4.2564],
        [ -9.3604,   0.4855,  -3.0158],
        [  8.1528,  10.7452,   9.1493],
        [ 10.9931,   7.2107,   7.7933],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2975,   0.0947,   5.1791],
        [  0.3715,  -4.7756,   0.2312],
        [ 10.9916,   7.2133,  -0.9179],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.1575,  12.2046,   3.3257],
        [ -0.4341, -10.9473,   7.4509],
        [  7.5064,  -8.9651,  -0.0994],
        [  5.6202,  -1.9508,   0.7471],
        [  5.4426, -10.1181,  -5.1268],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.2076,  13.8032,  -1.5099],
        [  6.2895,   0.0405,   5.7652],
        [ -4.0868,   7.2981,   8.1874],
        [  6.2630,   2.1057,  -5.2136],
        [  4.5057,  -3.3894,  -0.8160],
        [ -8.3062,  -5.6008,   0.0287],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.1751,  -2.2592,  -5.7751],
        [  5.6248,   3.9984,   8.4933],
        [  6.3460,   1.4252,   1.2291],
        [ 12.2218,   4.0953,  -8.2072],
        [ -4.6393,   7.0461,   1.1624],
        [  4.3554,   5.7393,  -9.0766],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4716,  -0.6673,   7.6068],
        [  4.4039,   5.6903,   8.7366],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -3.9858,  -9.8334,  -3.7323],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 12.6038,   0.7665,  -3.4439],
        [ -6.4620,  -8.0449,   6.2719],
        [  7.7794,  11.0629,   8.9069],
        [  1.6308,  13.9254,  -2.0984],
        [  2.3200,   7.1756,   1.1938],
        [  7.5930,  -8.9037,   7.2323],
        [  9.0652,   9.8597,  -3.5949],
        [  9.3424,  -7.3685,  -3.6807],
        [ -9.0961,   1.6529,   3.1553],
        [  2.5010,  -4.5736,  -0.8989],
        [  7.2340,  -9.1505,  -5.8321],
        [  5.9548,  -1.2273,  -2.6520],
        [ -0.2586,   7.8825,  -9.9840],
        [  0.2571,  -4.7594,   1.5618],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.4671,  -0.5553,   0.1117],
        [  4.7813,  -3.1139,   8.2691],
        [  9.1820,   9.7334,  -2.7932],
        [  1.3339,   7.5627,  -0.9034],
        [  6.2036,   2.4030,  -1.2500],
        [ -4.7334,   6.9982,  -0.4392],
        [ 12.5717,   1.9266,   7.4903],
        [ -1.9714,   7.8563,  -1.8084],
        [  6.1372,   2.6726,  -6.7885],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.4430,  -0.1897,   4.8383],
        [ 10.0567,   8.6697,   4.6329],
        [  1.6232,  -4.7717,  -2.5655],
        [  5.6672,  -1.8631,   8.1020],
        [ -1.9600,   7.8577,   6.9121],
        [ 12.5717,   1.9266,   7.4903],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.8323,  -1.5228,   5.1132],
        [ -7.4602,   4.7718,  -4.6277],
        [ 12.4837,   2.7510,   0.4449],
        [ 12.3759,  -1.3943,   6.4659],
        [  4.4915,   5.5990,   7.0405],
        [  0.6933,   7.7336,   2.1179]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0417, 0.0648,  ..., 0.3363, 0.1419, 0.2255],
        [0.0417, 0.0000, 0.0238,  ..., 0.3626, 0.1674, 0.2102],
        [0.0648, 0.0238, 0.0000,  ..., 0.3730, 0.1790, 0.2121],
        ...,
        [0.3363, 0.3626, 0.3730,  ..., 0.0000, 0.1954, 0.5597],
        [0.1419, 0.1674, 0.1790,  ..., 0.1954, 0.0000, 0.3671],
        [0.2255, 0.2102, 0.2121,  ..., 0.5597, 0.3671, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.50917551,  7.62232307,  7.66377554,  7.62011734,  7.62509582,
        7.61088125,  9.47369628,  5.69913817, 11.22142195, 13.41713661,
       10.51437947, 13.14690137, 12.19805394, 11.51557166,  5.68942572,
        9.15253344,  4.89869205,  6.28835101, 13.89974693,  6.35029274,
        9.17016857,  6.01356004, 11.99633819, 12.67576792,  8.06990483,
        7.01475376, 11.00508681,  6.48486171, 12.17451685,  6.80370254,
        5.91677188,  9.38232121,  9.37296102, 13.48809595, 13.14690137,
        6.33075682,  6.29821673,  4.79003361, 13.1471313 ,  9.84314712,
       12.4980845 ,  9.69629   , 13.66991027, 10.95593729, 11.69264243,
        5.94909709, 11.4890861 ,  9.20340031, 13.97857591,  6.28961988,
        8.36445758,  6.60752373,  5.63824971, 10.01804223, 12.77402351,
       12.3828957 ,  6.90115901,  6.50409372, 12.889695  ,  8.43626114,
        7.20482224, 13.36092624,  9.49511524,  7.19534146,  9.43112957,
       10.61047596, 11.785566  ,  9.75075273, 12.62711252, 10.31886066,
       13.52430176, 14.02058621,  7.54127425, 11.70168755, 13.39372426,
       11.89855267,  9.24501829,  5.21280156, 11.66453059,  6.07992623,
        7.88677538,  4.7663526 ,  6.01356004,  9.48337063,  5.70591986,
       13.38091121,  7.67939546,  6.652742  ,  8.4486771 , 12.71843462,
        8.09984231,  6.69389431, 13.79762415,  9.44486609, 13.2778378 ,
        5.04028538,  5.96556248,  8.09842704, 12.71843462,  5.39723241,
        6.02779201,  8.85572292, 12.78327037, 12.45418236,  7.17789666,
        7.76457095])}
{'P': tensor([[1.8332e-02, 2.7662e-01, 2.1220e-01,  ..., 2.1262e-06, 4.2693e-05,
         1.1130e-03],
        [7.3930e-04, 2.5567e-02, 3.6648e-01,  ..., 2.7019e-06, 5.5433e-05,
         6.2493e-04],
        [6.9685e-04, 1.2309e-03, 3.2447e-02,  ..., 3.2201e-06, 6.7422e-05,
         4.1435e-04],
        ...,
        [5.6785e-04, 3.6059e-04, 2.5742e-04,  ..., 9.6157e-02, 4.8458e-03,
         1.7057e-04],
        [1.8854e-03, 1.3335e-03, 1.0470e-03,  ..., 3.7148e-03, 5.5488e-02,
         2.2625e-04],
        [4.3430e-04, 1.0386e-03, 1.6464e-03,  ..., 3.3221e-08, 8.0805e-07,
         5.3006e-03]]), 'X': tensor([[  2.5375,   7.0676,  -4.3490],
        [  1.6408,   7.4591,  -4.2294],
        [  1.0000,   7.6591,  -4.5011],
        [  0.4996,   7.7738,  -4.4711],
        [  1.8719,   7.3713,  -3.7465],
        [  0.1497,   7.8334,  -5.4412],
        [  3.9022,  13.2906,   5.5788],
        [ -9.4753,  -0.8012,  -5.2050],
        [  5.9912,   3.1453,  -3.9989],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.0890,  -0.8382,   2.9166],
        [  0.1532,  -4.7422,  -7.0145],
        [ 10.1798,   8.5002,  -5.8893],
        [ 11.2847,  -4.5848,  -4.7503],
        [  7.3592,  11.3937,   6.2407],
        [ 12.5224,  -0.4497,   4.8448],
        [ -6.0844,  -8.3955,   0.7705],
        [ -1.6275, -10.7221,   3.2376],
        [  4.5954,  13.0099,  -4.4439],
        [ -4.3359,  -9.6431,  -9.0616],
        [  2.3589,  13.7668,  -3.6501],
        [  6.3515,   0.6418,  -6.5823],
        [  6.7565,  11.8237,  -4.9882],
        [  2.0729,   7.2878,   7.0880],
        [  1.7659,   7.4127,   4.7871],
        [  5.1787,  -2.6383,  -4.2533],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.9636,   2.0648,   8.1946],
        [  0.1206,  -4.7363,   8.7103],
        [  5.4732,  -2.2041,  -4.3830],
        [ -7.2986,  -7.1225,  -3.0814],
        [  2.3376,  -4.6231,   9.8436],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.6396,  14.0772,  -6.3786],
        [  6.8407,  11.7666,   3.0045],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.4630,  -1.5145,   0.2983],
        [  7.8375,  -8.7240,  -9.8913],
        [  0.3493,  -4.7727,  -6.5682],
        [  3.3069,   6.6089,   8.7377],
        [ -8.7409,  -4.6837,  -6.5840],
        [  3.6249, -10.7255,   5.6605],
        [ -8.8154,   2.4611,   2.5198],
        [  9.0004,  -7.7195,   4.6945],
        [  1.3705,  13.9723,   7.9266],
        [  5.8325,  -1.5222,  -2.1341],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.0604,   1.7707,   1.1066],
        [ -5.9938,  -8.4745,   3.9735],
        [  0.9788,  -4.8145,  -3.8661],
        [  6.2555,   2.1474,   4.1685],
        [ -0.3432,   7.8900,   7.1805],
        [  2.2076,  13.8032,  -1.5099],
        [ -1.6275, -10.7221,   3.2376],
        [ 12.2914,   3.8037,   6.0978],
        [ -7.9925,  -6.1423,   2.8990],
        [ -8.5149,  -5.1905,   1.5239],
        [  6.3542,   0.6934,  -3.3840],
        [  1.8320,   7.3871,   5.1452],
        [ 12.5832,   1.7543,  -7.6134],
        [ -4.2040,   7.2486,   2.9875],
        [ -9.3384,  -2.6161,  -4.7630],
        [  2.2076,  13.8032,  -1.5099],
        [ -1.6086,   7.8928,   4.3385],
        [ 12.5717,   1.9266,   7.4903],
        [  5.5240,  -2.1198,   6.2055],
        [ -9.4430,  -0.1897,   4.8383],
        [  3.0719, -10.8449,  -4.1673],
        [  3.3839,  -4.1966,   8.3574],
        [  6.3460,   1.4252,   1.2291],
        [ -6.0844,  -8.3955,   0.7705],
        [ -1.9600,   7.8577,   6.9121],
        [  1.4306,  13.9620,  -0.2075],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.2038,   2.4020,  -3.1692],
        [  1.7122,  -4.7591,  -3.6734],
        [ 12.5789,   0.1722,   1.6470],
        [  4.5503,   5.5359,   7.4735],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.2608,  -2.5256,  -9.1446],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.5456,   6.0861,   8.1101],
        [  2.7976,  -4.4682,  -4.2020],
        [  1.9241,   7.3503,   7.9004],
        [ -1.7009,   7.8851,  -8.2437],
        [  6.2630,   2.1057,  -5.2136],
        [  8.6271,  -8.0715,  -7.4813],
        [  7.5309,  11.2617,   7.2656],
        [ -8.7735,   2.5640,  -6.3313],
        [  0.5250, -11.0306,   5.5358],
        [ -3.5029,   7.5138,   8.1853],
        [  4.9531,   5.0576,   5.6794],
        [  6.8704,  -9.3809,   7.5769],
        [  4.5800,   5.5034,  -3.9432],
        [  2.4572,   7.1085,   7.7376],
        [  0.8700,  -4.8132,  -7.8042],
        [  6.3516,   0.6425,   0.3425],
        [  2.7976,  -4.4682,  -4.2020],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.4732,  -2.2041,  -4.3830],
        [ -0.5739,   7.9056,   0.6382],
        [  0.3354,  -4.7708,  -4.9125],
        [  8.0877,  -8.5297,  -1.1050],
        [  3.5965,  -4.0754,  -3.5035],
        [  4.9337,  12.8566,  -3.6582]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0382, 0.0634,  ..., 0.2208, 0.0906, 0.3871],
        [0.0382, 0.0000, 0.0275,  ..., 0.1982, 0.1246, 0.4251],
        [0.0634, 0.0275, 0.0000,  ..., 0.1997, 0.1519, 0.4476],
        ...,
        [0.2208, 0.1982, 0.1997,  ..., 0.0000, 0.2503, 0.5676],
        [0.0906, 0.1246, 0.1519,  ..., 0.2503, 0.0000, 0.3208],
        [0.3871, 0.4251, 0.4476,  ..., 0.5676, 0.3208, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.5092834 ,  7.6374603 ,  7.72415048,  7.78980299,  7.60529892,
        7.83486943, 13.85158417,  9.50913849,  6.76661595,  6.1561646 ,
        6.1463896 ,  4.74468406, 13.26209545, 12.18045978, 13.56367857,
       12.53047489, 10.36844813, 10.84493732, 13.79762415, 10.57302546,
       13.96746816,  6.383897  , 13.61801526,  7.57685908,  7.62011734,
        5.81199422, 14.07623266,  9.19837792,  4.73784222,  5.90035324,
       10.1979669 ,  5.1805345 ,  5.35774573, 14.09176259, 13.6105595 ,
       10.78533838,  9.58347456, 11.72749923,  4.78544875,  7.39004226,
        9.91668182, 11.32149259,  9.15253344, 11.85743425, 14.0393949 ,
        6.02789634,  8.62260327,  9.23178204, 10.37998078,  4.91296205,
        6.61386436,  7.89745107, 13.97857591, 10.84493732, 12.86648771,
       10.08001894,  9.97220799,  6.39187285,  7.61088125, 12.70489563,
        8.37954853,  9.69792513, 13.97857591,  8.0550286 , 12.71843462,
        5.91677188,  9.44486609, 11.27160035,  5.39098669,  6.50409372,
       10.36844813,  8.09842704, 14.03505624,  8.4486771 ,  6.65259775,
        5.05774295, 12.58006038,  7.16598273,  8.29022197,  8.47371868,
        5.83560009, 13.15525803, 13.05148826,  5.27178363,  7.59795445,
        8.06643487,  6.60752373, 11.81423161, 13.54774355,  9.14044984,
       11.04313641,  8.29022197,  7.07902036, 11.62770332,  7.15990676,
        7.52117132,  4.89121805,  6.38399789,  5.27178363,  9.53015376,
        5.90035324,  7.92644746,  4.78258092, 11.75438006,  5.43544626,
       13.77075174])}
{'P': tensor([[1.8251e-02, 1.5295e-01, 1.2999e-01,  ..., 3.9808e-05, 6.7368e-05,
         7.6970e-04],
        [8.2730e-04, 5.4864e-02, 3.5540e-03,  ..., 1.6234e-04, 1.1047e-04,
         6.9009e-03],
        [4.6850e-04, 8.2824e-02, 1.5744e-02,  ..., 6.4898e-05, 3.7432e-05,
         1.1105e-03],
        ...,
        [8.5868e-06, 1.4510e-05, 1.5651e-05,  ..., 7.1074e-02, 3.6671e-06,
         4.6124e-05],
        [2.4717e-02, 8.7027e-03, 1.1203e-02,  ..., 1.3010e-05, 5.4775e-02,
         1.3778e-04],
        [2.1797e-05, 2.9908e-05, 2.4483e-05,  ..., 6.8234e-06, 3.9471e-05,
         4.6175e-02]]), 'X': tensor([[  2.4932,   7.0903,  -5.4214],
        [  0.1497,   7.8334,  -5.4412],
        [  1.0000,   7.6591,  -4.5011],
        [  0.4996,   7.7738,  -4.4711],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.6323,   7.9085,  -5.9544],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.1504,  -4.6726,   8.7908],
        [  2.5128, -10.9365,   9.9645],
        [ -7.3333,   4.9246,  -7.9212],
        [ -2.4956, -10.4682,   3.5632],
        [  3.6703,   6.3443,  -5.4401],
        [ -4.8665,  -9.3204,  -5.2564],
        [  5.9026,   3.3837,  -0.6573],
        [ -8.3653,   3.4184,   9.3454],
        [ -3.2053, -10.1993,  -3.8736],
        [  7.0358,  -9.2784,  -9.8479],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -5.1974,   6.7396,   3.3484],
        [  9.6816,   9.1536,  -5.8543],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.5626,  -2.0536,   8.4433],
        [  3.2647, -10.8066,  -0.3768],
        [ -1.7290,   7.8825,  -3.3831],
        [ -8.8363,   2.4086,  -5.6296],
        [ -8.4824,  -5.2575,  -0.0795],
        [  3.9663,   6.1018,  -5.4104],
        [  2.4697,  -4.5836,   1.4593],
        [  0.7685,  -4.8098,  -2.9310],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -8.3539,  -5.5111,   0.3373],
        [ -0.1963,   7.8765,  -1.1371],
        [ -6.8466,  -7.6488,   1.9648],
        [  5.5424,  -2.0885,   7.2487],
        [ -6.4368,   5.8356,   3.3014],
        [ -9.2345,  -3.1345,   3.9715],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.2911,  -2.4824,  -3.6877],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.5939,   6.4029,  -1.8357],
        [ 11.6019,   5.9540,  -5.1715],
        [  3.0719, -10.8449,  -4.1673],
        [  4.4633,   5.6287,  -0.7754],
        [  0.6933,   7.7336,   2.1179],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.9770, -10.6341,   8.0783],
        [  4.5430,   5.5438,  -3.8669],
        [  2.5469,  13.7192,  -9.6935],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.4996,   7.7738,  -4.4711],
        [  7.1697,  11.5343,  -0.7276],
        [  0.6396,  14.0772,  -6.3786],
        [  0.6933,   7.7336,   2.1179],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.0582,  -2.6640,   0.7132],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.6613,   7.0021,   4.5783],
        [ -1.8823, -10.6558,  -3.8952],
        [  9.9058,  -6.7203,   5.6928],
        [ -9.3107,  -2.7697,  -3.2919],
        [  1.7000,  -4.7609,   2.7935],
        [  1.0000,   7.6591,  -4.5011],
        [ 12.1379,   4.4125,  -3.8833],
        [  2.6321,  -4.5296,  -2.5159],
        [  6.9284,  11.7061,  -0.3732],
        [ -7.4733,   4.7555,   6.2490],
        [  9.2692,   9.6370,   7.0746],
        [  3.1756, -10.8247,   6.8882],
        [  4.5430,   5.5438,  -3.8669],
        [  2.6310, -10.9195,   3.8054],
        [  9.2979,   9.6048,   7.3138],
        [  2.5813,  -4.5471,   5.3597],
        [  6.0047,   3.1062,  -6.7914],
        [  5.9655,   3.2172,  -7.5953],
        [  5.9548,  -1.2273,  -2.6520],
        [  3.7121,   6.3117,   0.6738],
        [  6.1984,  -0.4327,   1.5438],
        [  2.6078, -10.9229,   6.2869],
        [ 11.7795,  -3.4697,   3.5624],
        [ -0.8675, -10.8815,   6.8561],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.7218,  -9.9934,   6.5764],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.3048,   0.1469,  -9.1691],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.3419,   0.5972,  -1.2841],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.4624,  -0.4633,  -7.3918],
        [  6.0305,   3.0296,   0.1375],
        [  3.3069,   6.6089,   8.7377],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -3.2486,   7.5925,  -1.3719],
        [ 12.2218,   4.0953,  -8.2072],
        [ 11.9695,  -2.9407,  -6.3498],
        [  6.0345,   3.0175,  -1.4424],
        [  4.5814, -10.4485,   2.5496],
        [ -7.6293,  -6.6859,  -5.5161],
        [  2.7590,  13.6621,   9.4496],
        [ 11.3691,   6.4760,   1.0787],
        [ -6.7992,   5.4991,   9.9186],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.9033,   2.2329,  -8.8912]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0961, 0.0682,  ..., 1.0304, 0.1568, 0.7537],
        [0.0961, 0.0000, 0.0626,  ..., 0.9936, 0.2513, 0.6595],
        [0.0682, 0.0626, 0.0000,  ..., 0.9687, 0.2196, 0.7129],
        ...,
        [1.0304, 0.9936, 0.9687,  ..., 0.0000, 1.1340, 1.1155],
        [0.1568, 0.2513, 0.2196,  ..., 1.1340, 0.0000, 0.8994],
        [0.7537, 0.6595, 0.7129,  ..., 1.1155, 0.8994, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.5158489 ,  7.83486943,  7.72415048,  7.78980299,  7.822854  ,
        7.9337647 ,  5.52416368,  6.14086582,  5.14371005, 11.22142195,
        8.83341608, 10.76154256,  7.32950385, 10.51437947,  6.80370254,
        9.0368458 , 10.6910768 , 11.64436684,  9.43112957,  8.51088913,
       13.32374056, 12.13568598,  5.92952538, 11.28895489,  8.06990483,
        9.15866607,  9.97964013,  7.27759233,  5.20659892,  4.87083348,
        9.54154096, 10.00794468,  7.8788965 , 10.26545792,  5.92281142,
        8.68832398,  9.75200848, 12.46513646,  5.84450127,  9.9353921 ,
        7.34250948, 13.04051744, 11.27160035,  7.18354885,  7.76457095,
       12.91143491, 11.35345345,  7.16747878, 13.95361098, 12.62576026,
        9.94285511,  7.78980299, 13.58101794, 14.09176259,  7.76457095,
       12.33167382, 12.62576026, 12.3489324 ,  9.47369628, 12.10222204,
        7.49076899, 10.8207297 , 11.9702762 ,  9.71392179,  5.05534609,
        7.72415048, 12.91505111,  5.23878371, 13.60275211,  8.8580671 ,
       13.37121923, 11.28093507,  7.16747878, 11.23201619, 13.36800073,
        5.2287058 ,  6.76056625,  6.77777606,  6.07992623,  7.32233733,
        6.2134961 , 11.22993218, 12.27991991, 10.91601655, 13.04217487,
       11.51557166,  4.85995002,  6.30647746, 10.86693708,  9.36093562,
        7.71075226,  9.47369628,  6.74872981,  7.39004226,  9.99145096,
        8.25828409, 12.889695  , 12.32545637,  6.74686071, 11.4088043 ,
       10.14436912, 13.93790235, 13.08414689,  8.74470218,  6.383897  ,
        9.17905491])}
{'P': tensor([[1.4109e-02, 2.2735e-01, 2.0963e-01,  ..., 2.1250e-06, 2.2576e-02,
         1.0137e-05],
        [9.5750e-04, 2.5723e-02, 8.7613e-03,  ..., 3.9239e-06, 5.0237e-02,
         1.5486e-05],
        [1.0412e-03, 5.1171e-02, 2.3860e-02,  ..., 3.7209e-06, 4.9478e-02,
         1.7020e-05],
        ...,
        [1.2555e-04, 9.7632e-05, 1.0557e-04,  ..., 1.0454e-01, 2.0463e-05,
         1.8817e-05],
        [3.9395e-04, 4.8004e-04, 4.5639e-04,  ..., 3.2657e-05, 8.9221e-02,
         3.0168e-05],
        [2.3644e-04, 2.4495e-04, 2.7977e-04,  ..., 1.1810e-05, 9.1549e-04,
         3.4557e-01]]), 'X': tensor([[  2.4572,   7.1085,   7.7376],
        [  1.9241,   7.3503,   7.9004],
        [  2.0243,   7.3086,   7.5432],
        [ -0.3432,   7.8900,   7.1805],
        [ -0.5709,   7.9055,   7.6089],
        [  1.9204,   7.3518,   8.4781],
        [ -5.6423,   6.4537,  -1.2091],
        [  2.7420,  -4.4896,  -3.7139],
        [  1.4749,  -4.7892,  -0.8089],
        [  2.1824,  -4.6647,   1.1985],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -1.1885, -10.8211,   0.0865],
        [  6.2555,   2.1474,   4.1685],
        [ -0.5385,   7.9037,   6.3830],
        [ -9.2707,   0.9661,   4.4863],
        [ -7.4733,   4.7555,   6.2490],
        [  3.6817,   6.3354,   6.1233],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.0937,  13.8293,  -2.3533],
        [ 11.0275,  -5.0666,   8.4154],
        [  3.4009,  -4.1874,  -5.9751],
        [  9.5203,   9.3483,   5.8781],
        [  5.5240,  -2.1198,   6.2055],
        [ -1.6220,   7.8917,   0.0315],
        [ -3.2736, -10.1703,   3.8712],
        [  5.6928,  12.4710,   3.6001],
        [  4.8051, -10.3703,   0.5268],
        [  4.7739,  -3.1218,  -4.1188],
        [ -0.4341, -10.9473,   7.4509],
        [ 12.5199,  -0.4711,   3.0436],
        [  9.6816,   9.1536,  -5.8543],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.2644,  -4.6434,   9.7508],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.7151,  -1.7700,   3.1542],
        [  4.2339,  -3.6268,   6.3125],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.0047,   3.1062,  -6.7914],
        [ 12.4691,   2.8525,  -5.5110],
        [  2.3367,   7.1676,  -9.5263],
        [  5.1292,   4.8183,  -5.7997],
        [ -2.7572,   7.7194,   5.9580],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.2365,  -0.2578,  -6.0212],
        [  1.8447,  -4.7372,   8.1235],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.3419,   0.5972,  -1.2841],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.2939,   6.6325,  -5.8311],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.4378,  -1.8606,  -2.5668],
        [  0.9445,   7.6736,   0.5095],
        [ -2.2188,   7.8218,   8.6064],
        [  4.5325,  -3.3643,   6.7699],
        [ -3.5029,   7.5138,   8.1853],
        [ 12.3759,  -1.3943,   6.4659],
        [ -9.4766,  -1.1116,  -2.1706],
        [  0.1206,  -4.7363,   8.7103],
        [  6.3460,   1.4252,   1.2291],
        [  3.9663,   6.1018,  -5.4104],
        [  6.5428,  -9.5728,  -2.9897],
        [ -0.1626, -10.9796,   6.4629],
        [  5.1203,  -2.7151,   6.2110],
        [  3.9515,  -3.8425,   6.1729],
        [  5.1519, -10.2387,   9.8191],
        [ -0.1626, -10.9796,   6.4629],
        [  1.9204,   7.3518,   8.4781],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.5832,   1.7543,  -7.6134],
        [  5.3924, -10.1396,  -4.0863],
        [  6.9693,  -9.3200,  -8.3822],
        [  3.3633,  13.4794,   0.5193],
        [ -9.1433,  -3.5003,  -3.0209],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.1931,  -4.7491,   7.4385],
        [  3.1064,  13.5607,   6.7605],
        [ -3.9858,  -9.8334,  -3.7323],
        [ 10.0400,   8.6922,  -6.2847],
        [ 12.0042,  -2.8350,   2.6864],
        [ -3.5029,   7.5138,   8.1853],
        [ 11.5966,   5.9666,   6.0083],
        [ -0.4341, -10.9473,   7.4509],
        [  5.9912,   3.1453,  -3.9989],
        [ -7.2986,  -7.1225,  -3.0814],
        [  9.4516,  -7.2501,  -1.8868],
        [  4.5767,   5.5071,   3.9069],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.5428,  -9.5728,  -2.9897],
        [ -0.8885,   7.9160,   0.0571],
        [  6.3122,  -9.6994,  -6.3282],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.2199,   4.6861,  -1.9304],
        [  4.6357,   5.4415,   7.9067],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.8359,   4.2735,   7.2943],
        [  1.2273,  -4.8081,   9.8129],
        [ -3.4986,   7.5152,   9.0712],
        [ -4.5778,  -9.5013,  -5.0667],
        [  4.3459,  -3.5327,   0.3667],
        [  2.6310, -10.9195,   3.8054],
        [ -3.1671,   7.6158,   6.0765],
        [  2.3367,   7.1676,  -9.5263]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0355, 0.0296,  ..., 0.4534, 0.2225, 0.9937],
        [0.0355, 0.0000, 0.0131,  ..., 0.4247, 0.1908, 0.9984],
        [0.0296, 0.0131, 0.0000,  ..., 0.4248, 0.1930, 0.9869],
        ...,
        [0.4534, 0.4247, 0.4248,  ..., 0.0000, 0.2374, 0.8760],
        [0.2225, 0.1908, 0.1930,  ..., 0.2374, 0.0000, 0.9478],
        [0.9937, 0.9984, 0.9869,  ..., 0.8760, 0.9478, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.52117132,  7.59795445,  7.58378655,  7.89745107,  7.92607342,
        7.5984809 ,  8.5724042 ,  5.26067019,  5.01114719,  5.14999799,
        6.35029274,  9.86089579, 10.88618114,  6.61386436,  7.92200342,
        9.32094055,  8.8580671 ,  7.32754882, 10.31584102, 13.98691426,
       12.13568598,  5.39450305, 13.3426541 ,  5.91677188,  8.05668609,
       10.68416401, 13.70893492, 11.42947632,  5.70405525, 10.95593729,
       12.52875793, 13.32374056,  8.83341608,  5.16611782,  8.92395381,
        5.98285319,  5.5748624 , 10.23946247,  6.76056625, 12.79126785,
        7.53884104,  7.03733775,  8.19703512, 10.31886066,  6.24187044,
        5.08372149,  5.80312949,  9.36093562, 11.81423161, 12.53047489,
       13.09737053, 12.3828957 ,  9.09661212,  9.61942445,  7.73151309,
        8.1303945 ,  5.64464807,  8.29022197, 12.45418236,  9.54154096,
        4.73784222,  6.50409372,  7.27759233, 11.59512194, 10.98076873,
        5.79562691,  5.51177501, 11.46176259, 10.98076873,  7.5984809 ,
       10.00794468, 12.70489563, 11.48434767, 11.63765273, 13.89264822,
        9.79039371,  5.52416368,  4.75303127, 13.91198191, 10.61047596,
       13.27994304, 12.3344536 ,  8.29022197, 13.04156075, 10.95593729,
        6.76661595, 10.1979669 , 11.9120321 ,  7.16058587,  9.78147542,
        9.61942445, 11.59512194,  7.96575782, 11.57249984,  8.25828409,
        7.01475376,  7.14837514,  8.90405823,  8.92549752,  4.96231532,
        8.28968067, 10.54658597,  5.60064319, 11.23201619,  8.24808832,
        7.53884104])}
{'P': tensor([[9.7124e-03, 1.6902e-01, 1.1746e-01,  ..., 3.5274e-04, 2.9448e-05,
         7.6556e-06],
        [7.9910e-04, 2.0178e-02, 1.0577e-02,  ..., 8.0694e-04, 4.4884e-05,
         1.4909e-05],
        [1.1406e-03, 2.8332e-02, 1.9938e-02,  ..., 6.7325e-04, 4.5980e-05,
         1.3701e-05],
        ...,
        [3.0117e-04, 3.5424e-04, 3.2021e-04,  ..., 2.2907e-01, 1.3956e-05,
         2.5036e-04],
        [2.2814e-02, 2.3812e-02, 2.3022e-02,  ..., 6.1260e-04, 4.1350e-02,
         1.5844e-05],
        [2.9784e-03, 2.8577e-03, 2.5200e-03,  ..., 3.7354e-03, 3.6617e-03,
         2.2979e-01]]), 'X': tensor([[  2.3928,   7.1404,  -1.7035],
        [  2.0039,   7.3173,  -1.5885],
        [  2.0547,   7.2957,  -1.9343],
        [  0.9809,   7.6642,  -1.4217],
        [  1.1006,   7.6318,  -2.4877],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.2804,   2.0019,   0.8607],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.6308,  13.9254,  -2.0984],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.0190,   4.9705,  -7.9752],
        [  0.4996,   7.7738,  -4.4711],
        [ 12.2592,   3.9425,   2.0383],
        [  9.9750,  -6.6338,  -1.2440],
        [ -1.9714,   7.8563,  -1.8084],
        [ -3.3812, -10.1235,   7.4384],
        [  6.3535,   0.6792,  -7.5101],
        [ -1.6086,   7.8928,   4.3385],
        [ -3.9508,  -9.8515,   1.7923],
        [  5.7784,  -1.6401,   9.5587],
        [  5.9003,  12.3550,   2.1917],
        [ 11.7794,   5.5086,  -9.6929],
        [ -8.9033,   2.2329,  -8.8912],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.3616,  -2.4750,   3.7765],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -3.9700,   7.3453,  -4.1486],
        [  5.5158,  -2.1337,   3.5246],
        [  9.3560,   9.5391,   9.4943],
        [  4.7813,  -3.1139,   8.2691],
        [ -2.0840,   7.8415,  -7.8657],
        [  0.1460,   7.8340,  -9.4614],
        [  2.2849, -10.9655,  -7.6542],
        [  4.4633,   5.6287,  -0.7754],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.2345,  -3.1345,   3.9715],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.3534,   1.2990,  -7.2942],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.1756, -10.8247,   6.8882],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.2311,  -2.0445,   1.3545],
        [  9.8082,  -6.8396,   4.9615],
        [  1.3403,  -4.8011,  -1.1698],
        [  4.1999,  -3.6543,  -4.1998],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.8766,   7.3694,   1.1887],
        [  6.1989,  -9.7591,   0.8289],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.9314,  -9.3435,  -1.2729],
        [ -1.6086,   7.8928,   4.3385],
        [ 11.0891,  -4.9558,  -2.3178],
        [  4.1999,  -3.6543,  -4.1998],
        [ -5.9872,   6.2037,  -9.4477],
        [  9.3836,   9.5075,  -5.9278],
        [  8.9284,  10.0036,  -6.9210],
        [ -7.6719,  -6.6260,   7.1468],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.3539,  -5.5111,   0.3373],
        [  0.3038,   7.8092,   0.4164],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -7.5720,   4.6310,   2.1717],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.1204,  -5.9308,   4.0761],
        [  0.6952,  -4.8060,   8.8976],
        [ -5.0550,   6.8231,  -9.2502],
        [  9.3607,   9.5337,  -4.5412],
        [  2.0547,   7.2957,  -1.9343],
        [  5.8275,  -1.5334,   2.0124],
        [  6.2555,   2.1474,   4.1685],
        [  2.0870,  -4.6877,   9.6823],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.1905,   3.7253,   4.1175],
        [  4.9531,   5.0576,   5.6794],
        [  5.1196,  -2.7160,   4.2705],
        [  6.3122,  -9.6994,  -6.3282],
        [  9.2979,   9.6048,   7.3138],
        [  3.6154,  -4.0640,  -7.9698],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.3550,   0.7129,   0.4917],
        [  6.2905,   1.9366,   2.1169],
        [  4.6357,   5.4415,   7.9067],
        [  6.3535,   0.6792,  -7.5101],
        [ 10.9875,   7.2207,  -8.8549],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.8930,  12.3592,  -8.9201],
        [ -7.4808,   4.7462,   2.0266],
        [  2.1625,  -4.6697,   7.0658],
        [  2.1440, -10.9811,   8.5281],
        [ -8.6381,  -4.9238,  -4.4077],
        [  8.0877,  -8.5297,  -1.1050],
        [ 12.5650,   2.0132,   4.6272],
        [ -5.6174,   6.4708,  -1.7339],
        [  6.9433,  -9.3362,  -0.3421],
        [  3.3943,   6.5483,   1.0790],
        [  6.8704,  -9.3809,   7.5769],
        [  4.8674,  12.8875,   6.7622],
        [ 12.6000,   1.3839,  -3.2992],
        [ -8.8835,  -4.3178,   9.3080],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.3048,   0.1469,  -9.1691],
        [ 11.5456,   6.0861,   8.1101],
        [ -4.0868,   7.2981,   8.1874],
        [  9.6587,   9.1817,  -0.8212],
        [  6.1222,  -0.7261,   8.8204]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0251, 0.0249,  ..., 0.7297, 0.3962, 0.6718],
        [0.0251, 0.0000, 0.0199,  ..., 0.7131, 0.4186, 0.6698],
        [0.0249, 0.0199, 0.0000,  ..., 0.7322, 0.4199, 0.6885],
        ...,
        [0.7297, 0.7131, 0.7322,  ..., 0.0000, 0.9137, 0.4439],
        [0.3962, 0.4186, 0.4199,  ..., 0.9137, 0.0000, 0.6389],
        [0.6718, 0.6698, 0.6885,  ..., 0.4439, 0.6389, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.53063885,  7.58668144,  7.57946459,  7.72668467,  7.71075226,
        7.8575648 ,  6.59176018,  9.55593374, 14.02058621, 10.01804223,
        7.06372172,  7.78980299, 12.87752052, 11.97949273,  8.09984231,
       10.67322527,  6.38968357,  8.0550286 , 10.61417181,  6.00663314,
       13.69162795, 13.00380624,  9.17905491,  8.09704052,  9.68324553,
       10.32715715,  8.34949383,  5.91408504, 13.3614598 ,  5.70591986,
        8.1137452 ,  7.83534987, 11.20100795,  7.18354885,  9.23071698,
        9.75200848,  5.5205189 ,  6.48486171,  5.18047638, 11.28093507,
       13.04051744, 12.40074901, 11.95743558,  4.98461983,  5.5671389 ,
        5.19577146,  7.60463349, 11.56146873,  8.34949383, 11.63383221,
        8.0550286 , 12.1460931 ,  5.5671389 ,  8.62162372, 13.35832725,
       13.40850328, 10.1371552 ,  8.04718433, 10.00794468,  7.81509631,
        9.562174  ,  8.87587457,  7.58668144, 10.05558893,  4.85603886,
        8.49161154, 13.36092624,  7.57946459,  6.02588517,  6.61386436,
        5.13125513,  8.04664886, 10.00794468,  8.99791316,  7.07902036,
        5.79543038, 11.57249984, 13.36800073,  5.43943791,  5.90035324,
        6.3949043 ,  6.58184474,  7.14837514,  6.38968357, 13.14777433,
        7.88677538, 13.6922425 ,  8.859416  ,  5.14607014, 11.18839169,
        9.94285511, 11.75438006, 12.72524615,  8.56889738, 11.63502679,
        7.37576154, 11.62770332, 13.77604293, 12.67576792,  9.87720102,
        5.22903117,  6.30647746, 13.05148826,  8.36445758, 13.32645552,
        6.16512856])}
{'P': tensor([[2.2368e-02, 2.0307e-01, 1.6112e-01,  ..., 8.0645e-05, 1.3243e-06,
         3.4857e-06],
        [3.2881e-04, 2.0511e-02, 1.1660e-01,  ..., 3.6338e-05, 9.5414e-07,
         1.7368e-06],
        [4.6739e-04, 3.3738e-03, 3.3493e-02,  ..., 4.9374e-05, 1.2660e-06,
         2.4773e-06],
        ...,
        [1.5345e-02, 9.0494e-03, 6.6803e-03,  ..., 9.7710e-02, 1.7501e-04,
         5.2733e-04],
        [7.5785e-05, 5.4281e-05, 4.1311e-05,  ..., 1.9139e-03, 2.1504e-01,
         6.7386e-04],
        [1.3256e-03, 4.7383e-04, 3.5472e-04,  ..., 1.6594e-02, 4.9709e-05,
         3.9520e-02]]), 'X': tensor([[ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0724, 0.1020,  ..., 0.3180, 1.0254, 0.2882],
        [0.0724, 0.0000, 0.0297,  ..., 0.3771, 1.0560, 0.3590],
        [0.1020, 0.0297, 0.0000,  ..., 0.4040, 1.0724, 0.3886],
        ...,
        [0.3180, 0.3771, 0.4040,  ..., 0.0000, 0.7591, 0.1363],
        [1.0254, 1.0560, 1.0724,  ..., 0.7591, 0.0000, 0.8838],
        [0.2882, 0.3590, 0.3886,  ..., 0.1363, 0.8838, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.53884104,  7.83534987,  7.88677538,  8.05242591,  8.08320235,
        8.06643487,  9.75200848,  5.46244988,  7.72668467,  6.76056625,
        5.70591986,  9.15253344,  5.18277858, 13.14690137,  5.79562691,
        9.79039371,  9.55593374, 11.4424816 ,  5.26067019,  6.1561646 ,
        9.99211663,  9.6282763 , 12.44177641,  7.60529892,  8.22728399,
       10.8207297 ,  6.02789634,  9.73579738,  9.78286733, 11.81423161,
       12.04924159,  5.56809335,  7.5984809 , 11.74901778,  6.18668463,
       12.38881635,  7.4544866 , 12.25806559, 12.91143491,  6.83083863,
       10.84493732,  5.08910491, 11.35182589,  9.50654967,  4.9160962 ,
       10.66631784,  8.83341608,  7.32754882, 12.77402351,  6.10774544,
       11.51811603,  9.36378438, 13.45124917, 13.41713661,  5.90035324,
        9.48337063, 11.24434374, 12.85032836,  5.10446653, 11.73567759,
       13.32645552,  6.56833864,  9.7712321 ,  7.57946459,  8.95983586,
        9.69629   , 11.2845608 ,  5.48276038,  8.22728399,  9.54154096,
        6.61386436,  9.96152788,  8.67227386,  7.46248947,  8.7273569 ,
        5.60064319,  7.15100303,  7.46248947,  8.06643487,  7.32233733,
        7.96575782,  5.91854397, 11.8159254 ,  9.05843201,  4.8763135 ,
       12.30858677, 11.48663767,  6.81084724, 12.10942412,  5.92281142,
        7.4544866 , 11.28215429,  5.20659892,  9.50654967, 10.35464916,
        6.1561646 , 10.1979669 , 12.41366754,  5.40689332, 11.81423161,
        7.83534987,  8.09704052,  8.87007495,  6.61211224,  5.91854397,
       12.4217523 ])}
{'P': tensor([[2.9546e-02, 2.0106e-01, 1.7460e-01,  ..., 6.2573e-05, 1.4514e-04,
         1.7926e-05],
        [3.9829e-04, 3.2685e-02, 1.0534e-02,  ..., 1.8370e-04, 5.5784e-05,
         1.3658e-05],
        [4.2842e-04, 4.5764e-02, 2.9432e-02,  ..., 1.2223e-04, 4.6514e-05,
         1.0297e-05],
        ...,
        [4.3905e-06, 4.3647e-06, 3.7566e-06,  ..., 3.2877e-02, 2.1996e-05,
         2.4052e-04],
        [1.1049e-02, 6.4037e-03, 4.8165e-03,  ..., 9.9906e-05, 7.8832e-02,
         5.6352e-04],
        [5.1999e-04, 2.3741e-04, 1.9178e-04,  ..., 4.9268e-05, 2.6489e-02,
         7.1418e-02]]), 'X': tensor([[ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0831, 0.0880,  ..., 1.0880, 0.5245, 0.7537],
        [0.0831, 0.0000, 0.0370,  ..., 1.0384, 0.5675, 0.7886],
        [0.0880, 0.0370, 0.0000,  ..., 1.0707, 0.5964, 0.8200],
        ...,
        [1.0880, 1.0384, 1.0707,  ..., 0.0000, 0.9953, 0.9700],
        [0.5245, 0.5675, 0.5964,  ..., 0.9953, 0.0000, 0.2368],
        [0.7537, 0.7886, 0.8200,  ..., 0.9700, 0.2368, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.5403822 ,  7.92200342,  7.89745107,  7.92607342,  8.04664886,
        8.09842704, 11.76830981,  4.74468406,  8.49161154,  6.2322769 ,
        9.41244865, 12.17451685, 11.9182956 ,  8.62631801,  6.1561646 ,
       10.41909211, 11.95743558, 13.96746816, 11.51557166, 11.36089355,
       11.4890861 , 11.59512194,  6.18668463,  7.19513806,  7.50917551,
       13.46553623,  6.81084724, 10.95593729,  9.79039371,  5.04028538,
       13.77222752,  5.39450305, 11.4890861 ,  9.14973319, 11.83509372,
        6.90115901,  7.32950385, 11.85743425,  9.58347456, 13.6922425 ,
       13.72594022, 12.3344536 ,  5.10444097,  7.72291946,  9.14973319,
       11.00726483,  7.06272256,  6.7593917 ,  8.90405823, 11.20100795,
        5.16205087, 10.31584102,  9.66977802, 12.97370371,  8.30729352,
        9.17016857,  9.18690018, 13.88683436,  9.10945678, 10.39942497,
       11.42947632,  4.90676242,  5.91677188, 11.77124373,  8.09899874,
        6.56833864,  8.95133961, 12.70489563,  9.79814997, 13.39372426,
       13.15525803, 13.91198191,  9.32094055,  7.73692436, 11.37939695,
       14.03618649, 12.35421461,  7.32950385,  7.46248947, 12.11735334,
        7.98469166,  7.02253235,  7.72291946, 13.98691426,  8.31625171,
       10.27572673,  9.63742669,  8.04483495,  8.8580671 ,  8.62631801,
       13.98691426,  5.43943791,  5.94293185,  6.81117802, 11.33837396,
       10.01804223,  6.39187285, 10.71162681, 12.1460931 , 12.00995898,
       12.889695  ,  9.09502385,  7.49076899,  9.75401449,  6.80370254,
        5.81199422])}
{'P': tensor([[1.8532e-02, 3.5514e-01, 1.6599e-01,  ..., 3.3546e-06, 3.7299e-03,
         4.1002e-05],
        [9.6257e-04, 2.7716e-02, 2.3144e-01,  ..., 4.2145e-06, 6.7051e-03,
         5.8188e-05],
        [7.3148e-04, 1.1194e-03, 2.5497e-02,  ..., 3.0205e-06, 6.7756e-03,
         4.9480e-05],
        ...,
        [1.2137e-03, 1.0202e-03, 8.3065e-04,  ..., 3.3221e-02, 9.2402e-06,
         5.4723e-05],
        [1.6748e-04, 1.8828e-04, 2.3409e-04,  ..., 2.1342e-06, 1.3176e-01,
         1.9821e-04],
        [3.5752e-03, 2.9797e-03, 2.1102e-03,  ..., 4.3417e-03, 1.4147e-04,
         7.7956e-02]]), 'X': tensor([[  2.3200,   7.1756,   1.1938],
        [  1.8766,   7.3694,   1.1887],
        [  0.9445,   7.6736,   0.5095],
        [  0.3038,   7.8092,   0.4164],
        [  0.6933,   7.7336,   2.1179],
        [ -0.5739,   7.9056,   0.6382],
        [  5.0506,   4.9277,  -9.5094],
        [  6.1222,  -0.7261,   8.8204],
        [ -8.8363,   2.4086,  -5.6296],
        [ 12.5199,  -0.4711,   3.0436],
        [ -6.4620,  -8.0449,   6.2719],
        [  8.6420,  -8.0580,   6.5781],
        [ -6.4844,  -8.0230,   8.5209],
        [  6.3497,   1.3680,  -1.9076],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.1426,   3.8048,   9.3542],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.8318,   3.5567,  -4.7692],
        [  0.9074,  -4.8139,   8.1674],
        [ -8.6681,  -4.8556,   9.2518],
        [  1.7500,   7.4187,   4.4237],
        [  5.1472,  -2.6801,   8.1194],
        [  9.9217,  -6.7006,  -5.6728],
        [  5.5424,  -2.0885,   7.2487],
        [  6.1372,   2.6726,  -6.7885],
        [  5.3858,   4.4260,  -9.6769],
        [ -7.9925,  -6.1423,   2.8990],
        [  3.7992,  -3.9474,  -8.9019],
        [ 10.4758,  -5.9539,   1.3667],
        [ -1.0482, -10.8487,  -5.9066],
        [  3.4920,  -4.1366,  -8.3955],
        [  0.6142,  14.0802,   0.5407],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.9883,   3.1536,   6.2320],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.8318,   3.5567,  -4.7692],
        [  6.2675,  12.1379,  -8.9720],
        [  2.5375,   7.0676,  -4.3490],
        [  5.8617,   3.4853,  -9.8080],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.3376,  -4.6231,   9.8436],
        [ -9.4159,  -2.0739,   9.6841],
        [  2.2860,   7.1916,  -6.1011],
        [  4.7813,  -3.1139,   8.2691],
        [  6.3535,   0.6792,  -7.5101],
        [  6.3211,   1.7017,  -0.2164],
        [  3.8111, -10.6787,  -4.6539],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.7565,  11.8237,  -4.9882],
        [ -9.3604,   0.4855,  -3.0158],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4624,  -0.4633,  -7.3918],
        [  6.3460,   1.4252,   1.2291],
        [  5.4336,  -2.2676,   5.6933],
        [  4.4049,   5.6892,  -4.4394],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.7558,  -4.4844,   0.0636],
        [ -9.1706,   1.3863,   8.4032],
        [  4.8653,   5.1694,  -4.0823],
        [  2.5010,  -4.5736,  -0.8989],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.3554,   5.7393,  -9.0766],
        [  1.8447,  -4.7372,   8.1235],
        [  6.2905,   1.9366,   2.1169],
        [  7.5784,  -8.9141,   5.8410],
        [  3.9022,  13.2906,   5.5788],
        [ -8.8154,   2.4611,   2.5198],
        [ 10.4737,  -5.9570,  -1.2108],
        [  6.3535,   0.6792,  -7.5101],
        [  3.2809,  13.5061,  -4.0199],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.6019,   5.9540,  -5.1715],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.9526,   3.2525,  -4.3821],
        [ -8.0534,  -6.0429,   2.4583],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.3804,  -2.3504,   7.2744],
        [ 12.5789,   0.1722,   1.6470],
        [  2.6970,   6.9826,   9.7350],
        [ -8.3717,  -5.4769,  -9.2228],
        [  5.9003,  12.3550,   2.1917],
        [ -4.0868,   7.2981,   8.1874],
        [  4.5340,  -3.3629,   8.9652],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.6095,  -1.9702,  -7.8233],
        [  8.6420,  -8.0580,   6.5781],
        [  2.0449,  13.8402,  -7.1642],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.6681,  -4.8556,   9.2518],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.2076,  13.8032,  -1.5099],
        [ -1.8367,   7.8718,  -8.6523],
        [  9.3753,   9.5171,   3.9131],
        [ 11.3691,   6.4760,   1.0787],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.8930,  12.3592,  -8.9201],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.9152,  12.8653,  -1.4459],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.7420,  -4.4896,  -3.7139],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.5919,   0.4035,  -5.8460],
        [ -6.3306,   5.9272,   5.6935],
        [  4.3459,  -3.5327,   0.3667]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0260, 0.0800,  ..., 0.6149, 0.5274, 0.1224],
        [0.0260, 0.0000, 0.0603,  ..., 0.6236, 0.5107, 0.1434],
        [0.0800, 0.0603, 0.0000,  ..., 0.5984, 0.5235, 0.1641],
        ...,
        [0.6149, 0.6236, 0.5984,  ..., 0.0000, 1.1201, 0.5152],
        [0.5274, 0.5107, 0.5235,  ..., 1.1201, 0.0000, 0.6485],
        [0.1224, 0.1434, 0.1641,  ..., 0.5152, 0.6485, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.54127425,  7.60463349,  7.73151309,  7.81509631,  7.76457095,
        7.92644746,  7.05627461,  6.16512856,  9.15866607, 12.52875793,
       10.31886066, 11.8159254 , 10.31584102,  6.49538841,  5.62005629,
        8.98765644, 13.00380624,  6.83083863,  4.89869205,  9.9353921 ,
        7.62232307,  5.80312949, 11.97238624,  5.92281142,  6.69389431,
        6.97106912, 10.08001894,  5.47866221, 12.0495521 , 10.8992524 ,
        5.41348121, 14.09357547,  5.63824971,  6.76791483, 11.63724792,
        6.83083863, 13.66051884,  7.5092834 ,  6.81961996,  8.20436764,
        5.1805345 ,  9.64157461,  7.54620849,  5.70591986,  6.38968357,
        6.54615887, 11.33837396,  5.8828986 , 13.61801526,  9.37296102,
       13.25336001,  9.47369628,  6.50409372,  5.88782913,  7.19513806,
       11.72749923,  5.26342203,  9.27474386,  7.09887557,  5.21280156,
        8.5724042 ,  7.20482224,  5.08372149,  6.58184474, 11.70016217,
       13.85158417,  9.15253344, 12.04924159,  6.38968357, 13.89886469,
        5.95297459,  6.01356004, 13.04051744,  6.54615887, 12.22503754,
        6.78325151, 10.06850677, 10.1979669 ,  5.87132718, 12.58006038,
        7.48538905, 10.00410625, 13.69162795,  8.36445758,  5.64500512,
        8.22728399,  5.94541803, 11.8159254 , 13.99048094,  7.26074925,
        9.9353921 ,  8.25828409, 13.97857591,  8.08320235, 13.35927527,
       13.08414689,  6.18668463, 13.6922425 , 10.31584102, 13.77222752,
        9.97220799,  5.26067019, 11.99633819, 12.59840193,  8.67227386,
        5.60064319])}
{'P': tensor([[1.6372e-02, 1.5008e-01, 1.2912e-01,  ..., 9.9946e-07, 3.8712e-03,
         4.4736e-04],
        [1.0363e-03, 5.8901e-02, 5.1440e-02,  ..., 4.5003e-06, 3.9578e-02,
         2.8631e-04],
        [1.0892e-03, 2.9171e-02, 6.0811e-02,  ..., 6.0576e-06, 4.2911e-02,
         3.0698e-04],
        ...,
        [2.6533e-06, 1.4156e-06, 1.7615e-06,  ..., 2.5114e-02, 6.4107e-07,
         8.5345e-07],
        [7.5858e-05, 1.1682e-04, 1.1946e-04,  ..., 5.8893e-05, 7.4230e-02,
         4.5683e-06],
        [1.5746e-02, 4.8245e-02, 7.1807e-02,  ..., 3.2934e-06, 1.2789e-02,
         3.0359e-01]]), 'X': tensor([[ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0778, 0.0863,  ..., 1.1250, 0.5135, 0.4607],
        [0.0778, 0.0000, 0.0543,  ..., 1.1088, 0.4394, 0.5372],
        [0.0863, 0.0543, 0.0000,  ..., 1.0581, 0.4345, 0.5212],
        ...,
        [1.1250, 1.1088, 1.0581,  ..., 0.0000, 0.9712, 1.2382],
        [0.5135, 0.4394, 0.4345,  ..., 0.9712, 0.0000, 0.9464],
        [0.4607, 0.5372, 0.5212,  ..., 1.2382, 0.9464, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.54620849,  7.822854  ,  7.83486943,  7.80625981,  7.9337647 ,
        7.63726134, 11.48434767, 14.0393949 , 12.3828957 , 10.95593729,
        5.26342203,  6.00732193, 11.70168755,  7.05627461,  7.78980299,
        7.02496223,  4.89121805,  7.822854  , 12.35625694, 11.38774149,
       13.26590071,  9.71392179, 10.74655346,  9.34061938, 10.04916678,
        5.30158497, 10.04916678,  5.13125513, 10.31584102,  6.12886024,
       13.39372426,  5.96556248,  5.1805345 , 13.59480906, 10.60226614,
        4.74770899,  9.43737523, 11.63724792, 13.6829843 ,  9.76896904,
       10.7506423 , 12.3828957 , 10.78533838, 13.0493368 , 13.25017211,
        8.06517143,  7.57685908,  5.08910491,  5.40689332,  8.36995953,
       11.1913598 , 10.76154256,  7.59795445, 10.48721382,  5.63824971,
       13.46553623,  6.31576264,  7.15990676, 13.59480906,  5.87305005,
       11.87665501,  4.78258092, 11.70016217,  9.25336371, 12.1494023 ,
       13.13205357,  9.66977802,  9.77588801, 11.785566  ,  6.69389431,
        5.90035324, 11.1913598 , 10.01804223, 13.66051884, 11.87665501,
        5.47866221,  7.27759233,  8.92098752,  7.62011734,  5.14607014,
        5.91854397, 12.1494023 ,  7.50500064,  9.71392179,  7.80625981,
       12.3344536 , 12.5085995 ,  8.86082124,  9.41244865,  5.63824971,
        7.44805901,  7.05627461, 12.71843462, 12.88270018, 12.91505111,
       11.51811603, 13.45124917, 12.05395056,  6.81084724,  9.85330646,
       12.4217523 , 12.49291768,  9.562174  , 10.86693708,  8.8353674 ,
       13.77222752])}
{'P': tensor([[2.4222e-02, 2.1239e-01, 1.6027e-01,  ..., 1.5044e-04, 1.0420e-04,
         3.0473e-06],
        [5.1793e-04, 3.1493e-02, 1.2156e-03,  ..., 4.2104e-04, 2.8133e-05,
         2.3054e-06],
        [1.2084e-03, 3.1676e-01, 3.1404e-02,  ..., 2.5258e-04, 5.0593e-05,
         2.2378e-06],
        ...,
        [4.9784e-06, 5.1301e-06, 4.4746e-06,  ..., 2.3757e-02, 4.1172e-07,
         7.6698e-06],
        [8.4951e-03, 8.2752e-03, 8.3792e-03,  ..., 6.3136e-06, 1.6993e-02,
         1.6599e-06],
        [4.6713e-05, 3.0790e-05, 2.7211e-05,  ..., 1.4698e-05, 6.6920e-05,
         5.2143e-02]]), 'X': tensor([[ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 7.4366e-02, 6.0215e-02,  ..., 7.9108e-01, 5.9715e-01,
         9.9086e-01],
        [7.4366e-02, 0.0000e+00, 1.4835e-02,  ..., 7.3453e-01, 6.6155e-01,
         1.0229e+00],
        [6.0215e-02, 1.4835e-02, 0.0000e+00,  ..., 7.4782e-01, 6.4709e-01,
         1.0200e+00],
        ...,
        [7.9108e-01, 7.3453e-01, 7.4782e-01,  ..., 0.0000e+00, 1.3857e+00,
         1.0947e+00],
        [5.9715e-01, 6.6155e-01, 6.4709e-01,  ..., 1.3857e+00, 0.0000e+00,
         1.2272e+00],
        [9.9086e-01, 1.0229e+00, 1.0200e+00,  ..., 1.0947e+00, 1.2272e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.56617037,  7.85919687,  7.73692436,  8.04483495,  7.92607342,
        8.04718433, 10.42667093,  4.89121805,  6.61211224,  6.00732193,
       12.76408562,  6.4020404 ,  7.37902766,  7.61088125,  9.0368458 ,
        9.51074763,  5.98285319, 10.80728788, 10.00794468, 10.33712258,
        4.75303127,  5.92952538, 12.62711252, 13.36800073, 11.785566  ,
        7.61088125,  6.02588517, 14.0393949 ,  8.68979052,  8.50268322,
       11.35182589,  8.89458273,  8.89458273,  9.15866607, 10.09979747,
       13.103709  ,  9.94173708, 13.36800073, 10.37998078,  6.81084724,
       13.79676654, 13.92390689,  6.06416957,  6.53290663,  7.18354885,
       11.87412439,  6.74686071, 13.14690137,  7.53063885,  5.40689332,
        9.26455795, 13.3607379 ,  8.28968067, 10.24557047,  9.63742669,
        9.4997987 , 10.06850677,  8.8737556 ,  4.75303127,  9.10945678,
       11.69264243, 13.27717982, 11.62158554,  6.4020404 ,  8.91551231,
        5.19577146,  8.62329466, 12.67576792,  9.69792513, 11.48734978,
       12.4980845 , 13.04156075,  6.1463896 ,  5.51177501, 14.10045797,
        8.30125201, 10.91805683,  4.92497514,  5.05963217, 10.24557047,
        6.53250937,  4.96231532,  5.53255882,  7.73692436,  6.02779201,
       11.89105259,  8.7907923 ,  9.17905491,  8.99276564, 13.68672855,
       12.88270018, 12.30858677, 13.32613445, 13.98691426,  9.38232121,
        6.77777606,  7.53063885,  5.62005629,  7.03733775, 10.7167285 ,
        7.66377554,  5.69913817,  9.45995955,  9.60412735, 13.43127131,
        6.41202986])}
{'P': tensor([[2.0198e-02, 1.7087e-01, 1.4395e-01,  ..., 2.4973e-03, 1.9356e-05,
         8.8681e-05],
        [2.8863e-04, 1.9870e-02, 7.6517e-02,  ..., 6.7094e-03, 1.0566e-05,
         2.1784e-04],
        [2.7533e-04, 3.4671e-03, 2.0087e-02,  ..., 6.3933e-03, 9.2370e-06,
         2.1693e-04],
        ...,
        [5.3258e-05, 9.1914e-05, 9.5468e-05,  ..., 6.5959e-02, 3.3195e-05,
         6.8256e-02],
        [1.4730e-03, 8.6262e-04, 6.9538e-04,  ..., 2.5343e-04, 8.6031e-02,
         5.9365e-05],
        [3.7590e-06, 3.7701e-06, 3.6152e-06,  ..., 2.2833e-05, 2.5031e-05,
         1.9206e-02]]), 'X': tensor([[  2.0729,   7.2878,   7.0880],
        [ -0.3432,   7.8900,   7.1805],
        [ -0.5709,   7.9055,   7.6089],
        [ -0.5385,   7.9037,   6.3830],
        [ -1.5451,   7.8975,   7.6104],
        [ -1.6906,   7.8860,   7.7142],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.7298,  13.3537,  -9.4653],
        [  3.2366,  -4.2732,   5.1532],
        [ 11.2939,   6.6325,  -5.8311],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.3924, -10.1396,  -4.0863],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.0305,   3.0296,   0.1375],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.3889,  -4.6082,  -8.7291],
        [ -3.5173, -10.0623,   4.8280],
        [  2.6078, -10.9229,   6.2869],
        [ 11.4495,   6.3026,   6.8257],
        [  5.6928,  12.4710,   3.6001],
        [ -9.4671,  -0.5553,   0.1117],
        [ 10.1503,   8.5414,  -7.2681],
        [  5.5158,  -2.1337,   3.5246],
        [  9.3607,   9.5337,  -4.5412],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.4242, -10.1261,   3.9595],
        [  6.3583,   1.1729,   8.7835],
        [ -9.4671,  -0.5553,   0.1117],
        [  4.1537,   5.9340,  -6.0423],
        [  1.7659,   7.4127,   4.7871],
        [  6.0310,  12.2795,  -6.4528],
        [ -7.8126,   4.3066,   5.1791],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3549,   0.7096,   6.9198],
        [ 11.0275,  -5.0666,   8.4154],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.8325,  -1.5222,  -2.1341],
        [ -6.5295,  -7.9785,  -7.4429],
        [  5.6260,  -1.9401,   3.0259],
        [ -8.7409,  -4.6837,  -6.5840],
        [  5.5158,  -2.1337,   3.5246],
        [  5.6022,  -1.9833,   8.3081],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.5859,  -5.0393,   0.5476],
        [ -4.9217,   6.8979,  -1.7667],
        [  2.0243,   7.3086,   7.5432],
        [  6.3551,   0.7149,   1.3344],
        [  2.6310, -10.9195,   3.8054],
        [  5.6260,  -1.9401,   3.0259],
        [ 11.2767,  -4.6006,   0.7868],
        [ -2.0721,   7.8432,   6.6380],
        [  3.7298,  13.3537,  -9.4653],
        [  6.0345,   3.0175,  -1.4424],
        [ -7.0120,   5.2825,   7.7762],
        [  9.5203,   9.3483,   5.8781],
        [ -8.6050,   2.9451,  -0.7012],
        [  6.2242,  -0.3171,   7.4978],
        [  5.9548,  -1.2273,  -2.6520],
        [  3.3094,  13.4969,  -3.6507],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.0506,   4.9277,  -9.5094],
        [ -4.2040,   7.2486,   2.9875],
        [ -0.4341, -10.9473,   7.4509],
        [ -9.4735,  -1.2620,  -3.6742],
        [  0.5850,  -4.7982,   9.6639],
        [  1.0000,   7.6591,  -4.5011],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.2687,  -2.9794,   0.6436],
        [  8.3791,  10.5409,   5.5590],
        [ -0.5709,   7.9055,   7.6089],
        [  0.9036,   7.6840,   9.9237],
        [  5.6248,   3.9984,   8.4933],
        [  8.1528,  10.7452,   9.1493],
        [ -8.9178,   2.1936,   7.5388],
        [ -9.4430,  -0.1897,   4.8383],
        [  4.5340,  -3.3629,   8.9652],
        [  2.7420,  -4.4896,  -3.7139],
        [  2.3367,   7.1676,  -9.5263],
        [  4.2612, -10.5514,  -6.6268],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3200,   0.2690,   1.5900],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.6523,   3.9443,  -2.9449],
        [  3.6469,   6.3624,  -1.8414],
        [  2.7590,  13.6621,   9.4496],
        [ -8.8058,   2.4850,  -1.7425],
        [ -1.8823, -10.6558,  -3.8952],
        [  1.9241,   7.3503,   7.9004],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.1984,  -0.4327,   1.5438],
        [ 11.5568,   6.0602,   9.3244],
        [  2.3376,  -4.6231,   9.8436],
        [  7.3724,  -9.0577,  -8.9843],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -5.0550,   6.8231,  -9.2502],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.1940,  -2.1881,   7.1499],
        [ -7.7239,   4.4298,   3.1216],
        [  6.3542,   0.6934,  -3.3840],
        [ -9.2687,  -2.9794,   0.6436]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0589, 0.0710,  ..., 0.5384, 0.7522, 0.8781],
        [0.0589, 0.0000, 0.0165,  ..., 0.4993, 0.7844, 0.8395],
        [0.0710, 0.0165, 0.0000,  ..., 0.5020, 0.8008, 0.8420],
        ...,
        [0.5384, 0.4993, 0.5020,  ..., 0.0000, 0.8059, 0.3402],
        [0.7522, 0.7844, 0.8008,  ..., 0.8059, 0.0000, 0.9840],
        [0.8781, 0.8395, 0.8420,  ..., 0.3402, 0.9840, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.57685908,  7.89745107,  7.92607342,  7.92200342,  8.04718433,
        8.06517143,  5.24137909,  5.53255882, 13.86480175,  5.36061974,
       13.09737053,  9.95552626, 11.48434767,  8.56889738,  9.34061938,
        6.74872981, 10.00410625,  5.19065439, 10.65929968, 11.22993218,
       13.06957425, 13.70893492,  9.48337063, 13.26590071,  5.91408504,
       13.36092624,  6.00732193,  6.06416957, 12.49317563, 11.48734978,
        6.46560181,  9.48337063,  7.24326992,  7.62011734, 13.68063098,
        8.92098752,  6.02588517,  6.39438044, 12.13568598,  4.91296205,
        6.02789634, 10.30972193,  5.95111804,  9.91668182,  5.91408504,
        5.94293185,  5.91677188,  9.94285511,  9.95552626,  8.47371868,
        7.58378655,  6.39520397, 11.23201619,  5.95111804, 12.17901033,
        8.11227668, 13.86480175,  6.74686071,  8.77913428, 13.3426541 ,
        9.09502385,  6.2322769 ,  6.07992623, 13.89671314, 13.03002338,
        5.16077764,  7.05627461,  8.37954853, 10.95593729,  9.55720651,
        4.83371341,  7.72415048,  9.43112957,  9.73579738, 13.46553623,
        7.92607342,  7.73692436,  6.90115901, 13.48809595,  9.18358562,
        9.44486609,  5.64500512,  5.26067019,  7.53884104, 11.37939695,
        6.30647746,  6.32572323, 13.14777433,  6.89245352,  7.33350642,
       13.93790235,  9.14973319, 10.8207297 ,  7.59795445,  9.87720102,
        6.2134961 , 13.0493368 ,  5.1805345 , 11.67875855,  9.47369628,
        8.49161154,  8.83725081, 12.38881635,  8.90405823,  6.39187285,
        9.73579738])}
{'P': tensor([[1.2045e-02, 1.2981e-01, 1.2091e-01,  ..., 1.4179e-05, 1.2091e-01,
         2.8013e-06],
        [7.7224e-04, 2.3736e-02, 1.0417e-02,  ..., 1.3580e-05, 1.0417e-02,
         8.1656e-06],
        [1.7894e-03, 4.3769e-02, 4.7389e-02,  ..., 3.3422e-05, 4.7389e-02,
         1.2195e-05],
        ...,
        [1.0063e-03, 8.2171e-04, 1.2627e-03,  ..., 1.1953e-02, 1.2627e-03,
         2.3926e-08],
        [1.7894e-03, 4.3769e-02, 4.7389e-02,  ..., 3.3422e-05, 4.7389e-02,
         1.2195e-05],
        [2.5578e-06, 2.3355e-06, 1.9082e-06,  ..., 1.6617e-07, 1.9082e-06,
         2.8711e-02]]), 'X': tensor([[  2.0547,   7.2957,  -1.9343],
        [  0.9809,   7.6642,  -1.4217],
        [  1.1006,   7.6318,  -2.4877],
        [  1.1139,   7.6281,  -2.5680],
        [ -0.0282,   7.8575,  -1.1746],
        [ -0.1963,   7.8765,  -1.1371],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.1204,  -5.9308,   4.0761],
        [ -3.1204, -10.2345,   9.2288],
        [  4.5503,   5.5359,   7.4735],
        [  8.0877,  -8.5297,  -1.1050],
        [ -8.3062,  -5.6008,   0.0287],
        [ -0.4341, -10.9473,   7.4509],
        [ 11.7794,   5.5086,  -9.6929],
        [ -5.2589,   6.7024,  -5.5809],
        [ -1.6086,   7.8928,   4.3385],
        [  3.9770, -10.6341,   8.0783],
        [ 10.7566,   7.6212,  -2.3279],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.2707,   0.9661,   4.4863],
        [  1.1350,  14.0104,   0.6321],
        [  4.5814, -10.4485,   2.5496],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.2939,  -2.4784,  -4.5840],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.4242, -10.1261,   3.9595],
        [ -2.2188,   7.8218,   8.6064],
        [  1.8447,  -4.7372,   8.1235],
        [ -5.6423,   6.4537,  -1.2091],
        [ 10.4994,  -5.9193,  -7.5153],
        [  5.6093,  -1.9705,   6.4807],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.6220,   7.8917,   0.0315],
        [  2.3200,   7.1756,   1.1938],
        [  1.1350,  14.0104,   0.6321],
        [  4.9152,  12.8653,  -1.4459],
        [  6.3534,   1.2990,  -7.2942],
        [  5.5626,  -2.0536,   8.4433],
        [  5.7485,  -9.9811,  -3.2355],
        [  1.3705,  13.9723,   7.9266],
        [  2.7590,  13.6621,   9.4496],
        [ -3.4488, -10.0934,   2.5195],
        [ -4.5778,  -9.5013,  -5.0667],
        [  7.9403,  -8.6455,  -8.7433],
        [  9.0424,  -7.6779,  -7.5286],
        [ -9.0604,   1.7707,   1.1066],
        [  8.8091,  -7.9037,   0.3631],
        [  1.8766,   7.3694,   1.1887],
        [  1.7232,  -4.7574,   3.1270],
        [  0.5250, -11.0306,   5.5358],
        [ -5.6423,   6.4537,  -1.2091],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.3616,  -2.4750,   3.7765],
        [  4.0078,  -3.8018,  -6.2168],
        [  4.0591,   6.0202,  -1.6339],
        [  1.1006,   7.6318,  -2.4877],
        [  5.3602,  -2.3808,   7.2098],
        [  2.4932,   7.0903,  -5.4214],
        [ -3.5520,   7.4976,   9.1522],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.9942,  -3.9990,   6.0448],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -8.4297,  -5.3637,  -9.3962],
        [  4.6063,  13.0051,  -3.1456],
        [  6.3303,   1.6145,  -8.4083],
        [  6.2804,   2.0019,   0.8607],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.5149,  -5.1905,   1.5239],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.6142,  14.0802,   0.5407],
        [  1.7903,  -4.7467,   0.1168],
        [ -7.3470,  -7.0615,   5.0066],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.3119,   1.7797,   9.9186],
        [  1.3767, -11.0341,   9.9415],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.1752,   2.5242,   3.5874],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -8.5506,  -5.1154,   6.3423],
        [  4.4915,   5.5990,   7.0405],
        [  4.4633,   5.6287,  -0.7754],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.9193,  -1.3176,   5.1739],
        [  6.1579,  -0.5961,  -6.5595],
        [ -3.9508,  -9.8515,   1.7923],
        [  8.5198,  10.4091,  -8.6486],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.1120,  -4.6818,   8.6144],
        [ -6.4368,   5.8356,   3.3014],
        [  2.0243,   7.3086,   7.5432],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.3604,   0.9042,  -6.5901],
        [  9.5018,  -7.1945,  -1.1892],
        [ 10.4994,  -5.9193,  -7.5153],
        [  6.3119,   1.7797,   9.9186],
        [  5.0506,   4.9277,  -9.5094],
        [  2.4572,   7.1085,   7.7376],
        [  5.9026,   3.3837,  -0.6573],
        [  4.1297,  -3.7098,  -3.9838],
        [  7.8425,  -8.7202,   6.4530],
        [  6.1222,  -0.7261,   8.8204],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.0619,   8.6627,  -9.2884],
        [  1.1006,   7.6318,  -2.4877],
        [ -5.7347,  -8.6908,   9.2878]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0532, 0.0621,  ..., 0.6670, 0.0621, 1.0459],
        [0.0532, 0.0000, 0.0611,  ..., 0.7187, 0.0611, 0.9941],
        [0.0621, 0.0611, 0.0000,  ..., 0.6828, 0.0000, 1.0328],
        ...,
        [0.6670, 0.7187, 0.6828,  ..., 0.0000, 0.6828, 1.7128],
        [0.0621, 0.0611, 0.0000,  ..., 0.6828, 0.0000, 1.0328],
        [1.0459, 0.9941, 1.0328,  ..., 1.7128, 1.0328, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.57946459,  7.72668467,  7.71075226,  7.70897675,  7.8575648 ,
        7.8788965 ,  4.9160962 , 10.05558893, 10.69963954,  7.16598273,
       11.75438006, 10.01804223, 10.95593729, 13.00380624,  8.51928531,
        8.0550286 , 11.35345345, 13.18278606,  5.24461068,  9.32094055,
       14.05633441, 11.4088043 , 12.17451685,  5.84533345,  8.22728399,
       11.48734978,  8.1303945 ,  5.08372149,  8.5724042 , 12.05300943,
        5.94535529,  6.61211224,  8.05668609,  7.54127425, 14.05633441,
       13.77222752,  6.48486171,  5.92952538, 11.51811603, 14.0393949 ,
       13.93790235, 10.66631784, 10.54658597, 11.73848065, 11.86239961,
        9.23178204, 11.83509372,  7.60463349,  5.05989547, 11.04313641,
        8.5724042 ,  9.19837792,  9.68324553,  5.52416368,  7.26074925,
        7.71075226,  5.86519525,  7.5158489 ,  8.29640613,  9.85330646,
        9.84314712, 10.52587243,  9.99145096, 13.79676654,  6.53290663,
        6.59176018, 10.5575601 ,  9.97220799, 10.7167285 , 14.09357547,
        5.07306274, 10.19036081, 11.72749923,  6.55800267, 11.11969815,
        8.20436764,  6.67121748,  9.91668182,  9.96391045,  7.17789666,
        7.18354885,  9.79039371,  6.10774544,  6.06416957,  6.18668463,
       10.61417181, 13.45124917, 11.67875855,  5.13616224,  8.68832398,
        7.58378655, 12.22503754,  6.42439344, 11.9182956 , 12.05300943,
        6.55800267,  7.05627461,  7.52117132,  6.80370254,  5.5513072 ,
       11.72802491,  6.16512856,  7.71075226, 13.27717982,  7.71075226,
       10.412299  ])}
{'P': tensor([[1.5850e-02, 1.3172e-01, 1.2507e-01,  ..., 1.0735e-03, 7.9525e-06,
         7.1026e-05],
        [2.1764e-04, 1.4385e-02, 5.5396e-02,  ..., 3.0325e-03, 1.2939e-05,
         5.1207e-05],
        [1.7906e-04, 2.2935e-03, 1.3287e-02,  ..., 3.2282e-03, 1.2765e-05,
         3.7437e-05],
        ...,
        [4.9559e-05, 6.8575e-05, 7.1868e-05,  ..., 9.1370e-02, 6.2960e-03,
         9.5822e-06],
        [9.4532e-06, 5.1352e-06, 4.9097e-06,  ..., 3.5641e-06, 2.2291e-02,
         6.8936e-07],
        [7.6220e-05, 1.7089e-04, 1.5183e-04,  ..., 1.7320e-04, 7.9281e-06,
         6.7107e-02]]), 'X': tensor([[  2.0243,   7.3086,   7.5432],
        [ -0.3432,   7.8900,   7.1805],
        [ -0.5709,   7.9055,   7.6089],
        [ -0.5385,   7.9037,   6.3830],
        [ -1.5451,   7.8975,   7.6104],
        [ -1.6906,   7.8860,   7.7142],
        [  2.5813,  -4.5471,   5.3597],
        [  4.9957,  -2.8705,   6.5978],
        [  5.4426, -10.1181,  -5.1268],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 11.5083,  -4.1177,   0.2389],
        [ -0.5709,   7.9055,   7.6089],
        [ -6.1914,  -8.2997,   8.5380],
        [  5.8617,   3.4853,  -9.8080],
        [  5.9587,  12.3216,   9.1695],
        [  5.4732,  -2.2041,  -4.3830],
        [  0.3715,  -4.7756,   0.2312],
        [  6.0467,  -9.8369,  -6.5816],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.7078,   2.7182,   1.8146],
        [  2.1379,  -4.6757,   6.4981],
        [  7.1441,  11.5529,   2.3232],
        [ -1.8367,   7.8718,  -8.6523],
        [ 10.5678,   7.9261,  -0.6100],
        [ -8.5859,  -5.0393,   0.5476],
        [ 12.5154,   2.5072,  -9.3274],
        [  1.3103,  -4.8032,  -0.9300],
        [  0.6396,  14.0772,  -6.3786],
        [ -6.3489,   5.9117,   6.5161],
        [ -7.8126,   4.3066,   5.1791],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.7565,  11.8237,  -4.9882],
        [  9.5203,   9.3483,   5.8781],
        [  3.2647, -10.8066,  -0.3768],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.0608,   3.9362,  -5.9567],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.2508,   1.0572,   2.9978],
        [  9.1620,  -7.5573,  -9.9882],
        [ -1.3940, -10.7772,   6.7835],
        [  6.0140,  -1.0661,  -2.5040],
        [ -8.8934,   2.2597,   3.5831],
        [  5.6248,   3.9984,   8.4933],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.2895,   0.0405,   5.7652],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.7078,   2.7182,   1.8146],
        [ -1.6906,   7.8860,   7.7142],
        [  6.9314,  -9.3435,  -1.2729],
        [  2.1625,  -4.6697,   7.0658],
        [  2.0243,   7.3086,   7.5432],
        [  1.1644, -11.0394,   7.8511],
        [  3.9591, -10.6391,  -6.2557],
        [  3.2692,  13.5098,  -4.9032],
        [  2.0729,   7.2878,   7.0880],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.2895,   0.0405,   5.7652],
        [ 10.5955,   7.8824,  -3.7006],
        [ -3.7964,  -9.9294,   7.7147],
        [ 10.0567,   8.6697,   4.6329],
        [  3.7237,  -3.9966,  -5.0376],
        [  3.3094,  13.4969,  -3.6507],
        [ -1.9600,   7.8577,   6.9121],
        [ -3.2736, -10.1703,   3.8712],
        [ -9.3444,   0.5821,   6.9101],
        [ -8.0829,  -5.9940,   5.0564],
        [  5.7218,  -9.9934,   6.5764],
        [ -9.4735,  -1.2620,  -3.6742],
        [  5.6248,   3.9984,   8.4933],
        [ -0.3432,   7.8900,   7.1805],
        [  6.2555,   2.1474,   4.1685],
        [  1.3670,  -4.7990,   3.2603],
        [  2.1625,  -4.6697,   7.0658],
        [ 12.3025,  -1.7440,  -6.6632],
        [  4.0362,  13.2397,  -7.0030],
        [ -8.0611,  -6.0302,   4.6962],
        [ -0.0254, -10.9932,   3.2200],
        [  0.1460,   7.8340,  -9.4614],
        [  0.9659,  -4.8144,  -0.9594],
        [  8.3791,  10.5409,   5.5590],
        [ -8.6801,   2.7809,  -0.8443],
        [ -4.9217,   6.8979,  -1.7667],
        [ -6.3252,  -8.1760,  -1.0891],
        [  4.2612, -10.5514,  -6.6268],
        [  5.5101,  -2.1433,   9.5442],
        [ -8.9788,   2.0208,  -3.2867],
        [  4.2096,   5.8815,   1.9230],
        [ -3.4986,   7.5152,   9.0712],
        [ -9.1960,  -3.2964,   1.6921],
        [  2.3261,  13.7749,   5.9121],
        [  4.9152,  12.8653,  -1.4459],
        [  2.6970,   6.9826,   9.7350],
        [  5.6202,  -1.9508,   0.7471],
        [  1.8719,   7.3713,  -3.7465],
        [  4.5057,  -3.3894,  -0.8160],
        [ -6.6998,   5.5954,   0.0823],
        [  6.1984,  -0.4327,   1.5438],
        [  4.1537,   5.9340,  -6.0423],
        [  3.4401,  13.4540,   4.6576],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.9282,   2.1648,   8.4894],
        [ -3.3116, -10.1539,   7.5919],
        [ -1.0407,   7.9166,  -6.6021]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0528, 0.0617,  ..., 0.5855, 0.6432, 0.9228],
        [0.0528, 0.0000, 0.0165,  ..., 0.5330, 0.5907, 0.9181],
        [0.0617, 0.0165, 0.0000,  ..., 0.5247, 0.5821, 0.9326],
        ...,
        [0.5855, 0.5330, 0.5247,  ..., 0.0000, 0.0593, 1.0785],
        [0.6432, 0.5907, 0.5821,  ..., 0.0593, 0.0000, 1.1216],
        [0.9228, 0.9181, 0.9326,  ..., 1.0785, 1.1216, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.58378655,  7.89745107,  7.92607342,  7.92200342,  8.04718433,
        8.06517143,  5.2287058 ,  5.76165934, 11.4890861 ,  6.78325151,
        9.58072286, 12.22276278,  7.92607342, 10.35464916,  6.81961996,
       13.68672855,  5.90035324,  4.79003361, 11.54672755,  9.51074763,
       13.08414689,  9.12220271,  5.14124612, 13.58333857,  8.08320235,
       13.20988996,  9.95552626, 12.76408562,  4.978704  , 14.09176259,
        8.67502348,  8.92098752,  5.53255882, 13.61801526, 13.3426541 ,
       11.28895489,  8.22728399,  8.97052115, 10.27572673,  9.3109884 ,
       11.87665501, 10.86693708,  6.10774544,  9.17595899,  6.90115901,
        9.50913849,  6.28961988,  8.8737556 ,  9.12220271,  8.06517143,
       11.63383221,  5.14607014,  7.58378655, 11.10066126, 11.35182589,
       13.89974693,  7.57685908,  9.85384623,  9.97964013,  6.28961988,
       13.20597932, 10.63038565, 13.2778378 ,  5.46244988, 13.89671314,
        8.09842704, 10.68416401,  9.36256096, 10.06286074, 11.51557166,
        9.55720651,  6.90115901,  7.89745107,  6.61386436,  4.98989906,
        5.14607014, 12.42554808, 13.84126197, 10.06703358, 10.99326577,
        7.83534987,  4.91038884, 13.46553623,  9.11473391,  8.47371868,
       10.33712258, 11.37939695,  5.91222346,  9.20340031,  7.23276582,
        8.28968067,  9.76896904, 13.96987806, 13.77222752,  7.48538905,
        5.94909709,  7.60529892,  5.63824971,  8.72896381,  6.2134961 ,
        7.24326992, 13.88683436, 11.95743558,  9.18690018, 10.68031351,
        7.98469166])}
{'P': tensor([[1.6223e-02, 2.3870e-01, 1.5086e-01,  ..., 4.4436e-05, 1.0115e-03,
         8.3997e-04],
        [7.1757e-04, 2.6482e-02, 3.9791e-01,  ..., 5.4883e-05, 2.7591e-03,
         2.1074e-03],
        [5.9919e-04, 9.2485e-04, 3.3809e-02,  ..., 5.6068e-05, 5.8691e-03,
         3.9184e-03],
        ...,
        [7.3647e-03, 5.4803e-03, 4.1110e-03,  ..., 4.3696e-02, 1.2771e-04,
         4.5249e-05],
        [2.5679e-05, 2.9658e-05, 3.5314e-05,  ..., 3.9419e-05, 4.8459e-02,
         2.5620e-03],
        [5.6702e-05, 6.4454e-05, 7.5141e-05,  ..., 3.8887e-05, 6.1349e-03,
         1.1476e-01]]), 'X': tensor([[  2.0039,   7.3173,  -1.5885],
        [  0.9809,   7.6642,  -1.4217],
        [ -0.0282,   7.8575,  -1.1746],
        [ -0.1963,   7.8765,  -1.1371],
        [  1.3339,   7.5627,  -0.9034],
        [  1.1006,   7.6318,  -2.4877],
        [  6.3360,   0.4258,  -9.2509],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.2905,   1.9366,   2.1169],
        [  3.2226,  -4.2802,  -4.0428],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.9337,  12.8566,  -3.6582],
        [  6.2905,   1.9366,   2.1169],
        [  2.3589,  13.7668,  -3.6501],
        [  1.6422,   7.4586,  -6.8056],
        [  4.5503,   5.5359,   7.4735],
        [  7.0358,  -9.2784,  -9.8479],
        [  2.4572,   7.1085,   7.7376],
        [ -2.3317,   7.8034,  -3.6274],
        [ 12.3025,  -1.7440,  -6.6632],
        [  7.7750,  11.0665,   2.8364],
        [  5.5240,  -2.1198,   6.2055],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.1886,  -2.6250,   6.4677],
        [ -0.2586,   7.8825,  -9.9840],
        [  2.2849, -10.9655,  -7.6542],
        [  0.4996,   7.7738,  -4.4711],
        [ -3.1204, -10.2345,   9.2288],
        [  1.0000,   7.6591,  -4.5011],
        [ -3.5173, -10.0623,   4.8280],
        [  2.3376,  -4.6231,   9.8436],
        [ -7.5401,   4.6718,   9.0033],
        [  6.1752,   2.5242,   3.5874],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.3303,   1.6145,  -8.4083],
        [  0.9445,   7.6736,   0.5095],
        [  6.2905,   1.9366,   2.1169],
        [ 11.3118,   6.5957,  -4.5667],
        [ -2.5161,   7.7699,  -9.3531],
        [ -8.6426,  -4.9136,   9.4941],
        [  9.3624,   9.5318,   0.2779],
        [  9.9756,  -6.6330,  -3.3429],
        [ 12.2589,  -1.9312,   9.8657],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -7.4887,   4.7364,   8.9074],
        [  9.5261,  -7.1675,   1.4206],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.3724,   7.7974,  -6.8722],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.5789,   0.1722,   1.6470],
        [ -7.8280,   4.2848,  -7.1955],
        [ -3.4488, -10.0934,   2.5195],
        [  4.9152,  12.8653,  -1.4459],
        [  5.2608,  -2.5256,  -9.1446],
        [ -3.4488, -10.0934,   2.5195],
        [ 11.2209,  -4.7092,   2.6117],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.9445,   7.6736,   0.5095],
        [ -9.4203,  -2.0340,   2.5856],
        [  4.5057,  -3.3894,  -0.8160],
        [ -9.1459,   1.4780,  -4.2403],
        [ -5.6651,   6.4380,   1.3660],
        [ 11.0275,  -5.0666,   8.4154],
        [ 11.5966,   5.9666,   6.0083],
        [  8.3791,  10.5409,   5.5590],
        [  2.8890,  -4.4316,   1.1600],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.6470, -11.0352,  -1.1931],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.1006,   7.6318,  -2.4877],
        [ 12.4837,   2.7510,   0.4449],
        [  6.3301,   0.3638,  -7.5685],
        [  9.9058,  -6.7203,   5.6928],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.1787,  -2.6383,  -4.2533],
        [  8.3791,  10.5409,   5.5590],
        [ -9.4351,  -1.8891,   3.5642],
        [ 11.0373,   7.1299,  -8.5895],
        [ 12.1504,   4.3674,  -7.2998],
        [  2.3373,  -4.6232,  -5.2246],
        [  0.5960, -11.0335,   3.1144],
        [  5.0277,   4.9589,   7.0175],
        [  6.3606,   0.9204,   3.2525],
        [  4.1140,  -3.7220,  -0.1273],
        [ 11.0275,  -5.0666,   8.4154],
        [  0.6952,  -4.8060,   8.8976],
        [ -2.5161,   7.7699,  -9.3531],
        [ -9.3616,  -2.4749,   0.1793],
        [  0.2571,  -4.7594,   1.5618],
        [ -9.3810,  -2.3455,  -9.5194],
        [  2.3376,  -4.6231,   9.8436],
        [  2.8890,  -4.4316,   1.1600],
        [  6.2790,  -0.0259,  -5.4116],
        [ -2.5056,   7.7720,   5.5878],
        [  7.5064,  -8.9651,  -0.0994],
        [  7.5064,  -8.9651,  -0.0994],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.6308,  13.9254,  -2.0984],
        [  6.3606,   0.9204,   3.2525],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.8363,   2.4086,  -5.6296]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.3971e-02, 7.7500e-02,  ..., 3.9765e-01, 6.6083e-01,
         7.2789e-01],
        [4.3970e-02, 0.0000e+00, 3.3707e-02,  ..., 4.0927e-01, 6.1709e-01,
         6.9243e-01],
        [7.7500e-02, 3.3707e-02, 0.0000e+00,  ..., 4.1673e-01, 5.8340e-01,
         6.6908e-01],
        ...,
        [3.9765e-01, 4.0927e-01, 4.1673e-01,  ..., 0.0000e+00, 8.0924e-01,
         1.0559e+00],
        [6.6083e-01, 6.1709e-01, 5.8340e-01,  ..., 8.0924e-01, 0.0000e+00,
         4.8984e-01],
        [7.2789e-01, 6.9243e-01, 6.6908e-01,  ..., 1.0559e+00, 4.8984e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.58668144,  7.72668467,  7.8575648 ,  7.8788965 ,  7.67939546,
        7.71075226,  6.35029274,  5.1224079 ,  6.58184474,  5.35774573,
       12.9089559 , 13.77075174,  6.58184474, 13.96746816,  7.63726134,
        7.16598273, 11.64436684,  7.52117132,  8.14433542, 12.42554808,
       13.5247215 ,  5.91677188, 11.85743425, 12.41012046,  5.81481044,
        7.88677538, 11.20100795,  7.78980299, 10.69963954,  7.72415048,
       10.65929968,  5.1805345 ,  8.87007495,  6.67121748, 12.35421461,
       12.35421461,  6.53290663,  7.73151309,  6.58184474, 13.09426142,
        8.16714836,  9.94173708, 13.3607379 , 11.97957425, 12.41012046,
        5.04028538,  5.68942572,  9.60359158,  8.86082124, 11.92133129,
        8.83341608,  7.80625981,  4.92497514, 12.58006038,  8.92395381,
       10.66631784, 13.77222752,  5.83560009, 10.66631784, 12.16901615,
       10.78533838,  7.73151309,  9.63742669,  5.63824971,  9.26455795,
        8.57560275, 12.13568598, 13.04156075, 13.46553623,  5.29009836,
        5.5366698 , 11.05413545,  4.9160962 ,  6.07992623,  7.71075226,
       12.78327037,  6.34059021, 11.9702762 ,  5.18277858,  5.81199422,
       13.46553623,  9.62238828, 13.1399181 , 12.91143491,  5.18047638,
       11.04953767,  7.06169391,  6.42689705,  5.54776934, 12.13568598,
        4.85603886,  8.16714836,  9.68323686,  4.7663526 ,  9.66977802,
        5.1805345 ,  5.29009836,  6.27905292,  8.16584537, 11.69264243,
       11.69264243,  5.84450127, 14.02058621,  6.42689705,  9.15976746,
        9.15866607])}
{'P': tensor([[2.3716e-02, 1.9199e-01, 1.8004e-01,  ..., 1.9452e-05, 7.5082e-05,
         3.4408e-05],
        [3.4877e-04, 2.4593e-02, 4.2450e-03,  ..., 4.4061e-05, 4.3406e-05,
         9.0072e-05],
        [4.6239e-04, 1.0525e-01, 2.7331e-02,  ..., 5.4461e-05, 5.0687e-05,
         1.0141e-04],
        ...,
        [1.5215e-06, 2.0333e-06, 2.2025e-06,  ..., 3.1761e-02, 4.3424e-06,
         9.0604e-03],
        [2.3939e-02, 8.1995e-03, 9.1418e-03,  ..., 1.0075e-05, 6.5856e-02,
         2.1388e-05],
        [1.7612e-05, 1.5821e-05, 1.6864e-05,  ..., 3.1719e-04, 1.6775e-04,
         1.2271e-01]]), 'X': tensor([[  1.9241,   7.3503,   7.9004],
        [ -0.5709,   7.9055,   7.6089],
        [ -0.3432,   7.8900,   7.1805],
        [ -1.5451,   7.8975,   7.6104],
        [ -1.6906,   7.8860,   7.7142],
        [ -0.0410,   7.8591,   9.3457],
        [  0.7685,  -4.8098,  -2.9310],
        [ -9.0574,   1.7801,   8.8986],
        [  4.3523, -10.5232,   8.2138],
        [ -6.1914,  -8.2997,   8.5380],
        [ 11.2517,  -4.6495,   2.9800],
        [  0.1460,   7.8340,  -9.4614],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.0560,  -0.9421,  -5.0334],
        [  0.9809,  -4.8145,   6.8706],
        [  5.7485,  -9.9811,  -3.2355],
        [ 11.1085,  -4.9204,   0.0736],
        [  0.1047, -11.0046,   0.5188],
        [  6.3604,   0.9042,  -6.5901],
        [  6.2804,   2.0019,   0.8607],
        [  3.1271, -10.8343,   2.6442],
        [  4.6992,   5.3690,   3.0935],
        [  3.7992,  -3.9474,  -8.9019],
        [ -8.5900,   2.9767,   7.4811],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.3515,   0.6418,  -6.5823],
        [ 12.0845,  -2.5772,  -4.6073],
        [  0.8700,  -4.8132,  -7.8042],
        [  9.6662,   9.1725,   8.1056],
        [  3.9085,   6.1512,  -2.2560],
        [ -7.0825,   5.2073,  -5.0497],
        [  4.1348,   5.9514,  -0.8856],
        [ 10.4994,  -5.9193,  -7.5153],
        [  6.0073,   3.0986,   6.7937],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.8721,  -4.7322,   0.5524],
        [  5.1793,   4.7461,   0.3899],
        [  2.0937,  13.8293,  -2.3533],
        [ 12.5919,   0.4035,  -5.8460],
        [  8.3791,  10.5409,   5.5590],
        [ -9.4203,  -2.0340,   2.5856],
        [  1.7218,  -4.7576,  -3.3617],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.0140,  -1.0661,  -2.5040],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.6154,  -4.0640,  -7.9698],
        [ 11.8951,  -3.1570,   7.0742],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -8.3535,  -5.5118,   9.8399],
        [  3.4920,  -4.1366,  -8.3955],
        [  6.1575,  12.2046,   3.3257],
        [ -9.3616,  -2.4750,   3.7765],
        [ -6.7992,   5.4991,   9.9186],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.5900,   2.9767,   7.4811],
        [ -2.1533, -10.5778,   1.0112],
        [  0.7685,  -4.8098,  -2.9310],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.3589,  13.7668,  -3.6501],
        [  2.8890,  -4.4316,   1.1600],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -0.5709,   7.9055,   7.6089],
        [  1.4489,   7.5256,   4.4529],
        [ 10.1503,   8.5414,  -7.2681],
        [  9.0424,  -7.6779,  -7.5286],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 12.3453,  -1.5462,   7.0212],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.5620,   7.0549,   8.4825],
        [  6.1737,   2.5305,   3.9049],
        [  5.8450,  -9.9356,   3.0767],
        [  4.4290,  -3.4596,   1.4181],
        [  1.8766,   7.3694,   1.1887],
        [  3.5284, -10.7484,   5.3347],
        [  9.5261,  -7.1675,   1.4206],
        [  2.2644,  -4.6434,   9.7508],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.2555,   2.1474,   4.1685],
        [  5.8325,  -1.5222,  -2.1341],
        [ 12.4690,  -0.8528,  -9.3374],
        [  1.7301,   7.4262,   5.0522],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -3.0644,   7.6439,  -1.4322],
        [ -8.0766,  -6.0046,  -3.1873],
        [  4.3377,   5.7570,   1.3712],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.3266,   0.3292,   9.3502],
        [  4.7482, -10.3907,   5.9149],
        [ -9.1231,  -3.5739,   8.4708],
        [  5.0277,   4.9589,   7.0175],
        [  3.6249, -10.7255,   5.6605],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.3535,   0.6792,  -7.5101],
        [  5.9843,  -9.8679,  -6.1449],
        [  4.1348,   5.9514,  -0.8856],
        [ -7.4807,   4.7463,   2.0616],
        [ -8.6801,   2.7809,  -0.8443],
        [  1.6954,  -4.7616,   9.8198],
        [  2.5620,   7.0549,   8.4825],
        [  1.1274,  14.0116,   8.1753],
        [ -9.4697,  -0.6157,  -6.8472],
        [  6.3549,   0.7096,   6.9198],
        [ -8.3062,  -5.6008,   0.0287]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0539, 0.0482,  ..., 1.2478, 0.2236, 0.9321],
        [0.0539, 0.0000, 0.0165,  ..., 1.2185, 0.2768, 0.8907],
        [0.0482, 0.0165, 0.0000,  ..., 1.2106, 0.2670, 0.8874],
        ...,
        [1.2478, 1.2185, 1.2106,  ..., 0.0000, 1.3509, 0.4527],
        [0.2236, 0.2768, 0.2670,  ..., 1.3509, 0.0000, 1.0902],
        [0.9321, 0.8907, 0.8874,  ..., 0.4527, 1.0902, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.59795445,  7.92607342,  7.89745107,  8.04718433,  8.06517143,
        7.85919687,  4.87083348,  9.23071698, 11.38774149, 10.35464916,
       12.17451685,  7.83534987, 12.4217523 ,  9.43112957,  6.12886024,
        4.91336994, 11.51811603, 12.1494023 , 11.00508681,  6.42439344,
        6.59176018, 11.27656488,  7.13501264,  5.47866221,  9.09119351,
        8.89458273,  6.383897  , 12.35625694,  4.89121805, 13.32557133,
        7.28793663,  8.7907923 ,  7.24677664, 12.05300943,  6.7593917 ,
        8.06990483,  5.08910491,  7.02496223, 13.98691426, 12.59840193,
       13.46553623,  9.63742669,  5.05963217,  9.88571713,  6.10774544,
       13.09737053,  5.43943791, 12.30695098,  9.43737523, 10.00802627,
        5.41348121, 13.66991027,  9.68324553,  8.74470218, 11.83075005,
        9.09119351, 10.794755  ,  4.87083348, 12.07818871, 13.96746816,
        5.29009836,  6.83515201,  9.7712321 ,  7.92607342,  7.66377554,
       13.26590071, 11.86239961,  9.36378438,  9.70108409, 12.44177641,
        5.84450127,  7.50564223,  6.67218687, 11.52733581,  5.62005629,
        7.60463349, 11.31276109, 11.92133129,  5.16611782, 12.13568598,
        6.61386436,  6.02789634, 12.4980845 ,  7.62509582, 10.39942497,
        8.23525433, 10.06407839,  7.20826219, 10.06703358,  6.33517349,
       11.42420508,  9.79814997,  7.06169391, 11.32149259,  5.5513072 ,
        6.38968357, 11.54071859,  7.24677664,  8.85939626,  9.11473391,
        5.05444288,  7.50564223, 14.05687815,  9.48969978,  6.39438044,
       10.01804223])}
{'P': tensor([[2.8484e-02, 2.1286e-01, 1.8848e-01,  ..., 1.6398e-05, 1.6974e-03,
         1.2781e-05],
        [7.8138e-04, 3.7295e-02, 4.5547e-02,  ..., 3.4627e-05, 3.6878e-03,
         2.3861e-05],
        [4.9217e-04, 5.5890e-03, 3.1470e-02,  ..., 3.5174e-05, 6.7010e-03,
         3.5248e-05],
        ...,
        [3.3647e-05, 2.0252e-05, 1.8753e-05,  ..., 8.7922e-02, 1.1342e-05,
         4.1027e-04],
        [2.1944e-05, 2.9339e-05, 3.8801e-05,  ..., 1.2086e-03, 3.9552e-02,
         4.6630e-03],
        [5.0581e-06, 3.2476e-06, 4.1684e-06,  ..., 3.9992e-03, 1.1195e-05,
         4.2658e-02]]), 'X': tensor([[ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.3746e-02, 5.0318e-02,  ..., 6.5542e-01, 5.9871e-01,
         9.5902e-01],
        [7.3746e-02, 0.0000e+00, 4.0260e-02,  ..., 5.8392e-01, 5.5232e-01,
         9.2297e-01],
        [5.0318e-02, 4.0260e-02, 0.0000e+00,  ..., 6.0770e-01, 5.5080e-01,
         9.1473e-01],
        ...,
        [6.5542e-01, 5.8392e-01, 6.0770e-01,  ..., 0.0000e+00, 3.7928e-01,
         6.8313e-01],
        [5.9871e-01, 5.5232e-01, 5.5080e-01,  ..., 3.7928e-01, 0.0000e+00,
         3.8640e-01],
        [9.5902e-01, 9.2297e-01, 9.1473e-01,  ..., 6.8313e-01, 3.8640e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.5984809 ,  7.85919687,  7.92607342,  7.89745107,  8.04718433,
        8.04483495,  5.26342203, 10.57302546,  5.79562691, 10.412299  ,
       10.65929968,  6.10360073, 13.38091121,  6.49538841,  5.14371005,
       10.00410625, 13.58333857, 12.49317563, 13.25336001, 10.4606265 ,
        9.25336371, 13.77604293,  8.36445758, 13.36092624, 10.61472636,
       10.75443106,  9.88571713,  9.15976746, 10.09979747,  6.58184474,
        9.62958973,  5.68942572,  9.15866607,  6.31576264, 11.54071859,
        5.05963217, 12.11735334, 11.73567759,  7.20482224, 13.12299145,
       13.37121923,  9.54154096, 11.69264243, 10.65929968, 13.27717982,
        9.43737523,  7.39004226,  6.67218687,  7.78980299, 10.51437947,
        8.04664886,  6.53250937,  6.99097013, 13.26590071,  5.01114719,
       12.85032836, 11.73397309, 12.16901615,  8.66273171,  8.43626114,
       13.99048094,  8.86082124,  7.62509582,  9.50654967,  9.14044984,
        5.16077764,  7.16598273,  6.39187285, 13.89264822,  9.11473391,
        5.30158497, 12.49317563, 11.35345345, 11.63724792, 11.27160035,
        5.39098669,  5.94535529,  8.27853076,  5.62005629,  7.81509631,
       13.25017211,  5.54776934, 12.72524615,  4.79003361,  7.58378655,
        8.7273569 ,  5.05989547,  5.43544626, 11.22993218,  5.04028538,
        5.64500512,  9.94173708,  6.67108662,  6.97106912, 13.85158417,
       11.63502679, 10.35485331, 11.83570132,  9.09963894, 11.76830981,
       13.76047999,  6.41202986, 13.744387  , 10.63038565,  9.04255655,
       10.32715715])}
{'P': tensor([[2.3595e-02, 1.9881e-01, 1.9702e-01,  ..., 1.6514e-03, 2.0113e-04,
         3.7207e-05],
        [6.8959e-04, 2.6565e-02, 8.1311e-04,  ..., 4.7642e-04, 5.1178e-04,
         3.3971e-05],
        [8.8195e-04, 3.7622e-01, 2.0090e-02,  ..., 6.0550e-04, 2.9336e-04,
         2.8540e-05],
        ...,
        [6.3830e-03, 1.9127e-02, 1.2188e-02,  ..., 2.6324e-01, 3.3986e-04,
         1.6422e-05],
        [2.2054e-05, 2.6985e-05, 2.5349e-05,  ..., 8.8862e-07, 1.0174e-01,
         2.0873e-04],
        [3.2995e-03, 2.4281e-03, 2.9504e-03,  ..., 2.0285e-04, 6.4146e-05,
         6.3951e-02]]), 'X': tensor([[  1.8766,   7.3694,   1.1887],
        [  0.3038,   7.8092,   0.4164],
        [  0.9445,   7.6736,   0.5095],
        [ -0.5739,   7.9056,   0.6382],
        [  0.6933,   7.7336,   2.1179],
        [ -0.8885,   7.9160,   0.0571],
        [ 11.4703,  -4.2006,   7.4086],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 11.4703,  -4.2006,   7.4086],
        [  2.5128, -10.9365,   9.9645],
        [  6.0345,   3.0175,  -1.4424],
        [ -9.3419,   0.5972,  -1.2841],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.4915,   5.5990,   7.0405],
        [  5.1667,  -2.6543,   1.2879],
        [  6.1929,   2.4501,  -6.9352],
        [ -2.8162,   7.7059,  -9.6885],
        [  1.6422,   7.4586,  -6.8056],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.0961,   1.6529,   3.1553],
        [  2.3261,   7.1726,   6.2522],
        [ -8.0829,  -5.9940,   5.0564],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.9693,  -9.3200,  -8.3822],
        [  6.2770,  -0.0379,   2.4195],
        [ 12.3453,  -1.5462,   7.0212],
        [ -9.2974,   0.8371,   2.4852],
        [  9.5018,  -7.1945,  -1.1892],
        [ 12.5763,   0.1333,  -9.2650],
        [  1.9505,  -4.7171,   9.9868],
        [  3.4009,  -4.1874,  -5.9751],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.7558,  -4.4844,   0.0636],
        [  2.2659,  13.7894,   6.5436],
        [  1.8320,   7.3871,   5.1452],
        [  5.6314,  -1.9302,  -9.6543],
        [ 11.7349,  -3.5838,   2.1644],
        [ -5.1974,   6.7396,   3.3484],
        [  5.8836,   3.4315,  -4.0327],
        [ -8.9636,   2.0648,   8.1946],
        [  5.9587,  12.3216,   9.1695],
        [  1.1006,   7.6318,  -2.4877],
        [ 12.5717,   1.9266,   7.4903],
        [ -2.1382,   7.8339,   7.3524],
        [  6.0047,   3.1062,  -6.7914],
        [  1.8320,   7.3871,   5.1452],
        [ 10.0400,   8.6922,  -6.2847],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.7301,   7.4262,   5.0522],
        [ -3.3812, -10.1235,   7.4384],
        [ 12.1504,   4.3674,  -7.2998],
        [  8.7131,  10.2216,   7.7915],
        [  5.3118,   4.5452,   4.2114],
        [  6.2902,   0.0451,   3.7116],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.7976,  -4.4682,  -4.2020],
        [  7.0358,  -9.2784,  -9.8479],
        [ -5.6651,   6.4380,   1.3660],
        [  5.4179,  -2.2925,  -3.4641],
        [  5.1886,  -2.6250,   6.4677],
        [  9.4516,  -7.2501,  -1.8868],
        [ -3.0008, -10.2827,  -8.9480],
        [  1.7218,  -4.7576,  -3.3617],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 12.5996,   1.3971,  -1.0694],
        [  0.9074,  -4.8139,   8.1674],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.0788,  -0.8710,  -6.8719],
        [  1.5246,  -4.7838,  -5.2095],
        [  6.3122,  -9.6994,  -6.3282],
        [ -8.9788,   2.0208,  -3.2867],
        [  0.8940, -11.0402,  -0.9783],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.3542,   0.6934,  -3.3840],
        [  9.3624,   9.5318,   0.2779],
        [  1.7232,  -4.7574,   3.1270],
        [  0.1931,  -4.7491,   7.4385],
        [ -0.5709,   7.9055,   7.6089],
        [  2.8890,  -4.4316,   1.1600],
        [  5.8323,  -1.5228,   5.1132],
        [  6.0955,   2.8210,  -4.3321],
        [  2.2849, -10.9655,  -7.6542],
        [ 11.7162,   5.6728,  -8.9372],
        [ -8.0766,  -6.0046,  -3.1873],
        [  5.4867,  12.5816,   2.9549],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.3928,   7.1404,  -1.7035],
        [  9.5261,  -7.1675,   1.4206],
        [  6.3266,   0.3292,   9.3502],
        [  2.4572,   7.1085,   7.7376],
        [ -7.7842,   4.3466,  -1.7548],
        [  5.1886,  -2.6250,   6.4677],
        [  7.3724,  -9.0577,  -8.9843],
        [ 11.7795,  -3.4697,   3.5624],
        [ -8.0965,  -5.9712,   4.3466],
        [ 10.9931,   7.2107,   7.7933],
        [ -6.2200,  -8.2737,   5.5079],
        [  2.2182,  13.8007,   5.1889],
        [ -9.4731,  -0.7120,  -6.0271],
        [  6.0560,  -0.9421,  -5.0334]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0822, 0.0603,  ..., 0.6796, 0.9256, 0.4171],
        [0.0822, 0.0000, 0.0270,  ..., 0.7616, 0.8449, 0.3980],
        [0.0603, 0.0270, 0.0000,  ..., 0.7394, 0.8706, 0.3887],
        ...,
        [0.6796, 0.7616, 0.7394,  ..., 0.0000, 1.5909, 0.9298],
        [0.9256, 0.8449, 0.8706,  ..., 1.5909, 0.0000, 0.9587],
        [0.4171, 0.3980, 0.3887,  ..., 0.9298, 0.9587, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.60463349,  7.81509631,  7.73151309,  7.92644746,  7.76457095,
        7.96575782, 12.21532282, 10.60226614, 12.49317563, 12.21532282,
       11.22142195,  6.74686071,  9.36093562, 10.74409834,  7.17789666,
        5.80860165,  6.65991877,  8.20436764,  7.63726134,  9.75401449,
       10.74655346,  9.24501829,  7.5403822 , 10.06286074,  4.978704  ,
       11.63765273,  6.27715105, 12.44177641,  9.3349858 , 11.9182956 ,
       12.5769659 ,  5.10446653,  5.39450305, 12.25806559,  5.26342203,
       13.97430191,  7.61088125,  5.95297459, 12.26997165,  8.51088913,
        6.81117802,  9.19837792, 13.68672855,  7.71075226, 12.71843462,
        8.12043983,  6.76056625,  7.61088125, 13.27994304, 12.3828957 ,
        7.62509582, 10.67322527, 12.91143491, 13.43127131,  6.99097013,
        6.29034958,  6.12886024,  5.27178363, 11.64436684,  8.57560275,
        5.8828986 ,  5.81481044, 11.9120321 , 10.71162681,  5.05963217,
       10.39942497,  9.43112957, 12.67680439,  4.89869205,  8.47371868,
        6.14086582,  5.0209005 , 11.57249984,  9.20340031, 11.07637531,
       12.42554808, 11.63502679,  7.53884104,  9.78147542,  6.39187285,
       13.3607379 ,  5.05989547,  4.75303127,  7.92607342,  5.29009836,
        6.02779201,  6.7166297 , 11.20100795, 13.01728313, 10.06407839,
       13.72594022, 12.9089559 ,  7.53063885, 11.92133129,  6.33517349,
        7.52117132,  8.91551231,  5.81481044, 11.67875855, 12.27991991,
       10.06023665, 13.14690137, 10.35093485, 13.97779839,  9.4997987 ,
        6.12886024])}
{'P': tensor([[2.7886e-02, 2.4612e-01, 1.9910e-01,  ..., 4.4474e-04, 1.0412e-05,
         1.6400e-05],
        [1.1650e-03, 3.9616e-02, 1.2633e-03,  ..., 5.0841e-04, 1.1412e-05,
         1.1717e-05],
        [1.2107e-03, 4.7800e-01, 2.3495e-02,  ..., 2.8907e-04, 6.8282e-06,
         8.6935e-06],
        ...,
        [3.2182e-05, 3.9871e-05, 3.4002e-05,  ..., 4.9323e-02, 1.4002e-04,
         1.3361e-06],
        [6.7092e-04, 3.8435e-04, 4.2490e-04,  ..., 9.9461e-04, 3.4800e-01,
         2.2797e-03],
        [1.6976e-03, 9.1155e-04, 1.0976e-03,  ..., 1.6754e-05, 1.0829e-06,
         5.7589e-03]]), 'X': tensor([[  1.8719,   7.3713,  -3.7465],
        [  0.4996,   7.7738,  -4.4711],
        [  1.0000,   7.6591,  -4.5011],
        [ -1.7290,   7.8825,  -3.3831],
        [ -1.9487,   7.8590,  -3.3330],
        [ -2.3317,   7.8034,  -3.6274],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.4697,  -4.5836,   1.4593],
        [  2.1771, -10.9776,  -3.2007],
        [ 11.1085,  -4.9204,   0.0736],
        [  0.1532,  -4.7422,  -7.0145],
        [ -8.6426,  -4.9136,   9.4941],
        [  1.1274,  14.0116,   8.1753],
        [  6.2038,   2.4020,  -3.1692],
        [  2.6451,  -4.5250,  -0.5565],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 10.7566,   7.6212,  -2.3279],
        [  8.9284,  10.0036,  -6.9210],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.5128, -10.9365,   9.9645],
        [ -8.0087,   4.0173,   9.1091],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.3604,   0.9042,  -6.5901],
        [ -9.4203,  -2.0340,   2.5856],
        [  6.3505,   0.6235,  -8.5849],
        [  3.6817,   6.3354,   6.1233],
        [  6.1984,  -0.4327,   1.5438],
        [  5.6523,   3.9443,  -2.9449],
        [  3.2990,  -4.2415,  -7.4857],
        [ -7.9925,  -6.1423,   2.8990],
        [  5.1890,   4.7318,   0.9515],
        [  2.5813,  -4.5471,   5.3597],
        [ -8.3062,  -5.6008,   0.0287],
        [  3.1064,  13.5607,   6.7605],
        [ 12.2218,   4.0953,  -8.2072],
        [ -1.9714,   7.8563,  -1.8084],
        [  1.4489,   7.5256,   4.4529],
        [  0.1252, -11.0062,   7.2564],
        [  2.1477,   7.2550,   8.9688],
        [  9.6816,   9.1536,  -5.8543],
        [ -1.5875,   7.8944,  -9.4286],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.7298,  13.3537,  -9.4653],
        [ -6.2200,  -8.2737,   5.5079],
        [ -7.9925,  -6.1423,   2.8990],
        [  9.5261,  -7.1675,   1.4206],
        [  0.5175,  14.0910,   4.6540],
        [  2.3928,   7.1404,  -1.7035],
        [  0.1382, -11.0073,   1.0026],
        [  4.6357,   5.4415,   7.9067],
        [  0.1206,  -4.7363,   8.7103],
        [  4.2041,  -3.6509,  -4.9488],
        [  1.2728, -11.0372,   9.8121],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.4196,   0.0478,   9.4140],
        [  3.1480,  -4.3165,  -7.8002],
        [ -0.7599, -10.8995,  -6.0366],
        [ -5.3736,   6.6310,  -1.5174],
        [  4.8051, -10.3703,   0.5268],
        [  6.3246,   1.6696,  -0.1544],
        [ -3.9858,  -9.8334,  -3.7323],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.8093,  -9.4178,   8.4244],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 11.1420,   6.9331,  -2.4779],
        [  3.7298,  13.3537,  -9.4653],
        [ -8.8154,   2.4611,   2.5198],
        [ -3.1671,   7.6158,   6.0765],
        [  2.8474,   6.8979,   6.6457],
        [  3.3167,   6.6022,   6.3554],
        [ 10.0619,   8.6627,  -9.2884],
        [  0.4119,  14.1019,  -4.1870],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.2555,   2.1474,   4.1685],
        [  1.2728, -11.0372,   9.8121],
        [  4.7739,  -3.1218,  -4.1188],
        [  7.8425,  -8.7202,   6.4530],
        [ 12.2218,   4.0953,  -8.2072],
        [ -8.4887,  -5.2448,  -0.6585],
        [  2.7558,  -4.4844,   0.0636],
        [  7.3592,  11.3937,   6.2407],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.4157,   0.0821,  -5.0025],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.4117,   0.1160,   8.2574],
        [  3.2366,  -4.2732,   5.1532],
        [  6.1929,   2.4501,  -6.9352],
        [  3.0719, -10.8449,  -4.1673],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.2026,  12.1774,  -1.5887],
        [ -2.8162,   7.7059,  -9.6885],
        [ -8.3653,   3.4184,   9.3454],
        [  5.1667,  -2.6543,   1.2879],
        [ 10.5678,   7.9261,  -0.6100],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.8450,  -9.9356,   3.0767],
        [  0.1460,   7.8340,  -9.4614],
        [ -5.3382,   6.6533,   7.8054],
        [  3.3839,  -4.1966,   8.3574],
        [ 11.7418,   5.6070,  -1.4192]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0692, 0.0529,  ..., 0.8454, 0.7797, 0.4043],
        [0.0692, 0.0000, 0.0295,  ..., 0.8652, 0.8315, 0.4710],
        [0.0529, 0.0295, 0.0000,  ..., 0.8789, 0.8297, 0.4462],
        ...,
        [0.8454, 0.8652, 0.8789,  ..., 0.0000, 0.4575, 0.9499],
        [0.7797, 0.8315, 0.8297,  ..., 0.4575, 0.0000, 0.6729],
        [0.4043, 0.4710, 0.4462,  ..., 0.9499, 0.6729, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.60529892,  7.78980299,  7.72415048,  8.06990483,  8.09704052,
        8.14433542, 11.75807252,  5.20659892, 11.1913598 , 12.1494023 ,
        4.74468406,  9.94173708, 14.05687815,  6.65259775,  5.24137909,
       12.10942412, 13.18278606, 13.40850328,  5.1224079 , 11.22142195,
        8.95983586, 10.61472636, 13.08414689,  9.96391045,  9.45995955,
        6.42439344,  9.63742669,  6.38105015,  7.32754882,  6.2134961 ,
        6.89245352,  5.37343173, 10.08001894,  7.02253235,  5.2287058 ,
       10.01804223, 13.91198191, 12.889695  ,  8.09984231,  7.66377554,
       11.00694688,  7.56617037, 13.32374056,  8.05242591, 12.19805394,
       13.86480175, 10.35093485, 10.08001894, 11.92133129, 14.10045797,
        7.53063885, 11.00812996,  7.14837514,  4.73784222,  5.56809335,
       11.11038551, 11.70016217,  9.41970031,  5.34250225, 10.92597196,
        8.53501777, 11.42947632,  6.54127431, 10.61047596, 11.72749923,
       11.62158554, 12.10222204,  9.5014146 , 13.12299145, 13.86480175,
        9.15253344,  8.24808832,  7.46248947,  7.38844703, 13.27717982,
       14.10796466,  8.95983586,  9.36093562,  6.61386436, 11.11038551,
        5.70405525, 11.72802491, 12.889695  ,  9.97822473,  5.26342203,
       13.56367857, 10.01804223,  9.41605684,  9.97822473,  9.58072286,
        9.41244865,  5.36061974,  6.65991877, 11.27160035, 10.67132524,
       13.6660671 ,  8.20436764,  9.0368458 ,  5.80860165, 13.20988996,
        4.78258092, 11.52733581,  7.83534987,  8.53014915,  5.39098669,
       13.01187484])}
{'P': tensor([[2.9207e-02, 3.0877e-01, 1.6721e-01,  ..., 6.0673e-05, 4.6801e-04,
         1.4592e-05],
        [3.4545e-03, 4.1206e-02, 2.4621e-01,  ..., 8.2659e-05, 7.0027e-04,
         2.1184e-05],
        [3.7995e-04, 3.9582e-04, 3.8562e-02,  ..., 1.1921e-05, 7.1494e-04,
         4.6510e-05],
        ...,
        [2.9355e-04, 3.1630e-04, 4.4271e-04,  ..., 1.7174e-02, 1.2874e-03,
         2.4852e-07],
        [4.2058e-05, 4.4258e-05, 1.1237e-04,  ..., 6.8996e-06, 7.7271e-02,
         1.9133e-04],
        [3.4726e-05, 3.3883e-05, 1.6141e-05,  ..., 3.8389e-06, 5.8855e-06,
         9.5584e-02]]), 'X': tensor([[  1.8320,   7.3871,   5.1452],
        [  1.7301,   7.4262,   5.0522],
        [ -1.5408,   7.8978,   5.3470],
        [ -0.5385,   7.9037,   6.3830],
        [ -1.6086,   7.8928,   4.3385],
        [ -2.5056,   7.7720,   5.5878],
        [  6.2883,   0.0325,   5.2692],
        [  3.3474,  -4.2162,   5.2455],
        [  1.8721,  -4.7322,   0.5524],
        [ 12.4691,   2.8525,  -5.5110],
        [ -8.4573,   3.2448,   5.4370],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.0604,   1.7707,   1.1066],
        [  2.3367,   7.1676,  -9.5263],
        [  2.8602,   6.8905,   5.8779],
        [  5.8836,   3.4315,  -4.0327],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.1644, -11.0394,   7.8511],
        [  5.6248,   3.9984,   8.4933],
        [  6.8704,  -9.3809,   7.5769],
        [  3.2159, -10.8166,  -1.0406],
        [  6.3211,   1.7017,  -0.2164],
        [  8.1528,  10.7452,   9.1493],
        [  2.2373,  -4.6506,  -1.7334],
        [ 10.5955,   7.8824,  -3.7006],
        [  3.2647, -10.8066,  -0.3768],
        [  2.7558,  -4.4844,   0.0636],
        [  5.8201,   3.5841,  -8.0609],
        [ -3.3116, -10.1539,   7.5919],
        [ 12.4592,  -0.9168,  -6.7571],
        [  2.7121,   6.9743,  -8.4202],
        [ 11.9808,   4.9340,  -3.3715],
        [  1.7815,  -4.7481,   4.4289],
        [  9.8082,  -6.8396,   4.9615],
        [ -9.2508,   1.0572,   2.9978],
        [  0.7145,  -4.8071,  -2.3766],
        [ -9.4671,  -0.5553,   0.1117],
        [ -5.1398,  -9.1367,   6.2074],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.0233,   4.9647,  -3.1788],
        [  6.0305,   3.0296,   0.1375],
        [ -6.4464,   5.8272,   7.5038],
        [  5.2911,  -2.4824,  -3.6877],
        [ -7.8280,   4.2848,  -7.1955],
        [  8.8091,  -7.9037,   0.3631],
        [  9.1820,   9.7334,  -2.7932],
        [  5.3860,  -2.3418,   1.0826],
        [  4.2275,  -3.6320,  -0.2714],
        [ -8.8154,   2.4611,   2.5198],
        [ 10.8218,  -5.4177,   6.8573],
        [  4.9337,  12.8566,  -3.6582],
        [  7.4132,  11.3526,   0.3322],
        [  3.2366,  -4.2732,   5.1532],
        [ -1.9714,   7.8563,  -1.8084],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.3759,  -1.3943,   6.4659],
        [  5.9526,   3.2525,  -4.3821],
        [  4.8432,   5.1968,   1.5006],
        [  5.0233,   4.9647,  -3.1788],
        [ -6.3066,  -8.1936,   9.5822],
        [  1.9204,   7.3518,   8.4781],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.3048,   0.1469,  -9.1691],
        [  5.1196,  -2.7160,   4.2705],
        [  0.7145,  -4.8071,  -2.3766],
        [ -3.5173,   7.5091,  -8.9585],
        [ -6.4002,  -8.1048,  -0.9108],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.1637,  13.1898,  -1.9189],
        [  5.6258,  -1.9404,  -6.0715],
        [ -6.5295,  -7.9785,  -7.4429],
        [  6.0467,  -9.8369,  -6.5816],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.2911,  -2.4824,  -3.6877],
        [ -5.6169,  -8.7844,   8.9057],
        [  5.9655,   3.2172,  -7.5953],
        [ -6.3489,   5.9117,   6.5161],
        [  2.0899,  -4.6870,  -7.4260],
        [  8.7131,  10.2216,   7.7915],
        [  2.0729,   7.2878,   7.0880],
        [  6.9956,  11.6590,  -7.6876],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.0965,  -5.9712,   4.3466],
        [ -3.9700,   7.3453,  -4.1486],
        [ -0.5739,   7.9056,   0.6382],
        [ -3.2053, -10.1993,  -3.8736],
        [  4.4915,   5.5990,   7.0405],
        [  9.3836,   9.5075,  -5.9278],
        [  8.1043,  -8.5164,  -7.9290],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.7544,  -3.1425,  -1.5315],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.2811,  -7.4336,   5.1771],
        [  4.4049,   5.6892,  -4.4394],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.1623,  -3.4288,  -4.9961],
        [  9.1820,   9.7334,  -2.7932],
        [  1.6422,   7.4586,  -6.8056],
        [ -8.3535,  -5.5118,   9.8399],
        [  7.8375,  -8.7240,  -9.8913],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.4430,  -0.1897,   4.8383],
        [  7.0171,  11.6439,  -5.0178],
        [ -5.9987,   6.1949,  -6.2114],
        [ -3.4488, -10.0934,   2.5195]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0121, 0.1079,  ..., 0.9549, 0.8953, 0.6833],
        [0.0121, 0.0000, 0.0973,  ..., 0.9568, 0.8867, 0.6713],
        [0.1079, 0.0973, 0.0000,  ..., 1.0217, 0.8673, 0.5857],
        ...,
        [0.9549, 0.9568, 1.0217,  ..., 0.0000, 0.8438, 1.2898],
        [0.8953, 0.8867, 0.8673,  ..., 0.8438, 0.0000, 0.6760],
        [0.6833, 0.6713, 0.5857,  ..., 1.2898, 0.6760, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.61088125,  7.62509582,  8.04664886,  7.92200342,  8.0550286 ,
        8.16584537,  6.28835101,  5.38342077,  5.08910491, 12.79126785,
        9.05843201,  5.79391825,  9.23178204,  7.53884104,  7.46053281,
        6.81117802,  9.38232121, 11.10066126,  6.90115901, 11.62770332,
       11.2845608 ,  6.54615887, 13.48809595,  5.16077764, 13.20597932,
       11.28895489,  5.26342203,  6.83515201, 10.68031351, 12.49291768,
        7.48311002, 12.95702647,  5.0713364 , 11.95743558,  9.3109884 ,
        4.85995002,  9.48337063, 10.48315436,  9.85384623,  7.06272256,
        6.74872981,  8.68979052,  5.84450127,  8.92395381, 11.83509372,
       13.38091121,  5.87305005,  5.57341623,  9.15253344, 12.10222204,
       13.77075174, 13.55868684,  5.36061974,  8.09984231,  5.16611782,
       12.45418236,  6.78325151,  7.10378919,  7.06272256, 10.33958639,
        7.5984809 ,  5.34250225,  6.30647746,  5.79543038,  4.85995002,
        8.29203356, 10.32715715,  5.41348121, 13.83139024,  5.9510489 ,
       10.30972193, 11.54672755, 10.65929968,  9.68323686,  5.84450127,
       10.42667093,  6.77777606,  8.67502348,  5.13182506, 13.43127131,
        7.57685908, 13.59673406, 13.50484539, 10.06023665,  8.34949383,
        7.92644746, 10.6910768 ,  7.17789666, 13.35832725, 11.75618513,
        5.24137909,  5.69913817, 12.10222204, 11.89105259,  7.19513806,
       13.32645552,  9.78286733, 13.38091121,  7.63726134, 10.00802627,
       11.72749923,  5.40689332,  9.44486609, 13.59480906,  8.62329466,
       10.66631784])}
{'P': tensor([[3.0436e-02, 2.3914e-01, 1.7062e-01,  ..., 3.1648e-05, 2.1803e-04,
         6.8516e-05],
        [3.3156e-03, 3.8825e-02, 2.2611e-01,  ..., 4.0187e-05, 3.4700e-04,
         8.5407e-05],
        [1.5148e-03, 1.7933e-03, 1.4695e-01,  ..., 5.5787e-05, 1.9278e-03,
         3.1740e-04],
        ...,
        [7.6439e-04, 8.6336e-04, 3.7911e-04,  ..., 5.5202e-02, 3.1354e-04,
         5.1055e-05],
        [1.1638e-05, 1.3095e-05, 1.9508e-05,  ..., 3.8649e-05, 6.2024e-02,
         5.9743e-05],
        [3.6955e-03, 3.2620e-03, 1.2702e-03,  ..., 1.1949e-03, 4.1810e-05,
         1.6280e-01]]), 'X': tensor([[  1.7659,   7.4127,   4.7871],
        [  1.4489,   7.5256,   4.4529],
        [ -1.5408,   7.8978,   5.3470],
        [ -1.6086,   7.8928,   4.3385],
        [ -0.5385,   7.9037,   6.3830],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.4714,  -1.3301,  -3.1998],
        [  8.2422,  -8.4041,  -1.4962],
        [ -9.3616,  -2.4750,   3.7765],
        [  1.4749,  -4.7892,  -0.8089],
        [  8.4691,  10.4571,  -5.0650],
        [  5.5101,  -2.1433,   9.5442],
        [  4.1999,  -3.6543,  -4.1998],
        [  2.0937,  13.8293,  -2.3533],
        [  2.1504,  -4.6726,   8.7908],
        [  6.2975,   0.0947,   5.1791],
        [ -3.3116, -10.1539,   7.5919],
        [ -2.6735, -10.4062,  -1.5911],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.4887,   4.7364,   8.9074],
        [  5.8201,   3.5841,  -8.0609],
        [  2.7121,   6.9743,  -8.4202],
        [  7.5784,  -8.9141,   5.8410],
        [  2.8890,  -4.4316,   1.1600],
        [  6.0305,   3.0296,   0.1375],
        [  5.0233,   4.9647,  -3.1788],
        [  0.1532,  -4.7422,  -7.0145],
        [ -0.3432,   7.8900,   7.1805],
        [  6.1685,  12.1979,  -6.2473],
        [ -5.1369,   6.7755,   8.3836],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.2608,  12.6977,   4.9166],
        [ 12.4877,  -0.7222,   8.8347],
        [  3.2809,  13.5061,  -4.0199],
        [  0.6470, -11.0352,  -1.1931],
        [  5.3860,  -2.3418,   1.0826],
        [ -3.7964,  -9.9294,   7.7147],
        [  2.7558,  -4.4844,   0.0636],
        [  8.9284,  10.0036,  -6.9210],
        [  9.6662,   9.1725,   8.1056],
        [  3.3744,   6.5623,  -6.3636],
        [ 12.6038,   0.7665,  -3.4439],
        [  6.3305,   1.6119,   1.1538],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.3587,   1.1596,  -3.6629],
        [  5.7485,  -9.9811,  -3.2355],
        [ 12.2692,  -1.8882,   3.9253],
        [ -8.3539,  -5.5111,   0.3373],
        [  7.9141,  -8.6656,   9.6948],
        [  6.3360,   0.4258,  -9.2509],
        [ -2.9497, -10.3028,  -7.3479],
        [  1.6422,   7.4586,  -6.8056],
        [ -0.1626, -10.9796,   6.4629],
        [  7.0171,  11.6439,  -5.0178],
        [ -4.6393,   7.0461,   1.1624],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.2036,   2.4030,  -1.2500],
        [  2.2849, -10.9655,  -7.6542],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.4645,  -1.4880,  -2.0226],
        [  4.6992,   5.3690,   3.0935],
        [ 12.5770,   1.8512,  -0.6578],
        [ -7.6719,  -6.6260,   7.1468],
        [ -1.9487,   7.8590,  -3.3330],
        [ 12.4996,   2.6335,  -6.3166],
        [  6.2905,   1.9366,   2.1169],
        [  9.3424,  -7.3685,  -3.6807],
        [ -0.0410,   7.8591,   9.3457],
        [  8.2422,  -8.4041,  -1.4962],
        [ -9.4748,  -0.7765,  -5.6695],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.8450,  -9.9356,   3.0767],
        [  5.3858,   4.4260,  -9.6769],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.9655,   3.2172,  -7.5953],
        [  1.9241,   7.3503,   7.9004],
        [ -9.1706,   1.3863,   8.4032],
        [  2.9470,  13.6085,   2.9809],
        [ -0.2586,   7.8825,  -9.9840],
        [ -9.3419,   0.5972,  -1.2841],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.4426, -10.1181,  -5.1268],
        [  6.3048,   0.1469,  -9.1691],
        [  4.8653,   5.1694,  -4.0823],
        [  7.0358,  -9.2784,  -9.8479],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.4996,   7.7738,  -4.4711],
        [  0.6396,  14.0772,  -6.3786],
        [  2.2659,  13.7894,   6.5436],
        [  6.9956,  11.6590,  -7.6876],
        [ -5.5525,   6.5146,  -4.4984],
        [ 12.2914,   3.8037,   6.0978],
        [ -8.8631,  -4.3728,  -1.1746],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.9956,  11.6590,  -7.6876],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.2902,   0.0451,   3.7116],
        [  7.1697,  11.5343,  -0.7276],
        [  8.1528,  10.7452,   9.1493],
        [  0.9809,  -4.8145,   6.8706],
        [  5.9587,  12.3216,   9.1695],
        [  6.1575,  12.2046,   3.3257],
        [  7.0664,  -9.2590,   8.0879],
        [  6.1372,   2.6726,  -6.7885],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.3474,  -4.2162,   5.2455]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 2.2988e-02, 1.0592e-01,  ..., 7.9252e-01, 1.0251e+00,
         1.6430e-01],
        [2.2988e-02, 0.0000e+00, 9.7163e-02,  ..., 7.8216e-01, 1.0025e+00,
         1.8105e-01],
        [1.0592e-01, 9.7163e-02, 0.0000e+00,  ..., 8.5652e-01, 9.9750e-01,
         2.6778e-01],
        ...,
        [7.9252e-01, 7.8216e-01, 8.5652e-01,  ..., 0.0000e+00, 8.2597e-01,
         7.5695e-01],
        [1.0251e+00, 1.0025e+00, 9.9750e-01,  ..., 8.2597e-01, 0.0000e+00,
         1.1248e+00],
        [1.6430e-01, 1.8105e-01, 2.6778e-01,  ..., 7.5695e-01, 1.1248e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.62011734,  7.66377554,  8.04664886,  8.0550286 ,  7.92200342,
        8.09899874,  9.5642993 , 11.77124373,  9.68324553,  5.01114719,
       13.45642379,  5.91222346,  5.5671389 , 13.98691426,  5.14371005,
        6.29821673, 10.68031351, 10.74409834, 10.74655346,  8.86082124,
        6.83515201,  7.48311002, 11.70016217,  5.29009836,  6.74872981,
        7.06272256,  4.74468406,  7.89745107, 13.66896739,  8.50268322,
       11.81423161, 13.744387  , 12.5085995 , 13.89886469, 11.05413545,
        5.87305005, 10.63038565,  5.26342203, 13.40850328, 13.32557133,
        7.37902766, 12.62711252,  6.53250937, 12.41366754,  6.46356806,
       11.51811603, 12.41366754, 10.00794468, 11.73567759,  6.35029274,
       10.7167285 ,  7.63726134, 10.98076873, 13.59480906,  8.43626114,
        5.43544626,  6.652742  , 11.20100795, 13.77075174,  9.58072286,
        7.13501264, 12.7125079 , 10.1371552 ,  8.09704052, 12.77402351,
        6.58184474, 11.89855267,  7.85919687, 11.77124373,  9.50654967,
        5.68942572, 11.52733581,  6.97106912, 12.32408516,  6.77777606,
        7.59795445,  9.27474386, 13.92390689,  7.88677538,  9.36093562,
       10.39942497, 11.4890861 ,  6.30647746,  7.09887557, 11.64436684,
        5.35774573,  7.78980299, 14.09176259, 13.97430191, 13.59673406,
        8.55982105, 12.86648771,  9.88310614,  4.90505566, 13.59673406,
        6.24187044,  6.29034958, 13.58101794, 13.48809595,  4.91336994,
       13.68672855, 13.66991027, 11.64746882,  6.69389431,  9.14044984,
        5.38342077])}
{'P': tensor([[2.0497e-02, 4.2815e-01, 1.1972e-01,  ..., 9.7227e-06, 1.5264e-05,
         9.9215e-05],
        [1.2081e-03, 3.3054e-02, 2.2487e-01,  ..., 1.6816e-05, 2.0663e-05,
         1.7173e-04],
        [8.9719e-04, 1.0009e-03, 9.1017e-02,  ..., 1.0369e-04, 1.4342e-05,
         9.6204e-04],
        ...,
        [2.7347e-05, 2.5139e-05, 1.2842e-05,  ..., 7.8463e-02, 1.7763e-05,
         1.6163e-06],
        [2.2754e-03, 2.1269e-03, 7.9778e-04,  ..., 1.0134e-07, 1.0623e-02,
         7.3499e-06],
        [5.7332e-06, 5.9745e-06, 1.1532e-05,  ..., 3.7832e-05, 6.9389e-07,
         3.7674e-02]]), 'X': tensor([[  1.7500,   7.4187,   4.4237],
        [  1.4489,   7.5256,   4.4529],
        [ -1.6086,   7.8928,   4.3385],
        [ -1.5408,   7.8978,   5.3470],
        [ -1.9646,   7.8571,   3.3817],
        [ -2.5056,   7.7720,   5.5878],
        [  0.1041, -11.0045,   2.4311],
        [ -2.1817,   7.8274,  -4.6298],
        [  2.7590,  13.6621,   9.4496],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.4356,  -4.5942,  -5.0808],
        [ -2.6069, -10.4298,  -6.1995],
        [  0.9392,  -4.8143,  -3.4649],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.4505,  -1.7081,  -0.7039],
        [  6.9284,  11.7061,  -0.3732],
        [  2.1477,   7.2550,   8.9688],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.8704,  -9.3809,   7.5769],
        [  3.9515,  -3.8425,   6.1729],
        [ -1.9600,   7.8577,   6.9121],
        [  2.5318,  13.7231,   6.3692],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.0031,  12.2958,  -6.5992],
        [  4.1348,   5.9514,  -0.8856],
        [  9.0051,   9.9234,   5.4411],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.4297,  -5.3637,  -9.3962],
        [  9.7328,   9.0902,  -0.8917],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.7349,  -3.5838,   2.1644],
        [ -3.0004,   7.6607,  -7.2640],
        [  1.0093,   7.6567,   2.3526],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.9695,  -2.9407,  -6.3498],
        [  8.0381,  -8.5690,  -8.9523],
        [  9.2979,   9.6048,   7.3138],
        [  0.9659,  -4.8144,  -0.9594],
        [ -0.0282,   7.8575,  -1.1746],
        [  9.9756,  -6.6330,  -3.3429],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -7.0948,   5.1940,   9.8171],
        [  6.2902,   0.0451,   3.7116],
        [  5.8617,   3.4853,  -9.8080],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4721,  -1.3097,  -7.4377],
        [  8.0877,  -8.5297,  -1.1050],
        [ -4.0633,  -9.7927,  -3.0825],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.0729,   7.2878,   7.0880],
        [ 12.2914,   3.8037,   6.0978],
        [  2.7590,  13.6621,   9.4496],
        [ -4.9217,   6.8979,  -1.7667],
        [  8.2422,  -8.4041,  -1.4962],
        [ -1.7009,   7.8851,  -8.2437],
        [ -0.1963,   7.8765,  -1.1371],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -7.0948,   5.1940,   9.8171],
        [  4.3554,   5.7393,  -9.0766],
        [  2.0870,  -4.6877,   9.6823],
        [  6.3551,   0.7149,   1.3344],
        [  1.6422,   7.4586,  -6.8056],
        [ -3.5520,   7.4976,   9.1522],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.3474,  -4.2162,   5.2455],
        [ -5.6174,   6.4708,  -1.7339],
        [  6.3360,   0.4258,  -9.2509],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.4633,   5.6287,  -0.7754],
        [  0.3038,   7.8092,   0.4164],
        [  5.5294,  -2.1106,   9.0626],
        [ 11.4495,   6.3026,   6.8257],
        [  7.5309,  11.2617,   7.2656],
        [  5.6314,  -1.9302,  -9.6543],
        [  3.6154,  -4.0640,  -7.9698],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.1929,   2.4501,  -6.9352],
        [ -0.6323,   7.9085,  -5.9544],
        [  4.7482, -10.3907,   5.9149],
        [ -9.4117,   0.1160,   8.2574],
        [ -2.0445,   7.8469,  -0.5857],
        [ -5.3736,   6.6310,  -1.5174],
        [  6.2038,   2.4020,  -3.1692],
        [  6.1579,  -0.5961,  -6.5595],
        [  0.5960, -11.0335,   3.1144],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.1579,  -0.5961,  -6.5595],
        [ 12.1751,  -2.2592,  -5.7751],
        [  3.4009,  -4.1874,  -5.9751],
        [ -1.0482, -10.8487,  -5.9066],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.2732,  -4.7619,  -1.1577],
        [  4.6063,  13.0051,  -3.1456],
        [  2.5636, -10.9293,   9.6519],
        [ -7.6293,  -6.6859,  -5.5161],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.8979,   3.3958,  -3.8713],
        [  7.1821,  -9.1845,   9.8548],
        [ -3.3116, -10.1539,   7.5919],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.6112,   2.9319,  -7.9943]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0238, 0.1300,  ..., 0.6763, 0.5606, 1.0514],
        [0.0238, 0.0000, 0.1087,  ..., 0.6526, 0.5825, 1.0437],
        [0.1300, 0.1087, 0.0000,  ..., 0.5619, 0.6554, 0.9668],
        ...,
        [0.6763, 0.6526, 0.5619,  ..., 0.0000, 1.2145, 1.0637],
        [0.5606, 0.5825, 0.6554,  ..., 1.2145, 0.0000, 1.1539],
        [1.0514, 1.0437, 0.9668,  ..., 1.0637, 1.1539, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.62232307,  7.66377554,  8.0550286 ,  8.04664886,  8.09899874,
        8.16584537, 11.0050379 ,  8.12580976, 13.93790235, 11.75807252,
        5.19986797, 10.7506423 ,  4.90505566,  9.66977802,  9.60359158,
       13.60275211,  7.56617037, 10.52587243, 11.62770332,  5.51177501,
        8.09842704, 13.95472331,  9.51074763, 13.6829843 ,  7.24677664,
       13.40024224,  6.24614905,  9.61942445,  7.96181288,  8.04718433,
        9.99145096, 13.31764181,  6.1561646 , 12.26997165,  8.22728399,
        7.72291946, 13.66051884, 12.32545637, 11.74901778, 13.36800073,
        4.91038884,  7.8575648 , 11.97957425,  9.58072286, 12.05395056,
        8.79284466,  6.29034958,  6.81961996,  9.60080639,  9.562174  ,
       11.75438006, 10.60226614,  5.34250225,  7.57685908, 12.86648771,
       13.93790235,  8.47371868, 11.77124373,  8.06643487,  7.8788965 ,
       10.52357479,  8.79284466,  7.20482224,  5.13125513,  6.39520397,
        7.63726134,  8.29640613,  5.35774573,  5.38342077,  8.56889738,
        6.35029274, 12.9089559 ,  7.18354885,  7.81509631,  5.91854397,
       13.06957425, 13.54774355,  5.95297459,  5.43943791,  6.23262289,
        6.65991877,  7.9337647 , 11.42420508,  9.41244865,  8.10886021,
        8.53501777,  6.65259775,  6.18668463, 11.04953767, 10.63038565,
        6.18668463, 12.3828957 ,  5.39450305, 10.8992524 ,  9.43112957,
        7.96181288,  4.76969364, 13.79676654, 11.22596994, 10.14436912,
        6.27905292,  6.80559669, 11.65923396, 10.68031351, 13.13205357,
        9.09661212])}
{'P': tensor([[3.0231e-02, 1.8064e-01, 1.6154e-01,  ..., 3.6894e-06, 1.1764e-03,
         2.1729e-04],
        [3.3154e-04, 3.2299e-02, 2.6467e-02,  ..., 2.4052e-06, 2.2086e-04,
         8.4921e-04],
        [5.6024e-04, 2.8344e-02, 5.4362e-02,  ..., 4.1605e-06, 3.8002e-04,
         1.7584e-03],
        ...,
        [2.6077e-03, 4.9697e-04, 5.0832e-04,  ..., 1.4883e-01, 4.4911e-03,
         5.1462e-06],
        [1.9375e-01, 8.4912e-02, 6.5309e-02,  ..., 4.2182e-06, 2.6604e-02,
         1.0602e-04],
        [1.9985e-05, 2.1389e-05, 2.3226e-05,  ..., 6.4495e-05, 1.9663e-05,
         8.3095e-02]]), 'X': tensor([[ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0973, 0.1325,  ..., 0.1892, 0.0696, 0.8222],
        [0.0973, 0.0000, 0.0766,  ..., 0.2736, 0.1566, 0.7448],
        [0.1325, 0.0766, 0.0000,  ..., 0.2633, 0.2019, 0.6896],
        ...,
        [0.1892, 0.2736, 0.2633,  ..., 0.0000, 0.1871, 0.8994],
        [0.0696, 0.1566, 0.2019,  ..., 0.1871, 0.0000, 0.8914],
        [0.8222, 0.7448, 0.6896,  ..., 0.8994, 0.8914, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.62509582,  8.04664886,  8.0550286 ,  7.92200342,  8.16584537,
        8.11227668,  5.92952538,  6.53290663, 11.70016217, 10.23946247,
        7.58378655, 13.44593519, 11.66453059,  9.43737523,  5.88782913,
       12.22503754,  7.67939546,  6.35029274, 13.61600536,  8.95133961,
        7.38844703, 13.6105595 , 13.09737053, 12.49317563, 11.66453059,
        4.78258092, 10.54658597,  8.31625171,  8.06990483, 13.6105595 ,
       12.0495521 ,  4.98989906,  6.06416957,  6.18668463,  5.34250225,
        9.09661212, 10.84493732, 10.06023665,  5.39723241,  5.5513072 ,
        8.97052115,  8.67502348, 11.89855267, 13.79676654, 12.10942412,
       10.1979669 ,  7.62509582,  9.71392179,  5.53255882,  5.64865173,
        8.7273569 , 10.84493732, 10.30972193, 10.65929968, 13.36800073,
        5.05989547, 10.31886066,  9.05843201,  9.23178204,  7.54620849,
        9.15866607,  6.4020404 ,  8.68832398,  6.80559669, 14.09176259,
        4.75303127,  9.15976746, 11.22596994,  9.15866607, 12.0495521 ,
       13.48809595,  5.46244988,  4.74468406, 13.25017211, 12.22503754,
       11.37939695,  6.76056625,  7.67939546,  8.7907923 , 12.12448912,
       13.66896739, 12.32545637, 12.24467041,  7.39004226,  5.8828986 ,
        9.32094055,  5.29009836, 10.53368644, 12.17901033,  9.91668182,
        8.94395628, 10.95593729,  8.38252937, 14.05687815,  6.61211224,
       11.72802491,  6.28835101,  8.09899874, 13.98691426,  8.04483495,
       13.66896739, 12.00995898, 11.00812996, 12.17451685,  7.46053281,
        9.73579738])}
{'P': tensor([[1.8630e-02, 2.6423e-01, 2.2792e-01,  ..., 1.3981e-05, 4.6629e-07,
         6.2702e-05],
        [5.8511e-04, 2.5110e-02, 5.7728e-02,  ..., 1.3551e-05, 5.0452e-07,
         5.8319e-05],
        [5.0138e-04, 6.5127e-03, 2.1356e-02,  ..., 1.3038e-05, 4.5978e-07,
         4.8980e-05],
        ...,
        [1.0193e-04, 1.1901e-04, 1.4446e-04,  ..., 8.7575e-02, 1.2369e-05,
         5.2021e-05],
        [1.8042e-05, 1.1134e-05, 1.2039e-05,  ..., 4.1960e-04, 1.9542e-02,
         3.2071e-04],
        [2.8686e-02, 1.7145e-02, 1.7722e-02,  ..., 4.3109e-04, 5.0362e-05,
         7.9359e-02]]), 'X': tensor([[  1.6422,   7.4586,  -6.8056],
        [  0.3724,   7.7974,  -6.8722],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.0407,   7.9166,  -6.6021],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.3928,   7.1404,  -1.7035],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.8704,  -9.3809,   7.5769],
        [  1.4489,   7.5256,   4.4529],
        [  8.6420,  -8.0580,   6.5781],
        [ -7.6650,   4.5092,   0.2520],
        [  0.1460,   7.8340,  -9.4614],
        [  5.6248,   3.9984,   8.4933],
        [  6.3516,   0.6425,   0.3425],
        [ -5.1369,   6.7755,   8.3836],
        [  6.1973,  -0.4372,   6.5720],
        [  4.7739,  -3.1218,  -4.1188],
        [  0.1677,  -4.7447,  -6.1529],
        [  5.2911,  -2.4824,  -3.6877],
        [  4.8432,   5.1968,   1.5006],
        [  6.3497,   1.3680,  -1.9076],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.4732,  -2.2041,  -4.3830],
        [ -7.0825,   5.2073,  -5.0497],
        [ -9.0961,   1.6529,   3.1553],
        [  3.9895,   6.0816,  -6.5235],
        [  6.1579,  -0.5961,  -6.5595],
        [  3.0876,   6.7527,  -5.3540],
        [  2.4639, -10.9431,  -7.3718],
        [  2.7558,  -4.4844,   0.0636],
        [  9.0424,  -7.6779,  -7.5286],
        [ -8.1531,  -5.8748,   2.6934],
        [  6.3031,   1.8477,  -7.1706],
        [  5.3924, -10.1396,  -4.0863],
        [  2.1504,  -4.6726,   8.7908],
        [  3.8111, -10.6787,  -4.6539],
        [  2.1379,  -4.6757,   6.4981],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.5568,   6.0602,   9.3244],
        [  4.2275,  -3.6320,  -0.2714],
        [  1.4150,  13.9647,  -4.0915],
        [  6.3211,   1.7017,  -0.2164],
        [  7.7794,  11.0629,   8.9069],
        [ -4.4503,   7.1378,   3.3751],
        [  4.0591,   6.0202,  -1.6339],
        [ -7.6525,  -6.6534,   5.1721],
        [  2.6321,  -4.5296,  -2.5159],
        [ -5.6174,   6.4708,  -1.7339],
        [  0.7958,  -4.8109,  -8.2248],
        [ 11.2517,  -4.6495,   2.9800],
        [  8.5716,  10.3596,   6.8818],
        [  6.3515,   0.6418,  -6.5823],
        [ 12.4597,  -0.9136,  -4.8763],
        [  0.9659,  -4.8144,  -0.9594],
        [  6.1222,  -0.7261,   8.8204],
        [  6.2555,   2.1474,   4.1685],
        [  2.7685, -10.8982,  -6.8928],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.2373,  -3.1225,  -8.9763],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.4242, -10.1261,   3.9595],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.1231,  -3.5739,   8.4708],
        [  0.5850,  -4.7982,   9.6639],
        [  3.2159, -10.8166,  -1.0406],
        [  0.3715,  -4.7756,   0.2312],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.4749,  -4.7892,  -0.8089],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.6357,   5.4415,   7.9067],
        [  1.5246,  -4.7838,  -5.2095],
        [ -0.3432,   7.8900,   7.1805],
        [  6.3516,   0.6425,   0.3425],
        [ -9.3076,   0.7852,  -6.7010],
        [  2.2644,  -4.6434,   9.7508],
        [  5.3118,   4.5452,   4.2114],
        [  4.2096,   5.8815,   1.9230],
        [  8.2422,  -8.4041,  -1.4962],
        [ 10.0567,   8.6697,   4.6329],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.1940,  -2.1881,   7.1499],
        [ 10.3251,   8.2928,  -3.0006],
        [ -7.3554,   4.8985,  -0.6665],
        [  3.4401,  13.4540,   4.6576],
        [  6.1575,  12.2046,   3.3257],
        [  0.6952,  -4.8060,   8.8976],
        [ -0.3432,   7.8900,   7.1805],
        [ 10.7566,   7.6212,  -2.3279],
        [  5.8201,   3.5841,  -8.0609],
        [  0.1252, -11.0062,   7.2564],
        [  2.2076,  13.8032,  -1.5099],
        [  6.1755,   2.5233,  -2.0042],
        [  9.4516,  -7.2501,  -1.8868],
        [  4.7813,  -3.1139,   8.2691],
        [  4.8674,  12.8875,   6.7622],
        [  2.1477,   7.2550,   8.9688],
        [ 12.1379,   4.4125,  -3.8833],
        [  3.3069,   6.6089,   8.7377],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.9663,   6.1018,  -5.4104],
        [  4.2096,   5.8815,   1.9230],
        [  2.6970,   6.9826,   9.7350],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.3205,   1.7068,  -3.8619]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0468, 0.0453,  ..., 0.9582, 1.0334, 0.2345],
        [0.0468, 0.0000, 0.0147,  ..., 0.9678, 1.0448, 0.2770],
        [0.0453, 0.0147, 0.0000,  ..., 0.9531, 1.0301, 0.2683],
        ...,
        [0.9582, 0.9678, 0.9531,  ..., 0.0000, 0.0854, 0.8367],
        [1.0334, 1.0448, 1.0301,  ..., 0.0854, 0.0000, 0.9017],
        [0.2345, 0.2770, 0.2683,  ..., 0.8367, 0.9017, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.63726134,  7.80625981,  7.822854  ,  7.96181288,  7.98469166,
        7.9337647 ,  7.53063885, 12.53047489, 11.62770332,  7.66377554,
       11.8159254 ,  8.89301516,  7.83534987,  6.90115901,  6.38399789,
        8.50268322,  6.21275016,  5.70405525,  4.74770899,  5.84450127,
        7.10378919,  6.49538841,  5.53255882,  5.90035324,  8.7907923 ,
        9.24501829,  7.27340834,  6.18668463,  7.42512621, 11.21703654,
        5.26342203, 11.86239961, 10.04916678,  6.56833864, 11.48434767,
        5.14371005, 11.33837396,  5.14124612,  4.85603886, 13.0493368 ,
        5.57341623, 14.03618649,  6.54615887, 13.52430176,  8.41148722,
        7.26074925, 10.14044489,  5.23878371,  8.56889738,  4.8763135 ,
       12.17451685, 13.44593519,  6.383897  , 12.49317563,  4.91038884,
        6.16512856,  6.61386436, 11.24434374,  9.75075273,  9.75075273,
        5.79391825, 11.48734978,  9.95552626,  6.53290663,  9.79814997,
        4.83371341, 11.2845608 ,  4.79003361,  9.55593374,  5.01114719,
       10.31584102,  7.14837514,  5.0209005 ,  7.89745107,  6.38399789,
        9.34061938,  5.16611782,  6.99097013,  7.23276582, 11.77124373,
       13.2778378 ,  8.43626114, 12.38881635, 13.2430377 ,  8.83725081,
       13.88683436, 13.66991027,  4.85603886,  7.89745107, 13.18278606,
        6.83515201, 11.00694688, 13.97857591,  6.67108662, 11.9120321 ,
        5.70591986, 13.77604293,  7.56617037, 12.91505111,  7.39004226,
        9.14044984,  7.27759233,  7.23276582,  7.48538905, 12.10222204,
        6.54693987])}
{'P': tensor([[1.9563e-02, 2.8042e-01, 2.7290e-01,  ..., 6.2907e-06, 9.5883e-05,
         2.0205e-05],
        [8.5904e-04, 2.2646e-02, 4.6071e-01,  ..., 8.2352e-06, 7.0630e-05,
         1.7677e-05],
        [1.1551e-03, 1.4200e-03, 4.4528e-02,  ..., 1.7828e-05, 9.6472e-05,
         3.0343e-05],
        ...,
        [3.0294e-05, 2.4756e-05, 2.1236e-05,  ..., 5.9952e-02, 6.9697e-06,
         1.9272e-05],
        [1.7747e-03, 2.4469e-03, 2.7041e-03,  ..., 1.8162e-07, 1.4695e-02,
         1.5560e-06],
        [5.1281e-04, 4.4677e-04, 4.4266e-04,  ..., 2.0349e-06, 2.9001e-05,
         5.7139e-02]]), 'X': tensor([[ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0275, 0.0538,  ..., 0.5900, 0.5461, 0.8196],
        [0.0275, 0.0000, 0.0295,  ..., 0.5629, 0.5570, 0.8372],
        [0.0538, 0.0295, 0.0000,  ..., 0.5371, 0.5836, 0.8384],
        ...,
        [0.5900, 0.5629, 0.5371,  ..., 0.0000, 0.9966, 1.2169],
        [0.5461, 0.5570, 0.5836,  ..., 0.9966, 0.0000, 1.1845],
        [0.8196, 0.8372, 0.8384,  ..., 1.2169, 1.1845, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.6374603 ,  7.72415048,  7.78980299,  7.83486943,  8.06990483,
        8.12580976, 13.96746816, 13.06957425, 11.75618513,  9.88571713,
       11.83570132, 10.84493732,  7.23276582, 11.80092578,  6.42439344,
        8.95133961, 13.66896739,  5.95297459,  7.71075226,  9.76896904,
       13.5247215 ,  9.91668182,  9.0368458 ,  6.16512856,  9.23178204,
       12.62576026,  7.19513806, 13.45124917, 10.27716434, 10.86693708,
        9.23071698,  9.78147542,  7.96181288,  9.71392179, 12.33167382,
        7.19534146, 10.31886066,  9.25336371,  7.16058587,  5.90035324,
       13.68063098, 12.1494023 ,  6.69389431,  9.17016857,  9.44486609,
        9.20340031,  7.32754882, 11.10066126,  7.92200342, 14.10796466,
        9.55593374, 10.27572673, 12.84056604,  7.20826219,  5.40689332,
        9.23071698,  6.65991877,  6.652742  , 12.1494023 , 13.18278606,
        7.50564223,  7.02253235, 12.07225469, 10.65929968, 13.6922425 ,
        8.36445758, 10.48840476,  9.85384623, 11.76830981,  5.23878371,
        9.85384623,  6.33075682,  8.16714836,  9.36093562,  9.66977802,
        5.39098669,  5.81199422,  6.69389431,  7.32950385,  8.62260327,
       12.84056604,  7.6374603 ,  9.68323686, 10.1979669 ,  8.92395381,
        8.85939626,  9.10945678, 11.4424816 , 11.1913598 ,  7.32754882,
       11.77124373, 10.8992524 ,  6.99097013, 11.23201619,  7.69752262,
       12.41012046, 12.72524615,  5.94535529, 10.1979669 ,  9.23071698,
       13.97430191,  5.16077764, 10.42667093, 10.8992524 , 13.59056613,
        6.90115901])}
{'P': tensor([[3.0525e-02, 2.0766e-01, 1.7777e-01,  ..., 3.0181e-04, 1.2965e-02,
         3.1646e-03],
        [5.0682e-04, 4.6086e-02, 2.4029e-02,  ..., 1.6809e-03, 5.9860e-02,
         1.5288e-02],
        [3.0182e-04, 2.0267e-02, 2.4733e-02,  ..., 7.1613e-04, 4.7560e-02,
         5.2992e-03],
        ...,
        [1.0557e-05, 1.2337e-05, 1.1258e-05,  ..., 4.1858e-02, 2.5407e-05,
         6.6199e-05],
        [1.0503e-04, 2.7994e-04, 2.6348e-04,  ..., 6.8774e-03, 6.3669e-02,
         4.9114e-03],
        [1.9196e-05, 4.2094e-05, 3.7832e-05,  ..., 5.5117e-02, 2.6733e-04,
         2.5578e-02]]), 'X': tensor([[  1.4489,   7.5256,   4.4529],
        [ -1.6086,   7.8928,   4.3385],
        [ -1.5408,   7.8978,   5.3470],
        [ -1.9646,   7.8571,   3.3817],
        [ -2.5056,   7.7720,   5.5878],
        [ -0.5385,   7.9037,   6.3830],
        [ -8.7544,   2.6096,   3.3146],
        [  2.0937,  13.8293,  -2.3533],
        [  4.6063,  13.0051,  -3.1456],
        [  0.9809,  -4.8145,   6.8706],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -6.2668,   5.9808,   2.5614],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 11.5568,   6.0602,   9.3244],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.2328,  -2.0376,  -1.1234],
        [  3.9770, -10.6341,   8.0783],
        [ 12.5789,   0.1722,   1.6470],
        [ -6.1914,  -8.2997,   8.5380],
        [ -6.2668,   5.9808,   2.5614],
        [ 11.3367,  -4.4806,   0.4643],
        [  9.2811,  -7.4336,   5.1771],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.4505,  -1.7081,  -0.7039],
        [  6.3303,   1.6145,  -8.4083],
        [  3.3943,   6.5483,   1.0790],
        [  2.2659,  13.7894,   6.5436],
        [  2.3373,  13.7721,  -0.2442],
        [ -0.2586,   7.8825,  -9.9840],
        [ 10.3251,   8.2928,  -3.0006],
        [  4.5800,   5.5034,  -3.9432],
        [  0.5960, -11.0335,   3.1144],
        [  1.1644, -11.0394,   7.8511],
        [ -3.1671,   7.6158,   6.0765],
        [  1.6308,  13.9254,  -2.0984],
        [ 11.8951,  -3.1570,   7.0742],
        [  7.9403,  -8.6455,  -8.7433],
        [ -1.0407,   7.9166,  -6.6021],
        [  8.5716,  10.3596,   6.8818],
        [ -4.2271,   7.2387,  -1.8316],
        [ -2.5683, -10.4433,  -2.9182],
        [ 10.0567,   8.6697,   4.6329],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.4267,  -5.3697,   2.4172],
        [ -7.7842,   4.3466,  -1.7548],
        [  7.9403,  -8.6455,  -8.7433],
        [  7.7794,  11.0629,   8.9069],
        [ -7.6525,  -6.6534,   5.1721],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3031,   1.8477,  -7.1706],
        [ -5.1398,  -9.1367,   6.2074],
        [ -8.5900,   2.9767,   7.4811],
        [  0.7275,  14.0667,  -0.9815],
        [  9.4554,   9.4245,   9.8800],
        [ 12.4592,  -0.9168,  -6.7571],
        [  5.9003,  12.3550,   2.1917],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.4351,  -1.8891,   3.5642],
        [ 12.5770,   1.8512,  -0.6578],
        [  1.1994,   7.6035,   2.4295],
        [ -3.3812, -10.1235,   7.4384],
        [  8.4691,  10.4571,  -5.0650],
        [  3.7916,  -3.9524,  -7.9547],
        [  3.2366,  -4.2732,   5.1532],
        [  6.3124,   0.2057,  -0.4593],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.8363,   2.4086,  -5.6296],
        [  9.6587,   9.1817,  -0.8212],
        [ -8.4887,  -5.2448,  -0.6585],
        [  6.3205,   1.7068,  -3.8619],
        [  0.2480, -11.0153,  -3.0183],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.9809,   7.6642,  -1.4217],
        [ 11.4743,   6.2478,  -2.1100],
        [ -1.6275, -10.7221,   3.2376],
        [  8.4691,  10.4571,  -5.0650],
        [ -3.5029,   7.5138,   8.1853],
        [  1.7232,  -4.7574,   3.1270],
        [ -7.4602,   4.7718,  -4.6277],
        [ -9.4366,  -0.1189,  -2.6736],
        [  0.9074,  -4.8139,   8.1674],
        [ -7.8126,   4.3066,   5.1791],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.3534,   1.2990,  -7.2942],
        [  4.2612, -10.5514,  -6.6268],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.3367,   7.1676,  -9.5263],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.5424,  -2.0885,   7.2487],
        [ -8.6050,   2.9451,  -0.7012],
        [  1.5383, -11.0274,  -4.9666],
        [ -1.8823, -10.6558,  -3.8952],
        [  4.0078,  -3.8018,  -6.2168],
        [ 11.2571,   6.7072,   8.3693],
        [ -2.3317,   7.8034,  -3.6274],
        [  0.5600, -11.0321,  -0.8660],
        [  0.2571,  -4.7594,   1.5618],
        [  3.6703,   6.3443,  -5.4401],
        [ -9.3616,  -2.4749,   0.1793],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.9667,   4.0814,  -0.6397]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1087, 0.0972,  ..., 0.8042, 0.3641, 0.6572],
        [0.1087, 0.0000, 0.0766,  ..., 0.6993, 0.2856, 0.5676],
        [0.0972, 0.0766, 0.0000,  ..., 0.7565, 0.2676, 0.6390],
        ...,
        [0.8042, 0.6993, 0.7565,  ..., 0.0000, 0.6460, 0.2645],
        [0.3641, 0.2856, 0.2676,  ..., 0.6460, 0.0000, 0.6284],
        [0.6572, 0.5676, 0.6390,  ..., 0.2645, 0.6284, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.66377554,  8.0550286 ,  8.04664886,  8.09899874,  8.16584537,
        7.92200342,  9.13507321, 13.98691426, 13.79676654,  4.91336994,
       12.18045978,  8.66273171, 11.21703654, 12.49291768, 13.0493368 ,
       11.85743425, 12.4013153 , 11.35345345, 12.58006038, 10.35464916,
        8.66273171, 12.18998195, 11.89105259, 13.86480175,  9.60359158,
        6.53290663,  7.37576154, 13.97430191, 13.96905756,  7.88677538,
       13.2430377 ,  7.15990676, 11.04953767, 11.10066126,  8.24808832,
       14.02058621, 12.30695098, 11.73848065,  7.98469166, 13.44593519,
        8.38252937, 10.75443106, 13.2778378 ,  9.562174  ,  8.49161154,
        9.99211663,  8.91551231, 11.73848065, 13.52430176, 10.14044489,
        5.5366698 ,  6.56833864, 10.48315436,  9.09119351, 14.08549802,
       13.35013374, 12.49291768, 13.69162795,  6.78325151,  9.62238828,
       12.7125079 ,  7.69752262, 10.67322527, 13.45642379,  5.47702766,
        5.36061974,  6.31576264, 10.1371552 ,  9.15866607, 13.32645552,
        9.97822473,  6.54693987, 11.018084  , 12.30695098,  7.72668467,
       13.06498554, 10.84493732, 13.45642379,  8.29022197,  5.05989547,
        8.85572292,  9.43737523,  4.89869205,  8.92098752, 10.80728788,
        9.77588801,  6.48486171, 11.37939695,  7.83486943, 13.04217487,
        7.53884104, 12.12448912, 12.26997165,  5.92281142,  9.09502385,
       11.1341798 , 10.8207297 ,  5.52416368, 13.103709  ,  8.14433542,
       11.04629471,  4.7663526 ,  7.32950385,  9.68323686,  8.68979052,
        8.95133961])}
{'P': tensor([[1.5496e-02, 2.0277e-01, 1.9440e-01,  ..., 9.9910e-06, 3.7131e-06,
         2.1392e-05],
        [3.6985e-04, 1.5981e-02, 3.6662e-01,  ..., 5.2069e-06, 4.6991e-06,
         9.8587e-06],
        [5.2107e-04, 9.3018e-04, 2.4689e-02,  ..., 7.3724e-06, 7.5232e-06,
         1.3948e-05],
        ...,
        [6.7429e-04, 6.6947e-04, 6.4212e-04,  ..., 1.1994e-02, 4.4269e-08,
         6.2106e-04],
        [6.5543e-06, 4.1526e-06, 3.9712e-06,  ..., 2.0676e-06, 1.4438e-02,
         6.9301e-06],
        [7.3242e-03, 5.5318e-03, 5.2607e-03,  ..., 2.6935e-04, 1.2304e-07,
         1.2777e-02]]), 'X': tensor([[  1.3339,   7.5627,  -0.9034],
        [ -0.0282,   7.8575,  -1.1746],
        [ -0.1963,   7.8765,  -1.1371],
        [ -1.2454,   7.9127,  -0.5715],
        [ -0.8885,   7.9160,   0.0571],
        [ -2.0445,   7.8469,  -0.5857],
        [ -2.5683, -10.4433,  -2.9182],
        [ 11.8951,  -3.1570,   7.0742],
        [  4.6992,   5.3690,   3.0935],
        [  4.9152,  12.8653,  -1.4459],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.1252, -11.0062,   7.2564],
        [  8.5093,  -8.1764,  -3.2381],
        [  6.1575,  12.2046,   3.3257],
        [  0.2571,  -4.7594,   1.5618],
        [ -3.3999, -10.1152,  -1.2088],
        [  4.5340,  -3.3629,   8.9652],
        [ -2.0229, -10.6163,   8.6930],
        [  6.3460,   1.4252,   1.2291],
        [ 11.3367,  -4.4806,   0.4643],
        [  5.8979,   3.3958,  -3.8713],
        [  0.1460,   7.8340,  -9.4614],
        [  1.1006,   7.6318,  -2.4877],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.1420,   6.9331,  -2.4779],
        [  7.4132,  11.3526,   0.3322],
        [  5.8323,  -1.5228,   5.1132],
        [  6.2630,   2.1057,  -5.2136],
        [  3.5965,  -4.0754,  -3.5035],
        [ -4.0868,   7.2981,   8.1874],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.8745,   2.3097,   3.5600],
        [  6.0047,   3.1062,  -6.7914],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.5939,   6.4029,  -1.8357],
        [ 11.9641,  -2.9568,   7.7211],
        [  4.0587, -10.6112,   6.2273],
        [  5.1787,  -2.6383,  -4.2533],
        [ -8.6381,  -4.9238,  -4.4077],
        [  8.5198,  10.4091,  -8.6486],
        [  5.7963,  -1.6018,   1.8856],
        [  2.3367,   7.1676,  -9.5263],
        [  0.6363, -11.0348,   6.1414],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.2975,   0.0947,   5.1791],
        [  7.1441,  11.5529,   2.3232],
        [  4.9531,   5.0576,   5.6794],
        [ 10.9162,  -5.2599,  -7.0812],
        [  0.5600, -11.0321,  -0.8660],
        [ -4.6393,   7.0461,   1.1624],
        [ -0.2586,   7.8825,  -9.9840],
        [ -2.5056,   7.7720,   5.5878],
        [  8.2154,  -8.4262,   8.0171],
        [  0.1532,  -4.7422,  -7.0145],
        [ -3.3116, -10.1539,   7.5919],
        [  4.8653,   5.1694,  -4.0823],
        [  1.6954,  -4.7616,   9.8198],
        [  1.7903,  -4.7467,   0.1168],
        [  6.1741,  -0.5329,   3.1913],
        [  2.5382,   7.0672,   4.5097],
        [  4.5814, -10.4485,   2.5496],
        [ -8.9624,  -4.0944,  -4.9622],
        [  0.1931,  -4.7491,   7.4385],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.2883,   0.0325,   5.2692],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.3200,   0.2690,   1.5900],
        [  4.5814, -10.4485,   2.5496],
        [  8.3718,  -8.2953,   8.4666],
        [  2.6310, -10.9195,   3.8054],
        [  1.7659,   7.4127,   4.7871],
        [ -1.9600,   7.8577,   6.9121],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.7685,  -4.8098,  -2.9310],
        [  3.4009,  -4.1874,  -5.9751],
        [  0.9809,   7.6642,  -1.4217],
        [  4.4633,   5.6287,  -0.7754],
        [ -6.4464,   5.8272,   7.5038],
        [ -9.4283,  -1.9585,  -0.9129],
        [  1.4306,  13.9620,  -0.2075],
        [  1.1274,  14.0116,   8.1753],
        [  1.8721,  -4.7322,   0.5524],
        [  5.6022,  -1.9833,   8.3081],
        [  3.3839,  -4.1966,   8.3574],
        [  3.6154,  -4.0640,  -7.9698],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.1433,  -3.5003,  -3.0209],
        [  1.1350,  14.0104,   0.6321],
        [ -4.0868,   7.2981,   8.1874],
        [  2.3376,  -4.6231,   9.8436],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.8093,  -9.4178,   8.4244],
        [  0.7685,  -4.8098,  -2.9310],
        [  2.1625,  -4.6697,   7.0658],
        [ -7.8280,   4.2848,  -7.1955],
        [  8.0877,  -8.5297,  -1.1050],
        [ 11.0275,  -5.0666,   8.4154],
        [ -1.2454,   7.9127,  -0.5715],
        [  3.8111, -10.6787,  -4.6539],
        [  6.1730,  -0.5375,  -8.8861],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.4149,  -4.6005,  -4.3339],
        [ 10.0619,   8.6627,  -9.2884],
        [ -0.0254, -10.9932,   3.2200],
        [  9.6614,   9.1784,  -1.2087]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0457, 0.0593,  ..., 0.7354, 0.6428, 0.4430],
        [0.0457, 0.0000, 0.0157,  ..., 0.7490, 0.6211, 0.4835],
        [0.0593, 0.0157, 0.0000,  ..., 0.7616, 0.6074, 0.4990],
        ...,
        [0.7354, 0.7490, 0.7616,  ..., 0.0000, 1.3672, 0.5301],
        [0.6428, 0.6211, 0.6074,  ..., 1.3672, 0.0000, 1.0475],
        [0.4430, 0.4835, 0.4990,  ..., 0.5301, 1.0475, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.67939546,  7.8575648 ,  7.8788965 ,  8.01009083,  7.96575782,
        8.10886021, 10.75443106, 12.30695098,  7.13501264, 13.77222752,
        7.96181288, 11.00694688, 11.80092578, 13.66991027,  4.7663526 ,
       10.67132524,  5.64500512, 10.80728788,  6.50409372, 12.18998195,
        6.80559669,  7.83534987,  7.71075226,  8.08320235, 13.12299145,
       13.55868684,  6.02779201,  6.60752373,  5.43544626,  8.36445758,
        9.3109884 ,  9.17016857,  6.76056625, 11.77124373,  7.34250948,
       12.32408516, 11.36089355,  5.81199422,  9.94285511, 13.45124917,
        6.01356004,  7.53884104, 11.05317597,  9.97220799,  5.19065439,
        6.29821673, 13.58333857,  7.07902036, 12.11735334, 11.04629471,
        8.43626114,  7.88677538,  8.16584537, 11.76830981,  4.74468406,
       10.68031351,  7.09887557,  5.05444288,  5.07306274,  6.19707944,
        7.50917551, 11.4088043 ,  9.85330646,  4.75303127, 10.1371552 ,
        6.28835101, 12.7125079 ,  6.32572323, 11.4088043 , 11.785566  ,
       11.23201619,  7.62011734,  8.09842704, 10.26545792,  4.87083348,
        5.39450305,  7.72668467,  7.18354885,  8.68979052,  9.62958973,
       14.03505624, 14.05687815,  5.08910491,  5.94293185,  5.39098669,
        5.43943791,  9.75075273,  9.79039371, 14.05633441,  8.36445758,
        5.1805345 ,  4.85995002, 11.62158554,  4.87083348,  5.14607014,
        8.92395381, 11.75438006, 12.13568598,  8.01009083, 11.33837396,
        6.19633139, 12.40074901,  5.19577146, 13.27717982, 10.99326577,
       13.32613445])}
{'P': tensor([[1.9263e-02, 3.5287e-01, 2.3766e-01,  ..., 2.3970e-05, 4.3029e-06,
         4.2231e-05],
        [1.3706e-03, 3.0008e-02, 3.4784e-01,  ..., 3.6774e-05, 6.9236e-06,
         6.2646e-05],
        [2.1938e-03, 2.7801e-03, 4.5161e-02,  ..., 5.2921e-05, 1.1110e-05,
         8.6526e-05],
        ...,
        [8.8822e-04, 8.2892e-04, 7.2185e-04,  ..., 1.8692e-01, 4.0758e-04,
         3.8015e-02],
        [2.4436e-05, 2.3010e-05, 2.1061e-05,  ..., 4.4155e-04, 4.8522e-02,
         1.0549e-03],
        [2.4356e-02, 2.2221e-02, 1.8720e-02,  ..., 1.3307e-03, 5.6189e-05,
         1.6541e-01]]), 'X': tensor([[  1.1994,   7.6035,   2.4295],
        [  1.0093,   7.6567,   2.3526],
        [  0.6933,   7.7336,   2.1179],
        [ -1.9646,   7.8571,   3.3817],
        [ -1.6086,   7.8928,   4.3385],
        [ -0.5739,   7.9056,   0.6382],
        [  1.6308,  13.9254,  -2.0984],
        [ -5.0132,  -9.2234,  -8.3531],
        [ 11.6801,  -3.7196,  -0.9181],
        [  1.8766,   7.3694,   1.1887],
        [  6.3606,   0.9204,   3.2525],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.5101,  -2.1433,   9.5442],
        [ -8.9942,  -3.9990,   6.0448],
        [  4.6231,   5.4556,   8.4973],
        [  6.1984,  -0.4327,   1.5438],
        [  6.2576,   2.1359,  -3.3398],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -4.7642,  -9.3860,  -5.1133],
        [  0.2435,   7.8191,  -6.4709],
        [ 12.4901,  -0.7052,   1.7315],
        [ -7.3333,   4.9246,  -7.9212],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.0574,   1.7801,   8.8986],
        [  9.1409,  -7.5789,   1.5814],
        [  6.2535,  -0.1709,  -9.4102],
        [ -5.3736,   6.6310,  -1.5174],
        [ -8.6603,   2.8250,   5.9181],
        [  0.2571,  -4.7594,   1.5618],
        [ -5.9987,   6.1949,  -6.2114],
        [ -3.2486,   7.5925,  -1.3719],
        [  3.1271, -10.8343,   2.6442],
        [ 12.5919,   0.4035,  -5.8460],
        [ -6.4002,  -8.1048,  -0.9108],
        [  8.4691,  10.4571,  -5.0650],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.2437,  -4.6489,   6.1430],
        [ -4.6393,   7.0461,   1.1624],
        [  3.7121,   6.3117,   0.6738],
        [ -8.3062,  -5.6008,   0.0287],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.9003,  12.3550,   2.1917],
        [ -8.7735,   2.5640,  -6.3313],
        [ 11.6801,  -3.7196,  -0.9181],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.5650,   2.0132,   4.6272],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.0547,   7.2957,  -1.9343],
        [ -2.6069, -10.4298,  -6.1995],
        [  2.6310, -10.9195,   3.8054],
        [  6.3515,   0.6418,  -6.5823],
        [  5.4242, -10.1261,   3.9595],
        [ -5.6169,  -8.7844,   8.9057],
        [  5.4867,  12.5816,   2.9549],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.3062,  -5.6008,   0.0287],
        [ -3.5173, -10.0623,   4.8280],
        [  0.5960, -11.0335,   3.1144],
        [ -7.3446,   4.9113,  -5.9356],
        [  0.9659,  -4.8144,  -0.9594],
        [  4.2096,   5.8815,   1.9230],
        [  5.7784,  -1.6401,   9.5587],
        [  4.8051, -10.3703,   0.5268],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.3554,   4.8985,  -0.6665],
        [ -6.4464,   5.8272,   7.5038],
        [  5.3858,   4.4260,  -9.6769],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.9058,  -6.7203,   5.6928],
        [ -3.0004,   7.6607,  -7.2640],
        [  0.1206,  -4.7363,   8.7103],
        [  7.1697,  11.5343,  -0.7276],
        [ -9.3705,   0.4208,  -3.5815],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 11.4703,  -4.2006,   7.4086],
        [  3.1064,  13.5607,   6.7605],
        [  2.7558,  -4.4844,   0.0636],
        [ -7.9925,  -6.1423,   2.8990],
        [ -8.0829,  -5.9940,   5.0564],
        [ -4.4503,   7.1378,   3.3751],
        [  2.0547,   7.2957,  -1.9343],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.3589,  13.7668,  -3.6501],
        [  4.8653,   5.1694,  -4.0823],
        [  6.3534,   1.2990,  -7.2942],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 12.5224,  -0.4497,   4.8448],
        [  5.7151,  -1.7700,   3.1542],
        [  6.3589,   0.8239,  -8.4230],
        [  0.8700,  -4.8132,  -7.8042],
        [ 10.2470,   8.4054,  -0.1177],
        [  3.3943,   6.5483,   1.0790],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.3367,   7.1676,  -9.5263],
        [ -7.6525,  -6.6534,   5.1721],
        [ -6.3306,   5.9272,   5.6935],
        [  6.3497,   1.3680,  -1.9076],
        [  0.9445,   7.6736,   0.5095],
        [  6.6319,  11.9066,   3.5165],
        [ -0.8455, -10.8853,  -6.7400],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.2644,  -4.6434,   9.7508],
        [  0.6757, -11.0361,  -1.3047],
        [  6.2883,   0.0325,   5.2692]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0193, 0.0436,  ..., 0.3981, 0.5263, 0.3014],
        [0.0193, 0.0000, 0.0257,  ..., 0.4106, 0.5088, 0.3186],
        [0.0436, 0.0257, 0.0000,  ..., 0.4350, 0.4832, 0.3442],
        ...,
        [0.3981, 0.4106, 0.4350,  ..., 0.0000, 0.9006, 0.1653],
        [0.5263, 0.5088, 0.4832,  ..., 0.9006, 0.0000, 0.8273],
        [0.3014, 0.3186, 0.3442,  ..., 0.1653, 0.8273, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.69752262,  7.72291946,  7.76457095,  8.09899874,  8.0550286 ,
        7.92644746, 14.02058621, 10.4977132 , 12.25806559,  7.60463349,
        6.42689705, 12.17901033,  5.91222346,  9.84314712,  7.15100303,
        6.2134961 ,  6.61211224, 12.42554808, 10.52587243,  7.822854  ,
       12.50996828,  8.83341608, 12.19805394,  9.23071698, 11.87412439,
        6.25586297,  8.53501777,  9.10945678,  4.7663526 ,  8.62329466,
        8.25828409, 11.27656488, 12.59840193, 10.32715715, 13.45642379,
       13.01728313,  5.16205087,  8.43626114,  7.32233733, 10.01804223,
       12.41012046, 13.69162795,  9.14044984, 12.25806559,  6.38968357,
       12.72524615,  5.30158497,  7.57946459, 10.7506423 , 11.23201619,
        6.383897  , 11.48734978, 10.42667093, 13.72594022,  5.53255882,
       10.01804223, 10.65929968, 11.04953767,  8.8353674 ,  4.91038884,
        7.23276582,  6.00663314, 11.42947632, 12.85032836,  8.83725081,
        8.68979052,  6.97106912, 12.10222204, 11.9702762 ,  8.22728399,
        4.73784222, 13.58101794,  9.37989631, 12.07818871,  9.88310614,
       12.21532282, 13.91198191,  5.26342203, 10.08001894, 10.06286074,
        8.41148722,  7.57946459, 13.27994304,  9.93553424, 13.96746816,
        7.09887557,  6.48486171, 10.14436912, 12.53047489,  5.98285319,
        6.41202986,  4.89121805, 13.25336001,  7.37576154,  8.62329466,
        7.53884104, 10.14044489,  8.67227386,  6.49538841,  7.73151309,
       13.62896593, 10.91805683, 11.9120321 ,  5.16611782, 11.05672136,
        6.28835101])}
{'P': tensor([[2.5359e-02, 1.6369e-01, 1.5460e-01,  ..., 2.3566e-04, 2.6299e-06,
         6.2377e-06],
        [2.7152e-04, 2.1297e-02, 5.2022e-01,  ..., 5.3744e-05, 1.4917e-06,
         1.4636e-06],
        [6.7541e-04, 1.9977e-03, 5.9049e-02,  ..., 1.3996e-04, 4.0901e-06,
         3.7991e-06],
        ...,
        [1.2043e-03, 1.4021e-03, 1.4665e-03,  ..., 7.7057e-02, 5.9601e-06,
         5.7586e-05],
        [1.0822e-04, 2.5196e-05, 2.2874e-05,  ..., 6.5155e-05, 9.7926e-03,
         2.2856e-03],
        [2.7603e-03, 1.0058e-03, 9.5408e-04,  ..., 3.6403e-02, 6.3472e-06,
         9.0819e-02]]), 'X': tensor([[  1.1139,   7.6281,  -2.5680],
        [ -1.7290,   7.8825,  -3.3831],
        [ -1.9487,   7.8590,  -3.3330],
        [ -1.9714,   7.8563,  -1.8084],
        [ -2.3317,   7.8034,  -3.6274],
        [ -0.1963,   7.8765,  -1.1371],
        [ -9.1706,   1.3863,   8.4032],
        [  2.7121,   6.9743,  -8.4202],
        [ -1.5451,   7.8975,   7.6104],
        [ -5.9940,   6.1985,   5.9566],
        [  6.1575,  12.2046,   3.3257],
        [ -9.3616,  -2.4749,   0.1793],
        [ -3.6381,   7.4683,   7.3544],
        [  8.5198,  10.4091,  -8.6486],
        [  0.6470, -11.0352,  -1.1931],
        [ 12.1504,   4.3674,  -7.2998],
        [  1.7500,   7.4187,   4.4237],
        [  6.3460,   1.4252,   1.2291],
        [ 10.1798,   8.5002,  -5.8893],
        [ -6.3306,   5.9272,   5.6935],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3236,   0.3011,  -4.6668],
        [  6.9956,  11.6590,  -7.6876],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.3367,  -4.4806,   0.4643],
        [ -9.4645,  -1.4880,  -2.0226],
        [  7.1821,  -9.1845,   9.8548],
        [ -2.6069, -10.4298,  -6.1995],
        [  5.0348,  -2.8229,   3.5436],
        [ 11.3367,  -4.4806,   0.4643],
        [ -3.1671,   7.6158,   6.0765],
        [  3.4140,  -4.1803,  -8.8224],
        [ 11.9019,  -3.1379,  -3.4534],
        [  4.7482, -10.3907,   5.9149],
        [ -8.0829,  -5.9940,   5.0564],
        [  5.8902,  -1.3887,   4.0388],
        [  9.9756,  -6.6330,  -3.3429],
        [  3.6154,  -4.0640,  -7.9698],
        [ -5.8389,  -8.6056,  -3.5098],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.4167, -10.1293,   6.2156],
        [  6.1222,  -0.7261,   8.8204],
        [  4.2339,  -3.6268,   6.3125],
        [ -7.4807,   4.7463,   2.0616],
        [  4.7813,  -3.1139,   8.2691],
        [  1.9503,  -4.7172,  -6.2313],
        [ -0.1963,   7.8765,  -1.1371],
        [ -5.9987,   6.1949,  -6.2114],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 10.0995,  -6.4739,   2.0974],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.5294,  -2.1106,   9.0626],
        [  2.3261,  13.7749,   5.9121],
        [  2.5382,   7.0672,   4.5097],
        [  2.9470,  13.6085,   2.9809],
        [  1.8320,   7.3871,   5.1452],
        [  6.3266,   0.3292,   9.3502],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.9433,  -9.3362,  -0.3421],
        [  0.9945,  -4.8144,  -0.1897],
        [  2.0449,  13.8402,  -7.1642],
        [  9.7328,   9.0902,  -0.8917],
        [  0.1041, -11.0045,   2.4311],
        [  0.1931,  -4.7491,   7.4385],
        [  0.2571,  -4.7594,   1.5618],
        [ -9.4541,  -0.3327,   8.8246],
        [  1.8447,  -4.7372,   8.1235],
        [  2.6078, -10.9229,   6.2869],
        [ -6.3252,  -8.1760,  -1.0891],
        [  9.5261,  -7.1675,   1.4206],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.7735,   2.5640,  -6.3313],
        [ -1.3178, -10.7939,   0.4236],
        [ -9.1749,  -3.3803,   6.4629],
        [  3.7298,  13.3537,  -9.4653],
        [  2.8602,   6.8905,   5.8779],
        [  0.9809,   7.6642,  -1.4217],
        [  6.3246,   1.6696,  -0.1544],
        [  7.9816,  10.8938,  -0.5378],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -0.0282,   7.8575,  -1.1746],
        [ 12.1379,   4.4125,  -3.8833],
        [  2.7558,  -4.4844,   0.0636],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -8.6681,  -4.8556,   9.2518],
        [  7.2340,  -9.1505,  -5.8321],
        [  9.3624,   9.5318,   0.2779],
        [ 12.3357,   3.6001,   0.7903],
        [  6.0955,   2.8210,  -4.3321],
        [  2.5813,  -4.5471,   5.3597],
        [  9.0424,  -7.6779,  -7.5286],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.5428,  -9.5728,  -2.9897],
        [  1.1139,   7.6281,  -2.5680],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.1890,   4.7318,   0.9515],
        [ 12.4837,   2.7510,   0.4449],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.9241,   7.3503,   7.9004],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.4495,   6.3026,   6.8257]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1020, 0.1197,  ..., 0.6958, 0.1830, 0.9664],
        [0.1020, 0.0000, 0.0235,  ..., 0.7623, 0.2814, 1.0609],
        [0.1197, 0.0235, 0.0000,  ..., 0.7620, 0.2947, 1.0705],
        ...,
        [0.6958, 0.7623, 0.7620,  ..., 0.0000, 0.5524, 0.4932],
        [0.1830, 0.2814, 0.2947,  ..., 0.5524, 0.0000, 0.7841],
        [0.9664, 1.0609, 1.0705,  ..., 0.4932, 0.7841, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.70897675,  8.06990483,  8.09704052,  8.09984231,  8.14433542,
        7.8788965 ,  9.27474386,  7.48311002,  8.04718433,  8.62260327,
       13.66991027,  9.68323686,  8.30729352, 13.45124917, 11.05413545,
       12.91143491,  7.62232307,  6.50409372, 13.26209545,  8.67227386,
        6.12886024,  6.33075682, 13.59673406, 13.02818702, 12.18998195,
        9.58072286, 11.65923396, 10.7506423 ,  5.77219077, 12.18998195,
        8.24808832,  5.39723241, 12.30858677, 11.42420508, 10.06286074,
        6.05165248, 11.97957425,  5.43943791, 10.39942497, 11.72749923,
        9.97964013, 11.48663767,  6.16512856,  5.5748624 ,  8.85939626,
        5.70591986,  5.10444097,  7.8788965 ,  8.62329466, 10.30972193,
       10.76154256,  9.70108409, 11.99633819, 12.32408516, 13.09426142,
        5.91854397, 13.96987806,  7.50917551, 13.92390689,  7.61088125,
        6.33517349, 13.26590071, 11.63502679,  4.9160962 , 13.99048094,
       13.31764181, 11.0050379 ,  4.75303127,  4.7663526 ,  9.45995955,
        5.08372149, 11.22993218, 10.33712258, 11.92133129, 12.18045978,
       13.25336001,  9.14044984, 10.87409005,  9.77777346, 13.86480175,
        7.46053281,  7.72668467,  6.54127431, 13.50484539, 12.35421461,
        7.8575648 , 12.91505111,  5.26342203, 12.05395056,  9.9353921 ,
       11.66453059, 13.3607379 , 12.85032836,  6.7166297 ,  5.2287058 ,
       11.86239961, 10.42667093, 11.59512194,  7.70897675, 13.14777433,
        7.02253235, 12.78327037, 10.91805683,  7.59795445, 12.1494023 ,
       13.06957425])}
{'P': tensor([[2.1464e-02, 1.3463e-01, 1.3419e-01,  ..., 1.0491e-06, 4.1616e-05,
         1.0975e-04],
        [1.7325e-04, 1.3317e-02, 1.1419e-02,  ..., 5.5688e-07, 1.7559e-05,
         1.8984e-04],
        [2.1380e-04, 5.7961e-03, 1.8993e-02,  ..., 1.0576e-06, 2.2247e-05,
         3.6215e-04],
        ...,
        [3.7625e-05, 1.0608e-05, 1.3536e-05,  ..., 4.6117e-02, 3.6193e-04,
         1.6579e-06],
        [3.6762e-03, 1.3397e-03, 1.0274e-03,  ..., 2.0972e-05, 3.8825e-02,
         1.5651e-05],
        [4.5994e-06, 4.1220e-06, 4.5538e-06,  ..., 1.5766e-04, 6.0305e-05,
         2.0134e-02]]), 'X': tensor([[  1.1006,   7.6318,  -2.4877],
        [ -1.7290,   7.8825,  -3.3831],
        [ -1.9714,   7.8563,  -1.8084],
        [ -1.9487,   7.8590,  -3.3330],
        [ -2.3317,   7.8034,  -3.6274],
        [ -0.1963,   7.8765,  -1.1371],
        [ -8.0087,   4.0173,   9.1091],
        [  6.6319,  11.9066,   3.5165],
        [ -4.0409,   7.3169,  -3.5779],
        [ -9.2687,  -2.9794,   0.6436],
        [  9.9217,  -6.7006,  -5.6728],
        [  6.3048,   0.1469,  -9.1691],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.0449,  13.8402,  -7.1642],
        [  6.0345,   3.0175,  -1.4424],
        [  5.0348,  -2.8229,   3.5436],
        [  5.8318,   3.5567,  -4.7692],
        [  0.4996,   7.7738,  -4.4711],
        [ -4.9217,   6.8979,  -1.7667],
        [ -3.5029,   7.5138,   8.1853],
        [ 12.5650,   2.0132,   4.6272],
        [ -0.8569,   7.9156,  -6.5557],
        [ -9.1231,  -3.5739,   8.4708],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.3928,   7.1404,  -1.7035],
        [  1.2273,  -4.8081,   9.8129],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.7121,   6.9743,  -8.4202],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -7.5720,   4.6310,   2.1717],
        [ -8.3539,  -5.5111,   0.3373],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.8890,  -4.4316,   1.1600],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.6363, -11.0348,   6.1414],
        [  6.1741,  -0.5329,   3.1913],
        [  0.9074,  -4.8139,   8.1674],
        [  6.3246,   1.6696,  -0.1544],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.6314,  -1.9302,  -9.6543],
        [  0.4996,   7.7738,  -4.4711],
        [  6.5428,  -9.5728,  -2.9897],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 10.0567,   8.6697,   4.6329],
        [  3.3943,   6.5483,   1.0790],
        [ 11.8951,  -3.1570,   7.0742],
        [ -9.4283,  -1.9585,  -0.9129],
        [  1.9204,   7.3518,   8.4781],
        [  1.8719,   7.3713,  -3.7465],
        [  7.5064,  -8.9651,  -0.0994],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4157,   0.0821,  -5.0025],
        [ -8.0608,   3.9362,  -5.9567],
        [  0.9478,  -4.8144,   0.6371],
        [  6.3122,  -9.6994,  -6.3282],
        [ -0.0410,   7.8591,   9.3457],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.6801,   2.7809,  -0.8443],
        [ -3.9858,  -9.8334,  -3.7323],
        [  0.9392,  -4.8143,  -3.4649],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.1999,  -3.6543,  -4.1998],
        [  0.6396,  14.0772,  -6.3786],
        [  6.9284,  11.7061,  -0.3732],
        [  5.0233,   4.9647,  -3.1788],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.1382, -11.0073,   1.0026],
        [  2.2849, -10.9655,  -7.6542],
        [  6.3236,   0.3011,  -4.6668],
        [  0.3354,  -4.7708,  -4.9125],
        [ -6.7643,  -7.7372,  -9.6663],
        [  7.5930,  -8.9037,   7.2323],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.0547,   7.2957,  -1.9343],
        [  5.9587,  12.3216,   9.1695],
        [ -7.8126,   4.3066,   5.1791],
        [ -7.6293,  -6.6859,  -5.5161],
        [  2.3200,   7.1756,   1.1938],
        [  2.1771, -10.9776,  -3.2007],
        [ -5.9940,   6.1985,   5.9566],
        [  6.3497,   1.3680,  -1.9076],
        [  6.0890,  -0.8382,   2.9166],
        [ 11.3691,   6.4760,   1.0787],
        [ -3.2486,   7.5925,  -1.3719],
        [ -4.8751,  -9.3148,   4.0406],
        [  1.1139,   7.6281,  -2.5680],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.8934,   2.2597,   3.5831],
        [  2.8890,  -4.4316,   1.1600],
        [ 11.7796,  -3.4696,   9.0088],
        [ -4.9217,   6.8979,  -1.7667],
        [ -4.2271,   7.2387,  -1.8316],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.0093,   7.6567,   2.3526],
        [  1.0000,   7.6591,  -4.5011],
        [  6.2975,   0.0947,   5.1791],
        [  2.2373,  -4.6506,  -1.7334],
        [  7.5930,  -8.9037,   7.2323],
        [  6.0560,  -0.9421,  -5.0334],
        [ -8.5859,  -5.0393,   0.5476]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1049, 0.1054,  ..., 0.7873, 0.2489, 0.7764],
        [0.1049, 0.0000, 0.1069,  ..., 0.8438, 0.2966, 0.7200],
        [0.1054, 0.1069, 0.0000,  ..., 0.7374, 0.3526, 0.6719],
        ...,
        [0.7873, 0.8438, 0.7374,  ..., 0.0000, 0.9763, 0.8436],
        [0.2489, 0.2966, 0.3526,  ..., 0.9763, 0.0000, 1.0149],
        [0.7764, 0.7200, 0.6719,  ..., 0.8436, 1.0149, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.71075226,  8.06990483,  8.09984231,  8.09704052,  8.14433542,
        7.8788965 ,  8.95983586, 13.62896593,  8.35856487,  9.73579738,
       11.97238624,  6.30647746, 11.99633819, 13.99048094,  6.74686071,
        5.77219077,  6.83083863,  7.78980299,  8.47371868,  8.29022197,
       12.72524615,  7.96181288,  9.79814997,  8.89458273,  7.53063885,
        4.96231532,  7.9337647 ,  7.48311002, 10.37698435,  8.87587457,
       10.00794468, 10.71162681,  5.29009836, 12.3828957 ,  9.36093562,
       11.05317597,  6.19707944,  4.89869205,  6.54127431, 12.32408516,
        6.14086582,  5.95297459,  7.78980299, 11.59512194, 12.49317563,
       13.2778378 ,  7.37576154, 12.30695098,  9.62958973,  7.5984809 ,
        7.60529892, 11.69264243,  9.41244865,  9.41605684,  8.97052115,
        4.90676242, 11.57249984,  7.85919687,  6.35029274,  9.11473391,
       10.61047596,  4.90505566, 13.04217487, 13.00413829,  5.5671389 ,
       14.09176259, 13.60275211,  7.06272256, 10.78533838, 11.00812996,
       11.20100795,  6.33075682,  4.78258092, 10.27716434, 11.70168755,
        9.49511524,  7.57946459, 13.68672855,  8.92098752, 10.14436912,
        7.54127425, 11.1913598 ,  8.62260327,  6.49538841,  6.1463896 ,
       13.08414689,  8.25828409, 10.51340829,  7.70897675, 10.48721382,
        6.383897  ,  9.17595899,  5.29009836, 12.27993098,  8.47371868,
        8.38252937, 13.0743399 , 12.4013153 ,  6.27905292,  7.72291946,
        7.72415048,  6.29821673,  5.16077764, 11.70168755,  6.12886024,
        9.95552626])}
{'P': tensor([[2.8812e-02, 3.3398e-01, 1.6386e-01,  ..., 1.3434e-04, 2.2636e-04,
         4.0070e-05],
        [2.6716e-03, 4.3399e-02, 2.3961e-01,  ..., 2.4019e-04, 3.2839e-04,
         5.4367e-05],
        [8.0893e-04, 9.9231e-04, 6.5317e-02,  ..., 6.9306e-04, 1.2283e-04,
         1.5134e-05],
        ...,
        [6.2501e-06, 6.4492e-06, 6.1205e-06,  ..., 3.8255e-02, 1.0641e-05,
         5.8368e-07],
        [3.9446e-03, 4.7319e-03, 3.5915e-03,  ..., 9.1290e-04, 1.7625e-01,
         6.5272e-03],
        [4.5529e-04, 5.5244e-04, 4.5867e-04,  ..., 2.1151e-05, 5.1811e-03,
         4.0569e-02]]), 'X': tensor([[  1.0093,   7.6567,   2.3526],
        [  0.6933,   7.7336,   2.1179],
        [ -1.9646,   7.8571,   3.3817],
        [ -0.5739,   7.9056,   0.6382],
        [ -1.6086,   7.8928,   4.3385],
        [ -1.6220,   7.8917,   0.0315],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.3062,  -5.6008,   0.0287],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.5382,   7.0672,   4.5097],
        [ -6.3306,   5.9272,   5.6935],
        [  2.7976,  -4.4682,  -4.2020],
        [ -9.0604,   1.7707,   1.1066],
        [  7.8425,  -8.7202,   6.4530],
        [  3.9515,  -3.8425,   6.1729],
        [  1.9204,   7.3518,   8.4781],
        [  6.3303,   1.6145,  -8.4083],
        [ -1.8367,   7.8718,  -8.6523],
        [  4.7482, -10.3907,   5.9149],
        [  4.0362,  13.2397,  -7.0030],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.9033,   2.2329,  -8.8912],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.9526,   3.2525,  -4.3821],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.0750,  -6.0072,  -5.8267],
        [  4.5340,  -3.3629,   8.9652],
        [  6.0890,  -0.8382,   2.9166],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.8930,  12.3592,  -8.9201],
        [  1.6308,  13.9254,  -2.0984],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.4732,  -2.2041,  -4.3830],
        [ -6.4464,   5.8272,   7.5038],
        [  3.4140,  -4.1803,  -8.8224],
        [  6.7792,  11.8084,   1.6251],
        [ -6.0195,   6.1789,  -2.9496],
        [ 11.5198,  -4.0923,   5.5088],
        [  9.3753,   9.5171,   3.9131],
        [ -7.4887,   4.7364,   8.9074],
        [  4.8674,  12.8875,   6.7622],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -6.6895,   5.6051,   2.7899],
        [  5.0617,  12.7957,   8.3544],
        [  2.0729,   7.2878,   7.0880],
        [  3.6249, -10.7255,   5.6605],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.4297,  -5.3637,  -9.3962],
        [  4.4633,   5.6287,  -0.7754],
        [  4.8674,  12.8875,   6.7622],
        [ -9.1178,   1.5784,   2.1410],
        [  0.5960, -11.0335,   3.1144],
        [  8.8091,  -7.9037,   0.3631],
        [  0.3038,   7.8092,   0.4164],
        [  0.1252, -11.0062,   7.2564],
        [ -0.1626, -10.9796,   6.4629],
        [  5.5626,  -2.0536,   8.4433],
        [  4.5767,   5.5071,   3.9069],
        [ -9.3464,   0.5708,  -1.6397],
        [ -6.3066,  -8.1936,   9.5822],
        [ 12.5258,  -0.4200,  -9.3493],
        [  7.1441,  11.5529,   2.3232],
        [  6.2036,   2.4030,  -1.2500],
        [ 11.0373,   7.1299,  -8.5895],
        [ -9.2373,  -3.1225,  -8.9763],
        [  8.9284,  10.0036,  -6.9210],
        [ -6.6998,   5.5954,   0.0823],
        [  2.8474,   6.8979,   6.6457],
        [ -9.3616,  -2.4749,   0.1793],
        [ 12.6038,   0.7665,  -3.4439],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.4501,  -1.7133,   2.2980],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -6.6895,   5.6051,   2.7899],
        [ 12.0582,  -2.6640,   0.7132],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 11.0275,  -5.0666,   8.4154],
        [  0.3231,  -4.7691,  -1.6120],
        [ -7.7842,   4.3466,  -1.7548],
        [ -9.3331,  -2.6465,  -8.2028],
        [  8.7131,  10.2216,   7.7915],
        [  6.3124,   0.2057,  -0.4593],
        [  3.1892, -10.8220,  -6.8409],
        [  6.8704,  -9.3809,   7.5769],
        [  6.3360,   0.4258,  -9.2509],
        [  5.8617,   3.4853,  -9.8080],
        [  5.1292,   4.8183,  -5.7997],
        [  1.1006,   7.6318,  -2.4877],
        [  5.9655,   3.2172,  -7.5953],
        [  3.3744,   6.5623,  -6.3636],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.0391,  -4.8141,  -0.0768],
        [ -9.4645,  -1.4880,  -2.0226],
        [  0.9478,  -4.8144,   0.6371],
        [  3.9515,  -3.8425,   6.1729],
        [ -9.2508,   1.0572,   2.9978],
        [  1.7659,   7.4127,   4.7871],
        [  5.6258,  -1.9404,  -6.0715],
        [  0.2571,  -4.7594,   1.5618],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.5974,   7.0363,  -7.7810],
        [  5.8930,  12.3592,  -8.9201]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0257, 0.1463,  ..., 0.9675, 0.6001, 0.9368],
        [0.0257, 0.0000, 0.1458,  ..., 0.9419, 0.5853, 0.9324],
        [0.1463, 0.1458, 0.0000,  ..., 0.9492, 0.7147, 1.0775],
        ...,
        [0.9675, 0.9419, 0.9492,  ..., 0.0000, 0.8344, 1.2791],
        [0.6001, 0.5853, 0.7147,  ..., 0.8344, 0.0000, 0.4720],
        [0.9368, 0.9324, 1.0775,  ..., 1.2791, 0.4720, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.72291946,  7.76457095,  8.09899874,  7.92644746,  8.0550286 ,
        8.05668609,  8.19703512, 10.01804223,  7.89745107,  9.48969978,
        7.50917551,  8.67227386,  5.27178363,  9.23178204, 11.72802491,
        5.51177501,  7.5984809 ,  6.53290663,  8.08320235, 11.42420508,
       13.84126197, 10.1371552 ,  9.17905491, 12.52875793, 12.46513646,
        6.78325151,  7.26074925, 10.06437598,  5.64500512,  6.1463896 ,
        6.14086582, 13.6922425 , 14.02058621,  9.15866607,  5.90035324,
        8.68979052,  5.39723241, 13.61600536,  8.62631801, 12.22503754,
       13.35927527,  8.86082124, 13.77604293,  9.69792513,  8.7273569 ,
       13.76047999,  7.57685908, 11.32149259,  6.01356004,  9.99145096,
        7.18354885, 13.77604293,  9.25336371, 11.04953767, 11.83509372,
        7.81509631, 11.00694688, 10.98076873,  5.92952538,  7.16058587,
        9.36378438, 10.33958639, 12.53285633, 13.58333857,  6.652742  ,
       13.1399181 ,  9.75075273, 13.40850328,  8.72896381,  7.46248947,
        9.68323686, 12.62711252, 10.80728788,  9.60412735, 10.06407839,
        8.7273569 , 12.3489324 , 12.4217523 , 12.13568598,  4.78002665,
        8.91551231,  9.70108409, 13.43127131,  6.31576264, 11.28215429,
       11.62770332,  6.35029274,  6.81961996,  7.03733775,  7.71075226,
        6.77777606,  7.37902766, 10.86693708, 10.04916678,  4.92497514,
        9.58072286,  4.90676242,  5.51177501,  9.3109884 ,  7.62011734,
        5.9510489 ,  4.7663526 , 12.59969936,  9.78286733,  7.50035355,
       13.6922425 ])}
{'P': tensor([[1.7026e-02, 3.4639e-01, 1.1335e-01,  ..., 4.5513e-04, 4.9327e-05,
         5.7313e-07],
        [7.3203e-04, 2.2955e-02, 1.9471e-01,  ..., 4.0687e-04, 6.9252e-05,
         7.8288e-07],
        [4.3506e-04, 5.2102e-04, 4.2366e-02,  ..., 9.0951e-05, 1.6414e-04,
         1.6663e-06],
        ...,
        [1.2366e-02, 1.7770e-02, 6.1217e-02,  ..., 2.0909e-01, 2.3609e-05,
         2.1557e-07],
        [4.5387e-04, 4.0667e-04, 2.3684e-04,  ..., 2.2799e-05, 9.6803e-02,
         2.7409e-03],
        [4.3491e-06, 3.6723e-06, 1.5002e-06,  ..., 3.5583e-07, 7.6822e-05,
         1.6975e-02]]), 'X': tensor([[  1.0000,   7.6591,  -4.5011],
        [  0.4996,   7.7738,  -4.4711],
        [ -2.1817,   7.8274,  -4.6298],
        [ -1.7290,   7.8825,  -3.3831],
        [ -2.3317,   7.8034,  -3.6274],
        [ -1.9487,   7.8590,  -3.3330],
        [ -7.6734,   4.4980,  -5.4248],
        [  7.5309,  11.2617,   7.2656],
        [ 11.6165,  -3.8715,   6.4594],
        [  0.1287, -11.0065,   6.5757],
        [  1.9204,   7.3518,   8.4781],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.4753,  -0.8012,  -5.2050],
        [  1.6408,   7.4591,  -4.2294],
        [ -8.3653,   3.4184,   9.3454],
        [  4.2096,   5.8815,   1.9230],
        [  5.1886,  -2.6250,   6.4677],
        [  0.3354,  -4.7708,  -4.9125],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.1905,   3.7253,   4.1175],
        [ -1.8823, -10.6558,  -3.8952],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.3413,   0.4883,  -6.8416],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.3928,   7.1404,  -1.7035],
        [  3.1480,  -4.3165,  -7.8002],
        [ 12.4691,   2.8525,  -5.5110],
        [  1.4306,  13.9620,  -0.2075],
        [  4.6063,  13.0051,  -3.1456],
        [ -6.0195,   6.1789,  -2.9496],
        [ -7.3470,  -7.0615,   5.0066],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.8940, -11.0402,  -0.9783],
        [  2.2644,  -4.6434,   9.7508],
        [ 11.9808,   4.9340,  -3.3715],
        [ -3.0644,   7.6439,  -1.4322],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.3569,   0.7591,   6.5011],
        [ -3.0004,   7.6607,  -7.2640],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -5.6174,   6.4708,  -1.7339],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 11.5568,   6.0602,   9.3244],
        [  6.2883,   0.0325,   5.2692],
        [  7.1821,  -9.1845,   9.8548],
        [ -3.0644,   7.6439,  -1.4322],
        [ -8.6112,   2.9319,  -7.9943],
        [ -4.8751,  -9.3148,   4.0406],
        [ -9.4505,  -1.7081,  -0.7039],
        [  2.5128, -10.9365,   9.9645],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -4.2271,   7.2387,  -1.8316],
        [  2.4639, -10.9431,  -7.3718],
        [  6.2038,   2.4020,  -3.1692],
        [  0.3493,  -4.7727,  -6.5682],
        [ -7.5720,   4.6310,   2.1717],
        [  2.1477,   7.2550,   8.9688],
        [ -5.2589,   6.7024,  -5.5809],
        [ -1.6275, -10.7221,   3.2376],
        [ -1.5408,   7.8978,   5.3470],
        [  3.3474,  -4.2162,   5.2455],
        [  5.8902,  -1.3887,   4.0388],
        [  1.6308,  13.9254,  -2.0984],
        [ -1.5875,   7.8944,  -9.4286],
        [  0.5960, -11.0335,   3.1144],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.0965,  -5.9712,   4.3466],
        [  4.5430,   5.5438,  -3.8669],
        [ -8.6603,   2.8250,   5.9181],
        [  8.2422,  -8.4041,  -1.4962],
        [ -2.7572,   7.7194,   5.9580],
        [ -6.7745,  -7.7264,   8.4684],
        [ -5.3736,   6.6310,  -1.5174],
        [  6.1372,   2.6726,  -6.7885],
        [ -6.3489,   5.9117,   6.5161],
        [  1.0093,   7.6567,   2.3526],
        [  7.0358,  -9.2784,  -9.8479],
        [ -3.5520,   7.4976,   9.1522],
        [  6.3360,   0.4258,  -9.2509],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.1905,   3.7253,   4.1175],
        [ -1.6275, -10.7221,   3.2376],
        [  4.1348,   5.9514,  -0.8856],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.0000,   7.6591,  -4.5011],
        [  5.3860,  -2.3418,   1.0826],
        [  5.6314,  -1.9302,  -9.6543],
        [  7.5930,  -8.9037,   7.2323],
        [  3.6817,   6.3354,   6.1233],
        [ -7.4808,   4.7462,   2.0266],
        [  3.4920,  -4.1366,  -8.3955],
        [  0.3724,   7.7974,  -6.8722],
        [  0.5600, -11.0321,  -0.8660],
        [  3.2809,  13.5061,  -4.0199],
        [ -7.5603,   4.6459,   8.2002],
        [  2.8890,  -4.4316,   1.1600],
        [  4.3523, -10.5232,   8.2138],
        [  1.3339,   7.5627,  -0.9034],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.7295,  13.6703,  -8.9328],
        [  0.3715,  -4.7756,   0.2312],
        [  5.6414, -10.0303,   7.8525]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0295, 0.1423,  ..., 0.4741, 0.2865, 0.9706],
        [0.0295, 0.0000, 0.1138,  ..., 0.4997, 0.2887, 0.9583],
        [0.1423, 0.1138, 0.0000,  ..., 0.5859, 0.3497, 0.9496],
        ...,
        [0.4741, 0.4997, 0.5859,  ..., 0.0000, 0.6624, 1.3890],
        [0.2865, 0.2887, 0.3497,  ..., 0.6624, 0.0000, 0.7267],
        [0.9706, 0.9583, 0.9496,  ..., 1.3890, 0.7267, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.72415048,  7.78980299,  8.12580976,  8.06990483,  8.14433542,
        8.09704052,  8.89458273, 13.54774355, 12.24467041, 11.00726483,
        7.5984809 , 12.4980845 ,  9.50913849,  7.6374603 ,  9.0368458 ,
        7.23276582,  5.81481044,  4.78258092,  5.5671389 ,  8.99791316,
       10.8207297 ,  9.86089579,  6.36004393,  6.14086582,  7.53063885,
        5.34250225, 12.79126785, 14.03505624, 13.79676654,  8.62631801,
       10.19036081,  5.01114719, 11.07637531,  5.16611782, 12.95702647,
        8.23525433,  5.63824971,  6.4020404 ,  8.22728399, 10.794755  ,
        9.5642993 ,  8.56889738, 10.00410625, 13.0493368 ,  6.28835101,
       11.65923396,  8.23525433,  9.09661212, 10.51340829,  9.60359158,
       11.22142195, 12.10942412,  9.66977802,  8.38252937, 11.21703654,
        6.65259775,  4.78544875,  8.87587457,  7.56617037,  8.51928531,
       10.84493732,  8.04664886,  5.38342077,  6.05165248, 14.02058621,
        8.05242591, 11.04953767,  8.04718433, 10.06023665,  7.16747878,
        9.10945678, 11.77124373,  8.19703512, 10.27572673,  8.53501777,
        6.69389431,  8.67502348,  7.72291946, 11.64436684,  8.29640613,
        6.35029274, 12.10222204,  6.2322769 ,  8.99791316, 10.84493732,
        7.24677664,  9.75401449,  7.72415048,  5.87305005,  5.95297459,
       11.70168755,  7.32754882,  8.859416  ,  5.41348121,  7.80625981,
       11.04629471, 13.89886469,  8.8737556 ,  5.29009836, 11.38774149,
        7.67939546,  5.23878371, 12.17901033, 13.94008948,  4.79003361,
       11.50791859])}
{'P': tensor([[1.0581e-02, 1.5899e-01, 1.4841e-01,  ..., 1.3647e-03, 2.5602e-05,
         1.0585e-06],
        [2.8326e-04, 1.0355e-02, 2.3756e-01,  ..., 1.1024e-03, 3.7804e-05,
         1.0330e-06],
        [4.7936e-04, 7.1273e-04, 1.8918e-02,  ..., 1.9339e-03, 7.3974e-05,
         1.8937e-06],
        ...,
        [1.1260e-03, 1.6713e-03, 1.7881e-03,  ..., 2.3427e-01, 2.3150e-04,
         4.2256e-05],
        [1.5200e-05, 1.5168e-05, 1.5205e-05,  ..., 3.7349e-06, 1.1094e-01,
         2.6787e-04],
        [6.1172e-05, 3.9014e-05, 3.6330e-05,  ..., 1.4456e-05, 9.8452e-07,
         5.0810e-02]]), 'X': tensor([[  0.9809,   7.6642,  -1.4217],
        [ -0.0282,   7.8575,  -1.1746],
        [ -0.1963,   7.8765,  -1.1371],
        [ -1.2454,   7.9127,  -0.5715],
        [ -1.9714,   7.8563,  -1.8084],
        [ -2.0445,   7.8469,  -0.5857],
        [ 12.5770,   1.8512,  -0.6578],
        [ -4.0409,   7.3169,  -3.5779],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.5294,  -2.1106,   9.0626],
        [  5.0617,  12.7957,   8.3544],
        [  2.2182,  13.8007,   5.1889],
        [ -5.0943,  -9.1682,   0.5073],
        [  0.3493,  -4.7727,  -6.5682],
        [  6.0305,   3.0296,   0.1375],
        [  0.6396,  14.0772,  -6.3786],
        [ -9.1960,  -3.2964,   1.6921],
        [ -5.9987,   6.1949,  -6.2114],
        [  6.0310,  12.2795,  -6.4528],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.5318,  13.7231,   6.3692],
        [ -5.3736,   6.6310,  -1.5174],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.0277,   4.9589,   7.0175],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.0362,  13.2397,  -7.0030],
        [ -0.0282,   7.8575,  -1.1746],
        [  7.3592,  11.3937,   6.2407],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.7482, -10.3907,   5.9149],
        [  1.5662,  13.9375,   4.5142],
        [  2.7685, -10.8982,  -6.8928],
        [  4.7482, -10.3907,   5.9149],
        [ 10.9931,   7.2107,   7.7933],
        [  6.1575,  12.2046,   3.3257],
        [  6.2365,  -0.2578,  -6.0212],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 12.4996,   2.6335,  -6.3166],
        [  2.6321,  -4.5296,  -2.5159],
        [ -3.2736, -10.1703,   3.8712],
        [  1.5383, -11.0274,  -4.9666],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.0547,   7.2957,  -1.9343],
        [  5.5158,  -2.1337,   3.5246],
        [  9.0004,  -7.7195,   4.6945],
        [  0.3493,  -4.7727,  -6.5682],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.0047,   3.1062,  -6.7914],
        [  1.7903,  -4.7467,   0.1168],
        [  1.2728, -11.0372,   9.8121],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 12.4013,  -1.2597,  -7.6474],
        [  4.5814, -10.4485,   2.5496],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.8154,   2.4611,   2.5198],
        [ -3.6381,   7.4683,   7.3544],
        [  1.9503,  -4.7172,  -6.2313],
        [ -4.3963,   7.1629,   4.5263],
        [  1.9241,   7.3503,   7.9004],
        [ -3.5029,   7.5138,   8.1853],
        [  5.1292,   4.8183,  -5.7997],
        [  9.9750,  -6.6338,  -1.2440],
        [  5.1196,  -2.7160,   4.2705],
        [ -7.5603,   4.6459,   8.2002],
        [  0.5600, -11.0321,  -0.8660],
        [  0.3038,   7.8092,   0.4164],
        [  2.0937,  13.8293,  -2.3533],
        [ 11.9808,   4.9340,  -3.3715],
        [ -8.5900,   2.9767,   7.4811],
        [ -1.1885, -10.8211,   0.0865],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.3367,  -4.4806,   0.4643],
        [  1.0093,   7.6567,   2.3526],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.9587,  12.3216,   9.1695],
        [  1.1274,  14.0116,   8.1753],
        [  6.2576,   2.1359,  -3.3398],
        [  3.7121,   6.3117,   0.6738],
        [ -3.9508,  -9.8515,   1.7923],
        [ -6.0175,  -8.4541,  -9.8408],
        [  1.1139,   7.6281,  -2.5680],
        [  1.7228,  -4.7575,   8.3396],
        [  4.4039,   5.6903,   8.7366],
        [ -8.5506,  -5.1154,   6.3423],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.4429,  -2.2529,   9.7306],
        [ -1.6334, -10.7207,   7.1397],
        [ -0.4341, -10.9473,   7.4509],
        [  5.8318,   3.5567,  -4.7692],
        [ -0.1963,   7.8765,  -1.1371],
        [ -7.5603,   4.6459,   8.2002],
        [ -0.4341, -10.9473,   7.4509],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.3119,   1.7797,   9.9186],
        [  5.7485,  -9.9811,  -3.2355],
        [ 12.4996,   2.6335,  -6.3166],
        [ -3.5173,   7.5091,  -8.9585],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.0467,  -9.8369,  -6.5816]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0337, 0.0492,  ..., 0.4472, 0.9301, 0.4194],
        [0.0337, 0.0000, 0.0157,  ..., 0.4438, 0.8981, 0.4198],
        [0.0492, 0.0157, 0.0000,  ..., 0.4400, 0.8845, 0.4179],
        ...,
        [0.4472, 0.4438, 0.4400,  ..., 0.0000, 1.0975, 0.0554],
        [0.9301, 0.8981, 0.8845,  ..., 1.0975, 0.0000, 1.1208],
        [0.4194, 0.4198, 0.4179,  ..., 0.0554, 1.1208, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.72668467,  7.8575648 ,  7.8788965 ,  8.01009083,  8.09984231,
        8.10886021, 12.7125079 ,  8.35856487,  7.9337647 ,  5.91854397,
       13.76047999, 13.97779839, 10.48840476,  4.78544875,  6.74872981,
       14.09176259,  9.76896904,  8.62329466, 13.68063098, 12.9089559 ,
       13.95472331,  8.53501777,  5.83560009,  7.06169391,  5.8828986 ,
       13.84126197,  7.8575648 , 13.56367857,  5.41348121, 11.42420508,
       14.02526505, 11.24434374, 11.42420508, 13.14690137, 13.66991027,
        6.24187044, 12.41366754,  6.69389431,  9.69792513, 12.77402351,
        5.23878371, 10.68416401, 11.1341798 ,  8.23525433,  7.57946459,
        5.91408504, 11.85743425,  4.78544875,  8.34949383,  6.76056625,
        5.07306274, 11.11038551,  9.69629   , 12.46513646, 11.4088043 ,
        8.43626114,  9.15253344,  8.30729352,  5.10444097,  8.40445872,
        7.59795445,  8.29022197,  7.03733775, 11.97949273,  5.79543038,
        8.8737556 , 11.04629471,  7.81509631, 13.98691426, 12.95702647,
        9.09119351, 10.88618114, 13.18278606, 12.18998195,  7.72291946,
        5.70405525,  9.53015376, 13.68672855, 14.05687815,  6.61211224,
        7.32233733, 10.61417181, 10.37698435,  7.70897675,  5.05981914,
        7.19534146,  9.96391045, 11.97238624,  9.71392179,  5.89074104,
       10.84437705, 10.95593729,  6.83083863,  7.8788965 ,  8.8737556 ,
       10.95593729, 10.5575601 , 13.05148826,  9.69792513, 12.50996828,
        6.55800267, 11.51811603, 12.77402351,  8.29203356,  9.96391045,
       11.54672755])}
{'P': tensor([[1.2309e-02, 2.3052e-01, 1.6430e-01,  ..., 4.5491e-06, 1.5263e-03,
         2.3577e-04],
        [4.0906e-04, 1.3365e-02, 2.1927e-01,  ..., 5.6029e-06, 1.3322e-03,
         1.7555e-04],
        [7.5535e-04, 1.0449e-03, 3.6879e-02,  ..., 1.8706e-05, 3.0202e-03,
         3.1767e-04],
        ...,
        [4.0072e-05, 3.3031e-05, 2.5975e-05,  ..., 9.5532e-02, 1.2070e-05,
         6.3364e-05],
        [1.9524e-03, 2.6411e-03, 4.4587e-03,  ..., 4.7904e-05, 1.3637e-01,
         8.0363e-04],
        [1.0379e-02, 1.1098e-02, 1.3493e-02,  ..., 7.7199e-06, 9.3669e-02,
         4.3323e-02]]), 'X': tensor([[  0.9445,   7.6736,   0.5095],
        [  0.3038,   7.8092,   0.4164],
        [ -0.5739,   7.9056,   0.6382],
        [ -0.8885,   7.9160,   0.0571],
        [ -1.6220,   7.8917,   0.0315],
        [ -1.2454,   7.9127,  -0.5715],
        [ -7.4808,   4.7462,   2.0266],
        [ 10.8670,  -5.3428,  -2.8637],
        [  3.8182,  -3.9347,  -1.7961],
        [ -5.9225,  -8.5355,   2.4778],
        [  1.7228,  -4.7575,   8.3396],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.2311,  -2.0445,   1.3545],
        [ -6.1899,  -8.3011,  -7.9269],
        [  9.3424,  -7.3685,  -3.6807],
        [  7.3592,  11.3937,   6.2407],
        [  5.7784,  -1.6401,   9.5587],
        [  4.0362,  13.2397,  -7.0030],
        [ 10.9931,   7.2107,   7.7933],
        [  4.4915,   5.5990,   7.0405],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.9636,   2.0648,   8.1946],
        [  3.3839,  -4.1966,   8.3574],
        [ -3.0644,   7.6439,  -1.4322],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -9.2707,   0.9661,   4.4863],
        [ -2.0229, -10.6163,   8.6930],
        [  0.1041, -11.0045,   2.4311],
        [  1.3103,  -4.8032,  -0.9300],
        [ -8.3900,   3.3727,   3.1650],
        [ 12.5224,  -0.4497,   4.8448],
        [  2.2373,  -4.6506,  -1.7334],
        [  3.6154,  -4.0640,  -7.9698],
        [  4.6992,   5.3690,   3.0935],
        [ 10.3251,   8.2928,  -3.0006],
        [  1.0093,   7.6567,   2.3526],
        [  4.4915,   5.5990,   7.0405],
        [  2.3889,  -4.6082,  -8.7291],
        [ 12.2218,   4.0953,  -8.2072],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.1459,   1.4780,  -4.2403],
        [  6.3031,   1.8477,  -7.1706],
        [  4.0362,  13.2397,  -7.0030],
        [  0.5850,  -4.7982,   9.6639],
        [  2.5375,   7.0676,  -4.3490],
        [ -1.6334, -10.7207,   7.1397],
        [ -0.0282,   7.8575,  -1.1746],
        [ -5.1046,  -9.1610,   2.3418],
        [  5.0348,  -2.8229,   3.5436],
        [  2.5620,   7.0549,   8.4825],
        [  8.5716,  10.3596,   6.8818],
        [  6.5428,  -9.5728,  -2.9897],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.1861,   4.7361,  -3.7312],
        [  5.8979,   3.3958,  -3.8713],
        [  7.3724,  -9.0577,  -8.9843],
        [ -2.2188,   7.8218,   8.6064],
        [ -1.9487,   7.8590,  -3.3330],
        [  9.6816,   9.1536,  -5.8543],
        [  9.5018,  -7.1945,  -1.1892],
        [ -5.9225,  -8.5355,   2.4778],
        [ 12.2435,   4.0076,   8.1257],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.3444,   0.5821,   6.9101],
        [  0.9478,  -4.8144,   0.6371],
        [ -1.7290,   7.8825,  -3.3831],
        [ 11.7794,   5.5086,  -9.6929],
        [  2.2860,   7.1916,  -6.1011],
        [  5.4167, -10.1293,   6.2156],
        [  4.9337,  12.8566,  -3.6582],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -8.9942,  -3.9990,   6.0448],
        [  6.1372,   2.6726,  -6.7885],
        [  3.3094,  13.4969,  -3.6507],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -0.5739,   7.9056,   0.6382],
        [  4.9152,  12.8653,  -1.4459],
        [  6.2576,   2.1359,  -3.3398],
        [ -4.4520,   7.1370,  -8.7867],
        [ 11.7795,  -3.4697,   3.5624],
        [ -5.5525,   6.5146,  -4.4984],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.3261,   7.1726,   6.2522],
        [  5.2188,  -2.5839,   2.1833],
        [  0.3493,  -4.7727,  -6.5682],
        [  2.7420,  -4.4896,  -3.7139],
        [ -8.4824,  -5.2575,  -0.0795],
        [  1.7122,  -4.7591,  -3.6734],
        [ -6.8466,  -7.6488,   1.9648],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -5.9872,   6.2037,  -9.4477],
        [  2.5318,  13.7231,   6.3692],
        [ -4.7642,  -9.3860,  -5.1133],
        [  0.8700,  -4.8132,  -7.8042],
        [  0.1931,  -4.7491,   7.4385],
        [  1.8766,   7.3694,   1.1887],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -0.9715, -10.8630,   0.3555],
        [ -1.5451,   7.8975,   7.6104],
        [  3.3167,   6.6022,   6.3554]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.6978e-02, 7.1525e-02,  ..., 5.4460e-01, 5.1160e-01,
         4.4835e-01],
        [2.6978e-02, 0.0000e+00, 4.7994e-02,  ..., 5.1825e-01, 5.1458e-01,
         4.6055e-01],
        [7.1525e-02, 4.7994e-02, 4.7405e-06,  ..., 4.7468e-01, 4.9345e-01,
         4.5635e-01],
        ...,
        [5.4460e-01, 5.1825e-01, 4.7468e-01,  ..., 0.0000e+00, 6.9868e-01,
         7.8081e-01],
        [5.1160e-01, 5.1458e-01, 4.9345e-01,  ..., 6.9868e-01, 0.0000e+00,
         1.8180e-01],
        [4.4835e-01, 4.6055e-01, 4.5635e-01,  ..., 7.8081e-01, 1.8180e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.73151309,  7.81509631,  7.92644746,  7.96575782,  8.05668609,
        8.01009083,  8.859416  , 12.10942412,  5.48276038, 10.38897563,
        5.05981914,  4.8763135 , 12.40074901, 10.35485331, 11.89855267,
       13.56367857,  6.00663314, 13.84126197, 13.14690137,  7.17789666,
       10.06850677,  9.19837792,  5.39098669,  8.23525433,  8.83725081,
        9.97822473, 10.5575601 ,  9.32094055, 10.80728788, 11.0050379 ,
        4.978704  ,  9.04255655, 12.53047489,  5.16077764,  5.43943791,
        7.13501264, 13.2430377 ,  7.72291946,  7.17789666,  5.19065439,
       12.889695  , 11.83075005,  9.18690018,  9.26455795,  6.56833864,
       13.84126197,  4.83371341,  7.5092834 , 10.84437705,  7.8575648 ,
       10.48721382,  5.77219077,  7.50564223, 13.44593519, 11.59512194,
       13.14777433,  7.02326515,  6.80559669, 11.67875855,  8.1303945 ,
        8.09704052, 13.32374056, 11.9182956 , 10.38897563, 12.88270018,
       12.10222204,  9.36256096,  4.90676242,  8.06990483, 13.00380624,
        7.54620849, 11.48663767, 13.77075174, 12.53345445,  9.84314712,
        6.69389431, 13.89671314,  5.21280156,  9.70108409,  7.92644746,
       13.77222752,  6.61211224,  8.41171369, 12.27991991,  8.55982105,
       12.91143491, 12.59969936,  7.5403822 ,  5.82345867,  4.78544875,
        5.26067019,  9.97964013,  5.05774295, 10.26545792,  9.79039371,
        8.62162372, 13.95472331, 10.52587243,  4.89121805,  4.75303127,
        7.60463349, 10.32715715,  9.88310614, 10.90638066,  8.04718433,
        7.38844703])}
{'P': tensor([[2.3862e-02, 2.4069e-01, 1.9940e-01,  ..., 3.3551e-06, 1.4022e-06,
         2.4409e-04],
        [7.4027e-04, 1.9178e-02, 2.7935e-01,  ..., 3.0181e-06, 1.2709e-06,
         1.4476e-04],
        [3.3094e-04, 4.4549e-04, 2.2655e-02,  ..., 4.2490e-06, 1.3899e-06,
         6.9508e-05],
        ...,
        [2.2659e-05, 1.6467e-05, 9.7544e-06,  ..., 2.6311e-02, 8.0068e-05,
         1.8687e-06],
        [2.1313e-05, 2.2359e-05, 1.6838e-05,  ..., 3.3830e-05, 7.7504e-02,
         9.1716e-05],
        [1.8087e-03, 3.7726e-03, 7.4273e-03,  ..., 1.7232e-06, 1.4067e-05,
         2.3742e-01]]), 'X': tensor([[ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0148, 0.0709,  ..., 0.2850, 1.0022, 0.6335],
        [0.0148, 0.0000, 0.0562,  ..., 0.2702, 1.0049, 0.6425],
        [0.0709, 0.0562, 0.0000,  ..., 0.2141, 1.0237, 0.6836],
        ...,
        [0.2850, 0.2702, 0.2141,  ..., 0.0000, 1.1053, 0.8474],
        [1.0022, 1.0049, 1.0237,  ..., 1.1053, 0.0000, 0.5231],
        [0.6335, 0.6425, 0.6836,  ..., 0.8474, 0.5231, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.73692436,  7.85919687,  8.04483495,  8.1303945 ,  8.28968067,
        8.29640613,  6.81084724, 13.68063098, 13.95361098,  8.12043983,
        5.24461068, 12.97370371, 14.10045797,  7.06372172,  9.99211663,
        5.95297459,  8.09899874,  5.05444288,  9.24501829,  9.17905491,
       10.68416401,  5.21280156, 11.00812996, 10.8207297 , 10.92597196,
        4.91296205, 12.46513646,  7.02326515,  9.41244865, 10.35093485,
        5.95111804,  5.22903117,  6.28835101,  6.31576264,  5.62005629,
        5.30158497, 10.01804223,  8.37954853, 10.23946247, 10.31584102,
       11.54672755,  8.05242591,  9.44486609,  9.50913849, 10.48840476,
        9.18358562, 11.42947632,  4.91652556,  8.29022197,  6.54693987,
        5.70405525, 13.66991027,  8.01009083,  6.36004393,  5.87132718,
        4.75303127, 13.45642379,  8.16584537,  8.06517143,  7.70897675,
        8.79284466,  7.53884104, 13.6829843 , 12.5769659 , 10.00794468,
       11.54071859, 11.22142195,  4.96231532, 12.97370371,  9.05843201,
        5.22903117, 11.50791859,  7.26074925, 13.6660671 ,  8.04664886,
        9.77777346, 12.62711252,  7.03733775,  4.78544875, 11.42947632,
        7.16747878,  7.14837514, 12.88270018, 11.37939695,  6.27905292,
       10.9036226 ,  9.37296102, 13.40024224,  6.14086582,  7.5403822 ,
       11.9702762 ,  9.77588801, 13.25336001,  5.95297459, 11.57249984,
        7.44805901, 11.70168755,  9.62958973, 13.103709  , 13.88683436,
        8.05668609,  5.10446653, 10.06407839, 11.36089355,  5.68942572,
       14.05633441])}
{'P': tensor([[3.0419e-02, 1.6795e-01, 1.1284e-01,  ..., 3.4510e-04, 7.0904e-06,
         3.2493e-05],
        [7.2740e-04, 4.7880e-02, 2.9074e-03,  ..., 9.2446e-04, 2.1671e-05,
         4.2991e-05],
        [1.6674e-03, 3.9679e-02, 3.2062e-02,  ..., 8.8324e-04, 7.7918e-06,
         4.7979e-05],
        ...,
        [4.3660e-06, 4.9746e-06, 6.0118e-06,  ..., 2.0591e-02, 6.7332e-05,
         6.3245e-04],
        [6.2223e-05, 3.3977e-05, 3.5757e-05,  ..., 6.0249e-06, 1.8399e-01,
         6.0239e-04],
        [3.9367e-04, 1.6205e-04, 4.1124e-04,  ..., 1.6311e-04, 1.3398e-04,
         1.3204e-01]]), 'X': tensor([[  0.6933,   7.7336,   2.1179],
        [ -1.9646,   7.8571,   3.3817],
        [ -0.5739,   7.9056,   0.6382],
        [ -1.6220,   7.8917,   0.0315],
        [ -4.2040,   7.2486,   2.9875],
        [ -1.6086,   7.8928,   4.3385],
        [  2.8602,   6.8905,   5.8779],
        [  6.3569,   0.7591,   6.5011],
        [  8.5716,  10.3596,   6.8818],
        [  9.3607,   9.5337,  -4.5412],
        [  3.9022,  13.2906,   5.5788],
        [ -8.0588,  -6.0341,  -2.3006],
        [  6.1755,   2.5233,  -2.0042],
        [  7.7794,  11.0629,   8.9069],
        [ 12.3357,   3.6001,   0.7903],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.4203,  -2.0340,   2.5856],
        [  6.3551,   0.7149,   1.3344],
        [  4.0591,   6.0202,  -1.6339],
        [  6.3587,   1.1596,  -3.6629],
        [  5.8318,   3.5567,  -4.7692],
        [ 11.2847,  -4.5848,  -4.7503],
        [  3.7916,  -3.9524,  -7.9547],
        [ -9.3616,  -2.4749,   0.1793],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 12.1751,  -2.2592,  -5.7751],
        [  5.4867,  12.5816,   2.9549],
        [  5.5626,  -2.0536,   8.4433],
        [ -9.4697,  -0.6157,  -6.8472],
        [  7.1697,  11.5343,  -0.7276],
        [  6.3569,   0.7591,   6.5011],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4739,  -1.2497,   0.5483],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -9.4645,  -1.4880,  -2.0226],
        [  0.9036,   7.6840,   9.9237],
        [ -7.8280,   4.2848,  -7.1955],
        [ -9.3616,  -2.4749,   0.1793],
        [ -6.4464,   5.8272,   7.5038],
        [  3.3094,  13.4969,  -3.6507],
        [  6.0345,   3.0175,  -1.4424],
        [  2.5382,   7.0672,   4.5097],
        [ -5.1369,   6.7755,   8.3836],
        [  6.9433,  -9.3362,  -0.3421],
        [ -9.4541,  -0.3327,   8.8246],
        [ -2.2508, -10.5479,  -2.7235],
        [ -7.0120,   5.2825,   7.7762],
        [  3.6469,   6.3624,  -1.8414],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.4524,  -1.6813,  -6.6862],
        [  7.3592,  11.3937,   6.2407],
        [ -7.4733,   4.7555,   6.2490],
        [ -7.3470,  -7.0615,   5.0066],
        [  5.8318,   3.5567,  -4.7692],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.3653,   3.4184,   9.3454],
        [  2.6613,   7.0021,   4.5783],
        [ -4.6393,   7.0461,   1.1624],
        [  6.2038,   2.4020,  -3.1692],
        [ -7.3554,   4.8985,  -0.6665],
        [  4.2096,   5.8815,   1.9230],
        [ 12.4013,  -1.2597,  -7.6474],
        [  1.8766,   7.3694,   1.1887],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -7.9925,  -6.1423,   2.8990],
        [  8.3791,  10.5409,   5.5590],
        [ -0.5739,   7.9056,   0.6382],
        [ -1.0482, -10.8487,  -5.9066],
        [ -2.0445,   7.8469,  -0.5857],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.9653,  -9.3225,  -3.4727],
        [ -9.3444,   0.5821,   6.9101],
        [ -6.5295,  -7.9785,  -7.4429],
        [  4.5767,   5.5071,   3.9069],
        [  4.3554,   5.7393,  -9.0766],
        [ -8.9788,   2.0208,  -3.2867],
        [ -4.3359,  -9.6431,  -9.0616],
        [  2.9461,  -4.4077,  -0.4831],
        [  6.1685,  12.1979,  -6.2473],
        [ -8.8058,   2.4850,  -1.7425],
        [  9.2811,  -7.4336,   5.1771],
        [ -8.9606,  -4.0994,   1.7886],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.8447,  -4.7372,   8.1235],
        [  6.3211,   1.7017,  -0.2164],
        [ -3.5173,   7.5091,  -8.9585],
        [  8.6420,  -8.0580,   6.5781],
        [ 12.2914,   3.8037,   6.0978],
        [  6.6319,  11.9066,   3.5165],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.1379,  -4.6757,   6.4981],
        [  0.5600, -11.0321,  -0.8660],
        [ -0.5739,   7.9056,   0.6382],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -3.7964,  -9.9294,   7.7147],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.2692,  13.5098,  -4.9032],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.2026,  12.1774,  -1.5887],
        [  2.3373,  -4.6232,  -5.2246],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.1252, -11.0062,   7.2564],
        [  1.9503,  -4.7172,  -6.2313]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4575e-01, 1.2558e-01,  ..., 9.0316e-01, 6.2278e-01,
         5.2950e-01],
        [1.4575e-01, 0.0000e+00, 2.4572e-01,  ..., 9.1710e-01, 4.7876e-01,
         6.6901e-01],
        [1.2558e-01, 2.4572e-01, 4.7405e-06,  ..., 8.0907e-01, 6.9737e-01,
         4.2729e-01],
        ...,
        [9.0316e-01, 9.1710e-01, 8.0907e-01,  ..., 0.0000e+00, 1.0342e+00,
         8.7801e-01],
        [6.2278e-01, 4.7876e-01, 6.9737e-01,  ..., 1.0342e+00, 0.0000e+00,
         1.1213e+00],
        [5.2950e-01, 6.6901e-01, 4.2729e-01,  ..., 8.7801e-01, 1.1213e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.76457095,  8.09899874,  7.92644746,  8.05668609,  8.37954853,
        8.0550286 ,  7.46053281,  6.4020404 , 13.44593519, 13.36092624,
       13.85158417, 10.0674887 ,  6.67108662, 13.52430176, 12.85032836,
        6.83083863,  9.63742669,  6.39520397,  7.26074925,  6.46356806,
        6.83083863, 12.18045978,  5.47702766,  9.68323686,  9.78286733,
       12.3828957 , 13.72594022,  5.92952538,  9.48969978, 13.58101794,
        6.4020404 , 10.0674887 ,  9.55593374, 12.1460931 ,  9.58072286,
        7.73692436,  8.92395381,  9.68323686,  8.68979052, 13.89671314,
        6.74686071,  7.50917551,  8.50268322, 11.63502679,  9.45995955,
       10.78533838,  8.77913428,  7.33350642,  5.24137909,  9.60080639,
       13.56367857,  8.8580671 , 10.19036081,  6.83083863,  6.05113217,
        9.0368458 ,  7.49076899,  8.43626114,  6.65259775,  8.83725081,
        7.23276582, 12.46513646,  7.60463349, 12.35421461, 10.08001894,
       13.46553623,  7.92644746, 10.8992524 ,  8.10886021, 12.22503754,
       11.63724792,  9.36256096, 10.30972193,  7.16058587,  7.20482224,
        9.20340031, 10.57302546,  5.30158497, 13.66896739,  9.14973319,
       11.89105259,  9.85384623,  7.85919687, 10.06850677,  5.08372149,
        6.54615887,  8.29203356, 11.8159254 , 12.86648771, 13.62896593,
        5.41348121,  5.14124612, 11.04629471,  7.92644746,  9.78286733,
       10.63038565,  6.06416957,  9.47369628, 13.89974693, 10.35485331,
       13.6660671 ,  5.18047638,  5.0209005 ,  9.50654967, 11.00694688,
        5.10444097])}
{'P': tensor([[1.9587e-02, 1.6613e-01, 1.3107e-01,  ..., 6.6757e-05, 6.6799e-07,
         1.3792e-03],
        [4.0009e-04, 3.2533e-02, 5.8281e-03,  ..., 5.0415e-05, 1.2796e-06,
         8.1980e-03],
        [2.6930e-04, 2.6580e-02, 1.4549e-02,  ..., 2.7352e-05, 7.0855e-07,
         2.3895e-03],
        ...,
        [5.3109e-02, 1.9854e-02, 3.1423e-02,  ..., 1.4395e-01, 2.2657e-05,
         3.3794e-04],
        [3.4374e-05, 1.4042e-05, 2.1443e-05,  ..., 1.5803e-03, 1.5889e-01,
         1.6454e-06],
        [5.8928e-05, 8.9852e-05, 7.8464e-05,  ..., 4.6463e-05, 3.0975e-05,
         1.2391e-01]]), 'X': tensor([[ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1138, 0.0855,  ..., 0.3260, 0.9583, 0.6971],
        [0.1138, 0.0000, 0.0864,  ..., 0.4149, 0.9496, 0.5846],
        [0.0855, 0.0864, 0.0000,  ..., 0.3321, 0.8881, 0.6539],
        ...,
        [0.3260, 0.4149, 0.3321,  ..., 0.0000, 0.8070, 0.9776],
        [0.9583, 0.9496, 0.8881,  ..., 0.8070, 0.0000, 1.1831],
        [0.6971, 0.5846, 0.6539,  ..., 0.9776, 1.1831, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.78980299,  8.12580976,  8.06990483,  8.14433542,  8.09704052,
        8.34949383, 13.76047999, 14.05633441,  9.77777346,  7.72415048,
        8.29640613,  6.74872981, 10.36844813,  9.62958973, 12.17451685,
        9.71392179,  8.30125201,  5.87132718, 10.92597196, 10.91805683,
        7.06372172, 10.90638066, 10.90638066,  6.46560181,  6.69389431,
        5.23878371, 10.07238463,  8.8580671 ,  5.26067019, 13.59170809,
       12.52372629,  9.69629   ,  7.96575782, 11.87665501, 10.42667093,
       13.1399181 , 13.46553623,  6.50409372,  6.65368551,  9.68323686,
       11.63724792,  8.31625171, 11.42947632,  8.29022197, 12.35421461,
        7.15100303, 13.2430377 , 13.48809595,  8.04664886,  7.19513806,
        8.1137452 ,  6.33517349,  8.34949383,  6.02779201,  9.10945678,
       10.26327895,  9.71392179, 14.02532758, 13.99048094,  7.66377554,
       12.91505111,  6.02779201,  6.02789634, 13.05148826, 10.84437705,
       12.00995898,  4.78258092,  9.62238828, 12.84056604, 11.22596994,
        5.35774573,  8.28968067,  8.66273171,  5.80860165,  9.11473391,
       12.78327037, 13.83139024,  9.44486609,  8.09984231, 10.00802627,
       10.46739073,  7.62232307, 13.66051884,  4.91652556, 14.03618649,
        5.13125513, 11.31276109,  9.18358562,  8.12580976,  8.09842704,
        6.00732193, 11.69264243,  8.19703512,  5.70591986,  5.30158497,
       12.04924159,  5.96556248,  6.46356806, 12.26997165, 12.5085995 ,
       12.72524615, 14.09176259, 11.57249984,  6.80370254, 11.50791859,
        9.34061938])}
{'P': tensor([[1.6483e-02, 2.3125e-01, 2.2563e-01,  ..., 3.3092e-04, 2.0788e-04,
         7.1353e-05],
        [4.5571e-04, 1.8688e-02, 4.3472e-01,  ..., 5.7747e-04, 1.2928e-04,
         1.3432e-04],
        [6.9304e-04, 1.1713e-03, 3.1960e-02,  ..., 1.0080e-03, 2.0232e-04,
         2.4570e-04],
        ...,
        [6.5232e-06, 9.2614e-06, 9.6215e-06,  ..., 1.7666e-02, 8.0601e-06,
         2.6828e-03],
        [3.4392e-02, 4.0122e-02, 3.8411e-02,  ..., 2.2180e-03, 7.8053e-02,
         1.2149e-04],
        [1.1263e-05, 1.0849e-05, 1.0769e-05,  ..., 2.6969e-05, 2.0622e-05,
         5.6563e-02]]), 'X': tensor([[  0.3724,   7.7974,  -6.8722],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.0407,   7.9166,  -6.6021],
        [ -2.0840,   7.8415,  -7.8657],
        [ -0.6323,   7.9085,  -5.9544],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.5491,  -3.3485,   6.8412],
        [  6.3211,   1.7017,  -0.2164],
        [  1.9505,  -4.7171,   9.9868],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.0899,  -4.6870,  -7.4260],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.9875,   7.2207,  -8.8549],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.5083,  -4.1177,   0.2389],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -3.2053, -10.1993,  -3.8736],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.1737,   2.5305,   3.9049],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -2.5161,   7.7699,  -9.3531],
        [ -8.8154,   2.4611,   2.5198],
        [ -1.6334, -10.7207,   7.1397],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -0.0254, -10.9932,   3.2200],
        [  6.0310,  12.2795,  -6.4528],
        [  2.1625,  -4.6697,   7.0658],
        [  6.0305,   3.0296,   0.1375],
        [  4.4290,  -3.4596,   1.4181],
        [  6.3087,   1.8053,  -1.9481],
        [ -5.0943,  -9.1682,   0.5073],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.1907,  -3.3180,  -5.5943],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -1.8788, -10.6567,   3.7068],
        [  6.3031,   1.8477,  -7.1706],
        [  3.2701,   6.6338,   9.9209],
        [  5.5158,  -2.1337,   3.5246],
        [ -2.1533, -10.5778,   1.0112],
        [  1.5246,  -4.7838,  -5.2095],
        [ -6.3489,   5.9117,   6.5161],
        [  4.8674,  12.8875,   6.7622],
        [  4.7156,  -3.1831,  -8.0224],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.5375,   7.0676,  -4.3490],
        [ -8.5607,  -5.0938,  -5.4531],
        [  2.3367,   7.1676,  -9.5263],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -4.5778,  -9.5013,  -5.0667],
        [  5.4867,  12.5816,   2.9549],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.4572,   7.1085,   7.7376],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.8835,  -4.3178,   9.3080],
        [  4.4039,   5.6903,   8.7366],
        [  2.3373,  -4.6232,  -5.2246],
        [  0.5600, -11.0321,  -0.8660],
        [  5.3924, -10.1396,  -4.0863],
        [  3.9022,  13.2906,   5.5788],
        [ -9.1795,  -3.3623,   8.5259],
        [ 10.0995,  -6.4739,   2.0974],
        [ -4.0868,   7.2981,   8.1874],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.1178,   1.5784,   2.1410],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.3261,  13.7749,   5.9121],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2675,  12.1379,  -8.9720],
        [ -5.9940,   6.1985,   5.9566],
        [ -4.3359,  -9.6431,  -9.0616],
        [  4.5814, -10.4485,   2.5496],
        [ 10.8218,  -5.4177,   6.8573],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.1348,   5.9514,  -0.8856],
        [  0.1931,  -4.7491,   7.4385],
        [ 11.1469,   6.9237,   1.6160],
        [  3.3744,   6.5623,  -6.3636],
        [  6.0054,  -1.0902,   8.3079],
        [  2.1504,  -4.6726,   8.7908],
        [ 12.2692,  -1.8882,   3.9253],
        [ -8.9282,   2.1648,   8.4894],
        [ -2.8162,   7.7059,  -9.6885],
        [ -6.0195,   6.1789,  -2.9496],
        [ -9.0961,   1.6529,   3.1553],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.8602,   6.8905,   5.8779],
        [  2.1771, -10.9776,  -3.2007],
        [  3.6249, -10.7255,   5.6605],
        [  4.9452, -10.3187,  -7.8621],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.1929,   2.4501,  -6.9352],
        [  2.2373,  -4.6506,  -1.7334],
        [  2.5620,   7.0549,   8.4825],
        [  4.1999,  -3.6543,  -4.1998],
        [ -9.4630,  -1.5145,   0.2983],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.3900,   3.3727,   3.1650],
        [  3.6469,   6.3624,  -1.8414],
        [ -8.0766,  -6.0046,  -3.1873]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.3493e-02, 6.7095e-02,  ..., 8.0461e-01, 3.1197e-01,
         7.6707e-01],
        [4.3493e-02, 1.5170e-04, 2.4120e-02,  ..., 7.7305e-01, 3.2695e-01,
         7.2359e-01],
        [6.7095e-02, 2.4120e-02, 0.0000e+00,  ..., 7.6104e-01, 3.4250e-01,
         7.0016e-01],
        ...,
        [8.0461e-01, 7.7305e-01, 7.6104e-01,  ..., 0.0000e+00, 7.2450e-01,
         5.8543e-01],
        [3.1197e-01, 3.2695e-01, 3.4250e-01,  ..., 7.2450e-01, 0.0000e+00,
         9.1168e-01],
        [7.6707e-01, 7.2359e-01, 7.0016e-01,  ..., 5.8543e-01, 9.1168e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.80625981,  7.96181288,  7.98469166,  8.1137452 ,  7.9337647 ,
        8.22728399,  9.97964013,  5.64865173,  6.54615887,  5.10446653,
        8.97052115,  5.13182506,  7.71075226, 13.14777433,  8.04718433,
       12.22276278,  8.51928531, 12.3828957 , 10.6910768 , 13.20597932,
        6.23262289,  6.67218687,  9.75075273,  8.16714836,  9.15253344,
       10.84437705, 12.46513646,  9.93553424, 10.99326577, 13.68063098,
        5.14607014,  6.74872981,  5.62005629,  6.5618968 , 10.48840476,
       12.26997165,  9.7712321 , 12.07818871, 10.82106505,  6.56833864,
        7.39599986,  5.91408504, 10.794755  ,  5.0209005 ,  8.67502348,
       13.77604293,  5.68942572, 10.31886066,  7.5092834 ,  9.96152788,
        7.53884104,  9.9353921 ,  9.75075273, 10.54658597, 13.72594022,
        8.27853076,  9.37989631,  5.79391825,  7.52117132,  5.13125513,
        9.87720102,  7.19534146,  5.18047638, 11.04629471, 11.48434767,
       13.85158417,  9.77588801, 11.99633819,  8.36445758,  9.34127013,
        9.25336371, 10.35485331, 13.96987806,  6.33075682, 13.66051884,
        8.62260327, 10.57302546, 11.4088043 , 12.10222204, 10.63038565,
        7.24677664,  4.75303127, 13.12218013,  7.37902766,  6.10360073,
        5.14371005, 12.41366754,  9.18690018,  8.20436764,  8.62631801,
        9.24501829, 12.95702647,  7.46053281, 11.1913598 , 11.32149259,
       11.4424816 ,  9.55593374,  6.65991877,  5.16077764,  7.50564223,
        5.5671389 ,  9.58347456,  9.97220799,  9.04255655,  7.33350642,
       10.06407839])}
{'P': tensor([[1.2548e-02, 2.0588e-01, 1.7221e-01,  ..., 2.3870e-04, 2.4599e-05,
         8.9481e-05],
        [8.9024e-04, 3.1420e-02, 1.1548e-01,  ..., 3.7404e-04, 4.7380e-05,
         1.5071e-04],
        [5.8089e-04, 3.5604e-03, 2.2818e-02,  ..., 2.1316e-04, 3.4147e-05,
         9.9782e-05],
        ...,
        [2.9151e-02, 3.7849e-02, 2.7856e-02,  ..., 9.3928e-02, 1.3784e-04,
         7.0235e-04],
        [1.1658e-02, 8.2736e-03, 9.0005e-03,  ..., 1.8028e-03, 1.0191e-01,
         2.5808e-01],
        [6.2420e-02, 4.8356e-02, 5.8184e-02,  ..., 3.0307e-03, 1.4753e-03,
         1.0805e-01]]), 'X': tensor([[ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.7994e-02, 4.8783e-02,  ..., 4.3122e-01, 4.0443e-01,
         3.2885e-01],
        [4.7994e-02, 4.7405e-06, 4.3125e-02,  ..., 4.2508e-01, 4.4569e-01,
         3.6905e-01],
        [4.8783e-02, 4.3125e-02, 0.0000e+00,  ..., 4.6551e-01, 4.1128e-01,
         3.3392e-01],
        ...,
        [4.3122e-01, 4.2508e-01, 4.6551e-01,  ..., 0.0000e+00, 7.6625e-01,
         7.0540e-01],
        [4.0443e-01, 4.4569e-01, 4.1128e-01,  ..., 7.6625e-01, 0.0000e+00,
         7.7814e-02],
        [3.2885e-01, 3.6905e-01, 3.3392e-01,  ..., 7.0540e-01, 7.7814e-02,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.81509631,  7.92644746,  7.96575782,  8.05668609,  8.01009083,
        8.10886021,  6.42439344, 13.09737053, 13.99877413, 10.52587243,
        8.29022197,  8.8737556 ,  9.24501829,  5.86519525,  6.21275016,
       13.00413829,  8.62329466,  7.24326992,  9.84314712,  8.22728399,
       11.8159254 , 13.13888841,  6.48486171, 13.12299145,  5.37343173,
        5.87305005,  6.55800267,  4.79003361, 13.02818702,  5.52416368,
       13.62896593, 11.35182589, 12.25806559,  6.24614905,  9.26455795,
       10.46739073,  9.12220271,  6.10360073, 11.83570132,  7.15100303,
       12.7125079 ,  9.69792513, 11.10066126,  9.25336371, 12.5769659 ,
        5.5748624 ,  4.78002665, 10.71162681,  8.95133961,  5.87132718,
       11.28093507,  5.47866221,  7.92644746, 12.70489563, 12.9089559 ,
        5.95111804,  6.10360073, 14.03618649,  5.91854397,  6.46560181,
       11.27160035,  8.62162372,  8.51928531,  5.19577146,  5.96556248,
       10.66631784,  8.09899874, 10.31584102,  8.62162372, 13.13888841,
        5.18277858, 11.33837396,  6.90115901,  8.11227668, 10.35464916,
       13.26590071,  8.98765644,  5.16077764,  8.87587457,  5.05989547,
       13.66991027,  5.70591986,  7.8575648 , 11.66453059,  5.91854397,
        7.53884104,  6.39438044, 13.99877413, 12.49317563,  6.05113217,
       13.79676654,  8.53501777, 13.99048094,  4.89121805, 12.78327037,
        7.50917551,  6.27905292, 10.69963954,  5.19065439, 13.50484539,
        6.65259775,  9.88571713,  6.02588517,  7.46053281,  6.60752373,
        7.15990676])}
{'P': tensor([[1.5704e-02, 2.6085e-01, 2.3985e-01,  ..., 1.1365e-03, 2.7318e-05,
         6.4471e-06],
        [4.6382e-04, 1.9567e-02, 4.5518e-01,  ..., 6.5243e-04, 3.6144e-05,
         6.5064e-06],
        [7.5541e-04, 1.2567e-03, 3.4289e-02,  ..., 1.0034e-03, 6.3694e-05,
         1.0951e-05],
        ...,
        [1.4222e-02, 2.2925e-02, 2.3856e-02,  ..., 7.5505e-02, 2.4784e-04,
         1.0404e-04],
        [1.9014e-04, 1.4886e-04, 1.4227e-04,  ..., 4.8420e-04, 6.6456e-02,
         7.9888e-03],
        [8.7578e-04, 6.6543e-04, 6.2720e-04,  ..., 6.8076e-03, 8.8164e-04,
         1.4363e-01]]), 'X': tensor([[  0.2435,   7.8191,  -6.4709],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.0407,   7.9166,  -6.6021],
        [ -0.6323,   7.9085,  -5.9544],
        [ -3.0004,   7.6607,  -7.2640],
        [ -2.0840,   7.8415,  -7.8657],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.1751,  -2.2592,  -5.7751],
        [  9.2692,   9.6370,   7.0746],
        [  3.6249, -10.7255,   5.6605],
        [  3.9913,  -3.8139,  -6.8311],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.7794,   5.5086,  -9.6929],
        [ 12.4690,  -0.8528,  -9.3374],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.3360,   0.4258,  -9.2509],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 12.6035,   0.7494,   7.1536],
        [  6.1737,   2.5305,   3.9049],
        [  3.8111, -10.6787,  -4.6539],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.5717,   1.9266,   7.4903],
        [  7.5309,  11.2617,   7.2656],
        [  1.1006,   7.6318,  -2.4877],
        [ 12.2921,  -1.7902,  -4.0870],
        [  5.4732,  -2.2041,  -4.3830],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.3062,  -5.6008,   0.0287],
        [  9.6614,   9.1784,  -1.2087],
        [  9.0051,   9.9234,   5.4411],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -7.3554,   4.8985,  -0.6665],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.1824,  -4.6647,   1.1985],
        [  9.4554,   9.4245,   9.8800],
        [  0.6933,   7.7336,   2.1179],
        [ -0.0410,   7.8591,   9.3457],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -8.5149,  -5.1905,   1.5239],
        [  1.6954,  -4.7616,   9.8198],
        [  2.2644,  -4.6434,   9.7508],
        [  1.7903,  -4.7467,   0.1168],
        [  3.9895,   6.0816,  -6.5235],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.3587,   1.1596,  -3.6629],
        [ -9.3705,   0.4208,  -3.5815],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -1.6086,   7.8928,   4.3385],
        [  8.2154,  -8.4262,   8.0171],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.7796,  -3.4696,   9.0088],
        [ -1.6906,   7.8860,   7.7142],
        [  0.8940, -11.0402,  -0.9783],
        [ -4.4503,   7.1378,   3.3751],
        [ -4.7389,   6.9953,   3.7638],
        [  6.0054,  -1.0902,   8.3079],
        [ -1.0012, -10.8576,   2.2867],
        [ -5.1046,  -9.1610,   2.3418],
        [ -4.2040,   7.2486,   2.9875],
        [ -0.7599, -10.8995,  -6.0366],
        [  1.4489,   7.5256,   4.4529],
        [  6.3515,   0.6418,  -6.5823],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.0870,  -4.6877,   9.6823],
        [ -0.8885,   7.9160,   0.0571],
        [  3.5965,  -4.0754,  -3.5035],
        [  2.9260,  -4.4162,   8.3808],
        [ -9.4697,  -0.6157,  -6.8472],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -6.6998,   5.5954,   0.0823],
        [  5.2199,   4.6861,  -1.9304],
        [  6.0047,   3.1062,  -6.7914],
        [ 10.9931,   7.2107,   7.7933],
        [  6.3505,   0.6235,  -8.5849],
        [  2.7295,  13.6703,  -8.9328],
        [  2.6970,   6.9826,   9.7350],
        [ -7.4887,   4.7364,   8.9074],
        [ -1.0482, -10.8487,  -5.9066],
        [  3.1271, -10.8343,   2.6442],
        [ -7.3333,   4.9246,  -7.9212],
        [  1.1994,   7.6035,   2.4295],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.0054,  -1.0902,   8.3079],
        [  5.3602,  -2.3808,   7.2098],
        [  1.4150,  13.9647,  -4.0915],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.1861,   4.7361,  -3.7312],
        [ -6.7745,  -7.7264,   8.4684],
        [  2.4697,  -4.5836,   1.4593],
        [ -4.0868,   7.2981,   8.1874],
        [  0.2480, -11.0153,  -3.0183],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -5.3382,   6.6533,   7.8054],
        [  0.9809,   7.6642,  -1.4217],
        [  0.9478,  -4.8144,   0.6371],
        [  6.2883,   0.0325,   5.2692]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.4364e-02, 6.8437e-02,  ..., 2.9132e-01, 4.2299e-01,
         8.1049e-01],
        [4.4364e-02, 1.5170e-04, 2.4120e-02,  ..., 3.0593e-01, 4.4266e-01,
         8.3114e-01],
        [6.8437e-02, 2.4120e-02, 0.0000e+00,  ..., 3.1451e-01, 4.5318e-01,
         8.4149e-01],
        ...,
        [2.9132e-01, 3.0593e-01, 3.1451e-01,  ..., 0.0000e+00, 1.4157e-01,
         5.2777e-01],
        [4.2299e-01, 4.4266e-01, 4.5318e-01,  ..., 1.4157e-01, 0.0000e+00,
         3.8849e-01],
        [8.1049e-01, 8.3114e-01, 8.4149e-01,  ..., 5.2777e-01, 3.8849e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.822854  ,  7.96181288,  7.98469166,  7.9337647 ,  8.22728399,
        8.1137452 ,  4.92497514, 12.3828957 , 13.37121923, 11.32149259,
        5.5205189 ,  8.85572292, 13.00380624, 12.4980845 ,  5.47702766,
       12.84056604,  6.35029274,  5.52416368, 13.6660671 , 10.06407839,
       12.62576026,  6.67218687, 11.33837396,  8.37954853, 12.71843462,
       13.54774355,  7.71075226, 12.4217523 ,  5.90035324,  4.85995002,
       12.27993098, 10.01804223, 13.32613445, 13.40024224,  9.50654967,
        8.83725081,  9.49511524,  5.14999799, 13.35013374,  7.76457095,
        7.85919687,  8.22728399,  9.4997987 ,  9.97220799,  5.05444288,
        5.16611782,  5.07306274,  7.27340834, 13.01728313,  6.46356806,
        9.37989631,  9.7712321 ,  8.0550286 , 11.76830981, 13.02818702,
       12.27993098,  8.06517143, 11.07637531,  8.41148722,  8.44940661,
        6.10360073, 10.9036226 , 10.48721382,  8.37954853, 10.92597196,
        7.66377554,  6.383897  ,  9.47369628,  5.34250225,  6.1561646 ,
        5.13125513,  7.96575782,  5.43544626,  5.29755487,  9.48969978,
       12.52372629, 12.18045978,  8.72896381,  7.01475376,  6.76056625,
       13.14690137,  6.38105015, 13.94008948,  7.48538905,  8.86082124,
       10.8992524 , 11.27656488,  8.83341608,  7.69752262, 10.26327895,
        6.10360073,  5.86519525, 14.03618649,  9.69629   ,  9.5014146 ,
        9.97964013,  7.02326515, 10.27572673,  5.20659892,  8.36445758,
       11.018084  ,  9.55720651,  8.53014915,  7.72668467,  4.90676242,
        6.28835101])}
{'P': tensor([[2.4884e-02, 2.6263e-01, 1.9792e-01,  ..., 1.0771e-03, 1.6504e-06,
         1.2540e-04],
        [1.9039e-03, 4.4255e-02, 2.3726e-01,  ..., 1.9940e-03, 2.2637e-06,
         1.2050e-04],
        [1.1180e-03, 2.3788e-03, 7.0967e-02,  ..., 8.3993e-03, 3.2186e-06,
         9.5664e-05],
        ...,
        [1.5038e-05, 1.6497e-05, 3.0215e-05,  ..., 3.0494e-02, 1.9730e-06,
         1.3943e-06],
        [9.8801e-04, 6.2200e-04, 4.4552e-04,  ..., 2.7794e-05, 1.2064e-01,
         6.5169e-03],
        [2.5426e-03, 2.8439e-03, 7.6477e-03,  ..., 3.5826e-03, 1.0537e-06,
         9.4740e-02]]), 'X': tensor([[ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0468, 0.1069,  ..., 0.7265, 0.6807, 0.8887],
        [0.0468, 0.0000, 0.0886,  ..., 0.7195, 0.7220, 0.9320],
        [0.1069, 0.0886, 0.0000,  ..., 0.6310, 0.7001, 0.9175],
        ...,
        [0.7265, 0.7195, 0.6310,  ..., 0.0000, 0.8213, 1.0208],
        [0.6807, 0.7220, 0.7001,  ..., 0.8213, 0.0000, 0.2251],
        [0.8887, 0.9320, 0.9175,  ..., 1.0208, 0.2251, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.83486943,  7.9337647 ,  8.12580976,  7.98469166,  7.96181288,
        8.14433542, 11.42947632, 11.10066126,  8.51928531, 10.35464916,
        5.16611782,  7.59795445,  8.30729352,  9.49511524,  5.53175275,
        9.97822473, 10.54658597, 10.24557047,  5.2287058 ,  8.31625171,
       11.37939695,  5.05981914,  5.92952538,  9.11473391,  9.77588801,
        8.19703512, 10.7167285 ,  7.5092834 ,  6.00663314, 10.08001894,
        6.31576264,  5.9510489 ,  5.94541803,  7.8788965 , 10.87409005,
        7.57685908, 11.9120321 , 10.46238054,  6.46560181,  9.23178204,
        9.23071698,  8.92395381,  5.89074104, 11.87665501, 13.62896593,
        6.5618968 , 12.7125079 ,  8.23525433, 11.05317597, 11.89105259,
        5.13616224,  7.39599986, 13.40024224,  9.85330646,  5.16611782,
       14.10796466,  8.50278896,  5.94535529,  9.24501829,  5.91408504,
        7.53063885, 13.32613445,  9.62238828,  7.81509631,  8.7273569 ,
        5.14607014,  8.04664886,  6.1463896 , 10.35464916, 13.6829843 ,
        6.53290663,  8.66273171,  8.0550286 , 12.9089559 ,  8.83341608,
       13.96987806, 12.10942412,  8.10886021, 13.36800073, 12.53285633,
       11.73397309,  5.92952538,  4.78258092, 10.0674887 ,  9.99145096,
       11.85743425,  5.04028538, 13.96905756,  9.88310614,  7.53884104,
        9.91668182,  6.07992623, 10.32715715,  6.81084724, 14.03505624,
       10.8207297 , 11.11969815,  6.89245352, 12.27993098,  5.22903117,
        5.1224079 ,  5.10444097, 10.78533838,  9.09963894, 12.52875793,
       13.92390689])}
{'P': tensor([[1.8314e-02, 2.3818e-01, 1.6029e-01,  ..., 3.3486e-06, 1.2768e-06,
         1.0274e-05],
        [3.0168e-04, 1.6888e-02, 3.3049e-01,  ..., 1.6769e-06, 1.7874e-06,
         4.1477e-06],
        [3.2599e-04, 6.2245e-04, 2.6862e-02,  ..., 2.0322e-06, 3.6846e-06,
         4.3395e-06],
        ...,
        [3.9590e-04, 2.0140e-04, 1.3646e-04,  ..., 6.8735e-03, 5.3210e-08,
         3.0108e-03],
        [1.8937e-06, 1.6196e-06, 1.5648e-06,  ..., 1.3666e-05, 1.0958e-01,
         2.9047e-06],
        [1.9331e-04, 2.4880e-04, 2.6367e-04,  ..., 2.8153e-06, 1.3873e-06,
         3.0922e-02]]), 'X': tensor([[  0.1460,   7.8340,  -9.4614],
        [ -1.5875,   7.8944,  -9.4286],
        [ -2.5161,   7.7699,  -9.3531],
        [ -1.8367,   7.8718,  -8.6523],
        [ -2.8162,   7.7059,  -9.6885],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.2939,  -2.4784,  -4.5840],
        [ -4.7389,   6.9953,   3.7638],
        [  5.1519, -10.2387,   9.8191],
        [ -1.7290,   7.8825,  -3.3831],
        [ 11.5935,   5.9740,  -4.3952],
        [  3.4140,  -4.1803,  -8.8224],
        [ 11.5083,  -4.1177,   0.2389],
        [ -7.4807,   4.7463,   2.0616],
        [  5.1667,  -2.6543,   1.2879],
        [  6.3516,   0.6425,   0.3425],
        [  4.5340,  -3.3629,   8.9652],
        [ -2.0445,   7.8469,  -0.5857],
        [  1.6308,  13.9254,  -2.0984],
        [  4.5340,  -3.3629,   8.9652],
        [  4.5491,  -3.3485,   6.8412],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.3303,   1.6145,  -8.4083],
        [  6.3550,   0.7129,   0.4917],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.2701,   6.6338,   9.9209],
        [  7.9141,  -8.6656,   9.6948],
        [ -3.5173,   7.5091,  -8.9585],
        [ -7.6734,   4.4980,  -5.4248],
        [  3.3094,  13.4969,  -3.6507],
        [ -1.9646,   7.8571,   3.3817],
        [  7.0358,  -9.2784,  -9.8479],
        [  5.3118,   4.5452,   4.2114],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.3038,   7.8092,   0.4164],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.2199,   4.6861,  -1.9304],
        [ 10.9394,   7.3069,  -8.9365],
        [  3.0719, -10.8449,  -4.1673],
        [  6.1737,   2.5305,   3.9049],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.3604,   0.4855,  -3.0158],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.7792,  11.8084,   1.6251],
        [  4.1348,   5.9514,  -0.8856],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.8979,   3.3958,  -3.8713],
        [ -1.0482, -10.8487,  -5.9066],
        [  3.5284, -10.7484,   5.3347],
        [  7.0664,  -9.2590,   8.0879],
        [  5.8323,  -1.5228,   5.1132],
        [ -8.6229,   2.9068,   2.4429],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.7796,  -3.4696,   9.0088],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 10.5955,   7.8824,  -3.7006],
        [ 12.2592,   3.9425,   2.0383],
        [  3.4009,  -4.1874,  -5.9751],
        [  2.1477,   7.2550,   8.9688],
        [ -3.4488, -10.0934,   2.5195],
        [ -4.1296,   7.2803,  -2.7194],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.0277,   4.9589,   7.0175],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.6321,  -4.5296,  -2.5159],
        [  0.5175,  14.0910,   4.6540],
        [  1.4749,  -4.7892,  -0.8089],
        [  3.2366,  -4.2732,   5.1532],
        [ 12.2311,  -2.0445,   1.3545],
        [ -4.7847,  -9.3729,  -3.7741],
        [  3.1892, -10.8220,  -6.8409],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.5294,  -2.1106,   9.0626],
        [  2.6613,   7.0021,   4.5783],
        [  4.3554,   5.7393,  -9.0766],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.3444,   0.5821,   6.9101],
        [ 11.5083,  -4.1177,   0.2389],
        [  9.6614,   9.1784,  -1.2087],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 12.4996,   2.6335,  -6.3166],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.9655,   3.2172,  -7.5953],
        [  6.2675,  12.1379,  -8.9720],
        [  1.4489,   7.5256,   4.4529],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.4159,  -2.0739,   9.6841],
        [ 12.6038,   0.7665,  -3.4439],
        [  5.8450,  -9.9356,   3.0767],
        [  2.5663,   7.0526,  -6.8769],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.0330,  -6.0765,   6.7945],
        [ -5.0550,   6.8231,  -9.2502],
        [  0.5250, -11.0306,   5.5358],
        [ -6.8466,  -7.6488,   1.9648],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.6970,   6.9826,   9.7350],
        [  6.3122,  -9.6994,  -6.3282],
        [  0.9036,   7.6840,   9.9237],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.9808,   4.9340,  -3.3715],
        [ -5.3328,  -8.9992,   7.3440],
        [  5.6928,  12.4710,   3.6001]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.7480e-02, 9.4019e-02,  ..., 5.2419e-01, 1.1952e+00,
         1.1215e+00],
        [5.7480e-02, 2.1453e-04, 3.6898e-02,  ..., 5.8117e-01, 1.1800e+00,
         1.1627e+00],
        [9.4019e-02, 3.6898e-02, 0.0000e+00,  ..., 6.1648e-01, 1.1654e+00,
         1.1851e+00],
        ...,
        [5.2419e-01, 5.8117e-01, 6.1648e-01,  ..., 0.0000e+00, 1.3758e+00,
         7.7270e-01],
        [1.1952e+00, 1.1800e+00, 1.1654e+00,  ..., 1.3758e+00, 3.0339e-04,
         1.1849e+00],
        [1.1215e+00, 1.1627e+00, 1.1851e+00,  ..., 7.7270e-01, 1.1849e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.83534987,  8.05242591,  8.16714836,  8.08320235,  8.20436764,
        8.06643487, 12.91143491,  6.23262289,  5.84533345,  8.44940661,
       11.46176259,  8.06990483, 13.04217487,  5.39723241, 12.22276278,
        8.85939626,  5.80860165,  6.38399789,  5.64500512,  8.10886021,
       14.02058621,  5.64500512,  5.64865173,  5.18277858,  6.53290663,
        6.3949043 , 13.25017211,  7.39599986, 11.73567759,  8.29203356,
        8.89458273, 13.89671314,  8.09899874, 11.64436684,  6.99097013,
        5.26067019,  7.81509631,  6.00663314,  9.58347456,  7.01475376,
       13.15525803, 11.27160035,  6.67218687,  5.5748624 ,  9.37296102,
        5.5513072 , 13.61600536,  7.24677664, 12.50996828,  6.80559669,
       10.8992524 , 11.31276109, 11.64746882,  6.02779201,  9.09963894,
        8.8353674 , 12.27993098, 12.32545637, 13.20597932, 12.87752052,
        5.39450305,  7.56617037, 10.66631784,  8.36995953,  5.23878371,
        7.06169391, 12.22276278,  5.23878371, 14.10045797,  5.01114719,
        5.36061974, 12.40074901, 10.52357479, 11.28215429, 10.31613269,
        5.91854397,  7.49076899,  7.20482224,  9.09963894,  9.36256096,
       12.22276278, 13.32613445,  9.60359158, 12.77402351,  9.14973319,
        6.77777606, 13.66051884,  7.66377554, 12.4980845 ,  9.64157461,
       12.62711252, 11.52733581,  7.50500064,  5.05444288, 10.07238463,
        8.49161154, 11.04313641, 10.26545792, 13.27994304,  7.48538905,
       11.57249984,  7.73692436,  4.87083348, 12.95702647, 10.4606265 ,
       13.70893492])}
{'P': tensor([[1.4794e-02, 3.3941e-01, 1.6099e-01,  ..., 1.3110e-04, 8.7475e-06,
         1.0236e-04],
        [8.0656e-04, 2.1408e-02, 2.3863e-01,  ..., 1.7429e-04, 1.2569e-05,
         1.3829e-04],
        [7.1882e-04, 8.0770e-04, 2.4553e-02,  ..., 1.1990e-04, 1.4994e-05,
         1.1088e-04],
        ...,
        [5.1368e-02, 4.9582e-02, 3.2358e-02,  ..., 7.5897e-02, 2.8217e-05,
         6.5180e-04],
        [7.8302e-04, 7.5665e-04, 6.7035e-04,  ..., 4.5382e-04, 1.3704e-01,
         1.8146e-02],
        [1.0262e-02, 1.0563e-02, 1.4048e-02,  ..., 5.5400e-04, 2.2685e-04,
         7.8333e-02]]), 'X': tensor([[-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0157, 0.0667,  ..., 0.2818, 0.6477, 0.5588],
        [0.0157, 0.0000, 0.0536,  ..., 0.2923, 0.6513, 0.5602],
        [0.0667, 0.0536, 0.0000,  ..., 0.3452, 0.6320, 0.5345],
        ...,
        [0.2818, 0.2923, 0.3452,  ..., 0.0000, 0.8463, 0.7827],
        [0.6477, 0.6513, 0.6320,  ..., 0.8463, 0.0000, 0.1226],
        [0.5588, 0.5602, 0.5345,  ..., 0.7827, 0.1226, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.8575648 ,  7.8788965 ,  8.01009083,  8.10886021,  8.09984231,
        8.23525433,  8.09842704,  4.978704  , 11.64436684,  6.42689705,
        4.78002665,  5.27178363,  9.54154096,  8.04718433,  5.9510489 ,
        6.38968357,  6.29034958,  5.14999799,  4.85995002,  9.26455795,
       11.85743425, 11.73397309, 11.785566  ,  7.85919687,  4.96231532,
       13.99048094, 12.30695098,  6.59176018,  5.80312949, 12.67576792,
        5.91408504,  8.87587457, 12.17451685, 13.6660671 ,  7.53063885,
        9.69629   , 13.20597932, 11.04313641,  9.66977802, 10.91805683,
        6.2134961 ,  8.95133961, 13.79676654,  7.27340834, 13.89886469,
       11.77124373, 12.67576792,  7.49076899,  5.19577146, 13.2778378 ,
        5.19577146, 13.84126197,  5.43544626,  7.72415048,  6.67218687,
       10.30972193,  5.05981914, 11.21703654,  6.83083863,  5.40689332,
       10.39942497,  8.62631801,  9.43737523, 13.61801526, 11.42947632,
       13.95472331,  9.11473391, 11.28093507, 13.97857591,  9.94173708,
        7.56617037, 13.96987806,  6.38105015, 13.744387  , 13.66896739,
       13.2778378 ,  8.30125201, 10.36844813,  7.28793663,  8.95133961,
        7.14837514,  6.18668463,  6.39438044, 12.1494023 ,  9.36378438,
        6.41202986, 11.97949273,  5.94909709,  5.60064319,  6.39520397,
       12.91505111,  6.81084724,  7.01475376, 13.86480175, 10.19036081,
        5.27178363, 11.85890664,  9.34061938, 11.81423161,  8.34949383,
        5.07306274,  5.91408504,  7.73692436,  7.27759233,  5.80312949,
        7.32754882])}
{'P': tensor([[2.3671e-02, 3.4479e-01, 2.0585e-01,  ..., 2.6542e-05, 1.7286e-04,
         3.8823e-06],
        [5.0602e-04, 2.5733e-02, 1.5982e-01,  ..., 3.8465e-05, 6.4799e-05,
         4.9938e-06],
        [3.8459e-04, 1.5562e-03, 2.3166e-02,  ..., 5.4420e-05, 3.9716e-05,
         5.0549e-06],
        ...,
        [5.5802e-06, 6.9830e-06, 9.8686e-06,  ..., 1.1064e-01, 1.5342e-06,
         3.5957e-05],
        [1.0002e-02, 1.7690e-02, 2.7985e-02,  ..., 4.5345e-04, 1.5219e-01,
         9.9343e-07],
        [6.3643e-06, 3.7202e-06, 2.9953e-06,  ..., 9.0845e-08, 1.2163e-06,
         8.6513e-03]]), 'X': tensor([[ -0.0410,   7.8591,   9.3457],
        [ -1.5261,   7.8988,   9.3768],
        [ -2.2188,   7.8218,   8.6064],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.5520,   7.4976,   9.1522],
        [ -3.5029,   7.5138,   8.1853],
        [ -2.6485, -10.4151,   0.8766],
        [  9.0652,   9.8597,  -3.5949],
        [  5.4426, -10.1181,  -5.1268],
        [  6.6319,  11.9066,   3.5165],
        [  2.0870,  -4.6877,   9.6823],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.1459,   1.4780,  -4.2403],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.2373,  -4.6506,  -1.7334],
        [ -1.5875,   7.8944,  -9.4286],
        [  5.8450,  -9.9356,   3.0767],
        [  9.2692,   9.6370,   7.0746],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -5.6651,   6.4380,   1.3660],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.3460,   1.4252,   1.2291],
        [  6.3542,   0.6934,  -3.3840],
        [  2.2076,  13.8032,  -1.5099],
        [ -8.8154,   2.4611,   2.5198],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.1142,  -2.7230,  -4.1966],
        [  3.3474,  -4.2162,   5.2455],
        [  1.2728, -11.0372,   9.8121],
        [ -9.1795,  -3.3623,   8.5259],
        [  7.7794,  11.0629,   8.9069],
        [  7.5309,  11.2617,   7.2656],
        [  6.3360,   0.4258,  -9.2509],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -3.0008, -10.2827,  -8.9480],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.3943,   6.5483,   1.0790],
        [  5.3924, -10.1396,  -4.0863],
        [  2.9470,  13.6085,   2.9809],
        [  0.8940, -11.0402,  -0.9783],
        [  3.2809,  13.5061,  -4.0199],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.2159, -10.8166,  -1.0406],
        [ 11.5966,   5.9666,   6.0083],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.5258,  -0.4200,  -9.3493],
        [  3.7916,  -3.9524,  -7.9547],
        [  0.9809,  -4.8145,   6.8706],
        [  5.8836,   3.4315,  -4.0327],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.0277,   4.9589,   7.0175],
        [ 11.2571,   6.7072,   8.3693],
        [  6.1579,  -0.5961,  -6.5595],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.9470,  13.6085,   2.9809],
        [  4.9531,   5.0576,   5.6794],
        [  5.1890,   4.7318,   0.9515],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.2535,  -0.1709,  -9.4102],
        [  8.3791,  10.5409,   5.5590],
        [  3.7121,   6.3117,   0.6738],
        [  4.5325,  -3.3643,   6.7699],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.1637,  13.1898,  -1.9189],
        [  0.8940, -11.0402,  -0.9783],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -9.2974,   0.8371,   2.4852],
        [  2.2659,  13.7894,   6.5436],
        [  7.9141,  -8.6656,   9.6948],
        [  6.3589,   0.8239,  -8.4230],
        [  4.3459,  -3.5327,   0.3667],
        [  2.7558,  -4.4844,   0.0636],
        [  5.6414, -10.0303,   7.8525],
        [  3.2990,  -4.2415,  -7.4857],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.4743,   6.2478,  -2.1100],
        [ -5.1046,  -9.1610,   2.3418],
        [ 11.5083,  -4.1177,   0.2389],
        [  3.1756, -10.8247,   6.8882],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.8093,  -9.4178,   8.4244],
        [ -0.1963,   7.8765,  -1.1371],
        [ -5.6423,   6.4537,  -1.2091],
        [ -3.9858,  -9.8334,  -3.7323],
        [  4.7544,  -3.1425,  -1.5315],
        [  5.9912,   3.1453,  -3.9989],
        [ -8.3717,  -5.4769,  -9.2228],
        [  6.1579,  -0.5961,  -6.5595],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.3549,   0.7096,   6.9198],
        [ -0.5385,   7.9037,   6.3830],
        [  1.3103,  -4.8032,  -0.9300],
        [ -4.1296,   7.2803,  -2.7194],
        [  2.6310, -10.9195,   3.8054],
        [  6.3303,   1.6145,  -8.4083],
        [  6.1737,   2.5305,   3.9049],
        [ -9.3076,   0.7852,  -6.7010],
        [  2.9470,  13.6085,   2.9809],
        [  4.3523, -10.5232,   8.2138]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0562, 0.0636,  ..., 1.1959, 0.5536, 0.2390],
        [0.0562, 0.0000, 0.0114,  ..., 1.1696, 0.6081, 0.1864],
        [0.0636, 0.0114, 0.0000,  ..., 1.1583, 0.6126, 0.1843],
        ...,
        [1.1959, 1.1696, 1.1583,  ..., 0.0000, 1.4377, 1.1741],
        [0.5536, 0.6081, 0.6126,  ..., 1.4377, 0.0000, 0.7923],
        [0.2390, 0.1864, 0.1843,  ..., 1.1741, 0.7923, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.85919687,  8.04483495,  8.1303945 ,  8.28968067,  8.29640613,
        8.29022197, 10.74655346, 13.39372426, 11.4890861 , 13.62896593,
        5.13125513, 13.59673406,  9.26455795,  5.22903117, 12.17901033,
        5.16077764,  8.05242591, 11.52733581, 13.37121923, 12.05300943,
        8.57560275,  8.56889738,  9.79814997,  6.50409372,  6.39187285,
       13.97857591,  9.15253344, 10.31613269,  5.79391825,  5.38342077,
       11.11038551,  9.77588801, 13.52430176, 13.54774355,  6.35029274,
       10.33712258, 10.71162681, 10.6910768 ,  7.37576154, 11.48434767,
       13.92390689, 11.07637531, 13.89886469, 10.41909211, 11.2845608 ,
       13.04156075,  7.83486943, 12.53285633,  5.47702766,  4.91336994,
        6.81117802,  9.48337063,  7.06169391, 13.103709  ,  6.18668463,
        5.01114719,  6.1561646 , 12.17901033, 13.92390689,  7.07902036,
        7.02253235,  7.8575648 ,  9.62238828,  6.25586297, 13.46553623,
        7.32233733,  5.64464807,  5.53175275, 13.83139024, 11.07637531,
        9.85330646,  9.3349858 , 13.97430191, 11.73567759,  6.41202986,
        5.60064319,  5.26342203, 11.50791859,  5.37343173, 12.07225469,
       13.01728313,  6.1561646 , 13.06498554, 10.48721382, 12.22276278,
       11.28093507,  6.19633139, 11.62158554,  7.8788965 ,  8.5724042 ,
       10.61047596,  5.69913817,  6.76661595, 10.00410625,  6.18668463,
       10.63038565,  6.39438044,  7.92200342,  4.978704  ,  8.36995953,
       11.23201619,  6.53290663,  6.67218687,  9.34061938, 13.92390689,
       11.38774149])}
{'P': tensor([[1.5577e-02, 1.7363e-01, 1.6574e-01,  ..., 1.1072e-06, 2.8330e-06,
         1.6409e-06],
        [5.3585e-04, 1.6289e-02, 3.3465e-01,  ..., 1.2677e-06, 2.0850e-06,
         1.7368e-06],
        [4.7180e-04, 6.3065e-04, 2.5440e-02,  ..., 2.0626e-06, 2.4097e-06,
         2.9370e-06],
        ...,
        [2.1009e-05, 1.5190e-05, 1.1191e-05,  ..., 2.5122e-02, 1.1165e-03,
         7.2636e-05],
        [3.1718e-03, 2.9369e-03, 2.2579e-03,  ..., 1.9288e-06, 9.9510e-02,
         4.3879e-07],
        [1.9796e-05, 1.2404e-05, 9.3516e-06,  ..., 3.4012e-04, 3.3111e-05,
         2.6588e-02]]), 'X': tensor([[ -0.1963,   7.8765,  -1.1371],
        [ -1.2454,   7.9127,  -0.5715],
        [ -2.0445,   7.8469,  -0.5857],
        [ -1.9714,   7.8563,  -1.8084],
        [ -3.0644,   7.6439,  -1.4322],
        [ -3.2486,   7.5925,  -1.3719],
        [  0.2435,   7.8191,  -6.4709],
        [ 11.3118,   6.5957,  -4.5667],
        [  0.4119,  14.1019,  -4.1870],
        [ 10.6673,  -5.6649,  -4.1646],
        [  0.9392,  -4.8143,  -3.4649],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.1752,   2.5242,   3.5874],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.6248,   3.9984,   8.4933],
        [  2.1440, -10.9811,   8.5281],
        [  6.0047,   3.1062,  -6.7914],
        [  5.5158,  -2.1337,   3.5246],
        [ -0.5709,   7.9055,   7.6089],
        [ -8.6112,   2.9319,  -7.9943],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.9461,  -4.4077,  -0.4831],
        [  0.1041, -11.0045,   2.4311],
        [  6.3246,   1.6696,  -0.1544],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.6414, -10.0303,   7.8525],
        [  2.7976,  -4.4682,  -4.2020],
        [  3.2701,   6.6338,   9.9209],
        [ -8.7544,   2.6096,   3.3146],
        [  5.1667,  -2.6543,   1.2879],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -3.6381,   7.4683,   7.3544],
        [  6.1989,  -9.7591,   0.8289],
        [  7.8425,  -8.7202,   6.4530],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.4732,  -2.2041,  -4.3830],
        [ -0.9715, -10.8630,   0.3555],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.1960,  -3.2964,   1.6921],
        [  9.0004,  -7.7195,   4.6945],
        [ -9.1706,   1.3863,   8.4032],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.1795,  -3.3623,   8.5259],
        [ -4.4503,   7.1378,   3.3751],
        [  4.5954,  13.0099,  -4.4439],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.6249, -10.7255,   5.6605],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.4290,  -3.4596,   1.4181],
        [  7.9816,  10.8938,  -0.5378],
        [ -9.1658,  -3.4155,  -2.8880],
        [  4.0421,  -3.7765,  -8.3631],
        [  2.4932,   7.0903,  -5.4214],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.1666,   3.7653,   3.6075],
        [ -6.1914,  -8.2997,   8.5380],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -4.7334,   6.9982,  -0.4392],
        [  0.1460,   7.8340,  -9.4614],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.2790,  -0.0259,  -5.4116],
        [  7.8982,  -8.6778,   0.0239],
        [  5.0506,   4.9277,  -9.5094],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -8.0766,  -6.0046,  -3.1873],
        [  3.5939,   6.4029,  -1.8357],
        [ -6.3066,  -8.1936,   9.5822],
        [  0.7145,  -4.8071,  -2.3766],
        [ -8.0087,   4.0173,   9.1091],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -1.7290,   7.8825,  -3.3831],
        [  9.3753,   9.5171,   3.9131],
        [ -7.4887,   4.7364,   8.9074],
        [  2.1440, -10.9811,   8.5281],
        [  1.3767, -11.0341,   9.9415],
        [  0.8700,  -4.8132,  -7.8042],
        [  9.3624,   9.5318,   0.2779],
        [  5.3860,  -2.3418,   1.0826],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.6681,  -4.8556,   9.2518],
        [  7.9403,  -8.6455,  -8.7433],
        [ -3.0644,   7.6439,  -1.4322],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.3118,   4.5452,   4.2114],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.3062,  -5.6008,   0.0287],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.2975,   0.0947,   5.1791],
        [  6.7792,  11.8084,   1.6251],
        [  6.7792,  11.8084,   1.6251],
        [ -5.9987,   6.1949,  -6.2114],
        [  1.5653,  13.9377,   4.1773],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.8994,   6.8675,  -9.5987],
        [ -1.6086,   7.8928,   4.3385],
        [ -5.9938,  -8.4745,   3.9735],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.4572,   7.1085,   7.7376],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 11.0438,   7.1180,  -5.6991],
        [  7.5784,  -8.9141,   5.8410],
        [ 10.9931,   7.2107,   7.7933],
        [  5.9843,  -9.8679,  -6.1449]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0536, 0.0808,  ..., 0.6335, 0.9972, 0.4062],
        [0.0536, 0.0000, 0.0321,  ..., 0.5901, 0.9876, 0.4339],
        [0.0808, 0.0321, 0.0000,  ..., 0.5841, 1.0047, 0.4316],
        ...,
        [0.6335, 0.5901, 0.5841,  ..., 0.0000, 0.6953, 1.0110],
        [0.9972, 0.9876, 1.0047,  ..., 0.6953, 0.0000, 1.3921],
        [0.4062, 0.4339, 0.4316,  ..., 1.0110, 1.3921, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.8788965 ,  8.01009083,  8.10886021,  8.09984231,  8.23525433,
        8.25828409,  7.822854  , 13.09426142, 14.10796466, 12.07818871,
        4.90505566,  5.39450305,  6.67121748, 12.59969936,  9.69629   ,
        6.90115901, 11.18839169,  6.76056625,  5.91408504,  7.92607342,
        9.09661212, 11.75807252,  5.30158497, 11.0050379 ,  6.54127431,
       12.41012046, 11.50791859,  5.27178363,  7.39599986,  9.13507321,
        5.80860165, 10.31886066,  6.24614905,  9.79039371,  8.30729352,
       11.56146873, 11.72802491, 11.81423161,  5.90035324, 10.90638066,
        5.91854397,  9.76896904, 11.85743425,  9.27474386, 12.22276278,
        9.77588801,  8.41148722, 13.79762415, 10.8207297 , 11.32149259,
       12.70489563,  5.62005629, 13.50484539,  9.78147542,  5.53175275,
        7.5158489 ,  4.91038884,  8.99276564, 10.35464916, 10.06437598,
        8.4486771 ,  7.83534987,  6.00732193,  6.27905292, 11.73397309,
        7.05627461, 10.60226614, 10.06407839,  7.34250948, 10.33958639,
        4.85995002,  8.95983586, 10.53368644,  8.06990483, 13.35927527,
        8.86082124, 11.18839169, 11.11969815,  4.89121805, 13.3607379 ,
        5.87305005,  5.38342077,  9.9353921 , 11.73848065,  8.23525433,
       11.75807252,  6.99097013, 11.1913598 , 10.01804223, 10.1371552 ,
        6.29821673, 13.61600536, 13.61600536,  8.62329466, 14.02532758,
        9.85384623,  7.4544866 ,  8.0550286 , 10.37998078,  7.98469166,
        7.52117132, 10.51437947, 13.13888841, 11.70016217, 13.14690137,
       11.54071859])}
{'P': tensor([[1.7244e-02, 2.0037e-01, 1.5691e-01,  ..., 3.0448e-05, 2.5562e-03,
         1.2572e-05],
        [2.8749e-04, 1.1223e-02, 1.8055e-01,  ..., 1.5847e-05, 1.8553e-03,
         4.4153e-06],
        [4.5039e-04, 7.6742e-04, 3.5940e-02,  ..., 4.1415e-05, 3.5374e-03,
         7.3197e-06],
        ...,
        [1.0748e-02, 6.7789e-03, 4.0408e-03,  ..., 9.1977e-02, 1.6286e-03,
         3.3865e-03],
        [3.2735e-04, 7.1173e-04, 1.2964e-03,  ..., 3.9840e-05, 5.0330e-02,
         9.9368e-06],
        [6.6926e-04, 8.8081e-04, 6.9751e-04,  ..., 8.3263e-06, 3.6255e-03,
         1.9527e-02]]), 'X': tensor([[ -0.2586,   7.8825,  -9.9840],
        [ -1.5875,   7.8944,  -9.4286],
        [ -2.8162,   7.7059,  -9.6885],
        [ -2.5161,   7.7699,  -9.3531],
        [ -3.5173,   7.5091,  -8.9585],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.6928,  12.4710,   3.6001],
        [  9.3624,   9.5318,   0.2779],
        [  4.2275,  -3.6320,  -0.2714],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.1379,   4.4125,  -3.8833],
        [ 12.5770,   1.8512,  -0.6578],
        [  4.8051, -10.3703,   0.5268],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.4541,  -0.3327,   8.8246],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.8154,   2.4611,   2.5198],
        [  9.5261,  -7.1675,   1.4206],
        [ 12.4837,   2.7510,   0.4449],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.9515,  -3.8425,   6.1729],
        [  6.3305,   1.6119,   1.1538],
        [  5.3860,  -2.3418,   1.0826],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.4430,  -0.1897,   4.8383],
        [ 12.3357,   3.6001,   0.7903],
        [  4.5954,  13.0099,  -4.4439],
        [ -8.9636,   2.0648,   8.1946],
        [  5.7845,   3.6652,   0.9175],
        [ 10.3251,   8.2928,  -3.0006],
        [  2.0870,  -4.6877,   9.6823],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.7301,   7.4262,   5.0522],
        [  7.0171,  11.6439,  -5.0178],
        [  1.5662,  13.9375,   4.5142],
        [  2.1771, -10.9776,  -3.2007],
        [  3.3069,   6.6089,   8.7377],
        [ -6.4464,   5.8272,   7.5038],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.8201,   3.5841,  -8.0609],
        [ -2.1817,   7.8274,  -4.6298],
        [  4.3377,   5.7570,   1.3712],
        [  0.5600, -11.0321,  -0.8660],
        [ -7.7239,   4.4298,   3.1216],
        [ -6.3066,  -8.1936,   9.5822],
        [ 10.5955,   7.8824,  -3.7006],
        [  2.0729,   7.2878,   7.0880],
        [  9.0051,   9.9234,   5.4411],
        [  8.5198,  10.4091,  -8.6486],
        [ -1.6906,   7.8860,   7.7142],
        [ -8.3539,  -5.5111,   0.3373],
        [ -6.3306,   5.9272,   5.6935],
        [  2.8994,   6.8675,  -9.5987],
        [  4.1637,  13.1898,  -1.9189],
        [  3.1480,  -4.3165,  -7.8002],
        [ -9.4739,  -1.2497,   0.5483],
        [  5.6022,  -1.9833,   8.3081],
        [  9.5261,  -7.1675,   1.4206],
        [ -7.3333,   4.9246,  -7.9212],
        [  4.1537,   5.9340,  -6.0423],
        [ -3.7964,  -9.9294,   7.7147],
        [  0.4996,   7.7738,  -4.4711],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.1730,  -0.5375,  -8.8861],
        [  1.7228,  -4.7575,   8.3396],
        [ -4.7847,  -9.3729,  -3.7741],
        [  2.5813,  -4.5471,   5.3597],
        [  3.4604,  -4.1545,   1.8091],
        [  5.1203,  -2.7151,   6.2110],
        [  6.0890,  -0.8382,   2.9166],
        [  6.3549,   0.7096,   6.9198],
        [  0.5600, -11.0321,  -0.8660],
        [  9.0129,  -7.7072,   9.1400],
        [  2.5829,  -4.5466,  -8.4015],
        [  4.0362,  13.2397,  -7.0030],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -8.5149,  -5.1905,   1.5239],
        [  6.1222,  -0.7261,   8.8204],
        [  8.0877,  -8.5297,  -1.1050],
        [ -6.6895,   5.6051,   2.7899],
        [  1.1274,  14.0116,   8.1753],
        [ -5.3736,   6.6310,  -1.5174],
        [  9.3607,   9.5337,  -4.5412],
        [  5.0190,   4.9705,  -7.9752],
        [  5.6928,  12.4710,   3.6001],
        [  7.2340,  -9.1505,  -5.8321],
        [  7.0358,  -9.2784,  -9.8479],
        [ -9.4773,  -1.0024,   5.2599],
        [ 11.6544,   5.8273,  -3.1375],
        [ -7.4808,   4.7462,   2.0266],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.4592,  -0.9168,  -6.7571],
        [  7.5930,  -8.9037,   7.2323],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 11.0275,  -5.0666,   8.4154],
        [ -3.9700,   7.3453,  -4.1486],
        [  1.5383, -11.0274,  -4.9666],
        [ -5.3736,   6.6310,  -1.5174],
        [ -0.8675, -10.8815,   6.8561],
        [  9.0129,  -7.7072,   9.1400],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.3535,   0.6792,  -7.5101],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.9284,  11.7061,  -0.3732]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.7880e-02, 9.7805e-02,  ..., 3.3841e-01, 3.2061e-01,
         7.0402e-01],
        [2.7880e-02, 2.1453e-04, 7.0121e-02,  ..., 3.6607e-01, 3.1699e-01,
         7.2883e-01],
        [9.7805e-02, 7.0121e-02, 0.0000e+00,  ..., 4.3420e-01, 3.0885e-01,
         7.8750e-01],
        ...,
        [3.3841e-01, 3.6607e-01, 4.3420e-01,  ..., 0.0000e+00, 4.7611e-01,
         4.0366e-01],
        [3.2061e-01, 3.1699e-01, 3.0885e-01,  ..., 4.7611e-01, 0.0000e+00,
         6.7903e-01],
        [7.0402e-01, 7.2883e-01, 7.8750e-01,  ..., 4.0366e-01, 6.7903e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.88677538,  8.05242591,  8.20436764,  8.16714836,  8.29203356,
        8.30125201, 13.70893492, 13.3607379 ,  5.57341623,  9.11473391,
       12.91505111, 12.7125079 , 11.42947632, 10.794755  ,  9.45995955,
       10.8992524 ,  9.15253344, 11.92133129, 12.78327037,  9.47369628,
        5.51177501,  6.53250937,  5.87305005, 12.35625694,  9.44486609,
       12.85032836, 13.79762415,  9.19837792,  6.84796202, 13.2430377 ,
        5.13125513,  8.30125201,  7.62509582, 13.59480906, 14.02526505,
       11.1913598 ,  7.39004226,  8.68979052,  4.74468406,  6.83515201,
        8.12580976,  7.20826219, 11.04629471,  8.90405823, 10.33958639,
       13.20597932,  7.57685908, 13.40024224, 13.45124917,  8.06517143,
       10.00794468,  8.67227386,  7.4544866 , 13.83139024,  5.34250225,
        9.55593374,  5.94293185, 11.92133129,  8.83341608,  7.24326992,
       10.63038565,  7.78980299,  9.77588801,  6.19633139,  5.05981914,
       10.52357479,  5.2287058 ,  5.40689332,  5.79562691,  6.1463896 ,
        6.39438044, 11.04629471, 11.85890664,  5.22903117, 13.84126197,
       10.61047596,  9.97220799,  6.16512856, 11.75438006,  8.7273569 ,
       14.05687815,  8.53501777, 13.36092624,  7.06372172, 13.70893492,
       11.66453059, 11.64436684,  9.53015376, 13.03002338,  8.859416  ,
       12.5769659 , 12.49291768, 11.70168755,  5.39098669,  9.51074763,
       10.37698435, 12.13568598,  8.34949383, 11.1341798 ,  8.53501777,
       10.91601655, 11.85890664,  8.20436764,  6.38968357,  8.36995953,
       13.60275211])}
{'P': tensor([[1.2881e-02, 1.7125e-01, 1.7119e-01,  ..., 3.3926e-04, 4.6871e-02,
         1.5395e-04],
        [4.1843e-04, 1.6283e-02, 6.5015e-02,  ..., 2.0871e-04, 8.8292e-02,
         2.6394e-04],
        [7.1328e-04, 4.7565e-03, 3.7648e-02,  ..., 3.5770e-04, 9.1089e-02,
         8.8056e-04],
        ...,
        [1.3042e-01, 9.8371e-02, 1.3298e-01,  ..., 4.3634e-02, 3.3345e-02,
         3.6658e-04],
        [5.5948e-04, 1.0554e-03, 1.9919e-03,  ..., 2.2756e-04, 4.5691e-02,
         8.5404e-04],
        [6.9353e-06, 7.3999e-06, 8.6676e-06,  ..., 8.1605e-06, 8.0165e-06,
         3.5196e-02]]), 'X': tensor([[-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0445, 0.0705,  ..., 0.1187, 0.1733, 0.8914],
        [0.0445, 0.0000, 0.0413,  ..., 0.1552, 0.1292, 0.8715],
        [0.0705, 0.0413, 0.0000,  ..., 0.1537, 0.1151, 0.8304],
        ...,
        [0.1187, 0.1552, 0.1537,  ..., 0.0000, 0.2686, 0.8724],
        [0.1733, 0.1292, 0.1151,  ..., 0.2686, 0.0000, 0.8072],
        [0.8914, 0.8715, 0.8304,  ..., 0.8724, 0.8072, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.89745107,  8.04718433,  8.09842704,  8.12043983,  8.06517143,
        8.11227668, 10.6910768 ,  6.81084724, 11.63383221, 12.84056604,
        5.23878371, 11.36089355,  7.5403822 ,  6.1561646 ,  5.16077764,
       11.8159254 ,  7.69752262, 11.9702762 ,  5.19065439,  6.3949043 ,
       12.11735334,  7.09887557,  8.68832398,  9.97220799, 13.60275211,
        4.74468406, 12.84056604,  5.29009836,  7.28793663,  8.62260327,
        7.24326992, 13.66051884,  9.26455795, 10.36844813,  5.36061974,
        5.47866221,  5.35774573,  8.83341608,  9.0368458 , 14.08549802,
        9.3349858 ,  5.19065439, 13.18278606,  8.12580976,  7.16058587,
       12.30858677, 11.64436684,  7.53063885,  8.05668609, 13.09426142,
       11.97957425, 13.79676654, 13.744387  ,  7.16598273, 11.28895489,
        7.6374603 ,  7.02253235,  7.27340834,  9.88310614, 12.40074901,
       13.1399181 ,  7.48538905,  7.49076899,  8.53014915,  9.87720102,
       12.62576026, 13.89974693,  7.27340834,  6.76661595, 13.59480906,
        7.16747878,  5.14999799,  5.86519525,  7.42512621,  9.94173708,
       13.60275211,  7.73151309, 10.06286074,  8.29640613,  7.02326515,
        5.23878371, 12.97370371,  5.14371005,  9.14044984,  8.8580671 ,
       14.02058621, 11.38774149, 12.16901615, 13.41713661,  9.24501829,
        5.21280156,  5.56809335, 13.12299145, 12.95702647, 11.36089355,
        9.3109884 , 10.33712258, 10.91601655,  5.16611782,  9.9353921 ,
        6.46356806,  7.53063885, 13.2778378 ,  7.61088125,  8.28968067,
        9.60359158])}
{'P': tensor([[1.7278e-02, 2.4011e-01, 2.0697e-01,  ..., 6.8001e-06, 6.3621e-05,
         5.3360e-04],
        [6.7475e-04, 3.4722e-02, 8.4281e-03,  ..., 1.2772e-05, 1.5503e-04,
         2.1962e-03],
        [9.1008e-04, 1.2547e-01, 3.9083e-02,  ..., 1.3361e-05, 1.7662e-04,
         2.4430e-03],
        ...,
        [2.5087e-04, 1.6563e-04, 1.5626e-04,  ..., 1.6570e-01, 1.9074e-04,
         1.5106e-05],
        [8.6040e-04, 6.5208e-04, 6.8220e-04,  ..., 5.6236e-04, 2.0510e-01,
         3.9856e-04],
        [1.5332e-05, 1.8059e-05, 1.8062e-05,  ..., 3.1472e-05, 4.2284e-03,
         5.0374e-02]]), 'X': tensor([[ -0.5385,   7.9037,   6.3830],
        [ -2.0721,   7.8432,   6.6380],
        [ -1.9600,   7.8577,   6.9121],
        [ -2.7572,   7.7194,   5.9580],
        [ -2.5056,   7.7720,   5.5878],
        [ -3.1671,   7.6158,   6.0765],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.0467,  -9.8369,  -6.5816],
        [ -2.6735, -10.4062,  -1.5911],
        [  1.8766,   7.3694,   1.1887],
        [ 12.6038,   0.7665,  -3.4439],
        [  0.9074,  -4.8139,   8.1674],
        [  5.8325,  -1.5222,  -2.1341],
        [  2.2659,  13.7894,   6.5436],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.5428,  -9.5728,  -2.9897],
        [  0.2435,   7.8191,  -6.4709],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.1685,  12.1979,  -6.2473],
        [ -1.7009,   7.8851,  -8.2437],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3569,   0.7591,   6.5011],
        [ -9.4117,   0.1160,   8.2574],
        [  1.0000,   7.6591,  -4.5011],
        [  9.6662,   9.1725,   8.1056],
        [  7.8375,  -8.7240,  -9.8913],
        [  5.1861,   4.7361,  -3.7312],
        [  2.3928,   7.1404,  -1.7035],
        [  4.7813,  -3.1139,   8.2691],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.0330,  -6.0765,   6.7945],
        [  5.6523,   3.9443,  -2.9449],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.2883,   0.0325,   5.2692],
        [ -7.9298,   4.1367,  -3.9773],
        [  7.0642,  11.6104,  -9.8316],
        [  0.3354,  -4.7708,  -4.9125],
        [  8.6420,  -8.0580,   6.5781],
        [ 12.3609,   3.4769,  -8.5762],
        [ -9.4159,  -2.0739,   9.6841],
        [  2.2659,  13.7894,   6.5436],
        [ -8.8400,   2.3991,   1.3457],
        [ -1.6275, -10.7221,   3.2376],
        [ -5.5525,   6.5146,  -4.4984],
        [ -9.1178,   1.5784,   2.1410],
        [  8.0381,  -8.5690,  -8.9523],
        [  5.1886,  -2.6250,   6.4677],
        [ -8.9636,   2.0648,   8.1946],
        [  1.3767, -11.0341,   9.9415],
        [  6.9314,  -9.3435,  -1.2729],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.0750,  -6.0072,  -5.8267],
        [  3.2809,  13.5061,  -4.0199],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.2895,   0.0405,   5.7652],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.1222,  -0.7261,   8.8204],
        [  2.7590,  13.6621,   9.4496],
        [  0.1532,  -4.7422,  -7.0145],
        [  1.6308,  13.9254,  -2.0984],
        [  5.6928,  12.4710,   3.6001],
        [  2.4639, -10.9431,  -7.3718],
        [ 10.1798,   8.5002,  -5.8893],
        [ -9.0961,   1.6529,   3.1553],
        [  3.7237,  -3.9966,  -5.0376],
        [ -6.8618,  -7.6322,   3.7027],
        [  5.1196,  -2.7160,   4.2705],
        [ -7.0948,   5.1940,   9.8171],
        [ -8.6229,   2.9068,   2.4429],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.3549,   0.7096,   6.9198],
        [  6.2576,   2.1359,  -3.3398],
        [ -2.2508, -10.5479,  -2.7235],
        [  3.3474,  -4.2162,   5.2455],
        [  1.3403,  -4.8011,  -1.1698],
        [  4.2275,  -3.6320,  -0.2714],
        [  1.1994,   7.6035,   2.4295],
        [  4.9152,  12.8653,  -1.4459],
        [  6.0305,   3.0296,   0.1375],
        [ -7.8359,   4.2735,   7.2943],
        [ -1.0482, -10.8487,  -5.9066],
        [  8.8474,  10.0868,  -8.8644],
        [  6.2247,  -0.3150,  -9.3657],
        [  8.3791,  10.5409,   5.5590],
        [ -7.2986,  -7.1225,  -3.0814],
        [  6.0310,  12.2795,  -6.4528],
        [  0.1041, -11.0045,   2.4311],
        [  5.9843,  -9.8679,  -6.1449],
        [ -8.8934,   2.2597,   3.5831],
        [  0.7685,  -4.8098,  -2.9310],
        [ -1.8367,   7.8718,  -8.6523],
        [ 10.0995,  -6.4739,   2.0974],
        [  4.4049,   5.6892,  -4.4394],
        [ -5.9987,   6.1949,  -6.2114],
        [  6.1755,   2.5233,  -2.0042],
        [  2.6613,   7.0021,   4.5783],
        [  4.5954,  13.0099,  -4.4439],
        [  6.8407,  11.7666,   3.0045],
        [  4.4633,   5.6287,  -0.7754],
        [  3.7237,  -3.9966,  -5.0376],
        [  0.5850,  -4.7982,   9.6639],
        [ -9.4159,  -2.0739,   9.6841]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0658, 0.0633,  ..., 0.8043, 0.1890, 0.6645],
        [0.0658, 0.0000, 0.0143,  ..., 0.8319, 0.2501, 0.5987],
        [0.0633, 0.0143, 0.0000,  ..., 0.8424, 0.2422, 0.6024],
        ...,
        [0.8043, 0.8319, 0.8424,  ..., 0.0000, 0.8342, 1.2388],
        [0.1890, 0.2501, 0.2422,  ..., 0.8342, 0.0000, 0.8371],
        [0.6645, 0.5987, 0.6024,  ..., 1.2388, 0.8371, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.92200342,  8.11227668,  8.09842704,  8.19703512,  8.16584537,
        8.24808832, 10.99326577,  9.95552626, 11.54672755, 10.74409834,
        7.60463349, 12.62711252,  4.89869205,  6.02789634, 13.97430191,
        9.58072286, 11.59512194,  7.822854  , 12.32545637,  7.6374603 ,
       12.91143491, 13.66896739,  8.06643487, 10.06286074, 13.36092624,
        6.4020404 ,  9.41244865,  7.72415048, 13.32557133, 11.72749923,
        7.02326515,  7.53063885,  5.70591986,  8.01009083, 10.07238463,
        6.89245352,  5.19577146,  6.28835101,  8.94395628, 13.59056613,
        4.78258092, 11.8159254 , 12.84056604,  9.64157461, 13.97430191,
        9.15976746, 10.84493732,  8.55982105,  9.25336371, 11.74901778,
        5.81481044,  9.19837792, 11.11969815, 11.63383221,  9.3109884 ,
       10.06437598, 13.89886469, 10.00802627,  6.28961988,  4.90505566,
        6.16512856, 13.93790235,  4.74468406, 14.02058621, 13.70893492,
       11.21703654, 13.26209545,  9.24501829,  5.46244988, 10.26327895,
        5.79543038,  8.79284466,  9.09963894, 12.41366754,  6.39438044,
        6.61211224, 10.78533838,  5.38342077,  4.98461983,  5.57341623,
        7.69752262, 13.77222752,  6.74872981,  8.92549752, 10.8992524 ,
       13.41713661,  6.23262289, 13.46553623, 10.1979669 , 13.68063098,
       11.0050379 , 11.54071859,  9.17595899,  4.87083348,  8.08320235,
       11.99633819,  7.19513806,  8.62329466,  6.67108662,  7.49076899,
       13.79762415, 13.6105595 ,  7.18354885,  5.46244988,  4.83371341,
        9.64157461])}
{'P': tensor([[1.3196e-02, 2.4411e-01, 2.2621e-01,  ..., 5.5326e-06, 5.1864e-05,
         8.8221e-06],
        [4.8660e-04, 2.0432e-02, 2.9393e-01,  ..., 8.8468e-06, 1.1060e-04,
         8.2983e-06],
        [8.6860e-04, 2.2936e-03, 3.8363e-02,  ..., 1.6250e-05, 2.1050e-04,
         1.4534e-05],
        ...,
        [1.3229e-04, 1.0944e-04, 1.0328e-04,  ..., 1.4329e-01, 1.6601e-04,
         1.1485e-04],
        [1.3098e-06, 1.4083e-06, 1.3999e-06,  ..., 4.9840e-04, 1.2255e-02,
         1.1883e-07],
        [7.2992e-03, 5.4051e-03, 5.1366e-03,  ..., 8.2288e-07, 1.4446e-06,
         1.5325e-02]]), 'X': tensor([[ -0.5709,   7.9055,   7.6089],
        [ -1.5451,   7.8975,   7.6104],
        [ -1.6906,   7.8860,   7.7142],
        [ -2.1382,   7.8339,   7.3524],
        [ -1.9600,   7.8577,   6.9121],
        [ -2.0721,   7.8432,   6.6380],
        [ -4.2271,   7.2387,  -1.8316],
        [ 11.2571,   6.7072,   8.3693],
        [  0.6363, -11.0348,   6.1414],
        [  9.7328,   9.0902,  -0.8917],
        [  6.1741,  -0.5329,   3.1913],
        [  6.3031,   1.8477,  -7.1706],
        [  3.7916,  -3.9524,  -7.9547],
        [  8.7131,  10.2216,   7.7915],
        [  1.3403,  -4.8011,  -1.1698],
        [ -8.4824,  -5.2575,  -0.0795],
        [  9.5203,   9.3483,   5.8781],
        [  0.3354,  -4.7708,  -4.9125],
        [  9.3624,   9.5318,   0.2779],
        [  1.7228,  -4.7575,   8.3396],
        [ -2.0840,   7.8415,  -7.8657],
        [  1.8320,   7.3871,   5.1452],
        [  9.9058,  -6.7203,   5.6928],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.2435,   4.0076,   8.1257],
        [  5.6928,  12.4710,   3.6001],
        [  3.6469,   6.3624,  -1.8414],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -5.3382,   6.6533,   7.8054],
        [  5.6314,  -1.9302,  -9.6543],
        [  4.0362,  13.2397,  -7.0030],
        [ 11.6634,   5.8050,   9.1590],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -5.1377,   6.7751,  -0.4438],
        [  8.1528,  10.7452,   9.1493],
        [  2.5469,  13.7192,  -9.6935],
        [  9.6816,   9.1536,  -5.8543],
        [ -7.9667,   4.0814,  -0.6397],
        [ -9.1960,  -3.2964,   1.6921],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.3717,  -5.4769,  -9.2228],
        [  3.2990,  -4.2415,  -7.4857],
        [  7.1441,  11.5529,   2.3232],
        [  7.7794,  11.0629,   8.9069],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.1658,  -3.4155,  -2.8880],
        [  4.5767,   5.5071,   3.9069],
        [  6.3569,   0.7591,   6.5011],
        [  0.8700,  -4.8132,  -7.8042],
        [ -2.6485, -10.4151,   0.8766],
        [ -2.9497, -10.3028,  -7.3479],
        [  5.1142,  -2.7230,  -4.1966],
        [ -8.0611,  -6.0302,   4.6962],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.8093,  -9.4178,   8.4244],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.4267,  -5.3697,   2.4172],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.3119,   1.7797,   9.9186],
        [  1.7218,  -4.7576,  -3.3617],
        [  6.2419,  -0.2313,   4.2440],
        [ -2.3317,   7.8034,  -3.6274],
        [ -9.4309,  -0.0599,  -4.0830],
        [  1.4749,  -4.7892,  -0.8089],
        [ 12.2328,  -2.0376,  -1.1234],
        [  6.3119,   1.7797,   9.9186],
        [ 12.5224,  -0.4497,   4.8448],
        [  0.6142,  14.0802,   0.5407],
        [  8.1043,  -8.5164,  -7.9290],
        [ 11.6544,   5.8273,  -3.1375],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.1440, -10.9811,   8.5281],
        [  5.7784,  -1.6401,   9.5587],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.3048,   0.1469,  -9.1691],
        [ -3.2053, -10.1993,  -3.8736],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -5.2589,   6.7024,  -5.5809],
        [  3.9085,   6.1512,  -2.2560],
        [ 11.1420,   6.9331,  -2.4779],
        [ -9.2345,  -3.1345,   3.9715],
        [ 12.5224,  -0.4497,   4.8448],
        [  2.0937,  13.8293,  -2.3533],
        [ 12.0772,  -2.6014,  -7.1832],
        [  3.9770, -10.6341,   8.0783],
        [ 10.0400,   8.6922,  -6.2847],
        [  5.5424,  -2.0885,   7.2487],
        [  5.4426, -10.1181,  -5.1268],
        [  0.9036,   7.6840,   9.9237],
        [  4.8432,   5.1968,   1.5006],
        [  3.7992,  -3.9474,  -8.9019],
        [ -7.8359,   4.2735,   7.2943],
        [  5.2608,  12.6977,   4.9166],
        [ 12.6035,   0.7494,   7.1536],
        [ -7.4887,   4.7364,   8.9074],
        [ -1.6220,   7.8917,   0.0315],
        [ -9.4196,   0.0478,   9.4140],
        [ -0.2586,   7.8825,  -9.9840],
        [ -6.6998,   5.5954,   0.0823],
        [  6.3236,   0.3011,  -4.6668],
        [  2.5375,   7.0676,  -4.3490],
        [ -6.2668,   5.9808,   2.5614],
        [  2.4149,  -4.6005,  -4.3339],
        [ -9.2301,  -3.1537,  -2.7477],
        [  8.7131,  10.2216,   7.7915]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0342, 0.0507,  ..., 0.8168, 1.0765, 0.6522],
        [0.0342, 0.0000, 0.0165,  ..., 0.8266, 1.0547, 0.6837],
        [0.0507, 0.0165, 0.0000,  ..., 0.8330, 1.0456, 0.6984],
        ...,
        [0.8168, 0.8266, 0.8330,  ..., 0.0000, 0.9126, 1.1170],
        [1.0765, 1.0547, 1.0456,  ..., 0.9126, 0.0000, 1.6871],
        [0.6522, 0.6837, 0.6984,  ..., 1.1170, 1.6871, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.92607342,  8.04718433,  8.06517143,  8.12043983,  8.09842704,
        8.11227668,  8.38252937, 13.103709  , 11.05317597, 13.31764181,
        6.19707944,  6.56833864,  5.47702766, 13.43127131,  4.98461983,
        9.97964013, 13.3426541 ,  4.78258092, 13.3607379 ,  5.05981914,
        8.1137452 ,  7.61088125, 11.9702762 ,  9.23178204, 12.88270018,
       13.70893492,  7.33350642,  9.43737523,  8.53014915,  5.95297459,
       13.84126197, 13.02818702,  9.93553424,  8.50278896, 13.48809595,
       13.95361098, 13.32374056,  8.95133961,  9.76896904, 10.52357479,
       10.00410625,  5.37343173, 13.58333857, 13.52430176, 12.22276278,
        9.78147542,  7.16058587,  6.4020404 ,  4.89121805, 10.74655346,
       10.7167285 ,  5.79391825, 10.06703358,  8.16714836, 11.62158554,
        8.04664886,  9.99211663, 13.13205357,  6.55800267,  5.05963217,
        6.24614905,  8.14433542,  9.43112957,  5.01114719, 12.4013153 ,
        6.55800267, 12.53047489, 14.09357547, 11.75618513, 13.03002338,
       10.61417181, 11.18839169,  6.00663314, 10.42667093,  6.30647746,
       10.6910768 , 12.71843462,  9.86089579,  8.51928531,  7.28793663,
       13.12299145,  9.75200848, 12.53047489, 13.98691426, 12.35421461,
       11.35345345, 13.27994304,  5.92281142, 11.4890861 ,  7.73692436,
        7.10378919,  5.47866221,  8.92549752, 13.744387  , 12.62576026,
        8.86082124,  8.05668609,  9.41970031,  7.88677538,  8.72896381,
        6.33075682,  7.5092834 ,  8.66273171,  5.19577146,  9.75401449,
       13.43127131])}
{'P': tensor([[2.6287e-02, 2.8161e-01, 1.7568e-01,  ..., 2.7164e-06, 3.5090e-03,
         9.6935e-06],
        [8.6092e-04, 2.5550e-02, 3.0838e-01,  ..., 2.1092e-06, 4.0165e-03,
         1.1092e-05],
        [2.8088e-04, 3.9178e-04, 2.8497e-02,  ..., 1.5811e-06, 8.0950e-03,
         2.2497e-05],
        ...,
        [2.7414e-04, 1.7908e-04, 6.2840e-05,  ..., 1.0301e-02, 1.3980e-06,
         9.3065e-08],
        [9.1116e-05, 1.1118e-04, 2.1508e-04,  ..., 2.8590e-06, 1.1815e-01,
         2.7456e-03],
        [2.9509e-06, 2.1784e-06, 1.5367e-06,  ..., 3.0152e-05, 1.9761e-06,
         1.8023e-02]]), 'X': tensor([[-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[4.7405e-06, 5.3292e-02, 1.1707e-01,  ..., 5.3681e-01, 6.0887e-01,
         7.4771e-01],
        [5.3292e-02, 0.0000e+00, 7.5948e-02,  ..., 5.3720e-01, 6.1304e-01,
         7.5885e-01],
        [1.1707e-01, 7.5948e-02, 0.0000e+00,  ..., 6.0046e-01, 5.6295e-01,
         7.1581e-01],
        ...,
        [5.3681e-01, 5.3720e-01, 6.0046e-01,  ..., 0.0000e+00, 1.1457e+00,
         1.2817e+00],
        [6.0887e-01, 6.1304e-01, 5.6295e-01,  ..., 1.1457e+00, 2.1453e-04,
         1.6830e-01],
        [7.4771e-01, 7.5885e-01, 7.1581e-01,  ..., 1.2817e+00, 1.6830e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.92644746,  8.05668609,  8.31625171,  8.10886021,  8.43626114,
        8.4486771 ,  4.75303127, 13.26209545, 10.48721382,  8.83341608,
        9.66977802, 11.28215429, 13.18278606,  7.60463349, 10.95593729,
       12.5085995 ,  8.56889738, 10.35464916,  9.66977802,  6.27715105,
       12.49291768,  8.57560275,  9.34127013, 10.65929968, 11.63502679,
       11.92133129, 13.1471313 ,  5.29009836,  5.81199422,  7.92607342,
       11.62770332, 13.38091121,  9.96152788, 11.1341798 ,  9.6282763 ,
       13.46553623,  9.3349858 , 11.28215429,  7.09887557,  5.64865173,
        5.87132718, 12.91505111, 11.74901778,  7.49076899,  7.92607342,
       13.89886469, 10.99326577, 13.08414689, 11.20100795,  7.59795445,
        9.562174  ,  4.76969364,  9.34127013, 11.54071859, 13.89264822,
        6.27905292,  6.54127431, 13.66991027,  4.78544875, 11.48434767,
        5.19065439, 10.48721382, 11.785566  , 13.84126197, 10.91805683,
       11.33837396,  6.59176018,  5.05444288, 10.68031351,  6.33075682,
       13.66051884, 13.50484539,  9.97822473, 14.05687815,  9.62958973,
       11.33837396, 12.07818871, 13.99048094, 12.11735334,  5.53255882,
       12.91143491, 14.03505624,  5.90035324,  7.72668467, 10.08001894,
        7.23276582, 10.4606265 , 10.78533838,  7.72415048,  8.30125201,
       13.40024224, 12.64235359,  9.54154096, 12.35421461,  7.88677538,
       10.27572673, 14.08549802,  5.79562691,  5.16611782,  7.71075226,
       13.95361098,  7.10378919, 12.79126785, 12.67576792,  9.10945678,
       10.84437705])}
{'P': tensor([[2.8842e-02, 1.7109e-01, 1.5463e-01,  ..., 2.0976e-05, 1.8037e-06,
         1.2432e-05],
        [9.0610e-04, 5.2886e-02, 5.8983e-03,  ..., 2.3941e-05, 2.4260e-06,
         9.4891e-06],
        [1.4942e-03, 3.9078e-02, 4.4577e-02,  ..., 3.4693e-05, 2.7794e-06,
         1.8929e-05],
        ...,
        [1.1873e-03, 4.1136e-04, 1.0592e-03,  ..., 1.0723e-01, 3.2118e-03,
         3.6451e-02],
        [2.1136e-05, 5.9640e-06, 1.2783e-05,  ..., 9.3480e-04, 7.0236e-03,
         5.2988e-04],
        [9.0691e-04, 3.5660e-04, 1.2125e-03,  ..., 1.4644e-03, 9.9919e-05,
         1.3504e-01]]), 'X': tensor([[ -0.6323,   7.9085,  -5.9544],
        [ -3.0004,   7.6607,  -7.2640],
        [ -2.1817,   7.8274,  -4.6298],
        [ -1.0407,   7.9166,  -6.6021],
        [ -3.4100,   7.5436,  -7.6980],
        [ -3.9700,   7.3453,  -4.1486],
        [ -2.0229, -10.6163,   8.6930],
        [  6.3542,   0.6934,  -3.3840],
        [ -3.1204, -10.2345,   9.2288],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.8367,   7.8718,  -8.6523],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 12.2592,   3.9425,   2.0383],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -2.0229, -10.6163,   8.6930],
        [  3.2647, -10.8066,  -0.3768],
        [  2.4149,  -4.6005,  -4.3339],
        [ -5.1369,   6.7755,   8.3836],
        [  0.7275,  14.0667,  -0.9815],
        [ -8.3539,  -5.5111,   0.3373],
        [ 10.9394,   7.3069,  -8.9365],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3497,   1.3680,  -1.9076],
        [ 12.2435,   4.0076,   8.1257],
        [ 10.0995,  -6.4739,   2.0974],
        [  3.2692,  13.5098,  -4.9032],
        [ -7.4808,   4.7462,   2.0266],
        [ 11.6165,  -3.8715,   6.4594],
        [  0.5850,  -4.7982,   9.6639],
        [  4.8674,  12.8875,   6.7622],
        [  4.7482, -10.3907,   5.9149],
        [  5.2199,   4.6861,  -1.9304],
        [  0.3724,   7.7974,  -6.8722],
        [ -8.9606,  -4.0994,   1.7886],
        [ 12.2592,   3.9425,   2.0383],
        [  1.9505,  -4.7171,   9.9868],
        [  2.5974,   7.0363,  -7.7810],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.0073,   3.0986,   6.7937],
        [  4.2275,  -3.6320,  -0.2714],
        [ -1.9646,   7.8571,   3.3817],
        [ -2.0721,   7.8432,   6.6380],
        [ -2.3317,   7.8034,  -3.6274],
        [  3.7121,   6.3117,   0.6738],
        [ -1.5261,   7.8988,   9.3768],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.3464,   0.5708,  -1.6397],
        [ 10.1983,  -6.3430,  -0.3200],
        [  9.3753,   9.5171,   3.9131],
        [  0.5850,  -4.7982,   9.6639],
        [  3.1892, -10.8220,  -6.8409],
        [  5.1667,  -2.6543,   1.2879],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.5250, -11.0306,   5.5358],
        [ -9.3616,  -2.4750,   3.7765],
        [  7.5784,  -8.9141,   5.8410],
        [ -2.3317,   7.8034,  -3.6274],
        [  0.7145,  -4.8071,  -2.3766],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 12.2589,  -1.9312,   9.8657],
        [ -8.8631,  -4.3728,  -1.1746],
        [  6.2242,  -0.3171,   7.4978],
        [  3.9913,  -3.8139,  -6.8311],
        [  3.6817,   6.3354,   6.1233],
        [ 12.4901,  -0.7052,   1.7315],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.3497,   1.3680,  -1.9076],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.0449,  13.8402,  -7.1642],
        [  9.9217,  -6.7006,  -5.6728],
        [  3.9085,   6.1512,  -2.2560],
        [ -9.1231,  -3.5739,   8.4708],
        [  4.1348,   5.9514,  -0.8856],
        [ 11.9122,   5.1397,  -1.5498],
        [ -6.1914,  -8.2997,   8.5380],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.5506,  -5.1154,   6.3423],
        [  5.3804,  -2.3504,   7.2744],
        [  1.1139,   7.6281,  -2.5680],
        [ -9.2974,   0.8371,   2.4852],
        [  6.2804,   2.0019,   0.8607],
        [  0.6757, -11.0361,  -1.3047],
        [  8.3791,  10.5409,   5.5590],
        [ 12.5199,  -0.4711,   3.0436],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.7795,  -3.4697,   3.5624],
        [  3.9515,  -3.8425,   6.1729],
        [ -3.5173,   7.5091,  -8.9585],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.3453,  -1.5462,   7.0212],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.2790,  -0.0259,  -5.4116],
        [ -6.0175,  -8.4541,  -9.8408],
        [  5.1861,   4.7361,  -3.7312],
        [  3.9895,   6.0816,  -6.5235],
        [  1.0000,   7.6591,  -4.5011],
        [ -1.5408,   7.8978,   5.3470],
        [  0.3493,  -4.7727,  -6.5682],
        [  5.8275,  -1.5334,   2.0124],
        [ 11.1085,  -4.9204,   0.0736],
        [  6.0073,   3.0986,   6.7937]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1303, 0.0886,  ..., 0.5646, 0.3889, 0.8809],
        [0.1303, 0.0000, 0.1388,  ..., 0.6786, 0.5019, 0.9802],
        [0.0886, 0.1388, 0.0000,  ..., 0.5437, 0.3675, 0.8413],
        ...,
        [0.5646, 0.6786, 0.5437,  ..., 0.0000, 0.1767, 0.3445],
        [0.3889, 0.5019, 0.3675,  ..., 0.1767, 0.0000, 0.5001],
        [0.8809, 0.9802, 0.8413,  ..., 0.3445, 0.5001, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.9337647 ,  8.22728399,  8.12580976,  7.98469166,  8.27853076,
        8.34949383, 10.80728788,  6.39187285, 10.69963954, 10.6910768 ,
        8.08320235, 10.37698435, 12.87752052,  9.75401449, 10.35485331,
       10.80728788, 11.28895489,  5.19577146,  8.50268322, 14.08549802,
       10.00794468, 13.15525803, 10.23946247, 12.87752052,  6.49538841,
       12.88270018, 11.99633819, 13.89974693,  8.859416  , 12.24467041,
        4.83371341, 13.77604293, 11.42420508,  7.01475376,  7.80625981,
        9.85384623, 12.87752052,  5.10446653,  7.50035355, 10.31886066,
        6.7593917 ,  5.57341623,  8.09899874,  8.11227668,  8.14433542,
        7.32233733,  8.04483495,  5.39450305,  9.36378438, 12.00995898,
       13.35927527,  4.83371341, 11.28215429,  5.80860165,  8.62329466,
       11.04313641,  9.68324553, 11.70016217,  8.14433542,  4.85995002,
       12.07818871, 12.41012046,  9.88310614,  6.2322769 ,  5.5205189 ,
        7.32754882, 12.50996828,  9.79814997,  6.49538841,  6.12886024,
       13.99048094, 11.97238624,  7.28793663,  9.79814997,  7.24677664,
       12.97370371, 10.35464916,  9.94285511,  9.5642993 ,  9.96391045,
        5.87132718,  7.70897675,  9.3349858 ,  6.59176018, 11.05672136,
       13.46553623, 12.52875793,  5.05963217, 12.27991991,  5.51177501,
        8.29203356, 12.22276278, 13.04156075, 12.44177641, 10.31613269,
       10.26327895,  6.27905292, 10.37698435,  7.02326515,  7.27340834,
        7.72415048,  8.04664886,  4.78544875,  6.02588517, 12.1494023 ,
        6.7593917 ])}
{'P': tensor([[1.8501e-02, 4.3038e-01, 1.7147e-01,  ..., 7.1571e-04, 1.2943e-02,
         3.4451e-04],
        [1.1428e-03, 3.1183e-02, 3.1364e-01,  ..., 1.2179e-03, 1.8373e-02,
         6.2677e-04],
        [7.4807e-04, 8.0720e-04, 4.6889e-02,  ..., 1.6274e-03, 3.4574e-03,
         2.1468e-03],
        ...,
        [2.8880e-04, 3.0191e-04, 4.9424e-04,  ..., 3.6921e-01, 6.9842e-04,
         3.2523e-02],
        [8.2884e-03, 9.9219e-03, 4.4716e-02,  ..., 2.1192e-02, 1.0158e-01,
         3.0476e-03],
        [3.6867e-05, 3.7009e-05, 4.1036e-05,  ..., 7.9721e-05, 3.8618e-05,
         1.4552e-01]]), 'X': tensor([[ -0.8569,   7.9156,  -6.5557],
        [ -1.0407,   7.9166,  -6.6021],
        [ -3.0004,   7.6607,  -7.2640],
        [ -3.4100,   7.5436,  -7.6980],
        [ -2.0840,   7.8415,  -7.8657],
        [ -5.2589,   6.7024,  -5.5809],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.3246,   1.6696,  -0.1544],
        [  6.0345,   3.0175,  -1.4424],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.1929,   2.4501,  -6.9352],
        [  0.9809,   7.6642,  -1.4217],
        [  1.7500,   7.4187,   4.4237],
        [  2.7685, -10.8982,  -6.8928],
        [ -7.5603,   4.6459,   8.2002],
        [  4.2096,   5.8815,   1.9230],
        [  2.5663,   7.0526,  -6.8769],
        [ -3.9508,  -9.8515,   1.7923],
        [ -5.1974,   6.7396,   3.3484],
        [  4.8674,  12.8875,   6.7622],
        [  2.0899,  -4.6870,  -7.4260],
        [ -7.6650,   4.5092,   0.2520],
        [ -2.1382,   7.8339,   7.3524],
        [  6.0305,   3.0296,   0.1375],
        [ -0.8455, -10.8853,  -6.7400],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.5960, -11.0335,   3.1144],
        [ 10.4758,  -5.9539,   1.3667],
        [ -6.4620,  -8.0449,   6.2719],
        [ -7.4807,   4.7463,   2.0616],
        [  5.7802,  -1.6363,  -6.2782],
        [  1.7659,   7.4127,   4.7871],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.1677,  -4.7447,  -6.1529],
        [  1.8719,   7.3713,  -3.7465],
        [  4.0457,  -3.7737,  -0.4704],
        [ 10.9394,   7.3069,  -8.9365],
        [ -8.0608,   3.9362,  -5.9567],
        [  4.0362,  13.2397,  -7.0030],
        [  4.5503,   5.5359,   7.4735],
        [  3.6469,   6.3624,  -1.8414],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.6078, -10.9229,   6.2869],
        [  4.3377,   5.7570,   1.3712],
        [  5.9026,   3.3837,  -0.6573],
        [ -7.0120,   5.2825,   7.7762],
        [ -1.6220,   7.8917,   0.0315],
        [  4.1637,  13.1898,  -1.9189],
        [  9.9058,  -6.7203,   5.6928],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.0591,   6.0202,  -1.6339],
        [ -4.2040,   7.2486,   2.9875],
        [  8.3718,  -8.2953,   8.4666],
        [  5.9548,  -1.2273,  -2.6520],
        [  8.7715,  -7.9390,   5.2900],
        [  3.3943,   6.5483,   1.0790],
        [ -2.6069, -10.4298,  -6.1995],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -2.5056,   7.7720,   5.5878],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.6112,   2.9319,  -7.9943],
        [  7.1441,  11.5529,   2.3232],
        [  5.1472,  -2.6801,   8.1194],
        [  6.2905,   1.9366,   2.1169],
        [  3.8111, -10.6787,  -4.6539],
        [ -8.8363,   2.4086,  -5.6296],
        [ 11.1469,   6.9237,   1.6160],
        [  3.6249, -10.7255,   5.6605],
        [  3.2701,   6.6338,   9.9209],
        [ -1.6334, -10.7207,   7.1397],
        [ 11.4743,   6.2478,  -2.1100],
        [  1.3103,  -4.8032,  -0.9300],
        [  2.4932,   7.0903,  -5.4214],
        [ 12.0042,  -2.8350,   2.6864],
        [ -1.6334, -10.7207,   7.1397],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.4489,   7.5256,   4.4529],
        [ -9.4524,  -1.6813,  -6.6862],
        [  5.0233,   4.9647,  -3.1788],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.3705,  13.9723,   7.9266],
        [ -7.5720,   4.6310,   2.1717],
        [  2.5469,  13.7192,  -9.6935],
        [  4.5503,   5.5359,   7.4735],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.0870,  -4.6877,   9.6823],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.1644, -11.0394,   7.8511],
        [ -8.4297,  -5.3637,  -9.3962],
        [  6.3534,   1.2990,  -7.2942],
        [  4.5800,   5.5034,  -3.9432],
        [  3.9515,  -3.8425,   6.1729],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.2644,  -4.6434,   9.7508],
        [  1.7903,  -4.7467,   0.1168],
        [ -1.6334, -10.7207,   7.1397],
        [  4.2339,  -3.6268,   6.3125],
        [  2.7590,  13.6621,   9.4496],
        [ -7.7239,   4.4298,   3.1216],
        [ -1.9487,   7.8590,  -3.3330],
        [ -8.9624,  -4.0944,  -4.9622]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 2.4120e-02, 1.2148e-01,  ..., 7.4145e-01, 1.7686e-01,
         7.3893e-01],
        [2.4120e-02, 0.0000e+00, 9.7763e-02,  ..., 7.3086e-01, 1.7519e-01,
         7.1488e-01],
        [1.2148e-01, 9.7763e-02, 0.0000e+00,  ..., 7.1442e-01, 2.1719e-01,
         6.2161e-01],
        ...,
        [7.4145e-01, 7.3086e-01, 7.1442e-01,  ..., 0.0000e+00, 5.7709e-01,
         7.2626e-01],
        [1.7686e-01, 1.7519e-01, 2.1719e-01,  ..., 5.7709e-01, 0.0000e+00,
         7.2238e-01],
        [7.3893e-01, 7.1488e-01, 6.2161e-01,  ..., 7.2626e-01, 7.2238e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.96181288,  7.98469166,  8.22728399,  8.27853076,  8.1137452 ,
        8.51928531,  9.4997987 ,  4.76969364,  5.5205189 ,  6.54127431,
        6.74686071, 12.41366754,  6.65991877,  7.72668467,  7.62232307,
       11.24434374,  8.8737556 ,  7.23276582,  7.50500064, 10.61417181,
        8.51088913, 13.77604293,  5.13182506,  8.89301516,  8.12043983,
        6.74872981, 10.91805683, 12.35421461,  9.4997987 , 11.04953767,
       12.0495521 , 10.31886066,  8.85939626,  6.00732193,  7.62011734,
        4.91038884,  4.74770899,  7.60529892,  5.53255882, 13.15525803,
        8.97052115, 13.84126197,  7.16598273,  7.33350642, 10.27716434,
       11.22993218,  7.20826219,  6.80370254,  8.77913428,  8.05668609,
       13.83139024, 11.9702762 ,  5.41348121,  7.26074925,  8.37954853,
       11.785566  ,  6.07992623, 11.83075005,  7.37576154, 10.7506423 ,
       12.05300943, 12.49317563,  8.16584537,  7.53063885,  9.69792513,
        9.09661212, 13.58333857,  5.80312949,  6.58184474, 11.33837396,
        9.15866607, 13.12218013, 11.32149259,  7.39599986, 10.84437705,
       13.06498554,  4.978704  ,  7.5158489 , 12.3344536 , 10.84437705,
        8.30125201,  7.66377554,  9.60080639,  7.06272256,  5.84450127,
       14.0393949 ,  8.87587457, 13.95361098,  7.16598273, 12.38881635,
        5.13125513, 10.32715715, 11.10066126,  9.99145096,  6.48486171,
        7.15990676,  5.51177501, 12.59840193,  5.16611782,  5.07306274,
       10.84437705,  5.5748624 , 13.93790235,  8.90405823,  8.09704052,
        9.85330646])}
{'P': tensor([[1.7223e-02, 3.5638e-01, 1.8401e-01,  ..., 4.3056e-05, 1.5551e-06,
         3.2160e-05],
        [7.9385e-04, 3.0469e-02, 1.5055e-01,  ..., 5.8222e-05, 2.7120e-06,
         4.8826e-05],
        [8.3634e-04, 2.9340e-03, 2.7232e-02,  ..., 3.9412e-05, 2.3799e-06,
         4.0157e-05],
        ...,
        [2.1964e-03, 2.0401e-03, 1.4638e-03,  ..., 3.4849e-02, 4.7977e-06,
         1.5727e-04],
        [3.8121e-05, 2.7442e-05, 2.2207e-05,  ..., 2.4761e-04, 1.7905e-02,
         4.7698e-03],
        [1.2199e-02, 8.9410e-03, 7.4641e-03,  ..., 3.4138e-03, 8.2037e-05,
         6.8699e-02]]), 'X': tensor([[-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0338, 0.0577,  ..., 0.5718, 0.1408, 0.2738],
        [0.0338, 0.0000, 0.0434,  ..., 0.5832, 0.1639, 0.3075],
        [0.0577, 0.0434, 0.0000,  ..., 0.6252, 0.1985, 0.3148],
        ...,
        [0.5718, 0.5832, 0.6252,  ..., 0.0000, 0.4424, 0.5537],
        [0.1408, 0.1639, 0.1985,  ..., 0.4424, 0.0000, 0.2176],
        [0.2738, 0.3075, 0.3148,  ..., 0.5537, 0.2176, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.96575782,  8.05668609,  8.10886021,  8.31625171,  8.25828409,
        8.23525433,  7.96575782,  9.96391045,  7.13501264,  8.62162372,
       11.54672755,  7.02253235,  8.67227386,  6.65259775, 10.6910768 ,
       13.85158417,  5.26342203, 13.02818702, 10.42667093, 11.72802491,
        7.5984809 ,  7.52117132,  8.30125201,  9.78286733, 13.03002338,
       13.66991027, 12.4217523 , 12.1494023 ,  8.05242591, 11.2845608 ,
        6.50409372, 10.99326577, 14.02526505, 12.70489563, 10.82106505,
        7.70897675, 11.63765273, 13.45642379,  5.0713364 ,  6.24614905,
        9.12220271,  4.7663526 ,  9.45995955, 12.91143491,  9.85384623,
        5.83560009,  6.35029274,  8.16714836,  9.26455795, 12.10942412,
       12.22276278, 11.24434374,  6.10360073, 13.6829843 ,  5.43544626,
        6.19707944,  8.36995953,  5.23878371,  7.39004226,  7.85919687,
        9.70108409,  9.04255655,  7.73151309, 11.35345345,  7.8575648 ,
       12.18045978,  6.19633139, 13.91198191,  9.69629   ,  9.94173708,
        9.3349858 ,  7.61088125, 13.04156075, 13.13205357, 13.46553623,
        9.43112957, 12.84056604, 10.48315436,  9.77588801, 13.03002338,
       10.61417181,  7.96575782,  5.19065439, 13.12299145,  6.14086582,
       11.73567759,  7.5092834 ,  5.95111804, 13.58101794, 10.99326577,
       13.31764181, 10.61472636, 11.38774149, 11.63724792,  7.76457095,
        5.77219077, 13.15525803, 10.92597196,  7.60463349,  7.02253235,
        5.14371005, 11.70168755,  8.85572292,  7.19534146, 11.92133129,
        6.54615887])}
{'P': tensor([[2.3462e-02, 2.3598e-01, 1.6407e-01,  ..., 4.7136e-03, 4.3117e-05,
         2.6648e-05],
        [7.0270e-04, 4.0819e-02, 3.5314e-01,  ..., 1.5766e-02, 5.1294e-05,
         2.0703e-05],
        [1.0464e-03, 4.6628e-03, 6.7288e-02,  ..., 2.6255e-02, 7.8651e-05,
         2.6157e-05],
        ...,
        [4.9042e-05, 7.6063e-05, 8.4145e-05,  ..., 6.3730e-02, 3.8336e-05,
         1.1025e-05],
        [1.1673e-02, 5.1396e-03, 4.2867e-03,  ..., 1.8953e-04, 6.5889e-02,
         2.0997e-04],
        [3.7960e-04, 3.2191e-04, 2.6616e-04,  ..., 5.3323e-04, 3.0236e-05,
         8.4713e-02]]), 'X': tensor([[ -1.0407,   7.9166,  -6.6021],
        [ -3.0004,   7.6607,  -7.2640],
        [ -3.4100,   7.5436,  -7.6980],
        [ -2.0840,   7.8415,  -7.8657],
        [ -5.2589,   6.7024,  -5.5809],
        [ -4.4520,   7.1370,  -8.7867],
        [ -5.1046,  -9.1610,   2.3418],
        [ -7.5401,   4.6718,   9.0033],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.2437,  -4.6489,   6.1430],
        [ -9.1795,  -3.3623,   8.5259],
        [  0.9478,  -4.8144,   0.6371],
        [ -6.3066,  -8.1936,   9.5822],
        [  3.3839,  -4.1966,   8.3574],
        [  9.6587,   9.1817,  -0.8212],
        [ 12.6059,   0.9597,  -6.3264],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.1196,  -2.7160,   4.2705],
        [ -9.4296,  -1.9458,   4.0833],
        [  1.3705,  13.9723,   7.9266],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -6.6998,   5.5954,   0.0823],
        [  2.6078, -10.9229,   6.2869],
        [  9.5203,   9.3483,   5.8781],
        [  9.9750,  -6.6338,  -1.2440],
        [  6.1737,   2.5305,   3.9049],
        [  9.6614,   9.1784,  -1.2087],
        [ -4.3359,  -9.6431,  -9.0616],
        [  6.1730,  -0.5375,  -8.8861],
        [ 11.2939,   6.6325,  -5.8311],
        [ -1.0012, -10.8576,   2.2867],
        [  9.3607,   9.5337,  -4.5412],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 11.0891,  -4.9558,  -2.3178],
        [  0.2732,  -4.7619,  -1.1577],
        [  4.9452, -10.3187,  -7.8621],
        [ -4.5778,  -9.5013,  -5.0667],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.6035,   0.7494,   7.1536],
        [  4.3523, -10.5232,   8.2138],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.0619,   8.6627,  -9.2884],
        [  3.9022,  13.2906,   5.5788],
        [  1.4150,  13.9647,  -4.0915],
        [  1.0000,   7.6591,  -4.5011],
        [  3.1892, -10.8220,  -6.8409],
        [  6.3211,   1.7017,  -0.2164],
        [  1.9503,  -4.7172,  -6.2313],
        [  2.5469,  13.7192,  -9.6935],
        [  6.0955,   2.8210,  -4.3321],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.0330,  -6.0765,   6.7945],
        [  5.1667,  -2.6543,   1.2879],
        [  6.7792,  11.8084,   1.6251],
        [  4.1637,  13.1898,  -1.9189],
        [ 12.2589,  -1.9312,   9.8657],
        [ -0.3432,   7.8900,   7.1805],
        [  3.2692,  13.5098,  -4.9032],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.1929,   2.4501,  -6.9352],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.2247,  -0.3150,  -9.3657],
        [ -4.7389,   6.9953,   3.7638],
        [  9.3836,   9.5075,  -5.9278],
        [ 12.4877,  -0.7222,   8.8347],
        [ -6.2668,   5.9808,   2.5614],
        [  5.6258,  -1.9404,  -6.0715],
        [ -4.7334,   6.9982,  -0.4392],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -1.9487,   7.8590,  -3.3330],
        [ 12.3759,  -1.3943,   6.4659],
        [  2.2849, -10.9655,  -7.6542],
        [  2.0729,   7.2878,   7.0880],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.0073,   3.0986,   6.7937],
        [ -2.0229, -10.6163,   8.6930],
        [  6.0890,  -0.8382,   2.9166],
        [  8.8091,  -7.9037,   0.3631],
        [ -7.8359,   4.2735,   7.2943],
        [ -6.6998,   5.5954,   0.0823],
        [  0.7275,  14.0667,  -0.9815],
        [  2.3589,  13.7668,  -3.6501],
        [ -1.3940, -10.7772,   6.7835],
        [  4.8432,   5.1968,   1.5006],
        [ -2.6485, -10.4151,   0.8766],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.3118,   4.5452,   4.2114],
        [ -8.7735,   2.5640,  -6.3313],
        [  4.9152,  12.8653,  -1.4459],
        [ -3.9456,  -9.8542,   0.7466],
        [ -1.6334, -10.7207,   7.1397],
        [  4.6231,   5.4556,   8.4973],
        [ -8.1204,  -5.9308,   4.0761],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.2608,  -2.5256,  -9.1446],
        [  3.6249, -10.7255,   5.6605],
        [ -8.9788,   2.0208,  -3.2867],
        [ -8.8363,   2.4086,  -5.6296],
        [ -5.3736,   6.6310,  -1.5174],
        [ -9.1624,  -3.4286,   2.0647],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.3031,   1.8477,  -7.1706],
        [  2.0243,   7.3086,   7.5432]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0978, 0.1324,  ..., 0.5474, 0.3089, 0.9228],
        [0.0978, 0.0000, 0.0355,  ..., 0.4652, 0.4044, 0.9646],
        [0.1324, 0.0355, 0.0000,  ..., 0.4435, 0.4367, 0.9890],
        ...,
        [0.5474, 0.4652, 0.4435,  ..., 0.0000, 0.8489, 1.0114],
        [0.3089, 0.4044, 0.4367,  ..., 0.8489, 0.0000, 0.9293],
        [0.9228, 0.9646, 0.9890,  ..., 1.0114, 0.9293, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 7.98469166,  8.22728399,  8.27853076,  8.1137452 ,  8.51928531,
        8.41171369, 10.48721382,  8.87007495, 12.3828957 ,  5.16205087,
        9.77588801,  4.90676242, 10.33958639,  5.39098669, 13.32645552,
       12.64235359, 10.78533838,  9.66977802,  9.71392179,  5.79543038,
        9.6282763 , 14.0393949 ,  9.88310614,  8.72896381, 11.22993218,
       13.3426541 , 11.97949273,  6.67218687, 13.32613445, 10.57302546,
        6.19633139, 13.09737053, 10.9036226 , 13.36092624,  9.88571713,
       12.1460931 ,  4.76969364, 11.4424816 , 10.54658597, 11.62770332,
       12.62576026, 11.38774149,  5.53175275,  6.80559669, 13.27717982,
       13.85158417, 14.03618649,  7.72415048, 11.28215429,  6.54615887,
        5.10444097, 13.95361098,  6.7166297 ,  8.22728399, 10.07238463,
        5.80860165, 13.61600536, 13.83139024, 12.41012046,  7.89745107,
       13.89974693,  5.35774573,  6.65991877, 13.01187484,  6.23262289,
        8.44940661, 13.35832725, 12.5085995 ,  8.66273171,  5.9510489 ,
        8.4486771 , 12.30858677,  8.09704052, 12.45418236, 11.20100795,
        7.57685908, 11.9120321 ,  6.7593917 , 10.80728788,  6.1463896 ,
       11.83509372,  8.92549752,  8.72896381, 14.08549802, 13.96746816,
       10.86693708,  7.10378919, 10.74655346,  9.20340031,  6.99097013,
        9.14044984, 13.77222752, 10.61472636, 10.84437705,  7.15100303,
       10.05558893,  4.978704  ,  5.83560009, 11.32149259,  9.20340031,
        9.15866607,  8.53501777,  9.78285424,  9.20340031,  6.56833864,
        7.58378655])}
{'P': tensor([[1.4336e-02, 2.9453e-01, 1.3161e-01,  ..., 1.2559e-05, 1.9734e-04,
         2.2032e-04],
        [4.1750e-04, 1.6842e-02, 1.6095e-01,  ..., 1.2994e-05, 1.1617e-04,
         3.3081e-04],
        [4.1139e-04, 7.2456e-04, 2.3660e-02,  ..., 1.7393e-05, 6.2455e-05,
         4.8463e-04],
        ...,
        [2.6494e-03, 2.0697e-03, 1.7148e-03,  ..., 1.0853e-01, 1.9738e-03,
         7.7022e-06],
        [2.8948e-02, 4.6137e-02, 8.9295e-02,  ..., 2.6612e-05, 1.9438e-01,
         2.6501e-04],
        [1.7794e-05, 2.0352e-05, 2.2633e-05,  ..., 5.3851e-06, 6.3563e-07,
         7.9713e-02]]), 'X': tensor([[-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0321, 0.0877,  ..., 0.4525, 0.4606, 0.8538],
        [0.0321, 0.0000, 0.0661,  ..., 0.4762, 0.4878, 0.8298],
        [0.0877, 0.0661, 0.0000,  ..., 0.4713, 0.4940, 0.8434],
        ...,
        [0.4525, 0.4762, 0.4713,  ..., 0.0000, 0.0832, 1.3060],
        [0.4606, 0.4878, 0.4940,  ..., 0.0832, 0.0000, 1.3118],
        [0.8538, 0.8298, 0.8434,  ..., 1.3060, 1.3118, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.01009083,  8.10886021,  8.25828409,  8.31625171,  8.23525433,
        8.4486771 ,  9.17905491,  6.78325151, 11.75438006,  8.85939626,
        8.40445872,  7.5158489 , 10.1371552 ,  5.87305005,  7.60529892,
       13.72594022,  9.41970031,  8.16714836, 10.07238463,  9.75401449,
       10.6910768 ,  9.76896904,  6.1561646 ,  8.89458273,  8.89301516,
       10.91805683, 13.35013374, 12.38881635,  5.38342077,  8.47371868,
        7.66377554,  7.46053281,  9.3349858 ,  5.22903117,  9.95552626,
        6.00663314, 12.79126785,  9.09661212,  8.67227386,  9.27474386,
       11.99633819,  5.40689332, 13.04156075,  7.5092834 , 12.86648771,
       14.0393949 , 11.70016217,  8.11227668,  4.78258092, 11.73848065,
        7.58668144,  6.81961996,  6.652742  ,  5.79543038, 10.00802627,
        7.07902036,  4.7663526 ,  7.09887557, 10.91805683, 11.48734978,
       11.04313641,  7.72291946,  7.76457095,  8.1303945 ,  7.76457095,
        9.05843201, 12.95702647,  8.08320235,  9.88571713,  5.53175275,
        9.68323686, 13.66051884,  5.80860165, 12.27993098,  5.70591986,
        7.53884104,  6.00732193,  8.81930525, 11.54672755, 11.27160035,
        8.90405823, 13.40850328, 11.9182956 , 10.76154256,  9.68324553,
        5.79391825, 10.52587243, 11.38774149, 12.5085995 ,  7.34250948,
        6.25586297, 14.05687815, 11.80092578, 11.59512194,  6.60752373,
       11.31276109,  5.86519525,  8.50278896, 13.3614598 , 13.99048094,
        7.32950385, 10.68416401, 14.10045797,  6.34059021, 14.10796466,
        9.45995955])}
{'P': tensor([[1.6550e-02, 1.9977e-01, 1.9972e-01,  ..., 2.4789e-07, 2.5499e-07,
         7.8230e-03],
        [3.8444e-04, 2.5661e-02, 1.7720e-01,  ..., 2.3337e-07, 5.2183e-07,
         2.5005e-02],
        [4.7604e-04, 4.3970e-03, 3.2988e-02,  ..., 2.8952e-07, 6.6172e-07,
         3.1273e-02],
        ...,
        [1.5134e-05, 1.1302e-05, 1.0801e-05,  ..., 2.3501e-02, 3.6491e-07,
         7.5406e-06],
        [4.9181e-06, 3.1180e-06, 3.0092e-06,  ..., 2.1432e-03, 1.0617e-01,
         2.6010e-06],
        [1.0265e-04, 2.1380e-04, 2.1883e-04,  ..., 4.4265e-07, 8.6967e-06,
         5.5472e-02]]), 'X': tensor([[ -1.5261,   7.8988,   9.3768],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.5520,   7.4976,   9.1522],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.0868,   7.2981,   8.1874],
        [ -2.2188,   7.8218,   8.6064],
        [  6.3301,   0.3638,  -7.5685],
        [  4.1999,  -3.6543,  -4.1998],
        [ -7.9667,   4.0814,  -0.6397],
        [  0.4119,  14.1019,  -4.1870],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -2.0445,   7.8469,  -0.5857],
        [  4.9152,  12.8653,  -1.4459],
        [  6.1989,  -9.7591,   0.8289],
        [ 12.6038,   0.7665,  -3.4439],
        [  9.4925,   9.3811,   8.9842],
        [  2.1120,  -4.6818,   8.6144],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.1984,  -0.4327,   1.5438],
        [  4.8653,   5.1694,  -4.0823],
        [ 12.5832,   1.7543,  -7.6134],
        [  2.8474,   6.8979,   6.6457],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.2508,   1.0572,   2.9978],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.4630,  -1.5145,   0.2983],
        [  9.5261,  -7.1675,   1.4206],
        [  1.5246,  -4.7838,  -5.2095],
        [ -1.6086,   7.8928,   4.3385],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.8400,   2.3991,   1.3457],
        [  1.0000,   7.6591,  -4.5011],
        [ 10.0400,   8.6922,  -6.2847],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.1204,  -5.9308,   4.0761],
        [ -7.9925,  -6.1423,   2.8990],
        [ -7.5720,   4.6310,   2.1717],
        [ -4.0868,   7.2981,   8.1874],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.6314,  -1.9302,  -9.6543],
        [  4.6357,   5.4415,   7.9067],
        [ -8.0330,  -6.0765,   6.7945],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.0348,  -2.8229,   3.5436],
        [  0.6470, -11.0352,  -1.1931],
        [  6.3535,   0.6792,  -7.5101],
        [  9.0652,   9.8597,  -3.5949],
        [  5.0233,   4.9647,  -3.1788],
        [  0.1047, -11.0046,   0.5188],
        [  5.1890,   4.7318,   0.9515],
        [  6.3549,   0.7096,   6.9198],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.0870,  -4.6877,   9.6823],
        [ -7.5401,   4.6718,   9.0033],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.7156,  -3.1831,  -8.0224],
        [ -6.8618,  -7.6322,   3.7027],
        [ 12.5919,   0.4035,  -5.8460],
        [  4.1297,  -3.7098,  -3.9838],
        [  7.1821,  -9.1845,   9.8548],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.0087,   4.0173,   9.1091],
        [ -6.8618,  -7.6322,   3.7027],
        [ -7.9925,  -6.1423,   2.8990],
        [ 10.8670,  -5.3428,  -2.8637],
        [  1.5383, -11.0274,  -4.9666],
        [ -4.0633,  -9.7927,  -3.0825],
        [  1.0093,   7.6567,   2.3526],
        [  3.3839,  -4.1966,   8.3574],
        [  6.2804,   2.0019,   0.8607],
        [  6.2675,  12.1379,  -8.9720],
        [  6.3604,   0.9042,  -6.5901],
        [  1.3705,  13.9723,   7.9266],
        [  5.1667,  -2.6543,   1.2879],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -7.4887,   4.7364,   8.9074],
        [ 12.2435,   4.0076,   8.1257],
        [ -1.8788, -10.6567,   3.7068],
        [  3.8111, -10.6787,  -4.6539],
        [  5.3858,   4.4260,  -9.6769],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -2.5683, -10.4433,  -2.9182],
        [  7.9816,  10.8938,  -0.5378],
        [ 11.7349,  -3.5838,   2.1644],
        [  1.2728, -11.0372,   9.8121],
        [ -1.5261,   7.8988,   9.3768],
        [  6.3360,   0.4258,  -9.2509],
        [  0.1460,   7.8340,  -9.4614],
        [  7.0642,  11.6104,  -9.8316],
        [  0.9074,  -4.8139,   8.1674],
        [  5.8275,  -1.5334,   2.0124],
        [ 12.5927,   0.4199,  -9.4997],
        [  5.0617,  12.7957,   8.3544],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.3535,  -5.5118,   9.8399],
        [ 12.6000,   1.3839,  -3.2992],
        [ -8.8745,   2.3097,   3.5600],
        [ -9.3419,   0.5972,  -1.2841],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.0467,  -9.8369,  -6.5816],
        [ -7.8126,   4.3066,   5.1791]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.2107e-02, 9.4896e-02,  ..., 1.3149e+00, 1.0597e+00,
         3.8790e-01],
        [8.2107e-02, 0.0000e+00, 1.2815e-02,  ..., 1.3512e+00, 1.0569e+00,
         3.1661e-01],
        [9.4896e-02, 1.2815e-02, 0.0000e+00,  ..., 1.3580e+00, 1.0579e+00,
         3.0650e-01],
        ...,
        [1.3149e+00, 1.3512e+00, 1.3580e+00,  ..., 0.0000e+00, 6.4070e-01,
         1.3581e+00],
        [1.0597e+00, 1.0569e+00, 1.0579e+00,  ..., 6.4070e-01, 0.0000e+00,
         9.2335e-01],
        [3.8790e-01, 3.1661e-01, 3.0650e-01,  ..., 1.3581e+00, 9.2335e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.04483495,  8.28968067,  8.29640613,  8.29022197,  8.36445758,
        8.1303945 ,  6.34059021,  5.5671389 ,  8.95133961, 14.10796466,
        9.55720651,  8.10886021, 13.77222752, 11.56146873, 12.62711252,
       13.34587027,  5.13616224,  9.85330646,  6.2134961 ,  7.09887557,
       12.70489563,  7.46248947,  5.13616224,  9.3109884 , 12.19805394,
        9.58347456, 11.92133129,  5.0209005 ,  8.0550286 ,  8.86082124,
        9.15976746,  7.72415048, 13.27994304,  8.86082124, 10.05558893,
       10.08001894,  8.87587457,  8.36445758,  5.68942572,  5.95297459,
        7.14837514, 10.07238463,  9.84314712,  5.77219077, 11.05413545,
        6.38968357, 13.39372426,  7.06272256, 11.00508681,  7.02253235,
        6.39438044, 12.38881635,  5.13125513,  8.87007495,  5.18047638,
        5.68942572, 10.26327895, 12.59840193,  5.5513072 , 11.65923396,
        8.37954853,  8.95983586, 10.26327895, 10.08001894, 12.10942412,
       11.1341798 , 10.60226614,  7.72291946,  5.39098669,  6.59176018,
       13.66051884,  6.42439344, 14.0393949 ,  5.80860165, 12.42554808,
        8.86082124, 12.88270018, 10.82106505, 11.33837396,  6.97106912,
       10.8992524 ,  9.91668182, 10.75443106, 13.50484539, 12.26997165,
       11.11038551,  8.04483495,  6.35029274,  7.83534987, 13.59056613,
        4.89869205,  6.02588517, 12.59969936, 13.76047999,  9.54154096,
        5.94909709, 10.00802627, 12.67576792,  9.17016857,  9.36093562,
        9.70108409, 11.62158554,  9.5014146 , 13.1399181 , 11.54672755,
        8.92098752])}
{'P': tensor([[1.2989e-02, 2.2629e-01, 1.5059e-01,  ..., 2.7487e-04, 9.1594e-07,
         3.6851e-06],
        [6.4667e-04, 2.6183e-02, 1.5203e-01,  ..., 5.1710e-04, 1.6653e-06,
         6.2726e-06],
        [6.9745e-04, 3.0256e-03, 2.5668e-02,  ..., 4.2516e-04, 1.5670e-06,
         5.4597e-06],
        ...,
        [5.3016e-05, 6.7965e-05, 6.6675e-05,  ..., 6.4985e-02, 2.9158e-06,
         6.6433e-05],
        [2.3251e-04, 1.4369e-04, 1.2299e-04,  ..., 7.8614e-06, 5.4049e-02,
         1.0458e-03],
        [3.9162e-04, 2.8505e-04, 2.3583e-04,  ..., 5.5237e-04, 7.2556e-04,
         1.2165e-01]]), 'X': tensor([[ -1.5408,   7.8978,   5.3470],
        [ -2.5056,   7.7720,   5.5878],
        [ -2.7572,   7.7194,   5.9580],
        [ -3.1671,   7.6158,   6.0765],
        [ -4.3963,   7.1629,   4.5263],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.3759,  -1.3943,   6.4659],
        [  5.9026,   3.3837,  -0.6573],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 11.7796,  -3.4696,   9.0088],
        [ -7.6719,  -6.6260,   7.1468],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 12.5927,   0.4199,  -9.4997],
        [ 10.0567,   8.6697,   4.6329],
        [  6.1685,  12.1979,  -6.2473],
        [  2.0243,   7.3086,   7.5432],
        [ -1.9646,   7.8571,   3.3817],
        [  6.2242,  -0.3171,   7.4978],
        [  6.8093,  -9.4178,   8.4244],
        [ -5.2589,   6.7024,  -5.5809],
        [ -2.0445,   7.8469,  -0.5857],
        [  6.1222,  -0.7261,   8.8204],
        [  5.1886,  -2.6250,   6.4677],
        [  1.7228,  -4.7575,   8.3396],
        [ 12.6000,   1.3839,  -3.2992],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.4748,  -0.7765,  -5.6695],
        [  9.8082,  -6.8396,   4.9615],
        [  4.1297,  -3.7098,  -3.9838],
        [ -3.4100,   7.5436,  -7.6980],
        [ 11.4743,   6.2478,  -2.1100],
        [ -8.7078,   2.7182,   1.8146],
        [ 11.7795,  -3.4697,   3.5624],
        [ 11.7779,   5.5127,  -2.0699],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.0087,   4.0173,   9.1091],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 11.5456,   6.0861,   8.1101],
        [ -3.5029,   7.5138,   8.1853],
        [  2.1504,  -4.6726,   8.7908],
        [ -3.5520,   7.4976,   9.1522],
        [  5.1519, -10.2387,   9.8191],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.1222,  -0.7261,   8.8204],
        [ -0.8885,   7.9160,   0.0571],
        [ -2.5056,   7.7720,   5.5878],
        [  1.7903,  -4.7467,   0.1168],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -0.1963,   7.8765,  -1.1371],
        [  4.6357,   5.4415,   7.9067],
        [  2.0419,  -4.6978,  -5.0983],
        [ -1.6275, -10.7221,   3.2376],
        [ 10.9162,  -5.2599,  -7.0812],
        [  8.3791,  10.5409,   5.5590],
        [  0.1931,  -4.7491,   7.4385],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.2849, -10.9655,  -7.6542],
        [  0.3724,   7.7974,  -6.8722],
        [  2.2644,  -4.6434,   9.7508],
        [ -2.4956, -10.4682,   3.5632],
        [  5.3602,  -2.3808,   7.2098],
        [  9.0051,   9.9234,   5.4411],
        [ -9.4630,  -1.5145,   0.2983],
        [  6.1929,   2.4501,  -6.9352],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.4766,  -1.1116,  -2.1706],
        [  2.9260,  -4.4162,   8.3808],
        [  4.7739,  -3.1218,  -4.1188],
        [  3.6154,  -4.0640,  -7.9698],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.5829,  -4.5466,  -8.4015],
        [  5.7845,   3.6652,   0.9175],
        [  6.9653,  -9.3225,  -3.4727],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.8704,  -9.3809,   7.5769],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 10.9394,   7.3069,  -8.9365],
        [  1.9312,  13.8649,   8.3563],
        [  6.3124,   0.2057,  -0.4593],
        [ -5.9872,   6.2037,  -9.4477],
        [ 12.3357,   3.6001,   0.7903],
        [  6.2026,  12.1774,  -1.5887],
        [ 11.4235,   6.3594,   5.1663],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.6816,   9.1536,  -5.8543],
        [  4.4290,  -3.4596,   1.4181],
        [ -7.9298,   4.1367,  -3.9773],
        [  1.1274,  14.0116,   8.1753],
        [  5.8201,   3.5841,  -8.0609],
        [  5.6260,  -1.9401,   3.0259],
        [  8.5716,  10.3596,   6.8818],
        [ -9.1178,   1.5784,   2.1410],
        [  6.8704,  -9.3809,   7.5769],
        [  0.6396,  14.0772,  -6.3786],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.0604,   1.7707,   1.1066],
        [  4.0078,  -3.8018,  -6.2168],
        [  4.8674,  12.8875,   6.7622],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.1940,  -2.1881,   7.1499],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.1579,  -0.5961,  -6.5595]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0466, 0.0610,  ..., 0.8994, 0.3610, 0.8438],
        [0.0466, 0.0000, 0.0256,  ..., 0.8898, 0.4013, 0.8695],
        [0.0610, 0.0256, 0.0000,  ..., 0.9115, 0.4213, 0.8945],
        ...,
        [0.8994, 0.8898, 0.9115,  ..., 0.0000, 0.8565, 0.6782],
        [0.3610, 0.4013, 0.4213,  ..., 0.8565, 0.0000, 0.5424],
        [0.8438, 0.8695, 0.8945,  ..., 0.6782, 0.5424, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.04664886,  8.16584537,  8.19703512,  8.24808832,  8.40445872,
        8.44940661, 12.45418236,  6.80370254, 12.05300943, 12.27993098,
       10.1371552 ,  9.51074763, 12.59969936, 13.2778378 , 13.66896739,
        7.58378655,  8.09899874,  6.2322769 , 11.62158554,  8.51928531,
        8.10886021,  6.16512856,  5.81481044,  5.05981914, 12.67576792,
        7.89745107,  9.50654967, 11.95743558,  5.5513072 ,  8.27853076,
       13.06498554,  9.12220271, 12.27991991, 13.00413829,  9.58072286,
        8.43626114,  8.95983586, 12.05300943, 13.05148826,  8.29022197,
        5.14371005,  8.29640613, 11.46176259,  9.85384623,  6.16512856,
        7.96575782,  8.16584537,  5.07306274,  9.47369628,  7.8788965 ,
        7.14837514,  5.1224079 , 10.84493732, 12.11735334, 13.46553623,
        4.75303127,  9.55720651, 11.20100795,  7.80625981,  5.16611782,
       10.76154256,  5.86519525, 13.40024224,  9.58347456,  6.65991877,
        9.54154096, 10.27572673,  9.54154096,  5.29755487,  5.70405525,
        5.43943791, 10.35485331,  5.22903117,  6.84796202, 11.63724792,
       10.00802627, 11.62770332, 10.27716434, 13.15525803, 13.99877413,
        6.31576264,  8.62162372, 12.85032836, 13.6660671 , 13.0743399 ,
        8.7273569 ,  9.68324553, 13.32374056,  5.62005629,  8.94395628,
       14.05687815,  6.83515201,  5.95111804, 13.44593519,  9.25336371,
       11.62770332, 14.09176259, 13.66896739,  9.23178204,  5.52416368,
       13.77604293,  9.84314712, 12.38881635,  8.8353674 , 12.26997165,
        6.18668463])}
{'P': tensor([[1.4909e-02, 2.5065e-01, 2.1448e-01,  ..., 5.2651e-04, 2.5376e-04,
         1.8500e-05],
        [8.5314e-04, 2.4783e-02, 1.3956e-03,  ..., 1.1894e-03, 4.2493e-04,
         3.4308e-05],
        [1.2111e-03, 2.2121e-01, 2.0257e-02,  ..., 7.4274e-04, 3.2002e-04,
         2.4820e-05],
        ...,
        [9.4596e-06, 1.0397e-05, 9.5277e-06,  ..., 3.2057e-02, 1.2634e-05,
         1.2888e-03],
        [7.7365e-04, 1.0496e-03, 7.8581e-04,  ..., 1.5069e-02, 2.0587e-01,
         9.5855e-04],
        [1.1088e-04, 1.0572e-04, 1.0531e-04,  ..., 9.7205e-05, 1.9290e-04,
         7.5414e-02]]), 'X': tensor([[-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0381, 0.0165,  ..., 0.7057, 0.7948, 0.6472],
        [0.0381, 0.0000, 0.0235,  ..., 0.6689, 0.7808, 0.6500],
        [0.0165, 0.0235, 0.0000,  ..., 0.6924, 0.7943, 0.6539],
        ...,
        [0.7057, 0.6689, 0.6924,  ..., 0.0000, 0.6617, 0.8470],
        [0.7948, 0.7808, 0.7943,  ..., 0.6617, 0.0000, 0.3489],
        [0.6472, 0.6500, 0.6539,  ..., 0.8470, 0.3489, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.04718433,  8.12043983,  8.06517143,  8.30729352,  8.29022197,
        8.36445758,  7.73692436, 13.08414689, 11.04629471, 13.6829843 ,
       12.00995898, 13.89671314, 11.73567759,  8.36995953,  6.19633139,
        6.54127431, 10.48840476, 13.52430176,  5.23878371, 10.82106505,
        6.25586297, 10.9036226 ,  9.53015376, 10.23946247, 11.64746882,
        6.61211224, 12.25806559,  5.79391825,  9.13507321,  5.80860165,
       13.39372426, 10.06023665,  8.98765644, 13.6105595 ,  5.5671389 ,
       11.86239961,  5.48276038, 12.64235359,  6.76791483, 12.12448912,
        8.92098752, 13.0493368 , 13.15525803,  8.51088913,  5.88782913,
       13.2778378 , 10.27572673, 10.5575601 , 11.05672136,  9.34061938,
        5.08372149,  5.96556248, 10.14044489,  9.95552626, 11.89105259,
        7.37902766, 13.13888841, 12.33167382,  4.8763135 ,  7.6374603 ,
       11.42420508,  9.84314712,  4.90676242,  8.08320235,  5.26067019,
       10.14044489, 13.00413829, 11.83570132, 13.88683436,  5.80860165,
        5.16205087, 10.27716434,  5.80312949,  8.22728399,  5.64500512,
        8.99276564,  7.18354885, 12.27835996, 10.41909211,  7.16058587,
        8.12580976,  7.76457095,  7.59795445, 13.25336001, 13.6105595 ,
        5.63824971,  5.16205087,  6.55800267, 12.27835996, 13.85158417,
        8.4486771 ,  5.94541803,  7.72415048,  7.28793663,  9.47369628,
       10.26545792, 10.69963954,  6.24187044,  7.13501264,  7.42512621,
       13.46553623,  9.34061938, 11.05672136,  9.63742669,  8.35856487,
        4.98461983])}
{'P': tensor([[1.4723e-02, 2.8812e-01, 2.3685e-01,  ..., 4.5415e-05, 7.0470e-04,
         1.0666e-07],
        [4.4584e-04, 1.9240e-02, 1.5968e-01,  ..., 8.3510e-05, 6.2324e-04,
         1.4595e-07],
        [8.0570e-04, 3.1016e-03, 3.7733e-02,  ..., 1.6446e-04, 9.3049e-04,
         2.6529e-07],
        ...,
        [1.9134e-06, 2.0135e-06, 1.9340e-06,  ..., 2.0717e-02, 4.4467e-06,
         1.0056e-04],
        [1.1976e-03, 1.8694e-03, 1.8112e-03,  ..., 5.5658e-04, 3.5185e-02,
         1.4326e-06],
        [4.3403e-06, 3.3841e-06, 2.8039e-06,  ..., 3.0179e-06, 2.9953e-05,
         1.0883e-01]]), 'X': tensor([[ -1.5875,   7.8944,  -9.4286],
        [ -2.5161,   7.7699,  -9.3531],
        [ -2.8162,   7.7059,  -9.6885],
        [ -3.5173,   7.5091,  -8.9585],
        [ -3.5903,   7.4847,  -8.9112],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.3236,   0.3011,  -4.6668],
        [  2.7420,  -4.4896,  -3.7139],
        [  8.8474,  10.0868,  -8.8644],
        [  3.3943,   6.5483,   1.0790],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.2608,  -2.5256,  -9.1446],
        [  4.1297,  -3.7098,  -3.9838],
        [  2.4932,   7.0903,  -5.4214],
        [  0.6933,   7.7336,   2.1179],
        [  6.1984,  -0.4327,   1.5438],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -6.4620,  -8.0449,   6.2719],
        [ 12.5717,   1.9266,   7.4903],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.7992,  -3.9474,  -8.9019],
        [  5.6095,  -1.9702,  -7.8233],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.9843,  -9.8679,  -6.1449],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 11.1420,   6.9331,  -2.4779],
        [  6.2247,  -0.3150,  -9.3657],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.4426, -10.1181,  -5.1268],
        [  1.9503,  -4.7172,  -6.2313],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.1824,  -4.6647,   1.1985],
        [  0.8940, -11.0402,  -0.9783],
        [  3.2701,   6.6338,   9.9209],
        [  1.6408,   7.4591,  -4.2294],
        [  6.3266,   0.3292,   9.3502],
        [ -5.1377,   6.7751,  -0.4438],
        [ -8.7735,   2.5640,  -6.3313],
        [ -9.4296,  -1.9458,   4.0833],
        [ -0.8569,   7.9156,  -6.5557],
        [ 10.4737,  -5.9570,  -1.2108],
        [  2.6078, -10.9229,   6.2869],
        [  4.5814, -10.4485,   2.5496],
        [  7.5784,  -8.9141,   5.8410],
        [  6.3413,   0.4883,  -6.8416],
        [  5.1142,  -2.7230,  -4.1966],
        [  1.7301,   7.4262,   5.0522],
        [  4.3377,   5.7570,   1.3712],
        [  6.0031,  12.2958,  -6.5992],
        [ -9.0604,   1.7707,   1.1066],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.3739,   0.3982,  -4.2564],
        [  6.0310,  12.2795,  -6.4528],
        [  2.2076,  13.8032,  -1.5099],
        [  6.8704,  -9.3809,   7.5769],
        [  1.6422,   7.4586,  -6.8056],
        [  9.1409,  -7.5789,   1.5814],
        [  5.2608,  12.6977,   4.9166],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.9178,   2.1936,   7.5388],
        [  3.3839,  -4.1966,   8.3574],
        [ -3.2486,   7.5925,  -1.3719],
        [ 12.5199,  -0.4711,   3.0436],
        [  1.7228,  -4.7575,   8.3396],
        [ 12.1379,   4.4125,  -3.8833],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -1.6220,   7.8917,   0.0315],
        [  7.1821,  -9.1845,   9.8548],
        [  5.1667,  -2.6543,   1.2879],
        [  2.3376,  -4.6231,   9.8436],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.9260,  -4.4162,   8.3808],
        [  2.6970,   6.9826,   9.7350],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.5650,   2.0132,   4.6272],
        [  9.1620,  -7.5573,  -9.9882],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.8275,  -1.5334,   2.0124],
        [ 10.5955,   7.8824,  -3.7006],
        [ -5.0943,  -9.1682,   0.5073],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.8058,   2.4850,  -1.7425],
        [ 11.6634,   5.8050,   9.1590],
        [  0.1460,   7.8340,  -9.4614],
        [  6.0305,   3.0296,   0.1375],
        [  6.9314,  -9.3435,  -1.2729],
        [  5.8450,  -9.9356,   3.0767],
        [ -9.3705,   0.4208,  -3.5815],
        [ -2.0229, -10.6163,   8.6930],
        [  5.6414, -10.0303,   7.8525],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.3444,   0.5821,   6.9101],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.5663,   7.0526,  -6.8769],
        [  7.8425,  -8.7202,   6.4530],
        [  4.5430,   5.5438,  -3.8669],
        [ -3.3999, -10.1152,  -1.2088],
        [ -7.7239,   4.4298,   3.1216],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.3539,  -5.5111,   0.3373],
        [ -1.9714,   7.8563,  -1.8084],
        [  8.3718,  -8.2953,   8.4666]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 3.6898e-02, 7.0121e-02,  ..., 8.0431e-01, 3.7447e-01,
         1.1619e+00],
        [3.6898e-02, 0.0000e+00, 3.3244e-02,  ..., 7.7765e-01, 3.8131e-01,
         1.1664e+00],
        [7.0121e-02, 3.3244e-02, 0.0000e+00,  ..., 7.5322e-01, 3.8883e-01,
         1.1699e+00],
        ...,
        [8.0431e-01, 7.7765e-01, 7.5322e-01,  ..., 0.0000e+00, 6.6528e-01,
         9.3814e-01],
        [3.7447e-01, 3.8131e-01, 3.8883e-01,  ..., 6.6528e-01, 0.0000e+00,
         7.8774e-01],
        [1.1619e+00, 1.1664e+00, 1.1699e+00,  ..., 9.3814e-01, 7.8774e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.05242591,  8.16714836,  8.20436764,  8.29203356,  8.30125201,
        8.41171369,  6.33075682,  5.26067019, 13.41713661,  7.37576154,
        5.24137909,  5.83560009,  5.5513072 ,  7.5158489 ,  7.76457095,
        6.2134961 , 12.30858677, 10.31886066, 12.71843462,  5.13182506,
        5.47866221,  5.94541803, 11.67875855, 11.54071859,  9.97964013,
       13.12299145,  6.23262289, 11.86239961, 11.4890861 ,  5.10444097,
       13.12299145,  5.14999799, 11.07637531,  7.39599986,  7.6374603 ,
        6.33517349,  8.50278896,  9.14044984,  9.6282763 ,  7.96181288,
       12.04924159, 11.22993218, 11.4088043 , 11.70016217,  6.36004393,
        5.79391825,  7.62509582,  7.20826219, 13.6829843 ,  9.23178204,
       10.75443106,  9.38232121, 13.68063098, 13.97857591, 11.62770332,
        7.63726134, 11.87412439, 13.744387  , 13.09737053,  9.23071698,
        9.18358562,  5.39098669,  8.25828409, 12.52875793,  5.05981914,
       12.91505111, 10.32715715,  8.05668609, 11.65923396,  5.80860165,
        5.1805345 ,  5.27178363,  5.29755487,  7.48538905, 10.51437947,
        8.50278896, 12.72524615, 11.87665501, 10.24557047,  6.02588517,
       13.20597932, 10.48840476, 12.41012046,  5.84450127,  5.07306274,
        9.14973319, 13.02818702,  7.83534987,  6.74872981, 11.63383221,
       11.52733581,  9.37989631, 10.80728788, 11.50791859,  5.0713364 ,
        9.36256096,  9.75401449,  7.50500064, 11.72802491,  7.16747878,
       10.67132524,  8.90405823,  7.822854  , 10.00794468,  8.09984231,
       11.785566  ])}
{'P': tensor([[2.3338e-02, 2.0266e-01, 1.6172e-01,  ..., 4.5847e-05, 6.1477e-04,
         4.2633e-05],
        [5.0136e-04, 4.9440e-02, 1.6675e-01,  ..., 1.3298e-04, 2.8213e-03,
         3.4588e-05],
        [3.2212e-04, 5.2178e-03, 3.6059e-02,  ..., 8.9171e-05, 3.1285e-03,
         2.6555e-05],
        ...,
        [7.9348e-04, 4.6340e-04, 3.8487e-04,  ..., 2.5490e-01, 3.5994e-05,
         1.7543e-04],
        [7.6343e-06, 1.4278e-05, 1.7887e-05,  ..., 2.7082e-05, 2.5741e-02,
         9.9638e-06],
        [5.3752e-03, 3.4772e-03, 4.4424e-03,  ..., 2.4787e-05, 1.4113e-03,
         1.4044e-01]]), 'X': tensor([[ -1.6086,   7.8928,   4.3385],
        [ -4.3963,   7.1629,   4.5263],
        [ -4.7389,   6.9953,   3.7638],
        [ -4.4503,   7.1378,   3.3751],
        [ -4.2040,   7.2486,   2.9875],
        [ -5.1974,   6.7396,   3.3484],
        [  5.6523,   3.9443,  -2.9449],
        [ -0.1626, -10.9796,   6.4629],
        [  7.1821,  -9.1845,   9.8548],
        [ 11.6019,   5.9540,  -5.1715],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.1433,  -3.5003,  -3.0209],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.8902,  -1.3887,   4.0388],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.8432,   5.1968,   1.5006],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.4167, -10.1293,   6.2156],
        [  5.7151,  -1.7700,   3.1542],
        [ -3.9456,  -9.8542,   0.7466],
        [  1.9204,   7.3518,   8.4781],
        [  9.5261,  -7.1675,   1.4206],
        [  6.3246,   1.6696,  -0.1544],
        [  7.2340,  -9.1505,  -5.8321],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -1.9487,   7.8590,  -3.3330],
        [  4.5491,  -3.3485,   6.8412],
        [  1.7232,  -4.7574,   3.1270],
        [  5.1667,  -2.6543,   1.2879],
        [ -9.3419,   0.5972,  -1.2841],
        [  7.2340,  -9.1505,  -5.8321],
        [  2.3928,   7.1404,  -1.7035],
        [  0.1252, -11.0062,   7.2564],
        [ 10.9916,   7.2133,  -0.9179],
        [ -3.2053, -10.1993,  -3.8736],
        [  5.7485,  -9.9811,  -3.2355],
        [ -8.7735,   2.5640,  -6.3313],
        [ 12.4877,  -0.7222,   8.8347],
        [ -1.6334, -10.7207,   7.1397],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.1999,  -3.6543,  -4.1998],
        [ 12.0042,  -2.8350,   2.6864],
        [ -4.0868,   7.2981,   8.1874],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.1929,   2.4501,  -6.9352],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.8617,   3.4853,  -9.8080],
        [  2.4697,  -4.5836,   1.4593],
        [  4.0457,  -3.7737,  -0.4704],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.1178,   1.5784,   2.1410],
        [  4.1999,  -3.6543,  -4.1998],
        [  3.6469,   6.3624,  -1.8414],
        [  2.5620,   7.0549,   8.4825],
        [  1.0391,  -4.8141,  -0.0768],
        [ -9.1706,   1.3863,   8.4032],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.6248,   3.9984,   8.4933],
        [ -0.5385,   7.9037,   6.3830],
        [ -8.1666,   3.7653,   3.6075],
        [  1.7000,  -4.7609,   2.7935],
        [  7.1821,  -9.1845,   9.8548],
        [  2.1120,  -4.6818,   8.6144],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.3459,  -3.5327,   0.3667],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.0899,  -4.6870,  -7.4260],
        [  6.2905,   1.9366,   2.1169],
        [ -9.4714,  -1.3301,  -3.1998],
        [  9.6816,   9.1536,  -5.8543],
        [  5.8836,   3.4315,  -4.0327],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.3411,  -2.6004,  -2.7062],
        [  0.6952,  -4.8060,   8.8976],
        [  6.3360,   0.4258,  -9.2509],
        [  5.0277,   4.9589,   7.0175],
        [ -8.8400,   2.3991,   1.3457],
        [  5.6414, -10.0303,   7.8525],
        [  6.2675,  12.1379,  -8.9720],
        [ -2.0840,   7.8415,  -7.8657],
        [  3.7916,  -3.9524,  -7.9547],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.6314,  -1.9302,  -9.6543],
        [ -6.4002,  -8.1048,  -0.9108],
        [  4.5800,   5.5034,  -3.9432],
        [  4.7739,  -3.1218,  -4.1188],
        [ -7.5720,   4.6310,   2.1717],
        [  6.0031,  12.2958,  -6.5992],
        [  0.9788,  -4.8145,  -3.8661],
        [  3.1064,  13.5607,   6.7605],
        [ -1.9646,   7.8571,   3.3817],
        [ -8.6112,   2.9319,  -7.9943],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.8835,  -4.3178,   9.3080],
        [  9.5018,  -7.1945,  -1.1892],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.9284,  11.7061,  -0.3732],
        [ 10.9931,   7.2107,   7.7933],
        [  8.5198,  10.4091,  -8.6486],
        [  1.9505,  -4.7171,   9.9868],
        [ -9.4309,  -0.0599,  -4.0830],
        [  3.9663,   6.1018,  -5.4104]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1192, 0.1471,  ..., 0.3392, 0.9019, 0.7210],
        [0.1192, 0.0000, 0.0636,  ..., 0.4368, 0.8378, 0.7800],
        [0.1471, 0.0636, 0.0000,  ..., 0.4825, 0.7769, 0.7376],
        ...,
        [0.3392, 0.4368, 0.4825,  ..., 0.0000, 1.2339, 0.8711],
        [0.9019, 0.8378, 0.7769,  ..., 1.2339, 0.0000, 0.8442],
        [0.7210, 0.7800, 0.7376,  ..., 0.8711, 0.8442, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.0550286 ,  8.40445872,  8.44940661,  8.41148722,  8.37954853,
        8.51088913,  6.89245352, 10.98076873, 11.65923396, 13.04051744,
       12.10942412,  9.79039371,  5.39723241,  6.05165248,  4.78544875,
        9.26455795,  9.60412735,  7.10378919, 10.06850677, 11.48663767,
        5.98285319, 10.61472636,  7.5984809 , 11.92133129,  6.54127431,
       11.66453059, 12.07818871,  8.09704052,  5.64865173,  5.05989547,
        5.80860165,  9.36093562, 11.66453059,  7.53063885, 11.00694688,
       13.1471313 , 10.6910768 , 11.51811603,  9.14044984, 12.5085995 ,
       10.84437705, 13.00413829,  5.5671389 , 12.3344536 ,  8.36445758,
        4.78258092,  6.65991877,  5.23878371, 13.1399181 ,  6.81961996,
        5.20659892,  5.53255882,  8.09899874,  9.25336371,  5.5671389 ,
        7.33350642,  7.50564223,  4.92497514,  9.27474386,  8.1137452 ,
        6.90115901,  7.92200342,  8.99276564,  5.05534609, 11.65923396,
        5.13616224,  6.07992623, 11.51811603,  5.60064319,  8.35856487,
        5.13182506,  6.58184474,  9.5642993 , 13.32374056,  6.81117802,
       13.77222752,  9.69629   ,  4.85603886,  6.35029274,  7.06169391,
        9.15976746, 11.50791859, 13.66051884,  8.1137452 ,  5.47702766,
        9.15866607,  5.95297459, 10.32715715,  7.15990676,  5.70405525,
        8.87587457, 13.6829843 ,  4.91296205, 13.91198191,  8.09899874,
        9.09661212,  9.88310614,  9.87720102, 11.9182956 ,  7.88677538,
       13.60275211, 13.14690137, 13.45124917,  5.10446653,  9.43112957,
        7.27759233])}
{'P': tensor([[2.3795e-02, 2.8720e-01, 1.7028e-01,  ..., 2.8062e-06, 4.5470e-05,
         3.5405e-05],
        [3.3560e-04, 2.4410e-02, 3.8128e-01,  ..., 3.5350e-06, 2.3379e-05,
         7.6276e-05],
        [2.6025e-04, 6.8132e-04, 2.9200e-02,  ..., 4.9894e-06, 1.8914e-05,
         1.4479e-04],
        ...,
        [1.7654e-05, 7.9430e-06, 5.4864e-06,  ..., 2.1051e-02, 1.0214e-04,
         2.8714e-06],
        [5.5566e-03, 3.9112e-03, 2.5715e-03,  ..., 8.6378e-06, 9.4918e-02,
         8.4111e-06],
        [5.1444e-06, 4.0885e-06, 3.9305e-06,  ..., 9.8740e-03, 6.2700e-06,
         3.2033e-02]]), 'X': tensor([[-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.5948e-02, 1.3177e-01,  ..., 2.9150e-02, 5.8320e-01,
         6.1544e-01],
        [7.5948e-02, 0.0000e+00, 6.0403e-02,  ..., 5.5355e-02, 6.1825e-01,
         5.4627e-01],
        [1.3177e-01, 6.0403e-02, 7.5848e-05,  ..., 1.0564e-01, 6.7156e-01,
         4.8604e-01],
        ...,
        [2.9150e-02, 5.5355e-02, 1.0564e-01,  ..., 0.0000e+00, 6.0883e-01,
         5.8717e-01],
        [5.8320e-01, 6.1825e-01, 6.7156e-01,  ..., 6.0883e-01, 0.0000e+00,
         1.1157e+00],
        [6.1544e-01, 5.4627e-01, 4.8604e-01,  ..., 5.8717e-01, 1.1157e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.05668609,  8.31625171,  8.4486771 ,  8.43626114,  8.50278896,
        8.25828409, 13.97779839, 13.04156075, 12.41366754, 12.77402351,
        9.96391045,  8.41171369, 12.67576792,  6.23262289,  4.78258092,
       11.10066126,  8.90405823,  5.43544626,  4.96231532,  8.35856487,
        8.12043983, 13.2430377 ,  5.39723241, 13.66051884, 13.66896739,
       13.14777433, 13.96746816,  8.41148722,  4.74770899,  7.06372172,
        6.02588517,  7.5984809 ,  6.12886024,  6.35029274, 13.79762415,
       10.5575601 , 10.39942497, 11.74901778,  4.91038884,  5.30158497,
        5.19065439,  6.19707944, 13.97430191, 11.86239961, 13.92390689,
       10.54658597,  7.58668144, 13.25017211,  7.15100303, 10.8207297 ,
        5.70591986,  6.60752373,  6.27715105,  4.85603886,  6.7593917 ,
       12.97370371,  9.19837792, 12.27835996, 12.35421461,  9.44486609,
        7.88677538, 13.3614598 , 13.26590071,  8.09899874, 10.26545792,
        9.23071698,  4.87083348,  7.62011734, 13.15525803, 13.85158417,
       12.4013153 , 10.66631784, 13.83139024, 11.54672755, 11.28215429,
        9.41605684,  5.5671389 , 13.02818702,  6.65259775,  6.65259775,
       13.66051884,  4.8763135 , 11.65923396, 12.18998195, 11.57249984,
       10.9036226 ,  9.97220799,  9.78147542, 10.82106505, 13.0493368 ,
       12.00995898,  6.65368551,  6.80370254,  5.47866221, 11.62770332,
        5.92952538, 10.05558893,  6.48486171,  7.14837514,  5.82345867,
        9.09963894, 13.77604293, 11.00694688, 11.69264243,  7.19534146,
       10.60226614])}
{'P': tensor([[1.7252e-02, 2.0753e-01, 2.0697e-01,  ..., 6.7833e-04, 7.2280e-05,
         4.2429e-07],
        [6.5606e-04, 4.3328e-02, 1.6903e-02,  ..., 4.3740e-03, 4.2453e-05,
         6.1626e-07],
        [3.2487e-04, 2.4496e-02, 1.9426e-02,  ..., 1.5471e-03, 2.0631e-05,
         2.3312e-07],
        ...,
        [1.1691e-05, 1.5577e-05, 1.4424e-05,  ..., 3.8718e-02, 6.1943e-07,
         2.8556e-07],
        [5.9021e-02, 7.4727e-02, 6.5984e-02,  ..., 3.3128e-04, 1.2018e-01,
         4.0866e-06],
        [1.7116e-04, 1.3813e-04, 1.0209e-04,  ..., 6.9489e-06, 1.3945e-03,
         1.5208e-01]]), 'X': tensor([[ -1.6906,   7.8860,   7.7142],
        [ -3.6381,   7.4683,   7.3544],
        [ -3.5029,   7.5138,   8.1853],
        [ -2.1382,   7.8339,   7.3524],
        [ -4.0868,   7.2981,   8.1874],
        [ -5.1369,   6.7755,   8.3836],
        [  4.5814, -10.4485,   2.5496],
        [ -0.5385,   7.9037,   6.3830],
        [ -0.5739,   7.9056,   0.6382],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.3724,   7.7974,  -6.8722],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -2.3317,   7.8034,  -3.6274],
        [ -8.5506,  -5.1154,   6.3423],
        [  1.4489,   7.5256,   4.4529],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.0330,  -6.0765,   6.7945],
        [  5.8318,   3.5567,  -4.7692],
        [  8.6271,  -8.0715,  -7.4813],
        [  2.9260,  -4.4162,   8.3808],
        [  0.3715,  -4.7756,   0.2312],
        [  0.6142,  14.0802,   0.5407],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 11.6544,   5.8273,  -3.1375],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.1989,  -9.7591,   0.8289],
        [ 10.9875,   7.2207,  -8.8549],
        [  0.2480, -11.0153,  -3.0183],
        [  7.7750,  11.0665,   2.8364],
        [ 10.0400,   8.6922,  -6.2847],
        [  4.7813,  -3.1139,   8.2691],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.6613,   7.0021,   4.5783],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.8325,  -1.5222,  -2.1341],
        [ -4.5778,  -9.5013,  -5.0667],
        [  1.7232,  -4.7574,   3.1270],
        [  2.1120,  -4.6818,   8.6144],
        [ 11.7726,  -3.4876,  -6.7402],
        [  4.5814, -10.4485,   2.5496],
        [  6.3303,   1.6145,  -8.4083],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.1751,  -2.2592,  -5.7751],
        [  9.6587,   9.1817,  -0.8212],
        [  0.6142,  14.0802,   0.5407],
        [ -6.1899,  -8.3011,  -7.9269],
        [  3.1756, -10.8247,   6.8882],
        [ 12.5789,   0.1722,   1.6470],
        [  3.7992,  -3.9474,  -8.9019],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.4630,  -1.5145,   0.2983],
        [  6.1929,   2.4501,  -6.9352],
        [  7.7794,  11.0629,   8.9069],
        [  7.5784,  -8.9141,   5.8410],
        [ -5.1974,   6.7396,   3.3484],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.5813,  -4.5471,   5.3597],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -7.5603,   4.6459,   8.2002],
        [ -1.6220,   7.8917,   0.0315],
        [  6.2022,   2.4092,   3.0349],
        [ -4.7334,   6.9982,  -0.4392],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.0330,  -6.0765,   6.7945],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.4283,  -1.9585,  -0.9129],
        [  3.1480,  -4.3165,  -7.8002],
        [ -7.9925,  -6.1423,   2.8990],
        [  7.4132,  11.3526,   0.3322],
        [ -8.3900,   3.3727,   3.1650],
        [  7.0171,  11.6439,  -5.0178],
        [  4.0457,  -3.7737,  -0.4704],
        [ -0.0410,   7.8591,   9.3457],
        [  0.3038,   7.8092,   0.4164],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.3505,   0.6235,  -8.5849],
        [  2.2437,  -4.6489,   6.1430],
        [  5.7485,  -9.9811,  -3.2355],
        [ -2.1817,   7.8274,  -4.6298],
        [ 11.5083,  -4.1177,   0.2389],
        [ -1.5261,   7.8988,   9.3768],
        [ -8.5859,  -5.0393,   0.5476],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.4524,  -1.6813,  -6.6862],
        [  3.5965,  -4.0754,  -3.5035],
        [ -3.5173, -10.0623,   4.8280],
        [  5.9912,   3.1453,  -3.9989],
        [  6.2975,   0.0947,   5.1791],
        [  8.3791,  10.5409,   5.5590],
        [  5.8275,  -1.5334,   2.0124],
        [  0.6363, -11.0348,   6.1414],
        [ -2.5056,   7.7720,   5.5878],
        [  5.9193,  -1.3176,   5.1739],
        [ 10.2470,   8.4054,  -0.1177],
        [ -2.2508, -10.5479,  -2.7235],
        [  1.1139,   7.6281,  -2.5680],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.3261,  13.7749,   5.9121],
        [ 10.0619,   8.6627,  -9.2884]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0886, 0.0798,  ..., 0.6595, 0.5525, 1.3051],
        [0.0886, 0.0000, 0.0303,  ..., 0.5737, 0.6411, 1.3366],
        [0.0798, 0.0303, 0.0000,  ..., 0.5946, 0.6279, 1.3566],
        ...,
        [0.6595, 0.5737, 0.5946,  ..., 0.0000, 1.2041, 1.5170],
        [0.5525, 0.6411, 0.6279,  ..., 1.2041, 0.0000, 1.2451],
        [1.3051, 1.3366, 1.3566,  ..., 1.5170, 1.2451, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.06517143,  8.30729352,  8.29022197,  8.12043983,  8.36445758,
        8.50268322, 11.4088043 ,  7.92200342,  7.92644746, 11.89855267,
        7.80625981,  9.11473391, 10.31613269,  9.86089579,  9.562174  ,
        8.14433542,  9.96391045,  7.66377554,  5.07306274, 10.07238463,
        6.83083863, 11.81423161,  5.29755487,  4.79003361, 14.09357547,
        9.54154096, 13.03002338,  7.8575648 , 11.56146873, 13.14777433,
       11.018084  , 13.5247215 , 13.27994304,  5.70591986, 12.19805394,
        7.49076899,  4.978704  ,  6.02789634, 10.54658597,  5.05989547,
        5.13616224, 12.27835996, 11.4088043 ,  6.53290663, 12.59840193,
       12.3828957 , 13.32645552, 14.09357547, 10.35485331, 11.28093507,
       12.58006038,  5.47866221,  5.79391825,  9.58347456,  6.65991877,
       13.52430176, 11.70016217,  8.51088913, 11.74901778,  5.2287058 ,
       12.1494023 ,  9.88310614, 12.32545637,  8.8737556 ,  8.05668609,
        6.65368551,  8.4486771 ,  8.87007495,  9.93553424, 10.07238463,
        7.89745107,  9.62958973,  5.34250225, 10.08001894, 13.55868684,
        9.04255655, 13.59480906,  5.53255882,  7.85919687,  7.81509631,
       11.69264243,  6.38105015,  5.16205087, 11.51811603,  8.12580976,
       12.22276278,  8.04483495,  9.95552626,  9.41970031,  9.60080639,
        5.43544626, 10.65929968,  6.76661595,  6.29821673, 13.46553623,
        6.02588517, 11.05317597,  8.16584537,  6.06416957, 13.25336001,
       10.78533838,  7.70897675, 12.72524615,  9.68324553, 13.96987806,
       13.27717982])}
{'P': tensor([[1.5687e-02, 1.8540e-01, 1.6409e-01,  ..., 2.1865e-06, 2.0351e-05,
         3.7185e-04],
        [1.0398e-03, 5.7130e-02, 5.8185e-02,  ..., 8.5213e-06, 5.5686e-05,
         6.1198e-04],
        [6.3055e-04, 1.2862e-02, 3.6810e-02,  ..., 3.8708e-06, 3.0461e-05,
         2.7397e-04],
        ...,
        [9.3539e-05, 7.7547e-05, 4.9405e-05,  ..., 1.1443e-01, 3.8152e-03,
         1.3788e-03],
        [8.3864e-03, 5.1071e-03, 3.5961e-03,  ..., 4.9607e-04, 1.1136e-01,
         2.7348e-02],
        [6.0863e-03, 1.2923e-02, 6.9003e-03,  ..., 5.5158e-05, 1.4659e-04,
         6.5086e-02]]), 'X': tensor([[ -1.7009,   7.8851,  -8.2437],
        [ -3.4100,   7.5436,  -7.6980],
        [ -3.5903,   7.4847,  -8.9112],
        [ -3.5173,   7.5091,  -8.9585],
        [ -2.0840,   7.8415,  -7.8657],
        [ -4.4520,   7.1370,  -8.7867],
        [  5.9587,  12.3216,   9.1695],
        [ 10.2712,   8.3707,  -4.9743],
        [  0.6363, -11.0348,   6.1414],
        [  5.6672,  -1.8631,   8.1020],
        [ -5.1046,  -9.1610,   2.3418],
        [ -7.7239,   4.4298,   3.1216],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.6414, -10.0303,   7.8525],
        [ 11.9122,   5.1397,  -1.5498],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.3413,   0.4883,  -6.8416],
        [ -0.5739,   7.9056,   0.6382],
        [  1.6408,   7.4591,  -4.2294],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.6422,   7.4586,  -6.8056],
        [ -4.7847,  -9.3729,  -3.7741],
        [  8.8144,  -7.8988,  -0.0128],
        [  1.4489,   7.5256,   4.4529],
        [  7.7794,  11.0629,   8.9069],
        [  3.2701,   6.6338,   9.9209],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.6681,  -4.8556,   9.2518],
        [  4.3377,   5.7570,   1.3712],
        [  6.1752,   2.5242,   3.5874],
        [  6.9433,  -9.3362,  -0.3421],
        [  6.0788,  -0.8710,  -6.8719],
        [  6.2630,   2.1057,  -5.2136],
        [  6.2883,   0.0325,   5.2692],
        [ -9.4203,  -2.0340,   2.5856],
        [  7.7750,  11.0665,   2.8364],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.5375,   7.0676,  -4.3490],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.0345,   3.0175,  -1.4424],
        [ 11.5198,  -4.0923,   5.5088],
        [  3.2990,  -4.2415,  -7.4857],
        [  9.1409,  -7.5789,   1.5814],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.6970,   6.9826,   9.7350],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.3616,  -2.4750,   3.7765],
        [  3.9591, -10.6391,  -6.2557],
        [  6.3413,   0.4883,  -6.8416],
        [  3.4604,  -4.1545,   1.8091],
        [ 12.2435,   4.0076,   8.1257],
        [ -8.1905,   3.7253,   4.1175],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.7558,  -4.4844,   0.0636],
        [  2.5829,  -4.5466,  -8.4015],
        [  8.4691,  10.4571,  -5.0650],
        [ 12.0582,  -2.6640,   0.7132],
        [  0.9036,   7.6840,   9.9237],
        [  3.7916,  -3.9524,  -7.9547],
        [  4.0587, -10.6112,   6.2273],
        [  5.4429,  -2.2529,   9.7306],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -3.0008, -10.2827,  -8.9480],
        [  0.2571,  -4.7594,   1.5618],
        [ -7.0825,   5.2073,  -5.0497],
        [  0.1206,  -4.7363,   8.7103],
        [  7.3592,  11.3937,   6.2407],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.6933,   7.7336,   2.1179],
        [  1.7218,  -4.7576,  -3.3617],
        [  5.3602,  -2.3808,   7.2098],
        [  5.6022,  -1.9833,   8.3081],
        [ -0.5709,   7.9055,   7.6089],
        [  0.1041, -11.0045,   2.4311],
        [ -0.8885,   7.9160,   0.0571],
        [ -2.5683, -10.4433,  -2.9182],
        [ 12.5224,  -0.4497,   4.8448],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.6414, -10.0303,   7.8525],
        [ -7.9298,   4.1367,  -3.9773],
        [ -5.2589,   6.7024,  -5.5809],
        [ -2.7572,   7.7194,   5.9580],
        [  2.1824,  -4.6647,   1.1985],
        [ 12.1379,   4.4125,  -3.8833],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -3.0004,   7.6607,  -7.2640],
        [ -4.0409,   7.3169,  -3.5779],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.3604,   0.9042,  -6.5901],
        [  2.0039,   7.3173,  -1.5885],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.9788,   2.0208,  -3.2867],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -2.3317,   7.8034,  -3.6274],
        [ -8.4824,  -5.2575,  -0.0795],
        [  1.1644, -11.0394,   7.8511],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.8093,  -9.4178,   8.4244],
        [  5.5240,  -2.1198,   6.2055],
        [  6.2576,   2.1359,  -3.3398],
        [ -0.1963,   7.8765,  -1.1371]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0926, 0.1001,  ..., 0.9531, 0.4012, 0.3822],
        [0.0926, 0.0000, 0.0505,  ..., 0.9756, 0.4717, 0.3990],
        [0.1001, 0.0505, 0.0000,  ..., 1.0221, 0.4988, 0.4456],
        ...,
        [0.9531, 0.9756, 1.0221,  ..., 0.0000, 0.6594, 0.5767],
        [0.4012, 0.4717, 0.4988,  ..., 0.6594, 0.0000, 0.2796],
        [0.3822, 0.3990, 0.4456,  ..., 0.5767, 0.2796, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.06643487,  8.27853076,  8.30125201,  8.29203356,  8.1137452 ,
        8.41171369, 13.68672855, 13.25017211, 11.05317597,  5.96556248,
       10.48721382,  8.90405823,  4.8763135 ,  9.73579738, 11.50791859,
       12.97370371,  5.5513072 ,  6.36004393,  7.92644746,  7.6374603 ,
       10.27572673,  7.63726134, 10.52357479, 11.83570132,  7.66377554,
       13.52430176,  7.39599986, 10.36844813,  9.9353921 ,  7.20826219,
        6.67121748, 11.63502679,  6.14086582,  6.60752373,  6.28835101,
        9.63742669, 13.5247215 , 10.33958639,  7.5092834 , 13.13205357,
        6.74686071, 12.22503754,  5.37343173, 11.87412439,  4.76969364,
        7.48538905,  9.34127013,  9.68324553, 11.35182589,  6.36004393,
        5.40689332, 12.88270018,  8.99791316,  8.14433542,  5.19065439,
        5.26342203,  5.22903117, 13.45642379, 12.3489324 ,  7.73692436,
        5.47702766, 11.36089355,  5.89074104, 12.11735334, 10.71162681,
        4.7663526 ,  8.7907923 ,  4.73784222, 13.56367857, 11.785566  ,
        9.50654967,  7.76457095,  5.05963217,  5.86519525,  5.94293185,
        7.92607342, 11.0050379 ,  7.96575782, 10.75443106, 12.53047489,
       12.91505111, 11.50791859,  8.94395628,  8.51928531,  8.19703512,
        5.14999799, 12.91505111, 10.39942497, 12.53345445,  8.22728399,
        8.35856487, 10.27716434,  6.42439344,  7.58668144, 13.59480906,
        9.20340031, 10.46739073, 12.10942412,  8.14433542,  9.97964013,
       11.10066126,  9.75200848, 11.62158554,  5.91677188,  6.61211224,
        7.8788965 ])}
{'P': tensor([[1.0753e-02, 2.6265e-01, 1.8829e-01,  ..., 6.3454e-06, 8.9178e-06,
         3.0159e-05],
        [5.1732e-04, 1.5291e-02, 1.8498e-01,  ..., 9.1336e-06, 1.3443e-05,
         4.3465e-05],
        [5.4751e-04, 9.1357e-04, 1.6638e-02,  ..., 9.3303e-06, 1.7122e-05,
         5.0114e-05],
        ...,
        [2.8006e-04, 2.6931e-04, 2.2718e-04,  ..., 1.3247e-01, 2.1269e-05,
         4.5639e-04],
        [2.4875e-05, 2.3886e-05, 2.3259e-05,  ..., 3.5142e-04, 1.8712e-01,
         4.0901e-02],
        [6.0192e-04, 5.6918e-04, 5.3517e-04,  ..., 9.9046e-04, 7.9134e-04,
         1.3654e-01]]), 'X': tensor([[-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.3509e-02, 4.6759e-02,  ..., 7.1495e-01, 6.0086e-01,
         2.3735e-01],
        [2.3509e-02, 0.0000e+00, 2.7661e-02,  ..., 7.2082e-01, 5.8359e-01,
         2.5971e-01],
        [4.6759e-02, 2.7661e-02, 1.0727e-04,  ..., 7.4567e-01, 5.5602e-01,
         2.7244e-01],
        ...,
        [7.1495e-01, 7.2082e-01, 7.4567e-01,  ..., 0.0000e+00, 1.2571e+00,
         7.7386e-01],
        [6.0086e-01, 5.8359e-01, 5.5602e-01,  ..., 1.2571e+00, 0.0000e+00,
         7.4600e-01],
        [2.3735e-01, 2.5971e-01, 2.7244e-01,  ..., 7.7386e-01, 7.4600e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.06990483,  8.09704052,  8.14433542,  8.35856487,  8.36995953,
        8.34949383,  8.4486771 , 11.42420508,  6.81117802,  6.61211224,
        8.7907923 ,  8.37954853, 10.38897563, 11.54672755,  5.14371005,
        8.56889738,  6.54127431,  5.29009836, 10.41909211, 10.4606265 ,
       10.14436912, 12.24467041, 13.25336001,  9.87720102, 12.00995898,
       11.75618513,  6.1463896 ,  9.6282763 ,  8.12580976,  5.94293185,
       10.54658597, 12.18045978,  7.6374603 ,  8.34949383, 13.99048094,
       13.2778378 ,  7.56617037, 11.59512194, 10.90638066,  7.16747878,
       12.70489563, 11.48734978,  8.31625171, 13.34587027,  7.81509631,
       10.90638066, 11.31276109, 12.1460931 , 14.10045797,  8.08320235,
        8.01009083, 13.95472331,  6.29821673, 11.72802491,  6.60752373,
        7.06169391, 13.99048094,  5.47702766, 10.08001894,  9.88310614,
        7.16058587,  9.99211663,  9.34127013, 12.17451685,  6.00732193,
        5.51177501, 10.63038565,  6.38399789, 12.05300943,  4.8763135 ,
       10.06850677, 11.2845608 ,  7.57946459,  9.24501829, 13.6105595 ,
       11.80092578,  9.71392179, 10.86693708,  8.09842704,  7.57685908,
        4.91296205,  7.4544866 , 11.9702762 ,  6.65368551, 10.09979747,
       10.66631784, 10.4606265 ,  8.31625171,  6.19633139, 12.67680439,
        8.1303945 , 10.65929968, 13.05148826, 12.97370371,  6.18668463,
       11.72749923, 10.35093485,  6.35029274, 11.73397309, 13.15525803,
        9.86089579,  7.38844703, 13.35013374,  5.51177501, 10.71162681,
        5.10444097])}
{'P': tensor([[1.2099e-02, 1.6497e-01, 1.6324e-01,  ..., 1.0752e-03, 1.4582e-05,
         1.2313e-07],
        [1.6460e-04, 9.7549e-03, 1.8007e-01,  ..., 1.7371e-03, 1.9886e-05,
         1.0028e-07],
        [1.8236e-04, 5.6856e-04, 1.1324e-02,  ..., 2.1083e-03, 2.3895e-05,
         1.1856e-07],
        ...,
        [6.0999e-06, 8.1323e-06, 8.2836e-06,  ..., 1.3835e-02, 5.1586e-03,
         3.5314e-06],
        [1.4459e-05, 1.1716e-05, 1.1710e-05,  ..., 3.7487e-05, 1.1424e-01,
         1.9281e-03],
        [8.2726e-07, 4.6045e-07, 4.5697e-07,  ..., 1.8088e-07, 8.0535e-07,
         1.5921e-02]]), 'X': tensor([[-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0674, 0.0829,  ..., 0.6151, 0.5598, 1.1265],
        [0.0674, 0.0000, 0.0164,  ..., 0.5556, 0.5100, 1.1524],
        [0.0829, 0.0164, 0.0000,  ..., 0.5394, 0.4950, 1.1534],
        ...,
        [0.6151, 0.5556, 0.5394,  ..., 0.0000, 0.1473, 1.2305],
        [0.5598, 0.5100, 0.4950,  ..., 0.1473, 0.0000, 1.0837],
        [1.1265, 1.1524, 1.1534,  ..., 1.2305, 1.0837, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.08320235,  8.29203356,  8.30125201,  8.41171369,  8.27853076,
        8.16714836,  9.64157461,  5.54776934,  7.44805901,  7.92200342,
        7.72668467, 13.40850328, 13.32613445,  6.29034958, 12.33167382,
       13.59480906,  9.26455795,  6.33517349, 13.83139024, 11.76830981,
        8.89458273,  9.71392179,  7.57946459,  8.83341608, 11.85890664,
        5.80860165,  7.89745107,  9.7712321 , 12.62576026, 11.04313641,
        7.50917551, 12.59969936,  9.12220271,  6.29821673,  8.83341608,
       11.05317597, 12.52372629,  6.21275016,  6.39438044, 13.61801526,
        5.79543038,  9.73579738,  5.63824971, 12.58006038,  5.05989547,
       11.28895489,  5.29009836,  7.6374603 ,  9.64157461,  5.13182506,
        6.49538841,  9.0368458 , 12.21532282,  5.19577146,  5.91677188,
        5.68942572, 12.13568598, 13.13888841,  7.50035355, 11.24434374,
       13.39372426,  6.97106912, 10.74655346, 11.63724792,  8.31625171,
       10.1371552 ,  6.54693987, 10.09979747,  8.92395381, 10.794755  ,
        5.92281142,  9.75401449, 13.35013374,  7.78980299,  8.62162372,
       12.05300943,  8.41171369, 12.32545637,  7.06372172, 10.69963954,
        9.69792513,  6.25586297,  7.92644746,  7.56617037, 12.91143491,
       13.79762415,  6.39187285,  4.90676242, 12.4013153 , 10.14436912,
        6.50409372, 10.48315436,  8.12043983,  9.73579738, 12.07818871,
        6.53250937,  7.19534146,  9.47369628,  8.41171369, 10.01804223,
        9.88571713, 13.36092624,  7.39599986,  9.50913849, 10.60226614,
       11.76830981])}
{'P': tensor([[1.3965e-02, 1.6894e-01, 1.6500e-01,  ..., 4.7436e-05, 3.6752e-05,
         1.7401e-06],
        [7.8285e-04, 1.4257e-02, 2.0725e-01,  ..., 4.9129e-05, 3.7452e-05,
         1.5700e-06],
        [1.2676e-03, 1.4392e-03, 9.3370e-02,  ..., 4.0057e-04, 2.9863e-04,
         6.4474e-06],
        ...,
        [2.8631e-04, 2.5736e-04, 1.8458e-04,  ..., 7.3100e-02, 7.2497e-02,
         9.9615e-04],
        [3.5114e-04, 3.1091e-04, 2.2017e-04,  ..., 2.9108e-02, 9.5276e-02,
         1.4607e-03],
        [2.9711e-04, 2.4093e-04, 9.8839e-05,  ..., 1.5837e-06, 1.5761e-06,
         4.7620e-03]]), 'X': tensor([[-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.7661e-02, 1.1362e-01,  ..., 2.5636e-01, 3.2132e-01,
         5.2384e-01],
        [2.7661e-02, 1.0727e-04, 8.9926e-02,  ..., 2.8400e-01, 3.4869e-01,
         5.4696e-01],
        [1.1362e-01, 8.9926e-02, 0.0000e+00,  ..., 3.6241e-01, 4.1865e-01,
         6.3679e-01],
        ...,
        [2.5636e-01, 2.8400e-01, 3.6241e-01,  ..., 0.0000e+00, 8.4356e-02,
         3.3543e-01],
        [3.2132e-01, 3.4869e-01, 4.1865e-01,  ..., 8.4356e-02, 0.0000e+00,
         3.6157e-01],
        [5.2384e-01, 5.4696e-01, 6.3679e-01,  ..., 3.3543e-01, 3.6157e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.09704052,  8.14433542,  8.35856487,  8.36995953,  8.34949383,
        8.38252937, 12.24467041,  9.87720102,  6.29034958, 14.0393949 ,
        6.34059021, 13.35927527, 11.73397309, 10.04916678, 12.5769659 ,
        5.26067019, 13.72594022, 11.52733581,  8.8580671 , 11.10066126,
        7.81509631,  8.92098752, 13.96746816, 10.06437598,  9.43112957,
        7.48538905,  8.49161154, 11.52733581,  8.41148722,  5.70405525,
       11.07637531, 11.89855267, 11.57249984,  5.16077764,  9.37989631,
        9.11473391,  7.58378655,  7.9337647 , 14.0393949 ,  5.79391825,
        9.78286733,  7.62509582, 11.75618513, 12.05395056,  7.27759233,
       11.1913598 ,  6.02789634,  9.58072286, 13.89671314,  5.41348121,
        9.562174  , 13.09426142, 12.52372629, 11.75807252, 10.54658597,
        5.13616224, 11.83570132,  5.05774295,  9.97964013, 11.51811603,
        8.34949383, 12.00995898,  5.19065439, 14.10796466,  4.89121805,
        7.62011734,  9.17016857, 13.12299145, 12.35625694,  9.36256096,
        4.92497514, 13.97430191, 13.15525803, 11.04629471,  6.54615887,
        8.34949383, 10.68416401,  4.7663526 , 10.4606265 , 10.6910768 ,
       13.92390689,  8.29640613, 13.15525803,  5.86519525, 13.18278606,
       13.04051744,  6.61386436,  5.46244988, 10.19036081, 13.50484539,
        7.83486943, 11.05672136,  5.64500512,  5.23878371, 12.05300943,
        7.19534146, 13.00413829,  5.79391825, 13.94008948,  5.40689332,
        5.82345867,  8.8353674 , 12.11735334,  5.01114719,  5.08910491,
       13.00413829])}
{'P': tensor([[2.0142e-02, 2.5827e-01, 1.7997e-01,  ..., 4.1673e-05, 6.1186e-07,
         8.5747e-06],
        [7.4051e-04, 4.2403e-02, 6.8611e-03,  ..., 2.8253e-05, 1.1363e-06,
         1.7847e-05],
        [2.2377e-03, 1.3647e-01, 6.9917e-02,  ..., 6.3866e-05, 2.3387e-06,
         4.0394e-05],
        ...,
        [6.3117e-02, 5.2875e-02, 7.0903e-02,  ..., 1.1873e-01, 5.2121e-07,
         7.6888e-06],
        [3.1687e-05, 1.4807e-05, 2.3790e-05,  ..., 1.3467e-04, 5.0689e-02,
         1.9800e-04],
        [7.6737e-05, 5.0474e-05, 7.8308e-05,  ..., 5.2025e-05, 1.8138e-02,
         1.1649e-01]]), 'X': tensor([[ -1.9600,   7.8577,   6.9121],
        [ -3.6381,   7.4683,   7.3544],
        [ -3.1671,   7.6158,   6.0765],
        [ -4.0868,   7.2981,   8.1874],
        [ -3.5029,   7.5138,   8.1853],
        [ -5.3382,   6.6533,   7.8054],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.6703,   6.3443,  -5.4401],
        [ 11.0891,  -4.9558,  -2.3178],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.9653,  -9.3225,  -3.4727],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.4235,   6.3594,   5.1663],
        [  3.4604,  -4.1545,   1.8091],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.8572,  14.0501,  -8.3364],
        [ 10.3251,   8.2928,  -3.0006],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.6142,  14.0802,   0.5407],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.9653,  -9.3225,  -3.4727],
        [  6.8093,  -9.4178,   8.4244],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.2895,   0.0405,   5.7652],
        [ -5.6423,   6.4537,  -1.2091],
        [ 11.1469,   6.9237,   1.6160],
        [ -6.2200,  -8.2737,   5.5079],
        [  6.3200,   0.2690,   1.5900],
        [ -3.0644,   7.6439,  -1.4322],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.3551,   0.7149,   1.3344],
        [  6.3549,   0.7096,   6.9198],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.6634,   5.8050,   9.1590],
        [  9.9217,  -6.7006,  -5.6728],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.4572,   7.1085,   7.7376],
        [  6.3497,   1.3680,  -1.9076],
        [  5.4167, -10.1293,   6.2156],
        [ 12.3357,   3.6001,   0.7903],
        [  0.1677,  -4.7447,  -6.1529],
        [ -3.4488, -10.0934,   2.5195],
        [  3.6249, -10.7255,   5.6605],
        [  3.3474,  -4.2162,   5.2455],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.4242, -10.1261,   3.9595],
        [ 10.3251,   8.2928,  -3.0006],
        [  4.4049,   5.6892,  -4.4394],
        [  3.9085,   6.1512,  -2.2560],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.0000,   7.6591,  -4.5011],
        [  1.8721,  -4.7322,   0.5524],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -7.6719,  -6.6260,   7.1468],
        [  3.2692,  13.5098,  -4.9032],
        [ 12.3759,  -1.3943,   6.4659],
        [ -3.3812, -10.1235,   7.4384],
        [ 12.5927,   0.4199,  -9.4997],
        [ -5.2752,  -9.0409,  -8.2568],
        [  5.7151,  -1.7700,   3.1542],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -2.4956, -10.4682,   3.5632],
        [  2.5636, -10.9293,   9.6519],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.5424,  -2.0885,   7.2487],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.9410,   6.8428,   5.4458],
        [ -1.6906,   7.8860,   7.7142],
        [  5.0233,   4.9647,  -3.1788],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.2571,   6.7072,   8.3693],
        [  6.2895,   0.0405,   5.7652],
        [  6.2576,   2.1359,  -3.3398],
        [  7.5784,  -8.9141,   5.8410],
        [ 11.5198,  -4.0923,   5.5088],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.9925,  -6.1423,   2.8990],
        [  6.2555,   2.1474,   4.1685],
        [ -5.0132,  -9.2234,  -8.3531],
        [  3.1756, -10.8247,   6.8882],
        [  6.5428,  -9.5728,  -2.9897],
        [ -7.4887,   4.7364,   8.9074],
        [ -6.2200,  -8.2737,   5.5079],
        [ -1.8367,   7.8718,  -8.6523],
        [  4.3459,  -3.5327,   0.3667],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.3200,   0.2690,   1.5900],
        [ -8.1666,   3.7653,   3.6075],
        [  6.1755,   2.5233,  -2.0042],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 11.1469,   6.9237,   1.6160],
        [ -0.1963,   7.8765,  -1.1371],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.4932,   7.0903,  -5.4214],
        [  3.9913,  -3.8139,  -6.8311],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.2345,  -3.1345,   3.9715],
        [ 11.3367,  -4.4806,   0.4643],
        [  3.4401,  13.4540,   4.6576],
        [ 10.9599,  -5.1849,  -3.5526],
        [  2.3373,  -4.6232,  -5.2246]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.8729e-02, 7.4962e-02,  ..., 6.2833e-01, 8.6715e-01,
         8.3973e-01],
        [7.8729e-02, 0.0000e+00, 6.1118e-02,  ..., 7.0292e-01, 9.1054e-01,
         8.8310e-01],
        [7.4962e-02, 6.1118e-02, 0.0000e+00,  ..., 6.9596e-01, 8.5021e-01,
         8.2278e-01],
        ...,
        [6.2833e-01, 7.0292e-01, 6.9596e-01,  ..., 3.0339e-04, 9.6401e-01,
         9.4388e-01],
        [8.6715e-01, 9.1054e-01, 8.5021e-01,  ..., 9.6401e-01, 0.0000e+00,
         2.7448e-02],
        [8.3973e-01, 8.8310e-01, 8.2278e-01,  ..., 9.4388e-01, 2.7448e-02,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.09842704,  8.30729352,  8.24808832,  8.36445758,  8.29022197,
        8.53014915,  8.49161154,  7.32950385, 12.1460931 ,  6.38968357,
       12.52875793, 11.63724792,  7.83486943, 13.0743399 ,  5.40689332,
        5.5366698 , 14.07623266, 13.2430377 , 10.7167285 ,  9.94285511,
       14.09357547,  9.5014146 , 11.63724792, 11.62158554, 12.59969936,
        6.28961988,  8.5724042 , 13.12218013, 10.35093485,  6.32572323,
        8.23525433, 11.51811603,  6.39520397,  6.39438044, 11.27160035,
       13.02818702, 11.97238624, 10.412299  ,  7.52117132,  6.49538841,
       11.48663767, 12.85032836,  4.74770899, 10.66631784, 11.32149259,
        5.38342077, 12.32545637, 11.48734978, 13.2430377 ,  7.19513806,
        7.28793663, 10.27572673,  7.72415048,  5.08910491, 12.00995898,
       10.1371552 , 13.89974693, 12.45418236, 10.67322527, 12.59969936,
       10.46739073,  5.98285319, 12.53285633, 10.76154256, 11.22596994,
        9.88310614, 10.00794468,  5.92281142, 12.00995898,  7.44805901,
        8.06517143,  7.06272256, 12.91143491, 13.103709  ,  6.28961988,
        6.61211224, 11.70016217, 12.22503754, 10.06437598, 10.08001894,
        6.61386436, 10.4977132 , 11.28093507, 11.59512194,  8.86082124,
       10.35093485,  8.08320235,  5.60064319, 13.18278606,  6.32572323,
        8.99276564,  6.67108662,  9.4997987 , 13.12218013,  7.8788965 ,
       10.32715715,  5.18277858,  7.5158489 ,  5.5205189 ,  9.69629   ,
        8.29022197,  9.75200848, 12.18998195, 13.88683436, 12.12448912,
        5.18047638])}
{'P': tensor([[1.3795e-02, 1.4728e-01, 1.3985e-01,  ..., 4.8463e-05, 7.9885e-06,
         5.4404e-07],
        [1.7661e-04, 1.3664e-02, 8.3110e-02,  ..., 2.2756e-05, 9.2406e-06,
         3.1196e-07],
        [1.4407e-04, 1.4233e-03, 1.2883e-02,  ..., 1.9329e-05, 8.0993e-06,
         2.5787e-07],
        ...,
        [2.4466e-02, 1.3414e-02, 1.0889e-02,  ..., 5.5207e-02, 1.1500e-04,
         9.0502e-05],
        [2.2332e-04, 1.4594e-04, 1.2529e-04,  ..., 4.6263e-03, 1.1110e-01,
         5.5180e-03],
        [2.6955e-04, 1.4164e-04, 1.1342e-04,  ..., 3.4970e-03, 3.4055e-05,
         6.3194e-02]]), 'X': tensor([[ -1.9646,   7.8571,   3.3817],
        [ -4.2040,   7.2486,   2.9875],
        [ -4.4503,   7.1378,   3.3751],
        [ -4.7389,   6.9953,   3.7638],
        [ -5.1974,   6.7396,   3.3484],
        [ -4.3963,   7.1629,   4.5263],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.2902,   0.0451,   3.7116],
        [  0.5600, -11.0321,  -0.8660],
        [  7.9141,  -8.6656,   9.6948],
        [  4.0078,  -3.8018,  -6.2168],
        [ -0.5385,   7.9037,   6.3830],
        [ 11.4495,   6.3026,   6.8257],
        [  3.5965,  -4.0754,  -3.5035],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.8889,  -1.3916,   8.8993],
        [  3.3633,  13.4794,   0.5193],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -9.1623,  -3.4288,  -4.9961],
        [  1.1274,  14.0116,   8.1753],
        [  1.1994,   7.6035,   2.4295],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.6470, -11.0352,  -1.1931],
        [  6.0047,   3.1062,  -6.7914],
        [ -1.7290,   7.8825,  -3.3831],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.3118,   4.5452,   4.2114],
        [ -3.5520,   7.4976,   9.1522],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 10.9931,   7.2107,   7.7933],
        [  6.2790,  -0.0259,  -5.4116],
        [ -0.1963,   7.8765,  -1.1371],
        [ 11.7794,   5.5086,  -9.6929],
        [ -1.9714,   7.8563,  -1.8084],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 12.5224,  -0.4497,   4.8448],
        [ -5.0550,   6.8231,  -9.2502],
        [ -9.4351,  -1.8891,   3.5642],
        [  8.1528,  10.7452,   9.1493],
        [ -8.6603,   2.8250,   5.9181],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.5662,  13.9375,   4.5142],
        [ -4.2271,   7.2387,  -1.8316],
        [ -2.0229, -10.6163,   8.6930],
        [  6.3515,   0.6418,  -6.5823],
        [ -7.7842,   4.3466,  -1.7548],
        [ -9.2687,  -2.9794,   0.6436],
        [ -7.5603,   4.6459,   8.2002],
        [ -3.9700,   7.3453,  -4.1486],
        [ -2.0445,   7.8469,  -0.5857],
        [ 12.5123,  -0.5339,  -1.0728],
        [  5.1196,  -2.7160,   4.2705],
        [ -3.0008, -10.2827,  -8.9480],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.6895,   5.6051,   2.7899],
        [  0.7685,  -4.8098,  -2.9310],
        [  8.7131,  10.2216,   7.7915],
        [  5.1142,  -2.7230,  -4.1966],
        [ -2.1817,   7.8274,  -4.6298],
        [  4.8432,   5.1968,   1.5006],
        [ 12.1504,   4.3674,  -7.2998],
        [ -8.6675,  -4.8569,  -9.1769],
        [  9.4925,   9.3811,   8.9842],
        [ -8.5859,  -5.0393,   0.5476],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.1531,  -5.8748,   2.6934],
        [  2.5813,  -4.5471,   5.3597],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -8.1905,   3.7253,   4.1175],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.4735,  -0.7274,  -2.6749],
        [  4.5340,  -3.3629,   8.9652],
        [  1.0093,   7.6567,   2.3526],
        [  5.8889,  -1.3916,   8.8993],
        [  5.0233,   4.9647,  -3.1788],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.2345,  -3.1345,   3.9715],
        [ -4.6393,   7.0461,   1.1624],
        [  4.8653,   5.1694,  -4.0823],
        [  6.2804,   2.0019,   0.8607],
        [  9.6662,   9.1725,   8.1056],
        [ -1.6275, -10.7221,   3.2376],
        [  7.1697,  11.5343,  -0.7276],
        [  6.3122,  -9.6994,  -6.3282],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.1292,   4.8183,  -5.7997],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -3.5520,   7.4976,   9.1522],
        [ -4.4503,   7.1378,   3.3751],
        [  3.6703,   6.3443,  -5.4401],
        [  6.0345,   3.0175,  -1.4424],
        [  2.3589,  13.7668,  -3.6501],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.6672,  -1.8631,   8.1020],
        [ -5.1974,   6.7396,   3.3484],
        [  1.1006,   7.6318,  -2.4877],
        [ 12.0042,  -2.8350,   2.6864],
        [  3.9913,  -3.8139,  -6.8311],
        [ -8.9178,   2.1936,   7.5388],
        [ -3.2053, -10.1993,  -3.8736],
        [  6.9693,  -9.3200,  -8.3822],
        [ -7.6293,  -6.6859,  -5.5161],
        [  8.6420,  -8.0580,   6.5781],
        [  4.1348,   5.9514,  -0.8856],
        [  4.0078,  -3.8018,  -6.2168],
        [ 11.7162,   5.6728,  -8.9372]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0923, 0.0974,  ..., 0.4267, 0.6904, 1.0772],
        [0.0923, 0.0000, 0.0295,  ..., 0.4658, 0.7105, 1.1076],
        [0.0974, 0.0295, 0.0000,  ..., 0.4918, 0.7390, 1.1353],
        ...,
        [0.4267, 0.4658, 0.4918,  ..., 0.0000, 0.2822, 0.6512],
        [0.6904, 0.7105, 0.7390,  ..., 0.2822, 0.0000, 0.4052],
        [1.0772, 1.1076, 1.1353,  ..., 0.6512, 0.4052, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.09899874,  8.37954853,  8.41148722,  8.44940661,  8.51088913,
        8.40445872,  9.34061938,  6.29034958, 11.04629471, 11.73567759,
        5.52416368,  7.92200342, 13.06957425,  5.43544626, 10.74409834,
        6.05113217, 13.89264822, 10.23946247,  9.78286733, 14.05687815,
        7.69752262,  4.91038884, 11.05413545,  6.76056625,  8.06990483,
       10.00802627,  6.99097013,  8.29640613,  9.97964013, 13.14690137,
        6.27905292,  7.8788965 , 13.00380624,  8.09984231, 12.07818871,
       12.53047489,  8.49161154,  9.62238828, 13.48809595,  9.10945678,
        8.05242591, 14.02526505,  8.38252937, 10.80728788,  6.383897  ,
        8.91551231,  9.73579738,  8.8737556 ,  8.34949383,  8.10886021,
       12.52372629,  5.79543038, 10.71162681,  9.11473391,  8.7273569 ,
        4.87083348, 13.43127131,  5.79391825,  8.12580976,  7.10378919,
       12.91143491,  9.93553424, 13.34587027,  9.95552626, 13.27994304,
       10.04916678,  5.2287058 , 12.11735334,  8.99791316,  5.80312949,
        9.5014146 ,  5.64500512,  7.72291946,  6.05113217,  7.06272256,
       13.97857591,  9.75200848,  8.43626114,  7.09887557,  6.59176018,
       13.32557133, 10.84493732, 13.58101794, 11.57249984, 11.73848065,
        7.03733775,  9.69792513,  8.29640613,  8.41148722,  7.32950385,
        6.74686071, 13.96746816, 10.00802627,  5.96556248,  8.51088913,
        7.71075226, 12.3344536 ,  5.5205189 ,  9.18358562, 10.6910768 ,
       11.63765273, 10.14436912, 11.8159254 ,  7.24677664,  5.52416368,
       13.01728313])}
{'P': tensor([[1.2531e-02, 1.9196e-01, 1.7735e-01,  ..., 4.0517e-04, 1.2726e-05,
         1.2826e-03],
        [2.6182e-04, 1.0780e-02, 2.6459e-01,  ..., 1.5835e-04, 9.3160e-06,
         1.4629e-03],
        [3.2022e-04, 4.9609e-04, 1.4414e-02,  ..., 1.8792e-04, 1.2156e-05,
         2.0315e-03],
        ...,
        [6.9487e-02, 6.9643e-02, 6.8277e-02,  ..., 7.7392e-02, 1.3755e-04,
         1.0843e-03],
        [1.0979e-03, 6.9526e-04, 6.4436e-04,  ..., 4.1350e-03, 4.6850e-02,
         2.2356e-05],
        [1.4274e-04, 2.3384e-04, 2.5584e-04,  ..., 3.4426e-05, 1.3115e-05,
         8.0432e-02]]), 'X': tensor([[ -1.9714,   7.8563,  -1.8084],
        [ -3.0644,   7.6439,  -1.4322],
        [ -3.2486,   7.5925,  -1.3719],
        [ -4.2271,   7.2387,  -1.8316],
        [ -4.1296,   7.2803,  -2.7194],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.3569,   0.7591,   6.5011],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.4378,  -1.8606,  -2.5668],
        [  0.2480, -11.0153,  -3.0183],
        [ -6.7992,   5.4991,   9.9186],
        [  9.1620,  -7.5573,  -9.9882],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.3539,  -5.5111,   0.3373],
        [ -1.0012, -10.8576,   2.2867],
        [  1.1006,   7.6318,  -2.4877],
        [ -7.6719,  -6.6260,   7.1468],
        [ -7.9925,  -6.1423,   2.8990],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.7239,   4.4298,   3.1216],
        [ -4.6393,   7.0461,   1.1624],
        [  9.3836,   9.5075,  -5.9278],
        [ -7.7239,   4.4298,   3.1216],
        [  5.1519, -10.2387,   9.8191],
        [  5.9526,   3.2525,  -4.3821],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.5382,   7.0672,   4.5097],
        [  1.9241,   7.3503,   7.9004],
        [ -7.7842,   4.3466,  -1.7548],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -7.4808,   4.7462,   2.0266],
        [  6.1989,  -9.7591,   0.8289],
        [  5.7485,  -9.9811,  -3.2355],
        [ -9.1178,   1.5784,   2.1410],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.5057,  -3.3894,  -0.8160],
        [ 10.9599,  -5.1849,  -3.5526],
        [  1.3705,  13.9723,   7.9266],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.2790,  -0.0259,  -5.4116],
        [ 10.6283,  -5.7253,   7.4878],
        [  2.6078, -10.9229,   6.2869],
        [ -9.3384,  -2.6161,  -4.7630],
        [  6.2022,   2.4092,   3.0349],
        [ -9.1178,   1.5784,   2.1410],
        [  3.4401,  13.4540,   4.6576],
        [ -7.8359,   4.2735,   7.2943],
        [ -4.7334,   6.9982,  -0.4392],
        [ 12.6038,   0.7665,  -3.4439],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -8.3717,  -5.4769,  -9.2228],
        [  7.9816,  10.8938,  -0.5378],
        [  5.9912,   3.1453,  -3.9989],
        [  0.9967,  -4.8144,  -9.7370],
        [  8.4691,  10.4571,  -5.0650],
        [ 12.5258,  -0.4200,  -9.3493],
        [  0.5250, -11.0306,   5.5358],
        [ 12.4597,  -0.9136,  -4.8763],
        [  1.1644, -11.0394,   7.8511],
        [ -7.4808,   4.7462,   2.0266],
        [  1.7218,  -4.7576,  -3.3617],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.3736,   6.6310,  -1.5174],
        [  4.6231,   5.4556,   8.4973],
        [ -9.1459,   1.4780,  -4.2403],
        [ -3.4488, -10.0934,   2.5195],
        [  0.2571,  -4.7594,   1.5618],
        [  6.9284,  11.7061,  -0.3732],
        [ -3.5520,   7.4976,   9.1522],
        [  6.3460,   1.4252,   1.2291],
        [  1.9505,  -4.7171,   9.9868],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.6248,   3.9984,   8.4933],
        [  6.3413,   0.4883,  -6.8416],
        [ -7.5720,   4.6310,   2.1717],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -5.3328,  -8.9992,   7.3440],
        [ -5.7347,  -8.6908,   9.2878],
        [  4.5800,   5.5034,  -3.9432],
        [ 12.5717,   1.9266,   7.4903],
        [  8.5198,  10.4091,  -8.6486],
        [ -8.3900,   3.3727,   3.1650],
        [ -6.8466,  -7.6488,   1.9648],
        [  6.1575,  12.2046,   3.3257],
        [ -9.3616,  -2.4750,   3.7765],
        [  7.5064,  -8.9651,  -0.0994],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.0078,  -3.8018,  -6.2168],
        [ -6.3066,  -8.1936,   9.5822],
        [ 10.0619,   8.6627,  -9.2884],
        [ -2.5683, -10.4433,  -2.9182],
        [  1.7218,  -4.7576,  -3.3617],
        [  4.6231,   5.4556,   8.4973],
        [  2.1504,  -4.6726,   8.7908],
        [  1.1644, -11.0394,   7.8511],
        [ 11.7349,  -3.5838,   2.1644],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.5267,  -0.4125,  -8.7331],
        [  5.0506,   4.9277,  -9.5094],
        [ -8.5149,  -5.1905,   1.5239],
        [  0.1497,   7.8334,  -5.4412],
        [  6.1579,  -0.5961,  -6.5595],
        [ -7.4733,   4.7555,   6.2490]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0388, 0.0562,  ..., 0.2245, 0.3944, 0.6631],
        [0.0388, 0.0000, 0.0186,  ..., 0.2561, 0.4332, 0.6283],
        [0.0562, 0.0186, 0.0000,  ..., 0.2665, 0.4498, 0.6170],
        ...,
        [0.2245, 0.2561, 0.2665,  ..., 0.0000, 0.2483, 0.8834],
        [0.3944, 0.4332, 0.4498,  ..., 0.2483, 0.0000, 1.0386],
        [0.6631, 0.6283, 0.6170,  ..., 0.8834, 1.0386, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.09984231,  8.23525433,  8.25828409,  8.38252937,  8.36995953,
        8.47371868,  6.4020404 ,  6.12886024,  9.61942445, 11.018084  ,
        8.74470218, 11.87665501,  6.83515201, 10.00794468, 10.9036226 ,
        7.71075226, 10.1371552 , 10.08001894,  8.95133961,  8.90405823,
        8.43626114, 13.35832725,  8.90405823, 11.46176259,  6.78325151,
       12.45418236,  6.14086582, 12.40074901,  7.50917551,  7.59795445,
        8.91551231,  9.43112957,  8.859416  , 11.56146873, 11.51811603,
        9.25336371, 12.9089559 ,  5.63824971, 12.12448912, 14.0393949 ,
       10.37698435,  6.24187044,  6.27905292, 12.07225469, 11.22993218,
        9.69792513,  6.65368551,  9.25336371, 13.88683436,  8.92549752,
        8.4486771 , 12.62711252,  9.562174  , 10.00410625, 13.50484539,
        6.76661595,  4.91652556, 13.45642379, 12.53285633, 11.04313641,
       12.49317563, 11.10066126,  8.859416  ,  5.05963217,  8.24808832,
        8.53501777,  7.15100303,  9.26455795, 10.66631784,  4.7663526 ,
       13.60275211,  8.29640613,  6.50409372,  5.10446653,  8.1137452 ,
        6.90115901,  6.36004393,  8.87587457,  9.75075273, 10.4606265 ,
       10.412299  ,  7.15990676, 12.71843462, 13.45124917,  9.04255655,
       10.26545792, 13.66991027,  9.68324553, 11.69264243, 10.74409834,
        5.52416368, 10.33958639, 13.27717982, 10.75443106,  5.05963217,
        7.15100303,  5.14371005, 11.10066126, 12.26997165, 12.38881635,
       12.53345445,  7.05627461,  9.97220799,  7.83486943,  6.18668463,
        8.8580671 ])}
{'P': tensor([[1.6997e-02, 2.0037e-01, 1.6244e-01,  ..., 2.7758e-05, 6.2848e-05,
         5.2787e-05],
        [3.4366e-04, 1.8567e-02, 3.1841e-03,  ..., 6.3128e-05, 2.1020e-05,
         3.1846e-05],
        [8.9693e-04, 5.0294e-02, 2.9289e-02,  ..., 5.9033e-05, 4.1431e-05,
         6.3033e-05],
        ...,
        [4.7248e-06, 5.0561e-06, 4.0318e-06,  ..., 5.8504e-02, 6.6559e-07,
         5.9447e-06],
        [4.0917e-02, 6.7104e-02, 3.9828e-02,  ..., 3.5410e-05, 1.9853e-01,
         1.8860e-04],
        [2.8671e-02, 1.3552e-02, 2.4036e-02,  ..., 6.8984e-06, 2.2268e-03,
         6.3944e-02]]), 'X': tensor([[-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0814, 0.0661,  ..., 0.8886, 0.8038, 0.3406],
        [0.0814, 0.0000, 0.0962,  ..., 0.8114, 0.8291, 0.4198],
        [0.0661, 0.0962, 0.0000,  ..., 0.9007, 0.8695, 0.3407],
        ...,
        [0.8886, 0.8114, 0.9007,  ..., 0.0000, 1.1737, 1.2288],
        [0.8038, 0.8291, 0.8695,  ..., 1.1737, 0.0000, 0.8690],
        [0.3406, 0.4198, 0.3407,  ..., 1.2288, 0.8690, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.10886021,  8.31625171,  8.25828409,  8.4486771 ,  8.23525433,
        8.50278896,  5.70405525,  6.34059021,  9.77777346,  8.98765644,
        6.77777606, 12.41012046,  5.14371005,  7.39599986, 13.00413829,
        7.13501264,  6.24187044,  6.89245352,  6.99097013,  5.88782913,
        6.53250937, 13.08414689, 11.22596994, 14.07623266,  7.8575648 ,
        8.72896381, 11.36089355, 12.3489324 , 14.09176259,  8.28968067,
        6.42689705, 11.24434374, 10.8207297 ,  6.18668463,  8.87007495,
       11.42947632, 12.46513646,  8.89458273,  9.36093562,  6.76056625,
       10.4977132 , 12.889695  , 10.00802627,  7.50500064,  7.62011734,
       11.48734978, 13.92390689,  7.60529892,  6.19633139, 12.3828957 ,
        4.91652556, 10.75443106,  5.87305005,  9.562174  ,  6.81961996,
       11.05413545, 13.27994304,  7.71075226, 13.26590071, 13.25017211,
       11.69264243, 10.41909211, 12.0495521 ,  8.68832398,  9.34061938,
        9.87720102, 11.59512194, 11.38774149,  8.77913428, 13.88683436,
        6.67121748,  8.85572292, 10.71162681,  9.25336371, 11.76830981,
        8.57560275, 11.018084  ,  6.65991877,  6.33517349,  9.95552626,
        7.05627461, 13.0743399 ,  8.44940661,  8.859416  , 13.04051744,
       13.0743399 ,  6.25586297, 12.35625694,  8.92395381, 10.00802627,
       11.74901778,  6.89245352, 10.69963954,  4.73784222, 13.1471313 ,
       13.66051884,  7.39599986,  8.77913428, 10.36844813,  7.24326992,
        6.54127431, 13.06957425, 10.88618114, 10.31584102, 14.02532758,
        7.19513806])}
{'P': tensor([[1.7857e-02, 2.2173e-01, 1.9214e-01,  ..., 1.4916e-06, 1.0665e-05,
         1.2618e-06],
        [1.7411e-03, 6.1958e-02, 1.2094e-01,  ..., 6.2357e-06, 4.6040e-05,
         3.2366e-06],
        [8.2598e-04, 6.3351e-03, 3.9152e-02,  ..., 3.5335e-06, 2.1595e-05,
         1.3406e-06],
        ...,
        [6.6695e-06, 4.9337e-06, 3.4863e-06,  ..., 1.9056e-02, 4.3566e-04,
         1.4554e-04],
        [1.1496e-04, 1.0888e-04, 7.1171e-05,  ..., 8.6896e-03, 1.2463e-01,
         1.0027e-03],
        [8.6896e-04, 8.4050e-04, 4.5439e-04,  ..., 1.1759e-06, 2.7337e-05,
         1.1858e-01]]), 'X': tensor([[ -2.0721,   7.8432,   6.6380],
        [ -3.1671,   7.6158,   6.0765],
        [ -3.6381,   7.4683,   7.3544],
        [ -2.7572,   7.7194,   5.9580],
        [ -5.3382,   6.6533,   7.8054],
        [ -4.0868,   7.2981,   8.1874],
        [  7.9403,  -8.6455,  -8.7433],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -2.7572,   7.7194,   5.9580],
        [  6.1575,  12.2046,   3.3257],
        [  4.4915,   5.5990,   7.0405],
        [ -1.5875,   7.8944,  -9.4286],
        [ 12.5199,  -0.4711,   3.0436],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.0449,  13.8402,  -7.1642],
        [ -1.8367,   7.8718,  -8.6523],
        [  5.5158,  -2.1337,   3.5246],
        [ 12.1504,   4.3674,  -7.2998],
        [ -1.5451,   7.8975,   7.6104],
        [ -5.1377,   6.7751,  -0.4438],
        [ 11.6801,  -3.7196,  -0.9181],
        [  9.3560,   9.5391,   9.4943],
        [ -9.4157,   0.0821,  -5.0025],
        [ 11.9641,  -2.9568,   7.7211],
        [  8.7715,  -7.9390,   5.2900],
        [ -7.3554,   4.8985,  -0.6665],
        [ 11.3367,  -4.4806,   0.4643],
        [  2.2437,  -4.6489,   6.1430],
        [  2.7420,  -4.4896,  -3.7139],
        [ -1.2454,   7.9127,  -0.5715],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -4.6393,   7.0461,   1.1624],
        [ 11.1420,   6.9331,  -2.4779],
        [  5.8325,  -1.5222,  -2.1341],
        [ -3.4986,   7.5152,   9.0712],
        [  1.7301,   7.4262,   5.0522],
        [  3.1064,  13.5607,   6.7605],
        [  6.3305,   1.6119,   1.1538],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -8.1531,  -5.8748,   2.6934],
        [ -8.8154,   2.4611,   2.5198],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.6603,   2.8250,   5.9181],
        [  6.2555,   2.1474,   4.1685],
        [  3.4140,  -4.1803,  -8.8224],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.5900,   2.9767,   7.4811],
        [  2.8890,  -4.4316,   1.1600],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.2692,  -1.8882,   3.9253],
        [  5.0277,   4.9589,   7.0175],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -7.2986,  -7.1225,  -3.0814],
        [  6.0305,   3.0296,   0.1375],
        [ 11.5568,   6.0602,   9.3244],
        [ 12.0772,  -2.6014,  -7.1832],
        [  3.0876,   6.7527,  -5.3540],
        [  7.0664,  -9.2590,   8.0879],
        [  5.8318,   3.5567,  -4.7692],
        [ -2.1817,   7.8274,  -4.6298],
        [  1.3670,  -4.7990,   3.2603],
        [  4.9957,  -2.8705,   6.5978],
        [ -0.5385,   7.9037,   6.3830],
        [  5.5294,  -2.1106,   9.0626],
        [  5.9587,  12.3216,   9.1695],
        [ 12.2592,   3.9425,   2.0383],
        [ -8.6801,   2.7809,  -0.8443],
        [  3.9663,   6.1018,  -5.4104],
        [ -1.5451,   7.8975,   7.6104],
        [ -6.7745,  -7.7264,   8.4684],
        [  7.0515,  11.6194,   8.4289],
        [  1.7000,  -4.7609,   2.7935],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 10.9916,   7.2133,  -0.9179],
        [ -0.8569,   7.9156,  -6.5557],
        [ 11.0438,   7.1180,  -5.6991],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -4.9217,   6.8979,  -1.7667],
        [ 12.1751,  -2.2592,  -5.7751],
        [  7.1821,  -9.1845,   9.8548],
        [  5.3602,  -2.3808,   7.2098],
        [ -5.8389,  -8.6056,  -3.5098],
        [  3.9663,   6.1018,  -5.4104],
        [  5.0190,   4.9705,  -7.9752],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.3516,   0.6425,   0.3425],
        [ 10.0567,   8.6697,   4.6329],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.3900,   3.3727,   3.1650],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.9260,  -4.4162,   8.3808],
        [ -7.6525,  -6.6534,   5.1721],
        [  1.3403,  -4.8011,  -1.1698],
        [  0.5960, -11.0335,   3.1144],
        [ 11.5083,  -4.1177,   0.2389],
        [ -8.1426,   3.8048,   9.3542],
        [  6.3360,   0.4258,  -9.2509],
        [  2.3261,  13.7749,   5.9121],
        [  9.5018,  -7.1945,  -1.1892],
        [ -2.1533, -10.5778,   1.0112],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.6816,   9.1536,  -5.8543]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0636, 0.0777,  ..., 0.5336, 0.7899, 1.1422],
        [0.0636, 0.0000, 0.0611,  ..., 0.5048, 0.7844, 1.1514],
        [0.0777, 0.0611, 0.0000,  ..., 0.5650, 0.8446, 1.2080],
        ...,
        [0.5336, 0.5048, 0.5650,  ..., 0.0000, 0.3617, 0.7822],
        [0.7899, 0.7844, 0.8446,  ..., 0.3617, 0.0000, 0.4217],
        [1.1422, 1.1514, 1.2080,  ..., 0.7822, 0.4217, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.11227668,  8.24808832,  8.30729352,  8.19703512,  8.53014915,
        8.36445758, 11.73848065,  9.43737523,  8.19703512, 13.66991027,
        7.17789666,  8.05242591, 12.52875793, 13.27994304, 13.99048094,
        8.08320235,  5.91408504, 12.91143491,  8.04718433,  8.50278896,
       12.25806559, 13.3614598 ,  9.41605684, 12.32408516, 11.83075005,
        8.83725081, 12.18998195,  5.16205087,  5.26067019,  8.01009083,
       12.46513646,  8.43626114, 13.12299145,  6.02789634,  8.28968067,
        7.62509582, 13.91198191,  6.53250937,  9.60080639, 10.04916678,
        9.15253344, 10.5575601 ,  9.10945678,  6.61386436,  5.39723241,
        8.09984231,  9.09119351,  5.29009836, 10.33712258, 10.00794468,
       12.41366754,  7.06169391, 10.09979747, 10.1979669 ,  6.74872981,
       13.0493368 , 12.35421461,  7.42512621, 11.64746882,  6.83083863,
        8.12580976,  4.98989906,  5.76165934,  7.92200342,  5.91854397,
       13.68672855, 12.87752052,  9.11473391,  7.27759233,  8.04718433,
       10.27572673, 13.59170809,  5.05534609,  9.78147542, 13.1471313 ,
        7.96181288, 13.13888841,  9.75075273,  8.47371868, 12.3828957 ,
       11.65923396,  5.86519525, 10.39942497,  7.27759233,  7.06372172,
        9.97964013, 10.48721382,  6.38399789, 13.2778378 , 11.76830981,
        9.04255655,  9.78147542,  6.12886024,  5.29755487, 10.14044489,
        4.98461983, 11.04953767, 12.22276278,  8.98765644,  6.35029274,
       13.96987806, 11.9182956 , 10.794755  , 11.63502679,  5.18277858,
       13.32374056])}
{'P': tensor([[1.2455e-02, 2.0667e-01, 1.3674e-01,  ..., 2.8859e-07, 3.6993e-06,
         1.7099e-05],
        [8.0785e-04, 4.2425e-02, 2.9399e-03,  ..., 8.7378e-07, 6.8293e-06,
         5.4488e-05],
        [1.1265e-03, 2.7251e-01, 3.1498e-02,  ..., 7.3031e-07, 6.4663e-06,
         4.0949e-05],
        ...,
        [1.9196e-04, 1.1905e-04, 1.5767e-04,  ..., 2.0434e-01, 3.7639e-02,
         4.5628e-04],
        [7.2304e-04, 5.8654e-04, 7.5984e-04,  ..., 3.2297e-07, 1.2371e-02,
         2.0413e-06],
        [1.1639e-03, 7.2504e-04, 8.3329e-04,  ..., 2.9914e-04, 6.2555e-04,
         5.7484e-02]]), 'X': tensor([[ -2.0840,   7.8415,  -7.8657],
        [ -3.4100,   7.5436,  -7.6980],
        [ -3.0004,   7.6607,  -7.2640],
        [ -4.4520,   7.1370,  -8.7867],
        [ -3.5903,   7.4847,  -8.9112],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.1066,  -0.7799,  -4.1746],
        [  4.2339,  -3.6268,   6.3125],
        [  5.1519, -10.2387,   9.8191],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.3124,   0.2057,  -0.4593],
        [ -0.9715, -10.8630,   0.3555],
        [  4.3459,  -3.5327,   0.3667],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.3497,   1.3680,  -1.9076],
        [  1.9204,   7.3518,   8.4781],
        [  6.1222,  -0.7261,   8.8204],
        [ -7.0825,   5.2073,  -5.0497],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.2199,   4.6861,  -1.9304],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.8323,  -1.5228,   5.1132],
        [  6.8407,  11.7666,   3.0045],
        [  0.6933,   7.7336,   2.1179],
        [  4.9152,  12.8653,  -1.4459],
        [  5.7485,  -9.9811,  -3.2355],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.3616,  -2.4749,   0.1793],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.3118,   4.5452,   4.2114],
        [  0.5600, -11.0321,  -0.8660],
        [  5.6414, -10.0303,   7.8525],
        [ -6.3306,   5.9272,   5.6935],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -4.3963,   7.1629,   4.5263],
        [ -1.0012, -10.8576,   2.2867],
        [  2.5974,   7.0363,  -7.7810],
        [  0.9659,  -4.8144,  -0.9594],
        [  1.3705,  13.9723,   7.9266],
        [  4.8432,   5.1968,   1.5006],
        [ -7.3446,   4.9113,  -5.9356],
        [ -4.7389,   6.9953,   3.7638],
        [  3.5939,   6.4029,  -1.8357],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.0078,  -3.8018,  -6.2168],
        [ -9.1178,   1.5784,   2.1410],
        [  0.9809,  -4.8145,   6.8706],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.3266,   0.3292,   9.3502],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.9587,  12.3216,   9.1695],
        [ -1.9487,   7.8590,  -3.3330],
        [  7.8375,  -8.7240,  -9.8913],
        [  9.3560,   9.5391,   9.4943],
        [  0.9445,   7.6736,   0.5095],
        [  6.1737,   2.5305,   3.9049],
        [ -8.8363,   2.4086,  -5.6296],
        [ -0.0254, -10.9932,   3.2200],
        [  3.1756, -10.8247,   6.8882],
        [  5.1196,  -2.7160,   4.2705],
        [  0.6363, -11.0348,   6.1414],
        [  6.2242,  -0.3171,   7.4978],
        [  1.8320,   7.3871,   5.1452],
        [ -9.4524,  -1.6813,  -6.6862],
        [  1.3339,   7.5627,  -0.9034],
        [  2.3261,  13.7749,   5.9121],
        [  8.3718,  -8.2953,   8.4666],
        [  1.3705,  13.9723,   7.9266],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.4309,  -0.0599,  -4.0830],
        [  3.9515,  -3.8425,   6.1729],
        [  5.1886,  -2.6250,   6.4677],
        [ -9.3739,   0.3982,  -4.2564],
        [  6.3534,   1.2990,  -7.2942],
        [  4.9531,   5.0576,   5.6794],
        [ 11.7795,  -3.4697,   3.5624],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.9587,  12.3216,   9.1695],
        [  4.9152,  12.8653,  -1.4459],
        [  7.9141,  -8.6656,   9.6948],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.3360,   0.4258,  -9.2509],
        [  5.8201,   3.5841,  -8.0609],
        [  3.6249, -10.7255,   5.6605],
        [  6.1579,  -0.5961,  -6.5595],
        [  6.7792,  11.8084,   1.6251],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -2.0445,   7.8469,  -0.5857],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.2242,  -0.3171,   7.4978],
        [  4.1999,  -3.6543,  -4.1998],
        [  2.5813,  -4.5471,   5.3597],
        [  4.2612, -10.5514,  -6.6268],
        [ -6.2200,  -8.2737,   5.5079],
        [ -3.5903,   7.4847,  -8.9112],
        [ -5.0550,   6.8231,  -9.2502],
        [ 12.2589,  -1.9312,   9.8657],
        [ 11.6544,   5.8273,  -3.1375],
        [ 12.2692,  -1.8882,   3.9253],
        [  7.4132,  11.3526,   0.3322],
        [  5.6095,  -1.9702,  -7.8233]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.6551e-02, 5.0941e-02,  ..., 8.9538e-01, 7.4841e-01,
         3.5462e-01],
        [7.6551e-02, 0.0000e+00, 3.5478e-02,  ..., 9.2702e-01, 8.0814e-01,
         4.3000e-01],
        [5.0941e-02, 3.5478e-02, 0.0000e+00,  ..., 8.9641e-01, 7.7273e-01,
         3.9799e-01],
        ...,
        [8.9538e-01, 9.2702e-01, 8.9641e-01,  ..., 0.0000e+00, 4.1330e-01,
         7.6105e-01],
        [7.4841e-01, 8.0814e-01, 7.7273e-01,  ..., 4.1330e-01, 2.1453e-04,
         4.7291e-01],
        [3.5462e-01, 4.3000e-01, 3.9799e-01,  ..., 7.6105e-01, 4.7291e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.1137452 ,  8.27853076,  8.22728399,  8.41171369,  8.30125201,
        8.29203356,  6.1561646 ,  5.5748624 , 11.46176259,  6.00732193,
        6.31576264, 10.90638066,  5.60064319,  4.978704  ,  6.49538841,
        7.5984809 ,  6.16512856,  8.7907923 , 12.16901615,  7.01475376,
       10.39942497,  6.02779201, 13.6105595 ,  7.76457095, 13.77222752,
       11.51811603,  5.24461068,  9.68323686,  4.78258092,  6.99097013,
       11.04629471, 11.50791859,  8.67227386, 10.14436912,  8.40445872,
       10.9036226 ,  7.50035355,  4.91038884, 14.0393949 ,  7.10378919,
        8.8353674 ,  8.44940661,  7.34250948,  9.43112957,  9.27474386,
       10.01804223,  5.52416368,  9.25336371,  4.91336994,  5.8828986 ,
        6.33517349,  8.22728399, 13.68672855,  8.09704052, 11.72749923,
       13.3614598 ,  7.73151309,  6.67218687,  9.15866607, 10.99326577,
       11.28093507,  5.79543038, 11.05317597,  6.2322769 ,  7.61088125,
        9.60080639,  7.67939546, 13.96987806, 11.785566  , 14.0393949 ,
        5.1224079 ,  9.43112957,  5.51177501,  5.81481044,  9.38232121,
        6.48486171,  7.07902036, 12.27991991,  9.26455795, 13.68672855,
       13.77222752, 11.73567759,  9.75401449,  9.50654967,  6.35029274,
        6.83515201, 11.32149259,  6.18668463, 13.61600536,  9.77777346,
        9.96152788,  8.10886021,  6.34059021,  9.61942445,  6.2322769 ,
        5.5671389 ,  5.2287058 , 11.37939695, 10.35093485,  8.30125201,
        8.49161154, 12.41012046, 13.03002338, 12.41366754, 13.55868684,
        5.94541803])}
{'P': tensor([[1.3117e-02, 2.0837e-01, 1.2431e-01,  ..., 8.7925e-07, 2.8774e-06,
         1.4706e-05],
        [3.7059e-04, 2.1577e-02, 8.4179e-03,  ..., 9.3757e-07, 6.4982e-06,
         2.6958e-05],
        [2.8603e-04, 1.3477e-02, 1.0688e-02,  ..., 4.2749e-07, 2.8430e-06,
         1.1017e-05],
        ...,
        [1.9536e-04, 1.0669e-04, 8.8943e-05,  ..., 1.4850e-02, 2.0343e-07,
         5.8740e-06],
        [6.3346e-06, 4.1936e-06, 3.9870e-06,  ..., 2.2724e-04, 2.7302e-02,
         3.2691e-03],
        [3.0816e-04, 2.2438e-04, 1.9148e-04,  ..., 3.9351e-03, 3.2228e-03,
         1.5052e-01]]), 'X': tensor([[-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0669, 0.0632,  ..., 0.8336, 0.3925, 0.6232],
        [0.0669, 0.0000, 0.0303,  ..., 0.8766, 0.3535, 0.6500],
        [0.0632, 0.0303, 0.0000,  ..., 0.8914, 0.3833, 0.6713],
        ...,
        [0.8336, 0.8766, 0.8914,  ..., 0.0000, 0.8441, 0.3027],
        [0.3925, 0.3535, 0.3833,  ..., 0.8441, 0.0000, 0.5498],
        [0.6232, 0.6500, 0.6713,  ..., 0.3027, 0.5498, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.12043983,  8.30729352,  8.29022197,  8.36445758,  8.53014915,
        8.50268322,  5.91222346,  8.36445758, 10.53368644,  6.69389431,
       10.46238054,  7.05627461, 10.1371552 , 11.75618513,  9.97220799,
        8.99791316, 13.55868684,  5.19577146,  5.68942572, 13.09737053,
        6.97106912,  5.89074104, 11.48663767, 13.59056613, 11.9702762 ,
        8.37954853,  5.89074104,  5.81199422, 12.10942412,  6.05113217,
        5.37343173,  4.978704  ,  8.01009083,  9.99211663, 11.75807252,
        7.50564223, 11.56146873,  7.18354885,  8.72896381,  5.08372149,
       13.13205357, 13.25017211,  5.92952538,  8.68979052, 11.05672136,
       13.36800073,  9.11473391,  7.42512621,  5.82345867,  6.77777606,
        5.36061974,  5.18047638,  5.08372149, 13.04051744,  5.52416368,
       13.99877413,  8.29022197, 10.4977132 ,  9.0368458 ,  6.1561646 ,
        4.78258092,  8.31625171, 10.06703358,  6.60752373,  4.89869205,
        6.80370254,  7.62509582,  8.09899874, 12.84056604,  9.79039371,
        9.85330646, 13.39372426,  9.79039371, 11.75618513,  9.94173708,
       11.70016217, 13.35927527,  5.98285319,  6.42689705, 11.99633819,
       11.85743425,  9.64157461,  7.46053281, 13.58333857,  4.96231532,
        7.06272256,  9.68324553,  8.92098752, 13.99048094,  6.81084724,
        6.48486171, 12.32545637, 12.05300943,  9.64157461, 11.35182589,
       10.61417181,  9.7712321 , 11.00508681,  8.859416  ,  5.41348121,
        9.3349858 ,  7.01475376,  6.81117802, 13.01187484, 11.27656488,
        5.24137909])}
{'P': tensor([[2.0356e-02, 2.4883e-01, 1.6243e-01,  ..., 1.6679e-04, 1.4236e-06,
         3.2738e-05],
        [6.1555e-04, 3.7646e-02, 4.6924e-02,  ..., 7.1084e-04, 2.0792e-06,
         2.2110e-05],
        [1.0519e-03, 2.6326e-02, 5.3267e-02,  ..., 1.1925e-03, 2.8430e-06,
         3.2744e-05],
        ...,
        [1.3832e-06, 1.6807e-06, 1.8045e-06,  ..., 9.5096e-03, 2.0814e-06,
         1.1905e-07],
        [4.1625e-04, 1.6391e-04, 1.5094e-04,  ..., 8.1398e-07, 5.5419e-02,
         1.9032e-04],
        [1.1142e-02, 1.8507e-02, 2.3713e-02,  ..., 2.6335e-04, 1.1804e-06,
         3.1532e-01]]), 'X': tensor([[ -2.1817,   7.8274,  -4.6298],
        [ -3.9700,   7.3453,  -4.1486],
        [ -4.0409,   7.3169,  -3.5779],
        [ -5.5525,   6.5146,  -4.4984],
        [ -5.2589,   6.7024,  -5.5809],
        [ -4.1296,   7.2803,  -2.7194],
        [  4.4290,  -3.4596,   1.4181],
        [  6.1752,   2.5242,   3.5874],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.0788,  -0.8710,  -6.8719],
        [  7.0664,  -9.2590,   8.0879],
        [  5.9587,  12.3216,   9.1695],
        [  8.6271,  -8.0715,  -7.4813],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.0031,  12.2958,  -6.5992],
        [ -4.0409,   7.3169,  -3.5779],
        [  6.3031,   1.8477,  -7.1706],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.1531,  -5.8748,   2.6934],
        [  4.4633,   5.6287,  -0.7754],
        [ -1.6334, -10.7207,   7.1397],
        [  9.4925,   9.3811,   8.9842],
        [  6.2038,   2.4020,  -3.1692],
        [  8.2422,  -8.4041,  -1.4962],
        [ -5.3382,   6.6533,   7.8054],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.6954,  -4.7616,   9.8198],
        [ -5.9225,  -8.5355,   2.4778],
        [ -0.9715, -10.8630,   0.3555],
        [  8.5198,  10.4091,  -8.6486],
        [  3.3633,  13.4794,   0.5193],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.1620,  -7.5573,  -9.9882],
        [ -8.1204,  -5.9308,   4.0761],
        [  4.4039,   5.6903,   8.7366],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.4630,  -1.5145,   0.2983],
        [ -8.9379,  -4.1655,  -2.8114],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.1222,  -0.7261,   8.8204],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.2692,  -1.8882,   3.9253],
        [  7.9141,  -8.6656,   9.6948],
        [  2.6970,   6.9826,   9.7350],
        [ -1.0012, -10.8576,   2.2867],
        [  0.7275,  14.0667,  -0.9815],
        [ -6.7745,  -7.7264,   8.4684],
        [ 11.6634,   5.8050,   9.1590],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.6314,  -1.9302,  -9.6543],
        [ -5.3382,   6.6533,   7.8054],
        [ -1.0012, -10.8576,   2.2867],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -0.9715, -10.8630,   0.3555],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.9452, -10.3187,  -7.8621],
        [ -8.0534,  -6.0429,   2.4583],
        [  2.6614,  -4.5192,  -9.3641],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.9478,  -4.8144,   0.6371],
        [ -8.3653,   3.4184,   9.3454],
        [  1.3339,   7.5627,  -0.9034],
        [  1.7228,  -4.7575,   8.3396],
        [  3.3167,   6.6022,   6.3554],
        [  5.5101,  -2.1433,   9.5442],
        [ -1.9487,   7.8590,  -3.3330],
        [  3.9770, -10.6341,   8.0783],
        [ 11.3800,  -4.3919,   1.4066],
        [ -1.6275, -10.7221,   3.2376],
        [  2.9410,   6.8428,   5.4458],
        [  2.3376,  -4.6231,   9.8436],
        [  8.4691,  10.4571,  -5.0650],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.6895,   5.6051,   2.7899],
        [  6.6319,  11.9066,   3.5165],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.3583,   1.1729,   8.7835],
        [  6.3535,   0.6792,  -7.5101],
        [  6.7792,  11.8084,   1.6251],
        [ -4.2271,   7.2387,  -1.8316],
        [ -1.7009,   7.8851,  -8.2437],
        [ -9.4748,  -0.7765,  -5.6695],
        [  5.6022,  -1.9833,   8.3081],
        [  7.0171,  11.6439,  -5.0178],
        [  4.7482, -10.3907,   5.9149],
        [  0.2571,  -4.7594,   1.5618],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.4140,  -4.1803,  -8.8224],
        [  0.1677,  -4.7447,  -6.1529],
        [  2.3928,   7.1404,  -1.7035],
        [  6.3535,   0.6792,  -7.5101],
        [ -7.0948,   5.1940,   9.8171],
        [  9.3624,   9.5318,   0.2779],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.4267,  -5.3697,   2.4172],
        [ 12.4592,  -0.9168,  -6.7571],
        [  1.5662,  13.9375,   4.5142]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.3004e-02, 9.7097e-02,  ..., 7.8873e-01, 4.9452e-01,
         1.0532e+00],
        [8.3004e-02, 0.0000e+00, 3.6070e-02,  ..., 7.1738e-01, 5.7451e-01,
         1.0924e+00],
        [9.7097e-02, 3.6070e-02, 0.0000e+00,  ..., 6.9267e-01, 5.9135e-01,
         1.0659e+00],
        ...,
        [7.8873e-01, 7.1738e-01, 6.9267e-01,  ..., 0.0000e+00, 1.2761e+00,
         1.2830e+00],
        [4.9452e-01, 5.7451e-01, 5.9135e-01,  ..., 1.2761e+00, 0.0000e+00,
         1.0871e+00],
        [1.0532e+00, 1.0924e+00, 1.0659e+00,  ..., 1.2830e+00, 1.0871e+00,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.12580976,  8.34949383,  8.35856487,  8.55982105,  8.51928531,
        8.36995953,  5.62005629,  6.67121748,  5.47866221,  6.14086582,
       11.64746882, 13.68672855, 11.81423161,  9.76896904, 13.6829843 ,
        8.35856487,  6.56833864,  9.70108409, 10.04916678,  7.18354885,
       10.84437705, 13.34587027,  6.65259775, 11.77124373,  8.53014915,
        8.06990483,  5.05444288, 10.38897563, 10.90638066, 13.45124917,
       13.89264822, 12.46513646, 10.06286074, 11.87665501, 10.05558893,
        7.19534146, 13.02818702,  9.45995955,  9.58347456,  9.86089579,
        4.90505566,  6.16512856, 10.0674887 , 10.31613269, 12.45418236,
       12.41366754, 11.73567759,  7.48538905, 10.9036226 , 14.08549802,
       10.27572673, 13.02818702,  6.14086582, 12.76408562,  5.84450127,
        5.95297459,  8.53014915, 10.9036226 , 10.06407839, 10.90638066,
       11.51811603, 11.4424816 , 10.06850677,  5.24461068,  9.15976746,
        9.95552626,  4.90676242,  9.0368458 ,  7.67939546,  5.05981914,
        7.38844703,  5.91222346,  8.09704052, 11.35345345, 12.19805394,
       10.84493732,  7.44805901,  5.1805345 , 13.45642379, 11.28895489,
        8.7273569 , 13.62896593,  5.91854397,  9.62238828,  6.46560181,
        6.38968357, 13.61600536,  8.38252937,  8.06643487,  9.50654967,
        5.94293185, 13.59480906, 11.42420508,  4.7663526 , 13.25017211,
        5.39723241,  4.74770899,  7.53063885,  6.38968357,  8.79284466,
       13.3607379 ,  5.70405525,  9.50654967,  9.99211663, 12.49291768,
       14.02526505])}
{'P': tensor([[1.4666e-02, 2.1237e-01, 2.0453e-01,  ..., 1.3792e-06, 1.2362e-06,
         4.8870e-04],
        [4.1653e-04, 1.9359e-02, 1.3103e-02,  ..., 2.2328e-06, 1.7799e-06,
         8.4633e-04],
        [6.8596e-04, 2.1704e-02, 3.0954e-02,  ..., 2.6712e-06, 2.6865e-06,
         8.7415e-04],
        ...,
        [1.5613e-05, 1.4799e-05, 1.1035e-05,  ..., 1.1190e-01, 3.2013e-04,
         9.6142e-05],
        [3.0887e-05, 1.8786e-05, 1.7521e-05,  ..., 2.6255e-05, 3.2201e-02,
         2.3272e-06],
        [2.7175e-05, 4.7529e-05, 3.7849e-05,  ..., 8.9848e-05, 3.8901e-06,
         2.7425e-02]]), 'X': tensor([[-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0444, 0.0750,  ..., 1.0050, 0.1066, 0.7911],
        [0.0444, 0.0000, 0.0334,  ..., 1.0113, 0.1458, 0.7659],
        [0.0750, 0.0334, 0.0000,  ..., 1.0335, 0.1790, 0.7656],
        ...,
        [1.0050, 1.0113, 1.0335,  ..., 0.0000, 0.9400, 0.6853],
        [0.1066, 0.1458, 0.1790,  ..., 0.9400, 0.0000, 0.8034],
        [0.7911, 0.7659, 0.7656,  ..., 0.6853, 0.8034, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.1303945 ,  8.29022197,  8.28968067,  8.29640613,  8.36445758,
        8.50268322, 13.04156075,  5.60064319, 10.92597196,  5.14999799,
        9.78285424,  6.99097013, 12.58006038,  7.26074925,  9.85384623,
        6.39438044,  7.63726134, 13.97779839,  5.79543038, 11.83570132,
       11.48663767, 11.70168755,  5.5671389 , 12.86648771, 14.08549802,
        8.44940661,  6.3949043 ,  5.64464807, 11.89855267,  8.41171369,
       13.02818702,  6.65368551,  7.02496223,  4.98461983,  7.19513806,
        5.34250225,  7.62011734, 11.87665501, 12.35625694, 12.10222204,
        9.75075273,  5.05444288,  6.67121748, 12.79126785,  9.44486609,
       10.01804223,  8.62162372, 12.88270018, 13.0493368 ,  9.73579738,
        9.3109884 ,  8.92549752,  7.37576154,  5.5513072 , 11.56146873,
        5.37343173,  8.22728399,  9.11473391,  9.75200848,  9.55593374,
        9.70108409, 10.35485331,  7.62011734,  8.37954853,  9.37296102,
        6.4020404 ,  5.18277858,  6.65259775, 12.50996828,  5.30158497,
        6.69389431, 13.14777433,  5.79562691,  7.01475376,  7.69752262,
        7.96181288, 10.4977132 , 13.25017211,  7.38844703,  8.06517143,
        4.8763135 , 13.5247215 ,  6.00663314,  4.7663526 ,  7.66377554,
        9.18690018,  9.36093562,  9.79039371, 10.31886066, 10.84493732,
       13.26590071, 12.40074901,  6.83515201,  7.13501264, 11.83570132,
       13.14690137,  8.8580671 ,  7.20826219, 10.23946247,  9.77588801,
        9.58347456,  9.23071698,  8.83725081,  5.22903117, 11.89105259,
        8.91551231])}
{'P': tensor([[1.3562e-02, 1.9714e-01, 1.6971e-01,  ..., 6.2059e-07, 4.8954e-05,
         1.6519e-06],
        [4.5468e-04, 2.9497e-02, 1.4578e-02,  ..., 1.2330e-06, 1.3104e-04,
         2.1807e-06],
        [5.0106e-04, 3.4924e-02, 2.8019e-02,  ..., 1.0919e-06, 1.2278e-04,
         2.0800e-06],
        ...,
        [1.4241e-05, 6.7438e-06, 6.2852e-06,  ..., 8.2665e-03, 8.0423e-06,
         5.0303e-03],
        [4.5473e-04, 3.1776e-04, 3.2119e-04,  ..., 3.1252e-03, 1.0904e-01,
         1.3834e-03],
        [4.4838e-03, 1.8507e-03, 1.8496e-03,  ..., 8.8484e-06, 2.1608e-05,
         6.9841e-02]]), 'X': tensor([[ -2.3317,   7.8034,  -3.6274],
        [ -4.0409,   7.3169,  -3.5779],
        [ -3.9700,   7.3453,  -4.1486],
        [ -4.1296,   7.2803,  -2.7194],
        [ -5.5525,   6.5146,  -4.4984],
        [ -6.0195,   6.1789,  -2.9496],
        [  5.1667,  -2.6543,   1.2879],
        [ -4.0409,   7.3169,  -3.5779],
        [ -5.9987,   6.1949,  -6.2114],
        [  4.5767,   5.5071,   3.9069],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.0438,   7.1180,  -5.6991],
        [  1.1274,  14.0116,   8.1753],
        [  6.3604,   0.9042,  -6.5901],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -5.3736,   6.6310,  -1.5174],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.1752,   2.5242,   3.5874],
        [  6.2535,  -0.1709,  -9.4102],
        [ -1.3178, -10.7939,   0.4236],
        [ -9.4117,   0.1160,   8.2574],
        [ -6.4002,  -8.1048,  -0.9108],
        [  5.1861,   4.7361,  -3.7312],
        [  8.8091,  -7.9037,   0.3631],
        [ -6.0844,  -8.3955,   0.7705],
        [  7.0515,  11.6194,   8.4289],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.2630,   2.1057,  -5.2136],
        [ 10.6673,  -5.6649,  -4.1646],
        [  3.9913,  -3.8139,  -6.8311],
        [  0.9659,  -4.8144,  -0.9594],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.6681,  -4.8556,   9.2518],
        [ -1.5408,   7.8978,   5.3470],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -1.5875,   7.8944,  -9.4286],
        [  5.4426, -10.1181,  -5.1268],
        [  0.3354,  -4.7708,  -4.9125],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.4639, -10.9431,  -7.3718],
        [ -4.7389,   6.9953,   3.7638],
        [  5.7802,  -1.6363,  -6.2782],
        [ -8.6675,  -4.8569,  -9.1769],
        [  0.7685,  -4.8098,  -2.9310],
        [  1.0000,   7.6591,  -4.5011],
        [  6.9433,  -9.3362,  -0.3421],
        [  1.3103,  -4.8032,  -0.9300],
        [  1.5383, -11.0274,  -4.9666],
        [  5.8323,  -1.5228,   5.1132],
        [  1.5653,  13.9377,   4.1773],
        [ -7.2515,   5.0195,   4.6434],
        [  5.5240,  -2.1198,   6.2055],
        [  0.9809,  -4.8145,   6.8706],
        [ -1.0012, -10.8576,   2.2867],
        [  2.3367,   7.1676,  -9.5263],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.4430,  -0.1897,   4.8383],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.0890,  -0.8382,   2.9166],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.7992,  -3.9474,  -8.9019],
        [ -2.2188,   7.8218,   8.6064],
        [  3.1271, -10.8343,   2.6442],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.8450,  -9.9356,   3.0767],
        [  2.8474,   6.8979,   6.6457],
        [  2.2849, -10.9655,  -7.6542],
        [ 12.4691,   2.8525,  -5.5110],
        [ -3.3812, -10.1235,   7.4384],
        [ -7.3333,   4.9246,  -7.9212],
        [ 11.7418,   5.6070,  -1.4192],
        [  0.9074,  -4.8139,   8.1674],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.4630,  -1.5145,   0.2983],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.4758,  -5.9539,   1.3667],
        [  4.5800,   5.5034,  -3.9432],
        [  4.4915,   5.5990,   7.0405],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.3534,   1.2990,  -7.2942],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -8.6229,   2.9068,   2.4429],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.8940, -11.0402,  -0.9783],
        [  6.2630,   2.1057,  -5.2136],
        [  6.9693,  -9.3200,  -8.3822],
        [  5.6414, -10.0303,   7.8525],
        [  6.0345,   3.0175,  -1.4424],
        [  6.3542,   0.6934,  -3.3840],
        [  2.9461,  -4.4077,  -0.4831],
        [  3.2159, -10.8166,  -1.0406],
        [  9.2979,   9.6048,   7.3138],
        [ -5.3328,  -8.9992,   7.3440],
        [ -7.0948,   5.1940,   9.8171],
        [  4.8051, -10.3703,   0.5268],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.4196,   0.0478,   9.4140],
        [ -5.1046,  -9.1610,   2.3418],
        [ -5.3736,   6.6310,  -1.5174],
        [ -1.8788, -10.6567,   3.7068],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 12.5199,  -0.4711,   3.0436],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 11.7795,  -3.4697,   3.5624],
        [  1.7122,  -4.7591,  -3.6734],
        [ 11.4743,   6.2478,  -2.1100]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 8.9926e-02, 9.7740e-02,  ..., 6.4918e-01, 2.3683e-01,
         5.6411e-01],
        [8.9926e-02, 0.0000e+00, 3.6070e-02,  ..., 7.0200e-01, 3.2666e-01,
         6.5398e-01],
        [9.7740e-02, 3.6070e-02, 0.0000e+00,  ..., 7.3093e-01, 3.2814e-01,
         6.5706e-01],
        ...,
        [6.4918e-01, 7.0200e-01, 7.3093e-01,  ..., 0.0000e+00, 5.7026e-01,
         5.6542e-01],
        [2.3683e-01, 3.2666e-01, 3.2814e-01,  ..., 5.7026e-01, 0.0000e+00,
         3.2901e-01],
        [5.6411e-01, 6.5398e-01, 6.5706e-01,  ..., 5.6542e-01, 3.2901e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.14433542,  8.35856487,  8.34949383,  8.36995953,  8.55982105,
        8.62631801,  5.80860165,  8.35856487,  8.62329466,  7.16058587,
       12.53047489, 13.13888841, 14.05687815,  6.42439344, 10.54658597,
        8.53501777, 10.48721382,  6.67121748,  6.25586297, 10.87409005,
        9.41244865, 10.32715715,  7.02326515, 11.83509372, 10.36844813,
       13.59170809,  5.94541803,  6.60752373, 12.07818871,  5.5205189 ,
        4.91038884,  8.1137452 ,  9.9353921 ,  8.04664886, 10.51437947,
        8.05242591, 11.4890861 ,  4.78258092, 12.05395056, 11.21703654,
        8.44940661,  6.00732193,  9.93553424,  4.87083348,  7.72415048,
       11.63502679,  4.978704  , 11.1341798 ,  6.02779201, 14.02532758,
        8.81930525,  5.91677188,  4.91336994, 10.9036226 ,  7.53884104,
        5.70405525,  9.44486609, 12.52875793,  6.1463896 , 10.53368644,
        5.47866221,  8.1303945 , 11.27656488,  9.70108409, 11.52733581,
        7.46248947, 11.20100795, 12.79126785, 10.67322527,  8.83341608,
       13.01187484,  4.89869205, 12.35421461,  9.58347456, 13.45642379,
       12.0495521 ,  7.15990676,  7.17789666,  5.53175275,  6.48486171,
       10.14436912,  9.09963894, 13.06498554, 11.07637531,  6.60752373,
       11.63765273, 11.50791859,  6.74686071,  6.39187285,  5.30158497,
       11.2845608 , 13.36800073, 10.4606265 ,  8.79284466, 11.42947632,
        8.53014915,  9.41970031, 10.48721382,  8.53501777, 10.82106505,
       10.54658597, 12.52875793, 10.5575601 , 12.27991991,  5.05774295,
       13.06498554])}
{'P': tensor([[1.8095e-02, 2.0515e-01, 1.4517e-01,  ..., 7.6988e-05, 2.2104e-05,
         2.1183e-03],
        [1.1471e-03, 2.6306e-02, 1.0497e-01,  ..., 9.7055e-05, 2.9966e-05,
         1.9947e-03],
        [6.7155e-04, 1.6284e-03, 3.5841e-02,  ..., 2.9767e-04, 4.2658e-05,
         3.4842e-03],
        ...,
        [1.3919e-05, 1.4288e-05, 3.1945e-05,  ..., 7.9520e-02, 2.4643e-05,
         1.5958e-04],
        [9.4411e-04, 7.1051e-04, 5.2255e-04,  ..., 1.5269e-05, 5.6129e-02,
         2.9573e-04],
        [7.1247e-04, 8.7010e-04, 2.9554e-03,  ..., 1.0722e-02, 5.2446e-05,
         5.8591e-02]]), 'X': tensor([[ -2.5056,   7.7720,   5.5878],
        [ -3.1671,   7.6158,   6.0765],
        [ -4.3963,   7.1629,   4.5263],
        [ -5.9940,   6.1985,   5.9566],
        [ -2.7572,   7.7194,   5.9580],
        [ -6.3306,   5.9272,   5.6935],
        [  5.9883,   3.1536,   6.2320],
        [ 11.4743,   6.2478,  -2.1100],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.3444,   0.5821,   6.9101],
        [  4.2339,  -3.6268,   6.3125],
        [  4.0362,  13.2397,  -7.0030],
        [  5.8836,   3.4315,  -4.0327],
        [  4.1637,  13.1898,  -1.9189],
        [ -8.7078,   2.7182,   1.8146],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.8979,   3.3958,  -3.8713],
        [  4.8051, -10.3703,   0.5268],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -4.4779,  -9.5609,  -2.4505],
        [  4.4039,   5.6903,   8.7366],
        [ 10.3251,   8.2928,  -3.0006],
        [  5.7151,  -1.7700,   3.1542],
        [  9.1409,  -7.5789,   1.5814],
        [  6.1737,   2.5305,   3.9049],
        [ -7.4808,   4.7462,   2.0266],
        [  5.3602,  -2.3808,   7.2098],
        [  8.2422,  -8.4041,  -1.4962],
        [ -4.3963,   7.1629,   4.5263],
        [  3.4009,  -4.1874,  -5.9751],
        [ -6.4464,   5.8272,   7.5038],
        [ 12.5224,  -0.4497,   4.8448],
        [  4.9452, -10.3187,  -7.8621],
        [ -7.5401,   4.6718,   9.0033],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.5717,   1.9266,   7.4903],
        [  2.0039,   7.3173,  -1.5885],
        [  6.2247,  -0.3150,  -9.3657],
        [ -4.7847,  -9.3729,  -3.7741],
        [  3.7916,  -3.9524,  -7.9547],
        [ -1.8367,   7.8718,  -8.6523],
        [ -5.1046,  -9.1610,   2.3418],
        [ -0.8455, -10.8853,  -6.7400],
        [ -5.9987,   6.1949,  -6.2114],
        [  9.4554,   9.4245,   9.8800],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.0588,  -6.0341,  -2.3006],
        [  3.5284, -10.7484,   5.3347],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -7.7842,   4.3466,  -1.7548],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -4.7334,   6.9982,  -0.4392],
        [ -1.9714,   7.8563,  -1.8084],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.4501,  -1.7133,   2.2980],
        [ -8.1531,  -5.8748,   2.6934],
        [  3.1480,  -4.3165,  -7.8002],
        [  5.8889,  -1.3916,   8.8993],
        [  4.8051, -10.3703,   0.5268],
        [  5.1196,  -2.7160,   4.2705],
        [ -0.9715, -10.8630,   0.3555],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.3705,  13.9723,   7.9266],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.3087,   0.7792,  -7.5587],
        [  6.1929,   2.4501,  -6.9352],
        [ 11.7794,   5.5086,  -9.6929],
        [  6.5428,  -9.5728,  -2.9897],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.8889,  -1.3916,   8.8993],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.0445,   7.8469,  -0.5857],
        [ -0.9715, -10.8630,   0.3555],
        [  6.3606,   0.9204,   3.2525],
        [  7.9403,  -8.6455,  -8.7433],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.5158,  -2.1337,   3.5246],
        [ 12.5224,  -0.4497,   4.8448],
        [ -4.7334,   6.9982,  -0.4392],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.3551,   0.7149,   1.3344],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.2902,   0.0451,   3.7116],
        [ -7.0948,   5.1940,   9.8171],
        [ -5.3328,  -8.9992,   7.3440],
        [ 11.5568,   6.0602,   9.3244],
        [  0.2480, -11.0153,  -3.0183],
        [  8.2154,  -8.4262,   8.0171],
        [ -5.1398,  -9.1367,   6.2074],
        [ 11.7162,   5.6728,  -8.9372],
        [ -7.0120,   5.2825,   7.7762],
        [  0.2571,  -4.7594,   1.5618],
        [  7.0515,  11.6194,   8.4289],
        [  2.4356,  -4.5942,  -5.0808],
        [ 12.3357,   3.6001,   0.7903],
        [  1.2273,  -4.8081,   9.8129],
        [  6.0788,  -0.8710,  -6.8719],
        [ -0.5739,   7.9056,   0.6382],
        [ -9.4196,   0.0478,   9.4140],
        [  7.9816,  10.8938,  -0.5378],
        [  9.9756,  -6.6330,  -3.3429],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.7151,  -1.7700,   3.1542],
        [ -4.9217,   6.8979,  -1.7667]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0460, 0.1188,  ..., 1.0736, 0.4031, 0.5711],
        [0.0460, 0.0000, 0.1184,  ..., 1.0848, 0.4484, 0.5974],
        [0.1188, 0.1184, 0.0000,  ..., 0.9668, 0.4618, 0.4937],
        ...,
        [1.0736, 1.0848, 0.9668,  ..., 0.0000, 1.1333, 0.5720],
        [0.4031, 0.4484, 0.4618,  ..., 1.1333, 0.0000, 0.5614],
        [0.5711, 0.5974, 0.4937,  ..., 0.5720, 0.5614, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.16584537,  8.24808832,  8.40445872,  8.62260327,  8.19703512,
        8.67227386,  6.76791483, 13.06498554,  9.09661212,  9.36256096,
        5.5748624 , 13.84126197,  6.81117802, 13.83139024,  9.12220271,
       10.32715715,  9.84314712,  6.80559669, 11.42947632,  9.4997987 ,
       10.5575601 ,  7.19534146, 13.2430377 ,  5.98285319, 11.87412439,
        6.67218687,  8.859416  ,  5.86519525, 11.77124373,  8.40445872,
        5.39450305,  8.68979052, 12.53047489, 11.4424816 ,  8.87007495,
       11.81423161, 12.71843462,  7.58668144,  6.23262289, 10.52357479,
        5.47702766,  8.08320235, 10.48721382, 10.91805683,  8.62329466,
       13.35013374,  5.91677188, 10.0674887 , 11.31276109, 12.49291768,
        8.91551231, 12.00995898,  8.4486771 ,  8.09984231,  9.43737523,
        9.60412735, 10.04916678,  5.34250225,  6.05113217, 11.42947632,
        5.79543038, 10.90638066, 12.3828957 , 14.0393949 ,  6.69389431,
        9.34127013,  6.65991877, 13.00380624, 11.59512194,  5.41348121,
        6.05113217, 13.86480175,  8.10886021, 10.90638066,  6.42689705,
       11.73848065,  8.22728399,  5.91408504, 12.53047489,  8.4486771 ,
        9.11473391,  6.39520397, 12.3489324 ,  6.29034958,  8.79284466,
       10.4606265 , 13.0493368 , 11.018084  , 11.76830981, 10.48315436,
       13.01728313,  8.77913428,  4.7663526 , 13.59170809,  5.19986797,
       12.85032836,  4.96231532,  6.14086582,  7.92644746,  9.41970031,
       13.50484539, 11.97957425, 10.52587243,  9.17905491,  5.98285319,
        8.47371868])}
{'P': tensor([[1.3861e-02, 2.1441e-01, 2.0315e-01,  ..., 2.5596e-07, 1.7489e-06,
         1.8949e-06],
        [3.5069e-04, 1.3772e-02, 2.5423e-01,  ..., 3.4167e-07, 2.2751e-06,
         2.1218e-06],
        [4.4776e-04, 9.0138e-04, 1.7954e-02,  ..., 4.5812e-07, 3.0389e-06,
         2.7932e-06],
        ...,
        [1.4669e-06, 1.3527e-06, 1.3522e-06,  ..., 1.1279e-01, 4.6806e-03,
         1.2976e-04],
        [5.8211e-06, 4.8080e-06, 4.7583e-06,  ..., 6.2659e-04, 4.9558e-02,
         4.1932e-03],
        [1.2598e-04, 8.6802e-05, 8.4485e-05,  ..., 2.1975e-05, 4.8415e-04,
         2.3934e-01]]), 'X': tensor([[ -2.5161,   7.7699,  -9.3531],
        [ -3.5173,   7.5091,  -8.9585],
        [ -3.5903,   7.4847,  -8.9112],
        [ -4.4520,   7.1370,  -8.7867],
        [ -5.0550,   6.8231,  -9.2502],
        [ -2.8162,   7.7059,  -9.6885],
        [  0.2480, -11.0153,  -3.0183],
        [  6.3542,   0.6934,  -3.3840],
        [  2.7420,  -4.4896,  -3.7139],
        [ -7.4733,   4.7555,   6.2490],
        [  0.9788,  -4.8145,  -3.8661],
        [  1.8766,   7.3694,   1.1887],
        [  1.9312,  13.8649,   8.3563],
        [ -8.7544,   2.6096,   3.3146],
        [  6.3211,   1.7017,  -0.2164],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.3550,   0.7129,   0.4917],
        [  0.6396,  14.0772,  -6.3786],
        [ 11.5568,   6.0602,   9.3244],
        [  3.6154,  -4.0640,  -7.9698],
        [  2.6078, -10.9229,   6.2869],
        [ -9.4157,   0.0821,  -5.0025],
        [  7.1697,  11.5343,  -0.7276],
        [  1.6232,  -4.7717,  -2.5655],
        [  8.5198,  10.4091,  -8.6486],
        [ 10.1503,   8.5414,  -7.2681],
        [ -9.2687,  -2.9794,   0.6436],
        [ 12.5770,   1.8512,  -0.6578],
        [ 12.6059,   0.9597,  -6.3264],
        [  1.8766,   7.3694,   1.1887],
        [ -6.0844,  -8.3955,   0.7705],
        [  1.7000,  -4.7609,   2.7935],
        [  5.0190,   4.9705,  -7.9752],
        [  4.9957,  -2.8705,   6.5978],
        [  2.6613,   7.0021,   4.5783],
        [  9.6662,   9.1725,   8.1056],
        [ -5.0550,   6.8231,  -9.2502],
        [ -4.9217,   6.8979,  -1.7667],
        [  2.7295,  13.6703,  -8.9328],
        [  3.7121,   6.3117,   0.6738],
        [  3.6249, -10.7255,   5.6605],
        [  8.8091,  -7.9037,   0.3631],
        [  6.1755,   2.5233,  -2.0042],
        [ -4.2271,   7.2387,  -1.8316],
        [ -5.6423,   6.4537,  -1.2091],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.9260,  -4.4162,   8.3808],
        [ -3.4986,   7.5152,   9.0712],
        [  8.7131,  10.2216,   7.7915],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.1813,  -3.3550,   1.8827],
        [  2.3373,  -4.6232,  -5.2246],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.0608,   3.9362,  -5.9567],
        [ -2.6485, -10.4151,   0.8766],
        [  8.3791,  10.5409,   5.5590],
        [ -4.6942,  -9.4299,  -9.1846],
        [  0.6142,  14.0802,   0.5407],
        [ -1.8823, -10.6558,  -3.8952],
        [  9.1409,  -7.5789,   1.5814],
        [  3.9913,  -3.8139,  -6.8311],
        [ 12.2218,   4.0953,  -8.2072],
        [ 11.7794,   5.5086,  -9.6929],
        [  2.2659,  13.7894,   6.5436],
        [  6.3589,   0.8239,  -8.4230],
        [  6.3124,   0.2057,  -0.4593],
        [  2.0547,   7.2957,  -1.9343],
        [  4.5767,   5.5071,   3.9069],
        [  0.3715,  -4.7756,   0.2312],
        [ -7.3470,  -7.0615,   5.0066],
        [  2.6078, -10.9229,   6.2869],
        [ -9.3087,   0.7792,  -7.5587],
        [  6.3589,   0.8239,  -8.4230],
        [  6.1685,  12.1979,  -6.2473],
        [  5.0190,   4.9705,  -7.9752],
        [  5.3858,   4.4260,  -9.6769],
        [  2.3376,  -4.6231,   9.8436],
        [ -7.7239,   4.4298,   3.1216],
        [  3.3633,  13.4794,   0.5193],
        [  2.3367,   7.1676,  -9.5263],
        [  5.7151,  -1.7700,   3.1542],
        [  6.2419,  -0.2313,   4.2440],
        [  3.4401,  13.4540,   4.6576],
        [  3.2159, -10.8166,  -1.0406],
        [  9.9058,  -6.7203,   5.6928],
        [  9.9750,  -6.6338,  -1.2440],
        [ -9.3444,   0.5821,   6.9101],
        [  1.9241,   7.3503,   7.9004],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.1379,   4.4125,  -3.8833],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.6321,  -4.5296,  -2.5159],
        [  0.3231,  -4.7691,  -1.6120],
        [ -2.5056,   7.7720,   5.5878],
        [ -0.4341, -10.9473,   7.4509],
        [  4.9957,  -2.8705,   6.5978],
        [  1.7000,  -4.7609,   2.7935],
        [  6.1741,  -0.5329,   3.1913],
        [ 12.0845,  -2.5772,  -4.6073],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.5456,   6.0861,   8.1101],
        [  0.7958,  -4.8109,  -8.2248],
        [  0.3715,  -4.7756,   0.2312],
        [  2.1440, -10.9811,   8.5281],
        [  3.2647, -10.8066,  -0.3768],
        [  7.0358,  -9.2784,  -9.8479]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0338, 0.0502,  ..., 1.1521, 0.4763, 0.2160],
        [0.0338, 0.0000, 0.0164,  ..., 1.1450, 0.4636, 0.2485],
        [0.0502, 0.0164, 0.0000,  ..., 1.1419, 0.4582, 0.2644],
        ...,
        [1.1521, 1.1450, 1.1419,  ..., 0.0000, 0.6908, 1.2736],
        [0.4763, 0.4636, 0.4582,  ..., 0.6908, 0.0000, 0.6392],
        [0.2160, 0.2485, 0.2644,  ..., 1.2736, 0.6392, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.16714836,  8.29203356,  8.30125201,  8.41171369,  8.49161154,
        8.20436764, 11.018084  ,  6.39187285,  5.26067019,  8.8580671 ,
        4.91296205,  7.60463349, 13.99877413,  9.13507321,  6.54615887,
        5.9510489 ,  6.3949043 , 14.09176259, 13.0493368 ,  5.43943791,
       11.22993218,  9.41605684, 13.58101794,  5.04028538, 13.45124917,
       13.26590071,  9.73579738, 12.7125079 , 12.64235359,  7.60463349,
       10.36844813,  5.05534609,  7.06372172,  5.76165934,  7.49076899,
       13.32557133,  8.49161154,  8.47371868, 13.94008948,  7.32233733,
       11.32149259, 11.83509372,  6.67108662,  8.38252937,  8.5724042 ,
       12.3489324 ,  5.29755487,  8.28968067, 13.43127131,  6.53290663,
        9.7751189 ,  5.18047638,  8.95983586,  8.97052115, 10.74655346,
       13.46553623, 10.53368644, 14.09357547, 10.8207297 , 11.87412439,
        5.5205189 , 12.889695  , 13.00380624, 13.97430191,  6.41202986,
        6.31576264,  7.57946459,  7.16058587,  4.79003361, 10.19036081,
       11.22993218,  9.34127013,  6.41202986, 13.66896739,  7.06372172,
        6.97106912,  5.1805345 ,  8.90405823, 13.89264822,  7.53884104,
        5.98285319,  6.24614905, 13.88683436, 11.2845608 , 11.9702762 ,
       11.97949273,  9.36256096,  7.59795445,  9.66977802, 12.91505111,
        9.73579738,  5.23878371,  4.78002665,  8.16584537, 10.95593729,
        5.76165934,  5.05534609,  6.19707944, 12.35625694, 13.79762415,
       13.05148826,  4.8763135 ,  4.79003361, 11.18839169, 11.28895489,
       11.64436684])}
{'P': tensor([[1.5046e-02, 3.4091e-01, 1.0345e-01,  ..., 2.7608e-05, 1.9304e-06,
         5.5496e-06],
        [6.3847e-04, 2.0932e-02, 1.7569e-01,  ..., 3.2648e-05, 2.4700e-06,
         6.0870e-06],
        [1.0855e-04, 1.2574e-04, 1.5125e-02,  ..., 9.7690e-06, 1.4214e-06,
         1.0564e-06],
        ...,
        [1.3674e-02, 1.2047e-02, 5.8281e-03,  ..., 1.5101e-01, 6.3146e-04,
         1.6207e-02],
        [1.0474e-04, 9.1680e-05, 4.7740e-05,  ..., 9.3165e-03, 8.1917e-02,
         3.1124e-03],
        [7.9869e-05, 8.0285e-05, 8.3461e-05,  ..., 2.7312e-05, 2.2803e-06,
         1.1051e-02]]), 'X': tensor([[-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0295, 0.1655,  ..., 0.6324, 0.9477, 1.1049],
        [0.0295, 0.0000, 0.1374,  ..., 0.6507, 0.9648, 1.1237],
        [0.1655, 0.1374, 0.0000,  ..., 0.7101, 1.0133, 1.1799],
        ...,
        [0.6324, 0.6507, 0.7101,  ..., 0.0000, 0.3174, 0.4731],
        [0.9477, 0.9648, 1.0133,  ..., 0.3174, 0.0000, 0.1753],
        [1.1049, 1.1237, 1.1799,  ..., 0.4731, 0.1753, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.19703512,  8.24808832,  8.62260327,  8.67227386,  8.40445872,
        8.67502348, 10.1371552 ,  9.48337063, 10.09979747, 11.28215429,
        9.3109884 , 11.73397309,  5.86519525,  9.3349858 ,  8.27853076,
        6.38105015,  8.35856487,  8.8580671 , 12.49291768,  7.16058587,
        9.50913849, 10.33958639,  9.58347456, 11.018084  , 11.63502679,
       13.0743399 ,  8.87587457, 13.2778378 ,  4.85603886,  5.60064319,
        7.20482224, 12.72524615, 13.6660671 , 11.83075005,  6.97106912,
       12.10222204,  7.59795445, 10.39942497, 10.4977132 , 12.67680439,
       10.4977132 , 12.13568598, 13.04051744,  9.97220799, 13.13205357,
       13.03002338, 13.86480175,  8.53501777,  5.21280156, 11.32149259,
       11.63765273,  8.41148722,  9.75200848,  8.08320235,  6.39438044,
       13.55868684, 11.04629471, 10.61417181, 12.49291768,  7.5158489 ,
        8.89301516,  5.69913817, 10.99326577,  6.77777606,  4.79003361,
        6.90115901,  8.53014915,  8.37954853,  6.42689705, 11.83570132,
        9.77588801,  7.26074925, 12.44177641, 13.96987806, 11.86239961,
       11.92133129,  7.32950385,  8.67227386,  8.53501777,  9.27474386,
        8.57560275,  5.94293185, 13.79676654, 13.1471313 , 11.22596994,
       12.62576026,  5.63824971,  5.16205087, 11.54071859,  5.52416368,
        7.48538905, 14.09176259, 11.48663767,  5.91854397,  5.39723241,
       11.8159254 ,  6.54693987,  7.37902766,  6.24187044, 10.31613269,
        5.16077764,  9.7712321 , 13.3614598 ,  7.26074925,  6.38105015,
       13.99048094])}
{'P': tensor([[2.2937e-02, 2.4425e-01, 2.1535e-01,  ..., 2.8333e-06, 1.0100e-06,
         3.9891e-05],
        [5.1966e-04, 4.3335e-02, 1.6618e-03,  ..., 2.7484e-06, 1.0779e-06,
         1.0088e-04],
        [7.4463e-04, 4.7170e-01, 3.6143e-02,  ..., 2.7098e-06, 1.1396e-06,
         8.3276e-05],
        ...,
        [9.2944e-03, 2.7894e-03, 3.9478e-03,  ..., 1.2471e-01, 1.1683e-03,
         3.1705e-05],
        [2.1411e-04, 1.0497e-04, 1.5667e-04,  ..., 5.1811e-04, 3.5095e-02,
         8.4537e-06],
        [3.0437e-04, 1.9787e-04, 2.4047e-04,  ..., 1.2642e-03, 1.1305e-03,
         1.0392e-01]]), 'X': tensor([[ -2.8162,   7.7059,  -9.6885],
        [ -5.0550,   6.8231,  -9.2502],
        [ -4.4520,   7.1370,  -8.7867],
        [ -3.5903,   7.4847,  -8.9112],
        [ -3.5173,   7.5091,  -8.9585],
        [ -5.9872,   6.2037,  -9.4477],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.9379,  -4.1655,  -2.8114],
        [  1.8320,   7.3871,   5.1452],
        [ -9.3464,   0.5708,  -1.6397],
        [ -5.9225,  -8.5355,   2.4778],
        [  2.6078, -10.9229,   6.2869],
        [  6.2576,   2.1359,  -3.3398],
        [  6.3583,   1.1729,   8.7835],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.2608,  12.6977,   4.9166],
        [  6.3551,   0.7149,   1.3344],
        [  6.1741,  -0.5329,   3.1913],
        [  5.9026,   3.3837,  -0.6573],
        [ -7.5603,   4.6459,   8.2002],
        [ -3.9700,   7.3453,  -4.1486],
        [  9.9756,  -6.6330,  -3.3429],
        [  0.7145,  -4.8071,  -2.3766],
        [  5.1886,  -2.6250,   6.4677],
        [ -2.5056,   7.7720,   5.5878],
        [  2.0449,  13.8402,  -7.1642],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.4351,  -1.8891,   3.5642],
        [  3.4140,  -4.1803,  -8.8224],
        [  7.0664,  -9.2590,   8.0879],
        [  2.4572,   7.1085,   7.7376],
        [  4.9337,  12.8566,  -3.6582],
        [  5.1667,  -2.6543,   1.2879],
        [  2.3928,   7.1404,  -1.7035],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.1752,   2.5242,   3.5874],
        [  9.6587,   9.1817,  -0.8212],
        [  5.6258,  -1.9404,  -6.0715],
        [ -6.7643,  -7.7372,  -9.6663],
        [  3.2809,  13.5061,  -4.0199],
        [  1.5653,  13.9377,   4.1773],
        [  0.1047, -11.0046,   0.5188],
        [  2.0243,   7.3086,   7.5432],
        [  0.1041, -11.0045,   2.4311],
        [  0.1252, -11.0062,   7.2564],
        [ -8.6603,   2.8250,   5.9181],
        [  0.6396,  14.0772,  -6.3786],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.3464,   0.5708,  -1.6397],
        [ -8.3717,  -5.4769,  -9.2228],
        [  3.8111, -10.6787,  -4.6539],
        [  4.1637,  13.1898,  -1.9189],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.9693,  -9.3200,  -8.3822],
        [ -6.8466,  -7.6488,   1.9648],
        [ 11.0438,   7.1180,  -5.6991],
        [ -9.1231,  -3.5739,   8.4708],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.4541,  -0.3327,   8.8246],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.1755,   2.5233,  -2.0042],
        [  6.3303,   1.6145,  -8.4083],
        [  1.9241,   7.3503,   7.9004],
        [  6.3048,   0.1469,  -9.1691],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 12.5717,   1.9266,   7.4903],
        [  1.6232,  -4.7717,  -2.5655],
        [ -8.9282,   2.1648,   8.4894],
        [  9.9756,  -6.6330,  -3.3429],
        [  2.3261,   7.1726,   6.2522],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.3858,   4.4260,  -9.6769],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.3118,   4.5452,   4.2114],
        [  6.3236,   0.3011,  -4.6668],
        [  4.6231,   5.4556,   8.4973],
        [  1.2273,  -4.8081,   9.8129],
        [  2.5375,   7.0676,  -4.3490],
        [  0.7145,  -4.8071,  -2.3766],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.3464,   0.5708,  -1.6397],
        [  1.7218,  -4.7576,  -3.3617],
        [ -9.2508,   1.0572,   2.9978],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2804,   2.0019,   0.8607],
        [ -5.6651,   6.4380,   1.3660],
        [  5.1203,  -2.7151,   6.2110],
        [  7.5930,  -8.9037,   7.2323],
        [ -8.6603,   2.8250,   5.9181],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.2576,   2.1359,  -3.3398],
        [  6.2576,   2.1359,  -3.3398],
        [  0.9659,  -4.8144,  -0.9594],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.5149,  -5.1905,   1.5239],
        [  3.9022,  13.2906,   5.5788],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.1989,  -9.7591,   0.8289],
        [  7.5784,  -8.9141,   5.8410],
        [  1.4489,   7.5256,   4.4529],
        [  7.9141,  -8.6656,   9.6948],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.1469,   6.9237,   1.6160],
        [  2.7976,  -4.4682,  -4.2020]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0669e-01, 6.4464e-02,  ..., 7.0445e-01, 9.2271e-01,
         4.1262e-01],
        [1.0669e-01, 2.1453e-04, 4.2871e-02,  ..., 8.0986e-01, 9.9952e-01,
         5.0418e-01],
        [6.4464e-02, 4.2871e-02, 0.0000e+00,  ..., 7.6874e-01, 9.6432e-01,
         4.6411e-01],
        ...,
        [7.0445e-01, 8.0986e-01, 7.6874e-01,  ..., 0.0000e+00, 7.8513e-01,
         4.9876e-01],
        [9.2271e-01, 9.9952e-01, 9.6432e-01,  ..., 7.8513e-01, 0.0000e+00,
         5.2540e-01],
        [4.1262e-01, 5.0418e-01, 4.6411e-01,  ..., 4.9876e-01, 5.2540e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.20436764,  8.49161154,  8.41171369,  8.30125201,  8.29203356,
        8.62162372, 10.01804223,  9.86089579,  7.61088125,  9.36378438,
       10.38897563, 11.22993218,  6.61211224,  6.46560181,  5.9510489 ,
       13.744387  ,  6.39520397,  6.19707944,  6.80370254,  8.8737556 ,
        8.34949383, 11.97957425,  4.85995002,  5.81481044,  8.16584537,
       13.99048094,  6.1463896 ,  9.62238828,  5.39723241, 11.64746882,
        7.52117132, 13.77075174,  5.80860165,  7.53063885, 13.09426142,
        9.75401449,  6.67121748, 13.32645552,  5.9510489 , 10.27716434,
       13.89886469, 14.02532758, 11.00508681,  7.58378655, 11.0050379 ,
       11.00694688,  9.10945678, 14.09176259, 10.68031351,  9.36378438,
       10.00410625, 11.33837396, 13.83139024, 10.60226614, 11.63765273,
       10.26545792, 13.13888841,  9.79814997,  5.39450305,  9.45995955,
       13.13205357,  6.67108662,  6.53290663,  7.59795445,  6.30647746,
        9.70108409, 12.71843462,  5.04028538,  9.18690018, 11.97957425,
        7.5403822 , 13.1399181 ,  6.97106912,  9.14973319,  6.99097013,
        6.33075682,  7.15100303,  4.96231532,  7.5092834 ,  4.85995002,
       12.58006038,  9.36378438,  5.05963217,  9.3109884 ,  6.33075682,
        6.59176018,  8.57560275,  5.79562691, 11.70168755,  9.10945678,
        7.8575648 ,  6.61211224,  6.61211224,  4.91038884, 13.13205357,
        9.97220799, 13.85158417,  5.18047638, 11.56146873, 11.70016217,
        7.66377554, 11.73567759,  7.9337647 , 13.15525803, 13.12218013,
        5.27178363])}
{'P': tensor([[2.9347e-02, 2.5390e-01, 1.8072e-01,  ..., 7.1819e-06, 3.0846e-02,
         7.0090e-05],
        [2.6449e-03, 3.8168e-02, 2.1326e-01,  ..., 7.0881e-06, 3.7776e-02,
         8.7054e-05],
        [4.8895e-04, 6.5048e-04, 5.7200e-02,  ..., 3.8853e-06, 2.1559e-01,
         4.9171e-04],
        ...,
        [8.6934e-04, 6.9074e-04, 5.7838e-04,  ..., 2.3629e-02, 3.3358e-04,
         9.6261e-06],
        [1.7781e-04, 2.0890e-04, 7.9243e-04,  ..., 2.2124e-06, 6.1727e-02,
         2.4900e-03],
        [1.0524e-05, 1.0010e-05, 2.3185e-05,  ..., 2.5825e-06, 7.7260e-05,
         1.6845e-01]]), 'X': tensor([[ -3.0004,   7.6607,  -7.2640],
        [ -3.4100,   7.5436,  -7.6980],
        [ -5.9987,   6.1949,  -6.2114],
        [ -5.2589,   6.7024,  -5.5809],
        [ -4.4520,   7.1370,  -8.7867],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.2905,   1.9366,   2.1169],
        [  5.6414, -10.0303,   7.8525],
        [  5.3924, -10.1396,  -4.0863],
        [  2.6970,   6.9826,   9.7350],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -0.8675, -10.8815,   6.8561],
        [  5.8836,   3.4315,  -4.0327],
        [ 12.5832,   1.7543,  -7.6134],
        [ -7.8359,   4.2735,   7.2943],
        [  0.6470, -11.0352,  -1.1931],
        [  4.0642,  -3.7599,  -3.4565],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.8889,  -1.3916,   8.8993],
        [  6.1755,   2.5233,  -2.0042],
        [  5.3858,   4.4260,  -9.6769],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.9410,   6.8428,   5.4458],
        [  3.1756, -10.8247,   6.8882],
        [ -9.3616,  -2.4750,   3.7765],
        [ -8.0534,  -6.0429,   2.4583],
        [  4.0457,  -3.7737,  -0.4704],
        [  7.7750,  11.0665,   2.8364],
        [ 12.5650,   2.0132,   4.6272],
        [  7.5930,  -8.9037,   7.2323],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.1625,  -4.6697,   7.0658],
        [  6.3516,   0.6425,   0.3425],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.8111, -10.6787,  -4.6539],
        [ -3.2736, -10.1703,   3.8712],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.2883,   0.0325,   5.2692],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.1755,   2.5233,  -2.0042],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.0190,   4.9705,  -7.9752],
        [  2.1771, -10.9776,  -3.2007],
        [  5.5424,  -2.0885,   7.2487],
        [  4.5491,  -3.3485,   6.8412],
        [  0.1047, -11.0046,   0.5188],
        [ -9.1749,  -3.3803,   6.4629],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.0047,   3.1062,  -6.7914],
        [ -8.9282,   2.1648,   8.4894],
        [  5.8201,   3.5841,  -8.0609],
        [  2.1120,  -4.6818,   8.6144],
        [  5.3602,  -2.3808,   7.2098],
        [  5.6248,   3.9984,   8.4933],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.5814, -10.4485,   2.5496],
        [ -8.3062,  -5.6008,   0.0287],
        [ -7.5603,   4.6459,   8.2002],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.0348,  -2.8229,   3.5436],
        [  2.2373,  -4.6506,  -1.7334],
        [ 10.9931,   7.2107,   7.7933],
        [ 12.5763,   0.1333,  -9.2650],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 12.0042,  -2.8350,   2.6864],
        [ -3.3999, -10.1152,  -1.2088],
        [  9.3753,   9.5171,   3.9131],
        [  6.3549,   0.7096,   6.9198],
        [ -1.7290,   7.8825,  -3.3831],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 10.5057,  -5.9099,  -5.1599],
        [  3.3633,  13.4794,   0.5193],
        [ -8.4573,   3.2448,   5.4370],
        [  9.2692,   9.6370,   7.0746],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.5568,   6.0602,   9.3244],
        [  5.3804,  -2.3504,   7.2744],
        [  2.0937,  13.8293,  -2.3533],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 11.4235,   6.3594,   5.1663],
        [  4.1637,  13.1898,  -1.9189],
        [  5.6022,  -1.9833,   8.3081],
        [  0.5850,  -4.7982,   9.6639],
        [  5.5101,  -2.1433,   9.5442],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 10.9931,   7.2107,   7.7933],
        [ -8.3539,  -5.5111,   0.3373],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.2508,   1.0572,   2.9978],
        [  0.2732,  -4.7619,  -1.1577],
        [ -7.8359,   4.2735,   7.2943],
        [  5.8889,  -1.3916,   8.8993],
        [ 10.0567,   8.6697,   4.6329],
        [  5.6093,  -1.9705,   6.4807],
        [ 10.1798,   8.5002,  -5.8893],
        [  5.9003,  12.3550,   2.1917],
        [ -7.9298,   4.1367,  -3.9773],
        [ -8.9942,  -3.9990,   6.0448]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0355, 0.1743,  ..., 0.9613, 0.3599, 1.0202],
        [0.0355, 0.0000, 0.1483,  ..., 0.9954, 0.3393, 1.0203],
        [0.1743, 0.1483, 0.0000,  ..., 1.0712, 0.1946, 0.9190],
        ...,
        [0.9613, 0.9954, 1.0712,  ..., 0.0000, 1.1453, 1.2044],
        [0.3599, 0.3393, 0.1946,  ..., 1.1453, 0.0000, 0.7646],
        [1.0202, 1.0203, 0.9190,  ..., 1.2044, 0.7646, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.22728399,  8.27853076,  8.62329466,  8.51928531,  8.41171369,
        8.49161154,  6.58184474, 11.50791859, 11.48434767,  7.48538905,
       10.54658597, 10.91601655,  6.81117802, 12.70489563,  8.92549752,
       11.05413545,  5.5366698 , 11.80092578,  6.05113217,  6.67108662,
        6.97106912,  5.79391825,  7.44805901, 11.28093507,  9.68324553,
       10.06850677,  5.53255882, 13.5247215 , 12.72524615, 11.70168755,
       12.38881635,  5.14607014,  6.38399789,  9.97964013,  5.5366698 ,
       11.33837396, 10.68416401,  8.91551231,  6.28835101,  9.77777346,
        9.86089579,  6.67108662,  9.17905491,  7.06372172, 11.1913598 ,
        5.92281142,  5.64865173, 11.00508681,  9.77777346,  4.978704  ,
        6.76056625,  9.18690018,  6.83515201,  5.13616224,  5.86519525,
        6.90115901, 12.17901033, 11.4088043 , 10.01804223,  8.8737556 ,
        8.85572292, 11.48434767,  9.50654967, 10.53368644,  5.77219077,
        5.16077764, 13.14690137, 12.5769659 ,  9.69629   , 12.3344536 ,
       10.67132524, 13.35927527,  6.39438044,  8.06990483,  9.51074763,
       12.05395056, 13.89264822,  9.05843201, 13.37121923, 13.20597932,
       13.0493368 ,  5.87132718, 13.98691426, 10.53368644,  5.83560009,
        9.24501829,  9.43737523, 13.0743399 , 13.83139024,  5.94293185,
        4.83371341,  5.91222346,  9.88571713, 13.14690137, 10.00794468,
        5.01114719,  9.3109884 ,  4.76969364,  8.92549752,  6.05113217,
       13.2778378 ,  5.94535529, 13.26209545, 13.69162795,  8.94395628,
        9.84314712])}
{'P': tensor([[9.6559e-03, 2.3699e-01, 1.5064e-01,  ..., 1.5461e-06, 3.4747e-05,
         1.2611e-04],
        [4.3487e-04, 1.2635e-02, 1.8690e-01,  ..., 1.8757e-06, 4.2149e-05,
         1.4678e-04],
        [2.9717e-04, 3.5769e-04, 1.3778e-02,  ..., 1.2226e-06, 3.2358e-05,
         9.4201e-05],
        ...,
        [2.6024e-03, 2.5773e-03, 1.7275e-03,  ..., 9.6017e-02, 2.2752e-04,
         3.0077e-04],
        [2.7764e-02, 2.5727e-02, 2.2010e-02,  ..., 5.9796e-05, 4.8447e-02,
         2.3125e-01],
        [2.8782e-02, 2.8419e-02, 3.8007e-02,  ..., 1.3734e-05, 8.2622e-04,
         6.0042e-02]]), 'X': tensor([[ -3.0644,   7.6439,  -1.4322],
        [ -3.2486,   7.5925,  -1.3719],
        [ -4.2271,   7.2387,  -1.8316],
        [ -4.9217,   6.8979,  -1.7667],
        [ -5.3736,   6.6310,  -1.5174],
        [ -5.6174,   6.4708,  -1.7339],
        [ 11.5935,   5.9740,  -4.3952],
        [ -8.9942,  -3.9990,   6.0448],
        [  4.4633,   5.6287,  -0.7754],
        [ -8.6229,   2.9068,   2.4429],
        [  1.9503,  -4.7172,  -6.2313],
        [  1.6232,  -4.7717,  -2.5655],
        [  5.4179,  -2.2925,  -3.4641],
        [ -4.0868,   7.2981,   8.1874],
        [  0.1382, -11.0073,   1.0026],
        [  6.0073,   3.0986,   6.7937],
        [  6.3200,   0.2690,   1.5900],
        [  9.6614,   9.1784,  -1.2087],
        [  2.5375,   7.0676,  -4.3490],
        [  4.5057,  -3.3894,  -0.8160],
        [  3.9770, -10.6341,   8.0783],
        [ 12.1379,   4.4125,  -3.8833],
        [ 12.2328,  -2.0376,  -1.1234],
        [  8.2154,  -8.4262,   8.0171],
        [ 10.2712,   8.3707,  -4.9743],
        [ -7.2515,   5.0195,   4.6434],
        [  2.5813,  -4.5471,   5.3597],
        [  0.2435,   7.8191,  -6.4709],
        [  6.1755,   2.5233,  -2.0042],
        [ -2.5056,   7.7720,   5.5878],
        [  6.2770,  -0.0379,   2.4195],
        [  2.3261,   7.1726,   6.2522],
        [  1.6954,  -4.7616,   9.8198],
        [  6.3122,  -9.6994,  -6.3282],
        [ -5.3382,   6.6533,   7.8054],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.1477,   7.2550,   8.9688],
        [ -1.0012, -10.8576,   2.2867],
        [  1.3767, -11.0341,   9.9415],
        [ -1.3178, -10.7939,   0.4236],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.6035,   0.7494,   7.1536],
        [  5.8617,   3.4853,  -9.8080],
        [  6.0073,   3.0986,   6.7937],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3549,   0.7096,   6.9198],
        [  5.9655,   3.2172,  -7.5953],
        [  5.2199,   4.6861,  -1.9304],
        [  7.0171,  11.6439,  -5.0178],
        [  6.1755,   2.5233,  -2.0042],
        [ -4.2271,   7.2387,  -1.8316],
        [  0.2732,  -4.7619,  -1.1577],
        [  1.3403,  -4.8011,  -1.1698],
        [ -0.0254, -10.9932,   3.2200],
        [ -7.4807,   4.7463,   2.0616],
        [ -0.0410,   7.8591,   9.3457],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.9204,   7.3518,   8.4781],
        [ -1.8823, -10.6558,  -3.8952],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.3211,   1.7017,  -0.2164],
        [ 10.6283,  -5.7253,   7.4878],
        [ 12.1588,   4.3364,  -8.0867],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.8902,  -1.3887,   4.0388],
        [ -6.0175,  -8.4541,  -9.8408],
        [  0.5175,  14.0910,   4.6540],
        [ 12.2589,  -1.9312,   9.8657],
        [  2.5974,   7.0363,  -7.7810],
        [  4.8653,   5.1694,  -4.0823],
        [ -3.0004,   7.6607,  -7.2640],
        [ -1.5261,   7.8988,   9.3768],
        [  3.2809,  13.5061,  -4.0199],
        [ -9.4524,  -1.6813,  -6.6862],
        [  1.9503,  -4.7172,  -6.2313],
        [ 12.4592,  -0.9168,  -6.7571],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.1196,  -2.7160,   4.2705],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.9204,   7.3518,   8.4781],
        [ -8.0965,  -5.9712,   4.3466],
        [ -7.0825,   5.2073,  -5.0497],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.3107,  -2.7697,  -3.2919],
        [  7.0642,  11.6104,  -9.8316],
        [  1.8320,   7.3871,   5.1452],
        [ 12.4690,  -0.8528,  -9.3374],
        [  2.5620,   7.0549,   8.4825],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.3539,  -5.5111,   0.3373],
        [ 11.0891,  -4.9558,  -2.3178],
        [  9.9217,  -6.7006,  -5.6728],
        [  4.9152,  12.8653,  -1.4459],
        [  0.9036,   7.6840,   9.9237],
        [ -1.6334, -10.7207,   7.1397],
        [  5.3924, -10.1396,  -4.0863],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.3616,  -2.4749,   0.1793],
        [ -8.8835,  -4.3178,   9.3080],
        [  7.0515,  11.6194,   8.4289],
        [  3.0876,   6.7527,  -5.3540],
        [  0.2435,   7.8191,  -6.4709]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0186, 0.0671,  ..., 1.1508, 0.3196, 0.3093],
        [0.0186, 0.0000, 0.0530,  ..., 1.1597, 0.3348, 0.3190],
        [0.0671, 0.0530, 0.0000,  ..., 1.2111, 0.3475, 0.3108],
        ...,
        [1.1508, 1.1597, 1.2111,  ..., 0.0000, 1.2339, 1.3392],
        [0.3196, 0.3348, 0.3475,  ..., 1.2339, 0.0000, 0.1318],
        [0.3093, 0.3190, 0.3108,  ..., 1.3392, 0.1318, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.23525433,  8.25828409,  8.38252937,  8.47371868,  8.53501777,
        8.56889738, 13.04217487,  9.84314712,  7.18354885,  9.09963894,
        5.10444097,  5.04028538,  5.8828986 ,  8.36445758, 11.00812996,
        6.7593917 ,  6.32572323, 13.32613445,  7.5092834 ,  5.63824971,
       11.35345345, 12.91505111, 12.4013153 , 11.76830981, 13.25017211,
        8.81930525,  5.2287058 ,  7.822854  ,  6.67108662,  8.16584537,
        6.27715105,  7.5403822 ,  5.05444288, 11.57249984,  8.53014915,
        5.19065439,  7.56617037, 10.9036226 , 11.11969815, 10.87409005,
       10.33712258,  8.51928531, 12.62576026,  6.81961996,  6.7593917 ,
        5.5366698 ,  6.39438044,  6.77777606,  7.01475376, 13.59480906,
        6.67108662,  8.38252937,  4.76969364,  4.98461983, 10.99326577,
        8.85939626,  7.85919687, 12.32408516,  7.5984809 , 10.8207297 ,
        9.85330646,  9.71392179,  8.36995953,  6.54615887, 12.07225469,
       12.9089559 ,  9.69792513,  8.92395381,  6.05165248, 10.37698435,
       14.10045797, 12.41012046,  7.50035355,  7.09887557,  8.22728399,
        8.04483495, 13.89886469,  9.60080639,  5.10444097, 12.49291768,
        5.24137909,  5.79543038, 13.20988996,  7.5984809 , 10.06023665,
        8.7907923 , 10.67322527,  9.71392179, 13.59056613,  7.61088125,
       12.4980845 ,  7.50564223,  8.83725081, 10.00794468, 12.1460931 ,
       11.97238624, 13.77222752,  7.73692436, 10.84437705, 11.48434767,
       10.74655346,  9.68323686,  9.87720102, 13.59170809,  7.42512621,
        7.822854  ])}
{'P': tensor([[2.8058e-02, 2.3550e-01, 1.9084e-01,  ..., 3.1644e-05, 2.7814e-06,
         1.7198e-03],
        [2.5180e-04, 3.0290e-02, 5.3601e-01,  ..., 3.0665e-05, 1.0914e-06,
         7.9344e-03],
        [8.9094e-04, 4.6495e-03, 1.2595e-01,  ..., 1.2600e-04, 4.2023e-06,
         3.8332e-02],
        ...,
        [2.2947e-03, 7.9151e-04, 6.7926e-04,  ..., 1.2745e-01, 1.0097e-03,
         8.5096e-05],
        [1.6205e-03, 5.3517e-04, 4.8167e-04,  ..., 1.3593e-05, 6.5339e-02,
         3.9925e-06],
        [3.6263e-05, 6.0544e-05, 6.5932e-05,  ..., 4.3032e-04, 1.0902e-06,
         1.1499e-01]]), 'X': tensor([[ -3.1671,   7.6158,   6.0765],
        [ -5.9940,   6.1985,   5.9566],
        [ -6.3306,   5.9272,   5.6935],
        [ -6.3489,   5.9117,   6.5161],
        [ -5.3382,   6.6533,   7.8054],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.4602,   4.7718,  -4.6277],
        [  2.6613,   7.0021,   4.5783],
        [  0.1047, -11.0046,   0.5188],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.0047,   3.1062,  -6.7914],
        [ -6.2200,  -8.2737,   5.5079],
        [ -1.0407,   7.9166,  -6.6021],
        [  3.6817,   6.3354,   6.1233],
        [ -8.3653,   3.4184,   9.3454],
        [  5.5626,  -2.0536,   8.4433],
        [  6.0031,  12.2958,  -6.5992],
        [ 11.9122,   5.1397,  -1.5498],
        [ -1.5451,   7.8975,   7.6104],
        [ -4.2040,   7.2486,   2.9875],
        [ 11.0438,   7.1180,  -5.6991],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.8363,   2.4086,  -5.6296],
        [  9.9756,  -6.6330,  -3.3429],
        [  1.5246,  -4.7838,  -5.2095],
        [ -4.7334,   6.9982,  -0.4392],
        [  4.4915,   5.5990,   7.0405],
        [  6.3497,   1.3680,  -1.9076],
        [ 10.9394,   7.3069,  -8.9365],
        [ 12.5154,   2.5072,  -9.3274],
        [  3.0719, -10.8449,  -4.1673],
        [  6.1752,   2.5242,   3.5874],
        [ -8.6229,   2.9068,   2.4429],
        [ -6.8466,  -7.6488,   1.9648],
        [  4.4915,   5.5990,   7.0405],
        [  6.3413,   0.4883,  -6.8416],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.3118,   4.5452,   4.2114],
        [ -9.4505,  -1.7081,  -0.7039],
        [  4.8051, -10.3703,   0.5268],
        [  2.5469,  13.7192,  -9.6935],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.3589,   0.8239,  -8.4230],
        [  4.5430,   5.5438,  -3.8669],
        [  0.5850,  -4.7982,   9.6639],
        [  9.0129,  -7.7072,   9.1400],
        [  6.3119,   1.7797,   9.9186],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 12.5154,   2.5072,  -9.3274],
        [  4.5340,  -3.3629,   8.9652],
        [ 12.5763,   0.1333,  -9.2650],
        [  9.3836,   9.5075,  -5.9278],
        [ 10.1798,   8.5002,  -5.8893],
        [  6.2975,   0.0947,   5.1791],
        [  2.1379,  -4.6757,   6.4981],
        [  3.3744,   6.5623,  -6.3636],
        [  1.9312,  13.8649,   8.3563],
        [  6.1755,   2.5233,  -2.0042],
        [  2.3889,  -4.6082,  -8.7291],
        [  9.9217,  -6.7006,  -5.6728],
        [ -5.7347,  -8.6908,   9.2878],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.1196,  -2.7160,   4.2705],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.7122,  -4.7591,  -3.6734],
        [  8.7715,  -7.9390,   5.2900],
        [  2.7976,  -4.4682,  -4.2020],
        [  7.0358,  -9.2784,  -9.8479],
        [  6.0047,   3.1062,  -6.7914],
        [  6.2036,   2.4030,  -1.2500],
        [  5.1861,   4.7361,  -3.7312],
        [  8.0381,  -8.5690,  -8.9523],
        [  3.5284, -10.7484,   5.3347],
        [  6.3211,   1.7017,  -0.2164],
        [ -8.6050,   2.9451,  -0.7012],
        [ 12.2218,   4.0953,  -8.2072],
        [ -0.1963,   7.8765,  -1.1371],
        [  3.3744,   6.5623,  -6.3636],
        [  0.6952,  -4.8060,   8.8976],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.6258,  -1.9404,  -6.0715],
        [ 12.4690,  -0.8528,  -9.3374],
        [  6.3505,   0.6235,  -8.5849],
        [  2.6321,  -4.5296,  -2.5159],
        [  3.2647, -10.8066,  -0.3768],
        [ -9.4541,  -0.3327,   8.8246],
        [ 10.1503,   8.5414,  -7.2681],
        [  2.6970,   6.9826,   9.7350],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.5010,  -4.5736,  -0.8989],
        [ -2.1382,   7.8339,   7.3524],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.2790,  -0.0259,  -5.4116],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.8323,  -1.5228,   5.1132],
        [  6.3413,   0.4883,  -6.8416],
        [  3.3633,  13.4794,   0.5193],
        [  6.1737,   2.5305,   3.9049],
        [  5.4336,  -2.2676,   5.6933],
        [  6.0054,  -1.0902,   8.3079],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.1231,  -3.5739,   8.4708]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.3739e-01, 1.7637e-01,  ..., 4.0655e-01, 9.5078e-01,
         5.5753e-01],
        [1.3739e-01, 0.0000e+00, 4.1893e-02,  ..., 5.4394e-01, 1.0348e+00,
         4.2938e-01],
        [1.7637e-01, 4.1893e-02, 0.0000e+00,  ..., 5.8243e-01, 1.0474e+00,
         4.0297e-01],
        ...,
        [4.0655e-01, 5.4394e-01, 5.8243e-01,  ..., 2.1453e-04, 7.9787e-01,
         9.5131e-01],
        [9.5078e-01, 1.0348e+00, 1.0474e+00,  ..., 7.9787e-01, 0.0000e+00,
         1.4356e+00],
        [5.5753e-01, 4.2938e-01, 4.0297e-01,  ..., 9.5131e-01, 1.4356e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.24808832,  8.62260327,  8.67227386,  8.67502348,  8.53014915,
        8.68979052,  8.85572292,  7.49076899, 11.00508681, 10.33712258,
        6.76056625, 10.35093485,  7.98469166,  7.32754882,  9.0368458 ,
        5.92952538, 13.6829843 , 12.97370371,  8.04718433,  8.37954853,
       13.13888841, 12.24467041,  5.14999799,  9.15866607, 11.97957425,
        5.0209005 ,  8.4486771 ,  7.17789666,  6.49538841, 13.15525803,
       12.76408562, 11.27160035,  6.67121748,  9.09963894, 10.26545792,
        7.17789666,  6.36004393,  5.90035324,  6.99097013,  9.60359158,
       11.42947632, 13.95361098, 13.09737053,  6.41202986,  7.16747878,
        4.83371341, 11.85890664,  6.55800267, 10.48840476,  9.88571713,
       12.76408562,  5.64500512, 12.5769659 , 13.35832725, 13.26209545,
        6.29821673,  5.14124612,  7.37902766, 13.99877413,  6.67108662,
        5.19065439, 11.97238624, 10.412299  , 12.13568598,  5.79543038,
       11.86239961,  5.19577146,  5.05774295, 11.83075005,  5.27178363,
       11.64436684,  6.76056625,  6.652742  ,  7.02326515, 11.74901778,
       11.31276109,  6.54615887,  9.09502385, 12.889695  ,  7.8788965 ,
        7.37902766,  4.85603886,  8.56889738,  5.9510489 , 12.4980845 ,
        6.38105015,  5.23878371, 11.28895489,  9.45995955, 13.26590071,
        7.48538905, 12.3489324 ,  5.21280156,  8.12043983,  6.12886024,
        5.27178363,  6.27905292, 12.49317563,  6.02779201,  6.36004393,
       13.89264822,  6.67218687,  5.88782913,  6.10360073, 13.38091121,
        9.79814997])}
{'P': tensor([[1.1428e-02, 1.6905e-01, 1.5608e-01,  ..., 3.3011e-04, 1.3122e-06,
         6.2788e-04],
        [2.9704e-04, 1.1442e-02, 2.7465e-01,  ..., 5.0774e-04, 1.4766e-06,
         6.7069e-04],
        [2.1838e-04, 3.0767e-04, 1.4192e-02,  ..., 9.0257e-04, 2.1965e-06,
         1.0018e-03],
        ...,
        [7.3776e-06, 7.8966e-06, 8.6041e-06,  ..., 3.1990e-02, 9.9124e-04,
         5.1954e-05],
        [3.5578e-06, 2.5513e-06, 2.1879e-06,  ..., 2.7123e-06, 3.6662e-02,
         2.2882e-06],
        [1.6921e-05, 2.0847e-05, 2.6659e-05,  ..., 1.0245e-02, 2.2037e-04,
         3.2809e-02]]), 'X': tensor([[ -3.2486,   7.5925,  -1.3719],
        [ -4.2271,   7.2387,  -1.8316],
        [ -4.9217,   6.8979,  -1.7667],
        [ -5.3736,   6.6310,  -1.5174],
        [ -5.6423,   6.4537,  -1.2091],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.9470,  13.6085,   2.9809],
        [ -3.2736, -10.1703,   3.8712],
        [ -5.9872,   6.2037,  -9.4477],
        [ -0.6323,   7.9085,  -5.9544],
        [ -6.0844,  -8.3955,   0.7705],
        [ -3.0644,   7.6439,  -1.4322],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -4.0868,   7.2981,   8.1874],
        [  2.3367,   7.1676,  -9.5263],
        [ -7.4807,   4.7463,   2.0616],
        [ 12.5996,   1.3971,  -1.0694],
        [  0.1047, -11.0046,   0.5188],
        [ -6.4844,  -8.0230,   8.5209],
        [ -7.0120,   5.2825,   7.7762],
        [  2.2373,  -4.6506,  -1.7334],
        [ -2.0840,   7.8415,  -7.8657],
        [  8.4691,  10.4571,  -5.0650],
        [  6.2975,   0.0947,   5.1791],
        [  8.8474,  10.0868,  -8.8644],
        [  6.3505,   0.6235,  -8.5849],
        [  2.3373,  13.7721,  -0.2442],
        [  0.6757, -11.0361,  -1.3047],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.6165,  -3.8715,   6.4594],
        [  4.6231,   5.4556,   8.4973],
        [  1.8719,   7.3713,  -3.7465],
        [  8.7131,  10.2216,   7.7915],
        [ 11.7418,   5.6070,  -1.4192],
        [  9.3560,   9.5391,   9.4943],
        [ -7.4602,   4.7718,  -4.6277],
        [ -0.2586,   7.8825,  -9.9840],
        [ -9.4524,  -1.6813,  -6.6862],
        [  4.2275,  -3.6320,  -0.2714],
        [  3.2990,  -4.2415,  -7.4857],
        [  5.8318,   3.5567,  -4.7692],
        [  8.2154,  -8.4262,   8.0171],
        [ 11.7349,  -3.5838,   2.1644],
        [ -0.9715, -10.8630,   0.3555],
        [  5.1203,  -2.7151,   6.2110],
        [  1.6422,   7.4586,  -6.8056],
        [  7.5309,  11.2617,   7.2656],
        [ -8.6603,   2.8250,   5.9181],
        [ -0.1963,   7.8765,  -1.1371],
        [ -2.2508, -10.5479,  -2.7235],
        [  3.6249, -10.7255,   5.6605],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.3261,   7.1726,   6.2522],
        [ -4.2271,   7.2387,  -1.8316],
        [ -4.7389,   6.9953,   3.7638],
        [ -7.8359,   4.2735,   7.2943],
        [  9.0004,  -7.7195,   4.6945],
        [  6.2365,  -0.2578,  -6.0212],
        [  3.9085,   6.1512,  -2.2560],
        [  9.7328,   9.0902,  -0.8917],
        [  6.0310,  12.2795,  -6.4528],
        [  3.2226,  -4.2802,  -4.0428],
        [ -9.4541,  -0.3327,   8.8246],
        [ -6.0195,   6.1789,  -2.9496],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.8201,   3.5841,  -8.0609],
        [  0.1041, -11.0045,   2.4311],
        [ 11.5198,  -4.0923,   5.5088],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.9392,  -4.8143,  -3.4649],
        [  1.4749,  -4.7892,  -0.8089],
        [  1.6422,   7.4586,  -6.8056],
        [  2.4572,   7.1085,   7.7376],
        [  6.3569,   0.7591,   6.5011],
        [ -4.0868,   7.2981,   8.1874],
        [  1.7000,  -4.7609,   2.7935],
        [ -3.2486,   7.5925,  -1.3719],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.2770,  -0.0379,   2.4195],
        [  9.8082,  -6.8396,   4.9615],
        [  6.0560,  -0.9421,  -5.0334],
        [ -7.2986,  -7.1225,  -3.0814],
        [  0.1382, -11.0073,   1.0026],
        [ -6.4620,  -8.0449,   6.2719],
        [ 12.5650,   2.0132,   4.6272],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.5158,  -2.1337,   3.5246],
        [ -8.0750,  -6.0072,  -5.8267],
        [  7.1441,  11.5529,   2.3232],
        [  5.1203,  -2.7151,   6.2110],
        [  7.5064,  -8.9651,  -0.0994],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -8.0534,  -6.0429,   2.4583],
        [  7.9403,  -8.6455,  -8.7433],
        [ 12.3357,   3.6001,   0.7903],
        [  6.1575,  12.2046,   3.3257],
        [  0.6933,   7.7336,   2.1179],
        [ -9.1178,   1.5784,   2.1410],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.9379,  -4.1655,  -2.8114],
        [  0.6933,   7.7336,   2.1179],
        [ -8.3539,  -5.5111,   0.3373],
        [  3.9770, -10.6341,   8.0783],
        [ -9.3444,   0.5821,   6.9101]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0530, 0.0920,  ..., 0.6109, 0.7602, 0.7546],
        [0.0530, 0.0000, 0.0433,  ..., 0.5767, 0.7831, 0.7540],
        [0.0920, 0.0433, 0.0000,  ..., 0.5355, 0.7788, 0.7311],
        ...,
        [0.6109, 0.5767, 0.5355,  ..., 0.0000, 0.7567, 0.4800],
        [0.7602, 0.7831, 0.7788,  ..., 0.7567, 0.0000, 0.3476],
        [0.7546, 0.7540, 0.7311,  ..., 0.4800, 0.3476, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.25828409,  8.38252937,  8.47371868,  8.53501777,  8.5724042 ,
        8.56889738, 13.92390689, 10.68416401,  8.62162372,  7.9337647 ,
       10.36844813,  8.23525433, 12.12448912,  8.36445758,  7.53884104,
        8.85939626, 12.67680439, 11.00508681, 10.31584102,  8.77913428,
        5.16077764,  8.1137452 , 13.45642379,  6.29821673, 13.41713661,
        6.38105015, 13.96905756, 11.05672136,  7.83486943, 12.24467041,
        7.15100303,  7.60529892, 13.43127131, 13.01187484, 13.3614598 ,
        8.85572292,  7.88677538,  9.60080639,  5.57341623,  5.37343173,
        6.83083863, 11.76830981, 12.26997165, 10.90638066,  5.79562691,
        7.63726134, 13.54774355,  9.10945678,  7.8788965 , 10.78533838,
       11.32149259, 11.9120321 ,  7.5403822 ,  8.38252937,  8.44940661,
        8.92549752, 11.85743425,  6.24187044,  7.28793663, 13.31764181,
       13.68063098,  5.35774573,  9.45995955,  8.62631801, 12.17451685,
        6.83515201, 11.0050379 , 12.22503754, 13.00380624,  4.90505566,
        5.01114719,  7.63726134,  7.52117132,  6.4020404 ,  8.36445758,
        5.05534609,  8.25828409,  9.49511524,  6.27715105, 11.95743558,
        6.12886024, 10.1979669 , 11.00812996, 10.31886066, 12.72524615,
       10.38897563,  5.91408504, 10.06437598, 13.58333857,  5.79562691,
       11.69264243,  9.4997987 , 10.53368644, 10.06850677, 11.73848065,
       12.85032836, 13.66991027,  7.76457095,  9.25336371, 10.30972193,
        9.84314712,  9.86089579,  7.76457095, 10.00794468, 11.35345345,
        9.36256096])}
{'P': tensor([[2.7501e-02, 1.7238e-01, 1.5366e-01,  ..., 3.7570e-05, 1.4496e-04,
         2.1481e-05],
        [1.0315e-03, 2.4043e-02, 4.8953e-02,  ..., 4.1032e-05, 1.9861e-04,
         1.3214e-05],
        [5.7797e-04, 1.7655e-03, 5.0824e-02,  ..., 2.3657e-04, 8.4374e-04,
         3.1432e-05],
        ...,
        [2.7600e-06, 2.2108e-06, 2.4714e-06,  ..., 2.4203e-02, 2.0584e-04,
         1.6221e-04],
        [5.5841e-06, 5.3855e-06, 5.3332e-06,  ..., 2.4797e-02, 3.7194e-02,
         6.2144e-05],
        [3.1221e-03, 1.5927e-03, 1.6417e-03,  ..., 2.9681e-05, 2.1874e-05,
         1.0358e-01]]), 'X': tensor([[-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0686, 0.1483,  ..., 0.4221, 0.5275, 0.5510],
        [0.0686, 0.0000, 0.1289,  ..., 0.4027, 0.4703, 0.6195],
        [0.1483, 0.1289, 0.0000,  ..., 0.2765, 0.4041, 0.6422],
        ...,
        [0.4221, 0.4027, 0.2765,  ..., 0.0000, 0.3219, 0.8166],
        [0.5275, 0.4703, 0.4041,  ..., 0.3219, 0.0000, 1.0442],
        [0.5510, 0.6195, 0.6422,  ..., 0.8166, 1.0442, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.27853076,  8.41171369,  8.62329466,  8.49161154,  8.62162372,
        8.83341608, 10.61047596,  6.81084724, 12.44177641,  9.43112957,
       13.97430191,  8.92395381, 14.05687815,  9.58347456, 11.87412439,
        7.96181288, 10.88618114, 13.50484539,  4.978704  ,  5.05774295,
       13.61801526,  8.22728399,  8.74470218,  9.91668182,  8.43626114,
       14.09357547, 14.05687815,  9.99145096, 10.51340829,  5.94541803,
       11.83509372, 11.018084  ,  5.08372149, 11.67875855, 13.1471313 ,
       13.04217487,  5.5366698 ,  5.29009836, 10.95593729, 11.4890861 ,
        5.19577146,  8.05242591, 11.22596994,  5.05981914,  9.54154096,
        5.87132718,  9.36093562, 10.35485331,  6.10360073, 14.09176259,
        9.76896904,  5.51177501, 12.1494023 , 12.76408562,  9.68323686,
        9.94173708,  7.17789666, 13.6105595 ,  5.30158497, 13.36092624,
        5.87305005, 11.63724792, 13.83139024,  6.56833864,  4.89121805,
       10.39942497, 11.72802491, 14.10045797,  9.93553424,  6.27715105,
        7.13501264, 11.76830981, 10.91805683,  4.78258092,  7.09887557,
       13.88683436, 12.52372629,  5.35774573, 10.7506423 , 10.32715715,
        8.62329466,  7.39599986, 10.92597196,  7.37576154, 10.07238463,
        7.76457095, 13.96746816, 11.32149259,  6.05165248,  8.62260327,
       12.12448912, 10.84437705, 12.59969936,  6.28961988,  7.53884104,
       12.76408562, 11.56146873, 10.48721382, 12.38881635, 13.14777433,
        5.81481044,  5.34250225, 11.18839169, 10.75443106, 10.46739073,
        6.652742  ])}
{'P': tensor([[2.4525e-02, 2.8343e-01, 1.7287e-01,  ..., 7.8486e-07, 1.1171e-04,
         1.1301e-05],
        [5.9851e-04, 3.4927e-02, 1.0281e-01,  ..., 1.8804e-06, 8.6963e-05,
         1.3285e-05],
        [5.7784e-04, 6.1315e-03, 3.0517e-02,  ..., 2.0085e-06, 7.0588e-05,
         1.3454e-05],
        ...,
        [4.8626e-07, 4.0492e-07, 4.4547e-07,  ..., 1.9555e-02, 2.5950e-06,
         7.1942e-06],
        [1.3542e-01, 6.2987e-02, 5.5326e-02,  ..., 1.3503e-06, 6.0671e-02,
         8.9255e-05],
        [6.7128e-05, 1.1358e-04, 1.5083e-04,  ..., 1.4866e-05, 2.6930e-05,
         7.1241e-02]]), 'X': tensor([[ -3.4986,   7.5152,   9.0712],
        [ -5.1369,   6.7755,   8.3836],
        [ -5.3382,   6.6533,   7.8054],
        [ -3.5520,   7.4976,   9.1522],
        [ -6.7992,   5.4991,   9.9186],
        [ -6.4464,   5.8272,   7.5038],
        [ 10.1798,   8.5002,  -5.8893],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.6414, -10.0303,   7.8525],
        [ -0.3432,   7.8900,   7.1805],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.3200,   7.1756,   1.1938],
        [  8.3791,  10.5409,   5.5590],
        [ -9.2345,  -3.1345,   3.9715],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.2555,   2.1474,   4.1685],
        [  5.4732,  -2.2041,  -4.3830],
        [  0.9392,  -4.8143,  -3.4649],
        [  0.9074,  -4.8139,   8.1674],
        [  7.4132,  11.3526,   0.3322],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.9636,   2.0648,   8.1946],
        [  8.3718,  -8.2953,   8.4666],
        [  1.0093,   7.6567,   2.3526],
        [ -7.6650,   4.5092,   0.2520],
        [ -5.9938,  -8.4745,   3.9735],
        [  6.1984,  -0.4327,   1.5438],
        [  1.6308,  13.9254,  -2.0984],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 10.0567,   8.6697,   4.6329],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.0305,   3.0296,   0.1375],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.1929,   2.4501,  -6.9352],
        [ -8.0611,  -6.0302,   4.6962],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -2.6485, -10.4151,   0.8766],
        [  5.5424,  -2.0885,   7.2487],
        [  6.0788,  -0.8710,  -6.8719],
        [ 10.6283,  -5.7253,   7.4878],
        [  3.6469,   6.3624,  -1.8414],
        [ 11.7795,  -3.4697,   3.5624],
        [ -1.3940, -10.7772,   6.7835],
        [ 12.4592,  -0.9168,  -6.7571],
        [  0.4119,  14.1019,  -4.1870],
        [  9.4925,   9.3811,   8.9842],
        [  3.2990,  -4.2415,  -7.4857],
        [  5.1472,  -2.6801,   8.1194],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.7565,  11.8237,  -4.9882],
        [  1.7000,  -4.7609,   2.7935],
        [ -9.2373,  -3.1225,  -8.9763],
        [  1.6408,   7.4591,  -4.2294],
        [  2.2437,  -4.6489,   6.1430],
        [ -9.1813,  -3.3550,   1.8827],
        [  1.7301,   7.4262,   5.0522],
        [  1.7815,  -4.7481,   4.4289],
        [ -8.8400,   2.3991,   1.3457],
        [ 12.3759,  -1.3943,   6.4659],
        [  2.1771, -10.9776,  -3.2007],
        [  6.3301,   0.3638,  -7.5685],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.3246,   1.6696,  -0.1544],
        [ 10.0995,  -6.4739,   2.0974],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.0588,  -6.0341,  -2.3006],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.2419,  -0.2313,   4.2440],
        [  4.1999,  -3.6543,  -4.1998],
        [  0.5175,  14.0910,   4.6540],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.8058,   2.4850,  -1.7425],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.6334, -10.7207,   7.1397],
        [ -9.4697,  -0.6157,  -6.8472],
        [  1.3339,   7.5627,  -0.9034],
        [ -8.9178,   2.1936,   7.5388],
        [ 11.6634,   5.8050,   9.1590],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.0582,  -2.6640,   0.7132],
        [ -8.9033,   2.2329,  -8.8912],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 12.2592,   3.9425,   2.0383],
        [ 10.6673,  -5.6649,  -4.1646],
        [  3.0719, -10.8449,  -4.1673],
        [  5.7151,  -1.7700,   3.1542],
        [  0.9074,  -4.8139,   8.1674],
        [ -3.4100,   7.5436,  -7.6980],
        [  0.1252, -11.0062,   7.2564],
        [  1.3339,   7.5627,  -0.9034],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.2675,  12.1379,  -8.9720],
        [  4.1297,  -3.7098,  -3.9838],
        [  0.1041, -11.0045,   2.4311],
        [ -6.4368,   5.8356,   3.3014],
        [  1.9503,  -4.7172,  -6.2313],
        [ -4.4779,  -9.5609,  -2.4505],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.1287, -11.0065,   6.5757],
        [  4.4049,   5.6892,  -4.4394],
        [  3.9591, -10.6391,  -6.2557],
        [  1.9204,   7.3518,   8.4781],
        [ -1.7009,   7.8851,  -8.2437]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0636, 0.0670,  ..., 1.0455, 0.2095, 1.0272],
        [0.0636, 0.0000, 0.0160,  ..., 1.0377, 0.2722, 1.0280],
        [0.0670, 0.0160, 0.0000,  ..., 1.0217, 0.2718, 1.0121],
        ...,
        [1.0455, 1.0377, 1.0217,  ..., 0.0000, 1.0581, 0.1412],
        [0.2095, 0.2722, 0.2718,  ..., 1.0581, 0.0000, 1.0119],
        [1.0272, 1.0280, 1.0121,  ..., 0.1412, 1.0119, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.28968067,  8.50268322,  8.53014915,  8.29640613,  8.74470218,
        8.68979052, 13.26209545,  7.83534987,  9.78147542, 10.53368644,
       11.50791859,  7.89745107,  7.9337647 ,  7.54127425, 13.46553623,
        9.75200848, 13.27994304,  6.61386436,  5.90035324,  4.90505566,
        4.89869205, 13.55868684,  8.92098752,  9.19837792, 11.785566  ,
        7.72291946,  8.89301516, 10.37998078,  6.2134961 , 14.02058621,
        9.78286733, 13.2778378 , 12.41366754,  6.74872981, 13.00413829,
        6.65991877, 10.06703358,  9.75075273, 10.74655346,  5.92281142,
        6.14086582, 12.07225469,  7.33350642, 12.27991991, 10.86693708,
       12.49291768, 14.10796466, 13.34587027,  5.37343173,  5.80312949,
       12.16901615, 11.57249984, 13.61801526,  5.05534609,  9.75075273,
        7.6374603 ,  5.16205087,  9.7751189 ,  7.62509582,  5.0713364 ,
        9.15976746, 12.45418236, 11.1913598 ,  6.34059021, 12.64235359,
        6.54127431, 11.99633819,  8.19703512, 10.0674887 ,  5.01114719,
        6.24614905,  5.5671389 , 14.10045797,  9.85384623,  9.14973319,
       10.76154256, 10.84437705,  9.48969978,  7.67939546,  9.18358562,
       13.02818702,  4.92497514, 12.3489324 ,  9.17905491,  9.91668182,
       12.87752052, 12.07818871, 11.27160035,  5.98285319,  4.89869205,
        8.27853076, 11.00694688,  7.67939546, 10.46739073, 13.66051884,
        5.5513072 , 11.0050379 ,  8.68832398,  5.10444097, 10.5575601 ,
       11.89855267, 11.00726483,  7.19513806, 11.35182589,  7.5984809 ,
        8.06643487])}
{'P': tensor([[9.3405e-03, 2.3234e-01, 1.4000e-01,  ..., 7.5555e-08, 8.4509e-06,
         1.4331e-04],
        [3.2239e-04, 1.3539e-02, 2.6290e-01,  ..., 1.1578e-07, 1.1016e-05,
         2.4054e-04],
        [2.5900e-04, 3.3534e-04, 1.7710e-02,  ..., 1.6373e-07, 1.1860e-05,
         3.6699e-04],
        ...,
        [1.3758e-06, 1.1719e-06, 8.3497e-07,  ..., 6.9922e-02, 3.1019e-04,
         2.1278e-06],
        [7.7847e-03, 6.2476e-03, 3.9603e-03,  ..., 2.3195e-05, 1.8820e-01,
         2.3456e-04],
        [5.1812e-06, 5.9717e-06, 7.7363e-06,  ..., 2.5432e-05, 1.0969e-05,
         2.1812e-02]]), 'X': tensor([[ -3.5029,   7.5138,   8.1853],
        [ -4.0868,   7.2981,   8.1874],
        [ -5.1369,   6.7755,   8.3836],
        [ -5.3382,   6.6533,   7.8054],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.0120,   5.2825,   7.7762],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.1178,   1.5784,   2.1410],
        [ -7.5401,   4.6718,   9.0033],
        [  5.0506,   4.9277,  -9.5094],
        [ -5.8389,  -8.6056,  -3.5098],
        [  1.7815,  -4.7481,   4.4289],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.9636,   2.0648,   8.1946],
        [  1.3767, -11.0341,   9.9415],
        [  0.6933,   7.7336,   2.1179],
        [  9.7328,   9.0902,  -0.8917],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.1588,   4.3364,  -8.0867],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.4489,   7.5256,   4.4529],
        [  7.3724,  -9.0577,  -8.9843],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.1905,   3.7253,   4.1175],
        [  3.9663,   6.1018,  -5.4104],
        [  4.9337,  12.8566,  -3.6582],
        [  2.2437,  -4.6489,   6.1430],
        [  1.1644, -11.0394,   7.8511],
        [ 10.1503,   8.5414,  -7.2681],
        [ -7.7842,   4.3466,  -1.7548],
        [ -6.4620,  -8.0449,   6.2719],
        [  4.3459,  -3.5327,   0.3667],
        [  2.2860,   7.1916,  -6.1011],
        [  5.9026,   3.3837,  -0.6573],
        [ -1.8367,   7.8718,  -8.6523],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.2311,  -2.0445,   1.3545],
        [ -4.3963,   7.1629,   4.5263],
        [ -5.3736,   6.6310,  -1.5174],
        [  5.6202,  -1.9508,   0.7471],
        [ -3.1671,   7.6158,   6.0765],
        [  6.9693,  -9.3200,  -8.3822],
        [  6.3031,   1.8477,  -7.1706],
        [  2.0899,  -4.6870,  -7.4260],
        [ -7.5603,   4.6459,   8.2002],
        [ 11.7418,   5.6070,  -1.4192],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.9788,   2.0208,  -3.2867],
        [ 11.4235,   6.3594,   5.1663],
        [ -6.2668,   5.9808,   2.5614],
        [  7.7794,  11.0629,   8.9069],
        [  8.2154,  -8.4262,   8.0171],
        [  6.8093,  -9.4178,   8.4244],
        [ 12.1504,   4.3674,  -7.2998],
        [  0.3715,  -4.7756,   0.2312],
        [  2.2860,   7.1916,  -6.1011],
        [ 11.7795,  -3.4697,   3.5624],
        [ -8.6603,   2.8250,   5.9181],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.0042,  -2.8350,   2.6864],
        [  1.8320,   7.3871,   5.1452],
        [  7.0642,  11.6104,  -9.8316],
        [ -9.1706,   1.3863,   8.4032],
        [  0.3231,  -4.7691,  -1.6120],
        [ -8.0750,  -6.0072,  -5.8267],
        [  3.8111, -10.6787,  -4.6539],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -5.7347,  -8.6908,   9.2878],
        [  9.6587,   9.1817,  -0.8212],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.1929,   2.4501,  -6.9352],
        [ -5.1046,  -9.1610,   2.3418],
        [  3.4140,  -4.1803,  -8.8224],
        [  8.7715,  -7.9390,   5.2900],
        [  3.1064,  13.5607,   6.7605],
        [  1.9505,  -4.7171,   9.9868],
        [ -9.4773,  -1.0024,   5.2599],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -9.4714,  -1.3301,  -3.1998],
        [  9.8082,  -6.8396,   4.9615],
        [ 10.3251,   8.2928,  -3.0006],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.5258,  -0.4200,  -9.3493],
        [  1.5383, -11.0274,  -4.9666],
        [  2.4356,  -4.5942,  -5.0808],
        [  9.5203,   9.3483,   5.8781],
        [ -4.0868,   7.2981,   8.1874],
        [ -5.9938,  -8.4745,   3.9735],
        [ -7.4733,   4.7555,   6.2490],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.7349,  -3.5838,   2.1644],
        [  8.8091,  -7.9037,   0.3631],
        [  7.1697,  11.5343,  -0.7276],
        [  5.7485,  -9.9811,  -3.2355],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.3062,  -5.6008,   0.0287],
        [ 12.4996,   2.6335,  -6.3166],
        [ -2.5683, -10.4433,  -2.9182],
        [  4.5325,  -3.3643,   6.7699],
        [  2.8994,   6.8675,  -9.5987],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.2022,   2.4092,   3.0349],
        [ -9.4735,  -0.7274,  -2.6749]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0337, 0.0945,  ..., 1.1532, 0.4903, 0.9644],
        [0.0337, 0.0000, 0.0609,  ..., 1.1628, 0.5176, 0.9467],
        [0.0945, 0.0609, 0.0000,  ..., 1.1843, 0.5697, 0.9188],
        ...,
        [1.1532, 1.1628, 1.1843,  ..., 0.0000, 0.8194, 0.8833],
        [0.4903, 0.5176, 0.5697,  ..., 0.8194, 0.0000, 1.0482],
        [0.9644, 0.9467, 0.9188,  ..., 0.8833, 1.0482, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.29022197,  8.36445758,  8.50268322,  8.53014915,  8.68979052,
        8.77913428, 11.64746882,  9.25336371,  8.87007495,  7.05627461,
       10.39942497,  5.0713364 , 13.59480906,  9.19837792, 11.11969815,
        7.76457095, 13.31764181,  5.35774573,  9.97822473, 12.9089559 ,
        8.30125201,  7.66377554, 11.67875855,  8.37954853,  8.99791316,
        7.27759233, 13.77075174,  5.16205087, 11.10066126, 13.26590071,
        8.91551231, 10.31886066,  5.60064319,  7.54620849,  6.80370254,
        8.08320235,  4.83371341, 12.40074901,  8.40445872,  8.53501777,
        5.94909709,  8.24808832, 11.63765273,  6.56833864,  5.13182506,
        8.8737556 , 13.01187484, 10.33712258,  9.20340031, 13.0743399 ,
        8.66273171, 13.52430176, 11.76830981, 11.62158554, 12.91143491,
        4.79003361,  7.54620849, 12.27991991,  9.10945678, 11.33837396,
       12.3344536 ,  7.61088125, 13.59056613,  9.27474386,  4.78002665,
       10.06437598, 11.33837396,  9.66977802, 10.412299  , 13.32645552,
        4.8763135 ,  6.65991877, 10.48721382,  5.39723241, 11.83075005,
       13.91198191,  5.10446653,  9.53015376,  9.24501829,  9.78147542,
       12.05300943,  9.5642993 , 11.95743558, 13.2430377 , 12.00995898,
       12.53285633, 11.1341798 ,  5.19986797, 13.3426541 ,  8.36445758,
       10.37998078,  8.8580671 ,  7.02326515, 12.26997165, 11.83509372,
       13.58101794, 11.51811603,  7.78980299, 10.01804223, 12.77402351,
       10.75443106,  5.64464807,  7.4544866 , 11.72749923,  6.65368551,
        9.5014146 ])}
{'P': tensor([[1.1293e-02, 2.2755e-01, 2.0846e-01,  ..., 1.2895e-07, 5.5376e-05,
         5.5815e-06],
        [6.1440e-04, 2.9218e-02, 6.2262e-04,  ..., 3.9421e-07, 2.1434e-04,
         1.2224e-05],
        [6.8803e-04, 2.9223e-01, 1.3704e-02,  ..., 1.5981e-07, 6.9301e-05,
         6.7299e-06],
        ...,
        [1.1585e-06, 9.5306e-07, 1.1517e-06,  ..., 5.7114e-02, 1.0653e-06,
         9.0059e-04],
        [6.1180e-05, 5.8937e-05, 6.0959e-05,  ..., 7.0603e-04, 4.1643e-01,
         2.7828e-04],
        [7.7338e-04, 6.3873e-04, 7.7266e-04,  ..., 1.0297e-05, 7.8356e-06,
         5.4300e-02]]), 'X': tensor([[-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0638, 0.0164,  ..., 1.1546, 0.5021, 0.6996],
        [0.0638, 0.0000, 0.0474,  ..., 1.1543, 0.4395, 0.7342],
        [0.0164, 0.0474, 0.0000,  ..., 1.1544, 0.4859, 0.7082],
        ...,
        [1.1546, 1.1543, 1.1544,  ..., 0.0000, 1.3243, 0.6791],
        [0.5021, 0.4395, 0.4859,  ..., 1.3243, 0.0000, 1.0963],
        [0.6996, 0.7342, 0.7082,  ..., 0.6791, 1.0963, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.29203356,  8.41171369,  8.30125201,  8.49161154,  8.62162372,
        8.83341608,  7.5403822 ,  5.01114719,  8.09984231,  5.14371005,
       10.35464916,  8.41171369,  8.31625171,  5.79543038,  6.24614905,
        9.41970031,  7.02326515, 13.61600536, 13.38091121, 10.06286074,
        5.94535529,  9.79814997,  8.23525433,  7.50500064,  4.89121805,
       13.25336001, 12.12641969, 14.0393949 , 10.9036226 ,  5.91677188,
        7.09887557, 13.32645552, 13.09737053, 12.38881635,  8.19703512,
       12.30695098, 13.58101794, 14.07623266,  4.91336994, 13.14777433,
       12.91143491, 13.09426142,  5.62005629, 12.58006038,  9.41970031,
        9.91668182,  7.32754882,  7.24326992, 13.38091121,  5.29009836,
        8.04718433,  6.61211224, 13.32374056, 11.73397309, 12.7125079 ,
        8.8737556 ,  9.96391045, 13.96987806, 11.38774149,  7.8575648 ,
        9.17016857,  4.91038884,  9.48337063, 12.58006038,  5.05444288,
        6.83515201, 13.09737053, 13.40850328, 13.40850328, 11.9702762 ,
       13.35013374,  9.97822473, 10.26545792, 11.00812996,  9.79814997,
       11.73397309, 12.12641969, 10.35485331, 13.09737053,  6.32572323,
       11.70016217,  5.19986797,  5.20659892, 13.68672855,  6.18668463,
        7.50917551, 13.35013374,  7.67939546, 13.6660671 , 12.77402351,
       11.00726483, 12.25806559, 11.54071859,  9.55593374,  8.36445758,
       11.32149259, 11.35345345,  8.09842704,  7.02253235,  9.78147542,
       14.03505624, 11.83570132,  5.62005629, 11.62158554, 10.4977132 ,
        6.84796202])}
{'P': tensor([[2.6179e-02, 2.8622e-01, 1.7121e-01,  ..., 1.0302e-05, 3.0889e-04,
         2.1443e-06],
        [5.2633e-04, 2.8552e-02, 8.4045e-02,  ..., 1.4098e-05, 7.2565e-04,
         1.5754e-06],
        [5.4627e-04, 5.3671e-03, 2.6713e-02,  ..., 1.5191e-05, 7.4787e-04,
         1.3965e-06],
        ...,
        [1.3607e-04, 1.1343e-04, 1.2460e-04,  ..., 1.0449e-01, 2.7699e-05,
         8.5294e-04],
        [4.7155e-06, 5.0018e-06, 5.0563e-06,  ..., 1.5520e-04, 2.7352e-02,
         4.5941e-06],
        [2.6548e-03, 1.0815e-03, 9.4030e-04,  ..., 2.1591e-05, 2.2156e-06,
         8.5382e-02]]), 'X': tensor([[ -3.5520,   7.4976,   9.1522],
        [ -5.1369,   6.7755,   8.3836],
        [ -5.3382,   6.6533,   7.8054],
        [ -6.7992,   5.4991,   9.9186],
        [ -7.0948,   5.1940,   9.8171],
        [ -6.4464,   5.8272,   7.5038],
        [ 11.3367,  -4.4806,   0.4643],
        [ -5.6169,  -8.7844,   8.9057],
        [ -5.1046,  -9.1610,   2.3418],
        [  2.3589,  13.7668,  -3.6501],
        [ 10.2470,   8.4054,  -0.1177],
        [  0.3724,   7.7974,  -6.8722],
        [ -8.6675,  -4.8569,  -9.1769],
        [  4.8051, -10.3703,   0.5268],
        [  6.0890,  -0.8382,   2.9166],
        [ -3.2736, -10.1703,   3.8712],
        [  5.5626,  -2.0536,   8.4433],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.2770,  -0.0379,   2.4195],
        [ 12.2435,   4.0076,   8.1257],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.6426,  -4.9136,   9.4941],
        [  8.6420,  -8.0580,   6.5781],
        [  5.6523,   3.9443,  -2.9449],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.9956,  11.6590,  -7.6876],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.2659,  13.7894,   6.5436],
        [  1.7232,  -4.7574,   3.1270],
        [  5.6314,  -1.9302,  -9.6543],
        [  4.1348,   5.9514,  -0.8856],
        [ -9.2707,   0.9661,   4.4863],
        [  6.3048,   0.1469,  -9.1691],
        [  5.4732,  -2.2041,  -4.3830],
        [ 10.1798,   8.5002,  -5.8893],
        [ -7.4733,   4.7555,   6.2490],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.0233,   4.9647,  -3.1788],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.3118,   4.5452,   4.2114],
        [  5.0506,   4.9277,  -9.5094],
        [ 12.5763,   0.1333,  -9.2650],
        [  3.4401,  13.4540,   4.6576],
        [  2.3928,   7.1404,  -1.7035],
        [ 12.5789,   0.1722,   1.6470],
        [ -5.5525,   6.5146,  -4.4984],
        [  2.3589,  13.7668,  -3.6501],
        [  4.5430,   5.5438,  -3.8669],
        [  3.7121,   6.3117,   0.6738],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.1666,   3.7653,   3.6075],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 11.1469,   6.9237,   1.6160],
        [  5.4242, -10.1261,   3.9595],
        [  3.6469,   6.3624,  -1.8414],
        [  0.9074,  -4.8139,   8.1674],
        [  6.3460,   1.4252,   1.2291],
        [  5.9587,  12.3216,   9.1695],
        [ -8.0534,  -6.0429,   2.4583],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -3.5029,   7.5138,   8.1853],
        [ 11.3691,   6.4760,   1.0787],
        [ -2.6069, -10.4298,  -6.1995],
        [ 11.2517,  -4.6495,   2.9800],
        [ -7.3333,   4.9246,  -7.9212],
        [ -4.4520,   7.1370,  -8.7867],
        [  1.9503,  -4.7172,  -6.2313],
        [  2.1379,  -4.6757,   6.4981],
        [ 11.4495,   6.3026,   6.8257],
        [  6.3587,   1.1596,  -3.6629],
        [ 12.0042,  -2.8350,   2.6864],
        [  4.7482, -10.3907,   5.9149],
        [  4.4039,   5.6903,   8.7366],
        [  5.3804,  -2.3504,   7.2744],
        [ -9.3616,  -2.4749,   0.1793],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.3505,   0.6235,  -8.5849],
        [  5.3602,  -2.3808,   7.2098],
        [ -9.0574,   1.7801,   8.8986],
        [  4.9452, -10.3187,  -7.8621],
        [ -1.0482, -10.8487,  -5.9066],
        [  9.9217,  -6.7006,  -5.6728],
        [ -1.8823, -10.6558,  -3.8952],
        [  4.3377,   5.7570,   1.3712],
        [ -7.3470,  -7.0615,   5.0066],
        [ 12.3609,   3.4769,  -8.5762],
        [  4.4039,   5.6903,   8.7366],
        [  6.0345,   3.0175,  -1.4424],
        [  3.9085,   6.1512,  -2.2560],
        [  5.3804,  -2.3504,   7.2744],
        [  1.5383, -11.0274,  -4.9666],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.1625,  -4.6697,   7.0658],
        [  4.9452, -10.3187,  -7.8621],
        [  3.4604,  -4.1545,   1.8091],
        [  6.1929,   2.4501,  -6.9352],
        [  2.7590,  13.6621,   9.4496],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.3200,   0.2690,   1.5900],
        [  1.3670,  -4.7990,   3.2603],
        [  4.7544,  -3.1425,  -1.5315],
        [ -6.7745,  -7.7264,   8.4684],
        [ 11.6634,   5.8050,   9.1590]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0509, 0.0552,  ..., 0.7793, 0.5055, 0.7406],
        [0.0509, 0.0000, 0.0160,  ..., 0.8004, 0.4557, 0.7911],
        [0.0552, 0.0160, 0.0000,  ..., 0.7870, 0.4575, 0.7951],
        ...,
        [0.7793, 0.8004, 0.7870,  ..., 0.0000, 1.1417, 0.9254],
        [0.5055, 0.4557, 0.4575,  ..., 1.1417, 0.0000, 1.2297],
        [0.7406, 0.7911, 0.7951,  ..., 0.9254, 1.2297, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.29640613,  8.50268322,  8.53014915,  8.74470218,  8.79284466,
        8.68979052, 12.18998195, 10.42667093, 10.48721382, 13.96746816,
       13.25336001,  7.80625981,  9.93553424, 11.42947632,  6.1463896 ,
       10.68416401,  5.92952538, 10.91805683,  6.27715105, 12.88270018,
       10.06407839,  9.94173708, 11.8159254 ,  6.89245352, 12.45418236,
       13.59673406, 10.92597196, 13.97430191,  5.05989547,  5.95297459,
        7.24677664,  9.32094055,  6.30647746,  5.90035324, 13.26209545,
        8.8580671 ,  5.91854397,  9.562174  ,  7.06272256,  9.99211663,
        6.99097013,  7.05627461, 12.5769659 , 13.88683436,  7.53063885,
       12.58006038,  8.55982105, 13.96746816,  7.16747878,  7.32233733,
       13.94008948,  8.99276564, 10.35485331, 13.12218013, 11.48734978,
        7.33350642,  4.89869205,  6.50409372, 13.68672855, 10.06850677,
       10.46739073,  8.29022197, 13.08414689, 10.7506423 , 12.17451685,
        8.83341608,  8.41171369,  5.10444097,  5.14124612, 13.06957425,
        6.46356806, 12.3344536 , 11.42420508,  7.19534146,  5.87132718,
        9.68323686, 12.18998195,  6.38105015,  5.86519525,  9.23071698,
       11.4424816 , 10.8992524 , 11.97238624, 10.8207297 ,  7.20826219,
       10.19036081, 12.84056604,  7.19534146,  6.74686071,  7.28793663,
        5.87132718, 11.1341798 , 12.35625694, 12.0495521 ,  6.14086582,
        5.14607014, 11.4424816 ,  5.40689332,  6.65991877, 13.93790235,
       11.59512194,  6.32572323,  4.98989906,  5.69913817, 10.27572673,
       13.02818702])}
{'P': tensor([[1.7119e-02, 3.6506e-01, 2.6121e-01,  ..., 7.1683e-06, 3.2599e-06,
         3.1343e-06],
        [4.9917e-04, 2.3425e-02, 3.0572e-01,  ..., 9.8640e-06, 5.3783e-06,
         5.6940e-06],
        [4.3711e-04, 1.0186e-03, 2.6563e-02,  ..., 1.0072e-05, 6.7877e-06,
         6.4815e-06],
        ...,
        [3.6677e-04, 2.9892e-04, 2.2272e-04,  ..., 1.4409e-01, 3.0238e-04,
         4.2991e-05],
        [8.1741e-06, 6.3337e-06, 5.1124e-06,  ..., 3.5827e-03, 3.5706e-02,
         1.4625e-06],
        [2.7725e-07, 2.9315e-07, 2.6711e-07,  ..., 1.6361e-04, 1.4186e-04,
         2.2671e-02]]), 'X': tensor([[ -3.5903,   7.4847,  -8.9112],
        [ -4.4520,   7.1370,  -8.7867],
        [ -5.0550,   6.8231,  -9.2502],
        [ -5.9872,   6.2037,  -9.4477],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.8280,   4.2848,  -7.1955],
        [ -8.8835,  -4.3178,   9.3080],
        [  5.5158,  -2.1337,   3.5246],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.1755,   2.5233,  -2.0042],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.1795,  -3.3623,   8.5259],
        [ -0.3432,   7.8900,   7.1805],
        [ 12.0042,  -2.8350,   2.6864],
        [ -7.8280,   4.2848,  -7.1955],
        [ -8.0087,   4.0173,   9.1091],
        [ -7.5401,   4.6718,   9.0033],
        [  0.5960, -11.0335,   3.1144],
        [  2.6614,  -4.5192,  -9.3641],
        [ -8.7078,   2.7182,   1.8146],
        [  6.3303,   1.6145,  -8.4083],
        [  8.3718,  -8.2953,   8.4666],
        [  6.1973,  -0.4372,   6.5720],
        [  1.3339,   7.5627,  -0.9034],
        [ -4.1296,   7.2803,  -2.7194],
        [ -4.4503,   7.1378,   3.3751],
        [  6.0345,   3.0175,  -1.4424],
        [  5.9912,   3.1453,  -3.9989],
        [  3.9895,   6.0816,  -6.5235],
        [ -6.3306,   5.9272,   5.6935],
        [ -1.3178, -10.7939,   0.4236],
        [  0.8572,  14.0501,  -8.3364],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.6992,   5.3690,   3.0935],
        [  4.5057,  -3.3894,  -0.8160],
        [  3.6469,   6.3624,  -1.8414],
        [  6.3583,   1.1729,   8.7835],
        [  1.8766,   7.3694,   1.1887],
        [ -9.3616,  -2.4750,   3.7765],
        [  6.1579,  -0.5961,  -6.5595],
        [ -6.0844,  -8.3955,   0.7705],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.5424,  -2.0885,   7.2487],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.2480, -11.0153,  -3.0183],
        [ -3.5173,   7.5091,  -8.9585],
        [ -5.3382,   6.6533,   7.8054],
        [  8.5716,  10.3596,   6.8818],
        [  6.3301,   0.3638,  -7.5685],
        [ -8.1426,   3.8048,   9.3542],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.1531,  -5.8748,   2.6934],
        [ 11.1420,   6.9331,  -2.4779],
        [ -7.2986,  -7.1225,  -3.0814],
        [  6.1575,  12.2046,   3.3257],
        [  4.1537,   5.9340,  -6.0423],
        [ 12.1751,  -2.2592,  -5.7751],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.6681,  -4.8556,   9.2518],
        [  1.7122,  -4.7591,  -3.6734],
        [ 12.5927,   0.4199,  -9.4997],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.3505,   0.6235,  -8.5849],
        [  5.8323,  -1.5228,   5.1132],
        [ -7.6650,   4.5092,   0.2520],
        [ -7.0825,   5.2073,  -5.0497],
        [  5.6414, -10.0303,   7.8525],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 11.9936,  -2.8677,   6.0614],
        [ -3.1671,   7.6158,   6.0765],
        [  3.9085,   6.1512,  -2.2560],
        [ -5.1398,  -9.1367,   6.2074],
        [  4.9452, -10.3187,  -7.8621],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.3118,   4.5452,   4.2114],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.4691,   2.8525,  -5.5110],
        [ 10.9916,   7.2133,  -0.9179],
        [ -5.6169,  -8.7844,   8.9057],
        [  1.7232,  -4.7574,   3.1270],
        [ 12.2218,   4.0953,  -8.2072],
        [ -1.0012, -10.8576,   2.2867],
        [ -9.4541,  -0.3327,   8.8246],
        [ 11.5083,  -4.1177,   0.2389],
        [ -8.9624,  -4.0944,  -4.9622],
        [  3.5939,   6.4029,  -1.8357],
        [  3.6703,   6.3443,  -5.4401],
        [ 10.0995,  -6.4739,   2.0974],
        [ -3.0008, -10.2827,  -8.9480],
        [  0.6363, -11.0348,   6.1414],
        [  1.7815,  -4.7481,   4.4289],
        [ -5.9225,  -8.5355,   2.4778],
        [ -6.3066,  -8.1936,   9.5822],
        [  0.9809,   7.6642,  -1.4217],
        [  8.5198,  10.4091,  -8.6486],
        [  4.3523, -10.5232,   8.2138],
        [ 10.9916,   7.2133,  -0.9179],
        [ -8.9178,   2.1936,   7.5388],
        [  2.5663,   7.0526,  -6.8769],
        [ -1.8823, -10.6558,  -3.8952],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.6202,  -1.9508,   0.7471],
        [  5.4426, -10.1181,  -5.1268],
        [ -6.1914,  -8.2997,   8.5380]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.7383e-02, 8.9800e-02,  ..., 6.7746e-01, 1.0462e-01,
         1.1985e+00],
        [4.7383e-02, 0.0000e+00, 4.2871e-02,  ..., 7.0487e-01, 1.2916e-01,
         1.1775e+00],
        [8.9800e-02, 4.2871e-02, 2.1453e-04,  ..., 7.3749e-01, 1.6515e-01,
         1.1681e+00],
        ...,
        [6.7746e-01, 7.0487e-01, 7.3749e-01,  ..., 0.0000e+00, 5.7605e-01,
         1.0524e+00],
        [1.0462e-01, 1.2916e-01, 1.6515e-01,  ..., 5.7605e-01, 0.0000e+00,
         1.1310e+00],
        [1.1985e+00, 1.1775e+00, 1.1681e+00,  ..., 1.0524e+00, 1.1310e+00,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.30125201,  8.41171369,  8.49161154,  8.62162372,  8.83341608,
        8.92395381,  9.87720102,  5.91408504,  8.94395628,  6.67108662,
        9.0368458 ,  9.77588801,  7.89745107, 12.3344536 ,  8.92395381,
        8.95983586,  8.87007495, 11.04953767,  5.24461068,  9.12220271,
        6.53290663, 11.785566  ,  6.21275016,  7.67939546,  8.36995953,
        8.41148722,  6.74686071,  6.76661595,  7.27340834,  8.67227386,
       10.87409005, 14.07623266,  4.91296205,  7.13501264,  5.63824971,
        7.33350642,  6.46560181,  7.60463349,  9.68324553,  6.18668463,
       10.36844813, 12.76408562,  5.92281142, 12.19805394, 11.018084  ,
        8.29203356,  8.53014915, 13.44593519,  6.34059021,  8.98765644,
        5.22903117,  7.58668144, 10.04916678, 13.12299145, 10.1979669 ,
       13.66991027,  7.24326992, 12.3828957 ,  6.05113217,  9.9353921 ,
        5.05774295, 12.59969936,  5.35774573,  6.38105015,  6.02779201,
        8.89301516,  8.7907923 , 11.50791859,  9.75401449, 10.60226614,
       12.33167382,  8.24808832,  7.28793663, 10.48315436, 11.4424816 ,
       12.13568598,  6.99097013, 11.81423161, 12.79126785, 13.1471313 ,
       10.42667093,  5.05989547, 12.889695  , 10.9036226 ,  9.45995955,
       12.22276278,  9.85330646,  7.34250948,  7.32950385, 11.99633819,
       10.71162681, 11.05317597,  5.0713364 , 10.38897563, 10.33958639,
        7.72668467, 13.45124917, 11.38774149, 13.1471313 ,  9.18358562,
        7.50500064, 10.8207297 , 11.80092578,  5.94909709, 11.4890861 ,
       10.35464916])}
{'P': tensor([[1.3966e-02, 1.8490e-01, 1.1949e-01,  ..., 1.0254e-05, 9.7106e-03,
         1.4567e-05],
        [3.4345e-04, 2.3232e-02, 4.6678e-03,  ..., 6.0132e-06, 1.8142e-02,
         2.0629e-05],
        [6.2623e-04, 8.1060e-02, 2.7538e-02,  ..., 7.7101e-06, 1.5250e-02,
         2.3811e-05],
        ...,
        [5.1079e-02, 3.0944e-02, 2.7677e-02,  ..., 1.7046e-01, 8.4204e-03,
         3.4750e-05],
        [7.3813e-05, 1.6485e-04, 1.4299e-04,  ..., 1.5540e-06, 2.9290e-02,
         2.7008e-05],
        [2.1561e-03, 1.0548e-03, 1.0746e-03,  ..., 2.8938e-03, 2.6833e-04,
         8.8612e-02]]), 'X': tensor([[ -3.6381,   7.4683,   7.3544],
        [ -5.3382,   6.6533,   7.8054],
        [ -5.1369,   6.7755,   8.3836],
        [ -6.4464,   5.8272,   7.5038],
        [ -6.3489,   5.9117,   6.5161],
        [ -5.9940,   6.1985,   5.9566],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.2022,   2.4092,   3.0349],
        [  6.0031,  12.2958,  -6.5992],
        [  0.9074,  -4.8139,   8.1674],
        [ -6.1914,  -8.2997,   8.5380],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.9003,  12.3550,   2.1917],
        [ 11.9019,  -3.1379,  -3.4534],
        [  6.2365,  -0.2578,  -6.0212],
        [  4.5503,   5.5359,   7.4735],
        [  3.2701,   6.6338,   9.9209],
        [  4.5800,   5.5034,  -3.9432],
        [  6.1372,   2.6726,  -6.7885],
        [  4.5491,  -3.3485,   6.8412],
        [ 11.2571,   6.7072,   8.3693],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -0.1963,   7.8765,  -1.1371],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.3736,   6.6310,  -1.5174],
        [  0.1931,  -4.7491,   7.4385],
        [  5.8889,  -1.3916,   8.8993],
        [ -1.2454,   7.9127,  -0.5715],
        [  1.2273,  -4.8081,   9.8129],
        [ -9.2508,   1.0572,   2.9978],
        [ -1.7290,   7.8825,  -3.3831],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.9636,   2.0648,   8.1946],
        [ -5.3382,   6.6533,   7.8054],
        [  1.8320,   7.3871,   5.1452],
        [  5.9587,  12.3216,   9.1695],
        [  7.0171,  11.6439,  -5.0178],
        [  1.1994,   7.6035,   2.4295],
        [  3.2809,  13.5061,  -4.0199],
        [  2.4572,   7.1085,   7.7376],
        [  9.1409,  -7.5789,   1.5814],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -3.1204, -10.2345,   9.2288],
        [  5.1472,  -2.6801,   8.1194],
        [  0.3038,   7.8092,   0.4164],
        [  3.3633,  13.4794,   0.5193],
        [ 10.6673,  -5.6649,  -4.1646],
        [  4.8674,  12.8875,   6.7622],
        [  7.7750,  11.0665,   2.8364],
        [ 11.4235,   6.3594,   5.1663],
        [  0.8572,  14.0501,  -8.3364],
        [  6.2630,   2.1057,  -5.2136],
        [ -7.6650,   4.5092,   0.2520],
        [  6.3460,   1.4252,   1.2291],
        [ -8.8540,  -4.3971,  -5.2144],
        [  8.7131,  10.2216,   7.7915],
        [  6.2555,   2.1474,   4.1685],
        [ -8.6112,   2.9319,  -7.9943],
        [ 12.5650,   2.0132,   4.6272],
        [ 11.6544,   5.8273,  -3.1375],
        [ -5.9940,   6.1985,   5.9566],
        [ -7.4602,   4.7718,  -4.6277],
        [ -7.3446,   4.9113,  -5.9356],
        [ -3.5173, -10.0623,   4.8280],
        [ -7.4602,   4.7718,  -4.6277],
        [ -8.1426,   3.8048,   9.3542],
        [ -9.4541,  -0.3327,   8.8246],
        [  1.3705,  13.9723,   7.9266],
        [ -5.3736,   6.6310,  -1.5174],
        [ -0.4341, -10.9473,   7.4509],
        [  5.2911,  -2.4824,  -3.6877],
        [ -8.1204,  -5.9308,   4.0761],
        [  2.6321,  -4.5296,  -2.5159],
        [ -6.2668,   5.9808,   2.5614],
        [  0.3724,   7.7974,  -6.8722],
        [  9.8082,  -6.8396,   4.9615],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.3505,   0.6235,  -8.5849],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -0.8675, -10.8815,   6.8561],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.4157,   0.0821,  -5.0025],
        [  5.3858,   4.4260,  -9.6769],
        [  1.9204,   7.3518,   8.4781],
        [  0.6363, -11.0348,   6.1414],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.3076,   0.7852,  -6.7010],
        [ -8.9788,   2.0208,  -3.2867],
        [ 11.5966,   5.9666,   6.0083],
        [ -2.9497, -10.3028,  -7.3479],
        [  6.0031,  12.2958,  -6.5992],
        [ -1.9487,   7.8590,  -3.3330],
        [ -6.1914,  -8.2997,   8.5380],
        [ 12.6038,   0.7665,  -3.4439],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.3609,   3.4769,  -8.5762],
        [  3.9770, -10.6341,   8.0783],
        [  6.3583,   1.1729,   8.7835],
        [ -7.4807,   4.7463,   2.0616],
        [  2.9470,  13.6085,   2.9809],
        [ -8.3900,   3.3727,   3.1650],
        [  6.2883,   0.0325,   5.2692]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0868, 0.0917,  ..., 0.6566, 0.4347, 0.4292],
        [0.0868, 0.0000, 0.0160,  ..., 0.7433, 0.3899, 0.5155],
        [0.0917, 0.0160, 0.0000,  ..., 0.7466, 0.4026, 0.5208],
        ...,
        [0.6566, 0.7433, 0.7466,  ..., 0.0000, 0.9855, 0.2408],
        [0.4347, 0.3899, 0.4026,  ..., 0.9855, 0.0000, 0.7459],
        [0.4292, 0.5155, 0.5208,  ..., 0.2408, 0.7459, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.30729352,  8.53014915,  8.50268322,  8.68979052,  8.67502348,
        8.62260327,  9.86089579,  6.65368551, 13.6829843 ,  4.89869205,
       10.35464916, 13.09426142, 13.69162795, 12.30858677,  6.24187044,
        7.16598273,  7.39599986,  7.15990676,  6.69389431,  5.64865173,
       13.103709  ,  9.3349858 ,  9.75401449, 12.4013153 ,  9.60080639,
        7.8788965 , 10.51437947,  8.53501777,  4.75303127,  6.05113217,
        8.01009083,  4.96231532,  9.3109884 ,  8.06990483,  9.23071698,
        9.19837792,  8.53014915,  7.61088125, 13.68672855, 13.59480906,
        7.69752262, 13.89886469,  7.52117132, 11.87412439, 12.3828957 ,
       10.69963954,  5.80312949,  7.81509631, 13.89264822, 12.07818871,
       13.77604293, 13.5247215 , 13.0743399 , 14.07623266,  6.60752373,
        8.89301516,  6.50409372,  9.88571713, 13.43127131,  6.61386436,
        9.09661212, 12.72524615, 13.03002338,  8.62260327,  8.85572292,
        8.8353674 , 10.65929968,  8.85572292,  8.98765644,  9.45995955,
       14.0393949 ,  8.53501777, 10.95593729,  5.84450127, 10.05558893,
        5.23878371,  8.66273171,  7.80625981, 11.95743558,  9.68323686,
        6.38105015, 12.49291768, 10.91601655, 10.61047596,  9.41605684,
        6.97106912,  7.5984809 , 11.05317597, 12.05395056,  8.29022197,
        9.34061938,  9.20340031, 13.04156075, 10.7167285 , 13.6829843 ,
        8.09704052, 10.35464916, 12.62711252,  9.66977802, 12.84056604,
       11.35345345,  6.46560181,  8.85939626, 13.92390689,  9.04255655,
        6.28835101])}
{'P': tensor([[1.6630e-02, 2.5976e-01, 2.4220e-01,  ..., 1.6373e-05, 1.6373e-05,
         5.4539e-05],
        [3.6404e-04, 1.5602e-02, 4.4261e-01,  ..., 1.5418e-05, 1.5418e-05,
         7.6239e-05],
        [5.0630e-04, 6.9937e-04, 2.9260e-02,  ..., 3.0426e-05, 3.0426e-05,
         1.7148e-04],
        ...,
        [1.2044e-04, 9.1038e-05, 8.5218e-05,  ..., 1.3439e-01, 1.3439e-01,
         2.7862e-04],
        [1.2044e-04, 9.1038e-05, 8.5218e-05,  ..., 1.3439e-01, 1.3439e-01,
         2.7862e-04],
        [3.0362e-06, 2.7927e-06, 2.7475e-06,  ..., 8.7945e-04, 8.7945e-04,
         2.2502e-02]]), 'X': tensor([[-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.0403e-02, 1.0028e-01,  ..., 6.4629e-01, 6.4629e-01,
         5.0092e-01],
        [6.0403e-02, 7.5848e-05, 4.4166e-02,  ..., 7.0150e-01, 7.0150e-01,
         4.6434e-01],
        [1.0028e-01, 4.4166e-02, 0.0000e+00,  ..., 7.2697e-01, 7.2697e-01,
         4.2381e-01],
        ...,
        [6.4629e-01, 7.0150e-01, 7.2697e-01,  ..., 0.0000e+00, 0.0000e+00,
         9.1489e-01],
        [6.4629e-01, 7.0150e-01, 7.2697e-01,  ..., 0.0000e+00, 0.0000e+00,
         9.1489e-01],
        [5.0092e-01, 4.6434e-01, 4.2381e-01,  ..., 9.1489e-01, 9.1489e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.31625171,  8.4486771 ,  8.50278896,  8.5724042 ,  8.72896381,
        8.57560275,  5.29755487, 10.99326577, 11.32149259,  8.90405823,
       13.32613445,  9.41605684, 13.89974693, 12.40074901, 12.5085995 ,
       13.77075174,  5.20659892, 11.75618513, 10.82106505, 12.13568598,
       10.75443106, 13.96746816, 13.70893492, 14.03505624, 10.95593729,
       13.02818702,  5.76165934, 13.95472331, 12.85032836, 10.26545792,
       11.85743425,  5.13182506,  9.95552626,  9.17595899,  6.2134961 ,
       11.59512194,  5.94293185,  9.68324553,  9.5014146 , 11.00812996,
       13.35927527, 14.05687815,  5.01114719,  5.5366698 , 11.66453059,
       11.62158554,  9.97964013,  8.51928531, 13.91198191, 10.41909211,
        5.5366698 , 13.68063098, 12.52372629,  9.17905491, 10.05558893,
       12.10222204, 13.98691426, 13.93790235,  8.86082124,  9.62958973,
        5.21280156,  7.44805901, 10.53368644,  6.77777606,  6.80370254,
        5.35774573,  9.87720102,  4.83371341, 10.0674887 , 12.17901033,
        6.99097013,  6.56833864,  8.10886021,  6.80370254,  5.01114719,
       10.92597196,  5.13616224,  7.822854  , 11.05413545,  9.27474386,
       13.59056613, 10.75443106,  4.98461983,  5.79543038,  8.95983586,
       11.00726483,  8.89458273, 11.80092578,  8.8580671 ,  9.11473391,
        9.48969978,  9.62958973, 13.58333857,  5.41348121, 13.06498554,
        7.62011734,  8.09899874, 13.13205357,  7.72291946, 11.75618513,
        9.14973319, 11.22596994, 10.06286074,  5.13125513,  5.13125513,
       10.61417181])}
{'P': tensor([[2.0525e-02, 2.9957e-01, 1.5432e-01,  ..., 3.9584e-04, 6.1988e-05,
         1.8874e-05],
        [3.6813e-04, 2.4870e-02, 4.3295e-02,  ..., 4.0503e-04, 4.0777e-05,
         2.0505e-05],
        [6.5794e-04, 6.5545e-03, 4.5812e-02,  ..., 1.5344e-03, 5.3052e-05,
         3.1953e-05],
        ...,
        [2.8548e-05, 4.7309e-05, 9.1816e-05,  ..., 3.8369e-02, 1.6060e-06,
         3.1254e-06],
        [3.1298e-02, 2.3084e-02, 1.0931e-02,  ..., 6.9082e-05, 1.0110e-01,
         3.3035e-04],
        [1.9036e-03, 1.0621e-03, 7.0693e-04,  ..., 1.7595e-05, 2.9796e-02,
         9.0895e-02]]), 'X': tensor([[-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0981e-01, 1.5154e-01,  ..., 8.7158e-01, 2.7933e-01,
         4.4637e-01],
        [1.0981e-01, 1.5170e-04, 1.0201e-01,  ..., 8.6750e-01, 3.5201e-01,
         5.4496e-01],
        [1.5154e-01, 1.0201e-01, 0.0000e+00,  ..., 7.6562e-01, 4.2784e-01,
         5.9490e-01],
        ...,
        [8.7158e-01, 8.6750e-01, 7.6562e-01,  ..., 0.0000e+00, 1.1264e+00,
         1.1846e+00],
        [2.7933e-01, 3.5201e-01, 4.2784e-01,  ..., 1.1264e+00, 0.0000e+00,
         2.3659e-01],
        [4.4637e-01, 5.4496e-01, 5.9490e-01,  ..., 1.1846e+00, 2.3659e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.34949383,  8.55982105,  8.62631801,  8.7907923 ,  8.51928531,
        8.85572292,  6.39520397,  6.61386436,  5.88782913,  7.42512621,
        5.29755487,  5.81199422,  5.24137909,  8.1137452 , 11.75807252,
        8.77913428, 13.12299145, 13.20597932, 11.35182589,  7.49076899,
        9.66977802,  6.78325151,  8.20436764,  6.39187285,  9.45995955,
        8.1303945 ,  6.58184474, 11.97238624,  9.10945678,  8.22728399,
        8.09984231,  7.16747878,  6.5618968 , 10.48840476,  4.90676242,
        8.97052115, 10.65929968,  9.14973319,  4.75303127, 12.05300943,
        8.09984231, 10.69963954,  8.01009083,  8.859416  , 10.06703358,
       11.83570132, 11.73848065,  8.95983586,  8.68979052,  4.91652556,
       12.53285633,  6.59176018, 13.34587027,  7.8788965 , 11.89105259,
        7.52117132,  9.19837792,  6.77777606, 13.77604293,  8.11227668,
        5.46244988, 11.00726483, 13.59673406,  8.35856487,  7.34250948,
       11.97949273, 11.76830981,  8.56889738, 13.97857591,  7.38844703,
        5.8828986 , 10.75443106, 13.56367857,  6.42689705,  9.36378438,
        8.50268322,  9.79814997, 12.05300943,  8.50268322,  6.54693987,
       13.56367857, 10.32715715,  4.91296205,  5.5205189 ,  7.05627461,
       14.02058621, 13.31764181,  9.58072286, 13.36800073, 12.12448912,
        7.62509582,  5.77219077, 11.63502679, 13.0743399 , 11.75807252,
        9.68323686,  7.5984809 , 11.46176259, 11.72802491, 13.85158417,
        8.81930525, 10.37698435,  9.38232121,  9.09119351,  7.53884104,
        6.34059021])}
{'P': tensor([[2.1031e-02, 2.3039e-01, 2.0538e-01,  ..., 1.1389e-05, 8.6001e-03,
         7.1541e-06],
        [5.8875e-04, 4.7272e-02, 6.7635e-03,  ..., 3.4073e-05, 4.7846e-02,
         5.9627e-06],
        [5.1204e-04, 4.4650e-02, 2.5648e-02,  ..., 1.2227e-05, 1.0633e-02,
         3.0955e-06],
        ...,
        [7.0759e-05, 5.1839e-05, 3.9346e-05,  ..., 5.8009e-02, 4.6911e-05,
         1.4026e-04],
        [9.8485e-05, 2.3446e-04, 1.7333e-04,  ..., 2.2133e-04, 6.7620e-02,
         1.9590e-06],
        [6.1162e-03, 7.5103e-03, 4.0903e-03,  ..., 4.3294e-05, 5.3309e-03,
         3.6459e-01]]), 'X': tensor([[ -4.0409,   7.3169,  -3.5779],
        [ -6.0195,   6.1789,  -2.9496],
        [ -5.5525,   6.5146,  -4.4984],
        [ -7.0825,   5.2073,  -5.0497],
        [ -7.4602,   4.7718,  -4.6277],
        [ -5.6174,   6.4708,  -1.7339],
        [  4.5503,   5.5359,   7.4735],
        [ -0.1963,   7.8765,  -1.1371],
        [  0.6142,  14.0802,   0.5407],
        [  0.9036,   7.6840,   9.9237],
        [  6.3569,   0.7591,   6.5011],
        [  6.1755,   2.5233,  -2.0042],
        [  2.3589,  13.7668,  -3.6501],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.0870,  -4.6877,   9.6823],
        [  3.2647, -10.8066,  -0.3768],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.3246,   1.6696,  -0.1544],
        [  2.6613,   7.0021,   4.5783],
        [ 12.1751,  -2.2592,  -5.7751],
        [  5.8323,  -1.5228,   5.1132],
        [ -6.6998,   5.5954,   0.0823],
        [  8.1528,  10.7452,   9.1493],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.8719,   7.3713,  -3.7465],
        [ 12.2592,   3.9425,   2.0383],
        [  3.9663,   6.1018,  -5.4104],
        [  1.4306,  13.9620,  -0.2075],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -6.3306,   5.9272,   5.6935],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.8093,  -9.4178,   8.4244],
        [ -3.5903,   7.4847,  -8.9112],
        [ -8.6112,   2.9319,  -7.9943],
        [ 10.9875,   7.2207,  -8.8549],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.0419,  -4.6978,  -5.0983],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.5650,   2.0132,   4.6272],
        [  6.1973,  -0.4372,   6.5720],
        [ 10.0567,   8.6697,   4.6329],
        [  0.6933,   7.7336,   2.1179],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.9033,   2.2329,  -8.8912],
        [ -5.3736,   6.6310,  -1.5174],
        [ -6.6895,   5.6051,   2.7899],
        [  9.6816,   9.1536,  -5.8543],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.3516,   0.6425,   0.3425],
        [  6.3246,   1.6696,  -0.1544],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -5.6169,  -8.7844,   8.9057],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.9693,  -9.3200,  -8.3822],
        [  6.1741,  -0.5329,   3.1913],
        [  6.2535,  -0.1709,  -9.4102],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -9.1231,  -3.5739,   8.4708],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -0.7599, -10.8995,  -6.0366],
        [ -9.3384,  -2.6161,  -4.7630],
        [  8.1528,  10.7452,   9.1493],
        [  3.4009,  -4.1874,  -5.9751],
        [  0.1382, -11.0073,   1.0026],
        [ 10.7566,   7.6212,  -2.3279],
        [  1.5653,  13.9377,   4.1773],
        [ -7.6650,   4.5092,   0.2520],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.2883,   0.0325,   5.2692],
        [  9.9217,  -6.7006,  -5.6728],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.3739,   0.3982,  -4.2564],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.3087,   1.8053,  -1.9481],
        [  7.7750,  11.0665,   2.8364],
        [  0.2435,   7.8191,  -6.4709],
        [  2.5620,   7.0549,   8.4825],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 10.3251,   8.2928,  -3.0006],
        [  5.0233,   4.9647,  -3.1788],
        [  6.0305,   3.0296,   0.1375],
        [ -3.3812, -10.1235,   7.4384],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.2707,   0.9661,   4.4863],
        [ -0.8675, -10.8815,   6.8561],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.4714,  -1.3301,  -3.1998],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.6035,   0.7494,   7.1536],
        [  5.4242, -10.1261,   3.9595],
        [  4.9957,  -2.8705,   6.5978],
        [  1.5383, -11.0274,  -4.9666],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 11.2847,  -4.5848,  -4.7503],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.0604,   1.7707,   1.1066],
        [  1.3705,  13.9723,   7.9266]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.3716e-01, 1.2272e-01,  ..., 8.0189e-01, 5.1327e-01,
         1.1853e+00],
        [1.3716e-01, 0.0000e+00, 1.0201e-01,  ..., 8.5138e-01, 3.9376e-01,
         1.2366e+00],
        [1.2272e-01, 1.0201e-01, 1.5170e-04,  ..., 9.1246e-01, 4.8773e-01,
         1.2973e+00],
        ...,
        [8.0189e-01, 8.5138e-01, 9.1246e-01,  ..., 0.0000e+00, 8.8402e-01,
         3.8535e-01],
        [5.1327e-01, 3.9376e-01, 4.8773e-01,  ..., 8.8402e-01, 0.0000e+00,
         1.2375e+00],
        [1.1853e+00, 1.2366e+00, 1.2973e+00,  ..., 3.8535e-01, 1.2375e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.35856487,  8.62631801,  8.55982105,  8.7907923 ,  8.85572292,
        8.56889738,  7.16598273,  7.8788965 , 14.09357547,  7.73692436,
        6.4020404 ,  6.67108662, 13.96746816,  5.24137909,  5.13125513,
       11.28895489,  5.19065439, 10.00794468,  6.54127431,  7.49076899,
       12.3828957 ,  6.02779201,  8.72896381, 13.48809595, 11.97957425,
        9.17905491,  7.60529892, 12.87752052,  7.27759233, 14.03505624,
       10.27572673,  9.5014146 , 12.3489324 ,  5.94909709, 10.00410625,
        8.67227386,  8.34949383, 11.62158554,  8.30125201,  9.09661212,
       13.14777433,  9.68324553,  5.1224079 ,  5.05444288, 12.72524615,
        6.21275016, 13.2778378 ,  7.76457095,  8.12043983,  9.17905491,
        8.53501777,  8.7273569 , 13.32374056, 12.52875793,  6.38399789,
        6.54127431,  9.61942445, 10.42667093,  8.47371868,  4.76969364,
       11.63765273,  6.19707944,  6.25586297,  9.86089579, 12.25806559,
        9.79814997,  9.75401449, 10.92597196,  9.69792513, 13.48809595,
        5.39450305, 11.00812996, 13.18278606, 14.02532758,  8.89301516,
        5.90035324,  6.28835101, 11.97238624, 12.3828957 ,  9.38232121,
        5.8828986 ,  6.5618968 , 13.5247215 ,  7.822854  ,  7.50564223,
        7.72668467,  9.47369628, 13.2430377 ,  7.06272256,  6.74872981,
       10.67322527,  5.40689332,  9.32094055, 10.91601655, 13.79762415,
        9.5642993 , 13.79762415, 12.62576026, 11.48734978,  5.76165934,
       11.1341798 , 10.37698435, 12.18045978,  5.5748624 ,  9.23178204,
       14.0393949 ])}
{'P': tensor([[1.6129e-02, 3.1319e-01, 2.6006e-01,  ..., 6.1848e-07, 6.8936e-05,
         2.1137e-05],
        [5.4787e-04, 2.8935e-02, 8.5170e-02,  ..., 1.0553e-06, 1.8278e-04,
         2.7569e-05],
        [3.9554e-04, 4.3719e-03, 2.1760e-02,  ..., 8.5375e-07, 1.7249e-04,
         2.0769e-05],
        ...,
        [1.2645e-04, 7.5551e-05, 7.5877e-05,  ..., 1.3687e-01, 3.4074e-06,
         2.3162e-02],
        [1.1405e-06, 1.1479e-06, 1.2507e-06,  ..., 5.2967e-05, 1.1814e-02,
         8.3172e-06],
        [8.2602e-03, 5.2426e-03, 5.6647e-03,  ..., 3.2256e-05, 3.3020e-05,
         6.4826e-02]]), 'X': tensor([[ -4.0868,   7.2981,   8.1874],
        [ -5.1369,   6.7755,   8.3836],
        [ -5.3382,   6.6533,   7.8054],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.0120,   5.2825,   7.7762],
        [ -6.3489,   5.9117,   6.5161],
        [  5.7485,  -9.9811,  -3.2355],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 10.8218,  -5.4177,   6.8573],
        [  3.2366,  -4.2732,   5.1532],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.1292,   4.8183,  -5.7997],
        [  7.3592,  11.3937,   6.2407],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -3.5520,   7.4976,   9.1522],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 10.8218,  -5.4177,   6.8573],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -1.6086,   7.8928,   4.3385],
        [  1.3670,  -4.7990,   3.2603],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.3367,  -4.4806,   0.4643],
        [ -8.3900,   3.3727,   3.1650],
        [ -1.5408,   7.8978,   5.3470],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.9653,  -9.3225,  -3.4727],
        [  4.6063,  13.0051,  -3.1456],
        [ -8.9624,  -4.0944,  -4.9622],
        [  3.4009,  -4.1874,  -5.9751],
        [  1.2728, -11.0372,   9.8121],
        [ -8.0534,  -6.0429,   2.4583],
        [  8.6420,  -8.0580,   6.5781],
        [  2.5382,   7.0672,   4.5097],
        [  4.8674,  12.8875,   6.7622],
        [ -0.5709,   7.9055,   7.6089],
        [ -1.3178, -10.7939,   0.4236],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.1940,  -2.1881,   7.1499],
        [ -9.1813,  -3.3550,   1.8827],
        [ -9.4157,   0.0821,  -5.0025],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.4735,  -1.2620,  -3.6742],
        [  3.9085,   6.1512,  -2.2560],
        [ -4.2040,   7.2486,   2.9875],
        [ -2.0445,   7.8469,  -0.5857],
        [  3.2692,  13.5098,  -4.9032],
        [ -2.5683, -10.4433,  -2.9182],
        [ -7.3446,   4.9113,  -5.9356],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.3889,  -4.6082,  -8.7291],
        [ -1.6220,   7.8917,   0.0315],
        [  6.3606,   0.9204,   3.2525],
        [  4.5325,  -3.3643,   6.7699],
        [  5.1203,  -2.7151,   6.2110],
        [  1.8766,   7.3694,   1.1887],
        [ 12.4691,   2.8525,  -5.5110],
        [ -5.0943,  -9.1682,   0.5073],
        [  9.0424,  -7.6779,  -7.5286],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.2508,   1.0572,   2.9978],
        [  7.5930,  -8.9037,   7.2323],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -7.5720,   4.6310,   2.1717],
        [  5.9843,  -9.8679,  -6.1449],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -3.6381,   7.4683,   7.3544],
        [  6.2630,   2.1057,  -5.2136],
        [  1.9505,  -4.7171,   9.9868],
        [  6.9284,  11.7061,  -0.3732],
        [ -5.6792,  -8.7352,   4.0318],
        [  6.3031,   1.8477,  -7.1706],
        [ -7.4602,   4.7718,  -4.6277],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.4117,   0.1160,   8.2574],
        [  6.3305,   1.6119,   1.1538],
        [  8.2154,  -8.4262,   8.0171],
        [ -3.9456,  -9.8542,   0.7466],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.3602,  -2.3808,   7.2098],
        [ -0.8675, -10.8815,   6.8561],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.6954,  -4.7616,   9.8198],
        [  3.7121,   6.3117,   0.6738],
        [  5.4426, -10.1181,  -5.1268],
        [ -4.5778,  -9.5013,  -5.0667],
        [  9.3424,  -7.3685,  -3.6807],
        [ -9.4283,  -1.9585,  -0.9129],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.1973,  -0.4372,   6.5720],
        [ -8.0965,  -5.9712,   4.3466],
        [  9.1409,  -7.5789,   1.5814],
        [  3.2226,  -4.2802,  -4.0428],
        [ -2.7572,   7.7194,   5.9580],
        [ 10.5678,   7.9261,  -0.6100],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.0582,  -2.6640,   0.7132],
        [ -6.0844,  -8.3955,   0.7705],
        [  4.6992,   5.3690,   3.0935]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0609, 0.0617,  ..., 0.6553, 0.7133, 0.4593],
        [0.0609, 0.0000, 0.0160,  ..., 0.6978, 0.6745, 0.5096],
        [0.0617, 0.0160, 0.0000,  ..., 0.6859, 0.6628, 0.5001],
        ...,
        [0.6553, 0.6978, 0.6859,  ..., 0.0000, 0.9159, 0.2230],
        [0.7133, 0.6745, 0.6628,  ..., 0.9159, 0.0000, 0.8816],
        [0.4593, 0.5096, 0.5001,  ..., 0.2230, 0.8816, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.36445758,  8.50268322,  8.53014915,  8.68979052,  8.77913428,
        8.67502348, 11.51811603,  8.25828409,  5.18277858, 11.05413545,
        9.50654967, 12.10222204,  5.36061974,  8.47371868,  7.03733775,
       13.56367857, 12.32545637,  8.04664886, 10.00410625,  8.29640613,
        9.4997987 , 12.10222204, 10.30972193,  8.0550286 ,  4.98989906,
        8.85572292, 12.18998195,  9.04255655,  8.04664886,  5.18047638,
       11.63724792, 13.79676654,  9.85330646,  5.39450305, 11.11038551,
       10.06850677, 11.8159254 ,  7.50917551, 13.77604293,  7.92607342,
       10.87409005,  5.27178363, 12.38881635,  9.7751189 ,  9.41605684,
       12.4217523 ,  9.55720651,  7.28793663,  8.37954853,  8.10886021,
       13.89974693, 10.75443106,  8.8353674 , 11.9182956 ,  5.19065439,
        8.05668609,  6.42689705,  5.64464807,  5.79562691,  7.60463349,
       12.79126785, 10.48840476, 11.86239961,  7.98469166,  9.3109884 ,
       11.70168755,  9.91668182,  8.87587457, 11.54071859,  9.61942445,
        8.30729352,  6.60752373,  5.10446653, 13.60275211, 10.41909211,
        6.56833864,  8.85572292,  9.19837792,  9.41244865,  6.53250937,
       11.76830981, 10.61472636, 12.77402351,  5.86519525, 10.91601655,
       13.1399181 ,  6.00732193, 13.04217487,  5.05444288,  7.32233733,
       11.4890861 , 10.54658597, 11.89855267,  9.62958973,  5.13125513,
        9.34061938,  6.21275016, 10.06023665, 11.87412439,  5.35774573,
        8.19703512, 13.20988996, 11.99633819, 12.3489324 , 10.36844813,
        7.13501264])}
{'P': tensor([[2.1745e-02, 2.8589e-01, 1.9992e-01,  ..., 1.0759e-04, 4.7890e-05,
         2.6178e-03],
        [1.3866e-03, 1.1601e-01, 1.7157e-02,  ..., 2.2774e-04, 1.3636e-04,
         1.8929e-02],
        [1.9851e-03, 2.1254e-01, 9.4910e-02,  ..., 3.1634e-04, 1.1726e-04,
         2.6184e-02],
        ...,
        [4.4050e-03, 5.1940e-03, 9.1784e-03,  ..., 1.8190e-01, 1.1404e-04,
         3.2197e-02],
        [3.8585e-02, 1.8353e-02, 1.5894e-02,  ..., 5.0504e-04, 9.7449e-02,
         2.7784e-04],
        [6.0656e-05, 1.2570e-04, 1.2023e-04,  ..., 4.5395e-05, 9.0976e-06,
         6.8617e-02]]), 'X': tensor([[ -4.1296,   7.2803,  -2.7194],
        [ -6.0195,   6.1789,  -2.9496],
        [ -5.6174,   6.4708,  -1.7339],
        [ -5.3736,   6.6310,  -1.5174],
        [ -4.9217,   6.8979,  -1.7667],
        [ -5.6423,   6.4537,  -1.2091],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.4748,  -0.7765,  -5.6695],
        [  3.2692,  13.5098,  -4.9032],
        [ -3.0008, -10.2827,  -8.9480],
        [ 12.2914,   3.8037,   6.0978],
        [ -3.4488, -10.0934,   2.5195],
        [ -2.2508, -10.5479,  -2.7235],
        [  3.9515,  -3.8425,   6.1729],
        [ -1.5451,   7.8975,   7.6104],
        [  4.5325,  -3.3643,   6.7699],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.5456,   6.0861,   8.1101],
        [  6.3516,   0.6425,   0.3425],
        [  1.9503,  -4.7172,  -6.2313],
        [  1.6232,  -4.7717,  -2.5655],
        [ -3.4488, -10.0934,   2.5195],
        [ 11.9808,   4.9340,  -3.3715],
        [ -7.8280,   4.2848,  -7.1955],
        [  1.7228,  -4.7575,   8.3396],
        [  6.3266,   0.3292,   9.3502],
        [ -5.6792,  -8.7352,   4.0318],
        [ -0.8885,   7.9160,   0.0571],
        [ -3.4100,   7.5436,  -7.6980],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -9.2707,   0.9661,   4.4863],
        [  6.3087,   1.8053,  -1.9481],
        [  1.0093,   7.6567,   2.3526],
        [  3.7237,  -3.9966,  -5.0376],
        [  1.7122,  -4.7591,  -3.6734],
        [ -8.4824,  -5.2575,  -0.0795],
        [  3.4920,  -4.1366,  -8.3955],
        [  0.5960, -11.0335,   3.1144],
        [ 12.0845,  -2.5772,  -4.6073],
        [  6.3604,   0.9042,  -6.5901],
        [ -8.1905,   3.7253,   4.1175],
        [  2.4639, -10.9431,  -7.3718],
        [  4.5767,   5.5071,   3.9069],
        [  1.7122,  -4.7591,  -3.6734],
        [ -0.0410,   7.8591,   9.3457],
        [  5.5158,  -2.1337,   3.5246],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.5128, -10.9365,   9.9645],
        [  6.3301,   0.3638,  -7.5685],
        [  1.1139,   7.6281,  -2.5680],
        [  1.8719,   7.3713,  -3.7465],
        [ 12.2914,   3.8037,   6.0978],
        [ -6.1914,  -8.2997,   8.5380],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 10.9931,   7.2107,   7.7933],
        [  7.8982,  -8.6778,   0.0239],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.1813,  -3.3550,   1.8827],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.3554,   5.7393,  -9.0766],
        [  9.5018,  -7.1945,  -1.1892],
        [  6.3516,   0.6425,   0.3425],
        [  0.5600, -11.0321,  -0.8660],
        [  3.3069,   6.6089,   8.7377],
        [  6.0467,  -9.8369,  -6.5816],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.3542,   0.6934,  -3.3840],
        [  0.1047, -11.0046,   0.5188],
        [ 10.1798,   8.5002,  -5.8893],
        [ -9.1960,  -3.2964,   1.6921],
        [  2.9461,  -4.4077,  -0.4831],
        [  6.1989,  -9.7591,   0.8289],
        [  2.1824,  -4.6647,   1.1985],
        [ -1.7009,   7.8851,  -8.2437],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.4366,  -0.1189,  -2.6736],
        [  0.8572,  14.0501,  -8.3364],
        [  0.9036,   7.6840,   9.9237],
        [ 12.1940,  -2.1881,   7.1499],
        [ -0.9715, -10.8630,   0.3555],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.3124,   0.2057,  -0.4593],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.6248,   3.9984,   8.4933],
        [ -7.0948,   5.1940,   9.8171],
        [  7.8425,  -8.7202,   6.4530],
        [  6.2630,   2.1057,  -5.2136],
        [  5.8201,   3.5841,  -8.0609],
        [  5.7485,  -9.9811,  -3.2355],
        [ -9.3464,   0.5708,  -1.6397],
        [ -5.6651,   6.4380,   1.3660],
        [  2.2437,  -4.6489,   6.1430],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -2.0229, -10.6163,   8.6930],
        [  5.8323,  -1.5228,   5.1132],
        [  1.8721,  -4.7322,   0.5524],
        [ -1.6906,   7.8860,   7.7142],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.2707,   0.9661,   4.4863]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1407, 0.1147,  ..., 0.7339, 0.3549, 0.6526],
        [0.1407, 0.0000, 0.0942,  ..., 0.7937, 0.4690, 0.6011],
        [0.1147, 0.0942, 0.0000,  ..., 0.7001, 0.4696, 0.5470],
        ...,
        [0.7339, 0.7937, 0.7001,  ..., 0.0000, 0.9257, 0.5652],
        [0.3549, 0.4690, 0.4696,  ..., 0.9257, 0.0000, 0.9927],
        [0.6526, 0.6011, 0.5470,  ..., 0.5652, 0.9927, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.36995953,  8.62631801,  8.56889738,  8.53501777,  8.47371868,
        8.5724042 ,  7.8575648 ,  9.50654967, 13.89974693, 10.71162681,
       12.86648771, 10.66631784, 10.78533838,  5.51177501,  8.04718433,
        5.64464807,  6.83515201, 13.05148826,  6.38399789,  5.10444097,
        5.04028538, 10.66631784, 12.95702647,  8.92395381,  5.05981914,
        6.33517349, 10.41909211,  7.96575782,  8.27853076, 10.37698435,
        9.32094055,  6.5618968 ,  7.72291946,  5.46244988,  5.05774295,
        9.97964013,  5.41348121, 11.04953767, 12.35625694,  6.42439344,
        8.99791316, 11.21703654,  7.16058587,  5.05774295,  7.85919687,
        5.91408504,  8.41148722, 12.16901615, 11.22142195,  6.34059021,
        7.70897675,  7.60529892, 12.86648771, 10.35464916, 13.26590071,
       10.05558893,  9.95552626, 10.0674887 , 10.30972193, 13.14690137,
       11.73397309,  8.37954853,  9.91668182,  9.562174  ,  9.7751189 ,
        5.01114719,  7.20482224, 11.9182956 ,  6.38399789, 11.04629471,
        7.39004226, 11.54672755, 12.9089559 ,  6.39187285, 11.00508681,
       13.26209545,  9.76896904,  5.30158497, 11.56146873,  5.14999799,
        8.06643487, 11.05413545,  9.43737523, 14.07623266,  7.73692436,
       12.38881635, 10.90638066,  8.12580976,  6.31576264, 12.18045978,
        6.90115901,  8.79284466, 11.72802491,  6.60752373,  6.83515201,
       11.51811603,  9.36378438,  8.57560275,  5.16205087, 12.12641969,
       10.80728788,  6.02779201,  5.08910491,  8.06517143,  7.37902766,
        9.32094055])}
{'P': tensor([[9.2050e-03, 1.6011e-01, 1.0523e-01,  ..., 9.5826e-06, 5.3153e-07,
         1.8868e-03],
        [2.4059e-04, 1.1228e-02, 1.2680e-01,  ..., 1.0882e-05, 6.3919e-07,
         1.7906e-03],
        [1.0826e-04, 2.3950e-04, 9.7934e-03,  ..., 9.4711e-06, 6.2948e-07,
         1.8826e-03],
        ...,
        [1.0497e-03, 6.9569e-04, 4.6950e-04,  ..., 6.2073e-02, 1.1517e-03,
         1.4279e-04],
        [6.0572e-05, 3.6634e-05, 2.3438e-05,  ..., 9.3082e-03, 5.4961e-02,
         1.0030e-05],
        [2.3246e-04, 3.6844e-04, 8.7220e-04,  ..., 4.2627e-05, 5.2390e-06,
         6.0279e-02]]), 'X': tensor([[ -4.2040,   7.2486,   2.9875],
        [ -5.1974,   6.7396,   3.3484],
        [ -6.2668,   5.9808,   2.5614],
        [ -6.4368,   5.8356,   3.3014],
        [ -6.6895,   5.6051,   2.7899],
        [ -4.4503,   7.1378,   3.3751],
        [  9.6587,   9.1817,  -0.8212],
        [ -8.6426,  -4.9136,   9.4941],
        [  1.8721,  -4.7322,   0.5524],
        [ -5.0550,   6.8231,  -9.2502],
        [ -0.8569,   7.9156,  -6.5557],
        [  7.0664,  -9.2590,   8.0879],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.3038,   7.8092,   0.4164],
        [ -0.0254, -10.9932,   3.2200],
        [  8.5093,  -8.1764,  -3.2381],
        [ -7.7239,   4.4298,   3.1216],
        [ -5.9987,   6.1949,  -6.2114],
        [  6.3305,   1.6119,   1.1538],
        [  4.5503,   5.5359,   7.4735],
        [  9.3624,   9.5318,   0.2779],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.9178,   2.1936,   7.5388],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -5.1046,  -9.1610,   2.3418],
        [  5.2608,  -2.5256,  -9.1446],
        [ 10.1983,  -6.3430,  -0.3200],
        [  1.6408,   7.4591,  -4.2294],
        [  3.2701,   6.6338,   9.9209],
        [ 10.0400,   8.6922,  -6.2847],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 11.6019,   5.9540,  -5.1715],
        [ -9.1178,   1.5784,   2.1410],
        [  5.5626,  -2.0536,   8.4433],
        [  6.3549,   0.7096,   6.9198],
        [ 11.1420,   6.9331,  -2.4779],
        [ -4.7389,   6.9953,   3.7638],
        [ 11.1469,   6.9237,   1.6160],
        [ 10.4737,  -5.9570,  -1.2108],
        [  9.7328,   9.0902,  -0.8917],
        [  2.5382,   7.0672,   4.5097],
        [  2.4697,  -4.5836,   1.4593],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.8836,   3.4315,  -4.0327],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.9693,  -9.3200,  -8.3822],
        [ -1.9487,   7.8590,  -3.3330],
        [  2.3373,  -4.6232,  -5.2246],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.1813,  -3.3550,   1.8827],
        [ -9.4671,  -0.5553,   0.1117],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -8.3900,   3.3727,   3.1650],
        [  2.6310, -10.9195,   3.8054],
        [  4.5430,   5.5438,  -3.8669],
        [  3.7992,  -3.9474,  -8.9019],
        [  2.2076,  13.8032,  -1.5099],
        [  6.3246,   1.6696,  -0.1544],
        [ -4.7334,   6.9982,  -0.4392],
        [ -0.8885,   7.9160,   0.0571],
        [ 10.5678,   7.9261,  -0.6100],
        [ -5.1974,   6.7396,   3.3484],
        [  1.7500,   7.4187,   4.4237],
        [  9.8082,  -6.8396,   4.9615],
        [  9.3560,   9.5391,   9.4943],
        [ 11.2209,  -4.7092,   2.6117],
        [  9.3424,  -7.3685,  -3.6807],
        [ 12.5919,   0.4035,  -5.8460],
        [ -3.9858,  -9.8334,  -3.7323],
        [  7.8425,  -8.7202,   6.4530],
        [  4.5954,  13.0099,  -4.4439],
        [ -4.2271,   7.2387,  -1.8316],
        [  5.8836,   3.4315,  -4.0327],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.1440, -10.9811,   8.5281],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.1203,  -2.7151,   6.2110],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -8.7544,   2.6096,   3.3146],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.6321,  -4.5296,  -2.5159],
        [  3.3094,  13.4969,  -3.6507],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.0039,   7.3173,  -1.5885],
        [  2.3261,   7.1726,   6.2522],
        [  0.3724,   7.7974,  -6.8722],
        [ 10.9875,   7.2207,  -8.8549],
        [ -9.0604,   1.7707,   1.1066],
        [  6.3301,   0.3638,  -7.5685],
        [  4.7156,  -3.1831,  -8.0224],
        [  6.3516,   0.6425,   0.3425],
        [ -7.4807,   4.7463,   2.0616],
        [  7.9141,  -8.6656,   9.6948],
        [ -1.3178, -10.7939,   0.4236],
        [  1.9204,   7.3518,   8.4781],
        [ -6.6895,   5.6051,   2.7899],
        [  9.0652,   9.8597,  -3.5949],
        [  2.0729,   7.2878,   7.0880],
        [  6.1984,  -0.4327,   1.5438],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -7.9298,   4.1367,  -3.9773]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0626, 0.1283,  ..., 0.4547, 0.5263, 0.6213],
        [0.0626, 0.0000, 0.0898,  ..., 0.5172, 0.5856, 0.6175],
        [0.1283, 0.0898, 0.0000,  ..., 0.5636, 0.6012, 0.5360],
        ...,
        [0.4547, 0.5172, 0.5636,  ..., 0.0000, 0.2136, 0.8323],
        [0.5263, 0.5856, 0.6012,  ..., 0.2136, 0.0000, 0.7139],
        [0.6213, 0.6175, 0.5360,  ..., 0.8323, 0.7139, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.37954853,  8.51088913,  8.66273171,  8.68832398,  8.7273569 ,
        8.41148722, 13.32645552,  9.94173708,  5.08910491,  8.49161154,
        7.96181288, 11.64746882,  8.62631801,  7.81509631, 10.99326577,
       11.80092578,  8.90405823,  8.62329466,  6.53250937,  7.16598273,
       13.3607379 ,  9.26455795,  9.18358562,  6.06416957, 12.3828957 ,
       10.48721382,  5.83560009, 12.00995898,  7.6374603 ,  7.39599986,
       13.27994304, 10.51437947, 13.04051744,  9.25336371,  5.92952538,
        6.39438044, 13.12299145,  8.44940661, 13.12218013, 12.04924159,
       13.31764181,  7.50917551,  5.20659892, 13.04156075, 12.85032836,
        8.8737556 ,  9.97822473,  6.81117802,  4.978704  , 11.63765273,
        8.09704052,  5.18047638, 10.78533838,  9.7751189 ,  9.48337063,
       12.05395056,  9.04255655, 11.23201619,  7.16747878,  5.47866221,
       13.97857591,  6.54127431,  8.4486771 ,  7.96575782, 13.20988996,
        8.51088913,  7.62232307, 11.95743558, 13.3614598 , 12.16901615,
       11.89855267, 12.59840193, 10.61047596, 11.72802491, 13.79762415,
        8.38252937,  6.81117802, 11.83509372,  9.79814997, 11.18839169,
        6.02789634,  5.79562691, 12.32545637,  9.13507321, 10.41909211,
        5.23878371, 13.89671314,  6.19633139,  7.58668144,  7.5403822 ,
        7.80625981, 13.14777433,  9.23178204,  6.34059021,  5.68942572,
        6.38399789,  8.85939626, 11.73567759, 10.87409005,  7.5984809 ,
        8.7273569 , 13.39372426,  7.57685908,  6.2134961 , 12.25806559,
        8.94395628])}
{'P': tensor([[1.1070e-02, 2.6573e-01, 1.9385e-01,  ..., 8.4054e-07, 1.0607e-05,
         3.7252e-05],
        [2.8229e-04, 1.3022e-02, 2.5221e-01,  ..., 8.1852e-07, 1.1303e-05,
         4.8247e-05],
        [4.1176e-04, 7.0208e-04, 2.2171e-02,  ..., 1.2469e-06, 1.8357e-05,
         8.8646e-05],
        ...,
        [1.6329e-03, 1.1102e-03, 8.6337e-04,  ..., 5.2978e-02, 7.3715e-04,
         1.6554e-05],
        [1.7052e-03, 1.2788e-03, 1.0783e-03,  ..., 7.9771e-03, 4.1158e-02,
         2.5145e-04],
        [4.9574e-04, 4.2627e-04, 3.8909e-04,  ..., 2.6419e-03, 3.7179e-02,
         1.5893e-01]]), 'X': tensor([[ -4.2271,   7.2387,  -1.8316],
        [ -4.9217,   6.8979,  -1.7667],
        [ -5.3736,   6.6310,  -1.5174],
        [ -5.6174,   6.4708,  -1.7339],
        [ -5.6423,   6.4537,  -1.2091],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.3354,  -4.7708,  -4.9125],
        [ 11.4495,   6.3026,   6.8257],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.3087,   1.8053,  -1.9481],
        [  1.7301,   7.4262,   5.0522],
        [  1.1350,  14.0104,   0.6321],
        [ -8.3535,  -5.5118,   9.8399],
        [  1.0093,   7.6567,   2.3526],
        [ -5.3328,  -8.9992,   7.3440],
        [  8.7715,  -7.9390,   5.2900],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.9942,  -3.9990,   6.0448],
        [  1.3705,  13.9723,   7.9266],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.0305,   3.0296,   0.1375],
        [ 12.6059,   0.9597,  -6.3264],
        [ 11.2517,  -4.6495,   2.9800],
        [  6.3549,   0.7096,   6.9198],
        [ -8.6426,  -4.9136,   9.4941],
        [  3.6469,   6.3624,  -1.8414],
        [  4.8674,  12.8875,   6.7622],
        [ -6.7745,  -7.7264,   8.4684],
        [ -0.1626, -10.9796,   6.4629],
        [  1.7301,   7.4262,   5.0522],
        [  9.0051,   9.9234,   5.4411],
        [  4.3377,   5.7570,   1.3712],
        [  5.7963,  -1.6018,   1.8856],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.2096,   5.8815,   1.9230],
        [  5.3602,  -2.3808,   7.2098],
        [  2.6970,   6.9826,   9.7350],
        [  2.6078, -10.9229,   6.2869],
        [ -8.8400,   2.3991,   1.3457],
        [  3.7237,  -3.9966,  -5.0376],
        [ -4.2040,   7.2486,   2.9875],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.0449,  13.8402,  -7.1642],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.5506,  -5.1154,   6.3423],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.6319,  11.9066,   3.5165],
        [  3.5284, -10.7484,   5.3347],
        [  5.1886,  -2.6250,   6.4677],
        [  9.3753,   9.5171,   3.9131],
        [  5.2939,  -2.4784,  -4.5840],
        [ -4.2040,   7.2486,   2.9875],
        [  3.3474,  -4.2162,   5.2455],
        [  9.0424,  -7.6779,  -7.5286],
        [ -0.6323,   7.9085,  -5.9544],
        [  3.2366,  -4.2732,   5.1532],
        [  5.4429,  -2.2529,   9.7306],
        [  3.5939,   6.4029,  -1.8357],
        [ -4.6393,   7.0461,   1.1624],
        [  4.5954,  13.0099,  -4.4439],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.1890,   4.7318,   0.9515],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.3549,   0.7096,   6.9198],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.3589,   0.8239,  -8.4230],
        [  4.1348,   5.9514,  -0.8856],
        [ -8.6229,   2.9068,   2.4429],
        [  6.9693,  -9.3200,  -8.3822],
        [ -9.3087,   0.7792,  -7.5587],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.3549,   0.7096,   6.9198],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.4837,   2.7510,   0.4449],
        [  5.5424,  -2.0885,   7.2487],
        [  2.8474,   6.8979,   6.6457],
        [  7.5064,  -8.9651,  -0.0994],
        [  2.0870,  -4.6877,   9.6823],
        [  3.2159, -10.8166,  -1.0406],
        [  1.1006,   7.6318,  -2.4877],
        [  8.2154,  -8.4262,   8.0171],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.4505,  -1.7081,  -0.7039],
        [  1.9241,   7.3503,   7.9004],
        [ 11.0275,  -5.0666,   8.4154],
        [ -9.3076,   0.7852,  -6.7010],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.8902,  -1.3887,   4.0388],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.1929,   2.4501,  -6.9352],
        [  3.9591, -10.6391,  -6.2557],
        [  5.9655,   3.2172,  -7.5953],
        [ 12.2589,  -1.9312,   9.8657],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.8602,   6.8905,   5.8779],
        [ -6.3306,   5.9272,   5.6935],
        [  4.9531,   5.0576,   5.6794],
        [ 10.0619,   8.6627,  -9.2884],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.2804,   2.0019,   0.8607],
        [  2.5010,  -4.5736,  -0.8989]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.3255e-02, 8.6510e-02,  ..., 6.3966e-01, 4.4706e-01,
         3.1692e-01],
        [4.3255e-02, 0.0000e+00, 4.4619e-02,  ..., 6.8240e-01, 4.8665e-01,
         3.5975e-01],
        [8.6510e-02, 4.4619e-02, 0.0000e+00,  ..., 7.2180e-01, 5.2020e-01,
         3.9976e-01],
        ...,
        [6.3966e-01, 6.8240e-01, 7.2180e-01,  ..., 0.0000e+00, 2.4152e-01,
         3.2277e-01],
        [4.4706e-01, 4.8665e-01, 5.2020e-01,  ..., 2.4152e-01, 1.0727e-04,
         1.6807e-01],
        [3.1692e-01, 3.5975e-01, 3.9976e-01,  ..., 3.2277e-01, 1.6807e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.38252937,  8.47371868,  8.53501777,  8.56889738,  8.5724042 ,
        8.62631801,  4.78258092, 13.06957425,  9.23071698,  9.76896904,
        6.5618968 ,  7.62509582, 14.05633441, 10.00802627,  7.72291946,
       10.4606265 , 11.83075005,  6.02588517,  9.84314712, 14.0393949 ,
       12.50996828,  6.74872981, 12.64235359, 12.17451685,  6.39438044,
        9.94173708,  7.33350642, 13.77604293, 10.27572673, 10.98076873,
        7.62509582, 13.40024224,  7.20826219,  6.01356004, 12.9089559 ,
        7.23276582,  5.86519525,  7.48538905, 11.22993218,  9.15976746,
        5.46244988,  8.37954853, 10.71162681, 13.99048094,  8.40445872,
        9.96391045, 10.31886066, 13.62896593, 11.31276109,  5.81481044,
       13.35927527,  5.84533345,  8.37954853,  5.38342077, 11.86239961,
        7.9337647 ,  5.36061974,  5.89074104,  7.34250948,  8.43626114,
       13.79762415, 12.07818871, 12.22503754,  7.02253235, 10.35485331,
        6.39438044,  9.66977802,  6.41202986,  7.24677664,  9.09963894,
       11.63765273,  9.34127013, 10.23946247,  9.51074763,  8.91551231,
        6.39438044,  9.23178204, 12.78327037,  5.92281142,  7.46248947,
       11.69264243,  5.13125513, 11.2845608 ,  7.71075226, 11.76830981,
       12.10942412,  9.60359158,  7.59795445, 12.13568598,  9.34061938,
       13.27717982,  6.05165248, 13.00413829,  6.65991877, 11.35182589,
        6.77777606, 12.41012046,  9.99145096,  7.8788965 ,  7.46053281,
        8.67227386,  7.07902036, 13.27717982, 13.13205357,  6.59176018,
        5.21280156])}
{'P': tensor([[2.0444e-02, 1.6166e-01, 1.5343e-01,  ..., 2.2247e-03, 1.9258e-05,
         1.2420e-05],
        [8.0900e-04, 5.5832e-02, 4.6937e-01,  ..., 8.8144e-03, 6.3231e-05,
         5.8296e-05],
        [5.3809e-04, 2.3886e-03, 8.1556e-02,  ..., 2.8724e-02, 8.8795e-05,
         1.5733e-04],
        ...,
        [5.0272e-05, 8.0936e-05, 1.3501e-04,  ..., 1.0769e-01, 1.0258e-04,
         8.4471e-03],
        [3.8228e-04, 2.3702e-04, 1.4941e-04,  ..., 1.6156e-05, 7.0889e-02,
         2.8499e-05],
        [6.3532e-06, 4.1173e-06, 4.1773e-06,  ..., 7.6024e-06, 7.6736e-04,
         6.3001e-02]]), 'X': tensor([[ -4.3963,   7.1629,   4.5263],
        [ -6.3306,   5.9272,   5.6935],
        [ -7.2515,   5.0195,   4.6434],
        [ -6.4368,   5.8356,   3.3014],
        [ -5.9940,   6.1985,   5.9566],
        [ -6.6895,   5.6051,   2.7899],
        [  1.5653,  13.9377,   4.1773],
        [ -1.9714,   7.8563,  -1.8084],
        [ -1.5451,   7.8975,   7.6104],
        [  1.3403,  -4.8011,  -1.1698],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 11.0373,   7.1299,  -8.5895],
        [  4.1999,  -3.6543,  -4.1998],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.9284,  11.7061,  -0.3732],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.6050,   2.9451,  -0.7012],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.5056,   7.7720,   5.5878],
        [  8.1528,  10.7452,   9.1493],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.4639, -10.9431,  -7.3718],
        [ -7.6650,   4.5092,   0.2520],
        [  5.9883,   3.1536,   6.2320],
        [ -8.0965,  -5.9712,   4.3466],
        [  2.7976,  -4.4682,  -4.2020],
        [ 11.9808,   4.9340,  -3.3715],
        [ -1.6906,   7.8860,   7.7142],
        [  0.5850,  -4.7982,   9.6639],
        [  9.6816,   9.1536,  -5.8543],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.9606,  -4.0994,   1.7886],
        [  4.1140,  -3.7220,  -0.1273],
        [ 12.3453,  -1.5462,   7.0212],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.6078, -10.9229,   6.2869],
        [ 11.9019,  -3.1379,  -3.4534],
        [  6.0073,   3.0986,   6.7937],
        [  1.4150,  13.9647,  -4.0915],
        [  1.6308,  13.9254,  -2.0984],
        [ -9.4159,  -2.0739,   9.6841],
        [  4.5057,  -3.3894,  -0.8160],
        [  1.8447,  -4.7372,   8.1235],
        [  4.7813,  -3.1139,   8.2691],
        [  6.1989,  -9.7591,   0.8289],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.2911,  -2.4824,  -3.6877],
        [  4.6063,  13.0051,  -3.1456],
        [  2.1477,   7.2550,   8.9688],
        [  5.1196,  -2.7160,   4.2705],
        [  4.9152,  12.8653,  -1.4459],
        [  2.1504,  -4.6726,   8.7908],
        [ -9.4501,  -1.7133,   2.2980],
        [ -5.1974,   6.7396,   3.3484],
        [  3.2366,  -4.2732,   5.1532],
        [  3.9085,   6.1512,  -2.2560],
        [  8.2154,  -8.4262,   8.0171],
        [  1.7122,  -4.7591,  -3.6734],
        [ 12.5717,   1.9266,   7.4903],
        [  4.1637,  13.1898,  -1.9189],
        [ -6.2668,   5.9808,   2.5614],
        [  0.9036,   7.6840,   9.9237],
        [  4.6357,   5.4415,   7.9067],
        [  9.5018,  -7.1945,  -1.1892],
        [  5.0617,  12.7957,   8.3544],
        [ 12.4901,  -0.7052,   1.7315],
        [ -6.1914,  -8.2997,   8.5380],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.3858,   4.4260,  -9.6769],
        [  6.3542,   0.6934,  -3.3840],
        [  6.2242,  -0.3171,   7.4978],
        [  2.5813,  -4.5471,   5.3597],
        [  1.6308,  13.9254,  -2.0984],
        [  6.3550,   0.7129,   0.4917],
        [ -8.9636,   2.0648,   8.1946],
        [  2.0870,  -4.6877,   9.6823],
        [  6.3542,   0.6934,  -3.3840],
        [  0.9788,  -4.8145,  -3.8661],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.6613,   7.0021,   4.5783],
        [  2.1477,   7.2550,   8.9688],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -3.6381,   7.4683,   7.3544],
        [ 12.5650,   2.0132,   4.6272],
        [  2.5469,  13.7192,  -9.6935],
        [  6.0054,  -1.0902,   8.3079],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.0961,   1.6529,   3.1553],
        [ -8.0534,  -6.0429,   2.4583],
        [ -6.1914,  -8.2997,   8.5380],
        [  6.2895,   0.0405,   5.7652],
        [ -4.9217,   6.8979,  -1.7667],
        [  2.0870,  -4.6877,   9.6823],
        [  4.4915,   5.5990,   7.0405],
        [  6.2905,   1.9366,   2.1169],
        [ -9.3810,  -2.3455,  -9.5194],
        [  2.4639, -10.9431,  -7.3718],
        [  0.9036,   7.6840,   9.9237],
        [ 11.7162,   5.6728,  -8.9372],
        [ 10.2712,   8.3707,  -4.9743],
        [ -8.0750,  -6.0072,  -5.8267],
        [  4.4915,   5.5990,   7.0405],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.5101,  -2.1433,   9.5442],
        [  0.8940, -11.0402,  -0.9783]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1251, 0.1815,  ..., 0.6639, 0.4903, 0.5062],
        [0.1251, 0.0000, 0.1087,  ..., 0.6341, 0.5784, 0.5345],
        [0.1815, 0.1087, 0.0000,  ..., 0.5256, 0.6677, 0.4453],
        ...,
        [0.6639, 0.6341, 0.5256,  ..., 0.0000, 1.1359, 0.2994],
        [0.4903, 0.5784, 0.6677,  ..., 1.1359, 0.0000, 0.9172],
        [0.5062, 0.5345, 0.4453,  ..., 0.2994, 0.9172, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.40445872,  8.67227386,  8.81930525,  8.68832398,  8.62260327,
        8.7273569 , 14.02532758,  8.09984231,  8.04718433,  4.98461983,
       10.53368644, 13.1399181 ,  5.5671389 ,  9.51074763, 13.60275211,
        8.41148722,  9.09502385, 13.86480175,  8.16584537, 13.48809595,
        8.35856487, 11.21703654,  8.89301516,  6.76791483, 10.06023665,
        5.27178363, 12.95702647,  8.06517143,  4.83371341, 13.32374056,
        6.05113217,  9.85384623,  5.54776934, 12.44177641, 10.71162681,
       11.22993218, 12.30858677,  6.7593917 , 14.03618649, 14.02058621,
        9.64157461,  5.63824971,  5.08372149,  5.70591986, 11.56146873,
       11.54672755,  5.84450127, 13.79676654,  7.56617037,  5.79543038,
       13.77222752,  5.14371005,  9.60412735,  8.51088913,  5.36061974,
        7.28793663, 11.76830981,  5.05774295, 12.71843462, 13.83139024,
        8.66273171,  7.73692436,  7.14837514, 11.9182956 , 13.76047999,
       12.50996828, 10.35464916, 12.25806559,  6.97106912,  6.39187285,
        6.2322769 ,  5.2287058 , 14.02058621,  6.3949043 ,  9.19837792,
        5.13125513,  6.39187285,  4.91296205,  5.19065439,  7.49076899,
        7.56617037,  9.93553424,  8.30729352, 12.72524615, 13.95361098,
        6.10360073, 11.80092578,  9.24501829, 10.06850677, 10.35464916,
        6.28961988,  8.47371868,  5.13125513,  7.17789666,  6.58184474,
        9.66977802, 11.21703654,  7.73692436, 13.01728313, 13.25017211,
       10.06437598,  7.17789666, 10.65929968,  9.62958973,  5.91222346,
       11.07637531])}
{'P': tensor([[1.1676e-02, 2.7863e-01, 1.4749e-01,  ..., 6.7313e-04, 4.2848e-06,
         6.6120e-06],
        [2.8500e-04, 1.3870e-02, 2.6020e-01,  ..., 1.0941e-03, 5.3864e-06,
         7.2389e-06],
        [3.1743e-04, 4.3391e-04, 2.9612e-02,  ..., 3.9623e-03, 1.3135e-05,
         1.3929e-05],
        ...,
        [8.6092e-06, 1.0310e-05, 1.5218e-05,  ..., 2.5523e-02, 2.4120e-04,
         3.4023e-05],
        [6.2808e-05, 5.4001e-05, 4.2011e-05,  ..., 6.0699e-05, 9.7885e-02,
         2.5345e-02],
        [6.1722e-04, 4.7561e-04, 2.9860e-04,  ..., 5.1157e-05, 4.9495e-04,
         5.4725e-02]]), 'X': tensor([[ -4.4503,   7.1378,   3.3751],
        [ -5.1974,   6.7396,   3.3484],
        [ -6.4368,   5.8356,   3.3014],
        [ -6.2668,   5.9808,   2.5614],
        [ -6.6895,   5.6051,   2.7899],
        [ -4.7389,   6.9953,   3.7638],
        [  6.2902,   0.0451,   3.7116],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 10.6283,  -5.7253,   7.4878],
        [  2.3200,   7.1756,   1.1938],
        [  6.2022,   2.4092,   3.0349],
        [ -8.3539,  -5.5111,   0.3373],
        [  3.3633,  13.4794,   0.5193],
        [ -6.6895,   5.6051,   2.7899],
        [ -8.0534,  -6.0429,   2.4583],
        [ 11.1420,   6.9331,  -2.4779],
        [  0.1041, -11.0045,   2.4311],
        [ 12.2328,  -2.0376,  -1.1234],
        [  3.9913,  -3.8139,  -6.8311],
        [  5.4867,  12.5816,   2.9549],
        [  5.9548,  -1.2273,  -2.6520],
        [ -9.0604,   1.7707,   1.1066],
        [ -8.4267,  -5.3697,   2.4172],
        [ -9.4630,  -1.5145,   0.2983],
        [ -4.0868,   7.2981,   8.1874],
        [  4.8051, -10.3703,   0.5268],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.9282,   2.1648,   8.4894],
        [  6.2576,   2.1359,  -3.3398],
        [  7.3724,  -9.0577,  -8.9843],
        [ -3.9508,  -9.8515,   1.7923],
        [  1.7228,  -4.7575,   8.3396],
        [  8.2154,  -8.4262,   8.0171],
        [  2.7121,   6.9743,  -8.4202],
        [ -8.0534,  -6.0429,   2.4583],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -3.5520,   7.4976,   9.1522],
        [  6.3583,   1.1729,   8.7835],
        [ -7.4887,   4.7364,   8.9074],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -5.6169,  -8.7844,   8.9057],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.7544,   2.6096,   3.3146],
        [  4.0457,  -3.7737,  -0.4704],
        [  7.3724,  -9.0577,  -8.9843],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.9883,   3.1536,   6.2320],
        [  3.0876,   6.7527,  -5.3540],
        [  8.7131,  10.2216,   7.7915],
        [  3.4140,  -4.1803,  -8.8224],
        [ 12.2435,   4.0076,   8.1257],
        [  2.5010,  -4.5736,  -0.8989],
        [  1.3339,   7.5627,  -0.9034],
        [  4.8653,   5.1694,  -4.0823],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.6954,  -4.7616,   9.8198],
        [  9.3607,   9.5337,  -4.5412],
        [ 12.3025,  -1.7440,  -6.6632],
        [  2.5375,   7.0676,  -4.3490],
        [ -6.2200,  -8.2737,   5.5079],
        [  0.5175,  14.0910,   4.6540],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -1.5261,   7.8988,   9.3768],
        [  6.1984,  -0.4327,   1.5438],
        [ -4.8751,  -9.3148,   4.0406],
        [  6.2630,   2.1057,  -5.2136],
        [  2.5620,   7.0549,   8.4825],
        [  4.6063,  13.0051,  -3.1456],
        [  3.7237,  -3.9966,  -5.0376],
        [ 12.6038,   0.7665,  -3.4439],
        [ -5.1046,  -9.1610,   2.3418],
        [  0.3715,  -4.7756,   0.2312],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 12.3609,   3.4769,  -8.5762],
        [ 11.4743,   6.2478,  -2.1100],
        [ -9.1658,  -3.4155,  -2.8880],
        [  1.1350,  14.0104,   0.6321],
        [  8.3791,  10.5409,   5.5590],
        [  4.9337,  12.8566,  -3.6582],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.1459,   1.4780,  -4.2403],
        [ 10.0995,  -6.4739,   2.0974],
        [  5.7963,  -1.6018,   1.8856],
        [ -2.6735, -10.4062,  -1.5911],
        [ -2.3317,   7.8034,  -3.6274],
        [  1.9204,   7.3518,   8.4781],
        [  8.8474,  10.0868,  -8.8644],
        [ -3.9456,  -9.8542,   0.7466],
        [  9.6816,   9.1536,  -5.8543],
        [ 12.3357,   3.6001,   0.7903],
        [  6.5428,  -9.5728,  -2.9897],
        [  1.7228,  -4.7575,   8.3396],
        [  6.2902,   0.0451,   3.7116],
        [  9.4925,   9.3811,   8.9842],
        [  4.7156,  -3.1831,  -8.0224],
        [  2.1625,  -4.6697,   7.0658],
        [ -9.4541,  -0.3327,   8.8246],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.3107,  -2.7697,  -3.2919],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.3587,   1.1596,  -3.6629]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0485, 0.1270,  ..., 0.7818, 0.8051, 0.6815],
        [0.0485, 0.0000, 0.0785,  ..., 0.7458, 0.8283, 0.7107],
        [0.1270, 0.0785, 0.0000,  ..., 0.6867, 0.8651, 0.7576],
        ...,
        [0.7818, 0.7458, 0.6867,  ..., 0.0000, 0.9769, 0.9854],
        [0.8051, 0.8283, 0.8651,  ..., 0.9769, 0.0000, 0.1644],
        [0.6815, 0.7107, 0.7576,  ..., 0.9854, 0.1644, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.41148722,  8.51088913,  8.68832398,  8.66273171,  8.7273569 ,
        8.44940661,  6.29034958, 10.39942497,  9.96152788, 12.07225469,
        7.54127425,  6.65368551, 10.00794468, 13.89264822,  8.7273569 ,
       10.06850677, 13.12299145, 11.0050379 , 12.4013153 ,  5.5205189 ,
       13.72594022,  6.07992623,  9.23178204,  9.99211663,  9.58347456,
        8.36445758, 11.42947632, 13.25336001,  9.18690018,  6.61211224,
       11.67875855, 10.61417181,  5.05981914, 11.76830981,  7.48311002,
       10.06850677, 12.53345445,  8.29640613,  6.46560181,  8.86082124,
        5.27178363, 12.35421461, 10.42667093, 10.8992524 ,  9.13507321,
        5.53255882, 11.67875855, 11.63724792,  6.76791483,  7.42512621,
       13.43127131,  5.39723241, 12.88270018,  5.21280156,  7.67939546,
        7.09887557, 13.20988996,  5.05444288, 13.36092624, 12.42554808,
        7.5092834 , 10.35093485, 14.10045797, 12.4217523 ,  8.04483495,
        6.2134961 , 10.51340829,  6.60752373,  7.50564223, 13.79676654,
        5.46244988, 12.62711252, 10.48721382,  4.79003361, 11.785566  ,
        9.5642993 , 12.84056604, 13.06498554,  9.78147542, 14.05633441,
       13.46553623, 13.77075174,  9.93553424, 12.52372629,  8.92098752,
        9.26455795, 11.99633819,  6.01356004, 10.74409834,  8.14433542,
        7.5984809 , 13.41713661, 10.61472636, 13.32374056, 12.85032836,
       11.59512194,  5.05981914,  6.29034958, 13.34587027,  5.68942572,
        5.14607014,  9.45995955,  9.10945678,  9.71392179,  5.53175275,
        6.46356806])}
{'P': tensor([[2.5040e-02, 3.2680e-01, 3.0982e-01,  ..., 2.6613e-06, 2.9056e-06,
         1.3680e-04],
        [1.0594e-03, 2.7628e-02, 5.7480e-01,  ..., 2.4705e-06, 2.8250e-06,
         1.0850e-04],
        [9.8044e-04, 1.1219e-03, 6.0728e-02,  ..., 4.8821e-06, 6.0535e-06,
         1.4892e-04],
        ...,
        [1.5503e-04, 1.1361e-04, 8.3459e-05,  ..., 2.0719e-01, 5.2486e-03,
         2.3946e-03],
        [2.0431e-05, 1.5424e-05, 1.1820e-05,  ..., 1.9806e-01, 6.4017e-02,
         2.6534e-04],
        [1.0422e-01, 7.4592e-02, 5.4340e-02,  ..., 9.1375e-05, 6.6726e-05,
         1.4680e-01]]), 'X': tensor([[ -4.4520,   7.1370,  -8.7867],
        [ -5.0550,   6.8231,  -9.2502],
        [ -5.9872,   6.2037,  -9.4477],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.8280,   4.2848,  -7.1955],
        [ -8.6112,   2.9319,  -7.9943],
        [  7.9141,  -8.6656,   9.6948],
        [ 11.1469,   6.9237,   1.6160],
        [  1.9505,  -4.7171,   9.9868],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.7963,  -1.6018,   1.8856],
        [  6.3604,   0.9042,  -6.5901],
        [ -7.2515,   5.0195,   4.6434],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -8.4573,   3.2448,   5.4370],
        [ 11.8951,  -3.1570,   7.0742],
        [ 10.5955,   7.8824,  -3.7006],
        [  4.9337,  12.8566,  -3.6582],
        [  6.8704,  -9.3809,   7.5769],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -3.5029,   7.5138,   8.1853],
        [  6.9956,  11.6590,  -7.6876],
        [ 12.2328,  -2.0376,  -1.1234],
        [  3.7298,  13.3537,  -9.4653],
        [  0.6363, -11.0348,   6.1414],
        [ -6.0844,  -8.3955,   0.7705],
        [ -3.9456,  -9.8542,   0.7466],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.6231,   5.4556,   8.4973],
        [  5.7845,   3.6652,   0.9175],
        [  2.2860,   7.1916,  -6.1011],
        [ -2.5056,   7.7720,   5.5878],
        [ -9.1907,  -3.3180,  -5.5943],
        [  9.3753,   9.5171,   3.9131],
        [  1.4306,  13.9620,  -0.2075],
        [  3.3943,   6.5483,   1.0790],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.2517,  -4.6495,   2.9800],
        [ -8.7544,   2.6096,   3.3146],
        [  6.7792,  11.8084,   1.6251],
        [  2.8890,  -4.4316,   1.1600],
        [  6.3534,   1.2990,  -7.2942],
        [  6.1685,  12.1979,  -6.2473],
        [  6.0467,  -9.8369,  -6.5816],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.4489,   7.5256,   4.4529],
        [  6.2790,  -0.0259,  -5.4116],
        [ -2.6069, -10.4298,  -6.1995],
        [  4.2339,  -3.6268,   6.3125],
        [ -5.3179,  -9.0101,   9.4472],
        [ -1.3940, -10.7772,   6.7835],
        [  2.7295,  13.6703,  -8.9328],
        [  2.3589,  13.7668,  -3.6501],
        [  7.0358,  -9.2784,  -9.8479],
        [  7.8425,  -8.7202,   6.4530],
        [  6.0788,  -0.8710,  -6.8719],
        [  1.7659,   7.4127,   4.7871],
        [  1.7903,  -4.7467,   0.1168],
        [ -1.9646,   7.8571,   3.3817],
        [  0.1382, -11.0073,   1.0026],
        [ -6.4368,   5.8356,   3.3014],
        [  3.2159, -10.8166,  -1.0406],
        [ 12.4877,  -0.7222,   8.8347],
        [  3.1480,  -4.3165,  -7.8002],
        [ -2.0229, -10.6163,   8.6930],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.3739,   0.3982,  -4.2564],
        [ -6.8618,  -7.6322,   3.7027],
        [  3.3633,  13.4794,   0.5193],
        [ -9.2345,  -3.1345,   3.9715],
        [  1.9241,   7.3503,   7.9004],
        [  5.1861,   4.7361,  -3.7312],
        [  4.2339,  -3.6268,   6.3125],
        [ -2.1382,   7.8339,   7.3524],
        [ -2.8162,   7.7059,  -9.6885],
        [ -7.6525,  -6.6534,   5.1721],
        [ -8.9942,  -3.9990,   6.0448],
        [ 12.4690,  -0.8528,  -9.3374],
        [  9.0424,  -7.6779,  -7.5286],
        [  8.3718,  -8.2953,   8.4666],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.8889,  -1.3916,   8.8993],
        [  5.1196,  -2.7160,   4.2705],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.1984,  -0.4327,   1.5438],
        [  0.6933,   7.7336,   2.1179],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.6078, -10.9229,   6.2869],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 12.5650,   2.0132,   4.6272],
        [ -4.7389,   6.9953,   3.7638],
        [  6.3122,  -9.6994,  -6.3282],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.8323,  -1.5228,   5.1132],
        [ 12.3759,  -1.3943,   6.4659],
        [ 10.7566,   7.6212,  -2.3279],
        [  1.0000,   7.6591,  -4.5011],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.2895,   0.0405,   5.7652],
        [  5.1203,  -2.7151,   6.2110],
        [  0.4996,   7.7738,  -4.4711]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.2871e-02, 1.2579e-01,  ..., 1.0280e+00, 1.0376e+00,
         3.2117e-01],
        [4.2871e-02, 2.1453e-04, 8.3381e-02,  ..., 1.0539e+00, 1.0627e+00,
         3.5827e-01],
        [1.2579e-01, 8.3381e-02, 0.0000e+00,  ..., 1.0952e+00, 1.1023e+00,
         4.2656e-01],
        ...,
        [1.0280e+00, 1.0539e+00, 1.0952e+00,  ..., 0.0000e+00, 2.6363e-02,
         7.3950e-01],
        [1.0376e+00, 1.0627e+00, 1.1023e+00,  ..., 2.6363e-02, 0.0000e+00,
         7.5304e-01],
        [3.2117e-01, 3.5827e-01, 4.2656e-01,  ..., 7.3950e-01, 7.5304e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.41171369,  8.49161154,  8.62162372,  8.83341608,  8.92395381,
        9.09661212, 11.73567759, 13.12218013,  5.10446653,  6.1561646 ,
       12.30858677, 12.5085995 ,  6.01356004,  6.42439344,  8.81930525,
       12.53345445,  9.05843201, 12.30695098, 13.20597932, 13.77075174,
       11.62770332,  9.75401449,  8.29022197, 13.59673406, 12.4013153 ,
       13.86480175, 11.05317597, 10.36844813, 10.61472636, 10.4606265 ,
        7.15100303,  6.84796202,  7.54620849,  8.16584537,  9.7712321 ,
       13.35927527, 14.03505624,  7.37576154,  9.58072286,  6.54615887,
       12.17451685,  9.13507321, 13.61600536,  5.29009836,  6.48486171,
       13.66896739, 11.54672755,  8.30125201,  7.66377554,  6.27905292,
       10.7506423 ,  5.5748624 , 10.46238054, 10.86693708, 13.94008948,
       13.96746816, 11.64436684, 11.72802491,  6.14086582,  7.62011734,
        5.07306274,  8.09899874, 11.00812996,  8.68832398, 11.2845608 ,
       12.5085995 ,  5.34250225, 10.80728788,  6.06416957,  9.38232121,
       10.26327895, 13.89264822,  9.75200848,  7.59795445,  7.02326515,
        5.5748624 ,  8.12043983,  8.20436764, 10.14044489,  9.84314712,
       12.4980845 , 11.86239961, 11.785566  ,  8.47371868,  6.05113217,
        5.79543038,  5.39450305,  5.91854397,  9.61942445,  6.2134961 ,
        7.76457095,  9.49511524, 11.22993218, 12.10942412, 12.72524615,
        8.44940661, 11.57249984,  9.73579738,  6.02779201, 12.45418236,
       13.18278606,  7.72415048, 12.11735334,  6.28961988,  5.79562691,
        7.78980299])}
{'P': tensor([[1.5614e-02, 3.0912e-01, 1.2840e-01,  ..., 6.9609e-07, 1.4556e-05,
         9.8583e-06],
        [3.1434e-04, 1.7220e-02, 1.0617e-01,  ..., 7.5591e-07, 1.6155e-05,
         5.7162e-06],
        [4.1083e-04, 1.1586e-03, 3.3925e-02,  ..., 1.3363e-06, 2.8262e-05,
         5.7263e-06],
        ...,
        [3.0545e-05, 1.8756e-05, 9.1871e-06,  ..., 2.1871e-02, 8.1913e-04,
         3.7635e-04],
        [7.0117e-04, 4.9886e-04, 2.8388e-04,  ..., 9.3581e-03, 8.9301e-02,
         1.0439e-03],
        [7.7462e-02, 6.1462e-02, 3.0243e-02,  ..., 1.2910e-06, 2.3785e-05,
         1.7882e-01]]), 'X': tensor([[ -4.6393,   7.0461,   1.1624],
        [ -5.6651,   6.4380,   1.3660],
        [ -6.6998,   5.5954,   0.0823],
        [ -6.2668,   5.9808,   2.5614],
        [ -7.4808,   4.7462,   2.0266],
        [ -7.4807,   4.7463,   2.0616],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.6022,  -1.9833,   8.3081],
        [  5.8930,  12.3592,  -8.9201],
        [ 11.9808,   4.9340,  -3.3715],
        [  9.6614,   9.1784,  -1.2087],
        [  9.2692,   9.6370,   7.0746],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 11.6544,   5.8273,  -3.1375],
        [ 12.2435,   4.0076,   8.1257],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.2911,  -2.4824,  -3.6877],
        [ -1.6275, -10.7221,   3.2376],
        [ -3.1671,   7.6158,   6.0765],
        [  2.3261,   7.1726,   6.2522],
        [ -9.4753,  -0.8012,  -5.2050],
        [  5.7963,  -1.6018,   1.8856],
        [  5.8836,   3.4315,  -4.0327],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.5663,   7.0526,  -6.8769],
        [ -7.4808,   4.7462,   2.0266],
        [  3.9663,   6.1018,  -5.4104],
        [  2.6970,   6.9826,   9.7350],
        [  2.0547,   7.2957,  -1.9343],
        [ -7.8359,   4.2735,   7.2943],
        [  5.8318,   3.5567,  -4.7692],
        [  2.4572,   7.1085,   7.7376],
        [ -1.6220,   7.8917,   0.0315],
        [ -9.4159,  -2.0739,   9.6841],
        [  4.6063,  13.0051,  -3.1456],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.3900,   3.3727,   3.1650],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.3999, -10.1152,  -1.2088],
        [ -5.1398,  -9.1367,   6.2074],
        [ 12.5650,   2.0132,   4.6272],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.9653,  -9.3225,  -3.4727],
        [  6.3542,   0.6934,  -3.3840],
        [  7.5064,  -8.9651,  -0.0994],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.2535,  -0.1709,  -9.4102],
        [ 11.2571,   6.7072,   8.3693],
        [  1.8719,   7.3713,  -3.7465],
        [  9.4554,   9.4245,   9.8800],
        [  4.3459,  -3.5327,   0.3667],
        [  2.1504,  -4.6726,   8.7908],
        [  2.3200,   7.1756,   1.1938],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -5.7347,  -8.6908,   9.2878],
        [ 12.2592,   3.9425,   2.0383],
        [  0.9945,  -4.8144,  -0.1897],
        [  4.5800,   5.5034,  -3.9432],
        [  4.2339,  -3.6268,   6.3125],
        [  4.3377,   5.7570,   1.3712],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.1861,   4.7361,  -3.7312],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.5198,  -4.0923,   5.5088],
        [ -2.5056,   7.7720,   5.5878],
        [  7.0515,  11.6194,   8.4289],
        [  6.1989,  -9.7591,   0.8289],
        [ -6.1914,  -8.2997,   8.5380],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -3.5173,   7.5091,  -8.9585],
        [ -4.2271,   7.2387,  -1.8316],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.9655,   3.2172,  -7.5953],
        [  1.3670,  -4.7990,   3.2603],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 12.5258,  -0.4200,  -9.3493],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.2905,   1.9366,   2.1169],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.5101,  -2.1433,   9.5442],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.8400,   2.3991,   1.3457],
        [  5.6248,   3.9984,   8.4933],
        [ -6.3489,   5.9117,   6.5161],
        [  6.2895,   0.0405,   5.7652],
        [  9.4925,   9.3811,   8.9842],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.6050,   2.9451,  -0.7012],
        [ 11.6019,   5.9540,  -5.1715],
        [  0.3715,  -4.7756,   0.2312],
        [ -8.9636,   2.0648,   8.1946],
        [  4.0362,  13.2397,  -7.0030],
        [  1.3670,  -4.7990,   3.2603],
        [ -2.3317,   7.8034,  -3.6274],
        [  8.9284,  10.0036,  -6.9210],
        [ -9.4630,  -1.5145,   0.2983],
        [ 11.7796,  -3.4696,   9.0088],
        [ 12.3759,  -1.3943,   6.4659],
        [  5.9193,  -1.3176,   5.1739],
        [  1.5653,  13.9377,   4.1773]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0705, 0.1662,  ..., 0.7430, 0.5487, 0.8311],
        [0.0705, 0.0000, 0.1369,  ..., 0.7802, 0.6010, 0.8823],
        [0.1662, 0.1369, 0.0000,  ..., 0.9083, 0.7132, 0.9957],
        ...,
        [0.7430, 0.7802, 0.9083,  ..., 0.0000, 0.2633, 0.2289],
        [0.5487, 0.6010, 0.7132,  ..., 0.2633, 0.0000, 0.2825],
        [0.8311, 0.8823, 0.9957,  ..., 0.2289, 0.2825, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.43626114,  8.57560275,  8.72896381,  8.66273171,  8.859416  ,
        8.85939626,  9.85330646, 12.91143491,  6.10774544,  5.48276038,
        5.94293185, 13.6922425 , 12.95702647, 13.32613445, 13.37121923,
        9.60359158, 13.03002338, 12.88270018,  8.25828409,  5.84450127,
       10.84493732,  8.24808832,  7.5403822 ,  9.50913849,  6.01356004,
        6.81117802,  8.20436764,  7.50500064,  8.859416  ,  7.27759233,
        7.48538905,  7.57946459,  8.92549752,  6.83083863,  7.52117132,
        8.05668609,  9.64157461, 13.79676654, 11.83509372,  9.04255655,
       10.48721382, 10.67132524, 10.48315436, 12.72524615,  5.43544626,
       11.63724792,  6.39187285, 11.69264243, 12.76408562,  6.00732193,
        6.25586297, 13.103709  ,  7.60529892, 13.35013374,  5.60064319,
        5.14371005,  7.54127425,  9.50913849, 10.412299  , 12.87752052,
        4.9160962 ,  7.15990676,  5.5748624 ,  7.20826219,  4.91296205,
        7.02326515,  5.8828986 , 12.22503754,  8.16584537, 13.59170809,
       11.56146873, 10.35464916,  8.53014915,  9.562174  ,  8.29203356,
        8.38252937, 11.75618513,  6.77777606,  4.98989906, 12.27835996,
       12.53285633,  4.8763135 ,  6.58184474,  5.57341623,  5.91222346,
       12.04924159,  9.7712321 ,  9.15976746,  6.90115901,  8.67502348,
        6.28961988, 13.34587027,  9.0368458 ,  9.09502385, 13.04051744,
        4.79003361,  9.19837792, 13.84126197,  4.98989906,  8.14433542,
       13.40850328,  9.58347456, 12.27993098, 12.45418236,  6.06416957,
       14.02532758])}
{'P': tensor([[1.3824e-02, 3.9217e-01, 1.5758e-01,  ..., 6.6240e-02, 6.4726e-06,
         8.5103e-07],
        [5.1587e-04, 2.1583e-02, 2.9107e-01,  ..., 1.2259e-01, 9.7124e-06,
         1.1523e-06],
        [2.7528e-04, 3.3810e-04, 2.5095e-02,  ..., 1.0347e-01, 9.1447e-06,
         8.4882e-07],
        ...,
        [8.9210e-04, 1.0780e-03, 4.6309e-03,  ..., 1.5280e-01, 7.3724e-05,
         2.9679e-06],
        [8.7260e-04, 7.6066e-04, 3.7506e-04,  ..., 3.5555e-04, 1.2333e-01,
         1.9483e-04],
        [5.9077e-04, 4.9100e-04, 2.4316e-04,  ..., 7.6490e-05, 4.2164e-06,
         3.1285e-02]]), 'X': tensor([[-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[7.5848e-05, 4.4166e-02, 1.5181e-01,  ..., 2.7306e-01, 5.8760e-01,
         1.0946e+00],
        [4.4166e-02, 0.0000e+00, 1.0834e-01,  ..., 2.3377e-01, 6.2245e-01,
         1.1260e+00],
        [1.5181e-01, 1.0834e-01, 0.0000e+00,  ..., 1.6889e-01, 7.2239e-01,
         1.1901e+00],
        ...,
        [2.7306e-01, 2.3377e-01, 1.6889e-01,  ..., 0.0000e+00, 7.4803e-01,
         1.3538e+00],
        [5.8760e-01, 6.2245e-01, 7.2239e-01,  ..., 7.4803e-01, 0.0000e+00,
         1.1930e+00],
        [1.0946e+00, 1.1260e+00, 1.1901e+00,  ..., 1.3538e+00, 1.1930e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.4486771 ,  8.50278896,  8.72896381,  8.5724042 ,  8.83725081,
        8.89301516, 13.58333857,  6.60752373, 11.33837396,  8.95983586,
       13.2430377 ,  5.81199422, 11.72802491, 11.62770332, 10.06850677,
        6.56833864,  9.79039371, 11.63724792, 11.72802491,  7.46053281,
        5.29009836, 11.89855267, 11.65923396, 12.3828957 , 10.53368644,
       10.66631784, 12.05300943,  5.70591986, 13.41713661, 14.03618649,
       12.07818871, 10.98076873, 10.06023665, 10.06286074, 12.22276278,
       11.67875855,  6.34059021,  9.14044984,  5.13616224,  7.5092834 ,
       11.74901778, 11.50791859, 13.91198191, 11.10066126, 11.00812996,
       12.13568598,  6.81084724, 10.67132524, 13.97779839,  9.18358562,
        7.01475376,  5.21280156,  9.60080639,  5.5671389 , 12.05395056,
        6.29821673,  8.09899874,  7.78980299, 10.37698435,  6.38399789,
       13.20597932, 12.64235359, 12.26997165,  5.90035324, 14.10045797,
        6.383897  , 10.75443106,  5.19065439, 13.66051884, 13.04217487,
        8.11227668,  4.89869205, 10.04916678, 13.89974693,  6.2322769 ,
       10.74655346,  9.85330646, 10.1371552 ,  5.56809335,  5.29755487,
       13.70893492, 13.99048094,  9.86089579,  7.46053281, 10.66631784,
        6.35029274, 11.85890664,  6.84796202, 10.75443106,  9.77777346,
        8.81930525,  9.44486609, 13.55868684, 13.26209545, 13.02818702,
        7.59795445,  5.79391825, 13.40024224,  5.87305005, 10.61472636,
        6.41202986, 11.92133129, 10.06703358,  8.91551231,  6.41202986,
       13.36800073])}
{'P': tensor([[1.5047e-02, 1.9884e-01, 1.8241e-01,  ..., 2.1652e-04, 3.5518e-06,
         1.0454e-05],
        [3.2016e-04, 2.4916e-02, 3.6511e-04,  ..., 8.5633e-04, 5.8500e-06,
         1.2361e-05],
        [6.7600e-04, 2.7002e-01, 1.4393e-02,  ..., 2.5192e-04, 3.6022e-06,
         9.5042e-06],
        ...,
        [1.3361e-05, 1.4809e-05, 1.3411e-05,  ..., 8.4237e-02, 5.1683e-05,
         2.9327e-05],
        [3.0340e-04, 2.0636e-04, 3.0209e-04,  ..., 1.0139e-05, 1.3495e-01,
         8.1788e-02],
        [1.2131e-03, 7.3937e-04, 1.1878e-03,  ..., 6.8519e-06, 3.9416e-04,
         3.5043e-02]]), 'X': tensor([[ -4.7389,   6.9953,   3.7638],
        [ -6.4368,   5.8356,   3.3014],
        [ -5.1974,   6.7396,   3.3484],
        [ -6.6895,   5.6051,   2.7899],
        [ -6.2668,   5.9808,   2.5614],
        [ -7.2515,   5.0195,   4.6434],
        [  9.3753,   9.5171,   3.9131],
        [  2.0547,   7.2957,  -1.9343],
        [  9.1620,  -7.5573,  -9.9882],
        [ 10.0567,   8.6697,   4.6329],
        [  3.9663,   6.1018,  -5.4104],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.4573,   3.2448,   5.4370],
        [  5.1886,  -2.6250,   6.4677],
        [  6.2555,   2.1474,   4.1685],
        [  9.0424,  -7.6779,  -7.5286],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.9019,  -3.1379,  -3.4534],
        [  2.5620,   7.0549,   8.4825],
        [ -9.3464,   0.5708,  -1.6397],
        [  0.1931,  -4.7491,   7.4385],
        [  5.6314,  -1.9302,  -9.6543],
        [  1.7218,  -4.7576,  -3.3617],
        [  8.1528,  10.7452,   9.1493],
        [  5.6093,  -1.9705,   6.4807],
        [  1.3103,  -4.8032,  -0.9300],
        [ -8.1531,  -5.8748,   2.6934],
        [  5.6202,  -1.9508,   0.7471],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -3.3116, -10.1539,   7.5919],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.8154,   2.4611,   2.5198],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.8602,   6.8905,   5.8779],
        [ -1.6906,   7.8860,   7.7142],
        [ -9.1813,  -3.3550,   1.8827],
        [  2.8602,   6.8905,   5.8779],
        [ 12.4901,  -0.7052,   1.7315],
        [ -1.6220,   7.8917,   0.0315],
        [  7.7750,  11.0665,   2.8364],
        [  4.5800,   5.5034,  -3.9432],
        [ -8.3535,  -5.5118,   9.8399],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -2.1817,   7.8274,  -4.6298],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.2902,   0.0451,   3.7116],
        [ -4.3359,  -9.6431,  -9.0616],
        [  0.4996,   7.7738,  -4.4711],
        [  5.6095,  -1.9702,  -7.8233],
        [  1.0000,   7.6591,  -4.5011],
        [  5.8836,   3.4315,  -4.0327],
        [  5.6672,  -1.8631,   8.1020],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.3464,   0.5708,  -1.6397],
        [  9.3607,   9.5337,  -4.5412],
        [  3.2701,   6.6338,   9.9209],
        [  3.5939,   6.4029,  -1.8357],
        [ -9.4309,  -0.0599,  -4.0830],
        [  4.0587, -10.6112,   6.2273],
        [ -7.8849,  -6.3115,  -4.7479],
        [  3.9895,   6.0816,  -6.5235],
        [  3.6154,  -4.0640,  -7.9698],
        [  8.2422,  -8.4041,  -1.4962],
        [ -1.9600,   7.8577,   6.9121],
        [ -4.1296,   7.2803,  -2.7194],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.4401,  13.4540,   4.6576],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.6651,   6.4380,   1.3660],
        [  2.5128, -10.9365,   9.9645],
        [  1.9503,  -4.7172,  -6.2313],
        [  1.3705,  13.9723,   7.9266],
        [  0.7275,  14.0667,  -0.9815],
        [  7.1821,  -9.1845,   9.8548],
        [  1.0093,   7.6567,   2.3526],
        [  4.6231,   5.4556,   8.4973],
        [ -5.6169,  -8.7844,   8.9057],
        [ -8.1905,   3.7253,   4.1175],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -8.6603,   2.8250,   5.9181],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.0310,  12.2795,  -6.4528],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.9503,  -4.7172,  -6.2313],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.9379,  -4.1655,  -2.8114],
        [  1.5383, -11.0274,  -4.9666],
        [  3.6249, -10.7255,   5.6605],
        [ -8.5859,  -5.0393,   0.5476],
        [ -7.5401,   4.6718,   9.0033],
        [ 11.9808,   4.9340,  -3.3715],
        [  5.8318,   3.5567,  -4.7692],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.1426,   3.8048,   9.3542],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.9967,  -4.8144,  -9.7370],
        [ -6.2200,  -8.2737,   5.5079],
        [  6.3301,   0.3638,  -7.5685],
        [  5.1861,   4.7361,  -3.7312]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1116, 0.0392,  ..., 0.4593, 0.8411, 0.6982],
        [0.1116, 0.0000, 0.0785,  ..., 0.3767, 0.8716, 0.7363],
        [0.0392, 0.0785, 0.0000,  ..., 0.4435, 0.8322, 0.6919],
        ...,
        [0.4593, 0.3767, 0.4435,  ..., 0.0000, 1.2273, 1.1004],
        [0.8411, 0.8716, 0.8322,  ..., 1.2273, 0.0000, 0.1504],
        [0.6982, 0.7363, 0.6919,  ..., 1.1004, 0.1504, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.44940661,  8.68832398,  8.51088913,  8.7273569 ,  8.66273171,
        8.81930525, 13.35927527,  7.57946459, 11.87665501, 13.2778378 ,
        7.27759233, 13.95361098,  9.05843201,  5.81481044,  6.61386436,
       11.86239961, 12.04924159, 12.30858677,  7.50564223,  9.36378438,
        4.75303127,  5.95297459,  5.05963217, 13.48809595,  5.94535529,
        4.978704  , 10.04916678,  5.94909709, 12.42554808, 10.68031351,
       12.3828957 , 10.92597196,  9.15253344, 12.16901615,  7.46053281,
        8.06517143,  9.7751189 ,  7.46053281, 12.50996828,  8.05668609,
       13.5247215 ,  7.15990676, 10.00802627, 10.51437947,  8.12580976,
       13.79762415,  9.64157461,  6.29034958, 10.57302546,  7.78980299,
        5.94541803,  7.72415048,  6.81117802,  5.96556248,  9.69629   ,
        9.41970031,  9.36378438, 13.36092624,  7.39599986,  7.34250948,
        9.43112957, 11.36089355, 10.09979747,  7.27340834,  5.43943791,
       11.77124373,  8.09842704,  8.36995953, 11.9120321 , 13.88683436,
       12.53345445, 10.51437947,  8.57560275, 11.22142195,  5.10444097,
       14.0393949 , 14.08549802, 11.65923396,  7.72291946,  7.15100303,
       10.42667093,  8.99791316,  9.79039371,  9.58072286,  9.10945678,
       12.05395056, 12.35421461,  9.70108409, 13.68063098,  9.55593374,
        5.10444097, 12.30695098,  9.86089579, 11.1341798 , 11.32149259,
        9.95552626,  8.87007495, 12.95702647,  6.83083863, 13.27994304,
        8.98765644,  8.62329466,  4.91652556, 10.35093485,  6.34059021,
        7.02326515])}
{'P': tensor([[1.1090e-02, 2.7471e-01, 2.1480e-01,  ..., 2.9122e-06, 2.9596e-05,
         2.4415e-06],
        [7.0920e-04, 3.4933e-02, 1.7015e-03,  ..., 1.1054e-05, 1.0359e-04,
         5.2555e-06],
        [5.0464e-04, 2.4103e-01, 1.5936e-02,  ..., 4.6272e-06, 4.3690e-05,
         2.8206e-06],
        ...,
        [6.2868e-06, 5.1340e-06, 5.4205e-06,  ..., 3.9882e-02, 1.7531e-03,
         2.9118e-06],
        [2.3163e-04, 2.0316e-04, 2.0513e-04,  ..., 1.0764e-02, 1.4253e-01,
         1.8440e-05],
        [4.1021e-03, 3.5762e-03, 4.1220e-03,  ..., 1.9812e-07, 2.0585e-06,
         7.5727e-02]]), 'X': tensor([[ -4.9217,   6.8979,  -1.7667],
        [ -5.6174,   6.4708,  -1.7339],
        [ -5.3736,   6.6310,  -1.5174],
        [ -5.6423,   6.4537,  -1.2091],
        [ -7.3554,   4.8985,  -0.6665],
        [ -7.7842,   4.3466,  -1.7548],
        [  9.4516,  -7.2501,  -1.8868],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.2311,  -2.0445,   1.3545],
        [  0.4119,  14.1019,  -4.1870],
        [ 12.5224,  -0.4497,   4.8448],
        [  3.7992,  -3.9474,  -8.9019],
        [  2.3200,   7.1756,   1.1938],
        [  1.5662,  13.9375,   4.5142],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.1459,   1.4780,  -4.2403],
        [ 12.2914,   3.8037,   6.0978],
        [ -5.3382,   6.6533,   7.8054],
        [ -8.3717,  -5.4769,  -9.2228],
        [  5.1196,  -2.7160,   4.2705],
        [  4.4049,   5.6892,  -4.4394],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.4837,   2.7510,   0.4449],
        [ -2.4956, -10.4682,   3.5632],
        [  4.5430,   5.5438,  -3.8669],
        [  8.7131,  10.2216,   7.7915],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.3900,   3.3727,   3.1650],
        [ -9.0961,   1.6529,   3.1553],
        [  1.8766,   7.3694,   1.1887],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.1905,   3.7253,   4.1175],
        [ -8.4573,   3.2448,   5.4370],
        [  2.0547,   7.2957,  -1.9343],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.3589,  13.7668,  -3.6501],
        [ -9.4430,  -0.1897,   4.8383],
        [  0.5600, -11.0321,  -0.8660],
        [  6.3515,   0.6418,  -6.5823],
        [ -9.1907,  -3.3180,  -5.5943],
        [  4.5767,   5.5071,   3.9069],
        [  1.3670,  -4.7990,   3.2603],
        [  6.0310,  12.2795,  -6.4528],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -3.3116, -10.1539,   7.5919],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -1.6220,   7.8917,   0.0315],
        [ 10.1798,   8.5002,  -5.8893],
        [  6.3460,   1.4252,   1.2291],
        [  2.4149,  -4.6005,  -4.3339],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.9461,  -4.4077,  -0.4831],
        [  5.1142,  -2.7230,  -4.1966],
        [ 11.1469,   6.9237,   1.6160],
        [ -7.8126,   4.3066,   5.1791],
        [  2.1771, -10.9776,  -3.2007],
        [  6.2419,  -0.2313,   4.2440],
        [  0.3231,  -4.7691,  -1.6120],
        [  5.6414, -10.0303,   7.8525],
        [ -4.6393,   7.0461,   1.1624],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.0955,   2.8210,  -4.3321],
        [  0.1460,   7.8340,  -9.4614],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.5010,  -4.5736,  -0.8989],
        [ -2.1382,   7.8339,   7.3524],
        [  9.5018,  -7.1945,  -1.1892],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -7.8126,   4.3066,   5.1791],
        [  9.6662,   9.1725,   8.1056],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.9912,   3.1453,  -3.9989],
        [ 10.4994,  -5.9193,  -7.5153],
        [  2.3373,  -4.6232,  -5.2246],
        [  1.6422,   7.4586,  -6.8056],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.3124,   0.2057,  -0.4593],
        [  1.7228,  -4.7575,   8.3396],
        [  8.7715,  -7.9390,   5.2900],
        [ -6.1914,  -8.2997,   8.5380],
        [ -7.6719,  -6.6260,   7.1468],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.0348,  -2.8229,   3.5436],
        [ -8.6426,  -4.9136,   9.4941],
        [  0.1931,  -4.7491,   7.4385],
        [  2.4572,   7.1085,   7.7376],
        [  5.1890,   4.7318,   0.9515],
        [  8.2154,  -8.4262,   8.0171],
        [ -2.1533, -10.5778,   1.0112],
        [  6.3589,   0.8239,  -8.4230],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.3900,   3.3727,   3.1650],
        [  5.8979,   3.3958,  -3.8713],
        [ 11.2209,  -4.7092,   2.6117],
        [ -2.5056,   7.7720,   5.5878],
        [  1.7659,   7.4127,   4.7871],
        [  1.1644, -11.0394,   7.8511],
        [  2.8602,   6.8905,   5.8779],
        [ -9.4505,  -1.7081,  -0.7039],
        [  5.4426, -10.1181,  -5.1268],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1064,  13.5607,   6.7605]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0756, 0.0446,  ..., 0.2753, 0.4499, 1.1502],
        [0.0756, 0.0000, 0.0360,  ..., 0.3007, 0.5147, 1.2019],
        [0.0446, 0.0360, 0.0000,  ..., 0.3004, 0.4931, 1.1688],
        ...,
        [0.2753, 0.3007, 0.3004,  ..., 0.0000, 0.3172, 1.3305],
        [0.4499, 0.5147, 0.4931,  ..., 0.3172, 0.0000, 1.1731],
        [1.1502, 1.2019, 1.1688,  ..., 1.3305, 1.1731, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.47371868,  8.56889738,  8.53501777,  8.5724042 ,  8.83725081,
        8.91551231, 11.9120321 , 13.79762415, 12.40074901, 14.10796466,
       12.53047489,  5.47866221,  7.54127425, 14.02526505, 12.4980845 ,
        9.26455795, 12.86648771,  8.53014915, 10.00410625,  5.79543038,
        7.19513806,  8.68832398, 12.78327037, 10.76154256,  7.16747878,
       13.43127131,  5.07306274,  9.04255655,  9.24501829,  7.60463349,
        9.73579738,  8.99791316,  9.05843201,  7.57946459,  7.9337647 ,
       13.96746816,  9.44486609, 11.04629471,  6.383897  ,  9.7712321 ,
        7.16058587,  4.98989906, 13.68063098,  6.80559669,  9.75075273,
       10.68031351, 12.32545637,  8.05668609, 13.26209545,  6.50409372,
        5.19577146, 10.00410625,  5.30158497,  5.79391825, 13.12218013,
        8.92098752, 11.1913598 ,  6.24614905,  4.78002665, 11.50791859,
        8.43626114, 11.54672755,  6.7166297 ,  7.83534987, 12.12448912,
       12.24467041,  5.21280156,  8.12043983, 11.9182956 , 10.32715715,
        8.92098752, 13.32557133,  8.5724042 ,  9.64157461,  6.76661595,
       12.05300943,  5.18047638,  7.63726134,  4.92497514,  6.31576264,
        5.05981914, 11.83075005, 10.35464916, 10.1371552 ,  7.09887557,
       10.00794468,  5.77219077,  9.94173708,  4.75303127,  7.52117132,
        7.02253235, 11.76830981, 10.794755  ,  6.41202986, 12.71843462,
        9.04255655,  6.80559669, 12.16901615,  8.16584537,  7.62011734,
       11.10066126,  7.46053281,  9.60359158, 11.4890861 ,  5.13182506,
       13.91198191])}
{'P': tensor([[2.3302e-02, 4.8481e-01, 1.5386e-01,  ..., 1.6854e-06, 1.0306e-04,
         4.4442e-04],
        [7.6484e-04, 4.1402e-02, 2.4811e-01,  ..., 3.3297e-06, 2.3485e-04,
         5.8634e-04],
        [2.2565e-04, 5.0662e-04, 2.2943e-02,  ..., 2.9045e-06, 3.9157e-04,
         3.8023e-04],
        ...,
        [3.9305e-06, 2.6258e-06, 1.7601e-06,  ..., 1.3641e-02, 3.6691e-07,
         1.5279e-06],
        [3.8053e-06, 4.3076e-06, 8.3616e-06,  ..., 1.8994e-04, 4.9618e-02,
         2.1452e-05],
        [1.8470e-04, 2.6111e-04, 9.4212e-04,  ..., 8.2660e-06, 1.9829e-02,
         6.7518e-02]]), 'X': tensor([[ -5.0550,   6.8231,  -9.2502],
        [ -5.9872,   6.2037,  -9.4477],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.8280,   4.2848,  -7.1955],
        [ -8.6112,   2.9319,  -7.9943],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.7232,  -4.7574,   3.1270],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.8450,  -9.9356,   3.0767],
        [ -6.7992,   5.4991,   9.9186],
        [  2.0449,  13.8402,  -7.1642],
        [  2.5318,  13.7231,   6.3692],
        [  9.3607,   9.5337,  -4.5412],
        [ -2.0721,   7.8432,   6.6380],
        [  3.3633,  13.4794,   0.5193],
        [  4.9152,  12.8653,  -1.4459],
        [  1.7232,  -4.7574,   3.1270],
        [ -7.5603,   4.6459,   8.2002],
        [  7.9403,  -8.6455,  -8.7433],
        [ 11.7796,  -3.4696,   9.0088],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.7218,  -9.9934,   6.5764],
        [  5.5294,  -2.1106,   9.0626],
        [ -5.0943,  -9.1682,   0.5073],
        [ -7.5603,   4.6459,   8.2002],
        [  2.5636, -10.9293,   9.6519],
        [ -9.4735,  -0.7274,  -2.6749],
        [  2.6613,   7.0021,   4.5783],
        [  3.3744,   6.5623,  -6.3636],
        [  6.3551,   0.7149,   1.3344],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.2939,   6.6325,  -5.8311],
        [  0.1382, -11.0073,   1.0026],
        [  6.8704,  -9.3809,   7.5769],
        [ -5.6423,   6.4537,  -1.2091],
        [ 12.2914,   3.8037,   6.0978],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.0087,   4.0173,   9.1091],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.6952,  -4.8060,   8.8976],
        [  5.6928,  12.4710,   3.6001],
        [  2.8474,   6.8979,   6.6457],
        [  5.8325,  -1.5222,  -2.1341],
        [  9.9750,  -6.6338,  -1.2440],
        [  8.1043,  -8.5164,  -7.9290],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 11.0438,   7.1180,  -5.6991],
        [  8.3718,  -8.2953,   8.4666],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.3411,  -2.6004,  -2.7062],
        [  2.4697,  -4.5836,   1.4593],
        [ -9.1795,  -3.3623,   8.5259],
        [  5.4426, -10.1181,  -5.1268],
        [ -8.0608,   3.9362,  -5.9567],
        [  6.3542,   0.6934,  -3.3840],
        [  7.0642,  11.6104,  -9.8316],
        [ 12.5770,   1.8512,  -0.6578],
        [ 12.1940,  -2.1881,   7.1499],
        [  7.3592,  11.3937,   6.2407],
        [  6.3200,   0.2690,   1.5900],
        [  0.1041, -11.0045,   2.4311],
        [  3.5965,  -4.0754,  -3.5035],
        [ -5.2589,   6.7024,  -5.5809],
        [  2.8890,  -4.4316,   1.1600],
        [ 10.4737,  -5.9570,  -1.2108],
        [  2.2860,   7.1916,  -6.1011],
        [ -0.2586,   7.8825,  -9.9840],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -7.3446,   4.9113,  -5.9356],
        [  6.1929,   2.4501,  -6.9352],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -6.2200,  -8.2737,   5.5079],
        [  0.6952,  -4.8060,   8.8976],
        [  6.8407,  11.7666,   3.0045],
        [ 12.4837,   2.7510,   0.4449],
        [  4.5503,   5.5359,   7.4735],
        [ -9.1623,  -3.4288,  -4.9961],
        [  5.6314,  -1.9302,  -9.6543],
        [  1.1274,  14.0116,   8.1753],
        [ -5.6651,   6.4380,   1.3660],
        [  1.8721,  -4.7322,   0.5524],
        [ -0.8675, -10.8815,   6.8561],
        [ -8.1531,  -5.8748,   2.6934],
        [  3.4009,  -4.1874,  -5.9751],
        [  2.2182,  13.8007,   5.1889],
        [ -1.8788, -10.6567,   3.7068],
        [  4.8432,   5.1968,   1.5006],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.3376,  -4.6231,   9.8436],
        [  6.3516,   0.6425,   0.3425],
        [ -7.4808,   4.7462,   2.0266],
        [  3.5284, -10.7484,   5.3347],
        [  5.1203,  -2.7151,   6.2110],
        [  2.2644,  -4.6434,   9.7508],
        [  7.9816,  10.8938,  -0.5378],
        [  2.3373,  -4.6232,  -5.2246],
        [  9.6816,   9.1536,  -5.8543],
        [ -2.6485, -10.4151,   0.8766],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.1531,  -5.8748,   2.6934],
        [ -7.4808,   4.7462,   2.0266]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 8.3381e-02, 1.7426e-01,  ..., 4.1238e-01, 8.2718e-01,
         6.7374e-01],
        [8.3381e-02, 0.0000e+00, 9.2034e-02,  ..., 4.8660e-01, 7.8305e-01,
         6.5867e-01],
        [1.7426e-01, 9.2034e-02, 0.0000e+00,  ..., 5.6200e-01, 7.2394e-01,
         6.3304e-01],
        ...,
        [4.1238e-01, 4.8660e-01, 5.6200e-01,  ..., 0.0000e+00, 9.5458e-01,
         6.8798e-01],
        [8.2718e-01, 7.8305e-01, 7.2394e-01,  ..., 9.5458e-01, 0.0000e+00,
         3.0702e-01],
        [6.7374e-01, 6.5867e-01, 6.3304e-01,  ..., 6.8798e-01, 3.0702e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.49161154,  8.62162372,  8.83341608,  8.92395381,  9.09661212,
        9.17905491,  5.05989547, 12.53285633,  6.02789634, 11.52733581,
        8.74470218, 13.99048094, 13.95472331, 13.36092624,  8.11227668,
       13.89264822, 13.77222752,  5.05989547,  8.8737556 , 11.73848065,
       12.27993098, 11.81423161, 11.51557166,  5.91854397, 10.48840476,
        8.8737556 , 11.22596994,  9.5014146 ,  7.49076899,  7.37902766,
        6.39520397,  9.17016857,  9.96152788, 13.09737053, 11.00812996,
       11.62770332,  8.5724042 , 12.86648771,  9.14973319,  8.95983586,
        5.83560009,  9.37296102,  9.77588801,  9.4997987 ,  4.85603886,
       13.70893492,  7.46248947,  6.02789634, 11.97949273, 11.75618513,
       12.00995898, 13.13888841, 11.785566  ,  5.57341623,  9.69629   ,
        5.20659892,  9.77588801, 11.4890861 ,  8.97052115,  6.39187285,
       13.59056613, 12.7125079 , 12.38881635, 13.56367857,  6.32572323,
       11.0050379 ,  5.43544626,  8.51928531,  5.29009836, 12.04924159,
        7.54620849,  7.88677538, 12.10942412,  8.8353674 ,  6.65991877,
        9.58072286, 10.35093485,  4.85603886, 13.6105595 , 12.78327037,
        7.16598273,  9.78286733,  5.95297459, 14.05687815,  8.57560275,
        5.08910491, 10.91601655, 10.04916678,  5.39450305, 13.97779839,
       10.82106505,  7.10378919, 10.4606265 ,  5.1805345 ,  6.38399789,
        8.859416  , 11.31276109,  5.79562691,  5.16611782, 13.50484539,
        5.18047638, 13.32374056, 10.74655346, 11.97957425, 10.04916678,
        8.859416  ])}
{'P': tensor([[1.7612e-02, 1.9928e-01, 1.8779e-01,  ..., 1.7580e-06, 4.9330e-06,
         2.4523e-06],
        [2.8344e-04, 2.5682e-02, 5.7735e-04,  ..., 1.2212e-06, 5.1538e-06,
         5.3679e-06],
        [3.3561e-04, 3.7223e-01, 1.6839e-02,  ..., 1.0380e-06, 4.1506e-06,
         3.5461e-06],
        ...,
        [1.6342e-03, 7.1973e-04, 1.0793e-03,  ..., 3.5107e-02, 4.2581e-04,
         2.8540e-06],
        [4.7564e-04, 3.5502e-04, 4.7479e-04,  ..., 4.0895e-03, 7.0375e-02,
         1.8824e-04],
        [2.8035e-05, 2.4966e-05, 2.9719e-05,  ..., 3.3236e-04, 1.0657e-02,
         1.9687e-01]]), 'X': tensor([[ -5.1369,   6.7755,   8.3836],
        [ -7.0120,   5.2825,   7.7762],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.5603,   4.6459,   8.2002],
        [ -7.4887,   4.7364,   8.9074],
        [ -7.5401,   4.6718,   9.0033],
        [  3.1756, -10.8247,   6.8882],
        [ 12.1588,   4.3364,  -8.0867],
        [ -2.6735, -10.4062,  -1.5911],
        [  0.1460,   7.8340,  -9.4614],
        [ -6.6895,   5.6051,   2.7899],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 12.6038,   0.7665,  -3.4439],
        [ -3.4488, -10.0934,   2.5195],
        [  5.1292,   4.8183,  -5.7997],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.2659,  13.7894,   6.5436],
        [  4.8432,   5.1968,   1.5006],
        [  4.5491,  -3.3485,   6.8412],
        [ -1.6906,   7.8860,   7.7142],
        [  5.9548,  -1.2273,  -2.6520],
        [  2.0419,  -4.6978,  -5.0983],
        [  0.9659,  -4.8144,  -0.9594],
        [  9.5018,  -7.1945,  -1.1892],
        [  1.7659,   7.4127,   4.7871],
        [  4.5491,  -3.3485,   6.8412],
        [ 12.3759,  -1.3943,   6.4659],
        [  5.8889,  -1.3916,   8.8993],
        [  6.2535,  -0.1709,  -9.4102],
        [ -2.1382,   7.8339,   7.3524],
        [ -5.3328,  -8.9992,   7.3440],
        [  5.0506,   4.9277,  -9.5094],
        [ -6.4844,  -8.0230,   8.5209],
        [ -4.1296,   7.2803,  -2.7194],
        [  4.3459,  -3.5327,   0.3667],
        [  4.0591,   6.0202,  -1.6339],
        [  6.0890,  -0.8382,   2.9166],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.4203,  -2.0340,   2.5856],
        [  5.8275,  -1.5334,   2.0124],
        [ -6.6998,   5.5954,   0.0823],
        [  6.1737,   2.5305,   3.9049],
        [  6.2555,   2.1474,   4.1685],
        [  0.5250, -11.0306,   5.5358],
        [  5.5240,  -2.1198,   6.2055],
        [  5.2199,   4.6861,  -1.9304],
        [  0.7275,  14.0667,  -0.9815],
        [  1.1994,   7.6035,   2.4295],
        [  6.0047,   3.1062,  -6.7914],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.7145,  -4.8071,  -2.3766],
        [ -1.8823, -10.6558,  -3.8952],
        [  7.0171,  11.6439,  -5.0178],
        [  0.1931,  -4.7491,   7.4385],
        [  1.3103,  -4.8032,  -0.9300],
        [  3.2226,  -4.2802,  -4.0428],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.4524,  -1.6813,  -6.6862],
        [  2.4356,  -4.5942,  -5.0808],
        [ -0.1963,   7.8765,  -1.1371],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.2770,  -0.0379,   2.4195],
        [ -6.0195,   6.1789,  -2.9496],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.1771, -10.9776,  -3.2007],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -8.3062,  -5.6008,   0.0287],
        [ 11.0891,  -4.9558,  -2.3178],
        [  3.3069,   6.6089,   8.7377],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.2419,  -0.2313,   4.2440],
        [ -2.1382,   7.8339,   7.3524],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.0031,  12.2958,  -6.5992],
        [  5.1142,  -2.7230,  -4.1966],
        [  1.3339,   7.5627,  -0.9034],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.1960,  -3.2964,   1.6921],
        [  3.4401,  13.4540,   4.6576],
        [  7.4132,  11.3526,   0.3322],
        [  3.5939,   6.4029,  -1.8357],
        [ -9.4505,  -1.7081,  -0.7039],
        [  2.7558,  -4.4844,   0.0636],
        [ -8.6801,   2.7809,  -0.8443],
        [  2.1504,  -4.6726,   8.7908],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.5123,  -0.5339,  -1.0728],
        [  5.5294,  -2.1106,   9.0626],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -2.1817,   7.8274,  -4.6298],
        [ -9.2707,   0.9661,   4.4863],
        [ -7.8126,   4.3066,   5.1791],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.0140,  -1.0661,  -2.5040],
        [ -6.0175,  -8.4541,  -9.8408],
        [  3.9085,   6.1512,  -2.2560],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.6523,   3.9443,  -2.9449],
        [  5.7845,   3.6652,   0.9175],
        [  6.7792,  11.8084,   1.6251],
        [  4.5430,   5.5438,  -3.8669],
        [  3.2990,  -4.2415,  -7.4857]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1182, 0.0729,  ..., 0.8147, 0.9095, 1.0349],
        [0.1182, 0.0000, 0.0490,  ..., 0.9123, 0.9507, 1.0716],
        [0.0729, 0.0490, 0.0000,  ..., 0.8651, 0.9193, 1.0419],
        ...,
        [0.8147, 0.9123, 0.8651,  ..., 0.0000, 0.5457, 0.6429],
        [0.9095, 0.9507, 0.9193,  ..., 0.5457, 0.0000, 0.1301],
        [1.0349, 1.0716, 1.0419,  ..., 0.6429, 0.1301, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.50268322,  8.77913428,  8.68979052,  8.8737556 ,  8.86082124,
        8.87007495, 11.28093507, 12.9089559 , 10.74409834,  7.83534987,
        8.7273569 , 10.46739073, 12.62711252, 10.66631784,  7.03733775,
        9.85384623, 13.97430191,  7.10378919,  5.64865173,  8.06517143,
        6.07992623,  5.1224079 ,  4.91038884, 11.9182956 ,  7.62011734,
        5.64865173, 12.45418236,  6.05113217,  6.25586297,  8.12043983,
       10.4606265 ,  7.05627461, 10.31584102,  8.36995953,  5.60064319,
        7.26074925,  6.1463896 ,  7.53063885,  9.63742669,  6.02588517,
        8.72896381,  6.67218687,  6.61386436, 11.04313641,  5.91677188,
        7.01475376, 14.08549802,  7.69752262,  6.76056625,  9.64157461,
        4.85995002, 10.8207297 , 13.59480906,  4.75303127,  4.978704  ,
        5.35774573, 10.67322527,  9.60080639,  5.19986797,  7.8788965 ,
       13.15525803,  6.27715105,  8.62631801,  8.83341608, 11.1913598 ,
        9.60359158, 10.01804223, 12.1460931 ,  7.39004226,  8.62260327,
        9.60359158, 10.48315436,  6.24614905,  8.12043983, 13.26590071,
       13.6829843 ,  5.79391825,  7.67939546, 10.35485331,  9.23071698,
        9.76896904, 13.88683436, 13.55868684,  7.34250948,  9.60359158,
        5.26342203,  9.11473391,  5.14371005,  8.53501777, 12.52372629,
        5.91854397, 13.38091121,  9.50913849,  8.12580976,  9.32094055,
        8.92098752, 10.52357479,  6.10774544, 10.37698435,  7.28793663,
       12.76408562,  6.89245352,  6.84796202, 13.61600536,  7.16747878,
        5.37343173])}
{'P': tensor([[1.7307e-02, 2.3341e-01, 1.7890e-01,  ..., 1.9098e-05, 6.3294e-03,
         3.9686e-05],
        [2.3961e-04, 1.7784e-02, 1.9133e-01,  ..., 1.3848e-05, 1.2242e-02,
         2.2167e-05],
        [1.9777e-04, 7.9342e-04, 2.3724e-02,  ..., 1.5899e-05, 3.0354e-02,
         2.1043e-05],
        ...,
        [8.0439e-03, 3.6128e-03, 2.3956e-03,  ..., 6.3437e-02, 1.7563e-04,
         3.6558e-02],
        [5.0053e-05, 8.5660e-05, 1.1834e-04,  ..., 2.4295e-05, 6.3392e-02,
         1.0734e-05],
        [3.3824e-02, 2.3267e-02, 1.3330e-02,  ..., 1.1146e-03, 6.2394e-04,
         1.1565e-01]]), 'X': tensor([[-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0834e-01, 1.5711e-01,  ..., 4.4766e-01, 5.4572e-01,
         5.1056e-01],
        [1.0834e-01, 0.0000e+00, 7.7534e-02,  ..., 5.4924e-01, 4.7073e-01,
         5.5822e-01],
        [1.5711e-01, 7.7534e-02, 0.0000e+00,  ..., 6.0448e-01, 3.9628e-01,
         6.3527e-01],
        ...,
        [4.4766e-01, 5.4924e-01, 6.0448e-01,  ..., 0.0000e+00, 9.7948e-01,
         3.8340e-01],
        [5.4572e-01, 4.7073e-01, 3.9628e-01,  ..., 9.7948e-01, 0.0000e+00,
         1.0265e+00],
        [5.1056e-01, 5.5822e-01, 6.3527e-01,  ..., 3.8340e-01, 1.0265e+00,
         1.5170e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.50278896,  8.72896381,  8.83725081,  8.89301516,  8.95133961,
        8.5724042 , 12.72524615,  9.97964013,  7.72291946, 12.07225469,
       11.85743425, 13.52430176,  9.17905491,  4.92497514, 11.87412439,
        7.19534146, 10.90638066,  7.50500064,  8.68832398,  5.13182506,
        5.14371005,  9.36256096,  4.78258092,  7.58378655, 10.06850677,
        7.8788965 , 13.6922425 , 13.59673406, 11.04953767, 11.11038551,
       11.27160035, 10.00794468,  5.63824971,  9.79814997,  8.94395628,
        9.5014146 , 12.18998195,  7.81509631,  8.29640613, 11.04629471,
        5.91677188,  7.24677664, 11.72802491, 10.68416401,  9.88571713,
        9.3109884 , 11.00508681,  9.55593374, 10.68416401,  4.91336994,
        5.2287058 ,  8.66273171,  8.53501777,  7.09887557,  5.05774295,
        6.34059021, 11.83570132, 13.89886469, 13.89264822,  8.51928531,
       13.04156075,  7.46248947, 10.95593729, 13.13205357,  8.79284466,
       13.35832725,  6.34059021, 13.88683436,  5.91408504,  8.57560275,
       12.10222204,  7.88677538, 11.04313641,  5.05534609,  6.01356004,
        5.79562691, 11.72749923,  8.87587457, 12.38881635,  7.78980299,
        7.16598273,  5.38342077,  7.27340834,  7.8788965 ,  6.33517349,
       10.60226614,  9.36256096, 13.3426541 , 13.27717982, 14.09357547,
       11.36089355,  7.62011734, 10.06850677,  6.78325151,  4.83371341,
        5.91408504,  9.77777346, 10.7506423 ,  9.34127013,  8.01009083,
        6.12886024, 10.37998078,  7.83486943,  7.02496223,  9.55720651,
        7.62011734])}
{'P': tensor([[1.3132e-02, 2.4636e-01, 1.7830e-01,  ..., 1.7340e-06, 2.0328e-05,
         1.8832e-04],
        [2.4885e-04, 1.6982e-02, 9.9107e-02,  ..., 2.9029e-06, 2.8925e-05,
         1.0430e-04],
        [2.5602e-04, 1.8402e-03, 1.8022e-02,  ..., 3.2183e-06, 3.1346e-05,
         9.7607e-05],
        ...,
        [3.4001e-05, 2.1116e-05, 1.8728e-05,  ..., 1.2310e-01, 2.0535e-02,
         1.0295e-04],
        [4.7879e-04, 3.1924e-04, 2.9120e-04,  ..., 2.3138e-03, 9.7242e-02,
         1.0660e-03],
        [1.1377e-01, 8.9565e-02, 6.4641e-02,  ..., 3.6040e-06, 4.9725e-05,
         3.5200e-02]]), 'X': tensor([[ -5.1974,   6.7396,   3.3484],
        [ -6.4368,   5.8356,   3.3014],
        [ -6.6895,   5.6051,   2.7899],
        [ -6.2668,   5.9808,   2.5614],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.2515,   5.0195,   4.6434],
        [  1.6408,   7.4591,  -4.2294],
        [ -4.5778,  -9.5013,  -5.0667],
        [  9.6662,   9.1725,   8.1056],
        [  1.1644, -11.0394,   7.8511],
        [  9.6816,   9.1536,  -5.8543],
        [ -7.3470,  -7.0615,   5.0066],
        [  6.3124,   0.2057,  -0.4593],
        [  5.6248,   3.9984,   8.4933],
        [ -9.3616,  -2.4749,   0.1793],
        [ 11.1085,  -4.9204,   0.0736],
        [  4.4039,   5.6903,   8.7366],
        [  3.6469,   6.3624,  -1.8414],
        [  4.4290,  -3.4596,   1.4181],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4714,  -1.3301,  -3.1998],
        [  7.5784,  -8.9141,   5.8410],
        [ -3.5520,   7.4976,   9.1522],
        [  4.0078,  -3.8018,  -6.2168],
        [ -9.4430,  -0.1897,   4.8383],
        [ -8.7078,   2.7182,   1.8146],
        [ -4.7334,   6.9982,  -0.4392],
        [ -7.3446,   4.9113,  -5.9356],
        [  6.3497,   1.3680,  -1.9076],
        [  5.9912,   3.1453,  -3.9989],
        [ -7.9667,   4.0814,  -0.6397],
        [  3.9591, -10.6391,  -6.2557],
        [  9.0129,  -7.7072,   9.1400],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.3804,  -2.3504,   7.2744],
        [  5.6414, -10.0303,   7.8525],
        [  1.2273,  -4.8081,   9.8129],
        [  5.3602,  -2.3808,   7.2098],
        [  5.1472,  -2.6801,   8.1194],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -9.4624,  -0.4633,  -7.3918],
        [  1.3705,  13.9723,   7.9266],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.6523,   3.9443,  -2.9449],
        [  4.5503,   5.5359,   7.4735],
        [  2.2076,  13.8032,  -1.5099],
        [  7.3724,  -9.0577,  -8.9843],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -1.6275, -10.7221,   3.2376],
        [  5.0506,   4.9277,  -9.5094],
        [ -3.2736, -10.1703,   3.8712],
        [  6.8704,  -9.3809,   7.5769],
        [  3.4140,  -4.1803,  -8.8224],
        [ -4.4520,   7.1370,  -8.7867],
        [  1.9312,  13.8649,   8.3563],
        [  2.3261,   7.1726,   6.2522],
        [  4.7482, -10.3907,   5.9149],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.4242, -10.1261,   3.9595],
        [ -8.1666,   3.7653,   3.6075],
        [  4.5430,   5.5438,  -3.8669],
        [  3.6703,   6.3443,  -5.4401],
        [  5.1472,  -2.6801,   8.1194],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.6314,  -1.9302,  -9.6543],
        [  6.3031,   1.8477,  -7.1706],
        [  2.6078, -10.9229,   6.2869],
        [  4.5340,  -3.3629,   8.9652],
        [  6.1575,  12.2046,   3.3257],
        [  5.0277,   4.9589,   7.0175],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.6634,   5.8050,   9.1590],
        [  1.3403,  -4.8011,  -1.1698],
        [ -4.3963,   7.1629,   4.5263],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.3550,   0.7129,   0.4917],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.2345,  -3.1345,   3.9715],
        [  7.9816,  10.8938,  -0.5378],
        [  1.8320,   7.3871,   5.1452],
        [ -0.3432,   7.8900,   7.1805],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.2790,  -0.0259,  -5.4116],
        [  9.9217,  -6.7006,  -5.6728],
        [  6.0890,  -0.8382,   2.9166],
        [ 12.2328,  -2.0376,  -1.1234],
        [  1.0000,   7.6591,  -4.5011],
        [  2.4149,  -4.6005,  -4.3339],
        [  0.9945,  -4.8144,  -0.1897],
        [  6.2242,  -0.3171,   7.4978],
        [  0.5250, -11.0306,   5.5358],
        [ -9.1795,  -3.3623,   8.5259],
        [ -4.8751,  -9.3148,   4.0406],
        [ -5.1369,   6.7755,   8.3836],
        [  3.9591, -10.6391,  -6.2557],
        [  7.1697,  11.5343,  -0.7276],
        [  1.7815,  -4.7481,   4.4289],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.2687,  -2.9794,   0.6436],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.2790,  -0.0259,  -5.4116],
        [  2.6613,   7.0021,   4.5783],
        [  8.7715,  -7.9390,   5.2900],
        [  5.0348,  -2.8229,   3.5436],
        [ -2.5056,   7.7720,   5.5878]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0785, 0.1088,  ..., 0.3198, 0.4718, 0.2137],
        [0.0785, 0.0000, 0.0424,  ..., 0.3830, 0.5487, 0.2793],
        [0.1088, 0.0424, 0.0000,  ..., 0.4226, 0.5677, 0.3180],
        ...,
        [0.3198, 0.3830, 0.4226,  ..., 0.0000, 0.3786, 0.1063],
        [0.4718, 0.5487, 0.5677,  ..., 0.3786, 0.0000, 0.3774],
        [0.2137, 0.2793, 0.3180,  ..., 0.1063, 0.3774, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.51088913,  8.68832398,  8.7273569 ,  8.66273171,  8.90405823,
        8.81930525,  7.6374603 , 10.54658597, 13.32557133, 11.10066126,
       13.32374056, 10.19036081,  6.31576264,  6.90115901,  9.68323686,
       12.1494023 ,  7.19534146,  7.33350642,  5.62005629,  9.5014146 ,
        9.5642993 , 11.70016217,  8.29640613,  5.52416368,  9.44486609,
        9.12220271,  8.4486771 ,  8.8353674 ,  6.49538841,  6.76661595,
        8.95133961, 11.35182589, 11.85890664, 10.53368644,  5.87132718,
       11.50791859,  4.96231532,  5.86519525,  5.80312949, 12.52372629,
        9.47369628, 14.0393949 ,  7.6374603 , 12.41012046,  6.89245352,
        7.16598273, 13.97857591, 11.67875855, 10.0674887 , 10.84493732,
        7.05627461, 10.68416401, 11.62770332,  5.39723241,  8.41171369,
       13.99877413,  7.5403822 , 11.42420508,  9.64157461, 11.48734978,
        8.99276564,  7.16747878,  7.32950385,  5.80312949, 11.51811603,
        5.95297459,  6.56833864, 11.22993218,  5.64500512, 13.66991027,
        7.06169391,  5.8828986 , 13.02818702,  4.98461983,  8.40445872,
        8.41171369,  6.3949043 ,  8.09704052,  9.75200848, 13.50484539,
        7.61088125,  7.89745107,  5.84533345,  6.27905292, 11.97238624,
        6.1463896 , 12.4013153 ,  7.72415048,  5.19577146,  4.9160962 ,
        6.2322769 , 11.04313641,  9.77588801, 10.51340829,  8.50268322,
       11.35182589, 13.58101794,  5.0713364 ,  9.09963894,  9.73579738,
        5.53255882,  6.27905292,  7.49076899, 11.83075005,  5.77219077,
        8.16584537])}
{'P': tensor([[1.5854e-02, 1.8907e-01, 1.8646e-01,  ..., 1.4337e-06, 3.1645e-05,
         9.8383e-04],
        [1.7979e-04, 1.6386e-02, 4.3491e-04,  ..., 7.2090e-07, 1.7032e-05,
         8.4423e-04],
        [6.2596e-04, 1.4461e-01, 1.8521e-02,  ..., 1.1361e-06, 2.8749e-05,
         7.4520e-04],
        ...,
        [1.0735e-03, 4.8078e-04, 6.4007e-04,  ..., 2.7712e-02, 6.3815e-04,
         4.2136e-04],
        [9.1797e-02, 3.1347e-02, 6.5836e-02,  ..., 8.3871e-04, 1.4592e-01,
         1.4384e-03],
        [3.5616e-04, 1.2094e-03, 4.3780e-04,  ..., 3.9802e-06, 2.6278e-05,
         1.0613e-01]]), 'X': tensor([[-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1272, 0.0628,  ..., 0.9488, 0.3575, 0.6591],
        [0.1272, 0.0000, 0.0875,  ..., 1.0430, 0.4843, 0.6164],
        [0.0628, 0.0875, 0.0000,  ..., 1.0112, 0.4113, 0.6768],
        ...,
        [0.9488, 1.0430, 1.0112,  ..., 0.0000, 0.7107, 0.9536],
        [0.3575, 0.4843, 0.4113,  ..., 0.7107, 0.0000, 0.8491],
        [0.6591, 0.6164, 0.6768,  ..., 0.9536, 0.8491, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.51928531,  8.7907923 ,  8.62329466,  8.8353674 ,  8.89458273,
        8.85572292,  7.50035355,  7.92607342,  7.58668144, 13.52430176,
       10.9036226 , 13.02818702,  5.1224079 , 13.04156075,  5.23878371,
        5.16077764, 11.72802491,  5.94909709, 11.38774149,  8.0550286 ,
        7.27759233,  4.91296205, 13.98691426,  9.73579738, 12.05300943,
        7.56617037,  6.33075682, 10.35464916, 10.1371552 , 10.04916678,
       11.64746882, 14.08549802, 11.56146873,  5.94293185, 11.83570132,
       12.50996828,  9.37296102,  9.68324553,  8.98765644,  7.10378919,
        5.13616224, 10.88618114,  5.96556248, 11.33837396,  5.91408504,
        8.25828409, 11.1913598 ,  6.81117802, 13.01187484,  8.44940661,
        7.06372172, 12.16901615, 12.26997165,  6.65259775, 10.1371552 ,
        8.29203356, 13.32613445,  9.20340031,  6.1561646 ,  9.75075273,
        9.99145096, 12.67680439, 10.41909211,  8.8580671 ,  9.50654967,
       10.7167285 ,  6.56833864,  5.37343173,  9.14044984,  7.73151309,
       12.07225469,  4.78002665, 11.27656488,  6.42689705,  7.02496223,
        6.38105015,  9.96391045, 10.33712258,  7.24326992,  9.19837792,
        8.09899874, 11.35345345, 13.97857591,  5.80312949,  5.84450127,
       13.98691426,  5.57341623,  6.60752373,  5.96556248, 11.20100795,
        8.55982105,  5.20659892,  9.68323686, 11.51557166,  5.53255882,
        8.04718433, 11.28215429,  8.37954853, 11.1341798 ,  9.41605684,
        5.22903117,  5.47702766,  7.53884104, 13.61600536,  7.50500064,
        8.99276564])}
{'P': tensor([[1.4454e-02, 2.6494e-01, 2.1034e-01,  ..., 2.7478e-06, 4.5885e-06,
         4.7915e-07],
        [2.3340e-04, 1.4068e-02, 3.1099e-01,  ..., 1.5062e-06, 2.3297e-06,
         4.1290e-07],
        [2.0389e-04, 4.1187e-04, 1.8321e-02,  ..., 1.4034e-06, 2.0473e-06,
         4.9191e-07],
        ...,
        [1.3400e-03, 1.0054e-03, 7.0772e-04,  ..., 1.7597e-02, 7.9228e-01,
         4.9722e-07],
        [1.6638e-02, 1.5805e-02, 1.2075e-02,  ..., 4.8991e-04, 2.9679e-01,
         4.1072e-06],
        [1.6947e-04, 9.6744e-05, 6.5513e-05,  ..., 2.8718e-02, 7.9037e-03,
         6.1203e-02]]), 'X': tensor([[ -5.3382,   6.6533,   7.8054],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.0120,   5.2825,   7.7762],
        [ -7.5603,   4.6459,   8.2002],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.4887,   4.7364,   8.9074],
        [  7.7794,  11.0629,   8.9069],
        [ 12.5123,  -0.5339,  -1.0728],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.5960, -11.0335,   3.1144],
        [ -9.1459,   1.4780,  -4.2403],
        [  1.6408,   7.4591,  -4.2294],
        [ -3.9700,   7.3453,  -4.1486],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.0890,  -0.8382,   2.9166],
        [  3.8182,  -3.9347,  -1.7961],
        [  8.5198,  10.4091,  -8.6486],
        [  1.9204,   7.3518,   8.4781],
        [  5.8323,  -1.5228,   5.1132],
        [  4.4039,   5.6903,   8.7366],
        [ 12.5927,   0.4199,  -9.4997],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.1749,  -3.3803,   6.4629],
        [  0.1382, -11.0073,   1.0026],
        [ -3.2053, -10.1993,  -3.8736],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 12.5789,   0.1722,   1.6470],
        [  6.2790,  -0.0259,  -5.4116],
        [  8.7131,  10.2216,   7.7915],
        [  3.7992,  -3.9474,  -8.9019],
        [  0.7958,  -4.8109,  -8.2248],
        [  3.1892, -10.8220,  -6.8409],
        [  6.3534,   1.2990,  -7.2942],
        [ -7.3470,  -7.0615,   5.0066],
        [  7.0515,  11.6194,   8.4289],
        [  6.1984,  -0.4327,   1.5438],
        [  9.1409,  -7.5789,   1.5814],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.5010,  -4.5736,  -0.8989],
        [ -8.8058,   2.4850,  -1.7425],
        [ -1.6906,   7.8860,   7.7142],
        [  3.2226,  -4.2802,  -4.0428],
        [  5.2939,  -2.4784,  -4.5840],
        [  2.4572,   7.1085,   7.7376],
        [  5.9003,  12.3550,   2.1917],
        [ -1.6334, -10.7207,   7.1397],
        [  2.3589,  13.7668,  -3.6501],
        [  4.7156,  -3.1831,  -8.0224],
        [  2.6451,  -4.5250,  -0.5565],
        [ 11.1420,   6.9331,  -2.4779],
        [  5.9026,   3.3837,  -0.6573],
        [ 12.0582,  -2.6640,   0.7132],
        [ -0.1963,   7.8765,  -1.1371],
        [  0.2732,  -4.7619,  -1.1577],
        [  0.2571,  -4.7594,   1.5618],
        [  6.2576,   2.1359,  -3.3398],
        [  3.2366,  -4.2732,   5.1532],
        [ 12.0582,  -2.6640,   0.7132],
        [ 11.2517,  -4.6495,   2.9800],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.4739,  -1.2497,   0.5483],
        [  3.2692,  13.5098,  -4.9032],
        [  9.3836,   9.5075,  -5.9278],
        [ -2.0445,   7.8469,  -0.5857],
        [ -0.0282,   7.8575,  -1.1746],
        [  9.2692,   9.6370,   7.0746],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.6613,   7.0021,   4.5783],
        [  2.4149,  -4.6005,  -4.3339],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -8.6603,   2.8250,   5.9181],
        [  6.3246,   1.6696,  -0.1544],
        [ 12.3609,   3.4769,  -8.5762],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.1426,   3.8048,   9.3542],
        [ -4.0868,   7.2981,   8.1874],
        [  5.6523,   3.9443,  -2.9449],
        [ -9.1624,  -3.4286,   2.0647],
        [ -5.3382,   6.6533,   7.8054],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.6662,   9.1725,   8.1056],
        [  3.9515,  -3.8425,   6.1729],
        [  0.9074,  -4.8139,   8.1674],
        [  6.1372,   2.6726,  -6.7885],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 11.6801,  -3.7196,  -0.9181],
        [  4.9531,   5.0576,   5.6794],
        [ -5.1974,   6.7396,   3.3484],
        [ -9.4203,  -2.0340,   2.5856],
        [ -1.5261,   7.8988,   9.3768],
        [  1.7903,  -4.7467,   0.1168],
        [  6.0305,   3.0296,   0.1375],
        [ -7.9667,   4.0814,  -0.6397],
        [ 10.8670,  -5.3428,  -2.8637],
        [  1.1644, -11.0394,   7.8511],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.4996,   2.6335,  -6.3166],
        [  3.3633,  13.4794,   0.5193],
        [  0.7275,  14.0667,  -0.9815],
        [ 12.3357,   3.6001,   0.7903]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0687, 0.1166,  ..., 0.8361, 0.8053, 0.8009],
        [0.0687, 0.0000, 0.0490,  ..., 0.8872, 0.8421, 0.8503],
        [0.1166, 0.0490, 0.0000,  ..., 0.9321, 0.8805, 0.8945],
        ...,
        [0.8361, 0.8872, 0.9321,  ..., 0.0000, 0.2326, 0.0477],
        [0.8053, 0.8421, 0.8805,  ..., 0.2326, 0.0000, 0.1944],
        [0.8009, 0.8503, 0.8945,  ..., 0.0477, 0.1944, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.53014915,  8.68979052,  8.77913428,  8.8737556 ,  8.92549752,
        8.86082124, 13.52430176, 12.52372629,  5.26067019, 11.04953767,
        9.26455795,  7.6374603 ,  8.34949383,  9.562174  , 10.46739073,
        6.1463896 ,  5.48276038, 13.45124917,  7.5984809 ,  6.02779201,
        7.19534146, 12.59969936,  7.71075226,  9.77777346, 11.00812996,
       10.6910768 ,  8.97052115,  9.86089579, 12.58006038,  6.27905292,
       13.43127131,  5.47866221,  4.8763135 , 11.28215429,  6.48486171,
       10.19036081, 13.59170809,  6.2134961 , 11.87412439,  9.91668182,
        5.21280156,  9.14973319,  8.06517143,  5.35774573,  5.84533345,
        7.52117132, 13.69162795, 10.84437705, 13.96746816,  5.68942572,
        5.24137909, 13.12299145,  6.80370254, 12.3489324 ,  7.8788965 ,
        4.76969364,  4.7663526 ,  6.61211224,  5.36061974, 12.3489324 ,
       12.17451685, 10.67322527,  9.55593374, 13.89974693, 13.35832725,
        8.10886021,  7.8575648 , 13.37121923,  6.1561646 ,  7.49076899,
        5.19577146,  9.43112957,  9.10945678,  6.54127431, 12.84056604,
        9.47369628,  9.3349858 ,  8.98765644,  8.36445758,  6.89245352,
        9.78285424,  8.53014915, 10.5575601 ,  5.26067019, 13.32557133,
        5.51177501,  4.89869205,  6.69389431, 13.97857591,  9.75075273,
       12.35421461, 12.25806559,  7.07902036,  8.51088913,  9.63742669,
        8.04483495,  5.07306274,  6.74872981,  8.95133961, 12.10942412,
       11.10066126,  5.19577146, 12.77402351, 13.89264822, 14.08549802,
       12.85032836])}
{'P': tensor([[1.5710e-02, 2.3761e-01, 1.7203e-01,  ..., 1.2632e-05, 6.7623e-07,
         1.0915e-05],
        [1.2631e-03, 2.5934e-02, 2.5549e-02,  ..., 2.0305e-05, 9.9842e-07,
         1.7212e-05],
        [1.4932e-03, 1.5519e-02, 2.3792e-02,  ..., 2.1636e-05, 9.7377e-07,
         1.6679e-05],
        ...,
        [8.3160e-05, 7.5825e-05, 8.6408e-05,  ..., 1.7803e-01, 1.0287e-03,
         1.3024e-02],
        [1.1128e-04, 9.3187e-05, 1.0705e-04,  ..., 2.2713e-06, 1.0746e-02,
         1.7462e-05],
        [5.8219e-04, 5.0760e-04, 5.5079e-04,  ..., 2.1523e-04, 6.4454e-03,
         4.8301e-02]]), 'X': tensor([[ -5.3736,   6.6310,  -1.5174],
        [ -5.6174,   6.4708,  -1.7339],
        [ -5.6423,   6.4537,  -1.2091],
        [ -7.3554,   4.8985,  -0.6665],
        [ -7.7842,   4.3466,  -1.7548],
        [ -7.9667,   4.0814,  -0.6397],
        [ -2.6069, -10.4298,  -6.1995],
        [  5.0617,  12.7957,   8.3544],
        [ 10.4994,  -5.9193,  -7.5153],
        [  6.1741,  -0.5329,   3.1913],
        [ -7.6734,   4.4980,  -5.4248],
        [  4.2612, -10.5514,  -6.6268],
        [ 12.6038,   0.7665,  -3.4439],
        [  9.1409,  -7.5789,   1.5814],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.2804,   2.0019,   0.8607],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.6451,  -4.5250,  -0.5565],
        [  3.1480,  -4.3165,  -7.8002],
        [  5.2608,  12.6977,   4.9166],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.7916,  -3.9524,  -7.9547],
        [  4.2275,  -3.6320,  -0.2714],
        [ 12.4837,   2.7510,   0.4449],
        [ -4.4520,   7.1370,  -8.7867],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -4.3963,   7.1629,   4.5263],
        [ 12.2311,  -2.0445,   1.3545],
        [ -5.9225,  -8.5355,   2.4778],
        [ 12.5199,  -0.4711,   3.0436],
        [ 11.4703,  -4.2006,   7.4086],
        [  6.2022,   2.4092,   3.0349],
        [ 12.4691,   2.8525,  -5.5110],
        [  4.3554,   5.7393,  -9.0766],
        [ -8.8934,   2.2597,   3.5831],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.8318,   3.5567,  -4.7692],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.3800,  -4.3919,   1.4066],
        [  8.5093,  -8.1764,  -3.2381],
        [ -6.0195,   6.1789,  -2.9496],
        [  5.1890,   4.7318,   0.9515],
        [  2.0039,   7.3173,  -1.5885],
        [  1.2273,  -4.8081,   9.8129],
        [  7.9816,  10.8938,  -0.5378],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.6093,  -1.9705,   6.4807],
        [ -1.6334, -10.7207,   7.1397],
        [  0.7958,  -4.8109,  -8.2248],
        [  7.0515,  11.6194,   8.4289],
        [ -8.4824,  -5.2575,  -0.0795],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.2592,   3.9425,   2.0383],
        [  0.9392,  -4.8143,  -3.4649],
        [  2.4572,   7.1085,   7.7376],
        [ 11.1420,   6.9331,  -2.4779],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.7500,   7.4187,   4.4237],
        [ -8.8400,   2.3991,   1.3457],
        [  2.1504,  -4.6726,   8.7908],
        [  6.1752,   2.5242,   3.5874],
        [  5.7151,  -1.7700,   3.1542],
        [  2.1504,  -4.6726,   8.7908],
        [ -6.4620,  -8.0449,   6.2719],
        [ 10.9931,   7.2107,   7.7933],
        [  5.3858,   4.4260,  -9.6769],
        [  8.5093,  -8.1764,  -3.2381],
        [  4.0421,  -3.7765,  -8.3631],
        [ -9.4645,  -1.4880,  -2.0226],
        [  2.3589,  13.7668,  -3.6501],
        [ -9.1749,  -3.3803,   6.4629],
        [  2.7121,   6.9743,  -8.4202],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.8745,   2.3097,   3.5600],
        [  5.5294,  -2.1106,   9.0626],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.6229,   2.9068,   2.4429],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.7485,  -9.9811,  -3.2355],
        [ -5.3736,   6.6310,  -1.5174],
        [ -4.7642,  -9.3860,  -5.1133],
        [  2.0419,  -4.6978,  -5.0983],
        [ -2.7572,   7.7194,   5.9580],
        [  5.1142,  -2.7230,  -4.1966],
        [  3.3094,  13.4969,  -3.6507],
        [ 12.2589,  -1.9312,   9.8657],
        [  1.7232,  -4.7574,   3.1270],
        [ -3.6381,   7.4683,   7.3544],
        [  3.7237,  -3.9966,  -5.0376],
        [  0.7275,  14.0667,  -0.9815],
        [ 10.8670,  -5.3428,  -2.8637],
        [  4.1999,  -3.6543,  -4.1998],
        [ -6.7745,  -7.7264,   8.4684],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.4996,   2.6335,  -6.3166],
        [ -1.5408,   7.8978,   5.3470],
        [ -9.4671,  -0.5553,   0.1117],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -5.8389,  -8.6056,  -3.5098],
        [  2.0870,  -4.6877,   9.6823],
        [  9.2979,   9.6048,   7.3138],
        [  6.3606,   0.9204,   3.2525]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0360, 0.0405,  ..., 0.7927, 1.1834, 0.6158],
        [0.0360, 0.0000, 0.0358,  ..., 0.8231, 1.2166, 0.6508],
        [0.0405, 0.0358, 0.0000,  ..., 0.7947, 1.1921, 0.6311],
        ...,
        [0.7927, 0.8231, 0.7947,  ..., 0.0000, 0.4263, 0.2936],
        [1.1834, 1.2166, 1.1921,  ..., 0.4263, 0.0000, 0.5839],
        [0.6158, 0.6508, 0.6311,  ..., 0.2936, 0.5839, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.53501777,  8.56889738,  8.5724042 ,  8.83725081,  8.91551231,
        8.95133961, 10.7506423 , 13.76047999, 12.05300943,  6.19707944,
        8.89458273, 11.37939695, 12.62711252, 11.87412439,  9.50913849,
        6.59176018,  9.47369628,  9.4997987 ,  5.24137909,  5.34250225,
       13.744387  ,  5.18047638,  5.47702766,  5.57341623, 12.78327037,
        8.41171369, 12.00995898,  8.40445872, 12.40074901, 10.38897563,
       12.52875793, 12.21532282,  6.65368551, 12.79126785,  7.20482224,
        9.17595899,  6.18668463,  6.83083863,  8.16584537, 12.19805394,
       11.80092578,  8.62631801,  7.02253235,  7.58668144,  4.96231532,
       13.50484539,  8.30125201,  5.94535529, 10.84437705,  4.8763135 ,
       13.59170809,  9.97964013, 11.05672136, 12.87752052,  4.90505566,
        7.52117132, 13.12299145, 10.31613269, 12.17451685,  7.62232307,
        9.15976746,  5.14371005,  6.67121748,  5.98285319,  5.14371005,
       10.31886066, 13.14690137,  6.97106912, 11.80092578,  5.53175275,
        9.58072286, 13.96746816,  9.77777346,  7.48311002, 12.4013153 ,
        9.17016857,  5.91854397, 13.59480906,  9.09963894,  4.978704  ,
       11.51811603,  8.53501777, 10.52587243,  5.1224079 ,  8.19703512,
        5.79391825, 13.89671314, 12.41012046,  5.05989547,  8.30729352,
        5.46244988, 14.08549802, 12.10942412,  5.5671389 , 10.27572673,
       12.4217523 , 12.52875793, 12.77402351,  8.04664886,  9.48337063,
        9.93553424, 10.46739073, 10.39942497,  5.13125513, 13.36800073,
        6.42689705])}
{'P': tensor([[1.7730e-02, 2.3756e-01, 2.1729e-01,  ..., 3.1671e-07, 2.5258e-05,
         1.9509e-04],
        [2.1870e-04, 1.6957e-02, 2.4578e-01,  ..., 1.7044e-07, 2.7015e-05,
         1.3681e-04],
        [3.0077e-04, 1.2293e-03, 3.2501e-02,  ..., 3.1901e-07, 5.5272e-05,
         3.0353e-04],
        ...,
        [2.6968e-05, 1.1360e-05, 1.0616e-05,  ..., 1.3762e-02, 1.3392e-06,
         2.0890e-05],
        [7.2188e-04, 4.1730e-04, 3.6903e-04,  ..., 4.5189e-04, 1.4764e-01,
         3.2136e-05],
        [3.9199e-05, 6.3636e-05, 8.8681e-05,  ..., 1.8798e-06, 3.2213e-05,
         7.5379e-02]]), 'X': tensor([[ -5.5525,   6.5146,  -4.4984],
        [ -7.0825,   5.2073,  -5.0497],
        [ -7.4602,   4.7718,  -4.6277],
        [ -7.6734,   4.4980,  -5.4248],
        [ -7.9298,   4.1367,  -3.9773],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.1271, -10.8343,   2.6442],
        [  0.7685,  -4.8098,  -2.9310],
        [ -8.5149,  -5.1905,   1.5239],
        [  4.4633,   5.6287,  -0.7754],
        [ -2.2508, -10.5479,  -2.7235],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.2454,   7.9127,  -0.5715],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -8.0087,   4.0173,   9.1091],
        [  2.2659,  13.7894,   6.5436],
        [ -7.2986,  -7.1225,  -3.0814],
        [  6.2883,   0.0325,   5.2692],
        [ -8.6229,   2.9068,   2.4429],
        [ 10.6283,  -5.7253,   7.4878],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.4739,  -1.2497,   0.5483],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -3.9508,  -9.8515,   1.7923],
        [ -8.8058,   2.4850,  -1.7425],
        [ -6.0844,  -8.3955,   0.7705],
        [ -5.3382,   6.6533,   7.8054],
        [ -2.6735, -10.4062,  -1.5911],
        [  9.5203,   9.3483,   5.8781],
        [  0.9809,   7.6642,  -1.4217],
        [  9.3753,   9.5171,   3.9131],
        [  4.5503,   5.5359,   7.4735],
        [ 12.0582,  -2.6640,   0.7132],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.1892, -10.8220,  -6.8409],
        [ 10.9916,   7.2133,  -0.9179],
        [  2.3261,   7.1726,   6.2522],
        [ -2.1382,   7.8339,   7.3524],
        [ -0.4341, -10.9473,   7.4509],
        [  5.1519, -10.2387,   9.8191],
        [ -3.5029,   7.5138,   8.1853],
        [  4.6357,   5.4415,   7.9067],
        [  5.8889,  -1.3916,   8.8993],
        [  4.9531,   5.0576,   5.6794],
        [  3.1271, -10.8343,   2.6442],
        [  0.5250, -11.0306,   5.5358],
        [ -5.1377,   6.7751,  -0.4438],
        [  3.2701,   6.6338,   9.9209],
        [ -8.3717,  -5.4769,  -9.2228],
        [  9.8082,  -6.8396,   4.9615],
        [  0.6396,  14.0772,  -6.3786],
        [ 12.3453,  -1.5462,   7.0212],
        [ -4.0868,   7.2981,   8.1874],
        [ -3.1671,   7.6158,   6.0765],
        [ -9.3331,  -2.6465,  -8.2028],
        [  3.7992,  -3.9474,  -8.9019],
        [ -3.2736, -10.1703,   3.8712],
        [ -9.1813,  -3.3550,   1.8827],
        [  7.0515,  11.6194,   8.4289],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.8093,  -9.4178,   8.4244],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -3.5903,   7.4847,  -8.9112],
        [ -0.0410,   7.8591,   9.3457],
        [  9.2692,   9.6370,   7.0746],
        [  1.6308,  13.9254,  -2.0984],
        [  2.1504,  -4.6726,   8.7908],
        [  6.3516,   0.6425,   0.3425],
        [  6.1737,   2.5305,   3.9049],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -7.2515,   5.0195,   4.6434],
        [  9.2692,   9.6370,   7.0746],
        [ -7.2515,   5.0195,   4.6434],
        [  6.3236,   0.3011,  -4.6668],
        [  2.0419,  -4.6978,  -5.0983],
        [ -4.2040,   7.2486,   2.9875],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.3804,  -2.3504,   7.2744],
        [ 11.7418,   5.6070,  -1.4192],
        [  0.9945,  -4.8144,  -0.1897],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.0305,   3.0296,   0.1375],
        [ -2.1533, -10.5778,   1.0112],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.5900,   2.9767,   7.4811],
        [  7.1441,  11.5529,   2.3232],
        [  7.5930,  -8.9037,   7.2323],
        [ -2.6485, -10.4151,   0.8766],
        [  9.9750,  -6.6338,  -1.2440],
        [  0.3724,   7.7974,  -6.8722],
        [ -8.6426,  -4.9136,   9.4941],
        [ -0.1626, -10.9796,   6.4629],
        [ -3.2053, -10.1993,  -3.8736],
        [  0.4996,   7.7738,  -4.4711],
        [  5.4167, -10.1293,   6.2156],
        [  9.5018,  -7.1945,  -1.1892],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.7228,  -4.7575,   8.3396],
        [  1.8719,   7.3713,  -3.7465],
        [  9.4925,   9.3811,   8.9842],
        [  5.2911,  -2.4824,  -3.6877],
        [ -9.1706,   1.3863,   8.4032]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 1.1610e-01, 1.5690e-01,  ..., 1.4150e+00, 5.1160e-01,
         9.2637e-01],
        [1.1610e-01, 0.0000e+00, 5.2273e-02,  ..., 1.5143e+00, 6.2596e-01,
         9.3552e-01],
        [1.5690e-01, 5.2273e-02, 0.0000e+00,  ..., 1.5276e+00, 6.6834e-01,
         9.0420e-01],
        ...,
        [1.4150e+00, 1.5143e+00, 1.5276e+00,  ..., 0.0000e+00, 1.1006e+00,
         1.1918e+00],
        [5.1160e-01, 6.2596e-01, 6.6834e-01,  ..., 1.1006e+00, 0.0000e+00,
         1.1337e+00],
        [9.2637e-01, 9.3552e-01, 9.0420e-01,  ..., 1.1918e+00, 1.1337e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.55982105,  8.7907923 ,  8.85572292,  8.89458273,  8.94395628,
        8.8353674 , 11.27656488,  4.87083348,  9.97220799,  7.18354885,
       10.78533838,  7.96181288,  8.01009083,  9.69629   ,  8.95983586,
       13.97430191, 10.1979669 ,  6.28835101,  9.09963894, 12.07225469,
        7.71075226,  9.55593374,  9.85330646, 10.61417181,  9.14973319,
       10.36844813,  8.53014915, 10.74409834, 13.3426541 ,  7.72668467,
       13.35927527,  7.16598273, 12.3489324 ,  9.88310614, 11.28215429,
       13.1471313 ,  7.5403822 ,  8.12043983, 10.95593729, 11.46176259,
        8.29022197,  7.14837514,  6.05113217,  7.07902036, 11.27656488,
       11.04313641,  8.50278896,  7.39599986, 10.00410625, 11.95743558,
       14.09176259, 12.44177641,  8.36445758,  8.24808832,  9.70108409,
        5.47866221, 10.68416401,  9.7751189 , 13.59170809,  9.94285511,
       12.30695098, 11.62158554, 11.64746882,  9.69792513,  8.30125201,
        7.85919687, 13.37121923, 14.02058621,  5.14371005,  6.38399789,
        6.67218687, 12.22276278,  9.60080639,  8.81930525, 13.37121923,
        8.81930525,  6.33075682,  5.1224079 ,  8.37954853, 12.19805394,
        5.87132718, 13.01187484,  4.9160962 ,  9.64157461,  9.41605684,
        6.74872981, 10.794755  , 13.08414689,  9.09119351, 13.58333857,
       11.70168755, 10.74655346, 11.97949273,  7.80625981,  9.94173708,
       10.98076873, 10.6910768 ,  7.78980299, 11.48663767, 11.9182956 ,
        5.9510489 ,  5.05981914,  7.60529892, 13.34587027,  5.84450127,
        9.27474386])}
{'P': tensor([[1.8509e-02, 1.8864e-01, 1.6646e-01,  ..., 5.1562e-05, 7.4649e-07,
         1.7105e-05],
        [1.7066e-04, 2.2344e-02, 1.7488e-03,  ..., 1.0173e-04, 1.0068e-06,
         1.2839e-05],
        [1.7680e-04, 5.9736e-02, 1.3881e-02,  ..., 7.6133e-05, 5.2671e-07,
         9.1525e-06],
        ...,
        [6.4548e-05, 5.6169e-05, 7.1404e-05,  ..., 9.1204e-02, 7.9611e-05,
         1.1067e-03],
        [2.6548e-05, 8.0730e-06, 8.9815e-06,  ..., 2.3276e-06, 3.2572e-02,
         1.3336e-03],
        [6.5589e-03, 1.9805e-03, 2.8802e-03,  ..., 1.7346e-04, 4.3622e-05,
         7.9551e-02]]), 'X': tensor([[ -5.6174,   6.4708,  -1.7339],
        [ -7.7842,   4.3466,  -1.7548],
        [ -7.3554,   4.8985,  -0.6665],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.6650,   4.5092,   0.2520],
        [ -8.6050,   2.9451,  -0.7012],
        [  8.0381,  -8.5690,  -8.9523],
        [  4.0362,  13.2397,  -7.0030],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.6232,  -4.7717,  -2.5655],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.9193,  -1.3176,   5.1739],
        [ -1.8823, -10.6558,  -3.8952],
        [ 12.5763,   0.1333,  -9.2650],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.0348,  -2.8229,   3.5436],
        [ 11.7162,   5.6728,  -8.9372],
        [  3.3943,   6.5483,   1.0790],
        [  8.5198,  10.4091,  -8.6486],
        [  6.8093,  -9.4178,   8.4244],
        [  6.0467,  -9.8369,  -6.5816],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -7.0948,   5.1940,   9.8171],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.1120,  -4.6818,   8.6144],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.2199,   4.6861,  -1.9304],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.0870,  -4.6877,   9.6823],
        [  4.2339,  -3.6268,   6.3125],
        [  2.3373,  -4.6232,  -5.2246],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.2809,  13.5061,  -4.0199],
        [  6.3031,   1.8477,  -7.1706],
        [ -8.3535,  -5.5118,   9.8399],
        [  7.3592,  11.3937,   6.2407],
        [ -6.3306,   5.9272,   5.6935],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.8930,  12.3592,  -8.9201],
        [  5.2911,  -2.4824,  -3.6877],
        [  9.4516,  -7.2501,  -1.8868],
        [  5.6260,  -1.9401,   3.0259],
        [  3.8111, -10.6787,  -4.6539],
        [ 10.0567,   8.6697,   4.6329],
        [ 10.6283,  -5.7253,   7.4878],
        [  4.9957,  -2.8705,   6.5978],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.2630,   2.1057,  -5.2136],
        [  2.3367,   7.1676,  -9.5263],
        [  2.1440, -10.9811,   8.5281],
        [  2.2644,  -4.6434,   9.7508],
        [  6.2036,   2.4030,  -1.2500],
        [ 10.9394,   7.3069,  -8.9365],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.1737,   2.5305,   3.9049],
        [  0.4119,  14.1019,  -4.1870],
        [  5.8836,   3.4315,  -4.0327],
        [  0.9445,   7.6736,   0.5095],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.0000,   7.6591,  -4.5011],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -7.3554,   4.8985,  -0.6665],
        [  0.3354,  -4.7708,  -4.9125],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.1047, -11.0046,   0.5188],
        [  6.2026,  12.1774,  -1.5887],
        [ -5.0550,   6.8231,  -9.2502],
        [  2.5620,   7.0549,   8.4825],
        [  2.2437,  -4.6489,   6.1430],
        [  6.3551,   0.7149,   1.3344],
        [  5.7845,   3.6652,   0.9175],
        [ -8.0766,  -6.0046,  -3.1873],
        [  2.2659,  13.7894,   6.5436],
        [ -1.9646,   7.8571,   3.3817],
        [  0.4119,  14.1019,  -4.1870],
        [  9.8082,  -6.8396,   4.9615],
        [  1.2728, -11.0372,   9.8121],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4117,   0.1160,   8.2574],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4735,  -1.2620,  -3.6742],
        [  6.0031,  12.2958,  -6.5992],
        [  6.3535,   0.6792,  -7.5101],
        [ -8.6229,   2.9068,   2.4429],
        [ -4.7334,   6.9982,  -0.4392],
        [ 11.2571,   6.7072,   8.3693],
        [  0.1532,  -4.7422,  -7.0145],
        [ -3.2053, -10.1993,  -3.8736],
        [  2.4697,  -4.5836,   1.4593],
        [  6.3604,   0.9042,  -6.5901],
        [  6.0890,  -0.8382,   2.9166],
        [ -2.6069, -10.4298,  -6.1995],
        [ -8.9942,  -3.9990,   6.0448],
        [ -1.5451,   7.8975,   7.6104],
        [  2.5636, -10.9293,   9.6519],
        [ -4.3963,   7.1629,   4.5263],
        [  0.1931,  -4.7491,   7.4385],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.1890,   4.7318,   0.9515]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1696, 0.1284,  ..., 0.7340, 0.7521, 0.5241],
        [0.1696, 0.0000, 0.0914,  ..., 0.8554, 0.8893, 0.6881],
        [0.1284, 0.0914, 0.0000,  ..., 0.7650, 0.8796, 0.6159],
        ...,
        [0.7340, 0.8554, 0.7650,  ..., 0.0000, 1.0362, 0.3984],
        [0.7521, 0.8893, 0.8796,  ..., 1.0362, 0.0000, 0.6385],
        [0.5241, 0.6881, 0.6159,  ..., 0.3984, 0.6385, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.56889738,  8.91551231,  8.83725081,  8.95133961,  8.89301516,
        9.09502385, 11.74901778, 13.84126197, 13.20988996,  5.04028538,
       12.3489324 ,  5.18047638,  6.06416957, 10.8207297 , 12.5769659 ,
        5.27178363, 12.3828957 , 12.67576792,  5.77219077, 13.01728313,
        7.37576154, 13.45124917, 11.62158554, 11.54672755, 12.52372629,
        8.79284466, 12.3828957 ,  5.13616224, 11.75618513,  7.01475376,
       10.61417181,  5.13125513,  5.5748624 ,  5.18047638,  9.47369628,
       13.89886469,  6.56833864, 10.00802627, 13.56367857,  8.67227386,
       11.80092578, 13.6922425 ,  5.84450127, 11.9120321 ,  5.95111804,
       11.33837396, 13.2778378 , 12.07225469,  5.76165934,  9.85330646,
        9.48969978,  9.44486609,  6.60752373,  7.53884104, 11.18839169,
        5.16611782,  6.652742  , 13.15525803,  9.14973319,  6.67218687,
       14.10796466,  6.81117802,  7.73151309, 10.91805683,  7.72415048,
       10.61047596,  8.83725081,  4.78258092, 10.1979669 ,  9.36093562,
       11.00508681, 13.6660671 ,  8.49161154,  7.50564223,  5.16205087,
        6.39520397,  6.84796202, 10.06407839, 13.97430191,  8.09899874,
       14.10796466, 11.95743558, 11.11038551,  9.60080639,  9.41244865,
       11.05413545,  9.41244865,  9.55720651, 13.6829843 ,  6.38968357,
        9.09963894,  8.4486771 , 13.103709  ,  4.74468406, 10.6910768 ,
        5.20659892,  6.42439344,  6.1463896 , 10.7506423 ,  9.84314712,
        8.04718433, 11.22596994,  8.40445872,  4.75303127, 12.46513646,
        7.02253235])}
{'P': tensor([[2.1227e-02, 2.6137e-01, 2.0291e-01,  ..., 1.2114e-06, 2.3434e-05,
         2.4865e-05],
        [2.7117e-04, 2.4658e-02, 1.0612e-01,  ..., 8.5606e-07, 1.7436e-05,
         1.6035e-05],
        [6.9574e-04, 6.7958e-03, 8.6829e-02,  ..., 2.4456e-06, 5.1726e-05,
         5.3890e-05],
        ...,
        [2.0252e-04, 7.3815e-05, 4.4654e-05,  ..., 9.2850e-03, 1.1774e-03,
         1.8930e-04],
        [1.3518e-02, 5.6393e-03, 3.7227e-03,  ..., 2.3353e-03, 8.8940e-02,
         7.3126e-03],
        [2.3628e-02, 8.8093e-03, 8.5136e-03,  ..., 5.0446e-04, 5.6883e-03,
         1.5133e-01]]), 'X': tensor([[ -5.6423,   6.4537,  -1.2091],
        [ -7.3554,   4.8985,  -0.6665],
        [ -7.7842,   4.3466,  -1.7548],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.6650,   4.5092,   0.2520],
        [ -6.6998,   5.5954,   0.0823],
        [  6.7792,  11.8084,   1.6251],
        [ -3.4488, -10.0934,   2.5195],
        [  6.2630,   2.1057,  -5.2136],
        [ -3.1671,   7.6158,   6.0765],
        [  0.9967,  -4.8144,  -9.7370],
        [  7.9403,  -8.6455,  -8.7433],
        [  6.6319,  11.9066,   3.5165],
        [  1.1274,  14.0116,   8.1753],
        [  2.0899,  -4.6870,  -7.4260],
        [ -0.1963,   7.8765,  -1.1371],
        [ -6.4368,   5.8356,   3.3014],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.3551,   0.7149,   1.3344],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.3261,   7.1726,   6.2522],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.9515,  -3.8425,   6.1729],
        [ -5.9872,   6.2037,  -9.4477],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.2247,  -0.3150,  -9.3657],
        [  1.1139,   7.6281,  -2.5680],
        [  6.3211,   1.7017,  -0.2164],
        [  1.3705,  13.9723,   7.9266],
        [ 10.9916,   7.2133,  -0.9179],
        [  9.6614,   9.1784,  -1.2087],
        [  6.8093,  -9.4178,   8.4244],
        [ -1.9646,   7.8571,   3.3817],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.4732,  -2.2041,  -4.3830],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.8111, -10.6787,  -4.6539],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.2849, -10.9655,  -7.6542],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.2692,  -1.8882,   3.9253],
        [ -6.2200,  -8.2737,   5.5079],
        [  5.4167, -10.1293,   6.2156],
        [  2.1120,  -4.6818,   8.6144],
        [  1.6954,  -4.7616,   9.8198],
        [  7.0664,  -9.2590,   8.0879],
        [  5.2911,  -2.4824,  -3.6877],
        [ -3.0644,   7.6439,  -1.4322],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.2895,   0.0405,   5.7652],
        [  2.5636, -10.9293,   9.6519],
        [ -0.5739,   7.9056,   0.6382],
        [ 12.5224,  -0.4497,   4.8448],
        [ -9.2345,  -3.1345,   3.9715],
        [ 11.1420,   6.9331,  -2.4779],
        [  4.0457,  -3.7737,  -0.4704],
        [  8.2154,  -8.4262,   8.0171],
        [  7.0515,  11.6194,   8.4289],
        [  5.0277,   4.9589,   7.0175],
        [  2.2182,  13.8007,   5.1889],
        [ -5.3179,  -9.0101,   9.4472],
        [  6.2675,  12.1379,  -8.9720],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.3261,  13.7749,   5.9121],
        [ -4.0409,   7.3169,  -3.5779],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.9636,   2.0648,   8.1946],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 12.1504,   4.3674,  -7.2998],
        [  9.0004,  -7.7195,   4.6945],
        [ -9.3604,   0.4855,  -3.0158],
        [  9.9756,  -6.6330,  -3.3429],
        [  5.7963,  -1.6018,   1.8856],
        [ -4.5778,  -9.5013,  -5.0667],
        [  2.2076,  13.8032,  -1.5099],
        [ 11.5198,  -4.0923,   5.5088],
        [ -9.4697,  -0.6157,  -6.8472],
        [ 11.6019,   5.9540,  -5.1715],
        [  3.2366,  -4.2732,   5.1532],
        [  5.8930,  12.3592,  -8.9201],
        [  4.4633,   5.6287,  -0.7754],
        [  9.4516,  -7.2501,  -1.8868],
        [ -1.2454,   7.9127,  -0.5715],
        [  4.9337,  12.8566,  -3.6582],
        [  4.2339,  -3.6268,   6.3125],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.2576,   2.1359,  -3.3398],
        [  8.6420,  -8.0580,   6.5781],
        [ -7.0948,   5.1940,   9.8171],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.5224,  -0.4497,   4.8448],
        [ -2.0721,   7.8432,   6.6380],
        [  7.7794,  11.0629,   8.9069],
        [ -7.0120,   5.2825,   7.7762],
        [  4.9152,  12.8653,  -1.4459],
        [  7.5309,  11.2617,   7.2656],
        [  9.5203,   9.3483,   5.8781],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.1755,   2.5233,  -2.0042],
        [  5.3602,  -2.3808,   7.2098],
        [  7.0358,  -9.2784,  -9.8479],
        [  1.2273,  -4.8081,   9.8129],
        [  9.3753,   9.5171,   3.9131],
        [  4.3377,   5.7570,   1.3712],
        [  3.6703,   6.3443,  -5.4401]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1169, 0.1767,  ..., 0.9686, 0.4872, 0.4860],
        [0.1169, 0.0000, 0.0914,  ..., 1.0570, 0.5883, 0.5974],
        [0.1767, 0.0914, 0.0000,  ..., 1.1400, 0.6634, 0.6192],
        ...,
        [0.9686, 1.0570, 1.1400,  ..., 0.0000, 0.4948, 0.8625],
        [0.4872, 0.5883, 0.6634,  ..., 0.4948, 0.0000, 0.4360],
        [0.4860, 0.5974, 0.6192,  ..., 0.8625, 0.4360, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.5724042 ,  8.83725081,  8.91551231,  8.95133961,  8.89301516,
        8.72896381, 13.61600536, 10.66631784,  6.60752373,  8.24808832,
        4.91652556, 11.73848065, 13.62896593, 14.05687815,  5.13182506,
        7.8788965 ,  8.68832398, 10.52587243,  6.39520397, 12.9089559 ,
        7.5403822 ,  8.8353674 ,  5.51177501,  8.62162372,  5.48276038,
        6.23262289,  7.70897675,  6.54615887, 14.0393949 , 13.1471313 ,
       13.32613445, 11.62158554,  8.09899874, 10.52587243,  6.18668463,
        5.90035324, 11.89855267, 11.33837396, 13.25336001, 11.20100795,
       13.01728313, 12.41366754, 10.35093485, 11.48663767,  5.13616224,
        5.05444288, 11.64746882,  5.84450127,  8.23525433, 11.69264243,
        6.28961988, 11.22596994,  7.92644746, 12.53047489,  9.75200848,
       13.12299145,  5.53255882, 11.76830981, 13.59170809,  7.06169391,
       13.97779839, 10.46238054, 13.66051884,  9.75401449, 13.96987806,
        8.35856487,  6.02588517,  9.19837792, 10.60226614, 12.91143491,
       11.85743425,  9.37296102, 11.97957425,  6.01356004, 10.54658597,
       13.97857591, 12.22503754,  9.48969978, 13.04051744,  5.36061974,
       13.6922425 ,  7.18354885, 11.9120321 ,  8.01009083, 13.77075174,
        5.5748624 , 12.7125079 ,  6.61211224, 11.8159254 ,  8.79284466,
       13.04156075, 12.53047489,  8.11227668, 13.52430176,  8.77913428,
       13.77222752, 13.54774355, 13.3426541 ,  6.12886024,  6.67108662,
        5.86519525, 11.64436684,  4.96231532, 13.35927527,  7.20826219,
        7.32950385])}
{'P': tensor([[1.7653e-02, 1.9546e-01, 1.9230e-01,  ..., 4.5613e-05, 4.6655e-05,
         1.2973e-04],
        [2.3744e-04, 2.2815e-02, 2.2067e-02,  ..., 7.6423e-05, 7.8041e-05,
         2.5276e-04],
        [2.1300e-04, 2.0368e-02, 2.0289e-02,  ..., 6.7541e-05, 6.8923e-05,
         2.2388e-04],
        ...,
        [9.6580e-05, 6.1304e-05, 6.0989e-05,  ..., 5.8361e-02, 1.5141e-02,
         4.7061e-03],
        [8.4236e-05, 5.4771e-05, 5.4465e-05,  ..., 1.2284e-01, 5.7902e-02,
         5.4184e-03],
        [1.7677e-04, 1.3461e-04, 1.3424e-04,  ..., 7.0986e-02, 4.4236e-02,
         9.1443e-02]]), 'X': tensor([[ -5.6651,   6.4380,   1.3660],
        [ -7.4808,   4.7462,   2.0266],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.5720,   4.6310,   2.1717],
        [ -7.6650,   4.5092,   0.2520],
        [ -6.6998,   5.5954,   0.0823],
        [ -2.2508, -10.5479,  -2.7235],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 12.4691,   2.8525,  -5.5110],
        [  8.5093,  -8.1764,  -3.2381],
        [  6.3048,   0.1469,  -9.1691],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.2974,   0.8371,   2.4852],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -8.6229,   2.9068,   2.4429],
        [  2.2849, -10.9655,  -7.6542],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.2038,   2.4020,  -3.1692],
        [  9.8082,  -6.8396,   4.9615],
        [  6.0305,   3.0296,   0.1375],
        [  5.4242, -10.1261,   3.9595],
        [ -9.4309,  -0.0599,  -4.0830],
        [  9.3624,   9.5318,   0.2779],
        [ 12.2914,   3.8037,   6.0978],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.5198,  -4.0923,   5.5088],
        [  2.0729,   7.2878,   7.0880],
        [  6.2535,  -0.1709,  -9.4102],
        [ -3.1204, -10.2345,   9.2288],
        [ 11.7795,  -3.4697,   3.5624],
        [ -8.5859,  -5.0393,   0.5476],
        [  4.5325,  -3.3643,   6.7699],
        [ -1.0012, -10.8576,   2.2867],
        [  6.0467,  -9.8369,  -6.5816],
        [  9.2979,   9.6048,   7.3138],
        [  6.9284,  11.7061,  -0.3732],
        [ -0.8455, -10.8853,  -6.7400],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.2373,  -4.6506,  -1.7334],
        [ -3.0008, -10.2827,  -8.9480],
        [ -9.3604,   0.4855,  -3.0158],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.1271, -10.8343,   2.6442],
        [  5.6202,  -1.9508,   0.7471],
        [  4.6063,  13.0051,  -3.1456],
        [  4.5503,   5.5359,   7.4735],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.6035,   0.7494,   7.1536],
        [  3.9085,   6.1512,  -2.2560],
        [  5.8902,  -1.3887,   4.0388],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.7685, -10.8982,  -6.8928],
        [  9.2979,   9.6048,   7.3138],
        [  9.4554,   9.4245,   9.8800],
        [ -1.7009,   7.8851,  -8.2437],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -3.0008, -10.2827,  -8.9480],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -8.8835,  -4.3178,   9.3080],
        [ -6.4822,  -8.0252,  -3.8983],
        [  1.1350,  14.0104,   0.6321],
        [ -2.6485, -10.4151,   0.8766],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.0587, -10.6112,   6.2273],
        [  0.8940, -11.0402,  -0.9783],
        [ 11.8951,  -3.1570,   7.0742],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.3963,   7.1629,   4.5263],
        [  0.9478,  -4.8144,   0.6371],
        [  7.9816,  10.8938,  -0.5378],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.4735,  -0.7274,  -2.6749],
        [  8.2154,  -8.4262,   8.0171],
        [  7.0515,  11.6194,   8.4289],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.2707,   0.9661,   4.4863],
        [ -2.6735, -10.4062,  -1.5911],
        [ -8.6675,  -4.8569,  -9.1769],
        [  0.6363, -11.0348,   6.1414],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3542,   0.6934,  -3.3840],
        [  4.5340,  -3.3629,   8.9652],
        [ -1.8788, -10.6567,   3.7068],
        [ -3.0004,   7.6607,  -7.2640],
        [ -7.6525,  -6.6534,   5.1721],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 12.4013,  -1.2597,  -7.6474],
        [  9.0051,   9.9234,   5.4411],
        [ -8.8363,   2.4086,  -5.6296],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.5600, -11.0321,  -0.8660],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -0.5709,   7.9055,   7.6089],
        [  5.1787,  -2.6383,  -4.2533],
        [  6.3550,   0.7129,   0.4917],
        [  2.0243,   7.3086,   7.5432],
        [ -2.2508, -10.5479,  -2.7235],
        [  9.5203,   9.3483,   5.8781],
        [ 12.4877,  -0.7222,   8.8347],
        [ 11.2571,   6.7072,   8.3693],
        [  2.7420,  -4.4896,  -3.7139],
        [  2.4149,  -4.6005,  -4.3339],
        [  0.9659,  -4.8144,  -0.9594]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1319, 0.1311,  ..., 0.5741, 0.5864, 0.4449],
        [0.1319, 0.0000, 0.0038,  ..., 0.7028, 0.7139, 0.5767],
        [0.1311, 0.0038, 0.0000,  ..., 0.7027, 0.7140, 0.5760],
        ...,
        [0.5741, 0.7028, 0.7027,  ..., 0.0000, 0.0257, 0.1696],
        [0.5864, 0.7139, 0.7140,  ..., 0.0257, 0.0000, 0.1927],
        [0.4449, 0.5767, 0.5760,  ..., 0.1696, 0.1927, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.57560275,  8.859416  ,  8.85939626,  8.87587457,  8.89301516,
        8.72896381, 10.78533838, 11.97957425, 10.00410625, 12.79126785,
       11.80092578,  6.30647746,  5.80312949,  9.3349858 , 10.60226614,
        9.09963894, 11.20100795, 10.52587243,  6.65259775, 11.95743558,
        6.74872981, 11.48734978,  9.43112957, 13.3607379 , 12.86648771,
       12.07818871, 12.22503754,  7.57685908,  6.25586297, 10.69963954,
       12.27991991,  9.95552626,  5.64464807, 10.9036226 , 11.54672755,
       13.36800073, 13.60275211, 10.91805683,  5.22903117,  5.16077764,
       10.71162681,  9.37296102, 13.00413829, 11.27656488,  5.94909709,
       13.79676654,  7.16598273, 12.00995898, 12.62576026,  7.28793663,
        6.05165248, 11.9120321 , 11.24434374, 13.36800073, 13.35013374,
        8.06643487,  9.50654967, 10.71162681,  5.35774573, 10.06437598,
        9.87720102, 10.31613269, 14.05633441, 10.74655346, 13.09737053,
       11.36089355, 11.07637531, 12.30695098, 10.60226614,  8.40445872,
        4.90676242, 13.50484539,  6.80559669,  9.5014146 , 11.76830981,
       13.59170809, 10.4977132 ,  9.66977802,  9.32094055, 10.74409834,
        9.93553424, 11.05317597,  5.5366698 ,  6.39187285,  5.64500512,
       10.82106505,  8.22728399, 10.14044489, 10.31613269, 12.46513646,
       13.40024224,  9.15866607, 13.00380624, 11.04629471,  9.4997987 ,
        7.92607342,  5.81199422,  6.3949043 ,  7.58378655, 10.78533838,
       13.3426541 , 12.5085995 , 13.103709  ,  5.26067019,  5.19577146,
        4.91038884])}
{'P': tensor([[3.9277e-02, 2.3538e-01, 1.8177e-01,  ..., 1.3118e-06, 2.5310e-05,
         5.7967e-07],
        [4.0601e-04, 1.8387e-02, 2.7833e-01,  ..., 4.4301e-07, 3.4156e-05,
         5.9630e-07],
        [1.5736e-04, 2.7462e-04, 3.0490e-02,  ..., 3.4383e-07, 1.1807e-04,
         1.6816e-06],
        ...,
        [1.8898e-04, 2.6386e-04, 1.3786e-04,  ..., 8.2497e-02, 3.4892e-06,
         4.9371e-07],
        [3.8285e-07, 5.5517e-07, 6.8092e-07,  ..., 3.2428e-07, 1.4728e-02,
         6.1824e-03],
        [5.3601e-07, 5.7842e-07, 4.2049e-07,  ..., 2.2034e-05, 1.1937e-05,
         5.6466e-02]]), 'X': tensor([[-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0920, 0.2256,  ..., 1.4919, 0.8572, 1.1736],
        [0.0920, 0.0000, 0.1365,  ..., 1.5279, 0.8110, 1.1695],
        [0.2256, 0.1365, 0.0000,  ..., 1.6245, 0.8018, 1.2193],
        ...,
        [1.4919, 1.5279, 1.6245,  ..., 0.0000, 1.2103, 0.6792],
        [0.8572, 0.8110, 0.8018,  ..., 1.2103, 0.0000, 0.5794],
        [1.1736, 1.1695, 1.2193,  ..., 0.6792, 0.5794, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.62162372,  8.83341608,  9.09661212,  9.17905491,  8.92395381,
        9.14044984, 13.14690137, 10.91601655, 10.19036081,  9.41605684,
       13.36800073, 12.62711252,  6.23262289,  9.97220799,  7.73151309,
        9.91668182, 14.10796466, 10.66631784, 10.27716434,  6.81084724,
       11.42947632, 13.59170809,  9.50654967, 11.1341798 ,  5.79391825,
        8.86082124, 10.87409005, 10.35464916,  4.98461983,  9.36256096,
        6.12886024, 12.25806559,  6.39520397, 11.27160035, 13.06957425,
        8.0550286 , 10.61047596,  4.92497514, 10.1371552 ,  5.52416368,
        9.91668182,  6.24614905, 11.51557166, 12.18045978, 10.48315436,
        7.06169391, 14.09357547,  5.69913817,  9.34061938,  7.54127425,
       14.02526505,  8.23525433, 13.04051744,  9.53015376,  9.55720651,
        4.78002665,  6.53250937,  5.18277858, 11.00508681, 13.34587027,
       11.54071859, 11.73567759,  7.98469166,  8.79284466,  5.19577146,
        7.5092834 ,  5.14124612, 13.66991027,  9.38232121,  9.04255655,
        7.13501264,  7.50500064, 13.99877413, 14.05633441, 13.13205357,
        9.27474386,  5.79391825, 10.14044489,  9.78286733,  6.49538841,
        8.67502348,  7.96181288,  9.04255655,  7.61088125, 13.37121923,
       11.00812996,  5.53175275,  5.05981914, 13.6105595 ,  6.59176018,
        9.58072286, 10.05558893,  9.12220271,  7.19534146,  6.19633139,
        7.71075226,  7.03733775, 12.91143491, 10.82106505, 12.4980845 ,
       11.42420508,  4.978704  ,  6.07992623, 13.95472331, 10.68416401,
       11.62158554])}
{'P': tensor([[1.3156e-02, 2.3280e-01, 1.9842e-01,  ..., 2.6085e-02, 4.5940e-04,
         4.1229e-06],
        [6.7564e-04, 1.8302e-02, 2.7976e-01,  ..., 4.2615e-02, 4.0476e-04,
         6.0577e-06],
        [4.1111e-04, 5.4859e-04, 3.5898e-02,  ..., 1.9753e-01, 2.8414e-04,
         1.0771e-05],
        ...,
        [1.9742e-04, 2.2631e-04, 3.7882e-04,  ..., 1.0176e-01, 1.1618e-04,
         3.6329e-05],
        [5.1380e-02, 7.6481e-02, 5.6446e-02,  ..., 1.2338e-02, 2.5425e-02,
         1.3981e-05],
        [1.1827e-04, 1.1396e-04, 6.6504e-05,  ..., 2.2665e-05, 3.7908e-04,
         7.4858e-02]]), 'X': tensor([[ -5.9940,   6.1985,   5.9566],
        [ -6.3306,   5.9272,   5.6935],
        [ -7.4733,   4.7555,   6.2490],
        [ -7.8126,   4.3066,   5.1791],
        [ -7.8359,   4.2735,   7.2943],
        [ -6.3489,   5.9117,   6.5161],
        [  4.0587, -10.6112,   6.2273],
        [ -0.8675, -10.8815,   6.8561],
        [ -1.8788, -10.6567,   3.7068],
        [  7.8425,  -8.7202,   6.4530],
        [ 11.4495,   6.3026,   6.8257],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.8835,  -4.3178,   9.3080],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.3301,   0.3638,  -7.5685],
        [  0.7145,  -4.8071,  -2.3766],
        [  9.6662,   9.1725,   8.1056],
        [  1.8447,  -4.7372,   8.1235],
        [ -0.1963,   7.8765,  -1.1371],
        [  3.9591, -10.6391,  -6.2557],
        [  9.1409,  -7.5789,   1.5814],
        [  4.1637,  13.1898,  -1.9189],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.3505,   0.6235,  -8.5849],
        [  3.6154,  -4.0640,  -7.9698],
        [ -9.4624,  -0.4633,  -7.3918],
        [  2.9470,  13.6085,   2.9809],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.6202,  -1.9508,   0.7471],
        [ -5.6169,  -8.7844,   8.9057],
        [  0.7958,  -4.8109,  -8.2248],
        [  5.6414, -10.0303,   7.8525],
        [ -7.6734,   4.4980,  -5.4248],
        [ -9.2707,   0.9661,   4.4863],
        [  2.5974,   7.0363,  -7.7810],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.6229,   2.9068,   2.4429],
        [  2.0899,  -4.6870,  -7.4260],
        [  1.5653,  13.9377,   4.1773],
        [ -8.7544,   2.6096,   3.3146],
        [  5.2911,  -2.4824,  -3.6877],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.3413,   0.4883,  -6.8416],
        [  6.3048,   0.1469,  -9.1691],
        [  1.1274,  14.0116,   8.1753],
        [ -9.0604,   1.7707,   1.1066],
        [  9.0051,   9.9234,   5.4411],
        [ -8.4267,  -5.3697,   2.4172],
        [  9.2811,  -7.4336,   5.1771],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.6319,  11.9066,   3.5165],
        [  0.5960, -11.0335,   3.1144],
        [  1.9204,   7.3518,   8.4781],
        [  3.3744,   6.5623,  -6.3636],
        [ -4.3359,  -9.6431,  -9.0616],
        [  0.5600, -11.0321,  -0.8660],
        [ -8.4267,  -5.3697,   2.4172],
        [  3.6469,   6.3624,  -1.8414],
        [  6.2247,  -0.3150,  -9.3657],
        [ -9.1178,   1.5784,   2.1410],
        [ -6.4464,   5.8272,   7.5038],
        [  6.2804,   2.0019,   0.8607],
        [ -4.4520,   7.1370,  -8.7867],
        [ -5.2752,  -9.0409,  -8.2568],
        [  7.0358,  -9.2784,  -9.8479],
        [  1.3103,  -4.8032,  -0.9300],
        [ -4.6393,   7.0461,   1.1624],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.3542,   0.6934,  -3.3840],
        [  4.0591,   6.0202,  -1.6339],
        [  5.1196,  -2.7160,   4.2705],
        [ -3.2736, -10.1703,   3.8712],
        [  2.4149,  -4.6005,  -4.3339],
        [  5.1142,  -2.7230,  -4.1966],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.7685,  -4.8098,  -2.9310],
        [  4.6231,   5.4556,   8.4973],
        [  0.1460,   7.8340,  -9.4614],
        [  5.1519, -10.2387,   9.8191],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -6.9838,  -7.4965,   0.9796],
        [ -5.1369,   6.7755,   8.3836],
        [ -7.9925,  -6.1423,   2.8990],
        [  5.1142,  -2.7230,  -4.1966],
        [ -5.1377,   6.7751,  -0.4438],
        [ -9.4501,  -1.7133,   2.2980],
        [  0.9392,  -4.8143,  -3.4649],
        [ -5.3382,   6.6533,   7.8054],
        [  4.4633,   5.6287,  -0.7754],
        [  5.1886,  -2.6250,   6.4677],
        [ -3.0004,   7.6607,  -7.2640],
        [ 12.5224,  -0.4497,   4.8448],
        [  9.0051,   9.9234,   5.4411],
        [  5.9526,   3.2525,  -4.3821],
        [ -2.6735, -10.4062,  -1.5911],
        [  1.7815,  -4.7481,   4.4289],
        [ -6.4368,   5.8356,   3.3014],
        [  8.7131,  10.2216,   7.7915],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.3444,   0.5821,   6.9101],
        [ -4.4503,   7.1378,   3.3751],
        [  6.1066,  -0.7799,  -4.1746]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0419, 0.1221,  ..., 0.3228, 0.1848, 0.8756],
        [0.0419, 0.0000, 0.0906,  ..., 0.2912, 0.1840, 0.8796],
        [0.1221, 0.0906, 0.0000,  ..., 0.2013, 0.2633, 0.9562],
        ...,
        [0.3228, 0.2912, 0.2013,  ..., 0.0000, 0.4470, 1.1200],
        [0.1848, 0.1840, 0.2633,  ..., 0.4470, 0.0000, 0.6957],
        [0.8756, 0.8796, 0.9562,  ..., 1.1200, 0.6957, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.62260327,  8.67227386,  8.8580671 ,  8.92098752,  8.92549752,
        8.67502348, 11.36089355, 10.91601655, 10.82106505, 11.72802491,
       13.06957425,  9.66977802, 11.95743558,  9.87720102, 10.412299  ,
        6.34059021,  4.85995002, 13.32557133,  5.08372149,  7.8788965 ,
       11.35182589, 11.87412439, 13.83139024,  5.5205189 ,  6.38105015,
        5.43943791,  9.47369628, 13.92390689,  5.94541803,  5.94909709,
       10.42667093,  4.8763135 , 11.50791859,  8.89458273,  9.32094055,
        7.50035355,  8.95983586,  9.09963894,  5.13182506, 14.02532758,
        9.13507321,  5.84450127, 13.00413829,  6.36004393,  6.30647746,
       14.05687815,  9.23178204, 13.40024224,  9.99211663, 11.89105259,
        5.16077764, 13.62896593, 11.04953767,  7.5984809 ,  7.37902766,
       10.57302546, 11.04629471,  9.99211663,  7.33350642,  6.23262289,
        9.25336371,  8.68979052,  6.59176018,  8.41171369, 10.46739073,
       11.64436684,  4.978704  ,  8.43626114,  9.62958973,  6.39187285,
        7.26074925,  5.79543038, 10.68416401,  5.19577146,  5.79391825,
        6.14086582,  4.87083348,  7.15100303,  7.83534987, 11.46176259,
       11.05672136, 12.3828957 , 10.24557047,  8.50268322, 10.08001894,
        5.79391825,  8.50278896,  9.60412735,  4.90505566,  8.53014915,
        7.18354885,  5.81481044,  8.22728399, 12.53047489, 13.40024224,
        6.78325151, 10.74409834,  5.0713364 ,  8.68832398, 13.43127131,
       11.62158554,  9.85330646,  9.0368458 ,  9.36256096,  8.41148722,
        6.1561646 ])}
{'P': tensor([[1.5969e-02, 2.6192e-01, 1.7393e-01,  ..., 5.6647e-07, 1.2002e-05,
         7.0112e-07],
        [2.2590e-04, 1.8022e-02, 1.8964e-01,  ..., 7.2356e-07, 1.6235e-05,
         4.9232e-07],
        [3.1857e-04, 1.4690e-03, 2.8685e-02,  ..., 1.2824e-06, 2.9723e-05,
         7.3905e-07],
        ...,
        [2.5890e-05, 1.3481e-05, 1.2056e-05,  ..., 6.0018e-02, 8.2878e-04,
         1.6524e-02],
        [2.1363e-04, 1.4964e-04, 1.4758e-04,  ..., 2.2774e-02, 1.2264e-01,
         3.4120e-03],
        [8.7441e-05, 4.5232e-05, 4.1305e-05,  ..., 2.1621e-07, 1.9251e-06,
         7.3030e-03]]), 'X': tensor([[ -5.9987,   6.1949,  -6.2114],
        [ -7.3446,   4.9113,  -5.9356],
        [ -7.6734,   4.4980,  -5.4248],
        [ -8.0608,   3.9362,  -5.9567],
        [ -7.8280,   4.2848,  -7.1955],
        [ -7.0825,   5.2073,  -5.0497],
        [  0.1287, -11.0065,   6.5757],
        [  8.1528,  10.7452,   9.1493],
        [  5.9843,  -9.8679,  -6.1449],
        [  2.9260,  -4.4162,   8.3808],
        [  6.9433,  -9.3362,  -0.3421],
        [ -1.0482, -10.8487,  -5.9066],
        [ 11.7779,   5.5127,  -2.0699],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -5.6423,   6.4537,  -1.2091],
        [  7.2340,  -9.1505,  -5.8321],
        [  2.5663,   7.0526,  -6.8769],
        [ -6.3306,   5.9272,   5.6935],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.0373,   7.1299,  -8.5895],
        [ -9.1624,  -3.4286,   2.0647],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.2804,   2.0019,   0.8607],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.8323,  -1.5228,   5.1132],
        [  6.3413,   0.4883,  -6.8416],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.3303,   1.6145,  -8.4083],
        [  1.8320,   7.3871,   5.1452],
        [  7.7750,  11.0665,   2.8364],
        [  2.0870,  -4.6877,   9.6823],
        [ 10.9931,   7.2107,   7.7933],
        [ -7.8359,   4.2735,   7.2943],
        [ -1.5408,   7.8978,   5.3470],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.6357,   5.4415,   7.9067],
        [ 10.9394,   7.3069,  -8.9365],
        [ -0.5385,   7.9037,   6.3830],
        [  3.2647, -10.8066,  -0.3768],
        [  1.3705,  13.9723,   7.9266],
        [  2.2437,  -4.6489,   6.1430],
        [  5.9526,   3.2525,  -4.3821],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.5626,  -2.0536,   8.4433],
        [ -9.2373,  -3.1225,  -8.9763],
        [  1.4489,   7.5256,   4.4529],
        [  9.2811,  -7.4336,   5.1771],
        [  4.6992,   5.3690,   3.0935],
        [  1.5662,  13.9375,   4.5142],
        [  2.4572,   7.1085,   7.7376],
        [ -0.0410,   7.8591,   9.3457],
        [  4.4915,   5.5990,   7.0405],
        [  3.3943,   6.5483,   1.0790],
        [ -8.3535,  -5.5118,   9.8399],
        [ -8.5607,  -5.0938,  -5.4531],
        [  1.5653,  13.9377,   4.1773],
        [  4.3459,  -3.5327,   0.3667],
        [ -9.0604,   1.7707,   1.1066],
        [  9.0129,  -7.7072,   9.1400],
        [ 10.1798,   8.5002,  -5.8893],
        [  6.1752,   2.5242,   3.5874],
        [  6.3549,   0.7096,   6.9198],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.8363,   2.4086,  -5.6296],
        [  3.9591, -10.6391,  -6.2557],
        [ 11.9808,   4.9340,  -3.3715],
        [  0.1497,   7.8334,  -5.4412],
        [  9.9756,  -6.6330,  -3.3429],
        [ -2.5683, -10.4433,  -2.9182],
        [  2.3889,  -4.6082,  -8.7291],
        [ -1.0407,   7.9166,  -6.6021],
        [  3.6249, -10.7255,   5.6605],
        [ 12.5832,   1.7543,  -7.6134],
        [ 11.5083,  -4.1177,   0.2389],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.1588,   4.3364,  -8.0867],
        [  5.7485,  -9.9811,  -3.2355],
        [ -7.4733,   4.7555,   6.2490],
        [ 10.2712,   8.3707,  -4.9743],
        [  7.7794,  11.0629,   8.9069],
        [  5.8902,  -1.3887,   4.0388],
        [  5.7784,  -1.6401,   9.5587],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.1379,  -4.6757,   6.4981],
        [  3.2809,  13.5061,  -4.0199],
        [  4.8653,   5.1694,  -4.0823],
        [  3.3167,   6.6022,   6.3554],
        [  0.5850,  -4.7982,   9.6639],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.6757, -11.0361,  -1.3047],
        [ -6.0195,   6.1789,  -2.9496],
        [ -3.3116, -10.1539,   7.5919],
        [  1.1994,   7.6035,   2.4295],
        [  0.2480, -11.0153,  -3.0183],
        [  6.3497,   1.3680,  -1.9076],
        [ 12.5717,   1.9266,   7.4903],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -9.4378,  -1.8606,  -2.5668],
        [  1.4749,  -4.7892,  -0.8089],
        [  2.7558,  -4.4844,   0.0636],
        [  3.3633,  13.4794,   0.5193],
        [  6.6319,  11.9066,   3.5165],
        [ 12.2311,  -2.0445,   1.3545],
        [  5.1667,  -2.6543,   1.2879],
        [  7.7750,  11.0665,   2.8364]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0521e-01, 1.4461e-01,  ..., 7.5724e-01, 7.2359e-01,
         1.1550e+00],
        [1.0521e-01, 0.0000e+00, 4.2779e-02,  ..., 8.3567e-01, 8.0141e-01,
         1.2315e+00],
        [1.4461e-01, 4.2779e-02, 0.0000e+00,  ..., 8.5367e-01, 8.1921e-01,
         1.2476e+00],
        ...,
        [7.5724e-01, 8.3567e-01, 8.5367e-01,  ..., 1.0727e-04, 3.4644e-02,
         3.9802e-01],
        [7.2359e-01, 8.0141e-01, 8.1921e-01,  ..., 3.4644e-02, 0.0000e+00,
         4.3147e-01],
        [1.1550e+00, 1.2315e+00, 1.2476e+00,  ..., 3.9802e-01, 4.3147e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.62329466,  8.8353674 ,  8.89458273,  8.97052115,  8.92395381,
        8.7907923 , 11.00726483, 13.48809595, 11.54071859,  5.29755487,
       11.63502679, 10.8992524 , 13.00413829,  9.97964013,  8.5724042 ,
       11.66453059,  7.50500064,  8.67227386, 13.0743399 , 13.1399181 ,
        9.78285424, 12.79126785,  6.59176018, 10.00794468,  6.02779201,
        6.36004393,  9.14973319,  6.53290663,  7.61088125, 13.5247215 ,
        5.13125513, 13.14690137,  8.92549752,  8.04664886, 13.14777433,
        7.14837514, 13.15525803,  7.92200342, 11.28895489, 14.0393949 ,
        5.16205087,  6.78325151, 10.30972193, 13.06498554,  5.92952538,
        9.75075273,  7.66377554, 11.89105259,  7.13501264, 14.02526505,
        7.52117132,  7.85919687,  7.17789666,  7.37576154, 10.00802627,
        9.96152788, 14.02532758,  5.60064319,  9.23178204, 11.85890664,
       13.26209545,  6.67121748,  6.39438044,  5.5513072 ,  9.15866607,
       11.35182589, 12.95702647,  7.83486943, 11.97957425, 10.75443106,
        5.19065439,  7.98469166, 11.32149259, 12.70489563, 12.22276278,
        6.14086582, 12.9089559 , 11.51811603,  8.8580671 , 13.25017211,
       13.52430176,  6.05165248,  6.00663314,  5.21280156,  5.14124612,
       13.89886469,  7.09887557,  7.38844703,  4.83371341, 10.46238054,
       11.05672136,  8.62631801, 10.68031351,  7.69752262, 11.018084  ,
        6.49538841, 12.71843462, 12.05395056,  9.61942445,  5.01114719,
        5.26342203, 13.89264822, 13.62896593, 12.40074901,  5.80860165,
       13.5247215 ])}
{'P': tensor([[2.0206e-02, 1.7333e-01, 1.6511e-01,  ..., 7.7663e-07, 9.6427e-07,
         1.5272e-05],
        [2.6193e-04, 2.7556e-02, 7.0191e-03,  ..., 7.7049e-07, 1.3375e-06,
         1.6830e-05],
        [1.9681e-04, 1.0228e-02, 1.7112e-02,  ..., 4.0236e-07, 8.0247e-07,
         1.2047e-05],
        ...,
        [2.1138e-04, 6.6211e-05, 5.3087e-05,  ..., 3.0438e-02, 1.5003e-05,
         1.0760e-04],
        [1.4648e-04, 4.3788e-05, 4.7130e-05,  ..., 1.5704e-01, 8.5770e-02,
         1.1412e-02],
        [2.0429e-03, 6.9812e-04, 9.2660e-04,  ..., 3.2141e-03, 2.2959e-03,
         1.2257e-01]]), 'X': tensor([[-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1891, 0.1449,  ..., 0.9169, 0.6775, 0.5924],
        [0.1891, 0.0000, 0.1506,  ..., 1.0501, 0.8433, 0.7812],
        [0.1449, 0.1506, 0.0000,  ..., 1.0617, 0.8157, 0.6983],
        ...,
        [0.9169, 1.0501, 1.0617,  ..., 0.0000, 0.3330, 0.6543],
        [0.6775, 0.8433, 0.8157,  ..., 0.3330, 0.0000, 0.3228],
        [0.5924, 0.7812, 0.6983,  ..., 0.6543, 0.3228, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.62631801,  8.94395628,  8.91551231,  8.85572292,  9.14973319,
        9.20340031, 13.89974693,  9.63742669,  6.24187044,  8.67227386,
        8.27853076,  6.29034958,  6.39438044, 13.95361098, 13.35832725,
        6.00732193,  5.05981914,  6.7593917 , 14.09176259, 13.89886469,
        9.69792513, 10.61047596, 13.96987806,  9.60359158,  5.16611782,
        5.80312949, 11.57249984, 13.6922425 , 11.51811603,  6.00663314,
        7.32950385, 12.0495521 ,  6.24614905, 11.00726483,  8.53501777,
        6.25586297,  7.24326992, 10.9036226 , 11.62158554,  6.46560181,
       12.33167382,  4.90676242,  9.71392179,  9.78286733,  9.61942445,
        9.17016857,  7.32233733, 10.46739073,  7.80625981,  5.80312949,
        9.73579738,  8.12580976,  9.24501829, 13.2778378 , 13.89974693,
        9.84314712,  7.48538905, 12.45418236,  9.99211663,  8.27853076,
       12.4980845 ,  6.65991877,  4.978704  , 11.11969815,  9.77777346,
        8.4486771 , 13.89886469,  6.24614905, 13.02818702,  7.19513806,
        9.14973319, 12.3344536 , 11.70016217, 13.00380624,  5.1224079 ,
       11.70016217, 10.4606265 ,  6.38968357,  8.37954853,  7.48311002,
        6.78325151,  7.24677664,  7.88677538, 11.75807252, 13.35013374,
        7.06169391,  8.94395628, 10.48840476, 12.7125079 ,  5.20659892,
        7.9337647 ,  4.74770899,  5.05981914,  6.67218687, 12.13568598,
        8.23525433,  7.58378655, 12.12448912,  7.73692436,  4.85603886,
        8.31625171, 13.14690137,  9.55720651, 13.1399181 , 12.62711252,
        6.54615887])}
{'P': tensor([[6.9535e-03, 1.5603e-01, 1.1002e-01,  ..., 3.2725e-05, 1.4706e-03,
         1.5721e-04],
        [2.1019e-04, 7.8793e-03, 1.1827e-01,  ..., 4.0262e-05, 2.2502e-03,
         1.7912e-04],
        [1.4421e-04, 2.2481e-04, 1.1454e-02,  ..., 7.5352e-05, 4.8442e-03,
         4.2151e-04],
        ...,
        [5.7333e-04, 5.2286e-04, 4.4829e-04,  ..., 2.8606e-01, 2.7722e-04,
         1.1517e-04],
        [1.6168e-05, 1.8500e-05, 2.5193e-05,  ..., 1.0467e-03, 3.3327e-02,
         5.4299e-05],
        [7.2530e-05, 8.0771e-05, 1.3726e-04,  ..., 1.4297e-03, 8.6201e-04,
         2.3902e-01]]), 'X': tensor([[ -6.2668,   5.9808,   2.5614],
        [ -6.6895,   5.6051,   2.7899],
        [ -7.5720,   4.6310,   2.1717],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.4808,   4.7462,   2.0266],
        [ -7.7239,   4.4298,   3.1216],
        [  6.0073,   3.0986,   6.7937],
        [  3.8111, -10.6787,  -4.6539],
        [  5.0233,   4.9647,  -3.1788],
        [  1.6232,  -4.7717,  -2.5655],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.6970,   6.9826,   9.7350],
        [  1.7815,  -4.7481,   4.4289],
        [  0.9967,  -4.8144,  -9.7370],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 11.3118,   6.5957,  -4.5667],
        [ -5.6169,  -8.7844,   8.9057],
        [  2.8474,   6.8979,   6.6457],
        [ -2.5161,   7.7699,  -9.3531],
        [ 11.0275,  -5.0666,   8.4154],
        [ -1.2454,   7.9127,  -0.5715],
        [ 11.5198,  -4.0923,   5.5088],
        [  3.5284, -10.7484,   5.3347],
        [ -4.8665,  -9.3204,  -5.2564],
        [  5.0506,   4.9277,  -9.5094],
        [  4.3523, -10.5232,   8.2138],
        [ -8.1905,   3.7253,   4.1175],
        [  2.2659,  13.7894,   6.5436],
        [  2.0419,  -4.6978,  -5.0983],
        [  0.8572,  14.0501,  -8.3364],
        [ -9.4541,  -0.3327,   8.8246],
        [ -0.5385,   7.9037,   6.3830],
        [  5.9883,   3.1536,   6.2320],
        [  6.1372,   2.6726,  -6.7885],
        [  6.2555,   2.1474,   4.1685],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -8.7544,   2.6096,   3.3146],
        [  5.7845,   3.6652,   0.9175],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.4920,  -4.1366,  -8.3955],
        [  6.1755,   2.5233,  -2.0042],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.3464,   0.5708,  -1.6397],
        [ -7.4602,   4.7718,  -4.6277],
        [  4.3377,   5.7570,   1.3712],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -6.3066,  -8.1936,   9.5822],
        [ 12.2311,  -2.0445,   1.3545],
        [  1.2273,  -4.8081,   9.8129],
        [ 10.0400,   8.6922,  -6.2847],
        [  4.3377,   5.7570,   1.3712],
        [  2.4639, -10.9431,  -7.3718],
        [  0.9478,  -4.8144,   0.6371],
        [  3.9895,   6.0816,  -6.5235],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.6095,  -1.9702,  -7.8233],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.0277,   4.9589,   7.0175],
        [ -8.8154,   2.4611,   2.5198],
        [  4.0587, -10.6112,   6.2273],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.0729,   7.2878,   7.0880],
        [  5.4167, -10.1293,   6.2156],
        [ -8.8400,   2.3991,   1.3457],
        [ -1.9714,   7.8563,  -1.8084],
        [ -7.4808,   4.7462,   2.0266],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -6.2668,   5.9808,   2.5614],
        [ -3.9700,   7.3453,  -4.1486],
        [ 12.5154,   2.5072,  -9.3274],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.6634,   5.8050,   9.1590],
        [  7.0515,  11.6194,   8.4289],
        [ 11.0864,   7.0387,  -0.1238],
        [ -3.5520,   7.4976,   9.1522],
        [  5.8836,   3.4315,  -4.0327],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3604,   0.9042,  -6.5901],
        [ 12.6035,   0.7494,   7.1536],
        [ -7.4733,   4.7555,   6.2490],
        [  8.1528,  10.7452,   9.1493],
        [ -0.0282,   7.8575,  -1.1746],
        [  1.9503,  -4.7172,  -6.2313],
        [ -8.6229,   2.9068,   2.4429],
        [ -0.5739,   7.9056,   0.6382],
        [  1.1994,   7.6035,   2.4295],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.0330,  -6.0765,   6.7945],
        [ -9.3616,  -2.4749,   0.1793],
        [ -6.4844,  -8.0230,   8.5209],
        [ -3.6381,   7.4683,   7.3544],
        [  0.5175,  14.0910,   4.6540],
        [  1.4489,   7.5256,   4.4529],
        [  2.3367,   7.1676,  -9.5263],
        [  2.4697,  -4.5836,   1.4593],
        [ -8.9942,  -3.9990,   6.0448],
        [ -9.3331,  -2.6465,  -8.2028]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0341, 0.1150,  ..., 0.4895, 0.4405, 0.8623],
        [0.0341, 0.0000, 0.0915,  ..., 0.5235, 0.4072, 0.8564],
        [0.1150, 0.0915, 0.0000,  ..., 0.5876, 0.3791, 0.7760],
        ...,
        [0.4895, 0.5235, 0.5876,  ..., 0.0000, 0.9207, 1.1055],
        [0.4405, 0.4072, 0.3791,  ..., 0.9207, 0.0000, 0.9675],
        [0.8623, 0.8564, 0.7760,  ..., 1.1055, 0.9675, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.66273171,  8.7273569 ,  8.87587457,  8.85939626,  8.859416  ,
        8.90405823,  6.7593917 , 11.33837396,  7.06272256,  5.04028538,
       12.84056604,  7.48538905,  5.0713364 ,  4.91652556,  9.3109884 ,
        9.5642993 , 13.09426142, 10.42667093,  7.46248947,  8.16714836,
       12.13568598,  8.01009083, 12.22503754, 11.31276109, 10.51437947,
        7.05627461, 11.38774149,  8.99791316, 13.97430191,  5.1224079 ,
       14.07623266,  9.45995955,  7.92200342,  6.76791483,  6.69389431,
        6.61386436, 11.21703654, 12.35625694,  9.13507321,  6.84796202,
        8.22728399,  5.41348121,  6.67108662,  8.36995953,  9.36378438,
        8.85572292,  7.20826219, 12.49291768, 10.33958639, 12.40074901,
        4.96231532, 13.27994304,  7.20826219, 11.21703654,  4.90676242,
        7.27340834, 12.38881635,  5.94541803, 10.52357479,  7.06169391,
        9.15253344, 11.36089355, 13.25336001,  7.57685908, 11.48663767,
        9.15976746,  8.09984231,  8.859416  , 12.3828957 ,  9.78286733,
        8.66273171,  8.34949383, 12.76408562, 11.81423161, 13.02818702,
       13.59170809, 13.13205357,  8.29640613,  6.81117802,  6.1561646 ,
        9.17905491,  5.80312949,  9.73579738,  9.9353921 ,  9.09963894,
        6.42439344, 12.62576026,  8.8580671 , 13.48809595,  7.8575648 ,
        5.10444097,  9.09963894,  7.92644746,  7.69752262, 10.5575601 ,
       10.99326577, 10.07238463,  9.68323686, 10.31584102,  8.30729352,
       14.10045797,  7.66377554,  7.53884104,  5.20659892,  9.84314712,
        9.70108409])}
{'P': tensor([[1.6697e-02, 2.5523e-01, 2.2548e-01,  ..., 2.1999e-06, 1.0847e-06,
         8.3411e-07],
        [4.0559e-04, 2.6541e-02, 8.9947e-02,  ..., 4.6139e-06, 1.1527e-06,
         9.6409e-07],
        [2.5353e-04, 2.4371e-03, 2.2839e-02,  ..., 5.3025e-06, 8.3316e-07,
         7.0746e-07],
        ...,
        [1.5425e-06, 9.6967e-07, 9.8522e-07,  ..., 1.4838e-02, 2.7858e-05,
         5.0525e-05],
        [9.6261e-04, 4.5756e-04, 3.0768e-04,  ..., 1.2386e-07, 2.5528e-02,
         4.3088e-05],
        [1.7287e-04, 8.2722e-05, 5.5420e-05,  ..., 6.0156e-08, 8.3770e-01,
         9.5475e-03]]), 'X': tensor([[ -6.3306,   5.9272,   5.6935],
        [ -7.4733,   4.7555,   6.2490],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.4573,   3.2448,   5.4370],
        [ -7.2515,   5.0195,   4.6434],
        [ -8.6603,   2.8250,   5.9181],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 12.3357,   3.6001,   0.7903],
        [  4.0078,  -3.8018,  -6.2168],
        [  1.7301,   7.4262,   5.0522],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -9.1749,  -3.3803,   6.4629],
        [  6.3124,   0.2057,  -0.4593],
        [  3.0876,   6.7527,  -5.3540],
        [ -8.4824,  -5.2575,  -0.0795],
        [  8.7715,  -7.9390,   5.2900],
        [  2.5375,   7.0676,  -4.3490],
        [ -4.0868,   7.2981,   8.1874],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.6757, -11.0361,  -1.3047],
        [ -7.6650,   4.5092,   0.2520],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.1751,  -2.2592,  -5.7751],
        [  3.9022,  13.2906,   5.5788],
        [ 12.1588,   4.3364,  -8.0867],
        [  3.9895,   6.0816,  -6.5235],
        [  0.3724,   7.7974,  -6.8722],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.0073,   3.0986,   6.7937],
        [  1.5246,  -4.7838,  -5.2095],
        [ -2.5056,   7.7720,   5.5878],
        [  3.1271, -10.8343,   2.6442],
        [  4.0457,  -3.7737,  -0.4704],
        [  2.3928,   7.1404,  -1.7035],
        [ 11.4235,   6.3594,   5.1663],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.6613,   7.0021,   4.5783],
        [ -7.7842,   4.3466,  -1.7548],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.1730,  -0.5375,  -8.8861],
        [ -7.9298,   4.1367,  -3.9773],
        [ -7.4808,   4.7462,   2.0266],
        [ 11.7796,  -3.4696,   9.0088],
        [ -7.5720,   4.6310,   2.1717],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 11.2571,   6.7072,   8.3693],
        [ 10.2712,   8.3707,  -4.9743],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1480,  -4.3165,  -7.8002],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 10.9916,   7.2133,  -0.9179],
        [ -5.2752,  -9.0409,  -8.2568],
        [  1.3767, -11.0341,   9.9415],
        [ -1.5451,   7.8975,   7.6104],
        [  4.1637,  13.1898,  -1.9189],
        [  6.3124,   0.2057,  -0.4593],
        [  6.2022,   2.4092,   3.0349],
        [ -5.1398,  -9.1367,   6.2074],
        [ -1.6086,   7.8928,   4.3385],
        [  6.3360,   0.4258,  -9.2509],
        [  1.7228,  -4.7575,   8.3396],
        [  5.0233,   4.9647,  -3.1788],
        [  1.5653,  13.9377,   4.1773],
        [ -3.5173, -10.0623,   4.8280],
        [  6.3305,   1.6119,   1.1538],
        [ -3.5520,   7.4976,   9.1522],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.3653,   3.4184,   9.3454],
        [  4.6992,   5.3690,   3.0935],
        [  5.4429,  -2.2529,   9.7306],
        [  5.7784,  -1.6401,   9.5587],
        [  5.9912,   3.1453,  -3.9989],
        [  2.1504,  -4.6726,   8.7908],
        [ -8.1204,  -5.9308,   4.0761],
        [ -2.6485, -10.4151,   0.8766],
        [  6.9693,  -9.3200,  -8.3822],
        [  1.1274,  14.0116,   8.1753],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -3.1671,   7.6158,   6.0765],
        [ -8.0750,  -6.0072,  -5.8267],
        [  0.6757, -11.0361,  -1.3047],
        [ -5.8389,  -8.6056,  -3.5098],
        [  9.5018,  -7.1945,  -1.1892],
        [  4.8653,   5.1694,  -4.0823],
        [  8.6271,  -8.0715,  -7.4813],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.3464,   0.5708,  -1.6397],
        [  5.0277,   4.9589,   7.0175],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.0611,  -6.0302,   4.6962],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -0.8455, -10.8853,  -6.7400],
        [  2.5318,  13.7231,   6.3692],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -7.6719,  -6.6260,   7.1468],
        [  5.4429,  -2.2529,   9.7306],
        [  7.1821,  -9.1845,   9.8548],
        [  4.5325,  -3.3643,   6.7699],
        [  6.1973,  -0.4372,   6.5720],
        [ 10.0567,   8.6697,   4.6329],
        [  8.0877,  -8.5297,  -1.1050],
        [  7.7794,  11.0629,   8.9069],
        [  9.3560,   9.5391,   9.4943]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.0619e-02, 1.2682e-01,  ..., 5.9920e-01, 1.0113e+00,
         9.8575e-01],
        [9.0619e-02, 0.0000e+00, 8.0977e-02,  ..., 6.6685e-01, 1.0816e+00,
         1.0566e+00],
        [1.2682e-01, 8.0977e-02, 2.1453e-04,  ..., 6.2528e-01, 1.1381e+00,
         1.1126e+00],
        ...,
        [5.9920e-01, 6.6685e-01, 6.2528e-01,  ..., 0.0000e+00, 1.1313e+00,
         1.1058e+00],
        [1.0113e+00, 1.0816e+00, 1.1381e+00,  ..., 1.1313e+00, 0.0000e+00,
         2.6590e-02],
        [9.8575e-01, 1.0566e+00, 1.1126e+00,  ..., 1.1058e+00, 2.6590e-02,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.67227386,  8.8580671 ,  8.92098752,  9.05843201,  8.81930525,
        9.10945678, 10.35485331, 10.54658597, 12.85032836,  5.52416368,
        7.62509582, 10.52357479,  9.77777346,  6.31576264,  7.42512621,
        9.97964013, 11.83075005,  7.5092834 ,  8.36445758,  8.47371868,
       11.05672136,  8.89301516,  5.81199422, 12.3828957 , 13.85158417,
       12.9089559 ,  7.27340834,  7.80625981,  8.29022197,  9.94285511,
        6.7593917 ,  5.0209005 ,  8.16584537, 11.27656488,  5.53255882,
        7.53063885, 13.0743399 ,  9.78286733,  7.49076899,  8.91551231,
        9.95552626,  6.19633139,  8.94395628,  8.859416  , 12.27993098,
        8.87587457,  9.75401449, 13.103709  , 13.25017211,  5.13182506,
        5.34250225, 12.49291768, 13.1471313 , 10.46739073, 11.11969815,
        8.04718433, 13.83139024,  6.31576264,  6.65368551, 10.48315436,
        8.0550286 ,  6.35029274,  5.05981914,  7.06272256, 14.02532758,
       10.65929968,  6.53250937,  8.29640613,  9.73579738,  9.0368458 ,
        7.13501264,  5.89074104,  6.00663314,  6.76661595,  5.14371005,
       10.05558893, 10.74655346, 11.63765273, 14.05687815,  5.0713364 ,
        9.60359158,  8.24808832, 10.06437598, 11.05672136, 10.39942497,
       11.9182956 ,  7.09887557, 11.81423161, 10.1979669 ,  9.36378438,
        7.06169391, 13.41713661, 10.06703358,  9.69629   , 10.91805683,
       13.95472331, 10.5575601 , 10.1371552 ,  5.89074104, 11.65923396,
        5.64464807,  6.21275016, 13.2778378 , 11.75438006, 13.52430176,
       13.3614598 ])}
{'P': tensor([[1.5794e-02, 2.9046e-01, 1.8408e-01,  ..., 5.0737e-03, 8.9199e-05,
         1.0961e-04],
        [2.9449e-04, 2.1429e-02, 8.2599e-02,  ..., 1.4451e-03, 6.2746e-05,
         1.9530e-04],
        [4.3485e-04, 3.4269e-03, 3.6464e-02,  ..., 1.2563e-03, 8.4506e-05,
         4.2283e-04],
        ...,
        [3.1406e-03, 1.6371e-02, 1.1796e-02,  ..., 2.3749e-02, 1.0583e-04,
         9.4239e-05],
        [4.6920e-02, 2.5255e-02, 2.5903e-02,  ..., 6.3637e-03, 1.7905e-02,
         7.3425e-05],
        [1.5904e-04, 1.3432e-04, 1.4008e-04,  ..., 9.2191e-05, 3.7287e-04,
         9.7362e-02]]), 'X': tensor([[ -6.3489,   5.9117,   6.5161],
        [ -7.4733,   4.7555,   6.2490],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.4573,   3.2448,   5.4370],
        [ -8.6603,   2.8250,   5.9181],
        [  2.0899,  -4.6870,  -7.4260],
        [ 12.2218,   4.0953,  -8.2072],
        [  3.4140,  -4.1803,  -8.8224],
        [  0.1497,   7.8334,  -5.4412],
        [ -7.7239,   4.4298,   3.1216],
        [  9.1820,   9.7334,  -2.7932],
        [  4.8432,   5.1968,   1.5006],
        [ -3.5029,   7.5138,   8.1853],
        [ -0.0282,   7.8575,  -1.1746],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.1751,  -2.2592,  -5.7751],
        [  7.0515,  11.6194,   8.4289],
        [  6.1741,  -0.5329,   3.1913],
        [ 12.4996,   2.6335,  -6.3166],
        [ 11.2209,  -4.7092,   2.6117],
        [ -1.5875,   7.8944,  -9.4286],
        [  4.9152,  12.8653,  -1.4459],
        [ -3.5520,   7.4976,   9.1522],
        [ 11.7418,   5.6070,  -1.4192],
        [  2.1824,  -4.6647,   1.1985],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -6.2200,  -8.2737,   5.5079],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.3357,   3.6001,   0.7903],
        [  5.9587,  12.3216,   9.1695],
        [ 10.6283,  -5.7253,   7.4878],
        [  3.3069,   6.6089,   8.7377],
        [ -4.0409,   7.3169,  -3.5779],
        [  3.5965,  -4.0754,  -3.5035],
        [ -1.6220,   7.8917,   0.0315],
        [ 11.3367,  -4.4806,   0.4643],
        [  5.8318,   3.5567,  -4.7692],
        [ 10.4758,  -5.9539,   1.3667],
        [  4.2339,  -3.6268,   6.3125],
        [  5.9655,   3.2172,  -7.5953],
        [ -6.0175,  -8.4541,  -9.8408],
        [  9.2811,  -7.4336,   5.1771],
        [  0.1252, -11.0062,   7.2564],
        [  2.1440, -10.9811,   8.5281],
        [ -6.1914,  -8.2997,   8.5380],
        [ 12.1588,   4.3364,  -8.0867],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.1203,  -2.7151,   6.2110],
        [ -5.6423,   6.4537,  -1.2091],
        [ 10.7566,   7.6212,  -2.3279],
        [  1.7232,  -4.7574,   3.1270],
        [ -5.1369,   6.7755,   8.3836],
        [  5.6095,  -1.9702,  -7.8233],
        [  2.6970,   6.9826,   9.7350],
        [ 12.0582,  -2.6640,   0.7132],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 12.5258,  -0.4200,  -9.3493],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.0608,   3.9362,  -5.9567],
        [ -4.6393,   7.0461,   1.1624],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.7992,  -3.9474,  -8.9019],
        [  1.6954,  -4.7616,   9.8198],
        [  3.8111, -10.6787,  -4.6539],
        [  1.6232,  -4.7717,  -2.5655],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.8363,   2.4086,  -5.6296],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -8.0608,   3.9362,  -5.9567],
        [  0.9074,  -4.8139,   8.1674],
        [  6.9653,  -9.3225,  -3.4727],
        [ -1.5261,   7.8988,   9.3768],
        [ 10.0619,   8.6627,  -9.2884],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.5859,  -5.0393,   0.5476],
        [  2.1379,  -4.6757,   6.4981],
        [ -4.1296,   7.2803,  -2.7194],
        [  2.3261,   7.1726,   6.2522],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.8407,  11.7666,   3.0045],
        [ -7.7842,   4.3466,  -1.7548],
        [ 12.5770,   1.8512,  -0.6578],
        [ -8.3535,  -5.5118,   9.8399],
        [ -2.7572,   7.7194,   5.9580],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.5955,   7.8824,  -3.7006],
        [ -8.1426,   3.8048,   9.3542],
        [ -7.9298,   4.1367,  -3.9773],
        [  4.5800,   5.5034,  -3.9432],
        [ -9.0604,   1.7707,   1.1066],
        [ -4.0409,   7.3169,  -3.5779],
        [  3.3474,  -4.2162,   5.2455],
        [  9.1820,   9.7334,  -2.7932],
        [  4.9152,  12.8653,  -1.4459],
        [  7.7750,  11.0665,   2.8364],
        [ 11.4703,  -4.2006,   7.4086],
        [ -9.1178,   1.5784,   2.1410],
        [ -5.1369,   6.7755,   8.3836],
        [ -2.5161,   7.7699,  -9.3531],
        [ -9.3616,  -2.4749,   0.1793],
        [ -6.6895,   5.6051,   2.7899],
        [ -2.2188,   7.8218,   8.6064],
        [  1.2273,  -4.8081,   9.8129]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0923, 0.1270,  ..., 0.2547, 0.2259, 0.4929],
        [0.0923, 0.0000, 0.0699,  ..., 0.2350, 0.3177, 0.5847],
        [0.1270, 0.0699, 0.0000,  ..., 0.3034, 0.3332, 0.6094],
        ...,
        [0.2547, 0.2350, 0.3034,  ..., 0.0000, 0.4171, 0.6152],
        [0.2259, 0.3177, 0.3332,  ..., 0.4171, 0.0000, 0.2791],
        [0.4929, 0.5847, 0.6094,  ..., 0.6152, 0.2791, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.67502348,  8.8580671 ,  8.92549752,  8.92098752,  9.05843201,
        9.10945678,  5.13182506, 12.889695  ,  5.39723241,  7.83486943,
        8.90405823, 13.38091121,  7.10378919,  8.29022197,  7.8575648 ,
       11.28895489, 12.3828957 , 13.59170809,  6.19707944, 12.77402351,
       12.16901615,  8.05242591, 13.77222752,  8.29640613, 13.01187484,
        5.14999799, 10.31613269, 10.35093485,  7.83486943, 12.85032836,
       13.68672855, 12.07225469,  7.39004226,  8.35856487,  5.43544626,
        8.05668609, 12.18998195,  6.83083863, 12.0495521 ,  5.5748624 ,
        6.77777606, 10.37698435, 11.89105259, 11.00694688, 11.18839169,
       10.35464916, 12.9089559 ,  5.90035324,  5.79562691,  8.5724042 ,
       13.18278606,  5.05989547,  8.50268322,  5.94541803,  7.48538905,
       12.3489324 ,  9.43737523, 12.53285633, 11.72749923,  8.97052115,
        8.43626114,  6.10774544,  5.47866221,  5.05444288, 11.33837396,
        5.04028538,  9.47369628,  9.15866607, 10.39942497,  8.97052115,
        4.89869205, 11.63724792,  8.04483495, 13.27717982,  9.19837792,
        9.79814997, 11.1913598 ,  9.95552626,  5.14124612,  8.36995953,
        7.5403822 ,  9.11473391, 13.6105595 ,  8.91551231, 12.7125079 ,
       10.00802627,  8.19703512,  6.80559669, 13.20597932,  8.98765644,
        8.94395628,  7.15990676,  9.23178204,  8.35856487,  5.38342077,
       13.38091121, 13.77222752, 13.5247215 , 12.21532282,  9.25336371,
        8.50268322,  8.16714836,  9.68323686,  8.7273569 ,  8.1303945 ,
        4.96231532])}
{'P': tensor([[1.2504e-02, 2.1262e-01, 1.5030e-01,  ..., 1.8514e-06, 5.6667e-07,
         1.5080e-06],
        [1.7095e-04, 1.4554e-02, 2.3925e-01,  ..., 3.1565e-06, 3.8174e-07,
         2.4707e-06],
        [3.4761e-04, 1.1369e-03, 4.1031e-02,  ..., 1.0985e-05, 8.2019e-07,
         8.5077e-06],
        ...,
        [2.9507e-06, 1.7304e-06, 1.4548e-06,  ..., 1.5411e-02, 1.6566e-05,
         7.8949e-01],
        [7.3639e-04, 3.8685e-04, 2.4504e-04,  ..., 1.9497e-07, 5.8691e-02,
         1.9650e-07],
        [1.6143e-05, 9.1347e-06, 7.6052e-06,  ..., 5.8409e-04, 9.6607e-05,
         7.4200e-02]]), 'X': tensor([[ -6.4368,   5.8356,   3.3014],
        [ -7.7239,   4.4298,   3.1216],
        [ -8.1666,   3.7653,   3.6075],
        [ -8.3900,   3.3727,   3.1650],
        [ -8.1905,   3.7253,   4.1175],
        [ -7.5720,   4.6310,   2.1717],
        [ -9.4378,  -1.8606,  -2.5668],
        [  9.3560,   9.5391,   9.4943],
        [  3.7237,  -3.9966,  -5.0376],
        [  1.5383, -11.0274,  -4.9666],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.0073,   3.0986,   6.7937],
        [  9.2811,  -7.4336,   5.1771],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.4739,  -1.2497,   0.5483],
        [ 10.4758,  -5.9539,   1.3667],
        [  2.6613,   7.0021,   4.5783],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.9452, -10.3187,  -7.8621],
        [  5.0348,  -2.8229,   3.5436],
        [ 12.4996,   2.6335,  -6.3166],
        [  4.9531,   5.0576,   5.6794],
        [ -9.4735,  -1.2620,  -3.6742],
        [  8.0381,  -8.5690,  -8.9523],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.3583,   1.1729,   8.7835],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -4.8751,  -9.3148,   4.0406],
        [ 10.6673,  -5.6649,  -4.1646],
        [  4.8051, -10.3703,   0.5268],
        [ -6.8618,  -7.6322,   3.7027],
        [  7.5930,  -8.9037,   7.2323],
        [  6.9653,  -9.3225,  -3.4727],
        [  0.1287, -11.0065,   6.5757],
        [  4.5814, -10.4485,   2.5496],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -8.8400,   2.3991,   1.3457],
        [  4.5954,  13.0099,  -4.4439],
        [  2.1771, -10.9776,  -3.2007],
        [ -9.2687,  -2.9794,   0.6436],
        [ 10.2470,   8.4054,  -0.1177],
        [  5.9548,  -1.2273,  -2.6520],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -9.1459,   1.4780,  -4.2403],
        [  0.8700,  -4.8132,  -7.8042],
        [  9.6614,   9.1784,  -1.2087],
        [ -2.5056,   7.7720,   5.5878],
        [ -1.5261,   7.8988,   9.3768],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.6022,  -1.9833,   8.3081],
        [  4.1637,  13.1898,  -1.9189],
        [  5.9026,   3.3837,  -0.6573],
        [  5.5101,  -2.1433,   9.5442],
        [ 11.2571,   6.7072,   8.3693],
        [  6.0345,   3.0175,  -1.4424],
        [ -8.4887,  -5.2448,  -0.6585],
        [  6.3303,   1.6145,  -8.4083],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.1905,   3.7253,   4.1175],
        [ -6.4620,  -8.0449,   6.2719],
        [  3.3474,  -4.2162,   5.2455],
        [  6.3236,   0.3011,  -4.6668],
        [ -4.2040,   7.2486,   2.9875],
        [  2.1120,  -4.6818,   8.6144],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.2905,   1.9366,   2.1169],
        [ -6.4368,   5.8356,   3.3014],
        [  4.1348,   5.9514,  -0.8856],
        [ -9.3604,   0.4855,  -3.0158],
        [ -8.5900,   2.9767,   7.4811],
        [  1.0000,   7.6591,  -4.5011],
        [  6.3550,   0.7129,   0.4917],
        [  6.9956,  11.6590,  -7.6876],
        [  5.1861,   4.7361,  -3.7312],
        [ -5.9872,   6.2037,  -9.4477],
        [ -3.0008, -10.2827,  -8.9480],
        [  6.1984,  -0.4327,   1.5438],
        [ -3.5173, -10.0623,   4.8280],
        [ 11.4495,   6.3026,   6.8257],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.3928,   7.1404,  -1.7035],
        [ -1.1885, -10.8211,   0.0865],
        [  0.6142,  14.0802,   0.5407],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -7.9667,   4.0814,  -0.6397],
        [  4.9452, -10.3187,  -7.8621],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.3670,  -4.7990,   3.2603],
        [  2.8994,   6.8675,  -9.5987],
        [  6.3087,   1.8053,  -1.9481],
        [  0.3715,  -4.7756,   0.2312],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.3087,   1.8053,  -1.9481],
        [ 10.5678,   7.9261,  -0.6100],
        [  0.1252, -11.0062,   7.2564],
        [  6.0560,  -0.9421,  -5.0334],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.2608,  -2.5256,  -9.1446],
        [  9.2811,  -7.4336,   5.1771],
        [  6.0310,  12.2795,  -6.4528],
        [  9.9058,  -6.7203,   5.6928]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0967, 0.1335,  ..., 0.4143, 1.1053, 0.4771],
        [0.0967, 0.0000, 0.0470,  ..., 0.5044, 1.1599, 0.5664],
        [0.1335, 0.0470, 0.0000,  ..., 0.5254, 1.2066, 0.5864],
        ...,
        [0.4143, 0.5044, 0.5254,  ..., 0.0000, 1.1202, 0.0632],
        [1.1053, 1.1599, 1.2066,  ..., 1.1202, 0.0000, 1.1434],
        [0.4771, 0.5664, 0.5864,  ..., 0.0632, 1.1434, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.68832398,  8.90405823,  8.99276564,  9.04255655,  8.99791316,
        8.87587457,  9.61942445, 13.3614598 ,  5.46244988, 11.1341798 ,
       12.52875793,  6.7593917 , 11.89105259,  5.24137909,  9.55593374,
       12.0495521 ,  7.49076899,  9.38232121, 11.4424816 ,  5.77219077,
       12.77402351,  7.07902036,  9.55720651, 11.74901778, 10.06286074,
        6.46560181,  7.822854  ,  9.88571713, 10.51340829, 12.07818871,
       11.42947632, 10.26327895, 11.70168755, 11.63724792, 11.00726483,
       11.4088043 ,  8.68979052, 10.23946247,  9.15976746, 13.79762415,
       11.1913598 ,  9.73579738, 13.25336001,  6.07992623, 12.18045978,
        9.26455795,  4.89121805, 13.32613445,  8.16584537,  8.04483495,
        5.18277858,  5.94293185, 13.83139024,  6.80370254,  5.91222346,
       13.103709  ,  6.74686071,  9.97822473,  6.53290663, 12.52875793,
        9.94285511,  8.99791316, 10.31886066,  5.38342077,  6.33075682,
        8.37954853,  5.13616224, 10.33712258,  6.58184474,  8.68832398,
        7.24677664,  9.37296102,  9.09119351,  7.72415048,  6.3949043 ,
       13.59673406,  7.02326515,  8.62162372, 10.71162681,  6.2134961 ,
       10.65929968, 13.06957425,  7.85919687, 10.06407839,  8.56889738,
        7.53063885, 10.88618114, 14.09357547,  9.4997987 ,  8.95133961,
       11.4424816 , 12.3828957 ,  4.98989906,  7.4544866 ,  6.5618968 ,
        4.79003361, 11.59512194,  6.5618968 , 13.20988996, 11.00694688,
        6.12886024,  9.11473391,  5.83560009, 11.89105259, 13.68063098,
       11.9702762 ])}
{'P': tensor([[1.0197e-02, 2.2540e-01, 1.6046e-01,  ..., 4.9898e-07, 6.6539e-06,
         2.5250e-07],
        [2.4767e-04, 1.1017e-02, 1.9564e-01,  ..., 4.9258e-07, 5.9280e-06,
         2.5689e-07],
        [2.3719e-04, 3.8067e-04, 2.1751e-02,  ..., 1.1356e-06, 9.8949e-06,
         5.4654e-07],
        ...,
        [2.2189e-05, 1.7349e-05, 1.5954e-05,  ..., 1.1726e-01, 2.1544e-03,
         2.0932e-03],
        [3.4276e-03, 2.4367e-03, 1.7719e-03,  ..., 1.1715e-04, 7.6352e-02,
         7.2599e-05],
        [3.3149e-05, 2.2903e-05, 1.5114e-05,  ..., 3.9100e-04, 1.6215e-02,
         5.2004e-02]]), 'X': tensor([[-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.8983e-02, 1.2534e-01,  ..., 1.1156e+00, 6.3507e-01,
         9.1675e-01],
        [4.8983e-02, 0.0000e+00, 7.9029e-02,  ..., 1.1461e+00, 6.7576e-01,
         9.5648e-01],
        [1.2534e-01, 7.9029e-02, 0.0000e+00,  ..., 1.1695e+00, 7.2207e-01,
         9.9939e-01],
        ...,
        [1.1156e+00, 1.1461e+00, 1.1695e+00,  ..., 0.0000e+00, 5.4404e-01,
         3.4139e-01],
        [6.3507e-01, 6.7576e-01, 7.2207e-01,  ..., 5.4404e-01, 0.0000e+00,
         2.8255e-01],
        [9.1675e-01, 9.5648e-01, 9.9939e-01,  ..., 3.4139e-01, 2.8255e-01,
         1.5170e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.68979052,  8.77913428,  8.92549752,  8.8737556 ,  9.09119351,
        8.8580671 , 13.96987806,  9.69629   ,  9.77588801,  6.02779201,
        7.15100303,  9.7751189 ,  7.5403822 , 12.33167382, 13.99877413,
       11.85743425,  6.67121748, 11.11969815,  5.05989547,  5.1224079 ,
        9.32094055,  5.5671389 ,  7.18354885,  6.27905292,  6.10360073,
       11.42947632, 10.46238054,  6.38105015,  7.58668144, 10.8207297 ,
       13.103709  , 11.05317597, 10.412299  ,  8.10886021,  8.16584537,
       13.12218013,  6.32572323,  9.45995955, 11.27160035, 12.27835996,
       11.57249984,  6.48486171,  8.8737556 ,  8.91551231,  6.53250937,
       14.02058621,  9.5642993 ,  7.27340834,  9.18690018, 10.63038565,
        9.91668182,  9.62958973,  7.8575648 , 11.75618513,  9.96152788,
        7.52117132,  6.84796202,  8.16584537, 10.69963954,  8.0550286 ,
       13.79676654, 12.72524615, 10.794755  ,  5.63824971, 11.64746882,
        6.383897  , 13.83139024, 12.30695098, 13.0493368 ,  9.25336371,
        9.24501829, 13.31764181, 12.49291768,  5.14371005, 11.42947632,
       10.80728788, 12.70489563,  9.94173708,  8.1303945 , 12.05300943,
       11.83570132, 10.91601655, 10.9036226 , 12.67576792,  5.29009836,
        7.44805901,  5.23878371,  5.29009836,  7.18354885, 11.77124373,
        9.85330646, 10.48840476,  8.83341608,  7.69752262,  6.46560181,
        5.43943791, 12.18998195,  7.09887557, 10.06286074,  9.0368458 ,
       12.12448912, 13.97779839,  5.05534609,  6.25586297,  7.20826219,
       12.67680439])}
{'P': tensor([[1.2925e-02, 2.3902e-01, 1.9400e-01,  ..., 1.8087e-03, 5.0523e-04,
         1.0391e-04],
        [4.1936e-04, 2.9812e-02, 7.6003e-03,  ..., 3.6630e-03, 9.0479e-04,
         3.4715e-04],
        [9.5532e-04, 8.3503e-02, 4.8673e-02,  ..., 1.0059e-02, 2.4548e-03,
         4.3595e-04],
        ...,
        [7.0389e-05, 1.2888e-04, 1.3531e-04,  ..., 4.1961e-02, 4.8482e-03,
         8.2300e-05],
        [3.7715e-05, 6.3445e-05, 7.4189e-05,  ..., 2.6160e-02, 2.8137e-02,
         1.9021e-05],
        [4.9382e-04, 4.8902e-04, 4.2592e-04,  ..., 9.3165e-05, 4.2021e-05,
         4.1103e-01]]), 'X': tensor([[ -6.6895,   5.6051,   2.7899],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.5720,   4.6310,   2.1717],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.4808,   4.7462,   2.0266],
        [ -8.3900,   3.3727,   3.1650],
        [  3.7298,  13.3537,  -9.4653],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.1940,  -2.1881,   7.1499],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 12.5717,   1.9266,   7.4903],
        [  0.5250, -11.0306,   5.5358],
        [ -1.5408,   7.8978,   5.3470],
        [ -7.3446,   4.9113,  -5.9356],
        [ -2.5683, -10.4433,  -2.9182],
        [  5.2199,   4.6861,  -1.9304],
        [ 12.4597,  -0.9136,  -4.8763],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.7009,   7.8851,  -8.2437],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.7779,   5.5127,  -2.0699],
        [  9.9756,  -6.6330,  -3.3429],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.9591, -10.6391,  -6.2557],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.0591,   6.0202,  -1.6339],
        [  6.3266,   0.3292,   9.3502],
        [  2.9410,   6.8428,   5.4458],
        [  2.1824,  -4.6647,   1.1985],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.1667,  -2.6543,   1.2879],
        [  2.2860,   7.1916,  -6.1011],
        [  2.1477,   7.2550,   8.9688],
        [  2.6078, -10.9229,   6.2869],
        [ -1.9714,   7.8563,  -1.8084],
        [  3.0876,   6.7527,  -5.3540],
        [  3.3094,  13.4969,  -3.6507],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.3860,  -2.3418,   1.0826],
        [  8.8091,  -7.9037,   0.3631],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.1426,   3.8048,   9.3542],
        [  8.8474,  10.0868,  -8.8644],
        [  5.9883,   3.1536,   6.2320],
        [  6.3589,   0.8239,  -8.4230],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.3373,  -4.6232,  -5.2246],
        [  2.7976,  -4.4682,  -4.2020],
        [ -6.1914,  -8.2997,   8.5380],
        [  3.6249, -10.7255,   5.6605],
        [  6.9314,  -9.3435,  -1.2729],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.2939,   6.6325,  -5.8311],
        [  1.8721,  -4.7322,   0.5524],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 12.5650,   2.0132,   4.6272],
        [  2.5829,  -4.5466,  -8.4015],
        [ -8.8363,   2.4086,  -5.6296],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.0330,  -6.0765,   6.7945],
        [ -6.7992,   5.4991,   9.9186],
        [  2.4149,  -4.6005,  -4.3339],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.9526,   3.2525,  -4.3821],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.6928,  12.4710,   3.6001],
        [  7.3724,  -9.0577,  -8.9843],
        [  3.9022,  13.2906,   5.5788],
        [  1.0093,   7.6567,   2.3526],
        [  6.3360,   0.4258,  -9.2509],
        [  2.0039,   7.3173,  -1.5885],
        [ 11.5935,   5.9740,  -4.3952],
        [ -8.3717,  -5.4769,  -9.2228],
        [  1.7000,  -4.7609,   2.7935],
        [  5.9003,  12.3550,   2.1917],
        [  6.8093,  -9.4178,   8.4244],
        [  1.7500,   7.4187,   4.4237],
        [ 12.5832,   1.7543,  -7.6134],
        [  1.8721,  -4.7322,   0.5524],
        [  5.9526,   3.2525,  -4.3821],
        [  4.4049,   5.6892,  -4.4394],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.6933,   7.7336,   2.1179],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -0.0254, -10.9932,   3.2200],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.3705,  13.9723,   7.9266],
        [ 11.5935,   5.9740,  -4.3952],
        [ -9.4773,  -1.0024,   5.2599],
        [  9.1620,  -7.5573,  -9.9882],
        [  0.3354,  -4.7708,  -4.9125],
        [  4.0642,  -3.7599,  -3.4565],
        [ -2.6735, -10.4062,  -1.5911],
        [  7.5309,  11.2617,   7.2656],
        [ -3.0644,   7.6439,  -1.4322],
        [  6.3497,   1.3680,  -1.9076],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.3087,   0.7792,  -7.5587],
        [  0.1206,  -4.7363,   8.7103]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0760, 0.0915,  ..., 0.6847, 0.7782, 0.5716],
        [0.0760, 0.0000, 0.0638,  ..., 0.6640, 0.7623, 0.6307],
        [0.0915, 0.0638, 0.0000,  ..., 0.6050, 0.7020, 0.6631],
        ...,
        [0.6847, 0.6640, 0.6050,  ..., 0.0000, 0.1070, 1.2148],
        [0.7782, 0.7623, 0.7020,  ..., 0.1070, 0.0000, 1.2931],
        [0.5716, 0.6307, 0.6631,  ..., 1.2148, 1.2931, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.7273569 ,  8.90405823,  8.87587457,  8.85939626,  8.859416  ,
        9.04255655, 13.86480175, 13.04051744, 12.38881635,  9.79039371,
       12.71843462, 11.04313641,  8.04664886,  8.8353674 , 10.75443106,
        7.01475376, 12.49317563,  6.61211224,  8.06643487,  8.92098752,
       13.00413829, 11.97957425, 10.41909211, 11.35182589,  9.54154096,
        7.26074925,  6.33517349,  7.44805901,  5.14999799,  8.85572292,
        5.80860165,  7.54620849,  7.56617037, 11.22993218,  8.09984231,
        7.42512621, 13.89671314,  9.58072286,  5.87305005, 11.83509372,
        5.24137909,  8.98765644, 13.41713661,  6.76791483,  6.41202986,
       12.00995898,  8.23525433,  5.18047638,  5.27178363, 10.35464916,
       11.32149259, 11.63383221,  9.71392179, 12.12641969,  9.562174  ,
        9.78285424,  5.1224079 , 13.09737053,  5.08910491, 10.52357479,
       12.72524615,  5.22903117,  9.15866607,  8.29022197, 10.07238463,
        8.74470218,  5.19577146, 10.31613269,  6.78325151,  6.07992623,
       13.70893492, 11.67875855, 13.85158417,  7.72291946,  6.35029274,
        7.58668144, 13.04217487, 10.00410625,  5.05534609, 13.69162795,
       11.62158554,  7.62232307, 12.70489563,  5.08910491,  6.78325151,
        7.19513806,  5.52416368,  7.76457095,  9.43737523, 10.99326577,
        8.29203356,  9.66977802, 14.0393949 , 13.04217487,  9.53015376,
       11.87665501,  4.78258092,  5.5366698 , 10.74409834, 13.54774355,
        8.23525433,  6.49538841, 12.10222204,  9.41605684,  9.34127013,
        4.73784222])}
{'P': tensor([[1.4469e-02, 2.9951e-01, 1.8799e-01,  ..., 1.1509e-06, 1.2590e-04,
         5.6026e-06],
        [3.8187e-04, 2.7069e-02, 1.1570e-01,  ..., 1.2750e-06, 3.1221e-04,
         1.6742e-05],
        [2.1985e-04, 1.6413e-03, 1.6747e-02,  ..., 5.8246e-07, 2.2111e-04,
         1.0288e-05],
        ...,
        [8.0740e-03, 5.5019e-03, 3.4258e-03,  ..., 3.1287e-01, 2.2294e-05,
         2.6085e-06],
        [1.0606e-04, 9.2900e-05, 9.1228e-05,  ..., 1.4608e-05, 5.9652e-02,
         6.9324e-04],
        [2.8970e-06, 2.4664e-06, 2.0651e-06,  ..., 7.9800e-06, 4.1745e-04,
         5.3229e-02]]), 'X': tensor([[-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0871, 0.1256,  ..., 1.1601, 0.4760, 0.6606],
        [0.0871, 0.0000, 0.0782,  ..., 1.2173, 0.5632, 0.6636],
        [0.1256, 0.0782, 0.0000,  ..., 1.2830, 0.5798, 0.7417],
        ...,
        [1.1601, 1.2173, 1.2830,  ..., 0.0000, 0.9355, 0.7940],
        [0.4760, 0.5632, 0.5798,  ..., 0.9355, 0.0000, 0.8285],
        [0.6606, 0.6636, 0.7417,  ..., 0.7940, 0.8285, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.72896381,  8.89301516,  8.95133961,  8.83725081,  9.09502385,
        9.11473391, 14.09357547, 11.89105259,  6.67218687, 12.32545637,
        9.7712321 ,  6.05165248,  5.87305005, 10.06703358,  6.31576264,
        8.04483495,  7.96575782,  8.62329466,  5.23878371,  6.38399789,
        6.80370254, 12.52875793,  7.61088125, 11.00508681, 13.26590071,
        8.23525433, 12.45418236,  9.94173708,  7.15100303,  6.29034958,
       12.87752052, 13.79762415, 13.48809595,  6.46356806, 11.35345345,
        5.10444097,  6.81084724,  6.28961988,  5.16077764, 12.07818871,
        9.17595899,  6.10774544,  6.31576264, 13.99048094, 11.48663767,
        8.56889738, 12.22276278,  6.23262289,  8.99791316,  8.06990483,
        9.88310614, 11.63724792,  9.55593374, 10.52587243,  9.78286733,
       10.24557047,  9.94173708, 12.9089559 , 11.27656488,  8.74470218,
        7.24326992, 13.1399181 ,  7.71075226,  5.91677188,  8.68979052,
       11.10066126,  5.79391825, 12.79126785,  5.04028538, 12.40074901,
       12.05395056, 12.52875793,  4.96231532,  6.53250937, 10.69963954,
        5.05963217,  7.92200342,  5.88782913, 12.53345445,  9.87720102,
        7.50500064,  8.7273569 , 12.13568598,  5.82345867,  9.50654967,
       11.99633819, 10.67132524,  5.90035324, 10.1979669 , 14.02532758,
       11.28895489,  5.30158497, 13.26590071,  6.05165248,  8.44940661,
       11.97238624, 11.87665501,  9.43737523, 11.89855267, 13.06957425,
       13.08414689, 10.57302546, 10.63038565, 13.93790235,  4.978704  ,
       11.35345345])}
{'P': tensor([[1.0159e-02, 2.9841e-01, 1.3943e-01,  ..., 1.2106e-05, 1.5685e-05,
         2.8210e-06],
        [3.0041e-04, 1.2743e-02, 2.1311e-01,  ..., 1.3534e-05, 2.1977e-05,
         3.8436e-06],
        [1.1893e-04, 1.4064e-04, 1.0700e-02,  ..., 7.5052e-06, 2.8207e-05,
         4.6825e-06],
        ...,
        [1.7349e-02, 1.5544e-02, 9.4284e-03,  ..., 9.3769e-02, 7.5965e-05,
         6.0849e-05],
        [5.8206e-06, 6.5497e-06, 1.1266e-05,  ..., 8.5975e-06, 1.1876e-01,
         7.5185e-03],
        [1.5074e-05, 1.4969e-05, 1.5110e-05,  ..., 2.9151e-04, 1.5416e-04,
         1.1691e-01]]), 'X': tensor([[ -6.7992,   5.4991,   9.9186],
        [ -7.0948,   5.1940,   9.8171],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.3653,   3.4184,   9.3454],
        [ -7.5401,   4.6718,   9.0033],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.4159,  -2.0739,   9.6841],
        [ -1.9646,   7.8571,   3.3817],
        [  5.4429,  -2.2529,   9.7306],
        [ -9.4748,  -0.7765,  -5.6695],
        [  2.8994,   6.8675,  -9.5987],
        [  5.6523,   3.9443,  -2.9449],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -3.5903,   7.4847,  -8.9112],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.7735,   2.5640,  -6.3313],
        [  4.5325,  -3.3643,   6.7699],
        [  5.3924, -10.1396,  -4.0863],
        [ 12.3453,  -1.5462,   7.0212],
        [ -4.8751,  -9.3148,   4.0406],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.3542,   0.6934,  -3.3840],
        [  7.1441,  11.5529,   2.3232],
        [ -8.3653,   3.4184,   9.3454],
        [ -1.5875,   7.8944,  -9.4286],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.8745,   2.3097,   3.5600],
        [  1.7301,   7.4262,   5.0522],
        [  5.4167, -10.1293,   6.2156],
        [ 10.9875,   7.2207,  -8.8549],
        [ 12.5789,   0.1722,   1.6470],
        [  4.2339,  -3.6268,   6.3125],
        [  6.0073,   3.0986,   6.7937],
        [ 11.6801,  -3.7196,  -0.9181],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.5469,  13.7192,  -9.6935],
        [ -7.2515,   5.0195,   4.6434],
        [  2.5620,   7.0549,   8.4825],
        [  6.3569,   0.7591,   6.5011],
        [  1.6954,  -4.7616,   9.8198],
        [  1.1644, -11.0394,   7.8511],
        [ -7.0825,   5.2073,  -5.0497],
        [ -3.9508,  -9.8515,   1.7923],
        [  5.5158,  -2.1337,   3.5246],
        [  2.4572,   7.1085,   7.7376],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.3444,   0.5821,   6.9101],
        [ -8.9282,   2.1648,   8.4894],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -5.7347,  -8.6908,   9.2878],
        [ -4.6393,   7.0461,   1.1624],
        [  5.5626,  -2.0536,   8.4433],
        [  8.3718,  -8.2953,   8.4666],
        [  6.8093,  -9.4178,   8.4244],
        [ -3.4100,   7.5436,  -7.6980],
        [  4.1297,  -3.7098,  -3.9838],
        [ -1.8823, -10.6558,  -3.8952],
        [  1.3103,  -4.8032,  -0.9300],
        [  8.3718,  -8.2953,   8.4666],
        [  7.0358,  -9.2784,  -9.8479],
        [ -7.2986,  -7.1225,  -3.0814],
        [  2.0449,  13.8402,  -7.1642],
        [ -6.4620,  -8.0449,   6.2719],
        [  4.5503,   5.5359,   7.4735],
        [  5.2608,  12.6977,   4.9166],
        [  6.0310,  12.2795,  -6.4528],
        [ -3.6381,   7.4683,   7.3544],
        [  5.0348,  -2.8229,   3.5436],
        [  8.0877,  -8.5297,  -1.1050],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -7.2515,   5.0195,   4.6434],
        [ -6.8466,  -7.6488,   1.9648],
        [  2.6078, -10.9229,   6.2869],
        [  0.6952,  -4.8060,   8.8976],
        [  1.2728, -11.0372,   9.8121],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.7232,  -4.7574,   3.1270],
        [  2.2373,  -4.6506,  -1.7334],
        [ 11.9122,   5.1397,  -1.5498],
        [ -5.5525,   6.5146,  -4.4984],
        [ -0.7599, -10.8995,  -6.0366],
        [  4.0587, -10.6112,   6.2273],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -8.8540,  -4.3971,  -5.2144],
        [  3.3094,  13.4969,  -3.6507],
        [ 11.0864,   7.0387,  -0.1238],
        [  0.1041, -11.0045,   2.4311],
        [ 11.0275,  -5.0666,   8.4154],
        [  1.7659,   7.4127,   4.7871],
        [ -9.4753,  -0.8012,  -5.2050],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.2373,  -3.1225,  -8.9763],
        [  1.7815,  -4.7481,   4.4289],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.3493,  -4.7727,  -6.5682],
        [  3.6154,  -4.0640,  -7.9698],
        [  8.1217,  -8.5024,  -0.6149],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3497,   1.3680,  -1.9076],
        [  1.9241,   7.3503,   7.9004],
        [  1.7301,   7.4262,   5.0522],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.3373,  -4.6232,  -5.2246]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0328, 0.1027,  ..., 0.4455, 1.1131, 1.0356],
        [0.0328, 0.0000, 0.0700,  ..., 0.4756, 1.1020, 1.0528],
        [0.1027, 0.0700, 0.0000,  ..., 0.5417, 1.0830, 1.0934],
        ...,
        [0.4455, 0.4756, 0.5417,  ..., 0.0000, 1.1767, 0.7200],
        [1.1131, 1.1020, 1.0830,  ..., 1.1767, 0.0000, 0.9374],
        [1.0356, 1.0528, 1.0934,  ..., 0.7200, 0.9374, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.74470218,  8.79284466,  8.98765644,  8.95983586,  9.0368458 ,
        8.87007495, 10.53368644,  7.53884104,  9.64157461,  8.09899874,
        5.89074104,  9.50654967,  7.4544866 ,  6.89245352, 12.00995898,
        8.30125201,  5.53255882,  9.14044984,  5.64464807, 11.48434767,
       12.44177641, 10.51340829,  9.62958973,  6.39187285, 13.58333857,
        9.0368458 ,  8.05242591,  8.22728399,  9.17016857,  7.62509582,
       11.48663767, 13.14777433, 12.58006038,  5.5748624 ,  6.7593917 ,
       12.25806559, 11.86239961, 13.95361098,  8.81930525,  7.50564223,
        6.4020404 ,  5.05444288, 11.10066126,  8.7907923 , 10.61417181,
        5.91408504,  7.52117132,  8.90405823,  9.36256096,  9.18690018,
       12.07818871, 10.412299  ,  8.43626114,  5.92952538, 11.785566  ,
       11.62158554,  8.27853076,  5.5513072 , 10.8207297 ,  4.978704  ,
       11.785566  , 11.64436684, 10.1979669 , 13.99048094, 10.31886066,
        7.16598273, 13.744387  , 13.68063098,  8.30729352,  5.77219077,
       11.75438006, 12.32545637,  8.81930525, 10.26545792, 11.22993218,
        4.85603886, 11.11038551, 13.12299145,  5.05989547,  5.16077764,
       12.97370371,  8.55982105, 10.92597196, 11.36089355, 13.79762415,
       12.27835996,  9.88571713, 13.89671314, 13.13205357, 11.0050379 ,
       12.13568598,  7.62011734,  9.50913849,  4.8763135 ,  9.75075273,
        5.0713364 , 10.24557047,  4.78544875,  5.43943791, 11.75807252,
        5.5366698 ,  6.49538841,  7.59795445,  7.62509582,  9.91668182,
        5.18047638])}
{'P': tensor([[1.3718e-02, 2.4784e-01, 2.4360e-01,  ..., 1.4087e-07, 8.1392e-07,
         1.4044e-05],
        [5.0103e-04, 1.9983e-02, 7.0827e-02,  ..., 1.6415e-07, 9.0658e-07,
         1.4879e-05],
        [3.2439e-04, 2.2214e-03, 1.8536e-02,  ..., 1.7040e-07, 7.8770e-07,
         1.3725e-05],
        ...,
        [1.2765e-05, 8.2494e-06, 9.1673e-06,  ..., 1.6903e-02, 3.1348e-03,
         2.0245e-03],
        [4.0584e-04, 2.5437e-04, 2.3663e-04,  ..., 7.4544e-06, 1.1377e-02,
         3.7007e-03],
        [6.5049e-03, 4.4791e-03, 5.5764e-03,  ..., 4.5886e-05, 4.1575e-04,
         7.1875e-02]]), 'X': tensor([[-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0608, 0.0790,  ..., 1.3367, 0.9088, 0.6380],
        [0.0608, 0.0000, 0.0357,  ..., 1.3831, 0.9695, 0.6828],
        [0.0790, 0.0357, 0.0000,  ..., 1.3700, 0.9820, 0.6693],
        ...,
        [1.3367, 1.3831, 1.3700,  ..., 0.0000, 0.9229, 0.7007],
        [0.9088, 0.9695, 0.9820,  ..., 0.9229, 0.0000, 0.6627],
        [0.6380, 0.6828, 0.6693,  ..., 0.7007, 0.6627, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.77913428,  8.8737556 ,  8.92549752,  9.09119351,  9.18358562,
        9.19837792,  6.59176018,  7.16058587,  6.39520397, 13.98691426,
        7.8788965 , 13.66896739, 13.44593519, 11.62158554,  4.89121805,
        5.80860165, 10.412299  ,  8.31625171,  5.80312949, 12.4217523 ,
       13.56367857,  7.48311002,  6.83083863, 11.35182589, 11.9182956 ,
        8.09842704,  6.29821673, 13.6829843 , 14.08549802,  8.53014915,
       10.75443106,  5.76165934, 11.56146873, 11.63383221,  9.93553424,
       13.89264822,  5.05534609,  6.652742  , 10.80728788, 13.13888841,
       14.02526505,  7.14837514,  5.83560009,  8.37954853,  9.58072286,
        9.5014146 ,  6.74686071, 13.54774355,  5.52416368, 13.31764181,
        9.38232121,  9.97220799,  5.60064319,  5.14371005, 12.58006038,
       12.85032836,  9.562174  , 14.02526505,  7.20482224, 10.07238463,
        6.383897  ,  9.66977802,  9.50654967, 11.21703654, 14.0393949 ,
        8.89301516,  9.93553424,  6.18668463, 10.06437598, 11.67875855,
       13.09737053, 12.17901033,  7.24677664, 12.4217523 , 10.05558893,
        5.16077764,  9.88310614,  5.56809335,  5.5205189 ,  5.98285319,
        8.36445758, 13.89886469,  8.85572292,  6.18668463,  9.4997987 ,
        8.47371868,  7.5158489 , 10.06023665,  5.46244988,  8.34949383,
       10.61417181, 10.31584102,  9.34127013,  9.58072286,  6.10360073,
        8.98765644,  5.94541803,  5.21280156,  9.77777346,  8.41148722,
       11.27160035,  6.54615887,  4.76969364, 13.26209545, 13.66991027,
        7.73151309])}
{'P': tensor([[1.3915e-02, 2.7969e-01, 2.0169e-01,  ..., 6.6120e-07, 4.1622e-05,
         1.8383e-05],
        [5.5623e-04, 2.5855e-02, 4.0797e-03,  ..., 9.0273e-07, 8.2296e-05,
         3.8676e-05],
        [9.6165e-04, 7.7541e-02, 2.5426e-02,  ..., 1.0630e-06, 9.2388e-05,
         3.3662e-05],
        ...,
        [2.3373e-03, 1.4418e-03, 2.0411e-03,  ..., 1.1825e-01, 2.1999e-05,
         7.4679e-06],
        [1.2615e-04, 1.0892e-04, 1.2997e-04,  ..., 3.1689e-04, 1.6369e-01,
         1.5892e-03],
        [1.1101e-04, 9.3933e-05, 9.2431e-05,  ..., 1.1349e-04, 1.2103e-04,
         4.9526e-02]]), 'X': tensor([[ -7.0825,   5.2073,  -5.0497],
        [ -7.6734,   4.4980,  -5.4248],
        [ -7.4602,   4.7718,  -4.6277],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.8363,   2.4086,  -5.6296],
        [ -7.9298,   4.1367,  -3.9773],
        [  1.8719,   7.3713,  -3.7465],
        [  3.2701,   6.6338,   9.9209],
        [  5.5240,  -2.1198,   6.2055],
        [  9.6816,   9.1536,  -5.8543],
        [ 11.4703,  -4.2006,   7.4086],
        [ -4.3963,   7.1629,   4.5263],
        [ 10.0995,  -6.4739,   2.0974],
        [  3.0719, -10.8449,  -4.1673],
        [  5.5158,  -2.1337,   3.5246],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.0604,   1.7707,   1.1066],
        [  9.4925,   9.3811,   8.9842],
        [  6.3211,   1.7017,  -0.2164],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.6681,  -4.8556,   9.2518],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.1929,   2.4501,  -6.9352],
        [  7.1697,  11.5343,  -0.7276],
        [ -1.8788, -10.6567,   3.7068],
        [  1.9503,  -4.7172,  -6.2313],
        [ -6.7992,   5.4991,   9.9186],
        [  6.1685,  12.1979,  -6.2473],
        [  7.5064,  -8.9651,  -0.0994],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.7590,  13.6621,   9.4496],
        [ 11.9122,   5.1397,  -1.5498],
        [ -6.4620,  -8.0449,   6.2719],
        [ -9.3464,   0.5708,  -1.6397],
        [ -1.0482, -10.8487,  -5.9066],
        [  9.3560,   9.5391,   9.4943],
        [ -6.9838,  -7.4965,   0.9796],
        [ -7.4807,   4.7463,   2.0616],
        [  9.1409,  -7.5789,   1.5814],
        [  6.2022,   2.4092,   3.0349],
        [  1.8320,   7.3871,   5.1452],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.3606,   0.9204,   3.2525],
        [ -0.1963,   7.8765,  -1.1371],
        [ -6.7745,  -7.7264,   8.4684],
        [ -8.9636,   2.0648,   8.1946],
        [ 11.0373,   7.1299,  -8.5895],
        [  0.1047, -11.0046,   0.5188],
        [  8.2154,  -8.4262,   8.0171],
        [  4.5954,  13.0099,  -4.4439],
        [  1.4150,  13.9647,  -4.0915],
        [ 10.4994,  -5.9193,  -7.5153],
        [  2.1440, -10.9811,   8.5281],
        [ -2.0229, -10.6163,   8.6930],
        [  2.0039,   7.3173,  -1.5885],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.9883,   3.1536,   6.2320],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -3.1204, -10.2345,   9.2288],
        [ 11.0438,   7.1180,  -5.6991],
        [ -2.5056,   7.7720,   5.5878],
        [  8.5716,  10.3596,   6.8818],
        [  6.2535,  -0.1709,  -9.4102],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.7000,  -4.7609,   2.7935],
        [  6.3534,   1.2990,  -7.2942],
        [ -8.8835,  -4.3178,   9.3080],
        [ -1.8823, -10.6558,  -3.8952],
        [  7.7750,  11.0665,   2.8364],
        [  4.3523, -10.5232,   8.2138],
        [  6.3303,   1.6145,  -8.4083],
        [  2.0039,   7.3173,  -1.5885],
        [ -9.1624,  -3.4286,   2.0647],
        [ 11.0275,  -5.0666,   8.4154],
        [ -9.1624,  -3.4286,   2.0647],
        [  1.7301,   7.4262,   5.0522],
        [ -8.7078,   2.7182,   1.8146],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.2975,   0.0947,   5.1791],
        [ 12.5267,  -0.4125,  -8.7331],
        [  9.9058,  -6.7203,   5.6928],
        [ -5.0550,   6.8231,  -9.2502],
        [ -0.8885,   7.9160,   0.0571],
        [ -1.9714,   7.8563,  -1.8084],
        [ -7.6293,  -6.6859,  -5.5161],
        [  5.6314,  -1.9302,  -9.6543],
        [  1.0000,   7.6591,  -4.5011],
        [  1.1274,  14.0116,   8.1753],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.5900,   2.9767,   7.4811],
        [ -2.5161,   7.7699,  -9.3531],
        [ -2.8162,   7.7059,  -9.6885],
        [  4.9152,  12.8653,  -1.4459],
        [  6.8704,  -9.3809,   7.5769],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.8201,   3.5841,  -8.0609],
        [  6.2038,   2.4020,  -3.1692],
        [  5.9003,  12.3550,   2.1917],
        [  1.7000,  -4.7609,   2.7935],
        [  5.2608,  -2.5256,  -9.1446]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0736, 0.0523,  ..., 1.0985, 0.7600, 0.6476],
        [0.0736, 0.0000, 0.0518,  ..., 1.1689, 0.8264, 0.7156],
        [0.0523, 0.0518, 0.0000,  ..., 1.1240, 0.7783, 0.6966],
        ...,
        [1.0985, 1.1689, 1.1240,  ..., 0.0000, 0.3705, 0.7416],
        [0.7600, 0.8264, 0.7783,  ..., 0.3705, 0.0000, 0.6072],
        [0.6476, 0.7156, 0.6966,  ..., 0.7416, 0.6072, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.7907923 ,  8.89458273,  8.85572292,  8.97052115,  9.15866607,
        8.94395628,  7.60529892,  7.39599986,  5.91677188, 13.32374056,
       12.21532282,  8.40445872, 11.99633819, 11.27160035,  5.91408504,
        8.7273569 ,  9.23178204, 13.34587027,  6.54615887, 13.6660671 ,
        9.9353921 , 12.24467041,  6.65991877, 13.58101794, 10.82106505,
        5.10444097,  8.74470218, 13.66896739, 11.69264243,  5.84450127,
       13.93790235, 12.97370371, 10.31886066,  9.36378438, 10.8992524 ,
       13.3614598 , 10.24557047,  8.85939626, 11.87412439,  6.65368551,
        7.61088125,  9.97220799,  9.86089579,  9.45995955,  6.42689705,
        7.8788965 , 10.27572673,  9.19837792, 13.1399181 , 11.00508681,
       11.76830981, 13.79762415, 14.03618649, 12.05300943, 11.18839169,
       10.80728788,  7.58668144,  6.23262289,  5.84533345,  6.76791483,
       10.5575601 , 10.69963954, 13.13888841,  8.16584537, 13.44593519,
        6.25586297,  8.5724042 ,  9.77588801,  5.05534609,  6.48486171,
        9.87720102, 10.8207297 , 13.5247215 , 11.38774149,  6.53290663,
        7.58668144,  9.78285424, 12.13568598,  9.78285424,  7.62509582,
        9.12220271, 11.66453059,  6.29821673, 12.53345445, 11.9702762 ,
        8.49161154,  7.96575782,  8.09984231, 10.14436912,  5.95297459,
        7.72415048, 14.05687815,  9.37989631,  9.09119351,  8.16714836,
        8.20436764, 13.77222752, 11.62770332, 13.01728313, 12.52875793,
       10.0674887 ,  6.83515201,  6.65259775, 13.69162795,  5.05534609,
        5.83560009])}
{'P': tensor([[1.2679e-02, 2.1204e-01, 1.8412e-01,  ..., 5.7832e-07, 7.5136e-06,
         1.5692e-06],
        [1.8221e-04, 1.3862e-02, 4.9660e-01,  ..., 6.4308e-07, 1.0458e-05,
         9.1124e-07],
        [2.6588e-04, 4.8500e-04, 2.5288e-02,  ..., 1.1522e-06, 2.0013e-05,
         1.4308e-06],
        ...,
        [1.1052e-04, 9.0102e-05, 8.2113e-05,  ..., 2.9467e-01, 1.3490e-03,
         2.4048e-03],
        [1.5567e-04, 1.2117e-04, 1.1177e-04,  ..., 1.0198e-02, 1.4969e-01,
         7.9289e-04],
        [3.0533e-02, 1.7064e-02, 1.4069e-02,  ..., 4.1296e-05, 3.2561e-05,
         2.9791e-01]]), 'X': tensor([[ -7.0948,   5.1940,   9.8171],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.0087,   4.0173,   9.1091],
        [ -7.5401,   4.6718,   9.0033],
        [ -9.0574,   1.7801,   8.8986],
        [ -3.5173,   7.5091,  -8.9585],
        [  5.7784,  -1.6401,   9.5587],
        [ -7.5401,   4.6718,   9.0033],
        [  3.4401,  13.4540,   4.6576],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.4766,  -1.1116,  -2.1706],
        [  0.2571,  -4.7594,   1.5618],
        [  3.6469,   6.3624,  -1.8414],
        [  6.0467,  -9.8369,  -6.5816],
        [ 12.5832,   1.7543,  -7.6134],
        [ -4.0868,   7.2981,   8.1874],
        [  9.5203,   9.3483,   5.8781],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.2707,   0.9661,   4.4863],
        [ -8.9178,   2.1936,   7.5388],
        [  9.4925,   9.3811,   8.9842],
        [  4.5954,  13.0099,  -4.4439],
        [  5.4867,  12.5816,   2.9549],
        [ -9.1706,   1.3863,   8.4032],
        [ -1.9714,   7.8563,  -1.8084],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.8835,  -4.3178,   9.3080],
        [ 11.2767,  -4.6006,   0.7868],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.0042,  -2.8350,   2.6864],
        [ -4.3963,   7.1629,   4.5263],
        [  9.9756,  -6.6330,  -3.3429],
        [  7.5930,  -8.9037,   7.2323],
        [  5.9026,   3.3837,  -0.6573],
        [  0.1382, -11.0073,   1.0026],
        [  7.1697,  11.5343,  -0.7276],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.9314,  -9.3435,  -1.2729],
        [  5.0617,  12.7957,   8.3544],
        [ -8.0829,  -5.9940,   5.0564],
        [ -5.6169,  -8.7844,   8.9057],
        [ -7.8126,   4.3066,   5.1791],
        [ -5.1046,  -9.1610,   2.3418],
        [  7.1441,  11.5529,   2.3232],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.4735,  -1.2620,  -3.6742],
        [  0.6952,  -4.8060,   8.8976],
        [ -8.7735,   2.5640,  -6.3313],
        [  6.1752,   2.5242,   3.5874],
        [ 10.0995,  -6.4739,   2.0974],
        [  7.5784,  -8.9141,   5.8410],
        [  4.9452, -10.3187,  -7.8621],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.0591,   6.0202,  -1.6339],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -1.8367,   7.8718,  -8.6523],
        [ 12.2692,  -1.8882,   3.9253],
        [ -9.4430,  -0.1897,   4.8383],
        [  7.4132,  11.3526,   0.3322],
        [  4.8051, -10.3703,   0.5268],
        [  6.2975,   0.0947,   5.1791],
        [  7.8982,  -8.6778,   0.0239],
        [  0.6757, -11.0361,  -1.3047],
        [  1.7500,   7.4187,   4.4237],
        [  8.8091,  -7.9037,   0.3631],
        [  0.4119,  14.1019,  -4.1870],
        [  6.3505,   0.6235,  -8.5849],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 11.6019,   5.9540,  -5.1715],
        [  1.7903,  -4.7467,   0.1168],
        [ 10.9916,   7.2133,  -0.9179],
        [ -2.8162,   7.7059,  -9.6885],
        [ -5.6792,  -8.7352,   4.0318],
        [  5.6248,   3.9984,   8.4933],
        [  8.3791,  10.5409,   5.5590],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.1504,   4.3674,  -7.2998],
        [ -6.1899,  -8.3011,  -7.9269],
        [  9.9058,  -6.7203,   5.6928],
        [ 11.5568,   6.0602,   9.3244],
        [  3.1480,  -4.3165,  -7.8002],
        [  4.9957,  -2.8705,   6.5978],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.1531,  -5.8748,   2.6934],
        [  5.3860,  -2.3418,   1.0826],
        [ -7.4602,   4.7718,  -4.6277],
        [  0.9967,  -4.8144,  -9.7370],
        [ -0.8455, -10.8853,  -6.7400],
        [  9.5018,  -7.1945,  -1.1892],
        [  6.8407,  11.7666,   3.0045],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.1730,  -0.5375,  -8.8861],
        [ -4.4520,   7.1370,  -8.7867],
        [ -7.0948,   5.1940,   9.8171],
        [ -9.2974,   0.8371,   2.4852],
        [ -0.8455, -10.8853,  -6.7400],
        [  3.4604,  -4.1545,   1.8091],
        [  6.1737,   2.5305,   3.9049],
        [  6.0047,   3.1062,  -6.7914],
        [  5.3860,  -2.3418,   1.0826],
        [  1.5662,  13.9375,   4.5142]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.9993e-02, 9.7075e-02,  ..., 1.1537e+00, 8.1252e-01,
         9.1353e-01],
        [6.9993e-02, 0.0000e+00, 2.7082e-02,  ..., 1.1949e+00, 8.6841e-01,
         9.8353e-01],
        [9.7075e-02, 2.7082e-02, 0.0000e+00,  ..., 1.2113e+00, 8.9042e-01,
         1.0106e+00],
        ...,
        [1.1537e+00, 1.1949e+00, 1.2113e+00,  ..., 0.0000e+00, 4.4374e-01,
         9.8007e-01],
        [8.1252e-01, 8.6841e-01, 8.9042e-01,  ..., 4.4374e-01, 1.0727e-04,
         5.7393e-01],
        [9.1353e-01, 9.8353e-01, 1.0106e+00,  ..., 9.8007e-01, 5.7393e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.79284466,  8.98765644,  9.0368458 ,  8.95983586,  8.87007495,
        9.23071698,  8.29203356,  6.00663314,  8.87007495, 13.88683436,
        6.69389431,  9.43112957,  9.54154096,  4.7663526 ,  7.33350642,
       11.54672755, 12.70489563,  8.36445758, 13.3426541 ,  8.5724042 ,
        9.32094055,  9.18358562, 13.34587027, 13.79762415, 13.72594022,
        9.27474386,  8.09984231, 10.61472636,  9.87720102, 12.17901033,
        8.51928531, 12.3344536 ,  8.40445872, 11.97957425, 11.70168755,
        6.80370254, 11.00812996, 13.58101794, 10.0674887 ,  9.41605684,
       11.63383221, 13.76047999, 10.06286074, 10.42667093,  8.92098752,
       10.48721382, 13.58333857, 11.62158554,  9.55720651,  4.85603886,
        9.14044984,  6.67121748, 11.99633819, 11.70016217, 11.4424816 ,
        9.43737523, 11.51811603,  7.26074925, 12.49291768,  8.08320235,
       12.41366754,  9.44486609, 13.55868684, 11.42947632,  6.29821673,
       11.73397309, 11.05672136,  7.62232307, 11.83509372, 14.10796466,
        6.38105015,  5.83560009,  9.54154096, 13.04051744,  5.07306274,
       13.1471313 ,  8.20436764, 10.41909211,  6.90115901, 13.46553623,
       11.62770332, 12.91143491, 10.35485331, 11.9702762 , 13.0493368 ,
        5.34250225,  5.76165934,  9.85384623, 10.04916678,  5.87305005,
        8.85572292,  4.91652556, 10.91805683, 11.9182956 , 13.6105595 ,
       10.33712258,  6.19633139,  8.41171369,  8.79284466,  9.3349858 ,
       10.91805683,  5.40689332,  6.67218687,  6.76056625,  5.87305005,
       14.02526505])}
{'P': tensor([[1.2390e-02, 2.0690e-01, 1.9297e-01,  ..., 1.4530e-06, 3.8425e-04,
         4.3637e-03],
        [4.5378e-04, 3.1232e-02, 2.3664e-03,  ..., 3.2904e-06, 2.0315e-03,
         2.0187e-02],
        [4.1520e-04, 7.1262e-02, 1.5691e-02,  ..., 1.3907e-06, 5.5585e-04,
         5.2648e-03],
        ...,
        [3.4591e-04, 2.9114e-04, 2.3117e-04,  ..., 1.5138e-01, 8.5775e-05,
         2.5202e-04],
        [1.1795e-05, 1.7385e-05, 1.3068e-05,  ..., 2.2844e-05, 6.6335e-02,
         1.3041e-04],
        [5.2517e-05, 1.0036e-04, 7.0656e-05,  ..., 8.0599e-06, 1.3773e-01,
         4.1888e-02]]), 'X': tensor([[ -7.2515,   5.0195,   4.6434],
        [ -8.1905,   3.7253,   4.1175],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.4573,   3.2448,   5.4370],
        [ -8.1666,   3.7653,   3.6075],
        [ -8.6603,   2.8250,   5.9181],
        [  0.1382, -11.0073,   1.0026],
        [  0.7275,  14.0667,  -0.9815],
        [ 11.7162,   5.6728,  -8.9372],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.6310, -10.9195,   3.8054],
        [  6.3236,   0.3011,  -4.6668],
        [  3.0876,   6.7527,  -5.3540],
        [  7.1697,  11.5343,  -0.7276],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.4167, -10.1293,   6.2156],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.8836,   3.4315,  -4.0327],
        [  8.0877,  -8.5297,  -1.1050],
        [  9.1409,  -7.5789,   1.5814],
        [ -7.5720,   4.6310,   2.1717],
        [ -9.4524,  -1.6813,  -6.6862],
        [ 11.3367,  -4.4806,   0.4643],
        [ -0.0410,   7.8591,   9.3457],
        [ -7.0825,   5.2073,  -5.0497],
        [ -6.0195,   6.1789,  -2.9496],
        [  2.1771, -10.9776,  -3.2007],
        [ -0.2586,   7.8825,  -9.9840],
        [ -9.2707,   0.9661,   4.4863],
        [ -2.5161,   7.7699,  -9.3531],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.3606,   0.9204,   3.2525],
        [ -9.3705,   0.4208,  -3.5815],
        [ 11.7162,   5.6728,  -8.9372],
        [  3.8111, -10.6787,  -4.6539],
        [ -3.6381,   7.4683,   7.3544],
        [  3.1271, -10.8343,   2.6442],
        [  5.6314,  -1.9302,  -9.6543],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.0345,   3.0175,  -1.4424],
        [  5.7218,  -9.9934,   6.5764],
        [ 12.5650,   2.0132,   4.6272],
        [ -3.3812, -10.1235,   7.4384],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.0864,   7.0387,  -0.1238],
        [  5.6672,  -1.8631,   8.1020],
        [ -9.1795,  -3.3623,   8.5259],
        [ -4.2271,   7.2387,  -1.8316],
        [ 12.5789,   0.1722,   1.6470],
        [  6.3119,   1.7797,   9.9186],
        [ -7.5720,   4.6310,   2.1717],
        [  8.1217,  -8.5024,  -0.6149],
        [ -9.4351,  -1.8891,   3.5642],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -9.3464,   0.5708,  -1.6397],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.1204,  -5.9308,   4.0761],
        [ -7.3333,   4.9246,  -7.9212],
        [ -5.1974,   6.7396,   3.3484],
        [ 10.5955,   7.8824,  -3.7006],
        [  7.8375,  -8.7240,  -9.8913],
        [  7.8982,  -8.6778,   0.0239],
        [  1.3339,   7.5627,  -0.9034],
        [ 12.4597,  -0.9136,  -4.8763],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.3616,  -2.4750,   3.7765],
        [  5.0348,  -2.8229,   3.5436],
        [  9.5203,   9.3483,   5.8781],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.0937,  13.8293,  -2.3533],
        [ -2.2188,   7.8218,   8.6064],
        [  3.9895,   6.0816,  -6.5235],
        [  1.5662,  13.9375,   4.5142],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.5859,  -5.0393,   0.5476],
        [ -7.0825,   5.2073,  -5.0497],
        [  9.0004,  -7.7195,   4.6945],
        [  6.3550,   0.7129,   0.4917],
        [  2.2437,  -4.6489,   6.1430],
        [ 10.5678,   7.9261,  -0.6100],
        [  5.2199,   4.6861,  -1.9304],
        [  5.5158,  -2.1337,   3.5246],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.6319,  11.9066,   3.5165],
        [ -6.3066,  -8.1936,   9.5822],
        [  6.2022,   2.4092,   3.0349],
        [  6.0305,   3.0296,   0.1375],
        [  6.9433,  -9.3362,  -0.3421],
        [  0.1287, -11.0065,   6.5757],
        [  3.6817,   6.3354,   6.1233],
        [  1.5653,  13.9377,   4.1773],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -8.7078,   2.7182,   1.8146],
        [  3.7121,   6.3117,   0.6738],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4541,  -0.3327,   8.8246],
        [  5.5626,  -2.0536,   8.4433],
        [  2.8474,   6.8979,   6.6457],
        [ -2.2508, -10.5479,  -2.7235],
        [  7.1697,  11.5343,  -0.7276],
        [  4.3554,   5.7393,  -9.0766],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.4283,  -1.9585,  -0.9129]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.7247e-02, 5.8447e-02,  ..., 9.4450e-01, 7.1670e-01,
         5.2562e-01],
        [9.7247e-02, 0.0000e+00, 9.1487e-02,  ..., 9.5602e-01, 6.3507e-01,
         4.3806e-01],
        [5.8447e-02, 9.1487e-02, 2.1453e-04,  ..., 9.9980e-01, 7.2655e-01,
         5.2890e-01],
        ...,
        [9.4450e-01, 9.5602e-01, 9.9980e-01,  ..., 0.0000e+00, 8.9104e-01,
         9.1488e-01],
        [7.1670e-01, 6.3507e-01, 7.2655e-01,  ..., 8.9104e-01, 0.0000e+00,
         2.0672e-01],
        [5.2562e-01, 4.3806e-01, 5.2890e-01,  ..., 9.1488e-01, 2.0672e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.81930525,  8.99791316,  8.92098752,  9.05843201,  8.99276564,
        9.10945678, 11.00812996, 14.08549802, 13.01728313, 12.24467041,
       11.23201619,  6.33075682,  7.42512621, 13.58101794,  9.94173708,
       11.48663767, 13.04217487,  6.81117802, 11.75438006, 11.87412439,
        8.87587457,  9.60080639, 12.18998195,  7.85919687,  8.7907923 ,
        8.62631801, 11.1913598 ,  7.88677538,  9.32094055,  8.16714836,
        9.79039371,  6.42689705,  9.37989631, 13.01728313, 11.33837396,
        8.30729352, 11.27656488,  5.95297459,  8.20436764,  6.74686071,
       11.51557166, 12.72524615, 10.67322527,  6.01356004, 13.13205357,
        5.96556248,  9.77588801,  8.38252937, 12.58006038,  6.55800267,
        8.87587457, 11.75807252,  9.62238828, 10.35485331,  9.36378438,
        8.29022197, 10.05558893,  8.83341608,  8.51088913, 13.20597932,
       11.72749923, 11.73397309,  7.67939546, 12.49317563,  5.70405525,
        9.68324553,  5.77219077, 13.3426541 , 12.59969936, 13.98691426,
        8.1303945 ,  7.27340834, 14.02526505,  9.0368458 ,  9.95552626,
        8.7907923 , 11.85743425,  6.3949043 ,  5.16205087, 13.20988996,
        7.01475376,  5.91408504,  9.0368458 ,  9.7712321 , 13.62896593,
       10.33958639,  6.65368551,  6.74872981, 11.63502679, 11.00726483,
        7.32754882, 14.02532758, 10.75443106,  9.48969978,  9.12220271,
        7.32233733, 13.09737053,  9.79039371,  9.45995955,  5.92952538,
        7.46248947, 10.78533838, 13.58101794,  7.20482224, 10.1979669 ,
        9.62958973])}
{'P': tensor([[1.8355e-02, 2.7784e-01, 1.8351e-01,  ..., 2.7244e-05, 4.0142e-06,
         3.3452e-05],
        [2.6285e-04, 2.9183e-02, 4.9314e-04,  ..., 5.6635e-05, 3.8849e-06,
         1.2646e-04],
        [7.4016e-04, 2.4943e-01, 1.8461e-02,  ..., 3.1802e-05, 4.0107e-06,
         5.1303e-05],
        ...,
        [2.0028e-04, 1.2283e-04, 1.7967e-04,  ..., 9.5700e-02, 1.3697e-03,
         1.3783e-04],
        [7.8594e-04, 4.5565e-04, 8.8752e-04,  ..., 1.1427e-04, 1.2250e-01,
         1.9557e-05],
        [6.8030e-06, 6.5568e-06, 7.1576e-06,  ..., 1.7446e-02, 7.7438e-05,
         1.1523e-01]]), 'X': tensor([[-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1365, 0.0532,  ..., 0.6955, 0.9917, 0.3502],
        [0.1365, 0.0000, 0.0921,  ..., 0.8288, 1.0769, 0.3961],
        [0.0532, 0.0921, 0.0000,  ..., 0.7369, 1.0014, 0.3383],
        ...,
        [0.6955, 0.8288, 0.7369,  ..., 0.0000, 0.6500, 0.6450],
        [0.9917, 1.0769, 1.0014,  ..., 0.6500, 0.0000, 0.6950],
        [0.3502, 0.3961, 0.3383,  ..., 0.6450, 0.6950, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.83341608,  9.09661212,  8.92395381,  9.17905491,  9.14044984,
        9.34127013, 11.54071859,  7.44805901,  7.81509631,  9.94173708,
       10.91601655, 10.54658597, 11.97957425,  9.58347456,  9.79039371,
        7.6374603 ,  5.16077764,  9.15866607, 11.785566  , 13.83139024,
       12.53047489, 11.35182589, 13.6105595 , 14.09176259,  7.06372172,
        6.00732193,  5.56809335,  7.14837514, 10.37698435, 13.99048094,
       10.794755  , 12.50996828,  7.02253235,  8.24808832,  6.05165248,
       12.4013153 ,  5.56809335,  7.27340834,  7.37576154,  5.5513072 ,
        9.50654967, 10.68031351, 10.37998078,  5.35774573,  7.98469166,
        6.60752373,  8.74470218,  8.31625171,  7.03733775, 13.56367857,
       12.91143491,  6.76791483, 11.018084  , 10.1371552 , 10.51437947,
        8.31625171,  5.0209005 , 13.88683436,  6.02789634,  5.70405525,
       12.62576026,  6.54127431, 13.93790235,  5.87305005, 11.18839169,
        6.652742  , 11.54071859, 13.34587027, 10.37698435,  7.50917551,
       12.85032836, 12.59840193,  7.28793663, 13.744387  ,  7.38844703,
       10.26545792,  6.18668463,  7.56617037,  9.23071698,  9.84314712,
       12.25806559,  5.82345867, 10.46739073, 10.98076873,  4.75303127,
        6.54127431, 14.03505624, 10.52357479,  9.15976746,  8.8580671 ,
        4.978704  , 13.77075174, 10.412299  ,  8.67227386,  9.88571713,
        8.41171369,  8.41148722,  8.62329466,  9.25336371, 10.51437947,
       13.50484539, 11.35345345, 10.00410625,  5.81199422,  7.50917551,
       11.05413545])}
{'P': tensor([[1.3622e-02, 3.3541e-01, 1.6722e-01,  ..., 2.6918e-03, 1.9211e-04,
         6.4408e-05],
        [2.3545e-04, 1.6047e-02, 3.2142e-01,  ..., 5.4930e-03, 1.0656e-04,
         1.0552e-04],
        [1.7354e-04, 2.8834e-04, 2.2026e-02,  ..., 1.3656e-02, 6.8193e-05,
         1.8231e-04],
        ...,
        [2.8228e-05, 3.6576e-05, 5.3847e-05,  ..., 7.3847e-02, 1.3857e-05,
         3.6593e-03],
        [1.5159e-01, 1.5710e-01, 1.3915e-01,  ..., 2.4283e-03, 9.1057e-02,
         5.5317e-05],
        [8.5600e-06, 1.0449e-05, 1.3285e-05,  ..., 6.9180e-04, 2.9209e-06,
         1.3176e-01]]), 'X': tensor([[ -7.3446,   4.9113,  -5.9356],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.7735,   2.5640,  -6.3313],
        [ -8.8363,   2.4086,  -5.6296],
        [ -7.6734,   4.4980,  -5.4248],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.2036,   2.4030,  -1.2500],
        [  5.2199,   4.6861,  -1.9304],
        [ 10.9599,  -5.1849,  -3.5526],
        [  4.4915,   5.5990,   7.0405],
        [ 12.4877,  -0.7222,   8.8347],
        [ 11.1085,  -4.9204,   0.0736],
        [  3.0876,   6.7527,  -5.3540],
        [  4.2096,   5.8815,   1.9230],
        [ -2.0840,   7.8415,  -7.8657],
        [  1.9204,   7.3518,   8.4781],
        [ 11.9936,  -2.8677,   6.0614],
        [  2.3376,  -4.6231,   9.8436],
        [ 11.4703,  -4.2006,   7.4086],
        [ -5.6169,  -8.7844,   8.9057],
        [  2.5813,  -4.5471,   5.3597],
        [  3.3744,   6.5623,  -6.3636],
        [ -8.7544,   2.6096,   3.3146],
        [  3.9895,   6.0816,  -6.5235],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.2555,   2.1474,   4.1685],
        [ 12.3357,   3.6001,   0.7903],
        [ -9.4697,  -0.6157,  -6.8472],
        [  4.5800,   5.5034,  -3.9432],
        [  9.4925,   9.3811,   8.9842],
        [  2.5375,   7.0676,  -4.3490],
        [ 10.0567,   8.6697,   4.6329],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 12.2692,  -1.8882,   3.9253],
        [ -9.4117,   0.1160,   8.2574],
        [ -5.6174,   6.4708,  -1.7339],
        [ -6.8618,  -7.6322,   3.7027],
        [ -7.6293,  -6.6859,  -5.5161],
        [  8.5716,  10.3596,   6.8818],
        [ -9.4630,  -1.5145,   0.2983],
        [  1.1139,   7.6281,  -2.5680],
        [  7.9816,  10.8938,  -0.5378],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.2182,  13.8007,   5.1889],
        [  2.0419,  -4.6978,  -5.0983],
        [ -8.9282,   2.1648,   8.4894],
        [ 12.3759,  -1.3943,   6.4659],
        [  0.1041, -11.0045,   2.4311],
        [ 11.4495,   6.3026,   6.8257],
        [  4.2339,  -3.6268,   6.3125],
        [ -8.8631,  -4.3728,  -1.1746],
        [  0.5960, -11.0335,   3.1144],
        [  8.0381,  -8.5690,  -8.9523],
        [  1.0093,   7.6567,   2.3526],
        [ 11.2517,  -4.6495,   2.9800],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.5149,  -5.1905,   1.5239],
        [  0.1931,  -4.7491,   7.4385],
        [ -4.2271,   7.2387,  -1.8316],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.7903,  -4.7467,   0.1168],
        [  5.6095,  -1.9702,  -7.8233],
        [ -4.5778,  -9.5013,  -5.0667],
        [  0.9036,   7.6840,   9.9237],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -4.3963,   7.1629,   4.5263],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.9936,  -2.8677,   6.0614],
        [ -7.9667,   4.0814,  -0.6397],
        [ -8.0611,  -6.0302,   4.6962],
        [ 12.5789,   0.1722,   1.6470],
        [  3.1064,  13.5607,   6.7605],
        [  1.8320,   7.3871,   5.1452],
        [  9.1620,  -7.5573,  -9.9882],
        [  3.9022,  13.2906,   5.5788],
        [  5.5626,  -2.0536,   8.4433],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.5636, -10.9293,   9.6519],
        [  6.2675,  12.1379,  -8.9720],
        [  3.9663,   6.1018,  -5.4104],
        [ -0.7599, -10.8995,  -6.0366],
        [  9.0129,  -7.7072,   9.1400],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.9433,  -9.3362,  -0.3421],
        [  7.3592,  11.3937,   6.2407],
        [ 12.2328,  -2.0376,  -1.1234],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.2911,  -2.4824,  -3.6877],
        [  3.1064,  13.5607,   6.7605],
        [ -3.3999, -10.1152,  -1.2088],
        [ 10.7566,   7.6212,  -2.3279],
        [ -1.6086,   7.8928,   4.3385],
        [  4.5800,   5.5034,  -3.9432],
        [  3.7916,  -3.9524,  -7.9547],
        [  2.7558,  -4.4844,   0.0636],
        [  6.1752,   2.5242,   3.5874],
        [ -4.4503,   7.1378,   3.3751],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.5158,  -2.1337,   3.5246],
        [  1.4306,  13.9620,  -0.2075],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -5.0550,   6.8231,  -9.2502],
        [ -6.4620,  -8.0449,   6.2719]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.6740e-02, 1.5702e-01,  ..., 3.3833e-01, 1.8558e-01,
         9.0471e-01],
        [7.6740e-02, 0.0000e+00, 8.1259e-02,  ..., 2.6968e-01, 2.5099e-01,
         8.9311e-01],
        [1.5702e-01, 8.1259e-02, 0.0000e+00,  ..., 2.1711e-01, 3.1802e-01,
         9.0589e-01],
        ...,
        [3.3833e-01, 2.6968e-01, 2.1711e-01,  ..., 0.0000e+00, 5.2044e-01,
         7.6521e-01],
        [1.8558e-01, 2.5099e-01, 3.1802e-01,  ..., 5.2044e-01, 2.1453e-04,
         1.0507e+00],
        [9.0471e-01, 8.9311e-01, 9.0589e-01,  ..., 7.6521e-01, 1.0507e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.8353674 ,  8.97052115,  9.14044984,  9.15866607,  8.89458273,
        9.34061938,  6.652742  ,  7.01475376, 12.12448912,  7.17789666,
       12.5085995 , 12.1494023 ,  7.42512621,  7.23276582,  8.1137452 ,
        7.5984809 , 12.33167382,  5.1805345 , 12.21532282, 10.42667093,
        5.2287058 ,  7.37902766,  9.13507321,  7.27340834, 11.59512194,
        6.61386436, 12.85032836,  9.48969978,  7.15990676, 13.34587027,
        7.5092834 , 13.2778378 ,  9.75075273, 12.41366754,  9.41244865,
        8.56889738, 10.26327895, 10.14436912, 13.44593519,  9.58347456,
        7.70897675, 13.50484539,  7.98469166, 13.97779839,  5.1224079 ,
        9.18690018, 12.45418236, 11.0050379 , 13.06957425,  5.5748624 ,
        9.88310614, 11.04953767, 11.74901778,  7.72291946, 12.17451685,
        5.62005629,  9.97220799,  4.75303127,  8.38252937, 10.27572673,
        5.07306274,  5.94541803, 10.54658597,  7.73692436, 12.3828957 ,
        9.5014146 ,  8.40445872,  5.8828986 , 12.33167382,  8.95133961,
       10.06703358, 12.58006038, 13.91198191,  7.61088125, 11.87665501,
       13.85158417,  5.92952538,  9.75075273, 11.22596994, 13.66051884,
        7.27759233, 10.92597196, 11.85890664, 11.80092578,  5.1224079 ,
       11.63502679, 13.56367857, 12.4013153 , 11.73848065,  5.84450127,
       13.91198191, 10.67132524, 13.18278606,  8.0550286 ,  7.15990676,
        5.47702766,  5.26342203,  6.67121748,  8.41148722,  8.1137452 ,
        9.70108409,  5.91408504, 14.03505624, 10.06407839,  8.49161154,
       10.31886066])}
{'P': tensor([[1.2363e-02, 3.2900e-01, 1.9165e-01,  ..., 4.2733e-06, 5.7882e-07,
         1.0989e-06],
        [2.1051e-04, 1.3225e-02, 3.2765e-01,  ..., 6.1820e-06, 4.6880e-07,
         1.3650e-06],
        [2.2243e-04, 3.2679e-04, 2.4268e-02,  ..., 1.6909e-05, 6.4046e-07,
         3.1606e-06],
        ...,
        [1.2556e-06, 1.0830e-06, 9.2507e-07,  ..., 2.2270e-02, 3.4404e-06,
         1.1322e-02],
        [1.9163e-04, 1.2019e-04, 6.6568e-05,  ..., 1.9424e-08, 8.6865e-03,
         8.0646e-08],
        [5.7711e-06, 4.1248e-06, 2.7879e-06,  ..., 7.0852e-06, 3.8703e-04,
         2.2747e-02]]), 'X': tensor([[ -7.3554,   4.8985,  -0.6665],
        [ -7.9667,   4.0814,  -0.6397],
        [ -8.6050,   2.9451,  -0.7012],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.0604,   1.7707,   1.1066],
        [  4.2339,  -3.6268,   6.3125],
        [  6.2804,   2.0019,   0.8607],
        [ -3.4488, -10.0934,   2.5195],
        [  5.4429,  -2.2529,   9.7306],
        [ -9.1813,  -3.3550,   1.8827],
        [ -0.5709,   7.9055,   7.6089],
        [ -6.0195,   6.1789,  -2.9496],
        [  9.1620,  -7.5573,  -9.9882],
        [  2.3928,   7.1404,  -1.7035],
        [ 12.3759,  -1.3943,   6.4659],
        [  5.5240,  -2.1198,   6.2055],
        [  6.2242,  -0.3171,   7.4978],
        [  9.3753,   9.5171,   3.9131],
        [ -8.8745,   2.3097,   3.5600],
        [  1.3670,  -4.7990,   3.2603],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.1504,   4.3674,  -7.2998],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.4932,   7.0903,  -5.4214],
        [ -3.5520,   7.4976,   9.1522],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.8930,  12.3592,  -8.9201],
        [ -5.3736,   6.6310,  -1.5174],
        [  8.1217,  -8.5024,  -0.6149],
        [ -0.5709,   7.9055,   7.6089],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.3301,   0.3638,  -7.5685],
        [  9.2811,  -7.4336,   5.1771],
        [ 10.4737,  -5.9570,  -1.2108],
        [  9.3836,   9.5075,  -5.9278],
        [  6.7565,  11.8237,  -4.9882],
        [  4.0362,  13.2397,  -7.0030],
        [  2.4356,  -4.5942,  -5.0808],
        [ 10.8670,  -5.3428,  -2.8637],
        [  1.7500,   7.4187,   4.4237],
        [ 12.5763,   0.1333,  -9.2650],
        [  5.1292,   4.8183,  -5.7997],
        [ -5.3382,   6.6533,   7.8054],
        [  3.2159, -10.8166,  -1.0406],
        [ 12.3453,  -1.5462,   7.0212],
        [ -9.4541,  -0.3327,   8.8246],
        [  3.9085,   6.1512,  -2.2560],
        [  0.1041, -11.0045,   2.4311],
        [  6.1973,  -0.4372,   6.5720],
        [  9.2979,   9.6048,   7.3138],
        [  3.1892, -10.8220,  -6.8409],
        [  0.6933,   7.7336,   2.1179],
        [  1.9505,  -4.7171,   9.9868],
        [ -4.4503,   7.1378,   3.3751],
        [ -1.8823, -10.6558,  -3.8952],
        [ 11.7418,   5.6070,  -1.4192],
        [  7.4132,  11.3526,   0.3322],
        [  4.5340,  -3.3629,   8.9652],
        [  1.9505,  -4.7171,   9.9868],
        [  0.1206,  -4.7363,   8.7103],
        [  6.3583,   1.1729,   8.7835],
        [ 12.2914,   3.8037,   6.0978],
        [  9.4925,   9.3811,   8.9842],
        [  4.5325,  -3.3643,   6.7699],
        [  9.3624,   9.5318,   0.2779],
        [  5.3860,  -2.3418,   1.0826],
        [ -3.3116, -10.1539,   7.5919],
        [  3.3943,   6.5483,   1.0790],
        [  2.7590,  13.6621,   9.4496],
        [  3.6817,   6.3354,   6.1233],
        [ 10.0619,   8.6627,  -9.2884],
        [  7.4132,  11.3526,   0.3322],
        [ 12.2311,  -2.0445,   1.3545],
        [  9.3753,   9.5171,   3.9131],
        [ -8.8934,   2.2597,   3.5831],
        [  6.1752,   2.5242,   3.5874],
        [  0.6933,   7.7336,   2.1179],
        [  6.3515,   0.6418,  -6.5823],
        [ -0.8455, -10.8853,  -6.7400],
        [  9.9217,  -6.7006,  -5.6728],
        [  6.3460,   1.4252,   1.2291],
        [  9.9750,  -6.6338,  -1.2440],
        [ 11.6634,   5.8050,   9.1590],
        [  2.6310, -10.9195,   3.8054],
        [ -9.3464,   0.5708,  -1.6397],
        [ -4.4779,  -9.5609,  -2.4505],
        [  4.0587, -10.6112,   6.2273],
        [ 10.5678,   7.9261,  -0.6100],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.1666,   3.7653,   3.6075],
        [ -3.3116, -10.1539,   7.5919],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.3858,   4.4260,  -9.6769],
        [  2.3261,   7.1726,   6.2522],
        [  6.8704,  -9.3809,   7.5769],
        [ -2.2508, -10.5479,  -2.7235],
        [ -8.7544,   2.6096,   3.3146],
        [  1.7228,  -4.7575,   8.3396],
        [  6.1730,  -0.5375,  -8.8861],
        [ -0.0410,   7.8591,   9.3457],
        [ -0.1626, -10.9796,   6.4629],
        [  5.7218,  -9.9934,   6.5764],
        [  6.0031,  12.2958,  -6.5992],
        [ 11.5083,  -4.1177,   0.2389]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0610, 0.1322,  ..., 0.6808, 0.9730, 0.5788],
        [0.0610, 0.0000, 0.0712,  ..., 0.7019, 1.0270, 0.6396],
        [0.1322, 0.0712, 0.0000,  ..., 0.7360, 1.0891, 0.7109],
        ...,
        [0.6808, 0.7019, 0.7360,  ..., 0.0000, 1.2862, 0.7011],
        [0.9730, 1.0270, 1.0891,  ..., 1.2862, 0.0000, 0.5856],
        [0.5788, 0.6396, 0.7109,  ..., 0.7011, 0.5856, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.83725081,  8.95133961,  9.09502385,  9.11473391,  9.14973319,
        9.23178204,  5.5748624 ,  6.59176018, 10.66631784,  5.89074104,
        9.7751189 ,  7.92607342,  8.62631801, 11.87665501,  7.53063885,
       12.45418236,  5.91677188,  6.2322769 , 13.35927527,  9.17016857,
        4.98989906, 12.35421461, 12.91143491, 10.35485331,  7.5158489 ,
        8.29640613,  5.18277858, 13.6922425 ,  8.53501777, 11.75807252,
        7.92607342, 12.32545637,  6.00732193,  6.34059021, 11.89105259,
       12.04924159, 13.35832725, 13.61801526, 13.84126197,  5.19986797,
       12.10942412,  7.62232307, 12.5769659 ,  7.03733775,  8.53014915,
       11.2845608 , 12.44177641,  9.45995955,  7.28793663, 11.0050379 ,
        6.21275016, 13.36800073, 11.28215429,  7.76457095,  5.10446653,
        8.41148722, 10.8207297 , 13.01187484, 13.55868684,  5.64500512,
        5.10446653,  4.73784222,  6.46560181, 12.86648771, 13.34587027,
        5.64464807, 13.3607379 ,  5.87305005, 10.68031351,  7.37576154,
       13.93790235,  7.32754882, 13.27717982, 13.55868684, 12.40074901,
       13.35927527,  9.17595899,  6.67121748,  7.76457095,  6.383897  ,
       10.91805683, 11.97238624,  6.50409372, 11.97949273, 13.02818702,
       11.23201619,  9.36378438, 10.5575601 , 11.36089355, 13.20988996,
        4.91652556,  8.99276564, 10.68031351, 10.0674887 ,  6.97106912,
        7.5403822 , 11.62770332, 10.78533838,  9.13507321,  5.05981914,
        6.19633139,  7.85919687, 10.98076873, 11.51557166, 13.6829843 ,
       12.22276278])}
{'P': tensor([[1.9117e-02, 2.1910e-01, 1.8501e-01,  ..., 2.1110e-03, 1.2786e-05,
         1.0528e-05],
        [8.1381e-04, 2.2062e-02, 1.3291e-01,  ..., 3.7044e-03, 1.4960e-05,
         1.1365e-05],
        [5.5656e-04, 1.2058e-03, 6.2444e-02,  ..., 4.4310e-03, 3.8103e-05,
         1.1481e-05],
        ...,
        [6.1812e-05, 9.2055e-05, 1.5604e-04,  ..., 5.7413e-02, 3.8270e-05,
         1.8932e-05],
        [4.8186e-04, 3.9078e-04, 1.8293e-04,  ..., 4.7906e-05, 5.5866e-02,
         4.9204e-04],
        [5.3755e-04, 7.1239e-04, 2.8728e-04,  ..., 3.0408e-03, 2.6429e-05,
         6.1744e-02]]), 'X': tensor([[ -7.4602,   4.7718,  -4.6277],
        [ -7.9298,   4.1367,  -3.9773],
        [ -8.8363,   2.4086,  -5.6296],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.9788,   2.0208,  -3.2867],
        [ -8.7735,   2.5640,  -6.3313],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -1.1885, -10.8211,   0.0865],
        [  5.9655,   3.2172,  -7.5953],
        [  3.3094,  13.4969,  -3.6507],
        [  5.4336,  -2.2676,   5.6933],
        [ -8.6426,  -4.9136,   9.4941],
        [ -3.5173, -10.0623,   4.8280],
        [ -7.0120,   5.2825,   7.7762],
        [  5.3804,  -2.3504,   7.2744],
        [  6.2804,   2.0019,   0.8607],
        [  5.6414, -10.0303,   7.8525],
        [ -2.6485, -10.4151,   0.8766],
        [  1.4150,  13.9647,  -4.0915],
        [  2.3889,  -4.6082,  -8.7291],
        [ -1.6086,   7.8928,   4.3385],
        [ 12.5224,  -0.4497,   4.8448],
        [ -1.6220,   7.8917,   0.0315],
        [  0.7275,  14.0667,  -0.9815],
        [  9.3607,   9.5337,  -4.5412],
        [  0.9809,  -4.8145,   6.8706],
        [ -2.6069, -10.4298,  -6.1995],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.6970,   6.9826,   9.7350],
        [  6.3246,   1.6696,  -0.1544],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.6248,   3.9984,   8.4933],
        [  6.1755,   2.5233,  -2.0042],
        [ -1.6275, -10.7221,   3.2376],
        [  8.5093,  -8.1764,  -3.2381],
        [ -5.1369,   6.7755,   8.3836],
        [ -4.2040,   7.2486,   2.9875],
        [  2.3928,   7.1404,  -1.7035],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.5763,   0.1333,  -9.2650],
        [  0.8572,  14.0501,  -8.3364],
        [ 11.5083,  -4.1177,   0.2389],
        [  6.2895,   0.0405,   5.7652],
        [ 11.4743,   6.2478,  -2.1100],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.9938,  -8.4745,   3.9735],
        [ -1.6334, -10.7207,   7.1397],
        [  3.4401,  13.4540,   4.6576],
        [ -8.6112,   2.9319,  -7.9943],
        [  5.6928,  12.4710,   3.6001],
        [ -9.4203,  -2.0340,   2.5856],
        [  0.3493,  -4.7727,  -6.5682],
        [ -3.3812, -10.1235,   7.4384],
        [  8.5093,  -8.1764,  -3.2381],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.5240,  -2.1198,   6.2055],
        [  6.2022,   2.4092,   3.0349],
        [  0.8572,  14.0501,  -8.3364],
        [  1.3767, -11.0341,   9.9415],
        [  6.2902,   0.0451,   3.7116],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.2767,  -4.6006,   0.7868],
        [  9.0652,   9.8597,  -3.5949],
        [  7.3724,  -9.0577,  -8.9843],
        [ -6.2200,  -8.2737,   5.5079],
        [  2.0243,   7.3086,   7.5432],
        [  6.1737,   2.5305,   3.9049],
        [  6.1752,   2.5242,   3.5874],
        [ 12.5927,   0.4199,  -9.4997],
        [ -5.6651,   6.4380,   1.3660],
        [ -5.9940,   6.1985,   5.9566],
        [ -1.3940, -10.7772,   6.7835],
        [  5.4867,  12.5816,   2.9549],
        [  4.1537,   5.9340,  -6.0423],
        [  8.7715,  -7.9390,   5.2900],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.9843,  -9.8679,  -6.1449],
        [  0.1047, -11.0046,   0.5188],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.0890,  -0.8382,   2.9166],
        [  2.1504,  -4.6726,   8.7908],
        [ -8.8934,   2.2597,   3.5831],
        [ 11.0864,   7.0387,  -0.1238],
        [ 11.9808,   4.9340,  -3.3715],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.5175,  14.0910,   4.6540],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.4901,  -0.7052,   1.7315],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.6093,  -1.9705,   6.4807],
        [  6.3505,   0.6235,  -8.5849],
        [  0.4996,   7.7738,  -4.4711],
        [ -9.4283,  -1.9585,  -0.9129],
        [  2.1771, -10.9776,  -3.2007],
        [ -9.3384,  -2.6161,  -4.7630],
        [  9.3560,   9.5391,   9.4943],
        [ -8.6112,   2.9319,  -7.9943],
        [  5.6022,  -1.9833,   8.3081],
        [  7.5064,  -8.9651,  -0.0994],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.3087,   1.8053,  -1.9481],
        [ -2.2188,   7.8218,   8.6064]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.5146e-02, 1.5705e-01,  ..., 6.3633e-01, 7.0548e-01,
         9.5259e-01],
        [5.5146e-02, 0.0000e+00, 1.3900e-01,  ..., 5.9143e-01, 7.4074e-01,
         9.3988e-01],
        [1.5705e-01, 1.3900e-01, 3.0339e-04,  ..., 6.6877e-01, 8.6091e-01,
         1.0713e+00],
        ...,
        [6.3633e-01, 5.9143e-01, 6.6877e-01,  ..., 0.0000e+00, 9.8263e-01,
         6.3074e-01],
        [7.0548e-01, 7.4074e-01, 8.6091e-01,  ..., 9.8263e-01, 0.0000e+00,
         7.7203e-01],
        [9.5259e-01, 9.3988e-01, 1.0713e+00,  ..., 6.3074e-01, 7.7203e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.85572292,  8.94395628,  9.15866607,  9.26455795,  9.20340031,
        9.14044984, 10.54658597, 10.88618114,  6.77777606, 13.89671314,
        5.88782913,  9.94173708, 10.65929968,  8.77913428,  5.87132718,
        6.59176018, 11.50791859, 10.74655346, 14.03618649,  5.19065439,
        8.0550286 , 12.53047489,  8.05668609, 14.08549802, 13.36092624,
        4.91336994, 10.7506423 , 12.53047489, 13.06498554,  5.8828986 ,
        7.48538905,  6.54127431, 12.00995898,  6.90115901,  6.67108662,
       10.84493732, 11.80092578,  8.50268322,  8.37954853,  7.53063885,
       11.52733581, 12.5769659 , 14.07623266, 12.22276278,  6.28961988,
       13.06498554,  8.24808832, 10.37998078, 10.84437705, 13.88683436,
        9.09661212, 13.70893492,  9.63742669,  4.78544875, 10.67322527,
       11.80092578,  5.57341623,  5.91677188,  6.65368551, 14.07623266,
       11.11969815,  6.29034958,  7.83486943, 12.17901033, 13.39372426,
       11.67875855, 10.35093485,  7.58378655,  6.67218687,  6.67121748,
       12.59969936,  8.57560275,  8.62260327, 10.86693708, 13.72594022,
        7.24326992, 11.83075005,  9.36378438,  9.78285424, 11.54071859,
       11.00508681,  5.84533345,  6.1463896 ,  5.14371005,  9.17595899,
       13.13205357, 12.95702647,  9.94285511, 14.10045797,  8.41171369,
       12.50996828, 13.01187484,  5.94535529,  6.38105015,  7.78980299,
        9.62958973, 11.1913598 ,  9.69792513, 13.3614598 ,  9.09661212,
        5.94293185, 11.69264243, 10.33712258,  9.62238828,  6.5618968 ,
        8.1303945 ])}
{'P': tensor([[1.7443e-02, 2.6571e-01, 2.2694e-01,  ..., 8.9649e-07, 1.7360e-04,
         1.5653e-04],
        [3.5066e-04, 3.6688e-02, 1.9699e-03,  ..., 1.1054e-06, 1.2713e-04,
         1.0898e-04],
        [4.3050e-04, 3.1337e-01, 2.9566e-02,  ..., 1.0074e-06, 1.3988e-04,
         1.0344e-04],
        ...,
        [2.4693e-03, 8.5987e-04, 1.1440e-03,  ..., 8.2473e-02, 7.9626e-03,
         1.7326e-02],
        [6.0332e-03, 1.0178e-02, 1.2832e-02,  ..., 8.2122e-06, 6.8880e-02,
         3.4232e-03],
        [1.0072e-01, 6.7176e-02, 9.2619e-02,  ..., 6.6341e-06, 6.2407e-03,
         3.4272e-02]]), 'X': tensor([[ -7.4733,   4.7555,   6.2490],
        [ -8.6603,   2.8250,   5.9181],
        [ -8.4573,   3.2448,   5.4370],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.9178,   2.1936,   7.5388],
        [ -9.3444,   0.5821,   6.9101],
        [  9.8082,  -6.8396,   4.9615],
        [  5.7845,   3.6652,   0.9175],
        [ -8.8400,   2.3991,   1.3457],
        [ -2.6069, -10.4298,  -6.1995],
        [  5.5158,  -2.1337,   3.5246],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.9788,   2.0208,  -3.2867],
        [ -1.9646,   7.8571,   3.3817],
        [  6.6319,  11.9066,   3.5165],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.5506,  -5.1154,   6.3423],
        [  4.1537,   5.9340,  -6.0423],
        [ 10.9916,   7.2133,  -0.9179],
        [  9.0652,   9.8597,  -3.5949],
        [  0.6396,  14.0772,  -6.3786],
        [ -1.6086,   7.8928,   4.3385],
        [  6.0305,   3.0296,   0.1375],
        [  4.3554,   5.7393,  -9.0766],
        [ -9.4756,  -0.8166,  -9.6021],
        [  1.1139,   7.6281,  -2.5680],
        [  6.2419,  -0.2313,   4.2440],
        [ -8.4267,  -5.3697,   2.4172],
        [ -3.1671,   7.6158,   6.0765],
        [ -9.3331,  -2.6465,  -8.2028],
        [  2.3261,   7.1726,   6.2522],
        [ -9.4773,  -1.0024,   5.2599],
        [ 11.2517,  -4.6495,   2.9800],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.7485,  -9.9811,  -3.2355],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.7151,  -1.7700,   3.1542],
        [  6.3301,   0.3638,  -7.5685],
        [  9.4516,  -7.2501,  -1.8868],
        [ 11.2939,   6.6325,  -5.8311],
        [  7.8982,  -8.6778,   0.0239],
        [  3.7916,  -3.9524,  -7.9547],
        [ -8.5859,  -5.0393,   0.5476],
        [ -3.3116, -10.1539,   7.5919],
        [ -2.9497, -10.3028,  -7.3479],
        [  2.5663,   7.0526,  -6.8769],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.3691,   6.4760,   1.0787],
        [ -4.3963,   7.1629,   4.5263],
        [ -6.4368,   5.8356,   3.3014],
        [  6.2535,  -0.1709,  -9.4102],
        [ -1.8788, -10.6567,   3.7068],
        [ -1.5875,   7.8944,  -9.4286],
        [  2.0243,   7.3086,   7.5432],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.0506,   4.9277,  -9.5094],
        [  5.2608,  12.6977,   4.9166],
        [  0.8700,  -4.8132,  -7.8042],
        [  1.6308,  13.9254,  -2.0984],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.2914,   3.8037,   6.0978],
        [  2.7121,   6.9743,  -8.4202],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.3589,   0.8239,  -8.4230],
        [  5.7218,  -9.9934,   6.5764],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.8162,   7.7059,  -9.6885],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -4.0409,   7.3169,  -3.5779],
        [ -3.6381,   7.4683,   7.3544],
        [  5.5240,  -2.1198,   6.2055],
        [  2.7590,  13.6621,   9.4496],
        [  5.8275,  -1.5334,   2.0124],
        [  6.1730,  -0.5375,  -8.8861],
        [ -9.1178,   1.5784,   2.1410],
        [  6.9956,  11.6590,  -7.6876],
        [  5.3804,  -2.3504,   7.2744],
        [  3.9895,   6.0816,  -6.5235],
        [  5.1886,  -2.6250,   6.4677],
        [ -7.9298,   4.1367,  -3.9773],
        [  0.4119,  14.1019,  -4.1870],
        [ -9.3616,  -2.4749,   0.1793],
        [  1.8447,  -4.7372,   8.1235],
        [  6.1222,  -0.7261,   8.8204],
        [  3.6703,   6.3443,  -5.4401],
        [  4.5503,   5.5359,   7.4735],
        [  1.8447,  -4.7372,   8.1235],
        [ -1.5875,   7.8944,  -9.4286],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.0587, -10.6112,   6.2273],
        [  6.3535,   0.6792,  -7.5101],
        [ 10.9875,   7.2207,  -8.8549],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.3569,   0.7591,   6.5011],
        [  0.1252, -11.0062,   7.2564],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.8201,   3.5841,  -8.0609],
        [  5.6260,  -1.9401,   3.0259],
        [  7.0515,  11.6194,   8.4289],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.9003,  12.3550,   2.1917],
        [ -5.6174,   6.4708,  -1.7339],
        [ -4.7389,   6.9953,   3.7638]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2374e-01, 1.1442e-01,  ..., 9.1985e-01, 5.9478e-01,
         2.2974e-01],
        [1.2374e-01, 2.1453e-04, 3.7662e-02,  ..., 1.0317e+00, 5.9872e-01,
         3.1580e-01],
        [1.1442e-01, 3.7662e-02, 0.0000e+00,  ..., 1.0046e+00, 5.6246e-01,
         2.8376e-01],
        ...,
        [9.1985e-01, 1.0317e+00, 1.0046e+00,  ..., 0.0000e+00, 8.8125e-01,
         7.2822e-01],
        [5.9478e-01, 5.9872e-01, 5.6246e-01,  ..., 8.8125e-01, 0.0000e+00,
         4.3707e-01],
        [2.2974e-01, 3.1580e-01, 2.8376e-01,  ..., 7.2822e-01, 4.3707e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.8580671 ,  9.10945678,  9.05843201,  9.09119351,  9.18358562,
        9.36256096, 11.95743558,  6.84796202,  9.15976746, 10.7506423 ,
        5.91408504,  7.8575648 ,  9.20340031,  8.09899874, 13.62896593,
        9.37989631,  9.96391045,  7.24326992, 13.1471313 , 13.39372426,
       14.09176259,  8.0550286 ,  6.74872981,  7.20482224,  9.51074763,
        7.70897675,  6.24614905,  9.99211663,  8.24808832,  9.70108409,
        7.5403822 ,  9.53015376, 12.17451685,  5.39450305, 11.51811603,
       12.19805394,  5.98285319,  6.34059021, 11.9120321 , 13.09737053,
       11.73397309,  5.47702766,  9.95552626, 10.68031351, 10.7167285 ,
        7.50500064, 11.81423161, 13.08414689,  8.40445872,  8.68832398,
        6.25586297, 10.82106505,  8.05242591,  7.58378655,  5.26067019,
        7.05627461, 13.744387  ,  4.89121805, 14.02058621, 12.33167382,
       12.86648771,  7.48311002, 13.09426142,  6.02789634,  6.41202986,
       11.51557166, 13.86480175,  8.20436764, 12.0495521 ,  9.48969978,
        8.35856487,  8.30729352,  5.91677188, 13.93790235,  6.02588517,
        6.19633139,  9.25336371, 13.59673406,  5.87132718,  7.27340834,
        5.81481044,  8.94395628, 14.10796466,  9.68323686,  5.08372149,
        6.16512856,  7.32950385,  7.16598273,  5.08372149,  8.05242591,
        9.38232121, 11.36089355,  6.38968357, 13.14777433, 13.1399181 ,
       11.57249984,  6.4020404 , 11.00694688, 10.78533838,  6.83515201,
        5.95111804, 13.59170809,  5.79391825, 13.69162795,  8.56889738,
        8.44940661])}
{'P': tensor([[1.1169e-02, 2.0104e-01, 1.7391e-01,  ..., 3.3322e-06, 7.2791e-06,
         9.8395e-06],
        [6.2303e-04, 1.3089e-02, 2.2483e-01,  ..., 4.1250e-06, 8.3494e-06,
         1.1536e-05],
        [1.4899e-04, 1.5468e-04, 1.4752e-02,  ..., 8.7594e-06, 7.5897e-06,
         1.3914e-05],
        ...,
        [2.6564e-06, 2.6185e-06, 1.9769e-06,  ..., 4.0725e-02, 3.9131e-04,
         2.4367e-03],
        [2.0964e-03, 1.9669e-03, 8.9457e-04,  ..., 8.3796e-06, 6.0649e-02,
         6.5014e-04],
        [2.7072e-04, 2.5820e-04, 1.4253e-04,  ..., 3.7099e-05, 8.9801e-02,
         4.8066e-02]]), 'X': tensor([[ -7.4807,   4.7463,   2.0616],
        [ -7.5720,   4.6310,   2.1717],
        [ -8.6229,   2.9068,   2.4429],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.8154,   2.4611,   2.5198],
        [ -8.8400,   2.3991,   1.3457],
        [ -4.4779,  -9.5609,  -2.4505],
        [  1.3103,  -4.8032,  -0.9300],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.3589,  13.7668,  -3.6501],
        [  6.7565,  11.8237,  -4.9882],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.5199,  -0.4711,   3.0436],
        [  5.6314,  -1.9302,  -9.6543],
        [ -7.0120,   5.2825,   7.7762],
        [  6.2895,   0.0405,   5.7652],
        [  4.9957,  -2.8705,   6.5978],
        [  5.7151,  -1.7700,   3.1542],
        [ -8.0829,  -5.9940,   5.0564],
        [  7.3592,  11.3937,   6.2407],
        [ 11.3118,   6.5957,  -4.5667],
        [ -5.3382,   6.6533,   7.8054],
        [  6.3119,   1.7797,   9.9186],
        [  6.2883,   0.0325,   5.2692],
        [ -6.8466,  -7.6488,   1.9648],
        [  1.5662,  13.9375,   4.5142],
        [ 11.9808,   4.9340,  -3.3715],
        [ 12.5267,  -0.4125,  -8.7331],
        [  6.1066,  -0.7799,  -4.1746],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 10.6283,  -5.7253,   7.4878],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -8.0766,  -6.0046,  -3.1873],
        [  3.4009,  -4.1874,  -5.9751],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.1196,  -2.7160,   4.2705],
        [  2.6614,  -4.5192,  -9.3641],
        [  5.4242, -10.1261,   3.9595],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 11.1469,   6.9237,   1.6160],
        [  0.5250, -11.0306,   5.5358],
        [  6.3551,   0.7149,   1.3344],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.1196,  -2.7160,   4.2705],
        [  0.3493,  -4.7727,  -6.5682],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.7162,   5.6728,  -8.9372],
        [ -0.8569,   7.9156,  -6.5557],
        [ -4.5778,  -9.5013,  -5.0667],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.0039,   7.3173,  -1.5885],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -9.2508,   1.0572,   2.9978],
        [  3.9913,  -3.8139,  -6.8311],
        [ -8.7409,  -4.6837,  -6.5840],
        [  6.3606,   0.9204,   3.2525],
        [  1.1644, -11.0394,   7.8511],
        [ -1.6906,   7.8860,   7.7142],
        [ -8.8934,   2.2597,   3.5831],
        [ -3.7964,  -9.9294,   7.7147],
        [ -7.0120,   5.2825,   7.7762],
        [  5.2188,  -2.5839,   2.1833],
        [  6.6319,  11.9066,   3.5165],
        [  5.0233,   4.9647,  -3.1788],
        [ 10.8218,  -5.4177,   6.8573],
        [  4.6992,   5.3690,   3.0935],
        [ -5.0943,  -9.1682,   0.5073],
        [ -9.4117,   0.1160,   8.2574],
        [ -2.0229, -10.6163,   8.6930],
        [  4.2612, -10.5514,  -6.6268],
        [ -6.7992,   5.4991,   9.9186],
        [ -1.5875,   7.8944,  -9.4286],
        [  7.4132,  11.3526,   0.3322],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.0955,   2.8210,  -4.3321],
        [  5.9526,   3.2525,  -4.3821],
        [ -3.0004,   7.6607,  -7.2640],
        [  4.6231,   5.4556,   8.4973],
        [  6.3246,   1.6696,  -0.1544],
        [  0.9478,  -4.8144,   0.6371],
        [  2.2437,  -4.6489,   6.1430],
        [ -6.2668,   5.9808,   2.5614],
        [  2.0899,  -4.6870,  -7.4260],
        [ -8.9178,   2.1936,   7.5388],
        [  5.8318,   3.5567,  -4.7692],
        [  5.2199,   4.6861,  -1.9304],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.3589,   0.8239,  -8.4230],
        [ -1.6220,   7.8917,   0.0315],
        [  2.3367,   7.1676,  -9.5263],
        [  8.8474,  10.0868,  -8.8644],
        [  4.3377,   5.7570,   1.3712],
        [  5.3602,  -2.3808,   7.2098],
        [  2.9410,   6.8428,   5.4458],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.9019,  -3.1379,  -3.4534],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.8093,  -9.4178,   8.4244],
        [  5.1890,   4.7318,   0.9515],
        [  6.3200,   0.2690,   1.5900]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0156, 0.0990,  ..., 0.5616, 0.6144, 0.6535],
        [0.0156, 0.0000, 0.0841,  ..., 0.5620, 0.6292, 0.6679],
        [0.0990, 0.0841, 0.0000,  ..., 0.5975, 0.7133, 0.7518],
        ...,
        [0.5616, 0.5620, 0.5975,  ..., 0.0000, 0.6689, 0.6642],
        [0.6144, 0.6292, 0.7133,  ..., 0.6689, 0.0000, 0.0542],
        [0.6535, 0.6679, 0.7518,  ..., 0.6642, 0.0542, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.85939626,  8.87587457,  9.09963894,  9.12220271,  9.15253344,
        9.15976746, 10.5575601 ,  4.978704  , 13.00413829, 13.96746816,
       13.61801526,  4.92497514, 12.4980845 , 12.76408562, 12.52875793,
        5.95297459,  8.77913428,  6.28961988,  5.76165934,  5.98285319,
       10.06286074, 13.56367857, 13.09426142,  8.53014915,  6.55800267,
        6.28835101, 10.26545792, 14.02526505, 12.95702647, 12.53345445,
        6.1561646 , 12.4217523 , 12.07225469, 10.00410625, 10.06407839,
        5.39450305, 13.25336001,  8.95983586,  9.77777346, 10.01804223,
        5.79543038,  5.24461068, 11.48734978, 12.42554808, 13.12218013,
       11.04313641,  6.39520397,  9.79039371,  5.79543038,  4.78544875,
        5.04028538, 13.01728313,  7.96181288, 10.54658597,  5.84450127,
        7.58668144, 12.07818871,  9.3109884 ,  5.5205189 ,  9.91668182,
        6.42689705, 11.10066126,  8.06517143,  9.17595899, 10.63038565,
        8.77913428,  5.82345867, 13.62896593,  7.06272256, 12.10222204,
        7.13501264, 10.48840476,  9.41244865, 10.80728788, 11.37939695,
        8.74470218,  8.05242591, 13.55868684,  9.50913849,  6.7166297 ,
        6.78325151,  8.22728399,  7.15100303,  6.54127431,  4.90676242,
        5.16205087,  8.66273171,  5.13182506,  9.18358562,  6.83083863,
        7.01475376,  5.27178363,  6.41202986,  8.05668609,  7.53884104,
       13.41713661,  7.20826219,  5.86519525,  7.44805901,  5.22903117,
       12.30858677,  5.53175275,  6.02789634, 11.62158554,  7.02253235,
        6.32572323])}
{'P': tensor([[1.3173e-02, 2.0244e-01, 1.9803e-01,  ..., 1.3524e-05, 6.7921e-06,
         4.2095e-07],
        [2.0213e-04, 1.9854e-02, 4.9464e-02,  ..., 2.3312e-05, 8.3211e-06,
         4.7252e-07],
        [4.0458e-04, 1.0363e-02, 4.5040e-02,  ..., 5.8020e-05, 1.7380e-05,
         9.9649e-07],
        ...,
        [1.7934e-04, 1.0260e-04, 1.0557e-04,  ..., 7.1264e-02, 2.4725e-03,
         9.2560e-04],
        [3.7709e-03, 1.8808e-03, 1.5664e-03,  ..., 3.6470e-04, 1.6233e-01,
         1.3163e-02],
        [1.5822e-04, 6.0582e-05, 5.0426e-05,  ..., 5.9849e-06, 1.4300e-03,
         2.3584e-02]]), 'X': tensor([[-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0976, 0.1171,  ..., 0.7091, 0.6794, 1.1418],
        [0.0976, 0.0000, 0.0543,  ..., 0.8047, 0.7543, 1.2140],
        [0.1171, 0.0543, 0.0000,  ..., 0.8028, 0.7931, 1.2546],
        ...,
        [0.7091, 0.8047, 0.8028,  ..., 0.0000, 0.6379, 0.9518],
        [0.6794, 0.7543, 0.7931,  ..., 0.6379, 0.0000, 0.4631],
        [1.1418, 1.2140, 1.2546,  ..., 0.9518, 0.4631, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.859416  ,  9.09963894,  9.12220271,  8.87587457,  9.15253344,
        9.15976746, 11.04313641, 12.50996828,  6.67218687,  7.34250948,
        5.53175275, 10.35093485,  8.83341608,  8.53014915, 10.61417181,
        8.74470218, 10.07238463,  7.39004226, 11.75438006,  9.43737523,
       13.59673406, 11.37939695, 11.42947632,  9.47369628, 10.0674887 ,
       13.59673406, 11.04313641,  5.5366698 , 13.06957425, 13.27717982,
        4.98989906, 14.07623266,  7.37576154, 10.23946247,  6.46560181,
        8.50278896,  7.37902766, 10.4977132 ,  8.23525433, 12.78327037,
       13.58333857,  6.41202986, 12.27835996,  5.68942572,  6.80370254,
        8.7273569 ,  9.54154096,  9.20340031,  6.02588517, 10.14436912,
        9.69629   , 10.82106505, 10.91805683, 12.50996828,  7.26074925,
        6.81084724,  9.87720102, 13.98691426, 10.27716434,  5.5513072 ,
       13.0743399 ,  9.36378438, 13.52430176,  6.21275016,  7.822854  ,
       10.7167285 , 10.00410625, 13.06957425, 10.87409005,  6.39187285,
        8.41148722,  6.67108662, 11.63765273,  8.85572292,  9.69629   ,
       13.58333857,  6.38105015,  4.73784222, 11.89855267, 12.0495521 ,
        8.62162372,  5.37343173,  8.57560275, 11.65923396,  9.36256096,
        4.92497514, 11.75807252, 13.13205357,  5.94909709,  8.68979052,
       13.62896593,  6.12886024, 10.46739073,  6.54693987,  9.27474386,
        9.85330646,  9.69629   ,  4.98461983, 13.59480906, 10.14044489,
        8.35856487,  9.71392179, 13.92390689,  6.02789634,  7.06169391,
       13.36800073])}
{'P': tensor([[7.6521e-03, 1.9637e-01, 1.4641e-01,  ..., 2.4195e-05, 5.7675e-07,
         5.7373e-02],
        [2.0967e-04, 1.1571e-02, 1.4636e-01,  ..., 4.0139e-05, 9.6969e-07,
         7.7616e-02],
        [2.8745e-04, 9.0548e-04, 1.5387e-02,  ..., 5.1901e-05, 1.3238e-06,
         7.4776e-02],
        ...,
        [5.4613e-05, 5.0336e-05, 4.7346e-05,  ..., 9.2821e-02, 2.0153e-03,
         6.6232e-05],
        [7.8986e-05, 5.7780e-05, 5.2626e-05,  ..., 1.9515e-04, 2.8291e-01,
         3.2140e-05],
        [1.2535e-03, 2.4195e-03, 3.4095e-03,  ..., 1.0928e-03, 1.7409e-05,
         1.4137e-01]]), 'X': tensor([[ -7.4887,   4.7364,   8.9074],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.9282,   2.1648,   8.4894],
        [ -7.5401,   4.6718,   9.0033],
        [  4.3523, -10.5232,   8.2138],
        [  5.7784,  -1.6401,   9.5587],
        [ -5.6423,   6.4537,  -1.2091],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.5250, -11.0306,   5.5358],
        [  2.3490,  -4.6199,  -4.2489],
        [ -6.2668,   5.9808,   2.5614],
        [  3.5965,  -4.0754,  -3.5035],
        [ -5.0132,  -9.2234,  -8.3531],
        [  5.3924, -10.1396,  -4.0863],
        [ 10.0567,   8.6697,   4.6329],
        [ 10.2470,   8.4054,  -0.1177],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.4932,   7.0903,  -5.4214],
        [  9.3624,   9.5318,   0.2779],
        [ -8.5900,   2.9767,   7.4811],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.8704,  -9.3809,   7.5769],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.8450,  -9.9356,   3.0767],
        [  0.5850,  -4.7982,   9.6639],
        [ -6.3489,   5.9117,   6.5161],
        [  1.6422,   7.4586,  -6.8056],
        [  6.3413,   0.4883,  -6.8416],
        [ -7.5603,   4.6459,   8.2002],
        [ -7.6734,   4.4980,  -5.4248],
        [ -5.1377,   6.7751,  -0.4438],
        [ -9.1960,  -3.2964,   1.6921],
        [  5.0506,   4.9277,  -9.5094],
        [ -8.1426,   3.8048,   9.3542],
        [  4.8674,  12.8875,   6.7622],
        [  3.5939,   6.4029,  -1.8357],
        [  0.3038,   7.8092,   0.4164],
        [ -9.4739,  -1.2497,   0.5483],
        [  4.3459,  -3.5327,   0.3667],
        [ -0.0410,   7.8591,   9.3457],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.3087,   1.8053,  -1.9481],
        [  2.2644,  -4.6434,   9.7508],
        [  6.3505,   0.6235,  -8.5849],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.1287, -11.0065,   6.5757],
        [  0.3715,  -4.7756,   0.2312],
        [ -9.3464,   0.5708,  -1.6397],
        [  3.8182,  -3.9347,  -1.7961],
        [ 12.5996,   1.3971,  -1.0694],
        [ -2.3317,   7.8034,  -3.6274],
        [  0.5250, -11.0306,   5.5358],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.9967,  -4.8144,  -9.7370],
        [ -1.0482, -10.8487,  -5.9066],
        [  0.6396,  14.0772,  -6.3786],
        [  5.3602,  -2.3808,   7.2098],
        [  5.6248,   3.9984,   8.4933],
        [  3.5284, -10.7484,   5.3347],
        [ -9.4157,   0.0821,  -5.0025],
        [ -6.0844,  -8.3955,   0.7705],
        [ -5.2589,   6.7024,  -5.5809],
        [  2.5974,   7.0363,  -7.7810],
        [  2.3376,  -4.6231,   9.8436],
        [ -5.0550,   6.8231,  -9.2502],
        [ 10.0995,  -6.4739,   2.0974],
        [ -5.7347,  -8.6908,   9.2878],
        [ -8.9788,   2.0208,  -3.2867],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.1519, -10.2387,   9.8191],
        [  0.3038,   7.8092,   0.4164],
        [  2.2860,   7.1916,  -6.1011],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.8201,   3.5841,  -8.0609],
        [  0.9036,   7.6840,   9.9237],
        [  6.8407,  11.7666,   3.0045],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.9178,   2.1936,   7.5388],
        [  1.6308,  13.9254,  -2.0984],
        [ -6.4620,  -8.0449,   6.2719],
        [  4.8432,   5.1968,   1.5006],
        [  1.9503,  -4.7172,  -6.2313],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -8.8835,  -4.3178,   9.3080],
        [ -2.0840,   7.8415,  -7.8657],
        [ -3.5520,   7.4976,   9.1522],
        [  3.1064,  13.5607,   6.7605],
        [  6.2365,  -0.2578,  -6.0212],
        [  1.9204,   7.3518,   8.4781],
        [ -6.1899,  -8.3011,  -7.9269],
        [  5.0277,   4.9589,   7.0175],
        [ -9.4366,  -0.1189,  -2.6736],
        [  3.9591, -10.6391,  -6.2557],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.4297,  -5.3637,  -9.3962],
        [  3.2159, -10.8166,  -1.0406],
        [  2.5620,   7.0549,   8.4825],
        [  0.9945,  -4.8144,  -0.1897],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.9178,   2.1936,   7.5388]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0527, 0.0681,  ..., 0.8000, 0.6106, 0.1378],
        [0.0527, 0.0000, 0.0155,  ..., 0.8421, 0.6606, 0.0965],
        [0.0681, 0.0155, 0.0000,  ..., 0.8553, 0.6750, 0.0875],
        ...,
        [0.8000, 0.8421, 0.8553,  ..., 0.0000, 0.6977, 0.8591],
        [0.6106, 0.6606, 0.6750,  ..., 0.6977, 0.0000, 0.7462],
        [0.1378, 0.0965, 0.0875,  ..., 0.8591, 0.7462, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.86082124,  8.95983586,  8.98765644,  9.0368458 ,  9.18690018,
        8.87007495, 11.38774149,  6.00663314,  8.5724042 ,  7.96181288,
       11.04313641,  5.18277858,  8.66273171,  5.43544626, 10.4977132 ,
       11.48434767, 13.2778378 , 13.25336001, 12.0495521 , 11.63724792,
        7.5158489 , 13.3607379 ,  9.09119351,  6.25586297, 11.62770332,
       10.61472636, 11.52733581,  4.83371341,  8.67502348,  7.63726134,
        6.36004393,  8.8737556 ,  8.89458273,  8.50278896,  9.76896904,
        7.05627461,  8.98765644, 13.77604293,  7.34250948,  7.81509631,
        9.55593374,  5.60064319,  7.85919687, 12.07225469,  6.5618968 ,
        5.16611782,  6.38105015, 10.00410625, 11.00726483,  4.79003361,
        9.36378438,  5.48276038, 12.67680439,  8.14433542, 11.04313641,
       10.4606265 ,  5.57341623,  4.91652556, 10.8992524 , 14.09176259,
        5.86519525,  6.90115901, 11.31276109,  9.41605684, 10.36844813,
        8.51928531,  7.50035355,  5.1805345 ,  8.49161154, 11.99633819,
       10.412299  ,  9.20340031, 10.26545792,  4.91296205, 11.46176259,
        7.81509631,  7.54620849,  5.57341623,  6.83515201,  7.73692436,
       13.6105595 , 12.27993098,  9.18358562, 14.02058621, 10.31886066,
        7.10378919,  5.10444097, 12.52372629,  9.91668182,  9.87720102,
        8.1137452 ,  8.29640613, 13.91198191,  6.24187044,  7.5984809 ,
       10.35485331,  7.06169391,  9.43737523, 11.35182589,  8.92098752,
        9.99145096, 11.2845608 ,  7.50564223,  4.9160962 , 12.30695098,
        9.18358562])}
{'P': tensor([[9.6391e-03, 2.7471e-01, 2.0725e-01,  ..., 2.0951e-06, 1.0226e-03,
         3.6642e-08],
        [2.1531e-04, 1.2150e-02, 1.5368e-01,  ..., 3.1855e-06, 1.7853e-03,
         4.6610e-08],
        [2.3897e-04, 7.7104e-04, 1.3103e-02,  ..., 3.5887e-06, 2.0185e-03,
         4.7838e-08],
        ...,
        [1.2608e-05, 1.0405e-05, 9.6832e-06,  ..., 1.2383e-01, 6.0559e-06,
         1.1098e-03],
        [6.3873e-06, 7.4133e-06, 7.7033e-06,  ..., 5.9167e-04, 2.2949e-02,
         2.3917e-06],
        [1.8196e-06, 1.4009e-06, 1.2154e-06,  ..., 6.7376e-06, 4.4672e-07,
         8.8911e-02]]), 'X': tensor([[ -7.5401,   4.6718,   9.0033],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.0574,   1.7801,   8.8986],
        [  9.0051,   9.9234,   5.4411],
        [ -8.9636,   2.0648,   8.1946],
        [  4.8674,  12.8875,   6.7622],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.1892, -10.8220,  -6.8409],
        [  0.5250, -11.0306,   5.5358],
        [  3.4604,  -4.1545,   1.8091],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.3419,   0.5972,  -1.2841],
        [  9.2692,   9.6370,   7.0746],
        [  3.6469,   6.3624,  -1.8414],
        [  5.6523,   3.9443,  -2.9449],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.4140,  -4.1803,  -8.8224],
        [  0.3038,   7.8092,   0.4164],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -6.8466,  -7.6488,   1.9648],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.7544,  -3.1425,  -1.5315],
        [  4.6992,   5.3690,   3.0935],
        [  8.6420,  -8.0580,   6.5781],
        [  2.5128, -10.9365,   9.9645],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.0190,   4.9705,  -7.9752],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 11.0373,   7.1299,  -8.5895],
        [  2.1440, -10.9811,   8.5281],
        [  1.8447,  -4.7372,   8.1235],
        [  5.8836,   3.4315,  -4.0327],
        [  5.3924, -10.1396,  -4.0863],
        [ -5.1369,   6.7755,   8.3836],
        [  1.5662,  13.9375,   4.5142],
        [ -8.1905,   3.7253,   4.1175],
        [ 12.5199,  -0.4711,   3.0436],
        [ -5.0550,   6.8231,  -9.2502],
        [  5.2199,   4.6861,  -1.9304],
        [  3.3069,   6.6089,   8.7377],
        [  8.7715,  -7.9390,   5.2900],
        [  5.3858,   4.4260,  -9.6769],
        [  2.9260,  -4.4162,   8.3808],
        [ -7.4807,   4.7463,   2.0616],
        [  4.0421,  -3.7765,  -8.3631],
        [  2.3376,  -4.6231,   9.8436],
        [  6.3301,   0.3638,  -7.5685],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.3205,   1.7068,  -3.8619],
        [  8.7715,  -7.9390,   5.2900],
        [  2.5469,  13.7192,  -9.6935],
        [ -5.0132,  -9.2234,  -8.3531],
        [  4.1999,  -3.6543,  -4.1998],
        [  3.8111, -10.6787,  -4.6539],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -6.6895,   5.6051,   2.7899],
        [ 11.4743,   6.2478,  -2.1100],
        [  6.3301,   0.3638,  -7.5685],
        [  6.3301,   0.3638,  -7.5685],
        [ -3.2486,   7.5925,  -1.3719],
        [  7.7750,  11.0665,   2.8364],
        [  5.1890,   4.7318,   0.9515],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.3587,   1.1596,  -3.6629],
        [ -1.0407,   7.9166,  -6.6021],
        [ -7.9667,   4.0814,  -0.6397],
        [  5.8323,  -1.5228,   5.1132],
        [  8.0877,  -8.5297,  -1.1050],
        [  2.6451,  -4.5250,  -0.5565],
        [  6.3413,   0.4883,  -6.8416],
        [  4.4049,   5.6892,  -4.4394],
        [  3.4140,  -4.1803,  -8.8224],
        [  9.4554,   9.4245,   9.8800],
        [  0.6142,  14.0802,   0.5407],
        [  4.8674,  12.8875,   6.7622],
        [ -2.1382,   7.8339,   7.3524],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.4150,  13.9647,  -4.0915],
        [  2.8602,   6.8905,   5.8779],
        [ 10.9931,   7.2107,   7.7933],
        [  2.3490,  -4.6199,  -4.2489],
        [  3.3167,   6.6022,   6.3554],
        [  9.0424,  -7.6779,  -7.5286],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -1.9714,   7.8563,  -1.8084],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -3.9700,   7.3453,  -4.1486],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -5.1398,  -9.1367,   6.2074],
        [  2.4356,  -4.5942,  -5.0808],
        [ 11.3691,   6.4760,   1.0787],
        [  5.3860,  -2.3418,   1.0826],
        [ -0.8885,   7.9160,   0.0571],
        [  0.6363, -11.0348,   6.1414],
        [ -9.3384,  -2.6161,  -4.7630],
        [  1.3339,   7.5627,  -0.9034],
        [  5.3118,   4.5452,   4.2114],
        [  8.7715,  -7.9390,   5.2900],
        [ -7.3470,  -7.0615,   5.0066],
        [ 12.5258,  -0.4200,  -9.3493]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0424, 0.0578,  ..., 0.3816, 0.3420, 1.4452],
        [0.0424, 0.0000, 0.0155,  ..., 0.4240, 0.3119, 1.4748],
        [0.0578, 0.0155, 0.0000,  ..., 0.4394, 0.3030, 1.4869],
        ...,
        [0.3816, 0.4240, 0.4394,  ..., 0.0000, 0.6740, 1.2113],
        [0.3420, 0.3119, 0.3030,  ..., 0.6740, 0.0000, 1.4928],
        [1.4452, 1.4748, 1.4869,  ..., 1.2113, 1.4928, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.87007495,  8.95983586,  8.98765644,  9.0368458 ,  9.18690018,
        9.23071698, 13.40024224,  9.19837792, 13.77604293, 10.41909211,
       11.28215429, 11.04313641,  5.40689332,  6.61211224,  9.36093562,
       13.37121923,  7.33350642,  6.89245352, 11.9120321 ,  5.39723241,
        7.81509631,  9.34127013, 12.35625694, 10.26545792, 10.4606265 ,
        5.69913817,  7.13501264, 11.8159254 , 11.22142195,  9.78285424,
        7.06372172,  5.64464807, 12.53285633, 13.1399181 , 11.18839169,
        5.08372149,  6.81117802, 11.48434767,  8.50268322, 14.02526505,
        8.99791316, 12.52875793,  8.49161154,  7.01475376,  7.39004226,
       11.83075005,  6.97106912,  5.29755487,  8.85939626,  5.53175275,
        5.1805345 ,  6.34059021, 10.67132524,  6.54693987, 11.83075005,
       13.95361098, 10.4977132 ,  5.5671389 , 11.33837396,  9.94285511,
        8.7273569 , 13.06498554,  6.34059021,  6.34059021,  8.25828409,
       13.5247215 ,  7.02253235,  5.53175275,  6.46356806,  7.98469166,
        8.95133961,  6.02779201, 11.75438006,  5.24137909,  6.36004393,
        7.19513806,  5.39723241, 13.35013374, 14.09357547, 13.77604293,
        8.12043983,  8.05242591, 14.03618649,  7.46053281, 13.14690137,
        5.18277858,  7.38844703, 11.86239961, 12.04924159,  8.09984231,
       12.35625694,  8.34949383,  8.29022197,  9.50913849, 10.48315436,
        5.19986797, 13.08414689,  5.87305005,  7.96575782, 11.05317597,
        9.69792513,  7.67939546,  6.99097013, 11.83075005, 10.19036081,
       12.53285633])}
{'P': tensor([[1.4945e-02, 2.0653e-01, 1.8418e-01,  ..., 2.0526e-04, 1.8728e-06,
         8.7463e-05],
        [1.7898e-04, 1.4341e-02, 1.0046e-01,  ..., 5.1070e-04, 3.2574e-06,
         1.9395e-04],
        [2.8268e-04, 1.7717e-03, 3.8253e-02,  ..., 1.3644e-03, 9.3427e-06,
         7.4732e-04],
        ...,
        [5.5599e-06, 8.0244e-06, 8.3714e-06,  ..., 5.3747e-02, 5.7453e-03,
         1.8837e-02],
        [6.2826e-07, 4.6700e-07, 3.5175e-07,  ..., 7.4669e-07, 1.0209e-02,
         1.9030e-06],
        [2.2438e-06, 2.2981e-06, 2.3814e-06,  ..., 5.2504e-05, 4.3175e-03,
         2.4485e-02]]), 'X': tensor([[-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0833, 0.1257,  ..., 0.5681, 0.7009, 0.0936],
        [0.0833, 0.0000, 0.0579,  ..., 0.5202, 0.7383, 0.0715],
        [0.1257, 0.0579, 0.0000,  ..., 0.5460, 0.7960, 0.0571],
        ...,
        [0.5681, 0.5202, 0.5460,  ..., 0.0000, 0.6662, 0.5895],
        [0.7009, 0.7383, 0.7960,  ..., 0.6662, 0.0000, 0.7900],
        [0.0936, 0.0715, 0.0571,  ..., 0.5895, 0.7900, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.8737556 ,  9.09119351,  9.18690018,  9.19837792,  9.18358562,
        9.23071698, 13.96746816,  6.19633139,  9.87720102, 13.3607379 ,
       12.44177641, 12.27835996,  5.05774295,  8.89301516,  7.38844703,
       11.62770332, 11.73397309,  7.85919687, 10.68031351,  9.53015376,
       11.64436684,  9.14973319,  6.65991877, 11.32149259, 11.38774149,
        7.16598273,  6.83515201, 13.02818702,  5.83560009, 10.68031351,
       10.26327895, 12.95702647, 13.88683436,  6.54615887,  6.81117802,
       12.17451685, 12.58006038,  8.06517143,  7.02326515,  5.94909709,
       13.58101794,  7.73151309, 12.22276278,  5.5205189 , 12.18998195,
        4.75303127, 11.70168755, 13.18278606, 12.22503754,  9.7712321 ,
        9.95552626, 11.42420508,  8.41148722, 13.26590071,  6.1463896 ,
        9.32094055,  9.51074763, 11.63765273,  5.64865173,  4.98989906,
       12.95702647, 12.67680439, 10.71162681, 13.77075174,  7.63726134,
        6.28961988, 12.59840193,  8.85939626, 12.97370371, 11.9182956 ,
       12.25806559,  6.83515201,  7.5158489 ,  5.91854397,  5.79562691,
        5.5366698 ,  5.53255882,  5.39723241,  5.77219077,  7.63726134,
        4.91038884,  9.77777346, 11.21703654, 11.54071859, 10.35464916,
       12.44177641,  9.05843201, 10.88618114,  7.58668144,  6.29034958,
       12.7125079 ,  6.55800267,  6.16512856, 11.63724792,  8.40445872,
        5.16611782, 11.85743425, 13.59480906,  9.7712321 ,  8.51088913,
        6.77777606, 12.26997165, 10.91601655, 10.48840476, 11.83570132,
       10.95593729])}
{'P': tensor([[1.0042e-02, 1.7249e-01, 1.5530e-01,  ..., 1.2113e-03, 8.4814e-05,
         1.7249e-01],
        [1.8191e-04, 1.7349e-02, 4.3221e-02,  ..., 1.6386e-03, 6.0293e-05,
         1.7349e-02],
        [2.6928e-04, 6.3784e-03, 2.7723e-02,  ..., 3.5656e-03, 9.1690e-05,
         6.3784e-03],
        ...,
        [1.6958e-04, 3.9666e-04, 5.0611e-04,  ..., 8.9852e-02, 4.5686e-05,
         3.9666e-04],
        [1.7220e-01, 8.9526e-02, 7.1005e-02,  ..., 1.1152e-03, 2.5131e-02,
         8.9526e-02],
        [1.8191e-04, 1.7349e-02, 4.3221e-02,  ..., 1.6386e-03, 6.0293e-05,
         1.7349e-02]]), 'X': tensor([[ -7.5720,   4.6310,   2.1717],
        [ -8.6229,   2.9068,   2.4429],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.8154,   2.4611,   2.5198],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.3900,   3.3727,   3.1650],
        [ -9.4716,  -0.6673,   7.6068],
        [  4.2275,  -3.6320,  -0.2714],
        [ -5.5525,   6.5146,  -4.4984],
        [  6.3360,   0.4258,  -9.2509],
        [  2.6614,  -4.5192,  -9.3641],
        [ -5.3382,   6.6533,   7.8054],
        [ -7.5603,   4.6459,   8.2002],
        [ 12.5267,  -0.4125,  -8.7331],
        [  5.5626,  -2.0536,   8.4433],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.0772,  -2.6014,  -7.1832],
        [  9.9217,  -6.7006,  -5.6728],
        [ -5.9225,  -8.5355,   2.4778],
        [ -8.3717,  -5.4769,  -9.2228],
        [  9.0652,   9.8597,  -3.5949],
        [ -6.7643,  -7.7372,  -9.6663],
        [  5.4336,  -2.2676,   5.6933],
        [  0.1041, -11.0045,   2.4311],
        [  6.3301,   0.3638,  -7.5685],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.4179,  -2.2925,  -3.4641],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.3087,   1.8053,  -1.9481],
        [  3.4140,  -4.1803,  -8.8224],
        [  6.0047,   3.1062,  -6.7914],
        [ -8.5149,  -5.1905,   1.5239],
        [  6.0047,   3.1062,  -6.7914],
        [ -3.5029,   7.5138,   8.1853],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.4867,  12.5816,   2.9549],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -3.5520,   7.4976,   9.1522],
        [ 12.2914,   3.8037,   6.0978],
        [ -9.3616,  -2.4749,   0.1793],
        [ -6.0195,   6.1789,  -2.9496],
        [ 12.2592,   3.9425,   2.0383],
        [  4.3523, -10.5232,   8.2138],
        [ 10.0619,   8.6627,  -9.2884],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.3587,   1.1596,  -3.6629],
        [ 10.5057,  -5.9099,  -5.1599],
        [  0.6952,  -4.8060,   8.8976],
        [ -0.3432,   7.8900,   7.1805],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.0087,   4.0173,   9.1091],
        [ 10.2712,   8.3707,  -4.9743],
        [  2.6970,   6.9826,   9.7350],
        [  8.5198,  10.4091,  -8.6486],
        [ -3.1671,   7.6158,   6.0765],
        [ -1.6275, -10.7221,   3.2376],
        [ 11.6544,   5.8273,  -3.1375],
        [ -1.2454,   7.9127,  -0.5715],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.4157,   0.0821,  -5.0025],
        [ -7.0948,   5.1940,   9.8171],
        [ -3.5029,   7.5138,   8.1853],
        [  9.6816,   9.1536,  -5.8543],
        [  1.0000,   7.6591,  -4.5011],
        [ -9.4748,  -0.7765,  -5.6695],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3205,   1.7068,  -3.8619],
        [  4.5800,   5.5034,  -3.9432],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.5603,   4.6459,   8.2002],
        [  6.0310,  12.2795,  -6.4528],
        [ -8.5149,  -5.1905,   1.5239],
        [ 10.7566,   7.6212,  -2.3279],
        [  3.1756, -10.8247,   6.8882],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.1737,   2.5305,   3.9049],
        [ -9.0604,   1.7707,   1.1066],
        [  5.1890,   4.7318,   0.9515],
        [  8.9284,  10.0036,  -6.9210],
        [  9.4516,  -7.2501,  -1.8868],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.2905,   1.9366,   2.1169],
        [ -3.7964,  -9.9294,   7.7147],
        [  8.7131,  10.2216,   7.7915],
        [ -2.0229, -10.6163,   8.6930],
        [ -7.0253,  -7.4493,  -1.5978],
        [  7.0664,  -9.2590,   8.0879],
        [  8.0381,  -8.5690,  -8.9523],
        [  0.2480, -11.0153,  -3.0183],
        [ -3.4986,   7.5152,   9.0712],
        [  6.2036,   2.4030,  -1.2500],
        [ -6.3066,  -8.1936,   9.5822],
        [  6.2022,   2.4092,   3.0349],
        [  2.5318,  13.7231,   6.3692],
        [ 11.2209,  -4.7092,   2.6117],
        [ -1.6086,   7.8928,   4.3385],
        [ 12.4901,  -0.7052,   1.7315],
        [  0.8572,  14.0501,  -8.3364],
        [  0.1252, -11.0062,   7.2564],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -5.1974,   6.7396,   3.3484],
        [ -8.6229,   2.9068,   2.4429]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0841, 0.1090,  ..., 0.6636, 0.2001, 0.0841],
        [0.0841, 0.0000, 0.0543,  ..., 0.6441, 0.2705, 0.0000],
        [0.1090, 0.0543, 0.0000,  ..., 0.5898, 0.3080, 0.0543],
        ...,
        [0.6636, 0.6441, 0.5898,  ..., 0.0000, 0.8374, 0.6441],
        [0.2001, 0.2705, 0.3080,  ..., 0.8374, 0.0000, 0.2705],
        [0.0841, 0.0000, 0.0543,  ..., 0.6441, 0.2705, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.87587457,  9.09963894,  9.12220271,  9.15253344,  9.15976746,
        9.04255655,  9.49511524,  5.57341623,  8.55982105,  6.35029274,
        5.24461068,  8.53014915,  8.8737556 , 12.53345445,  5.92952538,
        9.61942445, 12.35421461, 11.97238624, 10.38897563, 10.00410625,
       13.39372426, 10.27716434,  5.88782913, 11.0050379 ,  6.34059021,
       10.74409834,  5.8828986 , 10.63038565,  6.23262289,  6.5618968 ,
        5.39723241,  6.76056625,  9.97220799,  6.76056625,  8.29022197,
        8.25828409, 13.72594022, 10.37698435,  8.29640613, 12.86648771,
        9.68323686,  8.62631801, 12.87752052, 11.38774149, 13.27717982,
        9.97822473,  8.29203356,  6.46356806, 12.05395056,  4.85603886,
        7.89745107, 10.31613269,  8.43626114,  8.95983586, 13.25017211,
        7.48538905, 13.45124917,  8.24808832, 10.84493732, 13.03002338,
        8.01009083, 13.59673406,  9.41605684,  8.79284466,  8.29022197,
       13.32374056,  7.72415048,  9.50654967,  7.60529892,  6.54693987,
        7.15990676,  8.85939626,  8.8737556 , 13.68063098,  9.97220799,
       13.18278606, 11.28093507, 13.26590071,  6.67218687,  9.23178204,
        7.02253235, 13.40850328, 11.9120321 , 11.99633819,  6.58184474,
       10.63038565, 13.43127131, 10.80728788, 10.23946247, 11.64746882,
       11.74901778, 11.018084  ,  8.28968067,  6.652742  , 10.33958639,
        6.65368551, 13.95472331, 12.16901615,  8.0550286 , 12.50996828,
       14.07623266, 11.00694688,  6.69389431,  9.4997987 ,  8.51088913,
        9.09963894])}
{'P': tensor([[1.6101e-02, 1.8654e-01, 1.6402e-01,  ..., 1.7238e-06, 1.0635e-05,
         2.0786e-05],
        [3.1224e-04, 2.0451e-02, 3.6658e-01,  ..., 3.2236e-06, 1.3083e-05,
         1.3303e-05],
        [4.5291e-04, 1.4495e-03, 3.1540e-02,  ..., 5.1941e-06, 1.9936e-05,
         1.9126e-05],
        ...,
        [9.3723e-06, 5.7665e-06, 5.5242e-06,  ..., 5.4293e-02, 4.2391e-04,
         9.7981e-05],
        [1.5928e-04, 9.0342e-05, 8.4781e-05,  ..., 6.3866e-04, 1.3681e-01,
         4.8156e-03],
        [1.6441e-02, 6.9941e-03, 6.3067e-03,  ..., 4.3120e-06, 1.4382e-04,
         9.5280e-02]]), 'X': tensor([[ -7.6650,   4.5092,   0.2520],
        [ -8.6050,   2.9451,  -0.7012],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.0604,   1.7707,   1.1066],
        [ -8.7078,   2.7182,   1.8146],
        [  1.8447,  -4.7372,   8.1235],
        [  5.4336,  -2.2676,   5.6933],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.4573,   3.2448,   5.4370],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.4766,  -1.1116,  -2.1706],
        [  8.8091,  -7.9037,   0.3631],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.3493,  -4.7727,  -6.5682],
        [ -8.5900,   2.9767,   7.4811],
        [ -3.0644,   7.6439,  -1.4322],
        [ -7.0120,   5.2825,   7.7762],
        [  6.3505,   0.6235,  -8.5849],
        [  7.5784,  -8.9141,   5.8410],
        [  0.1677,  -4.7447,  -6.1529],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.5258,  -0.4200,  -9.3493],
        [  8.1528,  10.7452,   9.1493],
        [ 12.2592,   3.9425,   2.0383],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.3373,  13.7721,  -0.2442],
        [ -2.2508, -10.5479,  -2.7235],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.9788,   2.0208,  -3.2867],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.2707,   0.9661,   4.4863],
        [  1.9505,  -4.7171,   9.9868],
        [  3.0876,   6.7527,  -5.3540],
        [ -7.0948,   5.1940,   9.8171],
        [  6.1929,   2.4501,  -6.9352],
        [  3.1480,  -4.3165,  -7.8002],
        [  0.5960, -11.0335,   3.1144],
        [ -2.0840,   7.8415,  -7.8657],
        [  8.5093,  -8.1764,  -3.2381],
        [ -6.1899,  -8.3011,  -7.9269],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.9026,   3.3837,  -0.6573],
        [ -7.5603,   4.6459,   8.2002],
        [  1.6308,  13.9254,  -2.0984],
        [  9.6614,   9.1784,  -1.2087],
        [ -1.6334, -10.7207,   7.1397],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.5966,   5.9666,   6.0083],
        [  1.1274,  14.0116,   8.1753],
        [ -9.4748,  -0.7765,  -5.6695],
        [  2.5382,   7.0672,   4.5097],
        [  3.1271, -10.8343,   2.6442],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.9282,   2.1648,   8.4894],
        [  7.0642,  11.6104,  -9.8316],
        [ -8.6681,  -4.8556,   9.2518],
        [ -0.5709,   7.9055,   7.6089],
        [ -1.6906,   7.8860,   7.7142],
        [  0.2480, -11.0153,  -3.0183],
        [ -4.0633,  -9.7927,  -3.0825],
        [  2.7685, -10.8982,  -6.8928],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.8093,  -9.4178,   8.4244],
        [  9.6614,   9.1784,  -1.2087],
        [ -9.4309,  -0.0599,  -4.0830],
        [  5.0348,  -2.8229,   3.5436],
        [  7.9403,  -8.6455,  -8.7433],
        [  6.3549,   0.7096,   6.9198],
        [  1.7000,  -4.7609,   2.7935],
        [  2.3928,   7.1404,  -1.7035],
        [  6.0890,  -0.8382,   2.9166],
        [ -1.0407,   7.9166,  -6.6021],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -4.4779,  -9.5609,  -2.4505],
        [  5.9587,  12.3216,   9.1695],
        [ -9.1813,  -3.3550,   1.8827],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.7735,   2.5640,  -6.3313],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.0348,  -2.8229,   3.5436],
        [  6.9284,  11.7061,  -0.3732],
        [  2.2860,   7.1916,  -6.1011],
        [  4.3554,   5.7393,  -9.0766],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.9433,  -9.3362,  -0.3421],
        [  9.7328,   9.0902,  -0.8917],
        [  1.5662,  13.9375,   4.5142],
        [  7.5784,  -8.9141,   5.8410],
        [  6.3516,   0.6425,   0.3425],
        [  6.2365,  -0.2578,  -6.0212],
        [ -2.4956, -10.4682,   3.5632],
        [  4.5430,   5.5438,  -3.8669],
        [  6.3119,   1.7797,   9.9186],
        [  6.0310,  12.2795,  -6.4528],
        [ -8.6681,  -4.8556,   9.2518],
        [  1.2273,  -4.8081,   9.8129],
        [  2.3261,   7.1726,   6.2522],
        [  9.2692,   9.6370,   7.0746],
        [ 11.0891,  -4.9558,  -2.3178],
        [  5.6672,  -1.8631,   8.1020],
        [ -0.5385,   7.9037,   6.3830]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1237, 0.1430,  ..., 0.6406, 0.8590, 0.6398],
        [0.1237, 0.0000, 0.0195,  ..., 0.7083, 0.9823, 0.7627],
        [0.1430, 0.0195, 0.0000,  ..., 0.7227, 1.0017, 0.7815],
        ...,
        [0.6406, 0.7083, 0.7227,  ..., 0.0000, 0.7305, 0.7229],
        [0.8590, 0.9823, 1.0017,  ..., 0.7305, 0.0000, 0.2737],
        [0.6398, 0.7627, 0.7815,  ..., 0.7229, 0.2737, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.89301516,  9.09502385,  9.11473391,  9.15976746,  9.23178204,
        9.12220271,  5.08372149,  5.88782913,  9.88571713,  9.05843201,
       10.19036081,  9.54154096, 11.83509372,  5.27178363,  4.78544875,
        9.09119351,  8.23525433,  8.77913428,  6.38105015, 11.70016217,
        4.74770899,  4.78544875, 12.53285633, 13.48809595, 12.87752052,
        9.5014146 , 12.59840193, 13.96905756, 10.78533838, 12.42554808,
        9.20340031,  5.39450305,  9.32094055,  5.10446653,  7.42512621,
        8.79284466,  6.65991877,  5.34250225, 11.04953767,  8.1137452 ,
       11.80092578, 10.35485331,  5.84450127,  6.80370254,  8.8737556 ,
       14.02058621, 13.32613445, 10.84437705,  5.16611782, 12.7125079 ,
       11.63502679, 13.0743399 , 13.04156075, 14.05687815,  9.50654967,
        7.50917551, 11.27656488, 10.00794468,  9.18690018, 13.59056613,
        9.9353921 ,  7.92607342,  8.06517143, 11.018084  , 10.60226614,
       11.24434374,  9.5014146 , 11.62158554, 13.32613445,  9.43112957,
        5.77219077, 11.73848065,  6.39438044,  5.05534609,  7.53063885,
        6.1463896 ,  7.98469166, 10.27716434, 10.5575601 , 13.68672855,
        9.7751189 ,  5.24137909,  9.14044984,  9.75200848,  5.77219077,
       13.60275211,  7.54620849,  7.20482224,  9.77588801, 11.63502679,
       13.31764181, 14.02526505, 11.70016217,  6.38399789,  6.24187044,
       10.76154256,  7.16747878,  6.55800267, 13.68063098,  9.9353921 ,
        4.96231532,  7.5403822 , 13.37121923, 12.1460931 ,  5.96556248,
        7.92200342])}
{'P': tensor([[1.5497e-02, 2.3468e-01, 2.0435e-01,  ..., 1.0326e-04, 6.9861e-05,
         3.0520e-06],
        [3.1313e-04, 3.5720e-02, 4.2133e-04,  ..., 2.3907e-04, 4.2648e-04,
         7.4847e-06],
        [5.6263e-04, 3.0025e-01, 1.5364e-02,  ..., 8.4561e-05, 8.7108e-05,
         2.7047e-06],
        ...,
        [6.2984e-05, 1.2219e-04, 6.5928e-05,  ..., 3.1607e-01, 3.5517e-03,
         8.0481e-04],
        [1.1527e-05, 1.1613e-05, 1.1109e-05,  ..., 9.2948e-06, 1.3836e-01,
         1.3182e-03],
        [7.2441e-05, 4.1849e-05, 5.3825e-05,  ..., 8.3966e-05, 2.9026e-05,
         1.2413e-01]]), 'X': tensor([[ -7.6734,   4.4980,  -5.4248],
        [ -8.8363,   2.4086,  -5.6296],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.7735,   2.5640,  -6.3313],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.3076,   0.7852,  -6.7010],
        [  7.9141,  -8.6656,   9.6948],
        [  6.1737,   2.5305,   3.9049],
        [ 12.3357,   3.6001,   0.7903],
        [ -9.4630,  -1.5145,   0.2983],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.1537,   5.9340,  -6.0423],
        [  4.1348,   5.9514,  -0.8856],
        [  9.5261,  -7.1675,   1.4206],
        [  6.3031,   1.8477,  -7.1706],
        [ -9.1749,  -3.3803,   6.4629],
        [ 11.5568,   6.0602,   9.3244],
        [  7.9141,  -8.6656,   9.6948],
        [ 12.5927,   0.4199,  -9.4997],
        [ 10.4994,  -5.9193,  -7.5153],
        [  6.3535,   0.6792,  -7.5101],
        [  7.9816,  10.8938,  -0.5378],
        [  2.5318,  13.7231,   6.3692],
        [  1.8320,   7.3871,   5.1452],
        [ 11.2571,   6.7072,   8.3693],
        [  1.1644, -11.0394,   7.8511],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.6319,  11.9066,   3.5165],
        [ 11.2517,  -4.6495,   2.9800],
        [ -8.1426,   3.8048,   9.3542],
        [  6.2419,  -0.2313,   4.2440],
        [  2.0449,  13.8402,  -7.1642],
        [  6.3236,   0.3011,  -4.6668],
        [ 12.4690,  -0.8528,  -9.3374],
        [  9.2692,   9.6370,   7.0746],
        [ -1.6906,   7.8860,   7.7142],
        [ -5.1046,  -9.1610,   2.3418],
        [  7.0171,  11.6439,  -5.0178],
        [ 11.7779,   5.5127,  -2.0699],
        [ -9.4645,  -1.4880,  -2.0226],
        [  0.8940, -11.0402,  -0.9783],
        [  9.1409,  -7.5789,   1.5814],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.9515,  -3.8425,   6.1729],
        [  3.4604,  -4.1545,   1.8091],
        [  5.6672,  -1.8631,   8.1020],
        [  4.5491,  -3.3485,   6.8412],
        [  4.6063,  13.0051,  -3.1456],
        [  6.1973,  -0.4372,   6.5720],
        [  9.6816,   9.1536,  -5.8543],
        [ -8.7544,   2.6096,   3.3146],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -7.7842,   4.3466,  -1.7548],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -7.9925,  -6.1423,   2.8990],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.8631,  -4.3728,  -1.1746],
        [  6.8093,  -9.4178,   8.4244],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.1469,   6.9237,   1.6160],
        [  6.3606,   0.9204,   3.2525],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.2188,  -2.5839,   2.1833],
        [  6.0955,   2.8210,  -4.3321],
        [  4.2096,   5.8815,   1.9230],
        [ 12.5996,   1.3971,  -1.0694],
        [  2.5010,  -4.5736,  -0.8989],
        [  0.3715,  -4.7756,   0.2312],
        [ 12.4996,   2.6335,  -6.3166],
        [  8.3718,  -8.2953,   8.4666],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.2975,   0.0947,   5.1791],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.3602,  -2.3808,   7.2098],
        [ -8.6801,   2.7809,  -0.8443],
        [  3.1480,  -4.3165,  -7.8002],
        [ -0.0410,   7.8591,   9.3457],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.6165,  -3.8715,   6.4594],
        [ 11.9641,  -2.9568,   7.7211],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.0310,  12.2795,  -6.4528],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.9085,   6.1512,  -2.2560],
        [  7.5784,  -8.9141,   5.8410],
        [ -8.4887,  -5.2448,  -0.6585],
        [  9.0004,  -7.7195,   4.6945],
        [ -6.4822,  -8.0252,  -3.8983],
        [  4.0587, -10.6112,   6.2273],
        [ -8.9788,   2.0208,  -3.2867],
        [ 12.3357,   3.6001,   0.7903],
        [  6.3589,   0.8239,  -8.4230],
        [ -3.0008, -10.2827,  -8.9480],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.5966,   5.9666,   6.0083],
        [  6.8704,  -9.3809,   7.5769],
        [  3.9770, -10.6341,   8.0783],
        [ -9.1795,  -3.3623,   8.5259],
        [  0.2480, -11.0153,  -3.0183],
        [  6.1973,  -0.4372,   6.5720]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2054e-01, 4.8001e-02,  ..., 9.7231e-01, 9.7424e-02,
         1.1025e+00],
        [1.2054e-01, 3.0339e-04, 7.9498e-02,  ..., 9.7307e-01, 1.9000e-01,
         1.1982e+00],
        [4.8001e-02, 7.9498e-02, 0.0000e+00,  ..., 9.9090e-01, 1.3970e-01,
         1.1490e+00],
        ...,
        [9.7231e-01, 9.7307e-01, 9.9090e-01,  ..., 0.0000e+00, 8.9335e-01,
         9.4721e-01],
        [9.7424e-02, 1.9000e-01, 1.3970e-01,  ..., 8.9335e-01, 0.0000e+00,
         1.0113e+00],
        [1.1025e+00, 1.1982e+00, 1.1490e+00,  ..., 9.4721e-01, 1.0113e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.89458273,  9.15866607,  8.97052115,  9.14044984,  9.26455795,
        9.34061938, 11.73567759,  6.67218687, 12.85032836,  9.58347456,
       13.01728313,  7.24326992,  7.24677664, 11.92133129,  6.56833864,
        9.77777346, 13.0493368 , 11.73567759, 12.59969936, 12.05300943,
        6.38968357, 13.50484539, 13.95472331,  7.61088125, 13.103709  ,
       11.10066126,  5.53255882,  9.75200848, 13.62896593, 12.17451685,
        8.98765644,  6.24614905, 13.99048094,  6.33075682, 12.4980845 ,
       13.37121923,  8.06517143, 10.48721382, 13.59480906, 13.00413829,
        9.58072286, 11.07637531, 11.87412439, 12.52372629,  5.51177501,
        5.40689332,  5.96556248,  5.64865173, 13.79676654,  6.21275016,
       13.32374056,  9.13507321, 12.10222204,  9.562174  ,  8.91551231,
       12.30858677, 10.35485331, 10.08001894,  5.05963217, 12.22276278,
        5.54776934,  9.88310614, 11.62158554,  4.85995002, 13.12218013,
        6.42689705,  8.16714836,  5.82345867,  6.7166297 ,  7.23276582,
       12.67680439,  5.21280156,  4.79003361, 12.77402351, 11.785566  ,
        5.39450305,  6.29821673, 10.7167285 ,  9.97220799,  5.86519525,
        9.11473391,  5.34250225,  7.85919687,  8.92098752, 12.24467041,
       12.32408516, 10.33712258, 13.68063098, 13.25017211,  7.28793663,
       11.70016217,  9.97822473, 11.85743425, 10.31613269, 11.36089355,
        9.20340031, 12.85032836,  6.41202986, 10.71162681,  6.00732193,
       13.04156075, 11.62770332, 11.35345345,  9.77588801, 11.018084  ,
        6.21275016])}
{'P': tensor([[8.4906e-03, 2.1131e-01, 1.4623e-01,  ..., 3.4819e-06, 2.4681e-07,
         1.6135e-02],
        [1.3554e-04, 9.6573e-03, 2.8284e-01,  ..., 5.7680e-06, 3.2294e-07,
         2.3337e-02],
        [1.5372e-04, 2.5697e-04, 1.5501e-02,  ..., 1.2324e-05, 5.7094e-07,
         3.4864e-02],
        ...,
        [4.9125e-06, 4.0232e-06, 3.6378e-06,  ..., 6.6759e-02, 5.3744e-04,
         2.1013e-06],
        [8.7311e-07, 5.9134e-07, 4.5285e-07,  ..., 6.5717e-07, 9.0377e-03,
         3.4114e-07],
        [1.0732e-04, 1.9383e-04, 3.2571e-04,  ..., 5.2456e-05, 4.0634e-06,
         3.3936e-02]]), 'X': tensor([[-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.8634e-02, 9.8188e-02,  ..., 4.7342e-01, 1.0486e+00,
         3.1863e-01],
        [5.8634e-02, 0.0000e+00, 4.0571e-02,  ..., 5.0899e-01, 1.0889e+00,
         2.7845e-01],
        [9.8188e-02, 4.0571e-02, 2.1453e-04,  ..., 5.2637e-01, 1.1246e+00,
         2.6573e-01],
        ...,
        [4.7342e-01, 5.0899e-01, 5.2637e-01,  ..., 0.0000e+00, 1.2451e+00,
         7.8742e-01],
        [1.0486e+00, 1.0889e+00, 1.1246e+00,  ..., 1.2451e+00, 0.0000e+00,
         1.0931e+00],
        [3.1863e-01, 2.7845e-01, 2.6573e-01,  ..., 7.8742e-01, 1.0931e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.90405823,  9.04255655,  9.13507321,  8.99276564,  9.09963894,
        9.17016857, 12.88270018,  8.91551231,  8.49161154, 13.48809595,
        8.29640613,  5.70405525, 10.54658597,  7.72415048,  9.26455795,
       13.27717982, 10.794755  , 14.10045797,  9.48969978, 13.0743399 ,
        7.48311002, 10.91805683, 12.24467041,  6.99097013,  9.88310614,
        5.76165934,  5.8828986 ,  5.1805345 ,  9.48969978,  9.17016857,
        8.49161154, 13.34587027, 11.57249984, 10.06407839, 10.4606265 ,
        5.82345867, 12.50996828,  9.49511524, 10.35093485, 13.84126197,
       10.37698435, 13.03002338,  6.14086582,  4.76969364, 11.64746882,
       12.95702647,  7.62509582,  8.30729352, 12.58006038,  8.7907923 ,
       12.35625694,  5.2287058 ,  5.29755487, 12.42554808, 13.0493368 ,
        4.8763135 ,  4.96231532,  7.34250948,  6.34059021,  7.28793663,
        7.69752262, 12.79126785, 13.00413829, 11.22142195, 10.71162681,
       10.00410625, 13.37121923,  6.84796202,  6.97106912,  9.3349858 ,
       11.4088043 , 12.59969936, 11.87665501,  4.90676242,  7.71075226,
        7.39004226,  7.02253235, 10.82106505, 10.98076873,  6.53250937,
        6.81084724,  9.5642993 , 11.97949273, 13.61600536, 13.03002338,
        9.85384623, 10.06850677,  9.61942445, 11.57249984, 11.28093507,
       12.49317563, 11.4890861 ,  9.99211663, 11.35182589, 12.19805394,
        5.13616224,  5.23878371,  9.75200848, 13.35927527, 10.51437947,
       12.84056604, 12.1494023 ,  6.3949043 , 11.62770332, 12.35421461,
        9.48337063])}
{'P': tensor([[1.3368e-02, 2.3274e-01, 1.6254e-01,  ..., 1.6676e-06, 1.9842e-04,
         2.9229e-06],
        [1.6948e-04, 1.7616e-02, 3.2562e-03,  ..., 3.8184e-06, 6.7090e-04,
         4.1236e-06],
        [3.1251e-04, 5.0438e-02, 2.0803e-02,  ..., 4.9078e-06, 7.4352e-04,
         5.8797e-06],
        ...,
        [1.8376e-06, 1.2885e-06, 1.5549e-06,  ..., 4.4617e-02, 5.5978e-06,
         1.3341e-02],
        [2.1049e-06, 2.4853e-06, 2.4787e-06,  ..., 2.4099e-03, 1.9328e-02,
         1.0527e-04],
        [1.7557e-04, 1.0898e-04, 1.4694e-04,  ..., 3.3354e-04, 3.9084e-05,
         2.0310e-01]]), 'X': tensor([[ -7.7842,   4.3466,  -1.7548],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.3069,   6.6089,   8.7377],
        [  5.7151,  -1.7700,   3.1542],
        [ -3.2486,   7.5925,  -1.3719],
        [  3.2701,   6.6338,   9.9209],
        [ -8.7544,   2.6096,   3.3146],
        [ -2.2508, -10.5479,  -2.7235],
        [ -4.1296,   7.2803,  -2.7194],
        [ -5.1046,  -9.1610,   2.3418],
        [  2.0870,  -4.6877,   9.6823],
        [  7.8425,  -8.7202,   6.4530],
        [  5.8450,  -9.9356,   3.0767],
        [  2.0899,  -4.6870,  -7.4260],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.4703,  -4.2006,   7.4086],
        [ -7.5603,   4.6459,   8.2002],
        [ -4.1296,   7.2803,  -2.7194],
        [  5.3858,   4.4260,  -9.6769],
        [  2.6078, -10.9229,   6.2869],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.6650,   4.5092,   0.2520],
        [ -0.5709,   7.9055,   7.6089],
        [ 12.2435,   4.0076,   8.1257],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.4140,  -4.1803,  -8.8224],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.4430,  -0.1897,   4.8383],
        [ -2.2508, -10.5479,  -2.7235],
        [  2.3261,   7.1726,   6.2522],
        [  6.9956,  11.6590,  -7.6876],
        [  0.6757, -11.0361,  -1.3047],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.8617,   3.4853,  -9.8080],
        [  2.1120,  -4.6818,   8.6144],
        [ 10.4994,  -5.9193,  -7.5153],
        [  2.7558,  -4.4844,   0.0636],
        [  6.2902,   0.0451,   3.7116],
        [  5.4429,  -2.2529,   9.7306],
        [  5.7845,   3.6652,   0.9175],
        [  4.5325,  -3.3643,   6.7699],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.4351,  -1.8891,   3.5642],
        [ 12.3453,  -1.5462,   7.0212],
        [  4.9452, -10.3187,  -7.8621],
        [  6.1752,   2.5242,   3.5874],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.0890,  -0.8382,   2.9166],
        [  1.1994,   7.6035,   2.4295],
        [ -9.4157,   0.0821,  -5.0025],
        [  9.2979,   9.6048,   7.3138],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.1741,  -0.5329,   3.1913],
        [  6.3569,   0.7591,   6.5011],
        [  9.7328,   9.0902,  -0.8917],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.4901,  -0.7052,   1.7315],
        [ -3.2736, -10.1703,   3.8712],
        [ -0.9715, -10.8630,   0.3555],
        [  6.3606,   0.9204,   3.2525],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.0955,   2.8210,  -4.3321],
        [  8.3791,  10.5409,   5.5590],
        [  1.9505,  -4.7171,   9.9868],
        [  5.5240,  -2.1198,   6.2055],
        [  6.1741,  -0.5329,   3.1913],
        [ -8.3653,   3.4184,   9.3454],
        [  3.1756, -10.8247,   6.8882],
        [  2.4697,  -4.5836,   1.4593],
        [ -5.9987,   6.1949,  -6.2114],
        [ -3.4100,   7.5436,  -7.6980],
        [  6.1984,  -0.4327,   1.5438],
        [  8.5198,  10.4091,  -8.6486],
        [  7.0642,  11.6104,  -9.8316],
        [ 11.0438,   7.1180,  -5.6991],
        [ -6.8618,  -7.6322,   3.7027],
        [ -2.0840,   7.8415,  -7.8657],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -6.4844,  -8.0230,   8.5209],
        [  0.3715,  -4.7756,   0.2312],
        [  6.1575,  12.2046,   3.3257],
        [  2.0729,   7.2878,   7.0880],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.1861,   4.7361,  -3.7312],
        [  1.6954,  -4.7616,   9.8198],
        [ -4.8665,  -9.3204,  -5.2564],
        [  3.1892, -10.8220,  -6.8409],
        [  2.8994,   6.8675,  -9.5987],
        [ -9.4524,  -1.6813,  -6.6862],
        [ 12.5789,   0.1722,   1.6470],
        [ -3.0008, -10.2827,  -8.9480],
        [  9.6662,   9.1725,   8.1056],
        [ -6.6998,   5.5954,   0.0823],
        [ -8.6381,  -4.9238,  -4.4077],
        [  8.2154,  -8.4262,   8.0171],
        [ -2.6485, -10.4151,   0.8766],
        [  6.3266,   0.3292,   9.3502]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1144, 0.1105,  ..., 0.8938, 0.2615, 0.9936],
        [0.1144, 0.0000, 0.0654,  ..., 0.9569, 0.2240, 1.0836],
        [0.1105, 0.0654, 0.0000,  ..., 0.8943, 0.1676, 1.0297],
        ...,
        [0.8938, 0.9569, 0.8943,  ..., 0.0000, 0.8051, 0.3333],
        [0.2615, 0.2240, 0.1676,  ..., 0.8051, 0.0000, 0.9892],
        [0.9936, 1.0836, 1.0297,  ..., 0.3333, 0.9892, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.91551231,  9.14973319,  9.11473391,  9.09502385,  9.36378438,
        9.36093562,  7.39004226,  5.98285319,  8.25828409,  7.39599986,
        9.13507321, 10.78533838,  8.36995953, 10.48721382,  5.13125513,
       11.72802491, 11.52733581,  5.13182506,  8.94395628, 13.61801526,
       12.21532282,  8.8737556 ,  8.36995953,  6.97106912, 11.22993218,
       11.9182956 ,  8.89301516,  7.92607342, 12.88270018, 12.91143491,
        5.39723241, 10.78533838,  9.55593374,  9.44486609, 10.78533838,
        7.5403822 , 13.59673406, 11.05672136,  5.9510489 ,  6.81961996,
        5.13616224, 12.05300943,  5.26342203,  6.29034958,  5.89074104,
        6.84796202,  5.64464807,  6.83515201,  9.70108409,  9.43112957,
        9.62238828, 12.44177641, 11.4424816 ,  6.67121748, 12.05395056,
        6.1463896 ,  7.69752262,  9.41605684, 13.36800073,  5.34250225,
        6.19707944,  6.4020404 , 13.31764181,  9.50654967,  6.25586297,
       12.50996828, 10.68416401, 10.90638066,  6.42689705,  4.978704  ,
        6.7166297 , 13.46553623,  5.10446653,  5.91677188,  6.19707944,
        9.0368458 , 11.28093507,  5.20659892,  8.62329466,  8.27853076,
        6.2134961 , 13.45124917, 13.59056613, 13.13888841, 10.26327895,
        8.1137452 , 10.52587243, 10.31584102,  4.79003361, 13.66991027,
        7.57685908,  9.20340031,  7.02326515,  5.05444288, 10.51437947,
       11.28215429,  7.4544866 ,  9.60080639, 12.58006038, 10.71162681,
       13.32557133,  8.72896381,  9.94285511, 11.76830981, 10.74655346,
        6.33517349])}
{'P': tensor([[1.2322e-02, 2.8400e-01, 1.7634e-01,  ..., 4.0053e-05, 8.6471e-05,
         3.5000e-03],
        [3.1365e-04, 2.1021e-02, 2.2280e-01,  ..., 3.9024e-05, 1.9663e-04,
         1.0130e-02],
        [7.0239e-04, 2.5813e-03, 4.8074e-02,  ..., 6.4553e-05, 5.3320e-04,
         2.5608e-02],
        ...,
        [1.6296e-03, 1.7552e-03, 1.4621e-03,  ..., 9.8785e-02, 2.1713e-05,
         1.5465e-03],
        [5.2123e-05, 4.9579e-05, 5.0964e-05,  ..., 4.7968e-06, 4.5175e-02,
         5.3242e-05],
        [1.6168e-05, 2.1391e-05, 2.3877e-05,  ..., 5.3055e-06, 1.3167e-03,
         3.7193e-02]]), 'X': tensor([[ -7.8126,   4.3066,   5.1791],
        [ -8.4573,   3.2448,   5.4370],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.2707,   0.9661,   4.4863],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.0829,  -5.9940,   5.0564],
        [  0.3493,  -4.7727,  -6.5682],
        [ -6.4620,  -8.0449,   6.2719],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.4824,  -5.2575,  -0.0795],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.9531,   5.0576,   5.6794],
        [ 10.9875,   7.2207,  -8.8549],
        [ -3.4488, -10.0934,   2.5195],
        [ -4.0633,  -9.7927,  -3.0825],
        [  9.5203,   9.3483,   5.8781],
        [ -9.4505,  -1.7081,  -0.7039],
        [  2.9470,  13.6085,   2.9809],
        [  4.5491,  -3.3485,   6.8412],
        [  6.2902,   0.0451,   3.7116],
        [ -3.4986,   7.5152,   9.0712],
        [ -8.8363,   2.4086,  -5.6296],
        [  1.7228,  -4.7575,   8.3396],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.1504,  -4.6726,   8.7908],
        [  0.6142,  14.0802,   0.5407],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.3419,   0.5972,  -1.2841],
        [  7.9403,  -8.6455,  -8.7433],
        [  4.8674,  12.8875,   6.7622],
        [  0.4119,  14.1019,  -4.1870],
        [ -6.4844,  -8.0230,   8.5209],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.6672,  -1.8631,   8.1020],
        [  1.9505,  -4.7171,   9.9868],
        [  8.6420,  -8.0580,   6.5781],
        [ 11.9808,   4.9340,  -3.3715],
        [  9.9756,  -6.6330,  -3.3429],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.9838,  -7.4965,   0.9796],
        [  1.0000,   7.6591,  -4.5011],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 12.4877,  -0.7222,   8.8347],
        [ -8.0534,  -6.0429,   2.4583],
        [ 10.9875,   7.2207,  -8.8549],
        [ 12.3453,  -1.5462,   7.0212],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.2849, -10.9655,  -7.6542],
        [  5.5158,  -2.1337,   3.5246],
        [  9.9750,  -6.6338,  -1.2440],
        [  2.7295,  13.6703,  -8.9328],
        [  4.3459,  -3.5327,   0.3667],
        [  2.2644,  -4.6434,   9.7508],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.9788,  -4.8145,  -3.8661],
        [  2.0899,  -4.6870,  -7.4260],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.6357,   5.4415,   7.9067],
        [  8.1217,  -8.5024,  -0.6149],
        [  0.9036,   7.6840,   9.9237],
        [ -8.6801,   2.7809,  -0.8443],
        [  8.2422,  -8.4041,  -1.4962],
        [  0.9036,   7.6840,   9.9237],
        [  7.8375,  -8.7240,  -9.8913],
        [ -2.0445,   7.8469,  -0.5857],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.2630,   2.1057,  -5.2136],
        [  5.2608,  12.6977,   4.9166],
        [  6.0467,  -9.8369,  -6.5816],
        [ -9.4721,  -1.3097,  -7.4377],
        [  8.3791,  10.5409,   5.5590],
        [  1.9505,  -4.7171,   9.9868],
        [  6.3236,   0.3011,  -4.6668],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.2517,  -4.6495,   2.9800],
        [  6.1372,   2.6726,  -6.7885],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.4501,  -1.7133,   2.2980],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.7565,  11.8237,  -4.9882],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.9531,   5.0576,   5.6794],
        [ -4.6393,   7.0461,   1.1624],
        [  4.5503,   5.5359,   7.4735],
        [  5.1793,   4.7461,   0.3899],
        [  5.6260,  -1.9401,   3.0259],
        [  4.5430,   5.5438,  -3.8669],
        [  7.2340,  -9.1505,  -5.8321],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.1706,   1.3863,   8.4032],
        [ -5.1398,  -9.1367,   6.2074],
        [ -3.7964,  -9.9294,   7.7147],
        [ 12.4996,   2.6335,  -6.3166],
        [ -3.0008, -10.2827,  -8.9480],
        [  4.0362,  13.2397,  -7.0030],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 10.0567,   8.6697,   4.6329],
        [ -5.9987,   6.1949,  -6.2114],
        [  1.6954,  -4.7616,   9.8198],
        [ -7.9925,  -6.1423,   2.8990]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 6.5122e-02, 9.6757e-02,  ..., 7.8620e-01, 6.4141e-01,
         3.2185e-01],
        [6.5122e-02, 0.0000e+00, 3.7662e-02,  ..., 8.0852e-01, 7.0079e-01,
         2.7522e-01],
        [9.6757e-02, 3.7662e-02, 2.1453e-04,  ..., 8.4232e-01, 7.1802e-01,
         2.7765e-01],
        ...,
        [7.8620e-01, 8.0852e-01, 8.4232e-01,  ..., 0.0000e+00, 1.0783e+00,
         7.2824e-01],
        [6.4141e-01, 7.0079e-01, 7.1802e-01,  ..., 1.0783e+00, 0.0000e+00,
         9.5629e-01],
        [3.2185e-01, 2.7522e-01, 2.7765e-01,  ..., 7.2824e-01, 9.5629e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.92098752,  9.05843201,  9.10945678,  9.32094055,  9.17595899,
        9.17016857, 10.06286074,  4.78544875, 10.31886066, 12.49291768,
        8.86082124,  9.97964013,  4.91296205,  7.07902036, 13.14777433,
       10.66631784, 10.60226614, 13.3426541 ,  9.60359158, 13.92390689,
        5.64865173,  6.29034958,  8.28968067,  9.15866607,  5.05981914,
       13.00413829,  5.14371005, 14.09357547, 11.59512194,  6.33075682,
        9.36093562, 11.73848065, 13.77604293, 14.10796466, 10.31584102,
       12.50996828,  5.96556248,  5.10446653, 11.8159254 , 12.95702647,
       11.97957425, 11.28895489, 10.24557047,  7.72415048, 10.30972193,
       12.5085995 , 10.06850677, 13.14777433, 12.44177641, 10.36844813,
        9.96391045, 11.20100795,  5.91408504, 11.97949273, 13.94008948,
        5.60064319,  5.16611782, 10.30972193,  4.91296205,  5.13182506,
       12.70489563,  7.14837514, 11.75807252,  7.73692436,  9.11473391,
       11.77124373,  7.73692436, 11.72749923,  8.10886021, 12.27993098,
        6.60752373, 13.744387  , 11.54672755,  9.562174  , 13.46553623,
        5.10446653,  6.33075682,  8.41148722, 12.17451685,  6.69389431,
        9.10945678,  9.60412735,  5.19065439, 13.61801526,  9.38232121,
        7.07902036,  8.43626114,  7.16598273,  7.02496223,  5.95111804,
        7.16747878, 11.66453059, 11.66453059,  9.27474386, 10.48315436,
       10.63038565, 12.77402351, 10.71162681, 13.84126197, 13.6660671 ,
        9.91668182, 12.42554808, 13.2778378 ,  8.62329466,  5.05444288,
       10.08001894])}
{'P': tensor([[1.3160e-02, 1.7781e-01, 1.6580e-01,  ..., 5.2763e-04, 5.0099e-02,
         2.7111e-07],
        [3.7699e-04, 2.2310e-02, 2.6608e-02,  ..., 2.0094e-04, 1.0970e-01,
         4.9445e-07],
        [1.9847e-04, 3.0213e-03, 1.5202e-02,  ..., 1.5899e-04, 1.6824e-01,
         5.5009e-07],
        ...,
        [1.6312e-01, 2.4697e-01, 1.0656e-01,  ..., 1.6315e-02, 3.6110e-02,
         2.5024e-07],
        [1.0264e-03, 2.1071e-03, 1.9402e-03,  ..., 8.0677e-04, 3.6389e-01,
         3.1497e-05],
        [3.6364e-07, 2.1321e-07, 2.8980e-07,  ..., 3.6986e-07, 1.8230e-07,
         1.7238e-02]]), 'X': tensor([[-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0921, 0.0919,  ..., 0.0532, 0.2176, 1.2475],
        [0.0921, 0.0000, 0.0522,  ..., 0.1365, 0.1392, 1.3146],
        [0.0919, 0.0522, 0.0000,  ..., 0.1447, 0.1274, 1.2703],
        ...,
        [0.0532, 0.1365, 0.1447,  ..., 0.0000, 0.2689, 1.2452],
        [0.2176, 0.1392, 0.1274,  ..., 0.2689, 0.0000, 1.3403],
        [1.2475, 1.3146, 1.2703,  ..., 1.2452, 1.3403, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.92395381,  9.09661212,  9.14044984,  9.34127013,  9.34061938,
        9.15866607,  8.12580976,  7.42512621,  8.05668609, 10.36844813,
        8.95133961,  9.14044984, 10.08001894,  6.54127431, 12.32408516,
       13.35927527, 11.04313641, 12.53345445, 13.3614598 , 10.31584102,
       12.07818871, 11.69264243,  9.58347456,  6.81117802, 14.02526505,
        5.01114719,  8.41171369,  6.74872981, 11.05413545,  7.32754882,
        8.5724042 , 12.0495521 , 12.42554808, 12.22503754,  5.19065439,
       13.06498554,  4.92497514,  9.60359158, 12.58006038, 12.27835996,
        9.27474386,  4.89121805, 12.27835996, 13.94008948,  5.94541803,
        7.58378655, 14.07623266, 10.38897563,  8.91551231,  6.00732193,
        6.55800267,  9.48337063,  5.14607014, 13.0493368 ,  7.27340834,
        9.63742669,  7.15990676, 13.95361098,  9.79814997,  6.74872981,
       13.59056613,  6.02588517,  6.58184474,  9.60412735, 12.76408562,
       10.51340829, 10.98076873, 10.00794468, 11.00726483,  7.83534987,
       13.59673406,  5.14371005,  6.652742  ,  5.63824971,  9.27474386,
        8.31625171, 13.89974693, 12.9089559 , 13.32557133, 11.28895489,
        4.78544875,  7.44805901, 11.4424816 ,  5.46244988,  5.14124612,
       10.74655346, 13.48809595,  7.42512621, 11.99633819, 10.31886066,
        7.19513806, 13.12218013, 11.56146873, 13.32613445,  7.23276582,
       14.0393949 ,  8.66273171, 13.95361098, 11.85743425, 13.04217487,
        9.34061938, 12.91505111,  6.67218687,  8.83341608,  9.4997987 ,
       12.07225469])}
{'P': tensor([[1.2729e-02, 2.7937e-01, 1.9757e-01,  ..., 4.1194e-04, 1.7092e-05,
         2.9499e-05],
        [1.5762e-04, 1.2341e-02, 3.8886e-01,  ..., 7.2204e-04, 9.1824e-06,
         1.7588e-05],
        [3.6358e-04, 6.3603e-04, 4.1004e-02,  ..., 3.3668e-03, 2.2531e-05,
         4.5449e-05],
        ...,
        [3.5564e-06, 4.3473e-06, 5.0350e-06,  ..., 2.4395e-02, 2.4039e-06,
         2.7406e-06],
        [2.0120e-03, 1.8654e-03, 1.7402e-03,  ..., 3.1459e-04, 9.6721e-02,
         2.6453e-04],
        [1.3980e-01, 7.5476e-02, 5.2444e-02,  ..., 1.7251e-04, 6.9272e-04,
         9.6636e-02]]), 'X': tensor([[ -7.8359,   4.2735,   7.2943],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.9178,   2.1936,   7.5388],
        [ -8.9636,   2.0648,   8.1946],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.1706,   1.3863,   8.4032],
        [  6.8704,  -9.3809,   7.5769],
        [ -5.6174,   6.4708,  -1.7339],
        [  6.1685,  12.1979,  -6.2473],
        [ -4.0633,  -9.7927,  -3.0825],
        [  7.3592,  11.3937,   6.2407],
        [ 12.5919,   0.4035,  -5.8460],
        [  4.4049,   5.6892,  -4.4394],
        [  6.3604,   0.9042,  -6.5901],
        [  6.3583,   1.1729,   8.7835],
        [ -2.0229, -10.6163,   8.6930],
        [  5.3860,  -2.3418,   1.0826],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -1.3178, -10.7939,   0.4236],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 10.2712,   8.3707,  -4.9743],
        [  6.7792,  11.8084,   1.6251],
        [ -6.7992,   5.4991,   9.9186],
        [ -8.4297,  -5.3637,  -9.3962],
        [  4.0587, -10.6112,   6.2273],
        [ 11.7726,  -3.4876,  -6.7402],
        [  0.1287, -11.0065,   6.5757],
        [  8.6420,  -8.0580,   6.5781],
        [  0.5250, -11.0306,   5.5358],
        [  1.7903,  -4.7467,   0.1168],
        [  4.3459,  -3.5327,   0.3667],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.2076,  13.8032,  -1.5099],
        [ -8.3900,   3.3727,   3.1650],
        [  3.9022,  13.2906,   5.5788],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -7.8849,  -6.3115,  -4.7479],
        [  1.5246,  -4.7838,  -5.2095],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.2188,  -2.5839,   2.1833],
        [ -9.4739,  -1.2497,   0.5483],
        [  3.1480,  -4.3165,  -7.8002],
        [ -7.3333,   4.9246,  -7.9212],
        [ 12.3609,   3.4769,  -8.5762],
        [ 10.9931,   7.2107,   7.7933],
        [  1.7500,   7.4187,   4.4237],
        [ -2.6485, -10.4151,   0.8766],
        [  6.2555,   2.1474,   4.1685],
        [  0.1460,   7.8340,  -9.4614],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.7500,   7.4187,   4.4237],
        [ 10.9916,   7.2133,  -0.9179],
        [  7.0171,  11.6439,  -5.0178],
        [  5.6202,  -1.9508,   0.7471],
        [ -9.1433,  -3.5003,  -3.0209],
        [  4.3377,   5.7570,   1.3712],
        [  1.8320,   7.3871,   5.1452],
        [  9.6614,   9.1784,  -1.2087],
        [ 10.6673,  -5.6649,  -4.1646],
        [  0.5250, -11.0306,   5.5358],
        [ -6.6895,   5.6051,   2.7899],
        [ 12.6038,   0.7665,  -3.4439],
        [  5.6414, -10.0303,   7.8525],
        [  6.1755,   2.5233,  -2.0042],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.3373,  -4.6232,  -5.2246],
        [ -4.8665,  -9.3204,  -5.2564],
        [  0.9659,  -4.8144,  -0.9594],
        [  8.3791,  10.5409,   5.5590],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.4501,  -1.7133,   2.2980],
        [  2.5469,  13.7192,  -9.6935],
        [ -7.4808,   4.7462,   2.0266],
        [  2.2860,   7.1916,  -6.1011],
        [  2.2860,   7.1916,  -6.1011],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.4996,   2.6335,  -6.3166],
        [ -6.8466,  -7.6488,   1.9648],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.2675,  12.1379,  -8.9720],
        [  2.7685, -10.8982,  -6.8928],
        [  0.9445,   7.6736,   0.5095],
        [  2.9470,  13.6085,   2.9809],
        [  4.5800,   5.5034,  -3.9432],
        [  1.5246,  -4.7838,  -5.2095],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.2675,  12.1379,  -8.9720],
        [  6.3589,   0.8239,  -8.4230],
        [  5.6260,  -1.9401,   3.0259],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.2188,  -2.5839,   2.1833],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.4366,  -0.1189,  -2.6736],
        [  4.2041,  -3.6509,  -4.9488],
        [ -7.6525,  -6.6534,   5.1721],
        [  5.8450,  -9.9356,   3.0767],
        [ -9.1706,   1.3863,   8.4032],
        [  6.0955,   2.8210,  -4.3321],
        [  3.9515,  -3.8425,   6.1729],
        [  0.1041, -11.0045,   2.4311],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.9700,   7.3453,  -4.1486],
        [ -2.2188,   7.8218,   8.6064]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0672, 0.1098,  ..., 0.4044, 0.8369, 0.3332],
        [0.0672, 0.0000, 0.0438,  ..., 0.3822, 0.8716, 0.3971],
        [0.1098, 0.0438, 0.0000,  ..., 0.3611, 0.8844, 0.4409],
        ...,
        [0.4044, 0.3822, 0.3611,  ..., 0.0000, 0.6751, 0.6784],
        [0.8369, 0.8716, 0.8844,  ..., 0.6751, 0.0000, 0.8490],
        [0.3332, 0.3971, 0.4409,  ..., 0.6784, 0.8490, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.92549752,  9.09119351,  9.18358562,  9.19837792,  9.18690018,
        9.27474386, 11.62770332,  8.56889738, 13.66896739, 10.60226614,
       13.56367857, 12.59840193,  7.19513806,  6.42439344,  6.46560181,
       10.80728788,  5.87305005, 12.27835996, 10.87409005,  9.54154096,
       13.25017211, 13.61600536,  8.74470218,  9.99145096, 11.36089355,
       12.27835996, 11.00726483, 11.8159254 , 11.04313641,  5.07306274,
        5.60064319,  8.35856487, 13.97857591,  9.04255655, 13.85158417,
        9.97822473, 10.09979747,  5.0209005 , 13.00413829,  5.82345867,
        9.55593374,  5.34250225,  8.83341608, 12.84056604, 13.14690137,
        7.62232307, 10.74655346,  6.61386436,  7.83534987,  6.07992623,
        7.62232307, 13.1471313 , 13.59480906,  5.94909709,  9.79039371,
        7.20826219,  7.61088125, 13.32613445, 12.07818871, 11.04313641,
        8.7273569 , 12.62711252, 11.50791859,  6.67108662, 12.4217523 ,
        5.18047638, 10.51437947,  4.91038884, 13.46553623, 12.52372629,
        8.92098752,  9.60412735, 13.95361098,  8.859416  ,  7.54620849,
        7.54620849, 11.57249984, 12.77402351, 10.26545792,  6.27905292,
       10.05558893, 13.66051884, 11.24434374,  7.73151309, 13.92390689,
        7.15990676,  5.0209005 ,  9.97964013, 13.66051884,  6.41202986,
        5.95111804,  8.56889738,  5.82345867, 13.86480175, 10.78533838,
        9.43737523,  5.56809335, 10.14044489, 11.52733581,  9.27474386,
        6.7166297 ,  5.51177501, 11.0050379 , 10.48721382,  8.34949383,
        8.1303945 ])}
{'P': tensor([[1.4448e-02, 1.8981e-01, 1.7735e-01,  ..., 1.3198e-05, 1.0029e-04,
         1.0217e-05],
        [8.6861e-05, 9.0448e-03, 4.0871e-02,  ..., 1.7547e-05, 2.3590e-05,
         8.2068e-06],
        [7.9440e-05, 1.0024e-03, 1.1525e-02,  ..., 3.0815e-05, 2.3771e-05,
         9.7230e-06],
        ...,
        [1.2264e-06, 9.4427e-07, 9.8256e-07,  ..., 1.8352e-02, 2.6356e-06,
         3.8402e-04],
        [2.7811e-01, 1.0088e-01, 1.0391e-01,  ..., 2.0132e-05, 5.7592e-02,
         3.1758e-05],
        [1.3417e-04, 8.3094e-05, 6.4153e-05,  ..., 2.7420e-05, 2.8021e-04,
         5.8326e-02]]), 'X': tensor([[ -7.9298,   4.1367,  -3.9773],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.3739,   0.3982,  -4.2564],
        [ -9.3705,   0.4208,  -3.5815],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.8093,  -9.4178,   8.4244],
        [  3.0719, -10.8449,  -4.1673],
        [ 10.2712,   8.3707,  -4.9743],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.9653,  -9.3225,  -3.4727],
        [  6.2026,  12.1774,  -1.5887],
        [ 11.6634,   5.8050,   9.1590],
        [  7.8425,  -8.7202,   6.4530],
        [  3.6154,  -4.0640,  -7.9698],
        [  0.5960, -11.0335,   3.1144],
        [  6.1222,  -0.7261,   8.8204],
        [  2.6613,   7.0021,   4.5783],
        [  6.2975,   0.0947,   5.1791],
        [  5.9655,   3.2172,  -7.5953],
        [  2.5382,   7.0672,   4.5097],
        [ -1.9487,   7.8590,  -3.3330],
        [ -2.5683, -10.4433,  -2.9182],
        [ 10.4758,  -5.9539,   1.3667],
        [ -6.2200,  -8.2737,   5.5079],
        [  3.6249, -10.7255,   5.6605],
        [  1.2728, -11.0372,   9.8121],
        [  2.5636, -10.9293,   9.6519],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.3076,   0.7852,  -6.7010],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.2895,   0.0405,   5.7652],
        [  0.1497,   7.8334,  -5.4412],
        [  9.3624,   9.5318,   0.2779],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -5.7347,  -8.6908,   9.2878],
        [ 12.3357,   3.6001,   0.7903],
        [  4.0421,  -3.7765,  -8.3631],
        [  0.9478,  -4.8144,   0.6371],
        [  9.7328,   9.0902,  -0.8917],
        [ -3.3812, -10.1235,   7.4384],
        [  6.2883,   0.0325,   5.2692],
        [ -9.1624,  -3.4286,   2.0647],
        [  7.9403,  -8.6455,  -8.7433],
        [  0.8940, -11.0402,  -0.9783],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.1771, -10.9776,  -3.2007],
        [  5.9003,  12.3550,   2.1917],
        [  3.9663,   6.1018,  -5.4104],
        [ 10.0400,   8.6922,  -6.2847],
        [  0.2732,  -4.7619,  -1.1577],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.3122,  -9.6994,  -6.3282],
        [ -7.2515,   5.0195,   4.6434],
        [  3.1892, -10.8220,  -6.8409],
        [  4.5430,   5.5438,  -3.8669],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4505,  -1.7081,  -0.7039],
        [  6.3200,   0.2690,   1.5900],
        [ -0.8675, -10.8815,   6.8561],
        [  9.2979,   9.6048,   7.3138],
        [  6.3266,   0.3292,   9.3502],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -2.6069, -10.4298,  -6.1995],
        [  1.9204,   7.3518,   8.4781],
        [ -8.5149,  -5.1905,   1.5239],
        [  9.6662,   9.1725,   8.1056],
        [  1.8320,   7.3871,   5.1452],
        [ -4.7334,   6.9982,  -0.4392],
        [ 12.4996,   2.6335,  -6.3166],
        [  7.0515,  11.6194,   8.4289],
        [  6.3122,  -9.6994,  -6.3282],
        [ -1.3178, -10.7939,   0.4236],
        [  0.7685,  -4.8098,  -2.9310],
        [ -5.1377,   6.7751,  -0.4438],
        [  5.6523,   3.9443,  -2.9449],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.3542,   0.6934,  -3.3840],
        [ -6.2200,  -8.2737,   5.5079],
        [ -8.5607,  -5.0938,  -5.4531],
        [  5.2939,  -2.4784,  -4.5840],
        [ -6.7992,   5.4991,   9.9186],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.1519, -10.2387,   9.8191],
        [ 12.1588,   4.3364,  -8.0867],
        [ -1.1885, -10.8211,   0.0865],
        [  0.1931,  -4.7491,   7.4385],
        [ -4.3963,   7.1629,   4.5263],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.1890,   4.7318,   0.9515],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.3124,   0.2057,  -0.4593],
        [ -6.4002,  -8.1048,  -0.9108],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.2630,   2.1057,  -5.2136],
        [ -6.6998,   5.5954,   0.0823],
        [  4.9452, -10.3187,  -7.8621],
        [ -5.2589,   6.7024,  -5.5809],
        [  6.1984,  -0.4327,   1.5438]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0575e-01, 1.3225e-01,  ..., 4.3133e-01, 2.2946e-01,
         8.3231e-01],
        [1.0575e-01, 0.0000e+00, 6.4606e-02,  ..., 5.3252e-01, 3.3501e-01,
         9.1174e-01],
        [1.3225e-01, 6.4606e-02, 3.0339e-04,  ..., 5.2950e-01, 3.4788e-01,
         9.6076e-01],
        ...,
        [4.3133e-01, 5.3252e-01, 5.2950e-01,  ..., 0.0000e+00, 2.1718e-01,
         7.3957e-01],
        [2.2946e-01, 3.3501e-01, 3.4788e-01,  ..., 2.1718e-01, 0.0000e+00,
         6.9933e-01],
        [8.3231e-01, 9.1174e-01, 9.6076e-01,  ..., 7.3957e-01, 6.9933e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.94395628,  9.20340031,  9.26455795,  9.38232121,  9.37989631,
        9.37296102, 11.62158554, 11.27160035, 13.25017211,  8.41171369,
       11.63724792, 13.6660671 , 13.02818702, 11.72802491,  5.43943791,
       11.04953767,  6.16512856,  7.49076899,  6.29821673,  6.77777606,
        7.50917551,  8.09704052, 10.75443106, 12.0495521 , 10.35093485,
       11.32149259, 11.11038551, 11.22596994,  8.92098752, 12.18045978,
        9.60359158,  9.34061938,  8.22728399,  6.28961988,  7.83486943,
       13.3607379 , 12.12448912, 10.412299  , 12.85032836,  5.53175275,
        4.90676242, 13.31764181, 10.67322527,  6.28835101,  9.78285424,
       11.73848065, 11.07637531,  4.8763135 , 11.1913598 , 13.69162795,
        7.27759233, 13.27994304,  4.76969364,  8.4486771 , 11.57249984,
        8.81930525, 11.28215429,  7.16747878,  5.13616224,  9.60359158,
        6.32572323, 10.91601655, 13.36800073,  6.33517349, 12.46513646,
       10.7506423 ,  7.5984809 ,  9.97220799, 13.32557133,  7.61088125,
        8.4486771 , 12.77402351, 13.59170809, 11.57249984, 10.87409005,
        4.87083348,  8.50278896,  6.89245352, 13.03002338,  9.4997987 ,
        9.7712321 ,  6.39187285, 10.35093485,  9.96152788,  5.84533345,
        8.74470218,  5.48276038, 11.46176259, 12.9089559 , 10.88618114,
        4.75303127,  8.40445872,  9.69629   , 10.60226614, 12.22503754,
        7.02253235,  9.75200848,  9.37296102,  6.31576264, 10.32715715,
        5.84450127,  6.60752373,  8.72896381, 11.4424816 ,  8.51928531,
        6.2134961 ])}
{'P': tensor([[1.0366e-02, 2.5681e-01, 2.2986e-01,  ..., 4.8266e-06, 4.7139e-04,
         3.0084e-06],
        [1.8889e-04, 1.4028e-02, 2.5145e-01,  ..., 5.7669e-06, 1.0426e-03,
         3.4259e-06],
        [2.3202e-04, 8.3341e-04, 1.8135e-02,  ..., 7.2461e-06, 1.3805e-03,
         4.2372e-06],
        ...,
        [8.2361e-04, 5.1750e-04, 4.7460e-04,  ..., 5.6518e-02, 2.6757e-05,
         2.6131e-02],
        [4.2720e-06, 5.4254e-06, 5.5589e-06,  ..., 2.3863e-05, 2.3506e-02,
         1.5568e-05],
        [4.2934e-04, 2.8285e-04, 2.5716e-04,  ..., 4.0317e-03, 2.5670e-05,
         4.8048e-02]]), 'X': tensor([[-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0712, 0.0891,  ..., 0.7756, 0.5446, 0.8543],
        [0.0712, 0.0000, 0.0195,  ..., 0.8401, 0.5284, 0.9115],
        [0.0891, 0.0195, 0.0000,  ..., 0.8596, 0.5345, 0.9308],
        ...,
        [0.7756, 0.8401, 0.8596,  ..., 0.0000, 0.8725, 0.1861],
        [0.5446, 0.5284, 0.5345,  ..., 0.8725, 0.0000, 0.8304],
        [0.8543, 0.9115, 0.9308,  ..., 0.1861, 0.8304, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.95133961,  9.09502385,  9.11473391,  9.14973319,  9.36093562,
        9.36378438,  9.7751189 ,  9.78286733, 10.06703358,  8.91551231,
        6.84796202,  4.75303127,  8.04664886,  4.91336994,  5.5513072 ,
       13.3614598 ,  7.32754882, 10.31886066,  9.96152788,  5.39723241,
        9.48337063,  7.02326515,  6.83083863,  7.20482224,  7.06169391,
        6.28835101,  6.61386436,  7.19534146, 10.36844813, 13.50484539,
       12.3489324 ,  8.62631801,  6.33517349, 10.14044489,  7.46248947,
        5.27178363, 13.3607379 ,  8.57560275,  8.66273171,  6.81084724,
        7.19534146,  8.28968067,  8.51928531, 10.4606265 ,  6.81117802,
        8.77913428, 10.41909211,  8.67227386,  9.23178204, 11.32149259,
        7.18354885, 11.83509372,  5.13182506, 10.00802627,  9.3349858 ,
        7.13501264, 10.794755  ,  6.14086582,  8.859416  ,  9.20340031,
        7.15990676,  9.3349858 ,  9.562174  ,  8.87587457,  6.1463896 ,
       11.4088043 , 13.5247215 ,  6.34059021,  7.57685908, 11.22142195,
        9.48969978, 12.79126785,  7.61088125,  5.88782913, 11.50791859,
       11.46176259,  6.97106912, 13.44593519, 14.10796466, 12.05395056,
        5.18277858,  4.89121805, 13.32374056,  6.5618968 , 11.00812996,
        6.80370254,  8.06990483,  9.69792513,  8.01009083, 12.52875793,
       10.48315436,  5.37343173, 11.83570132,  6.7593917 , 10.06703358,
       12.26997165,  8.25828409, 13.03002338, 11.05317597,  8.95983586,
       11.51557166, 13.2430377 , 12.4013153 ,  6.99097013, 10.35093485,
        7.14837514])}
{'P': tensor([[1.1862e-02, 3.0623e-01, 1.7848e-01,  ..., 2.2458e-05, 3.7270e-07,
         1.9090e-06],
        [4.7149e-04, 2.2609e-02, 4.1729e-01,  ..., 3.2002e-05, 5.8615e-07,
         2.9479e-06],
        [1.4945e-03, 2.0980e-03, 1.8114e-01,  ..., 1.7563e-04, 4.1133e-06,
         1.8781e-05],
        ...,
        [1.7841e-04, 1.8856e-04, 3.2060e-04,  ..., 6.9771e-02, 5.4176e-05,
         7.0106e-05],
        [7.2908e-05, 5.9471e-05, 5.2455e-05,  ..., 1.8679e-02, 1.7096e-01,
         2.8117e-02],
        [4.8250e-04, 3.8371e-04, 3.1141e-04,  ..., 2.7280e-02, 7.3234e-03,
         1.0859e-01]]), 'X': tensor([[ -8.0087,   4.0173,   9.1091],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.9282,   2.1648,   8.4894],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.9636,   2.0648,   8.1946],
        [  6.6319,  11.9066,   3.5165],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.5763,   0.1333,  -9.2650],
        [  6.1929,   2.4501,  -6.9352],
        [  2.3376,  -4.6231,   9.8436],
        [ -1.6275, -10.7221,   3.2376],
        [  6.1730,  -0.5375,  -8.8861],
        [  5.1793,   4.7461,   0.3899],
        [ -8.5859,  -5.0393,   0.5476],
        [ -6.4464,   5.8272,   7.5038],
        [  2.4572,   7.1085,   7.7376],
        [  9.6816,   9.1536,  -5.8543],
        [  1.5662,  13.9375,   4.5142],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.2301,  -3.1537,  -2.7477],
        [  0.9809,  -4.8145,   6.8706],
        [ -5.1369,   6.7755,   8.3836],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.3031,   1.8477,  -7.1706],
        [  6.1579,  -0.5961,  -6.5595],
        [ -8.4887,  -5.2448,  -0.6585],
        [  6.2883,   0.0325,   5.2692],
        [  6.3413,   0.4883,  -6.8416],
        [  5.9026,   3.3837,  -0.6573],
        [  2.4149,  -4.6005,  -4.3339],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.4753,  -0.8012,  -5.2050],
        [  7.3592,  11.3937,   6.2407],
        [  2.1477,   7.2550,   8.9688],
        [ -3.5520,   7.4976,   9.1522],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.7078,   2.7182,   1.8146],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.3331,  -2.6465,  -8.2028],
        [  4.7482, -10.3907,   5.9149],
        [  8.0877,  -8.5297,  -1.1050],
        [  1.1139,   7.6281,  -2.5680],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -8.3900,   3.3727,   3.1650],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.3583,   1.1729,   8.7835],
        [ -5.6423,   6.4537,  -1.2091],
        [  0.9036,   7.6840,   9.9237],
        [  4.5800,   5.5034,  -3.9432],
        [ -4.7389,   6.9953,   3.7638],
        [  1.7218,  -4.7576,  -3.3617],
        [  9.4554,   9.4245,   9.8800],
        [  6.2022,   2.4092,   3.0349],
        [ -4.2040,   7.2486,   2.9875],
        [ -9.4501,  -1.7133,   2.2980],
        [  8.9284,  10.0036,  -6.9210],
        [  4.4049,   5.6892,  -4.4394],
        [ 11.9122,   5.1397,  -1.5498],
        [ -4.0409,   7.3169,  -3.5779],
        [  7.9816,  10.8938,  -0.5378],
        [  5.4732,  -2.2041,  -4.3830],
        [ -7.7239,   4.4298,   3.1216],
        [  9.0051,   9.9234,   5.4411],
        [  5.8930,  12.3592,  -8.9201],
        [  1.6308,  13.9254,  -2.0984],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.3535,  -5.5118,   9.8399],
        [  2.9470,  13.6085,   2.9809],
        [ -8.1531,  -5.8748,   2.6934],
        [ -5.0132,  -9.2234,  -8.3531],
        [  2.0449,  13.8402,  -7.1642],
        [  0.8572,  14.0501,  -8.3364],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.3303,   1.6145,  -8.4083],
        [  6.2804,   2.0019,   0.8607],
        [ -7.0948,   5.1940,   9.8171],
        [  9.3624,   9.5318,   0.2779],
        [  1.5653,  13.9377,   4.1773],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.2508,   1.0572,   2.9978],
        [  6.3119,   1.7797,   9.9186],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.2687,  -2.9794,   0.6436],
        [  6.1989,  -9.7591,   0.8289],
        [  5.1787,  -2.6383,  -4.2533],
        [ -5.1046,  -9.1610,   2.3418],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.6022,  -1.9833,   8.3081],
        [ 12.2692,  -1.8882,   3.9253],
        [  9.1409,  -7.5789,   1.5814],
        [ -7.5603,   4.6459,   8.2002],
        [  6.3549,   0.7096,   6.9198],
        [  6.3542,   0.6934,  -3.3840],
        [  2.3589,  13.7668,  -3.6501],
        [  0.3038,   7.8092,   0.4164],
        [  6.3303,   1.6145,  -8.4083],
        [  5.5626,  -2.0536,   8.4433],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.0467,  -9.8369,  -6.5816],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.3367,   7.1676,  -9.5263],
        [  1.6408,   7.4591,  -4.2294]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0423, 0.1015,  ..., 0.9556, 1.1589, 1.0080],
        [0.0423, 0.0000, 0.0602,  ..., 0.9604, 1.1810, 1.0341],
        [0.1015, 0.0602, 0.0000,  ..., 0.9531, 1.1989, 1.0587],
        ...,
        [0.9556, 0.9604, 0.9531,  ..., 0.0000, 0.5049, 0.5426],
        [1.1589, 1.1810, 1.1989,  ..., 0.5049, 0.0000, 0.1952],
        [1.0080, 1.0341, 1.0587,  ..., 0.5426, 0.1952, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.95983586,  9.0368458 ,  9.23071698,  9.18690018,  8.98765644,
        9.19837792, 13.62896593,  8.06643487, 12.5769659 ,  6.65991877,
        5.1805345 , 10.84493732,  6.19633139,  7.02496223,  9.95552626,
        8.68979052,  7.52117132, 13.32374056, 14.02526505,  9.62238828,
        9.75401449,  4.91336994,  8.50268322,  8.36995953,  6.56833864,
        6.18668463,  9.97822473,  6.28835101,  6.36004393,  6.80370254,
        5.19577146,  5.83560009,  9.50913849, 13.56367857,  7.56617037,
        8.29640613,  4.91652556,  9.12220271,  9.43112957,  9.70108409,
       11.42420508, 11.75438006,  7.70897675,  9.79039371,  9.04255655,
       12.53047489,  6.46560181,  8.5724042 ,  7.73692436,  7.15990676,
        8.44940661,  5.05963217, 13.35013374,  6.65368551,  8.37954853,
        9.60412735, 13.40850328,  7.19513806, 12.97370371,  8.35856487,
       13.50484539,  5.90035324,  8.90405823, 13.40024224, 13.6922425 ,
       14.02058621, 10.54658597, 10.00802627, 13.92390689, 10.04916678,
       10.4977132 , 13.99048094, 14.07623266,  4.78544875,  9.76896904,
        6.53290663,  6.59176018,  8.79284466, 13.3607379 , 14.02532758,
        7.89745107,  9.3109884 ,  6.55800267, 12.46513646,  9.79039371,
        9.60412735,  9.73579738, 11.56146873,  5.81199422, 10.48721382,
        6.07992623,  5.94293185, 12.41366754, 11.87412439,  8.8737556 ,
        6.39438044,  6.39187285, 13.96746816,  7.81509631,  6.53290663,
        5.92952538,  9.95552626, 11.54672755,  8.89458273,  7.53884104,
        7.6374603 ])}
{'P': tensor([[1.3216e-02, 2.6472e-01, 2.5828e-01,  ..., 2.0121e-05, 5.7445e-07,
         1.2435e-07],
        [3.0134e-04, 2.3019e-02, 4.1343e-02,  ..., 4.2499e-05, 7.1736e-07,
         1.6484e-07],
        [2.7249e-04, 7.0096e-03, 2.3102e-02,  ..., 4.2028e-05, 6.9426e-07,
         1.8555e-07],
        ...,
        [2.8784e-04, 2.0989e-04, 1.9597e-04,  ..., 1.3356e-01, 3.4627e-03,
         1.9857e-04],
        [6.1997e-04, 2.8381e-04, 2.5797e-04,  ..., 5.8934e-06, 2.6136e-02,
         1.2934e-05],
        [5.9627e-06, 2.9904e-06, 3.4181e-06,  ..., 5.4915e-07, 1.1836e-04,
         8.7331e-03]]), 'X': tensor([[ -8.0608,   3.9362,  -5.9567],
        [ -8.7735,   2.5640,  -6.3313],
        [ -8.8363,   2.4086,  -5.6296],
        [ -9.3076,   0.7852,  -6.7010],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.1459,   1.4780,  -4.2403],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.4505,  -1.7081,  -0.7039],
        [  6.0310,  12.2795,  -6.4528],
        [ -3.9508,  -9.8515,   1.7923],
        [ -7.2515,   5.0195,   4.6434],
        [ 11.5966,   5.9666,   6.0083],
        [ -8.9624,  -4.0944,  -4.9622],
        [  4.4039,   5.6903,   8.7366],
        [  1.4150,  13.9647,  -4.0915],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.9193,  -1.3176,   5.1739],
        [  6.0031,  12.2958,  -6.5992],
        [ 10.9875,   7.2207,  -8.8549],
        [  0.6470, -11.0352,  -1.1931],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.4351,  -1.8891,   3.5642],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.9653,  -9.3225,  -3.4727],
        [  0.9445,   7.6736,   0.5095],
        [ -0.8885,   7.9160,   0.0571],
        [ -2.6735, -10.4062,  -1.5911],
        [  3.9663,   6.1018,  -5.4104],
        [ -9.1459,   1.4780,  -4.2403],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.5123,  -0.5339,  -1.0728],
        [  7.5309,  11.2617,   7.2656],
        [  0.3493,  -4.7727,  -6.5682],
        [  2.1625,  -4.6697,   7.0658],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.2571,  -4.7594,   1.5618],
        [  5.1861,   4.7361,  -3.7312],
        [ -3.0008, -10.2827,  -8.9480],
        [ -5.9987,   6.1949,  -6.2114],
        [  3.9895,   6.0816,  -6.5235],
        [  9.1409,  -7.5789,   1.5814],
        [ 12.3357,   3.6001,   0.7903],
        [ -5.7347,  -8.6908,   9.2878],
        [ -4.9217,   6.8979,  -1.7667],
        [  3.1480,  -4.3165,  -7.8002],
        [  1.5653,  13.9377,   4.1773],
        [  3.7121,   6.3117,   0.6738],
        [  5.6928,  12.4710,   3.6001],
        [  8.2154,  -8.4262,   8.0171],
        [ -0.5709,   7.9055,   7.6089],
        [  3.4604,  -4.1545,   1.8091],
        [  4.7156,  -3.1831,  -8.0224],
        [  2.5010,  -4.5736,  -0.8989],
        [ 11.6019,   5.9540,  -5.1715],
        [  7.9403,  -8.6455,  -8.7433],
        [ -2.9497, -10.3028,  -7.3479],
        [ 11.6165,  -3.8715,   6.4594],
        [ -3.7964,  -9.9294,   7.7147],
        [ -2.0229, -10.6163,   8.6930],
        [  0.9788,  -4.8145,  -3.8661],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.3569,   0.7591,   6.5011],
        [  6.3587,   1.1596,  -3.6629],
        [ 11.6634,   5.8050,   9.1590],
        [ -4.1296,   7.2803,  -2.7194],
        [  1.3103,  -4.8032,  -0.9300],
        [  7.5784,  -8.9141,   5.8410],
        [  7.1821,  -9.1845,   9.8548],
        [ -0.4341, -10.9473,   7.4509],
        [  6.1752,   2.5242,   3.5874],
        [ -6.4002,  -8.1048,  -0.9108],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.9033,   2.2329,  -8.8912],
        [ -6.7745,  -7.7264,   8.4684],
        [  2.8474,   6.8979,   6.6457],
        [ -9.4630,  -1.5145,   0.2983],
        [  4.5954,  13.0099,  -4.4439],
        [ -8.8058,   2.4850,  -1.7425],
        [  1.3670,  -4.7990,   3.2603],
        [  7.0171,  11.6439,  -5.0178],
        [  9.0652,   9.8597,  -3.5949],
        [  3.9913,  -3.8139,  -6.8311],
        [  4.1637,  13.1898,  -1.9189],
        [ 11.0438,   7.1180,  -5.6991],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.8450,  -9.9356,   3.0767],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -1.6275, -10.7221,   3.2376],
        [  4.2612, -10.5514,  -6.6268],
        [  2.3200,   7.1756,   1.1938],
        [ -8.0330,  -6.0765,   6.7945],
        [ -9.4196,   0.0478,   9.4140],
        [ -4.8751,  -9.3148,   4.0406],
        [  0.6142,  14.0802,   0.5407],
        [  0.9074,  -4.8139,   8.1674],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.3515,   0.6418,  -6.5823],
        [ -2.6069, -10.4298,  -6.1995],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.0310,  12.2795,  -6.4528],
        [  9.2692,   9.6370,   7.0746]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.1259e-02, 7.9498e-02,  ..., 7.5111e-01, 9.9361e-01,
         1.5586e+00],
        [8.1259e-02, 0.0000e+00, 3.0412e-02,  ..., 8.2937e-01, 1.0694e+00,
         1.6300e+00],
        [7.9498e-02, 3.0412e-02, 3.0339e-04,  ..., 8.3029e-01, 1.0731e+00,
         1.6098e+00],
        ...,
        [7.5111e-01, 8.2937e-01, 8.3029e-01,  ..., 0.0000e+00, 2.5662e-01,
         1.1656e+00],
        [9.9361e-01, 1.0694e+00, 1.0731e+00,  ..., 2.5662e-01, 0.0000e+00,
         1.2162e+00],
        [1.5586e+00, 1.6300e+00, 1.6098e+00,  ..., 1.1656e+00, 1.2162e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.97052115,  9.14044984,  9.15866607,  9.34061938,  9.41605684,
        9.26455795,  5.05981914,  9.60359158, 13.68063098, 10.61417181,
        8.81930525, 13.04156075,  9.85330646,  7.19534146, 14.03618649,
        6.60752373,  9.62238828,  6.06416957, 13.6829843 , 13.14777433,
       11.05413545,  6.80370254,  9.62238828,  5.5513072 , 11.63724792,
        7.73151309,  7.96575782, 10.74409834,  7.27759233,  9.26455795,
       12.53285633, 12.67576792, 12.52372629, 13.54774355,  4.78544875,
        5.14607014, 12.27993098, 10.00410625,  4.7663526 ,  7.02326515,
       10.71162681,  8.62329466,  7.27340834, 11.87412439, 12.85032836,
       10.412299  ,  8.47371868,  5.34250225, 14.02532758,  7.32233733,
       13.70893492, 11.76830981,  7.92607342,  5.40689332,  5.68942572,
        5.21280156, 13.04051744, 11.73848065, 10.7167285 , 12.24467041,
       10.63038565, 10.80728788,  4.91296205, 12.45418236,  6.4020404 ,
        6.46356806, 13.02818702,  8.36995953,  4.978704  , 11.70016217,
       11.65923396, 10.95593729,  6.67121748, 10.32715715,  5.35774573,
        9.17905491, 10.27572673,  7.46248947,  9.58347456, 13.79762415,
        9.14973319,  4.98989906, 13.59480906, 13.39372426,  5.5205189 ,
       13.83139024, 13.13888841,  4.98461983, 11.52733581,  9.51074763,
       12.30858677, 10.84493732, 11.37939695,  7.54127425, 10.07238463,
        9.41970031, 10.51340829, 14.09357547,  4.89869205,  9.87720102,
        6.383897  , 10.7506423 , 10.36844813,  5.94541803, 13.68063098,
       13.37121923])}
{'P': tensor([[1.1198e-02, 4.0115e-01, 1.7507e-01,  ..., 8.6029e-05, 1.4510e-04,
         4.0243e-04],
        [3.3688e-04, 1.7565e-02, 3.2419e-01,  ..., 1.6150e-04, 2.6061e-04,
         7.3114e-04],
        [5.4641e-04, 6.6700e-04, 5.7588e-02,  ..., 1.2019e-03, 1.6633e-03,
         4.8554e-03],
        ...,
        [1.0602e-05, 1.0869e-05, 1.3209e-05,  ..., 1.3899e-01, 2.7432e-04,
         1.6061e-04],
        [1.0110e-05, 1.1043e-05, 1.7911e-05,  ..., 1.4291e-01, 9.1754e-02,
         5.8797e-03],
        [1.8991e-05, 2.0968e-05, 3.5045e-05,  ..., 3.1576e-01, 2.2558e-01,
         9.8010e-02]]), 'X': tensor([[ -8.1426,   3.8048,   9.3542],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.9636,   2.0648,   8.1946],
        [  1.8721,  -4.7322,   0.5524],
        [  3.3167,   6.6022,   6.3554],
        [ 12.6000,   1.3839,  -3.2992],
        [ -1.6220,   7.8917,   0.0315],
        [  5.7802,  -1.6363,  -6.2782],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.6154,  -4.0640,  -7.9698],
        [  6.2555,   2.1474,   4.1685],
        [  9.0004,  -7.7195,   4.6945],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -6.4844,  -8.0230,   8.5209],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.2589,  -1.9312,   9.8657],
        [ -9.4501,  -1.7133,   2.2980],
        [  7.1441,  11.5529,   2.3232],
        [  7.8982,  -8.6778,   0.0239],
        [  2.6970,   6.9826,   9.7350],
        [  6.9284,  11.7061,  -0.3732],
        [ -6.4822,  -8.0252,  -3.8983],
        [  4.5800,   5.5034,  -3.9432],
        [ -9.4671,  -0.5553,   0.1117],
        [ -2.0840,   7.8415,  -7.8657],
        [ -6.2668,   5.9808,   2.5614],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.1755,   2.5233,  -2.0042],
        [ -7.4733,   4.7555,   6.2490],
        [  6.7565,  11.8237,  -4.9882],
        [  4.0078,  -3.8018,  -6.2168],
        [  1.6422,   7.4586,  -6.8056],
        [  2.3589,  13.7668,  -3.6501],
        [ -8.6112,   2.9319,  -7.9943],
        [  2.5974,   7.0363,  -7.7810],
        [ -9.3616,  -2.4749,   0.1793],
        [  1.8320,   7.3871,   5.1452],
        [  6.3604,   0.9042,  -6.5901],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.2242,  -0.3171,   7.4978],
        [ 11.5198,  -4.0923,   5.5088],
        [  0.3715,  -4.7756,   0.2312],
        [ -8.0965,  -5.9712,   4.3466],
        [  2.2849, -10.9655,  -7.6542],
        [  0.7958,  -4.8109,  -8.2248],
        [  5.2608,  -2.5256,  -9.1446],
        [ -6.3306,   5.9272,   5.6935],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.0961,   1.6529,   3.1553],
        [  2.5636, -10.9293,   9.6519],
        [ -9.3464,   0.5708,  -1.6397],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.4336,  -2.2676,   5.6933],
        [ -2.0445,   7.8469,  -0.5857],
        [  6.0345,   3.0175,  -1.4424],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.6414, -10.0303,   7.8525],
        [  2.5663,   7.0526,  -6.8769],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.0305,   3.0296,   0.1375],
        [ -9.4721,  -1.3097,  -7.4377],
        [  0.2571,  -4.7594,   1.5618],
        [  6.2036,   2.4030,  -1.2500],
        [  8.8091,  -7.9037,   0.3631],
        [  9.9750,  -6.6338,  -1.2440],
        [  6.3549,   0.7096,   6.9198],
        [ -2.2188,   7.8218,   8.6064],
        [ -2.0229, -10.6163,   8.6930],
        [ -5.0132,  -9.2234,  -8.3531],
        [ 11.0275,  -5.0666,   8.4154],
        [  0.3038,   7.8092,   0.4164],
        [  6.2419,  -0.2313,   4.2440],
        [  4.5325,  -3.3643,   6.7699],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -2.2188,   7.8218,   8.6064],
        [ -8.6381,  -4.9238,  -4.4077],
        [  1.1350,  14.0104,   0.6321],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -6.4464,   5.8272,   7.5038],
        [  8.1217,  -8.5024,  -0.6149],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.6319,  11.9066,   3.5165],
        [ 10.9717,  -5.1646,  -6.9039],
        [  3.9515,  -3.8425,   6.1729],
        [  0.3724,   7.7974,  -6.8722],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.8154,   2.4611,   2.5198],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.1752,   2.5242,   3.5874],
        [  2.5469,  13.7192,  -9.6935],
        [  3.9022,  13.2906,   5.5788],
        [ 10.9162,  -5.2599,  -7.0812],
        [  8.3718,  -8.2953,   8.4666],
        [  5.7784,  -1.6401,   9.5587],
        [ -7.5401,   4.6718,   9.0033],
        [  9.3836,   9.5075,  -5.9278],
        [  5.8902,  -1.3887,   4.0388],
        [  3.6817,   6.3354,   6.1233],
        [  2.4639, -10.9431,  -7.3718],
        [ -1.6275, -10.7221,   3.2376],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -6.9838,  -7.4965,   0.9796]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.7082e-02, 8.6947e-02,  ..., 3.0471e-01, 7.2470e-01,
         5.6798e-01],
        [2.7082e-02, 0.0000e+00, 6.0236e-02,  ..., 3.0027e-01, 7.1777e-01,
         5.5771e-01],
        [8.6947e-02, 6.0236e-02, 0.0000e+00,  ..., 2.8710e-01, 6.9427e-01,
         5.2787e-01],
        ...,
        [3.0471e-01, 3.0027e-01, 2.8710e-01,  ..., 0.0000e+00, 4.2121e-01,
         2.7637e-01],
        [7.2470e-01, 7.1777e-01, 6.9427e-01,  ..., 4.2121e-01, 3.0339e-04,
         1.7880e-01],
        [5.6798e-01, 5.5771e-01, 5.2787e-01,  ..., 2.7637e-01, 1.7880e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.98765644,  9.0368458 ,  9.23071698,  9.18690018,  9.27474386,
        9.19837792,  5.08910491,  7.38844703, 12.67576792,  8.05668609,
        6.00732193,  9.79814997,  9.88310614,  5.43943791,  6.61386436,
       11.85743425,  9.88310614, 10.31584102, 12.59840193, 12.41012046,
        9.60412735, 13.58333857, 11.73397309,  7.48538905, 13.60275211,
       10.31613269,  7.15990676,  9.48337063,  8.1137452 ,  8.66273171,
        5.56809335,  6.67108662,  8.8580671 , 13.61801526,  5.52416368,
        7.63726134, 13.96746816,  9.09661212,  7.50035355,  9.68323686,
        7.61088125,  6.42439344, 12.35421461,  6.2322769 , 12.22503754,
        4.79003361, 10.06023665, 11.20100795,  4.8763135 ,  5.83560009,
        8.67227386,  8.56889738,  9.24501829, 11.22596994,  9.36378438,
       10.31584102,  5.88782913,  8.10886021,  6.74686071, 12.22503754,
       11.50791859,  7.50500064,  8.85572292,  6.74872981,  9.562174  ,
        4.7663526 ,  6.652742  , 11.83509372, 11.97949273,  6.39438044,
        8.1303945 , 10.80728788, 10.4977132 , 12.13568598,  7.81509631,
        6.24614905,  5.64464807, 10.61047596,  8.1303945 ,  9.94285511,
       14.05633441, 10.54658597,  8.68979052, 11.75807252,  5.43544626,
       13.62896593, 12.12641969,  5.51177501,  7.80625981,  9.26455795,
        9.15253344, 13.27994304,  6.67121748, 13.95361098, 13.85158417,
       12.11735334, 11.785566  ,  6.00663314,  8.87007495, 13.35832725,
        6.05165248,  7.32754882, 11.21703654, 10.84493732, 10.33712258,
       10.24557047])}
{'P': tensor([[9.2569e-03, 2.1136e-01, 2.0537e-01,  ..., 1.1582e-06, 9.8212e-06,
         6.7473e-05],
        [1.4613e-04, 1.0210e-02, 1.8128e-01,  ..., 1.6698e-06, 7.5376e-06,
         9.4836e-05],
        [1.4585e-04, 5.0591e-04, 1.2638e-02,  ..., 2.2131e-06, 7.7813e-06,
         1.2787e-04],
        ...,
        [9.7158e-07, 6.8135e-07, 6.2491e-07,  ..., 6.6795e-03, 9.9999e-07,
         2.4265e-05],
        [9.5858e-04, 1.1825e-03, 1.0563e-03,  ..., 2.0568e-06, 9.3023e-02,
         2.2848e-05],
        [7.2322e-05, 6.3780e-05, 6.3111e-05,  ..., 7.1499e-03, 1.0472e-05,
         5.3435e-02]]), 'X': tensor([[ -8.1666,   3.7653,   3.6075],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2508,   1.0572,   2.9978],
        [  5.5294,  -2.1106,   9.0626],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.3087,   1.8053,  -1.9481],
        [  4.4915,   5.5990,   7.0405],
        [  4.6992,   5.3690,   3.0935],
        [  7.8425,  -8.7202,   6.4530],
        [ -7.3446,   4.9113,  -5.9356],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.1984,  -0.4327,   1.5438],
        [ -8.6050,   2.9451,  -0.7012],
        [  3.3094,  13.4969,  -3.6507],
        [ -8.7078,   2.7182,   1.8146],
        [  6.1989,  -9.7591,   0.8289],
        [  3.1756, -10.8247,   6.8882],
        [  5.1203,  -2.7151,   6.2110],
        [  0.6470, -11.0352,  -1.1931],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.5813,  -4.5471,   5.3597],
        [  4.1348,   5.9514,  -0.8856],
        [  0.5960, -11.0335,   3.1144],
        [  4.0587, -10.6112,   6.2273],
        [  9.1820,   9.7334,  -2.7932],
        [  6.9284,  11.7061,  -0.3732],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.1706,   1.3863,   8.4032],
        [ 10.4994,  -5.9193,  -7.5153],
        [  9.6587,   9.1817,  -0.8212],
        [  1.8719,   7.3713,  -3.7465],
        [  3.2159, -10.8166,  -1.0406],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -1.9646,   7.8571,   3.3817],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.1890,   4.7318,   0.9515],
        [  5.4336,  -2.2676,   5.6933],
        [  1.4489,   7.5256,   4.4529],
        [  5.6022,  -1.9833,   8.3081],
        [  5.7802,  -1.6363,  -6.2782],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.7685,  -4.8098,  -2.9310],
        [  5.2608,  12.6977,   4.9166],
        [  0.1252, -11.0062,   7.2564],
        [  0.9809,  -4.8145,   6.8706],
        [  6.0031,  12.2958,  -6.5992],
        [  6.3549,   0.7096,   6.9198],
        [ -2.1817,   7.8274,  -4.6298],
        [ 11.6165,  -3.8715,   6.4594],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -7.4733,   4.7555,   6.2490],
        [  8.3791,  10.5409,   5.5590],
        [  5.3602,  -2.3808,   7.2098],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -9.3331,  -2.6465,  -8.2028],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.4731,  -0.7120,  -6.0271],
        [  6.3200,   0.2690,   1.5900],
        [ 11.4703,  -4.2006,   7.4086],
        [  0.9809,  -4.8145,   6.8706],
        [ -0.5739,   7.9056,   0.6382],
        [  5.8617,   3.4853,  -9.8080],
        [  6.3236,   0.3011,  -4.6668],
        [  3.1892, -10.8220,  -6.8409],
        [  4.5954,  13.0099,  -4.4439],
        [  4.1537,   5.9340,  -6.0423],
        [ 10.0400,   8.6922,  -6.2847],
        [  3.4604,  -4.1545,   1.8091],
        [ 11.6019,   5.9540,  -5.1715],
        [  9.6816,   9.1536,  -5.8543],
        [ 11.1469,   6.9237,   1.6160],
        [ -5.6423,   6.4537,  -1.2091],
        [  1.7659,   7.4127,   4.7871],
        [  2.7420,  -4.4896,  -3.7139],
        [ 12.5717,   1.9266,   7.4903],
        [ -6.4822,  -8.0252,  -3.8983],
        [  3.3474,  -4.2162,   5.2455],
        [  1.1350,  14.0104,   0.6321],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.1793,   4.7461,   0.3899],
        [ 10.1798,   8.5002,  -5.8893],
        [ -7.3333,   4.9246,  -7.9212],
        [  1.0000,   7.6591,  -4.5011],
        [  0.1041, -11.0045,   2.4311],
        [ -0.8675, -10.8815,   6.8561],
        [  5.1890,   4.7318,   0.9515],
        [  6.0310,  12.2795,  -6.4528],
        [ -8.0750,  -6.0072,  -5.8267],
        [  9.3424,  -7.3685,  -3.6807],
        [ -9.1178,   1.5784,   2.1410],
        [  1.2273,  -4.8081,   9.8129],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.4808,   4.7462,   2.0266],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 10.9916,   7.2133,  -0.9179],
        [  1.6954,  -4.7616,   9.8198],
        [  2.6970,   6.9826,   9.7350],
        [ 11.2209,  -4.7092,   2.6117],
        [ -5.0550,   6.8231,  -9.2502],
        [  2.5813,  -4.5471,   5.3597]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.9337e-02, 7.1477e-02,  ..., 6.2419e-01, 7.8115e-01,
         6.5840e-01],
        [5.9337e-02, 2.1453e-04, 2.0252e-02,  ..., 6.8117e-01, 7.8332e-01,
         7.1765e-01],
        [7.1477e-02, 2.0252e-02, 0.0000e+00,  ..., 6.9560e-01, 8.0170e-01,
         7.2867e-01],
        ...,
        [6.2419e-01, 6.8117e-01, 6.9560e-01,  ..., 0.0000e+00, 8.4697e-01,
         1.5696e-01],
        [7.8115e-01, 7.8332e-01, 8.0170e-01,  ..., 8.4697e-01, 2.1453e-04,
         9.9390e-01],
        [6.5840e-01, 7.1765e-01, 7.2867e-01,  ..., 1.5696e-01, 9.9390e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.99276564,  9.13507321,  9.17016857,  9.17595899,  9.24501829,
        9.3109884 ,  5.91854397,  9.99211663,  6.5618968 ,  7.17789666,
        7.13501264, 11.72802491,  8.8353674 , 10.8207297 ,  6.2134961 ,
        9.09502385, 13.89671314,  9.12220271, 11.56146873, 11.28093507,
        5.79562691, 11.05413545,  8.56889738,  5.2287058 ,  7.24677664,
       11.04953767, 11.36089355, 13.38091121, 13.60275211,  6.1463896 ,
        9.27474386, 12.05300943, 13.32645552,  7.60529892, 11.2845608 ,
       10.1979669 ,  8.09899874, 12.0495521 ,  7.02253235,  5.88782913,
        7.66377554,  5.94293185,  6.00732193, 10.46238054,  4.87083348,
       13.744387  , 11.00694688,  4.91336994, 13.6829843 ,  6.39438044,
        8.12580976, 12.24467041, 12.26997165, 12.1460931 ,  8.8580671 ,
       13.46553623,  5.86519525,  9.93553424,  9.70108409,  5.18047638,
       12.3344536 ,  9.4997987 ,  6.32572323, 12.21532282,  4.91336994,
        7.92644746,  6.81961996,  6.33075682, 11.28215429, 13.79762415,
        7.24326992, 13.27994304,  5.40689332, 13.04051744, 13.32374056,
       13.12218013,  8.5724042 ,  7.62011734,  5.26067019, 12.71843462,
       10.31613269,  5.38342077, 14.05633441,  9.78286733,  9.69629   ,
        7.02496223, 13.26209545,  8.83341608,  7.72415048, 11.0050379 ,
       10.91601655,  7.02253235, 13.68063098, 10.06437598, 11.89855267,
        9.25336371,  4.96231532, 11.9182956 ,  8.859416  , 12.4980845 ,
       13.1471313 ,  5.05444288,  7.48538905, 12.16901615,  8.49161154,
        5.2287058 ])}
{'P': tensor([[1.3078e-02, 2.3191e-01, 2.2950e-01,  ..., 3.8034e-04, 3.3920e-06,
         1.7043e-05],
        [3.8729e-04, 2.8976e-02, 8.3586e-04,  ..., 1.8046e-03, 8.2690e-06,
         2.3804e-05],
        [2.0183e-04, 4.7576e-01, 1.4413e-02,  ..., 8.8760e-04, 4.1284e-06,
         1.2037e-05],
        ...,
        [9.4331e-06, 1.3814e-05, 1.3688e-05,  ..., 6.3902e-02, 2.0146e-04,
         1.0706e-05],
        [1.7474e-04, 1.2923e-04, 1.3175e-04,  ..., 3.4171e-05, 9.3413e-02,
         8.7534e-03],
        [3.0883e-02, 2.0935e-02, 2.1485e-02,  ..., 2.8021e-04, 1.5669e-04,
         1.0660e-01]]), 'X': tensor([[ -8.1905,   3.7253,   4.1175],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.2707,   0.9661,   4.4863],
        [ -9.0961,   1.6529,   3.1553],
        [  6.2036,   2.4030,  -1.2500],
        [ -6.3252,  -8.1760,  -1.0891],
        [  1.3705,  13.9723,   7.9266],
        [  0.5175,  14.0910,   4.6540],
        [  9.6816,   9.1536,  -5.8543],
        [  4.7482, -10.3907,   5.9149],
        [ -3.1204, -10.2345,   9.2288],
        [  6.3604,   0.9042,  -6.5901],
        [  9.1409,  -7.5789,   1.5814],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.4524,  -1.6813,  -6.6862],
        [  6.3087,   1.8053,  -1.9481],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -2.6735, -10.4062,  -1.5911],
        [ -6.8618,  -7.6322,   3.7027],
        [  5.9912,   3.1453,  -3.9989],
        [  8.9284,  10.0036,  -6.9210],
        [ -3.5173,   7.5091,  -8.9585],
        [ 10.5057,  -5.9099,  -5.1599],
        [  1.2273,  -4.8081,   9.8129],
        [ -7.5603,   4.6459,   8.2002],
        [  0.3231,  -4.7691,  -1.6120],
        [ 10.9931,   7.2107,   7.7933],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.3453,  -1.5462,   7.0212],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.2790,  -0.0259,  -5.4116],
        [ -7.8126,   4.3066,   5.1791],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.6319,  11.9066,   3.5165],
        [ -7.3554,   4.8985,  -0.6665],
        [  5.9843,  -9.8679,  -6.1449],
        [  9.3753,   9.5171,   3.9131],
        [  5.3118,   4.5452,   4.2114],
        [  5.9003,  12.3550,   2.1917],
        [  3.6817,   6.3354,   6.1233],
        [  0.9478,  -4.8144,   0.6371],
        [ -3.5903,   7.4847,  -8.9112],
        [ 11.0275,  -5.0666,   8.4154],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.8994,   6.8675,  -9.5987],
        [ -7.7239,   4.4298,   3.1216],
        [ 12.4597,  -0.9136,  -4.8763],
        [  1.6308,  13.9254,  -2.0984],
        [  4.7544,  -3.1425,  -1.5315],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.4013,  -1.2597,  -7.6474],
        [  4.4290,  -3.4596,   1.4181],
        [  2.5010,  -4.5736,  -0.8989],
        [  0.4996,   7.7738,  -4.4711],
        [ -2.3317,   7.8034,  -3.6274],
        [ 10.7566,   7.6212,  -2.3279],
        [  4.5954,  13.0099,  -4.4439],
        [  9.4516,  -7.2501,  -1.8868],
        [ -8.8400,   2.3991,   1.3457],
        [  0.1460,   7.8340,  -9.4614],
        [  1.1139,   7.6281,  -2.5680],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.2022,   2.4092,   3.0349],
        [ 10.5678,   7.9261,  -0.6100],
        [ 12.5996,   1.3971,  -1.0694],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.0611,  -6.0302,   4.6962],
        [  1.8719,   7.3713,  -3.7465],
        [  4.3377,   5.7570,   1.3712],
        [ -4.4503,   7.1378,   3.3751],
        [  0.9478,  -4.8144,   0.6371],
        [ -6.4464,   5.8272,   7.5038],
        [  2.4149,  -4.6005,  -4.3339],
        [ 11.0864,   7.0387,  -0.1238],
        [  4.2041,  -3.6509,  -4.9488],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.8474,   6.8979,   6.6457],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.2339,  -3.6268,   6.3125],
        [ -5.6169,  -8.7844,   8.9057],
        [  4.4633,   5.6287,  -0.7754],
        [  6.3516,   0.6425,   0.3425],
        [  2.1477,   7.2550,   8.9688],
        [  0.2480, -11.0153,  -3.0183],
        [ -7.3333,   4.9246,  -7.9212],
        [  9.9217,  -6.7006,  -5.6728],
        [  1.0093,   7.6567,   2.3526],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.2038,   2.4020,  -3.1692],
        [  4.4039,   5.6903,   8.7366],
        [ 12.2914,   3.8037,   6.0978],
        [ -5.7347,  -8.6908,   9.2878],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -3.1671,   7.6158,   6.0765],
        [  1.1274,  14.0116,   8.1753],
        [  5.4242, -10.1261,   3.9595],
        [  5.4336,  -2.2676,   5.6933],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.2630,   2.1057,  -5.2136],
        [ -1.2454,   7.9127,  -0.5715]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0828, 0.0717,  ..., 0.6482, 0.9270, 0.5446],
        [0.0828, 0.0000, 0.0121,  ..., 0.5928, 0.9670, 0.5884],
        [0.0717, 0.0121, 0.0000,  ..., 0.5964, 0.9573, 0.5782],
        ...,
        [0.6482, 0.5928, 0.5964,  ..., 0.0000, 0.9480, 0.7156],
        [0.9270, 0.9670, 0.9573,  ..., 0.9480, 0.0000, 0.3829],
        [0.5446, 0.5884, 0.5782,  ..., 0.7156, 0.3829, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 8.99791316,  9.17595899,  9.17016857,  9.13507321,  9.32094055,
        9.24501829,  6.652742  , 10.33712258, 14.0393949 , 14.10045797,
       13.32374056, 11.42420508, 10.69963954,  6.42439344, 11.87412439,
        8.94395628,  9.60080639,  6.5618968 , 12.05395056, 10.74409834,
       10.26327895,  6.76661595, 13.40850328,  8.29203356, 12.05395056,
        4.96231532,  8.8737556 ,  4.78002665, 13.14690137, 13.41713661,
       12.44177641,  7.822854  ,  9.69629   ,  8.85572292,  6.24187044,
        6.27905292,  8.92098752,  8.05242591, 13.62896593,  8.83725081,
       11.54071859, 13.35927527,  6.99097013, 13.69162795,  7.32754882,
        4.90676242,  8.30125201, 12.13568598,  8.89458273,  7.4544866 ,
        8.90405823, 12.49317563, 14.02058621,  5.69913817, 12.5769659 ,
       12.46513646,  5.62005629,  5.21280156,  7.78980299,  8.14433542,
       13.18278606, 13.79762415, 11.9120321 ,  9.15976746,  7.83534987,
        7.70897675,  9.66977802,  6.65368551, 13.20988996, 12.67680439,
        9.11473391, 12.62576026, 10.06703358,  7.60529892,  7.20826219,
        8.41148722,  4.90676242,  8.68979052,  5.19577146, 13.13205357,
        5.56809335, 13.04217487,  7.46248947, 13.01728313,  5.5748624 ,
       10.42667093,  7.18354885,  6.38399789,  7.56617037, 11.018084  ,
        8.83341608, 11.97238624,  7.72291946, 10.06407839,  6.65259775,
        7.19534146, 12.86648771, 10.412299  ,  9.562174  ,  8.24808832,
       14.05687815, 11.48734978,  5.88782913, 10.39942497,  6.60752373,
        8.01009083])}
{'P': tensor([[1.2135e-02, 2.2398e-01, 1.5447e-01,  ..., 5.5588e-05, 1.3592e-07,
         5.5583e-05],
        [3.0588e-04, 2.6409e-02, 1.3772e-03,  ..., 1.4678e-04, 4.3616e-07,
         5.7541e-05],
        [2.8280e-04, 1.6596e-01, 1.4055e-02,  ..., 9.6551e-05, 2.3963e-07,
         3.9053e-05],
        ...,
        [2.6399e-05, 5.2411e-05, 5.2263e-05,  ..., 1.1977e-01, 7.2668e-05,
         4.5264e-05],
        [2.5619e-07, 1.9030e-07, 2.2959e-07,  ..., 5.6632e-07, 1.6134e-02,
         5.3350e-06],
        [8.0971e-03, 1.0850e-02, 1.3620e-02,  ..., 4.9776e-03, 2.5841e-06,
         6.9037e-02]]), 'X': tensor([[ -8.3653,   3.4184,   9.3454],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4196,   0.0478,   9.4140],
        [ -8.9636,   2.0648,   8.1946],
        [  2.5813,  -4.5471,   5.3597],
        [  1.2728, -11.0372,   9.8121],
        [ -8.5900,   2.9767,   7.4811],
        [ 11.1420,   6.9331,  -2.4779],
        [ 12.2218,   4.0953,  -8.2072],
        [  7.8425,  -8.7202,   6.4530],
        [  3.8111, -10.6787,  -4.6539],
        [  1.1350,  14.0104,   0.6321],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.1203,  -2.7151,   6.2110],
        [ -5.1369,   6.7755,   8.3836],
        [  2.4697,  -4.5836,   1.4593],
        [ -0.5709,   7.9055,   7.6089],
        [ -4.0409,   7.3169,  -3.5779],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 10.5678,   7.9261,  -0.6100],
        [  2.3261,   7.1726,   6.2522],
        [  2.6451,  -4.5250,  -0.5565],
        [ -7.5603,   4.6459,   8.2002],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.4430,  -0.1897,   4.8383],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.5620,   7.0549,   8.4825],
        [ -4.5778,  -9.5013,  -5.0667],
        [  4.1999,  -3.6543,  -4.1998],
        [  5.6928,  12.4710,   3.6001],
        [ -1.0012, -10.8576,   2.2867],
        [  3.6249, -10.7255,   5.6605],
        [  3.0876,   6.7527,  -5.3540],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.0604,   1.7707,   1.1066],
        [ -8.6675,  -4.8569,  -9.1769],
        [  3.6817,   6.3354,   6.1233],
        [  2.6310, -10.9195,   3.8054],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.3767, -11.0341,   9.9415],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.4867,  12.5816,   2.9549],
        [  6.3569,   0.7591,   6.5011],
        [ 12.2218,   4.0953,  -8.2072],
        [ -1.9714,   7.8563,  -1.8084],
        [  5.9883,   3.1536,   6.2320],
        [  6.1737,   2.5305,   3.9049],
        [ -6.8618,  -7.6322,   3.7027],
        [ -6.7992,   5.4991,   9.9186],
        [  7.3724,  -9.0577,  -8.9843],
        [ -1.0012, -10.8576,   2.2867],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -5.6174,   6.4708,  -1.7339],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.4604,  -4.1545,   1.8091],
        [  5.0233,   4.9647,  -3.1788],
        [  0.4119,  14.1019,  -4.1870],
        [  2.4639, -10.9431,  -7.3718],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.4426, -10.1181,  -5.1268],
        [  2.2076,  13.8032,  -1.5099],
        [  6.0560,  -0.9421,  -5.0334],
        [ -8.9282,   2.1648,   8.4894],
        [  3.3744,   6.5623,  -6.3636],
        [ 12.2592,   3.9425,   2.0383],
        [  4.1348,   5.9514,  -0.8856],
        [  3.6703,   6.3443,  -5.4401],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.8323,  -1.5228,   5.1132],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.1426,   3.8048,   9.3542],
        [  5.8979,   3.3958,  -3.8713],
        [ 12.4837,   2.7510,   0.4449],
        [  5.6314,  -1.9302,  -9.6543],
        [  6.1929,   2.4501,  -6.9352],
        [  2.3261,   7.1726,   6.2522],
        [  5.5101,  -2.1433,   9.5442],
        [ -6.3306,   5.9272,   5.6935],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.5199,  -0.4711,   3.0436],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.0604,   1.7707,   1.1066],
        [  4.9152,  12.8653,  -1.4459],
        [  9.3607,   9.5337,  -4.5412],
        [  3.3069,   6.6089,   8.7377],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4430,  -0.1897,   4.8383],
        [ -8.8835,  -4.3178,   9.3080],
        [  2.4572,   7.1085,   7.7376],
        [ -0.4341, -10.9473,   7.4509],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.1787,  -2.6383,  -4.2533],
        [ 10.0567,   8.6697,   4.6329],
        [  0.1382, -11.0073,   1.0026],
        [ -5.0550,   6.8231,  -9.2502],
        [ -9.1907,  -3.3180,  -5.5943],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -5.6651,   6.4380,   1.3660]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0602, 0.0393,  ..., 1.0420, 1.3423, 0.5217],
        [0.0602, 0.0000, 0.0253,  ..., 1.0176, 1.3673, 0.5388],
        [0.0393, 0.0253, 0.0000,  ..., 1.0159, 1.3473, 0.5208],
        ...,
        [1.0420, 1.0176, 1.0159,  ..., 0.0000, 1.0214, 0.7145],
        [1.3423, 1.3673, 1.3473,  ..., 1.0214, 0.0000, 0.8334],
        [0.5217, 0.5388, 0.5208,  ..., 0.7145, 0.8334, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.0368458 ,  9.23071698,  9.18690018,  9.27474386,  9.41970031,
        9.19837792,  5.2287058 , 11.11038551,  9.09119351, 13.12299145,
       12.889695  , 11.72802491, 11.33837396, 14.05633441,  9.9353921 ,
        5.79562691,  8.50268322,  5.20659892,  7.92607342,  8.35856487,
        9.97964013, 13.20988996,  7.5403822 ,  5.24137909,  8.8737556 ,
       12.46513646,  9.66977802,  9.44486609,  9.78286733,  7.50564223,
       10.54658597,  5.5671389 , 13.70893492, 10.9036226 , 11.32149259,
        7.42512621, 13.77075174,  9.23178204,  9.93553424,  7.32754882,
       11.23201619, 13.12299145, 11.11969815,  9.562174  ,  9.20340031,
       13.72594022,  6.4020404 , 12.889695  ,  8.09984231,  6.76791483,
        6.67218687, 10.26327895,  8.74470218, 11.67875855, 10.9036226 ,
       10.4977132 ,  8.56889738, 12.52372629,  5.40689332,  7.06272256,
       14.10796466, 11.21703654,  6.80559669, 13.14777433, 11.4890861 ,
       13.97857591,  6.12886024,  9.18690018,  7.37902766, 12.87752052,
        7.24677664,  7.32950385, 10.63038565,  6.12886024,  6.02779201,
        5.35774573,  8.98765644,  6.80559669, 12.78327037,  5.95297459,
        6.65991877,  7.5403822 ,  5.91222346,  8.67227386,  4.83371341,
       12.52875793,  5.1224079 ,  9.5642993 ,  6.53290663,  9.23178204,
       13.77222752, 13.36092624,  7.39004226,  9.24501829,  9.44486609,
        9.87720102,  7.52117132, 10.95593729, 12.16901615,  5.81199422,
       13.2778378 , 11.00812996,  8.49161154,  9.7712321 , 12.27835996,
        8.57560275])}
{'P': tensor([[5.8835e-03, 1.7232e-01, 1.2816e-01,  ..., 2.9701e-06, 3.8970e-05,
         1.6605e-05],
        [1.4817e-04, 8.9377e-03, 1.5869e-01,  ..., 3.5258e-06, 3.7493e-05,
         3.2508e-05],
        [1.6488e-04, 4.0579e-04, 1.0137e-02,  ..., 3.5407e-06, 3.5776e-05,
         3.8366e-05],
        ...,
        [1.5932e-03, 1.2124e-03, 1.0158e-03,  ..., 4.2258e-02, 3.3442e-02,
         1.7438e-05],
        [5.4860e-02, 4.9610e-02, 4.2395e-02,  ..., 2.8553e-04, 8.0943e-02,
         1.4002e-04],
        [1.8761e-06, 1.9294e-06, 1.9007e-06,  ..., 1.4425e-05, 3.7649e-06,
         4.3716e-02]]), 'X': tensor([[ -8.3900,   3.3727,   3.1650],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2508,   1.0572,   2.9978],
        [ -5.3736,   6.6310,  -1.5174],
        [  4.5340,  -3.3629,   8.9652],
        [  5.1890,   4.7318,   0.9515],
        [ -9.4716,  -0.6673,   7.6068],
        [ 11.0438,   7.1180,  -5.6991],
        [  9.0424,  -7.6779,  -7.5286],
        [  4.9337,  12.8566,  -3.6582],
        [  0.9967,  -4.8144,  -9.7370],
        [  1.6308,  13.9254,  -2.0984],
        [  4.2041,  -3.6509,  -4.9488],
        [ 12.3609,   3.4769,  -8.5762],
        [ -3.6381,   7.4683,   7.3544],
        [  6.1579,  -0.5961,  -6.5595],
        [  4.3377,   5.7570,   1.3712],
        [  6.8704,  -9.3809,   7.5769],
        [  0.1382, -11.0073,   1.0026],
        [  5.3804,  -2.3504,   7.2744],
        [  6.2038,   2.4020,  -3.1692],
        [ 12.2592,   3.9425,   2.0383],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.3444,   0.5821,   6.9101],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.0449,  13.8402,  -7.1642],
        [  6.2905,   1.9366,   2.1169],
        [ -9.3616,  -2.4749,   0.1793],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.1989,  -9.7591,   0.8289],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -8.8154,   2.4611,   2.5198],
        [  2.4697,  -4.5836,   1.4593],
        [  1.4150,  13.9647,  -4.0915],
        [ 11.9695,  -2.9407,  -6.3498],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.2199,   4.6861,  -1.9304],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.1472,  -2.6801,   8.1194],
        [ -5.3382,   6.6533,   7.8054],
        [  9.6662,   9.1725,   8.1056],
        [  5.9655,   3.2172,  -7.5953],
        [  9.3753,   9.5171,   3.9131],
        [  3.2701,   6.6338,   9.9209],
        [ 12.5770,   1.8512,  -0.6578],
        [  1.1350,  14.0104,   0.6321],
        [  7.4132,  11.3526,   0.3322],
        [ -0.8675, -10.8815,   6.8561],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.1231,  -3.5739,   8.4708],
        [ -4.0409,   7.3169,  -3.5779],
        [ 10.6673,  -5.6649,  -4.1646],
        [  1.8447,  -4.7372,   8.1235],
        [ -0.8569,   7.9156,  -6.5557],
        [  5.7802,  -1.6363,  -6.2782],
        [  9.8082,  -6.8396,   4.9615],
        [ -3.2053, -10.1993,  -3.8736],
        [ -5.1377,   6.7751,  -0.4438],
        [ -1.5875,   7.8944,  -9.4286],
        [ -8.0965,  -5.9712,   4.3466],
        [  5.6248,   3.9984,   8.4933],
        [  6.0955,   2.8210,  -4.3321],
        [ 11.1420,   6.9331,  -2.4779],
        [ -0.8675, -10.8815,   6.8561],
        [  2.0449,  13.8402,  -7.1642],
        [ -6.0195,   6.1789,  -2.9496],
        [  3.1892, -10.8220,  -6.8409],
        [ -5.9225,  -8.5355,   2.4778],
        [ -0.5739,   7.9056,   0.6382],
        [  4.9152,  12.8653,  -1.4459],
        [  9.0129,  -7.7072,   9.1400],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -3.1671,   7.6158,   6.0765],
        [  4.6063,  13.0051,  -3.1456],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.7544,   2.6096,   3.3146],
        [  1.4150,  13.9647,  -4.0915],
        [ 11.3800,  -4.3919,   1.4066],
        [ -7.0825,   5.2073,  -5.0497],
        [ 12.3357,   3.6001,   0.7903],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.3505,   0.6235,  -8.5849],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.6202,  -1.9508,   0.7471],
        [ -9.4430,  -0.1897,   4.8383],
        [ 11.3800,  -4.3919,   1.4066],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.2860,   7.1916,  -6.1011],
        [  0.2480, -11.0153,  -3.0183],
        [ -3.2736, -10.1703,   3.8712],
        [ -2.3317,   7.8034,  -3.6274],
        [ -8.0750,  -6.0072,  -5.8267],
        [ 12.1588,   4.3364,  -8.0867],
        [  7.0171,  11.6439,  -5.0178],
        [ -1.0407,   7.9166,  -6.6021],
        [ 10.9931,   7.2107,   7.7933],
        [  5.8450,  -9.9356,   3.0767],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.1497,   7.8334,  -5.4412],
        [ -5.6174,   6.4708,  -1.7339],
        [ -0.8455, -10.8853,  -6.7400]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.0571e-02, 5.8893e-02,  ..., 7.4625e-01, 4.1792e-01,
         7.3680e-01],
        [4.0571e-02, 2.1453e-04, 2.0252e-02,  ..., 7.8022e-01, 4.4747e-01,
         7.5012e-01],
        [5.8893e-02, 2.0252e-02, 0.0000e+00,  ..., 8.0047e-01, 4.6739e-01,
         7.6604e-01],
        ...,
        [7.4625e-01, 7.8022e-01, 8.0047e-01,  ..., 0.0000e+00, 3.4580e-01,
         4.8726e-01],
        [4.1792e-01, 4.4747e-01, 4.6739e-01,  ..., 3.4580e-01, 0.0000e+00,
         4.1206e-01],
        [7.3680e-01, 7.5012e-01, 7.6604e-01,  ..., 4.8726e-01, 4.1206e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.04255655,  9.13507321,  9.17016857,  9.17595899,  9.24501829,
        9.3109884 ,  8.53501777,  5.64500512,  7.02253235,  9.49511524,
       13.13888841, 11.86239961, 13.77075174,  4.91652556, 14.02058621,
        5.56809335, 12.84056604,  8.30729352,  6.18668463,  7.20826219,
       11.62770332, 11.00812996,  5.87132718,  6.65259775, 12.87752052,
       13.27717982,  9.36256096, 11.99633819, 13.99048094,  6.58184474,
        9.68323686, 12.35421461, 11.56146873, 10.52587243,  9.15253344,
        5.20659892, 14.03618649, 12.32545637, 11.80092578,  9.7712321 ,
        7.01475376,  9.75200848,  5.80312949,  8.53014915, 13.32557133,
        6.77777606, 13.35927527,  7.39599986, 12.7125079 , 14.05633441,
       13.55868684, 10.91601655,  9.75200848,  9.79814997,  8.35856487,
       12.07818871,  5.08372149,  7.96181288,  6.00732193, 11.95743558,
       10.6910768 ,  8.50278896,  8.05242591, 10.06023665,  6.90115901,
        6.7166297 , 13.12299145, 10.91601655, 13.99048094,  8.62631801,
       11.28215429, 10.38897563,  7.92644746, 13.77222752, 11.85890664,
       10.60226614,  8.24808832, 13.79676654,  5.53255882,  6.02588517,
        9.13507321, 14.03618649, 12.19805394,  8.7907923 , 12.85032836,
       12.10942412,  6.38105015,  8.89458273,  5.94909709,  9.44486609,
       12.19805394, 12.40074901,  7.54620849, 11.018084  , 10.68416401,
        8.14433542, 10.06437598, 12.9089559 , 13.59480906,  7.98469166,
       13.14690137, 11.52733581, 10.24557047,  7.83486943,  8.56889738,
       10.91805683])}
{'P': tensor([[1.7372e-02, 2.0174e-01, 1.8412e-01,  ..., 9.6179e-07, 1.0929e-04,
         2.4369e-06],
        [1.5370e-04, 1.5904e-02, 2.6030e-04,  ..., 4.3759e-07, 1.4635e-04,
         3.2351e-06],
        [1.1589e-03, 1.7148e-01, 2.1584e-02,  ..., 1.0552e-06, 1.5566e-04,
         2.7138e-06],
        ...,
        [2.3565e-04, 6.4209e-05, 1.9852e-04,  ..., 4.4444e-03, 6.5006e-07,
         1.0096e-07],
        [1.1919e-04, 9.2034e-05, 1.1798e-04,  ..., 1.4056e-04, 1.0065e-01,
         1.6213e-04],
        [1.8429e-05, 1.4760e-05, 1.4545e-05,  ..., 3.4067e-05, 1.8537e-05,
         4.8409e-02]]), 'X': tensor([[ -8.4573,   3.2448,   5.4370],
        [ -9.2707,   0.9661,   4.4863],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.4430,  -0.1897,   4.8383],
        [ -9.3444,   0.5821,   6.9101],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.5408,   7.8978,   5.3470],
        [  6.1685,  12.1979,  -6.2473],
        [  1.7218,  -4.7576,  -3.3617],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.1252, -11.0062,   7.2564],
        [  9.1409,  -7.5789,   1.5814],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -4.3963,   7.1629,   4.5263],
        [  6.1372,   2.6726,  -6.7885],
        [  5.9526,   3.2525,  -4.3821],
        [  0.9945,  -4.8144,  -0.1897],
        [  3.9022,  13.2906,   5.5788],
        [  3.2990,  -4.2415,  -7.4857],
        [  2.5318,  13.7231,   6.3692],
        [  0.5960, -11.0335,   3.1144],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.9526,   3.2525,  -4.3821],
        [  6.0073,   3.0986,   6.7937],
        [  2.3373,  13.7721,  -0.2442],
        [ 10.1503,   8.5414,  -7.2681],
        [ -4.4520,   7.1370,  -8.7867],
        [ -9.0574,   1.7801,   8.8986],
        [ -7.7842,   4.3466,  -1.7548],
        [ 10.1503,   8.5414,  -7.2681],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.2644,  -4.6434,   9.7508],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3534,   1.2990,  -7.2942],
        [ 10.9931,   7.2107,   7.7933],
        [ -8.4297,  -5.3637,  -9.3962],
        [  1.1994,   7.6035,   2.4295],
        [  0.1382, -11.0073,   1.0026],
        [ -4.0409,   7.3169,  -3.5779],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.0729,   7.2878,   7.0880],
        [ -7.9298,   4.1367,  -3.9773],
        [ -5.1377,   6.7751,  -0.4438],
        [  5.3602,  -2.3808,   7.2098],
        [ -8.0608,   3.9362,  -5.9567],
        [ 10.6283,  -5.7253,   7.4878],
        [ -6.3489,   5.9117,   6.5161],
        [ 12.4877,  -0.7222,   8.8347],
        [ -3.0008, -10.2827,  -8.9480],
        [  3.7298,  13.3537,  -9.4653],
        [ -2.2188,   7.8218,   8.6064],
        [  6.1741,  -0.5329,   3.1913],
        [  4.0587, -10.6112,   6.2273],
        [ -3.6381,   7.4683,   7.3544],
        [  8.5093,  -8.1764,  -3.2381],
        [ -4.1296,   7.2803,  -2.7194],
        [ -9.4501,  -1.7133,   2.2980],
        [  3.3744,   6.5623,  -6.3636],
        [ -3.5029,   7.5138,   8.1853],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.1085,  -4.9204,   0.0736],
        [  4.4039,   5.6903,   8.7366],
        [ 12.5717,   1.9266,   7.4903],
        [  0.9967,  -4.8144,  -9.7370],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3604,   0.9042,  -6.5901],
        [  2.5318,  13.7231,   6.3692],
        [  3.7298,  13.3537,  -9.4653],
        [  9.5018,  -7.1945,  -1.1892],
        [  6.2022,   2.4092,   3.0349],
        [  3.8182,  -3.9347,  -1.7961],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.1274,  14.0116,   8.1753],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 12.5717,   1.9266,   7.4903],
        [ -4.8751,  -9.3148,   4.0406],
        [ -9.1960,  -3.2964,   1.6921],
        [ -5.0943,  -9.1682,   0.5073],
        [ -5.9940,   6.1985,   5.9566],
        [ -7.4807,   4.7463,   2.0616],
        [  4.2612, -10.5514,  -6.6268],
        [  6.1989,  -9.7591,   0.8289],
        [  5.6022,  -1.9833,   8.3081],
        [ -9.3604,   0.4855,  -3.0158],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.6093,  -1.9705,   6.4807],
        [ -0.9715, -10.8630,   0.3555],
        [  6.3542,   0.6934,  -3.3840],
        [  5.7802,  -1.6363,  -6.2782],
        [ -8.8934,   2.2597,   3.5831],
        [  6.3569,   0.7591,   6.5011],
        [  3.6817,   6.3354,   6.1233],
        [ -2.5161,   7.7699,  -9.3531],
        [ -4.9217,   6.8979,  -1.7667],
        [ -0.5385,   7.9037,   6.3830],
        [  9.3836,   9.5075,  -5.9278],
        [ -9.3616,  -2.4750,   3.7765],
        [  6.8093,  -9.4178,   8.4244],
        [  4.7156,  -3.1831,  -8.0224],
        [  0.4119,  14.1019,  -4.1870],
        [  1.9312,  13.8649,   8.3563],
        [  2.9260,  -4.4162,   8.3808],
        [  6.3301,   0.3638,  -7.5685]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1599e-01, 3.7662e-02,  ..., 1.0755e+00, 7.1680e-01,
         1.0789e+00],
        [1.1599e-01, 0.0000e+00, 1.2027e-01,  ..., 1.1881e+00, 8.2289e-01,
         1.1002e+00],
        [3.7662e-02, 1.2027e-01, 2.1453e-04,  ..., 1.0872e+00, 7.3541e-01,
         1.1164e+00],
        ...,
        [1.0755e+00, 1.1881e+00, 1.0872e+00,  ..., 0.0000e+00, 3.9664e-01,
         1.2020e+00],
        [7.1680e-01, 8.2289e-01, 7.3541e-01,  ..., 3.9664e-01, 0.0000e+00,
         9.1096e-01],
        [1.0789e+00, 1.1002e+00, 1.1164e+00,  ..., 1.2020e+00, 9.1096e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.05843201,  9.32094055,  9.10945678,  9.44486609,  9.36256096,
        9.53015376,  8.04664886, 13.66896739,  5.05963217, 10.30972193,
       11.00694688, 11.87412439,  9.69792513,  9.5642993 ,  8.40445872,
        6.69389431,  6.78325151,  4.9160962 , 13.85158417,  5.37343173,
       13.95472331, 11.04953767, 12.91505111,  6.78325151,  6.7593917 ,
       13.96905756, 13.26590071,  8.41171369,  9.23071698,  8.91551231,
       13.26590071,  8.23525433,  5.16611782,  5.76165934,  6.48486171,
       13.14690137,  9.99145096,  7.69752262, 11.00812996,  8.35856487,
        8.09704052,  9.50654967,  8.56889738,  7.57685908,  8.94395628,
        8.50278896,  5.86519525,  8.97052115, 12.07225469,  8.67502348,
       12.5085995 , 10.71162681, 13.86480175,  8.1303945 ,  6.19707944,
       11.36089355,  8.30729352, 11.80092578,  8.36995953,  9.60412735,
        7.37902766,  8.29022197, 13.79762415, 12.1494023 ,  7.19534146,
       12.71843462,  4.91652556, 10.8207297 ,  6.42439344, 13.95472331,
       13.86480175, 11.9182956 ,  6.65368551,  5.48276038,  9.66977802,
       14.05687815,  9.62958973, 12.71843462, 10.51340829,  9.76896904,
       10.48840476,  8.62260327,  8.85939626, 11.37939695, 11.56146873,
        5.94293185,  9.37296102, 13.00380624,  5.94535529, 10.90638066,
        6.39187285,  6.00732193,  9.17595899,  6.4020404 ,  7.32754882,
        8.16714836,  8.47371868,  7.92200342, 13.35832725,  9.68324553,
       11.62158554,  5.68942572, 14.10796466, 13.99877413,  5.29755487,
        6.34059021])}
{'P': tensor([[1.1004e-02, 3.4673e-01, 1.5492e-01,  ..., 4.4745e-06, 1.1651e-05,
         1.0700e-06],
        [2.9192e-04, 1.8820e-02, 3.3021e-01,  ..., 8.3132e-06, 2.2240e-05,
         2.1296e-06],
        [4.9614e-04, 7.7588e-04, 6.0949e-02,  ..., 3.9768e-05, 1.1498e-04,
         1.0004e-05],
        ...,
        [1.9212e-05, 1.6803e-05, 1.5983e-05,  ..., 6.5195e-02, 1.6285e-03,
         6.3574e-04],
        [1.9281e-05, 1.8468e-05, 2.1222e-05,  ..., 1.4013e-01, 8.9037e-02,
         6.4524e-04],
        [1.6202e-06, 1.2569e-06, 8.1973e-07,  ..., 3.3573e-05, 4.6627e-06,
         1.0794e-02]]), 'X': tensor([[-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0438, 0.1101,  ..., 1.0715, 1.0507, 0.7196],
        [0.0438, 0.0000, 0.0697,  ..., 1.0980, 1.0751, 0.7634],
        [0.1101, 0.0697, 0.0000,  ..., 1.1118, 1.0849, 0.8255],
        ...,
        [1.0715, 1.0980, 1.1118,  ..., 0.0000, 0.0686, 0.8740],
        [1.0507, 1.0751, 1.0849,  ..., 0.0686, 0.0000, 0.9025],
        [0.7196, 0.7634, 0.8255,  ..., 0.8740, 0.9025, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.09119351,  9.18358562,  9.36256096,  9.19837792,  9.27474386,
        9.41244865,  8.05668609, 13.92390689, 13.00380624, 11.63383221,
        6.48486171,  8.62631801, 11.28215429,  8.09899874,  6.29034958,
        7.5092834 ,  6.383897  , 13.744387  ,  9.88571713,  7.53884104,
        8.36445758,  7.32754882, 12.52372629,  7.85919687,  6.33075682,
        7.92200342,  7.24326992, 11.18839169, 11.48734978,  6.42689705,
       10.8992524 ,  9.41605684,  6.61211224,  5.81481044,  5.35774573,
        7.37576154,  5.47702766, 10.31613269, 11.63383221, 12.10222204,
        8.11227668, 12.24467041,  5.1805345 ,  9.53015376, 12.07818871,
        6.29034958,  6.65991877, 13.46553623,  6.10360073,  7.32233733,
        9.79814997, 11.9702762 ,  4.91296205,  7.16747878,  8.29640613,
       12.19805394,  7.02253235,  9.14973319, 10.33958639, 10.80728788,
        9.15253344, 13.01187484,  6.46560181,  8.5724042 ,  9.88310614,
       10.68416401, 12.45418236, 13.95472331, 12.38881635, 13.5247215 ,
        9.23178204,  4.78002665,  5.91222346, 12.67680439,  6.69389431,
        7.28793663, 13.46553623,  6.81084724, 12.53285633, 10.84437705,
        4.79003361,  7.50564223,  7.59795445,  5.08910491, 12.10942412,
        6.41202986,  4.90505566,  5.84533345, 13.70893492, 12.0495521 ,
       13.69162795, 12.40074901, 12.4980845 , 11.52733581,  7.72415048,
       12.27991991, 13.89886469, 12.49317563, 10.63038565,  5.62005629,
        7.33350642,  8.40445872,  6.99097013,  5.81199422,  5.1224079 ,
       12.33167382])}
{'P': tensor([[1.4338e-02, 2.5701e-01, 2.0458e-01,  ..., 1.2885e-06, 3.2447e-02,
         7.4112e-06],
        [8.1278e-04, 1.7686e-02, 2.8710e-01,  ..., 1.6173e-06, 4.5026e-02,
         9.1751e-06],
        [1.9311e-04, 2.0170e-04, 2.3132e-02,  ..., 3.3157e-06, 1.6394e-01,
         1.2141e-05],
        ...,
        [1.8225e-06, 1.7068e-06, 9.4194e-07,  ..., 2.1294e-02, 4.3537e-07,
         6.0078e-04],
        [7.9337e-05, 8.3914e-05, 1.9659e-04,  ..., 1.3735e-05, 4.1288e-02,
         2.8585e-05],
        [3.3617e-04, 3.2127e-04, 1.6098e-04,  ..., 2.3632e-05, 6.1525e-05,
         4.6444e-02]]), 'X': tensor([[ -8.6050,   2.9451,  -0.7012],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.3419,   0.5972,  -1.2841],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.4671,  -0.5553,   0.1117],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.0870,  -4.6877,   9.6823],
        [  4.7739,  -3.1218,  -4.1188],
        [  1.7122,  -4.7591,  -3.6734],
        [  4.5491,  -3.3485,   6.8412],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.4203,  -2.0340,   2.5856],
        [ -7.0948,   5.1940,   9.8171],
        [ -6.8618,  -7.6322,   3.7027],
        [ -2.6069, -10.4298,  -6.1995],
        [  6.3550,   0.7129,   0.4917],
        [ 12.0772,  -2.6014,  -7.1832],
        [  2.1824,  -4.6647,   1.1985],
        [  5.9587,  12.3216,   9.1695],
        [  7.4132,  11.3526,   0.3322],
        [ 12.5832,   1.7543,  -7.6134],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.9033,   2.2329,  -8.8912],
        [ -2.2188,   7.8218,   8.6064],
        [ 10.0619,   8.6627,  -9.2884],
        [  6.3087,   1.8053,  -1.9481],
        [ -8.5149,  -5.1905,   1.5239],
        [  0.5175,  14.0910,   4.6540],
        [  5.7151,  -1.7700,   3.1542],
        [ -7.6650,   4.5092,   0.2520],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.3587,   1.1596,  -3.6629],
        [  5.1519, -10.2387,   9.8191],
        [  0.5250, -11.0306,   5.5358],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.3373,  13.7721,  -0.2442],
        [  5.1292,   4.8183,  -5.7997],
        [  3.2809,  13.5061,  -4.0199],
        [  4.8653,   5.1694,  -4.0823],
        [ -0.9715, -10.8630,   0.3555],
        [  5.1203,  -2.7151,   6.2110],
        [ -0.0254, -10.9932,   3.2200],
        [  2.3373,  13.7721,  -0.2442],
        [ -9.4430,  -0.1897,   4.8383],
        [ -3.9700,   7.3453,  -4.1486],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.2026,  12.1774,  -1.5887],
        [  6.9314,  -9.3435,  -1.2729],
        [  4.3554,   5.7393,  -9.0766],
        [ -2.3317,   7.8034,  -3.6274],
        [  4.3523, -10.5232,   8.2138],
        [  3.9515,  -3.8425,   6.1729],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.5057,  -3.3894,  -0.8160],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -2.1533, -10.5778,   1.0112],
        [  3.9895,   6.0816,  -6.5235],
        [  1.3670,  -4.7990,   3.2603],
        [  5.6093,  -1.9705,   6.4807],
        [ -5.6423,   6.4537,  -1.2091],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.4159,  -2.0739,   9.6841],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.3031,   1.8477,  -7.1706],
        [  6.3266,   0.3292,   9.3502],
        [ -5.3179,  -9.0101,   9.4472],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.0937,  13.8293,  -2.3533],
        [ -3.5029,   7.5138,   8.1853],
        [  5.2188,  -2.5839,   2.1833],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.0419,  -4.6978,  -5.0983],
        [  7.5064,  -8.9651,  -0.0994],
        [  0.2480, -11.0153,  -3.0183],
        [ -5.0943,  -9.1682,   0.5073],
        [  3.3839,  -4.1966,   8.3574],
        [  0.4996,   7.7738,  -4.4711],
        [  6.1973,  -0.4372,   6.5720],
        [ 12.3453,  -1.5462,   7.0212],
        [  8.1043,  -8.5164,  -7.9290],
        [ -4.7389,   6.9953,   3.7638],
        [  2.2076,  13.8032,  -1.5099],
        [ -8.1204,  -5.9308,   4.0761],
        [ -7.0120,   5.2825,   7.7762],
        [ 11.6019,   5.9540,  -5.1715],
        [  1.5662,  13.9375,   4.5142],
        [ -3.5173, -10.0623,   4.8280],
        [  6.8093,  -9.4178,   8.4244],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.8902,  -1.3887,   4.0388],
        [ 11.9122,   5.1397,  -1.5498],
        [ -5.9940,   6.1985,   5.9566],
        [  5.3602,  -2.3808,   7.2098],
        [  5.0506,   4.9277,  -9.5094],
        [  1.1644, -11.0394,   7.8511],
        [ 12.6035,   0.7494,   7.1536],
        [  4.5430,   5.5438,  -3.8669],
        [ -9.4296,  -1.9458,   4.0833],
        [ -2.0840,   7.8415,  -7.8657],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.5198,  -4.0923,   5.5088],
        [ -9.2301,  -3.1537,  -2.7477],
        [  5.8979,   3.3958,  -3.8713]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0195, 0.1125,  ..., 0.9445, 0.2799, 0.7672],
        [0.0195, 0.0000, 0.0938,  ..., 0.9637, 0.2605, 0.7812],
        [0.1125, 0.0938, 0.0000,  ..., 1.0469, 0.1778, 0.8668],
        ...,
        [0.9445, 0.9637, 1.0469,  ..., 0.0000, 1.2231, 0.8097],
        [0.2799, 0.2605, 0.1778,  ..., 1.2231, 0.0000, 0.9741],
        [0.7672, 0.7812, 0.8668,  ..., 0.8097, 0.9741, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.09502385,  9.11473391,  9.36093562,  9.36378438,  9.48337063,
        9.60359158,  9.97220799,  5.13125513,  5.70405525,  5.05774295,
        5.64865173, 11.1341798 ,  9.63742669,  8.79284466, 10.26327895,
       10.7506423 ,  6.3949043 , 12.35421461,  5.14999799, 13.68672855,
       13.55868684, 12.70489563, 10.0674887 ,  9.17905491,  8.1303945 ,
       13.27717982,  6.5618968 ,  9.97220799, 14.10045797,  5.98285319,
        8.89301516, 11.9120321 ,  6.46356806, 11.46176259, 11.04313641,
        9.68324553, 13.96905756,  7.03733775, 13.89886469,  7.09887557,
       10.90638066,  5.79562691, 10.99326577, 13.96905756,  9.44486609,
        8.34949383, 10.74409834, 13.6660671 , 11.63383221,  7.20482224,
        8.14433542, 11.38774149,  5.51177501, 10.00802627,  5.63824971,
       12.32408516, 12.32545637, 10.794755  ,  7.27340834,  4.98989906,
        5.94535529,  8.5724042 , 10.35485331, 12.19805394,  9.64157461,
       10.35485331,  6.56833864,  6.33517349, 10.46238054, 10.1371552 ,
       13.98691426,  8.29022197,  5.82345867,  8.23525433,  5.1224079 ,
       11.69264243, 11.018084  , 10.48840476,  5.39098669,  7.78980299,
        6.21275016, 12.44177641, 11.75618513,  8.44940661, 13.97857591,
       10.05558893,  8.77913428, 13.04051744, 14.02526505, 10.65929968,
       11.62158554, 12.27993098,  6.05165248, 12.97370371,  8.62260327,
        5.86519525,  7.05627461, 11.10066126, 12.62576026,  7.16747878,
        9.6282763 ,  8.1137452 , 13.02818702, 12.22503754,  9.75401449,
        6.80559669])}
{'P': tensor([[1.3141e-02, 2.1351e-01, 1.2648e-01,  ..., 9.6157e-07, 3.3179e-06,
         3.8969e-05],
        [1.5127e-04, 1.7084e-02, 4.9228e-01,  ..., 1.7886e-06, 4.0675e-06,
         3.0034e-05],
        [5.0810e-04, 1.1642e-03, 9.7082e-02,  ..., 1.2405e-05, 2.3170e-05,
         1.1723e-04],
        ...,
        [2.2121e-05, 1.9164e-05, 1.7766e-05,  ..., 3.2511e-01, 1.2739e-02,
         4.7435e-04],
        [8.1252e-04, 4.5694e-04, 3.2958e-04,  ..., 4.0882e-04, 1.7638e-01,
         1.5094e-03],
        [2.0655e-04, 4.3590e-04, 5.8988e-04,  ..., 1.2538e-04, 2.8051e-05,
         7.4284e-02]]), 'X': tensor([[ -8.6112,   2.9319,  -7.9943],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.3076,   0.7852,  -6.7010],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -8.9033,   2.2329,  -8.8912],
        [  0.8572,  14.0501,  -8.3364],
        [  6.2535,  -0.1709,  -9.4102],
        [ -7.4807,   4.7463,   2.0616],
        [  2.5663,   7.0526,  -6.8769],
        [  0.1677,  -4.7447,  -6.1529],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.2707,   0.9661,   4.4863],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.6952,  -4.8060,   8.8976],
        [  2.1771, -10.9776,  -3.2007],
        [  1.6232,  -4.7717,  -2.5655],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.4877,  -0.7222,   8.8347],
        [  6.1989,  -9.7591,   0.8289],
        [  4.7156,  -3.1831,  -8.0224],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.1575,  12.2046,   3.3257],
        [  5.8617,   3.4853,  -9.8080],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.3715,  -4.7756,   0.2312],
        [  5.1472,  -2.6801,   8.1194],
        [  6.3413,   0.4883,  -6.8416],
        [  6.8704,  -9.3809,   7.5769],
        [  6.3587,   1.1596,  -3.6629],
        [  7.0515,  11.6194,   8.4289],
        [  5.6095,  -1.9702,  -7.8233],
        [ -8.8058,   2.4850,  -1.7425],
        [ -3.9700,   7.3453,  -4.1486],
        [ -4.2271,   7.2387,  -1.8316],
        [ 12.1588,   4.3364,  -8.0867],
        [ 12.1940,  -2.1881,   7.1499],
        [ -7.2986,  -7.1225,  -3.0814],
        [  2.6321,  -4.5296,  -2.5159],
        [ -9.4366,  -0.1189,  -2.6736],
        [  1.3103,  -4.8032,  -0.9300],
        [  2.1824,  -4.6647,   1.1985],
        [  2.6310, -10.9195,   3.8054],
        [ 10.4994,  -5.9193,  -7.5153],
        [  5.9026,   3.3837,  -0.6573],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.5813,  -4.5471,   5.3597],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.6681,  -4.8556,   9.2518],
        [ 11.7418,   5.6070,  -1.4192],
        [ 12.5650,   2.0132,   4.6272],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.2687,  -2.9794,   0.6436],
        [  1.6954,  -4.7616,   9.8198],
        [  6.3087,   1.8053,  -1.9481],
        [  9.8082,  -6.8396,   4.9615],
        [  1.6408,   7.4591,  -4.2294],
        [  7.4132,  11.3526,   0.3322],
        [  2.7590,  13.6621,   9.4496],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.7792,  11.8084,   1.6251],
        [  7.5930,  -8.9037,   7.2323],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.8363,   2.4086,  -5.6296],
        [  6.1752,   2.5242,   3.5874],
        [  5.7485,  -9.9811,  -3.2355],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.6363, -11.0348,   6.1414],
        [  9.5261,  -7.1675,   1.4206],
        [ -0.5739,   7.9056,   0.6382],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.4748,  -0.7765,  -5.6695],
        [  7.3592,  11.3937,   6.2407],
        [  0.8700,  -4.8132,  -7.8042],
        [  2.6078, -10.9229,   6.2869],
        [  9.9750,  -6.6338,  -1.2440],
        [  2.5010,  -4.5736,  -0.8989],
        [ -2.0445,   7.8469,  -0.5857],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.2555,   2.1474,   4.1685],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.8934,   2.2597,   3.5831],
        [ -2.5056,   7.7720,   5.5878],
        [  4.1637,  13.1898,  -1.9189],
        [  3.3474,  -4.2162,   5.2455],
        [  3.5965,  -4.0754,  -3.5035],
        [ 10.9916,   7.2133,  -0.9179],
        [  5.8325,  -1.5222,  -2.1341],
        [ -4.4503,   7.1378,   3.3751],
        [ -9.4739,  -1.2497,   0.5483],
        [ 11.6801,  -3.7196,  -0.9181],
        [  8.1528,  10.7452,   9.1493],
        [ -2.5161,   7.7699,  -9.3531],
        [ 11.4703,  -4.2006,   7.4086],
        [  8.8091,  -7.9037,   0.3631],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.0788,  -0.8710,  -6.8719],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -1.6086,   7.8928,   4.3385],
        [ -9.3087,   0.7792,  -7.5587],
        [  5.5294,  -2.1106,   9.0626],
        [  5.1793,   4.7461,   0.3899],
        [ -8.6603,   2.8250,   5.9181]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.1636e-02, 1.3408e-01,  ..., 1.3107e+00, 9.3798e-01,
         8.9748e-01],
        [9.1636e-02, 0.0000e+00, 4.2478e-02,  ..., 1.3747e+00, 1.0182e+00,
         9.1381e-01],
        [1.3408e-01, 4.2478e-02, 3.0339e-04,  ..., 1.4067e+00, 1.0568e+00,
         9.2625e-01],
        ...,
        [1.3107e+00, 1.3747e+00, 1.4067e+00,  ..., 0.0000e+00, 4.9465e-01,
         7.7935e-01],
        [9.3798e-01, 1.0182e+00, 1.0568e+00,  ..., 4.9465e-01, 0.0000e+00,
         8.1021e-01],
        [8.9748e-01, 9.1381e-01, 9.2625e-01,  ..., 7.7935e-01, 8.1021e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.09661212,  9.34127013,  9.47369628,  9.34061938,  9.48969978,
        9.17905491, 14.07623266,  6.25586297,  8.85939626,  7.50500064,
        4.74770899,  9.71392179,  9.32094055,  8.62631801,  4.85603886,
       11.1913598 ,  5.04028538, 11.85743425, 12.38881635, 12.5085995 ,
       11.56146873,  5.68942572,  8.85572292, 13.66991027,  6.81961996,
        9.36093562,  4.79003361,  5.80312949,  6.36004393, 11.62770332,
        6.46356806, 13.59170809,  5.94541803,  9.14973319,  8.34949383,
        8.38252937, 12.9089559 , 12.38881635, 10.1979669 ,  5.23878371,
        9.43737523,  4.978704  ,  5.14999799, 11.23201619, 12.05300943,
        6.80370254, 12.84056604,  5.2287058 ,  9.94285511,  9.9353921 ,
       13.01187484, 12.72524615, 11.99633819,  9.73579738,  5.05444288,
        6.5618968 , 11.95743558,  7.6374603 , 13.55868684, 13.93790235,
       12.30858677, 13.1399181 , 13.61600536, 11.70168755, 12.71843462,
        9.15866607,  6.67121748, 11.51811603, 10.33712258, 11.05317597,
       11.92133129,  7.92644746,  9.13507321,  9.50654967, 13.56367857,
        4.89121805, 11.22993218, 11.97949273,  5.21280156,  8.10886021,
        5.19065439,  6.61386436, 10.33958639,  9.17595899,  8.16584537,
       13.83139024,  5.38342077,  5.43544626, 13.1471313 ,  6.02789634,
        8.41148722,  9.55593374, 12.25806559, 13.48809595,  8.16714836,
       12.21532282, 11.83509372, 12.46513646,  6.14086582,  9.5014146 ,
        9.61942445,  8.0550286 ,  9.34127013,  5.91854397,  7.02496223,
        9.10945678])}
{'P': tensor([[8.1825e-03, 2.8807e-01, 1.8764e-01,  ..., 3.0568e-06, 8.2642e-05,
         1.2512e-06],
        [1.9683e-04, 1.0479e-02, 2.4160e-01,  ..., 3.8259e-06, 8.6261e-05,
         1.7774e-06],
        [2.6807e-04, 3.8687e-04, 2.0810e-02,  ..., 7.9375e-06, 1.1343e-04,
         4.4923e-06],
        ...,
        [1.7767e-04, 1.5108e-04, 1.2310e-04,  ..., 4.5539e-02, 3.8248e-04,
         4.2446e-05],
        [2.2422e-01, 1.8258e-01, 1.3140e-01,  ..., 6.3779e-06, 1.4399e-02,
         1.4962e-06],
        [2.0950e-06, 1.8435e-06, 1.5015e-06,  ..., 3.5191e-03, 4.0099e-06,
         2.0426e-02]]), 'X': tensor([[-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9527e-02, 7.5915e-02,  ..., 8.5804e-01, 9.9045e-02,
         6.7009e-01],
        [2.9527e-02, 0.0000e+00, 5.1247e-02,  ..., 8.8557e-01, 1.2855e-01,
         6.9935e-01],
        [7.5915e-02, 5.1247e-02, 2.1453e-04,  ..., 9.1047e-01, 1.6955e-01,
         7.3401e-01],
        ...,
        [8.5804e-01, 8.8557e-01, 9.1047e-01,  ..., 0.0000e+00, 7.6527e-01,
         2.6160e-01],
        [9.9045e-02, 1.2855e-01, 1.6955e-01,  ..., 7.6527e-01, 0.0000e+00,
         5.7158e-01],
        [6.7009e-01, 6.9935e-01, 7.3401e-01,  ..., 2.6160e-01, 5.7158e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.09963894,  9.15253344,  9.25336371,  9.3349858 ,  9.3109884 ,
        9.24501829, 13.01187484, 13.38091121, 12.3344536 , 10.00794468,
       13.2778378 ,  8.85939626,  5.84450127, 10.06703358,  9.37989631,
        5.5366698 ,  6.83515201,  7.5984809 ,  8.53014915, 13.3607379 ,
        6.54693987,  7.32754882,  5.10446653,  9.75200848, 10.36844813,
       11.62770332, 11.4424816 , 13.44593519, 10.76154256,  9.49511524,
        5.41348121,  9.53015376, 12.52372629,  9.85330646,  5.5748624 ,
        5.24137909,  8.68832398,  8.85939626,  8.56889738,  7.52117132,
       10.75443106, 13.68672855,  7.92607342, 12.4217523 , 11.22596994,
       12.11735334,  9.97220799,  5.1805345 , 13.26590071,  9.88310614,
        5.94909709,  9.11473391,  4.89869205, 13.0743399 ,  6.54615887,
        5.21280156, 11.66453059,  9.55720651,  8.72896381,  4.78544875,
       13.93790235,  7.15990676,  5.1224079 , 11.46176259,  6.65259775,
       11.83570132,  4.8763135 ,  5.63824971, 11.11038551,  7.4544866 ,
        5.63824971, 12.91505111, 12.58006038,  5.84450127, 13.09426142,
       13.2430377 ,  8.23525433,  7.52117132, 10.74409834, 12.86648771,
       11.1341798 ,  8.68979052, 11.99633819, 10.09979747, 13.52430176,
        8.35856487,  6.50409372,  8.95133961,  7.54620849,  6.54693987,
       11.95743558,  9.97964013,  7.09887557,  7.69752262,  7.06272256,
        8.50278896,  7.89745107,  5.41348121, 12.04924159,  5.05444288,
        5.70591986, 11.10066126, 13.6105595 ,  6.76661595,  8.85939626,
       12.1494023 ])}
{'P': tensor([[1.8770e-02, 2.1917e-01, 1.7406e-01,  ..., 1.2576e-05, 1.7045e-06,
         1.6342e-04],
        [2.0639e-04, 2.0529e-02, 3.2225e-02,  ..., 5.6592e-06, 8.8951e-07,
         6.3035e-05],
        [1.4647e-04, 2.1531e-03, 2.1755e-02,  ..., 7.1167e-06, 1.4319e-06,
         1.1936e-04],
        ...,
        [1.2110e-03, 5.0949e-04, 9.3878e-04,  ..., 2.9677e-02, 5.6890e-04,
         6.7247e-03],
        [2.5692e-04, 1.1367e-04, 2.1790e-04,  ..., 3.1528e-02, 1.2514e-01,
         9.6918e-03],
        [5.0918e-04, 7.2826e-04, 2.2517e-03,  ..., 1.5210e-04, 1.0214e-04,
         2.1066e-01]]), 'X': tensor([[-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 1.0390e-01, 1.2815e-01,  ..., 7.8427e-01, 1.0177e+00,
         8.8196e-01],
        [1.0390e-01, 0.0000e+00, 1.4561e-01,  ..., 8.8406e-01, 1.1147e+00,
         9.3908e-01],
        [1.2815e-01, 1.4561e-01, 3.0339e-04,  ..., 7.8548e-01, 1.0056e+00,
         7.9414e-01],
        ...,
        [7.8427e-01, 8.8406e-01, 7.8548e-01,  ..., 1.0727e-04, 2.4545e-01,
         6.0714e-01],
        [1.0177e+00, 1.1147e+00, 1.0056e+00,  ..., 2.4545e-01, 0.0000e+00,
         6.2184e-01],
        [8.8196e-01, 9.3908e-01, 7.9414e-01,  ..., 6.0714e-01, 6.2184e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.10945678,  9.36256096,  9.44486609,  9.53015376,  9.32094055,
        9.49511524,  6.58184474,  5.48276038,  8.53501777, 12.78327037,
        7.73151309,  7.72668467, 13.92390689,  5.48276038, 11.4890861 ,
       12.9089559 , 10.26545792,  8.06990483,  8.05668609,  5.05981914,
       10.06703358,  7.53063885, 13.46553623, 11.52733581,  5.05444288,
       12.40074901,  7.37576154, 10.05558893, 12.9089559 ,  4.92497514,
       13.97430191,  7.60463349,  9.94173708,  4.91652556, 13.96746816,
        6.00663314,  7.17789666,  7.5092834 ,  7.02326515, 12.9089559 ,
       13.40850328,  9.48337063,  7.8575648 ,  9.79814997,  8.41148722,
        5.41348121, 13.14690137, 13.58333857,  8.83725081, 12.0495521 ,
       14.07623266, 12.67576792, 10.42667093, 12.45418236,  7.27340834,
       10.84437705,  5.79543038,  5.92281142,  5.35774573,  6.61211224,
        5.26342203, 11.85743425,  7.72668467,  5.84533345, 13.3607379 ,
       10.38897563,  5.63824971,  7.92200342,  5.41348121,  7.15990676,
        6.07992623,  9.34061938,  6.31576264,  8.35856487, 12.24467041,
       12.24467041,  5.77219077, 10.67132524,  8.99276564,  9.71392179,
       10.63038565, 13.14690137,  9.17595899,  6.01356004,  4.75303127,
        6.65259775, 12.49317563,  5.54776934, 12.50996828, 11.67875855,
       11.9702762 ,  5.91222346, 11.10066126,  8.1137452 ,  8.55982105,
       11.83570132,  8.1303945 ,  9.0368458 , 12.76408562,  4.78258092,
        8.8353674 ,  9.77777346,  8.38252937,  8.14433542,  7.83534987,
        9.34061938])}
{'P': tensor([[1.3131e-02, 2.1316e-01, 1.7636e-01,  ..., 5.2528e-07, 4.4257e-03,
         2.2409e-03],
        [1.5867e-04, 1.8197e-02, 2.1027e-02,  ..., 4.1565e-07, 1.6532e-02,
         4.9742e-03],
        [1.7298e-04, 1.1619e-02, 1.8124e-02,  ..., 4.0269e-07, 1.5354e-02,
         4.2184e-03],
        ...,
        [3.6565e-03, 1.1351e-03, 1.0662e-03,  ..., 1.3693e-01, 2.6836e-05,
         4.0917e-05],
        [2.7389e-05, 4.8030e-05, 4.8324e-05,  ..., 9.6160e-07, 6.7512e-02,
         3.7880e-03],
        [3.7406e-05, 7.4717e-05, 7.3026e-05,  ..., 1.1007e-06, 3.8041e-02,
         8.1451e-02]]), 'X': tensor([[-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0938, 0.1095,  ..., 1.1492, 0.2329, 0.5164],
        [0.0938, 0.0000, 0.0309,  ..., 1.2411, 0.2007, 0.5179],
        [0.1095, 0.0309, 0.0000,  ..., 1.2585, 0.2277, 0.5475],
        ...,
        [1.1492, 1.2411, 1.2585,  ..., 0.0000, 1.2469, 1.1971],
        [0.2329, 0.2007, 0.2277,  ..., 1.2469, 0.0000, 0.3268],
        [0.5164, 0.5179, 0.5475,  ..., 1.1971, 0.3268, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.11473391,  9.36093562,  9.36378438,  9.48337063,  9.60359158,
        9.54154096,  8.68832398, 12.50996828,  7.89745107,  5.51177501,
        7.50035355, 12.13568598, 13.26209545,  7.03733775, 11.75438006,
        8.31625171,  7.69752262, 13.97857591,  7.62509582, 11.70168755,
       10.33712258, 11.51811603,  8.06517143,  6.46356806,  6.16512856,
        9.79814997, 11.95743558,  7.78980299,  5.13616224,  9.10945678,
       11.31276109, 14.07623266, 13.32557133,  5.77219077, 10.68031351,
       14.08549802, 13.79676654, 14.10796466, 10.27716434, 11.9702762 ,
       10.67322527, 10.75443106, 10.1979669 , 14.09176259,  6.42689705,
       12.9089559 ,  7.6374603 ,  8.99276564, 10.7506423 , 13.1471313 ,
        6.21275016,  7.89745107,  7.09887557, 12.79126785,  7.48538905,
       11.27656488,  5.13182506, 13.06957425, 11.51557166,  9.50913849,
       10.87409005, 13.27994304,  8.62260327,  5.5205189 ,  7.06372172,
        5.35774573,  7.32233733,  5.79543038, 13.0743399 , 11.00508681,
        8.09704052, 12.4980845 , 11.72802491,  7.822854  ,  5.51177501,
        9.75401449,  5.36061974, 12.52372629, 11.80092578, 13.59673406,
        9.45995955,  6.27905292,  7.23276582,  6.4020404 , 13.35832725,
       14.02058621,  7.58668144,  5.07306274,  7.8575648 ,  9.76896904,
        7.58668144, 13.3426541 ,  9.96391045,  9.87720102, 11.80092578,
        8.87587457,  5.04028538, 12.7125079 , 11.83075005,  5.05534609,
        7.58668144,  9.36093562,  6.80559669, 13.92390689, 10.24557047,
       10.14044489])}
{'P': tensor([[7.6546e-03, 1.8302e-01, 1.2059e-01,  ..., 5.0922e-05, 1.2533e-06,
         1.0946e-05],
        [2.9669e-04, 2.0383e-02, 4.9423e-01,  ..., 1.5782e-04, 4.3532e-06,
         3.6141e-05],
        [9.7710e-04, 1.8104e-03, 8.8759e-02,  ..., 7.2639e-04, 2.2003e-05,
         1.8579e-04],
        ...,
        [5.6878e-05, 5.2638e-05, 4.9582e-05,  ..., 7.5357e-02, 1.0795e-02,
         5.6691e-04],
        [1.4532e-06, 1.1451e-06, 9.8364e-07,  ..., 2.5868e-04, 4.2854e-02,
         8.0848e-05],
        [2.1177e-04, 1.8005e-04, 1.6717e-04,  ..., 3.7302e-04, 1.3380e-04,
         2.0159e-01]]), 'X': tensor([[ -8.7078,   2.7182,   1.8146],
        [ -9.1178,   1.5784,   2.1410],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.0604,   1.7707,   1.1066],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.8400,   2.3991,   1.3457],
        [  3.6703,   6.3443,  -5.4401],
        [  0.8940, -11.0402,  -0.9783],
        [ -1.1885, -10.8211,   0.0865],
        [  3.6249, -10.7255,   5.6605],
        [ -7.3554,   4.8985,  -0.6665],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.7544,   2.6096,   3.3146],
        [  2.5010,  -4.5736,  -0.8989],
        [  6.3516,   0.6425,   0.3425],
        [  8.0877,  -8.5297,  -1.1050],
        [ 11.6019,   5.9540,  -5.1715],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 12.5224,  -0.4497,   4.8448],
        [  3.3744,   6.5623,  -6.3636],
        [  7.2340,  -9.1505,  -5.8321],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.8721,  -4.7322,   0.5524],
        [  5.6095,  -1.9702,  -7.8233],
        [  9.2979,   9.6048,   7.3138],
        [ -5.3382,   6.6533,   7.8054],
        [  2.7558,  -4.4844,   0.0636],
        [ 11.7349,  -3.5838,   2.1644],
        [  0.1206,  -4.7363,   8.7103],
        [  2.5974,   7.0363,  -7.7810],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.6229,   2.9068,   2.4429],
        [ -0.7599, -10.8995,  -6.0366],
        [ 10.0567,   8.6697,   4.6329],
        [ 11.9122,   5.1397,  -1.5498],
        [ -2.0229, -10.6163,   8.6930],
        [  6.7565,  11.8237,  -4.9882],
        [ -3.1671,   7.6158,   6.0765],
        [  2.5663,   7.0526,  -6.8769],
        [ -4.4503,   7.1378,   3.3751],
        [  2.9410,   6.8428,   5.4458],
        [  2.0243,   7.3086,   7.5432],
        [  4.6357,   5.4415,   7.9067],
        [ -8.8400,   2.3991,   1.3457],
        [  7.0642,  11.6104,  -9.8316],
        [ -5.9940,   6.1985,   5.9566],
        [  3.5284, -10.7484,   5.3347],
        [  1.5662,  13.9375,   4.5142],
        [  9.4925,   9.3811,   8.9842],
        [  5.3804,  -2.3504,   7.2744],
        [  3.2366,  -4.2732,   5.1532],
        [ -1.5451,   7.8975,   7.6104],
        [  6.3604,   0.9042,  -6.5901],
        [  3.3167,   6.6022,   6.3554],
        [  6.3604,   0.9042,  -6.5901],
        [  6.3303,   1.6145,  -8.4083],
        [ 10.7566,   7.6212,  -2.3279],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.8979,   3.3958,  -3.8713],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -9.0604,   1.7707,   1.1066],
        [  6.3236,   0.3011,  -4.6668],
        [  5.2188,  -2.5839,   2.1833],
        [  6.3542,   0.6934,  -3.3840],
        [  8.2422,  -8.4041,  -1.4962],
        [  8.3791,  10.5409,   5.5590],
        [ 10.0619,   8.6627,  -9.2884],
        [ -8.0330,  -6.0765,   6.7945],
        [ -9.0604,   1.7707,   1.1066],
        [  2.3928,   7.1404,  -1.7035],
        [ -3.4986,   7.5152,   9.0712],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.2707,   0.9661,   4.4863],
        [  4.1537,   5.9340,  -6.0423],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.7976,  -4.4682,  -4.2020],
        [ -7.0120,   5.2825,   7.7762],
        [  6.3413,   0.4883,  -6.8416],
        [ -7.8280,   4.2848,  -7.1955],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.6933,   7.7336,   2.1179],
        [ -7.9667,   4.0814,  -0.6397],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.0890,  -0.8382,   2.9166],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.3602,  -2.3808,   7.2098],
        [  2.9470,  13.6085,   2.9809],
        [ -7.4808,   4.7462,   2.0266],
        [  5.4732,  -2.2041,  -4.3830],
        [ -9.3604,   0.4855,  -3.0158],
        [  0.1382, -11.0073,   1.0026],
        [  4.5800,   5.5034,  -3.9432],
        [  6.3551,   0.7149,   1.3344],
        [  5.2608,  -2.5256,  -9.1446],
        [ 12.0042,  -2.8350,   2.6864],
        [  1.5246,  -4.7838,  -5.2095],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.3804,  -2.3504,   7.2744]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.0639e-02, 9.6817e-02,  ..., 7.9703e-01, 9.0823e-01,
         8.4319e-01],
        [6.0639e-02, 2.1453e-04, 3.7217e-02,  ..., 8.5747e-01, 9.6553e-01,
         8.8154e-01],
        [9.6817e-02, 3.7217e-02, 0.0000e+00,  ..., 8.9381e-01, 1.0027e+00,
         8.9707e-01],
        ...,
        [7.9703e-01, 8.5747e-01, 8.9381e-01,  ..., 0.0000e+00, 2.4700e-01,
         7.8570e-01],
        [9.0823e-01, 9.6553e-01, 1.0027e+00,  ..., 2.4700e-01, 0.0000e+00,
         1.0315e+00],
        [8.4319e-01, 8.8154e-01, 8.9707e-01,  ..., 7.8570e-01, 1.0315e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.12220271,  9.25336371,  9.3349858 ,  9.23178204,  9.3109884 ,
        9.15976746,  7.32950385, 11.07637531, 10.88618114, 11.32149259,
        8.83725081, 11.97238624,  9.13507321,  5.21280156,  6.38399789,
       11.75438006, 13.04051744, 13.97857591,  9.3109884 ,  9.85330646,
       12.53047489,  7.37902766, 11.66453059, 12.32545637,  5.08910491,
        5.94541803, 13.36800073,  8.53014915,  5.26342203, 12.26997165,
        4.73784222,  7.50035355,  9.66977802,  6.27905292,  9.09963894,
       10.92597196, 13.2778378 , 12.97370371, 10.80728788, 13.61801526,
        8.24808832,  7.50500064,  8.41148722,  7.44805901,  7.58378655,
        7.14837514,  9.15976746, 13.59056613,  8.62260327, 11.31276109,
       14.02526505, 13.34587027,  5.87132718,  5.36061974,  8.04718433,
        6.42439344,  7.38844703,  6.42439344,  6.53290663, 13.18278606,
       10.60226614,  6.80559669, 11.62158554, 10.06407839,  9.23178204,
        6.33075682,  5.82345867,  6.39187285, 11.77124373, 13.46553623,
       13.27717982, 10.07238463,  9.23178204,  7.53063885,  8.28968067,
       13.36092624,  9.32094055,  7.24326992,  9.36378438,  9.7712321 ,
        6.1463896 ,  9.55593374,  5.27178363,  8.77913428,  6.36004393,
        8.92395381,  5.27178363,  7.76457095,  8.95133961,  5.48276038,
        6.1463896 ,  6.1561646 ,  9.97822473,  5.86519525, 13.92390689,
        8.859416  ,  5.90035324,  9.37296102, 11.00812996,  7.15990676,
        6.39520397,  5.83560009, 12.3344536 ,  5.0209005 , 11.86239961,
        5.87132718])}
{'P': tensor([[7.8808e-03, 2.2657e-01, 1.7031e-01,  ..., 3.9220e-06, 5.0015e-06,
         8.7371e-04],
        [2.8537e-04, 1.9488e-02, 6.3901e-01,  ..., 1.3579e-05, 1.2464e-05,
         3.3488e-03],
        [1.3132e-03, 2.0501e-03, 1.1473e-01,  ..., 9.9647e-05, 7.4400e-05,
         2.5288e-02],
        ...,
        [1.8429e-06, 1.6204e-06, 1.5263e-06,  ..., 3.2769e-02, 2.2731e-03,
         2.1372e-06],
        [4.9398e-04, 3.5805e-04, 2.9646e-04,  ..., 4.1945e-05, 9.0223e-02,
         2.9743e-05],
        [1.6048e-05, 1.9221e-05, 2.1695e-05,  ..., 3.3278e-03, 2.2281e-04,
         7.3061e-02]]), 'X': tensor([[ -8.7544,   2.6096,   3.3146],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.8745,   2.3097,   3.5600],
        [ -9.2974,   0.8371,   2.4852],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.6314,  -1.9302,  -9.6543],
        [  4.2275,  -3.6320,  -0.2714],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.9337,  12.8566,  -3.6582],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.1905,   3.7253,   4.1175],
        [  2.8994,   6.8675,  -9.5987],
        [  8.1043,  -8.5164,  -7.9290],
        [ 11.7794,   5.5086,  -9.6929],
        [ -9.1231,  -3.5739,   8.4708],
        [  4.8653,   5.1694,  -4.0823],
        [ -4.2271,   7.2387,  -1.8316],
        [  5.9912,   3.1453,  -3.9989],
        [  6.0560,  -0.9421,  -5.0334],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.6396,  14.0772,  -6.3786],
        [  9.1820,   9.7334,  -2.7932],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.4671,  -0.5553,   0.1117],
        [ 12.0042,  -2.8350,   2.6864],
        [ -1.8823, -10.6558,  -3.8952],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.0729,   7.2878,   7.0880],
        [ 11.9641,  -2.9568,   7.7211],
        [  4.5954,  13.0099,  -4.4439],
        [ 10.9599,  -5.1849,  -3.5526],
        [  0.3231,  -4.7691,  -1.6120],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.6817,   6.3354,   6.1233],
        [  0.3493,  -4.7727,  -6.5682],
        [  1.4150,  13.9647,  -4.0915],
        [  4.6992,   5.3690,   3.0935],
        [ -2.1533, -10.5778,   1.0112],
        [  3.5965,  -4.0754,  -3.5035],
        [  3.3474,  -4.2162,   5.2455],
        [  5.6248,   3.9984,   8.4933],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.8154,   2.4611,   2.5198],
        [  7.0358,  -9.2784,  -9.8479],
        [  4.7482, -10.3907,   5.9149],
        [  2.8890,  -4.4316,   1.1600],
        [  4.9452, -10.3187,  -7.8621],
        [  2.1379,  -4.6757,   6.4981],
        [ -5.9872,   6.2037,  -9.4477],
        [ 12.5832,   1.7543,  -7.6134],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.2975,   0.0947,   5.1791],
        [  2.0039,   7.3173,  -1.5885],
        [  5.4732,  -2.2041,  -4.3830],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.3604,   0.4855,  -3.0158],
        [  2.5469,  13.7192,  -9.6935],
        [  3.2809,  13.5061,  -4.0199],
        [ -8.8934,   2.2597,   3.5831],
        [ -3.7964,  -9.9294,   7.7147],
        [  8.4691,  10.4571,  -5.0650],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.4573,   3.2448,   5.4370],
        [  2.7590,  13.6621,   9.4496],
        [  6.7565,  11.8237,  -4.9882],
        [  4.5954,  13.0099,  -4.4439],
        [  9.0652,   9.8597,  -3.5949],
        [  0.8700,  -4.8132,  -7.8042],
        [  5.6248,   3.9984,   8.4933],
        [  4.5800,   5.5034,  -3.9432],
        [  4.1348,   5.9514,  -0.8856],
        [ -0.2586,   7.8825,  -9.9840],
        [ 12.2914,   3.8037,   6.0978],
        [  4.5503,   5.5359,   7.4735],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.2076,  13.8032,  -1.5099],
        [ -8.1666,   3.7653,   3.6075],
        [  2.0449,  13.8402,  -7.1642],
        [  3.4140,  -4.1803,  -8.8224],
        [  6.8093,  -9.4178,   8.4244],
        [  3.3069,   6.6089,   8.7377],
        [  5.9526,   3.2525,  -4.3821],
        [  0.3231,  -4.7691,  -1.6120],
        [ -7.8849,  -6.3115,  -4.7479],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.8400,   2.3991,   1.3457],
        [ 11.9122,   5.1397,  -1.5498],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.3549,   0.7096,   6.9198],
        [ -4.3963,   7.1629,   4.5263],
        [  1.9204,   7.3518,   8.4781],
        [  6.3549,   0.7096,   6.9198],
        [  9.2811,  -7.4336,   5.1771],
        [ -4.7389,   6.9953,   3.7638],
        [ -9.1433,  -3.5003,  -3.0209],
        [  3.4401,  13.4540,   4.6576],
        [  5.8889,  -1.3916,   8.8993],
        [  5.1787,  -2.6383,  -4.2533],
        [ -8.0608,   3.9362,  -5.9567],
        [ -1.0407,   7.9166,  -6.6021],
        [  8.0877,  -8.5297,  -1.1050],
        [  6.1737,   2.5305,   3.9049],
        [ -5.1398,  -9.1367,   6.2074]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 5.1106e-02, 7.9547e-02,  ..., 6.0005e-01, 7.5689e-01,
         2.6553e-01],
        [5.1106e-02, 2.1453e-04, 2.8654e-02,  ..., 6.3493e-01, 8.0753e-01,
         2.5884e-01],
        [7.9547e-02, 2.8654e-02, 2.1453e-04,  ..., 6.5219e-01, 8.3508e-01,
         2.6376e-01],
        ...,
        [6.0005e-01, 6.3493e-01, 6.5219e-01,  ..., 0.0000e+00, 4.9430e-01,
         8.4338e-01],
        [7.5689e-01, 8.0753e-01, 8.3508e-01,  ..., 4.9430e-01, 0.0000e+00,
         8.8757e-01],
        [2.6553e-01, 2.5884e-01, 2.6376e-01,  ..., 8.4338e-01, 8.8757e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.13507321,  9.24501829,  9.3109884 ,  9.17595899,  9.17016857,
        9.3349858 , 10.92597196,  5.95297459,  5.57341623,  5.24137909,
       13.77075174,  4.91038884,  8.99791316,  7.4544866 , 11.75618513,
       13.00380624,  9.79814997,  7.09887557,  8.38252937,  6.76661595,
        6.12886024, 12.49317563, 10.00410625, 14.09176259, 13.38091121,
        6.60752373,  9.48337063, 12.3344536 , 10.8207297 , 12.95702647,
        7.57685908, 12.32408516, 13.79762415, 12.12448912,  4.78002665,
       12.27993098,  7.32754882,  4.78544875, 14.03618649,  7.13501264,
       10.794755  ,  5.43544626,  5.38342077,  6.90115901,  7.822854  ,
        9.15253344, 11.64436684, 11.42420508,  5.29009836, 11.4424816 ,
        5.14124612,  8.62162372, 12.70489563, 10.42667093,  6.29821673,
        7.58668144,  5.90035324,  9.18690018,  9.37296102, 13.95361098,
       13.89886469,  9.17595899, 10.63038565, 13.45642379, 13.6660671 ,
        9.05843201, 13.93790235, 13.61801526, 13.79762415, 13.39372426,
        4.89121805,  6.90115901,  7.15990676,  7.24677664,  7.88677538,
       12.86648771,  7.16598273, 12.10942412, 13.97857591,  8.99276564,
       13.99048094,  5.39723241, 11.62158554,  7.39004226,  6.78325151,
        4.78002665, 10.09979747,  5.69913817,  9.15976746, 12.97370371,
        8.30125201,  6.39438044,  8.40445872,  7.5984809 ,  6.39438044,
       11.89105259,  8.44940661,  9.79039371, 13.88683436,  6.05113217,
        5.81199422,  8.97052115,  7.98469166, 11.75438006,  6.67218687,
       10.48315436])}
{'P': tensor([[1.5543e-02, 3.0433e-01, 1.7202e-01,  ..., 2.3665e-04, 1.3856e-07,
         4.0979e-05],
        [1.7007e-04, 1.6384e-02, 2.9523e-01,  ..., 4.5780e-04, 1.1980e-07,
         2.1027e-05],
        [8.9949e-04, 2.3356e-03, 1.6870e-01,  ..., 1.0179e-02, 1.3491e-06,
         1.6477e-04],
        ...,
        [3.0458e-06, 3.8067e-06, 5.4802e-06,  ..., 3.4248e-02, 5.8808e-06,
         3.1032e-06],
        [8.9478e-06, 4.0015e-06, 2.6987e-06,  ..., 2.5227e-07, 2.0513e-02,
         6.6459e-05],
        [1.7206e-04, 2.2280e-04, 3.8897e-04,  ..., 7.5358e-04, 1.4336e-06,
         5.0872e-02]]), 'X': tensor([[-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0833, 0.1274,  ..., 0.4858, 1.5546, 0.9571],
        [0.0833, 0.0000, 0.0559,  ..., 0.5165, 1.6309, 0.9941],
        [0.1274, 0.0559, 0.0000,  ..., 0.4978, 1.6496, 0.9765],
        ...,
        [0.4858, 0.5165, 0.4978,  ..., 0.0000, 1.2870, 0.4787],
        [1.5546, 1.6309, 1.6496,  ..., 1.2870, 0.0000, 1.0534],
        [0.9571, 0.9941, 0.9765,  ..., 0.4787, 1.0534, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.14044984,  9.34061938,  9.4997987 ,  9.48969978,  9.50654967,
        9.34127013,  7.5984809 , 14.02532758,  5.37343173, 11.59512194,
        9.36256096,  7.71075226,  5.84533345, 13.76047999,  8.79284466,
       13.97779839, 13.03002338,  8.62260327, 12.91143491,  7.37576154,
        8.34949383, 13.27994304,  9.84314712,  8.77913428,  7.50564223,
       10.14044489, 10.57302546,  9.75200848, 10.60226614, 12.78327037,
        4.978704  ,  5.29755487,  7.49076899,  5.16077764, 10.84493732,
        7.27759233, 10.48721382, 10.31613269,  6.59176018, 14.05687815,
       10.31584102, 12.58006038,  7.16747878, 11.69264243,  8.85939626,
        5.91408504, 11.75438006, 11.83570132,  8.7907923 ,  7.6374603 ,
        6.16512856,  6.53290663, 14.0393949 ,  5.13182506, 12.62711252,
        6.16512856, 11.00694688,  8.7907923 , 12.77402351,  8.53014915,
       10.98076873,  8.09704052, 11.51557166,  7.8575648 , 12.26997165,
        5.91222346,  5.40689332,  4.91652556, 12.3344536 , 11.0050379 ,
       11.22993218,  8.92549752,  8.8353674 ,  7.50500064, 13.92390689,
        5.52416368,  8.29640613,  5.05444288,  5.5205189 , 11.27656488,
        5.92281142,  8.34949383, 11.54071859,  9.17595899, 13.59673406,
        6.49538841, 13.72594022,  6.27905292,  6.25586297,  8.94395628,
        7.57685908, 13.58101794,  5.16077764,  5.64464807,  9.54154096,
       11.4424816 ,  6.00663314,  6.65259775, 12.44177641, 13.26209545,
        6.46356806, 11.83075005,  7.56617037, 10.74655346, 13.3426541 ,
        9.09119351])}
{'P': tensor([[8.3993e-03, 1.6974e-01, 1.5272e-01,  ..., 3.4242e-05, 3.4409e-05,
         5.7196e-06],
        [1.1362e-04, 1.2268e-02, 7.8651e-03,  ..., 1.1429e-04, 2.2712e-05,
         5.3619e-06],
        [1.8080e-04, 2.0702e-02, 1.7915e-02,  ..., 1.6582e-04, 3.6912e-05,
         7.6737e-06],
        ...,
        [1.8538e-06, 1.8456e-06, 1.8489e-06,  ..., 2.7210e-02, 9.1703e-07,
         1.3857e-05],
        [1.0350e-03, 1.4924e-03, 1.7809e-03,  ..., 2.1211e-05, 4.5812e-02,
         9.1382e-06],
        [5.8167e-03, 2.6147e-03, 2.4656e-03,  ..., 6.9817e-06, 1.7865e-04,
         4.3838e-02]]), 'X': tensor([[ -8.8058,   2.4850,  -1.7425],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.3419,   0.5972,  -1.2841],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.2199,   4.6861,  -1.9304],
        [  8.2154,  -8.4262,   8.0171],
        [ 12.5763,   0.1333,  -9.2650],
        [  5.4242, -10.1261,   3.9595],
        [  8.1043,  -8.5164,  -7.9290],
        [ -0.8885,   7.9160,   0.0571],
        [  6.0140,  -1.0661,  -2.5040],
        [  7.9816,  10.8938,  -0.5378],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.0729,   7.2878,   7.0880],
        [ -9.3739,   0.3982,  -4.2564],
        [  3.3633,  13.4794,   0.5193],
        [  6.3505,   0.6235,  -8.5849],
        [ -8.3539,  -5.5111,   0.3373],
        [ -9.4366,  -0.1189,  -2.6736],
        [  3.9022,  13.2906,   5.5788],
        [  3.1756, -10.8247,   6.8882],
        [ -1.3178, -10.7939,   0.4236],
        [  6.2905,   1.9366,   2.1169],
        [ -3.9700,   7.3453,  -4.1486],
        [ 12.4597,  -0.9136,  -4.8763],
        [  7.1821,  -9.1845,   9.8548],
        [  6.2022,   2.4092,   3.0349],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.0937,  13.8293,  -2.3533],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.3266,   0.3292,   9.3502],
        [  1.7903,  -4.7467,   0.1168],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 12.6038,   0.7665,  -3.4439],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.0574,   1.7801,   8.8986],
        [  6.3360,   0.4258,  -9.2509],
        [  4.5057,  -3.3894,  -0.8160],
        [ -9.4157,   0.0821,  -5.0025],
        [  3.3839,  -4.1966,   8.3574],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.3535,  -5.5118,   9.8399],
        [  7.7750,  11.0665,   2.8364],
        [  5.3858,   4.4260,  -9.6769],
        [  6.9284,  11.7061,  -0.3732],
        [  6.2804,   2.0019,   0.8607],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.3211,   1.7017,  -0.2164],
        [ 12.1588,   4.3364,  -8.0867],
        [ -8.1666,   3.7653,   3.6075],
        [  4.5503,   5.5359,   7.4735],
        [  1.3670,  -4.7990,   3.2603],
        [  6.3211,   1.7017,  -0.2164],
        [  0.4996,   7.7738,  -4.4711],
        [ -9.1178,   1.5784,   2.1410],
        [ -0.1963,   7.8765,  -1.1371],
        [  6.0310,  12.2795,  -6.4528],
        [ -0.6323,   7.9085,  -5.9544],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.4716,  -0.6673,   7.6068],
        [ -7.9667,   4.0814,  -0.6397],
        [  4.0457,  -3.7737,  -0.4704],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4196,   0.0478,   9.4140],
        [ -7.4733,   4.7555,   6.2490],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 10.6673,  -5.6649,  -4.1646],
        [  6.1752,   2.5242,   3.5874],
        [  6.1973,  -0.4372,   6.5720],
        [  6.2242,  -0.3171,   7.4978],
        [  2.4639, -10.9431,  -7.3718],
        [  2.3367,   7.1676,  -9.5263],
        [  6.3205,   1.7068,  -3.8619],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.2659,  13.7894,   6.5436],
        [ -8.6229,   2.9068,   2.4429],
        [ -8.0611,  -6.0302,   4.6962],
        [  4.4039,   5.6903,   8.7366],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -1.3178, -10.7939,   0.4236],
        [ 11.0275,  -5.0666,   8.4154],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.2770,  -0.0379,   2.4195],
        [  6.1066,  -0.7799,  -4.1746],
        [ 12.2328,  -2.0376,  -1.1234],
        [  2.3589,  13.7668,  -3.6501],
        [  7.8982,  -8.6778,   0.0239],
        [ -8.6681,  -4.8556,   9.2518],
        [  6.9956,  11.6590,  -7.6876],
        [  6.3535,   0.6792,  -7.5101],
        [ -4.4779,  -9.5609,  -2.4505],
        [  6.2247,  -0.3150,  -9.3657],
        [  9.0129,  -7.7072,   9.1400],
        [  2.0870,  -4.6877,   9.6823],
        [  2.0243,   7.3086,   7.5432],
        [  7.0171,  11.6439,  -5.0178],
        [  3.2159, -10.8166,  -1.0406],
        [ -7.5401,   4.6718,   9.0033],
        [  0.1497,   7.8334,  -5.4412]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0785, 0.0834,  ..., 0.2639, 0.7483, 0.5874],
        [0.0785, 0.0000, 0.0309,  ..., 0.3385, 0.7574, 0.6647],
        [0.0834, 0.0309, 0.0000,  ..., 0.3306, 0.7269, 0.6694],
        ...,
        [0.2639, 0.3385, 0.3306,  ..., 0.0000, 0.6848, 0.3849],
        [0.7483, 0.7574, 0.7269,  ..., 0.6848, 0.0000, 1.0037],
        [0.5874, 0.6647, 0.6694,  ..., 0.3849, 1.0037, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.14973319,  9.36378438,  9.36093562,  9.43737523,  9.5014146 ,
        9.54154096,  7.01475376, 11.76830981, 12.5769659 , 11.48734978,
       11.75618513,  7.96575782,  6.10774544, 13.50484539, 13.14777433,
        7.57685908,  9.38232121, 13.89264822,  6.38105015, 10.00794468,
        9.43737523, 13.85158417, 11.28093507, 10.87409005,  6.58184474,
        8.34949383, 12.49317563, 11.65923396,  6.65368551, 11.63502679,
       13.98691426, 12.84056604,  6.33517349,  5.07306274,  9.75075273,
       12.62711252,  7.98469166,  9.24501829,  9.23071698,  6.35029274,
        5.63824971,  9.41605684,  5.39098669, 10.52357479, 10.00802627,
       13.5247215 ,  6.97106912, 13.60275211,  6.59176018,  9.09661212,
        6.54615887, 12.9089559 ,  8.99276564,  7.16598273,  4.98989906,
        6.54615887,  7.78980299,  9.25336371,  7.8788965 , 13.68063098,
        7.9337647 ,  4.8763135 ,  9.49511524,  8.95133961,  5.53255882,
       11.05413545,  9.5014146 ,  9.41970031,  8.8580671 ,  9.5014146 ,
       12.07818871,  6.67121748,  6.21275016,  6.2322769 , 11.21703654,
        7.53884104,  6.54693987,  8.01009083, 13.97430191,  9.09963894,
       10.06703358,  7.19534146,  9.5014146 , 10.87409005, 12.13568598,
        9.4997987 ,  6.83515201, 12.18045978, 10.4977132 ,  6.27715105,
        6.1561646 , 12.4013153 , 13.96746816, 11.73397309,  9.9353921 ,
       13.59673406,  6.38968357, 10.5575601 ,  6.23262289, 11.85890664,
        5.13125513,  7.58378655, 13.59480906, 11.2845608 ,  8.87007495,
        7.83486943])}
{'P': tensor([[8.3425e-03, 1.9234e-01, 1.9202e-01,  ..., 2.4141e-06, 8.5193e-05,
         3.5085e-05],
        [3.4219e-04, 1.8407e-02, 4.4632e-01,  ..., 4.5222e-06, 2.4517e-04,
         9.5432e-05],
        [3.3807e-04, 6.6027e-04, 3.2372e-02,  ..., 6.0784e-06, 4.7497e-04,
         1.8903e-04],
        ...,
        [1.2938e-03, 1.2919e-03, 9.6339e-04,  ..., 1.1185e-01, 1.9374e-04,
         7.4091e-05],
        [1.1551e-04, 1.1830e-04, 1.1302e-04,  ..., 2.7459e-04, 1.1760e-01,
         1.4971e-01],
        [1.0146e-04, 9.3709e-05, 8.4934e-05,  ..., 2.4481e-04, 4.3162e-03,
         9.2423e-02]]), 'X': tensor([[ -8.8154,   2.4611,   2.5198],
        [ -9.1178,   1.5784,   2.1410],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.5954,  13.0099,  -4.4439],
        [  1.1274,  14.0116,   8.1753],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.0560,  -0.9421,  -5.0334],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3549,   0.7096,   6.9198],
        [  6.3569,   0.7591,   6.5011],
        [  8.5716,  10.3596,   6.8818],
        [  6.2555,   2.1474,   4.1685],
        [  2.3367,   7.1676,  -9.5263],
        [ -8.7735,   2.5640,  -6.3313],
        [ -6.8618,  -7.6322,   3.7027],
        [ -4.0409,   7.3169,  -3.5779],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.6260,  -1.9401,   3.0259],
        [ -9.4697,  -0.6157,  -6.8472],
        [  0.5250, -11.0306,   5.5358],
        [  4.5325,  -3.3643,   6.7699],
        [  6.3505,   0.6235,  -8.5849],
        [  2.6970,   6.9826,   9.7350],
        [ 12.4013,  -1.2597,  -7.6474],
        [ 12.4691,   2.8525,  -5.5110],
        [ -7.7239,   4.4298,   3.1216],
        [  7.5064,  -8.9651,  -0.0994],
        [ -4.6393,   7.0461,   1.1624],
        [  6.1066,  -0.7799,  -4.1746],
        [  1.8447,  -4.7372,   8.1235],
        [ -0.2586,   7.8825,  -9.9840],
        [  0.9659,  -4.8144,  -0.9594],
        [  6.8093,  -9.4178,   8.4244],
        [  1.3339,   7.5627,  -0.9034],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.1231,  -3.5739,   8.4708],
        [ -2.1533, -10.5778,   1.0112],
        [  6.0305,   3.0296,   0.1375],
        [ -9.4756,  -0.8166,  -9.6021],
        [  5.7845,   3.6652,   0.9175],
        [ -7.9667,   4.0814,  -0.6397],
        [ -3.4100,   7.5436,  -7.6980],
        [  6.3604,   0.9042,  -6.5901],
        [  2.7295,  13.6703,  -8.9328],
        [  3.9913,  -3.8139,  -6.8311],
        [ -9.4541,  -0.3327,   8.8246],
        [  1.5662,  13.9375,   4.5142],
        [ 10.2470,   8.4054,  -0.1177],
        [  6.6319,  11.9066,   3.5165],
        [  6.9693,  -9.3200,  -8.3822],
        [  5.5158,  -2.1337,   3.5246],
        [  4.4049,   5.6892,  -4.4394],
        [  2.5469,  13.7192,  -9.6935],
        [  4.6231,   5.4556,   8.4973],
        [  9.2692,   9.6370,   7.0746],
        [ -7.8280,   4.2848,  -7.1955],
        [  3.9663,   6.1018,  -5.4104],
        [  0.5960, -11.0335,   3.1144],
        [  2.3373,  13.7721,  -0.2442],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.8325,  -1.5222,  -2.1341],
        [  3.9770, -10.6341,   8.0783],
        [ -5.9940,   6.1985,   5.9566],
        [  6.9433,  -9.3362,  -0.3421],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.1178,   1.5784,   2.1410],
        [ 12.2914,   3.8037,   6.0978],
        [  6.0140,  -1.0661,  -2.5040],
        [ 12.3609,   3.4769,  -8.5762],
        [  3.9770, -10.6341,   8.0783],
        [ -8.7078,   2.7182,   1.8146],
        [  0.1252, -11.0062,   7.2564],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.1755,   2.5233,  -2.0042],
        [ -5.9987,   6.1949,  -6.2114],
        [ -9.1624,  -3.4286,   2.0647],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.2038,   2.4020,  -3.1692],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -6.3252,  -8.1760,  -1.0891],
        [  9.0424,  -7.6779,  -7.5286],
        [ -4.4779,  -9.5609,  -2.4505],
        [  1.1644, -11.0394,   7.8511],
        [  5.9587,  12.3216,   9.1695],
        [ -5.6651,   6.4380,   1.3660],
        [ -6.4368,   5.8356,   3.3014],
        [ -0.9715, -10.8630,   0.3555],
        [  1.6308,  13.9254,  -2.0984],
        [ 11.7795,  -3.4697,   3.5624],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.0955,   2.8210,  -4.3321],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.5829,  -4.5466,  -8.4015],
        [ 10.7566,   7.6212,  -2.3279],
        [  5.9026,   3.3837,  -0.6573],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.6260,  -1.9401,   3.0259],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.8363,   2.4086,  -5.6296],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.0457,  -3.7737,  -0.4704]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.1247e-02, 6.9630e-02,  ..., 8.1249e-01, 7.6396e-01,
         7.5462e-01],
        [5.1247e-02, 2.1453e-04, 3.7217e-02,  ..., 8.1457e-01, 7.9046e-01,
         7.8880e-01],
        [6.9630e-02, 3.7217e-02, 0.0000e+00,  ..., 8.5139e-01, 8.2552e-01,
         8.2107e-01],
        ...,
        [8.1249e-01, 8.1457e-01, 8.5139e-01,  ..., 2.1453e-04, 3.9730e-01,
         5.3731e-01],
        [7.6396e-01, 7.9046e-01, 8.2552e-01,  ..., 3.9730e-01, 0.0000e+00,
         1.4457e-01],
        [7.5462e-01, 7.8880e-01, 8.2107e-01,  ..., 5.3731e-01, 1.4457e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.15253344,  9.25336371,  9.3349858 ,  9.3109884 ,  9.24501829,
        9.60412735, 13.79762415, 14.05687815,  4.8763135 ,  6.12886024,
       11.33837396,  6.39438044,  6.4020404 , 13.44593519,  6.61386436,
        7.53884104,  9.14044984, 10.26327895,  8.35856487,  9.54154096,
        5.95111804,  9.48969978, 11.04313641,  5.64464807,  6.38105015,
        7.48538905, 12.46513646, 12.79126785,  8.90405823, 11.69264243,
        8.43626114,  6.1561646 ,  5.08372149,  7.88677538,  4.91038884,
       11.62158554,  7.67939546,  9.75401449,  9.45995955,  9.79814997,
       10.794755  ,  6.74872981,  9.51074763,  6.84796202,  8.95133961,
        8.27853076,  6.42439344, 13.94008948,  5.5205189 ,  9.45995955,
       14.02526505, 13.25336001, 13.62896593, 11.63765273,  5.91408504,
        7.19513806, 13.95361098,  7.15100303, 13.37121923,  8.92395381,
        7.27759233, 11.04953767, 13.96905756,  4.978704  ,  6.02789634,
       11.35345345,  8.62260327, 11.63502679,  7.822854  ,  9.25336371,
       12.86648771,  6.10774544, 12.84056604, 11.35345345,  9.12220271,
       11.00694688,  4.978704  ,  6.67108662,  8.62329466,  9.78285424,
        7.88677538,  6.65259775,  9.78147542, 10.33712258, 11.86239961,
       10.5575601 , 11.10066126, 13.68672855,  8.57560275,  8.68832398,
       10.90638066, 14.02058621, 12.27991991,  5.83560009,  6.7166297 ,
       10.4606265 ,  5.22903117, 13.18278606,  6.80370254, 12.16901615,
        5.95111804,  7.09887557,  9.15866607,  8.05242591,  5.04028538,
        5.53255882])}
{'P': tensor([[1.2508e-02, 1.8318e-01, 1.4891e-01,  ..., 1.2409e-06, 5.1989e-06,
         2.0847e-07],
        [1.1545e-04, 1.3601e-02, 1.9589e-01,  ..., 1.1208e-06, 9.3411e-06,
         1.6761e-07],
        [2.9785e-04, 1.5153e-03, 5.4550e-02,  ..., 3.5101e-06, 3.5262e-05,
         5.6119e-07],
        ...,
        [7.7894e-04, 5.3845e-04, 3.4457e-04,  ..., 3.2278e-01, 7.3372e-04,
         7.9139e-03],
        [2.3096e-05, 2.0710e-05, 1.6371e-05,  ..., 7.1857e-03, 9.9660e-02,
         1.1429e-03],
        [3.0927e-04, 1.2770e-04, 7.6675e-05,  ..., 7.3338e-03, 1.6621e-05,
         9.9460e-02]]), 'X': tensor([[ -8.8363,   2.4086,  -5.6296],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.3076,   0.7852,  -6.7010],
        [ -6.2668,   5.9808,   2.5614],
        [  0.5250, -11.0306,   5.5358],
        [  0.6470, -11.0352,  -1.1931],
        [  4.5814, -10.4485,   2.5496],
        [  5.2199,   4.6861,  -1.9304],
        [ -8.8154,   2.4611,   2.5198],
        [  0.5250, -11.0306,   5.5358],
        [  0.6396,  14.0772,  -6.3786],
        [  0.7145,  -4.8071,  -2.3766],
        [  4.6992,   5.3690,   3.0935],
        [  2.4572,   7.1085,   7.7376],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.6022,  -1.9833,   8.3081],
        [  2.1504,  -4.6726,   8.7908],
        [  2.5010,  -4.5736,  -0.8989],
        [  3.2366,  -4.2732,   5.1532],
        [  0.6470, -11.0352,  -1.1931],
        [  2.4149,  -4.6005,  -4.3339],
        [  8.9284,  10.0036,  -6.9210],
        [ -4.4503,   7.1378,   3.3751],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -5.0550,   6.8231,  -9.2502],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.4714,  -1.3301,  -3.1998],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.4877,  -0.7222,   8.8347],
        [ 11.4495,   6.3026,   6.8257],
        [ -1.1885, -10.8211,   0.0865],
        [  5.5294,  -2.1106,   9.0626],
        [ -2.0229, -10.6163,   8.6930],
        [  5.9003,  12.3550,   2.1917],
        [  3.3839,  -4.1966,   8.3574],
        [  0.1532,  -4.7422,  -7.0145],
        [  1.9503,  -4.7172,  -6.2313],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.3569,   0.7591,   6.5011],
        [ -5.0132,  -9.2234,  -8.3531],
        [  3.3633,  13.4794,   0.5193],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.0482, -10.8487,  -5.9066],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.2589,  -1.9312,   9.8657],
        [  3.7992,  -3.9474,  -8.9019],
        [  8.0877,  -8.5297,  -1.1050],
        [  3.1271, -10.8343,   2.6442],
        [  3.9663,   6.1018,  -5.4104],
        [ -1.6334, -10.7207,   7.1397],
        [ -2.6485, -10.4151,   0.8766],
        [  0.9445,   7.6736,   0.5095],
        [  2.3376,  -4.6231,   9.8436],
        [  5.8325,  -1.5222,  -2.1341],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 12.2311,  -2.0445,   1.3545],
        [  5.9548,  -1.2273,  -2.6520],
        [ 12.1940,  -2.1881,   7.1499],
        [  6.3413,   0.4883,  -6.8416],
        [ -6.2200,  -8.2737,   5.5079],
        [  2.2644,  -4.6434,   9.7508],
        [  5.1667,  -2.6543,   1.2879],
        [  5.2199,   4.6861,  -1.9304],
        [ -6.2200,  -8.2737,   5.5079],
        [ -6.2200,  -8.2737,   5.5079],
        [  6.3236,   0.3011,  -4.6668],
        [ -8.8400,   2.3991,   1.3457],
        [  2.7295,  13.6703,  -8.9328],
        [  4.0078,  -3.8018,  -6.2168],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -7.4808,   4.7462,   2.0266],
        [ 11.2517,  -4.6495,   2.9800],
        [ -9.4501,  -1.7133,   2.2980],
        [  3.9895,   6.0816,  -6.5235],
        [  0.1677,  -4.7447,  -6.1529],
        [ -8.5607,  -5.0938,  -5.4531],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.3516,   0.6425,   0.3425],
        [  0.2571,  -4.7594,   1.5618],
        [ -1.9487,   7.8590,  -3.3330],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.0054,  -1.0902,   8.3079],
        [  8.5093,  -8.1764,  -3.2381],
        [ 11.4703,  -4.2006,   7.4086],
        [  4.3523, -10.5232,   8.2138],
        [ 12.2692,  -1.8882,   3.9253],
        [  9.8082,  -6.8396,   4.9615],
        [  6.3236,   0.3011,  -4.6668],
        [  6.3360,   0.4258,  -9.2509],
        [ 10.5057,  -5.9099,  -5.1599],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.0719, -10.8449,  -4.1673],
        [  2.2860,   7.1916,  -6.1011],
        [  2.5128, -10.9365,   9.9645],
        [  0.3038,   7.8092,   0.4164],
        [  2.4572,   7.1085,   7.7376],
        [  4.9957,  -2.8705,   6.5978],
        [  5.6928,  12.4710,   3.6001]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 9.4550e-02, 1.2767e-01,  ..., 1.1310e+00, 1.1823e+00,
         1.4255e+00],
        [9.4550e-02, 0.0000e+00, 4.6743e-02,  ..., 1.1726e+00, 1.2353e+00,
         1.4889e+00],
        [1.2767e-01, 4.6743e-02, 0.0000e+00,  ..., 1.2189e+00, 1.2821e+00,
         1.5352e+00],
        ...,
        [1.1310e+00, 1.1726e+00, 1.2189e+00,  ..., 0.0000e+00, 1.7590e-01,
         4.7321e-01],
        [1.1823e+00, 1.2353e+00, 1.2821e+00,  ..., 1.7590e-01, 0.0000e+00,
         3.0917e-01],
        [1.4255e+00, 1.4889e+00, 1.5352e+00,  ..., 4.7321e-01, 3.0917e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.15866607,  9.41605684,  9.50654967,  9.4997987 ,  9.50913849,
        9.34061938,  8.66273171, 11.04313641, 11.05413545, 11.4088043 ,
        7.01475376,  9.15253344, 11.04313641, 14.09176259,  4.85995002,
        7.13501264,  7.52117132, 11.74901778, 11.59512194,  5.94293185,
        5.14371005,  5.21280156,  5.36061974, 11.05413545,  5.19577146,
       13.40850328,  8.41148722,  5.0713364 ,  9.5642993 ,  8.49161154,
        9.47369628, 13.08414689,  9.5642993 ,  5.27178363, 12.5085995 ,
       13.06957425, 10.88618114,  5.91854397, 10.80728788, 13.69162795,
        5.39098669,  4.74468406,  5.10444097, 10.05558893,  6.4020404 ,
       10.4977132 , 13.89264822,  6.61211224, 10.8992524 ,  7.6374603 ,
       12.41012046,  5.47866221, 11.75438006, 11.27656488,  7.27759233,
       10.84437705, 10.74655346,  7.73151309,  5.1805345 ,  6.02789634,
       12.30695098,  9.97964013, 12.40074901,  6.07992623, 12.38881635,
        6.36004393, 10.35093485,  5.16611782,  5.80860165,  7.01475376,
       10.35093485, 10.35093485,  6.33075682,  9.15976746, 13.94008948,
        5.52416368, 10.00410625,  9.99145096,  8.859416  , 12.17451685,
        9.60412735,  7.27340834,  4.74770899,  9.96152788,  6.19633139,
        6.38399789,  4.7663526 ,  8.09704052, 11.9120321 ,  6.10360073,
       11.80092578, 12.21532282, 11.38774149, 12.41366754, 11.95743558,
        6.33075682,  6.35029274, 12.05395056, 11.89855267, 11.27160035,
        7.54620849, 11.22142195,  7.81509631,  7.52117132,  5.76165934,
       13.70893492])}
{'P': tensor([[1.3155e-02, 3.6369e-01, 1.3637e-01,  ..., 9.4019e-05, 3.2775e-07,
         5.6253e-06],
        [4.7739e-04, 2.3938e-02, 3.2078e-02,  ..., 1.8616e-04, 6.5620e-07,
         1.1092e-05],
        [7.8197e-04, 6.7059e-03, 2.1807e-02,  ..., 2.1631e-04, 5.2031e-07,
         8.4399e-06],
        ...,
        [6.3634e-05, 5.9879e-05, 6.8112e-05,  ..., 7.4728e-02, 9.5344e-05,
         2.9033e-04],
        [3.0687e-06, 2.5552e-06, 1.9953e-06,  ..., 1.9128e-06, 2.3534e-02,
         1.2959e-03],
        [1.6732e-04, 1.4997e-04, 1.1557e-04,  ..., 9.8514e-05, 1.1600e-02,
         1.4089e-01]]), 'X': tensor([[ -8.8400,   2.3991,   1.3457],
        [ -9.0604,   1.7707,   1.1066],
        [ -9.1178,   1.5784,   2.1410],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.4671,  -0.5553,   0.1117],
        [ -9.4739,  -1.2497,   0.5483],
        [  0.3038,   7.8092,   0.4164],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.1140,  -3.7220,  -0.1273],
        [ -9.1907,  -3.3180,  -5.5943],
        [  2.1379,  -4.6757,   6.4981],
        [ -7.4887,   4.7364,   8.9074],
        [  5.2199,   4.6861,  -1.9304],
        [  9.6662,   9.1725,   8.1056],
        [  0.7685,  -4.8098,  -2.9310],
        [  9.1620,  -7.5573,  -9.9882],
        [  5.7784,  -1.6401,   9.5587],
        [  4.6063,  13.0051,  -3.1456],
        [  6.0467,  -9.8369,  -6.5816],
        [  3.5284, -10.7484,   5.3347],
        [  5.9587,  12.3216,   9.1695],
        [  6.3542,   0.6934,  -3.3840],
        [  6.2905,   1.9366,   2.1169],
        [ -8.6675,  -4.8569,  -9.1769],
        [  6.3550,   0.7129,   0.4917],
        [  5.0190,   4.9705,  -7.9752],
        [ -2.0445,   7.8469,  -0.5857],
        [  5.8889,  -1.3916,   8.8993],
        [  6.3119,   1.7797,   9.9186],
        [  2.8474,   6.8979,   6.6457],
        [ 12.1940,  -2.1881,   7.1499],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -3.5029,   7.5138,   8.1853],
        [  7.5309,  11.2617,   7.2656],
        [  6.0310,  12.2795,  -6.4528],
        [  5.5101,  -2.1433,   9.5442],
        [  0.9967,  -4.8144,  -9.7370],
        [ 10.9599,  -5.1849,  -3.5526],
        [  2.6078, -10.9229,   6.2869],
        [ -8.1666,   3.7653,   3.6075],
        [  6.9314,  -9.3435,  -1.2729],
        [  1.8719,   7.3713,  -3.7465],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.4297,  -5.3637,  -9.3962],
        [  9.8082,  -6.8396,   4.9615],
        [  8.6271,  -8.0715,  -7.4813],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.6019,   5.9540,  -5.1715],
        [ -8.7735,   2.5640,  -6.3313],
        [ 12.2692,  -1.8882,   3.9253],
        [ -4.0868,   7.2981,   8.1874],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 11.7349,  -3.5838,   2.1644],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -5.0550,   6.8231,  -9.2502],
        [  2.1120,  -4.6818,   8.6144],
        [  5.1793,   4.7461,   0.3899],
        [  2.7685, -10.8982,  -6.8928],
        [ 12.2589,  -1.9312,   9.8657],
        [  4.7813,  -3.1139,   8.2691],
        [  4.7482, -10.3907,   5.9149],
        [ 10.5955,   7.8824,  -3.7006],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.4573,   3.2448,   5.4370],
        [  4.5325,  -3.3643,   6.7699],
        [ 10.9394,   7.3069,  -8.9365],
        [  4.5814, -10.4485,   2.5496],
        [  1.7815,  -4.7481,   4.4289],
        [ -1.5408,   7.8978,   5.3470],
        [  3.2647, -10.8066,  -0.3768],
        [ -5.0550,   6.8231,  -9.2502],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.0400,   8.6922,  -6.2847],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -1.0012, -10.8576,   2.2867],
        [ -8.6681,  -4.8556,   9.2518],
        [ -7.2515,   5.0195,   4.6434],
        [  5.6928,  12.4710,   3.6001],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.9178,   2.1936,   7.5388],
        [  3.0876,   6.7527,  -5.3540],
        [  1.3103,  -4.8032,  -0.9300],
        [ -9.4773,  -1.0024,   5.2599],
        [  4.3523, -10.5232,   8.2138],
        [ -4.1296,   7.2803,  -2.7194],
        [  0.6470, -11.0352,  -1.1931],
        [  5.1203,  -2.7151,   6.2110],
        [  5.1519, -10.2387,   9.8191],
        [  3.9022,  13.2906,   5.5788],
        [  6.2675,  12.1379,  -8.9720],
        [ -7.4807,   4.7463,   2.0616],
        [ -1.9646,   7.8571,   3.3817],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.5900,   2.9767,   7.4811],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -6.4620,  -8.0449,   6.2719],
        [  1.6308,  13.9254,  -2.0984],
        [  2.2182,  13.8007,   5.1889],
        [ -9.4366,  -0.1189,  -2.6736],
        [  4.5814, -10.4485,   2.5496],
        [  2.1625,  -4.6697,   7.0658],
        [ 12.5832,   1.7543,  -7.6134],
        [  6.3515,   0.6418,  -6.5823]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.1209e-02, 7.7232e-02,  ..., 8.2140e-01, 1.1687e+00,
         9.1662e-01],
        [4.1209e-02, 0.0000e+00, 8.8703e-02,  ..., 8.6261e-01, 1.1896e+00,
         9.3983e-01],
        [7.7232e-02, 8.8703e-02, 2.1453e-04,  ..., 8.2254e-01, 1.2352e+00,
         9.8084e-01],
        ...,
        [8.2140e-01, 8.6261e-01, 8.2254e-01,  ..., 0.0000e+00, 1.0439e+00,
         8.3069e-01],
        [1.1687e+00, 1.1896e+00, 1.2352e+00,  ..., 1.0439e+00, 0.0000e+00,
         2.6000e-01],
        [9.1662e-01, 9.3983e-01, 9.8084e-01,  ..., 8.3069e-01, 2.6000e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.15976746,  9.23178204,  9.25336371,  9.3349858 ,  9.48337063,
        9.55593374,  7.81509631,  5.48276038,  5.54776934,  9.7712321 ,
        5.14124612,  8.86082124,  7.01475376, 13.32557133,  4.87083348,
       11.87665501,  6.00663314, 13.79676654, 11.54672755, 11.31276109,
       13.68672855,  6.39187285,  6.58184474,  9.93553424,  6.3949043 ,
        7.06372172,  8.10886021,  6.05113217,  6.55800267,  7.46248947,
       12.38881635,  9.69629   ,  8.29022197, 13.54774355, 13.68063098,
        5.91222346,  4.91652556, 12.12448912, 11.22993218,  8.99276564,
       11.63383221,  7.60529892,  7.09887557,  9.99145096, 11.95743558,
       11.81423161,  8.34949383, 11.63502679, 13.04051744,  9.14044984,
       12.41366754,  8.36445758, 12.12641969, 12.26997165, 13.79762415,
        9.62958973,  8.49161154,  5.13616224,  7.02496223, 11.24434374,
       12.41012046,  5.70591986, 11.42420508, 13.20597932, 10.1371552 ,
        9.05843201,  5.64464807, 13.15525803, 11.4088043 ,  5.0713364 ,
        8.04664886, 11.28895489,  8.49161154,  6.00663314, 13.27994304,
        9.43112957, 10.9036226 ,  9.9353921 ,  8.81930525, 13.70893492,
       11.95743558,  9.18358562,  7.42512621,  4.978704  ,  9.53015376,
       11.38774149,  8.36995953, 11.05413545,  5.79562691, 11.46176259,
       13.85158417, 13.66051884,  8.85939626,  8.09899874, 11.1913598 ,
        9.09119351, 11.81423161,  9.99145096, 10.31886066, 14.02058621,
       13.97779839,  9.43737523, 11.4088043 ,  5.14607014, 12.70489563,
        6.383897  ])}
{'P': tensor([[9.4030e-03, 3.1038e-01, 1.8613e-01,  ..., 1.0119e-02, 2.1994e-06,
         4.4418e-06],
        [4.0937e-04, 1.4191e-02, 2.5187e-01,  ..., 1.5479e-02, 3.2795e-06,
         6.8152e-06],
        [4.0624e-04, 4.6584e-04, 1.5798e-02,  ..., 2.5482e-02, 3.5848e-06,
         9.1063e-06],
        ...,
        [7.9616e-05, 8.1010e-05, 9.9602e-05,  ..., 6.8699e-02, 1.5113e-05,
         4.1328e-04],
        [1.4071e-03, 1.3748e-03, 1.2829e-03,  ..., 2.6769e-04, 1.6037e-01,
         9.5046e-06],
        [1.2569e-06, 1.2459e-06, 1.1111e-06,  ..., 6.6155e-07, 5.1386e-05,
         1.8811e-02]]), 'X': tensor([[ -8.8745,   2.3097,   3.5600],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.2707,   0.9661,   4.4863],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.0534,  -6.0429,   2.4583],
        [  2.0729,   7.2878,   7.0880],
        [ 10.3251,   8.2928,  -3.0006],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -8.9178,   2.1936,   7.5388],
        [  2.9470,  13.6085,   2.9809],
        [ 11.5198,  -4.0923,   5.5088],
        [  2.1379,  -4.6757,   6.4981],
        [  5.4429,  -2.2529,   9.7306],
        [  8.5093,  -8.1764,  -3.2381],
        [ -8.0829,  -5.9940,   5.0564],
        [ 11.6019,   5.9540,  -5.1715],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3535,   0.6792,  -7.5101],
        [  0.9392,  -4.8143,  -3.4649],
        [ 11.4703,  -4.2006,   7.4086],
        [ -2.0721,   7.8432,   6.6380],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.0965,  -5.9712,   4.3466],
        [ -6.7992,   5.4991,   9.9186],
        [  4.4633,   5.6287,  -0.7754],
        [  1.7228,  -4.7575,   8.3396],
        [  4.2612, -10.5514,  -6.6268],
        [ 11.7796,  -3.4696,   9.0088],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.4645,  -1.4880,  -2.0226],
        [  7.4132,  11.3526,   0.3322],
        [ 12.2589,  -1.9312,   9.8657],
        [  2.6970,   6.9826,   9.7350],
        [ -9.1623,  -3.4288,  -4.9961],
        [  6.3583,   1.1729,   8.7835],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.4716,  -0.6673,   7.6068],
        [  0.1047, -11.0046,   0.5188],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.3087,   1.8053,  -1.9481],
        [  6.3048,   0.1469,  -9.1691],
        [ -9.4714,  -1.3301,  -3.1998],
        [  0.4119,  14.1019,  -4.1870],
        [  1.8766,   7.3694,   1.1887],
        [  1.9312,  13.8649,   8.3563],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.4739,  -1.2497,   0.5483],
        [  4.0078,  -3.8018,  -6.2168],
        [ -5.9940,   6.1985,   5.9566],
        [ 10.7566,   7.6212,  -2.3279],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.8407,  11.7666,   3.0045],
        [ -8.0750,  -6.0072,  -5.8267],
        [  8.5716,  10.3596,   6.8818],
        [ 10.8218,  -5.4177,   6.8573],
        [  0.1047, -11.0046,   0.5188],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -2.8162,   7.7059,  -9.6885],
        [ -9.4645,  -1.4880,  -2.0226],
        [  9.3424,  -7.3685,  -3.6807],
        [  9.1820,   9.7334,  -2.7932],
        [  6.0310,  12.2795,  -6.4528],
        [  6.2675,  12.1379,  -8.9720],
        [  4.3459,  -3.5327,   0.3667],
        [  0.8940, -11.0402,  -0.9783],
        [  3.0719, -10.8449,  -4.1673],
        [  4.9337,  12.8566,  -3.6582],
        [  0.7275,  14.0667,  -0.9815],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.3589,   0.8239,  -8.4230],
        [ -7.3446,   4.9113,  -5.9356],
        [  1.1644, -11.0394,   7.8511],
        [  5.9883,   3.1536,   6.2320],
        [  5.5158,  -2.1337,   3.5246],
        [ 12.6000,   1.3839,  -3.2992],
        [ -7.4733,   4.7555,   6.2490],
        [ -6.6895,   5.6051,   2.7899],
        [ -3.3116, -10.1539,   7.5919],
        [ -5.6169,  -8.7844,   8.9057],
        [  2.3373,  13.7721,  -0.2442],
        [  5.0233,   4.9647,  -3.1788],
        [ -6.4464,   5.8272,   7.5038],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.7685, -10.8982,  -6.8928],
        [  0.7145,  -4.8071,  -2.3766],
        [ -0.8569,   7.9156,  -6.5557],
        [  4.6992,   5.3690,   3.0935],
        [  7.8375,  -8.7240,  -9.8913],
        [  1.0093,   7.6567,   2.3526],
        [ -2.2188,   7.8218,   8.6064],
        [  2.7558,  -4.4844,   0.0636],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.2592,   3.9425,   2.0383],
        [  6.9653,  -9.3225,  -3.4727],
        [  0.3715,  -4.7756,   0.2312],
        [  6.1737,   2.5305,   3.9049],
        [  0.3493,  -4.7727,  -6.5682],
        [ -8.5859,  -5.0393,   0.5476],
        [  3.0876,   6.7527,  -5.3540],
        [  9.0004,  -7.7195,   4.6945]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2123e-02, 4.4442e-02,  ..., 2.9307e-01, 8.8323e-01,
         5.6537e-01],
        [1.2123e-02, 0.0000e+00, 3.5298e-02,  ..., 2.8541e-01, 8.9225e-01,
         5.7674e-01],
        [4.4442e-02, 3.5298e-02, 2.1453e-04,  ..., 2.5025e-01, 8.9366e-01,
         6.0838e-01],
        ...,
        [2.9307e-01, 2.8541e-01, 2.5025e-01,  ..., 0.0000e+00, 9.2003e-01,
         8.3307e-01],
        [8.8323e-01, 8.9225e-01, 8.9366e-01,  ..., 9.2003e-01, 0.0000e+00,
         8.0089e-01],
        [5.6537e-01, 5.7674e-01, 6.0838e-01,  ..., 8.3307e-01, 8.0089e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.17016857,  9.17595899,  9.24501829,  9.3109884 ,  9.32094055,
        9.3349858 , 10.06850677,  7.57685908, 13.2430377 , 12.49317563,
        9.18358562, 13.92390689, 12.22503754,  5.14124612,  5.89074104,
       11.80092578, 10.06286074, 13.04051744,  9.37989631, 12.3489324 ,
        6.38968357,  4.90505566, 12.21532282,  8.11227668, 10.23946247,
       12.35625694, 12.50996828, 10.06023665,  8.74470218,  7.18354885,
        5.05981914, 11.37939695, 12.27993098,  9.7712321 ,  9.58072286,
       13.55868684, 12.41012046,  7.48538905,  9.78286733,  6.46560181,
        6.12886024,  9.49511524, 11.00508681,  5.83560009,  6.5618968 ,
        6.30647746,  9.5642993 , 14.10796466,  7.60463349, 13.99877413,
       12.42554808,  6.06416957,  9.55593374,  5.52416368,  8.62260327,
       13.18278606,  6.07992623, 13.6105595 , 10.06437598, 13.44593519,
       12.10222204, 11.00508681,  9.60080639,  8.20436764,  9.58072286,
       11.89855267, 13.38091121, 13.68063098, 13.66051884,  5.60064319,
       11.07637531, 11.27160035, 13.77075174, 14.08549802,  6.02789634,
        6.41202986,  8.8353674 , 11.10066126,  6.76791483,  5.91408504,
       12.67576792,  8.8580671 ,  8.7273569 , 10.68031351, 10.42667093,
       13.96905756,  7.06272256,  8.68979052,  9.85330646, 11.24434374,
        4.85995002,  7.96181288,  7.13501264, 11.72749923,  7.72291946,
        8.1303945 ,  5.26342203,  5.60064319, 12.87752052, 11.63724792,
        4.79003361,  6.67218687,  4.78544875,  9.95552626,  7.42512621,
       11.85743425])}
{'P': tensor([[1.3362e-02, 2.4602e-01, 2.3716e-01,  ..., 1.0288e-05, 2.2715e-06,
         2.8591e-04],
        [1.0055e-03, 5.9329e-02, 1.0602e-03,  ..., 6.6168e-05, 1.2941e-05,
         1.8567e-03],
        [4.6509e-04, 5.1718e-01, 1.5772e-02,  ..., 1.5229e-05, 3.1396e-06,
         4.2021e-04],
        ...,
        [1.1669e-05, 1.2534e-05, 1.2472e-05,  ..., 5.7359e-02, 6.7199e-02,
         8.1086e-04],
        [6.2207e-05, 5.8147e-05, 6.2192e-05,  ..., 3.4044e-03, 2.2821e-01,
         7.0458e-04],
        [1.6094e-04, 1.7815e-04, 1.7144e-04,  ..., 8.1936e-03, 3.6564e-03,
         1.2531e-01]]), 'X': tensor([[ -8.8934,   2.2597,   3.5831],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2707,   0.9661,   4.4863],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.4430,  -0.1897,   4.8383],
        [  5.6672,  -1.8631,   8.1020],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.1222,  -0.7261,   8.8204],
        [  1.3705,  13.9723,   7.9266],
        [ -1.8823, -10.6558,  -3.8952],
        [ -8.9033,   2.2329,  -8.8912],
        [ 11.2209,  -4.7092,   2.6117],
        [ 10.3251,   8.2928,  -3.0006],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -8.0534,  -6.0429,   2.4583],
        [  0.5960, -11.0335,   3.1144],
        [  5.4732,  -2.2041,  -4.3830],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.3889,  -4.6082,  -8.7291],
        [  4.3377,   5.7570,   1.3712],
        [  4.7739,  -3.1218,  -4.1188],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -4.0633,  -9.7927,  -3.0825],
        [  4.5340,  -3.3629,   8.9652],
        [ -9.4721,  -1.3097,  -7.4377],
        [  9.9756,  -6.6330,  -3.3429],
        [  7.0515,  11.6194,   8.4289],
        [ -4.2271,   7.2387,  -1.8316],
        [ 12.2692,  -1.8882,   3.9253],
        [  0.9392,  -4.8143,  -3.4649],
        [  4.1537,   5.9340,  -6.0423],
        [ -7.2515,   5.0195,   4.6434],
        [  7.9403,  -8.6455,  -8.7433],
        [  0.9809,   7.6642,  -1.4217],
        [  6.2770,  -0.0379,   2.4195],
        [ -8.0611,  -6.0302,   4.6962],
        [  0.5600, -11.0321,  -0.8660],
        [ 10.5955,   7.8824,  -3.7006],
        [ 12.5267,  -0.4125,  -8.7331],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.2767,  -4.6006,   0.7868],
        [  8.5093,  -8.1764,  -3.2381],
        [ -2.1533, -10.5778,   1.0112],
        [  7.0664,  -9.2590,   8.0879],
        [  8.5198,  10.4091,  -8.6486],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.8836,   3.4315,  -4.0327],
        [  6.1685,  12.1979,  -6.2473],
        [ 12.1379,   4.4125,  -3.8833],
        [  0.6952,  -4.8060,   8.8976],
        [  4.9957,  -2.8705,   6.5978],
        [  0.5250, -11.0306,   5.5358],
        [ -2.5056,   7.7720,   5.5878],
        [  6.3301,   0.3638,  -7.5685],
        [  7.3724,  -9.0577,  -8.9843],
        [ -7.9667,   4.0814,  -0.6397],
        [  6.2419,  -0.2313,   4.2440],
        [ -4.4503,   7.1378,   3.3751],
        [ 12.3357,   3.6001,   0.7903],
        [  6.3550,   0.7129,   0.4917],
        [  2.3200,   7.1756,   1.1938],
        [ 12.5996,   1.3971,  -1.0694],
        [  3.7992,  -3.9474,  -8.9019],
        [  5.8323,  -1.5228,   5.1132],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.9284,  11.7061,  -0.3732],
        [  3.0719, -10.8449,  -4.1673],
        [ -8.0087,   4.0173,   9.1091],
        [  9.0051,   9.9234,   5.4411],
        [  7.5930,  -8.9037,   7.2323],
        [ -5.2589,   6.7024,  -5.5809],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.3236,   0.3011,  -4.6668],
        [ -5.9872,   6.2037,  -9.4477],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.6258,  -1.9404,  -6.0715],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.5919,   0.4035,  -5.8460],
        [ -3.2053, -10.1993,  -3.8736],
        [ -7.4887,   4.7364,   8.9074],
        [ -7.4887,   4.7364,   8.9074],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 11.9695,  -2.9407,  -6.3498],
        [  2.9461,  -4.4077,  -0.4831],
        [  8.1217,  -8.5024,  -0.6149],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.0087,   4.0173,   9.1091],
        [ -3.2736, -10.1703,   3.8712],
        [  2.3261,   7.1726,   6.2522],
        [ 12.3357,   3.6001,   0.7903],
        [  1.6954,  -4.7616,   9.8198],
        [  2.7685, -10.8982,  -6.8928],
        [  5.5101,  -2.1433,   9.5442],
        [ 10.6283,  -5.7253,   7.4878],
        [  9.8082,  -6.8396,   4.9615],
        [  3.9663,   6.1018,  -5.4104],
        [ 11.4495,   6.3026,   6.8257],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -1.5451,   7.8975,   7.6104],
        [  3.4140,  -4.1803,  -8.8224],
        [  6.2247,  -0.3150,  -9.3657],
        [  0.2732,  -4.7619,  -1.1577]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.2335e-02, 3.5298e-02,  ..., 1.0118e+00, 1.0597e+00,
         7.2968e-01],
        [6.2335e-02, 2.1453e-04, 2.8654e-02,  ..., 1.0272e+00, 1.0759e+00,
         7.5854e-01],
        [3.5298e-02, 2.8654e-02, 2.1453e-04,  ..., 1.0130e+00, 1.0614e+00,
         7.3891e-01],
        ...,
        [1.0118e+00, 1.0272e+00, 1.0130e+00,  ..., 2.1453e-04, 5.0266e-02,
         3.4143e-01],
        [1.0597e+00, 1.0759e+00, 1.0614e+00,  ..., 5.0266e-02, 0.0000e+00,
         3.7771e-01],
        [7.2968e-01, 7.5854e-01, 7.3891e-01,  ..., 3.4143e-01, 3.7771e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.17595899,  9.3109884 ,  9.24501829,  9.32094055,  9.3349858 ,
        9.44486609,  5.96556248,  4.91652556,  6.16512856, 14.0393949 ,
       10.8207297 ,  9.17905491, 12.16901615, 13.2430377 , 12.32545637,
        9.96152788, 10.06850677, 11.04953767,  5.90035324, 12.25806559,
        9.48337063,  5.19065439,  7.20826219,  5.70405525,  9.85330646,
       10.60226614,  5.64500512,  9.562174  , 11.97957425, 13.59170809,
        8.38252937, 12.41366754,  4.90505566,  7.24326992,  8.81930525,
       11.73848065,  7.72668467,  6.27715105, 10.06703358, 11.04629471,
       13.20597932, 12.53345445,  5.1224079 , 12.17901033, 11.80092578,
       10.794755  , 11.64746882, 13.45124917, 10.74409834,  6.81117802,
       13.66896739, 12.91505111,  4.85603886,  5.76165934, 11.04313641,
        8.16584537,  6.34059021, 11.67875855,  8.95133961,  6.24614905,
        8.41148722, 12.85032836,  6.3949043 ,  7.54127425, 12.67680439,
        5.47866221,  6.02779201,  8.62162372, 13.60275211, 11.27160035,
        8.95983586, 13.40024224, 11.70168755,  8.51928531,  5.83560009,
        6.33075682,  8.62162372, 12.17901033,  5.9510489 , 12.35625694,
       12.59840193, 10.6910768 ,  8.86082124,  8.86082124, 10.09979747,
       12.32545637,  5.30158497, 11.75807252,  4.7663526 ,  8.95983586,
       10.68416401,  7.5403822 , 12.85032836,  5.05444288, 11.24434374,
        5.91222346, 12.07225469, 11.95743558,  7.27759233, 13.06957425,
        5.05444288,  9.88310614,  8.04718433,  5.39723241,  6.23262289,
        4.76969364])}
{'P': tensor([[1.5795e-02, 1.7728e-01, 1.2510e-01,  ..., 3.1300e-07, 1.0520e-06,
         4.4044e-05],
        [8.8421e-05, 1.4448e-02, 3.0091e-03,  ..., 1.7324e-07, 1.6352e-06,
         1.0498e-05],
        [1.6367e-04, 1.3237e-02, 1.8128e-02,  ..., 4.2555e-07, 2.2151e-06,
         1.9120e-05],
        ...,
        [6.0095e-05, 2.4961e-05, 6.0188e-05,  ..., 2.6296e-01, 1.3408e-05,
         2.9523e-03],
        [1.0606e-05, 4.1843e-06, 5.2833e-06,  ..., 1.7002e-04, 7.4910e-02,
         9.0859e-05],
        [8.8690e-03, 4.3176e-03, 1.1990e-02,  ..., 1.7987e-05, 2.2401e-06,
         4.5829e-02]]), 'X': tensor([[-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2978e-01, 7.9164e-02,  ..., 1.2784e+00, 9.7102e-01,
         5.1639e-01],
        [1.2978e-01, 0.0000e+00, 7.7393e-02,  ..., 1.3901e+00, 1.0949e+00,
         6.3891e-01],
        [7.9164e-02, 7.7393e-02, 3.0339e-04,  ..., 1.3131e+00, 1.0476e+00,
         5.6545e-01],
        ...,
        [1.2784e+00, 1.3901e+00, 1.3131e+00,  ..., 0.0000e+00, 9.9766e-01,
         7.7635e-01],
        [9.7102e-01, 1.0949e+00, 1.0476e+00,  ..., 9.9766e-01, 0.0000e+00,
         6.6366e-01],
        [5.1639e-01, 6.3891e-01, 5.6545e-01,  ..., 7.7635e-01, 6.6366e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.17905491,  9.51074763,  9.47369628,  9.34127013,  9.562174  ,
        9.66977802,  6.74872981, 11.1341798 , 11.35182589,  9.51074763,
        8.97052115,  6.4020404 , 13.96746816,  7.58378655,  7.24326992,
        6.99097013, 13.32557133, 11.018084  ,  5.10444097,  6.76791483,
        9.09502385, 10.05558893, 12.7125079 , 10.61417181,  8.53501777,
       13.69162795,  7.20826219,  5.92952538,  8.57560275,  6.34059021,
       13.72594022,  5.81481044, 13.1399181 , 11.21703654, 10.91805683,
        9.04255655, 13.59170809,  5.39723241,  5.29009836,  5.92952538,
        7.03733775,  5.47866221,  6.61386436, 13.48809595,  7.27340834,
        5.22903117,  9.50654967, 12.27991991,  8.06643487,  7.10378919,
        7.92644746, 11.92133129, 13.1399181 ,  6.81084724,  5.64865173,
       11.04629471,  6.29821673, 11.56146873, 12.10222204, 10.06023665,
        6.41202986,  6.02789634, 10.7167285 , 13.20597932, 10.65929968,
        8.06990483, 10.9036226 , 12.35625694, 11.62770332,  6.18668463,
        7.06169391, 12.41012046,  5.9510489 , 11.28215429,  5.40689332,
        6.24614905,  9.62958973,  5.77219077,  8.24808832, 10.04916678,
       13.97779839,  6.99097013,  5.47866221,  9.18690018,  5.60064319,
       11.04953767,  9.54154096, 11.28215429,  8.91551231, 12.05300943,
        5.19065439,  8.8353674 , 12.4980845 , 10.4977132 ,  5.14999799,
       12.12448912, 11.00694688,  9.99145096, 10.06850677,  4.78258092,
        9.66977802,  7.50917551, 10.33712258,  7.39599986, 12.49317563,
        8.56889738])}
{'P': tensor([[1.1411e-02, 2.0021e-01, 1.7211e-01,  ..., 3.8513e-05, 1.7791e-07,
         2.5105e-05],
        [3.3806e-04, 2.6557e-02, 5.3204e-02,  ..., 1.9053e-04, 3.4874e-07,
         9.2694e-05],
        [2.8645e-04, 4.3208e-03, 2.8765e-02,  ..., 2.0221e-04, 2.9828e-07,
         6.5233e-05],
        ...,
        [8.6804e-06, 9.2188e-06, 8.4301e-06,  ..., 1.5567e-01, 4.9626e-05,
         1.3983e-03],
        [4.0221e-04, 2.2641e-04, 1.3727e-04,  ..., 1.2135e-06, 1.6243e-01,
         3.4740e-05],
        [3.4217e-05, 4.2791e-05, 3.0693e-05,  ..., 2.8349e-03, 1.2444e-04,
         1.2308e-01]]), 'X': tensor([[ -8.9178,   2.1936,   7.5388],
        [ -9.3444,   0.5821,   6.9101],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.1706,   1.3863,   8.4032],
        [  5.0277,   4.9589,   7.0175],
        [  5.6414, -10.0303,   7.8525],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.0467,  -9.8369,  -6.5816],
        [ -2.6735, -10.4062,  -1.5911],
        [ -9.2345,  -3.1345,   3.9715],
        [ 10.2712,   8.3707,  -4.9743],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.2301,  -3.1537,  -2.7477],
        [  5.5626,  -2.0536,   8.4433],
        [ 11.2209,  -4.7092,   2.6117],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.0890,  -0.8382,   2.9166],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.9193,  -1.3176,   5.1739],
        [ -6.4822,  -8.0252,  -3.8983],
        [  0.6396,  14.0772,  -6.3786],
        [  4.2275,  -3.6320,  -0.2714],
        [ -5.9225,  -8.5355,   2.4778],
        [ -3.9456,  -9.8542,   0.7466],
        [  2.7121,   6.9743,  -8.4202],
        [  1.9503,  -4.7172,  -6.2313],
        [  6.1685,  12.1979,  -6.2473],
        [ -7.2515,   5.0195,   4.6434],
        [  8.8474,  10.0868,  -8.8644],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.7963,  -1.6018,   1.8856],
        [  7.3724,  -9.0577,  -8.9843],
        [ -6.4844,  -8.0230,   8.5209],
        [  1.3767, -11.0341,   9.9415],
        [  1.9505,  -4.7171,   9.9868],
        [  3.5965,  -4.0754,  -3.5035],
        [ -2.7572,   7.7194,   5.9580],
        [ -9.1623,  -3.4288,  -4.9961],
        [  1.7903,  -4.7467,   0.1168],
        [ -9.4117,   0.1160,   8.2574],
        [  6.0310,  12.2795,  -6.4528],
        [  6.1685,  12.1979,  -6.2473],
        [  4.4039,   5.6903,   8.7366],
        [  6.0031,  12.2958,  -6.5992],
        [ -4.1296,   7.2803,  -2.7194],
        [ -0.2586,   7.8825,  -9.9840],
        [  0.9809,  -4.8145,   6.8706],
        [ -5.0132,  -9.2234,  -8.3531],
        [  0.3715,  -4.7756,   0.2312],
        [  0.7685,  -4.8098,  -2.9310],
        [ -9.4697,  -0.6157,  -6.8472],
        [  6.3413,   0.4883,  -6.8416],
        [  5.8836,   3.4315,  -4.0327],
        [  6.3031,   1.8477,  -7.1706],
        [ -6.8618,  -7.6322,   3.7027],
        [  5.1886,  -2.6250,   6.4677],
        [  6.0031,  12.2958,  -6.5992],
        [  7.0358,  -9.2784,  -9.8479],
        [ 11.9122,   5.1397,  -1.5498],
        [  6.2038,   2.4020,  -3.1692],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.8902,  -1.3887,   4.0388],
        [ 11.5966,   5.9666,   6.0083],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.5506,  -5.1154,   6.3423],
        [  4.4633,   5.6287,  -0.7754],
        [  2.0870,  -4.6877,   9.6823],
        [  6.0890,  -0.8382,   2.9166],
        [  9.9217,  -6.7006,  -5.6728],
        [ -7.4807,   4.7463,   2.0616],
        [  5.9193,  -1.3176,   5.1739],
        [ 11.1469,   6.9237,   1.6160],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.2974,   0.8371,   2.4852],
        [  7.0642,  11.6104,  -9.8316],
        [  2.9410,   6.8428,   5.4458],
        [  7.5309,  11.2617,   7.2656],
        [  5.3118,   4.5452,   4.2114],
        [  5.5294,  -2.1106,   9.0626],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.3446,   4.9113,  -5.9356],
        [  1.7500,   7.4187,   4.4237],
        [  3.7992,  -3.9474,  -8.9019],
        [ -6.7643,  -7.7372,  -9.6663],
        [  4.5430,   5.5438,  -3.8669],
        [ -5.1398,  -9.1367,   6.2074],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.5294,  -2.1106,   9.0626],
        [  5.1886,  -2.6250,   6.4677],
        [ -4.9217,   6.8979,  -1.7667],
        [ 10.9162,  -5.2599,  -7.0812],
        [  8.9284,  10.0036,  -6.9210],
        [ 12.6059,   0.9597,  -6.3264],
        [ -8.9606,  -4.0994,   1.7886],
        [ -7.6293,  -6.6859,  -5.5161],
        [  5.8979,   3.3958,  -3.8713],
        [ 11.4743,   6.2478,  -2.1100],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.6310, -10.9195,   3.8054],
        [  2.5663,   7.0526,  -6.8769],
        [  3.1271, -10.8343,   2.6442],
        [  6.2026,  12.1774,  -1.5887],
        [  0.3354,  -4.7708,  -4.9125]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.9681e-02, 8.0925e-02,  ..., 3.7427e-01, 1.2301e+00,
         1.0383e+00],
        [6.9681e-02, 0.0000e+00, 7.3111e-02,  ..., 3.8135e-01, 1.2587e+00,
         1.0467e+00],
        [8.0925e-02, 7.3111e-02, 3.0339e-04,  ..., 4.4500e-01, 1.3099e+00,
         1.1107e+00],
        ...,
        [3.7427e-01, 3.8135e-01, 4.4500e-01,  ..., 0.0000e+00, 8.9477e-01,
         6.6578e-01],
        [1.2301e+00, 1.2587e+00, 1.3099e+00,  ..., 8.9477e-01, 0.0000e+00,
         3.9747e-01],
        [1.0383e+00, 1.0467e+00, 1.1107e+00,  ..., 6.6578e-01, 3.9747e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.18358562,  9.36256096,  9.41244865,  9.49511524,  9.45995955,
        9.27474386,  7.06169391, 11.50791859,  9.51074763, 11.54672755,
       10.74409834,  9.75200848, 13.25017211,  9.4997987 ,  9.75401449,
        5.92952538, 12.16901615,  4.78002665,  6.1463896 ,  5.57341623,
        6.06416957, 10.31613269, 14.09176259,  5.57341623, 10.38897563,
       10.61472636,  7.48311002,  5.10444097, 13.66896739,  8.81930525,
       13.41713661,  5.95297459,  6.01356004, 11.67875855, 10.31584102,
       11.11969815,  5.10446653,  5.43544626,  8.19703512,  9.78286733,
        5.07306274,  9.41244865, 13.68063098, 13.66896739,  7.19534146,
       13.6829843 ,  8.36995953,  7.88677538,  4.91336994, 10.4977132 ,
        4.79003361,  4.87083348,  9.48969978,  6.36004393,  6.81117802,
        6.56833864, 10.26327895,  5.81481044, 13.6829843 , 11.64436684,
       12.97370371,  6.65259775,  9.79039371,  6.05165248, 13.04156075,
       10.1371552 ,  9.96391045,  7.18354885,  5.13125513,  6.1463896 ,
       11.97238624,  8.85939626,  6.06416957, 13.12218013,  9.94285511,
       12.3344536 ,  9.3349858 , 13.59056613,  7.44805901, 13.54774355,
        6.99097013,  5.91854397,  8.85939626,  8.8353674 ,  7.62232307,
        5.47866221, 10.27716434,  7.16747878, 10.48315436,  9.7751189 ,
        5.91854397,  5.81481044,  8.47371868, 12.11735334, 13.40850328,
       12.64235359,  9.85384623, 10.14436912,  6.80559669, 13.06498554,
       10.27716434, 11.23201619,  7.50500064, 11.27656488, 13.6660671 ,
        4.78258092])}
{'P': tensor([[8.7309e-03, 2.8642e-01, 1.6690e-01,  ..., 2.8686e-08, 2.1917e-05,
         8.5904e-06],
        [1.6721e-04, 1.0934e-02, 3.0459e-01,  ..., 3.6025e-08, 3.1420e-05,
         8.2117e-06],
        [5.7712e-04, 7.8731e-04, 6.5541e-02,  ..., 2.2557e-07, 2.3830e-04,
         3.3303e-05],
        ...,
        [1.1068e-06, 8.7089e-07, 6.0187e-07,  ..., 2.1429e-02, 1.7568e-06,
         2.3730e-04],
        [1.6501e-04, 1.4533e-04, 1.2195e-04,  ..., 2.0198e-04, 1.3008e-01,
         5.5400e-04],
        [2.8342e-03, 2.5792e-03, 2.0883e-03,  ..., 1.9312e-06, 3.0768e-05,
         3.6860e-02]]), 'X': tensor([[-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.7577e-02, 7.5754e-02,  ..., 1.5709e+00, 7.8721e-01,
         6.9382e-01],
        [3.7577e-02, 0.0000e+00, 3.8724e-02,  ..., 1.5958e+00, 8.2424e-01,
         7.1502e-01],
        [7.5754e-02, 3.8724e-02, 3.0339e-04,  ..., 1.6145e+00, 8.5993e-01,
         7.3035e-01],
        ...,
        [1.5709e+00, 1.5958e+00, 1.6145e+00,  ..., 0.0000e+00, 1.0791e+00,
         8.8879e-01],
        [7.8721e-01, 8.2424e-01, 8.5993e-01,  ..., 1.0791e+00, 0.0000e+00,
         5.9511e-01],
        [6.9382e-01, 7.1502e-01, 7.3035e-01,  ..., 8.8879e-01, 5.9511e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.18690018,  9.27474386,  9.41244865,  9.45995955,  9.41970031,
        9.23071698, 12.86648771, 10.84493732, 13.94008948, 13.59673406,
       13.06498554,  6.10360073,  8.92549752,  9.0368458 ,  4.74468406,
        5.05774295,  8.30729352,  8.94395628,  7.58668144,  6.02588517,
       11.28093507,  8.50278896,  9.78285424,  6.99097013,  5.14371005,
       13.18278606,  8.68979052, 14.07623266,  7.8575648 ,  9.17016857,
        8.51928531,  7.06272256,  9.97964013,  8.95983586,  4.89869205,
        8.8353674 , 13.68063098,  7.62509582, 11.2845608 , 11.73567759,
       12.12641969,  7.23276582, 11.85890664, 12.25806559, 14.08549802,
        5.9510489 ,  7.66377554,  5.5205189 , 12.52875793,  7.15990676,
        4.90676242, 12.07225469, 11.9120321 , 10.84493732,  5.14371005,
       14.07623266,  8.95133961,  8.08320235, 13.15525803, 12.30858677,
        9.09963894, 12.38881635, 13.6660671 ,  6.38968357, 10.07238463,
       13.89886469,  6.49538841,  7.03733775,  8.7273569 ,  5.64500512,
        7.34250948,  8.09899874, 11.99633819,  5.07306274,  9.58347456,
       11.9702762 , 11.11969815,  4.78544875,  4.73784222,  9.99145096,
        9.96152788, 12.50996828, 10.46739073,  8.16714836,  6.39438044,
        9.58347456,  5.5205189 ,  6.19633139, 13.13205357, 11.63502679,
       12.18998195,  9.86089579,  9.14973319,  6.54693987, 13.98691426,
        9.41244865,  5.5366698 , 11.10066126, 11.9120321 ,  7.32233733,
       12.25806559, 11.9120321 , 11.57249984, 13.1399181 ,  5.81481044,
        8.31625171])}
{'P': tensor([[9.8072e-03, 2.9163e-01, 2.0219e-01,  ..., 3.9347e-04, 1.0788e-06,
         2.8834e-05],
        [2.0185e-04, 1.1181e-02, 3.1146e-01,  ..., 5.3081e-04, 1.1604e-06,
         3.6622e-05],
        [3.1884e-04, 4.2504e-04, 3.5383e-02,  ..., 2.2559e-03, 3.8466e-06,
         1.4782e-04],
        ...,
        [1.1322e-04, 1.1596e-04, 1.2351e-04,  ..., 6.4290e-02, 1.3445e-04,
         3.4337e-02],
        [4.9839e-05, 3.9541e-05, 3.0591e-05,  ..., 3.6750e-05, 7.6106e-02,
         1.2441e-04],
        [1.1027e-04, 9.8167e-05, 8.3351e-05,  ..., 1.4692e-03, 1.3548e-03,
         8.9609e-02]]), 'X': tensor([[ -8.9636,   2.0648,   8.1946],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.4196,   0.0478,   9.4140],
        [  0.6396,  14.0772,  -6.3786],
        [  8.6420,  -8.0580,   6.5781],
        [  4.4049,   5.6892,  -4.4394],
        [ 12.2592,   3.9425,   2.0383],
        [ -7.5401,   4.6718,   9.0033],
        [  1.1274,  14.0116,   8.1753],
        [  6.2883,   0.0325,   5.2692],
        [  0.6952,  -4.8060,   8.8976],
        [  9.6662,   9.1725,   8.1056],
        [ -0.7599, -10.8995,  -6.0366],
        [ -9.4196,   0.0478,   9.4140],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.1459,   1.4780,  -4.2403],
        [  9.1820,   9.7334,  -2.7932],
        [ -7.4807,   4.7463,   2.0616],
        [  7.5064,  -8.9651,  -0.0994],
        [ -2.6485, -10.4151,   0.8766],
        [ -2.5683, -10.4433,  -2.9182],
        [ -3.6381,   7.4683,   7.3544],
        [  3.1064,  13.5607,   6.7605],
        [  9.9750,  -6.6338,  -1.2440],
        [  1.0000,   7.6591,  -4.5011],
        [ -4.7334,   6.9982,  -0.4392],
        [  9.2692,   9.6370,   7.0746],
        [ -1.9600,   7.8577,   6.9121],
        [  6.3505,   0.6235,  -8.5849],
        [  5.8836,   3.4315,  -4.0327],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.5974,   7.0363,  -7.7810],
        [  6.2419,  -0.2313,   4.2440],
        [ -8.9178,   2.1936,   7.5388],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -7.9667,   4.0814,  -0.6397],
        [  3.2159, -10.8166,  -1.0406],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.5859,  -5.0393,   0.5476],
        [ -9.4309,  -0.0599,  -4.0830],
        [  2.5829,  -4.5466,  -8.4015],
        [  1.5246,  -4.7838,  -5.2095],
        [  6.3413,   0.4883,  -6.8416],
        [ 12.2592,   3.9425,   2.0383],
        [  1.8320,   7.3871,   5.1452],
        [  4.5954,  13.0099,  -4.4439],
        [  7.9141,  -8.6656,   9.6948],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.4488, -10.0934,   2.5195],
        [ 10.7566,   7.6212,  -2.3279],
        [ -7.3333,   4.9246,  -7.9212],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -6.4464,   5.8272,   7.5038],
        [  6.2790,  -0.0259,  -5.4116],
        [  2.7295,  13.6703,  -8.9328],
        [  4.9957,  -2.8705,   6.5978],
        [ -6.4620,  -8.0449,   6.2719],
        [ -8.6050,   2.9451,  -0.7012],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.1504,   4.3674,  -7.2998],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.7156,  -3.1831,  -8.0224],
        [  8.5093,  -8.1764,  -3.2381],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.3360,   0.4258,  -9.2509],
        [  0.9074,  -4.8139,   8.1674],
        [  7.8982,  -8.6778,   0.0239],
        [ -7.4602,   4.7718,  -4.6277],
        [  9.5203,   9.3483,   5.8781],
        [  0.5250, -11.0306,   5.5358],
        [  1.9312,  13.8649,   8.3563],
        [  4.8051, -10.3703,   0.5268],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -5.1377,   6.7751,  -0.4438],
        [  2.1504,  -4.6726,   8.7908],
        [  4.9152,  12.8653,  -1.4459],
        [  3.9591, -10.6391,  -6.2557],
        [ -2.1382,   7.8339,   7.3524],
        [ -4.4779,  -9.5609,  -2.4505],
        [  3.4140,  -4.1803,  -8.8224],
        [ -7.0948,   5.1940,   9.8171],
        [  3.4604,  -4.1545,   1.8091],
        [  4.0362,  13.2397,  -7.0030],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -2.8162,   7.7059,  -9.6885],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.2571,  -4.7594,   1.5618],
        [  5.6022,  -1.9833,   8.3081],
        [  4.0421,  -3.7765,  -8.3631],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.0965,  -5.9712,   4.3466],
        [  0.1252, -11.0062,   7.2564],
        [  6.3305,   1.6119,   1.1538],
        [  6.1730,  -0.5375,  -8.8861],
        [ -8.9788,   2.0208,  -3.2867],
        [ -2.0229, -10.6163,   8.6930],
        [  6.8704,  -9.3809,   7.5769],
        [  0.1206,  -4.7363,   8.7103],
        [  5.9526,   3.2525,  -4.3821],
        [  4.9957,  -2.8705,   6.5978]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.4685e-02, 7.0361e-02,  ..., 6.8766e-01, 1.1317e+00,
         7.8542e-01],
        [3.4685e-02, 0.0000e+00, 3.8724e-02,  ..., 7.2120e-01, 1.1622e+00,
         8.1951e-01],
        [7.0361e-02, 3.8724e-02, 3.0339e-04,  ..., 7.5797e-01, 1.1812e+00,
         8.5534e-01],
        ...,
        [6.8766e-01, 7.2120e-01, 7.5797e-01,  ..., 0.0000e+00, 7.9874e-01,
         1.1219e-01],
        [1.1317e+00, 1.1622e+00, 1.1812e+00,  ..., 7.9874e-01, 0.0000e+00,
         7.3088e-01],
        [7.8542e-01, 8.1951e-01, 8.5534e-01,  ..., 1.1219e-01, 7.3088e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.19837792,  9.27474386,  9.41244865,  9.45995955,  9.49511524,
        9.41970031, 14.09176259, 11.8159254 ,  7.19513806, 12.87752052,
        8.87007495, 14.05687815,  6.28835101,  4.85603886, 13.32557133,
       10.92597196,  9.41970031, 10.86693708,  9.26455795, 13.38091121,
        8.85939626, 11.69264243, 10.74655346, 10.75443106,  8.30729352,
       13.91198191, 11.97949273,  7.72415048,  8.4486771 , 13.37121923,
        8.09842704,  6.38105015,  6.81117802,  8.06990483,  7.50035355,
        6.24614905,  9.18358562, 10.1979669 ,  8.95133961, 11.2845608 ,
       12.4980845 ,  9.97220799,  9.95552626,  9.43112957,  5.22903117,
        5.0209005 ,  6.36004393, 12.87752052,  7.61088125, 13.79762415,
       11.73567759, 10.48840476, 10.66631784, 13.18278606,  8.83341608,
        9.43737523,  8.68979052,  6.27905292, 13.94008948,  5.76165934,
       10.31886066,  9.09502385, 12.44177641, 12.91143491,  5.68942572,
        5.68942572, 11.80092578,  8.20436764,  6.35029274,  4.89869205,
       11.73397309,  8.85572292, 13.3426541 , 11.04313641, 13.99877413,
       11.42947632,  6.2322769 ,  9.85330646,  8.50278896,  5.14371005,
       13.77222752, 11.35182589,  8.12043983, 10.5575601 ,  5.39723241,
        8.79284466,  5.40689332, 13.84126197,  9.562174  ,  8.20436764,
       11.86239961,  5.19986797,  4.7663526 ,  5.94293185,  5.53175275,
       10.48840476, 10.06023665, 11.00694688,  6.53250937,  6.19633139,
        9.20340031, 10.80728788, 11.62770332,  4.73784222,  6.78325151,
        5.76165934])}
{'P': tensor([[8.5525e-03, 1.9760e-01, 1.9109e-01,  ..., 2.0677e-05, 6.5714e-04,
         2.2473e-06],
        [1.2313e-04, 1.0979e-02, 1.3557e-02,  ..., 3.7744e-05, 1.3540e-03,
         4.5558e-06],
        [1.8597e-04, 8.5190e-03, 1.6984e-02,  ..., 5.1080e-05, 1.6228e-03,
         6.9994e-06],
        ...,
        [4.2060e-05, 4.0288e-05, 3.6053e-05,  ..., 8.7134e-02, 1.0386e-04,
         4.0510e-03],
        [2.8792e-05, 4.9648e-05, 4.6868e-05,  ..., 5.7204e-03, 7.5562e-02,
         3.6451e-04],
        [3.1153e-06, 2.3245e-06, 2.2514e-06,  ..., 2.4816e-04, 1.0992e-06,
         4.2947e-02]]), 'X': tensor([[ -8.9788,   2.0208,  -3.2867],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.3705,   0.4208,  -3.5815],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 11.5456,   6.0861,   8.1101],
        [  3.3474,  -4.2162,   5.2455],
        [ -0.0282,   7.8575,  -1.1746],
        [ 12.1588,   4.3364,  -8.0867],
        [ -3.2486,   7.5925,  -1.3719],
        [ -9.2687,  -2.9794,   0.6436],
        [  6.3246,   1.6696,  -0.1544],
        [  5.1793,   4.7461,   0.3899],
        [  6.3048,   0.1469,  -9.1691],
        [  2.3376,  -4.6231,   9.8436],
        [ 10.6673,  -5.6649,  -4.1646],
        [  4.6357,   5.4415,   7.9067],
        [  6.3048,   0.1469,  -9.1691],
        [ -6.4002,  -8.1048,  -0.9108],
        [  5.9912,   3.1453,  -3.9989],
        [  6.3200,   0.2690,   1.5900],
        [  6.1973,  -0.4372,   6.5720],
        [  3.2366,  -4.2732,   5.1532],
        [ -5.3328,  -8.9992,   7.3440],
        [ -3.6381,   7.4683,   7.3544],
        [  2.0039,   7.3173,  -1.5885],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.3118,   4.5452,   4.2114],
        [  5.7845,   3.6652,   0.9175],
        [ 10.4994,  -5.9193,  -7.5153],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -7.8359,   4.2735,   7.2943],
        [  0.8940, -11.0402,  -0.9783],
        [  5.0190,   4.9705,  -7.9752],
        [  6.3516,   0.6425,   0.3425],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.8407,  11.7666,   3.0045],
        [  1.8766,   7.3694,   1.1887],
        [  1.9312,  13.8649,   8.3563],
        [  7.1821,  -9.1845,   9.8548],
        [  1.9204,   7.3518,   8.4781],
        [  4.5491,  -3.3485,   6.8412],
        [  9.1409,  -7.5789,   1.5814],
        [  5.5158,  -2.1337,   3.5246],
        [  3.9770, -10.6341,   8.0783],
        [  6.3205,   1.7068,  -3.8619],
        [ -9.1749,  -3.3803,   6.4629],
        [  0.9945,  -4.8144,  -0.1897],
        [ 11.1085,  -4.9204,   0.0736],
        [ -1.9600,   7.8577,   6.9121],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.7992,  -3.9474,  -8.9019],
        [ -9.4630,  -1.5145,   0.2983],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.2608,  12.6977,   4.9166],
        [ -8.9178,   2.1936,   7.5388],
        [  3.3069,   6.6089,   8.7377],
        [ -1.6086,   7.8928,   4.3385],
        [  2.8474,   6.8979,   6.6457],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.9036,   7.6840,   9.9237],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.4749,  -4.7892,  -0.8089],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.9433,  -9.3362,  -0.3421],
        [ -1.6220,   7.8917,   0.0315],
        [  6.0054,  -1.0902,   8.3079],
        [  1.0093,   7.6567,   2.3526],
        [  5.8836,   3.4315,  -4.0327],
        [  7.1441,  11.5529,   2.3232],
        [  4.5814, -10.4485,   2.5496],
        [  7.4132,  11.3526,   0.3322],
        [ 11.9641,  -2.9568,   7.7211],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -2.8162,   7.7059,  -9.6885],
        [  5.4429,  -2.2529,   9.7306],
        [  1.5653,  13.9377,   4.1773],
        [ -9.4766,  -1.1116,  -2.1706],
        [  0.5600, -11.0321,  -0.8660],
        [ -0.0410,   7.8591,   9.3457],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.3604,   0.9042,  -6.5901],
        [  4.8653,   5.1694,  -4.0823],
        [ -3.1671,   7.6158,   6.0765],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3516,   0.6425,   0.3425],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.4049,   5.6892,  -4.4394],
        [ -8.7544,   2.6096,   3.3146],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.1379,  -4.6757,   6.4981],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -5.0943,  -9.1682,   0.5073],
        [ 11.9936,  -2.8677,   6.0614],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.3860,  -2.3418,   1.0826],
        [  8.0877,  -8.5297,  -1.1050],
        [ -8.9033,   2.2329,  -8.8912],
        [ -8.7735,   2.5640,  -6.3313],
        [ -9.4731,  -0.7120,  -6.0271],
        [  3.3474,  -4.2162,   5.2455],
        [ -7.3470,  -7.0615,   5.0066],
        [ 10.4737,  -5.9570,  -1.2108]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0647, 0.0725,  ..., 1.0238, 0.6621, 0.7114],
        [0.0647, 0.0000, 0.0380,  ..., 1.0618, 0.6364, 0.7686],
        [0.0725, 0.0380, 0.0000,  ..., 1.0904, 0.6729, 0.7838],
        ...,
        [1.0238, 1.0618, 1.0904,  ..., 0.0000, 0.9061, 0.5008],
        [0.6621, 0.6364, 0.6729,  ..., 0.9061, 0.0000, 0.9471],
        [0.7114, 0.7686, 0.7838,  ..., 0.5008, 0.9471, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.20340031,  9.37296102,  9.37989631,  9.43737523,  9.43112957,
        9.5014146 , 13.05148826,  5.38342077,  7.8575648 , 12.9089559 ,
        8.25828409,  9.73579738,  6.54127431,  7.02496223,  6.30647746,
        5.1805345 , 12.07818871,  7.14837514,  6.30647746, 10.32715715,
        6.76661595,  6.32572323,  6.21275016,  5.36061974, 10.4606265 ,
        8.30729352,  7.58668144, 11.80092578,  6.99097013,  6.84796202,
       12.05300943,  6.00663314,  9.71392179,  8.92549752, 11.07637531,
        7.06372172,  6.38399789, 10.91805683, 13.6105595 ,  7.60463349,
       13.99877413, 11.65923396,  7.5984809 ,  5.64865173, 11.87412439,
        5.91408504, 11.35345345,  6.54693987,  9.77777346,  4.9160962 ,
       12.1494023 ,  8.09842704,  5.9510489 ,  5.47866221,  9.58347456,
        8.16714836, 13.744387  ,  9.18358562,  7.39004226,  8.0550286 ,
        7.46248947,  9.4997987 ,  7.73692436, 12.59840193,  5.01114719,
       12.24467041, 11.63502679,  8.05668609,  6.10360073,  7.72291946,
        6.81117802, 13.58333857, 11.4088043 , 13.55868684, 12.32408516,
       10.00410625,  8.20436764,  5.89074104, 14.02532758,  9.54154096,
       11.04629471,  7.85919687,  8.83341608,  4.90505566,  6.42439344,
        7.09887557,  8.24808832,  6.48486171,  6.38399789,  5.41348121,
        7.19513806,  9.13507321, 10.61417181,  5.14124612,  9.70108409,
       10.48840476, 12.33167382,  5.16077764,  5.87305005, 11.75438006,
        9.17905491,  9.14044984,  9.4997987 ,  5.38342077, 10.19036081,
       12.04924159])}
{'P': tensor([[1.1177e-02, 2.1915e-01, 2.1412e-01,  ..., 4.4194e-06, 3.7732e-07,
         2.2923e-04],
        [2.8602e-04, 3.4260e-02, 2.9759e-03,  ..., 8.4587e-06, 7.3112e-07,
         4.6199e-04],
        [2.7969e-04, 1.6107e-01, 2.5264e-02,  ..., 6.1361e-06, 5.6371e-07,
         2.8625e-04],
        ...,
        [3.8050e-03, 2.0607e-03, 1.9079e-03,  ..., 1.1187e-01, 8.7368e-04,
         3.8697e-03],
        [3.8291e-04, 1.3271e-04, 1.5095e-04,  ..., 1.9376e-03, 1.7853e-02,
         2.7064e-05],
        [1.0031e-04, 2.4073e-04, 1.6744e-04,  ..., 1.6316e-05, 5.4739e-07,
         5.1413e-02]]), 'X': tensor([[ -9.0574,   1.7801,   8.8986],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.1706,   1.3863,   8.4032],
        [  3.9895,   6.0816,  -6.5235],
        [ 11.6019,   5.9540,  -5.1715],
        [  4.8653,   5.1694,  -4.0823],
        [  6.0955,   2.8210,  -4.3321],
        [  9.0004,  -7.7195,   4.6945],
        [ 10.0995,  -6.4739,   2.0974],
        [  0.9445,   7.6736,   0.5095],
        [  1.7659,   7.4127,   4.7871],
        [  6.3122,  -9.6994,  -6.3282],
        [ -2.8162,   7.7059,  -9.6885],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.8890,  -4.4316,   1.1600],
        [  5.8201,   3.5841,  -8.0609],
        [  6.3534,   1.2990,  -7.2942],
        [  2.5636, -10.9293,   9.6519],
        [  5.9912,   3.1453,  -3.9989],
        [ -7.4807,   4.7463,   2.0616],
        [ -0.2586,   7.8825,  -9.9840],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.7685,  -4.8098,  -2.9310],
        [ -1.0012, -10.8576,   2.2867],
        [ -9.3076,   0.7852,  -6.7010],
        [ 12.3025,  -1.7440,  -6.6632],
        [  2.3261,   7.1726,   6.2522],
        [ -2.0721,   7.8432,   6.6380],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.5175,  14.0910,   4.6540],
        [  5.1667,  -2.6543,   1.2879],
        [ -5.9987,   6.1949,  -6.2114],
        [  5.1472,  -2.6801,   8.1194],
        [  7.0515,  11.6194,   8.4289],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.1886,  -2.6250,   6.4677],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.1625,  -4.6697,   7.0658],
        [ -8.8631,  -4.3728,  -1.1746],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -7.9298,   4.1367,  -3.9773],
        [  5.4867,  12.5816,   2.9549],
        [  8.2422,  -8.4041,  -1.4962],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.4430,  -0.1897,   4.8383],
        [  4.6063,  13.0051,  -3.1456],
        [ 11.9641,  -2.9568,   7.7211],
        [ -1.1885, -10.8211,   0.0865],
        [  9.3607,   9.5337,  -4.5412],
        [  6.1066,  -0.7799,  -4.1746],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.3549,   0.7096,   6.9198],
        [ -6.3252,  -8.1760,  -1.0891],
        [  5.3118,   4.5452,   4.2114],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.7802,  -1.6363,  -6.2782],
        [ -8.3900,   3.3727,   3.1650],
        [  6.1929,   2.4501,  -6.9352],
        [ -6.1914,  -8.2997,   8.5380],
        [ -9.3076,   0.7852,  -6.7010],
        [ -5.9940,   6.1985,   5.9566],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -7.4807,   4.7463,   2.0616],
        [  2.0937,  13.8293,  -2.3533],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -5.7347,  -8.6908,   9.2878],
        [ -8.6801,   2.7809,  -0.8443],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3542,   0.6934,  -3.3840],
        [ -5.5525,   6.5146,  -4.4984],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.3515,   0.6418,  -6.5823],
        [  6.1752,   2.5242,   3.5874],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.6426,  -4.9136,   9.4941],
        [  2.2182,  13.8007,   5.1889],
        [ -3.2736, -10.1703,   3.8712],
        [  4.5340,  -3.3629,   8.9652],
        [ -7.3470,  -7.0615,   5.0066],
        [ 11.4743,   6.2478,  -2.1100],
        [ -9.4714,  -1.3301,  -3.1998],
        [  2.3200,   7.1756,   1.1938],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.2659,  13.7894,   6.5436],
        [ -9.1813,  -3.3550,   1.8827],
        [ -7.4602,   4.7718,  -4.6277],
        [  4.0457,  -3.7737,  -0.4704],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.1196,  -2.7160,   4.2705],
        [ -7.7239,   4.4298,   3.1216],
        [  2.0449,  13.8402,  -7.1642],
        [  5.6022,  -1.9833,   8.3081],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.2911,  -2.4824,  -3.6877],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -8.6112,   2.9319,  -7.9943],
        [  9.1820,   9.7334,  -2.7932],
        [ -0.5739,   7.9056,   0.6382],
        [  2.3261,  13.7749,   5.9121],
        [ -9.4378,  -1.8606,  -2.5668]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.8998e-02, 6.2963e-02,  ..., 7.4700e-01, 1.0443e+00,
         8.2973e-01],
        [6.8998e-02, 0.0000e+00, 1.0311e-02,  ..., 7.9466e-01, 1.1133e+00,
         8.1577e-01],
        [6.2963e-02, 1.0311e-02, 0.0000e+00,  ..., 7.9564e-01, 1.1068e+00,
         8.2471e-01],
        ...,
        [7.4700e-01, 7.9466e-01, 7.9564e-01,  ..., 4.7405e-06, 7.7937e-01,
         7.4170e-01],
        [1.0443e+00, 1.1133e+00, 1.1068e+00,  ..., 7.7937e-01, 0.0000e+00,
         1.4823e+00],
        [8.2973e-01, 8.1577e-01, 8.2471e-01,  ..., 7.4170e-01, 1.4823e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.23071698,  9.45995955,  9.41970031,  9.41244865,  9.49511524,
        9.27474386,  7.27340834, 13.04051744,  7.09887557,  6.7166297 ,
       11.85743425, 11.99633819,  7.73151309,  7.62011734, 11.57249984,
        8.20436764,  9.43737523,  5.29009836,  6.83515201,  6.48486171,
       11.22596994,  6.76661595,  8.85939626,  7.88677538,  9.50654967,
        4.87083348, 10.9036226 ,  9.34061938, 12.42554808,  7.5403822 ,
        8.11227668, 11.89855267, 14.10045797,  5.80860165,  8.62329466,
        5.80312949, 13.59170809,  6.00732193,  5.81481044,  9.41605684,
        5.14607014,  9.88310614,  5.53255882,  9.97964013,  8.94395628,
       13.72594022, 11.77124373, 13.79762415,  9.44486609, 13.79676654,
       12.32408516, 10.88618114, 13.36092624,  6.1561646 ,  7.8575648 ,
        6.39438044, 10.33712258,  6.99097013,  6.02789634,  6.00732193,
        9.04255655,  6.65991877, 10.35464916,  9.34061938,  8.62260327,
       12.46513646,  8.85939626, 13.98691426, 12.27835996, 10.412299  ,
        9.11473391, 12.27993098,  6.39187285,  8.55982105,  8.89458273,
        6.383897  ,  6.67121748,  6.2322769 ,  9.94173708, 13.97779839,
       10.68416401,  5.64500512, 10.19036081, 13.06498554,  9.5642993 ,
        7.54127425, 12.84056604, 13.97430191,  9.7751189 ,  8.85572292,
        5.53255882,  8.53501777, 12.77402351,  9.64157461,  5.79543038,
        8.90405823, 13.99048094,  5.94293185, 12.97370371,  5.84450127,
       12.00995898,  9.09661212, 13.38091121,  7.92644746, 13.96987806,
        9.61942445])}
{'P': tensor([[1.2549e-02, 1.5444e-01, 1.5294e-01,  ..., 9.7481e-06, 8.5606e-07,
         1.8211e-03],
        [9.0772e-05, 1.5155e-02, 3.6923e-04,  ..., 3.2642e-05, 1.7968e-06,
         1.0971e-04],
        [9.4052e-05, 2.0373e-01, 9.8975e-03,  ..., 1.5445e-05, 1.0001e-06,
         1.2213e-04],
        ...,
        [4.2932e-06, 3.3090e-06, 3.2618e-06,  ..., 8.9692e-02, 5.7722e-02,
         5.3548e-06],
        [7.3880e-06, 2.9832e-06, 3.5434e-06,  ..., 5.4574e-06, 4.2995e-02,
         8.6521e-06],
        [2.0915e-02, 6.6855e-02, 4.5589e-02,  ..., 2.0019e-05, 1.5488e-06,
         2.5267e-02]]), 'X': tensor([[ -9.0604,   1.7707,   1.1066],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.4671,  -0.5553,   0.1117],
        [ -9.4630,  -1.5145,   0.2983],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.3616,  -2.4749,   0.1793],
        [ -9.4735,  -1.2620,  -3.6742],
        [  6.3604,   0.9042,  -6.5901],
        [  6.3587,   1.1596,  -3.6629],
        [  1.1006,   7.6318,  -2.4877],
        [ 11.1469,   6.9237,   1.6160],
        [  6.2038,   2.4020,  -3.1692],
        [ 10.3251,   8.2928,  -3.0006],
        [ -6.0175,  -8.4541,  -9.8408],
        [  0.3715,  -4.7756,   0.2312],
        [  1.9503,  -4.7172,  -6.2313],
        [  6.3460,   1.4252,   1.2291],
        [ -2.7572,   7.7194,   5.9580],
        [  5.8323,  -1.5228,   5.1132],
        [ -5.1377,   6.7751,  -0.4438],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4501,  -1.7133,   2.2980],
        [ -3.5520,   7.4976,   9.1522],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.1886,  -2.6250,   6.4677],
        [  9.3607,   9.5337,  -4.5412],
        [  1.1350,  14.0104,   0.6321],
        [  1.1644, -11.0394,   7.8511],
        [ -5.1974,   6.7396,   3.3484],
        [  1.8719,   7.3713,  -3.7465],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.8930,  12.3592,  -8.9201],
        [  2.8474,   6.8979,   6.6457],
        [  4.3459,  -3.5327,   0.3667],
        [  9.9756,  -6.6330,  -3.3429],
        [  0.6952,  -4.8060,   8.8976],
        [  6.2804,   2.0019,   0.8607],
        [  6.0560,  -0.9421,  -5.0334],
        [ 12.4877,  -0.7222,   8.8347],
        [  6.0788,  -0.8710,  -6.8719],
        [  4.7739,  -3.1218,  -4.1188],
        [  9.3424,  -7.3685,  -3.6807],
        [ -5.7347,  -8.6908,   9.2878],
        [  4.4915,   5.5990,   7.0405],
        [  0.3724,   7.7974,  -6.8722],
        [ -2.2188,   7.8218,   8.6064],
        [ -9.1813,  -3.3550,   1.8827],
        [  0.7958,  -4.8109,  -8.2248],
        [ -8.7735,   2.5640,  -6.3313],
        [ 12.5650,   2.0132,   4.6272],
        [  4.7482, -10.3907,   5.9149],
        [  2.1504,  -4.6726,   8.7908],
        [  0.1497,   7.8334,  -5.4412],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.9410,   6.8428,   5.4458],
        [  7.1697,  11.5343,  -0.7276],
        [ 11.9641,  -2.9568,   7.7211],
        [ -9.2345,  -3.1345,   3.9715],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.3602,  -2.3808,   7.2098],
        [  6.0305,   3.0296,   0.1375],
        [ 12.1588,   4.3364,  -8.0867],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -0.8885,   7.9160,   0.0571],
        [ -8.9788,   2.0208,  -3.2867],
        [  0.9809,   7.6642,  -1.4217],
        [  4.2096,   5.8815,   1.9230],
        [  8.9284,  10.0036,  -6.9210],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 11.6544,   5.8273,  -3.1375],
        [  1.7228,  -4.7575,   8.3396],
        [  2.2849, -10.9655,  -7.6542],
        [  3.6703,   6.3443,  -5.4401],
        [  2.1504,  -4.6726,   8.7908],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.0277,   4.9589,   7.0175],
        [  9.5261,  -7.1675,   1.4206],
        [  2.4697,  -4.5836,   1.4593],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 10.9875,   7.2207,  -8.8549],
        [ 12.6059,   0.9597,  -6.3264],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.9217,  -6.7006,  -5.6728],
        [ -6.6895,   5.6051,   2.7899],
        [  6.3604,   0.9042,  -6.5901],
        [ -7.5401,   4.6718,   9.0033],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.3048,   0.1469,  -9.1691],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -6.8466,  -7.6488,   1.9648],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.0310,  12.2795,  -6.4528],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.6000,   1.3839,  -3.2992],
        [ -6.4464,   5.8272,   7.5038],
        [  5.4179,  -2.2925,  -3.4641],
        [  3.9895,   6.0816,  -6.5235],
        [  6.2630,   2.1057,  -5.2136],
        [  5.1787,  -2.6383,  -4.2533],
        [ -9.4203,  -2.0340,   2.5856],
        [  7.5784,  -8.9141,   5.8410],
        [ 12.5789,   0.1722,   1.6470],
        [ -8.8745,   2.3097,   3.5600]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0880, 0.0907,  ..., 0.6585, 0.8998, 0.1934],
        [0.0880, 0.0000, 0.0411,  ..., 0.7347, 0.9867, 0.2470],
        [0.0907, 0.0411, 0.0000,  ..., 0.7488, 0.9744, 0.2722],
        ...,
        [0.6585, 0.7347, 0.7488,  ..., 0.0000, 0.5987, 0.5013],
        [0.8998, 0.9867, 0.9744,  ..., 0.5987, 0.0000, 0.8602],
        [0.1934, 0.2470, 0.2722,  ..., 0.5013, 0.8602, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.23178204,  9.55593374,  9.48337063,  9.58347456,  9.60412735,
        9.68323686,  9.55720651,  6.42439344,  6.46356806,  7.71075226,
       13.12218013,  6.65259775, 13.2430377 , 10.37698435,  4.79003361,
        5.10444097,  6.50409372,  8.19703512,  6.02779201,  8.50278896,
        6.80370254,  9.27474386,  9.60412735,  8.29640613,  9.58072286,
        5.81481044, 13.36092624, 14.05633441, 11.10066126,  8.51088913,
        7.60529892,  9.14973319, 13.6922425 ,  7.46248947,  5.60064319,
       11.97957425,  4.85603886,  6.59176018,  6.12886024, 12.5085995 ,
        6.14086582,  5.70405525, 11.89855267, 10.412299  ,  7.17789666,
        7.80625981,  8.1303945 ,  9.7751189 ,  4.8763135 ,  9.14044984,
       12.72524615, 11.42420508,  5.14371005,  7.83486943, 11.86239961,
        7.44805901, 13.58101794, 12.32408516,  9.75200848, 12.84056604,
        5.86519525,  6.74872981, 12.9089559 , 12.35421461,  7.96575782,
        9.20340031,  7.72668467,  7.23276582, 13.40850328, 10.14436912,
       13.03002338,  5.05981914, 11.20100795,  7.32950385,  5.14371005,
        8.1137452 ,  7.06169391, 11.92133129,  5.20659892, 12.12448912,
       13.14777433, 12.64235359,  8.41171369, 12.49291768, 11.97238624,
        8.7273569 ,  6.42439344,  8.87007495,  5.24461068,  6.30647746,
       10.06407839, 10.26545792,  9.5014146 , 13.68063098,  5.81199422,
       12.59969936, 12.67576792,  8.68979052,  5.8828986 ,  7.27340834,
        6.60752373,  5.81199422,  9.63742669, 11.70016217, 12.58006038,
        9.17016857])}
{'P': tensor([[9.3059e-03, 3.0514e-01, 1.2315e-01,  ..., 8.8915e-05, 6.7809e-05,
         1.2378e-02],
        [2.2916e-04, 1.2824e-02, 4.8577e-02,  ..., 1.4241e-04, 1.0184e-04,
         1.7520e-02],
        [4.0355e-04, 2.1745e-03, 1.3192e-02,  ..., 1.6217e-04, 9.6758e-05,
         2.5076e-02],
        ...,
        [9.5612e-05, 9.3704e-05, 9.5957e-05,  ..., 9.2625e-02, 1.3719e-03,
         9.7384e-05],
        [1.3078e-04, 1.2587e-04, 1.1548e-04,  ..., 3.5703e-03, 1.7670e-01,
         6.1635e-05],
        [2.3669e-04, 3.5346e-04, 3.9995e-04,  ..., 8.0735e-04, 1.3895e-04,
         3.0870e-02]]), 'X': tensor([[ -9.0961,   1.6529,   3.1553],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.4203,  -2.0340,   2.5856],
        [ -8.0330,  -6.0765,   6.7945],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.1477,   7.2550,   8.9688],
        [  6.2242,  -0.3171,   7.4978],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -1.0482, -10.8487,  -5.9066],
        [  9.4516,  -7.2501,  -1.8868],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -6.4620,  -8.0449,   6.2719],
        [ -8.5149,  -5.1905,   1.5239],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.2038,   2.4020,  -3.1692],
        [  5.8617,   3.4853,  -9.8080],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.7151,  -1.7700,   3.1542],
        [  1.6954,  -4.7616,   9.8198],
        [  4.0457,  -3.7737,  -0.4704],
        [ -8.3539,  -5.5111,   0.3373],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.2975,   0.0947,   5.1791],
        [ -6.5295,  -7.9785,  -7.4429],
        [  9.6662,   9.1725,   8.1056],
        [  1.1006,   7.6318,  -2.4877],
        [ -3.4986,   7.5152,   9.0712],
        [  1.7815,  -4.7481,   4.4289],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.5294,  -2.1106,   9.0626],
        [ -3.7964,  -9.9294,   7.7147],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.1890,   4.7318,   0.9515],
        [ 10.0995,  -6.4739,   2.0974],
        [  7.9816,  10.8938,  -0.5378],
        [  6.3589,   0.8239,  -8.4230],
        [ 11.8951,  -3.1570,   7.0742],
        [  7.0358,  -9.2784,  -9.8479],
        [  9.3607,   9.5337,  -4.5412],
        [  5.3924, -10.1396,  -4.0863],
        [ -3.9508,  -9.8515,   1.7923],
        [ -7.7239,   4.4298,   3.1216],
        [  2.5636, -10.9293,   9.6519],
        [ -8.7409,  -4.6837,  -6.5840],
        [  6.3200,   0.2690,   1.5900],
        [  5.0617,  12.7957,   8.3544],
        [ -8.6050,   2.9451,  -0.7012],
        [  3.0876,   6.7527,  -5.3540],
        [  5.1203,  -2.7151,   6.2110],
        [ -6.0195,   6.1789,  -2.9496],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.1931,  -4.7491,   7.4385],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.3122,  -9.6994,  -6.3282],
        [  2.0243,   7.3086,   7.5432],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.3200,   0.2690,   1.5900],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.2675,  12.1379,  -8.9720],
        [  8.3718,  -8.2953,   8.4666],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -1.6220,   7.8917,   0.0315],
        [ -1.0012, -10.8576,   2.2867],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -9.4430,  -0.1897,   4.8383],
        [  7.8982,  -8.6778,   0.0239],
        [  5.3858,   4.4260,  -9.6769],
        [ 12.1504,   4.3674,  -7.2998],
        [ -1.6906,   7.8860,   7.7142],
        [ -9.3616,  -2.4750,   3.7765],
        [  1.5653,  13.9377,   4.1773],
        [  3.2366,  -4.2732,   5.1532],
        [ -6.0175,  -8.4541,  -9.8408],
        [  8.0877,  -8.5297,  -1.1050],
        [ -9.4756,  -0.8166,  -9.6021],
        [  1.7815,  -4.7481,   4.4289],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.0766,  -6.0046,  -3.1873],
        [  2.6310, -10.9195,   3.8054],
        [  9.3424,  -7.3685,  -3.6807],
        [  2.2659,  13.7894,   6.5436],
        [ 11.7796,  -3.4696,   9.0088],
        [ -9.1706,   1.3863,   8.4032],
        [  5.2911,  -2.4824,  -3.6877],
        [  0.5600, -11.0321,  -0.8660],
        [  3.0719, -10.8449,  -4.1673],
        [  5.1292,   4.8183,  -5.7997],
        [  0.3231,  -4.7691,  -1.6120],
        [  9.0652,   9.8597,  -3.5949],
        [ -4.7389,   6.9953,   3.7638],
        [  2.2182,  13.8007,   5.1889],
        [  3.3474,  -4.2162,   5.2455],
        [ 11.4703,  -4.2006,   7.4086],
        [ -9.4159,  -2.0739,   9.6841],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.0772,  -2.6014,  -7.1832],
        [  5.9193,  -1.3176,   5.1739],
        [  2.7558,  -4.4844,   0.0636],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.4283,  -1.9585,  -0.9129]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 2.8654e-02, 5.8188e-02,  ..., 7.6773e-01, 8.2422e-01,
         3.4036e-01],
        [2.8654e-02, 2.1453e-04, 3.6409e-02,  ..., 7.8983e-01, 8.5287e-01,
         3.2110e-01],
        [5.8188e-02, 3.6409e-02, 0.0000e+00,  ..., 7.8913e-01, 8.7512e-01,
         2.8479e-01],
        ...,
        [7.6773e-01, 7.8983e-01, 7.8913e-01,  ..., 0.0000e+00, 5.5317e-01,
         8.6041e-01],
        [8.2422e-01, 8.5287e-01, 8.7512e-01,  ..., 5.5317e-01, 0.0000e+00,
         1.0896e+00],
        [3.4036e-01, 3.2110e-01, 2.8479e-01,  ..., 8.6041e-01, 1.0896e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.24501829,  9.3109884 ,  9.3349858 ,  9.62238828,  9.60412735,
        9.63742669, 10.07238463,  8.56889738,  7.56617037,  6.2322769 ,
       10.32715715, 10.8992524 , 11.9120321 , 12.46513646, 10.31886066,
        9.97220799, 10.31886066,  6.65259775,  6.81961996,  5.84533345,
        5.98285319,  5.05444288,  5.53255882, 10.00794468,  5.19577146,
        6.29821673, 10.30972193, 13.32557133,  7.71075226,  8.28968067,
        5.0713364 , 10.61472636,  5.91854397, 10.63038565,  5.01114719,
        7.02253235, 11.99633819, 13.50484539,  6.41202986, 12.30695098,
       11.64436684, 13.36092624, 11.48434767, 10.61417181,  8.90405823,
       11.22596994,  9.91668182,  6.32572323, 13.76047999,  9.09502385,
        7.42512621,  5.79562691,  8.62631801, 12.42554808,  4.75303127,
        8.62162372, 11.57249984,  7.58378655,  9.68324553,  9.78147542,
        6.32572323,  9.61942445, 13.66051884, 11.785566  , 10.61047596,
       10.06437598, 10.46739073,  8.05668609, 10.9036226 , 12.42554808,
        9.44486609, 11.73397309,  6.97106912, 12.91143491,  8.06517143,
        9.68324553, 14.02532758,  5.36061974, 10.37698435, 11.75438006,
        9.51074763,  5.0713364 ,  6.35029274, 10.06407839, 11.23201619,
       11.89855267, 13.97430191, 12.27993098,  9.27474386,  5.84450127,
       11.04629471, 11.27160035,  7.03733775,  4.78002665, 13.39372426,
        8.44940661, 13.97779839,  5.38342077, 12.21532282,  9.64157461,
       13.77222752, 12.35421461,  6.06416957,  5.26342203,  5.13125513,
        9.62958973])}
{'P': tensor([[1.2256e-02, 2.9718e-01, 1.4713e-01,  ..., 6.1304e-03, 5.5218e-06,
         3.5291e-07],
        [2.9152e-04, 1.4293e-02, 2.4392e-01,  ..., 4.4902e-03, 6.0351e-06,
         3.2920e-07],
        [1.5569e-04, 2.0589e-04, 3.0329e-02,  ..., 2.2579e-03, 1.1629e-05,
         4.2662e-07],
        ...,
        [1.5625e-03, 2.4984e-03, 1.7109e-02,  ..., 1.3148e-01, 4.0938e-05,
         2.6161e-06],
        [1.2781e-03, 9.5138e-04, 3.9133e-04,  ..., 3.2402e-04, 8.2460e-02,
         9.3098e-03],
        [4.5964e-03, 3.0250e-03, 9.6733e-04,  ..., 2.0796e-03, 4.7214e-04,
         3.0503e-01]]), 'X': tensor([[ -9.1178,   1.5784,   2.1410],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.4203,  -2.0340,   2.5856],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.4351,  -1.8891,   3.5642],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 12.5927,   0.4199,  -9.4997],
        [  3.9663,   6.1018,  -5.4104],
        [  5.6414, -10.0303,   7.8525],
        [  3.2366,  -4.2732,   5.1532],
        [  5.7845,   3.6652,   0.9175],
        [ -7.9298,   4.1367,  -3.9773],
        [ -5.1398,  -9.1367,   6.2074],
        [  5.4426, -10.1181,  -5.1268],
        [ 12.3759,  -1.3943,   6.4659],
        [  0.6933,   7.7336,   2.1179],
        [ -6.7745,  -7.7264,   8.4684],
        [  9.4925,   9.3811,   8.9842],
        [  4.5430,   5.5438,  -3.8669],
        [  3.2159, -10.8166,  -1.0406],
        [  4.4633,   5.6287,  -0.7754],
        [ -7.3333,   4.9246,  -7.9212],
        [  6.2038,   2.4020,  -3.1692],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.4630,  -1.5145,   0.2983],
        [  3.6249, -10.7255,   5.6605],
        [  0.5250, -11.0306,   5.5358],
        [  0.6470, -11.0352,  -1.1931],
        [  3.0876,   6.7527,  -5.3540],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.7802,  -1.6363,  -6.2782],
        [  4.5491,  -3.3485,   6.8412],
        [ -0.5739,   7.9056,   0.6382],
        [  2.0729,   7.2878,   7.0880],
        [ -7.6719,  -6.6260,   7.1468],
        [  4.9152,  12.8653,  -1.4459],
        [  3.2701,   6.6338,   9.9209],
        [  5.9548,  -1.2273,  -2.6520],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.4867,  12.5816,   2.9549],
        [  4.9452, -10.3187,  -7.8621],
        [ -9.3616,  -2.4750,   3.7765],
        [  4.6357,   5.4415,   7.9067],
        [ -3.3116, -10.1539,   7.5919],
        [  2.6613,   7.0021,   4.5783],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -9.3464,   0.5708,  -1.6397],
        [  6.2975,   0.0947,   5.1791],
        [ -0.1963,   7.8765,  -1.1371],
        [ -1.3940, -10.7772,   6.7835],
        [  3.3474,  -4.2162,   5.2455],
        [  1.9312,  13.8649,   8.3563],
        [ 10.4758,  -5.9539,   1.3667],
        [  9.0004,  -7.7195,   4.6945],
        [  5.1861,   4.7361,  -3.7312],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -1.9600,   7.8577,   6.9121],
        [ -6.6895,   5.6051,   2.7899],
        [  7.4132,  11.3526,   0.3322],
        [ -8.0611,  -6.0302,   4.6962],
        [ -6.0844,  -8.3955,   0.7705],
        [ -3.9456,  -9.8542,   0.7466],
        [ -2.9497, -10.3028,  -7.3479],
        [  3.1064,  13.5607,   6.7605],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.7275,  14.0667,  -0.9815],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.0031,  12.2958,  -6.5992],
        [ -8.7735,   2.5640,  -6.3313],
        [  9.3424,  -7.3685,  -3.6807],
        [  2.2076,  13.8032,  -1.5099],
        [  4.5325,  -3.3643,   6.7699],
        [  8.7715,  -7.9390,   5.2900],
        [  6.0890,  -0.8382,   2.9166],
        [  5.2608,  12.6977,   4.9166],
        [  6.2026,  12.1774,  -1.5887],
        [  6.9653,  -9.3225,  -3.4727],
        [  4.5057,  -3.3894,  -0.8160],
        [ -3.3116, -10.1539,   7.5919],
        [ -8.7078,   2.7182,   1.8146],
        [  2.5318,  13.7231,   6.3692],
        [  8.0381,  -8.5690,  -8.9523],
        [  1.5246,  -4.7838,  -5.2095],
        [ -0.8455, -10.8853,  -6.7400],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3515,   0.6418,  -6.5823],
        [ -9.4697,  -0.6157,  -6.8472],
        [  7.8425,  -8.7202,   6.4530],
        [  0.1460,   7.8340,  -9.4614],
        [ -5.1369,   6.7755,   8.3836],
        [  2.4356,  -4.5942,  -5.0808],
        [  6.2790,  -0.0259,  -5.4116],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.9843,  -9.8679,  -6.1449],
        [  5.8902,  -1.3887,   4.0388],
        [  4.1637,  13.1898,  -1.9189],
        [ -3.4100,   7.5436,  -7.6980],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.4714,  -1.3301,  -3.1998],
        [  4.3377,   5.7570,   1.3712],
        [  1.4150,  13.9647,  -4.0915]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 3.7217e-02, 8.7770e-02,  ..., 4.6010e-01, 7.4532e-01,
         1.1524e+00],
        [3.7217e-02, 0.0000e+00, 5.9796e-02,  ..., 4.7862e-01, 7.7381e-01,
         1.1879e+00],
        [8.7770e-02, 5.9796e-02, 0.0000e+00,  ..., 4.6145e-01, 8.3222e-01,
         1.2375e+00],
        ...,
        [4.6010e-01, 4.7862e-01, 4.6145e-01,  ..., 0.0000e+00, 9.6051e-01,
         1.1639e+00],
        [7.4532e-01, 7.7381e-01, 8.3222e-01,  ..., 9.6051e-01, 0.0000e+00,
         5.5140e-01],
        [1.1524e+00, 1.1879e+00, 1.2375e+00,  ..., 1.1639e+00, 5.5140e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.25336371,  9.3349858 ,  9.60412735,  9.63742669,  9.55593374,
        9.62238828, 10.92597196,  9.87720102,  9.91668182, 12.59969936,
        7.27759233, 11.50791859,  5.36061974,  6.84796202,  8.94395628,
       10.48315436, 11.4890861 , 12.45418236,  7.76457095, 10.27572673,
       13.34587027,  7.16747878, 11.2845608 ,  7.18354885,  8.83341608,
        6.65259775,  5.83560009,  9.58347456, 11.32149259, 11.04313641,
       11.05413545,  7.42512621,  5.53255882,  6.00732193,  5.64865173,
        7.92644746,  7.57685908, 10.1371552 , 13.77222752,  7.39599986,
        6.07992623, 13.00413829,  5.27178363, 13.72594022, 11.4424816 ,
        9.68324553,  7.14837514, 10.68031351,  7.49076899, 12.35421461,
       10.57302546,  9.36378438,  6.29821673,  7.8788965 , 10.86693708,
        5.38342077, 13.99877413, 12.0495521 , 11.85743425,  7.02326515,
        9.60359158,  8.09842704,  8.7273569 , 13.55868684, 10.06703358,
       10.36844813, 10.61472636, 10.7167285 , 13.91198191,  4.91296205,
       14.08549802, 10.61417181, 13.6829843 ,  9.14044984, 11.89855267,
       13.97857591,  5.64464807, 11.83075005,  6.1463896 , 13.744387  ,
       13.6660671 , 11.63724792,  5.63824971, 10.68031351,  9.12220271,
       13.95472331, 11.74901778,  5.0209005 , 10.91805683,  5.5366698 ,
        6.383897  ,  9.48969978, 11.72802491,  7.83534987,  8.50268322,
        5.19986797,  6.27905292,  5.16077764, 11.54071859,  6.05165248,
       13.83139024,  8.27853076,  9.17595899,  9.5642993 ,  7.20826219,
       14.03618649])}
{'P': tensor([[7.1392e-03, 2.2005e-01, 1.7789e-01,  ..., 3.2856e-06, 2.7133e-06,
         1.3772e-06],
        [1.9920e-04, 1.5541e-02, 5.0204e-01,  ..., 7.7243e-06, 5.6765e-06,
         3.3157e-06],
        [3.2164e-04, 6.1489e-04, 3.0079e-02,  ..., 1.6035e-05, 1.1088e-05,
         7.0678e-06],
        ...,
        [6.1787e-05, 4.7892e-05, 4.4746e-05,  ..., 5.6241e-02, 6.2292e-02,
         2.3226e-03],
        [4.8506e-04, 3.3756e-04, 2.9881e-04,  ..., 1.1271e-03, 7.8491e-02,
         2.8611e-04],
        [4.2891e-05, 3.5808e-05, 3.5004e-05,  ..., 3.6656e-02, 1.1300e-02,
         2.0113e-01]]), 'X': tensor([[-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 4.6740e-02, 6.8597e-02,  ..., 1.0416e+00, 8.9487e-01,
         1.2342e+00],
        [4.6740e-02, 0.0000e+00, 2.3367e-02,  ..., 1.0821e+00, 9.3866e-01,
         1.2699e+00],
        [6.8597e-02, 2.3367e-02, 0.0000e+00,  ..., 1.0958e+00, 9.5524e-01,
         1.2799e+00],
        ...,
        [1.0416e+00, 1.0821e+00, 1.0958e+00,  ..., 0.0000e+00, 2.1761e-01,
         2.8320e-01],
        [8.9487e-01, 9.3866e-01, 9.5524e-01,  ..., 2.1761e-01, 0.0000e+00,
         4.9907e-01],
        [1.2342e+00, 1.2699e+00, 1.2799e+00,  ..., 2.8320e-01, 4.9907e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.26455795,  9.38232121,  9.43112957,  9.37989631,  9.41605684,
        9.55720651,  6.4020404 , 11.75438006,  8.09899874, 13.0493368 ,
       11.11038551,  4.7663526 , 12.19805394, 13.97779839, 12.32545637,
        9.20340031, 11.27160035, 13.91198191, 13.15525803,  8.83341608,
       14.10796466,  6.10774544,  7.83534987,  6.53290663,  9.36378438,
       11.66453059,  4.98989906,  5.22903117,  6.99097013, 13.59480906,
        7.78980299,  9.38232121, 12.53285633,  6.81117802,  8.53501777,
       11.11038551, 13.5247215 ,  6.65368551, 13.97779839,  5.04028538,
        6.61211224,  5.14371005,  9.58347456, 10.90638066, 11.63502679,
        8.41171369,  7.23276582,  4.78258092, 11.00694688, 13.79762415,
       10.51437947,  9.9353921 ,  9.0368458 ,  8.44940661,  5.5513072 ,
       12.64235359, 13.45642379,  5.87132718, 11.56146873,  7.92200342,
        5.1224079 ,  6.81961996, 14.07623266, 14.08549802,  9.15253344,
        8.49161154, 12.53345445,  7.15100303,  8.89458273, 12.84056604,
       12.78327037,  7.92607342,  9.97964013, 12.07818871,  7.71075226,
        6.81084724,  7.89745107,  9.78285424,  6.89245352,  8.94395628,
        6.38399789,  5.29755487,  9.11473391, 12.24467041,  7.92200342,
        7.57946459, 13.27717982,  7.58668144, 13.13888841,  9.76896904,
       12.12448912, 10.14436912,  6.42439344, 13.66896739, 12.85032836,
        5.64865173,  9.6282763 ,  6.81961996,  9.68323686, 13.2430377 ,
       11.04953767,  6.65259775, 12.49291768,  6.42689705,  7.02496223,
        6.16512856])}
{'P': tensor([[8.2422e-03, 2.2960e-01, 1.7195e-01,  ..., 5.3049e-07, 2.9591e-07,
         7.5720e-05],
        [2.3713e-04, 1.9740e-02, 1.6960e-01,  ..., 1.1142e-06, 5.1413e-07,
         2.7212e-04],
        [4.0748e-04, 2.4303e-03, 3.8411e-02,  ..., 1.7836e-06, 8.7668e-07,
         4.6737e-04],
        ...,
        [7.1063e-05, 5.5374e-05, 4.0628e-05,  ..., 5.4869e-02, 6.1105e-04,
         2.8112e-05],
        [6.2483e-03, 3.3944e-03, 2.5297e-03,  ..., 3.8654e-04, 2.2258e-01,
         1.1444e-05],
        [2.9982e-06, 4.0550e-06, 4.0356e-06,  ..., 1.4470e-05, 3.5534e-07,
         3.8616e-02]]), 'X': tensor([[ -9.1706,   1.3863,   8.4032],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.6022,  -1.9833,   8.3081],
        [ -8.9178,   2.1936,   7.5388],
        [  0.2480, -11.0153,  -3.0183],
        [ -9.4196,   0.0478,   9.4140],
        [  6.1222,  -0.7261,   8.8204],
        [ 12.4996,   2.6335,  -6.3166],
        [  1.1274,  14.0116,   8.1753],
        [  6.1579,  -0.5961,  -6.5595],
        [  1.1006,   7.6318,  -2.4877],
        [  6.2576,   2.1359,  -3.3398],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.8994,   6.8675,  -9.5987],
        [  1.6232,  -4.7717,  -2.5655],
        [ 10.9916,   7.2133,  -0.9179],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -6.2668,   5.9808,   2.5614],
        [  5.1890,   4.7318,   0.9515],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.7482, -10.3907,   5.9149],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.8400,   2.3991,   1.3457],
        [  5.2199,   4.6861,  -1.9304],
        [  6.1929,   2.4501,  -6.9352],
        [ -7.8280,   4.2848,  -7.1955],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 11.3118,   6.5957,  -4.5667],
        [ -5.9225,  -8.5355,   2.4778],
        [ -2.9497, -10.3028,  -7.3479],
        [ -0.8675, -10.8815,   6.8561],
        [ -8.5859,  -5.0393,   0.5476],
        [ 12.5832,   1.7543,  -7.6134],
        [  3.7237,  -3.9966,  -5.0376],
        [  6.8704,  -9.3809,   7.5769],
        [  3.5939,   6.4029,  -1.8357],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.8675, -10.8815,   6.8561],
        [  4.5491,  -3.3485,   6.8412],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.4597,  -0.9136,  -4.8763],
        [  7.8982,  -8.6778,   0.0239],
        [ 12.3609,   3.4769,  -8.5762],
        [  7.0358,  -9.2784,  -9.8479],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.1372,   2.6726,  -6.7885],
        [  2.3367,   7.1676,  -9.5263],
        [ -1.6086,   7.8928,   4.3385],
        [ -5.5525,   6.5146,  -4.4984],
        [ -5.1974,   6.7396,   3.3484],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.2644,  -4.6434,   9.7508],
        [ -2.0840,   7.8415,  -7.8657],
        [  3.2990,  -4.2415,  -7.4857],
        [ -7.4887,   4.7364,   8.9074],
        [  5.7151,  -1.7700,   3.1542],
        [  8.5093,  -8.1764,  -3.2381],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3534,   1.2990,  -7.2942],
        [ -2.0721,   7.8432,   6.6380],
        [  7.5784,  -8.9141,   5.8410],
        [  5.8323,  -1.5228,   5.1132],
        [  3.7298,  13.3537,  -9.4653],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.0275,  -5.0666,   8.4154],
        [  1.0391,  -4.8141,  -0.0768],
        [  0.6757, -11.0361,  -1.3047],
        [  5.2608,  12.6977,   4.9166],
        [  2.6613,   7.0021,   4.5783],
        [ -9.3705,   0.4208,  -3.5815],
        [ -4.0409,   7.3169,  -3.5779],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.3031,   1.8477,  -7.1706],
        [  5.2608,  -2.5256,  -9.1446],
        [  7.7794,  11.0629,   8.9069],
        [  5.0277,   4.9589,   7.0175],
        [ -2.8162,   7.7059,  -9.6885],
        [ -2.0445,   7.8469,  -0.5857],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -1.9600,   7.8577,   6.9121],
        [  6.2535,  -0.1709,  -9.4102],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.3360,   0.4258,  -9.2509],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.3087,   1.8053,  -1.9481],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.1477,   7.2550,   8.9688],
        [  5.6260,  -1.9401,   3.0259],
        [  5.6928,  12.4710,   3.6001],
        [ -5.6651,   6.4380,   1.3660],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -3.0644,   7.6439,  -1.4322],
        [  6.0305,   3.0296,   0.1375],
        [  1.9505,  -4.7171,   9.9868],
        [ -6.3066,  -8.1936,   9.5822],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.2860,   7.1916,  -6.1011],
        [  0.5175,  14.0910,   4.6540],
        [ -4.0633,  -9.7927,  -3.0825]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.8724e-02, 5.5894e-02,  ..., 1.1389e+00, 9.9105e-01,
         8.3609e-01],
        [3.8724e-02, 3.0339e-04, 2.3392e-02,  ..., 1.1529e+00, 1.0284e+00,
         8.2468e-01],
        [5.5894e-02, 2.3392e-02, 0.0000e+00,  ..., 1.1756e+00, 1.0468e+00,
         8.3917e-01],
        ...,
        [1.1389e+00, 1.1529e+00, 1.1756e+00,  ..., 0.0000e+00, 9.7740e-01,
         7.1400e-01],
        [9.9105e-01, 1.0284e+00, 1.0468e+00,  ..., 9.7740e-01, 0.0000e+00,
         1.3273e+00],
        [8.3609e-01, 8.2468e-01, 8.3917e-01,  ..., 7.1400e-01, 1.3273e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.27474386,  9.41244865,  9.45995955,  9.49511524,  9.41970031,
        9.64157461,  5.94293185,  9.18358562, 11.018084  ,  9.41970031,
        6.16512856, 12.77402351, 14.05687815,  6.18668463,  7.71075226,
        6.61211224,  6.06416957,  9.45995955,  7.4544866 ,  5.04028538,
       13.1471313 , 10.1979669 ,  8.66273171,  7.02253235,  5.01114719,
       11.42420508,  9.77777346,  9.15976746,  7.01475376,  6.65991877,
        8.92395381,  9.75401449, 13.09426142, 10.38897563, 10.7167285 ,
       10.91601655,  9.95552626, 12.70489563,  5.46244988, 11.62770332,
        7.34250948,  7.822854  , 10.91601655,  5.64865173, 11.77124373,
       11.28895489, 12.49317563, 11.73397309, 12.84056604, 11.64436684,
        6.00732193,  6.69389431,  7.53884104,  8.0550286 ,  8.55982105,
        8.51088913, 13.04217487,  5.16611782,  8.1137452 ,  5.37343173,
        8.86082124,  5.98285319, 11.80092578,  6.16512856,  6.48486171,
        8.11227668, 11.70016217,  6.02779201, 13.86480175, 12.4013153 ,
        9.96152788, 12.13568598,  4.92497514, 11.05672136, 13.744387  ,
        7.49076899,  9.37989631,  8.35856487,  8.85572292,  6.56833864,
        5.83560009, 13.52430176,  7.06169391,  8.20436764,  8.10886021,
       10.57302546,  8.09842704,  6.25586297,  8.47371868,  6.35029274,
       13.13888841,  6.5618968 ,  9.43737523,  7.56617037,  5.95111804,
       13.70893492,  8.57560275,  9.5014146 ,  8.23525433,  6.74872981,
        5.10446653, 10.33958639, 10.35485331,  7.54620849, 14.10045797,
       10.60226614])}
{'P': tensor([[1.3335e-02, 1.7773e-01, 1.7549e-01,  ..., 2.3533e-06, 3.5214e-07,
         1.6796e-04],
        [1.7656e-04, 2.6202e-02, 2.6664e-02,  ..., 1.1822e-05, 4.3142e-07,
         5.0370e-04],
        [7.6087e-05, 4.2094e-03, 1.2559e-02,  ..., 4.7769e-06, 1.8610e-07,
         3.3553e-04],
        ...,
        [2.2892e-07, 1.7087e-07, 1.4226e-07,  ..., 6.6231e-03, 1.5410e-05,
         2.6350e-06],
        [2.6695e-03, 7.5937e-04, 5.9759e-04,  ..., 1.0685e-06, 2.1078e-01,
         1.5342e-05],
        [1.1052e-04, 1.0643e-04, 1.2582e-04,  ..., 3.8365e-04, 3.4801e-05,
         1.2121e-01]]), 'X': tensor([[ -9.2508,   1.0572,   2.9978],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.4203,  -2.0340,   2.5856],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.3616,  -2.4750,   3.7765],
        [ -7.5401,   4.6718,   9.0033],
        [ 11.5456,   6.0861,   8.1101],
        [ 12.4592,  -0.9168,  -6.7571],
        [  6.3497,   1.3680,  -1.9076],
        [ 12.5123,  -0.5339,  -1.0728],
        [  2.3373,  -4.6232,  -5.2246],
        [  2.8474,   6.8979,   6.6457],
        [  0.2435,   7.8191,  -6.4709],
        [  2.8994,   6.8675,  -9.5987],
        [  6.3535,   0.6792,  -7.5101],
        [  6.0073,   3.0986,   6.7937],
        [ 11.5935,   5.9740,  -4.3952],
        [ -7.9667,   4.0814,  -0.6397],
        [ 12.2328,  -2.0376,  -1.1234],
        [  0.7958,  -4.8109,  -8.2248],
        [  3.2226,  -4.2802,  -4.0428],
        [ 10.9875,   7.2207,  -8.8549],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.5382,   7.0672,   4.5097],
        [  1.0000,   7.6591,  -4.5011],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.1989,  -9.7591,   0.8289],
        [ -5.9987,   6.1949,  -6.2114],
        [  6.3236,   0.3011,  -4.6668],
        [  3.2990,  -4.2415,  -7.4857],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.2895,   0.0405,   5.7652],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.3025,  -1.7440,  -6.6632],
        [  9.2811,  -7.4336,   5.1771],
        [ -5.1369,   6.7755,   8.3836],
        [  1.5653,  13.9377,   4.1773],
        [ 10.1503,   8.5414,  -7.2681],
        [ 10.4737,  -5.9570,  -1.2108],
        [  7.1441,  11.5529,   2.3232],
        [  3.3943,   6.5483,   1.0790],
        [ -8.5149,  -5.1905,   1.5239],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.4378,  -1.8606,  -2.5668],
        [  2.9461,  -4.4077,  -0.4831],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.2365,  -0.2578,  -6.0212],
        [ -9.1178,   1.5784,   2.1410],
        [  9.5018,  -7.1945,  -1.1892],
        [ 12.6000,   1.3839,  -3.2992],
        [  0.7145,  -4.8071,  -2.3766],
        [ -3.2486,   7.5925,  -1.3719],
        [  6.2036,   2.4030,  -1.2500],
        [  5.5101,  -2.1433,   9.5442],
        [  7.9141,  -8.6656,   9.6948],
        [  5.3924, -10.1396,  -4.0863],
        [  5.5294,  -2.1106,   9.0626],
        [  6.1575,  12.2046,   3.3257],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.5770,   1.8512,  -0.6578],
        [  7.7794,  11.0629,   8.9069],
        [  0.6933,   7.7336,   2.1179],
        [  6.2902,   0.0451,   3.7116],
        [ -4.7389,   6.9953,   3.7638],
        [  2.3589,  13.7668,  -3.6501],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.1085,  -4.9204,   0.0736],
        [  9.2692,   9.6370,   7.0746],
        [ -5.6651,   6.4380,   1.3660],
        [ -1.8788, -10.6567,   3.7068],
        [ -4.1296,   7.2803,  -2.7194],
        [  3.3094,  13.4969,  -3.6507],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.3602,  -2.3808,   7.2098],
        [  3.0719, -10.8449,  -4.1673],
        [  4.1537,   5.9340,  -6.0423],
        [  5.2911,  -2.4824,  -3.6877],
        [  7.0515,  11.6194,   8.4289],
        [  8.6271,  -8.0715,  -7.4813],
        [  6.1741,  -0.5329,   3.1913],
        [  8.9284,  10.0036,  -6.9210],
        [ -9.4714,  -1.3301,  -3.1998],
        [  8.6420,  -8.0580,   6.5781],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.0004,  -7.7195,   4.6945],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.4336,  -2.2676,   5.6933],
        [  2.4697,  -4.5836,   1.4593],
        [  1.4749,  -4.7892,  -0.8089],
        [  3.1064,  13.5607,   6.7605],
        [  9.0652,   9.8597,  -3.5949],
        [  4.6992,   5.3690,   3.0935],
        [  6.3551,   0.7149,   1.3344],
        [  2.7590,  13.6621,   9.4496],
        [  5.2188,  -2.5839,   2.1833],
        [  1.8719,   7.3713,  -3.7465],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.2076,  13.8032,  -1.5099],
        [  1.7228,  -4.7575,   8.3396]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 8.0605e-02, 7.9581e-02,  ..., 1.0036e+00, 1.0547e+00,
         8.2798e-01],
        [8.0605e-02, 0.0000e+00, 8.6979e-02,  ..., 1.0304e+00, 1.1078e+00,
         9.0720e-01],
        [7.9581e-02, 8.6979e-02, 0.0000e+00,  ..., 1.0826e+00, 1.1317e+00,
         8.7649e-01],
        ...,
        [1.0036e+00, 1.0304e+00, 1.0826e+00,  ..., 0.0000e+00, 4.0119e-01,
         9.4899e-01],
        [1.0547e+00, 1.1078e+00, 1.1317e+00,  ..., 4.0119e-01, 0.0000e+00,
         6.6719e-01],
        [8.2798e-01, 9.0720e-01, 8.7649e-01,  ..., 9.4899e-01, 6.6719e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.3109884 ,  9.60412735,  9.62238828,  9.63742669,  9.6282763 ,
        9.68324553,  8.87007495, 13.05148826, 12.49291768,  6.49538841,
       12.52372629,  5.18047638,  7.46248947,  7.822854  ,  7.4544866 ,
        6.38968357,  6.7593917 , 13.04217487,  8.95133961, 12.4013153 ,
        4.8763135 ,  5.35774573, 13.14777433, 12.18045978,  7.50917551,
        7.72415048,  6.1561646 ,  5.80312949,  9.68324553,  9.37296102,
       11.56146873,  8.62329466,  6.33075682,  5.37343173,  9.61942445,
        6.28961988,  8.89458273, 12.42554808, 11.89105259,  8.50268322,
       14.02532758, 13.26590071, 12.04924159, 13.58333857,  7.37576154,
        9.97220799, 10.33958639,  9.61942445,  5.30158497,  9.09661212,
        6.24187044,  9.25336371, 11.9182956 , 12.67576792,  4.85995002,
        8.25828409,  6.652742  ,  5.91222346, 11.73567759, 11.48434767,
        5.91854397, 13.66991027, 12.64235359, 12.7125079 , 13.52430176,
        7.76457095,  6.29034958,  8.44940661, 13.96746816, 12.42554808,
        8.41148722, 12.1494023 , 13.37121923,  8.57560275, 10.82106505,
        8.36995953, 13.89671314, 10.52357479,  5.86519525, 11.27160035,
        7.24326992,  5.84450127, 13.59170809, 11.81423161,  6.19707944,
       13.40850328,  9.5642993 , 11.8159254 ,  9.66977802, 11.85743425,
        6.12886024,  5.88782913,  5.20659892,  5.01114719, 13.91198191,
       13.39372426,  7.13501264,  6.39520397, 13.93790235,  5.82345867,
        7.60529892,  8.43626114, 12.35421461, 12.05395056, 13.97857591,
        5.05981914])}
{'P': tensor([[1.1014e-02, 2.8049e-01, 1.7369e-01,  ..., 3.0890e-07, 2.6056e-06,
         2.0176e-05],
        [1.7729e-04, 1.2607e-02, 2.8995e-01,  ..., 3.6582e-07, 3.0153e-06,
         1.3688e-05],
        [3.3480e-04, 6.3488e-04, 3.1561e-02,  ..., 9.4675e-07, 7.3450e-06,
         2.3839e-05],
        ...,
        [7.9805e-06, 4.8900e-06, 3.4824e-06,  ..., 1.7969e-02, 4.3468e-05,
         1.2910e-04],
        [2.4100e-05, 1.7716e-05, 1.3852e-05,  ..., 2.2695e-03, 6.4014e-02,
         5.8099e-04],
        [1.2573e-02, 9.2904e-03, 6.7254e-03,  ..., 3.2168e-06, 5.7183e-05,
         7.7060e-02]]), 'X': tensor([[-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.8654e-02, 7.5531e-02,  ..., 1.0651e+00, 1.0858e+00,
         6.0108e-01],
        [3.8654e-02, 3.0339e-04, 3.8538e-02,  ..., 1.1012e+00, 1.1242e+00,
         6.3470e-01],
        [7.5531e-02, 3.8538e-02, 3.0339e-04,  ..., 1.1290e+00, 1.1610e+00,
         6.7311e-01],
        ...,
        [1.0651e+00, 1.1012e+00, 1.1290e+00,  ..., 0.0000e+00, 5.2223e-01,
         8.3896e-01],
        [1.0858e+00, 1.1242e+00, 1.1610e+00,  ..., 5.2223e-01, 0.0000e+00,
         5.8752e-01],
        [6.0108e-01, 6.3470e-01, 6.7311e-01,  ..., 8.3896e-01, 5.8752e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.32094055,  9.44486609,  9.53015376,  9.6282763 ,  9.62238828,
        9.68324553, 11.59512194, 12.38881635,  7.88677538, 10.8992524 ,
        9.3109884 ,  6.38968357,  5.37343173, 13.15525803, 13.12218013,
        5.24137909,  6.74872981,  8.12043983,  8.8737556 , 13.66991027,
        9.51074763, 10.53368644, 12.45418236,  5.76165934, 13.59673406,
        5.68942572,  7.24677664, 12.04924159, 13.6105595 ,  9.43112957,
        9.71392179,  7.02496223,  9.78286733, 13.77075174,  5.05981914,
       10.37698435, 12.27993098,  9.75200848,  7.05627461,  6.65259775,
        9.94285511, 11.89855267,  6.76791483,  6.80559669,  7.50035355,
       13.77604293,  8.20436764, 11.04629471,  5.82345867,  5.89074104,
        6.00732193,  9.12220271,  6.53290663,  5.47702766,  8.89301516,
       13.09737053, 10.06850677,  7.10378919, 10.794755  , 13.2778378 ,
        6.81084724,  5.79391825, 11.73567759,  5.19577146, 11.42420508,
        6.60752373,  8.40445872, 12.42554808,  6.84796202,  4.74770899,
        5.98285319, 12.62711252,  7.32754882,  4.8763135 ,  7.32754882,
        6.49538841, 12.10222204,  6.81961996,  5.41348121,  8.89458273,
       12.3828957 , 11.21703654,  9.4997987 ,  8.0550286 ,  9.32094055,
        4.78544875, 13.04051744,  5.27178363,  5.14124612,  9.36093562,
        8.30729352,  7.83486943,  6.36004393, 10.69963954,  5.30158497,
        5.91408504, 13.1399181 , 13.09737053,  8.37954853,  8.09984231,
       12.889695  , 10.8207297 , 12.5085995 , 13.08414689,  6.36004393,
        8.62631801])}
{'P': tensor([[1.4361e-02, 2.4509e-01, 2.1598e-01,  ..., 2.9173e-05, 4.9584e-04,
         1.7807e-04],
        [1.1054e-04, 1.6283e-02, 2.7328e-01,  ..., 5.4923e-05, 2.1014e-04,
         3.4131e-04],
        [1.4627e-04, 1.0243e-03, 2.4723e-02,  ..., 8.0830e-05, 2.4689e-04,
         5.8877e-04],
        ...,
        [5.4608e-05, 5.4203e-05, 5.1020e-05,  ..., 1.7294e-01, 1.7215e-04,
         2.8143e-04],
        [3.8815e-04, 1.2661e-03, 1.2414e-03,  ..., 5.8012e-04, 3.7920e-02,
         4.2824e-05],
        [1.4303e-04, 1.4227e-04, 1.4829e-04,  ..., 3.3884e-04, 2.2083e-05,
         1.5380e-01]]), 'X': tensor([[ -9.2974,   0.8371,   2.4852],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.4203,  -2.0340,   2.5856],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.1813,  -3.3550,   1.8827],
        [  8.8474,  10.0868,  -8.8644],
        [  3.9591, -10.6391,  -6.2557],
        [ -3.3116, -10.1539,   7.5919],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.3205,   1.7068,  -3.8619],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.1206,  -4.7363,   8.7103],
        [ -0.9715, -10.8630,   0.3555],
        [  4.9337,  12.8566,  -3.6582],
        [  2.5382,   7.0672,   4.5097],
        [  5.6248,   3.9984,   8.4933],
        [  0.7275,  14.0667,  -0.9815],
        [  3.9895,   6.0816,  -6.5235],
        [  9.9217,  -6.7006,  -5.6728],
        [  1.3339,   7.5627,  -0.9034],
        [ -3.0644,   7.6439,  -1.4322],
        [  9.6816,   9.1536,  -5.8543],
        [  9.0004,  -7.7195,   4.6945],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.1666,   3.7653,   3.6075],
        [ -0.1626, -10.9796,   6.4629],
        [  9.4554,   9.4245,   9.8800],
        [ -1.8788, -10.6567,   3.7068],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.4837,   2.7510,   0.4449],
        [  2.5813,  -4.5471,   5.3597],
        [ -9.4541,  -0.3327,   8.8246],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.4996,   7.7738,  -4.4711],
        [  1.9312,  13.8649,   8.3563],
        [  7.1821,  -9.1845,   9.8548],
        [  5.6093,  -1.9705,   6.4807],
        [  6.8093,  -9.4178,   8.4244],
        [  0.9036,   7.6840,   9.9237],
        [ -3.9858,  -9.8334,  -3.7323],
        [  8.1528,  10.7452,   9.1493],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.1271, -10.8343,   2.6442],
        [ -2.1382,   7.8339,   7.3524],
        [  4.2096,   5.8815,   1.9230],
        [  5.0506,   4.9277,  -9.5094],
        [  1.1644, -11.0394,   7.8511],
        [ -9.1795,  -3.3623,   8.5259],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.1677,  -4.7447,  -6.1529],
        [  3.7237,  -3.9966,  -5.0376],
        [  5.6248,   3.9984,   8.4933],
        [  2.0729,   7.2878,   7.0880],
        [  3.1064,  13.5607,   6.7605],
        [  6.5428,  -9.5728,  -2.9897],
        [  2.1440, -10.9811,   8.5281],
        [ 11.9695,  -2.9407,  -6.3498],
        [  9.3836,   9.5075,  -5.9278],
        [  2.5813,  -4.5471,   5.3597],
        [  6.1984,  -0.4327,   1.5438],
        [  5.3924, -10.1396,  -4.0863],
        [ 11.0275,  -5.0666,   8.4154],
        [  3.2692,  13.5098,  -4.9032],
        [ -9.4159,  -2.0739,   9.6841],
        [ -8.8540,  -4.3971,  -5.2144],
        [  0.5175,  14.0910,   4.6540],
        [  4.1297,  -3.7098,  -3.9838],
        [ -9.4196,   0.0478,   9.4140],
        [ -1.9714,   7.8563,  -1.8084],
        [  6.2365,  -0.2578,  -6.0212],
        [  3.9895,   6.0816,  -6.5235],
        [  4.2096,   5.8815,   1.9230],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.6928,  12.4710,   3.6001],
        [ 12.3453,  -1.5462,   7.0212],
        [  2.0870,  -4.6877,   9.6823],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.3373,  13.7721,  -0.2442],
        [ -8.7544,   2.6096,   3.3146],
        [ 11.2209,  -4.7092,   2.6117],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.6613,   7.0021,   4.5783],
        [  6.0310,  12.2795,  -6.4528],
        [  0.6142,  14.0802,   0.5407],
        [ -8.0829,  -5.9940,   5.0564],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.7298,  13.3537,  -9.4653],
        [  6.2790,  -0.0259,  -5.4116],
        [  8.5198,  10.4091,  -8.6486],
        [ -0.3432,   7.8900,   7.1805],
        [ -8.1531,  -5.8748,   2.6934],
        [ -0.4341, -10.9473,   7.4509],
        [  5.7784,  -1.6401,   9.5587],
        [  4.1999,  -3.6543,  -4.1998],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.1504,   4.3674,  -7.2998],
        [ -9.1907,  -3.3180,  -5.5943],
        [  3.3633,  13.4794,   0.5193],
        [  3.1480,  -4.3165,  -7.8002],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.1625,  -4.6697,   7.0658]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0598, 0.0671,  ..., 0.9903, 0.6636, 0.8378],
        [0.0598, 0.0000, 0.0176,  ..., 1.0352, 0.6494, 0.8963],
        [0.0671, 0.0176, 0.0000,  ..., 1.0501, 0.6649, 0.8980],
        ...,
        [0.9903, 1.0352, 1.0501,  ..., 0.0000, 0.9180, 0.8553],
        [0.6636, 0.6494, 0.6649,  ..., 0.9180, 0.0000, 1.2962],
        [0.8378, 0.8963, 0.8980,  ..., 0.8553, 1.2962, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.3349858 ,  9.60412735,  9.63742669,  9.62238828,  9.68324553,
        9.7751189 , 13.41713661, 11.35182589, 10.68031351,  5.26067019,
        6.54693987,  8.25828409,  5.27178363,  4.73784222, 10.90638066,
       13.77075174,  7.50917551,  6.90115901, 14.08549802,  7.27340834,
       11.97238624,  7.67939546,  8.23525433, 13.32374056, 11.85743425,
       10.46238054,  4.7663526 ,  8.99276564, 10.98076873, 13.35013374,
       10.82106505, 10.06407839, 12.76408562, 12.78327037,  5.2287058 ,
        9.45995955, 10.5575601 ,  7.78980299, 13.99877413, 11.65923396,
        5.94535529, 11.62158554,  7.73692436, 10.61047596, 13.48809595,
        5.53255882, 11.27656488,  8.12043983,  7.23276582,  7.05627461,
       11.10066126,  9.77588801,  8.83341608,  4.74770899,  5.46244988,
        6.90115901,  7.57685908, 13.91198191, 11.59512194, 11.18839169,
       12.32545637, 13.35832725,  5.2287058 ,  6.2134961 , 11.48434767,
       12.13568598, 13.89974693,  9.64157461,  9.88571713, 14.10045797,
        5.5513072 ,  9.41970031,  8.09984231,  6.24187044,  7.27340834,
        7.23276582,  9.48337063, 13.70893492, 12.44177641,  5.13125513,
        5.8828986 , 13.96905756,  9.13507321, 12.16901615,  8.27853076,
        9.75075273,  7.49076899, 13.68063098, 14.09357547, 10.06286074,
        5.18047638, 13.86480175,  6.27905292, 13.45124917,  7.89745107,
       10.04916678, 10.95593729,  6.00663314,  5.5671389 , 13.13205357,
       12.91143491,  9.7712321 , 13.89264822,  5.34250225,  9.48969978,
        5.14607014])}
{'P': tensor([[6.9371e-03, 1.9001e-01, 1.2500e-01,  ..., 3.9437e-05, 2.5214e-03,
         4.5031e-07],
        [1.1479e-04, 1.0364e-02, 1.1644e-02,  ..., 9.1530e-05, 1.9342e-03,
         8.5374e-07],
        [2.3809e-04, 7.6411e-03, 1.7197e-02,  ..., 2.0493e-04, 4.8376e-03,
         1.5120e-06],
        ...,
        [2.8521e-06, 3.3406e-06, 3.8417e-06,  ..., 6.0276e-02, 7.8937e-06,
         5.8711e-04],
        [1.4300e-03, 3.6521e-03, 4.8316e-03,  ..., 8.7341e-03, 1.1397e-01,
         1.3101e-05],
        [2.1318e-06, 1.3903e-06, 1.4174e-06,  ..., 3.0729e-07, 1.0245e-06,
         1.7199e-02]]), 'X': tensor([[ -9.3076,   0.7852,  -6.7010],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 11.3691,   6.4760,   1.0787],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.1382, -11.0073,   1.0026],
        [ -3.5173,   7.5091,  -8.9585],
        [  1.6954,  -4.7616,   9.8198],
        [  2.0547,   7.2957,  -1.9343],
        [  2.6078, -10.9229,   6.2869],
        [  2.5974,   7.0363,  -7.7810],
        [  3.2159, -10.8166,  -1.0406],
        [  4.2096,   5.8815,   1.9230],
        [  6.2804,   2.0019,   0.8607],
        [ -3.9456,  -9.8542,   0.7466],
        [ -5.9987,   6.1949,  -6.2114],
        [ -7.6650,   4.5092,   0.2520],
        [  7.0642,  11.6104,  -9.8316],
        [  2.9260,  -4.4162,   8.3808],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.1047, -11.0046,   0.5188],
        [ -8.7544,   2.6096,   3.3146],
        [  9.0652,   9.8597,  -3.5949],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.3301,   0.3638,  -7.5685],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.0890,  -0.8382,   2.9166],
        [ -3.5520,   7.4976,   9.1522],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.8093,  -9.4178,   8.4244],
        [ -3.3116, -10.1539,   7.5919],
        [  0.3715,  -4.7756,   0.2312],
        [  1.6408,   7.4591,  -4.2294],
        [  2.3889,  -4.6082,  -8.7291],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.1372,   2.6726,  -6.7885],
        [ 11.9808,   4.9340,  -3.3715],
        [ 12.2589,  -1.9312,   9.8657],
        [ -1.9714,   7.8563,  -1.8084],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -6.7992,   5.4991,   9.9186],
        [  1.9503,  -4.7172,  -6.2313],
        [ -5.2589,   6.7024,  -5.5809],
        [  6.2026,  12.1774,  -1.5887],
        [  3.3069,   6.6089,   8.7377],
        [ -3.5520,   7.4976,   9.1522],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.4573,   3.2448,   5.4370],
        [ -7.4733,   4.7555,   6.2490],
        [  2.5636, -10.9293,   9.6519],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -0.9715, -10.8630,   0.3555],
        [  2.3261,   7.1726,   6.2522],
        [  2.2182,  13.8007,   5.1889],
        [ -9.4716,  -0.6673,   7.6068],
        [  5.9587,  12.3216,   9.1695],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4524,  -1.6813,  -6.6862],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.6321,  -4.5296,  -2.5159],
        [  1.5383, -11.0274,  -4.9666],
        [  0.6363, -11.0348,   6.1414],
        [  5.6928,  12.4710,   3.6001],
        [  2.2849, -10.9655,  -7.6542],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.3167,   6.6022,   6.3554],
        [  0.1252, -11.0062,   7.2564],
        [ 11.4743,   6.2478,  -2.1100],
        [ 12.5224,  -0.4497,   4.8448],
        [ -6.4620,  -8.0449,   6.2719],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.1741,  -0.5329,   3.1913],
        [ -5.6792,  -8.7352,   4.0318],
        [ -0.6323,   7.9085,  -5.9544],
        [  3.2809,  13.5061,  -4.0199],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.4167, -10.1293,   6.2156],
        [  7.0515,  11.6194,   8.4289],
        [  7.4132,  11.3526,   0.3322],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.3367,   7.1676,  -9.5263],
        [  6.2895,   0.0405,   5.7652],
        [  9.3424,  -7.3685,  -3.6807],
        [  9.3753,   9.5171,   3.9131],
        [  1.6308,  13.9254,  -2.0984],
        [  0.6363, -11.0348,   6.1414],
        [ -3.5903,   7.4847,  -8.9112],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -0.2586,   7.8825,  -9.9840],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.2975,   0.0947,   5.1791],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 12.6038,   0.7665,  -3.4439]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0538, 0.0559,  ..., 0.7659, 0.3748, 1.0290],
        [0.0538, 0.0000, 0.0400,  ..., 0.7917, 0.3858, 1.0826],
        [0.0559, 0.0400, 0.0000,  ..., 0.7525, 0.3458, 1.0737],
        ...,
        [0.7659, 0.7917, 0.7525,  ..., 0.0000, 0.4296, 1.0275],
        [0.3748, 0.3858, 0.3458,  ..., 0.4296, 0.0000, 1.0773],
        [1.0290, 1.0826, 1.0737,  ..., 1.0275, 1.0773, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.34061938,  9.48969978,  9.4997987 ,  9.60080639,  9.47369628,
        9.562174  , 13.08414689, 11.81423161, 11.00812996,  8.29203356,
        5.05444288,  7.57946459, 11.22993218,  7.50035355, 11.2845608 ,
        7.23276582,  6.59176018, 10.61472636,  8.62329466,  8.89301516,
       13.59056613,  5.29755487,  4.91296205, 11.00508681,  9.13507321,
       13.39372426, 10.95593729,  9.85330646,  6.34059021, 10.65929968,
        9.61942445, 12.42554808,  6.1463896 ,  8.29640613,  5.8828986 ,
       11.62158554, 10.68031351,  4.79003361,  7.6374603 ,  5.19065439,
       11.785566  ,  9.62958973,  9.5642993 ,  6.69389431, 12.95702647,
       12.41012046,  8.09984231,  9.47369628, 12.52372629,  8.74470218,
        5.10444097,  8.51928531, 13.6660671 ,  7.39004226,  8.29640613,
       10.67322527,  9.05843201,  8.8580671 , 11.22596994, 12.05300943,
       10.90638066,  7.5403822 , 13.97779839,  9.49511524, 13.68672855,
        7.5092834 ,  9.60080639,  4.78258092,  5.23878371, 11.1341798 ,
       11.05317597, 13.70893492, 11.20100795,  8.22728399,  7.38844703,
       11.00694688, 13.06498554, 12.53047489, 10.31886066,  8.30125201,
        6.19707944, 10.41909211,  7.9337647 , 13.89886469,  6.14086582,
       11.48663767, 13.59170809, 13.55868684, 12.22276278,  7.53884104,
        6.28961988, 11.89855267, 13.35927527, 14.02058621, 11.05317597,
        8.30125201,  9.62958973,  7.88677538,  9.60412735,  6.19633139,
        6.29821673,  5.91677188,  9.96152788, 10.99326577,  9.62958973,
       12.62711252])}
{'P': tensor([[8.6043e-03, 2.4793e-01, 1.7992e-01,  ..., 1.3475e-07, 2.2869e-06,
         7.5702e-06],
        [1.1505e-04, 9.5942e-03, 3.5668e-01,  ..., 2.0292e-07, 2.4496e-06,
         6.0977e-06],
        [2.0478e-04, 3.2998e-04, 2.5033e-02,  ..., 6.3150e-07, 6.2043e-06,
         1.3005e-05],
        ...,
        [3.9067e-07, 3.2762e-07, 2.8539e-07,  ..., 5.1064e-02, 8.0550e-04,
         6.2846e-06],
        [6.5252e-04, 4.5440e-04, 3.4713e-04,  ..., 2.2801e-05, 1.3242e-01,
         1.5910e-02],
        [1.7866e-02, 1.0299e-02, 7.1027e-03,  ..., 3.2779e-07, 6.9233e-05,
         4.0669e-02]]), 'X': tensor([[ -9.3087,   0.7792,  -7.5587],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.3331,  -2.6465,  -8.2028],
        [  2.0870,  -4.6877,   9.6823],
        [ -3.5173, -10.0623,   4.8280],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.3266,   0.3292,   9.3502],
        [  1.7659,   7.4127,   4.7871],
        [  6.3497,   1.3680,  -1.9076],
        [  1.4749,  -4.7892,  -0.8089],
        [ -3.4100,   7.5436,  -7.6980],
        [  6.0955,   2.8210,  -4.3321],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.4524,  -1.6813,  -6.6862],
        [  4.1637,  13.1898,  -1.9189],
        [  9.2811,  -7.4336,   5.1771],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.3858,   4.4260,  -9.6769],
        [  4.9337,  12.8566,  -3.6582],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.0506,   4.9277,  -9.5094],
        [ -2.6069, -10.4298,  -6.1995],
        [ -8.5900,   2.9767,   7.4811],
        [ -2.5056,   7.7720,   5.5878],
        [ -9.1623,  -3.4288,  -4.9961],
        [  6.3550,   0.7129,   0.4917],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -2.6485, -10.4151,   0.8766],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.2905,   1.9366,   2.1169],
        [  5.9883,   3.1536,   6.2320],
        [  2.5813,  -4.5471,   5.3597],
        [ -6.2200,  -8.2737,   5.5079],
        [ -3.5173,   7.5091,  -8.9585],
        [  1.2728, -11.0372,   9.8121],
        [  6.0890,  -0.8382,   2.9166],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.7963,  -1.6018,   1.8856],
        [  4.8653,   5.1694,  -4.0823],
        [  0.1382, -11.0073,   1.0026],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.7156,  -3.1831,  -8.0224],
        [ -5.1974,   6.7396,   3.3484],
        [  2.1625,  -4.6697,   7.0658],
        [  5.5240,  -2.1198,   6.2055],
        [ -6.1899,  -8.3011,  -7.9269],
        [  9.3624,   9.5318,   0.2779],
        [  4.8653,   5.1694,  -4.0823],
        [  3.4920,  -4.1366,  -8.3955],
        [  9.3836,   9.5075,  -5.9278],
        [  8.6271,  -8.0715,  -7.4813],
        [ -5.9225,  -8.5355,   2.4778],
        [ -7.6734,   4.4980,  -5.4248],
        [  1.3767, -11.0341,   9.9415],
        [  9.3424,  -7.3685,  -3.6807],
        [  5.2199,   4.6861,  -1.9304],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.8951,  -3.1570,   7.0742],
        [  1.7301,   7.4262,   5.0522],
        [  5.7485,  -9.9811,  -3.2355],
        [  3.3633,  13.4794,   0.5193],
        [  4.9152,  12.8653,  -1.4459],
        [  9.4925,   9.3811,   8.9842],
        [  6.2026,  12.1774,  -1.5887],
        [  5.4336,  -2.2676,   5.6933],
        [ 11.6634,   5.8050,   9.1590],
        [ -1.9487,   7.8590,  -3.3330],
        [  6.2675,  12.1379,  -8.9720],
        [ -7.0825,   5.2073,  -5.0497],
        [  2.0243,   7.3086,   7.5432],
        [  2.1477,   7.2550,   8.9688],
        [  4.7156,  -3.1831,  -8.0224],
        [  6.8407,  11.7666,   3.0045],
        [  6.3542,   0.6934,  -3.3840],
        [ -2.1382,   7.8339,   7.3524],
        [  8.6420,  -8.0580,   6.5781],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.6928,  12.4710,   3.6001],
        [  5.3860,  -2.3418,   1.0826],
        [ 12.5258,  -0.4200,  -9.3493],
        [  6.9653,  -9.3225,  -3.4727],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.3616,  -2.4750,   3.7765],
        [ -5.9987,   6.1949,  -6.2114],
        [ -5.6423,   6.4537,  -1.2091],
        [  7.0642,  11.6104,  -9.8316],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.8201,   3.5841,  -8.0609],
        [  0.1497,   7.8334,  -5.4412],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.3717,  -5.4769,  -9.2228],
        [  6.9956,  11.6590,  -7.6876],
        [  6.3266,   0.3292,   9.3502],
        [  5.8318,   3.5567,  -4.7692],
        [ 12.5717,   1.9266,   7.4903],
        [  0.3493,  -4.7727,  -6.5682],
        [  1.0093,   7.6567,   2.3526],
        [ -7.7842,   4.3466,  -1.7548],
        [  9.9058,  -6.7203,   5.6928],
        [ 11.7796,  -3.4696,   9.0088],
        [  4.4633,   5.6287,  -0.7754],
        [ -3.0004,   7.6607,  -7.2640]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.2478e-02, 6.8788e-02,  ..., 1.4914e+00, 9.5119e-01,
         5.2516e-01],
        [4.2478e-02, 3.0339e-04, 2.6322e-02,  ..., 1.5209e+00, 9.9094e-01,
         5.6764e-01],
        [6.8788e-02, 2.6322e-02, 0.0000e+00,  ..., 1.5401e+00, 1.0160e+00,
         5.9395e-01],
        ...,
        [1.4914e+00, 1.5209e+00, 1.5401e+00,  ..., 3.0339e-04, 7.5422e-01,
         1.1909e+00],
        [9.5119e-01, 9.9094e-01, 1.0160e+00,  ..., 7.5422e-01, 0.0000e+00,
         4.9659e-01],
        [5.2516e-01, 5.6764e-01, 5.9395e-01,  ..., 1.1909e+00, 4.9659e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.34127013,  9.47369628,  9.562174  ,  9.48969978,  9.60080639,
        9.70108409,  5.13125513, 10.65929968,  8.4486771 ,  6.33517349,
        7.62011734,  6.49538841,  5.01114719,  8.27853076,  6.7166297 ,
        6.83083863,  9.60080639, 13.83139024, 11.89105259,  8.62260327,
        9.48337063,  6.97106912, 13.77075174, 12.76408562,  7.05627461,
       10.7506423 ,  9.09119351,  8.16584537,  9.78286733,  6.3949043 ,
        9.69792513, 10.74655346, 10.63038565,  6.58184474,  6.76791483,
        5.2287058 , 10.35093485,  8.29203356, 11.11038551,  6.1463896 ,
       12.42554808,  6.01356004,  7.09887557, 11.00812996, 13.14777433,
        5.68942572,  8.51088913,  5.14607014,  5.91677188, 10.35485331,
       13.3607379 ,  7.09887557,  5.41348121, 13.35832725, 11.81423161,
       10.38897563,  8.89458273, 11.11969815, 11.89855267,  7.01475376,
        5.22903117, 12.30695098,  7.62509582, 11.51811603, 13.89264822,
       13.77222752, 13.34587027, 13.6660671 ,  5.88782913, 13.02818702,
        8.09704052, 13.66051884,  8.7907923 ,  7.58378655,  7.56617037,
        5.68942572, 13.6105595 ,  6.39187285,  8.12043983, 11.8159254 ,
        5.9510489 , 13.70893492,  5.87305005, 12.53285633, 11.63724792,
        9.14973319,  9.68324553,  8.62329466,  8.5724042 , 13.59056613,
       12.53285633,  6.83515201,  7.83486943,  9.76896904, 10.00410625,
       13.59673406,  6.33517349,  6.83083863, 12.71843462,  4.78544875,
        7.72291946,  8.91551231, 11.9702762 , 12.27993098,  7.18354885,
        8.22728399])}
{'P': tensor([[1.0875e-02, 1.8200e-01, 1.8034e-01,  ..., 6.0064e-06, 1.0201e-05,
         1.1641e-07],
        [1.8687e-04, 2.3159e-02, 4.4523e-01,  ..., 3.0610e-05, 2.9249e-05,
         2.6888e-07],
        [2.7431e-04, 1.5804e-03, 4.0002e-02,  ..., 5.6266e-05, 5.1663e-05,
         4.5395e-07],
        ...,
        [3.3033e-07, 2.8603e-07, 2.7713e-07,  ..., 8.6615e-03, 1.9142e-03,
         2.2332e-04],
        [1.0708e-04, 6.6910e-05, 6.2969e-05,  ..., 1.0404e-04, 6.5617e-02,
         2.7198e-03],
        [2.0398e-05, 9.4912e-06, 8.3166e-06,  ..., 8.9783e-07, 6.5317e-05,
         6.4451e-02]]), 'X': tensor([[ -9.3419,   0.5972,  -1.2841],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 10.5057,  -5.9099,  -5.1599],
        [  3.3744,   6.5623,  -6.3636],
        [  5.6248,   3.9984,   8.4933],
        [  4.1537,   5.9340,  -6.0423],
        [  9.8082,  -6.8396,   4.9615],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.9809,   7.6642,  -1.4217],
        [  4.6357,   5.4415,   7.9067],
        [  0.6952,  -4.8060,   8.8976],
        [ -7.8849,  -6.3115,  -4.7479],
        [  0.5250, -11.0306,   5.5358],
        [  2.0870,  -4.6877,   9.6823],
        [  0.1041, -11.0045,   2.4311],
        [ -7.6734,   4.4980,  -5.4248],
        [ -9.3107,  -2.7697,  -3.2919],
        [  6.3515,   0.6418,  -6.5823],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.6801,   2.7809,  -0.8443],
        [ 11.4703,  -4.2006,   7.4086],
        [  1.4150,  13.9647,  -4.0915],
        [  5.9193,  -1.3176,   5.1739],
        [ 10.6283,  -5.7253,   7.4878],
        [  1.7232,  -4.7574,   3.1270],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 11.0275,  -5.0666,   8.4154],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.0995,  -6.4739,   2.0974],
        [ -7.9667,   4.0814,  -0.6397],
        [  4.2096,   5.8815,   1.9230],
        [  3.4920,  -4.1366,  -8.3955],
        [  3.9515,  -3.8425,   6.1729],
        [  0.3038,   7.8092,   0.4164],
        [  1.6954,  -4.7616,   9.8198],
        [  2.7590,  13.6621,   9.4496],
        [ 11.0373,   7.1299,  -8.5895],
        [  2.0899,  -4.6870,  -7.4260],
        [ -8.3535,  -5.5118,   9.8399],
        [  7.9141,  -8.6656,   9.6948],
        [  3.2809,  13.5061,  -4.0199],
        [  2.7558,  -4.4844,   0.0636],
        [ 10.1503,   8.5414,  -7.2681],
        [  5.1142,  -2.7230,  -4.1966],
        [ 11.6019,   5.9540,  -5.1715],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.6634,   5.8050,   9.1590],
        [  6.3124,   0.2057,  -0.4593],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.2188,  -2.5839,   2.1833],
        [  9.9756,  -6.6330,  -3.3429],
        [ 11.7418,   5.6070,  -1.4192],
        [  9.1409,  -7.5789,   1.5814],
        [  3.3094,  13.4969,  -3.6507],
        [ -8.0766,  -6.0046,  -3.1873],
        [  5.3924, -10.1396,  -4.0863],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -3.4100,   7.5436,  -7.6980],
        [ -8.7735,   2.5640,  -6.3313],
        [  0.6933,   7.7336,   2.1179],
        [ -9.2707,   0.9661,   4.4863],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.8940, -11.0402,  -0.9783],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.3305,   1.6119,   1.1538],
        [ 12.5789,   0.1722,   1.6470],
        [ -3.6381,   7.4683,   7.3544],
        [ 11.4743,   6.2478,  -2.1100],
        [  3.4140,  -4.1803,  -8.8224],
        [ 12.4877,  -0.7222,   8.8347],
        [  0.1287, -11.0065,   6.5757],
        [ -1.9714,   7.8563,  -1.8084],
        [  8.7715,  -7.9390,   5.2900],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.2345,  -3.1345,   3.9715],
        [ 10.5678,   7.9261,  -0.6100],
        [ -7.8126,   4.3066,   5.1791],
        [ 10.4994,  -5.9193,  -7.5153],
        [  5.8902,  -1.3887,   4.0388],
        [  2.0243,   7.3086,   7.5432],
        [ -9.4351,  -1.8891,   3.5642],
        [  3.9591, -10.6391,  -6.2557],
        [ 11.4495,   6.3026,   6.8257],
        [  9.6816,   9.1536,  -5.8543],
        [ 12.5832,   1.7543,  -7.6134],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -1.6086,   7.8928,   4.3385],
        [  1.1350,  14.0104,   0.6321],
        [  2.6321,  -4.5296,  -2.5159],
        [  2.0729,   7.2878,   7.0880],
        [  2.2076,  13.8032,  -1.5099],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.5456,   6.0861,   8.1101],
        [ -0.5709,   7.9055,   7.6089],
        [  1.1274,  14.0116,   8.1753],
        [  9.0004,  -7.7195,   4.6945],
        [  6.3551,   0.7149,   1.3344],
        [  9.4925,   9.3811,   8.9842]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.2331e-02, 7.6850e-02,  ..., 8.5774e-01, 9.0520e-01,
         1.5293e+00],
        [8.2331e-02, 0.0000e+00, 2.2764e-02,  ..., 8.6034e-01, 9.4602e-01,
         1.5389e+00],
        [7.6850e-02, 2.2764e-02, 0.0000e+00,  ..., 8.8080e-01, 9.5968e-01,
         1.5586e+00],
        ...,
        [8.5774e-01, 8.6034e-01, 8.8080e-01,  ..., 0.0000e+00, 4.3535e-01,
         6.8244e-01],
        [9.0520e-01, 9.4602e-01, 9.5968e-01,  ..., 4.3535e-01, 1.0727e-04,
         7.7301e-01],
        [1.5293e+00, 1.5389e+00, 1.5586e+00,  ..., 6.8244e-01, 7.7301e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.36093562,  9.60359158,  9.62958973,  9.58072286,  9.54154096,
        9.61942445, 12.05395056,  7.37902766,  6.90115901,  7.24326992,
       11.95743558,  5.52416368,  7.72668467,  7.14837514,  4.85603886,
       10.09979747, 11.04313641,  5.13125513, 11.0050379 ,  8.89458273,
        9.71392179,  6.383897  , 12.53345445, 10.41909211,  7.28793663,
        9.11473391, 12.21532282, 14.03618649,  6.06416957, 12.07225469,
        5.05989547,  9.78286733, 12.13568598, 12.13568598, 11.99633819,
        8.95133961,  7.23276582,  5.41348121,  5.51177501,  7.81509631,
        5.05444288, 13.93790235, 13.1399181 ,  5.13182506, 10.00802627,
       11.73567759, 13.89886469,  5.26342203, 13.26590071,  5.79391825,
       13.04051744, 10.31613269, 13.02818702,  6.31576264, 12.38881635,
        5.82345867, 11.97957425, 13.01187484, 11.87412439, 13.89671314,
       10.06407839, 11.48434767, 11.1913598 ,  9.97964013, 10.46739073,
        8.27853076,  9.14044984,  7.76457095,  9.32094055, 10.78533838,
       11.07637531,  5.53175275,  6.53250937, 12.58006038,  8.30729352,
       13.06498554,  5.39723241, 12.5085995 , 11.00726483,  8.09984231,
       11.83075005,  5.21280156,  9.75200848, 13.20988996,  8.92098752,
       12.05300943,  6.05165248,  7.58378655,  9.62238828, 11.35182589,
       13.06957425, 13.32374056, 12.70489563, 12.49317563,  8.0550286 ,
       14.05633441,  5.23878371,  7.57685908, 13.97857591,  8.41148722,
       13.05148826,  7.92607342, 14.05687815, 11.85743425,  6.39520397,
       13.34587027])}
{'P': tensor([[1.3903e-02, 2.3736e-01, 1.2715e-01,  ..., 8.8401e-08, 1.3064e-06,
         7.3026e-06],
        [3.1758e-04, 1.8152e-02, 2.0750e-01,  ..., 8.8391e-08, 1.9527e-06,
         9.1460e-06],
        [9.3826e-05, 1.7888e-04, 2.4656e-02,  ..., 1.2983e-07, 8.4241e-06,
         2.7919e-05],
        ...,
        [4.2980e-05, 2.2007e-05, 1.0788e-05,  ..., 9.9050e-02, 6.9897e-07,
         1.6816e-05],
        [4.5410e-07, 3.6353e-07, 4.7451e-07,  ..., 1.7677e-04, 6.2215e-02,
         5.2449e-03],
        [8.6442e-06, 6.7496e-06, 8.7772e-06,  ..., 4.9851e-04, 3.3523e-03,
         6.8544e-02]]), 'X': tensor([[-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.6616e-02, 8.3809e-02,  ..., 1.4507e+00, 1.2299e+00,
         1.1696e+00],
        [5.6616e-02, 0.0000e+00, 7.8017e-02,  ..., 1.5069e+00, 1.2860e+00,
         1.2261e+00],
        [8.3809e-02, 7.8017e-02, 3.0339e-04,  ..., 1.4984e+00, 1.2580e+00,
         1.2120e+00],
        ...,
        [1.4507e+00, 1.5069e+00, 1.4984e+00,  ..., 0.0000e+00, 4.2385e-01,
         2.9992e-01],
        [1.2299e+00, 1.2860e+00, 1.2580e+00,  ..., 4.2385e-01, 0.0000e+00,
         2.3721e-01],
        [1.1696e+00, 1.2261e+00, 1.2120e+00,  ..., 2.9992e-01, 2.3721e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.36256096,  9.49511524,  9.77777346,  9.84314712,  9.77588801,
        9.79814997, 14.10045797,  8.04483495, 10.57302546, 13.38091121,
        5.47866221, 12.46513646,  8.81930525, 10.33712258, 10.05558893,
       10.52587243,  5.91408504,  5.21280156,  4.98461983,  9.17595899,
       13.06957425, 10.57302546, 12.13568598, 12.17451685,  9.20340031,
        6.90115901,  6.18668463,  5.51177501, 11.785566  ,  9.71392179,
        5.53175275, 10.8992524 , 13.66896739,  9.50913849, 11.36089355,
        9.70108409, 11.48734978, 13.43127131,  5.23878371,  6.19707944,
        8.8737556 , 13.32374056,  7.58378655,  5.10446653,  8.31625171,
       11.75618513, 11.2845608 , 11.57249984,  5.18047638, 12.3828957 ,
        7.01475376, 10.61047596,  9.49511524,  6.36004393,  5.1805345 ,
       10.84493732, 11.4088043 ,  9.61942445,  5.94541803,  9.3109884 ,
        9.64157461, 13.25336001, 13.35013374, 12.1460931 ,  8.28968067,
       11.05317597, 10.68031351, 13.99877413,  7.06372172,  9.37296102,
       13.96987806, 13.68672855, 10.33712258,  8.28968067, 10.19036081,
       11.65923396,  5.94541803,  7.20482224,  9.41244865, 12.84056604,
       13.26590071, 11.28215429, 13.86480175, 11.89855267,  7.5403822 ,
        4.83371341, 12.19805394, 11.9182956 , 13.32645552,  5.87132718,
       13.12218013, 13.43127131,  6.33075682, 10.76154256, 11.11969815,
       12.18045978, 10.61472636, 14.05633441, 12.97370371,  6.90115901,
        8.5724042 ,  6.27905292,  5.23878371, 13.66896739, 11.73848065,
        5.5205189 ])}
{'P': tensor([[9.0940e-03, 1.8504e-01, 1.8095e-01,  ..., 1.1305e-04, 1.2539e-02,
         4.3233e-05],
        [1.0308e-04, 9.2906e-03, 3.2414e-01,  ..., 1.7831e-04, 2.2982e-02,
         6.8361e-05],
        [1.5017e-04, 3.5580e-04, 1.7449e-02,  ..., 4.0045e-04, 3.9701e-02,
         1.5333e-04],
        ...,
        [1.8839e-06, 2.2845e-06, 2.5032e-06,  ..., 2.7010e-02, 3.4599e-06,
         6.4471e-01],
        [3.4779e-04, 7.6544e-04, 9.7073e-04,  ..., 2.3704e-03, 1.5575e-01,
         1.0940e-03],
        [4.7202e-06, 5.2366e-06, 5.6470e-06,  ..., 6.8528e-04, 6.2262e-06,
         1.1026e-01]]), 'X': tensor([[ -9.3464,   0.5708,  -1.6397],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.3167,   6.6022,   6.3554],
        [ -2.5056,   7.7720,   5.5878],
        [  3.2692,  13.5098,  -4.9032],
        [  6.3589,   0.8239,  -8.4230],
        [  3.9515,  -3.8425,   6.1729],
        [  6.9314,  -9.3435,  -1.2729],
        [ -8.8363,   2.4086,  -5.6296],
        [ -9.4159,  -2.0739,   9.6841],
        [ -8.0534,  -6.0429,   2.4583],
        [ 10.9394,   7.3069,  -8.9365],
        [  7.5064,  -8.9651,  -0.0994],
        [  1.3705,  13.9723,   7.9266],
        [ -3.7964,  -9.9294,   7.7147],
        [ 12.3453,  -1.5462,   7.0212],
        [  5.8450,  -9.9356,   3.0767],
        [  2.1625,  -4.6697,   7.0658],
        [  6.3303,   1.6145,  -8.4083],
        [  2.6310, -10.9195,   3.8054],
        [  2.1771, -10.9776,  -3.2007],
        [  5.0190,   4.9705,  -7.9752],
        [  5.4429,  -2.2529,   9.7306],
        [  8.5716,  10.3596,   6.8818],
        [  5.3118,   4.5452,   4.2114],
        [ 12.4837,   2.7510,   0.4449],
        [ -9.1960,  -3.2964,   1.6921],
        [  9.9058,  -6.7203,   5.6928],
        [ -8.8631,  -4.3728,  -1.1746],
        [  6.0054,  -1.0902,   8.3079],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.6675,  -4.8569,  -9.1769],
        [  3.1064,  13.5607,   6.7605],
        [  8.5716,  10.3596,   6.8818],
        [  8.7131,  10.2216,   7.7915],
        [  5.4867,  12.5816,   2.9549],
        [  4.7544,  -3.1425,  -1.5315],
        [  7.9403,  -8.6455,  -8.7433],
        [  0.1460,   7.8340,  -9.4614],
        [ -1.2454,   7.9127,  -0.5715],
        [  6.2576,   2.1359,  -3.3398],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.7078,   2.7182,   1.8146],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.9152,  12.8653,  -1.4459],
        [  9.9217,  -6.7006,  -5.6728],
        [  6.3583,   1.1729,   8.7835],
        [ -4.3359,  -9.6431,  -9.0616],
        [  1.6308,  13.9254,  -2.0984],
        [  1.6308,  13.9254,  -2.0984],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.5154,   2.5072,  -9.3274],
        [  1.8447,  -4.7372,   8.1235],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 11.9122,   5.1397,  -1.5498],
        [  4.3459,  -3.5327,   0.3667],
        [  2.8602,   6.8905,   5.8779],
        [  6.2247,  -0.3150,  -9.3657],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3550,   0.7129,   0.4917],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.1477,   7.2550,   8.9688],
        [ 10.7566,   7.6212,  -2.3279],
        [  5.5240,  -2.1198,   6.2055],
        [ -5.6174,   6.4708,  -1.7339],
        [  3.7121,   6.3117,   0.6738],
        [  3.1271, -10.8343,   2.6442],
        [ -8.3653,   3.4184,   9.3454],
        [ 11.8951,  -3.1570,   7.0742],
        [  9.3560,   9.5391,   9.4943],
        [  2.0729,   7.2878,   7.0880],
        [  9.6662,   9.1725,   8.1056],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -2.4956, -10.4682,   3.5632],
        [  9.9058,  -6.7203,   5.6928],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.3804,  -2.3504,   7.2744],
        [ -7.6650,   4.5092,   0.2520],
        [  4.5057,  -3.3894,  -0.8160],
        [ -6.0175,  -8.4541,  -9.8408],
        [  3.4009,  -4.1874,  -5.9751],
        [  7.2340,  -9.1505,  -5.8321],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -3.9508,  -9.8515,   1.7923],
        [ -9.0961,   1.6529,   3.1553],
        [  1.9312,  13.8649,   8.3563],
        [  5.3118,   4.5452,   4.2114],
        [  9.1409,  -7.5789,   1.5814],
        [ -9.3384,  -2.6161,  -4.7630],
        [  0.6757, -11.0361,  -1.3047],
        [ -7.6734,   4.4980,  -5.4248],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.8363,   2.4086,  -5.6296],
        [  3.4140,  -4.1803,  -8.8224],
        [  1.0391,  -4.8141,  -0.0768],
        [ -3.4488, -10.0934,   2.5195],
        [ -8.0608,   3.9362,  -5.9567],
        [ -0.8675, -10.8815,   6.8561],
        [ -8.5607,  -5.0938,  -5.4531],
        [  1.1644, -11.0394,   7.8511]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0803, 0.0834,  ..., 0.7344, 0.3034, 0.7994],
        [0.0803, 0.0000, 0.0137,  ..., 0.7977, 0.2305, 0.8686],
        [0.0834, 0.0137, 0.0000,  ..., 0.7914, 0.2356, 0.8639],
        ...,
        [0.7344, 0.7977, 0.7914,  ..., 0.0000, 1.0267, 0.1244],
        [0.3034, 0.2305, 0.2356,  ..., 1.0267, 0.0000, 1.0991],
        [0.7994, 0.8686, 0.8639,  ..., 0.1244, 1.0991, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.36378438,  9.54154096,  9.58072286,  9.62958973,  9.60359158,
        9.61942445,  9.88310614,  7.38844703,  8.16584537, 13.89974693,
        6.41202986,  5.51177501, 11.63383221,  9.15866607,  9.64157461,
       10.06850677, 13.15525803, 11.69264243, 14.0393949 , 10.63038565,
       12.44177641, 11.52733581,  5.14607014,  6.53290663, 11.23201619,
       11.1913598 ,  7.06372172,  5.89074104, 13.44593519,  6.99097013,
       12.78327037,  9.76896904, 11.9702762 ,  9.88310614,  6.10360073,
       11.92133129,  9.93553424, 13.91198191, 13.44593519, 13.43127131,
       13.72594022,  5.69913817, 11.73848065,  7.83534987,  8.01009083,
        6.61211224, 10.06850677,  9.12220271,  5.18047638, 13.77222752,
       11.97238624,  6.46560181, 10.57302546, 14.02058621, 14.02058621,
       13.01728313, 12.76408562,  5.08372149,  9.94285511, 12.97370371,
        5.60064319,  7.46053281,  6.23262289, 12.87752052,  6.3949043 ,
       10.08001894,  7.56617037, 13.18278606,  5.91677188,  8.56889738,
        7.32233733, 11.27656488,  9.0368458 , 12.30695098, 13.3614598 ,
        7.57685908, 13.32557133,  9.50913849, 10.76154256, 11.9702762 ,
       10.00794468,  5.87132718,  8.89301516,  5.63824971, 10.37698435,
        5.39450305, 11.66453059, 12.32545637, 10.61417181,  9.24501829,
       13.99877413,  6.99097013, 11.87412439,  9.69792513, 11.05672136,
        8.89458273, 13.13205357, 12.59969936,  9.15866607,  5.39723241,
        4.92497514, 10.66631784,  8.97052115, 10.91601655,  9.96152788,
       11.10066126])}
{'P': tensor([[6.4089e-03, 1.6725e-01, 1.5012e-01,  ..., 1.0050e-02, 4.8822e-06,
         5.5355e-06],
        [1.9799e-04, 1.4978e-02, 2.1624e-01,  ..., 3.8469e-02, 1.4953e-05,
         1.8822e-05],
        [5.5019e-04, 2.1217e-03, 6.0007e-02,  ..., 1.2590e-01, 5.7656e-05,
         1.0188e-04],
        ...,
        [3.7689e-04, 6.2148e-04, 8.8040e-04,  ..., 2.0363e-01, 1.0601e-03,
         8.5245e-04],
        [2.1643e-05, 2.0372e-05, 1.7174e-05,  ..., 2.5856e-05, 7.0750e-02,
         2.3480e-05],
        [1.5627e-06, 1.3874e-06, 1.4223e-06,  ..., 1.0510e-06, 4.9910e-05,
         4.1508e-02]]), 'X': tensor([[ -9.3604,   0.4855,  -3.0158],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.4378,  -1.8606,  -2.5668],
        [  1.7228,  -4.7575,   8.3396],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.2311,  -2.0445,   1.3545],
        [  4.2096,   5.8815,   1.9230],
        [ -7.7842,   4.3466,  -1.7548],
        [  8.2154,  -8.4262,   8.0171],
        [  6.3534,   1.2990,  -7.2942],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.2770,  -0.0379,   2.4195],
        [ 11.0891,  -4.9558,  -2.3178],
        [  9.3753,   9.5171,   3.9131],
        [  4.5057,  -3.3894,  -0.8160],
        [ 12.5996,   1.3971,  -1.0694],
        [ -8.0611,  -6.0302,   4.6962],
        [ -9.1749,  -3.3803,   6.4629],
        [  6.3200,   0.2690,   1.5900],
        [  2.4697,  -4.5836,   1.4593],
        [ 12.5154,   2.5072,  -9.3274],
        [ -9.3464,   0.5708,  -1.6397],
        [  6.0890,  -0.8382,   2.9166],
        [ 10.4737,  -5.9570,  -1.2108],
        [  9.9750,  -6.6338,  -1.2440],
        [  6.3460,   1.4252,   1.2291],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.0140,  -1.0661,  -2.5040],
        [ 12.2218,   4.0953,  -8.2072],
        [  8.9284,  10.0036,  -6.9210],
        [ -9.4731,  -0.7120,  -6.0271],
        [  4.5767,   5.5071,   3.9069],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.4296,  -1.9458,   4.0833],
        [ 11.4703,  -4.2006,   7.4086],
        [ -9.3705,   0.4208,  -3.5815],
        [ 10.4758,  -5.9539,   1.3667],
        [ -5.1377,   6.7751,  -0.4438],
        [  4.7813,  -3.1139,   8.2691],
        [  1.6954,  -4.7616,   9.8198],
        [  0.1252, -11.0062,   7.2564],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 12.2692,  -1.8882,   3.9253],
        [  7.9141,  -8.6656,   9.6948],
        [ -4.9217,   6.8979,  -1.7667],
        [  3.4920,  -4.1366,  -8.3955],
        [ -7.4602,   4.7718,  -4.6277],
        [ -6.3489,   5.9117,   6.5161],
        [ -7.3333,   4.9246,  -7.9212],
        [ -3.9700,   7.3453,  -4.1486],
        [  0.1287, -11.0065,   6.5757],
        [ 10.4994,  -5.9193,  -7.5153],
        [  1.8320,   7.3871,   5.1452],
        [  9.0129,  -7.7072,   9.1400],
        [ -8.3717,  -5.4769,  -9.2228],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.4932,   7.0903,  -5.4214],
        [  3.2809,  13.5061,  -4.0199],
        [  2.0243,   7.3086,   7.5432],
        [  3.2647, -10.8066,  -0.3768],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.7794,   5.5086,  -9.6929],
        [  3.0719, -10.8449,  -4.1673],
        [  5.8450,  -9.9356,   3.0767],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.8318,   3.5567,  -4.7692],
        [  5.8323,  -1.5228,   5.1132],
        [  1.7301,   7.4262,   5.0522],
        [  4.5430,   5.5438,  -3.8669],
        [  2.7685, -10.8982,  -6.8928],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.3523, -10.5232,   8.2138],
        [  0.9659,  -4.8144,  -0.9594],
        [  6.2975,   0.0947,   5.1791],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.2849, -10.9655,  -7.6542],
        [  5.4426, -10.1181,  -5.1268],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.5996,   1.3971,  -1.0694],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.6681,  -4.8556,   9.2518],
        [ -7.8280,   4.2848,  -7.1955],
        [ 11.6165,  -3.8715,   6.4594],
        [ -9.4735,  -1.2620,  -3.6742],
        [  6.2038,   2.4020,  -3.1692],
        [ -4.8751,  -9.3148,   4.0406],
        [  2.7685, -10.8982,  -6.8928],
        [ -7.5603,   4.6459,   8.2002],
        [  4.6357,   5.4415,   7.9067],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -9.3604,   0.4855,  -3.0158],
        [  9.4516,  -7.2501,  -1.8868],
        [ 10.4758,  -5.9539,   1.3667],
        [ 11.2571,   6.7072,   8.3693],
        [  4.1348,   5.9514,  -0.8856],
        [  5.9843,  -9.8679,  -6.1449],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.3602,  -2.3808,   7.2098],
        [  8.0381,  -8.5690,  -8.9523]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.5046e-02, 6.8227e-02,  ..., 2.7803e-01, 1.1552e+00,
         8.0065e-01],
        [4.5046e-02, 0.0000e+00, 4.2179e-02,  ..., 2.5898e-01, 1.1815e+00,
         8.4546e-01],
        [6.8227e-02, 4.2179e-02, 0.0000e+00,  ..., 2.9509e-01, 1.2202e+00,
         8.5169e-01],
        ...,
        [2.7803e-01, 2.5898e-01, 2.9509e-01,  ..., 0.0000e+00, 1.0637e+00,
         9.8814e-01],
        [1.1552e+00, 1.1815e+00, 1.2202e+00,  ..., 1.0637e+00, 2.1453e-04,
         1.0426e+00],
        [8.0065e-01, 8.4546e-01, 8.5169e-01,  ..., 9.8814e-01, 1.0426e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.37296102,  9.5014146 ,  9.5642993 ,  9.43737523,  9.55720651,
        9.61942445,  5.05981914,  6.25586297, 12.40074901,  7.23276582,
        8.91551231, 11.76830981,  6.48486171,  9.37989631,  6.27715105,
       12.1460931 , 13.35927527,  5.63824971, 12.67680439, 10.06703358,
        9.77777346,  6.32572323,  5.20659892, 12.76408562,  9.36378438,
        6.1463896 , 12.04924159, 11.97949273,  6.50409372, 11.54672755,
        6.10774544, 12.889695  , 13.40850328,  9.4997987 ,  7.16058587,
        9.9353921 ,  5.94535529,  9.6282763 , 12.21532282,  9.37989631,
       12.0495521 ,  8.50278896,  5.70591986,  5.05444288, 11.00694688,
        9.60080639,  9.55720651, 12.41366754, 11.73567759,  8.47371868,
        5.41348121,  8.85572292,  8.67502348,  8.83341608,  8.34949383,
       11.00726483, 12.05300943,  7.61088125, 11.85890664, 10.00410625,
       11.80092578,  7.5158489 , 13.89886469,  7.58378655, 11.28895489,
       13.04217487, 13.00380624, 11.27160035, 11.52733581, 12.17451685,
        6.83083863,  6.02779201,  7.62509582,  7.16747878, 11.24434374,
        9.60412735, 11.38774149,  4.91038884,  6.29821673,  9.9353921 ,
       11.20100795, 11.4890861 ,  8.41171369, 12.67680439,  6.2322769 ,
        9.97220799,  9.9353921 ,  8.92395381, 12.24467041,  9.55720651,
        6.65259775, 10.51340829, 11.24434374,  8.8737556 ,  7.14837514,
        6.2134961 , 12.52372629,  9.37296102, 11.9120321 , 12.0495521 ,
       13.103709  ,  7.24677664, 11.54071859, 10.00794468,  5.86519525,
       11.74901778])}
{'P': tensor([[9.4302e-03, 2.3819e-01, 2.0940e-01,  ..., 2.4135e-04, 4.1141e-04,
         1.4310e-05],
        [1.9724e-04, 2.0624e-02, 2.9782e-02,  ..., 1.1264e-03, 1.1197e-03,
         5.5762e-05],
        [1.4486e-04, 6.7381e-03, 1.4563e-02,  ..., 8.5895e-04, 1.0038e-03,
         4.5682e-05],
        ...,
        [3.1863e-06, 3.8267e-06, 3.9173e-06,  ..., 3.4758e-02, 1.8519e-05,
         1.0454e-02],
        [1.9551e-05, 3.3573e-05, 3.7513e-05,  ..., 1.2920e-02, 6.2410e-02,
         2.4067e-02],
        [3.5888e-07, 3.7198e-07, 4.0023e-07,  ..., 3.8563e-05, 2.1343e-06,
         1.2659e-02]]), 'X': tensor([[ -9.3705,   0.4208,  -3.5815],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -9.4735,  -0.7274,  -2.6749],
        [  4.1140,  -3.7220,  -0.1273],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.7349,  -3.5838,   2.1644],
        [ -5.0943,  -9.1682,   0.5073],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.5198,  -4.0923,   5.5088],
        [  9.2811,  -7.4336,   5.1771],
        [ 11.9695,  -2.9407,  -6.3498],
        [  2.1625,  -4.6697,   7.0658],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.2226,  -4.2802,  -4.0428],
        [  9.3836,   9.5075,  -5.9278],
        [  3.2692,  13.5098,  -4.9032],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.4167, -10.1293,   6.2156],
        [  6.2365,  -0.2578,  -6.0212],
        [ -8.9178,   2.1936,   7.5388],
        [  7.7794,  11.0629,   8.9069],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.4732,  -2.2041,  -4.3830],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.2707,   0.9661,   4.4863],
        [  0.3231,  -4.7691,  -1.6120],
        [ -1.7009,   7.8851,  -8.2437],
        [ -6.4464,   5.8272,   7.5038],
        [  6.3048,   0.1469,  -9.1691],
        [ -0.9715, -10.8630,   0.3555],
        [  6.2902,   0.0451,   3.7116],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.3924, -10.1396,  -4.0863],
        [  4.6231,   5.4556,   8.4973],
        [  9.0652,   9.8597,  -3.5949],
        [  6.3087,   1.8053,  -1.9481],
        [ -4.0868,   7.2981,   8.1874],
        [  6.3031,   1.8477,  -7.1706],
        [ -0.8885,   7.9160,   0.0571],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.5240,  -2.1198,   6.2055],
        [  2.5318,  13.7231,   6.3692],
        [  0.8940, -11.0402,  -0.9783],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.3261,  13.7749,   5.9121],
        [  5.2188,  -2.5839,   2.1833],
        [  1.7659,   7.4127,   4.7871],
        [  4.5954,  13.0099,  -4.4439],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 10.5955,   7.8824,  -3.7006],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.5224,  -0.4497,   4.8448],
        [  4.2041,  -3.6509,  -4.9488],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.2345,  -3.1345,   3.9715],
        [ -1.8788, -10.6567,   3.7068],
        [ -7.6734,   4.4980,  -5.4248],
        [ -0.0254, -10.9932,   3.2200],
        [ -2.4956, -10.4682,   3.5632],
        [  4.4915,   5.5990,   7.0405],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.5636, -10.9293,   9.6519],
        [  6.9314,  -9.3435,  -1.2729],
        [  1.3339,   7.5627,  -0.9034],
        [ 11.9936,  -2.8677,   6.0614],
        [  4.9337,  12.8566,  -3.6582],
        [  2.3373,  -4.6232,  -5.2246],
        [ 10.2712,   8.3707,  -4.9743],
        [ 12.0772,  -2.6014,  -7.1832],
        [  9.3607,   9.5337,  -4.5412],
        [  1.6954,  -4.7616,   9.8198],
        [  2.7558,  -4.4844,   0.0636],
        [ 10.9394,   7.3069,  -8.9365],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.0330,  -6.0765,   6.7945],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.0560,  -0.9421,  -5.0334],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.3569,   0.7591,   6.5011],
        [  6.3211,   1.7017,  -0.2164],
        [ -5.6423,   6.4537,  -1.2091],
        [  0.3038,   7.8092,   0.4164],
        [ -8.6229,   2.9068,   2.4429],
        [  4.5503,   5.5359,   7.4735],
        [  4.5057,  -3.3894,  -0.8160],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 12.4592,  -0.9168,  -6.7571],
        [  5.2608,  -2.5256,  -9.1446],
        [ -5.9940,   6.1985,   5.9566],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.3515,   0.6418,  -6.5823],
        [ -9.3411,  -2.6004,  -2.7062],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3305,   1.6119,   1.1538],
        [ 11.0891,  -4.9558,  -2.3178],
        [  0.6757, -11.0361,  -1.3047],
        [ -6.4620,  -8.0449,   6.2719],
        [  4.7482, -10.3907,   5.9149]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0596, 0.0574,  ..., 0.3076, 0.7653, 0.9235],
        [0.0596, 0.0000, 0.0300,  ..., 0.3658, 0.7841, 0.9642],
        [0.0574, 0.0300, 0.0000,  ..., 0.3514, 0.7541, 0.9372],
        ...,
        [0.3076, 0.3658, 0.3514,  ..., 0.0000, 0.6661, 0.6918],
        [0.7653, 0.7841, 0.7541,  ..., 0.6661, 0.0000, 0.4030],
        [0.9235, 0.9642, 0.9372,  ..., 0.6918, 0.4030, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.37989631,  9.55720651,  9.5642993 ,  9.71392179,  9.61942445,
        9.5014146 ,  5.54776934,  6.81961996, 12.26997165, 10.48840476,
       13.79762415, 12.22503754, 11.89105259, 12.32545637,  5.14607014,
        9.9353921 ,  5.35774573, 13.35832725, 13.89974693,  9.7751189 ,
       11.48663767,  6.24187044,  9.18358562, 13.52430176,  6.14086582,
        5.90035324,  9.43737523,  9.32094055,  4.78002665,  8.06643487,
        8.68979052,  6.30647746, 10.90638066,  6.29034958,  5.94541803,
       11.48434767,  7.15100303, 13.39372426,  6.5618968 ,  8.36445758,
        6.56833864,  7.96575782, 12.79126785,  5.91677188, 13.95472331,
       11.07637531,  5.24137909, 13.96987806,  5.82345867,  7.62011734,
       13.79762415, 12.05300943, 13.20597932,  5.47702766, 12.53047489,
        5.56809335, 10.31886066,  7.53884104,  9.75200848, 10.82106505,
        8.89458273, 10.99326577, 10.76154256,  7.17789666, 10.33958639,
       11.22596994, 11.63383221,  7.67939546, 12.33167382, 13.77075174,
        5.18047638, 13.25017211, 12.35421461, 13.36092624,  5.05444288,
        5.26342203, 13.15525803,  8.37954853, 10.07238463,  5.38342077,
        9.11473391,  9.79814997,  6.12886024, 10.32715715,  6.4020404 ,
        6.54615887,  8.5724042 ,  7.81509631,  9.09963894,  7.16598273,
        5.63824971, 12.05395056, 12.49291768,  5.83560009,  8.62260327,
       13.27717982,  9.70108409,  9.53015376,  6.383897  ,  9.69629   ,
        5.76165934,  6.53250937, 12.1460931 , 11.05672136, 10.31886066,
       11.42420508])}
{'P': tensor([[7.2409e-03, 2.3392e-01, 1.4305e-01,  ..., 8.3214e-06, 4.9192e-05,
         1.0582e-06],
        [2.6417e-04, 1.2923e-02, 3.2304e-01,  ..., 1.5420e-05, 9.6758e-05,
         1.8579e-06],
        [3.5348e-04, 4.2252e-04, 3.0757e-02,  ..., 4.1169e-05, 2.9859e-04,
         4.2853e-06],
        ...,
        [1.2398e-04, 1.1118e-04, 8.5595e-05,  ..., 1.2330e-01, 2.3958e-03,
         3.0836e-04],
        [1.0780e-04, 1.0362e-04, 9.5062e-05,  ..., 3.7429e-02, 1.0989e-01,
         7.1810e-04],
        [2.5279e-04, 2.3263e-04, 1.8310e-04,  ..., 1.9107e-05, 4.6887e-05,
         6.8941e-02]]), 'X': tensor([[-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0234, 0.0600,  ..., 0.9240, 0.8642, 1.0840],
        [0.0234, 0.0000, 0.0367,  ..., 0.9465, 0.8852, 1.0931],
        [0.0600, 0.0367, 0.0000,  ..., 0.9808, 0.9169, 1.1051],
        ...,
        [0.9240, 0.9465, 0.9808,  ..., 0.0000, 0.1666, 0.8804],
        [0.8642, 0.8852, 0.9169,  ..., 0.1666, 0.0000, 0.7168],
        [1.0840, 1.0931, 1.1051,  ..., 0.8804, 0.7168, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.38232121,  9.43112957,  9.55720651,  9.69792513,  9.5642993 ,
        9.50913849, 10.51340829, 11.54071859,  6.54615887, 11.83570132,
        9.97964013, 13.14777433, 13.79676654,  7.5403822 , 11.04313641,
        5.80860165,  5.8828986 , 13.77604293,  9.24501829, 14.05687815,
        7.48311002, 10.42667093, 10.37698435,  5.64500512, 10.42667093,
        5.92952538, 11.59512194, 13.98691426,  7.06272256,  8.62162372,
        9.76896904,  8.30125201,  8.50268322, 10.31613269,  9.55593374,
        5.64464807,  4.91336994,  6.59176018, 11.87412439,  8.14433542,
        7.63726134, 10.36844813, 11.785566  ,  7.54127425, 10.48315436,
       12.3344536 , 11.65923396, 13.26209545,  9.43112957,  8.16714836,
        5.64500512,  4.89869205,  4.92497514,  9.17595899,  5.10444097,
       10.14044489,  6.80559669, 10.24557047, 11.20100795, 13.68063098,
        7.56617037,  6.34059021,  5.53175275, 10.06286074, 13.37121923,
        7.27340834,  7.34250948,  8.55982105,  9.9353921 ,  5.5366698 ,
       10.69963954,  7.73692436,  9.68323686, 11.27160035,  4.78258092,
        5.14371005, 11.89105259,  7.14837514, 13.0743399 ,  8.06990483,
        7.50500064, 10.90638066,  5.53175275, 12.27993098, 12.32545637,
       10.87409005, 10.07238463,  6.83515201, 13.96905756, 13.09737053,
       11.76830981,  8.68832398,  5.38342077,  5.86519525,  5.82345867,
       12.22503754,  9.3349858 ,  5.1224079 ,  9.78286733,  5.70591986,
       10.794755  ,  9.37989631,  4.98989906,  6.19633139,  5.5671389 ,
        7.46053281])}
{'P': tensor([[1.3953e-02, 1.9054e-01, 1.6410e-01,  ..., 1.1835e-04, 7.5071e-06,
         4.6513e-03],
        [5.9575e-04, 1.9192e-02, 2.5229e-01,  ..., 2.2651e-04, 9.0578e-06,
         8.5386e-03],
        [3.9705e-05, 6.0561e-05, 8.8488e-03,  ..., 5.5596e-05, 1.6465e-06,
         8.1307e-04],
        ...,
        [1.9347e-05, 2.9907e-05, 5.8166e-05,  ..., 7.3430e-02, 1.4296e-05,
         5.8077e-04],
        [3.8175e-03, 4.4732e-03, 1.6835e-03,  ..., 3.1391e-03, 1.0702e-01,
         1.0377e-02],
        [2.3720e-04, 3.9862e-04, 1.9192e-03,  ..., 8.9741e-03, 1.8360e-05,
         5.2984e-02]]), 'X': tensor([[ -9.4117,   0.1160,   8.2574],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.1231,  -3.5739,   8.4708],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.4541,  -0.3327,   8.8246],
        [ 11.2517,  -4.6495,   2.9800],
        [ -9.4203,  -2.0340,   2.5856],
        [  1.1994,   7.6035,   2.4295],
        [ 12.2218,   4.0953,  -8.2072],
        [ -8.7078,   2.7182,   1.8146],
        [ -4.0409,   7.3169,  -3.5779],
        [ -3.5173,   7.5091,  -8.9585],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.8450,  -9.9356,   3.0767],
        [  6.3515,   0.6418,  -6.5823],
        [  9.2811,  -7.4336,   5.1771],
        [  4.4915,   5.5990,   7.0405],
        [  5.5240,  -2.1198,   6.2055],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.3569,   0.7591,   6.5011],
        [  1.4489,   7.5256,   4.4529],
        [ 11.5966,   5.9666,   6.0083],
        [  0.1497,   7.8334,  -5.4412],
        [ 10.0619,   8.6627,  -9.2884],
        [  4.6063,  13.0051,  -3.1456],
        [ -5.9940,   6.1985,   5.9566],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.4867,  12.5816,   2.9549],
        [ -8.1426,   3.8048,   9.3542],
        [  9.1820,   9.7334,  -2.7932],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.7295,  13.6703,  -8.9328],
        [ -1.6086,   7.8928,   4.3385],
        [ -7.9667,   4.0814,  -0.6397],
        [  6.0310,  12.2795,  -6.4528],
        [ -7.6525,  -6.6534,   5.1721],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.9284,  11.7061,  -0.3732],
        [ 11.6165,  -3.8715,   6.4594],
        [ -7.6719,  -6.6260,   7.1468],
        [  0.1287, -11.0065,   6.5757],
        [  3.3069,   6.6089,   8.7377],
        [  6.8704,  -9.3809,   7.5769],
        [  5.2939,  -2.4784,  -4.5840],
        [  1.7122,  -4.7591,  -3.6734],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.9461,  -4.4077,  -0.4831],
        [  1.5383, -11.0274,  -4.9666],
        [  4.3523, -10.5232,   8.2138],
        [ -1.6275, -10.7221,   3.2376],
        [  3.2701,   6.6338,   9.9209],
        [ -9.1231,  -3.5739,   8.4708],
        [  1.9505,  -4.7171,   9.9868],
        [ 10.0400,   8.6922,  -6.2847],
        [  4.8051, -10.3703,   0.5268],
        [ -6.8618,  -7.6322,   3.7027],
        [  8.9284,  10.0036,  -6.9210],
        [  6.1737,   2.5305,   3.9049],
        [  4.6231,   5.4556,   8.4973],
        [  5.4426, -10.1181,  -5.1268],
        [  5.0506,   4.9277,  -9.5094],
        [  2.1625,  -4.6697,   7.0658],
        [  5.3804,  -2.3504,   7.2744],
        [  2.3373,  13.7721,  -0.2442],
        [ -1.7009,   7.8851,  -8.2437],
        [  4.6231,   5.4556,   8.4973],
        [  3.2701,   6.6338,   9.9209],
        [  2.5375,   7.0676,  -4.3490],
        [  0.3493,  -4.7727,  -6.5682],
        [  8.4691,  10.4571,  -5.0650],
        [  5.1292,   4.8183,  -5.7997],
        [ -3.5173, -10.0623,   4.8280],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.5424,  -2.0885,   7.2487],
        [ -0.1963,   7.8765,  -1.1371],
        [  6.7565,  11.8237,  -4.9882],
        [ 12.4837,   2.7510,   0.4449],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.6019,   5.9540,  -5.1715],
        [  8.2154,  -8.4262,   8.0171],
        [  9.1409,  -7.5789,   1.5814],
        [  0.6952,  -4.8060,   8.8976],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -4.7847,  -9.3729,  -3.7741],
        [  4.5800,   5.5034,  -3.9432],
        [  3.5965,  -4.0754,  -3.5035],
        [ 12.4597,  -0.9136,  -4.8763],
        [  1.7000,  -4.7609,   2.7935],
        [  1.1274,  14.0116,   8.1753],
        [  9.0004,  -7.7195,   4.6945],
        [ -6.3489,   5.9117,   6.5161],
        [  7.7794,  11.0629,   8.9069],
        [ -1.9487,   7.8590,  -3.3330],
        [  2.1379,  -4.6757,   6.4981],
        [ -5.5525,   6.5146,  -4.4984],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -9.4203,  -2.0340,   2.5856],
        [  1.8721,  -4.7322,   0.5524],
        [  2.2659,  13.7894,   6.5436],
        [ -5.2589,   6.7024,  -5.5809],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -4.9217,   6.8979,  -1.7667],
        [ -8.9606,  -4.0994,   1.7886]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 3.5175e-02, 7.1470e-02,  ..., 9.8455e-01, 8.0035e-01,
         4.2024e-01],
        [3.5175e-02, 0.0000e+00, 6.6987e-02,  ..., 9.5360e-01, 7.8719e-01,
         3.8769e-01],
        [7.1470e-02, 6.6987e-02, 0.0000e+00,  ..., 9.9479e-01, 8.5253e-01,
         4.2584e-01],
        ...,
        [9.8455e-01, 9.5360e-01, 9.9479e-01,  ..., 0.0000e+00, 6.2052e-01,
         5.6907e-01],
        [8.0035e-01, 7.8719e-01, 8.5253e-01,  ..., 6.2052e-01, 0.0000e+00,
         5.8820e-01],
        [4.2024e-01, 3.8769e-01, 4.2584e-01,  ..., 5.6907e-01, 5.8820e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.41244865,  9.49511524,  9.77588801,  9.79814997,  9.64157461,
        9.45995955, 12.17451685,  9.63742669,  7.69752262, 12.889695  ,
        9.12220271,  8.35856487,  8.29203356,  7.83486943, 12.53047489,
        6.1561646 , 11.52733581,  6.383897  , 11.89105259,  7.17789666,
        5.91677188, 12.26997165,  6.4020404 ,  7.66377554, 13.04156075,
        7.83486943, 13.27717982, 13.79676654,  8.62260327, 13.09426142,
       13.72594022,  8.98765644, 13.38091121,  9.85330646, 13.94008948,
        8.0550286 ,  8.95133961, 13.68063098, 10.14044489, 13.27994304,
       13.60275211, 12.24467041, 10.1371552 , 11.00726483,  7.39004226,
       11.62770332,  5.84533345,  5.05774295, 10.41909211,  5.30158497,
       11.1341798 , 11.38774149, 10.84493732,  7.39599986,  9.79814997,
        5.10446653, 13.27994304, 11.42947632, 10.26327895, 13.40850328,
        6.67218687,  7.15100303, 11.4890861 ,  7.05627461,  5.14607014,
        5.87132718, 13.96905756,  8.06643487,  7.15100303,  7.39599986,
        7.5092834 ,  4.78544875, 13.45642379,  7.03733775, 10.65929968,
       10.61472636,  5.92281142,  7.8788965 , 13.61801526, 12.78327037,
       12.32408516, 13.04051744, 11.76830981, 11.87412439,  4.85603886,
       12.35421461, 10.52357479,  7.15990676,  5.43544626, 12.49317563,
        5.05534609, 14.05687815, 11.85743425,  8.67502348, 13.52430176,
        8.09704052,  5.14124612,  8.55982105, 10.53368644,  9.63742669,
        5.08910491, 13.97430191,  8.51928531, 10.14436912,  8.47371868,
        9.85384623])}
{'P': tensor([[1.0477e-02, 3.3762e-01, 1.7278e-01,  ..., 4.4971e-06, 4.5117e-06,
         1.2451e-04],
        [2.1631e-04, 1.4968e-02, 3.3202e-01,  ..., 5.5876e-06, 6.3245e-06,
         2.2367e-04],
        [1.0727e-04, 1.6128e-04, 1.7297e-02,  ..., 4.9695e-06, 7.6720e-06,
         3.5955e-04],
        ...,
        [5.2193e-03, 3.6946e-03, 1.7275e-03,  ..., 9.0230e-02, 5.2116e-04,
         1.9991e-04],
        [6.0551e-04, 4.4705e-04, 2.6083e-04,  ..., 7.2365e-02, 1.0604e-01,
         2.2130e-04],
        [1.3175e-04, 1.3291e-04, 1.2997e-04,  ..., 1.4659e-03, 4.2946e-03,
         1.5094e-01]]), 'X': tensor([[ -9.4157,   0.0821,  -5.0025],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -3.2486,   7.5925,  -1.3719],
        [ 10.5057,  -5.9099,  -5.1599],
        [  0.9392,  -4.8143,  -3.4649],
        [ 11.9808,   4.9340,  -3.3715],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.3535,   0.6792,  -7.5101],
        [ 11.6165,  -3.8715,   6.4594],
        [ 10.5955,   7.8824,  -3.7006],
        [ 12.0772,  -2.6014,  -7.1832],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.5650,   2.0132,   4.6272],
        [  4.8432,   5.1968,   1.5006],
        [  3.4604,  -4.1545,   1.8091],
        [  5.4336,  -2.2676,   5.6933],
        [  5.2608,  -2.5256,  -9.1446],
        [ -7.6734,   4.4980,  -5.4248],
        [ -2.2188,   7.8218,   8.6064],
        [  5.6093,  -1.9705,   6.4807],
        [  2.0899,  -4.6870,  -7.4260],
        [ -3.3999, -10.1152,  -1.2088],
        [  4.9152,  12.8653,  -1.4459],
        [  5.9548,  -1.2273,  -2.6520],
        [  9.3753,   9.5171,   3.9131],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.2555,   2.1474,   4.1685],
        [ -8.3900,   3.3727,   3.1650],
        [  9.3753,   9.5171,   3.9131],
        [  2.0899,  -4.6870,  -7.4260],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.1824,  -4.6647,   1.1985],
        [ 11.5456,   6.0861,   8.1101],
        [ -8.7078,   2.7182,   1.8146],
        [  6.2630,   2.1057,  -5.2136],
        [ -1.7009,   7.8851,  -8.2437],
        [ -1.1885, -10.8211,   0.0865],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.3739,   0.3982,  -4.2564],
        [ -9.4196,   0.0478,   9.4140],
        [  5.2199,   4.6861,  -1.9304],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.3604,   0.4855,  -3.0158],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.8275,  -1.5334,   2.0124],
        [ -8.5506,  -5.1154,   6.3423],
        [  9.6662,   9.1725,   8.1056],
        [  5.8902,  -1.3887,   4.0388],
        [  5.8930,  12.3592,  -8.9201],
        [  1.3339,   7.5627,  -0.9034],
        [  2.6321,  -4.5296,  -2.5159],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 10.9394,   7.3069,  -8.9365],
        [  0.8940, -11.0402,  -0.9783],
        [  3.8111, -10.6787,  -4.6539],
        [  5.9526,   3.2525,  -4.3821],
        [  3.7121,   6.3117,   0.6738],
        [  6.3587,   1.1596,  -3.6629],
        [ -5.9225,  -8.5355,   2.4778],
        [  2.1477,   7.2550,   8.9688],
        [  2.3889,  -4.6082,  -8.7291],
        [ -7.7842,   4.3466,  -1.7548],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.0845,  -2.5772,  -4.6073],
        [  6.1755,   2.5233,  -2.0042],
        [  9.8082,  -6.8396,   4.9615],
        [  6.8407,  11.7666,   3.0045],
        [  6.9284,  11.7061,  -0.3732],
        [ -9.3087,   0.7792,  -7.5587],
        [  5.3924, -10.1396,  -4.0863],
        [  3.5284, -10.7484,   5.3347],
        [  6.3200,   0.2690,   1.5900],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.8318,   3.5567,  -4.7692],
        [  6.2790,  -0.0259,  -5.4116],
        [  9.3560,   9.5391,   9.4943],
        [ -8.3062,  -5.6008,   0.0287],
        [ -7.5720,   4.6310,   2.1717],
        [  4.1348,   5.9514,  -0.8856],
        [  0.1287, -11.0065,   6.5757],
        [  7.3724,  -9.0577,  -8.9843],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -4.8665,  -9.3204,  -5.2564],
        [  8.8474,  10.0868,  -8.8644],
        [ 10.1503,   8.5414,  -7.2681],
        [  5.5101,  -2.1433,   9.5442],
        [ -0.1963,   7.8765,  -1.1371],
        [  6.0345,   3.0175,  -1.4424],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.0560,  -0.9421,  -5.0334],
        [  1.0093,   7.6567,   2.3526],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.7078,   2.7182,   1.8146],
        [  5.6260,  -1.9401,   3.0259],
        [  5.6414, -10.0303,   7.8525],
        [ -6.4002,  -8.1048,  -0.9108],
        [  0.3724,   7.7974,  -6.8722],
        [  5.0233,   4.9647,  -3.1788],
        [  2.6614,  -4.5192,  -9.3641]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0369, 0.0752,  ..., 0.6873, 0.8879, 0.8797],
        [0.0369, 0.0000, 0.0395,  ..., 0.7220, 0.9245, 0.9144],
        [0.0752, 0.0395, 0.0000,  ..., 0.7615, 0.9630, 0.9539],
        ...,
        [0.6873, 0.7220, 0.7615,  ..., 0.0000, 0.2685, 0.1924],
        [0.8879, 0.9245, 0.9630,  ..., 0.2685, 0.0000, 0.2007],
        [0.8797, 0.9144, 0.9539,  ..., 0.1924, 0.2007, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.41605684,  9.50913849,  9.69792513,  9.50654967,  9.78286733,
        9.7712321 ,  9.62958973,  8.25828409, 12.05395056,  4.90505566,
       12.95702647, 12.52875793,  6.38968357, 12.24467041, 13.20597932,
       12.35421461, 13.99048094, 12.72524615,  7.10378919,  5.40689332,
        5.88782913,  5.83560009,  8.89458273,  8.1303945 ,  5.94535529,
        5.13182506, 10.67132524, 13.77222752,  6.07992623, 13.35927527,
       12.26997165, 13.13205357,  6.61386436,  9.04255655, 13.35927527,
        5.13182506, 10.0674887 ,  5.14999799, 13.05148826,  9.12220271,
        6.60752373,  8.06643487, 10.88618114,  8.29203356,  9.38232121,
        9.41970031,  7.01475376,  5.5748624 ,  9.37296102, 10.48840476,
        6.02588517,  9.96391045, 13.32557133,  6.05165248, 13.6922425 ,
        7.67939546,  5.23878371,  9.58072286, 13.15525803, 11.07637531,
       11.33837396,  6.78325151,  7.32233733,  6.46356806, 10.38897563,
        7.56617037,  5.19065439,  8.91551231, 13.04051744, 12.35625694,
        6.67108662, 11.95743558, 13.6105595 , 13.60275211,  9.34127013,
       11.48434767, 11.31276109,  6.32572323,  5.18277858,  6.83083863,
        6.27905292, 13.3614598 , 10.01804223,  8.87587457,  7.24677664,
       11.00726483, 11.67875855, 10.23946247, 10.51437947, 13.41713661,
       13.26590071,  5.91222346,  7.8788965 ,  6.74686071, 12.67576792,
        6.12886024,  7.72291946, 12.79126785, 13.6660671 ,  9.12220271,
        5.95111804, 11.50791859, 10.32715715,  7.80625981,  7.06272256,
        5.24461068])}
{'P': tensor([[1.6932e-02, 3.5083e-01, 1.7553e-01,  ..., 2.9516e-07, 1.6750e-07,
         5.9564e-06],
        [1.8534e-04, 2.3223e-02, 2.3190e-01,  ..., 2.8779e-07, 2.5127e-07,
         1.2927e-05],
        [7.5509e-05, 3.5155e-04, 1.5285e-02,  ..., 1.8516e-07, 2.3608e-07,
         1.8526e-05],
        ...,
        [1.2975e-03, 4.6826e-04, 3.0978e-04,  ..., 2.8880e-01, 1.8720e-05,
         1.3829e-06],
        [2.1593e-06, 1.0095e-06, 8.5728e-07,  ..., 1.6131e-02, 2.3238e-02,
         2.4335e-07],
        [3.7273e-07, 4.3190e-07, 6.9205e-07,  ..., 1.6618e-06, 1.3820e-04,
         4.3499e-02]]), 'X': tensor([[ -9.4196,   0.0478,   9.4140],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.9314,  -9.3435,  -1.2729],
        [  4.8051, -10.3703,   0.5268],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -2.5683, -10.4433,  -2.9182],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.7794,   5.5086,  -9.6929],
        [  6.0345,   3.0175,  -1.4424],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.5636, -10.9293,   9.6519],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 11.5568,   6.0602,   9.3244],
        [ -9.2345,  -3.1345,   3.9715],
        [  2.0937,  13.8293,  -2.3533],
        [ -7.4887,   4.7364,   8.9074],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -8.4297,  -5.3637,  -9.3962],
        [  0.5960, -11.0335,   3.1144],
        [ -8.4297,  -5.3637,  -9.3962],
        [  9.5261,  -7.1675,   1.4206],
        [  4.0457,  -3.7737,  -0.4704],
        [  0.1532,  -4.7422,  -7.0145],
        [  0.3493,  -4.7727,  -6.5682],
        [ -3.5029,   7.5138,   8.1853],
        [ 12.6059,   0.9597,  -6.3264],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.4887,  -5.2448,  -0.6585],
        [  1.7232,  -4.7574,   3.1270],
        [  4.5954,  13.0099,  -4.4439],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.7485,  -9.9811,  -3.2355],
        [ -9.3419,   0.5972,  -1.2841],
        [ -9.4196,   0.0478,   9.4140],
        [  2.1625,  -4.6697,   7.0658],
        [  2.4149,  -4.6005,  -4.3339],
        [  2.4697,  -4.5836,   1.4593],
        [  2.5375,   7.0676,  -4.3490],
        [  6.1984,  -0.4327,   1.5438],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -2.5056,   7.7720,   5.5878],
        [  7.0664,  -9.2590,   8.0879],
        [  5.5240,  -2.1198,   6.2055],
        [ -3.3999, -10.1152,  -1.2088],
        [ 11.3691,   6.4760,   1.0787],
        [  1.2728, -11.0372,   9.8121],
        [ -9.3087,   0.7792,  -7.5587],
        [  5.3804,  -2.3504,   7.2744],
        [  4.8051, -10.3703,   0.5268],
        [  8.1528,  10.7452,   9.1493],
        [ -0.3432,   7.8900,   7.1805],
        [ -7.4887,   4.7364,   8.9074],
        [  6.9284,  11.7061,  -0.3732],
        [  2.2373,  -4.6506,  -1.7334],
        [  4.4049,   5.6892,  -4.4394],
        [ -2.5683, -10.4433,  -2.9182],
        [  6.3122,  -9.6994,  -6.3282],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.1973,  -0.4372,   6.5720],
        [ -6.7643,  -7.7372,  -9.6663],
        [  4.1637,  13.1898,  -1.9189],
        [  1.7301,   7.4262,   5.0522],
        [  4.4049,   5.6892,  -4.4394],
        [  6.9284,  11.7061,  -0.3732],
        [  1.7903,  -4.7467,   0.1168],
        [  6.3542,   0.6934,  -3.3840],
        [  5.4242, -10.1261,   3.9595],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.0310,  12.2795,  -6.4528],
        [ -2.0840,   7.8415,  -7.8657],
        [ 12.2592,   3.9425,   2.0383],
        [ -6.3489,   5.9117,   6.5161],
        [ -3.0008, -10.2827,  -8.9480],
        [  4.5491,  -3.3485,   6.8412],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -4.4779,  -9.5609,  -2.4505],
        [  6.3200,   0.2690,   1.5900],
        [  0.1047, -11.0046,   0.5188],
        [  0.3038,   7.8092,   0.4164],
        [  9.2811,  -7.4336,   5.1771],
        [  0.7145,  -4.8071,  -2.3766],
        [ -9.4296,  -1.9458,   4.0833],
        [  4.1537,   5.9340,  -6.0423],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.5128, -10.9365,   9.9645],
        [  1.1994,   7.6035,   2.4295],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.4157,   0.0821,  -5.0025],
        [  0.3354,  -4.7708,  -4.9125],
        [ -3.4986,   7.5152,   9.0712],
        [  4.5503,   5.5359,   7.4735],
        [  6.1984,  -0.4327,   1.5438],
        [  4.9152,  12.8653,  -1.4459],
        [  6.2630,   2.1057,  -5.2136],
        [ -1.9714,   7.8563,  -1.8084],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.9470,  13.6085,   2.9809],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.4639, -10.9431,  -7.3718]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0489, 0.0574,  ..., 1.1317, 1.3583, 1.1412],
        [0.0489, 0.0000, 0.0208,  ..., 1.1803, 1.4044, 1.1678],
        [0.0574, 0.0208, 0.0000,  ..., 1.1874, 1.4022, 1.1538],
        ...,
        [1.1317, 1.1803, 1.1874,  ..., 0.0000, 0.6155, 1.2125],
        [1.3583, 1.4044, 1.4022,  ..., 0.6155, 0.0000, 0.8352],
        [1.1412, 1.1678, 1.1538,  ..., 1.2125, 0.8352, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.41970031,  9.64157461,  9.77588801,  9.79814997,  9.87720102,
        9.45995955, 11.63383221, 11.42947632, 10.61047596, 10.75443106,
       12.04924159, 13.00380624,  6.74686071, 11.75807252, 11.22596994,
        9.75401449, 13.0493368 ,  9.75200848, 13.98691426,  8.86082124,
        9.69629   ,  9.99145096, 11.04953767,  9.99145096, 11.92133129,
        5.53255882,  4.74468406,  4.78544875,  8.29022197, 12.64235359,
       13.103709  ,  9.97822473,  5.05989547, 13.79762415,  5.24137909,
       11.51811603,  9.36093562,  9.41970031,  5.14607014,  5.19577146,
        5.20659892,  7.5092834 ,  6.2134961 , 12.12448912,  8.16584537,
       11.64746882,  5.91677188, 10.67132524, 13.08414689, 11.11038551,
        9.34127013,  5.87132718, 11.42947632, 13.48809595,  7.89745107,
        8.86082124, 13.60275211,  5.16077764,  7.19513806, 10.75443106,
       11.57249984,  7.8575648 ,  6.21275016, 10.27716434, 13.83139024,
        7.62509582,  7.19513806, 13.60275211,  5.07306274,  6.39187285,
       11.48734978,  9.75200848, 13.68063098,  8.1137452 , 12.87752052,
        8.67502348, 10.71162681,  5.64865173,  9.69792513,  9.99145096,
       10.5575601 ,  6.32572323, 11.00508681,  7.81509631, 11.89105259,
        4.85995002,  9.6282763 ,  7.24326992,  5.95297459, 11.22142195,
        7.69752262,  5.21280156,  5.16611782,  9.34127013,  9.41605684,
        4.78258092,  8.28968067,  7.16598273,  6.2134961 , 13.77222752,
        6.60752373,  8.09984231,  9.53015376, 13.92390689, 13.12299145,
       11.21703654])}
{'P': tensor([[9.7770e-03, 2.4440e-01, 1.4830e-01,  ..., 6.2995e-06, 3.3536e-03,
         1.1415e-05],
        [1.8108e-04, 1.3181e-02, 1.4596e-01,  ..., 9.5007e-06, 8.4201e-03,
         2.6037e-05],
        [1.2782e-04, 3.7455e-04, 1.7974e-02,  ..., 1.3632e-05, 1.8293e-02,
         4.2186e-05],
        ...,
        [1.8315e-04, 1.3808e-04, 9.1810e-05,  ..., 8.8540e-02, 2.8655e-05,
         2.1354e-05],
        [1.9307e-05, 2.5607e-05, 3.6568e-05,  ..., 2.4798e-04, 6.0437e-02,
         5.2978e-03],
        [8.5975e-07, 9.4079e-07, 8.5193e-07,  ..., 6.9172e-04, 6.4096e-06,
         3.6332e-02]]), 'X': tensor([[ -9.4309,  -0.0599,  -4.0830],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -3.5029,   7.5138,   8.1853],
        [ 12.1588,   4.3364,  -8.0867],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -6.5295,  -7.9785,  -7.4429],
        [  8.7131,  10.2216,   7.7915],
        [  5.8323,  -1.5228,   5.1132],
        [ 12.1588,   4.3364,  -8.0867],
        [ -7.3333,   4.9246,  -7.9212],
        [  5.6314,  -1.9302,  -9.6543],
        [  8.7715,  -7.9390,   5.2900],
        [ 12.4691,   2.8525,  -5.5110],
        [  3.5939,   6.4029,  -1.8357],
        [  0.1532,  -4.7422,  -7.0145],
        [ -9.4117,   0.1160,   8.2574],
        [  5.0233,   4.9647,  -3.1788],
        [  1.9204,   7.3518,   8.4781],
        [  6.3551,   0.7149,   1.3344],
        [  2.1625,  -4.6697,   7.0658],
        [ -0.6323,   7.9085,  -5.9544],
        [ -9.4309,  -0.0599,  -4.0830],
        [  2.1477,   7.2550,   8.9688],
        [ -4.2040,   7.2486,   2.9875],
        [  4.1140,  -3.7220,  -0.1273],
        [  5.7963,  -1.6018,   1.8856],
        [  9.6816,   9.1536,  -5.8543],
        [ -9.4157,   0.0821,  -5.0025],
        [  4.5340,  -3.3629,   8.9652],
        [  5.9883,   3.1536,   6.2320],
        [ -1.6220,   7.8917,   0.0315],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.5294,  -2.1106,   9.0626],
        [  5.1142,  -2.7230,  -4.1966],
        [  4.9337,  12.8566,  -3.6582],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.1575,  12.2046,   3.3257],
        [ -4.8751,  -9.3148,   4.0406],
        [  0.6470, -11.0352,  -1.1931],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -9.4716,  -0.6673,   7.6068],
        [  5.6314,  -1.9302,  -9.6543],
        [ -1.6906,   7.8860,   7.7142],
        [  4.1140,  -3.7220,  -0.1273],
        [  0.8940, -11.0402,  -0.9783],
        [  2.5813,  -4.5471,   5.3597],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.5663,   7.0526,  -6.8769],
        [ 11.6801,  -3.7196,  -0.9181],
        [  6.2905,   1.9366,   2.1169],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.2701,   6.6338,   9.9209],
        [ -7.7842,   4.3466,  -1.7548],
        [ -3.0008, -10.2827,  -8.9480],
        [ 11.5935,   5.9740,  -4.3952],
        [  9.4925,   9.3811,   8.9842],
        [  3.6249, -10.7255,   5.6605],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -1.6220,   7.8917,   0.0315],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.6248,   3.9984,   8.4933],
        [ -8.3717,  -5.4769,  -9.2228],
        [  5.7784,  -1.6401,   9.5587],
        [ 11.5966,   5.9666,   6.0083],
        [  5.2939,  -2.4784,  -4.5840],
        [  2.6613,   7.0021,   4.5783],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.1497,   7.8334,  -5.4412],
        [  0.1287, -11.0065,   6.5757],
        [  3.7298,  13.3537,  -9.4653],
        [ 12.5763,   0.1333,  -9.2650],
        [  5.4732,  -2.2041,  -4.3830],
        [ -5.3382,   6.6533,   7.8054],
        [  3.6469,   6.3624,  -1.8414],
        [ 12.2218,   4.0953,  -8.2072],
        [  3.9770, -10.6341,   8.0783],
        [ -2.0840,   7.8415,  -7.8657],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.5996,   1.3971,  -1.0694],
        [  6.3031,   1.8477,  -7.1706],
        [  1.3705,  13.9723,   7.9266],
        [ -3.5029,   7.5138,   8.1853],
        [  9.6614,   9.1784,  -1.2087],
        [  1.1994,   7.6035,   2.4295],
        [  3.6154,  -4.0640,  -7.9698],
        [  1.1644, -11.0394,   7.8511],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.0348,  -2.8229,   3.5436],
        [  0.9036,   7.6840,   9.9237],
        [  6.3211,   1.7017,  -0.2164],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.6095,  -1.9702,  -7.8233],
        [ -8.9606,  -4.0994,   1.7886],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3534,   1.2990,  -7.2942],
        [ -1.6220,   7.8917,   0.0315],
        [  5.3804,  -2.3504,   7.2744],
        [  6.2036,   2.4030,  -1.2500],
        [ -4.4779,  -9.5609,  -2.4505],
        [  5.4167, -10.1293,   6.2156]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0367, 0.0930,  ..., 0.9466, 0.1167, 0.9945],
        [0.0367, 0.0000, 0.0875,  ..., 0.9742, 0.1214, 0.9951],
        [0.0930, 0.0875, 0.0000,  ..., 1.0297, 0.2060, 1.0823],
        ...,
        [0.9466, 0.9742, 1.0297,  ..., 0.0000, 0.8755, 0.7931],
        [0.1167, 0.1214, 0.2060,  ..., 0.8755, 0.0000, 0.8778],
        [0.9945, 0.9951, 1.0823,  ..., 0.7931, 0.8778, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.43112957,  9.55720651,  9.69792513,  9.71392179,  9.5642993 ,
        9.78286733,  8.29022197, 12.9089559 ,  9.58072286, 10.30972193,
       13.43127131,  6.02779201, 12.9089559 ,  8.83341608,  5.95297459,
       11.83075005, 12.79126785,  7.34250948,  4.74468406,  9.41244865,
        7.06272256,  7.5984809 ,  6.39520397,  5.14607014,  7.9337647 ,
        9.43112957,  7.56617037,  8.37954853,  5.54776934,  6.01356004,
       13.32374056,  9.41605684,  5.64500512,  6.76791483,  8.05668609,
       12.27993098,  5.91854397,  5.79391825, 13.77075174, 11.99633819,
       13.66991027, 10.51340829, 11.05413545, 10.06437598,  9.49511524,
        5.95297459,  8.06517143,  5.54776934, 11.07637531,  5.2287058 ,
        7.8788965 ,  7.50500064, 12.25806559,  6.58184474,  5.53255882,
        7.39599986,  8.91551231, 10.71162681, 13.04217487, 13.34587027,
       11.32149259, 12.27835996,  8.05668609,  5.0209005 ,  6.90115901,
       10.00410625,  6.00663314, 13.04156075,  5.84533345,  7.49076899,
        6.14086582,  7.83486943, 11.00726483, 13.86480175, 12.5769659 ,
        5.90035324,  8.53014915,  7.33350642, 12.889695  , 11.35345345,
        8.1137452 , 10.37998078,  9.5014146 , 11.52733581, 12.67680439,
        6.56833864, 14.0393949 ,  8.29022197, 13.32613445,  7.69752262,
        5.43943791, 11.10066126,  9.79039371,  5.77219077,  7.73692436,
        6.54615887,  9.48969978,  5.94541803,  9.85384623,  5.5366698 ,
        6.48486171,  8.05668609,  5.87132718,  6.652742  , 10.5575601 ,
       11.48663767])}
{'P': tensor([[6.3415e-03, 2.5728e-01, 1.6004e-01,  ..., 4.1681e-06, 8.7286e-06,
         3.4928e-05],
        [1.5606e-04, 1.0575e-02, 3.4876e-01,  ..., 8.5144e-06, 1.1508e-05,
         4.1938e-05],
        [3.7532e-04, 4.7824e-04, 4.1118e-02,  ..., 4.9416e-05, 3.0043e-05,
         9.4060e-05],
        ...,
        [9.9642e-06, 9.2255e-06, 8.2360e-06,  ..., 2.0713e-01, 8.9644e-06,
         1.7904e-05],
        [4.4846e-04, 4.7193e-04, 5.1820e-04,  ..., 1.6749e-06, 2.4165e-02,
         2.9143e-03],
        [1.8962e-02, 1.9785e-02, 2.0430e-02,  ..., 9.4738e-06, 4.5959e-03,
         5.7143e-02]]), 'X': tensor([[ -9.4366,  -0.1189,  -2.6736],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -8.9788,   2.0208,  -3.2867],
        [  3.6154,  -4.0640,  -7.9698],
        [ -9.2373,  -3.1225,  -8.9763],
        [  4.0642,  -3.7599,  -3.4565],
        [  9.5261,  -7.1675,   1.4206],
        [  7.0642,  11.6104,  -9.8316],
        [  9.8082,  -6.8396,   4.9615],
        [  6.2895,   0.0405,   5.7652],
        [ -9.3464,   0.5708,  -1.6397],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4721,  -1.3097,  -7.4377],
        [  9.0129,  -7.7072,   9.1400],
        [ -9.4541,  -0.3327,   8.8246],
        [ -7.8849,  -6.3115,  -4.7479],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 12.3609,   3.4769,  -8.5762],
        [  9.2979,   9.6048,   7.3138],
        [ -4.7334,   6.9982,  -0.4392],
        [  8.2422,  -8.4041,  -1.4962],
        [ -0.8885,   7.9160,   0.0571],
        [  6.8407,  11.7666,   3.0045],
        [ -1.3178, -10.7939,   0.4236],
        [  9.9217,  -6.7006,  -5.6728],
        [  7.1441,  11.5529,   2.3232],
        [  2.8602,   6.8905,   5.8779],
        [  0.6363, -11.0348,   6.1414],
        [  2.3376,  -4.6231,   9.8436],
        [  7.9141,  -8.6656,   9.6948],
        [  2.4639, -10.9431,  -7.3718],
        [ -0.0410,   7.8591,   9.3457],
        [ -4.0868,   7.2981,   8.1874],
        [  6.2902,   0.0451,   3.7116],
        [  6.3535,   0.6792,  -7.5101],
        [  9.4554,   9.4245,   9.8800],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.1588,   4.3364,  -8.0867],
        [  8.6420,  -8.0580,   6.5781],
        [  6.3305,   1.6119,   1.1538],
        [  5.7963,  -1.6018,   1.8856],
        [  2.3261,  13.7749,   5.9121],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.5653,  13.9377,   4.1773],
        [ 12.5763,   0.1333,  -9.2650],
        [  0.1041, -11.0045,   2.4311],
        [ 11.0864,   7.0387,  -0.1238],
        [ -9.3705,   0.4208,  -3.5815],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.8201,   3.5841,  -8.0609],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.3569,   0.7591,   6.5011],
        [  6.2902,   0.0451,   3.7116],
        [  2.1771, -10.9776,  -3.2007],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.2883,   0.0325,   5.2692],
        [  2.0870,  -4.6877,   9.6823],
        [  2.7295,  13.6703,  -8.9328],
        [  9.5261,  -7.1675,   1.4206],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.2804,   2.0019,   0.8607],
        [  7.1697,  11.5343,  -0.7276],
        [  2.5128, -10.9365,   9.9645],
        [  5.6093,  -1.9705,   6.4807],
        [ -8.9178,   2.1936,   7.5388],
        [  5.2188,  -2.5839,   2.1833],
        [ 11.1469,   6.9237,   1.6160],
        [  0.2571,  -4.7594,   1.5618],
        [  1.1274,  14.0116,   8.1753],
        [  2.2659,  13.7894,   6.5436],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.3376,  -4.6231,   9.8436],
        [  1.1006,   7.6318,  -2.4877],
        [  1.2728, -11.0372,   9.8121],
        [  1.6408,   7.4591,  -4.2294],
        [  2.1771, -10.9776,  -3.2007],
        [  6.1989,  -9.7591,   0.8289],
        [  5.7218,  -9.9934,   6.5764],
        [  5.4867,  12.5816,   2.9549],
        [  6.5428,  -9.5728,  -2.9897],
        [  3.7121,   6.3117,   0.6738],
        [ -9.0574,   1.7801,   8.8986],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.1706,   1.3863,   8.4032],
        [ 10.5955,   7.8824,  -3.7006],
        [ -8.8154,   2.4611,   2.5198],
        [  7.0515,  11.6194,   8.4289],
        [ -2.3317,   7.8034,  -3.6274],
        [  6.1685,  12.1979,  -6.2473],
        [ -0.9715, -10.8630,   0.3555],
        [ -6.4620,  -8.0449,   6.2719],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.3303,   1.6145,  -8.4083],
        [  5.1472,  -2.6801,   8.1194],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.1666,   3.7653,   3.6075]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0238, 0.0536,  ..., 0.7450, 0.8412, 0.5379],
        [0.0238, 0.0000, 0.0300,  ..., 0.7688, 0.8523, 0.5522],
        [0.0536, 0.0300, 0.0000,  ..., 0.7986, 0.8623, 0.5669],
        ...,
        [0.7450, 0.7688, 0.7986,  ..., 0.0000, 0.8721, 0.6447],
        [0.8412, 0.8523, 0.8623,  ..., 0.8721, 0.0000, 0.3201],
        [0.5379, 0.5522, 0.5669,  ..., 0.6447, 0.3201, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.43737523,  9.5014146 ,  9.61942445,  9.5642993 ,  9.54154096,
        9.58072286,  9.20340031,  5.43943791,  9.75075273,  5.5366698 ,
       11.92133129, 13.59056613, 11.95743558,  6.28961988,  9.36378438,
       11.1913598 , 10.0674887 ,  9.562174  , 11.85890664,  9.45995955,
       10.09979747,  6.60752373,  9.55720651, 12.84056604, 13.36800073,
        8.4486771 , 11.77124373,  7.96575782, 13.6105595 , 10.87409005,
       11.97238624, 13.58333857,  7.46053281, 11.05317597,  5.1805345 ,
       11.73567759, 11.21703654,  7.85919687,  8.36445758,  6.29034958,
        6.38968357, 13.35013374,  5.05444288, 12.9089559 , 11.8159254 ,
        6.53250937,  6.01356004, 13.96987806,  9.88571713, 14.02532758,
       12.5769659 , 11.0050379 , 13.13205357,  9.37989631,  5.43544626,
        9.69792513,  8.8353674 ,  6.83515201, 10.39942497,  6.4020404 ,
        6.29034958, 11.1913598 ,  5.90035324,  6.28835101,  5.13125513,
       13.94008948, 11.92133129,  8.47371868,  6.59176018, 13.58101794,
       11.22142195,  5.94535529,  9.18358562,  5.82345867, 13.12218013,
        4.7663526 , 14.05687815, 13.97430191, 10.32715715,  5.1805345 ,
        7.71075226, 11.11038551,  7.6374603 , 11.1913598 , 11.56146873,
       11.51557166, 13.72594022, 11.59512194,  7.32233733,  9.23071698,
        9.9353921 ,  9.27474386, 13.20597932,  9.15253344, 13.59170809,
        8.14433542, 13.66896739, 10.90638066, 10.31886066,  4.98461983,
        5.83560009,  6.53290663,  5.80312949, 11.97949273,  8.95983586,
        8.99276564])}
{'P': tensor([[1.2651e-02, 2.9096e-01, 2.1779e-01,  ..., 1.4230e-07, 1.1448e-04,
         1.3736e-05],
        [6.5181e-04, 3.2402e-02, 2.1054e-01,  ..., 3.6426e-07, 2.7472e-04,
         2.4475e-05],
        [1.7575e-04, 6.5341e-04, 1.5111e-02,  ..., 2.4276e-07, 2.4779e-04,
         9.6719e-06],
        ...,
        [7.6657e-07, 5.4820e-07, 5.4151e-07,  ..., 1.9118e-02, 3.4895e-07,
         6.1412e-05],
        [4.1668e-06, 4.7090e-06, 8.2456e-06,  ..., 1.8116e-05, 3.5423e-02,
         4.1305e-06],
        [9.1879e-03, 6.6512e-03, 9.0424e-03,  ..., 2.7538e-06, 8.3036e-04,
         7.6210e-02]]), 'X': tensor([[ -9.4430,  -0.1897,   4.8383],
        [ -9.4773,  -1.0024,   5.2599],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.4351,  -1.8891,   3.5642],
        [  2.7295,  13.6703,  -8.9328],
        [  3.1271, -10.8343,   2.6442],
        [  2.9461,  -4.4077,  -0.4831],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.4639, -10.9431,  -7.3718],
        [  6.2895,   0.0405,   5.7652],
        [  2.4697,  -4.5836,   1.4593],
        [ -9.3444,   0.5821,   6.9101],
        [ -4.1296,   7.2803,  -2.7194],
        [  3.9085,   6.1512,  -2.2560],
        [ -7.5603,   4.6459,   8.2002],
        [  8.3791,  10.5409,   5.5590],
        [ 11.0438,   7.1180,  -5.6991],
        [  3.9663,   6.1018,  -5.4104],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.2182,  13.8007,   5.1889],
        [  3.3474,  -4.2162,   5.2455],
        [  4.7156,  -3.1831,  -8.0224],
        [  1.6408,   7.4591,  -4.2294],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.5919,   0.4035,  -5.8460],
        [  6.1989,  -9.7591,   0.8289],
        [  2.3490,  -4.6199,  -4.2489],
        [ -0.6323,   7.9085,  -5.9544],
        [  1.0000,   7.6591,  -4.5011],
        [  0.1532,  -4.7422,  -7.0145],
        [ 10.0400,   8.6922,  -6.2847],
        [  4.6992,   5.3690,   3.0935],
        [  6.0305,   3.0296,   0.1375],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.6202,  -1.9508,   0.7471],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.3331,  -2.6465,  -8.2028],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.9282,   2.1648,   8.4894],
        [ -2.3317,   7.8034,  -3.6274],
        [ 12.5199,  -0.4711,   3.0436],
        [  8.1043,  -8.5164,  -7.9290],
        [  9.9750,  -6.6338,  -1.2440],
        [  7.1697,  11.5343,  -0.7276],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.0305,   3.0296,   0.1375],
        [ 12.2914,   3.8037,   6.0978],
        [  9.9058,  -6.7203,   5.6928],
        [ 11.1420,   6.9331,  -2.4779],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.1372,   2.6726,  -6.7885],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -8.7544,   2.6096,   3.3146],
        [ -8.7735,   2.5640,  -6.3313],
        [  8.8091,  -7.9037,   0.3631],
        [  4.5325,  -3.3643,   6.7699],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.1989,  -9.7591,   0.8289],
        [  4.3459,  -3.5327,   0.3667],
        [  0.7145,  -4.8071,  -2.3766],
        [  5.8902,  -1.3887,   4.0388],
        [ 10.2470,   8.4054,  -0.1177],
        [  6.3550,   0.7129,   0.4917],
        [ -2.9497, -10.3028,  -7.3479],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.5974,   7.0363,  -7.7810],
        [  5.7218,  -9.9934,   6.5764],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.6321,  -4.5296,  -2.5159],
        [ -8.3653,   3.4184,   9.3454],
        [ 11.1085,  -4.9204,   0.0736],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.1372,   2.6726,  -6.7885],
        [ 10.8218,  -5.4177,   6.8573],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.2608,  12.6977,   4.9166],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1271, -10.8343,   2.6442],
        [ 10.7566,   7.6212,  -2.3279],
        [ -8.4824,  -5.2575,  -0.0795],
        [  9.6587,   9.1817,  -0.8212],
        [  1.7000,  -4.7609,   2.7935],
        [  6.2247,  -0.3150,  -9.3657],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.1755,   2.5233,  -2.0042],
        [ -8.6229,   2.9068,   2.4429],
        [ -7.8849,  -6.3115,  -4.7479],
        [  9.3753,   9.5171,   3.9131],
        [ 12.0042,  -2.8350,   2.6864],
        [  2.3200,   7.1756,   1.1938],
        [ 11.4495,   6.3026,   6.8257],
        [  9.6587,   9.1817,  -0.8212],
        [ -7.5720,   4.6310,   2.1717],
        [  2.5128, -10.9365,   9.9645],
        [ -7.6650,   4.5092,   0.2520],
        [  5.3804,  -2.3504,   7.2744],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.1588,   4.3364,  -8.0867],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -6.0195,   6.1789,  -2.9496]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 3.8538e-02, 5.4931e-02,  ..., 1.4418e+00, 9.0214e-01,
         6.3470e-01],
        [3.8538e-02, 3.0339e-04, 7.1809e-02,  ..., 1.4787e+00, 9.3265e-01,
         6.7311e-01],
        [5.4931e-02, 7.1809e-02, 0.0000e+00,  ..., 1.4505e+00, 8.6259e-01,
         6.2393e-01],
        ...,
        [1.4418e+00, 1.4787e+00, 1.4505e+00,  ..., 0.0000e+00, 1.1876e+00,
         8.8402e-01],
        [9.0214e-01, 9.3265e-01, 8.6259e-01,  ..., 1.1876e+00, 0.0000e+00,
         5.1197e-01],
        [6.3470e-01, 6.7311e-01, 6.2393e-01,  ..., 8.8402e-01, 5.1197e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.44486609,  9.53015376,  9.6282763 ,  9.68324553,  9.75200848,
        9.62238828, 13.94008948, 11.27656488,  5.30158497,  8.20436764,
       11.21703654,  6.28961988,  5.20659892,  9.36256096,  8.36995953,
        7.28793663,  8.8737556 , 13.46553623, 13.13888841,  7.27759233,
       11.72749923, 13.97779839,  5.38342077,  5.68942572,  7.6374603 ,
       10.74409834,  6.14086582, 12.59840193, 11.56146873,  5.18277858,
        7.9337647 ,  7.72415048,  4.74468406, 13.27994304,  7.13501264,
        6.74872981, 10.36844813,  5.94909709, 13.09737053,  9.70108409,
       11.21703654,  9.18690018,  8.14433542, 12.52875793, 11.75618513,
       11.97949273, 13.58101794, 13.01187484,  6.74872981, 12.86648771,
       11.9702762 , 13.12299145, 12.19805394,  9.7712321 ,  8.89458273,
        6.69389431, 12.32545637,  9.13507321,  9.14044984, 11.83509372,
        5.64464807, 10.61047596, 10.00794468, 11.56146873,  5.60064319,
        4.85995002,  6.05165248, 13.25336001,  6.3949043 , 10.7167285 ,
        8.06990483,  7.50035355, 11.51557166,  9.75075273,  5.23878371,
        9.0368458 , 12.1494023 ,  5.24461068,  6.69389431, 12.10222204,
        8.89458273, 13.744387  ,  5.13182506, 11.27656488, 13.18278606,
        9.97964013, 13.32645552,  5.05534609,  6.23262289, 12.16901615,
        6.67108662,  9.09963894, 10.09979747, 13.35927527, 12.3344536 ,
        7.54127425, 13.06957425, 13.32645552,  8.87587457, 11.22142195,
        8.89301516,  5.87132718, 11.57249984, 12.9089559 , 10.46739073,
        8.62631801])}
{'P': tensor([[1.3261e-02, 2.0772e-01, 1.9124e-01,  ..., 4.6710e-05, 2.6132e-05,
         1.2890e-04],
        [1.4003e-04, 1.1240e-02, 1.1224e-01,  ..., 5.2707e-05, 4.0147e-05,
         3.8345e-05],
        [4.3760e-05, 1.8997e-04, 8.2598e-03,  ..., 5.5157e-05, 5.6557e-05,
         2.2228e-05],
        ...,
        [1.9471e-04, 1.5347e-04, 1.4502e-04,  ..., 2.1510e-01, 1.8841e-03,
         9.5799e-05],
        [3.9638e-07, 3.5810e-07, 3.8559e-07,  ..., 2.7149e-03, 8.8400e-03,
         2.9715e-07],
        [1.8196e-03, 1.9918e-03, 4.5842e-03,  ..., 1.0221e-04, 7.5052e-05,
         9.0614e-02]]), 'X': tensor([[ -9.4541,  -0.3327,   8.8246],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.6681,  -4.8556,   9.2518],
        [ -0.8569,   7.9156,  -6.5557],
        [ -2.8162,   7.7059,  -9.6885],
        [  0.3493,  -4.7727,  -6.5682],
        [  9.2692,   9.6370,   7.0746],
        [ -9.2508,   1.0572,   2.9978],
        [ -1.1885, -10.8211,   0.0865],
        [  1.7659,   7.4127,   4.7871],
        [ 12.5996,   1.3971,  -1.0694],
        [ -8.3900,   3.3727,   3.1650],
        [  9.0004,  -7.7195,   4.6945],
        [  2.7685, -10.8982,  -6.8928],
        [ 11.2767,  -4.6006,   0.7868],
        [ -8.1204,  -5.9308,   4.0761],
        [ -4.0409,   7.3169,  -3.5779],
        [  1.0391,  -4.8141,  -0.0768],
        [ -5.6423,   6.4537,  -1.2091],
        [ -7.2515,   5.0195,   4.6434],
        [ -3.2053, -10.1993,  -3.8736],
        [  9.2811,  -7.4336,   5.1771],
        [ -0.2586,   7.8825,  -9.9840],
        [ 12.2921,  -1.7902,  -4.0870],
        [  5.4867,  12.5816,   2.9549],
        [ -8.7409,  -4.6837,  -6.5840],
        [  1.3403,  -4.8011,  -1.1698],
        [  3.6249, -10.7255,   5.6605],
        [ -3.7964,  -9.9294,   7.7147],
        [ -5.0550,   6.8231,  -9.2502],
        [  7.5930,  -8.9037,   7.2323],
        [ 12.3759,  -1.3943,   6.4659],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -1.5408,   7.8978,   5.3470],
        [ 12.1588,   4.3364,  -8.0867],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.2902,   0.0451,   3.7116],
        [ -7.8359,   4.2735,   7.2943],
        [  6.1752,   2.5242,   3.5874],
        [  7.0642,  11.6104,  -9.8316],
        [  0.1677,  -4.7447,  -6.1529],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 11.5568,   6.0602,   9.3244],
        [  3.2226,  -4.2802,  -4.0428],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -7.6719,  -6.6260,   7.1468],
        [  0.1382, -11.0073,   1.0026],
        [  2.6614,  -4.5192,  -9.3641],
        [ 10.8218,  -5.4177,   6.8573],
        [  5.2608,  12.6977,   4.9166],
        [  2.3589,  13.7668,  -3.6501],
        [ -7.5603,   4.6459,   8.2002],
        [ 12.2592,   3.9425,   2.0383],
        [  3.1271, -10.8343,   2.6442],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.3124,   0.2057,  -0.4593],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.5430,   5.5438,  -3.8669],
        [ -3.1204, -10.2345,   9.2288],
        [  1.7659,   7.4127,   4.7871],
        [  1.9241,   7.3503,   7.9004],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.3360,   0.4258,  -9.2509],
        [  5.3860,  -2.3418,   1.0826],
        [ 11.0864,   7.0387,  -0.1238],
        [  8.2422,  -8.4041,  -1.4962],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -0.0282,   7.8575,  -1.1746],
        [ 12.6035,   0.7494,   7.1536],
        [ -1.3178, -10.7939,   0.4236],
        [  0.9945,  -4.8144,  -0.1897],
        [ 12.6000,   1.3839,  -3.2992],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.9912,   3.1453,  -3.9989],
        [  2.3367,   7.1676,  -9.5263],
        [  9.6614,   9.1784,  -1.2087],
        [  7.5309,  11.2617,   7.2656],
        [  6.1984,  -0.4327,   1.5438],
        [  1.9505,  -4.7171,   9.9868],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.1292,   4.8183,  -5.7997],
        [  1.6422,   7.4586,  -6.8056],
        [  1.4749,  -4.7892,  -0.8089],
        [  1.9241,   7.3503,   7.9004],
        [ 11.9936,  -2.8677,   6.0614],
        [ 11.9936,  -2.8677,   6.0614],
        [  2.7295,  13.6703,  -8.9328],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 11.9641,  -2.9568,   7.7211],
        [  7.7750,  11.0665,   2.8364],
        [  4.1140,  -3.7220,  -0.1273],
        [  5.9587,  12.3216,   9.1695],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.2339,  -3.6268,   6.3125],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.2480, -11.0153,  -3.0183],
        [  5.3602,  -2.3808,   7.2098],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.3419,   0.5972,  -1.2841]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.6200e-02, 5.1053e-02,  ..., 8.8091e-01, 5.1960e-01,
         6.9032e-01],
        [4.6200e-02, 0.0000e+00, 2.0764e-02,  ..., 9.2410e-01, 5.6219e-01,
         7.1268e-01],
        [5.1053e-02, 2.0764e-02, 0.0000e+00,  ..., 9.3191e-01, 5.7065e-01,
         6.9512e-01],
        ...,
        [8.8091e-01, 9.2410e-01, 9.3191e-01,  ..., 2.1453e-04, 3.6247e-01,
         1.0613e+00],
        [5.1960e-01, 5.6219e-01, 5.7065e-01,  ..., 3.6247e-01, 0.0000e+00,
         8.3821e-01],
        [6.9032e-01, 7.1268e-01, 6.9512e-01,  ..., 1.0613e+00, 8.3821e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.45995955,  9.64157461,  9.77588801,  9.79814997,  9.87720102,
        9.9353921 ,  7.96181288,  8.20436764,  4.78544875, 13.37121923,
        9.3109884 , 10.88618114,  7.62011734, 12.67680439,  9.04255655,
       11.85743425, 11.24434374, 12.17901033, 10.05558893,  8.35856487,
        4.92497514,  8.5724042 ,  8.81930525, 10.6910768 , 11.89105259,
        7.88677538, 12.4217523 , 13.72594022,  9.91668182,  4.98461983,
       11.32149259, 10.63038565,  8.49161154, 11.70168755, 12.45418236,
       10.1979669 ,  8.04664886, 12.9089559 ,  9.99211663,  6.29034958,
        8.92549752,  6.67121748, 13.59056613,  4.74770899, 10.39942497,
       13.0493368 ,  5.35774573,  6.83083863,  9.43737523, 10.1371552 ,
       11.00812996,  5.24461068, 12.10222204, 13.744387  , 13.96746816,
        8.8737556 , 12.87752052, 11.27656488,  9.44486609,  6.31576264,
       10.00802627,  7.16747878, 10.69963954,  7.62011734,  7.59795445,
       12.1494023 , 12.30695098,  6.35029274,  5.87305005, 13.13205357,
       11.77124373, 13.41713661, 12.4013153 ,  7.8575648 , 12.62576026,
       10.87409005,  4.9160962 , 12.67576792,  9.99211663,  6.76661595,
        7.53884104, 13.32613445, 13.54774355,  6.2134961 ,  5.10446653,
        9.15866607,  7.03733775,  7.63726134,  5.01114719,  7.59795445,
       12.33167382, 12.33167382, 13.94008948, 12.12641969, 12.32408516,
       13.5247215 ,  5.54776934, 13.68672855,  4.91652556, 10.00802627,
        5.5748624 ,  8.62631801, 11.018084  ,  5.86519525, 11.70016217,
        9.36093562])}
{'P': tensor([[1.0691e-02, 3.9746e-01, 1.8550e-01,  ..., 8.1611e-07, 1.2910e-07,
         1.9684e-05],
        [2.9712e-04, 2.2541e-02, 9.3814e-02,  ..., 1.7583e-06, 3.1799e-07,
         5.7382e-05],
        [6.4988e-04, 4.3732e-03, 3.6856e-02,  ..., 3.5663e-06, 6.8939e-07,
         1.0658e-04],
        ...,
        [2.3804e-05, 1.9766e-05, 2.2755e-05,  ..., 1.0926e-01, 2.5959e-04,
         7.1198e-06],
        [1.1894e-06, 1.0144e-06, 1.1488e-06,  ..., 6.2113e-02, 2.0345e-01,
         8.4131e-06],
        [4.6174e-06, 4.4056e-06, 4.2957e-06,  ..., 1.9250e-04, 1.9034e-04,
         9.8499e-02]]), 'X': tensor([[ -9.4624,  -0.4633,  -7.3918],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.7409,  -4.6837,  -6.5840],
        [  3.3744,   6.5623,  -6.3636],
        [  5.4242, -10.1261,   3.9595],
        [  0.1497,   7.8334,  -5.4412],
        [ -8.9942,  -3.9990,   6.0448],
        [ -9.1795,  -3.3623,   8.5259],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.3062,  -5.6008,   0.0287],
        [  0.1206,  -4.7363,   8.7103],
        [  2.6614,  -4.5192,  -9.3641],
        [  1.9204,   7.3518,   8.4781],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.3860,  -2.3418,   1.0826],
        [  3.7121,   6.3117,   0.6738],
        [ 10.2712,   8.3707,  -4.9743],
        [  6.0467,  -9.8369,  -6.5816],
        [  2.3373,  -4.6232,  -5.2246],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.7845,   3.6652,   0.9175],
        [  6.3606,   0.9204,   3.2525],
        [ -3.0644,   7.6439,  -1.4322],
        [ -5.1377,   6.7751,  -0.4438],
        [ -6.6895,   5.6051,   2.7899],
        [ 11.6165,  -3.8715,   6.4594],
        [ -2.3317,   7.8034,  -3.6274],
        [  6.3211,   1.7017,  -0.2164],
        [  9.5261,  -7.1675,   1.4206],
        [  1.1350,  14.0104,   0.6321],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.2199,   4.6861,  -1.9304],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.2419,  -0.2313,   4.2440],
        [  6.2419,  -0.2313,   4.2440],
        [ -1.7009,   7.8851,  -8.2437],
        [ -8.0611,  -6.0302,   4.6962],
        [ -3.4986,   7.5152,   9.0712],
        [  3.4604,  -4.1545,   1.8091],
        [ 12.2589,  -1.9312,   9.8657],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 11.9808,   4.9340,  -3.3715],
        [ -9.4773,  -1.0024,   5.2599],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.5927,   0.4199,  -9.4997],
        [  4.2612, -10.5514,  -6.6268],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.4597,  -0.9136,  -4.8763],
        [  3.3943,   6.5483,   1.0790],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.9945,  -4.8144,  -0.1897],
        [  9.4554,   9.4245,   9.8800],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.1624,  -3.4286,   2.0647],
        [ 10.1798,   8.5002,  -5.8893],
        [  0.1460,   7.8340,  -9.4614],
        [  5.8979,   3.3958,  -3.8713],
        [ -1.9600,   7.8577,   6.9121],
        [  1.4306,  13.9620,  -0.2075],
        [  5.0617,  12.7957,   8.3544],
        [ -9.4697,  -0.6157,  -6.8472],
        [  7.5309,  11.2617,   7.2656],
        [ -8.4824,  -5.2575,  -0.0795],
        [  2.5636, -10.9293,   9.6519],
        [  5.7802,  -1.6363,  -6.2782],
        [  3.4920,  -4.1366,  -8.3955],
        [ -8.0534,  -6.0429,   2.4583],
        [  2.0039,   7.3173,  -1.5885],
        [  2.5974,   7.0363,  -7.7810],
        [ -9.4378,  -1.8606,  -2.5668],
        [  2.3373,  -4.6232,  -5.2246],
        [ -4.8751,  -9.3148,   4.0406],
        [  2.4572,   7.1085,   7.7376],
        [  0.2435,   7.8191,  -6.4709],
        [  3.4140,  -4.1803,  -8.8224],
        [  2.1379,  -4.6757,   6.4981],
        [  0.5600, -11.0321,  -0.8660],
        [  0.8940, -11.0402,  -0.9783],
        [  4.7739,  -3.1218,  -4.1188],
        [  1.6308,  13.9254,  -2.0984],
        [  3.2701,   6.6338,   9.9209],
        [  2.6970,   6.9826,   9.7350],
        [ -0.6323,   7.9085,  -5.9544],
        [  4.4049,   5.6892,  -4.4394],
        [ 10.6673,  -5.6649,  -4.1646],
        [  8.5716,  10.3596,   6.8818],
        [  5.1861,   4.7361,  -3.7312],
        [  1.7500,   7.4187,   4.4237],
        [ -4.3359,  -9.6431,  -9.0616],
        [  9.7328,   9.0902,  -0.8917],
        [ -7.3333,   4.9246,  -7.9212],
        [ -2.0229, -10.6163,   8.6930],
        [  1.7000,  -4.7609,   2.7935],
        [  0.9945,  -4.8144,  -0.1897],
        [  8.3791,  10.5409,   5.5590],
        [  6.0073,   3.0986,   6.7937],
        [ 12.2589,  -1.9312,   9.8657],
        [  8.1043,  -8.5164,  -7.9290]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 2.6322e-02, 3.9857e-02,  ..., 1.3503e+00, 1.5954e+00,
         7.5255e-01],
        [2.6322e-02, 0.0000e+00, 3.0803e-02,  ..., 1.3713e+00, 1.6154e+00,
         7.7837e-01],
        [3.9857e-02, 3.0803e-02, 0.0000e+00,  ..., 1.3522e+00, 1.5949e+00,
         7.8385e-01],
        ...,
        [1.3503e+00, 1.3713e+00, 1.3522e+00,  ..., 0.0000e+00, 2.5835e-01,
         1.0276e+00],
        [1.5954e+00, 1.6154e+00, 1.5949e+00,  ..., 2.5835e-01, 0.0000e+00,
         1.2803e+00],
        [7.5255e-01, 7.7837e-01, 7.8385e-01,  ..., 1.0276e+00, 1.2803e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.47369628,  9.562174  ,  9.60080639,  9.70108409,  9.75075273,
        9.91668182,  7.37902766, 11.48734978,  7.83486943,  9.84314712,
        9.77588801,  4.7663526 , 10.01804223,  4.73784222,  5.24461068,
        7.5984809 ,  9.61942445,  5.87305005,  7.32233733, 13.25017211,
       11.54672755,  5.18047638,  9.7751189 ,  6.18668463,  6.84796202,
        6.42689705,  8.23525433,  8.50278896,  8.7273569 , 12.24467041,
        8.14433542,  6.54615887, 11.92133129, 14.05633441, 10.39942497,
       10.92597196,  7.01475376, 12.67576792,  6.24614905,  6.24614905,
        8.06643487, 10.06703358,  8.28968067,  5.40689332, 12.41012046,
        9.58072286, 12.95702647,  9.53015376, 12.67576792, 12.59969936,
       11.37939695,  9.50654967, 10.8992524 , 12.49317563,  7.37576154,
        8.24808832, 10.46238054,  4.9160962 , 13.35013374,  6.19707944,
        9.62238828,  9.78285424, 13.26209545,  7.83534987,  6.80559669,
        8.09842704, 14.03505624, 13.76047999,  9.48969978, 13.54774355,
        9.97964013, 11.22596994,  6.00732193,  5.41348121, 10.06850677,
        7.58668144,  7.50035355,  9.61942445,  5.18047638, 10.51340829,
        7.52117132,  7.822854  ,  5.39723241,  5.14124612, 11.04629471,
       11.07637531,  5.70405525, 14.02058621,  7.39599986,  7.48538905,
        7.9337647 ,  7.19513806, 12.07818871, 13.44593519,  7.02326515,
        7.62232307, 10.57302546, 13.31764181,  8.83341608, 10.80728788,
        5.05534609,  4.9160962 , 13.46553623,  6.7593917 , 12.41012046,
       11.75618513])}
{'P': tensor([[5.9245e-03, 1.9604e-01, 1.4214e-01,  ..., 1.0229e-06, 8.4664e-02,
         5.7023e-06],
        [1.3444e-04, 1.0122e-02, 3.6539e-01,  ..., 1.4497e-06, 1.4129e-02,
         1.0783e-05],
        [2.2000e-04, 3.2535e-04, 2.6633e-02,  ..., 3.4101e-06, 1.1449e-03,
         3.3473e-05],
        ...,
        [3.2765e-03, 2.3219e-03, 1.8006e-03,  ..., 3.8119e-01, 2.9676e-03,
         6.1913e-04],
        [6.5278e-04, 9.2896e-03, 2.6537e-01,  ..., 5.0643e-06, 3.0609e-02,
         3.9076e-05],
        [6.6765e-05, 5.5038e-05, 4.8214e-05,  ..., 7.0807e-03, 6.1306e-05,
         9.3132e-02]]), 'X': tensor([[-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.1934e-02, 4.8606e-02,  ..., 7.8798e-01, 8.4000e-02,
         9.6645e-01],
        [3.1934e-02, 3.0339e-04, 2.7048e-02,  ..., 8.1979e-01, 9.4242e-02,
         9.9766e-01],
        [4.8606e-02, 2.7048e-02, 0.0000e+00,  ..., 8.2757e-01, 7.9145e-02,
         1.0126e+00],
        ...,
        [7.8798e-01, 8.1979e-01, 8.2757e-01,  ..., 0.0000e+00, 7.7236e-01,
         3.2079e-01],
        [8.4000e-02, 9.4242e-02, 7.9145e-02,  ..., 7.7236e-01, 0.0000e+00,
         9.7568e-01],
        [9.6645e-01, 9.9766e-01, 1.0126e+00,  ..., 3.2079e-01, 9.7568e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.48337063,  9.58347456,  9.68323686,  9.55593374,  9.73579738,
        9.60359158,  5.83560009,  6.24187044, 13.06498554, 10.33958639,
        8.47371868,  6.14086582,  5.87305005,  5.22903117, 11.77124373,
       11.69264243,  7.5984809 , 11.11038551, 10.46238054, 11.87665501,
        5.70591986,  7.85919687,  9.96391045,  4.87083348, 13.6922425 ,
       11.63724792, 13.35013374,  4.74468406, 12.95702647,  9.94285511,
       11.50791859,  9.38232121,  6.42689705,  9.73579738, 12.18998195,
       13.3614598 ,  8.74470218,  9.48969978, 12.42554808,  7.15990676,
        8.87587457,  9.60412735,  6.33075682, 11.62770332,  5.16077764,
       12.85032836, 11.54672755,  6.61211224,  8.31625171,  8.10886021,
        6.32572323,  7.17789666,  7.52117132,  9.55593374,  9.41605684,
       13.56367857,  7.17789666,  8.28968067, 11.00812996, 11.69264243,
       13.6105595 , 14.05633441, 12.3344536 ,  7.6374603 , 12.88270018,
        6.39187285,  9.43737523, 13.88683436, 12.42554808,  5.37343173,
       11.018084  ,  7.06272256, 12.32408516,  5.13616224,  5.19986797,
        6.46356806, 11.57249984, 10.23946247,  4.91038884, 13.27717982,
        7.96575782, 12.70489563, 13.25017211,  6.2322769 , 11.11969815,
       11.48734978,  9.34061938,  5.56809335, 12.88270018,  6.65368551,
       12.59840193,  7.92607342, 13.54774355, 13.93790235, 13.6660671 ,
       13.85158417, 12.91143491, 13.55868684,  4.74468406,  5.96556248,
       13.45124917,  7.02496223,  7.98469166,  7.83534987,  9.60359158,
        6.24187044])}
{'P': tensor([[1.1339e-02, 3.7150e-01, 1.6050e-01,  ..., 6.6117e-07, 1.9309e-04,
         1.4679e-05],
        [2.8850e-04, 2.3408e-02, 2.7775e-03,  ..., 1.2777e-06, 5.1019e-04,
         3.4733e-05],
        [7.0962e-04, 8.9986e-02, 2.1621e-02,  ..., 1.0144e-06, 3.5444e-04,
         3.1744e-05],
        ...,
        [1.2916e-04, 1.0456e-04, 8.8744e-05,  ..., 1.2356e-01, 1.9413e-04,
         1.9957e-05],
        [1.0397e-04, 1.1350e-04, 9.8031e-05,  ..., 5.2200e-04, 1.1659e-01,
         2.6513e-03],
        [5.8259e-05, 4.6988e-05, 5.2436e-05,  ..., 6.7943e-05, 1.2816e-04,
         5.1317e-02]]), 'X': tensor([[ -9.4697,  -0.6157,  -6.8472],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.3900,   3.3727,   3.1650],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.3542,   0.6934,  -3.3840],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.3333,   4.9246,  -7.9212],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.4401,  13.4540,   4.6576],
        [  2.3261,  13.7749,   5.9121],
        [ -8.3653,   3.4184,   9.3454],
        [  4.8674,  12.8875,   6.7622],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.6426,  -4.9136,   9.4941],
        [ -6.2668,   5.9808,   2.5614],
        [  6.2770,  -0.0379,   2.4195],
        [  1.4306,  13.9620,  -0.2075],
        [ -7.5720,   4.6310,   2.1717],
        [  5.1861,   4.7361,  -3.7312],
        [  6.8704,  -9.3809,   7.5769],
        [  5.9003,  12.3550,   2.1917],
        [  1.8721,  -4.7322,   0.5524],
        [ -2.9497, -10.3028,  -7.3479],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.1579,  -0.5961,  -6.5595],
        [ -5.1974,   6.7396,   3.3484],
        [  4.5340,  -3.3629,   8.9652],
        [  6.0955,   2.8210,  -4.3321],
        [  2.6613,   7.0021,   4.5783],
        [ 10.0567,   8.6697,   4.6329],
        [ -3.7964,  -9.9294,   7.7147],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.6063,  13.0051,  -3.1456],
        [  6.3497,   1.3680,  -1.9076],
        [  5.5240,  -2.1198,   6.2055],
        [  4.8432,   5.1968,   1.5006],
        [ -1.6334, -10.7207,   7.1397],
        [  6.3583,   1.1729,   8.7835],
        [ -4.7334,   6.9982,  -0.4392],
        [ -7.8849,  -6.3115,  -4.7479],
        [  1.5653,  13.9377,   4.1773],
        [ -5.1398,  -9.1367,   6.2074],
        [  5.9193,  -1.3176,   5.1739],
        [  5.8275,  -1.5334,   2.0124],
        [ 12.5717,   1.9266,   7.4903],
        [ -1.7290,   7.8825,  -3.3831],
        [  9.9058,  -6.7203,   5.6928],
        [  6.0305,   3.0296,   0.1375],
        [  1.4150,  13.9647,  -4.0915],
        [  2.1625,  -4.6697,   7.0658],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.0190,   4.9705,  -7.9752],
        [  5.1861,   4.7361,  -3.7312],
        [  3.4140,  -4.1803,  -8.8224],
        [ 12.4691,   2.8525,  -5.5110],
        [  7.0358,  -9.2784,  -9.8479],
        [ -8.6112,   2.9319,  -7.9943],
        [ -7.8280,   4.2848,  -7.1955],
        [  2.3261,  13.7749,   5.9121],
        [  0.6363, -11.0348,   6.1414],
        [  3.2692,  13.5098,  -4.9032],
        [ -0.0410,   7.8591,   9.3457],
        [  8.5716,  10.3596,   6.8818],
        [ -9.4766,  -1.1116,  -2.1706],
        [  6.9956,  11.6590,  -7.6876],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.1066,  -0.7799,  -4.1746],
        [ -0.7599, -10.8995,  -6.0366],
        [ -5.9938,  -8.4745,   3.9735],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.3360,   0.4258,  -9.2509],
        [ -3.5903,   7.4847,  -8.9112],
        [ -5.3179,  -9.0101,   9.4472],
        [  3.0876,   6.7527,  -5.3540],
        [ -7.8280,   4.2848,  -7.1955],
        [  2.1625,  -4.6697,   7.0658],
        [  0.3493,  -4.7727,  -6.5682],
        [ -2.3317,   7.8034,  -3.6274],
        [  4.5325,  -3.3643,   6.7699],
        [  3.5284, -10.7484,   5.3347],
        [  3.9770, -10.6341,   8.0783],
        [ -3.0008, -10.2827,  -8.9480],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -3.5029,   7.5138,   8.1853],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.3384,  -2.6161,  -4.7630],
        [  2.6613,   7.0021,   4.5783],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.8617,   3.4853,  -9.8080],
        [  6.0890,  -0.8382,   2.9166],
        [  5.1142,  -2.7230,  -4.1966],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.0305,   3.0296,   0.1375],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.0330,  -6.0765,   6.7945],
        [  2.0729,   7.2878,   7.0880],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.3048,   0.1469,  -9.1691]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0308, 0.0377,  ..., 1.2312, 0.9432, 0.9762],
        [0.0308, 0.0000, 0.0308,  ..., 1.2469, 0.9683, 1.0065],
        [0.0377, 0.0308, 0.0000,  ..., 1.2686, 0.9796, 1.0047],
        ...,
        [1.2312, 1.2469, 1.2686,  ..., 0.0000, 0.5297, 0.9528],
        [0.9432, 0.9683, 0.9796,  ..., 0.5297, 0.0000, 0.4281],
        [0.9762, 1.0065, 1.0047,  ..., 0.9528, 0.4281, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.48969978,  9.60080639,  9.562174  ,  9.7712321 ,  9.70108409,
        9.91668182,  8.41148722,  9.04255655,  8.89458273,  6.39187285,
        8.83341608,  8.83341608, 11.77124373, 13.88683436, 13.96987806,
        9.0368458 , 13.77604293,  6.10774544,  9.50654967,  9.94173708,
        8.66273171,  6.27715105, 14.03505624,  8.87587457,  7.02326515,
       11.62770332, 13.69162795,  5.08910491, 10.7167285 ,  8.89458273,
        6.18668463,  8.51088913,  5.64500512,  6.7166297 ,  7.49076899,
       13.2778378 , 10.63038565, 10.01804223, 13.79676654,  6.49538841,
        5.91677188,  7.10378919, 10.84437705,  6.46560181,  8.4486771 ,
       10.09979747, 14.02532758, 10.48315436,  6.06416957,  6.02588517,
       12.71843462,  8.06990483, 11.9702762 ,  6.74872981, 14.03618649,
        5.14607014, 11.51811603,  7.06372172,  7.02326515,  5.39723241,
       12.79126785, 11.64436684,  9.09661212,  8.92395381, 13.96987806,
       11.05317597, 13.89974693,  7.85919687, 13.44593519,  9.54154096,
       13.59673406, 12.11735334,  6.1561646 , 10.92597196, 10.37998078,
       10.06286074,  6.10774544,  6.35029274,  8.30125201, 10.46238054,
        7.42512621,  8.92395381,  5.14607014,  4.78544875,  8.14433542,
        5.64464807, 11.31276109, 11.35345345, 10.71162681, 12.12641969,
        8.29022197,  7.53884104,  9.69792513,  7.49076899, 10.54658597,
       10.31886066,  6.81961996,  6.1463896 ,  5.79391825,  9.95552626,
        6.74872981,  7.09887557, 10.07238463,  7.57685908,  4.92497514,
        6.30647746])}
{'P': tensor([[1.4294e-02, 1.8790e-01, 1.8167e-01,  ..., 6.2761e-06, 4.3296e-05,
         8.6308e-06],
        [9.7816e-05, 1.4292e-02, 6.9004e-01,  ..., 2.0363e-06, 5.6727e-05,
         1.0287e-05],
        [3.8348e-04, 1.0887e-03, 6.3564e-02,  ..., 8.6578e-06, 2.6958e-04,
         4.7876e-05],
        ...,
        [8.4848e-05, 7.1856e-05, 7.4354e-05,  ..., 4.9601e-02, 9.3881e-04,
         2.1361e-05],
        [1.0297e-06, 1.4372e-06, 1.5325e-06,  ..., 2.9690e-06, 3.1013e-02,
         5.7056e-04],
        [3.3812e-05, 1.8409e-05, 1.8075e-05,  ..., 5.7701e-05, 2.0197e-05,
         6.2341e-02]]), 'X': tensor([[ -9.4716,  -0.6673,   7.6068],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.1231,  -3.5739,   8.4708],
        [ -9.1749,  -3.3803,   6.4629],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.2535,  -0.1709,  -9.4102],
        [ -9.3810,  -2.3455,  -9.5194],
        [  5.9193,  -1.3176,   5.1739],
        [  0.1287, -11.0065,   6.5757],
        [  5.1886,  -2.6250,   6.4677],
        [ -6.0844,  -8.3955,   0.7705],
        [  6.3266,   0.3292,   9.3502],
        [  2.5663,   7.0526,  -6.8769],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.4267,  -5.3697,   2.4172],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 11.7726,  -3.4876,  -6.7402],
        [  5.5158,  -2.1337,   3.5246],
        [  3.1271, -10.8343,   2.6442],
        [ -3.5903,   7.4847,  -8.9112],
        [ -0.3432,   7.8900,   7.1805],
        [ -4.0868,   7.2981,   8.1874],
        [ -9.3087,   0.7792,  -7.5587],
        [  3.8111, -10.6787,  -4.6539],
        [  2.6614,  -4.5192,  -9.3641],
        [ -1.3178, -10.7939,   0.4236],
        [ 10.5057,  -5.9099,  -5.1599],
        [  9.3753,   9.5171,   3.9131],
        [  5.0233,   4.9647,  -3.1788],
        [ 11.9019,  -3.1379,  -3.4534],
        [  6.2555,   2.1474,   4.1685],
        [  1.1274,  14.0116,   8.1753],
        [ 11.6544,   5.8273,  -3.1375],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.9026,   3.3837,  -0.6573],
        [  6.3360,   0.4258,  -9.2509],
        [ -9.1795,  -3.3623,   8.5259],
        [ -8.6675,  -4.8569,  -9.1769],
        [  6.3124,   0.2057,  -0.4593],
        [  2.0937,  13.8293,  -2.3533],
        [  0.9445,   7.6736,   0.5095],
        [ -1.8788, -10.6567,   3.7068],
        [  4.7482, -10.3907,   5.9149],
        [  8.5198,  10.4091,  -8.6486],
        [ -4.0409,   7.3169,  -3.5779],
        [ -1.7009,   7.8851,  -8.2437],
        [  2.1771, -10.9776,  -3.2007],
        [  9.0652,   9.8597,  -3.5949],
        [  5.9193,  -1.3176,   5.1739],
        [ -2.1533, -10.5778,   1.0112],
        [  5.8836,   3.4315,  -4.0327],
        [ -8.9178,   2.1936,   7.5388],
        [  1.1139,   7.6281,  -2.5680],
        [  5.4732,  -2.2041,  -4.3830],
        [  9.1620,  -7.5573,  -9.9882],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.5240,  -2.1198,   6.2055],
        [ -1.5261,   7.8988,   9.3768],
        [ -7.5720,   4.6310,   2.1717],
        [ -5.1369,   6.7755,   8.3836],
        [ -3.5029,   7.5138,   8.1853],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.8890,  -4.4316,   1.1600],
        [  0.6470, -11.0352,  -1.1931],
        [  7.8375,  -8.7240,  -9.8913],
        [ -3.6381,   7.4683,   7.3544],
        [ -8.7735,   2.5640,  -6.3313],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.7301,   7.4262,   5.0522],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.3124,   0.2057,  -0.4593],
        [  3.9895,   6.0816,  -6.5235],
        [  5.6523,   3.9443,  -2.9449],
        [  4.0078,  -3.8018,  -6.2168],
        [ 10.5955,   7.8824,  -3.7006],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.9843,  -9.8679,  -6.1449],
        [  4.1637,  13.1898,  -1.9189],
        [  2.4932,   7.0903,  -5.4214],
        [  4.0078,  -3.8018,  -6.2168],
        [  7.7794,  11.0629,   8.9069],
        [  3.9663,   6.1018,  -5.4104],
        [  2.1120,  -4.6818,   8.6144],
        [  1.3103,  -4.8032,  -0.9300],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.4179,  -2.2925,  -3.4641],
        [  9.2811,  -7.4336,   5.1771],
        [  4.2339,  -3.6268,   6.3125],
        [  5.9883,   3.1536,   6.2320],
        [ 11.4743,   6.2478,  -2.1100],
        [  2.4572,   7.1085,   7.7376],
        [  6.1755,   2.5233,  -2.0042],
        [  6.3606,   0.9204,   3.2525],
        [  5.1472,  -2.6801,   8.1194],
        [ 12.4691,   2.8525,  -5.5110],
        [  0.6757, -11.0361,  -1.3047],
        [  4.4039,   5.6903,   8.7366],
        [  3.8182,  -3.9347,  -1.7961],
        [  2.3589,  13.7668,  -3.6501],
        [ -7.0825,   5.2073,  -5.0497],
        [ -8.9033,   2.2329,  -8.8912],
        [ -0.7599, -10.8995,  -6.0366],
        [  6.3124,   0.2057,  -0.4593]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0670, 0.0689,  ..., 1.0272, 1.0135, 1.0350],
        [0.0670, 0.0000, 0.0050,  ..., 1.0741, 1.0680, 1.1009],
        [0.0689, 0.0050, 0.0000,  ..., 1.0721, 1.0667, 1.1035],
        ...,
        [1.0272, 1.0741, 1.0721,  ..., 0.0000, 0.1797, 0.9922],
        [1.0135, 1.0680, 1.0667,  ..., 0.1797, 0.0000, 0.8244],
        [1.0350, 1.1009, 1.1035,  ..., 0.9922, 0.8244, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.49511524,  9.77588801,  9.79814997,  9.77777346,  9.84314712,
        9.87720102,  6.25586297,  9.66977802,  6.06416957, 11.00726483,
        5.81481044, 10.36844813,  6.33517349,  7.50500064, 11.97957425,
        9.99211663, 12.00995898, 12.27835996,  5.91408504, 11.27656488,
        8.30125201,  7.89745107,  8.36445758,  9.34127013, 11.33837396,
        5.24461068, 10.87409005, 12.05395056, 13.35927527,  7.06272256,
       12.30858677,  6.61386436, 14.05687815, 13.03002338, 11.67875855,
        6.80370254,  6.35029274,  9.77588801,  9.93553424,  6.31576264,
       13.98691426,  7.73151309, 10.82106505, 11.42420508, 13.45124917,
        8.35856487,  8.06643487, 11.1913598 , 13.39372426,  6.06416957,
       10.794755  ,  6.81117802,  9.18358562,  7.70897675,  5.90035324,
       11.87665501,  4.91652556,  5.91677188,  8.04483495,  8.87587457,
        8.50268322,  8.29022197, 11.72749923,  5.29009836, 11.05413545,
       11.72749923,  8.30729352,  9.14044984,  9.88571713,  7.62509582,
       10.06703358,  6.31576264,  7.27340834,  6.89245352,  5.52416368,
       13.20597932,  9.99145096, 10.38897563, 11.51811603, 11.54071859,
       13.83139024,  7.5158489 ,  5.52416368, 13.52430176,  7.27759233,
        5.13616224,  4.978704  , 12.46513646,  5.8828986 , 11.89105259,
        5.5748624 ,  6.76791483, 13.06498554,  7.52117132,  6.67108662,
        6.42689705,  5.80312949, 12.79126785, 11.05672136,  7.19534146,
        5.48276038, 13.96746816,  8.7907923 ,  9.17905491, 10.92597196,
        6.31576264])}
{'P': tensor([[1.1796e-02, 1.9854e-01, 1.9534e-01,  ..., 2.3070e-07, 6.7317e-06,
         4.2537e-05],
        [1.0406e-04, 1.6375e-02, 2.4377e-04,  ..., 2.8169e-07, 2.2169e-05,
         1.4568e-04],
        [4.7425e-04, 2.6631e-01, 2.1059e-02,  ..., 3.8525e-07, 1.2226e-05,
         7.8401e-05],
        ...,
        [7.1995e-05, 2.1578e-05, 4.5930e-05,  ..., 3.4576e-02, 4.7140e-08,
         1.2042e-07],
        [4.0200e-07, 6.2004e-07, 3.7966e-07,  ..., 8.4187e-07, 3.6586e-02,
         4.4553e-04],
        [1.5176e-06, 2.6727e-06, 1.5497e-06,  ..., 1.2196e-06, 1.1028e-01,
         4.3881e-02]]), 'X': tensor([[ -9.4731,  -0.7120,  -6.0271],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -0.0410,   7.8591,   9.3457],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.1222,  -0.7261,   8.8204],
        [  7.3592,  11.3937,   6.2407],
        [ -7.6293,  -6.6859,  -5.5161],
        [  5.7218,  -9.9934,   6.5764],
        [  5.1203,  -2.7151,   6.2110],
        [  4.7813,  -3.1139,   8.2691],
        [  8.1528,  10.7452,   9.1493],
        [  4.7813,  -3.1139,   8.2691],
        [ 12.5919,   0.4035,  -5.8460],
        [  6.2576,   2.1359,  -3.3398],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.7784,  -1.6401,   9.5587],
        [  6.9693,  -9.3200,  -8.3822],
        [ 12.6035,   0.7494,   7.1536],
        [  0.9074,  -4.8139,   8.1674],
        [  6.3550,   0.7129,   0.4917],
        [  6.3515,   0.6418,  -6.5823],
        [  6.2675,  12.1379,  -8.9720],
        [  6.9956,  11.6590,  -7.6876],
        [  6.3303,   1.6145,  -8.4083],
        [  1.9241,   7.3503,   7.9004],
        [  1.8766,   7.3694,   1.1887],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -9.0604,   1.7707,   1.1066],
        [ -0.8455, -10.8853,  -6.7400],
        [  5.8318,   3.5567,  -4.7692],
        [ 12.0042,  -2.8350,   2.6864],
        [  2.8890,  -4.4316,   1.1600],
        [  2.3889,  -4.6082,  -8.7291],
        [  5.1292,   4.8183,  -5.7997],
        [  7.0664,  -9.2590,   8.0879],
        [  5.2199,   4.6861,  -1.9304],
        [  0.4996,   7.7738,  -4.4711],
        [ 11.9122,   5.1397,  -1.5498],
        [ 12.3609,   3.4769,  -8.5762],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.5158,  -2.1337,   3.5246],
        [ -9.4773,  -1.0024,   5.2599],
        [ -8.0087,   4.0173,   9.1091],
        [ 11.3800,  -4.3919,   1.4066],
        [  6.8093,  -9.4178,   8.4244],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.6022,  -1.9833,   8.3081],
        [ -6.8466,  -7.6488,   1.9648],
        [ 11.6801,  -3.7196,  -0.9181],
        [  1.8447,  -4.7372,   8.1235],
        [  0.9478,  -4.8144,   0.6371],
        [  1.6308,  13.9254,  -2.0984],
        [  8.1528,  10.7452,   9.1493],
        [  3.3094,  13.4969,  -3.6507],
        [  2.0243,   7.3086,   7.5432],
        [ -4.1296,   7.2803,  -2.7194],
        [  1.0093,   7.6567,   2.3526],
        [  3.7298,  13.3537,  -9.4653],
        [  2.2644,  -4.6434,   9.7508],
        [  2.4639, -10.9431,  -7.3718],
        [ -7.6525,  -6.6534,   5.1721],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.2247,  -0.3150,  -9.3657],
        [  2.8994,   6.8675,  -9.5987],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.8930,  12.3592,  -8.9201],
        [  6.1929,   2.4501,  -6.9352],
        [  5.3118,   4.5452,   4.2114],
        [ 10.9931,   7.2107,   7.7933],
        [ -0.3432,   7.8900,   7.1805],
        [  1.3339,   7.5627,  -0.9034],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.3261,  13.7749,   5.9121],
        [ -4.0868,   7.2981,   8.1874],
        [ -0.1963,   7.8765,  -1.1371],
        [  5.4429,  -2.2529,   9.7306],
        [  6.0310,  12.2795,  -6.4528],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.3604,   0.4855,  -3.0158],
        [  2.2076,  13.8032,  -1.5099],
        [  3.3633,  13.4794,   0.5193],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3551,   0.7149,   1.3344],
        [ -9.3739,   0.3982,  -4.2564],
        [ -5.1377,   6.7751,  -0.4438],
        [  9.2979,   9.6048,   7.3138],
        [ -3.2736, -10.1703,   3.8712],
        [ 11.0438,   7.1180,  -5.6991],
        [  9.9217,  -6.7006,  -5.6728],
        [  3.2647, -10.8066,  -0.3768],
        [  1.7228,  -4.7575,   8.3396],
        [  6.2895,   0.0405,   5.7652],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.0031,  12.2958,  -6.5992],
        [  2.5128, -10.9365,   9.9645],
        [  0.1287, -11.0065,   6.5757]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0530, 0.0492,  ..., 1.1957, 1.1866, 1.0203],
        [0.0530, 0.0000, 0.0407,  ..., 1.2487, 1.2067, 1.0337],
        [0.0492, 0.0407, 0.0000,  ..., 1.2283, 1.2324, 1.0632],
        ...,
        [1.1957, 1.2487, 1.2283,  ..., 0.0000, 1.3955, 1.4247],
        [1.1866, 1.2067, 1.2324,  ..., 1.3955, 0.0000, 0.2247],
        [1.0203, 1.0337, 1.0632,  ..., 1.4247, 0.2247, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.4997987 ,  9.7712321 ,  9.60080639,  9.78286733,  9.69792513,
        9.85330646,  7.85919687, 12.49291768, 10.42667093,  6.16512856,
       13.56367857, 10.14436912, 11.51557166,  5.79562691,  5.70591986,
       13.48809595,  5.70591986, 12.59840193,  6.61211224,  9.97964013,
        6.00663314, 11.63765273, 12.62576026,  4.89869205,  6.3949043 ,
        6.383897  , 13.66051884, 13.59673406,  6.53290663,  7.59795445,
        7.60463349, 12.53345445,  9.23178204, 10.91805683,  6.83083863,
       12.3344536 ,  5.29009836,  5.19065439,  7.03733775, 11.64746882,
        7.01475376,  7.78980299, 12.97370371, 12.84056604,  9.62958973,
        5.91408504,  9.53015376,  8.95983586, 12.19805394, 11.62158554,
        4.91652556,  5.94293185, 10.26545792, 12.25806559,  5.08372149,
        4.90676242, 14.02058621, 13.48809595, 13.89671314,  7.58378655,
        8.36995953,  7.72291946, 13.86480175,  5.16611782, 11.21703654,
       10.14044489, 11.66453059,  6.23262289,  7.4544866 , 12.64235359,
       13.6922425 ,  6.65991877,  6.99097013, 13.14690137,  7.89745107,
        7.67939546, 12.40074901, 13.96987806,  8.36445758,  7.8788965 ,
        5.89074104, 13.68063098,  9.96152788,  9.75401449,  9.79039371,
        6.00732193, 13.02818702,  9.37296102, 13.97857591, 13.89264822,
       12.3489324 ,  6.39520397,  9.38232121,  8.50278896, 13.36800073,
       10.68416401, 13.13888841, 11.97238624, 11.28895489,  5.05981914,
        6.28961988,  9.58072286,  8.91551231, 13.6829843 , 11.22142195,
       11.00726483])}
{'P': tensor([[5.8607e-03, 1.9329e-01, 1.4471e-01,  ..., 9.9022e-05, 2.7691e-05,
         1.6144e-06],
        [5.8873e-05, 5.0618e-03, 1.9480e-01,  ..., 1.4136e-04, 3.0734e-05,
         1.2557e-06],
        [8.4952e-05, 1.3972e-04, 1.0184e-02,  ..., 4.0358e-04, 6.9117e-05,
         2.4476e-06],
        ...,
        [2.9430e-06, 3.2118e-06, 3.4811e-06,  ..., 4.0818e-02, 7.5396e-04,
         5.9593e-05],
        [9.3198e-05, 9.3470e-05, 9.1420e-05,  ..., 6.6318e-04, 1.9772e-01,
         3.2660e-04],
        [1.0010e-03, 6.6218e-04, 5.2607e-04,  ..., 1.4938e-05, 3.1117e-05,
         1.0157e-01]]), 'X': tensor([[ -9.4735,  -0.7274,  -2.6749],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.1658,  -3.4155,  -2.8880],
        [  5.7802,  -1.6363,  -6.2782],
        [ 10.6283,  -5.7253,   7.4878],
        [  3.4401,  13.4540,   4.6576],
        [  3.4920,  -4.1366,  -8.3955],
        [  3.3943,   6.5483,   1.0790],
        [ -3.9508,  -9.8515,   1.7923],
        [  3.7121,   6.3117,   0.6738],
        [ -9.3419,   0.5972,  -1.2841],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.3900,   3.3727,   3.1650],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -6.4822,  -8.0252,  -3.8983],
        [  6.1575,  12.2046,   3.3257],
        [ -8.0087,   4.0173,   9.1091],
        [  5.9526,   3.2525,  -4.3821],
        [ -5.1046,  -9.1610,   2.3418],
        [ -9.4748,  -0.7765,  -5.6695],
        [  1.3103,  -4.8032,  -0.9300],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.1793,   4.7461,   0.3899],
        [  2.6310, -10.9195,   3.8054],
        [  5.9912,   3.1453,  -3.9989],
        [  4.6357,   5.4415,   7.9067],
        [ -2.7572,   7.7194,   5.9580],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.5626,  -2.0536,   8.4433],
        [  4.5430,   5.5438,  -3.8669],
        [ 11.9936,  -2.8677,   6.0614],
        [ -4.2040,   7.2486,   2.9875],
        [ 11.6165,  -3.8715,   6.4594],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 12.3357,   3.6001,   0.7903],
        [ -5.1398,  -9.1367,   6.2074],
        [  1.9241,   7.3503,   7.9004],
        [  4.9152,  12.8653,  -1.4459],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.1531,  -5.8748,   2.6934],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.1588,   4.3364,  -8.0867],
        [  9.2979,   9.6048,   7.3138],
        [ 11.3118,   6.5957,  -4.5667],
        [ -1.0482, -10.8487,  -5.9066],
        [ 10.7566,   7.6212,  -2.3279],
        [  5.3858,   4.4260,  -9.6769],
        [  1.4749,  -4.7892,  -0.8089],
        [ 11.9641,  -2.9568,   7.7211],
        [  3.2809,  13.5061,  -4.0199],
        [ -9.3616,  -2.4749,   0.1793],
        [ -2.2508, -10.5479,  -2.7235],
        [ -7.4808,   4.7462,   2.0266],
        [  6.9653,  -9.3225,  -3.4727],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.9022,  13.2906,   5.5788],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.3616,  -2.4750,   3.7765],
        [  4.2275,  -3.6320,  -0.2714],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.3464,   0.5708,  -1.6397],
        [  4.4633,   5.6287,  -0.7754],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -2.0229, -10.6163,   8.6930],
        [  4.3459,  -3.5327,   0.3667],
        [ -6.3306,   5.9272,   5.6935],
        [  7.7750,  11.0665,   2.8364],
        [ -8.8363,   2.4086,  -5.6296],
        [  6.8407,  11.7666,   3.0045],
        [ -7.4887,   4.7364,   8.9074],
        [  7.1697,  11.5343,  -0.7276],
        [  1.1644, -11.0394,   7.8511],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.7544,   2.6096,   3.3146],
        [  6.2576,   2.1359,  -3.3398],
        [  5.3860,  -2.3418,   1.0826],
        [  1.1994,   7.6035,   2.4295],
        [  9.4554,   9.4245,   9.8800],
        [  6.3205,   1.7068,  -3.8619],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.2218,   4.0953,  -8.2072],
        [  5.6314,  -1.9302,  -9.6543],
        [ -9.4351,  -1.8891,   3.5642],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.2939,  -2.4784,  -4.5840],
        [  1.5246,  -4.7838,  -5.2095],
        [ -5.6423,   6.4537,  -1.2091],
        [ -1.0407,   7.9166,  -6.6021],
        [ -7.5603,   4.6459,   8.2002],
        [ -3.6381,   7.4683,   7.3544],
        [  0.1047, -11.0046,   0.5188],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.8890,  -4.4316,   1.1600],
        [  6.2036,   2.4030,  -1.2500],
        [ -5.6423,   6.4537,  -1.2091],
        [ 12.5996,   1.3971,  -1.0694],
        [ -3.3812, -10.1235,   7.4384],
        [  2.5663,   7.0526,  -6.8769],
        [  3.0719, -10.8449,  -4.1673],
        [  3.3474,  -4.2162,   5.2455],
        [  2.5974,   7.0363,  -7.7810]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0300, 0.0488,  ..., 0.4193, 1.0899, 0.7824],
        [0.0300, 0.0000, 0.0205,  ..., 0.4481, 1.1155, 0.8114],
        [0.0488, 0.0205, 0.0000,  ..., 0.4625, 1.1360, 0.8258],
        ...,
        [0.4193, 0.4481, 0.4625,  ..., 0.0000, 0.8735, 0.3633],
        [1.0899, 1.1155, 1.1360,  ..., 0.8735, 0.0000, 0.8056],
        [0.7824, 0.8114, 0.8258,  ..., 0.3633, 0.8056, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.5014146 ,  9.61942445,  9.69629   ,  9.75401449,  9.71392179,
        9.78147542,  6.00732193, 12.07225469, 13.88683436,  5.41348121,
        7.37576154, 10.61417181,  7.32233733,  9.36093562, 12.21532282,
        9.04255655, 12.4013153 , 10.31613269, 13.66991027,  8.95983586,
        6.78325151, 10.48721382,  9.50654967,  4.978704  ,  9.77777346,
        7.02496223, 11.23201619,  6.76661595,  7.14837514,  8.19703512,
        9.69629   ,  5.92952538,  7.16747878, 12.33167382,  8.37954853,
       12.24467041, 10.32715715, 12.85032836, 10.48315436,  7.59795445,
       13.77222752,  7.28793663, 10.04916678, 12.91143491, 12.9089559 ,
       13.36800073, 13.09426142, 10.8992524 , 13.18278606,  6.97106912,
        5.01114719, 12.32408516, 13.89886469,  9.68323686, 10.78533838,
        8.859416  , 11.63724792, 12.19805394, 13.85158417, 11.75618513,
        9.68324553,  5.57341623,  6.10774544,  5.91854397,  9.36378438,
        7.18354885,  9.69629   , 10.80728788,  5.60064319,  8.67227386,
       13.5247215 ,  9.15866607, 13.6105595 ,  8.86082124, 13.58101794,
       11.10066126, 10.95593729,  9.13507321,  6.61211224,  5.87305005,
        7.69752262, 13.35013374,  6.54693987, 12.53285633,  8.06643487,
       12.889695  ,  5.95297459,  9.62238828,  5.68942572,  5.84533345,
        5.0209005 ,  8.5724042 ,  7.98469166,  8.8737556 ,  8.30729352,
       11.00508681,  9.9353921 ,  5.29009836,  6.652742  ,  8.5724042 ,
       12.67680439, 10.67322527,  7.50500064, 11.27160035,  5.38342077,
        7.50035355])}
{'P': tensor([[8.8684e-03, 1.6435e-01, 1.3524e-01,  ..., 3.4787e-06, 1.3344e-05,
         1.7610e-05],
        [6.2042e-05, 9.8957e-03, 4.3645e-04,  ..., 9.2618e-06, 2.2338e-05,
         8.9867e-06],
        [6.7405e-05, 5.2158e-02, 5.7940e-03,  ..., 4.8806e-06, 1.1495e-05,
         9.0964e-06],
        ...,
        [3.5306e-07, 2.9263e-07, 3.3475e-07,  ..., 1.2972e-02, 7.4163e-04,
         4.2180e-07],
        [1.5166e-04, 9.7470e-05, 1.0843e-04,  ..., 3.0777e-04, 1.3245e-01,
         1.6790e-05],
        [2.0186e-04, 3.6490e-04, 4.6880e-04,  ..., 3.2166e-05, 1.1086e-05,
         7.9725e-02]]), 'X': tensor([[-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0545, 0.0457,  ..., 0.9890, 0.9613, 0.9809],
        [0.0545, 0.0000, 0.0402,  ..., 1.0410, 1.0146, 1.0029],
        [0.0457, 0.0402, 0.0000,  ..., 1.0098, 0.9994, 0.9633],
        ...,
        [0.9890, 1.0410, 1.0098,  ..., 0.0000, 0.5046, 0.8283],
        [0.9613, 1.0146, 0.9994,  ..., 0.5046, 0.0000, 1.2443],
        [0.9809, 1.0029, 0.9633,  ..., 0.8283, 1.2443, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.50654967,  9.7712321 ,  9.69792513,  9.78286733,  9.85330646,
        9.88571713,  5.20659892, 12.88270018, 13.40850328,  5.88782913,
       11.83509372,  9.9353921 ,  5.05534609, 13.76047999,  4.91038884,
        8.95133961, 12.05300943, 12.97370371,  7.85919687,  8.95133961,
        5.79543038, 11.62770332,  8.44940661,  5.84450127,  7.14837514,
        5.14607014, 12.22276278, 11.51811603,  7.53063885, 10.24557047,
        9.48337063,  9.49511524, 11.52733581,  8.7907923 ,  7.72415048,
       10.37998078, 11.64746882,  5.04028538,  9.91668182,  8.79284466,
        9.50654967, 11.83570132,  6.28961988,  5.70591986, 12.52875793,
        9.95552626,  7.71075226, 11.00726483, 12.27991991, 10.0674887 ,
       11.31276109, 12.52875793,  5.05444288, 11.785566  , 13.66896739,
       12.27835996,  7.98469166,  7.05627461,  9.78286733, 12.52875793,
        8.85572292, 14.09357547, 12.97370371,  5.19065439, 10.68416401,
        7.8788965 , 13.61600536,  7.50500064, 13.52430176, 12.1494023 ,
        9.48969978,  7.27340834,  6.2134961 , 13.00380624, 13.39372426,
        5.30158497,  9.18358562, 14.05687815,  8.55982105,  5.14607014,
        4.91336994,  8.62162372, 13.59673406, 13.52430176, 10.31584102,
       10.46739073,  7.61088125, 11.63383221,  5.2287058 ,  9.17905491,
        6.69389431, 11.05317597, 13.95472331,  4.7663526 ,  9.88571713,
        5.26342203, 11.20100795, 10.61047596,  5.82345867, 13.00413829,
        7.54127425,  5.01114719,  5.2287058 , 11.99633819,  6.27905292,
        9.45995955])}
{'P': tensor([[1.1416e-02, 2.5324e-01, 2.0222e-01,  ..., 2.6127e-07, 3.4466e-05,
         1.2867e-05],
        [8.7036e-05, 9.3343e-03, 8.4027e-02,  ..., 1.5474e-07, 5.6241e-05,
         1.5722e-05],
        [1.1467e-04, 7.7367e-04, 1.7542e-02,  ..., 2.6796e-07, 1.5551e-04,
         2.7840e-05],
        ...,
        [4.0658e-03, 1.6080e-03, 1.1643e-03,  ..., 3.6028e-01, 2.1449e-06,
         1.8867e-05],
        [3.4321e-06, 3.3461e-06, 3.5729e-06,  ..., 1.6591e-05, 7.6130e-02,
         7.8041e-04],
        [9.1109e-05, 8.0880e-05, 6.5407e-05,  ..., 4.2700e-04, 1.9381e-04,
         2.2744e-01]]), 'X': tensor([[-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0395, 0.0632,  ..., 0.9616, 0.7143, 1.1272],
        [0.0395, 0.0000, 0.0402,  ..., 1.0011, 0.7531, 1.1600],
        [0.0632, 0.0402, 0.0000,  ..., 1.0109, 0.7575, 1.1904],
        ...,
        [0.9616, 1.0011, 1.0109,  ..., 0.0000, 0.2783, 0.6762],
        [0.7143, 0.7531, 0.7575,  ..., 0.2783, 0.0000, 0.8079],
        [1.1272, 1.1600, 1.1904,  ..., 0.6762, 0.8079, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.50913849,  9.69792513,  9.7712321 ,  9.78286733,  9.85330646,
        9.88571713, 11.48434767,  5.07306274, 14.07623266, 13.0493368 ,
        6.76056625,  8.51088913, 13.83139024, 10.74409834, 13.59170809,
        7.92644746, 14.03618649,  8.06517143,  6.89245352,  6.48486171,
       11.76830981, 11.77124373,  5.92952538,  6.81084724, 10.14044489,
        4.91038884,  4.89869205,  8.8580671 , 11.83075005,  4.83371341,
        7.72291946,  7.19534146, 11.97957425, 10.75443106,  9.18358562,
        7.8788965 ,  5.41348121,  7.53884104, 11.85743425,  6.56833864,
        9.23071698, 11.75807252,  7.24326992, 11.57249984, 10.48315436,
        7.54127425,  9.3349858 , 14.05633441, 13.96746816, 14.03618649,
       13.00413829,  9.10945678,  9.19837792,  8.85939626, 10.88618114,
        6.05113217,  8.09984231, 10.65929968,  9.23178204,  7.20482224,
        8.0550286 , 12.85032836, 10.1979669 ,  9.09502385,  5.30158497,
        8.57560275, 11.05672136,  7.07902036, 11.66453059,  8.04664886,
       13.62896593,  8.86082124,  8.19703512, 13.103709  ,  4.91038884,
        6.16512856, 10.82106505, 12.3489324 , 13.13888841,  7.15990676,
        5.5671389 , 11.33837396, 10.88618114, 10.52357479,  9.3109884 ,
        7.06272256,  9.50654967,  7.83534987, 11.24434374,  9.15866607,
       12.12641969,  4.978704  ,  7.09887557,  8.29640613, 13.45642379,
        7.15100303, 12.44177641,  4.91652556,  8.27853076, 10.14044489,
       13.91198191,  8.67502348,  4.78002665, 14.09176259, 11.63765273,
        5.98285319])}
{'P': tensor([[7.8977e-03, 2.2482e-01, 1.3682e-01,  ..., 1.5275e-07, 1.9465e-04,
         3.0692e-06],
        [1.0686e-04, 1.1057e-02, 2.0970e-01,  ..., 1.9448e-07, 2.9879e-04,
         3.4502e-06],
        [1.0717e-04, 3.2944e-04, 1.3999e-02,  ..., 2.4148e-07, 4.8361e-04,
         3.9708e-06],
        ...,
        [6.9788e-04, 3.4348e-04, 2.4247e-04,  ..., 3.3996e-01, 1.0969e-05,
         1.9786e-01],
        [2.1564e-05, 3.8161e-05, 5.9592e-05,  ..., 4.8134e-07, 5.3379e-02,
         3.2973e-06],
        [5.9727e-03, 3.1991e-03, 2.3280e-03,  ..., 4.4535e-04, 6.6356e-05,
         1.2410e-01]]), 'X': tensor([[ -9.4756,  -0.8166,  -9.6021],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.4297,  -5.3637,  -9.3962],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.0243,   7.3086,   7.5432],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -9.1907,  -3.3180,  -5.5943],
        [  8.7131,  10.2216,   7.7915],
        [ -9.3087,   0.7792,  -7.5587],
        [  9.1409,  -7.5789,   1.5814],
        [  2.7590,  13.6621,   9.4496],
        [  3.2647, -10.8066,  -0.3768],
        [  2.5636, -10.9293,   9.6519],
        [ -4.0868,   7.2981,   8.1874],
        [  4.1999,  -3.6543,  -4.1998],
        [ 12.5650,   2.0132,   4.6272],
        [  5.8275,  -1.5334,   2.0124],
        [  4.6992,   5.3690,   3.0935],
        [  0.2480, -11.0153,  -3.0183],
        [  7.1441,  11.5529,   2.3232],
        [  0.1287, -11.0065,   6.5757],
        [ -3.5173,   7.5091,  -8.9585],
        [  2.3928,   7.1404,  -1.7035],
        [ -5.1369,   6.7755,   8.3836],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -7.4602,   4.7718,  -4.6277],
        [  3.4401,  13.4540,   4.6576],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.2687,  -2.9794,   0.6436],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.1666,   3.7653,   3.6075],
        [ -9.4630,  -1.5145,   0.2983],
        [ -7.4733,   4.7555,   6.2490],
        [  5.8201,   3.5841,  -8.0609],
        [ -7.5720,   4.6310,   2.1717],
        [  5.6248,   3.9984,   8.4933],
        [  6.3604,   0.9042,  -6.5901],
        [  3.3069,   6.6089,   8.7377],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.1477,   7.2550,   8.9688],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.7544,   2.6096,   3.3146],
        [ -4.1296,   7.2803,  -2.7194],
        [ 11.0275,  -5.0666,   8.4154],
        [  4.8653,   5.1694,  -4.0823],
        [  2.7685, -10.8982,  -6.8928],
        [ -4.2271,   7.2387,  -1.8316],
        [  4.1140,  -3.7220,  -0.1273],
        [  3.9663,   6.1018,  -5.4104],
        [  6.2790,  -0.0259,  -5.4116],
        [  3.3167,   6.6022,   6.3554],
        [  9.3836,   9.5075,  -5.9278],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.5919,   0.4035,  -5.8460],
        [  0.1931,  -4.7491,   7.4385],
        [  5.8450,  -9.9356,   3.0767],
        [ -5.0943,  -9.1682,   0.5073],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.1252, -11.0062,   7.2564],
        [  6.3122,  -9.6994,  -6.3282],
        [  5.4167, -10.1293,   6.2156],
        [  0.7145,  -4.8071,  -2.3766],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.3261,   7.1726,   6.2522],
        [ -5.6169,  -8.7844,   8.9057],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.4541,  -0.3327,   8.8246],
        [ -7.4733,   4.7555,   6.2490],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.2373,  -3.1225,  -8.9763],
        [  1.3767, -11.0341,   9.9415],
        [  5.5626,  -2.0536,   8.4433],
        [  8.6271,  -8.0715,  -7.4813],
        [  1.8766,   7.3694,   1.1887],
        [  6.2022,   2.4092,   3.0349],
        [ -0.5385,   7.9037,   6.3830],
        [  3.7992,  -3.9474,  -8.9019],
        [ -1.3940, -10.7772,   6.7835],
        [ -1.9714,   7.8563,  -1.8084],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -4.2040,   7.2486,   2.9875],
        [  4.8432,   5.1968,   1.5006],
        [ -3.1204, -10.2345,   9.2288],
        [  2.9410,   6.8428,   5.4458],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.1379,  -4.6757,   6.4981],
        [  7.8982,  -8.6778,   0.0239],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.9033,   2.2329,  -8.8912],
        [ -5.3736,   6.6310,  -1.5174],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.6587,   9.1817,  -0.8212],
        [  6.2975,   0.0947,   5.1791],
        [ -1.7290,   7.8825,  -3.3831],
        [  5.0617,  12.7957,   8.3544],
        [  6.8407,  11.7666,   3.0045],
        [  2.5663,   7.0526,  -6.8769],
        [  5.8930,  12.3592,  -8.9201],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.1460,   7.8340,  -9.4614]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0306, 0.0352,  ..., 1.2511, 0.7059, 0.7186],
        [0.0306, 0.0000, 0.0071,  ..., 1.2774, 0.7226, 0.7470],
        [0.0352, 0.0071, 0.0000,  ..., 1.2839, 0.7202, 0.7530],
        ...,
        [1.2511, 1.2774, 1.2839,  ..., 0.0000, 1.4227, 0.5517],
        [0.7059, 0.7226, 0.7202,  ..., 1.4227, 0.0000, 0.9157],
        [0.7186, 0.7470, 0.7530,  ..., 0.5517, 0.9157, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.51074763,  9.66977802,  9.75075273,  9.93553424,  9.70108409,
        9.99145096, 12.41366754,  7.58378655, 10.00410625,  9.7712321 ,
       13.43127131,  9.34127013, 11.87412439, 13.93790235, 11.28895489,
       11.22596994,  8.36445758,  5.5671389 , 12.72524615,  6.02588517,
        7.13501264, 11.018084  , 13.58333857, 11.00726483,  8.29203356,
        7.53063885,  8.50268322, 12.49291768,  8.85572292, 13.88683436,
        5.1224079 , 13.66896739,  9.73579738,  6.14086582,  8.99276564,
        9.58347456,  8.8580671 ,  6.83515201,  8.87587457,  6.90115901,
        6.42439344,  7.39004226,  9.96391045,  9.58347456,  7.56617037,
       11.62158554,  9.13507321,  8.36995953, 12.13568598,  7.09887557,
       11.24434374,  8.38252937,  5.54776934,  7.27759233,  6.27905292,
        7.38844703, 13.35832725,  9.34127013, 12.59840193,  4.75303127,
       11.52733581, 10.48840476,  5.52416368, 11.00694688, 11.57249984,
       11.48663767,  4.85995002, 12.7125079 ,  7.5403822 , 10.42667093,
        5.0209005 ,  9.45995955,  8.8580671 ,  6.12886024,  9.75075273,
       11.11969815,  5.92952538, 11.81423161,  7.60463349,  6.65368551,
        7.92200342,  5.47866221, 10.86693708,  8.09984231, 12.27835996,
        8.37954853,  7.10378919, 10.69963954,  7.44805901, 11.99633819,
        5.14124612, 11.73397309, 10.92597196,  9.17905491,  8.53501777,
        9.3109884 ,  9.66977802, 13.32645552,  6.29821673,  8.06990483,
       13.76047999, 13.6105595 ,  7.50500064, 13.6922425 , 10.26545792,
        7.83534987])}
{'P': tensor([[1.7087e-02, 2.2034e-01, 1.9883e-01,  ..., 1.9943e-05, 2.9416e-05,
         3.1800e-03],
        [1.3040e-04, 2.3512e-02, 5.4429e-04,  ..., 8.0007e-05, 1.5063e-05,
         1.9604e-02],
        [1.5451e-04, 3.8860e-01, 1.7170e-02,  ..., 4.2940e-05, 1.3333e-05,
         1.2107e-02],
        ...,
        [1.0537e-06, 8.8840e-07, 8.6460e-07,  ..., 2.6853e-02, 2.8142e-06,
         1.9376e-06],
        [6.0687e-02, 2.0886e-02, 2.9096e-02,  ..., 5.6798e-06, 4.4839e-02,
         5.2550e-04],
        [9.2437e-06, 2.0898e-05, 1.7422e-05,  ..., 2.2573e-03, 4.3861e-06,
         3.8069e-02]]), 'X': tensor([[ -9.4773,  -1.0024,   5.2599],
        [ -8.9942,  -3.9990,   6.0448],
        [ -9.1749,  -3.3803,   6.4629],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.3616,  -2.4750,   3.7765],
        [ -2.5056,   7.7720,   5.5878],
        [  4.6231,   5.4556,   8.4973],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.2365,  -0.2578,  -6.0212],
        [ -9.3616,  -2.4749,   0.1793],
        [  8.5093,  -8.1764,  -3.2381],
        [ -5.0132,  -9.2234,  -8.3531],
        [  3.3094,  13.4969,  -3.6507],
        [  2.6614,  -4.5192,  -9.3641],
        [ 11.3691,   6.4760,   1.0787],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.1755,   2.5233,  -2.0042],
        [  9.0004,  -7.7195,   4.6945],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.4159,  -2.0739,   9.6841],
        [ -4.9217,   6.8979,  -1.7667],
        [  3.2809,  13.5061,  -4.0199],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.2767,  -4.6006,   0.7868],
        [  0.9809,   7.6642,  -1.4217],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.1271, -10.8343,   2.6442],
        [ -8.7409,  -4.6837,  -6.5840],
        [  4.5814, -10.4485,   2.5496],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.9193,  -1.3176,   5.1739],
        [  0.6470, -11.0352,  -1.1931],
        [ -6.2668,   5.9808,   2.5614],
        [  6.2905,   1.9366,   2.1169],
        [ -9.4430,  -0.1897,   4.8383],
        [ -8.4573,   3.2448,   5.4370],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.0000,   7.6591,  -4.5011],
        [  5.4179,  -2.2925,  -3.4641],
        [ -0.1626, -10.9796,   6.4629],
        [ 11.5198,  -4.0923,   5.5088],
        [  0.8940, -11.0402,  -0.9783],
        [  1.3103,  -4.8032,  -0.9300],
        [  9.3424,  -7.3685,  -3.6807],
        [ -8.3062,  -5.6008,   0.0287],
        [ 11.7779,   5.5127,  -2.0699],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.9756,  -6.6330,  -3.3429],
        [  0.5850,  -4.7982,   9.6639],
        [ -1.3940, -10.7772,   6.7835],
        [ 12.5996,   1.3971,  -1.0694],
        [ 10.9931,   7.2107,   7.7933],
        [ -2.4956, -10.4682,   3.5632],
        [ -3.1204, -10.2345,   9.2288],
        [ 12.4597,  -0.9136,  -4.8763],
        [  1.9503,  -4.7172,  -6.2313],
        [  2.0937,  13.8293,  -2.3533],
        [ 12.2435,   4.0076,   8.1257],
        [  1.3670,  -4.7990,   3.2603],
        [ -5.6792,  -8.7352,   4.0318],
        [  1.1644, -11.0394,   7.8511],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -8.6229,   2.9068,   2.4429],
        [ -8.7544,   2.6096,   3.3146],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.8474,   6.8979,   6.6457],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -9.0961,   1.6529,   3.1553],
        [  0.1206,  -4.7363,   8.7103],
        [ -7.4733,   4.7555,   6.2490],
        [  0.6933,   7.7336,   2.1179],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.7121,   6.9743,  -8.4202],
        [  1.9312,  13.8649,   8.3563],
        [  5.8930,  12.3592,  -8.9201],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -2.3317,   7.8034,  -3.6274],
        [  7.4132,  11.3526,   0.3322],
        [  7.8425,  -8.7202,   6.4530],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.1504,  -4.6726,   8.7908],
        [  3.6249, -10.7255,   5.6605],
        [  0.1497,   7.8334,  -5.4412],
        [  6.3031,   1.8477,  -7.1706],
        [  6.1755,   2.5233,  -2.0042],
        [  6.0073,   3.0986,   6.7937],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.9193,  -1.3176,   5.1739],
        [  4.9337,  12.8566,  -3.6582],
        [  5.6022,  -1.9833,   8.3081],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.3236,   0.3011,  -4.6668],
        [ -5.1377,   6.7751,  -0.4438],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.1196,  -2.7160,   4.2705],
        [ -2.2188,   7.8218,   8.6064],
        [ -9.4351,  -1.8891,   3.5642],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.9193,  -1.3176,   5.1739],
        [  9.1409,  -7.5789,   1.5814],
        [ -7.0120,   5.2825,   7.7762],
        [ -3.1204, -10.2345,   9.2288]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 8.3429e-02, 9.6980e-02,  ..., 7.0768e-01, 3.3062e-01,
         2.7301e-01],
        [8.3429e-02, 0.0000e+00, 2.3540e-02,  ..., 7.7753e-01, 3.4754e-01,
         2.3393e-01],
        [9.6980e-02, 2.3540e-02, 3.0339e-04,  ..., 7.7680e-01, 3.3203e-01,
         2.1075e-01],
        ...,
        [7.0768e-01, 7.7753e-01, 7.7680e-01,  ..., 0.0000e+00, 5.5820e-01,
         7.6314e-01],
        [3.3062e-01, 3.4754e-01, 3.3203e-01,  ..., 5.5820e-01, 0.0000e+00,
         2.1668e-01],
        [2.7301e-01, 2.3393e-01, 2.1075e-01,  ..., 7.6314e-01, 2.1668e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.53015376,  9.84314712,  9.77777346,  9.75200848,  9.96391045,
        9.68324553,  8.16584537,  7.15100303,  8.22728399,  6.24187044,
        9.68323686, 11.80092578, 10.4977132 , 13.89671314,  5.24461068,
       13.08414689,  5.26067019,  6.67108662, 11.85743425,  9.75075273,
        9.64157461,  8.47371868, 13.89886469,  8.08320235, 12.17901033,
        7.72668467,  6.25586297, 11.27656488,  9.91668182, 11.4088043 ,
       10.01804223,  6.06416957, 11.05413545,  8.66273171,  6.58184474,
        9.44486609,  9.05843201,  9.38232121,  7.72415048,  5.8828986 ,
       10.98076873, 12.22503754, 11.07637531,  4.978704  , 11.89855267,
       10.01804223, 13.00413829,  9.66977802, 11.97957425,  4.83371341,
       10.86693708, 12.67680439, 13.14690137, 10.76154256, 10.69963954,
       12.49317563,  5.10444097, 13.98691426, 12.88270018,  4.98989906,
       10.41909211, 11.10066126,  9.48969978,  9.09963894,  9.13507321,
       12.9089559 ,  7.46248947,  6.383897  ,  8.98765644,  9.88571713,
        9.24501829,  4.73784222,  8.8580671 ,  7.76457095, 11.67875855,
        7.48311002, 13.99877413, 13.6922425 , 10.06407839,  8.14433542,
       13.55868684, 11.72802491,  7.9337647 ,  5.1224079 ,  5.14371005,
       11.32149259,  7.83486943,  6.56833864,  6.67108662,  6.7593917 ,
        4.98461983,  6.06416957, 13.77075174,  5.94293185, 12.13568598,
        6.33075682,  8.50278896,  9.36093562,  5.79543038,  8.1303945 ,
        9.62238828, 12.42554808,  6.06416957, 11.87412439,  8.77913428,
       10.69963954])}
{'P': tensor([[6.9958e-03, 2.4408e-01, 1.6904e-01,  ..., 2.7429e-06, 2.1517e-06,
         7.4303e-04],
        [2.4195e-04, 1.1866e-02, 8.7230e-02,  ..., 4.6058e-06, 3.5630e-06,
         1.2936e-03],
        [1.5606e-04, 6.0094e-04, 7.6983e-03,  ..., 3.0441e-06, 2.1586e-06,
         5.8436e-04],
        ...,
        [1.5081e-03, 1.3209e-03, 1.2036e-03,  ..., 3.0566e-01, 1.1024e-01,
         1.2333e-04],
        [1.5550e-03, 1.3782e-03, 1.1252e-03,  ..., 2.4668e-03, 1.2275e-01,
         2.9705e-04],
        [1.1555e-04, 1.4365e-04, 1.5198e-04,  ..., 5.1549e-06, 7.5052e-06,
         3.4917e-02]]), 'X': tensor([[-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0137, 0.0457,  ..., 0.8799, 0.8389, 0.6705],
        [0.0137, 0.0000, 0.0484,  ..., 0.8909, 0.8458, 0.6618],
        [0.0457, 0.0484, 0.0000,  ..., 0.9061, 0.8776, 0.7076],
        ...,
        [0.8799, 0.8909, 0.9061,  ..., 0.0000, 0.3507, 1.1357],
        [0.8389, 0.8458, 0.8776,  ..., 0.3507, 0.0000, 0.8747],
        [0.6705, 0.6618, 0.7076,  ..., 1.1357, 0.8747, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.54154096,  9.58072286,  9.61942445,  9.69629   ,  9.75401449,
        9.78147542,  5.05981914, 11.62158554,  6.83083863, 10.51437947,
        9.68324553, 10.91805683, 10.74655346,  6.42439344,  6.38399789,
        6.54693987,  6.05113217,  9.97220799, 12.04924159,  6.5618968 ,
       13.06957425,  5.30158497,  6.29821673,  5.95297459,  7.10378919,
       10.95593729, 12.53345445,  5.60064319, 14.07623266,  8.37954853,
       13.32645552, 10.61417181,  6.10360073, 10.14044489,  8.99276564,
        9.60359158,  5.14124612,  9.04255655, 11.59512194,  8.09899874,
        9.41244865,  5.87132718,  8.87007495, 11.785566  , 11.59512194,
       10.06286074,  7.14837514,  5.19577146, 12.71843462, 11.27160035,
       13.13205357,  9.5014146 ,  8.25828409,  7.85919687,  5.30158497,
        9.79039371, 10.65929968, 11.77124373, 13.97430191, 13.03002338,
        6.12886024, 11.69264243, 13.43127131, 10.48721382,  9.41605684,
        9.17016857, 12.26997165, 12.50996828,  5.47702766,  8.68832398,
       13.34587027, 10.06437598, 13.97779839,  5.81199422,  6.39187285,
        5.48276038,  5.53175275,  8.68832398,  4.74468406, 12.0495521 ,
       10.8207297 ,  5.5513072 ,  7.83534987, 13.39372426,  9.62238828,
       11.97949273, 10.4977132 ,  7.92607342, 13.95361098,  5.14607014,
        8.90405823,  8.57560275, 13.94008948, 13.2778378 , 13.32557133,
       12.1494023 , 10.74655346, 12.21532282,  8.67227386, 12.35421461,
       11.83570132,  5.19065439, 13.36092624,  7.15990676,  7.54127425,
        9.96391045])}
{'P': tensor([[8.7104e-03, 2.3787e-01, 2.3144e-01,  ..., 8.4475e-04, 2.2683e-05,
         6.8989e-05],
        [2.5512e-04, 2.0531e-02, 3.1176e-01,  ..., 1.0501e-03, 6.9085e-05,
         2.2504e-04],
        [2.1637e-04, 9.1787e-04, 2.4947e-02,  ..., 1.2617e-03, 8.9556e-05,
         2.8748e-04],
        ...,
        [8.2916e-04, 1.4411e-03, 2.2503e-03,  ..., 8.4131e-02, 7.9137e-05,
         1.5061e-04],
        [6.7542e-05, 5.8969e-05, 5.4132e-05,  ..., 1.6211e-05, 8.4008e-02,
         2.7229e-03],
        [4.1023e-05, 4.0494e-05, 3.8444e-05,  ..., 1.0022e-05, 1.2359e-01,
         5.6417e-02]]), 'X': tensor([[-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0484, 0.0378,  ..., 0.5164, 0.9171, 0.9074],
        [0.0484, 0.0000, 0.0304,  ..., 0.5517, 0.9422, 0.9285],
        [0.0378, 0.0304, 0.0000,  ..., 0.5218, 0.9524, 0.9414],
        ...,
        [0.5164, 0.5517, 0.5218,  ..., 0.0000, 1.1642, 1.1917],
        [0.9171, 0.9422, 0.9524,  ..., 1.1642, 0.0000, 0.0875],
        [0.9074, 0.9285, 0.9414,  ..., 1.1917, 0.0875, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.55593374,  9.68323686,  9.73579738,  9.58347456,  9.76896904,
        9.85384623, 10.53368644, 13.09737053, 12.79126785,  6.76056625,
       13.06498554,  7.71075226, 10.33712258, 11.87412439,  4.978704  ,
        8.09842704,  9.37296102,  6.65991877,  5.64464807,  8.62329466,
       11.64436684,  5.05534609, 10.04916678,  9.96391045,  8.50278896,
        6.97106912,  9.96391045, 11.27656488, 13.52430176,  9.19837792,
        9.37989631, 10.37698435, 10.08001894, 12.77402351,  7.19513806,
        9.19837792,  8.95983586, 13.84126197, 13.40024224,  6.00663314,
       13.94008948,  7.02253235,  9.51074763,  5.26067019,  7.57946459,
        7.72668467,  5.80860165,  8.22728399,  5.1805345 , 10.74409834,
        5.35774573,  9.84314712,  6.67108662, 10.68416401, 12.95702647,
        8.20436764,  5.40689332,  6.10360073,  7.06272256,  8.57560275,
       14.09357547, 11.89855267,  4.74468406,  7.58378655,  7.80625981,
        5.84450127,  6.34059021,  6.61386436,  7.32754882, 11.83570132,
        9.96152788,  9.0368458 , 12.78327037,  5.47866221,  7.24677664,
        9.62238828,  5.16077764, 13.06498554,  8.68832398,  9.09119351,
        8.41171369,  8.68979052,  5.89074104,  9.55593374, 12.4980845 ,
        5.13182506, 12.21532282, 13.31764181, 10.06407839, 12.1494023 ,
        5.8828986 , 12.0495521 , 12.67576792, 11.92133129, 11.97238624,
       11.73848065, 11.2845608 , 13.68063098, 13.88683436,  6.39187285,
       14.08549802,  8.90405823,  9.99211663,  9.79814997,  6.02789634,
        5.5366698 ])}
{'P': tensor([[1.0187e-02, 2.5997e-01, 1.7997e-01,  ..., 6.4114e-06, 5.3839e-06,
         4.1748e-07],
        [1.3809e-04, 1.3138e-02, 4.3498e-01,  ..., 5.7337e-06, 5.0094e-06,
         7.1831e-07],
        [6.7446e-04, 1.3349e-03, 8.6422e-02,  ..., 3.1959e-05, 2.8632e-05,
         5.6076e-06],
        ...,
        [1.9281e-02, 9.7700e-03, 6.9680e-03,  ..., 6.6235e-02, 7.8798e-03,
         3.1364e-06],
        [9.2503e-03, 5.1161e-03, 3.8279e-03,  ..., 4.0275e-02, 5.4030e-02,
         3.7036e-06],
        [4.8500e-06, 3.2187e-06, 2.6959e-06,  ..., 3.8381e-04, 8.6558e-04,
         5.3280e-02]]), 'X': tensor([[-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0340, 0.0519,  ..., 0.5363, 0.6360, 1.2300],
        [0.0340, 0.0000, 0.0186,  ..., 0.5627, 0.6551, 1.2465],
        [0.0519, 0.0186, 0.0000,  ..., 0.5730, 0.6605, 1.2500],
        ...,
        [0.5363, 0.5627, 0.5730,  ..., 0.0000, 0.1990, 0.7569],
        [0.6360, 0.6551, 0.6605,  ..., 0.1990, 0.0000, 0.5981],
        [1.2300, 1.2465, 1.2500,  ..., 0.7569, 0.5981, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.55720651,  9.71392179,  9.79039371,  9.78147542,  9.75401449,
        9.69629   , 11.85743425,  6.76661595, 10.00410625,  9.97822473,
        7.39599986,  8.92098752,  9.13507321,  7.92607342,  6.02779201,
        9.93553424,  8.92549752,  8.31625171,  5.14607014, 11.42947632,
        6.00732193,  9.87720102,  5.52416368, 11.18839169, 12.3828957 ,
       11.36089355, 12.26997165, 11.59512194,  6.81117802, 13.45642379,
       11.4088043 , 13.48809595,  7.16598273,  5.76165934,  9.62958973,
        8.27853076, 13.58101794,  8.44940661,  8.62329466,  9.05843201,
       13.1471313 , 10.42667093, 13.59056613, 10.53368644,  7.37576154,
        8.0550286 ,  5.0209005 , 12.27993098,  4.78002665, 13.38091121,
        7.98469166,  6.97106912, 11.07637531, 13.97430191,  5.35774573,
        5.81481044,  5.64500512,  7.09887557, 11.07637531, 12.41012046,
        6.27715105, 10.27716434,  9.7751189 , 10.01804223,  6.5618968 ,
        8.5724042 ,  7.17789666,  4.98461983,  8.29022197,  5.63824971,
        5.8828986 ,  5.43943791, 10.54658597,  5.81481044,  5.87305005,
       11.2845608 ,  5.13125513, 13.96987806,  4.9160962 , 12.5769659 ,
        5.47866221,  8.08320235, 10.35093485,  7.02326515,  8.19703512,
        8.67502348,  9.09661212, 11.97949273,  4.91652556,  7.19534146,
       10.87409005,  5.5205189 ,  8.7907923 , 11.28093507,  7.02326515,
        8.1137452 , 12.05395056,  7.72668467, 13.96987806,  9.84314712,
       10.84437705, 12.17901033,  8.12043983,  8.34949383,  8.25828409,
       12.87752052])}
{'P': tensor([[1.4108e-02, 2.4228e-01, 1.5515e-01,  ..., 3.2771e-06, 1.4833e-06,
         5.4410e-06],
        [3.6485e-04, 2.3018e-02, 1.9211e-01,  ..., 5.0017e-06, 1.5437e-06,
         6.4350e-06],
        [3.8784e-04, 1.0786e-03, 8.5002e-02,  ..., 4.3997e-05, 5.7938e-06,
         1.9263e-05],
        ...,
        [1.0030e-05, 7.4831e-06, 1.1052e-05,  ..., 1.5277e-01, 1.7113e-03,
         2.7241e-04],
        [2.8294e-04, 1.5582e-04, 1.6472e-04,  ..., 8.6539e-05, 6.4488e-02,
         1.2120e-03],
        [5.9349e-03, 3.0258e-03, 1.7565e-03,  ..., 2.7810e-05, 2.5434e-03,
         8.8223e-02]]), 'X': tensor([[ -9.4721,  -1.3097,  -7.4377],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.8093,  -9.4178,   8.4244],
        [  5.1667,  -2.6543,   1.2879],
        [  4.5954,  13.0099,  -4.4439],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.3062,  -5.6008,   0.0287],
        [  1.6308,  13.9254,  -2.0984],
        [  1.9241,   7.3503,   7.9004],
        [ -9.1795,  -3.3623,   8.5259],
        [  2.6970,   6.9826,   9.7350],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.0506,   4.9277,  -9.5094],
        [ -7.3446,   4.9113,  -5.9356],
        [ -1.8788, -10.6567,   3.7068],
        [  0.1287, -11.0065,   6.5757],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.4732,  -2.2041,  -4.3830],
        [ 12.5199,  -0.4711,   3.0436],
        [  4.5954,  13.0099,  -4.4439],
        [  3.6469,   6.3624,  -1.8414],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.3460,   1.4252,   1.2291],
        [  5.9003,  12.3550,   2.1917],
        [ -8.7409,  -4.6837,  -6.5840],
        [  4.1637,  13.1898,  -1.9189],
        [ -5.9940,   6.1985,   5.9566],
        [ -2.7572,   7.7194,   5.9580],
        [  2.1824,  -4.6647,   1.1985],
        [ -1.3940, -10.7772,   6.7835],
        [  4.7482, -10.3907,   5.9149],
        [ -1.0407,   7.9166,  -6.6021],
        [  4.0591,   6.0202,  -1.6339],
        [  0.9074,  -4.8139,   8.1674],
        [  7.9403,  -8.6455,  -8.7433],
        [ -3.2486,   7.5925,  -1.3719],
        [  0.9036,   7.6840,   9.9237],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -4.7334,   6.9982,  -0.4392],
        [  3.7121,   6.3117,   0.6738],
        [ -9.2687,  -2.9794,   0.6436],
        [  0.1041, -11.0045,   2.4311],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.2707,   0.9661,   4.4863],
        [ -6.7745,  -7.7264,   8.4684],
        [  4.8674,  12.8875,   6.7622],
        [  7.4132,  11.3526,   0.3322],
        [  3.3744,   6.5623,  -6.3636],
        [  3.9515,  -3.8425,   6.1729],
        [ 11.5966,   5.9666,   6.0083],
        [  4.0591,   6.0202,  -1.6339],
        [ -2.2508, -10.5479,  -2.7235],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -9.4714,  -1.3301,  -3.1998],
        [  9.4925,   9.3811,   8.9842],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.0275,  -5.0666,   8.4154],
        [  2.3589,  13.7668,  -3.6501],
        [ -1.7009,   7.8851,  -8.2437],
        [  1.4306,  13.9620,  -0.2075],
        [  6.3119,   1.7797,   9.9186],
        [  7.0664,  -9.2590,   8.0879],
        [  5.1861,   4.7361,  -3.7312],
        [  9.8082,  -6.8396,   4.9615],
        [ 12.5717,   1.9266,   7.4903],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.4573,   3.2448,   5.4370],
        [  5.6202,  -1.9508,   0.7471],
        [  1.3103,  -4.8032,  -0.9300],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.1519, -10.2387,   9.8191],
        [  0.7685,  -4.8098,  -2.9310],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.3464,   0.5708,  -1.6397],
        [ -4.8665,  -9.3204,  -5.2564],
        [  4.8674,  12.8875,   6.7622],
        [  6.1989,  -9.7591,   0.8289],
        [ -6.9838,  -7.4965,   0.9796],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.5294,  -2.1106,   9.0626],
        [  2.2182,  13.8007,   5.1889],
        [ -8.8631,  -4.3728,  -1.1746],
        [  2.9260,  -4.4162,   8.3808],
        [ -9.1749,  -3.3803,   6.4629],
        [  6.1066,  -0.7799,  -4.1746],
        [  7.8982,  -8.6778,   0.0239],
        [  9.9217,  -6.7006,  -5.6728],
        [ -5.2589,   6.7024,  -5.5809],
        [  2.2182,  13.8007,   5.1889],
        [  0.1047, -11.0046,   0.5188],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.2508,   1.0572,   2.9978],
        [  6.6319,  11.9066,   3.5165],
        [ -1.5261,   7.8988,   9.3768],
        [  5.1472,  -2.6801,   8.1194],
        [ -2.5056,   7.7720,   5.5878],
        [ -1.9487,   7.8590,  -3.3330]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0403, 0.0332,  ..., 1.4087, 1.1212, 0.7214],
        [0.0403, 0.0000, 0.0623,  ..., 1.4486, 1.1614, 0.7561],
        [0.0332, 0.0623, 0.0000,  ..., 1.3987, 1.1059, 0.7255],
        ...,
        [1.4087, 1.4486, 1.3987,  ..., 0.0000, 0.3601, 0.8038],
        [1.1212, 1.1614, 1.1059,  ..., 0.3601, 0.0000, 0.6578],
        [0.7214, 0.7561, 0.7255,  ..., 0.8038, 0.6578, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.562174  ,  9.70108409,  9.91668182,  9.75075273,  9.93553424,
        9.7712321 , 11.62158554,  5.80860165, 13.79762415,  8.06643487,
       12.62576026, 10.01804223, 14.02058621,  7.59795445,  9.77588801,
        7.48538905, 13.06498554,  7.05627461,  8.8353674 , 10.82106505,
       11.00726483, 12.5085995 ,  5.90035324, 12.52875793, 13.79762415,
        7.33350642,  9.78147542,  6.50409372, 13.69162795,  9.91668182,
       13.83139024,  8.62260327,  8.19703512,  5.14999799, 10.86693708,
       11.42420508,  7.98469166,  7.26074925,  4.89869205, 11.73848065,
        8.25828409,  7.73692436, 10.67322527,  9.51074763,  8.4486771 ,
        7.32233733,  9.73579738, 11.0050379 ,  5.24461068,  9.50913849,
        9.7712321 ,  9.32094055, 10.27572673, 13.77604293, 13.55868684,
        7.37902766,  5.51177501, 13.04156075,  7.26074925, 10.78533838,
       10.54658597,  9.5642993 , 13.34587027, 14.10796466, 12.13568598,
       13.96746816,  8.06643487, 14.03505624,  6.55800267, 11.64746882,
        7.02326515, 11.95743558, 12.71843462, 11.21703654,  9.05843201,
        5.94909709,  4.978704  , 10.61417181,  6.23262289, 11.46176259,
        4.87083348, 13.77075174,  9.36378438, 10.51437947, 13.77604293,
       11.56146873, 10.24557047,  5.56809335,  5.91854397, 13.97779839,
        9.88310614,  5.29755487,  9.77777346,  6.1561646 , 11.73397309,
       11.97238624,  8.51928531, 13.97779839, 11.00508681,  9.45995955,
        9.3109884 , 13.62896593,  8.04483495,  5.80312949,  8.16584537,
        8.09704052])}
{'P': tensor([[6.0394e-03, 1.8154e-01, 1.4441e-01,  ..., 1.5815e-05, 7.8579e-04,
         1.2846e-05],
        [1.0648e-04, 1.0856e-02, 2.6738e-03,  ..., 3.7975e-05, 1.2435e-03,
         1.3533e-05],
        [9.0564e-05, 1.6645e-02, 6.8520e-03,  ..., 2.1263e-05, 1.1063e-03,
         9.1022e-06],
        ...,
        [4.4182e-05, 3.8258e-05, 3.6418e-05,  ..., 6.7886e-02, 6.8121e-06,
         1.6621e-04],
        [1.3915e-04, 2.6796e-04, 2.7523e-04,  ..., 8.8044e-05, 8.7769e-02,
         1.3112e-05],
        [5.8926e-02, 3.4918e-02, 3.1877e-02,  ..., 4.0959e-05, 4.7312e-04,
         2.9537e-02]]), 'X': tensor([[ -9.4714,  -1.3301,  -3.1998],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -8.6229,   2.9068,   2.4429],
        [ 12.1940,  -2.1881,   7.1499],
        [  6.3413,   0.4883,  -6.8416],
        [ -2.0229, -10.6163,   8.6930],
        [ -0.6323,   7.9085,  -5.9544],
        [  0.1497,   7.8334,  -5.4412],
        [ 10.5955,   7.8824,  -3.7006],
        [ -6.7643,  -7.7372,  -9.6663],
        [  0.9659,  -4.8144,  -0.9594],
        [  5.8323,  -1.5228,   5.1132],
        [ -3.9700,   7.3453,  -4.1486],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.7218,  -9.9934,   6.5764],
        [  1.0093,   7.6567,   2.3526],
        [ -7.6650,   4.5092,   0.2520],
        [  7.9403,  -8.6455,  -8.7433],
        [  1.5246,  -4.7838,  -5.2095],
        [ 12.1504,   4.3674,  -7.2998],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -5.2752,  -9.0409,  -8.2568],
        [  9.3753,   9.5171,   3.9131],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -6.1914,  -8.2997,   8.5380],
        [ 12.4877,  -0.7222,   8.8347],
        [  6.3266,   0.3292,   9.3502],
        [ -9.3444,   0.5821,   6.9101],
        [ -0.8569,   7.9156,  -6.5557],
        [  9.7328,   9.0902,  -0.8917],
        [  1.7232,  -4.7574,   3.1270],
        [  5.5158,  -2.1337,   3.5246],
        [  9.3424,  -7.3685,  -3.6807],
        [  8.9284,  10.0036,  -6.9210],
        [  2.5318,  13.7231,   6.3692],
        [  4.1637,  13.1898,  -1.9189],
        [  9.4516,  -7.2501,  -1.8868],
        [  1.3705,  13.9723,   7.9266],
        [ -8.6681,  -4.8556,   9.2518],
        [  1.3403,  -4.8011,  -1.1698],
        [ -9.3107,  -2.7697,  -3.2919],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.4996,   7.7738,  -4.4711],
        [  2.8994,   6.8675,  -9.5987],
        [  4.2096,   5.8815,   1.9230],
        [ 12.2914,   3.8037,   6.0978],
        [ -1.6275, -10.7221,   3.2376],
        [  0.1382, -11.0073,   1.0026],
        [  6.2804,   2.0019,   0.8607],
        [  5.5240,  -2.1198,   6.2055],
        [ -7.6650,   4.5092,   0.2520],
        [  9.0129,  -7.7072,   9.1400],
        [  2.0899,  -4.6870,  -7.4260],
        [  5.2608,  -2.5256,  -9.1446],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.3200,   7.1756,   1.1938],
        [  9.0051,   9.9234,   5.4411],
        [  2.8474,   6.8979,   6.6457],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.9433,  -9.3362,  -0.3421],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.1752,   2.5242,   3.5874],
        [  6.3606,   0.9204,   3.2525],
        [  2.7121,   6.9743,  -8.4202],
        [ -4.0868,   7.2981,   8.1874],
        [ -5.2589,   6.7024,  -5.5809],
        [ 11.0438,   7.1180,  -5.6991],
        [ -9.4157,   0.0821,  -5.0025],
        [ 10.1798,   8.5002,  -5.8893],
        [ -4.6393,   7.0461,   1.1624],
        [ -3.3999, -10.1152,  -1.2088],
        [  5.5424,  -2.0885,   7.2487],
        [  5.5424,  -2.0885,   7.2487],
        [ 11.7795,  -3.4697,   3.5624],
        [ -7.5401,   4.6718,   9.0033],
        [ 12.5154,   2.5072,  -9.3274],
        [ -8.8363,   2.4086,  -5.6296],
        [  8.5093,  -8.1764,  -3.2381],
        [ 12.0845,  -2.5772,  -4.6073],
        [  0.6952,  -4.8060,   8.8976],
        [  0.9445,   7.6736,   0.5095],
        [  6.2630,   2.1057,  -5.2136],
        [ 11.2939,   6.6325,  -5.8311],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.3589,   0.8239,  -8.4230],
        [ 10.1503,   8.5414,  -7.2681],
        [ -1.8367,   7.8718,  -8.6523],
        [ -8.0588,  -6.0341,  -2.3006],
        [  9.3624,   9.5318,   0.2779],
        [  6.3589,   0.8239,  -8.4230],
        [ -0.8569,   7.9156,  -6.5557],
        [  3.3744,   6.5623,  -6.3636],
        [  3.1064,  13.5607,   6.7605],
        [  5.6258,  -1.9404,  -6.0715],
        [  0.6952,  -4.8060,   8.8976],
        [  0.1041, -11.0045,   2.4311],
        [  0.1931,  -4.7491,   7.4385],
        [  5.6095,  -1.9702,  -7.8233],
        [ -7.6525,  -6.6534,   5.1721],
        [ -7.6734,   4.4980,  -5.4248]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0294, 0.0361,  ..., 0.9682, 0.6685, 0.2681],
        [0.0294, 0.0000, 0.0333,  ..., 0.9949, 0.6786, 0.2914],
        [0.0361, 0.0333, 0.0000,  ..., 0.9980, 0.6453, 0.3022],
        ...,
        [0.9682, 0.9949, 0.9980,  ..., 0.0000, 1.2348, 0.7118],
        [0.6685, 0.6786, 0.6453,  ..., 1.2348, 0.0000, 0.8036],
        [0.2681, 0.2914, 0.3022,  ..., 0.7118, 0.8036, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.5642993 ,  9.71392179,  9.69629   ,  9.75401449,  9.78147542,
        9.79039371,  9.09963894, 12.38881635,  6.36004393, 10.80728788,
        7.9337647 ,  7.83486943, 13.20597932, 10.27716434,  4.91038884,
        6.02779201,  8.34949383, 10.48840476, 11.51557166,  7.72291946,
        8.89301516, 11.73848065,  5.0209005 , 12.91143491,  9.55720651,
       10.46739073, 13.35927527, 10.1979669 , 10.35464916, 12.5085995 ,
        6.33517349,  9.36256096,  7.96181288, 13.31764181,  5.05989547,
        5.91408504, 11.89855267, 13.40850328, 13.95472331, 13.83139024,
       11.9120321 , 14.0393949 ,  9.9353921 ,  4.98461983,  9.71392179,
        5.69913817,  7.78980299,  7.4544866 ,  7.23276582, 12.86648771,
       10.84493732, 11.00812996,  6.59176018,  5.91677188,  8.89301516,
       11.85890664,  5.13182506,  5.83560009,  5.24461068,  7.54127425,
       13.40024224,  7.46248947, 10.32715715,  4.978704  , 11.63502679,
       10.75443106,  9.51074763,  6.67121748,  6.42689705,  7.48311002,
        8.36445758,  8.51928531, 13.13888841,  9.41605684, 13.26209545,
        8.43626114, 10.67132524,  5.92281142,  5.92281142, 12.27991991,
        8.87007495, 12.76408562,  9.15866607, 11.80092578, 12.35625694,
        4.85603886,  7.73151309,  6.60752373, 13.09737053,  9.87720102,
        6.41202986, 13.26590071,  8.08320235, 10.0674887 , 13.3607379 ,
        6.41202986,  7.96181288,  7.37902766, 13.91198191,  5.9510489 ,
        4.85603886, 11.0050379 ,  4.75303127,  5.94541803, 10.14044489,
        8.89458273])}
{'P': tensor([[1.0860e-02, 2.0340e-01, 1.9429e-01,  ..., 6.5951e-05, 8.4151e-06,
         1.1726e-06],
        [1.1027e-04, 9.6450e-03, 1.3151e-04,  ..., 8.6962e-05, 3.8596e-06,
         1.5791e-06],
        [1.2042e-04, 2.9592e-01, 6.4948e-03,  ..., 4.8576e-05, 3.1213e-06,
         9.2203e-07],
        ...,
        [1.2595e-06, 1.3864e-06, 1.2928e-06,  ..., 3.0521e-02, 3.0778e-06,
         3.6502e-04],
        [5.2876e-03, 2.7057e-03, 3.1285e-03,  ..., 4.2840e-05, 4.8651e-02,
         1.1496e-06],
        [2.6427e-06, 1.7945e-06, 2.0351e-06,  ..., 7.2812e-07, 1.9683e-05,
         3.7823e-02]]), 'X': tensor([[ -9.4645,  -1.4880,  -2.0226],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 12.2592,   3.9425,   2.0383],
        [  5.5294,  -2.1106,   9.0626],
        [  6.3549,   0.7096,   6.9198],
        [  2.6614,  -4.5192,  -9.3641],
        [ -2.0445,   7.8469,  -0.5857],
        [ -9.1960,  -3.2964,   1.6921],
        [ -9.3604,   0.4855,  -3.0158],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.6426,  -4.9136,   9.4941],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.0574,   1.7801,   8.8986],
        [ 12.0845,  -2.5772,  -4.6073],
        [  5.6672,  -1.8631,   8.1020],
        [  5.6314,  -1.9302,  -9.6543],
        [ -2.1533, -10.5778,   1.0112],
        [  0.9788,  -4.8145,  -3.8661],
        [  8.5716,  10.3596,   6.8818],
        [ -6.1914,  -8.2997,   8.5380],
        [  2.9260,  -4.4162,   8.3808],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.5767,   5.5071,   3.9069],
        [  1.8766,   7.3694,   1.1887],
        [  1.2728, -11.0372,   9.8121],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.9883,   3.1536,   6.2320],
        [  5.0617,  12.7957,   8.3544],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.6059,   0.9597,  -6.3264],
        [ -0.1626, -10.9796,   6.4629],
        [ -7.8280,   4.2848,  -7.1955],
        [  4.5954,  13.0099,  -4.4439],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -5.9872,   6.2037,  -9.4477],
        [ -0.7599, -10.8995,  -6.0366],
        [ 12.1751,  -2.2592,  -5.7751],
        [  6.3200,   0.2690,   1.5900],
        [ 12.3453,  -1.5462,   7.0212],
        [ -2.6485, -10.4151,   0.8766],
        [  5.4867,  12.5816,   2.9549],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.3589,   0.8239,  -8.4230],
        [  1.7232,  -4.7574,   3.1270],
        [  3.2647, -10.8066,  -0.3768],
        [  4.7156,  -3.1831,  -8.0224],
        [  7.0642,  11.6104,  -9.8316],
        [  9.3424,  -7.3685,  -3.6807],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.1459,   1.4780,  -4.2403],
        [ -2.5161,   7.7699,  -9.3531],
        [ -1.0482, -10.8487,  -5.9066],
        [ -1.0012, -10.8576,   2.2867],
        [  6.0788,  -0.8710,  -6.8719],
        [  0.9478,  -4.8144,   0.6371],
        [  5.8930,  12.3592,  -8.9201],
        [  2.5620,   7.0549,   8.4825],
        [  2.0449,  13.8402,  -7.1642],
        [  5.5101,  -2.1433,   9.5442],
        [ -9.2508,   1.0572,   2.9978],
        [  5.6523,   3.9443,  -2.9449],
        [  6.0073,   3.0986,   6.7937],
        [  7.0171,  11.6439,  -5.0178],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.2435,   7.8191,  -6.4709],
        [  0.5600, -11.0321,  -0.8660],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.8318,   3.5567,  -4.7692],
        [  1.7232,  -4.7574,   3.1270],
        [ -7.8126,   4.3066,   5.1791],
        [  6.3246,   1.6696,  -0.1544],
        [  1.6422,   7.4586,  -6.8056],
        [  5.8323,  -1.5228,   5.1132],
        [  5.6523,   3.9443,  -2.9449],
        [  6.3505,   0.6235,  -8.5849],
        [ -1.0012, -10.8576,   2.2867],
        [  6.3549,   0.7096,   6.9198],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.9026,   3.3837,  -0.6573],
        [  5.4179,  -2.2925,  -3.4641],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -0.7599, -10.8995,  -6.0366],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.0073,   3.0986,   6.7937],
        [ 12.2592,   3.9425,   2.0383],
        [  5.4732,  -2.2041,  -4.3830],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 12.2921,  -1.7902,  -4.0870],
        [  1.7301,   7.4262,   5.0522],
        [  6.2365,  -0.2578,  -6.0212],
        [  1.7500,   7.4187,   4.4237],
        [  0.3715,  -4.7756,   0.2312],
        [  4.0587, -10.6112,   6.2273],
        [  7.0358,  -9.2784,  -9.8479],
        [  3.1756, -10.8247,   6.8882],
        [ -6.3306,   5.9272,   5.6935],
        [ 12.6038,   0.7665,  -3.4439]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0722, 0.0661,  ..., 0.8660, 0.7349, 1.0695],
        [0.0722, 0.0000, 0.0079,  ..., 0.9373, 0.8071, 1.1068],
        [0.0661, 0.0079, 0.0000,  ..., 0.9319, 0.8010, 1.0994],
        ...,
        [0.8660, 0.9373, 0.9319,  ..., 0.0000, 0.1651, 1.1292],
        [0.7349, 0.8071, 0.8010,  ..., 0.1651, 0.0000, 0.9885],
        [1.0695, 1.1068, 1.0994,  ..., 1.1292, 0.9885, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.58072286,  9.75401449,  9.69629   ,  9.78147542,  9.86089579,
        9.79039371, 12.87752052,  5.91854397,  6.39438044,  5.24461068,
        8.10886021,  9.76896904,  9.37296102,  6.05113217,  9.94173708,
       12.77402351,  9.60080639,  9.23071698, 12.35625694,  5.96556248,
        5.95297459, 10.794755  ,  4.91296205, 13.44593519, 10.35464916,
        5.29755487, 10.01804223,  7.16058587,  7.60463349, 11.11038551,
       10.52357479,  6.76791483, 13.76047999,  5.91677188, 12.64235359,
       10.98076873,  8.92395381, 13.79762415, 12.05300943,  8.62162372,
       10.92597196, 12.3828957 ,  6.32572323, 12.44177641, 10.74655346,
       13.72594022, 12.00995898,  6.41202986,  5.05989547, 11.28895489,
        5.68942572, 13.59056613, 11.89855267,  9.0368458 ,  9.26455795,
        8.16714836, 10.8992524 , 10.9036226 ,  6.14086582,  4.90676242,
       13.6922425 ,  7.50564223, 13.99048094,  5.91222346,  9.3109884 ,
        6.89245352,  6.7593917 , 13.59480906, 12.91143491, 12.42554808,
        7.822854  , 11.04629471,  4.91296205,  5.63824971,  6.83083863,
        5.05989547,  8.92098752,  6.54127431,  7.63726134,  6.02779201,
        6.89245352,  6.38105015, 10.9036226 ,  6.39438044, 12.4013153 ,
       12.77402351,  6.80370254,  5.8828986 ,  9.66977802, 10.92597196,
       10.52357479,  6.7593917 , 12.87752052,  5.90035324, 13.2430377 ,
        9.4997987 , 12.4217523 ,  7.62509582,  6.24187044,  7.62232307,
        4.79003361, 11.36089355, 11.64436684, 11.28093507,  8.67227386,
       12.62711252])}
{'P': tensor([[8.8199e-03, 3.1839e-01, 2.3581e-01,  ..., 3.2293e-06, 5.1858e-04,
         8.1167e-02],
        [1.4352e-04, 1.1749e-02, 1.7840e-01,  ..., 5.8259e-06, 8.6688e-04,
         1.2318e-01],
        [1.3921e-04, 4.9098e-04, 1.3344e-02,  ..., 7.7104e-06, 1.0930e-03,
         6.4476e-02],
        ...,
        [4.8283e-07, 4.1674e-07, 3.8484e-07,  ..., 1.0660e-02, 1.4989e-05,
         3.3523e-07],
        [8.0887e-05, 8.6607e-05, 8.9939e-05,  ..., 4.2974e-03, 5.0351e-02,
         1.0353e-04],
        [1.9177e-04, 3.5500e-04, 8.0862e-04,  ..., 2.7130e-05, 3.0537e-03,
         2.7663e-02]]), 'X': tensor([[ -9.4630,  -1.5145,   0.2983],
        [ -9.3616,  -2.4749,   0.1793],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -8.3539,  -5.5111,   0.3373],
        [ 10.8218,  -5.4177,   6.8573],
        [  5.0348,  -2.8229,   3.5436],
        [ -8.8154,   2.4611,   2.5198],
        [ -6.1914,  -8.2997,   8.5380],
        [ -6.2200,  -8.2737,   5.5079],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.2905,   1.9366,   2.1169],
        [  0.9036,   7.6840,   9.9237],
        [  5.6093,  -1.9705,   6.4807],
        [  0.6363, -11.0348,   6.1414],
        [  0.1041, -11.0045,   2.4311],
        [  6.5428,  -9.5728,  -2.9897],
        [  4.4049,   5.6892,  -4.4394],
        [  5.9883,   3.1536,   6.2320],
        [ 11.3691,   6.4760,   1.0787],
        [  5.1196,  -2.7160,   4.2705],
        [ -9.2707,   0.9661,   4.4863],
        [  3.6154,  -4.0640,  -7.9698],
        [  3.2809,  13.5061,  -4.0199],
        [  2.5636, -10.9293,   9.6519],
        [ -0.4341, -10.9473,   7.4509],
        [  5.2939,  -2.4784,  -4.5840],
        [  3.2226,  -4.2802,  -4.0428],
        [ -8.9636,   2.0648,   8.1946],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.1433,  -3.5003,  -3.0209],
        [  7.0642,  11.6104,  -9.8316],
        [  2.5469,  13.7192,  -9.6935],
        [ 12.3357,   3.6001,   0.7903],
        [ -4.1296,   7.2803,  -2.7194],
        [ -1.2454,   7.9127,  -0.5715],
        [  5.3804,  -2.3504,   7.2744],
        [  0.2732,  -4.7619,  -1.1577],
        [  4.9152,  12.8653,  -1.4459],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.8407,  11.7666,   3.0045],
        [ -2.3317,   7.8034,  -3.6274],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.9470,  13.6085,   2.9809],
        [ -2.0840,   7.8415,  -7.8657],
        [ 12.5832,   1.7543,  -7.6134],
        [  7.5309,  11.2617,   7.2656],
        [  5.6202,  -1.9508,   0.7471],
        [ -1.9646,   7.8571,   3.3817],
        [ -1.6086,   7.8928,   4.3385],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.5954,  13.0099,  -4.4439],
        [  0.7145,  -4.8071,  -2.3766],
        [ -5.1974,   6.7396,   3.3484],
        [ 11.9641,  -2.9568,   7.7211],
        [  9.0129,  -7.7072,   9.1400],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.4630,  -1.5145,   0.2983],
        [  8.1043,  -8.5164,  -7.9290],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.3516,   0.6425,   0.3425],
        [ 11.2517,  -4.6495,   2.9800],
        [  6.2022,   2.4092,   3.0349],
        [ -5.6169,  -8.7844,   8.9057],
        [  2.6310, -10.9195,   3.8054],
        [  5.0277,   4.9589,   7.0175],
        [  7.5064,  -8.9651,  -0.0994],
        [  0.1382, -11.0073,   1.0026],
        [  6.1741,  -0.5329,   3.1913],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.6093,  -1.9705,   6.4807],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -8.6426,  -4.9136,   9.4941],
        [  8.7715,  -7.9390,   5.2900],
        [  1.0093,   7.6567,   2.3526],
        [  0.3354,  -4.7708,  -4.9125],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.7558,  -4.4844,   0.0636],
        [  9.2692,   9.6370,   7.0746],
        [  2.6970,   6.9826,   9.7350],
        [  3.7121,   6.3117,   0.6738],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.1667,  -2.6543,   1.2879],
        [ -8.6112,   2.9319,  -7.9943],
        [ -7.6525,  -6.6534,   5.1721],
        [ -4.2271,   7.2387,  -1.8316],
        [  4.2612, -10.5514,  -6.6268],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.9036,   7.6840,   9.9237],
        [  6.3122,  -9.6994,  -6.3282],
        [ 11.0373,   7.1299,  -8.5895],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.9461,  -4.4077,  -0.4831],
        [ -2.8162,   7.7059,  -9.6885],
        [ 11.6165,  -3.8715,   6.4594],
        [ -0.4341, -10.9473,   7.4509],
        [ -0.0254, -10.9932,   3.2200],
        [  6.2883,   0.0325,   5.2692],
        [ 11.2517,  -4.6495,   2.9800],
        [  4.7544,  -3.1425,  -1.5315],
        [ 11.6801,  -3.7196,  -0.9181],
        [  1.0391,  -4.8141,  -0.0768],
        [ -8.8631,  -4.3728,  -1.1746]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 2.7048e-02, 3.0367e-02,  ..., 8.8828e-01, 7.9885e-01,
         1.5752e-01],
        [2.7048e-02, 0.0000e+00, 3.0432e-02,  ..., 9.0793e-01, 8.1995e-01,
         1.3689e-01],
        [3.0367e-02, 3.0432e-02, 0.0000e+00,  ..., 9.1693e-01, 8.2660e-01,
         1.6587e-01],
        ...,
        [8.8828e-01, 9.0793e-01, 9.1693e-01,  ..., 0.0000e+00, 1.1299e-01,
         9.2605e-01],
        [7.9885e-01, 8.1995e-01, 8.2660e-01,  ..., 1.1299e-01, 0.0000e+00,
         8.5000e-01],
        [1.5752e-01, 1.3689e-01, 1.6587e-01,  ..., 9.2605e-01, 8.5000e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.58347456,  9.68323686,  9.73579738,  9.95552626,  9.97964013,
       10.00794468, 12.10222204,  5.77219077,  9.15253344, 10.35464916,
       10.35093485,  9.5014146 ,  6.58184474,  7.73692436,  5.94535529,
       11.05317597, 11.0050379 , 11.59512194,  7.19513806,  6.76791483,
       13.08414689,  5.79543038,  9.32094055,  5.43943791, 13.89886469,
       11.22596994, 10.95593729,  5.84533345,  5.35774573,  9.19837792,
        8.06643487, 12.4013153 ,  9.79039371, 13.59056613, 13.95361098,
       12.85032836,  8.36995953,  8.01009083,  5.87132718,  4.76969364,
       13.77222752,  5.48276038, 13.6105595 ,  8.14433542, 12.95702647,
       13.92390689,  8.1137452 , 12.70489563, 13.54774355,  5.94909709,
        8.09899874,  8.0550286 , 12.22276278, 13.79762415,  4.85995002,
        8.51088913, 12.32408516, 11.85890664,  4.78002665,  9.58347456,
       11.75618513, 12.3344536 ,  9.55593374,  6.38399789, 12.17451685,
        6.65368551, 10.42667093, 11.23201619,  7.06169391, 11.69264243,
       11.00812996,  6.19707944,  5.21280156,  5.94535529, 10.23946247,
        9.94173708, 11.83075005,  7.72291946,  4.78258092,  5.34250225,
        5.26342203, 13.37121923,  7.48538905,  7.32233733, 10.61047596,
        5.80860165,  9.09661212, 10.14044489,  8.38252937, 11.37939695,
        9.36093562,  7.73692436, 11.57249984, 13.1399181 , 10.92597196,
        5.30158497,  8.20436764, 12.24467041, 10.95593729, 10.99326577,
        6.28835101, 12.17451685,  5.69913817, 12.25806559,  4.92497514,
        9.88310614])}
{'P': tensor([[1.3622e-02, 2.0917e-01, 1.7225e-01,  ..., 2.5280e-05, 1.1085e-04,
         6.7145e-05],
        [3.4828e-04, 7.1823e-02, 1.1679e-03,  ..., 4.6181e-04, 4.0173e-04,
         8.5570e-05],
        [1.3933e-04, 1.3251e-01, 9.3594e-03,  ..., 3.1832e-05, 1.0939e-04,
         2.5433e-05],
        ...,
        [4.8585e-06, 4.5641e-06, 4.4453e-06,  ..., 1.1644e-01, 1.0542e-06,
         3.1114e-06],
        [1.5375e-05, 4.7261e-05, 3.9336e-05,  ..., 2.5764e-05, 3.1945e-02,
         2.3196e-05],
        [5.6033e-03, 7.5688e-03, 1.2658e-02,  ..., 1.6267e-05, 2.9471e-02,
         7.6168e-02]]), 'X': tensor([[ -9.4524,  -1.6813,  -6.6862],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.7685, -10.8982,  -6.8928],
        [  4.5340,  -3.3629,   8.9652],
        [  6.9653,  -9.3225,  -3.4727],
        [ -9.3616,  -2.4749,   0.1793],
        [  9.2692,   9.6370,   7.0746],
        [  2.1625,  -4.6697,   7.0658],
        [  6.2026,  12.1774,  -1.5887],
        [  2.6970,   6.9826,   9.7350],
        [  0.9659,  -4.8144,  -0.9594],
        [  9.2979,   9.6048,   7.3138],
        [ -8.0611,  -6.0302,   4.6962],
        [ -8.6603,   2.8250,   5.9181],
        [  9.3624,   9.5318,   0.2779],
        [  0.9659,  -4.8144,  -0.9594],
        [ 12.4592,  -0.9168,  -6.7571],
        [  4.0587, -10.6112,   6.2273],
        [  1.0391,  -4.8141,  -0.0768],
        [ -8.6426,  -4.9136,   9.4941],
        [ -4.7334,   6.9982,  -0.4392],
        [ -9.0604,   1.7707,   1.1066],
        [  5.1886,  -2.6250,   6.4677],
        [ -7.5401,   4.6718,   9.0033],
        [ -5.3179,  -9.0101,   9.4472],
        [  8.4691,  10.4571,  -5.0650],
        [ -8.1426,   3.8048,   9.3542],
        [ -1.5408,   7.8978,   5.3470],
        [  3.5284, -10.7484,   5.3347],
        [  7.1441,  11.5529,   2.3232],
        [  0.6470, -11.0352,  -1.1931],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.8093,  -9.4178,   8.4244],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.8154,   2.4611,   2.5198],
        [  0.9036,   7.6840,   9.9237],
        [  4.0642,  -3.7599,  -3.4565],
        [ 10.8670,  -5.3428,  -2.8637],
        [  3.4920,  -4.1366,  -8.3955],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.1472,  -2.6801,   8.1194],
        [ 11.6019,   5.9540,  -5.1715],
        [ -3.3116, -10.1539,   7.5919],
        [  5.6093,  -1.9705,   6.4807],
        [  2.1120,  -4.6818,   8.6144],
        [ 11.2939,   6.6325,  -5.8311],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.4877,  -0.7222,   8.8347],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.9942,  -3.9990,   6.0448],
        [ -5.0132,  -9.2234,  -8.3531],
        [  9.4925,   9.3811,   8.9842],
        [  6.9653,  -9.3225,  -3.4727],
        [ -0.0282,   7.8575,  -1.1746],
        [ -1.2454,   7.9127,  -0.5715],
        [ -3.3812, -10.1235,   7.4384],
        [ 11.0438,   7.1180,  -5.6991],
        [  5.6022,  -1.9833,   8.3081],
        [  5.6523,   3.9443,  -2.9449],
        [  9.3753,   9.5171,   3.9131],
        [  6.1984,  -0.4327,   1.5438],
        [  4.0362,  13.2397,  -7.0030],
        [  7.7794,  11.0629,   8.9069],
        [ -4.8751,  -9.3148,   4.0406],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.8363,   2.4086,  -5.6296],
        [  0.9074,  -4.8139,   8.1674],
        [  4.6063,  13.0051,  -3.1456],
        [  5.1519, -10.2387,   9.8191],
        [ -9.4716,  -0.6673,   7.6068],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.2480, -11.0153,  -3.0183],
        [  0.8700,  -4.8132,  -7.8042],
        [ -8.6112,   2.9319,  -7.9943],
        [  2.9410,   6.8428,   5.4458],
        [  0.8572,  14.0501,  -8.3364],
        [  3.2701,   6.6338,   9.9209],
        [ -8.7409,  -4.6837,  -6.5840],
        [  9.3753,   9.5171,   3.9131],
        [ 12.3759,  -1.3943,   6.4659],
        [ -8.9282,   2.1648,   8.4894],
        [  0.6396,  14.0772,  -6.3786],
        [ 11.9641,  -2.9568,   7.7211],
        [ -5.0132,  -9.2234,  -8.3531],
        [  4.3523, -10.5232,   8.2138],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.9548,  -1.2273,  -2.6520],
        [  2.7558,  -4.4844,   0.0636],
        [ 11.4743,   6.2478,  -2.1100],
        [  2.8602,   6.8905,   5.8779],
        [ -1.5408,   7.8978,   5.3470],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.0330,  -6.0765,   6.7945],
        [ -3.3812, -10.1235,   7.4384],
        [ 11.9936,  -2.8677,   6.0614],
        [  1.9503,  -4.7172,  -6.2313],
        [ 12.2592,   3.9425,   2.0383],
        [  9.0424,  -7.6779,  -7.5286],
        [ -7.6719,  -6.6260,   7.1468],
        [ -9.1178,   1.5784,   2.1410]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2113e-02, 4.0685e-02,  ..., 8.3484e-01, 9.8798e-01,
         6.4548e-01],
        [1.2113e-02, 0.0000e+00, 3.7845e-02,  ..., 8.4675e-01, 9.8989e-01,
         6.4981e-01],
        [4.0685e-02, 3.7845e-02, 0.0000e+00,  ..., 8.5755e-01, 9.5325e-01,
         6.1608e-01],
        ...,
        [8.3484e-01, 8.4675e-01, 8.5755e-01,  ..., 0.0000e+00, 1.3155e+00,
         9.6553e-01],
        [9.8798e-01, 9.8989e-01, 9.5325e-01,  ..., 1.3155e+00, 0.0000e+00,
         3.7915e-01],
        [6.4548e-01, 6.4981e-01, 6.1608e-01,  ..., 9.6553e-01, 3.7915e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.60080639,  9.91668182,  9.7712321 ,  9.88571713,  9.96152788,
        9.85330646, 11.24434374,  5.64500512, 11.63724792,  9.68323686,
       13.37121923,  5.14607014, 13.6660671 ,  7.48538905,  4.91038884,
       13.36800073, 10.06703358,  9.10945678, 13.3607379 ,  4.91038884,
       12.49291768, 11.36089355,  4.92497514,  9.94173708,  8.4486771 ,
        9.23178204,  5.81481044,  8.87007495, 10.46238054, 13.45642379,
        8.98765644,  8.04664886, 11.31276109, 13.58333857, 11.05413545,
       10.07238463, 11.62158554, 11.81423161,  9.15253344,  7.73692436,
        5.5366698 , 12.10942412,  5.41348121,  9.5014146 ,  5.80312949,
       13.04051744, 10.68031351,  5.94535529,  5.13616224, 13.09737053,
        8.06643487, 12.5085995 ,  7.58668144,  9.84314712, 10.4977132 ,
       13.34587027, 11.63724792,  7.8575648 ,  8.01009083, 10.67322527,
       13.13888841,  5.94293185,  6.89245352, 13.35927527,  6.2134961 ,
       13.84126197, 13.52430176, 10.51340829,  8.1137452 ,  9.15866607,
        4.89869205, 13.79676654, 11.46176259,  9.49511524,  8.62162372,
        6.2134961 , 12.41012046, 11.018084  ,  4.89121805,  9.09661212,
        7.44805901, 14.07623266,  7.39599986,  9.91668182, 13.35927527,
       12.45418236,  9.18690018, 14.09176259, 12.32408516, 10.4977132 ,
       11.38774149,  9.26455795,  6.07992623,  5.26342203, 13.06498554,
        7.46053281,  8.04664886, 13.95361098, 10.07238463, 10.67322527,
       12.33167382,  5.10444097, 12.87752052, 11.86239961, 10.1371552 ,
        9.25336371])}
{'P': tensor([[1.0591e-02, 2.0360e-01, 1.7653e-01,  ..., 5.2896e-04, 3.2340e-05,
         3.8435e-07],
        [4.9421e-04, 1.2509e-02, 2.4589e-01,  ..., 4.8923e-04, 4.3435e-05,
         4.6373e-07],
        [1.0504e-04, 1.1232e-04, 1.7990e-02,  ..., 1.2800e-04, 1.7926e-04,
         9.5440e-07],
        ...,
        [9.7327e-03, 1.0147e-02, 2.3190e-02,  ..., 7.6057e-02, 5.2344e-05,
         2.2815e-06],
        [1.0142e-06, 1.0309e-06, 1.1453e-06,  ..., 4.7438e-07, 2.6576e-02,
         6.0357e-04],
        [3.8967e-06, 3.5378e-06, 1.5739e-06,  ..., 4.3445e-06, 3.7215e-07,
         2.2417e-02]]), 'X': tensor([[ -9.4505,  -1.7081,  -0.7039],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -8.5859,  -5.0393,   0.5476],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.5974,   7.0363,  -7.7810],
        [  9.3607,   9.5337,  -4.5412],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.1667,  -2.6543,   1.2879],
        [ 10.5678,   7.9261,  -0.6100],
        [  0.5600, -11.0321,  -0.8660],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.3587,   1.1596,  -3.6629],
        [ 12.2218,   4.0953,  -8.2072],
        [ -8.6229,   2.9068,   2.4429],
        [  6.1372,   2.6726,  -6.7885],
        [ -2.8162,   7.7059,  -9.6885],
        [ -9.3464,   0.5708,  -1.6397],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.3858,   4.4260,  -9.6769],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.3604,   0.4855,  -3.0158],
        [  0.1460,   7.8340,  -9.4614],
        [  4.9957,  -2.8705,   6.5978],
        [ -8.4297,  -5.3637,  -9.3962],
        [  8.3718,  -8.2953,   8.4666],
        [ -8.4824,  -5.2575,  -0.0795],
        [  1.8721,  -4.7322,   0.5524],
        [  5.2199,   4.6861,  -1.9304],
        [  6.0054,  -1.0902,   8.3079],
        [ 10.0567,   8.6697,   4.6329],
        [  7.0515,  11.6194,   8.4289],
        [  6.0073,   3.0986,   6.7937],
        [ -9.2508,   1.0572,   2.9978],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.3211,   1.7017,  -0.2164],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.5267,  -0.4125,  -8.7331],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.1751,  -2.2592,  -5.7751],
        [  4.9152,  12.8653,  -1.4459],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -2.8162,   7.7059,  -9.6885],
        [  3.8182,  -3.9347,  -1.7961],
        [ -5.1398,  -9.1367,   6.2074],
        [  0.9967,  -4.8144,  -9.7370],
        [  2.4356,  -4.5942,  -5.0808],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.1297,  -3.7098,  -3.9838],
        [  8.1217,  -8.5024,  -0.6149],
        [ -8.1666,   3.7653,   3.6075],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.3124,   0.2057,  -0.4593],
        [  0.6757, -11.0361,  -1.3047],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.0438,   7.1180,  -5.6991],
        [ -8.4297,  -5.3637,  -9.3962],
        [  3.3094,  13.4969,  -3.6507],
        [  7.1441,  11.5529,   2.3232],
        [ 10.9717,  -5.1646,  -6.9039],
        [  4.6063,  13.0051,  -3.1456],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.3549,   0.7096,   6.9198],
        [  6.1984,  -0.4327,   1.5438],
        [  4.4039,   5.6903,   8.7366],
        [  1.7301,   7.4262,   5.0522],
        [  6.2902,   0.0451,   3.7116],
        [ 12.4691,   2.8525,  -5.5110],
        [  3.6703,   6.3443,  -5.4401],
        [ -1.9714,   7.8563,  -1.8084],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.4267,  -5.3697,   2.4172],
        [ -5.7347,  -8.6908,   9.2878],
        [ 11.2571,   6.7072,   8.3693],
        [  0.5850,  -4.7982,   9.6639],
        [  0.1047, -11.0046,   0.5188],
        [  3.9913,  -3.8139,  -6.8311],
        [  3.1271, -10.8343,   2.6442],
        [  4.5503,   5.5359,   7.4735],
        [  2.4932,   7.0903,  -5.4214],
        [  3.5939,   6.4029,  -1.8357],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.2767,  -4.6006,   0.7868],
        [  0.6363, -11.0348,   6.1414],
        [ 11.7726,  -3.4876,  -6.7402],
        [  2.4572,   7.1085,   7.7376],
        [  0.9392,  -4.8143,  -3.4649],
        [  2.5636, -10.9293,   9.6519],
        [ -3.0008, -10.2827,  -8.9480],
        [  7.0642,  11.6104,  -9.8316],
        [  1.7301,   7.4262,   5.0522],
        [ -9.3419,   0.5972,  -1.2841],
        [ -9.4351,  -1.8891,   3.5642],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -8.8154,   2.4611,   2.5198],
        [  4.5767,   5.5071,   3.9069],
        [  7.0664,  -9.2590,   8.0879],
        [  1.6408,   7.4591,  -4.2294],
        [  4.5057,  -3.3894,  -0.8160],
        [  9.4516,  -7.2501,  -1.8868],
        [ -8.3539,  -5.5111,   0.3373],
        [ -9.4773,  -1.0024,   5.2599],
        [  7.2340,  -9.1505,  -5.8321],
        [ 11.3691,   6.4760,   1.0787]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.2764e-02, 6.8158e-02,  ..., 4.3406e-01, 7.5620e-01,
         1.1568e+00],
        [2.2764e-02, 0.0000e+00, 4.5735e-02,  ..., 4.5548e-01, 7.5448e-01,
         1.1699e+00],
        [6.8158e-02, 4.5735e-02, 0.0000e+00,  ..., 4.9509e-01, 7.6269e-01,
         1.2042e+00],
        ...,
        [4.3406e-01, 4.5548e-01, 4.9509e-01,  ..., 3.0339e-04, 1.0211e+00,
         1.1290e+00],
        [7.5620e-01, 7.5448e-01, 7.6269e-01,  ..., 1.0211e+00, 0.0000e+00,
         7.5482e-01],
        [1.1568e+00, 1.1699e+00, 1.2042e+00,  ..., 1.1290e+00, 7.5482e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.60359158,  9.62958973,  9.88310614,  9.97822473,  9.97964013,
        9.95552626, 11.9120321 ,  7.50035355, 13.36092624, 10.74409834,
        5.80860165, 13.20988996, 11.04629471,  8.30125201,  6.46356806,
       12.889695  ,  9.09963894,  6.69389431,  8.20436764,  9.36378438,
        8.1137452 ,  6.97106912,  5.53255882,  9.37296102,  7.83534987,
        5.76165934,  9.99145096, 11.785566  ,  9.97964013,  5.08910491,
        7.01475376,  6.10360073, 13.2778378 , 13.59170809,  6.7593917 ,
        9.3109884 , 12.889695  ,  6.54615887,  4.78544875, 12.53345445,
       11.05672136, 12.3828957 , 13.77222752, 12.12448912,  8.20436764,
        5.48276038, 10.48315436,  4.91652556,  5.19986797, 12.22276278,
        5.5513072 , 11.75807252,  8.99276564, 12.889695  ,  6.31576264,
       11.05672136, 13.0743399 , 13.13888841,  9.99145096, 13.89671314,
       13.58333857, 12.12641969, 13.79676654,  5.5205189 ,  6.39438044,
        6.2134961 ,  7.19534146,  7.62509582,  6.29034958, 12.79126785,
        7.32950385,  8.09984231,  8.01009083,  9.99211663, 10.412299  ,
       13.103709  ,  4.83371341, 11.00508681,  5.5205189 , 11.27656488,
        7.16598273,  7.5158489 ,  7.34250948,  6.81961996, 12.17901033,
       11.05317597, 12.27835996,  7.52117132,  4.90505566, 11.22596994,
       10.71162681, 13.59056613,  7.62509582,  9.36093562,  9.62238828,
       10.06437598,  9.15253344,  7.16058587, 11.64746882,  7.6374603 ,
        5.63824971, 11.9120321 , 10.00794468,  9.53015376, 11.66453059,
       13.08414689])}
{'P': tensor([[6.7507e-03, 1.7688e-01, 1.6530e-01,  ..., 1.1330e-01, 1.6463e-06,
         2.0302e-06],
        [1.1897e-04, 1.3646e-02, 2.5925e-03,  ..., 1.6102e-04, 2.9271e-06,
         4.0834e-06],
        [1.1665e-04, 5.3898e-02, 1.2088e-02,  ..., 1.7640e-04, 2.5469e-06,
         3.5325e-06],
        ...,
        [3.9938e-04, 2.2913e-01, 1.8458e-01,  ..., 9.6395e-03, 2.3663e-06,
         2.9878e-06],
        [2.2857e-04, 1.2247e-04, 1.2160e-04,  ..., 2.1527e-04, 3.2687e-02,
         5.3981e-04],
        [9.7302e-05, 5.6164e-05, 5.5843e-05,  ..., 9.1986e-05, 2.2400e-01,
         3.3855e-02]]), 'X': tensor([[ -9.4501,  -1.7133,   2.2980],
        [ -9.1624,  -3.4286,   2.0647],
        [ -9.1813,  -3.3550,   1.8827],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.9606,  -4.0994,   1.7886],
        [ -9.4203,  -2.0340,   2.5856],
        [  2.4572,   7.1085,   7.7376],
        [ -9.1749,  -3.3803,   6.4629],
        [  2.2182,  13.8007,   5.1889],
        [  0.7685,  -4.8098,  -2.9310],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.9445,   7.6736,   0.5095],
        [  0.1047, -11.0046,   0.5188],
        [  7.8375,  -8.7240,  -9.8913],
        [ 10.9875,   7.2207,  -8.8549],
        [  3.2366,  -4.2732,   5.1532],
        [ -7.8126,   4.3066,   5.1791],
        [  2.3589,  13.7668,  -3.6501],
        [  6.0310,  12.2795,  -6.4528],
        [  4.7482, -10.3907,   5.9149],
        [ -3.6381,   7.4683,   7.3544],
        [  7.4132,  11.3526,   0.3322],
        [  5.0190,   4.9705,  -7.9752],
        [ 11.7418,   5.6070,  -1.4192],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.4267,  -5.3697,   2.4172],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.3038,   7.8092,   0.4164],
        [  5.4429,  -2.2529,   9.7306],
        [  0.4996,   7.7738,  -4.4711],
        [  1.5383, -11.0274,  -4.9666],
        [ -8.4573,   3.2448,   5.4370],
        [  9.1820,   9.7334,  -2.7932],
        [ 12.4901,  -0.7052,   1.7315],
        [ -3.5903,   7.4847,  -8.9112],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.6525,  -6.6534,   5.1721],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.4297,  -5.3637,  -9.3962],
        [ 10.1503,   8.5414,  -7.2681],
        [  3.2226,  -4.2802,  -4.0428],
        [ -1.8823, -10.6558,  -3.8952],
        [  4.5767,   5.5071,   3.9069],
        [  5.6672,  -1.8631,   8.1020],
        [ -1.8788, -10.6567,   3.7068],
        [  6.0955,   2.8210,  -4.3321],
        [  6.9653,  -9.3225,  -3.4727],
        [  1.5653,  13.9377,   4.1773],
        [  7.9403,  -8.6455,  -8.7433],
        [  1.7218,  -4.7576,  -3.3617],
        [ -1.9600,   7.8577,   6.9121],
        [  0.9809,  -4.8145,   6.8706],
        [  5.8836,   3.4315,  -4.0327],
        [  1.9503,  -4.7172,  -6.2313],
        [ -2.1533, -10.5778,   1.0112],
        [ 11.6801,  -3.7196,  -0.9181],
        [  9.9750,  -6.6338,  -1.2440],
        [  0.1931,  -4.7491,   7.4385],
        [  5.3804,  -2.3504,   7.2744],
        [  3.6469,   6.3624,  -1.8414],
        [  2.2076,  13.8032,  -1.5099],
        [  3.3069,   6.6089,   8.7377],
        [  2.0243,   7.3086,   7.5432],
        [  4.2612, -10.5514,  -6.6268],
        [  4.3377,   5.7570,   1.3712],
        [  5.1472,  -2.6801,   8.1194],
        [  2.7558,  -4.4844,   0.0636],
        [  2.3261,  13.7749,   5.9121],
        [  1.8447,  -4.7372,   8.1235],
        [ -0.0254, -10.9932,   3.2200],
        [ 11.6634,   5.8050,   9.1590],
        [ -3.5029,   7.5138,   8.1853],
        [ -2.2188,   7.8218,   8.6064],
        [  5.1667,  -2.6543,   1.2879],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.3602,  -2.3808,   7.2098],
        [  3.3474,  -4.2162,   5.2455],
        [ -2.4956, -10.4682,   3.5632],
        [ -7.4733,   4.7555,   6.2490],
        [ -1.0012, -10.8576,   2.2867],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.4697,  -0.6157,  -6.8472],
        [  4.5057,  -3.3894,  -0.8160],
        [ -8.3653,   3.4184,   9.3454],
        [  3.6154,  -4.0640,  -7.9698],
        [ 11.2209,  -4.7092,   2.6117],
        [ -2.0721,   7.8432,   6.6380],
        [ 12.5224,  -0.4497,   4.8448],
        [ -3.9508,  -9.8515,   1.7923],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.4296,  -1.9458,   4.0833],
        [  6.2675,  12.1379,  -8.9720],
        [ -4.0409,   7.3169,  -3.5779],
        [  3.0876,   6.7527,  -5.3540],
        [ 12.3357,   3.6001,   0.7903],
        [ -9.2373,  -3.1225,  -8.9763],
        [  5.5158,  -2.1337,   3.5246],
        [  5.8318,   3.5567,  -4.7692],
        [  2.4149,  -4.6005,  -4.3339],
        [ -8.1905,   3.7253,   4.1175],
        [ -8.1204,  -5.9308,   4.0761],
        [ 11.7794,   5.5086,  -9.6929],
        [  2.4639, -10.9431,  -7.3718],
        [ -9.4203,  -2.0340,   2.5856],
        [  2.4572,   7.1085,   7.7376],
        [  4.5503,   5.5359,   7.4735]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0408, 0.0471,  ..., 0.0176, 0.8120, 0.8807],
        [0.0408, 0.0000, 0.0138,  ..., 0.0433, 0.8527, 0.9215],
        [0.0471, 0.0138, 0.0000,  ..., 0.0536, 0.8583, 0.9264],
        ...,
        [0.0176, 0.0433, 0.0536,  ..., 0.0000, 0.8119, 0.8816],
        [0.8120, 0.8527, 0.8583,  ..., 0.8119, 0.0000, 0.0866],
        [0.8807, 0.9215, 0.9264,  ..., 0.8816, 0.0866, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.60412735,  9.78285424,  9.7751189 ,  9.76896904,  9.85384623,
        9.63742669,  7.52117132,  9.77777346, 13.97779839,  4.87083348,
       12.0495521 ,  7.73151309, 11.00508681, 11.72749923, 13.14777433,
        5.36061974,  8.92098752, 13.96746816, 13.68063098, 11.42420508,
        8.30729352, 13.55868684,  7.06372172, 13.01187484,  8.95983586,
        9.99211663,  5.69913817,  7.81509631,  5.89074104,  7.78980299,
       11.1341798 ,  9.05843201, 13.38091121, 12.50996828,  8.30125201,
        8.16714836, 10.14044489,  4.91652556,  9.99145096, 13.26590071,
        5.35774573, 10.8207297 ,  7.16058587,  5.96556248, 10.82106505,
        6.7166297 , 11.63724792, 14.02532758, 11.73848065,  5.05963217,
        8.09842704,  4.91336994,  6.81117802,  5.10444097, 10.794755  ,
       12.25806559, 11.97949273,  4.75303127,  5.87132718,  7.33350642,
       13.97857591,  7.39004226,  7.58378655, 11.37939695,  7.20826219,
        5.80312949,  5.26342203, 13.96987806,  5.08372149, 10.99326577,
       13.02818702,  8.29022197,  8.1303945 ,  5.80860165, 13.03002338,
        5.86519525,  5.38342077, 10.76154256,  8.8580671 , 10.9036226 ,
       13.03002338,  9.48969978,  5.63824971,  9.0368458 ,  5.43943791,
       12.16901615,  8.11227668, 12.53047489, 10.61417181, 10.80728788,
        9.6282763 , 13.66051884,  8.35856487,  7.42512621, 12.85032836,
        9.75075273,  5.91408504,  6.83083863,  5.19577146,  8.99791316,
       10.05558893, 13.00380624, 11.21703654,  9.63742669,  7.52117132,
        7.16598273])}
{'P': tensor([[6.7215e-03, 2.5868e-01, 2.1173e-01,  ..., 4.9321e-08, 2.9668e-04,
         2.2091e-07],
        [9.0883e-05, 6.6242e-03, 3.0181e-01,  ..., 4.5205e-08, 3.5194e-04,
         2.3487e-07],
        [1.0482e-04, 1.3693e-04, 1.0042e-02,  ..., 6.6226e-08, 6.1127e-04,
         3.8188e-07],
        ...,
        [1.0091e-04, 7.2614e-05, 5.7669e-05,  ..., 2.4804e-01, 1.1713e-05,
         3.8903e-05],
        [1.1604e-04, 1.2100e-04, 1.2554e-04,  ..., 2.4707e-05, 8.6785e-02,
         7.4980e-04],
        [4.1179e-06, 3.1425e-06, 2.5910e-06,  ..., 6.5299e-03, 5.7666e-06,
         2.2605e-02]]), 'X': tensor([[ -9.4378,  -1.8606,  -2.5668],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -0.1626, -10.9796,   6.4629],
        [ -4.7334,   6.9982,  -0.4392],
        [ -3.1671,   7.6158,   6.0765],
        [  4.9957,  -2.8705,   6.5978],
        [ 10.4758,  -5.9539,   1.3667],
        [ -3.5029,   7.5138,   8.1853],
        [  2.0937,  13.8293,  -2.3533],
        [ -3.2053, -10.1993,  -3.8736],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.8617,   3.4853,  -9.8080],
        [  5.2608,  12.6977,   4.9166],
        [ 11.3691,   6.4760,   1.0787],
        [  0.1677,  -4.7447,  -6.1529],
        [  9.2979,   9.6048,   7.3138],
        [ -8.3062,  -5.6008,   0.0287],
        [  8.1217,  -8.5024,  -0.6149],
        [  6.2905,   1.9366,   2.1169],
        [  2.1477,   7.2550,   8.9688],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.7078,   2.7182,   1.8146],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 11.2939,   6.6325,  -5.8311],
        [  5.7784,  -1.6401,   9.5587],
        [  1.9241,   7.3503,   7.9004],
        [ -8.9636,   2.0648,   8.1946],
        [  4.9452, -10.3187,  -7.8621],
        [ -2.1382,   7.8339,   7.3524],
        [  4.7156,  -3.1831,  -8.0224],
        [ -5.3382,   6.6533,   7.8054],
        [  6.2804,   2.0019,   0.8607],
        [  2.5829,  -4.5466,  -8.4015],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.1984,  -0.4327,   1.5438],
        [  1.4749,  -4.7892,  -0.8089],
        [  1.7815,  -4.7481,   4.4289],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.9033,   2.2329,  -8.8912],
        [  3.2366,  -4.2732,   5.1532],
        [  9.3836,   9.5075,  -5.9278],
        [  3.2226,  -4.2802,  -4.0428],
        [ -9.4430,  -0.1897,   4.8383],
        [  1.6422,   7.4586,  -6.8056],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.0004,  -7.7195,   4.6945],
        [  8.9284,  10.0036,  -6.9210],
        [  6.3119,   1.7797,   9.9186],
        [  2.2849, -10.9655,  -7.6542],
        [  1.3670,  -4.7990,   3.2603],
        [ -6.6998,   5.5954,   0.0823],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.2692,  13.5098,  -4.9032],
        [ -0.6323,   7.9085,  -5.9544],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.7963,  -1.6018,   1.8856],
        [  6.2804,   2.0019,   0.8607],
        [  1.7232,  -4.7574,   3.1270],
        [  8.3791,  10.5409,   5.5590],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.2508,   1.0572,   2.9978],
        [  3.2647, -10.8066,  -0.3768],
        [  9.3424,  -7.3685,  -3.6807],
        [ -8.9178,   2.1936,   7.5388],
        [ -5.1369,   6.7755,   8.3836],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.8407,  11.7666,   3.0045],
        [ 11.6019,   5.9540,  -5.1715],
        [  7.7794,  11.0629,   8.9069],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.4196,   0.0478,   9.4140],
        [ -7.5401,   4.6718,   9.0033],
        [ -0.6323,   7.9085,  -5.9544],
        [  9.9217,  -6.7006,  -5.6728],
        [  2.5382,   7.0672,   4.5097],
        [  6.9284,  11.7061,  -0.3732],
        [  1.3767, -11.0341,   9.9415],
        [ -0.5739,   7.9056,   0.6382],
        [ -9.4735,  -1.2620,  -3.6742],
        [  8.1043,  -8.5164,  -7.9290],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.2589,  -1.9312,   9.8657],
        [ 11.7726,  -3.4876,  -6.7402],
        [  2.4697,  -4.5836,   1.4593],
        [ 10.1503,   8.5414,  -7.2681],
        [  4.0587, -10.6112,   6.2273],
        [  7.1441,  11.5529,   2.3232],
        [  6.3211,   1.7017,  -0.2164],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.8889,  -1.3916,   8.8993],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.1203,  -2.7151,   6.2110],
        [  6.3266,   0.3292,   9.3502],
        [  7.0515,  11.6194,   8.4289],
        [  5.0617,  12.7957,   8.3544],
        [  1.7122,  -4.7591,  -3.6734],
        [ 11.0864,   7.0387,  -0.1238]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0205, 0.0282,  ..., 1.6733, 0.8689, 1.2174],
        [0.0205, 0.0000, 0.0079,  ..., 1.6938, 0.8862, 1.2365],
        [0.0282, 0.0079, 0.0000,  ..., 1.7012, 0.8938, 1.2444],
        ...,
        [1.6733, 1.6938, 1.7012,  ..., 0.0000, 1.1266, 0.8098],
        [0.8689, 0.8862, 0.8938,  ..., 1.1266, 0.0000, 0.4010],
        [1.2174, 1.2365, 1.2444,  ..., 0.8098, 0.4010, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.61942445,  9.69629   ,  9.75401449,  9.78147542,  9.79039371,
        9.86089579, 10.98076873,  8.4486771 ,  8.24808832,  5.76165934,
       12.0495521 ,  8.29022197, 13.98691426, 10.6910768 ,  5.16077764,
        6.81961996, 13.744387  , 13.08414689,  4.74770899, 13.36800073,
       10.01804223, 11.75807252,  6.58184474,  7.56617037, 13.12299145,
        9.12220271,  8.09899874,  9.70108409, 13.09737053,  6.00663314,
        7.59795445,  9.19837792, 11.4424816 ,  8.12043983,  5.68942572,
        8.53014915,  6.59176018,  5.22903117,  4.78002665,  6.2134961 ,
        5.01114719,  5.0713364 ,  6.01356004,  9.17905491,  5.36061974,
       13.35832725,  5.35774573,  9.44486609,  7.63726134,  9.68324553,
       11.85743425, 13.40850328,  6.55800267, 11.20100795,  4.98989906,
        8.72896381,  9.77588801,  6.10774544, 13.89974693,  7.9337647 ,
        9.20340031,  6.01356004,  6.59176018,  5.05989547, 13.46553623,
        4.78002665,  9.3109884 , 11.28895489, 11.89855267,  9.18358562,
        8.50268322,  9.99211663, 13.6105595 , 13.04051744, 13.52430176,
       12.77402351,  9.26455795,  9.41970031,  8.87007495,  7.9337647 ,
       11.97238624,  7.50917551, 13.60275211, 11.11969815,  7.92644746,
        9.55720651, 11.75618513, 12.4217523 , 12.41012046, 12.27835996,
        5.20659892, 13.26590071, 11.36089355, 13.58333857,  6.54615887,
       11.9120321 ,  5.19986797,  6.05113217,  4.89121805,  9.62238828,
        5.79562691,  6.33517349, 13.59170809, 13.76047999,  5.05774295,
       13.13205357])}
{'P': tensor([[7.0130e-03, 2.4051e-01, 1.8856e-01,  ..., 4.9715e-04, 1.2471e-07,
         2.7297e-05],
        [1.8758e-04, 1.0907e-02, 4.0268e-01,  ..., 6.0485e-04, 2.0007e-07,
         4.3378e-05],
        [2.0219e-04, 2.4938e-04, 1.6583e-02,  ..., 6.9058e-04, 3.2051e-07,
         6.8328e-05],
        ...,
        [1.9771e-04, 2.4901e-04, 3.2541e-04,  ..., 4.5498e-02, 2.8217e-06,
         3.4147e-03],
        [2.1757e-06, 1.7354e-06, 1.3651e-06,  ..., 1.7706e-06, 2.9342e-02,
         9.0082e-06],
        [5.6831e-05, 5.5493e-05, 5.4933e-05,  ..., 2.6114e-04, 2.5303e-03,
         1.7905e-01]]), 'X': tensor([[ -9.4351,  -1.8891,   3.5642],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.1531,  -5.8748,   2.6934],
        [ -6.3306,   5.9272,   5.6935],
        [  9.2692,   9.6370,   7.0746],
        [  5.6672,  -1.8631,   8.1020],
        [ -7.3554,   4.8985,  -0.6665],
        [ 11.4703,  -4.2006,   7.4086],
        [ -1.8367,   7.8718,  -8.6523],
        [ -6.2668,   5.9808,   2.5614],
        [  0.9809,   7.6642,  -1.4217],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -1.0407,   7.9166,  -6.6021],
        [  1.7815,  -4.7481,   4.4289],
        [  8.7715,  -7.9390,   5.2900],
        [ -7.7842,   4.3466,  -1.7548],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -9.3419,   0.5972,  -1.2841],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.3860,  -2.3418,   1.0826],
        [  9.2979,   9.6048,   7.3138],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.0829,  -5.9940,   5.0564],
        [ -1.5451,   7.8975,   7.6104],
        [  2.1477,   7.2550,   8.9688],
        [  6.1685,  12.1979,  -6.2473],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.3943,   6.5483,   1.0790],
        [  2.3261,  13.7749,   5.9121],
        [  1.2728, -11.0372,   9.8121],
        [  0.2571,  -4.7594,   1.5618],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -3.2736, -10.1703,   3.8712],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -7.9298,   4.1367,  -3.9773],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.1890,   4.7318,   0.9515],
        [  1.7000,  -4.7609,   2.7935],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.8407,  11.7666,   3.0045],
        [  5.1890,   4.7318,   0.9515],
        [  3.0719, -10.8449,  -4.1673],
        [  2.3589,  13.7668,  -3.6501],
        [ -8.5900,   2.9767,   7.4811],
        [  0.9392,  -4.8143,  -3.4649],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.5506,  -5.1154,   6.3423],
        [ -0.0410,   7.8591,   9.3457],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.2687,  -2.9794,   0.6436],
        [ -9.1433,  -3.5003,  -3.0209],
        [  2.6970,   6.9826,   9.7350],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.9945,  -4.8144,  -0.1897],
        [  3.2692,  13.5098,  -4.9032],
        [  0.9478,  -4.8144,   0.6371],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.3490,  -4.6199,  -4.2489],
        [ 12.2218,   4.0953,  -8.2072],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.1685,  12.1979,  -6.2473],
        [ -5.6423,   6.4537,  -1.2091],
        [  5.4732,  -2.2041,  -4.3830],
        [  1.8766,   7.3694,   1.1887],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.7802,  -1.6363,  -6.2782],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.3211,   1.7017,  -0.2164],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.9809,  -4.8145,   6.8706],
        [  6.7792,  11.8084,   1.6251],
        [  2.3373,  -4.6232,  -5.2246],
        [ -3.0008, -10.2827,  -8.9480],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.1519, -10.2387,   9.8191],
        [  6.0305,   3.0296,   0.1375],
        [  9.3560,   9.5391,   9.4943],
        [  3.3094,  13.4969,  -3.6507],
        [  6.9314,  -9.3435,  -1.2729],
        [  4.5954,  13.0099,  -4.4439],
        [  3.3633,  13.4794,   0.5193],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.9942,  -3.9990,   6.0448],
        [  0.9809,   7.6642,  -1.4217],
        [  6.3606,   0.9204,   3.2525],
        [  9.5018,  -7.1945,  -1.1892],
        [  7.0642,  11.6104,  -9.8316],
        [  2.2644,  -4.6434,   9.7508],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.3616,  -2.4750,   3.7765],
        [  6.3515,   0.6418,  -6.5823],
        [  5.8979,   3.3958,  -3.8713],
        [ 11.1469,   6.9237,   1.6160],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.2939,   6.6325,  -5.8311],
        [  2.0899,  -4.6870,  -7.4260]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0259, 0.0448,  ..., 0.6998, 1.3735, 1.0558],
        [0.0259, 0.0000, 0.0189,  ..., 0.7165, 1.3984, 1.0813],
        [0.0448, 0.0189, 0.0000,  ..., 0.7290, 1.4166, 1.0998],
        ...,
        [0.6998, 0.7165, 0.7290,  ..., 0.0000, 1.2696, 0.9518],
        [1.3735, 1.3984, 1.4166,  ..., 1.2696, 0.0000, 0.3331],
        [1.0558, 1.0813, 1.0998,  ..., 0.9518, 0.3331, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.62238828,  9.68324553,  9.75200848,  9.99211663, 10.05558893,
       10.04916678,  8.67227386, 13.37121923,  5.96556248,  8.83725081,
       12.21532282,  8.08320235,  8.66273171,  7.72668467, 12.25806559,
        7.98469166,  5.0713364 , 11.83075005,  8.91551231,  9.61942445,
        9.36093562, 12.97370371,  5.87305005, 13.36800073,  4.91336994,
       12.42554808, 10.06286074,  8.04718433,  7.56617037, 13.66896739,
       10.53368644,  7.37576154, 13.96987806, 11.11038551,  4.7663526 ,
       10.4977132 , 10.68416401, 12.12641969,  8.94395628, 12.40074901,
       11.59512194,  7.02253235,  5.05534609, 12.33167382, 13.6105595 ,
        7.02253235, 11.27160035, 13.96746816,  9.09119351,  4.90505566,
       10.0674887 , 10.30972193,  9.96391045,  7.85919687,  9.45995955,
        9.73579738,  9.79039371,  7.48538905, 10.46238054,  4.9160962 ,
       13.89974693,  4.90676242, 12.42554808,  4.76969364,  5.18277858,
       12.889695  , 11.51811603, 13.66896739,  8.5724042 ,  5.90035324,
        7.60463349,  9.77777346,  6.00732193,  5.27178363,  6.54615887,
        5.57341623,  4.91336994, 13.61600536,  5.18047638, 10.71162681,
        4.74468406, 11.46176259,  6.74872981, 13.3614598 , 13.89671314,
       11.63383221, 13.79762415, 13.89264822,  9.69792513,  9.84314712,
        7.72668467,  6.42689705, 11.9182956 , 13.59056613,  5.16611782,
        5.05981914,  9.79039371,  9.68324553,  6.383897  ,  6.80559669,
       13.12218013, 10.39942497, 12.44177641,  9.96152788, 13.09737053,
        5.13182506])}
{'P': tensor([[9.8216e-03, 3.3210e-01, 2.5143e-01,  ..., 2.7696e-06, 2.9520e-06,
         4.3291e-05],
        [1.5570e-04, 1.4569e-02, 2.1909e-04,  ..., 3.7975e-06, 6.2790e-06,
         8.2804e-05],
        [2.6018e-04, 4.1271e-01, 1.1179e-02,  ..., 2.9728e-06, 3.8169e-06,
         5.5202e-05],
        ...,
        [1.0921e-03, 7.0667e-04, 8.6968e-04,  ..., 1.2864e-01, 2.1491e-05,
         2.5454e-04],
        [6.9953e-07, 5.8450e-07, 6.2893e-07,  ..., 7.9888e-04, 1.9875e-02,
         4.2449e-04],
        [7.1691e-05, 6.3933e-05, 6.8076e-05,  ..., 6.2026e-03, 3.5843e-03,
         9.2253e-02]]), 'X': tensor([[ -9.4296,  -1.9458,   4.0833],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.3616,  -2.4750,   3.7765],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.0965,  -5.9712,   4.3466],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.3516,   0.6425,   0.3425],
        [  6.0560,  -0.9421,  -5.0334],
        [  0.1931,  -4.7491,   7.4385],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.2687,  -2.9794,   0.6436],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.3266,   0.3292,   9.3502],
        [  1.8721,  -4.7322,   0.5524],
        [  6.3550,   0.7129,   0.4917],
        [  5.0233,   4.9647,  -3.1788],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.8889,  -1.3916,   8.8993],
        [  8.9284,  10.0036,  -6.9210],
        [  6.7565,  11.8237,  -4.9882],
        [ -5.9872,   6.2037,  -9.4477],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.3474,  -4.2162,   5.2455],
        [  4.3377,   5.7570,   1.3712],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 12.0582,  -2.6640,   0.7132],
        [  8.1528,  10.7452,   9.1493],
        [  3.9663,   6.1018,  -5.4104],
        [  0.8940, -11.0402,  -0.9783],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.2939,   6.6325,  -5.8311],
        [  7.0171,  11.6439,  -5.0178],
        [  5.2911,  -2.4824,  -3.6877],
        [ -1.5261,   7.8988,   9.3768],
        [  1.4489,   7.5256,   4.4529],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3087,   1.8053,  -1.9481],
        [ -8.4297,  -5.3637,  -9.3962],
        [  2.3261,   7.1726,   6.2522],
        [ -9.4721,  -1.3097,  -7.4377],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.4766,  -1.1116,  -2.1706],
        [  1.7218,  -4.7576,  -3.3617],
        [  0.8700,  -4.8132,  -7.8042],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.7482, -10.3907,   5.9149],
        [ -8.1531,  -5.8748,   2.6934],
        [  5.8836,   3.4315,  -4.0327],
        [  9.5018,  -7.1945,  -1.1892],
        [  4.4049,   5.6892,  -4.4394],
        [  4.8051, -10.3703,   0.5268],
        [  5.6260,  -1.9401,   3.0259],
        [ -6.8466,  -7.6488,   1.9648],
        [ -0.5739,   7.9056,   0.6382],
        [  0.5175,  14.0910,   4.6540],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.3550,   0.7129,   0.4917],
        [  2.6614,  -4.5192,  -9.3641],
        [  5.2939,  -2.4784,  -4.5840],
        [ -4.7642,  -9.3860,  -5.1133],
        [  4.7739,  -3.1218,  -4.1188],
        [ -8.8835,  -4.3178,   9.3080],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.3384,  -2.6161,  -4.7630],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.8450,  -9.9356,   3.0767],
        [  9.9750,  -6.6338,  -1.2440],
        [  5.1861,   4.7361,  -3.7312],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.0438,   7.1180,  -5.6991],
        [  6.0140,  -1.0661,  -2.5040],
        [ 11.0275,  -5.0666,   8.4154],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.2038,   2.4020,  -3.1692],
        [  5.1886,  -2.6250,   6.4677],
        [  4.2275,  -3.6320,  -0.2714],
        [ 11.8951,  -3.1570,   7.0742],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.3317,   7.8034,  -3.6274],
        [  6.3604,   0.9042,  -6.5901],
        [  2.5375,   7.0676,  -4.3490],
        [  5.8325,  -1.5222,  -2.1341],
        [ 12.1379,   4.4125,  -3.8833],
        [  0.6396,  14.0772,  -6.3786],
        [ 12.2589,  -1.9312,   9.8657],
        [  3.5939,   6.4029,  -1.8357],
        [ 12.5224,  -0.4497,   4.8448],
        [ 10.2712,   8.3707,  -4.9743],
        [  6.3535,   0.6792,  -7.5101],
        [  3.3069,   6.6089,   8.7377],
        [  2.8602,   6.8905,   5.8779],
        [  3.0876,   6.7527,  -5.3540],
        [  6.3087,   1.8053,  -1.9481],
        [  4.9452, -10.3187,  -7.8621],
        [  2.0449,  13.8402,  -7.1642],
        [ -3.5520,   7.4976,   9.1522],
        [  5.5158,  -2.1337,   3.5246],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.6816,   9.1536,  -5.8543],
        [  7.0515,  11.6194,   8.4289],
        [ -3.4488, -10.0934,   2.5195],
        [  2.4572,   7.1085,   7.7376],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.5158,  -2.1337,   3.5246]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.8738e-02, 2.1831e-02,  ..., 7.6684e-01, 9.7839e-01,
         8.9491e-01],
        [2.8738e-02, 0.0000e+00, 1.8918e-02,  ..., 7.9448e-01, 1.0048e+00,
         9.2358e-01],
        [2.1831e-02, 1.8918e-02, 0.0000e+00,  ..., 7.8668e-01, 9.9920e-01,
         9.1040e-01],
        ...,
        [7.6684e-01, 7.9448e-01, 7.8668e-01,  ..., 0.0000e+00, 2.3812e-01,
         3.1874e-01],
        [9.7839e-01, 1.0048e+00, 9.9920e-01,  ..., 2.3812e-01, 3.0339e-04,
         4.5302e-01],
        [8.9491e-01, 9.2358e-01, 9.1040e-01,  ..., 3.1874e-01, 4.5302e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.6282763 ,  9.75200848,  9.68324553, 10.05558893, 10.06023665,
       10.06703358,  6.38399789,  6.12886024,  4.75303127,  9.41970031,
        9.73579738,  6.23262289,  6.33517349,  5.08910491,  6.3949043 ,
        7.06272256, 12.10942412,  6.05113217, 13.40850328, 13.61801526,
        8.62162372, 10.41909211,  5.38342077,  7.20826219,  8.95983586,
        9.4997987 , 12.3489324 , 13.48809595,  7.27759233, 11.07637531,
       13.20597932, 13.09737053, 13.59480906,  5.84450127,  8.04483495,
        7.66377554, 12.3489324 ,  6.5618968 ,  9.99145096,  7.5403822 ,
        9.562174  ,  4.78002665,  9.54154096,  5.05963217,  4.89121805,
        5.8828986 , 11.42420508, 10.04916678,  6.81117802, 11.9182956 ,
        7.19513806, 11.42947632,  5.95111804, 10.26545792,  7.92644746,
       14.10045797,  7.88677538,  6.02789634,  6.3949043 ,  5.24461068,
        5.84533345, 10.52587243,  5.70405525,  9.87720102,  8.92098752,
        9.69792513,  5.18047638, 11.52733581, 11.97949273,  7.02326515,
        5.35774573, 13.13888841,  6.10774544, 12.13568598, 12.18045978,
        6.65259775,  5.81481044,  5.57341623, 12.30695098,  9.34127013,
        8.14433542,  6.42439344,  7.5092834 ,  6.02789634, 12.91505111,
       14.09176259, 12.41012046,  7.34250948, 12.53047489, 13.25017211,
        6.38968357,  7.39004226,  7.46053281,  7.42512621,  6.5618968 ,
       11.4424816 , 13.99048094,  8.29640613,  5.91408504, 12.10222204,
       13.32374056, 13.59170809, 10.66631784,  7.52117132, 12.27993098,
        5.91408504])}
{'P': tensor([[1.2307e-02, 2.4193e-01, 1.6356e-01,  ..., 1.1661e-04, 1.2115e-05,
         1.7986e-07],
        [9.8715e-05, 1.5812e-02, 3.8517e-01,  ..., 2.5930e-04, 2.2274e-05,
         2.2990e-07],
        [2.0273e-04, 7.9692e-04, 4.8175e-02,  ..., 7.9684e-04, 8.9872e-05,
         6.9573e-07],
        ...,
        [6.1337e-05, 7.3010e-05, 7.2102e-05,  ..., 1.0744e-01, 2.2881e-04,
         2.8563e-04],
        [5.8720e-05, 4.1217e-05, 4.0961e-05,  ..., 6.7469e-05, 1.3098e-01,
         4.2012e-04],
        [1.0394e-05, 3.7674e-06, 2.4526e-06,  ..., 2.1733e-06, 1.0912e-06,
         1.3216e-02]]), 'X': tensor([[ -9.4283,  -1.9585,  -0.9129],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.2435,   7.8191,  -6.4709],
        [  4.3377,   5.7570,   1.3712],
        [  5.3804,  -2.3504,   7.2744],
        [  3.3069,   6.6089,   8.7377],
        [  6.0310,  12.2795,  -6.4528],
        [ 12.4996,   2.6335,  -6.3166],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -1.9600,   7.8577,   6.9121],
        [  5.8930,  12.3592,  -8.9201],
        [  5.8889,  -1.3916,   8.8993],
        [  4.5430,   5.5438,  -3.8669],
        [  0.3038,   7.8092,   0.4164],
        [  7.9141,  -8.6656,   9.6948],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.4159,  -2.0739,   9.6841],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.3459,  -3.5327,   0.3667],
        [  0.9945,  -4.8144,  -0.1897],
        [  4.9337,  12.8566,  -3.6582],
        [  3.3094,  13.4969,  -3.6507],
        [  6.3211,   1.7017,  -0.2164],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.1504,  -4.6726,   8.7908],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [ 10.2470,   8.4054,  -0.1177],
        [  8.4691,  10.4571,  -5.0650],
        [  5.2188,  -2.5839,   2.1833],
        [ 12.5832,   1.7543,  -7.6134],
        [ -3.5029,   7.5138,   8.1853],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.0608,   3.9362,  -5.9567],
        [  6.9433,  -9.3362,  -0.3421],
        [ -2.2508, -10.5479,  -2.7235],
        [  6.3360,   0.4258,  -9.2509],
        [  5.9193,  -1.3176,   5.1739],
        [ -5.9938,  -8.4745,   3.9735],
        [ 11.4235,   6.3594,   5.1663],
        [  7.5309,  11.2617,   7.2656],
        [  7.9816,  10.8938,  -0.5378],
        [ -6.2200,  -8.2737,   5.5079],
        [ -6.8618,  -7.6322,   3.7027],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.9695,  -2.9407,  -6.3498],
        [  0.6363, -11.0348,   6.1414],
        [  6.3246,   1.6696,  -0.1544],
        [ -1.9646,   7.8571,   3.3817],
        [  6.2555,   2.1474,   4.1685],
        [  5.5626,  -2.0536,   8.4433],
        [ 11.6019,   5.9540,  -5.1715],
        [  4.5503,   5.5359,   7.4735],
        [  5.4167, -10.1293,   6.2156],
        [ 10.1983,  -6.3430,  -0.3200],
        [  7.1441,  11.5529,   2.3232],
        [  0.1041, -11.0045,   2.4311],
        [  2.6321,  -4.5296,  -2.5159],
        [  3.9591, -10.6391,  -6.2557],
        [ -9.4671,  -0.5553,   0.1117],
        [ -9.2508,   1.0572,   2.9978],
        [  3.4009,  -4.1874,  -5.9751],
        [ -5.3382,   6.6533,   7.8054],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.7792,  11.8084,   1.6251],
        [  3.4920,  -4.1366,  -8.3955],
        [ -8.4824,  -5.2575,  -0.0795],
        [  0.9445,   7.6736,   0.5095],
        [  2.3200,   7.1756,   1.1938],
        [  1.6308,  13.9254,  -2.0984],
        [ -7.5720,   4.6310,   2.1717],
        [ 11.5568,   6.0602,   9.3244],
        [  0.2480, -11.0153,  -3.0183],
        [  0.4119,  14.1019,  -4.1870],
        [  9.3624,   9.5318,   0.2779],
        [ 12.1379,   4.4125,  -3.8833],
        [  3.5965,  -4.0754,  -3.5035],
        [ 10.2712,   8.3707,  -4.9743],
        [  5.4732,  -2.2041,  -4.3830],
        [ 11.6801,  -3.7196,  -0.9181],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.4541,  -0.3327,   8.8246],
        [  0.7275,  14.0667,  -0.9815],
        [ 11.6019,   5.9540,  -5.1715],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.7802,  -1.6363,  -6.2782],
        [ -5.9938,  -8.4745,   3.9735],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4716,  -0.6673,   7.6068],
        [ -5.3328,  -8.9992,   7.3440],
        [  6.2975,   0.0947,   5.1791],
        [  9.6662,   9.1725,   8.1056],
        [ -0.0282,   7.8575,  -1.1746],
        [ -1.9487,   7.8590,  -3.3330],
        [  7.7750,  11.0665,   2.8364],
        [ -4.8751,  -9.3148,   4.0406],
        [ -7.5603,   4.6459,   8.2002],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.1973,  -0.4372,   6.5720],
        [  6.1685,  12.1979,  -6.2473]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0457, 0.0197,  ..., 0.9462, 1.0952, 1.2514],
        [0.0457, 0.0000, 0.0391,  ..., 0.9616, 1.1377, 1.2677],
        [0.0197, 0.0391, 0.0000,  ..., 0.9654, 1.1112, 1.2707],
        ...,
        [0.9462, 0.9616, 0.9654,  ..., 0.0000, 0.8652, 0.3066],
        [1.0952, 1.1377, 1.1112,  ..., 0.8652, 0.0000, 0.9634],
        [1.2514, 1.2677, 1.2707,  ..., 0.3066, 0.9634, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.62958973,  9.88310614,  9.97822473,  9.97964013, 10.01804223,
        9.95552626,  7.822854  ,  7.20826219,  5.87132718,  7.39004226,
       13.68063098, 12.77402351,  9.96152788,  8.09842704, 13.6922425 ,
        6.05113217,  7.16747878,  7.81509631, 11.73567759, 13.08414689,
        9.64157461,  5.01114719,  5.60064319,  4.9160962 , 13.77075174,
       13.89671314,  6.54615887,  6.02588517,  9.41244865,  9.58347456,
        5.14371005, 13.45642379, 13.59480906, 13.25336001, 13.45642379,
        5.82345867, 12.70489563,  8.29022197, 10.61472636,  8.97052115,
       11.63502679, 10.78533838,  6.35029274,  6.06416957, 10.37998078,
       13.0743399 , 13.54774355, 13.50484539, 10.35093485, 10.26327895,
       11.27160035, 12.32545637, 11.05317597,  6.54127431,  8.09899874,
        6.61386436,  5.92952538, 13.04051744,  7.16598273, 11.48663767,
       12.00995898, 13.58333857, 11.0050379 ,  5.23878371, 11.35182589,
        9.48337063,  9.3109884 ,  5.39450305,  8.53014915,  5.18277858,
       13.61600536,  5.41348121,  9.97964013,  7.73151309,  7.54127425,
       14.02058621,  8.87587457, 13.0493368 , 11.018084  , 14.10796466,
       13.3607379 , 12.91505111,  5.43544626, 13.25017211,  5.90035324,
       12.25806559,  5.0713364 ,  9.45995955, 14.08549802, 13.04051744,
       11.67875855,  6.00732193, 10.37998078,  6.00663314,  9.49511524,
       10.4606265 ,  6.29821673, 13.32557133,  7.8575648 ,  8.09704052,
       13.5247215 , 10.51340829,  8.8737556 ,  5.19065439,  6.21275016,
       13.66896739])}
{'P': tensor([[1.0849e-02, 2.5787e-01, 2.0774e-01,  ..., 8.8016e-06, 1.7613e-06,
         8.0792e-03],
        [1.9086e-04, 1.6175e-02, 3.0730e-03,  ..., 1.7758e-05, 2.2985e-06,
         1.0234e-03],
        [2.0110e-04, 6.1445e-02, 1.3780e-02,  ..., 1.5423e-05, 2.0367e-06,
         9.1934e-04],
        ...,
        [4.9682e-05, 4.3367e-05, 4.5074e-05,  ..., 9.9191e-02, 7.6481e-03,
         6.5123e-05],
        [8.2031e-04, 6.1591e-04, 6.6446e-04,  ..., 1.3237e-04, 8.7886e-02,
         1.4195e-03],
        [6.6272e-03, 1.0252e-01, 1.1471e-01,  ..., 4.8569e-05, 8.3196e-06,
         3.8679e-02]]), 'X': tensor([[ -9.4203,  -2.0340,   2.5856],
        [ -9.1624,  -3.4286,   2.0647],
        [ -9.1813,  -3.3550,   1.8827],
        [ -8.9606,  -4.0994,   1.7886],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.4267,  -5.3697,   2.4172],
        [ -3.2736, -10.1703,   3.8712],
        [  1.3670,  -4.7990,   3.2603],
        [  3.9895,   6.0816,  -6.5235],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.3515,   0.6418,  -6.5823],
        [  3.3633,  13.4794,   0.5193],
        [  3.5284, -10.7484,   5.3347],
        [  2.1771, -10.9776,  -3.2007],
        [ -0.0254, -10.9932,   3.2200],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.8540,  -4.3971,  -5.2144],
        [  7.0515,  11.6194,   8.4289],
        [  2.9461,  -4.4077,  -0.4831],
        [  5.0617,  12.7957,   8.3544],
        [ -0.9715, -10.8630,   0.3555],
        [ 12.6038,   0.7665,  -3.4439],
        [ 11.3691,   6.4760,   1.0787],
        [ -7.0825,   5.2073,  -5.0497],
        [ -1.3940, -10.7772,   6.7835],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -7.6734,   4.4980,  -5.4248],
        [  1.1006,   7.6318,  -2.4877],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.8602,   6.8905,   5.8779],
        [  6.3246,   1.6696,  -0.1544],
        [  9.6614,   9.1784,  -1.2087],
        [  0.1677,  -4.7447,  -6.1529],
        [  0.2435,   7.8191,  -6.4709],
        [  2.8474,   6.8979,   6.6457],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.1120,  -4.6818,   8.6144],
        [  1.1994,   7.6035,   2.4295],
        [  1.0093,   7.6567,   2.3526],
        [ -7.6734,   4.4980,  -5.4248],
        [  4.0591,   6.0202,  -1.6339],
        [ -3.0008, -10.2827,  -8.9480],
        [ 11.5456,   6.0861,   8.1101],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -9.0604,   1.7707,   1.1066],
        [  9.8082,  -6.8396,   4.9615],
        [  1.0093,   7.6567,   2.3526],
        [  0.6952,  -4.8060,   8.8976],
        [ -1.9600,   7.8577,   6.9121],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.0042,  -2.8350,   2.6864],
        [ -3.0008, -10.2827,  -8.9480],
        [  7.5930,  -8.9037,   7.2323],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.2911,  -2.4824,  -3.6877],
        [ -8.0087,   4.0173,   9.1091],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.4296,  -1.9458,   4.0833],
        [ 12.2692,  -1.8882,   3.9253],
        [ -9.4366,  -0.1189,  -2.6736],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.6634,   5.8050,   9.1590],
        [  6.2242,  -0.3171,   7.4978],
        [  9.5203,   9.3483,   5.8781],
        [  7.9816,  10.8938,  -0.5378],
        [  1.5383, -11.0274,  -4.9666],
        [  5.1292,   4.8183,  -5.7997],
        [ -4.7389,   6.9953,   3.7638],
        [ -8.8058,   2.4850,  -1.7425],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.4119,  14.1019,  -4.1870],
        [ -3.5029,   7.5138,   8.1853],
        [  5.7784,  -1.6401,   9.5587],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.3534,   1.2990,  -7.2942],
        [  1.1274,  14.0116,   8.1753],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -5.3328,  -8.9992,   7.3440],
        [ -7.8126,   4.3066,   5.1791],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3569,   0.7591,   6.5011],
        [  3.2990,  -4.2415,  -7.4857],
        [ -8.0608,   3.9362,  -5.9567],
        [  3.4140,  -4.1803,  -8.8224],
        [ 10.0567,   8.6697,   4.6329],
        [  6.1575,  12.2046,   3.3257],
        [  5.5240,  -2.1198,   6.2055],
        [  6.2036,   2.4030,  -1.2500],
        [ -4.0409,   7.3169,  -3.5779],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.2199,   4.6861,  -1.9304],
        [  1.7301,   7.4262,   5.0522],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.3236,   0.3011,  -4.6668],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.3616,  -2.4749,   0.1793]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0433, 0.0536,  ..., 1.0314, 0.8554, 0.1819],
        [0.0433, 0.0000, 0.0138,  ..., 1.0461, 0.8581, 0.1496],
        [0.0536, 0.0138, 0.0000,  ..., 1.0400, 0.8488, 0.1358],
        ...,
        [1.0314, 1.0461, 1.0400,  ..., 0.0000, 0.3224, 0.9895],
        [0.8554, 0.8581, 0.8488,  ..., 0.3224, 0.0000, 0.7662],
        [0.1819, 0.1496, 0.1358,  ..., 0.9895, 0.7662, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.63742669,  9.78285424,  9.7751189 ,  9.85384623,  9.76896904,
        9.99211663, 10.68416401,  4.98989906,  7.27340834, 13.14777433,
        5.19065439,  7.27340834,  9.562174  , 10.00802627,  6.383897  ,
       13.89264822, 11.31276109, 11.1913598 , 10.99326577,  6.05113217,
        9.88571713, 13.59170809,  5.30158497, 13.76047999, 10.90638066,
       12.62711252, 13.08414689,  8.7907923 , 10.86693708, 10.57302546,
        9.50654967,  8.89458273,  7.71075226, 11.67875855,  7.46053281,
        6.54127431, 13.32613445,  4.74770899,  7.822854  ,  7.46248947,
        9.79039371,  6.10774544,  5.24137909,  5.13616224,  7.69752262,
        7.72291946,  8.89458273,  7.26074925, 10.71162681, 13.05148826,
       12.19805394,  9.51074763,  9.23178204, 11.95743558,  7.72291946,
        4.85603886,  8.09842704,  8.51928531, 12.3344536 , 10.71162681,
       11.70168755,  5.68942572,  5.84450127,  8.95983586, 12.27835996,
       12.97370371,  9.6282763 , 12.41366754,  9.43737523, 11.70168755,
       13.02818702,  6.2322769 , 13.3426541 , 13.50484539, 11.1341798 ,
        7.03733775,  8.44940661,  9.14973319,  5.52416368, 14.10796466,
        8.29022197,  6.00663314,  5.35774573,  5.35774573,  6.48486171,
       14.05687815,  9.99145096, 10.4606265 ,  8.92098752, 12.87752052,
        6.4020404 ,  5.37343173,  8.97052115,  5.39723241, 13.2778378 ,
       13.66991027,  5.91677188,  6.652742  ,  8.35856487, 12.50996828,
        7.01475376,  7.62509582, 12.67576792,  6.33075682,  7.98469166,
        9.68323686])}
{'P': tensor([[1.2073e-02, 2.4731e-01, 1.9895e-01,  ..., 4.0665e-06, 5.0091e-06,
         1.2763e-05],
        [5.3726e-05, 7.8597e-03, 3.1800e-01,  ..., 2.5367e-06, 3.7167e-06,
         1.3466e-05],
        [1.1098e-04, 3.0782e-04, 2.1868e-02,  ..., 7.1348e-06, 1.0749e-05,
         3.8444e-05],
        ...,
        [6.9307e-04, 3.1643e-04, 2.5904e-04,  ..., 9.8003e-02, 2.9684e-03,
         2.4467e-05],
        [1.5724e-04, 8.3829e-05, 7.0091e-05,  ..., 7.9554e-02, 5.8195e-02,
         4.3612e-05],
        [2.2324e-05, 2.6255e-05, 2.5809e-05,  ..., 2.3178e-04, 1.1391e-03,
         1.7314e-01]]), 'X': tensor([[ -9.4159,  -2.0739,   9.6841],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.1231,  -3.5739,   8.4708],
        [  1.7500,   7.4187,   4.4237],
        [  7.8375,  -8.7240,  -9.8913],
        [  2.5636, -10.9293,   9.6519],
        [ 12.5717,   1.9266,   7.4903],
        [ -9.4541,  -0.3327,   8.8246],
        [ -0.0410,   7.8591,   9.3457],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.4242, -10.1261,   3.9595],
        [  3.9515,  -3.8425,   6.1729],
        [  1.3705,  13.9723,   7.9266],
        [  6.3266,   0.3292,   9.3502],
        [ 10.0400,   8.6922,  -6.2847],
        [  7.9141,  -8.6656,   9.6948],
        [  2.7121,   6.9743,  -8.4202],
        [ -8.0829,  -5.9940,   5.0564],
        [ -1.2454,   7.9127,  -0.5715],
        [ -7.4733,   4.7555,   6.2490],
        [  9.3624,   9.5318,   0.2779],
        [  3.9770, -10.6341,   8.0783],
        [  6.2026,  12.1774,  -1.5887],
        [  9.5261,  -7.1675,   1.4206],
        [ -1.9487,   7.8590,  -3.3330],
        [  6.9956,  11.6590,  -7.6876],
        [ 10.1798,   8.5002,  -5.8893],
        [  2.3261,  13.7749,   5.9121],
        [  4.0078,  -3.8018,  -6.2168],
        [  4.6231,   5.4556,   8.4973],
        [  6.5428,  -9.5728,  -2.9897],
        [ -3.2053, -10.1993,  -3.8736],
        [  8.1043,  -8.5164,  -7.9290],
        [ 11.4495,   6.3026,   6.8257],
        [ -5.3179,  -9.0101,   9.4472],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.3031,   1.8477,  -7.1706],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.0419,  -4.6978,  -5.0983],
        [ -1.9646,   7.8571,   3.3817],
        [  0.7685,  -4.8098,  -2.9310],
        [  2.7295,  13.6703,  -8.9328],
        [ -9.4157,   0.0821,  -5.0025],
        [  8.1528,  10.7452,   9.1493],
        [  0.2480, -11.0153,  -3.0183],
        [ -7.7239,   4.4298,   3.1216],
        [  0.1532,  -4.7422,  -7.0145],
        [ -5.3179,  -9.0101,   9.4472],
        [ -4.6393,   7.0461,   1.1624],
        [  5.0277,   4.9589,   7.0175],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.9241,   7.3503,   7.9004],
        [ -7.6650,   4.5092,   0.2520],
        [  1.8320,   7.3871,   5.1452],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.2182,  13.8007,   5.1889],
        [  6.0560,  -0.9421,  -5.0334],
        [ 11.5966,   5.9666,   6.0083],
        [ -3.0644,   7.6439,  -1.4322],
        [  1.6232,  -4.7717,  -2.5655],
        [ -1.8788, -10.6567,   3.7068],
        [  0.3038,   7.8092,   0.4164],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.5960, -11.0335,   3.1144],
        [ 11.0891,  -4.9558,  -2.3178],
        [  4.4049,   5.6892,  -4.4394],
        [  9.3560,   9.5391,   9.4943],
        [ -2.1817,   7.8274,  -4.6298],
        [ -7.4887,   4.7364,   8.9074],
        [  4.4049,   5.6892,  -4.4394],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 12.5650,   2.0132,   4.6272],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.2218,   4.0953,  -8.2072],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.3200,   0.2690,   1.5900],
        [ 11.7779,   5.5127,  -2.0699],
        [ -8.1666,   3.7653,   3.6075],
        [ 11.9808,   4.9340,  -3.3715],
        [ -7.0948,   5.1940,   9.8171],
        [  1.3670,  -4.7990,   3.2603],
        [ -9.2687,  -2.9794,   0.6436],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.9693,  -9.3200,  -8.3822],
        [  5.9655,   3.2172,  -7.5953],
        [  2.6613,   7.0021,   4.5783],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.3411,  -2.6004,  -2.7062],
        [  9.6614,   9.1784,  -1.2087],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -8.1666,   3.7653,   3.6075],
        [ -9.1749,  -3.3803,   6.4629],
        [  4.7813,  -3.1139,   8.2691],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.5199,  -0.4711,   3.0436],
        [ 10.0567,   8.6697,   4.6329],
        [  3.2701,   6.6338,   9.9209],
        [  5.0277,   4.9589,   7.0175],
        [  3.2226,  -4.2802,  -4.0428]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0272, 0.0250,  ..., 0.7475, 0.8660, 1.2020],
        [0.0272, 0.0000, 0.0041,  ..., 0.7736, 0.8927, 1.2264],
        [0.0250, 0.0041, 0.0000,  ..., 0.7705, 0.8897, 1.2256],
        ...,
        [0.7475, 0.7736, 0.7705,  ..., 0.0000, 0.1354, 0.8247],
        [0.8660, 0.8927, 0.8897,  ..., 0.1354, 0.0000, 0.7431],
        [1.2020, 1.2264, 1.2256,  ..., 0.8247, 0.7431, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.64157461,  9.87720102,  9.94173708,  9.9353921 , 10.00802627,
        9.79814997,  7.62232307, 11.72749923, 11.22596994, 12.71843462,
        9.45995955,  7.85919687, 13.09426142, 11.48734978,  5.51177501,
       14.0393949 ,  6.33517349, 13.27994304, 11.73567759,  7.48311002,
       10.06286074,  8.01009083,  8.8580671 , 13.3607379 , 11.35345345,
       13.6660671 , 11.92133129,  8.09704052, 13.59673406, 13.26209545,
       13.96987806,  5.52416368,  7.15100303, 11.59512194, 10.6910768 ,
       11.75618513, 13.06957425, 10.46238054, 12.53047489,  6.56833864,
       10.0674887 ,  5.1224079 ,  8.09899874,  4.87083348, 13.94008948,
        9.41605684, 13.48809595, 11.018084  ,  8.90405823,  4.74468406,
       10.46238054,  8.43626114,  7.06169391,  9.66977802,  7.59795445,
        8.89301516,  7.61088125, 11.63502679, 13.97779839,  6.12886024,
       13.04156075,  8.23525433,  5.04028538, 10.82106505,  7.81509631,
        7.96181288, 11.04953767, 12.1460931 ,  7.19513806, 13.3614598 ,
        8.12580976,  8.86082124,  7.19513806, 10.31613269, 12.72524615,
       13.99048094, 12.889695  ,  5.23878371, 13.03002338,  5.27178363,
        6.32572323, 13.00413829,  8.99276564, 12.95702647,  8.79284466,
        4.98989906,  9.73579738,  8.20436764, 11.63765273,  6.77777606,
        7.49076899, 11.48434767,  9.69629   , 13.32613445, 12.49317563,
        8.99276564,  9.77777346,  5.70591986, 13.0743399 , 12.45418236,
       12.4013153 , 12.52875793, 13.2778378 ,  7.39599986,  7.06169391,
        5.35774573])}
{'P': tensor([[8.9593e-03, 1.6991e-01, 1.6671e-01,  ..., 4.1757e-06, 2.3890e-06,
         6.2765e-05],
        [2.2244e-04, 9.4525e-03, 2.5749e-01,  ..., 5.6703e-06, 2.3848e-06,
         8.7671e-05],
        [5.6490e-05, 7.5430e-05, 9.6343e-03,  ..., 7.3637e-06, 2.0293e-06,
         1.2844e-04],
        ...,
        [1.9458e-05, 1.9973e-05, 1.5910e-05,  ..., 9.4297e-02, 2.6663e-03,
         1.5573e-03],
        [2.0968e-03, 1.6850e-03, 8.3867e-04,  ..., 1.1876e-04, 1.1382e-01,
         2.0602e-04],
        [6.3800e-05, 6.7543e-05, 6.6604e-05,  ..., 3.2914e-02, 1.8925e-03,
         1.1596e-01]]), 'X': tensor([[ -9.3810,  -2.3455,  -9.5194],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.6321,  -4.5296,  -2.5159],
        [ -9.4739,  -1.2497,   0.5483],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.3419,   0.5972,  -1.2841],
        [  3.7916,  -3.9524,  -7.9547],
        [  3.2990,  -4.2415,  -7.4857],
        [ -8.8934,   2.2597,   3.5831],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.2373,  -3.1225,  -8.9763],
        [  4.7482, -10.3907,   5.9149],
        [  0.9809,  -4.8145,   6.8706],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.1472,  -2.6801,   8.1194],
        [  6.0560,  -0.9421,  -5.0334],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -3.3116, -10.1539,   7.5919],
        [  5.1861,   4.7361,  -3.7312],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.6603,   2.8250,   5.9181],
        [  2.7558,  -4.4844,   0.0636],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.2373,  -4.6506,  -1.7334],
        [  8.7715,  -7.9390,   5.2900],
        [  2.3261,   7.1726,   6.2522],
        [  5.0617,  12.7957,   8.3544],
        [ -8.3062,  -5.6008,   0.0287],
        [  1.5662,  13.9375,   4.5142],
        [  8.1528,  10.7452,   9.1493],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.3266,   0.3292,   9.3502],
        [ 12.5650,   2.0132,   4.6272],
        [ -7.6650,   4.5092,   0.2520],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.5158,  -2.1337,   3.5246],
        [ 12.5996,   1.3971,  -1.0694],
        [ 11.5568,   6.0602,   9.3244],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.4296,  -1.9458,   4.0833],
        [  2.2849, -10.9655,  -7.6542],
        [ -0.9715, -10.8630,   0.3555],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.9410,   6.8428,   5.4458],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.3497,   1.3680,  -1.9076],
        [  1.3767, -11.0341,   9.9415],
        [ -2.5683, -10.4433,  -2.9182],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.7792,  11.8084,   1.6251],
        [  2.1120,  -4.6818,   8.6144],
        [ 12.5770,   1.8512,  -0.6578],
        [ -8.6681,  -4.8556,   9.2518],
        [ 12.1504,   4.3674,  -7.2998],
        [  1.4749,  -4.7892,  -0.8089],
        [ -7.2515,   5.0195,   4.6434],
        [  1.0000,   7.6591,  -4.5011],
        [  0.7958,  -4.8109,  -8.2248],
        [ -7.9925,  -6.1423,   2.8990],
        [ 10.5057,  -5.9099,  -5.1599],
        [  5.7151,  -1.7700,   3.1542],
        [  6.2975,   0.0947,   5.1791],
        [  5.4242, -10.1261,   3.9595],
        [  6.2895,   0.0405,   5.7652],
        [ -3.9456,  -9.8542,   0.7466],
        [ 10.4758,  -5.9539,   1.3667],
        [ -8.6675,  -4.8569,  -9.1769],
        [  3.3633,  13.4794,   0.5193],
        [  2.8602,   6.8905,   5.8779],
        [ -9.1459,   1.4780,  -4.2403],
        [  3.2701,   6.6338,   9.9209],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.3535,  -5.5118,   9.8399],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.1666,   3.7653,   3.6075],
        [  4.1297,  -3.7098,  -3.9838],
        [ 11.4743,   6.2478,  -2.1100],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.2728, -11.0372,   9.8121],
        [  0.9659,  -4.8144,  -0.9594],
        [  4.0642,  -3.7599,  -3.4565],
        [ -7.8849,  -6.3115,  -4.7479],
        [  0.2435,   7.8191,  -6.4709],
        [ -3.3116, -10.1539,   7.5919],
        [  6.3305,   1.6119,   1.1538],
        [ 10.5678,   7.9261,  -0.6100],
        [ -3.1204, -10.2345,   9.2288],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.0233,   4.9647,  -3.1788],
        [ 12.1504,   4.3674,  -7.2998],
        [  0.1041, -11.0045,   2.4311],
        [  2.1379,  -4.6757,   6.4981],
        [ 11.4495,   6.3026,   6.8257],
        [  2.6613,   7.0021,   4.5783],
        [  7.8982,  -8.6778,   0.0239],
        [  7.0642,  11.6104,  -9.8316],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.1984,  -0.4327,   1.5438],
        [  0.1497,   7.8334,  -5.4412],
        [  4.0642,  -3.7599,  -3.4565]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0071, 0.0244,  ..., 1.2263, 0.8177, 1.0340],
        [0.0071, 0.0000, 0.0222,  ..., 1.2299, 0.8229, 1.0390],
        [0.0244, 0.0222, 0.0000,  ..., 1.2507, 0.8405, 1.0572],
        ...,
        [1.2263, 1.2299, 1.2507,  ..., 0.0000, 0.5041, 0.3411],
        [0.8177, 0.8229, 0.8405,  ..., 0.5041, 0.0000, 0.2202],
        [1.0340, 1.0390, 1.0572,  ..., 0.3411, 0.2202, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.66977802,  9.75075273,  9.93553424,  9.99145096, 10.00410625,
       10.27716434,  5.23878371,  9.55593374, 13.86480175,  9.36093562,
        5.47702766,  5.37343173,  9.17595899, 12.32545637,  8.29203356,
        9.75075273, 11.42420508,  4.91336994,  9.61942445,  5.80312949,
        6.12886024, 12.49291768, 10.68031351,  7.02326515,  8.95983586,
        9.10945678,  5.26342203, 12.76408562,  5.16077764, 11.83075005,
        7.5403822 , 13.76047999, 10.01804223, 14.02526505, 13.48809595,
       11.51811603,  6.33517349, 12.72524615,  8.89301516, 12.13568598,
        5.91408504, 12.67680439, 13.0493368 , 11.785566  ,  9.6282763 ,
       11.20100795, 10.90638066,  7.98469166,  7.44805901,  9.96391045,
        6.49538841, 11.11969815, 10.75443106, 13.1399181 , 13.61600536,
        5.13616224, 12.7125079 ,  9.9353921 , 12.91143491,  5.01114719,
        8.81930525,  7.72415048,  4.8763135 , 10.08001894, 12.05395056,
        5.98285319,  6.29821673, 11.48734978,  6.28961988, 10.61472636,
       12.0495521 ,  9.93553424, 13.89264822,  7.46053281,  9.26455795,
        7.39599986, 12.45418236, 12.46513646, 10.00802627, 10.33958639,
        8.99276564,  5.5513072 , 13.06498554, 10.32715715, 11.11038551,
        4.91038884,  5.5366698 , 10.09979747,  7.822854  , 10.68031351,
        6.53250937, 13.20988996, 10.69963954,  9.5014146 ,  7.06272256,
       12.91143491, 11.0050379 ,  5.14124612, 13.06957425,  7.49076899,
       11.73397309, 13.59056613,  9.44486609,  6.2134961 ,  7.83486943,
        5.5366698 ])}
{'P': tensor([[8.9280e-03, 1.6153e-01, 1.4987e-01,  ..., 5.0139e-04, 1.5945e-05,
         1.4881e-06],
        [6.7834e-05, 1.1942e-02, 4.0322e-02,  ..., 5.1550e-04, 6.4603e-05,
         1.7309e-06],
        [6.1012e-05, 2.0270e-03, 1.2047e-02,  ..., 6.9848e-04, 6.3666e-05,
         1.9282e-06],
        ...,
        [2.5202e-05, 7.0233e-05, 9.1872e-05,  ..., 3.8564e-02, 3.7399e-05,
         1.3103e-05],
        [1.8508e-06, 1.9927e-06, 1.8999e-06,  ..., 5.7814e-07, 6.2064e-02,
         3.2964e-05],
        [3.0024e-04, 1.2405e-04, 1.3200e-04,  ..., 8.7803e-05, 2.2045e-06,
         9.8841e-02]]), 'X': tensor([[ -9.3616,  -2.4749,   0.1793],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.1426,   3.8048,   9.3542],
        [  5.0617,  12.7957,   8.3544],
        [  6.0031,  12.2958,  -6.5992],
        [ -4.8751,  -9.3148,   4.0406],
        [  9.5203,   9.3483,   5.8781],
        [  6.2902,   0.0451,   3.7116],
        [ -9.4196,   0.0478,   9.4140],
        [ 11.1420,   6.9331,  -2.4779],
        [ -5.9872,   6.2037,  -9.4477],
        [  7.0171,  11.6439,  -5.0178],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.7659,   7.4127,   4.7871],
        [ -6.7992,   5.4991,   9.9186],
        [ -0.1626, -10.9796,   6.4629],
        [  1.7301,   7.4262,   5.0522],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.2038,   2.4020,  -3.1692],
        [ -6.0195,   6.1789,  -2.9496],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.8700,  -4.8132,  -7.8042],
        [ -7.2986,  -7.1225,  -3.0814],
        [  4.3523, -10.5232,   8.2138],
        [  6.5428,  -9.5728,  -2.9897],
        [  0.8572,  14.0501,  -8.3364],
        [ 12.5832,   1.7543,  -7.6134],
        [  1.9204,   7.3518,   8.4781],
        [ -5.7347,  -8.6908,   9.2878],
        [  5.8889,  -1.3916,   8.8993],
        [  7.9816,  10.8938,  -0.5378],
        [ -9.4203,  -2.0340,   2.5856],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.5636, -10.9293,   9.6519],
        [ 12.4592,  -0.9168,  -6.7571],
        [  5.3804,  -2.3504,   7.2744],
        [ 11.7349,  -3.5838,   2.1644],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 11.2517,  -4.6495,   2.9800],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.3411,  -2.6004,  -2.7062],
        [  3.3839,  -4.1966,   8.3574],
        [ -4.7847,  -9.3729,  -3.7741],
        [  1.7000,  -4.7609,   2.7935],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -8.3535,  -5.5118,   9.8399],
        [ 11.3800,  -4.3919,   1.4066],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.1907,  -3.3180,  -5.5943],
        [  3.4401,  13.4540,   4.6576],
        [  7.2340,  -9.1505,  -5.8321],
        [ -7.8126,   4.3066,   5.1791],
        [  0.7275,  14.0667,  -0.9815],
        [  6.3550,   0.7129,   0.4917],
        [  6.3200,   0.2690,   1.5900],
        [  2.3367,   7.1676,  -9.5263],
        [  2.8602,   6.8905,   5.8779],
        [  3.1480,  -4.3165,  -7.8002],
        [ -1.9646,   7.8571,   3.3817],
        [  1.8320,   7.3871,   5.1452],
        [  9.6614,   9.1784,  -1.2087],
        [ 12.5154,   2.5072,  -9.3274],
        [  0.4119,  14.1019,  -4.1870],
        [  1.7000,  -4.7609,   2.7935],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.7349,  -3.5838,   2.1644],
        [  9.6614,   9.1784,  -1.2087],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.1531,  -5.8748,   2.6934],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.3305,   1.6119,   1.1538],
        [  6.3516,   0.6425,   0.3425],
        [ -7.8126,   4.3066,   5.1791],
        [ -5.9872,   6.2037,  -9.4477],
        [  2.6321,  -4.5296,  -2.5159],
        [  7.9403,  -8.6455,  -8.7433],
        [  6.3246,   1.6696,  -0.1544],
        [ -8.6229,   2.9068,   2.4429],
        [ 12.4691,   2.8525,  -5.5110],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.4773,  -1.0024,   5.2599],
        [  4.4049,   5.6892,  -4.4394],
        [  7.9141,  -8.6656,   9.6948],
        [ 11.0373,   7.1299,  -8.5895],
        [  2.3589,  13.7668,  -3.6501],
        [  6.3200,   0.2690,   1.5900],
        [ -1.9487,   7.8590,  -3.3330],
        [  0.7275,  14.0667,  -0.9815],
        [  2.3373,  -4.6232,  -5.2246],
        [ -0.6323,   7.9085,  -5.9544],
        [  0.9659,  -4.8144,  -0.9594],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.3118,   4.5452,   4.2114],
        [ -1.5261,   7.8988,   9.3768],
        [ -1.6906,   7.8860,   7.7142],
        [  2.5974,   7.0363,  -7.7810],
        [ -6.0175,  -8.4541,  -9.8408],
        [  8.6420,  -8.0580,   6.5781],
        [  2.5974,   7.0363,  -7.7810]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0250, 0.0448,  ..., 0.6101, 0.8768, 0.8818],
        [0.0250, 0.0000, 0.0561,  ..., 0.6266, 0.8804, 0.9062],
        [0.0448, 0.0561, 0.0000,  ..., 0.5706, 0.9189, 0.8818],
        ...,
        [0.6101, 0.6266, 0.5706,  ..., 0.0000, 1.3634, 0.7952],
        [0.8768, 0.8804, 0.9189,  ..., 1.3634, 0.0000, 1.0196],
        [0.8818, 0.9062, 0.8818,  ..., 0.7952, 1.0196, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.68323686,  9.95552626,  9.97964013,  9.73579738, 10.00794468,
       10.01804223,  8.98765644, 13.76047999, 13.6829843 , 10.51340829,
       13.3426541 ,  6.29034958,  9.41970031, 13.12299145,  8.62162372,
       13.59480906, 10.32715715,  7.62011734,  8.74470218, 10.98076873,
        7.62509582,  8.30125201,  6.65259775,  8.62631801, 10.52357479,
        4.89121805, 10.1979669 , 11.38774149, 11.59512194, 14.07623266,
       12.70489563,  7.5984809 , 10.412299  ,  6.05113217, 13.50484539,
        9.63742669,  8.83341608, 11.22596994, 12.49291768,  5.87132718,
       12.26997165, 10.14436912,  9.50654967,  9.97964013, 12.17451685,
       10.80728788,  9.75075273,  9.69629   ,  5.39098669, 10.52357479,
        5.05534609, 10.57302546, 10.00802627, 12.19805394, 12.10942412,
        9.7712321 , 13.88683436, 11.66453059,  8.92098752, 14.08549802,
        6.3949043 ,  6.32572323,  7.53884104,  7.46053281,  5.34250225,
        8.09899874,  7.61088125, 13.32613445, 12.76408562, 14.10796466,
        5.05534609, 12.53047489, 12.26997165, 13.32613445, 11.62158554,
       10.04916678, 10.74409834,  6.53250937,  6.38399789,  8.92098752,
        8.62162372,  5.23878371, 11.73848065,  6.54127431,  9.09963894,
       12.79126785,  9.09963894,  9.53015376,  7.19513806, 11.73567759,
       13.1399181 , 13.96746816,  6.32572323,  8.09704052, 14.08549802,
        5.18047638,  7.9337647 ,  4.91038884, 12.76408562,  6.99097013,
        8.04483495,  8.06517143,  7.50035355, 10.37698435, 11.8159254 ,
        7.50035355])}
{'P': tensor([[1.0642e-02, 3.9289e-01, 1.2901e-01,  ..., 3.6811e-06, 1.6459e-05,
         4.3062e-05],
        [2.1330e-04, 1.4184e-02, 2.3974e-01,  ..., 5.0991e-06, 2.2864e-05,
         6.6999e-05],
        [7.3184e-05, 9.2409e-05, 1.9688e-02,  ..., 1.0872e-05, 4.8528e-05,
         1.8536e-04],
        ...,
        [2.7279e-05, 2.3747e-05, 1.6710e-05,  ..., 1.3360e-01, 1.5848e-03,
         2.1604e-04],
        [2.5972e-05, 2.4973e-05, 2.7733e-05,  ..., 1.2309e-01, 1.5481e-01,
         3.4056e-04],
        [1.8114e-04, 1.7198e-04, 1.4403e-04,  ..., 1.0179e-03, 2.3890e-04,
         2.6733e-01]]), 'X': tensor([[ -9.3616,  -2.4750,   3.7765],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.0965,  -5.9712,   4.3466],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.1531,  -5.8748,   2.6934],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.2038,   2.4020,  -3.1692],
        [  3.3633,  13.4794,   0.5193],
        [  5.6202,  -1.9508,   0.7471],
        [ -6.7745,  -7.7264,   8.4684],
        [  2.3373,  13.7721,  -0.2442],
        [  0.1041, -11.0045,   2.4311],
        [  2.1625,  -4.6697,   7.0658],
        [ -0.9715, -10.8630,   0.3555],
        [ -9.4296,  -1.9458,   4.0833],
        [  4.1348,   5.9514,  -0.8856],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 10.4737,  -5.9570,  -1.2108],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.8407,  11.7666,   3.0045],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.3103,  -4.8032,  -0.9300],
        [ 12.3453,  -1.5462,   7.0212],
        [  2.1625,  -4.6697,   7.0658],
        [  8.4691,  10.4571,  -5.0650],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.4630,  -1.5145,   0.2983],
        [ 10.1503,   8.5414,  -7.2681],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.4766,  -1.1116,  -2.1706],
        [  6.3497,   1.3680,  -1.9076],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -7.4808,   4.7462,   2.0266],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.3357,   3.6001,   0.7903],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.4179,  -2.2925,  -3.4641],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.9003,  12.3550,   2.1917],
        [  5.4426, -10.1181,  -5.1268],
        [  5.6022,  -1.9833,   8.3081],
        [ -8.4573,   3.2448,   5.4370],
        [  5.7963,  -1.6018,   1.8856],
        [  7.7750,  11.0665,   2.8364],
        [  5.8275,  -1.5334,   2.0124],
        [  6.0345,   3.0175,  -1.4424],
        [  4.1297,  -3.7098,  -3.9838],
        [ 11.7796,  -3.4696,   9.0088],
        [  1.7903,  -4.7467,   0.1168],
        [  8.5716,  10.3596,   6.8818],
        [ -8.0766,  -6.0046,  -3.1873],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.6422,   7.4586,  -6.8056],
        [  4.9152,  12.8653,  -1.4459],
        [  2.4639, -10.9431,  -7.3718],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.3413,   0.4883,  -6.8416],
        [ -0.2586,   7.8825,  -9.9840],
        [  4.8674,  12.8875,   6.7622],
        [  6.0467,  -9.8369,  -6.5816],
        [  2.7976,  -4.4682,  -4.2020],
        [ -2.5683, -10.4433,  -2.9182],
        [  8.4691,  10.4571,  -5.0650],
        [  3.4401,  13.4540,   4.6576],
        [ -7.3446,   4.9113,  -5.9356],
        [  1.6954,  -4.7616,   9.8198],
        [ -7.6719,  -6.6260,   7.1468],
        [  5.1203,  -2.7151,   6.2110],
        [  3.2692,  13.5098,  -4.9032],
        [ 12.2311,  -2.0445,   1.3545],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.8745,   2.3097,   3.5600],
        [  1.0093,   7.6567,   2.3526],
        [  6.5428,  -9.5728,  -2.9897],
        [  2.6614,  -4.5192,  -9.3641],
        [ 11.9936,  -2.8677,   6.0614],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.3653,   3.4184,   9.3454],
        [  0.3724,   7.7974,  -6.8722],
        [  0.2571,  -4.7594,   1.5618],
        [ 11.2209,  -4.7092,   2.6117],
        [ 10.0567,   8.6697,   4.6329],
        [  1.3103,  -4.8032,  -0.9300],
        [ -6.8618,  -7.6322,   3.7027],
        [  0.9788,  -4.8145,  -3.8661],
        [ -4.0409,   7.3169,  -3.5779],
        [ -8.5607,  -5.0938,  -5.4531],
        [  3.4009,  -4.1874,  -5.9751],
        [ 12.5224,  -0.4497,   4.8448],
        [ -5.9938,  -8.4745,   3.9735],
        [  0.7685,  -4.8098,  -2.9310],
        [  6.3515,   0.6418,  -6.5823],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.2576,   2.1359,  -3.3398],
        [ -8.8540,  -4.3971,  -5.2144],
        [  5.1203,  -2.7151,   6.2110],
        [ -1.6086,   7.8928,   4.3385],
        [  4.1348,   5.9514,  -0.8856],
        [  3.8111, -10.6787,  -4.6539],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.3301,   0.3638,  -7.5685],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.6093,  -1.9705,   6.4807]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.8918e-02, 2.7689e-02,  ..., 1.1516e+00, 1.1320e+00,
         9.3837e-01],
        [1.8918e-02, 0.0000e+00, 8.7880e-03,  ..., 1.1700e+00, 1.1506e+00,
         9.4840e-01],
        [2.7689e-02, 8.7880e-03, 3.0339e-04,  ..., 1.1787e+00, 1.1593e+00,
         9.5366e-01],
        ...,
        [1.1516e+00, 1.1700e+00, 1.1787e+00,  ..., 0.0000e+00, 4.3547e-02,
         8.3617e-01],
        [1.1320e+00, 1.1506e+00, 1.1593e+00,  ..., 4.3547e-02, 2.1453e-04,
         8.5606e-01],
        [9.3837e-01, 9.4840e-01, 9.5366e-01,  ..., 8.3617e-01, 8.5606e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.68324553,  9.75200848, 10.05558893, 10.06023665,  9.99211663,
       10.04916678, 12.64235359,  6.65259775, 13.89264822,  5.94909709,
       10.27572673, 13.96905756, 11.0050379 ,  5.14607014, 10.90638066,
        9.6282763 ,  7.24677664, 12.3344536 ,  9.54154096, 12.04924159,
        5.5671389 , 13.6105595 , 12.3828957 ,  4.978704  , 12.44177641,
        5.14607014, 13.45642379,  9.11473391,  9.58347456, 13.26590071,
       12.16901615,  6.34059021,  9.54154096,  6.49538841, 10.27716434,
        8.859416  , 11.57249984, 12.85032836,  5.27178363,  5.8828986 ,
       11.67875855, 13.69162795, 11.4890861 ,  5.94293185,  9.05843201,
        6.01356004, 13.5247215 ,  6.02588517,  6.74686071,  5.5513072 ,
       12.27993098,  5.07306274, 13.44593519, 10.06407839,  6.07992623,
        7.63726134, 13.77222752, 11.21703654, 11.51811603,  6.36004393,
        7.88677538, 13.77604293, 11.54672755,  5.27178363, 10.75443106,
       13.45642379, 13.88683436,  8.8353674 ,  5.05444288, 10.1371552 ,
        5.79562691, 13.89974693, 12.40074901,  4.91652556,  9.17016857,
        7.72291946, 11.59512194,  5.24461068, 12.33167382,  8.40445872,
        9.0368458 ,  7.80625981,  4.7663526 , 12.16901615, 13.2778378 ,
        4.978704  , 10.26327895,  4.91296205,  8.35856487,  9.96152788,
        5.39450305, 12.53047489, 10.37998078,  4.87083348,  6.383897  ,
       12.10222204,  6.61211224,  9.88571713,  5.79562691,  8.0550286 ,
        7.24677664, 11.33837396,  8.62631801,  6.34059021,  5.39723241,
        5.94535529])}
{'P': tensor([[4.9021e-03, 2.2335e-01, 1.8291e-01,  ..., 1.1967e-05, 1.8572e-06,
         1.3189e-07],
        [9.1007e-05, 6.6744e-03, 2.2056e-01,  ..., 1.9609e-05, 2.5550e-06,
         1.8965e-07],
        [1.2886e-04, 2.3100e-04, 9.8581e-03,  ..., 3.0510e-05, 3.8174e-06,
         2.9020e-07],
        ...,
        [9.7354e-06, 1.0096e-05, 1.0104e-05,  ..., 4.0068e-01, 4.4372e-04,
         2.9730e-04],
        [4.2677e-04, 3.5881e-04, 3.3269e-04,  ..., 5.7338e-06, 1.2830e-01,
         2.1736e-03],
        [7.7600e-07, 6.3373e-07, 5.8084e-07,  ..., 1.3983e-08, 9.3843e-04,
         4.0487e-03]]), 'X': tensor([[ -9.3411,  -2.6004,  -2.7062],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.2883,   0.0325,   5.2692],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -3.7964,  -9.9294,   7.7147],
        [  3.3094,  13.4969,  -3.6507],
        [  6.3266,   0.3292,   9.3502],
        [  0.8572,  14.0501,  -8.3364],
        [ -0.8885,   7.9160,   0.0571],
        [ -4.2040,   7.2486,   2.9875],
        [  3.3744,   6.5623,  -6.3636],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.2076,  13.8032,  -1.5099],
        [ -7.5720,   4.6310,   2.1717],
        [  8.8474,  10.0868,  -8.8644],
        [  0.1252, -11.0062,   7.2564],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.0438,   7.1180,  -5.6991],
        [ -3.0004,   7.6607,  -7.2640],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.6000,   1.3839,  -3.2992],
        [  1.7301,   7.4262,   5.0522],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -7.3333,   4.9246,  -7.9212],
        [ 11.6019,   5.9540,  -5.1715],
        [ -1.8788, -10.6567,   3.7068],
        [  2.5813,  -4.5471,   5.3597],
        [  6.3604,   0.9042,  -6.5901],
        [ -8.0766,  -6.0046,  -3.1873],
        [  2.5318,  13.7231,   6.3692],
        [ -9.4753,  -0.8012,  -5.2050],
        [  4.5954,  13.0099,  -4.4439],
        [  6.1929,   2.4501,  -6.9352],
        [ -1.1885, -10.8211,   0.0865],
        [  9.6816,   9.1536,  -5.8543],
        [  1.1994,   7.6035,   2.4295],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.3497,   1.3680,  -1.9076],
        [  6.9653,  -9.3225,  -3.4727],
        [ -9.1433,  -3.5003,  -3.0209],
        [  7.3724,  -9.0577,  -8.9843],
        [  1.2728, -11.0372,   9.8121],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.6634,   5.8050,   9.1590],
        [  1.0391,  -4.8141,  -0.0768],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.1204,  -5.9308,   4.0761],
        [  1.7659,   7.4127,   4.7871],
        [ 12.2218,   4.0953,  -8.2072],
        [  5.4167, -10.1293,   6.2156],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.6357,   5.4415,   7.9067],
        [ -9.4196,   0.0478,   9.4140],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.1041, -11.0045,   2.4311],
        [  1.7301,   7.4262,   5.0522],
        [  6.2902,   0.0451,   3.7116],
        [ 12.1379,   4.4125,  -3.8833],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.4697,  -0.6157,  -6.8472],
        [  9.6614,   9.1784,  -1.2087],
        [  1.3705,  13.9723,   7.9266],
        [  6.3587,   1.1596,  -3.6629],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.2630,   2.1057,  -5.2136],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.2373,  -3.1225,  -8.9763],
        [  6.3211,   1.7017,  -0.2164],
        [  4.5814, -10.4485,   2.5496],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.3604,   0.4855,  -3.0158],
        [ 10.1503,   8.5414,  -7.2681],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.3858,   4.4260,  -9.6769],
        [ -2.1533, -10.5778,   1.0112],
        [ -6.0175,  -8.4541,  -9.8408],
        [  0.1206,  -4.7363,   8.7103],
        [  1.0093,   7.6567,   2.3526],
        [  1.5383, -11.0274,  -4.9666],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.2437,  -4.6489,   6.1430],
        [  6.2790,  -0.0259,  -5.4116],
        [  2.3373,  13.7721,  -0.2442],
        [ -8.1531,  -5.8748,   2.6934],
        [ 11.5456,   6.0861,   8.1101],
        [ -1.6086,   7.8928,   4.3385],
        [  5.6022,  -1.9833,   8.3081],
        [  2.7590,  13.6621,   9.4496],
        [ 11.2847,  -4.5848,  -4.7503],
        [  1.7000,  -4.7609,   2.7935],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -1.9487,   7.8590,  -3.3330],
        [  4.3523, -10.5232,   8.2138],
        [  2.4639, -10.9431,  -7.3718],
        [  5.6414, -10.0303,   7.8525],
        [  4.4049,   5.6892,  -4.4394],
        [ 10.1798,   8.5002,  -5.8893]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0079, 0.0152,  ..., 1.0376, 0.9142, 1.2764],
        [0.0079, 0.0000, 0.0087,  ..., 1.0435, 0.9217, 1.2835],
        [0.0152, 0.0087, 0.0000,  ..., 1.0522, 0.9252, 1.2858],
        ...,
        [1.0376, 1.0435, 1.0522,  ..., 0.0000, 1.0032, 1.3410],
        [0.9142, 0.9217, 0.9252,  ..., 1.0032, 0.0000, 0.3959],
        [1.2764, 1.2835, 1.2858,  ..., 1.3410, 0.3959, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.69629   ,  9.75401449,  9.78147542,  9.79039371,  9.86089579,
       10.06407839,  6.28835101,  9.75075273, 10.63038565, 13.89671314,
        6.33517349, 14.07623266,  7.96575782,  8.37954853,  7.37902766,
        8.20436764, 13.97857591,  8.87587457, 13.41713661, 11.00694688,
        5.1224079 , 13.13888841,  8.22728399, 13.99048094, 12.67576792,
        7.62509582, 12.97370371,  9.69792513,  8.83341608, 13.04051744,
       10.82106505,  5.2287058 ,  6.42439344, 10.06407839, 13.95472331,
        9.50913849, 13.79762415,  6.65991877, 10.88618114, 13.32374056,
        7.69752262, 12.49291768, 10.07238463,  6.49538841, 11.63724792,
        9.79039371, 11.67875855, 11.11038551,  6.80370254,  9.5642993 ,
       13.61801526, 13.02818702,  4.92497514, 13.26590071, 10.05558893,
        7.62011734, 12.889695  , 11.48663767, 10.37998078,  7.14837514,
        9.41970031,  6.27905292, 11.0050379 ,  7.62509582,  6.29034958,
       12.91505111, 10.37998078,  9.48969978, 13.32613445, 14.0393949 ,
        6.46356806,  9.60412735,  6.60752373,  4.8763135 ,  9.75075273,
        6.54615887, 11.4088043 , 10.65929968,  9.37296102, 13.26590071,
       12.32545637,  6.97106912, 10.794755  , 10.37698435,  4.73784222,
        7.72291946, 11.1341798 , 13.00413829,  5.16205087,  6.27905292,
       13.96905756, 10.04916678, 13.05148826,  8.0550286 ,  5.94293185,
       13.93790235, 12.18045978,  5.05534609,  9.27474386,  9.43112957,
        8.09704052, 11.38774149, 11.21703654, 11.50791859,  7.19513806,
       13.26209545])}
{'P': tensor([[5.6521e-03, 1.9836e-01, 1.6887e-01,  ..., 1.4103e-06, 4.2262e-04,
         9.6769e-06],
        [6.8086e-05, 4.9411e-03, 2.2284e-01,  ..., 8.9599e-07, 4.8840e-04,
         9.7024e-06],
        [8.5648e-05, 1.1527e-04, 9.3357e-03,  ..., 1.3913e-06, 1.2298e-03,
         2.2228e-05],
        ...,
        [2.2631e-04, 1.9190e-04, 1.8220e-04,  ..., 3.0286e-02, 5.1847e-05,
         5.9593e-05],
        [4.9776e-06, 5.7916e-06, 6.7860e-06,  ..., 1.7899e-06, 3.7442e-02,
         2.3339e-03],
        [7.5319e-07, 8.1163e-07, 9.2093e-07,  ..., 3.1350e-06, 2.1634e-04,
         3.8552e-02]]), 'X': tensor([[ -9.3384,  -2.6161,  -4.7630],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.5607,  -5.0938,  -5.4531],
        [  9.6587,   9.1817,  -0.8212],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.5456,   6.0861,   8.1101],
        [  5.1886,  -2.6250,   6.4677],
        [  1.7122,  -4.7591,  -3.6734],
        [  5.3858,   4.4260,  -9.6769],
        [  5.3858,   4.4260,  -9.6769],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.7232,  -4.7574,   3.1270],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.3653,   3.4184,   9.3454],
        [  4.4915,   5.5990,   7.0405],
        [ -0.9715, -10.8630,   0.3555],
        [ -2.3317,   7.8034,  -3.6274],
        [ 10.8218,  -5.4177,   6.8573],
        [  4.7544,  -3.1425,  -1.5315],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.1231,  -3.5739,   8.4708],
        [  6.2036,   2.4030,  -1.2500],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.5813,  -4.5471,   5.3597],
        [  6.2038,   2.4020,  -3.1692],
        [  0.6396,  14.0772,  -6.3786],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.9026,   3.3837,  -0.6573],
        [  2.7420,  -4.4896,  -3.7139],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.4368,   5.8356,   3.3014],
        [  6.3516,   0.6425,   0.3425],
        [  3.3943,   6.5483,   1.0790],
        [  5.0617,  12.7957,   8.3544],
        [ -7.5401,   4.6718,   9.0033],
        [  1.2728, -11.0372,   9.8121],
        [  6.0031,  12.2958,  -6.5992],
        [ -0.6323,   7.9085,  -5.9544],
        [  3.6703,   6.3443,  -5.4401],
        [  3.9085,   6.1512,  -2.2560],
        [  2.4356,  -4.5942,  -5.0808],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.3261,  13.7749,   5.9121],
        [  5.8836,   3.4315,  -4.0327],
        [  4.0078,  -3.8018,  -6.2168],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 12.4877,  -0.7222,   8.8347],
        [  0.1497,   7.8334,  -5.4412],
        [ -7.8849,  -6.3115,  -4.7479],
        [  4.4039,   5.6903,   8.7366],
        [  6.2975,   0.0947,   5.1791],
        [  6.2895,   0.0405,   5.7652],
        [  6.0305,   3.0296,   0.1375],
        [ -8.3539,  -5.5111,   0.3373],
        [  1.5383, -11.0274,  -4.9666],
        [ 10.1798,   8.5002,  -5.8893],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.2849, -10.9655,  -7.6542],
        [  5.6672,  -1.8631,   8.1020],
        [ -2.1382,   7.8339,   7.3524],
        [  2.0899,  -4.6870,  -7.4260],
        [ 10.1798,   8.5002,  -5.8893],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -2.3317,   7.8034,  -3.6274],
        [ -8.5149,  -5.1905,   1.5239],
        [ -2.2188,   7.8218,   8.6064],
        [  0.1677,  -4.7447,  -6.1529],
        [  5.3860,  -2.3418,   1.0826],
        [ -2.2508, -10.5479,  -2.7235],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.8979,   3.3958,  -3.8713],
        [  5.1793,   4.7461,   0.3899],
        [  5.7802,  -1.6363,  -6.2782],
        [  7.5309,  11.2617,   7.2656],
        [  0.4996,   7.7738,  -4.4711],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.9452, -10.3187,  -7.8621],
        [  9.0129,  -7.7072,   9.1400],
        [  2.1477,   7.2550,   8.9688],
        [ 12.5224,  -0.4497,   4.8448],
        [ -8.5859,  -5.0393,   0.5476],
        [ -3.1671,   7.6158,   6.0765],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.3444,   0.5821,   6.9101],
        [ -9.4203,  -2.0340,   2.5856],
        [  6.3413,   0.4883,  -6.8416],
        [  2.4356,  -4.5942,  -5.0808],
        [  9.6587,   9.1817,  -0.8212],
        [ -7.9925,  -6.1423,   2.8990],
        [  7.5784,  -8.9141,   5.8410],
        [  6.0073,   3.0986,   6.7937],
        [ -9.4196,   0.0478,   9.4140],
        [  4.3523, -10.5232,   8.2138],
        [  1.1139,   7.6281,  -2.5680],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.4901,  -0.7052,   1.7315],
        [  8.9284,  10.0036,  -6.9210],
        [ -7.5401,   4.6718,   9.0033],
        [  0.1047, -11.0046,   0.5188],
        [  2.5636, -10.9293,   9.6519]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0207, 0.0243,  ..., 1.0164, 0.5147, 1.1589],
        [0.0207, 0.0000, 0.0057,  ..., 1.0349, 0.5352, 1.1782],
        [0.0243, 0.0057, 0.0000,  ..., 1.0356, 0.5377, 1.1794],
        ...,
        [1.0164, 1.0349, 1.0356,  ..., 0.0000, 0.5537, 0.1723],
        [0.5147, 0.5352, 0.5377,  ..., 0.5537, 0.0000, 0.6694],
        [1.1589, 1.1782, 1.1794,  ..., 0.1723, 0.6694, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.69792513,  9.78286733,  9.85330646,  9.88571713,  9.94285511,
        9.96152788, 13.32645552,  8.87007495,  9.85330646, 13.05148826,
        5.81481044,  5.05774295,  6.97106912,  6.97106912, 10.1979669 ,
        5.05989547,  6.14086582,  9.0368458 ,  7.17789666, 10.90638066,
        8.14433542, 12.10222204,  5.69913817, 12.35625694,  9.79814997,
        6.652742  ,  6.10774544,  5.2287058 ,  6.65259775, 14.09176259,
        4.90505566,  6.80370254,  5.26067019, 11.28895489,  8.68832398,
        6.38399789,  7.37576154, 13.76047999,  8.87007495, 11.11038551,
       13.6829843 ,  7.9337647 ,  7.32950385,  7.28793663,  5.19986797,
       12.10942412, 13.96987806,  6.81117802,  5.52416368, 10.51437947,
       12.5085995 ,  7.83486943, 10.09979747,  7.19534146,  6.29821673,
        6.28961988,  6.74872981, 10.00794468, 11.1341798 , 13.26209545,
       13.14777433, 11.20100795,  5.96556248,  8.12043983,  5.13182506,
       13.26209545,  4.92497514, 12.4013153 ,  9.54154096,  8.14433542,
        9.97220799,  8.1303945 ,  4.74770899,  5.87305005, 10.78533838,
       12.91505111,  6.80559669,  7.02496223,  6.00732193, 13.54774355,
        7.78980299,  5.53175275, 11.4424816 , 11.85890664,  7.56617037,
       12.53047489,  9.95552626,  8.24808832,  6.10774544,  9.36256096,
        9.63742669,  6.36004393,  5.19986797, 13.32645552, 10.08001894,
       11.70016217,  6.7593917 ,  9.41970031, 11.38774149,  7.70897675,
       10.1371552 , 12.50996828, 13.40850328,  8.87007495, 11.00508681,
       11.22596994])}
{'P': tensor([[1.3961e-02, 2.0888e-01, 1.7503e-01,  ..., 2.1662e-06, 4.9186e-06,
         1.8547e-06],
        [5.9973e-05, 7.3866e-03, 3.5478e-01,  ..., 9.6454e-07, 2.5960e-06,
         1.7389e-06],
        [3.6433e-04, 7.5978e-04, 6.5128e-02,  ..., 8.6111e-06, 2.3601e-05,
         1.8776e-05],
        ...,
        [9.6157e-05, 4.0894e-05, 3.4046e-05,  ..., 6.3641e-02, 2.6370e-03,
         2.5543e-05],
        [4.6812e-04, 2.1594e-04, 1.7708e-04,  ..., 6.5879e-03, 1.1730e-01,
         8.1339e-05],
        [1.5540e-06, 9.3385e-07, 8.3264e-07,  ..., 1.9476e-03, 9.1912e-03,
         2.9362e-02]]), 'X': tensor([[ -9.3331,  -2.6465,  -8.2028],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.8890,  -4.4316,   1.1600],
        [ -8.0611,  -6.0302,   4.6962],
        [  4.9957,  -2.8705,   6.5978],
        [  2.9470,  13.6085,   2.9809],
        [  5.2911,  -2.4824,  -3.6877],
        [ -5.3382,   6.6533,   7.8054],
        [  6.1575,  12.2046,   3.3257],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.8836,   3.4315,  -4.0327],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.1140,  -3.7220,  -0.1273],
        [  9.3624,   9.5318,   0.2779],
        [  5.1472,  -2.6801,   8.1194],
        [  4.2612, -10.5514,  -6.6268],
        [ -9.2345,  -3.1345,   3.9715],
        [ -3.5903,   7.4847,  -8.9112],
        [  9.4925,   9.3811,   8.9842],
        [  7.0171,  11.6439,  -5.0178],
        [  5.8617,   3.4853,  -9.8080],
        [  6.3303,   1.6145,  -8.4083],
        [ -8.9379,  -4.1655,  -2.8114],
        [  4.0421,  -3.7765,  -8.3631],
        [ -1.3940, -10.7772,   6.7835],
        [  4.1297,  -3.7098,  -3.9838],
        [  1.3339,   7.5627,  -0.9034],
        [  1.9312,  13.8649,   8.3563],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.3587,   1.1596,  -3.6629],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -1.3178, -10.7939,   0.4236],
        [ -0.8569,   7.9156,  -6.5557],
        [  5.6202,  -1.9508,   0.7471],
        [  0.9445,   7.6736,   0.5095],
        [ -2.5683, -10.4433,  -2.9182],
        [  4.7544,  -3.1425,  -1.5315],
        [ 12.6000,   1.3839,  -3.2992],
        [ -9.4351,  -1.8891,   3.5642],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.8111, -10.6787,  -4.6539],
        [  7.5930,  -8.9037,   7.2323],
        [ -0.0410,   7.8591,   9.3457],
        [  5.1793,   4.7461,   0.3899],
        [  2.5813,  -4.5471,   5.3597],
        [  3.7992,  -3.9474,  -8.9019],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.3087,   0.7792,  -7.5587],
        [  0.9478,  -4.8144,   0.6371],
        [ 12.2589,  -1.9312,   9.8657],
        [  1.3403,  -4.8011,  -1.1698],
        [  6.3583,   1.1729,   8.7835],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -9.1231,  -3.5739,   8.4708],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.2419,  -0.2313,   4.2440],
        [  5.1793,   4.7461,   0.3899],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 10.9162,  -5.2599,  -7.0812],
        [  4.2275,  -3.6320,  -0.2714],
        [  7.7750,  11.0665,   2.8364],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -4.3963,   7.1629,   4.5263],
        [ -2.0721,   7.8432,   6.6380],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.9587,  12.3216,   9.1695],
        [ -9.1706,   1.3863,   8.4032],
        [  0.5850,  -4.7982,   9.6639],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3569,   0.7591,   6.5011],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.3373,  13.7721,  -0.2442],
        [  1.1006,   7.6318,  -2.4877],
        [  4.8653,   5.1694,  -4.0823],
        [  4.8653,   5.1694,  -4.0823],
        [ -0.9715, -10.8630,   0.3555],
        [ -8.3062,  -5.6008,   0.0287],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.0845,  -2.5772,  -4.6073],
        [  1.7903,  -4.7467,   0.1168],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -5.6651,   6.4380,   1.3660],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 12.3609,   3.4769,  -8.5762],
        [ -1.9487,   7.8590,  -3.3330],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4714,  -1.3301,  -3.1998],
        [  2.1625,  -4.6697,   7.0658],
        [ -9.4157,   0.0821,  -5.0025],
        [ -6.8618,  -7.6322,   3.7027],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.5428,  -9.5728,  -2.9897],
        [  3.7121,   6.3117,   0.6738],
        [  3.9895,   6.0816,  -6.5235],
        [ 12.6059,   0.9597,  -6.3264]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0696, 0.0582,  ..., 1.0752, 0.9299, 1.1567],
        [0.0696, 0.0000, 0.0210,  ..., 1.1383, 0.9770, 1.1964],
        [0.0582, 0.0210, 0.0000,  ..., 1.1194, 0.9560, 1.1755],
        ...,
        [1.0752, 1.1383, 1.1194,  ..., 0.0000, 0.4275, 0.5990],
        [0.9299, 0.9770, 0.9560,  ..., 0.4275, 0.0000, 0.2658],
        [1.1567, 1.1964, 1.1755,  ..., 0.5990, 0.2658, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.70108409,  9.93553424, 10.00410625,  9.99145096,  9.91668182,
        9.75075273, 12.17901033,  5.29009836, 10.06703358,  5.76165934,
       13.92390689,  5.84450127,  8.53014915, 13.66991027, 12.4013153 ,
       10.39942497,  6.81117802,  9.75401449,  5.54776934, 13.3607379 ,
        5.80312949, 11.37939695,  9.75200848,  8.30125201, 13.34587027,
       13.59480906,  6.81961996,  6.53290663,  9.86089579,  5.53175275,
       10.86693708,  5.5513072 ,  7.67939546, 13.99877413,  8.34949383,
        6.46356806,  9.5642993 , 10.87409005,  7.96181288,  5.94909709,
        7.73151309, 10.75443106,  5.69913817, 12.67576792,  9.62238828,
        8.92549752,  8.8353674 , 11.33837396, 11.70168755,  7.85919687,
        7.02496223,  5.2287058 ,  5.47866221,  9.68324553,  9.34127013,
        4.90676242, 12.41012046,  4.98461983,  6.46560181,  9.99145096,
        9.79814997,  8.05242591,  6.24614905,  7.02496223, 12.00995898,
       12.11735334,  5.57341623, 13.5247215 , 10.06407839,  8.40445872,
        8.11227668, 12.84056604, 13.68672855,  9.27474386,  4.83371341,
       13.36092624,  6.4020404 ,  9.37296102,  9.45995955, 13.96905756,
        7.71075226,  7.09887557,  7.09887557, 10.90638066, 10.01804223,
       12.1494023 ,  9.97822473,  5.81199422, 12.35625694,  5.07306274,
       10.37698435,  8.57560275, 12.25806559, 12.84056604,  8.09704052,
        8.09704052,  9.5642993 ,  5.14607014,  9.41605684, 10.26327895,
        5.91677188, 12.35421461, 11.59512194,  7.32233733,  7.27340834,
       12.64235359])}
{'P': tensor([[7.2697e-03, 2.4069e-01, 1.8480e-01,  ..., 4.2644e-06, 3.5025e-07,
         8.0142e-07],
        [1.5647e-04, 1.0130e-02, 3.8183e-01,  ..., 6.3088e-06, 5.5090e-07,
         1.0467e-06],
        [1.6725e-04, 2.2027e-04, 1.5306e-02,  ..., 1.0160e-05, 9.4281e-07,
         1.4914e-06],
        ...,
        [8.1013e-05, 6.6740e-05, 5.6133e-05,  ..., 4.6870e-02, 1.3892e-03,
         8.3309e-04],
        [8.1710e-06, 6.4666e-06, 5.2609e-06,  ..., 1.2692e-02, 7.4996e-02,
         6.5890e-04],
        [1.3038e-04, 1.1523e-04, 1.0048e-04,  ..., 1.8850e-05, 1.5933e-06,
         2.5806e-02]]), 'X': tensor([[ -9.3107,  -2.7697,  -3.2919],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.6757, -11.0361,  -1.3047],
        [ -6.4464,   5.8272,   7.5038],
        [ -8.0588,  -6.0341,  -2.3006],
        [  3.5939,   6.4029,  -1.8357],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.1532,  -4.7422,  -7.0145],
        [ -4.4520,   7.1370,  -8.7867],
        [  4.7813,  -3.1139,   8.2691],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.9074,  -4.8139,   8.1674],
        [ -0.1963,   7.8765,  -1.1371],
        [  1.7659,   7.4127,   4.7871],
        [  6.2535,  -0.1709,  -9.4102],
        [  0.5175,  14.0910,   4.6540],
        [  2.7121,   6.9743,  -8.4202],
        [ -1.6220,   7.8917,   0.0315],
        [  5.1519, -10.2387,   9.8191],
        [ 11.4703,  -4.2006,   7.4086],
        [ -3.4488, -10.0934,   2.5195],
        [  4.4915,   5.5990,   7.0405],
        [ -9.4351,  -1.8891,   3.5642],
        [  4.4633,   5.6287,  -0.7754],
        [ -7.6293,  -6.6859,  -5.5161],
        [  2.8994,   6.8675,  -9.5987],
        [  6.3550,   0.7129,   0.4917],
        [ 12.5996,   1.3971,  -1.0694],
        [  1.1139,   7.6281,  -2.5680],
        [  5.9193,  -1.3176,   5.1739],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.7735,   2.5640,  -6.3313],
        [  6.1222,  -0.7261,   8.8204],
        [  1.3705,  13.9723,   7.9266],
        [ -7.6525,  -6.6534,   5.1721],
        [  1.1994,   7.6035,   2.4295],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.4524,  -1.6813,  -6.6862],
        [  4.1537,   5.9340,  -6.0423],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.0190,   4.9705,  -7.9752],
        [ 12.1751,  -2.2592,  -5.7751],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.9788,   2.0208,  -3.2867],
        [ -4.0868,   7.2981,   8.1874],
        [  5.8930,  12.3592,  -8.9201],
        [  1.5383, -11.0274,  -4.9666],
        [  0.1497,   7.8334,  -5.4412],
        [ -8.3539,  -5.5111,   0.3373],
        [  0.9945,  -4.8144,  -0.1897],
        [ 11.2517,  -4.6495,   2.9800],
        [ -2.0229, -10.6163,   8.6930],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.3549,   0.7096,   6.9198],
        [ 10.9931,   7.2107,   7.7933],
        [ 11.6165,  -3.8715,   6.4594],
        [  9.9750,  -6.6338,  -1.2440],
        [  4.2096,   5.8815,   1.9230],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -0.8675, -10.8815,   6.8561],
        [  2.9260,  -4.4162,   8.3808],
        [ 11.7726,  -3.4876,  -6.7402],
        [  5.6093,  -1.9705,   6.4807],
        [  2.7976,  -4.4682,  -4.2020],
        [  1.0391,  -4.8141,  -0.0768],
        [  1.7218,  -4.7576,  -3.3617],
        [  0.9788,  -4.8145,  -3.8661],
        [ -8.5149,  -5.1905,   1.5239],
        [ 11.4743,   6.2478,  -2.1100],
        [  4.7482, -10.3907,   5.9149],
        [  6.1752,   2.5242,   3.5874],
        [  3.2226,  -4.2802,  -4.0428],
        [  8.7715,  -7.9390,   5.2900],
        [  9.1620,  -7.5573,  -9.9882],
        [  4.4633,   5.6287,  -0.7754],
        [ -4.0868,   7.2981,   8.1874],
        [  6.2576,   2.1359,  -3.3398],
        [  5.8201,   3.5841,  -8.0609],
        [  8.5716,  10.3596,   6.8818],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.3360,   0.4258,  -9.2509],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -2.1382,   7.8339,   7.3524],
        [  1.7000,  -4.7609,   2.7935],
        [  5.9193,  -1.3176,   5.1739],
        [  6.2770,  -0.0379,   2.4195],
        [  1.2273,  -4.8081,   9.8129],
        [  5.7802,  -1.6363,  -6.2782],
        [  0.3493,  -4.7727,  -6.5682],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.8323,  -1.5228,   5.1132],
        [ 12.3357,   3.6001,   0.7903],
        [  1.1274,  14.0116,   8.1753],
        [  6.6319,  11.9066,   3.5165],
        [  5.6523,   3.9443,  -2.9449],
        [ 11.6544,   5.8273,  -3.1375],
        [ -1.5451,   7.8975,   7.6104]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0186, 0.0295,  ..., 0.9623, 1.2077, 1.0739],
        [0.0186, 0.0000, 0.0109,  ..., 0.9711, 1.2178, 1.0686],
        [0.0295, 0.0109, 0.0000,  ..., 0.9764, 1.2238, 1.0658],
        ...,
        [0.9623, 0.9711, 0.9764,  ..., 0.0000, 0.2604, 0.7792],
        [1.2077, 1.2178, 1.2238,  ..., 0.2604, 0.0000, 0.9578],
        [1.0739, 1.0686, 1.0658,  ..., 0.7792, 0.9578, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.71392179,  9.79039371,  9.86089579,  9.78147542, 10.06407839,
        9.94285511, 11.05672136,  8.68979052, 10.0674887 ,  7.34250948,
        8.83341608,  4.74468406,  8.41171369,  5.70591986,  7.8575648 ,
        4.89869205,  7.8788965 ,  7.62011734,  6.25586297, 14.10045797,
        7.48311002,  8.05668609, 11.46176259, 12.21532282, 10.66631784,
        7.17789666,  9.62238828,  7.18354885, 10.14436912,  7.4544866 ,
        6.3949043 , 12.67680439,  7.70897675,  6.06416957,  6.00663314,
        9.14044984,  6.16512856, 14.0393949 , 10.14044489,  7.69752262,
        5.40689332,  9.36378438,  9.60080639,  7.24326992, 13.00413829,
        7.06372172, 12.3828957 ,  6.35029274,  9.20340031,  8.36445758,
       13.6922425 , 11.1341798 ,  7.83486943, 10.00794468,  4.9160962 ,
       12.17451685, 10.80728788,  7.6374603 , 12.3828957 ,  9.34061938,
        6.39438044, 13.14690137, 12.24467041, 11.97949273,  7.23276582,
       10.06437598, 10.91601655,  5.29755487, 12.27835996,  5.94535529,
        5.27178363,  4.92497514,  5.05963217,  4.91296205,  9.97220799,
       13.06498554, 11.42420508,  6.67121748,  5.35774573, 11.83075005,
       11.87665501,  7.18354885,  8.36445758,  6.61211224,  6.83515201,
       13.44593519,  5.24461068,  6.35029274, 12.18045978, 10.60226614,
        8.12043983,  5.05534609,  6.06416957,  6.27715105,  4.96231532,
        6.00732193,  4.78544875,  7.26074925,  9.97964013,  6.02779201,
       12.85032836, 14.05687815, 13.62896593,  6.89245352, 13.03002338,
        8.04718433])}
{'P': tensor([[9.2354e-03, 2.2157e-01, 1.7246e-01,  ..., 3.4508e-07, 3.6176e-06,
         1.3038e-04],
        [9.0717e-05, 1.3535e-02, 5.1557e-01,  ..., 7.1081e-07, 6.8070e-06,
         3.2662e-04],
        [3.3168e-04, 9.8664e-04, 6.0519e-02,  ..., 3.5297e-06, 3.3763e-05,
         1.6773e-03],
        ...,
        [1.2602e-06, 6.3579e-07, 5.5026e-07,  ..., 9.5436e-03, 1.5581e-04,
         4.3279e-06],
        [1.2367e-04, 8.0943e-05, 7.6152e-05,  ..., 7.5040e-03, 1.8992e-01,
         7.3471e-04],
        [6.7166e-05, 6.6564e-05, 6.7489e-05,  ..., 3.9130e-03, 7.0941e-03,
         8.4237e-02]]), 'X': tensor([[ -9.2687,  -2.9794,   0.6436],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.5149,  -5.1905,   1.5239],
        [  4.9152,  12.8653,  -1.4459],
        [  9.1409,  -7.5789,   1.5814],
        [  6.1372,   2.6726,  -6.7885],
        [  9.3424,  -7.3685,  -3.6807],
        [  5.1667,  -2.6543,   1.2879],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.4732,  -2.2041,  -4.3830],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.4430,  -0.1897,   4.8383],
        [  2.6613,   7.0021,   4.5783],
        [  5.1472,  -2.6801,   8.1194],
        [  6.3505,   0.6235,  -8.5849],
        [  4.7544,  -3.1425,  -1.5315],
        [ -1.9487,   7.8590,  -3.3330],
        [  2.5318,  13.7231,   6.3692],
        [ 10.9931,   7.2107,   7.7933],
        [  2.4356,  -4.5942,  -5.0808],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -1.5408,   7.8978,   5.3470],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.0608,   3.9362,  -5.9567],
        [  5.8617,   3.4853,  -9.8080],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.0534,  -6.0429,   2.4583],
        [  0.9659,  -4.8144,  -0.9594],
        [  8.8474,  10.0868,  -8.8644],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.1178,   1.5784,   2.1410],
        [  1.7659,   7.4127,   4.7871],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -0.8885,   7.9160,   0.0571],
        [ -5.6174,   6.4708,  -1.7339],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.2345,  -3.1345,   3.9715],
        [ 11.2767,  -4.6006,   0.7868],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.2576,   2.1359,  -3.3398],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 11.7794,   5.5086,  -9.6929],
        [ 11.6634,   5.8050,   9.1590],
        [ -8.1666,   3.7653,   3.6075],
        [ -1.7009,   7.8851,  -8.2437],
        [  1.7122,  -4.7591,  -3.6734],
        [  5.6202,  -1.9508,   0.7471],
        [  3.4920,  -4.1366,  -8.3955],
        [  3.3943,   6.5483,   1.0790],
        [  1.6422,   7.4586,  -6.8056],
        [  3.2226,  -4.2802,  -4.0428],
        [ 10.2470,   8.4054,  -0.1177],
        [ -3.3999, -10.1152,  -1.2088],
        [  7.9816,  10.8938,  -0.5378],
        [ -5.9872,   6.2037,  -9.4477],
        [  2.3376,  -4.6231,   9.8436],
        [  1.0391,  -4.8141,  -0.0768],
        [ -4.4520,   7.1370,  -8.7867],
        [ -7.4602,   4.7718,  -4.6277],
        [ -1.6906,   7.8860,   7.7142],
        [  4.7482, -10.3907,   5.9149],
        [  0.1532,  -4.7422,  -7.0145],
        [ 10.6673,  -5.6649,  -4.1646],
        [  5.1196,  -2.7160,   4.2705],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -7.6650,   4.5092,   0.2520],
        [  3.4140,  -4.1803,  -8.8224],
        [  2.4356,  -4.5942,  -5.0808],
        [  3.1892, -10.8220,  -6.8409],
        [ -3.3999, -10.1152,  -1.2088],
        [ 11.9641,  -2.9568,   7.7211],
        [  8.7715,  -7.9390,   5.2900],
        [ 11.9695,  -2.9407,  -6.3498],
        [  8.3718,  -8.2953,   8.4666],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.3858,   4.4260,  -9.6769],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.0087,   4.0173,   9.1091],
        [ -7.8126,   4.3066,   5.1791],
        [  5.8323,  -1.5228,   5.1132],
        [  9.4554,   9.4245,   9.8800],
        [ 10.2712,   8.3707,  -4.9743],
        [ -8.1204,  -5.9308,   4.0761],
        [ -9.4296,  -1.9458,   4.0833],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -9.4157,   0.0821,  -5.0025],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -4.0868,   7.2981,   8.1874],
        [  0.9788,  -4.8145,  -3.8661],
        [  6.8704,  -9.3809,   7.5769],
        [  2.1477,   7.2550,   8.9688],
        [  9.9750,  -6.6338,  -1.2440],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.4703,  -4.2006,   7.4086],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.0047,   3.1062,  -6.7914],
        [  4.2275,  -3.6320,  -0.2714]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0197, 0.0322,  ..., 1.1851, 1.0294, 0.8993],
        [0.0197, 0.0000, 0.0168,  ..., 1.1980, 1.0369, 0.9125],
        [0.0322, 0.0168, 0.0000,  ..., 1.1937, 1.0276, 0.9086],
        ...,
        [1.1851, 1.1980, 1.1937,  ..., 0.0000, 0.3941, 0.2863],
        [1.0294, 1.0369, 1.0276,  ..., 0.3941, 0.0000, 0.3555],
        [0.8993, 0.9125, 0.9086,  ..., 0.2863, 0.3555, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.73579738,  9.95552626, 10.00794468,  9.97964013, 10.01804223,
        9.97220799, 13.77222752, 11.87412439,  6.69389431, 11.89855267,
        5.80860165,  9.94173708,  5.90035324, 10.06286074,  9.44486609,
        7.49076899,  5.80312949,  6.38105015,  5.69913817,  8.09704052,
       13.95472331, 13.14690137,  5.19986797, 10.5575601 ,  8.04664886,
       11.76830981,  8.97052115,  6.81961996, 10.7167285 , 10.06850677,
        4.91038884, 13.41713661,  5.79391825,  9.25336371,  7.62011734,
        9.91668182,  7.96575782,  8.56889738,  4.78258092,  5.1224079 ,
        9.75200848, 12.17901033, 12.13568598,  6.61211224, 12.18045978,
       13.00380624, 13.02818702,  8.99276564,  8.06643487,  5.05774295,
        5.94909709,  5.41348121,  7.37576154,  7.63726134,  5.35774573,
       13.25336001, 10.67132524, 13.50484539,  8.62162372,  5.1805345 ,
        4.92497514,  8.41171369,  8.85572292,  8.06517143, 11.42420508,
        4.74468406, 12.07818871,  5.79543038, 13.08414689,  9.66977802,
        8.89301516,  5.39723241,  5.19986797, 11.28215429, 10.67132524,
       12.32408516, 11.83075005, 12.32545637, 11.785566  , 12.00995898,
        6.97106912,  4.91652556,  8.95983586,  8.92098752,  6.02779201,
       13.35013374, 13.25017211, 10.05558893,  9.6282763 , 12.12448912,
        9.41605684, 12.11735334, 10.37698435, 12.05395056,  8.36445758,
        4.91296205, 11.62770332,  7.56617037, 11.97949273, 12.1460931 ,
       13.04217487,  5.05963217, 12.21532282, 13.1471313 ,  6.76056625,
        5.57341623])}
{'P': tensor([[1.0824e-02, 2.9486e-01, 2.2209e-01,  ..., 9.6705e-08, 2.6100e-04,
         2.0966e-07],
        [5.9937e-05, 7.6555e-03, 2.8837e-01,  ..., 5.6523e-08, 2.4252e-04,
         1.7093e-07],
        [9.6651e-05, 2.4297e-04, 1.5178e-02,  ..., 1.0554e-07, 4.9286e-04,
         3.5486e-07],
        ...,
        [2.2774e-04, 1.0071e-04, 7.6546e-05,  ..., 2.1650e-01, 3.9837e-05,
         2.6748e-05],
        [8.5658e-05, 9.8219e-05, 1.0107e-04,  ..., 6.2801e-05, 9.1721e-02,
         4.1283e-04],
        [5.0532e-06, 2.5693e-06, 2.0822e-06,  ..., 8.5203e-02, 5.7442e-06,
         2.6453e-02]]), 'X': tensor([[ -9.2373,  -3.1225,  -8.9763],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -0.9715, -10.8630,   0.3555],
        [ -8.9788,   2.0208,  -3.2867],
        [  9.1409,  -7.5789,   1.5814],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.3048,   0.1469,  -9.1691],
        [ 12.5650,   2.0132,   4.6272],
        [ -8.8154,   2.4611,   2.5198],
        [  5.4167, -10.1293,   6.2156],
        [  5.1861,   4.7361,  -3.7312],
        [ -7.9298,   4.1367,  -3.9773],
        [ 11.3118,   6.5957,  -4.5667],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -7.4807,   4.7463,   2.0616],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.1892, -10.8220,  -6.8409],
        [ -0.9715, -10.8630,   0.3555],
        [  4.5767,   5.5071,   3.9069],
        [ -8.9624,  -4.0944,  -4.9622],
        [  4.0362,  13.2397,  -7.0030],
        [ -0.8885,   7.9160,   0.0571],
        [  0.6363, -11.0348,   6.1414],
        [ -3.3116, -10.1539,   7.5919],
        [  3.1480,  -4.3165,  -7.8002],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.3384,  -2.6161,  -4.7630],
        [  0.4119,  14.1019,  -4.1870],
        [  2.2437,  -4.6489,   6.1430],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 11.9641,  -2.9568,   7.7211],
        [  3.3474,  -4.2162,   5.2455],
        [  6.1755,   2.5233,  -2.0042],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.1755,   2.5233,  -2.0042],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.0937,  13.8293,  -2.3533],
        [  6.0955,   2.8210,  -4.3321],
        [ -0.7599, -10.8995,  -6.0366],
        [ -6.0175,  -8.4541,  -9.8408],
        [  1.7659,   7.4127,   4.7871],
        [ -7.0825,   5.2073,  -5.0497],
        [ -6.8618,  -7.6322,   3.7027],
        [  2.1440, -10.9811,   8.5281],
        [  7.5064,  -8.9651,  -0.0994],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.1984,  -0.4327,   1.5438],
        [  2.1379,  -4.6757,   6.4981],
        [ -8.0766,  -6.0046,  -3.1873],
        [  0.8572,  14.0501,  -8.3364],
        [  9.2979,   9.6048,   7.3138],
        [  5.1472,  -2.6801,   8.1194],
        [ -5.1369,   6.7755,   8.3836],
        [  8.0381,  -8.5690,  -8.9523],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -4.7389,   6.9953,   3.7638],
        [ -1.9646,   7.8571,   3.3817],
        [  1.1006,   7.6318,  -2.4877],
        [  9.5261,  -7.1675,   1.4206],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.2692,  -1.8882,   3.9253],
        [  0.9809,   7.6642,  -1.4217],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.4901,  -0.7052,   1.7315],
        [  1.9312,  13.8649,   8.3563],
        [ -8.9606,  -4.0994,   1.7886],
        [ -5.3179,  -9.0101,   9.4472],
        [  6.3534,   1.2990,  -7.2942],
        [ -5.3736,   6.6310,  -1.5174],
        [  9.6614,   9.1784,  -1.2087],
        [  7.8982,  -8.6778,   0.0239],
        [  1.9312,  13.8649,   8.3563],
        [  4.4915,   5.5990,   7.0405],
        [ -5.2589,   6.7024,  -5.5809],
        [ -1.6334, -10.7207,   7.1397],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -4.3359,  -9.6431,  -9.0616],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.8994,   6.8675,  -9.5987],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.0766,  -6.0046,  -3.1873],
        [  5.7218,  -9.9934,   6.5764],
        [ -7.4733,   4.7555,   6.2490],
        [  1.7301,   7.4262,   5.0522],
        [ -9.3087,   0.7792,  -7.5587],
        [  4.5767,   5.5071,   3.9069],
        [  4.4290,  -3.4596,   1.4181],
        [ -6.9838,  -7.4965,   0.9796],
        [  3.2159, -10.8166,  -1.0406],
        [ -9.3464,   0.5708,  -1.6397],
        [  3.3633,  13.4794,   0.5193],
        [ 12.5258,  -0.4200,  -9.3493],
        [  2.6321,  -4.5296,  -2.5159],
        [  1.6308,  13.9254,  -2.0984],
        [  1.4306,  13.9620,  -0.2075],
        [  1.6232,  -4.7717,  -2.5655],
        [  9.0652,   9.8597,  -3.5949]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0222, 0.0171,  ..., 1.3070, 1.0059, 1.3116],
        [0.0222, 0.0000, 0.0208,  ..., 1.3266, 1.0249, 1.3279],
        [0.0171, 0.0208, 0.0000,  ..., 1.3070, 1.0049, 1.3071],
        ...,
        [1.3070, 1.3266, 1.3070,  ..., 0.0000, 0.3090, 0.3476],
        [1.0059, 1.0249, 1.0049,  ..., 0.3090, 0.0000, 0.3860],
        [1.3116, 1.3279, 1.3071,  ..., 0.3476, 0.3860, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.75075273,  9.93553424,  9.99145096, 10.00410625, 10.27716434,
       10.30972193, 10.90638066,  9.20340031, 11.87412439, 13.1399181 ,
        6.30647746, 12.72524615,  9.15253344, 11.48663767,  7.02326515,
        8.94395628, 13.09426142, 10.51437947,  8.85939626, 11.9120321 ,
       11.28215429, 10.90638066,  7.16058587,  9.85330646, 13.84126197,
        7.96575782, 11.05317597, 10.68031351,  5.34250225, 10.33958639,
        9.69792513, 14.10796466,  5.16205087,  9.97822473, 12.32408516,
        5.38342077,  6.67108662, 11.83509372,  9.11473391,  6.67108662,
        5.22903117, 13.98691426,  6.7166297 , 10.92597196, 10.37698435,
        7.62011734,  8.7907923 , 10.26327895, 11.18839169, 11.69264243,
        9.34061938,  6.2134961 ,  5.14124612, 10.06407839, 14.07623266,
       13.36800073,  5.80312949,  8.50268322, 11.74901778,  9.69629   ,
        8.44940661,  8.09899874,  7.71075226, 11.92133129,  9.6282763 ,
        9.61942445, 12.41366754,  7.72668467,  9.88571713, 10.46739073,
       12.5769659 , 12.50996828, 13.99877413,  9.85384623, 10.46238054,
        6.48486171,  8.53501777, 13.32613445, 11.73397309, 13.99877413,
        7.17789666,  8.51928531, 10.84437705,  9.75075273, 10.57302546,
        5.39450305,  5.8828986 ,  7.4544866 , 13.94008948, 10.06407839,
       11.51557166,  8.8580671 ,  7.62509582,  9.34127013,  7.16058587,
        5.62005629, 10.24557047, 11.2845608 ,  9.36378438, 13.89264822,
       12.53285633,  5.23878371, 14.02058621, 14.03505624,  5.04028538,
       13.39372426])}
{'P': tensor([[1.3138e-02, 2.2206e-01, 2.0976e-01,  ..., 3.5241e-06, 3.1290e-05,
         3.2411e-06],
        [4.4691e-04, 9.5217e-02, 1.7434e-01,  ..., 1.7368e-05, 4.1127e-04,
         2.2934e-05],
        [5.3091e-04, 4.9028e-02, 1.1310e-01,  ..., 2.0109e-05, 4.6766e-04,
         2.7931e-05],
        ...,
        [1.9167e-03, 6.0232e-04, 5.6954e-04,  ..., 6.5210e-02, 8.5669e-05,
         2.9763e-04],
        [4.8301e-05, 3.8593e-05, 3.7122e-05,  ..., 2.8403e-03, 1.0724e-01,
         1.1288e-03],
        [2.4677e-04, 9.5629e-05, 9.7349e-05,  ..., 3.9387e-03, 3.6593e-05,
         5.1757e-02]]), 'X': tensor([[ -9.2345,  -3.1345,   3.9715],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.0965,  -5.9712,   4.3466],
        [ -8.0611,  -6.0302,   4.6962],
        [ -8.0829,  -5.9940,   5.0564],
        [ -7.9925,  -6.1423,   2.8990],
        [ -6.4464,   5.8272,   7.5038],
        [ -1.5261,   7.8988,   9.3768],
        [  3.1064,  13.5607,   6.7605],
        [  9.2811,  -7.4336,   5.1771],
        [  6.9956,  11.6590,  -7.6876],
        [  7.8982,  -8.6778,   0.0239],
        [  2.2659,  13.7894,   6.5436],
        [ -2.0721,   7.8432,   6.6380],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.9026,   3.3837,  -0.6573],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.7078,   2.7182,   1.8146],
        [  4.9531,   5.0576,   5.6794],
        [  5.1886,  -2.6250,   6.4677],
        [  5.8836,   3.4315,  -4.0327],
        [ -0.5385,   7.9037,   6.3830],
        [ -7.5720,   4.6310,   2.1717],
        [ -3.4100,   7.5436,  -7.6980],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.1504,  -4.6726,   8.7908],
        [ 11.4235,   6.3594,   5.1663],
        [  5.6260,  -1.9401,   3.0259],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.7784,  -1.6401,   9.5587],
        [ -7.8280,   4.2848,  -7.1955],
        [ -7.4808,   4.7462,   2.0266],
        [ -1.6275, -10.7221,   3.2376],
        [  4.5954,  13.0099,  -4.4439],
        [  8.9284,  10.0036,  -6.9210],
        [ 12.5650,   2.0132,   4.6272],
        [ -4.0868,   7.2981,   8.1874],
        [  3.5939,   6.4029,  -1.8357],
        [ -7.0948,   5.1940,   9.8171],
        [ -2.5056,   7.7720,   5.5878],
        [  5.6258,  -1.9404,  -6.0715],
        [ -9.4296,  -1.9458,   4.0833],
        [ -2.3317,   7.8034,  -3.6274],
        [  7.3592,  11.3937,   6.2407],
        [  1.9505,  -4.7171,   9.9868],
        [  0.9074,  -4.8139,   8.1674],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.4336,  -2.2676,   5.6933],
        [  5.6928,  12.4710,   3.6001],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.8745,   2.3097,   3.5600],
        [ 10.0995,  -6.4739,   2.0974],
        [ -1.6086,   7.8928,   4.3385],
        [ -9.1706,   1.3863,   8.4032],
        [  3.4604,  -4.1545,   1.8091],
        [  1.5383, -11.0274,  -4.9666],
        [ -4.2271,   7.2387,  -1.8316],
        [  1.9312,  13.8649,   8.3563],
        [ 12.0582,  -2.6640,   0.7132],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.9808,   4.9340,  -3.3715],
        [  0.6470, -11.0352,  -1.1931],
        [  5.8318,   3.5567,  -4.7692],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.3453,  -1.5462,   7.0212],
        [  5.7845,   3.6652,   0.9175],
        [  2.1625,  -4.6697,   7.0658],
        [  1.7232,  -4.7574,   3.1270],
        [  4.1537,   5.9340,  -6.0423],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.1348,   5.9514,  -0.8856],
        [  2.6310, -10.9195,   3.8054],
        [  6.3535,   0.6792,  -7.5101],
        [  5.6095,  -1.9702,  -7.8233],
        [ -6.2668,   5.9808,   2.5614],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.3231,  -4.7691,  -1.6120],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -2.1817,   7.8274,  -4.6298],
        [ 10.9931,   7.2107,   7.7933],
        [  1.7000,  -4.7609,   2.7935],
        [ -3.5029,   7.5138,   8.1853],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.3534,   1.2990,  -7.2942],
        [ -5.9987,   6.1949,  -6.2114],
        [ 11.5083,  -4.1177,   0.2389],
        [ -2.6069, -10.4298,  -6.1995],
        [  4.5814, -10.4485,   2.5496],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.1931,  -4.7491,   7.4385],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.5158,  -2.1337,   3.5246],
        [  5.4732,  -2.2041,  -4.3830],
        [  1.4150,  13.9647,  -4.0915],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.6248,   3.9984,   8.4933],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.1787,  -2.6383,  -4.2533],
        [ -5.0132,  -9.2234,  -8.3531],
        [ 11.3800,  -4.3919,   1.4066],
        [ -1.6220,   7.8917,   0.0315],
        [  5.8325,  -1.5222,  -2.1341],
        [  2.5620,   7.0549,   8.4825]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.7880e-03, 2.8003e-02,  ..., 6.9005e-01, 1.0221e+00,
         7.9793e-01],
        [8.7880e-03, 3.0339e-04, 2.1074e-02,  ..., 6.9839e-01, 1.0305e+00,
         8.0205e-01],
        [2.8003e-02, 2.1074e-02, 0.0000e+00,  ..., 7.0820e-01, 1.0402e+00,
         7.9632e-01],
        ...,
        [6.9005e-01, 6.9839e-01, 7.0820e-01,  ..., 0.0000e+00, 3.3210e-01,
         5.6423e-01],
        [1.0221e+00, 1.0305e+00, 1.0402e+00,  ..., 3.3210e-01, 0.0000e+00,
         7.1133e-01],
        [7.9793e-01, 8.0205e-01, 7.9632e-01,  ..., 5.6423e-01, 7.1133e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.75200848, 10.05558893, 10.06023665, 10.06703358, 10.06286074,
       10.08001894,  8.68979052,  8.04483495, 13.91198191, 11.89105259,
       13.59673406, 11.73397309, 13.97430191,  8.11227668,  6.02789634,
        6.80370254, 11.62158554,  9.12220271,  7.07902036,  5.81481044,
        6.81117802,  7.92200342,  8.87587457,  8.27853076,  8.35856487,
        5.14371005, 13.0743399 ,  5.95111804, 12.79126785,  6.00663314,
        8.92395381,  8.859416  , 10.84493732, 13.79762415, 13.40850328,
       12.72524615,  8.36445758,  7.34250948,  8.79284466,  8.16584537,
        5.9510489 ,  9.6282763 ,  8.14433542, 13.56367857,  5.10446653,
        4.89869205, 14.09176259,  9.85384623,  5.88782913, 13.70893492,
        8.12043983,  9.17016857, 11.99633819,  8.0550286 ,  9.27474386,
        5.40689332, 11.1341798 ,  8.38252937, 13.99877413, 12.3489324 ,
        8.53014915,  9.41244865, 12.95702647, 11.05413545,  6.83083863,
        8.83725081, 12.67576792, 12.44177641,  6.84796202,  5.14607014,
        5.05989547,  7.24326992, 10.01804223,  7.24677664, 11.23201619,
        6.38968357,  5.94541803,  8.66273171, 11.89855267,  4.78002665,
        9.88571713,  8.12580976, 13.14690137,  5.05534609,  8.29022197,
       12.79126785,  6.48486171,  8.62329466, 12.22276278, 10.7506423 ,
       11.4088043 ,  8.83341608,  4.75303127,  9.54154096,  5.91408504,
        5.90035324, 14.03618649, 12.3489324 ,  6.90115901,  9.71392179,
        5.81199422, 10.4977132 , 12.19805394,  8.05668609,  6.02789634,
        7.50564223])}
{'P': tensor([[7.5798e-03, 2.9084e-01, 2.5048e-01,  ..., 6.9382e-06, 6.1071e-06,
         1.1441e-06],
        [1.9258e-04, 1.7952e-02, 2.1628e-04,  ..., 1.8763e-05, 1.6925e-05,
         2.8770e-06],
        [2.4842e-04, 4.5363e-01, 1.0601e-02,  ..., 9.9714e-06, 8.6294e-06,
         1.5770e-06],
        ...,
        [5.6704e-05, 4.5396e-05, 5.3199e-05,  ..., 5.4200e-02, 6.7161e-04,
         1.3665e-03],
        [1.8076e-05, 1.5889e-05, 1.7030e-05,  ..., 4.8949e-03, 6.6624e-02,
         9.0506e-02],
        [3.6505e-05, 2.8868e-05, 3.3290e-05,  ..., 1.0402e-03, 1.3209e-03,
         1.0778e-01]]), 'X': tensor([[-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0115, 0.0087,  ..., 0.9961, 1.2328, 1.2373],
        [0.0115, 0.0000, 0.0061,  ..., 1.0067, 1.2442, 1.2487],
        [0.0087, 0.0061, 0.0000,  ..., 1.0012, 1.2403, 1.2451],
        ...,
        [0.9961, 1.0067, 1.0012,  ..., 0.0000, 0.5376, 0.6109],
        [1.2328, 1.2442, 1.2403,  ..., 0.5376, 0.0000, 0.0891],
        [1.2373, 1.2487, 1.2451,  ..., 0.6109, 0.0891, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.75401449,  9.86089579,  9.78147542,  9.79039371, 10.06407839,
       10.0674887 ,  9.69792513, 10.00794468,  6.00732193,  5.14124612,
       13.89671314,  6.76791483, 10.54658597,  5.29009836,  4.8763135 ,
        9.66977802,  5.84533345,  9.17595899,  5.39450305, 12.04924159,
        5.64500512,  4.92497514, 11.22142195, 12.04924159, 10.61472636,
       12.70489563,  6.54127431,  9.18690018,  8.1137452 , 12.1494023 ,
        5.36061974, 13.89974693,  9.15976746, 11.83570132, 13.15525803,
        6.69389431, 11.66453059,  7.33350642, 12.18998195,  4.79003361,
        4.91652556, 10.65929968, 10.92597196,  9.75075273,  8.04483495,
       10.65929968,  6.10774544,  8.90405823,  9.04255655,  9.18690018,
        6.74686071, 13.96987806, 11.92133129,  8.81930525,  7.69752262,
       11.77124373,  6.00663314,  7.52117132,  7.60529892, 12.17901033,
        5.16611782, 13.15525803, 14.09176259,  6.76056625, 13.91198191,
        7.822854  ,  7.8575648 ,  7.92607342,  7.85919687,  7.46248947,
        9.23071698,  7.63726134, 11.97238624, 12.32408516, 11.22142195,
        6.30647746,  6.39438044, 12.42554808, 13.13888841, 10.00802627,
        5.14124612, 11.00812996, 13.77604293, 11.46176259, 13.0743399 ,
        7.67939546,  5.05989547, 11.28093507, 10.07238463,  9.75075273,
       10.66631784, 14.03505624, 12.32408516,  9.43112957, 13.103709  ,
        7.42512621, 13.0743399 , 14.09176259, 12.25806559,  8.1303945 ,
        8.62631801, 13.35832725,  9.91668182,  6.652742  ,  6.21275016,
        6.90115901])}
{'P': tensor([[8.1266e-03, 3.3570e-01, 2.0868e-01,  ..., 2.7990e-05, 2.5206e-07,
         2.2501e-05],
        [1.1337e-04, 9.5132e-03, 3.2559e-01,  ..., 3.8447e-05, 3.2497e-07,
         1.9256e-05],
        [2.1323e-04, 3.1677e-04, 2.9955e-02,  ..., 1.6120e-04, 1.2248e-06,
         4.0070e-05],
        ...,
        [3.6902e-05, 3.3899e-05, 3.1839e-05,  ..., 6.4228e-02, 7.1986e-03,
         7.1953e-05],
        [9.9608e-06, 7.4292e-06, 5.1165e-06,  ..., 5.2581e-05, 6.4295e-02,
         7.1625e-05],
        [1.4723e-01, 1.0318e-01, 6.0202e-02,  ..., 3.6002e-05, 6.3743e-07,
         2.5079e-02]]), 'X': tensor([[ -9.1960,  -3.2964,   1.6921],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 10.4737,  -5.9570,  -1.2108],
        [  4.5340,  -3.3629,   8.9652],
        [  9.1409,  -7.5789,   1.5814],
        [  1.1274,  14.0116,   8.1753],
        [ 11.5198,  -4.0923,   5.5088],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.8447,  -4.7372,   8.1235],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.5057,  -3.3894,  -0.8160],
        [ -5.1046,  -9.1610,   2.3418],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.2328,  -2.0376,  -1.1234],
        [  5.6414, -10.0303,   7.8525],
        [ -8.8934,   2.2597,   3.5831],
        [ -8.8934,   2.2597,   3.5831],
        [  5.1203,  -2.7151,   6.2110],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 10.0995,  -6.4739,   2.0974],
        [ -2.8162,   7.7059,  -9.6885],
        [ 11.7795,  -3.4697,   3.5624],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.3236,   0.3011,  -4.6668],
        [  5.1196,  -2.7160,   4.2705],
        [  4.0421,  -3.7765,  -8.3631],
        [  4.9531,   5.0576,   5.6794],
        [ 12.6035,   0.7494,   7.1536],
        [  6.3305,   1.6119,   1.1538],
        [  5.8930,  12.3592,  -8.9201],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.3124,   0.2057,  -0.4593],
        [  0.9659,  -4.8144,  -0.9594],
        [  2.7420,  -4.4896,  -3.7139],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -1.1885, -10.8211,   0.0865],
        [  3.4140,  -4.1803,  -8.8224],
        [  6.3301,   0.3638,  -7.5685],
        [  9.9756,  -6.6330,  -3.3429],
        [ -1.6220,   7.8917,   0.0315],
        [  1.3403,  -4.8011,  -1.1698],
        [  0.5250, -11.0306,   5.5358],
        [  8.2154,  -8.4262,   8.0171],
        [  6.0345,   3.0175,  -1.4424],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.5770,   1.8512,  -0.6578],
        [ -9.2345,  -3.1345,   3.9715],
        [ -7.6293,  -6.6859,  -5.5161],
        [  3.2692,  13.5098,  -4.9032],
        [ -1.6275, -10.7221,   3.2376],
        [  6.1737,   2.5305,   3.9049],
        [  6.3124,   0.2057,  -0.4593],
        [ 12.4837,   2.7510,   0.4449],
        [  3.4920,  -4.1366,  -8.3955],
        [  3.3167,   6.6022,   6.3554],
        [ -6.4844,  -8.0230,   8.5209],
        [  6.3515,   0.6418,  -6.5823],
        [  9.5018,  -7.1945,  -1.1892],
        [  4.1637,  13.1898,  -1.9189],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.9033,   2.2329,  -8.8912],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.7685, -10.8982,  -6.8928],
        [  1.3403,  -4.8011,  -1.1698],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.1644, -11.0394,   7.8511],
        [  3.2701,   6.6338,   9.9209],
        [  6.1222,  -0.7261,   8.8204],
        [  3.1892, -10.8220,  -6.8409],
        [ 12.1504,   4.3674,  -7.2998],
        [ -6.7992,   5.4991,   9.9186],
        [  5.8325,  -1.5222,  -2.1341],
        [ -1.7009,   7.8851,  -8.2437],
        [ -1.0012, -10.8576,   2.2867],
        [  5.9587,  12.3216,   9.1695],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.1625,  -4.6697,   7.0658],
        [ -7.5401,   4.6718,   9.0033],
        [  9.2811,  -7.4336,   5.1771],
        [  6.2905,   1.9366,   2.1169],
        [ -1.1885, -10.8211,   0.0865],
        [  3.0876,   6.7527,  -5.3540],
        [  9.0652,   9.8597,  -3.5949],
        [  4.4633,   5.6287,  -0.7754],
        [ -1.1885, -10.8211,   0.0865],
        [  3.2701,   6.6338,   9.9209],
        [  3.4604,  -4.1545,   1.8091],
        [  0.7145,  -4.8071,  -2.3766],
        [  1.1994,   7.6035,   2.4295],
        [  6.3305,   1.6119,   1.1538],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.2182,  13.8007,   5.1889],
        [  5.3924, -10.1396,  -4.0863],
        [  5.8325,  -1.5222,  -2.1341],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.1178,   1.5784,   2.1410]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.0463e-02, 2.0101e-02,  ..., 9.7183e-01, 1.1991e+00,
         1.2549e-01],
        [2.0463e-02, 0.0000e+00, 2.3821e-02,  ..., 9.9209e-01, 1.2185e+00,
         1.4255e-01],
        [2.0101e-02, 2.3821e-02, 0.0000e+00,  ..., 9.8086e-01, 1.2113e+00,
         1.4226e-01],
        ...,
        [9.7183e-01, 9.9209e-01, 9.8086e-01,  ..., 0.0000e+00, 3.0536e-01,
         8.6294e-01],
        [1.1991e+00, 1.2185e+00, 1.2113e+00,  ..., 3.0536e-01, 0.0000e+00,
         1.0798e+00],
        [1.2549e-01, 1.4255e-01, 1.4226e-01,  ..., 8.6294e-01, 1.0798e+00,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.76896904,  9.85384623,  9.97220799,  9.99211663, 10.06850677,
       10.04916678,  5.05989547,  9.85330646, 12.04924159,  5.64500512,
       11.87412439, 14.05687815, 12.22503754, 12.17451685,  5.08372149,
        5.39098669,  9.60412735,  5.63824971, 10.48721382, 10.1371552 ,
       12.4013153 , 11.50791859,  9.17595899,  9.17595899,  5.79562691,
        9.54154096, 11.99633819,  8.20436764, 12.27991991,  9.09661212,
        6.33075682,  5.79543038,  5.53175275,  7.07902036, 12.62576026,
        6.53250937, 13.6922425 ,  8.16714836,  6.31576264,  4.91038884,
        5.26067019,  9.93553424, 10.88618114,  5.39723241,  6.34059021,
       11.97957425,  8.05668609,  4.98461983, 11.04313641, 11.76830981,
        6.74686071,  5.18047638, 12.7125079 ,  9.75200848, 10.14436912,
       13.89974693, 10.84493732,  6.67218687,  6.31576264, 12.78327037,
        5.41348121,  7.38844703, 10.31584102,  6.383897  , 11.9182956 ,
       13.83139024,  5.5513072 ,  9.17905491,  9.88310614,  5.46244988,
       11.24434374,  4.98461983,  5.24461068,  9.77588801, 11.10066126,
        7.39599986,  6.16512856, 11.28215429, 12.91143491,  8.74470218,
        6.02789634,  8.06643487, 10.9036226 , 13.68672855,  4.90505566,
        6.1561646 ,  5.14607014,  8.87007495, 11.89105259,  6.58184474,
       10.88618114,  7.42512621, 13.39372426,  7.18354885, 10.88618114,
        7.39599986,  5.40689332,  4.85995002,  7.69752262,  6.53250937,
        5.95297459, 13.97779839, 11.48434767,  6.02789634, 13.25336001,
        9.25336371])}
{'P': tensor([[6.9681e-03, 2.0692e-01, 1.8971e-01,  ..., 2.3806e-06, 2.3803e-04,
         9.3533e-04],
        [1.2472e-04, 1.0101e-02, 3.7889e-01,  ..., 4.2373e-06, 4.2582e-04,
         2.3259e-03],
        [1.0263e-04, 1.8745e-04, 1.3698e-02,  ..., 6.0297e-06, 7.1251e-04,
         3.7308e-03],
        ...,
        [2.3533e-05, 2.0633e-05, 1.7326e-05,  ..., 6.1027e-02, 5.2426e-05,
         1.5595e-05],
        [6.8535e-05, 7.1015e-05, 7.6201e-05,  ..., 4.2488e-04, 6.5288e-02,
         1.5010e-04],
        [2.3845e-06, 3.3160e-06, 4.0897e-06,  ..., 8.5379e-05, 3.0152e-04,
         1.0991e-02]]), 'X': tensor([[-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0174, 0.0124,  ..., 1.2260, 0.9571, 0.4966],
        [0.0174, 0.0000, 0.0093,  ..., 1.2283, 0.9686, 0.4906],
        [0.0124, 0.0093, 0.0000,  ..., 1.2338, 0.9690, 0.4990],
        ...,
        [1.2260, 1.2283, 1.2338,  ..., 0.0000, 0.6883, 0.8207],
        [0.9571, 0.9686, 0.9690,  ..., 0.6883, 0.0000, 0.8455],
        [0.4966, 0.4906, 0.4990,  ..., 0.8207, 0.8455, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.7712321 ,  9.88571713,  9.96152788, 10.06437598,  9.85330646,
        9.91668182,  9.09963894,  7.16058587,  7.06272256, 10.14044489,
        6.53250937, 11.63765273,  5.13182506,  6.19707944, 11.63724792,
        9.15253344, 10.37998078, 11.38774149,  4.85603886, 10.68416401,
        5.92281142,  9.38232121, 13.32557133,  8.77913428, 12.4980845 ,
        9.15976746,  4.7663526 ,  8.99791316,  8.72896381,  5.18277858,
        6.2134961 ,  4.98461983, 11.35182589,  6.58184474,  5.9510489 ,
       10.31584102,  6.97106912, 11.31276109, 11.76830981, 14.10796466,
        5.18277858,  7.50917551,  6.10360073,  7.50035355,  9.85330646,
        6.54693987, 10.52587243,  7.62011734, 11.42420508,  5.43943791,
       11.9120321 , 10.46739073, 13.41713661, 11.1341798 ,  8.87007495,
       11.62158554,  5.21280156, 10.8992524 ,  6.02588517,  6.99097013,
       10.88618114,  9.15866607,  9.27474386,  5.2287058 , 13.59170809,
        5.43544626, 13.89886469,  8.36995953, 11.37939695,  6.89245352,
        8.14433542,  5.10444097, 12.59840193,  7.92644746,  9.36093562,
        9.76896904,  9.48337063, 13.62896593,  9.60412735,  7.58378655,
       11.32149259,  9.04255655,  5.41348121, 10.90638066, 13.66991027,
       12.88270018,  9.51074763, 12.4217523 , 13.46553623, 12.04924159,
        4.78544875, 12.05395056,  9.7751189 ,  5.64865173, 13.70893492,
       10.09979747,  6.81084724,  5.1805345 ,  6.35029274, 12.1494023 ,
        7.69752262, 10.51437947, 11.8159254 ,  6.67121748,  5.22903117,
       10.87409005])}
{'P': tensor([[6.8160e-03, 2.8851e-01, 1.6877e-01,  ..., 5.0717e-03, 7.2393e-06,
         1.7428e-06],
        [9.8958e-05, 8.0887e-03, 2.7684e-01,  ..., 8.8448e-03, 9.9784e-06,
         1.9149e-06],
        [2.6210e-04, 3.6474e-04, 3.4492e-02,  ..., 6.8321e-02, 5.5532e-05,
         7.4946e-06],
        ...,
        [1.1159e-05, 1.3635e-05, 1.8886e-05,  ..., 2.8389e-02, 2.9885e-04,
         6.8542e-06],
        [3.1821e-05, 2.9488e-05, 2.7310e-05,  ..., 2.5113e-05, 9.1219e-02,
         4.4452e-03],
        [6.3721e-04, 4.7962e-04, 3.1781e-04,  ..., 2.6053e-05, 5.0794e-05,
         3.8214e-02]]), 'X': tensor([[ -9.1813,  -3.3550,   1.8827],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.1531,  -5.8748,   2.6934],
        [ -7.4733,   4.7555,   6.2490],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.2902,   0.0451,   3.7116],
        [  9.2811,  -7.4336,   5.1771],
        [  2.1504,  -4.6726,   8.7908],
        [  6.5428,  -9.5728,  -2.9897],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.4049,   5.6892,  -4.4394],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.8323,  -1.5228,   5.1132],
        [ -1.6906,   7.8860,   7.7142],
        [  4.8432,   5.1968,   1.5006],
        [ -4.0409,   7.3169,  -3.5779],
        [  9.6816,   9.1536,  -5.8543],
        [ -9.4430,  -0.1897,   4.8383],
        [  1.6954,  -4.7616,   9.8198],
        [  6.2036,   2.4030,  -1.2500],
        [  9.4554,   9.4245,   9.8800],
        [  1.6408,   7.4591,  -4.2294],
        [  1.7218,  -4.7576,  -3.3617],
        [ 12.5770,   1.8512,  -0.6578],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.2612, -10.5514,  -6.6268],
        [  6.3542,   0.6934,  -3.3840],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -9.4671,  -0.5553,   0.1117],
        [  6.3266,   0.3292,   9.3502],
        [ 11.4743,   6.2478,  -2.1100],
        [ -8.7078,   2.7182,   1.8146],
        [ 12.5224,  -0.4497,   4.8448],
        [  5.3602,  -2.3808,   7.2098],
        [ -7.4808,   4.7462,   2.0266],
        [  3.6817,   6.3354,   6.1233],
        [  6.2555,   2.1474,   4.1685],
        [  4.4049,   5.6892,  -4.4394],
        [ -7.0120,   5.2825,   7.7762],
        [ 12.5996,   1.3971,  -1.0694],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.4630,  -1.5145,   0.2983],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.4242, -10.1261,   3.9595],
        [ -9.4756,  -0.8166,  -9.6021],
        [  5.0233,   4.9647,  -3.1788],
        [  7.5064,  -8.9651,  -0.0994],
        [  2.0729,   7.2878,   7.0880],
        [ -0.8675, -10.8815,   6.8561],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.8201,   3.5841,  -8.0609],
        [  0.1497,   7.8334,  -5.4412],
        [  2.4572,   7.1085,   7.7376],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.8363,   2.4086,  -5.6296],
        [ 10.5057,  -5.9099,  -5.1599],
        [  0.2732,  -4.7619,  -1.1577],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.6451,  -4.5250,  -0.5565],
        [  3.8182,  -3.9347,  -1.7961],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.1625,  -4.6697,   7.0658],
        [  2.3200,   7.1756,   1.1938],
        [  0.1047, -11.0046,   0.5188],
        [  8.4691,  10.4571,  -5.0650],
        [  5.1793,   4.7461,   0.3899],
        [  6.0955,   2.8210,  -4.3321],
        [ -8.1905,   3.7253,   4.1175],
        [ 11.0373,   7.1299,  -8.5895],
        [ -0.7599, -10.8995,  -6.0366],
        [  6.2419,  -0.2313,   4.2440],
        [ -3.5903,   7.4847,  -8.9112],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.1350,  14.0104,   0.6321],
        [ 11.4495,   6.3026,   6.8257],
        [  2.2373,  -4.6506,  -1.7334],
        [ -4.2271,   7.2387,  -1.8316],
        [ -9.4157,   0.0821,  -5.0025],
        [ -1.5451,   7.8975,   7.6104],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.2508,   1.0572,   2.9978],
        [  0.3231,  -4.7691,  -1.6120],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.6050,   2.9451,  -0.7012],
        [  4.3554,   5.7393,  -9.0766],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.1706,   1.3863,   8.4032],
        [  4.9531,   5.0576,   5.6794],
        [  8.7131,  10.2216,   7.7915],
        [ -7.4602,   4.7718,  -4.6277],
        [ -0.5709,   7.9055,   7.6089],
        [ -2.6485, -10.4151,   0.8766],
        [  6.2365,  -0.2578,  -6.0212],
        [ -0.1963,   7.8765,  -1.1371]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0192, 0.0326,  ..., 0.1394, 1.0694, 0.7361],
        [0.0192, 0.0000, 0.0238,  ..., 0.1494, 1.0826, 0.7513],
        [0.0326, 0.0238, 0.0000,  ..., 0.1328, 1.0681, 0.7398],
        ...,
        [0.1394, 0.1494, 0.1328,  ..., 0.0000, 0.9356, 0.6111],
        [1.0694, 1.0826, 1.0681,  ..., 0.9356, 0.0000, 0.3579],
        [0.7361, 0.7513, 0.7398,  ..., 0.6111, 0.3579, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.7751189 ,  9.85384623,  9.97220799,  9.99211663, 10.06850677,
       10.04916678,  8.8580671 , 10.74409834,  6.29034958, 11.89105259,
        5.14371005, 11.59512194,  5.18047638,  7.19513806,  6.23262289,
        6.02779201,  8.06517143,  7.10378919,  8.35856487, 13.32374056,
        9.44486609,  5.05444288,  6.652742  , 13.35013374,  7.6374603 ,
        5.05963217, 12.7125079 ,  9.38232121, 11.37939695,  6.39187285,
        8.43626114, 12.49291768,  9.48337063,  6.33517349, 13.06498554,
        9.12220271, 12.53047489,  5.86519525,  8.859416  ,  7.32754882,
        6.61386436,  7.19513806,  8.77913428, 12.67680439, 13.86480175,
        9.58347456,  5.68942572,  9.48337063, 11.48734978,  9.51074763,
        7.06272256, 11.69264243,  7.57685908, 10.91601655,  5.63824971,
        6.83515201,  7.83486943,  7.52117132, 13.66051884, 12.27993098,
        9.99211663,  9.15866607, 12.05395056,  4.76969364,  9.88310614,
       10.8207297 ,  5.24137909,  5.48276038,  8.01009083,  5.14607014,
        7.54127425, 11.00508681, 13.45642379,  7.02496223,  6.7166297 ,
        8.99791316, 13.1399181 , 10.92597196,  6.24614905,  8.30125201,
        8.06990483, 14.05633441, 13.06957425,  5.16077764,  8.38252937,
        9.41605684,  8.04718433, 12.64235359,  9.3109884 ,  4.78002665,
       13.02818702, 12.19805394,  9.51074763, 10.74655346,  9.75200848,
        9.09502385,  7.20482224, 12.4013153 ,  9.27474386,  7.07902036,
       13.43127131,  8.85572292,  7.92607342, 10.74655346,  6.24187044,
        7.8788965 ])}
{'P': tensor([[8.3132e-03, 4.0137e-01, 1.6776e-01,  ..., 1.2398e-04, 8.8059e-06,
         3.1439e-03],
        [2.7005e-04, 1.5767e-02, 2.9027e-01,  ..., 2.4927e-04, 1.5721e-05,
         5.7252e-03],
        [3.5652e-04, 4.5052e-04, 3.1032e-02,  ..., 5.0455e-04, 1.4815e-05,
         3.8606e-03],
        ...,
        [1.1092e-04, 1.1323e-04, 1.0917e-04,  ..., 1.6486e-01, 1.1329e-04,
         2.3398e-04],
        [6.9769e-04, 7.4016e-04, 5.9233e-04,  ..., 8.0221e-04, 9.3409e-02,
         1.6585e-02],
        [2.6516e-04, 3.1044e-04, 6.6757e-04,  ..., 1.8353e-03, 3.2856e-05,
         2.2481e-02]]), 'X': tensor([[ -9.1795,  -3.3623,   8.5259],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.3535,  -5.5118,   9.8399],
        [ 10.6673,  -5.6649,  -4.1646],
        [  0.6363, -11.0348,   6.1414],
        [ -9.3464,   0.5708,  -1.6397],
        [  1.7815,  -4.7481,   4.4289],
        [  3.6817,   6.3354,   6.1233],
        [  8.1217,  -8.5024,  -0.6149],
        [ -6.6998,   5.5954,   0.0823],
        [  6.9653,  -9.3225,  -3.4727],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.5428,  -9.5728,  -2.9897],
        [ -5.9940,   6.1985,   5.9566],
        [ -5.7347,  -8.6908,   9.2878],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.2038,   2.4020,  -3.1692],
        [ -8.8363,   2.4086,  -5.6296],
        [ -1.9714,   7.8563,  -1.8084],
        [ -6.8466,  -7.6488,   1.9648],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.8617,   3.4853,  -9.8080],
        [  5.8889,  -1.3916,   8.8993],
        [  2.8602,   6.8905,   5.8779],
        [  3.2366,  -4.2732,   5.1532],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.5859,  -5.0393,   0.5476],
        [  2.3200,   7.1756,   1.1938],
        [ 12.3025,  -1.7440,  -6.6632],
        [  4.4633,   5.6287,  -0.7754],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.3413,   0.4883,  -6.8416],
        [  9.6662,   9.1725,   8.1056],
        [  5.0348,  -2.8229,   3.5436],
        [  5.6258,  -1.9404,  -6.0715],
        [  2.5382,   7.0672,   4.5097],
        [  4.0591,   6.0202,  -1.6339],
        [ -9.4159,  -2.0739,   9.6841],
        [ 10.2712,   8.3707,  -4.9743],
        [  5.3602,  -2.3808,   7.2098],
        [  1.3705,  13.9723,   7.9266],
        [  9.0129,  -7.7072,   9.1400],
        [ -6.8618,  -7.6322,   3.7027],
        [ -8.0829,  -5.9940,   5.0564],
        [  3.3069,   6.6089,   8.7377],
        [ -4.5778,  -9.5013,  -5.0667],
        [  4.2612, -10.5514,  -6.6268],
        [  9.0051,   9.9234,   5.4411],
        [ 12.3759,  -1.3943,   6.4659],
        [ -8.5149,  -5.1905,   1.5239],
        [ 11.0438,   7.1180,  -5.6991],
        [ 12.5224,  -0.4497,   4.8448],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.0547,   7.2957,  -1.9343],
        [ 11.4495,   6.3026,   6.8257],
        [  6.3031,   1.8477,  -7.1706],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.5318,  13.7231,   6.3692],
        [ 10.0567,   8.6697,   4.6329],
        [  8.4691,  10.4571,  -5.0650],
        [  2.6613,   7.0021,   4.5783],
        [  1.5383, -11.0274,  -4.9666],
        [ -8.9606,  -4.0994,   1.7886],
        [ -1.9600,   7.8577,   6.9121],
        [  5.6022,  -1.9833,   8.3081],
        [  1.3767, -11.0341,   9.9415],
        [ 11.4703,  -4.2006,   7.4086],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.2373,  -3.1225,  -8.9763],
        [  5.1787,  -2.6383,  -4.2533],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.6363, -11.0348,   6.1414],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.1737,   2.5305,   3.9049],
        [ -3.0008, -10.2827,  -8.9480],
        [ -2.1533, -10.5778,   1.0112],
        [ -5.9872,   6.2037,  -9.4477],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.0467,  -9.8369,  -6.5816],
        [  7.5784,  -8.9141,   5.8410],
        [  4.0587, -10.6112,   6.2273],
        [  3.9770, -10.6341,   8.0783],
        [ -8.7409,  -4.6837,  -6.5840],
        [  4.5814, -10.4485,   2.5496],
        [  1.9503,  -4.7172,  -6.2313],
        [ -9.3464,   0.5708,  -1.6397],
        [ 12.4592,  -0.9168,  -6.7571],
        [  6.2038,   2.4020,  -3.1692],
        [ 12.2218,   4.0953,  -8.2072],
        [  0.5250, -11.0306,   5.5358],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -8.7544,   2.6096,   3.3146],
        [ -2.3317,   7.8034,  -3.6274],
        [  5.9026,   3.3837,  -0.6573],
        [  1.1006,   7.6318,  -2.4877],
        [  2.5663,   7.0526,  -6.8769],
        [  2.4697,  -4.5836,   1.4593],
        [ -9.4157,   0.0821,  -5.0025],
        [ -8.1531,  -5.8748,   2.6934]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0050, 0.0268,  ..., 0.9748, 0.9490, 0.3721],
        [0.0050, 0.0000, 0.0272,  ..., 0.9776, 0.9468, 0.3695],
        [0.0268, 0.0272, 0.0000,  ..., 0.9977, 0.9729, 0.3949],
        ...,
        [0.9748, 0.9776, 0.9977,  ..., 0.0000, 0.9538, 0.8816],
        [0.9490, 0.9468, 0.9729,  ..., 0.9538, 0.0000, 0.5794],
        [0.3721, 0.3695, 0.3949,  ..., 0.8816, 0.5794, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.77588801,  9.79814997,  9.9353921 ,  9.94173708,  9.87720102,
       10.00802627, 12.07818871, 11.05317597,  9.36378438,  5.0713364 ,
        7.32754882, 11.75807252,  8.72896381, 11.63724792, 13.13205357,
       11.59512194,  8.62260327, 10.412299  ,  7.9337647 , 12.10222204,
        6.65259775,  9.15866607,  8.09984231, 10.26545792,  9.95552626,
        6.81961996,  6.05113217,  7.46053281,  5.36061974,  5.5513072 ,
        9.95552626,  7.54127425, 12.42554808,  7.18354885, 10.92597196,
        5.90035324,  6.36004393, 13.32557133,  5.77219077,  5.9510489 ,
        7.50917551,  7.26074925,  9.64157461, 13.25017211,  5.86519525,
       14.0393949 , 11.85890664, 10.26327895, 10.06286074,  7.39004226,
       10.54658597, 11.37939695, 13.40024224, 12.45418236,  9.97220799,
       13.13888841, 12.53047489, 11.99633819,  7.57946459, 13.06957425,
        6.56833864,  8.62329466, 13.95472331, 13.2778378 , 13.45642379,
        7.49076899, 11.1341798 ,  9.85384623,  8.09842704,  5.94293185,
       11.11969815, 12.21532282,  7.72668467,  9.75075273,  5.81199422,
        5.19986797, 11.05317597,  9.09661212,  6.67218687, 10.71162681,
       10.794755  ,  8.62162372, 12.00995898, 12.64235359, 11.54672755,
       11.70016217, 11.36089355, 11.35345345,  9.91668182, 11.4088043 ,
        5.10444097,  9.36378438, 12.49291768,  6.65259775, 12.889695  ,
       11.04313641,  5.22903117, 12.1460931 ,  9.13507321,  8.14433542,
        6.80370254,  7.71075226,  7.50500064,  5.20659892,  9.41605684,
       10.04916678])}
{'P': tensor([[1.0446e-02, 2.9099e-01, 2.3642e-01,  ..., 8.7098e-06, 2.3646e-05,
         1.5738e-07],
        [1.1882e-04, 1.5670e-02, 1.6779e-04,  ..., 1.6983e-05, 7.4821e-05,
         2.1078e-07],
        [2.6556e-04, 3.7818e-01, 1.1472e-02,  ..., 1.0736e-05, 3.3695e-05,
         1.6399e-07],
        ...,
        [7.0656e-05, 4.9832e-05, 6.4010e-05,  ..., 8.0042e-02, 7.3357e-05,
         5.9989e-03],
        [4.5788e-06, 4.4331e-06, 4.4960e-06,  ..., 2.5613e-02, 1.2568e-01,
         2.3120e-04],
        [1.2036e-04, 5.1271e-05, 8.8622e-05,  ..., 2.1865e-05, 3.3843e-07,
         6.2741e-02]]), 'X': tensor([[ -9.1749,  -3.3803,   6.4629],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.0330,  -6.0765,   6.7945],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.1989,  -9.7591,   0.8289],
        [ -7.4807,   4.7463,   2.0616],
        [  2.2182,  13.8007,   5.1889],
        [  0.9392,  -4.8143,  -3.4649],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.8745,   2.3097,   3.5600],
        [  7.5930,  -8.9037,   7.2323],
        [ -3.2486,   7.5925,  -1.3719],
        [  1.3339,   7.5627,  -0.9034],
        [  2.7685, -10.8982,  -6.8928],
        [  8.4691,  10.4571,  -5.0650],
        [  6.0788,  -0.8710,  -6.8719],
        [ -9.1624,  -3.4286,   2.0647],
        [  1.1274,  14.0116,   8.1753],
        [ -2.5683, -10.4433,  -2.9182],
        [  5.7845,   3.6652,   0.9175],
        [  6.3516,   0.6425,   0.3425],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.2860,   7.1916,  -6.1011],
        [  9.6614,   9.1784,  -1.2087],
        [  5.2911,  -2.4824,  -3.6877],
        [  8.1528,  10.7452,   9.1493],
        [  2.5813,  -4.5471,   5.3597],
        [  4.6992,   5.3690,   3.0935],
        [  3.6249, -10.7255,   5.6605],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.8325,  -1.5222,  -2.1341],
        [  9.3560,   9.5391,   9.4943],
        [ -3.5173,   7.5091,  -8.9585],
        [  4.5340,  -3.3629,   8.9652],
        [ -9.3076,   0.7852,  -6.7010],
        [  3.3094,  13.4969,  -3.6507],
        [  4.3554,   5.7393,  -9.0766],
        [  6.1579,  -0.5961,  -6.5595],
        [  4.3377,   5.7570,   1.3712],
        [  4.7482, -10.3907,   5.9149],
        [ -9.2687,  -2.9794,   0.6436],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -8.1666,   3.7653,   3.6075],
        [  5.7963,  -1.6018,   1.8856],
        [ 12.4996,   2.6335,  -6.3166],
        [ -2.8162,   7.7059,  -9.6885],
        [  0.7145,  -4.8071,  -2.3766],
        [  4.2275,  -3.6320,  -0.2714],
        [  7.7794,  11.0629,   8.9069],
        [  2.6321,  -4.5296,  -2.5159],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.1644, -11.0394,   7.8511],
        [  5.9912,   3.1453,  -3.9989],
        [  1.5383, -11.0274,  -4.9666],
        [  2.7590,  13.6621,   9.4496],
        [ -8.0588,  -6.0341,  -2.3006],
        [ 11.1085,  -4.9204,   0.0736],
        [ -1.3178, -10.7939,   0.4236],
        [  4.0457,  -3.7737,  -0.4704],
        [ -7.3333,   4.9246,  -7.9212],
        [ 12.5919,   0.4035,  -5.8460],
        [  9.7328,   9.0902,  -0.8917],
        [  1.7232,  -4.7574,   3.1270],
        [  6.3604,   0.9042,  -6.5901],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -2.0721,   7.8432,   6.6380],
        [  4.5814, -10.4485,   2.5496],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.1196,  -2.7160,   4.2705],
        [  3.9913,  -3.8139,  -6.8311],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.2555,   2.1474,   4.1685],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.4356,  -4.5942,  -5.0808],
        [  2.0243,   7.3086,   7.5432],
        [ -8.4887,  -5.2448,  -0.6585],
        [  4.5954,  13.0099,  -4.4439],
        [  6.0310,  12.2795,  -6.4528],
        [ 11.6634,   5.8050,   9.1590],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 12.5832,   1.7543,  -7.6134],
        [  8.3791,  10.5409,   5.5590],
        [ -0.0282,   7.8575,  -1.1746],
        [  1.7659,   7.4127,   4.7871],
        [  2.1504,  -4.6726,   8.7908],
        [ -1.7009,   7.8851,  -8.2437],
        [  9.3560,   9.5391,   9.4943],
        [ -2.1817,   7.8274,  -4.6298],
        [  3.9770, -10.6341,   8.0783],
        [  4.3523, -10.5232,   8.2138],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.5900,   2.9767,   7.4811],
        [  5.8979,   3.3958,  -3.8713],
        [  2.0449,  13.8402,  -7.1642],
        [  7.9141,  -8.6656,   9.6948],
        [ 11.7349,  -3.5838,   2.1644],
        [ -3.0008, -10.2827,  -8.9480],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.2022,   2.4092,   3.0349],
        [  9.9058,  -6.7203,   5.6928],
        [  2.2659,  13.7894,   6.5436]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 1.8308e-02, 2.3540e-02,  ..., 9.4019e-01, 7.2716e-01,
         1.1950e+00],
        [1.8308e-02, 0.0000e+00, 2.6297e-02,  ..., 9.5849e-01, 7.4423e-01,
         1.2117e+00],
        [2.3540e-02, 2.6297e-02, 0.0000e+00,  ..., 9.4582e-01, 7.4081e-01,
         1.2096e+00],
        ...,
        [9.4019e-01, 9.5849e-01, 9.4582e-01,  ..., 0.0000e+00, 3.7517e-01,
         5.1991e-01],
        [7.2716e-01, 7.4423e-01, 7.4081e-01,  ..., 3.7517e-01, 0.0000e+00,
         4.7066e-01],
        [1.1950e+00, 1.2117e+00, 1.2096e+00,  ..., 5.1991e-01, 4.7066e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.77777346,  9.96391045,  9.84314712, 10.07238463, 10.1371552 ,
       10.06286074, 11.56146873,  8.85939626, 13.97779839,  4.90505566,
       14.09176259,  9.17016857, 11.70168755,  8.25828409,  7.67939546,
       11.24434374, 13.45642379,  6.14086582,  9.78285424, 14.05687815,
       10.75443106,  6.84796202,  6.38399789,  9.60359158, 10.41909211,
        7.54620849, 13.32613445,  5.84450127, 13.48809595,  5.2287058 ,
        7.13501264, 11.32149259, 10.36844813,  6.02789634, 13.3614598 ,
        8.29203356,  5.64500512,  9.34061938, 13.89671314,  7.20482224,
        6.18668463,  7.20826219, 11.42420508,  9.73579738, 12.3828957 ,
        8.99276564,  6.01356004, 12.77402351,  8.20436764,  4.85995002,
        5.57341623, 13.52430176,  5.23878371,  5.24137909, 11.10066126,
        6.76661595, 11.1341798 , 13.93790235, 10.0674887 , 12.1494023 ,
       10.87409005,  5.53255882,  8.83341608, 12.59840193, 13.31764181,
        5.05989547,  6.42439344, 12.1460931 ,  8.11227668, 11.4088043 ,
        5.39450305,  5.79543038,  5.5205189 ,  9.97964013,  6.61386436,
        8.35856487,  5.19986797,  7.58378655,  9.97822473, 13.79762415,
       13.68063098, 13.02818702,  9.86089579,  9.26455795,  9.50654967,
       12.70489563, 13.46553623,  7.8575648 ,  7.62011734,  5.14371005,
        8.06643487, 13.3614598 ,  8.12580976, 11.35345345, 11.38774149,
        5.54776934,  9.09119351,  6.80559669, 13.99048094, 11.73567759,
       12.26997165, 10.71162681,  9.68323686,  6.65368551, 11.9702762 ,
       13.97430191])}
{'P': tensor([[9.4235e-03, 4.0323e-01, 1.7125e-01,  ..., 1.8150e-04, 2.9013e-05,
         1.2426e-06],
        [1.6546e-04, 1.3734e-02, 3.4038e-01,  ..., 1.9041e-04, 2.8881e-05,
         2.2929e-06],
        [8.9084e-05, 1.2375e-04, 1.7433e-02,  ..., 7.9177e-05, 1.4889e-05,
         5.0325e-06],
        ...,
        [1.7529e-03, 2.5195e-03, 4.0909e-03,  ..., 8.8159e-02, 2.8325e-04,
         2.8340e-05],
        [2.2173e-02, 2.2272e-02, 1.3571e-02,  ..., 1.0199e-01, 5.2044e-02,
         4.4004e-06],
        [3.9725e-06, 3.5951e-06, 2.7416e-06,  ..., 8.2630e-06, 2.4486e-05,
         2.1493e-01]]), 'X': tensor([[ -9.1658,  -3.4155,  -2.8880],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.1624,  -3.4286,   2.0647],
        [ -9.4541,  -0.3327,   8.8246],
        [  9.5018,  -7.1945,  -1.1892],
        [  5.8323,  -1.5228,   5.1132],
        [  3.6249, -10.7255,   5.6605],
        [  4.9957,  -2.8705,   6.5978],
        [  5.1890,   4.7318,   0.9515],
        [ 12.2218,   4.0953,  -8.2072],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.6165,  -3.8715,   6.4594],
        [ -5.1377,   6.7751,  -0.4438],
        [  7.7750,  11.0665,   2.8364],
        [  2.2860,   7.1916,  -6.1011],
        [ 12.3453,  -1.5462,   7.0212],
        [ -9.3604,   0.4855,  -3.0158],
        [  9.9756,  -6.6330,  -3.3429],
        [  3.3633,  13.4794,   0.5193],
        [ -1.0012, -10.8576,   2.2867],
        [  4.5491,  -3.3485,   6.8412],
        [  0.1206,  -4.7363,   8.7103],
        [ -7.3554,   4.8985,  -0.6665],
        [  8.4691,  10.4571,  -5.0650],
        [  5.5101,  -2.1433,   9.5442],
        [  8.0381,  -8.5690,  -8.9523],
        [  8.0381,  -8.5690,  -8.9523],
        [ 11.8951,  -3.1570,   7.0742],
        [ -3.6381,   7.4683,   7.3544],
        [  9.3560,   9.5391,   9.4943],
        [ -9.1749,  -3.3803,   6.4629],
        [  7.5784,  -8.9141,   5.8410],
        [  4.5800,   5.5034,  -3.9432],
        [ 12.4837,   2.7510,   0.4449],
        [ -8.4297,  -5.3637,  -9.3962],
        [  4.5503,   5.5359,   7.4735],
        [  6.2038,   2.4020,  -3.1692],
        [  6.9284,  11.7061,  -0.3732],
        [  5.8323,  -1.5228,   5.1132],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.1350,  14.0104,   0.6321],
        [  6.3200,   0.2690,   1.5900],
        [  3.2366,  -4.2732,   5.1532],
        [  2.1440, -10.9811,   8.5281],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.1379,   4.4125,  -3.8833],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -8.8934,   2.2597,   3.5831],
        [  6.2902,   0.0451,   3.7116],
        [ -0.5385,   7.9037,   6.3830],
        [  1.6954,  -4.7616,   9.8198],
        [  2.3200,   7.1756,   1.1938],
        [ -7.6650,   4.5092,   0.2520],
        [ -7.8126,   4.3066,   5.1791],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -5.9938,  -8.4745,   3.9735],
        [  1.9505,  -4.7171,   9.9868],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.9503,  -4.7172,  -6.2313],
        [ -8.4824,  -5.2575,  -0.0795],
        [  2.5375,   7.0676,  -4.3490],
        [  1.1274,  14.0116,   8.1753],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -3.9456,  -9.8542,   0.7466],
        [  7.7750,  11.0665,   2.8364],
        [ -2.6735, -10.4062,  -1.5911],
        [  2.7295,  13.6703,  -8.9328],
        [  5.4732,  -2.2041,  -4.3830],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.1203,  -2.7151,   6.2110],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.5294,  -2.1106,   9.0626],
        [  5.9003,  12.3550,   2.1917],
        [  7.3724,  -9.0577,  -8.9843],
        [  0.4996,   7.7738,  -4.4711],
        [  0.9659,  -4.8144,  -0.9594],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.7295,  13.6703,  -8.9328],
        [ -1.9487,   7.8590,  -3.3330],
        [ 11.7162,   5.6728,  -8.9372],
        [  8.0381,  -8.5690,  -8.9523],
        [ -7.4807,   4.7463,   2.0616],
        [  5.1472,  -2.6801,   8.1194],
        [  2.9470,  13.6085,   2.9809],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.1886,  -2.6250,   6.4677],
        [  0.6470, -11.0352,  -1.1931],
        [  6.3505,   0.6235,  -8.5849],
        [  6.2247,  -0.3150,  -9.3657],
        [ -6.2200,  -8.2737,   5.5079],
        [ -9.1749,  -3.3803,   6.4629],
        [ -9.0961,   1.6529,   3.1553],
        [ 12.3759,  -1.3943,   6.4659]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.1358e-03, 2.2901e-02,  ..., 7.2821e-01, 5.3071e-01,
         1.3741e+00],
        [6.1358e-03, 0.0000e+00, 2.6363e-02,  ..., 7.2720e-01, 5.3127e-01,
         1.3775e+00],
        [2.2901e-02, 2.6363e-02, 0.0000e+00,  ..., 7.4995e-01, 5.5003e-01,
         1.3851e+00],
        ...,
        [7.2821e-01, 7.2720e-01, 7.4995e-01,  ..., 3.0339e-04, 2.5271e-01,
         9.7898e-01],
        [5.3071e-01, 5.3127e-01, 5.5003e-01,  ..., 2.5271e-01, 2.1453e-04,
         9.4289e-01],
        [1.3741e+00, 1.3775e+00, 1.3851e+00,  ..., 9.7898e-01, 9.4289e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.78147542,  9.86089579, 10.06407839, 10.0674887 , 10.1979669 ,
        9.79039371,  9.78285424,  9.45995955, 11.9182956 ,  6.02779201,
       11.32149259,  5.76165934,  7.02253235, 12.889695  ,  8.04718433,
       12.24467041,  8.50278896, 13.5247215 ,  7.54620849, 12.44177641,
        9.37296102, 11.97957425, 13.89264822, 10.9036226 ,  5.64865173,
        4.73784222,  8.83725081, 13.45642379,  5.91222346, 11.74901778,
       11.74901778, 12.30695098,  8.30729352, 13.3614598 ,  9.77777346,
       11.70016217,  7.15990676, 12.78327037,  9.99145096,  7.16598273,
        6.65259775, 13.60275211,  6.02779201,  6.33075682,  9.71392179,
        9.66977802, 14.05633441,  6.32572323,  5.36061974, 11.18839169,
       12.35625694, 12.91505111, 12.49317563,  9.17595899,  6.29034958,
        7.92200342,  5.05444288,  7.54127425,  8.89301516,  8.92098752,
       10.61047596, 10.37998078,  5.10446653, 10.92597196,  9.19837792,
        9.55593374,  5.10444097,  9.97964013,  7.5092834 , 14.05687815,
       11.59512194,  9.562174  , 10.61472636, 13.5247215 , 10.74409834,
       13.94008948,  5.90035324,  5.56809335,  5.79562691,  9.50913849,
       10.6910768 ,  8.09704052,  9.47369628,  9.61942445,  5.91854397,
       13.69162795, 11.67875855,  7.78980299,  4.91038884,  9.37989631,
       13.94008948,  8.09704052, 13.01728313, 11.74901778,  8.85939626,
        5.80312949, 13.92390689,  5.21280156,  5.81481044, 11.05413545,
        6.38105015,  6.23262289, 10.35093485,  9.77777346,  9.24501829,
       12.45418236])}
{'P': tensor([[4.7821e-03, 1.5897e-01, 1.1403e-01,  ..., 3.6497e-06, 5.3702e-08,
         1.5897e-01],
        [1.2837e-04, 8.0402e-03, 1.8438e-01,  ..., 6.8772e-06, 8.2858e-08,
         8.0402e-03],
        [7.0166e-05, 1.1996e-04, 9.8377e-03,  ..., 9.5735e-06, 9.9433e-08,
         1.1996e-04],
        ...,
        [5.6979e-05, 5.0692e-05, 3.6832e-05,  ..., 6.7783e-02, 7.0006e-04,
         5.0692e-05],
        [5.2726e-04, 3.6991e-04, 2.2234e-04,  ..., 6.1177e-05, 4.2071e-01,
         3.6991e-04],
        [1.2837e-04, 8.0402e-03, 1.8438e-01,  ..., 6.8772e-06, 8.2858e-08,
         8.0402e-03]]), 'X': tensor([[ -9.1624,  -3.4286,   2.0647],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.1531,  -5.8748,   2.6934],
        [  9.9058,  -6.7203,   5.6928],
        [ -8.3539,  -5.5111,   0.3373],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.3535,   0.6792,  -7.5101],
        [  0.8572,  14.0501,  -8.3364],
        [ 11.5198,  -4.0923,   5.5088],
        [ -6.4844,  -8.0230,   8.5209],
        [  0.2571,  -4.7594,   1.5618],
        [ -0.8455, -10.8853,  -6.7400],
        [ 11.1469,   6.9237,   1.6160],
        [  4.1297,  -3.7098,  -3.9838],
        [  0.8940, -11.0402,  -0.9783],
        [  1.7301,   7.4262,   5.0522],
        [  3.4009,  -4.1874,  -5.9751],
        [ -0.8675, -10.8815,   6.8561],
        [  2.5974,   7.0363,  -7.7810],
        [  5.1519, -10.2387,   9.8191],
        [ 10.9394,   7.3069,  -8.9365],
        [ -6.8618,  -7.6322,   3.7027],
        [ -8.9636,   2.0648,   8.1946],
        [  9.0004,  -7.7195,   4.6945],
        [ -2.1817,   7.8274,  -4.6298],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.4572,   7.1085,   7.7376],
        [  1.6308,  13.9254,  -2.0984],
        [  8.0877,  -8.5297,  -1.1050],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.2311,  -2.0445,   1.3545],
        [ -0.2586,   7.8825,  -9.9840],
        [  8.5716,  10.3596,   6.8818],
        [  2.6451,  -4.5250,  -0.5565],
        [  6.2365,  -0.2578,  -6.0212],
        [ -3.0004,   7.6607,  -7.2640],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.0345,   3.0175,  -1.4424],
        [ -5.2589,   6.7024,  -5.5809],
        [  3.6249, -10.7255,   5.6605],
        [  5.8318,   3.5567,  -4.7692],
        [  2.5469,  13.7192,  -9.6935],
        [  6.8407,  11.7666,   3.0045],
        [ -0.5709,   7.9055,   7.6089],
        [ -3.3999, -10.1152,  -1.2088],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.0611,  -6.0302,   4.6962],
        [  7.8982,  -8.6778,   0.0239],
        [  5.4336,  -2.2676,   5.6933],
        [  1.3403,  -4.8011,  -1.1698],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.0330,  -6.0765,   6.7945],
        [  0.3354,  -4.7708,  -4.9125],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.2608,  -2.5256,  -9.1446],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.0955,   2.8210,  -4.3321],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.3464,   0.5708,  -1.6397],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.5224,  -0.4497,   4.8448],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.3739,   0.3982,  -4.2564],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.5626,  -2.0536,   8.4433],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.1824,  -4.6647,   1.1985],
        [ -9.3616,  -2.4749,   0.1793],
        [  2.7558,  -4.4844,   0.0636],
        [  5.7218,  -9.9934,   6.5764],
        [ -0.8455, -10.8853,  -6.7400],
        [  4.0362,  13.2397,  -7.0030],
        [ -8.9788,   2.0208,  -3.2867],
        [ 12.2589,  -1.9312,   9.8657],
        [  4.3554,   5.7393,  -9.0766],
        [ -6.2668,   5.9808,   2.5614],
        [  4.9337,  12.8566,  -3.6582],
        [ 11.4495,   6.3026,   6.8257],
        [  2.2849, -10.9655,  -7.6542],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.0054,  -1.0902,   8.3079],
        [  5.8979,   3.3958,  -3.8713],
        [  6.3236,   0.3011,  -4.6668],
        [ -0.5709,   7.9055,   7.6089],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.5927,   0.4199,  -9.4997],
        [  5.6248,   3.9984,   8.4933],
        [  5.9026,   3.3837,  -0.6573],
        [  3.1271, -10.8343,   2.6442],
        [  2.1504,  -4.6726,   8.7908],
        [ -5.3736,   6.6310,  -1.5174],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.1139,   7.6281,  -2.5680],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3534,   1.2990,  -7.2942],
        [  2.6970,   6.9826,   9.7350],
        [ -7.0120,   5.2825,   7.7762],
        [  2.3928,   7.1404,  -1.7035],
        [  6.1755,   2.5233,  -2.0042],
        [  1.9312,  13.8649,   8.3563],
        [ -8.9606,  -4.0994,   1.7886]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0269, 0.0268,  ..., 0.9796, 1.3552, 0.0269],
        [0.0269, 0.0000, 0.0392,  ..., 0.9906, 1.3802, 0.0000],
        [0.0268, 0.0392, 0.0000,  ..., 1.0035, 1.3635, 0.0392],
        ...,
        [0.9796, 0.9906, 1.0035,  ..., 0.0000, 0.9622, 0.9906],
        [1.3552, 1.3802, 1.3635,  ..., 0.9622, 0.0000, 1.3802],
        [0.0269, 0.0000, 0.0392,  ..., 0.9906, 1.3802, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.78285424,  9.85384623,  9.99211663,  9.97220799, 10.06850677,
       10.04916678, 11.9702762 , 10.00794468,  9.43112957,  6.38968357,
       14.07623266, 12.22503754, 10.31584102,  4.7663526 , 10.91805683,
       13.12218013,  5.5513072 , 11.07637531,  7.62509582,  5.39450305,
       10.91601655,  7.50035355, 11.46176259, 13.15525803, 10.26327895,
        9.19837792, 11.85743425,  8.12580976,  8.35856487,  7.52117132,
       14.02058621, 11.75438006,  8.68832398, 12.40074901,  7.88677538,
       13.44593519,  5.24137909,  6.24187044,  8.22728399,  4.78258092,
        6.74686071,  8.51928531, 11.32149259,  6.83083863, 13.95361098,
       13.6105595 ,  7.92607342, 10.67132524,  9.70108409, 10.06703358,
       11.73397309,  5.88782913,  4.98461983, 10.01804223,  9.85384623,
       10.07238463,  4.78258092,  8.47371868,  5.83560009,  5.47866221,
        6.7166297 , 12.35625694,  9.36378438,  5.16611782, 12.53047489,
        5.05444288,  8.95983586,  9.38232121, 13.14777433,  5.92952538,
        9.20340031,  5.14999799,  9.68323686,  5.26342203, 11.51557166,
       10.91805683, 13.84126197,  9.20340031, 12.41012046,  7.20482224,
        8.66273171, 13.77075174, 13.06957425, 11.20100795, 11.74901778,
        6.10360073,  6.80559669,  6.33075682,  7.92607342, 11.33837396,
       12.59969936,  6.90115901,  6.80370254, 11.27656488,  5.14371005,
        8.53501777,  9.66977802,  7.70897675, 12.27993098,  6.48486171,
        7.48538905,  8.77913428,  7.53063885,  6.67108662, 13.99877413,
        9.85384623])}
{'P': tensor([[5.2511e-03, 2.3682e-01, 1.8864e-01,  ..., 8.0172e-05, 1.1891e-03,
         1.4192e-06],
        [9.8995e-05, 8.0176e-03, 1.7407e-01,  ..., 1.3509e-04, 2.1608e-03,
         1.8309e-06],
        [1.4342e-04, 4.0347e-04, 1.2354e-02,  ..., 1.9035e-04, 3.1392e-03,
         2.5179e-06],
        ...,
        [1.4220e-05, 1.7804e-05, 1.8630e-05,  ..., 8.1768e-02, 4.5666e-03,
         8.6184e-06],
        [9.7955e-05, 1.2618e-04, 1.3947e-04,  ..., 2.7094e-02, 1.8024e-01,
         2.2486e-05],
        [3.2629e-03, 2.6999e-03, 2.2493e-03,  ..., 3.6944e-04, 5.7995e-04,
         9.2358e-02]]), 'X': tensor([[ -9.1623,  -3.4288,  -4.9961],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -7.8849,  -6.3115,  -4.7479],
        [  2.0449,  13.8402,  -7.1642],
        [  6.3550,   0.7129,   0.4917],
        [  6.3569,   0.7591,   6.5011],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.3705,  13.9723,   7.9266],
        [  2.0547,   7.2957,  -1.9343],
        [  2.6078, -10.9229,   6.2869],
        [ -6.4822,  -8.0252,  -3.8983],
        [  1.4489,   7.5256,   4.4529],
        [ -9.4117,   0.1160,   8.2574],
        [  3.2809,  13.5061,  -4.0199],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.2365,  -0.2578,  -6.0212],
        [  2.2373,  -4.6506,  -1.7334],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.3616,  -2.4750,   3.7765],
        [  3.0719, -10.8449,  -4.1673],
        [ -5.0550,   6.8231,  -9.2502],
        [ -5.9938,  -8.4745,   3.9735],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.0506,   4.9277,  -9.5094],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.8323,  -1.5228,   5.1132],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.0560,  -0.9421,  -5.0334],
        [ -1.9487,   7.8590,  -3.3330],
        [  7.0358,  -9.2784,  -9.8479],
        [  3.0876,   6.7527,  -5.3540],
        [  3.5965,  -4.0754,  -3.5035],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -1.6086,   7.8928,   4.3385],
        [  2.2437,  -4.6489,   6.1430],
        [  5.9843,  -9.8679,  -6.1449],
        [  2.0419,  -4.6978,  -5.0983],
        [ -4.2271,   7.2387,  -1.8316],
        [ 11.0438,   7.1180,  -5.6991],
        [ -4.4503,   7.1378,   3.3751],
        [ -9.3384,  -2.6161,  -4.7630],
        [  4.7739,  -3.1218,  -4.1188],
        [ 11.9936,  -2.8677,   6.0614],
        [ -9.1907,  -3.3180,  -5.5943],
        [ 12.4597,  -0.9136,  -4.8763],
        [  9.9058,  -6.7203,   5.6928],
        [ -3.0004,   7.6607,  -7.2640],
        [ 11.4495,   6.3026,   6.8257],
        [ -5.6651,   6.4380,   1.3660],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.9655,   3.2172,  -7.5953],
        [  0.6396,  14.0772,  -6.3786],
        [  2.8474,   6.8979,   6.6457],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.5636, -10.9293,   9.6519],
        [  6.3542,   0.6934,  -3.3840],
        [ -7.9925,  -6.1423,   2.8990],
        [  6.9314,  -9.3435,  -1.2729],
        [  0.3715,  -4.7756,   0.2312],
        [  6.1372,   2.6726,  -6.7885],
        [ -7.4807,   4.7463,   2.0616],
        [  6.9956,  11.6590,  -7.6876],
        [  2.8602,   6.8905,   5.8779],
        [ -8.4267,  -5.3697,   2.4172],
        [ -3.9700,   7.3453,  -4.1486],
        [ 11.9019,  -3.1379,  -3.4534],
        [  6.9693,  -9.3200,  -8.3822],
        [  3.2226,  -4.2802,  -4.0428],
        [  5.1886,  -2.6250,   6.4677],
        [ 12.5650,   2.0132,   4.6272],
        [  6.3550,   0.7129,   0.4917],
        [ 11.5456,   6.0861,   8.1101],
        [  4.4633,   5.6287,  -0.7754],
        [  3.7298,  13.3537,  -9.4653],
        [ -6.8466,  -7.6488,   1.9648],
        [ -1.5261,   7.8988,   9.3768],
        [ -3.5029,   7.5138,   8.1853],
        [ -1.9487,   7.8590,  -3.3330],
        [  2.1120,  -4.6818,   8.6144],
        [ -2.0840,   7.8415,  -7.8657],
        [ -7.9298,   4.1367,  -3.9773],
        [  2.6613,   7.0021,   4.5783],
        [  1.3339,   7.5627,  -0.9034],
        [  3.9022,  13.2906,   5.5788],
        [  9.0051,   9.9234,   5.4411],
        [ -3.4986,   7.5152,   9.0712],
        [ 10.9394,   7.3069,  -8.9365],
        [  4.9452, -10.3187,  -7.8621],
        [  4.9452, -10.3187,  -7.8621],
        [  8.9284,  10.0036,  -6.9210],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.3606,   0.9204,   3.2525],
        [  6.0890,  -0.8382,   2.9166],
        [ 12.2592,   3.9425,   2.0383],
        [ -3.3812, -10.1235,   7.4384],
        [ -3.9508,  -9.8515,   1.7923],
        [ -5.1974,   6.7396,   3.3484]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0057, 0.0141,  ..., 0.9930, 0.5598, 0.8498],
        [0.0057, 0.0000, 0.0128,  ..., 0.9928, 0.5601, 0.8522],
        [0.0141, 0.0128, 0.0000,  ..., 1.0055, 0.5729, 0.8639],
        ...,
        [0.9930, 0.9928, 1.0055,  ..., 0.0000, 0.4376, 0.4618],
        [0.5598, 0.5601, 0.5729,  ..., 0.4376, 0.0000, 0.3985],
        [0.8498, 0.8522, 0.8639,  ..., 0.4618, 0.3985, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.78286733,  9.85330646,  9.88571713,  9.96152788,  9.94285511,
       10.09979747, 13.99048094,  6.3949043 ,  6.4020404 ,  5.24137909,
       14.0393949 ,  7.57946459, 11.22993218, 10.31613269,  7.66377554,
        9.41244865, 13.89886469, 10.53368644, 10.91805683,  5.0209005 ,
        9.58072286,  6.24187044,  5.16077764, 12.58006038,  9.68324553,
       11.27160035,  8.49161154, 10.37998078,  8.30125201,  7.05627461,
       13.06498554,  6.02779201,  9.88571713, 12.05395056, 13.01187484,
        6.12886024,  8.09704052, 11.64436684,  7.42512621,  5.43544626,
        5.1224079 ,  9.48969978,  8.0550286 ,  5.16205087, 11.54071859,
        5.1224079 ,  8.38252937, 13.13888841,  8.41148722,  9.69792513,
        5.70405525, 12.33167382,  9.7712321 , 12.49317563, 11.9702762 ,
        8.22728399, 13.06957425,  8.57560275,  5.79391825,  6.77777606,
       14.09176259,  7.46248947,  9.55720651, 11.22596994,  6.39187285,
       10.08001894, 11.63383221,  4.79003361,  6.69389431,  8.85939626,
       13.59673406,  7.46053281,  9.99211663,  8.34949383, 12.30858677,
       11.63765273,  5.35774573,  5.81481044, 12.72524615,  6.3949043 ,
       13.05148826,  7.18354885, 13.86480175, 10.26545792,  8.04483495,
        8.29022197,  8.09704052,  5.13616224,  8.1137452 ,  8.94395628,
        7.49076899,  7.67939546, 13.85158417, 13.40024224,  8.28968067,
       13.15525803, 11.4424816 , 11.4424816 , 13.40850328,  8.4486771 ,
        6.42689705,  6.1463896 , 12.87752052, 10.67322527, 10.61417181,
        8.51088913])}
{'P': tensor([[1.0369e-02, 3.9086e-01, 2.0029e-01,  ..., 1.1101e-05, 4.8720e-06,
         1.3595e-07],
        [2.0705e-04, 1.4387e-02, 3.5657e-01,  ..., 1.6920e-05, 8.4300e-06,
         1.9513e-07],
        [1.1699e-04, 1.5911e-04, 2.2415e-02,  ..., 3.6639e-05, 2.7382e-05,
         3.2806e-07],
        ...,
        [9.4193e-05, 8.1816e-05, 5.7572e-05,  ..., 9.9378e-02, 3.9862e-04,
         5.4780e-03],
        [4.2823e-07, 3.8903e-07, 3.2489e-07,  ..., 5.1124e-03, 1.2281e-02,
         3.7888e-04],
        [1.5247e-06, 1.1891e-06, 5.2904e-07,  ..., 3.9790e-06, 6.7413e-08,
         5.0671e-03]]), 'X': tensor([[ -9.1433,  -3.5003,  -3.0209],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 11.7418,   5.6070,  -1.4192],
        [  0.6757, -11.0361,  -1.3047],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.5294,  -2.1106,   9.0626],
        [  5.5294,  -2.1106,   9.0626],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -4.2271,   7.2387,  -1.8316],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 12.3357,   3.6001,   0.7903],
        [  4.6992,   5.3690,   3.0935],
        [  0.2732,  -4.7619,  -1.1577],
        [  5.9883,   3.1536,   6.2320],
        [  9.5261,  -7.1675,   1.4206],
        [ -1.3940, -10.7772,   6.7835],
        [  4.6063,  13.0051,  -3.1456],
        [ 11.4743,   6.2478,  -2.1100],
        [  6.3515,   0.6418,  -6.5823],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.0054,  -1.0902,   8.3079],
        [ -9.3705,   0.4208,  -3.5815],
        [ -5.1974,   6.7396,   3.3484],
        [  6.2790,  -0.0259,  -5.4116],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 11.3367,  -4.4806,   0.4643],
        [  0.9809,   7.6642,  -1.4217],
        [ -1.0482, -10.8487,  -5.9066],
        [  5.8450,  -9.9356,   3.0767],
        [  5.5101,  -2.1433,   9.5442],
        [  3.1892, -10.8220,  -6.8409],
        [ -9.4716,  -0.6673,   7.6068],
        [  9.3424,  -7.3685,  -3.6807],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.2076,  13.8032,  -1.5099],
        [ -3.0008, -10.2827,  -8.9480],
        [  8.3791,  10.5409,   5.5590],
        [  6.2975,   0.0947,   5.1791],
        [  4.9957,  -2.8705,   6.5978],
        [  0.9478,  -4.8144,   0.6371],
        [  3.9085,   6.1512,  -2.2560],
        [  9.2692,   9.6370,   7.0746],
        [  9.5261,  -7.1675,   1.4206],
        [  6.2535,  -0.1709,  -9.4102],
        [  1.1644, -11.0394,   7.8511],
        [ -8.1426,   3.8048,   9.3542],
        [  7.1441,  11.5529,   2.3232],
        [ -9.4366,  -0.1189,  -2.6736],
        [  8.2154,  -8.4262,   8.0171],
        [  7.8982,  -8.6778,   0.0239],
        [  7.3724,  -9.0577,  -8.9843],
        [  1.6408,   7.4591,  -4.2294],
        [  6.1989,  -9.7591,   0.8289],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.3464,   0.5708,  -1.6397],
        [ -8.1426,   3.8048,   9.3542],
        [  5.6022,  -1.9833,   8.3081],
        [ 12.5919,   0.4035,  -5.8460],
        [  9.3836,   9.5075,  -5.9278],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.7784,  -1.6401,   9.5587],
        [  6.3587,   1.1596,  -3.6629],
        [ 11.2571,   6.7072,   8.3693],
        [  3.3094,  13.4969,  -3.6507],
        [  5.7784,  -1.6401,   9.5587],
        [ 12.6059,   0.9597,  -6.3264],
        [  1.1644, -11.0394,   7.8511],
        [ 11.7779,   5.5127,  -2.0699],
        [  0.1931,  -4.7491,   7.4385],
        [  1.0391,  -4.8141,  -0.0768],
        [  8.9284,  10.0036,  -6.9210],
        [  9.2692,   9.6370,   7.0746],
        [ 12.3357,   3.6001,   0.7903],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.1793,   4.7461,   0.3899],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.8602,   6.8905,   5.8779],
        [  5.9843,  -9.8679,  -6.1449],
        [ -9.3076,   0.7852,  -6.7010],
        [ 11.5456,   6.0861,   8.1101],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3413,   0.4883,  -6.8416],
        [ -0.0410,   7.8591,   9.3457],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.2701,   6.6338,   9.9209],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.4049,   5.6892,  -4.4394],
        [  2.7558,  -4.4844,   0.0636],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.4336,  -2.2676,   5.6933],
        [ -0.8885,   7.9160,   0.0571],
        [  5.6928,  12.4710,   3.6001],
        [ -7.9667,   4.0814,  -0.6397],
        [  6.7792,  11.8084,   1.6251],
        [  6.2576,   2.1359,  -3.3398],
        [ 12.0845,  -2.5772,  -4.6073],
        [  6.6319,  11.9066,   3.5165]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0109, 0.0157,  ..., 0.9907, 1.0283, 1.4917],
        [0.0109, 0.0000, 0.0264,  ..., 0.9963, 1.0353, 1.4925],
        [0.0157, 0.0264, 0.0000,  ..., 0.9865, 1.0217, 1.4947],
        ...,
        [0.9907, 0.9963, 0.9865,  ..., 0.0000, 0.1649, 0.7528],
        [1.0283, 1.0353, 1.0217,  ..., 0.1649, 0.0000, 0.8921],
        [1.4917, 1.4925, 1.4947,  ..., 0.7528, 0.8921, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.79039371,  9.86089579, 10.06407839, 10.0674887 , 10.1979669 ,
        9.94285511, 13.01187484, 11.05672136, 11.63724792,  5.91854397,
        5.91854397, 10.30972193,  8.38252937,  9.75075273, 12.85032836,
        7.13501264,  4.76969364,  6.76791483, 11.92133129, 10.86693708,
       13.79676654, 13.06498554,  6.383897  ,  8.29203356,  6.10360073,
        9.37989631,  8.51088913,  6.27905292,  8.89458273, 12.49291768,
       12.18998195,  7.72668467, 10.8992524 , 11.52733581,  5.91222346,
       11.28215429,  9.49511524, 11.89855267, 12.10942412, 12.12448912,
       12.95702647, 13.97857591, 10.71162681, 13.46553623,  6.29821673,
        5.76165934,  4.90676242,  7.28793663, 13.37121923, 11.92133129,
        6.25586297, 11.10066126,  8.98765644, 13.58333857,  9.43737523,
       11.76830981, 11.73397309, 11.67875855,  7.6374603 , 11.56146873,
        8.22728399,  9.36378438,  8.98765644,  5.94293185, 12.59840193,
       13.35832725, 13.79762415,  9.23071698,  9.53015376,  6.00663314,
        6.46356806, 13.103709  , 13.89671314,  6.00663314, 12.64235359,
       11.10066126, 13.00413829,  4.75303127,  4.92497514, 13.40850328,
       13.37121923, 12.85032836,  8.25828409,  7.02496223,  6.10774544,
        7.46053281, 11.54071859,  9.34061938, 13.05148826, 11.33837396,
        6.36004393,  7.85919687,  8.05242591,  7.39599986, 11.66453059,
        7.19513806,  5.26342203,  5.90035324,  5.88782913,  7.96575782,
       13.70893492,  8.95133961, 13.61600536,  6.61211224, 12.35625694,
       13.62896593])}
{'P': tensor([[1.5233e-02, 2.8044e-01, 2.0670e-01,  ..., 1.2714e-03, 5.8411e-05,
         3.1628e-05],
        [5.2569e-04, 3.6210e-02, 9.3358e-02,  ..., 9.8449e-04, 1.5152e-04,
         9.0331e-05],
        [4.6521e-04, 8.6669e-03, 2.7588e-02,  ..., 6.6269e-04, 1.0668e-04,
         6.7923e-05],
        ...,
        [7.8553e-03, 1.2278e-02, 1.1281e-02,  ..., 1.8475e-01, 7.1818e-03,
         3.7545e-04],
        [7.1035e-07, 7.6979e-07, 7.4063e-07,  ..., 1.9993e-06, 2.4030e-02,
         2.8365e-04],
        [1.0144e-04, 8.0284e-05, 7.8108e-05,  ..., 8.5082e-05, 4.0599e-05,
         1.5557e-01]]), 'X': tensor([[ -9.1231,  -3.5739,   8.4708],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.3535,  -5.5118,   9.8399],
        [ -8.8835,  -4.3178,   9.3080],
        [ -7.6719,  -6.6260,   7.1468],
        [  9.3836,   9.5075,  -5.9278],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.5770,   1.8512,  -0.6578],
        [  3.3094,  13.4969,  -3.6507],
        [  7.7750,  11.0665,   2.8364],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.3515,   0.6418,  -6.5823],
        [  3.2226,  -4.2802,  -4.0428],
        [  7.0171,  11.6439,  -5.0178],
        [ -2.5161,   7.7699,  -9.3531],
        [ -1.6906,   7.8860,   7.7142],
        [  6.0345,   3.0175,  -1.4424],
        [ -9.4203,  -2.0340,   2.5856],
        [ -5.9872,   6.2037,  -9.4477],
        [ -9.4773,  -1.0024,   5.2599],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.2419,  -0.2313,   4.2440],
        [  6.3266,   0.3292,   9.3502],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -0.8569,   7.9156,  -6.5557],
        [  8.9284,  10.0036,  -6.9210],
        [  0.1206,  -4.7363,   8.7103],
        [  3.3069,   6.6089,   8.7377],
        [  3.4401,  13.4540,   4.6576],
        [  2.7976,  -4.4682,  -4.2020],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.1350,  14.0104,   0.6321],
        [  5.6260,  -1.9401,   3.0259],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.4996,   2.6335,  -6.3166],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.0190,   4.9705,  -7.9752],
        [ -1.6906,   7.8860,   7.7142],
        [  6.2036,   2.4030,  -1.2500],
        [  1.5246,  -4.7838,  -5.2095],
        [  0.9392,  -4.8143,  -3.4649],
        [  1.7000,  -4.7609,   2.7935],
        [  2.6310, -10.9195,   3.8054],
        [ -3.5173,   7.5091,  -8.9585],
        [  5.1519, -10.2387,   9.8191],
        [  8.7715,  -7.9390,   5.2900],
        [  6.3119,   1.7797,   9.9186],
        [  6.9284,  11.7061,  -0.3732],
        [ -4.2040,   7.2486,   2.9875],
        [ -9.2687,  -2.9794,   0.6436],
        [  2.0870,  -4.6877,   9.6823],
        [  4.5503,   5.5359,   7.4735],
        [ 12.2311,  -2.0445,   1.3545],
        [  9.3560,   9.5391,   9.4943],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.5456,   6.0861,   8.1101],
        [  5.5294,  -2.1106,   9.0626],
        [  4.9337,  12.8566,  -3.6582],
        [  9.0129,  -7.7072,   9.1400],
        [ -0.6323,   7.9085,  -5.9544],
        [ -4.8751,  -9.3148,   4.0406],
        [ 12.2589,  -1.9312,   9.8657],
        [  6.7565,  11.8237,  -4.9882],
        [ 10.7566,   7.6212,  -2.3279],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 11.4235,   6.3594,   5.1663],
        [  6.3542,   0.6934,  -3.3840],
        [ -3.2486,   7.5925,  -1.3719],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.0348,  -2.8229,   3.5436],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.0608,   3.9362,  -5.9567],
        [  5.0617,  12.7957,   8.3544],
        [  9.7328,   9.0902,  -0.8917],
        [  5.0506,   4.9277,  -9.5094],
        [  5.8979,   3.3958,  -3.8713],
        [ -6.1899,  -8.3011,  -7.9269],
        [  0.1497,   7.8334,  -5.4412],
        [ 10.9916,   7.2133,  -0.9179],
        [ -7.3333,   4.9246,  -7.9212],
        [ -3.2486,   7.5925,  -1.3719],
        [ -8.9282,   2.1648,   8.4894],
        [ -8.6112,   2.9319,  -7.9943],
        [  2.4572,   7.1085,   7.7376],
        [  2.9410,   6.8428,   5.4458],
        [ -9.1459,   1.4780,  -4.2403],
        [ 12.1588,   4.3364,  -8.0867],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.6634,   5.8050,   9.1590],
        [ -4.7334,   6.9982,  -0.4392],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 12.5224,  -0.4497,   4.8448],
        [  4.2339,  -3.6268,   6.3125],
        [  3.2701,   6.6338,   9.9209],
        [  3.7916,  -3.9524,  -7.9547],
        [ -9.3107,  -2.7697,  -3.2919],
        [  4.1140,  -3.7220,  -0.1273],
        [  2.5829,  -4.5466,  -8.4015],
        [ 10.1983,  -6.3430,  -0.3200],
        [  8.5716,  10.3596,   6.8818],
        [  3.8111, -10.6787,  -4.6539],
        [ -8.9606,  -4.0994,   1.7886],
        [  3.0719, -10.8449,  -4.1673],
        [  6.2895,   0.0405,   5.7652]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0272, 0.0324,  ..., 0.4233, 0.9973, 0.9540],
        [0.0272, 0.0000, 0.0054,  ..., 0.4486, 1.0243, 0.9665],
        [0.0324, 0.0054, 0.0000,  ..., 0.4541, 1.0292, 0.9675],
        ...,
        [0.4233, 0.4486, 0.4541,  ..., 0.0000, 0.6536, 1.0083],
        [0.9973, 1.0243, 1.0292,  ..., 0.6536, 0.0000, 0.9319],
        [0.9540, 0.9665, 0.9675,  ..., 1.0083, 0.9319, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.79814997,  9.9353921 ,  9.94173708, 10.00802627,  9.87720102,
       10.1371552 , 13.35832725,  5.27178363, 12.7125079 , 13.89671314,
       13.5247215 ,  9.75401449,  6.383897  ,  5.35774573, 13.59480906,
        8.16714836,  8.06517143,  6.74686071,  9.63742669,  8.62162372,
        9.53015376, 12.42554808,  6.24614905,  6.33517349, 10.52587243,
        7.96181288, 13.40850328,  4.73784222,  7.39004226, 13.88683436,
        5.27178363,  9.88571713, 14.05633441,  5.95111804,  9.5642993 ,
        8.50278896, 12.77402351,  5.24137909,  7.06372172,  8.06517143,
        6.652742  ,  5.0209005 ,  4.90505566,  5.05534609, 11.23201619,
        8.29203356, 11.46176259, 11.83075005,  6.55800267, 13.60275211,
        8.37954853,  9.73579738,  5.13125513,  7.16598273, 12.40074901,
       13.3614598 , 12.07225469, 13.05148826,  5.91854397, 13.77075174,
       11.85890664,  7.9337647 , 10.51340829, 12.41012046, 13.61801526,
       13.18278606, 12.52372629, 13.0743399 ,  6.39187285,  8.25828409,
        9.37989631,  5.77219077,  9.73579738,  8.97052115, 13.76047999,
       13.31764181,  7.05627461,  6.80559669, 10.35485331,  7.83486943,
       13.1471313 ,  8.83341608,  8.25828409,  9.18690018,  9.09661212,
        7.52117132,  7.44805901,  9.26455795, 12.9089559 ,  8.41148722,
       13.02818702,  8.4486771 ,  9.93553424, 12.53047489,  5.5748624 ,
        7.39599986,  5.47702766,  9.71392179,  5.54776934,  5.22903117,
       12.00995898, 13.44593519, 11.33837396,  9.85384623, 11.27160035,
        6.28961988])}
{'P': tensor([[8.4371e-03, 2.7814e-01, 1.5665e-01,  ..., 3.2173e-07, 5.4051e-06,
         1.5113e-05],
        [1.0431e-04, 9.7419e-03, 2.8370e-01,  ..., 4.5040e-07, 4.5168e-06,
         9.9032e-06],
        [8.8467e-05, 1.5839e-04, 1.2464e-02,  ..., 6.8567e-07, 4.4358e-06,
         7.3271e-06],
        ...,
        [2.7114e-06, 1.9108e-06, 1.3978e-06,  ..., 6.2961e-02, 7.3948e-05,
         1.0709e-05],
        [2.4136e-02, 1.4857e-02, 9.5596e-03,  ..., 1.0271e-06, 5.1816e-02,
         3.8718e-04],
        [4.3748e-04, 4.4928e-04, 3.9274e-04,  ..., 9.4742e-07, 1.0563e-05,
         2.9453e-02]]), 'X': tensor([[ -8.9942,  -3.9990,   6.0448],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.0330,  -6.0765,   6.7945],
        [ -8.0829,  -5.9940,   5.0564],
        [ -7.6525,  -6.6534,   5.1721],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.2675,  12.1379,  -8.9720],
        [  2.7558,  -4.4844,   0.0636],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -8.6426,  -4.9136,   9.4941],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.1637,  13.1898,  -1.9189],
        [ -9.1706,   1.3863,   8.4032],
        [  0.1532,  -4.7422,  -7.0145],
        [  4.3554,   5.7393,  -9.0766],
        [ -7.9925,  -6.1423,   2.8990],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.2849, -10.9655,  -7.6542],
        [  5.4732,  -2.2041,  -4.3830],
        [ -9.3087,   0.7792,  -7.5587],
        [  7.8425,  -8.7202,   6.4530],
        [ -5.3179,  -9.0101,   9.4472],
        [ -8.8745,   2.3097,   3.5600],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.2571,  -4.7594,   1.5618],
        [  0.4996,   7.7738,  -4.4711],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.3705,   0.4208,  -3.5815],
        [  4.5430,   5.5438,  -3.8669],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.1813,  -3.3550,   1.8827],
        [ -8.4573,   3.2448,   5.4370],
        [ 12.2914,   3.8037,   6.0978],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -1.5408,   7.8978,   5.3470],
        [  2.4149,  -4.6005,  -4.3339],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.3367,   7.1676,  -9.5263],
        [ -6.7992,   5.4991,   9.9186],
        [ -4.6393,   7.0461,   1.1624],
        [  9.2692,   9.6370,   7.0746],
        [  4.9337,  12.8566,  -3.6582],
        [  0.1047, -11.0046,   0.5188],
        [ -3.9508,  -9.8515,   1.7923],
        [ 12.1588,   4.3364,  -8.0867],
        [  1.0093,   7.6567,   2.3526],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.1379,  -4.6757,   6.4981],
        [ -0.8675, -10.8815,   6.8561],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.9122,   5.1397,  -1.5498],
        [ -7.0948,   5.1940,   9.8171],
        [ 12.4877,  -0.7222,   8.8347],
        [ -9.1178,   1.5784,   2.1410],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.9895,   6.0816,  -6.5235],
        [  4.5800,   5.5034,  -3.9432],
        [  8.9284,  10.0036,  -6.9210],
        [  0.6142,  14.0802,   0.5407],
        [  6.9653,  -9.3225,  -3.4727],
        [  1.6954,  -4.7616,   9.8198],
        [  0.6933,   7.7336,   2.1179],
        [  2.6613,   7.0021,   4.5783],
        [ -1.0012, -10.8576,   2.2867],
        [  9.0051,   9.9234,   5.4411],
        [  7.7750,  11.0665,   2.8364],
        [ 11.3367,  -4.4806,   0.4643],
        [  2.1379,  -4.6757,   6.4981],
        [ 12.4996,   2.6335,  -6.3166],
        [  3.3069,   6.6089,   8.7377],
        [ -8.6050,   2.9451,  -0.7012],
        [ -1.0012, -10.8576,   2.2867],
        [  4.5767,   5.5071,   3.9069],
        [  0.7145,  -4.8071,  -2.3766],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -7.6293,  -6.6859,  -5.5161],
        [  5.8979,   3.3958,  -3.8713],
        [  2.6614,  -4.5192,  -9.3641],
        [  3.9913,  -3.8139,  -6.8311],
        [ -7.5720,   4.6310,   2.1717],
        [  0.1041, -11.0045,   2.4311],
        [  1.6422,   7.4586,  -6.8056],
        [  2.1440, -10.9811,   8.5281],
        [  2.3589,  13.7668,  -3.6501],
        [  6.3550,   0.7129,   0.4917],
        [  2.3200,   7.1756,   1.1938],
        [  2.8602,   6.8905,   5.8779],
        [ -1.6220,   7.8917,   0.0315],
        [  0.6142,  14.0802,   0.5407],
        [  2.4932,   7.0903,  -5.4214],
        [  6.1066,  -0.7799,  -4.1746],
        [ -6.4620,  -8.0449,   6.2719],
        [ -1.0012, -10.8576,   2.2867],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.2365,  -0.2578,  -6.0212],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4541,  -0.3327,   8.8246],
        [ 11.7162,   5.6728,  -8.9372],
        [ -9.2707,   0.9661,   4.4863],
        [ 11.7418,   5.6070,  -1.4192],
        [ -7.0120,   5.2825,   7.7762],
        [ -9.4731,  -0.7120,  -6.0271]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0263, 0.0608,  ..., 1.2670, 0.3475, 0.8666],
        [0.0263, 0.0000, 0.0357,  ..., 1.2847, 0.3485, 0.8917],
        [0.0608, 0.0357, 0.0000,  ..., 1.2974, 0.3390, 0.9272],
        ...,
        [1.2670, 1.2847, 1.2974,  ..., 0.0000, 1.0344, 1.2046],
        [0.3475, 0.3485, 0.3390,  ..., 1.0344, 0.0000, 1.0261],
        [0.8666, 0.8917, 0.9272,  ..., 1.2046, 1.0261, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.84314712,  9.96391045, 10.07238463, 10.06286074, 10.14044489,
       10.1371552 , 12.91143491, 13.66051884,  5.26342203, 12.12641969,
        9.94173708,  5.01114719, 13.83139024,  9.27474386,  4.74468406,
        7.20482224, 10.08001894, 10.31584102, 11.20100795,  5.90035324,
        9.34127013, 11.72802491, 10.46238054,  9.17016857, 10.26545792,
        4.7663526 ,  7.78980299,  9.50913849,  9.37989631,  7.16747878,
       10.68031351,  9.7751189 ,  9.05843201, 12.86648771, 10.61047596,
        9.78147542, 10.06437598,  9.60080639,  8.04664886,  5.19577146,
        9.20340031,  7.53884104,  8.74470218,  8.43626114, 13.37121923,
       13.77075174, 11.00508681, 10.61417181, 12.9089559 ,  7.72291946,
        8.06990483,  5.14124612, 10.91601655,  5.23878371, 12.97370371,
        8.79284466, 12.5085995 ,  9.25336371,  9.47369628,  7.27340834,
        7.15990676, 13.40850328, 14.09357547, 11.63724792,  5.05444288,
        7.76457095,  7.49076899, 10.9036226 , 13.40024224, 13.5247215 ,
       12.18998195,  5.14124612, 12.77402351,  7.39004226,  9.09502385,
       10.9036226 ,  7.16058587,  4.85995002, 12.4217523 , 10.14436912,
        6.80559669,  5.24461068,  5.5205189 ,  8.87587457, 11.0050379 ,
        7.63726134, 11.18839169, 13.96746816,  6.3949043 ,  7.54127425,
        7.46053281,  8.05668609, 14.09357547,  7.5158489 ,  6.1561646 ,
       10.31886066, 10.9036226 ,  8.49161154,  6.24187044,  9.79039371,
        9.45995955, 13.01728313,  9.32094055, 13.01187484,  8.77913428,
        9.4997987 ])}
{'P': tensor([[7.2518e-03, 1.9079e-01, 1.5897e-01,  ..., 1.6767e-05, 1.6197e-07,
         3.9718e-04],
        [1.4206e-04, 1.0118e-02, 2.2021e-01,  ..., 2.5513e-05, 2.5012e-07,
         5.8373e-04],
        [1.6498e-04, 3.5832e-04, 2.9001e-02,  ..., 1.1606e-04, 1.0722e-06,
         2.5155e-03],
        ...,
        [2.3645e-05, 2.0619e-05, 2.2694e-05,  ..., 8.1599e-02, 1.0834e-02,
         1.1273e-03],
        [1.8923e-06, 1.3120e-06, 1.0654e-06,  ..., 6.7323e-05, 6.0101e-02,
         1.3620e-05],
        [1.3214e-04, 1.4206e-04, 1.8886e-04,  ..., 3.0081e-02, 4.8542e-04,
         8.6159e-02]]), 'X': tensor([[ -8.9624,  -4.0944,  -4.9622],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3413,   0.4883,  -6.8416],
        [ -2.2188,   7.8218,   8.6064],
        [  6.0890,  -0.8382,   2.9166],
        [  0.1677,  -4.7447,  -6.1529],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.5720,   4.6310,   2.1717],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.2182,  13.8007,   5.1889],
        [  6.2630,   2.1057,  -5.2136],
        [ -1.6086,   7.8928,   4.3385],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -9.3076,   0.7852,  -6.7010],
        [ -8.8934,   2.2597,   3.5831],
        [  3.4140,  -4.1803,  -8.8224],
        [  1.5246,  -4.7838,  -5.2095],
        [ -7.2515,   5.0195,   4.6434],
        [ -2.1533, -10.5778,   1.0112],
        [  5.0277,   4.9589,   7.0175],
        [  6.3413,   0.4883,  -6.8416],
        [  5.3602,  -2.3808,   7.2098],
        [ -5.9225,  -8.5355,   2.4778],
        [ 12.2592,   3.9425,   2.0383],
        [  3.6469,   6.3624,  -1.8414],
        [  7.5309,  11.2617,   7.2656],
        [ -5.3328,  -8.9992,   7.3440],
        [ 11.4743,   6.2478,  -2.1100],
        [  1.7815,  -4.7481,   4.4289],
        [ -7.9298,   4.1367,  -3.9773],
        [  3.3094,  13.4969,  -3.6507],
        [  6.3497,   1.3680,  -1.9076],
        [  1.6954,  -4.7616,   9.8198],
        [  3.5284, -10.7484,   5.3347],
        [  7.0171,  11.6439,  -5.0178],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.6093,  -1.9705,   6.4807],
        [ -5.1377,   6.7751,  -0.4438],
        [  2.3200,   7.1756,   1.1938],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.2732,  -4.7619,  -1.1577],
        [ -7.5401,   4.6718,   9.0033],
        [  1.7228,  -4.7575,   8.3396],
        [  6.0560,  -0.9421,  -5.0334],
        [  3.3744,   6.5623,  -6.3636],
        [  2.6310, -10.9195,   3.8054],
        [ 11.8951,  -3.1570,   7.0742],
        [  2.8474,   6.8979,   6.6457],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.8617,   3.4853,  -9.8080],
        [  3.2692,  13.5098,  -4.9032],
        [ -2.1382,   7.8339,   7.3524],
        [  0.2435,   7.8191,  -6.4709],
        [ -7.6650,   4.5092,   0.2520],
        [ -7.6293,  -6.6859,  -5.5161],
        [  3.5965,  -4.0754,  -3.5035],
        [ -7.6650,   4.5092,   0.2520],
        [ 11.7162,   5.6728,  -8.9372],
        [  9.0129,  -7.7072,   9.1400],
        [  6.3534,   1.2990,  -7.2942],
        [  9.0004,  -7.7195,   4.6945],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.7963,  -1.6018,   1.8856],
        [  6.0031,  12.2958,  -6.5992],
        [  3.9085,   6.1512,  -2.2560],
        [ -9.3076,   0.7852,  -6.7010],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.5375,   7.0676,  -4.3490],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.6314,  -1.9302,  -9.6543],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 12.5199,  -0.4711,   3.0436],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.5424,  -2.0885,   7.2487],
        [  5.6414, -10.0303,   7.8525],
        [  5.4242, -10.1261,   3.9595],
        [  1.8766,   7.3694,   1.1887],
        [ -5.9987,   6.1949,  -6.2114],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.7784,  -1.6401,   9.5587],
        [ -3.4100,   7.5436,  -7.6980],
        [  2.5318,  13.7231,   6.3692],
        [  6.2905,   1.9366,   2.1169],
        [ -2.1382,   7.8339,   7.3524],
        [ -2.2188,   7.8218,   8.6064],
        [ 12.5224,  -0.4497,   4.8448],
        [ -5.3328,  -8.9992,   7.3440],
        [ -6.7992,   5.4991,   9.9186],
        [  1.8766,   7.3694,   1.1887],
        [ -9.3604,   0.4855,  -3.0158],
        [  5.6258,  -1.9404,  -6.0715],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.7903,  -4.7467,   0.1168],
        [  5.5158,  -2.1337,   3.5246],
        [ 11.4235,   6.3594,   5.1663],
        [  1.0391,  -4.8141,  -0.0768]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0220, 0.0291,  ..., 1.1953, 1.5897, 0.9673],
        [0.0220, 0.0000, 0.0485,  ..., 1.2100, 1.6056, 0.9790],
        [0.0291, 0.0485, 0.0000,  ..., 1.1669, 1.5609, 0.9404],
        ...,
        [1.1953, 1.2100, 1.1669,  ..., 0.0000, 0.4092, 0.2944],
        [1.5897, 1.6056, 1.5609,  ..., 0.4092, 0.0000, 0.6963],
        [0.9673, 0.9790, 0.9404,  ..., 0.2944, 0.6963, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.85330646,  9.96152788, 10.09979747,  9.88571713,  9.94285511,
       10.06437598,  7.96181288,  6.36004393,  8.1303945 ,  6.1463896 ,
        4.74770899,  8.83341608,  8.87587457,  5.05774295, 13.97779839,
        6.60752373,  8.0550286 , 12.32545637,  9.34061938,  9.17595899,
        5.39723241,  5.0209005 ,  8.81930525, 10.794755  ,  7.06169391,
        6.36004393,  5.86519525, 10.38897563, 12.87752052,  7.33350642,
       13.54774355, 10.4606265 , 13.06498554,  5.0713364 ,  8.94395628,
       13.89671314,  6.49538841,  5.05444288, 11.31276109, 13.59480906,
        9.5642993 , 12.27993098,  5.94535529,  8.50278896,  7.54127425,
       12.41012046,  4.76969364,  8.87007495,  5.05981914,  6.12886024,
        7.37902766, 11.23201619, 12.30695098,  7.46248947,  4.78258092,
        6.81961996, 13.89974693,  8.12043983,  7.822854  ,  8.89301516,
       10.14436912,  5.43544626,  8.89301516, 13.01728313, 11.85890664,
        6.48486171, 11.85743425, 12.00995898,  6.01356004, 13.6829843 ,
        7.28793663,  9.34061938,  8.47371868,  4.8763135 ,  9.78285424,
        7.5092834 ,  5.19986797,  5.95297459, 10.60226614, 12.52875793,
       10.31613269,  5.26067019,  5.92281142, 11.50791859, 11.48734978,
        7.60463349,  8.62329466,  6.18668463,  6.00663314,  8.27853076,
       13.95472331,  6.58184474,  8.12043983,  8.1303945 , 12.53047489,
       10.4606265 ,  8.74470218,  7.60463349,  9.37296102,  5.9510489 ,
        9.17016857, 10.04916678,  5.07306274,  5.91408504, 13.0743399 ,
        4.92497514])}
{'P': tensor([[7.0252e-03, 2.4044e-01, 1.6111e-01,  ..., 1.6763e-06, 1.2945e-07,
         6.2390e-04],
        [2.5412e-04, 2.4031e-02, 4.0691e-02,  ..., 5.5728e-06, 5.5867e-07,
         3.3454e-03],
        [9.7768e-05, 2.1921e-03, 8.0175e-03,  ..., 2.0334e-06, 1.6519e-07,
         7.9933e-04],
        ...,
        [2.1635e-04, 1.4476e-04, 1.5313e-04,  ..., 7.4182e-02, 1.3188e-05,
         4.3641e-06],
        [2.5667e-06, 1.9467e-06, 1.5810e-06,  ..., 2.0072e-04, 7.4704e-02,
         8.8227e-07],
        [2.0539e-05, 2.9884e-05, 2.8767e-05,  ..., 1.4108e-05, 8.7079e-05,
         9.7357e-02]]), 'X': tensor([[ -8.9606,  -4.0994,   1.7886],
        [ -8.5149,  -5.1905,   1.5239],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.1531,  -5.8748,   2.6934],
        [ -7.9925,  -6.1423,   2.8990],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.0039,   7.3173,  -1.5885],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.6038,   0.7665,  -3.4439],
        [  2.2659,  13.7894,   6.5436],
        [  6.3549,   0.7096,   6.9198],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.3551,   0.7149,   1.3344],
        [  5.2939,  -2.4784,  -4.5840],
        [ 12.3453,  -1.5462,   7.0212],
        [  5.9526,   3.2525,  -4.3821],
        [ -8.9379,  -4.1655,  -2.8114],
        [  1.4489,   7.5256,   4.4529],
        [  7.4132,  11.3526,   0.3322],
        [  7.2340,  -9.1505,  -5.8321],
        [  7.2340,  -9.1505,  -5.8321],
        [  2.2373,  -4.6506,  -1.7334],
        [  1.6308,  13.9254,  -2.0984],
        [  4.2275,  -3.6320,  -0.2714],
        [ -5.6651,   6.4380,   1.3660],
        [ -4.2271,   7.2387,  -1.8316],
        [ 12.4996,   2.6335,  -6.3166],
        [  6.2555,   2.1474,   4.1685],
        [  1.8721,  -4.7322,   0.5524],
        [  1.8719,   7.3713,  -3.7465],
        [  1.1994,   7.6035,   2.4295],
        [  5.6022,  -1.9833,   8.3081],
        [ 11.2209,  -4.7092,   2.6117],
        [  8.3718,  -8.2953,   8.4666],
        [ 11.3118,   6.5957,  -4.5667],
        [  2.7590,  13.6621,   9.4496],
        [ -5.5525,   6.5146,  -4.4984],
        [ -8.9178,   2.1936,   7.5388],
        [  5.9587,  12.3216,   9.1695],
        [  2.6614,  -4.5192,  -9.3641],
        [  3.6469,   6.3624,  -1.8414],
        [ -8.3717,  -5.4769,  -9.2228],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.5382,   7.0672,   4.5097],
        [  1.0391,  -4.8141,  -0.0768],
        [  5.1787,  -2.6383,  -4.2533],
        [  7.0515,  11.6194,   8.4289],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.1752,   2.5242,   3.5874],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.9284,  11.7061,  -0.3732],
        [  0.3493,  -4.7727,  -6.5682],
        [  6.3124,   0.2057,  -0.4593],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4645,  -1.4880,  -2.0226],
        [  9.8082,  -6.8396,   4.9615],
        [  1.1139,   7.6281,  -2.5680],
        [  5.6260,  -1.9401,   3.0259],
        [  8.0877,  -8.5297,  -1.1050],
        [  5.9655,   3.2172,  -7.5953],
        [ -1.6334, -10.7207,   7.1397],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -9.1178,   1.5784,   2.1410],
        [ -7.4807,   4.7463,   2.0616],
        [ -5.3179,  -9.0101,   9.4472],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.0547,   7.2957,  -1.9343],
        [  9.6816,   9.1536,  -5.8543],
        [  6.8704,  -9.3809,   7.5769],
        [ -1.6086,   7.8928,   4.3385],
        [  2.5469,  13.7192,  -9.6935],
        [  2.3261,  13.7749,   5.9121],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.8836,   3.4315,  -4.0327],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.4267,  -5.3697,   2.4172],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.0243,   7.3086,   7.5432],
        [ -8.5900,   2.9767,   7.4811],
        [  0.3715,  -4.7756,   0.2312],
        [  6.3587,   1.1596,  -3.6629],
        [  4.4290,  -3.4596,   1.4181],
        [ -2.2508, -10.5479,  -2.7235],
        [ 12.4901,  -0.7052,   1.7315],
        [  9.0424,  -7.6779,  -7.5286],
        [ -4.2271,   7.2387,  -1.8316],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.0047,   3.1062,  -6.7914],
        [ -7.8280,   4.2848,  -7.1955],
        [  0.3231,  -4.7691,  -1.6120],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.8835,  -4.3178,   9.3080],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.3246,   1.6696,  -0.1544],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.6231,   5.4556,   8.4973],
        [ -1.0407,   7.9166,  -6.6021],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.8474,   6.8979,   6.6457],
        [ 11.0373,   7.1299,  -8.5895],
        [ -1.0482, -10.8487,  -5.9066]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0238, 0.0392,  ..., 0.8792, 1.4445, 0.6556],
        [0.0238, 0.0000, 0.0628,  ..., 0.8841, 1.4286, 0.6321],
        [0.0392, 0.0628, 0.0000,  ..., 0.8660, 1.4659, 0.6928],
        ...,
        [0.8792, 0.8841, 0.8660,  ..., 0.0000, 1.1586, 1.1152],
        [1.4445, 1.4286, 1.4659,  ..., 1.1586, 0.0000, 1.0190],
        [0.6556, 0.6321, 0.6928,  ..., 1.1152, 1.0190, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.85384623,  9.97220799,  9.99211663, 10.06850677, 10.04916678,
       10.08001894,  9.51074763,  7.58668144,  9.23178204, 12.62711252,
       13.97430191,  6.39438044, 10.78533838, 12.11735334,  6.39520397,
        5.84533345, 12.44177641,  6.78325151,  9.86089579,  7.66377554,
       13.55868684, 11.66453059, 11.66453059,  5.16077764, 14.02058621,
        5.57341623,  8.57560275,  8.38252937, 12.77402351,  6.61386436,
        5.08910491,  7.60529892,  7.69752262,  5.94293185, 12.16901615,
       11.785566  , 13.09426142, 13.93790235,  8.55982105,  9.18358562,
       13.68672855,  5.24461068,  7.33350642, 10.00410625,  5.0209005 ,
        9.58347456,  7.50917551,  4.92497514,  5.81199422, 13.59170809,
        9.76896904,  6.67121748,  5.52416368, 13.60275211,  4.78544875,
        6.31576264,  5.13616224,  9.58072286, 11.95743558,  7.70897675,
        5.95111804, 11.75438006,  6.77777606, 10.84437705,  9.88571713,
        9.25336371,  8.85939626, 10.46238054, 12.67680439,  5.79391825,
        7.57946459, 13.32374056, 11.62770332,  8.0550286 , 13.95361098,
       13.96987806, 10.48840476,  6.81117802, 11.62158554,  9.99211663,
       10.41909211,  7.58378655,  9.09119351,  4.79003361,  6.46356806,
        5.62005629, 10.78533838, 12.50996828, 11.86239961,  8.38252937,
        9.34061938,  6.76056625,  8.92395381,  4.78002665,  5.05989547,
        9.87720102,  8.44940661, 12.889695  ,  6.54127431, 11.75618513,
        7.15100303,  7.98469166,  9.55720651,  7.46248947, 13.1399181 ,
       10.8992524 ])}
{'P': tensor([[1.0795e-02, 2.6754e-01, 2.4742e-01,  ..., 6.4212e-06, 2.0334e-06,
         2.6770e-04],
        [8.3627e-05, 1.1781e-02, 9.5547e-03,  ..., 1.4882e-05, 1.7245e-06,
         4.8155e-04],
        [1.0003e-04, 8.0245e-03, 1.3730e-02,  ..., 1.6216e-05, 1.7899e-06,
         4.4060e-04],
        ...,
        [6.0678e-07, 5.2192e-07, 4.9522e-07,  ..., 2.0823e-02, 5.7861e-05,
         4.0839e-05],
        [1.5526e-03, 8.3532e-04, 6.7110e-04,  ..., 5.5370e-06, 9.0273e-02,
         2.7084e-04],
        [7.7783e-05, 9.7536e-05, 8.6863e-05,  ..., 1.0164e-02, 3.8282e-04,
         9.0735e-02]]), 'X': tensor([[ -8.9379,  -4.1655,  -2.8114],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.3376,  -4.6231,   9.8436],
        [ 12.3453,  -1.5462,   7.0212],
        [  7.5784,  -8.9141,   5.8410],
        [  1.3670,  -4.7990,   3.2603],
        [  0.5850,  -4.7982,   9.6639],
        [  7.0171,  11.6439,  -5.0178],
        [  5.9526,   3.2525,  -4.3821],
        [  5.1519, -10.2387,   9.8191],
        [  5.4167, -10.1293,   6.2156],
        [ 11.4495,   6.3026,   6.8257],
        [ -6.4844,  -8.0230,   8.5209],
        [ -8.3062,  -5.6008,   0.0287],
        [ -2.5161,   7.7699,  -9.3531],
        [  2.1824,  -4.6647,   1.1985],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.4573,   3.2448,   5.4370],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.1624,  -3.4286,   2.0647],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -5.9225,  -8.5355,   2.4778],
        [  4.4049,   5.6892,  -4.4394],
        [  6.0788,  -0.8710,  -6.8719],
        [ -6.4464,   5.8272,   7.5038],
        [ -3.1204, -10.2345,   9.2288],
        [  8.3791,  10.5409,   5.5590],
        [ 11.2939,   6.6325,  -5.8311],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.0051,   9.9234,   5.4411],
        [  2.1504,  -4.6726,   8.7908],
        [  0.3493,  -4.7727,  -6.5682],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.3107,  -2.7697,  -3.2919],
        [  7.9403,  -8.6455,  -8.7433],
        [  6.1372,   2.6726,  -6.7885],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.0608,   3.9362,  -5.9567],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.0243,   7.3086,   7.5432],
        [ -9.1658,  -3.4155,  -2.8880],
        [  1.9312,  13.8649,   8.3563],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.8979,   3.3958,  -3.8713],
        [  2.6078, -10.9229,   6.2869],
        [  3.4401,  13.4540,   4.6576],
        [ -9.4117,   0.1160,   8.2574],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.5859,  -5.0393,   0.5476],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.7298,  13.3537,  -9.4653],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.1706,   1.3863,   8.4032],
        [  6.5428,  -9.5728,  -2.9897],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.2182,  13.8007,   5.1889],
        [  4.5800,   5.5034,  -3.9432],
        [  4.7544,  -3.1425,  -1.5315],
        [  8.7715,  -7.9390,   5.2900],
        [ -9.1658,  -3.4155,  -2.8880],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.1222,  -0.7261,   8.8204],
        [  7.0664,  -9.2590,   8.0879],
        [  9.6587,   9.1817,  -0.8212],
        [  2.4572,   7.1085,   7.7376],
        [ 10.3251,   8.2928,  -3.0006],
        [ -5.3382,   6.6533,   7.8054],
        [  3.8111, -10.6787,  -4.6539],
        [ -7.7842,   4.3466,  -1.7548],
        [  0.6757, -11.0361,  -1.3047],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 12.5650,   2.0132,   4.6272],
        [  6.3413,   0.4883,  -6.8416],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.9033,   2.2329,  -8.8912],
        [  4.5503,   5.5359,   7.4735],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.4572,   7.1085,   7.7376],
        [  3.9663,   6.1018,  -5.4104],
        [  3.4401,  13.4540,   4.6576],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.0534,  -6.0429,   2.4583],
        [  3.3069,   6.6089,   8.7377],
        [  1.7815,  -4.7481,   4.4289],
        [  4.0591,   6.0202,  -1.6339],
        [ -7.7239,   4.4298,   3.1216],
        [  5.6258,  -1.9404,  -6.0715],
        [  2.6613,   7.0021,   4.5783],
        [  5.9883,   3.1536,   6.2320],
        [  2.2437,  -4.6489,   6.1430],
        [ 12.5919,   0.4035,  -5.8460],
        [  9.3753,   9.5171,   3.9131],
        [  0.5960, -11.0335,   3.1144],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.3889,  -4.6082,  -8.7291]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0264, 0.0458,  ..., 1.0603, 0.6378, 0.9611],
        [0.0264, 0.0000, 0.0650,  ..., 1.0444, 0.6190, 0.9469],
        [0.0458, 0.0650, 0.0000,  ..., 1.0485, 0.6334, 0.9459],
        ...,
        [1.0603, 1.0444, 1.0485,  ..., 0.0000, 0.4402, 0.1270],
        [0.6378, 0.6190, 0.6334,  ..., 0.4402, 0.0000, 0.3695],
        [0.9611, 0.9469, 0.9459,  ..., 0.1270, 0.3695, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.86089579, 10.06407839, 10.0674887 , 10.1979669 , 10.23946247,
       10.31613269,  5.1805345 , 12.44177641, 11.70016217,  4.98989906,
        4.83371341, 13.59480906,  6.78325151, 11.46176259, 11.48663767,
       13.06957425, 10.31584102, 10.01804223,  8.16714836,  5.14999799,
       13.41713661,  9.05843201,  7.37902766,  9.78285424, 12.00995898,
       10.38897563,  7.19513806,  6.14086582,  8.68979052, 10.69963954,
       13.46553623, 13.09737053,  5.26067019, 13.40024224,  5.14371005,
        4.78544875,  6.80370254,  9.71392179, 11.73848065,  6.69389431,
       12.00995898, 10.36844813,  8.97052115, 12.04924159,  8.30125201,
        7.58378655,  9.78147542, 13.99877413,  9.5014146 ,  7.88677538,
        6.80559669, 11.22993218, 13.88683436,  9.41244865,  8.83725081,
        9.95552626, 10.54658597,  9.14044984, 13.86480175,  6.02588517,
        9.27474386, 11.59512194,  7.83486943, 13.13205357, 13.97779839,
        7.15990676,  5.69913817, 11.83075005,  9.78147542,  4.8763135 ,
        6.16512856, 11.64746882, 13.32645552,  7.52117132, 13.2430377 ,
        8.53014915, 11.33837396,  8.91551231, 11.05672136, 10.5575601 ,
       12.72524615,  6.36004393, 13.27994304,  9.17905491,  7.16598273,
        5.41348121,  7.52117132,  7.27759233, 13.88683436,  9.62958973,
        5.83560009, 10.06850677,  7.39004226,  5.0713364 ,  7.26074925,
        8.90405823,  5.9510489 ,  7.49076899,  6.76791483,  5.16205087,
       12.59840193, 13.35927527, 11.04953767, 12.32545637,  8.20436764,
        5.19065439])}
{'P': tensor([[6.4765e-03, 3.1373e-01, 2.6204e-01,  ..., 9.0605e-06, 2.8831e-07,
         2.1565e-04],
        [1.7837e-04, 1.4100e-02, 3.6353e-02,  ..., 1.6849e-05, 6.4488e-07,
         5.4341e-04],
        [1.7822e-04, 3.9774e-03, 1.2661e-02,  ..., 1.3498e-05, 5.4464e-07,
         4.8993e-04],
        ...,
        [7.9513e-05, 9.3425e-05, 8.5006e-05,  ..., 3.1984e-02, 1.1584e-05,
         3.5771e-05],
        [1.7084e-05, 1.5311e-05, 1.4054e-05,  ..., 3.1639e-04, 9.3797e-02,
         2.7108e-05],
        [1.2443e-04, 1.2738e-04, 1.2705e-04,  ..., 1.5645e-05, 2.3256e-04,
         1.2987e-01]]), 'X': tensor([[ -8.8835,  -4.3178,   9.3080],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.3535,  -5.5118,   9.8399],
        [ -6.7745,  -7.7264,   8.4684],
        [ -6.4844,  -8.0230,   8.5209],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -6.3489,   5.9117,   6.5161],
        [ -9.4714,  -1.3301,  -3.1998],
        [  0.1382, -11.0073,   1.0026],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.1440, -10.9811,   8.5281],
        [ -8.3653,   3.4184,   9.3454],
        [ 11.7726,  -3.4876,  -6.7402],
        [  1.5246,  -4.7838,  -5.2095],
        [ -0.7599, -10.8995,  -6.0366],
        [  1.9312,  13.8649,   8.3563],
        [  2.5318,  13.7231,   6.3692],
        [ -4.2040,   7.2486,   2.9875],
        [  1.8320,   7.3871,   5.1452],
        [  0.2435,   7.8191,  -6.4709],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.0619,   8.6627,  -9.2884],
        [  9.9058,  -6.7203,   5.6928],
        [  9.0004,  -7.7195,   4.6945],
        [  2.3367,   7.1676,  -9.5263],
        [  3.6817,   6.3354,   6.1233],
        [ 10.4737,  -5.9570,  -1.2108],
        [  5.0277,   4.9589,   7.0175],
        [ 12.5919,   0.4035,  -5.8460],
        [ -7.6734,   4.4980,  -5.4248],
        [ -8.5859,  -5.0393,   0.5476],
        [  9.2811,  -7.4336,   5.1771],
        [  1.9204,   7.3518,   8.4781],
        [  9.4554,   9.4245,   9.8800],
        [ -8.1531,  -5.8748,   2.6934],
        [ -0.0410,   7.8591,   9.3457],
        [  5.9912,   3.1453,  -3.9989],
        [  4.9452, -10.3187,  -7.8621],
        [  2.8602,   6.8905,   5.8779],
        [ -0.8569,   7.9156,  -6.5557],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.9282,   2.1648,   8.4894],
        [  6.3587,   1.1596,  -3.6629],
        [  2.7976,  -4.4682,  -4.2020],
        [  1.3103,  -4.8032,  -0.9300],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.9967,  -4.8144,  -9.7370],
        [  7.3592,  11.3937,   6.2407],
        [ 11.9641,  -2.9568,   7.7211],
        [  8.7131,  10.2216,   7.7915],
        [  7.5064,  -8.9651,  -0.0994],
        [ -8.6603,   2.8250,   5.9181],
        [ 12.2328,  -2.0376,  -1.1234],
        [  0.8700,  -4.8132,  -7.8042],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.8602,   6.8905,   5.8779],
        [ 12.2914,   3.8037,   6.0978],
        [  5.1793,   4.7461,   0.3899],
        [ 12.5996,   1.3971,  -1.0694],
        [  2.4572,   7.1085,   7.7376],
        [  4.3523, -10.5232,   8.2138],
        [  5.4732,  -2.2041,  -4.3830],
        [  1.9204,   7.3518,   8.4781],
        [ 12.5154,   2.5072,  -9.3274],
        [  6.0890,  -0.8382,   2.9166],
        [  0.7275,  14.0667,  -0.9815],
        [  6.0047,   3.1062,  -6.7914],
        [ -6.3252,  -8.1760,  -1.0891],
        [  4.2096,   5.8815,   1.9230],
        [  1.0093,   7.6567,   2.3526],
        [  3.7121,   6.3117,   0.6738],
        [  5.8930,  12.3592,  -8.9201],
        [  5.8979,   3.3958,  -3.8713],
        [  0.3493,  -4.7727,  -6.5682],
        [ 11.2767,  -4.6006,   0.7868],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.6614,  -4.5192,  -9.3641],
        [  9.1820,   9.7334,  -2.7932],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.4351,  -1.8891,   3.5642],
        [ -2.6485, -10.4151,   0.8766],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.3118,   4.5452,   4.2114],
        [ -5.0943,  -9.1682,   0.5073],
        [  1.8766,   7.3694,   1.1887],
        [  8.8474,  10.0868,  -8.8644],
        [  6.3549,   0.7096,   6.9198],
        [  1.4150,  13.9647,  -4.0915],
        [ -6.2668,   5.9808,   2.5614],
        [  6.3031,   1.8477,  -7.1706],
        [ -6.7745,  -7.7264,   8.4684],
        [  9.9750,  -6.6338,  -1.2440],
        [  6.1222,  -0.7261,   8.8204],
        [  9.5261,  -7.1675,   1.4206],
        [  3.2647, -10.8066,  -0.3768],
        [  6.2905,   1.9366,   2.1169],
        [  5.4336,  -2.2676,   5.6933],
        [  5.1886,  -2.6250,   6.4677],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.5158,  -2.1337,   3.5246],
        [ -9.1907,  -3.3180,  -5.5943],
        [  3.9663,   6.1018,  -5.4104],
        [  3.3839,  -4.1966,   8.3574]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0045, 0.0041,  ..., 1.0260, 1.2639, 0.9212],
        [0.0045, 0.0000, 0.0054,  ..., 1.0232, 1.2594, 0.9176],
        [0.0041, 0.0054, 0.0000,  ..., 1.0286, 1.2633, 0.9181],
        ...,
        [1.0260, 1.0232, 1.0286,  ..., 0.0000, 0.9294, 1.3633],
        [1.2639, 1.2594, 1.2633,  ..., 0.9294, 0.0000, 0.8497],
        [0.9212, 0.9176, 0.9181,  ..., 1.3633, 0.8497, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.87720102,  9.9353921 ,  9.94173708, 10.00802627, 10.27572673,
       10.31584102,  9.96152788,  8.67502348,  9.5642993 , 11.00812996,
        9.53015376, 11.18839169,  9.0368458 , 12.27835996,  5.0209005 ,
       10.92597196, 13.99877413, 13.95472331,  8.37954853,  7.61088125,
        7.822854  , 13.45642379, 13.27717982, 11.9702762 , 11.85743425,
        7.53884104,  7.32754882, 12.04924159,  7.06169391, 12.59840193,
        8.89458273,  9.95552626, 11.89105259,  7.5984809 , 13.35013374,
       10.04916678,  7.85919687,  6.76661595, 11.4424816 ,  7.46053281,
        7.96181288, 10.33958639,  9.18690018,  6.46356806,  5.27178363,
        4.978704  , 12.25806559,  4.91652556, 13.56367857, 12.32408516,
       13.43127131, 11.69264243,  9.10945678, 12.4013153 ,  4.89121805,
       12.59840193,  7.46053281, 12.86648771,  7.02496223, 12.67680439,
        7.52117132, 11.38774149,  5.90035324,  7.5984809 , 12.76408562,
        6.1463896 , 14.08549802,  6.76056625, 10.33712258,  7.23276582,
        7.72291946,  7.32233733, 13.6922425 ,  6.80559669,  4.78544875,
       12.17901033, 12.40074901,  5.24461068, 13.38091121,  8.94395628,
        9.62238828, 10.74655346, 12.52372629,  9.61942445,  6.99097013,
       10.48840476,  7.60463349, 13.41713661,  6.39438044, 14.03618649,
        8.66273171,  6.56833864, 10.27572673, 11.97949273,  6.16512856,
       11.92133129, 11.28895489,  6.58184474,  5.88782913,  5.81481044,
       12.05395056,  7.9337647 ,  5.91408504,  9.7712321 ,  7.27759233,
        5.39098669])}
{'P': tensor([[1.1618e-02, 2.8301e-01, 1.6275e-01,  ..., 1.2406e-02, 7.4979e-06,
         9.6575e-05],
        [2.8590e-04, 1.7283e-02, 5.8830e-02,  ..., 3.1467e-02, 8.2409e-06,
         1.7039e-04],
        [1.8407e-04, 9.6699e-04, 1.5901e-02,  ..., 2.6650e-02, 6.3292e-06,
         2.2298e-04],
        ...,
        [2.5843e-05, 3.3926e-05, 4.7329e-05,  ..., 4.9608e-02, 4.3345e-06,
         2.5827e-03],
        [2.6998e-02, 1.6454e-02, 1.6019e-02,  ..., 5.6140e-04, 7.0709e-02,
         1.2025e-04],
        [4.1528e-05, 3.8822e-05, 4.1323e-05,  ..., 4.9253e-05, 1.7140e-04,
         6.8573e-02]]), 'X': tensor([[ -8.8631,  -4.3728,  -1.1746],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.6954,  -4.7616,   9.8198],
        [  3.7298,  13.3537,  -9.4653],
        [  6.3460,   1.4252,   1.2291],
        [  2.3367,   7.1676,  -9.5263],
        [ -3.3812, -10.1235,   7.4384],
        [  7.3592,  11.3937,   6.2407],
        [  5.1472,  -2.6801,   8.1194],
        [  1.7301,   7.4262,   5.0522],
        [ -0.1626, -10.9796,   6.4629],
        [ -9.1178,   1.5784,   2.1410],
        [ -2.3317,   7.8034,  -3.6274],
        [  3.8182,  -3.9347,  -1.7961],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.1771, -10.9776,  -3.2007],
        [ -5.1046,  -9.1610,   2.3418],
        [  7.2340,  -9.1505,  -5.8321],
        [  8.6271,  -8.0715,  -7.4813],
        [  6.2630,   2.1057,  -5.2136],
        [  6.9433,  -9.3362,  -0.3421],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -8.0611,  -6.0302,   4.6962],
        [  5.8836,   3.4315,  -4.0327],
        [  2.9470,  13.6085,   2.9809],
        [  6.3542,   0.6934,  -3.3840],
        [ 11.4235,   6.3594,   5.1663],
        [  5.0190,   4.9705,  -7.9752],
        [ -1.7290,   7.8825,  -3.3831],
        [  1.3705,  13.9723,   7.9266],
        [ 10.3251,   8.2928,  -3.0006],
        [ -5.9938,  -8.4745,   3.9735],
        [  6.3549,   0.7096,   6.9198],
        [  2.1440, -10.9811,   8.5281],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.0400,   8.6922,  -6.2847],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.4501,  -1.7133,   2.2980],
        [  5.7963,  -1.6018,   1.8856],
        [  2.2860,   7.1916,  -6.1011],
        [  6.3534,   1.2990,  -7.2942],
        [  4.8432,   5.1968,   1.5006],
        [ -9.4541,  -0.3327,   8.8246],
        [  7.1441,  11.5529,   2.3232],
        [ -8.6681,  -4.8556,   9.2518],
        [  1.9204,   7.3518,   8.4781],
        [  5.7218,  -9.9934,   6.5764],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 12.5996,   1.3971,  -1.0694],
        [  9.0004,  -7.7195,   4.6945],
        [  5.8318,   3.5567,  -4.7692],
        [  5.8979,   3.3958,  -3.8713],
        [ -3.4100,   7.5436,  -7.6980],
        [  3.9591, -10.6391,  -6.2557],
        [  9.8082,  -6.8396,   4.9615],
        [  1.3705,  13.9723,   7.9266],
        [  0.9809,  -4.8145,   6.8706],
        [  1.8320,   7.3871,   5.1452],
        [ -4.4520,   7.1370,  -8.7867],
        [ 11.2939,   6.6325,  -5.8311],
        [ -1.5451,   7.8975,   7.6104],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.4572,   7.1085,   7.7376],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.9178,   2.1936,   7.5388],
        [  5.6258,  -1.9404,  -6.0715],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.0890,  -0.8382,   2.9166],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -8.0965,  -5.9712,   4.3466],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.4645,  -1.4880,  -2.0226],
        [  7.3592,  11.3937,   6.2407],
        [  3.9895,   6.0816,  -6.5235],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.0190,   4.9705,  -7.9752],
        [  2.3889,  -4.6082,  -8.7291],
        [  4.6357,   5.4415,   7.9067],
        [  2.1771, -10.9776,  -3.2007],
        [ -5.2752,  -9.0409,  -8.2568],
        [  2.9410,   6.8428,   5.4458],
        [ -9.3604,   0.4855,  -3.0158],
        [  1.7301,   7.4262,   5.0522],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.1222,  -0.7261,   8.8204],
        [  0.8940, -11.0402,  -0.9783],
        [ -7.9667,   4.0814,  -0.6397],
        [  1.9505,  -4.7171,   9.9868],
        [  9.9217,  -6.7006,  -5.6728],
        [ 10.9599,  -5.1849,  -3.5526],
        [  2.6321,  -4.5296,  -2.5159],
        [ -2.2188,   7.8218,   8.6064],
        [ -9.1813,  -3.3550,   1.8827],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.2911,  -2.4824,  -3.6877]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0391, 0.1057,  ..., 0.2165, 0.3548, 0.9544],
        [0.0391, 0.0000, 0.1445,  ..., 0.1811, 0.3724, 0.9514],
        [0.1057, 0.1445, 0.0000,  ..., 0.3125, 0.3136, 0.9571],
        ...,
        [0.2165, 0.1811, 0.3125,  ..., 0.0000, 0.4189, 0.8725],
        [0.3548, 0.3724, 0.3136,  ..., 0.4189, 0.0000, 0.6683],
        [0.9544, 0.9514, 0.9571,  ..., 0.8725, 0.6683, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.88310614,  9.97822473, 10.0674887 , 10.23946247, 10.01804223,
       10.32715715,  5.05444288, 13.86480175,  6.50409372,  7.53884104,
       10.67322527, 13.56367857,  5.80312949,  7.62509582, 10.98076873,
        9.25336371,  8.14433542,  5.48276038, 13.12299145, 11.1913598 ,
       10.48721382, 11.66453059, 11.81423161,  6.60752373, 11.63502679,
        9.50654967, 10.52357479,  8.7273569 ,  9.66977802, 10.06703358,
        6.81117802, 13.92390689,  6.39187285, 13.0743399 ,  7.06372172,
        8.06990483, 14.0393949 , 13.2430377 , 10.37998078,  6.39438044,
       11.18839169,  5.48276038,  6.00663314, 13.27994304,  9.62238828,
        9.60412735,  6.01356004,  7.54620849,  6.48486171,  7.10378919,
        9.45995955, 13.58333857,  9.9353921 ,  7.5984809 , 11.51557166,
       12.04924159, 12.67680439, 11.85743425,  6.83083863,  6.80559669,
        8.27853076, 11.35182589, 11.95743558, 14.0393949 ,  4.91336994,
        7.61088125,  8.41171369, 13.09737053,  8.04718433,  9.55593374,
        7.52117132, 11.62158554,  9.18358562,  5.9510489 , 10.60226614,
        6.1463896 , 10.52587243, 10.06023665, 10.52587243, 10.37998078,
        9.58072286, 13.56367857,  7.27340834,  6.07992623,  7.06372172,
        5.19065439,  7.14837514, 11.1913598 , 10.46739073,  7.44805901,
        9.37296102,  7.62509582,  6.00663314,  9.11473391,  6.16512856,
       11.07637531,  8.95133961,  5.10446653, 11.97238624, 12.12448912,
        5.23878371,  8.1303945 ,  9.7751189 , 10.74655346,  8.85572292,
        5.84450127])}
{'P': tensor([[7.9219e-03, 2.9714e-01, 1.8301e-01,  ..., 5.4933e-04, 2.7278e-05,
         1.5741e-05],
        [1.2430e-04, 9.0839e-03, 1.9770e-01,  ..., 3.8895e-04, 4.2764e-05,
         2.1808e-05],
        [1.2218e-04, 2.2124e-04, 1.7907e-02,  ..., 4.5244e-04, 1.5042e-04,
         8.6424e-05],
        ...,
        [2.0498e-03, 2.8747e-03, 8.2719e-03,  ..., 9.2938e-02, 1.9685e-04,
         1.4860e-03],
        [2.7605e-06, 2.7534e-06, 2.8247e-06,  ..., 1.6702e-06, 7.8003e-02,
         1.3556e-04],
        [1.3435e-06, 1.3704e-06, 1.7853e-06,  ..., 3.3579e-06, 1.7385e-03,
         7.0654e-02]]), 'X': tensor([[ -8.8540,  -4.3971,  -5.2144],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -6.4822,  -8.0252,  -3.8983],
        [  6.2675,  12.1379,  -8.9720],
        [  4.0078,  -3.8018,  -6.2168],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.0608,   3.9362,  -5.9567],
        [  3.9663,   6.1018,  -5.4104],
        [  5.6314,  -1.9302,  -9.6543],
        [ 10.0400,   8.6922,  -6.2847],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.1231,  -3.5739,   8.4708],
        [ -5.8389,  -8.6056,  -3.5098],
        [  2.4697,  -4.5836,   1.4593],
        [ 12.1379,   4.4125,  -3.8833],
        [ 12.2328,  -2.0376,  -1.1234],
        [  5.9003,  12.3550,   2.1917],
        [  5.7151,  -1.7700,   3.1542],
        [ 12.3357,   3.6001,   0.7903],
        [  9.3424,  -7.3685,  -3.6807],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.4925,   9.3811,   8.9842],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.9945,  -4.8144,  -0.1897],
        [ -9.0574,   1.7801,   8.8986],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.2470,   8.4054,  -0.1177],
        [ 12.1588,   4.3364,  -8.0867],
        [  1.9505,  -4.7171,   9.9868],
        [ -2.5683, -10.4433,  -2.9182],
        [  2.8602,   6.8905,   5.8779],
        [  6.0047,   3.1062,  -6.7914],
        [  2.4356,  -4.5942,  -5.0808],
        [ 10.1503,   8.5414,  -7.2681],
        [  9.2979,   9.6048,   7.3138],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.3124,   0.2057,  -0.4593],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.5424,  -2.0885,   7.2487],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.4157,   0.0821,  -5.0025],
        [ -1.1885, -10.8211,   0.0865],
        [  0.1252, -11.0062,   7.2564],
        [  0.9967,  -4.8144,  -9.7370],
        [  0.5250, -11.0306,   5.5358],
        [ -3.0008, -10.2827,  -8.9480],
        [ -3.3999, -10.1152,  -1.2088],
        [ 12.4996,   2.6335,  -6.3166],
        [  3.2366,  -4.2732,   5.1532],
        [  3.5965,  -4.0754,  -3.5035],
        [ -7.7239,   4.4298,   3.1216],
        [  5.6928,  12.4710,   3.6001],
        [  5.9587,  12.3216,   9.1695],
        [ -2.2188,   7.8218,   8.6064],
        [  2.1625,  -4.6697,   7.0658],
        [ -4.1296,   7.2803,  -2.7194],
        [  3.7121,   6.3117,   0.6738],
        [  1.6232,  -4.7717,  -2.5655],
        [  6.3542,   0.6934,  -3.3840],
        [  3.9515,  -3.8425,   6.1729],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.3554,   4.8985,  -0.6665],
        [  1.9505,  -4.7171,   9.9868],
        [  0.7958,  -4.8109,  -8.2248],
        [  4.2339,  -3.6268,   6.3125],
        [ 11.7795,  -3.4697,   3.5624],
        [ 11.2939,   6.6325,  -5.8311],
        [ -4.1296,   7.2803,  -2.7194],
        [  3.4920,  -4.1366,  -8.3955],
        [  9.6816,   9.1536,  -5.8543],
        [ -0.0254, -10.9932,   3.2200],
        [  3.7916,  -3.9524,  -7.9547],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.0965,  -5.9712,   4.3466],
        [  0.1532,  -4.7422,  -7.0145],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.4309,  -0.0599,  -4.0830],
        [  9.0129,  -7.7072,   9.1400],
        [  6.2902,   0.0451,   3.7116],
        [  6.3205,   1.7068,  -3.8619],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.3200,   7.1756,   1.1938],
        [ -2.0840,   7.8415,  -7.8657],
        [  0.6757, -11.0361,  -1.3047],
        [  5.8201,   3.5841,  -8.0609],
        [ -0.1626, -10.9796,   6.4629],
        [  6.6319,  11.9066,   3.5165],
        [ -9.0574,   1.7801,   8.8986],
        [  3.3069,   6.6089,   8.7377],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.7485,  -9.9811,  -3.2355],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -3.9858,  -9.8334,  -3.7323],
        [  0.1047, -11.0046,   0.5188],
        [  8.8474,  10.0868,  -8.8644],
        [  6.2675,  12.1379,  -8.9720],
        [  5.9655,   3.2172,  -7.5953],
        [  5.8979,   3.3958,  -3.8713],
        [ -8.4267,  -5.3697,   2.4172],
        [  9.9217,  -6.7006,  -5.6728],
        [  7.5784,  -8.9141,   5.8410]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0093, 0.0405,  ..., 0.6017, 0.8950, 1.1574],
        [0.0093, 0.0000, 0.0485,  ..., 0.6111, 0.8948, 1.1651],
        [0.0405, 0.0485, 0.0000,  ..., 0.5672, 0.8738, 1.1168],
        ...,
        [0.6017, 0.6111, 0.5672,  ..., 0.0000, 1.0598, 0.7004],
        [0.8950, 0.8948, 0.8738,  ..., 1.0598, 0.0000, 1.0251],
        [1.1574, 1.1651, 1.1168,  ..., 0.7004, 1.0251, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.88571713,  9.96152788, 10.09979747, 10.06437598, 10.14436912,
       10.31613269, 13.66051884,  5.52416368, 11.1913598 ,  8.97052115,
        7.27759233,  5.95297459, 13.27994304,  4.78544875,  9.79814997,
       10.39942497,  5.20659892, 12.91505111, 12.4013153 , 13.69162795,
        5.98285319, 12.85032836, 11.89855267,  9.68324553, 13.34587027,
        5.57341623,  4.9160962 ,  9.23071698,  7.71075226, 13.25336001,
       12.9089559 ,  5.10446653, 10.75443106,  7.46053281,  6.76056625,
        5.19986797, 13.26590071, 13.36800073,  9.97964013,  6.31576264,
        8.04664886,  9.14973319,  5.92281142, 12.72524615,  9.41605684,
       10.88618114, 11.00694688,  4.91652556, 11.04313641, 10.71162681,
       10.67132524, 12.77402351,  5.36061974,  5.43544626,  8.90405823,
       13.70893492, 13.68672855,  8.1303945 ,  5.14607014,  8.36995953,
        7.32233733,  5.04028538,  6.39187285,  5.51177501, 12.85032836,
        8.83725081,  5.10446653,  4.8763135 ,  5.5748624 , 12.27991991,
       13.09737053,  8.36995953,  5.41348121, 13.32374056, 10.99326577,
        5.47702766,  9.47369628, 10.06023665,  4.74468406, 11.99633819,
        9.43112957, 11.85890664,  6.29034958,  6.54693987,  8.25828409,
        7.54127425,  8.1137452 , 11.05672136,  6.83515201, 10.98076873,
       13.62896593,  9.23071698,  7.39004226, 12.67576792, 11.51811603,
        9.11473391, 10.27716434, 10.61047596, 11.00508681, 13.41713661,
       13.66051884,  6.77777606,  6.80559669,  9.99211663, 11.97238624,
       11.70016217])}
{'P': tensor([[1.3023e-02, 2.6510e-01, 1.9720e-01,  ..., 1.0514e-04, 1.0812e-04,
         1.0355e-02],
        [1.5149e-04, 1.2505e-02, 4.4947e-01,  ..., 1.8691e-04, 1.3116e-04,
         9.3996e-03],
        [1.9111e-04, 3.2469e-04, 2.4253e-02,  ..., 5.1086e-04, 2.6027e-04,
         1.2401e-02],
        ...,
        [9.1684e-06, 9.7708e-06, 1.0358e-05,  ..., 2.2057e-01, 3.4488e-06,
         9.2381e-06],
        [7.1844e-06, 1.5509e-05, 2.3380e-05,  ..., 4.0231e-05, 1.7668e-02,
         1.2461e-04],
        [3.0933e-04, 1.0315e-03, 2.4134e-03,  ..., 6.7644e-04, 2.2130e-03,
         3.6862e-02]]), 'X': tensor([[ -8.7409,  -4.6837,  -6.5840],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -7.9925,  -6.1423,   2.8990],
        [  8.3791,  10.5409,   5.5590],
        [  4.3523, -10.5232,   8.2138],
        [  2.5620,   7.0549,   8.4825],
        [ 12.5789,   0.1722,   1.6470],
        [ -8.9033,   2.2329,  -8.8912],
        [ -7.5720,   4.6310,   2.1717],
        [  1.7232,  -4.7574,   3.1270],
        [ -9.2707,   0.9661,   4.4863],
        [  6.2576,   2.1359,  -3.3398],
        [ -4.6393,   7.0461,   1.1624],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.0449,  13.8402,  -7.1642],
        [  5.5424,  -2.0885,   7.2487],
        [  2.5829,  -4.5466,  -8.4015],
        [  3.3069,   6.6089,   8.7377],
        [ -8.9942,  -3.9990,   6.0448],
        [ 11.7418,   5.6070,  -1.4192],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.3587,   1.1596,  -3.6629],
        [  6.7792,  11.8084,   1.6251],
        [  2.8994,   6.8675,  -9.5987],
        [ -1.6906,   7.8860,   7.7142],
        [ -7.3446,   4.9113,  -5.9356],
        [ -2.6735, -10.4062,  -1.5911],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.3246,   1.6696,  -0.1544],
        [ 10.7566,   7.6212,  -2.3279],
        [  9.1409,  -7.5789,   1.5814],
        [ -0.9715, -10.8630,   0.3555],
        [  0.2435,   7.8191,  -6.4709],
        [  5.2608,  -2.5256,  -9.1446],
        [  2.8474,   6.8979,   6.6457],
        [  5.4167, -10.1293,   6.2156],
        [ -9.4117,   0.1160,   8.2574],
        [ -0.4341, -10.9473,   7.4509],
        [  5.8979,   3.3958,  -3.8713],
        [  0.2480, -11.0153,  -3.0183],
        [ -2.0721,   7.8432,   6.6380],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.1755,   2.5233,  -2.0042],
        [  1.9204,   7.3518,   8.4781],
        [  2.2182,  13.8007,   5.1889],
        [ 12.3759,  -1.3943,   6.4659],
        [  3.9591, -10.6391,  -6.2557],
        [ -4.6942,  -9.4299,  -9.1846],
        [  1.6408,   7.4591,  -4.2294],
        [  0.9659,  -4.8144,  -0.9594],
        [  3.8182,  -3.9347,  -1.7961],
        [  0.4996,   7.7738,  -4.4711],
        [  5.9026,   3.3837,  -0.6573],
        [  1.5653,  13.9377,   4.1773],
        [ -8.3717,  -5.4769,  -9.2228],
        [  5.4242, -10.1261,   3.9595],
        [  7.1441,  11.5529,   2.3232],
        [  5.2939,  -2.4784,  -4.5840],
        [  0.9809,  -4.8145,   6.8706],
        [  6.3200,   0.2690,   1.5900],
        [  2.9260,  -4.4162,   8.3808],
        [  2.5128, -10.9365,   9.9645],
        [ -2.9497, -10.3028,  -7.3479],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.5240,  -2.1198,   6.2055],
        [ -5.6423,   6.4537,  -1.2091],
        [  0.8572,  14.0501,  -8.3364],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.1047, -11.0046,   0.5188],
        [ -3.3116, -10.1539,   7.5919],
        [  2.3373,  -4.6232,  -5.2246],
        [ -8.9178,   2.1936,   7.5388],
        [  3.4401,  13.4540,   4.6576],
        [  1.8719,   7.3713,  -3.7465],
        [ -8.8154,   2.4611,   2.5198],
        [ -5.0132,  -9.2234,  -8.3531],
        [  4.0078,  -3.8018,  -6.2168],
        [ -2.0721,   7.8432,   6.6380],
        [ -6.3066,  -8.1936,   9.5822],
        [ 11.4235,   6.3594,   5.1663],
        [  9.5203,   9.3483,   5.8781],
        [  5.6258,  -1.9404,  -6.0715],
        [ -2.5683, -10.4433,  -2.9182],
        [ 11.1420,   6.9331,  -2.4779],
        [ -2.0721,   7.8432,   6.6380],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -4.2040,   7.2486,   2.9875],
        [  5.1142,  -2.7230,  -4.1966],
        [ -1.3940, -10.7772,   6.7835],
        [  5.7784,  -1.6401,   9.5587],
        [  6.2038,   2.4020,  -3.1692],
        [ 11.1420,   6.9331,  -2.4779],
        [  9.9756,  -6.6330,  -3.3429],
        [  2.2373,  -4.6506,  -1.7334],
        [  9.1820,   9.7334,  -2.7932],
        [ -5.5525,   6.5146,  -4.4984],
        [ 12.5650,   2.0132,   4.6272],
        [  6.0310,  12.2795,  -6.4528],
        [ 11.9122,   5.1397,  -1.5498],
        [  8.6271,  -8.0715,  -7.4813],
        [ -5.3328,  -8.9992,   7.3440],
        [ -7.0253,  -7.4493,  -1.5978]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.6963e-02, 4.2474e-02,  ..., 8.0950e-01, 1.0222e+00,
         3.1104e-01],
        [2.6963e-02, 0.0000e+00, 2.1803e-02,  ..., 8.1991e-01, 9.9758e-01,
         2.8694e-01],
        [4.2473e-02, 2.1803e-02, 0.0000e+00,  ..., 8.0740e-01, 9.7979e-01,
         2.6857e-01],
        ...,
        [8.0950e-01, 8.1991e-01, 8.0740e-01,  ..., 0.0000e+00, 1.2839e+00,
         8.4844e-01],
        [1.0222e+00, 9.9758e-01, 9.7979e-01,  ..., 1.2839e+00, 3.0339e-04,
         7.1142e-01],
        [3.1104e-01, 2.8694e-01, 2.6857e-01,  ..., 8.4844e-01, 7.1142e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.91668182, 10.06437598, 10.14436912, 10.30972193, 10.35485331,
       10.09979747, 10.08001894, 13.46553623, 11.38774149,  7.50564223,
       12.58006038,  9.17905491,  8.87587457,  5.05989547,  9.32094055,
        6.61211224,  8.43626114,  4.92497514, 13.99048094,  5.92281142,
        5.22903117,  7.39004226,  9.84314712, 13.01187484,  4.8763135 ,
        6.46356806, 13.61600536,  7.4544866 ,  8.06517143,  8.8353674 ,
       10.74409834, 10.26327895,  6.54127431, 13.18278606, 11.87412439,
       10.90638066,  7.822854  ,  5.83560009,  7.46248947, 11.48663767,
        9.41244865, 10.95593729,  6.80559669, 11.018084  ,  8.11227668,
       12.27991991,  6.67108662,  7.5984809 , 13.97779839, 12.45418236,
       11.35182589, 10.53368644,  7.6374603 ,  4.91038884,  5.48276038,
        7.78980299,  6.80370254, 14.02532758, 10.00410625, 11.48734978,
       13.58333857,  5.84533345,  4.91336994,  6.32572323,  5.29755487,
       11.22142195, 10.7167285 ,  9.48337063,  5.91677188,  8.5724042 ,
       14.07623266,  5.19986797, 11.00508681, 10.68031351,  5.18047638,
        9.18358562, 13.88683436,  7.60529892,  9.15253344, 10.4977132 ,
        5.52416368,  8.11227668, 10.33958639, 13.0743399 , 13.3426541 ,
        5.9510489 , 10.75443106, 13.12299145,  8.11227668, 10.54658597,
        8.37954853,  5.79391825, 10.86693708,  6.00663314,  6.65259775,
       13.12299145, 11.97957425,  5.16077764, 13.38091121,  8.55982105,
       12.72524615, 13.68063098, 12.97370371, 11.81423161, 10.4606265 ,
       10.23946247])}
{'P': tensor([[1.3571e-02, 2.3477e-01, 1.7924e-01,  ..., 4.7041e-05, 1.1397e-07,
         6.8822e-07],
        [4.2719e-04, 1.6238e-02, 1.9302e-01,  ..., 6.9798e-05, 1.3439e-07,
         6.0384e-07],
        [2.7534e-05, 4.5857e-05, 7.0714e-03,  ..., 1.1564e-04, 1.6594e-07,
         2.2573e-07],
        ...,
        [1.5372e-06, 1.4905e-06, 1.9048e-06,  ..., 4.9574e-02, 5.0271e-04,
         3.8562e-06],
        [2.5661e-07, 1.9800e-07, 2.0480e-07,  ..., 4.4864e-06, 9.4385e-02,
         2.3078e-04],
        [4.0301e-04, 2.7719e-04, 2.6384e-04,  ..., 8.0403e-06, 8.7847e-06,
         2.1054e-01]]), 'X': tensor([[ -8.6681,  -4.8556,   9.2518],
        [ -8.3535,  -5.5118,   9.8399],
        [ -6.7745,  -7.7264,   8.4684],
        [ -6.4844,  -8.0230,   8.5209],
        [ -6.3066,  -8.1936,   9.5822],
        [ -6.1914,  -8.2997,   8.5380],
        [  6.0955,   2.8210,  -4.3321],
        [  2.6614,  -4.5192,  -9.3641],
        [ -2.2188,   7.8218,   8.6064],
        [ 12.5199,  -0.4711,   3.0436],
        [  3.2159, -10.8166,  -1.0406],
        [ -9.1624,  -3.4286,   2.0647],
        [  6.3301,   0.3638,  -7.5685],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.7916,  -3.9524,  -7.9547],
        [ -7.8359,   4.2735,   7.2943],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -0.5709,   7.9055,   7.6089],
        [  7.5930,  -8.9037,   7.2323],
        [  5.3860,  -2.3418,   1.0826],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -1.6220,   7.8917,   0.0315],
        [ -2.6485, -10.4151,   0.8766],
        [ 11.3800,  -4.3919,   1.4066],
        [  6.3583,   1.1729,   8.7835],
        [  2.8474,   6.8979,   6.6457],
        [  5.5424,  -2.0885,   7.2487],
        [  7.1821,  -9.1845,   9.8548],
        [  0.3724,   7.7974,  -6.8722],
        [  6.0073,   3.0986,   6.7937],
        [  2.3261,  13.7749,   5.9121],
        [  7.3592,  11.3937,   6.2407],
        [ -4.4520,   7.1370,  -8.7867],
        [ -9.4203,  -2.0340,   2.5856],
        [  9.3836,   9.5075,  -5.9278],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.3569,   0.7591,   6.5011],
        [  6.6319,  11.9066,   3.5165],
        [  0.1287, -11.0065,   6.5757],
        [ 11.6019,   5.9540,  -5.1715],
        [ -1.2454,   7.9127,  -0.5715],
        [  3.3943,   6.5483,   1.0790],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.2419,  -0.2313,   4.2440],
        [  2.1379,  -4.6757,   6.4981],
        [  7.0664,  -9.2590,   8.0879],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.9942,  -3.9990,   6.0448],
        [  6.3303,   1.6145,  -8.4083],
        [ -5.6174,   6.4708,  -1.7339],
        [  1.7122,  -4.7591,  -3.6734],
        [ -6.3489,   5.9117,   6.5161],
        [ -8.0611,  -6.0302,   4.6962],
        [  7.4132,  11.3526,   0.3322],
        [ -8.1666,   3.7653,   3.6075],
        [ -4.7334,   6.9982,  -0.4392],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -1.9714,   7.8563,  -1.8084],
        [ -4.2040,   7.2486,   2.9875],
        [  6.3569,   0.7591,   6.5011],
        [ -1.7009,   7.8851,  -8.2437],
        [  3.9895,   6.0816,  -6.5235],
        [ -5.3179,  -9.0101,   9.4472],
        [  1.7903,  -4.7467,   0.1168],
        [ 12.2311,  -2.0445,   1.3545],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.3587,   1.1596,  -3.6629],
        [  6.1730,  -0.5375,  -8.8861],
        [  9.0652,   9.8597,  -3.5949],
        [  6.6319,  11.9066,   3.5165],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4630,  -1.5145,   0.2983],
        [  6.3535,   0.6792,  -7.5101],
        [ -9.3444,   0.5821,   6.9101],
        [ -3.4986,   7.5152,   9.0712],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.2026,  12.1774,  -1.5887],
        [  9.6614,   9.1784,  -1.2087],
        [ -7.0948,   5.1940,   9.8171],
        [  0.3038,   7.8092,   0.4164],
        [  2.4572,   7.1085,   7.7376],
        [  2.3261,  13.7749,   5.9121],
        [  8.7715,  -7.9390,   5.2900],
        [ -0.7599, -10.8995,  -6.0366],
        [  0.9445,   7.6736,   0.5095],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.7685, -10.8982,  -6.8928],
        [  0.1532,  -4.7422,  -7.0145],
        [ 12.5789,   0.1722,   1.6470],
        [ -7.5603,   4.6459,   8.2002],
        [  4.3459,  -3.5327,   0.3667],
        [ -8.6050,   2.9451,  -0.7012],
        [  9.1620,  -7.5573,  -9.9882],
        [ -8.3900,   3.3727,   3.1650],
        [  0.6470, -11.0352,  -1.1931],
        [ -5.6792,  -8.7352,   4.0318],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.3236,   0.3011,  -4.6668],
        [ -6.5295,  -7.9785,  -7.4429],
        [  9.8082,  -6.8396,   4.9615],
        [ 12.5154,   2.5072,  -9.3274],
        [ -5.9987,   6.1949,  -6.2114]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0144, 0.0038,  ..., 0.7511, 1.6783, 1.0659],
        [0.0144, 0.0000, 0.0128,  ..., 0.7510, 1.6871, 1.0785],
        [0.0038, 0.0128, 0.0000,  ..., 0.7479, 1.6770, 1.0661],
        ...,
        [0.7511, 0.7510, 0.7479,  ..., 0.0000, 1.1862, 0.9784],
        [1.6783, 1.6871, 1.6770,  ..., 1.1862, 0.0000, 0.8253],
        [1.0659, 1.0785, 1.0661,  ..., 0.9784, 0.8253, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.9353921 , 10.00802627, 10.27572673, 10.31584102, 10.33958639,
       10.35464916,  6.7166297 ,  5.24461068,  8.1303945 , 12.52875793,
       11.2845608 ,  9.78285424,  6.34059021,  9.9353921 ,  5.47702766,
        8.92549752,  4.78544875,  9.69792513,  7.92607342, 11.70168755,
        5.87305005, 10.39942497,  8.05668609, 10.74655346, 12.19805394,
        6.46560181,  7.46248947,  5.92281142, 11.65923396,  7.80625981,
        6.7593917 , 13.96987806, 13.56367857,  8.41171369,  9.63742669,
       13.35832725,  8.05242591,  6.4020404 , 13.62896593, 11.00726483,
       13.04051744,  8.01009083,  7.37576154, 12.12448912,  6.24614905,
        5.14124612, 11.64746882, 12.4013153 ,  9.85384623,  9.84314712,
        6.53290663,  8.56889738,  5.05774295,  8.67502348, 10.06703358,
       13.55868684,  8.99276564,  8.4486771 ,  9.48969978,  8.09984231,
        8.37954853,  6.4020404 ,  8.06643487,  7.27340834, 10.46238054,
        5.07306274, 12.40074901, 10.54658597, 12.46513646,  6.46356806,
        6.19633139, 13.39372426, 13.62896593, 13.25336001,  9.58347456,
        6.38968357,  9.36256096,  8.28968067,  9.43737523,  5.18277858,
       13.6660671 , 13.32613445,  8.79284466,  7.81509631,  7.52117132,
       13.96987806, 11.83075005, 10.92597196,  7.73151309, 11.67875855,
       11.24434374,  4.74468406, 12.58006038,  8.8737556 ,  5.60064319,
        9.09502385, 11.87665501,  9.04255655, 11.05413545, 10.41909211,
        4.78258092,  6.33075682, 10.30972193, 11.95743558, 12.76408562,
        8.62329466])}
{'P': tensor([[7.1832e-03, 3.4501e-01, 2.7058e-01,  ..., 6.7539e-07, 1.7600e-07,
         3.3797e-06],
        [1.6647e-04, 1.4269e-02, 1.0976e-03,  ..., 1.3101e-06, 3.9386e-07,
         7.2711e-06],
        [2.0975e-04, 1.4215e-01, 1.3103e-02,  ..., 1.1652e-06, 3.4247e-07,
         6.3606e-06],
        ...,
        [5.6635e-05, 4.6300e-05, 4.5841e-05,  ..., 5.4567e-02, 4.1914e-05,
         2.8669e-04],
        [3.9432e-07, 3.2957e-07, 3.3000e-07,  ..., 6.6662e-03, 1.5611e-02,
         6.5520e-04],
        [3.3488e-05, 2.9372e-05, 2.9131e-05,  ..., 6.2269e-02, 6.2089e-03,
         8.2230e-02]]), 'X': tensor([[-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.0967e-02, 2.0831e-02,  ..., 1.0914e+00, 1.4374e+00,
         1.1374e+00],
        [2.0967e-02, 0.0000e+00, 6.0679e-04,  ..., 1.0730e+00, 1.4177e+00,
         1.1173e+00],
        [2.0831e-02, 6.0679e-04, 0.0000e+00,  ..., 1.0734e+00, 1.4181e+00,
         1.1176e+00],
        ...,
        [1.0914e+00, 1.0730e+00, 1.0734e+00,  ..., 0.0000e+00, 3.9312e-01,
         2.3617e-01],
        [1.4374e+00, 1.4177e+00, 1.4181e+00,  ..., 3.9312e-01, 0.0000e+00,
         3.0946e-01],
        [1.1374e+00, 1.1173e+00, 1.1176e+00,  ..., 2.3617e-01, 3.0946e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.93553424, 10.00410625,  9.99145096, 10.27716434, 10.37698435,
       10.35485331,  9.78285424, 12.19805394, 13.0493368 , 13.04217487,
       13.00413829,  9.88310614, 12.17901033,  7.32754882, 11.22993218,
        7.5092834 ,  6.41202986, 12.1494023 , 13.3614598 ,  4.9160962 ,
       13.03002338,  7.5092834 , 13.26209545, 13.26590071,  9.47369628,
        5.21280156,  6.65368551, 10.04916678,  9.64157461, 12.91143491,
        9.96152788, 10.78533838, 11.75807252, 14.10045797, 10.60226614,
       10.0674887 ,  9.44486609,  7.81509631,  9.61942445, 13.99048094,
        5.14607014, 13.2778378 , 10.46739073, 12.42554808,  9.15866607,
       12.53047489,  5.92952538, 11.9120321 ,  9.26455795, 12.18045978,
       13.85158417,  7.20826219,  5.16077764, 12.91143491,  9.78147542,
        6.4020404 ,  9.15253344,  7.44805901, 13.20988996, 12.70489563,
        5.0713364 ,  9.97964013,  6.83515201, 10.4977132 ,  5.77219077,
       11.73567759,  7.72415048, 13.26590071, 13.20988996, 11.76830981,
       11.99633819, 13.50484539,  5.08910491,  9.32094055, 13.56367857,
       10.67322527,  9.69792513,  7.73692436,  7.96575782, 12.4217523 ,
        9.97822473,  7.4544866 ,  8.31625171,  7.71075226, 10.48840476,
        5.92281142,  7.16747878,  9.09119351,  7.06372172,  5.07306274,
       10.07238463, 11.56146873, 14.09176259,  9.60359158,  7.6374603 ,
       11.22993218,  7.03733775, 12.22276278,  5.16205087, 10.35485331,
        9.68323686, 11.50791859,  6.42439344,  7.60463349, 13.08414689,
        6.74686071])}
{'P': tensor([[7.4963e-03, 2.2165e-01, 9.2017e-02,  ..., 1.4356e-06, 1.5933e-06,
         8.7874e-07],
        [2.1436e-04, 1.1902e-02, 1.4148e-01,  ..., 2.2594e-06, 2.4134e-06,
         1.3676e-06],
        [1.7896e-05, 2.7744e-05, 4.2782e-03,  ..., 1.8025e-06, 1.0211e-06,
         1.0758e-06],
        ...,
        [5.0044e-06, 4.2184e-06, 4.5010e-06,  ..., 5.0034e-02, 2.1827e-03,
         6.0369e-02],
        [1.8157e-04, 1.3952e-04, 7.1990e-05,  ..., 8.9493e-05, 5.3371e-02,
         7.2632e-05],
        [6.2392e-06, 5.2160e-06, 5.6148e-06,  ..., 2.1738e-02, 2.2335e-03,
         8.5935e-02]]), 'X': tensor([[ -8.6426,  -4.9136,   9.4941],
        [ -8.3535,  -5.5118,   9.8399],
        [ -6.7745,  -7.7264,   8.4684],
        [ -6.3066,  -8.1936,   9.5822],
        [ -6.4844,  -8.0230,   8.5209],
        [ -6.1914,  -8.2997,   8.5380],
        [  6.1973,  -0.4372,   6.5720],
        [  8.3791,  10.5409,   5.5590],
        [  6.1752,   2.5242,   3.5874],
        [  3.6469,   6.3624,  -1.8414],
        [ -6.6895,   5.6051,   2.7899],
        [  2.9260,  -4.4162,   8.3808],
        [  6.3516,   0.6425,   0.3425],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -0.5385,   7.9037,   6.3830],
        [ -6.2668,   5.9808,   2.5614],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -8.6801,   2.7809,  -0.8443],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.8719,   7.3713,  -3.7465],
        [ -9.4731,  -0.7120,  -6.0271],
        [  4.4039,   5.6903,   8.7366],
        [  4.7156,  -3.1831,  -8.0224],
        [  1.3103,  -4.8032,  -0.9300],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.3301,   0.3638,  -7.5685],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.7813,  -3.1139,   8.2691],
        [ -9.4541,  -0.3327,   8.8246],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.0617,  12.7957,   8.3544],
        [  2.0449,  13.8402,  -7.1642],
        [  6.1579,  -0.5961,  -6.5595],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.5340,  -3.3629,   8.9652],
        [ -2.7572,   7.7194,   5.9580],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -5.3328,  -8.9992,   7.3440],
        [ 12.5650,   2.0132,   4.6272],
        [ -6.0195,   6.1789,  -2.9496],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -4.7389,   6.9953,   3.7638],
        [  2.7685, -10.8982,  -6.8928],
        [  7.7750,  11.0665,   2.8364],
        [  5.4336,  -2.2676,   5.6933],
        [  5.8979,   3.3958,  -3.8713],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.1749,  -3.3803,   6.4629],
        [ -1.5408,   7.8978,   5.3470],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.2555,   2.1474,   4.1685],
        [  3.6249, -10.7255,   5.6605],
        [ -8.5900,   2.9767,   7.4811],
        [  2.0937,  13.8293,  -2.3533],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.4203,  -2.0340,   2.5856],
        [ -8.5859,  -5.0393,   0.5476],
        [ -6.4844,  -8.0230,   8.5209],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.1755,   2.5233,  -2.0042],
        [ 12.2592,   3.9425,   2.0383],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.3303,   1.6145,  -8.4083],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.4721,  -1.3097,  -7.4377],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.4837,   2.7510,   0.4449],
        [  6.9653,  -9.3225,  -3.4727],
        [ 12.3357,   3.6001,   0.7903],
        [ -1.2454,   7.9127,  -0.5715],
        [ -0.6323,   7.9085,  -5.9544],
        [  0.7958,  -4.8109,  -8.2248],
        [  1.8719,   7.3713,  -3.7465],
        [  1.5662,  13.9375,   4.5142],
        [ -1.5261,   7.8988,   9.3768],
        [  0.5250, -11.0306,   5.5358],
        [  6.3205,   1.7068,  -3.8619],
        [ 10.1798,   8.5002,  -5.8893],
        [  5.5424,  -2.0885,   7.2487],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.2975,   0.0947,   5.1791],
        [  5.1472,  -2.6801,   8.1194],
        [ 10.4758,  -5.9539,   1.3667],
        [ -3.3999, -10.1152,  -1.2088],
        [ 12.2921,  -1.7902,  -4.0870],
        [  1.6308,  13.9254,  -2.0984],
        [  7.7794,  11.0629,   8.9069],
        [  5.2188,  -2.5839,   2.1833],
        [  5.6093,  -1.9705,   6.4807],
        [ -3.7964,  -9.9294,   7.7147],
        [  2.3367,   7.1676,  -9.5263],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.1741,  -0.5329,   3.1913],
        [ 10.1798,   8.5002,  -5.8893],
        [  5.0233,   4.9647,  -3.1788],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.2914,   3.8037,   6.0978],
        [ -7.2515,   5.0195,   4.6434],
        [ -8.3535,  -5.5118,   9.8399],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.3205,   1.7068,  -3.8619],
        [  1.7659,   7.4127,   4.7871],
        [  6.2630,   2.1057,  -5.2136]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.0358e-03, 5.1844e-03,  ..., 1.2666e+00, 7.6510e-01,
         1.3039e+00],
        [9.0384e-03, 0.0000e+00, 1.2777e-02,  ..., 1.2711e+00, 7.6567e-01,
         1.3086e+00],
        [5.1844e-03, 1.2775e-02, 0.0000e+00,  ..., 1.2614e+00, 7.6074e-01,
         1.2987e+00],
        ...,
        [1.2666e+00, 1.2711e+00, 1.2614e+00,  ..., 0.0000e+00, 6.7115e-01,
         5.6202e-02],
        [7.6510e-01, 7.6567e-01, 7.6074e-01,  ..., 6.7115e-01, 1.5170e-04,
         7.2447e-01],
        [1.3039e+00, 1.3086e+00, 1.2987e+00,  ..., 5.6202e-02, 7.2447e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.94173708, 10.00802627, 10.27572673, 10.33958639, 10.31584102,
       10.35464916,  6.21275016, 13.46553623,  6.67121748,  7.33350642,
        8.7273569 ,  5.29755487,  6.38399789, 10.53368644,  7.92200342,
        8.66273171, 12.49291768,  9.11473391, 10.35093485,  7.60529892,
        9.4997987 ,  7.19534146,  5.68942572,  4.978704  , 12.64235359,
        6.34059021, 12.9089559 ,  5.70591986,  9.45995955, 13.00413829,
       13.76047999, 13.99048094,  6.18668463,  8.04483495,  9.54154096,
        5.64500512,  8.19703512, 10.35485331, 10.4606265 , 12.72524615,
        8.62631801, 12.05300943,  8.44940661, 11.24434374, 13.5247215 ,
        5.88782913,  6.80559669, 10.26327895,  6.21275016,  9.77777346,
        8.04664886, 10.412299  ,  6.61386436, 11.32149259,  9.09119351,
       13.98691426,  9.60412735,  9.63742669,  9.95552626, 10.31584102,
        9.37989631,  6.67108662, 12.87752052,  8.12580976,  6.53290663,
        5.43943791, 12.4217523 ,  9.562174  ,  4.92497514, 12.78327037,
       11.63724792, 12.85032836,  8.01009083,  7.9337647 ,  4.8763135 ,
        7.60529892, 14.02526505,  8.04483495, 11.04313641,  6.54693987,
       13.26209545,  5.92281142, 12.22503754,  6.29821673,  5.80312949,
       12.0495521 , 10.67132524, 12.4217523 , 14.02058621, 13.52430176,
        5.82345867,  5.94535529, 10.63038565,  7.53884104, 12.24467041,
        6.19707944, 13.26209545,  7.06272256, 11.81423161, 12.86648771,
        8.81930525, 10.00802627,  9.09661212,  6.54693987,  7.62011734,
        6.60752373])}
{'P': tensor([[8.6876e-03, 2.6257e-01, 1.2904e-01,  ..., 1.2508e-04, 2.3082e-05,
         8.9168e-06],
        [2.2216e-04, 2.6133e-02, 1.5462e-01,  ..., 1.1921e-04, 3.1479e-05,
         3.5010e-05],
        [4.6792e-04, 3.8837e-03, 4.8670e-02,  ..., 1.7018e-04, 4.0726e-05,
         7.2947e-05],
        ...,
        [1.7249e-01, 1.8203e-01, 1.7850e-01,  ..., 1.4193e-02, 5.7258e-05,
         1.2330e-05],
        [1.8575e-02, 1.8055e-02, 1.2374e-02,  ..., 7.1646e-03, 3.2937e-01,
         4.7613e-05],
        [4.5567e-05, 3.5074e-05, 3.3617e-05,  ..., 6.8319e-05, 2.0509e-05,
         6.3479e-02]]), 'X': tensor([[ -8.6381,  -4.9238,  -4.4077],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.0766,  -6.0046,  -3.1873],
        [  8.1528,  10.7452,   9.1493],
        [  9.5203,   9.3483,   5.8781],
        [  4.9152,  12.8653,  -1.4459],
        [ -4.2271,   7.2387,  -1.8316],
        [  5.3860,  -2.3418,   1.0826],
        [ -0.1626, -10.9796,   6.4629],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.5954,  13.0099,  -4.4439],
        [  5.9655,   3.2172,  -7.5953],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.9314,  -9.3435,  -1.2729],
        [  8.8474,  10.0868,  -8.8644],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.5996,   1.3971,  -1.0694],
        [  6.7792,  11.8084,   1.6251],
        [  7.0642,  11.6104,  -9.8316],
        [  1.6954,  -4.7616,   9.8198],
        [  1.6308,  13.9254,  -2.0984],
        [ -8.1666,   3.7653,   3.6075],
        [  9.0004,  -7.7195,   4.6945],
        [  5.5240,  -2.1198,   6.2055],
        [ -3.4100,   7.5436,  -7.6980],
        [  4.6063,  13.0051,  -3.1456],
        [ -0.4341, -10.9473,   7.4509],
        [  2.5010,  -4.5736,  -0.8989],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.6248,   3.9984,   8.4933],
        [  5.0348,  -2.8229,   3.5436],
        [  0.6396,  14.0772,  -6.3786],
        [  8.7131,  10.2216,   7.7915],
        [ -0.0254, -10.9932,   3.2200],
        [  2.2860,   7.1916,  -6.1011],
        [  3.5965,  -4.0754,  -3.5035],
        [  5.3858,   4.4260,  -9.6769],
        [ -8.8058,   2.4850,  -1.7425],
        [ -6.1914,  -8.2997,   8.5380],
        [  5.1861,   4.7361,  -3.7312],
        [  3.3633,  13.4794,   0.5193],
        [ -8.3900,   3.3727,   3.1650],
        [ 12.2692,  -1.8882,   3.9253],
        [  3.9770, -10.6341,   8.0783],
        [  5.9003,  12.3550,   2.1917],
        [  5.8617,   3.4853,  -9.8080],
        [ -6.4464,   5.8272,   7.5038],
        [ 12.5832,   1.7543,  -7.6134],
        [  6.1929,   2.4501,  -6.9352],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.3200,   7.1756,   1.1938],
        [ 11.9936,  -2.8677,   6.0614],
        [ -7.6734,   4.4980,  -5.4248],
        [ -1.5451,   7.8975,   7.6104],
        [  2.5128, -10.9365,   9.9645],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 12.4877,  -0.7222,   8.8347],
        [  4.5767,   5.5071,   3.9069],
        [  2.3261,  13.7749,   5.9121],
        [  6.3460,   1.4252,   1.2291],
        [ -2.2188,   7.8218,   8.6064],
        [  1.7000,  -4.7609,   2.7935],
        [  5.1667,  -2.6543,   1.2879],
        [  1.1350,  14.0104,   0.6321],
        [ -1.7009,   7.8851,  -8.2437],
        [ -8.6229,   2.9068,   2.4429],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.9497, -10.3028,  -7.3479],
        [  6.1752,   2.5242,   3.5874],
        [  2.0899,  -4.6870,  -7.4260],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.9945,  -4.8144,  -0.1897],
        [ -4.4503,   7.1378,   3.3751],
        [  3.2226,  -4.2802,  -4.0428],
        [ -7.3554,   4.8985,  -0.6665],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -7.8849,  -6.3115,  -4.7479],
        [  0.6470, -11.0352,  -1.1931],
        [ -1.5408,   7.8978,   5.3470],
        [  2.8474,   6.8979,   6.6457],
        [  2.8994,   6.8675,  -9.5987],
        [  6.0955,   2.8210,  -4.3321],
        [  2.8994,   6.8675,  -9.5987],
        [  6.1737,   2.5305,   3.9049],
        [  4.7482, -10.3907,   5.9149],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.1575,  12.2046,   3.3257],
        [ 10.2470,   8.4054,  -0.1177],
        [  3.3633,  13.4794,   0.5193],
        [ -3.9700,   7.3453,  -4.1486],
        [ -8.4297,  -5.3637,  -9.3962],
        [  7.1441,  11.5529,   2.3232],
        [  3.4920,  -4.1366,  -8.3955],
        [  8.4691,  10.4571,  -5.0650],
        [ -8.7735,   2.5640,  -6.3313],
        [ 12.6038,   0.7665,  -3.4439],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.1372,   2.6726,  -6.7885]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.0404e-02, 6.0234e-02,  ..., 6.4059e-02, 7.4251e-01,
         1.0021e+00],
        [2.0404e-02, 0.0000e+00, 4.1038e-02,  ..., 4.7237e-02, 7.5922e-01,
         9.9293e-01],
        [6.0234e-02, 4.1038e-02, 0.0000e+00,  ..., 1.4264e-02, 7.9962e-01,
         9.9238e-01],
        ...,
        [6.4059e-02, 4.7237e-02, 1.4264e-02,  ..., 0.0000e+00, 8.0618e-01,
         1.0061e+00],
        [7.4251e-01, 7.5922e-01, 7.9962e-01,  ..., 8.0618e-01, 3.0339e-04,
         1.1581e+00],
        [1.0021e+00, 9.9293e-01, 9.9238e-01,  ..., 1.0061e+00, 1.1581e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.94285511, 10.09979747, 10.14436912, 10.31613269, 10.1979669 ,
       10.06407839, 13.48809595, 13.3426541 , 13.77222752,  8.38252937,
        5.87305005, 10.98076873, 11.66453059, 13.79762415,  6.77777606,
       10.35485331,  5.26067019, 11.63383221, 13.41713661,  5.64464807,
       12.67680439, 13.61600536, 13.59056613,  5.05444288, 14.02058621,
        8.99276564, 11.85743425,  5.91677188,  8.27853076, 13.79676654,
       10.95593729,  5.21280156, 12.16901615,  6.90115901,  5.77219077,
       14.09176259, 13.43127131, 10.99326577,  7.54620849,  5.43544626,
        6.97106912,  9.14973319, 10.35464916,  7.02326515, 13.89264822,
        9.04255655, 12.41366754, 11.35345345, 13.69162795,  6.81961996,
        8.68979052, 12.70489563,  6.65991877, 12.4013153 ,  9.48969978,
        7.54127425, 12.33167382,  8.89458273,  8.04718433, 11.22142195,
        9.43737523, 12.5085995 ,  7.16058587, 13.96987806,  6.50409372,
        8.1303945 ,  5.05534609,  5.80860165, 14.05633441,  8.06643487,
        9.09963894, 11.785566  ,  9.34127013, 10.7167285 ,  6.67121748,
        5.13182506,  4.91038884,  4.9160962 ,  8.41148722,  5.35774573,
        8.83725081,  7.27340834,  9.71392179, 10.09979747, 11.05413545,
        8.04664886,  7.46248947,  7.4544866 ,  6.7166297 ,  7.4544866 ,
        6.67218687, 11.42420508,  7.8575648 , 13.66991027, 13.25336001,
       13.89264822,  8.34949383,  9.99145096, 13.58333857,  5.41348121,
       13.45642379,  9.14044984, 12.62711252,  9.7712321 ,  9.53015376,
        6.69389431])}
{'P': tensor([[6.8889e-03, 2.6240e-01, 1.2414e-01,  ..., 1.4329e-06, 8.6606e-03,
         2.2826e-04],
        [1.6057e-04, 9.8494e-03, 2.0130e-01,  ..., 2.0659e-06, 1.3079e-02,
         3.8270e-04],
        [4.3144e-05, 5.6431e-05, 9.4215e-03,  ..., 1.8894e-06, 2.2340e-03,
         5.1070e-04],
        ...,
        [1.3811e-04, 1.1988e-04, 5.2542e-05,  ..., 4.0925e-02, 6.0862e-05,
         9.8332e-05],
        [6.8518e-04, 9.7529e-04, 7.2536e-03,  ..., 2.9848e-05, 1.0906e-01,
         1.4652e-02],
        [1.7054e-04, 1.8142e-04, 2.1711e-04,  ..., 3.1957e-03, 3.1819e-04,
         1.8366e-01]]), 'X': tensor([[ -8.5859,  -5.0393,   0.5476],
        [ -8.3539,  -5.5111,   0.3373],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.0844,  -8.3955,   0.7705],
        [ -6.8466,  -7.6488,   1.9648],
        [  5.8318,   3.5567,  -4.7692],
        [  9.2979,   9.6048,   7.3138],
        [ -3.3116, -10.1539,   7.5919],
        [ -1.7290,   7.8825,  -3.3831],
        [  5.8902,  -1.3887,   4.0388],
        [ 12.5789,   0.1722,   1.6470],
        [  6.3551,   0.7149,   1.3344],
        [ -8.9636,   2.0648,   8.1946],
        [ -6.9838,  -7.4965,   0.9796],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -0.8885,   7.9160,   0.0571],
        [  5.9526,   3.2525,  -4.3821],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.9526,   3.2525,  -4.3821],
        [  8.0381,  -8.5690,  -8.9523],
        [ -3.9508,  -9.8515,   1.7923],
        [  4.0362,  13.2397,  -7.0030],
        [ 12.4597,  -0.9136,  -4.8763],
        [  9.6662,   9.1725,   8.1056],
        [  3.3167,   6.6022,   6.3554],
        [  0.6363, -11.0348,   6.1414],
        [  8.6420,  -8.0580,   6.5781],
        [  1.7232,  -4.7574,   3.1270],
        [ -2.6069, -10.4298,  -6.1995],
        [ 11.2767,  -4.6006,   0.7868],
        [ -9.2345,  -3.1345,   3.9715],
        [  1.0000,   7.6591,  -4.5011],
        [  0.1287, -11.0065,   6.5757],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.5974,   7.0363,  -7.7810],
        [  0.9445,   7.6736,   0.5095],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 10.9931,   7.2107,   7.7933],
        [  5.5294,  -2.1106,   9.0626],
        [  2.5620,   7.0549,   8.4825],
        [  6.8704,  -9.3809,   7.5769],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -6.8466,  -7.6488,   1.9648],
        [  5.9193,  -1.3176,   5.1739],
        [  4.5340,  -3.3629,   8.9652],
        [ -8.9282,   2.1648,   8.4894],
        [  9.8082,  -6.8396,   4.9615],
        [ -5.6169,  -8.7844,   8.9057],
        [  4.5503,   5.5359,   7.4735],
        [  5.1292,   4.8183,  -5.7997],
        [ -8.6112,   2.9319,  -7.9943],
        [  7.8375,  -8.7240,  -9.8913],
        [ 11.2571,   6.7072,   8.3693],
        [  4.8051, -10.3703,   0.5268],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.3200,   7.1756,   1.1938],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.2218,   4.0953,  -8.2072],
        [  7.0664,  -9.2590,   8.0879],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.1504,   4.3674,  -7.2998],
        [  0.4119,  14.1019,  -4.1870],
        [  5.2608,  12.6977,   4.9166],
        [ -6.3066,  -8.1936,   9.5822],
        [  4.1140,  -3.7220,  -0.1273],
        [ 12.3453,  -1.5462,   7.0212],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.9655,   3.2172,  -7.5953],
        [  0.6933,   7.7336,   2.1179],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -8.0087,   4.0173,   9.1091],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.3305,   1.6119,   1.1538],
        [  0.1677,  -4.7447,  -6.1529],
        [  5.0233,   4.9647,  -3.1788],
        [  5.9883,   3.1536,   6.2320],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.1460,   7.8340,  -9.4614],
        [  6.2555,   2.1474,   4.1685],
        [ -9.0574,   1.7801,   8.8986],
        [  5.0617,  12.7957,   8.3544],
        [ -7.3470,  -7.0615,   5.0066],
        [  7.5064,  -8.9651,  -0.0994],
        [ 12.0845,  -2.5772,  -4.6073],
        [  3.9085,   6.1512,  -2.2560],
        [  3.4604,  -4.1545,   1.8091],
        [  2.5813,  -4.5471,   5.3597],
        [ -2.0229, -10.6163,   8.6930],
        [  2.5974,   7.0363,  -7.7810],
        [ 11.2767,  -4.6006,   0.7868],
        [ -2.7572,   7.7194,   5.9580],
        [  1.5653,  13.9377,   4.1773],
        [ 11.7349,  -3.5838,   2.1644],
        [  1.1139,   7.6281,  -2.5680],
        [ -7.8280,   4.2848,  -7.1955],
        [ -7.0948,   5.1940,   9.8171],
        [ 11.6544,   5.8273,  -3.1375],
        [  3.7121,   6.3117,   0.6738],
        [ -8.8363,   2.4086,  -5.6296],
        [ -4.8751,  -9.3148,   4.0406],
        [  1.9312,  13.8649,   8.3563],
        [  1.6408,   7.4591,  -4.2294],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.7976,  -4.4682,  -4.2020]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0168, 0.0319,  ..., 0.8461, 0.3695, 0.9381],
        [0.0168, 0.0000, 0.0472,  ..., 0.8373, 0.3531, 0.9301],
        [0.0319, 0.0472, 0.0000,  ..., 0.8477, 0.4000, 0.9379],
        ...,
        [0.8461, 0.8373, 0.8477,  ..., 0.0000, 0.7986, 0.1049],
        [0.3695, 0.3531, 0.4000,  ..., 0.7986, 0.0000, 0.9032],
        [0.9381, 0.9301, 0.9379,  ..., 0.1049, 0.9032, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.95552626, 10.00794468, 10.24557047, 10.01804223, 10.36844813,
       10.26545792,  6.83083863, 13.36800073, 10.68031351,  8.06990483,
        6.05165248, 12.58006038,  6.39520397,  9.19837792, 10.24557047,
       12.4980845 ,  7.96575782,  6.78325151, 13.01187484,  6.78325151,
       11.74901778, 10.61417181, 13.84126197, 12.49317563, 13.32557133,
        7.38844703, 11.05317597, 11.8159254 ,  5.05989547, 10.7506423 ,
       12.17901033,  9.75200848,  7.72415048, 11.00726483,  9.51074763,
        7.50035355,  7.73151309, 10.30972193, 13.14690137,  5.91854397,
        7.50564223, 11.62770332, 12.25806559, 10.26545792,  6.06416957,
        5.64500512,  9.18690018, 11.95743558, 10.42667093,  7.16598273,
        7.03733775,  9.09661212, 11.72749923, 13.103709  , 11.42947632,
        8.20436764,  7.54127425,  5.64464807, 12.4013153 , 12.889695  ,
       11.64746882, 12.77402351, 12.91143491, 14.10796466, 13.744387  ,
       10.33958639,  5.54776934, 12.44177641, 10.36844813,  6.77777606,
        7.76457095,  9.86089579,  8.95983586,  5.9510489 ,  6.53250937,
        4.74770899,  7.06272256,  6.76791483, 12.0495521 ,  7.83534987,
        6.61386436,  9.23071698, 13.76047999, 10.19036081, 11.69264243,
       12.35625694,  7.28793663,  5.40689332,  5.2287058 , 10.80728788,
        7.50035355, 12.17901033,  8.19703512, 14.02532758, 12.26997165,
        7.70897675,  8.92395381,  8.79284466, 13.03002338,  7.32233733,
        9.15866607, 10.51340829, 13.99877413,  7.6374603 , 10.31613269,
        5.27178363])}
{'P': tensor([[8.9894e-03, 2.9013e-01, 2.7252e-01,  ..., 1.3960e-03, 1.4772e-03,
         2.9017e-07],
        [1.7403e-04, 1.4170e-02, 5.0934e-01,  ..., 3.7151e-03, 1.5797e-03,
         3.9086e-07],
        [2.1697e-04, 4.1605e-04, 3.1078e-02,  ..., 1.1765e-02, 3.2938e-03,
         8.9334e-07],
        ...,
        [3.0714e-06, 3.7745e-06, 4.4803e-06,  ..., 2.2883e-02, 5.5650e-06,
         8.9893e-07],
        [1.6690e-04, 2.6658e-04, 4.5029e-04,  ..., 4.0362e-03, 5.5086e-02,
         7.2907e-06],
        [2.2967e-04, 1.5744e-04, 1.4271e-04,  ..., 9.6371e-06, 5.4772e-04,
         5.0305e-01]]), 'X': tensor([[ -8.5607,  -5.0938,  -5.4531],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 11.0438,   7.1180,  -5.6991],
        [ 11.6019,   5.9540,  -5.1715],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.1984,  -0.4327,   1.5438],
        [  5.7963,  -1.6018,   1.8856],
        [ 12.3357,   3.6001,   0.7903],
        [  0.1047, -11.0046,   0.5188],
        [  8.7715,  -7.9390,   5.2900],
        [ -5.1398,  -9.1367,   6.2074],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 12.6038,   0.7665,  -3.4439],
        [  1.8721,  -4.7322,   0.5524],
        [ -7.0825,   5.2073,  -5.0497],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.1519, -10.2387,   9.8191],
        [  2.0449,  13.8402,  -7.1642],
        [ -0.0282,   7.8575,  -1.1746],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.8602,   6.8905,   5.8779],
        [  3.4401,  13.4540,   4.6576],
        [ 12.4691,   2.8525,  -5.5110],
        [  2.7590,  13.6621,   9.4496],
        [  4.2041,  -3.6509,  -4.9488],
        [  8.5198,  10.4091,  -8.6486],
        [  1.9312,  13.8649,   8.3563],
        [  5.4426, -10.1181,  -5.1268],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 11.6634,   5.8050,   9.1590],
        [  6.0955,   2.8210,  -4.3321],
        [  5.3118,   4.5452,   4.2114],
        [  5.1203,  -2.7151,   6.2110],
        [  6.0890,  -0.8382,   2.9166],
        [ -7.4807,   4.7463,   2.0616],
        [  6.3031,   1.8477,  -7.1706],
        [ 12.4690,  -0.8528,  -9.3374],
        [  4.1348,   5.9514,  -0.8856],
        [  6.3569,   0.7591,   6.5011],
        [ 11.4235,   6.3594,   5.1663],
        [ -1.8788, -10.6567,   3.7068],
        [ 10.7566,   7.6212,  -2.3279],
        [  3.9022,  13.2906,   5.5788],
        [  2.3373,  -4.6232,  -5.2246],
        [ -4.4520,   7.1370,  -8.7867],
        [  9.9750,  -6.6338,  -1.2440],
        [  0.1460,   7.8340,  -9.4614],
        [  6.3246,   1.6696,  -0.1544],
        [  3.9913,  -3.8139,  -6.8311],
        [ -1.6334, -10.7207,   7.1397],
        [ -8.5149,  -5.1905,   1.5239],
        [ -7.3470,  -7.0615,   5.0066],
        [ -8.6603,   2.8250,   5.9181],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.1196,  -2.7160,   4.2705],
        [ -7.4733,   4.7555,   6.2490],
        [ -3.9700,   7.3453,  -4.1486],
        [ -2.5683, -10.4433,  -2.9182],
        [  0.2435,   7.8191,  -6.4709],
        [  3.2692,  13.5098,  -4.9032],
        [ 10.4994,  -5.9193,  -7.5153],
        [  5.8902,  -1.3887,   4.0388],
        [  7.2340,  -9.1505,  -5.8321],
        [  1.6422,   7.4586,  -6.8056],
        [  4.5057,  -3.3894,  -0.8160],
        [ -8.6801,   2.7809,  -0.8443],
        [  3.2692,  13.5098,  -4.9032],
        [  6.9956,  11.6590,  -7.6876],
        [ 11.0275,  -5.0666,   8.4154],
        [  8.6271,  -8.0715,  -7.4813],
        [  6.3583,   1.1729,   8.7835],
        [  6.5428,  -9.5728,  -2.9897],
        [  9.9217,  -6.7006,  -5.6728],
        [  5.5424,  -2.0885,   7.2487],
        [  5.7963,  -1.6018,   1.8856],
        [  8.5716,  10.3596,   6.8818],
        [  6.3303,   1.6145,  -8.4083],
        [ 12.2592,   3.9425,   2.0383],
        [  2.7121,   6.9743,  -8.4202],
        [  3.4604,  -4.1545,   1.8091],
        [  6.3534,   1.2990,  -7.2942],
        [  5.3602,  -2.3808,   7.2098],
        [  0.9036,   7.6840,   9.9237],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.9606,  -4.0994,   1.7886],
        [ -3.0644,   7.6439,  -1.4322],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.7784,  -1.6401,   9.5587],
        [ 12.5770,   1.8512,  -0.6578],
        [  3.3839,  -4.1966,   8.3574],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.0305,   3.0296,   0.1375],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.8275,  -1.5334,   2.0124],
        [  9.6587,   9.1817,  -0.8212],
        [ -5.2589,   6.7024,  -5.5809],
        [  3.3069,   6.6089,   8.7377],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.0310,  12.2795,  -6.4528],
        [  2.7685, -10.8982,  -6.8928],
        [ -5.9225,  -8.5355,   2.4778],
        [ -0.0410,   7.8591,   9.3457]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0184, 0.0254,  ..., 0.4707, 0.6173, 1.2334],
        [0.0184, 0.0000, 0.0218,  ..., 0.4576, 0.6306, 1.2391],
        [0.0254, 0.0218, 0.0000,  ..., 0.4460, 0.6129, 1.2174],
        ...,
        [0.4707, 0.4576, 0.4460,  ..., 0.0000, 0.7897, 1.1082],
        [0.6173, 0.6306, 0.6129,  ..., 0.7897, 0.0000, 0.7501],
        [1.2334, 1.2391, 1.2174,  ..., 1.1082, 0.7501, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.96152788, 10.06437598, 10.14436912, 10.09979747, 10.31613269,
       10.30972193, 13.13888841, 13.04051744, 12.22503754,  6.2134961 ,
        6.01356004, 12.85032836, 11.00508681, 11.83075005, 10.48315436,
       10.46739073, 12.62711252,  5.08910491,  8.7907923 ,  5.39723241,
       11.46176259, 13.99048094,  7.8575648 , 12.95702647,  7.46053281,
       13.88683436, 12.79126785, 13.93790235,  5.56809335, 13.45124917,
       13.99877413, 11.4890861 , 10.52357479, 13.02818702,  6.7166297 ,
        6.99097013,  5.79562691,  6.1463896 ,  8.85939626,  6.56833864,
       12.4980845 ,  7.24677664,  6.4020404 , 13.0743399 , 10.82106505,
       13.18278606, 13.85158417,  5.18047638,  8.41171369, 11.97949273,
        7.83534987,  6.54127431,  5.5205189 , 10.84437705,  9.97220799,
       10.19036081,  9.10945678, 10.31886066,  5.79543038,  8.8580671 ,
        8.34949383, 10.75443106,  7.822854  , 13.89974693, 12.05300943,
        6.05165248, 11.66453059,  7.63726134,  5.63824971,  9.11473391,
       13.89974693, 13.59673406, 12.13568598, 11.81423161,  6.46560181,
       11.59512194, 11.97238624,  5.92281142,  6.01356004, 13.44593519,
        6.53290663, 12.87752052,  7.48311002,  5.40689332,  6.48486171,
        5.86519525,  7.73692436,  4.9160962 ,  9.85384623,  8.23525433,
        5.01114719,  6.00663314, 12.7125079 ,  5.39098669,  5.63824971,
        6.74872981,  5.94541803,  6.02588517, 13.32645552,  8.51928531,
        7.39004226,  5.70405525, 13.68063098, 11.24434374, 10.38897563,
        7.85919687])}
{'P': tensor([[9.0060e-03, 2.6227e-01, 1.8052e-01,  ..., 2.5391e-08, 3.6450e-08,
         6.3548e-05],
        [1.4686e-04, 1.1558e-02, 3.5088e-01,  ..., 2.8029e-08, 4.3268e-08,
         5.4087e-05],
        [1.9682e-04, 3.5009e-04, 1.9909e-02,  ..., 4.3988e-08, 7.0937e-08,
         6.8715e-05],
        ...,
        [5.3984e-05, 3.2178e-05, 2.3204e-05,  ..., 4.4634e-01, 4.5972e-04,
         1.1716e-04],
        [1.5450e-06, 9.4312e-07, 6.9332e-07,  ..., 9.3677e-01, 1.8987e-02,
         2.3182e-06],
        [3.3097e-05, 4.1517e-05, 4.5450e-05,  ..., 4.4729e-07, 6.5514e-07,
         3.6789e-02]]), 'X': tensor([[ -8.5506,  -5.1154,   6.3423],
        [ -8.0330,  -6.0765,   6.7945],
        [ -7.6719,  -6.6260,   7.1468],
        [ -6.4620,  -8.0449,   6.2719],
        [ -7.6525,  -6.6534,   5.1721],
        [ -7.3470,  -7.0615,   5.0066],
        [  1.8447,  -4.7372,   8.1235],
        [  1.8320,   7.3871,   5.1452],
        [ 10.3251,   8.2928,  -3.0006],
        [  5.1667,  -2.6543,   1.2879],
        [ 12.4592,  -0.9168,  -6.7571],
        [  4.4915,   5.5990,   7.0405],
        [  7.8982,  -8.6778,   0.0239],
        [ 11.5966,   5.9666,   6.0083],
        [  6.2630,   2.1057,  -5.2136],
        [  8.5716,  10.3596,   6.8818],
        [ 11.2847,  -4.5848,  -4.7503],
        [  4.5057,  -3.3894,  -0.8160],
        [ 12.2311,  -2.0445,   1.3545],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.4489,   7.5256,   4.4529],
        [ 11.9641,  -2.9568,   7.7211],
        [  0.1460,   7.8340,  -9.4614],
        [ -3.9508,  -9.8515,   1.7923],
        [ -7.8126,   4.3066,   5.1791],
        [  6.3551,   0.7149,   1.3344],
        [ -1.6086,   7.8928,   4.3385],
        [  1.2273,  -4.8081,   9.8129],
        [  0.9967,  -4.8144,  -9.7370],
        [ -2.2508, -10.5479,  -2.7235],
        [ -4.9217,   6.8979,  -1.7667],
        [  4.2041,  -3.6509,  -4.9488],
        [  1.7228,  -4.7575,   8.3396],
        [  3.2990,  -4.2415,  -7.4857],
        [ -7.2986,  -7.1225,  -3.0814],
        [  3.8111, -10.6787,  -4.6539],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.6414, -10.0303,   7.8525],
        [ -1.5875,   7.8944,  -9.4286],
        [  0.6933,   7.7336,   2.1179],
        [ 11.4235,   6.3594,   5.1663],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.4430,  -0.1897,   4.8383],
        [ -5.0132,  -9.2234,  -8.3531],
        [  9.6614,   9.1784,  -1.2087],
        [  5.4167, -10.1293,   6.2156],
        [  5.1203,  -2.7151,   6.2110],
        [  6.2247,  -0.3150,  -9.3657],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.2811,  -7.4336,   5.1771],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.3038,   7.8092,   0.4164],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.3460,   1.4252,   1.2291],
        [  5.9655,   3.2172,  -7.5953],
        [  4.6063,  13.0051,  -3.1456],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.8617,   3.4853,  -9.8080],
        [  5.5101,  -2.1433,   9.5442],
        [  7.7794,  11.0629,   8.9069],
        [  6.3606,   0.9204,   3.2525],
        [  6.9284,  11.7061,  -0.3732],
        [  6.1579,  -0.5961,  -6.5595],
        [ 10.3251,   8.2928,  -3.0006],
        [  3.3069,   6.6089,   8.7377],
        [  6.2883,   0.0325,   5.2692],
        [ -4.2040,   7.2486,   2.9875],
        [  2.3373,  13.7721,  -0.2442],
        [  2.3200,   7.1756,   1.1938],
        [ 10.9717,  -5.1646,  -6.9039],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.1178,   1.5784,   2.1410],
        [  7.5064,  -8.9651,  -0.0994],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.6078, -10.9229,   6.2869],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.9433,  -9.3362,  -0.3421],
        [ -3.1204, -10.2345,   9.2288],
        [  9.1820,   9.7334,  -2.7932],
        [  0.2480, -11.0153,  -3.0183],
        [  6.9314,  -9.3435,  -1.2729],
        [  2.3373,  -4.6232,  -5.2246],
        [  0.1206,  -4.7363,   8.7103],
        [  8.4691,  10.4571,  -5.0650],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.4633,   5.6287,  -0.7754],
        [ -1.7290,   7.8825,  -3.3831],
        [  7.1697,  11.5343,  -0.7276],
        [  5.2199,   4.6861,  -1.9304],
        [  2.2849, -10.9655,  -7.6542],
        [  5.5626,  -2.0536,   8.4433],
        [ -1.5875,   7.8944,  -9.4286],
        [  8.8091,  -7.9037,   0.3631],
        [ 11.6019,   5.9540,  -5.1715],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.4735,  -1.2620,  -3.6742],
        [  4.7739,  -3.1218,  -4.1188],
        [  0.5600, -11.0321,  -0.8660],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.9548,  -1.2273,  -2.6520],
        [  0.8572,  14.0501,  -8.3364],
        [  4.0362,  13.2397,  -7.0030],
        [ -6.5295,  -7.9785,  -7.4429]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0357, 0.0600,  ..., 1.4158, 1.5254, 0.9822],
        [0.0357, 0.0000, 0.0244,  ..., 1.4388, 1.5457, 1.0177],
        [0.0600, 0.0244, 0.0000,  ..., 1.4532, 1.5581, 1.0417],
        ...,
        [1.4158, 1.4388, 1.4532,  ..., 0.0000, 0.1784, 1.0019],
        [1.5254, 1.5457, 1.5581,  ..., 0.1784, 0.0000, 1.1775],
        [0.9822, 1.0177, 1.0417,  ..., 1.0019, 1.1775, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.96391045, 10.07238463, 10.1371552 , 10.31886066, 10.14044489,
       10.19036081,  5.08372149,  7.61088125, 13.2430377 ,  5.80860165,
       12.49291768,  7.17789666, 11.73397309, 13.04156075,  6.60752373,
       13.44593519, 12.18045978,  5.63824971, 12.40074901, 10.1979669 ,
        7.66377554, 12.32408516,  7.83534987, 10.61417181,  8.92098752,
        6.39520397,  8.0550286 ,  4.96231532,  4.91652556, 10.78533838,
        8.47371868,  5.56809335,  5.05981914,  5.37343173, 10.1979669 ,
       11.33837396, 13.01728313,  5.70405525, 11.50791859,  8.05242591,
        7.76457095, 13.0743399 ,  9.85330646,  8.90405823,  9.44486609,
       10.4977132 , 13.32613445, 11.48663767,  5.79562691,  6.23262289,
        5.18277858, 11.89105259, 10.35093485,  5.01114719,  7.81509631,
        9.20340031,  6.50409372,  6.77777606, 13.79676654, 12.84056604,
        6.81961996,  5.91222346, 13.52430176,  6.42689705, 13.60275211,
        6.18668463, 13.2430377 ,  7.39004226,  6.28835101,  8.37954853,
       13.96905756,  7.54127425, 12.12641969,  6.80559669,  9.25336371,
       11.69264243,  5.34250225, 11.22993218,  5.1224079 , 11.63502679,
       10.69963954, 13.38091121, 11.018084  , 11.63383221,  5.18047638,
        4.73784222, 13.45642379, 10.31584102,  7.18354885,  8.06990483,
       13.58101794,  7.01475376, 11.20100795,  5.92952538,  8.05242591,
       11.83509372, 13.04051744, 12.10222204,  9.55720651,  5.70405525,
       11.04629471, 13.00413829,  6.07992623, 14.07623266, 13.84126197,
       10.30972193])}
{'P': tensor([[1.1570e-02, 2.1590e-01, 2.0636e-01,  ..., 1.4807e-07, 1.1060e-05,
         3.7748e-03],
        [6.8698e-05, 1.3763e-02, 2.1605e-02,  ..., 1.6917e-07, 4.6253e-06,
         6.3024e-05],
        [7.2560e-05, 4.4248e-03, 1.6198e-02,  ..., 1.9760e-07, 5.6785e-06,
         8.4324e-05],
        ...,
        [2.6985e-04, 8.0679e-05, 7.2838e-05,  ..., 2.4962e-01, 3.5754e-03,
         2.9422e-04],
        [2.5416e-02, 1.0871e-02, 1.4881e-02,  ..., 3.7171e-06, 1.2306e-01,
         1.6327e-02],
        [2.0743e-02, 2.8064e-01, 1.6535e-01,  ..., 2.0242e-07, 1.4113e-05,
         1.5573e-02]]), 'X': tensor([[-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0600, 0.0275,  ..., 1.1149, 0.4749, 0.0897],
        [0.0600, 0.0000, 0.0849,  ..., 1.1083, 0.5285, 0.0319],
        [0.0275, 0.0849, 0.0000,  ..., 1.1049, 0.4474, 0.1158],
        ...,
        [1.1149, 1.1083, 1.1049,  ..., 0.0000, 1.0503, 1.1202],
        [0.4749, 0.5285, 0.4474,  ..., 1.0503, 0.0000, 0.5604],
        [0.0897, 0.0319, 0.1158,  ..., 1.1202, 0.5604, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.97220799, 10.24557047, 10.26545792, 10.06850677, 10.36844813,
       10.38897563,  9.12220271,  9.9353921 ,  4.85603886,  5.83560009,
       10.95593729,  8.5724042 ,  7.02496223, 13.48809595,  9.99145096,
       13.89264822,  9.32094055,  8.62162372,  6.61386436,  8.79284466,
        5.69913817, 10.67132524,  6.4020404 ,  9.66977802, 10.92597196,
        7.44805901,  5.48276038, 12.59969936, 13.00413829,  5.83560009,
        5.37343173, 13.58333857,  6.00732193,  8.1303945 , 13.95472331,
        8.51088913, 11.72802491, 10.794755  ,  8.40445872,  6.53250937,
       10.31613269, 13.37121923, 13.6829843 , 10.09979747,  7.78980299,
        8.09984231,  6.1561646 , 13.3614598 ,  9.94173708, 13.69162795,
       12.27835996,  7.85919687,  6.74872981,  7.13501264, 13.66991027,
        5.5748624 , 12.53345445, 10.92597196, 12.32408516,  7.39599986,
       13.1399181 ,  8.92549752, 12.59840193,  9.69629   , 11.83570132,
        7.73692436, 11.1913598 , 10.52357479, 10.69963954, 11.65923396,
        8.36995953, 10.19036081, 10.14436912,  7.17789666,  7.01475376,
        8.37954853, 10.06023665,  7.27759233, 12.04924159, 11.07637531,
       10.04916678, 13.0493368 ,  6.383897  , 12.71843462,  5.94535529,
       12.50996828,  6.27715105, 11.52733581,  5.96556248, 13.6105595 ,
       12.40074901, 12.70489563,  9.34127013, 11.37939695,  6.39438044,
        5.53175275,  5.47702766, 13.04156075, 13.36092624, 10.68416401,
       13.99048094,  5.05989547,  7.16058587, 14.05633441,  9.18690018,
        9.95552626])}
{'P': tensor([[1.3120e-02, 2.0992e-01, 1.9155e-01,  ..., 2.2410e-04, 3.9768e-05,
         7.2104e-06],
        [7.3912e-05, 1.2554e-02, 2.3960e-01,  ..., 4.5140e-04, 9.4671e-06,
         8.6338e-06],
        [1.1636e-04, 6.7816e-04, 3.3685e-02,  ..., 1.1002e-03, 2.0696e-05,
         3.0679e-05],
        ...,
        [7.2428e-05, 1.1199e-04, 1.1334e-04,  ..., 1.0858e-01, 4.3844e-06,
         2.1578e-04],
        [1.7036e-03, 1.4244e-03, 2.0087e-03,  ..., 1.2364e-05, 4.7891e-02,
         2.8333e-05],
        [4.6184e-05, 2.5856e-05, 2.5606e-05,  ..., 3.5548e-05, 1.4782e-04,
         1.0425e-01]]), 'X': tensor([[-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.7031e-02, 3.8770e-02,  ..., 9.6608e-01, 6.6979e-01,
         1.0837e+00],
        [7.7031e-02, 0.0000e+00, 6.7137e-02,  ..., 9.3701e-01, 7.4607e-01,
         1.1179e+00],
        [3.8770e-02, 6.7137e-02, 3.0339e-04,  ..., 9.2747e-01, 6.8458e-01,
         1.0613e+00],
        ...,
        [9.6608e-01, 9.3701e-01, 9.2747e-01,  ..., 0.0000e+00, 1.3171e+00,
         8.5474e-01],
        [6.6979e-01, 7.4607e-01, 6.8458e-01,  ..., 1.3171e+00, 0.0000e+00,
         9.1174e-01],
        [1.0837e+00, 1.1179e+00, 1.0613e+00,  ..., 8.5474e-01, 9.1174e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.97822473, 10.23946247, 10.32715715, 10.33712258, 10.24557047,
       10.36844813, 12.91505111,  8.28968067, 12.38881635,  6.33517349,
        6.60752373,  6.42689705,  5.43943791, 11.0050379 ,  4.79003361,
       11.75807252, 11.92133129,  8.97052115, 11.018084  ,  6.07992623,
        7.62232307, 13.14777433, 12.44177641, 10.05558893, 13.13205357,
       13.89671314,  8.51088913,  4.91652556, 10.7506423 ,  7.13501264,
       11.8159254 , 11.59512194,  9.43112957,  9.15866607,  5.22903117,
        8.0550286 , 12.4013153 , 13.26590071,  9.562174  ,  9.48337063,
       13.83139024, 12.25806559,  9.97822473,  9.26455795,  5.94909709,
        7.60529892,  9.50913849,  9.7712321 ,  8.92549752, 12.35421461,
        4.91038884,  7.28793663,  9.69629   , 13.96905756,  5.63824971,
       14.0393949 ,  5.1805345 ,  8.62631801, 12.84056604,  9.96391045,
       11.66453059,  8.92098752,  7.18354885,  5.92952538,  7.72291946,
        7.17789666,  4.96231532, 10.52357479,  5.19986797, 11.22142195,
       10.61047596,  5.16077764,  5.05989547, 11.22596994, 10.39942497,
       12.49317563,  4.91296205, 11.64436684, 12.88270018,  6.65991877,
       11.0050379 , 13.2778378 , 11.73397309, 12.86648771,  8.22728399,
        7.67939546,  7.83486943,  6.35029274,  8.89458273, 13.89671314,
       11.07637531,  9.26455795,  5.51177501,  5.14124612, 13.35013374,
        8.27853076,  7.5158489 ,  5.91854397,  6.59176018, 13.20988996,
        6.81084724,  9.9353921 ,  4.98461983,  5.22903117,  9.64157461,
        6.76791483])}
{'P': tensor([[7.6486e-03, 3.6319e-01, 1.1234e-01,  ..., 1.8541e-06, 1.8541e-06,
         1.1244e-07],
        [1.6141e-04, 1.0738e-02, 1.8483e-01,  ..., 2.8895e-06, 2.8895e-06,
         1.5824e-07],
        [8.3312e-05, 9.9168e-05, 1.3979e-02,  ..., 7.0188e-06, 7.0188e-06,
         2.2374e-07],
        ...,
        [7.1504e-07, 6.7030e-07, 4.7715e-07,  ..., 3.0630e-02, 3.0630e-02,
         3.1450e-03],
        [7.1504e-07, 6.7030e-07, 4.7715e-07,  ..., 3.0630e-02, 3.0630e-02,
         3.1450e-03],
        [9.0478e-06, 7.6634e-06, 2.9384e-06,  ..., 3.4324e-07, 3.4324e-07,
         2.1083e-02]]), 'X': tensor([[ -8.4824,  -5.2575,  -0.0795],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.9838,  -7.4965,   0.9796],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -6.0844,  -8.3955,   0.7705],
        [  2.6614,  -4.5192,  -9.3641],
        [ -3.2736, -10.1703,   3.8712],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.7122,  -4.7591,  -3.6734],
        [ -1.8823, -10.6558,  -3.8952],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -8.4824,  -5.2575,  -0.0795],
        [  0.5175,  14.0910,   4.6540],
        [ -3.4986,   7.5152,   9.0712],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.2659,  13.7894,   6.5436],
        [  0.8572,  14.0501,  -8.3364],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -2.2188,   7.8218,   8.6064],
        [ -8.5900,   2.9767,   7.4811],
        [ -0.5739,   7.9056,   0.6382],
        [ -6.3489,   5.9117,   6.5161],
        [  5.6248,   3.9984,   8.4933],
        [  3.3094,  13.4969,  -3.6507],
        [ -9.0961,   1.6529,   3.1553],
        [  7.2340,  -9.1505,  -5.8321],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 12.5789,   0.1722,   1.6470],
        [  1.7500,   7.4187,   4.4237],
        [  3.5965,  -4.0754,  -3.5035],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.5318,  13.7231,   6.3692],
        [  5.6202,  -1.9508,   0.7471],
        [ -6.3066,  -8.1936,   9.5822],
        [  5.2911,  -2.4824,  -3.6877],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.0534,  -6.0429,   2.4583],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 12.5832,   1.7543,  -7.6134],
        [  9.3560,   9.5391,   9.4943],
        [ -6.8466,  -7.6488,   1.9648],
        [ 12.5154,   2.5072,  -9.3274],
        [  4.5491,  -3.3485,   6.8412],
        [ -6.3066,  -8.1936,   9.5822],
        [  8.1043,  -8.5164,  -7.9290],
        [ -7.4602,   4.7718,  -4.6277],
        [ -7.7239,   4.4298,   3.1216],
        [  1.7903,  -4.7467,   0.1168],
        [  5.7218,  -9.9934,   6.5764],
        [ 11.3367,  -4.4806,   0.4643],
        [  5.0190,   4.9705,  -7.9752],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.1795,  -3.3623,   8.5259],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.3119,   1.7797,   9.9186],
        [  0.3724,   7.7974,  -6.8722],
        [  2.5974,   7.0363,  -7.7810],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.9936,  -2.8677,   6.0614],
        [  9.2811,  -7.4336,   5.1771],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.6587,   9.1817,  -0.8212],
        [ -8.6112,   2.9319,  -7.9943],
        [  3.2159, -10.8166,  -1.0406],
        [  8.9284,  10.0036,  -6.9210],
        [  6.3236,   0.3011,  -4.6668],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.1625,  -4.6697,   7.0658],
        [  0.1047, -11.0046,   0.5188],
        [  2.8602,   6.8905,   5.8779],
        [  6.3266,   0.3292,   9.3502],
        [ -9.3616,  -2.4750,   3.7765],
        [  0.9967,  -4.8144,  -9.7370],
        [  2.5829,  -4.5466,  -8.4015],
        [ -8.3900,   3.3727,   3.1650],
        [  6.3211,   1.7017,  -0.2164],
        [ -1.8367,   7.8718,  -8.6523],
        [ -2.6485, -10.4151,   0.8766],
        [  4.0457,  -3.7737,  -0.4704],
        [ -4.7389,   6.9953,   3.7638],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.9033,   2.2329,  -8.8912],
        [  0.6470, -11.0352,  -1.1931],
        [ -2.9497, -10.3028,  -7.3479],
        [ -9.1459,   1.4780,  -4.2403],
        [  1.4306,  13.9620,  -0.2075],
        [  5.3804,  -2.3504,   7.2744],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -7.5401,   4.6718,   9.0033],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.8474,   6.8979,   6.6457],
        [  4.8674,  12.8875,   6.7622],
        [ -9.4203,  -2.0340,   2.5856],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.5919,   0.4035,  -5.8460],
        [  4.1637,  13.1898,  -1.9189]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0083, 0.0870,  ..., 1.1782, 1.1782, 1.1915],
        [0.0083, 0.0000, 0.0787,  ..., 1.1801, 1.1801, 1.1912],
        [0.0870, 0.0787, 0.0000,  ..., 1.2014, 1.2014, 1.1913],
        ...,
        [1.1782, 1.1801, 1.2014,  ..., 0.0000, 0.0000, 0.3234],
        [1.1782, 1.1801, 1.2014,  ..., 0.0000, 0.0000, 0.3234],
        [1.1915, 1.1912, 1.1913,  ..., 0.3234, 0.3234, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.97964013, 10.01804223, 10.24557047, 10.32715715, 10.33712258,
       10.36844813,  5.24461068, 10.68416401, 13.04217487,  5.05774295,
       10.8207297 , 12.35421461,  9.97964013, 14.10045797,  8.28968067,
       10.53368644, 13.97430191, 14.07623266, 10.76154256,  9.41605684,
        9.60080639, 10.00410625,  8.1303945 ,  9.09119351,  7.92644746,
        8.67502348,  6.90115901, 13.89671314,  9.24501829, 11.66453059,
       12.49291768, 12.58006038,  7.62232307,  5.43544626, 11.67875855,
       13.95472331,  5.94909709, 10.33958639,  5.84450127,  8.40445872,
       10.06850677, 10.52587243, 12.70489563, 13.3614598 , 10.26545792,
       12.76408562,  5.64865173, 10.33958639, 11.75618513,  8.85572292,
        8.90405823,  5.07306274, 11.51557166, 12.18998195,  7.06372172,
        7.822854  ,  9.77588801, 10.09979747, 12.18998195,  6.55800267,
        7.80625981,  7.50035355,  5.23878371, 12.33167382, 11.89105259,
       12.59969936,  9.60412735,  9.68324553, 13.32645552,  9.09661212,
       11.2845608 , 13.40850328,  6.33075682, 11.28895489, 10.27716434,
        9.41605684,  5.14607014, 11.00508681,  7.46053281,  6.33517349,
        9.68324553,  4.91652556,  5.22903117,  9.04255655,  6.54615887,
        8.08320235, 10.74655346,  5.53255882,  8.44940661, 14.07623266,
        9.17905491, 11.05413545, 10.7167285 ,  9.26455795, 14.03505624,
        5.87132718, 12.32545637,  8.87007495, 10.24557047,  7.46248947,
       13.77604293,  9.63742669,  9.47369628, 12.59840193, 12.59840193,
       13.83139024])}
{'P': tensor([[1.3330e-02, 2.5616e-01, 1.6197e-01,  ..., 9.1274e-08, 1.6474e-04,
         1.2053e-06],
        [6.8014e-05, 1.5148e-02, 6.8873e-01,  ..., 1.0831e-07, 3.8810e-04,
         1.8762e-06],
        [2.8753e-04, 8.7832e-04, 9.6598e-02,  ..., 7.2991e-07, 3.2210e-03,
         1.3022e-05],
        ...,
        [2.4798e-05, 8.0815e-06, 5.5250e-06,  ..., 1.3302e-01, 8.2902e-06,
         1.3979e-05],
        [5.3177e-05, 5.3224e-05, 5.5166e-05,  ..., 1.8451e-04, 8.4604e-02,
         5.5294e-05],
        [1.5885e-06, 1.3399e-06, 1.2676e-06,  ..., 5.3224e-04, 2.6517e-05,
         6.9755e-02]]), 'X': tensor([[ -8.4297,  -5.3637,  -9.3962],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 11.4235,   6.3594,   5.1663],
        [  6.0073,   3.0986,   6.7937],
        [  6.1372,   2.6726,  -6.7885],
        [  2.5010,  -4.5736,  -0.8989],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.3266,   0.3292,   9.3502],
        [ 10.2470,   8.4054,  -0.1177],
        [  6.1929,   2.4501,  -6.9352],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.4495,   6.3026,   6.8257],
        [ 10.0619,   8.6627,  -9.2884],
        [ -7.4808,   4.7462,   2.0266],
        [  5.8889,  -1.3916,   8.8993],
        [  5.9912,   3.1453,  -3.9989],
        [  1.6422,   7.4586,  -6.8056],
        [  1.2728, -11.0372,   9.8121],
        [  1.3670,  -4.7990,   3.2603],
        [ -5.0132,  -9.2234,  -8.3531],
        [  5.9548,  -1.2273,  -2.6520],
        [  4.1140,  -3.7220,  -0.1273],
        [  1.8721,  -4.7322,   0.5524],
        [  5.9003,  12.3550,   2.1917],
        [  5.4426, -10.1181,  -5.1268],
        [ -1.8367,   7.8718,  -8.6523],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.2895,   0.0405,   5.7652],
        [ 12.5717,   1.9266,   7.4903],
        [ -7.3470,  -7.0615,   5.0066],
        [ -7.7239,   4.4298,   3.1216],
        [  0.7685,  -4.8098,  -2.9310],
        [  1.8320,   7.3871,   5.1452],
        [  6.2675,  12.1379,  -8.9720],
        [  5.6258,  -1.9404,  -6.0715],
        [ -8.0750,  -6.0072,  -5.8267],
        [  2.3261,   7.1726,   6.2522],
        [ 12.4691,   2.8525,  -5.5110],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.6523,   3.9443,  -2.9449],
        [  5.7784,  -1.6401,   9.5587],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -9.2974,   0.8371,   2.4852],
        [ -1.8788, -10.6567,   3.7068],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.3900,   3.3727,   3.1650],
        [ -1.9714,   7.8563,  -1.8084],
        [  0.1677,  -4.7447,  -6.1529],
        [ -1.9714,   7.8563,  -1.8084],
        [  6.3048,   0.1469,  -9.1691],
        [  9.4516,  -7.2501,  -1.8868],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.3542,   0.6934,  -3.3840],
        [  1.6232,  -4.7717,  -2.5655],
        [  0.6363, -11.0348,   6.1414],
        [ -5.1377,   6.7751,  -0.4438],
        [ -7.0825,   5.2073,  -5.0497],
        [ -7.6650,   4.5092,   0.2520],
        [  6.9284,  11.7061,  -0.3732],
        [  4.0362,  13.2397,  -7.0030],
        [  3.2809,  13.5061,  -4.0199],
        [ -4.3963,   7.1629,   4.5263],
        [ -9.1795,  -3.3623,   8.5259],
        [ 11.6019,   5.9540,  -5.1715],
        [ -7.4733,   4.7555,   6.2490],
        [ -9.3087,   0.7792,  -7.5587],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.1771, -10.9776,  -3.2007],
        [  4.0642,  -3.7599,  -3.4565],
        [ -3.6381,   7.4683,   7.3544],
        [ -4.0868,   7.2981,   8.1874],
        [  5.0190,   4.9705,  -7.9752],
        [  3.7992,  -3.9474,  -8.9019],
        [  2.0937,  13.8293,  -2.3533],
        [  0.9478,  -4.8144,   0.6371],
        [ -8.4824,  -5.2575,  -0.0795],
        [  0.1287, -11.0065,   6.5757],
        [  1.9312,  13.8649,   8.3563],
        [  6.3606,   0.9204,   3.2525],
        [  2.1477,   7.2550,   8.9688],
        [ -7.7842,   4.3466,  -1.7548],
        [  4.1348,   5.9514,  -0.8856],
        [  3.2366,  -4.2732,   5.1532],
        [ -5.6792,  -8.7352,   4.0318],
        [ -5.1377,   6.7751,  -0.4438],
        [  2.3261,  13.7749,   5.9121],
        [  6.3031,   1.8477,  -7.1706],
        [ -3.5173, -10.0623,   4.8280],
        [ 10.0400,   8.6922,  -6.2847],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.3119,   1.7797,   9.9186],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.3604,   0.9042,  -6.5901],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.9956,  11.6590,  -7.6876],
        [  3.3633,  13.4794,   0.5193],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.1222,  -0.7261,   8.8204]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0345, 0.0743,  ..., 1.3953, 1.0424, 1.5302],
        [0.0345, 0.0000, 0.0422,  ..., 1.3744, 1.0137, 1.5193],
        [0.0743, 0.0422, 0.0000,  ..., 1.3362, 0.9718, 1.4892],
        ...,
        [1.3953, 1.3744, 1.3362,  ..., 0.0000, 0.5310, 0.5005],
        [1.0424, 1.0137, 0.9718,  ..., 0.5310, 0.0000, 0.9547],
        [1.5302, 1.5193, 1.4892,  ..., 0.5005, 0.9547, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.99145096, 10.27716434, 10.37698435, 10.00410625, 10.35485331,
       10.30972193, 13.0743399 ,  6.7593917 ,  6.69389431,  5.21280156,
       10.1371552 ,  6.33517349, 13.25336001,  6.65991877, 12.10942412,
       13.06957425, 13.27717982,  8.859416  ,  6.05113217,  6.76661595,
        7.63726134, 11.11038551,  4.98989906, 10.4977132 ,  6.07992623,
        5.54776934,  5.08910491, 13.69162795, 11.4890861 ,  8.08320235,
        9.60412735,  6.28961988, 12.71843462, 10.19036081,  8.90405823,
        4.87083348,  7.61088125, 13.66051884,  5.9510489 , 10.06437598,
        7.5403822 , 12.79126785,  9.75075273, 10.99326577,  9.37989631,
        6.89245352,  6.00663314, 12.42554808,  9.3349858 , 10.82106505,
        5.91677188,  9.04255655,  8.09984231,  4.74770899,  8.09984231,
        6.30647746, 11.9120321 ,  4.8763135 , 12.67576792,  6.39187285,
        5.04028538, 11.05317597,  8.50278896,  8.7907923 ,  8.89301516,
       13.60275211, 13.84126197, 13.89886469,  8.40445872,  9.77588801,
       13.04051744,  8.8580671 ,  9.34127013, 10.67132524, 11.1913598 ,
        5.5366698 ,  8.30729352,  8.36445758,  7.06372172,  5.47866221,
       13.98691426,  4.90676242,  9.97964013, 11.00726483, 13.99877413,
        6.42689705,  7.56617037,  8.91551231,  7.24677664,  5.36061974,
       10.41909211,  8.50278896, 13.96987806,  6.56833864, 10.65929968,
       13.27994304,  9.75401449,  6.55800267, 10.30972193,  9.55593374,
        6.42439344,  5.83560009, 13.59673406, 13.89264822,  5.83560009,
        6.16512856])}
{'P': tensor([[6.2943e-03, 3.0037e-01, 2.1066e-01,  ..., 1.2054e-05, 6.5681e-07,
         1.4111e-05],
        [1.7074e-04, 1.5587e-02, 9.2863e-04,  ..., 2.1505e-05, 1.3636e-06,
         4.7556e-05],
        [1.6805e-04, 1.2478e-01, 9.6159e-03,  ..., 1.4403e-05, 8.3479e-07,
         2.7915e-05],
        ...,
        [8.7016e-02, 6.9944e-02, 6.6704e-02,  ..., 6.4444e-02, 8.8856e-05,
         1.1082e-05],
        [2.4991e-03, 2.0387e-03, 1.9573e-03,  ..., 3.7819e-02, 3.4505e-01,
         3.3042e-06],
        [4.1042e-06, 4.0923e-06, 4.1565e-06,  ..., 5.0398e-06, 4.4478e-06,
         1.5205e-01]]), 'X': tensor([[ -8.4267,  -5.3697,   2.4172],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.1531,  -5.8748,   2.6934],
        [ -7.9925,  -6.1423,   2.8990],
        [ -6.8466,  -7.6488,   1.9648],
        [ -6.8618,  -7.6322,   3.7027],
        [ 11.1420,   6.9331,  -2.4779],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.2022,   2.4092,   3.0349],
        [  3.2647, -10.8066,  -0.3768],
        [ -4.8751,  -9.3148,   4.0406],
        [ 11.5456,   6.0861,   8.1101],
        [  0.3724,   7.7974,  -6.8722],
        [  7.0358,  -9.2784,  -9.8479],
        [ -9.4296,  -1.9458,   4.0833],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -5.1369,   6.7755,   8.3836],
        [ 10.0567,   8.6697,   4.6329],
        [  1.1350,  14.0104,   0.6321],
        [ -9.2508,   1.0572,   2.9978],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.7121,   6.3117,   0.6738],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3542,   0.6934,  -3.3840],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.9433,  -9.3362,  -0.3421],
        [  5.4426, -10.1181,  -5.1268],
        [  6.2419,  -0.2313,   4.2440],
        [  3.7121,   6.3117,   0.6738],
        [ -1.0407,   7.9166,  -6.6021],
        [  7.2340,  -9.1505,  -5.8321],
        [  2.7295,  13.6703,  -8.9328],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.6050,   2.9451,  -0.7012],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.2345,  -3.1345,   3.9715],
        [ -5.9938,  -8.4745,   3.9735],
        [  3.2159, -10.8166,  -1.0406],
        [  6.3360,   0.4258,  -9.2509],
        [ -5.9940,   6.1985,   5.9566],
        [  2.8602,   6.8905,   5.8779],
        [  6.7792,  11.8084,   1.6251],
        [  6.2630,   2.1057,  -5.2136],
        [  3.9663,   6.1018,  -5.4104],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.7784,  -1.6401,   9.5587],
        [ -4.6393,   7.0461,   1.1624],
        [ -7.3470,  -7.0615,   5.0066],
        [  2.7558,  -4.4844,   0.0636],
        [ -9.4501,  -1.7133,   2.2980],
        [ -7.8359,   4.2735,   7.2943],
        [  9.3624,   9.5318,   0.2779],
        [ -8.6603,   2.8250,   5.9181],
        [  6.1737,   2.5305,   3.9049],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -9.1706,   1.3863,   8.4032],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.8836,   3.4315,  -4.0327],
        [  4.4039,   5.6903,   8.7366],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.5900,   2.9767,   7.4811],
        [  6.3549,   0.7096,   6.9198],
        [ -8.9942,  -3.9990,   6.0448],
        [  6.0788,  -0.8710,  -6.8719],
        [ -0.5739,   7.9056,   0.6382],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.5267,  -0.4125,  -8.7331],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.1196,  -2.7160,   4.2705],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.6095,  -1.9702,  -7.8233],
        [ -0.1626, -10.9796,   6.4629],
        [  2.0039,   7.3173,  -1.5885],
        [ -9.3810,  -2.3455,  -9.5194],
        [  0.1497,   7.8334,  -5.4412],
        [  2.2860,   7.1916,  -6.1011],
        [  2.5469,  13.7192,  -9.6935],
        [  9.3424,  -7.3685,  -3.6807],
        [  6.1984,  -0.4327,   1.5438],
        [  4.9957,  -2.8705,   6.5978],
        [  9.2811,  -7.4336,   5.1771],
        [  4.5800,   5.5034,  -3.9432],
        [  0.4996,   7.7738,  -4.4711],
        [  2.1440, -10.9811,   8.5281],
        [ -4.4520,   7.1370,  -8.7867],
        [ 11.4495,   6.3026,   6.8257],
        [ -2.1382,   7.8339,   7.3524],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -9.4430,  -0.1897,   4.8383],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.5456,   6.0861,   8.1101],
        [  0.2571,  -4.7594,   1.5618],
        [  2.3373,  13.7721,  -0.2442],
        [  2.8994,   6.8675,  -9.5987],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.6093,  -1.9705,   6.4807],
        [  0.8940, -11.0402,  -0.9783],
        [ -9.3419,   0.5972,  -1.2841],
        [ -5.9872,   6.2037,  -9.4477],
        [ 10.8218,  -5.4177,   6.8573]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0075, 0.0155,  ..., 0.3203, 0.7730, 0.9247],
        [0.0075, 0.0000, 0.0178,  ..., 0.3153, 0.7668, 0.9205],
        [0.0155, 0.0178, 0.0000,  ..., 0.3327, 0.7830, 0.9126],
        ...,
        [0.3203, 0.3153, 0.3327,  ..., 0.0000, 0.4647, 1.0455],
        [0.7730, 0.7668, 0.7830,  ..., 0.4647, 0.0000, 1.2162],
        [0.9247, 0.9205, 0.9126,  ..., 1.0455, 1.2162, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([ 9.99211663, 10.06850677, 10.04916678, 10.08001894, 10.26545792,
       10.26327895, 13.12299145,  5.48276038,  6.65368551, 11.28895489,
       10.51340829, 13.05148826,  7.80625981, 11.64436684,  9.6282763 ,
       11.64746882,  9.69792513, 12.04924159,  8.50268322, 13.2778378 ,
       14.05633441,  9.3109884 , 11.77124373,  7.32233733, 13.36092624,
        6.39187285, 12.42554808,  4.78002665,  6.25586297, 11.63502679,
       11.4890861 ,  6.24614905,  7.32233733,  7.98469166, 11.66453059,
       13.94008948, 13.06957425,  9.09502385, 12.22276278,  9.75200848,
       10.37998078, 11.2845608 ,  6.35029274,  8.62260327,  7.46053281,
       13.61600536,  6.60752373,  7.27759233,  9.15976746,  9.53015376,
        6.00663314,  8.43626114, 10.19036081,  5.26342203,  9.60412735,
        8.92549752, 13.3607379 ,  9.10945678,  6.67218687, 12.49291768,
        9.27474386, 10.1979669 ,  6.81117802,  7.19534146,  8.86082124,
        9.09119351,  6.39438044,  9.84314712,  6.14086582,  7.92644746,
        8.37954853, 12.53345445, 12.53285633,  5.79543038,  9.7712321 ,
        5.94541803, 10.98076873,  7.58668144,  9.66977802,  7.83486943,
        7.54620849, 13.95361098, 11.89855267,  6.2134961 ,  5.76165934,
       11.89105259,  7.15990676,  7.78980299, 11.18839169,  8.41171369,
       13.06957425,  8.12043983,  7.71075226,  9.50913849,  9.44486609,
       13.20597932, 13.05148826,  4.7663526 , 13.96905756,  7.4544866 ,
       11.54672755,  5.94535529, 11.07637531,  9.36093562,  8.62162372,
       12.10222204])}
{'P': tensor([[1.2772e-02, 2.4919e-01, 1.5863e-01,  ..., 6.7677e-06, 2.5978e-07,
         8.0877e-03],
        [3.9904e-05, 8.2729e-03, 3.7615e-01,  ..., 1.1563e-05, 2.5600e-07,
         9.8127e-04],
        [1.2807e-04, 3.6767e-04, 4.0436e-02,  ..., 8.3312e-05, 1.5225e-06,
         2.0697e-03],
        ...,
        [1.0279e-06, 8.0504e-07, 7.7256e-07,  ..., 3.8016e-02, 1.7750e-02,
         7.9391e-07],
        [2.3371e-06, 1.0311e-06, 7.9383e-07,  ..., 4.3230e-06, 5.0803e-02,
         1.6628e-06],
        [2.5352e-04, 1.9204e-03, 3.7800e-03,  ..., 4.8361e-05, 1.8264e-06,
         2.9532e-02]]), 'X': tensor([[ -8.3717,  -5.4769,  -9.2228],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.7565,  11.8237,  -4.9882],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.1984,  -0.4327,   1.5438],
        [  3.2366,  -4.2732,   5.1532],
        [ -6.4464,   5.8272,   7.5038],
        [  2.2182,  13.8007,   5.1889],
        [  2.1379,  -4.6757,   6.4981],
        [  8.0877,  -8.5297,  -1.1050],
        [  6.3211,   1.7017,  -0.2164],
        [  5.1793,   4.7461,   0.3899],
        [  4.3523, -10.5232,   8.2138],
        [ 12.5258,  -0.4200,  -9.3493],
        [  1.3103,  -4.8032,  -0.9300],
        [  4.1297,  -3.7098,  -3.9838],
        [ -2.2508, -10.5479,  -2.7235],
        [ -6.0175,  -8.4541,  -9.8408],
        [  2.0870,  -4.6877,   9.6823],
        [ -1.7009,   7.8851,  -8.2437],
        [  1.0000,   7.6591,  -4.5011],
        [ 12.2592,   3.9425,   2.0383],
        [  2.7976,  -4.4682,  -4.2020],
        [ -9.4430,  -0.1897,   4.8383],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.1274,  14.0116,   8.1753],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.4306,  13.9620,  -0.2075],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.2644,  -4.6434,   9.7508],
        [  3.9895,   6.0816,  -6.5235],
        [  0.9445,   7.6736,   0.5095],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -3.0008, -10.2827,  -8.9480],
        [  4.5340,  -3.3629,   8.9652],
        [ -8.3062,  -5.6008,   0.0287],
        [ 11.1420,   6.9331,  -2.4779],
        [  9.9756,  -6.6330,  -3.3429],
        [ -1.1885, -10.8211,   0.0865],
        [ -8.3539,  -5.5111,   0.3373],
        [ -9.3411,  -2.6004,  -2.7062],
        [  2.5974,   7.0363,  -7.7810],
        [ -3.9456,  -9.8542,   0.7466],
        [ -7.0120,   5.2825,   7.7762],
        [ -1.6275, -10.7221,   3.2376],
        [ -8.3062,  -5.6008,   0.0287],
        [ -5.1974,   6.7396,   3.3484],
        [  3.6703,   6.3443,  -5.4401],
        [ 11.1085,  -4.9204,   0.0736],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 11.1085,  -4.9204,   0.0736],
        [ 10.9916,   7.2133,  -0.9179],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.3124,   0.2057,  -0.4593],
        [  5.9843,  -9.8679,  -6.1449],
        [ -3.1204, -10.2345,   9.2288],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -6.6998,   5.5954,   0.0823],
        [  6.3604,   0.9042,  -6.5901],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.7685, -10.8982,  -6.8928],
        [  8.2154,  -8.4262,   8.0171],
        [  7.0171,  11.6439,  -5.0178],
        [  3.6469,   6.3624,  -1.8414],
        [  0.6142,  14.0802,   0.5407],
        [  5.9193,  -1.3176,   5.1739],
        [  8.1217,  -8.5024,  -0.6149],
        [ -9.1907,  -3.3180,  -5.5943],
        [  8.6271,  -8.0715,  -7.4813],
        [  1.6408,   7.4591,  -4.2294],
        [  8.2422,  -8.4041,  -1.4962],
        [ -7.4807,   4.7463,   2.0616],
        [ 11.0864,   7.0387,  -0.1238],
        [  9.2692,   9.6370,   7.0746],
        [  2.3490,  -4.6199,  -4.2489],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.3413,   0.4883,  -6.8416],
        [ -5.5525,   6.5146,  -4.4984],
        [ -7.6719,  -6.6260,   7.1468],
        [ 11.5966,   5.9666,   6.0083],
        [  3.6249, -10.7255,   5.6605],
        [ 12.1751,  -2.2592,  -5.7751],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.1348,   5.9514,  -0.8856],
        [ -9.3419,   0.5972,  -1.2841],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.5158,  -2.1337,   3.5246],
        [  9.0004,  -7.7195,   4.6945],
        [ -9.1795,  -3.3623,   8.5259],
        [  2.9260,  -4.4162,   8.3808],
        [  4.5340,  -3.3629,   8.9652],
        [ -5.6169,  -8.7844,   8.9057],
        [  0.6396,  14.0772,  -6.3786],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.6587,   9.1817,  -0.8212],
        [ -6.4822,  -8.0252,  -3.8983]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0347, 0.0744,  ..., 1.1316, 1.3835, 0.2511],
        [0.0347, 0.0000, 0.0422,  ..., 1.1013, 1.3602, 0.2611],
        [0.0744, 0.0422, 0.0000,  ..., 1.0591, 1.3204, 0.2579],
        ...,
        [1.1316, 1.1013, 1.0591,  ..., 0.0000, 0.4769, 1.0911],
        [1.3835, 1.3602, 1.3204,  ..., 0.4769, 0.0000, 1.2595],
        [0.2511, 0.2611, 0.2579,  ..., 1.0911, 1.2595, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.00410625, 10.27716434, 10.37698435, 10.35485331, 10.30972193,
       10.46739073, 13.61801526,  6.54615887, 12.26997165,  6.2134961 ,
        5.36061974,  8.68979052, 13.97779839,  5.14124612, 11.75438006,
        6.54615887,  7.02496223, 11.38774149, 12.53285633,  4.978704  ,
        5.5513072 , 10.78533838, 10.37698435,  5.13125513,  8.06643487,
        7.72415048, 12.87752052,  5.27178363,  9.44486609,  9.85384623,
       14.05687815,  6.07992623, 14.03505624,  9.79814997,  5.16611782,
        7.27340834,  7.73151309, 12.49291768, 10.71162681,  5.64500512,
       10.01804223, 13.12299145, 11.97957425, 10.88618114, 10.00794468,
        9.69629   ,  7.50035355, 10.61472636,  8.77913428, 10.84493732,
       10.01804223,  8.51088913,  7.32950385, 12.1494023 , 10.54658597,
       12.1494023 , 13.1471313 ,  5.19065439, 10.06286074,  6.31576264,
       11.54071859, 10.69963954,  8.7273569 ,  9.75075273,  9.75401449,
        8.72896381,  6.42439344,  9.4997987 , 11.24434374, 11.76830981,
       13.59480906,  7.33350642, 14.09357547,  6.06416957, 11.75807252,
        9.7712321 , 11.81423161,  7.6374603 , 11.77124373,  8.85939626,
       13.13205357, 13.37121923,  5.18277858, 12.889695  ,  6.36004393,
        8.55982105, 10.1371552 , 13.04156075, 11.32149259, 12.3828957 ,
       11.1341798 ,  9.54154096,  7.24677664,  9.36093562, 13.25336001,
        9.99211663,  5.91408504, 11.85743425,  9.77588801,  5.29755487,
        5.64500512, 10.42667093, 14.09176259, 12.49291768, 13.32645552,
       10.31613269])}
{'P': tensor([[1.1985e-02, 2.4495e-01, 1.6329e-01,  ..., 1.7243e-04, 1.5895e-04,
         2.3001e-05],
        [7.3592e-05, 1.2287e-02, 5.0108e-01,  ..., 3.2895e-04, 2.2577e-04,
         3.1081e-05],
        [9.9308e-05, 2.5139e-04, 3.0319e-02,  ..., 9.1729e-04, 7.5625e-04,
         1.0431e-04],
        ...,
        [1.2917e-04, 1.8278e-04, 2.0800e-04,  ..., 2.2242e-01, 5.6562e-04,
         1.4973e-04],
        [5.1895e-05, 5.4613e-05, 6.3433e-05,  ..., 1.6971e-04, 9.7354e-02,
         6.3644e-02],
        [1.7776e-05, 1.5223e-05, 1.6346e-05,  ..., 2.4816e-05, 3.8415e-03,
         8.8759e-02]]), 'X': tensor([[-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0472, 0.0456,  ..., 1.0529, 0.9909, 1.0690],
        [0.0472, 0.0000, 0.0230,  ..., 1.0246, 1.0015, 1.0824],
        [0.0456, 0.0230, 0.0000,  ..., 1.0113, 0.9785, 1.0594],
        ...,
        [1.0529, 1.0246, 1.0113,  ..., 0.0000, 0.8572, 0.9479],
        [0.9909, 1.0015, 0.9785,  ..., 0.8572, 0.0000, 0.1012],
        [1.0690, 1.0824, 1.0594,  ..., 0.9479, 0.1012, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.00794468, 10.24557047, 10.36844813, 10.32715715, 10.33712258,
       10.26545792, 11.28215429,  7.37902766,  9.64157461,  9.18358562,
        6.25586297,  9.15976746,  6.39438044, 10.69963954, 13.06957425,
        8.30729352,  5.82345867, 12.78327037, 13.05148826,  7.13501264,
       12.12448912, 10.09979747,  7.09887557, 13.76047999,  4.90505566,
       11.38774149,  9.36256096,  9.44486609, 12.84056604,  9.34061938,
        6.27905292,  7.06169391, 11.28215429,  7.50564223, 13.61600536,
       10.19036081,  9.73579738,  7.48538905, 13.6922425 ,  5.35774573,
       11.99633819,  9.34061938,  5.79562691,  7.06169391,  8.1137452 ,
        7.96181288,  6.36004393,  8.24808832,  9.50654967, 11.05672136,
       10.88618114, 10.98076873, 10.66631784,  9.32094055, 11.4088043 ,
        5.47866221,  5.41348121,  4.85603886,  7.39004226,  8.38252937,
        5.16205087,  8.20436764,  8.14433542, 13.95472331, 13.6105595 ,
       12.77402351,  9.79814997,  8.77913428, 13.70893492, 10.48721382,
        8.62631801, 12.70489563, 11.76830981,  9.75401449,  5.8828986 ,
        8.91551231, 11.57249984, 12.5085995 , 11.75618513,  9.15866607,
        7.28793663,  7.19513806, 11.83570132,  9.63742669,  8.83725081,
        8.4486771 , 12.19805394, 10.5575601 , 13.15525803, 11.10066126,
       13.97857591, 12.5085995 ,  7.05627461,  4.90505566,  5.81199422,
       11.63502679,  6.56833864,  5.43943791,  5.14371005, 10.37998078,
       11.00694688,  8.67227386,  5.10446653,  5.29755487,  5.52416368,
        5.95297459])}
{'P': tensor([[9.8200e-03, 1.5588e-01, 1.1673e-01,  ..., 4.2124e-06, 9.9327e-08,
         2.9033e-02],
        [3.0668e-05, 8.3890e-03, 5.3790e-04,  ..., 1.7186e-06, 9.7964e-08,
         1.2684e-01],
        [3.1163e-05, 2.5203e-02, 4.8055e-03,  ..., 1.1020e-06, 7.5621e-08,
         4.9840e-02],
        ...,
        [1.8522e-02, 4.2311e-03, 5.3739e-03,  ..., 6.9018e-02, 2.2813e-06,
         8.7900e-04],
        [2.5372e-05, 1.4438e-05, 2.3299e-05,  ..., 2.0426e-04, 5.3200e-01,
         8.6215e-06],
        [3.5470e-05, 1.0911e-04, 8.8523e-05,  ..., 4.7359e-06, 6.8761e-07,
         4.0368e-02]]), 'X': tensor([[-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0379, 0.0128,  ..., 0.2880, 1.2995, 0.1445],
        [0.0379, 0.0000, 0.0417,  ..., 0.2527, 1.2875, 0.1070],
        [0.0128, 0.0417, 0.0000,  ..., 0.2856, 1.2882, 0.1453],
        ...,
        [0.2880, 0.2527, 0.2856,  ..., 0.0000, 1.1390, 0.1508],
        [1.2995, 1.2875, 1.2882,  ..., 1.1390, 0.0000, 1.2424],
        [0.1445, 0.1070, 0.1453,  ..., 0.1508, 1.2424, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.00802627, 10.33958639, 10.27572673, 10.31584102, 10.412299  ,
       10.35464916, 10.00794468,  5.87132718, 13.92390689,  5.95111804,
        9.45995955,  7.19513806, 13.13205357,  5.18277858, 11.75438006,
        9.41244865,  5.69913817,  9.88310614, 11.65923396, 13.5247215 ,
        8.1303945 , 10.74409834, 13.88683436,  5.38342077,  9.55593374,
        5.0209005 , 13.50484539,  7.50564223,  6.42439344, 11.2845608 ,
       10.04916678, 10.31886066,  7.92200342,  7.53063885, 10.4606265 ,
       11.83570132,  5.19577146, 10.95593729, 10.67322527,  9.76896904,
        7.46248947, 11.28215429,  9.88571713, 14.02058621,  7.46053281,
       11.72749923, 10.46238054,  9.17016857,  9.41244865,  7.76457095,
        7.06169391, 13.35013374, 11.10066126,  5.23878371,  8.81930525,
       10.24557047,  9.45995955,  7.48311002,  5.05774295, 14.10045797,
        9.48969978,  5.8828986 ,  7.06272256, 12.12448912,  4.978704  ,
        9.05843201,  6.02588517,  6.99097013, 13.89974693,  5.5513072 ,
       11.75807252, 13.01728313, 13.31764181, 12.84056604, 13.86480175,
        5.57341623,  6.02588517,  5.81199422,  9.94285511, 12.59969936,
        9.75401449,  5.13182506,  5.92281142, 13.14690137,  9.41605684,
        7.92607342, 13.35927527, 12.1494023 ,  7.03733775,  7.72415048,
        7.18354885, 12.30695098,  5.87305005, 11.42947632, 11.75618513,
        6.61386436,  6.39520397, 10.33958639,  4.98461983, 10.1371552 ,
        7.06372172,  5.05774295, 11.27656488,  8.79284466,  7.53884104,
       10.69963954])}
{'P': tensor([[1.3158e-02, 2.2647e-01, 1.8625e-01,  ..., 9.5973e-06, 6.1139e-05,
         7.8314e-06],
        [1.2676e-04, 1.7868e-02, 4.3159e-02,  ..., 1.5702e-05, 1.7285e-04,
         1.2775e-05],
        [1.5450e-04, 2.5151e-03, 3.4662e-02,  ..., 5.3973e-05, 2.3564e-04,
         3.9661e-05],
        ...,
        [1.5344e-05, 9.9334e-06, 1.2800e-05,  ..., 5.7737e-02, 2.3137e-05,
         1.4931e-01],
        [8.1269e-05, 9.7734e-05, 7.3797e-05,  ..., 3.5831e-04, 2.5814e-01,
         7.1033e-04],
        [3.2251e-05, 1.9285e-05, 2.2757e-05,  ..., 4.6435e-03, 3.9794e-05,
         7.7998e-02]]), 'X': tensor([[ -8.3062,  -5.6008,   0.0287],
        [ -6.9838,  -7.4965,   0.9796],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -6.0844,  -8.3955,   0.7705],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -7.0253,  -7.4493,  -1.5978],
        [  5.8318,   3.5567,  -4.7692],
        [ -1.8367,   7.8718,  -8.6523],
        [ -1.5408,   7.8978,   5.3470],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.3523, -10.5232,   8.2138],
        [  5.8450,  -9.9356,   3.0767],
        [  2.0449,  13.8402,  -7.1642],
        [  3.6703,   6.3443,  -5.4401],
        [  0.5960, -11.0335,   3.1144],
        [  5.3858,   4.4260,  -9.6769],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.2687,  -2.9794,   0.6436],
        [  7.9816,  10.8938,  -0.5378],
        [  6.3200,   0.2690,   1.5900],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 11.2571,   6.7072,   8.3693],
        [  9.2979,   9.6048,   7.3138],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.2939,  -2.4784,  -4.5840],
        [  0.2480, -11.0153,  -3.0183],
        [  0.6142,  14.0802,   0.5407],
        [ -5.1377,   6.7751,  -0.4438],
        [ -9.1178,   1.5784,   2.1410],
        [  7.9141,  -8.6656,   9.6948],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.2435,   4.0076,   8.1257],
        [  8.8091,  -7.9037,   0.3631],
        [  0.7145,  -4.8071,  -2.3766],
        [ -8.9379,  -4.1655,  -2.8114],
        [  0.1252, -11.0062,   7.2564],
        [ 11.2517,  -4.6495,   2.9800],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.8318,   3.5567,  -4.7692],
        [ -8.7735,   2.5640,  -6.3313],
        [  1.5653,  13.9377,   4.1773],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -0.0410,   7.8591,   9.3457],
        [ -2.6069, -10.4298,  -6.1995],
        [ -8.0965,  -5.9712,   4.3466],
        [ -0.5709,   7.9055,   7.6089],
        [  4.1348,   5.9514,  -0.8856],
        [ 11.4743,   6.2478,  -2.1100],
        [  3.2701,   6.6338,   9.9209],
        [ -8.0750,  -6.0072,  -5.8267],
        [  4.5503,   5.5359,   7.4735],
        [  3.9022,  13.2906,   5.5788],
        [  1.9312,  13.8649,   8.3563],
        [  8.6271,  -8.0715,  -7.4813],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.1204,  -5.9308,   4.0761],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.2692,  13.5098,  -4.9032],
        [  6.0890,  -0.8382,   2.9166],
        [  6.9956,  11.6590,  -7.6876],
        [  0.3724,   7.7974,  -6.8722],
        [  2.7590,  13.6621,   9.4496],
        [  4.9531,   5.0576,   5.6794],
        [ -8.1426,   3.8048,   9.3542],
        [  2.0937,  13.8293,  -2.3533],
        [  5.3860,  -2.3418,   1.0826],
        [  5.3924, -10.1396,  -4.0863],
        [  4.0457,  -3.7737,  -0.4704],
        [  8.0381,  -8.5690,  -8.9523],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.8631,  -4.3728,  -1.1746],
        [  6.1575,  12.2046,   3.3257],
        [  2.4356,  -4.5942,  -5.0808],
        [  3.0876,   6.7527,  -5.3540],
        [ 10.2712,   8.3707,  -4.9743],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -6.5295,  -7.9785,  -7.4429],
        [  3.3943,   6.5483,   1.0790],
        [  6.2036,   2.4030,  -1.2500],
        [ -5.6651,   6.4380,   1.3660],
        [ 10.9599,  -5.1849,  -3.5526],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.8325,  -1.5222,  -2.1341],
        [ -7.4602,   4.7718,  -4.6277],
        [  0.6952,  -4.8060,   8.8976],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.6142,  14.0802,   0.5407],
        [  3.3069,   6.6089,   8.7377],
        [  2.1120,  -4.6818,   8.6144],
        [ -8.6050,   2.9451,  -0.7012],
        [ -1.6086,   7.8928,   4.3385],
        [  1.1644, -11.0394,   7.8511],
        [  6.3542,   0.6934,  -3.3840],
        [  5.8323,  -1.5228,   5.1132],
        [  7.4132,  11.3526,   0.3322],
        [ -9.3616,  -2.4749,   0.1793],
        [  3.9085,   6.1512,  -2.2560],
        [  5.8836,   3.4315,  -4.0327],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.8979,   3.3958,  -3.8713],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.9655,   3.2172,  -7.5953],
        [  6.3303,   1.6145,  -8.4083],
        [  5.1472,  -2.6801,   8.1194],
        [  6.0047,   3.1062,  -6.7914]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.8722e-02, 8.9515e-02,  ..., 1.0521e+00, 1.0985e+00,
         1.0198e+00],
        [7.8722e-02, 0.0000e+00, 1.6109e-01,  ..., 1.0738e+00, 1.0526e+00,
         1.0394e+00],
        [8.9515e-02, 1.6109e-01, 3.0339e-04,  ..., 9.8793e-01, 1.1080e+00,
         9.5759e-01],
        ...,
        [1.0521e+00, 1.0738e+00, 9.8793e-01,  ..., 0.0000e+00, 9.3174e-01,
         4.4153e-02],
        [1.0985e+00, 1.0526e+00, 1.1080e+00,  ..., 9.3174e-01, 0.0000e+00,
         8.9184e-01],
        [1.0198e+00, 1.0394e+00, 9.5759e-01,  ..., 4.4153e-02, 8.9184e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.01804223, 10.24557047, 10.32715715, 10.36844813, 10.33712258,
       10.23946247,  6.83083863,  8.08320235,  8.04664886,  5.04028538,
       11.38774149, 11.52733581, 13.99048094,  7.32950385, 11.04953767,
        6.97106912,  6.10774544,  9.73579738, 13.50484539,  6.32572323,
       12.27835996, 13.103709  , 13.36800073,  5.26067019,  5.84533345,
       11.018084  , 14.09357547,  8.50278896,  9.25336371, 11.73567759,
        9.66977802, 12.88270018, 11.83509372,  4.85995002,  9.86089579,
       11.00694688, 12.17451685,  9.36093562,  6.83083863,  9.14044984,
       14.02532758, 10.46739073,  7.85919687, 10.7506423 , 10.06023665,
        7.92607342,  7.24677664, 13.06498554,  7.39599986, 10.06437598,
        7.16598273, 13.85158417, 13.99877413, 11.81423161, 11.8159254 ,
       10.05558893, 10.41909211, 13.89974693,  6.1463896 , 13.59673406,
        7.80625981, 13.93790235,  7.07902036,  8.98765644, 13.98691426,
        5.87305005, 11.48434767,  5.53255882, 11.74901778,  5.54776934,
        9.88310614, 13.66991027,  5.19986797,  7.42512621, 13.25017211,
        9.70108409, 10.30972193,  7.37576154,  6.652742  ,  8.57560275,
       12.12448912,  5.57341623,  9.64157461,  6.02789634,  8.85572292,
        4.85603886, 10.5575601 , 14.09357547,  7.39004226,  5.13616224,
        9.09502385,  8.0550286 , 11.10066126,  6.39187285,  6.02779201,
       13.55868684,  9.68323686,  7.28793663,  6.81117802,  9.17905491,
        6.80559669, 12.27993098,  6.77777606,  6.53290663,  5.80312949,
        6.76056625])}
{'P': tensor([[8.3148e-03, 2.3957e-01, 1.7336e-01,  ..., 2.3555e-04, 1.4457e-04,
         4.8081e-06],
        [2.9766e-04, 1.1744e-02, 2.1659e-01,  ..., 3.9812e-04, 2.4159e-04,
         6.7478e-06],
        [1.0895e-04, 1.4695e-04, 1.6004e-02,  ..., 8.2167e-04, 6.2655e-04,
         1.6844e-05],
        ...,
        [9.5912e-07, 1.0334e-06, 1.3793e-06,  ..., 1.4962e-02, 1.0318e-01,
         5.4341e-06],
        [1.8882e-06, 1.9764e-06, 2.3869e-06,  ..., 2.3921e-04, 4.0327e-02,
         4.9025e-05],
        [1.8281e-05, 1.7060e-05, 1.9186e-05,  ..., 5.0930e-06, 1.7328e-05,
         1.6163e-01]]), 'X': tensor([[ -8.1531,  -5.8748,   2.6934],
        [ -7.9925,  -6.1423,   2.8990],
        [ -6.8466,  -7.6488,   1.9648],
        [ -6.8618,  -7.6322,   3.7027],
        [ -5.9225,  -8.5355,   2.4778],
        [ -8.0534,  -6.0429,   2.4583],
        [  2.8890,  -4.4316,   1.1600],
        [ 10.7566,   7.6212,  -2.3279],
        [  4.2275,  -3.6320,  -0.2714],
        [  9.0051,   9.9234,   5.4411],
        [  7.7750,  11.0665,   2.8364],
        [  4.0591,   6.0202,  -1.6339],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.5128, -10.9365,   9.9645],
        [  7.8425,  -8.7202,   6.4530],
        [  5.1861,   4.7361,  -3.7312],
        [  5.0277,   4.9589,   7.0175],
        [ -8.3900,   3.3727,   3.1650],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.5996,   1.3971,  -1.0694],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4671,  -0.5553,   0.1117],
        [  6.3534,   1.2990,  -7.2942],
        [  4.6992,   5.3690,   3.0935],
        [  3.6817,   6.3354,   6.1233],
        [  8.9284,  10.0036,  -6.9210],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.5294,  -2.1106,   9.0626],
        [  5.3804,  -2.3504,   7.2744],
        [  4.4290,  -3.4596,   1.4181],
        [  6.2804,   2.0019,   0.8607],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -4.4520,   7.1370,  -8.7867],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.1752,   2.5242,   3.5874],
        [  4.9152,  12.8653,  -1.4459],
        [  0.6363, -11.0348,   6.1414],
        [  3.4140,  -4.1803,  -8.8224],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.9026,   3.3837,  -0.6573],
        [ 11.1420,   6.9331,  -2.4779],
        [  7.8425,  -8.7202,   6.4530],
        [ -8.8540,  -4.3971,  -5.2144],
        [  8.7131,  10.2216,   7.7915],
        [  2.2644,  -4.6434,   9.7508],
        [  6.3211,   1.7017,  -0.2164],
        [  4.9957,  -2.8705,   6.5978],
        [  5.3602,  -2.3808,   7.2098],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.0604,   1.7707,   1.1066],
        [  1.4489,   7.5256,   4.4529],
        [  0.1206,  -4.7363,   8.7103],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.0419,  -4.6978,  -5.0983],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.5240,  -2.1198,   6.2055],
        [  0.1931,  -4.7491,   7.4385],
        [  7.1821,  -9.1845,   9.8548],
        [  6.3534,   1.2990,  -7.2942],
        [ -3.1204, -10.2345,   9.2288],
        [ -3.4100,   7.5436,  -7.6980],
        [ -1.0482, -10.8487,  -5.9066],
        [  1.8721,  -4.7322,   0.5524],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.4336,  -2.2676,   5.6933],
        [  6.3583,   1.1729,   8.7835],
        [  2.5620,   7.0549,   8.4825],
        [  8.4691,  10.4571,  -5.0650],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 12.3759,  -1.3943,   6.4659],
        [ -5.0943,  -9.1682,   0.5073],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -2.1533, -10.5778,   1.0112],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.1624,  -3.4286,   2.0647],
        [ 10.0567,   8.6697,   4.6329],
        [  6.2804,   2.0019,   0.8607],
        [  0.7958,  -4.8109,  -8.2248],
        [  0.7145,  -4.8071,  -2.3766],
        [  3.3744,   6.5623,  -6.3636],
        [  2.5813,  -4.5471,   5.3597],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.1497,   7.8334,  -5.4412],
        [  5.3118,   4.5452,   4.2114],
        [ 11.7795,  -3.4697,   3.5624],
        [ -9.4624,  -0.4633,  -7.3918],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.6275, -10.7221,   3.2376],
        [ -8.0588,  -6.0341,  -2.3006],
        [  4.5430,   5.5438,  -3.8669],
        [ -8.0330,  -6.0765,   6.7945],
        [  8.6271,  -8.0715,  -7.4813],
        [ -3.2053, -10.1993,  -3.8736],
        [ -5.7347,  -8.6908,   9.2878],
        [ -5.9938,  -8.4745,   3.9735],
        [ 12.5267,  -0.4125,  -8.7331],
        [  5.9587,  12.3216,   9.1695],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.1426,   3.8048,   9.3542],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.1519, -10.2387,   9.8191],
        [  7.5930,  -8.9037,   7.2323],
        [  6.2535,  -0.1709,  -9.4102]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0150, 0.0528,  ..., 0.7204, 0.7358, 1.1622],
        [0.0150, 0.0000, 0.0658,  ..., 0.7070, 0.7234, 1.1651],
        [0.0528, 0.0658, 0.0000,  ..., 0.7485, 0.7574, 1.1256],
        ...,
        [0.7204, 0.7070, 0.7485,  ..., 0.0000, 0.1038, 1.1995],
        [0.7358, 0.7234, 0.7574,  ..., 0.1038, 0.0000, 1.1062],
        [1.1622, 1.1651, 1.1256,  ..., 1.1995, 1.1062, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.04916678, 10.08001894, 10.26545792, 10.26327895, 10.38897563,
       10.06850677,  5.29009836, 13.18278606,  5.57341623, 13.40024224,
       13.5247215 ,  7.26074925,  9.78286733, 11.22142195, 11.72802491,
        7.02326515,  7.06169391,  9.04255655, 10.1371552 , 12.67680439,
        9.79039371,  9.48337063,  6.48486171,  7.13501264,  7.32754882,
       13.40850328, 12.25806559,  5.91854397,  5.87132718,  5.62005629,
        6.59176018, 12.35625694,  8.41171369, 12.33167382,  6.67121748,
       13.77222752, 11.05317597,  5.39723241,  7.13501264,  9.562174  ,
        6.80370254, 13.12299145, 11.72802491,  9.88571713, 13.43127131,
        5.16611782,  6.54615887,  5.76165934,  5.86519525,  7.09887557,
        9.23178204,  7.66377554,  4.73784222, 12.16901615,  6.19633139,
        5.1224079 , 10.31584102,  5.91677188,  4.75303127, 11.65923396,
        6.48486171, 10.69963954,  8.27853076, 10.8992524 ,  5.08910491,
       13.04217487,  5.88782913,  6.46560181,  7.50564223, 13.45642379,
        9.562174  , 12.45418236, 10.48840476, 12.11735334, 10.794755  ,
        9.09661212,  9.78285424, 13.2778378 ,  6.59176018,  4.8763135 ,
        4.85995002,  7.37902766,  5.2287058 ,  5.01114719,  7.83486943,
        6.99097013, 12.27991991,  9.47369628,  4.75303127, 10.84493732,
       10.0674887 ,  7.16747878, 10.07238463, 11.81423161, 10.6910768 ,
       10.412299  , 10.37998078, 12.53345445, 13.68672855,  5.43544626,
        9.61942445,  8.98765644, 10.53368644, 11.46176259, 11.70168755,
        6.25586297])}
{'P': tensor([[8.4528e-03, 2.2058e-01, 1.4233e-01,  ..., 6.3410e-08, 8.7908e-08,
         4.8453e-02],
        [5.5822e-05, 8.8841e-03, 3.5535e-01,  ..., 8.9394e-08, 9.8735e-08,
         4.5207e-03],
        [3.1595e-05, 7.6397e-05, 8.6698e-03,  ..., 9.9294e-08, 1.0151e-07,
         3.6163e-04],
        ...,
        [1.6117e-06, 9.0497e-07, 6.1526e-07,  ..., 3.8444e-02, 5.0191e-02,
         1.2227e-06],
        [2.7757e-05, 1.2298e-05, 7.5824e-06,  ..., 5.9405e-05, 7.0973e-02,
         1.4200e-05],
        [4.1516e-04, 1.3582e-02, 1.0042e-01,  ..., 4.0009e-07, 3.7248e-07,
         2.9830e-02]]), 'X': tensor([[-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 2.5425e-02, 3.1383e-03,  ..., 1.4757e+00, 1.1791e+00,
         1.5901e-01],
        [2.5425e-02, 0.0000e+00, 2.8526e-02,  ..., 1.4695e+00, 1.1795e+00,
         1.3513e-01],
        [3.1383e-03, 2.8526e-02, 0.0000e+00,  ..., 1.4760e+00, 1.1787e+00,
         1.6181e-01],
        ...,
        [1.4757e+00, 1.4695e+00, 1.4760e+00,  ..., 0.0000e+00, 4.4768e-01,
         1.3955e+00],
        [1.1791e+00, 1.1795e+00, 1.1787e+00,  ..., 4.4768e-01, 0.0000e+00,
         1.1391e+00],
        [1.5901e-01, 1.3513e-01, 1.6181e-01,  ..., 1.3955e+00, 1.1391e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.05558893, 10.26327895, 10.37998078, 10.19036081, 10.41909211,
       10.35093485,  6.65259775,  9.5014146 , 10.92597196,  5.35774573,
        4.91336994,  8.08320235,  5.81199422, 13.2430377 , 11.33837396,
        6.84796202, 11.80092578, 11.48434767, 12.19805394,  7.83534987,
        9.15253344, 11.87412439, 13.0743399 , 10.68031351,  5.21280156,
        7.37576154,  6.31576264,  6.81084724, 13.39372426,  8.09842704,
        6.23262289,  6.77777606, 12.86648771,  6.48486171, 10.08001894,
        7.28793663,  5.47866221,  7.39599986,  8.74470218,  8.68832398,
        5.10446653,  7.92200342,  5.13125513, 12.1494023 , 13.89886469,
       12.24467041,  9.86089579, 12.86648771, 12.3344536 ,  5.05774295,
       11.73397309,  7.92607342,  6.65368551, 10.06703358,  8.09842704,
        7.44805901, 13.89974693, 13.97779839,  5.5513072 ,  9.50654967,
       13.20597932,  7.48538905, 10.4606265 ,  8.22728399,  8.62162372,
       10.26545792,  8.83341608,  5.80312949,  9.68323686,  5.94541803,
       10.6910768 ,  8.67502348, 11.81423161,  6.34059021, 12.21532282,
        5.80860165,  8.29203356, 13.58333857, 11.95743558,  5.91854397,
        9.27474386,  6.81117802,  7.18354885, 11.63765273,  7.85919687,
       14.02526505, 11.85890664,  5.81481044,  8.859416  ,  8.06990483,
       10.61472636,  6.67108662, 10.06286074, 11.46176259, 11.63502679,
       11.31276109,  8.97052115, 10.51340829,  8.50268322,  5.79543038,
       14.03505624,  7.44805901,  6.24187044, 13.59673406, 13.97857591,
       10.26545792])}
{'P': tensor([[9.7531e-03, 2.3058e-01, 2.2292e-01,  ..., 1.1112e-07, 1.9638e-05,
         2.3747e-06],
        [1.3873e-04, 1.1379e-02, 5.0943e-02,  ..., 1.3249e-07, 2.8068e-05,
         3.2001e-06],
        [7.5030e-05, 6.6955e-04, 1.0623e-02,  ..., 1.1704e-07, 5.2197e-05,
         2.6991e-06],
        ...,
        [1.3938e-04, 8.2991e-05, 5.5339e-05,  ..., 2.3602e-01, 2.4649e-07,
         1.2241e-03],
        [2.2512e-07, 2.3283e-07, 3.2855e-07,  ..., 5.9102e-07, 1.3855e-02,
         2.4980e-06],
        [5.1063e-05, 3.9065e-05, 2.7041e-05,  ..., 1.1052e-02, 4.2760e-07,
         4.1922e-02]]), 'X': tensor([[-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0529, 0.0457,  ..., 1.2255, 1.1307, 0.8036],
        [0.0529, 0.0000, 0.0972,  ..., 1.2155, 1.1757, 0.7971],
        [0.0457, 0.0972, 0.0000,  ..., 1.2499, 1.1030, 0.8261],
        ...,
        [1.2255, 1.2155, 1.2499,  ..., 0.0000, 1.3491, 0.4257],
        [1.1307, 1.1757, 1.1030,  ..., 1.3491, 0.0000, 1.0946],
        [0.8036, 0.7971, 0.8261,  ..., 0.4257, 1.0946, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.06023665, 10.19036081, 10.26327895, 10.37998078, 10.41909211,
       10.35093485,  7.18354885, 12.87752052, 12.91505111,  8.40445872,
       10.80728788,  5.2287058 , 10.65929968, 11.018084  ,  6.38399789,
        5.88782913,  5.14607014, 13.59170809,  9.50913849, 12.13568598,
       12.4980845 ,  7.19534146,  7.5158489 ,  8.49161154, 12.64235359,
       13.15525803,  8.04483495,  6.46560181, 13.32613445, 11.54071859,
       13.92390689,  5.05963217, 11.86239961,  8.01009083, 11.4890861 ,
        5.36061974,  9.85330646,  6.1561646 ,  5.0209005 ,  5.88782913,
       11.87412439,  7.17789666, 13.91198191,  5.1805345 , 13.6829843 ,
        8.81930525,  9.0368458 ,  6.02789634, 13.34587027,  9.0368458 ,
       13.52430176,  8.14433542,  6.10360073,  5.83560009,  6.25586297,
       14.02526505,  4.74770899, 11.83570132, 13.35013374, 13.26590071,
       11.92133129,  7.23276582,  5.79562691,  8.77913428,  7.09887557,
        5.07306274,  8.41171369, 11.1341798 ,  7.17789666, 11.35345345,
       10.04916678,  7.83486943,  6.32572323,  7.52117132,  8.30729352,
        7.46053281,  5.64865173,  9.18358562, 10.69963954,  7.48538905,
        8.37954853,  6.4020404 , 12.53285633, 11.51557166,  5.37343173,
        7.81509631,  4.74770899,  5.91677188,  8.94395628,  9.09502385,
        7.19513806,  7.92200342,  7.62232307,  7.33350642,  8.0550286 ,
       11.75807252, 10.63038565, 13.50484539,  5.56809335,  9.75200848,
        8.06990483,  9.27474386, 10.08001894, 14.0393949 , 11.75618513,
        7.39599986])}
{'P': tensor([[7.0643e-03, 3.3066e-01, 2.8058e-01,  ..., 6.2848e-04, 1.6541e-05,
         2.5465e-06],
        [9.7121e-05, 8.8800e-03, 4.0551e-01,  ..., 9.4045e-04, 2.1596e-05,
         4.0602e-06],
        [1.2483e-04, 2.0605e-04, 1.4771e-02,  ..., 1.6658e-03, 4.0331e-05,
         8.3094e-06],
        ...,
        [1.7603e-04, 1.9653e-04, 2.0903e-04,  ..., 1.0452e-01, 9.3118e-05,
         4.4511e-04],
        [6.3519e-05, 6.6360e-05, 7.3315e-05,  ..., 2.5538e-04, 3.2990e-01,
         2.0045e-02],
        [3.7357e-07, 3.4976e-07, 3.4947e-07,  ..., 6.8184e-06, 3.3287e-05,
         2.7094e-02]]), 'X': tensor([[ -8.0829,  -5.9940,   5.0564],
        [ -7.6525,  -6.6534,   5.1721],
        [ -7.3470,  -7.0615,   5.0066],
        [ -6.2200,  -8.2737,   5.5079],
        [ -6.4620,  -8.0449,   6.2719],
        [ -5.9938,  -8.4745,   3.9735],
        [  0.4996,   7.7738,  -4.4711],
        [  6.1973,  -0.4372,   6.5720],
        [ -6.4464,   5.8272,   7.5038],
        [ 12.4690,  -0.8528,  -9.3374],
        [  7.0358,  -9.2784,  -9.8479],
        [ -3.5520,   7.4976,   9.1522],
        [  5.0277,   4.9589,   7.0175],
        [  5.5626,  -2.0536,   8.4433],
        [ 12.4877,  -0.7222,   8.8347],
        [  4.5325,  -3.3643,   6.7699],
        [ -2.1817,   7.8274,  -4.6298],
        [  0.7145,  -4.8071,  -2.3766],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.2038,   2.4020,  -3.1692],
        [  4.5430,   5.5438,  -3.8669],
        [ -9.4541,  -0.3327,   8.8246],
        [  5.1861,   4.7361,  -3.7312],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.6229,   2.9068,   2.4429],
        [ 11.7795,  -3.4697,   3.5624],
        [ -3.0004,   7.6607,  -7.2640],
        [  8.1043,  -8.5164,  -7.9290],
        [ -3.9456,  -9.8542,   0.7466],
        [  2.2849, -10.9655,  -7.6542],
        [  8.7131,  10.2216,   7.7915],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.7485,  -9.9811,  -3.2355],
        [  0.2435,   7.8191,  -6.4709],
        [ -0.1626, -10.9796,   6.4629],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.2038,   2.4020,  -3.1692],
        [  3.9895,   6.0816,  -6.5235],
        [  6.2026,  12.1774,  -1.5887],
        [  0.6363, -11.0348,   6.1414],
        [  8.5716,  10.3596,   6.8818],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 10.9931,   7.2107,   7.7933],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 10.5678,   7.9261,  -0.6100],
        [  8.8474,  10.0868,  -8.8644],
        [ 11.9808,   4.9340,  -3.3715],
        [ -5.0943,  -9.1682,   0.5073],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.4242, -10.1261,   3.9595],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.8745,   2.3097,   3.5600],
        [  0.4119,  14.1019,  -4.1870],
        [  1.1350,  14.0104,   0.6321],
        [ -9.3384,  -2.6161,  -4.7630],
        [  3.2159, -10.8166,  -1.0406],
        [  5.1886,  -2.6250,   6.4677],
        [ -7.4887,   4.7364,   8.9074],
        [ -9.3810,  -2.3455,  -9.5194],
        [  4.1637,  13.1898,  -1.9189],
        [  5.1203,  -2.7151,   6.2110],
        [  6.3266,   0.3292,   9.3502],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.3534,   1.2990,  -7.2942],
        [  9.3753,   9.5171,   3.9131],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.8902,  -1.3887,   4.0388],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -3.4100,   7.5436,  -7.6980],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.4117,   0.1160,   8.2574],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3236,   0.3011,  -4.6668],
        [ 11.6165,  -3.8715,   6.4594],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.7565,  11.8237,  -4.9882],
        [ -3.2053, -10.1993,  -3.8736],
        [ -4.4503,   7.1378,   3.3751],
        [ -9.3739,   0.3982,  -4.2564],
        [ 12.5717,   1.9266,   7.4903],
        [  0.7958,  -4.8109,  -8.2248],
        [ -8.0965,  -5.9712,   4.3466],
        [ 11.1085,  -4.9204,   0.0736],
        [  3.2647, -10.8066,  -0.3768],
        [  6.3246,   1.6696,  -0.1544],
        [  2.2437,  -4.6489,   6.1430],
        [  3.7298,  13.3537,  -9.4653],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.4996,   2.6335,  -6.3166],
        [  0.9945,  -4.8144,  -0.1897],
        [  8.0877,  -8.5297,  -1.1050],
        [ 11.5935,   5.9740,  -4.3952],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -1.0012, -10.8576,   2.2867],
        [  4.7544,  -3.1425,  -1.5315],
        [ -3.5173,   7.5091,  -8.9585],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.2026,  12.1774,  -1.5887],
        [  6.1929,   2.4501,  -6.9352],
        [  1.8447,  -4.7372,   8.1235],
        [  2.3889,  -4.6082,  -8.7291],
        [ 12.2921,  -1.7902,  -4.0870]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0094, 0.0025,  ..., 0.8940, 1.1907, 1.2357],
        [0.0094, 0.0000, 0.0098,  ..., 0.8969, 1.1993, 1.2440],
        [0.0025, 0.0098, 0.0000,  ..., 0.8964, 1.1916, 1.2369],
        ...,
        [0.8940, 0.8969, 0.8964,  ..., 0.0000, 0.9098, 0.8617],
        [1.1907, 1.1993, 1.1916,  ..., 0.9098, 0.0000, 0.1160],
        [1.2357, 1.2440, 1.2369,  ..., 0.8617, 0.1160, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.06286074, 10.14044489, 10.19036081, 10.35093485, 10.31886066,
       10.37998078,  7.78980299,  6.21275016,  8.68979052, 12.4980845 ,
       11.64436684,  8.29640613,  7.06169391,  5.92952538, 12.5085995 ,
        5.64464807,  8.12580976,  4.85995002,  9.50654967,  6.65259775,
        7.16747878,  9.45995955,  7.02326515, 12.71843462,  9.09963894,
       12.27991991,  8.22728399, 11.75618513, 10.61472636, 11.20100795,
       13.43127131, 10.14436912, 12.27993098, 11.51811603,  7.822854  ,
       10.98076873, 12.33167382,  6.65259775,  7.27340834, 13.6660671 ,
       11.05317597, 13.44593519,  9.94285511, 13.14690137, 10.36844813,
        7.72668467,  9.75401449, 13.20988996, 13.41713661, 12.95702647,
       10.48840476,  9.48969978, 11.48734978, 13.41713661,  9.17016857,
       14.10796466, 14.05633441,  9.69792513, 11.2845608 ,  5.81481044,
        8.86082124,  9.66977802, 13.83139024,  5.79562691,  6.33517349,
       10.46739073,  6.48486171, 13.35927527, 12.35625694,  9.53015376,
        6.05165248,  9.69629   ,  8.27853076, 13.2430377 ,  9.41244865,
        6.16512856,  6.33075682, 12.24467041,  8.30125201, 13.61801526,
       10.6910768 ,  8.41148722,  9.38232121, 12.71843462,  4.8763135 ,
       10.06023665, 12.1494023 , 11.28895489,  6.54127431,  5.16205087,
       13.86480175,  4.83371341, 12.77402351,  4.9160962 , 11.75438006,
       13.04217487,  9.48969978, 10.9036226 ,  5.69913817,  8.29203356,
       10.33712258, 13.6660671 ,  6.65991877,  5.08372149,  5.19065439,
       12.4217523 ])}
{'P': tensor([[1.0246e-02, 3.9195e-01, 1.9691e-01,  ..., 1.4409e-06, 1.4386e-06,
         2.0507e-06],
        [1.1693e-04, 1.4118e-02, 2.6781e-01,  ..., 2.0199e-06, 3.0835e-06,
         2.6246e-06],
        [7.5134e-05, 2.0971e-04, 1.3655e-02,  ..., 1.7742e-06, 3.5650e-06,
         2.5675e-06],
        ...,
        [9.6517e-05, 6.6345e-05, 4.0518e-05,  ..., 6.1994e-02, 8.9109e-06,
         1.3521e-03],
        [7.2134e-07, 6.6671e-07, 5.4929e-07,  ..., 3.0192e-03, 8.3374e-02,
         6.4927e-05],
        [1.0501e-03, 6.4244e-04, 4.5152e-04,  ..., 9.0593e-04, 2.5273e-06,
         1.5011e-01]]), 'X': tensor([[ -8.0766,  -6.0046,  -3.1873],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -7.0253,  -7.4493,  -1.5978],
        [  3.3633,  13.4794,   0.5193],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.8407,  11.7666,   3.0045],
        [ -3.2486,   7.5925,  -1.3719],
        [ -0.5385,   7.9037,   6.3830],
        [  5.6248,   3.9984,   8.4933],
        [  2.4572,   7.1085,   7.7376],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.7295,  13.6703,  -8.9328],
        [  5.2199,   4.6861,  -1.9304],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.0864,   7.0387,  -0.1238],
        [ -1.2454,   7.9127,  -0.5715],
        [  9.3607,   9.5337,  -4.5412],
        [ -8.4573,   3.2448,   5.4370],
        [ 12.6059,   0.9597,  -6.3264],
        [  7.3592,  11.3937,   6.2407],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -7.3446,   4.9113,  -5.9356],
        [  6.0560,  -0.9421,  -5.0334],
        [  2.5318,  13.7231,   6.3692],
        [ -5.0132,  -9.2234,  -8.3531],
        [  2.3367,   7.1676,  -9.5263],
        [ -3.4986,   7.5152,   9.0712],
        [ -6.8618,  -7.6322,   3.7027],
        [ -9.3107,  -2.7697,  -3.2919],
        [  2.8890,  -4.4316,   1.1600],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.6681,  -4.8556,   9.2518],
        [  0.5600, -11.0321,  -0.8660],
        [  2.0419,  -4.6978,  -5.0983],
        [ 12.2328,  -2.0376,  -1.1234],
        [  0.8572,  14.0501,  -8.3364],
        [  2.2437,  -4.6489,   6.1430],
        [ 12.6000,   1.3839,  -3.2992],
        [  7.0358,  -9.2784,  -9.8479],
        [  2.5382,   7.0672,   4.5097],
        [ -3.4986,   7.5152,   9.0712],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -7.6650,   4.5092,   0.2520],
        [ 11.1420,   6.9331,  -2.4779],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -3.1671,   7.6158,   6.0765],
        [ -9.1178,   1.5784,   2.1410],
        [ 12.6038,   0.7665,  -3.4439],
        [ -7.7239,   4.4298,   3.1216],
        [ -3.5173,   7.5091,  -8.9585],
        [  4.8653,   5.1694,  -4.0823],
        [  3.8182,  -3.9347,  -1.7961],
        [ -8.9942,  -3.9990,   6.0448],
        [  2.5375,   7.0676,  -4.3490],
        [ -5.1046,  -9.1610,   2.3418],
        [  9.3560,   9.5391,   9.4943],
        [  9.3753,   9.5171,   3.9131],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -9.4541,  -0.3327,   8.8246],
        [  0.9392,  -4.8143,  -3.4649],
        [  2.5813,  -4.5471,   5.3597],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.2188,  -2.5839,   2.1833],
        [  4.4039,   5.6903,   8.7366],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.8093,  -9.4178,   8.4244],
        [  6.9693,  -9.3200,  -8.3822],
        [  3.4009,  -4.1874,  -5.9751],
        [  3.1892, -10.8220,  -6.8409],
        [ -1.5408,   7.8978,   5.3470],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.3860,  -2.3418,   1.0826],
        [  6.9956,  11.6590,  -7.6876],
        [  6.9956,  11.6590,  -7.6876],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.2022,   2.4092,   3.0349],
        [  0.9478,  -4.8144,   0.6371],
        [ -0.9715, -10.8630,   0.3555],
        [  2.2644,  -4.6434,   9.7508],
        [  4.9531,   5.0576,   5.6794],
        [ -9.3616,  -2.4749,   0.1793],
        [ 11.0864,   7.0387,  -0.1238],
        [ 11.7795,  -3.4697,   3.5624],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.3860,  -2.3418,   1.0826],
        [ 11.9808,   4.9340,  -3.3715],
        [ -8.3062,  -5.6008,   0.0287],
        [  2.7295,  13.6703,  -8.9328],
        [ -9.2687,  -2.9794,   0.6436],
        [ 12.4996,   2.6335,  -6.3166],
        [ -2.0229, -10.6163,   8.6930],
        [ 11.4495,   6.3026,   6.8257],
        [ 12.2589,  -1.9312,   9.8657],
        [ -3.2486,   7.5925,  -1.3719],
        [ -7.3333,   4.9246,  -7.9212],
        [ -9.4378,  -1.8606,  -2.5668],
        [  6.1372,   2.6726,  -6.7885],
        [ -6.7992,   5.4991,   9.9186],
        [  1.4489,   7.5256,   4.4529],
        [ 12.4877,  -0.7222,   8.8347],
        [ -0.6323,   7.9085,  -5.9544]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0138, 0.0531,  ..., 1.0130, 1.4853, 0.7148],
        [0.0138, 0.0000, 0.0513,  ..., 1.0017, 1.4742, 0.7014],
        [0.0531, 0.0513, 0.0000,  ..., 1.0316, 1.5050, 0.6917],
        ...,
        [1.0130, 1.0017, 1.0316,  ..., 0.0000, 0.4738, 0.7381],
        [1.4853, 1.4742, 1.5050,  ..., 0.4738, 0.0000, 1.1443],
        [0.7148, 0.7014, 0.6917,  ..., 0.7381, 1.1443, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.06407839, 10.1979669 , 10.31613269, 10.39942497, 10.52357479,
       10.23946247, 13.89264822,  9.53015376, 13.6105595 ,  8.25828409,
        7.92200342,  6.90115901,  7.52117132, 10.08001894, 13.94008948,
        7.01475376,  7.02326515, 13.13205357,  8.01009083, 13.36092624,
        9.05843201, 12.64235359, 13.56367857, 12.3828957 ,  8.8353674 ,
        6.12886024, 13.95472331, 10.4977132 ,  7.53884104,  8.28968067,
       10.26327895,  9.71392179,  5.29009836,  9.3349858 ,  9.9353921 ,
       11.04629471,  5.1224079 , 12.4013153 , 14.07623266,  5.16205087,
       12.67576792, 11.64436684,  7.50917551,  8.28968067, 12.12641969,
        8.89301516, 13.12299145, 10.35485331,  8.24808832,  9.25336371,
       12.62711252,  8.90405823,  8.29203356,  7.09887557,  5.48276038,
        9.84314712,  7.5092834 , 10.48721382, 13.3614598 , 13.35927527,
        9.43112957, 10.4977132 ,  9.45995955,  4.90505566,  5.2287058 ,
        9.66977802, 12.46513646,  5.82345867,  7.19534146, 12.11735334,
       11.62158554, 11.63765273,  5.39450305, 11.28215429,  8.04664886,
        5.90035324,  5.87305005, 13.59673406, 13.59673406,  5.52416368,
        6.65368551,  4.90676242, 10.90638066,  5.16611782,  7.07902036,
        9.68323686, 13.13205357, 12.27991991, 12.84056604,  5.87305005,
       12.95702647, 10.01804223, 13.94008948,  9.73579738, 12.77402351,
       10.80728788, 13.06957425, 12.41012046,  8.25828409,  8.83341608,
        9.61942445,  6.69389431,  8.74470218,  7.66377554, 12.5085995 ,
        7.9337647 ])}
{'P': tensor([[1.1054e-02, 3.9731e-01, 1.1051e-01,  ..., 9.9198e-06, 3.0545e-03,
         4.7961e-07],
        [2.1405e-04, 1.5989e-02, 9.1114e-02,  ..., 1.5881e-05, 6.6756e-03,
         8.6023e-07],
        [3.2229e-04, 8.0067e-04, 3.9579e-02,  ..., 5.6641e-05, 1.3122e-02,
         2.4128e-06],
        ...,
        [5.4281e-05, 4.5799e-05, 3.7096e-05,  ..., 8.1437e-02, 1.2100e-05,
         6.4259e-04],
        [2.5725e-05, 3.2759e-05, 5.2367e-05,  ..., 4.2678e-04, 1.0150e-01,
         2.1416e-04],
        [2.7501e-06, 2.3542e-06, 1.2868e-06,  ..., 7.1471e-04, 1.0630e-06,
         9.2882e-02]]), 'X': tensor([[ -8.0750,  -6.0072,  -5.8267],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -4.5778,  -9.5013,  -5.0667],
        [  6.3124,   0.2057,  -0.4593],
        [  4.4915,   5.5990,   7.0405],
        [  5.8979,   3.3958,  -3.8713],
        [  9.8082,  -6.8396,   4.9615],
        [ -9.4203,  -2.0340,   2.5856],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 12.1940,  -2.1881,   7.1499],
        [ -2.1817,   7.8274,  -4.6298],
        [ -7.5603,   4.6459,   8.2002],
        [  6.3119,   1.7797,   9.9186],
        [  5.5158,  -2.1337,   3.5246],
        [ 10.7566,   7.6212,  -2.3279],
        [  9.6662,   9.1725,   8.1056],
        [ -4.4520,   7.1370,  -8.7867],
        [ -1.1885, -10.8211,   0.0865],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.7482, -10.3907,   5.9149],
        [  2.5636, -10.9293,   9.6519],
        [  5.5294,  -2.1106,   9.0626],
        [ -3.2736, -10.1703,   3.8712],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.4501,  -1.7133,   2.2980],
        [  0.2571,  -4.7594,   1.5618],
        [  5.5158,  -2.1337,   3.5246],
        [ 12.4901,  -0.7052,   1.7315],
        [ -6.5295,  -7.9785,  -7.4429],
        [  1.6308,  13.9254,  -2.0984],
        [ -1.1885, -10.8211,   0.0865],
        [ -7.8280,   4.2848,  -7.1955],
        [ 10.7566,   7.6212,  -2.3279],
        [  9.1620,  -7.5573,  -9.9882],
        [  2.3373,  13.7721,  -0.2442],
        [ -8.0611,  -6.0302,   4.6962],
        [ -5.9938,  -8.4745,   3.9735],
        [  5.6248,   3.9984,   8.4933],
        [  2.4149,  -4.6005,  -4.3339],
        [  5.6202,  -1.9508,   0.7471],
        [ -1.3178, -10.7939,   0.4236],
        [  5.6672,  -1.8631,   8.1020],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.6093,  -1.9705,   6.4807],
        [ 11.4703,  -4.2006,   7.4086],
        [  1.7500,   7.4187,   4.4237],
        [  4.2275,  -3.6320,  -0.2714],
        [ -5.1398,  -9.1367,   6.2074],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.8704,  -9.3809,   7.5769],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.2770,  -0.0379,   2.4195],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3087,   1.8053,  -1.9481],
        [ -1.6334, -10.7207,   7.1397],
        [ 12.5650,   2.0132,   4.6272],
        [ -7.7239,   4.4298,   3.1216],
        [ -2.3317,   7.8034,  -3.6274],
        [  1.5246,  -4.7838,  -5.2095],
        [  1.5246,  -4.7838,  -5.2095],
        [  6.1685,  12.1979,  -6.2473],
        [  2.3376,  -4.6231,   9.8436],
        [  5.2188,  -2.5839,   2.1833],
        [ 12.6035,   0.7494,   7.1536],
        [  5.3924, -10.1396,  -4.0863],
        [  5.0506,   4.9277,  -9.5094],
        [  6.3604,   0.9042,  -6.5901],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.1459,   1.4780,  -4.2403],
        [ -1.6220,   7.8917,   0.0315],
        [ -9.3739,   0.3982,  -4.2564],
        [  0.2480, -11.0153,  -3.0183],
        [  8.4691,  10.4571,  -5.0650],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.3413,   0.4883,  -6.8416],
        [  6.3211,   1.7017,  -0.2164],
        [  3.3744,   6.5623,  -6.3636],
        [  2.2373,  -4.6506,  -1.7334],
        [ -0.1963,   7.8765,  -1.1371],
        [  5.8979,   3.3958,  -3.8713],
        [  2.7976,  -4.4682,  -4.2020],
        [ -3.5520,   7.4976,   9.1522],
        [ -7.8359,   4.2735,   7.2943],
        [ -9.2687,  -2.9794,   0.6436],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.0876,   6.7527,  -5.3540],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.2571,   6.7072,   8.3693],
        [ 12.5650,   2.0132,   4.6272],
        [ -3.3812, -10.1235,   7.4384],
        [ -7.5720,   4.6310,   2.1717],
        [ -9.1231,  -3.5739,   8.4708],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.2036,   2.4030,  -1.2500],
        [  3.3094,  13.4969,  -3.6507],
        [  2.7121,   6.9743,  -8.4202],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.8201,   3.5841,  -8.0609],
        [  0.8940, -11.0402,  -0.9783],
        [ 11.1469,   6.9237,   1.6160]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0218, 0.0923,  ..., 0.9862, 0.4944, 1.3741],
        [0.0218, 0.0000, 0.1009,  ..., 0.9712, 0.4727, 1.3539],
        [0.0923, 0.1009, 0.0000,  ..., 0.9452, 0.5213, 1.3707],
        ...,
        [0.9862, 0.9712, 0.9452,  ..., 0.0000, 0.6912, 0.6321],
        [0.4944, 0.4727, 0.5213,  ..., 0.6912, 0.0000, 0.9090],
        [1.3741, 1.3539, 1.3707,  ..., 0.6321, 0.9090, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.06437598, 10.14436912, 10.30972193, 10.51437947, 10.52587243,
       10.54658597,  6.31576264,  7.17789666,  6.80559669, 11.95743558,
        9.63742669, 12.07818871, 12.38881635,  8.12580976,  8.8737556 ,
        6.55800267,  5.91408504, 13.18278606, 13.32557133,  8.41171369,
       10.88618114, 11.75618513, 11.42420508, 11.22596994,  5.91854397,
       10.68416401, 13.38091121,  9.60412735,  4.7663526 ,  5.91408504,
       12.50996828, 10.30972193, 14.02058621, 10.88618114,  8.92395381,
       13.18278606, 11.87665501, 13.96905756, 10.06703358, 10.37998078,
        6.90115901,  5.19577146,  5.94909709, 10.87409005,  5.96556248,
       10.60226614,  5.94535529, 12.21532282,  7.62232307,  5.57341623,
       10.48315436,  5.01114719, 11.62770332, 10.06407839,  6.27715105,
       10.8207297 ,  6.5618968 , 10.84437705, 12.72524615,  8.90405823,
        8.14433542,  5.0209005 ,  5.0209005 , 13.66896739,  5.1805345 ,
        5.82345867, 12.62576026, 11.48434767,  7.05627461,  6.42439344,
       10.67322527,  9.26455795,  8.05668609,  9.38232121, 11.018084  ,
       13.45642379,  4.76969364,  6.36004393,  6.54615887,  7.37902766,
        5.16077764,  7.8788965 ,  6.80559669,  5.27178363,  8.29640613,
        8.92549752,  9.73579738,  8.8353674 ,  7.42512621,  8.19703512,
        9.9353921 ,  9.41244865, 13.103709  , 12.72524615, 10.67322527,
        8.87587457,  9.79814997,  6.01356004,  9.94285511,  6.652742  ,
       13.89671314,  7.48311002,  9.15866607,  6.83515201, 11.07637531,
       13.12218013])}
{'P': tensor([[7.9971e-03, 2.8250e-01, 1.9433e-01,  ..., 2.7023e-07, 1.8817e-04,
         3.0776e-06],
        [1.3461e-04, 1.4105e-02, 1.9676e-04,  ..., 5.8734e-07, 4.3643e-04,
         8.3912e-06],
        [1.7240e-04, 4.0254e-01, 8.8150e-03,  ..., 3.3639e-07, 2.1994e-04,
         4.4409e-06],
        ...,
        [3.3540e-07, 2.2858e-07, 2.6662e-07,  ..., 3.9367e-03, 1.8065e-08,
         4.0408e-07],
        [2.6440e-06, 3.3732e-06, 2.9133e-06,  ..., 2.0613e-05, 3.7255e-02,
         3.2633e-04],
        [1.1319e-06, 9.8655e-07, 1.0489e-06,  ..., 9.3411e-02, 7.6594e-07,
         4.3304e-02]]), 'X': tensor([[ -8.0611,  -6.0302,   4.6962],
        [ -7.3470,  -7.0615,   5.0066],
        [ -7.6525,  -6.6534,   5.1721],
        [ -6.8618,  -7.6322,   3.7027],
        [ -6.2200,  -8.2737,   5.5079],
        [ -5.9938,  -8.4745,   3.9735],
        [  6.2895,   0.0405,   5.7652],
        [  5.9526,   3.2525,  -4.3821],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.2188,  -2.5839,   2.1833],
        [  3.4604,  -4.1545,   1.8091],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.7218,  -9.9934,   6.5764],
        [  4.5340,  -3.3629,   8.9652],
        [ -2.6485, -10.4151,   0.8766],
        [  8.7131,  10.2216,   7.7915],
        [  0.3038,   7.8092,   0.4164],
        [ -7.4602,   4.7718,  -4.6277],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.1771, -10.9776,  -3.2007],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.4351,  -1.8891,   3.5642],
        [ 12.6059,   0.9597,  -6.3264],
        [ -8.8934,   2.2597,   3.5831],
        [ -1.5408,   7.8978,   5.3470],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.1984,  -0.4327,   1.5438],
        [  7.5309,  11.2617,   7.2656],
        [  2.0547,   7.2957,  -1.9343],
        [  6.3583,   1.1729,   8.7835],
        [  4.4633,   5.6287,  -0.7754],
        [  2.5469,  13.7192,  -9.6935],
        [ -7.9667,   4.0814,  -0.6397],
        [  1.8766,   7.3694,   1.1887],
        [  6.8407,  11.7666,   3.0045],
        [  8.3718,  -8.2953,   8.4666],
        [  5.8450,  -9.9356,   3.0767],
        [  4.1297,  -3.7098,  -3.9838],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.7784,  -1.6401,   9.5587],
        [  0.3715,  -4.7756,   0.2312],
        [ -9.1178,   1.5784,   2.1410],
        [  2.9410,   6.8428,   5.4458],
        [ 11.5568,   6.0602,   9.3244],
        [  8.1528,  10.7452,   9.1493],
        [  6.3515,   0.6418,  -6.5823],
        [  3.7121,   6.3117,   0.6738],
        [  4.1999,  -3.6543,  -4.1998],
        [ 12.5199,  -0.4711,   3.0436],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.1588,   4.3364,  -8.0867],
        [  5.6523,   3.9443,  -2.9449],
        [  7.8425,  -8.7202,   6.4530],
        [  0.4996,   7.7738,  -4.4711],
        [  1.9312,  13.8649,   8.3563],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -5.1377,   6.7751,  -0.4438],
        [ -5.3328,  -8.9992,   7.3440],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.3076,   0.7852,  -6.7010],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.3413,   0.4883,  -6.8416],
        [  2.2182,  13.8007,   5.1889],
        [  6.0054,  -1.0902,   8.3079],
        [  0.2480, -11.0153,  -3.0183],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 12.5154,   2.5072,  -9.3274],
        [ 10.6673,  -5.6649,  -4.1646],
        [  9.9058,  -6.7203,   5.6928],
        [  4.5491,  -3.3485,   6.8412],
        [  8.3791,  10.5409,   5.5590],
        [ -7.7842,   4.3466,  -1.7548],
        [ -3.4488, -10.0934,   2.5195],
        [ -1.1885, -10.8211,   0.0865],
        [ -8.3539,  -5.5111,   0.3373],
        [  9.9058,  -6.7203,   5.6928],
        [  7.0515,  11.6194,   8.4289],
        [  4.5814, -10.4485,   2.5496],
        [ -5.9940,   6.1985,   5.9566],
        [  0.1460,   7.8340,  -9.4614],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.0345,   3.0175,  -1.4424],
        [ -2.0721,   7.8432,   6.6380],
        [ -7.0253,  -7.4493,  -1.5978],
        [  2.5974,   7.0363,  -7.7810],
        [ -9.1178,   1.5784,   2.1410],
        [ 12.3357,   3.6001,   0.7903],
        [  0.9478,  -4.8144,   0.6371],
        [ 11.9936,  -2.8677,   6.0614],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.5829,  -4.5466,  -8.4015],
        [ -5.2589,   6.7024,  -5.5809],
        [  6.3460,   1.4252,   1.2291],
        [  2.2182,  13.8007,   5.1889],
        [ -6.0844,  -8.3955,   0.7705],
        [ -7.0948,   5.1940,   9.8171],
        [  4.5954,  13.0099,  -4.4439],
        [  5.5240,  -2.1198,   6.2055],
        [  9.0051,   9.9234,   5.4411],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.6035,   0.7494,   7.1536]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0255, 0.0353,  ..., 1.2787, 0.8543, 1.0999],
        [0.0255, 0.0000, 0.0098,  ..., 1.2790, 0.8782, 1.0976],
        [0.0353, 0.0098, 0.0000,  ..., 1.2791, 0.8874, 1.0966],
        ...,
        [1.2787, 1.2790, 1.2791,  ..., 0.0000, 1.2664, 0.2169],
        [0.8543, 0.8782, 0.8874,  ..., 1.2664, 0.0000, 1.2126],
        [1.0999, 1.0976, 1.0966,  ..., 0.2169, 1.2126, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.06703358, 10.19036081, 10.14044489, 10.26327895, 10.35093485,
       10.37998078,  6.28961988,  6.78325151,  6.14086582,  5.82345867,
        5.40689332, 13.06498554, 11.51557166,  5.64500512, 10.74655346,
       13.43127131,  7.81509631,  8.85572292,  9.60080639, 10.37698435,
       12.59969936, 11.1913598 , 11.66453059,  9.62238828, 12.64235359,
        9.17595899,  8.04664886,  8.09704052,  9.7712321 ,  6.2134961 ,
       13.54774355,  7.57946459,  6.46560181,  7.18354885, 13.95361098,
        8.95133961,  7.60463349, 13.6105595 , 11.785566  , 11.52733581,
        5.5513072 , 12.42554808,  6.00663314,  4.79003361,  9.25336371,
        7.44805901, 13.0493368 , 13.48809595,  6.383897  ,  7.32233733,
        5.5671389 , 12.52875793, 11.99633819, 12.9089559 ,  6.89245352,
       11.72802491,  7.78980299, 13.99877413,  9.4997987 ,  8.50278896,
       10.4606265 ,  5.43544626,  9.34061938,  8.50268322, 13.09737053,
        6.36004393, 13.97779839,  6.10360073, 11.018084  , 12.05300943,
       12.76408562, 12.07818871, 11.9702762 ,  5.64865173, 13.46553623,
        8.91551231, 10.66631784, 10.88618114, 10.00794468, 11.9702762 ,
       13.59170809, 11.4088043 ,  8.62260327,  7.83534987, 10.63038565,
        9.62238828,  6.74686071,  8.11227668, 10.23946247,  7.50035355,
        9.25336371, 12.85032836,  4.90676242, 12.33167382,  9.14973319,
        5.22903117,  8.51928531,  6.50409372, 13.97779839, 10.36844813,
        8.79284466, 13.79762415,  5.91677188, 13.40024224, 11.33837396,
       12.62576026])}
{'P': tensor([[1.1161e-02, 2.5690e-01, 2.2434e-01,  ..., 1.3595e-02, 2.3566e-04,
         1.5494e-05],
        [1.2908e-04, 1.4549e-02, 2.3340e-03,  ..., 2.1168e-02, 5.2110e-04,
         3.6324e-05],
        [2.8440e-04, 3.5533e-02, 2.0889e-02,  ..., 1.2237e-02, 5.2782e-04,
         5.2634e-05],
        ...,
        [1.5518e-04, 5.0935e-04, 3.6115e-04,  ..., 3.8829e-02, 3.2147e-03,
         1.2188e-04],
        [6.5292e-05, 7.4389e-05, 6.4418e-05,  ..., 1.0036e-04, 9.8301e-02,
         2.5581e-03],
        [2.5973e-07, 2.3806e-07, 2.6335e-07,  ..., 1.7399e-07, 4.1170e-05,
         1.0986e-02]]), 'X': tensor([[-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0713, 0.0625,  ..., 0.4115, 0.9875, 1.0350],
        [0.0713, 0.0000, 0.1310,  ..., 0.3419, 0.9434, 1.0321],
        [0.0625, 0.1310, 0.0000,  ..., 0.4728, 1.0069, 1.0164],
        ...,
        [0.4115, 0.3419, 0.4728,  ..., 0.0000, 0.8707, 1.1598],
        [0.9875, 0.9434, 1.0069,  ..., 0.8707, 0.0000, 0.6170],
        [1.0350, 1.0321, 1.0164,  ..., 1.1598, 0.6170, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.0674887 , 10.23946247, 10.1979669 , 10.33712258, 10.32715715,
       10.39942497,  5.13616224,  7.39599986,  5.14999799,  8.16714836,
        5.83560009,  6.46560181, 11.77124373,  5.07306274, 12.35421461,
        9.84314712, 10.08001894, 11.85890664,  5.2287058 ,  9.12220271,
       11.22142195,  9.97822473,  8.01009083,  5.98285319, 11.38774149,
        5.14371005, 12.59969936,  5.01114719,  6.1463896 ,  7.32950385,
       13.70893492,  9.45995955,  9.7712321 , 12.41366754,  9.44486609,
       13.66051884,  6.74872981, 11.83570132, 11.04953767, 13.46553623,
       13.3426541 ,  5.38342077,  9.79814997, 11.35182589,  6.84796202,
        9.18358562, 10.84493732,  7.88677538,  4.78258092, 11.62158554,
       13.45642379,  5.24137909,  9.77777346, 11.24434374, 10.74409834,
       13.32557133,  7.8575648 , 12.59969936,  9.68324553,  7.96575782,
        6.29821673,  8.67502348, 13.48809595, 13.96746816, 10.71162681,
        7.72668467,  5.04028538,  8.08320235, 10.00410625,  7.50500064,
        6.69389431,  6.81117802, 10.30972193,  6.12886024, 12.3489324 ,
        8.67502348,  9.51074763, 12.7125079 , 12.50996828, 11.85890664,
       12.50996828,  9.09661212,  7.72415048,  4.98461983,  7.53063885,
        6.23262289, 11.21703654, 10.71162681, 13.6660671 , 13.40024224,
        8.27853076,  5.69913817,  6.42689705,  8.53014915,  9.87720102,
        5.39098669, 11.97957425, 13.20988996, 12.71843462,  9.15253344,
       11.50791859,  4.85603886, 13.54774355, 10.48721382,  5.62005629,
       12.27835996])}
{'P': tensor([[1.0894e-02, 2.7906e-01, 1.8786e-01,  ..., 5.7154e-03, 6.0386e-07,
         4.8281e-06],
        [8.3017e-05, 1.2120e-02, 3.7277e-01,  ..., 1.2714e-02, 1.0215e-06,
         3.1752e-06],
        [4.6833e-05, 1.2983e-04, 1.2748e-02,  ..., 2.5791e-02, 1.6137e-06,
         2.5867e-06],
        ...,
        [6.1293e-06, 1.1083e-05, 1.7187e-05,  ..., 2.0116e-02, 3.9407e-05,
         1.7195e-06],
        [3.7497e-07, 2.2803e-07, 1.8985e-07,  ..., 1.2787e-07, 1.1082e-02,
         2.6979e-05],
        [1.8894e-02, 7.5591e-03, 4.8101e-03,  ..., 3.3276e-04, 2.1309e-06,
         6.8253e-02]]), 'X': tensor([[ -8.0534,  -6.0429,   2.4583],
        [ -6.8466,  -7.6488,   1.9648],
        [ -5.9225,  -8.5355,   2.4778],
        [ -6.8618,  -7.6322,   3.7027],
        [ -5.1046,  -9.1610,   2.3418],
        [ -5.9938,  -8.4745,   3.9735],
        [ 11.2209,  -4.7092,   2.6117],
        [  1.9312,  13.8649,   8.3563],
        [  1.7000,  -4.7609,   2.7935],
        [  4.4633,   5.6287,  -0.7754],
        [  1.7500,   7.4187,   4.4237],
        [  4.5325,  -3.3643,   6.7699],
        [ -8.6426,  -4.9136,   9.4941],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.1886,  -2.6250,   6.4677],
        [ -8.6801,   2.7809,  -0.8443],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.0574,   1.7801,   8.8986],
        [  2.8890,  -4.4316,   1.1600],
        [  5.1861,   4.7361,  -3.7312],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -0.5385,   7.9037,   6.3830],
        [  6.0955,   2.8210,  -4.3321],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.9452, -10.3187,  -7.8621],
        [  1.7228,  -4.7575,   8.3396],
        [  2.2860,   7.1916,  -6.1011],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.1041, -11.0045,   2.4311],
        [  7.7794,  11.0629,   8.9069],
        [ -9.1907,  -3.3180,  -5.5943],
        [  0.9036,   7.6840,   9.9237],
        [  6.2630,   2.1057,  -5.2136],
        [  5.1861,   4.7361,  -3.7312],
        [  9.0051,   9.9234,   5.4411],
        [ -2.0721,   7.8432,   6.6380],
        [  4.7482, -10.3907,   5.9149],
        [ -4.6393,   7.0461,   1.1624],
        [  3.9663,   6.1018,  -5.4104],
        [ 11.9695,  -2.9407,  -6.3498],
        [  4.2612, -10.5514,  -6.6268],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.1222,  -0.7261,   8.8204],
        [ -8.8934,   2.2597,   3.5831],
        [  5.9526,   3.2525,  -4.3821],
        [ 11.3691,   6.4760,   1.0787],
        [  0.1931,  -4.7491,   7.4385],
        [  3.9770, -10.6341,   8.0783],
        [  9.3560,   9.5391,   9.4943],
        [ -2.0229, -10.6163,   8.6930],
        [  6.2770,  -0.0379,   2.4195],
        [  0.2571,  -4.7594,   1.5618],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.4901,  -0.7052,   1.7315],
        [ -3.5173, -10.0623,   4.8280],
        [  7.3592,  11.3937,   6.2407],
        [  7.5930,  -8.9037,   7.2323],
        [  6.3460,   1.4252,   1.2291],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.4167, -10.1293,   6.2156],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3200,   0.2690,   1.5900],
        [  6.1737,   2.5305,   3.9049],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.4501,  -1.7133,   2.2980],
        [  5.8318,   3.5567,  -4.7692],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.7295,  13.6703,  -8.9328],
        [  4.1637,  13.1898,  -1.9189],
        [  1.7228,  -4.7575,   8.3396],
        [  4.3523, -10.5232,   8.2138],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.3924, -10.1396,  -4.0863],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.2365,  -0.2578,  -6.0212],
        [  1.4150,  13.9647,  -4.0915],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.3900,   3.3727,   3.1650],
        [  3.1480,  -4.3165,  -7.8002],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 11.4495,   6.3026,   6.8257],
        [  4.5767,   5.5071,   3.9069],
        [ -2.2508, -10.5479,  -2.7235],
        [  4.0642,  -3.7599,  -3.4565],
        [  5.7963,  -1.6018,   1.8856],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.4691,   2.8525,  -5.5110],
        [ -0.8455, -10.8853,  -6.7400],
        [ -9.1623,  -3.4288,  -4.9961],
        [  1.5662,  13.9375,   4.5142],
        [ -7.5720,   4.6310,   2.1717],
        [  3.4401,  13.4540,   4.6576],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.3474,  -4.2162,   5.2455],
        [ -3.1204, -10.2345,   9.2288],
        [ -5.6174,   6.4708,  -1.7339],
        [ -2.0721,   7.8432,   6.6380],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.0870,  -4.6877,   9.6823],
        [ -3.0644,   7.6439,  -1.4322],
        [ -2.1533, -10.5778,   1.0112],
        [  0.5250, -11.0306,   5.5358],
        [ 11.2571,   6.7072,   8.3693],
        [ -7.8126,   4.3066,   5.1791]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.5335e-02, 2.8569e-02,  ..., 4.2594e-01, 1.3834e+00,
         3.4264e-01],
        [3.5335e-02, 0.0000e+00, 4.3317e-02,  ..., 4.4482e-01, 1.3887e+00,
         3.5504e-01],
        [2.8569e-02, 4.3317e-02, 0.0000e+00,  ..., 4.0307e-01, 1.3556e+00,
         3.1657e-01],
        ...,
        [4.2594e-01, 4.4482e-01, 4.0307e-01,  ..., 0.0000e+00, 1.0025e+00,
         1.1778e-01],
        [1.3834e+00, 1.3887e+00, 1.3556e+00,  ..., 1.0025e+00, 0.0000e+00,
         1.0474e+00],
        [3.4264e-01, 3.5504e-01, 3.1657e-01,  ..., 1.1778e-01, 1.0474e+00,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.06850677, 10.26545792, 10.38897563, 10.26327895, 10.48721382,
       10.37998078, 12.16901615, 13.99877413,  5.05534609,  7.18354885,
        7.62232307,  5.64464807,  9.94173708, 12.50996828,  5.81481044,
        9.11473391, 11.99633819,  9.23071698,  5.29009836,  7.02326515,
       10.32715715,  7.92200342,  6.7166297 , 12.17901033, 11.4424816 ,
        5.05981914,  7.54620849, 11.89855267, 11.0050379 , 13.52430176,
        9.7712321 ,  7.73692436,  6.60752373,  7.02326515, 13.40024224,
        8.11227668, 11.42420508,  8.43626114,  7.27759233, 12.32545637,
       11.37939695,  5.83560009,  6.16512856,  9.17595899,  6.78325151,
       13.08414689,  4.75303127, 11.35345345, 13.3614598 , 10.80728788,
        6.27715105,  4.7663526 ,  8.51928531, 12.50996828, 10.65929968,
       13.56367857, 11.70168755,  6.50409372,  5.18047638, 11.48663767,
        7.96181288,  6.32572323,  6.67218687,  6.60752373,  6.34059021,
        9.60412735,  6.83083863,  4.8763135 , 13.94008948, 13.83139024,
        5.05981914, 11.38774149,  4.91652556, 11.48434767, 12.00995898,
        6.24187044, 14.03618649,  5.05989547,  9.04255655,  5.34250225,
       10.53368644, 13.06957425,  7.16058587, 10.78533838,  5.5366698 ,
        6.01356004, 11.28895489, 12.79126785, 10.91805683,  9.78286733,
       14.02526505,  8.87587457, 13.88683436,  5.19065439,  5.38342077,
       10.69963954,  8.56889738,  8.11227668, 10.5575601 , 12.07818871,
        5.13125513,  8.23525433, 10.794755  , 11.04313641, 13.103709  ,
        8.92098752])}
{'P': tensor([[9.6902e-03, 2.9419e-01, 2.0779e-01,  ..., 1.5585e-03, 4.1481e-02,
         1.5818e-06],
        [2.3523e-04, 1.3378e-02, 2.6712e-01,  ..., 2.4926e-03, 1.5050e-02,
         1.8527e-06],
        [7.2861e-05, 1.2042e-04, 1.3604e-02,  ..., 2.6835e-03, 2.8470e-04,
         1.5798e-06],
        ...,
        [4.0335e-04, 4.5864e-04, 6.3622e-04,  ..., 1.3345e-01, 4.4095e-04,
         9.4855e-05],
        [5.2027e-04, 1.8909e-03, 1.3963e-01,  ..., 1.7727e-03, 1.3634e-02,
         2.1981e-06],
        [1.5519e-03, 1.1253e-03, 7.3483e-04,  ..., 1.6022e-04, 1.5669e-03,
         7.8722e-02]]), 'X': tensor([[-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0244, 0.0357,  ..., 0.8529, 0.1198, 0.7034],
        [0.0244, 0.0000, 0.0533,  ..., 0.8477, 0.1440, 0.7188],
        [0.0357, 0.0533, 0.0000,  ..., 0.8290, 0.1005, 0.6678],
        ...,
        [0.8529, 0.8477, 0.8290,  ..., 0.0000, 0.8754, 0.6914],
        [0.1198, 0.1440, 0.1005,  ..., 0.8754, 0.0000, 0.6249],
        [0.7034, 0.7188, 0.6678,  ..., 0.6914, 0.6249, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.07238463, 10.1371552 , 10.31886066, 10.35093485, 10.4606265 ,
       10.48315436, 11.92133129,  8.62329466,  4.96231532, 10.7167285 ,
        5.26067019, 12.59969936, 10.48840476,  9.20340031,  6.30647746,
        4.91038884, 12.62576026, 13.26590071,  9.20340031,  5.5748624 ,
       13.95361098,  8.79284466,  4.78544875,  9.48337063, 13.36092624,
       13.103709  , 12.27993098,  9.99211663,  5.18047638, 11.89105259,
        6.00732193,  9.48969978,  9.75200848,  5.5748624 , 10.52357479,
        8.99791316, 10.39942497, 14.07623266,  9.99211663, 11.64436684,
        6.2134961 ,  8.06517143,  6.652742  ,  6.54127431,  5.35774573,
       13.66991027, 13.52430176,  6.35029274,  8.14433542,  6.30647746,
        8.87587457,  5.98285319,  7.61088125,  6.59176018,  7.8575648 ,
       13.32557133,  7.6374603 ,  5.36061974,  9.10945678,  9.3349858 ,
        5.48276038, 13.96746816,  9.09661212,  4.75303127,  6.05165248,
       13.18278606,  4.7663526 ,  6.00663314,  9.17595899,  8.66273171,
        5.95111804, 10.46238054,  6.05113217, 12.22503754, 11.9182956 ,
        5.5366698 ,  6.81084724,  5.47702766, 11.75807252, 13.09737053,
        6.24187044,  5.79543038, 10.68416401, 11.75807252,  7.5158489 ,
       13.15525803,  7.19534146, 13.95472331,  7.46248947,  6.83515201,
        6.54693987,  7.5984809 , 12.58006038,  6.39187285,  5.19065439,
       13.32645552,  4.7663526 , 10.84437705, 13.69162795,  7.37576154,
        6.65259775, 11.52733581,  9.60359158,  4.85603886, 10.19036081,
        8.50278896])}
{'P': tensor([[9.8710e-03, 2.0626e-01, 1.8204e-01,  ..., 5.1715e-07, 1.0653e-05,
         1.8801e-06],
        [6.7214e-05, 7.7870e-03, 3.9624e-03,  ..., 5.6861e-07, 4.6370e-06,
         1.4469e-06],
        [1.3983e-04, 6.9341e-03, 1.5229e-02,  ..., 1.4514e-06, 9.0008e-06,
         2.5055e-06],
        ...,
        [3.4224e-07, 2.1091e-07, 2.6212e-07,  ..., 1.4778e-02, 3.2955e-06,
         1.3018e-04],
        [8.2950e-02, 3.5067e-02, 3.3836e-02,  ..., 7.9004e-07, 4.1951e-02,
         2.3014e-05],
        [1.7860e-04, 1.0092e-04, 8.3587e-05,  ..., 2.3418e-06, 3.2139e-03,
         3.0189e-02]]), 'X': tensor([[ -7.9925,  -6.1423,   2.8990],
        [ -6.8618,  -7.6322,   3.7027],
        [ -6.8466,  -7.6488,   1.9648],
        [ -5.9225,  -8.5355,   2.4778],
        [ -5.9938,  -8.4745,   3.9735],
        [ -5.6792,  -8.7352,   4.0318],
        [ -7.4602,   4.7718,  -4.6277],
        [  3.7916,  -3.9524,  -7.9547],
        [  0.6396,  14.0772,  -6.3786],
        [ -5.5525,   6.5146,  -4.4984],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.3122,  -9.6994,  -6.3282],
        [  5.2188,  -2.5839,   2.1833],
        [  9.5203,   9.3483,   5.8781],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.6672,  -1.8631,   8.1020],
        [  5.7802,  -1.6363,  -6.2782],
        [  0.5250, -11.0306,   5.5358],
        [  5.3602,  -2.3808,   7.2098],
        [  6.9956,  11.6590,  -7.6876],
        [  2.4572,   7.1085,   7.7376],
        [  0.9074,  -4.8139,   8.1674],
        [  8.1528,  10.7452,   9.1493],
        [  7.3592,  11.3937,   6.2407],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -8.3535,  -5.5118,   9.8399],
        [ -1.3178, -10.7939,   0.4236],
        [  7.8375,  -8.7240,  -9.8913],
        [  3.1756, -10.8247,   6.8882],
        [  3.9085,   6.1512,  -2.2560],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.9505,  -4.7171,   9.9868],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.6078, -10.9229,   6.2869],
        [  0.1047, -11.0046,   0.5188],
        [  8.5198,  10.4091,  -8.6486],
        [ -4.8751,  -9.3148,   4.0406],
        [  4.5325,  -3.3643,   6.7699],
        [ -9.3616,  -2.4749,   0.1793],
        [ -7.9298,   4.1367,  -3.9773],
        [ -6.4368,   5.8356,   3.3014],
        [ -5.9940,   6.1985,   5.9566],
        [ -2.6069, -10.4298,  -6.1995],
        [  1.3403,  -4.8011,  -1.1698],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -9.1624,  -3.4286,   2.0647],
        [  6.3569,   0.7591,   6.5011],
        [  2.1504,  -4.6726,   8.7908],
        [  2.3928,   7.1404,  -1.7035],
        [  4.6357,   5.4415,   7.9067],
        [ -7.5720,   4.6310,   2.1717],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -2.0721,   7.8432,   6.6380],
        [  6.3583,   1.1729,   8.7835],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -6.3489,   5.9117,   6.5161],
        [  0.9074,  -4.8139,   8.1674],
        [ 10.9931,   7.2107,   7.7933],
        [ 11.3118,   6.5957,  -4.5667],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.1041, -11.0045,   2.4311],
        [ 12.1940,  -2.1881,   7.1499],
        [  8.0381,  -8.5690,  -8.9523],
        [  1.6422,   7.4586,  -6.8056],
        [  5.1861,   4.7361,  -3.7312],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.9548,  -1.2273,  -2.6520],
        [  4.9957,  -2.8705,   6.5978],
        [ 11.6634,   5.8050,   9.1590],
        [  5.9655,   3.2172,  -7.5953],
        [  3.7916,  -3.9524,  -7.9547],
        [ 11.1420,   6.9331,  -2.4779],
        [ -1.6275, -10.7221,   3.2376],
        [ -1.8823, -10.6558,  -3.8952],
        [ -7.8280,   4.2848,  -7.1955],
        [  6.0955,   2.8210,  -4.3321],
        [  6.3119,   1.7797,   9.9186],
        [  0.9788,  -4.8145,  -3.8661],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.1575,  12.2046,   3.3257],
        [ -1.6086,   7.8928,   4.3385],
        [  6.1737,   2.5305,   3.9049],
        [  3.3094,  13.4969,  -3.6507],
        [ -8.3900,   3.3727,   3.1650],
        [  3.2809,  13.5061,  -4.0199],
        [ -3.6381,   7.4683,   7.3544],
        [  1.6232,  -4.7717,  -2.5655],
        [ -1.5451,   7.8975,   7.6104],
        [  3.2809,  13.5061,  -4.0199],
        [  2.9410,   6.8428,   5.4458],
        [  0.4119,  14.1019,  -4.1870],
        [  4.2096,   5.8815,   1.9230],
        [  0.4119,  14.1019,  -4.1870],
        [  5.9003,  12.3550,   2.1917],
        [  9.9756,  -6.6330,  -3.3429],
        [ 11.9641,  -2.9568,   7.7211],
        [  2.6310, -10.9195,   3.8054],
        [ -9.1813,  -3.3550,   1.8827],
        [ -9.3464,   0.5708,  -1.6397],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.2036,   2.4030,  -1.2500],
        [  2.6078, -10.9229,   6.2869],
        [ 11.6019,   5.9540,  -5.1715],
        [ -9.2508,   1.0572,   2.9978],
        [ -0.5385,   7.9037,   6.3830]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.9669e-02, 6.5776e-02,  ..., 1.3539e+00, 1.1493e-01,
         7.4064e-01],
        [6.9669e-02, 0.0000e+00, 1.3513e-01,  ..., 1.3855e+00, 1.2643e-01,
         7.1481e-01],
        [6.5776e-02, 1.3513e-01, 0.0000e+00,  ..., 1.3188e+00, 1.3208e-01,
         7.6192e-01],
        ...,
        [1.3539e+00, 1.3855e+00, 1.3188e+00,  ..., 0.0000e+00, 1.2592e+00,
         1.0445e+00],
        [1.1493e-01, 1.2643e-01, 1.3208e-01,  ..., 1.2592e+00, 2.1453e-04,
         6.3151e-01],
        [7.4064e-01, 7.1481e-01, 7.6192e-01,  ..., 1.0445e+00, 6.3151e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.08001894, 10.26327895, 10.26545792, 10.38897563, 10.37998078,
       10.41909211,  8.85572292,  5.47702766, 14.09176259,  8.55982105,
        5.84450127, 11.57249984,  5.82345867, 13.3426541 ,  9.70108409,
        5.96556248,  6.00732193, 11.04313641,  5.86519525, 13.59673406,
        7.52117132,  4.89869205, 13.48809595, 13.56367857, 13.97857591,
        9.5014146 , 10.00802627, 10.87409005, 11.72749923, 11.28093507,
        7.28793663, 10.91805683,  5.10446653, 10.4606265 , 11.22993218,
       11.00508681, 13.45124917, 10.51340829,  5.64464807,  9.68323686,
        8.94395628,  8.68832398,  8.62260327, 10.7506423 ,  4.98461983,
       12.25806559,  9.78285424,  6.4020404 ,  5.14371005,  7.53063885,
        7.14837514,  8.87587457, 12.35421461,  8.11227668,  6.46560181,
        9.55720651,  8.67502348,  4.89869205, 13.14690137, 13.09426142,
        5.26067019, 11.0050379 , 12.38881635, 11.74901778,  7.63726134,
        7.02326515, 12.67576792,  6.07992623,  5.76165934, 13.02818702,
        6.77777606,  5.47702766, 13.12299145, 10.84493732, 10.8207297 ,
        8.92395381,  6.7166297 ,  6.55800267,  4.91296205,  4.92497514,
       13.66991027,  8.0550286 ,  6.67218687, 13.89671314,  9.04255655,
       13.89886469,  8.30729352,  5.04028538,  8.04718433, 13.89886469,
        7.44805901, 14.10796466,  7.23276582, 14.10796466, 13.69162795,
       11.97957425, 12.32408516, 11.23201619,  9.7751189 ,  9.36378438,
       12.9089559 ,  6.652742  , 11.22993218, 13.04051744,  9.3109884 ,
        7.92200342])}
{'P': tensor([[1.3237e-02, 2.5772e-01, 1.6808e-01,  ..., 5.2417e-06, 9.1473e-04,
         5.1534e-03],
        [6.6985e-05, 9.5948e-03, 3.3554e-01,  ..., 7.3713e-06, 1.7448e-03,
         1.1763e-02],
        [8.4781e-05, 2.2503e-04, 1.7496e-02,  ..., 1.7463e-05, 4.5158e-03,
         3.3394e-02],
        ...,
        [6.7547e-07, 5.1852e-07, 4.8055e-07,  ..., 3.1104e-02, 2.2364e-06,
         8.3494e-07],
        [8.8099e-06, 1.2706e-05, 1.5105e-05,  ..., 3.5594e-02, 1.0993e-01,
         3.1870e-04],
        [7.8033e-06, 1.3596e-05, 1.7265e-05,  ..., 1.1288e-03, 2.0303e-01,
         3.7547e-02]]), 'X': tensor([[-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0503, 0.0822,  ..., 1.2223, 0.6285, 0.4394],
        [0.0503, 0.0000, 0.0322,  ..., 1.2090, 0.6145, 0.4090],
        [0.0822, 0.0322, 0.0000,  ..., 1.1962, 0.6022, 0.3871],
        ...,
        [1.2223, 1.2090, 1.1962,  ..., 0.0000, 0.5945, 0.8376],
        [0.6285, 0.6145, 0.6022,  ..., 0.5945, 0.0000, 0.2725],
        [0.4394, 0.4090, 0.3871,  ..., 0.8376, 0.2725, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.09979747, 10.31613269, 10.39942497, 10.51437947, 10.52587243,
       10.54658597, 11.75807252, 12.72524615,  6.48486171,  6.53290663,
       13.13888841,  5.46244988,  9.88571713,  8.53501777,  7.9337647 ,
        7.02253235, 11.76830981,  9.10945678,  5.41348121, 10.91601655,
        7.33350642,  5.05963217,  5.54776934, 10.69963954, 13.25017211,
        7.26074925,  7.56617037, 12.78327037, 11.28215429,  5.5513072 ,
       11.9182956 , 10.04916678,  9.78147542, 13.84126197,  9.37989631,
       10.33712258,  8.34949383, 13.14690137, 11.48663767,  7.50917551,
        8.27853076,  8.87007495,  9.04255655, 10.33958639,  8.35856487,
       11.50791859,  7.5092834 , 13.98691426,  7.13501264,  9.5014146 ,
       13.41713661,  8.94395628,  5.05534609,  6.83083863, 11.89105259,
       11.72749923,  6.28835101, 10.37698435,  5.35774573,  6.29821673,
        8.85939626, 13.83139024, 13.00380624, 12.71843462,  9.15976746,
       10.57302546,  8.74470218, 13.25336001, 12.0495521 ,  6.80559669,
       11.38774149,  9.12220271,  5.26067019, 13.59673406,  8.1303945 ,
        9.15866607,  5.94541803, 13.69162795, 13.35013374, 11.37939695,
        9.94285511,  5.96556248, 11.75438006,  6.00663314, 11.32149259,
        9.36256096,  5.9510489 , 11.04953767,  8.5724042 , 12.5085995 ,
        8.83341608,  6.65991877,  9.14973319,  7.60463349,  5.91222346,
       11.37939695, 10.60226614,  5.52416368, 10.84437705,  5.9510489 ,
        5.07306274,  5.14607014,  6.81084724, 12.59969936, 11.54071859,
       11.1913598 ])}
{'P': tensor([[8.2346e-03, 1.6442e-01, 1.4145e-01,  ..., 3.2330e-06, 1.5349e-06,
         5.2369e-07],
        [1.2023e-04, 1.3583e-02, 2.1137e-01,  ..., 9.6766e-06, 3.8439e-06,
         1.4926e-06],
        [7.6858e-05, 3.0269e-04, 2.1897e-02,  ..., 1.2816e-05, 5.9956e-06,
         3.2750e-06],
        ...,
        [3.1058e-06, 4.1844e-06, 3.3399e-06,  ..., 6.8068e-02, 8.8075e-03,
         5.8370e-04],
        [8.8095e-06, 7.7178e-06, 4.9611e-06,  ..., 1.6498e-04, 4.3555e-02,
         5.6332e-04],
        [4.4232e-07, 3.1968e-07, 2.2616e-07,  ..., 3.7108e-06, 1.0357e-03,
         1.4462e-02]]), 'X': tensor([[ -7.6719,  -6.6260,   7.1468],
        [ -6.4620,  -8.0449,   6.2719],
        [ -5.3328,  -8.9992,   7.3440],
        [ -5.1398,  -9.1367,   6.2074],
        [ -6.1914,  -8.2997,   8.5380],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.4639, -10.9431,  -7.3718],
        [  9.9756,  -6.6330,  -3.3429],
        [ -0.8569,   7.9156,  -6.5557],
        [  5.6314,  -1.9302,  -9.6543],
        [ 11.1469,   6.9237,   1.6160],
        [  5.1472,  -2.6801,   8.1194],
        [  6.7565,  11.8237,  -4.9882],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.8323,  -1.5228,   5.1132],
        [  3.7916,  -3.9524,  -7.9547],
        [ -4.1296,   7.2803,  -2.7194],
        [  0.9809,  -4.8145,   6.8706],
        [  3.7121,   6.3117,   0.6738],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.0467,  -9.8369,  -6.5816],
        [ -6.1914,  -8.2997,   8.5380],
        [ -9.4501,  -1.7133,   2.2980],
        [ -7.9298,   4.1367,  -3.9773],
        [  6.3360,   0.4258,  -9.2509],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.9808,   4.9340,  -3.3715],
        [ -9.2707,   0.9661,   4.4863],
        [  7.7794,  11.0629,   8.9069],
        [  1.1994,   7.6035,   2.4295],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.5900,   2.9767,   7.4811],
        [  4.1537,   5.9340,  -6.0423],
        [  1.7903,  -4.7467,   0.1168],
        [  2.4639, -10.9431,  -7.3718],
        [  4.1140,  -3.7220,  -0.1273],
        [ -3.5029,   7.5138,   8.1853],
        [  5.9193,  -1.3176,   5.1739],
        [  6.0305,   3.0296,   0.1375],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.4495,   6.3026,   6.8257],
        [  4.9337,  12.8566,  -3.6582],
        [  6.3413,   0.4883,  -6.8416],
        [ -3.5903,   7.4847,  -8.9112],
        [  8.5093,  -8.1764,  -3.2381],
        [ 11.0275,  -5.0666,   8.4154],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.8154,   2.4611,   2.5198],
        [  5.2608,  -2.5256,  -9.1446],
        [  7.5930,  -8.9037,   7.2323],
        [  3.7298,  13.3537,  -9.4653],
        [  8.8474,  10.0868,  -8.8644],
        [  3.3167,   6.6022,   6.3554],
        [ -8.0750,  -6.0072,  -5.8267],
        [ 11.5568,   6.0602,   9.3244],
        [ 11.5198,  -4.0923,   5.5088],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.2608,  -2.5256,  -9.1446],
        [ -0.6323,   7.9085,  -5.9544],
        [ -4.3359,  -9.6431,  -9.0616],
        [  7.7750,  11.0665,   2.8364],
        [  6.8093,  -9.4178,   8.4244],
        [  6.0047,   3.1062,  -6.7914],
        [ -1.5451,   7.8975,   7.6104],
        [  4.4633,   5.6287,  -0.7754],
        [  1.1139,   7.6281,  -2.5680],
        [ -4.4779,  -9.5609,  -2.4505],
        [  3.6817,   6.3354,   6.1233],
        [  9.3560,   9.5391,   9.4943],
        [ -3.7964,  -9.9294,   7.7147],
        [  3.2701,   6.6338,   9.9209],
        [  9.2811,  -7.4336,   5.1771],
        [  6.0305,   3.0296,   0.1375],
        [  3.2226,  -4.2802,  -4.0428],
        [  1.7903,  -4.7467,   0.1168],
        [  4.7482, -10.3907,   5.9149],
        [ -3.2053, -10.1993,  -3.8736],
        [ -8.4297,  -5.3637,  -9.3962],
        [  6.9653,  -9.3225,  -3.4727],
        [  0.6363, -11.0348,   6.1414],
        [  3.9022,  13.2906,   5.5788],
        [ -6.0195,   6.1789,  -2.9496],
        [ -4.7847,  -9.3729,  -3.7741],
        [  2.0039,   7.3173,  -1.5885],
        [  6.0310,  12.2795,  -6.4528],
        [ -6.3066,  -8.1936,   9.5822],
        [ -6.4620,  -8.0449,   6.2719],
        [  9.5261,  -7.1675,   1.4206],
        [ 12.6059,   0.9597,  -6.3264],
        [  2.5128, -10.9365,   9.9645],
        [  4.8653,   5.1694,  -4.0823],
        [  1.4749,  -4.7892,  -0.8089],
        [  9.3624,   9.5318,   0.2779],
        [ -8.9033,   2.2329,  -8.8912],
        [ -0.7599, -10.8995,  -6.0366],
        [  1.6954,  -4.7616,   9.8198],
        [  5.1886,  -2.6250,   6.4677],
        [  5.3602,  -2.3808,   7.2098],
        [  2.4932,   7.0903,  -5.4214],
        [ 12.3357,   3.6001,   0.7903],
        [  3.2366,  -4.2732,   5.1532],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.1469,   6.9237,   1.6160]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.3309e-02, 5.6299e-02,  ..., 1.3294e+00, 1.1814e+00,
         1.2062e+00],
        [5.3309e-02, 0.0000e+00, 8.2432e-02,  ..., 1.2804e+00, 1.1342e+00,
         1.1733e+00],
        [5.6299e-02, 8.2432e-02, 3.0339e-04,  ..., 1.3160e+00, 1.1639e+00,
         1.1683e+00],
        ...,
        [1.3294e+00, 1.2804e+00, 1.3160e+00,  ..., 0.0000e+00, 1.7708e-01,
         6.3169e-01],
        [1.1814e+00, 1.1342e+00, 1.1639e+00,  ..., 1.7708e-01, 0.0000e+00,
         4.9445e-01],
        [1.2062e+00, 1.1733e+00, 1.1683e+00,  ..., 6.3169e-01, 4.9445e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.1371552 , 10.31886066, 10.4606265 , 10.48315436, 10.35464916,
       10.31584102, 11.21703654, 11.97957425,  7.96181288,  5.95297459,
       13.12218013,  5.80312949, 13.61801526, 11.75807252,  6.02779201,
        5.47702766,  8.36995953,  4.91336994,  7.32233733, 12.13568598,
       11.54672755, 10.35464916,  9.60412735,  8.94395628,  6.35029274,
       12.91143491, 12.95702647,  9.32094055, 13.52430176,  7.69752262,
       12.04924159, 13.06957425,  9.09119351,  7.24326992,  5.07306274,
       11.21703654,  5.54776934,  8.29022197,  6.06416957,  6.74872981,
       14.10796466, 13.06957425, 13.77075174,  6.36004393,  8.30125201,
       11.80092578, 12.13568598, 12.49291768,  8.12043983,  9.15253344,
        5.83560009, 11.70168755, 13.86480175, 13.41713661,  7.38844703,
       10.06437598, 13.0493368 , 12.22503754, 13.01187484,  5.83560009,
        7.9337647 , 10.57302546, 13.5247215 , 11.62158554,  6.76056625,
        8.04718433,  7.18354885,  7.70897675, 10.5575601 ,  7.32754882,
       13.3614598 , 10.63038565,  7.39599986, 11.89105259,  6.74872981,
        5.35774573,  5.07306274, 11.42420508, 10.6910768 ,  9.99145096,
       11.63724792, 11.05317597, 13.85158417,  8.62631801, 10.52357479,
        7.58668144, 13.68063098, 10.33958639, 10.31886066, 11.92133129,
       12.64235359, 11.22142195,  7.09887557,  5.01114719, 13.3607379 ,
        9.17905491, 10.92597196,  5.05444288,  5.81481044,  5.86519525,
        7.5158489 , 12.85032836,  5.36061974,  5.83560009,  7.02326515,
       13.12218013])}
{'P': tensor([[7.8464e-03, 3.5830e-01, 2.1157e-01,  ..., 3.8158e-06, 4.0263e-06,
         3.3916e-06],
        [1.6041e-04, 1.1499e-02, 3.3599e-01,  ..., 6.3163e-06, 6.7286e-06,
         4.2602e-06],
        [2.3312e-04, 3.1662e-04, 3.9639e-02,  ..., 2.5542e-05, 2.6966e-05,
         9.8323e-06],
        ...,
        [1.9631e-05, 1.9163e-05, 1.4726e-05,  ..., 1.2195e-01, 9.5499e-03,
         4.4940e-05],
        [1.2382e-05, 1.2341e-05, 9.9495e-06,  ..., 2.9759e-01, 1.0865e-01,
         1.9519e-05],
        [9.3708e-03, 7.4074e-03, 4.3731e-03,  ..., 2.9964e-06, 2.3113e-06,
         3.9781e-02]]), 'X': tensor([[ -7.6525,  -6.6534,   5.1721],
        [ -7.3470,  -7.0615,   5.0066],
        [ -6.2200,  -8.2737,   5.5079],
        [ -6.4620,  -8.0449,   6.2719],
        [ -5.9938,  -8.4745,   3.9735],
        [ -5.6792,  -8.7352,   4.0318],
        [ 12.6038,   0.7665,  -3.4439],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.3459,  -3.5327,   0.3667],
        [  7.1697,  11.5343,  -0.7276],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.7218,  -9.9934,   6.5764],
        [ -1.3940, -10.7772,   6.7835],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3497,   1.3680,  -1.9076],
        [  3.4920,  -4.1366,  -8.3955],
        [  1.7659,   7.4127,   4.7871],
        [ 12.6038,   0.7665,  -3.4439],
        [ -1.8367,   7.8718,  -8.6523],
        [  4.5954,  13.0099,  -4.4439],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.9936,  -2.8677,   6.0614],
        [  0.1252, -11.0062,   7.2564],
        [ 11.2517,  -4.6495,   2.9800],
        [ -8.3653,   3.4184,   9.3454],
        [  4.9531,   5.0576,   5.6794],
        [ 12.4690,  -0.8528,  -9.3374],
        [  0.5175,  14.0910,   4.6540],
        [  6.3587,   1.1596,  -3.6629],
        [ -9.1706,   1.3863,   8.4032],
        [  0.9809,   7.6642,  -1.4217],
        [ -1.3178, -10.7939,   0.4236],
        [ -1.7009,   7.8851,  -8.2437],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.3515,   0.6418,  -6.5823],
        [ -7.4887,   4.7364,   8.9074],
        [  2.3376,  -4.6231,   9.8436],
        [  5.6414, -10.0303,   7.8525],
        [  2.5128, -10.9365,   9.9645],
        [  1.5662,  13.9375,   4.5142],
        [  2.9470,  13.6085,   2.9809],
        [ -8.5149,  -5.1905,   1.5239],
        [  1.7659,   7.4127,   4.7871],
        [ -8.9636,   2.0648,   8.1946],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.7815,  -4.7481,   4.4289],
        [ 12.4877,  -0.7222,   8.8347],
        [  2.3373,  -4.6232,  -5.2246],
        [ -9.4756,  -0.8166,  -9.6021],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.4877,  -0.7222,   8.8347],
        [ -7.0120,   5.2825,   7.7762],
        [  9.5203,   9.3483,   5.8781],
        [ -8.1666,   3.7653,   3.6075],
        [  3.3744,   6.5623,  -6.3636],
        [ -8.3062,  -5.6008,   0.0287],
        [  3.6703,   6.3443,  -5.4401],
        [ -6.7992,   5.4991,   9.9186],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.1667,  -2.6543,   1.2879],
        [ 12.5763,   0.1333,  -9.2650],
        [ -8.8631,  -4.3728,  -1.1746],
        [  2.1625,  -4.6697,   7.0658],
        [ -2.1382,   7.8339,   7.3524],
        [ -2.4956, -10.4682,   3.5632],
        [ 11.2571,   6.7072,   8.3693],
        [  0.9809,   7.6642,  -1.4217],
        [ -0.5385,   7.9037,   6.3830],
        [ -9.1433,  -3.5003,  -3.0209],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.5424,  -2.0885,   7.2487],
        [ 11.7796,  -3.4696,   9.0088],
        [  4.2612, -10.5514,  -6.6268],
        [ -8.0330,  -6.0765,   6.7945],
        [  9.3836,   9.5075,  -5.9278],
        [  0.1677,  -4.7447,  -6.1529],
        [  7.3592,  11.3937,   6.2407],
        [ -7.6734,   4.4980,  -5.4248],
        [ -3.4100,   7.5436,  -7.6980],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.4501,  -1.7133,   2.2980],
        [  1.3670,  -4.7990,   3.2603],
        [  8.4691,  10.4571,  -5.0650],
        [  4.1297,  -3.7098,  -3.9838],
        [ -9.3384,  -2.6161,  -4.7630],
        [  1.8719,   7.3713,  -3.7465],
        [  0.1532,  -4.7422,  -7.0145],
        [ 10.9916,   7.2133,  -0.9179],
        [ 11.7794,   5.5086,  -9.6929],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 11.5456,   6.0861,   8.1101],
        [ -4.7389,   6.9953,   3.7638],
        [  5.4336,  -2.2676,   5.6933],
        [ 11.9641,  -2.9568,   7.7211],
        [  0.6396,  14.0772,  -6.3786],
        [  5.0506,   4.9277,  -9.5094],
        [ 12.3609,   3.4769,  -8.5762],
        [ -9.4748,  -0.7765,  -5.6695],
        [  9.2811,  -7.4336,   5.1771],
        [  9.4554,   9.4245,   9.8800],
        [ -1.6086,   7.8928,   4.3385],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3604,   0.9042,  -6.5901],
        [ -8.3653,   3.4184,   9.3454]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0098, 0.0481,  ..., 1.1746, 1.2192, 0.2780],
        [0.0098, 0.0000, 0.0534,  ..., 1.1675, 1.2119, 0.2836],
        [0.0481, 0.0534, 0.0000,  ..., 1.1628, 1.2088, 0.2301],
        ...,
        [1.1746, 1.1675, 1.1628,  ..., 0.0000, 0.0577, 1.1609],
        [1.2192, 1.2119, 1.2088,  ..., 0.0577, 0.0000, 1.2128],
        [0.2780, 0.2836, 0.2301,  ..., 1.1609, 1.2128, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.14044489, 10.19036081, 10.35093485, 10.31886066, 10.37998078,
       10.41909211, 12.62711252, 12.05300943,  5.60064319, 13.58101794,
        4.91296205, 11.51557166, 10.86693708, 11.62158554,  6.49538841,
        5.41348121,  7.62011734, 12.62711252,  8.08320235, 13.79762415,
        4.87083348, 12.33167382, 11.00694688, 12.17451685,  9.0368458 ,
        7.07902036, 12.4980845 , 14.10045797,  6.46356806,  9.27474386,
        7.72668467, 10.87409005,  8.06643487,  7.89745107,  9.50913849,
        6.383897  ,  8.86082124,  5.1805345 , 11.50791859, 11.22142195,
       14.02526505, 13.92390689,  9.97220799,  7.62011734,  9.19837792,
        9.88571713,  5.0713364 , 12.5085995 ,  5.18047638,  9.51074763,
        5.81199422, 12.5085995 ,  8.77913428, 13.3426541 ,  8.99276564,
        7.37902766, 10.01804223,  7.32950385,  8.74470218,  5.39723241,
        5.80860165, 12.5769659 ,  9.88310614,  5.14607014,  8.12043983,
       10.76154256, 13.103709  ,  7.72668467,  7.92200342,  9.79039371,
        5.18277858,  5.92281142, 12.27993098, 11.37939695, 10.07238463,
       13.35832725,  4.74770899, 13.56367857,  8.89458273,  8.27853076,
       13.6660671 ,  9.60412735,  4.98989906, 13.45642379,  5.5513072 ,
        9.69792513,  7.60529892,  4.74468406, 13.1471313 , 13.00380624,
       10.33712258, 13.05148826,  8.44940661,  5.88782913, 12.32408516,
       14.09176259,  7.05627461, 12.84056604,  9.50654967, 11.89105259,
       13.35013374,  8.0550286 , 10.06407839,  6.60752373,  6.42439344,
        9.0368458 ])}
{'P': tensor([[1.5695e-02, 2.1795e-01, 2.0328e-01,  ..., 1.7206e-05, 9.0917e-07,
         2.8596e-04],
        [4.7012e-05, 1.6330e-02, 2.6095e-01,  ..., 8.3881e-05, 2.4562e-06,
         7.5063e-04],
        [3.7589e-05, 7.2657e-04, 1.3597e-02,  ..., 7.4765e-05, 2.1197e-06,
         6.8035e-04],
        ...,
        [9.3068e-07, 9.2378e-07, 9.3646e-07,  ..., 4.2985e-02, 3.7461e-02,
         2.8823e-06],
        [7.7245e-07, 3.2685e-07, 3.2007e-07,  ..., 1.3861e-06, 1.5852e-02,
         8.3308e-08],
        [6.1973e-06, 2.0052e-05, 2.1461e-05,  ..., 1.0544e-02, 7.9274e-05,
         8.5184e-02]]), 'X': tensor([[-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0986, 0.0959,  ..., 1.0507, 1.2551, 1.0171],
        [0.0986, 0.0000, 0.0055,  ..., 0.9530, 1.1580, 0.9630],
        [0.0959, 0.0055, 0.0000,  ..., 0.9552, 1.1618, 0.9605],
        ...,
        [1.0507, 0.9530, 0.9552,  ..., 0.0000, 0.3978, 0.8289],
        [1.2551, 1.1580, 1.1618,  ..., 0.3978, 0.0000, 1.2261],
        [1.0171, 0.9630, 0.9605,  ..., 0.8289, 1.2261, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.14436912, 10.51437947, 10.52587243, 10.54658597, 10.31613269,
       10.52357479,  8.95983586,  5.39450305, 10.66631784, 14.10045797,
       11.75807252,  9.69792513,  6.39520397,  6.80559669, 10.23946247,
       12.10222204,  8.50278896,  5.64865173,  4.7663526 ,  5.63824971,
       13.32374056, 10.42667093,  5.10446653, 12.52875793, 10.35464916,
        5.79543038,  6.83083863,  7.39004226,  7.83486943,  8.31625171,
        7.20826219, 11.35345345,  7.8788965 ,  8.49161154,  6.02588517,
        4.9160962 , 11.63383221, 13.96987806,  5.14607014,  8.31625171,
        6.46356806,  6.05113217,  8.67502348,  5.1224079 , 13.50484539,
        7.02326515, 11.22142195,  7.9337647 , 12.67576792, 13.77075174,
       11.83570132,  8.99791316,  8.92549752,  7.69752262,  6.3949043 ,
        8.01009083,  8.85939626, 13.744387  , 13.46553623, 13.02818702,
        9.95552626,  5.68942572, 12.76408562,  5.8828986 ,  7.02496223,
        8.43626114, 13.88683436, 11.07637531,  9.48969978,  8.12580976,
       10.06023665,  9.61942445, 10.30972193,  7.57946459,  6.1561646 ,
        7.62509582, 13.59056613,  7.27759233, 12.35625694,  5.21280156,
       12.35421461,  8.30125201, 13.31764181,  4.85603886,  7.16058587,
        8.41148722,  6.50409372,  5.47702766, 12.16901615, 11.85890664,
        5.98285319, 11.23201619, 13.60275211,  7.50035355,  5.57341623,
        6.36004393,  7.80625981, 14.10796466,  8.86082124,  8.06990483,
        5.04028538,  8.29022197,  9.18690018, 12.4013153 , 13.25017211,
       11.31276109])}
{'P': tensor([[9.4137e-03, 2.7506e-01, 1.6887e-01,  ..., 4.5555e-05, 7.4922e-05,
         4.1872e-04],
        [1.3068e-04, 1.6361e-02, 5.4146e-02,  ..., 7.3655e-05, 1.7480e-04,
         3.6823e-04],
        [8.1375e-05, 9.5469e-04, 1.2539e-02,  ..., 1.0754e-04, 2.9248e-04,
         4.4327e-04],
        ...,
        [5.9423e-05, 6.3982e-05, 1.0572e-04,  ..., 1.6893e-01, 1.8610e-02,
         5.7985e-04],
        [9.5028e-07, 1.2209e-06, 1.8479e-06,  ..., 9.2376e-04, 2.7272e-02,
         4.6849e-06],
        [1.6598e-04, 3.0716e-04, 7.2916e-04,  ..., 1.2314e-02, 2.9494e-02,
         5.9973e-02]]), 'X': tensor([[ -7.3470,  -7.0615,   5.0066],
        [ -6.2200,  -8.2737,   5.5079],
        [ -5.6792,  -8.7352,   4.0318],
        [ -5.9938,  -8.4745,   3.9735],
        [ -4.8751,  -9.3148,   4.0406],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.2022,   2.4092,   3.0349],
        [  2.7295,  13.6703,  -8.9328],
        [  6.2770,  -0.0379,   2.4195],
        [ -9.4541,  -0.3327,   8.8246],
        [  5.0190,   4.9705,  -7.9752],
        [  6.3587,   1.1596,  -3.6629],
        [ -4.3359,  -9.6431,  -9.0616],
        [  5.2608,  12.6977,   4.9166],
        [  0.2732,  -4.7619,  -1.1577],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.5203,   9.3483,   5.8781],
        [  4.2041,  -3.6509,  -4.9488],
        [ -8.7544,   2.6096,   3.3146],
        [ -3.3812, -10.1235,   7.4384],
        [  4.5325,  -3.3643,   6.7699],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.3549,   0.7096,   6.9198],
        [  7.0358,  -9.2784,  -9.8479],
        [  0.5960, -11.0335,   3.1144],
        [  0.9788,  -4.8145,  -3.8661],
        [  1.5383, -11.0274,  -4.9666],
        [  2.6310, -10.9195,   3.8054],
        [ -8.0330,  -6.0765,   6.7945],
        [ 10.2470,   8.4054,  -0.1177],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -5.6423,   6.4537,  -1.2091],
        [ 12.6038,   0.7665,  -3.4439],
        [  2.7295,  13.6703,  -8.9328],
        [ -5.5525,   6.5146,  -4.4984],
        [ -2.6735, -10.4062,  -1.5911],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.4489,   7.5256,   4.4529],
        [  1.9204,   7.3518,   8.4781],
        [  7.3724,  -9.0577,  -8.9843],
        [  6.3604,   0.9042,  -6.5901],
        [ -1.6275, -10.7221,   3.2376],
        [ -2.3317,   7.8034,  -3.6274],
        [  0.9659,  -4.8144,  -0.9594],
        [  5.7845,   3.6652,   0.9175],
        [  2.9260,  -4.4162,   8.3808],
        [  8.6271,  -8.0715,  -7.4813],
        [  7.0515,  11.6194,   8.4289],
        [  6.3535,   0.6792,  -7.5101],
        [  2.8602,   6.8905,   5.8779],
        [  2.5636, -10.9293,   9.6519],
        [  1.4489,   7.5256,   4.4529],
        [  0.2571,  -4.7594,   1.5618],
        [  5.5158,  -2.1337,   3.5246],
        [  7.7750,  11.0665,   2.8364],
        [  0.3038,   7.8092,   0.4164],
        [  6.1741,  -0.5329,   3.1913],
        [  2.5620,   7.0549,   8.4825],
        [  5.1861,   4.7361,  -3.7312],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.3554,   5.7393,  -9.0766],
        [ 12.1751,  -2.2592,  -5.7751],
        [  3.2366,  -4.2732,   5.1532],
        [ 11.9936,  -2.8677,   6.0614],
        [  0.5600, -11.0321,  -0.8660],
        [  3.4401,  13.4540,   4.6576],
        [  4.8432,   5.1968,   1.5006],
        [  4.8051, -10.3703,   0.5268],
        [  6.3604,   0.9042,  -6.5901],
        [ -7.3470,  -7.0615,   5.0066],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.3200,   7.1756,   1.1938],
        [ -4.7642,  -9.3860,  -5.1133],
        [  0.1047, -11.0046,   0.5188],
        [  0.7958,  -4.8109,  -8.2248],
        [  6.3542,   0.6934,  -3.3840],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.4996,   2.6335,  -6.3166],
        [ -6.2668,   5.9808,   2.5614],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.1532,  -4.7422,  -7.0145],
        [  3.6703,   6.3443,  -5.4401],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.1752,   2.5242,   3.5874],
        [  8.9284,  10.0036,  -6.9210],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.2275,  -3.6320,  -0.2714],
        [  4.1999,  -3.6543,  -4.1998],
        [  7.8982,  -8.6778,   0.0239],
        [ -9.0961,   1.6529,   3.1553],
        [ -3.5029,   7.5138,   8.1853],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 12.3357,   3.6001,   0.7903],
        [  2.3200,   7.1756,   1.1938],
        [ 10.8670,  -5.3428,  -2.8637],
        [  4.9531,   5.0576,   5.6794],
        [  3.2692,  13.5098,  -4.9032],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -5.9872,   6.2037,  -9.4477],
        [ -1.5408,   7.8978,   5.3470],
        [  2.0899,  -4.6870,  -7.4260],
        [  6.3122,  -9.6994,  -6.3282],
        [ -4.7642,  -9.3860,  -5.1133]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0534, 0.0671,  ..., 1.1611, 1.0557, 0.7828],
        [0.0534, 0.0000, 0.1041,  ..., 1.1607, 1.0684, 0.8209],
        [0.0671, 0.1041, 0.0000,  ..., 1.1040, 0.9916, 0.7183],
        ...,
        [1.1611, 1.1607, 1.1040,  ..., 0.0000, 0.2904, 0.8321],
        [1.0557, 1.0684, 0.9916,  ..., 0.2904, 0.0000, 0.5667],
        [0.7828, 0.8209, 0.7183,  ..., 0.8321, 0.5667, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.19036081, 10.35093485, 10.41909211, 10.37998078, 10.51340829,
       10.48315436,  6.65368551, 13.94008948,  6.27715105,  9.45995955,
        7.06372172,  6.46356806, 10.57302546, 13.744387  ,  4.76969364,
       10.46739073, 12.10222204, 13.3426541 ,  5.56809335,  9.13507321,
       10.67322527,  5.64464807, 13.01728313,  6.39438044, 11.64436684,
       11.04953767,  4.91296205, 11.1341798 , 11.23201619, 10.07238463,
       13.25336001, 10.33712258,  8.5724042 , 12.62711252, 13.94008948,
        8.55982105, 10.74409834, 12.17451685,  7.66377554,  7.5984809 ,
       11.67875855,  6.42439344, 10.84493732,  8.14433542,  4.91038884,
        6.84796202,  5.29755487, 11.81423161, 13.59170809,  6.38968357,
        7.46053281, 11.22596994,  7.66377554,  4.7663526 ,  5.91408504,
       13.5247215 ,  7.81509631,  6.19707944,  7.50564223,  7.02326515,
        5.01114719,  7.20482224, 12.3828957 ,  5.36061974, 12.33167382,
       11.04629471, 13.88683436,  7.10378919, 11.42947632,  6.42439344,
       10.19036081, 10.71162681,  7.54127425, 10.52587243, 11.00508681,
        4.8763135 ,  6.39187285,  9.61942445, 12.77402351,  8.66273171,
        9.36093562,  4.74468406,  7.32950385,  9.55593374,  6.67121748,
       13.40850328, 13.14777433,  5.57341623,  5.5671389 , 11.73397309,
        9.24501829,  8.29022197,  7.37902766,  9.37989631,  9.91668182,
       12.85032836,  7.54127425, 12.10942412,  7.07902036, 13.89974693,
       10.23946247,  8.62162372,  8.04664886,  5.13182506, 11.57249984,
       10.52587243])}
{'P': tensor([[8.7089e-03, 2.3806e-01, 1.6520e-01,  ..., 9.2520e-07, 2.2891e-05,
         1.1118e-05],
        [4.4215e-05, 7.2212e-03, 9.2877e-05,  ..., 4.2694e-07, 4.5208e-05,
         1.3091e-05],
        [1.1380e-04, 2.5914e-01, 7.4100e-03,  ..., 4.8798e-07, 3.2051e-05,
         1.0898e-05],
        ...,
        [2.4193e-03, 1.2873e-03, 1.3452e-03,  ..., 2.9297e-01, 6.2248e-06,
         7.2575e-05],
        [1.3639e-06, 1.3838e-06, 1.3561e-06,  ..., 2.1847e-06, 5.2613e-02,
         1.3170e-02],
        [3.4586e-05, 2.5293e-05, 2.7226e-05,  ..., 2.0005e-04, 1.2700e-04,
         7.5715e-02]]), 'X': tensor([[ -7.2986,  -7.1225,  -3.0814],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -4.0633,  -9.7927,  -3.0825],
        [  1.3670,  -4.7990,   3.2603],
        [ -7.9925,  -6.1423,   2.8990],
        [ 10.5955,   7.8824,  -3.7006],
        [  1.3767, -11.0341,   9.9415],
        [ 10.8670,  -5.3428,  -2.8637],
        [  3.3943,   6.5483,   1.0790],
        [  1.7659,   7.4127,   4.7871],
        [  4.6992,   5.3690,   3.0935],
        [  6.2804,   2.0019,   0.8607],
        [  6.3542,   0.6934,  -3.3840],
        [ -3.5520,   7.4976,   9.1522],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 10.1503,   8.5414,  -7.2681],
        [ -2.7572,   7.7194,   5.9580],
        [  9.2979,   9.6048,   7.3138],
        [  2.0729,   7.2878,   7.0880],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.6050,   2.9451,  -0.7012],
        [  5.5626,  -2.0536,   8.4433],
        [  2.6613,   7.0021,   4.5783],
        [ -7.9925,  -6.1423,   2.8990],
        [  6.9433,  -9.3362,  -0.3421],
        [  6.9653,  -9.3225,  -3.4727],
        [ -8.8400,   2.3991,   1.3457],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.1931,  -4.7491,   7.4385],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.7815,  -4.7481,   4.4289],
        [  6.2419,  -0.2313,   4.2440],
        [  1.8320,   7.3871,   5.1452],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 10.0400,   8.6922,  -6.2847],
        [  9.1820,   9.7334,  -2.7932],
        [  6.6319,  11.9066,   3.5165],
        [ 11.2571,   6.7072,   8.3693],
        [  6.0955,   2.8210,  -4.3321],
        [ -9.3464,   0.5708,  -1.6397],
        [  9.9756,  -6.6330,  -3.3429],
        [ -4.2271,   7.2387,  -1.8316],
        [ -7.0948,   5.1940,   9.8171],
        [  6.0467,  -9.8369,  -6.5816],
        [ -8.0330,  -6.0765,   6.7945],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.2419,  -0.2313,   4.2440],
        [  4.2096,   5.8815,   1.9230],
        [ -0.8675, -10.8815,   6.8561],
        [  3.2701,   6.6338,   9.9209],
        [  3.9913,  -3.8139,  -6.8311],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 11.4495,   6.3026,   6.8257],
        [  4.7156,  -3.1831,  -8.0224],
        [ -7.6525,  -6.6534,   5.1721],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.5789,   0.1722,   1.6470],
        [  5.9526,   3.2525,  -4.3821],
        [  4.2339,  -3.6268,   6.3125],
        [  4.0421,  -3.7765,  -8.3631],
        [ -8.0534,  -6.0429,   2.4583],
        [ -7.9298,   4.1367,  -3.9773],
        [  4.5503,   5.5359,   7.4735],
        [ -9.3705,   0.4208,  -3.5815],
        [ -7.4733,   4.7555,   6.2490],
        [  2.7295,  13.6703,  -8.9328],
        [  0.9036,   7.6840,   9.9237],
        [  3.4009,  -4.1874,  -5.9751],
        [ -4.7389,   6.9953,   3.7638],
        [  5.6202,  -1.9508,   0.7471],
        [  4.5503,   5.5359,   7.4735],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -1.6086,   7.8928,   4.3385],
        [  3.2809,  13.5061,  -4.0199],
        [  2.2860,   7.1916,  -6.1011],
        [  6.2675,  12.1379,  -8.9720],
        [  0.2571,  -4.7594,   1.5618],
        [ 12.3609,   3.4769,  -8.5762],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.9788,   2.0208,  -3.2867],
        [  0.5600, -11.0321,  -0.8660],
        [ -2.0840,   7.8415,  -7.8657],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.1667,  -2.6543,   1.2879],
        [ -9.3087,   0.7792,  -7.5587],
        [  4.2096,   5.8815,   1.9230],
        [  6.0890,  -0.8382,   2.9166],
        [  3.2647, -10.8066,  -0.3768],
        [  1.9241,   7.3503,   7.9004],
        [  6.3305,   1.6119,   1.1538],
        [ -8.8835,  -4.3178,   9.3080],
        [  3.7916,  -3.9524,  -7.9547],
        [ -0.5385,   7.9037,   6.3830],
        [ -2.7572,   7.7194,   5.9580],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.1620,  -7.5573,  -9.9882],
        [ -8.1426,   3.8048,   9.3542],
        [ 11.9019,  -3.1379,  -3.4534],
        [  6.0305,   3.0296,   0.1375]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0356, 0.0513,  ..., 0.9049, 0.9770, 1.0149],
        [0.0356, 0.0000, 0.0322,  ..., 0.9224, 0.9514, 0.9992],
        [0.0513, 0.0322, 0.0000,  ..., 0.9523, 0.9709, 1.0263],
        ...,
        [0.9049, 0.9224, 0.9523,  ..., 0.0000, 1.1512, 0.8991],
        [0.9770, 0.9514, 0.9709,  ..., 1.1512, 0.0000, 0.3449],
        [1.0149, 0.9992, 1.0263,  ..., 0.8991, 0.3449, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.1979669 , 10.39942497, 10.31613269, 10.52357479, 10.5575601 ,
       10.60226614,  4.98989906, 10.08001894, 13.20597932, 11.11969815,
       12.10942412,  7.37576154,  7.62011734,  7.13501264,  6.59176018,
        6.39187285,  8.29640613,  9.50654967,  9.94285511, 13.26590071,
        8.19703512, 13.36800073,  7.57685908, 10.33712258,  9.09502385,
        5.92952538,  7.49076899, 10.08001894, 11.63502679, 11.63724792,
        9.15976746, 10.7167285 ,  4.75303127, 12.32545637,  5.0713364 ,
        6.24614905,  7.61088125, 10.52357479, 13.27994304, 13.38091121,
       13.62896593, 13.103709  ,  6.7166297 ,  9.36378438, 11.97957425,
        8.38252937,  8.79284466, 11.54672755, 10.07238463, 12.30695098,
        6.24614905,  7.23276582, 10.91601655,  7.39599986,  5.5205189 ,
       12.30858677, 13.06957425,  5.68942572, 10.14044489, 13.41713661,
       12.58006038,  6.78325151,  5.5748624 ,  5.53175275, 10.06850677,
        8.94395628,  7.16598273,  9.37989631,  8.8580671 , 13.94008948,
        7.73692436,  5.39450305,  8.44940661,  5.94909709,  7.16598273,
       12.18045978,  8.0550286 , 13.89886469,  7.54620849, 13.66051884,
        4.7663526 , 12.84056604, 11.97957425,  9.20340031, 11.04629471,
        8.1137452 , 10.61472636, 13.04217487,  5.80860165,  9.34127013,
        7.23276582,  6.1463896 , 11.28895489,  7.59795445,  6.53250937,
        9.87720102,  5.47702766,  7.92200342,  8.19703512, 12.18045978,
       12.889695  , 12.49291768, 11.87665501,  8.98765644, 12.30858677,
        6.74872981])}
{'P': tensor([[1.0403e-02, 3.0150e-01, 1.9855e-01,  ..., 2.4198e-06, 1.7448e-02,
         1.3376e-05],
        [2.4816e-04, 1.8611e-02, 1.9472e-03,  ..., 6.1688e-06, 5.4908e-02,
         1.6670e-05],
        [3.5264e-04, 1.4076e-01, 1.7516e-02,  ..., 5.6804e-06, 5.1178e-02,
         1.6320e-05],
        ...,
        [5.3184e-06, 4.7449e-06, 4.8507e-06,  ..., 2.4195e-01, 3.7473e-06,
         2.2025e-05],
        [2.7476e-05, 3.7387e-05, 3.5959e-05,  ..., 2.6570e-04, 4.7396e-02,
         5.3735e-06],
        [1.0637e-01, 6.2888e-02, 6.2042e-02,  ..., 1.9752e-06, 2.5495e-03,
         4.8199e-02]]), 'X': tensor([[ -7.0253,  -7.4493,  -1.5978],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -3.3999, -10.1152,  -1.2088],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.3924, -10.1396,  -4.0863],
        [  8.2154,  -8.4262,   8.0171],
        [ 11.9695,  -2.9407,  -6.3498],
        [  0.6363, -11.0348,   6.1414],
        [  2.4639, -10.9431,  -7.3718],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -9.4716,  -0.6673,   7.6068],
        [  3.3167,   6.6022,   6.3554],
        [ -3.5173,   7.5091,  -8.9585],
        [ -5.6651,   6.4380,   1.3660],
        [  2.3200,   7.1756,   1.1938],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.0310,  12.2795,  -6.4528],
        [ -5.0550,   6.8231,  -9.2502],
        [ -6.4822,  -8.0252,  -3.8983],
        [  6.1929,   2.4501,  -6.9352],
        [  2.8994,   6.8675,  -9.5987],
        [ -3.5173,   7.5091,  -8.9585],
        [  2.1440, -10.9811,   8.5281],
        [  3.2692,  13.5098,  -4.9032],
        [  5.2608,  -2.5256,  -9.1446],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.5763,   0.1333,  -9.2650],
        [ -9.1813,  -3.3550,   1.8827],
        [  7.1821,  -9.1845,   9.8548],
        [  2.5829,  -4.5466,  -8.4015],
        [  4.3523, -10.5232,   8.2138],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -9.3464,   0.5708,  -1.6397],
        [  7.0664,  -9.2590,   8.0879],
        [  0.9445,   7.6736,   0.5095],
        [  5.8275,  -1.5334,   2.0124],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -6.0195,   6.1789,  -2.9496],
        [ 12.3025,  -1.7440,  -6.6632],
        [  2.6451,  -4.5250,  -0.5565],
        [ -3.2486,   7.5925,  -1.3719],
        [  1.9505,  -4.7171,   9.9868],
        [  4.3377,   5.7570,   1.3712],
        [ 11.7162,   5.6728,  -8.9372],
        [  4.7739,  -3.1218,  -4.1188],
        [  1.3403,  -4.8011,  -1.1698],
        [  3.1271, -10.8343,   2.6442],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.0042,  -2.8350,   2.6864],
        [  5.2199,   4.6861,  -1.9304],
        [ -5.6651,   6.4380,   1.3660],
        [  6.1984,  -0.4327,   1.5438],
        [  4.4290,  -3.4596,   1.4181],
        [  0.5250, -11.0306,   5.5358],
        [  6.3087,   1.8053,  -1.9481],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.8836,   3.4315,  -4.0327],
        [  3.3744,   6.5623,  -6.3636],
        [  2.7295,  13.6703,  -8.9328],
        [  5.7218,  -9.9934,   6.5764],
        [ -5.1974,   6.7396,   3.3484],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.4267,  -5.3697,   2.4172],
        [ -9.2687,  -2.9794,   0.6436],
        [ -6.3306,   5.9272,   5.6935],
        [ -5.6423,   6.4537,  -1.2091],
        [  6.3587,   1.1596,  -3.6629],
        [  5.7151,  -1.7700,   3.1542],
        [  6.3505,   0.6235,  -8.5849],
        [  1.1139,   7.6281,  -2.5680],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.2311,  -2.0445,   1.3545],
        [ 11.4703,  -4.2006,   7.4086],
        [  4.1999,  -3.6543,  -4.1998],
        [ -2.6069, -10.4298,  -6.1995],
        [ -2.2188,   7.8218,   8.6064],
        [  0.1047, -11.0046,   0.5188],
        [  3.0719, -10.8449,  -4.1673],
        [ -8.0534,  -6.0429,   2.4583],
        [ -3.4986,   7.5152,   9.0712],
        [ 11.5198,  -4.0923,   5.5088],
        [  2.9410,   6.8428,   5.4458],
        [ -9.2687,  -2.9794,   0.6436],
        [ 11.9641,  -2.9568,   7.7211],
        [ -7.3470,  -7.0615,   5.0066],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.3549,   0.7096,   6.9198],
        [  6.3583,   1.1729,   8.7835],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.4691,   2.8525,  -5.5110],
        [  4.5430,   5.5438,  -3.8669],
        [ -7.4807,   4.7463,   2.0616],
        [  6.3550,   0.7129,   0.4917],
        [  3.6703,   6.3443,  -5.4401],
        [ -6.4368,   5.8356,   3.3014],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.0467,  -9.8369,  -6.5816],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.2592,   3.9425,   2.0383],
        [  0.1047, -11.0046,   0.5188],
        [ -9.4735,  -1.2620,  -3.6742]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.2726e-02, 6.7137e-02,  ..., 1.1717e+00, 3.2498e-01,
         1.5214e-01],
        [5.2726e-02, 3.0339e-04, 1.7743e-02,  ..., 1.1304e+00, 2.7675e-01,
         1.9830e-01],
        [6.7137e-02, 1.7743e-02, 3.0339e-04,  ..., 1.1274e+00, 2.7029e-01,
         2.1541e-01],
        ...,
        [1.1717e+00, 1.1304e+00, 1.1274e+00,  ..., 0.0000e+00, 8.6319e-01,
         1.2300e+00],
        [3.2498e-01, 2.7675e-01, 2.7029e-01,  ..., 8.6319e-01, 0.0000e+00,
         4.3024e-01],
        [1.5214e-01, 1.9830e-01, 2.1541e-01,  ..., 1.2300e+00, 4.3024e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.23946247, 10.33712258, 10.32715715, 10.5575601 , 10.67132524,
       10.60226614, 11.48434767, 11.76830981, 12.32545637, 11.05317597,
       11.21703654, 12.07818871,  9.49511524,  7.38844703,  8.29203356,
        8.57560275,  7.54127425,  5.47866221, 13.68063098,  8.49161154,
       10.31613269,  6.65991877,  7.4544866 ,  8.29203356, 11.18839169,
       13.89974693,  5.83560009, 11.05672136, 12.5769659 ,  9.7751189 ,
       11.65923396,  5.22903117, 11.38774149, 10.23946247,  9.36378438,
       11.64746882,  7.73151309,  6.02588517, 11.66453059,  9.43737523,
        8.62631801, 12.42554808,  5.24137909,  8.25828409,  5.10446653,
        7.20826219, 13.01728313,  5.70405525,  4.98461983, 11.27656488,
        7.6374603 , 12.3344536 ,  7.01475376,  8.57560275,  6.2134961 ,
        5.62005629, 11.04313641,  6.5618968 , 13.04217487,  6.81117802,
        7.37902766, 13.94008948, 11.51557166,  8.51088913,  7.85919687,
        9.99211663,  9.73579738,  8.67227386,  8.5724042 ,  6.46356806,
        5.98285319,  6.38105015,  7.70897675,  5.19577146, 12.40074901,
       12.21532282,  5.5671389 , 10.7506423 ,  8.1303945 , 11.00508681,
       11.27160035, 10.06850677,  8.28968067, 12.22503754,  7.44805901,
        9.73579738, 12.32408516, 10.19036081,  7.88677538,  6.39438044,
        6.46560181,  8.09984231,  9.88310614, 10.00794468, 12.79126785,
        7.16747878,  8.85939626,  6.3949043 ,  7.32950385,  8.68832398,
        5.24461068, 11.54672755, 12.49317563, 12.87752052, 11.00508681,
        9.55720651])}
{'P': tensor([[8.9344e-03, 3.6437e-01, 2.0962e-01,  ..., 9.9070e-05, 1.3764e-07,
         2.4887e-05],
        [8.1355e-05, 9.8119e-03, 4.0755e-01,  ..., 1.8476e-04, 1.6820e-07,
         1.4115e-05],
        [7.2135e-05, 1.1996e-04, 1.4668e-02,  ..., 4.7146e-04, 2.9082e-07,
         1.1647e-05],
        ...,
        [2.3340e-06, 2.9498e-06, 3.8949e-06,  ..., 6.7034e-02, 3.0429e-05,
         3.7746e-07],
        [2.1985e-05, 1.3695e-05, 8.8502e-06,  ..., 4.3510e-07, 1.1964e-01,
         7.6889e-05],
        [1.4263e-02, 1.3288e-02, 1.0736e-02,  ..., 1.4874e-05, 6.6689e-07,
         6.9820e-02]]), 'X': tensor([[ -6.9838,  -7.4965,   0.9796],
        [ -6.0844,  -8.3955,   0.7705],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.9456,  -9.8542,   0.7466],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.9508,  -9.8515,   1.7923],
        [  8.7131,  10.2216,   7.7915],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.5800,   5.5034,  -3.9432],
        [ 10.5057,  -5.9099,  -5.1599],
        [  5.4426, -10.1181,  -5.1268],
        [  6.3211,   1.7017,  -0.2164],
        [ -9.3464,   0.5708,  -1.6397],
        [  2.0547,   7.2957,  -1.9343],
        [  5.6022,  -1.9833,   8.3081],
        [  0.9788,  -4.8145,  -3.8661],
        [ -9.3411,  -2.6004,  -2.7062],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.1623,  -3.4288,  -4.9961],
        [  4.0642,  -3.7599,  -3.4565],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.9410,   6.8428,   5.4458],
        [  5.1203,  -2.7151,   6.2110],
        [  2.4932,   7.0903,  -5.4214],
        [  4.7544,  -3.1425,  -1.5315],
        [ -9.4366,  -0.1189,  -2.6736],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.7500,   7.4187,   4.4237],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.0042,  -2.8350,   2.6864],
        [  5.9548,  -1.2273,  -2.6520],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.8201,   3.5841,  -8.0609],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.0330,  -6.0765,   6.7945],
        [  5.5424,  -2.0885,   7.2487],
        [  9.7328,   9.0902,  -0.8917],
        [ -1.9646,   7.8571,   3.3817],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.2242,  -0.3171,   7.4978],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.9655,   3.2172,  -7.5953],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.4602,   4.7718,  -4.6277],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.2555,   2.1474,   4.1685],
        [ 11.7794,   5.5086,  -9.6929],
        [  3.1756, -10.8247,   6.8882],
        [  5.2911,  -2.4824,  -3.6877],
        [ 12.1940,  -2.1881,   7.1499],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.1677,  -4.7447,  -6.1529],
        [ -5.6651,   6.4380,   1.3660],
        [ -8.8631,  -4.3728,  -1.1746],
        [  2.5813,  -4.5471,   5.3597],
        [  0.5960, -11.0335,   3.1144],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -8.9942,  -3.9990,   6.0448],
        [ 10.9875,   7.2207,  -8.8549],
        [  9.8082,  -6.8396,   4.9615],
        [  2.2659,  13.7894,   6.5436],
        [  0.6470, -11.0352,  -1.1931],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.4196,   0.0478,   9.4140],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -6.2668,   5.9808,   2.5614],
        [ 10.5678,   7.9261,  -0.6100],
        [ -4.6393,   7.0461,   1.1624],
        [  5.3860,  -2.3418,   1.0826],
        [ -0.1963,   7.8765,  -1.1371],
        [  3.2701,   6.6338,   9.9209],
        [  5.4336,  -2.2676,   5.6933],
        [ -4.7847,  -9.3729,  -3.7741],
        [  5.9193,  -1.3176,   5.1739],
        [  4.1537,   5.9340,  -6.0423],
        [  6.2242,  -0.3171,   7.4978],
        [ 10.9394,   7.3069,  -8.9365],
        [  9.3624,   9.5318,   0.2779],
        [ 12.2311,  -2.0445,   1.3545],
        [  3.9770, -10.6341,   8.0783],
        [  0.8572,  14.0501,  -8.3364],
        [  6.2365,  -0.2578,  -6.0212],
        [  0.1287, -11.0065,   6.5757],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.9956,  11.6590,  -7.6876],
        [ -3.7964,  -9.9294,   7.7147],
        [  2.0243,   7.3086,   7.5432],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.1984,  -0.4327,   1.5438],
        [  1.9241,   7.3503,   7.9004],
        [  0.5175,  14.0910,   4.6540],
        [  3.3633,  13.4794,   0.5193],
        [  2.6310, -10.9195,   3.8054],
        [  6.1741,  -0.5329,   3.1913],
        [  9.0051,   9.9234,   5.4411],
        [ -9.1813,  -3.3550,   1.8827],
        [  8.1043,  -8.5164,  -7.9290],
        [  3.3633,  13.4794,   0.5193],
        [ -9.1795,  -3.3623,   8.5259]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0230, 0.0557,  ..., 0.9538, 1.1603, 0.5079],
        [0.0230, 0.0000, 0.0329,  ..., 0.9312, 1.1400, 0.5180],
        [0.0557, 0.0329, 0.0000,  ..., 0.8983, 1.1147, 0.5389],
        ...,
        [0.9538, 0.9312, 0.8983,  ..., 0.0000, 0.7331, 1.3181],
        [1.1603, 1.1400, 1.1147,  ..., 0.7331, 0.0000, 1.2415],
        [0.5079, 0.5180, 0.5389,  ..., 1.3181, 1.2415, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.24557047, 10.36844813, 10.48840476, 10.61472636, 10.48721382,
       10.61417181, 13.43127131,  5.01114719,  7.15990676, 12.05395056,
       11.4890861 ,  6.54615887,  9.36378438,  7.57946459,  5.94293185,
        4.91296205,  9.69629   ,  5.40689332,  9.78286733,  5.5366698 ,
       13.13205357,  7.44805901,  5.79562691,  7.5158489 ,  5.69913817,
        9.43737523,  5.64464807, 12.52875793, 10.04916678,  7.62232307,
        8.92098752,  9.55720651,  5.19577146, 12.3344536 ,  6.07992623,
        9.77777346,  6.83515201,  4.91652556, 10.07238463,  5.92281142,
       13.31764181,  8.09899874, 12.53047489,  6.2322769 , 12.53285633,
        6.77777606,  8.16714836,  8.85572292,  8.4486771 ,  6.61386436,
       13.00380624, 11.28093507,  5.84450127, 12.38881635, 10.27716434,
        6.27905292,  4.74770899,  8.57560275,  9.88310614,  5.2287058 ,
       11.04953767,  9.58072286,  9.84314712, 13.14777433, 11.95743558,
       13.97430191, 11.05413545,  8.50268322, 12.26997165,  9.55593374,
        9.41970031, 12.12448912,  8.66273171, 13.20988996,  8.43626114,
        5.87305005,  7.8788965 ,  7.39599986,  5.88782913, 10.52357479,
        6.06416957,  7.24326992,  6.2322769 , 13.15525803, 13.3607379 ,
       12.40074901, 11.35345345, 14.07623266,  6.24187044, 11.00726483,
       13.1471313 , 13.59673406, 10.63038565,  7.58378655,  8.20436764,
        6.2134961 ,  7.59795445, 14.10045797, 13.89264822, 11.23201619,
        6.19707944, 13.40024224,  9.7751189 , 11.75618513, 13.89264822,
        9.77588801])}
{'P': tensor([[5.1152e-03, 2.0460e-01, 1.7558e-01,  ..., 7.1472e-03, 8.8461e-06,
         2.9520e-06],
        [3.4169e-05, 3.8776e-03, 2.4270e-01,  ..., 9.7681e-03, 1.0109e-05,
         2.5825e-06],
        [4.6543e-05, 7.2306e-05, 6.4046e-03,  ..., 1.9452e-02, 1.9329e-05,
         4.5166e-06],
        ...,
        [1.5719e-05, 2.3078e-05, 2.6490e-05,  ..., 3.0173e-02, 6.6679e-04,
         5.6279e-05],
        [8.3555e-07, 8.3114e-07, 8.3962e-07,  ..., 1.6072e-06, 4.7791e-02,
         1.1129e-02],
        [7.4889e-05, 5.8666e-05, 5.4563e-05,  ..., 2.1187e-05, 1.2594e-04,
         1.8529e-01]]), 'X': tensor([[-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0285, 0.0406,  ..., 0.3730, 1.1209, 0.9926],
        [0.0285, 0.0000, 0.0141,  ..., 0.3471, 1.1263, 0.9941],
        [0.0406, 0.0141, 0.0000,  ..., 0.3333, 1.1202, 0.9864],
        ...,
        [0.3730, 0.3471, 0.3333,  ..., 0.0000, 1.0951, 0.9236],
        [1.1209, 1.1263, 1.1202,  ..., 1.0951, 0.0000, 0.1941],
        [0.9926, 0.9941, 0.9864,  ..., 0.9236, 0.1941, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.26327895, 10.37998078, 10.41909211, 10.51340829, 10.48721382,
       10.68416401,  8.89458273, 12.27993098, 13.97430191,  7.92200342,
       11.86239961, 11.77124373,  5.35774573, 12.42554808,  7.71075226,
       12.53047489,  7.32754882, 10.51340829,  9.15976746,  8.47371868,
       13.97779839,  6.06416957, 10.51437947,  5.8828986 ,  9.4997987 ,
        6.65368551, 13.12299145,  6.49538841,  9.7712321 ,  6.00663314,
       13.2778378 ,  5.47702766, 10.86693708,  9.15253344, 10.27716434,
        9.95552626,  5.14371005, 10.6910768 , 10.9036226 ,  5.37343173,
       11.22596994, 14.03505624, 13.61801526, 13.2778378 , 11.32149259,
       11.05672136, 10.14044489, 13.94008948,  9.91668182,  6.74872981,
        6.81084724,  8.36445758,  6.19633139, 12.10942412,  5.47702766,
       12.24467041, 10.67132524,  9.15866607, 10.09979747,  8.85939626,
        5.92281142, 11.4890861 , 11.83570132,  6.67121748, 13.36800073,
        8.67502348,  9.68324553,  5.86519525,  9.43112957,  6.99097013,
       13.95472331,  5.08910491,  6.35029274, 10.41909211, 13.09737053,
        6.28835101, 11.81423161, 12.40074901, 14.09176259,  8.85939626,
        7.14837514,  5.91408504,  8.0550286 ,  9.88571713, 13.3426541 ,
       13.96905756, 10.06407839, 10.27572673, 13.36800073,  7.98469166,
        6.7166297 ,  5.16611782, 13.34587027,  9.20340031,  6.38105015,
        5.39450305, 11.04953767,  4.76969364, 12.52372629,  5.40689332,
       13.26590071, 11.05672136,  6.24187044, 11.05317597, 12.30858677,
        7.01475376])}
{'P': tensor([[9.3302e-03, 2.8698e-01, 2.4750e-01,  ..., 2.2807e-05, 4.3547e-05,
         8.3655e-07],
        [9.4056e-05, 9.2354e-03, 4.5229e-01,  ..., 1.1287e-05, 1.9950e-05,
         1.2682e-06],
        [6.4436e-05, 1.0410e-04, 1.2216e-02,  ..., 9.6817e-06, 1.6240e-05,
         2.2416e-06],
        ...,
        [2.9579e-03, 2.2859e-03, 2.2822e-03,  ..., 1.4312e-02, 4.2499e-01,
         1.1640e-07],
        [8.0911e-03, 7.0424e-03, 7.6795e-03,  ..., 4.3694e-04, 3.5388e-02,
         3.7045e-07],
        [1.3697e-06, 1.1712e-06, 9.9931e-07,  ..., 1.0925e-06, 8.9803e-07,
         7.8558e-02]]), 'X': tensor([[ -6.8466,  -7.6488,   1.9648],
        [ -5.9225,  -8.5355,   2.4778],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.9508,  -9.8515,   1.7923],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.4488, -10.0934,   2.5195],
        [ 12.6035,   0.7494,   7.1536],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -5.9938,  -8.4745,   3.9735],
        [  5.6022,  -1.9833,   8.3081],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.2419,  -0.2313,   4.2440],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.1459,   1.4780,  -4.2403],
        [  1.9505,  -4.7171,   9.9868],
        [ -0.5739,   7.9056,   0.6382],
        [  6.9653,  -9.3225,  -3.4727],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.0047,   3.1062,  -6.7914],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 12.3759,  -1.3943,   6.4659],
        [ -1.9487,   7.8590,  -3.3330],
        [  0.1931,  -4.7491,   7.4385],
        [  6.2675,  12.1379,  -8.9720],
        [  2.3376,  -4.6231,   9.8436],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -8.3717,  -5.4769,  -9.2228],
        [  3.2159, -10.8166,  -1.0406],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.3357,   3.6001,   0.7903],
        [  9.5203,   9.3483,   5.8781],
        [ -5.7347,  -8.6908,   9.2878],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -2.0721,   7.8432,   6.6380],
        [ 12.4877,  -0.7222,   8.8347],
        [  2.2076,  13.8032,  -1.5099],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.6038,   0.7665,  -3.4439],
        [ 11.5568,   6.0602,   9.3244],
        [  2.5813,  -4.5471,   5.3597],
        [  2.6310, -10.9195,   3.8054],
        [  2.6321,  -4.5296,  -2.5159],
        [ -9.4159,  -2.0739,   9.6841],
        [  1.9505,  -4.7171,   9.9868],
        [  6.9653,  -9.3225,  -3.4727],
        [  3.1064,  13.5607,   6.7605],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.2038,   2.4020,  -3.1692],
        [ -7.4807,   4.7463,   2.0616],
        [  2.6613,   7.0021,   4.5783],
        [  5.6672,  -1.8631,   8.1020],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.4150,  13.9647,  -4.0915],
        [ 11.6544,   5.8273,  -3.1375],
        [ -6.4368,   5.8356,   3.3014],
        [  2.8890,  -4.4316,   1.1600],
        [  5.2939,  -2.4784,  -4.5840],
        [  2.8994,   6.8675,  -9.5987],
        [  8.7715,  -7.9390,   5.2900],
        [  2.8602,   6.8905,   5.8779],
        [ -9.4735,  -1.2620,  -3.6742],
        [  4.9337,  12.8566,  -3.6582],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.1231,  -3.5739,   8.4708],
        [ -2.2188,   7.8218,   8.6064],
        [  9.9756,  -6.6330,  -3.3429],
        [  7.8982,  -8.6778,   0.0239],
        [ -7.4808,   4.7462,   2.0266],
        [  2.3261,  13.7749,   5.9121],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.2345,  -3.1345,   3.9715],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.7565,  11.8237,  -4.9882],
        [  5.4336,  -2.2676,   5.6933],
        [ -8.1666,   3.7653,   3.6075],
        [ 12.5717,   1.9266,   7.4903],
        [  2.1440, -10.9811,   8.5281],
        [ -9.1795,  -3.3623,   8.5259],
        [  4.1637,  13.1898,  -1.9189],
        [  3.1064,  13.5607,   6.7605],
        [ -7.9298,   4.1367,  -3.9773],
        [  3.9913,  -3.8139,  -6.8311],
        [  5.6258,  -1.9404,  -6.0715],
        [ -1.0482, -10.8487,  -5.9066],
        [ -0.2586,   7.8825,  -9.9840],
        [ 11.2209,  -4.7092,   2.6117],
        [ 10.1983,  -6.3430,  -0.3200],
        [  0.7958,  -4.8109,  -8.2248],
        [  4.0587, -10.6112,   6.2273],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -2.0229, -10.6163,   8.6930],
        [  5.8902,  -1.3887,   4.0388],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.8934,   2.2597,   3.5831],
        [ -1.8367,   7.8718,  -8.6523],
        [ 12.2914,   3.8037,   6.0978],
        [  0.1047, -11.0046,   0.5188],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 11.5568,   6.0602,   9.3244]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0433, 0.0514,  ..., 0.5779, 0.5665, 1.3920],
        [0.0433, 0.0000, 0.0207,  ..., 0.6191, 0.6080, 1.3586],
        [0.0514, 0.0207, 0.0000,  ..., 0.6170, 0.6064, 1.3426],
        ...,
        [0.5779, 0.6191, 0.6170,  ..., 0.0000, 0.0174, 1.7697],
        [0.5665, 0.6080, 0.6064,  ..., 0.0174, 0.0000, 1.7698],
        [1.3920, 1.3586, 1.3426,  ..., 1.7697, 1.7698, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.26545792, 10.38897563, 10.48721382, 10.61417181, 10.48840476,
       10.66631784, 12.62576026, 13.12299145,  9.97964013, 10.37998078,
        5.94293185,  4.91652556,  6.24614905, 13.79676654,  9.26455795,
        5.10446653,  7.92644746, 11.63724792, 12.62711252, 12.91505111,
        6.76056625,  5.05989547,  9.85330646, 12.45418236,  8.09704052,
        4.75303127, 13.66051884,  5.1805345 ,  9.66977802, 10.00410625,
       11.2845608 ,  6.14086582, 12.52875793, 12.77402351, 12.85032836,
       13.3426541 , 10.412299  ,  9.93553424,  8.11227668, 12.5085995 ,
       13.97857591, 13.77222752, 12.62711252, 13.0493368 ,  5.2287058 ,
       11.23201619,  5.23878371,  9.64157461,  5.10446653, 11.63724792,
       13.91198191, 10.06437598,  6.65259775,  8.85939626,  7.49076899,
        5.96556248, 13.12299145, 14.03618649, 13.03002338,  8.68832398,
        5.29009836,  5.84533345,  7.4544866 , 11.83075005,  7.46053281,
        9.55720651, 13.77075174, 12.64235359,  9.79814997,  8.1303945 ,
       11.97957425, 11.73397309,  8.859416  , 13.96987806,  8.97052115,
        9.75200848,  8.47371868, 13.61801526,  5.88782913,  8.99276564,
       12.71843462, 11.18839169,  9.77588801, 13.83139024, 13.91198191,
        8.94395628,  5.5205189 ,  5.9510489 , 10.8992524 ,  7.88677538,
       12.16901615, 12.00995898,  4.8763135 , 11.36089355, 12.3828957 ,
       12.53285633, 10.80728788,  6.05165248,  9.70108409,  9.17595899,
        8.08320235, 12.86648771, 11.00508681,  9.7712321 ,  9.88571713,
       13.0493368 ])}
{'P': tensor([[5.0332e-03, 3.0373e-01, 2.6270e-01,  ..., 1.0917e-06, 2.9609e-04,
         7.1336e-06],
        [7.6305e-05, 6.5467e-03, 4.1663e-01,  ..., 1.3172e-06, 3.5826e-04,
         7.9607e-06],
        [1.2403e-04, 1.4452e-04, 1.2913e-02,  ..., 2.4216e-06, 6.6186e-04,
         1.3600e-05],
        ...,
        [9.0164e-03, 7.5961e-03, 6.4222e-03,  ..., 1.4219e-01, 1.8443e-04,
         1.5487e-01],
        [5.0648e-05, 6.0385e-05, 7.2429e-05,  ..., 1.3155e-06, 3.6527e-02,
         4.3128e-06],
        [1.9399e-01, 1.6466e-01, 1.3915e-01,  ..., 6.2496e-05, 1.0122e-03,
         5.7799e-02]]), 'X': tensor([[ -6.7745,  -7.7264,   8.4684],
        [ -6.4844,  -8.0230,   8.5209],
        [ -6.1914,  -8.2997,   8.5380],
        [ -5.6169,  -8.7844,   8.9057],
        [ -5.7347,  -8.6908,   9.2878],
        [ -5.3179,  -9.0101,   9.4472],
        [ -7.9925,  -6.1423,   2.8990],
        [  1.8721,  -4.7322,   0.5524],
        [ -3.2736, -10.1703,   3.8712],
        [ -3.2736, -10.1703,   3.8712],
        [ -4.8751,  -9.3148,   4.0406],
        [ -9.1813,  -3.3550,   1.8827],
        [ -7.6293,  -6.6859,  -5.5161],
        [  3.3069,   6.6089,   8.7377],
        [  9.3836,   9.5075,  -5.9278],
        [  6.2630,   2.1057,  -5.2136],
        [  4.1140,  -3.7220,  -0.1273],
        [ -9.4157,   0.0821,  -5.0025],
        [  5.1292,   4.8183,  -5.7997],
        [ -3.4488, -10.0934,   2.5195],
        [ -1.0407,   7.9166,  -6.6021],
        [ 12.2692,  -1.8882,   3.9253],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.8363,   2.4086,  -5.6296],
        [ 11.9122,   5.1397,  -1.5498],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.2811,  -7.4336,   5.1771],
        [  3.2809,  13.5061,  -4.0199],
        [ 12.5770,   1.8512,  -0.6578],
        [ -1.8823, -10.6558,  -3.8952],
        [  4.5503,   5.5359,   7.4735],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.9074,  -4.8139,   8.1674],
        [  3.6249, -10.7255,   5.6605],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.5600, -11.0321,  -0.8660],
        [  5.8201,   3.5841,  -8.0609],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.9646,   7.8571,   3.3817],
        [ -3.3999, -10.1152,  -1.2088],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -4.0868,   7.2981,   8.1874],
        [ 10.5678,   7.9261,  -0.6100],
        [  4.9531,   5.0576,   5.6794],
        [  2.3261,   7.1726,   6.2522],
        [  2.1120,  -4.6818,   8.6144],
        [  9.1820,   9.7334,  -2.7932],
        [  4.9337,  12.8566,  -3.6582],
        [  4.0362,  13.2397,  -7.0030],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.4203,  -2.0340,   2.5856],
        [  6.2675,  12.1379,  -8.9720],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.9058,  -6.7203,   5.6928],
        [  3.1480,  -4.3165,  -7.8002],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.0310,  12.2795,  -6.4528],
        [  4.8653,   5.1694,  -4.0823],
        [  0.4996,   7.7738,  -4.4711],
        [  5.8201,   3.5841,  -8.0609],
        [  8.5716,  10.3596,   6.8818],
        [  6.1741,  -0.5329,   3.1913],
        [ -2.8162,   7.7059,  -9.6885],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.5859,  -5.0393,   0.5476],
        [  9.2811,  -7.4336,   5.1771],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 11.0275,  -5.0666,   8.4154],
        [ -3.5173,   7.5091,  -8.9585],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.5650,   2.0132,   4.6272],
        [  2.3373,  -4.6232,  -5.2246],
        [  2.1120,  -4.6818,   8.6144],
        [ -8.9178,   2.1936,   7.5388],
        [  0.9945,  -4.8144,  -0.1897],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.1929,   2.4501,  -6.9352],
        [ -9.2345,  -3.1345,   3.9715],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.2707,   0.9661,   4.4863],
        [  4.1637,  13.1898,  -1.9189],
        [  2.3200,   7.1756,   1.1938],
        [  2.3373,  -4.6232,  -5.2246],
        [ 10.9717,  -5.1646,  -6.9039],
        [  5.4426, -10.1181,  -5.1268],
        [ -9.3604,   0.4855,  -3.0158],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.6314,  -1.9302,  -9.6543],
        [  4.6992,   5.3690,   3.0935],
        [ -2.0229, -10.6163,   8.6930],
        [ 12.1940,  -2.1881,   7.1499],
        [ 11.0438,   7.1180,  -5.6991],
        [ 12.3759,  -1.3943,   6.4659],
        [ -9.1178,   1.5784,   2.1410],
        [  1.9312,  13.8649,   8.3563],
        [  1.3670,  -4.7990,   3.2603],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.1929,   2.4501,  -6.9352],
        [ -4.1296,   7.2803,  -2.7194],
        [  0.9809,   7.6642,  -1.4217],
        [  2.8890,  -4.4316,   1.1600],
        [ -6.7992,   5.4991,   9.9186],
        [ -2.6735, -10.4062,  -1.5911],
        [ -9.4159,  -2.0739,   9.6841]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.0785e-03, 2.3834e-02,  ..., 3.1833e-01, 7.5851e-01,
         2.0335e-02],
        [8.0785e-03, 0.0000e+00, 1.6153e-02,  ..., 3.1104e-01, 7.6011e-01,
         1.5403e-02],
        [2.3834e-02, 1.6153e-02, 3.0339e-04,  ..., 2.9492e-01, 7.5739e-01,
         1.1213e-02],
        ...,
        [3.1833e-01, 3.1104e-01, 2.9492e-01,  ..., 0.0000e+00, 7.4804e-01,
         2.9901e-01],
        [7.5851e-01, 7.6011e-01, 7.5739e-01,  ..., 7.4804e-01, 2.1453e-04,
         7.4805e-01],
        [2.0335e-02, 1.5403e-02, 1.1213e-02,  ..., 2.9901e-01, 7.4805e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.27572673, 10.31584102, 10.35464916, 10.42667093, 10.412299  ,
       10.46238054, 10.08001894,  5.08910491, 10.68416401, 10.68416401,
       10.51340829,  9.7751189 , 10.14436912,  7.39004226, 13.35832725,
        6.60752373,  5.54776934,  9.41605684,  7.03733775, 10.66631784,
        7.98469166, 12.41366754, 12.27993098,  9.15866607, 12.97370371,
        9.85384623,  5.18277858, 11.89105259, 13.89886469, 12.7125079 ,
       10.8207297 ,  7.16598273, 10.33712258,  4.89869205, 11.32149259,
        9.95552626, 11.04629471,  6.83515201,  6.61211224,  8.09899874,
       10.67132524, 12.62576026,  9.75075273,  8.36445758, 13.20988996,
        7.07902036,  7.5403822 ,  5.13616224, 13.38091121, 13.77075174,
       13.84126197, 12.35421461,  9.63742669, 13.66051884,  5.18277858,
       11.9702762 ,  5.34250225, 12.16901615, 13.68063098,  7.09887557,
        7.78980299,  6.83515201, 13.44593519,  6.19707944,  8.20436764,
       11.81423161,  9.95552626, 11.89105259, 10.60226614, 12.13568598,
        8.29203356, 12.87752052, 12.72524615,  5.18047638,  5.13616224,
        9.18358562,  4.9160962 ,  9.68323686,  6.65991877,  9.75200848,
       13.20597932,  9.32094055, 13.83139024,  7.54127425,  5.18047638,
       12.12641969, 11.4890861 ,  9.37296102, 11.80092578,  5.95297459,
        7.13501264, 10.80728788, 12.38881635, 13.13888841, 12.45418236,
        9.25336371, 13.99877413,  4.98989906, 12.12448912,  6.65991877,
        8.36995953,  7.72668467,  5.29009836,  8.74470218, 10.74409834,
        9.64157461])}
{'P': tensor([[8.6981e-03, 3.9549e-01, 1.9494e-01,  ..., 1.0594e-06, 5.5824e-07,
         3.1208e-06],
        [1.3978e-04, 1.5373e-02, 4.2061e-01,  ..., 1.8715e-06, 1.2487e-06,
         5.9669e-06],
        [7.4351e-05, 1.3509e-04, 1.6391e-02,  ..., 2.4115e-06, 2.1521e-06,
         1.0190e-05],
        ...,
        [6.0314e-05, 4.3993e-05, 3.2676e-05,  ..., 7.0566e-02, 5.6359e-05,
         5.3247e-05],
        [8.6900e-07, 6.7765e-07, 5.0595e-07,  ..., 5.5483e-03, 2.6439e-02,
         2.5442e-06],
        [3.6664e-06, 3.6224e-06, 4.8010e-06,  ..., 1.0048e-03, 2.4538e-04,
         9.6782e-02]]), 'X': tensor([[ -6.7643,  -7.7372,  -9.6663],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -3.5903,   7.4847,  -8.9112],
        [ -9.3107,  -2.7697,  -3.2919],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.3569,   0.7591,   6.5011],
        [  0.6757, -11.0361,  -1.3047],
        [  2.7420,  -4.4896,  -3.7139],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.7544,   2.6096,   3.3146],
        [  5.9193,  -1.3176,   5.1739],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.5101,  -2.1433,   9.5442],
        [ 11.2571,   6.7072,   8.3693],
        [  4.4633,   5.6287,  -0.7754],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.4732,  -2.2041,  -4.3830],
        [ -4.3963,   7.1629,   4.5263],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.8363,   2.4086,  -5.6296],
        [ -5.6169,  -8.7844,   8.9057],
        [  4.1999,  -3.6543,  -4.1998],
        [  5.3924, -10.1396,  -4.0863],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.6319,  11.9066,   3.5165],
        [  6.3534,   1.2990,  -7.2942],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.4721,  -1.3097,  -7.4377],
        [  6.8407,  11.7666,   3.0045],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -3.0644,   7.6439,  -1.4322],
        [ 12.4013,  -1.2597,  -7.6474],
        [  3.2159, -10.8166,  -1.0406],
        [  5.8318,   3.5567,  -4.7692],
        [  2.4356,  -4.5942,  -5.0808],
        [ -4.6393,   7.0461,   1.1624],
        [  1.3103,  -4.8032,  -0.9300],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.3357,   3.6001,   0.7903],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.3246,   1.6696,  -0.1544],
        [ 11.9641,  -2.9568,   7.7211],
        [  2.1824,  -4.6647,   1.1985],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 12.1504,   4.3674,  -7.2998],
        [ 10.2712,   8.3707,  -4.9743],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 12.3357,   3.6001,   0.7903],
        [  0.7958,  -4.8109,  -8.2248],
        [ -8.6229,   2.9068,   2.4429],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.2188,  -2.5839,   2.1833],
        [  5.7845,   3.6652,   0.9175],
        [  5.4429,  -2.2529,   9.7306],
        [  2.6321,  -4.5296,  -2.5159],
        [  6.0047,   3.1062,  -6.7914],
        [  2.9461,  -4.4077,  -0.4831],
        [  7.8425,  -8.7202,   6.4530],
        [  0.5250, -11.0306,   5.5358],
        [ -8.1905,   3.7253,   4.1175],
        [  4.1297,  -3.7098,  -3.9838],
        [ -1.0407,   7.9166,  -6.6021],
        [ -6.7992,   5.4991,   9.9186],
        [  0.1041, -11.0045,   2.4311],
        [ 12.5154,   2.5072,  -9.3274],
        [  6.3589,   0.8239,  -8.4230],
        [  6.3305,   1.6119,   1.1538],
        [  0.9445,   7.6736,   0.5095],
        [  3.6817,   6.3354,   6.1233],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -2.3317,   7.8034,  -3.6274],
        [  0.9788,  -4.8145,  -3.8661],
        [ 11.9641,  -2.9568,   7.7211],
        [  3.2809,  13.5061,  -4.0199],
        [  2.5318,  13.7231,   6.3692],
        [  9.3424,  -7.3685,  -3.6807],
        [  4.3554,   5.7393,  -9.0766],
        [ -9.4721,  -1.3097,  -7.4377],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.6165,  -3.8715,   6.4594],
        [  3.4401,  13.4540,   4.6576],
        [ -8.7544,   2.6096,   3.3146],
        [  3.1271, -10.8343,   2.6442],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.6093,  -1.9705,   6.4807],
        [  4.2339,  -3.6268,   6.3125],
        [ 11.3691,   6.4760,   1.0787],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.8631,  -4.3728,  -1.1746],
        [  7.1697,  11.5343,  -0.7276],
        [ 10.8218,  -5.4177,   6.8573],
        [ -6.6998,   5.5954,   0.0823],
        [ 12.5224,  -0.4497,   4.8448],
        [ -8.0829,  -5.9940,   5.0564],
        [  4.4039,   5.6903,   8.7366],
        [  0.9809,   7.6642,  -1.4217],
        [ 10.0400,   8.6922,  -6.2847],
        [  5.3804,  -2.3504,   7.2744]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0422, 0.0800,  ..., 0.9333, 1.2669, 1.4647],
        [0.0422, 0.0000, 0.0378,  ..., 0.8962, 1.2247, 1.4341],
        [0.0800, 0.0378, 0.0000,  ..., 0.8616, 1.1870, 1.4048],
        ...,
        [0.9333, 0.8962, 0.8616,  ..., 0.0000, 0.6132, 0.6153],
        [1.2669, 1.2247, 1.1870,  ..., 0.6132, 0.0000, 1.0096],
        [1.4647, 1.4341, 1.4048,  ..., 0.6153, 1.0096, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.27716434, 10.37698435, 10.53368644, 10.57302546, 10.4977132 ,
       10.46739073, 10.31613269,  8.30125201,  9.71392179,  5.01114719,
        6.4020404 , 11.05672136,  5.26067019,  9.70108409,  9.69792513,
        9.13507321,  6.06416957,  5.0209005 ,  5.91222346, 13.103709  ,
        7.18354885, 12.77402351,  5.90035324,  8.40445872, 13.03002338,
        5.14999799,  9.15866607, 10.42667093,  5.5671389 , 11.48434767,
       13.00413829, 13.62896593,  6.48486171, 13.66896739,  9.562174  ,
       13.6105595 , 11.81423161, 12.4217523 ,  8.23525433, 12.46513646,
       11.2845608 ,  6.83083863,  5.19986797,  8.43626114,  4.978704  ,
       11.52733581, 12.85032836,  8.12580976,  6.54127431, 12.32408516,
        5.14999799,  9.70108409, 12.91143491, 13.25017211, 10.39942497,
       12.85032836,  4.8763135 ,  9.09963894, 12.27993098,  5.82345867,
        6.84796202,  5.89074104,  5.23878371,  6.76056625,  5.30158497,
       11.72802491, 11.04313641,  8.99791316,  5.5513072 ,  7.98469166,
        8.74470218, 11.0050379 , 12.76408562,  6.41202986,  6.53250937,
        7.73151309,  7.32754882, 10.23946247,  8.14433542,  4.91296205,
       12.32408516, 13.89886469, 13.95472331, 11.89855267,  7.20482224,
        9.562174  ,  4.85603886, 12.24467041, 13.88683436,  9.13507321,
       11.27656488,  9.95552626,  5.94535529,  5.5748624 , 13.08414689,
        8.41148722,  9.88310614, 13.58101794, 12.10222204,  8.72896381,
       12.53047489, 10.06286074,  7.19534146,  7.72668467, 13.27994304,
        5.87132718])}
{'P': tensor([[9.1408e-03, 2.0588e-01, 1.8816e-01,  ..., 7.5108e-07, 4.6065e-07,
         2.6814e-06],
        [5.6723e-05, 5.9790e-03, 3.4745e-01,  ..., 3.3141e-07, 1.6340e-07,
         2.9953e-06],
        [1.5646e-04, 2.5828e-04, 1.9870e-02,  ..., 1.0378e-06, 4.9321e-07,
         1.1102e-05],
        ...,
        [3.7275e-04, 1.7563e-04, 1.5494e-04,  ..., 5.7842e-02, 2.2448e-02,
         1.1529e-06],
        [7.5861e-04, 4.1679e-04, 3.7919e-04,  ..., 1.5731e-03, 3.4610e-01,
         1.8170e-06],
        [7.1724e-07, 5.7991e-07, 5.6079e-07,  ..., 3.5400e-05, 1.3423e-06,
         3.1981e-02]]), 'X': tensor([[ -6.5295,  -7.9785,  -7.4429],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -2.9497, -10.3028,  -7.3479],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -8.0611,  -6.0302,   4.6962],
        [  3.3744,   6.5623,  -6.3636],
        [ 12.2589,  -1.9312,   9.8657],
        [  8.9284,  10.0036,  -6.9210],
        [  4.7544,  -3.1425,  -1.5315],
        [ -5.9940,   6.1985,   5.9566],
        [  6.8407,  11.7666,   3.0045],
        [  6.2770,  -0.0379,   2.4195],
        [ 11.5935,   5.9740,  -4.3952],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.6992,   5.3690,   3.0935],
        [ 12.0582,  -2.6640,   0.7132],
        [ -6.2200,  -8.2737,   5.5079],
        [ -2.0445,   7.8469,  -0.5857],
        [  7.9403,  -8.6455,  -8.7433],
        [ -0.8885,   7.9160,   0.0571],
        [  5.4167, -10.1293,   6.2156],
        [  4.9957,  -2.8705,   6.5978],
        [  0.2571,  -4.7594,   1.5618],
        [ 12.5224,  -0.4497,   4.8448],
        [  9.4554,   9.4245,   9.8800],
        [  6.3534,   1.2990,  -7.2942],
        [  6.2975,   0.0947,   5.1791],
        [  4.1348,   5.9514,  -0.8856],
        [ -9.2974,   0.8371,   2.4852],
        [ -1.6334, -10.7207,   7.1397],
        [  3.8182,  -3.9347,  -1.7961],
        [  8.1217,  -8.5024,  -0.6149],
        [ -9.2974,   0.8371,   2.4852],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.1741,  -0.5329,   3.1913],
        [  3.9913,  -3.8139,  -6.8311],
        [  5.4429,  -2.2529,   9.7306],
        [ 12.2589,  -1.9312,   9.8657],
        [ 10.8218,  -5.4177,   6.8573],
        [  4.5491,  -3.3485,   6.8412],
        [  2.9260,  -4.4162,   8.3808],
        [ -8.3653,   3.4184,   9.3454],
        [  7.1441,  11.5529,   2.3232],
        [  6.3549,   0.7096,   6.9198],
        [ -8.8745,   2.3097,   3.5600],
        [  6.3516,   0.6425,   0.3425],
        [  9.6587,   9.1817,  -0.8212],
        [  3.4140,  -4.1803,  -8.8224],
        [ -9.3616,  -2.4749,   0.1793],
        [  0.7275,  14.0667,  -0.9815],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.1379,  -4.6757,   6.4981],
        [  4.9452, -10.3187,  -7.8621],
        [  6.3205,   1.7068,  -3.8619],
        [  7.0358,  -9.2784,  -9.8479],
        [  6.2036,   2.4030,  -1.2500],
        [  2.4149,  -4.6005,  -4.3339],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.1196,  -2.7160,   4.2705],
        [  5.9526,   3.2525,  -4.3821],
        [  3.3744,   6.5623,  -6.3636],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.1426,   3.8048,   9.3542],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.2675,  12.1379,  -8.9720],
        [  4.6992,   5.3690,   3.0935],
        [  3.2226,  -4.2802,  -4.0428],
        [ 11.3800,  -4.3919,   1.4066],
        [  6.0073,   3.0986,   6.7937],
        [  1.9204,   7.3518,   8.4781],
        [  6.3048,   0.1469,  -9.1691],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -1.6906,   7.8860,   7.7142],
        [ -0.8455, -10.8853,  -6.7400],
        [  9.6587,   9.1817,  -0.8212],
        [  4.4290,  -3.4596,   1.4181],
        [  5.6202,  -1.9508,   0.7471],
        [  0.5960, -11.0335,   3.1144],
        [  9.3624,   9.5318,   0.2779],
        [  1.7815,  -4.7481,   4.4289],
        [ -0.1626, -10.9796,   6.4629],
        [  2.9470,  13.6085,   2.9809],
        [  7.0642,  11.6104,  -9.8316],
        [  6.3360,   0.4258,  -9.2509],
        [ -2.0840,   7.8415,  -7.8657],
        [  3.0876,   6.7527,  -5.3540],
        [ -9.0961,   1.6529,   3.1553],
        [  4.5491,  -3.3485,   6.8412],
        [  3.6703,   6.3443,  -5.4401],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.3122,  -9.6994,  -6.3282],
        [  0.7145,  -4.8071,  -2.3766],
        [  2.3261,  13.7749,   5.9121],
        [ 12.5789,   0.1722,   1.6470],
        [  8.1043,  -8.5164,  -7.9290],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.4572,   7.1085,   7.7376],
        [  3.2692,  13.5098,  -4.9032],
        [ -3.6381,   7.4683,   7.3544],
        [  6.0054,  -1.0902,   8.3079],
        [ -6.2668,   5.9808,   2.5614],
        [ -8.9282,   2.1648,   8.4894],
        [ 12.1504,   4.3674,  -7.2998]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0482, 0.0668,  ..., 0.8314, 1.0966, 1.1915],
        [0.0482, 0.0000, 0.0310,  ..., 0.8595, 1.1359, 1.1702],
        [0.0668, 0.0310, 0.0000,  ..., 0.8446, 1.1304, 1.1392],
        ...,
        [0.8314, 0.8595, 0.8446,  ..., 0.0000, 0.4212, 1.0810],
        [1.0966, 1.1359, 1.1304,  ..., 0.4212, 0.0000, 1.4928],
        [1.1915, 1.1702, 1.1392,  ..., 1.0810, 1.4928, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.30972193, 10.46739073, 10.4977132 , 10.35485331, 10.7167285 ,
       10.57302546, 10.06703358,  7.37902766, 12.41012046, 13.40850328,
        5.69913817,  8.62260327, 13.6105595 ,  6.27715105, 13.04217487,
       10.01804223,  7.13501264, 12.3489324 , 10.35093485,  8.10886021,
       11.73848065,  7.96575782, 11.48663767,  5.76165934,  4.7663526 ,
       12.53047489, 13.35013374,  6.48486171,  6.29821673,  7.24677664,
        9.3349858 , 10.84437705,  5.48276038, 11.75807252,  9.3349858 ,
       13.18278606,  6.19707944,  5.5205189 ,  5.89074104, 12.41012046,
       12.10222204,  5.64865173,  5.29755487,  9.0368458 , 13.58333857,
        6.39438044,  9.17016857,  6.38399789, 13.32645552,  5.39723241,
        9.68323686, 14.08549802,  5.1224079 ,  5.14124612, 11.4424816 ,
        6.54693987, 11.64436684,  6.652742  ,  5.19577146, 12.0495521 ,
        5.79543038,  6.78325151,  7.37902766,  5.13125513,  8.98765644,
       10.4977132 ,  6.25586297, 13.66051884,  7.13501264,  5.35774573,
       12.19805394,  6.7593917 ,  7.5984809 ,  6.30647746,  9.75075273,
        8.06517143, 10.91805683, 13.32645552,  5.62005629,  5.94909709,
       11.04953767, 13.3607379 ,  5.0713364 , 10.98076873, 13.92390689,
       13.59056613,  6.35029274,  8.1137452 ,  7.42512621,  9.24501829,
        5.64865173,  7.32950385, 12.50996828, 11.57249984,  4.85995002,
       13.96987806, 12.58006038, 11.75618513, 12.22276278,  7.52117132,
       13.89974693,  8.30729352,  6.10360073,  8.66273171,  9.18690018,
       12.91143491])}
{'P': tensor([[5.2262e-03, 3.3259e-01, 1.9160e-01,  ..., 4.7526e-07, 2.7147e-07,
         2.0423e-06],
        [9.0970e-05, 8.1282e-03, 2.8412e-01,  ..., 8.2800e-07, 4.3567e-07,
         2.8562e-06],
        [1.2730e-04, 1.8380e-04, 1.2990e-02,  ..., 1.4897e-06, 6.7696e-07,
         3.7706e-06],
        ...,
        [2.2652e-07, 2.2139e-07, 1.9988e-07,  ..., 5.5042e-02, 1.7230e-02,
         1.6236e-06],
        [2.3412e-06, 2.1743e-06, 1.7368e-06,  ..., 2.3853e-05, 3.4399e-02,
         2.6155e-05],
        [3.1761e-02, 2.6542e-02, 1.8911e-02,  ..., 7.7830e-07, 4.6137e-06,
         8.3580e-02]]), 'X': tensor([[ -6.4844,  -8.0230,   8.5209],
        [ -6.1914,  -8.2997,   8.5380],
        [ -5.6169,  -8.7844,   8.9057],
        [ -5.3179,  -9.0101,   9.4472],
        [ -5.7347,  -8.6908,   9.2878],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.0591,   6.0202,  -1.6339],
        [ -9.3616,  -2.4750,   3.7765],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -9.3444,   0.5821,   6.9101],
        [  6.3516,   0.6425,   0.3425],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.2608,  12.6977,   4.9166],
        [  4.9337,  12.8566,  -3.6582],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.6038,   0.7665,  -3.4439],
        [  2.5318,  13.7231,   6.3692],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.7228,  -4.7575,   8.3396],
        [ -7.2515,   5.0195,   4.6434],
        [  3.4604,  -4.1545,   1.8091],
        [  6.3535,   0.6792,  -7.5101],
        [  1.3403,  -4.8011,  -1.1698],
        [ -9.4157,   0.0821,  -5.0025],
        [ -2.1382,   7.8339,   7.3524],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.6613,   7.0021,   4.5783],
        [ -3.1204, -10.2345,   9.2288],
        [  7.1821,  -9.1845,   9.8548],
        [  4.3377,   5.7570,   1.3712],
        [ -9.3384,  -2.6161,  -4.7630],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.4501,  -1.7133,   2.2980],
        [  1.6232,  -4.7717,  -2.5655],
        [  3.9022,  13.2906,   5.5788],
        [ -7.7239,   4.4298,   3.1216],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.7590,  13.6621,   9.4496],
        [  2.3589,  13.7668,  -3.6501],
        [  4.9957,  -2.8705,   6.5978],
        [ 11.7349,  -3.5838,   2.1644],
        [ -7.3470,  -7.0615,   5.0066],
        [  9.4925,   9.3811,   8.9842],
        [ -1.5261,   7.8988,   9.3768],
        [  7.5930,  -8.9037,   7.2323],
        [ -8.8400,   2.3991,   1.3457],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.3691,   6.4760,   1.0787],
        [ -0.1963,   7.8765,  -1.1371],
        [ -3.1671,   7.6158,   6.0765],
        [ -7.3333,   4.9246,  -7.9212],
        [ 11.4235,   6.3594,   5.1663],
        [  9.8082,  -6.8396,   4.9615],
        [ -3.9456,  -9.8542,   0.7466],
        [  0.1252, -11.0062,   7.2564],
        [  3.2990,  -4.2415,  -7.4857],
        [  6.3587,   1.1596,  -3.6629],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.5375,   7.0676,  -4.3490],
        [  1.7903,  -4.7467,   0.1168],
        [ -7.9298,   4.1367,  -3.9773],
        [ 12.5919,   0.4035,  -5.8460],
        [  6.3497,   1.3680,  -1.9076],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -0.2586,   7.8825,  -9.9840],
        [ -1.0012, -10.8576,   2.2867],
        [  9.5203,   9.3483,   5.8781],
        [  5.8450,  -9.9356,   3.0767],
        [  7.5930,  -8.9037,   7.2323],
        [  5.4242, -10.1261,   3.9595],
        [  0.9659,  -4.8144,  -0.9594],
        [  3.3744,   6.5623,  -6.3636],
        [ -2.1817,   7.8274,  -4.6298],
        [  0.1252, -11.0062,   7.2564],
        [ -8.3539,  -5.5111,   0.3373],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.0955,   2.8210,  -4.3321],
        [ -7.6719,  -6.6260,   7.1468],
        [ -3.5520,   7.4976,   9.1522],
        [  9.8082,  -6.8396,   4.9615],
        [  5.8979,   3.3958,  -3.8713],
        [  6.9653,  -9.3225,  -3.4727],
        [ -2.8162,   7.7059,  -9.6885],
        [ 12.5224,  -0.4497,   4.8448],
        [  3.8111, -10.6787,  -4.6539],
        [  6.1737,   2.5305,   3.9049],
        [  2.5375,   7.0676,  -4.3490],
        [ -5.5525,   6.5146,  -4.4984],
        [  8.6271,  -8.0715,  -7.4813],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 10.8218,  -5.4177,   6.8573],
        [ 12.5770,   1.8512,  -0.6578],
        [  4.9152,  12.8653,  -1.4459],
        [ -7.9925,  -6.1423,   2.8990],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.0073,   3.0986,   6.7937],
        [  6.9956,  11.6590,  -7.6876],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.6059,   0.9597,  -6.3264],
        [  3.9895,   6.0816,  -6.5235],
        [ -8.9282,   2.1648,   8.4894]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.6153e-02, 4.8410e-02,  ..., 1.5302e+00, 1.2824e+00,
         1.5006e-01],
        [1.6153e-02, 3.0339e-04, 3.2533e-02,  ..., 1.5190e+00, 1.2720e+00,
         1.3444e-01],
        [4.8410e-02, 3.2533e-02, 0.0000e+00,  ..., 1.5026e+00, 1.2577e+00,
         1.0700e-01],
        ...,
        [1.5302e+00, 1.5190e+00, 1.5026e+00,  ..., 0.0000e+00, 2.6581e-01,
         1.4051e+00],
        [1.2824e+00, 1.2720e+00, 1.2577e+00,  ..., 2.6581e-01, 0.0000e+00,
         1.1641e+00],
        [1.5006e-01, 1.3444e-01, 1.0700e-01,  ..., 1.4051e+00, 1.1641e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.31584102, 10.35464916, 10.42667093, 10.46238054, 10.412299  ,
       10.63038565,  7.26074925,  9.68324553, 10.32715715,  9.36256096,
        6.38399789, 13.03002338, 13.744387  , 13.77075174,  5.64464807,
       12.62711252, 13.95472331, 13.12299145,  5.05981914,  8.81930525,
        5.40689332,  6.38968357,  4.98461983,  9.41605684,  8.12043983,
        7.9337647 , 13.25336001,  7.49076899, 10.69963954, 11.65923396,
        7.20826219,  9.69792513, 11.80092578,  9.60412735,  5.04028538,
       13.85158417,  8.90405823,  9.91668182, 13.93790235, 13.96746816,
        5.76165934, 12.26997165, 10.19036081, 13.34587027,  8.04483495,
       11.70168755,  9.15976746, 13.61801526, 13.08414689,  7.8788965 ,
        8.24808832,  8.83341608, 13.0743399 , 11.95743558, 10.61472636,
       11.00694688,  5.37343173,  6.46356806,  7.98469166,  5.21280156,
        7.5092834 ,  5.07306274,  8.94395628, 12.59840193,  6.49538841,
        4.78544875, 12.52372629,  7.88677538, 10.9036226 , 13.3426541 ,
       11.52733581, 11.70168755, 11.48734978,  4.91038884,  7.37902766,
        8.12580976, 11.00694688, 10.00794468,  7.88677538,  6.7166297 ,
       10.1371552 ,  8.29640613, 11.95743558,  6.80559669, 11.63724792,
        8.20436764, 12.53047489, 11.33837396,  6.67218687,  7.5092834 ,
        8.55982105, 11.81423161, 10.48840476, 10.63038565,  9.562174  ,
       12.10222204, 12.7125079 , 13.77222752, 10.08001894,  5.90035324,
        6.7593917 , 13.59673406,  6.14086582, 12.64235359,  7.27340834,
        9.18690018])}
{'P': tensor([[7.5508e-03, 2.6406e-01, 2.3065e-01,  ..., 1.4802e-05, 1.6561e-04,
         1.9728e-03],
        [1.0026e-04, 7.7954e-03, 3.2350e-01,  ..., 2.0135e-05, 2.5587e-04,
         2.9822e-03],
        [5.1179e-05, 7.6418e-05, 9.7129e-03,  ..., 3.0789e-05, 5.8470e-04,
         7.1424e-03],
        ...,
        [2.8610e-05, 2.9909e-05, 2.8154e-05,  ..., 1.5984e-01, 2.7235e-04,
         5.5423e-05],
        [1.7688e-06, 1.9385e-06, 2.2906e-06,  ..., 1.5657e-03, 4.2400e-02,
         3.5931e-05],
        [8.8493e-06, 1.0643e-05, 1.4613e-05,  ..., 8.9644e-04, 4.9426e-01,
         6.4667e-02]]), 'X': tensor([[ -6.4822,  -8.0252,  -3.8983],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.5506,  -5.1154,   6.3423],
        [  1.9241,   7.3503,   7.9004],
        [  5.7802,  -1.6363,  -6.2782],
        [  1.5662,  13.9375,   4.5142],
        [  5.5101,  -2.1433,   9.5442],
        [  5.5158,  -2.1337,   3.5246],
        [  1.4150,  13.9647,  -4.0915],
        [  5.7963,  -1.6018,   1.8856],
        [  6.3205,   1.7068,  -3.8619],
        [ 12.3609,   3.4769,  -8.5762],
        [  1.7500,   7.4187,   4.4237],
        [  6.0054,  -1.0902,   8.3079],
        [ -8.8835,  -4.3178,   9.3080],
        [  3.9770, -10.6341,   8.0783],
        [ -8.1204,  -5.9308,   4.0761],
        [ -5.3382,   6.6533,   7.8054],
        [  5.6928,  12.4710,   3.6001],
        [ 12.6059,   0.9597,  -6.3264],
        [  9.1620,  -7.5573,  -9.9882],
        [  6.1372,   2.6726,  -6.7885],
        [ -1.8788, -10.6567,   3.7068],
        [ -9.3107,  -2.7697,  -3.2919],
        [  4.2612, -10.5514,  -6.6268],
        [  1.8766,   7.3694,   1.1887],
        [ 11.5966,   5.9666,   6.0083],
        [ 11.6165,  -3.8715,   6.4594],
        [ -5.9938,  -8.4745,   3.9735],
        [ 11.6165,  -3.8715,   6.4594],
        [  1.4489,   7.5256,   4.4529],
        [ 10.9916,   7.2133,  -0.9179],
        [ 12.4690,  -0.8528,  -9.3374],
        [  7.9141,  -8.6656,   9.6948],
        [  5.4429,  -2.2529,   9.7306],
        [ -6.7643,  -7.7372,  -9.6663],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.3900,   3.3727,   3.1650],
        [  9.6816,   9.1536,  -5.8543],
        [ 11.6634,   5.8050,   9.1590],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.8363,   2.4086,  -5.6296],
        [  6.3087,   1.8053,  -1.9481],
        [ -3.5173,   7.5091,  -8.9585],
        [  0.7685,  -4.8098,  -2.9310],
        [  6.1575,  12.2046,   3.3257],
        [ 12.1588,   4.3364,  -8.0867],
        [  6.7565,  11.8237,  -4.9882],
        [  5.9526,   3.2525,  -4.3821],
        [ -7.8280,   4.2848,  -7.1955],
        [  6.2902,   0.0451,   3.7116],
        [  5.1890,   4.7318,   0.9515],
        [  8.6420,  -8.0580,   6.5781],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.0617,  12.7957,   8.3544],
        [  1.7000,  -4.7609,   2.7935],
        [ -4.2271,   7.2387,  -1.8316],
        [ 12.0582,  -2.6640,   0.7132],
        [  7.7750,  11.0665,   2.8364],
        [ 12.3609,   3.4769,  -8.5762],
        [ -3.5173,   7.5091,  -8.9585],
        [  5.1886,  -2.6250,   6.4677],
        [  6.2242,  -0.3171,   7.4978],
        [  3.1480,  -4.3165,  -7.8002],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -8.0750,  -6.0072,  -5.8267],
        [  3.3943,   6.5483,   1.0790],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.8450,  -9.9356,   3.0767],
        [  6.6319,  11.9066,   3.5165],
        [ -8.0330,  -6.0765,   6.7945],
        [  7.3592,  11.3937,   6.2407],
        [ -6.3066,  -8.1936,   9.5822],
        [  1.1644, -11.0394,   7.8511],
        [  0.2571,  -4.7594,   1.5618],
        [  2.7121,   6.9743,  -8.4202],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.4179,  -2.2925,  -3.4641],
        [ -5.3382,   6.6533,   7.8054],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.8835,  -4.3178,   9.3080],
        [  1.8719,   7.3713,  -3.7465],
        [  9.4925,   9.3811,   8.9842],
        [  6.0310,  12.2795,  -6.4528],
        [  6.2895,   0.0405,   5.7652],
        [  6.1989,  -9.7591,   0.8289],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.8700,  -4.8132,  -7.8042],
        [  6.0047,   3.1062,  -6.7914],
        [  6.8407,  11.7666,   3.0045],
        [ 12.3453,  -1.5462,   7.0212],
        [  3.7298,  13.3537,  -9.4653],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.5240,  -2.1198,   6.2055],
        [  8.7131,  10.2216,   7.7915],
        [  8.5198,  10.4091,  -8.6486],
        [ 12.4690,  -0.8528,  -9.3374],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.2883,   0.0325,   5.2692],
        [  9.9756,  -6.6330,  -3.3429],
        [  5.4426, -10.1181,  -5.1268]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0322, 0.0586,  ..., 1.2156, 0.8217, 0.5912],
        [0.0322, 0.0000, 0.0377,  ..., 1.1848, 0.8019, 0.5754],
        [0.0586, 0.0377, 0.0000,  ..., 1.1605, 0.7653, 0.5378],
        ...,
        [1.2156, 1.1848, 1.1605,  ..., 0.0000, 0.7101, 0.8762],
        [0.8217, 0.8019, 0.7653,  ..., 0.7101, 0.0000, 0.2508],
        [0.5912, 0.5754, 0.5378,  ..., 0.8762, 0.2508, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.31613269, 10.39942497, 10.52357479, 10.61047596, 10.60226614,
       10.54658597,  9.96391045,  7.59795445,  6.00732193, 14.02526505,
        5.91222346,  5.91408504, 14.03618649,  6.01356004,  6.54693987,
       12.84056604,  7.62232307,  6.10360073,  9.87720102, 11.35345345,
       10.05558893,  8.53014915, 13.70893492, 12.64235359, 11.87665501,
        6.69389431, 10.82106505,  9.71392179, 11.37939695,  7.60463349,
       13.04156075, 12.24467041, 10.37998078, 12.24467041,  7.66377554,
       13.1471313 , 12.4980845 , 11.73567759,  5.89074104, 10.27716434,
       11.92133129,  9.04255655, 13.32374056, 13.02818702, 14.07623266,
        9.15866607,  6.5618968 ,  8.29203356,  4.87083348, 13.66991027,
       12.9089559 , 13.61801526,  6.78325151,  8.92395381,  6.29034958,
        7.02253235, 11.8159254 , 12.17451685, 13.76047999,  5.05534609,
        8.38252937, 12.3489324 , 13.5247215 , 12.84056604,  8.29203356,
        5.81481044,  6.2322769 ,  5.34250225,  9.97822473, 10.06437598,
        7.37576154, 10.75443106,  9.48337063, 11.52733581, 13.62896593,
       10.07238463, 13.56367857, 10.33958639, 11.10066126,  4.7663526 ,
        7.48311002,  8.25828409,  5.8828986 ,  8.53014915, 11.8159254 ,
        9.87720102,  7.60529892, 13.34587027, 13.68063098,  6.28961988,
       11.56146873, 12.25806559,  4.89121805,  6.76056625, 13.6105595 ,
       12.44177641, 13.86480175, 10.1979669 ,  5.91677188, 13.43127131,
       13.45124917, 12.4980845 ,  5.5205189 ,  6.28835101, 11.97957425,
       11.4890861 ])}
{'P': tensor([[9.9389e-03, 3.6948e-01, 1.5467e-01,  ..., 4.8044e-06, 4.0872e-07,
         2.4053e-04],
        [1.1624e-04, 1.8308e-02, 4.5900e-03,  ..., 1.1691e-05, 1.1382e-06,
         9.8268e-04],
        [1.1936e-04, 1.2966e-02, 8.6349e-03,  ..., 3.4784e-06, 4.2982e-07,
         3.8076e-04],
        ...,
        [5.7801e-06, 6.2748e-06, 4.4936e-06,  ..., 1.2224e-01, 2.0809e-03,
         8.8059e-05],
        [7.6992e-07, 5.3644e-07, 4.8682e-07,  ..., 1.0135e-05, 3.5013e-02,
         5.5792e-07],
        [1.4802e-06, 1.8205e-06, 1.7159e-06,  ..., 6.9235e-05, 1.1482e-03,
         3.4260e-02]]), 'X': tensor([[-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.4361e-02, 8.2432e-02,  ..., 1.2913e+00, 1.2922e+00,
         7.1660e-01],
        [3.4361e-02, 3.0339e-04, 6.8515e-02,  ..., 1.2742e+00, 1.2665e+00,
         6.8315e-01],
        [8.2432e-02, 6.8515e-02, 3.0339e-04,  ..., 1.3269e+00, 1.3029e+00,
         6.8927e-01],
        ...,
        [1.2913e+00, 1.2742e+00, 1.3269e+00,  ..., 0.0000e+00, 4.0998e-01,
         9.5147e-01],
        [1.2922e+00, 1.2665e+00, 1.3029e+00,  ..., 4.0998e-01, 0.0000e+00,
         7.4200e-01],
        [7.1660e-01, 6.8315e-01, 6.8927e-01,  ..., 9.5147e-01, 7.4200e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.31886066, 10.48315436, 10.4606265 , 10.67322527, 10.63038565,
       10.68031351, 13.96987806,  9.50913849,  5.05774295,  5.13616224,
        8.30729352,  7.69752262,  5.69913817,  8.06643487, 10.69963954,
        6.27905292, 13.59170809, 11.46176259, 13.59170809,  7.34250948,
       10.412299  ,  7.60529892, 13.5247215 ,  6.46560181, 12.30858677,
        9.79814997,  9.84314712,  7.37902766, 14.08549802,  6.76661595,
        8.36995953, 10.90638066, 10.87409005, 11.83570132,  8.74470218,
       13.59056613, 11.31276109,  8.89301516, 13.15525803, 13.0743399 ,
        6.67108662,  9.18358562, 13.43127131, 10.06023665,  8.5724042 ,
       13.44593519,  9.23178204,  6.89245352,  9.5642993 ,  6.65991877,
        6.76661595, 11.28895489,  9.93553424, 12.1494023 , 13.76047999,
       13.13205357,  9.75401449,  9.4997987 , 13.32557133, 10.06703358,
       13.77075174,  9.58347456, 11.57249984,  9.09119351,  5.57341623,
       12.35625694,  8.7907923 ,  9.12220271, 12.52875793, 10.36844813,
       13.76047999, 11.11038551,  9.41244865,  8.91551231, 14.07623266,
       10.06703358, 11.018084  , 13.31764181, 13.79762415, 11.62158554,
        6.38399789, 13.04217487,  6.46356806,  7.01475376, 12.12641969,
        8.08320235, 10.99326577,  6.59176018, 11.28215429,  8.97052115,
       10.06437598,  4.91336994,  8.92098752,  5.64464807,  6.35029274,
       13.36800073,  8.99276564,  5.64500512, 12.11735334, 12.27991991,
        8.0550286 , 10.48840476, 11.00812996,  5.95297459, 13.32613445,
       11.95743558])}
{'P': tensor([[1.3515e-02, 2.2049e-01, 1.5777e-01,  ..., 7.2111e-05, 2.8965e-06,
         6.3911e-02],
        [3.8648e-05, 1.3301e-02, 2.8864e-04,  ..., 1.7887e-04, 3.7021e-06,
         1.7317e-01],
        [1.4521e-04, 2.0141e-01, 1.8821e-02,  ..., 2.1982e-04, 5.0526e-06,
         9.0798e-02],
        ...,
        [1.9315e-05, 1.9243e-05, 2.0766e-05,  ..., 5.5910e-02, 9.4277e-03,
         1.8548e-05],
        [9.4716e-05, 3.6467e-05, 5.3755e-05,  ..., 1.4276e-04, 6.3762e-02,
         1.7996e-05],
        [4.2671e-05, 2.3584e-04, 1.5328e-04,  ..., 7.0317e-04, 1.2390e-05,
         3.7799e-02]]), 'X': tensor([[ -6.4002,  -8.1048,  -0.9108],
        [ -3.3999, -10.1152,  -1.2088],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -2.6735, -10.4062,  -1.5911],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.9456,  -9.8542,   0.7466],
        [ -7.9298,   4.1367,  -3.9773],
        [  8.8091,  -7.9037,   0.3631],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.1297,  -3.7098,  -3.9838],
        [ -2.5056,   7.7720,   5.5878],
        [ -9.1960,  -3.2964,   1.6921],
        [  7.0515,  11.6194,   8.4289],
        [  1.3670,  -4.7990,   3.2603],
        [  0.1497,   7.8334,  -5.4412],
        [  9.2811,  -7.4336,   5.1771],
        [ 11.5966,   5.9666,   6.0083],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.3301,   0.3638,  -7.5685],
        [  3.4604,  -4.1545,   1.8091],
        [  9.0129,  -7.7072,   9.1400],
        [ -9.2707,   0.9661,   4.4863],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.9600,   7.8577,   6.9121],
        [ 11.4703,  -4.2006,   7.4086],
        [  3.2701,   6.6338,   9.9209],
        [ -1.3940, -10.7772,   6.7835],
        [  9.3560,   9.5391,   9.4943],
        [  2.2860,   7.1916,  -6.1011],
        [ -3.4100,   7.5436,  -7.6980],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.0277,   4.9589,   7.0175],
        [  8.5198,  10.4091,  -8.6486],
        [  2.6078, -10.9229,   6.2869],
        [  6.9433,  -9.3362,  -0.3421],
        [ -3.1671,   7.6158,   6.0765],
        [  9.3836,   9.5075,  -5.9278],
        [ -2.3317,   7.8034,  -3.6274],
        [ 11.1469,   6.9237,   1.6160],
        [ -8.6603,   2.8250,   5.9181],
        [  2.2860,   7.1916,  -6.1011],
        [ -8.8154,   2.4611,   2.5198],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.1066,  -0.7799,  -4.1746],
        [  3.6469,   6.3624,  -1.8414],
        [  1.5653,  13.9377,   4.1773],
        [  6.3460,   1.4252,   1.2291],
        [ -8.0608,   3.9362,  -5.9567],
        [ -5.6792,  -8.7352,   4.0318],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -2.9497, -10.3028,  -7.3479],
        [  5.3860,  -2.3418,   1.0826],
        [  1.3670,  -4.7990,   3.2603],
        [  6.1752,   2.5242,   3.5874],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.3076,   0.7852,  -6.7010],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.3583,   1.1729,   8.7835],
        [  2.4572,   7.1085,   7.7376],
        [ 12.5258,  -0.4200,  -9.3493],
        [  6.2630,   2.1057,  -5.2136],
        [  0.3715,  -4.7756,   0.2312],
        [  6.3542,   0.6934,  -3.3840],
        [ -5.6792,  -8.7352,   4.0318],
        [  4.5491,  -3.3485,   6.8412],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.8474,   6.8979,   6.6457],
        [  0.8700,  -4.8132,  -7.8042],
        [ -2.2188,   7.8218,   8.6064],
        [  1.8721,  -4.7322,   0.5524],
        [ -8.9788,   2.0208,  -3.2867],
        [ -8.8835,  -4.3178,   9.3080],
        [  0.6933,   7.7336,   2.1179],
        [ -4.0868,   7.2981,   8.1874],
        [ -5.6174,   6.4708,  -1.7339],
        [ -8.8745,   2.3097,   3.5600],
        [  1.8721,  -4.7322,   0.5524],
        [ -8.5607,  -5.0938,  -5.4531],
        [  2.3376,  -4.6231,   9.8436],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.4117,   0.1160,   8.2574],
        [  9.3560,   9.5391,   9.4943],
        [ -5.0550,   6.8231,  -9.2502],
        [  0.5600, -11.0321,  -0.8660],
        [  2.9461,  -4.4077,  -0.4831],
        [  0.2571,  -4.7594,   1.5618],
        [ -7.2515,   5.0195,   4.6434],
        [  3.5939,   6.4029,  -1.8357],
        [  7.1821,  -9.1845,   9.8548],
        [  1.7301,   7.4262,   5.0522],
        [  8.6420,  -8.0580,   6.5781],
        [ -6.7992,   5.4991,   9.9186],
        [  5.6523,   3.9443,  -2.9449],
        [ -8.9636,   2.0648,   8.1946],
        [  5.2608,  12.6977,   4.9166],
        [ -8.6603,   2.8250,   5.9181],
        [  5.5240,  -2.1198,   6.2055],
        [  0.8572,  14.0501,  -8.3364],
        [  3.0719, -10.8449,  -4.1673],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.0000,   7.6591,  -4.5011],
        [ -0.9715, -10.8630,   0.3555]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 1.0122e-01, 1.3308e-01,  ..., 9.5224e-01, 7.5177e-01,
         1.9919e-01],
        [1.0122e-01, 0.0000e+00, 1.1837e-01,  ..., 8.5123e-01, 6.5105e-01,
         1.4064e-01],
        [1.3308e-01, 1.1837e-01, 0.0000e+00,  ..., 8.8338e-01, 6.8105e-01,
         2.5886e-01],
        ...,
        [9.5224e-01, 8.5123e-01, 8.8338e-01,  ..., 0.0000e+00, 2.0233e-01,
         8.2384e-01],
        [7.5177e-01, 6.5105e-01, 6.8105e-01,  ..., 2.0233e-01, 0.0000e+00,
         6.3281e-01],
        [1.9919e-01, 1.4064e-01, 2.5886e-01,  ..., 8.2384e-01, 6.3281e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.32715715, 10.67132524, 10.5575601 , 10.74409834, 10.48840476,
       10.61472636,  8.94395628, 11.83509372, 12.22276278,  5.5513072 ,
        8.16584537,  9.76896904, 13.59170809,  4.98989906,  7.83486943,
       11.89105259, 13.04156075, 12.26997165, 12.27991991,  6.34059021,
        5.40689332, 11.85890664,  9.32094055,  4.75303127,  8.09842704,
       12.21532282,  7.39599986, 10.86693708, 13.3614598 ,  7.54620849,
        8.27853076,  5.18047638,  5.90035324,  7.06169391, 13.45124917,
       11.22993218, 11.63502679,  8.24808832, 13.35832725,  8.14433542,
       13.12218013,  9.10945678,  7.54620849,  9.15253344,  6.07992623,
        6.1561646 ,  7.33350642, 14.02532758,  6.50409372,  8.97052115,
       10.41909211, 10.32715715, 10.7167285 ,  5.87305005,  4.98989906,
        6.67121748,  9.45995955,  9.34061938,  9.76896904,  6.46560181,
        7.52117132, 12.53285633,  6.60752373,  4.79003361,  6.39187285,
       10.41909211,  5.64865173,  9.23178204, 12.41366754,  7.46248947,
        4.89121805,  8.1303945 ,  5.08910491,  9.20340031,  9.87720102,
        7.76457095,  8.36445758,  8.56889738,  9.17016857,  5.08910491,
        9.96152788,  5.1805345 , 10.48315436,  6.60752373,  9.41244865,
       13.3614598 ,  8.49161154, 11.04629471,  5.30158497,  4.7663526 ,
        8.81930525,  7.34250948, 11.65923396,  7.62509582, 11.8159254 ,
        8.74470218,  6.89245352,  9.19837792, 13.744387  ,  9.10945678,
        5.91677188, 14.07623266, 11.27160035,  6.27905292,  7.72415048,
       10.90638066])}
{'P': tensor([[1.2673e-02, 2.2067e-01, 1.7371e-01,  ..., 4.3454e-06, 2.1040e-06,
         1.4691e-06],
        [5.4969e-05, 1.8373e-02, 3.9871e-04,  ..., 2.8760e-06, 2.2850e-06,
         6.2402e-06],
        [9.2691e-05, 1.2965e-01, 1.2116e-02,  ..., 2.4142e-06, 1.6162e-06,
         2.3130e-06],
        ...,
        [1.1206e-02, 2.4153e-03, 4.0526e-03,  ..., 3.9026e-02, 1.0225e-04,
         8.1761e-07],
        [7.9284e-04, 2.0762e-04, 3.2373e-04,  ..., 1.2311e-01, 4.5366e-02,
         1.9012e-06],
        [1.2113e-07, 7.5657e-08, 7.5002e-08,  ..., 3.7734e-06, 5.2881e-05,
         6.8579e-03]]), 'X': tensor([[-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 9.8036e-02, 1.1612e-01,  ..., 1.3061e-01, 4.7929e-01,
         1.3529e+00],
        [9.8035e-02, 0.0000e+00, 1.1837e-01,  ..., 5.4763e-02, 3.8137e-01,
         1.2687e+00],
        [1.1612e-01, 1.1837e-01, 0.0000e+00,  ..., 9.3501e-02, 4.4085e-01,
         1.3704e+00],
        ...,
        [1.3061e-01, 5.4763e-02, 9.3501e-02,  ..., 0.0000e+00, 3.5926e-01,
         1.2770e+00],
        [4.7929e-01, 3.8137e-01, 4.4085e-01,  ..., 3.5926e-01, 0.0000e+00,
         9.8160e-01],
        [1.3529e+00, 1.2687e+00, 1.3704e+00,  ..., 1.2770e+00, 9.8160e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.33712258, 10.67132524, 10.5575601 , 10.74409834, 10.61472636,
       10.48840476, 11.92133129,  6.383897  ,  8.31625171, 12.22276278,
        8.28968067,  6.78325151, 11.70168755,  5.48276038,  9.96152788,
        6.25586297, 13.14690137, 13.25017211, 13.36800073,  8.87587457,
       12.07818871,  8.10886021,  5.27178363,  9.4997987 , 12.35421461,
        6.41202986,  4.91652556, 12.0495521 , 10.76154256,  9.34061938,
        6.33075682, 10.23946247,  6.10360073,  9.04255655, 11.87665501,
        8.41171369, 13.12218013,  5.84533345, 13.84126197, 12.9089559 ,
       12.22503754, 10.86693708, 10.38897563,  9.84314712, 12.18045978,
        6.02588517, 10.33958639, 11.85743425,  6.55800267, 13.34587027,
        9.60359158,  4.8763135 ,  7.85919687, 12.52372629, 12.05395056,
        5.91677188,  9.77588801, 12.33167382,  5.24137909, 11.92133129,
        7.05627461,  6.38399789, 13.20988996,  8.44940661,  8.43626114,
       13.35832725,  6.02789634,  5.24137909,  6.39438044,  8.5724042 ,
       11.22993218,  7.20482224,  4.91652556,  6.77777606, 10.8207297 ,
        6.7166297 , 13.61801526,  7.14837514,  5.19986797, 11.52733581,
       10.78533838,  9.79039371,  8.95983586,  5.26342203,  9.26455795,
       10.26545792,  8.62260327,  6.33075682,  8.27853076,  7.14837514,
        8.35856487, 13.2778378 ,  4.74770899, 12.91505111,  5.48276038,
        9.58072286,  5.64464807,  8.41171369, 10.06407839, 13.3426541 ,
       10.37998078, 10.66631784, 13.26590071,  9.14973319,  8.47371868,
       13.0743399 ])}
{'P': tensor([[6.8127e-03, 2.9875e-01, 2.7701e-01,  ..., 2.4905e-05, 8.5784e-06,
         2.1873e-06],
        [1.2691e-04, 1.6467e-02, 1.9978e-04,  ..., 3.6404e-05, 2.6405e-05,
         7.1957e-06],
        [1.1093e-04, 4.6435e-01, 8.9389e-03,  ..., 2.4544e-05, 1.3015e-05,
         3.7783e-06],
        ...,
        [2.2749e-01, 1.2979e-01, 1.5819e-01,  ..., 1.4331e-02, 1.0760e-05,
         1.9082e-06],
        [1.5522e-04, 1.2905e-04, 1.3930e-04,  ..., 2.9120e-04, 3.4436e-01,
         6.0043e-06],
        [6.9994e-08, 9.1377e-08, 8.6483e-08,  ..., 4.2063e-08, 4.9830e-06,
         1.4069e-02]]), 'X': tensor([[-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0366, 0.0136,  ..., 0.0379, 0.8683, 1.4233],
        [0.0366, 0.0000, 0.0232,  ..., 0.0743, 0.8327, 1.4076],
        [0.0136, 0.0232, 0.0000,  ..., 0.0512, 0.8548, 1.4156],
        ...,
        [0.0379, 0.0743, 0.0512,  ..., 0.0000, 0.9039, 1.4359],
        [0.8683, 0.8327, 0.8548,  ..., 0.9039, 0.0000, 1.1031],
        [1.4233, 1.4076, 1.4156,  ..., 1.4359, 1.1031, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.33958639, 10.46238054, 10.412299  , 10.42667093, 10.69963954,
       10.63038565,  8.86082124, 13.46553623, 13.32374056,  9.78286733,
        9.09963894,  9.60359158, 13.5247215 , 10.35464916,  8.28968067,
        8.62162372,  5.18277858,  5.16611782,  5.27178363, 12.1494023 ,
        6.42439344, 13.02818702,  9.13507321, 10.41909211,  6.14086582,
        5.87305005,  6.53290663, 11.67875855, 10.7506423 ,  4.74770899,
        8.8737556 ,  9.63742669,  5.81481044,  6.42689705, 10.35485331,
       10.51340829,  7.53884104, 11.11969815, 13.6829843 , 10.48315436,
        5.39098669, 10.88618114,  4.91652556, 14.09357547, 11.63765273,
       11.11969815, 11.00694688, 12.44177641, 10.61472636, 10.67132524,
       13.27994304, 10.91805683,  9.37989631,  8.31625171,  5.80312949,
        8.35856487,  9.60080639,  7.54127425, 13.13888841, 11.22993218,
       13.59056613,  9.60359158, 13.69162795, 13.25017211, 13.52430176,
       12.49291768, 13.56367857, 13.77222752,  5.14124612,  6.41202986,
       12.5085995 , 13.66051884,  5.47702766,  5.19065439, 13.61801526,
        7.57946459,  9.91668182, 13.3614598 ,  9.4997987 , 11.87665501,
        9.77777346, 13.61600536,  8.27853076, 12.04924159,  9.97964013,
        5.5366698 , 13.32557133, 10.91601655,  8.67227386,  9.91668182,
       10.26545792,  8.38252937, 10.91601655,  8.01009083,  9.77588801,
        6.81961996, 12.33167382,  4.74468406,  5.16077764,  4.91652556,
        8.41148722,  9.69792513, 12.49291768, 10.00802627,  6.90115901,
       11.72749923])}
{'P': tensor([[1.0687e-02, 2.7071e-01, 1.8679e-01,  ..., 2.8919e-07, 3.3374e-05,
         4.9777e-05],
        [2.1506e-04, 2.0063e-02, 2.6233e-01,  ..., 6.5290e-07, 3.4945e-05,
         1.0367e-04],
        [3.0035e-04, 9.2303e-04, 8.5142e-02,  ..., 5.6964e-06, 7.7454e-05,
         9.5943e-04],
        ...,
        [7.9484e-07, 5.3020e-07, 4.7166e-07,  ..., 5.3201e-02, 1.9700e-06,
         3.5102e-05],
        [2.6473e-01, 1.6415e-01, 6.1274e-02,  ..., 2.9616e-07, 2.4190e-02,
         4.1015e-05],
        [2.8157e-05, 2.5594e-05, 3.4883e-05,  ..., 1.1451e-02, 2.7187e-05,
         1.1481e-01]]), 'X': tensor([[-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.1479e-02, 7.7589e-02,  ..., 1.4296e+00, 5.3070e-02,
         1.0831e+00],
        [7.1479e-02, 3.0339e-04, 9.2949e-02,  ..., 1.4401e+00, 6.5133e-02,
         1.0932e+00],
        [7.7589e-02, 9.2949e-02, 0.0000e+00,  ..., 1.3570e+00, 1.2229e-01,
         1.0102e+00],
        ...,
        [1.4296e+00, 1.4401e+00, 1.3570e+00,  ..., 0.0000e+00, 1.4790e+00,
         3.4692e-01],
        [5.3070e-02, 6.5133e-02, 1.2229e-01,  ..., 1.4790e+00, 0.0000e+00,
         1.1322e+00],
        [1.0831e+00, 1.0932e+00, 1.0102e+00,  ..., 3.4692e-01, 1.1322e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.35093485, 10.48315436, 10.65929968, 10.51340829, 10.68416401,
       10.67322527,  5.94541803,  5.64464807,  5.89074104,  7.48311002,
        4.78002665,  7.6374603 ,  5.36061974,  7.61088125, 11.00508681,
       13.14777433, 12.12641969,  5.91854397, 11.81423161,  5.69913817,
        6.65259775, 13.59056613, 11.77124373,  4.91038884,  8.51928531,
        9.7751189 , 10.8992524 ,  8.24808832, 13.27994304, 10.26327895,
        9.86089579,  8.62260327, 13.44593519,  9.53015376,  5.86519525,
        5.79562691,  8.92549752,  6.01356004, 12.35421461, 14.0393949 ,
       12.16901615, 11.75438006,  5.84450127, 10.5575601 ,  8.7273569 ,
        9.05843201, 12.78327037,  8.5724042 , 13.99048094, 14.10796466,
        5.98285319,  9.04255655,  5.39723241,  5.08910491,  5.05444288,
        9.68323686, 12.95702647,  6.1463896 ,  6.65368551,  6.32572323,
        8.62631801, 12.24467041, 10.51340829, 11.56146873,  8.1303945 ,
        7.44805901, 11.20100795,  8.8737556 , 11.80092578,  7.83534987,
        6.81084724,  5.39098669,  5.39098669, 13.14690137,  7.53884104,
        5.96556248,  6.48486171, 12.30858677, 12.72524615,  8.40445872,
        5.92952538,  9.37296102,  7.02496223, 12.07225469, 10.63038565,
        7.16598273,  7.9337647 ,  8.04483495, 10.00410625, 13.52430176,
       12.59840193,  9.84314712, 11.018084  , 13.3614598 , 10.67132524,
       10.00794468, 11.27656488, 13.13205357,  9.44486609, 13.39372426,
        7.28793663,  7.06169391, 11.74901778, 13.36092624,  9.96391045,
        6.07992623])}
{'P': tensor([[9.2170e-03, 3.2218e-01, 1.7076e-01,  ..., 2.7006e-06, 1.2921e-07,
         3.8106e-06],
        [1.9233e-04, 1.3592e-02, 2.2419e-01,  ..., 3.1756e-06, 2.0152e-07,
         3.8532e-06],
        [4.5425e-05, 8.1269e-05, 1.0606e-02,  ..., 1.6358e-06, 2.0086e-07,
         2.0538e-06],
        ...,
        [6.1532e-03, 4.2251e-03, 2.4351e-03,  ..., 4.4562e-02, 1.6577e-06,
         2.0579e-03],
        [4.0833e-05, 3.0673e-05, 1.4469e-05,  ..., 5.9222e-03, 2.4191e-01,
         2.5143e-05],
        [2.1683e-04, 1.9593e-04, 3.7287e-04,  ..., 3.4909e-05, 1.4336e-07,
         5.1261e-02]]), 'X': tensor([[-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 3.2533e-02, 5.0263e-02,  ..., 3.6682e-01, 1.1885e+00,
         1.0666e+00],
        [3.2533e-02, 0.0000e+00, 2.7497e-02,  ..., 3.6344e-01, 1.1560e+00,
         1.0772e+00],
        [5.0263e-02, 2.7497e-02, 0.0000e+00,  ..., 3.3773e-01, 1.1433e+00,
         1.0574e+00],
        ...,
        [3.6682e-01, 3.6344e-01, 3.3773e-01,  ..., 0.0000e+00, 1.1619e+00,
         7.5713e-01],
        [1.1885e+00, 1.1560e+00, 1.1433e+00,  ..., 1.1619e+00, 0.0000e+00,
         1.7745e+00],
        [1.0666e+00, 1.0772e+00, 1.0574e+00,  ..., 7.5713e-01, 1.7745e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.35464916, 10.42667093, 10.63038565, 10.46238054, 10.69963954,
       10.68031351, 13.99048094, 12.49317563,  8.23525433, 13.77075174,
       13.72594022,  9.99145096, 13.83139024,  6.76791483,  5.48276038,
       10.00794468, 10.35485331,  9.96391045, 13.79676654, 11.28895489,
        6.81084724,  6.33075682,  9.36378438,  9.27474386, 12.50996828,
       13.97430191,  9.62238828,  7.14837514, 11.8159254 ,  7.54127425,
        9.50913849,  7.32950385,  7.92200342,  5.68942572,  4.75303127,
       11.63383221, 10.53368644, 10.95593729,  8.1303945 ,  5.52416368,
        8.12580976,  7.73692436, 11.9182956 ,  9.97220799,  9.09963894,
        6.27715105,  7.26074925,  8.86082124,  9.97822473, 13.03002338,
        8.30729352, 12.62711252, 13.14777433,  6.02588517, 13.03002338,
       10.51437947,  8.74470218,  6.65991877, 11.72802491,  7.57685908,
        6.01356004,  7.73151309, 10.31886066, 10.99326577, 11.76830981,
       12.10942412,  7.822854  ,  5.1805345 , 13.89886469, 10.35093485,
        5.24461068, 11.22993218, 12.38881635,  5.52416368, 13.52430176,
       10.06407839,  7.32233733, 12.91143491,  8.68979052,  7.92607342,
       13.66991027,  5.79543038,  7.39599986,  5.37343173,  5.87132718,
        9.53015376, 10.88618114,  8.53501777,  7.16747878,  9.93553424,
        6.80559669, 10.01804223,  4.91336994, 12.19805394, 10.06407839,
        6.10774544,  7.24677664,  9.71392179, 11.62770332, 13.62896593,
        8.91551231,  8.68832398,  7.50564223,  9.17016857, 13.99877413,
        9.91668182])}
{'P': tensor([[6.9416e-03, 2.7440e-01, 2.3784e-01,  ..., 3.4112e-04, 1.0660e-05,
         1.0665e-05],
        [6.4768e-05, 7.1570e-03, 4.1590e-01,  ..., 4.7382e-04, 1.5798e-05,
         7.8570e-06],
        [1.0046e-04, 1.6548e-04, 1.2731e-02,  ..., 9.1190e-04, 3.0944e-05,
         1.2623e-05],
        ...,
        [9.1988e-05, 1.0083e-04, 1.0376e-04,  ..., 1.1408e-01, 2.4162e-03,
         6.8808e-06],
        [1.0959e-06, 1.0949e-06, 1.1001e-06,  ..., 2.4307e-04, 7.6864e-02,
         1.4132e-06],
        [2.4935e-05, 3.1460e-05, 3.3284e-05,  ..., 6.6511e-05, 6.1816e-04,
         9.7871e-02]]), 'X': tensor([[ -6.1899,  -8.3011,  -7.9269],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -2.9497, -10.3028,  -7.3479],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -2.5683, -10.4433,  -2.9182],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.2975,   0.0947,   5.1791],
        [  5.8323,  -1.5228,   5.1132],
        [  0.1287, -11.0065,   6.5757],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.2535,  -0.1709,  -9.4102],
        [  4.8653,   5.1694,  -4.0823],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.2804,   2.0019,   0.8607],
        [  6.1973,  -0.4372,   6.5720],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.8323,  -1.5228,   5.1132],
        [  1.5653,  13.9377,   4.1773],
        [ -8.0611,  -6.0302,   4.6962],
        [  4.2041,  -3.6509,  -4.9488],
        [ -8.3062,  -5.6008,   0.0287],
        [ -6.9838,  -7.4965,   0.9796],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.3604,   0.9042,  -6.5901],
        [  9.2979,   9.6048,   7.3138],
        [  5.3118,   4.5452,   4.2114],
        [  8.0381,  -8.5690,  -8.9523],
        [  5.1667,  -2.6543,   1.2879],
        [ -1.8823, -10.6558,  -3.8952],
        [ -9.1433,  -3.5003,  -3.0209],
        [  4.7482, -10.3907,   5.9149],
        [  6.2022,   2.4092,   3.0349],
        [  9.2979,   9.6048,   7.3138],
        [  6.3048,   0.1469,  -9.1691],
        [  2.5382,   7.0672,   4.5097],
        [ 10.0567,   8.6697,   4.6329],
        [ -6.7992,   5.4991,   9.9186],
        [ -7.8359,   4.2735,   7.2943],
        [  5.0190,   4.9705,  -7.9752],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.1771, -10.9776,  -3.2007],
        [ -7.3446,   4.9113,  -5.9356],
        [  9.9756,  -6.6330,  -3.3429],
        [  3.1064,  13.5607,   6.7605],
        [ 11.9019,  -3.1379,  -3.4534],
        [  2.2849, -10.9655,  -7.6542],
        [  5.5626,  -2.0536,   8.4433],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.9036,   7.6840,   9.9237],
        [ -7.8126,   4.3066,   5.1791],
        [  9.2811,  -7.4336,   5.1771],
        [ 12.2218,   4.0953,  -8.2072],
        [  4.2096,   5.8815,   1.9230],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 12.2218,   4.0953,  -8.2072],
        [ -9.1795,  -3.3623,   8.5259],
        [ 12.5154,   2.5072,  -9.3274],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -7.0253,  -7.4493,  -1.5978],
        [  2.3200,   7.1756,   1.1938],
        [  3.3744,   6.5623,  -6.3636],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.3589,  13.7668,  -3.6501],
        [  4.8051, -10.3703,   0.5268],
        [ -8.7078,   2.7182,   1.8146],
        [  9.5261,  -7.1675,   1.4206],
        [  5.6248,   3.9984,   8.4933],
        [  1.5246,  -4.7838,  -5.2095],
        [  4.5800,   5.5034,  -3.9432],
        [  0.6396,  14.0772,  -6.3786],
        [ -3.1204, -10.2345,   9.2288],
        [  3.9591, -10.6391,  -6.2557],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.3474,  -4.2162,   5.2455],
        [  5.6928,  12.4710,   3.6001],
        [ 11.4743,   6.2478,  -2.1100],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.3539,  -5.5111,   0.3373],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.9217,  -6.7006,  -5.6728],
        [ -1.6220,   7.8917,   0.0315],
        [  0.1382, -11.0073,   1.0026],
        [  0.6470, -11.0352,  -1.1931],
        [ 11.2847,  -4.5848,  -4.7503],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.1426,   3.8048,   9.3542],
        [ -4.4520,   7.1370,  -8.7867],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.2345,  -3.1345,   3.9715],
        [ -3.9508,  -9.8515,   1.7923],
        [  5.5240,  -2.1198,   6.2055],
        [  3.6154,  -4.0640,  -7.9698],
        [  9.3753,   9.5171,   3.9131],
        [ -2.2508, -10.5479,  -2.7235],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.3205,   1.7068,  -3.8619],
        [ 11.7418,   5.6070,  -1.4192],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.4297,  -5.3637,  -9.3962],
        [  4.7156,  -3.1831,  -8.0224],
        [ 12.0582,  -2.6640,   0.7132],
        [ -3.1204, -10.2345,   9.2288]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0268, 0.0547,  ..., 0.9644, 1.1120, 1.1852],
        [0.0268, 0.0000, 0.0310,  ..., 0.9450, 1.1015, 1.1974],
        [0.0547, 0.0310, 0.0000,  ..., 0.9142, 1.0747, 1.1923],
        ...,
        [0.9644, 0.9450, 0.9142,  ..., 0.0000, 0.4400, 1.2957],
        [1.1120, 1.1015, 1.0747,  ..., 0.4400, 0.0000, 0.9774],
        [1.1852, 1.1974, 1.1923,  ..., 1.2957, 0.9774, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.35485331, 10.46739073, 10.4977132 , 10.57302546, 10.7167285 ,
       10.53368644, 10.75443106,  8.29203356,  6.29821673,  6.02779201,
       11.00726483, 10.00802627,  6.25586297,  7.09887557,  4.85995002,
        6.59176018,  6.21275016, 10.23946247, 12.49317563,  6.02779201,
       14.02532758, 10.06703358,  5.56809335, 10.01804223, 10.24557047,
        9.62238828,  6.42439344, 13.36800073,  6.99097013, 11.74901778,
        5.80860165, 10.8207297 ,  9.79039371, 11.42420508,  6.65368551,
       13.36800073,  6.30647746,  7.50917551, 13.2778378 ,  8.74470218,
        8.92549752,  7.06372172, 10.33958639, 11.1913598 ,  8.8353674 ,
       11.97957425, 13.91198191, 12.30858677, 11.20100795,  5.92952538,
       12.19805394,  7.73692436,  8.92098752, 11.89105259, 12.889695  ,
        7.23276582,  5.91677188,  9.88571713, 12.889695  ,  9.77588801,
       12.76408562, 10.52587243, 10.23946247,  7.54127425,  7.37902766,
       10.71162681, 13.96746816, 11.42947632,  9.12220271, 11.92133129,
        6.90115901,  5.0209005 ,  7.15990676, 14.09176259, 10.69963954,
       11.35182589,  5.5366698 ,  5.38342077, 13.70893492, 13.06498554,
        5.14999799, 10.00794468,  5.26067019, 11.97238624,  8.05668609,
       11.00812996, 11.05413545, 12.18045978,  7.26074925,  8.98765644,
        8.41171369,  8.1137452 ,  9.75200848, 10.61417181,  5.91677188,
        5.43943791, 13.35927527, 10.78533838, 11.66453059,  6.54693987,
       13.01187484, 13.12299145,  9.99145096,  5.68942572, 12.3489324 ,
       10.69963954])}
{'P': tensor([[7.4094e-03, 3.0776e-01, 1.9509e-01,  ..., 1.8346e-06, 1.8037e-07,
         4.4379e-06],
        [6.7174e-05, 8.2133e-03, 3.5609e-01,  ..., 3.1280e-06, 2.5704e-07,
         6.4820e-06],
        [3.7740e-05, 6.4503e-05, 8.7757e-03,  ..., 5.2031e-06, 3.5509e-07,
         8.4716e-06],
        ...,
        [7.4317e-08, 6.4685e-08, 5.6374e-08,  ..., 4.0435e-03, 1.7762e-02,
         3.1736e-04],
        [6.5526e-07, 4.7263e-07, 3.2985e-07,  ..., 2.2253e-06, 1.6008e-02,
         4.6476e-05],
        [1.5353e-05, 1.3936e-05, 1.1816e-05,  ..., 1.4487e-03, 1.7107e-02,
         8.4631e-02]]), 'X': tensor([[ -6.0844,  -8.3955,   0.7705],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.9456,  -9.8542,   0.7466],
        [ -3.9508,  -9.8515,   1.7923],
        [ -2.6485, -10.4151,   0.8766],
        [ -2.1533, -10.5778,   1.0112],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.2366,  -4.2732,   5.1532],
        [ -8.3653,   3.4184,   9.3454],
        [ 11.7794,   5.5086,  -9.6929],
        [  2.2373,  -4.6506,  -1.7334],
        [  2.3376,  -4.6231,   9.8436],
        [  5.6093,  -1.9705,   6.4807],
        [  5.6414, -10.0303,   7.8525],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.3604,   0.9042,  -6.5901],
        [  7.0171,  11.6439,  -5.0178],
        [ -9.0574,   1.7801,   8.8986],
        [ -6.6998,   5.5954,   0.0823],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.3124,   0.2057,  -0.4593],
        [ -9.4714,  -1.3301,  -3.1998],
        [  3.4920,  -4.1366,  -8.3955],
        [  4.8674,  12.8875,   6.7622],
        [  5.2911,  -2.4824,  -3.6877],
        [ -7.4807,   4.7463,   2.0616],
        [ -9.0961,   1.6529,   3.1553],
        [  9.5203,   9.3483,   5.8781],
        [ -6.3489,   5.9117,   6.5161],
        [ 10.2470,   8.4054,  -0.1177],
        [  4.2339,  -3.6268,   6.3125],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.7151,  -1.7700,   3.1542],
        [  6.3606,   0.9204,   3.2525],
        [  5.5158,  -2.1337,   3.5246],
        [  5.6248,   3.9984,   8.4933],
        [ -5.1046,  -9.1610,   2.3418],
        [ 10.0400,   8.6922,  -6.2847],
        [ -6.3252,  -8.1760,  -1.0891],
        [  9.0051,   9.9234,   5.4411],
        [ -9.1178,   1.5784,   2.1410],
        [  3.2692,  13.5098,  -4.9032],
        [  5.0190,   4.9705,  -7.9752],
        [  6.1984,  -0.4327,   1.5438],
        [ -1.8788, -10.6567,   3.7068],
        [  5.4429,  -2.2529,   9.7306],
        [ -9.2707,   0.9661,   4.4863],
        [ 12.5832,   1.7543,  -7.6134],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.6928,  12.4710,   3.6001],
        [  9.4516,  -7.2501,  -1.8868],
        [ -5.0550,   6.8231,  -9.2502],
        [  6.1752,   2.5242,   3.5874],
        [  0.3038,   7.8092,   0.4164],
        [  6.3460,   1.4252,   1.2291],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -7.0948,   5.1940,   9.8171],
        [  1.9312,  13.8649,   8.3563],
        [  6.7565,  11.8237,  -4.9882],
        [ -7.8359,   4.2735,   7.2943],
        [ -6.9838,  -7.4965,   0.9796],
        [  9.9058,  -6.7203,   5.6928],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -2.2508, -10.5479,  -2.7235],
        [ 12.3453,  -1.5462,   7.0212],
        [  6.3205,   1.7068,  -3.8619],
        [ 12.5770,   1.8512,  -0.6578],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.4735,  -1.2620,  -3.6742],
        [  4.5325,  -3.3643,   6.7699],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.7779,   5.5127,  -2.0699],
        [ -8.3535,  -5.5118,   9.8399],
        [  9.6587,   9.1817,  -0.8212],
        [  2.6970,   6.9826,   9.7350],
        [ -1.1885, -10.8211,   0.0865],
        [ -2.0721,   7.8432,   6.6380],
        [  0.7958,  -4.8109,  -8.2248],
        [ -6.1914,  -8.2997,   8.5380],
        [  1.3767, -11.0341,   9.9415],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.3551,   0.7149,   1.3344],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.9788,  -4.8145,  -3.8661],
        [  3.3474,  -4.2162,   5.2455],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3236,   0.3011,  -4.6668],
        [  0.5850,  -4.7982,   9.6639],
        [  6.8407,  11.7666,   3.0045],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.8700,  -4.8132,  -7.8042],
        [  6.3542,   0.6934,  -3.3840],
        [  4.7813,  -3.1139,   8.2691],
        [ -8.3653,   3.4184,   9.3454],
        [  5.1519, -10.2387,   9.8191],
        [ -3.1671,   7.6158,   6.0765],
        [ 10.0567,   8.6697,   4.6329],
        [ -4.3963,   7.1629,   4.5263],
        [ -2.8162,   7.7059,  -9.6885],
        [ 11.9808,   4.9340,  -3.3715],
        [  6.7565,  11.8237,  -4.9882],
        [  5.8201,   3.5841,  -8.0609]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0329, 0.0567,  ..., 1.1867, 1.2878, 1.0199],
        [0.0329, 0.0000, 0.0348,  ..., 1.1562, 1.2564, 0.9876],
        [0.0567, 0.0348, 0.0000,  ..., 1.1314, 1.2336, 0.9677],
        ...,
        [1.1867, 1.1562, 1.1314,  ..., 0.0000, 0.1357, 0.2552],
        [1.2878, 1.2564, 1.2336,  ..., 0.1357, 0.0000, 0.2924],
        [1.0199, 0.9876, 0.9677,  ..., 0.2552, 0.2924, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.36844813, 10.48840476, 10.61472636, 10.61417181, 10.74655346,
       10.794755  , 13.00413829,  5.36061974,  9.0368458 , 13.00380624,
        5.16077764,  5.1805345 ,  5.94535529, 11.50791859,  9.11473391,
       10.00794468, 10.07238463,  6.42439344, 13.59480906,  9.23071698,
        8.72896381, 10.412299  ,  6.31576264,  9.5642993 ,  5.41348121,
       13.77604293,  5.84450127,  8.85939626,  9.24501829, 13.3426541 ,
        8.67502348, 13.25336001,  5.5748624 , 13.03002338,  5.98285319,
        6.42689705,  5.91408504,  6.90115901, 10.48721382, 13.27994304,
       10.33712258, 13.40024224,  9.25336371, 13.89974693,  7.06372172,
        6.2134961 , 10.82106505,  5.89074104,  9.32094055, 12.70489563,
       11.99633819, 12.7125079 , 13.70893492, 11.9120321 ,  8.49161154,
        6.67121748,  7.81509631,  6.50409372, 12.35421461,  8.79284466,
       13.99877413, 13.61801526,  8.92549752, 10.24557047, 11.9702762 ,
       12.10942412, 10.78533838, 12.44177641,  6.54693987, 12.7125079 ,
        7.8575648 ,  9.55720651,  5.64464807, 13.20597932, 13.00413829,
       10.00802627, 13.32645552,  7.48538905, 10.88618114,  8.11227668,
        4.8763135 , 10.35464916, 11.11969815, 10.61047596,  6.39520397,
       10.26545792,  4.91296205,  5.38342077,  6.12886024,  6.33075682,
        4.83371341, 13.6105595 ,  6.14086582,  5.26067019,  4.89121805,
        6.39187285,  5.70591986,  9.0368458 , 11.46176259,  8.24808832,
       13.2778378 ,  8.40445872,  8.20436764, 12.95702647, 13.61801526,
        6.83515201])}
{'P': tensor([[9.0964e-03, 2.4889e-01, 2.1369e-01,  ..., 9.9300e-06, 2.1610e-04,
         2.5145e-06],
        [7.5309e-05, 9.1376e-03, 5.2743e-01,  ..., 1.6592e-05, 3.6382e-04,
         5.1437e-06],
        [1.1286e-04, 2.0317e-04, 1.6906e-02,  ..., 3.4567e-05, 7.5795e-04,
         1.1237e-05],
        ...,
        [8.7325e-06, 9.7855e-06, 1.0040e-05,  ..., 7.9852e-02, 1.2404e-03,
         6.0032e-04],
        [2.5062e-05, 3.1192e-05, 3.3018e-05,  ..., 9.1700e-03, 4.4135e-02,
         3.9483e-04],
        [1.3889e-07, 1.8300e-07, 1.9626e-07,  ..., 4.0538e-03, 8.9940e-05,
         3.2898e-02]]), 'X': tensor([[-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0378, 0.0693,  ..., 1.1867, 0.9749, 1.4965],
        [0.0378, 0.0000, 0.0315,  ..., 1.1531, 0.9377, 1.4699],
        [0.0693, 0.0315, 0.0000,  ..., 1.1253, 0.9067, 1.4480],
        ...,
        [1.1867, 1.1531, 1.1253,  ..., 0.0000, 0.3825, 0.5164],
        [0.9749, 0.9377, 0.9067,  ..., 0.3825, 0.0000, 0.8921],
        [1.4965, 1.4699, 1.4480,  ..., 0.5164, 0.8921, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.37698435, 10.53368644, 10.57302546, 10.71162681, 10.4977132 ,
       10.7167285 ,  8.92098752,  9.04255655,  5.95111804,  8.11227668,
       13.6105595 ,  7.10378919, 11.28093507, 11.77124373,  5.13616224,
       13.43127131, 12.71843462, 10.92597196,  8.09704052, 12.35625694,
        7.17789666,  9.5642993 , 10.88618114, 13.0493368 ,  7.32754882,
        7.8788965 , 12.42554808,  9.5014146 ,  7.24326992,  5.84533345,
        5.70405525, 12.87752052,  5.77219077, 10.9036226 ,  5.94293185,
       13.96905756,  6.35029274, 13.88683436,  8.30729352,  9.63742669,
       12.77402351,  5.35774573,  7.38844703,  5.92281142, 13.50484539,
       12.10222204,  7.32754882,  4.90676242, 12.27835996,  7.38844703,
       10.86693708,  4.83371341,  9.36378438,  5.08372149,  5.47702766,
        9.41244865, 10.99326577,  8.09842704, 13.39372426,  6.10774544,
       13.5247215 ,  6.80370254,  7.69752262,  5.19065439,  5.80312949,
        6.05165248,  8.83341608, 10.00410625,  5.79391825,  8.79284466,
       13.02818702,  6.29821673,  5.13125513,  9.15976746,  8.77913428,
        6.80370254, 11.76830981,  5.39450305, 13.97430191, 13.13205357,
        8.27853076,  7.42512621,  5.29755487, 12.72524615,  6.41202986,
       12.59969936,  9.85330646, 10.06437598,  5.13182506,  5.69913817,
       13.32557133, 12.21532282,  5.80312949, 12.21532282, 10.53368644,
        8.62329466, 11.97957425, 10.06023665, 11.70168755, 12.21532282,
        6.81084724,  6.31576264, 10.88618114,  6.32572323,  5.70405525,
       12.24467041])}
{'P': tensor([[5.5716e-03, 3.4872e-01, 2.4184e-01,  ..., 2.9664e-07, 1.3814e-07,
         1.9329e-04],
        [8.1502e-05, 7.2191e-03, 3.7903e-01,  ..., 4.2516e-07, 1.9016e-07,
         2.6136e-04],
        [5.8811e-05, 6.8620e-05, 8.2958e-03,  ..., 6.2506e-07, 2.5796e-07,
         3.4251e-04],
        ...,
        [3.4926e-06, 3.1982e-06, 2.5930e-06,  ..., 1.4105e-01, 1.2980e-01,
         2.2491e-07],
        [4.2642e-06, 3.7397e-06, 2.7680e-06,  ..., 6.9291e-05, 6.2422e-02,
         2.6759e-07],
        [1.8216e-05, 2.0567e-05, 2.8831e-05,  ..., 1.1262e-05, 1.3018e-05,
         1.1355e-01]]), 'X': tensor([[ -5.9938,  -8.4745,   3.9735],
        [ -5.6792,  -8.7352,   4.0318],
        [ -4.8751,  -9.3148,   4.0406],
        [ -3.2736, -10.1703,   3.8712],
        [ -3.5173, -10.0623,   4.8280],
        [ -2.4956, -10.4682,   3.5632],
        [ -0.3432,   7.8900,   7.1805],
        [  5.2911,  -2.4824,  -3.6877],
        [ -8.8154,   2.4611,   2.5198],
        [  2.5375,   7.0676,  -4.3490],
        [ -6.0844,  -8.3955,   0.7705],
        [ 10.9916,   7.2133,  -0.9179],
        [ -9.3076,   0.7852,  -6.7010],
        [ 12.2914,   3.8037,   6.0978],
        [ -5.6423,   6.4537,  -1.2091],
        [  0.1206,  -4.7363,   8.7103],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 11.0275,  -5.0666,   8.4154],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -1.0012, -10.8576,   2.2867],
        [  3.4009,  -4.1874,  -5.9751],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.5128, -10.9365,   9.9645],
        [  4.5503,   5.5359,   7.4735],
        [  0.9809,  -4.8145,   6.8706],
        [ -6.7992,   5.4991,   9.9186],
        [  2.2860,   7.1916,  -6.1011],
        [  7.8375,  -8.7240,  -9.8913],
        [  5.3804,  -2.3504,   7.2744],
        [ -8.7409,  -4.6837,  -6.5840],
        [  4.6063,  13.0051,  -3.1456],
        [ -7.5401,   4.6718,   9.0033],
        [ -1.6086,   7.8928,   4.3385],
        [ -1.0407,   7.9166,  -6.6021],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.0890,  -0.8382,   2.9166],
        [  3.1756, -10.8247,   6.8882],
        [ -4.0409,   7.3169,  -3.5779],
        [  5.0348,  -2.8229,   3.5436],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.3062,  -5.6008,   0.0287],
        [ 12.5763,   0.1333,  -9.2650],
        [  2.5382,   7.0672,   4.5097],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.2592,   3.9425,   2.0383],
        [  1.5383, -11.0274,  -4.9666],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.5128, -10.9365,   9.9645],
        [ -9.3604,   0.4855,  -3.0158],
        [ 11.9122,   5.1397,  -1.5498],
        [  2.3261,  13.7749,   5.9121],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.9058,  -6.7203,   5.6928],
        [ -4.0633,  -9.7927,  -3.0825],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.6681,  -4.8556,   9.2518],
        [ -1.5261,   7.8988,   9.3768],
        [  3.3943,   6.5483,   1.0790],
        [ 11.3367,  -4.4806,   0.4643],
        [  5.1196,  -2.7160,   4.2705],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4196,   0.0478,   9.4140],
        [  2.6451,  -4.5250,  -0.5565],
        [ -7.5603,   4.6459,   8.2002],
        [  6.3048,   0.1469,  -9.1691],
        [ -1.0407,   7.9166,  -6.6021],
        [  3.7992,  -3.9474,  -8.9019],
        [ -8.8934,   2.2597,   3.5831],
        [  4.5814, -10.4485,   2.5496],
        [  2.1440, -10.9811,   8.5281],
        [ -8.3653,   3.4184,   9.3454],
        [  7.1821,  -9.1845,   9.8548],
        [ 12.0772,  -2.6014,  -7.1832],
        [  4.9337,  12.8566,  -3.6582],
        [  0.9445,   7.6736,   0.5095],
        [  6.3535,   0.6792,  -7.5101],
        [ -2.3317,   7.8034,  -3.6274],
        [ -2.0721,   7.8432,   6.6380],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.3497,   1.3680,  -1.9076],
        [ -7.5603,   4.6459,   8.2002],
        [ 11.1469,   6.9237,   1.6160],
        [  6.3535,   0.6792,  -7.5101],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.5663,   7.0526,  -6.8769],
        [  6.2022,   2.4092,   3.0349],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.3464,   0.5708,  -1.6397],
        [  1.1139,   7.6281,  -2.5680],
        [  2.7558,  -4.4844,   0.0636],
        [  5.1472,  -2.6801,   8.1194],
        [  5.8617,   3.4853,  -9.8080],
        [ -2.1382,   7.8339,   7.3524],
        [ -3.0008, -10.2827,  -8.9480],
        [ -1.6275, -10.7221,   3.2376],
        [  2.3928,   7.1404,  -1.7035],
        [ 12.1379,   4.4125,  -3.8833],
        [ -1.9714,   7.8563,  -1.8084],
        [  6.1989,  -9.7591,   0.8289],
        [ -3.1204, -10.2345,   9.2288],
        [  9.3560,   9.5391,   9.4943],
        [  5.4867,  12.5816,   2.9549],
        [  3.1892, -10.8220,  -6.8409]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0141, 0.0380,  ..., 1.3820, 1.2281, 0.9197],
        [0.0141, 0.0000, 0.0241,  ..., 1.3681, 1.2155, 0.9207],
        [0.0380, 0.0241, 0.0000,  ..., 1.3440, 1.1923, 0.9179],
        ...,
        [1.3820, 1.3681, 1.3440,  ..., 0.0000, 0.4111, 1.5777],
        [1.2281, 1.2155, 1.1923,  ..., 0.4111, 0.0000, 1.2153],
        [0.9197, 0.9207, 0.9179,  ..., 1.5777, 1.2153, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.37998078, 10.41909211, 10.51340829, 10.68416401, 10.65929968,
       10.76154256,  7.89745107,  5.84450127,  9.15253344,  7.5092834 ,
       10.36844813, 13.1471313 ,  9.34061938, 12.86648771,  8.5724042 ,
        4.73784222,  9.69629   , 12.13568598, 10.5575601 , 10.9036226 ,
        5.39450305, 10.1371552 ,  5.30158497, 11.22142195,  7.16598273,
        4.91336994,  8.74470218,  7.54620849, 11.72749923,  5.87132718,
        9.91668182, 13.79676654,  8.87007495,  8.0550286 ,  7.98469166,
        5.39450305,  6.1463896 , 11.28093507,  8.35856487,  5.77219077,
        4.89121805,  9.79814997, 10.01804223, 12.5769659 ,  7.50917551,
        6.14086582, 12.87752052, 11.1341798 ,  5.19577146, 12.76408562,
       11.22142195,  9.37296102, 12.97370371, 13.96987806, 12.49291768,
       11.9702762 , 10.60226614, 13.94008948,  9.9353921 ,  8.04483495,
        7.37576154, 12.18998195,  5.79543038,  5.91854397,  9.41970031,
        5.24137909,  8.8737556 ,  6.30647746,  7.98469166,  5.47866221,
        9.17595899, 11.4088043 , 11.18839169,  9.0368458 , 11.65923396,
       12.35421461, 13.77075174,  7.73151309,  6.38968357,  8.14433542,
        8.11227668,  9.55593374,  6.49538841,  8.8737556 , 13.12218013,
        6.38968357, 10.4606265 ,  7.50500064,  6.65368551,  5.57341623,
        9.36378438,  7.70897675,  5.26342203,  5.80312949,  6.81961996,
        8.12043983, 10.71162681, 10.84493732,  7.53063885, 12.91505111,
        8.09984231, 11.56146873, 10.69963954, 13.3614598 , 13.72594022,
       11.28215429])}
{'P': tensor([[7.8039e-03, 3.8218e-01, 1.8389e-01,  ..., 3.7830e-06, 2.3224e-04,
         4.3465e-07],
        [8.1530e-05, 9.5673e-03, 3.0775e-01,  ..., 3.2458e-06, 3.7479e-04,
         5.6208e-07],
        [1.7464e-04, 3.0054e-04, 3.2814e-02,  ..., 7.0172e-06, 1.8687e-03,
         2.2978e-06],
        ...,
        [9.7697e-03, 6.9516e-03, 3.7987e-03,  ..., 4.7734e-02, 7.2138e-05,
         1.0289e-06],
        [3.5383e-05, 3.7427e-05, 4.1668e-05,  ..., 2.6415e-05, 5.7707e-02,
         2.6145e-04],
        [4.0830e-05, 3.2787e-05, 2.7243e-05,  ..., 4.3292e-05, 5.5274e-05,
         1.2165e-01]]), 'X': tensor([[ -5.9225,  -8.5355,   2.4778],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.9508,  -9.8515,   1.7923],
        [ -3.4488, -10.0934,   2.5195],
        [ -3.2736, -10.1703,   3.8712],
        [ -2.4956, -10.4682,   3.5632],
        [  3.3069,   6.6089,   8.7377],
        [  4.4915,   5.5990,   7.0405],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.5154,   2.5072,  -9.3274],
        [ 10.9875,   7.2207,  -8.8549],
        [  1.8447,  -4.7372,   8.1235],
        [  0.1460,   7.8340,  -9.4614],
        [ 12.5770,   1.8512,  -0.6578],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.6672,  -1.8631,   8.1020],
        [  6.2555,   2.1474,   4.1685],
        [  2.0547,   7.2957,  -1.9343],
        [ -1.8823, -10.6558,  -3.8952],
        [ -1.9646,   7.8571,   3.3817],
        [ -1.7290,   7.8825,  -3.3831],
        [ -4.8665,  -9.3204,  -5.2564],
        [  6.1752,   2.5242,   3.5874],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.5626,  -2.0536,   8.4433],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.1440, -10.9811,   8.5281],
        [ -3.5903,   7.4847,  -8.9112],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 10.9394,   7.3069,  -8.9365],
        [  5.0506,   4.9277,  -9.5094],
        [ -1.6220,   7.8917,   0.0315],
        [  5.5424,  -2.0885,   7.2487],
        [ -8.8154,   2.4611,   2.5198],
        [  5.8450,  -9.9356,   3.0767],
        [  7.8982,  -8.6778,   0.0239],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.9433,  -9.3362,  -0.3421],
        [  5.6095,  -1.9702,  -7.8233],
        [  0.6363, -11.0348,   6.1414],
        [ -9.4716,  -0.6673,   7.6068],
        [ -1.9714,   7.8563,  -1.8084],
        [  4.1140,  -3.7220,  -0.1273],
        [  2.1625,  -4.6697,   7.0658],
        [  5.6202,  -1.9508,   0.7471],
        [  4.2339,  -3.6268,   6.3125],
        [  9.3624,   9.5318,   0.2779],
        [ -6.7992,   5.4991,   9.9186],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.3516,   0.6425,   0.3425],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 10.5057,  -5.9099,  -5.1599],
        [  3.2159, -10.8166,  -1.0406],
        [  2.8474,   6.8979,   6.6457],
        [  0.2480, -11.0153,  -3.0183],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -3.7964,  -9.9294,   7.7147],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.3122,  -9.6994,  -6.3282],
        [  8.3791,  10.5409,   5.5590],
        [  1.6308,  13.9254,  -2.0984],
        [ 10.8670,  -5.3428,  -2.8637],
        [  3.0719, -10.8449,  -4.1673],
        [  0.3724,   7.7974,  -6.8722],
        [  8.1043,  -8.5164,  -7.9290],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -7.2515,   5.0195,   4.6434],
        [  4.0457,  -3.7737,  -0.4704],
        [  4.4290,  -3.4596,   1.4181],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.3266,   0.3292,   9.3502],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.5469,  13.7192,  -9.6935],
        [ -7.9667,   4.0814,  -0.6397],
        [ -8.6381,  -4.9238,  -4.4077],
        [  4.4915,   5.5990,   7.0405],
        [ -8.5900,   2.9767,   7.4811],
        [ 11.4743,   6.2478,  -2.1100],
        [ -1.9487,   7.8590,  -3.3330],
        [ 10.1983,  -6.3430,  -0.3200],
        [  4.1297,  -3.7098,  -3.9838],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.3670,  -4.7990,   3.2603],
        [  4.1537,   5.9340,  -6.0423],
        [ -2.7572,   7.7194,   5.9580],
        [ -9.2707,   0.9661,   4.4863],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -2.6069, -10.4298,  -6.1995],
        [ -5.9938,  -8.4745,   3.9735],
        [  6.3266,   0.3292,   9.3502],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.3497,   1.3680,  -1.9076],
        [  6.3246,   1.6696,  -0.1544],
        [ -9.4541,  -0.3327,   8.8246],
        [  5.3860,  -2.3418,   1.0826],
        [  0.1460,   7.8340,  -9.4614]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.0692e-02, 7.7404e-02,  ..., 3.8386e-01, 9.1395e-01,
         9.3069e-01],
        [2.0692e-02, 0.0000e+00, 5.9282e-02,  ..., 3.8653e-01, 8.9338e-01,
         9.1327e-01],
        [7.7404e-02, 5.9282e-02, 0.0000e+00,  ..., 4.2868e-01, 8.4858e-01,
         8.5399e-01],
        ...,
        [3.8386e-01, 3.8653e-01, 4.2868e-01,  ..., 0.0000e+00, 9.8975e-01,
         1.2064e+00],
        [9.1395e-01, 8.9338e-01, 8.4858e-01,  ..., 9.8975e-01, 1.0727e-04,
         6.0776e-01],
        [9.3069e-01, 9.1327e-01, 8.5399e-01,  ..., 1.2064e+00, 6.0776e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.38897563, 10.48721382, 10.61417181, 10.66631784, 10.68416401,
       10.76154256,  7.39004226,  7.17789666,  5.47702766, 12.76408562,
       13.14777433,  5.08372149,  7.83534987, 12.7125079 , 13.09426142,
        5.96556248,  6.61386436,  7.57946459, 10.8207297 ,  8.09899874,
        8.06990483, 10.51437947,  6.67121748,  4.78258092,  5.92952538,
       12.4217523 , 11.18839169,  8.30125201,  9.50654967, 13.15525803,
        7.05627461,  8.05668609,  5.92281142,  9.15253344, 11.52733581,
       11.73397309,  9.62238828, 11.63502679,  5.94541803, 11.05317597,
        9.49511524,  8.09984231,  5.54776934,  5.14607014,  5.94909709,
        5.5748624 , 13.3607379 ,  8.74470218,  6.10774544,  5.24461068,
        6.38399789,  9.43737523, 12.05395056, 11.2845608 ,  7.46248947,
       11.018084  ,  9.62958973, 10.63038565, 12.18045978, 11.57249984,
       13.46553623, 14.02058621, 12.10942412, 11.27160035,  7.80625981,
       11.75618513, 12.4980845 ,  8.81930525,  5.53255882,  5.62005629,
        6.27905292,  4.85995002,  6.33517349, 10.1371552 , 10.01804223,
        9.9353921 , 13.95361098,  8.95133961,  9.94285511,  7.17789666,
        9.09119351, 13.06498554,  8.09704052, 12.00995898,  5.5513072 ,
        5.84450127,  4.98989906,  7.24326992,  8.19703512,  9.32094055,
       12.40074901,  6.49538841, 10.00794468, 10.06286074,  8.95983586,
        9.47369628, 10.7506423 , 10.37998078,  6.33517349,  6.83515201,
       10.00802627,  6.49538841,  6.54127431,  9.45995955,  5.87305005,
        7.83534987])}
{'P': tensor([[6.2731e-03, 2.6033e-01, 1.9044e-01,  ..., 1.2769e-06, 1.0846e-03,
         4.6165e-06],
        [5.7606e-05, 7.3219e-03, 4.1133e-01,  ..., 1.1618e-06, 1.2329e-03,
         7.0403e-06],
        [6.4514e-05, 1.0505e-04, 1.2830e-02,  ..., 1.7546e-06, 2.1507e-03,
         1.4592e-05],
        ...,
        [2.7450e-03, 1.5523e-03, 1.0650e-03,  ..., 3.7654e-02, 5.1748e-04,
         1.2023e-05],
        [3.3631e-04, 4.4583e-04, 5.6612e-04,  ..., 1.1416e-04, 7.3625e-02,
         5.1886e-04],
        [9.5103e-05, 8.1996e-05, 7.2957e-05,  ..., 2.5194e-03, 3.7594e-04,
         3.3222e-01]]), 'X': tensor([[ -5.8389,  -8.6056,  -3.5098],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -3.2053, -10.1993,  -3.8736],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -1.1885, -10.8211,   0.0865],
        [  9.4554,   9.4245,   9.8800],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 10.5678,   7.9261,  -0.6100],
        [  3.3094,  13.4969,  -3.6507],
        [ -6.4368,   5.8356,   3.3014],
        [  3.6469,   6.3624,  -1.8414],
        [  7.5064,  -8.9651,  -0.0994],
        [  7.7794,  11.0629,   8.9069],
        [  5.9912,   3.1453,  -3.9989],
        [  1.7232,  -4.7574,   3.1270],
        [ -1.5875,   7.8944,  -9.4286],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 10.8670,  -5.3428,  -2.8637],
        [  7.7794,  11.0629,   8.9069],
        [  6.2902,   0.0451,   3.7116],
        [  6.2242,  -0.3171,   7.4978],
        [  5.4167, -10.1293,   6.2156],
        [ -5.3382,   6.6533,   7.8054],
        [ -6.3066,  -8.1936,   9.5822],
        [  4.5491,  -3.3485,   6.8412],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.8934,   2.2597,   3.5831],
        [  6.3200,   0.2690,   1.5900],
        [ -9.3604,   0.4855,  -3.0158],
        [ -0.8455, -10.8853,  -6.7400],
        [  7.3724,  -9.0577,  -8.9843],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -7.3554,   4.8985,  -0.6665],
        [  2.5636, -10.9293,   9.6519],
        [ -3.1204, -10.2345,   9.2288],
        [ 11.6019,   5.9540,  -5.1715],
        [  0.6757, -11.0361,  -1.3047],
        [ -8.8363,   2.4086,  -5.6296],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.2571,   6.7072,   8.3693],
        [ -9.2301,  -3.1537,  -2.7477],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.7802,  -1.6363,  -6.2782],
        [  3.2692,  13.5098,  -4.9032],
        [  5.4179,  -2.2925,  -3.4641],
        [ 12.2921,  -1.7902,  -4.0870],
        [  0.6952,  -4.8060,   8.8976],
        [ -3.2736, -10.1703,   3.8712],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 12.2592,   3.9425,   2.0383],
        [ 11.6019,   5.9540,  -5.1715],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.4766,  -1.1116,  -2.1706],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.1984,  -0.4327,   1.5438],
        [  7.0642,  11.6104,  -9.8316],
        [  6.1737,   2.5305,   3.9049],
        [ -2.0840,   7.8415,  -7.8657],
        [  0.2571,  -4.7594,   1.5618],
        [  2.2076,  13.8032,  -1.5099],
        [  9.6587,   9.1817,  -0.8212],
        [ -6.7643,  -7.7372,  -9.6663],
        [  0.3354,  -4.7708,  -4.9125],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.9282,   2.1648,   8.4894],
        [  4.0078,  -3.8018,  -6.2168],
        [ -7.3554,   4.8985,  -0.6665],
        [ -3.3999, -10.1152,  -1.2088],
        [  0.1677,  -4.7447,  -6.1529],
        [  9.5261,  -7.1675,   1.4206],
        [ 11.5456,   6.0861,   8.1101],
        [ -5.1974,   6.7396,   3.3484],
        [  0.8940, -11.0402,  -0.9783],
        [  4.0457,  -3.7737,  -0.4704],
        [  7.5930,  -8.9037,   7.2323],
        [  2.0729,   7.2878,   7.0880],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.5770,   1.8512,  -0.6578],
        [ 11.5966,   5.9666,   6.0083],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -8.9942,  -3.9990,   6.0448],
        [  0.9967,  -4.8144,  -9.7370],
        [  9.9756,  -6.6330,  -3.3429],
        [  5.6095,  -1.9702,  -7.8233],
        [ 10.0400,   8.6922,  -6.2847],
        [  9.2811,  -7.4336,   5.1771],
        [  1.8766,   7.3694,   1.1887],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.4690,  -0.8528,  -9.3374],
        [  7.5309,  11.2617,   7.2656],
        [  9.6662,   9.1725,   8.1056],
        [ -9.4505,  -1.7081,  -0.7039],
        [  1.8721,  -4.7322,   0.5524],
        [  2.4697,  -4.5836,   1.4593],
        [  2.1379,  -4.6757,   6.4981],
        [  1.0391,  -4.8141,  -0.0768],
        [ -7.7842,   4.3466,  -1.7548],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.3858,   4.4260,  -9.6769]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0377, 0.0753,  ..., 0.3070, 0.8720, 0.9097],
        [0.0377, 0.0000, 0.0380,  ..., 0.2812, 0.8403, 0.8722],
        [0.0753, 0.0380, 0.0000,  ..., 0.2521, 0.8050, 0.8344],
        ...,
        [0.3070, 0.2812, 0.2521,  ..., 0.0000, 0.5753, 0.6996],
        [0.8720, 0.8403, 0.8050,  ..., 0.5753, 0.0000, 0.4207],
        [0.9097, 0.8722, 0.8344,  ..., 0.6996, 0.4207, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.39942497, 10.52357479, 10.61047596, 10.60226614, 10.6910768 ,
       10.5575601 , 10.88618114, 13.35013374,  5.38342077, 10.06407839,
       13.20988996, 13.89671314,  8.68832398,  7.33350642, 11.69264243,
       13.52430176,  6.76661595,  5.05989547,  8.05242591, 12.30858677,
        9.43737523, 12.10942412, 13.52430176,  6.29034958,  6.2322769 ,
       11.48663767,  8.53014915, 10.33958639,  5.64865173, 14.09176259,
        9.17595899,  6.32572323,  9.37296102, 10.91805683, 11.67875855,
       10.27716434, 11.57249984, 12.53285633,  8.83725081, 11.22596994,
       10.69963954, 13.04051744, 11.05672136,  9.15866607, 13.15525803,
       13.103709  ,  9.75401449,  5.9510489 ,  6.00732193, 13.89974693,
        5.8828986 , 12.4217523 ,  4.85603886, 10.68416401, 12.53285633,
       12.87752052, 13.04051744, 10.33712258,  9.54154096, 13.09737053,
        6.2134961 , 13.59056613,  6.67218687,  8.1137452 ,  4.7663526 ,
       13.97857591, 13.32645552, 10.27716434,  4.78258092, 12.59969936,
        9.18690018,  5.52416368,  8.83725081, 10.67132524,  4.74770899,
       11.92133129, 13.05148826,  8.51088913, 11.07637531,  5.53255882,
       11.70168755,  7.57685908,  9.66977802, 12.7125079 , 13.04156075,
        9.37989631,  9.97822473,  9.84314712,  4.91652556, 11.97957425,
        5.94541803, 13.27994304, 11.89105259,  7.60463349, 12.67576792,
       12.4980845 , 13.54774355, 13.32557133,  9.60359158,  5.08910491,
        5.20659892,  5.14124612,  4.92497514,  8.91551231,  4.9160962 ,
        6.97106912])}
{'P': tensor([[7.6432e-03, 3.9704e-01, 1.6908e-01,  ..., 4.3581e-05, 2.2043e-06,
         3.1550e-07],
        [1.4337e-04, 1.1818e-02, 3.2729e-01,  ..., 4.8181e-05, 3.8087e-06,
         4.0841e-07],
        [4.2362e-05, 5.2037e-05, 1.2061e-02,  ..., 1.3495e-05, 9.0489e-06,
         2.4732e-07],
        ...,
        [4.1447e-03, 4.1845e-03, 5.1210e-03,  ..., 1.5168e-02, 7.7225e-06,
         1.0406e-05],
        [5.4331e-08, 5.3700e-08, 6.8260e-08,  ..., 1.8079e-07, 1.1566e-02,
         9.7450e-07],
        [6.8544e-04, 5.9947e-04, 4.4071e-04,  ..., 2.2110e-02, 5.6549e-05,
         4.2960e-01]]), 'X': tensor([[ -5.7347,  -8.6908,   9.2878],
        [ -5.3179,  -9.0101,   9.4472],
        [ -3.1204, -10.2345,   9.2288],
        [ -2.0229, -10.6163,   8.6930],
        [ -3.7964,  -9.9294,   7.7147],
        [ -3.3116, -10.1539,   7.5919],
        [  5.0190,   4.9705,  -7.9752],
        [  3.3633,  13.4794,   0.5193],
        [  6.2419,  -0.2313,   4.2440],
        [  6.3534,   1.2990,  -7.2942],
        [ -5.1398,  -9.1367,   6.2074],
        [ -4.4503,   7.1378,   3.3751],
        [ -1.9600,   7.8577,   6.9121],
        [  3.1756, -10.8247,   6.8882],
        [  0.1047, -11.0046,   0.5188],
        [  1.4489,   7.5256,   4.4529],
        [  3.3474,  -4.2162,   5.2455],
        [ -5.5525,   6.5146,  -4.4984],
        [  4.6231,   5.4556,   8.4973],
        [  6.3087,   1.8053,  -1.9481],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.8902,  -1.3887,   4.0388],
        [ 11.7418,   5.6070,  -1.4192],
        [  7.3724,  -9.0577,  -8.9843],
        [ -7.4808,   4.7462,   2.0266],
        [ 11.5568,   6.0602,   9.3244],
        [  5.7802,  -1.6363,  -6.2782],
        [ -0.5385,   7.9037,   6.3830],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.8930,  12.3592,  -8.9201],
        [  6.2975,   0.0947,   5.1791],
        [ -8.3900,   3.3727,   3.1650],
        [ -4.0633,  -9.7927,  -3.0825],
        [  1.9241,   7.3503,   7.9004],
        [  5.4429,  -2.2529,   9.7306],
        [  5.5424,  -2.0885,   7.2487],
        [ 11.4743,   6.2478,  -2.1100],
        [ 10.8218,  -5.4177,   6.8573],
        [ -8.9788,   2.0208,  -3.2867],
        [ 12.0772,  -2.6014,  -7.1832],
        [  1.9505,  -4.7171,   9.9868],
        [ 11.1420,   6.9331,  -2.4779],
        [  2.7976,  -4.4682,  -4.2020],
        [ -9.4739,  -1.2497,   0.5483],
        [ -1.6906,   7.8860,   7.7142],
        [  9.3424,  -7.3685,  -3.6807],
        [ -6.7745,  -7.7264,   8.4684],
        [ -8.0611,  -6.0302,   4.6962],
        [  2.6613,   7.0021,   4.5783],
        [  5.5294,  -2.1106,   9.0626],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.1706,   1.3863,   8.4032],
        [ 12.6038,   0.7665,  -3.4439],
        [  3.6817,   6.3354,   6.1233],
        [ -7.6719,  -6.6260,   7.1468],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -0.8675, -10.8815,   6.8561],
        [  5.3804,  -2.3504,   7.2744],
        [  9.1620,  -7.5573,  -9.9882],
        [  6.2555,   2.1474,   4.1685],
        [ -4.9217,   6.8979,  -1.7667],
        [ 12.5832,   1.7543,  -7.6134],
        [  5.1292,   4.8183,  -5.7997],
        [ -8.8154,   2.4611,   2.5198],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.7659,   7.4127,   4.7871],
        [  7.5930,  -8.9037,   7.2323],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.2790,  -0.0259,  -5.4116],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.2608,  12.6977,   4.9166],
        [  5.1667,  -2.6543,   1.2879],
        [  4.4049,   5.6892,  -4.4394],
        [  5.9883,   3.1536,   6.2320],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -1.0482, -10.8487,  -5.9066],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.3376,  -4.6231,   9.8436],
        [  3.3633,  13.4794,   0.5193],
        [  1.7000,  -4.7609,   2.7935],
        [  4.4049,   5.6892,  -4.4394],
        [ 12.6059,   0.9597,  -6.3264],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.3200,   0.2690,   1.5900],
        [  5.6093,  -1.9705,   6.4807],
        [  6.1984,  -0.4327,   1.5438],
        [ -3.2736, -10.1703,   3.8712],
        [  2.4149,  -4.6005,  -4.3339],
        [  3.3167,   6.6022,   6.3554],
        [ 12.0582,  -2.6640,   0.7132],
        [  9.0129,  -7.7072,   9.1400],
        [  7.3592,  11.3937,   6.2407],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.9808,   4.9340,  -3.3715],
        [ -5.6792,  -8.7352,   4.0318],
        [ -5.1046,  -9.1610,   2.3418],
        [ 11.5966,   5.9666,   6.0083],
        [ 11.4495,   6.3026,   6.8257],
        [  2.0547,   7.2957,  -1.9343],
        [  5.9883,   3.1536,   6.2320],
        [  0.7145,  -4.8071,  -2.3766],
        [  2.0870,  -4.6877,   9.6823],
        [  8.0381,  -8.5690,  -8.9523],
        [ -8.0534,  -6.0429,   2.4583],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.3087,   0.7792,  -7.5587]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0232, 0.0934,  ..., 0.4414, 1.4340, 1.1036],
        [0.0232, 0.0000, 0.0704,  ..., 0.4503, 1.4230, 1.1093],
        [0.0934, 0.0704, 0.0000,  ..., 0.4745, 1.3826, 1.1197],
        ...,
        [0.4414, 0.4503, 0.4745,  ..., 0.0000, 1.1998, 0.6685],
        [1.4340, 1.4230, 1.3826,  ..., 1.1998, 0.0000, 0.9708],
        [1.1036, 1.1093, 1.1197,  ..., 0.6685, 0.9708, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.412299  , 10.46238054, 10.69963954, 10.80728788, 10.63038565,
       10.68031351,  7.06372172, 13.89264822,  6.24614905,  6.48486171,
       10.48315436,  8.41148722,  8.09842704, 11.28093507, 11.00508681,
        7.66377554,  5.38342077,  8.55982105,  7.15100303,  6.5618968 ,
        5.19986797,  6.05165248, 13.01187484, 11.67875855,  8.859416  ,
       13.0493368 ,  6.00732193,  7.92200342,  4.91652556, 13.6922425 ,
        6.29821673,  9.04255655, 10.60226614,  7.59795445,  5.89074104,
        5.92281142, 13.06498554, 12.10222204,  9.20340031, 12.35421461,
        5.10446653, 13.12299145,  5.27178363,  9.55593374,  8.06517143,
       11.89855267, 10.27572673, 10.06703358,  7.49076899,  5.91854397,
       10.80728788,  9.27474386, 12.62711252,  7.32754882, 10.1371552 ,
       10.60226614, 10.91601655,  5.87132718, 11.87665501,  6.61386436,
        8.47371868, 12.70489563,  7.03733775,  9.15253344,  6.27905292,
        7.62011734, 11.70168755,  9.68323686,  6.27905292, 10.38897563,
       13.744387  ,  5.80860165,  7.19513806,  6.76791483, 10.31613269,
       10.8992524 , 12.16901615,  5.1805345 , 13.89264822,  5.05534609,
        7.19513806, 12.64235359, 10.26327895,  6.32572323,  5.94535529,
        6.2134961 , 10.68416401,  5.19577146,  7.38844703, 12.3489324 ,
       11.85890664, 13.56367857, 12.53047489, 12.95702647, 10.41909211,
       10.48721382, 13.04156075, 13.06957425,  7.57946459,  6.76791483,
        4.85995002,  5.13125513, 11.74901778, 10.06850677, 12.3828957 ,
        9.34127013])}
{'P': tensor([[6.8979e-03, 3.6217e-01, 1.6794e-01,  ..., 1.1210e-06, 3.9011e-03,
         1.8658e-06],
        [6.3070e-05, 7.6249e-03, 2.8941e-01,  ..., 1.2789e-06, 6.8848e-03,
         1.6453e-06],
        [2.9088e-05, 4.2635e-05, 7.5049e-03,  ..., 1.4255e-06, 1.5076e-02,
         1.1265e-06],
        ...,
        [2.1691e-04, 1.6271e-04, 1.0175e-04,  ..., 1.6932e-01, 1.6273e-05,
         2.1071e-02],
        [1.2369e-05, 1.5997e-05, 2.7214e-05,  ..., 2.6156e-05, 5.8025e-02,
         3.5295e-06],
        [2.3922e-02, 1.5499e-02, 7.1182e-03,  ..., 8.9674e-05, 2.6017e-04,
         1.3000e-01]]), 'X': tensor([[ -5.6792,  -8.7352,   4.0318],
        [ -4.8751,  -9.3148,   4.0406],
        [ -3.2736, -10.1703,   3.8712],
        [ -3.5173, -10.0623,   4.8280],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.8788, -10.6567,   3.7068],
        [  3.9770, -10.6341,   8.0783],
        [  4.3459,  -3.5327,   0.3667],
        [  5.3860,  -2.3418,   1.0826],
        [ -6.0175,  -8.4541,  -9.8408],
        [  0.6470, -11.0352,  -1.1931],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.2767,  -4.6006,   0.7868],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.4932,   7.0903,  -5.4214],
        [  4.7482, -10.3907,   5.9149],
        [  3.3474,  -4.2162,   5.2455],
        [  3.6817,   6.3354,   6.1233],
        [  9.9217,  -6.7006,  -5.6728],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.3505,   0.6235,  -8.5849],
        [  0.5175,  14.0910,   4.6540],
        [  9.3753,   9.5171,   3.9131],
        [  0.3354,  -4.7708,  -4.9125],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.1531,  -5.8748,   2.6934],
        [  6.0560,  -0.9421,  -5.0334],
        [  8.7131,  10.2216,   7.7915],
        [  5.8201,   3.5841,  -8.0609],
        [  1.1274,  14.0116,   8.1753],
        [  9.4925,   9.3811,   8.9842],
        [  5.9843,  -9.8679,  -6.1449],
        [  2.5469,  13.7192,  -9.6935],
        [  5.5424,  -2.0885,   7.2487],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.3691,   6.4760,   1.0787],
        [  1.5246,  -4.7838,  -5.2095],
        [ -6.2668,   5.9808,   2.5614],
        [  6.2022,   2.4092,   3.0349],
        [  3.6469,   6.3624,  -1.8414],
        [  3.1480,  -4.3165,  -7.8002],
        [ -4.9217,   6.8979,  -1.7667],
        [  4.1999,  -3.6543,  -4.1998],
        [ -9.1706,   1.3863,   8.4032],
        [  6.0345,   3.0175,  -1.4424],
        [  4.3377,   5.7570,   1.3712],
        [ -5.3179,  -9.0101,   9.4472],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.0560,  -0.9421,  -5.0334],
        [  3.9770, -10.6341,   8.0783],
        [  0.6757, -11.0361,  -1.3047],
        [  1.1139,   7.6281,  -2.5680],
        [  0.6757, -11.0361,  -1.3047],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.3739,   0.3982,  -4.2564],
        [  9.2692,   9.6370,   7.0746],
        [  5.4179,  -2.2925,  -3.4641],
        [  3.7298,  13.3537,  -9.4653],
        [  7.5309,  11.2617,   7.2656],
        [  3.7298,  13.3537,  -9.4653],
        [ 10.9162,  -5.2599,  -7.0812],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.5267,  -0.4125,  -8.7331],
        [  1.8766,   7.3694,   1.1887],
        [  2.4572,   7.1085,   7.7376],
        [ -2.5056,   7.7720,   5.5878],
        [  6.3535,   0.6792,  -7.5101],
        [  8.5198,  10.4091,  -8.6486],
        [ 12.5650,   2.0132,   4.6272],
        [  3.4140,  -4.1803,  -8.8224],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.4877,  -0.7222,   8.8347],
        [  3.3069,   6.6089,   8.7377],
        [ -9.4671,  -0.5553,   0.1117],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.6319,  11.9066,   3.5165],
        [  3.2692,  13.5098,  -4.9032],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.4505,  -1.7081,  -0.7039],
        [  3.0876,   6.7527,  -5.3540],
        [ 11.7796,  -3.4696,   9.0088],
        [ -3.4986,   7.5152,   9.0712],
        [  6.1984,  -0.4327,   1.5438],
        [  3.0876,   6.7527,  -5.3540],
        [  1.6232,  -4.7717,  -2.5655],
        [  2.6310, -10.9195,   3.8054],
        [  4.3377,   5.7570,   1.3712],
        [  8.2422,  -8.4041,  -1.4962],
        [ -1.9600,   7.8577,   6.9121],
        [  3.6469,   6.3624,  -1.8414],
        [  1.4749,  -4.7892,  -0.8089],
        [ -2.1533, -10.5778,   1.0112],
        [ -0.3432,   7.8900,   7.1805],
        [  1.8719,   7.3713,  -3.7465],
        [  1.1006,   7.6318,  -2.4877],
        [  8.2154,  -8.4262,   8.0171],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.0345,   3.0175,  -1.4424],
        [  4.5057,  -3.3894,  -0.8160],
        [  2.3589,  13.7668,  -3.6501],
        [ -0.0254, -10.9932,   3.2200],
        [  3.2366,  -4.2732,   5.1532],
        [ -0.0282,   7.8575,  -1.1746],
        [  4.7482, -10.3907,   5.9149],
        [ -8.7544,   2.6096,   3.3146]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.4080e-02, 6.9141e-02,  ..., 8.0309e-01, 4.7139e-01,
         1.7961e-01],
        [2.4080e-02, 0.0000e+00, 4.6307e-02,  ..., 7.8756e-01, 4.4773e-01,
         1.6268e-01],
        [6.9141e-02, 4.6307e-02, 0.0000e+00,  ..., 7.4689e-01, 4.1008e-01,
         1.2252e-01],
        ...,
        [8.0309e-01, 7.8756e-01, 7.4689e-01,  ..., 0.0000e+00, 6.9641e-01,
         6.2504e-01],
        [4.7139e-01, 4.4773e-01, 4.1008e-01,  ..., 6.9641e-01, 0.0000e+00,
         3.9067e-01],
        [1.7961e-01, 1.6268e-01, 1.2252e-01,  ..., 6.2504e-01, 3.9067e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.41909211, 10.51340829, 10.68416401, 10.65929968, 10.76154256,
       10.82106505, 11.35345345,  5.60064319,  5.87305005, 10.37698435,
       11.05413545, 11.81423161, 12.17901033, 13.13205357,  7.5158489 ,
       11.42420508,  5.38342077,  7.32754882, 11.97238624, 13.01187484,
        6.38105015, 14.10045797, 13.35927527,  4.78258092, 12.46513646,
       10.04916678,  6.12886024, 13.43127131,  6.83515201, 14.05687815,
       13.34587027, 11.54071859, 13.95361098,  5.92281142,  5.1224079 ,
       13.08414689,  5.0209005 ,  8.66273171,  6.65368551,  7.33350642,
        5.34250225,  8.47371868,  5.5671389 ,  9.27474386,  6.74686071,
        7.20826219, 10.46238054, 10.05558893,  6.12886024, 11.35345345,
       11.05672136,  7.70897675, 11.05672136,  8.97052115,  9.38232121,
       13.37121923,  5.8828986 , 13.86480175, 13.54774355, 13.86480175,
       12.11735334, 11.21703654, 12.53345445,  7.60463349,  7.52117132,
        8.16584537,  6.38968357, 13.45124917, 12.72524615,  5.39723241,
       13.41713661, 12.5085995 ,  7.39004226,  9.48337063, 12.26997165,
       13.62896593, 13.89974693, 13.01187484,  9.60359158,  7.42512621,
       12.27993098,  8.28968067,  6.2134961 ,  7.42512621,  5.04028538,
       11.23201619,  7.20826219, 11.77124373,  8.09842704,  7.33350642,
        5.01114719, 10.794755  ,  7.89745107,  7.60529892,  7.71075226,
       11.76830981, 12.35421461, 11.57249984,  6.74686071,  5.63824971,
       13.96746816, 10.99326577,  5.36061974,  7.8575648 , 11.42420508,
        9.13507321])}
{'P': tensor([[1.0732e-02, 2.4621e-01, 1.7701e-01,  ..., 1.5669e-06, 2.1453e-06,
         7.7837e-07],
        [7.0411e-05, 1.8614e-02, 1.2850e-03,  ..., 2.6041e-06, 5.4420e-06,
         2.9141e-06],
        [5.0689e-05, 2.3195e-02, 6.6150e-03,  ..., 9.9736e-07, 2.6371e-06,
         1.2808e-06],
        ...,
        [2.2682e-04, 8.5419e-05, 1.2131e-04,  ..., 6.2212e-02, 3.4288e-05,
         1.6404e-06],
        [4.7499e-06, 3.5167e-06, 5.2593e-06,  ..., 1.1601e-03, 7.9871e-02,
         7.4261e-04],
        [4.5385e-08, 3.9659e-08, 5.3759e-08,  ..., 2.8208e-05, 4.2225e-03,
         1.3114e-02]]), 'X': tensor([[ -5.6169,  -8.7844,   8.9057],
        [ -3.1204, -10.2345,   9.2288],
        [ -3.7964,  -9.9294,   7.7147],
        [ -3.3116, -10.1539,   7.5919],
        [ -2.0229, -10.6163,   8.6930],
        [ -3.3812, -10.1235,   7.4384],
        [ -7.9298,   4.1367,  -3.9773],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 11.4743,   6.2478,  -2.1100],
        [  1.7228,  -4.7575,   8.3396],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.6314,  -1.9302,  -9.6543],
        [ -9.4203,  -2.0340,   2.5856],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.4203,  -2.0340,   2.5856],
        [ 11.3691,   6.4760,   1.0787],
        [ -1.2454,   7.9127,  -0.5715],
        [ 11.5568,   6.0602,   9.3244],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.9945,  -4.8144,  -0.1897],
        [ -2.4956, -10.4682,   3.5632],
        [  3.4604,  -4.1545,   1.8091],
        [  6.3535,   0.6792,  -7.5101],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.2301,  -3.1537,  -2.7477],
        [  3.4920,  -4.1366,  -8.3955],
        [ -6.2200,  -8.2737,   5.5079],
        [  3.1271, -10.8343,   2.6442],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.2644,  -4.6434,   9.7508],
        [ -0.1963,   7.8765,  -1.1371],
        [  9.1820,   9.7334,  -2.7932],
        [  6.3583,   1.1729,   8.7835],
        [  4.5814, -10.4485,   2.5496],
        [  0.9392,  -4.8143,  -3.4649],
        [ -4.9217,   6.8979,  -1.7667],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.6231,   5.4556,   8.4973],
        [  9.5018,  -7.1945,  -1.1892],
        [ -4.8751,  -9.3148,   4.0406],
        [ 11.2571,   6.7072,   8.3693],
        [  1.7815,  -4.7481,   4.4289],
        [  6.0140,  -1.0661,  -2.5040],
        [ -4.5778,  -9.5013,  -5.0667],
        [  0.7145,  -4.8071,  -2.3766],
        [ -3.9508,  -9.8515,   1.7923],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.1666,   3.7653,   3.6075],
        [ -7.8359,   4.2735,   7.2943],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.5240,  -2.1198,   6.2055],
        [  2.6613,   7.0021,   4.5783],
        [  5.6093,  -1.9705,   6.4807],
        [ 12.5832,   1.7543,  -7.6134],
        [ 11.0438,   7.1180,  -5.6991],
        [  5.4867,  12.5816,   2.9549],
        [ -5.3736,   6.6310,  -1.5174],
        [ -3.6381,   7.4683,   7.3544],
        [  6.7792,  11.8084,   1.6251],
        [ -6.3306,   5.9272,   5.6935],
        [  0.6757, -11.0361,  -1.3047],
        [ -5.6169,  -8.7844,   8.9057],
        [ -8.8835,  -4.3178,   9.3080],
        [  4.1297,  -3.7098,  -3.9838],
        [  9.1409,  -7.5789,   1.5814],
        [ -1.7290,   7.8825,  -3.3831],
        [  4.3377,   5.7570,   1.3712],
        [ -4.0868,   7.2981,   8.1874],
        [  9.4516,  -7.2501,  -1.8868],
        [  5.1196,  -2.7160,   4.2705],
        [  5.0506,   4.9277,  -9.5094],
        [ 12.0772,  -2.6014,  -7.1832],
        [  5.9587,  12.3216,   9.1695],
        [  9.3624,   9.5318,   0.2779],
        [ -9.0574,   1.7801,   8.8986],
        [  5.7485,  -9.9811,  -3.2355],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.2437,  -4.6489,   6.1430],
        [ -7.3470,  -7.0615,   5.0066],
        [ -2.8162,   7.7059,  -9.6885],
        [ 11.5568,   6.0602,   9.3244],
        [ -0.4341, -10.9473,   7.4509],
        [  5.0617,  12.7957,   8.3544],
        [ -2.5056,   7.7720,   5.5878],
        [  1.7659,   7.4127,   4.7871],
        [  6.1575,  12.2046,   3.3257],
        [ -6.2200,  -8.2737,   5.5079],
        [  5.6414, -10.0303,   7.8525],
        [  4.0587, -10.6112,   6.2273],
        [ 11.7779,   5.5127,  -2.0699],
        [ -9.3616,  -2.4749,   0.1793],
        [  2.0937,  13.8293,  -2.3533],
        [  4.0457,  -3.7737,  -0.4704],
        [ -0.8885,   7.9160,   0.0571],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.9260,  -4.4162,   8.3808],
        [  7.9403,  -8.6455,  -8.7433],
        [  3.6703,   6.3443,  -5.4401],
        [  5.6314,  -1.9302,  -9.6543],
        [ -2.5056,   7.7720,   5.5878],
        [  5.8318,   3.5567,  -4.7692],
        [ 12.4996,   2.6335,  -6.3166]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0891, 0.0275,  ..., 0.5590, 1.2443, 1.5274],
        [0.0891, 0.0000, 0.0822,  ..., 0.4857, 1.2023, 1.4840],
        [0.0275, 0.0822, 0.0000,  ..., 0.5366, 1.2170, 1.5001],
        ...,
        [0.5590, 0.4857, 0.5366,  ..., 0.0000, 0.7957, 1.0624],
        [1.2443, 1.2023, 1.2170,  ..., 0.7957, 0.0000, 0.2836],
        [1.5274, 1.4840, 1.5001,  ..., 1.0624, 0.2836, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.42667093, 10.69963954, 10.63038565, 10.68031351, 10.80728788,
       10.67322527,  8.94395628, 13.25336001,  9.19837792,  9.69629   ,
       13.06498554,  5.05981914, 13.09737053,  9.4997987 ,  5.95297459,
        9.63742669,  8.09899874,  9.63742669, 13.08414689,  8.01009083,
       13.0493368 ,  6.27905292,  4.9160962 , 10.76154256,  5.40689332,
        6.38968357, 10.19036081,  9.75401449,  5.41348121, 10.35093485,
       11.27656488, 10.0674887 ,  5.16611782,  7.8788965 , 13.38091121,
        6.46560181, 11.4088043 ,  4.90505566,  8.47371868,  9.97964013,
        7.15100303, 11.9182956 , 10.51340829, 13.103709  ,  5.0713364 ,
        6.10774544, 10.54658597,  4.85995002, 10.61417181,  8.09984231,
        8.99276564,  8.92549752,  7.02326515, 12.19805394,  5.91677188,
        7.49076899,  5.94535529, 12.70489563, 13.13888841, 13.72594022,
        8.53501777,  8.30729352, 13.61600536,  8.67227386, 11.05672136,
       10.42667093,  9.87720102,  5.5513072 , 11.87412439,  8.06990483,
        7.20826219,  8.36445758, 11.9120321 ,  5.79543038,  7.05627461,
       12.35421461, 13.68672855, 13.3607379 ,  9.23071698, 11.51811603,
        9.14973319,  5.16205087, 10.19036081,  8.20436764, 13.0493368 ,
       10.95593729, 13.76047999,  8.16584537,  7.62011734, 13.66991027,
       10.35093485, 11.50791859, 11.36089355, 13.00413829,  9.68323686,
       13.98691426,  5.53255882,  7.96575782, 12.24467041,  5.29755487,
       11.73848065,  7.32950385,  5.95297459,  8.16584537,  6.83083863,
       12.77402351])}
{'P': tensor([[6.1711e-03, 2.1229e-01, 1.9403e-01,  ..., 1.3557e-06, 1.3564e-06,
         6.9249e-05],
        [2.8055e-05, 4.5153e-03, 1.6440e-01,  ..., 6.5470e-07, 5.4840e-07,
         3.2628e-05],
        [4.9739e-05, 1.8244e-04, 1.0420e-02,  ..., 1.4074e-06, 1.1725e-06,
         7.8220e-05],
        ...,
        [1.9301e-03, 8.7972e-04, 7.7230e-04,  ..., 1.9424e-02, 6.1438e-03,
         8.3949e-05],
        [5.4768e-04, 3.1577e-04, 3.1370e-04,  ..., 1.9027e-04, 1.5735e-02,
         8.1903e-04],
        [1.0152e-05, 1.7910e-05, 2.3467e-05,  ..., 9.0747e-07, 2.2374e-06,
         1.8621e-02]]), 'X': tensor([[ -5.3328,  -8.9992,   7.3440],
        [ -3.7964,  -9.9294,   7.7147],
        [ -3.3812, -10.1235,   7.4384],
        [ -3.3116, -10.1539,   7.5919],
        [ -1.6334, -10.7207,   7.1397],
        [ -1.3940, -10.7772,   6.7835],
        [ -6.2668,   5.9808,   2.5614],
        [ -9.4773,  -1.0024,   5.2599],
        [  4.7544,  -3.1425,  -1.5315],
        [  6.3606,   0.9204,   3.2525],
        [ -9.3616,  -2.4749,   0.1793],
        [ -9.1178,   1.5784,   2.1410],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.5832,   1.7543,  -7.6134],
        [ 10.7566,   7.6212,  -2.3279],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.0829,  -5.9940,   5.0564],
        [  5.8450,  -9.9356,   3.0767],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.5240,  -2.1198,   6.2055],
        [  2.5318,  13.7231,   6.3692],
        [ -5.1398,  -9.1367,   6.2074],
        [  5.4732,  -2.2041,  -4.3830],
        [  1.1350,  14.0104,   0.6321],
        [  8.7131,  10.2216,   7.7915],
        [  2.2849, -10.9655,  -7.6542],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.5128, -10.9365,   9.9645],
        [  0.9478,  -4.8144,   0.6371],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.6095,  -1.9702,  -7.8233],
        [  2.6970,   6.9826,   9.7350],
        [  6.1741,  -0.5329,   3.1913],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.5375,   7.0676,  -4.3490],
        [  6.2555,   2.1474,   4.1685],
        [ -0.8569,   7.9156,  -6.5557],
        [ -9.4430,  -0.1897,   4.8383],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.9526,   3.2525,  -4.3821],
        [  2.7685, -10.8982,  -6.8928],
        [  6.1579,  -0.5961,  -6.5595],
        [ -9.3076,   0.7852,  -6.7010],
        [  0.8700,  -4.8132,  -7.8042],
        [ 12.6059,   0.9597,  -6.3264],
        [ -6.2200,  -8.2737,   5.5079],
        [ -3.4100,   7.5436,  -7.6980],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.4309,  -0.0599,  -4.0830],
        [  5.7151,  -1.7700,   3.1542],
        [  9.6587,   9.1817,  -0.8212],
        [  3.6817,   6.3354,   6.1233],
        [ -5.3328,  -8.9992,   7.3440],
        [ -5.1046,  -9.1610,   2.3418],
        [ -9.1706,   1.3863,   8.4032],
        [ 12.4690,  -0.8528,  -9.3374],
        [  2.8994,   6.8675,  -9.5987],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 11.0275,  -5.0666,   8.4154],
        [  1.0093,   7.6567,   2.3526],
        [  9.5261,  -7.1675,   1.4206],
        [ -7.7239,   4.4298,   3.1216],
        [  8.1043,  -8.5164,  -7.9290],
        [ -2.2508, -10.5479,  -2.7235],
        [ -3.5173,   7.5091,  -8.9585],
        [  4.3554,   5.7393,  -9.0766],
        [  5.2939,  -2.4784,  -4.5840],
        [ -8.1426,   3.8048,   9.3542],
        [  6.2022,   2.4092,   3.0349],
        [  4.3377,   5.7570,   1.3712],
        [  3.7992,  -3.9474,  -8.9019],
        [  1.0000,   7.6591,  -4.5011],
        [  4.7156,  -3.1831,  -8.0224],
        [  3.8182,  -3.9347,  -1.7961],
        [ -9.4739,  -1.2497,   0.5483],
        [ -0.5709,   7.9055,   7.6089],
        [  0.5250, -11.0306,   5.5358],
        [ 11.3691,   6.4760,   1.0787],
        [ 10.0995,  -6.4739,   2.0974],
        [  7.4132,  11.3526,   0.3322],
        [ -7.6734,   4.4980,  -5.4248],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.3739,   0.3982,  -4.2564],
        [ -9.3739,   0.3982,  -4.2564],
        [ -6.3306,   5.9272,   5.6935],
        [  1.6954,  -4.7616,   9.8198],
        [  6.3604,   0.9042,  -6.5901],
        [  6.3569,   0.7591,   6.5011],
        [  0.2435,   7.8191,  -6.4709],
        [  2.2644,  -4.6434,   9.7508],
        [  1.8766,   7.3694,   1.1887],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.4639, -10.9431,  -7.3718],
        [  8.6271,  -8.0715,  -7.4813],
        [  3.9022,  13.2906,   5.5788],
        [  1.9312,  13.8649,   8.3563],
        [ -7.9667,   4.0814,  -0.6397],
        [ -0.2586,   7.8825,  -9.9840],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -0.7599, -10.8995,  -6.0366]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 3.7563e-02, 4.8160e-02,  ..., 3.2997e-01, 7.8831e-01,
         1.0405e+00],
        [3.7563e-02, 0.0000e+00, 2.5823e-02,  ..., 3.5575e-01, 8.2246e-01,
         1.0706e+00],
        [4.8160e-02, 2.5823e-02, 0.0000e+00,  ..., 3.4073e-01, 8.1533e-01,
         1.0583e+00],
        ...,
        [3.2997e-01, 3.5575e-01, 3.4073e-01,  ..., 2.1453e-04, 5.0441e-01,
         7.2028e-01],
        [7.8831e-01, 8.2246e-01, 8.1533e-01,  ..., 5.0441e-01, 0.0000e+00,
         3.1807e-01],
        [1.0405e+00, 1.0706e+00, 1.0583e+00,  ..., 7.2028e-01, 3.1807e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.4606265 , 10.63038565, 10.67322527, 10.68031351, 10.84437705,
       10.86693708,  8.66273171,  9.53015376,  5.69913817,  6.42689705,
        9.68323686,  9.25336371,  8.22728399,  9.37989631, 12.67680439,
       12.70489563, 13.18278606,  9.87720102, 10.06286074, 11.52733581,
       13.1399181 ,  5.91677188, 13.95472331, 10.48315436,  5.90035324,
       14.05633441, 13.43127131, 11.20100795, 12.18045978, 11.22142195,
        4.90676242,  4.90505566,  5.94541803,  7.48538905,  6.19707944,
       13.14777433,  7.5092834 ,  6.61386436,  7.96181288,  9.44486609,
        9.5014146 ,  6.78325151, 11.24434374,  6.18668463,  9.34061938,
        4.89121805, 12.64235359, 10.35093485,  8.27853076,  7.53063885,
        9.43112957,  5.98285319, 13.32645552,  7.32754882, 10.4606265 ,
       10.48721382,  9.27474386, 12.4980845 ,  7.4544866 ,  8.09899874,
        9.3109884 ,  9.91668182, 12.13568598,  7.72291946, 11.92133129,
        8.90405823, 11.75618513, 10.78533838,  8.29203356,  7.20482224,
        5.84533345,  8.98765644,  6.65368551,  7.20826219,  5.47866221,
        7.72415048,  5.68942572,  5.48276038,  9.55593374,  7.92607342,
       11.04313641, 13.08414689, 11.99633819, 13.55868684,  8.89458273,
       13.20597932,  9.38232121,  9.38232121,  8.67227386,  5.05444288,
        6.42439344,  6.4020404 ,  7.822854  ,  5.16611782,  7.60463349,
       10.68031351,  9.75401449, 11.21703654, 11.81423161, 13.85158417,
       13.99877413,  8.95133961,  7.88677538,  9.13507321,  9.43737523,
       10.92597196])}
{'P': tensor([[1.2526e-02, 3.4692e-01, 1.9073e-01,  ..., 6.4210e-06, 1.1770e-05,
         1.0453e-03],
        [8.4133e-05, 1.9500e-02, 6.7623e-01,  ..., 2.3933e-05, 2.4475e-05,
         5.4941e-03],
        [1.4470e-04, 5.0455e-04, 5.1350e-02,  ..., 1.0912e-04, 8.5824e-05,
         2.5339e-02],
        ...,
        [7.9267e-07, 1.0204e-06, 1.3091e-06,  ..., 1.2996e-01, 6.2058e-04,
         3.3333e-05],
        [6.9874e-06, 9.8973e-06, 1.3581e-05,  ..., 1.4078e-02, 8.4629e-02,
         6.6918e-05],
        [1.5138e-06, 2.4105e-06, 3.1195e-06,  ..., 2.9774e-04, 1.5206e-05,
         2.2949e-02]]), 'X': tensor([[ -5.3179,  -9.0101,   9.4472],
        [ -3.1204, -10.2345,   9.2288],
        [ -2.0229, -10.6163,   8.6930],
        [ -3.3116, -10.1539,   7.5919],
        [ -3.7964,  -9.9294,   7.7147],
        [ -3.3812, -10.1235,   7.4384],
        [  4.9957,  -2.8705,   6.5978],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.8617,   3.4853,  -9.8080],
        [ -1.3940, -10.7772,   6.7835],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.9461,  -4.4077,  -0.4831],
        [  6.9956,  11.6590,  -7.6876],
        [ -2.2508, -10.5479,  -2.7235],
        [  6.8407,  11.7666,   3.0045],
        [  3.1271, -10.8343,   2.6442],
        [  4.7739,  -3.1218,  -4.1188],
        [  7.8375,  -8.7240,  -9.8913],
        [ 10.2470,   8.4054,  -0.1177],
        [ -6.4822,  -8.0252,  -3.8983],
        [  6.2242,  -0.3171,   7.4978],
        [ -3.4488, -10.0934,   2.5195],
        [  6.3048,   0.1469,  -9.1691],
        [ 11.4495,   6.3026,   6.8257],
        [  0.6396,  14.0772,  -6.3786],
        [  3.4140,  -4.1803,  -8.8224],
        [ -7.0825,   5.2073,  -5.0497],
        [  4.8674,  12.8875,   6.7622],
        [  4.0078,  -3.8018,  -6.2168],
        [  2.0899,  -4.6870,  -7.4260],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -1.6275, -10.7221,   3.2376],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.5503,   5.5359,   7.4735],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.1929,   2.4501,  -6.9352],
        [  4.0421,  -3.7765,  -8.3631],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -1.5261,   7.8988,   9.3768],
        [  0.3231,  -4.7691,  -1.6120],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.4932,   7.0903,  -5.4214],
        [  0.1931,  -4.7491,   7.4385],
        [ -8.4297,  -5.3637,  -9.3962],
        [  1.0000,   7.6591,  -4.5011],
        [ -5.6792,  -8.7352,   4.0318],
        [ 12.5770,   1.8512,  -0.6578],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.3331,  -2.6465,  -8.2028],
        [  1.4306,  13.9620,  -0.2075],
        [ 12.2589,  -1.9312,   9.8657],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.2365,  -0.2578,  -6.0212],
        [  0.1677,  -4.7447,  -6.1529],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.6095,  -1.9702,  -7.8233],
        [  3.2701,   6.6338,   9.9209],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.5900,   2.9767,   7.4811],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.1737,   2.5305,   3.9049],
        [  6.5428,  -9.5728,  -2.9897],
        [  2.7295,  13.6703,  -8.9328],
        [ -1.9714,   7.8563,  -1.8084],
        [ 12.6035,   0.7494,   7.1536],
        [  0.6933,   7.7336,   2.1179],
        [  3.9770, -10.6341,   8.0783],
        [  1.9312,  13.8649,   8.3563],
        [  3.9663,   6.1018,  -5.4104],
        [ -5.9940,   6.1985,   5.9566],
        [  6.1752,   2.5242,   3.5874],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.5568,   6.0602,   9.3244],
        [  8.3718,  -8.2953,   8.4666],
        [ -4.0633,  -9.7927,  -3.0825],
        [  3.2701,   6.6338,   9.9209],
        [ -6.3252,  -8.1760,  -1.0891],
        [  5.9587,  12.3216,   9.1695],
        [  0.5850,  -4.7982,   9.6639],
        [ -2.8162,   7.7059,  -9.6885],
        [ 10.4758,  -5.9539,   1.3667],
        [  7.4132,  11.3526,   0.3322],
        [  9.3753,   9.5171,   3.9131],
        [  5.9026,   3.3837,  -0.6573],
        [  5.4242, -10.1261,   3.9595],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.6363, -11.0348,   6.1414],
        [  0.3231,  -4.7691,  -1.6120],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.1793,   4.7461,   0.3899],
        [ -1.0012, -10.8576,   2.2867],
        [ -7.8849,  -6.3115,  -4.7479],
        [  4.1637,  13.1898,  -1.9189],
        [  2.7295,  13.6703,  -8.9328],
        [ -1.8788, -10.6567,   3.7068],
        [ 12.0845,  -2.5772,  -4.6073],
        [  3.9913,  -3.8139,  -6.8311],
        [  9.0129,  -7.7072,   9.1400]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0704, 0.1055,  ..., 1.3478, 1.2898, 0.6398],
        [0.0704, 0.0000, 0.0356,  ..., 1.3064, 1.2495, 0.5694],
        [0.1055, 0.0356, 0.0000,  ..., 1.2811, 1.2247, 0.5349],
        ...,
        [1.3478, 1.3064, 1.2811,  ..., 0.0000, 0.0621, 1.1061],
        [1.2898, 1.2495, 1.2247,  ..., 0.0621, 0.0000, 1.0661],
        [0.6398, 0.5694, 0.5349,  ..., 1.1061, 1.0661, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.46238054, 10.69963954, 10.80728788, 10.68031351, 10.63038565,
       10.67322527,  5.76165934,  5.0209005 ,  6.81961996, 10.86693708,
        6.1561646 ,  5.30158497, 13.59673406, 10.78533838, 13.6105595 ,
       11.27656488,  5.70405525, 11.72749923, 13.25336001, 10.31613269,
        6.2322769 , 10.66631784,  6.30647746, 13.06957425, 14.09176259,
        5.39723241,  8.7907923 , 13.77604293,  5.52416368,  5.13182506,
       10.1979669 , 10.09979747, 10.84493732,  5.5513072 ,  7.16598273,
       12.13568598,  6.65991877,  5.53175275,  9.69792513,  8.04483495,
        4.78002665, 10.31584102,  7.5158489 ,  4.75303127,  9.99145096,
        7.72415048, 10.41909211, 12.7125079 , 10.46739073, 10.36844813,
        7.822854  ,  9.70108409, 14.03505624, 12.41012046,  9.62238828,
        6.24187044,  4.74770899,  5.39723241,  5.94541803,  7.39599986,
        9.60359158,  8.1137452 ,  9.09119351, 10.32715715,  6.67218687,
       11.59512194, 13.94008948,  8.09984231, 12.62576026,  7.76457095,
       11.35345345, 13.99877413,  7.27759233,  8.62260327,  6.67121748,
       10.46739073, 10.53368644,  8.16584537, 13.0493368 , 11.785566  ,
       10.60226614,  7.39599986, 10.33712258, 13.68672855,  4.83371341,
        8.20436764, 12.0495521 , 13.55868684, 13.35927527,  6.80370254,
       11.48734978,  5.69913817, 10.00410625, 11.05317597,  4.78002665,
       13.05148826,  9.5014146 ,  7.02496223, 10.9036226 , 10.09979747,
       13.83139024, 13.94008948, 10.82106505, 12.35625694,  5.5205189 ,
       11.85890664])}
{'P': tensor([[7.3540e-03, 4.2735e-01, 1.6734e-01,  ..., 6.1986e-07, 8.1201e-08,
         5.5758e-04],
        [1.7311e-04, 1.3317e-02, 3.5505e-01,  ..., 1.0739e-06, 1.5006e-07,
         1.0493e-03],
        [3.8241e-04, 4.3128e-04, 8.3092e-02,  ..., 5.2305e-06, 1.1673e-06,
         7.5006e-03],
        ...,
        [3.0731e-04, 2.7010e-04, 1.1278e-04,  ..., 7.2347e-02, 3.3415e-05,
         2.6596e-05],
        [1.5719e-05, 1.3901e-05, 6.1927e-06,  ..., 9.4247e-02, 1.9360e-01,
         1.4563e-06],
        [9.4856e-06, 1.0342e-05, 2.1139e-05,  ..., 8.9169e-06, 7.6868e-06,
         5.8405e-02]]), 'X': tensor([[ -5.2752,  -9.0409,  -8.2568],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -3.0008, -10.2827,  -8.9480],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -2.9497, -10.3028,  -7.3479],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.9770, -10.6341,   8.0783],
        [ -2.8162,   7.7059,  -9.6885],
        [ -1.9646,   7.8571,   3.3817],
        [  6.9284,  11.7061,  -0.3732],
        [ -8.0750,  -6.0072,  -5.8267],
        [  1.9503,  -4.7172,  -6.2313],
        [  3.0719, -10.8449,  -4.1673],
        [ 12.1940,  -2.1881,   7.1499],
        [  0.1532,  -4.7422,  -7.0145],
        [ 11.5198,  -4.0923,   5.5088],
        [  0.9445,   7.6736,   0.5095],
        [  0.5850,  -4.7982,   9.6639],
        [ 11.7794,   5.5086,  -9.6929],
        [ 11.0275,  -5.0666,   8.4154],
        [ -1.9600,   7.8577,   6.9121],
        [  1.1006,   7.6318,  -2.4877],
        [  6.2675,  12.1379,  -8.9720],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.9074,  -4.8139,   8.1674],
        [ -1.5408,   7.8978,   5.3470],
        [  9.9756,  -6.6330,  -3.3429],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.8788, -10.6567,   3.7068],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.2707,   0.9661,   4.4863],
        [ -3.5173,   7.5091,  -8.9585],
        [  5.9193,  -1.3176,   5.1739],
        [  3.3943,   6.5483,   1.0790],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4196,   0.0478,   9.4140],
        [  5.9655,   3.2172,  -7.5953],
        [  0.8940, -11.0402,  -0.9783],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.1047, -11.0046,   0.5188],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 11.1469,   6.9237,   1.6160],
        [  9.9750,  -6.6338,  -1.2440],
        [ 12.2435,   4.0076,   8.1257],
        [ -4.1296,   7.2803,  -2.7194],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -0.0282,   7.8575,  -1.1746],
        [ 11.7162,   5.6728,  -8.9372],
        [ -7.0825,   5.2073,  -5.0497],
        [  3.5939,   6.4029,  -1.8357],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.2647, -10.8066,  -0.3768],
        [ -9.1178,   1.5784,   2.1410],
        [  0.2435,   7.8191,  -6.4709],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.4356,  -4.5942,  -5.0808],
        [  8.9284,  10.0036,  -6.9210],
        [ -6.7992,   5.4991,   9.9186],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.3569,   0.7591,   6.5011],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.4572,   7.1085,   7.7376],
        [ -5.9938,  -8.4745,   3.9735],
        [ -8.6426,  -4.9136,   9.4941],
        [ -6.2668,   5.9808,   2.5614],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.4117,   0.1160,   8.2574],
        [  2.7590,  13.6621,   9.4496],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.4739,  -1.2497,   0.5483],
        [  2.3373,  13.7721,  -0.2442],
        [ -8.6229,   2.9068,   2.4429],
        [ -5.3382,   6.6533,   7.8054],
        [ 10.6673,  -5.6649,  -4.1646],
        [  3.2701,   6.6338,   9.9209],
        [  9.0424,  -7.6779,  -7.5286],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.2939,  -2.4784,  -4.5840],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.2589,  -1.9312,   9.8657],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.1905,   3.7253,   4.1175],
        [  3.4920,  -4.1366,  -8.3955],
        [  1.6232,  -4.7717,  -2.5655],
        [  0.5250, -11.0306,   5.5358],
        [  1.0391,  -4.8141,  -0.0768],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.4639, -10.9431,  -7.3718],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -6.0195,   6.1789,  -2.9496],
        [ -0.5709,   7.9055,   7.6089],
        [  1.3339,   7.5627,  -0.9034],
        [  5.5424,  -2.0885,   7.2487],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.9178,   2.1936,   7.5388],
        [ 12.5927,   0.4199,  -9.4997],
        [  9.3424,  -7.3685,  -3.6807],
        [ -5.6651,   6.4380,   1.3660],
        [ -6.6895,   5.6051,   2.7899],
        [  0.9945,  -4.8144,  -0.1897],
        [ -5.1377,   6.7751,  -0.4438],
        [  1.6308,  13.9254,  -2.0984],
        [  4.8051, -10.3703,   0.5268]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.0953e-02, 1.4907e-01,  ..., 6.9185e-01, 1.1270e+00,
         7.2812e-01],
        [3.0954e-02, 0.0000e+00, 1.1823e-01,  ..., 6.7058e-01, 1.0974e+00,
         7.0754e-01],
        [1.4907e-01, 1.1823e-01, 0.0000e+00,  ..., 6.0671e-01, 9.8939e-01,
         6.4639e-01],
        ...,
        [6.9185e-01, 6.7058e-01, 6.0671e-01,  ..., 0.0000e+00, 6.1854e-01,
         4.2363e-02],
        [1.1270e+00, 1.0974e+00, 9.8939e-01,  ..., 6.1854e-01, 0.0000e+00,
         6.2255e-01],
        [7.2812e-01, 7.0754e-01, 6.4639e-01,  ..., 4.2363e-02, 6.2255e-01,
         7.5848e-05]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.46739073, 10.4977132 , 10.71162681, 10.57302546, 10.7167285 ,
       10.53368644, 11.35345345,  8.20436764,  8.09899874, 13.60275211,
       10.06437598,  5.10444097, 11.27160035, 12.38881635,  4.74468406,
       12.22503754,  7.73151309,  4.83371341, 13.00380624, 12.13568598,
        8.09842704,  7.71075226, 13.66051884,  8.47371868,  4.89869205,
        8.04664886, 11.97957425,  4.75303127, 10.82106505, 10.61047596,
        9.32094055,  8.29203356,  6.06416957,  7.37576154,  8.04483495,
        9.41970031,  6.77777606, 11.07637531, 13.06498554, 11.00508681,
        9.60359158, 13.12218013, 11.97949273, 12.88270018,  8.36995953,
       12.07818871,  7.8575648 , 13.01728313,  8.7907923 ,  7.34250948,
        6.25586297, 11.28895489,  9.25336371,  7.822854  ,  6.83083863,
        9.79814997,  5.19986797, 13.40850328,  8.74470218,  9.77588801,
        6.4020404 , 12.59969936,  7.52117132, 10.37998078,  9.94173708,
        8.66273171,  8.1137452 ,  9.41244865, 13.93790235,  5.18047638,
        5.53255882,  9.55593374, 13.96905756,  9.09963894,  8.53014915,
       12.07818871,  7.39599986, 11.86239961, 13.0743399 , 12.49317563,
        5.84533345, 13.41713661, 12.41012046, 11.81423161,  8.99791316,
        5.41348121,  5.04028538, 11.04313641,  4.92497514, 13.13205357,
       11.21703654, 10.27716434,  8.62631801,  7.92607342,  7.67939546,
        5.92281142,  9.3349858 ,  9.18358562, 12.59969936, 11.89855267,
        8.57560275,  8.7273569 ,  4.9160962 ,  8.50278896, 14.02058621,
       11.42947632])}
{'P': tensor([[8.8747e-03, 1.4165e-01, 1.2217e-01,  ..., 1.3783e-06, 7.8918e-08,
         1.5719e-05],
        [9.2205e-05, 1.0742e-02, 7.2742e-02,  ..., 1.5647e-06, 1.0672e-07,
         1.2926e-05],
        [1.1055e-04, 1.6063e-03, 1.2563e-02,  ..., 1.8112e-06, 1.2256e-07,
         1.4219e-05],
        ...,
        [4.5908e-04, 2.1649e-04, 2.0819e-04,  ..., 1.7943e-01, 6.6055e-05,
         6.0572e-05],
        [7.1729e-07, 3.4859e-07, 3.3237e-07,  ..., 1.9438e-03, 6.2387e-02,
         9.8338e-06],
        [1.5250e-05, 1.3448e-05, 1.2936e-05,  ..., 8.1641e-05, 4.5627e-04,
         1.0116e-01]]), 'X': tensor([[-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 9.9832e-02, 1.0768e-01,  ..., 5.6918e-01, 1.4889e+00,
         1.2325e+00],
        [9.9832e-02, 0.0000e+00, 8.0127e-03,  ..., 5.7688e-01, 1.5252e+00,
         1.2773e+00],
        [1.0768e-01, 8.0127e-03, 0.0000e+00,  ..., 5.7982e-01, 1.5298e+00,
         1.2826e+00],
        ...,
        [5.6918e-01, 5.7688e-01, 5.7982e-01,  ..., 0.0000e+00, 9.7435e-01,
         7.5743e-01],
        [1.4889e+00, 1.5252e+00, 1.5298e+00,  ..., 9.7435e-01, 0.0000e+00,
         2.8326e-01],
        [1.2325e+00, 1.2773e+00, 1.2826e+00,  ..., 7.5743e-01, 2.8326e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.48315436, 10.67322527, 10.68031351, 10.86693708, 10.84437705,
       10.65929968,  8.87007495, 10.65929968, 11.69264243, 13.6105595 ,
       13.96746816, 10.06286074, 11.73397309,  5.83560009,  7.24326992,
        8.55982105,  7.50564223,  7.15100303, 10.08001894,  9.5014146 ,
       10.52357479, 11.05317597, 11.97949273,  6.1561646 ,  8.8353674 ,
        5.95297459,  4.9160962 , 11.63724792, 10.794755  , 10.0674887 ,
       10.57302546, 13.61801526,  8.0550286 , 11.87665501, 11.86239961,
        8.22728399,  6.54615887,  7.10378919, 11.2845608 , 11.83570132,
       10.07238463, 11.00508681,  8.92549752,  8.29640613,  8.06990483,
       10.87409005, 12.16901615, 10.67322527,  5.39723241,  7.06169391,
       12.77402351, 12.97370371, 12.76408562,  7.66377554, 10.06437598,
        5.94909709,  8.53501777, 10.35093485, 11.66453059,  9.34061938,
        7.37902766,  7.5984809 ,  5.82345867, 10.1371552 , 10.5575601 ,
        6.28835101,  7.60529892,  5.13616224, 12.07225469,  5.07306274,
       11.62158554, 12.889695  ,  8.38252937,  8.85572292, 10.5575601 ,
        5.05981914,  6.24187044,  6.39438044,  7.81509631, 11.04953767,
       10.71162681,  6.81117802,  7.52117132,  5.5748624 , 12.17901033,
        7.6374603 , 10.57302546,  7.8575648 , 10.31613269,  7.52117132,
        9.97964013,  8.67502348,  8.20436764,  9.38232121, 12.30858677,
        5.70591986,  7.57685908,  4.73784222,  7.62509582,  7.92200342,
        7.24677664, 10.37998078,  5.8828986 ,  8.09899874, 13.6829843 ,
        5.39723241])}
{'P': tensor([[6.4943e-03, 2.3524e-01, 2.0890e-01,  ..., 1.4277e-06, 5.7990e-07,
         1.3232e-05],
        [9.7386e-05, 1.7573e-02, 7.9565e-04,  ..., 4.9710e-06, 2.6402e-06,
         6.9078e-05],
        [1.3862e-04, 1.5465e-01, 1.5135e-02,  ..., 4.6946e-06, 1.9863e-06,
         5.8561e-05],
        ...,
        [2.6684e-05, 2.0533e-05, 2.5543e-05,  ..., 2.8255e-01, 2.2658e-03,
         1.8405e-04],
        [7.5456e-07, 5.2335e-07, 5.8909e-07,  ..., 2.1857e-04, 3.1507e-02,
         1.2356e-06],
        [8.8182e-07, 9.9914e-07, 1.0437e-06,  ..., 1.4149e-02, 8.5246e-03,
         6.6451e-02]]), 'X': tensor([[ -5.1046,  -9.1610,   2.3418],
        [ -3.4488, -10.0934,   2.5195],
        [ -3.9508,  -9.8515,   1.7923],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.6275, -10.7221,   3.2376],
        [ -2.1533, -10.5778,   1.0112],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.0829,  -5.9940,   5.0564],
        [ -3.9508,  -9.8515,   1.7923],
        [ -3.9700,   7.3453,  -4.1486],
        [ 11.6165,  -3.8715,   6.4594],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.4756,  -0.8166,  -9.6021],
        [  1.7000,  -4.7609,   2.7935],
        [  1.1274,  14.0116,   8.1753],
        [  0.2480, -11.0153,  -3.0183],
        [  0.1532,  -4.7422,  -7.0145],
        [  9.0424,  -7.6779,  -7.5286],
        [  3.4401,  13.4540,   4.6576],
        [ -1.3940, -10.7772,   6.7835],
        [  0.1677,  -4.7447,  -6.1529],
        [  4.2041,  -3.6509,  -4.9488],
        [  0.9967,  -4.8144,  -9.7370],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.3360,   0.4258,  -9.2509],
        [ -0.8569,   7.9156,  -6.5557],
        [  8.2154,  -8.4262,   8.0171],
        [  5.9526,   3.2525,  -4.3821],
        [ -2.0445,   7.8469,  -0.5857],
        [  4.3523, -10.5232,   8.2138],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.7802,  -1.6363,  -6.2782],
        [ 12.2921,  -1.7902,  -4.0870],
        [  5.9526,   3.2525,  -4.3821],
        [  6.3205,   1.7068,  -3.8619],
        [  5.5424,  -2.0885,   7.2487],
        [  1.3339,   7.5627,  -0.9034],
        [  8.5093,  -8.1764,  -3.2381],
        [ 10.3251,   8.2928,  -3.0006],
        [  1.5662,  13.9375,   4.5142],
        [  5.5626,  -2.0536,   8.4433],
        [  9.3607,   9.5337,  -4.5412],
        [  6.1929,   2.4501,  -6.9352],
        [ -3.5173, -10.0623,   4.8280],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.6928,  12.4710,   3.6001],
        [  3.7992,  -3.9474,  -8.9019],
        [ 12.5199,  -0.4711,   3.0436],
        [ -1.9600,   7.8577,   6.9121],
        [ -7.4887,   4.7364,   8.9074],
        [  9.2979,   9.6048,   7.3138],
        [  2.5620,   7.0549,   8.4825],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.7558,  -4.4844,   0.0636],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.5935,   5.9740,  -4.3952],
        [ -2.2508, -10.5479,  -2.7235],
        [ 11.6544,   5.8273,  -3.1375],
        [  9.9756,  -6.6330,  -3.3429],
        [ -4.4503,   7.1378,   3.3751],
        [  2.5010,  -4.5736,  -0.8989],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.2188,  -2.5839,   2.1833],
        [  2.3373,  13.7721,  -0.2442],
        [  4.1348,   5.9514,  -0.8856],
        [ 12.4877,  -0.7222,   8.8347],
        [ -7.6293,  -6.6859,  -5.5161],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.4296,  -1.9458,   4.0833],
        [  7.0664,  -9.2590,   8.0879],
        [ -3.5029,   7.5138,   8.1853],
        [ -6.2200,  -8.2737,   5.5079],
        [  5.1667,  -2.6543,   1.2879],
        [  7.3724,  -9.0577,  -8.9843],
        [  4.1140,  -3.7220,  -0.1273],
        [  1.3767, -11.0341,   9.9415],
        [ -7.2515,   5.0195,   4.6434],
        [ -8.3900,   3.3727,   3.1650],
        [  2.5128, -10.9365,   9.9645],
        [  1.8766,   7.3694,   1.1887],
        [  2.3261,   7.1726,   6.2522],
        [  1.3767, -11.0341,   9.9415],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.9470,  13.6085,   2.9809],
        [ -5.6174,   6.4708,  -1.7339],
        [ 10.2470,   8.4054,  -0.1177],
        [ -8.8934,   2.2597,   3.5831],
        [  2.4932,   7.0903,  -5.4214],
        [  6.3606,   0.9204,   3.2525],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4117,   0.1160,   8.2574],
        [ -4.6393,   7.0461,   1.1624],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.1440, -10.9811,   8.5281],
        [ -5.2589,   6.7024,  -5.5809],
        [  1.9241,   7.3503,   7.9004],
        [  5.0506,   4.9277,  -9.5094],
        [  9.3624,   9.5318,   0.2779],
        [ 12.0772,  -2.6014,  -7.1832]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0645, 0.0593,  ..., 1.0476, 1.1745, 1.1976],
        [0.0645, 0.0000, 0.0728,  ..., 1.0185, 1.1204, 1.1681],
        [0.0593, 0.0728, 0.0000,  ..., 0.9893, 1.1296, 1.1394],
        ...,
        [1.0476, 1.0185, 0.9893,  ..., 0.0000, 0.5794, 0.1501],
        [1.1745, 1.1204, 1.1296,  ..., 0.5794, 0.0000, 0.6002],
        [1.1976, 1.1681, 1.1394,  ..., 0.1501, 0.6002, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.48721382, 10.66631784, 10.61417181, 10.76154256, 10.84493732,
       10.794755  ,  6.83515201,  9.69792513, 10.35485331,  6.383897  ,
       10.06286074, 10.61417181,  8.34949383, 12.24467041,  8.30125201,
        6.61211224,  9.51074763,  5.05534609, 14.05687815, 11.018084  ,
        4.74468406, 11.86239961, 13.88683436, 10.86693708,  4.74770899,
        5.56809335,  4.91652556, 12.27993098,  9.87720102,  6.35029274,
        7.96181288, 11.76830981,  6.78325151,  8.10886021, 11.38774149,
       10.53368644,  6.00732193, 12.4217523 ,  6.78325151,  6.54693987,
        5.92281142,  7.67939546, 11.80092578, 13.2430377 , 14.02526505,
        5.92952538, 13.36092624,  6.65991877, 10.65929968,  5.83560009,
       13.70893492,  5.47866221, 12.52875793,  8.09842704,  8.86082124,
       13.36800073,  7.50564223,  7.8788965 ,  5.26342203, 12.91505111,
       13.04217487, 10.78533838, 13.03002338, 11.97957425,  8.41148722,
        5.21280156,  4.91652556,  8.97052115,  9.71392179,  5.82345867,
       13.96905756,  7.24677664, 12.5085995 , 10.14436912,  6.34059021,
        9.6282763 , 11.64746882,  8.29022197, 10.35093485,  5.80860165,
       11.67875855,  5.54776934, 11.11969815,  8.81930525,  9.04255655,
       11.22142195,  7.60463349,  7.5403822 , 11.11969815,  8.53501777,
       13.92390689,  8.56889738, 13.25336001,  9.17595899,  7.5158489 ,
        6.42689705,  9.79039371,  9.41244865,  8.43626114, 12.16901615,
       11.18839169,  8.51928531,  7.59795445,  7.05627461, 13.3607379 ,
       12.35421461])}
{'P': tensor([[7.3757e-03, 3.1978e-01, 1.8182e-01,  ..., 3.4478e-05, 9.6324e-05,
         1.2930e-04],
        [4.7012e-05, 6.3960e-03, 2.9997e-01,  ..., 5.4071e-05, 1.5337e-04,
         1.5251e-04],
        [2.5699e-05, 4.2092e-05, 6.5149e-03,  ..., 1.0370e-04, 2.8991e-04,
         2.1769e-04],
        ...,
        [7.9846e-07, 8.5982e-07, 9.6752e-07,  ..., 3.6656e-02, 8.6490e-05,
         6.5726e-04],
        [2.1425e-06, 2.6073e-06, 3.3242e-06,  ..., 5.3179e-01, 7.1881e-02,
         1.1543e-03],
        [4.7644e-05, 4.9216e-05, 5.3134e-05,  ..., 2.7036e-03, 4.2387e-04,
         1.0614e-01]]), 'X': tensor([[-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0348, 0.0679,  ..., 0.9896, 0.9619, 0.9190],
        [0.0348, 0.0000, 0.0334,  ..., 0.9551, 0.9278, 0.8932],
        [0.0679, 0.0334, 0.0000,  ..., 0.9218, 0.8972, 0.8649],
        ...,
        [0.9896, 0.9551, 0.9218,  ..., 0.0000, 0.3287, 0.5495],
        [0.9619, 0.9278, 0.8972,  ..., 0.3287, 0.0000, 0.8333],
        [0.9190, 0.8932, 0.8649,  ..., 0.5495, 0.8333, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.48840476, 10.61472636, 10.74655346, 10.794755  , 10.87409005,
       10.88618114, 13.04156075, 11.42947632, 13.01187484,  9.78286733,
       12.67576792,  6.29821673, 12.49291768, 13.70893492, 10.84493732,
       10.53368644,  8.89458273, 10.32715715, 11.80092578,  6.77777606,
        9.94285511, 12.87752052,  8.09704052,  9.41970031,  5.62005629,
        5.77219077,  5.91222346,  7.50917551,  8.06643487, 11.63724792,
       10.69963954,  6.31576264,  7.83534987, 10.06850677, 10.7167285 ,
       10.41909211,  6.97106912,  5.87132718, 12.05395056, 10.75443106,
       13.12299145,  5.0209005 , 10.9036226 , 12.27835996, 12.59840193,
        5.43943791, 13.31764181,  7.37902766, 13.59673406, 10.1979669 ,
       11.63383221,  6.78325151,  6.38105015,  5.95297459,  6.48486171,
       10.42667093, 11.51557166,  8.66273171,  9.7751189 , 13.97779839,
       12.84056604, 10.6910768 ,  9.95552626,  7.48538905, 14.02526505,
       13.70893492,  9.70108409,  5.29755487, 11.85890664,  8.50268322,
       14.05633441, 13.2778378 ,  5.07306274,  7.48538905,  5.94535529,
        8.62162372,  6.80559669,  9.70108409,  8.16584537,  8.23525433,
       13.79762415, 11.97957425,  9.68324553,  7.98469166,  9.562174  ,
       12.40074901,  8.72896381, 11.83570132, 11.70168755,  9.73579738,
        6.4020404 , 12.78327037, 10.09979747,  5.38342077, 10.48721382,
       10.07238463,  6.74872981, 10.69963954,  8.90405823,  7.07902036,
        9.43112957,  9.09119351, 10.38897563, 12.41366754, 12.10222204,
        6.07992623])}
{'P': tensor([[9.5711e-03, 2.5517e-01, 1.8978e-01,  ..., 1.3376e-07, 5.5413e-04,
         3.7980e-06],
        [4.1666e-04, 8.0275e-02, 4.7743e-02,  ..., 8.7022e-07, 4.1519e-03,
         3.8133e-05],
        [1.5121e-04, 1.1075e-02, 2.5332e-02,  ..., 4.1198e-07, 2.3628e-03,
         2.1057e-05],
        ...,
        [4.6681e-06, 2.2348e-06, 3.6657e-06,  ..., 2.3104e-02, 2.1342e-05,
         2.3141e-05],
        [2.5589e-04, 3.6404e-04, 4.8383e-04,  ..., 1.0848e-04, 1.3433e-01,
         5.1515e-03],
        [1.0951e-05, 9.4080e-06, 1.4154e-05,  ..., 6.9281e-03, 7.9395e-04,
         2.4305e-01]]), 'X': tensor([[ -5.0132,  -9.2234,  -8.3531],
        [ -3.0008, -10.2827,  -8.9480],
        [ -2.9497, -10.3028,  -7.3479],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -0.8455, -10.8853,  -6.7400],
        [ -2.6069, -10.4298,  -6.1995],
        [  6.2804,   2.0019,   0.8607],
        [ -0.8675, -10.8815,   6.8561],
        [ 12.5832,   1.7543,  -7.6134],
        [  7.9141,  -8.6656,   9.6948],
        [  5.9548,  -1.2273,  -2.6520],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.0772,  -2.6014,  -7.1832],
        [  5.4336,  -2.2676,   5.6933],
        [  5.9026,   3.3837,  -0.6573],
        [  6.1752,   2.5242,   3.5874],
        [  9.0129,  -7.7072,   9.1400],
        [  3.2701,   6.6338,   9.9209],
        [  0.9788,  -4.8145,  -3.8661],
        [  2.8474,   6.8979,   6.6457],
        [ -0.0410,   7.8591,   9.3457],
        [ -3.0644,   7.6439,  -1.4322],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 10.9875,   7.2207,  -8.8549],
        [ -5.1974,   6.7396,   3.3484],
        [  0.9036,   7.6840,   9.9237],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 11.5083,  -4.1177,   0.2389],
        [ -4.8751,  -9.3148,   4.0406],
        [ -1.8788, -10.6567,   3.7068],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.7409,  -4.6837,  -6.5840],
        [  6.2026,  12.1774,  -1.5887],
        [  0.7145,  -4.8071,  -2.3766],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.9641,  -2.9568,   7.7211],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.8889,  -1.3916,   8.8993],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.1575,  12.2046,   3.3257],
        [ -8.1905,   3.7253,   4.1175],
        [  1.6408,   7.4591,  -4.2294],
        [ -9.3076,   0.7852,  -6.7010],
        [ 10.9875,   7.2207,  -8.8549],
        [  9.3836,   9.5075,  -5.9278],
        [  3.4604,  -4.1545,   1.8091],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.2974,   0.8371,   2.4852],
        [  6.9956,  11.6590,  -7.6876],
        [ 10.1798,   8.5002,  -5.8893],
        [ -2.9497, -10.3028,  -7.3479],
        [  1.0391,  -4.8141,  -0.0768],
        [  5.6095,  -1.9702,  -7.8233],
        [  3.2692,  13.5098,  -4.9032],
        [  0.3231,  -4.7691,  -1.6120],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -5.1369,   6.7755,   8.3836],
        [ -8.1905,   3.7253,   4.1175],
        [ -8.1666,   3.7653,   3.6075],
        [  2.4149,  -4.6005,  -4.3339],
        [ -9.2508,   1.0572,   2.9978],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -6.1914,  -8.2997,   8.5380],
        [  8.8091,  -7.9037,   0.3631],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.9770, -10.6341,   8.0783],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.7976,  -4.4682,  -4.2020],
        [ 12.5770,   1.8512,  -0.6578],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.3490,  -4.6199,  -4.2489],
        [ -6.4368,   5.8356,   3.3014],
        [ -8.5900,   2.9767,   7.4811],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -5.6174,   6.4708,  -1.7339],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.3616,  -2.4749,   0.1793],
        [ -5.3382,   6.6533,   7.8054],
        [  4.0587, -10.6112,   6.2273],
        [  3.4009,  -4.1874,  -5.9751],
        [  9.9217,  -6.7006,  -5.6728],
        [ -3.2736, -10.1703,   3.8712],
        [  8.5198,  10.4091,  -8.6486],
        [ 10.9875,   7.2207,  -8.8549],
        [  0.2480, -11.0153,  -3.0183],
        [ 12.4837,   2.7510,   0.4449],
        [ -2.0840,   7.8415,  -7.8657],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.1271, -10.8343,   2.6442],
        [  0.6142,  14.0802,   0.5407],
        [  0.9478,  -4.8144,   0.6371],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.4932,   7.0903,  -5.4214],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.8979,   3.3958,  -3.8713],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -4.0868,   7.2981,   8.1874],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.3569,   0.7591,   6.5011]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1182, 0.1265,  ..., 1.2131, 0.9246, 1.3691],
        [0.1182, 0.0000, 0.0457,  ..., 1.1750, 0.8249, 1.2964],
        [0.1265, 0.0457, 0.0000,  ..., 1.1312, 0.8001, 1.2598],
        ...,
        [1.2131, 1.1750, 1.1312,  ..., 0.0000, 0.7493, 0.4426],
        [0.9246, 0.8249, 0.8001,  ..., 0.7493, 0.0000, 0.5870],
        [1.3691, 1.2964, 1.2598,  ..., 0.4426, 0.5870, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.4977132 , 10.71162681, 10.7167285 , 10.57302546, 10.91805683,
       10.7506423 ,  6.59176018, 10.91601655, 12.70489563, 11.73567759,
        6.07992623, 12.889695  , 12.35421461,  5.88782913,  6.80370254,
        6.67121748, 11.85890664,  7.39599986,  4.91296205,  7.46248947,
        7.85919687,  8.23525433, 12.12448912, 13.14777433,  8.51088913,
        7.73692436, 10.06407839,  6.35029274,  9.93553424, 12.22276278,
       10.51340829, 10.82106505,  8.09704052,  9.75200848,  9.91668182,
       13.6660671 ,  4.85995002,  5.05963217, 12.32408516, 10.78533838,
        6.05113217,  8.62631801, 13.66991027,  8.99791316,  7.6374603 ,
        9.34061938, 13.14777433, 13.35832725,  5.40689332,  6.61211224,
        9.562174  , 11.99633819,  9.3349858 , 13.59673406, 13.26209545,
       10.7167285 ,  4.92497514,  5.94541803, 13.89974693,  4.78002665,
       12.53285633,  8.50268322,  8.99791316,  8.99276564,  5.19577146,
        9.3109884 , 10.61047596, 10.35464916, 11.83509372,  5.9510489 ,
       11.35345345, 13.79676654,  9.45995955,  5.27178363, 12.7125079 ,
       12.35421461,  9.20340031,  5.18277858,  8.68832398,  9.09119351,
       12.07818871,  8.56889738,  9.09502385,  9.68323686,  8.53014915,
       11.36089355,  5.39450305, 11.97238624, 10.68416401, 13.45124917,
       13.14777433, 11.018084  , 12.78327037,  8.1137452 , 10.6910768 ,
       11.27656488, 14.09357547,  4.90676242, 10.35485331,  7.5158489 ,
       11.67875855,  6.80559669, 12.49317563,  8.36445758,  5.16077764,
        6.4020404 ])}
{'P': tensor([[6.0034e-03, 2.2787e-01, 1.5025e-01,  ..., 2.0978e-04, 1.1643e-06,
         1.6444e-03],
        [2.8975e-05, 5.1004e-03, 2.3631e-01,  ..., 3.1237e-05, 1.7509e-06,
         1.4633e-03],
        [2.9584e-05, 7.2589e-05, 7.1009e-03,  ..., 2.3754e-05, 3.1574e-06,
         1.9777e-03],
        ...,
        [3.2262e-02, 8.3456e-02, 1.0449e-01,  ..., 2.3420e-02, 2.0816e-06,
         4.4777e-03],
        [3.0186e-07, 2.3675e-07, 2.1448e-07,  ..., 2.9345e-07, 1.7804e-02,
         4.0379e-06],
        [6.5333e-04, 1.0502e-03, 1.3439e-03,  ..., 4.4017e-04, 7.3091e-04,
         1.2087e-01]]), 'X': tensor([[-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0463, 0.0712,  ..., 0.2733, 1.2794, 0.8047],
        [0.0463, 0.0000, 0.0275,  ..., 0.2787, 1.2369, 0.7585],
        [0.0712, 0.0275, 0.0000,  ..., 0.2704, 1.2215, 0.7342],
        ...,
        [0.2733, 0.2787, 0.2704,  ..., 0.0000, 1.4161, 0.8409],
        [1.2794, 1.2369, 1.2215,  ..., 1.4161, 0.0000, 0.6681],
        [0.8047, 0.7585, 0.7342,  ..., 0.8409, 0.6681, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.51340829, 10.68416401, 10.76154256, 10.65929968, 10.82106505,
       10.84493732, 11.28215429, 11.05672136, 11.4424816 ,  7.14837514,
       13.03002338,  4.76969364,  6.10774544,  8.22728399,  9.5642993 ,
        7.61088125,  8.86082124,  5.56809335, 10.26545792,  6.39187285,
       10.76154256,  8.09899874,  9.50654967,  9.62238828,  7.24677664,
        6.38105015, 11.75807252,  6.39187285, 13.96905756,  9.43112957,
       11.92133129,  7.34250948,  5.64500512, 11.23201619, 10.68031351,
        9.69792513,  5.20659892,  7.50564223, 11.10066126, 10.52587243,
       11.83570132, 12.32545637,  9.76896904,  5.05534609,  7.92607342,
        8.50278896,  8.67502348,  6.1561646 ,  7.54127425, 13.43127131,
        7.10378919, 12.59969936, 11.48663767,  8.10886021,  9.45995955,
        4.87083348, 12.84056604, 11.1913598 , 12.88270018,  7.03733775,
       11.64746882, 10.00794468,  9.26455795,  9.58347456,  9.48969978,
        9.68324553,  5.91222346,  7.57685908,  6.61386436, 12.77402351,
       11.32149259,  8.50278896, 10.87409005,  7.32950385, 11.05413545,
       12.32545637, 12.1460931 ,  5.14371005, 12.22276278,  9.73579738,
        8.62260327, 10.00802627,  8.97052115,  9.27474386,  7.56617037,
        5.16205087, 11.38774149,  7.6374603 ,  7.13501264,  7.822854  ,
        8.36445758, 13.04217487,  8.87007495, 11.50791859,  8.72896381,
        4.85995002,  6.7593917 ,  9.18690018,  7.50035355,  6.65368551,
       13.14690137, 11.52733581,  7.42512621, 10.36844813, 13.14690137,
        4.98989906])}
{'P': tensor([[9.2231e-03, 3.6870e-01, 1.8711e-01,  ..., 2.4133e-04, 8.4205e-05,
         8.6626e-07],
        [3.2825e-04, 1.8217e-02, 3.3218e-01,  ..., 5.0647e-04, 1.8227e-04,
         1.9063e-06],
        [1.0322e-04, 1.3966e-04, 2.0100e-02,  ..., 2.1187e-04, 1.2845e-04,
         3.5912e-06],
        ...,
        [4.3018e-05, 5.2214e-05, 9.2459e-05,  ..., 5.0740e-02, 1.4152e-02,
         1.6125e-05],
        [5.6381e-05, 6.2617e-05, 6.5678e-05,  ..., 9.6861e-04, 8.1110e-02,
         2.9470e-04],
        [6.6137e-07, 6.2824e-07, 4.0750e-07,  ..., 2.6405e-07, 5.0816e-06,
         3.4658e-02]]), 'X': tensor([[ -4.8665,  -9.3204,  -5.2564],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -2.6069, -10.4298,  -6.1995],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -1.0482, -10.8487,  -5.9066],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3211,   1.7017,  -0.2164],
        [  8.5716,  10.3596,   6.8818],
        [  3.5939,   6.4029,  -1.8357],
        [ -6.6998,   5.5954,   0.0823],
        [  5.1203,  -2.7151,   6.2110],
        [ 12.0042,  -2.8350,   2.6864],
        [  5.5158,  -2.1337,   3.5246],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -1.3940, -10.7772,   6.7835],
        [  2.3589,  13.7668,  -3.6501],
        [  6.0140,  -1.0661,  -2.5040],
        [ -0.0410,   7.8591,   9.3457],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.5600, -11.0321,  -0.8660],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 12.5832,   1.7543,  -7.6134],
        [  2.2373,  -4.6506,  -1.7334],
        [  3.3474,  -4.2162,   5.2455],
        [  4.5954,  13.0099,  -4.4439],
        [ -0.3432,   7.8900,   7.1805],
        [ -7.6650,   4.5092,   0.2520],
        [  8.5716,  10.3596,   6.8818],
        [  0.6142,  14.0802,   0.5407],
        [ -8.0750,  -6.0072,  -5.8267],
        [ 12.5996,   1.3971,  -1.0694],
        [  6.2905,   1.9366,   2.1169],
        [  7.3724,  -9.0577,  -8.9843],
        [  1.9503,  -4.7172,  -6.2313],
        [  0.3038,   7.8092,   0.4164],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.3705,   0.4208,  -3.5815],
        [  4.5954,  13.0099,  -4.4439],
        [  5.9587,  12.3216,   9.1695],
        [ -8.6603,   2.8250,   5.9181],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.0054,  -1.0902,   8.3079],
        [ 12.5789,   0.1722,   1.6470],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 12.6038,   0.7665,  -3.4439],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.5636, -10.9293,   9.6519],
        [ -1.3940, -10.7772,   6.7835],
        [  5.7784,  -1.6401,   9.5587],
        [  3.3167,   6.6022,   6.3554],
        [  1.4306,  13.9620,  -0.2075],
        [ -8.8835,  -4.3178,   9.3080],
        [ 10.9599,  -5.1849,  -3.5526],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.7590,  13.6621,   9.4496],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.5763,   0.1333,  -9.2650],
        [  1.7122,  -4.7591,  -3.6734],
        [ -9.3087,   0.7792,  -7.5587],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.4716,  -0.6673,   7.6068],
        [  0.5175,  14.0910,   4.6540],
        [  7.3724,  -9.0577,  -8.9843],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 12.2328,  -2.0376,  -1.1234],
        [  2.5813,  -4.5471,   5.3597],
        [ -8.8745,   2.3097,   3.5600],
        [  2.9470,  13.6085,   2.9809],
        [ -9.1706,   1.3863,   8.4032],
        [ -4.3963,   7.1629,   4.5263],
        [  8.5198,  10.4091,  -8.6486],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.2022,   2.4092,   3.0349],
        [ -1.5408,   7.8978,   5.3470],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.4356,  -4.5942,  -5.0808],
        [  4.5430,   5.5438,  -3.8669],
        [ -3.7964,  -9.9294,   7.7147],
        [ -3.1671,   7.6158,   6.0765],
        [  3.9770, -10.6341,   8.0783],
        [  8.3791,  10.5409,   5.5590],
        [ -9.1706,   1.3863,   8.4032],
        [ -9.2373,  -3.1225,  -8.9763],
        [ 11.0438,   7.1180,  -5.6991],
        [ -8.8934,   2.2597,   3.5831],
        [ -1.0012, -10.8576,   2.2867],
        [ -9.4117,   0.1160,   8.2574],
        [  4.4290,  -3.4596,   1.4181],
        [  5.6022,  -1.9833,   8.3081],
        [  6.0955,   2.8210,  -4.3321],
        [ -5.9940,   6.1985,   5.9566],
        [ -8.9624,  -4.0944,  -4.9622],
        [  8.2154,  -8.4262,   8.0171],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 12.4837,   2.7510,   0.4449],
        [ -3.7964,  -9.9294,   7.7147],
        [ -2.2188,   7.8218,   8.6064],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 10.9916,   7.2133,  -0.9179],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -0.1626, -10.9796,   6.4629],
        [  2.2437,  -4.6489,   6.1430],
        [  9.6587,   9.1817,  -0.8212]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0207, 0.1224,  ..., 0.9737, 1.1776, 1.2035],
        [0.0207, 0.0000, 0.1094,  ..., 0.9660, 1.1601, 1.1829],
        [0.1224, 0.1094, 0.0000,  ..., 1.0306, 1.1603, 1.1302],
        ...,
        [0.9737, 0.9660, 1.0306,  ..., 0.0000, 0.6867, 1.1656],
        [1.1776, 1.1601, 1.1603,  ..., 0.6867, 0.0000, 0.5777],
        [1.2035, 1.1829, 1.1302,  ..., 1.1656, 0.5777, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.51437947, 10.54658597, 10.7506423 , 10.52587243, 10.8992524 ,
       10.8207297 ,  6.54615887, 13.44593519,  7.34250948,  8.72896381,
        5.79562691, 12.3344536 ,  5.91408504, 12.00995898, 10.86693708,
       13.96746816,  6.10774544,  7.85919687,  8.47371868, 11.04629471,
       12.10942412, 12.70489563,  5.16077764,  5.38342077, 13.79762415,
        7.89745107,  8.89301516, 13.44593519, 14.09357547, 10.06437598,
       12.67680439,  6.58184474, 11.67875855,  5.10444097,  7.81509631,
       12.11735334,  9.37989631, 13.79762415, 13.68672855,  9.10945678,
       12.26997165,  6.10360073, 12.58006038, 11.97238624,  9.5014146 ,
       12.62711252, 12.95702647, 11.22596994, 10.86693708,  6.00663314,
        7.38844703, 14.03505624,  9.87720102, 12.12448912,  4.8763135 ,
       13.93790235,  6.01356004, 13.13205357, 12.5769659 ,  5.05774295,
        9.34127013, 10.00802627,  9.49511524, 14.10045797, 11.67875855,
        9.60359158, 12.4013153 ,  5.2287058 ,  9.17016857, 13.92390689,
        9.27474386,  8.40445872, 13.45124917,  9.4997987 ,  4.78258092,
        6.65368551,  8.04664886,  9.37989631,  5.19986797,  7.16747878,
       10.63038565,  8.24808832, 11.35345345, 13.46553623,  9.27474386,
        9.75075273, 13.13888841,  9.17595899, 10.9036226 ,  9.41244865,
        5.62005629,  5.94293185,  6.7166297 ,  8.62260327,  9.85330646,
       11.76830981,  9.60359158, 12.78327037, 10.63038565,  8.1303945 ,
       12.49291768, 13.1471313 , 10.5575601 , 10.98076873,  5.16205087,
       13.32645552])}
{'P': tensor([[5.7742e-03, 3.2438e-01, 2.2641e-01,  ..., 1.1010e-06, 8.1706e-06,
         1.1549e-05],
        [6.1155e-05, 7.4693e-03, 4.2072e-01,  ..., 1.5421e-06, 1.2674e-05,
         2.0978e-05],
        [6.8961e-05, 1.0339e-04, 1.2144e-02,  ..., 2.6728e-06, 2.5353e-05,
         4.4304e-05],
        ...,
        [1.0534e-04, 8.4500e-05, 6.7794e-05,  ..., 1.8191e-01, 1.6377e-04,
         1.8877e-05],
        [7.0734e-05, 6.4584e-05, 6.1460e-05,  ..., 3.2809e-03, 3.0658e-01,
         2.6333e-05],
        [9.5341e-08, 1.0424e-07, 1.1230e-07,  ..., 1.2120e-04, 9.4914e-06,
         7.8948e-03]]), 'X': tensor([[-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0380, 0.0803,  ..., 0.8600, 0.8944, 1.3079],
        [0.0380, 0.0000, 0.0430,  ..., 0.8301, 0.8566, 1.2838],
        [0.0803, 0.0430, 0.0000,  ..., 0.8057, 0.8141, 1.2674],
        ...,
        [0.8600, 0.8301, 0.8057,  ..., 0.0000, 0.6544, 0.5061],
        [0.8944, 0.8566, 0.8141,  ..., 0.6544, 0.0000, 1.1044],
        [1.3079, 1.2838, 1.2674,  ..., 0.5061, 1.1044, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.52357479, 10.61047596, 10.6910768 , 10.75443106, 10.8207297 ,
       10.60226614,  6.02779201, 13.27717982,  9.64157461,  5.05444288,
       11.92133129,  7.18354885, 14.10045797,  4.89121805, 13.6105595 ,
       11.9702762 , 11.35182589, 12.30858677, 13.35013374,  5.91677188,
        4.74468406, 13.61801526,  5.1224079 , 13.89886469, 12.24467041,
        9.76896904,  5.43544626, 12.59969936, 11.32149259, 10.92597196,
       10.27716434,  8.77913428, 12.53285633,  5.92952538,  8.62260327,
        9.37296102, 12.16901615,  5.96556248, 11.75618513, 12.44177641,
        8.62631801, 14.09176259,  9.36378438, 11.62158554, 11.87665501,
        7.15990676, 11.83075005,  9.62238828, 11.63765273,  9.0368458 ,
       11.62770332,  9.38232121, 13.03002338, 11.1913598 ,  8.53014915,
        4.83371341, 12.87752052, 12.62576026, 13.18278606,  9.61942445,
        6.65259775,  8.31625171, 10.53368644,  8.62329466,  8.47371868,
       10.35464916, 13.1399181 , 13.66051884, 10.46739073, 13.77222752,
       13.6660671 ,  7.15990676, 11.59512194,  5.54776934, 12.79126785,
       13.83139024,  5.18047638, 12.41012046, 13.32374056, 12.17451685,
       13.95361098,  6.69389431, 12.12641969,  9.50913849,  6.21275016,
       12.12448912, 10.38897563, 13.60275211,  5.86519525,  5.05981914,
       13.97430191, 11.87412439,  9.88310614,  8.16714836, 10.24557047,
        6.59176018,  8.67502348, 11.51557166, 12.32545637, 11.75807252,
       13.76047999, 11.63724792, 12.22503754,  7.69752262,  6.81961996,
       12.30695098])}
{'P': tensor([[6.7389e-03, 4.5922e-01, 1.2564e-01,  ..., 1.9056e-06, 6.0685e-07,
         6.7389e-03],
        [1.5366e-04, 1.2644e-02, 2.3056e-01,  ..., 3.3576e-06, 1.2076e-06,
         1.5366e-04],
        [9.9053e-05, 1.1741e-04, 1.6899e-02,  ..., 1.5718e-06, 2.8774e-06,
         9.9053e-05],
        ...,
        [2.3070e-03, 2.3815e-03, 1.3774e-03,  ..., 3.7472e-01, 2.8110e-06,
         2.3070e-03],
        [6.1145e-07, 5.8488e-07, 3.8135e-07,  ..., 7.1430e-07, 2.6199e-02,
         6.1145e-07],
        [6.7389e-03, 4.5922e-01, 1.2564e-01,  ..., 1.9056e-06, 6.0685e-07,
         6.7389e-03]]), 'X': tensor([[ -4.7642,  -9.3860,  -5.1133],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -2.6069, -10.4298,  -6.1995],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.8823, -10.6558,  -3.8952],
        [ -1.0482, -10.8487,  -5.9066],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -9.2345,  -3.1345,   3.9715],
        [  9.4554,   9.4245,   9.8800],
        [  4.0591,   6.0202,  -1.6339],
        [ -6.9838,  -7.4965,   0.9796],
        [  1.3403,  -4.8011,  -1.1698],
        [ -2.2188,   7.8218,   8.6064],
        [  7.1697,  11.5343,  -0.7276],
        [  4.7156,  -3.1831,  -8.0224],
        [  8.3791,  10.5409,   5.5590],
        [  6.3542,   0.6934,  -3.3840],
        [  6.1575,  12.2046,   3.3257],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 12.3357,   3.6001,   0.7903],
        [  1.9503,  -4.7172,  -6.2313],
        [  3.5965,  -4.0754,  -3.5035],
        [  5.0617,  12.7957,   8.3544],
        [ -3.4100,   7.5436,  -7.6980],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.3724,   7.7974,  -6.8722],
        [ -1.8367,   7.8718,  -8.6523],
        [ -4.7642,  -9.3860,  -5.1133],
        [  6.0047,   3.1062,  -6.7914],
        [  6.1579,  -0.5961,  -6.5595],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.0233,   4.9647,  -3.1788],
        [ 10.0995,  -6.4739,   2.0974],
        [  8.2154,  -8.4262,   8.0171],
        [ 10.9931,   7.2107,   7.7933],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.8058,   2.4850,  -1.7425],
        [  3.1892, -10.8220,  -6.8409],
        [ -0.8569,   7.9156,  -6.5557],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.7794,   5.5086,  -9.6929],
        [  3.6469,   6.3624,  -1.8414],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.8979,   3.3958,  -3.8713],
        [ -6.3306,   5.9272,   5.6935],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.0961,   1.6529,   3.1553],
        [ -1.7009,   7.8851,  -8.2437],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.4602,   4.7718,  -4.6277],
        [  2.5663,   7.0526,  -6.8769],
        [  0.2480, -11.0153,  -3.0183],
        [ -8.1204,  -5.9308,   4.0761],
        [ -1.9646,   7.8571,   3.3817],
        [ -8.3900,   3.3727,   3.1650],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.3261,   7.1726,   6.2522],
        [  1.7301,   7.4262,   5.0522],
        [ -8.5607,  -5.0938,  -5.4531],
        [  0.7275,  14.0667,  -0.9815],
        [ -9.4196,   0.0478,   9.4140],
        [ -0.1626, -10.9796,   6.4629],
        [ -5.9225,  -8.5355,   2.4778],
        [  0.9445,   7.6736,   0.5095],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.3602,  -2.3808,   7.2098],
        [  2.2644,  -4.6434,   9.7508],
        [  7.7794,  11.0629,   8.9069],
        [ -9.3444,   0.5821,   6.9101],
        [  8.4691,  10.4571,  -5.0650],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.6928,  12.4710,   3.6001],
        [  9.5261,  -7.1675,   1.4206],
        [ 12.5267,  -0.4125,  -8.7331],
        [  5.1667,  -2.6543,   1.2879],
        [ 11.0275,  -5.0666,   8.4154],
        [  1.7218,  -4.7576,  -3.3617],
        [  6.1741,  -0.5329,   3.1913],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -3.5029,   7.5138,   8.1853],
        [  4.9957,  -2.8705,   6.5978],
        [  2.3589,  13.7668,  -3.6501],
        [ -1.5875,   7.8944,  -9.4286],
        [ -7.5401,   4.6718,   9.0033],
        [  6.0560,  -0.9421,  -5.0334],
        [  4.4039,   5.6903,   8.7366],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -8.4267,  -5.3697,   2.4172],
        [  1.7232,  -4.7574,   3.1270],
        [  7.9141,  -8.6656,   9.6948],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 10.4737,  -5.9570,  -1.2108],
        [  3.3633,  13.4794,   0.5193],
        [  1.9204,   7.3518,   8.4781],
        [ -9.4731,  -0.7120,  -6.0271],
        [  1.3767, -11.0341,   9.9415],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.5607,  -5.0938,  -5.4531],
        [  4.2096,   5.8815,   1.9230],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.0608,   3.9362,  -5.9567],
        [  0.9788,  -4.8145,  -3.8661],
        [ -8.6426,  -4.9136,   9.4941],
        [  9.0652,   9.8597,  -3.5949],
        [ -4.7642,  -9.3860,  -5.1133]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0238, 0.1278,  ..., 1.0015, 1.1578, 0.0000],
        [0.0238, 0.0000, 0.1094,  ..., 1.0023, 1.1340, 0.0238],
        [0.1278, 0.1094, 0.0000,  ..., 1.0834, 1.0617, 0.1278],
        ...,
        [1.0015, 1.0023, 1.0834,  ..., 0.0000, 1.5040, 1.0015],
        [1.1578, 1.1340, 1.0617,  ..., 1.5040, 0.0000, 1.1578],
        [0.0000, 0.0238, 0.1278,  ..., 1.0015, 1.1578, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.52587243, 10.54658597, 10.7506423 , 10.6910768 , 10.8207297 ,
       10.8992524 , 10.27716434,  9.75200848, 13.35013374,  7.26074925,
       10.24557047,  4.98461983,  8.1303945 , 13.58101794,  5.68942572,
       13.46553623,  6.39187285, 13.66991027,  9.96152788, 12.85032836,
        5.10444097,  5.43544626, 13.76047999,  8.27853076,  8.62631801,
        7.80625981,  8.08320235, 10.52587243,  6.76056625,  6.18668463,
       12.79126785,  7.06272256, 11.99633819, 11.76830981, 13.14690137,
        5.24137909,  9.14973319, 11.28215429,  7.96181288,  9.85330646,
       13.00380624,  7.33350642, 12.38881635,  6.80559669,  8.67227386,
        9.47369628,  9.24501829,  8.06643487,  8.90405823,  8.85572292,
        7.50500064, 11.018084  , 10.05558893,  8.09899874,  9.04255655,
       11.86239961,  7.5403822 ,  7.62509582,  9.96152788, 14.08549802,
        9.41970031, 10.98076873, 10.38897563,  7.73151309, 11.59512194,
        5.86519525,  5.16611782, 13.52430176,  9.36256096, 13.45642379,
        9.43737523, 13.70893492, 11.92133129, 12.53345445,  5.80860165,
       12.13568598,  5.05963217,  6.19707944,  6.14086582,  9.85330646,
        8.29022197,  5.76165934, 13.96746816,  8.05242591,  8.87007495,
        6.12886024,  7.19534146, 10.61047596,  9.99211663,  5.05989547,
       11.73567759, 12.07818871, 12.04924159, 13.89264822,  7.5984809 ,
        9.4997987 , 11.11969815,  5.94909709,  9.96152788,  7.23276582,
        7.78980299,  8.97052115,  4.91296205,  9.94173708, 13.39372426,
       10.52587243])}
{'P': tensor([[8.4509e-03, 4.8779e-01, 3.0725e-01,  ..., 2.3100e-05, 5.2457e-08,
         8.8944e-07],
        [1.7085e-04, 1.4217e-02, 6.4125e-01,  ..., 4.6055e-05, 9.8483e-08,
         1.4837e-06],
        [3.2306e-04, 3.7267e-04, 5.8619e-02,  ..., 3.5424e-04, 5.6855e-07,
         5.9062e-06],
        ...,
        [3.1592e-06, 3.1547e-06, 3.2525e-06,  ..., 1.6116e-01, 4.7210e-04,
         4.4514e-04],
        [8.1325e-08, 7.6987e-08, 6.0376e-08,  ..., 2.2494e-07, 2.2134e-02,
         3.0496e-04],
        [3.6545e-04, 3.2365e-04, 2.0183e-04,  ..., 4.7312e-06, 8.5531e-06,
         1.1030e-01]]), 'X': tensor([[-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0315, 0.1254,  ..., 1.1210, 1.8359, 0.5927],
        [0.0315, 0.0000, 0.0939,  ..., 1.0900, 1.8108, 0.5642],
        [0.1254, 0.0939, 0.0000,  ..., 0.9985, 1.7359, 0.4807],
        ...,
        [1.1210, 1.0900, 0.9985,  ..., 0.0000, 1.3581, 0.7245],
        [1.8359, 1.8108, 1.7359,  ..., 1.3581, 0.0000, 1.2623],
        [0.5927, 0.5642, 0.4807,  ..., 0.7245, 1.2623, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.53368644, 10.57302546, 10.71162681, 10.7167285 , 10.91805683,
       10.92597196, 13.46553623, 11.75807252, 13.44593519,  9.77777346,
       13.35013374,  5.05981914, 14.03618649,  5.07306274,  4.78258092,
       11.04953767,  8.29022197, 11.22142195,  8.43626114,  7.73692436,
        8.41148722,  5.26342203,  6.383897  , 10.1371552 , 13.6105595 ,
       14.03618649,  8.30729352,  7.88677538,  5.30158497,  7.17789666,
       11.48434767,  6.383897  , 13.98691426, 12.3828957 ,  8.7273569 ,
       13.70893492, 10.36844813, 12.13568598,  7.50564223, 11.8159254 ,
       13.97779839, 13.97430191, 10.04916678, 12.33167382,  5.13182506,
        5.76165934,  7.24326992,  6.42689705,  9.09661212, 12.18998195,
       13.01187484,  6.76661595,  8.67227386,  5.5205189 , 13.69162795,
        5.30158497, 11.75618513,  9.37296102, 11.07637531,  8.99276564,
        7.05627461,  9.36256096,  5.13125513, 11.48734978,  7.72415048,
       12.67576792,  5.30158497, 13.32645552, 12.78327037,  7.98469166,
       11.4088043 ,  7.15100303,  6.74872981,  6.76661595,  7.16598273,
       11.00726483,  8.92395381, 10.8992524 ,  6.74686071,  9.55720651,
       12.52372629,  5.82345867, 12.95702647,  5.26342203, 11.785566  ,
        9.70108409,  8.1303945 , 11.75438006,  4.78258092,  8.8353674 ,
       10.61047596,  8.92395381, 13.45124917, 10.30972193,  6.81084724,
       12.10222204,  9.32094055, 11.73567759,  5.05963217, 13.00413829,
        7.27340834, 12.18045978,  4.87083348, 12.59969936, 13.54774355,
        8.35856487])}
{'P': tensor([[9.7524e-03, 1.7783e-01, 1.5842e-01,  ..., 1.1221e-06, 1.4094e-03,
         2.0948e-06],
        [7.4993e-05, 1.0793e-02, 1.5974e-02,  ..., 1.1827e-06, 9.4085e-04,
         5.1745e-06],
        [2.4431e-04, 4.9625e-03, 5.6551e-02,  ..., 5.0786e-06, 8.8201e-03,
         2.9154e-05],
        ...,
        [9.4096e-04, 4.9062e-04, 2.8954e-04,  ..., 1.5655e-01, 3.7231e-04,
         9.8447e-06],
        [3.3673e-04, 5.0041e-04, 8.6340e-04,  ..., 4.9314e-05, 5.4317e-02,
         1.4264e-04],
        [1.2520e-06, 1.0286e-06, 8.4866e-07,  ..., 1.6874e-03, 1.1567e-05,
         7.7853e-02]]), 'X': tensor([[ -4.5778,  -9.5013,  -5.0667],
        [ -2.6069, -10.4298,  -6.1995],
        [ -1.8823, -10.6558,  -3.8952],
        [ -1.0482, -10.8487,  -5.9066],
        [ -3.2053, -10.1993,  -3.8736],
        [ -0.7599, -10.8995,  -6.0366],
        [ -5.6792,  -8.7352,   4.0318],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.6381,  -4.9238,  -4.4077],
        [  1.2273,  -4.8081,   9.8129],
        [ -2.5056,   7.7720,   5.5878],
        [ 12.2921,  -1.7902,  -4.0870],
        [  9.1820,   9.7334,  -2.7932],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.2571,   6.7072,   8.3693],
        [ 10.7566,   7.6212,  -2.3279],
        [  2.1477,   7.2550,   8.9688],
        [  2.4356,  -4.5942,  -5.0808],
        [ -7.6525,  -6.6534,   5.1721],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.1741,  -0.5329,   3.1913],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.2608,  12.6977,   4.9166],
        [ 12.5763,   0.1333,  -9.2650],
        [ 11.7794,   5.5086,  -9.6929],
        [ -9.4117,   0.1160,   8.2574],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.2218,   4.0953,  -8.2072],
        [  7.7794,  11.0629,   8.9069],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.5974,   7.0363,  -7.7810],
        [ 12.2921,  -1.7902,  -4.0870],
        [  3.7121,   6.3117,   0.6738],
        [  5.4179,  -2.2925,  -3.4641],
        [ -6.1899,  -8.3011,  -7.9269],
        [  0.8700,  -4.8132,  -7.8042],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.3124,   0.2057,  -0.4593],
        [  3.0876,   6.7527,  -5.3540],
        [  5.4179,  -2.2925,  -3.4641],
        [  4.2041,  -3.6509,  -4.9488],
        [ 12.3609,   3.4769,  -8.5762],
        [  0.9809,   7.6642,  -1.4217],
        [  4.7482, -10.3907,   5.9149],
        [  5.6672,  -1.8631,   8.1020],
        [ -5.0943,  -9.1682,   0.5073],
        [  6.9284,  11.7061,  -0.3732],
        [  6.0073,   3.0986,   6.7937],
        [  5.8275,  -1.5334,   2.0124],
        [  9.3836,   9.5075,  -5.9278],
        [ -3.5173,   7.5091,  -8.9585],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.5318,  13.7231,   6.3692],
        [  7.3592,  11.3937,   6.2407],
        [  2.7121,   6.9743,  -8.4202],
        [  2.6310, -10.9195,   3.8054],
        [ -8.0829,  -5.9940,   5.0564],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.4524,  -1.6813,  -6.6862],
        [  3.6817,   6.3354,   6.1233],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.3087,   0.7792,  -7.5587],
        [  4.5057,  -3.3894,  -0.8160],
        [ 11.5198,  -4.0923,   5.5088],
        [  4.5814, -10.4485,   2.5496],
        [ -9.3464,   0.5708,  -1.6397],
        [ 11.3367,  -4.4806,   0.4643],
        [  2.5829,  -4.5466,  -8.4015],
        [  1.3670,  -4.7990,   3.2603],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -1.2454,   7.9127,  -0.5715],
        [  3.3839,  -4.1966,   8.3574],
        [  2.5375,   7.0676,  -4.3490],
        [ -1.8823, -10.6558,  -3.8952],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.1737,   2.5305,   3.9049],
        [  4.3377,   5.7570,   1.3712],
        [ -1.5875,   7.8944,  -9.4286],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.8602,   6.8905,   5.8779],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.6112,   2.9319,  -7.9943],
        [ -8.1905,   3.7253,   4.1175],
        [ -7.4808,   4.7462,   2.0266],
        [  9.1620,  -7.5573,  -9.9882],
        [ -7.9925,  -6.1423,   2.8990],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.2366,  -4.2732,   5.1532],
        [ 10.9717,  -5.1646,  -6.9039],
        [  9.7328,   9.0902,  -0.8917],
        [ -8.6681,  -4.8556,   9.2518],
        [  0.6470, -11.0352,  -1.1931],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.5382,   7.0672,   4.5097],
        [ -3.5173,   7.5091,  -8.9585],
        [  2.2373,  -4.6506,  -1.7334],
        [ -7.4807,   4.7463,   2.0616],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.1504,   4.3674,  -7.2998],
        [ -1.6220,   7.8917,   0.0315],
        [  3.3943,   6.5483,   1.0790],
        [ -5.0550,   6.8231,  -9.2502],
        [  0.9659,  -4.8144,  -0.9594],
        [ 10.9875,   7.2207,  -8.8549]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0935e-01, 1.3042e-01,  ..., 3.7278e-01, 8.1350e-01,
         1.1769e+00],
        [1.0935e-01, 0.0000e+00, 1.4281e-01,  ..., 2.7944e-01, 7.7441e-01,
         1.0857e+00],
        [1.3042e-01, 1.4281e-01, 0.0000e+00,  ..., 2.9699e-01, 6.8427e-01,
         1.0810e+00],
        ...,
        [3.7278e-01, 2.7944e-01, 2.9699e-01,  ..., 2.1453e-04, 5.7227e-01,
         8.0645e-01],
        [8.1350e-01, 7.7441e-01, 6.8427e-01,  ..., 5.7227e-01, 0.0000e+00,
         6.6958e-01],
        [1.1769e+00, 1.0857e+00, 1.0810e+00,  ..., 8.0645e-01, 6.6958e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.54658597, 10.7506423 , 10.8207297 , 10.8992524 , 10.6910768 ,
       10.92597196, 10.41909211, 12.62576026,  9.94285511,  4.96231532,
        8.16584537, 12.4217523 , 13.38091121,  9.85330646, 13.103709  ,
       13.18278606,  7.56617037,  5.19986797, 10.14044489,  5.90035324,
        6.19707944, 10.61047596,  5.90035324, 13.744387  , 12.5769659 ,
       13.00380624,  9.41244865, 11.57249984,  6.2134961 , 12.889695  ,
       13.52430176,  9.96391045,  7.50035355, 12.4217523 ,  7.32233733,
        5.8828986 , 10.35485331,  4.89121805,  5.01114719,  6.31576264,
        7.42512621,  5.8828986 ,  5.56809335, 12.84056604,  7.72668467,
       11.42420508,  5.96556248, 10.48840476, 13.60275211,  6.7593917 ,
        6.02588517, 13.35832725,  8.29203356, 10.41909211, 13.95472331,
       13.56367857,  7.48311002, 11.23201619, 10.06286074,  5.53255882,
        9.60080639,  7.32754882,  7.53063885,  9.34127013,  5.63824971,
       12.22503754, 11.4088043 ,  9.36378438, 12.18998195,  5.22903117,
        4.98989906,  9.60080639,  8.01009083,  5.39098669,  7.5092834 ,
       10.8207297 , 12.84056604,  6.67218687,  7.20826219,  8.05242591,
       12.10222204,  7.46053281,  9.27474386,  9.09661212,  8.99791316,
        8.859416  , 11.87665501, 10.08001894,  8.22728399,  5.36061974,
       12.12641969, 13.31764181,  9.9353921 , 11.05413545, 12.07818871,
        7.50917551,  8.29203356,  5.16077764,  8.85939626, 12.50996828,
       12.91143491,  8.05668609,  7.37576154,  8.49161154,  4.91038884,
       13.14777433])}
{'P': tensor([[7.4909e-03, 2.3105e-01, 2.2036e-01,  ..., 5.6907e-05, 1.3156e-03,
         3.1921e-07],
        [7.2564e-05, 1.3719e-02, 5.7484e-01,  ..., 1.8359e-04, 2.7684e-03,
         5.0935e-07],
        [2.2457e-04, 8.9552e-04, 5.1958e-02,  ..., 8.3061e-04, 1.1196e-02,
         2.0088e-06],
        ...,
        [5.5772e-07, 7.5259e-07, 8.2922e-07,  ..., 2.4061e-02, 7.1869e-05,
         2.4164e-05],
        [1.7979e-04, 2.7237e-04, 2.9840e-04,  ..., 2.5203e-03, 7.1881e-02,
         8.5138e-05],
        [7.0176e-05, 3.4886e-05, 3.3613e-05,  ..., 8.1123e-06, 9.7537e-05,
         1.0513e-01]]), 'X': tensor([[ -4.4779,  -9.5609,  -2.4505],
        [ -2.5683, -10.4433,  -2.9182],
        [ -2.2508, -10.5479,  -2.7235],
        [ -2.6735, -10.4062,  -1.5911],
        [ -1.8823, -10.6558,  -3.8952],
        [ -3.3999, -10.1152,  -1.2088],
        [ -6.0195,   6.1789,  -2.9496],
        [ -2.1533, -10.5778,   1.0112],
        [ -5.9940,   6.1985,   5.9566],
        [ -3.5520,   7.4976,   9.1522],
        [  2.6451,  -4.5250,  -0.5565],
        [  0.8700,  -4.8132,  -7.8042],
        [  3.3943,   6.5483,   1.0790],
        [  6.2247,  -0.3150,  -9.3657],
        [ 11.6165,  -3.8715,   6.4594],
        [  0.2732,  -4.7619,  -1.1577],
        [ -6.0844,  -8.3955,   0.7705],
        [  1.7218,  -4.7576,  -3.3617],
        [  6.3587,   1.1596,  -3.6629],
        [  2.7558,  -4.4844,   0.0636],
        [  0.3715,  -4.7756,   0.2312],
        [ 11.3691,   6.4760,   1.0787],
        [  2.3376,  -4.6231,   9.8436],
        [  6.2895,   0.0405,   5.7652],
        [  6.3460,   1.4252,   1.2291],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.3606,   0.9204,   3.2525],
        [ -7.2515,   5.0195,   4.6434],
        [  2.0243,   7.3086,   7.5432],
        [  8.8474,  10.0868,  -8.8644],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.3943,   6.5483,   1.0790],
        [ -6.3306,   5.9272,   5.6935],
        [ -8.0330,  -6.0765,   6.7945],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.5128, -10.9365,   9.9645],
        [  3.1892, -10.8220,  -6.8409],
        [  4.4915,   5.5990,   7.0405],
        [ -9.3604,   0.4855,  -3.0158],
        [ -1.0012, -10.8576,   2.2867],
        [  5.4242, -10.1261,   3.9595],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.0582,  -2.6640,   0.7132],
        [  1.5662,  13.9375,   4.5142],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.3860,  -2.3418,   1.0826],
        [  3.9913,  -3.8139,  -6.8311],
        [  3.2159, -10.8166,  -1.0406],
        [  1.5653,  13.9377,   4.1773],
        [ -7.4807,   4.7463,   2.0616],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.1196,  -2.7160,   4.2705],
        [ -5.1046,  -9.1610,   2.3418],
        [  1.4489,   7.5256,   4.4529],
        [ -8.4267,  -5.3697,   2.4172],
        [ -9.4714,  -1.3301,  -3.1998],
        [  2.5010,  -4.5736,  -0.8989],
        [  1.8719,   7.3713,  -3.7465],
        [  5.6022,  -1.9833,   8.3081],
        [  6.2883,   0.0325,   5.2692],
        [ -7.4602,   4.7718,  -4.6277],
        [ -8.3900,   3.3727,   3.1650],
        [ -8.3653,   3.4184,   9.3454],
        [ -4.7389,   6.9953,   3.7638],
        [ -7.3554,   4.8985,  -0.6665],
        [ -3.5173, -10.0623,   4.8280],
        [ -5.9872,   6.2037,  -9.4477],
        [  0.1677,  -4.7447,  -6.1529],
        [  2.3200,   7.1756,   1.1938],
        [ -9.4430,  -0.1897,   4.8383],
        [ -9.4739,  -1.2497,   0.5483],
        [ -8.6112,   2.9319,  -7.9943],
        [  7.5930,  -8.9037,   7.2323],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.6634,   5.8050,   9.1590],
        [  2.9470,  13.6085,   2.9809],
        [ -1.3940, -10.7772,   6.7835],
        [ 10.4737,  -5.9570,  -1.2108],
        [  0.4996,   7.7738,  -4.4711],
        [  6.0890,  -0.8382,   2.9166],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.5158,  -2.1337,   3.5246],
        [  1.6232,  -4.7717,  -2.5655],
        [ 10.6283,  -5.7253,   7.4878],
        [ 12.0845,  -2.5772,  -4.6073],
        [  0.1677,  -4.7447,  -6.1529],
        [ 12.2218,   4.0953,  -8.2072],
        [ 11.6634,   5.8050,   9.1590],
        [ -8.6675,  -4.8569,  -9.1769],
        [  5.9655,   3.2172,  -7.5953],
        [  1.1274,  14.0116,   8.1753],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.4296,  -1.9458,   4.0833],
        [ -7.9925,  -6.1423,   2.8990],
        [  0.3038,   7.8092,   0.4164],
        [  7.0515,  11.6194,   8.4289],
        [ -1.8823, -10.6558,  -3.8952],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.0190,   4.9705,  -7.9752],
        [ 12.2218,   4.0953,  -8.2072],
        [ 10.6283,  -5.7253,   7.4878],
        [  2.7558,  -4.4844,   0.0636],
        [ -3.5520,   7.4976,   9.1522]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0843, 0.1040,  ..., 1.1282, 0.8202, 0.9002],
        [0.0843, 0.0000, 0.0321,  ..., 1.1251, 0.7678, 0.9113],
        [0.1040, 0.0321, 0.0000,  ..., 1.0973, 0.7359, 0.8869],
        ...,
        [1.1282, 1.1251, 1.0973,  ..., 0.0000, 0.6770, 0.2792],
        [0.8202, 0.7678, 0.7359,  ..., 0.6770, 0.0000, 0.6720],
        [0.9002, 0.9113, 0.8869,  ..., 0.2792, 0.6720, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.5575601 , 10.75443106, 10.78533838, 10.74409834, 10.8207297 ,
       10.67132524,  8.62631801, 10.794755  ,  8.62260327,  8.29640613,
        5.24137909,  4.89121805,  7.37576154,  6.23262289, 12.24467041,
        4.76969364, 10.36844813,  5.05963217,  6.46356806,  5.26342203,
        4.79003361, 13.08414689,  5.1805345 ,  6.28961988,  6.50409372,
       13.20597932,  6.42689705,  8.81930525,  7.58378655, 13.41713661,
        5.35774573,  7.37576154,  8.67227386, 10.07238463,  5.18277858,
       11.22142195, 11.28215429,  7.17789666,  9.37296102, 10.9036226 ,
       11.48734978,  9.97822473, 12.3489324 , 14.02526505,  8.90405823,
        8.8353674 ,  5.87305005,  5.5205189 , 11.2845608 , 14.02532758,
        8.85939626, 13.61801526, 12.18045978,  5.84533345,  5.79543038,
       10.48721382,  7.66377554,  9.99211663,  9.5642993 ,  5.21280156,
        7.60529892,  5.94293185,  6.28835101,  8.85572292,  9.04255655,
        9.0368458 ,  8.44940661,  8.83725081, 10.65929968,  8.62162372,
        4.74770899,  7.54127425,  9.44486609,  9.55593374,  9.09661212,
       11.70168755,  5.60064319, 13.02818702, 13.92390689, 10.86693708,
       12.04924159,  7.78980299,  6.1463896 ,  5.41348121,  5.91408504,
        5.04028538, 12.07225469, 12.35625694,  4.74770899, 12.889695  ,
       13.02818702,  9.93553424,  6.77777606, 14.05687815,  9.78286733,
        9.6282763 , 10.08001894,  7.81509631, 13.59170809, 10.8207297 ,
       10.36844813,  7.06372172, 12.889695  , 12.07225469,  5.26342203,
        8.29640613])}
{'P': tensor([[1.3992e-02, 6.3107e-01, 1.0330e-01,  ..., 1.6888e-06, 4.5290e-08,
         3.7318e-05],
        [2.6510e-04, 4.1700e-02, 2.4801e-02,  ..., 5.1834e-06, 1.6781e-07,
         1.5861e-04],
        [3.0668e-04, 6.8590e-03, 1.5689e-02,  ..., 2.2485e-06, 8.9025e-08,
         8.2311e-05],
        ...,
        [4.3452e-05, 2.8138e-05, 3.3304e-05,  ..., 2.7342e-02, 2.4266e-05,
         3.7514e-05],
        [1.4677e-06, 1.0170e-06, 1.4198e-06,  ..., 4.5868e-03, 1.6286e-01,
         2.9557e-05],
        [1.2562e-05, 1.2213e-05, 1.4867e-05,  ..., 5.9070e-03, 1.1353e-03,
         6.3208e-02]]), 'X': tensor([[ -4.3359,  -9.6431,  -9.0616],
        [ -3.0008, -10.2827,  -8.9480],
        [ -2.9497, -10.3028,  -7.3479],
        [ -0.8455, -10.8853,  -6.7400],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.2849, -10.9655,  -7.6542],
        [  6.2902,   0.0451,   3.7116],
        [  0.2571,  -4.7594,   1.5618],
        [  3.9085,   6.1512,  -2.2560],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.0345,   3.0175,  -1.4424],
        [ 12.1379,   4.4125,  -3.8833],
        [ -8.3717,  -5.4769,  -9.2228],
        [  0.1677,  -4.7447,  -6.1529],
        [  0.4119,  14.1019,  -4.1870],
        [ -9.4773,  -1.0024,   5.2599],
        [ -2.1533, -10.5778,   1.0112],
        [  3.1480,  -4.3165,  -7.8002],
        [ -2.7572,   7.7194,   5.9580],
        [  2.9470,  13.6085,   2.9809],
        [  4.5814, -10.4485,   2.5496],
        [ -1.1885, -10.8211,   0.0865],
        [ -9.1459,   1.4780,  -4.2403],
        [ -3.7964,  -9.9294,   7.7147],
        [ -0.1626, -10.9796,   6.4629],
        [  6.7792,  11.8084,   1.6251],
        [  9.4554,   9.4245,   9.8800],
        [  5.4167, -10.1293,   6.2156],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 10.9931,   7.2107,   7.7933],
        [ -9.2707,   0.9661,   4.4863],
        [  5.2911,  -2.4824,  -3.6877],
        [ -3.0644,   7.6439,  -1.4322],
        [  4.2041,  -3.6509,  -4.9488],
        [ -5.9872,   6.2037,  -9.4477],
        [ -8.6050,   2.9451,  -0.7012],
        [  5.8930,  12.3592,  -8.9201],
        [  2.0729,   7.2878,   7.0880],
        [  6.3497,   1.3680,  -1.9076],
        [  8.1043,  -8.5164,  -7.9290],
        [  3.9591, -10.6391,  -6.2557],
        [ -6.6998,   5.5954,   0.0823],
        [ -3.5173,   7.5091,  -8.9585],
        [  5.0233,   4.9647,  -3.1788],
        [  2.5318,  13.7231,   6.3692],
        [  5.8617,   3.4853,  -9.8080],
        [ -9.3107,  -2.7697,  -3.2919],
        [  9.4925,   9.3811,   8.9842],
        [  2.6613,   7.0021,   4.5783],
        [  1.3767, -11.0341,   9.9415],
        [  6.1984,  -0.4327,   1.5438],
        [  1.1994,   7.6035,   2.4295],
        [  2.1504,  -4.6726,   8.7908],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.4630,  -1.5145,   0.2983],
        [ -9.1795,  -3.3623,   8.5259],
        [  4.9957,  -2.8705,   6.5978],
        [ -9.2345,  -3.1345,   3.9715],
        [  8.3791,  10.5409,   5.5590],
        [  0.9809,   7.6642,  -1.4217],
        [  6.2419,  -0.2313,   4.2440],
        [  3.9770, -10.6341,   8.0783],
        [ -5.9225,  -8.5355,   2.4778],
        [  7.7750,  11.0665,   2.8364],
        [ -7.3470,  -7.0615,   5.0066],
        [  6.1730,  -0.5375,  -8.8861],
        [ -5.2752,  -9.0409,  -8.2568],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.8617,   3.4853,  -9.8080],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.6928,  12.4710,   3.6001],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.0054,  -1.0902,   8.3079],
        [  5.9193,  -1.3176,   5.1739],
        [ -4.0409,   7.3169,  -3.5779],
        [  3.7237,  -3.9966,  -5.0376],
        [ -1.9487,   7.8590,  -3.3330],
        [  1.9312,  13.8649,   8.3563],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.8323,  -1.5228,   5.1132],
        [  8.5198,  10.4091,  -8.6486],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.4149,  -4.6005,  -4.3339],
        [ -1.9646,   7.8571,   3.3817],
        [  5.5101,  -2.1433,   9.5442],
        [ -2.0229, -10.6163,   8.6930],
        [  5.3804,  -2.3504,   7.2744],
        [  1.3103,  -4.8032,  -0.9300],
        [ -8.0965,  -5.9712,   4.3466],
        [ -9.3616,  -2.4750,   3.7765],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.2921,  -1.7902,  -4.0870],
        [  9.2811,  -7.4336,   5.1771],
        [  6.3549,   0.7096,   6.9198],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.3515,   0.6418,  -6.5823],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.9655,   3.2172,  -7.5953],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -4.0409,   7.3169,  -3.5779],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.1480,  -4.3165,  -7.8002],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.5318,  13.7231,   6.3692],
        [  6.3087,   1.8053,  -1.9481]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.3869e-02, 1.1313e-01,  ..., 6.4280e-01, 1.7381e+00,
         9.8198e-01],
        [9.3869e-02, 0.0000e+00, 4.5737e-02,  ..., 5.5602e-01, 1.6636e+00,
         8.9213e-01],
        [1.1313e-01, 4.5737e-02, 0.0000e+00,  ..., 5.2979e-01, 1.6282e+00,
         8.6986e-01],
        ...,
        [6.4280e-01, 5.5602e-01, 5.2979e-01,  ..., 1.0727e-04, 1.1327e+00,
         3.4553e-01],
        [1.7381e+00, 1.6636e+00, 1.6282e+00,  ..., 1.1327e+00, 0.0000e+00,
         8.8710e-01],
        [9.8198e-01, 8.9213e-01, 8.6986e-01,  ..., 3.4553e-01, 8.8710e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.57302546, 10.71162681, 10.7167285 , 10.91805683, 10.92597196,
       11.20100795,  6.29034958,  4.7663526 ,  7.28793663,  8.91551231,
        6.74686071, 12.91505111, 10.00410625,  4.74770899, 14.10796466,
        9.53015376, 10.794755  ,  5.34250225,  8.19703512, 13.92390689,
       11.4088043 , 10.88618114,  9.26455795, 10.63038565, 10.98076873,
       13.61600536, 13.35013374, 11.48663767, 10.14436912, 13.14690137,
        9.32094055,  5.84450127,  8.23525433,  5.56809335,  8.62162372,
        9.09502385, 13.6922425 ,  7.57685908,  6.49538841, 11.75618513,
       11.35182589,  8.72896381,  8.29203356,  7.06272256, 13.95472331,
        6.81961996,  9.71392179, 13.34587027,  7.49076899, 11.11969815,
        6.2134961 ,  7.69752262,  5.14371005,  8.29203356,  9.58347456,
        9.77588801,  5.76165934,  9.75200848, 13.46553623,  7.72668467,
        6.24614905, 11.35345345, 10.38897563, 13.5247215 , 10.19036081,
        6.19633139, 10.46739073,  4.9160962 ,  6.81961996, 12.42554808,
       13.70893492, 10.48315436,  6.10360073,  6.06416957,  8.35856487,
        5.46244988,  8.09704052, 13.99877413,  6.02789634,  6.02779201,
       13.45124917, 10.53368644,  5.19577146,  8.09899874,  5.91222346,
       10.80728788,  5.87132718,  4.978704  , 10.06023665,  9.68324553,
       11.80092578,  5.18047638, 12.4217523 , 11.89105259,  6.39438044,
        8.29203356,  6.383897  , 13.00380624,  6.77777606, 10.61047596,
        8.35856487,  8.49161154,  5.34250225,  8.14433542, 13.95472331,
        6.5618968 ])}
{'P': tensor([[6.7475e-03, 2.8626e-01, 2.3438e-01,  ..., 4.0552e-02, 7.1952e-07,
         2.0225e-06],
        [6.1465e-05, 1.0351e-02, 4.3374e-01,  ..., 9.3746e-02, 1.3593e-06,
         1.9926e-06],
        [1.0125e-04, 3.2505e-04, 1.8859e-02,  ..., 2.0117e-01, 2.6877e-06,
         3.3092e-06],
        ...,
        [1.8686e-04, 4.3345e-04, 5.0821e-04,  ..., 9.5019e-02, 2.9888e-05,
         7.8540e-06],
        [1.4760e-05, 1.1190e-05, 1.0982e-05,  ..., 9.8940e-06, 8.9408e-02,
         7.3334e-05],
        [1.7379e-02, 8.7771e-03, 7.3732e-03,  ..., 1.6199e-03, 3.3501e-06,
         6.9934e-02]]), 'X': tensor([[-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0606, 0.0912,  ..., 0.2589, 1.0385, 0.2154],
        [0.0606, 0.0000, 0.0321,  ..., 0.2146, 0.9977, 0.2501],
        [0.0912, 0.0321, 0.0000,  ..., 0.1863, 0.9692, 0.2790],
        ...,
        [0.2589, 0.2146, 0.1863,  ..., 0.0000, 0.7832, 0.4646],
        [1.0385, 0.9977, 0.9692,  ..., 0.7832, 0.0000, 1.2478],
        [0.2154, 0.2501, 0.2790,  ..., 0.4646, 1.2478, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.60226614, 10.75443106, 10.78533838, 10.8207297 , 10.6910768 ,
       11.018084  , 11.22993218,  5.91677188,  7.02253235,  9.53015376,
        5.70591986,  6.24614905, 12.35421461, 12.86648771,  9.34127013,
       13.36092624,  5.23878371,  9.88571713, 13.52430176, 12.50996828,
       10.54658597, 12.27991991,  5.39098669, 13.89264822,  5.94535529,
       12.4013153 ,  5.5366698 ,  9.5014146 ,  8.09899874,  8.43626114,
        9.36378438,  8.08320235, 12.9089559 , 11.77124373,  7.61088125,
       11.51557166, 13.98691426, 13.94008948,  5.13182506,  6.39520397,
        6.00732193, 13.97430191,  5.19577146,  4.73784222, 14.10796466,
        5.38342077, 13.32645552,  8.50268322, 12.53047489, 10.92597196,
        4.78258092,  5.64500512, 12.10942412, 10.38897563,  7.88677538,
        9.17016857,  9.5014146 , 11.35182589,  6.38399789, 12.52372629,
        8.11227668, 10.91601655,  5.52416368,  7.76457095,  8.11227668,
        7.56617037,  9.24501829,  6.27715105, 12.59969936,  9.79039371,
       13.95361098,  5.05774295, 13.98691426,  9.50654967,  8.83341608,
       13.59170809,  9.43737523,  9.27474386,  9.48337063,  9.53015376,
        5.94293185, 13.77222752,  6.06416957,  5.37343173, 10.48840476,
        6.05165248, 10.14044489,  5.5748624 ,  6.21275016,  7.4544866 ,
        9.37296102,  4.91652556, 13.68672855,  9.15253344, 11.83570132,
        5.79543038, 10.9036226 , 11.73848065,  9.27474386,  9.86089579,
       12.05395056,  6.76791483, 10.38897563, 11.04629471,  7.59795445,
        9.48969978])}
{'P': tensor([[4.4068e-03, 2.4822e-01, 1.4581e-01,  ..., 2.5772e-06, 6.9338e-07,
         6.3940e-06],
        [7.7072e-05, 9.0528e-03, 4.5013e-01,  ..., 4.0422e-06, 1.5366e-06,
         8.9890e-06],
        [8.5483e-05, 1.1775e-04, 1.9009e-02,  ..., 5.6311e-06, 3.8949e-06,
         1.0841e-05],
        ...,
        [8.8045e-02, 5.9368e-02, 3.1922e-02,  ..., 6.8494e-02, 8.8149e-06,
         1.7143e-01],
        [1.9532e-05, 1.6284e-05, 1.2957e-05,  ..., 2.0133e-04, 1.2670e-01,
         1.8499e-04],
        [7.4146e-02, 5.6827e-02, 3.7585e-02,  ..., 7.4488e-04, 1.0758e-05,
         5.3077e-02]]), 'X': tensor([[ -3.9858,  -9.8334,  -3.7323],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.8823, -10.6558,  -3.8952],
        [ -2.5683, -10.4433,  -2.9182],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.2480, -11.0153,  -3.0183],
        [ -8.6112,   2.9319,  -7.9943],
        [ 12.3357,   3.6001,   0.7903],
        [  6.3236,   0.3011,  -4.6668],
        [ -8.8934,   2.2597,   3.5831],
        [  6.1685,  12.1979,  -6.2473],
        [  6.1741,  -0.5329,   3.1913],
        [ -2.0721,   7.8432,   6.6380],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.9883,   3.1536,   6.2320],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.2911,  -2.4824,  -3.6877],
        [ -9.2373,  -3.1225,  -8.9763],
        [  3.0876,   6.7527,  -5.3540],
        [ -7.9298,   4.1367,  -3.9773],
        [  2.6310, -10.9195,   3.8054],
        [ 11.7162,   5.6728,  -8.9372],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.4351,  -1.8891,   3.5642],
        [  0.3715,  -4.7756,   0.2312],
        [  6.3122,  -9.6994,  -6.3282],
        [ 10.1983,  -6.3430,  -0.3200],
        [  1.3670,  -4.7990,   3.2603],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.4430,  -0.1897,   4.8383],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.5636, -10.9293,   9.6519],
        [ -3.4986,   7.5152,   9.0712],
        [ -8.8934,   2.2597,   3.5831],
        [  3.6154,  -4.0640,  -7.9698],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.2809,  13.5061,  -4.0199],
        [  5.8836,   3.4315,  -4.0327],
        [ -9.4541,  -0.3327,   8.8246],
        [ 12.4837,   2.7510,   0.4449],
        [ -8.6112,   2.9319,  -7.9943],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.9587,  12.3216,   9.1695],
        [ -9.4541,  -0.3327,   8.8246],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -5.9938,  -8.4745,   3.9735],
        [ -6.3489,   5.9117,   6.5161],
        [  7.5309,  11.2617,   7.2656],
        [ 10.8670,  -5.3428,  -2.8637],
        [  3.9770, -10.6341,   8.0783],
        [  2.6078, -10.9229,   6.2869],
        [  2.5318,  13.7231,   6.3692],
        [  6.3460,   1.4252,   1.2291],
        [  9.3836,   9.5075,  -5.9278],
        [ -8.7735,   2.5640,  -6.3313],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -9.3616,  -2.4750,   3.7765],
        [  6.2419,  -0.2313,   4.2440],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -1.7290,   7.8825,  -3.3831],
        [ -6.7745,  -7.7264,   8.4684],
        [  3.6817,   6.3354,   6.1233],
        [  8.1043,  -8.5164,  -7.9290],
        [ -3.4488, -10.0934,   2.5195],
        [ 11.3118,   6.5957,  -4.5667],
        [  4.7739,  -3.1218,  -4.1188],
        [  2.5375,   7.0676,  -4.3490],
        [ -1.8788, -10.6567,   3.7068],
        [  2.1379,  -4.6757,   6.4981],
        [  9.9750,  -6.6338,  -1.2440],
        [  9.1409,  -7.5789,   1.5814],
        [  2.6970,   6.9826,   9.7350],
        [  3.2990,  -4.2415,  -7.4857],
        [  2.5829,  -4.5466,  -8.4015],
        [  5.9548,  -1.2273,  -2.6520],
        [ -9.4645,  -1.4880,  -2.0226],
        [  1.2728, -11.0372,   9.8121],
        [  4.9957,  -2.8705,   6.5978],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.0604,   1.7707,   1.1066],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.4235,   6.3594,   5.1663],
        [  4.3377,   5.7570,   1.3712],
        [  1.5383, -11.0274,  -4.9666],
        [ 10.9916,   7.2133,  -0.9179],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.1382, -11.0073,   1.0026],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.1793,   4.7461,   0.3899],
        [  0.9392,  -4.8143,  -3.4649],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.2437,  -4.6489,   6.1430],
        [  5.1142,  -2.7230,  -4.1966],
        [  1.4489,   7.5256,   4.4529],
        [ -9.4203,  -2.0340,   2.5856],
        [  2.5469,  13.7192,  -9.6935],
        [  2.0547,   7.2957,  -1.9343],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.8474,   6.8979,   6.6457],
        [ -8.3539,  -5.5111,   0.3373]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0430, 0.1154,  ..., 0.1313, 1.0831, 0.3465],
        [0.0430, 0.0000, 0.0733,  ..., 0.1736, 1.0653, 0.3701],
        [0.1154, 0.0733, 0.0000,  ..., 0.2423, 1.0212, 0.4021],
        ...,
        [0.1313, 0.1736, 0.2423,  ..., 0.0000, 1.1205, 0.2753],
        [1.0831, 1.0653, 1.0212,  ..., 1.1205, 0.0000, 0.9331],
        [0.3465, 0.3701, 0.4021,  ..., 0.2753, 0.9331, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.61047596, 10.6910768 , 10.8207297 , 10.75443106, 10.78533838,
       11.018084  ,  9.09661212, 12.85032836,  6.33075682,  9.17595899,
       13.66896739,  6.19707944,  8.11227668, 12.77402351,  6.76791483,
        6.1561646 ,  5.84450127,  9.75075273,  7.42512621,  8.94395628,
       11.23201619, 13.01728313,  9.78286733,  9.62238828,  4.79003361,
       11.57249984, 12.00995898,  4.98989906,  9.5642993 ,  9.44486609,
       10.74409834,  6.60752373,  9.69629   , 12.50996828, 10.00410625,
       12.59840193, 11.22596994,  8.28968067,  9.17595899,  5.43943791,
       10.6910768 , 13.89886469,  6.81117802,  9.45995955, 12.78327037,
        9.09661212, 10.92597196, 13.68672855,  9.45995955, 10.51437947,
       10.37998078,  8.67502348, 13.54774355, 12.10942412, 11.35345345,
       11.22993218, 13.95472331,  6.50409372, 13.35832725,  9.14044984,
       10.27716434,  9.68324553,  6.24614905,  9.91668182,  8.06990483,
       10.27572673,  7.32754882, 11.75618513, 10.66631784, 13.09426142,
        5.70405525,  7.5092834 , 10.82106505,  5.14124612, 11.97949273,
       11.87412439,  7.48538905,  5.37343173,  5.22903117,  6.07992623,
        9.58072286, 11.11038551,  5.76165934,  9.68324553,  9.23178204,
        8.1137452 ,  5.8828986 , 13.0743399 ,  7.20826219, 11.1341798 ,
       13.1471313 ,  5.18277858, 11.00812996,  5.57341623,  7.02496223,
        4.90505566, 12.41366754,  5.16205087,  5.79391825,  7.66377554,
        9.63742669, 13.95361098,  7.57946459,  9.75401449,  7.46248947,
       10.00794468])}
{'P': tensor([[7.9004e-03, 2.0629e-01, 1.7130e-01,  ..., 8.1859e-06, 2.0070e-05,
         1.2938e-05],
        [7.6009e-05, 1.1520e-02, 1.2164e-01,  ..., 5.6557e-06, 1.1962e-05,
         8.7497e-06],
        [1.6024e-04, 1.0814e-03, 4.9419e-02,  ..., 1.4612e-05, 3.1378e-05,
         2.4305e-05],
        ...,
        [2.1644e-02, 2.0282e-02, 8.8062e-03,  ..., 5.1698e-02, 1.9744e-03,
         2.1974e-03],
        [4.5024e-02, 2.0670e-02, 2.3217e-02,  ..., 1.1852e-04, 1.7020e-02,
         5.3816e-04],
        [4.4353e-02, 1.7779e-02, 1.6134e-02,  ..., 1.9059e-04, 1.6321e-01,
         1.3685e-02]]), 'X': tensor([[ -3.9508,  -9.8515,   1.7923],
        [ -2.1533, -10.5778,   1.0112],
        [ -1.0012, -10.8576,   2.2867],
        [ -2.6485, -10.4151,   0.8766],
        [ -1.3178, -10.7939,   0.4236],
        [ -1.6275, -10.7221,   3.2376],
        [  0.1382, -11.0073,   1.0026],
        [  5.8323,  -1.5228,   5.1132],
        [  2.4572,   7.1085,   7.7376],
        [ 12.1940,  -2.1881,   7.1499],
        [  1.1644, -11.0394,   7.8511],
        [  4.5325,  -3.3643,   6.7699],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.8058,   2.4850,  -1.7425],
        [ -6.3489,   5.9117,   6.5161],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.8745,   2.3097,   3.5600],
        [  0.6952,  -4.8060,   8.8976],
        [  5.9003,  12.3550,   2.1917],
        [ 10.1798,   8.5002,  -5.8893],
        [  6.3460,   1.4252,   1.2291],
        [  0.9074,  -4.8139,   8.1674],
        [  3.2159, -10.8166,  -1.0406],
        [  2.5663,   7.0526,  -6.8769],
        [ -7.6650,   4.5092,   0.2520],
        [ -8.7544,   2.6096,   3.3146],
        [ -3.5520,   7.4976,   9.1522],
        [ -8.3062,  -5.6008,   0.0287],
        [  2.3261,   7.1726,   6.2522],
        [  4.3523, -10.5232,   8.2138],
        [  0.9036,   7.6840,   9.9237],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.5954,  13.0099,  -4.4439],
        [ -8.0750,  -6.0072,  -5.8267],
        [  5.6095,  -1.9702,  -7.8233],
        [  1.7232,  -4.7574,   3.1270],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.2608,  12.6977,   4.9166],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -9.4196,   0.0478,   9.4140],
        [ -4.2271,   7.2387,  -1.8316],
        [  6.7565,  11.8237,  -4.9882],
        [  6.3087,   1.8053,  -1.9481],
        [  6.3301,   0.3638,  -7.5685],
        [ -6.1914,  -8.2997,   8.5380],
        [ -1.5875,   7.8944,  -9.4286],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -9.3087,   0.7792,  -7.5587],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3497,   1.3680,  -1.9076],
        [  6.3460,   1.4252,   1.2291],
        [  5.0277,   4.9589,   7.0175],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.2373,  -4.6506,  -1.7334],
        [  7.3724,  -9.0577,  -8.9843],
        [  6.3589,   0.8239,  -8.4230],
        [ -8.0608,   3.9362,  -5.9567],
        [  7.3724,  -9.0577,  -8.9843],
        [  4.5954,  13.0099,  -4.4439],
        [  3.7237,  -3.9966,  -5.0376],
        [ 11.7726,  -3.4876,  -6.7402],
        [  2.0937,  13.8293,  -2.3533],
        [  2.1771, -10.9776,  -3.2007],
        [ -3.3999, -10.1152,  -1.2088],
        [ -8.9282,   2.1648,   8.4894],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -9.4766,  -1.1116,  -2.1706],
        [  3.3744,   6.5623,  -6.3636],
        [  6.9693,  -9.3200,  -8.3822],
        [ -2.2188,   7.8218,   8.6064],
        [  5.9526,   3.2525,  -4.3821],
        [  5.1861,   4.7361,  -3.7312],
        [  6.3606,   0.9204,   3.2525],
        [  7.9403,  -8.6455,  -8.7433],
        [ 12.6000,   1.3839,  -3.2992],
        [  1.7228,  -4.7575,   8.3396],
        [  8.5198,  10.4091,  -8.6486],
        [  6.2365,  -0.2578,  -6.0212],
        [ 11.2209,  -4.7092,   2.6117],
        [ -3.9700,   7.3453,  -4.1486],
        [ -6.9838,  -7.4965,   0.9796],
        [ 12.5123,  -0.5339,  -1.0728],
        [  7.3592,  11.3937,   6.2407],
        [  7.8982,  -8.6778,   0.0239],
        [ 11.3691,   6.4760,   1.0787],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -3.9858,  -9.8334,  -3.7323],
        [  4.3377,   5.7570,   1.3712],
        [  2.0243,   7.3086,   7.5432],
        [  1.3705,  13.9723,   7.9266],
        [ -1.8788, -10.6567,   3.7068],
        [ 11.7795,  -3.4697,   3.5624],
        [ -4.8751,  -9.3148,   4.0406],
        [  0.9967,  -4.8144,  -9.7370],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.0362,  13.2397,  -7.0030],
        [ -5.5525,   6.5146,  -4.4984],
        [ 12.1751,  -2.2592,  -5.7751],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.9636,   2.0648,   8.1946],
        [ -9.4351,  -1.8891,   3.5642],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -7.6525,  -6.6534,   5.1721],
        [ -8.0965,  -5.9712,   4.3466]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0907, 0.1350,  ..., 0.5197, 0.2635, 0.2018],
        [0.0907, 0.0000, 0.1527,  ..., 0.4707, 0.3500, 0.2897],
        [0.1350, 0.1527, 0.0000,  ..., 0.6234, 0.2679, 0.2225],
        ...,
        [0.5197, 0.4707, 0.6234,  ..., 0.0000, 0.7548, 0.6937],
        [0.2635, 0.3500, 0.2679,  ..., 0.7548, 0.0000, 0.0627],
        [0.2018, 0.2897, 0.2225,  ..., 0.6937, 0.0627, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.61417181, 10.794755  , 10.9036226 , 10.74655346, 10.87409005,
       10.84493732, 11.00812996,  6.02779201,  7.52117132, 12.38881635,
       11.10066126,  5.64464807, 11.81423161,  9.14973319,  8.67502348,
        5.13125513,  9.17016857,  4.85603886, 13.69162795, 13.26209545,
        6.50409372,  4.89869205, 11.2845608 ,  7.50500064,  8.89301516,
        9.13507321,  8.29640613, 10.01804223,  7.5403822 , 11.38774149,
        7.73692436, 10.37998078, 13.79762415, 10.06437598,  5.94541803,
        5.05989547, 13.08414689, 13.06498554, 13.744387  , 12.52372629,
        9.41970031,  8.38252937, 13.61801526,  6.5618968 ,  6.34059021,
       10.35464916,  8.05242591, 10.4977132 ,  9.34127013, 13.36092624,
        6.49538841,  6.50409372,  7.06169391,  6.1561646 ,  5.16077764,
       11.67875855,  6.41202986,  8.97052115, 11.67875855, 13.79762415,
        5.46244988, 12.27835996, 13.98691426, 11.1913598 , 10.67132524,
        9.18690018,  9.99145096,  9.54154096,  7.37902766, 11.63765273,
        8.1303945 ,  6.78325151,  7.02326515,  6.42689705, 11.73848065,
       12.67576792,  5.05981914, 13.45124917,  6.24187044, 12.16901615,
        8.34949383, 10.24557047, 12.52372629, 13.56367857, 11.73397309,
       13.08414689, 10.27716434, 10.61047596,  7.20826219,  7.58378655,
       14.0393949 , 10.82106505, 12.27991991, 10.51340829,  4.91652556,
       10.37998078, 13.84126197,  8.55982105, 12.3828957 ,  5.5513072 ,
        9.19837792,  9.62238828, 11.97238624,  9.94285511, 10.14044489,
       10.06023665])}
{'P': tensor([[5.3720e-03, 2.5195e-01, 1.9728e-01,  ..., 6.7531e-07, 5.4946e-04,
         8.8192e-07],
        [3.2253e-05, 4.9920e-03, 3.1371e-01,  ..., 6.0051e-07, 6.3367e-04,
         8.5101e-07],
        [5.3849e-05, 1.0328e-04, 1.0209e-02,  ..., 1.2145e-06, 1.4189e-03,
         1.7765e-06],
        ...,
        [2.0472e-04, 1.2472e-04, 1.0385e-04,  ..., 5.3425e-02, 6.7519e-05,
         9.0641e-04],
        [1.3564e-04, 1.7950e-04, 2.0266e-04,  ..., 1.3915e-04, 7.7666e-02,
         2.9838e-04],
        [1.1118e-04, 7.1629e-05, 6.1144e-05,  ..., 4.0762e-01, 8.3706e-05,
         6.0908e-02]]), 'X': tensor([[ -3.9456,  -9.8542,   0.7466],
        [ -2.6485, -10.4151,   0.8766],
        [ -2.1533, -10.5778,   1.0112],
        [ -1.3178, -10.7939,   0.4236],
        [ -0.9715, -10.8630,   0.3555],
        [ -1.1885, -10.8211,   0.0865],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.1006,   7.6318,  -2.4877],
        [  7.0515,  11.6194,   8.4289],
        [  2.4639, -10.9431,  -7.3718],
        [ 11.9019,  -3.1379,  -3.4534],
        [  0.6933,   7.7336,   2.1179],
        [  2.3261,  13.7749,   5.9121],
        [  6.3606,   0.9204,   3.2525],
        [  6.2036,   2.4030,  -1.2500],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.6314,  -1.9302,  -9.6543],
        [  1.8447,  -4.7372,   8.1235],
        [  9.9217,  -6.7006,  -5.6728],
        [ 10.9916,   7.2133,  -0.9179],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.6675,  -4.8569,  -9.1769],
        [  8.1043,  -8.5164,  -7.9290],
        [  3.8111, -10.6787,  -4.6539],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.0899,  -4.6870,  -7.4260],
        [  2.5375,   7.0676,  -4.3490],
        [ -0.2586,   7.8825,  -9.9840],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.8450,  -9.9356,   3.0767],
        [  6.2902,   0.0451,   3.7116],
        [ -8.0608,   3.9362,  -5.9567],
        [ -3.5520,   7.4976,   9.1522],
        [ -6.0844,  -8.3955,   0.7705],
        [  6.2576,   2.1359,  -3.3398],
        [  5.1142,  -2.7230,  -4.1966],
        [ -1.8788, -10.6567,   3.7068],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.3087,   1.8053,  -1.9481],
        [ -1.1885, -10.8211,   0.0865],
        [  2.2659,  13.7894,   6.5436],
        [ -9.4117,   0.1160,   8.2574],
        [  3.4920,  -4.1366,  -8.3955],
        [ 11.9641,  -2.9568,   7.7211],
        [ -2.0721,   7.8432,   6.6380],
        [  2.3261,   7.1726,   6.2522],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.8154,   2.4611,   2.5198],
        [  0.4119,  14.1019,  -4.1870],
        [  1.1644, -11.0394,   7.8511],
        [  3.2701,   6.6338,   9.9209],
        [ -9.1813,  -3.3550,   1.8827],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.1196,  -2.7160,   4.2705],
        [  6.3119,   1.7797,   9.9186],
        [  5.1196,  -2.7160,   4.2705],
        [ 12.4837,   2.7510,   0.4449],
        [ -1.5261,   7.8988,   9.3768],
        [  6.0031,  12.2958,  -6.5992],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.6019,   5.9540,  -5.1715],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.9693,  -9.3200,  -8.3822],
        [ 11.3367,  -4.4806,   0.4643],
        [ -9.4630,  -1.5145,   0.2983],
        [ 10.0400,   8.6922,  -6.2847],
        [ -5.6423,   6.4537,  -1.2091],
        [  5.0348,  -2.8229,   3.5436],
        [  5.8836,   3.4315,  -4.0327],
        [ -8.9636,   2.0648,   8.1946],
        [ -7.0948,   5.1940,   9.8171],
        [ -9.4739,  -1.2497,   0.5483],
        [  5.0190,   4.9705,  -7.9752],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.8836,   3.4315,  -4.0327],
        [  9.1620,  -7.5573,  -9.9882],
        [ -6.6895,   5.6051,   2.7899],
        [  6.3583,   1.1729,   8.7835],
        [ -9.4541,  -0.3327,   8.8246],
        [ -2.4956, -10.4682,   3.5632],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.9282,   2.1648,   8.4894],
        [ 11.7418,   5.6070,  -1.4192],
        [ -0.5385,   7.9037,   6.3830],
        [ -0.6323,   7.9085,  -5.9544],
        [  4.8674,  12.8875,   6.7622],
        [  6.0310,  12.2795,  -6.4528],
        [  5.2939,  -2.4784,  -4.5840],
        [ 11.5198,  -4.0923,   5.5088],
        [  9.3753,   9.5171,   3.9131],
        [  4.5430,   5.5438,  -3.8669],
        [ -5.3179,  -9.0101,   9.4472],
        [ -3.4986,   7.5152,   9.0712],
        [  5.4167, -10.1293,   6.2156],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.0190,   4.9705,  -7.9752],
        [  2.4932,   7.0903,  -5.4214],
        [  5.4426, -10.1181,  -5.1268],
        [  6.2975,   0.0947,   5.1791],
        [  5.6248,   3.9984,   8.4933],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.0506,   4.9277,  -9.5094],
        [ -4.2271,   7.2387,  -1.8316],
        [  3.3474,  -4.2162,   5.2455],
        [ -2.0445,   7.8469,  -0.5857]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0334, 0.0644,  ..., 0.5056, 0.8961, 0.5642],
        [0.0334, 0.0000, 0.0311,  ..., 0.4794, 0.8629, 0.5344],
        [0.0644, 0.0311, 0.0000,  ..., 0.4566, 0.8319, 0.5076],
        ...,
        [0.5056, 0.4794, 0.4566,  ..., 0.0000, 0.6227, 0.1181],
        [0.8961, 0.8629, 0.8319,  ..., 0.6227, 0.0000, 0.5065],
        [0.5642, 0.5344, 0.5076,  ..., 0.1181, 0.5065, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.61472636, 10.74655346, 10.794755  , 10.87409005, 10.90638066,
       10.88618114, 10.1979669 ,  7.71075226, 13.59170809, 11.21703654,
       12.30858677,  7.76457095, 13.96987806,  6.42689705,  6.652742  ,
        9.78285424,  5.95297459,  5.08372149, 11.97238624, 13.1471313 ,
       11.95743558,  9.93553424, 11.75618513, 11.33837396, 12.35625694,
        5.13182506,  7.5092834 ,  7.88677538, 12.49317563, 11.52733581,
        6.29034958,  8.97052115,  8.29640613, 10.36844813,  6.61211224,
        5.79391825, 10.82106505, 10.63038565,  6.5618968 , 10.88618114,
       13.97430191,  9.41244865,  5.41348121, 12.32408516,  8.11227668,
        7.5403822 ,  6.1561646 ,  9.15253344, 14.10796466, 11.10066126,
        7.39599986,  9.7751189 ,  5.23878371,  5.79543038,  6.55800267,
        5.79543038, 12.78327037,  8.04483495, 13.6829843 ,  7.83486943,
       13.04051744,  5.19065439, 11.63765273, 12.18998195,  9.58347456,
       13.27994304,  8.5724042 ,  5.77219077,  6.81117802,  9.19837792,
        8.79284466,  9.55593374,  7.06372172, 11.59512194,  5.84450127,
        6.81117802, 11.87665501,  8.7273569 ,  6.46560181,  9.45995955,
       10.76154256, 12.42554808,  9.18690018, 13.01187484,  7.92200342,
        7.9337647 , 13.77604293, 13.68063098,  5.84533345, 12.22503754,
       13.35927527,  7.16747878, 10.46238054,  8.28968067, 11.48663767,
        8.89458273,  7.06372172,  7.5158489 , 11.4890861 ,  6.29821673,
        6.90115901,  9.73579738,  7.05627461,  8.38252937,  5.38342077,
        8.10886021])}
{'P': tensor([[4.6210e-03, 2.8376e-01, 1.6825e-01,  ..., 6.7891e-06, 6.8965e-05,
         2.8149e-06],
        [1.1798e-04, 1.1490e-02, 1.4692e-03,  ..., 1.9158e-05, 2.1939e-04,
         7.5741e-06],
        [1.7938e-04, 6.9377e-02, 1.0245e-02,  ..., 1.7873e-05, 1.9679e-04,
         6.6431e-06],
        ...,
        [1.0650e-05, 1.1405e-05, 1.1793e-05,  ..., 1.2347e-01, 1.6259e-03,
         2.4530e-04],
        [1.2381e-06, 1.3498e-06, 1.3539e-06,  ..., 1.1703e-03, 4.2986e-02,
         5.7924e-04],
        [3.3758e-05, 3.0412e-05, 3.0715e-05,  ..., 1.8969e-05, 9.2492e-06,
         7.9548e-02]]), 'X': tensor([[ -3.7964,  -9.9294,   7.7147],
        [ -3.3116, -10.1539,   7.5919],
        [ -3.3812, -10.1235,   7.4384],
        [ -1.6334, -10.7207,   7.1397],
        [ -2.0229, -10.6163,   8.6930],
        [ -1.3940, -10.7772,   6.7835],
        [  6.1984,  -0.4327,   1.5438],
        [  7.9141,  -8.6656,   9.6948],
        [  2.6613,   7.0021,   4.5783],
        [  5.8889,  -1.3916,   8.8993],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.2905,   1.9366,   2.1169],
        [ -4.7389,   6.9953,   3.7638],
        [  0.9809,  -4.8145,   6.8706],
        [  1.4150,  13.9647,  -4.0915],
        [  1.8719,   7.3713,  -3.7465],
        [  1.9241,   7.3503,   7.9004],
        [  5.1142,  -2.7230,  -4.1966],
        [ -8.9942,  -3.9990,   6.0448],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -2.1817,   7.8274,  -4.6298],
        [  0.8572,  14.0501,  -8.3364],
        [ -3.1204, -10.2345,   9.2288],
        [  3.7121,   6.3117,   0.6738],
        [  2.2373,  -4.6506,  -1.7334],
        [ -6.7992,   5.4991,   9.9186],
        [ -9.3464,   0.5708,  -1.6397],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.3205,   1.7068,  -3.8619],
        [  5.3858,   4.4260,  -9.6769],
        [  5.9003,  12.3550,   2.1917],
        [ -0.5385,   7.9037,   6.3830],
        [  5.3860,  -2.3418,   1.0826],
        [  3.9591, -10.6391,  -6.2557],
        [  5.1203,  -2.7151,   6.2110],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.0547,   7.2957,  -1.9343],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.4837,   2.7510,   0.4449],
        [  5.8325,  -1.5222,  -2.1341],
        [ 11.1085,  -4.9204,   0.0736],
        [ -6.9838,  -7.4965,   0.9796],
        [ 10.4737,  -5.9570,  -1.2108],
        [  2.7121,   6.9743,  -8.4202],
        [  5.4426, -10.1181,  -5.1268],
        [  6.1066,  -0.7799,  -4.1746],
        [ -1.6275, -10.7221,   3.2376],
        [ -9.4645,  -1.4880,  -2.0226],
        [  3.0719, -10.8449,  -4.1673],
        [ -2.5056,   7.7720,   5.5878],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.7796,  -3.4696,   9.0088],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 12.3025,  -1.7440,  -6.6632],
        [  4.4039,   5.6903,   8.7366],
        [  9.6816,   9.1536,  -5.8543],
        [ 11.1085,  -4.9204,   0.0736],
        [ -9.3739,   0.3982,  -4.2564],
        [  5.5294,  -2.1106,   9.0626],
        [  9.6662,   9.1725,   8.1056],
        [  8.5093,  -8.1764,  -3.2381],
        [  8.0381,  -8.5690,  -8.9523],
        [ -4.4520,   7.1370,  -8.7867],
        [  0.1206,  -4.7363,   8.7103],
        [  5.1667,  -2.6543,   1.2879],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -1.0012, -10.8576,   2.2867],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.3589,  13.7668,  -3.6501],
        [  4.5340,  -3.3629,   8.9652],
        [ -8.8363,   2.4086,  -5.6296],
        [  4.9531,   5.0576,   5.6794],
        [  0.1041, -11.0045,   2.4311],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.1706,   1.3863,   8.4032],
        [ -7.4887,   4.7364,   8.9074],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.4690,  -0.8528,  -9.3374],
        [  5.9526,   3.2525,  -4.3821],
        [  5.6672,  -1.8631,   8.1020],
        [  4.2612, -10.5514,  -6.6268],
        [  3.5284, -10.7484,   5.3347],
        [ 11.4703,  -4.2006,   7.4086],
        [  3.5965,  -4.0754,  -3.5035],
        [  3.1064,  13.5607,   6.7605],
        [ -2.6069, -10.4298,  -6.1995],
        [ -9.4541,  -0.3327,   8.8246],
        [  5.1890,   4.7318,   0.9515],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.2974,   0.8371,   2.4852],
        [  7.5309,  11.2617,   7.2656],
        [  3.9895,   6.0816,  -6.5235],
        [ -9.4541,  -0.3327,   8.8246],
        [ -5.6792,  -8.7352,   4.0318],
        [  5.6928,  12.4710,   3.6001],
        [  6.1730,  -0.5375,  -8.8861],
        [  2.1771, -10.9776,  -3.2007],
        [  5.7802,  -1.6363,  -6.2782],
        [ 11.2767,  -4.6006,   0.7868],
        [  3.3069,   6.6089,   8.7377]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0256, 0.0258,  ..., 1.2731, 0.9488, 0.7244],
        [0.0256, 0.0000, 0.0080,  ..., 1.2551, 0.9271, 0.6988],
        [0.0258, 0.0080, 0.0000,  ..., 1.2501, 0.9238, 0.6998],
        ...,
        [1.2731, 1.2551, 1.2501,  ..., 0.0000, 0.4083, 0.9087],
        [0.9488, 0.9271, 0.9238,  ..., 0.4083, 0.0000, 0.5011],
        [0.7244, 0.6988, 0.6998,  ..., 0.9087, 0.5011, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.63038565, 10.68031351, 10.67322527, 10.84437705, 10.80728788,
       10.86693708,  6.2134961 , 11.73567759,  7.49076899,  6.05113217,
       10.31886066,  6.58184474,  8.44940661,  4.91336994, 14.03618649,
        7.60529892,  7.59795445,  5.79391825,  9.84314712,  7.26074925,
        9.99145096,  8.12580976, 14.07623266, 10.69963954,  7.32233733,
        5.16077764,  8.74470218,  9.36378438, 12.27993098,  5.34250225,
        6.54693987,  6.97106912, 13.69162795,  7.92200342,  5.87305005,
       11.35182589,  5.79562691, 11.86239961,  7.57946459, 12.52875793,
       12.78327037,  6.02789634, 12.1494023 , 10.24557047, 12.04924159,
        7.48311002, 11.4890861 ,  6.1561646 , 10.84493732,  9.58072286,
       11.27160035,  8.16584537, 13.66896739,  9.41244865, 12.27993098,
       12.27835996, 12.42554808,  7.19534146, 13.32374056, 12.1494023 ,
        9.38232121,  5.91854397, 13.32557133, 11.80092578, 11.74901778,
        8.41171369,  4.73784222,  5.80860165, 12.42554808, 10.9036226 ,
        5.19577146, 11.63502679, 13.96746816,  5.64500512,  9.15866607,
        7.07902036, 11.0050379 , 10.67322527,  9.27474386,  8.86082124,
        9.5014146 ,  9.66977802, 12.4980845 ,  6.78325151,  5.96556248,
       11.37939695, 11.31276109, 12.21532282,  5.43544626, 13.91198191,
       10.7506423 ,  9.45995955,  7.02253235,  7.83486943, 12.62576026,
        9.3349858 , 13.54774355,  7.27340834,  9.45995955, 10.41909211,
       13.70893492,  6.19633139, 11.1913598 ,  6.00732193, 12.17901033,
        7.39004226])}
{'P': tensor([[1.0431e-02, 1.8943e-01, 1.4387e-01,  ..., 8.0898e-04, 4.2035e-04,
         1.7562e-04],
        [1.1545e-04, 1.1988e-02, 1.2665e-01,  ..., 1.2403e-03, 5.8046e-04,
         3.8093e-04],
        [6.9178e-05, 2.4231e-04, 3.3139e-02,  ..., 6.8580e-03, 3.9415e-03,
         4.0997e-03],
        ...,
        [9.7415e-05, 1.1560e-04, 1.8041e-04,  ..., 1.1284e-01, 9.4008e-02,
         1.6716e-02],
        [6.1302e-05, 6.6435e-05, 1.0613e-04,  ..., 1.6714e-02, 1.0227e-01,
         1.1203e-02],
        [4.7050e-07, 5.4841e-07, 8.7445e-07,  ..., 3.0141e-04, 6.2461e-04,
         1.6301e-02]]), 'X': tensor([[-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0904, 0.1939,  ..., 0.8287, 0.8526, 0.8540],
        [0.0904, 0.0000, 0.2132,  ..., 0.7924, 0.8249, 0.8431],
        [0.1939, 0.2132, 0.0000,  ..., 0.6600, 0.6742, 0.6635],
        ...,
        [0.8287, 0.7924, 0.6600,  ..., 0.0000, 0.0875, 0.2438],
        [0.8526, 0.8249, 0.6742,  ..., 0.0875, 0.0000, 0.1611],
        [0.8540, 0.8431, 0.6635,  ..., 0.2438, 0.1611, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.65929968, 10.82106505, 11.04313641, 10.84493732, 10.98076873,
       10.99326577,  8.24808832, 10.51437947,  6.41202986,  8.34949383,
        5.13616224,  6.25586297, 12.24467041, 13.99048094,  9.6282763 ,
       13.99048094, 11.23201619, 12.3828957 ,  5.19986797, 12.41012046,
        8.53501777,  8.09704052,  9.50913849,  5.43544626, 11.07637531,
        5.95111804,  8.29640613, 11.018084  ,  4.78002665, 11.87412439,
        6.36004393,  9.50913849,  7.13501264,  8.87007495,  9.17595899,
        5.52416368, 12.05300943,  8.90405823, 13.09426142, 10.33958639,
        5.1805345 ,  6.53290663, 11.05672136,  8.50268322, 11.8159254 ,
        6.48486171,  5.96556248,  6.05165248, 11.63724792, 10.4977132 ,
       10.91601655,  9.50913849,  4.78544875, 12.38881635,  4.85603886,
        8.40445872,  7.34250948, 11.48734978,  9.9353921 , 10.01804223,
        6.42439344,  9.50654967, 11.76830981, 13.27717982,  7.62011734,
       12.40074901,  5.51177501, 13.52430176,  7.67939546, 12.26997165,
        6.5618968 ,  8.19703512,  8.49161154,  6.76661595,  8.25828409,
       13.2778378 ,  8.90405823,  7.62232307, 12.4980845 , 13.66051884,
        5.5366698 , 10.91601655,  8.67227386,  6.83083863,  6.80370254,
       10.31886066, 13.02818702, 14.10796466,  7.16747878,  8.53014915,
        8.31625171,  5.79562691, 10.68031351, 12.5769659 , 11.69264243,
       10.51437947,  4.7663526 ,  6.33517349,  6.67108662, 11.4890861 ,
        5.47866221, 10.1371552 , 12.72524615,  5.79562691,  5.92952538,
       12.32408516])}
{'P': tensor([[6.4305e-03, 1.8963e-01, 1.8139e-01,  ..., 1.2282e-06, 8.9939e-07,
         1.1735e-06],
        [3.2492e-05, 8.3606e-03, 4.0521e-04,  ..., 9.3238e-07, 7.8598e-07,
         1.8650e-06],
        [4.6981e-05, 5.8782e-02, 7.4760e-03,  ..., 8.8716e-07, 7.2722e-07,
         1.7044e-06],
        ...,
        [2.3065e-03, 8.6218e-04, 9.2951e-04,  ..., 2.2339e-02, 2.1320e-04,
         5.8118e-06],
        [1.3141e-03, 4.8216e-04, 5.4695e-04,  ..., 4.1392e-01, 3.0817e-02,
         1.5935e-05],
        [2.5760e-05, 1.3249e-05, 1.6947e-05,  ..., 6.5126e-04, 1.6557e-03,
         3.5862e-02]]), 'X': tensor([[ -3.4488, -10.0934,   2.5195],
        [ -1.0012, -10.8576,   2.2867],
        [ -1.6275, -10.7221,   3.2376],
        [  0.1041, -11.0045,   2.4311],
        [ -0.0254, -10.9932,   3.2200],
        [ -1.8788, -10.6567,   3.7068],
        [  1.5662,  13.9375,   4.5142],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -7.5720,   4.6310,   2.1717],
        [ 11.7796,  -3.4696,   9.0088],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 11.5456,   6.0861,   8.1101],
        [  2.3490,  -4.6199,  -4.2489],
        [  4.9337,  12.8566,  -3.6582],
        [ -3.6381,   7.4683,   7.3544],
        [  6.3589,   0.8239,  -8.4230],
        [  5.1203,  -2.7151,   6.2110],
        [ -7.6525,  -6.6534,   5.1721],
        [ 11.0891,  -4.9558,  -2.3178],
        [  6.7792,  11.8084,   1.6251],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.4624,  -0.4633,  -7.3918],
        [  2.9410,   6.8428,   5.4458],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.0345,   3.0175,  -1.4424],
        [ -4.3963,   7.1629,   4.5263],
        [ 11.2571,   6.7072,   8.3693],
        [ -2.3317,   7.8034,  -3.6274],
        [ -2.0721,   7.8432,   6.6380],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.8631,  -4.3728,  -1.1746],
        [  7.5064,  -8.9651,  -0.0994],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -9.4157,   0.0821,  -5.0025],
        [  0.5850,  -4.7982,   9.6639],
        [ -0.9715, -10.8630,   0.3555],
        [  9.7328,   9.0902,  -0.8917],
        [  4.0362,  13.2397,  -7.0030],
        [  2.1504,  -4.6726,   8.7908],
        [  0.2571,  -4.7594,   1.5618],
        [  4.9152,  12.8653,  -1.4459],
        [  6.0305,   3.0296,   0.1375],
        [ -6.2668,   5.9808,   2.5614],
        [  5.8889,  -1.3916,   8.8993],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 11.4235,   6.3594,   5.1663],
        [  0.5960, -11.0335,   3.1144],
        [  0.7275,  14.0667,  -0.9815],
        [  0.3493,  -4.7727,  -6.5682],
        [  7.0515,  11.6194,   8.4289],
        [ 11.8951,  -3.1570,   7.0742],
        [  4.0421,  -3.7765,  -8.3631],
        [  1.7815,  -4.7481,   4.4289],
        [  7.8425,  -8.7202,   6.4530],
        [  5.3804,  -2.3504,   7.2744],
        [  1.7903,  -4.7467,   0.1168],
        [  2.5375,   7.0676,  -4.3490],
        [  0.1382, -11.0073,   1.0026],
        [  6.3549,   0.7096,   6.9198],
        [  9.1820,   9.7334,  -2.7932],
        [  4.4049,   5.6892,  -4.4394],
        [  4.1537,   5.9340,  -6.0423],
        [ 11.5935,   5.9740,  -4.3952],
        [ 10.9394,   7.3069,  -8.9365],
        [  3.3069,   6.6089,   8.7377],
        [  1.3767, -11.0341,   9.9415],
        [ -1.9646,   7.8571,   3.3817],
        [  0.7275,  14.0667,  -0.9815],
        [  0.1460,   7.8340,  -9.4614],
        [ 12.5650,   2.0132,   4.6272],
        [ -0.9715, -10.8630,   0.3555],
        [ -2.2188,   7.8218,   8.6064],
        [  3.7237,  -3.9966,  -5.0376],
        [  6.2790,  -0.0259,  -5.4116],
        [ 11.0891,  -4.9558,  -2.3178],
        [  0.5175,  14.0910,   4.6540],
        [  6.3534,   1.2990,  -7.2942],
        [  5.4179,  -2.2925,  -3.4641],
        [  1.7228,  -4.7575,   8.3396],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.8363,   2.4086,  -5.6296],
        [  1.9505,  -4.7171,   9.9868],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.6544,   5.8273,  -3.1375],
        [  6.1741,  -0.5329,   3.1913],
        [  6.3266,   0.3292,   9.3502],
        [ 12.4901,  -0.7052,   1.7315],
        [ 10.9394,   7.3069,  -8.9365],
        [ -2.0840,   7.8415,  -7.8657],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.5832,   1.7543,  -7.6134],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.5224,  -0.4497,   4.8448],
        [  7.8425,  -8.7202,   6.4530],
        [  9.0652,   9.8597,  -3.5949],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.4697,  -4.5836,   1.4593],
        [  3.2990,  -4.2415,  -7.4857],
        [ -9.3464,   0.5708,  -1.6397],
        [ -8.6801,   2.7809,  -0.8443],
        [ -0.3432,   7.8900,   7.1805]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0979, 0.0989,  ..., 0.3640, 0.3161, 0.6724],
        [0.0979, 0.0000, 0.0841,  ..., 0.3829, 0.3091, 0.5880],
        [0.0989, 0.0841, 0.0000,  ..., 0.4480, 0.3845, 0.5821],
        ...,
        [0.3640, 0.3829, 0.4480,  ..., 0.0000, 0.1095, 0.9183],
        [0.3161, 0.3091, 0.3845,  ..., 0.1095, 0.0000, 0.8146],
        [0.6724, 0.5880, 0.5821,  ..., 0.9183, 0.8146, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.66631784, 10.9036226 , 10.84493732, 11.0050379 , 10.99326577,
       10.82106505, 14.02526505, 12.3828957 ,  8.87587457, 12.27993098,
       10.60226614,  9.69792513,  9.20340031,  9.5642993 , 13.05148826,
        5.18277858, 13.77075174,  8.30729352,  6.41202986,  5.79562691,
       10.14044489, 12.1460931 , 13.61600536, 13.2430377 ,  9.47369628,
        7.44805901,  8.83725081,  6.74686071,  8.40445872, 13.103709  ,
        8.14433542,  8.11227668,  6.652742  ,  9.88310614, 11.69264243,
       12.49317563,  9.41605684,  4.83371341, 10.90638066, 13.31764181,
       13.84126197,  5.14371005,  4.7663526 , 13.77222752,  6.74872981,
        8.66273171,  6.05113217, 12.27835996, 13.0743399 , 11.04953767,
       14.08549802,  4.78544875, 13.59170809, 12.30695098,  5.53175275,
        5.0713364 , 11.72802491,  5.87132718,  5.07306274,  7.5092834 ,
       11.00812996,  6.39438044, 13.38091121,  7.19513806,  7.24326992,
       13.04217487, 13.15525803,  7.39004226, 11.11969815,  8.09899874,
       14.08549802,  7.83534987, 12.72524615, 10.90638066,  8.1303945 ,
        5.46244988,  6.27905292, 12.1460931 , 14.10045797,  6.48486171,
        5.8828986 ,  5.05981914, 14.09176259,  9.15866607,  5.10446653,
       12.10942412, 13.03002338,  6.19707944,  6.33517349, 12.50996828,
       13.15525803,  8.1137452 ,  7.822854  ,  9.97822473, 12.70489563,
       12.52875793, 12.53047489, 11.72802491, 13.39372426,  9.58072286,
       10.0674887 ,  5.20659892,  5.37343173,  9.36378438,  9.11473391,
        7.89745107])}
{'P': tensor([[7.2258e-03, 2.8095e-01, 1.1901e-01,  ..., 7.0098e-05, 5.0729e-06,
         5.4234e-07],
        [1.1332e-04, 9.1682e-03, 7.4818e-02,  ..., 1.3289e-04, 8.6038e-06,
         8.3526e-07],
        [6.7815e-05, 1.8837e-04, 1.0455e-02,  ..., 1.9664e-04, 1.7348e-05,
         1.5359e-06],
        ...,
        [2.1811e-07, 2.4432e-07, 2.6016e-07,  ..., 9.5570e-03, 8.6236e-03,
         4.9134e-04],
        [7.0044e-07, 6.6457e-07, 6.1538e-07,  ..., 1.1109e-05, 4.6565e-02,
         3.7938e-01],
        [4.6289e-07, 3.8998e-07, 3.0484e-07,  ..., 3.9286e-07, 5.3183e-06,
         2.1986e-02]]), 'X': tensor([[ -3.3999, -10.1152,  -1.2088],
        [ -2.6735, -10.4062,  -1.5911],
        [ -1.1885, -10.8211,   0.0865],
        [  0.5600, -11.0321,  -0.8660],
        [  0.6470, -11.0352,  -1.1931],
        [  0.6757, -11.0361,  -1.3047],
        [  2.5128, -10.9365,   9.9645],
        [ -7.4808,   4.7462,   2.0266],
        [ 11.7794,   5.5086,  -9.6929],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.2979,   9.6048,   7.3138],
        [  5.6202,  -1.9508,   0.7471],
        [  4.1537,   5.9340,  -6.0423],
        [ 12.5770,   1.8512,  -0.6578],
        [  4.9957,  -2.8705,   6.5978],
        [  5.6260,  -1.9401,   3.0259],
        [ -6.9838,  -7.4965,   0.9796],
        [ -3.0008, -10.2827,  -8.9480],
        [ -5.6792,  -8.7352,   4.0318],
        [ -8.3653,   3.4184,   9.3454],
        [  6.0890,  -0.8382,   2.9166],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.8890,  -4.4316,   1.1600],
        [ -0.0254, -10.9932,   3.2200],
        [ -3.9700,   7.3453,  -4.1486],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.2437,  -4.6489,   6.1430],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.0611,  -6.0302,   4.6962],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.4573,   3.2448,   5.4370],
        [ -4.7334,   6.9982,  -0.4392],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.9445,   7.6736,   0.5095],
        [ -3.5029,   7.5138,   8.1853],
        [ -5.9940,   6.1985,   5.9566],
        [  2.7685, -10.8982,  -6.8928],
        [  5.5240,  -2.1198,   6.2055],
        [  3.7237,  -3.9966,  -5.0376],
        [  4.7739,  -3.1218,  -4.1188],
        [ 12.2692,  -1.8882,   3.9253],
        [  9.0129,  -7.7072,   9.1400],
        [  1.0000,   7.6591,  -4.5011],
        [  0.6757, -11.0361,  -1.3047],
        [  8.5716,  10.3596,   6.8818],
        [ -7.5401,   4.6718,   9.0033],
        [ 12.3453,  -1.5462,   7.0212],
        [  6.0345,   3.0175,  -1.4424],
        [  2.6078, -10.9229,   6.2869],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.8602,   6.8905,   5.8779],
        [  3.9591, -10.6391,  -6.2557],
        [  3.7121,   6.3117,   0.6738],
        [  5.1519, -10.2387,   9.8191],
        [ -0.0282,   7.8575,  -1.1746],
        [  1.6954,  -4.7616,   9.8198],
        [ -3.3116, -10.1539,   7.5919],
        [  2.1440, -10.9811,   8.5281],
        [  2.5663,   7.0526,  -6.8769],
        [  5.4867,  12.5816,   2.9549],
        [ -5.3736,   6.6310,  -1.5174],
        [ -0.9715, -10.8630,   0.3555],
        [ -7.0253,  -7.4493,  -1.5978],
        [  5.7784,  -1.6401,   9.5587],
        [ -7.4808,   4.7462,   2.0266],
        [ 11.2517,  -4.6495,   2.9800],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.1755,   2.5233,  -2.0042],
        [  0.6933,   7.7336,   2.1179],
        [ -8.8835,  -4.3178,   9.3080],
        [  2.1504,  -4.6726,   8.7908],
        [  6.3200,   0.2690,   1.5900],
        [  3.8111, -10.6787,  -4.6539],
        [ -6.4844,  -8.0230,   8.5209],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.7792,  11.8084,   1.6251],
        [  7.8982,  -8.6778,   0.0239],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -3.0644,   7.6439,  -1.4322],
        [ -8.4573,   3.2448,   5.4370],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.2242,  -0.3171,   7.4978],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -9.4541,  -0.3327,   8.8246],
        [  3.2701,   6.6338,   9.9209],
        [  8.5198,  10.4091,  -8.6486],
        [  4.7482, -10.3907,   5.9149],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.1685,  12.1979,  -6.2473],
        [  4.1637,  13.1898,  -1.9189],
        [  6.2902,   0.0451,   3.7116],
        [  0.8572,  14.0501,  -8.3364],
        [  9.4925,   9.3811,   8.9842],
        [ -8.0087,   4.0173,   9.1091],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.3464,   0.5708,  -1.6397],
        [ -7.6525,  -6.6534,   5.1721],
        [ -9.4630,  -1.5145,   0.2983],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 11.7418,   5.6070,  -1.4192],
        [  7.4132,  11.3526,   0.3322]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.3049e-02, 1.2019e-01,  ..., 9.7232e-01, 1.0341e+00,
         1.0748e+00],
        [5.3049e-02, 2.1453e-04, 1.3451e-01,  ..., 9.2119e-01, 9.9456e-01,
         1.0395e+00],
        [1.2019e-01, 1.3451e-01, 0.0000e+00,  ..., 9.5599e-01, 9.6656e-01,
         9.9589e-01],
        ...,
        [9.7232e-01, 9.2119e-01, 9.5599e-01,  ..., 0.0000e+00, 4.4999e-01,
         5.7582e-01],
        [1.0341e+00, 9.9456e-01, 9.6656e-01,  ..., 4.4999e-01, 0.0000e+00,
         1.2620e-01],
        [1.0748e+00, 1.0395e+00, 9.9589e-01,  ..., 5.7582e-01, 1.2620e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.67132524, 10.74409834, 10.88618114, 11.04629471, 11.05413545,
       11.05672136, 11.22142195,  8.859416  , 13.00380624, 12.49291768,
       13.36800073,  5.94909709,  7.24326992, 12.7125079 ,  5.76165934,
        5.95111804, 10.24557047, 10.71162681, 10.41909211,  9.0368458 ,
        6.1463896 ,  6.14086582,  5.29009836, 10.99326577,  8.34949383,
        9.3349858 ,  9.78147542,  5.16205087, 12.59969936, 10.06703358,
       11.21703654,  9.05843201,  8.4486771 ,  9.38232121, 10.00802627,
        9.50654967,  7.73151309,  8.29022197,  8.62260327, 11.24434374,
        5.91677188,  5.46244988,  5.70405525, 12.41366754, 11.85890664,
        7.72415048, 11.05672136, 13.44593519,  8.87007495, 12.44177641,
        6.74686071, 11.22993218,  9.58072286, 10.92597196,  7.46053281,
       11.35182589,  7.32233733, 11.46176259,  7.8575648 ,  5.05444288,
       10.68031351, 11.18839169,  7.50500064, 13.72594022,  8.53501777,
       10.90638066, 10.23946247,  6.00663314,  8.859416  , 12.17451685,
        5.52416368,  6.67108662,  7.76457095,  9.87720102,  5.14371005,
        6.32572323, 11.33837396, 10.31584102,  8.12580976, 13.61600536,
       11.73397309,  9.43737523,  8.23525433,  9.05843201, 10.48721382,
        6.2322769 , 12.53285633,  9.45995955,  7.39599986, 13.45124917,
       11.42420508, 12.50996828, 13.66896739, 13.83139024,  6.29034958,
       14.07623266, 13.34587027,  8.95983586,  8.94395628,  9.36378438,
       10.14044489,  9.58347456,  9.91668182, 12.35421461, 13.01187484,
       13.55868684])}
{'P': tensor([[6.7671e-03, 2.5912e-01, 1.9536e-01,  ..., 7.9977e-07, 8.7533e-07,
         9.6832e-07],
        [3.2512e-05, 6.0093e-03, 6.6367e-02,  ..., 1.3727e-06, 5.0549e-07,
         8.9447e-07],
        [3.6683e-05, 4.2131e-04, 6.6969e-03,  ..., 1.8125e-06, 5.6147e-07,
         1.0123e-06],
        ...,
        [1.8106e-07, 1.7535e-07, 1.8520e-07,  ..., 3.8791e-02, 8.2712e-06,
         1.7154e-04],
        [8.1066e-04, 5.5460e-04, 5.8522e-04,  ..., 1.1893e-06, 3.4212e-02,
         1.1098e-05],
        [1.8291e-04, 1.0276e-04, 9.6167e-05,  ..., 2.9047e-06, 2.6627e-03,
         8.3051e-02]]), 'X': tensor([[ -3.3812, -10.1235,   7.4384],
        [ -1.6334, -10.7207,   7.1397],
        [ -1.3940, -10.7772,   6.7835],
        [ -0.8675, -10.8815,   6.8561],
        [ -0.4341, -10.9473,   7.4509],
        [  0.1252, -11.0062,   7.2564],
        [  5.1667,  -2.6543,   1.2879],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.6248,   3.9984,   8.4933],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.5974,   7.0363,  -7.7810],
        [  3.3167,   6.6022,   6.3554],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.1984,  -0.4327,   1.5438],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.0567,   8.6697,   4.6329],
        [  5.3118,   4.5452,   4.2114],
        [  2.3490,  -4.6199,  -4.2489],
        [ -3.5029,   7.5138,   8.1853],
        [  5.5294,  -2.1106,   9.0626],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -6.2200,  -8.2737,   5.5079],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.0891,  -4.9558,  -2.3178],
        [  3.3474,  -4.2162,   5.2455],
        [ -3.3812, -10.1235,   7.4384],
        [ -8.8934,   2.2597,   3.5831],
        [ -1.8788, -10.6567,   3.7068],
        [  0.6396,  14.0772,  -6.3786],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 11.1420,   6.9331,  -2.4779],
        [  6.2576,   2.1359,  -3.3398],
        [  5.6928,  12.4710,   3.6001],
        [ -5.6792,  -8.7352,   4.0318],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.1752,   2.5242,   3.5874],
        [  5.8450,  -9.9356,   3.0767],
        [ -8.7544,   2.6096,   3.3146],
        [ -2.2188,   7.8218,   8.6064],
        [ -9.4196,   0.0478,   9.4140],
        [  6.2804,   2.0019,   0.8607],
        [ -3.1671,   7.6158,   6.0765],
        [  6.7792,  11.8084,   1.6251],
        [  6.1685,  12.1979,  -6.2473],
        [  7.5309,  11.2617,   7.2656],
        [ 12.1379,   4.4125,  -3.8833],
        [  0.2480, -11.0153,  -3.0183],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.4297,  -5.3637,  -9.3962],
        [ 10.9394,   7.3069,  -8.9365],
        [  5.5158,  -2.1337,   3.5246],
        [ -6.8466,  -7.6488,   1.9648],
        [  1.8447,  -4.7372,   8.1235],
        [  3.3094,  13.4969,  -3.6507],
        [  9.0424,  -7.6779,  -7.5286],
        [  3.7992,  -3.9474,  -8.9019],
        [  5.1142,  -2.7230,  -4.1966],
        [  4.6357,   5.4415,   7.9067],
        [  1.7659,   7.4127,   4.7871],
        [  1.5662,  13.9375,   4.5142],
        [  1.7903,  -4.7467,   0.1168],
        [  4.9957,  -2.8705,   6.5978],
        [  8.7131,  10.2216,   7.7915],
        [  6.7565,  11.8237,  -4.9882],
        [  9.6662,   9.1725,   8.1056],
        [ -7.4807,   4.7463,   2.0616],
        [ -4.7389,   6.9953,   3.7638],
        [ 10.2470,   8.4054,  -0.1177],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -7.6525,  -6.6534,   5.1721],
        [  3.3167,   6.6022,   6.3554],
        [ -9.4716,  -0.6673,   7.6068],
        [  4.9531,   5.0576,   5.6794],
        [  9.5261,  -7.1675,   1.4206],
        [  8.7131,  10.2216,   7.7915],
        [  6.2576,   2.1359,  -3.3398],
        [  7.8425,  -8.7202,   6.4530],
        [  5.7784,  -1.6401,   9.5587],
        [  0.7145,  -4.8071,  -2.3766],
        [  4.0421,  -3.7765,  -8.3631],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.1929,   2.4501,  -6.9352],
        [  5.8201,   3.5841,  -8.0609],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.2804,   2.0019,   0.8607],
        [ 11.6019,   5.9540,  -5.1715],
        [ -2.7572,   7.7194,   5.9580],
        [  6.3606,   0.9204,   3.2525],
        [ -8.8835,  -4.3178,   9.3080],
        [ -7.4733,   4.7555,   6.2490],
        [  9.0424,  -7.6779,  -7.5286],
        [  0.5850,  -4.7982,   9.6639],
        [ -7.6650,   4.5092,   0.2520],
        [  1.2728, -11.0372,   9.8121],
        [  2.8994,   6.8675,  -9.5987],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.3205,   1.7068,  -3.8619],
        [  5.7784,  -1.6401,   9.5587],
        [  6.3515,   0.6418,  -6.5823],
        [ 11.6019,   5.9540,  -5.1715],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -3.1671,   7.6158,   6.0765]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0405, 0.0480,  ..., 1.4776, 0.8230, 0.4700],
        [0.0405, 0.0000, 0.0155,  ..., 1.4509, 0.8321, 0.4319],
        [0.0480, 0.0155, 0.0000,  ..., 1.4364, 0.8193, 0.4220],
        ...,
        [1.4776, 1.4509, 1.4364,  ..., 0.0000, 1.1886, 1.0926],
        [0.8230, 0.8321, 0.8193,  ..., 1.1886, 0.0000, 0.8953],
        [0.4700, 0.4319, 0.4220,  ..., 1.0926, 0.8953, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.67322527, 10.84437705, 10.86693708, 10.91601655, 10.95593729,
       11.00694688,  5.80860165,  4.91652556,  6.90115901, 10.41909211,
        7.50035355,  7.38844703, 12.25806559,  4.8763135 , 12.59969936,
        6.2134961 , 10.78533838, 13.2778378 ,  6.99097013,  5.18277858,
        8.29022197,  5.91854397, 12.12448912, 10.35093485, 14.10796466,
       12.1460931 ,  5.38342077, 10.67322527,  9.17595899, 10.82106505,
       14.09176259, 10.27716434, 13.12299145,  6.61211224, 13.70893492,
       10.41909211,  9.88571713,  9.62958973,  6.67121748, 11.52733581,
        9.13507321,  8.1303945 ,  9.41970031,  6.59176018,  8.24808832,
       13.61600536, 13.66896739, 13.54774355, 12.91505111, 11.018084  ,
        9.7712321 ,  5.83560009,  9.99145096, 13.15525803,  5.91408504,
       10.26545792,  5.08372149, 13.89671314, 11.86239961,  5.47866221,
        5.79391825,  7.14837514,  7.62011734, 14.02526505,  5.07306274,
        5.76165934, 13.43127131, 13.61801526, 13.32557133,  8.85939626,
        8.44940661, 13.25336001, 10.60226614, 10.14044489,  7.38844703,
        9.49511524,  7.07902036, 11.92133129, 13.43127131,  6.61211224,
       11.72802491,  6.00663314,  4.85995002,  5.53175275,  9.5642993 ,
        6.65991877,  6.83515201,  4.978704  ,  6.59176018, 13.04051744,
        8.19703512,  6.42689705,  9.87720102,  8.8580671 , 11.86239961,
        4.83371341,  8.89301516, 11.11038551,  7.4544866 ,  4.78258092,
        6.54693987,  6.00663314,  6.383897  , 13.04051744,  9.5014146 ,
        8.24808832])}
{'P': tensor([[6.7055e-03, 2.4154e-01, 1.7407e-01,  ..., 6.6793e-07, 6.3577e-07,
         1.1953e-04],
        [3.1936e-05, 5.3558e-03, 5.9150e-02,  ..., 1.0443e-06, 4.6702e-07,
         8.9838e-05],
        [3.4642e-05, 3.5185e-04, 5.5929e-03,  ..., 1.3195e-06, 4.9558e-07,
         1.0290e-04],
        ...,
        [2.5970e-07, 3.0232e-07, 3.3066e-07,  ..., 1.1483e-01, 4.8419e-05,
         2.6530e-05],
        [3.8484e-04, 2.1900e-04, 2.1037e-04,  ..., 9.6714e-07, 6.7826e-02,
         6.1570e-04],
        [4.7830e-04, 7.7603e-04, 9.0860e-04,  ..., 6.0112e-04, 1.7226e-04,
         1.1150e-01]]), 'X': tensor([[ -3.3116, -10.1539,   7.5919],
        [ -1.6334, -10.7207,   7.1397],
        [ -1.3940, -10.7772,   6.7835],
        [ -0.4341, -10.9473,   7.4509],
        [ -0.8675, -10.8815,   6.8561],
        [  0.1252, -11.0062,   7.2564],
        [ -9.1624,  -3.4286,   2.0647],
        [ -2.0721,   7.8432,   6.6380],
        [  3.7121,   6.3117,   0.6738],
        [ -2.0721,   7.8432,   6.6380],
        [  6.3119,   1.7797,   9.9186],
        [ -1.0012, -10.8576,   2.2867],
        [  5.9843,  -9.8679,  -6.1449],
        [  6.3589,   0.8239,  -8.4230],
        [  9.9058,  -6.7203,   5.6928],
        [  9.4925,   9.3811,   8.9842],
        [  6.0310,  12.2795,  -6.4528],
        [  6.2242,  -0.3171,   7.4978],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.1372,   2.6726,  -6.7885],
        [ -0.8455, -10.8853,  -6.7400],
        [ 12.5224,  -0.4497,   4.8448],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.3367,   7.1676,  -9.5263],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.1771, -10.9776,  -3.2007],
        [ -7.6525,  -6.6534,   5.1721],
        [  1.8320,   7.3871,   5.1452],
        [ -6.0195,   6.1789,  -2.9496],
        [ -3.6381,   7.4683,   7.3544],
        [  5.1890,   4.7318,   0.9515],
        [ -0.8455, -10.8853,  -6.7400],
        [  5.4867,  12.5816,   2.9549],
        [  0.6363, -11.0348,   6.1414],
        [ -9.4378,  -1.8606,  -2.5668],
        [  1.6954,  -4.7616,   9.8198],
        [  5.9883,   3.1536,   6.2320],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.6059,   0.9597,  -6.3264],
        [ -8.6426,  -4.9136,   9.4941],
        [  6.3200,   0.2690,   1.5900],
        [ -8.0608,   3.9362,  -5.9567],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.7078,   2.7182,   1.8146],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.5158,  -2.1337,   3.5246],
        [ 12.0582,  -2.6640,   0.7132],
        [  4.2275,  -3.6320,  -0.2714],
        [  3.3943,   6.5483,   1.0790],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.3589,   0.8239,  -8.4230],
        [ 11.9122,   5.1397,  -1.5498],
        [ -2.5683, -10.4433,  -2.9182],
        [ -7.3554,   4.8985,  -0.6665],
        [ 11.7779,   5.5127,  -2.0699],
        [  0.9967,  -4.8144,  -9.7370],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.3744,   6.5623,  -6.3636],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.0829,  -5.9940,   5.0564],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.6019,   5.9540,  -5.1715],
        [  4.9337,  12.8566,  -3.6582],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.0829,  -5.9940,   5.0564],
        [  3.3943,   6.5483,   1.0790],
        [  7.3724,  -9.0577,  -8.9843],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 12.5832,   1.7543,  -7.6134],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.2692,   9.6370,   7.0746],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3236,   0.3011,  -4.6668],
        [ -6.3252,  -8.1760,  -1.0891],
        [  9.3624,   9.5318,   0.2779],
        [  4.1297,  -3.7098,  -3.9838],
        [ -3.5173, -10.0623,   4.8280],
        [  2.5128, -10.9365,   9.9645],
        [  4.6992,   5.3690,   3.0935],
        [  4.8674,  12.8875,   6.7622],
        [  8.6420,  -8.0580,   6.5781],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.2905,   1.9366,   2.1169],
        [ -6.7992,   5.4991,   9.9186],
        [  6.3031,   1.8477,  -7.1706],
        [ 12.5717,   1.9266,   7.4903],
        [  2.5318,  13.7231,   6.3692],
        [  0.9809,   7.6642,  -1.4217],
        [  1.8766,   7.3694,   1.1887],
        [ -4.8665,  -9.3204,  -5.2564],
        [  2.5829,  -4.5466,  -8.4015],
        [  0.3724,   7.7974,  -6.8722],
        [ -9.4196,   0.0478,   9.4140],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.6258,  -1.9404,  -6.0715],
        [ -9.2345,  -3.1345,   3.9715],
        [  0.2435,   7.8191,  -6.4709],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.4149,  -4.6005,  -4.3339],
        [  9.3753,   9.5171,   3.9131],
        [ -6.2200,  -8.2737,   5.5079],
        [ 12.5154,   2.5072,  -9.3274],
        [ -6.2668,   5.9808,   2.5614],
        [  0.3231,  -4.7691,  -1.6120]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0392, 0.0491,  ..., 1.6015, 0.4691, 0.9867],
        [0.0392, 0.0000, 0.0155,  ..., 1.5723, 0.4430, 0.9564],
        [0.0491, 0.0155, 0.0000,  ..., 1.5574, 0.4277, 0.9417],
        ...,
        [1.6015, 1.5723, 1.5574,  ..., 0.0000, 1.1350, 0.6178],
        [0.4691, 0.4430, 0.4277,  ..., 1.1350, 0.0000, 0.5259],
        [0.9867, 0.9564, 0.9417,  ..., 0.6178, 0.5259, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.68031351, 10.84437705, 10.86693708, 10.95593729, 10.91601655,
       11.00694688,  9.78285424,  8.11227668,  7.32233733,  8.11227668,
        6.55800267, 10.9036226 , 11.54071859,  6.41202986, 11.9702762 ,
       13.34587027, 13.68063098,  6.2322769 , 12.12448912,  6.69389431,
       10.91805683, 12.53047489,  5.19065439,  7.53884104,  8.8353674 ,
       11.1913598 , 10.14044489,  7.61088125,  8.62631801,  8.30729352,
        7.02253235, 10.91805683, 13.72594022, 11.05317597,  9.61942445,
        5.05444288,  6.76791483,  9.34127013, 12.64235359,  9.94173708,
        6.32572323,  8.97052115,  9.9353921 ,  9.12220271, 11.75618513,
        5.91408504, 12.3489324 ,  5.57341623,  7.37576154, 13.20597932,
        6.41202986, 12.97370371, 10.75443106,  8.83725081, 13.00413829,
        4.91652556, 10.41909211,  7.37902766,  8.1137452 , 10.06286074,
       11.81423161, 13.04051744, 13.77075174,  8.62260327,  9.24501829,
        9.7712321 , 10.06286074,  7.37576154, 11.67875855, 10.33712258,
       12.70489563, 10.06286074, 13.37121923, 12.27993098,  6.33075682,
       10.33712258, 13.3607379 ,  5.5513072 , 10.65929968, 11.22142195,
        7.13501264, 13.77604293, 11.8159254 ,  5.5671389 ,  6.58184474,
        8.74470218,  6.56833864, 12.71843462, 13.95472331,  7.72668467,
        7.60463349, 10.51437947,  5.22903117,  7.80625981,  9.41970031,
        5.21280156,  5.9510489 ,  9.75200848,  7.822854  , 12.05395056,
        5.19577146, 13.35927527, 10.35093485, 12.76408562,  8.66273171,
        4.78002665])}
{'P': tensor([[6.0308e-03, 2.8072e-01, 2.7942e-01,  ..., 1.8776e-06, 2.2923e-05,
         6.4105e-03],
        [7.8425e-05, 1.2749e-02, 1.1732e-04,  ..., 5.2608e-06, 2.2256e-05,
         1.0219e-02],
        [6.0195e-05, 3.6699e-01, 5.8885e-03,  ..., 2.2442e-06, 1.4752e-05,
         6.5703e-03],
        ...,
        [1.2672e-05, 1.0853e-05, 1.2044e-05,  ..., 1.0471e-01, 6.9648e-05,
         1.6912e-05],
        [1.3863e-02, 1.8284e-02, 1.8853e-02,  ..., 4.0607e-05, 7.7937e-02,
         1.2704e-01],
        [3.5115e-04, 1.0244e-03, 6.2027e-04,  ..., 1.4141e-04, 5.3191e-05,
         8.5804e-02]]), 'X': tensor([[ -3.2736, -10.1703,   3.8712],
        [ -1.8788, -10.6567,   3.7068],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.6275, -10.7221,   3.2376],
        [ -0.0254, -10.9932,   3.2200],
        [  0.5960, -11.0335,   3.1144],
        [  5.8836,   3.4315,  -4.0327],
        [ 12.6035,   0.7494,   7.1536],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.4753,  -0.8012,  -5.2050],
        [  5.2608,  12.6977,   4.9166],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.9526,   3.2525,  -4.3821],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.3705,   0.4208,  -3.5815],
        [  9.2811,  -7.4336,   5.1771],
        [  3.1892, -10.8220,  -6.8409],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.1120,  -4.6818,   8.6144],
        [  5.8889,  -1.3916,   8.8993],
        [ -9.3604,   0.4855,  -3.0158],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.8994,   6.8675,  -9.5987],
        [ -5.6169,  -8.7844,   8.9057],
        [  2.7590,  13.6621,   9.4496],
        [  6.2535,  -0.1709,  -9.4102],
        [ -1.5261,   7.8988,   9.3768],
        [ -7.3333,   4.9246,  -7.9212],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.7078,   2.7182,   1.8146],
        [  1.5653,  13.9377,   4.1773],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.8320,   7.3871,   5.1452],
        [  5.8902,  -1.3887,   4.0388],
        [ -2.6485, -10.4151,   0.8766],
        [  6.8704,  -9.3809,   7.5769],
        [ -8.6050,   2.9451,  -0.7012],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.4524,  -1.6813,  -6.6862],
        [  5.2608,  12.6977,   4.9166],
        [ -9.4159,  -2.0739,   9.6841],
        [  2.6970,   6.9826,   9.7350],
        [ -0.6323,   7.9085,  -5.9544],
        [ -1.1885, -10.8211,   0.0865],
        [ 10.6283,  -5.7253,   7.4878],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.0004,  -7.7195,   4.6945],
        [  3.9913,  -3.8139,  -6.8311],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.2675,  12.1379,  -8.9720],
        [  6.3587,   1.1596,  -3.6629],
        [  4.0642,  -3.7599,  -3.4565],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 12.2311,  -2.0445,   1.3545],
        [  2.1771, -10.9776,  -3.2007],
        [  6.2022,   2.4092,   3.0349],
        [ -9.3331,  -2.6465,  -8.2028],
        [  1.4306,  13.9620,  -0.2075],
        [  3.5939,   6.4029,  -1.8357],
        [  6.2576,   2.1359,  -3.3398],
        [  6.1752,   2.5242,   3.5874],
        [  5.8930,  12.3592,  -8.9201],
        [  3.3839,  -4.1966,   8.3574],
        [  1.7659,   7.4127,   4.7871],
        [ 11.6019,   5.9540,  -5.1715],
        [  6.3534,   1.2990,  -7.2942],
        [ 11.7795,  -3.4697,   3.5624],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -8.8058,   2.4850,  -1.7425],
        [ -0.0282,   7.8575,  -1.1746],
        [  8.6271,  -8.0715,  -7.4813],
        [  7.5309,  11.2617,   7.2656],
        [  1.4306,  13.9620,  -0.2075],
        [ 12.1588,   4.3364,  -8.0867],
        [  5.7784,  -1.6401,   9.5587],
        [  5.1519, -10.2387,   9.8191],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.6154,  -4.0640,  -7.9698],
        [  5.8201,   3.5841,  -8.0609],
        [ -7.7239,   4.4298,   3.1216],
        [ 11.0438,   7.1180,  -5.6991],
        [ -9.3705,   0.4208,  -3.5815],
        [  4.5503,   5.5359,   7.4735],
        [  0.8572,  14.0501,  -8.3364],
        [  6.1685,  12.1979,  -6.2473],
        [  9.4554,   9.4245,   9.8800],
        [ 11.7726,  -3.4876,  -6.7402],
        [  0.7685,  -4.8098,  -2.9310],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.6308,  13.9254,  -2.0984],
        [  3.4604,  -4.1545,   1.8091],
        [  2.3367,   7.1676,  -9.5263],
        [ -8.1426,   3.8048,   9.3542],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.8902,  -1.3887,   4.0388],
        [  2.3928,   7.1404,  -1.7035],
        [  1.4749,  -4.7892,  -0.8089],
        [  2.6970,   6.9826,   9.7350],
        [ -7.7842,   4.3466,  -1.7548],
        [  1.9241,   7.3503,   7.9004],
        [  4.1537,   5.9340,  -6.0423],
        [ -6.4822,  -8.0252,  -3.8983],
        [  0.6470, -11.0352,  -1.1931]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0645, 0.0275,  ..., 1.0283, 0.6674, 0.4905],
        [0.0645, 0.0000, 0.0450,  ..., 0.9803, 0.6812, 0.4668],
        [0.0275, 0.0450, 0.0000,  ..., 1.0010, 0.6559, 0.4668],
        ...,
        [1.0283, 0.9803, 1.0010,  ..., 0.0000, 0.8910, 0.6146],
        [0.6674, 0.6812, 0.6559,  ..., 0.8910, 0.0000, 0.3847],
        [0.4905, 0.4668, 0.4668,  ..., 0.6146, 0.3847, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.68416401, 10.82106505, 10.76154256, 10.84493732, 10.99326577,
       11.04953767,  6.81117802, 12.62576026,  5.56809335,  9.50913849,
       13.744387  ,  7.28793663,  9.96391045,  9.79039371,  6.78325151,
        5.39450305,  9.37989631, 11.89105259, 11.28215429, 13.6660671 ,
        9.49511524,  5.13616224,  6.05113217,  9.37296102, 10.1371552 ,
        7.4544866 , 10.42667093, 13.93790235,  6.25586297,  8.04483495,
        8.83341608, 14.07623266,  9.12220271, 14.02532758,  5.24137909,
        7.61088125,  6.05165248, 10.74655346, 11.62770332,  9.09502385,
       12.3828957 ,  9.60080639, 13.744387  ,  9.64157461,  7.48538905,
        7.9337647 , 10.88618114, 12.07225469, 10.06286074, 11.85743425,
        5.5205189 ,  7.88677538, 13.66051884,  6.46356806,  5.5366698 ,
       12.27835996, 12.40074901, 11.1913598 ,  6.65368551,  9.70108409,
       14.03505624,  7.34250948,  6.61211224,  6.67121748, 13.6922425 ,
        5.39098669,  7.62011734, 13.04051744,  6.48486171, 12.27991991,
       10.32715715,  9.14973319,  7.8575648 , 11.81423161, 13.54774355,
       14.03505624, 12.9089559 ,  6.00663314, 11.46176259,  8.22728399,
        5.43943791,  6.83515201,  8.90405823, 13.13888841,  9.37989631,
        7.16598273, 14.07623266, 13.66896739, 13.35013374, 12.27835996,
        4.87083348, 10.91805683, 14.02058621,  5.40689332,  7.53884104,
        8.98765644, 12.12448912,  6.05165248,  7.53063885,  5.01114719,
        7.48538905,  8.91551231,  7.59795445,  7.24326992, 10.31613269,
       11.05413545])}
{'P': tensor([[5.0986e-03, 2.5351e-01, 8.5997e-02,  ..., 4.0087e-02, 5.0092e-04,
         1.1810e-07],
        [6.8313e-05, 1.1028e-02, 2.8072e-01,  ..., 2.6080e-02, 1.4601e-03,
         3.3869e-07],
        [1.9695e-04, 3.9572e-04, 7.1636e-02,  ..., 2.8541e-03, 1.5889e-02,
         3.3478e-06],
        ...,
        [2.9247e-04, 1.9855e-03, 5.1489e-02,  ..., 2.9543e-02, 3.0318e-03,
         1.2902e-06],
        [1.2471e-04, 1.5237e-04, 2.3088e-04,  ..., 1.5365e-04, 1.1701e-01,
         2.9388e-04],
        [4.1426e-07, 2.8808e-07, 1.6425e-07,  ..., 2.7265e-07, 1.2143e-06,
         1.5254e-02]]), 'X': tensor([[ -3.2053, -10.1993,  -3.8736],
        [ -1.8823, -10.6558,  -3.8952],
        [  0.2480, -11.0153,  -3.0183],
        [ -2.2508, -10.5479,  -2.7235],
        [  1.5383, -11.0274,  -4.9666],
        [ -2.5683, -10.4433,  -2.9182],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.0876,   6.7527,  -5.3540],
        [  1.6408,   7.4591,  -4.2294],
        [ -5.1377,   6.7751,  -0.4438],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.0031,  12.2958,  -6.5992],
        [ -2.6485, -10.4151,   0.8766],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 11.2767,  -4.6006,   0.7868],
        [ 12.2311,  -2.0445,   1.3545],
        [  0.2480, -11.0153,  -3.0183],
        [ -3.1671,   7.6158,   6.0765],
        [ -8.9606,  -4.0994,   1.7886],
        [ 12.4996,   2.6335,  -6.3166],
        [ -2.0840,   7.8415,  -7.8657],
        [  7.4132,  11.3526,   0.3322],
        [  9.3836,   9.5075,  -5.9278],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.6258,  -1.9404,  -6.0715],
        [ -5.0550,   6.8231,  -9.2502],
        [  8.5716,  10.3596,   6.8818],
        [  8.7715,  -7.9390,   5.2900],
        [ -1.1885, -10.8211,   0.0865],
        [  3.6817,   6.3354,   6.1233],
        [ -1.0012, -10.8576,   2.2867],
        [  6.0031,  12.2958,  -6.5992],
        [  0.3231,  -4.7691,  -1.6120],
        [  5.0506,   4.9277,  -9.5094],
        [  4.1348,   5.9514,  -0.8856],
        [  5.2188,  -2.5839,   2.1833],
        [  6.1755,   2.5233,  -2.0042],
        [ -1.9487,   7.8590,  -3.3330],
        [ 10.1798,   8.5002,  -5.8893],
        [  4.9531,   5.0576,   5.6794],
        [  6.3087,   1.8053,  -1.9481],
        [  8.8474,  10.0868,  -8.8644],
        [ -5.3382,   6.6533,   7.8054],
        [  4.0591,   6.0202,  -1.6339],
        [ -0.9715, -10.8630,   0.3555],
        [ -7.8849,  -6.3115,  -4.7479],
        [  6.3266,   0.3292,   9.3502],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.1667,  -2.6543,   1.2879],
        [  5.8201,   3.5841,  -8.0609],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.3261,   7.1726,   6.2522],
        [ 12.2921,  -1.7902,  -4.0870],
        [  3.2366,  -4.2732,   5.1532],
        [ -3.2053, -10.1993,  -3.8736],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.6310, -10.9195,   3.8054],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.9470,  13.6085,   2.9809],
        [ 12.1379,   4.4125,  -3.8833],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.0093,   7.6567,   2.3526],
        [  4.3377,   5.7570,   1.3712],
        [ 12.4592,  -0.9168,  -6.7571],
        [  6.8407,  11.7666,   3.0045],
        [  1.4489,   7.5256,   4.4529],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.2555,   2.1474,   4.1685],
        [  5.8450,  -9.9356,   3.0767],
        [  9.3836,   9.5075,  -5.9278],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.0042,  -2.8350,   2.6864],
        [  4.1537,   5.9340,  -6.0423],
        [ -1.0012, -10.8576,   2.2867],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.3303,   1.6145,  -8.4083],
        [  3.7298,  13.3537,  -9.4653],
        [  6.2770,  -0.0379,   2.4195],
        [ -8.9178,   2.1936,   7.5388],
        [  1.1350,  14.0104,   0.6321],
        [ -0.8455, -10.8853,  -6.7400],
        [ -9.3444,   0.5821,   6.9101],
        [  0.5600, -11.0321,  -0.8660],
        [  5.4867,  12.5816,   2.9549],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.4524,  -1.6813,  -6.6862],
        [  5.3858,   4.4260,  -9.6769],
        [ 11.0373,   7.1299,  -8.5895],
        [ -8.6603,   2.8250,   5.9181],
        [  0.5960, -11.0335,   3.1144],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.6757, -11.0361,  -1.3047],
        [  2.4356,  -4.5942,  -5.0808],
        [  1.1139,   7.6281,  -2.5680],
        [  5.7963,  -1.6018,   1.8856],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.8572,  14.0501,  -8.3364],
        [ -3.4986,   7.5152,   9.0712],
        [  3.9085,   6.1512,  -2.2560],
        [ -1.0482, -10.8487,  -5.9066],
        [  4.7544,  -3.1425,  -1.5315],
        [  3.7298,  13.3537,  -9.4653]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0733, 0.1694,  ..., 0.1701, 0.8282, 1.0930],
        [0.0733, 0.0000, 0.1019,  ..., 0.1367, 0.7584, 1.0220],
        [0.1694, 0.1019, 0.0000,  ..., 0.1918, 0.6588, 0.9527],
        ...,
        [0.1701, 0.1367, 0.1918,  ..., 0.0000, 0.7818, 0.9633],
        [0.8282, 0.7584, 0.6588,  ..., 0.7818, 0.0000, 0.5750],
        [1.0930, 1.0220, 0.9527,  ..., 0.9633, 0.5750, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.6910768 , 10.8207297 , 11.018084  , 10.78533838, 11.1341798 ,
       10.75443106, 10.8207297 ,  7.42512621,  7.6374603 ,  8.50278896,
        5.70405525, 13.6829843 , 10.74655346, 12.11735334, 12.17901033,
       12.40074901, 11.018084  ,  8.24808832,  9.85384623, 12.77402351,
        8.1137452 , 13.55868684, 13.35832725,  9.70108409,  5.9510489 ,
        8.49161154, 13.44593519, 11.83075005, 10.88618114,  7.32754882,
       10.9036226 , 13.6829843 ,  4.78002665,  7.05627461,  7.24677664,
        5.82345867,  6.67108662,  8.09704052, 13.26209545,  7.07902036,
        6.5618968 , 13.41713661,  8.53014915,  7.26074925, 10.90638066,
       10.09979747,  6.33517349,  7.9337647 ,  5.80860165,  6.83515201,
        5.19065439,  7.5403822 , 12.4217523 ,  5.36061974, 10.6910768 ,
       12.3828957 , 11.23201619, 10.31584102, 13.92390689, 12.91505111,
        8.05242591,  7.72291946,  7.20826219, 12.49291768, 13.6105595 ,
        7.66377554, 11.05413545,  9.3109884 ,  8.95983586,  9.79039371,
        6.61386436, 11.52733581, 13.35832725,  5.60064319, 12.3344536 ,
        7.24326992, 10.9036226 ,  9.58072286,  6.53290663, 13.86480175,
        6.27715105,  9.18358562, 14.05633441, 10.91805683,  9.36256096,
       11.04629471, 13.72594022,  6.77777606,  9.5642993 ,  9.60080639,
        6.97106912, 13.1399181 ,  9.10945678, 11.04953767, 13.00380624,
       11.05672136,  5.19986797,  7.70897675,  6.01356004, 10.26545792,
       14.07623266,  8.28968067,  7.28793663, 10.8992524 ,  5.69913817,
       13.86480175])}
{'P': tensor([[1.1304e-02, 3.9200e-01, 1.4318e-01,  ..., 1.3598e-05, 2.9434e-06,
         1.1225e-06],
        [1.2227e-04, 1.2444e-02, 2.0995e-01,  ..., 2.1842e-05, 2.4108e-06,
         9.9884e-07],
        [5.4261e-05, 9.4329e-05, 3.0399e-02,  ..., 5.4197e-05, 2.3477e-06,
         7.2603e-07],
        ...,
        [9.2972e-06, 1.1511e-05, 1.1882e-05,  ..., 1.3025e-01, 5.4059e-05,
         7.5175e-05],
        [3.0466e-03, 2.5124e-03, 5.6605e-04,  ..., 2.8622e-05, 2.4582e-02,
         1.3974e-03],
        [1.4023e-04, 1.6833e-04, 6.7692e-05,  ..., 1.7207e-04, 2.2463e-04,
         4.0058e-02]]), 'X': tensor([[ -3.1204, -10.2345,   9.2288],
        [ -2.0229, -10.6163,   8.6930],
        [  1.2728, -11.0372,   9.8121],
        [  1.3767, -11.0341,   9.9415],
        [ -0.4341, -10.9473,   7.4509],
        [  1.1644, -11.0394,   7.8511],
        [  4.6992,   5.3690,   3.0935],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4671,  -0.5553,   0.1117],
        [  8.9284,  10.0036,  -6.9210],
        [ -7.4733,   4.7555,   6.2490],
        [  5.6258,  -1.9404,  -6.0715],
        [ -8.3539,  -5.5111,   0.3373],
        [  7.2340,  -9.1505,  -5.8321],
        [ -4.0409,   7.3169,  -3.5779],
        [ -8.4267,  -5.3697,   2.4172],
        [ -1.5451,   7.8975,   7.6104],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -8.3535,  -5.5118,   9.8399],
        [ -8.8934,   2.2597,   3.5831],
        [  3.6249, -10.7255,   5.6605],
        [  2.4572,   7.1085,   7.7376],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.6523,   3.9443,  -2.9449],
        [  1.7218,  -4.7576,  -3.3617],
        [  1.7659,   7.4127,   4.7871],
        [  7.9816,  10.8938,  -0.5378],
        [ -6.8466,  -7.6488,   1.9648],
        [ -5.6169,  -8.7844,   8.9057],
        [  0.3231,  -4.7691,  -1.6120],
        [ 11.3118,   6.5957,  -4.5667],
        [  7.9403,  -8.6455,  -8.7433],
        [  3.2990,  -4.2415,  -7.4857],
        [ 12.4901,  -0.7052,   1.7315],
        [  4.7544,  -3.1425,  -1.5315],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.4426, -10.1181,  -5.1268],
        [  3.4920,  -4.1366,  -8.3955],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.2038,   2.4020,  -3.1692],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -7.6719,  -6.6260,   7.1468],
        [ -9.4756,  -0.8166,  -9.6021],
        [  7.7750,  11.0665,   2.8364],
        [ -5.1974,   6.7396,   3.3484],
        [ -8.0330,  -6.0765,   6.7945],
        [ 11.4235,   6.3594,   5.1663],
        [ -6.4368,   5.8356,   3.3014],
        [  3.2366,  -4.2732,   5.1532],
        [  9.6614,   9.1784,  -1.2087],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -7.4807,   4.7463,   2.0616],
        [  3.0876,   6.7527,  -5.3540],
        [  0.1532,  -4.7422,  -7.0145],
        [ 11.0373,   7.1299,  -8.5895],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.1886,  -2.6250,   6.4677],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.1751,  -2.2592,  -5.7751],
        [  3.5939,   6.4029,  -1.8357],
        [ -8.3062,  -5.6008,   0.0287],
        [  9.8082,  -6.8396,   4.9615],
        [ 11.2209,  -4.7092,   2.6117],
        [  1.8447,  -4.7372,   8.1235],
        [ 12.5770,   1.8512,  -0.6578],
        [ -7.3333,   4.9246,  -7.9212],
        [ -8.0750,  -6.0072,  -5.8267],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.2209,  -4.7092,   2.6117],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.1685,  12.1979,  -6.2473],
        [  5.3858,   4.4260,  -9.6769],
        [  3.7916,  -3.9524,  -7.9547],
        [  2.4149,  -4.6005,  -4.3339],
        [ 10.5955,   7.8824,  -3.7006],
        [ -8.0829,  -5.9940,   5.0564],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.8275,  -1.5334,   2.0124],
        [ -1.6275, -10.7221,   3.2376],
        [  0.1460,   7.8340,  -9.4614],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.8280,   4.2848,  -7.1955],
        [  5.2199,   4.6861,  -1.9304],
        [  1.2273,  -4.8081,   9.8129],
        [ -3.1671,   7.6158,   6.0765],
        [  5.4167, -10.1293,   6.2156],
        [ -0.5385,   7.9037,   6.3830],
        [  6.0031,  12.2958,  -6.5992],
        [  2.3373,  13.7721,  -0.2442],
        [  7.8425,  -8.7202,   6.4530],
        [ -0.2586,   7.8825,  -9.9840],
        [  7.5784,  -8.9141,   5.8410],
        [  4.1140,  -3.7220,  -0.1273],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.3739,   0.3982,  -4.2564],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.4167, -10.1293,   6.2156],
        [  4.5340,  -3.3629,   8.9652],
        [ 12.3357,   3.6001,   0.7903],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.4203,  -2.0340,   2.5856],
        [ -8.7409,  -4.6837,  -6.5840]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0356, 0.2094,  ..., 1.2195, 0.4477, 1.1109],
        [0.0356, 0.0000, 0.1758,  ..., 1.1935, 0.4542, 1.1137],
        [0.2094, 0.1758, 0.0000,  ..., 1.1302, 0.5716, 1.2030],
        ...,
        [1.2195, 1.1935, 1.1302,  ..., 0.0000, 1.0367, 1.0099],
        [0.4477, 0.4542, 0.5716,  ..., 1.0367, 0.0000, 0.6658],
        [1.1109, 1.1137, 1.2030,  ..., 1.0099, 0.6658, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.69963954, 10.80728788, 11.11038551, 11.11969815, 10.95593729,
       11.10066126,  7.13501264, 13.36092624,  9.48337063, 13.40850328,
        8.8580671 ,  5.9510489 , 10.00794468, 11.66453059,  8.35856487,
        9.99211663,  8.04718433,  9.7712321 , 10.00802627,  9.17595899,
       11.32149259,  7.52117132, 12.64235359,  6.89245352,  5.05963217,
        7.62011734, 13.50484539, 10.26545792, 10.42667093,  4.78002665,
       13.09426142, 11.73848065,  5.37343173, 12.50996828,  5.69913817,
        9.43737523, 12.1460931 , 12.33167382, 11.4890861 ,  5.41348121,
       12.49317563, 10.91805683,  6.65259775,  9.79039371, 10.1371552 ,
        9.51074763, 13.5247215 ,  8.51088913, 10.07238463, 13.0743399 ,
        8.68832398,  5.36061974, 13.32613445,  9.66977802,  8.85939626,
        7.42512621,  4.74468406, 13.1399181 , 13.14777433,  5.81481044,
        5.60064319, 12.3828957 ,  7.34250948, 10.01804223, 11.95743558,
       12.16901615,  5.08372149, 12.7125079 ,  8.83341608, 10.06437598,
        5.1224079 , 12.16901615, 10.60226614, 13.66896739,  6.97106912,
        5.47702766,  5.19577146, 13.20597932, 10.06286074,  5.24137909,
        6.02588517, 10.84493732,  7.83534987, 10.74655346,  8.92395381,
        7.01475376,  4.96231532,  8.24808832, 11.48663767,  7.92200342,
       13.6829843 , 13.96905756, 11.72802491,  7.88677538, 11.70016217,
        5.54776934,  8.1137452 ,  9.38232121, 13.89974693, 10.01804223,
       11.48663767,  5.64500512, 12.85032836,  6.12886024,  9.63742669,
        9.91668182])}
{'P': tensor([[3.0414e-02, 2.2904e-01, 2.0032e-01,  ..., 1.5236e-05, 3.7805e-06,
         4.1551e-04],
        [2.9547e-05, 2.2458e-02, 6.9689e-05,  ..., 2.4626e-06, 4.7060e-06,
         1.3034e-03],
        [1.2477e-04, 2.3974e-01, 1.1575e-02,  ..., 3.0644e-06, 1.9231e-06,
         2.7344e-04],
        ...,
        [4.1970e-02, 1.0228e-02, 4.3162e-02,  ..., 1.4480e-01, 3.9685e-05,
         1.7887e-04],
        [1.1306e-04, 3.2014e-05, 7.5195e-05,  ..., 3.9693e-03, 1.0986e-01,
         1.1166e-04],
        [3.3419e-05, 4.3661e-05, 3.4345e-05,  ..., 4.4864e-05, 3.3835e-03,
         9.3374e-02]]), 'X': tensor([[ -3.0008, -10.2827,  -8.9480],
        [  2.2849, -10.9655,  -7.6542],
        [ -0.8455, -10.8853,  -6.7400],
        [  2.4639, -10.9431,  -7.3718],
        [  2.7685, -10.8982,  -6.8928],
        [  3.1892, -10.8220,  -6.8409],
        [  5.6248,   3.9984,   8.4933],
        [  4.3523, -10.5232,   8.2138],
        [ 12.2435,   4.0076,   8.1257],
        [ -5.3382,   6.6533,   7.8054],
        [  5.1203,  -2.7151,   6.2110],
        [  5.3804,  -2.3504,   7.2744],
        [  1.2273,  -4.8081,   9.8129],
        [  0.8940, -11.0402,  -0.9783],
        [  2.5318,  13.7231,   6.3692],
        [ -6.8618,  -7.6322,   3.7027],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.3569,   0.7591,   6.5011],
        [ -3.3812, -10.1235,   7.4384],
        [  0.3231,  -4.7691,  -1.6120],
        [  4.2275,  -3.6320,  -0.2714],
        [  2.3261,  13.7749,   5.9121],
        [  5.8836,   3.4315,  -4.0327],
        [  7.8425,  -8.7202,   6.4530],
        [ -7.4808,   4.7462,   2.0266],
        [ -2.6485, -10.4151,   0.8766],
        [  2.0039,   7.3173,  -1.5885],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.4159,  -2.0739,   9.6841],
        [ -8.9636,   2.0648,   8.1946],
        [  4.1537,   5.9340,  -6.0423],
        [  4.0587, -10.6112,   6.2273],
        [  2.6078, -10.9229,   6.2869],
        [  6.8704,  -9.3809,   7.5769],
        [  5.2188,  -2.5839,   2.1833],
        [  3.9515,  -3.8425,   6.1729],
        [  6.3360,   0.4258,  -9.2509],
        [  6.2038,   2.4020,  -3.1692],
        [  4.2096,   5.8815,   1.9230],
        [  5.1203,  -2.7151,   6.2110],
        [  2.7590,  13.6621,   9.4496],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.9284,  11.7061,  -0.3732],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4203,  -2.0340,   2.5856],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.5382,   7.0672,   4.5097],
        [ -2.6069, -10.4298,  -6.1995],
        [ -9.2508,   1.0572,   2.9978],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.0870,  -4.6877,   9.6823],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.1861,   4.7361,  -3.7312],
        [  2.1379,  -4.6757,   6.4981],
        [  6.3360,   0.4258,  -9.2509],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.2687,  -2.9794,   0.6436],
        [  6.0073,   3.0986,   6.7937],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.3048,   0.1469,  -9.1691],
        [  2.3200,   7.1756,   1.1938],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.7078,   2.7182,   1.8146],
        [  2.5813,  -4.5471,   5.3597],
        [ -1.8367,   7.8718,  -8.6523],
        [ -1.6906,   7.8860,   7.7142],
        [ -2.2188,   7.8218,   8.6064],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.1497,   7.8334,  -5.4412],
        [ -7.3446,   4.9113,  -5.9356],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.3535,  -5.5118,   9.8399],
        [  7.9141,  -8.6656,   9.6948],
        [ -9.2373,  -3.1225,  -8.9763],
        [  4.8674,  12.8875,   6.7622],
        [ -2.5161,   7.7699,  -9.3531],
        [  3.9895,   6.0816,  -6.5235],
        [  0.1287, -11.0065,   6.5757],
        [ -8.5607,  -5.0938,  -5.4531],
        [  5.2911,  -2.4824,  -3.6877],
        [ 10.0995,  -6.4739,   2.0974],
        [  1.7903,  -4.7467,   0.1168],
        [  5.3860,  -2.3418,   1.0826],
        [  0.2480, -11.0153,  -3.0183],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.3107,  -2.7697,  -3.2919],
        [  6.9956,  11.6590,  -7.6876],
        [ -8.0829,  -5.9940,   5.0564],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.9875,   7.2207,  -8.8549],
        [ -0.3432,   7.8900,   7.1805],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.9653,  -9.3225,  -3.4727],
        [  3.4009,  -4.1874,  -5.9751],
        [  1.1139,   7.6281,  -2.5680],
        [  2.7420,  -4.4896,  -3.7139],
        [ -5.2589,   6.7024,  -5.5809],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -2.3317,   7.8034,  -3.6274],
        [  6.3589,   0.8239,  -8.4230]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.5495e-01, 1.0865e-01,  ..., 3.6789e-01, 5.5602e-01,
         8.1682e-01],
        [2.5495e-01, 0.0000e+00, 1.6226e-01,  ..., 5.3101e-01, 3.4480e-01,
         5.6194e-01],
        [1.0865e-01, 1.6226e-01, 3.0339e-04,  ..., 3.8624e-01, 4.4742e-01,
         7.1822e-01],
        ...,
        [3.6789e-01, 5.3101e-01, 3.8624e-01,  ..., 0.0000e+00, 6.6671e-01,
         1.0262e+00],
        [5.5602e-01, 3.4480e-01, 4.4742e-01,  ..., 6.6671e-01, 1.0727e-04,
         3.8826e-01],
        [8.1682e-01, 5.6194e-01, 7.1822e-01,  ..., 1.0262e+00, 3.8826e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.71162681, 11.20100795, 10.91805683, 11.21703654, 11.24434374,
       11.28215429,  6.90115901, 11.38774149, 12.88270018,  8.53014915,
        5.79562691,  5.87132718,  4.96231532, 11.07637531, 13.95472331,
       10.26327895, 10.06023665,  6.4020404 , 10.67322527,  4.78002665,
        5.57341623, 13.96987806,  6.81117802, 11.72802491,  8.859416  ,
       10.74655346,  7.58668144, 13.05148826,  9.64157461,  9.19837792,
        7.24326992, 11.36089355, 11.22993218, 11.62770332,  5.82345867,
        5.51177501,  6.35029274,  6.65259775,  7.23276582,  5.79562691,
       13.93790235, 13.09737053, 13.60275211, 10.27716434,  8.09704052,
        9.63742669,  9.48337063,  7.50917551, 10.7506423 ,  9.3109884 ,
       10.27716434, 12.35625694,  5.13125513,  8.27853076,  7.02326515,
        5.14124612,  6.35029274,  9.48969978,  6.77777606,  9.73579738,
        6.7593917 , 13.26590071,  6.30647746,  7.54127425, 10.33958639,
        9.26455795,  9.12220271,  5.2287058 ,  8.08320235,  8.06517143,
        8.1303945 , 12.26997165, 12.30695098,  7.83486943,  8.8353674 ,
        9.12220271, 10.00802627, 11.73567759,  9.75075273, 13.77604293,
        8.16714836,  7.27340834, 11.00726483,  9.96152788,  5.84450127,
       11.99633819,  5.07306274,  5.87305005, 11.018084  ,  4.89869205,
        9.71392179, 13.59673406, 10.06286074, 13.45642379, 13.14777433,
        7.89745107,  9.75200848, 11.63724792,  5.39450305,  7.70897675,
        5.26067019,  8.51928531,  6.35029274,  9.86089579,  8.14433542,
        6.41202986])}
{'P': tensor([[8.2917e-03, 2.4586e-01, 1.3644e-01,  ..., 1.6798e-05, 3.2225e-05,
         7.4964e-07],
        [6.9609e-05, 1.3895e-02, 1.8063e-02,  ..., 5.2952e-05, 9.9749e-05,
         9.1232e-07],
        [1.3297e-04, 8.4140e-03, 1.9856e-02,  ..., 9.2643e-05, 1.8223e-04,
         1.4002e-06],
        ...,
        [2.4890e-05, 2.7784e-05, 3.1189e-05,  ..., 1.9393e-01, 5.7220e-03,
         6.5684e-04],
        [8.8241e-06, 1.1572e-05, 1.3291e-05,  ..., 1.9972e-01, 5.6925e-02,
         9.6913e-05],
        [5.5934e-04, 3.1814e-04, 3.6856e-04,  ..., 2.4135e-05, 4.1857e-05,
         3.4208e-02]]), 'X': tensor([[ -2.9497, -10.3028,  -7.3479],
        [ -0.8455, -10.8853,  -6.7400],
        [ -0.7599, -10.8995,  -6.0366],
        [ -1.0482, -10.8487,  -5.9066],
        [  2.2849, -10.9655,  -7.6542],
        [  2.4639, -10.9431,  -7.3718],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.0582,  -2.6640,   0.7132],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -2.0445,   7.8469,  -0.5857],
        [ -3.1671,   7.6158,   6.0765],
        [ -9.3419,   0.5972,  -1.2841],
        [  2.3589,  13.7668,  -3.6501],
        [ -5.9225,  -8.5355,   2.4778],
        [ -8.8934,   2.2597,   3.5831],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.3900,   3.3727,   3.1650],
        [  2.7558,  -4.4844,   0.0636],
        [ 11.3691,   6.4760,   1.0787],
        [  5.4179,  -2.2925,  -3.4641],
        [  6.2630,   2.1057,  -5.2136],
        [  9.4925,   9.3811,   8.9842],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -4.0409,   7.3169,  -3.5779],
        [  6.0073,   3.0986,   6.7937],
        [ 12.2328,  -2.0376,  -1.1234],
        [  1.3767, -11.0341,   9.9415],
        [ 11.2571,   6.7072,   8.3693],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.9809,  -4.8145,   6.8706],
        [  6.2675,  12.1379,  -8.9720],
        [  5.5101,  -2.1433,   9.5442],
        [ 10.7566,   7.6212,  -2.3279],
        [  1.3339,   7.5627,  -0.9034],
        [  5.8275,  -1.5334,   2.0124],
        [  6.9433,  -9.3362,  -0.3421],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.2849, -10.9655,  -7.6542],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.1459,   1.4780,  -4.2403],
        [  7.3592,  11.3937,   6.2407],
        [  6.3413,   0.4883,  -6.8416],
        [ 12.4592,  -0.9168,  -6.7571],
        [  3.2809,  13.5061,  -4.0199],
        [  7.0358,  -9.2784,  -9.8479],
        [ -1.5875,   7.8944,  -9.4286],
        [ -9.4283,  -1.9585,  -0.9129],
        [  0.6757, -11.0361,  -1.3047],
        [ -7.9667,   4.0814,  -0.6397],
        [  7.1697,  11.5343,  -0.7276],
        [ -9.1706,   1.3863,   8.4032],
        [ 10.9599,  -5.1849,  -3.5526],
        [  1.6232,  -4.7717,  -2.5655],
        [  2.4932,   7.0903,  -5.4214],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.9942,  -3.9990,   6.0448],
        [ -3.3812, -10.1235,   7.4384],
        [  5.4336,  -2.2676,   5.6933],
        [ 12.5770,   1.8512,  -0.6578],
        [ -9.4630,  -1.5145,   0.2983],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.6469,   6.3624,  -1.8414],
        [  9.3607,   9.5337,  -4.5412],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.7802,  -1.6363,  -6.2782],
        [  6.3551,   0.7149,   1.3344],
        [ -9.4716,  -0.6673,   7.6068],
        [  3.9663,   6.1018,  -5.4104],
        [ -0.9715, -10.8630,   0.3555],
        [ -7.6650,   4.5092,   0.2520],
        [  5.4179,  -2.2925,  -3.4641],
        [ -2.3317,   7.8034,  -3.6274],
        [  2.2860,   7.1916,  -6.1011],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.4697,  -4.5836,   1.4593],
        [  6.3606,   0.9204,   3.2525],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.9655,   3.2172,  -7.5953],
        [ -8.0330,  -6.0765,   6.7945],
        [ -0.8455, -10.8853,  -6.7400],
        [  3.3094,  13.4969,  -3.6507],
        [  2.7558,  -4.4844,   0.0636],
        [  2.3261,  13.7749,   5.9121],
        [  8.5093,  -8.1764,  -3.2381],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.5101,  -2.1433,   9.5442],
        [  3.2692,  13.5098,  -4.9032],
        [ -7.6650,   4.5092,   0.2520],
        [ -5.8389,  -8.6056,  -3.5098],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.3550,   0.7129,   0.4917],
        [  5.3860,  -2.3418,   1.0826],
        [  6.0345,   3.0175,  -1.4424],
        [  5.5626,  -2.0536,   8.4433],
        [ -1.9600,   7.8577,   6.9121],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -5.0550,   6.8231,  -9.2502],
        [ -1.8788, -10.6567,   3.7068],
        [ -7.0120,   5.2825,   7.7762],
        [  6.2905,   1.9366,   2.1169],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.0604,   1.7707,   1.1066]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.0151e-02, 9.6677e-02,  ..., 1.0413e+00, 1.0882e+00,
         5.9100e-01],
        [9.0151e-02, 3.0339e-04, 2.9847e-02,  ..., 9.6590e-01, 1.0148e+00,
         5.8869e-01],
        [9.6677e-02, 2.9847e-02, 0.0000e+00,  ..., 9.4777e-01, 9.9570e-01,
         5.5904e-01],
        ...,
        [1.0413e+00, 9.6590e-01, 9.4777e-01,  ..., 0.0000e+00, 6.2750e-02,
         8.2572e-01],
        [1.0882e+00, 1.0148e+00, 9.9570e-01,  ..., 6.2750e-02, 1.5170e-04,
         8.4172e-01],
        [5.9100e-01, 5.8869e-01, 5.5904e-01,  ..., 8.2572e-01, 8.4172e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.7167285 , 10.91805683, 10.92597196, 10.8992524 , 11.20100795,
       11.21703654, 11.62770332, 12.3489324 , 10.06407839,  8.10886021,
        8.24808832,  9.36093562, 13.96746816, 10.38897563,  9.17595899,
       13.27994304,  9.04255655,  5.26342203, 13.08414689,  5.8828986 ,
        6.60752373, 13.34587027,  9.79039371,  8.35856487,  6.7593917 ,
       12.4013153 , 11.11969815, 13.103709  , 12.53047489, 13.00380624,
        4.91336994, 13.66051884,  5.91222346, 13.18278606,  7.67939546,
        6.02588517, 11.63502679,  5.41348121, 11.20100795,  8.09899874,
        9.26455795, 13.56367857,  6.36004393, 12.49291768, 13.89886469,
       11.64436684,  8.05242591,  9.62958973, 11.05672136,  8.95133961,
       13.58101794,  9.27474386, 12.12448912,  5.04028538,  7.5158489 ,
       10.33712258,  9.84314712, 10.67322527,  5.88782913, 12.7125079 ,
        9.58347456,  9.14044984,  7.33350642, 13.36092624, 13.27717982,
        6.00732193,  6.39520397,  9.49511524,  7.27759233, 10.90638066,
        8.89301516,  5.8828986 ,  8.14433542,  7.54620849,  8.01009083,
        5.24461068,  5.20659892,  6.42689705, 12.77402351,  6.77777606,
       10.07238463, 10.91805683, 13.89671314,  5.26342203, 13.96987806,
       11.80092578, 13.00380624,  5.91222346, 13.89974693,  8.89301516,
       10.39942497, 13.77222752,  9.76896904,  6.3949043 ,  5.87305005,
        6.74686071,  5.92952538,  8.09842704, 11.97238624,  9.86089579,
        8.49161154, 10.82106505,  8.77913428,  6.58184474,  6.19707944,
        9.23178204])}
{'P': tensor([[9.1217e-03, 1.9369e-01, 1.9223e-01,  ..., 9.1889e-06, 6.1925e-05,
         2.2860e-06],
        [5.3854e-05, 2.0318e-02, 6.1057e-03,  ..., 7.5164e-05, 5.2187e-04,
         2.2615e-06],
        [3.8174e-05, 4.2558e-02, 1.4091e-02,  ..., 5.0559e-05, 3.7203e-04,
         1.5865e-06],
        ...,
        [3.6079e-07, 3.6863e-07, 3.6465e-07,  ..., 3.0699e-02, 2.9498e-06,
         1.5178e-06],
        [4.6694e-07, 8.7090e-07, 8.7722e-07,  ..., 2.0444e-04, 2.5117e-02,
         9.0718e-07],
        [1.3757e-02, 3.6570e-03, 3.7668e-03,  ..., 1.5712e-06, 1.4660e-05,
         6.4329e-02]]), 'X': tensor([[ -2.6735, -10.4062,  -1.5911],
        [  0.6757, -11.0361,  -1.3047],
        [  0.6470, -11.0352,  -1.1931],
        [  0.5600, -11.0321,  -0.8660],
        [  0.8940, -11.0402,  -0.9783],
        [  0.2480, -11.0153,  -3.0183],
        [  9.0004,  -7.7195,   4.6945],
        [ -6.6998,   5.5954,   0.0823],
        [  4.7482, -10.3907,   5.9149],
        [  9.5018,  -7.1945,  -1.1892],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 12.5258,  -0.4200,  -9.3493],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.6523,   3.9443,  -2.9449],
        [  4.2339,  -3.6268,   6.3125],
        [ -8.8745,   2.3097,   3.5600],
        [  7.1441,  11.5529,   2.3232],
        [  6.3542,   0.6934,  -3.3840],
        [ -9.3419,   0.5972,  -1.2841],
        [  9.6614,   9.1784,  -1.2087],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.6672,  -1.8631,   8.1020],
        [  5.9655,   3.2172,  -7.5953],
        [ -6.1914,  -8.2997,   8.5380],
        [  4.0421,  -3.7765,  -8.3631],
        [ -8.0829,  -5.9940,   5.0564],
        [  3.6469,   6.3624,  -1.8414],
        [  0.1532,  -4.7422,  -7.0145],
        [  2.7590,  13.6621,   9.4496],
        [  4.5430,   5.5438,  -3.8669],
        [ -6.6998,   5.5954,   0.0823],
        [  0.8700,  -4.8132,  -7.8042],
        [  1.9241,   7.3503,   7.9004],
        [ -1.9600,   7.8577,   6.9121],
        [  7.1441,  11.5529,   2.3232],
        [ 11.5568,   6.0602,   9.3244],
        [  2.3376,  -4.6231,   9.8436],
        [  5.1292,   4.8183,  -5.7997],
        [ 11.1420,   6.9331,  -2.4779],
        [ -5.6169,  -8.7844,   8.9057],
        [  8.5198,  10.4091,  -8.6486],
        [  4.9957,  -2.8705,   6.5978],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.5653,  13.9377,   4.1773],
        [  6.3360,   0.4258,  -9.2509],
        [  5.1793,   4.7461,   0.3899],
        [ -9.0961,   1.6529,   3.1553],
        [  1.2273,  -4.8081,   9.8129],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -5.5525,   6.5146,  -4.4984],
        [  1.7000,  -4.7609,   2.7935],
        [  6.2555,   2.1474,   4.1685],
        [  4.0078,  -3.8018,  -6.2168],
        [ -7.4808,   4.7462,   2.0266],
        [  6.3497,   1.3680,  -1.9076],
        [  5.1292,   4.8183,  -5.7997],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.3691,   6.4760,   1.0787],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.0604,   1.7707,   1.1066],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.3419,   0.5972,  -1.2841],
        [ -2.5056,   7.7720,   5.5878],
        [ -7.9298,   4.1367,  -3.9773],
        [  3.4604,  -4.1545,   1.8091],
        [ -8.1204,  -5.9308,   4.0761],
        [ 12.5996,   1.3971,  -1.0694],
        [  3.2647, -10.8066,  -0.3768],
        [  5.5101,  -2.1433,   9.5442],
        [ 10.5955,   7.8824,  -3.7006],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.2508,   1.0572,   2.9978],
        [  0.3715,  -4.7756,   0.2312],
        [ -5.9872,   6.2037,  -9.4477],
        [ -0.1963,   7.8765,  -1.1371],
        [ -9.3444,   0.5821,   6.9101],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.8325,  -1.5222,  -2.1341],
        [  0.7958,  -4.8109,  -8.2248],
        [  9.0129,  -7.7072,   9.1400],
        [ -8.6681,  -4.8556,   9.2518],
        [ 12.6035,   0.7494,   7.1536],
        [ -7.5720,   4.6310,   2.1717],
        [  2.2849, -10.9655,  -7.6542],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.1907,  -3.3180,  -5.5943],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.3413,   0.4883,  -6.8416],
        [  2.8602,   6.8905,   5.8779],
        [  9.4925,   9.3811,   8.9842],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.5408,   7.8978,   5.3470],
        [ -0.5385,   7.9037,   6.3830],
        [  8.8474,  10.0868,  -8.8644],
        [  1.5662,  13.9375,   4.5142],
        [ -9.3705,   0.4208,  -3.5815],
        [  0.5175,  14.0910,   4.6540],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.6363, -11.0348,   6.1414],
        [  5.1142,  -2.7230,  -4.1966],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.9641,  -2.9568,   7.7211],
        [ -9.4501,  -1.7133,   2.2980]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 1.4774e-01, 1.3622e-01,  ..., 1.0858e+00, 1.1191e+00,
         3.4932e-01],
        [1.4774e-01, 0.0000e+00, 1.6066e-02,  ..., 9.5541e-01, 1.0142e+00,
         4.1285e-01],
        [1.3622e-01, 1.6066e-02, 0.0000e+00,  ..., 9.7114e-01, 1.0144e+00,
         3.9721e-01],
        ...,
        [1.0858e+00, 9.5541e-01, 9.7114e-01,  ..., 0.0000e+00, 1.2354e+00,
         1.3414e+00],
        [1.1191e+00, 1.0142e+00, 1.0144e+00,  ..., 1.2354e+00, 0.0000e+00,
         1.0092e+00],
        [3.4932e-01, 4.1285e-01, 3.9721e-01,  ..., 1.3414e+00, 1.0092e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.74409834, 11.05672136, 11.05413545, 11.04629471, 11.07637531,
       11.018084  , 11.85743425,  8.72896381, 11.42420508, 11.9182956 ,
        9.51074763, 12.53285633, 11.81423161, 12.16901615,  6.89245352,
        5.5748624 ,  9.17016857, 13.58333857,  6.39187285,  9.36093562,
       13.32613445, 10.24557047,  5.96556248,  6.77777606, 10.35464916,
        5.53175275, 10.06286074,  7.33350642,  4.74468406, 13.93790235,
        7.16747878,  8.72896381,  4.89121805,  7.59795445,  8.09842704,
       13.58333857, 13.0493368 ,  5.1805345 ,  7.03733775, 13.12299145,
       10.42667093, 13.45124917,  5.76165934, 13.12299145, 14.02532758,
        6.35029274,  7.02496223,  9.24501829,  4.96231532, 10.27572673,
        9.47369628,  8.55982105,  5.05534609,  6.61386436,  5.52416368,
        8.859416  ,  6.49538841,  7.03733775, 12.19805394, 13.08414689,
       11.59512194,  9.23178204,  6.61211224,  9.36093562,  8.16584537,
        8.94395628,  5.40689332, 10.05558893, 12.67680439, 11.28895489,
        5.91222346, 13.20597932,  4.78002665,  9.3109884 ,  4.79003361,
        8.62162372,  7.8788965 ,  9.36256096, 10.78533838,  6.02789634,
        4.8763135 , 11.85890664,  9.9353921 , 12.62576026,  8.87587457,
       11.20100795, 11.72802491,  9.7712321 , 12.11735334,  6.36004393,
        7.46053281, 13.34587027, 12.12448912,  9.53015376,  8.04664886,
        7.92200342, 13.41713661, 14.02526505,  9.37989631, 14.10045797,
       12.30695098, 11.05317597,  5.79391825, 12.91143491, 12.32408516,
        9.60412735])}
{'P': tensor([[4.7894e-03, 3.0097e-01, 1.9032e-01,  ..., 1.6124e-03, 5.2118e-06,
         1.7164e-06],
        [8.1286e-05, 8.0354e-03, 2.1990e-01,  ..., 2.6097e-03, 6.8939e-06,
         2.4821e-06],
        [5.2912e-05, 1.0865e-04, 7.1572e-03,  ..., 3.2901e-03, 4.3921e-06,
         1.7484e-06],
        ...,
        [1.3994e-05, 1.7592e-05, 2.8402e-05,  ..., 1.6855e-02, 2.1888e-06,
         9.1199e-07],
        [2.8022e-02, 2.4025e-02, 2.5926e-02,  ..., 8.1945e-03, 4.6499e-02,
         3.1356e-04],
        [1.8838e-02, 1.4867e-02, 1.0137e-02,  ..., 4.0458e-04, 1.8569e-02,
         2.9388e-02]]), 'X': tensor([[ -2.6485, -10.4151,   0.8766],
        [ -2.1533, -10.5778,   1.0112],
        [ -1.3178, -10.7939,   0.4236],
        [ -0.9715, -10.8630,   0.3555],
        [  0.1382, -11.0073,   1.0026],
        [ -1.1885, -10.8211,   0.0865],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.1813,  -3.3550,   1.8827],
        [ -5.7347,  -8.6908,   9.2878],
        [ -8.1905,   3.7253,   4.1175],
        [  5.0233,   4.9647,  -3.1788],
        [  5.9193,  -1.3176,   5.1739],
        [ -5.1369,   6.7755,   8.3836],
        [ -8.6426,  -4.9136,   9.4941],
        [  1.5653,  13.9377,   4.1773],
        [ -0.7599, -10.8995,  -6.0366],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.7145,  -4.8071,  -2.3766],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.9284,  11.7061,  -0.3732],
        [  1.4489,   7.5256,   4.4529],
        [ -8.9788,   2.0208,  -3.2867],
        [  4.2339,  -3.6268,   6.3125],
        [  5.9026,   3.3837,  -0.6573],
        [ -7.0120,   5.2825,   7.7762],
        [  4.0078,  -3.8018,  -6.2168],
        [  7.5930,  -8.9037,   7.2323],
        [  0.3038,   7.8092,   0.4164],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -2.5161,   7.7699,  -9.3531],
        [  0.9478,  -4.8144,   0.6371],
        [ -2.6069, -10.4298,  -6.1995],
        [ -8.4887,  -5.2448,  -0.6585],
        [  3.2990,  -4.2415,  -7.4857],
        [  5.8902,  -1.3887,   4.0388],
        [  4.8674,  12.8875,   6.7622],
        [ -4.8751,  -9.3148,   4.0406],
        [  8.5198,  10.4091,  -8.6486],
        [  6.3031,   1.8477,  -7.1706],
        [  7.0515,  11.6194,   8.4289],
        [ -8.6603,   2.8250,   5.9181],
        [  3.2366,  -4.2732,   5.1532],
        [ 11.2517,  -4.6495,   2.9800],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -7.0120,   5.2825,   7.7762],
        [ -6.1899,  -8.3011,  -7.9269],
        [  3.4140,  -4.1803,  -8.8224],
        [ -7.5401,   4.6718,   9.0033],
        [  6.2555,   2.1474,   4.1685],
        [  5.5424,  -2.0885,   7.2487],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2883,   0.0325,   5.2692],
        [  7.0358,  -9.2784,  -9.8479],
        [ -5.1398,  -9.1367,   6.2074],
        [ 11.3800,  -4.3919,   1.4066],
        [  0.6952,  -4.8060,   8.8976],
        [  0.7275,  14.0667,  -0.9815],
        [  5.3924, -10.1396,  -4.0863],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.5456,   6.0861,   8.1101],
        [  4.9957,  -2.8705,   6.5978],
        [  1.3339,   7.5627,  -0.9034],
        [  4.5800,   5.5034,  -3.9432],
        [  0.5600, -11.0321,  -0.8660],
        [  5.1886,  -2.6250,   6.4677],
        [  2.3373,  -4.6232,  -5.2246],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.9515,  -3.8425,   6.1729],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.1460,   7.8340,  -9.4614],
        [ -7.0948,   5.1940,   9.8171],
        [  2.0039,   7.3173,  -1.5885],
        [  5.9843,  -9.8679,  -6.1449],
        [ -9.4501,  -1.7133,   2.2980],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.0190,   4.9705,  -7.9752],
        [ -8.8934,   2.2597,   3.5831],
        [  8.8474,  10.0868,  -8.8644],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 11.5198,  -4.0923,   5.5088],
        [ -7.0120,   5.2825,   7.7762],
        [ -1.5408,   7.8978,   5.3470],
        [ -4.0633,  -9.7927,  -3.0825],
        [  1.8447,  -4.7372,   8.1235],
        [  7.1441,  11.5529,   2.3232],
        [ -1.6220,   7.8917,   0.0315],
        [  3.4604,  -4.1545,   1.8091],
        [ -2.9497, -10.3028,  -7.3479],
        [  8.9284,  10.0036,  -6.9210],
        [  5.9587,  12.3216,   9.1695],
        [ -3.5029,   7.5138,   8.1853],
        [ 11.5083,  -4.1177,   0.2389],
        [ -8.0611,  -6.0302,   4.6962],
        [ 12.1379,   4.4125,  -3.8833],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.1206,  -4.7363,   8.7103],
        [  7.8425,  -8.7202,   6.4530],
        [  9.3753,   9.5171,   3.9131],
        [  3.9591, -10.6391,  -6.2557],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -9.2687,  -2.9794,   0.6436]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0311, 0.0711,  ..., 0.6534, 0.4645, 0.1061],
        [0.0311, 0.0000, 0.0580,  ..., 0.6454, 0.4840, 0.1366],
        [0.0711, 0.0580, 0.0000,  ..., 0.5875, 0.4505, 0.1665],
        ...,
        [0.6534, 0.6454, 0.5875,  ..., 0.0000, 0.5312, 0.7101],
        [0.4645, 0.4840, 0.4505,  ..., 0.5312, 0.0000, 0.4304],
        [0.1061, 0.1366, 0.1665,  ..., 0.7101, 0.4304, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.74655346, 10.794755  , 10.87409005, 10.90638066, 11.00812996,
       10.88618114,  9.78147542,  9.85384623,  6.10774544,  9.7751189 ,
       10.412299  ,  8.99791316,  7.06272256,  6.06416957,  8.50268322,
        9.94173708, 14.02532758, 10.92597196,  7.8575648 ,  4.85995002,
        8.89458273, 13.60275211,  7.66377554,  9.20340031,  5.5748624 ,
        6.80370254,  8.77913428,  5.52416368, 11.70168755,  7.81509631,
        9.61942445,  8.16714836,  4.90676242, 10.7506423 ,  9.97822473,
        5.37343173,  6.05165248, 13.77604293, 10.51340829, 13.45124917,
        6.56833864, 13.59170809,  9.10945678,  5.36061974, 12.17451685,
       10.06407839,  8.77913428, 10.35485331,  5.39723241,  8.87007495,
        6.61386436,  5.92281142,  6.33075682,  6.28835101, 11.64436684,
       10.48315436, 12.19805394,  4.85603886, 14.08549802, 11.48434767,
       13.50484539,  9.85330646, 13.05148826,  5.76165934,  7.67939546,
        7.15990676, 11.04629471,  5.81481044,  5.18047638,  7.8788965 ,
       12.59969936,  9.9353921 ,  5.51177501,  8.62631801,  7.83534987,
        8.79284466,  7.58668144, 11.54071859,  9.60412735,  5.16077764,
        7.06372172,  9.17595899, 13.41713661, 12.25806559, 12.22503754,
        8.77913428,  8.04664886, 10.60226614,  5.08372149, 13.58333857,
        8.05668609,  5.40689332, 10.7167285 , 13.40850328, 13.68672855,
        8.29022197, 12.22276278, 10.06703358, 12.91505111,  5.35774573,
        4.73784222, 11.72802491, 13.35927527, 11.35182589, 10.09979747,
        9.73579738])}
{'P': tensor([[6.0534e-03, 2.5977e-01, 2.4521e-01,  ..., 3.4635e-05, 1.9520e-04,
         3.6151e-05],
        [4.1437e-05, 7.0693e-03, 4.0295e-01,  ..., 1.6457e-05, 3.5723e-04,
         6.5415e-05],
        [5.7621e-05, 1.5636e-04, 1.1669e-02,  ..., 2.3049e-05, 6.2920e-04,
         1.1502e-04],
        ...,
        [2.0490e-01, 1.4031e-01, 1.3967e-01,  ..., 2.5160e-02, 2.9195e-04,
         5.8066e-05],
        [4.7114e-05, 5.3528e-05, 5.4693e-05,  ..., 3.7509e-05, 9.7669e-02,
         2.8013e-01],
        [3.2754e-05, 3.2364e-05, 3.2118e-05,  ..., 3.2745e-05, 2.5607e-03,
         1.2184e-01]]), 'X': tensor([[ -2.6069, -10.4298,  -6.1995],
        [ -1.0482, -10.8487,  -5.9066],
        [ -0.7599, -10.8995,  -6.0366],
        [ -0.8455, -10.8853,  -6.7400],
        [  1.5383, -11.0274,  -4.9666],
        [  2.4639, -10.9431,  -7.3718],
        [ -4.7389,   6.9953,   3.7638],
        [ -5.9225,  -8.5355,   2.4778],
        [  3.2159, -10.8166,  -1.0406],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.4283,  -1.9585,  -0.9129],
        [  9.9058,  -6.7203,   5.6928],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -2.6735, -10.4062,  -1.5911],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.5424,  -2.0885,   7.2487],
        [  6.2905,   1.9366,   2.1169],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.5607,  -5.0938,  -5.4531],
        [  6.0073,   3.0986,   6.7937],
        [ 11.3800,  -4.3919,   1.4066],
        [ -7.3470,  -7.0615,   5.0066],
        [  4.0591,   6.0202,  -1.6339],
        [  1.6408,   7.4591,  -4.2294],
        [  7.8982,  -8.6778,   0.0239],
        [  2.7420,  -4.4896,  -3.7139],
        [  7.5930,  -8.9037,   7.2323],
        [  1.4489,   7.5256,   4.4529],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 11.0438,   7.1180,  -5.6991],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.4731,  -0.7120,  -6.0271],
        [  6.8704,  -9.3809,   7.5769],
        [  6.1575,  12.2046,   3.3257],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 11.2571,   6.7072,   8.3693],
        [ 11.7726,  -3.4876,  -6.7402],
        [  6.3301,   0.3638,  -7.5685],
        [  3.6817,   6.3354,   6.1233],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.8704,  -9.3809,   7.5769],
        [  7.0515,  11.6194,   8.4289],
        [ 11.6165,  -3.8715,   6.4594],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.6248,   3.9984,   8.4933],
        [ -2.0229, -10.6163,   8.6930],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 10.6283,  -5.7253,   7.4878],
        [ 12.5763,   0.1333,  -9.2650],
        [ 11.2517,  -4.6495,   2.9800],
        [ -6.9838,  -7.4965,   0.9796],
        [  1.1644, -11.0394,   7.8511],
        [  5.1787,  -2.6383,  -4.2533],
        [ -2.6485, -10.4151,   0.8766],
        [  5.8902,  -1.3887,   4.0388],
        [  9.0652,   9.8597,  -3.5949],
        [  2.7685, -10.8982,  -6.8928],
        [ 11.1085,  -4.9204,   0.0736],
        [ 12.2328,  -2.0376,  -1.1234],
        [  3.9022,  13.2906,   5.5788],
        [ -0.0254, -10.9932,   3.2200],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.3384,  -2.6161,  -4.7630],
        [  3.7237,  -3.9966,  -5.0376],
        [ -5.3179,  -9.0101,   9.4472],
        [ -1.3178, -10.7939,   0.4236],
        [ -7.8280,   4.2848,  -7.1955],
        [  6.3535,   0.6792,  -7.5101],
        [  6.3497,   1.3680,  -1.9076],
        [ -4.9217,   6.8979,  -1.7667],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.7000,  -4.7609,   2.7935],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.1793,   4.7461,   0.3899],
        [  2.0243,   7.3086,   7.5432],
        [  1.5383, -11.0274,  -4.9666],
        [  2.7558,  -4.4844,   0.0636],
        [ 12.2435,   4.0076,   8.1257],
        [ -8.3900,   3.3727,   3.1650],
        [ -6.5295,  -7.9785,  -7.4429],
        [  6.2535,  -0.1709,  -9.4102],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.3589,  13.7668,  -3.6501],
        [ -1.8788, -10.6567,   3.7068],
        [  6.0140,  -1.0661,  -2.5040],
        [ -2.0445,   7.8469,  -0.5857],
        [  1.1350,  14.0104,   0.6321],
        [  9.2811,  -7.4336,   5.1771],
        [  7.0358,  -9.2784,  -9.8479],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.3367,   7.1676,  -9.5263],
        [ -7.4807,   4.7463,   2.0616],
        [  4.4039,   5.6903,   8.7366],
        [  4.7156,  -3.1831,  -8.0224],
        [  0.9788,  -4.8145,  -3.8661],
        [ -5.9225,  -8.5355,   2.4778],
        [ -5.6651,   6.4380,   1.3660],
        [  5.1292,   4.8183,  -5.7997],
        [ -8.7544,   2.6096,   3.3146],
        [ 11.6165,  -3.8715,   6.4594],
        [  5.8930,  12.3592,  -8.9201],
        [  2.1379,  -4.6757,   6.4981],
        [  9.2811,  -7.4336,   5.1771],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.2038,   2.4020,  -3.1692]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0611, 0.0916,  ..., 0.1688, 0.8171, 0.8258],
        [0.0611, 0.0000, 0.0313,  ..., 0.2255, 0.7569, 0.7663],
        [0.0916, 0.0313, 0.0000,  ..., 0.2520, 0.7289, 0.7391],
        ...,
        [0.1688, 0.2255, 0.2520,  ..., 0.0000, 0.9805, 0.9911],
        [0.8171, 0.7569, 0.7289,  ..., 0.9805, 0.0000, 0.0462],
        [0.8258, 0.7663, 0.7391,  ..., 0.9911, 0.0462, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.7506423 , 10.8992524 , 10.92597196, 10.91805683, 11.1341798 ,
       11.21703654,  8.44940661, 10.38897563, 11.2845608 ,  5.24137909,
        9.62958973, 11.9702762 ,  9.60359158, 10.74409834,  9.77777346,
        5.92281142,  6.58184474, 13.27994304,  9.96152788,  6.7593917 ,
       12.19805394, 10.19036081,  7.26074925,  7.6374603 , 11.73397309,
        5.26067019, 11.70168755,  7.66377554,  9.71392179, 13.13888841,
        9.09502385,  9.4997987 , 11.62770332, 13.66991027, 12.35625694,
       13.103709  , 12.27835996,  6.34059021,  7.32754882, 10.32715715,
       11.62770332, 13.59170809, 12.24467041,  8.30125201,  6.90115901,
       10.80728788, 10.54658597, 12.07225469, 12.5769659 , 12.17451685,
       10.24557047, 11.10066126,  5.81199422, 10.74655346,  6.05165248,
       13.39372426, 11.24434374, 12.1494023 , 12.4013153 , 13.85158417,
       10.99326577, 13.79676654,  9.69792513,  5.46244988, 10.46238054,
       10.87409005,  8.92395381,  6.38968357,  6.49538841,  8.47371868,
        9.77588801,  5.05534609,  8.8353674 ,  7.02496223,  7.58378655,
       11.1341798 ,  5.26342203, 12.88270018,  9.04255655, 10.30972193,
        6.25586297, 10.8207297 , 13.96746816, 10.82106505,  6.10774544,
        8.10886021, 14.05633441, 11.89105259, 11.64436684, 12.10942412,
        7.53884104,  8.85939626,  7.19534146,  5.68942572,  4.91296205,
       10.38897563,  8.57560275,  7.03733775,  9.13507321, 12.24467041,
       13.6922425 ,  5.14124612, 11.89105259, 10.46739073,  6.1561646 ,
        6.65259775])}
{'P': tensor([[8.9331e-03, 3.7431e-01, 2.4415e-01,  ..., 3.8666e-04, 7.4355e-08,
         1.4650e-02],
        [2.1582e-04, 1.2522e-02, 3.8624e-01,  ..., 5.6987e-04, 1.1319e-07,
         1.8224e-02],
        [6.0046e-05, 6.9431e-05, 1.8444e-02,  ..., 1.6072e-03, 2.5227e-07,
         8.4775e-04],
        ...,
        [2.5934e-05, 2.6377e-05, 3.6111e-05,  ..., 4.4792e-02, 2.6030e-05,
         2.2201e-05],
        [5.9244e-06, 5.5813e-06, 2.8056e-06,  ..., 8.2787e-06, 3.6344e-01,
         6.1430e-06],
        [8.6655e-04, 1.2754e-03, 3.2955e-02,  ..., 9.2725e-04, 5.4790e-07,
         3.1070e-02]]), 'X': tensor([[-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0321, 0.1347,  ..., 0.8090, 1.4389, 0.2815],
        [0.0321, 0.0000, 0.1095,  ..., 0.7819, 1.4086, 0.2637],
        [0.1347, 0.1095, 0.0000,  ..., 0.6744, 1.3619, 0.3067],
        ...,
        [0.8090, 0.7819, 0.6744,  ..., 0.0000, 1.1217, 0.8356],
        [1.4389, 1.4086, 1.3619,  ..., 1.1217, 0.0000, 1.2177],
        [0.2815, 0.2637, 0.3067,  ..., 0.8356, 1.2177, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.75443106, 10.78533838, 11.018084  , 11.05672136, 11.1913598 ,
       11.05413545,  8.22728399,  6.81084724, 10.7167285 ,  7.46053281,
        6.99097013,  5.92952538,  5.94535529,  4.91336994, 13.31764181,
       11.75807252, 13.12218013,  4.98461983,  8.57560275,  6.35029274,
        9.76896904, 13.40850328,  9.93553424,  7.59795445,  6.99097013,
       12.13568598,  9.51074763, 12.889695  , 10.26327895,  5.80312949,
        6.69389431,  6.00663314, 10.06407839,  7.62232307,  6.16512856,
        5.05963217,  9.75075273,  8.5724042 ,  4.91652556,  6.25586297,
        6.33075682,  5.64500512,  9.93553424, 10.06286074,  6.39187285,
       13.89264822,  9.88310614,  6.69389431,  9.70108409,  6.38105015,
        7.53884104, 10.31613269, 11.67875855,  6.14086582,  5.14371005,
       13.76047999,  6.33075682, 12.12641969,  5.64464807, 13.32557133,
       13.99048094,  7.15100303,  5.27178363,  8.30729352,  4.91296205,
       10.7506423 ,  8.04664886, 12.87752052,  8.79284466, 12.79126785,
       13.58333857, 12.24467041, 13.12218013, 12.17901033,  7.66377554,
       12.7125079 ,  8.05242591,  9.0368458 , 13.88683436,  8.25828409,
       10.74409834,  9.75401449,  5.24137909,  7.17789666,  9.79039371,
        5.80860165,  9.62958973,  8.92098752, 10.38897563, 10.48315436,
       11.04313641, 12.35625694,  4.89869205, 11.04953767, 13.54774355,
        6.24614905, 13.6922425 ,  6.12886024,  5.35774573, 12.04924159,
       10.35485331,  8.16714836,  4.79003361,  6.00732193, 14.0393949 ,
       10.90638066])}
{'P': tensor([[5.3101e-03, 3.3095e-01, 2.4947e-01,  ..., 1.0375e-07, 4.0704e-04,
         9.6634e-04],
        [9.5915e-05, 1.0423e-02, 7.4864e-02,  ..., 2.2698e-07, 9.3906e-04,
         1.9603e-03],
        [7.8247e-05, 7.3181e-04, 8.2538e-03,  ..., 2.0100e-07, 7.9208e-04,
         1.3505e-03],
        ...,
        [9.3453e-06, 7.7883e-06, 7.7295e-06,  ..., 4.8840e-01, 4.3618e-05,
         2.5834e-05],
        [5.8565e-05, 6.2821e-05, 6.4732e-05,  ..., 3.3748e-04, 7.5236e-02,
         1.4771e-03],
        [7.3043e-04, 8.7597e-04, 8.9769e-04,  ..., 4.1137e-05, 6.7000e-02,
         1.3416e-01]]), 'X': tensor([[ -2.4956, -10.4682,   3.5632],
        [ -1.8788, -10.6567,   3.7068],
        [ -1.6275, -10.7221,   3.2376],
        [ -0.0254, -10.9932,   3.2200],
        [  0.5960, -11.0335,   3.1144],
        [  0.1041, -11.0045,   2.4311],
        [ 11.6019,   5.9540,  -5.1715],
        [  2.7558,  -4.4844,   0.0636],
        [  2.2849, -10.9655,  -7.6542],
        [  7.8375,  -8.7240,  -9.8913],
        [  5.0190,   4.9705,  -7.9752],
        [  3.1480,  -4.3165,  -7.8002],
        [ -5.6169,  -8.7844,   8.9057],
        [  1.1994,   7.6035,   2.4295],
        [  5.2608,  -2.5256,  -9.1446],
        [  4.5767,   5.5071,   3.9069],
        [  6.2365,  -0.2578,  -6.0212],
        [ -3.0644,   7.6439,  -1.4322],
        [ -8.8154,   2.4611,   2.5198],
        [  5.9587,  12.3216,   9.1695],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.2592,   3.9425,   2.0383],
        [ 10.9931,   7.2107,   7.7933],
        [  9.6662,   9.1725,   8.1056],
        [  9.8082,  -6.8396,   4.9615],
        [ 11.2847,  -4.5848,  -4.7503],
        [  9.4554,   9.4245,   9.8800],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.7121,   6.3117,   0.6738],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.5636, -10.9293,   9.6519],
        [ -6.8466,  -7.6488,   1.9648],
        [  2.5829,  -4.5466,  -8.4015],
        [ -6.3489,   5.9117,   6.5161],
        [  6.1737,   2.5305,   3.9049],
        [ 12.2592,   3.9425,   2.0383],
        [ -3.7964,  -9.9294,   7.7147],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.5154,   2.5072,  -9.3274],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.2790,  -0.0259,  -5.4116],
        [ -9.1459,   1.4780,  -4.2403],
        [ 10.4758,  -5.9539,   1.3667],
        [  1.1994,   7.6035,   2.4295],
        [  3.5965,  -4.0754,  -3.5035],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -6.3306,   5.9272,   5.6935],
        [  9.0051,   9.9234,   5.4411],
        [  6.3124,   0.2057,  -0.4593],
        [  3.9663,   6.1018,  -5.4104],
        [  6.3550,   0.7129,   0.4917],
        [ -7.4602,   4.7718,  -4.6277],
        [  9.0652,   9.8597,  -3.5949],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.4731,  -0.7120,  -6.0271],
        [  1.6232,  -4.7717,  -2.5655],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.1222,  -0.7261,   8.8204],
        [  3.4009,  -4.1874,  -5.9751],
        [  7.8425,  -8.7202,   6.4530],
        [  4.2339,  -3.6268,   6.3125],
        [  3.2159, -10.8166,  -1.0406],
        [ -5.6792,  -8.7352,   4.0318],
        [  5.1142,  -2.7230,  -4.1966],
        [  3.5939,   6.4029,  -1.8357],
        [  2.8994,   6.8675,  -9.5987],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3589,   0.8239,  -8.4230],
        [ -6.1899,  -8.3011,  -7.9269],
        [  8.2422,  -8.4041,  -1.4962],
        [  6.3119,   1.7797,   9.9186],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.2895,   0.0405,   5.7652],
        [ 10.0567,   8.6697,   4.6329],
        [ -7.7239,   4.4298,   3.1216],
        [  4.5057,  -3.3894,  -0.8160],
        [ 10.4994,  -5.9193,  -7.5153],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.4932,   7.0903,  -5.4214],
        [ -8.3900,   3.3727,   3.1650],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.9284,  11.7061,  -0.3732],
        [  4.7156,  -3.1831,  -8.0224],
        [  8.5716,  10.3596,   6.8818],
        [ 11.7418,   5.6070,  -1.4192],
        [ 11.7349,  -3.5838,   2.1644],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.4716,  -0.6673,   7.6068],
        [ 12.5832,   1.7543,  -7.6134],
        [ -8.1426,   3.8048,   9.3542],
        [ -0.8455, -10.8853,  -6.7400],
        [ 10.6283,  -5.7253,   7.4878],
        [  1.7903,  -4.7467,   0.1168],
        [  2.3261,  13.7749,   5.9121],
        [ -8.9942,  -3.9990,   6.0448],
        [  6.5428,  -9.5728,  -2.9897],
        [  1.3339,   7.5627,  -0.9034],
        [  6.2902,   0.0451,   3.7116],
        [  5.8979,   3.3958,  -3.8713],
        [  3.2809,  13.5061,  -4.0199],
        [  5.5158,  -2.1337,   3.5246],
        [  2.1625,  -4.6697,   7.0658]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0450, 0.0524,  ..., 1.2604, 0.8253, 0.7940],
        [0.0450, 0.0000, 0.0366,  ..., 1.2334, 0.7846, 0.7495],
        [0.0524, 0.0366, 0.0000,  ..., 1.2086, 0.7759, 0.7513],
        ...,
        [1.2604, 1.2334, 1.2086,  ..., 0.0000, 0.6575, 0.8775],
        [0.8253, 0.7846, 0.7759,  ..., 0.6575, 0.0000, 0.2226],
        [0.7940, 0.7495, 0.7513,  ..., 0.8775, 0.2226, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.76154256, 10.82106505, 10.84493732, 10.99326577, 11.04953767,
       11.0050379 , 13.04051744,  5.26342203, 11.20100795, 11.72749923,
        7.06372172,  5.34250225, 10.42667093,  7.69752262,  5.83560009,
        7.16058587,  6.24187044,  8.23525433,  9.15253344, 13.68672855,
       12.86648771, 12.87752052, 13.14690137, 13.32557133, 11.95743558,
       12.18045978, 13.35013374,  5.9510489 ,  7.32233733,  5.24137909,
       11.22596994, 10.26545792,  5.22903117,  8.67502348,  6.67218687,
       12.87752052, 10.63038565, 12.3828957 , 13.13205357, 12.76408562,
       10.4977132 ,  6.27905292,  9.26455795, 12.0495521 ,  7.69752262,
        5.43544626, 12.52372629,  9.97822473,  8.67227386, 13.40024224,
        6.31576264,  7.27759233,  6.3949043 ,  8.85572292, 13.39372426,
        8.94395628,  9.4997987 ,  5.04028538,  6.23262289,  6.16512856,
        5.39450305, 11.72802491,  5.5748624 , 11.2845608 , 10.41909211,
        5.79391825,  7.34250948,  7.4544866 , 11.33837396,  6.41202986,
       10.35485331, 11.77124373,  6.55800267,  9.7751189 ,  6.28961988,
       13.2778378 ,  8.90405823,  5.63824971, 12.05300943,  5.05963217,
        7.5158489 ,  9.04255655, 10.37698435, 10.30972193, 12.40074901,
       13.60275211,  5.68942572, 13.44593519, 13.01187484, 12.26997165,
        5.1224079 ,  9.49511524, 12.70489563,  8.98765644, 10.91805683,
       12.07225469,  5.07306274, 13.96987806,  9.84314712, 11.59512194,
        7.67939546,  6.29034958,  6.80559669, 13.89886469,  5.91408504,
        5.14607014])}
{'P': tensor([[8.9544e-03, 2.7620e-01, 1.3399e-01,  ..., 1.4581e-04, 5.3874e-06,
         5.4946e-07],
        [4.0348e-05, 1.0718e-02, 1.5639e-02,  ..., 9.2988e-05, 2.8317e-06,
         1.3231e-06],
        [5.7567e-05, 2.1254e-03, 1.3706e-02,  ..., 2.3781e-04, 3.2878e-06,
         2.7150e-06],
        ...,
        [4.3565e-05, 1.1251e-04, 2.4744e-04,  ..., 6.3844e-02, 4.7682e-06,
         5.6897e-05],
        [4.7823e-02, 1.6104e-02, 1.7226e-02,  ..., 7.9647e-04, 3.9652e-02,
         6.7236e-07],
        [1.2424e-07, 8.9694e-08, 1.1044e-07,  ..., 1.7355e-07, 1.3351e-06,
         1.9263e-02]]), 'X': tensor([[ -2.2508, -10.5479,  -2.7235],
        [  0.2480, -11.0153,  -3.0183],
        [  0.6757, -11.0361,  -1.3047],
        [  0.6470, -11.0352,  -1.1931],
        [  2.1771, -10.9776,  -3.2007],
        [  0.8940, -11.0402,  -0.9783],
        [  6.1575,  12.2046,   3.3257],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.3474,  -4.2162,   5.2455],
        [  4.6357,   5.4415,   7.9067],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -0.8885,   7.9160,   0.0571],
        [ -6.0844,  -8.3955,   0.7705],
        [ 10.1503,   8.5414,  -7.2681],
        [ -1.6334, -10.7207,   7.1397],
        [  9.3607,   9.5337,  -4.5412],
        [  6.1372,   2.6726,  -6.7885],
        [ -7.6293,  -6.6859,  -5.5161],
        [  1.9312,  13.8649,   8.3563],
        [ -9.1813,  -3.3550,   1.8827],
        [ -6.8618,  -7.6322,   3.7027],
        [  0.9809,   7.6642,  -1.4217],
        [  5.8318,   3.5567,  -4.7692],
        [  5.3118,   4.5452,   4.2114],
        [  2.5010,  -4.5736,  -0.8989],
        [ -2.5683, -10.4433,  -2.9182],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.2508,   1.0572,   2.9978],
        [  4.8051, -10.3703,   0.5268],
        [ -5.2589,   6.7024,  -5.5809],
        [  1.3103,  -4.8032,  -0.9300],
        [  1.1139,   7.6281,  -2.5680],
        [ -7.4808,   4.7462,   2.0266],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.4697,  -0.6157,  -6.8472],
        [  9.0004,  -7.7195,   4.6945],
        [  2.1440, -10.9811,   8.5281],
        [  5.4867,  12.5816,   2.9549],
        [  0.9967,  -4.8144,  -9.7370],
        [ -3.5029,   7.5138,   8.1853],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.3419,   0.5972,  -1.2841],
        [ -7.4808,   4.7462,   2.0266],
        [  5.9193,  -1.3176,   5.1739],
        [  3.5284, -10.7484,   5.3347],
        [  0.3724,   7.7974,  -6.8722],
        [  3.4401,  13.4540,   4.6576],
        [  0.9809,   7.6642,  -1.4217],
        [ -0.9715, -10.8630,   0.3555],
        [  1.5246,  -4.7838,  -5.2095],
        [ -4.9217,   6.8979,  -1.7667],
        [ -4.0409,   7.3169,  -3.5779],
        [ -4.7334,   6.9982,  -0.4392],
        [  3.1892, -10.8220,  -6.8409],
        [  2.5128, -10.9365,   9.9645],
        [  6.3246,   1.6696,  -0.1544],
        [  6.0305,   3.0296,   0.1375],
        [ -5.1974,   6.7396,   3.3484],
        [ -8.9379,  -4.1655,  -2.8114],
        [  4.7544,  -3.1425,  -1.5315],
        [  9.0051,   9.9234,   5.4411],
        [ -2.0721,   7.8432,   6.6380],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.1459,   1.4780,  -4.2403],
        [  6.1989,  -9.7591,   0.8289],
        [  2.7685, -10.8982,  -6.8928],
        [  2.5375,   7.0676,  -4.3490],
        [  6.3587,   1.1596,  -3.6629],
        [  8.3718,  -8.2953,   8.4666],
        [  2.5318,  13.7231,   6.3692],
        [  5.7784,  -1.6401,   9.5587],
        [  8.1217,  -8.5024,  -0.6149],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.9788,   2.0208,  -3.2867],
        [ -1.6906,   7.8860,   7.7142],
        [  3.0719, -10.8449,  -4.1673],
        [ -6.4368,   5.8356,   3.3014],
        [ -4.4503,   7.1378,   3.3751],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -1.9714,   7.8563,  -1.8084],
        [  3.2647, -10.8066,  -0.3768],
        [  2.7121,   6.9743,  -8.4202],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.1231,  -3.5739,   8.4708],
        [  9.6662,   9.1725,   8.1056],
        [ -8.7544,   2.6096,   3.3146],
        [ 12.3357,   3.6001,   0.7903],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -0.5385,   7.9037,   6.3830],
        [ -9.4309,  -0.0599,  -4.0830],
        [  8.6420,  -8.0580,   6.5781],
        [  3.1064,  13.5607,   6.7605],
        [  6.0467,  -9.8369,  -6.5816],
        [ -8.1531,  -5.8748,   2.6934],
        [  2.3589,  13.7668,  -3.6501],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.1287, -11.0065,   6.5757],
        [  3.6154,  -4.0640,  -7.9698],
        [ -2.2508, -10.5479,  -2.7235],
        [ -7.3470,  -7.0615,   5.0066],
        [ -0.5739,   7.9056,   0.6382],
        [ -8.4824,  -5.2575,  -0.0795],
        [  2.5128, -10.9365,   9.9645],
        [ -8.4824,  -5.2575,  -0.0795],
        [  8.7131,  10.2216,   7.7915]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1095, 0.1843,  ..., 0.9649, 0.2962, 1.5199],
        [0.1095, 0.0000, 0.1566,  ..., 0.9737, 0.3921, 1.4673],
        [0.1843, 0.1566, 0.0000,  ..., 0.8175, 0.3445, 1.3356],
        ...,
        [0.9649, 0.9737, 0.8175,  ..., 0.0000, 0.8305, 0.8670],
        [0.2962, 0.3921, 0.3445,  ..., 0.8305, 0.0000, 1.5321],
        [1.5199, 1.4673, 1.3356,  ..., 0.8670, 1.5321, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.78533838, 11.018084  , 11.05672136, 11.05413545, 11.1913598 ,
       11.07637531, 13.66991027, 12.19805394,  5.38342077,  7.14837514,
       12.18045978,  7.96575782, 10.36844813, 13.26590071, 10.84437705,
       13.36092624,  6.69389431, 10.14436912, 13.99877413,  9.7751189 ,
       10.26327895,  7.72668467,  6.83083863,  6.99097013,  5.21280156,
       10.75443106, 11.72802491,  9.3109884 , 11.42947632,  8.51928531,
        4.978704  ,  7.70897675,  8.859416  ,  6.78325151,  9.48969978,
       11.85743425, 11.18839169, 13.72594022,  4.91652556,  8.29022197,
        8.5724042 ,  9.36093562,  8.859416  ,  6.06416957, 11.31276109,
        7.80625981, 13.88683436,  7.72668467, 10.90638066,  5.0209005 ,
        8.47371868,  8.35856487,  8.4486771 , 11.28215429, 11.22142195,
        6.54127431,  6.74872981,  8.51088913,  9.86089579,  5.69913817,
       13.40024224,  8.11227668, 12.30858677,  9.78286733,  9.26455795,
       11.56146873, 11.24434374,  7.5092834 ,  6.46356806, 11.785566  ,
       13.95472331,  6.00663314, 11.75807252,  5.69913817,  9.20340031,
        8.06517143, 11.27160035,  8.68832398,  8.41148722, 12.53285633,
        8.09984231, 11.28895489,  7.48311002,  6.34059021,  9.79814997,
       13.32557133,  9.13507321, 12.85032836,  9.97964013,  7.92200342,
        9.43112957, 11.8159254 , 13.91198191, 11.54672755, 10.04916678,
       13.96746816,  5.5366698 , 11.00726483,  5.43943791, 10.78533838,
       10.19036081,  7.92644746,  9.97964013, 11.22142195,  9.97964013,
       13.43127131])}
{'P': tensor([[6.5628e-03, 2.3073e-01, 2.0700e-01,  ..., 1.0670e-07, 1.9713e-06,
         1.6221e-04],
        [5.3155e-04, 1.3287e-01, 7.8695e-02,  ..., 3.4385e-06, 9.0927e-05,
         2.7207e-03],
        [5.2162e-04, 1.3343e-01, 1.1884e-01,  ..., 2.8708e-06, 7.6934e-05,
         2.0137e-03],
        ...,
        [2.6592e-07, 1.4711e-07, 1.3999e-07,  ..., 1.2034e-02, 1.0552e-06,
         2.0813e-06],
        [1.0759e-07, 8.8551e-08, 8.6323e-08,  ..., 1.6681e-02, 8.7649e-03,
         1.6220e-06],
        [4.6814e-04, 7.6643e-04, 6.6237e-04,  ..., 1.5982e-04, 8.8385e-04,
         1.8094e-01]]), 'X': tensor([[ -2.1533, -10.5778,   1.0112],
        [  0.1382, -11.0073,   1.0026],
        [  0.1047, -11.0046,   0.5188],
        [ -0.9715, -10.8630,   0.3555],
        [ -1.3178, -10.7939,   0.4236],
        [ -1.1885, -10.8211,   0.0865],
        [  7.3592,  11.3937,   6.2407],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.5955,   7.8824,  -3.7006],
        [  0.7275,  14.0667,  -0.9815],
        [  0.1206,  -4.7363,   8.7103],
        [  2.9461,  -4.4077,  -0.4831],
        [ -8.1905,   3.7253,   4.1175],
        [  6.3549,   0.7096,   6.9198],
        [  1.7815,  -4.7481,   4.4289],
        [  6.2804,   2.0019,   0.8607],
        [ -0.4341, -10.9473,   7.4509],
        [  5.2199,   4.6861,  -1.9304],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.3515,   0.6418,  -6.5823],
        [  0.1931,  -4.7491,   7.4385],
        [  2.0243,   7.3086,   7.5432],
        [  9.7328,   9.0902,  -0.8917],
        [  0.9809,  -4.8145,   6.8706],
        [  0.6363, -11.0348,   6.1414],
        [ 10.1798,   8.5002,  -5.8893],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.5650,   2.0132,   4.6272],
        [  5.1203,  -2.7151,   6.2110],
        [  6.9284,  11.7061,  -0.3732],
        [ -3.4488, -10.0934,   2.5195],
        [  2.5318,  13.7231,   6.3692],
        [  1.7000,  -4.7609,   2.7935],
        [ -7.8280,   4.2848,  -7.1955],
        [ -7.9667,   4.0814,  -0.6397],
        [  0.2732,  -4.7619,  -1.1577],
        [  9.6587,   9.1817,  -0.8212],
        [ -7.9925,  -6.1423,   2.8990],
        [ -2.9497, -10.3028,  -7.3479],
        [ -9.1706,   1.3863,   8.4032],
        [ 10.4737,  -5.9570,  -1.2108],
        [  4.1999,  -3.6543,  -4.1998],
        [ 12.5154,   2.5072,  -9.3274],
        [ -9.4773,  -1.0024,   5.2599],
        [ 12.2914,   3.8037,   6.0978],
        [  2.7976,  -4.4682,  -4.2020],
        [  8.1043,  -8.5164,  -7.9290],
        [  8.7131,  10.2216,   7.7915],
        [  3.3167,   6.6022,   6.3554],
        [ 11.7796,  -3.4696,   9.0088],
        [  4.8674,  12.8875,   6.7622],
        [  5.3804,  -2.3504,   7.2744],
        [  5.6523,   3.9443,  -2.9449],
        [ -7.3333,   4.9246,  -7.9212],
        [  9.6816,   9.1536,  -5.8543],
        [  4.7739,  -3.1218,  -4.1188],
        [  1.3705,  13.9723,   7.9266],
        [  6.1973,  -0.4372,   6.5720],
        [  5.4336,  -2.2676,   5.6933],
        [  5.6248,   3.9984,   8.4933],
        [ -8.8363,   2.4086,  -5.6296],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.5428,  -9.5728,  -2.9897],
        [  8.4691,  10.4571,  -5.0650],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 11.5935,   5.9740,  -4.3952],
        [  6.2905,   1.9366,   2.1169],
        [  6.3119,   1.7797,   9.9186],
        [  9.4554,   9.4245,   9.8800],
        [ -4.2040,   7.2486,   2.9875],
        [ 10.6283,  -5.7253,   7.4878],
        [  7.0358,  -9.2784,  -9.8479],
        [  4.3523, -10.5232,   8.2138],
        [  5.9003,  12.3550,   2.1917],
        [  3.1064,  13.5607,   6.7605],
        [ -4.8665,  -9.3204,  -5.2564],
        [  0.3493,  -4.7727,  -6.5682],
        [  6.3301,   0.3638,  -7.5685],
        [  6.7792,  11.8084,   1.6251],
        [  5.6260,  -1.9401,   3.0259],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.9298,   4.1367,  -3.9773],
        [  1.8719,   7.3713,  -3.7465],
        [  6.6319,  11.9066,   3.5165],
        [  5.6314,  -1.9302,  -9.6543],
        [  6.0788,  -0.8710,  -6.8719],
        [ -9.0604,   1.7707,   1.1066],
        [  3.9515,  -3.8425,   6.1729],
        [ -8.8835,  -4.3178,   9.3080],
        [ -3.4100,   7.5436,  -7.6980],
        [  3.4920,  -4.1366,  -8.3955],
        [ 12.5996,   1.3971,  -1.0694],
        [ 11.4743,   6.2478,  -2.1100],
        [ -7.6525,  -6.6534,   5.1721],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -7.8126,   4.3066,   5.1791],
        [ -0.4341, -10.9473,   7.4509],
        [  6.0345,   3.0175,  -1.4424],
        [ -6.6895,   5.6051,   2.7899],
        [  5.1886,  -2.6250,   6.4677],
        [  1.9312,  13.8649,   8.3563],
        [ 10.0567,   8.6697,   4.6329],
        [  1.9505,  -4.7171,   9.9868]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1457, 0.1408,  ..., 1.2912, 1.1886, 0.8993],
        [0.1457, 0.0000, 0.0485,  ..., 1.1738, 1.0566, 0.7803],
        [0.1408, 0.0485, 0.0000,  ..., 1.2133, 1.0897, 0.8200],
        ...,
        [1.2912, 1.1738, 1.2133,  ..., 0.0000, 0.2556, 0.3935],
        [1.1886, 1.0566, 1.0897,  ..., 0.2556, 0.0000, 0.3315],
        [0.8993, 0.7803, 0.8200,  ..., 0.3935, 0.3315, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.794755  , 11.00812996, 11.00508681, 10.90638066, 10.87409005,
       10.88618114, 13.56367857, 13.32374056, 13.20597932, 14.08549802,
        4.73784222,  5.30158497,  8.99791316,  6.39438044,  5.0713364 ,
        6.59176018, 10.95593729,  7.01475376,  6.02789634,  6.383897  ,
        4.75303127,  7.58378655, 13.31764181,  4.91336994, 11.05317597,
       13.26209545,  8.53501777, 12.72524615,  5.79562691, 13.60275211,
       10.66631784, 13.95472331,  5.05534609,  8.92395381,  8.95133961,
        4.76969364, 13.32645552, 10.08001894, 10.7167285 ,  9.27474386,
       12.04924159,  5.5671389 , 12.76408562,  9.53015376, 12.86648771,
        5.27178363, 11.75618513, 13.43127131,  7.38844703, 12.27993098,
       13.77604293,  5.87132718,  6.89245352,  8.83341608, 13.32374056,
        5.70405525, 14.0393949 ,  6.21275016,  5.88782913,  6.90115901,
        9.15866607, 12.10942412, 11.59512194, 13.45642379, 13.15525803,
       12.30858677, 13.04217487,  6.58184474,  6.55800267, 13.35013374,
        8.37954853, 12.07225469, 11.64436684, 11.38774149, 13.69162795,
       13.91198191, 10.51437947,  4.78544875,  6.34059021, 13.61600536,
        5.95111804,  5.79391825,  9.7712321 ,  8.94395628,  7.60529892,
       13.62896593,  5.95297459,  6.14086582,  9.23178204,  5.51177501,
        9.87720102,  8.27853076,  5.41348121, 12.67680439, 13.06498554,
       10.14044489, 10.99326577,  9.69629   ,  8.92098752, 10.95593729,
        6.74686071,  8.7273569 ,  5.81481044, 13.99877413, 13.2778378 ,
        5.10446653])}
{'P': tensor([[9.5462e-03, 1.9040e-01, 1.6106e-01,  ..., 9.6532e-08, 4.3349e-06,
         8.9428e-07],
        [7.6851e-05, 2.5116e-02, 1.4213e-02,  ..., 7.1220e-07, 5.7434e-06,
         7.0776e-06],
        [6.7490e-05, 7.3631e-03, 2.1749e-02,  ..., 3.7451e-07, 2.4045e-06,
         4.9826e-06],
        ...,
        [3.8973e-08, 3.1288e-08, 2.0314e-08,  ..., 1.6699e-02, 3.4034e-07,
         1.1560e-04],
        [1.2629e-04, 3.3579e-04, 1.3679e-04,  ..., 1.2133e-05, 1.6428e-01,
         4.6240e-06],
        [1.0492e-06, 7.3879e-07, 6.1647e-07,  ..., 1.7536e-03, 5.6897e-07,
         8.8360e-02]]), 'X': tensor([[ -2.0229, -10.6163,   8.6930],
        [  1.1644, -11.0394,   7.8511],
        [  1.2728, -11.0372,   9.8121],
        [  2.1440, -10.9811,   8.5281],
        [ -0.4341, -10.9473,   7.4509],
        [  1.3767, -11.0341,   9.9415],
        [  6.3087,   1.8053,  -1.9481],
        [ -1.8367,   7.8718,  -8.6523],
        [  7.7794,  11.0629,   8.9069],
        [  6.3305,   1.6119,   1.1538],
        [  9.1409,  -7.5789,   1.5814],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.4505,  -1.7081,  -0.7039],
        [  6.3119,   1.7797,   9.9186],
        [  5.3118,   4.5452,   4.2114],
        [  4.2041,  -3.6509,  -4.9488],
        [  3.2701,   6.6338,   9.9209],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.6050,   2.9451,  -0.7012],
        [  6.2975,   0.0947,   5.1791],
        [  3.7992,  -3.9474,  -8.9019],
        [  5.9843,  -9.8679,  -6.1449],
        [ -2.0445,   7.8469,  -0.5857],
        [  7.1697,  11.5343,  -0.7276],
        [  6.2365,  -0.2578,  -6.0212],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.2038,   2.4020,  -3.1692],
        [  1.3670,  -4.7990,   3.2603],
        [ 11.7162,   5.6728,  -8.9372],
        [ 11.1469,   6.9237,   1.6160],
        [  1.1139,   7.6281,  -2.5680],
        [ -7.3446,   4.9113,  -5.9356],
        [  1.9241,   7.3503,   7.9004],
        [  0.9074,  -4.8139,   8.1674],
        [ -7.0825,   5.2073,  -5.0497],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.1206,  -4.7363,   8.7103],
        [ -8.0330,  -6.0765,   6.7945],
        [ -3.9700,   7.3453,  -4.1486],
        [  2.8474,   6.8979,   6.6457],
        [  4.9337,  12.8566,  -3.6582],
        [  7.8375,  -8.7240,  -9.8913],
        [ 12.0582,  -2.6640,   0.7132],
        [ -1.8823, -10.6558,  -3.8952],
        [ -7.9667,   4.0814,  -0.6397],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.3419,   0.5972,  -1.2841],
        [ 10.9916,   7.2133,  -0.9179],
        [  2.0449,  13.8402,  -7.1642],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.1297,  -3.7098,  -3.9838],
        [  8.6271,  -8.0715,  -7.4813],
        [  9.4925,   9.3811,   8.9842],
        [ 11.6165,  -3.8715,   6.4594],
        [  5.5424,  -2.0885,   7.2487],
        [  5.3602,  -2.3808,   7.2098],
        [  2.5620,   7.0549,   8.4825],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.9641,  -2.9568,   7.7211],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.4743,   6.2478,  -2.1100],
        [  3.1064,  13.5607,   6.7605],
        [  6.1741,  -0.5329,   3.1913],
        [  3.7121,   6.3117,   0.6738],
        [  2.3261,   7.1726,   6.2522],
        [  6.9433,  -9.3362,  -0.3421],
        [  1.6954,  -4.7616,   9.8198],
        [  4.1140,  -3.7220,  -0.1273],
        [  7.8982,  -8.6778,   0.0239],
        [ -7.9667,   4.0814,  -0.6397],
        [  7.3724,  -9.0577,  -8.9843],
        [  5.9587,  12.3216,   9.1695],
        [ 11.0373,   7.1299,  -8.5895],
        [ -3.9508,  -9.8515,   1.7923],
        [ -6.8466,  -7.6488,   1.9648],
        [  6.0310,  12.2795,  -6.4528],
        [ 11.3118,   6.5957,  -4.5667],
        [ -5.2589,   6.7024,  -5.5809],
        [ 10.0619,   8.6627,  -9.2884],
        [ 12.6038,   0.7665,  -3.4439],
        [  1.3339,   7.5627,  -0.9034],
        [  1.4489,   7.5256,   4.4529],
        [  1.6232,  -4.7717,  -2.5655],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -1.8367,   7.8718,  -8.6523],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.9653,  -9.3225,  -3.4727],
        [  3.9085,   6.1512,  -2.2560],
        [  1.3767, -11.0341,   9.9415],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.8474,   6.8979,   6.6457],
        [  0.9967,  -4.8144,  -9.7370],
        [ 11.6019,   5.9540,  -5.1715],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.6381,  -4.9238,  -4.4077],
        [  3.5284, -10.7484,   5.3347],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.9956,  11.6590,  -7.6876],
        [ -3.0008, -10.2827,  -8.9480],
        [  7.7750,  11.0665,   2.8364]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1073, 0.1758,  ..., 1.5571, 1.1874, 1.0804],
        [0.1073, 0.0000, 0.0832,  ..., 1.4814, 1.1847, 0.9743],
        [0.1758, 0.0832, 0.0000,  ..., 1.4868, 1.2466, 0.9286],
        ...,
        [1.5571, 1.4814, 1.4868,  ..., 0.0000, 1.0488, 0.9251],
        [1.1874, 1.1847, 1.2466,  ..., 1.0488, 0.0000, 1.4281],
        [1.0804, 0.9743, 0.9286,  ..., 0.9251, 1.4281, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.80728788, 11.10066126, 11.11038551, 11.18839169, 10.95593729,
       11.11969815,  6.5618968 ,  8.08320235, 13.52430176,  6.53250937,
       11.87412439, 10.23946247, 13.59673406,  9.60359158,  6.55800267,
        6.99097013,  5.56809335,  7.39599986,  8.87007495,  9.09502385,
        6.29821673,  5.47866221, 11.54071859,  8.10886021, 13.58101794,
        6.24187044,  9.71392179, 11.99633819,  6.65259775,  4.98989906,
       13.01728313, 13.12218013,  7.70897675,  8.8353674 ,  7.59795445,
        4.89869205,  8.7907923 ,  9.94285511,  4.73784222, 10.07238463,
        8.34949383,  7.46248947, 13.77075174, 11.72749923, 12.3489324 ,
       10.8207297 ,  8.95133961,  4.78544875,  9.36093562, 13.1471313 ,
       13.99048094,  9.43737523,  5.16611782,  9.5014146 , 10.00802627,
        5.5513072 , 11.81423161, 13.34587027, 12.24467041,  5.92281142,
        5.86519525,  7.50564223, 11.81423161, 12.32408516,  4.85995002,
       13.06498554, 13.91198191,  6.19707944,  7.32233733,  7.5403822 ,
       11.63502679,  5.05444288,  5.54776934, 11.73397309,  8.95133961,
       11.67875855, 13.68672855, 13.1399181 , 10.61417181, 10.26545792,
       13.68063098, 13.09426142,  8.51928531, 13.27717982, 12.62711252,
        7.67939546,  7.66377554,  5.04028538,  9.43737523,  8.08320235,
        8.85572292, 11.63724792,  7.28793663, 11.11969815,  9.66977802,
        7.9337647 ,  7.46248947,  4.91652556, 13.04051744,  6.14086582,
        9.94285511, 11.31276109,  9.53015376, 13.59673406, 10.71162681,
       13.5247215 ])}
{'P': tensor([[8.4956e-03, 2.1625e-01, 1.4262e-01,  ..., 4.5511e-07, 3.1861e-06,
         9.9788e-06],
        [5.1565e-05, 9.3347e-03, 4.5096e-02,  ..., 6.8080e-07, 1.8874e-06,
         2.6280e-05],
        [2.6408e-05, 2.6643e-04, 9.1942e-03,  ..., 5.0228e-07, 1.3745e-06,
         4.2547e-05],
        ...,
        [1.6677e-05, 1.2227e-05, 6.0473e-06,  ..., 7.2374e-02, 1.8625e-04,
         3.8531e-06],
        [1.3665e-01, 4.8234e-02, 3.0611e-02,  ..., 1.2741e-05, 1.8127e-01,
         1.0691e-05],
        [1.5700e-07, 1.5203e-07, 1.5316e-07,  ..., 9.2050e-05, 8.0693e-07,
         1.1598e-02]]), 'X': tensor([[ -1.8823, -10.6558,  -3.8952],
        [  0.2480, -11.0153,  -3.0183],
        [  1.5383, -11.0274,  -4.9666],
        [  2.1771, -10.9776,  -3.2007],
        [  3.0719, -10.8449,  -4.1673],
        [  3.8111, -10.6787,  -4.6539],
        [  5.2939,  -2.4784,  -4.5840],
        [ -1.6906,   7.8860,   7.7142],
        [ -8.6603,   2.8250,   5.9181],
        [ 11.7796,  -3.4696,   9.0088],
        [  4.5057,  -3.3894,  -0.8160],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.0310,  12.2795,  -6.4528],
        [  8.8091,  -7.9037,   0.3631],
        [  2.1824,  -4.6647,   1.1985],
        [  5.5158,  -2.1337,   3.5246],
        [ -3.9456,  -9.8542,   0.7466],
        [ -0.1626, -10.9796,   6.4629],
        [ 11.7162,   5.6728,  -8.9372],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.7964,  -9.9294,   7.7147],
        [ -5.6174,   6.4708,  -1.7339],
        [  9.6614,   9.1784,  -1.2087],
        [  9.4554,   9.4245,   9.8800],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.8093,  -9.4178,   8.4244],
        [ -6.4464,   5.8272,   7.5038],
        [  2.7420,  -4.4896,  -3.7139],
        [ 12.5919,   0.4035,  -5.8460],
        [ 11.3367,  -4.4806,   0.4643],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.7078,   2.7182,   1.8146],
        [  0.2480, -11.0153,  -3.0183],
        [  9.9756,  -6.6330,  -3.3429],
        [  4.6992,   5.3690,   3.0935],
        [ 12.5996,   1.3971,  -1.0694],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.7121,   6.9743,  -8.4202],
        [  4.4039,   5.6903,   8.7366],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.0611,  -6.0302,   4.6962],
        [  7.0171,  11.6439,  -5.0178],
        [  2.8602,   6.8905,   5.8779],
        [ -7.4733,   4.7555,   6.2490],
        [  0.8940, -11.0402,  -0.9783],
        [  5.4167, -10.1293,   6.2156],
        [  4.7544,  -3.1425,  -1.5315],
        [  8.3718,  -8.2953,   8.4666],
        [  5.3924, -10.1396,  -4.0863],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.6202,  -1.9508,   0.7471],
        [  4.8674,  12.8875,   6.7622],
        [  9.9217,  -6.7006,  -5.6728],
        [  6.1989,  -9.7591,   0.8289],
        [  6.1741,  -0.5329,   3.1913],
        [ -8.1531,  -5.8748,   2.6934],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.9641,  -2.9568,   7.7211],
        [ 10.2470,   8.4054,  -0.1177],
        [ 12.5789,   0.1722,   1.6470],
        [  5.9655,   3.2172,  -7.5953],
        [  2.3376,  -4.6231,   9.8436],
        [  5.0348,  -2.8229,   3.5436],
        [  5.6414, -10.0303,   7.8525],
        [  2.8474,   6.8979,   6.6457],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.0031,  12.2958,  -6.5992],
        [  2.8474,   6.8979,   6.6457],
        [  9.2811,  -7.4336,   5.1771],
        [ -8.4887,  -5.2448,  -0.6585],
        [  0.4119,  14.1019,  -4.1870],
        [  1.7301,   7.4262,   5.0522],
        [  9.0652,   9.8597,  -3.5949],
        [ 11.5935,   5.9740,  -4.3952],
        [ -2.9497, -10.3028,  -7.3479],
        [  5.0190,   4.9705,  -7.9752],
        [  2.9470,  13.6085,   2.9809],
        [  2.7121,   6.9743,  -8.4202],
        [  3.2809,  13.5061,  -4.0199],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.3119,   1.7797,   9.9186],
        [  6.3606,   0.9204,   3.2525],
        [ 10.9394,   7.3069,  -8.9365],
        [ 12.3759,  -1.3943,   6.4659],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -5.6423,   6.4537,  -1.2091],
        [  0.6933,   7.7336,   2.1179],
        [ -8.6229,   2.9068,   2.4429],
        [ 11.3118,   6.5957,  -4.5667],
        [ -7.7842,   4.3466,  -1.7548],
        [  5.3118,   4.5452,   4.2114],
        [ 11.6801,  -3.7196,  -0.9181],
        [  3.7916,  -3.9524,  -7.9547],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.1890,   4.7318,   0.9515],
        [ 12.5224,  -0.4497,   4.8448],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.9452, -10.3187,  -7.8621],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.6544,   5.8273,  -3.1375]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1019, 0.1725,  ..., 0.9618, 0.2607, 0.9866],
        [0.1019, 0.0000, 0.1414,  ..., 0.8801, 0.3605, 0.8953],
        [0.1725, 0.1414, 0.0000,  ..., 0.9766, 0.4064, 0.8417],
        ...,
        [0.9618, 0.8801, 0.9766,  ..., 0.0000, 1.1515, 0.9710],
        [0.2607, 0.3605, 0.4064,  ..., 1.1515, 0.0000, 1.2433],
        [0.9866, 0.8953, 0.8417,  ..., 0.9710, 1.2433, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.8207297 , 11.018084  , 11.1341798 , 11.1913598 , 11.27160035,
       11.33837396,  5.84533345,  8.06517143,  9.10945678, 12.27993098,
        5.63824971, 11.74901778, 13.68063098, 11.83509372,  5.14999799,
        5.91408504, 10.61472636, 10.98076873, 13.01728313,  8.28968067,
       10.63038565,  8.56889738, 13.32613445, 13.35013374,  5.35774573,
       11.62158554,  8.68979052,  5.26067019, 12.59840193, 12.18998195,
        8.51928531, 12.44177641,  9.12220271, 11.018084  , 11.97957425,
        7.13501264, 12.67680439, 11.9182956 ,  7.48311002,  7.19534146,
        5.5205189 ,  6.27905292, 10.06703358, 13.59480906,  7.46053281,
        8.8580671 , 11.07637531, 11.48663767,  5.69913817, 11.785566  ,
       11.48434767,  6.77777606,  9.43737523,  5.94909709, 13.77604293,
       11.97238624, 11.56146873,  6.19707944, 10.04916678,  8.16584537,
       12.32408516, 13.25336001, 12.58006038,  6.77777606,  5.1805345 ,
        5.77219077, 11.50791859,  7.46248947,  4.90505566, 13.6829843 ,
        7.46248947, 11.89105259,  9.97822473, 14.10796466,  7.62509582,
       13.39372426, 13.04217487, 10.7167285 ,  7.06372172, 13.92390689,
        7.48311002, 13.89886469, 10.26327895,  6.55800267,  6.42689705,
       13.15525803, 12.45418236,  9.78286733,  8.5724042 ,  7.76457095,
        9.09963894, 13.09426142,  8.91551231,  6.99097013, 12.25806559,
        5.47702766,  8.95983586,  9.76896904,  6.23262289,  7.02253235,
       12.53047489, 11.75618513, 11.4424816 ,  8.12043983,  9.85330646,
       13.03002338])}
{'P': tensor([[7.7597e-03, 2.8085e-01, 2.2040e-01,  ..., 1.8289e-06, 7.1122e-07,
         3.2856e-03],
        [3.6155e-05, 6.7255e-03, 4.8833e-01,  ..., 3.1152e-06, 4.0594e-07,
         7.4200e-03],
        [4.5917e-05, 9.1949e-05, 1.1358e-02,  ..., 6.6328e-06, 5.9883e-07,
         1.7063e-02],
        ...,
        [1.7830e-07, 1.5020e-07, 1.4409e-07,  ..., 1.4432e-02, 1.4349e-06,
         2.1924e-07],
        [3.7110e-03, 3.1107e-03, 2.8003e-03,  ..., 4.6720e-06, 3.5932e-01,
         2.6905e-04],
        [3.3805e-06, 5.5123e-06, 6.5637e-06,  ..., 9.0258e-04, 4.1657e-07,
         2.5585e-02]]), 'X': tensor([[-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0743, 0.1174,  ..., 1.0873, 0.9198, 0.6034],
        [0.0743, 0.0000, 0.0437,  ..., 1.0141, 0.9100, 0.5296],
        [0.1174, 0.0437, 0.0000,  ..., 0.9704, 0.9160, 0.4861],
        ...,
        [1.0873, 1.0141, 0.9704,  ..., 0.0000, 1.4202, 0.4866],
        [0.9198, 0.9100, 0.9160,  ..., 1.4202, 0.0000, 1.0645],
        [0.6034, 0.5296, 0.4861,  ..., 0.4866, 1.0645, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.82106505, 10.99326577, 11.04953767, 11.0050379 , 11.23201619,
       11.04313641, 11.95743558,  6.5618968 ,  6.49538841, 11.89855267,
        8.27853076,  4.74770899, 13.27717982, 10.23946247,  4.8763135 ,
        9.45995955,  7.44805901, 13.60275211, 13.66896739,  7.09887557,
        8.36995953,  5.94541803,  5.41348121, 12.87752052,  7.80625981,
       11.63765273,  8.1303945 ,  6.12886024, 13.97779839,  5.91222346,
       10.76154256, 10.48721382,  6.39187285,  8.41148722, 12.86648771,
        5.92281142,  7.9337647 , 12.52875793, 11.50791859,  6.2134961 ,
        9.79814997, 13.06498554,  7.72415048,  8.1303945 ,  5.80312949,
        5.84450127,  9.14973319,  6.05165248, 11.83509372,  7.69752262,
        7.34250948, 13.76047999,  4.78002665,  9.48337063,  9.36256096,
       13.84126197, 12.19805394,  9.23071698, 10.37998078,  9.50654967,
        6.81961996, 12.32545637,  6.61386436,  6.36004393,  9.49511524,
        5.01114719,  5.39723241,  8.12043983, 13.91198191, 11.46176259,
        9.48337063,  6.90115901,  5.48276038,  4.978704  , 11.32149259,
       13.14777433, 10.60226614, 12.27993098, 11.56146873, 10.01804223,
        8.06517143,  6.05113217,  6.42689705,  9.37296102,  5.16077764,
        6.7593917 , 11.75618513,  6.2322769 ,  8.36445758, 10.1979669 ,
        6.3949043 ,  5.23878371,  7.69752262,  8.97052115,  8.7907923 ,
       13.35013374, 11.27656488,  9.55593374,  7.60463349,  6.76661595,
       10.42667093, 10.74409834, 11.83570132, 13.3607379 ,  9.75075273,
       11.87412439])}
{'P': tensor([[3.6898e-03, 1.8828e-01, 1.8010e-01,  ..., 6.3036e-03, 7.8974e-07,
         1.3782e-06],
        [4.5719e-05, 4.6995e-03, 5.2163e-02,  ..., 9.5098e-03, 1.2921e-06,
         1.3666e-06],
        [5.2005e-05, 3.5441e-04, 7.3246e-03,  ..., 9.1326e-03, 2.3528e-06,
         1.7159e-06],
        ...,
        [2.2099e-04, 3.4874e-04, 4.8723e-04,  ..., 9.3287e-02, 8.1586e-05,
         1.1240e-05],
        [8.7504e-07, 7.8785e-07, 7.5335e-07,  ..., 4.8993e-07, 6.4732e-02,
         4.1338e-06],
        [6.3712e-03, 6.0553e-03, 4.2531e-03,  ..., 5.2779e-03, 7.2719e-07,
         4.6321e-02]]), 'X': tensor([[ -1.6334, -10.7207,   7.1397],
        [ -0.8675, -10.8815,   6.8561],
        [ -0.4341, -10.9473,   7.4509],
        [  0.1252, -11.0062,   7.2564],
        [  0.1287, -11.0065,   6.5757],
        [ -0.1626, -10.9796,   6.4629],
        [  0.3493,  -4.7727,  -6.5682],
        [ -4.4503,   7.1378,   3.3751],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.1440, -10.9811,   8.5281],
        [  2.6613,   7.0021,   4.5783],
        [  5.8902,  -1.3887,   4.0388],
        [  3.9591, -10.6391,  -6.2557],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -8.1426,   3.8048,   9.3542],
        [  1.6308,  13.9254,  -2.0984],
        [ 11.9122,   5.1397,  -1.5498],
        [ -8.1531,  -5.8748,   2.6934],
        [ -9.3444,   0.5821,   6.9101],
        [  1.2728, -11.0372,   9.8121],
        [  2.6078, -10.9229,   6.2869],
        [  5.9843,  -9.8679,  -6.1449],
        [  6.0310,  12.2795,  -6.4528],
        [  5.0277,   4.9589,   7.0175],
        [  1.9312,  13.8649,   8.3563],
        [  0.8940, -11.0402,  -0.9783],
        [  6.3550,   0.7129,   0.4917],
        [  4.5340,  -3.3629,   8.9652],
        [  6.7792,  11.8084,   1.6251],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3587,   1.1596,  -3.6629],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.1459,   1.4780,  -4.2403],
        [ -7.5603,   4.6459,   8.2002],
        [ -3.5520,   7.4976,   9.1522],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.5636, -10.9293,   9.6519],
        [ -8.7735,   2.5640,  -6.3313],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.2076,  13.8032,  -1.5099],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.7000,  -4.7609,   2.7935],
        [  1.3339,   7.5627,  -0.9034],
        [  2.6451,  -4.5250,  -0.5565],
        [ -5.6174,   6.4708,  -1.7339],
        [ -1.5875,   7.8944,  -9.4286],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -2.1817,   7.8274,  -4.6298],
        [  4.0591,   6.0202,  -1.6339],
        [  5.6022,  -1.9833,   8.3081],
        [ 10.0400,   8.6922,  -6.2847],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.2860,   7.1916,  -6.1011],
        [  1.7903,  -4.7467,   0.1168],
        [  5.1667,  -2.6543,   1.2879],
        [ -8.8154,   2.4611,   2.5198],
        [ -8.6229,   2.9068,   2.4429],
        [ -1.9646,   7.8571,   3.3817],
        [  4.3377,   5.7570,   1.3712],
        [  0.5250, -11.0306,   5.5358],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.4714,  -1.3301,  -3.1998],
        [  9.0652,   9.8597,  -3.5949],
        [  3.3744,   6.5623,  -6.3636],
        [  8.8091,  -7.9037,   0.3631],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.2644,  -4.6434,   9.7508],
        [  2.7295,  13.6703,  -8.9328],
        [ -3.4986,   7.5152,   9.0712],
        [ -4.2271,   7.2387,  -1.8316],
        [ -4.6942,  -9.4299,  -9.1846],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.7963,  -1.6018,   1.8856],
        [  5.1292,   4.8183,  -5.7997],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.4157,   0.0821,  -5.0025],
        [  1.1139,   7.6281,  -2.5680],
        [  5.1203,  -2.7151,   6.2110],
        [ 11.9019,  -3.1379,  -3.4534],
        [  4.2096,   5.8815,   1.9230],
        [  6.3551,   0.7149,   1.3344],
        [ -3.2736, -10.1703,   3.8712],
        [  5.1519, -10.2387,   9.8191],
        [  4.7813,  -3.1139,   8.2691],
        [ 11.7795,  -3.4697,   3.5624],
        [ -1.8788, -10.6567,   3.7068],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -0.5385,   7.9037,   6.3830],
        [  3.2226,  -4.2802,  -4.0428],
        [  3.3943,   6.5483,   1.0790],
        [ -3.2053, -10.1993,  -3.8736],
        [  1.3103,  -4.8032,  -0.9300],
        [ -7.3554,   4.8985,  -0.6665],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.5506,  -5.1154,   6.3423],
        [  5.9843,  -9.8679,  -6.1449],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.6319,  11.9066,   3.5165],
        [  3.1271, -10.8343,   2.6442],
        [  8.7131,  10.2216,   7.7915],
        [ -8.4887,  -5.2448,  -0.6585]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0304, 0.0664,  ..., 0.4318, 1.1646, 0.6779],
        [0.0304, 0.0000, 0.0417,  ..., 0.4113, 1.1362, 0.6797],
        [0.0664, 0.0417, 0.0000,  ..., 0.4176, 1.0986, 0.7127],
        ...,
        [0.4318, 0.4113, 0.4176,  ..., 0.0000, 1.0727, 0.4880],
        [1.1646, 1.1362, 1.0986,  ..., 1.0727, 0.0000, 1.5589],
        [0.6779, 0.6797, 0.7127,  ..., 0.4880, 1.5589, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.84437705, 10.91601655, 10.95593729, 11.00694688, 11.00726483,
       10.98076873,  4.78544875,  8.41148722,  5.21280156, 11.18839169,
        7.49076899,  6.05165248, 11.35182589,  9.91668182,  8.98765644,
       14.02058621, 12.97370371, 10.04916678,  9.36256096, 11.11038551,
       11.22993218, 11.54071859, 13.68063098,  7.06169391, 13.99877413,
       11.07637531,  6.3949043 ,  5.64500512, 13.61600536, 11.33837396,
        6.46356806,  5.39098669,  9.26455795,  8.8737556 ,  8.29640613,
       10.24557047, 11.22596994,  9.14044984,  9.9353921 , 13.97857591,
        5.84450127,  5.05534609,  7.67939546,  5.24137909,  8.56889738,
        8.05242591, 10.31613269,  8.12580976,  7.26074925,  5.94293185,
       13.27994304, 13.00413829,  7.54620849,  5.07306274,  5.80860165,
        9.15253344,  9.09963894,  8.09899874,  7.20826219, 11.04313641,
       12.22276278,  9.5642993 , 13.39372426,  7.37902766, 11.83509372,
       11.74901778,  5.16611782, 13.94008948,  8.28968067,  8.38252937,
       10.53368644,  4.74468406,  6.00732193,  6.01356004,  7.03733775,
        5.91854397,  9.562174  ,  9.41605684,  7.70897675,  5.79562691,
       12.30858677,  7.23276582,  6.39520397, 10.68416401, 11.46176259,
        5.70591986, 12.27991991, 10.82106505, 10.61047596,  9.58072286,
        7.92200342,  5.35774573,  7.37576154, 10.6910768 ,  4.978704  ,
        8.83725081,  9.61942445,  9.96391045, 11.54071859, 12.18045978,
       12.24467041, 11.54672755, 13.62896593, 11.27656488, 13.43127131,
        9.97822473])}
{'P': tensor([[6.4113e-03, 3.0977e-01, 2.3681e-01,  ..., 1.2985e-04, 2.2750e-05,
         4.8903e-07],
        [3.5120e-05, 6.6591e-03, 4.8351e-01,  ..., 2.7796e-04, 4.1892e-05,
         4.2734e-07],
        [4.4173e-05, 8.7594e-05, 1.0820e-02,  ..., 5.8741e-04, 8.7163e-05,
         6.5219e-07],
        ...,
        [4.7505e-06, 5.8967e-06, 6.4781e-06,  ..., 1.7617e-01, 4.0789e-04,
         4.4532e-05],
        [2.5763e-07, 3.2487e-07, 3.6642e-07,  ..., 3.8673e-05, 2.0559e-02,
         4.0038e-07],
        [3.6408e-04, 2.0247e-04, 1.5956e-04,  ..., 3.0811e-06, 2.9380e-07,
         4.7390e-02]]), 'X': tensor([[ -1.6275, -10.7221,   3.2376],
        [ -0.0254, -10.9932,   3.2200],
        [  0.5960, -11.0335,   3.1144],
        [  0.1041, -11.0045,   2.4311],
        [  2.6310, -10.9195,   3.8054],
        [  3.1271, -10.8343,   2.6442],
        [  9.0004,  -7.7195,   4.6945],
        [  5.5101,  -2.1433,   9.5442],
        [  9.1409,  -7.5789,   1.5814],
        [  0.7275,  14.0667,  -0.9815],
        [  5.1793,   4.7461,   0.3899],
        [ -9.4501,  -1.7133,   2.2980],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.3087,   1.8053,  -1.9481],
        [ -2.2188,   7.8218,   8.6064],
        [  5.7963,  -1.6018,   1.8856],
        [  6.3360,   0.4258,  -9.2509],
        [ -7.6525,  -6.6534,   5.1721],
        [ -1.5875,   7.8944,  -9.4286],
        [  0.9809,   7.6642,  -1.4217],
        [ -0.6323,   7.9085,  -5.9544],
        [ -6.6998,   5.5954,   0.0823],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.4716,  -0.6673,   7.6068],
        [  5.8450,  -9.9356,   3.0767],
        [  0.3231,  -4.7691,  -1.6120],
        [  2.4356,  -4.5942,  -5.0808],
        [  2.6970,   6.9826,   9.7350],
        [ -4.2040,   7.2486,   2.9875],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -2.0445,   7.8469,  -0.5857],
        [ -4.4520,   7.1370,  -8.7867],
        [  4.0078,  -3.8018,  -6.2168],
        [ -3.4100,   7.5436,  -7.6980],
        [ -8.9788,   2.0208,  -3.2867],
        [ -1.2454,   7.9127,  -0.5715],
        [  0.1047, -11.0046,   0.5188],
        [  2.3589,  13.7668,  -3.6501],
        [ -9.1623,  -3.4288,  -4.9961],
        [  8.8474,  10.0868,  -8.8644],
        [  6.9693,  -9.3200,  -8.3822],
        [ -5.7347,  -8.6908,   9.2878],
        [ -8.8934,   2.2597,   3.5831],
        [  0.6142,  14.0802,   0.5407],
        [  8.5198,  10.4091,  -8.6486],
        [ -3.1204, -10.2345,   9.2288],
        [  1.7122,  -4.7591,  -3.6734],
        [  8.2422,  -8.4041,  -1.4962],
        [  4.1348,   5.9514,  -0.8856],
        [ 12.3609,   3.4769,  -8.5762],
        [  3.4009,  -4.1874,  -5.9751],
        [  4.7482, -10.3907,   5.9149],
        [  6.3587,   1.1596,  -3.6629],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.3924, -10.1396,  -4.0863],
        [  0.2732,  -4.7619,  -1.1577],
        [  8.4691,  10.4571,  -5.0650],
        [  6.9693,  -9.3200,  -8.3822],
        [  1.4306,  13.9620,  -0.2075],
        [  8.3718,  -8.2953,   8.4666],
        [  1.7500,   7.4187,   4.4237],
        [  5.7485,  -9.9811,  -3.2355],
        [ 12.5154,   2.5072,  -9.3274],
        [  3.1892, -10.8220,  -6.8409],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.9843,  -9.8679,  -6.1449],
        [ -1.9714,   7.8563,  -1.8084],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.1685,  12.1979,  -6.2473],
        [  1.9312,  13.8649,   8.3563],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -1.9600,   7.8577,   6.9121],
        [  5.4867,  12.5816,   2.9549],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.3515,   0.6418,  -6.5823],
        [  3.3943,   6.5483,   1.0790],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.2188,  -2.5839,   2.1833],
        [ -9.4716,  -0.6673,   7.6068],
        [ -4.8751,  -9.3148,   4.0406],
        [ -9.3384,  -2.6161,  -4.7630],
        [  3.9591, -10.6391,  -6.2557],
        [ -4.8751,  -9.3148,   4.0406],
        [  6.8704,  -9.3809,   7.5769],
        [ -8.0087,   4.0173,   9.1091],
        [  4.0457,  -3.7737,  -0.4704],
        [ 12.1940,  -2.1881,   7.1499],
        [ 10.2470,   8.4054,  -0.1177],
        [  8.5093,  -8.1764,  -3.2381],
        [  3.9515,  -3.8425,   6.1729],
        [  1.4150,  13.9647,  -4.0915],
        [  4.8653,   5.1694,  -4.0823],
        [  2.7685, -10.8982,  -6.8928],
        [  3.9022,  13.2906,   5.5788],
        [  5.8617,   3.4853,  -9.8080],
        [  5.5158,  -2.1337,   3.5246],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.1426,   3.8048,   9.3542]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0614, 0.1039,  ..., 0.9071, 1.2187, 0.3047],
        [0.0614, 0.0000, 0.0437,  ..., 0.8499, 1.1775, 0.2930],
        [0.1039, 0.0437, 0.0000,  ..., 0.8168, 1.1404, 0.3055],
        ...,
        [0.9071, 0.8499, 0.8168,  ..., 0.0000, 1.2096, 0.7611],
        [1.2187, 1.1775, 1.1404,  ..., 1.2096, 0.0000, 1.3965],
        [0.3047, 0.2930, 0.3055,  ..., 0.7611, 1.3965, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.84493732, 10.99326577, 11.04953767, 11.0050379 , 11.23201619,
       11.27656488, 11.85743425,  5.91222346, 11.87412439, 14.08549802,
        7.02496223,  9.60412735,  5.5366698 ,  6.5618968 ,  8.1303945 ,
        6.01356004,  6.35029274, 10.14044489,  8.05242591,  7.72668467,
        7.9337647 ,  8.72896381,  5.39450305,  9.49511524, 11.52733581,
        4.78002665,  5.19986797,  7.48538905,  8.37954853,  8.04483495,
        9.27474386,  9.91668182,  8.10886021,  8.41171369,  5.52416368,
        8.27853076,  9.20340031,  8.01009083, 11.00508681, 13.96746816,
        9.78286733, 13.41713661, 11.63765273, 10.412299  ,  9.17595899,
       14.09357547, 13.45124917, 10.69963954,  5.05774295, 11.77124373,
        7.24677664, 12.84056604,  5.39450305, 11.42420508,  6.46356806,
       12.58006038,  9.78147542, 12.12448912, 11.48434767,  4.76969364,
       13.45642379, 11.63765273, 14.03505624, 11.785566  ,  7.62232307,
       11.51811603, 12.76408562, 11.28215429,  5.0209005 , 11.54071859,
        8.09984231, 11.72749923, 13.66896739, 13.99877413, 10.14436912,
        8.09842704, 13.72594022, 12.45418236,  6.23262289,  6.383897  ,
        7.37576154, 10.31886066,  5.82345867,  9.49511524, 10.51340829,
        9.69792513, 11.35182589, 10.51340829, 11.62770332,  8.95983586,
        5.53255882, 12.38881635, 13.25336001, 11.80092578,  5.51177501,
       14.03618649,  7.09887557, 11.24434374, 13.85158417,  6.81961996,
        5.91408504, 13.36092624,  9.58072286, 12.44177641, 12.46513646,
        8.98765644])}
{'P': tensor([[3.9078e-03, 2.8579e-01, 1.8981e-01,  ..., 7.9507e-06, 7.7938e-07,
         1.1877e-06],
        [4.7084e-05, 5.4311e-03, 2.1328e-01,  ..., 8.2954e-06, 1.2768e-06,
         1.9829e-06],
        [5.2144e-05, 9.6210e-05, 7.4890e-03,  ..., 8.1801e-06, 2.3799e-06,
         3.4890e-06],
        ...,
        [6.0091e-02, 5.7025e-02, 6.6340e-02,  ..., 1.0153e-01, 7.5947e-06,
         4.6181e-06],
        [6.7633e-08, 6.3828e-08, 6.3003e-08,  ..., 2.5217e-07, 8.2694e-03,
         3.9221e-04],
        [6.1158e-08, 5.7661e-08, 5.3650e-08,  ..., 1.1158e-07, 1.0618e-04,
         4.8037e-03]]), 'X': tensor([[-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0246, 0.0611,  ..., 0.5305, 1.2223, 1.1394],
        [0.0246, 0.0000, 0.0411,  ..., 0.5446, 1.2065, 1.1152],
        [0.0611, 0.0411, 0.0000,  ..., 0.5397, 1.1656, 1.0816],
        ...,
        [0.5305, 0.5446, 0.5397,  ..., 0.0000, 1.1931, 1.4363],
        [1.2223, 1.2065, 1.1656,  ..., 1.1931, 0.0000, 0.7982],
        [1.1394, 1.1152, 1.0816,  ..., 1.4363, 0.7982, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.86693708, 10.91601655, 10.98076873, 11.00726483, 11.00694688,
       11.05317597,  6.14086582, 12.10222204,  9.97220799, 12.40074901,
        5.34250225, 12.10942412,  6.56833864,  8.06643487,  7.16598273,
       12.12448912,  7.59795445,  7.5984809 , 13.44593519, 13.89671314,
        6.46560181,  9.43737523, 10.0674887 , 13.35832725, 11.22142195,
       14.07623266,  9.58072286,  9.43737523,  7.69752262, 13.69162795,
       13.32374056,  8.62631801,  5.64464807, 11.73397309, 11.80092578,
        7.26074925,  7.02253235, 10.99326577,  8.31625171,  5.95111804,
       11.62158554,  6.46356806,  7.98469166,  7.52117132, 12.10942412,
       13.43127131,  6.12886024,  9.73579738, 10.19036081,  8.85939626,
       13.59673406, 12.10942412, 11.28215429,  6.67218687, 11.97957425,
       12.12448912,  4.89869205,  7.56617037,  6.81117802, 13.32374056,
        7.46053281,  9.69629   , 10.06703358,  5.30158497,  8.05668609,
       12.62576026,  8.12580976, 12.5769659 , 11.87412439, 12.52372629,
       13.50484539,  7.71075226,  6.83083863, 10.35485331, 10.42667093,
       11.27656488, 11.51811603, 13.97857591,  6.81961996,  6.39520397,
        5.91677188, 11.73848065,  6.02789634,  9.44486609,  9.7712321 ,
        6.10774544,  9.75401449, 12.07818871,  8.7273569 ,  9.09963894,
        8.10886021,  9.50654967, 13.04051744,  4.90505566,  9.09661212,
        8.29640613, 12.91143491, 12.05395056, 11.64746882, 10.39942497,
       11.81423161,  6.65991877,  7.14837514, 10.36844813, 13.32645552,
       13.36800073])}
{'P': tensor([[4.1774e-03, 3.1650e-01, 2.1843e-01,  ..., 2.2646e-06, 3.3522e-07,
         1.2646e-06],
        [5.9751e-05, 6.1266e-03, 3.5923e-01,  ..., 3.5268e-06, 4.8251e-07,
         1.9882e-06],
        [2.1893e-04, 2.6065e-04, 3.8486e-02,  ..., 2.6476e-05, 3.0147e-06,
         1.4522e-05],
        ...,
        [3.7412e-05, 3.5187e-05, 2.9056e-05,  ..., 1.1895e-01, 9.6954e-03,
         4.6031e-03],
        [3.9299e-04, 3.4580e-04, 2.4866e-04,  ..., 1.9907e-04, 1.7725e-01,
         3.4524e-05],
        [1.9781e-05, 1.9002e-05, 1.5941e-05,  ..., 1.1943e-02, 5.4964e-04,
         1.5051e-01]]), 'X': tensor([[ -1.3178, -10.7939,   0.4236],
        [ -0.9715, -10.8630,   0.3555],
        [  0.1047, -11.0046,   0.5188],
        [  0.1382, -11.0073,   1.0026],
        [  0.5600, -11.0321,  -0.8660],
        [  0.8940, -11.0402,  -0.9783],
        [  2.3373,  13.7721,  -0.2442],
        [  2.6613,   7.0021,   4.5783],
        [  4.5340,  -3.3629,   8.9652],
        [  4.5767,   5.5071,   3.9069],
        [  3.0876,   6.7527,  -5.3540],
        [  1.6954,  -4.7616,   9.8198],
        [ -9.3604,   0.4855,  -3.0158],
        [  5.9843,  -9.8679,  -6.1449],
        [ 11.9641,  -2.9568,   7.7211],
        [ -1.6906,   7.8860,   7.7142],
        [  4.3377,   5.7570,   1.3712],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -4.9217,   6.8979,  -1.7667],
        [  1.9312,  13.8649,   8.3563],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.2592,   3.9425,   2.0383],
        [  5.2608,  12.6977,   4.9166],
        [ -3.4986,   7.5152,   9.0712],
        [  5.0277,   4.9589,   7.0175],
        [  6.3031,   1.8477,  -7.1706],
        [  8.0381,  -8.5690,  -8.9523],
        [  1.1994,   7.6035,   2.4295],
        [ -1.6275, -10.7221,   3.2376],
        [ 12.3759,  -1.3943,   6.4659],
        [ -2.1382,   7.8339,   7.3524],
        [ 12.1504,   4.3674,  -7.2998],
        [  1.2273,  -4.8081,   9.8129],
        [  0.3354,  -4.7708,  -4.9125],
        [  0.1206,  -4.7363,   8.7103],
        [ -7.4602,   4.7718,  -4.6277],
        [ -2.1533, -10.5778,   1.0112],
        [  6.1741,  -0.5329,   3.1913],
        [ -5.1974,   6.7396,   3.3484],
        [  5.4429,  -2.2529,   9.7306],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.1575,  12.2046,   3.3257],
        [ -6.4844,  -8.0230,   8.5209],
        [  6.1973,  -0.4372,   6.5720],
        [  5.8323,  -1.5228,   5.1132],
        [  2.5636, -10.9293,   9.6519],
        [ -9.1178,   1.5784,   2.1410],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.0829,  -5.9940,   5.0564],
        [  5.9912,   3.1453,  -3.9989],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.4013,  -1.2597,  -7.6474],
        [  0.8940, -11.0402,  -0.9783],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.8318,   3.5567,  -4.7692],
        [ 11.0891,  -4.9558,  -2.3178],
        [  5.8889,  -1.3916,   8.8993],
        [  9.2979,   9.6048,   7.3138],
        [  4.9152,  12.8653,  -1.4459],
        [  2.7590,  13.6621,   9.4496],
        [  0.3354,  -4.7708,  -4.9125],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.3303,   1.6145,  -8.4083],
        [  7.9141,  -8.6656,   9.6948],
        [  3.4140,  -4.1803,  -8.8224],
        [ -1.0407,   7.9166,  -6.6021],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.0190,   4.9705,  -7.9752],
        [  4.0078,  -3.8018,  -6.2168],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.4773,  -1.0024,   5.2599],
        [  4.7482, -10.3907,   5.9149],
        [  9.3836,   9.5075,  -5.9278],
        [ -8.7735,   2.5640,  -6.3313],
        [  2.6310, -10.9195,   3.8054],
        [  3.4401,  13.4540,   4.6576],
        [ -5.3382,   6.6533,   7.8054],
        [  9.9750,  -6.6338,  -1.2440],
        [  1.2273,  -4.8081,   9.8129],
        [  5.9526,   3.2525,  -4.3821],
        [  7.8425,  -8.7202,   6.4530],
        [ -2.0445,   7.8469,  -0.5857],
        [  0.6396,  14.0772,  -6.3786],
        [  5.1861,   4.7361,  -3.7312],
        [ -0.0254, -10.9932,   3.2200],
        [  2.3928,   7.1404,  -1.7035],
        [  5.1787,  -2.6383,  -4.2533],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -2.6735, -10.4062,  -1.5911],
        [ -5.3328,  -8.9992,   7.3440],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.3266,   0.3292,   9.3502],
        [  5.1890,   4.7318,   0.9515],
        [  1.9204,   7.3518,   8.4781],
        [  5.9548,  -1.2273,  -2.6520],
        [  1.7228,  -4.7575,   8.3396],
        [ 12.3357,   3.6001,   0.7903],
        [  0.8940, -11.0402,  -0.9783],
        [  6.2242,  -0.3171,   7.4978],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.2373,  -4.6506,  -1.7334],
        [ 12.1504,   4.3674,  -7.2998],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.0039,   7.3173,  -1.5885],
        [ -5.9940,   6.1985,   5.9566],
        [  2.7121,   6.9743,  -8.4202]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0291, 0.1035,  ..., 0.6383, 0.5019, 0.7549],
        [0.0291, 0.0000, 0.0761,  ..., 0.6093, 0.4946, 0.7279],
        [0.1035, 0.0761, 0.0000,  ..., 0.5379, 0.4542, 0.6764],
        ...,
        [0.6383, 0.6093, 0.5379,  ..., 0.0000, 0.6628, 0.3254],
        [0.5019, 0.4946, 0.4542,  ..., 0.6628, 0.0000, 0.9436],
        [0.7549, 0.7279, 0.6764,  ..., 0.3254, 0.9436, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.87409005, 10.90638066, 11.00508681, 11.00812996, 11.04629471,
       11.07637531, 13.96905756,  7.49076899,  5.64500512,  7.16058587,
        7.42512621,  5.05444288,  9.37296102, 11.54071859, 12.32408516,
        8.06517143,  7.20826219, 10.52357479,  8.47371868, 13.99877413,
       13.99048094, 12.87752052, 13.744387  ,  8.28968067,  7.06169391,
        6.56833864, 11.74901778,  7.69752262, 10.84493732, 12.45418236,
        8.12043983, 12.91143491,  4.96231532,  4.78258092,  4.73784222,
        8.85572292, 10.794755  ,  6.19707944,  8.51088913,  5.89074104,
        9.14973319, 13.66991027, 10.31584102,  6.21275016,  6.02779201,
       11.22596994,  9.25336371,  8.87007495, 10.06286074,  6.76661595,
       13.01728313, 12.46513646, 11.07637531,  8.30125201,  6.83083863,
       12.1460931 ,  6.05113217, 13.36800073, 13.77222752, 13.93790235,
        4.78258092,  5.56809335,  6.53290663, 11.73567759,  5.39723241,
        7.98469166, 10.00802627,  7.06372172,  5.52416368, 12.11735334,
        9.53015376, 11.42420508, 13.35832725,  9.14044984, 11.23201619,
       13.88683436,  8.53014915, 11.97949273,  4.96231532,  6.78325151,
       11.72802491,  8.10886021, 14.09176259,  7.02326515, 10.99326577,
        7.53063885,  5.81199422, 10.60226614, 10.74409834, 10.4606265 ,
        6.12886024,  6.33517349,  7.02253235,  7.5984809 ,  6.07992623,
        5.05981914, 12.85032836, 11.07637531,  6.2322769 , 12.16901615,
        5.16077764, 12.91143491, 11.9182956 ,  7.58668144,  8.62260327,
        7.48311002])}
{'P': tensor([[5.5801e-03, 2.4127e-01, 1.3495e-01,  ..., 8.3314e-04, 1.6618e-04,
         2.9940e-05],
        [1.3385e-04, 1.8416e-02, 4.2300e-02,  ..., 2.3107e-03, 8.1355e-04,
         1.7512e-04],
        [4.0584e-05, 6.7546e-04, 5.2907e-03,  ..., 9.7922e-04, 4.1427e-04,
         5.9935e-05],
        ...,
        [5.6185e-04, 7.5134e-04, 1.0029e-03,  ..., 1.2038e-01, 1.5914e-02,
         4.0903e-03],
        [1.8717e-06, 2.3102e-06, 2.8440e-06,  ..., 1.5279e-04, 5.3978e-02,
         2.7706e-02],
        [1.6927e-07, 1.8140e-07, 1.8535e-07,  ..., 5.4703e-06, 6.3344e-06,
         9.9718e-03]]), 'X': tensor([[-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0811, 0.1127,  ..., 0.7244, 0.9169, 0.8920],
        [0.0811, 0.0000, 0.1176,  ..., 0.6708, 0.8793, 0.8178],
        [0.1127, 0.1176, 0.0000,  ..., 0.6254, 0.8073, 0.8236],
        ...,
        [0.7244, 0.6708, 0.6254,  ..., 0.0000, 0.2751, 0.3658],
        [0.9169, 0.8793, 0.8073,  ..., 0.2751, 0.0000, 0.5773],
        [0.8920, 0.8178, 0.8236,  ..., 0.3658, 0.5773, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.88618114, 11.00508681, 11.04629471, 11.00812996, 11.07637531,
       11.05413545, 13.62896593,  9.09119351, 10.80728788, 10.06023665,
        7.33350642,  8.04483495,  7.92607342,  7.20482224, 11.74901778,
       14.03505624,  6.02588517,  9.37296102, 11.73567759, 10.06023665,
       12.19805394,  8.0550286 , 12.25806559,  6.36004393,  7.60529892,
       10.67322527,  6.31576264,  7.73692436,  6.7166297 ,  9.13507321,
       11.22993218, 12.30695098,  4.83371341, 11.75618513,  9.6282763 ,
       13.89886469,  5.52416368, 13.26590071,  6.81117802, 13.09737053,
       11.46176259,  9.85384623, 13.32645552, 12.87752052, 12.41366754,
       13.85158417,  9.41244865, 11.95743558, 12.76408562, 10.7167285 ,
       11.27656488, 10.57302546, 11.07637531, 13.38091121, 13.38091121,
        4.978704  ,  5.18277858, 11.83570132, 13.5247215 , 13.85158417,
        7.73151309, 10.48840476,  7.61088125,  6.02789634, 10.68031351,
       13.12218013, 10.35464916,  6.652742  ,  5.29755487, 13.6660671 ,
        9.68323686, 11.28895489,  5.05534609,  5.19065439,  9.36256096,
       13.13205357, 11.86239961,  5.34250225,  6.27715105, 12.00995898,
       13.03002338, 12.22276278,  5.60064319, 13.36800073, 11.1341798 ,
        5.79543038, 12.7125079 , 11.66453059, 13.13888841,  7.57685908,
        6.29034958,  5.26067019,  6.76056625,  9.97220799, 13.40850328,
        7.19513806, 11.48434767, 10.52357479,  7.23276582,  5.64865173,
        5.05963217,  9.96391045,  7.18354885,  5.27178363, 12.27835996,
       12.78327037])}
{'P': tensor([[7.6019e-03, 4.3331e-01, 1.7718e-01,  ..., 2.7458e-04, 1.0222e-05,
         6.4089e-05],
        [1.6699e-04, 1.2463e-02, 2.7659e-01,  ..., 5.1042e-04, 1.5558e-05,
         1.1000e-04],
        [1.9797e-04, 2.5010e-04, 4.6969e-02,  ..., 6.0946e-03, 7.3625e-05,
         9.0698e-04],
        ...,
        [1.4617e-06, 1.5153e-06, 2.2335e-06,  ..., 4.4889e-02, 6.9231e-06,
         8.8501e-03],
        [4.9383e-05, 4.9139e-05, 1.0036e-04,  ..., 2.4926e-04, 1.1617e-01,
         1.5734e-03],
        [2.8466e-05, 2.8231e-05, 3.4779e-05,  ..., 7.9276e-04, 1.9080e-04,
         1.2310e-01]]), 'X': tensor([[ -1.0482, -10.8487,  -5.9066],
        [ -0.7599, -10.8995,  -6.0366],
        [  1.5383, -11.0274,  -4.9666],
        [  2.7685, -10.8982,  -6.8928],
        [  3.1892, -10.8220,  -6.8409],
        [  2.4639, -10.9431,  -7.3718],
        [ -9.0574,   1.7801,   8.8986],
        [  7.0642,  11.6104,  -9.8316],
        [ -1.9600,   7.8577,   6.9121],
        [  5.8318,   3.5567,  -4.7692],
        [  1.5653,  13.9377,   4.1773],
        [ -6.4002,  -8.1048,  -0.9108],
        [  0.6757, -11.0361,  -1.3047],
        [  5.4179,  -2.2925,  -3.4641],
        [ -2.1817,   7.8274,  -4.6298],
        [  0.5250, -11.0306,   5.5358],
        [ -2.5683, -10.4433,  -2.9182],
        [ -7.9667,   4.0814,  -0.6397],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.1741,  -0.5329,   3.1913],
        [ -1.6275, -10.7221,   3.2376],
        [  1.1006,   7.6318,  -2.4877],
        [  8.5716,  10.3596,   6.8818],
        [ -1.8823, -10.6558,  -3.8952],
        [  5.6248,   3.9984,   8.4933],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.2480, -11.0153,  -3.0183],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.0243,   7.3086,   7.5432],
        [ -6.3489,   5.9117,   6.5161],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -9.2508,   1.0572,   2.9978],
        [ 12.5717,   1.9266,   7.4903],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 10.0995,  -6.4739,   2.0974],
        [ -4.0409,   7.3169,  -3.5779],
        [ -7.8849,  -6.3115,  -4.7479],
        [  0.6952,  -4.8060,   8.8976],
        [  4.8674,  12.8875,   6.7622],
        [ -8.0611,  -6.0302,   4.6962],
        [ -8.7078,   2.7182,   1.8146],
        [  3.3744,   6.5623,  -6.3636],
        [ 11.8951,  -3.1570,   7.0742],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.5340,  -3.3629,   8.9652],
        [  3.3839,  -4.1966,   8.3574],
        [ 10.5057,  -5.9099,  -5.1599],
        [  3.9085,   6.1512,  -2.2560],
        [  2.5375,   7.0676,  -4.3490],
        [ -1.6275, -10.7221,   3.2376],
        [  8.2154,  -8.4262,   8.0171],
        [  6.0054,  -1.0902,   8.3079],
        [ -4.7642,  -9.3860,  -5.1133],
        [  7.0171,  11.6439,  -5.0178],
        [  4.1297,  -3.7098,  -3.9838],
        [  8.3718,  -8.2953,   8.4666],
        [ -5.5525,   6.5146,  -4.4984],
        [ 12.2435,   4.0076,   8.1257],
        [ -0.2586,   7.8825,  -9.9840],
        [  2.2076,  13.8032,  -1.5099],
        [  2.2373,  -4.6506,  -1.7334],
        [ 12.6035,   0.7494,   7.1536],
        [  2.3889,  -4.6082,  -8.7291],
        [  0.6952,  -4.8060,   8.8976],
        [  0.4996,   7.7738,  -4.4711],
        [  1.6308,  13.9254,  -2.0984],
        [  6.9284,  11.7061,  -0.3732],
        [  0.6470, -11.0352,  -1.1931],
        [ -1.2454,   7.9127,  -0.5715],
        [  7.3724,  -9.0577,  -8.9843],
        [  9.3607,   9.5337,  -4.5412],
        [ 10.0567,   8.6697,   4.6329],
        [ 12.5154,   2.5072,  -9.3274],
        [ -8.6801,   2.7809,  -0.8443],
        [  4.0591,   6.0202,  -1.6339],
        [  5.1472,  -2.6801,   8.1194],
        [ 11.7418,   5.6070,  -1.4192],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.1231,  -3.5739,   8.4708],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.3602,  -2.3808,   7.2098],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 11.6801,  -3.7196,  -0.9181],
        [  7.1697,  11.5343,  -0.7276],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.1139,   7.6281,  -2.5680],
        [  2.1824,  -4.6647,   1.1985],
        [  2.2076,  13.8032,  -1.5099],
        [ 11.9122,   5.1397,  -1.5498],
        [ -8.9178,   2.1936,   7.5388],
        [  0.5250, -11.0306,   5.5358],
        [  4.5954,  13.0099,  -4.4439],
        [ -9.2687,  -2.9794,   0.6436],
        [ -1.3178, -10.7939,   0.4236],
        [  5.0277,   4.9589,   7.0175],
        [  4.1140,  -3.7220,  -0.1273],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.6952,  -4.8060,   8.8976],
        [  2.5620,   7.0549,   8.4825],
        [ -4.7334,   6.9982,  -0.4392],
        [ 10.9394,   7.3069,  -8.9365],
        [  2.0243,   7.3086,   7.5432],
        [ 12.2921,  -1.7902,  -4.0870],
        [  1.6954,  -4.7616,   9.8198],
        [  6.3211,   1.7017,  -0.2164]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0313, 0.1371,  ..., 0.7927, 1.2194, 0.8626],
        [0.0313, 0.0000, 0.1126,  ..., 0.7631, 1.2071, 0.8398],
        [0.1371, 0.1126, 0.0000,  ..., 0.6619, 1.1054, 0.7272],
        ...,
        [0.7927, 0.7631, 0.6619,  ..., 0.0000, 0.9027, 0.3509],
        [1.2194, 1.2071, 1.1054,  ..., 0.9027, 0.0000, 0.5523],
        [0.8626, 0.8398, 0.7272,  ..., 0.3509, 0.5523, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.8992524 , 10.92597196, 11.1341798 , 11.24434374, 11.28215429,
       11.21703654,  9.23071698, 13.59056613,  8.09842704,  6.83083863,
       14.02532758, 10.32715715, 11.05672136,  5.8828986 ,  8.12580976,
       11.04313641, 10.75443106,  8.95133961,  7.9337647 ,  6.19707944,
       10.84493732,  7.71075226, 13.44593519, 10.8207297 ,  6.90115901,
        5.19986797, 11.018084  ,  4.78258092,  7.58378655,  8.67502348,
       10.52357479,  9.3109884 , 12.71843462, 10.27716434, 11.99633819,
        8.35856487, 10.09979747,  4.85603886, 13.77604293, 10.06703358,
        9.12220271,  7.37902766, 12.30695098, 10.37998078,  5.64500512,
        5.39098669, 12.05395056,  7.28793663,  7.5092834 , 10.84493732,
       11.76830981,  6.10360073, 10.52587243, 13.59480906,  5.5513072 ,
       11.785566  ,  8.55982105, 12.88270018,  7.88677538, 13.97857591,
        5.16077764, 12.62576026,  5.19065439,  4.85603886,  7.78980299,
       14.02058621, 13.60275211, 11.05413545,  8.01009083, 11.67875855,
       13.36092624, 13.2778378 , 12.76408562,  9.11473391,  7.26074925,
        5.80312949, 13.01187484,  5.68942572,  9.79814997, 12.97370371,
        5.95297459,  5.86519525, 10.5575601 , 12.25806559, 13.58101794,
        8.05242591,  7.70897675,  5.14999799, 13.97857591, 12.97370371,
        9.18358562, 11.04313641, 13.79762415,  9.73579738, 10.87409005,
        7.06169391,  5.54776934,  5.18277858,  4.85603886,  7.50564223,
        8.4486771 , 13.15525803,  7.58378655, 12.4217523 ,  5.05444288,
        6.54615887])}
{'P': tensor([[6.2285e-03, 3.6554e-01, 1.7172e-01,  ..., 1.5909e-05, 4.9267e-07,
         1.4562e-07],
        [6.7049e-05, 1.0134e-02, 1.0596e-01,  ..., 3.1803e-05, 7.8223e-07,
         3.0537e-07],
        [6.7890e-05, 4.0322e-04, 8.5644e-03,  ..., 2.4736e-05, 6.9443e-07,
         3.0880e-07],
        ...,
        [9.3519e-06, 9.1902e-06, 8.2219e-06,  ..., 7.5433e-02, 7.1230e-05,
         1.2855e-04],
        [1.3213e-04, 9.2692e-05, 8.5905e-05,  ..., 4.8033e-06, 5.1239e-02,
         1.0079e-05],
        [4.5336e-06, 3.5006e-06, 3.3233e-06,  ..., 1.2780e-05, 6.9373e-02,
         2.4472e-01]]), 'X': tensor([[ -1.0012, -10.8576,   2.2867],
        [  0.1041, -11.0045,   2.4311],
        [  0.5960, -11.0335,   3.1144],
        [  3.1271, -10.8343,   2.6442],
        [ -0.0254, -10.9932,   3.2200],
        [  2.6310, -10.9195,   3.8054],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.5428,  -9.5728,  -2.9897],
        [ -8.3653,   3.4184,   9.3454],
        [ -2.3317,   7.8034,  -3.6274],
        [  5.3860,  -2.3418,   1.0826],
        [  5.8617,   3.4853,  -9.8080],
        [ -5.3382,   6.6533,   7.8054],
        [  4.0362,  13.2397,  -7.0030],
        [ -2.9497, -10.3028,  -7.3479],
        [  5.5626,  -2.0536,   8.4433],
        [ 11.4235,   6.3594,   5.1663],
        [ 12.5789,   0.1722,   1.6470],
        [  6.3266,   0.3292,   9.3502],
        [  0.5600, -11.0321,  -0.8660],
        [  6.3587,   1.1596,  -3.6629],
        [ -7.3470,  -7.0615,   5.0066],
        [  8.7715,  -7.9390,   5.2900],
        [ -9.1706,   1.3863,   8.4032],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.2974,   0.8371,   2.4852],
        [ -0.8455, -10.8853,  -6.7400],
        [  5.7845,   3.6652,   0.9175],
        [  9.2979,   9.6048,   7.3138],
        [ -1.6220,   7.8917,   0.0315],
        [  7.1697,  11.5343,  -0.7276],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -0.5385,   7.9037,   6.3830],
        [  5.1292,   4.8183,  -5.7997],
        [ -1.9646,   7.8571,   3.3817],
        [  6.3266,   0.3292,   9.3502],
        [ -9.3604,   0.4855,  -3.0158],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.7298,  13.3537,  -9.4653],
        [ 11.2209,  -4.7092,   2.6117],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.0305,   3.0296,   0.1375],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -9.4159,  -2.0739,   9.6841],
        [ -5.9940,   6.1985,   5.9566],
        [  1.8721,  -4.7322,   0.5524],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.4196,   0.0478,   9.4140],
        [  2.2076,  13.8032,  -1.5099],
        [  8.2154,  -8.4262,   8.0171],
        [  8.2154,  -8.4262,   8.0171],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.5919,   0.4035,  -5.8460],
        [  6.1755,   2.5233,  -2.0042],
        [ -5.0132,  -9.2234,  -8.3531],
        [  4.7739,  -3.1218,  -4.1188],
        [  0.9945,  -4.8144,  -0.1897],
        [  4.1537,   5.9340,  -6.0423],
        [  5.1203,  -2.7151,   6.2110],
        [ -7.0825,   5.2073,  -5.0497],
        [  2.3376,  -4.6231,   9.8436],
        [ -0.3432,   7.8900,   7.1805],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.7218,  -9.9934,   6.5764],
        [ -7.6525,  -6.6534,   5.1721],
        [ -9.1706,   1.3863,   8.4032],
        [  8.2154,  -8.4262,   8.0171],
        [  9.3624,   9.5318,   0.2779],
        [  6.0073,   3.0986,   6.7937],
        [  1.5383, -11.0274,  -4.9666],
        [  6.3589,   0.8239,  -8.4230],
        [  5.5626,  -2.0536,   8.4433],
        [  1.1994,   7.6035,   2.4295],
        [  4.5954,  13.0099,  -4.4439],
        [  2.2373,  -4.6506,  -1.7334],
        [ -9.2707,   0.9661,   4.4863],
        [  6.0073,   3.0986,   6.7937],
        [  4.7482, -10.3907,   5.9149],
        [  1.8766,   7.3694,   1.1887],
        [ -0.1963,   7.8765,  -1.1371],
        [  1.9312,  13.8649,   8.3563],
        [  6.0031,  12.2958,  -6.5992],
        [  3.4401,  13.4540,   4.6576],
        [  4.5954,  13.0099,  -4.4439],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.4691,   2.8525,  -5.5110],
        [  3.0719, -10.8449,  -4.1673],
        [  2.2437,  -4.6489,   6.1430],
        [  6.3569,   0.7591,   6.5011],
        [ -9.2974,   0.8371,   2.4852],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.3303,   1.6145,  -8.4083],
        [  7.8375,  -8.7240,  -9.8913],
        [ 10.4994,  -5.9193,  -7.5153],
        [  3.1271, -10.8343,   2.6442],
        [  2.5375,   7.0676,  -4.3490],
        [  5.3858,   4.4260,  -9.6769],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.4920,  -4.1366,  -8.3955],
        [  8.8091,  -7.9037,   0.3631],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.1372,   2.6726,  -6.7885],
        [ -6.4464,   5.8272,   7.5038],
        [  1.9312,  13.8649,   8.3563]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.7371e-02, 1.0036e-01,  ..., 9.5255e-01, 3.8000e-01,
         1.1661e+00],
        [6.7371e-02, 2.1453e-04, 5.8276e-02,  ..., 9.1041e-01, 3.3684e-01,
         1.1022e+00],
        [1.0036e-01, 5.8276e-02, 0.0000e+00,  ..., 9.4672e-01, 2.8269e-01,
         1.0700e+00],
        ...,
        [9.5255e-01, 9.1041e-01, 9.4672e-01,  ..., 0.0000e+00, 1.0607e+00,
         1.1924e+00],
        [3.8000e-01, 3.3684e-01, 2.8269e-01,  ..., 1.0607e+00, 0.0000e+00,
         8.5032e-01],
        [1.1661e+00, 1.1022e+00, 1.0700e+00,  ..., 1.1924e+00, 8.5032e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.9036226 , 11.0050379 , 11.04953767, 11.27656488, 10.99326577,
       11.23201619, 10.46739073, 11.59512194,  9.0368458 ,  8.14433542,
        5.87305005,  6.81961996,  8.53014915, 13.84126197, 10.7167285 ,
        5.92952538, 13.0743399 , 12.58006038,  6.33517349, 11.04629471,
        6.46356806, 10.19036081, 11.83075005,  9.27474386,  9.87720102,
        9.3349858 , 10.91805683,  6.84796202, 13.36800073,  8.05668609,
       13.58101794,  9.5642993 ,  7.92200342,  7.03733775,  8.09899874,
        6.33517349,  9.37296102, 10.52587243, 13.86480175, 12.16901615,
       12.12448912,  6.74872981, 10.14436912,  9.64157461,  8.62260327,
        5.08910491,  5.16611782,  9.41970031, 13.97857591, 11.76830981,
       11.76830981, 10.23946247,  6.25586297, 12.59840193,  6.67108662,
       10.4977132 ,  5.70405525,  4.9160962 ,  7.24326992,  5.79562691,
        8.7907923 ,  5.1805345 ,  7.89745107, 12.46513646, 11.51557166,
       10.14044489,  9.27474386, 11.76830981, 13.3607379 ,  6.7593917 ,
       11.1341798 ,  6.41202986,  5.92952538,  7.69752262, 13.79762415,
        5.16077764,  9.32094055,  6.7593917 , 11.42420508,  7.60463349,
        7.8788965 , 13.99877413, 13.6829843 , 13.88683436, 13.79762415,
        8.83725081, 12.79126785, 11.27160035,  5.16205087,  6.4020404 ,
        9.3349858 ,  8.62162372,  6.53290663, 11.72749923, 12.05300943,
       11.27656488,  7.5092834 ,  6.97106912,  8.8353674 ,  5.41348121,
       11.83509372, 10.54658597, 12.53047489,  6.69389431,  8.68979052,
       13.99877413])}
{'P': tensor([[5.2013e-03, 3.0498e-01, 1.5976e-01,  ..., 1.4278e-03, 3.9849e-04,
         1.0289e-04],
        [1.3474e-04, 1.9894e-02, 2.2337e-02,  ..., 4.6971e-03, 2.0198e-03,
         6.5424e-04],
        [2.7376e-04, 1.4777e-02, 2.4951e-02,  ..., 5.8003e-03, 2.6440e-03,
         8.6964e-04],
        ...,
        [6.8810e-04, 9.8261e-04, 9.9040e-04,  ..., 5.5806e-02, 4.2921e-02,
         3.2439e-03],
        [3.6443e-05, 4.1751e-05, 4.2402e-05,  ..., 7.7192e-04, 5.0980e-02,
         7.8286e-03],
        [1.5436e-06, 1.7662e-06, 1.7988e-06,  ..., 5.3911e-05, 1.2407e-03,
         6.0880e-02]]), 'X': tensor([[-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0761, 0.1053,  ..., 0.6535, 0.7561, 0.8641],
        [0.0761, 0.0000, 0.0485,  ..., 0.5781, 0.6801, 0.7893],
        [0.1053, 0.0485, 0.0000,  ..., 0.5668, 0.6606, 0.7629],
        ...,
        [0.6535, 0.5781, 0.5668,  ..., 0.0000, 0.1484, 0.3049],
        [0.7561, 0.6801, 0.6606,  ..., 0.1484, 0.0000, 0.1575],
        [0.8641, 0.7893, 0.7629,  ..., 0.3049, 0.1575, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.90638066, 11.00508681, 11.00812996, 11.07637531, 11.04629471,
       11.28895489,  6.54127431, 13.44593519, 11.73397309,  7.71075226,
       12.27991991,  9.19837792,  8.62631801, 12.19805394,  8.38252937,
       13.98691426,  9.23178204,  8.37954853, 12.53345445,  9.78286733,
       12.97370371,  8.11227668,  5.29755487, 10.84493732, 13.99877413,
        5.79562691, 10.52357479,  4.74468406,  6.38399789, 11.83570132,
       13.77222752, 10.37698435, 10.8207297 , 11.56146873, 13.70893492,
       11.00812996, 10.51437947,  8.50278896,  5.26067019,  9.20340031,
        4.7663526 ,  5.94535529,  5.21280156,  9.3109884 ,  8.89458273,
       10.67132524, 12.45418236,  6.02779201, 13.88683436, 10.06023665,
        8.20436764, 13.76047999, 12.97370371,  6.23262289,  6.05113217,
        6.2134961 , 10.06437598, 11.22142195, 12.58006038,  4.78544875,
        5.24137909,  6.46356806, 13.62896593,  5.24137909,  4.76969364,
        5.76165934,  5.92952538,  9.15976746, 12.52372629,  6.2322769 ,
       11.05413545,  5.80860165, 12.40074901, 10.80728788, 11.50791859,
       13.14690137,  5.94541803,  5.29755487,  6.1561646 ,  7.59795445,
        7.5984809 , 13.6922425 ,  5.54776934, 10.86693708, 12.5085995 ,
       13.56367857, 13.59480906,  5.14999799,  6.32572323, 13.89974693,
        8.31625171,  8.62631801, 10.30972193,  5.19986797, 13.70893492,
       14.10045797, 10.31886066, 10.00410625, 13.58101794, 11.35182589,
        6.81117802, 11.1913598 ,  8.14433542,  5.08910491,  6.02588517,
       12.52875793])}
{'P': tensor([[5.0079e-03, 2.6796e-01, 2.2519e-01,  ..., 1.0548e-07, 6.0568e-06,
         2.4139e-06],
        [1.7728e-04, 2.2802e-02, 1.6722e-02,  ..., 6.5992e-07, 4.0519e-05,
         1.4279e-05],
        [1.8919e-04, 1.6200e-02, 2.1064e-02,  ..., 5.2329e-07, 3.8342e-05,
         1.1105e-05],
        ...,
        [4.7048e-08, 4.2650e-08, 3.7592e-08,  ..., 1.1943e-02, 9.1037e-07,
         1.5026e-04],
        [1.1432e-06, 1.1081e-06, 1.1240e-06,  ..., 9.7948e-04, 9.2517e-02,
         6.6680e-04],
        [7.1317e-06, 6.9894e-06, 6.1163e-06,  ..., 1.5131e-02, 9.9095e-05,
         9.3646e-02]]), 'X': tensor([[-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0632, 0.0677,  ..., 1.4685, 1.0880, 1.0849],
        [0.0632, 0.0000, 0.0309,  ..., 1.4187, 1.0284, 1.0375],
        [0.0677, 0.0309, 0.0000,  ..., 1.4405, 1.0211, 1.0607],
        ...,
        [1.4685, 1.4187, 1.4405,  ..., 0.0000, 1.2469, 0.3894],
        [1.0880, 1.0284, 1.0211,  ..., 1.2469, 0.0000, 1.0710],
        [1.0849, 1.0375, 1.0607,  ..., 0.3894, 1.0710, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.91601655, 11.00726483, 11.00694688, 10.98076873, 11.05317597,
       11.10066126,  7.06372172, 12.07225469, 11.73848065,  6.81084724,
        9.63742669,  6.76661595,  4.98461983,  7.85919687,  6.27905292,
        9.44486609, 10.84437705,  6.65368551,  4.98461983, 10.06437598,
        8.87007495, 11.62158554,  5.92952538, 11.92133129, 12.19805394,
        8.06643487,  7.67939546,  5.98285319, 12.22503754, 14.02532758,
        7.32233733, 13.95472331,  9.75075273,  6.42689705, 13.1399181 ,
        5.14607014, 11.74901778, 13.60275211, 10.1371552 ,  5.04028538,
       12.5085995 , 10.794755  ,  7.50500064,  7.72668467, 12.16901615,
       14.02526505,  4.75303127,  7.15990676,  9.41244865, 12.32545637,
        7.01475376,  9.3349858 , 12.16901615,  7.98469166,  9.49511524,
        8.90405823, 12.3489324 ,  5.18277858,  7.8575648 ,  6.39520397,
        4.89869205,  7.09887557, 10.51437947,  9.32094055, 10.1371552 ,
       12.91143491,  7.60463349,  5.5205189 ,  5.23878371, 13.89974693,
       10.00794468, 13.20988996, 10.23946247,  9.58347456,  4.75303127,
       10.06703358, 13.97430191,  9.41244865,  5.13182506,  5.13125513,
       13.77222752, 11.92133129, 13.66991027,  9.04255655,  9.94173708,
       13.89974693, 12.95702647, 10.1371552 , 11.66453059, 10.14436912,
        6.60752373,  5.70591986, 10.31613269, 12.72524615, 13.70893492,
       11.05672136, 10.42667093, 13.69162795,  9.50654967, 13.36092624,
        8.62329466,  9.60412735,  5.16611782, 13.68063098, 13.103709  ,
        7.16747878])}
{'P': tensor([[8.2854e-03, 1.8134e-01, 1.7577e-01,  ..., 1.8138e-07, 2.1865e-04,
         3.9227e-07],
        [3.1920e-05, 1.1899e-02, 1.1923e-01,  ..., 2.8554e-07, 3.3891e-04,
         4.9369e-07],
        [1.5849e-05, 4.3593e-04, 7.0529e-03,  ..., 1.9814e-07, 2.5762e-04,
         3.1959e-07],
        ...,
        [9.2930e-06, 4.1412e-06, 4.4652e-06,  ..., 1.4589e-01, 1.2370e-05,
         4.6050e-02],
        [1.6267e-06, 4.5813e-06, 5.7281e-06,  ..., 1.5482e-05, 1.8470e-02,
         3.5221e-06],
        [5.5072e-05, 1.8261e-05, 1.8588e-05,  ..., 3.0635e-04, 8.1531e-06,
         5.3406e-02]]), 'X': tensor([[-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 1.5955e-01, 1.4454e-01,  ..., 1.1199e+00, 1.0654e+00,
         7.2636e-01],
        [1.5955e-01, 0.0000e+00, 1.6767e-02,  ..., 1.0932e+00, 1.0210e+00,
         7.2573e-01],
        [1.4454e-01, 1.6767e-02, 0.0000e+00,  ..., 1.0877e+00, 1.0173e+00,
         7.1719e-01],
        ...,
        [1.1199e+00, 1.0932e+00, 1.0877e+00,  ..., 0.0000e+00, 1.3773e-01,
         4.0935e-01],
        [1.0654e+00, 1.0210e+00, 1.0173e+00,  ..., 1.3773e-01, 0.0000e+00,
         3.9942e-01],
        [7.2636e-01, 7.2573e-01, 7.1719e-01,  ..., 4.0935e-01, 3.9942e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.91805683, 11.21703654, 11.24434374, 11.20100795, 11.28215429,
       11.35182589,  6.7593917 ,  8.25828409,  6.24187044, 13.96905756,
        5.80860165, 12.49291768,  7.62509582, 13.00380624,  4.9160962 ,
        9.62958973,  9.68323686,  7.5092834 , 11.80092578, 11.89105259,
        5.70405525,  8.04483495, 13.97857591, 12.59840193, 13.34587027,
       11.65923396,  7.06372172,  7.5092834 ,  7.34250948,  6.2134961 ,
        7.58668144,  6.69389431, 12.79126785,  4.79003361,  9.66977802,
        8.06517143, 14.10045797,  6.10774544,  6.81084724,  8.90405823,
        5.79543038, 10.68031351, 12.00995898,  6.99097013,  5.47702766,
        6.10774544, 11.00812996,  7.58668144, 13.54774355, 10.48840476,
        8.98765644, 13.61801526, 12.32408516,  7.53063885,  5.18277858,
        9.78286733,  8.57560275, 13.89886469, 13.99048094, 12.97370371,
       14.03505624,  5.07306274,  7.83534987, 11.8159254 , 12.4217523 ,
        9.91668182,  9.55720651, 10.51437947,  4.79003361, 10.8207297 ,
       13.0493368 ,  4.89121805,  8.01009083, 12.79126785,  9.23071698,
        8.06990483, 11.70168755,  5.76165934, 11.24434374,  9.75200848,
        7.19534146, 13.61600536,  9.48337063, 11.74901778, 13.37121923,
       11.89105259,  5.39098669,  4.85995002,  7.39599986,  8.79284466,
        5.37343173,  6.81961996,  5.05534609,  9.18358562, 12.30695098,
       12.10942412,  5.94541803,  6.39520397,  7.48311002,  6.54127431,
       11.28895489,  7.5403822 , 13.25336001,  8.1303945 , 11.85743425,
        8.66273171])}
{'P': tensor([[1.0238e-02, 2.2720e-01, 1.9143e-01,  ..., 1.0775e-06, 1.4915e-04,
         6.3848e-03],
        [1.2111e-04, 2.2746e-02, 1.0895e-01,  ..., 2.3412e-06, 7.2766e-04,
         4.7767e-02],
        [2.2171e-05, 2.6760e-04, 8.7954e-03,  ..., 1.0966e-06, 1.9206e-04,
         9.3884e-03],
        ...,
        [4.4419e-04, 1.9767e-04, 1.6197e-04,  ..., 2.7276e-01, 7.3711e-05,
         2.3115e-05],
        [3.0347e-05, 4.4807e-05, 3.9838e-05,  ..., 1.9616e-04, 1.0204e-01,
         2.4854e-04],
        [1.4335e-05, 3.1929e-05, 5.8262e-05,  ..., 8.0689e-06, 1.4268e-02,
         5.5792e-02]]), 'X': tensor([[-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1258e-01, 1.5151e-01,  ..., 1.8843e-01, 9.0913e-01,
         5.5189e-01],
        [1.1258e-01, 0.0000e+00, 1.1041e-01,  ..., 1.2772e-01, 7.9732e-01,
         4.3963e-01],
        [1.5151e-01, 1.1041e-01, 0.0000e+00,  ..., 4.0079e-02, 8.2657e-01,
         4.7017e-01],
        ...,
        [1.8843e-01, 1.2772e-01, 4.0079e-02,  ..., 2.1453e-04, 7.9433e-01,
         4.3982e-01],
        [9.0913e-01, 7.9732e-01, 8.2657e-01,  ..., 7.9433e-01, 0.0000e+00,
         3.5850e-01],
        [5.5189e-01, 4.3963e-01, 4.7017e-01,  ..., 4.3982e-01, 3.5850e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.92597196, 11.1341798 , 11.24434374, 11.28215429, 11.21703654,
       11.20100795,  5.90035324,  5.05444288, 13.44593519, 11.59512194,
       10.24557047, 11.31276109, 13.45642379,  7.48538905,  5.94909709,
       12.44177641,  5.19577146, 13.3607379 , 13.01187484, 10.7167285 ,
       12.33167382,  6.06416957, 13.06498554, 12.95702647, 12.50996828,
       12.52875793, 13.59673406, 10.84437705, 13.6105595 ,  4.98461983,
       11.31276109, 10.06703358, 11.85743425, 11.63724792, 14.02058621,
        9.69792513, 13.32374056,  7.10378919, 10.27572673,  9.36093562,
       12.38881635,  9.88571713,  7.38844703,  9.17016857, 13.27994304,
        9.20340031, 11.32149259,  7.16598273,  5.19577146,  7.57946459,
        6.28961988, 13.38091121, 11.62158554,  5.13616224,  4.91336994,
        6.5618968 , 12.52372629,  8.7907923 ,  5.01114719,  9.50913849,
       13.62896593,  6.99097013,  6.67218687, 11.83570132,  7.20482224,
        7.72415048,  7.6374603 , 12.62576026, 10.31584102,  9.48337063,
       13.35832725,  5.10446653, 13.56367857,  9.94285511, 10.19036081,
        5.05534609,  4.85995002,  9.87720102, 13.6105595 ,  8.31625171,
        4.98461983, 10.46238054, 11.04313641,  9.20340031,  9.75075273,
       12.53047489, 11.99633819,  7.02496223,  5.53175275, 11.05672136,
       10.95593729,  7.80625981,  5.57341623,  9.85330646,  6.12886024,
        9.49511524,  4.92497514, 13.3614598 , 13.12299145,  7.26074925,
       10.35093485, 10.27716434, 11.66453059,  8.49161154,  6.2134961 ,
       11.77124373])}
{'P': tensor([[5.5563e-03, 3.3403e-01, 2.4258e-01,  ..., 1.2900e-03, 5.0675e-07,
         2.3587e-04],
        [8.4878e-05, 8.4185e-03, 2.8208e-01,  ..., 2.1573e-03, 7.4455e-07,
         4.6854e-04],
        [6.6586e-05, 1.2697e-04, 1.1655e-02,  ..., 3.3077e-03, 9.6929e-07,
         1.0306e-03],
        ...,
        [1.3793e-04, 1.5489e-04, 1.9322e-04,  ..., 6.7155e-02, 1.3556e-04,
         1.3991e-02],
        [2.7568e-04, 2.2531e-04, 1.5123e-04,  ..., 1.2565e-04, 5.0548e-02,
         3.0280e-06],
        [9.4974e-07, 1.0304e-06, 1.2060e-06,  ..., 1.5128e-04, 1.2154e-05,
         2.8553e-02]]), 'X': tensor([[ -0.4341, -10.9473,   7.4509],
        [  0.1252, -11.0062,   7.2564],
        [  1.1644, -11.0394,   7.8511],
        [  2.1440, -10.9811,   8.5281],
        [  3.1756, -10.8247,   6.8882],
        [  2.6078, -10.9229,   6.2869],
        [  9.2979,   9.6048,   7.3138],
        [ -6.4002,  -8.1048,  -0.9108],
        [  3.8182,  -3.9347,  -1.7961],
        [  9.4554,   9.4245,   9.8800],
        [  9.6662,   9.1725,   8.1056],
        [ 12.4837,   2.7510,   0.4449],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -8.5900,   2.9767,   7.4811],
        [ -4.3963,   7.1629,   4.5263],
        [  6.1222,  -0.7261,   8.8204],
        [ -8.7409,  -4.6837,  -6.5840],
        [  1.7122,  -4.7591,  -3.6734],
        [  0.6933,   7.7336,   2.1179],
        [  2.9470,  13.6085,   2.9809],
        [ -0.8675, -10.8815,   6.8561],
        [  3.5939,   6.4029,  -1.8357],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.5149,  -5.1905,   1.5239],
        [ -7.5603,   4.6459,   8.2002],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.1292,   4.8183,  -5.7997],
        [ -3.7964,  -9.9294,   7.7147],
        [ -6.1899,  -8.3011,  -7.9269],
        [  0.2480, -11.0153,  -3.0183],
        [  2.0449,  13.8402,  -7.1642],
        [  2.6613,   7.0021,   4.5783],
        [ -8.5607,  -5.0938,  -5.4531],
        [  6.3515,   0.6418,  -6.5823],
        [  1.7000,  -4.7609,   2.7935],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.0604,   1.7707,   1.1066],
        [  6.3413,   0.4883,  -6.8416],
        [  7.7794,  11.0629,   8.9069],
        [ -8.9942,  -3.9990,   6.0448],
        [  5.5294,  -2.1106,   9.0626],
        [  7.8425,  -8.7202,   6.4530],
        [  0.6470, -11.0352,  -1.1931],
        [  9.7328,   9.0902,  -0.8917],
        [  1.4150,  13.9647,  -4.0915],
        [  6.3119,   1.7797,   9.9186],
        [  8.3718,  -8.2953,   8.4666],
        [ -8.0750,  -6.0072,  -5.8267],
        [  9.0424,  -7.6779,  -7.5286],
        [  0.4119,  14.1019,  -4.1870],
        [  6.0054,  -1.0902,   8.3079],
        [ -9.0604,   1.7707,   1.1066],
        [  5.6248,   3.9984,   8.4933],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 11.7726,  -3.4876,  -6.7402],
        [  7.0358,  -9.2784,  -9.8479],
        [  4.4049,   5.6892,  -4.4394],
        [ 11.5966,   5.9666,   6.0083],
        [  2.8602,   6.8905,   5.8779],
        [  0.6952,  -4.8060,   8.8976],
        [ 12.2914,   3.8037,   6.0978],
        [  2.3367,   7.1676,  -9.5263],
        [  1.3705,  13.9723,   7.9266],
        [ 10.1503,   8.5414,  -7.2681],
        [  3.5939,   6.4029,  -1.8357],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.1519, -10.2387,   9.8191],
        [  3.8182,  -3.9347,  -1.7961],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.4351,  -1.8891,   3.5642],
        [  4.9531,   5.0576,   5.6794],
        [ -8.5149,  -5.1905,   1.5239],
        [ -3.2486,   7.5925,  -1.3719],
        [  9.1820,   9.7334,  -2.7932],
        [  3.0876,   6.7527,  -5.3540],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.5375,   7.0676,  -4.3490],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.1203,  -2.7151,   6.2110],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.1989,  -9.7591,   0.8289],
        [  6.3569,   0.7591,   6.5011],
        [  2.8994,   6.8675,  -9.5987],
        [  0.5250, -11.0306,   5.5358],
        [  1.2273,  -4.8081,   9.8129],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.4773,  -1.0024,   5.2599],
        [ -6.2200,  -8.2737,   5.5079],
        [  0.8700,  -4.8132,  -7.8042],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.4296,  -1.9458,   4.0833],
        [ 11.2517,  -4.6495,   2.9800],
        [  6.3124,   0.2057,  -0.4593],
        [  2.6613,   7.0021,   4.5783],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.4501,  -1.7133,   2.2980],
        [  0.1041, -11.0045,   2.4311],
        [ -4.4520,   7.1370,  -8.7867],
        [ 10.5678,   7.9261,  -0.6100],
        [ -8.3062,  -5.6008,   0.0287],
        [  6.5428,  -9.5728,  -2.9897],
        [ -1.6334, -10.7207,   7.1397],
        [  4.5325,  -3.3643,   6.7699],
        [ -7.4733,   4.7555,   6.2490],
        [ 12.3759,  -1.3943,   6.4659]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0304, 0.0851,  ..., 0.7326, 0.1746, 0.7779],
        [0.0304, 0.0000, 0.0568,  ..., 0.7023, 0.1576, 0.7479],
        [0.0851, 0.0568, 0.0000,  ..., 0.6541, 0.1639, 0.6937],
        ...,
        [0.7326, 0.7023, 0.6541,  ..., 0.0000, 0.6268, 0.1835],
        [0.1746, 0.1576, 0.1639,  ..., 0.6268, 0.0000, 0.7086],
        [0.7779, 0.7479, 0.6937,  ..., 0.1835, 0.7086, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.95593729, 11.00694688, 11.10066126, 11.18839169, 11.28093507,
       11.22993218, 13.36800073, 10.32715715,  5.48276038, 13.35013374,
       13.32557133, 12.78327037, 12.00995898,  9.09119351,  8.40445872,
        6.16512856,  9.91668182,  5.05774295,  7.76457095, 13.92390689,
       10.91601655,  7.34250948,  7.53063885,  9.97220799,  8.8737556 ,
        9.73579738,  7.03733775, 10.63038565, 10.35485331, 11.018084  ,
       13.99048094,  7.49076899,  9.96152788,  6.383897  ,  5.05534609,
        8.29022197,  9.23178204,  6.36004393, 13.52430176,  9.84314712,
        5.91854397, 11.72802491, 11.05413545, 13.31764181, 14.03618649,
        6.55800267, 11.785566  , 10.06437598, 11.86239961, 14.10796466,
        6.10360073,  9.23178204,  6.90115901, 12.27835996, 12.27835996,
       11.64436684,  7.19513806, 13.04156075,  7.46053281,  4.85603886,
       12.86648771,  7.53884104, 14.0393949 , 13.26590071,  7.34250948,
        9.26455795, 11.46176259,  5.48276038, 10.63038565,  9.62238828,
        7.07902036,  9.97220799,  8.25828409, 13.38091121,  7.42512621,
       12.41366754,  7.5092834 , 10.00794468,  5.79391825,  5.79562691,
       12.30695098, 11.56146873,  6.4020404 ,  7.4544866 , 11.04313641,
        4.96231532,  5.39098669,  9.53015376, 10.35093485,  4.89121805,
       12.4980845 ,  9.6282763 , 12.17451685,  6.31576264,  7.49076899,
       10.78533838,  9.60412735, 11.0050379 ,  8.41171369, 13.20988996,
       10.01804223, 11.59512194, 10.84437705,  5.64464807,  8.8580671 ,
       12.45418236])}
{'P': tensor([[3.8405e-03, 2.4732e-01, 1.9310e-01,  ..., 1.8780e-05, 1.1257e-06,
         5.5872e-07],
        [6.4534e-05, 5.4103e-03, 1.3701e-01,  ..., 2.9242e-05, 1.6284e-06,
         7.3492e-07],
        [5.0349e-05, 1.1772e-04, 5.2324e-03,  ..., 3.7712e-05, 1.8298e-06,
         6.4587e-07],
        ...,
        [5.0537e-07, 5.1105e-07, 5.5118e-07,  ..., 3.6271e-02, 4.8317e-03,
         2.6590e-06],
        [8.5161e-06, 7.9712e-06, 7.8209e-06,  ..., 1.3154e-05, 5.1074e-02,
         5.7069e-04],
        [1.0409e-02, 9.0678e-03, 7.6658e-03,  ..., 6.7141e-06, 2.4265e-05,
         9.7359e-02]]), 'X': tensor([[ -0.1626, -10.9796,   6.4629],
        [  0.1287, -11.0065,   6.5757],
        [  0.6363, -11.0348,   6.1414],
        [  2.6078, -10.9229,   6.2869],
        [  3.1756, -10.8247,   6.8882],
        [  3.6249, -10.7255,   5.6605],
        [  5.8318,   3.5567,  -4.7692],
        [ -8.8400,   2.3991,   1.3457],
        [  5.2199,   4.6861,  -1.9304],
        [  6.9314,  -9.3435,  -1.2729],
        [  3.5939,   6.4029,  -1.8357],
        [ -0.8675, -10.8815,   6.8561],
        [  1.1644, -11.0394,   7.8511],
        [  5.4242, -10.1261,   3.9595],
        [  5.6414, -10.0303,   7.8525],
        [  1.1994,   7.6035,   2.4295],
        [  2.5382,   7.0672,   4.5097],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.2242,  -0.3171,   7.4978],
        [ -3.4100,   7.5436,  -7.6980],
        [ -4.7389,   6.9953,   3.7638],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.8902,  -1.3887,   4.0388],
        [ -7.9667,   4.0814,  -0.6397],
        [ 10.6283,  -5.7253,   7.4878],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.1886,  -2.6250,   6.4677],
        [  1.4749,  -4.7892,  -0.8089],
        [  9.8082,  -6.8396,   4.9615],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.7784,  -1.6401,   9.5587],
        [  3.2366,  -4.2732,   5.1532],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.4119,  14.1019,  -4.1870],
        [ -1.6275, -10.7221,   3.2376],
        [  5.5240,  -2.1198,   6.2055],
        [ -7.5603,   4.6459,   8.2002],
        [ -9.3107,  -2.7697,  -3.2919],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.3124,   0.2057,  -0.4593],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.1771, -10.9776,  -3.2007],
        [ 12.5258,  -0.4200,  -9.3493],
        [  2.6078, -10.9229,   6.2869],
        [  3.3633,  13.4794,   0.5193],
        [  4.7482, -10.3907,   5.9149],
        [ -0.8675, -10.8815,   6.8561],
        [  2.9470,  13.6085,   2.9809],
        [ -5.6423,   6.4537,  -1.2091],
        [  5.2188,  -2.5839,   2.1833],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -0.1963,   7.8765,  -1.1371],
        [  6.2883,   0.0325,   5.2692],
        [  4.2339,  -3.6268,   6.3125],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 12.1588,   4.3364,  -8.0867],
        [ 12.4691,   2.8525,  -5.5110],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.5763,   0.1333,  -9.2650],
        [ -3.9700,   7.3453,  -4.1486],
        [ -4.7847,  -9.3729,  -3.7741],
        [  1.7301,   7.4262,   5.0522],
        [  7.8982,  -8.6778,   0.0239],
        [  2.2182,  13.8007,   5.1889],
        [  6.3587,   1.1596,  -3.6629],
        [  6.2770,  -0.0379,   2.4195],
        [  5.6022,  -1.9833,   8.3081],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -0.5385,   7.9037,   6.3830],
        [  3.3633,  13.4794,   0.5193],
        [ -3.2053, -10.1993,  -3.8736],
        [  2.5382,   7.0672,   4.5097],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.2849, -10.9655,  -7.6542],
        [  6.0310,  12.2795,  -6.4528],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.6314,  -1.9302,  -9.6543],
        [  6.3549,   0.7096,   6.9198],
        [ -0.6323,   7.9085,  -5.9544],
        [ 12.3453,  -1.5462,   7.0212],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.8450,  -9.9356,   3.0767],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.9178,   2.1936,   7.5388],
        [ -8.5859,  -5.0393,   0.5476],
        [ -1.0482, -10.8487,  -5.9066],
        [  5.1196,  -2.7160,   4.2705],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -0.9715, -10.8630,   0.3555],
        [ -9.0574,   1.7801,   8.8986],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.7328,   9.0902,  -0.8917],
        [ 12.4837,   2.7510,   0.4449],
        [  1.3339,   7.5627,  -0.9034],
        [ -9.4296,  -1.9458,   4.0833]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0268, 0.0539,  ..., 1.0123, 0.8269, 0.3060],
        [0.0268, 0.0000, 0.0336,  ..., 0.9946, 0.8166, 0.3273],
        [0.0539, 0.0336, 0.0000,  ..., 0.9611, 0.7840, 0.3293],
        ...,
        [1.0123, 0.9946, 0.9611,  ..., 0.0000, 0.3467, 1.0741],
        [0.8269, 0.8166, 0.7840,  ..., 0.3467, 0.0000, 0.8032],
        [0.3060, 0.3273, 0.3293,  ..., 1.0741, 0.8032, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.98076873, 11.00726483, 11.05317597, 11.22993218, 11.28093507,
       11.32149259,  6.83083863,  9.15976746,  7.01475376, 11.63383221,
        7.34250948, 10.91601655, 11.10066126, 11.48734978, 11.50791859,
        7.69752262,  7.50917551,  8.47371868,  4.90505566,  6.2322769 ,
        8.27853076,  8.44940661, 12.26997165,  6.05165248,  8.95133961,
       12.07225469, 11.75807252,  5.81481044,  5.01114719, 11.95743558,
        5.56809335,  6.00663314,  5.36061974, 10.33712258,  4.85995002,
       13.06498554, 14.10796466, 10.84493732,  5.91677188,  8.8737556 ,
        9.71392179,  4.90505566,  6.24187044,  6.31576264,  5.04028538,
       13.04156075, 12.50996828,  9.85384623, 11.1913598 , 12.53285633,
       11.22993218, 13.89264822, 11.42420508, 10.91601655, 13.92390689,
        8.5724042 ,  5.82345867, 12.1460931 ,  7.8788965 ,  6.28835101,
        5.5748624 , 11.99633819,  9.5642993 , 12.9089559 , 12.79126785,
       13.79762415, 12.5769659 ,  8.34949383, 10.52357479,  7.62509582,
       11.73397309, 13.97779839,  6.46356806,  6.27715105,  5.94293185,
       10.30972193,  7.92200342, 13.89264822, 10.6910768 ,  7.50917551,
       12.84056604, 11.20100795, 13.68063098,  9.51074763, 10.46739073,
        9.94173708,  5.95297459,  6.39438044,  7.9337647 , 12.44177641,
        4.9160962 , 11.52733581, 11.97949273,  9.18358562,  9.95552626,
       10.8992524 ,  5.79543038, 12.49291768, 10.90638066,  9.23071698,
       13.77075174,  9.66977802, 13.31764181, 12.78327037,  7.67939546,
        9.6282763 ])}
{'P': tensor([[6.1792e-03, 4.4867e-01, 1.7961e-01,  ..., 9.9763e-06, 3.5313e-06,
         7.9794e-03],
        [7.5776e-05, 9.3602e-03, 3.0900e-01,  ..., 1.1248e-05, 5.9116e-06,
         1.4859e-02],
        [7.6618e-05, 1.0888e-04, 2.1981e-02,  ..., 9.5077e-06, 2.2196e-05,
         2.8705e-02],
        ...,
        [9.6927e-03, 1.0664e-02, 6.9204e-03,  ..., 6.5058e-02, 4.3637e-06,
         1.4232e-02],
        [1.1422e-05, 1.0281e-05, 8.3175e-06,  ..., 6.7583e-06, 5.3334e-02,
         3.2879e-06],
        [1.3017e-05, 1.6887e-05, 4.9696e-05,  ..., 2.6364e-06, 2.5505e-05,
         2.0467e-02]]), 'X': tensor([[ -0.0254, -10.9932,   3.2200],
        [  0.5960, -11.0335,   3.1144],
        [  2.6310, -10.9195,   3.8054],
        [  3.1271, -10.8343,   2.6442],
        [  4.5814, -10.4485,   2.5496],
        [  3.5284, -10.7484,   5.3347],
        [ -5.1046,  -9.1610,   2.3418],
        [ -5.6792,  -8.7352,   4.0318],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.5198,  -4.0923,   5.5088],
        [  7.9141,  -8.6656,   9.6948],
        [ 11.0864,   7.0387,  -0.1238],
        [  0.6470, -11.0352,  -1.1931],
        [ -9.1813,  -3.3550,   1.8827],
        [  9.2979,   9.6048,   7.3138],
        [ 11.6165,  -3.8715,   6.4594],
        [ -4.8751,  -9.3148,   4.0406],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.1755,   2.5233,  -2.0042],
        [  6.3301,   0.3638,  -7.5685],
        [  3.3839,  -4.1966,   8.3574],
        [ -7.4808,   4.7462,   2.0266],
        [  5.7802,  -1.6363,  -6.2782],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.8631,  -4.3728,  -1.1746],
        [  9.0129,  -7.7072,   9.1400],
        [ -1.0012, -10.8576,   2.2867],
        [  9.4925,   9.3811,   8.9842],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.9217,   6.8979,  -1.7667],
        [ -8.1531,  -5.8748,   2.6934],
        [ -5.0550,   6.8231,  -9.2502],
        [  5.3118,   4.5452,   4.2114],
        [  0.2571,  -4.7594,   1.5618],
        [  9.5261,  -7.1675,   1.4206],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.9809,   7.6642,  -1.4217],
        [ -5.9872,   6.2037,  -9.4477],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.7232,  -4.7574,   3.1270],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.1477,   7.2550,   8.9688],
        [ 11.9019,  -3.1379,  -3.4534],
        [  9.3424,  -7.3685,  -3.6807],
        [  4.5491,  -3.3485,   6.8412],
        [  2.4639, -10.9431,  -7.3718],
        [  2.0899,  -4.6870,  -7.4260],
        [ -7.4807,   4.7463,   2.0616],
        [ 11.1420,   6.9331,  -2.4779],
        [ -4.7334,   6.9982,  -0.4392],
        [ 12.4592,  -0.9168,  -6.7571],
        [  4.3459,  -3.5327,   0.3667],
        [  2.0870,  -4.6877,   9.6823],
        [  2.1477,   7.2550,   8.9688],
        [  6.3549,   0.7096,   6.9198],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.6544,   5.8273,  -3.1375],
        [ -1.5451,   7.8975,   7.6104],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.7796,  -3.4696,   9.0088],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -0.5385,   7.9037,   6.3830],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.9193,  -1.3176,   5.1739],
        [ 12.5199,  -0.4711,   3.0436],
        [  1.6308,  13.9254,  -2.0984],
        [  3.3744,   6.5623,  -6.3636],
        [  0.9809,  -4.8145,   6.8706],
        [  6.3497,   1.3680,  -1.9076],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.0617,  12.7957,   8.3544],
        [ -3.4488, -10.0934,   2.5195],
        [  0.1931,  -4.7491,   7.4385],
        [ 11.7418,   5.6070,  -1.4192],
        [ -2.2508, -10.5479,  -2.7235],
        [ -7.0825,   5.2073,  -5.0497],
        [  4.9531,   5.0576,   5.6794],
        [  1.3103,  -4.8032,  -0.9300],
        [ -7.6293,  -6.6859,  -5.5161],
        [  7.3724,  -9.0577,  -8.9843],
        [  4.3377,   5.7570,   1.3712],
        [ 11.0864,   7.0387,  -0.1238],
        [ -9.3464,   0.5708,  -1.6397],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.4690,  -0.8528,  -9.3374],
        [  0.1206,  -4.7363,   8.7103],
        [ -0.0410,   7.8591,   9.3457],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.9314,  -9.3435,  -1.2729],
        [ -8.8835,  -4.3178,   9.3080],
        [ -1.1885, -10.8211,   0.0865],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.5763,   0.1333,  -9.2650],
        [ -0.8569,   7.9156,  -6.5557],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.3858,   4.4260,  -9.6769],
        [ 11.9122,   5.1397,  -1.5498],
        [  3.2647, -10.8066,  -0.3768],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -4.8665,  -9.3204,  -5.2564],
        [  2.0243,   7.3086,   7.5432],
        [  6.9314,  -9.3435,  -1.2729]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0437, 0.1506,  ..., 0.7155, 0.5726, 0.5490],
        [0.0437, 0.0000, 0.1135,  ..., 0.7173, 0.5382, 0.5133],
        [0.1506, 0.1135, 0.0000,  ..., 0.7970, 0.4248, 0.4939],
        ...,
        [0.7155, 0.7173, 0.7970,  ..., 0.0000, 1.1557, 0.6196],
        [0.5726, 0.5382, 0.4248,  ..., 1.1557, 0.0000, 0.6467],
        [0.5490, 0.5133, 0.4939,  ..., 0.6196, 0.6467, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([10.99326577, 11.04953767, 11.23201619, 11.27656488, 11.4088043 ,
       11.31276109, 10.48721382, 10.41909211, 11.63502679, 12.22503754,
       11.73567759, 13.13205357, 11.05413545,  9.7751189 , 13.36800073,
       12.24467041, 10.51340829,  8.34949383,  6.67108662,  6.34059021,
        5.39098669,  8.859416  ,  6.00732193, 12.52875793,  9.88310614,
       11.85890664, 10.9036226 , 13.34587027,  9.78147542,  8.29022197,
        8.47371868, 10.04916678,  8.49161154,  6.99097013,  4.7663526 ,
       11.92133129, 12.30695098,  7.72668467,  8.62162372, 12.00995898,
        9.99145096, 10.35093485,  5.05989547, 10.41909211,  7.56617037,
       12.30858677, 11.89855267,  5.64865173, 11.21703654,  5.13182506,
        8.85939626, 13.12299145,  8.4486771 , 12.49291768,  5.60064319,
        5.13125513,  7.56617037,  6.39438044,  4.85603886, 13.03002338,
        8.04718433,  9.41244865, 12.27993098,  9.71392179,  7.92200342,
       12.26997165,  6.06416957, 12.52875793, 14.02058621,  7.37902766,
        4.91336994,  6.49538841,  6.10774544, 13.76047999, 10.66631784,
        4.75303127, 13.01187484, 10.78533838,  8.7907923 ,  7.07902036,
        4.978704  , 10.14436912, 11.67875855,  7.20826219, 13.13205357,
        9.36378438, 12.4217523 , 12.4980845 ,  4.73784222,  7.85919687,
        5.18277858, 11.63383221,  9.87720102, 10.88618114,  5.19577146,
       12.52875793, 12.5769659 ,  7.96181288,  5.53255882,  6.97106912,
       12.97370371, 11.28895489, 10.0674887 , 10.51437947,  7.58378655,
       11.63383221])}
{'P': tensor([[9.3292e-03, 2.6229e-01, 1.6223e-01,  ..., 5.3076e-04, 1.1423e-03,
         1.1546e-05],
        [2.1525e-05, 7.7796e-03, 7.0055e-04,  ..., 2.2100e-04, 1.7086e-03,
         1.8290e-05],
        [6.6827e-05, 5.0096e-02, 1.3084e-02,  ..., 2.8475e-04, 2.1649e-03,
         2.1484e-05],
        ...,
        [1.0251e-03, 2.0563e-03, 1.3077e-03,  ..., 1.0248e-01, 2.9594e-02,
         3.5726e-03],
        [2.0013e-04, 3.3731e-04, 2.8994e-04,  ..., 2.9067e-03, 2.0498e-01,
         2.5461e-02],
        [1.0936e-05, 9.0246e-06, 7.7735e-06,  ..., 2.6154e-04, 2.6002e-04,
         8.1732e-02]]), 'X': tensor([[  0.1041, -11.0045,   2.4311],
        [  3.1271, -10.8343,   2.6442],
        [  2.6310, -10.9195,   3.8054],
        [  4.5814, -10.4485,   2.5496],
        [  0.5960, -11.0335,   3.1144],
        [  4.8051, -10.3703,   0.5268],
        [  6.3534,   1.2990,  -7.2942],
        [ -9.2301,  -3.1537,  -2.7477],
        [  0.9809,   7.6642,  -1.4217],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -8.7078,   2.7182,   1.8146],
        [  2.3367,   7.1676,  -9.5263],
        [ 12.1940,  -2.1881,   7.1499],
        [ 11.7796,  -3.4696,   9.0088],
        [  0.5175,  14.0910,   4.6540],
        [ -8.8154,   2.4611,   2.5198],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.8934,   2.2597,   3.5831],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.7275,  14.0667,  -0.9815],
        [  4.8051, -10.3703,   0.5268],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.1677,  -4.7447,  -6.1529],
        [  2.6614,  -4.5192,  -9.3641],
        [ 10.2470,   8.4054,  -0.1177],
        [  5.6523,   3.9443,  -2.9449],
        [  0.9659,  -4.8144,  -0.9594],
        [  9.9756,  -6.6330,  -3.3429],
        [ 10.0995,  -6.4739,   2.0974],
        [ -2.4956, -10.4682,   3.5632],
        [  3.7237,  -3.9966,  -5.0376],
        [  8.3718,  -8.2953,   8.4666],
        [  2.3261,  13.7749,   5.9121],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.0829,  -5.9940,   5.0564],
        [  9.3753,   9.5171,   3.9131],
        [ 11.5083,  -4.1177,   0.2389],
        [ -6.8466,  -7.6488,   1.9648],
        [ -5.6169,  -8.7844,   8.9057],
        [  9.9217,  -6.7006,  -5.6728],
        [ -0.6323,   7.9085,  -5.9544],
        [ -8.8154,   2.4611,   2.5198],
        [ -3.5173,   7.5091,  -8.9585],
        [  1.7232,  -4.7574,   3.1270],
        [ 12.4597,  -0.9136,  -4.8763],
        [  7.5784,  -8.9141,   5.8410],
        [  9.0652,   9.8597,  -3.5949],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -7.4807,   4.7463,   2.0616],
        [  5.9655,   3.2172,  -7.5953],
        [  5.6314,  -1.9302,  -9.6543],
        [  9.2692,   9.6370,   7.0746],
        [  6.3266,   0.3292,   9.3502],
        [  1.4489,   7.5256,   4.4529],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.6703,   6.3443,  -5.4401],
        [ -9.3616,  -2.4750,   3.7765],
        [ -7.4887,   4.7364,   8.9074],
        [  2.8474,   6.8979,   6.6457],
        [ -0.8569,   7.9156,  -6.5557],
        [ -3.3116, -10.1539,   7.5919],
        [  6.9284,  11.7061,  -0.3732],
        [  1.3767, -11.0341,   9.9415],
        [  9.9058,  -6.7203,   5.6928],
        [ -1.5875,   7.8944,  -9.4286],
        [  4.2096,   5.8815,   1.9230],
        [ -8.7735,   2.5640,  -6.3313],
        [ -7.7842,   4.3466,  -1.7548],
        [  4.6992,   5.3690,   3.0935],
        [ -5.6174,   6.4708,  -1.7339],
        [ -7.4733,   4.7555,   6.2490],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.7659,   7.4127,   4.7871],
        [ 10.0400,   8.6922,  -6.2847],
        [  8.7131,  10.2216,   7.7915],
        [  6.0310,  12.2795,  -6.4528],
        [ 12.4837,   2.7510,   0.4449],
        [  8.5716,  10.3596,   6.8818],
        [ -8.0965,  -5.9712,   4.3466],
        [  3.2159, -10.8166,  -1.0406],
        [  3.1892, -10.8220,  -6.8409],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 12.1940,  -2.1881,   7.1499],
        [  4.2096,   5.8815,   1.9230],
        [  9.9750,  -6.6338,  -1.2440],
        [  1.0093,   7.6567,   2.3526],
        [  4.1297,  -3.7098,  -3.9838],
        [  5.9883,   3.1536,   6.2320],
        [ 11.7779,   5.5127,  -2.0699],
        [ -6.2200,  -8.2737,   5.5079],
        [  4.2275,  -3.6320,  -0.2714],
        [  8.7131,  10.2216,   7.7915],
        [  9.6662,   9.1725,   8.1056],
        [  5.6022,  -1.9833,   8.3081],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -1.9600,   7.8577,   6.9121],
        [ -7.3554,   4.8985,  -0.6665],
        [ -9.3616,  -2.4750,   3.7765],
        [ -5.9940,   6.1985,   5.9566],
        [  1.7122,  -4.7591,  -3.6734],
        [  5.6202,  -1.9508,   0.7471],
        [  4.4049,   5.6892,  -4.4394]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 1.1638e-01, 1.5011e-01,  ..., 7.4517e-01, 6.9101e-01,
         7.9428e-01],
        [1.1638e-01, 0.0000e+00, 1.1270e-01,  ..., 6.5211e-01, 5.7719e-01,
         7.0264e-01],
        [1.5011e-01, 1.1270e-01, 0.0000e+00,  ..., 7.3974e-01, 6.2705e-01,
         7.9119e-01],
        ...,
        [7.4517e-01, 6.5211e-01, 7.3974e-01,  ..., 0.0000e+00, 2.9279e-01,
         5.1930e-02],
        [6.9101e-01, 5.7719e-01, 6.2705e-01,  ..., 2.9279e-01, 0.0000e+00,
         3.2959e-01],
        [7.9428e-01, 7.0264e-01, 7.9119e-01,  ..., 5.1930e-02, 3.2959e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.0050379 , 11.27656488, 11.23201619, 11.4088043 , 11.04953767,
       11.42947632,  6.48486171,  9.75401449,  7.72668467, 10.51437947,
        9.12220271,  7.53884104, 12.38881635, 12.27993098, 14.10045797,
        9.15253344, 11.83509372,  9.17595899, 10.78533838, 14.08549802,
       11.42947632,  9.94285511,  9.562174  ,  8.62329466,  4.74770899,
        5.24461068, 13.25336001,  6.89245352,  4.91038884, 11.97957425,
       11.99633819, 10.76154256,  5.46244988, 11.785566  , 13.96987806,
       11.8159254 , 10.06286074, 13.35927527, 12.22276278, 10.26545792,
       10.42667093, 11.97238624,  7.9337647 ,  9.15253344,  8.29203356,
        5.05989547, 12.49317563, 11.70016217, 13.39372426, 10.32715715,
        8.85939626,  6.77777606,  5.95297459, 13.37121923,  6.33517349,
        7.66377554,  5.9510489 , 11.52733581, 12.52372629,  7.32950385,
        9.68324553,  8.86082124,  7.46248947,  7.96181288, 10.68031351,
       13.60275211, 11.11969815, 11.9702762 ,  8.05242591,  7.23276582,
        9.14044984,  8.91551231,  7.13501264,  8.56889738,  8.8580671 ,
       10.1979669 ,  7.62011734, 13.27994304, 13.43127131, 13.68063098,
       12.78327037, 13.44593519, 10.06023665, 11.2845608 , 11.28215429,
       10.52587243, 12.38881635,  7.23276582, 11.97949273,  7.72291946,
        5.5513072 ,  6.76791483, 13.00413829, 10.35093485,  5.57341623,
       13.43127131, 13.32557133,  5.94293185, 12.4217523 ,  8.09842704,
        8.83725081,  9.68324553,  8.62260327,  5.05774295,  5.94909709,
        7.19513806])}
{'P': tensor([[1.1877e-02, 2.5441e-01, 1.7130e-01,  ..., 1.2051e-02, 4.6906e-04,
         3.0736e-06],
        [6.7479e-05, 2.3031e-02, 1.2220e-02,  ..., 6.9498e-03, 1.0140e-03,
         1.1006e-05],
        [1.3754e-04, 4.0086e-02, 3.7238e-02,  ..., 1.7561e-02, 2.1192e-03,
         1.8964e-05],
        ...,
        [1.5696e-04, 2.9819e-03, 3.2945e-03,  ..., 3.3918e-02, 5.4468e-03,
         2.4210e-05],
        [4.9587e-04, 1.3041e-03, 1.4973e-03,  ..., 3.2562e-03, 1.9720e-01,
         2.3290e-03],
        [8.3237e-06, 5.3027e-06, 5.6900e-06,  ..., 6.2815e-06, 7.2470e-05,
         3.5727e-02]]), 'X': tensor([[  0.1047, -11.0046,   0.5188],
        [  3.2647, -10.8066,  -0.3768],
        [  3.2159, -10.8166,  -1.0406],
        [  4.8051, -10.3703,   0.5268],
        [  3.1271, -10.8343,   2.6442],
        [  4.5814, -10.4485,   2.5496],
        [ -8.3717,  -5.4769,  -9.2228],
        [  6.2419,  -0.2313,   4.2440],
        [  0.2571,  -4.7594,   1.5618],
        [  2.0729,   7.2878,   7.0880],
        [  6.0788,  -0.8710,  -6.8719],
        [  7.0642,  11.6104,  -9.8316],
        [  0.3354,  -4.7708,  -4.9125],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.3444,   0.5821,   6.9101],
        [  9.0004,  -7.7195,   4.6945],
        [  9.0652,   9.8597,  -3.5949],
        [ -6.1914,  -8.2997,   8.5380],
        [  9.9750,  -6.6338,  -1.2440],
        [  2.0547,   7.2957,  -1.9343],
        [  5.8902,  -1.3887,   4.0388],
        [  0.8940, -11.0402,  -0.9783],
        [  2.8994,   6.8675,  -9.5987],
        [  1.3339,   7.5627,  -0.9034],
        [ -7.2515,   5.0195,   4.6434],
        [  2.8474,   6.8979,   6.6457],
        [  8.1043,  -8.5164,  -7.9290],
        [ 10.5955,   7.8824,  -3.7006],
        [  9.6662,   9.1725,   8.1056],
        [ -9.1706,   1.3863,   8.4032],
        [  4.1637,  13.1898,  -1.9189],
        [  6.2038,   2.4020,  -3.1692],
        [  1.9312,  13.8649,   8.3563],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 10.1798,   8.5002,  -5.8893],
        [  9.7328,   9.0902,  -0.8917],
        [  5.3860,  -2.3418,   1.0826],
        [  2.1477,   7.2550,   8.9688],
        [  0.9036,   7.6840,   9.9237],
        [ -1.3940, -10.7772,   6.7835],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.2555,   2.1474,   4.1685],
        [ -7.9925,  -6.1423,   2.8990],
        [ 11.0438,   7.1180,  -5.6991],
        [  2.5375,   7.0676,  -4.3490],
        [ 11.4235,   6.3594,   5.1663],
        [  8.7131,  10.2216,   7.7915],
        [  6.9956,  11.6590,  -7.6876],
        [  6.1222,  -0.7261,   8.8204],
        [ 12.3357,   3.6001,   0.7903],
        [  5.6928,  12.4710,   3.6001],
        [  1.3705,  13.9723,   7.9266],
        [  2.2437,  -4.6489,   6.1430],
        [ -2.4956, -10.4682,   3.5632],
        [  6.1066,  -0.7799,  -4.1746],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -3.0004,   7.6607,  -7.2640],
        [ -5.3736,   6.6310,  -1.5174],
        [ -3.5903,   7.4847,  -8.9112],
        [  6.2883,   0.0325,   5.2692],
        [  5.7845,   3.6652,   0.9175],
        [ 11.6634,   5.8050,   9.1590],
        [  2.6970,   6.9826,   9.7350],
        [  1.5383, -11.0274,  -4.9666],
        [  3.0719, -10.8449,  -4.1673],
        [  3.5939,   6.4029,  -1.8357],
        [  1.2273,  -4.8081,   9.8129],
        [ -3.5520,   7.4976,   9.1522],
        [  5.8201,   3.5841,  -8.0609],
        [ -2.3317,   7.8034,  -3.6274],
        [  6.8407,  11.7666,   3.0045],
        [  3.3839,  -4.1966,   8.3574],
        [ -2.8162,   7.7059,  -9.6885],
        [  9.3836,   9.5075,  -5.9278],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.8940, -11.0402,  -0.9783],
        [  1.8766,   7.3694,   1.1887],
        [  3.9022,  13.2906,   5.5788],
        [ -4.0868,   7.2981,   8.1874],
        [  0.9392,  -4.8143,  -3.4649],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.7544,   2.6096,   3.3146],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.5800,   5.5034,  -3.9432],
        [  5.7218,  -9.9934,   6.5764],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.4697,  -0.6157,  -6.8472],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.7078,   2.7182,   1.8146],
        [  4.0587, -10.6112,   6.2273],
        [  4.4633,   5.6287,  -0.7754],
        [  1.1006,   7.6318,  -2.4877],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.1741,  -0.5329,   3.1913],
        [  9.1820,   9.7334,  -2.7932],
        [ -1.0407,   7.9166,  -6.6021],
        [  5.3602,  -2.3808,   7.2098],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.7218,  -4.7576,  -3.3617],
        [ -2.2188,   7.8218,   8.6064],
        [  3.8111, -10.6787,  -4.6539],
        [  3.2990,  -4.2415,  -7.4857],
        [  0.1497,   7.8334,  -5.4412]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1496, 0.1811,  ..., 0.4333, 0.7539, 0.5657],
        [0.1496, 0.0000, 0.0466,  ..., 0.3209, 0.6044, 0.4187],
        [0.1811, 0.0466, 0.0000,  ..., 0.2744, 0.5763, 0.3849],
        ...,
        [0.4333, 0.3209, 0.2744,  ..., 0.0000, 0.4512, 0.2524],
        [0.7539, 0.6044, 0.5763,  ..., 0.4512, 0.0000, 0.2052],
        [0.5657, 0.4187, 0.3849,  ..., 0.2524, 0.2052, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.00508681, 11.28895489, 11.2845608 , 11.42947632, 11.27656488,
       11.4088043 , 10.00410625,  6.24614905,  4.7663526 ,  7.57685908,
        6.14086582, 13.59056613,  4.78258092,  6.10774544,  9.36256096,
       11.85743425, 13.39372426, 10.35464916, 11.97949273,  7.57946459,
        6.05165248, 11.07637531,  7.4544866 ,  7.67939546,  8.81930525,
        7.46248947, 11.75618513, 13.20597932, 13.32557133,  9.27474386,
       13.83139024,  6.65259775, 13.99877413,  9.51074763, 13.26209545,
       13.31764181,  5.87305005,  7.56617037,  7.73692436, 10.86693708,
        5.19065439,  9.86089579,  9.36093562,  6.61386436, 10.08001894,
       13.13888841,  7.5092834 , 13.0743399 , 13.43127131, 13.59673406,
        6.16512856, 12.85032836, 13.70893492, 14.0393949 ,  5.16205087,
       10.76154256,  6.1561646 , 10.54658597,  8.22728399,  8.53501777,
        8.30125201,  6.28835101,  6.84796202, 13.02818702,  7.48538905,
       11.1341798 , 11.27160035,  7.34250948,  4.96231532,  8.29640613,
        6.83515201,  8.14433542, 13.6105595 ,  5.39098669,  8.20436764,
       13.35832725,  5.27178363, 11.07637531,  7.60463349, 13.85158417,
        8.36445758,  4.90505566,  5.69913817,  9.13507321, 11.66453059,
        7.15990676, 11.51557166,  8.29022197,  9.48969978, 12.04924159,
       10.33712258,  9.12220271, 11.36089355,  7.18354885,  7.71075226,
        7.9337647 ,  6.19707944, 13.38091121,  7.98469166,  5.86519525,
        9.38232121,  5.05963217,  8.1303945 , 11.33837396,  5.37343173,
        7.83486943])}
{'P': tensor([[7.4197e-03, 2.4862e-01, 2.0057e-01,  ..., 2.6886e-06, 1.0073e-07,
         9.5062e-07],
        [9.4621e-05, 8.6857e-03, 2.2947e-01,  ..., 3.6250e-06, 9.2616e-08,
         8.6373e-07],
        [6.5196e-05, 1.3500e-04, 2.3485e-02,  ..., 2.5548e-05, 2.7137e-07,
         1.7093e-06],
        ...,
        [2.5586e-08, 2.4319e-08, 3.5822e-08,  ..., 7.3113e-03, 5.0377e-06,
         2.7354e-07],
        [3.5997e-05, 2.2498e-05, 1.8032e-05,  ..., 2.7463e-06, 7.6843e-02,
         1.7064e-03],
        [7.0049e-03, 4.1360e-03, 2.1209e-03,  ..., 1.8665e-06, 1.0781e-04,
         9.2297e-02]]), 'X': tensor([[  0.1252, -11.0062,   7.2564],
        [  1.1644, -11.0394,   7.8511],
        [  3.1756, -10.8247,   6.8882],
        [  2.6078, -10.9229,   6.2869],
        [  2.1440, -10.9811,   8.5281],
        [  3.9770, -10.6341,   8.0783],
        [  5.9587,  12.3216,   9.1695],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.2675,  12.1379,  -8.9720],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.9241,   7.3503,   7.9004],
        [  2.5318,  13.7231,   6.3692],
        [  9.3560,   9.5391,   9.4943],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.4837,   2.7510,   0.4449],
        [  7.1441,  11.5529,   2.3232],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -8.4573,   3.2448,   5.4370],
        [  0.1532,  -4.7422,  -7.0145],
        [  0.6142,  14.0802,   0.5407],
        [ -9.3705,   0.4208,  -3.5815],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -6.5295,  -7.9785,  -7.4429],
        [  8.6271,  -8.0715,  -7.4813],
        [  9.0129,  -7.7072,   9.1400],
        [  0.5960, -11.0335,   3.1144],
        [  6.9653,  -9.3225,  -3.4727],
        [ -1.8788, -10.6567,   3.7068],
        [ 11.6634,   5.8050,   9.1590],
        [  2.5318,  13.7231,   6.3692],
        [  6.3516,   0.6425,   0.3425],
        [  4.8653,   5.1694,  -4.0823],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.5382,   7.0672,   4.5097],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.6229,   2.9068,   2.4429],
        [  2.9461,  -4.4077,  -0.4831],
        [  8.4691,  10.4571,  -5.0650],
        [ -9.3616,  -2.4749,   0.1793],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -5.6174,   6.4708,  -1.7339],
        [  6.3048,   0.1469,  -9.1691],
        [  2.1477,   7.2550,   8.9688],
        [ -8.9942,  -3.9990,   6.0448],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.6422,   7.4586,  -6.8056],
        [  3.6154,  -4.0640,  -7.9698],
        [  3.3094,  13.4969,  -3.6507],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.9659,  -4.8144,  -0.9594],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.9548,  -1.2273,  -2.6520],
        [  3.7992,  -3.9474,  -8.9019],
        [ 11.0438,   7.1180,  -5.6991],
        [  4.8432,   5.1968,   1.5006],
        [  3.4009,  -4.1874,  -5.9751],
        [ 11.2939,   6.6325,  -5.8311],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.4753,  -0.8012,  -5.2050],
        [  3.4009,  -4.1874,  -5.9751],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 11.7418,   5.6070,  -1.4192],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -6.6998,   5.5954,   0.0823],
        [ -3.0004,   7.6607,  -7.2640],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.1706,   1.3863,   8.4032],
        [  8.2154,  -8.4262,   8.0171],
        [  4.6357,   5.4415,   7.9067],
        [ 10.9394,   7.3069,  -8.9365],
        [  5.3924, -10.1396,  -4.0863],
        [  4.6992,   5.3690,   3.0935],
        [ -2.6069, -10.4298,  -6.1995],
        [  4.4633,   5.6287,  -0.7754],
        [  6.0073,   3.0986,   6.7937],
        [  2.5974,   7.0363,  -7.7810],
        [  9.1409,  -7.5789,   1.5814],
        [  5.4242, -10.1261,   3.9595],
        [ 11.7162,   5.6728,  -8.9372],
        [ -4.4503,   7.1378,   3.3751],
        [  1.9312,  13.8649,   8.3563],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.1625,  -4.6697,   7.0658],
        [ -5.3328,  -8.9992,   7.3440],
        [ 12.2692,  -1.8882,   3.9253],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 12.0772,  -2.6014,  -7.1832],
        [  8.5093,  -8.1764,  -3.2381],
        [  8.9284,  10.0036,  -6.9210],
        [  1.8320,   7.3871,   5.1452],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.0277,   4.9589,   7.0175],
        [  6.3124,   0.2057,  -0.4593],
        [  6.8093,  -9.4178,   8.4244],
        [ -1.7009,   7.8851,  -8.2437],
        [  4.5954,  13.0099,  -4.4439],
        [  6.3413,   0.4883,  -6.8416],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.3122,  -9.6994,  -6.3282],
        [ 11.3118,   6.5957,  -4.5667],
        [ 12.1504,   4.3674,  -7.2998],
        [ -8.6112,   2.9319,  -7.9943],
        [ -9.4351,  -1.8891,   3.5642]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0568, 0.1251,  ..., 1.4641, 1.0803, 0.3819],
        [0.0568, 0.0000, 0.0766,  ..., 1.4459, 1.1065, 0.4275],
        [0.1251, 0.0766, 0.0000,  ..., 1.3809, 1.0929, 0.4533],
        ...,
        [1.4641, 1.4459, 1.3809,  ..., 0.0000, 1.0564, 1.3871],
        [1.0803, 1.1065, 1.0929,  ..., 1.0564, 0.0000, 0.7511],
        [0.3819, 0.4275, 0.4533,  ..., 1.3871, 0.7511, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.00694688, 11.10066126, 11.28093507, 11.22993218, 11.18839169,
       11.35345345, 13.68672855, 10.61417181, 13.66051884, 12.3828957 ,
        5.24137909,  7.59795445, 13.95472331, 13.3614598 ,  8.53501777,
       12.78327037, 13.58333857, 10.33712258,  9.97964013,  9.05843201,
        4.74468406, 14.09357547,  9.37989631, 10.5575601 , 10.30972193,
       11.81423161, 11.85890664, 11.04953767, 11.63724792, 10.82106505,
       13.02818702, 13.95472331,  6.38399789,  7.09887557,  9.93553424,
        7.50917551, 10.54658597,  9.09963894,  5.30158497, 13.45642379,
        9.68323686,  9.79039371,  8.56889738,  6.30647746,  7.56617037,
        9.84314712, 12.04924159,  7.63726134,  5.43943791, 13.89671314,
       12.42554808,  4.91038884,  9.4997987 ,  6.07992623,  5.47866221,
       13.13888841,  7.10378919,  5.39450305, 13.09737053, 12.11735334,
        9.50913849,  5.39450305, 12.12641969, 13.01187484,  9.86089579,
        8.72896381,  8.22728399,  7.48311002,  9.27474386, 11.76830981,
        7.14837514, 13.15525803, 11.48434767,  7.13501264, 10.7506423 ,
        7.18354885,  6.7593917 ,  7.50035355, 11.87412439, 11.48734978,
       13.01728313,  8.41148722, 13.99877413,  8.01009083,  5.14607014,
       10.4606265 , 12.41366754, 12.05300943, 12.35421461, 11.80092578,
       13.40850328,  7.61088125, 10.60226614,  7.06169391,  6.31576264,
       11.62158554,  8.06643487, 13.79762415,  6.36004393, 11.28895489,
       12.889695  , 11.57249984, 13.09426142, 12.91143491,  9.09661212,
        9.62238828])}
{'P': tensor([[6.6685e-03, 2.3165e-01, 1.8269e-01,  ..., 5.0372e-07, 5.7399e-04,
         3.3109e-04],
        [2.0537e-05, 5.8323e-03, 1.0726e-01,  ..., 9.9014e-07, 3.5829e-04,
         6.8461e-05],
        [3.4695e-05, 3.3900e-04, 1.2564e-02,  ..., 2.3387e-06, 5.5847e-04,
         8.7567e-05],
        ...,
        [1.9699e-07, 1.5139e-07, 1.3449e-07,  ..., 2.6554e-02, 4.3818e-06,
         3.0000e-07],
        [1.0514e-03, 2.2202e-03, 2.2528e-03,  ..., 2.8851e-04, 1.2193e-01,
         1.8192e-03],
        [1.2374e-03, 5.5279e-03, 5.0204e-03,  ..., 7.9794e-06, 8.7081e-03,
         4.7303e-02]]), 'X': tensor([[ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0842, 0.1288,  ..., 1.1919, 0.7935, 0.7000],
        [0.0842, 0.0000, 0.0512,  ..., 1.1223, 0.7267, 0.6912],
        [0.1288, 0.0512, 0.0000,  ..., 1.1078, 0.7163, 0.7213],
        ...,
        [1.1919, 1.1223, 1.1078,  ..., 0.0000, 0.4006, 0.8853],
        [0.7935, 0.7267, 0.7163,  ..., 0.4006, 0.0000, 0.5784],
        [0.7000, 0.6912, 0.7213,  ..., 0.8853, 0.5784, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.00726483, 11.22993218, 11.28093507, 11.05317597, 11.32149259,
       11.36089355,  7.01475376,  7.46248947,  5.27178363,  9.79039371,
        9.34127013, 11.46176259, 11.35182589, 11.83570132,  8.22728399,
        5.62005629,  5.04028538, 11.65923396,  5.16077764, 13.103709  ,
        7.85919687, 13.89974693,  5.5205189 ,  9.95552626,  7.32950385,
       11.97957425, 13.01728313, 10.00802627, 12.10942412,  5.41348121,
        9.50654967, 13.69162795, 13.27994304,  5.39723241,  9.55593374,
       11.33837396, 13.94008948,  8.1303945 , 12.04924159, 10.1371552 ,
        8.53014915,  9.32094055,  7.56617037,  6.78325151, 13.59056613,
       13.58333857,  8.98765644,  9.17905491, 11.05413545,  7.69752262,
       11.89105259,  6.31576264,  7.57946459,  8.0550286 ,  7.92200342,
       11.75438006,  7.02253235, 12.44177641,  9.86089579, 13.59056613,
        9.78285424,  4.75303127, 11.42420508, 13.99048094,  5.10446653,
        5.2287058 , 11.89105259,  7.60529892,  5.05981914,  8.29022197,
       10.67132524,  8.25828409, 13.41713661,  8.08320235,  6.05165248,
       13.20988996,  6.7593917 ,  9.23178204,  5.0713364 ,  7.34250948,
       13.01187484,  9.45995955, 13.13888841,  6.36004393, 11.27160035,
        9.85384623, 12.62576026,  8.23525433, 13.04156075,  9.3109884 ,
        9.58072286, 12.17901033,  6.99097013, 11.018084  ,  7.19534146,
        9.66977802, 13.01187484,  5.47702766,  6.61211224, 13.89264822,
       13.43127131,  5.14607014, 13.35013374, 13.6660671 ,  5.14999799,
       11.05672136])}
{'P': tensor([[1.3618e-02, 2.2230e-01, 1.8706e-01,  ..., 1.3400e-03, 2.7055e-05,
         1.4351e-03],
        [5.9301e-05, 1.7246e-02, 3.8804e-03,  ..., 3.3290e-03, 1.4195e-04,
         1.1573e-03],
        [4.9943e-05, 4.2337e-03, 1.2133e-02,  ..., 8.3904e-04, 6.9902e-05,
         2.3200e-03],
        ...,
        [2.3518e-04, 5.4675e-04, 3.6223e-04,  ..., 1.5667e-01, 1.2147e-02,
         1.6324e-03],
        [6.6714e-08, 7.7811e-08, 6.7600e-08,  ..., 3.8864e-06, 5.6435e-03,
         1.5832e-06],
        [2.2798e-04, 3.9311e-04, 5.7577e-04,  ..., 1.5326e-03, 1.3995e-03,
         1.0579e-01]]), 'X': tensor([[ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1679, 0.1765,  ..., 0.7153, 0.9238, 0.7470],
        [0.1679, 0.0000, 0.2696,  ..., 0.5588, 0.7847, 0.7263],
        [0.1765, 0.2696, 0.0000,  ..., 0.7196, 0.8853, 0.5942],
        ...,
        [0.7153, 0.5588, 0.7196,  ..., 0.0000, 0.2719, 0.7073],
        [0.9238, 0.7847, 0.8853,  ..., 0.2719, 0.0000, 0.6661],
        [0.7470, 0.7263, 0.5942,  ..., 0.7073, 0.6661, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.00812996, 11.28895489, 11.27656488, 11.42947632, 11.4088043 ,
       11.2845608 , 13.31764181, 12.49317563,  8.72896381,  8.51088913,
       11.46176259,  5.56809335,  4.90505566,  5.35774573, 12.18045978,
       10.35093485,  9.55720651, 12.40074901, 10.26545792, 11.00508681,
       10.52587243, 10.65929968, 13.05148826,  9.7712321 , 12.91505111,
       10.61047596,  7.28793663, 13.2430377 , 12.889695  , 13.95361098,
        9.04255655,  8.77913428, 12.58006038,  9.9353921 ,  7.23276582,
        6.652742  , 12.4013153 , 11.64436684, 12.53285633, 11.75807252,
       11.18839169,  8.53014915, 13.32613445,  6.81084724,  6.46356806,
       10.84437705,  5.05444288, 13.50484539, 11.89855267, 13.95361098,
       13.59673406,  8.16584537,  5.70591986,  9.75401449,  5.05774295,
        9.75075273,  8.89301516, 10.78533838, 13.14777433,  5.51177501,
       10.09979747, 13.15525803,  8.87587457,  5.88782913,  9.69629   ,
        5.94293185,  9.70108409,  9.54154096,  9.18690018,  7.98469166,
       12.17451685, 11.95743558, 10.31584102,  5.13125513,  6.76661595,
        8.29022197,  7.32233733, 11.05672136,  9.17905491, 13.744387  ,
       12.91505111, 11.1341798 , 13.95472331, 13.86480175, 13.95472331,
       12.79126785, 11.66453059,  8.12043983,  7.14837514,  8.08320235,
       13.55868684,  5.43544626, 10.04916678, 13.77222752, 10.68416401,
       12.27991991, 11.9182956 ,  7.49076899,  8.87007495,  7.23276582,
        8.1303945 ,  9.99145096, 12.19805394,  5.70405525, 13.06498554,
        5.5748624 ])}
{'P': tensor([[9.3873e-03, 4.1928e-01, 1.8984e-01,  ..., 6.3607e-03, 6.9443e-04,
         1.4069e-04],
        [7.0433e-05, 1.6067e-02, 2.2883e-01,  ..., 8.3567e-03, 1.9469e-03,
         4.0930e-04],
        [4.2328e-05, 2.5758e-04, 1.1174e-02,  ..., 2.4561e-03, 1.7253e-03,
         3.5807e-04],
        ...,
        [6.1765e-05, 1.8670e-04, 2.9873e-04,  ..., 4.6883e-02, 2.5955e-03,
         1.3906e-03],
        [5.9415e-05, 7.5938e-05, 8.7708e-05,  ..., 9.1425e-05, 9.3973e-02,
         2.6953e-01],
        [1.9543e-05, 2.0863e-05, 2.1778e-05,  ..., 2.0145e-05, 1.4242e-03,
         6.7766e-02]]), 'X': tensor([[ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1074, 0.1539,  ..., 0.6121, 0.6689, 0.6675],
        [0.1074, 0.0000, 0.0717,  ..., 0.5681, 0.5616, 0.5604],
        [0.1539, 0.0717, 0.0000,  ..., 0.6196, 0.5328, 0.5373],
        ...,
        [0.6121, 0.5681, 0.6196,  ..., 0.0000, 0.6528, 0.6137],
        [0.6689, 0.5616, 0.5328,  ..., 0.6528, 0.0000, 0.0462],
        [0.6675, 0.5604, 0.5373,  ..., 0.6137, 0.0462, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.018084  , 11.1913598 , 11.27160035, 11.33837396, 11.2845608 ,
       11.48434767,  8.94395628,  4.89869205,  6.27715105,  6.58184474,
        5.80312949, 10.67132524,  8.27853076,  9.55593374, 10.78533838,
       13.96905756,  7.52117132,  9.53015376,  5.05774295,  5.95111804,
        8.87587457,  8.8737556 ,  4.73784222,  5.80860165, 11.75618513,
        5.0209005 , 13.27994304, 13.744387  ,  7.60463349,  8.68832398,
       13.25017211,  8.04664886,  4.87083348, 12.13568598,  7.9337647 ,
        6.54615887, 12.17901033,  9.18358562,  7.32754882, 10.66631784,
        8.22728399,  9.88571713, 10.27716434,  8.92549752, 13.66051884,
        9.17905491, 10.75443106, 12.35421461,  8.41171369,  9.75401449,
       12.12641969,  9.3349858 ,  6.55800267,  5.26342203,  5.07306274,
        7.19513806,  9.78286733, 13.3426541 ,  7.81509631, 12.17451685,
        5.54776934,  6.24187044, 10.01804223, 11.00726483,  7.60529892,
        8.29022197, 13.6922425 ,  9.68324553,  7.15990676,  8.56889738,
       10.68416401, 11.05413545, 11.83570132, 11.23201619,  9.71392179,
       10.95593729, 12.05300943,  5.51177501,  8.66273171,  5.38342077,
        9.3349858 ,  4.79003361,  9.09502385,  6.18668463,  6.39520397,
       13.61600536,  6.54127431,  8.51088913, 11.04629471, 11.35345345,
        9.0368458 , 12.41366754, 12.10222204, 11.00694688,  7.57685908,
        8.87587457, 13.89974693, 13.59170809,  7.06372172,  5.5671389 ,
        7.15990676,  8.85939626,  7.67939546, 11.52733581,  6.1561646 ,
        6.65259775])}
{'P': tensor([[7.4033e-03, 2.3049e-01, 2.1283e-01,  ..., 1.1095e-06, 7.8336e-06,
         6.0365e-06],
        [3.3898e-05, 6.8136e-03, 1.0636e-01,  ..., 1.1674e-06, 9.7889e-06,
         1.0968e-05],
        [9.3073e-05, 7.0895e-04, 3.4053e-02,  ..., 7.5099e-06, 7.3363e-05,
         9.3232e-05],
        ...,
        [2.6619e-05, 1.5137e-05, 1.4241e-05,  ..., 8.1828e-02, 3.0460e-04,
         1.4882e-05],
        [7.1367e-06, 5.5850e-06, 6.1322e-06,  ..., 6.7449e-02, 5.7422e-02,
         4.6591e-04],
        [9.2817e-08, 8.6427e-08, 9.6979e-08,  ..., 1.4614e-03, 4.1464e-03,
         1.0793e-02]]), 'X': tensor([[  0.5250, -11.0306,   5.5358],
        [  2.6078, -10.9229,   6.2869],
        [  3.5284, -10.7484,   5.3347],
        [  3.6249, -10.7255,   5.6605],
        [  4.0587, -10.6112,   6.2273],
        [  3.1756, -10.8247,   6.8882],
        [  2.9470,  13.6085,   2.9809],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.3464,   0.5708,  -1.6397],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -3.9508,  -9.8515,   1.7923],
        [ 12.2592,   3.9425,   2.0383],
        [  5.2188,  -2.5839,   2.1833],
        [  5.7845,   3.6652,   0.9175],
        [  4.3554,   5.7393,  -9.0766],
        [ -6.5295,  -7.9785,  -7.4429],
        [  9.9750,  -6.6338,  -1.2440],
        [ 12.5770,   1.8512,  -0.6578],
        [  2.8994,   6.8675,  -9.5987],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.7151,  -1.7700,   3.1542],
        [  1.8719,   7.3713,  -3.7465],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.6142,  14.0802,   0.5407],
        [ 12.3357,   3.6001,   0.7903],
        [ -5.6423,   6.4537,  -1.2091],
        [  9.4925,   9.3811,   8.9842],
        [ -0.3432,   7.8900,   7.1805],
        [ 11.9641,  -2.9568,   7.7211],
        [ -5.7347,  -8.6908,   9.2878],
        [  8.3791,  10.5409,   5.5590],
        [  6.3301,   0.3638,  -7.5685],
        [ 11.2767,  -4.6006,   0.7868],
        [  9.9058,  -6.7203,   5.6928],
        [  6.2038,   2.4020,  -3.1692],
        [  6.2905,   1.9366,   2.1169],
        [ 10.0400,   8.6922,  -6.2847],
        [  5.9655,   3.2172,  -7.5953],
        [  3.3744,   6.5623,  -6.3636],
        [  2.0243,   7.3086,   7.5432],
        [  5.2911,  -2.4824,  -3.6877],
        [  0.6933,   7.7336,   2.1179],
        [ 11.5568,   6.0602,   9.3244],
        [  6.3542,   0.6934,  -3.3840],
        [  0.6142,  14.0802,   0.5407],
        [  2.2182,  13.8007,   5.1889],
        [ -9.3444,   0.5821,   6.9101],
        [  1.8447,  -4.7372,   8.1235],
        [  9.6816,   9.1536,  -5.8543],
        [ -8.0611,  -6.0302,   4.6962],
        [  0.8940, -11.0402,  -0.9783],
        [  9.5203,   9.3483,   5.8781],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4766,  -1.1116,  -2.1706],
        [  6.2895,   0.0405,   5.7652],
        [  3.8111, -10.6787,  -4.6539],
        [  4.1348,   5.9514,  -0.8856],
        [ -7.8126,   4.3066,   5.1791],
        [  6.1066,  -0.7799,  -4.1746],
        [  0.9945,  -4.8144,  -0.1897],
        [  6.9956,  11.6590,  -7.6876],
        [  6.7792,  11.8084,   1.6251],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.2860,   7.1916,  -6.1011],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.3266,   0.3292,   9.3502],
        [  1.7122,  -4.7591,  -3.6734],
        [ 11.7779,   5.5127,  -2.0699],
        [ -9.4671,  -0.5553,   0.1117],
        [  0.9809,   7.6642,  -1.4217],
        [  2.5128, -10.9365,   9.9645],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -0.4341, -10.9473,   7.4509],
        [  6.8704,  -9.3809,   7.5769],
        [  1.6308,  13.9254,  -2.0984],
        [  0.9036,   7.6840,   9.9237],
        [  3.6703,   6.3443,  -5.4401],
        [ -3.6381,   7.4683,   7.3544],
        [ -9.4351,  -1.8891,   3.5642],
        [ -5.6651,   6.4380,   1.3660],
        [  9.3624,   9.5318,   0.2779],
        [  0.1206,  -4.7363,   8.7103],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.8631,  -4.3728,  -1.1746],
        [  1.9312,  13.8649,   8.3563],
        [  4.4633,   5.6287,  -0.7754],
        [  6.3305,   1.6119,   1.1538],
        [  2.3261,  13.7749,   5.9121],
        [ -4.7847,  -9.3729,  -3.7741],
        [ 11.7418,   5.6070,  -1.4192],
        [  2.9410,   6.8428,   5.4458],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -9.3419,   0.5972,  -1.2841],
        [  1.9241,   7.3503,   7.9004],
        [ -8.8400,   2.3991,   1.3457],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.8201,   3.5841,  -8.0609],
        [  0.5850,  -4.7982,   9.6639],
        [ -2.0445,   7.8469,  -0.5857],
        [  4.0591,   6.0202,  -1.6339],
        [ 10.5678,   7.9261,  -0.6100]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0772, 0.1137,  ..., 0.6407, 0.8411, 1.0557],
        [0.0772, 0.0000, 0.0962,  ..., 0.6768, 0.8628, 1.0562],
        [0.1137, 0.0962, 0.0000,  ..., 0.5887, 0.7682, 0.9606],
        ...,
        [0.6407, 0.6768, 0.5887,  ..., 0.0000, 0.2457, 0.5626],
        [0.8411, 0.8628, 0.7682,  ..., 0.2457, 0.0000, 0.3338],
        [1.0557, 1.0562, 0.9606,  ..., 0.5626, 0.3338, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.04313641, 11.22993218, 11.31276109, 11.32149259, 11.36089355,
       11.28093507, 13.92390689,  9.96152788, 10.76154256,  9.36378438,
       10.5575601 , 10.61417181, 12.87752052,  5.82345867,  6.84796202,
        7.20482224, 10.30972193, 11.97949273, 12.7125079 ,  7.4544866 ,
       10.74409834,  5.98285319,  7.60529892,  7.8575648 , 14.09357547,
       12.85032836,  8.5724042 , 13.34587027,  7.89745107, 12.32408516,
       10.412299  , 13.46553623,  6.34059021, 12.17901033, 11.9702762 ,
        6.65259775,  6.58184474, 13.27994304,  6.77777606,  7.37902766,
        7.58378655,  5.84450127,  7.76457095, 13.0493368 ,  6.39187285,
       14.09357547, 13.97779839,  9.36256096,  5.08372149, 13.32374056,
       10.06703358, 11.07637531, 13.3426541 ,  5.34250225,  5.13616224,
        9.54154096,  6.28961988, 11.33837396,  7.24677664,  8.92098752,
        6.1561646 ,  4.9160962 , 13.59673406, 13.61600536, 12.59969936,
        9.20340031,  7.54620849, 12.12641969,  6.33517349,  5.05774295,
       13.00413829,  9.48337063,  7.72668467, 11.22142195,  9.79039371,
       10.95593729, 11.62770332, 14.02058621,  7.73692436,  7.32950385,
        8.30729352,  9.62238828,  8.57560275, 13.3607379 ,  4.73784222,
       13.02818702, 12.1494023 ,  9.88310614, 13.99877413,  7.18354885,
        6.53250937, 13.96987806, 10.52357479, 13.01187484,  7.44805901,
        9.71392179,  9.88310614,  9.36093562,  7.59795445,  9.15976746,
        6.02789634,  6.83515201,  4.83371341,  8.10886021,  7.26074925,
       13.20988996])}
{'P': tensor([[5.2808e-03, 3.6810e-01, 1.7692e-01,  ..., 3.8440e-04, 6.0612e-07,
         6.0567e-04],
        [8.0358e-05, 7.5347e-03, 2.9747e-01,  ..., 4.8755e-04, 9.3994e-07,
         9.4274e-04],
        [3.1487e-05, 3.5472e-05, 9.9750e-03,  ..., 2.7129e-04, 2.6511e-06,
         2.0779e-03],
        ...,
        [7.5393e-03, 8.2839e-03, 1.3825e-02,  ..., 1.0850e-01, 5.0951e-05,
         4.1153e-02],
        [4.0545e-08, 3.8664e-08, 3.2947e-08,  ..., 6.7376e-07, 7.6125e-03,
         9.4271e-07],
        [6.9297e-05, 7.2817e-05, 1.0739e-04,  ..., 8.7778e-04, 4.4553e-04,
         7.9512e-02]]), 'X': tensor([[ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0336, 0.1295,  ..., 0.5287, 1.3941, 0.6612],
        [0.0336, 0.0000, 0.0967,  ..., 0.4961, 1.3744, 0.6279],
        [0.1295, 0.0967, 0.0000,  ..., 0.3994, 1.3021, 0.5319],
        ...,
        [0.5287, 0.4961, 0.3994,  ..., 0.0000, 1.0486, 0.1439],
        [1.3941, 1.3744, 1.3021,  ..., 1.0486, 0.0000, 1.0466],
        [0.6612, 0.6279, 0.5319,  ..., 0.1439, 1.0466, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.04629471, 11.07637531, 11.2845608 , 11.28895489, 11.42947632,
       11.56146873,  5.56809335,  7.67939546,  8.4486771 , 12.4013153 ,
        5.19986797,  9.60412735,  7.50917551, 13.36800073,  5.92952538,
       12.58006038,  9.18358562,  5.19065439,  7.20826219, 13.1471313 ,
        9.14973319, 14.05633441, 11.63502679,  5.5366698 ,  6.81084724,
        5.10446653,  9.51074763, 12.45418236, 11.80092578,  6.05113217,
       11.89855267,  8.97052115, 12.44177641,  8.89301516, 11.63724792,
       11.75807252, 12.05300943, 10.06850677,  7.92644746,  5.37343173,
        7.58668144,  6.7166297 , 12.04924159, 10.07238463, 13.68672855,
       12.62711252,  8.94395628, 11.87665501, 13.59480906, 13.68063098,
       11.27656488, 12.35421461, 10.82106505, 11.83509372,  6.38968357,
       13.91198191, 10.66631784,  6.76661595, 13.3426541 , 10.04916678,
        5.43544626, 10.8992524 ,  7.20482224, 10.46739073, 10.61472636,
        8.28968067,  8.1137452 , 12.12641969,  6.53290663,  6.46560181,
        5.91222346, 11.22993218, 10.78533838, 11.42947632, 11.66453059,
        6.16512856,  8.04718433,  5.69913817, 10.38897563,  6.80559669,
       12.07225469, 12.42554808,  7.6374603 ,  9.47369628, 13.55868684,
        6.77777606,  5.5513072 ,  7.19513806, 13.35927527, 12.53345445,
        8.38252937, 10.69963954,  5.87132718,  6.29034958, 12.78327037,
       13.50484539,  7.92607342,  9.41244865,  6.5618968 , 12.24467041,
        5.5671389 ,  7.02496223,  7.46053281,  4.76969364, 13.52430176,
        6.02789634])}
{'P': tensor([[7.8737e-03, 2.5992e-01, 2.5635e-01,  ..., 1.5035e-06, 3.2816e-04,
         1.6029e-06],
        [3.7182e-05, 7.5060e-03, 2.8740e-02,  ..., 6.1649e-07, 5.7837e-04,
         2.7102e-06],
        [2.3360e-05, 6.0133e-04, 6.6777e-03,  ..., 5.9946e-07, 4.4308e-04,
         3.5589e-06],
        ...,
        [7.9976e-04, 4.7685e-04, 7.4559e-04,  ..., 5.0814e-02, 8.9121e-06,
         1.7803e-06],
        [1.2877e-04, 1.7364e-04, 1.7011e-04,  ..., 7.4205e-06, 2.7149e-01,
         1.3296e-03],
        [7.3169e-07, 5.9539e-07, 6.7009e-07,  ..., 4.7571e-06, 1.8253e-05,
         1.1327e-01]]), 'X': tensor([[  0.5960, -11.0335,   3.1144],
        [  2.6310, -10.9195,   3.8054],
        [  3.1271, -10.8343,   2.6442],
        [  4.5814, -10.4485,   2.5496],
        [  5.4242, -10.1261,   3.9595],
        [  5.8450,  -9.9356,   3.0767],
        [ -3.5520,   7.4976,   9.1522],
        [  2.5974,   7.0363,  -7.7810],
        [  5.9587,  12.3216,   9.1695],
        [ -3.0008, -10.2827,  -8.9480],
        [ -9.1623,  -3.4288,  -4.9961],
        [  4.5325,  -3.3643,   6.7699],
        [  5.9655,   3.2172,  -7.5953],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.9942,  -3.9990,   6.0448],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -8.1531,  -5.8748,   2.6934],
        [  5.9193,  -1.3176,   5.1739],
        [  7.2340,  -9.1505,  -5.8321],
        [  1.5383, -11.0274,  -4.9666],
        [  0.4119,  14.1019,  -4.1870],
        [ -5.1398,  -9.1367,   6.2074],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -1.6275, -10.7221,   3.2376],
        [  3.2226,  -4.2802,  -4.0428],
        [  9.2979,   9.6048,   7.3138],
        [  7.1821,  -9.1845,   9.8548],
        [  5.8902,  -1.3887,   4.0388],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.0093,   7.6567,   2.3526],
        [ -8.9606,  -4.0994,   1.7886],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.5506,  -5.1154,   6.3423],
        [ -4.8665,  -9.3204,  -5.2564],
        [  5.3804,  -2.3504,   7.2744],
        [  3.3839,  -4.1966,   8.3574],
        [  5.9912,   3.1453,  -3.9989],
        [ -9.4159,  -2.0739,   9.6841],
        [  2.5829,  -4.5466,  -8.4015],
        [ -7.7842,   4.3466,  -1.7548],
        [ -2.8162,   7.7059,  -9.6885],
        [ -8.1204,  -5.9308,   4.0761],
        [ 11.6634,   5.8050,   9.1590],
        [  5.4242, -10.1261,   3.9595],
        [ 11.5198,  -4.0923,   5.5088],
        [ 12.6059,   0.9597,  -6.3264],
        [ -3.4986,   7.5152,   9.0712],
        [  6.7792,  11.8084,   1.6251],
        [  0.3493,  -4.7727,  -6.5682],
        [ 12.3453,  -1.5462,   7.0212],
        [ -1.0012, -10.8576,   2.2867],
        [  3.4604,  -4.1545,   1.8091],
        [ -0.0254, -10.9932,   3.2200],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.0574,   1.7801,   8.8986],
        [ -1.9714,   7.8563,  -1.8084],
        [  2.2373,  -4.6506,  -1.7334],
        [ -8.3653,   3.4184,   9.3454],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.0788,  -0.8710,  -6.8719],
        [  4.3523, -10.5232,   8.2138],
        [ 10.0400,   8.6922,  -6.2847],
        [ -2.6735, -10.4062,  -1.5911],
        [  8.7715,  -7.9390,   5.2900],
        [  1.3767, -11.0341,   9.9415],
        [ 11.5966,   5.9666,   6.0083],
        [ -9.4117,   0.1160,   8.2574],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.2692,  -1.8882,   3.9253],
        [ -9.1658,  -3.4155,  -2.8880],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.1178,   1.5784,   2.1410],
        [  2.5382,   7.0672,   4.5097],
        [  7.0358,  -9.2784,  -9.8479],
        [ 12.0042,  -2.8350,   2.6864],
        [  7.5784,  -8.9141,   5.8410],
        [  3.9591, -10.6391,  -6.2557],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.5154,   2.5072,  -9.3274],
        [ -1.8788, -10.6567,   3.7068],
        [ -2.9497, -10.3028,  -7.3479],
        [  4.2041,  -3.6509,  -4.9488],
        [  9.2979,   9.6048,   7.3138],
        [ -5.6423,   6.4537,  -1.2091],
        [ -1.9714,   7.8563,  -1.8084],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.1579,  -0.5961,  -6.5595],
        [ -0.0254, -10.9932,   3.2200],
        [  5.7963,  -1.6018,   1.8856],
        [ -5.0943,  -9.1682,   0.5073],
        [  2.1440, -10.9811,   8.5281],
        [ -3.6381,   7.4683,   7.3544],
        [ -9.3810,  -2.3455,  -9.5194],
        [  4.0591,   6.0202,  -1.6339],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.1532,  -4.7422,  -7.0145],
        [  4.5767,   5.5071,   3.9069],
        [  5.6314,  -1.9302,  -9.6543],
        [ -9.3705,   0.4208,  -3.5815],
        [  1.7500,   7.4187,   4.4237],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.2242,  -0.3171,   7.4978],
        [  8.4691,  10.4571,  -5.0650]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1135, 0.1299,  ..., 0.9294, 0.7301, 1.1942],
        [0.1135, 0.0000, 0.1127,  ..., 1.0112, 0.6177, 1.1429],
        [0.1299, 0.1127, 0.0000,  ..., 0.9137, 0.6494, 1.0651],
        ...,
        [0.9294, 1.0112, 0.9137,  ..., 0.0000, 1.4848, 1.2814],
        [0.7301, 0.6177, 0.6494,  ..., 1.4848, 0.0000, 0.9600],
        [1.1942, 1.1429, 1.0651,  ..., 1.2814, 0.9600, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.04953767, 11.23201619, 11.27656488, 11.4088043 , 11.48734978,
       11.52733581,  8.29640613,  7.50035355, 13.68672855, 10.71162681,
        9.78286733,  5.64464807,  6.77777606, 11.97238624,  9.84314712,
        9.60359158, 10.04916678,  6.06416957, 11.66453059, 11.1341798 ,
       14.10796466, 10.48315436,  9.94285511, 10.00410625, 10.84493732,
        5.35774573, 13.36800073, 11.65923396,  6.05165248, 13.20988996,
        7.72291946,  9.85384623,  8.09984231,  9.96391045, 10.51437947,
        5.87132718,  5.39098669,  6.76661595,  9.64157461,  5.22903117,
        8.91551231,  8.20436764, 10.05558893, 13.02818702, 11.48734978,
       12.22503754, 12.64235359,  8.28968067, 13.61600536,  4.78544875,
       12.44177641, 10.9036226 ,  5.40689332, 10.99326577, 12.35625694,
        9.23071698,  8.09984231,  5.16077764,  9.0368458 , 13.103709  ,
        9.11473391,  6.14086582, 11.38774149, 13.27994304, 10.74409834,
       11.83075005, 11.11969815, 13.04156075,  9.41244865,  9.14973319,
        9.11473391, 12.41366754,  9.78147542,  7.822854  ,  9.25336371,
        7.50917551, 11.64436684, 12.3344536 , 11.70016217, 11.35182589,
       11.99633819, 12.76408562, 10.82106505, 10.7167285 ,  5.56809335,
       13.36800073,  8.5724042 ,  8.09984231,  9.49511524,  6.18668463,
       10.99326577,  6.01356004, 10.48840476, 11.18839169,  8.30729352,
        9.66977802,  7.26074925, 10.78533838,  4.74468406,  7.16058587,
        5.95297459,  9.37989631,  7.62232307, 10.27716434,  6.2322769 ,
       13.45642379])}
{'P': tensor([[5.8777e-03, 2.6389e-01, 1.6650e-01,  ..., 8.2801e-04, 2.6999e-04,
         1.7862e-06],
        [2.7738e-05, 6.5883e-03, 1.2116e-01,  ..., 2.4156e-03, 7.6960e-04,
         1.1572e-06],
        [6.9603e-05, 4.9670e-04, 1.8409e-02,  ..., 8.7550e-03, 3.0148e-03,
         2.7945e-06],
        ...,
        [2.7377e-06, 4.1056e-06, 4.6788e-06,  ..., 4.9872e-02, 2.8293e-01,
         1.3380e-06],
        [6.4922e-06, 9.1064e-06, 1.0326e-05,  ..., 1.7825e-03, 1.9895e-01,
         5.0785e-06],
        [7.1152e-02, 2.9573e-02, 2.3047e-02,  ..., 1.4044e-04, 6.3083e-05,
         9.8178e-02]]), 'X': tensor([[  0.6363, -11.0348,   6.1414],
        [  2.6078, -10.9229,   6.2869],
        [  3.1756, -10.8247,   6.8882],
        [  3.6249, -10.7255,   5.6605],
        [  3.5284, -10.7484,   5.3347],
        [  4.0587, -10.6112,   6.2273],
        [  6.3604,   0.9042,  -6.5901],
        [  1.3403,  -4.8011,  -1.1698],
        [  9.6614,   9.1784,  -1.2087],
        [ -9.4196,   0.0478,   9.4140],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.2589,  -1.9312,   9.8657],
        [  6.2576,   2.1359,  -3.3398],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.6310, -10.9195,   3.8054],
        [  2.0547,   7.2957,  -1.9343],
        [ -9.1706,   1.3863,   8.4032],
        [  9.3624,   9.5318,   0.2779],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.3604,   0.9042,  -6.5901],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.4150,  13.9647,  -4.0915],
        [  9.9217,  -6.7006,  -5.6728],
        [ -7.6525,  -6.6534,   5.1721],
        [ -9.2345,  -3.1345,   3.9715],
        [ -7.4602,   4.7718,  -4.6277],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.0961,   1.6529,   3.1553],
        [  3.6703,   6.3443,  -5.4401],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -8.9788,   2.0208,  -3.2867],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.4429,  -2.2529,   9.7306],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3516,   0.6425,   0.3425],
        [ -3.6381,   7.4683,   7.3544],
        [  2.1824,  -4.6647,   1.1985],
        [  6.9653,  -9.3225,  -3.4727],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -4.7642,  -9.3860,  -5.1133],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -7.3554,   4.8985,  -0.6665],
        [  1.4306,  13.9620,  -0.2075],
        [  6.3604,   0.9042,  -6.5901],
        [  2.5813,  -4.5471,   5.3597],
        [  4.5340,  -3.3629,   8.9652],
        [  4.0362,  13.2397,  -7.0030],
        [  6.0031,  12.2958,  -6.5992],
        [ 12.6038,   0.7665,  -3.4439],
        [  7.5784,  -8.9141,   5.8410],
        [  6.1579,  -0.5961,  -6.5595],
        [ -8.0766,  -6.0046,  -3.1873],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.0772,  -2.6014,  -7.1832],
        [  3.7121,   6.3117,   0.6738],
        [  9.2692,   9.6370,   7.0746],
        [ -7.6650,   4.5092,   0.2520],
        [  0.1677,  -4.7447,  -6.1529],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.4671,  -0.5553,   0.1117],
        [  0.5960, -11.0335,   3.1144],
        [  9.3560,   9.5391,   9.4943],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.7735,   2.5640,  -6.3313],
        [ 10.5955,   7.8824,  -3.7006],
        [  3.3094,  13.4969,  -3.6507],
        [  1.7301,   7.4262,   5.0522],
        [  6.1984,  -0.4327,   1.5438],
        [ -6.3252,  -8.1760,  -1.0891],
        [  4.6357,   5.4415,   7.9067],
        [ 10.0400,   8.6922,  -6.2847],
        [  0.3715,  -4.7756,   0.2312],
        [ 11.7726,  -3.4876,  -6.7402],
        [  9.9217,  -6.7006,  -5.6728],
        [ -7.8126,   4.3066,   5.1791],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.4242, -10.1261,   3.9595],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.4013,  -1.2597,  -7.6474],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.4732,  -2.2041,  -4.3830],
        [ 10.9394,   7.3069,  -8.9365],
        [  9.7328,   9.0902,  -0.8917],
        [  5.7845,   3.6652,   0.9175],
        [  1.7228,  -4.7575,   8.3396],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.4203,  -2.0340,   2.5856],
        [  4.5954,  13.0099,  -4.4439],
        [  3.2701,   6.6338,   9.9209],
        [  5.3924, -10.1396,  -4.0863],
        [  0.3038,   7.8092,   0.4164],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.6229,   2.9068,   2.4429],
        [  5.1861,   4.7361,  -3.7312],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -3.1204, -10.2345,   9.2288],
        [ 10.5955,   7.8824,  -3.7006],
        [ -7.8280,   4.2848,  -7.1955],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.6672,  -1.8631,   8.1020],
        [  6.1741,  -0.5329,   3.1913],
        [  3.8182,  -3.9347,  -1.7961],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.4877,  -0.7222,   8.8347],
        [ -8.0330,  -6.0765,   6.7945]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0588, 0.1085,  ..., 0.6590, 0.8063, 0.2586],
        [0.0588, 0.0000, 0.0512,  ..., 0.6010, 0.7476, 0.3172],
        [0.1085, 0.0512, 0.0000,  ..., 0.5567, 0.7001, 0.3650],
        ...,
        [0.6590, 0.6010, 0.5567,  ..., 0.0000, 0.1714, 0.9170],
        [0.8063, 0.7476, 0.7001,  ..., 0.1714, 0.0000, 1.0647],
        [0.2586, 0.3172, 0.3650,  ..., 0.9170, 1.0647, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.05317597, 11.22993218, 11.28093507, 11.32149259, 11.31276109,
       11.36089355,  6.42439344,  4.98461983, 13.32613445,  9.41970031,
       13.41713661, 12.41012046,  6.61211224,  8.89458273, 11.23201619,
        7.57946459,  9.27474386, 13.3607379 , 10.06703358,  6.42439344,
       10.06850677, 14.03618649, 11.97238624, 10.14044489,  9.75200848,
        8.85572292, 12.64235359,  9.24501829,  7.32950385, 12.04924159,
        9.20340031, 13.04217487,  5.89074104, 10.8207297 ,  6.38399789,
        8.30729352,  5.14999799, 11.63724792, 10.54658597, 10.52587243,
        5.60064319, 12.53345445,  8.83725081, 14.03505624,  6.42439344,
        5.2287058 ,  5.64500512, 13.84126197, 13.6829843 , 12.62711252,
       11.70016217,  6.18668463, 10.06407839,  4.91336994, 12.35421461,
        7.32233733, 13.37121923,  8.89301516,  4.74770899,  6.24614905,
        9.48337063, 11.04953767, 13.3614598 ,  5.14999799,  9.14044984,
       13.20597932, 13.89671314,  7.62509582,  6.2134961 , 10.33712258,
        7.14837514, 13.27994304,  4.79003361, 12.27835996, 11.97238624,
        8.92098752, 12.77402351, 11.48734978, 12.30858677, 12.46513646,
       12.38881635,  5.90035324, 13.15525803, 13.31764181,  6.84796202,
        5.05981914,  9.0368458 ,  9.63742669, 13.79762415,  7.39599986,
       11.48434767,  7.81509631,  9.9353921 ,  9.09963894,  7.02326515,
       10.35485331, 10.69963954, 13.20597932,  8.92395381,  9.85384623,
        5.96556248,  6.19707944,  5.48276038, 12.33167382, 12.5085995 ,
       10.07238463])}
{'P': tensor([[7.9010e-03, 2.7618e-01, 2.2758e-01,  ..., 2.2787e-04, 1.0771e-03,
         4.0643e-06],
        [5.9497e-05, 1.8269e-02, 6.4968e-05,  ..., 1.3663e-04, 8.1546e-05,
         3.8689e-06],
        [2.3784e-04, 3.5624e-01, 9.0234e-03,  ..., 2.5307e-04, 7.1049e-04,
         4.2487e-06],
        ...,
        [2.2668e-04, 8.2487e-04, 2.8181e-04,  ..., 1.3353e-02, 3.6518e-04,
         5.7547e-06],
        [1.7927e-02, 1.6234e-01, 3.4732e-02,  ..., 1.7556e-03, 1.6283e-02,
         1.0184e-05],
        [4.0784e-02, 1.8470e-02, 4.1282e-02,  ..., 4.5978e-02, 1.0544e-01,
         6.6878e-02]]), 'X': tensor([[  0.6470, -11.0352,  -1.1931],
        [  3.2159, -10.8166,  -1.0406],
        [  0.8940, -11.0402,  -0.9783],
        [  3.2647, -10.8066,  -0.3768],
        [  4.8051, -10.3703,   0.5268],
        [  5.7485,  -9.9811,  -3.2355],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.2366,  -4.2732,   5.1532],
        [  2.6310, -10.9195,   3.8054],
        [  2.4356,  -4.5942,  -5.0808],
        [ -6.3489,   5.9117,   6.5161],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -1.6086,   7.8928,   4.3385],
        [ -6.6998,   5.5954,   0.0823],
        [  1.4306,  13.9620,  -0.2075],
        [  0.6933,   7.7336,   2.1179],
        [  5.9883,   3.1536,   6.2320],
        [ 12.0582,  -2.6640,   0.7132],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.9770, -10.6341,   8.0783],
        [  2.3261,  13.7749,   5.9121],
        [  5.6414, -10.0303,   7.8525],
        [  2.5636, -10.9293,   9.6519],
        [  6.3549,   0.7096,   6.9198],
        [  7.7750,  11.0665,   2.8364],
        [ -6.6998,   5.5954,   0.0823],
        [  2.0937,  13.8293,  -2.3533],
        [ -2.0445,   7.8469,  -0.5857],
        [  2.1477,   7.2550,   8.9688],
        [  7.2340,  -9.1505,  -5.8321],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.7565,  11.8237,  -4.9882],
        [  6.2419,  -0.2313,   4.2440],
        [ -8.9636,   2.0648,   8.1946],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.6038,   0.7665,  -3.4439],
        [  6.3549,   0.7096,   6.9198],
        [ -9.4773,  -1.0024,   5.2599],
        [  3.7237,  -3.9966,  -5.0376],
        [  5.7485,  -9.9811,  -3.2355],
        [  4.9957,  -2.8705,   6.5978],
        [  5.7218,  -9.9934,   6.5764],
        [ 11.5456,   6.0861,   8.1101],
        [  8.8091,  -7.9037,   0.3631],
        [  8.8091,  -7.9037,   0.3631],
        [ -6.8618,  -7.6322,   3.7027],
        [  5.8325,  -1.5222,  -2.1341],
        [ -5.1398,  -9.1367,   6.2074],
        [  2.7121,   6.9743,  -8.4202],
        [  8.1528,  10.7452,   9.1493],
        [  0.1677,  -4.7447,  -6.1529],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.0419,  -4.6978,  -5.0983],
        [  7.3724,  -9.0577,  -8.9843],
        [ -8.3653,   3.4184,   9.3454],
        [ -3.3999, -10.1152,  -1.2088],
        [  0.3231,  -4.7691,  -1.6120],
        [  2.1379,  -4.6757,   6.4981],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.5294,  -2.1106,   9.0626],
        [  5.4732,  -2.2041,  -4.3830],
        [ -3.4100,   7.5436,  -7.6980],
        [ -7.6525,  -6.6534,   5.1721],
        [  9.5261,  -7.1675,   1.4206],
        [ -4.8665,  -9.3204,  -5.2564],
        [  3.1480,  -4.3165,  -7.8002],
        [  1.6232,  -4.7717,  -2.5655],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.2182,  13.8007,   5.1889],
        [  3.3094,  13.4969,  -3.6507],
        [  2.5813,  -4.5471,   5.3597],
        [ -3.0008, -10.2827,  -8.9480],
        [  7.5930,  -8.9037,   7.2323],
        [  0.2732,  -4.7619,  -1.1577],
        [ 11.6019,   5.9540,  -5.1715],
        [ -2.8162,   7.7059,  -9.6885],
        [ -9.3419,   0.5972,  -1.2841],
        [  9.4554,   9.4245,   9.8800],
        [ 11.6544,   5.8273,  -3.1375],
        [ -6.3306,   5.9272,   5.6935],
        [ 11.4743,   6.2478,  -2.1100],
        [  6.7792,  11.8084,   1.6251],
        [ -6.6895,   5.6051,   2.7899],
        [ -8.8154,   2.4611,   2.5198],
        [  6.3087,   1.8053,  -1.9481],
        [ 10.9916,   7.2133,  -0.9179],
        [ 11.1420,   6.9331,  -2.4779],
        [  9.1620,  -7.5573,  -9.9882],
        [  7.4132,  11.3526,   0.3322],
        [ -3.5173,   7.5091,  -8.9585],
        [  3.2701,   6.6338,   9.9209],
        [  3.6469,   6.3624,  -1.8414],
        [ 10.3251,   8.2928,  -3.0006],
        [  5.9548,  -1.2273,  -2.6520],
        [ -8.0766,  -6.0046,  -3.1873],
        [  0.1497,   7.8334,  -5.4412],
        [  2.7590,  13.6621,   9.4496],
        [  1.6308,  13.9254,  -2.0984],
        [ -7.4807,   4.7463,   2.0616],
        [ -8.5506,  -5.1154,   6.3423],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 12.4691,   2.8525,  -5.5110],
        [  9.1620,  -7.5573,  -9.9882],
        [  1.1644, -11.0394,   7.8511],
        [  0.1382, -11.0073,   1.0026],
        [ -6.8618,  -7.6322,   3.7027]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1076, 0.0206,  ..., 0.7371, 0.1906, 0.5044],
        [0.1076, 0.0000, 0.0967,  ..., 0.7227, 0.2069, 0.5572],
        [0.0206, 0.0967, 0.0000,  ..., 0.7184, 0.1730, 0.4947],
        ...,
        [0.7371, 0.7227, 0.7184,  ..., 0.0000, 0.5474, 0.4433],
        [0.1906, 0.2069, 0.1730,  ..., 0.5474, 0.0000, 0.3556],
        [0.5044, 0.5572, 0.4947,  ..., 0.4433, 0.3556, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.05413545, 11.2845608 , 11.07637531, 11.28895489, 11.42947632,
       11.51811603,  5.34250225,  5.36061974, 11.23201619,  5.19986797,
        8.67502348, 12.12641969,  8.0550286 ,  8.72896381, 14.03505624,
        7.76457095,  6.76791483, 12.3489324 , 12.19805394, 11.35345345,
       13.96987806, 11.50791859, 11.22596994,  6.39438044, 13.5247215 ,
        8.72896381, 13.98691426,  8.10886021,  7.56617037, 11.66453059,
       12.45418236, 13.61801526,  6.24614905,  9.19837792, 13.96905756,
       12.62711252,  6.39438044,  9.53015376,  5.46244988, 11.51811603,
        5.76165934, 11.51557166, 13.05148826, 11.83509372, 11.83509372,
       10.26327895,  6.02789634, 10.48315436,  7.48311002, 13.48809595,
        4.74770899, 11.86239961,  5.1224079 , 11.67875855,  9.0368458 ,
       10.67132524,  4.78002665,  5.14124612, 10.78533838,  5.91854397,
        5.90035324,  8.27853076, 10.14044489, 11.92133129, 10.51437947,
        5.34250225,  5.04028538,  5.18277858, 13.97779839, 13.89671314,
        5.2287058 , 10.71162681, 11.70168755,  4.76969364, 13.04051744,
        8.20436764,  9.36093562, 13.35013374, 13.03002338,  8.67227386,
       13.06498554, 13.61600536,  8.7273569 ,  9.15253344,  6.5618968 ,
       13.1471313 , 13.12299145, 11.87665501, 13.55868684,  8.29203356,
        7.39599986,  7.33350642, 13.2430377 ,  6.07992623, 10.06407839,
        7.83486943, 13.93790235, 14.02058621,  8.85939626,  9.96391045,
       12.3828957 , 12.79126785, 11.87665501, 11.10066126, 11.00812996,
       10.26327895])}
{'P': tensor([[8.1031e-03, 2.8120e-01, 2.0238e-01,  ..., 8.2275e-04, 3.4133e-04,
         3.1272e-05],
        [8.4379e-05, 2.5307e-02, 2.7243e-02,  ..., 4.4780e-03, 1.4411e-03,
         2.8681e-05],
        [7.3541e-05, 9.9366e-03, 1.8728e-02,  ..., 3.0753e-03, 1.3720e-03,
         2.3480e-05],
        ...,
        [1.0302e-04, 1.5146e-04, 1.4947e-04,  ..., 1.3166e-01, 1.8639e-03,
         2.2022e-05],
        [5.4333e-05, 9.5053e-05, 1.0704e-04,  ..., 2.0794e-03, 5.8460e-02,
         5.8733e-05],
        [1.6994e-03, 3.2339e-03, 4.6647e-03,  ..., 1.8222e-04, 4.0314e-03,
         4.0162e-02]]), 'X': tensor([[  0.6757, -11.0361,  -1.3047],
        [  3.2159, -10.8166,  -1.0406],
        [  3.2647, -10.8066,  -0.3768],
        [  0.8940, -11.0402,  -0.9783],
        [  4.8051, -10.3703,   0.5268],
        [  5.7485,  -9.9811,  -3.2355],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.0275,  -5.0666,   8.4154],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4157,   0.0821,  -5.0025],
        [  4.2612, -10.5514,  -6.6268],
        [  6.3205,   1.7068,  -3.8619],
        [ 10.0995,  -6.4739,   2.0974],
        [  5.1667,  -2.6543,   1.2879],
        [  5.9193,  -1.3176,   5.1739],
        [  5.8889,  -1.3916,   8.8993],
        [  9.2811,  -7.4336,   5.1771],
        [ -6.0175,  -8.4541,  -9.8408],
        [  5.5294,  -2.1106,   9.0626],
        [ 10.9875,   7.2207,  -8.8549],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.2026,  12.1774,  -1.5887],
        [ -5.1369,   6.7755,   8.3836],
        [ -9.1907,  -3.3180,  -5.5943],
        [  9.9058,  -6.7203,   5.6928],
        [  5.5294,  -2.1106,   9.0626],
        [ -3.0008, -10.2827,  -8.9480],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.1667,  -2.6543,   1.2879],
        [  5.0348,  -2.8229,   3.5436],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -6.4822,  -8.0252,  -3.8983],
        [  7.5784,  -8.9141,   5.8410],
        [  5.9193,  -1.3176,   5.1739],
        [  0.7275,  14.0667,  -0.9815],
        [  7.5309,  11.2617,   7.2656],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -5.1398,  -9.1367,   6.2074],
        [ -0.0254, -10.9932,   3.2200],
        [  9.0004,  -7.7195,   4.6945],
        [  5.4429,  -2.2529,   9.7306],
        [  5.3602,  -2.3808,   7.2098],
        [  9.1409,  -7.5789,   1.5814],
        [ 10.1798,   8.5002,  -5.8893],
        [  1.7815,  -4.7481,   4.4289],
        [  5.1890,   4.7318,   0.9515],
        [  4.3554,   5.7393,  -9.0766],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 12.4691,   2.8525,  -5.5110],
        [  9.9217,  -6.7006,  -5.6728],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.1222,  -0.7261,   8.8204],
        [  5.9587,  12.3216,   9.1695],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.3122,  -9.6994,  -6.3282],
        [  2.5663,   7.0526,  -6.8769],
        [ -9.2508,   1.0572,   2.9978],
        [  0.1497,   7.8334,  -5.4412],
        [  2.7685, -10.8982,  -6.8928],
        [ -8.3900,   3.3727,   3.1650],
        [  2.5469,  13.7192,  -9.6935],
        [  4.3459,  -3.5327,   0.3667],
        [  5.5101,  -2.1433,   9.5442],
        [ 10.7566,   7.6212,  -2.3279],
        [ -3.1671,   7.6158,   6.0765],
        [  3.6469,   6.3624,  -1.8414],
        [  5.7802,  -1.6363,  -6.2782],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.9663,   6.1018,  -5.4104],
        [  3.6249, -10.7255,   5.6605],
        [  6.3583,   1.1729,   8.7835],
        [ -7.4733,   4.7555,   6.2490],
        [  4.8674,  12.8875,   6.7622],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -5.9987,   6.1949,  -6.2114],
        [ 10.9916,   7.2133,  -0.9179],
        [  7.0642,  11.6104,  -9.8316],
        [  3.3943,   6.5483,   1.0790],
        [ 12.1588,   4.3364,  -8.0867],
        [ -7.0825,   5.2073,  -5.0497],
        [  8.6420,  -8.0580,   6.5781],
        [ 12.5927,   0.4199,  -9.4997],
        [ -2.0721,   7.8432,   6.6380],
        [  2.0449,  13.8402,  -7.1642],
        [  1.9312,  13.8649,   8.3563],
        [  2.7295,  13.6703,  -8.9328],
        [  4.0362,  13.2397,  -7.0030],
        [ -7.6719,  -6.6260,   7.1468],
        [ -9.3616,  -2.4749,   0.1793],
        [ 12.5717,   1.9266,   7.4903],
        [  3.2647, -10.8066,  -0.3768],
        [  1.4489,   7.5256,   4.4529],
        [ 11.6634,   5.8050,   9.1590],
        [ -0.4341, -10.9473,   7.4509],
        [ -5.6423,   6.4537,  -1.2091],
        [ -1.6220,   7.8917,   0.0315],
        [ -1.0012, -10.8576,   2.2867],
        [ 11.3691,   6.4760,   1.0787],
        [  6.0073,   3.0986,   6.7937],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.1203,  -2.7151,   6.2110],
        [ -2.0229, -10.6163,   8.6930]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0977, 0.1171,  ..., 0.6281, 0.8477, 0.7667],
        [0.0977, 0.0000, 0.0466,  ..., 0.5401, 0.7587, 0.7560],
        [0.1171, 0.0466, 0.0000,  ..., 0.5542, 0.7306, 0.7101],
        ...,
        [0.6281, 0.5401, 0.5542,  ..., 0.0000, 0.6180, 1.1007],
        [0.8477, 0.7587, 0.7306,  ..., 0.6180, 0.0000, 0.7753],
        [0.7667, 0.7560, 0.7101,  ..., 1.1007, 0.7753, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.05672136, 11.2845608 , 11.28895489, 11.07637531, 11.42947632,
       11.51811603, 12.07818871, 12.22276278, 12.13568598,  5.57341623,
        9.41605684, 11.37939695,  6.54693987, 11.99633819,  5.80860165,
        6.06416957,  6.05113217, 11.89105259, 10.37698435,  5.91854397,
       13.14777433,  9.78147542, 13.6660671 ,  8.50268322,  9.7712321 ,
       11.9702762 ,  5.91854397, 10.71162681,  9.41244865, 13.00413829,
        5.80860165,  5.77219077, 10.00410625, 10.31613269, 11.70016217,
        6.06416957, 14.08549802, 13.54774355, 12.4217523 , 10.48315436,
       10.99326577, 11.85743425,  5.89074104,  5.86519525, 11.87412439,
       13.26209545,  5.0713364 ,  7.02253235,  7.20482224, 12.53285633,
       12.52372629, 12.79126785, 11.97238624, 10.33712258,  6.16512856,
       13.68672855, 10.39942497,  5.94541803, 11.57249984,  7.50500064,
        9.3109884 ,  7.83486943, 11.24434374,  9.04255655, 13.95361098,
        5.60064319,  5.91222346, 13.18278606,  8.24808832,  7.33350642,
        6.00732193,  8.49161154,  7.27759233, 11.32149259,  6.46560181,
        8.8580671 , 13.77604293, 10.52357479,  8.62329466, 13.1471313 ,
       13.59056613,  7.37576154, 12.9089559 ,  8.7907923 , 11.8159254 ,
       12.59969936,  8.11227668, 13.99048094, 13.99877413, 13.94008948,
       13.84126197, 10.1371552 ,  9.68323686, 12.71843462, 11.28895489,
        7.66377554, 13.02818702, 10.95593729,  8.5724042 ,  8.05668609,
       10.9036226 , 13.08414689,  6.7593917 ,  6.07992623,  5.79562691,
       10.80728788])}
{'P': tensor([[8.3625e-03, 3.3015e-01, 2.7241e-01,  ..., 2.1624e-06, 1.8971e-05,
         2.6553e-03],
        [5.2723e-05, 1.4826e-02, 1.5960e-02,  ..., 8.5444e-06, 5.9484e-05,
         4.5132e-03],
        [3.6780e-05, 5.0235e-03, 9.4680e-03,  ..., 6.4226e-06, 3.7758e-05,
         2.6784e-03],
        ...,
        [2.3784e-07, 2.2533e-07, 2.4609e-07,  ..., 4.6352e-02, 4.8211e-04,
         4.2464e-06],
        [8.8919e-06, 7.6288e-06, 7.5215e-06,  ..., 1.2717e-04, 4.4620e-02,
         1.3041e-04],
        [8.5304e-04, 1.7118e-03, 1.7087e-03,  ..., 5.0071e-04, 1.4170e-02,
         1.8040e-01]]), 'X': tensor([[ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0967, 0.1041,  ..., 1.3380, 0.5605, 0.5827],
        [0.0967, 0.0000, 0.0466,  ..., 1.2656, 0.4640, 0.4864],
        [0.1041, 0.0466, 0.0000,  ..., 1.2370, 0.4726, 0.4960],
        ...,
        [1.3380, 1.2656, 1.2370,  ..., 0.0000, 1.0045, 1.0066],
        [0.5605, 0.4640, 0.4726,  ..., 1.0045, 0.0000, 0.0252],
        [0.5827, 0.4864, 0.4960,  ..., 1.0066, 0.0252, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.07637531, 11.2845608 , 11.28895489, 11.42947632, 11.56146873,
       11.63383221, 12.35421461, 13.66896739,  7.34250948,  5.19577146,
       11.10066126,  7.02326515,  9.87720102,  5.64865173, 13.97430191,
       10.71162681,  5.21280156,  9.17595899,  7.54620849, 13.2778378 ,
       13.103709  ,  8.40445872,  5.94541803,  5.16611782,  5.21280156,
        8.68979052,  9.48969978, 14.03618649,  5.94541803,  5.91408504,
        8.86082124,  9.94285511, 12.84056604,  8.72896381,  7.39599986,
        5.13616224, 13.58333857,  6.19633139, 13.79762415, 11.38774149,
       13.89264822,  8.53501777,  8.09842704,  9.99211663,  6.50409372,
       13.50484539, 12.97370371, 12.12641969,  4.74468406, 13.26590071,
       10.32715715, 13.66896739, 10.00794468, 13.27717982,  4.73784222,
       13.3614598 , 13.3614598 ,  5.76165934, 14.02058621,  8.1137452 ,
       13.59673406, 11.35182589,  5.39098669,  5.01114719, 13.66051884,
       10.06437598,  8.72896381, 12.22276278, 10.04916678, 12.16901615,
        9.93553424, 11.42947632, 13.79762415, 10.88618114,  8.98765644,
        8.92098752,  7.44805901,  9.47369628,  9.88571713,  5.35774573,
        5.0209005 , 10.31886066,  6.42689705,  7.06169391,  8.12043983,
        6.84796202, 11.35182589,  5.5513072 , 12.05395056,  8.04718433,
       12.04924159,  6.25586297,  7.70897675, 10.794755  ,  8.50268322,
       11.07637531, 11.10066126,  6.81084724,  8.62260327, 13.09426142,
        5.41348121,  9.11473391,  9.87720102, 13.34587027,  7.26074925,
        5.48276038])}
{'P': tensor([[7.4450e-03, 2.3508e-01, 2.0976e-01,  ..., 1.1202e-05, 1.5687e-03,
         2.6147e-05],
        [2.0112e-05, 6.9899e-03, 3.2115e-05,  ..., 1.7066e-05, 8.7014e-04,
         4.1362e-05],
        [1.1035e-04, 3.6711e-01, 8.4771e-03,  ..., 1.2463e-05, 1.1658e-03,
         2.9797e-05],
        ...,
        [6.3527e-06, 6.8777e-06, 5.6565e-06,  ..., 9.1341e-02, 3.0796e-05,
         1.5937e-02],
        [2.1836e-05, 9.7126e-05, 3.2112e-05,  ..., 1.2188e-03, 2.3928e-02,
         2.1432e-03],
        [8.4670e-06, 9.6593e-06, 7.8044e-06,  ..., 7.2273e-02, 3.3899e-05,
         7.3644e-02]]), 'X': tensor([[ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1234, 0.0721,  ..., 1.0814, 0.7136, 0.9858],
        [0.1234, 0.0000, 0.0517,  ..., 1.0397, 0.7030, 0.9363],
        [0.0721, 0.0517, 0.0000,  ..., 1.0603, 0.7098, 0.9600],
        ...,
        [1.0814, 1.0397, 1.0603,  ..., 0.0000, 0.4181, 0.1211],
        [0.7136, 0.7030, 0.7098,  ..., 0.4181, 0.0000, 0.3694],
        [0.9858, 0.9363, 0.9600,  ..., 0.1211, 0.3694, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.10066126, 11.35345345, 11.18839169, 11.28093507, 11.38774149,
       11.50791859,  8.09704052, 12.3489324 ,  6.38399789,  8.49161154,
        6.65368551,  8.8580671 ,  8.38252937,  6.54127431,  4.90505566,
       12.22503754,  9.27474386, 13.20988996,  6.1463896 ,  8.51088913,
       12.58006038,  5.04028538, 13.43127131,  8.24808832,  9.60080639,
        5.05963217,  9.6282763 ,  7.10378919,  5.47702766,  7.58378655,
        6.19633139, 11.52733581, 11.04313641,  8.66273171,  6.32572323,
        8.62162372, 11.86239961,  6.34059021,  6.38968357, 10.7506423 ,
        7.32754882, 10.35464916, 13.77222752, 13.39372426,  6.652742  ,
        4.75303127,  8.81930525,  5.1805345 ,  7.18354885, 11.24434374,
        8.20436764, 11.77124373,  9.23071698,  8.53014915, 13.25017211,
       14.0393949 , 12.87752052, 13.32374056,  7.38844703,  7.92607342,
        8.47371868, 11.48434767,  5.94909709, 11.1913598 , 13.92390689,
        9.50654967, 12.4013153 ,  9.78285424,  9.73579738, 13.69162795,
        8.31625171,  5.41348121,  8.34949383, 11.67875855, 13.94008948,
        9.12220271, 14.02526505,  6.80370254,  8.29640613, 13.66051884,
       13.3426541 , 13.89264822, 12.19805394, 11.67875855,  7.19513806,
        8.22728399, 13.91198191, 11.2845608 ,  4.78002665, 13.01728313,
        6.54127431,  9.41970031, 11.8159254 , 13.83139024, 13.88683436,
        9.43112957, 11.00812996, 14.05687815,  7.80625981, 11.73848065,
        8.41171369, 12.3344536 ,  6.18668463,  6.76661595, 11.63502679,
        6.67108662])}
{'P': tensor([[5.1363e-03, 3.1283e-01, 3.0601e-01,  ..., 2.8049e-07, 6.5148e-06,
         3.5146e-06],
        [8.3749e-05, 1.4644e-02, 2.7433e-02,  ..., 5.3943e-07, 1.0058e-05,
         6.5641e-06],
        [7.3488e-05, 5.3686e-03, 1.3128e-02,  ..., 5.1482e-07, 9.2480e-06,
         5.7730e-06],
        ...,
        [3.5280e-04, 2.8134e-04, 3.2018e-04,  ..., 8.9966e-02, 4.2731e-02,
         4.2867e-04],
        [7.7726e-03, 7.4445e-03, 8.7659e-03,  ..., 1.8507e-04, 6.9236e-02,
         7.1783e-04],
        [9.8635e-02, 5.6213e-02, 5.6643e-02,  ..., 2.7848e-05, 2.5980e-03,
         3.1519e-02]]), 'X': tensor([[  1.2728, -11.0372,   9.8121],
        [  2.5128, -10.9365,   9.9645],
        [  2.5636, -10.9293,   9.6519],
        [  5.1519, -10.2387,   9.8191],
        [  1.3767, -11.0341,   9.9415],
        [  4.3523, -10.5232,   8.2138],
        [ -8.9282,   2.1648,   8.4894],
        [  3.3839,  -4.1966,   8.3574],
        [ -9.4196,   0.0478,   9.4140],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -8.6229,   2.9068,   2.4429],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.3616,  -2.4749,   0.1793],
        [  2.2644,  -4.6434,   9.7508],
        [  6.9284,  11.7061,  -0.3732],
        [  6.3048,   0.1469,  -9.1691],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 12.4690,  -0.8528,  -9.3374],
        [  5.0617,  12.7957,   8.3544],
        [  8.9284,  10.0036,  -6.9210],
        [  7.5784,  -8.9141,   5.8410],
        [  2.9461,  -4.4077,  -0.4831],
        [ 11.5456,   6.0861,   8.1101],
        [ -0.5385,   7.9037,   6.3830],
        [ -9.3419,   0.5972,  -1.2841],
        [  4.6231,   5.4556,   8.4973],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.8889,  -1.3916,   8.8993],
        [ -2.6069, -10.4298,  -6.1995],
        [ -2.2188,   7.8218,   8.6064],
        [  2.2373,  -4.6506,  -1.7334],
        [ -7.5720,   4.6310,   2.1717],
        [ 12.2328,  -2.0376,  -1.1234],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.0345,   3.0175,  -1.4424],
        [ -8.5149,  -5.1905,   1.5239],
        [ -5.2589,   6.7024,  -5.5809],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.3460,   1.4252,   1.2291],
        [  2.3373,  -4.6232,  -5.2246],
        [ -5.6174,   6.4708,  -1.7339],
        [  6.1755,   2.5233,  -2.0042],
        [  7.4132,  11.3526,   0.3322],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.6952,  -4.8060,   8.8976],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.0829,  -5.9940,   5.0564],
        [ 10.9599,  -5.1849,  -3.5526],
        [  1.8766,   7.3694,   1.1887],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.5662,  13.9375,   4.5142],
        [  4.8051, -10.3703,   0.5268],
        [  2.2373,  -4.6506,  -1.7334],
        [ 10.4994,  -5.9193,  -7.5153],
        [  6.0560,  -0.9421,  -5.0334],
        [  5.8930,  12.3592,  -8.9201],
        [  3.4604,  -4.1545,   1.8091],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.1140,  -3.7220,  -0.1273],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.2508,   1.0572,   2.9978],
        [ -2.5683, -10.4433,  -2.9182],
        [  6.3460,   1.4252,   1.2291],
        [ -6.3489,   5.9117,   6.5161],
        [  6.9956,  11.6590,  -7.6876],
        [  3.8182,  -3.9347,  -1.7961],
        [  1.9503,  -4.7172,  -6.2313],
        [ -9.0604,   1.7707,   1.1066],
        [ -6.3489,   5.9117,   6.5161],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.3444,   0.5821,   6.9101],
        [  3.2647, -10.8066,  -0.3768],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -5.9872,   6.2037,  -9.4477],
        [ 10.7566,   7.6212,  -2.3279],
        [ -9.3444,   0.5821,   6.9101],
        [  2.7295,  13.6703,  -8.9328],
        [  5.1472,  -2.6801,   8.1194],
        [ -8.7544,   2.6096,   3.3146],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.3928,   7.1404,  -1.7035],
        [  4.6063,  13.0051,  -3.1456],
        [  7.0358,  -9.2784,  -9.8479],
        [ -8.6801,   2.7809,  -0.8443],
        [  8.9284,  10.0036,  -6.9210],
        [  6.0345,   3.0175,  -1.4424],
        [ -8.1905,   3.7253,   4.1175],
        [  9.3753,   9.5171,   3.9131],
        [  3.3633,  13.4794,   0.5193],
        [  3.9770, -10.6341,   8.0783],
        [ -8.3535,  -5.5118,   9.8399],
        [  3.7121,   6.3117,   0.6738],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.0719, -10.8449,  -4.1673],
        [  7.9141,  -8.6656,   9.6948],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.0140,  -1.0661,  -2.5040],
        [ -1.8788, -10.6567,   3.7068],
        [  8.5198,  10.4091,  -8.6486],
        [ 10.1798,   8.5002,  -5.8893],
        [  6.3301,   0.3638,  -7.5685],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -3.9456,  -9.8542,   0.7466],
        [ -5.6169,  -8.7844,   8.9057]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0765, 0.0674,  ..., 1.1613, 0.6810, 0.2984],
        [0.0765, 0.0000, 0.0100,  ..., 1.1970, 0.7229, 0.3746],
        [0.0674, 0.0100, 0.0000,  ..., 1.1891, 0.7144, 0.3651],
        ...,
        [1.1613, 1.1970, 1.1891,  ..., 0.0000, 0.4852, 1.0348],
        [0.6810, 0.7229, 0.7144,  ..., 0.4852, 0.0000, 0.5584],
        [0.2984, 0.3746, 0.3651,  ..., 1.0348, 0.5584, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.11038551, 11.22142195, 11.22596994, 11.46176259, 11.11969815,
       11.38774149,  9.18690018,  5.39098669,  9.41970031, 12.04924159,
        9.09963894, 13.05148826,  9.68323686,  5.16611782, 13.60275211,
        6.30647746,  9.78286733, 12.4980845 , 13.76047999, 13.40850328,
       11.70016217,  5.30158497, 13.05148826,  7.92200342,  9.36093562,
        7.15100303, 12.12641969, 12.17901033,  6.05113217, 10.7506423 ,
        8.1303945 ,  5.16077764,  8.87587457, 12.4013153 , 11.72749923,
        6.74686071,  9.97220799,  8.51928531, 12.18998195,  6.50409372,
        5.18047638,  8.56889738,  6.67108662, 13.55868684,  5.69913817,
        4.85603886, 12.77402351, 12.71843462, 10.06286074, 12.12448912,
        7.60463349, 10.27572673, 14.02526505, 11.42947632,  5.16077764,
       12.05300943,  6.12886024, 13.6922425 ,  5.40689332, 10.63038565,
        5.54776934, 10.1979669 ,  9.3109884 , 10.75443106,  6.50409372,
        8.67502348, 13.59673406,  5.48276038,  5.10444097,  9.23178204,
        8.67502348,  5.0209005 ,  9.36256096, 11.28895489, 12.00995898,
        8.62162372, 13.18278606,  9.36256096, 13.94008948,  5.80312949,
        9.13507321,  5.84450127,  7.53063885, 13.79676654, 11.64436684,
        9.11473391, 13.40850328,  6.74686071,  8.99791316, 13.35927527,
       13.89264822, 11.35345345, 10.00802627,  7.32233733,  5.9510489 ,
       11.27160035, 11.73567759,  9.14973319,  6.10774544, 10.82106505,
       13.45124917, 13.26209545,  6.34059021, 10.14436912, 10.61472636,
       10.42667093])}
{'P': tensor([[5.5623e-03, 3.6858e-01, 3.1193e-01,  ..., 5.0122e-07, 7.7499e-07,
         3.6281e-05],
        [8.4906e-05, 1.4689e-02, 2.7517e-02,  ..., 1.0482e-06, 1.5656e-06,
         8.2427e-05],
        [8.1375e-05, 5.2361e-03, 1.2804e-02,  ..., 9.3087e-07, 1.3707e-06,
         8.0757e-05],
        ...,
        [1.2496e-03, 8.1377e-04, 8.5793e-04,  ..., 3.5621e-02, 1.1313e-01,
         7.2547e-04],
        [1.3049e-02, 8.1851e-03, 8.4696e-03,  ..., 3.5134e-03, 1.1352e-01,
         1.7158e-03],
        [2.2966e-04, 2.7872e-04, 3.1298e-04,  ..., 9.5454e-05, 7.4233e-05,
         1.1161e-01]]), 'X': tensor([[ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0637, 0.0548,  ..., 0.5155, 0.4247, 0.9273],
        [0.0637, 0.0000, 0.0100,  ..., 0.5628, 0.4813, 0.8995],
        [0.0548, 0.0100, 0.0000,  ..., 0.5532, 0.4713, 0.8995],
        ...,
        [0.5155, 0.5628, 0.5532,  ..., 0.0000, 0.1529, 0.9221],
        [0.4247, 0.4813, 0.4713,  ..., 0.1529, 0.0000, 1.0089],
        [0.9273, 0.8995, 0.8995,  ..., 0.9221, 1.0089, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.11969815, 11.22142195, 11.22596994, 11.46176259, 11.38774149,
       11.35345345, 12.38881635,  8.7907923 , 10.31613269,  9.5642993 ,
        8.31625171,  9.38232121, 12.50996828,  5.16077764, 12.91505111,
       12.18045978,  6.2134961 , 13.77222752,  5.26342203, 10.51437947,
       13.59170809,  4.85603886, 10.08001894, 13.6922425 , 14.07623266,
        5.79391825, 11.50791859, 13.1471313 , 13.39372426,  8.16584537,
       13.15525803,  6.4020404 , 14.02532758, 13.56367857,  7.57946459,
       13.59673406, 13.6922425 ,  4.7663526 ,  5.30158497,  9.44486609,
        8.23525433,  8.83341608, 11.51811603,  9.97822473, 10.71162681,
       10.08001894,  6.27715105,  6.00732193,  4.73784222, 10.61417181,
       12.0495521 ,  5.18047638,  7.24326992,  6.05165248,  9.68324553,
       13.41713661, 10.66631784, 10.27716434,  5.0209005 ,  5.83560009,
       13.744387  ,  6.39438044,  7.62232307,  4.75303127,  8.04718433,
        8.06990483,  5.36061974,  8.44940661, 12.53285633,  6.23262289,
       11.4890861 ,  5.14607014, 13.14777433, 13.103709  , 11.54672755,
       11.48434767,  7.5092834 , 12.32408516, 12.77402351,  7.15990676,
       13.58101794, 13.27994304, 13.3426541 ,  9.69629   ,  8.95133961,
        5.5205189 ,  7.53063885, 13.20988996,  6.25586297, 13.96746816,
        7.01475376,  8.20436764,  6.97106912,  7.39599986, 11.10066126,
       12.10222204,  8.36445758,  8.62260327,  5.20659892, 13.04217487,
        8.89301516,  7.83486943,  9.04255655,  9.68324553,  9.84314712,
        5.16077764])}
{'P': tensor([[6.2146e-03, 2.4263e-01, 1.9141e-01,  ..., 4.9348e-05, 2.3453e-06,
         7.5190e-03],
        [2.3176e-05, 6.3752e-03, 8.1255e-05,  ..., 8.6438e-05, 3.4768e-06,
         7.6492e-03],
        [5.0191e-05, 2.4558e-01, 6.6770e-03,  ..., 6.7947e-05, 2.9931e-06,
         6.0631e-03],
        ...,
        [1.6640e-05, 1.7366e-05, 1.5984e-05,  ..., 8.5153e-02, 5.8940e-02,
         3.2479e-05],
        [7.9327e-05, 5.3292e-05, 5.6706e-05,  ..., 1.2938e-03, 3.7968e-01,
         4.1296e-05],
        [3.7365e-05, 1.2567e-04, 8.0877e-05,  ..., 3.9343e-03, 9.6866e-05,
         4.9625e-02]]), 'X': tensor([[  1.5383, -11.0274,  -4.9666],
        [  3.8111, -10.6787,  -4.6539],
        [  3.0719, -10.8449,  -4.1673],
        [  3.9591, -10.6391,  -6.2557],
        [  5.4426, -10.1181,  -5.1268],
        [  5.3924, -10.1396,  -4.0863],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -2.5683, -10.4433,  -2.9182],
        [ -2.2188,   7.8218,   8.6064],
        [ -4.2040,   7.2486,   2.9875],
        [  4.5340,  -3.3629,   8.9652],
        [  7.3724,  -9.0577,  -8.9843],
        [ -3.4488, -10.0934,   2.5195],
        [ -3.4488, -10.0934,   2.5195],
        [ 10.4994,  -5.9193,  -7.5153],
        [  7.5309,  11.2617,   7.2656],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.2038,   2.4020,  -3.1692],
        [ -8.8835,  -4.3178,   9.3080],
        [ 10.3251,   8.2928,  -3.0006],
        [  0.1252, -11.0062,   7.2564],
        [ -2.2508, -10.5479,  -2.7235],
        [ -8.7078,   2.7182,   1.8146],
        [  0.6396,  14.0772,  -6.3786],
        [  0.2435,   7.8191,  -6.4709],
        [  5.8979,   3.3958,  -3.8713],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.0506,   4.9277,  -9.5094],
        [  4.0362,  13.2397,  -7.0030],
        [ 12.5650,   2.0132,   4.6272],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.8450,  -9.9356,   3.0767],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.0039,   7.3173,  -1.5885],
        [ -9.4753,  -0.8012,  -5.2050],
        [  7.5930,  -8.9037,   7.2323],
        [  5.7963,  -1.6018,   1.8856],
        [  4.2275,  -3.6320,  -0.2714],
        [  3.1756, -10.8247,   6.8882],
        [  2.1771, -10.9776,  -3.2007],
        [ 11.1085,  -4.9204,   0.0736],
        [ -7.0825,   5.2073,  -5.0497],
        [  5.5158,  -2.1337,   3.5246],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.4524,  -1.6813,  -6.6862],
        [  5.9883,   3.1536,   6.2320],
        [  5.0617,  12.7957,   8.3544],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.1497,   7.8334,  -5.4412],
        [ -7.8359,   4.2735,   7.2943],
        [ -5.1974,   6.7396,   3.3484],
        [  6.2975,   0.0947,   5.1791],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.5382,   7.0672,   4.5097],
        [ -7.3446,   4.9113,  -5.9356],
        [  7.3724,  -9.0577,  -8.9843],
        [ -8.1666,   3.7653,   3.6075],
        [  3.4604,  -4.1545,   1.8091],
        [  0.1677,  -4.7447,  -6.1529],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.6426,  -4.9136,   9.4941],
        [  6.6319,  11.9066,   3.5165],
        [  1.1006,   7.6318,  -2.4877],
        [  6.2242,  -0.3171,   7.4978],
        [  5.4426, -10.1181,  -5.1268],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.2630,   2.1057,  -5.2136],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 11.9695,  -2.9407,  -6.3498],
        [  6.1372,   2.6726,  -6.7885],
        [ -8.5149,  -5.1905,   1.5239],
        [ 12.4996,   2.6335,  -6.3166],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.1196,  -2.7160,   4.2705],
        [  5.2199,   4.6861,  -1.9304],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.1575,  12.2046,   3.3257],
        [  0.9478,  -4.8144,   0.6371],
        [ -5.1046,  -9.1610,   2.3418],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.7792,  11.8084,   1.6251],
        [  6.0890,  -0.8382,   2.9166],
        [ -6.8466,  -7.6488,   1.9648],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.9659,  -4.8144,  -0.9594],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.0574,   1.7801,   8.8986],
        [  2.4356,  -4.5942,  -5.0808],
        [ -9.4731,  -0.7120,  -6.0271],
        [  4.7739,  -3.1218,  -4.1188],
        [ -8.6050,   2.9451,  -0.7012],
        [ -4.7334,   6.9982,  -0.4392],
        [  0.6933,   7.7336,   2.1179],
        [  7.4132,  11.3526,   0.3322],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.9470,  13.6085,   2.9809],
        [  4.5814, -10.4485,   2.5496],
        [  4.5814, -10.4485,   2.5496],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.8617,   3.4853,  -9.8080],
        [  0.1460,   7.8340,  -9.4614],
        [  7.8375,  -8.7240,  -9.8913]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1240, 0.1021,  ..., 0.5798, 0.3330, 0.5159],
        [0.1240, 0.0000, 0.0473,  ..., 0.4675, 0.2366, 0.4237],
        [0.1021, 0.0473, 0.0000,  ..., 0.5110, 0.2838, 0.4709],
        ...,
        [0.5798, 0.4675, 0.5110,  ..., 0.0000, 0.2570, 0.1554],
        [0.3330, 0.2366, 0.2838,  ..., 0.2570, 0.0000, 0.1871],
        [0.5159, 0.4237, 0.4709,  ..., 0.1554, 0.1871, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.1341798 , 11.33837396, 11.27160035, 11.35182589, 11.4890861 ,
       11.48434767, 10.0674887 , 10.75443106,  8.1303945 ,  8.37954853,
        5.64500512, 11.67875855, 10.66631784, 10.66631784, 12.05300943,
       13.54774355, 12.00995898,  6.65259775,  9.87720102, 13.2430377 ,
       11.00694688, 10.78533838,  9.12220271, 14.09176259,  7.822854  ,
        6.80559669,  9.7751189 ,  7.05627461, 13.84126197, 12.72524615,
        9.97964013, 11.52733581, 10.33958639,  7.58668144,  9.50913849,
       11.70168755,  6.01356004,  5.57341623, 11.28093507, 11.1913598 ,
       12.1494023 ,  8.7907923 ,  5.91408504, 12.97370371,  9.60080639,
        6.76791483, 13.76047999,  5.69913817,  7.83486943,  8.92549752,
        8.51088913,  6.29821673,  5.46244988,  7.50917551,  8.8353674 ,
       11.67875855,  8.99276564,  5.40689332,  4.74770899, 11.83075005,
        9.94173708, 13.62896593,  7.71075226,  6.2322769 , 11.4890861 ,
        8.29022197,  9.62958973,  6.60752373, 12.3828957 , 12.32545637,
        6.69389431,  9.97220799, 12.77402351, 10.61472636,  5.79543038,
        7.01475376, 12.889695  , 12.45418236, 13.66991027,  4.90676242,
       10.48721382, 11.97238624,  9.37296102, 13.61600536,  6.1463896 ,
       10.26545792,  6.27905292,  4.91038884, 12.77402351,  9.23071698,
        5.19986797,  9.4997987 ,  5.70405525,  9.09502385,  8.4486771 ,
        7.76457095, 13.55868684,  5.95297459,  5.19065439, 13.92390689,
       11.4088043 , 11.4088043 , 12.49317563,  6.81961996,  7.83534987,
       11.72749923])}
{'P': tensor([[6.3508e-03, 2.7503e-01, 2.5777e-01,  ..., 3.9673e-07, 3.6559e-07,
         2.1589e-05],
        [2.2265e-05, 4.8460e-03, 3.6726e-01,  ..., 6.0331e-07, 3.8221e-07,
         1.1585e-05],
        [2.5356e-05, 6.5104e-05, 6.9908e-03,  ..., 9.5386e-07, 5.6333e-07,
         1.3845e-05],
        ...,
        [4.1843e-08, 4.1025e-08, 3.9616e-08,  ..., 1.6446e-02, 6.5912e-03,
         1.3219e-07],
        [7.9261e-06, 6.2147e-06, 5.6505e-06,  ..., 4.6353e-05, 9.9264e-02,
         2.2280e-05],
        [1.7520e-05, 4.0066e-05, 4.2942e-05,  ..., 1.2102e-05, 4.8217e-06,
         1.6555e-02]]), 'X': tensor([[  2.1440, -10.9811,   8.5281],
        [  3.9770, -10.6341,   8.0783],
        [  4.3523, -10.5232,   8.2138],
        [  5.6414, -10.0303,   7.8525],
        [  5.1519, -10.2387,   9.8191],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3266,   0.3292,   9.3502],
        [  5.5240,  -2.1198,   6.2055],
        [ -0.8675, -10.8815,   6.8561],
        [ -0.8885,   7.9160,   0.0571],
        [  8.2154,  -8.4262,   8.0171],
        [  2.3367,   7.1676,  -9.5263],
        [ -8.6381,  -4.9238,  -4.4077],
        [  5.6248,   3.9984,   8.4933],
        [ -5.1974,   6.7396,   3.3484],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -2.5683, -10.4433,  -2.9182],
        [  8.2422,  -8.4041,  -1.4962],
        [  4.8432,   5.1968,   1.5006],
        [  6.3048,   0.1469,  -9.1691],
        [  1.9241,   7.3503,   7.9004],
        [  2.5128, -10.9365,   9.9645],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.8318,   3.5567,  -4.7692],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.2939,  -2.4784,  -4.5840],
        [ -3.9456,  -9.8542,   0.7466],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.4356,  -4.5942,  -5.0808],
        [ 11.7349,  -3.5838,   2.1644],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.5506,  -5.1154,   6.3423],
        [ 11.7162,   5.6728,  -8.9372],
        [  1.4306,  13.9620,  -0.2075],
        [ 11.7349,  -3.5838,   2.1644],
        [ -5.3328,  -8.9992,   7.3440],
        [  0.2571,  -4.7594,   1.5618],
        [  0.1287, -11.0065,   6.5757],
        [  0.7685,  -4.8098,  -2.9310],
        [ 12.4837,   2.7510,   0.4449],
        [  0.5250, -11.0306,   5.5358],
        [  6.0047,   3.1062,  -6.7914],
        [  8.1043,  -8.5164,  -7.9290],
        [  7.0358,  -9.2784,  -9.8479],
        [  5.8325,  -1.5222,  -2.1341],
        [  8.6271,  -8.0715,  -7.4813],
        [ -8.3653,   3.4184,   9.3454],
        [ -0.2586,   7.8825,  -9.9840],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.9883,   3.1536,   6.2320],
        [  9.4516,  -7.2501,  -1.8868],
        [  5.6672,  -1.8631,   8.1020],
        [ 10.5057,  -5.9099,  -5.1599],
        [  9.4516,  -7.2501,  -1.8868],
        [  4.4049,   5.6892,  -4.4394],
        [  0.3493,  -4.7727,  -6.5682],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.1890,   4.7318,   0.9515],
        [  5.4336,  -2.2676,   5.6933],
        [  9.6662,   9.1725,   8.1056],
        [ -8.9033,   2.2329,  -8.8912],
        [ -6.4368,   5.8356,   3.3014],
        [  6.3505,   0.6235,  -8.5849],
        [  7.8982,  -8.6778,   0.0239],
        [  1.7903,  -4.7467,   0.1168],
        [  2.3200,   7.1756,   1.1938],
        [  7.5930,  -8.9037,   7.2323],
        [  5.5424,  -2.0885,   7.2487],
        [ -4.7389,   6.9953,   3.7638],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -9.4773,  -1.0024,   5.2599],
        [  1.6408,   7.4591,  -4.2294],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 10.9931,   7.2107,   7.7933],
        [  4.9337,  12.8566,  -3.6582],
        [  6.0345,   3.0175,  -1.4424],
        [  5.4426, -10.1181,  -5.1268],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.3354,  -4.7708,  -4.9125],
        [  1.3705,  13.9723,   7.9266],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.2022,   2.4092,   3.0349],
        [  2.3376,  -4.6231,   9.8436],
        [ -6.4844,  -8.0230,   8.5209],
        [ -0.8569,   7.9156,  -6.5557],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.0547,   7.2957,  -1.9343],
        [  2.6078, -10.9229,   6.2869],
        [ -9.1960,  -3.2964,   1.6921],
        [ -5.5525,   6.5146,  -4.4984],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3124,   0.2057,  -0.4593],
        [  2.1120,  -4.6818,   8.6144],
        [ 12.4996,   2.6335,  -6.3166],
        [  6.1929,   2.4501,  -6.9352],
        [  6.2902,   0.0451,   3.7116],
        [  1.3403,  -4.8011,  -1.1698],
        [  1.5246,  -4.7838,  -5.2095],
        [  2.1824,  -4.6647,   1.1985],
        [  2.6451,  -4.5250,  -0.5565],
        [  0.9445,   7.6736,   0.5095],
        [  7.0171,  11.6439,  -5.0178],
        [ -2.3317,   7.8034,  -3.6274],
        [  3.9591, -10.6391,  -6.2557]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.1737e-02, 8.0044e-02,  ..., 1.3593e+00, 9.1237e-01,
         1.1213e+00],
        [5.1737e-02, 0.0000e+00, 2.8441e-02,  ..., 1.3330e+00, 9.0665e-01,
         1.1245e+00],
        [8.0044e-02, 2.8441e-02, 0.0000e+00,  ..., 1.3223e+00, 9.0818e-01,
         1.1306e+00],
        ...,
        [1.3593e+00, 1.3330e+00, 1.3223e+00,  ..., 0.0000e+00, 6.3690e-01,
         7.5239e-01],
        [9.1237e-01, 9.0665e-01, 9.0818e-01,  ..., 6.3690e-01, 1.0727e-04,
         2.7448e-01],
        [1.1213e+00, 1.1245e+00, 1.1306e+00,  ..., 7.5239e-01, 2.7448e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.18839169, 11.35345345, 11.38774149, 11.50791859, 11.46176259,
       11.62158554,  6.33517349,  5.91677188, 10.91601655,  7.96575782,
       11.76830981,  7.53884104,  9.94285511,  6.90115901,  8.51088913,
        9.75401449, 10.75443106, 11.77124373,  7.10378919,  6.30647746,
        7.59795445, 11.22142195,  9.78285424,  6.83083863,  9.94173708,
        5.84533345, 10.61472636, 10.27716434,  5.19986797, 12.26997165,
       11.1913598 , 10.0674887 ,  9.96391045, 13.01728313, 14.03505624,
       12.26997165, 10.4606265 ,  4.7663526 , 11.00726483,  4.87083348,
       12.78327037, 11.04313641,  6.76056625, 11.75618513, 11.64436684,
        6.02789634, 11.81423161,  9.0368458 ,  7.88677538, 11.75807252,
        6.76791483, 11.9120321 ,  5.96556248, 12.05395056, 11.9120321 ,
        7.19513806,  4.78544875, 12.22503754,  7.02253235,  5.88782913,
       13.32557133,  9.17905491,  8.68832398,  6.38105015, 11.73397309,
        5.07306274,  7.54127425, 11.70168755,  5.92281142,  8.44940661,
       12.32545637,  9.53015376,  7.6374603 , 13.97857591,  9.43737523,
       13.14690137, 13.77075174,  6.74686071, 11.4890861 , 10.5575601 ,
        4.78258092, 14.0393949 , 12.18045978,  6.65368551,  5.1805345 ,
       10.31584102,  7.96181288,  8.8353674 ,  7.57946459, 11.22993218,
        9.76896904,  8.55982105,  7.96181288,  6.31576264,  5.13616224,
       12.77402351,  6.65991877,  6.29034958,  4.98461983,  5.0209005 ,
        5.14999799,  5.24137909,  7.73151309, 13.59480906,  8.14433542,
       11.35182589])}
{'P': tensor([[9.6787e-03, 2.2233e-01, 2.1370e-01,  ..., 5.8911e-04, 2.3145e-03,
         1.0495e-06],
        [2.3476e-05, 1.2210e-02, 1.1063e-03,  ..., 1.8446e-04, 4.1193e-03,
         1.7377e-06],
        [7.3918e-05, 1.4793e-01, 2.9460e-02,  ..., 5.0039e-04, 1.0610e-02,
         3.8669e-06],
        ...,
        [6.4329e-03, 1.2639e-02, 1.1586e-02,  ..., 8.0649e-02, 5.4061e-02,
         1.6145e-04],
        [2.9781e-04, 7.5914e-04, 6.8667e-04,  ..., 2.6109e-03, 1.5805e-01,
         6.3142e-04],
        [5.3345e-05, 1.9213e-05, 2.0221e-05,  ..., 5.0380e-04, 7.4910e-05,
         6.0389e-02]]), 'X': tensor([[ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1657, 0.1689,  ..., 0.4485, 0.5470, 0.1912],
        [0.1657, 0.0000, 0.0571,  ..., 0.2835, 0.3831, 0.2041],
        [0.1689, 0.0571, 0.0000,  ..., 0.3017, 0.3839, 0.2528],
        ...,
        [0.4485, 0.2835, 0.3017,  ..., 0.0000, 0.1381, 0.4071],
        [0.5470, 0.3831, 0.3839,  ..., 0.1381, 0.0000, 0.5354],
        [0.1912, 0.2041, 0.2528,  ..., 0.4071, 0.5354, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.1913598 , 11.51811603, 11.48434767, 11.59512194, 11.27160035,
       11.33837396, 10.8207297 ,  8.29203356,  7.16598273,  4.90505566,
       10.6910768 , 13.46553623,  8.94395628,  6.1463896 ,  7.89745107,
       10.01804223, 11.54672755, 13.59056613,  5.14124612,  9.43737523,
        6.65991877, 13.76047999, 13.85158417,  9.41970031,  9.41244865,
        8.53501777,  6.83083863,  9.15976746,  5.08372149, 12.95702647,
       10.84493732,  5.79543038,  5.53175275,  6.23262289,  6.21275016,
        7.32233733, 10.51437947, 13.59056613, 13.04156075, 12.62711252,
        7.96575782,  8.29640613,  8.67502348, 10.31584102,  8.77913428,
        6.67121748, 10.4977132 , 13.35013374, 13.2430377 , 11.62770332,
       14.02526505,  9.94173708,  8.51928531, 13.59480906, 11.42420508,
        6.35029274,  4.75303127,  8.74470218,  8.47371868, 11.85743425,
        4.89869205, 10.60226614,  7.96575782,  8.40445872,  7.32233733,
        9.7712321 ,  8.89301516,  9.15866607,  6.7166297 ,  7.48538905,
        9.17016857, 10.09979747, 13.2778378 ,  4.85603886,  5.30158497,
        7.62232307, 12.67576792, 10.06023665,  8.68832398,  5.5366698 ,
        9.75075273, 11.87665501, 13.59673406, 13.04156075, 12.38881635,
       12.4980845 ,  9.84314712, 13.09426142, 12.95702647, 11.54672755,
        8.66273171, 13.3607379 ,  5.05963217,  5.64500512, 13.54774355,
       13.58333857, 11.4424816 , 11.83570132,  6.28961988,  4.7663526 ,
       12.32408516, 14.03618649,  7.46248947,  4.87083348,  5.84533345,
        8.5724042 ])}
{'P': tensor([[7.6736e-03, 2.6473e-01, 1.9724e-01,  ..., 5.0001e-06, 1.6722e-06,
         2.3406e-07],
        [7.9297e-05, 2.7563e-02, 1.3517e-03,  ..., 4.2584e-05, 1.2905e-05,
         4.6504e-07],
        [6.7642e-05, 7.1543e-02, 1.1696e-02,  ..., 2.0718e-05, 6.6938e-06,
         2.8841e-07],
        ...,
        [8.9391e-08, 1.1041e-07, 1.2473e-07,  ..., 2.0861e-02, 1.0347e-02,
         1.6794e-06],
        [3.7319e-08, 4.5000e-08, 5.3549e-08,  ..., 2.2324e-03, 1.6715e-02,
         1.3739e-06],
        [1.5691e-04, 7.2818e-05, 1.3748e-04,  ..., 2.3331e-06, 2.0781e-06,
         4.4234e-02]]), 'X': tensor([[  2.2849, -10.9655,  -7.6542],
        [  4.9452, -10.3187,  -7.8621],
        [  4.2612, -10.5514,  -6.6268],
        [  3.1892, -10.8220,  -6.8409],
        [  6.0467,  -9.8369,  -6.5816],
        [  3.9591, -10.6391,  -6.2557],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.4732,  -2.2041,  -4.3830],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.3534,   1.2990,  -7.2942],
        [ 12.5199,  -0.4711,   3.0436],
        [ -9.1813,  -3.3550,   1.8827],
        [ 11.3118,   6.5957,  -4.5667],
        [ -4.0633,  -9.7927,  -3.0825],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.1907,  -3.3180,  -5.5943],
        [  1.6308,  13.9254,  -2.0984],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.8400,   2.3991,   1.3457],
        [ 12.5224,  -0.4497,   4.8448],
        [  5.5240,  -2.1198,   6.2055],
        [  1.7301,   7.4262,   5.0522],
        [ -9.3331,  -2.6465,  -8.2028],
        [  1.7500,   7.4187,   4.4237],
        [  8.5716,  10.3596,   6.8818],
        [ 12.1588,   4.3364,  -8.0867],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.2659,  13.7894,   6.5436],
        [  1.6232,  -4.7717,  -2.5655],
        [ -6.2668,   5.9808,   2.5614],
        [ -5.5525,   6.5146,  -4.4984],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.6801,   2.7809,  -0.8443],
        [  3.9895,   6.0816,  -6.5235],
        [  2.3261,   7.1726,   6.2522],
        [  2.4932,   7.0903,  -5.4214],
        [  6.3569,   0.7591,   6.5011],
        [  8.1528,  10.7452,   9.1493],
        [ 12.1379,   4.4125,  -3.8833],
        [  7.8425,  -8.7202,   6.4530],
        [  4.8051, -10.3703,   0.5268],
        [ 11.9808,   4.9340,  -3.3715],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -8.0611,  -6.0302,   4.6962],
        [  5.0506,   4.9277,  -9.5094],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.0243,   7.3086,   7.5432],
        [  2.4356,  -4.5942,  -5.0808],
        [  6.3303,   1.6145,  -8.4083],
        [ -8.1204,  -5.9308,   4.0761],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.4149,  -4.6005,  -4.3339],
        [  4.8432,   5.1968,   1.5006],
        [ -8.6603,   2.8250,   5.9181],
        [ 12.3357,   3.6001,   0.7903],
        [ -0.5739,   7.9056,   0.6382],
        [  2.6970,   6.9826,   9.7350],
        [  5.4242, -10.1261,   3.9595],
        [  5.0506,   4.9277,  -9.5094],
        [  3.0876,   6.7527,  -5.3540],
        [  4.8432,   5.1968,   1.5006],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3305,   1.6119,   1.1538],
        [  5.4426, -10.1181,  -5.1268],
        [  7.8375,  -8.7240,  -9.8913],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.8979,   3.3958,  -3.8713],
        [  3.5965,  -4.0754,  -3.5035],
        [ -8.1426,   3.8048,   9.3542],
        [  3.6469,   6.3624,  -1.8414],
        [  5.3858,   4.4260,  -9.6769],
        [  6.8407,  11.7666,   3.0045],
        [  4.6231,   5.4556,   8.4973],
        [ -0.1963,   7.8765,  -1.1371],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.2022,   2.4092,   3.0349],
        [  6.7792,  11.8084,   1.6251],
        [ -9.2345,  -3.1345,   3.9715],
        [ -1.3178, -10.7939,   0.4236],
        [ -7.0948,   5.1940,   9.8171],
        [ -5.9225,  -8.5355,   2.4778],
        [ -7.4733,   4.7555,   6.2490],
        [  7.0642,  11.6104,  -9.8316],
        [  5.4429,  -2.2529,   9.7306],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.1624,  -3.4286,   2.0647],
        [  3.6817,   6.3354,   6.1233],
        [ 10.5057,  -5.9099,  -5.1599],
        [  0.8940, -11.0402,  -0.9783],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.1477,   7.2550,   8.9688],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.1292,   4.8183,  -5.7997],
        [  9.5261,  -7.1675,   1.4206],
        [  2.5128, -10.9365,   9.9645],
        [  6.7792,  11.8084,   1.6251],
        [  4.7739,  -3.1218,  -4.1188],
        [  2.1379,  -4.6757,   6.4981],
        [  2.1379,  -4.6757,   6.4981],
        [ 11.1469,   6.9237,   1.6160],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.2036,   2.4030,  -1.2500],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.9942,  -3.9990,   6.0448]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1694, 0.0816,  ..., 1.3408, 1.5284, 1.0385],
        [0.1694, 0.0000, 0.0958,  ..., 1.2612, 1.4545, 1.1254],
        [0.0816, 0.0958, 0.0000,  ..., 1.2793, 1.4693, 1.0532],
        ...,
        [1.3408, 1.2612, 1.2793,  ..., 0.0000, 0.1979, 1.1870],
        [1.5284, 1.4545, 1.4693,  ..., 0.1979, 0.0000, 1.2747],
        [1.0385, 1.1254, 1.0532,  ..., 1.1870, 1.2747, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.20100795, 11.4424816 , 11.37939695, 11.28215429, 11.54672755,
       11.35182589, 12.18045978,  5.90035324, 12.91505111,  6.48486171,
       12.52875793,  9.7751189 , 13.09426142, 10.60226614, 13.09737053,
        9.7712321 , 14.02058621, 11.1913598 ,  9.15976746, 12.53047489,
        5.91677188,  7.62509582,  9.70108409,  7.62232307, 13.44593519,
       12.9089559 ,  9.43737523, 13.97430191,  5.04028538,  8.66273171,
        8.55982105,  7.53063885,  9.11473391,  7.27340834,  7.5403822 ,
        7.5158489 ,  6.4020404 , 13.48809595, 12.91505111, 11.72802491,
       11.42947632, 12.95702647, 10.23946247, 10.06703358,  7.05627461,
        8.8353674 ,  7.58378655,  5.19986797,  6.53290663, 10.05558893,
        8.62329466,  5.19577146,  7.10378919,  9.10945678, 12.85032836,
        7.92644746,  7.48538905, 11.48734978,  7.05627461,  7.42512621,
        7.10378919,  7.60529892,  6.53250937, 11.4890861 , 11.72749923,
        9.58072286,  6.80559669,  5.43544626,  8.98765644,  7.33350642,
        6.97106912, 13.6105595 ,  7.15100303,  7.8788965 ,  8.20436764,
        6.65368551, 13.61600536,  9.75200848, 10.87409005,  8.79284466,
       10.38897563,  8.8580671 , 13.59056613,  5.89074104,  9.9353921 ,
        9.78285424,  7.32754882, 12.05395056, 11.07637531,  9.75401449,
        7.56617037, 12.12448912,  7.03733775, 11.92133129, 11.22142195,
       13.61600536,  5.70405525,  5.14124612,  5.14124612, 13.12218013,
       12.07818871,  5.19065439,  6.652742  , 13.0743399 , 13.103709  ,
        9.84314712])}
{'P': tensor([[4.8157e-03, 1.6578e-01, 1.6549e-01,  ..., 8.2069e-06, 1.6578e-01,
         3.3759e-07],
        [5.1180e-05, 9.3572e-03, 5.7237e-02,  ..., 2.7482e-05, 9.3572e-03,
         6.6704e-07],
        [3.1929e-05, 4.8174e-04, 9.8232e-03,  ..., 2.8406e-05, 4.8174e-04,
         6.5642e-07],
        ...,
        [6.4609e-06, 6.3038e-06, 5.4412e-06,  ..., 4.9473e-02, 6.3038e-06,
         3.1984e-03],
        [5.1180e-05, 9.3572e-03, 5.7237e-02,  ..., 2.7482e-05, 9.3572e-03,
         6.6704e-07],
        [1.4731e-04, 8.4216e-05, 6.0371e-05,  ..., 4.2837e-05, 8.4216e-05,
         6.6090e-02]]), 'X': tensor([[  2.4639, -10.9431,  -7.3718],
        [  4.2612, -10.5514,  -6.6268],
        [  4.9452, -10.3187,  -7.8621],
        [  3.1892, -10.8220,  -6.8409],
        [  3.9591, -10.6391,  -6.2557],
        [  6.0467,  -9.8369,  -6.5816],
        [ 11.9936,  -2.8677,   6.0614],
        [ 10.6283,  -5.7253,   7.4878],
        [  7.0358,  -9.2784,  -9.8479],
        [ -4.7642,  -9.3860,  -5.1133],
        [  1.8766,   7.3694,   1.1887],
        [  0.1047, -11.0046,   0.5188],
        [  3.5284, -10.7484,   5.3347],
        [  4.5954,  13.0099,  -4.4439],
        [  4.0421,  -3.7765,  -8.3631],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.0766,  -6.0046,  -3.1873],
        [  2.2849, -10.9655,  -7.6542],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.3515,   0.6418,  -6.5823],
        [  1.0093,   7.6567,   2.3526],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.2242,  -0.3171,   7.4978],
        [ -8.9282,   2.1648,   8.4894],
        [ 12.0042,  -2.8350,   2.6864],
        [ -8.6229,   2.9068,   2.4429],
        [ -8.4887,  -5.2448,  -0.6585],
        [  4.7544,  -3.1425,  -1.5315],
        [  5.7963,  -1.6018,   1.8856],
        [  4.0421,  -3.7765,  -8.3631],
        [ -2.5683, -10.4433,  -2.9182],
        [  8.0381,  -8.5690,  -8.9523],
        [ 11.5083,  -4.1177,   0.2389],
        [  7.0642,  11.6104,  -9.8316],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 12.3759,  -1.3943,   6.4659],
        [  0.1497,   7.8334,  -5.4412],
        [  0.1252, -11.0062,   7.2564],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.3339,   7.5627,  -0.9034],
        [ -9.2974,   0.8371,   2.4852],
        [  6.0890,  -0.8382,   2.9166],
        [  6.2419,  -0.2313,   4.2440],
        [ -8.1531,  -5.8748,   2.6934],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.3691,   6.4760,   1.0787],
        [  2.3589,  13.7668,  -3.6501],
        [  9.0004,  -7.7195,   4.6945],
        [  6.1984,  -0.4327,   1.5438],
        [  1.0093,   7.6567,   2.3526],
        [  6.2883,   0.0325,   5.2692],
        [ -9.3419,   0.5972,  -1.2841],
        [  1.4489,   7.5256,   4.4529],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -8.8631,  -4.3728,  -1.1746],
        [  1.3339,   7.5627,  -0.9034],
        [ 11.5456,   6.0861,   8.1101],
        [ -8.5149,  -5.1905,   1.5239],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.1222,  -0.7261,   8.8204],
        [  9.2692,   9.6370,   7.0746],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.6675,  -4.8569,  -9.1769],
        [  6.1685,  12.1979,  -6.2473],
        [ 10.5678,   7.9261,  -0.6100],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.3535,   0.6792,  -7.5101],
        [ -4.0868,   7.2981,   8.1874],
        [ -8.1204,  -5.9308,   4.0761],
        [  3.6469,   6.3624,  -1.8414],
        [ -9.4430,  -0.1897,   4.8383],
        [ -0.8885,   7.9160,   0.0571],
        [  5.0348,  -2.8229,   3.5436],
        [ 10.0619,   8.6627,  -9.2884],
        [ -5.9872,   6.2037,  -9.4477],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.9655,   3.2172,  -7.5953],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -2.6069, -10.4298,  -6.1995],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.1984,  -0.4327,   1.5438],
        [  4.0362,  13.2397,  -7.0030],
        [  5.1292,   4.8183,  -5.7997],
        [  5.3118,   4.5452,   4.2114],
        [  2.7590,  13.6621,   9.4496],
        [  0.5175,  14.0910,   4.6540],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.2199,   4.6861,  -1.9304],
        [  6.1741,  -0.5329,   3.1913],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.5010,  -4.5736,  -0.8989],
        [ -1.1885, -10.8211,   0.0865],
        [  2.3367,   7.1676,  -9.5263],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.5175,  14.0910,   4.6540],
        [  3.6469,   6.3624,  -1.8414],
        [  4.2612, -10.5514,  -6.6268],
        [ -7.9298,   4.1367,  -3.9773]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0826, 0.1721,  ..., 0.5932, 0.0826, 0.2888],
        [0.0826, 0.0000, 0.0958,  ..., 0.5240, 0.0000, 0.3386],
        [0.1721, 0.0958, 0.0000,  ..., 0.4933, 0.0958, 0.4313],
        ...,
        [0.5932, 0.5240, 0.4933,  ..., 0.0000, 0.5240, 0.6591],
        [0.0826, 0.0000, 0.0958,  ..., 0.5240, 0.0000, 0.3386],
        [0.2888, 0.3386, 0.4313,  ..., 0.6591, 0.3386, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.21703654, 11.37939695, 11.4424816 , 11.28215429, 11.35182589,
       11.54672755, 12.33167382, 12.07225469, 11.64436684, 10.52587243,
        7.60463349, 11.00508681, 11.31276109, 13.79762415,  5.53175275,
       11.21703654, 10.06407839, 11.20100795,  9.62958973,  6.383897  ,
        7.72291946, 11.75618513,  6.2322769 ,  9.18690018, 12.3344536 ,
        9.09963894,  9.97822473,  5.69913817,  6.01356004,  5.53175275,
       10.75443106, 11.74901778, 12.22276278, 13.59056613,  9.69792513,
       12.45418236,  7.83486943, 11.00694688, 10.1979669 ,  7.67939546,
        9.3349858 ,  6.1463896 ,  6.24614905, 10.04916678, 10.46238054,
        5.1224079 , 13.08414689, 13.96746816, 11.85743425,  6.2134961 ,
        7.72291946,  6.28835101,  9.36093562,  7.66377554,  9.86089579,
        9.88310614,  7.67939546, 13.05148826,  9.97220799, 11.75618513,
        5.94541803,  6.16512856, 13.37121923,  9.23071698,  9.50654967,
        9.93553424, 13.66896739, 13.20988996,  5.21280156,  5.9510489 ,
        6.38968357,  8.36445758, 10.05558893,  7.33350642,  9.44486609,
        7.96575782,  5.77219077, 13.27717982,  8.62162372, 12.12448912,
       10.31886066,  6.77777606, 10.33712258, 10.7506423 , 10.00794468,
       12.3489324 ,  6.2134961 , 13.84126197,  7.03733775,  6.99097013,
       13.93790235, 14.10045797, 12.7125079 ,  7.01475376,  6.19707944,
       10.53368644,  5.21280156, 10.88618114,  7.53884104,  5.1224079 ,
        5.5366698 ,  4.85995002, 14.10045797,  7.33350642, 11.37939695,
        8.94395628])}
{'P': tensor([[8.1681e-03, 2.8765e-01, 1.0772e-01,  ..., 3.3505e-06, 2.3848e-06,
         4.3182e-06],
        [3.4294e-05, 1.2045e-02, 5.7078e-01,  ..., 1.3617e-05, 1.6367e-06,
         1.1249e-05],
        [3.8918e-05, 1.0505e-04, 3.4124e-02,  ..., 8.8441e-05, 2.7019e-06,
         4.1425e-05],
        ...,
        [2.2230e-07, 2.4515e-07, 2.9611e-07,  ..., 5.0129e-02, 9.5602e-07,
         4.6223e-06],
        [2.5351e-02, 7.7154e-03, 2.8197e-03,  ..., 9.1377e-07, 3.1717e-02,
         1.8600e-06],
        [4.9426e-08, 9.2894e-08, 1.6774e-07,  ..., 2.7600e-03, 3.0405e-08,
         1.1721e-02]]), 'X': tensor([[  2.5128, -10.9365,   9.9645],
        [  5.1519, -10.2387,   9.8191],
        [  7.1821,  -9.1845,   9.8548],
        [  6.8093,  -9.4178,   8.4244],
        [  4.3523, -10.5232,   8.2138],
        [  5.6414, -10.0303,   7.8525],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -3.1671,   7.6158,   6.0765],
        [  6.3542,   0.6934,  -3.3840],
        [  5.1793,   4.7461,   0.3899],
        [  9.1620,  -7.5573,  -9.9882],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.0750,  -6.0072,  -5.8267],
        [  2.5375,   7.0676,  -4.3490],
        [  8.5198,  10.4091,  -8.6486],
        [ -9.1231,  -3.5739,   8.4708],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.4336,  -2.2676,   5.6933],
        [  6.3534,   1.2990,  -7.2942],
        [ -1.5451,   7.8975,   7.6104],
        [ -7.4887,   4.7364,   8.9074],
        [ 12.4597,  -0.9136,  -4.8763],
        [  0.2571,  -4.7594,   1.5618],
        [ -7.6650,   4.5092,   0.2520],
        [ -5.7347,  -8.6908,   9.2878],
        [  5.6928,  12.4710,   3.6001],
        [  1.9241,   7.3503,   7.9004],
        [  3.9022,  13.2906,   5.5788],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.3739,   0.3982,  -4.2564],
        [  5.1519, -10.2387,   9.8191],
        [ 11.9641,  -2.9568,   7.7211],
        [  3.9895,   6.0816,  -6.5235],
        [  6.1973,  -0.4372,   6.5720],
        [ 12.5770,   1.8512,  -0.6578],
        [ -5.0943,  -9.1682,   0.5073],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3211,   1.7017,  -0.2164],
        [ 10.9875,   7.2207,  -8.8549],
        [  6.3515,   0.6418,  -6.5823],
        [ -1.1885, -10.8211,   0.0865],
        [  6.1929,   2.4501,  -6.9352],
        [  0.3038,   7.8092,   0.4164],
        [  9.5261,  -7.1675,   1.4206],
        [  9.7328,   9.0902,  -0.8917],
        [  3.2809,  13.5061,  -4.0199],
        [ 10.6283,  -5.7253,   7.4878],
        [ 12.5224,  -0.4497,   4.8448],
        [  2.5620,   7.0549,   8.4825],
        [  6.7565,  11.8237,  -4.9882],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.1292,   4.8183,  -5.7997],
        [  2.5829,  -4.5466,  -8.4015],
        [  0.9945,  -4.8144,  -0.1897],
        [  6.3124,   0.2057,  -0.4593],
        [  5.2199,   4.6861,  -1.9304],
        [  5.1886,  -2.6250,   6.4677],
        [  6.3031,   1.8477,  -7.1706],
        [  9.8082,  -6.8396,   4.9615],
        [  2.5974,   7.0363,  -7.7810],
        [ -0.1626, -10.9796,   6.4629],
        [  9.1409,  -7.5789,   1.5814],
        [ 11.5456,   6.0861,   8.1101],
        [ -8.6112,   2.9319,  -7.9943],
        [  5.7845,   3.6652,   0.9175],
        [  6.1685,  12.1979,  -6.2473],
        [  0.5250, -11.0306,   5.5358],
        [  0.9478,  -4.8144,   0.6371],
        [  6.3505,   0.6235,  -8.5849],
        [  1.2728, -11.0372,   9.8121],
        [  0.5250, -11.0306,   5.5358],
        [  2.0039,   7.3173,  -1.5885],
        [ -4.2271,   7.2387,  -1.8316],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.6933,   7.7336,   2.1179],
        [  5.5240,  -2.1198,   6.2055],
        [  6.3516,   0.6425,   0.3425],
        [ 12.0582,  -2.6640,   0.7132],
        [  0.1677,  -4.7447,  -6.1529],
        [  1.8320,   7.3871,   5.1452],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.5382,   7.0672,   4.5097],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.3119,   1.7797,   9.9186],
        [ -9.2707,   0.9661,   4.4863],
        [  2.5010,  -4.5736,  -0.8989],
        [  3.2990,  -4.2415,  -7.4857],
        [  6.3497,   1.3680,  -1.9076],
        [  2.3376,  -4.6231,   9.8436],
        [  3.2647, -10.8066,  -0.3768],
        [ -0.0282,   7.8575,  -1.1746],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.9606,  -4.0994,   1.7886],
        [  4.5325,  -3.3643,   6.7699],
        [ -2.2508, -10.5479,  -2.7235],
        [  6.3122,  -9.6994,  -6.3282],
        [ 10.6283,  -5.7253,   7.4878],
        [ -4.0868,   7.2981,   8.1874],
        [  9.6587,   9.1817,  -0.8212],
        [ -6.7745,  -7.7264,   8.4684],
        [ 12.5267,  -0.4125,  -8.7331]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0886, 0.1847,  ..., 1.0636, 0.4309, 1.4162],
        [0.0886, 0.0000, 0.0961,  ..., 1.0090, 0.5194, 1.3851],
        [0.1847, 0.0961, 0.0000,  ..., 0.9520, 0.6150, 1.3532],
        ...,
        [1.0636, 1.0090, 0.9520,  ..., 0.0000, 1.3616, 0.5175],
        [0.4309, 0.5194, 0.6150,  ..., 1.3616, 0.0000, 1.6033],
        [1.4162, 1.3851, 1.3532,  ..., 0.5175, 1.6033, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.22142195, 11.46176259, 11.65923396, 11.62158554, 11.38774149,
       11.50791859,  9.15976746,  9.97822473,  8.24808832,  6.39187285,
        7.02496223, 11.87665501, 12.21532282, 10.06437598,  7.5092834 ,
       13.45124917,  9.79814997, 12.49317563, 12.84056604, 12.59969936,
       12.79126785,  5.88782913,  6.48486171,  8.04718433,  8.86082124,
       12.49317563,  4.7663526 ,  8.89301516, 10.412299  , 13.70893492,
        7.59795445, 13.85158417, 12.26997165,  9.38232121, 11.46176259,
       12.32408516,  7.27340834,  6.21275016, 12.7125079 , 10.48840476,
        6.27905292,  6.54615887, 13.14777433,  6.383897  , 10.88618114,
        6.65991877,  7.81509631, 11.92133129, 13.31764181, 13.89886469,
       12.07225469, 12.53047489,  7.50564223, 13.61801526, 10.78533838,
        9.68323686,  7.03733775,  5.22903117,  4.9160962 ,  6.31576264,
        7.01475376,  5.81481044,  6.56833864, 11.95743558,  7.50035355,
       10.98076873, 11.87412439, 13.05148826,  9.09661212,  6.84796202,
       13.66896739, 11.04313641,  4.90676242,  6.38105015, 11.11038551,
       11.04313641,  7.58668144,  8.38252937, 10.36844813,  7.76457095,
        5.91677188,  6.38399789, 12.3489324 ,  4.74770899,  7.61088125,
        9.78147542,  7.50917551,  8.29203356,  6.55800267,  9.32094055,
        5.21280156,  5.37343173,  6.49538841,  5.1805345 , 11.28895489,
        7.8575648 ,  5.19065439,  9.85384623,  5.64464807, 10.78533838,
       11.57249984, 12.07225469,  8.36445758, 13.32645552, 10.27572673,
       12.53345445])}
{'P': tensor([[1.0156e-02, 3.6508e-01, 1.5663e-01,  ..., 1.0296e-06, 2.4771e-04,
         3.1178e-06],
        [4.0674e-05, 1.3988e-02, 6.9633e-04,  ..., 1.8004e-06, 8.5918e-04,
         5.1047e-06],
        [7.4198e-05, 3.5452e-02, 8.1325e-03,  ..., 1.1387e-06, 6.1555e-04,
         5.3878e-06],
        ...,
        [3.8580e-05, 1.9940e-05, 2.9237e-05,  ..., 1.0870e-01, 5.8174e-06,
         1.9195e-05],
        [1.7044e-06, 3.0339e-06, 2.8411e-06,  ..., 2.1621e-04, 6.5444e-02,
         3.8267e-04],
        [3.1927e-06, 4.0566e-06, 5.9963e-06,  ..., 2.3667e-04, 6.4385e-04,
         1.4652e-01]]), 'X': tensor([[  2.5636, -10.9293,   9.6519],
        [  5.1519, -10.2387,   9.8191],
        [  4.3523, -10.5232,   8.2138],
        [  7.1821,  -9.1845,   9.8548],
        [  5.6414, -10.0303,   7.8525],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4430,  -0.1897,   4.8383],
        [ 11.5966,   5.9666,   6.0083],
        [ -1.5408,   7.8978,   5.3470],
        [ 11.6634,   5.8050,   9.1590],
        [ -7.4887,   4.7364,   8.9074],
        [  5.7802,  -1.6363,  -6.2782],
        [  5.4867,  12.5816,   2.9549],
        [ -9.3464,   0.5708,  -1.6397],
        [  2.4697,  -4.5836,   1.4593],
        [  6.3583,   1.1729,   8.7835],
        [  5.6928,  12.4710,   3.6001],
        [  3.4920,  -4.1366,  -8.3955],
        [  1.9312,  13.8649,   8.3563],
        [ -2.3317,   7.8034,  -3.6274],
        [  5.1667,  -2.6543,   1.2879],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 10.9394,   7.3069,  -8.9365],
        [  0.9809,   7.6642,  -1.4217],
        [  6.0073,   3.0986,   6.7937],
        [ -9.4773,  -1.0024,   5.2599],
        [  4.5954,  13.0099,  -4.4439],
        [  5.9587,  12.3216,   9.1695],
        [ -1.6275, -10.7221,   3.2376],
        [  5.2939,  -2.4784,  -4.5840],
        [  9.6662,   9.1725,   8.1056],
        [  2.7976,  -4.4682,  -4.2020],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -2.9497, -10.3028,  -7.3479],
        [  2.3373,  13.7721,  -0.2442],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.3266,   0.3292,   9.3502],
        [  8.7715,  -7.9390,   5.2900],
        [ -4.9217,   6.8979,  -1.7667],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -8.1426,   3.8048,   9.3542],
        [  5.8836,   3.4315,  -4.0327],
        [ -3.1204, -10.2345,   9.2288],
        [ -5.3328,  -8.9992,   7.3440],
        [ -3.0008, -10.2827,  -8.9480],
        [ -4.1296,   7.2803,  -2.7194],
        [ 12.5154,   2.5072,  -9.3274],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.2535,  -0.1709,  -9.4102],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -1.8823, -10.6558,  -3.8952],
        [ -2.9497, -10.3028,  -7.3479],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.4572,   7.1085,   7.7376],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -9.1178,   1.5784,   2.1410],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.6000,   1.3839,  -3.2992],
        [  4.0587, -10.6112,   6.2273],
        [  8.7131,  10.2216,   7.7915],
        [ 12.6035,   0.7494,   7.1536],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.0617,  12.7957,   8.3544],
        [  5.3924, -10.1396,  -4.0863],
        [  6.2022,   2.4092,   3.0349],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.9193,  -1.3176,   5.1739],
        [ -6.0844,  -8.3955,   0.7705],
        [  6.1372,   2.6726,  -6.7885],
        [ 11.0438,   7.1180,  -5.6991],
        [  4.2275,  -3.6320,  -0.2714],
        [  3.6249, -10.7255,   5.6605],
        [ -9.4196,   0.0478,   9.4140],
        [  2.3376,  -4.6231,   9.8436],
        [  7.9816,  10.8938,  -0.5378],
        [  5.4336,  -2.2676,   5.6933],
        [  3.5939,   6.4029,  -1.8357],
        [  6.2905,   1.9366,   2.1169],
        [  6.3589,   0.8239,  -8.4230],
        [  1.8766,   7.3694,   1.1887],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.1905,   3.7253,   4.1175],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.4721,  -1.3097,  -7.4377],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.1737,   2.5305,   3.9049],
        [  1.1139,   7.6281,  -2.5680],
        [ -3.9700,   7.3453,  -4.1486],
        [  4.1999,  -3.6543,  -4.1998],
        [ -1.6906,   7.8860,   7.7142],
        [  2.5469,  13.7192,  -9.6935],
        [  5.8889,  -1.3916,   8.8993],
        [  8.5716,  10.3596,   6.8818],
        [  3.1756, -10.8247,   6.8882],
        [  3.2701,   6.6338,   9.9209],
        [  9.6614,   9.1784,  -1.2087],
        [  1.7000,  -4.7609,   2.7935],
        [  2.5636, -10.9293,   9.6519],
        [ -4.4503,   7.1378,   3.3751],
        [ 12.5789,   0.1722,   1.6470],
        [  6.3360,   0.4258,  -9.2509]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0980, 0.0265,  ..., 0.4010, 0.8094, 1.2239],
        [0.0980, 0.0000, 0.0877,  ..., 0.4085, 0.7519, 1.1979],
        [0.0265, 0.0877, 0.0000,  ..., 0.3777, 0.7832, 1.1980],
        ...,
        [0.4010, 0.4085, 0.3777,  ..., 0.0000, 0.5409, 0.8431],
        [0.8094, 0.7519, 0.7832,  ..., 0.5409, 0.0000, 0.5534],
        [1.2239, 1.1979, 1.1980,  ..., 0.8431, 0.5534, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.22596994, 11.46176259, 11.38774149, 11.65923396, 11.50791859,
       11.62158554,  8.95983586,  9.44486609, 13.04156075,  8.04664886,
       13.02818702,  8.86082124,  6.00732193, 13.72594022,  9.36378438,
        5.20659892,  6.46560181, 13.70893492,  5.41348121, 13.99877413,
        8.14433542,  5.80860165,  9.43112957, 12.10942412, 13.15525803,
        7.72668467,  6.7593917 ,  9.53015376, 13.79762415, 13.68672855,
       10.84493732,  5.84533345, 13.32557133,  5.27178363, 10.0674887 ,
       10.7167285 , 13.96905756,  9.50654967,  6.33517349, 11.83075005,
        8.47371868, 10.23946247,  8.98765644,  6.81117802, 10.69963954,
       10.4606265 , 10.71162681,  8.36995953, 12.76408562, 10.06286074,
        9.64157461,  6.25586297, 10.54658597, 10.61047596, 10.8207297 ,
       10.7167285 , 12.19805394,  7.52117132,  9.94285511,  9.25336371,
       12.67680439, 12.67576792, 11.36089355, 13.43127131, 12.62576026,
       11.73848065, 13.76047999, 11.48434767,  6.65368551, 10.31886066,
        6.06416957, 10.36844813,  6.69389431, 13.13888841,  5.57341623,
       11.32149259,  9.41970031,  5.1805345 , 13.50484539,  5.88782913,
        7.34250948,  6.58184474,  6.41202986,  7.60463349,  7.85919687,
        8.99791316, 12.50996828,  6.34059021,  9.562174  , 13.26590071,
        6.67218687,  7.70897675,  8.34949383,  5.5671389 ,  8.06517143,
       13.95361098,  6.05113217, 13.44593519, 11.28093507,  7.39599986,
       13.32613445,  5.05534609, 11.22596994,  8.41148722, 12.58006038,
        6.35029274])}
{'P': tensor([[3.0325e-03, 1.8197e-01, 1.2480e-01,  ..., 6.7883e-06, 1.7521e-05,
         8.4735e-03],
        [1.5651e-05, 3.2765e-03, 1.8802e-01,  ..., 3.5253e-06, 2.1418e-05,
         1.4623e-02],
        [1.8871e-05, 4.5773e-05, 5.0992e-03,  ..., 3.9594e-06, 3.9227e-05,
         2.2747e-02],
        ...,
        [5.4092e-02, 3.8723e-02, 3.4070e-02,  ..., 9.4787e-03, 8.1487e-05,
         2.0000e-03],
        [2.7724e-05, 3.5922e-05, 4.4305e-05,  ..., 2.7441e-05, 1.0723e-01,
         4.2007e-05],
        [2.5400e-05, 4.6488e-05, 6.8293e-05,  ..., 5.8787e-06, 8.0878e-05,
         4.6753e-02]]), 'X': tensor([[  2.6078, -10.9229,   6.2869],
        [  4.0587, -10.6112,   6.2273],
        [  4.7482, -10.3907,   5.9149],
        [  3.6249, -10.7255,   5.6605],
        [  5.4167, -10.1293,   6.2156],
        [  5.7218,  -9.9934,   6.5764],
        [  0.1677,  -4.7447,  -6.1529],
        [ -7.5720,   4.6310,   2.1717],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.5318,  13.7231,   6.3692],
        [ 11.9641,  -2.9568,   7.7211],
        [  0.9788,  -4.8145,  -3.8661],
        [ -2.6485, -10.4151,   0.8766],
        [  6.2038,   2.4020,  -3.1692],
        [  3.3167,   6.6022,   6.3554],
        [ -9.3444,   0.5821,   6.9101],
        [  6.0140,  -1.0661,  -2.5040],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.7218,  -9.9934,   6.5764],
        [ 12.4901,  -0.7052,   1.7315],
        [ 11.9808,   4.9340,  -3.3715],
        [  6.1989,  -9.7591,   0.8289],
        [ -2.0445,   7.8469,  -0.5857],
        [  5.2188,  -2.5839,   2.1833],
        [  6.3122,  -9.6994,  -6.3282],
        [ -7.7239,   4.4298,   3.1216],
        [ -3.2736, -10.1703,   3.8712],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.8930,  12.3592,  -8.9201],
        [ 11.5456,   6.0861,   8.1101],
        [  0.5600, -11.0321,  -0.8660],
        [  2.5375,   7.0676,  -4.3490],
        [  5.5158,  -2.1337,   3.5246],
        [  2.4572,   7.1085,   7.7376],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.0449,  13.8402,  -7.1642],
        [  6.3549,   0.7096,   6.9198],
        [  2.2659,  13.7894,   6.5436],
        [ -9.0604,   1.7707,   1.1066],
        [  6.3534,   1.2990,  -7.2942],
        [ 10.1503,   8.5414,  -7.2681],
        [ 11.9695,  -2.9407,  -6.3498],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.5149,  -5.1905,   1.5239],
        [  9.1409,  -7.5789,   1.5814],
        [ -9.4283,  -1.9585,  -0.9129],
        [  2.7558,  -4.4844,   0.0636],
        [  0.7275,  14.0667,  -0.9815],
        [  9.3560,   9.5391,   9.4943],
        [  5.2911,  -2.4824,  -3.6877],
        [ -0.0254, -10.9932,   3.2200],
        [  2.0419,  -4.6978,  -5.0983],
        [ 12.1940,  -2.1881,   7.1499],
        [  4.5057,  -3.3894,  -0.8160],
        [ -3.1204, -10.2345,   9.2288],
        [ -0.9715, -10.8630,   0.3555],
        [  6.1755,   2.5233,  -2.0042],
        [ -8.4573,   3.2448,   5.4370],
        [  6.3542,   0.6934,  -3.3840],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.2902,   0.0451,   3.7116],
        [  2.7976,  -4.4682,  -4.2020],
        [ -5.9872,   6.2037,  -9.4477],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.7802,  -1.6363,  -6.2782],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.2767,  -4.6006,   0.7868],
        [ 11.5568,   6.0602,   9.3244],
        [  2.1625,  -4.6697,   7.0658],
        [  4.4915,   5.5990,   7.0405],
        [ -7.9667,   4.0814,  -0.6397],
        [ -9.3739,   0.3982,  -4.2564],
        [ -6.8618,  -7.6322,   3.7027],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.0438,   7.1180,  -5.6991],
        [  2.4932,   7.0903,  -5.4214],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.1252, -11.0062,   7.2564],
        [  3.4401,  13.4540,   4.6576],
        [  4.1637,  13.1898,  -1.9189],
        [ -7.2515,   5.0195,   4.6434],
        [  5.7218,  -9.9934,   6.5764],
        [ -0.0410,   7.8591,   9.3457],
        [  0.5960, -11.0335,   3.1144],
        [  4.0587, -10.6112,   6.2273],
        [ -9.3444,   0.5821,   6.9101],
        [  0.6142,  14.0802,   0.5407],
        [ -3.1204, -10.2345,   9.2288],
        [  2.6970,   6.9826,   9.7350],
        [ -6.4368,   5.8356,   3.3014],
        [ -6.1914,  -8.2997,   8.5380],
        [  2.6970,   6.9826,   9.7350],
        [ -8.0087,   4.0173,   9.1091],
        [ 12.5996,   1.3971,  -1.0694],
        [ -6.0844,  -8.3955,   0.7705],
        [ -9.1749,  -3.3803,   6.4629],
        [ -4.7334,   6.9982,  -0.4392],
        [  2.0547,   7.2957,  -1.9343],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.4149,  -4.6005,  -4.3339],
        [ 12.5224,  -0.4497,   4.8448],
        [  2.1120,  -4.6818,   8.6144],
        [ -1.8788, -10.6567,   3.7068],
        [  5.7802,  -1.6363,  -6.2782],
        [  9.0129,  -7.7072,   9.1400]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0728, 0.1100,  ..., 0.2903, 1.1015, 0.4164],
        [0.0728, 0.0000, 0.0394,  ..., 0.3401, 1.0651, 0.3468],
        [0.1100, 0.0394, 0.0000,  ..., 0.3582, 1.0345, 0.3186],
        ...,
        [0.2903, 0.3401, 0.3582,  ..., 0.0000, 1.0375, 0.6725],
        [1.1015, 1.0651, 1.0345,  ..., 1.0375, 0.0000, 1.0636],
        [0.4164, 0.3468, 0.3186,  ..., 0.6725, 1.0636, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.22993218, 11.36089355, 11.42420508, 11.32149259, 11.48663767,
       11.51557166,  4.74770899,  8.87587457, 12.05395056, 13.95472331,
       12.32408516,  4.91296205, 10.74655346,  6.65259775,  7.38844703,
        9.36256096,  6.10774544,  4.78258092, 11.51557166, 12.50996828,
       12.95702647, 11.56146873,  8.10886021,  5.82345867, 11.57249984,
        8.90405823, 10.68416401,  9.70108409, 13.6922425 , 13.05148826,
       11.04629471,  7.5092834 ,  5.91408504,  7.52117132, 10.31584102,
       13.99048094,  6.39438044, 13.97430191,  9.23178204,  6.48486171,
       13.26590071, 12.32545637, 13.89974693,  9.97220799, 11.87412439,
        9.62958973,  5.26342203, 14.08549802, 13.3614598 ,  5.84450127,
       10.99326577,  5.1224079 , 12.38881635,  5.63824971, 10.69963954,
       10.90638066,  6.67108662,  9.05843201,  6.39187285,  6.10774544,
        6.29034958,  5.27178363,  8.62162372,  5.39450305,  6.00732193,
        9.85330646, 12.17901033, 13.0493368 ,  5.14607014,  7.17789666,
        8.95133961,  9.38232121, 10.26327895, 11.70168755, 13.13888841,
        7.5158489 , 10.24557047, 11.00694688, 13.88683436, 13.83139024,
        8.81930525, 11.51557166,  7.85919687, 11.04953767, 11.36089355,
        9.36256096, 14.09357547, 10.69963954,  7.48538905,  8.68832398,
       10.35464916,  7.48538905,  8.95983586, 12.67680439, 10.36844813,
        9.77777346,  8.4486771 ,  7.57946459,  9.99211663,  9.91668182,
        5.19577146, 12.53047489,  5.13616224, 10.82106505,  6.00732193,
       11.85890664])}
{'P': tensor([[8.5141e-03, 2.7953e-01, 2.0119e-01,  ..., 1.6665e-06, 1.4785e-03,
         1.7567e-04],
        [5.8787e-05, 2.2288e-02, 1.3703e-01,  ..., 6.4718e-06, 6.1207e-03,
         8.4117e-04],
        [5.9889e-05, 1.7978e-03, 2.5557e-02,  ..., 7.6010e-06, 8.2382e-03,
         1.5787e-03],
        ...,
        [2.2492e-05, 1.2055e-05, 1.0544e-05,  ..., 6.1450e-02, 6.2103e-05,
         9.4695e-07],
        [1.1415e-04, 2.0778e-04, 2.3770e-04,  ..., 1.1396e-03, 9.5291e-02,
         9.8478e-04],
        [2.7500e-06, 6.3218e-06, 8.4737e-06,  ..., 2.5028e-05, 5.8015e-04,
         7.2303e-02]]), 'X': tensor([[ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1619, 0.1944,  ..., 0.2374, 0.6013, 1.0653],
        [0.1619, 0.0000, 0.1037,  ..., 0.1082, 0.4701, 1.0102],
        [0.1944, 0.1037, 0.0000,  ..., 0.2031, 0.4069, 0.9100],
        ...,
        [0.2374, 0.1082, 0.2031,  ..., 0.0000, 0.4997, 1.0864],
        [0.6013, 0.4701, 0.4069,  ..., 0.4997, 0.0000, 0.6673],
        [1.0653, 1.0102, 0.9100,  ..., 1.0864, 0.6673, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.23201619, 11.48734978, 11.52733581, 11.4088043 , 11.42420508,
       11.48663767,  5.81199422, 13.13205357, 13.3607379 ,  7.81509631,
       11.42947632, 10.90638066,  9.58072286,  7.66377554, 11.63383221,
        5.19065439,  8.57560275,  6.61386436, 11.1341798 , 10.99326577,
        6.28961988, 11.04629471, 13.72594022, 13.61600536,  8.68979052,
        8.04483495,  9.97964013, 13.89671314, 11.57249984, 10.38897563,
        7.8575648 ,  7.37576154, 10.42667093,  6.38105015,  6.42689705,
        8.79284466, 11.65923396, 12.4217523 , 14.08549802, 12.97370371,
       12.13568598,  6.55800267,  9.75075273,  4.76969364, 11.87665501,
        8.99276564,  9.53015376, 13.45124917,  9.54154096,  5.0209005 ,
        8.41148722, 13.01187484, 11.83075005,  8.8737556 , 13.18278606,
       11.73567759,  9.36256096, 12.17451685,  6.00663314,  8.10886021,
        9.43112957,  7.17789666, 10.41909211,  7.96575782, 10.84437705,
       11.64746882,  8.5724042 , 13.14690137,  5.91677188, 10.48315436,
        8.50268322, 13.1471313 , 12.49317563,  7.01475376,  8.0550286 ,
        7.39004226, 12.41366754,  6.81084724,  5.81199422, 12.88270018,
       11.83570132,  9.69629   , 10.27716434, 11.28895489, 11.99633819,
       11.36089355, 12.53047489, 13.59056613,  6.1561646 , 10.51340829,
       10.91805683,  9.70108409, 10.51437947,  5.53255882,  6.02779201,
       13.96905756,  5.10444097,  7.63726134, 10.7167285 ,  5.9510489 ,
       13.66991027,  8.5724042 , 12.87752052,  8.24808832,  6.02588517,
       12.35421461])}
{'P': tensor([[3.4769e-03, 3.1402e-01, 1.9085e-01,  ..., 1.9819e-05, 3.6086e-06,
         1.5378e-03],
        [3.7017e-05, 4.8032e-03, 3.1470e-01,  ..., 3.0771e-05, 5.5029e-06,
         2.2688e-03],
        [4.4783e-05, 5.3837e-05, 9.0982e-03,  ..., 8.0000e-05, 1.3726e-05,
         4.9669e-03],
        ...,
        [7.2512e-06, 7.2452e-06, 7.3687e-06,  ..., 4.2279e-02, 1.3343e-01,
         1.0054e-05],
        [8.6964e-06, 8.3864e-06, 7.7982e-06,  ..., 9.0681e-04, 6.5386e-02,
         7.9876e-06],
        [1.7919e-05, 2.1969e-05, 3.9213e-05,  ..., 1.0220e-03, 2.1207e-04,
         2.6338e-02]]), 'X': tensor([[ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0350, 0.0961,  ..., 0.5877, 0.5733, 0.5822],
        [0.0350, 0.0000, 0.0614,  ..., 0.5587, 0.5493, 0.5655],
        [0.0961, 0.0614, 0.0000,  ..., 0.5037, 0.5024, 0.5308],
        ...,
        [0.5877, 0.5587, 0.5037,  ..., 0.0000, 0.1270, 0.2729],
        [0.5733, 0.5493, 0.5024,  ..., 0.1270, 0.0000, 0.1460],
        [0.5822, 0.5655, 0.5308,  ..., 0.2729, 0.1460, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.24434374, 11.28215429, 11.37939695, 11.35182589, 11.4424816 ,
       11.54672755, 10.74655346,  7.02253235, 10.98076873, 12.17451685,
        8.98765644, 11.66453059,  5.39723241, 12.42554808, 13.12299145,
       12.91505111,  6.97106912,  9.97220799, 12.79126785,  7.19513806,
        9.23178204,  7.32950385, 10.90638066, 13.66896739,  7.72415048,
       11.75807252, 10.66631784,  9.61942445, 13.60275211, 11.62770332,
        7.96575782, 12.41366754, 13.95361098,  7.52117132, 12.18998195,
        7.15990676,  9.87720102,  6.30647746, 10.38897563,  9.62958973,
       11.018084  , 11.8159254 ,  8.09704052, 11.4424816 , 10.57302546,
       13.43127131,  5.2287058 ,  5.91677188, 11.04313641, 13.27994304,
       13.14690137, 13.34587027,  9.23071698,  6.06416957, 10.01804223,
        5.47866221, 12.45418236,  6.32572323,  5.43544626,  6.67121748,
        7.92200342,  7.23276582, 13.58333857,  9.71392179,  8.30125201,
        9.17595899,  5.64865173,  8.68832398,  5.79543038,  9.97220799,
        8.34949383,  8.30125201, 13.95472331,  7.26074925,  8.08320235,
        5.24461068,  8.98765644,  5.90035324, 13.20597932, 10.48840476,
        6.76791483,  6.39187285, 12.7125079 ,  8.95983586,  6.48486171,
       13.58333857,  8.01009083,  6.60752373,  8.09899874,  8.86082124,
        5.20659892,  9.17595899,  4.74770899, 11.75618513,  7.20482224,
        4.73784222,  7.98469166, 12.53345445,  9.78285424,  8.92395381,
        5.64464807, 10.31613269,  4.8763135 ,  7.06272256,  7.53063885,
       11.83570132])}
{'P': tensor([[5.9809e-03, 2.4616e-01, 2.1997e-01,  ..., 1.3103e-04, 1.6217e-05,
         1.6803e-06],
        [8.0643e-05, 2.5570e-02, 1.3309e-04,  ..., 6.9361e-04, 2.2462e-05,
         3.1653e-06],
        [9.1511e-05, 3.1258e-01, 7.1798e-03,  ..., 1.3951e-04, 1.2164e-05,
         1.4452e-06],
        ...,
        [3.2264e-05, 5.2301e-05, 3.3478e-05,  ..., 5.7611e-02, 1.2575e-05,
         2.9636e-05],
        [1.6712e-01, 7.8691e-02, 1.7314e-01,  ..., 1.0771e-04, 1.6376e-02,
         1.7555e-05],
        [4.0990e-03, 2.3715e-03, 2.8342e-03,  ..., 6.6031e-04, 2.0416e-03,
         3.2395e-02]]), 'X': tensor([[  3.0719, -10.8449,  -4.1673],
        [  5.3924, -10.1396,  -4.0863],
        [  3.8111, -10.6787,  -4.6539],
        [  5.4426, -10.1181,  -5.1268],
        [  5.7485,  -9.9811,  -3.2355],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.5424,  -2.0885,   7.2487],
        [  4.9452, -10.3187,  -7.8621],
        [ -7.4808,   4.7462,   2.0266],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -9.4697,  -0.6157,  -6.8472],
        [  4.4915,   5.5990,   7.0405],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.8450,  -9.9356,   3.0767],
        [  6.1222,  -0.7261,   8.8204],
        [ -2.4956, -10.4682,   3.5632],
        [  7.0642,  11.6104,  -9.8316],
        [  5.1292,   4.8183,  -5.7997],
        [ -8.0087,   4.0173,   9.1091],
        [  6.1984,  -0.4327,   1.5438],
        [  4.7482, -10.3907,   5.9149],
        [ -7.8359,   4.2735,   7.2943],
        [  7.0664,  -9.2590,   8.0879],
        [  2.8890,  -4.4316,   1.1600],
        [  2.5318,  13.7231,   6.3692],
        [  5.9193,  -1.3176,   5.1739],
        [  2.1477,   7.2550,   8.9688],
        [  0.3038,   7.8092,   0.4164],
        [  6.9314,  -9.3435,  -1.2729],
        [ -0.0254, -10.9932,   3.2200],
        [ -5.1046,  -9.1610,   2.3418],
        [ 10.8218,  -5.4177,   6.8573],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.1469,   6.9237,   1.6160],
        [  6.3031,   1.8477,  -7.1706],
        [  7.3724,  -9.0577,  -8.9843],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -8.5900,   2.9767,   7.4811],
        [  3.3069,   6.6089,   8.7377],
        [  4.0642,  -3.7599,  -3.4565],
        [  9.6614,   9.1784,  -1.2087],
        [  4.9531,   5.0576,   5.6794],
        [  6.3301,   0.3638,  -7.5685],
        [  5.1142,  -2.7230,  -4.1966],
        [  6.1752,   2.5242,   3.5874],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.8704,  -9.3809,   7.5769],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -2.0840,   7.8415,  -7.8657],
        [ -4.4503,   7.1378,   3.3751],
        [ 12.5919,   0.4035,  -5.8460],
        [ -8.8400,   2.3991,   1.3457],
        [ 12.4877,  -0.7222,   8.8347],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.0190,   4.9705,  -7.9752],
        [  0.2732,  -4.7619,  -1.1577],
        [  9.3560,   9.5391,   9.4943],
        [  1.2273,  -4.8081,   9.8129],
        [  2.0937,  13.8293,  -2.3533],
        [ -7.8359,   4.2735,   7.2943],
        [ -4.6393,   7.0461,   1.1624],
        [  7.0358,  -9.2784,  -9.8479],
        [ -9.1624,  -3.4286,   2.0647],
        [  8.1528,  10.7452,   9.1493],
        [ 12.4691,   2.8525,  -5.5110],
        [ -5.3328,  -8.9992,   7.3440],
        [ -8.1204,  -5.9308,   4.0761],
        [ -3.5173,   7.5091,  -8.9585],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.0305,   3.0296,   0.1375],
        [ -7.4733,   4.7555,   6.2490],
        [ -6.2200,  -8.2737,   5.5079],
        [ 10.0995,  -6.4739,   2.0974],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.3858,   4.4260,  -9.6769],
        [  0.7275,  14.0667,  -0.9815],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -6.6895,   5.6051,   2.7899],
        [  1.2728, -11.0372,   9.8121],
        [ -5.8389,  -8.6056,  -3.5098],
        [  0.5600, -11.0321,  -0.8660],
        [  2.4149,  -4.6005,  -4.3339],
        [  4.6231,   5.4556,   8.4973],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.6093,  -1.9705,   6.4807],
        [  5.6523,   3.9443,  -2.9449],
        [  6.1929,   2.4501,  -6.9352],
        [ -7.5401,   4.6718,   9.0033],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.0190,   4.9705,  -7.9752],
        [ -3.3999, -10.1152,  -1.2088],
        [  5.6202,  -1.9508,   0.7471],
        [  2.3373,  13.7721,  -0.2442],
        [ 11.4743,   6.2478,  -2.1100],
        [  4.2612, -10.5514,  -6.6268],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.4309,  -0.0599,  -4.0830],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.6229,   2.9068,   2.4429],
        [  4.3459,  -3.5327,   0.3667],
        [ -3.1671,   7.6158,   6.0765],
        [  5.8323,  -1.5228,   5.1132],
        [ -0.7599, -10.8995,  -6.0366],
        [ -5.6792,  -8.7352,   4.0318]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1326, 0.0473,  ..., 0.8935, 0.2142, 0.7703],
        [0.1326, 0.0000, 0.1039,  ..., 0.8002, 0.3385, 0.8244],
        [0.0473, 0.1039, 0.0000,  ..., 0.8916, 0.2348, 0.8130],
        ...,
        [0.8935, 0.8002, 0.8916,  ..., 0.0000, 1.1006, 0.9295],
        [0.2142, 0.3385, 0.2348,  ..., 1.1006, 0.0000, 0.8211],
        [0.7703, 0.8244, 0.8130,  ..., 0.9295, 0.8211, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.27160035, 11.48434767, 11.33837396, 11.4890861 , 11.51811603,
       11.59512194,  5.92281142, 11.4424816 ,  8.859416  , 12.00995898,
       12.52372629,  9.48969978,  7.17789666,  5.48276038, 11.52733581,
        6.16512856, 10.76154256, 13.59056613,  7.03733775,  8.95983586,
        6.2134961 , 11.42420508,  8.92549752, 11.64746882,  5.29009836,
       13.95472331,  6.06416957,  7.56617037,  7.81509631, 11.63383221,
       10.99326577, 10.48721382, 12.10222204,  5.95297459,  5.1224079 ,
       13.12218013,  6.56833864, 11.67875855, 10.09979747,  9.09119351,
        7.39004226,  5.5366698 , 13.32613445,  7.07902036,  6.34059021,
        5.79391825,  6.67121748,  9.5014146 , 11.62770332,  9.75075273,
        8.1137452 ,  8.41148722, 12.59840193,  9.15976746, 12.5085995 ,
       12.97370371,  7.06372172,  4.76969364, 13.3614598 ,  4.96231532,
       13.98691426,  8.92549752,  8.43626114, 11.64436684,  9.78285424,
       13.48809595, 12.79126785, 10.4606265 , 10.05558893,  8.29203356,
        5.48276038,  6.74872981,  8.8580671 , 10.35093485, 11.99633819,
       12.7125079 ,  6.97106912, 14.08549802, 12.30858677,  8.7273569 ,
       11.11038551, 10.39942497, 11.04629471,  5.19577146,  7.15100303,
        6.23262289,  5.94535529,  6.89245352,  6.65991877,  8.87007495,
       12.5085995 ,  7.06372172, 10.67132524,  5.94909709, 13.96905756,
       13.06498554, 11.37939695, 13.02818702,  9.43112957,  4.7663526 ,
        9.09963894,  5.60064319,  8.24808832,  6.02779201, 10.92597196,
       10.41909211])}
{'P': tensor([[5.4673e-03, 3.2962e-01, 1.7557e-01,  ..., 3.2044e-04, 1.4889e-03,
         1.2488e-07],
        [3.2600e-05, 6.9880e-03, 3.3166e-01,  ..., 6.7954e-04, 1.7969e-03,
         1.4909e-07],
        [6.0899e-05, 1.3759e-04, 2.2344e-02,  ..., 2.4800e-03, 4.2506e-03,
         4.1422e-07],
        ...,
        [5.5433e-07, 8.5247e-07, 1.2500e-06,  ..., 1.0194e-02, 3.7241e-05,
         1.2281e-06],
        [4.9736e-04, 7.9489e-04, 1.2122e-03,  ..., 5.4135e-03, 9.1582e-02,
         3.9353e-05],
        [1.9629e-04, 1.2876e-04, 7.5863e-05,  ..., 2.2523e-05, 3.0667e-04,
         2.4570e-01]]), 'X': tensor([[  3.1271, -10.8343,   2.6442],
        [  4.5814, -10.4485,   2.5496],
        [  5.8450,  -9.9356,   3.0767],
        [  5.4242, -10.1261,   3.9595],
        [  6.1989,  -9.7591,   0.8289],
        [  4.8051, -10.3703,   0.5268],
        [ -4.1296,   7.2803,  -2.7194],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.6587,   9.1817,  -0.8212],
        [  4.0591,   6.0202,  -1.6339],
        [  4.0362,  13.2397,  -7.0030],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -5.1974,   6.7396,   3.3484],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 12.2218,   4.0953,  -8.2072],
        [  5.1292,   4.8183,  -5.7997],
        [  6.3360,   0.4258,  -9.2509],
        [  5.0348,  -2.8229,   3.5436],
        [ -5.0132,  -9.2234,  -8.3531],
        [  5.9526,   3.2525,  -4.3821],
        [  2.4697,  -4.5836,   1.4593],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.8450,  -9.9356,   3.0767],
        [  5.1890,   4.7318,   0.9515],
        [  9.5203,   9.3483,   5.8781],
        [  3.6469,   6.3624,  -1.8414],
        [  1.1644, -11.0394,   7.8511],
        [  5.3860,  -2.3418,   1.0826],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.3534,   1.2990,  -7.2942],
        [ -0.5709,   7.9055,   7.6089],
        [ 11.7349,  -3.5838,   2.1644],
        [ -9.4630,  -1.5145,   0.2983],
        [  3.6703,   6.3443,  -5.4401],
        [  6.3305,   1.6119,   1.1538],
        [ 11.7795,  -3.4697,   3.5624],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.5401,   4.6718,   9.0033],
        [  1.5653,  13.9377,   4.1773],
        [ -3.2053, -10.1993,  -3.8736],
        [ 10.9394,   7.3069,  -8.9365],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.2218,   4.0953,  -8.2072],
        [  9.3424,  -7.3685,  -3.6807],
        [ -8.8058,   2.4850,  -1.7425],
        [  9.2811,  -7.4336,   5.1771],
        [  2.4697,  -4.5836,   1.4593],
        [ -8.1666,   3.7653,   3.6075],
        [ -7.9667,   4.0814,  -0.6397],
        [  0.4119,  14.1019,  -4.1870],
        [ -7.4602,   4.7718,  -4.6277],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.1886,  -2.6250,   6.4677],
        [ 11.0864,   7.0387,  -0.1238],
        [  4.5800,   5.5034,  -3.9432],
        [ -8.0330,  -6.0765,   6.7945],
        [  2.0449,  13.8402,  -7.1642],
        [  1.0391,  -4.8141,  -0.0768],
        [  3.2226,  -4.2802,  -4.0428],
        [  1.0000,   7.6591,  -4.5011],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.3118,   4.5452,   4.2114],
        [ -9.1624,  -3.4286,   2.0647],
        [ -7.6650,   4.5092,   0.2520],
        [  9.3560,   9.5391,   9.4943],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3236,   0.3011,  -4.6668],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -3.0008, -10.2827,  -8.9480],
        [ -8.0611,  -6.0302,   4.6962],
        [ -7.6650,   4.5092,   0.2520],
        [  9.9058,  -6.7203,   5.6928],
        [ -2.7572,   7.7194,   5.9580],
        [  4.5340,  -3.3629,   8.9652],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.5832,   1.7543,  -7.6134],
        [ 10.5678,   7.9261,  -0.6100],
        [ 12.5919,   0.4035,  -5.8460],
        [ -4.4503,   7.1378,   3.3751],
        [  3.7121,   6.3117,   0.6738],
        [  5.9655,   3.2172,  -7.5953],
        [  5.1861,   4.7361,  -3.7312],
        [  4.9531,   5.0576,   5.6794],
        [  0.3724,   7.7974,  -6.8722],
        [ 10.4758,  -5.9539,   1.3667],
        [ 12.4901,  -0.7052,   1.7315],
        [ -2.1817,   7.8274,  -4.6298],
        [ -8.9033,   2.2329,  -8.8912],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.3490,  -4.6199,  -4.2489],
        [  9.0652,   9.8597,  -3.5949],
        [  0.9809,   7.6642,  -1.4217],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.8474,   6.8979,   6.6457],
        [ -3.5029,   7.5138,   8.1853],
        [  3.3069,   6.6089,   8.7377],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.3589,  13.7668,  -3.6501],
        [ -6.3489,   5.9117,   6.5161],
        [ -9.4541,  -0.3327,   8.8246],
        [  1.7659,   7.4127,   4.7871],
        [  6.2975,   0.0947,   5.1791],
        [ 12.1751,  -2.2592,  -5.7751],
        [  4.3459,  -3.5327,   0.3667],
        [ -8.9033,   2.2329,  -8.8912]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0875, 0.1802,  ..., 0.9212, 0.5573, 0.7804],
        [0.0875, 0.0000, 0.1030,  ..., 0.8575, 0.4731, 0.8110],
        [0.1802, 0.1030, 0.0000,  ..., 0.8517, 0.4228, 0.9029],
        ...,
        [0.9212, 0.8575, 0.8517,  ..., 0.0000, 0.4938, 0.9355],
        [0.5573, 0.4731, 0.4228,  ..., 0.4938, 0.0000, 0.9828],
        [0.7804, 0.8110, 0.9029,  ..., 0.9355, 0.9828, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.27656488, 11.4088043 , 11.52733581, 11.48734978, 11.56146873,
       11.42947632,  8.36995953, 11.9182956 , 13.32645552,  7.26074925,
       13.84126197, 12.1460931 ,  8.51088913, 12.00995898, 12.889695  ,
        7.03733775,  6.35029274,  5.77219077, 10.4977132 ,  6.78325151,
        5.20659892,  5.53255882, 11.52733581,  7.02253235, 13.3426541 ,
        7.33350642, 11.10066126,  5.87305005,  9.60412735,  6.48486171,
        7.92607342, 12.26997165,  9.58347456,  7.32950385,  6.53250937,
       12.27991991,  8.16714836,  8.87007495, 14.02532758, 10.6910768 ,
       13.15525803,  5.05444288, 12.889695  , 11.89855267,  9.14973319,
       11.89105259,  5.20659892,  8.99276564,  8.95133961, 14.10796466,
        8.85572292,  9.58347456,  5.81481044, 13.13205357,  7.15990676,
       10.07238463, 13.99048094,  4.92497514,  5.35774573,  7.72415048,
        9.75200848,  6.99097013,  9.78285424,  8.89301516, 13.3614598 ,
       12.87752052,  6.33075682,  9.85330646, 10.71162681, 10.06703358,
        8.89301516, 11.9702762 ,  8.19703512,  5.64500512,  9.69629   ,
        9.11473391, 12.70489563, 13.20988996, 12.59840193,  8.41148722,
        7.32233733,  6.77777606,  7.02326515,  7.07902036,  7.80625981,
       12.0495521 , 12.50996828,  8.12580976,  9.17905491, 11.9182956 ,
        5.18277858, 13.39372426,  7.72668467,  8.20436764,  7.46248947,
        8.29022197,  7.39004226,  5.30158497, 13.96746816,  8.67502348,
        9.45995955,  7.62011734,  6.29821673, 12.3828957 ,  5.60064319,
        9.17905491])}
{'P': tensor([[6.1824e-03, 2.2119e-01, 2.1194e-01,  ..., 8.8053e-04, 9.5017e-08,
         6.0218e-07],
        [2.2731e-04, 7.8624e-02, 2.4194e-03,  ..., 3.4846e-02, 9.2676e-07,
         1.8245e-05],
        [1.0936e-04, 6.2808e-01, 2.9229e-02,  ..., 1.2906e-02, 3.8423e-07,
         6.1096e-06],
        ...,
        [4.0879e-06, 8.7136e-06, 7.9562e-06,  ..., 6.2811e-02, 1.7450e-06,
         9.0756e-04],
        [5.5196e-05, 2.9199e-05, 3.5971e-05,  ..., 5.3056e-06, 5.3590e-02,
         3.3535e-07],
        [2.0951e-06, 1.4715e-06, 1.5351e-06,  ..., 1.5327e-06, 1.4858e-04,
         2.1682e-01]]), 'X': tensor([[  3.1756, -10.8247,   6.8882],
        [  5.7218,  -9.9934,   6.5764],
        [  5.4167, -10.1293,   6.2156],
        [  4.0587, -10.6112,   6.2273],
        [  5.6414, -10.0303,   7.8525],
        [  4.7482, -10.3907,   5.9149],
        [ 12.6038,   0.7665,  -3.4439],
        [  6.3413,   0.4883,  -6.8416],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -5.9940,   6.1985,   5.9566],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.3301,   0.3638,  -7.5685],
        [  3.3474,  -4.2162,   5.2455],
        [  6.7565,  11.8237,  -4.9882],
        [  9.0652,   9.8597,  -3.5949],
        [  6.3205,   1.7068,  -3.8619],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.6038,   0.7665,  -3.4439],
        [ 10.9394,   7.3069,  -8.9365],
        [  0.9967,  -4.8144,  -9.7370],
        [ -4.4520,   7.1370,  -8.7867],
        [  2.5128, -10.9365,   9.9645],
        [ -2.5161,   7.7699,  -9.3531],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.2767,  -4.6006,   0.7868],
        [ -8.0608,   3.9362,  -5.9567],
        [ -2.0229, -10.6163,   8.6930],
        [ -4.0409,   7.3169,  -3.5779],
        [  8.0877,  -8.5297,  -1.1050],
        [  1.1644, -11.0394,   7.8511],
        [ -6.7992,   5.4991,   9.9186],
        [  0.4119,  14.1019,  -4.1870],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.1433,  -3.5003,  -3.0209],
        [  2.5974,   7.0363,  -7.7810],
        [ -6.3489,   5.9117,   6.5161],
        [ 11.4743,   6.2478,  -2.1100],
        [ -9.1459,   1.4780,  -4.2403],
        [ -1.3940, -10.7772,   6.7835],
        [  7.4132,  11.3526,   0.3322],
        [  5.2608,  12.6977,   4.9166],
        [  0.7685,  -4.8098,  -2.9310],
        [ -9.1624,  -3.4286,   2.0647],
        [ -1.5261,   7.8988,   9.3768],
        [  9.5018,  -7.1945,  -1.1892],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -1.6334, -10.7207,   7.1397],
        [ -9.2373,  -3.1225,  -8.9763],
        [  1.1994,   7.6035,   2.4295],
        [  5.0190,   4.9705,  -7.9752],
        [ -1.9714,   7.8563,  -1.8084],
        [  5.9003,  12.3550,   2.1917],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.3943,   6.5483,   1.0790],
        [  6.3124,   0.2057,  -0.4593],
        [ -5.6651,   6.4380,   1.3660],
        [  5.9548,  -1.2273,  -2.6520],
        [  4.1140,  -3.7220,  -0.1273],
        [ -1.8788, -10.6567,   3.7068],
        [  7.2340,  -9.1505,  -5.8321],
        [  7.3592,  11.3937,   6.2407],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -4.0633,  -9.7927,  -3.0825],
        [  2.8994,   6.8675,  -9.5987],
        [  2.3261,  13.7749,   5.9121],
        [  6.2419,  -0.2313,   4.2440],
        [  0.2571,  -4.7594,   1.5618],
        [  4.2096,   5.8815,   1.9230],
        [  3.2366,  -4.2732,   5.1532],
        [  6.3031,   1.8477,  -7.1706],
        [ -5.1369,   6.7755,   8.3836],
        [ -8.0611,  -6.0302,   4.6962],
        [ -8.0750,  -6.0072,  -5.8267],
        [  9.6587,   9.1817,  -0.8212],
        [  1.5246,  -4.7838,  -5.2095],
        [ -2.1817,   7.8274,  -4.6298],
        [  8.0877,  -8.5297,  -1.1050],
        [  2.3376,  -4.6231,   9.8436],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -7.9667,   4.0814,  -0.6397],
        [ 11.0864,   7.0387,  -0.1238],
        [  5.8275,  -1.5334,   2.0124],
        [  4.9452, -10.3187,  -7.8621],
        [  6.1575,  12.2046,   3.3257],
        [ -7.4733,   4.7555,   6.2490],
        [  0.3038,   7.8092,   0.4164],
        [  3.9895,   6.0816,  -6.5235],
        [  5.6314,  -1.9302,  -9.6543],
        [ -7.5720,   4.6310,   2.1717],
        [ 11.5568,   6.0602,   9.3244],
        [ 10.8218,  -5.4177,   6.8573],
        [ 10.0619,   8.6627,  -9.2884],
        [ 10.3251,   8.2928,  -3.0006],
        [ -4.2040,   7.2486,   2.9875],
        [  6.8407,  11.7666,   3.0045],
        [ 12.5919,   0.4035,  -5.8460],
        [ -7.9667,   4.0814,  -0.6397],
        [  5.1196,  -2.7160,   4.2705],
        [  5.4242, -10.1261,   3.9595],
        [  6.3122,  -9.6994,  -6.3282],
        [ 11.7795,  -3.4697,   3.5624],
        [ -6.4844,  -8.0230,   8.5209],
        [ 12.2311,  -2.0445,   1.3545],
        [ -9.4721,  -1.3097,  -7.4377],
        [  2.2182,  13.8007,   5.1889]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2214e-01, 1.0672e-01,  ..., 7.6213e-01, 1.1519e+00,
         8.5183e-01],
        [1.2214e-01, 0.0000e+00, 2.4510e-02,  ..., 6.6849e-01, 1.1940e+00,
         7.2994e-01],
        [1.0672e-01, 2.4510e-02, 0.0000e+00,  ..., 6.6912e-01, 1.1711e+00,
         7.4820e-01],
        ...,
        [7.6213e-01, 6.6849e-01, 6.6912e-01,  ..., 1.0727e-04, 1.1491e+00,
         5.6156e-01],
        [1.1519e+00, 1.1940e+00, 1.1711e+00,  ..., 1.1491e+00, 0.0000e+00,
         1.6572e+00],
        [8.5183e-01, 7.2994e-01, 7.4820e-01,  ..., 5.6156e-01, 1.6572e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.28093507, 11.51557166, 11.48663767, 11.36089355, 11.50791859,
       11.42420508, 12.62711252,  6.36004393, 10.0674887 ,  8.62260327,
       12.42554808,  6.34059021,  5.38342077, 13.61801526, 13.39372426,
        6.54693987,  9.66977802, 12.62711252, 13.15525803,  4.91652556,
        8.41171369, 11.22142195,  8.16714836, 12.86648771, 12.17901033,
        8.97052115, 10.80728788,  8.35856487, 11.75438006, 11.10066126,
        8.74470218, 14.10796466,  6.33075682,  9.79039371,  7.50035355,
        8.67502348, 13.06498554,  9.26455795, 10.86693708, 13.55868684,
       13.744387  ,  4.87083348,  9.78285424,  8.04483495, 11.9182956 ,
       10.5575601 , 10.84437705,  9.75075273,  7.69752262,  7.06372172,
        8.09984231, 13.69162795,  6.10774544,  7.37576154,  6.31576264,
        8.57560275,  6.07992623,  5.54776934, 10.82106505, 11.66453059,
       13.56367857,  8.09899874,  9.43112957, 10.60226614,  7.4544866 ,
       13.96987806,  6.24614905,  4.7663526 ,  7.23276582,  5.36061974,
        6.56833864,  8.50268322, 10.06703358, 10.06437598, 13.32645552,
        5.0209005 ,  8.12580976, 11.75438006,  5.1805345 , 12.35625694,
        8.95133961, 13.13205357,  6.02588517, 11.4424816 , 13.66991027,
        8.8580671 ,  7.81509631,  7.27340834,  5.95297459,  8.87587457,
       13.0493368 , 12.10222204, 13.27717982, 13.2430377 ,  8.37954853,
       13.6105595 , 12.59840193,  8.95133961,  5.79543038, 11.48734978,
       11.57249984, 12.27991991, 10.31584102, 12.40074901,  9.562174  ,
       13.97779839])}
{'P': tensor([[4.0618e-03, 2.6612e-01, 1.2766e-01,  ..., 7.1528e-05, 2.1034e-05,
         2.6051e-07],
        [2.8289e-05, 4.7806e-03, 2.5496e-01,  ..., 7.8195e-05, 2.8107e-05,
         2.4014e-07],
        [4.2639e-05, 6.3939e-05, 1.7205e-02,  ..., 1.7507e-04, 1.0302e-04,
         5.6775e-07],
        ...,
        [4.4496e-06, 7.6085e-06, 1.7926e-05,  ..., 7.0317e-03, 2.5197e-03,
         3.7707e-07],
        [2.5403e-05, 3.2642e-05, 4.7276e-05,  ..., 5.7709e-04, 8.1771e-02,
         4.5492e-05],
        [9.4209e-04, 7.6132e-04, 4.4134e-04,  ..., 1.5585e-03, 6.5557e-04,
         6.2602e-02]]), 'X': tensor([[ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.1440e-02, 1.6648e-01,  ..., 1.0589e+00, 1.0768e+00,
         9.1799e-01],
        [6.1440e-02, 0.0000e+00, 1.0506e-01,  ..., 1.0434e+00, 1.0415e+00,
         9.4056e-01],
        [1.6648e-01, 1.0506e-01, 0.0000e+00,  ..., 1.0282e+00, 9.9014e-01,
         9.8970e-01],
        ...,
        [1.0589e+00, 1.0434e+00, 1.0282e+00,  ..., 0.0000e+00, 3.7366e-01,
         6.2535e-01],
        [1.0768e+00, 1.0415e+00, 9.9014e-01,  ..., 3.7366e-01, 0.0000e+00,
         9.4880e-01],
        [9.1799e-01, 9.4056e-01, 9.8970e-01,  ..., 6.2535e-01, 9.4880e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.28215429, 11.37939695, 11.54672755, 11.35182589, 11.54071859,
       11.57249984,  9.62238828, 11.05317597,  9.0368458 , 13.59170809,
       12.3344536 ,  5.01114719,  6.54693987,  9.27474386,  5.24137909,
        9.64157461,  7.15100303,  5.08910491,  5.37343173, 12.78327037,
       13.59170809, 11.33837396, 14.02532758,  6.33075682,  9.66977802,
       12.53047489, 11.42420508,  7.06169391,  5.13182506, 10.88618114,
        6.89245352, 12.71843462, 13.77222752, 13.97857591, 11.4424816 ,
        5.79562691,  5.0209005 , 11.22142195, 13.39372426,  9.10945678,
       11.89105259,  8.35856487,  6.02779201, 11.83075005,  6.00663314,
        6.29034958, 10.00802627,  6.5618968 ,  4.92497514,  6.39520397,
        6.10774544, 12.4013153 ,  8.97052115,  4.74770899, 11.73567759,
        8.10886021,  6.76661595,  7.10378919, 11.51557166, 11.54071859,
        6.46560181,  5.82345867, 11.9702762 ,  6.00663314, 11.27656488,
        6.7593917 , 11.00508681, 10.42667093,  9.85330646,  5.84450127,
       11.76830981, 11.97238624,  6.78325151, 13.98691426, 11.35345345,
       13.39372426, 11.97957425, 12.04924159, 10.38897563,  9.9353921 ,
       10.51437947,  6.76056625,  7.62011734,  5.30158497, 11.42947632,
        6.61386436,  5.16205087, 11.24434374, 11.83570132,  5.89074104,
       10.91601655,  8.7273569 ,  5.08910491,  8.89301516, 12.62576026,
       13.03002338,  4.89121805,  7.17789666,  7.32950385, 13.25336001,
        5.5205189 ,  4.91336994,  8.8737556 , 11.70016217,  5.91677188,
       10.05558893])}
{'P': tensor([[1.1594e-02, 2.4151e-01, 2.2056e-01,  ..., 5.9042e-06, 4.6424e-04,
         7.6574e-06],
        [8.3215e-05, 5.0331e-02, 3.6602e-02,  ..., 6.0322e-05, 3.5547e-03,
         7.7932e-05],
        [1.9456e-05, 7.1463e-03, 1.1373e-02,  ..., 1.2251e-05, 1.0998e-03,
         1.8213e-05],
        ...,
        [3.7642e-05, 2.3836e-05, 2.1963e-05,  ..., 2.3636e-01, 1.2285e-04,
         1.8419e-02],
        [3.7629e-05, 7.3508e-05, 8.2503e-05,  ..., 7.0418e-04, 7.9331e-02,
         5.2333e-03],
        [9.1035e-06, 5.6337e-06, 5.7502e-06,  ..., 4.5223e-03, 7.3916e-05,
         4.8772e-02]]), 'X': tensor([[  3.2159, -10.8166,  -1.0406],
        [  6.9314,  -9.3435,  -1.2729],
        [  6.9433,  -9.3362,  -0.3421],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.1989,  -9.7591,   0.8289],
        [  4.8051, -10.3703,   0.5268],
        [  3.1756, -10.8247,   6.8882],
        [  2.2849, -10.9655,  -7.6542],
        [ -7.4733,   4.7555,   6.2490],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.6613,   7.0021,   4.5783],
        [ -1.9646,   7.8571,   3.3817],
        [ 11.6165,  -3.8715,   6.4594],
        [ -0.8885,   7.9160,   0.0571],
        [  1.1644, -11.0394,   7.8511],
        [  2.7590,  13.6621,   9.4496],
        [ -0.8455, -10.8853,  -6.7400],
        [  5.7784,  -1.6401,   9.5587],
        [  5.0277,   4.9589,   7.0175],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -2.0721,   7.8432,   6.6380],
        [  6.1984,  -0.4327,   1.5438],
        [  5.8323,  -1.5228,   5.1132],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -5.0943,  -9.1682,   0.5073],
        [  2.1440, -10.9811,   8.5281],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -9.4196,   0.0478,   9.4140],
        [ -0.9715, -10.8630,   0.3555],
        [ -9.1658,  -3.4155,  -2.8880],
        [  3.5965,  -4.0754,  -3.5035],
        [  5.1472,  -2.6801,   8.1194],
        [  2.5663,   7.0526,  -6.8769],
        [ 12.4592,  -0.9168,  -6.7571],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.8450,  -9.9356,   3.0767],
        [ -4.4779,  -9.5609,  -2.4505],
        [  6.3569,   0.7591,   6.5011],
        [  5.9843,  -9.8679,  -6.1449],
        [  7.5309,  11.2617,   7.2656],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.8902,  -1.3887,   4.0388],
        [  6.1929,   2.4501,  -6.9352],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.0305,   3.0296,   0.1375],
        [ -0.4341, -10.9473,   7.4509],
        [  2.5469,  13.7192,  -9.6935],
        [  4.4290,  -3.4596,   1.4181],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -4.4503,   7.1378,   3.3751],
        [  1.5662,  13.9375,   4.5142],
        [  6.2022,   2.4092,   3.0349],
        [ 11.6019,   5.9540,  -5.1715],
        [ 11.2767,  -4.6006,   0.7868],
        [ 12.5717,   1.9266,   7.4903],
        [  1.0093,   7.6567,   2.3526],
        [ -9.3616,  -2.4750,   3.7765],
        [ 12.5763,   0.1333,  -9.2650],
        [ -9.0604,   1.7707,   1.1066],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 12.2914,   3.8037,   6.0978],
        [ -2.7572,   7.7194,   5.9580],
        [ -7.5603,   4.6459,   8.2002],
        [  3.9895,   6.0816,  -6.5235],
        [ -1.5261,   7.8988,   9.3768],
        [  6.3460,   1.4252,   1.2291],
        [  9.3560,   9.5391,   9.4943],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.0330,  -6.0765,   6.7945],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.6954,  -4.7616,   9.8198],
        [  8.4691,  10.4571,  -5.0650],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.5424,  -2.0885,   7.2487],
        [ 12.5832,   1.7543,  -7.6134],
        [ -3.4488, -10.0934,   2.5195],
        [  4.5325,  -3.3643,   6.7699],
        [  4.6231,   5.4556,   8.4973],
        [  2.2849, -10.9655,  -7.6542],
        [  3.4140,  -4.1803,  -8.8224],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.1787,  -2.6383,  -4.2533],
        [ -5.1369,   6.7755,   8.3836],
        [ 10.1503,   8.5414,  -7.2681],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.3493,  -4.7727,  -6.5682],
        [ -6.9838,  -7.4965,   0.9796],
        [ 12.2435,   4.0076,   8.1257],
        [  6.2905,   1.9366,   2.1169],
        [ -1.7009,   7.8851,  -8.2437],
        [ 12.0845,  -2.5772,  -4.6073],
        [  0.1460,   7.8340,  -9.4614],
        [  4.3459,  -3.5327,   0.3667],
        [  6.3200,   0.2690,   1.5900],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.0788,  -0.8710,  -6.8719],
        [ 11.4235,   6.3594,   5.1663],
        [  4.9531,   5.0576,   5.6794],
        [  2.0729,   7.2878,   7.0880],
        [ 10.6283,  -5.7253,   7.4878],
        [  0.1497,   7.8334,  -5.4412],
        [  6.2883,   0.0325,   5.2692],
        [  0.3038,   7.8092,   0.4164]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1958, 0.1850,  ..., 0.3849, 0.7360, 0.3316],
        [0.1958, 0.0000, 0.0724,  ..., 0.2506, 0.6108, 0.1744],
        [0.1850, 0.0724, 0.0000,  ..., 0.3222, 0.5694, 0.1473],
        ...,
        [0.3849, 0.2506, 0.3222,  ..., 0.0000, 0.7625, 0.3593],
        [0.7360, 0.6108, 0.5694,  ..., 0.7625, 0.0000, 0.4370],
        [0.3316, 0.1744, 0.1473,  ..., 0.3593, 0.4370, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.2845608 , 11.63383221, 11.63502679, 11.69264243, 11.56146873,
       11.42947632, 11.28093507, 11.20100795,  8.8580671 ,  4.78258092,
        7.49076899,  8.09899874, 12.24467041,  7.96575782, 11.10066126,
       13.93790235, 10.91805683,  6.00663314,  7.06169391,  9.60080639,
        8.11227668,  6.2134961 ,  6.02779201,  7.83534987,  9.71392179,
       10.48840476, 11.18839169, 12.53285633,  9.41970031, 10.90638066,
        9.78147542,  5.43544626,  5.80312949,  7.50500064, 12.49291768,
       11.75618513, 11.52733581, 10.5575601 ,  6.4020404 , 11.54071859,
       13.54774355,  9.71392179,  6.05165248,  6.65991877,  6.23262289,
        6.74872981, 10.95593729, 13.95361098,  5.62005629, 10.37698435,
        8.41148722, 14.02526505,  6.65368551, 13.04051744, 12.17901033,
       12.71843462,  7.72291946,  9.68324553, 12.5769659 ,  9.23178204,
        9.55720651, 12.86648771,  8.19703512,  8.8737556 ,  7.27340834,
        8.04483495,  6.50409372, 13.3614598 ,  9.76896904, 10.07238463,
        7.28793663,  9.85384623,  5.05444288, 13.45642379, 12.41012046,
        5.92281142, 12.70489563, 10.66631784,  5.64464807,  7.15100303,
       11.20100795,  5.39723241,  8.97052115,  5.26067019,  5.81199422,
        8.50268322, 13.26590071,  7.8575648 ,  4.78544875, 10.24557047,
       12.88270018,  6.58184474,  8.06643487, 12.35625694,  7.83534987,
        5.60064319,  6.32572323,  5.24461068,  6.14086582, 13.0743399 ,
        7.07902036,  7.57685908, 12.07225469,  7.83486943,  6.28835101,
        7.81509631])}
{'P': tensor([[8.7506e-03, 2.5241e-01, 1.8744e-01,  ..., 3.8888e-03, 5.7661e-07,
         9.4125e-05],
        [5.0741e-05, 6.8734e-03, 2.1673e-01,  ..., 4.9745e-03, 4.1851e-07,
         1.0001e-04],
        [1.5921e-05, 3.6106e-05, 9.5593e-03,  ..., 1.9142e-03, 5.4372e-07,
         4.2155e-04],
        ...,
        [9.3023e-05, 2.3514e-04, 1.0844e-03,  ..., 9.5951e-02, 4.4401e-06,
         3.7259e-04],
        [4.3435e-04, 2.4654e-04, 1.0503e-04,  ..., 3.8431e-05, 8.2034e-02,
         1.5536e-06],
        [2.9515e-07, 3.5064e-07, 6.6520e-07,  ..., 3.6306e-07, 1.6972e-06,
         1.8557e-02]]), 'X': tensor([[ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.3126e-02, 1.8087e-01,  ..., 7.0400e-01, 3.0191e-01,
         9.5820e-01],
        [9.3126e-02, 7.5848e-05, 1.1598e-01,  ..., 6.2488e-01, 3.5399e-01,
         9.3545e-01],
        [1.8087e-01, 1.1598e-01, 0.0000e+00,  ..., 6.4474e-01, 4.6832e-01,
         8.2590e-01],
        ...,
        [7.0400e-01, 6.2488e-01, 6.4474e-01,  ..., 0.0000e+00, 7.5223e-01,
         1.3097e+00],
        [3.0191e-01, 3.5399e-01, 4.6832e-01,  ..., 7.5223e-01, 0.0000e+00,
         1.2566e+00],
        [9.5820e-01, 9.3545e-01, 8.2590e-01,  ..., 1.3097e+00, 1.2566e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.28895489, 11.42947632, 11.63502679, 11.56146873, 11.63383221,
       11.69264243,  5.18277858,  8.49161154, 12.40074901,  9.34127013,
        9.0368458 , 13.2430377 ,  5.52416368, 12.41366754,  5.81199422,
        5.36061974,  7.14837514,  5.34250225, 12.46513646,  7.38844703,
        6.38399789, 10.33958639, 13.72594022,  5.79391825,  9.71392179,
        4.98461983,  7.5158489 , 13.05148826,  7.19534146,  5.90035324,
        5.54776934,  6.2134961 ,  7.54620849,  7.19513806, 11.23201619,
        5.0713364 , 10.00794468,  6.78325151,  6.76056625, 10.14436912,
        7.17789666,  9.78147542, 13.744387  ,  9.48969978, 13.27994304,
       12.84056604,  8.05668609,  7.10378919,  7.8788965 ,  8.35856487,
        5.47866221,  7.50035355, 12.25806559, 14.09357547, 12.1460931 ,
        4.78002665,  8.27853076,  7.57946459,  9.88310614,  8.859416  ,
       10.66631784,  7.50035355, 13.89974693,  7.06169391,  8.36995953,
       11.28895489,  7.19534146, 12.25806559,  9.84314712,  5.46244988,
       11.11969815,  8.53501777,  5.70405525,  9.68324553,  8.10886021,
        5.95111804,  8.72896381, 10.95593729, 12.67680439, 13.79676654,
       12.19805394,  7.27340834,  5.19986797,  6.19633139,  5.05963217,
       10.84493732,  9.18358562,  6.78325151,  7.28793663, 10.82106505,
        8.31625171,  6.36004393, 13.0743399 , 13.3426541 ,  7.16598273,
        7.13501264,  6.06416957,  5.83560009,  6.81117802,  5.47866221,
       13.45642379,  4.90505566,  5.87132718, 11.8159254 ,  9.23178204,
       12.76408562])}
{'P': tensor([[6.8967e-03, 2.9015e-01, 2.1429e-01,  ..., 1.7901e-06, 2.4695e-07,
         7.8536e-07],
        [5.3564e-05, 7.3020e-03, 4.3990e-01,  ..., 2.1031e-06, 2.3718e-07,
         6.1310e-07],
        [7.3845e-05, 1.2322e-04, 1.4288e-02,  ..., 4.3999e-06, 4.4561e-07,
         1.0218e-06],
        ...,
        [4.4843e-06, 3.4860e-06, 3.0715e-06,  ..., 6.3598e-02, 2.4535e-02,
         2.0066e-04],
        [2.9038e-05, 1.7687e-05, 1.3561e-05,  ..., 5.3964e-05, 4.7632e-02,
         6.5540e-03],
        [1.1221e-02, 6.0490e-03, 4.2979e-03,  ..., 1.5388e-05, 4.8409e-05,
         1.0323e-01]]), 'X': tensor([[  3.5284, -10.7484,   5.3347],
        [  4.7482, -10.3907,   5.9149],
        [  5.4167, -10.1293,   6.2156],
        [  5.7218,  -9.9934,   6.5764],
        [  5.4242, -10.1261,   3.9595],
        [  7.5784,  -8.9141,   5.8410],
        [ 11.1420,   6.9331,  -2.4779],
        [  0.3493,  -4.7727,  -6.5682],
        [  6.2419,  -0.2313,   4.2440],
        [  5.1142,  -2.7230,  -4.1966],
        [  9.4554,   9.4245,   9.8800],
        [  3.1064,  13.5607,   6.7605],
        [  2.7685, -10.8982,  -6.8928],
        [  3.7121,   6.3117,   0.6738],
        [ -3.5903,   7.4847,  -8.9112],
        [  0.7145,  -4.8071,  -2.3766],
        [ -8.1426,   3.8048,   9.3542],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.3246,   1.6696,  -0.1544],
        [  2.8602,   6.8905,   5.8779],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.4429,  -2.2529,   9.7306],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.6928,  12.4710,   3.6001],
        [  1.6308,  13.9254,  -2.0984],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.2571,   6.7072,   8.3693],
        [  7.8375,  -8.7240,  -9.8913],
        [ -6.4822,  -8.0252,  -3.8983],
        [  9.5261,  -7.1675,   1.4206],
        [ -3.4986,   7.5152,   9.0712],
        [  7.0515,  11.6194,   8.4289],
        [  2.0547,   7.2957,  -1.9343],
        [  6.1579,  -0.5961,  -6.5595],
        [ -9.3107,  -2.7697,  -3.2919],
        [  6.2242,  -0.3171,   7.4978],
        [ -3.3999, -10.1152,  -1.2088],
        [ -9.4739,  -1.2497,   0.5483],
        [  5.0506,   4.9277,  -9.5094],
        [  6.3583,   1.1729,   8.7835],
        [  0.1206,  -4.7363,   8.7103],
        [  6.8407,  11.7666,   3.0045],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.9282,   2.1648,   8.4894],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.3860,  -2.3418,   1.0826],
        [  2.7976,  -4.4682,  -4.2020],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.2974,   0.8371,   2.4852],
        [ -0.8675, -10.8815,   6.8561],
        [  5.1519, -10.2387,   9.8191],
        [ -2.1382,   7.8339,   7.3524],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.0457,  -3.7737,  -0.4704],
        [ -3.1204, -10.2345,   9.2288],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -0.0282,   7.8575,  -1.1746],
        [  2.2860,   7.1916,  -6.1011],
        [  4.1348,   5.9514,  -0.8856],
        [ -4.4503,   7.1378,   3.3751],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.2199,   4.6861,  -1.9304],
        [ -5.9987,   6.1949,  -6.2114],
        [ -8.6603,   2.8250,   5.9181],
        [  5.9883,   3.1536,   6.2320],
        [ -8.3535,  -5.5118,   9.8399],
        [  5.7963,  -1.6018,   1.8856],
        [ 12.2218,   4.0953,  -8.2072],
        [  5.5240,  -2.1198,   6.2055],
        [  0.5960, -11.0335,   3.1144],
        [ -8.0608,   3.9362,  -5.9567],
        [  8.1043,  -8.5164,  -7.9290],
        [  8.7131,  10.2216,   7.7915],
        [ 11.0891,  -4.9558,  -2.3178],
        [  5.9655,   3.2172,  -7.5953],
        [ 12.4837,   2.7510,   0.4449],
        [  2.2437,  -4.6489,   6.1430],
        [  5.8450,  -9.9356,   3.0767],
        [ -5.6174,   6.4708,  -1.7339],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.1890,   4.7318,   0.9515],
        [ -8.3062,  -5.6008,   0.0287],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 11.0373,   7.1299,  -8.5895],
        [  0.3354,  -4.7708,  -4.9125],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.3550,   0.7129,   0.4917],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.2975,   0.0947,   5.1791],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.4932,   7.0903,  -5.4214],
        [  3.9022,  13.2906,   5.5788],
        [ -9.3604,   0.4855,  -3.0158],
        [  2.7590,  13.6621,   9.4496],
        [ -8.6381,  -4.9238,  -4.4077],
        [  5.8323,  -1.5228,   5.1132],
        [ -2.1817,   7.8274,  -4.6298],
        [  3.4604,  -4.1545,   1.8091],
        [ -2.5161,   7.7699,  -9.3531],
        [  0.4996,   7.7738,  -4.4711],
        [ -7.9298,   4.1367,  -3.9773],
        [ -8.4267,  -5.3697,   2.4172]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0646, 0.1007,  ..., 0.8450, 0.8124, 0.5151],
        [0.0646, 0.0000, 0.0403,  ..., 0.8814, 0.8725, 0.5768],
        [0.1007, 0.0403, 0.0000,  ..., 0.8843, 0.8968, 0.6155],
        ...,
        [0.8450, 0.8814, 0.8843,  ..., 0.0000, 0.5019, 0.8738],
        [0.8124, 0.8725, 0.8968,  ..., 0.5019, 0.0000, 0.5372],
        [0.5151, 0.5768, 0.6155,  ..., 0.8738, 0.5372, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.31276109, 11.42420508, 11.48663767, 11.51557166, 11.48734978,
       11.70016217, 13.12299145,  4.78544875,  6.24614905,  5.79391825,
       13.35013374, 13.91198191, 11.24434374,  7.32233733,  8.30125201,
        4.85995002,  8.98765644, 12.59969936,  6.54127431,  7.46053281,
        9.43112957, 12.4217523 , 12.25806559,  5.89074104,  9.97220799,
       13.70893492, 14.02058621, 12.04924159, 13.103709  , 11.72749923,
       10.31613269, 11.92133129,  8.28968067, 13.59170809,  7.57946459,
        6.18668463,  9.71392179,  6.2322769 , 10.67132524,  9.55593374,
        7.05627461,  6.46560181,  4.73784222, 13.6105595 ,  9.84314712,
        9.18690018, 11.80092578,  5.18277858,  5.87305005,  5.27178363,
       13.36092624,  9.3349858 , 10.91601655, 11.46176259,  8.12043983,
        5.1224079 ,  5.53255882, 10.69963954, 10.06286074,  9.88571713,
        7.8575648 ,  7.54620849,  7.24677664,  8.41148722, 11.81423161,
        7.01475376,  8.62329466,  9.10945678,  6.76791483, 10.00802627,
        6.01356004, 12.889695  ,  5.91677188, 11.04953767,  8.97052115,
       11.75618513, 13.43127131, 12.1460931 ,  6.77777606, 12.78327037,
        5.16205087, 11.52733581,  8.56889738,  4.9160962 ,  7.02253235,
       10.01804223, 10.5575601 , 13.1399181 ,  4.78258092, 10.06023665,
        6.3949043 , 10.06437598,  6.29821673,  8.62329466,  7.5158489 ,
       13.85158417,  9.37296102, 13.93790235,  9.94285511,  6.02779201,
        8.12580976,  5.40689332,  8.16714836,  7.78980299,  8.94395628,
        9.99211663])}
{'P': tensor([[4.2361e-03, 2.6785e-01, 1.7857e-01,  ..., 7.1645e-07, 2.2512e-07,
         1.6568e-04],
        [2.3256e-05, 4.0339e-03, 2.4302e-01,  ..., 9.7505e-07, 1.7742e-07,
         1.0501e-04],
        [3.1099e-05, 5.9227e-05, 6.8675e-03,  ..., 2.0327e-06, 2.6816e-07,
         1.3359e-04],
        ...,
        [4.6125e-07, 4.0778e-07, 3.7816e-07,  ..., 8.0760e-02, 1.4994e-04,
         1.3890e-05],
        [4.6285e-04, 2.8531e-04, 2.0686e-04,  ..., 7.8020e-07, 4.3978e-02,
         1.7085e-03],
        [3.6364e-03, 4.4571e-03, 4.7162e-03,  ..., 1.9910e-04, 1.6020e-04,
         1.2614e-01]]), 'X': tensor([[  3.6249, -10.7255,   5.6605],
        [  4.7482, -10.3907,   5.9149],
        [  5.4167, -10.1293,   6.2156],
        [  5.7218,  -9.9934,   6.5764],
        [  7.5784,  -8.9141,   5.8410],
        [  7.8425,  -8.7202,   6.4530],
        [ 12.4901,  -0.7052,   1.7315],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 12.2435,   4.0076,   8.1257],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.3087,   0.7792,  -7.5587],
        [  3.9085,   6.1512,  -2.2560],
        [  3.5965,  -4.0754,  -3.5035],
        [  0.9478,  -4.8144,   0.6371],
        [  6.2026,  12.1774,  -1.5887],
        [  0.9392,  -4.8143,  -3.4649],
        [  4.8432,   5.1968,   1.5006],
        [  3.4920,  -4.1366,  -8.3955],
        [  9.2811,  -7.4336,   5.1771],
        [ 11.2939,   6.6325,  -5.8311],
        [ -8.9636,   2.0648,   8.1946],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.2038,   2.4020,  -3.1692],
        [ -9.4351,  -1.8891,   3.5642],
        [ -4.0633,  -9.7927,  -3.0825],
        [  0.1287, -11.0065,   6.5757],
        [  6.0345,   3.0175,  -1.4424],
        [ -8.0766,  -6.0046,  -3.1873],
        [  4.0078,  -3.8018,  -6.2168],
        [ -5.6423,   6.4537,  -1.2091],
        [  6.2975,   0.0947,   5.1791],
        [  3.4401,  13.4540,   4.6576],
        [  6.3516,   0.6425,   0.3425],
        [ -9.2508,   1.0572,   2.9978],
        [  6.3048,   0.1469,  -9.1691],
        [  0.9967,  -4.8144,  -9.7370],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -7.6525,  -6.6534,   5.1721],
        [  7.1821,  -9.1845,   9.8548],
        [  6.3535,   0.6792,  -7.5101],
        [  6.1741,  -0.5329,   3.1913],
        [ 10.8218,  -5.4177,   6.8573],
        [ 11.6634,   5.8050,   9.1590],
        [ -3.4986,   7.5152,   9.0712],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.4297,  -5.3637,  -9.3962],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -0.7599, -10.8995,  -6.0366],
        [  3.2692,  13.5098,  -4.9032],
        [  9.5203,   9.3483,   5.8781],
        [  6.3587,   1.1596,  -3.6629],
        [  0.6933,   7.7336,   2.1179],
        [  3.2809,  13.5061,  -4.0199],
        [ -7.4887,   4.7364,   8.9074],
        [  6.3124,   0.2057,  -0.4593],
        [  9.6662,   9.1725,   8.1056],
        [ 11.6544,   5.8273,  -3.1375],
        [  6.1579,  -0.5961,  -6.5595],
        [  2.5128, -10.9365,   9.9645],
        [ -5.1369,   6.7755,   8.3836],
        [  1.9503,  -4.7172,  -6.2313],
        [ 10.1983,  -6.3430,  -0.3200],
        [  3.5284, -10.7484,   5.3347],
        [  2.1477,   7.2550,   8.9688],
        [ -8.8154,   2.4611,   2.5198],
        [ 11.3691,   6.4760,   1.0787],
        [ 12.4877,  -0.7222,   8.8347],
        [  3.7992,  -3.9474,  -8.9019],
        [  8.1217,  -8.5024,  -0.6149],
        [ -0.8675, -10.8815,   6.8561],
        [  8.2154,  -8.4262,   8.0171],
        [  2.2860,   7.1916,  -6.1011],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.3200,   7.1756,   1.1938],
        [  5.7784,  -1.6401,   9.5587],
        [  6.3535,   0.6792,  -7.5101],
        [  5.5158,  -2.1337,   3.5246],
        [ -9.1623,  -3.4288,  -4.9961],
        [  8.6271,  -8.0715,  -7.4813],
        [ 11.0373,   7.1299,  -8.5895],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.3836,   9.5075,  -5.9278],
        [  3.7298,  13.3537,  -9.4653],
        [  0.9445,   7.6736,   0.5095],
        [ -2.5161,   7.7699,  -9.3531],
        [ 10.0995,  -6.4739,   2.0974],
        [  0.3493,  -4.7727,  -6.5682],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.2644,  -4.6434,   9.7508],
        [  6.1752,   2.5242,   3.5874],
        [  5.9587,  12.3216,   9.1695],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.4167, -10.1293,   6.2156],
        [ -9.1231,  -3.5739,   8.4708],
        [ -8.0766,  -6.0046,  -3.1873],
        [  4.3523, -10.5232,   8.2138],
        [  0.1382, -11.0073,   1.0026],
        [  5.1667,  -2.6543,   1.2879],
        [  6.1741,  -0.5329,   3.1913],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.4505,  -1.7081,  -0.7039],
        [  1.8721,  -4.7322,   0.5524]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0519, 0.0921,  ..., 1.0261, 0.6922, 0.6840],
        [0.0519, 0.0000, 0.0403,  ..., 1.0062, 0.7372, 0.6723],
        [0.0921, 0.0403, 0.0000,  ..., 0.9890, 0.7710, 0.6623],
        ...,
        [1.0261, 1.0062, 0.9890,  ..., 0.0000, 1.1667, 0.3650],
        [0.6922, 0.7372, 0.7710,  ..., 1.1667, 0.0000, 0.8318],
        [0.6840, 0.6723, 0.6623,  ..., 0.3650, 0.8318, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.32149259, 11.42420508, 11.48663767, 11.51557166, 11.70016217,
       11.72802491, 12.50996828, 13.32645552,  9.61942445, 12.88270018,
        8.41148722, 10.01804223,  9.34127013,  7.28793663,  5.43544626,
        4.90676242, 13.6660671 ,  4.90505566,  7.10378919,  5.41348121,
       11.89105259, 13.09737053,  9.19837792, 12.16901615,  6.65259775,
        9.62238828, 10.60226614, 11.00726483,  6.74686071, 10.06407839,
        5.52416368,  8.5724042 ,  6.29821673, 13.88683436,  6.38399789,
        9.3109884 ,  6.30647746,  4.91652556, 10.54658597, 10.14044489,
       11.65923396,  6.38968357,  6.19707944, 12.10222204, 13.02818702,
        8.28968067,  9.5642993 ,  9.99145096, 12.3828957 , 10.92597196,
       13.89974693, 13.3426541 ,  6.46356806,  7.76457095, 13.89886469,
        8.86082124,  6.31576264, 13.32557133, 13.03002338,  6.18668463,
       11.22142195,  8.50268322,  5.10444097, 12.00995898, 11.31276109,
        7.56617037,  9.15253344, 13.08414689, 12.5085995 ,  5.47866221,
       11.75807252, 10.91601655, 11.76830981,  7.54620849,  9.85330646,
        7.54127425,  6.00663314,  6.38968357,  5.91408504,  9.78286733,
       11.81423161, 13.1399181 , 11.80092578, 13.35832725, 13.86480175,
        7.73151309,  8.16714836, 11.99633819,  4.78544875, 13.59480906,
        9.14973319,  5.16611782,  6.67121748, 13.68672855,  9.562174  ,
       11.48663767,  9.79814997, 10.06407839, 11.38774149, 11.00812996,
        5.80860165,  6.19707944, 12.70489563, 13.77222752,  9.60359158,
        5.08910491])}
{'P': tensor([[5.5359e-03, 2.5946e-01, 2.4101e-01,  ..., 2.5619e-05, 2.2939e-06,
         2.4167e-06],
        [3.9784e-05, 8.3986e-03, 4.2094e-03,  ..., 1.3211e-05, 2.0231e-06,
         2.0694e-06],
        [5.3325e-05, 8.5070e-03, 1.0245e-02,  ..., 1.6316e-05, 2.5073e-06,
         2.5817e-06],
        ...,
        [5.4183e-02, 4.7139e-02, 8.7868e-02,  ..., 5.3400e-02, 1.2786e-04,
         1.1729e-04],
        [1.5738e-01, 7.2962e-02, 7.1871e-02,  ..., 6.5293e-02, 8.1915e-02,
         3.3126e-02],
        [1.0550e-01, 4.6996e-02, 5.7605e-02,  ..., 2.6195e-01, 1.8201e-02,
         6.6647e-02]]), 'X': tensor([[  3.8111, -10.6787,  -4.6539],
        [  5.4426, -10.1181,  -5.1268],
        [  5.3924, -10.1396,  -4.0863],
        [  6.9653,  -9.3225,  -3.4727],
        [  7.2340,  -9.1505,  -5.8321],
        [  5.7485,  -9.9811,  -3.2355],
        [ -5.1046,  -9.1610,   2.3418],
        [ 12.1379,   4.4125,  -3.8833],
        [  4.4290,  -3.4596,   1.4181],
        [  3.5939,   6.4029,  -1.8357],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.8719,   7.3713,  -3.7465],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.0087,   4.0173,   9.1091],
        [  5.6523,   3.9443,  -2.9449],
        [  4.7739,  -3.1218,  -4.1188],
        [  2.8474,   6.8979,   6.6457],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.8280,   4.2848,  -7.1955],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -7.8849,  -6.3115,  -4.7479],
        [  2.4149,  -4.6005,  -4.3339],
        [ 11.9019,  -3.1379,  -3.4534],
        [  3.9895,   6.0816,  -6.5235],
        [  0.6952,  -4.8060,   8.8976],
        [  7.8425,  -8.7202,   6.4530],
        [ -8.9379,  -4.1655,  -2.8114],
        [  9.3624,   9.5318,   0.2779],
        [  4.5491,  -3.3485,   6.8412],
        [ -8.8400,   2.3991,   1.3457],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.2242,  -0.3171,   7.4978],
        [  3.0719, -10.8449,  -4.1673],
        [ 10.9394,   7.3069,  -8.9365],
        [ -0.3432,   7.8900,   7.1805],
        [ -8.9636,   2.0648,   8.1946],
        [ -2.2188,   7.8218,   8.6064],
        [ 12.5789,   0.1722,   1.6470],
        [  4.4049,   5.6892,  -4.4394],
        [  2.6613,   7.0021,   4.5783],
        [  6.1685,  12.1979,  -6.2473],
        [ 10.9162,  -5.2599,  -7.0812],
        [  9.9217,  -6.7006,  -5.6728],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.2209,  -4.7092,   2.6117],
        [ 11.2209,  -4.7092,   2.6117],
        [ -5.9940,   6.1985,   5.9566],
        [  9.0129,  -7.7072,   9.1400],
        [  2.3373,  13.7721,  -0.2442],
        [  5.9587,  12.3216,   9.1695],
        [ -0.5385,   7.9037,   6.3830],
        [  5.1472,  -2.6801,   8.1194],
        [ -9.4645,  -1.4880,  -2.0226],
        [  3.9022,  13.2906,   5.5788],
        [  6.1685,  12.1979,  -6.2473],
        [  3.3839,  -4.1966,   8.3574],
        [  5.9003,  12.3550,   2.1917],
        [ -6.8618,  -7.6322,   3.7027],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.2589,  -1.9312,   9.8657],
        [  6.2247,  -0.3150,  -9.3657],
        [ -7.4602,   4.7718,  -4.6277],
        [  5.5158,  -2.1337,   3.5246],
        [  1.7500,   7.4187,   4.4237],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.1372,   2.6726,  -6.7885],
        [  4.0078,  -3.8018,  -6.2168],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.1531,  -5.8748,   2.6934],
        [  3.3744,   6.5623,  -6.3636],
        [  2.1824,  -4.6647,   1.1985],
        [  7.5064,  -8.9651,  -0.0994],
        [  1.0391,  -4.8141,  -0.0768],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.3339,   7.5627,  -0.9034],
        [ -6.4002,  -8.1048,  -0.9108],
        [  1.3670,  -4.7990,   3.2603],
        [  1.7218,  -4.7576,  -3.3617],
        [  9.8082,  -6.8396,   4.9615],
        [ 12.4691,   2.8525,  -5.5110],
        [ -7.7239,   4.4298,   3.1216],
        [  9.0652,   9.8597,  -3.5949],
        [  3.3167,   6.6022,   6.3554],
        [ -8.5900,   2.9767,   7.4811],
        [  5.8325,  -1.5222,  -2.1341],
        [  4.5491,  -3.3485,   6.8412],
        [  2.8474,   6.8979,   6.6457],
        [ -9.3419,   0.5972,  -1.2841],
        [ -8.0611,  -6.0302,   4.6962],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.4645,  -1.4880,  -2.0226],
        [  7.1697,  11.5343,  -0.7276],
        [ -5.2589,   6.7024,  -5.5809],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.2275,  -3.6320,  -0.2714],
        [  2.0547,   7.2957,  -1.9343],
        [ -8.1666,   3.7653,   3.6075],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.1047, -11.0046,   0.5188],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -4.4779,  -9.5609,  -2.4505]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1118, 0.1039,  ..., 0.4333, 0.3850, 0.4133],
        [0.1118, 0.0000, 0.0831,  ..., 0.5141, 0.4916, 0.5247],
        [0.1039, 0.0831, 0.0000,  ..., 0.4389, 0.4811, 0.4854],
        ...,
        [0.4333, 0.5141, 0.4389,  ..., 0.0000, 0.4631, 0.3091],
        [0.3850, 0.4916, 0.4811,  ..., 0.4631, 0.0000, 0.1810],
        [0.4133, 0.5247, 0.4854,  ..., 0.3091, 0.1810, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.33837396, 11.4890861 , 11.48434767, 11.63724792, 11.66453059,
       11.51811603, 10.48721382, 12.91505111,  5.62005629,  7.34250948,
        9.11473391,  9.75401449,  7.60529892, 10.37698435,  9.85330646,
        8.95983586,  6.89245352,  5.70405525,  7.46248947,  8.68979052,
        8.92395381,  9.50913849, 10.09979747,  5.19577146, 12.30858677,
        7.27340834,  4.85603886, 11.72802491,  9.86089579, 13.3607379 ,
        5.64865173,  9.15976746, 12.35421461,  9.09661212,  6.2322769 ,
       11.27160035, 13.15525803,  7.89745107,  9.19837792,  8.1303945 ,
       12.58006038,  7.19513806,  7.49076899, 13.66896739, 12.11735334,
       11.97238624,  5.04028538, 12.16901615, 12.16901615,  8.62260327,
       11.85890664, 13.96905756, 13.68672855,  7.92200342,  5.80312949,
        9.58072286, 13.85158417, 13.66896739,  5.39098669, 13.69162795,
       10.26327895, 13.04051744, 12.41012046,  6.23262289,  8.85572292,
        5.91408504,  7.62232307,  9.61942445,  9.43112957,  6.69389431,
        5.52416368,  8.40445872, 10.04916678,  7.37902766,  5.14999799,
       11.69264243,  4.92497514,  8.56889738,  9.75401449,  7.67939546,
       10.32715715,  4.98989906,  5.05963217, 11.95743558, 12.79126785,
        8.90405823, 13.39372426,  7.38844703,  9.09119351,  6.02789634,
        5.64865173,  7.46248947,  9.36093562, 10.06703358, 11.75618513,
        9.58072286, 13.58101794,  8.51928531, 11.66453059,  5.57341623,
        7.57946459,  8.99276564,  5.5366698 , 11.00508681, 10.54658597,
       10.5575601 ])}
{'P': tensor([[4.2044e-03, 1.9883e-01, 1.8323e-01,  ..., 1.2246e-04, 2.0887e-05,
         1.1542e-04],
        [2.3086e-05, 6.7765e-03, 1.2052e-02,  ..., 5.3667e-04, 6.0869e-05,
         8.7402e-05],
        [3.8356e-05, 4.3381e-03, 1.1193e-02,  ..., 9.4907e-04, 8.4752e-05,
         1.4890e-04],
        ...,
        [2.9973e-07, 4.1439e-07, 4.2301e-07,  ..., 1.4282e-02, 6.1440e-06,
         5.2960e-06],
        [1.9925e-07, 3.4107e-07, 3.1951e-07,  ..., 6.0018e-05, 1.4391e-02,
         5.6210e-07],
        [1.4726e-02, 1.9681e-02, 2.1964e-02,  ..., 2.9960e-03, 9.1363e-05,
         2.0518e-01]]), 'X': tensor([[  3.9591, -10.6391,  -6.2557],
        [  5.9843,  -9.8679,  -6.1449],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.3122,  -9.6994,  -6.3282],
        [  7.2340,  -9.1505,  -5.8321],
        [  5.4426, -10.1181,  -5.1268],
        [  2.5974,   7.0363,  -7.7810],
        [  4.3554,   5.7393,  -9.0766],
        [  4.2339,  -3.6268,   6.3125],
        [  6.0031,  12.2958,  -6.5992],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.0373,   7.1299,  -8.5895],
        [  8.5198,  10.4091,  -8.6486],
        [ -6.3489,   5.9117,   6.5161],
        [ 11.4703,  -4.2006,   7.4086],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -7.6719,  -6.6260,   7.1468],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.7845,   3.6652,   0.9175],
        [  6.3301,   0.3638,  -7.5685],
        [  1.9241,   7.3503,   7.9004],
        [ 12.0042,  -2.8350,   2.6864],
        [  6.2895,   0.0405,   5.7652],
        [  4.7544,  -3.1425,  -1.5315],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.0876,   6.7527,  -5.3540],
        [  5.6523,   3.9443,  -2.9449],
        [  4.2339,  -3.6268,   6.3125],
        [ -2.0721,   7.8432,   6.6380],
        [  1.9505,  -4.7171,   9.9868],
        [ -9.4630,  -1.5145,   0.2983],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.3900,   3.3727,   3.1650],
        [  1.3403,  -4.8011,  -1.1698],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -6.8466,  -7.6488,   1.9648],
        [  4.7544,  -3.1425,  -1.5315],
        [  3.2809,  13.5061,  -4.0199],
        [  9.5203,   9.3483,   5.8781],
        [ -6.4368,   5.8356,   3.3014],
        [ -4.7334,   6.9982,  -0.4392],
        [  2.0729,   7.2878,   7.0880],
        [  4.0457,  -3.7737,  -0.4704],
        [  8.5716,  10.3596,   6.8818],
        [ -6.6895,   5.6051,   2.7899],
        [ -0.0410,   7.8591,   9.3457],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.4739,  -1.2497,   0.5483],
        [ -8.1905,   3.7253,   4.1175],
        [  6.1973,  -0.4372,   6.5720],
        [  1.1644, -11.0394,   7.8511],
        [ 11.3691,   6.4760,   1.0787],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.2041,  -3.6509,  -4.9488],
        [ -4.1296,   7.2803,  -2.7194],
        [  5.5158,  -2.1337,   3.5246],
        [  2.0547,   7.2957,  -1.9343],
        [ -5.1398,  -9.1367,   6.2074],
        [ -7.8359,   4.2735,   7.2943],
        [  2.3589,  13.7668,  -3.6501],
        [ -3.3116, -10.1539,   7.5919],
        [  2.0243,   7.3086,   7.5432],
        [  4.4633,   5.6287,  -0.7754],
        [  5.9912,   3.1453,  -3.9989],
        [ -4.5778,  -9.5013,  -5.0667],
        [  7.4132,  11.3526,   0.3322],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.4235,   6.3594,   5.1663],
        [ -6.2668,   5.9808,   2.5614],
        [  5.5626,  -2.0536,   8.4433],
        [  6.2883,   0.0325,   5.2692],
        [  1.5653,  13.9377,   4.1773],
        [  2.5469,  13.7192,  -9.6935],
        [ -5.6169,  -8.7844,   8.9057],
        [  5.9843,  -9.8679,  -6.1449],
        [ -7.3470,  -7.0615,   5.0066],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.2437,  -4.6489,   6.1430],
        [  3.4009,  -4.1874,  -5.9751],
        [ -2.7572,   7.7194,   5.9580],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 10.6283,  -5.7253,   7.4878],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.3602,  -2.3808,   7.2098],
        [  2.3373,  13.7721,  -0.2442],
        [ -1.8823, -10.6558,  -3.8952],
        [  0.6363, -11.0348,   6.1414],
        [ 12.2589,  -1.9312,   9.8657],
        [ 10.9931,   7.2107,   7.7933],
        [  0.7685,  -4.8098,  -2.9310],
        [  8.7715,  -7.9390,   5.2900],
        [  4.0421,  -3.7765,  -8.3631],
        [ 11.9122,   5.1397,  -1.5498],
        [  2.1440, -10.9811,   8.5281],
        [  2.3376,  -4.6231,   9.8436],
        [  7.1441,  11.5529,   2.3232],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.6035,   0.7494,   7.1536],
        [  0.9967,  -4.8144,  -9.7370]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1000, 0.1183,  ..., 0.7218, 1.3011, 0.4031],
        [0.1000, 0.0000, 0.0250,  ..., 0.6243, 1.2535, 0.3038],
        [0.1183, 0.0250, 0.0000,  ..., 0.6038, 1.2629, 0.2906],
        ...,
        [0.7218, 0.6243, 0.6038,  ..., 0.0000, 1.2734, 0.3614],
        [1.3011, 1.2535, 1.2629,  ..., 1.2734, 0.0000, 1.1094],
        [0.4031, 0.3038, 0.2906,  ..., 0.3614, 1.1094, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.35182589, 11.54071859, 11.54672755, 11.57249984, 11.66453059,
       11.4890861 ,  7.50035355,  7.20482224,  5.5748624 , 13.6829843 ,
       12.04924159,  8.8353674 , 13.1399181 , 13.45124917,  8.67502348,
       12.21532282,  8.22728399,  9.4997987 , 10.1371552 ,  5.05981914,
        9.60412735, 11.59512194,  6.84796202,  6.34059021,  7.59795445,
       12.3344536 ,  6.28961988,  5.69913817,  8.22728399,  7.42512621,
        6.89245352,  5.5748624 ,  8.11227668,  5.10446653,  9.58347456,
        8.04664886,  9.04255655,  4.98461983,  9.70108409, 10.26545792,
        5.69913817, 13.89886469, 13.3426541 ,  8.68832398,  8.4486771 ,
        7.57685908,  5.53255882, 13.44593519,  8.7273569 ,  7.85919687,
       10.86693708,  9.55593374,  8.99791316,  6.21275016, 11.10066126,
       13.08414689, 10.31584102,  5.56809335,  8.36995953,  5.91408504,
        7.57946459, 10.48315436,  8.92549752, 13.96746816, 10.68031351,
        7.58378655,  7.18354885,  6.76661595, 10.54658597, 13.55868684,
       12.07818871, 13.0743399 ,  8.66273171,  5.92952538,  6.28835101,
       14.02532758, 13.95361098, 10.42667093, 11.54071859, 10.19036081,
       12.11735334,  6.1561646 ,  5.16205087,  5.39450305,  8.19703512,
       13.96905756, 12.86648771, 12.4013153 , 12.07225469, 12.00995898,
        5.86519525, 13.96905756, 10.8207297 , 11.05317597, 12.41012046,
       13.14690137,  4.87083348, 11.83075005,  5.53175275, 12.97370371,
       11.18839169,  5.1805345 , 13.58333857, 12.889695  , 12.62576026,
        4.91652556])}
{'P': tensor([[4.3230e-03, 3.2762e-01, 2.3291e-01,  ..., 1.8849e-05, 2.5424e-06,
         3.1549e-06],
        [5.4582e-05, 5.8610e-03, 3.3061e-01,  ..., 2.5402e-05, 3.0766e-06,
         4.1857e-06],
        [2.3266e-05, 3.0040e-05, 5.6280e-03,  ..., 3.2703e-05, 2.6441e-06,
         5.1865e-06],
        ...,
        [1.0573e-05, 1.0786e-05, 1.4423e-05,  ..., 7.2380e-02, 4.8005e-04,
         9.9366e-03],
        [1.5467e-04, 1.4968e-04, 1.8416e-04,  ..., 1.5290e-02, 1.1009e-01,
         1.6751e-02],
        [9.2618e-06, 9.3281e-06, 1.2455e-05,  ..., 3.3636e-02, 1.4159e-03,
         2.0444e-01]]), 'X': tensor([[  3.9770, -10.6341,   8.0783],
        [  4.3523, -10.5232,   8.2138],
        [  5.6414, -10.0303,   7.8525],
        [  6.8093,  -9.4178,   8.4244],
        [  6.8704,  -9.3809,   7.5769],
        [  7.0664,  -9.2590,   8.0879],
        [  8.0381,  -8.5690,  -8.9523],
        [  6.1066,  -0.7799,  -4.1746],
        [  9.6662,   9.1725,   8.1056],
        [  3.2990,  -4.2415,  -7.4857],
        [ -9.4196,   0.0478,   9.4140],
        [ 10.9599,  -5.1849,  -3.5526],
        [  2.1625,  -4.6697,   7.0658],
        [ 12.3759,  -1.3943,   6.4659],
        [ -9.0604,   1.7707,   1.1066],
        [  6.9284,  11.7061,  -0.3732],
        [ -2.5683, -10.4433,  -2.9182],
        [ 12.5123,  -0.5339,  -1.0728],
        [  2.8890,  -4.4316,   1.1600],
        [  1.1006,   7.6318,  -2.4877],
        [  0.1931,  -4.7491,   7.4385],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.3384,  -2.6161,  -4.7630],
        [  6.7565,  11.8237,  -4.9882],
        [ -1.6334, -10.7207,   7.1397],
        [  2.1379,  -4.6757,   6.4981],
        [  0.1677,  -4.7447,  -6.1529],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.3087,   0.7792,  -7.5587],
        [ -3.4986,   7.5152,   9.0712],
        [ -0.9715, -10.8630,   0.3555],
        [  6.3246,   1.6696,  -0.1544],
        [  4.7482, -10.3907,   5.9149],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.1497,   7.8334,  -5.4412],
        [  2.1477,   7.2550,   8.9688],
        [  6.3534,   1.2990,  -7.2942],
        [  8.5198,  10.4091,  -8.6486],
        [ 10.9717,  -5.1646,  -6.9039],
        [  4.4915,   5.5990,   7.0405],
        [ -7.7239,   4.4298,   3.1216],
        [ -4.3359,  -9.6431,  -9.0616],
        [  0.6470, -11.0352,  -1.1931],
        [  7.9403,  -8.6455,  -8.7433],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.1973,  -0.4372,   6.5720],
        [ -2.2188,   7.8218,   8.6064],
        [  6.2576,   2.1359,  -3.3398],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.8902,  -1.3887,   4.0388],
        [ -9.3419,   0.5972,  -1.2841],
        [ -1.3940, -10.7772,   6.7835],
        [  5.6260,  -1.9401,   3.0259],
        [  6.7792,  11.8084,   1.6251],
        [ 12.1588,   4.3364,  -8.0867],
        [  1.7500,   7.4187,   4.4237],
        [ -9.4283,  -1.9585,  -0.9129],
        [  7.0171,  11.6439,  -5.0178],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.2419,  -0.2313,   4.2440],
        [  5.1519, -10.2387,   9.8191],
        [  9.1409,  -7.5789,   1.5814],
        [  5.6523,   3.9443,  -2.9449],
        [ -0.1963,   7.8765,  -1.1371],
        [ -2.9497, -10.3028,  -7.3479],
        [  1.7815,  -4.7481,   4.4289],
        [ 11.2939,   6.6325,  -5.8311],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.2612, -10.5514,  -6.6268],
        [ -7.8359,   4.2735,   7.2943],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.3076,   0.7852,  -6.7010],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.1460,   7.8340,  -9.4614],
        [  6.2036,   2.4030,  -1.2500],
        [  5.0277,   4.9589,   7.0175],
        [ 12.0042,  -2.8350,   2.6864],
        [ -4.6942,  -9.4299,  -9.1846],
        [  3.9591, -10.6391,  -6.2557],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -2.1382,   7.8339,   7.3524],
        [  0.5175,  14.0910,   4.6540],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 12.4837,   2.7510,   0.4449],
        [  0.9945,  -4.8144,  -0.1897],
        [ 12.5919,   0.4035,  -5.8460],
        [  4.5954,  13.0099,  -4.4439],
        [  7.7794,  11.0629,   8.9069],
        [  9.4554,   9.4245,   9.8800],
        [ -7.0253,  -7.4493,  -1.5978],
        [  2.1771, -10.9776,  -3.2007],
        [  0.7958,  -4.8109,  -8.2248],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.3470,  -7.0615,   5.0066],
        [  9.4925,   9.3811,   8.9842],
        [  5.1196,  -2.7160,   4.2705],
        [ -5.1398,  -9.1367,   6.2074],
        [ 11.1420,   6.9331,  -2.4779],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.3587,   1.1596,  -3.6629],
        [  0.2571,  -4.7594,   1.5618],
        [  6.3236,   0.3011,  -4.6668],
        [  0.1532,  -4.7422,  -7.0145],
        [  6.3048,   0.1469,  -9.1691]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0284, 0.0695,  ..., 1.0668, 1.0793, 1.2035],
        [0.0284, 0.0000, 0.0425,  ..., 1.0597, 1.0758, 1.1977],
        [0.0695, 0.0425, 0.0000,  ..., 1.0346, 1.0553, 1.1741],
        ...,
        [1.0668, 1.0597, 1.0346,  ..., 0.0000, 0.1388, 0.1468],
        [1.0793, 1.0758, 1.0553,  ..., 0.1388, 0.0000, 0.1552],
        [1.2035, 1.1977, 1.1741,  ..., 0.1468, 0.1552, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.35345345, 11.38774149, 11.50791859, 11.62158554, 11.62770332,
       11.64746882, 11.74901778,  6.1561646 , 13.32557133,  5.37343173,
        9.41970031, 12.12448912,  5.14607014, 12.45418236,  9.23178204,
       13.60275211, 10.75443106, 12.52372629,  5.29009836,  7.71075226,
        4.75303127,  9.17595899,  9.69792513, 13.61801526, 10.84437705,
        5.14124612,  4.74770899,  9.10945678,  9.34127013,  8.28968067,
       10.90638066,  6.54127431, 11.42420508,  5.57341623,  7.83486943,
        7.56617037,  6.48486171, 13.45124917, 12.12641969,  7.17789666,
        8.90405823, 10.57302546, 11.05413545, 11.73848065, 13.09737053,
        6.21275016,  8.1303945 ,  6.61211224,  5.79391825,  6.05165248,
        9.36093562, 10.86693708,  5.95111804, 13.61600536, 12.9089559 ,
        7.62232307,  9.62958973, 13.59480906, 12.18998195,  6.24614905,
       11.46176259, 11.87412439,  6.89245352,  7.8788965 , 10.7167285 ,
        5.0713364 , 13.09737053,  5.18047638, 11.37939695,  8.92549752,
       12.11735334,  9.34061938, 12.30695098,  7.83534987,  6.652742  ,
        7.06169391, 12.3344536 , 10.53368644, 11.35182589, 10.00794468,
        9.88310614,  9.96152788,  8.12043983, 14.10045797, 10.14436912,
       12.78327037,  4.9160962 , 12.59840193, 13.79762415, 13.52430176,
       13.35013374, 10.23946247, 11.1913598 ,  4.8763135 ,  8.92549752,
       10.19036081, 13.34587027,  5.79543038, 10.48315436, 13.12299145,
       12.41366754,  6.46356806,  4.7663526 ,  6.33075682,  4.74468406,
        6.30647746])}
{'P': tensor([[4.0944e-03, 2.6538e-01, 2.3495e-01,  ..., 1.7554e-03, 7.7025e-04,
         1.2345e-07],
        [3.2243e-05, 7.0255e-03, 6.0589e-05,  ..., 4.3226e-03, 1.6888e-03,
         1.9915e-07],
        [4.8745e-05, 3.2714e-01, 5.4303e-03,  ..., 3.2549e-03, 1.1086e-03,
         1.6556e-07],
        ...,
        [3.8252e-06, 6.6721e-06, 5.0559e-06,  ..., 1.6475e-02, 4.1636e-04,
         1.2624e-06],
        [3.6795e-04, 4.9401e-04, 4.1945e-04,  ..., 1.8307e-03, 2.9624e-01,
         9.6858e-05],
        [2.6718e-05, 1.7107e-05, 2.2859e-05,  ..., 4.4516e-06, 7.3270e-06,
         2.4837e-02]]), 'X': tensor([[  4.0587, -10.6112,   6.2273],
        [  5.4167, -10.1293,   6.2156],
        [  4.7482, -10.3907,   5.9149],
        [  5.7218,  -9.9934,   6.5764],
        [  7.5784,  -8.9141,   5.8410],
        [  7.8425,  -8.7202,   6.4530],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.5383, -11.0274,  -4.9666],
        [  4.5491,  -3.3485,   6.8412],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.3505,   0.6235,  -8.5849],
        [  0.9809,   7.6642,  -1.4217],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.1085,  -4.9204,   0.0736],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.9410,   6.8428,   5.4458],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.9936,  -2.8677,   6.0614],
        [  0.9809,  -4.8145,   6.8706],
        [  8.0877,  -8.5297,  -1.1050],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.3038,   7.8092,   0.4164],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.0617,  12.7957,   8.3544],
        [  6.2026,  12.1774,  -1.5887],
        [ -3.9456,  -9.8542,   0.7466],
        [  3.8182,  -3.9347,  -1.7961],
        [  2.7558,  -4.4844,   0.0636],
        [ -0.5739,   7.9056,   0.6382],
        [  4.5800,   5.5034,  -3.9432],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -3.0004,   7.6607,  -7.2640],
        [  9.0652,   9.8597,  -3.5949],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.1886,  -2.6250,   6.4677],
        [ -9.4766,  -1.1116,  -2.1706],
        [  3.6154,  -4.0640,  -7.9698],
        [  5.0506,   4.9277,  -9.5094],
        [  7.5930,  -8.9037,   7.2323],
        [  6.3604,   0.9042,  -6.5901],
        [  0.2480, -11.0153,  -3.0183],
        [  2.2659,  13.7894,   6.5436],
        [  4.0591,   6.0202,  -1.6339],
        [  3.9663,   6.1018,  -5.4104],
        [ -0.5385,   7.9037,   6.3830],
        [  2.0899,  -4.6870,  -7.4260],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.9942,  -3.9990,   6.0448],
        [  2.5128, -10.9365,   9.9645],
        [ -9.3076,   0.7852,  -6.7010],
        [ 12.5224,  -0.4497,   4.8448],
        [ -1.5261,   7.8988,   9.3768],
        [  7.1441,  11.5529,   2.3232],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -3.5173, -10.0623,   4.8280],
        [ -7.4733,   4.7555,   6.2490],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.1795,  -3.3623,   8.5259],
        [  5.7963,  -1.6018,   1.8856],
        [  3.2159, -10.8166,  -1.0406],
        [  1.7659,   7.4127,   4.7871],
        [ 11.2209,  -4.7092,   2.6117],
        [ -9.4524,  -1.6813,  -6.6862],
        [  5.4867,  12.5816,   2.9549],
        [ 11.0373,   7.1299,  -8.5895],
        [ 12.5123,  -0.5339,  -1.0728],
        [  0.1252, -11.0062,   7.2564],
        [ 12.2692,  -1.8882,   3.9253],
        [  1.7500,   7.4187,   4.4237],
        [  7.9816,  10.8938,  -0.5378],
        [ -1.5261,   7.8988,   9.3768],
        [  3.1756, -10.8247,   6.8882],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.4430,  -0.1897,   4.8383],
        [  8.3791,  10.5409,   5.5590],
        [  4.6357,   5.4415,   7.9067],
        [  2.3373,  13.7721,  -0.2442],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.9392,  -4.8143,  -3.4649],
        [  0.9967,  -4.8144,  -9.7370],
        [ 11.2571,   6.7072,   8.3693],
        [  1.5662,  13.9375,   4.5142],
        [  4.5340,  -3.3629,   8.9652],
        [ -3.4986,   7.5152,   9.0712],
        [  5.8979,   3.3958,  -3.8713],
        [  1.7232,  -4.7574,   3.1270],
        [  3.1064,  13.5607,   6.7605],
        [ -4.0868,   7.2981,   8.1874],
        [  5.1196,  -2.7160,   4.2705],
        [  2.2860,   7.1916,  -6.1011],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.3122,  -9.6994,  -6.3282],
        [ -2.6485, -10.4151,   0.8766],
        [  7.9141,  -8.6656,   9.6948],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.3739,   0.3982,  -4.2564],
        [  7.5309,  11.2617,   7.2656],
        [ 11.1085,  -4.9204,   0.0736],
        [  6.0054,  -1.0902,   8.3079],
        [ -8.9788,   2.0208,  -3.2867]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.4269e-02, 3.9357e-02,  ..., 7.2671e-01, 5.7105e-01,
         8.7886e-01],
        [7.4269e-02, 0.0000e+00, 4.0298e-02,  ..., 6.8490e-01, 4.9712e-01,
         9.0637e-01],
        [3.9357e-02, 4.0298e-02, 0.0000e+00,  ..., 6.9358e-01, 5.3340e-01,
         8.7840e-01],
        ...,
        [7.2671e-01, 6.8490e-01, 6.9358e-01,  ..., 0.0000e+00, 5.4843e-01,
         7.6834e-01],
        [5.7105e-01, 4.9712e-01, 5.3340e-01,  ..., 5.4843e-01, 2.1453e-04,
         1.1611e+00],
        [8.7886e-01, 9.0637e-01, 8.7840e-01,  ..., 7.6834e-01, 1.1611e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.36089355, 11.48663767, 11.42420508, 11.51557166, 11.70016217,
       11.72802491, 13.59480906,  9.17905491,  5.05444288,  9.17905491,
       11.1341798 ,  5.64865173, 10.33712258,  6.38105015,  7.72668467,
       12.1494023 , 12.1494023 ,  7.8575648 ,  9.93553424,  7.44805901,
        5.05963217, 12.33167382,  4.91336994, 11.75438006,  5.57341623,
        7.81509631, 13.03002338, 13.76047999, 13.6660671 , 10.61472636,
        5.48276038,  5.26342203,  7.92644746,  7.15990676, 13.66051884,
       12.30858677,  8.22728399, 13.39372426,  5.21280156,  5.81481044,
        9.54154096,  5.43943791,  7.05627461, 11.70168755,  6.42439344,
       11.018084  , 13.97430191,  7.26074925,  7.27759233,  7.92200342,
        5.13182506, 12.42554808,  9.84314712, 11.22142195,  9.34061938,
       12.53047489,  8.04483495, 13.58333857, 12.12448912, 10.65929968,
        8.8580671 ,  6.60752373,  9.77588801,  6.01356004, 11.2845608 ,
        7.62011734, 12.16901615,  9.60080639, 13.72594022, 13.1399181 ,
       12.52372629, 11.00694688, 12.41366754,  7.62232307, 13.50484539,
        8.04483495, 11.28093507,  5.70405525,  9.44486609, 13.46553623,
        7.14837514, 13.96905756,  4.85995002,  4.90505566,  4.91652556,
       13.103709  , 14.02526505,  5.64500512,  8.28968067,  6.80559669,
        5.05989547, 13.91198191,  8.36445758,  5.79543038,  7.54620849,
        9.58072286, 12.40074901, 11.57249984, 10.74655346, 11.73567759,
        9.36378438,  9.38232121, 13.54774355, 12.1494023 ,  6.10360073,
        9.20340031])}
{'P': tensor([[4.6739e-03, 2.4926e-01, 2.1470e-01,  ..., 1.0469e-05, 3.5642e-03,
         2.0050e-02],
        [3.5491e-05, 9.5501e-03, 3.7015e-03,  ..., 4.7470e-05, 2.4415e-03,
         1.1388e-02],
        [2.7351e-05, 1.1196e-02, 6.2951e-03,  ..., 3.2557e-05, 2.2088e-03,
         1.2173e-02],
        ...,
        [5.2584e-07, 5.8596e-07, 6.0149e-07,  ..., 6.4645e-02, 7.8549e-07,
         7.5672e-07],
        [1.4120e-04, 7.5846e-04, 7.8265e-04,  ..., 2.3084e-04, 1.8297e-02,
         7.9217e-03],
        [1.6225e-04, 1.4124e-03, 1.2031e-03,  ..., 3.0811e-04, 9.6691e-03,
         2.6654e-02]]), 'X': tensor([[  4.2612, -10.5514,  -6.6268],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.9843,  -9.8679,  -6.1449],
        [  6.3122,  -9.6994,  -6.3282],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.5375,   7.0676,  -4.3490],
        [ -8.3900,   3.3727,   3.1650],
        [ 11.5966,   5.9666,   6.0083],
        [ -1.5875,   7.8944,  -9.4286],
        [  8.9284,  10.0036,  -6.9210],
        [ -3.9508,  -9.8515,   1.7923],
        [ 11.7795,  -3.4697,   3.5624],
        [  2.9461,  -4.4077,  -0.4831],
        [  5.0348,  -2.8229,   3.5436],
        [  8.7715,  -7.9390,   5.2900],
        [  5.5424,  -2.0885,   7.2487],
        [ 12.5763,   0.1333,  -9.2650],
        [  2.4149,  -4.6005,  -4.3339],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.1459,   1.4780,  -4.2403],
        [ 11.2517,  -4.6495,   2.9800],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.0825,   5.2073,  -5.0497],
        [  8.5198,  10.4091,  -8.6486],
        [ -9.4671,  -0.5553,   0.1117],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.3373,  -4.6232,  -5.2246],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.1771, -10.9776,  -3.2007],
        [  6.3583,   1.1729,   8.7835],
        [ 10.8670,  -5.3428,  -2.8637],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.8407,  11.7666,   3.0045],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.4887,  -5.2448,  -0.6585],
        [  2.5382,   7.0672,   4.5097],
        [  6.5428,  -9.5728,  -2.9897],
        [  0.9392,  -4.8143,  -3.4649],
        [ 10.9916,   7.2133,  -0.9179],
        [  2.0449,  13.8402,  -7.1642],
        [  6.3551,   0.7149,   1.3344],
        [  2.9470,  13.6085,   2.9809],
        [  5.0277,   4.9589,   7.0175],
        [  6.2902,   0.0451,   3.7116],
        [  0.5960, -11.0335,   3.1144],
        [  0.5600, -11.0321,  -0.8660],
        [  1.1350,  14.0104,   0.6321],
        [ -3.9700,   7.3453,  -4.1486],
        [ -8.4573,   3.2448,   5.4370],
        [  2.5375,   7.0676,  -4.3490],
        [  1.7000,  -4.7609,   2.7935],
        [  2.5813,  -4.5471,   5.3597],
        [ -8.9178,   2.1936,   7.5388],
        [ -6.8466,  -7.6488,   1.9648],
        [ 10.5057,  -5.9099,  -5.1599],
        [  1.9204,   7.3518,   8.4781],
        [  0.1041, -11.0045,   2.4311],
        [  5.2199,   4.6861,  -1.9304],
        [ -3.2736, -10.1703,   3.8712],
        [ 12.4691,   2.8525,  -5.5110],
        [  6.0047,   3.1062,  -6.7914],
        [ 12.2921,  -1.7902,  -4.0870],
        [  7.8375,  -8.7240,  -9.8913],
        [  6.2555,   2.1474,   4.1685],
        [ 11.0275,  -5.0666,   8.4154],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.3604,   0.9042,  -6.5901],
        [  3.8182,  -3.9347,  -1.7961],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.2630,   2.1057,  -5.2136],
        [ -1.1885, -10.8211,   0.0865],
        [ 11.7162,   5.6728,  -8.9372],
        [ -8.0588,  -6.0341,  -2.3006],
        [  2.5469,  13.7192,  -9.6935],
        [ -6.4464,   5.8272,   7.5038],
        [  7.1821,  -9.1845,   9.8548],
        [ 11.8951,  -3.1570,   7.0742],
        [ 10.8218,  -5.4177,   6.8573],
        [ -6.7992,   5.4991,   9.9186],
        [  7.5064,  -8.9651,  -0.0994],
        [ -8.9282,   2.1648,   8.4894],
        [  0.8940, -11.0402,  -0.9783],
        [ -9.1749,  -3.3803,   6.4629],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.3236,   0.3011,  -4.6668],
        [  6.2247,  -0.3150,  -9.3657],
        [  5.6523,   3.9443,  -2.9449],
        [  3.0719, -10.8449,  -4.1673],
        [ -1.6086,   7.8928,   4.3385],
        [ 11.8951,  -3.1570,   7.0742],
        [  1.0000,   7.6591,  -4.5011],
        [  5.4179,  -2.2925,  -3.4641],
        [ -5.1974,   6.7396,   3.3484],
        [  3.7916,  -3.9524,  -7.9547],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -9.2974,   0.8371,   2.4852],
        [ -2.0721,   7.8432,   6.6380],
        [  0.9392,  -4.8143,  -3.4649],
        [ -7.4807,   4.7463,   2.0616],
        [  7.8982,  -8.6778,   0.0239],
        [  6.2883,   0.0325,   5.2692],
        [  9.7328,   9.0902,  -0.8917],
        [  6.9314,  -9.3435,  -1.2729],
        [  6.9653,  -9.3225,  -3.4727]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1051, 0.0909,  ..., 0.8370, 0.4054, 0.2529],
        [0.1051, 0.0000, 0.0250,  ..., 0.7439, 0.3801, 0.2031],
        [0.0909, 0.0250, 0.0000,  ..., 0.7490, 0.3635, 0.1903],
        ...,
        [0.8370, 0.7439, 0.7490,  ..., 0.0000, 0.6063, 0.6163],
        [0.4054, 0.3801, 0.3635,  ..., 0.6063, 0.0000, 0.1831],
        [0.2529, 0.2031, 0.1903,  ..., 0.6163, 0.1831, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.37939695, 11.54672755, 11.54071859, 11.57249984, 11.66453059,
       11.63765273,  7.5092834 ,  9.04255655, 13.04156075,  8.05242591,
       13.40850328, 10.61417181, 12.27991991,  5.30158497,  5.77219077,
       11.83075005,  5.92281142, 12.5769659 ,  5.19577146, 10.1979669 ,
        9.26455795, 12.17451685,  8.95133961,  8.7907923 , 13.45124917,
        9.48337063, 11.74901778,  5.18047638, 12.25806559, 11.1913598 ,
        6.46560181, 12.10942412, 11.69264243, 13.6105595 ,  7.822854  ,
        9.97822473,  7.50917551, 11.59512194,  4.90505566, 13.1471313 ,
       13.99048094,  6.39520397, 13.92390689,  7.06169391,  6.29034958,
       11.04953767, 11.04629471, 14.05633441,  8.34949383,  9.05843201,
        7.5092834 ,  5.05534609,  5.2287058 ,  9.18358562, 10.26545792,
       12.05395056,  7.5984809 , 11.0050379 ,  7.01475376, 10.68416401,
       12.79126785,  6.76056625, 12.4217523 , 11.72749923,  6.61386436,
       12.13568598, 12.50996828,  6.42439344,  5.48276038,  5.18047638,
        6.60752373, 10.88618114, 13.01728313, 10.0674887 , 13.95361098,
        8.68979052, 11.65923396, 12.30695098, 12.10222204,  8.74470218,
       11.69264243,  9.18690018, 11.07637531,  9.77777346,  7.48311002,
        9.36093562,  6.33075682,  6.23262289,  6.89245352, 11.27160035,
        8.0550286 , 12.30695098,  7.72415048,  5.8828986 ,  8.51088913,
        5.47702766, 10.27716434,  9.3349858 ,  8.11227668,  4.90505566,
        8.85939626, 11.73397309,  6.28835101, 13.31764181, 11.63383221,
       11.63724792])}
{'P': tensor([[3.7399e-03, 2.1096e-01, 1.4203e-01,  ..., 7.9407e-05, 4.0622e-05,
         9.7633e-06],
        [2.1227e-05, 3.9768e-03, 1.8409e-01,  ..., 1.5376e-04, 6.4247e-05,
         1.4925e-05],
        [4.2430e-05, 1.0091e-04, 1.5655e-02,  ..., 7.4170e-04, 3.5785e-04,
         8.0989e-05],
        ...,
        [5.7902e-07, 8.2415e-07, 1.0857e-06,  ..., 2.3953e-02, 1.3303e-03,
         1.0220e-03],
        [7.7449e-06, 8.0514e-06, 8.4514e-06,  ..., 4.0512e-05, 4.0207e-02,
         2.2081e-01],
        [2.2678e-05, 2.1294e-05, 2.0149e-05,  ..., 3.8799e-05, 2.3599e-03,
         1.5395e-01]]), 'X': tensor([[  4.3523, -10.5232,   8.2138],
        [  5.6414, -10.0303,   7.8525],
        [  6.8093,  -9.4178,   8.4244],
        [  7.0664,  -9.2590,   8.0879],
        [  6.8704,  -9.3809,   7.5769],
        [  7.5930,  -8.9037,   7.2323],
        [ -6.1899,  -8.3011,  -7.9269],
        [  5.3924, -10.1396,  -4.0863],
        [  1.7218,  -4.7576,  -3.3617],
        [  2.7295,  13.6703,  -8.9328],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.4159,  -2.0739,   9.6841],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.9026,   3.3837,  -0.6573],
        [ -8.6229,   2.9068,   2.4429],
        [ 12.3609,   3.4769,  -8.5762],
        [  0.1677,  -4.7447,  -6.1529],
        [  1.4150,  13.9647,  -4.0915],
        [  3.8111, -10.6787,  -4.6539],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.0617,  12.7957,   8.3544],
        [  3.6817,   6.3354,   6.1233],
        [  6.3583,   1.1729,   8.7835],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.6422,   7.4586,  -6.8056],
        [ -1.9646,   7.8571,   3.3817],
        [ 12.5123,  -0.5339,  -1.0728],
        [  2.5469,  13.7192,  -9.6935],
        [  7.5784,  -8.9141,   5.8410],
        [  7.5064,  -8.9651,  -0.0994],
        [ -3.4100,   7.5436,  -7.6980],
        [  3.2990,  -4.2415,  -7.4857],
        [  9.2692,   9.6370,   7.0746],
        [  6.1973,  -0.4372,   6.5720],
        [ -3.0008, -10.2827,  -8.9480],
        [ -5.9987,   6.1949,  -6.2114],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.3551,   0.7149,   1.3344],
        [  4.2339,  -3.6268,   6.3125],
        [  6.2883,   0.0325,   5.2692],
        [ 12.0845,  -2.5772,  -4.6073],
        [  0.7275,  14.0667,  -0.9815],
        [  8.6420,  -8.0580,   6.5781],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.4167, -10.1293,   6.2156],
        [ -6.4620,  -8.0449,   6.2719],
        [ -7.3554,   4.8985,  -0.6665],
        [ -0.8569,   7.9156,  -6.5557],
        [  5.5101,  -2.1433,   9.5442],
        [ -5.1377,   6.7751,  -0.4438],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.8745,   2.3097,   3.5600],
        [  5.6414, -10.0303,   7.8525],
        [  2.4932,   7.0903,  -5.4214],
        [  6.9693,  -9.3200,  -8.3822],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.9312,  13.8649,   8.3563],
        [  2.3200,   7.1756,   1.1938],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -4.6393,   7.0461,   1.1624],
        [ -6.7992,   5.4991,   9.9186],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -0.8885,   7.9160,   0.0571],
        [  4.2041,  -3.6509,  -4.9488],
        [  2.2849, -10.9655,  -7.6542],
        [ -7.6525,  -6.6534,   5.1721],
        [ -4.4503,   7.1378,   3.3751],
        [  3.3094,  13.4969,  -3.6507],
        [  5.2199,   4.6861,  -1.9304],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.2076,  13.8032,  -1.5099],
        [  6.3119,   1.7797,   9.9186],
        [ -8.9178,   2.1936,   7.5388],
        [  6.2902,   0.0451,   3.7116],
        [ -1.8788, -10.6567,   3.7068],
        [  6.0054,  -1.0902,   8.3079],
        [ -7.0120,   5.2825,   7.7762],
        [  5.4732,  -2.2041,  -4.3830],
        [ -9.0961,   1.6529,   3.1553],
        [ -0.9715, -10.8630,   0.3555],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -1.5261,   7.8988,   9.3768],
        [ 12.5650,   2.0132,   4.6272],
        [ -7.8280,   4.2848,  -7.1955],
        [  7.8375,  -8.7240,  -9.8913],
        [ -3.9858,  -9.8334,  -3.7323],
        [ 11.0438,   7.1180,  -5.6991],
        [  3.6817,   6.3354,   6.1233],
        [  0.6142,  14.0802,   0.5407],
        [  3.9022,  13.2906,   5.5788],
        [ 10.5678,   7.9261,  -0.6100],
        [ -9.4756,  -0.8166,  -9.6021],
        [  7.3724,  -9.0577,  -8.9843],
        [ -0.8675, -10.8815,   6.8561],
        [  2.6321,  -4.5296,  -2.5159],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3305,   1.6119,   1.1538],
        [ -7.2986,  -7.1225,  -3.0814],
        [  3.3069,   6.6089,   8.7377],
        [ -7.5603,   4.6459,   8.2002],
        [ -6.5295,  -7.9785,  -7.4429],
        [  2.1120,  -4.6818,   8.6144],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.0277,   4.9589,   7.0175],
        [  2.8474,   6.8979,   6.6457]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.2465e-02, 1.2059e-01,  ..., 9.7360e-01, 4.7625e-01,
         3.9214e-01],
        [4.2465e-02, 0.0000e+00, 7.9703e-02,  ..., 9.4237e-01, 4.3469e-01,
         3.5162e-01],
        [1.2059e-01, 7.9703e-02, 0.0000e+00,  ..., 9.0882e-01, 3.6930e-01,
         2.9273e-01],
        ...,
        [9.7360e-01, 9.4237e-01, 9.0882e-01,  ..., 1.5170e-04, 6.1301e-01,
         6.4154e-01],
        [4.7625e-01, 4.3469e-01, 3.6930e-01,  ..., 6.1301e-01, 0.0000e+00,
         9.2712e-02],
        [3.9214e-01, 3.5162e-01, 2.9273e-01,  ..., 6.4154e-01, 9.2712e-02,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.38774149, 11.50791859, 11.62158554, 11.64746882, 11.62770332,
       11.70168755, 10.35485331, 11.48434767,  5.05963217, 13.94008948,
        6.61211224,  9.64157461, 12.64235359,  6.80370254,  9.09963894,
       12.84056604,  4.74770899, 14.03618649, 11.33837396, 12.07225469,
       13.76047999,  7.32754882,  6.46560181, 13.20988996,  7.63726134,
        8.09899874, 12.52372629, 13.95361098, 11.70016217, 11.69264243,
        8.27853076,  5.37343173, 13.37121923,  6.21275016, 10.71162681,
        8.62329466,  9.75200848,  6.39520397,  5.5748624 ,  6.28835101,
       12.35625694, 14.08549802, 11.8159254 ,  5.70405525, 11.48663767,
       10.31886066,  8.83725081,  7.96181288,  5.91222346,  8.50278896,
       13.12299145,  9.17016857, 11.50791859,  7.5158489 , 11.63765273,
       10.06850677, 13.99877413,  7.54127425,  9.5014146 ,  8.43626114,
        8.74470218, 12.53345445,  7.96575782,  5.56809335, 11.20100795,
       10.14044489,  8.41148722, 13.89671314,  7.01475376, 12.00995898,
       13.97857591,  6.55800267,  9.18358562,  6.29034958, 10.82106505,
        6.10360073,  8.77913428,  5.90035324,  9.24501829, 10.90638066,
       12.05300943,  8.04483495, 12.72524615,  8.92395381, 11.72749923,
       10.61047596, 13.13888841,  7.32754882, 14.09357547, 13.85158417,
       13.20988996,  9.51074763, 11.67875855, 10.91601655,  5.23878371,
       12.00995898,  6.27905292,  6.53250937, 10.1979669 ,  7.39004226,
        8.8737556 , 10.30972193,  5.13616224, 12.67680439,  7.06169391,
        7.46248947])}
{'P': tensor([[1.0125e-02, 4.8054e-01, 1.1106e-01,  ..., 6.5340e-04, 3.7059e-06,
         7.4629e-04],
        [1.1125e-04, 1.8067e-02, 1.5994e-02,  ..., 2.3654e-03, 7.5037e-06,
         2.0032e-03],
        [1.1316e-04, 1.6283e-03, 9.8353e-03,  ..., 8.4330e-04, 8.0739e-06,
         6.7546e-04],
        ...,
        [6.6726e-07, 9.0563e-07, 8.6973e-07,  ..., 2.0479e-02, 2.0925e-05,
         1.1609e-03],
        [4.8764e-06, 4.0948e-06, 6.2767e-06,  ..., 3.8457e-06, 1.2553e-01,
         2.5663e-05],
        [3.8480e-05, 4.9081e-05, 4.3601e-05,  ..., 5.6381e-03, 1.1395e-04,
         6.9713e-02]]), 'X': tensor([[  4.5814, -10.4485,   2.5496],
        [  5.8450,  -9.9356,   3.0767],
        [  6.1989,  -9.7591,   0.8289],
        [  9.1409,  -7.5789,   1.5814],
        [  9.5261,  -7.1675,   1.4206],
        [  8.8091,  -7.9037,   0.3631],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.3076,   0.7852,  -6.7010],
        [ -8.6603,   2.8250,   5.9181],
        [  4.5503,   5.5359,   7.4735],
        [  6.2365,  -0.2578,  -6.0212],
        [  3.7298,  13.3537,  -9.4653],
        [  2.6310, -10.9195,   3.8054],
        [ 11.4235,   6.3594,   5.1663],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.4824,  -5.2575,  -0.0795],
        [  5.6248,   3.9984,   8.4933],
        [  6.3534,   1.2990,  -7.2942],
        [  1.3403,  -4.8011,  -1.1698],
        [  6.1973,  -0.4372,   6.5720],
        [  0.1532,  -4.7422,  -7.0145],
        [  1.9503,  -4.7172,  -6.2313],
        [  3.5284, -10.7484,   5.3347],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.9178,   2.1936,   7.5388],
        [ 11.2571,   6.7072,   8.3693],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 11.0864,   7.0387,  -0.1238],
        [ -9.1706,   1.3863,   8.4032],
        [  8.5716,  10.3596,   6.8818],
        [ -9.2508,   1.0572,   2.9978],
        [  0.9445,   7.6736,   0.5095],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.6321,  -4.5296,  -2.5159],
        [  2.0039,   7.3173,  -1.5885],
        [  4.1637,  13.1898,  -1.9189],
        [ -7.0120,   5.2825,   7.7762],
        [  0.9036,   7.6840,   9.9237],
        [ 11.5083,  -4.1177,   0.2389],
        [  3.3167,   6.6022,   6.3554],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.4645,  -1.4880,  -2.0226],
        [  0.1382, -11.0073,   1.0026],
        [  6.0073,   3.0986,   6.7937],
        [ 12.0772,  -2.6014,  -7.1832],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.5428,  -9.5728,  -2.9897],
        [  3.7298,  13.3537,  -9.4653],
        [  3.9591, -10.6391,  -6.2557],
        [  0.2571,  -4.7594,   1.5618],
        [  1.1274,  14.0116,   8.1753],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.1973,  -0.4372,   6.5720],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.3305,   1.6119,   1.1538],
        [  6.2555,   2.1474,   4.1685],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.0373,   7.1299,  -8.5895],
        [ -0.0410,   7.8591,   9.3457],
        [ -9.1231,  -3.5739,   8.4708],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.6970,   6.9826,   9.7350],
        [ -2.5161,   7.7699,  -9.3531],
        [  4.1637,  13.1898,  -1.9189],
        [ -4.3963,   7.1629,   4.5263],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.1579,  -0.5961,  -6.5595],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.2849, -10.9655,  -7.6542],
        [  7.0515,  11.6194,   8.4289],
        [ -5.6423,   6.4537,  -1.2091],
        [  1.8721,  -4.7322,   0.5524],
        [  5.8889,  -1.3916,   8.8993],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -6.3306,   5.9272,   5.6935],
        [ 12.2589,  -1.9312,   9.8657],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.3444,   0.5821,   6.9101],
        [ -7.2515,   5.0195,   4.6434],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.0449,  13.8402,  -7.1642],
        [  3.3839,  -4.1966,   8.3574],
        [  2.5974,   7.0363,  -7.7810],
        [ -8.8154,   2.4611,   2.5198],
        [  7.8375,  -8.7240,  -9.8913],
        [  4.5340,  -3.3629,   8.9652],
        [  2.6614,  -4.5192,  -9.3641],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -9.3076,   0.7852,  -6.7010],
        [  5.8836,   3.4315,  -4.0327],
        [  2.7121,   6.9743,  -8.4202],
        [  9.0051,   9.9234,   5.4411],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.1905,   3.7253,   4.1175],
        [ 11.2767,  -4.6006,   0.7868],
        [ -0.8675, -10.8815,   6.8561],
        [ -7.3470,  -7.0615,   5.0066],
        [ 11.1469,   6.9237,   1.6160],
        [  3.3839,  -4.1966,   8.3574],
        [ 12.5717,   1.9266,   7.4903],
        [  2.3367,   7.1676,  -9.5263],
        [  6.3569,   0.7591,   6.5011]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0303e-01, 1.9456e-01,  ..., 8.0682e-01, 7.2672e-01,
         5.5393e-01],
        [1.0303e-01, 0.0000e+00, 2.2858e-01,  ..., 7.0435e-01, 7.5199e-01,
         4.5142e-01],
        [1.9456e-01, 2.2858e-01, 5.3633e-05,  ..., 8.6527e-01, 5.3312e-01,
         5.8784e-01],
        ...,
        [8.0682e-01, 7.0435e-01, 8.6527e-01,  ..., 0.0000e+00, 1.2247e+00,
         2.8228e-01],
        [7.2672e-01, 7.5199e-01, 5.3312e-01,  ..., 1.2247e+00, 0.0000e+00,
         9.5219e-01],
        [5.5393e-01, 4.5142e-01, 5.8784e-01,  ..., 2.8228e-01, 9.5219e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.4088043 , 11.52733581, 11.56146873, 11.87412439, 11.92133129,
       11.83509372, 12.59969936,  9.34061938,  9.10945678,  7.16598273,
        6.24187044, 13.86480175, 11.23201619, 13.0743399 ,  8.87007495,
        9.97964013,  6.90115901,  6.48486171,  4.98461983,  6.21275016,
        4.74468406,  5.10444097, 11.31276109, 10.27716434, 12.42554808,
        6.1561646 ,  9.18358562, 13.103709  , 10.39942497, 13.13205357,
        9.27474386, 13.44593519,  9.3109884 ,  7.73151309, 12.05395056,
        5.23878371,  7.58668144, 13.83139024,  8.77913428,  7.73692436,
       12.22276278,  7.38844703, 10.06286074,  9.58072286, 11.00812996,
        6.7593917 , 12.35421461,  5.53255882, 11.59512194, 13.86480175,
       11.35182589,  4.7663526 , 14.05687815,  9.95552626,  6.21275016,
       10.1371552 ,  6.53250937,  6.61386436, 12.32408516, 13.1399181 ,
        7.85919687,  9.79814997, 11.99633819,  7.48538905,  8.16714836,
       13.83139024,  8.40445872,  8.62631801,  6.18668463,  5.35774573,
        5.18277858, 11.20100795, 13.59170809,  8.5724042 ,  5.08910491,
        6.05113217,  9.66977802,  8.67227386, 12.41012046,  8.16714836,
        6.83515201,  9.36256096,  8.81930525, 10.41909211, 13.99048094,
        5.39098669,  7.50035355,  9.15253344, 11.72749923,  5.64500512,
        5.24461068, 10.57302546,  9.34061938,  6.81117802,  7.48311002,
       13.40024224,  9.93553424,  8.99791316, 12.17901033, 10.91601655,
       10.19036081, 13.12218013,  5.39098669, 12.71843462,  7.53884104,
        6.4020404 ])}
{'P': tensor([[5.2535e-03, 3.1649e-01, 1.8143e-01,  ..., 4.9044e-05, 3.8025e-04,
         3.7880e-05],
        [5.6941e-05, 6.6024e-03, 1.4187e-01,  ..., 3.5931e-05, 3.8619e-04,
         3.7080e-05],
        [6.9407e-05, 2.0978e-04, 6.8172e-03,  ..., 3.1777e-05, 3.3065e-04,
         3.1484e-05],
        ...,
        [1.7571e-01, 1.6278e-01, 1.2629e-01,  ..., 2.1085e-02, 3.7260e-03,
         4.9675e-04],
        [8.5125e-04, 9.7797e-04, 9.8827e-04,  ..., 6.2120e-04, 1.2192e-01,
         6.9275e-03],
        [5.9898e-04, 6.0450e-04, 5.5697e-04,  ..., 7.6781e-04, 3.7138e-02,
         9.8891e-02]]), 'X': tensor([[ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0403, 0.0637,  ..., 0.2302, 0.7262, 0.9168],
        [0.0403, 0.0000, 0.0245,  ..., 0.2570, 0.7152, 0.9132],
        [0.0637, 0.0245, 0.0000,  ..., 0.2797, 0.7186, 0.9210],
        ...,
        [0.2302, 0.2570, 0.2797,  ..., 0.0000, 0.6250, 0.7690],
        [0.7262, 0.7152, 0.7186,  ..., 0.6250, 0.0000, 0.2477],
        [0.9168, 0.9132, 0.9210,  ..., 0.7690, 0.2477, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.42420508, 11.48663767, 11.51557166, 11.70016217, 11.72802491,
       11.70168755, 11.04953767, 14.03618649, 11.04313641,  9.43737523,
       12.44177641,  5.89074104, 11.9702762 ,  5.40689332,  7.18354885,
       13.45124917, 11.72749923,  8.31625171,  7.70897675,  5.79543038,
       13.36092624, 13.89671314,  7.38844703, 10.95593729, 14.05633441,
        6.81961996, 12.21532282,  8.40445872, 10.61472636, 11.59512194,
        7.822854  , 12.58006038, 13.38091121, 13.3614598 , 12.17901033,
        8.67502348, 13.83139024, 13.20597932,  5.48276038, 13.60275211,
       14.0393949 ,  4.73784222, 10.06437598,  5.76165934, 10.00410625,
       13.09426142,  5.53175275,  5.94909709,  8.62631801, 13.59170809,
        9.15866607,  6.10360073, 13.77222752,  8.95983586,  8.10886021,
       11.65923396, 13.6660671 , 12.41366754, 11.86239961, 10.67132524,
        9.77588801, 12.95702647,  7.34250948,  9.97964013, 13.55868684,
        9.32094055, 13.13205357,  8.7273569 , 13.96746816, 13.43127131,
       11.37939695,  5.52416368,  8.36995953, 12.53285633, 12.58006038,
       13.09737053, 13.744387  ,  8.23525433,  4.78258092, 14.10796466,
        5.04028538,  6.81084724,  5.51177501, 10.4977132 ,  5.76165934,
        5.29755487,  7.53063885, 13.61801526,  9.88310614,  6.34059021,
        5.5671389 , 11.28093507, 13.27717982, 13.91198191, 12.18998195,
       12.71843462,  5.96556248, 13.69162795,  7.02496223,  7.62011734,
        5.07306274, 13.36092624,  5.87132718, 11.23201619,  5.54776934,
        5.18277858])}
{'P': tensor([[5.7770e-03, 3.3572e-01, 1.8215e-01,  ..., 4.0291e-07, 9.3676e-05,
         1.7367e-05],
        [4.2544e-05, 9.1218e-03, 8.7133e-02,  ..., 6.7152e-07, 1.8687e-04,
         1.1182e-05],
        [1.7352e-05, 1.4192e-04, 4.5941e-03,  ..., 4.0549e-07, 1.5914e-04,
         3.6739e-06],
        ...,
        [8.4558e-05, 5.3401e-05, 4.8834e-05,  ..., 1.6410e-01, 1.1275e-04,
         1.1809e-04],
        [2.1659e-05, 2.4613e-05, 3.2182e-05,  ..., 1.4143e-03, 9.7624e-02,
         6.5551e-06],
        [6.7785e-02, 8.0670e-02, 3.8542e-02,  ..., 1.8316e-06, 7.5669e-05,
         5.3541e-02]]), 'X': tensor([[  4.8051, -10.3703,   0.5268],
        [  6.1989,  -9.7591,   0.8289],
        [  6.9433,  -9.3362,  -0.3421],
        [  7.5064,  -8.9651,  -0.0994],
        [  7.8982,  -8.6778,   0.0239],
        [  8.1217,  -8.5024,  -0.6149],
        [ 11.1420,   6.9331,  -2.4779],
        [  6.3497,   1.3680,  -1.9076],
        [  3.3744,   6.5623,  -6.3636],
        [  4.6063,  13.0051,  -3.1456],
        [ 11.4495,   6.3026,   6.8257],
        [  0.3231,  -4.7691,  -1.6120],
        [  4.4633,   5.6287,  -0.7754],
        [  7.2340,  -9.1505,  -5.8321],
        [  3.8111, -10.6787,  -4.6539],
        [  0.9392,  -4.8143,  -3.4649],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.4721,  -1.3097,  -7.4377],
        [  1.7232,  -4.7574,   3.1270],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.5267,  -0.4125,  -8.7331],
        [  6.3505,   0.6235,  -8.5849],
        [  1.8447,  -4.7372,   8.1235],
        [ -8.9624,  -4.0944,  -4.9622],
        [  2.4932,   7.0903,  -5.4214],
        [ -9.3087,   0.7792,  -7.5587],
        [  6.6319,  11.9066,   3.5165],
        [  3.3094,  13.4969,  -3.6507],
        [ -3.1671,   7.6158,   6.0765],
        [  8.3791,  10.5409,   5.5590],
        [  0.1047, -11.0046,   0.5188],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.8704,  -9.3809,   7.5769],
        [ -6.4822,  -8.0252,  -3.8983],
        [  6.3497,   1.3680,  -1.9076],
        [  4.8432,   5.1968,   1.5006],
        [  6.6319,  11.9066,   3.5165],
        [ -0.8569,   7.9156,  -6.5557],
        [  5.1519, -10.2387,   9.8191],
        [  6.2770,  -0.0379,   2.4195],
        [  7.0515,  11.6194,   8.4289],
        [ 12.1379,   4.4125,  -3.8833],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.4336,  -2.2676,   5.6933],
        [ 10.0619,   8.6627,  -9.2884],
        [  4.1637,  13.1898,  -1.9189],
        [  5.8325,  -1.5222,  -2.1341],
        [ 10.9916,   7.2133,  -0.9179],
        [ 12.5123,  -0.5339,  -1.0728],
        [  4.3377,   5.7570,   1.3712],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.3804,  -2.3504,   7.2744],
        [ 10.1798,   8.5002,  -5.8893],
        [  3.3839,  -4.1966,   8.3574],
        [  0.1287, -11.0065,   6.5757],
        [  4.7813,  -3.1139,   8.2691],
        [  4.9337,  12.8566,  -3.6582],
        [ 12.1588,   4.3364,  -8.0867],
        [ 12.1588,   4.3364,  -8.0867],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.0506,   4.9277,  -9.5094],
        [ -8.8540,  -4.3971,  -5.2144],
        [  3.3167,   6.6022,   6.3554],
        [  0.6142,  14.0802,   0.5407],
        [  9.5203,   9.3483,   5.8781],
        [  9.8082,  -6.8396,   4.9615],
        [  6.3266,   0.3292,   9.3502],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.1824,  -4.6647,   1.1985],
        [  7.1441,  11.5529,   2.3232],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.1204,  -5.9308,   4.0761],
        [ -2.3317,   7.8034,  -3.6274],
        [ -5.9987,   6.1949,  -6.2114],
        [  8.3791,  10.5409,   5.5590],
        [  2.3889,  -4.6082,  -8.7291],
        [  9.3753,   9.5171,   3.9131],
        [  2.5128, -10.9365,   9.9645],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -4.8665,  -9.3204,  -5.2564],
        [  3.3633,  13.4794,   0.5193],
        [ -8.3653,   3.4184,   9.3454],
        [  4.6357,   5.4415,   7.9067],
        [ -9.4716,  -0.6673,   7.6068],
        [  2.7121,   6.9743,  -8.4202],
        [  6.8407,  11.7666,   3.0045],
        [  6.3031,   1.8477,  -7.1706],
        [  1.7500,   7.4187,   4.4237],
        [  9.0129,  -7.7072,   9.1400],
        [  1.9204,   7.3518,   8.4781],
        [ -5.9938,  -8.4745,   3.9735],
        [ -3.4488, -10.0934,   2.5195],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.3303,   1.6145,  -8.4083],
        [ -8.6603,   2.8250,   5.9181],
        [ -2.3317,   7.8034,  -3.6274],
        [ 11.8951,  -3.1570,   7.0742],
        [  5.9883,   3.1536,   6.2320],
        [  2.5375,   7.0676,  -4.3490],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.2659,  13.7894,   6.5436],
        [  4.4633,   5.6287,  -0.7754],
        [ -7.0825,   5.2073,  -5.0497],
        [  6.1929,   2.4501,  -6.9352],
        [  0.5250, -11.0306,   5.5358]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[7.5848e-05, 8.6798e-02, 1.1598e-01,  ..., 3.9126e-01, 6.1576e-01,
         5.2471e-01],
        [8.6798e-02, 5.3633e-05, 7.7467e-02,  ..., 4.5557e-01, 5.7467e-01,
         5.3419e-01],
        [1.1598e-01, 7.7467e-02, 0.0000e+00,  ..., 4.1155e-01, 5.0543e-01,
         6.1035e-01],
        ...,
        [3.9126e-01, 4.5557e-01, 4.1155e-01,  ..., 0.0000e+00, 6.4202e-01,
         8.4696e-01],
        [6.1576e-01, 5.7467e-01, 5.0543e-01,  ..., 6.4202e-01, 0.0000e+00,
         1.1011e+00],
        [5.2471e-01, 5.3419e-01, 6.1035e-01,  ..., 8.4696e-01, 1.1011e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.42947632, 11.56146873, 11.63502679, 11.69264243, 11.73397309,
       11.75807252, 13.12299145,  6.49538841,  7.37902766, 13.79676654,
       13.06957425,  4.78002665,  7.18354885, 11.66453059, 11.33837396,
        4.90505566, 10.68031351,  9.562174  ,  5.05989547, 11.28895489,
       12.53345445,  6.38105015,  5.08372149,  9.85330646,  7.5158489 ,
        9.34127013, 13.62896593, 13.89671314,  8.24808832, 13.46553623,
       11.00508681,  5.22903117, 11.62770332, 10.31613269,  6.49538841,
        7.10378919, 13.62896593,  7.96181288, 11.46176259,  6.27715105,
       13.59170809, 12.91505111, 12.07225469, 12.19805394,  5.88782913,
       13.27717982, 13.83139024,  6.02789634, 13.1471313 , 12.52372629,
        7.20826219, 12.67576792,  5.87132718, 13.26209545,  5.39098669,
       11.00726483,  5.70591986, 13.77075174, 12.9089559 , 12.9089559 ,
       10.31886066,  7.05627461,  9.88571713,  7.38844703, 14.09357547,
       13.3426541 , 11.95743558,  6.33517349,  5.79391825,  5.14999799,
       13.58333857,  8.12043983, 10.05558893,  8.14433542,  8.62329466,
       13.46553623,  5.19065439, 13.35927527, 11.22142195, 12.30858677,
       10.51437947, 13.89264822,  9.0368458 ,  7.14837514,  9.49511524,
        7.48311002, 13.6105595 ,  6.56833864,  7.62232307, 11.85890664,
        7.5984809 , 10.37998078, 10.66631784, 10.67132524,  6.53290663,
        9.10945678,  8.14433542, 12.30695098,  6.76791483,  7.5092834 ,
       13.01728313, 13.97430191,  7.18354885,  8.7907923 ,  6.65991877,
       11.04313641])}
{'P': tensor([[6.0280e-03, 2.4776e-01, 1.5864e-01,  ..., 1.0641e-03, 2.8385e-07,
         4.8131e-07],
        [1.7229e-05, 4.9712e-03, 2.8191e-01,  ..., 6.9436e-04, 1.8679e-07,
         4.9888e-07],
        [1.2714e-04, 3.6783e-04, 7.1408e-02,  ..., 9.7467e-03, 2.6225e-06,
         1.0323e-05],
        ...,
        [2.4984e-04, 4.5086e-04, 7.4431e-04,  ..., 7.7845e-02, 6.8069e-05,
         3.4331e-04],
        [2.4127e-04, 9.8494e-05, 6.5034e-05,  ..., 1.8780e-04, 4.5812e-02,
         1.2287e-05],
        [1.4855e-06, 1.0065e-06, 9.7710e-07,  ..., 2.3579e-05, 1.0313e-03,
         4.2751e-02]]), 'X': tensor([[ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1118, 0.1629,  ..., 0.4544, 0.6397, 0.9606],
        [0.1118, 0.0000, 0.0537,  ..., 0.3859, 0.7450, 0.9715],
        [0.1629, 0.0537, 0.0000,  ..., 0.3444, 0.7869, 0.9610],
        ...,
        [0.4544, 0.3859, 0.3444,  ..., 0.0000, 0.9149, 0.7222],
        [0.6397, 0.7450, 0.7869,  ..., 0.9149, 0.0000, 0.9175],
        [0.9606, 0.9715, 0.9610,  ..., 0.7222, 0.9175, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.4424816 , 11.63765273, 11.75618513, 11.67875855, 11.73848065,
       11.74901778,  7.4544866 , 11.70016217,  9.77588801,  5.57341623,
        8.14433542, 12.40074901, 10.92597196, 13.98691426,  6.55800267,
       11.72749923,  5.10444097, 11.22993218,  8.51928531,  5.18277858,
       11.33837396, 14.02058621,  9.78286733, 12.04924159,  5.04028538,
       13.03002338,  6.21275016, 12.50996828,  9.15976746, 13.18278606,
        9.23071698, 10.37998078,  5.21280156,  5.92281142, 11.9120321 ,
        6.53290663, 13.44593519,  9.87720102, 10.05558893,  6.65259775,
        9.3109884 , 11.83570132,  6.67121748, 13.26590071,  8.62329466,
        7.61088125,  5.29755487,  8.06643487, 11.48734978, 12.87752052,
       10.5575601 , 12.77402351, 11.28215429,  6.38399789, 11.62158554,
        7.16598273, 12.87752052,  6.81084724,  8.16584537,  8.28968067,
        5.14371005, 12.45418236,  8.89458273,  7.42512621,  7.76457095,
        8.68832398,  6.90115901, 11.48734978, 11.65923396,  4.85995002,
        6.19707944,  5.91408504,  6.1561646 , 11.27160035,  6.39438044,
        6.5618968 ,  9.58072286,  9.64157461,  7.5092834 , 10.06703358,
       11.69264243, 11.4890861 , 11.64436684, 10.52357479, 10.75443106,
        5.19577146,  8.22728399, 13.43127131,  7.02496223,  5.41348121,
       12.1494023 ,  6.65991877,  8.89458273,  9.24501829,  8.99791316,
       11.018084  , 12.27835996, 13.35832725,  6.35029274, 12.84056604,
        4.98461983, 11.28895489, 13.13205357,  5.84450127,  9.5014146 ,
        8.04664886])}
{'P': tensor([[7.3364e-03, 3.4764e-01, 2.3658e-01,  ..., 7.3482e-05, 4.4183e-07,
         8.0929e-03],
        [2.1098e-05, 6.8531e-03, 5.8091e-01,  ..., 9.6860e-05, 8.3832e-07,
         1.9566e-02],
        [3.2718e-05, 8.4031e-05, 1.5335e-02,  ..., 2.5891e-04, 2.6222e-06,
         6.5094e-02],
        ...,
        [2.2844e-05, 3.1364e-05, 3.7523e-05,  ..., 1.1880e-01, 9.1536e-03,
         2.9539e-04],
        [6.3866e-08, 5.9234e-08, 6.1202e-08,  ..., 1.6237e-05, 3.5764e-02,
         1.6873e-07],
        [2.4782e-05, 5.2199e-05, 7.2183e-05,  ..., 9.0795e-03, 5.5722e-04,
         1.4060e-01]]), 'X': tensor([[  5.1519, -10.2387,   9.8191],
        [  7.1821,  -9.1845,   9.8548],
        [  7.9141,  -8.6656,   9.6948],
        [  6.8093,  -9.4178,   8.4244],
        [  8.3718,  -8.2953,   8.4666],
        [  9.0129,  -7.7072,   9.1400],
        [  1.4306,  13.9620,  -0.2075],
        [ -8.0608,   3.9362,  -5.9567],
        [ -4.3963,   7.1629,   4.5263],
        [ -9.4739,  -1.2497,   0.5483],
        [  9.1620,  -7.5573,  -9.9882],
        [  5.2188,  -2.5839,   2.1833],
        [ 12.5919,   0.4035,  -5.8460],
        [ -5.9872,   6.2037,  -9.4477],
        [  4.3523, -10.5232,   8.2138],
        [  2.5318,  13.7231,   6.3692],
        [  2.5636, -10.9293,   9.6519],
        [ -6.4464,   5.8272,   7.5038],
        [  9.1409,  -7.5789,   1.5814],
        [  6.1741,  -0.5329,   3.1913],
        [  6.7792,  11.8084,   1.6251],
        [  6.1685,  12.1979,  -6.2473],
        [ 10.9599,  -5.1849,  -3.5526],
        [  2.7121,   6.9743,  -8.4202],
        [ -0.7599, -10.8995,  -6.0366],
        [  3.9591, -10.6391,  -6.2557],
        [  4.8432,   5.1968,   1.5006],
        [ -9.1231,  -3.5739,   8.4708],
        [  9.3836,   9.5075,  -5.9278],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.6672,  -1.8631,   8.1020],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -6.0195,   6.1789,  -2.9496],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.6523,   3.9443,  -2.9449],
        [  6.3515,   0.6418,  -6.5823],
        [  5.4426, -10.1181,  -5.1268],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.4604,  -4.1545,   1.8091],
        [  0.7275,  14.0667,  -0.9815],
        [  3.6817,   6.3354,   6.1233],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 11.3367,  -4.4806,   0.4643],
        [ 10.0400,   8.6922,  -6.2847],
        [  9.3424,  -7.3685,  -3.6807],
        [  1.5662,  13.9375,   4.5142],
        [ -5.0943,  -9.1682,   0.5073],
        [ -3.1671,   7.6158,   6.0765],
        [  0.3715,  -4.7756,   0.2312],
        [ -6.7992,   5.4991,   9.9186],
        [ -3.9700,   7.3453,  -4.1486],
        [  9.5018,  -7.1945,  -1.1892],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 11.0373,   7.1299,  -8.5895],
        [  2.1379,  -4.6757,   6.4981],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.5149,  -5.1905,   1.5239],
        [ -1.7290,   7.8825,  -3.3831],
        [ -8.6381,  -4.9238,  -4.4077],
        [  4.3554,   5.7393,  -9.0766],
        [  6.3534,   1.2990,  -7.2942],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.9003,  12.3550,   2.1917],
        [ -1.7009,   7.8851,  -8.2437],
        [ -9.4735,  -0.7274,  -2.6749],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.2508,   1.0572,   2.9978],
        [ 12.1379,   4.4125,  -3.8833],
        [  4.5325,  -3.3643,   6.7699],
        [  4.6231,   5.4556,   8.4973],
        [ 12.2692,  -1.8882,   3.9253],
        [ -1.3178, -10.7939,   0.4236],
        [  7.4132,  11.3526,   0.3322],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.3653,   3.4184,   9.3454],
        [  0.3715,  -4.7756,   0.2312],
        [ -7.7842,   4.3466,  -1.7548],
        [  2.8602,   6.8905,   5.8779],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.3124,   0.2057,  -0.4593],
        [  1.7301,   7.4262,   5.0522],
        [  6.2905,   1.9366,   2.1169],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -1.8823, -10.6558,  -3.8952],
        [  9.0051,   9.9234,   5.4411],
        [  3.7298,  13.3537,  -9.4653],
        [  6.0560,  -0.9421,  -5.0334],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4766,  -1.1116,  -2.1706],
        [  6.2770,  -0.0379,   2.4195],
        [  9.2811,  -7.4336,   5.1771],
        [ -3.1671,   7.6158,   6.0765],
        [  2.0449,  13.8402,  -7.1642],
        [  6.9693,  -9.3200,  -8.3822],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.8979,   3.3958,  -3.8713],
        [  2.4932,   7.0903,  -5.4214],
        [  3.4401,  13.4540,   4.6576],
        [  2.7685, -10.8982,  -6.8928],
        [ -9.3810,  -2.3455,  -9.5194],
        [  6.3211,   1.7017,  -0.2164],
        [  4.5954,  13.0099,  -4.4439],
        [ 11.8951,  -3.1570,   7.0742]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0961, 0.1492,  ..., 0.8064, 1.2650, 0.3745],
        [0.0961, 0.0000, 0.0532,  ..., 0.7670, 1.2224, 0.2795],
        [0.1492, 0.0532, 0.0000,  ..., 0.7451, 1.1971, 0.2265],
        ...,
        [0.8064, 0.7670, 0.7451,  ..., 0.0000, 0.4588, 0.6683],
        [1.2650, 1.2224, 1.1971,  ..., 0.4588, 0.0000, 1.0904],
        [0.3745, 0.2795, 0.2265,  ..., 0.6683, 1.0904, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.46176259, 11.65923396, 11.73567759, 11.62158554, 11.785566  ,
       11.85890664, 14.03505624,  8.97052115,  8.40445872,  9.55593374,
       11.87665501,  5.82345867, 12.59840193,  8.62162372, 11.38774149,
       13.95472331, 11.22596994,  8.68979052, 11.87412439,  6.19707944,
       13.61600536, 13.66896739, 12.12448912,  7.48311002, 10.92597196,
       11.35182589,  7.10378919,  9.79814997, 13.35832725, 12.1494023 ,
        5.96556248,  9.75075273,  8.62631801,  8.47371868,  6.89245352,
        6.383897  , 11.4890861 , 13.09737053,  5.40689332, 14.08549802,
        7.32754882, 10.33712258, 12.18998195, 13.27994304, 11.89855267,
       14.02526505, 10.48840476,  8.24808832,  4.79003361,  8.74470218,
        8.34949383, 11.9182956 , 11.48434767,  9.55720651, 13.1399181 ,
        5.14124612,  8.98765644,  9.97220799,  8.06990483,  9.94285511,
        7.20482224,  6.48486171, 11.75807252, 13.69162795,  8.06643487,
        9.5014146 , 13.02818702,  9.3109884 , 12.91505111,  5.64464807,
        7.15100303, 12.41366754, 10.87409005, 13.55868684,  5.05444288,
        9.0368458 ,  4.79003361,  8.91551231,  7.46053281,  9.09661212,
        6.31576264,  7.62509582,  6.58184474,  9.77588801,  9.75401449,
       10.8207297 , 13.40024224, 13.86480175,  6.12886024,  8.95983586,
        9.54154096,  6.27715105, 11.89105259,  8.24808832, 13.99048094,
       11.63765273,  9.88571713,  9.69629   ,  6.80559669,  7.5158489 ,
       13.88683436, 11.24434374,  9.66977802,  6.54615887, 13.79762415,
       12.30695098])}
{'P': tensor([[7.1660e-03, 3.1277e-01, 1.5990e-01,  ..., 6.6015e-06, 1.7411e-05,
         7.6044e-05],
        [4.5253e-05, 9.5715e-03, 5.4642e-01,  ..., 1.4557e-05, 2.0137e-05,
         1.6652e-04],
        [2.2955e-05, 4.2297e-05, 1.2323e-02,  ..., 3.1129e-05, 1.8404e-05,
         3.6955e-04],
        ...,
        [1.1741e-05, 1.1119e-05, 1.0536e-05,  ..., 1.2843e-01, 2.2813e-04,
         1.1212e-03],
        [6.8269e-04, 1.0448e-03, 1.3155e-03,  ..., 1.1130e-03, 1.8524e-01,
         4.9622e-04],
        [1.0413e-05, 1.0890e-05, 1.2113e-05,  ..., 1.2048e-02, 2.9557e-05,
         5.8521e-02]]), 'X': tensor([[ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0872, 0.1605,  ..., 0.5360, 0.8913, 0.3454],
        [0.0872, 0.0000, 0.0761,  ..., 0.4868, 0.8368, 0.2593],
        [0.1605, 0.0761, 0.0000,  ..., 0.4811, 0.8205, 0.1852],
        ...,
        [0.5360, 0.4868, 0.4811,  ..., 0.0000, 0.3568, 0.4675],
        [0.8913, 0.8368, 0.8205,  ..., 0.3568, 0.0000, 0.7596],
        [0.3454, 0.2593, 0.1852,  ..., 0.4675, 0.7596, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.48434767, 11.63724792, 11.80092578, 11.59512194, 11.89855267,
       11.66453059, 12.84056604,  7.46248947,  6.05165248,  7.02326515,
       12.50996828,  6.38105015, 11.33837396, 10.95593729,  7.67939546,
       10.46238054,  6.76056625, 12.59969936,  9.68324553,  7.28793663,
        6.48486171,  5.05444288,  9.34061938,  9.77588801,  8.14433542,
        9.4997987 ,  6.54615887,  9.75075273,  6.00663314,  6.652742  ,
        5.94541803, 13.04051744,  8.83725081,  4.91038884,  8.90405823,
        6.81084724, 10.04916678,  7.63726134, 13.77604293,  6.1561646 ,
        6.24187044, 10.74409834,  9.36093562, 11.48663767, 14.02526505,
        7.5984809 ,  5.05444288,  9.25336371, 11.59512194,  8.81930525,
       11.07637531,  8.37954853,  5.94293185,  6.78325151,  7.58378655,
       13.6660671 , 13.85158417,  6.39520397,  9.47369628,  9.78147542,
       11.1913598 ,  4.92497514,  4.96231532, 12.30858677, 13.62896593,
       11.32149259, 13.2430377 , 13.32613445, 13.84126197,  9.47369628,
        5.98285319,  7.01475376,  9.71392179,  9.37989631, 12.24467041,
        6.19633139,  7.42512621,  9.14044984,  5.5366698 , 11.10066126,
       13.18278606,  5.84533345,  5.20659892,  8.56889738, 11.23201619,
        8.20436764,  4.91296205,  7.6374603 , 13.0493368 , 13.1471313 ,
       10.19036081, 12.26997165,  4.74770899,  8.36445758, 12.52372629,
       11.00726483,  7.66377554,  8.92098752,  8.87007495, 13.50484539,
        5.84450127, 13.48809595,  8.50278896,  7.69752262,  5.08372149,
        7.15990676])}
{'P': tensor([[5.0514e-03, 2.0884e-01, 1.9006e-01,  ..., 8.6454e-08, 4.6612e-06,
         5.2877e-06],
        [2.5369e-05, 8.6521e-03, 6.5531e-02,  ..., 1.5993e-07, 2.1119e-05,
         2.5513e-05],
        [1.7583e-05, 5.2040e-04, 7.0682e-03,  ..., 1.1658e-07, 1.5821e-05,
         2.2914e-05],
        ...,
        [1.9971e-05, 1.1228e-05, 8.8783e-06,  ..., 7.3357e-02, 2.0351e-06,
         1.0797e-06],
        [2.0578e-07, 2.8503e-07, 2.7018e-07,  ..., 1.0776e-04, 5.0767e-02,
         5.7786e-03],
        [6.4084e-07, 6.5801e-07, 6.5190e-07,  ..., 1.9069e-04, 8.4167e-05,
         8.9297e-02]]), 'X': tensor([[ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1221, 0.1348,  ..., 1.0496, 1.2932, 0.7886],
        [0.1221, 0.0000, 0.0419,  ..., 1.0719, 1.2061, 0.6670],
        [0.1348, 0.0419, 0.0000,  ..., 1.1137, 1.2324, 0.6582],
        ...,
        [1.0496, 1.0719, 1.1137,  ..., 0.0000, 1.0980, 1.4364],
        [1.2932, 1.2061, 1.2324,  ..., 1.0980, 0.0000, 0.9523],
        [0.7886, 0.6670, 0.6582,  ..., 1.4364, 0.9523, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.48663767, 11.70016217, 11.72802491, 11.70168755, 11.8159254 ,
       11.51557166,  9.43112957, 10.14436912, 13.52430176, 10.06407839,
        5.64464807,  5.53255882, 11.32149259,  5.19577146, 12.52372629,
       10.06023665,  5.14607014,  8.37954853,  6.12886024, 10.00794468,
        7.96181288,  9.04255655,  8.90405823, 10.39942497,  9.75401449,
       13.04156075, 12.07225469, 12.85032836,  6.99097013, 11.63383221,
       13.98691426,  7.18354885, 13.37121923,  5.5748624 ,  9.58072286,
       11.89105259, 11.54672755, 12.59969936, 10.14436912,  8.1137452 ,
       10.63038565, 11.62770332, 12.44177641,  8.0550286 , 12.78327037,
       10.52587243, 11.72749923,  7.44805901, 13.84126197, 10.04916678,
       10.95593729, 10.26327895,  6.54615887,  7.54127425,  5.19065439,
        7.20482224,  8.04718433, 10.09979747,  5.95111804,  7.18354885,
        4.8763135 , 10.78533838, 11.92133129,  8.86082124,  8.87007495,
        8.99276564,  7.92200342, 14.10796466,  7.62011734, 11.54071859,
        5.05774295, 12.4013153 , 10.84437705,  6.21275016,  7.39004226,
       13.5247215 ,  9.76896904,  7.92200342,  5.24461068,  7.96575782,
        8.87587457, 11.92133129, 10.69963954,  9.23178204,  6.53290663,
       10.00410625, 14.08549802,  7.32950385, 13.6660671 ,  5.94535529,
        8.31625171, 11.04313641,  8.5724042 , 12.45418236,  5.07306274,
       10.53368644, 12.71843462,  5.94535529,  5.23878371,  9.93553424,
        4.91296205,  7.66377554, 10.80728788,  9.14044984, 13.27994304,
       13.62896593])}
{'P': tensor([[1.1909e-02, 2.1648e-01, 1.9492e-01,  ..., 4.6302e-04, 2.2224e-04,
         9.2792e-05],
        [1.3109e-05, 9.4384e-03, 6.8874e-04,  ..., 1.2348e-03, 4.9196e-04,
         1.0368e-04],
        [9.6035e-06, 4.2316e-02, 5.4473e-03,  ..., 6.8705e-04, 2.0337e-04,
         4.3898e-05],
        ...,
        [3.7870e-05, 6.5729e-05, 6.3301e-05,  ..., 1.2453e-01, 2.1941e-04,
         3.5074e-05],
        [5.1349e-07, 1.4384e-06, 1.2005e-06,  ..., 1.7520e-04, 2.4647e-02,
         2.1794e-04],
        [1.1999e-06, 3.8327e-06, 2.9963e-06,  ..., 9.5908e-05, 1.2337e-01,
         5.5654e-02]]), 'X': tensor([[  5.4242, -10.1261,   3.9595],
        [  9.0004,  -7.7195,   4.6945],
        [  8.7715,  -7.9390,   5.2900],
        [  9.2811,  -7.4336,   5.1771],
        [  7.5784,  -8.9141,   5.8410],
        [  9.8082,  -6.8396,   4.9615],
        [ -2.4956, -10.4682,   3.5632],
        [ 12.5789,   0.1722,   1.6470],
        [  6.3413,   0.4883,  -6.8416],
        [  9.3836,   9.5075,  -5.9278],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 11.1420,   6.9331,  -2.4779],
        [  7.3592,  11.3937,   6.2407],
        [  8.0877,  -8.5297,  -1.1050],
        [ -9.0604,   1.7707,   1.1066],
        [ -3.9508,  -9.8515,   1.7923],
        [  5.8889,  -1.3916,   8.8993],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.9655,   3.2172,  -7.5953],
        [  0.2480, -11.0153,  -3.0183],
        [ -1.2454,   7.9127,  -0.5715],
        [  5.9548,  -1.2273,  -2.6520],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.6928,  12.4710,   3.6001],
        [ -7.8126,   4.3066,   5.1791],
        [  3.8182,  -3.9347,  -1.7961],
        [  3.7992,  -3.9474,  -8.9019],
        [ 12.6000,   1.3839,  -3.2992],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.3705,  13.9723,   7.9266],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -8.4573,   3.2448,   5.4370],
        [ -2.6069, -10.4298,  -6.1995],
        [  1.7000,  -4.7609,   2.7935],
        [  2.4639, -10.9431,  -7.3718],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -3.7964,  -9.9294,   7.7147],
        [ -8.6801,   2.7809,  -0.8443],
        [  3.5965,  -4.0754,  -3.5035],
        [  0.9809,  -4.8145,   6.8706],
        [ -2.6735, -10.4062,  -1.5911],
        [  1.0093,   7.6567,   2.3526],
        [  5.1292,   4.8183,  -5.7997],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.6093,  -1.9705,   6.4807],
        [ 12.6059,   0.9597,  -6.3264],
        [  1.8320,   7.3871,   5.1452],
        [  2.2437,  -4.6489,   6.1430],
        [ -2.4956, -10.4682,   3.5632],
        [  9.9750,  -6.6338,  -1.2440],
        [  6.1575,  12.2046,   3.3257],
        [ -3.2486,   7.5925,  -1.3719],
        [  6.3124,   0.2057,  -0.4593],
        [  2.3589,  13.7668,  -3.6501],
        [  6.0047,   3.1062,  -6.7914],
        [  7.1697,  11.5343,  -0.7276],
        [  2.4572,   7.1085,   7.7376],
        [ -5.1369,   6.7755,   8.3836],
        [ -5.6423,   6.4537,  -1.2091],
        [  2.4639, -10.9431,  -7.3718],
        [  6.3124,   0.2057,  -0.4593],
        [  6.3587,   1.1596,  -3.6629],
        [  9.0051,   9.9234,   5.4411],
        [ -8.6229,   2.9068,   2.4429],
        [  2.9260,  -4.4162,   8.3808],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.8450,  -9.9356,   3.0767],
        [  6.7565,  11.8237,  -4.9882],
        [ -7.6734,   4.4980,  -5.4248],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.6035,   0.7494,   7.1536],
        [ -3.7964,  -9.9294,   7.7147],
        [ -8.9624,  -4.0944,  -4.9622],
        [  1.2728, -11.0372,   9.8121],
        [  4.5814, -10.4485,   2.5496],
        [  2.9410,   6.8428,   5.4458],
        [ -5.9987,   6.1949,  -6.2114],
        [ -2.0445,   7.8469,  -0.5857],
        [ -7.8359,   4.2735,   7.2943],
        [  4.0591,   6.0202,  -1.6339],
        [  3.4401,  13.4540,   4.6576],
        [  6.2905,   1.9366,   2.1169],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -5.9938,  -8.4745,   3.9735],
        [  5.7802,  -1.6363,  -6.2782],
        [ 12.5789,   0.1722,   1.6470],
        [ 10.9599,  -5.1849,  -3.5526],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.8934,   2.2597,   3.5831],
        [  4.2041,  -3.6509,  -4.9488],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.0039,   7.3173,  -1.5885],
        [ 12.2592,   3.9425,   2.0383],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.4735,  -0.7274,  -2.6749],
        [  1.7815,  -4.7481,   4.4289],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -8.9282,   2.1648,   8.4894],
        [  2.3373,  -4.6232,  -5.2246],
        [  5.9883,   3.1536,   6.2320],
        [ 12.4691,   2.8525,  -5.5110],
        [ 12.5927,   0.4199,  -9.4997]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.2005, 0.2055,  ..., 0.4148, 1.0012, 1.1502],
        [0.2005, 0.0000, 0.0680,  ..., 0.2314, 0.9974, 1.1575],
        [0.2055, 0.0680, 0.0000,  ..., 0.2784, 1.0644, 1.2240],
        ...,
        [0.4148, 0.2314, 0.2784,  ..., 0.0000, 0.9127, 1.0787],
        [1.0012, 0.9974, 1.0644,  ..., 0.9127, 0.0000, 0.1664],
        [1.1502, 1.1575, 1.2240,  ..., 1.0787, 0.1664, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.48734978, 11.85743425, 11.83075005, 11.89105259, 11.70016217,
       11.95743558, 10.76154256, 12.58006038,  6.36004393, 13.35832725,
        9.62958973, 13.12299145, 13.56367857, 11.75438006,  9.23178204,
       10.61417181,  6.05113217, 12.67576792,  6.77777606, 11.018084  ,
        8.01009083,  6.07992623, 10.51340829, 13.70893492,  8.92098752,
        5.48276038,  5.47866221, 12.67576792, 13.12299145, 14.0393949 ,
        9.91668182,  9.05843201, 10.7506423 ,  5.05534609, 11.21703654,
        5.24137909,  9.50913849, 10.63038565,  9.11473391,  5.43544626,
        4.91336994, 10.74409834,  7.72291946,  7.03733775, 10.32715715,
       13.1399181 ,  5.94535529, 12.64235359,  7.61088125,  5.16205087,
       10.76154256, 11.97949273, 13.66991027,  8.25828409,  6.31576264,
       13.96746816,  6.76056625, 13.58101794,  7.52117132,  8.50268322,
        8.5724042 , 11.21703654,  6.31576264,  6.46356806, 13.40024224,
        9.09963894,  5.29755487,  9.48969978,  5.24137909, 11.52733581,
       13.61801526,  8.89458273,  9.97822473, 12.62576026, 10.63038565,
        9.85330646, 11.11038551, 11.4088043 ,  7.44805901,  8.62329466,
        8.10886021,  8.92549752,  7.26074925, 13.88683436,  6.58184474,
        9.70108409,  9.96152788, 10.37998078,  6.00732193, 12.58006038,
       12.12448912,  5.54776934,  9.17595899,  5.56809335, 10.08001894,
        7.58668144, 12.87752052,  6.06416957,  9.5014146 ,  5.0713364 ,
       12.11735334,  9.18690018,  5.18047638,  6.76791483, 12.79126785,
       12.59969936])}
{'P': tensor([[1.0236e-02, 3.9802e-01, 1.3181e-01,  ..., 2.3877e-06, 1.4989e-06,
         2.4179e-03],
        [1.4764e-04, 3.5180e-02, 1.2885e-01,  ..., 1.2873e-05, 6.4669e-06,
         2.3781e-02],
        [3.0606e-05, 2.5845e-04, 1.2324e-02,  ..., 1.7192e-05, 2.8719e-06,
         6.2188e-03],
        ...,
        [1.1675e-08, 1.1360e-08, 2.0701e-08,  ..., 5.3141e-03, 1.4088e-05,
         4.2765e-08],
        [5.5923e-05, 3.4130e-05, 2.4536e-05,  ..., 3.5028e-06, 1.5935e-01,
         6.6354e-06],
        [7.6586e-06, 1.3339e-05, 2.1256e-05,  ..., 1.6864e-04, 1.6219e-04,
         1.1614e-01]]), 'X': tensor([[  5.4426, -10.1181,  -5.1268],
        [  7.2340,  -9.1505,  -5.8321],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.3424,  -7.3685,  -3.6807],
        [  6.9653,  -9.3225,  -3.4727],
        [  6.3122,  -9.6994,  -6.3282],
        [  8.4691,  10.4571,  -5.0650],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.0587, -10.6112,   6.2273],
        [  1.9204,   7.3518,   8.4781],
        [  5.1519, -10.2387,   9.8191],
        [ 10.0995,  -6.4739,   2.0974],
        [  4.5503,   5.5359,   7.4735],
        [  6.1973,  -0.4372,   6.5720],
        [  4.7156,  -3.1831,  -8.0224],
        [ 11.2847,  -4.5848,  -4.7503],
        [  3.3474,  -4.2162,   5.2455],
        [  4.3523, -10.5232,   8.2138],
        [ -3.1671,   7.6158,   6.0765],
        [ -2.9497, -10.3028,  -7.3479],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.7566,   7.6212,  -2.3279],
        [ -3.3999, -10.1152,  -1.2088],
        [ 11.7779,   5.5127,  -2.0699],
        [  8.0877,  -8.5297,  -1.1050],
        [  7.8375,  -8.7240,  -9.8913],
        [ -7.4887,   4.7364,   8.9074],
        [ 11.7796,  -3.4696,   9.0088],
        [  2.0547,   7.2957,  -1.9343],
        [ -8.6229,   2.9068,   2.4429],
        [ -7.3470,  -7.0615,   5.0066],
        [  9.9058,  -6.7203,   5.6928],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.3515,   0.6418,  -6.5823],
        [  3.8111, -10.6787,  -4.6539],
        [  2.8994,   6.8675,  -9.5987],
        [ -1.6906,   7.8860,   7.7142],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.4573,   3.2448,   5.4370],
        [ 11.4703,  -4.2006,   7.4086],
        [  5.0617,  12.7957,   8.3544],
        [  2.2182,  13.8007,   5.1889],
        [  6.3583,   1.1729,   8.7835],
        [  3.7992,  -3.9474,  -8.9019],
        [  2.4356,  -4.5942,  -5.0808],
        [ 11.0275,  -5.0666,   8.4154],
        [  2.9260,  -4.4162,   8.3808],
        [  4.3377,   5.7570,   1.3712],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -0.5709,   7.9055,   7.6089],
        [  8.4691,  10.4571,  -5.0650],
        [  5.6672,  -1.8631,   8.1020],
        [  6.3569,   0.7591,   6.5011],
        [  8.6420,  -8.0580,   6.5781],
        [ -0.5709,   7.9055,   7.6089],
        [ 11.9808,   4.9340,  -3.3715],
        [  7.7794,  11.0629,   8.9069],
        [ -9.1459,   1.4780,  -4.2403],
        [  3.7992,  -3.9474,  -8.9019],
        [  3.4009,  -4.1874,  -5.9751],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.1041, -11.0045,   2.4311],
        [  1.7000,  -4.7609,   2.7935],
        [  5.6414, -10.0303,   7.8525],
        [ -4.7389,   6.9953,   3.7638],
        [  2.6310, -10.9195,   3.8054],
        [  4.8653,   5.1694,  -4.0823],
        [ 11.2209,  -4.7092,   2.6117],
        [  6.3505,   0.6235,  -8.5849],
        [  2.4149,  -4.6005,  -4.3339],
        [  7.0664,  -9.2590,   8.0879],
        [  7.0171,  11.6439,  -5.0178],
        [  4.1537,   5.9340,  -6.0423],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -8.0829,  -5.9940,   5.0564],
        [ -3.5173,   7.5091,  -8.9585],
        [  3.4140,  -4.1803,  -8.8224],
        [  8.1528,  10.7452,   9.1493],
        [  1.7232,  -4.7574,   3.1270],
        [  3.2990,  -4.2415,  -7.4857],
        [ 11.0864,   7.0387,  -0.1238],
        [  4.1637,  13.1898,  -1.9189],
        [ 11.4235,   6.3594,   5.1663],
        [  3.2647, -10.8066,  -0.3768],
        [  4.6231,   5.4556,   8.4973],
        [ -2.6735, -10.4062,  -1.5911],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3301,   0.3638,  -7.5685],
        [ -3.3116, -10.1539,   7.5919],
        [ -5.9987,   6.1949,  -6.2114],
        [  8.3718,  -8.2953,   8.4666],
        [ -7.3470,  -7.0615,   5.0066],
        [  1.1644, -11.0394,   7.8511],
        [ 11.5198,  -4.0923,   5.5088],
        [  2.0419,  -4.6978,  -5.0983],
        [  3.9913,  -3.8139,  -6.8311],
        [ -5.9938,  -8.4745,   3.9735],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.5240,  -2.1198,   6.2055],
        [  8.1528,  10.7452,   9.1493],
        [ -5.5525,   6.5146,  -4.4984],
        [ 12.5763,   0.1333,  -9.2650]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1294e-01, 1.7582e-01,  ..., 1.4407e+00, 1.6802e-01,
         6.1308e-01],
        [1.1294e-01, 0.0000e+00, 1.6582e-01,  ..., 1.4407e+00, 2.8095e-01,
         5.0245e-01],
        [1.7582e-01, 1.6582e-01, 0.0000e+00,  ..., 1.2791e+00, 2.8718e-01,
         5.3348e-01],
        ...,
        [1.4407e+00, 1.4407e+00, 1.2791e+00,  ..., 0.0000e+00, 1.4547e+00,
         1.4358e+00],
        [1.6802e-01, 2.8095e-01, 2.8718e-01,  ..., 1.4547e+00, 1.5170e-04,
         7.7855e-01],
        [6.1308e-01, 5.0245e-01, 5.3348e-01,  ..., 1.4358e+00, 7.7855e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.4890861 , 11.66453059, 11.80092578, 11.89855267, 11.63724792,
       11.57249984, 13.45642379,  8.05242591,  5.04028538, 11.36089355,
        7.5984809 , 11.46176259, 11.99633819,  7.16598273,  6.21275016,
        5.68942572, 12.18045978,  5.38342077, 11.38774149,  8.24808832,
       10.7167285 ,  5.41348121,  6.80559669, 13.18278606, 10.67132524,
       13.00413829, 11.75438006, 11.72749923,  8.86082124, 12.27993098,
        7.57946459,  9.09963894, 10.19036081, 11.9702762 , 13.27994304,
        6.383897  , 11.33837396,  7.4544866 ,  8.06517143,  5.19065439,
        9.05843201, 12.21532282, 13.76047999, 13.97779839,  6.46560181,
        5.47866221,  5.19986797, 12.13568598,  5.29755487,  7.20826219,
        9.58072286,  8.29022197, 10.60226614,  7.92607342, 13.45642379,
        5.96556248,  6.4020404 , 11.8159254 ,  7.92607342, 12.95702647,
       13.52430176,  9.26455795,  5.47866221,  5.39450305, 10.26545792,
       11.0050379 ,  5.05534609, 11.50791859,  8.44940661, 11.23201619,
        7.09887557, 12.16901615,  6.38105015,  5.19577146, 11.64746882,
       13.59480906,  7.24326992,  9.58072286, 10.06286074,  8.29203356,
        5.39723241, 13.48809595,  5.05989547,  5.37343173, 13.13205357,
       13.83139024, 13.0743399 , 11.28895489,  7.15100303, 10.74409834,
       12.27993098,  6.34059021, 10.68031351,  8.62329466, 11.785566  ,
       10.19036081, 11.10066126, 12.22503754,  5.1224079 ,  5.5205189 ,
       10.37998078, 12.33167382,  5.91677188, 13.48809595,  8.55982105,
       12.5769659 ])}
{'P': tensor([[3.5167e-03, 2.2329e-01, 2.1251e-01,  ..., 8.7463e-08, 8.5164e-06,
         3.1079e-07],
        [3.0870e-05, 6.4574e-03, 3.8005e-02,  ..., 1.3861e-07, 2.1522e-05,
         7.5981e-07],
        [4.2042e-05, 1.0680e-03, 1.0074e-02,  ..., 1.9320e-07, 2.7236e-05,
         1.0889e-06],
        ...,
        [4.4305e-04, 3.0897e-04, 2.4822e-04,  ..., 1.4456e-01, 2.5799e-05,
         4.2922e-05],
        [3.0423e-07, 4.7846e-07, 4.5039e-07,  ..., 6.2148e-07, 2.0466e-02,
         1.4584e-04],
        [5.1516e-06, 4.5464e-06, 3.9670e-06,  ..., 5.6337e-03, 8.3973e-06,
         1.3098e-01]]), 'X': tensor([[  5.6414, -10.0303,   7.8525],
        [  6.8704,  -9.3809,   7.5769],
        [  7.0664,  -9.2590,   8.0879],
        [  6.8093,  -9.4178,   8.4244],
        [  7.5930,  -8.9037,   7.2323],
        [  8.2154,  -8.4262,   8.0171],
        [  4.0642,  -3.7599,  -3.4565],
        [  1.7228,  -4.7575,   8.3396],
        [  5.8930,  12.3592,  -8.9201],
        [  5.4732,  -2.2041,  -4.3830],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.3534,   1.2990,  -7.2942],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.2770,  -0.0379,   2.4195],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -2.0229, -10.6163,   8.6930],
        [  7.3592,  11.3937,   6.2407],
        [  2.1625,  -4.6697,   7.0658],
        [ -7.4602,   4.7718,  -4.6277],
        [ -6.1914,  -8.2997,   8.5380],
        [ 12.5258,  -0.4200,  -9.3493],
        [  4.5800,   5.5034,  -3.9432],
        [  2.3200,   7.1756,   1.1938],
        [  5.3924, -10.1396,  -4.0863],
        [  0.9478,  -4.8144,   0.6371],
        [ -1.6086,   7.8928,   4.3385],
        [  0.1532,  -4.7422,  -7.0145],
        [  6.3535,   0.6792,  -7.5101],
        [ -2.3317,   7.8034,  -3.6274],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.2770,  -0.0379,   2.4195],
        [ -9.4366,  -0.1189,  -2.6736],
        [  8.1217,  -8.5024,  -0.6149],
        [  3.4009,  -4.1874,  -5.9751],
        [ 12.6035,   0.7494,   7.1536],
        [  2.7295,  13.6703,  -8.9328],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.7295,  13.6703,  -8.9328],
        [  1.9503,  -4.7172,  -6.2313],
        [  1.7301,   7.4262,   5.0522],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.9503,  -4.7172,  -6.2313],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -1.6220,   7.8917,   0.0315],
        [  1.6408,   7.4591,  -4.2294],
        [  2.8474,   6.8979,   6.6457],
        [  1.7659,   7.4127,   4.7871],
        [ -8.3535,  -5.5118,   9.8399],
        [  7.1821,  -9.1845,   9.8548],
        [  4.2096,   5.8815,   1.9230],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.4735,  -1.2620,  -3.6742],
        [  1.6408,   7.4591,  -4.2294],
        [  3.3474,  -4.2162,   5.2455],
        [  3.5965,  -4.0754,  -3.5035],
        [  2.0547,   7.2957,  -1.9343],
        [ -9.4766,  -1.1116,  -2.1706],
        [  0.9809,  -4.8145,   6.8706],
        [  6.2630,   2.1057,  -5.2136],
        [ 10.1503,   8.5414,  -7.2681],
        [ -6.4368,   5.8356,   3.3014],
        [  6.0890,  -0.8382,   2.9166],
        [ -7.6293,  -6.6859,  -5.5161],
        [  2.4932,   7.0903,  -5.4214],
        [  5.1886,  -2.6250,   6.4677],
        [  3.6154,  -4.0640,  -7.9698],
        [  2.6970,   6.9826,   9.7350],
        [ -8.7078,   2.7182,   1.8146],
        [  3.9895,   6.0816,  -6.5235],
        [ 12.5650,   2.0132,   4.6272],
        [ 11.5935,   5.9740,  -4.3952],
        [  6.1222,  -0.7261,   8.8204],
        [  8.7715,  -7.9390,   5.2900],
        [  0.1206,  -4.7363,   8.7103],
        [  1.9241,   7.3503,   7.9004],
        [  2.0937,  13.8293,  -2.3533],
        [  7.7794,  11.0629,   8.9069],
        [  0.9967,  -4.8144,  -9.7370],
        [  3.9022,  13.2906,   5.5788],
        [ -7.3554,   4.8985,  -0.6665],
        [  1.9505,  -4.7171,   9.9868],
        [  4.1637,  13.1898,  -1.9189],
        [  2.0243,   7.3086,   7.5432],
        [  1.0093,   7.6567,   2.3526],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.1292,   4.8183,  -5.7997],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.3858,   4.4260,  -9.6769],
        [ 12.5832,   1.7543,  -7.6134],
        [  0.3715,  -4.7756,   0.2312],
        [ 11.3691,   6.4760,   1.0787],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.4739,  -1.2497,   0.5483],
        [ -0.0254, -10.9932,   3.2200],
        [  6.3266,   0.3292,   9.3502],
        [  1.5653,  13.9377,   4.1773],
        [ -3.0644,   7.6439,  -1.4322],
        [ -6.4620,  -8.0449,   6.2719],
        [ -8.8835,  -4.3178,   9.3080],
        [  3.7298,  13.3537,  -9.4653],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -2.1817,   7.8274,  -4.6298]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0645, 0.0874,  ..., 1.0585, 1.3648, 0.9496],
        [0.0645, 0.0000, 0.0326,  ..., 1.0788, 1.3233, 0.9285],
        [0.0874, 0.0326, 0.0000,  ..., 1.1103, 1.3327, 0.9488],
        ...,
        [1.0585, 1.0788, 1.1103,  ..., 0.0000, 1.1989, 0.6554],
        [1.3648, 1.3233, 1.3327,  ..., 1.1989, 0.0000, 0.5740],
        [0.9496, 0.9285, 0.9488,  ..., 0.6554, 0.5740, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.50791859, 11.62770332, 11.64746882, 11.62158554, 11.70168755,
       11.76830981,  5.5366698 ,  5.05981914, 13.6922425 ,  5.90035324,
        9.20340031,  6.48486171,  5.18277858,  6.27715105,  8.90405823,
        9.55720651, 10.80728788, 13.56367857,  5.14607014,  8.85572292,
       10.35464916, 12.53285633,  7.15990676,  7.54127425, 11.48434767,
        4.90676242,  8.0550286 ,  4.74468406,  6.38968357,  8.14433542,
       10.53368644,  6.27715105,  9.43737523, 11.75807252,  5.39450305,
       12.62576026, 13.94008948, 12.05395056, 13.94008948,  5.10444097,
        7.62509582,  8.05242591,  5.10444097,  9.69629   ,  8.05668609,
        7.6374603 ,  7.46248947,  7.62011734, 10.00802627, 11.65923396,
        7.23276582,  5.13616224,  9.55720651,  7.6374603 ,  5.38342077,
        5.43544626,  7.57946459,  9.54154096,  4.91336994,  6.60752373,
       13.26590071,  8.68832398,  6.1463896 , 10.14436912,  7.5158489 ,
        5.81481044,  5.43943791,  7.48538905,  9.12220271,  7.27340834,
       12.72524615, 13.04217487,  6.16512856, 11.83075005,  4.73784222,
        7.59795445, 13.98691426, 13.52430176,  4.91652556, 13.85158417,
        8.83725081,  5.10446653, 13.83139024,  7.58378655,  7.72291946,
       12.7125079 ,  7.03733775,  9.48969978,  5.95297459,  6.97106912,
       12.70489563,  4.79003361, 13.08414689,  7.822854  ,  9.55593374,
       10.99326577,  6.33517349, 14.02532758,  8.23525433, 10.31886066,
        9.87720102, 13.86480175, 11.72802491,  9.79039371, 12.53345445,
        8.12580976])}
{'P': tensor([[5.6298e-03, 2.5053e-01, 2.2421e-01,  ..., 2.1645e-07, 3.6995e-07,
         9.2073e-05],
        [3.0675e-05, 1.0967e-02, 1.2573e-03,  ..., 9.2516e-07, 4.8035e-07,
         5.0103e-05],
        [6.9458e-05, 7.9960e-02, 1.8561e-02,  ..., 1.1669e-06, 8.0821e-07,
         7.8701e-05],
        ...,
        [6.2747e-07, 5.4833e-07, 4.8250e-07,  ..., 3.3444e-01, 9.3051e-05,
         2.0606e-06],
        [9.7100e-04, 3.6694e-04, 3.6814e-04,  ..., 1.9068e-07, 2.8325e-02,
         7.1970e-04],
        [1.9329e-04, 5.9464e-04, 3.5811e-04,  ..., 2.3462e-06, 7.1592e-07,
         1.2166e-02]]), 'X': tensor([[  5.7218,  -9.9934,   6.5764],
        [  7.8425,  -8.7202,   6.4530],
        [  7.5930,  -8.9037,   7.2323],
        [  7.5784,  -8.9141,   5.8410],
        [  8.6420,  -8.0580,   6.5781],
        [  6.8704,  -9.3809,   7.5769],
        [ -8.8400,   2.3991,   1.3457],
        [ 11.7162,   5.6728,  -8.9372],
        [ -9.1706,   1.3863,   8.4032],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.5158,  -2.1337,   3.5246],
        [  2.5469,  13.7192,  -9.6935],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -4.5778,  -9.5013,  -5.0667],
        [  7.8982,  -8.6778,   0.0239],
        [  9.9217,  -6.7006,  -5.6728],
        [ 11.3367,  -4.4806,   0.4643],
        [ -8.8058,   2.4850,  -1.7425],
        [ -3.5029,   7.5138,   8.1853],
        [ -8.9636,   2.0648,   8.1946],
        [ -3.9508,  -9.8515,   1.7923],
        [ -9.4721,  -1.3097,  -7.4377],
        [  6.3535,   0.6792,  -7.5101],
        [  5.4732,  -2.2041,  -4.3830],
        [  3.0876,   6.7527,  -5.3540],
        [  2.0729,   7.2878,   7.0880],
        [ -7.2515,   5.0195,   4.6434],
        [  4.9152,  12.8653,  -1.4459],
        [  2.2849, -10.9655,  -7.6542],
        [ -8.8058,   2.4850,  -1.7425],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.4867,  12.5816,   2.9549],
        [ -6.2668,   5.9808,   2.5614],
        [  5.0233,   4.9647,  -3.1788],
        [  0.9945,  -4.8144,  -0.1897],
        [  7.0515,  11.6194,   8.4289],
        [  1.3670,  -4.7990,   3.2603],
        [ -3.4488, -10.0934,   2.5195],
        [  1.9241,   7.3503,   7.9004],
        [ -9.1960,  -3.2964,   1.6921],
        [  5.2608,  12.6977,   4.9166],
        [  4.1348,   5.9514,  -0.8856],
        [ -6.6998,   5.5954,   0.0823],
        [ -0.5385,   7.9037,   6.3830],
        [ -6.3306,   5.9272,   5.6935],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.3900,   3.3727,   3.1650],
        [  6.3119,   1.7797,   9.9186],
        [ -7.0120,   5.2825,   7.7762],
        [  5.7485,  -9.9811,  -3.2355],
        [ -8.8631,  -4.3728,  -1.1746],
        [  8.4691,  10.4571,  -5.0650],
        [  2.7685, -10.8982,  -6.8928],
        [  6.1973,  -0.4372,   6.5720],
        [ -6.6895,   5.6051,   2.7899],
        [ -8.8058,   2.4850,  -1.7425],
        [ 10.7566,   7.6212,  -2.3279],
        [ -6.4620,  -8.0449,   6.2719],
        [ -8.6681,  -4.8556,   9.2518],
        [  9.3560,   9.5391,   9.4943],
        [  1.3767, -11.0341,   9.9415],
        [  3.6703,   6.3443,  -5.4401],
        [ -5.0550,   6.8231,  -9.2502],
        [ -6.9838,  -7.4965,   0.9796],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.3539,  -5.5111,   0.3373],
        [  1.6308,  13.9254,  -2.0984],
        [  6.0467,  -9.8369,  -6.5816],
        [  3.9591, -10.6391,  -6.2557],
        [ -4.6393,   7.0461,   1.1624],
        [ -5.3328,  -8.9992,   7.3440],
        [  6.3124,   0.2057,  -0.4593],
        [ -9.1960,  -3.2964,   1.6921],
        [ -1.7009,   7.8851,  -8.2437],
        [ -3.4986,   7.5152,   9.0712],
        [ -6.6895,   5.6051,   2.7899],
        [  3.2647, -10.8066,  -0.3768],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.2902,   0.0451,   3.7116],
        [  1.3767, -11.0341,   9.9415],
        [ -2.9497, -10.3028,  -7.3479],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.8154,   2.4611,   2.5198],
        [  0.3231,  -4.7691,  -1.6120],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 11.7162,   5.6728,  -8.9372],
        [  1.6954,  -4.7616,   9.8198],
        [  2.3261,  13.7749,   5.9121],
        [  2.2076,  13.8032,  -1.5099],
        [ -1.9714,   7.8563,  -1.8084],
        [  7.0171,  11.6439,  -5.0178],
        [ -2.4956, -10.4682,   3.5632],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.8154,   2.4611,   2.5198],
        [ -5.9938,  -8.4745,   3.9735],
        [ -7.0120,   5.2825,   7.7762],
        [  2.3889,  -4.6082,  -8.7291],
        [ -7.5401,   4.6718,   9.0033],
        [  6.1685,  12.1979,  -6.2473],
        [ -8.8363,   2.4086,  -5.6296],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.6248,   3.9984,   8.4933],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.7485,  -9.9811,  -3.2355]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1163, 0.1152,  ..., 1.2617, 0.6399, 0.8431],
        [0.1163, 0.0000, 0.0379,  ..., 1.2069, 0.7333, 0.8400],
        [0.1152, 0.0379, 0.0000,  ..., 1.2437, 0.7487, 0.8769],
        ...,
        [1.2617, 1.2069, 1.2437,  ..., 0.0000, 1.3219, 0.6120],
        [0.6399, 0.7333, 0.7487,  ..., 1.3219, 0.0000, 0.7184],
        [0.8431, 0.8400, 0.8769,  ..., 0.6120, 0.7184, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.51557166, 11.72802491, 11.70168755, 11.70016217, 11.8159254 ,
       11.62770332,  9.15976746, 13.01728313,  9.27474386,  4.90505566,
        5.91408504, 13.95361098, 12.4217523 ,  9.66977802, 10.54658597,
       11.73397309, 11.97238624, 12.18998195,  9.14973319,  8.29022197,
        9.19837792, 10.61417181,  9.562174  ,  6.38968357,  5.90035324,
        7.42512621,  7.57685908,  8.81930525, 13.77222752, 11.20100795,
        9.14973319, 12.5085995 , 13.72594022,  8.66273171,  7.06272256,
        4.9160962 , 13.59170809,  4.98989906, 10.66631784,  7.59795445,
        9.76896904, 13.744387  ,  7.24677664,  8.72896381,  7.92200342,
        8.67227386,  5.13125513,  9.04255655,  6.55800267,  8.77913428,
       11.51811603,  9.88310614, 13.45642379, 11.24434374,  6.21275016,
        8.7273569 ,  9.14973319, 13.18278606, 10.31886066,  9.9353921 ,
       13.3614598 , 11.11969815,  7.32950385,  8.49161154, 10.24557047,
       13.50484539, 10.00794468, 14.02058621, 11.54672755, 11.35182589,
        8.43626114, 10.4606265 ,  6.31576264,  9.76896904,  8.06643487,
        8.28968067,  8.7273569 , 11.28895489, 10.39942497,  6.29034958,
       11.11969815, 10.7167285 , 10.30972193,  9.15253344,  4.78002665,
       10.00410625, 13.01728313,  5.05444288, 13.96987806, 13.97857591,
        8.09984231, 13.59480906, 10.76154256,  8.83725081,  9.15253344,
       10.37998078,  8.77913428,  5.19065439,  8.87007495, 13.66896739,
        9.15866607, 11.63724792,  6.90115901, 13.99048094,  9.99211663,
       11.51811603])}
{'P': tensor([[4.6426e-03, 3.3431e-01, 3.0456e-01,  ..., 9.4728e-04, 7.9988e-05,
         5.5409e-06],
        [4.9824e-05, 7.8212e-03, 1.8823e-01,  ..., 2.2851e-03, 9.7532e-05,
         1.3782e-05],
        [4.1459e-05, 2.0623e-04, 8.8606e-03,  ..., 2.3649e-03, 9.3537e-05,
         1.6224e-05],
        ...,
        [1.7340e-06, 2.3801e-06, 2.7519e-06,  ..., 1.4471e-02, 7.5090e-06,
         8.3575e-03],
        [1.6676e-02, 1.7835e-02, 1.8429e-02,  ..., 1.8481e-03, 1.0939e-01,
         7.2349e-05],
        [4.6526e-07, 5.2020e-07, 5.0867e-07,  ..., 6.5682e-06, 6.6535e-06,
         1.0072e-01]]), 'X': tensor([[  5.7485,  -9.9811,  -3.2355],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.9653,  -9.3225,  -3.4727],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.3424,  -7.3685,  -3.6807],
        [  8.2422,  -8.4041,  -1.4962],
        [ -8.0087,   4.0173,   9.1091],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.2209,  -4.7092,   2.6117],
        [ -8.9636,   2.0648,   8.1946],
        [  0.6142,  14.0802,   0.5407],
        [  6.7792,  11.8084,   1.6251],
        [  5.7784,  -1.6401,   9.5587],
        [  2.2373,  -4.6506,  -1.7334],
        [ -0.4341, -10.9473,   7.4509],
        [  5.1890,   4.7318,   0.9515],
        [  6.1989,  -9.7591,   0.8289],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -8.0611,  -6.0302,   4.6962],
        [  9.8082,  -6.8396,   4.9615],
        [ -6.6998,   5.5954,   0.0823],
        [ -0.1963,   7.8765,  -1.1371],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.1749,  -3.3803,   6.4629],
        [ 12.2592,   3.9425,   2.0383],
        [ -8.9178,   2.1936,   7.5388],
        [  5.3602,  -2.3808,   7.2098],
        [ -4.4520,   7.1370,  -8.7867],
        [  1.3670,  -4.7990,   3.2603],
        [  2.9470,  13.6085,   2.9809],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 12.5224,  -0.4497,   4.8448],
        [ -4.8751,  -9.3148,   4.0406],
        [  0.9659,  -4.8144,  -0.9594],
        [  3.7298,  13.3537,  -9.4653],
        [  6.3200,   0.2690,   1.5900],
        [  7.5784,  -8.9141,   5.8410],
        [  4.0587, -10.6112,   6.2273],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.7078,   2.7182,   1.8146],
        [  3.9085,   6.1512,  -2.2560],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.1730,  -0.5375,  -8.8861],
        [ 10.8218,  -5.4177,   6.8573],
        [ -2.1382,   7.8339,   7.3524],
        [ 12.4901,  -0.7052,   1.7315],
        [ 10.2470,   8.4054,  -0.1177],
        [ 12.2435,   4.0076,   8.1257],
        [ 11.5935,   5.9740,  -4.3952],
        [ -1.9714,   7.8563,  -1.8084],
        [ -8.6603,   2.8250,   5.9181],
        [  5.1890,   4.7318,   0.9515],
        [  3.8182,  -3.9347,  -1.7961],
        [ 11.2517,  -4.6495,   2.9800],
        [ -6.7992,   5.4991,   9.9186],
        [ 11.6634,   5.8050,   9.1590],
        [  0.9809,  -4.8145,   6.8706],
        [  1.8447,  -4.7372,   8.1235],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.1047, -11.0046,   0.5188],
        [ -2.6069, -10.4298,  -6.1995],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.1658,  -3.4155,  -2.8880],
        [  4.3459,  -3.5327,   0.3667],
        [  8.8091,  -7.9037,   0.3631],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.1292,   4.8183,  -5.7997],
        [ -4.6393,   7.0461,   1.1624],
        [  0.2571,  -4.7594,   1.5618],
        [ -4.2271,   7.2387,  -1.8316],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.4429,  -2.2529,   9.7306],
        [  6.3497,   1.3680,  -1.9076],
        [  3.9913,  -3.8139,  -6.8311],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -0.8569,   7.9156,  -6.5557],
        [  0.1047, -11.0046,   0.5188],
        [  2.2659,  13.7894,   6.5436],
        [  4.9152,  12.8653,  -1.4459],
        [  0.6757, -11.0361,  -1.3047],
        [  3.4604,  -4.1545,   1.8091],
        [  2.4697,  -4.5836,   1.4593],
        [  4.2612, -10.5514,  -6.6268],
        [  3.9895,   6.0816,  -6.5235],
        [  2.8602,   6.8905,   5.8779],
        [ -3.2053, -10.1993,  -3.8736],
        [ 12.4837,   2.7510,   0.4449],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 10.0400,   8.6922,  -6.2847],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.1960,  -3.2964,   1.6921],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -2.2508, -10.5479,  -2.7235],
        [  6.0047,   3.1062,  -6.7914],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -7.9667,   4.0814,  -0.6397],
        [  5.4732,  -2.2041,  -4.3830],
        [ 10.0995,  -6.4739,   2.0974],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.1755,   2.5233,  -2.0042],
        [ 12.2692,  -1.8882,   3.9253],
        [  0.9392,  -4.8143,  -3.4649],
        [  9.6662,   9.1725,   8.1056]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0576, 0.0884,  ..., 0.7589, 0.2849, 1.2681],
        [0.0576, 0.0000, 0.0480,  ..., 0.7196, 0.2274, 1.2273],
        [0.0884, 0.0480, 0.0000,  ..., 0.7413, 0.2102, 1.2465],
        ...,
        [0.7589, 0.7196, 0.7413,  ..., 0.0000, 0.6065, 0.5097],
        [0.2849, 0.2274, 0.2102,  ..., 0.6065, 0.0000, 1.0925],
        [1.2681, 1.2273, 1.2465,  ..., 0.5097, 1.0925, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.51811603, 11.59512194, 11.63724792, 11.80092578, 11.89855267,
       11.77124373,  8.95983586, 13.66051884, 12.16901615,  9.19837792,
       14.09357547, 13.61600536,  6.00663314,  5.16077764, 10.95593729,
        7.02253235, 11.56146873,  9.43737523, 10.06703358, 11.95743558,
        8.72896381,  7.8788965 ,  9.69629   ,  9.77777346, 12.87752052,
        9.18358562,  5.86519525,  8.41171369,  4.98989906, 13.92390689,
       12.32545637, 12.53047489, 10.51340829,  4.91038884, 13.86480175,
        6.32572323, 11.70016217, 11.36089355,  5.91677188,  9.12220271,
        7.28793663, 10.48315436,  6.19633139, 12.10222204,  8.12043983,
       12.50996828, 13.25336001, 12.88270018, 13.04217487,  8.09984231,
        9.10945678,  7.02253235,  5.48276038, 12.17451685,  8.74470218,
       13.02818702,  4.91336994,  5.08372149, 10.52357479, 11.00508681,
       10.7506423 ,  6.60752373,  9.78147542,  5.60064319, 11.83509372,
       12.41012046,  7.03733775,  8.43626114,  4.7663526 ,  8.38252937,
       10.99326577, 10.00410625,  9.7712321 ,  5.89074104,  6.49538841,
        5.5205189 , 12.4980845 ,  9.60080639,  7.96181288, 11.00508681,
       13.97430191, 13.77222752, 11.05672136,  5.40689332,  5.20659892,
       11.37939695,  7.27340834,  7.46053281, 10.6910768 , 12.78327037,
       10.32715715, 13.27994304,  7.72668467,  9.76896904, 12.42554808,
       10.78533838,  6.76056625, 10.39942497,  8.95133961,  5.90035324,
       11.99633819,  9.88571713,  6.67108662, 12.41366754,  4.90505566,
       13.32557133])}
{'P': tensor([[1.0306e-02, 1.5559e-01, 1.5306e-01,  ..., 5.7244e-04, 1.2553e-03,
         1.2294e-04],
        [7.2525e-06, 4.1450e-03, 8.1278e-04,  ..., 2.2430e-04, 2.1359e-04,
         1.2738e-04],
        [7.5281e-05, 1.3935e-02, 3.7706e-02,  ..., 6.8072e-04, 5.9176e-03,
         3.1362e-04],
        ...,
        [1.7843e-04, 7.7516e-04, 3.1905e-04,  ..., 5.6040e-02, 3.1731e-04,
         4.2175e-02],
        [7.6218e-04, 1.8489e-03, 2.9894e-03,  ..., 1.1818e-03, 2.8414e-01,
         4.8145e-04],
        [3.1912e-05, 9.3790e-05, 4.3906e-05,  ..., 2.4645e-03, 8.1541e-05,
         1.1528e-01]]), 'X': tensor([[ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.2209, 0.2506,  ..., 0.6448, 0.4808, 0.7671],
        [0.2209, 0.0000, 0.3385,  ..., 0.4382, 0.4580, 0.5645],
        [0.2506, 0.3385, 0.0000,  ..., 0.7622, 0.2622, 0.8900],
        ...,
        [0.6448, 0.4382, 0.7622,  ..., 0.0000, 0.7883, 0.1278],
        [0.4808, 0.4580, 0.2622,  ..., 0.7883, 0.0000, 0.9101],
        [0.7671, 0.5645, 0.8900,  ..., 0.1278, 0.9101, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.52733581, 11.87412439, 11.85743425, 11.92133129, 11.99633819,
       11.83075005,  7.66377554,  8.25828409, 13.40024224,  6.84796202,
        9.96152788, 11.18839169, 11.00508681, 13.13205357,  8.09899874,
        9.79039371, 14.02058621, 10.14044489, 11.85743425, 11.80092578,
       12.16901615, 11.18839169,  9.78147542,  5.70405525,  4.78002665,
       10.0674887 ,  5.0713364 ,  5.5513072 , 13.26209545,  4.90505566,
        7.9337647 ,  8.31625171, 13.3426541 , 11.05672136,  9.37989631,
        6.652742  , 10.01804223, 13.01728313,  7.72291946,  6.80559669,
        7.37902766,  8.67227386,  7.18354885,  7.62509582,  9.26455795,
        5.29755487,  7.60529892, 11.11969815,  8.57560275, 12.5769659 ,
        6.5618968 ,  4.89869205,  9.09502385,  6.23262289, 10.33712258,
       13.3607379 ,  5.79391825, 12.18045978, 14.03505624,  5.84533345,
       13.04217487, 13.09426142,  8.51088913,  5.9510489 ,  9.04255655,
        7.06272256,  4.91038884,  9.7712321 , 10.63038565,  7.58378655,
        6.38105015, 14.09176259, 11.00694688,  6.1463896 ,  6.39520397,
       10.27716434, 12.35421461,  9.79039371,  7.5984809 , 13.97857591,
       13.92390689, 11.99633819,  7.57946459,  6.97106912, 14.10796466,
       12.85032836,  6.78325151,  4.978704  ,  9.94173708,  5.84533345,
       10.54658597,  5.87305005,  9.15253344,  5.14999799, 10.37698435,
       10.39942497, 10.61472636, 10.80728788,  7.72668467,  6.28961988,
       11.04953767,  7.50917551, 13.68063098,  5.79391825,  5.92952538,
        6.38968357])}
{'P': tensor([[5.3458e-03, 3.4362e-01, 3.3001e-01,  ..., 4.7104e-07, 1.5396e-03,
         2.8733e-06],
        [9.0735e-05, 8.2465e-03, 4.1479e-01,  ..., 6.1692e-07, 2.5329e-03,
         4.8028e-06],
        [1.1831e-04, 1.8572e-04, 2.5253e-02,  ..., 1.4939e-06, 8.5693e-03,
         1.7831e-05],
        ...,
        [3.5340e-04, 3.1238e-04, 3.3970e-04,  ..., 3.3329e-02, 3.6841e-05,
         6.5212e-07],
        [1.0392e-04, 1.1324e-04, 1.4645e-04,  ..., 4.0652e-06, 5.9496e-02,
         2.1854e-03],
        [1.3822e-05, 1.3190e-05, 1.1175e-05,  ..., 8.7582e-06, 1.2792e-04,
         9.5749e-02]]), 'X': tensor([[ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0352, 0.0891,  ..., 1.1496, 0.3918, 0.0423],
        [0.0352, 0.0000, 0.0590,  ..., 1.1712, 0.3612, 0.0227],
        [0.0891, 0.0590, 0.0000,  ..., 1.1717, 0.3029, 0.0474],
        ...,
        [1.1496, 1.1712, 1.1717,  ..., 0.0000, 1.2697, 1.1538],
        [0.3918, 0.3612, 0.3029,  ..., 1.2697, 0.0000, 0.3496],
        [0.0423, 0.0227, 0.0474,  ..., 1.1538, 0.3496, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.54071859, 11.57249984, 11.66453059, 11.81423161, 11.86239961,
       11.75618513, 10.794755  ,  6.36004393,  9.17016857,  7.62011734,
       12.3344536 , 12.50996828, 10.48315436, 13.2778378 ,  6.42689705,
        9.88310614,  9.60080639, 10.7506423 ,  5.19065439, 13.2778378 ,
        6.06416957, 13.35832725, 10.7506423 ,  8.66273171,  5.68942572,
       12.76408562,  7.06169391, 13.12218013,  9.68323686,  9.44486609,
        5.53255882,  6.1463896 ,  7.5092834 , 14.03505624,  6.58184474,
        9.55593374,  7.03733775,  8.56889738,  7.59795445,  5.19065439,
       12.46513646,  9.86089579, 10.00802627,  5.86519525,  9.78285424,
        6.16512856, 10.48721382,  6.33075682,  8.22728399, 12.45418236,
        5.29755487, 10.5575601 , 11.83570132, 14.09357547,  4.91038884,
        5.1805345 ,  7.8788965 , 10.91601655,  7.16747878,  6.25586297,
        6.05165248,  7.48538905, 10.1979669 ,  9.49511524,  6.77777606,
        7.50917551,  9.86089579,  8.72896381,  8.09704052,  8.72896381,
       12.91505111,  4.85603886,  9.7712321 , 13.52430176,  7.85919687,
        7.80625981, 10.01804223,  7.20482224, 13.13205357,  5.9510489 ,
        4.9160962 ,  5.80860165,  7.83534987,  7.73151309, 11.69264243,
       10.42667093, 11.83509372,  9.41605684, 11.51557166, 10.14044489,
       11.67875855,  5.53255882,  8.05242591,  5.08910491, 11.33837396,
        6.14086582,  8.38252937, 13.35013374, 11.99633819, 13.0743399 ,
        8.50278896,  7.49076899,  7.38844703, 10.67322527,  6.14086582,
        8.06643487])}
{'P': tensor([[5.7763e-03, 2.1214e-01, 1.9734e-01,  ..., 4.4139e-05, 2.9976e-06,
         1.5567e-04],
        [1.1488e-04, 1.5714e-02, 1.1556e-04,  ..., 9.4054e-05, 1.0258e-05,
         8.8905e-04],
        [1.4468e-04, 3.4574e-01, 6.8738e-03,  ..., 5.0960e-05, 3.7294e-06,
         2.2125e-04],
        ...,
        [2.7887e-03, 3.9415e-03, 3.1197e-03,  ..., 6.9998e-02, 9.7910e-05,
         7.3187e-04],
        [5.3086e-06, 4.2244e-06, 5.0210e-06,  ..., 5.3045e-05, 3.6591e-02,
         3.2761e-06],
        [2.3717e-07, 3.1054e-07, 2.5227e-07,  ..., 4.0683e-06, 2.9851e-04,
         1.2858e-02]]), 'X': tensor([[  6.0467,  -9.8369,  -6.5816],
        [  7.2340,  -9.1505,  -5.8321],
        [  6.3122,  -9.6994,  -6.3282],
        [  8.6271,  -8.0715,  -7.4813],
        [  9.0424,  -7.6779,  -7.5286],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -5.8389,  -8.6056,  -3.5098],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.7963,  -1.6018,   1.8856],
        [  3.3474,  -4.2162,   5.2455],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.1931,  -4.7491,   7.4385],
        [  9.0424,  -7.6779,  -7.5286],
        [ -5.0550,   6.8231,  -9.2502],
        [ -0.0254, -10.9932,   3.2200],
        [ 11.2517,  -4.6495,   2.9800],
        [  0.1252, -11.0062,   7.2564],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.0438,   7.1180,  -5.6991],
        [ -0.7599, -10.8995,  -6.0366],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.5620,   7.0549,   8.4825],
        [ -5.9987,   6.1949,  -6.2114],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 12.5996,   1.3971,  -1.0694],
        [  9.0051,   9.9234,   5.4411],
        [ -0.8569,   7.9156,  -6.5557],
        [  1.7500,   7.4187,   4.4237],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.4296,  -1.9458,   4.0833],
        [  6.9284,  11.7061,  -0.3732],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -3.5029,   7.5138,   8.1853],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.1041, -11.0045,   2.4311],
        [ -8.5506,  -5.1154,   6.3423],
        [ 11.9808,   4.9340,  -3.3715],
        [ -3.4100,   7.5436,  -7.6980],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.1372,   2.6726,  -6.7885],
        [ 11.2939,   6.6325,  -5.8311],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.0190,   4.9705,  -7.9752],
        [  5.6414, -10.0303,   7.8525],
        [  2.9470,  13.6085,   2.9809],
        [ -9.3705,   0.4208,  -3.5815],
        [ 11.7162,   5.6728,  -8.9372],
        [  0.3038,   7.8092,   0.4164],
        [  2.6613,   7.0021,   4.5783],
        [ -6.7992,   5.4991,   9.9186],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.0588,  -6.0341,  -2.3006],
        [  6.2883,   0.0325,   5.2692],
        [ 11.9936,  -2.8677,   6.0614],
        [  1.8721,  -4.7322,   0.5524],
        [  2.3589,  13.7668,  -3.6501],
        [ -4.4503,   7.1378,   3.3751],
        [  6.3266,   0.3292,   9.3502],
        [ -9.1459,   1.4780,  -4.2403],
        [  1.3103,  -4.8032,  -0.9300],
        [ -9.4117,   0.1160,   8.2574],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.6523,   3.9443,  -2.9449],
        [  1.8320,   7.3871,   5.1452],
        [  5.8930,  12.3592,  -8.9201],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.2188,  -2.5839,   2.1833],
        [  8.4691,  10.4571,  -5.0650],
        [  5.9026,   3.3837,  -0.6573],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.1519, -10.2387,   9.8191],
        [ -6.6895,   5.6051,   2.7899],
        [ 10.7566,   7.6212,  -2.3279],
        [ -8.4824,  -5.2575,  -0.0795],
        [  7.1821,  -9.1845,   9.8548],
        [ -3.9456,  -9.8542,   0.7466],
        [  4.3459,  -3.5327,   0.3667],
        [ -7.8126,   4.3066,   5.1791],
        [  5.8275,  -1.5334,   2.0124],
        [  2.3200,   7.1756,   1.1938],
        [  1.6232,  -4.7717,  -2.5655],
        [  6.2790,  -0.0259,  -5.4116],
        [ -5.2752,  -9.0409,  -8.2568],
        [  9.9756,  -6.6330,  -3.3429],
        [ -4.2271,   7.2387,  -1.8316],
        [  6.2576,   2.1359,  -3.3398],
        [ -5.0943,  -9.1682,   0.5073],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.1292,   4.8183,  -5.7997],
        [  6.2790,  -0.0259,  -5.4116],
        [ 10.9931,   7.2107,   7.7933],
        [ 10.0619,   8.6627,  -9.2884],
        [  8.8091,  -7.9037,   0.3631],
        [ -1.0407,   7.9166,  -6.6021],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.4009,  -4.1874,  -5.9751],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 12.2435,   4.0076,   8.1257],
        [  6.2535,  -0.1709,  -9.4102],
        [  1.0391,  -4.8141,  -0.0768],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.7418,   5.6070,  -1.4192]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0778, 0.0188,  ..., 0.5210, 0.0184, 0.6804],
        [0.0778, 0.0000, 0.0590,  ..., 0.4609, 0.0635, 0.6032],
        [0.0188, 0.0590, 0.0000,  ..., 0.5055, 0.0102, 0.6616],
        ...,
        [0.5210, 0.4609, 0.5055,  ..., 0.0000, 0.5026, 0.3441],
        [0.0184, 0.0635, 0.0102,  ..., 0.5026, 0.0000, 0.6641],
        [0.6804, 0.6032, 0.6616,  ..., 0.3441, 0.6641, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.54672755, 11.66453059, 11.57249984, 11.81423161, 11.86239961,
       11.75618513,  9.5014146 , 10.39942497,  5.01114719,  6.01356004,
        5.38342077,  5.35774573,  4.75303127, 11.86239961,  8.49161154,
       10.99326577, 12.17451685, 11.00694688,  6.54615887, 13.13888841,
       10.92597196,  9.78286733,  7.50564223,  8.62329466, 10.27716434,
       12.67680439, 13.40024224,  7.96181288,  7.62232307,  9.58072286,
        9.6282763 , 13.60275211, 12.00995898,  8.97052115,  9.51074763,
        8.29022197,  7.8575648 , 11.0050379 ,  9.96391045, 12.95702647,
        8.27853076,  8.05242591,  6.69389431, 13.09737053, 12.67680439,
       12.64235359,  7.06372172, 11.50791859, 13.92390689,  9.37989631,
       13.01728313,  7.81509631,  7.49076899,  8.74470218, 10.0674887 ,
       10.0674887 ,  6.28835101, 12.33167382,  5.08910491, 13.96746816,
        8.41148722,  6.33517349,  9.26455795,  4.978704  ,  9.41244865,
       10.74409834,  6.89245352,  7.61088125, 13.6922425 ,  8.56889738,
        5.82345867, 13.45642379,  6.80370254, 12.00995898, 11.46176259,
        8.7273569 , 13.18278606,  9.97964013, 11.65923396, 10.61472636,
        5.60064319,  8.92098752,  6.02588517,  7.54127425,  5.04028538,
        6.27905292, 10.46739073, 11.97957425,  8.38252937,  6.61211224,
       10.48840476,  8.47371868,  7.03733775,  6.27905292, 13.14690137,
       13.27717982, 11.83509372,  7.98469166,  8.49161154,  5.39450305,
       10.1979669 , 12.88270018,  6.25586297,  4.92497514,  8.08320235,
       13.01187484])}
{'P': tensor([[5.5753e-03, 2.0484e-01, 1.7478e-01,  ..., 1.7337e-04, 2.0041e-05,
         5.0595e-02],
        [2.0441e-05, 4.6584e-03, 3.1798e-01,  ..., 7.6660e-05, 5.1108e-06,
         1.1269e-01],
        [3.6328e-05, 8.9276e-05, 1.7336e-02,  ..., 2.5383e-04, 1.2101e-05,
         2.4537e-01],
        ...,
        [2.0324e-03, 2.6307e-03, 3.8285e-03,  ..., 4.2759e-01, 6.0553e-04,
         3.0710e-03],
        [1.5326e-01, 1.3281e-01, 7.5799e-02,  ..., 2.2320e-04, 1.2141e-02,
         4.1736e-02],
        [2.4860e-05, 5.6641e-05, 1.9316e-04,  ..., 1.1190e-04, 7.1830e-06,
         1.9861e-02]]), 'X': tensor([[ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[5.3633e-05, 1.0061e-01, 1.5962e-01,  ..., 6.3467e-01, 2.0020e-01,
         2.6171e-01],
        [1.0061e-01, 0.0000e+00, 7.2199e-02,  ..., 6.2006e-01, 2.5766e-01,
         1.6328e-01],
        [1.5962e-01, 7.2199e-02, 0.0000e+00,  ..., 5.6712e-01, 3.2985e-01,
         1.4285e-01],
        ...,
        [6.3467e-01, 6.2006e-01, 5.6712e-01,  ..., 0.0000e+00, 8.3215e-01,
         6.7172e-01],
        [2.0020e-01, 2.5766e-01, 3.2985e-01,  ..., 8.3215e-01, 0.0000e+00,
         3.6664e-01],
        [2.6171e-01, 1.6328e-01, 1.4285e-01,  ..., 6.7172e-01, 3.6664e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.56146873, 11.73397309, 11.83509372, 11.69264243, 11.83570132,
       11.87412439,  8.72896381,  5.83560009,  8.7907923 , 10.48840476,
        6.74872981,  7.8575648 , 13.77604293,  6.01356004, 12.97370371,
       12.88270018,  8.01009083,  9.63742669, 12.52372629,  9.61942445,
        6.80370254,  8.7907923 , 10.66631784, 11.73848065,  6.25586297,
       13.3614598 , 13.61600536,  7.83486943, 10.38897563, 11.63502679,
       12.21532282,  4.90505566,  8.87007495, 10.37698435, 11.11969815,
       14.0393949 , 13.20988996,  8.7907923 ,  9.49511524,  8.68979052,
        8.1303945 ,  7.32950385,  7.15990676, 14.09357547,  5.18277858,
       11.62158554,  8.74470218, 11.28895489,  8.49161154, 12.50996828,
        5.53175275, 11.1913598 , 13.6922425 , 13.20988996,  6.27715105,
        6.77777606, 13.96987806, 13.41713661,  5.5671389 ,  5.19986797,
       12.18998195,  8.29640613,  8.66273171,  6.76661595, 13.99877413,
       13.3426541 ,  8.81930525,  5.13182506,  7.16747878, 10.06703358,
       12.62711252,  6.67218687,  7.62509582,  6.80370254,  5.24461068,
        8.40445872, 13.56367857, 12.49317563, 13.91198191, 10.76154256,
        8.05242591, 10.74655346,  6.10774544, 12.35625694,  7.54620849,
        9.79039371,  5.22903117,  7.61088125,  5.14371005, 12.46513646,
        4.79003361, 10.06407839,  6.33075682,  7.83486943, 10.68031351,
        7.76457095,  7.70897675, 11.0050379 , 13.35013374, 12.32545637,
        7.92200342,  7.24677664, 11.07637531,  5.64500512, 11.2845608 ,
       11.97949273])}
{'P': tensor([[5.9247e-03, 2.9801e-01, 1.4829e-01,  ..., 2.2093e-07, 1.6764e-06,
         3.0491e-07],
        [8.3360e-05, 1.1335e-02, 1.4769e-01,  ..., 4.5338e-07, 4.5888e-06,
         5.3190e-07],
        [2.1449e-05, 7.0737e-05, 7.0219e-03,  ..., 2.3677e-07, 2.4104e-06,
         2.9900e-07],
        ...,
        [2.4882e-05, 1.9584e-05, 8.8922e-06,  ..., 3.8656e-02, 1.5410e-05,
         4.9777e-03],
        [9.9402e-07, 1.0791e-06, 7.3482e-07,  ..., 1.2881e-03, 4.1719e-02,
         3.0722e-05],
        [1.0860e-03, 7.1195e-04, 3.7094e-04,  ..., 6.3392e-04, 7.9496e-06,
         1.6268e-01]]), 'X': tensor([[  6.3122,  -9.6994,  -6.3282],
        [  7.2340,  -9.1505,  -5.8321],
        [  8.6271,  -8.0715,  -7.4813],
        [  9.0424,  -7.6779,  -7.5286],
        [  9.9217,  -6.7006,  -5.6728],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.0031,  12.2958,  -6.5992],
        [  4.1297,  -3.7098,  -3.9838],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.7295,  13.6703,  -8.9328],
        [  1.5662,  13.9375,   4.5142],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.4429,  -2.2529,   9.7306],
        [  4.6357,   5.4415,   7.9067],
        [  4.5325,  -3.3643,   6.7699],
        [ -9.1749,  -3.3803,   6.4629],
        [  1.9312,  13.8649,   8.3563],
        [ -8.8400,   2.3991,   1.3457],
        [ 11.1085,  -4.9204,   0.0736],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.1929,   2.4501,  -6.9352],
        [ 10.6283,  -5.7253,   7.4878],
        [ -5.1046,  -9.1610,   2.3418],
        [  9.6816,   9.1536,  -5.8543],
        [  3.1756, -10.8247,   6.8882],
        [  3.1756, -10.8247,   6.8882],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.4671,  -0.5553,   0.1117],
        [  6.3604,   0.9042,  -6.5901],
        [  1.5653,  13.9377,   4.1773],
        [ -5.2589,   6.7024,  -5.5809],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -8.8934,   2.2597,   3.5831],
        [  8.8474,  10.0868,  -8.8644],
        [ -3.4986,   7.5152,   9.0712],
        [  5.5101,  -2.1433,   9.5442],
        [  0.9478,  -4.8144,   0.6371],
        [ 11.5198,  -4.0923,   5.5088],
        [  1.8721,  -4.7322,   0.5524],
        [  1.5383, -11.0274,  -4.9666],
        [  1.6954,  -4.7616,   9.8198],
        [ -8.9379,  -4.1655,  -2.8114],
        [  8.0381,  -8.5690,  -8.9523],
        [ -3.2053, -10.1993,  -3.8736],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.7813,  -3.1139,   8.2691],
        [ -8.8934,   2.2597,   3.5831],
        [  0.7685,  -4.8098,  -2.9310],
        [  8.6420,  -8.0580,   6.5781],
        [  1.1274,  14.0116,   8.1753],
        [  4.2339,  -3.6268,   6.3125],
        [ -1.6086,   7.8928,   4.3385],
        [  3.9895,   6.0816,  -6.5235],
        [  5.5294,  -2.1106,   9.0626],
        [  1.7500,   7.4187,   4.4237],
        [  0.9036,   7.6840,   9.9237],
        [ -2.2188,   7.8218,   8.6064],
        [  5.1142,  -2.7230,  -4.1966],
        [  3.7992,  -3.9474,  -8.9019],
        [  1.3403,  -4.8011,  -1.1698],
        [ -5.1046,  -9.1610,   2.3418],
        [  0.9809,  -4.8145,   6.8706],
        [  1.8766,   7.3694,   1.1887],
        [  3.4920,  -4.1366,  -8.3955],
        [ 11.7726,  -3.4876,  -6.7402],
        [  8.6271,  -8.0715,  -7.4813],
        [  5.2608,  -2.5256,  -9.1446],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.8450,  -9.9356,   3.0767],
        [ -0.8455, -10.8853,  -6.7400],
        [ -8.9606,  -4.0994,   1.7886],
        [ -7.7239,   4.4298,   3.1216],
        [  4.0421,  -3.7765,  -8.3631],
        [  2.1379,  -4.6757,   6.4981],
        [  2.5375,   7.0676,  -4.3490],
        [  0.2480, -11.0153,  -3.0183],
        [ -0.2586,   7.8825,  -9.9840],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.3124,   0.2057,  -0.4593],
        [ 11.2209,  -4.7092,   2.6117],
        [  0.1047, -11.0046,   0.5188],
        [  3.2159, -10.8166,  -1.0406],
        [ -0.5709,   7.9055,   7.6089],
        [  0.1047, -11.0046,   0.5188],
        [  5.5101,  -2.1433,   9.5442],
        [  0.8700,  -4.8132,  -7.8042],
        [ 10.3251,   8.2928,  -3.0006],
        [ -6.1914,  -8.2997,   8.5380],
        [  5.4867,  12.5816,   2.9549],
        [ 10.3251,   8.2928,  -3.0006],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.1666,   3.7653,   3.6075],
        [ -3.9508,  -9.8515,   1.7923],
        [  1.7659,   7.4127,   4.7871],
        [  3.4604,  -4.1545,   1.8091],
        [  6.3587,   1.1596,  -3.6629],
        [  5.8889,  -1.3916,   8.8993],
        [  0.8572,  14.0501,  -8.3364],
        [  6.9314,  -9.3435,  -1.2729],
        [  3.5284, -10.7484,   5.3347],
        [ -8.8400,   2.3991,   1.3457],
        [  2.0243,   7.3086,   7.5432],
        [ -9.4524,  -1.6813,  -6.6862]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0590, 0.1537,  ..., 0.7284, 1.0093, 0.6493],
        [0.0590, 0.0000, 0.1196,  ..., 0.7426, 0.9773, 0.7020],
        [0.1537, 0.1196, 0.0000,  ..., 0.8612, 1.0491, 0.7976],
        ...,
        [0.7284, 0.7426, 0.8612,  ..., 0.0000, 0.7267, 0.5886],
        [1.0093, 0.9773, 1.0491,  ..., 0.7267, 0.0000, 1.2577],
        [0.6493, 0.7020, 0.7976,  ..., 0.5886, 1.2577, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.57249984, 11.66453059, 11.81423161, 11.86239961, 11.97238624,
       11.75618513, 13.6829843 ,  5.5513072 ,  9.78285424, 13.94008948,
       14.02526505, 12.49317563,  5.89074104,  7.14837514,  5.64464807,
        9.77777346, 13.99877413,  9.15976746, 12.1494023 ,  5.63824971,
        6.65991877, 12.07225469, 10.48721382, 13.32374056, 11.28093507,
       11.28093507, 10.80728788,  9.48337063,  6.42439344, 14.02532758,
        8.51928531, 12.25806559,  9.17595899, 13.41713661,  8.28968067,
        5.91222346,  4.90676242, 12.22503754,  5.08910491, 11.1341798 ,
        5.05444288,  9.86089579, 11.74901778, 10.6910768 ,  5.68942572,
        5.70591986,  9.17595899,  4.87083348, 11.8159254 , 14.05687815,
        5.5748624 ,  8.0550286 ,  7.27340834,  5.91854397,  7.62232307,
        7.73692436,  8.1303945 ,  5.79391825,  5.47866221,  4.98461983,
       10.48721382,  4.91336994,  7.60463349,  5.41348121, 12.27835996,
       11.81423161,  5.83560009,  7.8575648 ,  9.69792513,  9.68323686,
       11.52733581, 10.91805683,  9.85384623,  8.90405823,  5.53175275,
        5.14124612,  7.5092834 , 11.018084  ,  7.88677538, 13.04051744,
       12.3489324 ,  6.31576264, 12.16901615, 11.00508681, 11.2845608 ,
        7.92607342, 11.00508681,  5.91222346,  4.89121805, 13.2430377 ,
       10.35464916, 13.72594022, 13.2430377 ,  8.04718433,  8.99276564,
       10.61417181,  7.62011734,  5.40689332,  6.46356806,  6.05113217,
       14.07623266, 11.63383221, 11.31276109,  9.15976746,  7.58378655,
        9.60080639])}
{'P': tensor([[6.3330e-03, 2.9212e-01, 1.6347e-01,  ..., 2.0238e-04, 1.3048e-05,
         6.3800e-05],
        [1.7069e-05, 6.0554e-03, 3.7025e-01,  ..., 1.5038e-05, 2.3050e-05,
         2.8088e-05],
        [1.6959e-05, 5.3376e-05, 9.2795e-03,  ..., 1.1543e-05, 4.7706e-05,
         2.9580e-05],
        ...,
        [5.9746e-02, 1.9557e-01, 1.8398e-01,  ..., 1.5188e-02, 2.1357e-05,
         1.5684e-04],
        [1.7663e-05, 1.5542e-05, 1.5030e-05,  ..., 1.8312e-05, 1.4359e-01,
         4.3059e-04],
        [1.0558e-02, 1.0344e-02, 8.7719e-03,  ..., 5.0832e-03, 7.1406e-04,
         1.0471e-01]]), 'X': tensor([[  6.5428,  -9.5728,  -2.9897],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.3424,  -7.3685,  -3.6807],
        [  6.9653,  -9.3225,  -3.4727],
        [  9.4516,  -7.2501,  -1.8868],
        [  9.9756,  -6.6330,  -3.3429],
        [ -2.0445,   7.8469,  -0.5857],
        [  2.7590,  13.6621,   9.4496],
        [ 12.2311,  -2.0445,   1.3545],
        [  5.9548,  -1.2273,  -2.6520],
        [ -4.7389,   6.9953,   3.7638],
        [  5.1519, -10.2387,   9.8191],
        [ 12.3453,  -1.5462,   7.0212],
        [  3.3744,   6.5623,  -6.3636],
        [  9.2692,   9.6370,   7.0746],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.9770, -10.6341,   8.0783],
        [  2.5829,  -4.5466,  -8.4015],
        [ -6.0844,  -8.3955,   0.7705],
        [  5.7784,  -1.6401,   9.5587],
        [  2.0729,   7.2878,   7.0880],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.3266,   0.3292,   9.3502],
        [  5.4242, -10.1261,   3.9595],
        [  5.6202,  -1.9508,   0.7471],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.1755,   2.5233,  -2.0042],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -5.9940,   6.1985,   5.9566],
        [  5.8902,  -1.3887,   4.0388],
        [  4.6231,   5.4556,   8.4973],
        [  2.8994,   6.8675,  -9.5987],
        [  4.5954,  13.0099,  -4.4439],
        [  3.3094,  13.4969,  -3.6507],
        [ -0.8455, -10.8853,  -6.7400],
        [ -0.0410,   7.8591,   9.3457],
        [  2.6451,  -4.5250,  -0.5565],
        [  0.1532,  -4.7422,  -7.0145],
        [ -9.1658,  -3.4155,  -2.8880],
        [  9.5203,   9.3483,   5.8781],
        [ -3.0008, -10.2827,  -8.9480],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.4336,  -2.2676,   5.6933],
        [  2.4639, -10.9431,  -7.3718],
        [ 10.6283,  -5.7253,   7.4878],
        [  9.5261,  -7.1675,   1.4206],
        [ 12.0845,  -2.5772,  -4.6073],
        [  3.3943,   6.5483,   1.0790],
        [  4.4039,   5.6903,   8.7366],
        [ -9.3705,   0.4208,  -3.5815],
        [ -1.8367,   7.8718,  -8.6523],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.9379,  -4.1655,  -2.8114],
        [  5.5626,  -2.0536,   8.4433],
        [  5.7802,  -1.6363,  -6.2782],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.3858,   4.4260,  -9.6769],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.5927,   0.4199,  -9.4997],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.6672,  -1.8631,   8.1020],
        [  6.3360,   0.4258,  -9.2509],
        [  3.1892, -10.8220,  -6.8409],
        [ 12.2311,  -2.0445,   1.3545],
        [  1.8447,  -4.7372,   8.1235],
        [ -9.3384,  -2.6161,  -4.7630],
        [  6.2419,  -0.2313,   4.2440],
        [ -8.5149,  -5.1905,   1.5239],
        [ 11.0438,   7.1180,  -5.6991],
        [  1.9204,   7.3518,   8.4781],
        [ -1.3178, -10.7939,   0.4236],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 12.5927,   0.4199,  -9.4997],
        [  1.0391,  -4.8141,  -0.0768],
        [  0.3231,  -4.7691,  -1.6120],
        [  4.8432,   5.1968,   1.5006],
        [  4.0078,  -3.8018,  -6.2168],
        [  6.1575,  12.2046,   3.3257],
        [  0.3231,  -4.7691,  -1.6120],
        [  0.5175,  14.0910,   4.6540],
        [  1.1994,   7.6035,   2.4295],
        [  4.1537,   5.9340,  -6.0423],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.2974,   0.8371,   2.4852],
        [  1.8766,   7.3694,   1.1887],
        [  5.9526,   3.2525,  -4.3821],
        [  6.0955,   2.8210,  -4.3321],
        [ -4.7389,   6.9953,   3.7638],
        [ -9.4524,  -1.6813,  -6.6862],
        [  6.2770,  -0.0379,   2.4195],
        [  6.3606,   0.9204,   3.2525],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -7.6734,   4.4980,  -5.4248],
        [ -7.6719,  -6.6260,   7.1468],
        [  4.3554,   5.7393,  -9.0766],
        [ -3.3812, -10.1235,   7.4384],
        [  6.2790,  -0.0259,  -5.4116],
        [ -7.8359,   4.2735,   7.2943],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.3376,  -4.6231,   9.8436],
        [  5.4426, -10.1181,  -5.1268],
        [  1.1006,   7.6318,  -2.4877],
        [  0.9478,  -4.8144,   0.6371]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1116, 0.1736,  ..., 0.1536, 0.1621, 0.3480],
        [0.1116, 0.0000, 0.0625,  ..., 0.1758, 0.1381, 0.3354],
        [0.1736, 0.0625, 0.0000,  ..., 0.2068, 0.1689, 0.3536],
        ...,
        [0.1536, 0.1758, 0.2068,  ..., 0.0000, 0.2952, 0.4921],
        [0.1621, 0.1381, 0.1689,  ..., 0.2952, 0.0000, 0.1990],
        [0.3480, 0.3354, 0.3536,  ..., 0.4921, 0.1990, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.59512194, 11.80092578, 11.89855267, 11.63724792, 11.9120321 ,
       11.97957425,  8.10886021, 13.93790235, 12.40074901,  6.07992623,
        8.44940661, 11.46176259, 12.44177641,  7.37902766, 13.37121923,
        5.13182506, 11.35345345,  5.22903117, 10.36844813,  6.00663314,
        7.57685908,  9.43112957,  6.33517349, 11.48734978,  5.94909709,
        5.19065439,  6.12886024,  6.67108662, 10.37698435, 10.52357479,
        8.62260327,  6.05165248,  7.15100303,  7.4544866 , 13.79762415,
       13.89671314, 10.91805683,  7.85919687,  5.24137909,  4.74468406,
        9.78147542, 13.3426541 , 10.71162681,  8.16714836,  5.88782913,
       11.21703654, 12.07225469, 11.92133129, 12.35625694,  7.37576154,
        7.19534146,  9.37989631,  8.08320235,  5.13125513,  9.86089579,
        5.92952538,  6.00732193,  9.26455795,  6.97106912, 13.79762415,
       12.59969936, 12.00995898,  5.96556248,  6.35029274, 11.28215429,
       12.40074901,  5.08372149,  9.69792513,  6.24614905,  9.97220799,
       13.13888841,  7.5984809 , 10.87409005, 12.11735334, 12.59969936,
        4.92497514,  4.78002665,  7.10378919,  5.52416368, 13.66991027,
        4.78002665, 14.10045797,  7.69752262,  7.24326992,  8.22728399,
        9.3349858 ,  7.60463349,  6.78325151,  6.7166297 ,  8.44940661,
        9.60080639,  6.27715105,  6.42689705,  5.8828986 , 12.25806559,
        8.89458273, 10.1371552 ,  7.20482224, 10.67322527,  6.27905292,
        8.92549752, 10.67132524,  5.1805345 , 11.4890861 ,  7.71075226,
        4.90676242])}
{'P': tensor([[4.2929e-03, 2.5346e-01, 2.3754e-01,  ..., 7.1411e-06, 2.0838e-07,
         4.0141e-06],
        [7.2496e-05, 2.1610e-02, 2.2835e-03,  ..., 1.7472e-05, 9.1398e-07,
         9.7244e-06],
        [9.7090e-05, 1.4858e-01, 2.3782e-02,  ..., 2.4487e-05, 1.0068e-06,
         1.1693e-05],
        ...,
        [1.9693e-04, 3.3118e-04, 3.8828e-04,  ..., 8.1280e-02, 5.9474e-07,
         1.5618e-05],
        [1.4324e-03, 6.9146e-04, 7.3208e-04,  ..., 6.9193e-05, 2.3234e-01,
         1.6334e-02],
        [5.6936e-02, 2.6125e-02, 2.5937e-02,  ..., 2.5866e-05, 8.9087e-07,
         7.8371e-03]]), 'X': tensor([[ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1051, 0.0884,  ..., 1.1407, 0.5370, 0.2045],
        [0.1051, 0.0000, 0.0220,  ..., 1.1479, 0.6420, 0.3092],
        [0.0884, 0.0220, 0.0000,  ..., 1.1330, 0.6237, 0.2928],
        ...,
        [1.1407, 1.1479, 1.1330,  ..., 0.0000, 1.2176, 1.1761],
        [0.5370, 0.6420, 0.6237,  ..., 1.2176, 0.0000, 0.3375],
        [0.2045, 0.3092, 0.2928,  ..., 1.1761, 0.3375, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.62158554, 11.785566  , 11.76830981, 11.85890664, 11.64746882,
       11.73567759,  9.96152788, 12.0495521 ,  8.89301516, 12.5769659 ,
        7.5984809 ,  7.49076899,  5.91222346,  8.08320235,  5.79543038,
       10.31886066, 13.77075174,  7.50500064,  6.3949043 ,  8.8737556 ,
        7.34250948,  8.36445758,  9.23071698, 11.62770332, 10.76154256,
       13.26590071,  6.46560181, 10.68031351, 11.9702762 ,  7.44805901,
        9.43737523,  7.48311002, 11.42420508,  9.96152788,  7.56617037,
       11.72749923, 11.97949273, 11.73567759, 13.31764181, 10.32715715,
       13.95361098,  9.15976746,  8.27853076,  6.77777606,  7.57685908,
        5.70405525, 13.01728313,  5.91222346,  9.95552626, 13.86480175,
        5.81481044, 11.05672136, 12.3828957 ,  9.14973319, 11.59512194,
        9.3109884 ,  8.51928531,  7.19534146,  6.33075682, 12.59969936,
        9.17905491, 14.05687815, 11.73567759, 13.89886469,  9.14973319,
       10.48315436,  8.91551231,  4.89121805,  8.40445872, 10.09979747,
       13.0493368 , 11.63502679, 11.50791859, 11.18839169,  8.95983586,
       11.42420508,  9.36378438,  5.2287058 ,  6.81084724, 13.77075174,
        6.39187285, 12.95702647,  9.50913849,  4.85995002, 11.22142195,
        5.81481044,  9.88571713, 12.10942412,  8.8580671 , 13.77222752,
        9.70108409,  6.21275016, 12.22503754, 10.52357479, 13.06498554,
       13.88683436,  6.33075682,  5.29009836, 13.62896593, 13.99048094,
       10.39942497,  6.06416957,  6.25586297, 11.57249984,  9.64157461,
       11.11038551])}
{'P': tensor([[4.2815e-03, 2.7089e-01, 2.3551e-01,  ..., 1.0956e-06, 4.7455e-07,
         8.5744e-08],
        [6.6826e-05, 9.2620e-03, 1.8763e-01,  ..., 3.2055e-06, 8.2611e-07,
         2.1230e-07],
        [9.4509e-05, 4.2424e-04, 2.1862e-02,  ..., 1.0187e-05, 1.6550e-06,
         4.4805e-07],
        ...,
        [1.1275e-06, 1.0427e-06, 9.9465e-07,  ..., 1.8379e-01, 2.9472e-05,
         7.6163e-04],
        [2.1940e-02, 1.4535e-02, 1.1025e-02,  ..., 1.4459e-06, 1.0987e-01,
         3.0113e-06],
        [5.0720e-06, 4.5516e-06, 3.1863e-06,  ..., 2.4247e-06, 8.9861e-05,
         7.4856e-02]]), 'X': tensor([[  6.8704,  -9.3809,   7.5769],
        [  7.5930,  -8.9037,   7.2323],
        [  8.2154,  -8.4262,   8.0171],
        [  8.3718,  -8.2953,   8.4666],
        [  8.6420,  -8.0580,   6.5781],
        [  9.0129,  -7.7072,   9.1400],
        [ 12.2921,  -1.7902,  -4.0870],
        [  5.6414, -10.0303,   7.8525],
        [  4.5800,   5.5034,  -3.9432],
        [ -3.2053, -10.1993,  -3.8736],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -9.1231,  -3.5739,   8.4708],
        [  1.7903,  -4.7467,   0.1168],
        [  7.4132,  11.3526,   0.3322],
        [  1.5246,  -4.7838,  -5.2095],
        [  1.6232,  -4.7717,  -2.5655],
        [ -3.4488, -10.0934,   2.5195],
        [  1.2728, -11.0372,   9.8121],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.0391,  -4.8141,  -0.0768],
        [ -1.9646,   7.8571,   3.3817],
        [  2.3373,  13.7721,  -0.2442],
        [  6.7565,  11.8237,  -4.9882],
        [  5.8325,  -1.5222,  -2.1341],
        [  9.0129,  -7.7072,   9.1400],
        [ 11.3367,  -4.4806,   0.4643],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.1960,  -3.2964,   1.6921],
        [  6.2804,   2.0019,   0.8607],
        [ -3.4986,   7.5152,   9.0712],
        [  2.3376,  -4.6231,   9.8436],
        [  4.5954,  13.0099,  -4.4439],
        [  5.3860,  -2.3418,   1.0826],
        [ 10.7566,   7.6212,  -2.3279],
        [  4.9452, -10.3187,  -7.8621],
        [  9.1409,  -7.5789,   1.5814],
        [  1.5383, -11.0274,  -4.9666],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.0965,  -5.9712,   4.3466],
        [ 11.7349,  -3.5838,   2.1644],
        [ -3.9456,  -9.8542,   0.7466],
        [ -9.3616,  -2.4749,   0.1793],
        [  9.3753,   9.5171,   3.9131],
        [  0.1497,   7.8334,  -5.4412],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -9.4671,  -0.5553,   0.1117],
        [  8.7715,  -7.9390,   5.2900],
        [  6.3303,   1.6145,  -8.4083],
        [ -6.0195,   6.1789,  -2.9496],
        [  5.1886,  -2.6250,   6.4677],
        [ -1.5408,   7.8978,   5.3470],
        [ -9.0961,   1.6529,   3.1553],
        [  2.5636, -10.9293,   9.6519],
        [  5.8318,   3.5567,  -4.7692],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -7.4602,   4.7718,  -4.6277],
        [ -2.4956, -10.4682,   3.5632],
        [  2.9410,   6.8428,   5.4458],
        [  8.4691,  10.4571,  -5.0650],
        [ -1.5408,   7.8978,   5.3470],
        [  5.5101,  -2.1433,   9.5442],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.3542,   0.6934,  -3.3840],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -3.2486,   7.5925,  -1.3719],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -1.0482, -10.8487,  -5.9066],
        [  2.0937,  13.8293,  -2.3533],
        [ -7.4733,   4.7555,   6.2490],
        [ -9.3810,  -2.3455,  -9.5194],
        [  5.9587,  12.3216,   9.1695],
        [  6.9314,  -9.3435,  -1.2729],
        [  2.4639, -10.9431,  -7.3718],
        [ -6.9838,  -7.4965,   0.9796],
        [ -1.6275, -10.7221,   3.2376],
        [ -7.4807,   4.7463,   2.0616],
        [ -0.5709,   7.9055,   7.6089],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -7.0120,   5.2825,   7.7762],
        [ -0.5385,   7.9037,   6.3830],
        [  6.2022,   2.4092,   3.0349],
        [  6.1929,   2.4501,  -6.9352],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.4296,  -1.9458,   4.0833],
        [ -9.4671,  -0.5553,   0.1117],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -5.1377,   6.7751,  -0.4438],
        [  1.1274,  14.0116,   8.1753],
        [  4.4049,   5.6892,  -4.4394],
        [  2.7590,  13.6621,   9.4496],
        [ -4.4503,   7.1378,   3.3751],
        [ 10.9875,   7.2207,  -8.8549],
        [  6.3031,   1.8477,  -7.1706],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.2076,  13.8032,  -1.5099],
        [ -8.8934,   2.2597,   3.5831],
        [ -7.6525,  -6.6534,   5.1721],
        [  6.3551,   0.7149,   1.3344],
        [  0.2571,  -4.7594,   1.5618],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.3360,   0.4258,  -9.2509],
        [  8.9284,  10.0036,  -6.9210],
        [ -5.1369,   6.7755,   8.3836],
        [  1.1274,  14.0116,   8.1753],
        [ -5.7347,  -8.6908,   9.2878],
        [ -7.3446,   4.9113,  -5.9356]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0532, 0.1060,  ..., 0.5364, 0.4874, 1.0612],
        [0.0532, 0.0000, 0.0641,  ..., 0.4892, 0.5385, 1.0617],
        [0.1060, 0.0641, 0.0000,  ..., 0.4304, 0.5918, 1.1133],
        ...,
        [0.5364, 0.4892, 0.4304,  ..., 0.0000, 1.0204, 1.3780],
        [0.4874, 0.5385, 0.5918,  ..., 1.0204, 0.0000, 1.0390],
        [1.0612, 1.0617, 1.1133,  ..., 1.3780, 1.0390, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.62770332, 11.70168755, 11.76830981, 11.785566  , 11.8159254 ,
       11.85890664, 12.4217523 , 11.50791859,  7.15990676, 10.6910768 ,
       10.32715715,  9.79814997,  5.07306274, 13.55868684,  5.0209005 ,
        5.04028538, 10.66631784, 11.11038551, 10.27572673,  4.92497514,
        8.09899874, 13.96905756, 13.61801526,  6.02789634, 11.85890664,
       12.18998195,  5.05981914,  9.76896904,  6.59176018,  8.28968067,
        5.1805345 , 13.79762415,  5.87305005, 13.18278606, 11.4424816 ,
       11.87412439, 11.1341798 , 11.72749923, 10.06023665, 12.26997165,
       10.61472636,  9.68323686, 13.35927527,  7.83486943,  9.47369628,
        9.48337063, 11.83075005,  6.53290663,  8.62631801,  5.81481044,
        8.04664886,  9.24501829, 11.22596994,  6.83083863,  9.62958973,
        8.85572292, 10.76154256,  7.44805901, 13.45642379,  8.04664886,
        5.91222346,  4.91652556,  6.39187285,  9.85330646,  8.25828409,
        9.4997987 , 10.8992524 , 13.98691426,  8.8580671 ,  9.66977802,
       13.68672855, 11.63383221, 11.21703654, 10.24557047, 10.84493732,
        8.85939626,  7.92607342,  9.94285511,  8.77913428,  7.92200342,
        6.65368551,  6.65991877, 10.61047596,  9.6282763 ,  9.48337063,
        9.60080639,  8.50278896, 14.05687815,  7.19513806, 13.93790235,
        8.41148722, 13.14777433,  6.56833864,  5.22903117, 13.97857591,
        9.17595899, 10.14044489,  6.39520397,  4.7663526 ,  5.39450305,
        6.35029274, 13.40850328,  8.50268322, 14.05687815, 10.412299  ,
        8.8353674 ])}
{'P': tensor([[3.8140e-03, 2.6994e-01, 2.4494e-01,  ..., 3.1937e-04, 3.6353e-07,
         5.8963e-06],
        [3.0342e-05, 7.1157e-03, 5.5369e-02,  ..., 2.4134e-04, 4.9720e-07,
         8.4104e-06],
        [2.6010e-05, 5.8384e-04, 6.5723e-03,  ..., 1.6847e-04, 4.4036e-07,
         6.7535e-06],
        ...,
        [5.5501e-04, 1.5955e-03, 1.4957e-03,  ..., 1.9145e-02, 1.1230e-06,
         8.4950e-05],
        [4.9598e-03, 2.6635e-03, 2.4326e-03,  ..., 1.2207e-03, 3.4500e-02,
         5.8499e-04],
        [1.1954e-03, 1.2510e-03, 1.0713e-03,  ..., 1.1239e-02, 1.9813e-04,
         9.8324e-02]]), 'X': tensor([[  6.9314,  -9.3435,  -1.2729],
        [  8.0877,  -8.5297,  -1.1050],
        [  8.2422,  -8.4041,  -1.4962],
        [  8.1217,  -8.5024,  -0.6149],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.4306,  13.9620,  -0.2075],
        [ 12.1588,   4.3364,  -8.0867],
        [  1.9505,  -4.7171,   9.9868],
        [ 11.5456,   6.0861,   8.1101],
        [ -5.3736,   6.6310,  -1.5174],
        [  6.0305,   3.0296,   0.1375],
        [  4.0587, -10.6112,   6.2273],
        [ 12.6000,   1.3839,  -3.2992],
        [  1.5662,  13.9375,   4.5142],
        [  6.3031,   1.8477,  -7.1706],
        [ -7.6650,   4.5092,   0.2520],
        [  7.4132,  11.3526,   0.3322],
        [ -4.3963,   7.1629,   4.5263],
        [ 10.6673,  -5.6649,  -4.1646],
        [  5.5294,  -2.1106,   9.0626],
        [ -2.1533, -10.5778,   1.0112],
        [  7.7794,  11.0629,   8.9069],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 11.2209,  -4.7092,   2.6117],
        [  3.9515,  -3.8425,   6.1729],
        [  5.6414, -10.0303,   7.8525],
        [ -3.3999, -10.1152,  -1.2088],
        [ -2.7572,   7.7194,   5.9580],
        [  6.9956,  11.6590,  -7.6876],
        [  5.0190,   4.9705,  -7.9752],
        [ -1.9646,   7.8571,   3.3817],
        [ -6.3489,   5.9117,   6.5161],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4159,  -2.0739,   9.6841],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.1420,   6.9331,  -2.4779],
        [  9.6614,   9.1784,  -1.2087],
        [ -5.9225,  -8.5355,   2.4778],
        [  1.1994,   7.6035,   2.4295],
        [  7.7794,  11.0629,   8.9069],
        [ -1.5261,   7.8988,   9.3768],
        [ -8.6801,   2.7809,  -0.8443],
        [  0.3715,  -4.7756,   0.2312],
        [ 10.0567,   8.6697,   4.6329],
        [  6.3550,   0.7129,   0.4917],
        [  1.6422,   7.4586,  -6.8056],
        [  6.3266,   0.3292,   9.3502],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.1999,  -3.6543,  -4.1998],
        [ -1.0407,   7.9166,  -6.6021],
        [  9.0051,   9.9234,   5.4411],
        [  3.9770, -10.6341,   8.0783],
        [  5.2939,  -2.4784,  -4.5840],
        [  2.6613,   7.0021,   4.5783],
        [  7.8425,  -8.7202,   6.4530],
        [  9.0652,   9.8597,  -3.5949],
        [ 11.2517,  -4.6495,   2.9800],
        [ -2.5161,   7.7699,  -9.3531],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -8.7544,   2.6096,   3.3146],
        [  2.3928,   7.1404,  -1.7035],
        [  2.6321,  -4.5296,  -2.5159],
        [  4.8653,   5.1694,  -4.0823],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.5789,   0.1722,   1.6470],
        [ 11.2939,   6.6325,  -5.8311],
        [ -2.2188,   7.8218,   8.6064],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.3301,   0.3638,  -7.5685],
        [  2.5813,  -4.5471,   5.3597],
        [ -1.5451,   7.8975,   7.6104],
        [  5.8450,  -9.9356,   3.0767],
        [  9.3560,   9.5391,   9.4943],
        [  3.1892, -10.8220,  -6.8409],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.2022,   2.4092,   3.0349],
        [  2.5469,  13.7192,  -9.6935],
        [  2.5469,  13.7192,  -9.6935],
        [  2.7558,  -4.4844,   0.0636],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.8602,   6.8905,   5.8779],
        [  5.3118,   4.5452,   4.2114],
        [ -6.3489,   5.9117,   6.5161],
        [  5.9587,  12.3216,   9.1695],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.4157,   0.0821,  -5.0025],
        [ 10.7566,   7.6212,  -2.3279],
        [  4.4290,  -3.4596,   1.4181],
        [  0.9074,  -4.8139,   8.1674],
        [ -6.0844,  -8.3955,   0.7705],
        [  9.1820,   9.7334,  -2.7932],
        [  6.3497,   1.3680,  -1.9076],
        [  4.7482, -10.3907,   5.9149],
        [  9.8082,  -6.8396,   4.9615],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.5083,  -4.1177,   0.2389],
        [ -2.0445,   7.8469,  -0.5857],
        [  6.1730,  -0.5375,  -8.8861],
        [ -2.0445,   7.8469,  -0.5857],
        [  7.5784,  -8.9141,   5.8410],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.1206,  -4.7363,   8.7103]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.5762e-02, 9.1263e-02,  ..., 6.4818e-01, 5.7320e-01,
         6.7771e-01],
        [7.5762e-02, 0.0000e+00, 3.5148e-02,  ..., 6.4828e-01, 6.4882e-01,
         6.4868e-01],
        [9.1263e-02, 3.5148e-02, 0.0000e+00,  ..., 6.8279e-01, 6.6055e-01,
         6.7804e-01],
        ...,
        [6.4818e-01, 6.4828e-01, 6.8279e-01,  ..., 0.0000e+00, 8.6151e-01,
         2.5750e-01],
        [5.7320e-01, 6.4882e-01, 6.6055e-01,  ..., 8.6151e-01, 3.0339e-04,
         1.0372e+00],
        [6.7771e-01, 6.4868e-01, 6.7804e-01,  ..., 2.5750e-01, 1.0372e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.63383221, 11.75438006, 11.77124373, 11.75807252, 11.9182956 ,
       11.9120321 ,  9.55593374, 14.03505624, 12.9089559 ,  5.10446653,
       13.05148826,  8.53501777,  6.74872981, 11.36089355, 12.67576792,
       14.02526505,  6.56833864,  8.89301516, 13.55868684,  8.40445872,
       12.07818871,  5.91854397, 10.794755  , 13.52430176, 10.00410625,
       12.16901615,  5.51177501, 11.50791859, 10.67132524,  8.19703512,
       13.59673406,  7.06372172,  8.09899874,  8.67502348,  6.00663314,
        9.64157461, 11.27160035, 13.12299145, 13.32613445, 10.38897563,
        7.69752262, 13.52430176,  8.04483495,  9.11473391,  4.79003361,
       13.2778378 ,  6.3949043 ,  7.63726134,  6.33517349,  5.5513072 ,
        5.5671389 ,  7.98469166, 13.40024224, 11.35345345,  5.84533345,
        7.49076899, 11.72802491, 13.39372426, 12.17451685,  8.16714836,
       10.06437598,  9.13507321,  7.53063885,  5.23878371,  7.09887557,
       10.91805683,  6.38968357, 12.58006038, 13.09737053,  8.1303945 ,
       13.01187484,  6.34059021,  5.2287058 ,  8.04718433, 11.52733581,
       13.3614598 , 11.28215429, 12.91505111,  6.65368551, 13.95361098,
       13.95361098,  5.26342203, 10.08001894,  7.46053281,  6.99097013,
        8.67502348, 13.68672855,  8.29022197,  9.41605684, 13.18278606,
        5.62005629,  4.89869205, 10.36844813, 13.38091121,  6.49538841,
       11.42420508, 11.95743558,  5.94541803,  7.02326515, 12.22276278,
        8.10886021,  6.19633139,  8.10886021, 11.70016217, 10.33712258,
        4.73784222])}
{'P': tensor([[2.9376e-03, 2.1573e-01, 1.9398e-01,  ..., 1.1085e-07, 4.6391e-04,
         5.8180e-04],
        [2.6262e-05, 3.5249e-03, 1.4139e-01,  ..., 1.2544e-07, 5.9952e-04,
         7.9629e-04],
        [5.6141e-05, 1.4589e-04, 1.2760e-02,  ..., 4.6221e-07, 2.7413e-03,
         2.6703e-03],
        ...,
        [1.7687e-04, 1.3485e-04, 1.1185e-04,  ..., 8.1582e-02, 8.8207e-05,
         4.8805e-05],
        [9.9611e-05, 1.1230e-04, 1.3559e-04,  ..., 1.2964e-04, 1.0776e-01,
         3.9869e-03],
        [2.2704e-04, 2.7129e-04, 3.1993e-04,  ..., 5.2151e-05, 1.3101e-02,
         1.3502e-01]]), 'X': tensor([[ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.0080e-02, 9.3721e-02,  ..., 6.9877e-01, 4.1159e-01,
         4.6124e-01],
        [4.0080e-02, 0.0000e+00, 7.0441e-02,  ..., 7.3881e-01, 3.9152e-01,
         4.2135e-01],
        [9.3721e-02, 7.0441e-02, 0.0000e+00,  ..., 7.6650e-01, 3.2134e-01,
         4.0591e-01],
        ...,
        [6.9877e-01, 7.3881e-01, 7.6650e-01,  ..., 0.0000e+00, 9.7183e-01,
         1.1597e+00],
        [4.1159e-01, 3.9152e-01, 3.2134e-01,  ..., 9.7183e-01, 0.0000e+00,
         4.5629e-01],
        [4.6124e-01, 4.2135e-01, 4.0591e-01,  ..., 1.1597e+00, 4.5629e-01,
         1.5170e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.63502679, 11.69264243, 11.75807252, 11.73397309, 11.83570132,
       11.83509372,  9.75401449,  9.95552626, 12.5085995 ,  7.01475376,
        9.79039371, 14.03505624, 11.63502679, 13.32613445, 13.39372426,
        5.0209005 , 13.36800073,  6.67218687, 13.77075174, 11.64746882,
        9.50654967,  5.92952538,  6.12886024,  7.42512621,  9.69629   ,
       12.22276278,  9.62238828,  8.50268322,  5.01114719, 13.70893492,
        4.7663526 , 13.45642379,  5.13125513, 10.66631784, 13.18278606,
       10.27716434,  5.43943791,  8.1137452 , 10.53368644, 12.53345445,
       12.62576026, 11.63765273, 12.4217523 ,  5.05774295, 11.85890664,
        9.15976746,  4.91652556,  6.31576264,  9.14973319,  7.01475376,
       10.05558893, 10.412299  ,  7.67939546,  7.83534987,  8.8353674 ,
       13.27717982, 10.60226614, 12.32545637,  4.7663526 ,  6.02588517,
       11.04629471,  8.51928531, 13.01728313, 10.86693708, 10.06286074,
       11.75618513, 10.65929968,  5.56809335,  4.85603886,  6.97106912,
       11.07637531, 11.75438006,  8.24808832,  4.85603886,  5.2287058 ,
       13.04051744,  8.98765644,  6.25586297,  5.8828986 ,  8.16584537,
        9.47369628,  8.29022197, 13.68063098,  7.54127425,  8.36995953,
        8.08320235,  5.14607014,  6.33517349,  5.91677188, 12.53285633,
       12.67576792, 10.46739073, 11.65923396,  6.76056625,  9.68324553,
        7.14837514, 11.73567759, 12.50996828, 10.1371552 , 13.01728313,
        5.36061974, 10.84437705, 12.62576026,  9.4997987 ,  6.39187285,
        6.19707944])}
{'P': tensor([[6.2392e-03, 3.5618e-01, 2.2421e-01,  ..., 1.0145e-05, 1.1364e-03,
         2.9589e-07],
        [1.8953e-05, 5.5217e-03, 3.3761e-01,  ..., 1.4910e-05, 2.4732e-03,
         2.0277e-07],
        [1.3743e-05, 3.8054e-05, 6.6158e-03,  ..., 2.0841e-05, 4.3065e-03,
         2.0602e-07],
        ...,
        [7.7430e-06, 8.1735e-06, 7.9116e-06,  ..., 1.2633e-01, 5.4394e-05,
         2.7165e-04],
        [2.9192e-06, 4.7333e-06, 6.5279e-06,  ..., 6.8707e-03, 6.0696e-02,
         2.1901e-06],
        [7.2797e-04, 3.5879e-04, 2.1597e-04,  ..., 1.1742e-05, 7.9550e-06,
         4.6853e-02]]), 'X': tensor([[  6.9653,  -9.3225,  -3.4727],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.3424,  -7.3685,  -3.6807],
        [  9.9756,  -6.6330,  -3.3429],
        [  9.4516,  -7.2501,  -1.8868],
        [ 10.6673,  -5.6649,  -4.1646],
        [  6.2036,   2.4030,  -1.2500],
        [  6.3550,   0.7129,   0.4917],
        [ -9.3076,   0.7852,  -6.7010],
        [  9.4554,   9.4245,   9.8800],
        [ -9.1459,   1.4780,  -4.2403],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -7.6734,   4.4980,  -5.4248],
        [  3.1480,  -4.3165,  -7.8002],
        [  7.0642,  11.6104,  -9.8316],
        [  3.6249, -10.7255,   5.6605],
        [  6.0310,  12.2795,  -6.4528],
        [  2.5974,   7.0363,  -7.7810],
        [ -4.2040,   7.2486,   2.9875],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.1532,  -4.7422,  -7.0145],
        [ -3.5173, -10.0623,   4.8280],
        [ -8.0087,   4.0173,   9.1091],
        [ -9.4541,  -0.3327,   8.8246],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.4009,  -4.1874,  -5.9751],
        [  6.2247,  -0.3150,  -9.3657],
        [ 12.0845,  -2.5772,  -4.6073],
        [  6.2883,   0.0325,   5.2692],
        [  8.1043,  -8.5164,  -7.9290],
        [ 12.5789,   0.1722,   1.6470],
        [  0.6933,   7.7336,   2.1179],
        [  0.5250, -11.0306,   5.5358],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.7845,   3.6652,   0.9175],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.6220,   7.8917,   0.0315],
        [  9.5261,  -7.1675,   1.4206],
        [ -2.6735, -10.4062,  -1.5911],
        [  0.3724,   7.7974,  -6.8722],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.7685,  -4.8098,  -2.9310],
        [  7.1821,  -9.1845,   9.8548],
        [  0.9809,   7.6642,  -1.4217],
        [  3.0876,   6.7527,  -5.3540],
        [ -3.3999, -10.1152,  -1.2088],
        [ -5.1398,  -9.1367,   6.2074],
        [  1.5246,  -4.7838,  -5.2095],
        [ -1.5451,   7.8975,   7.6104],
        [  5.2911,  -2.4824,  -3.6877],
        [ -7.3446,   4.9113,  -5.9356],
        [ -6.4844,  -8.0230,   8.5209],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.1204,  -5.9308,   4.0761],
        [ -7.2986,  -7.1225,  -3.0814],
        [  9.5203,   9.3483,   5.8781],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.8745,   2.3097,   3.5600],
        [  2.5829,  -4.5466,  -8.4015],
        [  1.1644, -11.0394,   7.8511],
        [ -2.2508, -10.5479,  -2.7235],
        [  4.0421,  -3.7765,  -8.3631],
        [  5.4429,  -2.2529,   9.7306],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.1477,   7.2550,   8.9688],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.4714,  -1.3301,  -3.1998],
        [ 12.5717,   1.9266,   7.4903],
        [  4.4039,   5.6903,   8.7366],
        [  4.9957,  -2.8705,   6.5978],
        [ -7.4733,   4.7555,   6.2490],
        [  0.1497,   7.8334,  -5.4412],
        [  0.1041, -11.0045,   2.4311],
        [ -0.5739,   7.9056,   0.6382],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.3606,   0.9204,   3.2525],
        [ -9.1795,  -3.3623,   8.5259],
        [  5.2939,  -2.4784,  -4.5840],
        [ 10.6283,  -5.7253,   7.4878],
        [  2.4932,   7.0903,  -5.4214],
        [  5.9587,  12.3216,   9.1695],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.3118,   4.5452,   4.2114],
        [ 12.2589,  -1.9312,   9.8657],
        [ -7.4602,   4.7718,  -4.6277],
        [ -8.6426,  -4.9136,   9.4941],
        [  6.1973,  -0.4372,   6.5720],
        [  2.3376,  -4.6231,   9.8436],
        [ -6.4464,   5.8272,   7.5038],
        [  5.9193,  -1.3176,   5.1739],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.1667,  -2.6543,   1.2879],
        [  1.9505,  -4.7171,   9.9868],
        [ -8.3539,  -5.5111,   0.3373],
        [ -7.4807,   4.7463,   2.0616],
        [ -4.3963,   7.1629,   4.5263],
        [  5.5424,  -2.0885,   7.2487],
        [ -2.0840,   7.8415,  -7.8657],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -6.8618,  -7.6322,   3.7027],
        [  2.8994,   6.8675,  -9.5987],
        [  2.5375,   7.0676,  -4.3490],
        [  2.8602,   6.8905,   5.8779],
        [ 12.4837,   2.7510,   0.4449],
        [ -8.5149,  -5.1905,   1.5239]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0761, 0.1341,  ..., 0.7550, 0.5630, 0.7622],
        [0.0761, 0.0000, 0.0625,  ..., 0.7458, 0.5061, 0.8290],
        [0.1341, 0.0625, 0.0000,  ..., 0.7744, 0.4922, 0.8914],
        ...,
        [0.7550, 0.7458, 0.7744,  ..., 0.0000, 0.4966, 0.8698],
        [0.5630, 0.5061, 0.4922,  ..., 0.4966, 0.0000, 1.0984],
        [0.7622, 0.8290, 0.8914,  ..., 0.8698, 1.0984, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.63724792, 11.80092578, 11.89855267, 11.97957425, 11.9120321 ,
       12.07818871,  6.652742  ,  6.3949043 ,  9.34061938, 13.35013374,
        9.26455795, 12.25806559,  8.89458273,  5.34250225, 13.59056613,
       11.32149259, 13.68063098,  7.50035355,  8.37954853, 10.78533838,
        4.74468406, 10.65929968,  8.95983586,  9.45995955,  4.76969364,
        5.39450305,  6.23262289, 12.35625694,  6.28835101, 11.75618513,
       12.58006038,  7.76457095, 11.04313641, 11.59512194,  6.84796202,
        6.61211224,  8.05668609, 11.92133129, 10.74409834,  7.80625981,
        5.27178363,  4.87083348, 11.65923396,  7.72668467,  7.42512621,
       10.67132524, 10.48315436,  5.0209005 ,  8.04718433,  5.84450127,
        8.8353674 , 10.31584102, 10.86693708, 10.05558893, 10.1979669 ,
       13.3426541 ,  5.19065439,  9.17016857,  5.22903117, 11.10066126,
       10.78533838,  5.53175275,  5.89074104,  4.76969364,  7.56617037,
       10.78533838,  9.5642993 , 12.71843462,  7.19534146,  5.76165934,
        8.8580671 ,  7.83486943, 11.0050379 ,  7.92644746,  5.90035324,
        6.42689705,  9.77588801,  5.84533345, 12.07225469,  7.5158489 ,
       13.68672855,  9.37989631,  6.99097013, 12.41012046,  8.85572292,
        9.94173708,  6.21275016,  5.1805345 ,  8.68979052,  6.06416957,
       13.00413829,  5.80860165,  5.10446653, 10.00794468,  8.85939626,
        8.40445872,  5.92281142,  8.1137452 ,  9.96152788,  9.99145096,
       10.26327895,  7.4544866 ,  7.5092834 ,  7.46053281, 12.78327037,
        9.97220799])}
{'P': tensor([[3.7634e-03, 2.4098e-01, 2.1341e-01,  ..., 2.9814e-04, 6.0441e-04,
         2.3848e-07],
        [6.9287e-05, 1.2553e-02, 2.9932e-02,  ..., 6.4725e-04, 1.5248e-03,
         5.1572e-07],
        [3.8404e-05, 1.5415e-03, 7.4557e-03,  ..., 4.4723e-04, 1.1205e-03,
         3.3981e-07],
        ...,
        [1.1200e-03, 1.4905e-03, 1.8935e-03,  ..., 5.6494e-02, 2.9579e-01,
         3.7495e-05],
        [7.9672e-04, 1.1117e-03, 1.3313e-03,  ..., 5.9580e-03, 1.1239e-01,
         4.1201e-05],
        [2.2749e-03, 1.5982e-03, 2.1783e-03,  ..., 4.4841e-03, 2.6807e-03,
         2.1272e-01]]), 'X': tensor([[  6.9693,  -9.3200,  -8.3822],
        [  7.9403,  -8.6455,  -8.7433],
        [  8.1043,  -8.5164,  -7.9290],
        [  8.0381,  -8.5690,  -8.9523],
        [  8.6271,  -8.0715,  -7.4813],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.3804,  -2.3504,   7.2744],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.2644,  -4.6434,   9.7508],
        [  6.0140,  -1.0661,  -2.5040],
        [ -2.6069, -10.4298,  -6.1995],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.3767, -11.0341,   9.9415],
        [ -5.0550,   6.8231,  -9.2502],
        [ -7.0825,   5.2073,  -5.0497],
        [  2.3889,  -4.6082,  -8.7291],
        [  0.1460,   7.8340,  -9.4614],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.9695,  -2.9407,  -6.3498],
        [  0.3493,  -4.7727,  -6.5682],
        [ -0.5385,   7.9037,   6.3830],
        [  4.7813,  -3.1139,   8.2691],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.0305,   3.0296,   0.1375],
        [  2.7295,  13.6703,  -8.9328],
        [  6.8407,  11.7666,   3.0045],
        [  5.9912,   3.1453,  -3.9989],
        [  2.0937,  13.8293,  -2.3533],
        [ -1.6906,   7.8860,   7.7142],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.8363,   2.4086,  -5.6296],
        [ 10.6283,  -5.7253,   7.4878],
        [  4.2339,  -3.6268,   6.3125],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.5813,  -4.5471,   5.3597],
        [ -8.4824,  -5.2575,  -0.0795],
        [ 10.0400,   8.6922,  -6.2847],
        [ -7.9298,   4.1367,  -3.9773],
        [ -3.6381,   7.4683,   7.3544],
        [  5.6260,  -1.9401,   3.0259],
        [ 12.5650,   2.0132,   4.6272],
        [  7.5930,  -8.9037,   7.2323],
        [  2.0729,   7.2878,   7.0880],
        [ -9.4524,  -1.6813,  -6.6862],
        [  6.3211,   1.7017,  -0.2164],
        [  2.8474,   6.8979,   6.6457],
        [  6.1984,  -0.4327,   1.5438],
        [  4.9152,  12.8653,  -1.4459],
        [  6.3119,   1.7797,   9.9186],
        [  2.3373,  13.7721,  -0.2442],
        [  2.3367,   7.1676,  -9.5263],
        [  3.9913,  -3.8139,  -6.8311],
        [ -8.0766,  -6.0046,  -3.1873],
        [ 11.4495,   6.3026,   6.8257],
        [  4.9152,  12.8653,  -1.4459],
        [ -7.7239,   4.4298,   3.1216],
        [  5.3860,  -2.3418,   1.0826],
        [  0.4996,   7.7738,  -4.4711],
        [  6.3549,   0.7096,   6.9198],
        [  6.1575,  12.2046,   3.3257],
        [  1.6422,   7.4586,  -6.8056],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.4697,  -4.5836,   1.4593],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.3200,   7.1756,   1.1938],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -4.1296,   7.2803,  -2.7194],
        [  5.1787,  -2.6383,  -4.2533],
        [ -8.0608,   3.9362,  -5.9567],
        [ -4.7334,   6.9982,  -0.4392],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -0.8885,   7.9160,   0.0571],
        [  3.5939,   6.4029,  -1.8357],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.3118,   4.5452,   4.2114],
        [  4.9337,  12.8566,  -3.6582],
        [ 11.4743,   6.2478,  -2.1100],
        [  5.5626,  -2.0536,   8.4433],
        [  5.3602,  -2.3808,   7.2098],
        [ 11.7779,   5.5127,  -2.0699],
        [  9.4516,  -7.2501,  -1.8868],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.1620,  -7.5573,  -9.9882],
        [ -8.0608,   3.9362,  -5.9567],
        [ -6.1914,  -8.2997,   8.5380],
        [ 12.4690,  -0.8528,  -9.3374],
        [  3.9515,  -3.8425,   6.1729],
        [  1.4150,  13.9647,  -4.0915],
        [ 12.3025,  -1.7440,  -6.6632],
        [  9.3753,   9.5171,   3.9131],
        [ 10.1798,   8.5002,  -5.8893],
        [ -3.1204, -10.2345,   9.2288],
        [  8.3791,  10.5409,   5.5590],
        [  6.3303,   1.6145,  -8.4083],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.2883,   0.0325,   5.2692],
        [  4.3554,   5.7393,  -9.0766],
        [ -5.9940,   6.1985,   5.9566],
        [  5.0506,   4.9277,  -9.5094],
        [  1.7000,  -4.7609,   2.7935],
        [ 12.2435,   4.0076,   8.1257],
        [  4.3459,  -3.5327,   0.3667],
        [  2.0449,  13.8402,  -7.1642],
        [  4.1297,  -3.7098,  -3.9838],
        [  5.1787,  -2.6383,  -4.2533],
        [ -3.5173, -10.0623,   4.8280]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0871, 0.0537,  ..., 0.3610, 0.3641, 1.1096],
        [0.0871, 0.0000, 0.0390,  ..., 0.3193, 0.3165, 1.1589],
        [0.0537, 0.0390, 0.0000,  ..., 0.3206, 0.3210, 1.1254],
        ...,
        [0.3610, 0.3193, 0.3206,  ..., 0.0000, 0.0272, 1.0270],
        [0.3641, 0.3165, 0.3210,  ..., 0.0272, 0.0000, 1.0539],
        [1.1096, 1.1589, 1.1254,  ..., 1.0270, 1.0539, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.63765273, 11.73848065, 11.75618513, 11.74901778, 11.81423161,
       11.86239961,  5.87132718,  8.53501777,  5.16611782,  6.10774544,
       10.7506423 , 10.04916678, 11.11969815,  8.49161154,  8.7907923 ,
        5.19065439,  7.83534987, 12.19805394, 12.32545637,  4.78544875,
        7.92200342,  5.70591986,  9.79039371,  6.74872981, 13.94008948,
       13.6105595 ,  6.76661595, 13.98691426,  8.06517143, 14.07623266,
        9.15866607, 12.07225469,  5.5748624 , 10.00410625,  5.2287058 ,
        9.97964013, 13.27994304,  8.94395628,  8.30729352,  5.95111804,
       12.72524615, 11.70168755,  7.57685908,  9.60080639,  6.54615887,
        7.46248947,  6.2134961 , 13.77222752,  6.55800267, 13.96905756,
        7.53884104,  5.5205189 , 10.06407839, 13.06957425, 13.77222752,
        8.90405823,  5.87305005,  7.78980299,  6.39438044, 13.66991027,
        7.63726134, 12.4217523 ,  5.20659892,  5.27178363,  7.54127425,
       10.33712258,  8.36995953,  5.81199422,  8.97052115,  8.4486771 ,
        9.50654967,  7.96575782,  7.34250948,  5.41348121,  6.99097013,
       13.77075174, 13.06498554,  5.92952538,  5.86519525, 13.00413829,
       11.9120321 , 12.49291768, 11.87665501,  8.97052115, 10.35464916,
       12.4980845 ,  5.51177501, 14.03618649, 12.42554808, 13.35927527,
       13.26209545, 10.69963954, 13.46553623,  6.53290663,  8.22728399,
        6.28835101,  7.20482224,  8.62260327,  7.05627461,  5.05534609,
       12.88270018,  5.60064319, 13.99048094,  5.5513072 ,  5.81199422,
       10.65929968])}
{'P': tensor([[4.7205e-03, 4.1592e-01, 1.9821e-01,  ..., 3.9501e-05, 2.2438e-03,
         1.6393e-07],
        [4.8162e-05, 9.6440e-03, 6.3183e-01,  ..., 1.2301e-04, 4.0882e-03,
         2.9918e-07],
        [6.3252e-05, 9.1795e-05, 2.6611e-02,  ..., 7.3411e-04, 6.5892e-03,
         7.0241e-07],
        ...,
        [6.9694e-07, 7.6537e-07, 9.5399e-07,  ..., 6.7661e-02, 1.6093e-06,
         1.2402e-05],
        [1.9242e-05, 3.3382e-05, 1.0416e-04,  ..., 1.8388e-03, 1.6140e-02,
         5.5779e-07],
        [4.6782e-04, 3.2649e-04, 1.7198e-04,  ..., 2.3070e-06, 1.2142e-04,
         9.2427e-02]]), 'X': tensor([[ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0548, 0.1115,  ..., 0.4615, 0.3162, 0.8759],
        [0.0548, 0.0000, 0.0567,  ..., 0.4077, 0.3048, 0.9275],
        [0.1115, 0.0567, 0.0000,  ..., 0.3525, 0.3032, 0.9812],
        ...,
        [0.4615, 0.4077, 0.3525,  ..., 0.0000, 0.4215, 1.2931],
        [0.3162, 0.3048, 0.3032,  ..., 0.4215, 0.0000, 0.9127],
        [0.8759, 0.9275, 0.9812,  ..., 1.2931, 0.9127, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.64436684, 11.72749923, 11.87665501, 11.74901778, 11.73848065,
       11.86239961,  6.32572323, 11.83570132, 12.85032836,  8.92549752,
        5.89074104, 13.1399181 , 13.32557133,  9.9353921 ,  5.27178363,
        8.0550286 , 10.63038565,  8.51088913, 12.53285633,  6.25586297,
        7.8788965 , 12.18045978, 12.84056604,  6.89245352, 13.92390689,
        9.77588801,  7.5403822 , 11.10066126,  6.46560181, 12.52875793,
        6.32572323,  9.97822473,  5.07306274, 10.01804223, 12.49291768,
        9.36378438,  4.96231532,  7.37576154, 13.6660671 , 13.69162795,
        6.83515201,  8.85939626, 12.35625694, 10.60226614,  9.18690018,
        8.28968067,  9.37989631, 11.75618513, 11.37939695,  4.83371341,
        7.60463349,  6.27715105, 12.4980845 , 12.27835996, 10.35485331,
       11.22596994,  8.89458273, 10.8992524 , 11.51557166, 14.08549802,
        9.64157461,  9.48969978, 10.42667093,  6.24614905, 10.7506423 ,
        6.00663314,  6.50409372,  4.75303127,  7.28793663,  5.5205189 ,
        7.20826219,  8.67227386, 12.07225469, 13.40024224, 11.48734978,
        9.58072286,  8.7273569 ,  5.23878371, 12.4217523 , 14.10796466,
        7.50564223, 13.35832725, 11.97957425,  9.20340031,  6.58184474,
        9.58347456,  7.44805901, 13.97857591, 13.54774355,  9.11473391,
        6.28961988, 13.96905756, 12.35421461,  8.62162372, 12.05300943,
        6.76056625,  7.15990676,  9.13507321,  6.83515201,  6.74872981,
        9.09963894,  6.42439344,  6.58184474, 13.35832725, 12.10942412,
        9.79039371])}
{'P': tensor([[3.6577e-03, 2.6812e-01, 2.1420e-01,  ..., 2.1622e-05, 5.1118e-05,
         1.8514e-07],
        [6.1722e-05, 1.4953e-02, 9.3421e-02,  ..., 7.6266e-05, 1.6785e-04,
         7.3284e-07],
        [6.4546e-05, 1.6228e-03, 1.5357e-02,  ..., 6.0914e-05, 1.3204e-04,
         7.5150e-07],
        ...,
        [3.6812e-06, 6.4514e-06, 6.0104e-06,  ..., 1.7375e-02, 1.1086e-03,
         1.4002e-07],
        [4.5467e-06, 8.4883e-06, 7.9948e-06,  ..., 2.4594e-02, 8.9502e-03,
         1.0636e-07],
        [2.7337e-04, 1.6615e-04, 1.5502e-04,  ..., 4.4899e-06, 8.8424e-06,
         1.1715e-01]]), 'X': tensor([[  7.0664,  -9.2590,   8.0879],
        [  8.2154,  -8.4262,   8.0171],
        [  8.3718,  -8.2953,   8.4666],
        [  9.0129,  -7.7072,   9.1400],
        [  8.6420,  -8.0580,   6.5781],
        [ 10.6283,  -5.7253,   7.4878],
        [  9.5018,  -7.1945,  -1.1892],
        [  4.1537,   5.9340,  -6.0423],
        [  0.3354,  -4.7708,  -4.9125],
        [  1.9505,  -4.7171,   9.9868],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -0.7599, -10.8995,  -6.0366],
        [  6.2242,  -0.3171,   7.4978],
        [ -9.3444,   0.5821,   6.9101],
        [  3.7298,  13.3537,  -9.4653],
        [  9.2979,   9.6048,   7.3138],
        [  1.1644, -11.0394,   7.8511],
        [ 11.6544,   5.8273,  -3.1375],
        [  7.9403,  -8.6455,  -8.7433],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.4887,  -5.2448,  -0.6585],
        [  7.9403,  -8.6455,  -8.7433],
        [  2.9260,  -4.4162,   8.3808],
        [ -5.1377,   6.7751,  -0.4438],
        [ -8.7735,   2.5640,  -6.3313],
        [  2.2076,  13.8032,  -1.5099],
        [  6.3122,  -9.6994,  -6.3282],
        [  5.9193,  -1.3176,   5.1739],
        [ -1.9487,   7.8590,  -3.3330],
        [  1.7301,   7.4262,   5.0522],
        [  4.6231,   5.4556,   8.4973],
        [  5.1787,  -2.6383,  -4.2533],
        [ -3.4488, -10.0934,   2.5195],
        [  3.2809,  13.5061,  -4.0199],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 12.5770,   1.8512,  -0.6578],
        [  6.3535,   0.6792,  -7.5101],
        [  0.1047, -11.0046,   0.5188],
        [ -8.1905,   3.7253,   4.1175],
        [ 12.6000,   1.3839,  -3.2992],
        [ 11.4743,   6.2478,  -2.1100],
        [  1.4306,  13.9620,  -0.2075],
        [ 11.9695,  -2.9407,  -6.3498],
        [  8.3718,  -8.2953,   8.4666],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.1532,  -4.7422,  -7.0145],
        [  1.3705,  13.9723,   7.9266],
        [  6.3535,   0.6792,  -7.5101],
        [  4.6357,   5.4415,   7.9067],
        [  6.1973,  -0.4372,   6.5720],
        [  0.8700,  -4.8132,  -7.8042],
        [  4.2096,   5.8815,   1.9230],
        [  0.3493,  -4.7727,  -6.5682],
        [  2.5128, -10.9365,   9.9645],
        [  5.8902,  -1.3887,   4.0388],
        [ -9.3616,  -2.4750,   3.7765],
        [ -2.7572,   7.7194,   5.9580],
        [  7.0358,  -9.2784,  -9.8479],
        [  1.7815,  -4.7481,   4.4289],
        [  5.2608,  12.6977,   4.9166],
        [  6.2630,   2.1057,  -5.2136],
        [  3.3744,   6.5623,  -6.3636],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.0588,  -6.0341,  -2.3006],
        [  7.5930,  -8.9037,   7.2323],
        [ -4.7389,   6.9953,   3.7638],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.6154,  -4.0640,  -7.9698],
        [ -5.8389,  -8.6056,  -3.5098],
        [  2.3490,  -4.6199,  -4.2489],
        [ 11.2209,  -4.7092,   2.6117],
        [  0.9392,  -4.8143,  -3.4649],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -5.6423,   6.4537,  -1.2091],
        [  7.3592,  11.3937,   6.2407],
        [  2.9470,  13.6085,   2.9809],
        [  3.3167,   6.6022,   6.3554],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.6613,   7.0021,   4.5783],
        [  0.1206,  -4.7363,   8.7103],
        [ -4.4503,   7.1378,   3.3751],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.0608,   3.9362,  -5.9567],
        [  9.2811,  -7.4336,   5.1771],
        [ 12.4691,   2.8525,  -5.5110],
        [ -7.4887,   4.7364,   8.9074],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.6681,  -4.8556,   9.2518],
        [ -9.4721,  -1.3097,  -7.4377],
        [  0.9445,   7.6736,   0.5095],
        [ 11.0438,   7.1180,  -5.6991],
        [  4.5340,  -3.3629,   8.9652],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.1905,   3.7253,   4.1175],
        [ -8.0087,   4.0173,   9.1091],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.2974,   0.8371,   2.4852],
        [  5.1787,  -2.6383,  -4.2533],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -9.4117,   0.1160,   8.2574]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.9225e-02, 9.7226e-02,  ..., 1.2297e+00, 1.1258e+00,
         4.9149e-01],
        [7.9225e-02, 0.0000e+00, 2.2026e-02,  ..., 1.2115e+00, 1.1085e+00,
         5.6953e-01],
        [9.7226e-02, 2.2025e-02, 0.0000e+00,  ..., 1.2221e+00, 1.1194e+00,
         5.8865e-01],
        ...,
        [1.2297e+00, 1.2115e+00, 1.2221e+00,  ..., 0.0000e+00, 1.0425e-01,
         1.3584e+00],
        [1.1258e+00, 1.1085e+00, 1.1194e+00,  ..., 1.0425e-01, 0.0000e+00,
         1.2587e+00],
        [4.9149e-01, 5.6953e-01, 5.8865e-01,  ..., 1.3584e+00, 1.2587e+00,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.64746882, 11.76830981, 11.785566  , 11.85890664, 11.8159254 ,
       12.07225469, 11.9182956 ,  7.24326992,  4.78258092,  5.10446653,
        6.01356004,  9.66977802, 10.92597196,  6.2322769 ,  9.36256096,
       13.86480175, 13.36800073, 11.10066126, 13.03002338, 11.73848065,
       10.24557047,  9.97822473, 11.73848065,  5.29755487,  8.50278896,
        9.14044984, 13.97857591, 11.57249984,  6.06416957,  8.09704052,
        7.62509582,  7.15100303,  5.81199422, 10.66631784, 13.89886469,
       12.18045978, 12.7125079 ,  6.38968357, 11.00508681,  8.99791316,
       12.67576792, 13.06498554, 14.03505624, 12.32545637, 11.785566  ,
        8.62329466,  4.85995002,  4.74468406, 14.0393949 ,  6.38968357,
        7.14837514,  6.21275016,  4.89121805,  7.23276582,  4.78544875,
       11.22142195,  6.05165248,  9.68324553,  8.19703512, 11.64436684,
        5.0713364 , 13.744387  ,  6.60752373,  7.37902766, 12.1494023 ,
        9.15976746, 10.0674887 , 11.70168755,  8.44940661, 10.8207297 ,
        5.43943791, 10.39942497,  5.18277858, 12.16901615,  4.90505566,
       10.30972193,  8.5724042 , 13.56367857, 13.92390689,  7.38844703,
        5.8828986 ,  7.49076899,  4.73784222,  8.41148722, 10.63038565,
        9.47369628,  8.97052115, 11.89105259, 12.79126785,  8.86082124,
       13.26590071,  9.9353921 ,  9.562174  ,  7.73151309, 13.13888841,
        5.64500512,  5.24137909,  8.99791316,  8.95983586, 13.6660671 ,
        9.3349858 ,  5.81199422, 10.38897563, 12.11735334, 12.05395056,
        9.41244865])}
{'P': tensor([[6.0254e-03, 5.1075e-01, 2.3370e-01,  ..., 3.6934e-07, 2.2695e-05,
         7.5637e-08],
        [5.8070e-05, 1.0597e-02, 5.5408e-01,  ..., 5.4377e-07, 3.6003e-05,
         1.2624e-07],
        [4.6022e-05, 7.2717e-05, 1.4127e-02,  ..., 5.7482e-07, 4.2649e-05,
         1.6837e-07],
        ...,
        [7.2996e-04, 5.4099e-04, 3.4741e-04,  ..., 1.9135e-02, 8.3291e-05,
         3.8855e-05],
        [6.5575e-06, 9.6751e-06, 2.0422e-05,  ..., 3.5766e-07, 1.6795e-02,
         6.0557e-07],
        [5.9869e-05, 4.8287e-05, 3.6764e-05,  ..., 2.4262e-02, 1.4093e-04,
         8.5380e-02]]), 'X': tensor([[  7.1821,  -9.1845,   9.8548],
        [  7.9141,  -8.6656,   9.6948],
        [  9.0129,  -7.7072,   9.1400],
        [  8.3718,  -8.2953,   8.4666],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.6283,  -5.7253,   7.4878],
        [  4.4049,   5.6892,  -4.4394],
        [  2.0243,   7.3086,   7.5432],
        [ -0.3432,   7.8900,   7.1805],
        [ -7.3333,   4.9246,  -7.9212],
        [  4.0591,   6.0202,  -1.6339],
        [  2.3367,   7.1676,  -9.5263],
        [  4.7482, -10.3907,   5.9149],
        [  1.3670,  -4.7990,   3.2603],
        [ -9.3705,   0.4208,  -3.5815],
        [ 11.0864,   7.0387,  -0.1238],
        [ 11.2767,  -4.6006,   0.7868],
        [  0.2732,  -4.7619,  -1.1577],
        [ -6.8466,  -7.6488,   1.9648],
        [  6.3305,   1.6119,   1.1538],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.8154,   2.4611,   2.5198],
        [  6.1989,  -9.7591,   0.8289],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.2365,  -0.2578,  -6.0212],
        [ -3.2736, -10.1703,   3.8712],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -8.6426,  -4.9136,   9.4941],
        [  1.9204,   7.3518,   8.4781],
        [ -9.0574,   1.7801,   8.8986],
        [  5.6095,  -1.9702,  -7.8233],
        [  3.4920,  -4.1366,  -8.3955],
        [  6.3246,   1.6696,  -0.1544],
        [ 12.5123,  -0.5339,  -1.0728],
        [  8.4691,  10.4571,  -5.0650],
        [  0.2571,  -4.7594,   1.5618],
        [  9.0004,  -7.7195,   4.6945],
        [ -7.6650,   4.5092,   0.2520],
        [  6.3550,   0.7129,   0.4917],
        [ -9.3444,   0.5821,   6.9101],
        [ 11.4703,  -4.2006,   7.4086],
        [  1.7228,  -4.7575,   8.3396],
        [  0.7958,  -4.8109,  -8.2248],
        [ -8.9942,  -3.9990,   6.0448],
        [  2.2373,  -4.6506,  -1.7334],
        [  3.2366,  -4.2732,   5.1532],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3119,   1.7797,   9.9186],
        [ 11.5083,  -4.1177,   0.2389],
        [  2.8994,   6.8675,  -9.5987],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.9843,  -9.8679,  -6.1449],
        [  3.6469,   6.3624,  -1.8414],
        [  0.8572,  14.0501,  -8.3364],
        [  3.9770, -10.6341,   8.0783],
        [  6.2036,   2.4030,  -1.2500],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.6757, -11.0361,  -1.3047],
        [  3.2159, -10.8166,  -1.0406],
        [ 12.4691,   2.8525,  -5.5110],
        [ -0.8455, -10.8853,  -6.7400],
        [  0.6363, -11.0348,   6.1414],
        [  5.1861,   4.7361,  -3.7312],
        [ 12.6035,   0.7494,   7.1536],
        [  9.6587,   9.1817,  -0.8212],
        [  4.6357,   5.4415,   7.9067],
        [ -4.4520,   7.1370,  -8.7867],
        [  2.5128, -10.9365,   9.9645],
        [  9.1820,   9.7334,  -2.7932],
        [  1.1274,  14.0116,   8.1753],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.8323,  -1.5228,   5.1132],
        [  6.3211,   1.7017,  -0.2164],
        [ -4.7389,   6.9953,   3.7638],
        [ -6.9838,  -7.4965,   0.9796],
        [  4.0362,  13.2397,  -7.0030],
        [  5.0506,   4.9277,  -9.5094],
        [  2.0729,   7.2878,   7.0880],
        [  6.1752,   2.5242,   3.5874],
        [  6.0345,   3.0175,  -1.4424],
        [  4.9957,  -2.8705,   6.5978],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 11.9695,  -2.9407,  -6.3498],
        [  3.7298,  13.3537,  -9.4653],
        [  2.9260,  -4.4162,   8.3808],
        [  2.1625,  -4.6697,   7.0658],
        [  6.2576,   2.1359,  -3.3398],
        [ -1.5451,   7.8975,   7.6104],
        [ 12.5717,   1.9266,   7.4903],
        [  6.1755,   2.5233,  -2.0042],
        [ -4.9217,   6.8979,  -1.7667],
        [ -7.4808,   4.7462,   2.0266],
        [  6.1755,   2.5233,  -2.0042],
        [ -8.5900,   2.9767,   7.4811],
        [  3.4920,  -4.1366,  -8.3955],
        [  5.0190,   4.9705,  -7.9752],
        [ 11.4495,   6.3026,   6.8257],
        [  0.9809,   7.6642,  -1.4217],
        [  2.1120,  -4.6818,   8.6144],
        [ -5.6792,  -8.7352,   4.0318],
        [ 12.5919,   0.4035,  -5.8460],
        [ 11.2517,  -4.6495,   2.9800],
        [ -6.8618,  -7.6322,   3.7027],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.3107,  -2.7697,  -3.2919]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0532, 0.1005,  ..., 0.7418, 1.1706, 1.1786],
        [0.0532, 0.0000, 0.0473,  ..., 0.7861, 1.1614, 1.2081],
        [0.1005, 0.0473, 0.0000,  ..., 0.8259, 1.1540, 1.2348],
        ...,
        [0.7418, 0.7861, 0.8259,  ..., 0.0000, 1.1110, 0.5720],
        [1.1706, 1.1614, 1.1540,  ..., 1.1110, 0.0000, 0.8791],
        [1.1786, 1.2081, 1.2348,  ..., 0.5720, 0.8791, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.65923396, 11.73567759, 11.85890664, 11.785566  , 12.13568598,
       12.07225469,  7.19513806,  7.58378655,  7.89745107,  8.83341608,
        7.26074925,  7.53884104, 11.42420508,  4.98989906,  9.37989631,
       13.13205357, 12.17901033,  4.76969364, 10.26545792,  6.53250937,
        9.38232121,  9.15253344, 11.56146873,  9.75200848,  6.24187044,
       10.68416401, 10.61047596,  9.94173708,  7.5984809 ,  9.23071698,
        5.94541803,  5.41348121,  6.54127431, 12.52372629, 13.45642379,
        4.7663526 , 11.85743425,  8.89301516,  6.3949043 ,  9.36256096,
       12.21532282,  5.05981914,  4.8763135 ,  9.84314712,  5.16077764,
        5.36061974,  6.16512856,  6.55800267, 12.22276278,  7.4544866 ,
       12.32545637, 11.54071859,  7.33350642, 14.07623266, 11.35345345,
        6.652742  ,  4.91038884, 11.05672136, 11.2845608 , 12.79126785,
       10.91805683, 11.05317597,  7.02326515, 12.62576026, 13.32645552,
        7.14837514,  8.41171369, 11.22142195, 13.38091121, 14.05687815,
        9.7712321 ,  6.02779201,  6.54615887,  8.44940661, 10.24557047,
       13.84126197,  7.05627461,  7.57685908,  6.67121748,  6.74686071,
        5.76165934,  9.71392179,  9.88310614, 12.32545637, 13.86480175,
        5.29755487,  5.14607014,  6.61211224,  8.04718433, 12.71843462,
        6.67108662,  8.47371868,  8.859416  ,  6.67108662,  9.09119351,
        5.41348121,  7.06372172, 13.06957425,  7.72668467,  5.13616224,
       10.41909211, 12.59840193, 12.17451685, 10.26327895, 11.97238624,
        9.71392179])}
{'P': tensor([[8.5122e-03, 2.4479e-01, 1.4977e-01,  ..., 5.1002e-03, 2.8033e-04,
         9.0382e-04],
        [1.3074e-05, 8.0413e-03, 4.3679e-01,  ..., 2.6853e-03, 1.3880e-03,
         1.8950e-03],
        [1.2875e-05, 8.2332e-05, 1.1367e-02,  ..., 3.0154e-03, 2.7961e-03,
         3.3919e-03],
        ...,
        [6.1318e-06, 6.2559e-05, 1.5288e-04,  ..., 7.0924e-03, 4.4013e-04,
         1.4892e-03],
        [2.7072e-07, 5.2047e-07, 6.4226e-07,  ..., 6.5787e-07, 1.5346e-02,
         1.3134e-04],
        [5.9187e-05, 1.1656e-04, 1.4841e-04,  ..., 1.9706e-04, 7.3095e-03,
         1.0882e-01]]), 'X': tensor([[ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1614, 0.2048,  ..., 0.5445, 0.5884, 0.3611],
        [0.1614, 0.0000, 0.0489,  ..., 0.5122, 0.4325, 0.2625],
        [0.2048, 0.0489, 0.0000,  ..., 0.4863, 0.4022, 0.2235],
        ...,
        [0.5445, 0.5122, 0.4863,  ..., 0.0000, 0.7503, 0.2860],
        [0.5884, 0.4325, 0.4022,  ..., 0.7503, 0.0000, 0.4774],
        [0.3611, 0.2625, 0.2235,  ..., 0.2860, 0.4774, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.66453059, 11.97238624, 12.05395056, 11.86239961, 11.81423161,
       12.05300943, 13.36092624, 13.41713661, 13.3607379 , 10.30972193,
       10.33712258,  6.35029274,  8.7273569 ,  8.67502348,  9.50913849,
       13.91198191,  8.98765644,  9.97964013,  6.18668463,  7.09887557,
        5.9510489 , 11.63502679, 14.02526505,  9.94285511, 12.49317563,
        5.2287058 ,  5.29009836, 13.26209545, 12.30695098,  5.43943791,
        7.50500064,  7.8788965 , 13.09426142,  7.48311002, 12.22276278,
        9.62238828,  9.09661212, 12.53345445,  6.18668463, 13.61600536,
        6.02588517,  4.92497514, 13.97857591, 10.39942497, 12.12448912,
        6.69389431,  6.38105015,  6.49538841, 11.63765273,  4.74468406,
        9.75075273,  6.60752373,  6.33517349,  9.85384623,  5.07306274,
        9.60080639,  8.92549752,  7.13501264, 12.12448912,  8.98765644,
       10.38897563,  8.34949383,  8.04664886,  6.67121748,  8.31625171,
       12.35625694,  7.10378919, 13.48809595, 10.26327895, 11.92133129,
        9.94285511, 13.6660671 ,  9.09661212, 13.89264822,  9.9353921 ,
       13.88683436,  5.46244988,  4.89121805,  5.79543038, 12.3489324 ,
       10.33958639,  4.78258092, 10.8992524 ,  5.05444288, 11.62158554,
        7.52117132, 12.33167382, 14.03618649,  5.13125513,  8.72896381,
        5.68942572,  6.02779201,  6.24614905,  4.91296205, 10.08001894,
       12.4217523 ,  5.70405525, 12.52875793,  8.62162372,  4.91652556,
       10.35485331,  9.78285424, 12.22276278, 12.17901033, 13.14777433,
        6.61211224])}
{'P': tensor([[3.8240e-03, 3.6389e-01, 2.8965e-01,  ..., 6.4101e-06, 5.3035e-03,
         6.2805e-08],
        [5.8588e-05, 1.1175e-02, 8.1579e-04,  ..., 1.5867e-05, 2.0012e-02,
         1.8747e-07],
        [6.6966e-05, 1.0572e-01, 9.4692e-03,  ..., 1.4739e-05, 1.8029e-02,
         1.6508e-07],
        ...,
        [7.3580e-04, 7.8612e-04, 8.4840e-04,  ..., 1.3160e-01, 3.2075e-03,
         1.7398e-04],
        [1.5527e-05, 2.1577e-05, 2.0343e-05,  ..., 3.5891e-05, 4.6696e-02,
         3.1320e-06],
        [1.7839e-05, 1.4488e-05, 1.5747e-05,  ..., 1.8493e-03, 6.2787e-06,
         1.0988e-01]]), 'X': tensor([[ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0608, 0.0424,  ..., 0.7462, 0.2925, 0.9992],
        [0.0608, 0.0000, 0.0198,  ..., 0.7348, 0.2426, 1.0318],
        [0.0424, 0.0198, 0.0000,  ..., 0.7319, 0.2541, 1.0164],
        ...,
        [0.7462, 0.7348, 0.7319,  ..., 0.0000, 0.5805, 0.6954],
        [0.2925, 0.2426, 0.2541,  ..., 0.5805, 0.0000, 1.0395],
        [0.9992, 1.0318, 1.0164,  ..., 0.6954, 1.0395, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.67875855, 11.74901778, 11.73848065, 11.87665501, 11.86239961,
       12.05300943, 13.62896593, 11.22596994,  5.18047638, 13.89886469,
       13.0743399 , 12.07818871,  5.27178363, 13.32645552,  7.28793663,
        6.65991877, 10.33712258, 13.89974693,  7.63726134, 12.10942412,
       11.42420508, 12.05300943, 11.69264243,  9.94173708, 13.25336001,
        8.09704052,  8.99276564,  6.27905292, 12.1494023 ,  4.74770899,
        7.44805901,  9.0368458 ,  4.89121805,  5.0713364 , 11.05317597,
        8.81930525,  9.18690018,  5.14371005, 11.97949273,  8.38252937,
        5.54776934, 10.9036226 , 14.05687815,  9.7751189 , 10.33712258,
        6.61211224, 13.35013374,  9.97964013,  8.1303945 ,  7.60463349,
        5.36061974,  8.27853076, 13.40024224,  6.77777606, 10.41909211,
        8.77913428,  6.7166297 , 12.32408516,  6.00663314, 10.82106505,
        5.16205087, 11.32149259, 14.0393949 ,  5.0209005 , 11.67875855,
       12.889695  , 12.49291768, 11.62158554,  8.62329466,  7.24326992,
       13.54774355, 13.09426142,  6.35029274,  5.14371005,  7.38844703,
        7.15990676,  6.54693987, 10.54658597,  5.19577146, 10.42667093,
       10.1371552 , 12.35625694, 12.16901615,  5.79391825, 13.39372426,
        9.58072286,  5.0713364 , 10.04916678, 13.6660671 ,  8.31625171,
        8.06643487, 10.27716434, 10.68416401,  5.91408504, 10.63038565,
       13.96746816, 10.1979669 ,  5.48276038, 13.32557133,  7.69752262,
        7.37576154, 10.91601655, 10.38897563,  5.05534609, 12.35625694,
        9.17595899])}
{'P': tensor([[3.0112e-03, 2.8410e-01, 2.0103e-01,  ..., 3.3846e-07, 3.6327e-04,
         5.5745e-04],
        [2.9907e-05, 4.3020e-03, 3.5927e-01,  ..., 5.2707e-07, 6.0546e-04,
         7.1918e-04],
        [5.8104e-05, 6.9950e-05, 1.6077e-02,  ..., 2.3461e-06, 2.6774e-03,
         2.1753e-03],
        ...,
        [1.2367e-05, 1.1480e-05, 9.2025e-06,  ..., 1.1928e-01, 6.2762e-06,
         3.4974e-05],
        [4.4824e-06, 5.4226e-06, 8.6685e-06,  ..., 5.6611e-05, 3.4136e-02,
         3.4171e-05],
        [5.9797e-04, 6.9372e-04, 1.0532e-03,  ..., 6.0216e-05, 3.5727e-04,
         7.4132e-02]]), 'X': tensor([[ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0333, 0.0953,  ..., 0.5810, 0.8581, 0.3830],
        [0.0333, 0.0000, 0.0640,  ..., 0.5798, 0.8331, 0.3641],
        [0.0953, 0.0640, 0.0000,  ..., 0.6089, 0.8100, 0.3123],
        ...,
        [0.5810, 0.5798, 0.6089,  ..., 0.0000, 0.6143, 0.9076],
        [0.8581, 0.8331, 0.8100,  ..., 0.6143, 0.0000, 0.9640],
        [0.3830, 0.3641, 0.3123,  ..., 0.9076, 0.9640, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.69264243, 11.73397309, 11.83570132, 11.83509372, 11.75807252,
       11.9182956 ,  9.54154096, 13.58101794,  8.47371868,  8.95133961,
       12.87752052, 10.35464916,  7.14837514, 10.88618114,  9.15253344,
        5.39098669,  5.79543038, 10.06407839, 13.03002338,  4.89869205,
        9.60080639, 13.12299145,  7.14837514, 10.86693708,  6.18668463,
        8.36445758,  6.01356004,  9.63742669,  9.14044984,  9.75075273,
       13.88683436,  6.54693987,  7.83486943, 11.00812996, 13.37121923,
       11.97238624,  8.29022197, 13.70893492,  8.5724042 ,  8.1303945 ,
        5.94293185, 11.9182956 , 11.018084  ,  5.63824971,  8.74470218,
       11.86239961,  6.27905292,  5.38342077,  6.65991877, 12.49317563,
        6.67108662,  6.38968357,  8.04718433,  5.5366698 ,  7.98469166,
        8.55982105, 10.61472636,  6.55800267, 10.67322527,  6.35029274,
        6.05165248,  8.01009083,  7.05627461,  4.89869205,  7.06372172,
        5.94293185,  8.38252937,  8.50268322,  6.01356004,  5.88782913,
        4.91652556,  5.01114719,  5.19065439, 10.4977132 , 13.04051744,
       11.85890664,  7.17789666,  8.92098752, 10.51437947,  6.67218687,
        6.33517349,  8.30125201,  6.28961988,  4.74468406,  8.04483495,
        8.95133961,  6.39187285,  7.72291946,  8.05242591,  8.25828409,
       13.25017211,  5.60064319,  7.83486943, 11.48434767,  4.87083348,
        5.0713364 , 12.88270018, 13.54774355, 13.56367857,  5.05774295,
        7.54127425, 12.1494023 ,  7.37576154,  8.53014915, 12.5085995 ,
        5.84450127])}
{'P': tensor([[3.4505e-03, 1.7837e-01, 1.4413e-01,  ..., 3.1748e-06, 4.8726e-03,
         2.4689e-07],
        [1.2168e-05, 2.4923e-03, 2.4888e-01,  ..., 3.5682e-06, 1.7495e-03,
         2.2142e-07],
        [1.0049e-05, 1.6501e-05, 3.0366e-03,  ..., 5.0267e-06, 1.0015e-03,
         2.9710e-07],
        ...,
        [9.7689e-06, 1.3096e-05, 1.4630e-05,  ..., 2.9779e-01, 3.5936e-05,
         2.0475e-03],
        [1.4012e-04, 7.2294e-04, 1.8880e-03,  ..., 1.0851e-04, 1.4903e-02,
         2.0419e-06],
        [1.4850e-05, 1.1011e-05, 9.4945e-06,  ..., 1.9427e-05, 1.1840e-05,
         6.7684e-02]]), 'X': tensor([[ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0736, 0.1131,  ..., 1.0286, 0.4598, 0.6751],
        [0.0736, 0.0000, 0.0406,  ..., 0.9905, 0.4199, 0.6900],
        [0.1131, 0.0406, 0.0000,  ..., 0.9824, 0.4133, 0.7120],
        ...,
        [1.0286, 0.9905, 0.9824,  ..., 0.0000, 0.5707, 0.6899],
        [0.4598, 0.4199, 0.4133,  ..., 0.5707, 0.0000, 0.4814],
        [0.6751, 0.6900, 0.7120,  ..., 0.6899, 0.4814, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.70016217, 11.83075005, 11.89105259, 11.8159254 , 11.9702762 ,
       11.95743558, 13.99048094,  6.01356004,  7.46248947,  9.69792513,
       10.06023665, 13.89264822, 13.36800073,  5.92952538, 10.26327895,
        7.42512621, 13.5247215 ,  6.33075682, 10.06023665,  6.49538841,
        5.77219077, 10.74655346, 12.62576026,  6.1463896 ,  8.62162372,
        9.15976746, 13.12299145, 11.00694688, 11.72749923,  9.41970031,
        6.81961996, 11.018084  ,  8.55982105,  5.79562691, 10.51437947,
        5.38342077,  9.5642993 ,  9.87720102,  5.39450305, 13.79676654,
        6.89245352,  8.92395381,  9.09661212,  8.83725081, 10.9036226 ,
        6.21275016, 13.60275211, 12.53345445,  8.51928531, 11.83570132,
        9.9353921 , 10.31613269, 13.1399181 , 12.3489324 , 11.00812996,
        5.89074104,  7.92644746,  9.91668182, 12.24467041,  7.56617037,
       11.70168755,  8.68832398, 13.00413829,  5.83560009,  5.54776934,
        7.33350642,  6.27905292, 12.49291768, 12.05300943, 10.51437947,
        8.51928531, 11.52733581,  4.75303127,  8.91551231,  9.86089579,
        6.69389431,  7.34250948,  8.24808832, 13.32557133,  5.64464807,
        7.03733775, 11.57249984,  7.46248947,  5.10446653, 12.17451685,
       10.8992524 , 13.2778378 ,  8.62260327,  6.1463896 , 13.66991027,
       11.95743558,  8.09842704,  7.62509582,  7.72415048, 12.17451685,
       13.98691426,  8.55982105, 11.9702762 ,  5.1224079 , 12.1460931 ,
       13.99877413, 10.90638066,  7.06169391,  6.97106912, 11.92133129,
        8.83725081])}
{'P': tensor([[7.3280e-03, 3.2648e-01, 1.5269e-01,  ..., 6.7971e-07, 5.5602e-06,
         1.0772e-05],
        [5.8995e-05, 9.2678e-03, 2.8214e-01,  ..., 1.2297e-06, 7.4424e-06,
         2.4624e-05],
        [8.7835e-05, 1.6589e-04, 7.4320e-02,  ..., 1.4162e-05, 2.5358e-05,
         2.3732e-04],
        ...,
        [6.6531e-06, 6.2339e-06, 3.7450e-06,  ..., 1.5160e-01, 2.3771e-04,
         9.2347e-06],
        [8.6457e-05, 1.3160e-04, 1.2156e-04,  ..., 4.3812e-04, 1.3744e-01,
         7.1791e-03],
        [6.5465e-08, 1.0036e-07, 1.8161e-07,  ..., 1.8652e-04, 6.4406e-06,
         1.2007e-02]]), 'X': tensor([[  7.5930,  -8.9037,   7.2323],
        [  8.6420,  -8.0580,   6.5781],
        [ 10.6283,  -5.7253,   7.4878],
        [  8.2154,  -8.4262,   8.0171],
        [  9.9058,  -6.7203,   5.6928],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.2419,  -0.2313,   4.2440],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.7162,   5.6728,  -8.9372],
        [  9.6816,   9.1536,  -5.8543],
        [  3.2366,  -4.2732,   5.1532],
        [  5.4167, -10.1293,   6.2156],
        [  1.8447,  -4.7372,   8.1235],
        [  4.2041,  -3.6509,  -4.9488],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.6202,  -1.9508,   0.7471],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.4739,  -1.2497,   0.5483],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.1379,  -4.6757,   6.4981],
        [ 11.5568,   6.0602,   9.3244],
        [ -3.5173, -10.0623,   4.8280],
        [ 10.9931,   7.2107,   7.7933],
        [  7.9141,  -8.6656,   9.6948],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.1749,  -3.3803,   6.4629],
        [  6.3236,   0.3011,  -4.6668],
        [  3.2692,  13.5098,  -4.9032],
        [ 11.3367,  -4.4806,   0.4643],
        [  3.2159, -10.8166,  -1.0406],
        [ -7.5720,   4.6310,   2.1717],
        [ -0.6323,   7.9085,  -5.9544],
        [ -0.1626, -10.9796,   6.4629],
        [  6.1989,  -9.7591,   0.8289],
        [ -9.4157,   0.0821,  -5.0025],
        [  8.8474,  10.0868,  -8.8644],
        [  1.2273,  -4.8081,   9.8129],
        [ -8.6603,   2.8250,   5.9181],
        [  3.9913,  -3.8139,  -6.8311],
        [ -8.0611,  -6.0302,   4.6962],
        [  2.6614,  -4.5192,  -9.3641],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -1.8823, -10.6558,  -3.8952],
        [  8.8474,  10.0868,  -8.8644],
        [ -0.5739,   7.9056,   0.6382],
        [  7.8425,  -8.7202,   6.4530],
        [ -8.1666,   3.7653,   3.6075],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.1892, -10.8220,  -6.8409],
        [ -4.7389,   6.9953,   3.7638],
        [  3.9591, -10.6391,  -6.2557],
        [ -1.5261,   7.8988,   9.3768],
        [ -1.2454,   7.9127,  -0.5715],
        [  8.0877,  -8.5297,  -1.1050],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.9193,  -1.3176,   5.1739],
        [ -9.1907,  -3.3180,  -5.5943],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -1.7290,   7.8825,  -3.3831],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.3200,   0.2690,   1.5900],
        [  9.0004,  -7.7195,   4.6945],
        [  0.9809,  -4.8145,   6.8706],
        [ 11.0891,  -4.9558,  -2.3178],
        [  4.0457,  -3.7737,  -0.4704],
        [  6.3246,   1.6696,  -0.1544],
        [ -1.5451,   7.8975,   7.6104],
        [  6.2555,   2.1474,   4.1685],
        [ -9.1623,  -3.4288,  -4.9961],
        [  1.9503,  -4.7172,  -6.2313],
        [ -4.6393,   7.0461,   1.1624],
        [ -8.6050,   2.9451,  -0.7012],
        [  2.4697,  -4.5836,   1.4593],
        [  3.2366,  -4.2732,   5.1532],
        [ -8.4824,  -5.2575,  -0.0795],
        [  7.0642,  11.6104,  -9.8316],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.8889,  -1.3916,   8.8993],
        [  1.8766,   7.3694,   1.1887],
        [ 11.9122,   5.1397,  -1.5498],
        [  1.0000,   7.6591,  -4.5011],
        [  2.3367,   7.1676,  -9.5263],
        [ -3.7964,  -9.9294,   7.7147],
        [ -7.4887,   4.7364,   8.9074],
        [  6.3589,   0.8239,  -8.4230],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.3535,  -5.5118,   9.8399],
        [ -2.3317,   7.8034,  -3.6274],
        [  6.2975,   0.0947,   5.1791],
        [  0.6952,  -4.8060,   8.8976],
        [  5.6523,   3.9443,  -2.9449],
        [  0.6363, -11.0348,   6.1414],
        [  5.4426, -10.1181,  -5.1268],
        [ -8.0330,  -6.0765,   6.7945],
        [  1.8320,   7.3871,   5.1452],
        [  2.7420,  -4.4896,  -3.7139],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.4356,  -4.5942,  -5.0808],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -3.0004,   7.6607,  -7.2640],
        [  4.6357,   5.4415,   7.9067],
        [ -3.9700,   7.3453,  -4.1486],
        [  3.7992,  -3.9474,  -8.9019],
        [ 11.7162,   5.6728,  -8.9372]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0655, 0.1850,  ..., 0.9026, 1.0765, 1.3672],
        [0.0655, 0.0000, 0.1317,  ..., 0.8941, 1.0431, 1.3249],
        [0.1850, 0.1317, 0.0000,  ..., 0.9800, 1.0823, 1.3403],
        ...,
        [0.9026, 0.8941, 0.9800,  ..., 0.0000, 0.4240, 0.7640],
        [1.0765, 1.0431, 1.0823,  ..., 0.4240, 0.0000, 0.3481],
        [1.3672, 1.3249, 1.3403,  ..., 0.7640, 0.3481, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.70168755, 11.8159254 , 12.07225469, 11.76830981, 11.9702762 ,
       12.10222204,  6.24614905,  8.41148722, 13.01728313, 13.32374056,
        5.36061974, 11.48663767,  5.08372149,  5.56809335, 12.59840193,
        5.94909709,  5.43544626,  9.55593374,  5.95297459,  5.14124612,
       13.0493368 , 10.65929968, 13.14690137, 11.73567759, 12.22276278,
        9.77777346,  6.33075682, 13.89974693, 12.18998195, 11.2845608 ,
        8.87587457,  7.9337647 , 10.98076873, 11.56146873,  9.41605684,
       13.41713661,  4.96231532,  9.10945678,  5.5205189 , 10.06703358,
        5.24461068,  9.86089579, 10.8207297 , 13.41713661,  7.92644746,
       11.72802491,  8.99276564, 11.77124373, 11.28215429,  8.44940661,
       11.35182589,  8.04483495,  8.01009083, 11.75438006, 13.27994304,
        9.11473391,  6.06416957,  9.7712321 , 12.53285633,  9.94285511,
        8.06990483, 11.75618513,  6.32572323, 11.85743425,  4.91336994,
       12.1460931 ,  5.53255882,  6.54127431,  8.04718433,  6.61386436,
        9.78286733,  5.10444097,  8.43626114,  9.09502385,  5.20659892,
        5.36061974,  9.97964013, 13.59056613, 12.25806559,  6.05113217,
        7.60463349, 12.97370371,  7.72415048,  7.53884104, 10.63038565,
        8.86082124,  6.41202986,  7.8575648 , 10.00802627,  8.14433542,
        6.29821673,  4.85603886,  6.89245352, 11.05317597, 11.4890861 ,
       10.07238463,  7.61088125,  5.26067019,  8.53501777,  5.19986797,
       10.33712258,  8.22728399,  7.14837514,  8.34949383,  5.47866221,
       13.01728313])}
{'P': tensor([[9.7098e-03, 6.3614e-01, 9.5012e-02,  ..., 4.8453e-07, 1.0769e-05,
         2.9261e-06],
        [1.1379e-04, 3.2988e-02, 2.1582e-01,  ..., 1.6735e-06, 2.4730e-05,
         1.3323e-05],
        [1.6414e-05, 7.2353e-05, 7.3132e-03,  ..., 4.3039e-07, 6.9602e-06,
         1.0362e-05],
        ...,
        [9.5039e-05, 5.3292e-05, 3.0159e-05,  ..., 7.7577e-02, 2.7365e-04,
         5.7643e-06],
        [4.4024e-04, 4.5590e-04, 1.2258e-03,  ..., 4.0633e-05, 1.0902e-01,
         1.6124e-03],
        [8.4464e-07, 8.6652e-07, 1.7382e-06,  ..., 7.1590e-05, 1.3234e-04,
         5.2016e-02]]), 'X': tensor([[  7.8375,  -8.7240,  -9.8913],
        [  9.1620,  -7.5573,  -9.9882],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 10.9717,  -5.1646,  -6.9039],
        [  9.0424,  -7.6779,  -7.5286],
        [  6.3211,   1.7017,  -0.2164],
        [  9.2811,  -7.4336,   5.1771],
        [ 12.5267,  -0.4125,  -8.7331],
        [  6.2555,   2.1474,   4.1685],
        [  8.8091,  -7.9037,   0.3631],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.7349,  -3.5838,   2.1644],
        [  1.7659,   7.4127,   4.7871],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.3470,  -7.0615,   5.0066],
        [  3.1271, -10.8343,   2.6442],
        [  3.0876,   6.7527,  -5.3540],
        [  6.0345,   3.0175,  -1.4424],
        [ -9.3616,  -2.4749,   0.1793],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 11.5966,   5.9666,   6.0083],
        [  7.5784,  -8.9141,   5.8410],
        [ 10.7566,   7.6212,  -2.3279],
        [ -3.4488, -10.0934,   2.5195],
        [  3.4604,  -4.1545,   1.8091],
        [  3.6469,   6.3624,  -1.8414],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.5859,  -5.0393,   0.5476],
        [  2.4639, -10.9431,  -7.3718],
        [  0.6933,   7.7336,   2.1179],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 11.4235,   6.3594,   5.1663],
        [  1.7815,  -4.7481,   4.4289],
        [  3.6817,   6.3354,   6.1233],
        [ 11.5198,  -4.0923,   5.5088],
        [  6.0140,  -1.0661,  -2.5040],
        [ -7.2515,   5.0195,   4.6434],
        [ -7.0120,   5.2825,   7.7762],
        [  5.1886,  -2.6250,   6.4677],
        [ -2.6069, -10.4298,  -6.1995],
        [ 11.5935,   5.9740,  -4.3952],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.1929,   2.4501,  -6.9352],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.0073,   3.0986,   6.7937],
        [  2.3200,   7.1756,   1.1938],
        [ -2.2188,   7.8218,   8.6064],
        [  3.1271, -10.8343,   2.6442],
        [ 11.9936,  -2.8677,   6.0614],
        [  4.4915,   5.5990,   7.0405],
        [  0.1041, -11.0045,   2.4311],
        [  5.9843,  -9.8679,  -6.1449],
        [ 12.6000,   1.3839,  -3.2992],
        [  2.3589,  13.7668,  -3.6501],
        [  3.2366,  -4.2732,   5.1532],
        [ 12.2218,   4.0953,  -8.2072],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.1379,  -4.6757,   6.4981],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -7.2986,  -7.1225,  -3.0814],
        [  7.3592,  11.3937,   6.2407],
        [  4.1637,  13.1898,  -1.9189],
        [  2.5636, -10.9293,   9.6519],
        [  3.3839,  -4.1966,   8.3574],
        [  4.4633,   5.6287,  -0.7754],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.1120,  -4.6818,   8.6144],
        [  7.5064,  -8.9651,  -0.0994],
        [ -1.7009,   7.8851,  -8.2437],
        [  1.7659,   7.4127,   4.7871],
        [  3.5965,  -4.0754,  -3.5035],
        [ -1.2454,   7.9127,  -0.5715],
        [  7.8982,  -8.6778,   0.0239],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -3.5173, -10.0623,   4.8280],
        [  4.4049,   5.6892,  -4.4394],
        [ 11.5966,   5.9666,   6.0083],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.1813,  -3.3550,   1.8827],
        [ -5.5525,   6.5146,  -4.4984],
        [  0.1382, -11.0073,   1.0026],
        [ -2.5056,   7.7720,   5.5878],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -1.6906,   7.8860,   7.7142],
        [  0.1931,  -4.7491,   7.4385],
        [ -8.8400,   2.3991,   1.3457],
        [  6.3119,   1.7797,   9.9186],
        [  6.0955,   2.8210,  -4.3321],
        [ 12.5996,   1.3971,  -1.0694],
        [  4.5800,   5.5034,  -3.9432],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.0995,  -6.4739,   2.0974],
        [  5.4732,  -2.2041,  -4.3830],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -8.8934,   2.2597,   3.5831],
        [ -7.4887,   4.7364,   8.9074],
        [  3.9022,  13.2906,   5.5788],
        [ 10.9394,   7.3069,  -8.9365],
        [ -2.8162,   7.7059,  -9.6885],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.6523,   3.9443,  -2.9449],
        [ -9.3087,   0.7792,  -7.5587],
        [  1.7000,  -4.7609,   2.7935],
        [  2.8602,   6.8905,   5.8779]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0567, 0.0462,  ..., 0.8026, 0.7489, 1.0145],
        [0.0567, 0.0000, 0.0281,  ..., 0.8590, 0.7490, 1.0174],
        [0.0462, 0.0281, 0.0000,  ..., 0.8404, 0.7258, 0.9935],
        ...,
        [0.8026, 0.8590, 0.8404,  ..., 0.0000, 1.0545, 1.2045],
        [0.7489, 0.7490, 0.7258,  ..., 1.0545, 0.0000, 0.2706],
        [1.0145, 1.0174, 0.9935,  ..., 1.2045, 0.2706, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.72749923, 11.87665501, 12.05300943, 12.11735334, 12.12641969,
       11.86239961,  6.54615887, 11.89105259, 12.53345445,  6.61386436,
       11.83509372, 13.15525803, 12.26997165,  7.62011734, 10.74655346,
       10.19036081, 11.27656488,  7.42512621,  6.74686071,  9.68323686,
        9.78286733, 13.04156075, 11.70016217, 13.18278606, 10.66631784,
        5.40689332,  7.33350642, 11.72749923,  9.95552626, 11.21703654,
        7.76457095, 10.33712258, 13.0743399 ,  5.0713364 ,  7.32754882,
       12.22503754,  6.10774544,  8.81930525,  8.77913428,  5.81481044,
       10.7506423 , 13.04217487,  6.25586297,  6.65991877, 12.46513646,
        6.7593917 ,  7.54127425,  8.1303945 , 11.27656488, 12.33167382,
        7.17789666, 11.0050379 , 11.54071859, 12.67576792, 13.96746816,
        5.36061974, 12.889695  , 11.74901778,  5.14124612,  9.5642993 ,
       10.1979669 , 13.56367857, 13.83139024, 11.22596994,  5.39098669,
        7.18354885,  5.18277858,  5.13616224, 11.69264243,  8.06643487,
        7.62011734,  5.43544626,  8.01009083, 11.73397309,  9.66977802,
       10.65929968,  7.19513806, 13.04156075,  8.09704052,  9.7751189 ,
        8.55982105, 11.00812996,  8.16584537, 10.39942497,  8.06517143,
        4.75303127,  9.15976746,  6.55800267,  6.7166297 , 12.67680439,
        7.15990676,  9.60359158, 12.13568598, 11.99633819,  5.90035324,
       12.52372629,  9.17595899,  8.86082124, 13.85158417, 13.15525803,
        8.20436764,  4.90505566,  6.89245352,  9.34127013,  5.05534609,
        7.46053281])}
{'P': tensor([[4.5306e-03, 3.8964e-01, 1.5244e-01,  ..., 2.6726e-06, 1.4687e-05,
         1.5155e-02],
        [3.5359e-05, 7.3299e-03, 2.7372e-01,  ..., 6.4604e-06, 3.6634e-05,
         2.3621e-02],
        [4.1459e-05, 8.1704e-05, 1.7009e-02,  ..., 3.1290e-05, 2.0522e-04,
         7.2703e-02],
        ...,
        [1.4406e-06, 1.3862e-06, 1.3646e-06,  ..., 2.5297e-01, 1.9210e-04,
         1.3676e-06],
        [5.7067e-08, 5.9902e-08, 7.1664e-08,  ..., 8.9742e-02, 7.8170e-03,
         9.8791e-08],
        [1.1652e-05, 2.1874e-05, 8.0820e-05,  ..., 3.8665e-05, 3.6066e-04,
         9.2499e-03]]), 'X': tensor([[  7.8425,  -8.7202,   6.4530],
        [  8.6420,  -8.0580,   6.5781],
        [  9.9058,  -6.7203,   5.6928],
        [  9.2811,  -7.4336,   5.1771],
        [  9.8082,  -6.8396,   4.9615],
        [ 10.8218,  -5.4177,   6.8573],
        [  8.3718,  -8.2953,   8.4666],
        [ -4.8751,  -9.3148,   4.0406],
        [  7.4132,  11.3526,   0.3322],
        [  0.1287, -11.0065,   6.5757],
        [  5.2188,  -2.5839,   2.1833],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -2.0229, -10.6163,   8.6930],
        [  0.9945,  -4.8144,  -0.1897],
        [  1.7232,  -4.7574,   3.1270],
        [ -4.0409,   7.3169,  -3.5779],
        [ 11.1420,   6.9331,  -2.4779],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -2.2188,   7.8218,   8.6064],
        [ -5.6169,  -8.7844,   8.9057],
        [ -4.6393,   7.0461,   1.1624],
        [  9.9750,  -6.6338,  -1.2440],
        [ -9.4721,  -1.3097,  -7.4377],
        [  3.5284, -10.7484,   5.3347],
        [  3.6469,   6.3624,  -1.8414],
        [  3.5939,   6.4029,  -1.8357],
        [  7.7750,  11.0665,   2.8364],
        [  5.6523,   3.9443,  -2.9449],
        [ 12.3357,   3.6001,   0.7903],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.0588,  -6.0341,  -2.3006],
        [  3.7237,  -3.9966,  -5.0376],
        [ -8.8540,  -4.3971,  -5.2144],
        [  9.3560,   9.5391,   9.4943],
        [ 10.1983,  -6.3430,  -0.3200],
        [  5.1292,   4.8183,  -5.7997],
        [  9.0129,  -7.7072,   9.1400],
        [ 11.0438,   7.1180,  -5.6991],
        [ -8.3062,  -5.6008,   0.0287],
        [ -0.2586,   7.8825,  -9.9840],
        [ 12.2692,  -1.8882,   3.9253],
        [ -1.9487,   7.8590,  -3.3330],
        [ -7.0120,   5.2825,   7.7762],
        [ -2.2508, -10.5479,  -2.7235],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.2860,   7.1916,  -6.1011],
        [ -2.7572,   7.7194,   5.9580],
        [ -7.7842,   4.3466,  -1.7548],
        [ -3.9456,  -9.8542,   0.7466],
        [  6.3266,   0.3292,   9.3502],
        [  1.4306,  13.9620,  -0.2075],
        [ -7.6719,  -6.6260,   7.1468],
        [  0.9478,  -4.8144,   0.6371],
        [ 11.4235,   6.3594,   5.1663],
        [ -5.9872,   6.2037,  -9.4477],
        [ 11.4495,   6.3026,   6.8257],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -2.2188,   7.8218,   8.6064],
        [  5.4867,  12.5816,   2.9549],
        [  8.0381,  -8.5690,  -8.9523],
        [  0.2571,  -4.7594,   1.5618],
        [  6.9284,  11.7061,  -0.3732],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.2675,  12.1379,  -8.9720],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.2555,   2.1474,   4.1685],
        [  6.0054,  -1.0902,   8.3079],
        [  2.1379,  -4.6757,   6.4981],
        [ -8.9033,   2.2329,  -8.8912],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -9.0574,   1.7801,   8.8986],
        [  6.1752,   2.5242,   3.5874],
        [ -9.0574,   1.7801,   8.8986],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -8.0534,  -6.0429,   2.4583],
        [  3.5965,  -4.0754,  -3.5035],
        [  0.3493,  -4.7727,  -6.5682],
        [  9.5018,  -7.1945,  -1.1892],
        [  0.5250, -11.0306,   5.5358],
        [ -5.0550,   6.8231,  -9.2502],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.0042,  -2.8350,   2.6864],
        [  7.5784,  -8.9141,   5.8410],
        [  2.4356,  -4.5942,  -5.0808],
        [  2.3928,   7.1404,  -1.7035],
        [  9.6614,   9.1784,  -1.2087],
        [  6.0955,   2.8210,  -4.3321],
        [  8.1528,  10.7452,   9.1493],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.2804,   2.0019,   0.8607],
        [  3.4604,  -4.1545,   1.8091],
        [  6.9956,  11.6590,  -7.6876],
        [ -1.5875,   7.8944,  -9.4286],
        [  6.2895,   0.0405,   5.7652],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.8994,   6.8675,  -9.5987],
        [ -4.3963,   7.1629,   4.5263],
        [  2.2659,  13.7894,   6.5436],
        [  6.6319,  11.9066,   3.5165],
        [ 11.2209,  -4.7092,   2.6117]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0610, 0.1413,  ..., 0.6043, 0.6582, 0.4368],
        [0.0610, 0.0000, 0.0903,  ..., 0.5442, 0.6065, 0.4234],
        [0.1413, 0.0903, 0.0000,  ..., 0.4707, 0.5176, 0.3573],
        ...,
        [0.6043, 0.5442, 0.4707,  ..., 0.0000, 0.2416, 0.5719],
        [0.6582, 0.6065, 0.5176,  ..., 0.2416, 0.0000, 0.4493],
        [0.4368, 0.4234, 0.3573,  ..., 0.5719, 0.4493, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.72802491, 11.8159254 , 11.9702762 , 11.89105259, 11.95743558,
       12.10222204, 11.785566  , 10.51340829, 13.55868684, 11.00726483,
        5.82345867, 10.46739073, 10.80728788,  4.9160962 ,  5.05989547,
        8.35856487, 13.12299145, 12.32545637,  8.1303945 , 10.42667093,
        8.43626114, 11.97949273,  9.562174  , 11.31276109,  7.33350642,
        7.34250948, 13.5247215 ,  6.89245352, 12.85032836,  9.97220799,
        6.02588517,  9.79039371,  6.652742  , 10.0674887 ,  5.46244988,
        9.88571713, 13.3614598 , 12.00995898,  7.03733775, 11.85890664,
       13.13888841, 10.01804223,  7.88677538, 12.41366754,  8.09704052,
        8.77913428, 10.78533838,  5.05774295,  7.54620849,  8.19703512,
        8.91551231, 10.61472636,  6.33517349, 14.03505624, 10.1371552 ,
        4.90676242, 13.0743399 ,  8.62162372, 13.06957425, 12.49291768,
        8.1303945 , 13.72594022, 11.74901778,  4.7663526 , 13.60275211,
       12.07225469, 13.66051884,  8.91551231,  6.61386436,  6.10360073,
        5.14124612,  9.17905491,  9.97964013,  9.23071698,  6.67121748,
        9.23071698,  9.61942445,  9.97822473, 10.06850677,  5.43544626,
        4.78544875, 11.9182956 , 11.04313641,  8.49161154, 10.8992524 ,
       12.3344536 , 11.70016217,  5.19986797,  7.53063885, 13.32613445,
        6.7166297 , 13.48809595, 11.69264243,  6.59176018,  5.40689332,
       13.59673406,  8.05242591,  6.28961988, 10.61417181,  5.22903117,
       13.13205357,  7.4544866 ,  8.40445872, 13.97430191, 13.62896593,
       12.16901615])}
{'P': tensor([[3.2594e-03, 2.7221e-01, 2.2249e-01,  ..., 1.5072e-06, 1.3178e-05,
         2.5804e-06],
        [2.2986e-05, 5.2829e-03, 4.1004e-03,  ..., 1.6104e-06, 3.3139e-05,
         6.6596e-06],
        [3.5168e-05, 6.4123e-03, 6.8290e-03,  ..., 2.0964e-06, 4.7600e-05,
         9.0182e-06],
        ...,
        [1.6299e-01, 9.7618e-02, 9.2447e-02,  ..., 4.9607e-02, 9.0196e-06,
         2.5833e-06],
        [4.0448e-07, 4.7415e-07, 4.9612e-07,  ..., 2.5263e-07, 5.8418e-02,
         2.1245e-01],
        [7.4576e-08, 7.4455e-08, 7.6477e-08,  ..., 1.4102e-07, 4.8799e-06,
         1.5094e-02]]), 'X': tensor([[ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0640, 0.0722,  ..., 0.3553, 1.0428, 0.9019],
        [0.0640, 0.0000, 0.0331,  ..., 0.4135, 1.0101, 0.8651],
        [0.0722, 0.0331, 0.0000,  ..., 0.4275, 0.9812, 0.8375],
        ...,
        [0.3553, 0.4135, 0.4275,  ..., 0.0000, 1.3593, 1.2300],
        [1.0428, 1.0101, 0.9812,  ..., 1.3593, 0.0000, 0.1610],
        [0.9019, 0.8651, 0.8375,  ..., 1.2300, 0.1610, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.73397309, 11.83570132, 11.83509372, 12.00995898, 11.9182956 ,
       11.97949273,  8.62631801, 11.51557166,  7.38844703, 12.3489324 ,
        8.99791316,  9.37296102,  9.60359158,  8.25828409, 11.75618513,
       11.89855267,  9.48969978,  6.81084724, 13.14777433, 10.65929968,
       14.10045797, 13.36092624, 10.61472636, 10.7167285 ,  8.62162372,
        7.48538905, 11.05413545,  7.46248947, 14.05633441,  9.27474386,
        5.26342203,  5.47866221, 12.18045978,  9.70108409,  7.32754882,
       10.9036226 , 13.27994304,  9.58072286, 11.51557166, 12.05395056,
        9.562174  ,  8.37954853, 12.11735334,  5.13182506,  9.43112957,
       11.9702762 ,  7.50500064,  5.87132718,  5.21280156, 12.25806559,
        5.79543038,  9.23071698, 11.76830981,  4.96231532,  7.5092834 ,
        5.05981914, 11.9120321 ,  7.44805901, 12.30858677,  8.12043983,
       10.63038565, 13.26209545,  9.48337063, 12.88270018, 10.48721382,
        9.78286733, 13.62896593,  9.24501829, 10.71162681, 10.05558893,
        5.48276038,  8.09704052, 13.02818702, 13.59480906,  8.29640613,
        7.06372172,  6.81117802, 13.05148826,  8.74470218,  9.0368458 ,
        5.79562691,  6.76791483, 11.04313641,  5.63824971, 14.03505624,
       13.68063098, 11.28215429, 12.00995898,  8.29203356,  5.2287058 ,
        5.76165934,  6.28961988, 13.55868684, 13.83139024,  9.94173708,
        4.83371341, 13.68063098,  9.18690018,  8.41171369, 13.77604293,
       11.97238624, 10.0674887 ,  9.15976746, 11.05672136, 13.35013374,
       13.85158417])}
{'P': tensor([[9.7876e-03, 5.1174e-01, 1.4017e-01,  ..., 5.2875e-05, 3.5878e-07,
         1.4605e-05],
        [5.4582e-05, 1.0604e-02, 3.2955e-01,  ..., 7.7024e-05, 4.2157e-07,
         1.8344e-05],
        [1.3250e-05, 2.2090e-05, 1.1990e-02,  ..., 1.1344e-04, 6.1650e-07,
         1.9442e-05],
        ...,
        [9.4351e-06, 1.4286e-05, 3.4906e-05,  ..., 4.0875e-02, 4.7074e-05,
         1.6771e-03],
        [6.8245e-05, 4.7454e-05, 2.0427e-05,  ..., 6.1294e-05, 1.0248e-01,
         6.8965e-05],
        [2.8919e-05, 4.4901e-05, 1.0281e-04,  ..., 1.1816e-01, 5.0424e-05,
         1.1461e-01]]), 'X': tensor([[ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.7297e-02, 1.5432e-01,  ..., 9.5632e-01, 7.3099e-01,
         1.0768e+00],
        [4.7297e-02, 0.0000e+00, 1.0703e-01,  ..., 9.4182e-01, 7.6661e-01,
         1.0636e+00],
        [1.5432e-01, 1.0703e-01, 0.0000e+00,  ..., 9.2042e-01, 8.5326e-01,
         1.0445e+00],
        ...,
        [9.5632e-01, 9.4182e-01, 9.2042e-01,  ..., 0.0000e+00, 9.4832e-01,
         1.2501e-01],
        [7.3099e-01, 7.6661e-01, 8.5326e-01,  ..., 9.4832e-01, 2.1453e-04,
         1.0136e+00],
        [1.0768e+00, 1.0636e+00, 1.0445e+00,  ..., 1.2501e-01, 1.0136e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.73567759, 11.85890664, 12.13568598, 12.07225469, 12.27993098,
       12.21532282, 13.25336001,  7.92607342,  7.46053281,  5.5513072 ,
        8.14433542,  7.07902036,  6.39187285, 11.22993218, 10.66631784,
       12.62711252,  7.50035355,  8.8580671 , 10.26327895, 11.83509372,
       10.04916678,  8.53014915,  7.37576154,  5.9510489 , 12.91505111,
        9.94173708, 13.25017211, 11.75438006, 10.36844813,  5.80312949,
        9.17016857,  6.55800267, 11.83570132,  7.49076899, 13.14690137,
        7.5984809 ,  6.89245352, 13.58101794,  9.12220271, 10.75443106,
       10.95593729, 10.48721382, 10.08001894, 13.35832725,  6.65259775,
        8.43626114,  6.67108662,  7.46248947, 13.56367857, 11.22596994,
       11.11969815,  8.4486771 ,  4.9160962 ,  6.97106912, 13.85158417,
       11.73848065, 11.70016217,  6.54615887,  9.58072286,  8.72896381,
        7.50917551,  9.43737523, 13.08414689, 11.22993218,  8.94395628,
       10.31886066, 12.7125079 ,  8.74470218,  6.97106912,  8.90405823,
        6.59176018, 10.61472636,  5.5748624 ,  9.5014146 , 11.04629471,
        7.33350642,  7.46248947, 14.10045797, 12.0495521 ,  5.76165934,
        7.9337647 ,  6.78325151,  8.7273569 ,  6.06416957,  5.34250225,
       13.36092624,  7.23276582,  6.02588517,  4.74770899, 11.4424816 ,
        4.92497514, 11.4424816 ,  8.99791316, 12.10222204, 13.44593519,
       11.05672136, 12.88270018,  9.43112957, 11.00508681, 11.72749923,
       10.66631784,  6.67121748,  9.9353921 ,  6.39187285,  9.3109884 ,
        6.00732193])}
{'P': tensor([[9.3616e-03, 2.0058e-01, 1.9243e-01,  ..., 8.5907e-08, 7.6906e-05,
         8.4253e-07],
        [2.7203e-04, 3.2866e-02, 2.1502e-01,  ..., 1.5611e-07, 9.9885e-05,
         1.8093e-06],
        [1.3030e-05, 7.2659e-05, 7.3442e-03,  ..., 4.3763e-08, 5.1917e-06,
         3.0578e-07],
        ...,
        [8.6025e-05, 3.7697e-05, 6.0548e-05,  ..., 1.0702e-01, 3.1750e-04,
         7.2088e-04],
        [1.2584e-01, 6.1263e-02, 1.2825e-01,  ..., 3.8378e-07, 1.4164e-02,
         2.5864e-06],
        [3.2830e-03, 1.2886e-03, 9.1888e-04,  ..., 9.8343e-05, 1.3052e-02,
         2.9463e-02]]), 'X': tensor([[  7.9403,  -8.6455,  -8.7433],
        [  9.1620,  -7.5573,  -9.9882],
        [ 10.4994,  -5.9193,  -7.5153],
        [  9.0424,  -7.6779,  -7.5286],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.3048,   0.1469,  -9.1691],
        [ -9.1231,  -3.5739,   8.4708],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.1519, -10.2387,   9.8191],
        [  1.1274,  14.0116,   8.1753],
        [ -9.3076,   0.7852,  -6.7010],
        [  9.6614,   9.1784,  -1.2087],
        [  3.1892, -10.8220,  -6.8409],
        [  2.7685, -10.8982,  -6.8928],
        [  5.5240,  -2.1198,   6.2055],
        [  4.9957,  -2.8705,   6.5978],
        [  6.0054,  -1.0902,   8.3079],
        [  4.8674,  12.8875,   6.7622],
        [  6.2242,  -0.3171,   7.4978],
        [  1.3767, -11.0341,   9.9415],
        [  5.2939,  -2.4784,  -4.5840],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.7685, -10.8982,  -6.8928],
        [ 12.1940,  -2.1881,   7.1499],
        [  9.0129,  -7.7072,   9.1400],
        [  9.7328,   9.0902,  -0.8917],
        [ -7.4887,   4.7364,   8.9074],
        [ -0.0254, -10.9932,   3.2200],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.9788,   2.0208,  -3.2867],
        [ -8.9282,   2.1648,   8.4894],
        [  4.6063,  13.0051,  -3.1456],
        [  5.9587,  12.3216,   9.1695],
        [ -1.3178, -10.7939,   0.4236],
        [ -3.5173, -10.0623,   4.8280],
        [  5.6258,  -1.9404,  -6.0715],
        [  9.3607,   9.5337,  -4.5412],
        [ -7.6734,   4.4980,  -5.4248],
        [ -4.2040,   7.2486,   2.9875],
        [  2.2860,   7.1916,  -6.1011],
        [ 11.2209,  -4.7092,   2.6117],
        [ -8.3062,  -5.6008,   0.0287],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.2770,  -0.0379,   2.4195],
        [ -1.6906,   7.8860,   7.7142],
        [ -7.0948,   5.1940,   9.8171],
        [ -2.0840,   7.8415,  -7.8657],
        [  6.3583,   1.1729,   8.7835],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.4267,  -5.3697,   2.4172],
        [  1.3403,  -4.8011,  -1.1698],
        [ 12.4013,  -1.2597,  -7.6474],
        [  3.7298,  13.3537,  -9.4653],
        [  6.3515,   0.6418,  -6.5823],
        [  3.3744,   6.5623,  -6.3636],
        [  0.1041, -11.0045,   2.4311],
        [  6.3200,   0.2690,   1.5900],
        [  0.3038,   7.8092,   0.4164],
        [ 12.3609,   3.4769,  -8.5762],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.3589,   0.8239,  -8.4230],
        [ -5.7347,  -8.6908,   9.2878],
        [  4.1537,   5.9340,  -6.0423],
        [  1.1350,  14.0104,   0.6321],
        [  8.0381,  -8.5690,  -8.9523],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -6.5295,  -7.9785,  -7.4429],
        [  0.1041, -11.0045,   2.4311],
        [ -7.2515,   5.0195,   4.6434],
        [ -9.1178,   1.5784,   2.1410],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.5198,  -4.0923,   5.5088],
        [ -3.0644,   7.6439,  -1.4322],
        [  8.8474,  10.0868,  -8.8644],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.3246,   1.6696,  -0.1544],
        [ -9.1459,   1.4780,  -4.2403],
        [  1.5246,  -4.7838,  -5.2095],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.2373,  -4.6506,  -1.7334],
        [ -8.9942,  -3.9990,   6.0448],
        [  4.2041,  -3.6509,  -4.9488],
        [  9.6662,   9.1725,   8.1056],
        [  5.8450,  -9.9356,   3.0767],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.9548,  -1.2273,  -2.6520],
        [ -2.1382,   7.8339,   7.3524],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.2188,  -2.5839,   2.1833],
        [ 11.9122,   5.1397,  -1.5498],
        [ -3.2736, -10.1703,   3.8712],
        [ -0.6323,   7.9085,  -5.9544],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.7796,  -3.4696,   9.0088],
        [  2.1120,  -4.6818,   8.6144],
        [ -1.3178, -10.7939,   0.4236],
        [  5.1292,   4.8183,  -5.7997],
        [  1.7301,   7.4262,   5.0522],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.0467,  -9.8369,  -6.5816],
        [ -4.7847,  -9.3729,  -3.7741]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1122, 0.0937,  ..., 1.3568, 0.1923, 0.7789],
        [0.1122, 0.0000, 0.0281,  ..., 1.4341, 0.3039, 0.8904],
        [0.0937, 0.0281, 0.0000,  ..., 1.4060, 0.2825, 0.8682],
        ...,
        [1.3568, 1.4341, 1.4060,  ..., 0.0000, 1.2181, 0.9266],
        [0.1923, 0.3039, 0.2825,  ..., 1.2181, 0.0000, 0.5867],
        [0.7789, 0.8904, 0.8682,  ..., 0.9266, 0.5867, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.73848065, 11.87665501, 12.05300943, 11.86239961, 12.11735334,
       12.12641969,  6.30647746,  9.79814997, 10.61472636, 11.46176259,
       14.05687815,  9.34061938, 13.32613445, 11.28215429, 11.24434374,
        5.91677188,  5.76165934,  6.10360073, 13.77604293,  6.2322769 ,
       11.11969815,  5.84533345, 12.59840193, 11.24434374, 12.38881635,
       11.85890664, 13.31764181,  8.86082124, 10.99326577, 13.59480906,
        9.20340031,  9.18690018, 13.79676654, 13.68672855, 10.87409005,
       10.65929968,  5.9510489 , 13.36092624,  8.89458273,  8.37954853,
        7.54620849, 12.16901615, 10.01804223, 12.3489324 ,  5.16077764,
        6.27715105,  8.06517143,  8.79284466,  8.1137452 ,  6.46560181,
       11.21703654,  9.99211663,  4.98461983, 12.46513646, 13.86480175,
        6.383897  ,  7.37902766, 11.0050379 ,  6.32572323,  7.81509631,
       12.84056604,  7.09887557,  9.60359158, 12.59969936, 10.06286074,
        6.41202986, 10.412299  ,  7.24326992, 14.05633441, 11.74901778,
        9.71392179, 10.30972193, 11.0050379 ,  8.81930525,  9.25336371,
       14.10796466, 12.22503754,  8.23525433, 13.41713661, 10.60226614,
        6.54127431,  9.26455795,  5.0209005 ,  5.34250225,  5.16077764,
        9.84314712,  5.56809335, 13.32557133, 11.52733581,  9.48969978,
        6.07992623,  8.12043983,  9.71392179,  5.82345867, 12.97370371,
       10.68416401,  7.9337647 , 12.30695098, 12.27993098,  5.13616224,
       10.87409005,  7.03733775,  7.62509582, 10.42667093, 11.54672755,
       10.52357479])}
{'P': tensor([[1.0584e-02, 2.8857e-01, 2.0128e-01,  ..., 1.9200e-07, 1.2689e-07,
         2.2991e-07],
        [2.5803e-04, 3.2887e-02, 2.1516e-01,  ..., 3.9550e-07, 2.8560e-07,
         5.7500e-07],
        [1.2623e-05, 6.2744e-05, 6.3420e-03,  ..., 9.1668e-08, 8.0179e-08,
         1.5918e-07],
        ...,
        [1.1523e-04, 5.2644e-05, 5.5370e-05,  ..., 4.5549e-02, 2.2072e-04,
         1.1623e-04],
        [2.9004e-05, 1.3793e-05, 1.4556e-05,  ..., 5.4044e-01, 4.6784e-02,
         1.2811e-03],
        [1.6494e-05, 8.1697e-06, 7.1918e-06,  ..., 7.0331e-02, 2.7657e-02,
         2.0412e-02]]), 'X': tensor([[  8.0381,  -8.5690,  -8.9523],
        [  9.1620,  -7.5573,  -9.9882],
        [ 10.4994,  -5.9193,  -7.5153],
        [  9.0424,  -7.6779,  -7.5286],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 11.0373,   7.1299,  -8.5895],
        [  7.1821,  -9.1845,   9.8548],
        [ -8.1666,   3.7653,   3.6075],
        [ -7.4807,   4.7463,   2.0616],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -0.5385,   7.9037,   6.3830],
        [  6.1066,  -0.7799,  -4.1746],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.2687,  -2.9794,   0.6436],
        [  3.7121,   6.3117,   0.6738],
        [  3.6703,   6.3443,  -5.4401],
        [ -6.1914,  -8.2997,   8.5380],
        [  6.3122,  -9.6994,  -6.3282],
        [  2.5636, -10.9293,   9.6519],
        [  1.7500,   7.4187,   4.4237],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 12.0042,  -2.8350,   2.6864],
        [  6.3549,   0.7096,   6.9198],
        [  6.8407,  11.7666,   3.0045],
        [  2.2076,  13.8032,  -1.5099],
        [  4.5814, -10.4485,   2.5496],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.1795,  -3.3623,   8.5259],
        [ 10.2470,   8.4054,  -0.1177],
        [ -3.3812, -10.1235,   7.4384],
        [  0.1931,  -4.7491,   7.4385],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.0870,  -4.6877,   9.6823],
        [  3.4009,  -4.1874,  -5.9751],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.1624,  -3.4286,   2.0647],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.6801,   2.7809,  -0.8443],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.8474,   6.8979,   6.6457],
        [ -2.2508, -10.5479,  -2.7235],
        [  3.6817,   6.3354,   6.1233],
        [ -6.4368,   5.8356,   3.3014],
        [  0.1047, -11.0046,   0.5188],
        [ -1.8823, -10.6558,  -3.8952],
        [  0.6142,  14.0802,   0.5407],
        [ 12.2435,   4.0076,   8.1257],
        [  2.0870,  -4.6877,   9.6823],
        [  1.3767, -11.0341,   9.9415],
        [ 12.5996,   1.3971,  -1.0694],
        [  4.5767,   5.5071,   3.9069],
        [  4.8674,  12.8875,   6.7622],
        [  4.1637,  13.1898,  -1.9189],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 12.3025,  -1.7440,  -6.6632],
        [  1.3670,  -4.7990,   3.2603],
        [  1.0391,  -4.8141,  -0.0768],
        [ 11.9808,   4.9340,  -3.3715],
        [  4.2612, -10.5514,  -6.6268],
        [  5.0190,   4.9705,  -7.9752],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.2571,  -4.7594,   1.5618],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.9548,  -1.2273,  -2.6520],
        [  6.3583,   1.1729,   8.7835],
        [  0.3715,  -4.7756,   0.2312],
        [  5.9843,  -9.8679,  -6.1449],
        [  0.5175,  14.0910,   4.6540],
        [  2.7590,  13.6621,   9.4496],
        [  7.8375,  -8.7240,  -9.8913],
        [ 11.2209,  -4.7092,   2.6117],
        [  0.6757, -11.0361,  -1.3047],
        [  4.8653,   5.1694,  -4.0823],
        [  5.1203,  -2.7151,   6.2110],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.8836,   3.4315,  -4.0327],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.9461,  -4.4077,  -0.4831],
        [  6.3266,   0.3292,   9.3502],
        [ -5.1046,  -9.1610,   2.3418],
        [ 12.1940,  -2.1881,   7.1499],
        [  4.5057,  -3.3894,  -0.8160],
        [  1.7122,  -4.7591,  -3.6734],
        [  6.3505,   0.6235,  -8.5849],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.0899,  -4.6870,  -7.4260],
        [  5.6202,  -1.9508,   0.7471],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.6202,  -1.9508,   0.7471],
        [  6.3266,   0.3292,   9.3502],
        [  8.3791,  10.5409,   5.5590],
        [  6.2883,   0.0325,   5.2692],
        [  2.0547,   7.2957,  -1.9343],
        [ -3.0008, -10.2827,  -8.9480],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.0054,  -1.0902,   8.3079],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.7735,   2.5640,  -6.3313],
        [ -7.9925,  -6.1423,   2.8990],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.4739,  -1.2497,   0.5483]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0924, 0.0748,  ..., 1.1089, 1.1578, 0.9709],
        [0.0924, 0.0000, 0.0281,  ..., 1.1865, 1.2320, 1.0526],
        [0.0748, 0.0281, 0.0000,  ..., 1.1591, 1.2043, 1.0259],
        ...,
        [1.1089, 1.1865, 1.1591,  ..., 0.0000, 0.0892, 0.1666],
        [1.1578, 1.2320, 1.2043,  ..., 0.0892, 0.0000, 0.2494],
        [0.9709, 1.0526, 1.0259,  ..., 0.1666, 0.2494, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.74901778, 11.87665501, 12.05300943, 11.86239961, 12.11735334,
       12.12641969, 13.1399181 , 11.65923396,  8.99276564,  8.85939626,
       10.51437947,  7.92200342,  6.1561646 , 12.12448912,  9.37296102,
        9.73579738,  7.32233733,  7.32950385, 10.35464916, 11.57249984,
       11.22596994,  7.62232307,  9.43112957, 12.3344536 ,  6.39438044,
       13.6105595 , 13.97857591, 11.4088043 , 12.35421461,  9.77588801,
       13.25336001, 10.67322527,  4.75303127, 10.32715715,  9.53015376,
        5.13125513,  5.39450305,  5.24461068,  9.78285424, 13.27994304,
        9.11473391,  4.92497514,  7.46248947, 10.78533838,  7.32754882,
        8.68832398, 11.00508681, 10.8207297 , 14.09357547, 12.88270018,
        5.13125513, 11.11969815, 12.67680439,  7.16058587, 13.77604293,
       13.83139024, 10.32715715, 12.42554808,  4.98989906,  4.92497514,
       12.95702647, 11.37939695,  7.06372172, 10.7167285 ,  4.7663526 ,
       12.13568598,  6.07992623,  6.46560181,  4.79003361, 11.54071859,
       14.10045797, 13.93790235, 11.72749923, 12.16901615, 11.05672136,
        7.09887557,  5.79562691, 11.51811603,  6.81117802, 12.95702647,
        5.30158497,  6.33517349, 10.48721382, 12.38881635,  5.63824971,
        5.05774295,  6.38105015,  9.45995955,  5.13182506,  5.94909709,
        7.83486943, 12.7125079 ,  5.94909709,  6.33517349, 13.46553623,
        6.28835101,  7.57946459, 10.71162681, 12.25806559, 10.33712258,
        6.10360073,  8.95983586,  9.14044984, 10.08001894,  9.75200848,
        9.55593374])}
{'P': tensor([[3.5403e-03, 2.1822e-01, 1.5973e-01,  ..., 6.7883e-04, 5.3952e-02,
         1.0154e-06],
        [1.2212e-05, 3.8723e-03, 4.1874e-01,  ..., 1.0929e-03, 1.0265e-01,
         1.6223e-06],
        [1.5352e-05, 3.2848e-05, 6.5964e-03,  ..., 2.0942e-03, 1.6432e-01,
         3.1919e-06],
        ...,
        [1.0582e-04, 1.5630e-04, 1.8476e-04,  ..., 1.3033e-01, 3.1903e-04,
         1.8877e-03],
        [1.2738e-04, 3.3142e-04, 5.9259e-04,  ..., 4.8357e-02, 1.1507e-01,
         1.2099e-04],
        [7.5138e-06, 6.3885e-06, 6.0769e-06,  ..., 1.1204e-04, 7.0530e-06,
         1.8775e-01]]), 'X': tensor([[  8.0877,  -8.5297,  -1.1050],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.9750,  -6.6338,  -1.2440],
        [  9.4516,  -7.2501,  -1.8868],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 10.4737,  -5.9570,  -1.2108],
        [  4.8432,   5.1968,   1.5006],
        [  9.4925,   9.3811,   8.9842],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.1929,   2.4501,  -6.9352],
        [ -4.4779,  -9.5609,  -2.4505],
        [  4.9957,  -2.8705,   6.5978],
        [ 12.5650,   2.0132,   4.6272],
        [  2.0243,   7.3086,   7.5432],
        [ -6.1899,  -8.3011,  -7.9269],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.1575,  12.2046,   3.3257],
        [  2.5974,   7.0363,  -7.7810],
        [  6.1984,  -0.4327,   1.5438],
        [  5.1196,  -2.7160,   4.2705],
        [  9.1620,  -7.5573,  -9.9882],
        [ -1.9714,   7.8563,  -1.8084],
        [  3.7121,   6.3117,   0.6738],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -9.3076,   0.7852,  -6.7010],
        [  3.8182,  -3.9347,  -1.7961],
        [ -0.0410,   7.8591,   9.3457],
        [ -6.3489,   5.9117,   6.5161],
        [  3.3069,   6.6089,   8.7377],
        [  0.7145,  -4.8071,  -2.3766],
        [ -7.6650,   4.5092,   0.2520],
        [  6.0467,  -9.8369,  -6.5816],
        [ -9.4735,  -0.7274,  -2.6749],
        [  7.9403,  -8.6455,  -8.7433],
        [ -2.6069, -10.4298,  -6.1995],
        [ -3.9508,  -9.8515,   1.7923],
        [  4.5491,  -3.3485,   6.8412],
        [ -8.6050,   2.9451,  -0.7012],
        [ -5.1974,   6.7396,   3.3484],
        [  5.9548,  -1.2273,  -2.6520],
        [  2.3589,  13.7668,  -3.6501],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.2571,  -4.7594,   1.5618],
        [ -4.8751,  -9.3148,   4.0406],
        [  2.1771, -10.9776,  -3.2007],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.8934,   2.2597,   3.5831],
        [ -1.5408,   7.8978,   5.3470],
        [ -9.3076,   0.7852,  -6.7010],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.8323,  -1.5228,   5.1132],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.1274,  14.0116,   8.1753],
        [ -5.6174,   6.4708,  -1.7339],
        [  6.3460,   1.4252,   1.2291],
        [  3.9770, -10.6341,   8.0783],
        [ 11.5935,   5.9740,  -4.3952],
        [  9.2811,  -7.4336,   5.1771],
        [  4.8674,  12.8875,   6.7622],
        [ -9.4157,   0.0821,  -5.0025],
        [ -8.3653,   3.4184,   9.3454],
        [  6.3534,   1.2990,  -7.2942],
        [ -0.2586,   7.8825,  -9.9840],
        [ 12.5267,  -0.4125,  -8.7331],
        [  3.2692,  13.5098,  -4.9032],
        [ -0.8675, -10.8815,   6.8561],
        [  7.5309,  11.2617,   7.2656],
        [  3.5284, -10.7484,   5.3347],
        [  0.7275,  14.0667,  -0.9815],
        [  6.3266,   0.3292,   9.3502],
        [ -8.0588,  -6.0341,  -2.3006],
        [ 11.4703,  -4.2006,   7.4086],
        [  2.7590,  13.6621,   9.4496],
        [ -9.1658,  -3.4155,  -2.8880],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.3900,   3.3727,   3.1650],
        [  2.2373,  -4.6506,  -1.7334],
        [ -8.6112,   2.9319,  -7.9943],
        [  6.1929,   2.4501,  -6.9352],
        [  6.2555,   2.1474,   4.1685],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.5057,  -3.3894,  -0.8160],
        [  3.0876,   6.7527,  -5.3540],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -9.1623,  -3.4288,  -4.9961],
        [  6.2895,   0.0405,   5.7652],
        [ -6.7643,  -7.7372,  -9.6663],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.3744,   6.5623,  -6.3636],
        [ -8.1426,   3.8048,   9.3542],
        [  5.4732,  -2.2041,  -4.3830],
        [ -7.2986,  -7.1225,  -3.0814],
        [  3.3069,   6.6089,   8.7377],
        [ 12.1504,   4.3674,  -7.2998],
        [  3.4140,  -4.1803,  -8.8224],
        [ -8.8934,   2.2597,   3.5831],
        [ -1.7290,   7.8825,  -3.3831],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.1795,  -3.3623,   8.5259],
        [ 12.4597,  -0.9136,  -4.8763],
        [  1.0093,   7.6567,   2.3526],
        [  5.4867,  12.5816,   2.9549],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.1085,  -4.9204,   0.0736],
        [ -1.5261,   7.8988,   9.3768]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0621, 0.0957,  ..., 0.2871, 0.1899, 0.6772],
        [0.0621, 0.0000, 0.0337,  ..., 0.2322, 0.1418, 0.6901],
        [0.0957, 0.0337, 0.0000,  ..., 0.2060, 0.1237, 0.7020],
        ...,
        [0.2871, 0.2322, 0.2060,  ..., 0.0000, 0.1026, 0.6616],
        [0.1899, 0.1418, 0.1237,  ..., 0.1026, 0.0000, 0.6262],
        [0.6772, 0.6901, 0.7020,  ..., 0.6616, 0.6262, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.75438006, 11.9182956 , 11.97949273, 11.9120321 , 12.00995898,
       12.04924159,  7.10378919, 13.34587027,  5.5205189 ,  6.65991877,
       10.5575601 ,  5.76165934, 12.72524615,  7.58378655, 10.35485331,
       12.24467041, 13.66991027,  7.50035355,  6.2134961 ,  5.79543038,
       11.87665501,  8.09984231,  7.32233733,  9.97822473,  9.34061938,
        5.48276038,  7.85919687,  8.67502348,  7.39004226,  4.85995002,
        8.89301516, 11.54672755,  9.5014146 , 11.73848065, 10.7506423 ,
       10.61417181,  5.64865173,  9.09502385,  8.51088913,  6.07992623,
       13.96746816,  9.96152788, 10.36844813,  4.7663526 , 10.51340829,
       11.1913598 ,  6.27905292,  9.17595899,  8.04664886,  9.34061938,
       12.12448912,  6.02779201,  9.66977802, 14.05687815,  8.56889738,
        6.50409372, 11.35345345, 13.04217487, 11.89105259, 13.77604293,
        9.41605684,  9.0368458 ,  6.48486171,  7.88677538, 12.53345445,
       13.89974693, 10.91601655, 13.54774355, 11.31276109, 14.08549802,
        6.33517349, 10.0674887 , 12.21532282, 13.93790235,  9.78147542,
       13.89974693,  9.04255655,  5.16077764,  9.09661212,  6.65991877,
        6.61386436, 10.63038565,  5.63824971,  7.42512621,  9.51074763,
        9.78286733,  6.28961988, 10.27716434,  5.5366698 ,  7.37902766,
        8.98765644,  5.90035324, 10.1979669 ,  7.39004226, 12.91143491,
        5.39723241,  9.17595899,  8.06990483, 11.9120321 ,  9.77588801,
       12.49317563,  7.72291946, 13.72594022,  6.54615887, 12.1494023 ,
        8.04483495])}
{'P': tensor([[5.3262e-03, 3.3953e-01, 2.3601e-01,  ..., 2.9662e-04, 1.7303e-04,
         1.3923e-06],
        [3.4827e-05, 6.7430e-03, 3.7733e-05,  ..., 6.9641e-04, 6.0461e-05,
         2.9564e-06],
        [4.9850e-05, 4.8526e-01, 4.4018e-03,  ..., 3.6546e-04, 6.9070e-05,
         1.5773e-06],
        ...,
        [2.9141e-07, 3.9706e-07, 3.4895e-07,  ..., 1.1122e-02, 2.7636e-07,
         5.6103e-03],
        [2.2329e-02, 6.9837e-02, 5.2608e-02,  ..., 2.4831e-03, 3.6890e-02,
         9.7775e-06],
        [1.0080e-06, 1.0016e-06, 1.0237e-06,  ..., 7.3958e-06, 1.5550e-06,
         2.7625e-01]]), 'X': tensor([[  8.1043,  -8.5164,  -7.9290],
        [  9.0424,  -7.6779,  -7.5286],
        [  8.6271,  -8.0715,  -7.4813],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -6.3489,   5.9117,   6.5161],
        [  2.5974,   7.0363,  -7.7810],
        [  6.3550,   0.7129,   0.4917],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -6.9838,  -7.4965,   0.9796],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.1755,   2.5233,  -2.0042],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 11.5568,   6.0602,   9.3244],
        [  0.2480, -11.0153,  -3.0183],
        [ -6.6895,   5.6051,   2.7899],
        [  1.1274,  14.0116,   8.1753],
        [ 12.4597,  -0.9136,  -4.8763],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.9003,  12.3550,   2.1917],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -3.4488, -10.0934,   2.5195],
        [  6.3587,   1.1596,  -3.6629],
        [  0.4996,   7.7738,  -4.4711],
        [ -7.8280,   4.2848,  -7.1955],
        [ 12.4996,   2.6335,  -6.3166],
        [ -9.2974,   0.8371,   2.4852],
        [ -3.4488, -10.0934,   2.5195],
        [  1.9241,   7.3503,   7.9004],
        [ -6.0844,  -8.3955,   0.7705],
        [ -4.7642,  -9.3860,  -5.1133],
        [  0.6396,  14.0772,  -6.3786],
        [ 11.9695,  -2.9407,  -6.3498],
        [  4.0421,  -3.7765,  -8.3631],
        [ -1.8788, -10.6567,   3.7068],
        [  2.0449,  13.8402,  -7.1642],
        [ -5.6651,   6.4380,   1.3660],
        [  3.2809,  13.5061,  -4.0199],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.6801,  -3.7196,  -0.9181],
        [  6.0073,   3.0986,   6.7937],
        [ -9.4716,  -0.6673,   7.6068],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.6634,   5.8050,   9.1590],
        [  3.5965,  -4.0754,  -3.5035],
        [  0.9967,  -4.8144,  -9.7370],
        [ 10.0995,  -6.4739,   2.0974],
        [  9.6816,   9.1536,  -5.8543],
        [  9.1409,  -7.5789,   1.5814],
        [ -2.0840,   7.8415,  -7.8657],
        [  6.2022,   2.4092,   3.0349],
        [ -0.8885,   7.9160,   0.0571],
        [ -8.1426,   3.8048,   9.3542],
        [ -3.5029,   7.5138,   8.1853],
        [ -9.1749,  -3.3803,   6.4629],
        [ 11.7779,   5.5127,  -2.0699],
        [  5.8617,   3.4853,  -9.8080],
        [  2.1771, -10.9776,  -3.2007],
        [ -6.4620,  -8.0449,   6.2719],
        [ -2.2508, -10.5479,  -2.7235],
        [ -9.3464,   0.5708,  -1.6397],
        [  5.6523,   3.9443,  -2.9449],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.6970,   6.9826,   9.7350],
        [ 10.3251,   8.2928,  -3.0006],
        [ -9.4541,  -0.3327,   8.8246],
        [  3.5965,  -4.0754,  -3.5035],
        [ -9.4351,  -1.8891,   3.5642],
        [  6.3542,   0.6934,  -3.3840],
        [ -5.0943,  -9.1682,   0.5073],
        [ 12.4691,   2.8525,  -5.5110],
        [  1.8320,   7.3871,   5.1452],
        [  5.6928,  12.4710,   3.6001],
        [ -6.3066,  -8.1936,   9.5822],
        [ -0.8885,   7.9160,   0.0571],
        [  5.1787,  -2.6383,  -4.2533],
        [ -3.5173, -10.0623,   4.8280],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.1433,  -3.5003,  -3.0209],
        [  9.0004,  -7.7195,   4.6945],
        [  6.2675,  12.1379,  -8.9720],
        [  5.1667,  -2.6543,   1.2879],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -0.2586,   7.8825,  -9.9840],
        [  2.8994,   6.8675,  -9.5987],
        [  5.2199,   4.6861,  -1.9304],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.6672,  -1.8631,   8.1020],
        [  3.4920,  -4.1366,  -8.3955],
        [ -8.6675,  -4.8569,  -9.1769],
        [  0.5600, -11.0321,  -0.8660],
        [  9.6614,   9.1784,  -1.2087],
        [  1.7228,  -4.7575,   8.3396],
        [  5.1793,   4.7461,   0.3899],
        [  2.3889,  -4.6082,  -8.7291],
        [  6.1066,  -0.7799,  -4.1746],
        [  6.7792,  11.8084,   1.6251],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.9653,  -9.3225,  -3.4727],
        [  1.4306,  13.9620,  -0.2075]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0510, 0.0140,  ..., 0.5926, 0.2658, 0.6473],
        [0.0510, 0.0000, 0.0375,  ..., 0.5633, 0.2962, 0.6275],
        [0.0140, 0.0375, 0.0000,  ..., 0.5820, 0.2707, 0.6389],
        ...,
        [0.5926, 0.5633, 0.5820,  ..., 0.0000, 0.5577, 0.1432],
        [0.2658, 0.2962, 0.2707,  ..., 0.5577, 0.0000, 0.5504],
        [0.6473, 0.6275, 0.6389,  ..., 0.1432, 0.5504, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.75618513, 11.86239961, 11.81423161, 12.05300943, 12.11735334,
       12.12641969,  9.60080639,  8.67502348,  7.50035355,  6.3949043 ,
        5.05989547,  9.85330646, 10.24557047,  8.4486771 ,  6.67108662,
        9.70108409, 13.0493368 , 11.018084  ,  8.7273569 , 14.05687815,
       12.49317563,  4.91652556, 13.69162795,  9.7712321 , 10.66631784,
        6.46356806,  7.78980299,  8.92395381, 12.77402351,  9.3349858 ,
       10.66631784,  7.59795445, 10.36844813, 10.52587243, 14.09176259,
       12.32545637,  5.53175275, 10.82106505, 13.99048094,  8.57560275,
       13.89886469, 13.00413829,  6.81961996, 12.25806559,  6.7593917 ,
        9.49511524,  7.83486943, 13.02818702,  5.43544626,  4.91652556,
       11.99633819, 13.32374056, 11.87412439,  8.1137452 ,  6.65368551,
        7.96575782,  8.98765644,  8.29022197,  9.77777346, 13.00413829,
        6.81961996, 11.1913598 , 10.31886066, 10.78533838,  9.36378438,
        6.89245352,  9.97220799,  7.48538905, 13.2430377 ,  9.45995955,
        5.43544626,  9.62238828,  6.39187285, 10.48840476, 12.79126785,
        7.61088125, 13.70893492, 10.33958639,  7.96575782,  5.81199422,
       10.65929968, 13.79676654,  9.79039371, 11.85743425, 13.66051884,
        5.80860165, 10.5575601 ,  9.55720651,  7.88677538,  7.4544866 ,
        7.01475376, 10.78533838,  5.96556248,  5.41348121,  9.93553424,
       11.04629471, 13.32613445,  5.05981914,  7.02496223,  5.19065439,
        6.1561646 , 13.61600536,  6.01356004, 13.01187484, 11.63724792,
       14.03505624])}
{'P': tensor([[4.2072e-03, 2.0775e-01, 1.6527e-01,  ..., 1.2857e-05, 2.2313e-05,
         5.9564e-06],
        [1.5559e-05, 4.0862e-03, 8.8074e-05,  ..., 2.0257e-05, 3.7485e-05,
         1.3386e-05],
        [6.6284e-05, 8.3763e-02, 6.5617e-03,  ..., 2.5932e-05, 4.3389e-05,
         1.2674e-05],
        ...,
        [4.5855e-06, 4.2823e-06, 4.4747e-06,  ..., 4.2264e-02, 3.4079e-03,
         4.1929e-04],
        [6.2346e-06, 6.3094e-06, 6.0859e-06,  ..., 6.6812e-02, 5.2700e-02,
         2.0251e-03],
        [6.6351e-07, 7.1646e-07, 6.5160e-07,  ..., 1.2234e-02, 1.0314e-02,
         1.1489e-01]]), 'X': tensor([[ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0819, 0.0589,  ..., 0.1136, 0.1395, 0.6190],
        [0.0819, 0.0000, 0.0968,  ..., 0.1392, 0.0590, 0.5374],
        [0.0589, 0.0968, 0.0000,  ..., 0.0550, 0.1400, 0.6125],
        ...,
        [0.1136, 0.1392, 0.0550,  ..., 0.0000, 0.1679, 0.6197],
        [0.1395, 0.0590, 0.1400,  ..., 0.1679, 0.0000, 0.4797],
        [0.6190, 0.5374, 0.6125,  ..., 0.6197, 0.4797, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.75807252, 11.9182956 , 11.83570132, 11.97949273, 12.00995898,
       12.04924159, 11.87665501,  6.54693987, 10.86693708,  7.98469166,
        9.19837792, 13.26590071, 13.6105595 ,  9.77588801,  9.18358562,
        5.10444097, 11.00812996, 11.11038551, 13.45124917,  8.06990483,
       11.9120321 , 13.32645552,  8.06643487,  6.76791483,  9.09502385,
       10.01804223, 13.41713661,  8.12580976, 11.35345345, 11.97238624,
        7.54127425, 10.48721382,  6.48486171,  6.53250937,  6.50409372,
       12.22276278, 13.83139024,  9.78285424,  8.47371868,  7.27759233,
       11.35182589,  6.25586297, 11.27656488,  8.91551231,  8.16714836,
        7.72291946,  6.50409372, 13.35013374,  6.07992623,  7.19534146,
       13.1399181 , 10.46238054,  7.83534987,  7.16598273,  6.07992623,
        7.50917551,  8.04483495,  6.23262289,  8.97052115,  8.85939626,
        8.05668609, 13.84126197, 11.785566  ,  5.63824971,  5.41348121,
        7.83486943,  9.18690018,  6.32572323,  5.64500512, 13.79676654,
       13.35832725, 12.67680439, 13.25017211,  6.67108662, 10.35093485,
       10.61047596, 10.8992524 , 12.77402351,  8.50268322,  5.81481044,
       13.72594022, 10.66631784,  6.83083863,  8.04718433, 11.11969815,
       10.63038565, 10.26545792,  5.18047638,  9.75200848, 13.55868684,
        9.97822473, 12.58006038,  5.29755487, 13.99877413,  9.85330646,
       11.8159254 ,  8.57560275,  5.1805345 ,  7.39599986,  7.53063885,
        7.83486943,  9.91668182,  8.05668609,  7.73151309,  7.57946459,
       13.79762415])}
{'P': tensor([[7.2422e-03, 2.0586e-01, 1.5362e-01,  ..., 3.3401e-07, 1.9039e-07,
         2.3091e-05],
        [9.8260e-06, 6.0744e-03, 9.3423e-02,  ..., 2.8587e-07, 1.4853e-07,
         5.4925e-05],
        [1.7990e-05, 3.6438e-04, 1.5455e-02,  ..., 7.4784e-07, 3.3665e-07,
         1.3800e-04],
        ...,
        [7.1909e-05, 2.1544e-05, 1.6915e-05,  ..., 3.7986e-02, 3.9246e-03,
         9.0955e-06],
        [1.6728e-04, 5.6023e-05, 3.7607e-05,  ..., 1.5717e-03, 1.0093e-01,
         3.0933e-05],
        [1.2155e-05, 1.4334e-05, 1.3535e-05,  ..., 1.2978e-03, 1.3724e-03,
         1.4891e-01]]), 'X': tensor([[  8.2154,  -8.4262,   8.0171],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.0129,  -7.7072,   9.1400],
        [ 11.4703,  -4.2006,   7.4086],
        [  1.6954,  -4.7616,   9.8198],
        [  0.2435,   7.8191,  -6.4709],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.2804,   2.0019,   0.8607],
        [  0.5250, -11.0306,   5.5358],
        [  6.3413,   0.4883,  -6.8416],
        [  2.9260,  -4.4162,   8.3808],
        [ -3.4100,   7.5436,  -7.6980],
        [ -2.4956, -10.4682,   3.5632],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.2847,  -4.5848,  -4.7503],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.9936,  -2.8677,   6.0614],
        [  3.6817,   6.3354,   6.1233],
        [  9.6587,   9.1817,  -0.8212],
        [  2.6970,   6.9826,   9.7350],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.9956,  11.6590,  -7.6876],
        [ 12.0042,  -2.8350,   2.6864],
        [  6.1575,  12.2046,   3.3257],
        [ -8.9624,  -4.0944,  -4.9622],
        [  0.1382, -11.0073,   1.0026],
        [  5.4867,  12.5816,   2.9549],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.1206,  -4.7363,   8.7103],
        [ 12.0845,  -2.5772,  -4.6073],
        [  6.1752,   2.5242,   3.5874],
        [ -5.0132,  -9.2234,  -8.3531],
        [ 12.6059,   0.9597,  -6.3264],
        [  2.5620,   7.0549,   8.4825],
        [  6.2630,   2.1057,  -5.2136],
        [ -0.0254, -10.9932,   3.2200],
        [  6.9314,  -9.3435,  -1.2729],
        [  5.4429,  -2.2529,   9.7306],
        [ -7.2515,   5.0195,   4.6434],
        [  1.2728, -11.0372,   9.8121],
        [ -9.4351,  -1.8891,   3.5642],
        [ 10.3251,   8.2928,  -3.0006],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 10.0567,   8.6697,   4.6329],
        [  9.3836,   9.5075,  -5.9278],
        [ -9.4159,  -2.0739,   9.6841],
        [  8.9284,  10.0036,  -6.9210],
        [ -8.3717,  -5.4769,  -9.2228],
        [  8.6420,  -8.0580,   6.5781],
        [  5.0617,  12.7957,   8.3544],
        [  2.1824,  -4.6647,   1.1985],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.1433,  -3.5003,  -3.0209],
        [  0.2732,  -4.7619,  -1.1577],
        [  1.9505,  -4.7171,   9.9868],
        [  6.2022,   2.4092,   3.0349],
        [ 12.2592,   3.9425,   2.0383],
        [  2.0899,  -4.6870,  -7.4260],
        [ -8.1204,  -5.9308,   4.0761],
        [ 11.7162,   5.6728,  -8.9372],
        [  0.9788,  -4.8145,  -3.8661],
        [  8.1528,  10.7452,   9.1493],
        [ -9.0961,   1.6529,   3.1553],
        [  6.0467,  -9.8369,  -6.5816],
        [ -8.8835,  -4.3178,   9.3080],
        [  9.7328,   9.0902,  -0.8917],
        [ 12.5770,   1.8512,  -0.6578],
        [  3.2809,  13.5061,  -4.0199],
        [ -9.2974,   0.8371,   2.4852],
        [ -8.6426,  -4.9136,   9.4941],
        [ 12.2589,  -1.9312,   9.8657],
        [ -8.8934,   2.2597,   3.5831],
        [  9.0129,  -7.7072,   9.1400],
        [ 11.2209,  -4.7092,   2.6117],
        [  1.0391,  -4.8141,  -0.0768],
        [  7.3724,  -9.0577,  -8.9843],
        [ 10.6673,  -5.6649,  -4.1646],
        [  7.0171,  11.6439,  -5.0178],
        [  3.1064,  13.5607,   6.7605],
        [  0.1460,   7.8340,  -9.4614],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.2644,  -4.6434,   9.7508],
        [ -8.8058,   2.4850,  -1.7425],
        [  1.6422,   7.4586,  -6.8056],
        [  1.3767, -11.0341,   9.9415],
        [  7.9816,  10.8938,  -0.5378],
        [  5.6928,  12.4710,   3.6001],
        [  5.0277,   4.9589,   7.0175],
        [ 10.9875,   7.2207,  -8.8549],
        [  6.3301,   0.3638,  -7.5685],
        [  1.3339,   7.5627,  -0.9034],
        [  6.3460,   1.4252,   1.2291],
        [  4.0421,  -3.7765,  -8.3631],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -8.0829,  -5.9940,   5.0564],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.6249, -10.7255,   5.6605],
        [ -8.8745,   2.3097,   3.5600],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 11.2767,  -4.6006,   0.7868],
        [ -9.4716,  -0.6673,   7.6068],
        [ -9.3616,  -2.4749,   0.1793],
        [  2.3200,   7.1756,   1.1938]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1269, 0.1790,  ..., 0.5951, 0.9048, 0.5852],
        [0.1269, 0.0000, 0.0552,  ..., 0.7211, 1.0057, 0.5779],
        [0.1790, 0.0552, 0.0000,  ..., 0.7741, 1.0597, 0.6041],
        ...,
        [0.5951, 0.7211, 0.7741,  ..., 0.0000, 0.5120, 0.8497],
        [0.9048, 1.0057, 1.0597,  ..., 0.5120, 0.0000, 0.7930],
        [0.5852, 0.5779, 0.6041,  ..., 0.8497, 0.7930, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.76830981, 12.07225469, 12.13568598, 12.10222204, 11.85890664,
       12.21532282,  5.05444288,  7.822854  , 10.60226614,  6.59176018,
       11.04313641,  6.36004393,  5.29755487,  8.27853076, 10.76154256,
        8.16584537, 12.18045978, 11.70168755, 12.33167382,  7.32754882,
       13.32645552,  7.48538905, 12.33167382, 13.59673406, 12.3344536 ,
       13.66991027,  9.85330646, 11.00812996, 13.72594022, 12.42554808,
        4.73784222, 12.35625694,  6.67121748, 10.4977132 , 12.64235359,
        7.50564223,  6.60752373, 10.99326577, 11.63383221,  5.89074104,
        8.81930525, 11.11038551,  9.62238828, 13.2430377 , 10.31613269,
       13.2778378 , 13.35832725,  9.64157461, 13.40850328, 10.00410625,
       11.8159254 , 13.76047999,  5.14999799, 10.7167285 , 10.00802627,
        9.79039371,  4.76969364,  5.10446653,  6.65368551, 12.87752052,
        5.13182506, 10.05558893, 13.01728313,  4.91296205, 13.48809595,
        9.24501829, 11.54672755,  9.87720102, 13.31764181, 12.7125079 ,
       13.89886469,  9.3349858 ,  9.94173708, 12.41012046,  9.17595899,
       11.85890664, 12.16901615,  4.92497514, 11.67875855, 12.07818871,
       13.59480906, 13.91198191,  7.83534987,  7.8788965 ,  5.16611782,
        9.14973319,  7.63726134, 11.11969815, 13.50484539, 13.70893492,
        7.06169391, 13.14777433,  6.34059021,  7.67939546,  6.50409372,
        5.53175275, 10.60226614, 10.06286074, 10.41909211, 11.32149259,
        9.17016857,  9.43737523, 12.17901033,  9.49511524,  9.68323686,
        7.54127425])}
{'P': tensor([[3.2745e-03, 2.0380e-01, 1.9746e-01,  ..., 5.1753e-05, 3.7267e-04,
         8.3740e-04],
        [1.5866e-05, 4.3899e-03, 1.9174e-03,  ..., 2.5829e-05, 1.1445e-03,
         1.2690e-03],
        [3.7143e-05, 1.2047e-02, 9.4956e-03,  ..., 4.8757e-05, 2.4399e-03,
         2.2980e-03],
        ...,
        [5.5702e-03, 1.4519e-02, 9.6308e-03,  ..., 1.4062e-01, 2.0885e-03,
         1.5391e-02],
        [6.1093e-07, 8.6562e-07, 8.5422e-07,  ..., 3.1660e-07, 2.1345e-02,
         5.9586e-05],
        [1.4905e-04, 2.5127e-04, 2.3518e-04,  ..., 9.6468e-05, 9.0358e-03,
         6.1197e-02]]), 'X': tensor([[  8.2422,  -8.4041,  -1.4962],
        [  9.5018,  -7.1945,  -1.1892],
        [  9.4516,  -7.2501,  -1.8868],
        [  9.9750,  -6.6338,  -1.2440],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.4501,  -1.7133,   2.2980],
        [  2.4639, -10.9431,  -7.3718],
        [ -2.8162,   7.7059,  -9.6885],
        [ -0.8675, -10.8815,   6.8561],
        [  4.9337,  12.8566,  -3.6582],
        [ -6.2668,   5.9808,   2.5614],
        [  0.3354,  -4.7708,  -4.9125],
        [  4.0587, -10.6112,   6.2273],
        [  5.6022,  -1.9833,   8.3081],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -8.8058,   2.4850,  -1.7425],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -3.5029,   7.5138,   8.1853],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.0305,   3.0296,   0.1375],
        [  6.5428,  -9.5728,  -2.9897],
        [  9.4516,  -7.2501,  -1.8868],
        [ -7.7239,   4.4298,   3.1216],
        [ -8.8400,   2.3991,   1.3457],
        [  3.6817,   6.3354,   6.1233],
        [ -0.0410,   7.8591,   9.3457],
        [ -1.7009,   7.8851,  -8.2437],
        [  5.1203,  -2.7151,   6.2110],
        [  2.7590,  13.6621,   9.4496],
        [ 12.3609,   3.4769,  -8.5762],
        [  6.2975,   0.0947,   5.1791],
        [  2.1625,  -4.6697,   7.0658],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -8.0330,  -6.0765,   6.7945],
        [ 10.0567,   8.6697,   4.6329],
        [ -1.6220,   7.8917,   0.0315],
        [ -7.8359,   4.2735,   7.2943],
        [  8.8474,  10.0868,  -8.8644],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.4697,  -4.5836,   1.4593],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -7.5720,   4.6310,   2.1717],
        [ 12.6038,   0.7665,  -3.4439],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.3604,   0.4855,  -3.0158],
        [  3.1064,  13.5607,   6.7605],
        [  1.3339,   7.5627,  -0.9034],
        [ -3.4488, -10.0934,   2.5195],
        [  7.5930,  -8.9037,   7.2323],
        [  3.9515,  -3.8425,   6.1729],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -8.1204,  -5.9308,   4.0761],
        [  0.6363, -11.0348,   6.1414],
        [  3.7298,  13.3537,  -9.4653],
        [ -3.3116, -10.1539,   7.5919],
        [ -4.7334,   6.9982,  -0.4392],
        [  3.7298,  13.3537,  -9.4653],
        [  7.4132,  11.3526,   0.3322],
        [  0.9445,   7.6736,   0.5095],
        [  6.3200,   0.2690,   1.5900],
        [  6.2242,  -0.3171,   7.4978],
        [  0.2435,   7.8191,  -6.4709],
        [  6.3413,   0.4883,  -6.8416],
        [ -6.2668,   5.9808,   2.5614],
        [  2.6321,  -4.5296,  -2.5159],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.9433,  -9.3362,  -0.3421],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.8201,   3.5841,  -8.0609],
        [  0.9659,  -4.8144,  -0.9594],
        [  5.8323,  -1.5228,   5.1132],
        [ -5.6792,  -8.7352,   4.0318],
        [ 12.4691,   2.8525,  -5.5110],
        [  9.3560,   9.5391,   9.4943],
        [  2.2373,  -4.6506,  -1.7334],
        [  6.9956,  11.6590,  -7.6876],
        [ -3.1671,   7.6158,   6.0765],
        [  7.5930,  -8.9037,   7.2323],
        [  1.3103,  -4.8032,  -0.9300],
        [ 10.9394,   7.3069,  -8.9365],
        [ -8.9178,   2.1936,   7.5388],
        [  4.1140,  -3.7220,  -0.1273],
        [ -7.4602,   4.7718,  -4.6277],
        [  2.1477,   7.2550,   8.9688],
        [  9.0652,   9.8597,  -3.5949],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.5101,  -2.1433,   9.5442],
        [  2.6613,   7.0021,   4.5783],
        [ -9.4735,  -0.7274,  -2.6749],
        [  5.1787,  -2.6383,  -4.2533],
        [  2.7420,  -4.4896,  -3.7139],
        [  4.4633,   5.6287,  -0.7754],
        [  5.2608,  -2.5256,  -9.1446],
        [  0.3354,  -4.7708,  -4.9125],
        [ -5.0550,   6.8231,  -9.2502],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.4753,  -0.8012,  -5.2050],
        [  1.5662,  13.9375,   4.5142],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.4743,   6.2478,  -2.1100],
        [  6.1984,  -0.4327,   1.5438]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0584, 0.0759,  ..., 0.7578, 0.4891, 0.3585],
        [0.0584, 0.0000, 0.0625,  ..., 0.7368, 0.4419, 0.3021],
        [0.0759, 0.0625, 0.0000,  ..., 0.7989, 0.4184, 0.3366],
        ...,
        [0.7578, 0.7368, 0.7989,  ..., 0.0000, 0.9759, 0.6138],
        [0.4891, 0.4419, 0.4184,  ..., 0.9759, 0.0000, 0.3627],
        [0.3585, 0.3021, 0.3366,  ..., 0.6138, 0.3627, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.77124373, 11.9182956 , 11.9120321 , 11.97949273, 12.04924159,
       12.00995898,  9.14973319,  9.60412735, 11.21703654,  8.20436764,
       10.91601655, 13.77075174,  8.66273171,  4.78258092, 11.36089355,
        5.94293185,  9.60359158,  9.14973319, 12.49291768,  8.29022197,
       10.67132524,  5.24461068,  6.74872981, 11.59512194, 11.9120321 ,
        8.90405823,  9.15976746,  7.32754882,  7.85919687,  8.06643487,
        5.79562691, 13.93790235, 12.84056604,  6.29821673,  5.14607014,
        9.96152788, 12.32545637, 10.07238463, 13.2778378 ,  8.05668609,
        8.92549752, 13.41713661, 10.78533838, 13.27994304,  5.20659892,
       10.39942497,  8.87587457, 12.62711252, 12.0495521 ,  9.37296102,
       13.91198191,  7.67939546, 10.66631784, 11.70168755,  5.51177501,
        9.60359158, 10.05558893, 11.05317597, 13.86480175, 10.68031351,
        8.4486771 , 13.86480175, 13.55868684,  7.73151309,  6.32572323,
        6.2322769 ,  7.822854  ,  6.36004393,  8.66273171,  5.23878371,
        9.41605684, 11.63502679,  5.94541803,  6.83515201,  4.91038884,
        6.02779201, 10.41909211, 12.79126785, 13.3614598 ,  5.16077764,
       13.59673406,  8.24808832, 11.70168755,  4.978704  , 13.15525803,
        9.18358562,  5.54776934,  8.85572292,  7.56617037, 13.39372426,
       10.39942497,  5.91222346,  7.49076899,  9.5014146 ,  5.81199422,
        5.26067019,  7.18354885,  5.83560009,  4.78258092,  8.49161154,
        5.01114719,  9.50913849, 14.02526505, 11.70168755, 13.06498554,
        6.2134961 ])}
{'P': tensor([[6.9281e-03, 2.1900e-01, 1.8073e-01,  ..., 1.1311e-06, 1.2055e-05,
         2.0110e-06],
        [1.1347e-04, 9.0505e-03, 1.6586e-01,  ..., 1.0181e-06, 1.2825e-05,
         2.9353e-06],
        [9.6809e-06, 2.1221e-05, 5.0371e-03,  ..., 3.0967e-07, 1.6647e-05,
         6.3597e-06],
        ...,
        [2.9191e-03, 1.7866e-03, 1.5197e-03,  ..., 2.3403e-02, 9.2689e-05,
         1.0912e-06],
        [8.4471e-06, 7.9406e-06, 1.9288e-05,  ..., 1.9888e-05, 1.1898e-01,
         1.6846e-02],
        [6.9490e-08, 6.3900e-08, 9.9976e-08,  ..., 5.0790e-07, 2.2552e-05,
         3.2469e-02]]), 'X': tensor([[  8.3718,  -8.2953,   8.4666],
        [  9.0129,  -7.7072,   9.1400],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.8218,  -5.4177,   6.8573],
        [ 11.4703,  -4.2006,   7.4086],
        [ -9.4159,  -2.0739,   9.6841],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -8.8363,   2.4086,  -5.6296],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -2.7572,   7.7194,   5.9580],
        [ -6.6998,   5.5954,   0.0823],
        [ -5.9940,   6.1985,   5.9566],
        [ 12.3453,  -1.5462,   7.0212],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.9936,  -2.8677,   6.0614],
        [ 11.2767,  -4.6006,   0.7868],
        [  1.3705,  13.9723,   7.9266],
        [ -2.3317,   7.8034,  -3.6274],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.9003,  12.3550,   2.1917],
        [ 12.2592,   3.9425,   2.0383],
        [  6.2535,  -0.1709,  -9.4102],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.6650,   4.5092,   0.2520],
        [  1.4306,  13.9620,  -0.2075],
        [  2.3261,   7.1726,   6.2522],
        [  0.8940, -11.0402,  -0.9783],
        [  2.0547,   7.2957,  -1.9343],
        [ -7.2986,  -7.1225,  -3.0814],
        [  4.7156,  -3.1831,  -8.0224],
        [  1.6408,   7.4591,  -4.2294],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.3535,   0.6792,  -7.5101],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.2675,  12.1379,  -8.9720],
        [  5.6093,  -1.9705,   6.4807],
        [  5.8450,  -9.9356,   3.0767],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.8325,  -1.5222,  -2.1341],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4541,  -0.3327,   8.8246],
        [ -6.2200,  -8.2737,   5.5079],
        [  9.8082,  -6.8396,   4.9615],
        [  3.2990,  -4.2415,  -7.4857],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -8.9282,   2.1648,   8.4894],
        [  3.7916,  -3.9524,  -7.9547],
        [  9.8082,  -6.8396,   4.9615],
        [ -1.6086,   7.8928,   4.3385],
        [ -5.1377,   6.7751,  -0.4438],
        [ -8.0611,  -6.0302,   4.6962],
        [ 12.3453,  -1.5462,   7.0212],
        [  7.2340,  -9.1505,  -5.8321],
        [ -5.1046,  -9.1610,   2.3418],
        [ -2.3317,   7.8034,  -3.6274],
        [ -9.3464,   0.5708,  -1.6397],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -2.1817,   7.8274,  -4.6298],
        [  4.2096,   5.8815,   1.9230],
        [  2.8602,   6.8905,   5.8779],
        [ -9.3616,  -2.4750,   3.7765],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.2555,   2.1474,   4.1685],
        [ -3.9700,   7.3453,  -4.1486],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.4932,   7.0903,  -5.4214],
        [ -1.6086,   7.8928,   4.3385],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.3653,   3.4184,   9.3454],
        [  1.6308,  13.9254,  -2.0984],
        [  6.3236,   0.3011,  -4.6668],
        [  2.7121,   6.9743,  -8.4202],
        [ -6.4620,  -8.0449,   6.2719],
        [  1.5246,  -4.7838,  -5.2095],
        [  3.2647, -10.8066,  -0.3768],
        [ -2.9497, -10.3028,  -7.3479],
        [  3.1271, -10.8343,   2.6442],
        [  9.3607,   9.5337,  -4.5412],
        [  0.1252, -11.0062,   7.2564],
        [  2.2182,  13.8007,   5.1889],
        [  5.2188,  -2.5839,   2.1833],
        [  5.4167, -10.1293,   6.2156],
        [  1.1350,  14.0104,   0.6321],
        [ 12.5199,  -0.4711,   3.0436],
        [  5.6414, -10.0303,   7.8525],
        [ -0.0410,   7.8591,   9.3457],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -2.0229, -10.6163,   8.6930],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.3474,  -4.2162,   5.2455],
        [  4.5430,   5.5438,  -3.8669],
        [  6.1741,  -0.5329,   3.1913],
        [ -0.6323,   7.9085,  -5.9544],
        [  0.5850,  -4.7982,   9.6639],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.1426,   3.8048,   9.3542],
        [ -1.3178, -10.7939,   0.4236],
        [  6.1929,   2.4501,  -6.9352],
        [  4.5954,  13.0099,  -4.4439]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0565, 0.1097,  ..., 0.8200, 1.0846, 1.1760],
        [0.0565, 0.0000, 0.0642,  ..., 0.8706, 1.0954, 1.1752],
        [0.1097, 0.0642, 0.0000,  ..., 0.8889, 1.0577, 1.1275],
        ...,
        [0.8200, 0.8706, 0.8889,  ..., 0.0000, 0.8826, 1.1088],
        [1.0846, 1.0954, 1.0577,  ..., 0.8826, 0.0000, 0.2462],
        [1.1760, 1.1752, 1.1275,  ..., 1.1088, 0.2462, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.785566  , 11.85890664, 12.07225469, 12.13568598, 12.10222204,
       12.21532282,  9.64157461, 10.52357479, 10.51437947,  9.15866607,
        9.78286733,  8.19703512,  8.72896381,  8.62260327, 12.44177641,
       12.26997165, 12.33167382, 12.17901033, 14.0393949 ,  8.14433542,
       12.59840193, 13.69162795, 12.87752052,  6.25586297,  8.5724042 ,
        7.09887557,  9.7712321 ,  8.89301516, 14.03505624,  7.5403822 ,
       11.07637531,  7.57946459, 10.1979669 ,  5.68942572,  7.6374603 ,
        6.33075682,  9.64157461,  6.38968357, 12.0495521 , 13.66051884,
        5.94535529, 11.52733581, 10.33958639,  9.14973319,  6.02789634,
        8.04483495,  9.45995955, 10.35093485, 11.95743558,  5.37343173,
        9.79039371,  9.18690018,  5.47702766, 11.95743558,  8.0550286 ,
        8.50278896, 10.06703358, 12.44177641, 11.66453059, 10.48721382,
        8.14433542,  9.36378438, 12.52372629,  8.12580976,  7.23276582,
        7.46053281,  9.68324553,  9.51074763, 10.61417181,  6.61386436,
        8.34949383, 10.24557047,  7.5158489 ,  8.0550286 ,  7.58668144,
        9.0368458 , 14.02058621,  6.33075682,  7.48311002, 10.31886066,
        5.0209005 , 11.28895489, 10.7167285 , 11.27656488, 13.36092624,
       11.00694688, 13.97779839,  5.82345867, 11.48663767, 14.05633441,
       12.52875793, 11.50791859,  7.85919687,  9.47369628, 10.80728788,
       12.27993098,  5.38342077,  7.16747878,  6.19707944,  7.9337647 ,
        4.83371341,  6.01356004,  8.98765644, 10.87409005,  6.65991877,
       13.79762415])}
{'P': tensor([[4.4347e-03, 2.7115e-01, 2.5754e-01,  ..., 2.4118e-07, 6.0319e-07,
         4.8752e-04],
        [2.5140e-05, 4.3707e-03, 3.1916e-01,  ..., 2.4510e-07, 7.4678e-07,
         8.5782e-04],
        [1.5987e-05, 3.2184e-05, 5.5791e-03,  ..., 3.5199e-07, 1.0740e-06,
         1.7142e-03],
        ...,
        [2.2685e-05, 1.5682e-05, 1.2757e-05,  ..., 6.7714e-02, 1.0066e-04,
         1.4033e-06],
        [2.4924e-05, 2.1066e-05, 1.7143e-05,  ..., 3.5160e-03, 2.1316e-01,
         8.5032e-06],
        [8.6761e-07, 1.0936e-06, 1.3441e-06,  ..., 2.1892e-05, 2.1115e-04,
         3.1251e-02]]), 'X': tensor([[ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0625, 0.0932,  ..., 0.6967, 0.4093, 0.4961],
        [0.0625, 0.0000, 0.0443,  ..., 0.7581, 0.4424, 0.4382],
        [0.0932, 0.0443, 0.0000,  ..., 0.7745, 0.4866, 0.4031],
        ...,
        [0.6967, 0.7581, 0.7745,  ..., 0.0000, 0.6957, 1.1484],
        [0.4093, 0.4424, 0.4866,  ..., 0.6957, 0.0000, 0.8487],
        [0.4961, 0.4382, 0.4031,  ..., 1.1484, 0.8487, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.80092578, 11.89855267, 11.97957425, 12.10942412, 12.07818871,
       12.12448912, 13.66991027, 10.27572673, 11.80092578,  5.04028538,
        5.19986797,  9.77777346,  7.6374603 ,  5.54776934,  6.23262289,
       13.50484539, 13.97430191,  6.67108662,  8.20436764,  5.64464807,
       10.75443106,  6.83515201,  7.44805901,  7.34250948,  9.77588801,
        9.18358562,  6.90115901, 13.6105595 ,  9.38232121,  4.73784222,
       12.4980845 ,  5.1805345 ,  7.20482224,  6.61386436, 11.92133129,
       10.53368644, 11.83509372,  8.62162372,  8.06517143, 11.97238624,
        7.53884104, 10.06407839,  8.92098752, 11.28215429, 11.22142195,
       12.11735334,  5.64464807, 11.00726483, 13.14777433,  8.95983586,
       12.27835996,  8.77913428,  9.47369628, 12.86648771,  7.5984809 ,
       11.83570132,  5.05534609,  5.16611782,  6.54127431,  6.69389431,
        6.27905292,  5.64500512,  8.34949383,  8.09899874,  6.16512856,
        4.85603886, 12.0495521 , 11.9702762 , 11.48734978,  7.56617037,
       10.74409834,  6.35029274,  6.33517349,  9.97220799, 11.95743558,
       11.21703654, 11.73567759,  9.63742669,  5.20659892,  8.89458273,
       13.60275211,  5.1805345 ,  5.22903117,  6.39438044, 13.98691426,
       12.17451685,  5.05981914,  7.50035355, 13.55868684,  9.41244865,
        6.06416957,  9.11473391, 11.83509372,  7.16058587,  5.13182506,
        5.64500512,  9.48337063, 11.95743558, 13.59480906, 10.31886066,
       13.84126197,  6.3949043 , 12.45418236,  9.09963894,  8.62162372,
       13.09426142])}
{'P': tensor([[4.2917e-03, 4.7311e-01, 1.8549e-01,  ..., 3.2450e-04, 1.1499e-02,
         5.8104e-07],
        [3.5963e-05, 6.4267e-03, 3.6126e-01,  ..., 6.1230e-04, 2.1373e-02,
         1.0382e-06],
        [1.3230e-05, 1.6419e-05, 6.3045e-03,  ..., 1.5418e-03, 4.6017e-02,
         2.2098e-06],
        ...,
        [1.6390e-06, 1.8473e-06, 3.2448e-06,  ..., 5.9904e-02, 1.8984e-05,
         9.9140e-03],
        [3.5382e-06, 4.3333e-06, 1.0992e-05,  ..., 5.7996e-02, 1.2376e-02,
         6.3851e-05],
        [2.0515e-07, 2.0296e-07, 2.1568e-07,  ..., 4.3582e-06, 5.6737e-07,
         1.2190e-01]]), 'X': tensor([[ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0375, 0.1162,  ..., 0.5483, 0.3070, 1.2314],
        [0.0375, 0.0000, 0.0788,  ..., 0.5264, 0.2779, 1.2266],
        [0.1162, 0.0788, 0.0000,  ..., 0.4783, 0.2178, 1.2108],
        ...,
        [0.5483, 0.5264, 0.4783,  ..., 0.0000, 0.2660, 0.7989],
        [0.3070, 0.2779, 0.2178,  ..., 0.2660, 0.0000, 1.0368],
        [1.2314, 1.2266, 1.2108,  ..., 0.7989, 1.0368, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.81423161, 11.86239961, 12.05300943, 12.11735334, 12.12641969,
       12.27835996,  6.39520397,  6.61386436,  8.77913428, 11.22993218,
       13.59056613, 11.2845608 , 13.89671314,  7.48311002, 13.37121923,
        7.54620849, 11.05413545, 13.04051744,  9.85384623,  7.9337647 ,
       10.27572673, 13.83139024,  6.83083863,  6.67121748,  4.96231532,
        7.23276582, 13.83139024,  8.62631801,  7.73151309,  7.92607342,
        9.05843201,  6.49538841, 12.30858677,  6.05165248,  7.72415048,
       11.07637531,  7.83486943,  7.16058587, 12.71843462, 12.67680439,
        6.39438044, 10.33712258, 12.44177641, 12.13568598, 10.48721382,
       10.37698435,  6.4020404 ,  6.50409372,  7.5158489 ,  6.24614905,
        8.62260327,  6.24187044,  8.41171369,  4.74770899,  8.83725081,
        7.14837514,  8.91551231, 12.91143491, 13.85158417,  9.562174  ,
        5.64865173, 12.50996828,  7.39004226,  8.94395628,  5.87132718,
       10.60226614, 14.05687815,  5.76165934, 12.45418236,  4.91038884,
       12.84056604, 10.78533838,  4.87083348,  6.81084724, 12.88270018,
       11.00726483,  8.7273569 ,  5.86519525, 13.04051744,  5.43544626,
       10.05558893, 11.1341798 , 11.07637531, 13.37121923,  4.79003361,
       11.73567759,  7.07902036,  8.87007495, 10.06407839,  6.24187044,
       11.87412439,  4.91652556, 10.91601655, 12.86648771, 10.54658597,
       12.79126785,  8.67227386, 13.32374056,  7.44805901, 11.48663767,
        9.62238828,  9.78147542,  7.13501264, 13.06498554, 12.49317563,
       13.96987806])}
{'P': tensor([[4.9202e-03, 1.8373e-01, 1.6296e-01,  ..., 1.5985e-05, 1.1210e-03,
         5.4394e-02],
        [4.3102e-05, 8.9728e-03, 2.2780e-01,  ..., 6.9000e-05, 1.0561e-03,
         1.6962e-01],
        [8.7834e-06, 3.9221e-05, 5.2361e-03,  ..., 7.4260e-05, 1.1354e-04,
         2.9001e-01],
        ...,
        [1.4996e-07, 1.7287e-07, 1.9441e-07,  ..., 2.2068e-02, 1.6828e-07,
         2.7363e-07],
        [1.0082e-04, 5.5623e-04, 1.0710e-03,  ..., 8.4150e-05, 1.0471e-02,
         3.0001e-03],
        [6.9155e-05, 1.5475e-04, 2.3944e-04,  ..., 4.7156e-03, 3.7736e-04,
         1.0977e-01]]), 'X': tensor([[  8.6420,  -8.0580,   6.5781],
        [  9.9058,  -6.7203,   5.6928],
        [ 10.8218,  -5.4177,   6.8573],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.4703,  -4.2006,   7.4086],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.6614,  -4.5192,  -9.3641],
        [  8.4691,  10.4571,  -5.0650],
        [  6.0788,  -0.8710,  -6.8719],
        [  6.1685,  12.1979,  -6.2473],
        [  4.0587, -10.6112,   6.2273],
        [  9.3624,   9.5318,   0.2779],
        [  4.5325,  -3.3643,   6.7699],
        [ 11.7162,   5.6728,  -8.9372],
        [ -5.1369,   6.7755,   8.3836],
        [  0.7275,  14.0667,  -0.9815],
        [ -9.2707,   0.9661,   4.4863],
        [  5.6202,  -1.9508,   0.7471],
        [ -9.2707,   0.9661,   4.4863],
        [  4.1537,   5.9340,  -6.0423],
        [  0.1206,  -4.7363,   8.7103],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.5469,  13.7192,  -9.6935],
        [  4.5491,  -3.3485,   6.8412],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.3889,  -4.6082,  -8.7291],
        [ -0.8885,   7.9160,   0.0571],
        [  5.1203,  -2.7151,   6.2110],
        [ -9.1178,   1.5784,   2.1410],
        [  7.0642,  11.6104,  -9.8316],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 10.2712,   8.3707,  -4.9743],
        [  3.3943,   6.5483,   1.0790],
        [  2.7590,  13.6621,   9.4496],
        [ 11.1085,  -4.9204,   0.0736],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -9.2508,   1.0572,   2.9978],
        [  3.9022,  13.2906,   5.5788],
        [  9.1620,  -7.5573,  -9.9882],
        [  4.6231,   5.4556,   8.4973],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.7121,   6.9743,  -8.4202],
        [ -1.9646,   7.8571,   3.3817],
        [ -8.3717,  -5.4769,  -9.2228],
        [  4.2339,  -3.6268,   6.3125],
        [  6.2026,  12.1774,  -1.5887],
        [ -9.4159,  -2.0739,   9.6841],
        [  9.8082,  -6.8396,   4.9615],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.1929,   2.4501,  -6.9352],
        [ 12.4013,  -1.2597,  -7.6474],
        [  2.7590,  13.6621,   9.4496],
        [  5.8201,   3.5841,  -8.0609],
        [  1.1139,   7.6281,  -2.5680],
        [  5.5294,  -2.1106,   9.0626],
        [ 10.2470,   8.4054,  -0.1177],
        [ 11.2571,   6.7072,   8.3693],
        [  9.6614,   9.1784,  -1.2087],
        [  3.8111, -10.6787,  -4.6539],
        [  8.6420,  -8.0580,   6.5781],
        [ -7.4807,   4.7463,   2.0616],
        [  1.6954,  -4.7616,   9.8198],
        [  6.2975,   0.0947,   5.1791],
        [  6.3031,   1.8477,  -7.1706],
        [  8.6420,  -8.0580,   6.5781],
        [ -5.9872,   6.2037,  -9.4477],
        [  4.8653,   5.1694,  -4.0823],
        [  3.7916,  -3.9524,  -7.9547],
        [  7.9141,  -8.6656,   9.6948],
        [  9.4516,  -7.2501,  -1.8868],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.1231,  -3.5739,   8.4708],
        [ -3.5520,   7.4976,   9.1522],
        [  9.9058,  -6.7203,   5.6928],
        [ -6.1899,  -8.3011,  -7.9269],
        [  0.8940, -11.0402,  -0.9783],
        [ -0.3432,   7.8900,   7.1805],
        [ -7.0825,   5.2073,  -5.0497],
        [  3.7237,  -3.9966,  -5.0376],
        [  6.3048,   0.1469,  -9.1691],
        [ -2.8162,   7.7059,  -9.6885],
        [ -6.2668,   5.9808,   2.5614],
        [  6.3534,   1.2990,  -7.2942],
        [ 10.6673,  -5.6649,  -4.1646],
        [  8.5716,  10.3596,   6.8818],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.0965,  -5.9712,   4.3466],
        [ -7.0948,   5.1940,   9.8171],
        [  1.2273,  -4.8081,   9.8129],
        [  6.3211,   1.7017,  -0.2164],
        [ 10.5057,  -5.9099,  -5.1599],
        [  1.8766,   7.3694,   1.1887],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 12.5199,  -0.4711,   3.0436],
        [  8.9284,  10.0036,  -6.9210],
        [ -2.0229, -10.6163,   8.6930],
        [  4.7156,  -3.1831,  -8.0224],
        [  4.1999,  -3.6543,  -4.1998],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.9022,  13.2906,   5.5788],
        [  5.6928,  12.4710,   3.6001],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 11.8951,  -3.1570,   7.0742]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0903, 0.1351,  ..., 0.5986, 0.6440, 0.2436],
        [0.0903, 0.0000, 0.0867,  ..., 0.5102, 0.5993, 0.1811],
        [0.1351, 0.0867, 0.0000,  ..., 0.4844, 0.6718, 0.1086],
        ...,
        [0.5986, 0.5102, 0.4844,  ..., 0.0000, 0.6900, 0.3980],
        [0.6440, 0.5993, 0.6718,  ..., 0.6900, 0.0000, 0.7067],
        [0.2436, 0.1811, 0.1086,  ..., 0.3980, 0.7067, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.8159254 , 11.9702762 , 12.10222204, 12.07225469, 12.21532282,
       12.24467041,  5.24461068, 13.45642379,  6.14086582, 13.66896739,
       11.36089355, 13.3607379 ,  5.64464807, 13.01728313,  8.50268322,
       14.08549802,  9.32094055,  5.94909709,  9.32094055,  7.24326992,
        4.73784222, 13.27994304, 13.95361098,  5.64865173,  8.83341608,
        5.19065439,  7.96575782,  5.79562691,  9.25336371, 13.59056613,
       12.1460931 , 13.25017211,  7.37576154, 13.93790235, 12.1494023 ,
       11.27160035, 12.1460931 ,  9.3109884 , 13.85158417, 11.87665501,
        7.15100303, 12.10942412,  9.75401449,  7.48311002,  8.09899874,
       10.00410625,  5.5748624 , 13.6660671 ,  9.64157461, 11.95743558,
        5.01114719,  6.65991877, 12.46513646, 13.93790235,  6.83515201,
        7.70897675,  5.91854397, 13.25336001, 13.103709  , 13.32613445,
       11.33837396, 11.8159254 ,  8.85939626,  5.05444288,  6.29821673,
        6.56833864, 11.8159254 ,  8.62162372,  7.09887557,  5.47702766,
       11.73567759, 11.9120321 ,  8.5724042 ,  9.79814997,  8.29640613,
       11.9702762 , 10.35485331, 11.07637531,  7.89745107,  8.7907923 ,
        5.46244988,  6.30647746,  8.20436764,  8.66273171,  6.48486171,
       12.07818871, 13.44593519, 13.103709  , 10.06023665,  8.79284466,
        4.96231532,  6.54615887, 12.05395056,  7.60463349, 13.01187484,
        9.43737523, 12.52875793, 13.40850328, 10.80728788,  5.68942572,
        5.5671389 , 11.77124373, 13.85158417, 13.70893492, 12.00995898,
       12.30695098])}
{'P': tensor([[3.3817e-03, 3.3771e-01, 2.2821e-01,  ..., 3.6316e-07, 3.3174e-07,
         1.0290e-05],
        [2.4796e-05, 4.5629e-03, 4.0072e-01,  ..., 5.3634e-07, 5.2197e-07,
         1.1715e-05],
        [4.4139e-05, 6.2340e-05, 1.0941e-02,  ..., 1.4420e-06, 1.5271e-06,
         2.3712e-05],
        ...,
        [2.1255e-05, 1.8086e-05, 1.5336e-05,  ..., 9.6739e-02, 2.0830e-04,
         1.0718e-03],
        [4.9213e-06, 4.6676e-06, 4.5219e-06,  ..., 9.7121e-03, 2.0223e-01,
         5.3206e-04],
        [4.8382e-03, 4.9067e-03, 5.0323e-03,  ..., 1.4582e-04, 1.5475e-04,
         9.9057e-02]]), 'X': tensor([[ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0406, 0.0935,  ..., 0.5114, 0.9721, 0.6159],
        [0.0406, 0.0000, 0.0541,  ..., 0.5417, 0.9793, 0.6074],
        [0.0935, 0.0541, 0.0000,  ..., 0.5715, 0.9741, 0.5832],
        ...,
        [0.5114, 0.5417, 0.5715,  ..., 0.0000, 0.6477, 0.5767],
        [0.9721, 0.9793, 0.9741,  ..., 0.6477, 0.0000, 0.4715],
        [0.6159, 0.6074, 0.5832,  ..., 0.5767, 0.4715, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.83075005, 11.89105259, 11.95743558, 11.9702762 , 12.10222204,
       12.22503754,  9.17905491,  8.92098752,  6.2134961 , 11.37939695,
        6.55800267, 11.37939695,  9.99145096, 11.73397309,  7.62011734,
       13.14777433,  9.85384623,  7.92200342,  7.70897675, 12.62711252,
        6.53250937, 11.63502679,  8.09899874,  5.91854397, 11.27656488,
       11.28895489,  5.21280156,  9.88571713, 11.56146873,  5.84533345,
        7.5984809 , 12.53047489,  9.05843201,  6.24614905,  9.93553424,
        6.39438044, 13.01728313,  7.92200342, 13.12218013,  8.30125201,
       10.27572673, 12.12641969, 12.32408516, 11.4890861 ,  8.53501777,
       12.86648771,  7.33350642,  9.93553424,  6.05113217,  8.04664886,
       12.49317563, 10.42667093, 10.78533838,  9.75200848,  6.19633139,
        4.91296205, 12.49291768,  8.90405823,  8.19703512,  6.48486171,
        7.02326515,  6.80559669, 13.32613445, 13.20988996, 11.04629471,
        7.32950385,  6.38105015,  8.12580976, 12.53047489,  6.652742  ,
        8.86082124, 13.72594022,  5.86519525,  9.88571713,  9.41244865,
       12.30695098,  7.56617037,  5.34250225,  8.56889738,  6.42439344,
        9.34061938,  7.24326992,  9.62238828,  8.53014915,  6.02789634,
       12.40074901,  9.85384623, 11.63765273, 11.74901778,  8.5724042 ,
        8.31625171, 13.89264822, 13.15525803, 13.44593519,  5.56809335,
       10.69963954,  8.79284466,  9.37989631,  8.36445758,  5.8828986 ,
        4.98989906,  7.71075226, 11.22596994,  8.859416  ,  8.27853076,
        4.91038884])}
{'P': tensor([[5.5272e-03, 2.4894e-01, 1.7878e-01,  ..., 2.3837e-03, 7.8398e-05,
         2.3153e-05],
        [1.8416e-05, 5.2728e-03, 9.9708e-03,  ..., 8.1581e-04, 3.7876e-05,
         8.0859e-06],
        [1.7932e-05, 7.1231e-04, 5.8875e-03,  ..., 2.0090e-03, 3.1267e-05,
         8.7684e-06],
        ...,
        [4.2003e-05, 1.4721e-04, 2.9125e-04,  ..., 2.1117e-02, 2.1567e-05,
         1.6139e-05],
        [7.2670e-03, 9.5825e-03, 6.7084e-03,  ..., 8.4186e-04, 6.1926e-02,
         6.0286e-03],
        [2.6873e-01, 1.3399e-01, 7.5813e-02,  ..., 1.7685e-03, 1.3647e-04,
         4.8782e-03]]), 'X': tensor([[ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0880, 0.1267,  ..., 0.6665, 0.2427, 0.1055],
        [0.0880, 0.0000, 0.1355,  ..., 0.7053, 0.1598, 0.1721],
        [0.1267, 0.1355, 0.0000,  ..., 0.5703, 0.2771, 0.2283],
        ...,
        [0.6665, 0.7053, 0.5703,  ..., 0.0000, 0.8397, 0.7277],
        [0.2427, 0.1598, 0.2771,  ..., 0.8397, 0.0000, 0.2952],
        [0.1055, 0.1721, 0.2283,  ..., 0.7277, 0.2952, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.83509372, 12.00995898, 12.0495521 , 12.1494023 , 12.17901033,
       12.18998195,  8.74470218,  7.20482224,  5.80312949,  6.78325151,
        6.83515201, 12.11735334, 13.31764181, 11.1341798 , 13.40024224,
       11.8159254 ,  6.65259775, 12.35625694, 12.59969936,  5.53255882,
        7.28793663, 10.36844813,  6.25586297,  6.65991877,  6.42439344,
       12.67680439,  7.5158489 ,  6.99097013, 13.15525803,  8.53501777,
        9.58347456,  8.16584537, 12.59840193,  5.21280156,  9.95552626,
        6.81084724,  7.50035355,  5.53255882, 13.59673406,  6.42689705,
        4.96231532, 12.52875793,  5.94541803,  7.96181288,  5.05981914,
        4.73784222,  6.81084724, 10.37698435,  9.68323686, 13.60275211,
        9.62238828,  8.30729352,  5.38342077,  9.09661212,  9.14044984,
       12.07818871,  7.50564223,  7.8788965 ,  5.5366698 , 11.83509372,
       12.97370371,  5.91677188,  7.8575648 ,  6.54693987,  7.06169391,
        5.07306274,  9.50913849,  9.45995955, 12.45418236,  8.56889738,
       12.85032836,  8.16714836, 12.35421461, 10.06407839, 13.6105595 ,
       11.54071859, 13.96746816,  8.10886021,  6.2322769 , 11.9702762 ,
        7.98469166,  9.05843201,  5.07306274,  8.8737556 ,  8.12580976,
        8.99791316, 12.3489324 ,  5.90035324,  4.74770899,  8.68979052,
       11.50791859,  5.96556248,  8.24808832, 10.67322527,  5.10446653,
       12.24467041,  9.62238828,  6.54615887,  8.09899874, 12.7125079 ,
        5.5366698 , 10.24557047,  6.33517349, 12.21532282,  5.23878371,
       11.69264243])}
{'P': tensor([[5.3823e-03, 3.0594e-01, 1.5841e-01,  ..., 4.4439e-07, 1.2972e-06,
         2.1494e-05],
        [2.9102e-05, 9.7208e-03, 6.4222e-01,  ..., 9.9970e-07, 3.4107e-06,
         7.9264e-05],
        [3.3514e-05, 8.2129e-05, 2.3340e-02,  ..., 3.2508e-06, 1.1404e-05,
         4.0262e-04],
        ...,
        [1.0811e-05, 6.6196e-06, 5.0380e-06,  ..., 6.7137e-02, 7.8153e-04,
         2.7737e-06],
        [1.7682e-05, 1.2385e-05, 9.6348e-06,  ..., 1.4001e-02, 1.2192e-01,
         4.7244e-06],
        [2.9458e-07, 3.7676e-07, 5.3120e-07,  ..., 6.9656e-04, 3.8115e-04,
         4.6249e-02]]), 'X': tensor([[ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 5.1886e+00, -2.6250e+00,  6.4677e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0782, 0.1327,  ..., 0.6168, 0.2910, 1.0433],
        [0.0782, 0.0000, 0.0599,  ..., 0.6812, 0.3617, 1.0147],
        [0.1327, 0.0599, 0.0000,  ..., 0.7082, 0.4209, 0.9681],
        ...,
        [0.6168, 0.6812, 0.7082,  ..., 0.0000, 0.5596, 1.0837],
        [0.2910, 0.3617, 0.4209,  ..., 0.5596, 0.0000, 1.2844],
        [1.0433, 1.0147, 0.9681,  ..., 1.0837, 1.2844, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.83570132, 12.00995898, 12.1494023 , 12.04924159, 12.0495521 ,
       11.97949273, 11.89105259, 12.7125079 ,  7.06372172, 12.52372629,
       11.92133129, 13.40850328, 13.89264822,  9.562174  ,  7.61088125,
        9.5014146 ,  5.80860165,  8.05242591, 13.95472331, 13.35832725,
       13.20597932,  6.5618968 ,  6.2322769 , 10.35485331, 13.35013374,
        9.88310614, 13.35832725,  5.18277858,  8.53014915, 10.05558893,
        5.81481044, 13.68063098, 10.31613269, 13.61801526,  8.20436764,
        4.87083348,  8.08320235, 10.66631784, 13.50484539, 13.2430377 ,
       10.61472636, 11.27160035,  9.73579738,  5.29009836,  7.07902036,
       13.99048094,  8.92549752,  5.5748624 ,  6.61211224, 14.02058621,
       11.76830981,  6.42439344, 13.58101794,  5.88782913, 13.14777433,
        5.80860165, 11.21703654,  7.27340834, 14.0393949 , 13.39372426,
        6.05165248,  9.20340031,  9.95552626,  9.27474386,  9.88571713,
       10.67132524,  6.38399789, 12.21532282, 10.06703358,  5.19986797,
       12.3344536 ,  5.16077764,  7.33350642, 10.6910768 ,  7.57946459,
       10.98076873,  9.37989631,  6.81117802, 11.4424816 ,  9.78286733,
        9.3349858 , 13.59056613, 10.30972193,  9.75401449, 14.09357547,
       12.10222204,  6.05165248,  7.50917551,  7.06272256, 12.53345445,
        6.7593917 , 12.67576792, 13.04051744,  9.41244865, 11.51811603,
        7.39004226,  8.55982105,  7.38844703, 12.5085995 ,  5.1224079 ,
       11.70016217,  6.54615887,  7.48311002,  8.8580671 ,  8.5724042 ,
       13.48809595])}
{'P': tensor([[4.4617e-03, 3.4583e-01, 1.2119e-01,  ..., 2.0512e-06, 1.8367e-07,
         9.3994e-06],
        [5.1628e-05, 1.1207e-02, 2.3028e-02,  ..., 6.3849e-06, 3.3611e-07,
         1.7134e-05],
        [1.2105e-04, 3.5004e-03, 1.3489e-02,  ..., 7.0866e-06, 3.5061e-07,
         1.7770e-05],
        ...,
        [3.6381e-06, 3.1083e-06, 2.8046e-06,  ..., 5.0595e-02, 1.4320e-04,
         3.4216e-04],
        [5.6018e-04, 3.5463e-04, 2.6448e-04,  ..., 4.2480e-06, 3.2206e-02,
         4.4378e-03],
        [4.6076e-03, 3.8242e-03, 3.0069e-03,  ..., 2.8209e-04, 9.4694e-04,
         9.4586e-02]]), 'X': tensor([[ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0657, 0.1109,  ..., 0.5703, 1.0666, 0.5601],
        [0.0657, 0.0000, 0.0622,  ..., 0.6195, 1.1296, 0.5925],
        [0.1109, 0.0622, 0.0000,  ..., 0.6781, 1.1752, 0.6547],
        ...,
        [0.5703, 0.6195, 0.6781,  ..., 0.0000, 0.6130, 0.1945],
        [1.0666, 1.1296, 1.1752,  ..., 0.6130, 0.0000, 0.7927],
        [0.5601, 0.5925, 0.6547,  ..., 0.1945, 0.7927, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.85743425, 11.95743558, 11.9702762 , 12.22503754, 11.89105259,
       12.17451685, 11.63502679,  6.61211224,  9.7712321 ,  8.10886021,
       11.50791859, 13.00380624, 13.27717982,  9.36256096, 12.4980845 ,
       10.04916678,  9.84314712,  6.78325151, 10.08001894,  9.48969978,
       12.45418236, 13.39372426,  5.35774573,  9.23178204,  9.76896904,
       13.03002338,  9.37989631, 10.7167285 ,  5.56809335,  9.36093562,
       11.8159254 , 10.39942497,  7.73151309, 13.27994304, 11.87412439,
        6.76791483,  8.20436764,  7.89745107, 11.69264243,  8.4486771 ,
       12.53345445, 11.22993218,  6.50409372, 10.30972193,  7.32950385,
        6.383897  ,  7.16058587,  8.81930525,  8.859416  ,  6.89245352,
        4.978704  ,  9.43737523,  7.10378919,  6.83515201, 10.7167285 ,
       11.32149259, 13.96987806, 11.63383221, 13.0493368 , 10.9036226 ,
       11.48663767,  9.36093562, 10.92597196,  5.98285319,  9.11473391,
        5.54776934, 10.86693708, 11.04629471, 12.27835996, 10.52587243,
       10.06286074, 10.66631784, 12.16901615, 11.56146873, 12.1460931 ,
        9.51074763,  6.30647746,  8.92395381, 11.9702762 ,  9.62238828,
        6.02779201,  7.9337647 ,  7.69752262,  8.83725081, 11.64746882,
        8.22728399,  9.05843201,  8.12580976,  9.14044984, 13.52430176,
        9.41605684, 10.61047596, 12.17451685, 10.39942497,  9.55593374,
        7.02326515, 12.05395056,  6.29821673, 13.39372426,  5.2287058 ,
        9.45995955, 11.83570132,  7.24677664,  8.23525433, 10.54658597,
        4.76969364])}
{'P': tensor([[6.7903e-03, 2.1103e-01, 1.2444e-01,  ..., 2.1369e-04, 2.8869e-06,
         1.4168e-05],
        [1.1012e-05, 5.9771e-03, 1.4092e-04,  ..., 5.0597e-04, 2.2711e-06,
         2.8613e-05],
        [1.7778e-05, 6.4902e-02, 4.2200e-03,  ..., 3.0116e-04, 2.5875e-06,
         2.5112e-05],
        ...,
        [7.4426e-05, 1.0859e-04, 9.9761e-05,  ..., 3.6452e-01, 6.8998e-05,
         3.4632e-03],
        [1.6257e-05, 3.3198e-05, 4.3462e-05,  ..., 1.0411e-04, 8.2781e-02,
         8.6610e-03],
        [2.9724e-06, 5.0639e-06, 5.5458e-06,  ..., 6.0990e-04, 1.7891e-04,
         5.9806e-02]]), 'X': tensor([[  9.0129,  -7.7072,   9.1400],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.7796,  -3.4696,   9.0088],
        [ 11.4703,  -4.2006,   7.4086],
        [ 10.8218,  -5.4177,   6.8573],
        [ -2.6735, -10.4062,  -1.5911],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.1984,  -0.4327,   1.5438],
        [  4.5430,   5.5438,  -3.8669],
        [ -9.4296,  -1.9458,   4.0833],
        [  7.0515,  11.6194,   8.4289],
        [ -9.3107,  -2.7697,  -3.2919],
        [  3.4401,  13.4540,   4.6576],
        [  5.9883,   3.1536,   6.2320],
        [ 12.6035,   0.7494,   7.1536],
        [  2.4639, -10.9431,  -7.3718],
        [  9.4554,   9.4245,   9.8800],
        [  9.0129,  -7.7072,   9.1400],
        [ -4.3963,   7.1629,   4.5263],
        [  5.5158,  -2.1337,   3.5246],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -3.5029,   7.5138,   8.1853],
        [ 11.3800,  -4.3919,   1.4066],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.0604,   1.7707,   1.1066],
        [  0.1460,   7.8340,  -9.4614],
        [  5.1519, -10.2387,   9.8191],
        [ -4.6942,  -9.4299,  -9.1846],
        [  2.5829,  -4.5466,  -8.4015],
        [ -5.1974,   6.7396,   3.3484],
        [ -6.9838,  -7.4965,   0.9796],
        [ -9.3739,   0.3982,  -4.2564],
        [  6.0310,  12.2795,  -6.4528],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 10.4737,  -5.9570,  -1.2108],
        [  6.1929,   2.4501,  -6.9352],
        [ -9.4645,  -1.4880,  -2.0226],
        [  3.1271, -10.8343,   2.6442],
        [ -5.3328,  -8.9992,   7.3440],
        [ -9.1706,   1.3863,   8.4032],
        [  5.4732,  -2.2041,  -4.3830],
        [  3.8182,  -3.9347,  -1.7961],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.3587,   1.1596,  -3.6629],
        [  9.1409,  -7.5789,   1.5814],
        [  4.8051, -10.3703,   0.5268],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.2770,  -0.0379,   2.4195],
        [  6.1685,  12.1979,  -6.2473],
        [  2.2076,  13.8032,  -1.5099],
        [  0.6396,  14.0772,  -6.3786],
        [ -9.4157,   0.0821,  -5.0025],
        [ 11.6019,   5.9540,  -5.1715],
        [  5.6095,  -1.9702,  -7.8233],
        [  9.3607,   9.5337,  -4.5412],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.3413,   0.4883,  -6.8416],
        [  5.1292,   4.8183,  -5.7997],
        [  3.9663,   6.1018,  -5.4104],
        [ -1.5875,   7.8944,  -9.4286],
        [  9.8082,  -6.8396,   4.9615],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -5.6792,  -8.7352,   4.0318],
        [  6.7565,  11.8237,  -4.9882],
        [  5.2911,  -2.4824,  -3.6877],
        [ 11.4703,  -4.2006,   7.4086],
        [  5.8201,   3.5841,  -8.0609],
        [ 10.6283,  -5.7253,   7.4878],
        [ -2.1817,   7.8274,  -4.6298],
        [ -8.8400,   2.3991,   1.3457],
        [ 11.7796,  -3.4696,   9.0088],
        [ 11.9122,   5.1397,  -1.5498],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.5506,  -5.1154,   6.3423],
        [  0.9788,  -4.8145,  -3.8661],
        [ -7.8849,  -6.3115,  -4.7479],
        [  1.6954,  -4.7616,   9.8198],
        [  6.3048,   0.1469,  -9.1691],
        [ 11.3367,  -4.4806,   0.4643],
        [  6.7792,  11.8084,   1.6251],
        [ -9.3739,   0.3982,  -4.2564],
        [ 10.5057,  -5.9099,  -5.1599],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.2939,  -2.4784,  -4.5840],
        [  0.6952,  -4.8060,   8.8976],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.1771, -10.9776,  -3.2007],
        [ -3.9456,  -9.8542,   0.7466],
        [  7.7794,  11.0629,   8.9069],
        [ -3.0004,   7.6607,  -7.2640],
        [  3.9085,   6.1512,  -2.2560],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.1644, -11.0394,   7.8511],
        [  7.7750,  11.0665,   2.8364],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.3031,   1.8477,  -7.1706],
        [ -1.1885, -10.8211,   0.0865],
        [  4.4915,   5.5990,   7.0405],
        [  5.2608,  -2.5256,  -9.1446],
        [  4.5800,   5.5034,  -3.9432]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1070, 0.0642,  ..., 0.2463, 1.1320, 0.9582],
        [0.1070, 0.0000, 0.0552,  ..., 0.2018, 1.1132, 0.9439],
        [0.0642, 0.0552, 0.0000,  ..., 0.1928, 1.0945, 0.9227],
        ...,
        [0.2463, 0.2018, 0.1928,  ..., 0.0000, 0.9120, 0.7439],
        [1.1320, 1.1132, 1.0945,  ..., 0.9120, 0.0000, 0.1773],
        [0.9582, 0.9439, 0.9227,  ..., 0.7439, 0.1773, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.85890664, 12.13568598, 12.07225469, 12.27993098, 12.21532282,
       12.10222204, 10.74409834,  9.50654967,  6.2134961 ,  7.16747878,
        9.6282763 , 13.59170809,  9.71392179, 13.88683436,  6.76791483,
       12.62576026, 11.21703654, 13.35013374, 11.85890664,  8.40445872,
        5.91408504, 10.61047596,  9.69792513,  8.29022197, 12.19805394,
        7.71075226,  9.23178204,  7.83534987, 11.46176259, 10.53368644,
        5.22903117,  8.51088913, 10.24557047,  9.38232121, 13.68063098,
        9.4997987 , 12.04924159,  6.65991877,  9.58072286, 11.27656488,
       10.4606265 ,  9.27474386,  5.90035324,  5.48276038,  4.76969364,
        6.46356806, 11.87412439, 11.42947632, 12.27991991,  6.27715105,
       13.66896739, 13.97857591, 14.09176259,  9.41605684, 13.04051744,
        5.94541803, 13.36092624, 10.4977132 ,  6.36004393,  7.03733775,
        7.27759233,  8.05242591, 11.95743558,  9.69792513, 10.41909211,
       13.61801526,  5.84450127, 12.21532282,  6.83515201, 12.07225469,
        8.12580976,  9.15976746, 12.27993098, 12.97370371, 13.95361098,
        9.96391045,  4.91296205, 10.09979747,  5.05444288,  6.30647746,
       12.18998195, 13.61600536,  9.38232121, 12.05395056,  5.56809335,
        9.34127013, 10.78533838,  5.84533345,  4.85603886, 11.75807252,
       11.1913598 , 10.61472636, 13.52430176,  8.22728399,  7.28793663,
        5.5671389 , 10.04916678, 11.10066126, 13.5247215 , 12.97370371,
        9.62958973,  6.56833864, 10.88618114,  7.17789666,  5.83560009,
        7.15990676])}
{'P': tensor([[4.9809e-03, 2.7999e-01, 1.8827e-01,  ..., 2.9221e-06, 4.6717e-04,
         2.1839e-06],
        [1.0980e-05, 4.2162e-03, 2.4187e-01,  ..., 6.1217e-06, 5.1638e-04,
         4.4129e-06],
        [1.4059e-05, 5.8913e-05, 7.1632e-03,  ..., 1.4512e-05, 1.0710e-03,
         1.0980e-05],
        ...,
        [1.4140e-06, 1.3542e-06, 1.3758e-06,  ..., 2.8244e-01, 5.3229e-05,
         9.6590e-03],
        [8.7603e-05, 1.4729e-04, 1.8833e-04,  ..., 3.8393e-03, 1.1416e-01,
         6.6905e-03],
        [6.0871e-07, 6.1067e-07, 6.5207e-07,  ..., 2.6136e-02, 4.2527e-05,
         1.6788e-01]]), 'X': tensor([[ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0788, 0.1112,  ..., 0.1810, 0.4307, 0.6275],
        [0.0788, 0.0000, 0.0346,  ..., 0.1225, 0.4173, 0.6015],
        [0.1112, 0.0346, 0.0000,  ..., 0.0934, 0.4014, 0.5794],
        ...,
        [0.1810, 0.1225, 0.0934,  ..., 0.0000, 0.3205, 0.4889],
        [0.4307, 0.4173, 0.4014,  ..., 0.3205, 0.0000, 0.2092],
        [0.6275, 0.6015, 0.5794,  ..., 0.4889, 0.2092, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.86239961, 12.05300943, 12.11735334, 12.12641969, 12.27835996,
       12.32545637, 12.97370371,  9.04255655, 12.17901033,  4.76969364,
       11.83570132, 10.48315436,  6.90115901,  5.96556248,  6.00663314,
        6.35029274, 11.9702762 ,  6.38968357, 13.6922425 , 10.412299  ,
        9.78285424,  5.86519525,  7.5403822 , 10.26327895,  8.8580671 ,
        9.24501829,  8.25828409,  8.68832398,  8.31625171,  9.68324553,
        6.2134961 , 13.0493368 , 10.42667093, 13.25336001, 12.91505111,
        6.10360073, 12.11735334, 13.91198191,  9.05843201,  6.84796202,
       10.4606265 ,  4.9160962 , 13.02818702, 12.4980845 ,  7.44805901,
        9.19837792,  8.41148722, 10.71162681,  5.84450127,  5.51177501,
       14.10796466,  7.85919687,  9.6282763 ,  9.23178204,  7.16747878,
       11.2845608 ,  9.09661212,  7.8788965 , 13.6922425 ,  5.83560009,
        8.89301516, 10.38897563,  8.74470218, 10.06850677,  5.84533345,
       13.58101794,  9.68323686, 13.66991027,  8.44940661, 11.4088043 ,
        8.62162372, 12.41366754, 12.41012046, 11.54071859, 12.19805394,
        8.34949383,  9.37296102, 10.86693708, 10.23946247,  9.78285424,
       11.18839169,  5.52416368, 11.20100795,  8.44940661, 13.79762415,
       11.35182589, 10.06023665, 14.02532758,  7.56617037,  9.20340031,
        5.53255882,  9.53015376, 10.74409834, 13.99877413,  5.95111804,
       13.1471313 ,  7.03733775,  7.39599986,  8.12043983,  6.81117802,
       13.14690137,  7.39599986, 11.24434374, 14.09176259,  6.5618968 ,
       14.03505624])}
{'P': tensor([[2.6531e-03, 2.5063e-01, 1.5693e-01,  ..., 1.8455e-07, 7.6317e-08,
         1.9051e-05],
        [2.1080e-05, 3.2607e-03, 2.5580e-01,  ..., 2.3578e-07, 8.6117e-08,
         1.9619e-05],
        [1.0566e-05, 1.2890e-05, 3.7008e-03,  ..., 3.0529e-07, 7.5710e-08,
         1.3939e-05],
        ...,
        [1.1311e-05, 9.6457e-06, 6.3721e-06,  ..., 2.5727e-02, 7.9012e-04,
         3.9288e-04],
        [9.4431e-04, 8.2200e-04, 4.9561e-04,  ..., 1.9065e-05, 1.0742e-01,
         3.4847e-03],
        [1.1355e-02, 1.1476e-02, 1.0630e-02,  ..., 8.9814e-05, 9.7451e-05,
         6.9032e-02]]), 'X': tensor([[  9.1409,  -7.5789,   1.5814],
        [  9.5261,  -7.1675,   1.4206],
        [ 10.4758,  -5.9539,   1.3667],
        [ 10.0995,  -6.4739,   2.0974],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.2767,  -4.6006,   0.7868],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.0233,   4.9647,  -3.1788],
        [ 12.1504,   4.3674,  -7.2998],
        [ -0.0410,   7.8591,   9.3457],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.1575,  12.2046,   3.3257],
        [  0.5960, -11.0335,   3.1144],
        [  6.3266,   0.3292,   9.3502],
        [  0.9788,  -4.8145,  -3.8661],
        [ -9.2687,  -2.9794,   0.6436],
        [  0.1677,  -4.7447,  -6.1529],
        [  6.2905,   1.9366,   2.1169],
        [  1.6232,  -4.7717,  -2.5655],
        [  1.1006,   7.6318,  -2.4877],
        [  9.4554,   9.4245,   9.8800],
        [  9.0129,  -7.7072,   9.1400],
        [  6.1984,  -0.4327,   1.5438],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.1140,  -3.7220,  -0.1273],
        [ -5.1974,   6.7396,   3.3484],
        [  4.2339,  -3.6268,   6.3125],
        [ -3.5173, -10.0623,   4.8280],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.2437,  -4.6489,   6.1430],
        [ -5.1398,  -9.1367,   6.2074],
        [ 12.5154,   2.5072,  -9.3274],
        [ -9.4203,  -2.0340,   2.5856],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.1579,  -0.5961,  -6.5595],
        [ -0.8675, -10.8815,   6.8561],
        [ -9.1178,   1.5784,   2.1410],
        [ -3.0004,   7.6607,  -7.2640],
        [ -7.6650,   4.5092,   0.2520],
        [ -7.3554,   4.8985,  -0.6665],
        [  9.9750,  -6.6338,  -1.2440],
        [  5.8889,  -1.3916,   8.8993],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.4739,  -1.2497,   0.5483],
        [  4.2275,  -3.6320,  -0.2714],
        [ 11.3691,   6.4760,   1.0787],
        [ -3.5520,   7.4976,   9.1522],
        [  1.4150,  13.9647,  -4.0915],
        [ -7.3446,   4.9113,  -5.9356],
        [  0.9445,   7.6736,   0.5095],
        [  6.3048,   0.1469,  -9.1691],
        [ -4.3963,   7.1629,   4.5263],
        [ 11.2209,  -4.7092,   2.6117],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.1973,  -0.4372,   6.5720],
        [ -8.8400,   2.3991,   1.3457],
        [  5.9912,   3.1453,  -3.9989],
        [ 11.4235,   6.3594,   5.1663],
        [  9.3607,   9.5337,  -4.5412],
        [  1.9312,  13.8649,   8.3563],
        [  2.6310, -10.9195,   3.8054],
        [ -3.2053, -10.1993,  -3.8736],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.9036,   7.6840,   9.9237],
        [  1.7000,  -4.7609,   2.7935],
        [ -8.0087,   4.0173,   9.1091],
        [ -7.8849,  -6.3115,  -4.7479],
        [  9.5018,  -7.1945,  -1.1892],
        [  2.6613,   7.0021,   4.5783],
        [  8.2422,  -8.4041,  -1.4962],
        [  6.1730,  -0.5375,  -8.8861],
        [  3.9022,  13.2906,   5.5788],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -5.6423,   6.4537,  -1.2091],
        [  6.2905,   1.9366,   2.1169],
        [  6.3460,   1.4252,   1.2291],
        [  3.4401,  13.4540,   4.6576],
        [ 12.4837,   2.7510,   0.4449],
        [  6.3305,   1.6119,   1.1538],
        [ 11.5568,   6.0602,   9.3244],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.3505,   0.6235,  -8.5849],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.4748,  -0.7765,  -5.6695],
        [  0.9445,   7.6736,   0.5095],
        [ -6.7643,  -7.7372,  -9.6663],
        [ 11.9695,  -2.9407,  -6.3498],
        [  2.4932,   7.0903,  -5.4214],
        [  5.1203,  -2.7151,   6.2110],
        [  3.9515,  -3.8425,   6.1729],
        [  0.9788,  -4.8145,  -3.8661],
        [ 11.2209,  -4.7092,   2.6117],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.1624,  -3.4286,   2.0647],
        [  3.2701,   6.6338,   9.9209],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.8572,  14.0501,  -8.3364],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.8400,   2.3991,   1.3457],
        [ -6.1899,  -8.3011,  -7.9269],
        [  1.7903,  -4.7467,   0.1168]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0337, 0.0843,  ..., 0.5669, 0.9972, 0.1791],
        [0.0337, 0.0000, 0.0517,  ..., 0.5957, 1.0087, 0.1480],
        [0.0843, 0.0517, 0.0000,  ..., 0.6463, 1.0422, 0.1162],
        ...,
        [0.5669, 0.5957, 0.6463,  ..., 0.0000, 0.6596, 0.6904],
        [0.9972, 1.0087, 1.0422,  ..., 0.6596, 0.0000, 1.0086],
        [0.1791, 0.1480, 0.1162,  ..., 0.6904, 1.0086, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.87412439, 11.92133129, 12.0495521 , 11.99633819, 12.19805394,
       12.17901033, 12.17451685,  7.06272256, 12.91143491,  7.85919687,
        9.85330646, 13.66991027, 11.04953767,  6.33517349,  4.91296205,
        9.73579738,  4.74770899,  6.58184474,  5.04028538,  7.71075226,
       13.35013374, 11.85890664,  6.2134961 ,  4.91296205,  5.54776934,
        8.51088913,  5.5748624 , 10.65929968, 10.00410625,  5.16205087,
       10.48315436, 12.76408562,  9.63742669,  8.62162372,  6.18668463,
       10.91601655,  9.25336371,  8.22728399,  8.89301516,  8.83725081,
       11.97949273,  6.05113217,  5.21280156,  9.55593374,  5.57341623,
       13.08414689,  8.29640613, 14.03618649,  8.8353674 ,  7.73151309,
        6.30647746,  8.40445872, 12.16901615, 10.52587243, 13.00413829,
        6.21275016,  9.15976746,  6.76661595, 13.0743399 , 13.36092624,
       13.99877413, 11.23201619, 10.6910768 ,  5.01114719,  7.73692436,
        5.05534609,  8.95983586, 10.09979747, 11.9182956 ,  7.49076899,
       11.77124373,  6.19633139, 13.85158417, 10.51437947,  9.17016857,
        9.94285511,  8.5724042 ,  6.58184474,  6.50409372, 13.88683436,
       12.78327037,  6.53250937, 13.0493368 ,  9.58072286,  6.38105015,
        8.04483495,  9.50654967,  7.73151309, 10.27716434, 12.32545637,
        7.5158489 ,  5.79562691,  5.51177501,  4.91296205, 12.16901615,
       12.19805394,  9.78285424,  7.39599986, 12.1460931 , 13.09426142,
        9.64157461, 14.07623266, 11.83509372,  9.15976746, 10.35485331,
        5.07306274])}
{'P': tensor([[2.3967e-02, 1.5680e-01, 1.4439e-01,  ..., 7.4197e-08, 3.0495e-05,
         3.3569e-06],
        [6.4047e-05, 6.4737e-03, 3.7138e-01,  ..., 3.2309e-08, 2.5301e-05,
         7.7853e-07],
        [6.1947e-05, 9.6607e-05, 1.1746e-02,  ..., 6.3102e-08, 6.6310e-05,
         1.1609e-06],
        ...,
        [2.7077e-06, 3.4845e-06, 3.2274e-06,  ..., 3.3746e-02, 1.4045e-07,
         1.4752e-03],
        [3.4458e-08, 4.1831e-08, 4.4494e-08,  ..., 1.2570e-06, 7.0542e-03,
         2.7364e-08],
        [1.4037e-04, 4.7986e-04, 5.6195e-04,  ..., 2.4491e-06, 1.6543e-06,
         2.9411e-02]]), 'X': tensor([[  9.1620,  -7.5573,  -9.9882],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 10.9717,  -5.1646,  -6.9039],
        [ -8.1531,  -5.8748,   2.6934],
        [ -3.4488, -10.0934,   2.5195],
        [ -1.3940, -10.7772,   6.7835],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.9003,  12.3550,   2.1917],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.0604,   1.7707,   1.1066],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.8093,  -9.4178,   8.4244],
        [  7.5784,  -8.9141,   5.8410],
        [  4.3523, -10.5232,   8.2138],
        [  0.6396,  14.0772,  -6.3786],
        [  5.6093,  -1.9705,   6.4807],
        [  0.6757, -11.0361,  -1.3047],
        [  2.9410,   6.8428,   5.4458],
        [ -0.9715, -10.8630,   0.3555],
        [ -9.2974,   0.8371,   2.4852],
        [ 12.1940,  -2.1881,   7.1499],
        [  1.5246,  -4.7838,  -5.2095],
        [ -3.9456,  -9.8542,   0.7466],
        [  0.2435,   7.8191,  -6.4709],
        [  4.8674,  12.8875,   6.7622],
        [  6.1755,   2.5233,  -2.0042],
        [ -9.1623,  -3.4288,  -4.9961],
        [  5.5240,  -2.1198,   6.2055],
        [  5.7802,  -1.6363,  -6.2782],
        [  7.9816,  10.8938,  -0.5378],
        [  6.1372,   2.6726,  -6.7885],
        [  2.3490,  -4.6199,  -4.2489],
        [ 11.0373,   7.1299,  -8.5895],
        [ -4.3359,  -9.6431,  -9.0616],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.5763,   0.1333,  -9.2650],
        [  5.4732,  -2.2041,  -4.3830],
        [  0.5600, -11.0321,  -0.8660],
        [  2.8890,  -4.4316,   1.1600],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.3604,   0.9042,  -6.5901],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -9.1624,  -3.4286,   2.0647],
        [ 11.4703,  -4.2006,   7.4086],
        [ -2.1533, -10.5778,   1.0112],
        [  2.2182,  13.8007,   5.1889],
        [ 10.8670,  -5.3428,  -2.8637],
        [  1.7000,  -4.7609,   2.7935],
        [ -0.8885,   7.9160,   0.0571],
        [ -7.4808,   4.7462,   2.0266],
        [  0.6363, -11.0348,   6.1414],
        [ -0.3432,   7.8900,   7.1805],
        [  4.8674,  12.8875,   6.7622],
        [  7.1821,  -9.1845,   9.8548],
        [  6.2630,   2.1057,  -5.2136],
        [  1.1139,   7.6281,  -2.5680],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.8934,   2.2597,   3.5831],
        [ -3.3116, -10.1539,   7.5919],
        [ -5.1398,  -9.1367,   6.2074],
        [  8.5198,  10.4091,  -8.6486],
        [  5.6258,  -1.9404,  -6.0715],
        [ 12.4901,  -0.7052,   1.7315],
        [ -6.2668,   5.9808,   2.5614],
        [  3.2692,  13.5098,  -4.9032],
        [  4.5800,   5.5034,  -3.9432],
        [ -1.7009,   7.8851,  -8.2437],
        [  4.5325,  -3.3643,   6.7699],
        [ -8.0588,  -6.0341,  -2.3006],
        [  0.5250, -11.0306,   5.5358],
        [  2.5974,   7.0363,  -7.7810],
        [ -3.9456,  -9.8542,   0.7466],
        [  7.0515,  11.6194,   8.4289],
        [  5.7845,   3.6652,   0.9175],
        [  5.9548,  -1.2273,  -2.6520],
        [ -1.9487,   7.8590,  -3.3330],
        [  5.7151,  -1.7700,   3.1542],
        [  1.4150,  13.9647,  -4.0915],
        [ -9.4296,  -1.9458,   4.0833],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 11.4235,   6.3594,   5.1663],
        [  4.5954,  13.0099,  -4.4439],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.4290,  -3.4596,   1.4181],
        [ -4.7389,   6.9953,   3.7638],
        [  0.3724,   7.7974,  -6.8722],
        [  6.1984,  -0.4327,   1.5438],
        [ -6.3489,   5.9117,   6.5161],
        [ -8.8835,  -4.3178,   9.3080],
        [ -1.9600,   7.8577,   6.9121],
        [  8.5093,  -8.1764,  -3.2381],
        [  3.6249, -10.7255,   5.6605],
        [ -1.0012, -10.8576,   2.2867],
        [ -8.0330,  -6.0765,   6.7945],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.2905,   1.9366,   2.1169],
        [ -5.6423,   6.4537,  -1.2091],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 11.5568,   6.0602,   9.3244],
        [ -3.4986,   7.5152,   9.0712],
        [ -9.1231,  -3.5739,   8.4708],
        [  3.2692,  13.5098,  -4.9032],
        [  3.6249, -10.7255,   5.6605]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0281, 0.0379,  ..., 1.4329, 0.3122, 1.2020],
        [0.0281, 0.0000, 0.0346,  ..., 1.4048, 0.3160, 1.1745],
        [0.0379, 0.0346, 0.0000,  ..., 1.4156, 0.2816, 1.1777],
        ...,
        [1.4329, 1.4048, 1.4156,  ..., 0.0000, 1.5003, 0.3738],
        [0.3122, 0.3160, 0.2816,  ..., 1.5003, 0.0000, 1.2076],
        [1.2020, 1.1745, 1.1777,  ..., 0.3738, 1.2076, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.87665501, 12.05300943, 12.11735334, 12.35421461, 12.4980845 ,
       12.12641969, 10.04916678, 10.66631784, 10.86693708,  9.15866607,
       13.69162795, 10.61047596,  9.23178204, 10.31886066, 11.62158554,
       11.70016217, 11.38774149, 14.09176259,  5.94535529, 11.05672136,
        7.44805901, 10.90638066,  9.3349858 , 12.38881635,  5.0209005 ,
       10.61472636,  7.822854  , 13.77604293,  6.67108662,  9.78286733,
        5.91677188,  6.00732193, 13.50484539,  6.69389431,  5.18277858,
       13.1399181 , 10.57302546, 11.21703654, 12.5769659 ,  5.90035324,
       11.04629471,  5.29009836,  9.5014146 ,  6.42439344,  9.562174  ,
        9.78285424, 12.21532282, 10.794755  , 13.97779839, 12.10942412,
        5.05534609,  7.96575782,  8.859416  , 11.05317597,  7.89745107,
       13.77604293, 11.65923396,  6.60752373,  7.70897675, 13.08414689,
        9.17595899, 10.68031351, 10.48315436, 13.45124917,  5.9510489 ,
       12.50996828,  8.66273171, 13.89974693,  7.15990676,  8.06643487,
        5.64464807, 10.0674887 , 11.04313641,  7.50035355, 10.61472636,
       13.59170809,  6.84796202,  6.07992623,  8.09704052,  5.98285319,
       14.03618649,  9.6282763 , 12.25806559, 13.0743399 , 13.79762415,
       11.75618513,  5.62005629,  8.44940661,  7.80625981,  6.2134961 ,
        8.67502348,  9.87720102,  8.09842704, 11.80092578, 11.32149259,
       10.9036226 , 10.07238463,  7.96181288,  6.58184474,  8.5724042 ,
       12.05300943, 13.0493368 ,  8.28968067,  9.79814997, 13.89974693,
       11.32149259])}
{'P': tensor([[4.3627e-03, 3.8315e-01, 2.2350e-01,  ..., 3.4560e-06, 4.8703e-06,
         2.1991e-05],
        [5.7579e-05, 1.0106e-02, 2.0766e-02,  ..., 9.8996e-06, 8.2131e-06,
         4.0650e-05],
        [5.8536e-05, 1.9297e-03, 7.4362e-03,  ..., 6.7848e-06, 4.6629e-06,
         2.7566e-05],
        ...,
        [9.3918e-06, 9.0085e-06, 8.2482e-06,  ..., 1.3599e-01, 1.4794e-05,
         4.2386e-04],
        [5.1094e-04, 6.5887e-04, 4.6059e-04,  ..., 6.6912e-06, 1.9865e-02,
         7.3733e-05],
        [1.6711e-02, 1.6395e-02, 1.4603e-02,  ..., 3.7237e-04, 1.2480e-03,
         1.1292e-01]]), 'X': tensor([[ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0541, 0.0632,  ..., 0.5680, 1.0148, 0.3333],
        [0.0541, 0.0000, 0.0622,  ..., 0.5608, 1.0054, 0.2993],
        [0.0632, 0.0622, 0.0000,  ..., 0.6197, 1.0653, 0.3607],
        ...,
        [0.5680, 0.5608, 0.6197,  ..., 0.0000, 0.4470, 0.3367],
        [1.0148, 1.0054, 1.0653,  ..., 0.4470, 0.0000, 0.7494],
        [0.3333, 0.2993, 0.3607,  ..., 0.3367, 0.7494, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.89105259, 11.95743558, 11.9702762 , 12.22503754, 12.24467041,
       12.10222204,  9.37989631,  6.14086582, 13.89974693,  8.99276564,
        5.14999799, 13.68672855,  7.48538905,  7.02326515,  6.69389431,
       13.6105595 ,  7.60529892,  6.54615887, 12.41366754,  8.40445872,
        6.42439344, 11.28895489,  5.87132718,  6.81117802,  8.68979052,
        5.07306274, 13.99048094,  7.72291946,  4.78002665,  5.5366698 ,
       12.42554808, 10.06703358, 11.77124373,  6.3949043 ,  6.54615887,
        6.67218687, 13.68672855, 11.57249984,  9.09119351, 10.39942497,
        6.24187044,  6.83083863,  5.13125513,  6.27905292, 10.06703358,
        9.77588801, 11.54672755, 10.76154256, 13.97779839,  6.46356806,
       11.38774149,  6.81084724, 13.96987806,  8.22728399,  7.19513806,
        8.35856487, 11.22993218, 10.35093485,  6.29034958, 13.68672855,
        7.01475376,  7.28793663,  8.68979052, 12.50996828,  9.19837792,
       12.21532282,  7.5092834 , 11.72749923,  8.83725081, 13.66896739,
       10.07238463,  8.68832398,  6.42439344,  6.31576264, 12.12641969,
       13.66991027, 11.99633819,  9.562174  , 13.70893492,  5.76165934,
        7.80625981, 13.26209545, 14.09357547, 10.75443106, 10.52357479,
        9.64157461, 10.00794468,  7.15100303, 10.37698435, 10.46739073,
       13.91198191,  7.01475376, 11.70168755,  5.69913817,  6.5618968 ,
        5.91408504,  7.54620849, 12.67680439,  5.62005629, 10.66631784,
        5.10444097, 11.00694688,  5.47866221,  8.10886021, 11.54672755,
        4.98989906])}
{'P': tensor([[3.9856e-03, 2.9104e-01, 2.0776e-01,  ..., 8.9100e-08, 7.8617e-04,
         5.4550e-05],
        [2.7679e-05, 4.7982e-03, 1.6477e-01,  ..., 1.0016e-07, 6.8709e-04,
         5.1014e-05],
        [2.7383e-05, 7.9167e-05, 9.1770e-03,  ..., 1.5325e-07, 8.3137e-04,
         5.6776e-05],
        ...,
        [6.7191e-04, 5.2044e-04, 2.8741e-04,  ..., 2.0756e-01, 8.0612e-04,
         8.2942e-03],
        [1.7142e-03, 2.5234e-03, 3.9973e-03,  ..., 1.9103e-05, 7.0527e-02,
         2.1924e-03],
        [7.0740e-03, 9.0760e-03, 7.8647e-03,  ..., 1.3100e-04, 2.8565e-02,
         1.4184e-01]]), 'X': tensor([[  9.3424,  -7.3685,  -3.6807],
        [  9.9756,  -6.6330,  -3.3429],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.3236,   0.3011,  -4.6668],
        [  6.0310,  12.2795,  -6.4528],
        [  5.9548,  -1.2273,  -2.6520],
        [ 10.1798,   8.5002,  -5.8893],
        [ -1.9600,   7.8577,   6.9121],
        [ -4.6393,   7.0461,   1.1624],
        [  3.9022,  13.2906,   5.5788],
        [ -9.4501,  -1.7133,   2.2980],
        [ 12.6000,   1.3839,  -3.2992],
        [ -9.3464,   0.5708,  -1.6397],
        [  5.6672,  -1.8631,   8.1020],
        [  2.2076,  13.8032,  -1.5099],
        [  3.7237,  -3.9966,  -5.0376],
        [ -9.1813,  -3.3550,   1.8827],
        [  3.3069,   6.6089,   8.7377],
        [ -9.0961,   1.6529,   3.1553],
        [ 10.0400,   8.6922,  -6.2847],
        [  6.2242,  -0.3171,   7.4978],
        [  0.3038,   7.8092,   0.4164],
        [  7.1697,  11.5343,  -0.7276],
        [ 11.3118,   6.5957,  -4.5667],
        [ -8.6426,  -4.9136,   9.4941],
        [  4.1637,  13.1898,  -1.9189],
        [ 12.4877,  -0.7222,   8.8347],
        [  2.6970,   6.9826,   9.7350],
        [  0.9478,  -4.8144,   0.6371],
        [  5.9912,   3.1453,  -3.9989],
        [ -1.9714,   7.8563,  -1.8084],
        [  0.2480, -11.0153,  -3.0183],
        [  9.0424,  -7.6779,  -7.5286],
        [ 11.1420,   6.9331,  -2.4779],
        [  0.1047, -11.0046,   0.5188],
        [  5.8201,   3.5841,  -8.0609],
        [  5.9587,  12.3216,   9.1695],
        [  6.9653,  -9.3225,  -3.4727],
        [  1.0000,   7.6591,  -4.5011],
        [ -6.6998,   5.5954,   0.0823],
        [  4.8432,   5.1968,   1.5006],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.7218,  -9.9934,   6.5764],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.3261,  13.7749,   5.9121],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.9193,  -1.3176,   5.1739],
        [ -8.0829,  -5.9940,   5.0564],
        [ -4.0409,   7.3169,  -3.5779],
        [  2.1379,  -4.6757,   6.4981],
        [  8.8091,  -7.9037,   0.3631],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -4.3963,   7.1629,   4.5263],
        [  6.0890,  -0.8382,   2.9166],
        [ -8.1666,   3.7653,   3.6075],
        [  1.6308,  13.9254,  -2.0984],
        [ -1.1885, -10.8211,   0.0865],
        [ 11.0438,   7.1180,  -5.6991],
        [ -7.6650,   4.5092,   0.2520],
        [  2.3261,  13.7749,   5.9121],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.5662,  13.9375,   4.5142],
        [  3.1892, -10.8220,  -6.8409],
        [ -2.0840,   7.8415,  -7.8657],
        [ -4.0868,   7.2981,   8.1874],
        [  3.1064,  13.5607,   6.7605],
        [  5.0506,   4.9277,  -9.5094],
        [ 11.3691,   6.4760,   1.0787],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.3118,   6.5957,  -4.5667],
        [  3.9591, -10.6391,  -6.2557],
        [  5.6095,  -1.9702,  -7.8233],
        [  4.0587, -10.6112,   6.2273],
        [ -8.5900,   2.9767,   7.4811],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 10.2470,   8.4054,  -0.1177],
        [ -4.3359,  -9.6431,  -9.0616],
        [  5.8323,  -1.5228,   5.1132],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.9515,  -3.8425,   6.1729],
        [  6.0310,  12.2795,  -6.4528],
        [ 12.6059,   0.9597,  -6.3264],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.2555,   2.1474,   4.1685],
        [ -5.3382,   6.6533,   7.8054],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 10.9394,   7.3069,  -8.9365],
        [  2.4639, -10.9431,  -7.3718],
        [ -2.2508, -10.5479,  -2.7235],
        [ -0.0254, -10.9932,   3.2200],
        [ 11.5568,   6.0602,   9.3244],
        [  3.4920,  -4.1366,  -8.3955],
        [ -1.6906,   7.8860,   7.7142],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.4039,   5.6903,   8.7366],
        [ 10.3251,   8.2928,  -3.0006],
        [ -8.0829,  -5.9940,   5.0564],
        [  5.2911,  -2.4824,  -3.6877],
        [  3.4604,  -4.1545,   1.8091]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.4304e-02, 1.1221e-01,  ..., 1.0370e+00, 1.8841e-01,
         4.6056e-01],
        [4.4304e-02, 0.0000e+00, 8.8764e-02,  ..., 1.0527e+00, 1.4411e-01,
         4.3128e-01],
        [1.1221e-01, 8.8764e-02, 0.0000e+00,  ..., 1.1414e+00, 1.4124e-01,
         4.8225e-01],
        ...,
        [1.0370e+00, 1.0527e+00, 1.1414e+00,  ..., 0.0000e+00, 1.1130e+00,
         9.2414e-01],
        [1.8841e-01, 1.4411e-01, 1.4124e-01,  ..., 1.1130e+00, 0.0000e+00,
         3.5669e-01],
        [4.6056e-01, 4.3128e-01, 4.8225e-01,  ..., 9.2414e-01, 3.5669e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.89855267, 11.97957425, 12.07818871, 12.12448912, 12.10942412,
       12.18045978,  6.33075682, 13.68063098,  6.07992623, 13.26209545,
        8.09842704,  8.43626114, 13.85158417,  9.60412735, 12.67576792,
        9.36378438,  5.96556248, 13.97857591,  5.46244988,  9.7751189 ,
        7.39004226,  9.24501829, 13.27994304,  6.2322769 ,  7.81509631,
       13.58101794, 13.09426142,  9.94173708, 13.83139024, 12.5085995 ,
        7.48538905,  4.90676242,  6.76661595,  8.09984231, 11.018084  ,
       11.86239961, 13.12299145, 11.00508681,  6.83515201, 13.68672855,
       11.63724792,  7.72415048,  8.72896381,  7.10378919,  6.00663314,
        9.15866607, 11.51557166,  6.12886024,  9.79814997, 13.96987806,
       12.42554808,  6.06416957, 10.06286074,  8.35856487,  5.14124612,
       11.83509372, 10.35485331,  8.40445872,  6.1463896 ,  8.99276564,
       14.02058621, 10.88618114, 13.13888841,  8.89301516, 13.96987806,
        5.24137909, 14.02526505, 11.28215429,  8.1137452 ,  8.36445758,
       13.91198191,  7.05627461, 13.08414689,  8.85572292, 13.09426142,
       11.35182589,  5.94541803, 11.36089355,  9.09119351,  9.58072286,
       13.25336001, 10.57302546,  6.02779201,  9.9353921 ,  5.51177501,
       13.68063098, 12.64235359,  8.62260327,  9.43112957,  6.61386436,
        8.53014915,  5.18047638, 12.4980845 , 13.15525803, 11.21703654,
       10.78533838, 10.99326577, 13.0493368 ,  5.41348121,  8.06517143,
        9.54154096,  7.19534146, 13.2430377 , 10.06286074,  5.84450127,
        5.40689332])}
{'P': tensor([[5.2661e-03, 2.5666e-01, 2.2417e-01,  ..., 1.0271e-03, 1.5131e-06,
         7.2737e-04],
        [4.8796e-05, 1.0647e-02, 2.1565e-01,  ..., 1.5302e-03, 3.9463e-06,
         7.4992e-04],
        [5.4703e-05, 3.2550e-04, 2.5647e-02,  ..., 2.0060e-03, 1.3363e-05,
         1.4169e-03],
        ...,
        [1.5259e-03, 3.0467e-03, 4.1309e-03,  ..., 1.1212e-01, 3.8646e-04,
         8.8409e-03],
        [2.1237e-05, 1.5755e-05, 1.4651e-05,  ..., 1.5761e-04, 2.0696e-01,
         2.8936e-04],
        [2.7586e-03, 4.5212e-03, 8.0389e-03,  ..., 1.7252e-02, 6.6878e-04,
         1.2440e-01]]), 'X': tensor([[  9.4516,  -7.2501,  -1.8868],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 10.8670,  -5.3428,  -2.8637],
        [  9.9750,  -6.6338,  -1.2440],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -8.4573,   3.2448,   5.4370],
        [  6.6319,  11.9066,   3.5165],
        [  9.0652,   9.8597,  -3.5949],
        [ -3.2486,   7.5925,  -1.3719],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.6165,  -3.8715,   6.4594],
        [ -9.3464,   0.5708,  -1.6397],
        [ 12.5123,  -0.5339,  -1.0728],
        [  0.1931,  -4.7491,   7.4385],
        [ -2.5683, -10.4433,  -2.9182],
        [  6.0467,  -9.8369,  -6.5816],
        [ -6.4844,  -8.0230,   8.5209],
        [  1.1274,  14.0116,   8.1753],
        [ -9.4630,  -1.5145,   0.2983],
        [ -2.9497, -10.3028,  -7.3479],
        [  3.6249, -10.7255,   5.6605],
        [  9.0424,  -7.6779,  -7.5286],
        [  8.8091,  -7.9037,   0.3631],
        [ 10.1798,   8.5002,  -5.8893],
        [ -7.6719,  -6.6260,   7.1468],
        [ -6.3252,  -8.1760,  -1.0891],
        [  5.6022,  -1.9833,   8.3081],
        [  6.7565,  11.8237,  -4.9882],
        [ 12.5650,   2.0132,   4.6272],
        [ 12.1504,   4.3674,  -7.2998],
        [  5.2608,  12.6977,   4.9166],
        [ -8.6681,  -4.8556,   9.2518],
        [  6.1984,  -0.4327,   1.5438],
        [ -9.3604,   0.4855,  -3.0158],
        [  7.1441,  11.5529,   2.3232],
        [  5.1292,   4.8183,  -5.7997],
        [ -5.9225,  -8.5355,   2.4778],
        [ 11.5456,   6.0861,   8.1101],
        [  5.6258,  -1.9404,  -6.0715],
        [  3.2366,  -4.2732,   5.1532],
        [  3.3744,   6.5623,  -6.3636],
        [ -6.3489,   5.9117,   6.5161],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -3.9456,  -9.8542,   0.7466],
        [ -9.3464,   0.5708,  -1.6397],
        [  3.2990,  -4.2415,  -7.4857],
        [  2.3928,   7.1404,  -1.7035],
        [ -0.6323,   7.9085,  -5.9544],
        [ -3.4488, -10.0934,   2.5195],
        [  6.2555,   2.1474,   4.1685],
        [  2.3373,  13.7721,  -0.2442],
        [  5.9003,  12.3550,   2.1917],
        [  5.4429,  -2.2529,   9.7306],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.2247,  -0.3150,  -9.3657],
        [  1.6308,  13.9254,  -2.0984],
        [  2.1440, -10.9811,   8.5281],
        [  3.3633,  13.4794,   0.5193],
        [ -9.4196,   0.0478,   9.4140],
        [  5.1203,  -2.7151,   6.2110],
        [ 11.6019,   5.9540,  -5.1715],
        [  7.8982,  -8.6778,   0.0239],
        [ -7.3554,   4.8985,  -0.6665],
        [  1.4489,   7.5256,   4.4529],
        [ 10.0567,   8.6697,   4.6329],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.5294,  -2.1106,   9.0626],
        [ -9.3411,  -2.6004,  -2.7062],
        [  1.5246,  -4.7838,  -5.2095],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.1204,  -5.9308,   4.0761],
        [ -9.2707,   0.9661,   4.4863],
        [  0.6363, -11.0348,   6.1414],
        [  3.4920,  -4.1366,  -8.3955],
        [ -3.3999, -10.1152,  -1.2088],
        [ 10.2712,   8.3707,  -4.9743],
        [  0.9445,   7.6736,   0.5095],
        [ -4.7334,   6.9982,  -0.4392],
        [  3.4920,  -4.1366,  -8.3955],
        [ -3.1671,   7.6158,   6.0765],
        [  1.7500,   7.4187,   4.4237],
        [  8.9284,  10.0036,  -6.9210],
        [ -4.4503,   7.1378,   3.3751],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.1196,  -2.7160,   4.2705],
        [  6.1989,  -9.7591,   0.8289],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.1824,  -4.6647,   1.1985],
        [ -6.3252,  -8.1760,  -1.0891],
        [  4.8051, -10.3703,   0.5268],
        [  4.2612, -10.5514,  -6.6268],
        [  2.1379,  -4.6757,   6.4981],
        [  0.1931,  -4.7491,   7.4385],
        [  2.3490,  -4.6199,  -4.2489],
        [  2.5974,   7.0363,  -7.7810],
        [  0.9445,   7.6736,   0.5095],
        [ -8.9282,   2.1648,   8.4894],
        [  9.2811,  -7.4336,   5.1771],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.9636,   2.0648,   8.1946],
        [  5.6202,  -1.9508,   0.7471],
        [ -5.2589,   6.7024,  -5.5809],
        [  5.2939,  -2.4784,  -4.5840]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0794, 0.1307,  ..., 0.2777, 0.3771, 0.2079],
        [0.0794, 0.0000, 0.1228,  ..., 0.1987, 0.4546, 0.2076],
        [0.1307, 0.1228, 0.0000,  ..., 0.2662, 0.4680, 0.0850],
        ...,
        [0.2777, 0.1987, 0.2662,  ..., 0.0000, 0.6464, 0.3279],
        [0.3771, 0.4546, 0.4680,  ..., 0.6464, 0.0000, 0.5014],
        [0.2079, 0.2076, 0.0850,  ..., 0.3279, 0.5014, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.9120321 , 12.04924159, 12.1460931 , 12.10942412, 11.97949273,
       12.25806559,  9.05843201, 13.62896593, 13.39372426,  8.25828409,
        5.57341623,  4.85995002, 12.24467041,  9.36378438, 12.52372629,
        4.75303127, 10.75443106, 11.54672755, 10.31584102, 14.05687815,
        9.58347456, 10.7167285 , 11.32149259, 11.86239961, 11.83509372,
       13.26209545, 10.1371552 , 10.33712258,  5.94293185, 13.61801526,
       12.72524615, 12.91143491, 13.744387  ,  9.9353921 ,  6.2134961 ,
        9.37296102, 13.58333857,  7.03733775, 10.38897563, 13.05148826,
        5.9510489 ,  5.36061974,  7.37902766,  8.67502348,  9.71392179,
       10.61472636,  9.36378438,  5.37343173,  7.53063885,  7.9337647 ,
       10.66631784,  6.61386436, 13.96905756, 13.69162795,  5.89074104,
       10.32715715,  6.23262289, 14.02058621, 11.18839169, 13.89264822,
        9.41970031,  5.79562691, 13.04051744, 11.73397309,  8.83725081,
        7.66377554, 13.2778378 , 11.63724792,  5.91854397,  9.69629   ,
        5.0209005 , 11.8159254 , 10.05558893,  9.32094055, 11.05317597,
        5.41348121, 10.67132524, 13.25017211,  7.73151309,  8.4486771 ,
        5.41348121,  8.24808832,  7.62232307, 13.40850328,  8.41148722,
        7.88677538,  5.79543038, 11.56146873,  6.10774544,  6.14086582,
        5.14999799, 10.33712258, 11.42947632, 11.37939695,  5.14124612,
        4.75303127,  5.18277858,  7.50035355,  7.73151309,  9.18690018,
       11.89105259,  8.49161154,  9.19837792,  5.94909709,  8.51928531,
        5.84533345])}
{'P': tensor([[3.2210e-03, 3.4831e-01, 2.4970e-01,  ..., 2.0401e-07, 8.7093e-06,
         1.2451e-06],
        [2.2534e-05, 4.5253e-03, 4.8567e-01,  ..., 2.3024e-07, 1.4917e-05,
         2.0215e-06],
        [3.1669e-05, 4.2359e-05, 9.2275e-03,  ..., 3.7552e-07, 3.7885e-05,
         4.9003e-06],
        ...,
        [3.1254e-03, 2.5176e-03, 2.0111e-03,  ..., 9.6166e-02, 4.0429e-06,
         4.2807e-06],
        [1.2008e-05, 1.1843e-05, 1.1698e-05,  ..., 1.0233e-05, 1.4560e-01,
         1.4075e-01],
        [1.1749e-05, 1.0529e-05, 9.3747e-06,  ..., 7.0980e-05, 5.3644e-04,
         1.0681e-01]]), 'X': tensor([[  9.5018,  -7.1945,  -1.1892],
        [  9.9750,  -6.6338,  -1.2440],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 10.1983,  -6.3430,  -0.3200],
        [ 11.1085,  -4.9204,   0.0736],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.3124,   0.2057,  -0.4593],
        [  6.3587,   1.1596,  -3.6629],
        [  7.5309,  11.2617,   7.2656],
        [  4.5340,  -3.3629,   8.9652],
        [ 12.5123,  -0.5339,  -1.0728],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.2905,   1.9366,   2.1169],
        [  5.9843,  -9.8679,  -6.1449],
        [ -6.4620,  -8.0449,   6.2719],
        [ 11.9936,  -2.8677,   6.0614],
        [ -2.3317,   7.8034,  -3.6274],
        [ -7.6525,  -6.6534,   5.1721],
        [  1.7228,  -4.7575,   8.3396],
        [  6.3516,   0.6425,   0.3425],
        [  4.0587, -10.6112,   6.2273],
        [  1.9312,  13.8649,   8.3563],
        [  7.8425,  -8.7202,   6.4530],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.1066,  -0.7799,  -4.1746],
        [ -9.4203,  -2.0340,   2.5856],
        [  0.9478,  -4.8144,   0.6371],
        [  5.1861,   4.7361,  -3.7312],
        [  5.6928,  12.4710,   3.6001],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4117,   0.1160,   8.2574],
        [  8.1528,  10.7452,   9.1493],
        [  4.5057,  -3.3894,  -0.8160],
        [  8.3791,  10.5409,   5.5590],
        [  7.0515,  11.6194,   8.4289],
        [  0.2480, -11.0153,  -3.0183],
        [ -1.9600,   7.8577,   6.9121],
        [  4.9337,  12.8566,  -3.6582],
        [  1.3705,  13.9723,   7.9266],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -7.5603,   4.6459,   8.2002],
        [  6.3200,   0.2690,   1.5900],
        [  2.1120,  -4.6818,   8.6144],
        [ -9.3087,   0.7792,  -7.5587],
        [ -4.7389,   6.9953,   3.7638],
        [  6.3236,   0.3011,  -4.6668],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.3606,   0.9204,   3.2525],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.5763,   0.1333,  -9.2650],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.3087,   0.7792,  -7.5587],
        [ -4.7389,   6.9953,   3.7638],
        [ -8.0087,   4.0173,   9.1091],
        [  6.3604,   0.9042,  -6.5901],
        [  3.3943,   6.5483,   1.0790],
        [ -8.1905,   3.7253,   4.1175],
        [  5.1142,  -2.7230,  -4.1966],
        [ 12.1504,   4.3674,  -7.2998],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -6.8466,  -7.6488,   1.9648],
        [ -3.3812, -10.1235,   7.4384],
        [  3.1892, -10.8220,  -6.8409],
        [ -7.6650,   4.5092,   0.2520],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.3928,   7.1404,  -1.7035],
        [  6.3550,   0.7129,   0.4917],
        [ -6.4368,   5.8356,   3.3014],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -9.3616,  -2.4749,   0.1793],
        [  3.9895,   6.0816,  -6.5235],
        [ -8.4267,  -5.3697,   2.4172],
        [  4.8674,  12.8875,   6.7622],
        [  5.2199,   4.6861,  -1.9304],
        [ -3.5520,   7.4976,   9.1522],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.3200,   7.1756,   1.1938],
        [ -8.5900,   2.9767,   7.4811],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.4624,  -0.4633,  -7.3918],
        [  7.8425,  -8.7202,   6.4530],
        [  6.2555,   2.1474,   4.1685],
        [  2.9461,  -4.4077,  -0.4831],
        [  0.7958,  -4.8109,  -8.2248],
        [  0.6757, -11.0361,  -1.3047],
        [  4.4049,   5.6892,  -4.4394],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.7218,  -9.9934,   6.5764],
        [  9.5203,   9.3483,   5.8781],
        [ -5.1398,  -9.1367,   6.2074],
        [  9.0652,   9.8597,  -3.5949],
        [ -8.9282,   2.1648,   8.4894],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -9.0574,   1.7801,   8.8986],
        [  4.5325,  -3.3643,   6.7699],
        [  2.3373,  13.7721,  -0.2442],
        [  6.0031,  12.2958,  -6.5992],
        [ 12.4996,   2.6335,  -6.3166],
        [ -2.0229, -10.6163,   8.6930],
        [  0.1497,   7.8334,  -5.4412],
        [ -4.9217,   6.8979,  -1.7667]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0337, 0.0738,  ..., 0.9157, 0.2285, 0.2328],
        [0.0337, 0.0000, 0.0406,  ..., 0.9401, 0.2299, 0.2656],
        [0.0738, 0.0406, 0.0000,  ..., 0.9636, 0.2465, 0.3062],
        ...,
        [0.9157, 0.9401, 0.9636,  ..., 0.0000, 1.0917, 0.8186],
        [0.2285, 0.2299, 0.2465,  ..., 1.0917, 0.0000, 0.2888],
        [0.2328, 0.2656, 0.3062,  ..., 0.8186, 0.2888, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.9182956 , 11.97949273, 12.04924159, 12.1460931 , 12.00995898,
       12.1494023 , 10.52357479,  6.31576264,  6.46356806, 13.54774355,
        5.64500512, 12.52372629, 11.97957425,  9.94285511,  6.58184474,
       11.54071859, 10.31886066, 12.33167382,  8.14433542, 10.14044489,
        5.05981914,  6.38399789, 11.36089355, 13.99877413, 11.72802491,
        5.83560009,  6.1561646 ,  9.63742669,  4.90676242,  7.02326515,
       13.70893492,  7.5092834 ,  9.41244865, 13.48809595,  5.63824971,
       13.46553623, 13.59170809, 11.018084  ,  8.09842704, 13.77075174,
       14.0393949 , 10.46739073,  8.8737556 ,  6.32572323,  5.13616224,
        9.34127013,  8.44940661,  6.33075682,  5.01114719,  6.02789634,
        6.42689705,  5.91677188, 12.5769659 ,  6.02588517,  9.34127013,
        8.44940661,  8.95983586,  6.42439344,  7.37576154,  8.99791316,
        5.79391825, 12.91143491, 10.53368644, 10.26545792, 10.67322527,
       11.28215429,  8.89301516,  9.41605684,  7.53063885,  6.3949043 ,
        8.68832398, 10.1979669 ,  9.68323686,  7.27340834,  9.99211663,
       13.77604293,  7.01475376,  8.29640613, 12.25806559,  7.54127425,
        9.09119351,  5.47702766, 12.62576026,  9.47369628, 11.72802491,
        6.61386436,  5.30158497,  4.8763135 , 11.05672136,  7.19513806,
       10.53368644, 11.51557166, 13.3426541 , 10.48315436, 13.39372426,
        9.18690018,  9.15976746,  9.75075273,  9.23071698,  5.64464807,
       13.96905756, 13.6829843 , 12.77402351, 10.80728788,  7.83486943,
        8.47371868])}
{'P': tensor([[4.0624e-03, 3.1869e-01, 1.4912e-01,  ..., 8.2348e-06, 1.7922e-07,
         1.2314e-05],
        [1.5876e-05, 4.5580e-03, 3.3869e-01,  ..., 3.8578e-06, 2.2426e-07,
         9.1202e-06],
        [2.2854e-05, 4.0992e-05, 1.3911e-02,  ..., 5.0571e-06, 8.2587e-07,
         1.7987e-05],
        ...,
        [1.7600e-02, 1.6362e-02, 1.2077e-02,  ..., 1.2920e-02, 7.6176e-07,
         3.9553e-04],
        [3.9524e-05, 2.6339e-05, 1.7279e-05,  ..., 2.9175e-04, 1.6870e-01,
         2.9065e-03],
        [3.2602e-03, 3.1071e-03, 2.8086e-03,  ..., 8.4721e-03, 1.0662e-04,
         8.0546e-02]]), 'X': tensor([[  9.5261,  -7.1675,   1.4206],
        [ 10.4758,  -5.9539,   1.3667],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.2767,  -4.6006,   0.7868],
        [ 10.0995,  -6.4739,   2.0974],
        [ 11.3367,  -4.4806,   0.4643],
        [  8.5716,  10.3596,   6.8818],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -7.4807,   4.7463,   2.0616],
        [ -9.3616,  -2.4750,   3.7765],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.1575,  12.2046,   3.3257],
        [ -3.1204, -10.2345,   9.2288],
        [  3.2159, -10.8166,  -1.0406],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.9394,   7.3069,  -8.9365],
        [  5.1886,  -2.6250,   6.4677],
        [  2.7295,  13.6703,  -8.9328],
        [  3.3943,   6.5483,   1.0790],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 12.2914,   3.8037,   6.0978],
        [  1.6422,   7.4586,  -6.8056],
        [  7.9141,  -8.6656,   9.6948],
        [  6.2895,   0.0405,   5.7652],
        [  5.9026,   3.3837,  -0.6573],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.8450,  -9.9356,   3.0767],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -7.8126,   4.3066,   5.1791],
        [  6.9693,  -9.3200,  -8.3822],
        [ 12.2692,  -1.8882,   3.9253],
        [  7.5784,  -8.9141,   5.8410],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.4501,  -1.7133,   2.2980],
        [  0.1252, -11.0062,   7.2564],
        [  8.1217,  -8.5024,  -0.6149],
        [ -6.0195,   6.1789,  -2.9496],
        [  4.5491,  -3.3485,   6.8412],
        [ -5.1377,   6.7751,  -0.4438],
        [ -8.5506,  -5.1154,   6.3423],
        [  7.4132,  11.3526,   0.3322],
        [ -2.6735, -10.4062,  -1.5911],
        [ -2.8162,   7.7059,  -9.6885],
        [ -8.3653,   3.4184,   9.3454],
        [  6.3497,   1.3680,  -1.9076],
        [  2.6970,   6.9826,   9.7350],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.3305,   1.6119,   1.1538],
        [  6.3236,   0.3011,  -4.6668],
        [ -0.0410,   7.8591,   9.3457],
        [  5.6093,  -1.9705,   6.4807],
        [  5.4179,  -2.2925,  -3.4641],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.3516,   0.6425,   0.3425],
        [  5.9193,  -1.3176,   5.1739],
        [ -6.6895,   5.6051,   2.7899],
        [  5.8617,   3.4853,  -9.8080],
        [  4.5491,  -3.3485,   6.8412],
        [ -0.8885,   7.9160,   0.0571],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.3261,  13.7749,   5.9121],
        [ -8.5900,   2.9767,   7.4811],
        [  0.4996,   7.7738,  -4.4711],
        [  4.3459,  -3.5327,   0.3667],
        [  2.6451,  -4.5250,  -0.5565],
        [  3.9022,  13.2906,   5.5788],
        [ -3.5173, -10.0623,   4.8280],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.9606,  -4.0994,   1.7886],
        [ -7.3446,   4.9113,  -5.9356],
        [ -9.4753,  -0.8012,  -5.2050],
        [  5.7485,  -9.9811,  -3.2355],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.0547,   7.2957,  -1.9343],
        [  0.2435,   7.8191,  -6.4709],
        [  0.3354,  -4.7708,  -4.9125],
        [  0.6933,   7.7336,   2.1179],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.3303,   1.6145,  -8.4083],
        [  6.2022,   2.4092,   3.0349],
        [ -8.6603,   2.8250,   5.9181],
        [ 11.6544,   5.8273,  -3.1375],
        [ 11.1420,   6.9331,  -2.4779],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.5424,  -2.0885,   7.2487],
        [  9.4554,   9.4245,   9.8800],
        [  0.5250, -11.0306,   5.5358],
        [  1.0093,   7.6567,   2.3526],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.5424,  -2.0885,   7.2487],
        [  2.1379,  -4.6757,   6.4981],
        [  2.1440, -10.9811,   8.5281],
        [  6.1579,  -0.5961,  -6.5595],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.9659,  -4.8144,  -0.9594],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.3542,   0.6934,  -3.3840],
        [  2.2182,  13.8007,   5.1889],
        [  4.5325,  -3.3643,   6.7699],
        [  5.7218,  -9.9934,   6.5764],
        [ -8.9282,   2.1648,   8.4894],
        [  1.7228,  -4.7575,   8.3396]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0517, 0.1219,  ..., 0.5418, 0.7468, 0.4244],
        [0.0517, 0.0000, 0.0715,  ..., 0.5763, 0.7930, 0.4207],
        [0.1219, 0.0715, 0.0000,  ..., 0.6123, 0.8474, 0.4055],
        ...,
        [0.5418, 0.5763, 0.6123,  ..., 0.0000, 0.3242, 0.4031],
        [0.7468, 0.7930, 0.8474,  ..., 0.3242, 0.0000, 0.7232],
        [0.4244, 0.4207, 0.4055,  ..., 0.4031, 0.7232, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.92133129, 12.0495521 , 12.19805394, 12.17901033, 11.99633819,
       12.18998195, 13.44593519, 12.49317563,  8.85939626,  9.68324553,
       13.02818702,  9.50913849, 13.66991027, 10.69963954, 11.2845608 ,
       13.32374056, 13.15525803,  5.81481044, 13.94008948,  7.37576154,
       12.1460931 , 12.86648771,  7.63726134, 11.73567759,  6.28961988,
        6.80370254, 11.76830981,  9.97822473, 11.52733581,  8.19703512,
        9.86089579,  8.92098752, 11.63765273, 12.41366754, 11.70016217,
        8.56889738,  9.60412735, 11.00694688, 11.75807252,  8.62631801,
        5.64865173,  8.50278896,  9.96391045, 13.55868684, 10.74409834,
        8.20436764,  9.0368458 ,  6.49538841,  7.48538905, 11.54672755,
        6.53250937,  6.33075682,  7.85919687,  5.94535529,  5.8828986 ,
        5.01114719,  6.38399789,  6.06416957,  8.7273569 ,  6.81961996,
        5.64865173,  7.96575782, 12.3489324 , 13.96987806,  9.09119351,
        7.78980299,  5.60064319,  5.24137909, 13.85158417, 10.65929968,
       12.42554808,  9.85384623,  8.8353674 ,  9.50913849, 11.51811603,
        9.58347456,  7.57946459,  7.822854  ,  4.78258092,  7.76457095,
       12.24467041,  6.53290663,  6.65368551,  9.10945678, 13.03002338,
       13.12299145, 12.53285633,  5.92281142, 13.35013374, 11.04313641,
        7.72291946, 13.04217487,  5.92281142,  5.14124612, 11.18839169,
        6.18668463, 13.06498554,  4.91038884,  9.75200848,  9.7751189 ,
        6.39187285, 13.97779839,  5.64464807, 11.51557166,  9.18690018,
        5.05981914])}
{'P': tensor([[6.6423e-03, 2.3644e-01, 1.3593e-01,  ..., 4.5235e-03, 5.0272e-05,
         3.6716e-04],
        [1.2302e-05, 6.9332e-03, 1.6191e-02,  ..., 2.1829e-02, 1.1959e-04,
         7.0135e-05],
        [1.2052e-05, 1.1175e-03, 5.9555e-03,  ..., 2.6556e-02, 9.2727e-05,
         3.6836e-05],
        ...,
        [7.7308e-07, 2.1149e-06, 2.2084e-06,  ..., 8.8089e-03, 5.6793e-04,
         8.2675e-07],
        [1.2546e-05, 1.4070e-05, 1.3277e-05,  ..., 3.5599e-05, 1.2061e-01,
         1.1780e-05],
        [1.2731e-04, 5.9053e-04, 4.2932e-04,  ..., 7.7744e-04, 3.7435e-04,
         3.3902e-02]]), 'X': tensor([[ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1288, 0.1678,  ..., 0.4706, 0.4224, 0.9273],
        [0.1288, 0.0000, 0.0702,  ..., 0.3420, 0.4876, 0.9695],
        [0.1678, 0.0702, 0.0000,  ..., 0.3179, 0.5555, 1.0397],
        ...,
        [0.4706, 0.3420, 0.3179,  ..., 0.0000, 0.7584, 1.1651],
        [0.4224, 0.4876, 0.5555,  ..., 0.7584, 0.0000, 0.5120],
        [0.9273, 0.9695, 1.0397,  ..., 1.1651, 0.5120, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.95743558, 12.22503754, 12.24467041, 12.27991991, 12.33167382,
       12.17451685, 10.48315436,  5.91677188,  8.62260327, 11.38774149,
        5.26342203,  9.45995955, 12.49291768,  7.18354885, 10.4606265 ,
       12.33167382,  5.98285319, 10.31584102,  5.92281142, 13.12218013,
        5.5513072 ,  8.31625171, 11.62158554,  7.62509582, 12.16901615,
        6.39520397,  7.96181288,  9.69629   ,  4.90505566,  9.15253344,
       12.0495521 ,  5.84450127, 10.31886066,  6.18668463,  9.15253344,
        9.19837792, 12.50996828,  6.67218687,  5.0209005 ,  6.74872981,
       11.67875855,  7.34250948, 10.60226614,  6.28961988, 11.73848065,
        9.27474386,  5.79391825, 12.97370371,  7.48538905,  7.5984809 ,
       13.05148826,  5.14607014,  7.66377554, 12.84056604,  9.49511524,
        9.93553424,  7.66377554,  8.8737556 ,  4.78002665,  9.69792513,
       11.04313641, 13.3614598 , 12.13568598,  6.67218687,  9.36256096,
       13.09737053,  8.8737556 ,  4.74770899,  8.11227668, 10.35485331,
       13.88683436, 11.63724792,  5.38342077,  7.38844703,  6.27715105,
       13.52430176, 13.96905756,  7.57685908,  9.79814997, 13.32613445,
        6.27715105, 11.83570132, 11.37939695, 13.76047999,  8.47371868,
        6.28835101,  4.90676242, 11.75618513, 13.03002338, 13.32374056,
        9.41605684, 11.9702762 , 11.74901778,  9.68324553,  6.42689705,
       12.70489563,  8.12043983,  8.68979052,  8.16584537,  5.81199422,
        9.62238828,  7.72668467,  8.22728399, 12.88270018,  7.60463349,
       12.18045978])}
{'P': tensor([[5.7664e-03, 2.3839e-01, 1.8433e-01,  ..., 6.7350e-06, 7.4720e-06,
         2.5620e-05],
        [1.4920e-05, 8.3003e-03, 1.9383e-02,  ..., 5.2583e-06, 3.6223e-05,
         1.7218e-05],
        [9.7960e-06, 1.0356e-03, 5.5187e-03,  ..., 2.7717e-06, 2.0061e-05,
         8.4631e-06],
        ...,
        [2.5716e-03, 2.0816e-03, 1.4676e-03,  ..., 1.3224e-01, 1.2610e-04,
         2.7971e-01],
        [2.0765e-08, 2.9108e-08, 2.6616e-08,  ..., 1.2650e-06, 5.8095e-03,
         8.5947e-07],
        [2.5679e-03, 3.4973e-03, 2.4308e-03,  ..., 6.7945e-03, 6.4902e-04,
         1.7597e-01]]), 'X': tensor([[  9.9058,  -6.7203,   5.6928],
        [ 11.5198,  -4.0923,   5.5088],
        [ 11.6165,  -3.8715,   6.4594],
        [ 10.8218,  -5.4177,   6.8573],
        [ 11.9936,  -2.8677,   6.0614],
        [ 11.8951,  -3.1570,   7.0742],
        [ -7.6525,  -6.6534,   5.1721],
        [  4.1348,   5.9514,  -0.8856],
        [ -4.9217,   6.8979,  -1.7667],
        [ -9.1178,   1.5784,   2.1410],
        [ -4.4520,   7.1370,  -8.7867],
        [ -8.0965,  -5.9712,   4.3466],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.1666,   3.7653,   3.6075],
        [  0.5960, -11.0335,   3.1144],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.4714,  -1.3301,  -3.1998],
        [  8.1528,  10.7452,   9.1493],
        [  2.9410,   6.8428,   5.4458],
        [  7.0358,  -9.2784,  -9.8479],
        [ 10.3251,   8.2928,  -3.0006],
        [  5.8889,  -1.3916,   8.8993],
        [  3.4401,  13.4540,   4.6576],
        [ -9.4296,  -1.9458,   4.0833],
        [ -3.6381,   7.4683,   7.3544],
        [ -3.2486,   7.5925,  -1.3719],
        [  0.7958,  -4.8109,  -8.2248],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.4297,  -5.3637,  -9.3962],
        [  4.1537,   5.9340,  -6.0423],
        [  1.1644, -11.0394,   7.8511],
        [  0.6396,  14.0772,  -6.3786],
        [  3.3744,   6.5623,  -6.3636],
        [  6.3087,   1.8053,  -1.9481],
        [ -0.0410,   7.8591,   9.3457],
        [ -3.0008, -10.2827,  -8.9480],
        [ 12.5267,  -0.4125,  -8.7331],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.3616,  -2.4749,   0.1793],
        [ -7.0825,   5.2073,  -5.0497],
        [ -5.1398,  -9.1367,   6.2074],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.3076,   0.7852,  -6.7010],
        [ -5.1398,  -9.1367,   6.2074],
        [ -6.2200,  -8.2737,   5.5079],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 11.3691,   6.4760,   1.0787],
        [  7.0515,  11.6194,   8.4289],
        [  7.9403,  -8.6455,  -8.7433],
        [  2.1625,  -4.6697,   7.0658],
        [ -6.3066,  -8.1936,   9.5822],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -5.3736,   6.6310,  -1.5174],
        [  5.2188,  -2.5839,   2.1833],
        [  3.9770, -10.6341,   8.0783],
        [  2.4932,   7.0903,  -5.4214],
        [  6.2675,  12.1379,  -8.9720],
        [  4.7544,  -3.1425,  -1.5315],
        [  1.1274,  14.0116,   8.1753],
        [  5.8889,  -1.3916,   8.8993],
        [  4.8653,   5.1694,  -4.0823],
        [ 12.5224,  -0.4497,   4.8448],
        [  5.1472,  -2.6801,   8.1194],
        [ -3.9858,  -9.8334,  -3.7323],
        [  1.6422,   7.4586,  -6.8056],
        [  5.9526,   3.2525,  -4.3821],
        [ -4.4520,   7.1370,  -8.7867],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.5469,  13.7192,  -9.6935],
        [  9.3607,   9.5337,  -4.5412],
        [ -5.0943,  -9.1682,   0.5073],
        [  3.2809,  13.5061,  -4.0199],
        [ 11.6801,  -3.7196,  -0.9181],
        [  6.1755,   2.5233,  -2.0042],
        [ -5.6651,   6.4380,   1.3660],
        [ -1.6275, -10.7221,   3.2376],
        [  6.3549,   0.7096,   6.9198],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.6231,   5.4556,   8.4973],
        [  3.3167,   6.6022,   6.3554],
        [ -3.2486,   7.5925,  -1.3719],
        [  1.7500,   7.4187,   4.4237],
        [ -9.4766,  -1.1116,  -2.1706],
        [  1.8320,   7.3871,   5.1452],
        [ 11.5083,  -4.1177,   0.2389],
        [  0.9392,  -4.8143,  -3.4649],
        [ -9.1813,  -3.3550,   1.8827],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.3715,  -4.7756,   0.2312],
        [  6.2242,  -0.3171,   7.4978],
        [  8.0877,  -8.5297,  -1.1050],
        [  6.1929,   2.4501,  -6.9352],
        [  1.7815,  -4.7481,   4.4289],
        [  4.9152,  12.8653,  -1.4459],
        [  6.2975,   0.0947,   5.1791],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.5720,   4.6310,   2.1717],
        [ -8.6050,   2.9451,  -0.7012],
        [  9.3836,   9.5075,  -5.9278],
        [ 12.2914,   3.8037,   6.0978],
        [ -5.0943,  -9.1682,   0.5073],
        [ -4.0409,   7.3169,  -3.5779],
        [  0.7685,  -4.8098,  -2.9310],
        [  4.5954,  13.0099,  -4.4439],
        [  3.5965,  -4.0754,  -3.5035]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1242, 0.1351,  ..., 0.7622, 1.0507, 0.8088],
        [0.1242, 0.0000, 0.0702,  ..., 0.7533, 0.9975, 0.7896],
        [0.1351, 0.0702, 0.0000,  ..., 0.8230, 1.0654, 0.8597],
        ...,
        [0.7622, 0.7533, 0.8230,  ..., 0.0000, 0.4207, 0.0771],
        [1.0507, 0.9975, 1.0654,  ..., 0.4207, 0.0000, 0.3436],
        [0.8088, 0.7896, 0.8597,  ..., 0.0771, 0.3436, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.9702762 , 12.22503754, 12.24467041, 12.10222204, 12.33167382,
       12.30695098, 10.14044489,  7.24677664,  8.47371868,  9.25336371,
        8.41171369, 10.06023665,  8.29022197,  8.41148722,  8.99276564,
       11.04953767,  9.55720651,  9.5642993 , 13.48809595,  7.44805901,
       11.64436684, 13.2430377 ,  6.05113217, 13.88683436,  9.6282763 ,
        8.30729352,  8.25828409,  4.8763135 , 10.92597196,  9.99145096,
        7.24326992, 11.10066126, 14.09176259,  7.37902766,  6.5618968 ,
        7.85919687, 10.71162681, 12.53345445,  6.19707944,  9.68323686,
        8.7907923 , 10.48315436,  7.53884104,  9.34061938, 10.48315436,
       10.35093485, 12.12448912, 13.08414689, 13.59170809, 11.73848065,
        5.14607014, 10.33958639, 10.61047596,  8.53501777,  5.82345867,
       11.35345345,  7.5158489 , 13.66051884,  5.69913817, 14.05687815,
        6.05113217,  7.09887557, 12.53047489,  5.80312949, 10.61047596,
        7.63726134,  6.78325151,  8.41171369, 13.04217487, 13.95361098,
       13.36092624, 10.48840476, 13.89886469, 12.25806559,  6.67108662,
        8.57560275, 10.84493732,  6.39438044, 13.00413829,  7.15100303,
        7.38844703,  8.25828409,  7.62232307,  9.54154096,  7.61088125,
       12.22276278,  4.90505566,  9.7751189 , 12.30695098,  4.79003361,
        6.2322769 , 11.75438006,  6.65991877,  5.0713364 , 13.77222752,
        6.29821673, 10.0674887 ,  8.87587457,  9.09502385, 13.35832725,
       12.86648771, 10.48840476,  8.35856487,  4.87083348, 13.79762415,
        5.43544626])}
{'P': tensor([[4.5291e-03, 2.4602e-01, 1.6578e-01,  ..., 8.5131e-04, 9.8110e-05,
         6.4199e-07],
        [3.7625e-05, 5.1946e-03, 3.5689e-01,  ..., 1.4048e-03, 1.0576e-04,
         1.0304e-06],
        [2.6878e-05, 3.5821e-05, 9.3841e-03,  ..., 4.5882e-03, 1.4081e-04,
         2.8609e-06],
        ...,
        [9.2048e-07, 1.1384e-06, 1.6759e-06,  ..., 2.6240e-02, 6.2399e-06,
         2.0382e-05],
        [9.8781e-04, 1.4699e-03, 2.3860e-03,  ..., 8.1582e-04, 9.9691e-02,
         6.7860e-04],
        [1.5737e-06, 1.6741e-06, 1.7321e-06,  ..., 1.5286e-06, 8.9013e-05,
         1.0287e-01]]), 'X': tensor([[  9.9217,  -6.7006,  -5.6728],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 10.9717,  -5.1646,  -6.9039],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 12.2692,  -1.8882,   3.9253],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.8940, -11.0402,  -0.9783],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.6363, -11.0348,   6.1414],
        [  0.1677,  -4.7447,  -6.1529],
        [  0.4119,  14.1019,  -4.1870],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.7121,   6.9743,  -8.4202],
        [  0.8940, -11.0402,  -0.9783],
        [  4.0421,  -3.7765,  -8.3631],
        [  1.7000,  -4.7609,   2.7935],
        [ 11.7794,   5.5086,  -9.6929],
        [  6.3606,   0.9204,   3.2525],
        [  2.9461,  -4.4077,  -0.4831],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -1.6086,   7.8928,   4.3385],
        [  1.9503,  -4.7172,  -6.2313],
        [  5.9587,  12.3216,   9.1695],
        [ -2.1817,   7.8274,  -4.6298],
        [  5.8902,  -1.3887,   4.0388],
        [  5.8979,   3.3958,  -3.8713],
        [ -0.5385,   7.9037,   6.3830],
        [  7.1821,  -9.1845,   9.8548],
        [  5.7784,  -1.6401,   9.5587],
        [ 10.1503,   8.5414,  -7.2681],
        [  5.8617,   3.4853,  -9.8080],
        [  2.8602,   6.8905,   5.8779],
        [  1.8320,   7.3871,   5.1452],
        [ 12.0582,  -2.6640,   0.7132],
        [ -3.5173,   7.5091,  -8.9585],
        [  6.9433,  -9.3362,  -0.3421],
        [  1.3705,  13.9723,   7.9266],
        [ 10.6283,  -5.7253,   7.4878],
        [ -3.2053, -10.1993,  -3.8736],
        [ 12.5199,  -0.4711,   3.0436],
        [ -9.1459,   1.4780,  -4.2403],
        [ -8.8835,  -4.3178,   9.3080],
        [  9.0004,  -7.7195,   4.6945],
        [  1.9503,  -4.7172,  -6.2313],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.5919,   0.4035,  -5.8460],
        [ 11.0275,  -5.0666,   8.4154],
        [ -8.9282,   2.1648,   8.4894],
        [  2.6970,   6.9826,   9.7350],
        [ 12.5224,  -0.4497,   4.8448],
        [ -1.6220,   7.8917,   0.0315],
        [  5.5424,  -2.0885,   7.2487],
        [ -3.3812, -10.1235,   7.4384],
        [  8.2154,  -8.4262,   8.0171],
        [  9.0051,   9.9234,   5.4411],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.3305,   1.6119,   1.1538],
        [ -5.6651,   6.4380,   1.3660],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.1625,  -4.6697,   7.0658],
        [ 12.0772,  -2.6014,  -7.1832],
        [  4.5954,  13.0099,  -4.4439],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -7.0120,   5.2825,   7.7762],
        [  2.3928,   7.1404,  -1.7035],
        [  2.4356,  -4.5942,  -5.0808],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.2939,   6.6325,  -5.8311],
        [  5.7845,   3.6652,   0.9175],
        [  6.2576,   2.1359,  -3.3398],
        [ -8.1204,  -5.9308,   4.0761],
        [ -1.3940, -10.7772,   6.7835],
        [  7.1441,  11.5529,   2.3232],
        [ -7.9925,  -6.1423,   2.8990],
        [  5.9587,  12.3216,   9.1695],
        [  1.4150,  13.9647,  -4.0915],
        [ -6.0175,  -8.4541,  -9.8408],
        [  9.6614,   9.1784,  -1.2087],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -6.4464,   5.8272,   7.5038],
        [  5.4336,  -2.2676,   5.6933],
        [ -6.8466,  -7.6488,   1.9648],
        [ -9.4753,  -0.8012,  -5.2050],
        [  1.3767, -11.0341,   9.9415],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.1271, -10.8343,   2.6442],
        [ 10.8218,  -5.4177,   6.8573],
        [  3.3839,  -4.1966,   8.3574],
        [  9.6587,   9.1817,  -0.8212],
        [  0.7275,  14.0667,  -0.9815],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.3739,   0.3982,  -4.2564],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 11.5935,   5.9740,  -4.3952],
        [  0.9809,   7.6642,  -1.4217],
        [ -7.4733,   4.7555,   6.2490],
        [  4.3377,   5.7570,   1.3712],
        [ -9.4378,  -1.8606,  -2.5668],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.2188,  -2.5839,   2.1833],
        [ -2.2188,   7.8218,   8.6064]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0489, 0.1023,  ..., 0.4232, 0.6129, 1.0440],
        [0.0489, 0.0000, 0.0534,  ..., 0.3920, 0.5834, 1.0305],
        [0.1023, 0.0534, 0.0000,  ..., 0.3613, 0.5556, 1.0197],
        ...,
        [0.4232, 0.3920, 0.3613,  ..., 0.0000, 0.7965, 1.3104],
        [0.6129, 0.5834, 0.5556,  ..., 0.7965, 0.0000, 0.5253],
        [1.0440, 1.0305, 1.0197,  ..., 1.3104, 0.5253, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.97238624, 12.05395056, 12.18045978, 12.12641969, 12.27835996,
       12.32545637, 12.41366754, 12.30695098, 11.07637531, 13.38091121,
        9.55720651,  9.4997987 , 11.05317597,  4.74770899, 14.10796466,
       10.8207297 ,  7.48311002, 11.07637531,  5.53175275,  5.05534609,
       13.00380624,  6.42689705,  5.30158497, 10.57302546,  8.0550286 ,
        5.10444097, 13.68672855,  8.12580976,  6.05165248,  6.80559669,
        7.92200342, 11.65923396,  6.00663314, 13.26590071,  6.81961996,
        7.46053281,  7.61088125, 12.3489324 ,  8.29203356, 11.63502679,
       14.0393949 , 12.07225469, 10.6910768 , 12.52875793,  9.26455795,
        9.87720102, 11.85743425,  5.10444097, 12.38881635, 12.59840193,
       12.13568598,  9.18690018,  7.48538905, 12.53047489,  8.05668609,
        5.92281142, 10.67322527, 11.76830981, 13.40024224,  8.36995953,
        6.53250937,  8.57560275,  7.98469166,  5.14607014, 12.35421461,
       13.79762415,  5.81199422, 12.42554808,  8.77913428,  7.53063885,
        5.19986797,  4.85603886, 13.09737053,  6.84796202,  6.61211224,
       10.05558893, 10.86693708, 13.58333857, 10.08001894, 13.68672855,
       14.03618649, 10.37698435, 13.32613445, 12.00995898,  8.68979052,
        5.88782913, 10.26545792,  9.50913849, 11.11969815,  9.14044984,
       11.27656488, 12.10222204,  5.39098669, 13.32645552, 14.08549802,
        9.66977802,  9.38232121, 12.3828957 , 13.04217487,  7.72668467,
        8.8580671 ,  7.20826219,  9.61942445, 13.1399181 ,  5.82345867,
        8.1303945 ])}
{'P': tensor([[4.1820e-03, 4.4883e-01, 1.5017e-01,  ..., 5.1387e-04, 1.4770e-07,
         1.3765e-07],
        [3.6225e-05, 7.8912e-03, 4.1311e-01,  ..., 8.7640e-04, 2.8730e-07,
         2.3766e-07],
        [2.1447e-04, 2.9660e-04, 1.3367e-01,  ..., 8.2799e-03, 5.8454e-06,
         3.2851e-06],
        ...,
        [5.7942e-04, 7.5066e-04, 1.6284e-03,  ..., 1.2763e-01, 3.2431e-05,
         6.2371e-05],
        [9.0795e-06, 7.1073e-06, 3.6418e-06,  ..., 6.8994e-06, 2.0196e-02,
         1.3567e-03],
        [5.7480e-04, 4.1374e-04, 1.5553e-04,  ..., 4.5705e-04, 8.0582e-05,
         8.4794e-02]]), 'X': tensor([[  9.9750,  -6.6338,  -1.2440],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.2939,   6.6325,  -5.8311],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3569,   0.7591,   6.5011],
        [ -7.3554,   4.8985,  -0.6665],
        [ -0.5385,   7.9037,   6.3830],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -0.0410,   7.8591,   9.3457],
        [  5.1196,  -2.7160,   4.2705],
        [  9.3560,   9.5391,   9.4943],
        [ -7.3554,   4.8985,  -0.6665],
        [ -3.9508,  -9.8515,   1.7923],
        [  1.5662,  13.9375,   4.5142],
        [  2.4356,  -4.5942,  -5.0808],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -8.1426,   3.8048,   9.3542],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -1.6086,   7.8928,   4.3385],
        [  0.3493,  -4.7727,  -6.5682],
        [  4.5325,  -3.3643,   6.7699],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.2804,   2.0019,   0.8607],
        [  1.4749,  -4.7892,  -0.8089],
        [  4.5814, -10.4485,   2.5496],
        [ -8.9636,   2.0648,   8.1946],
        [ -8.9942,  -3.9990,   6.0448],
        [  8.1043,  -8.5164,  -7.9290],
        [ -6.1914,  -8.2997,   8.5380],
        [  6.3505,   0.6235,  -8.5849],
        [ -3.2053, -10.1993,  -3.8736],
        [  0.1206,  -4.7363,   8.7103],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.3497,   1.3680,  -1.9076],
        [ -0.4341, -10.9473,   7.4509],
        [ -9.3616,  -2.4750,   3.7765],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.0348,  -2.8229,   3.5436],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.3609,   3.4769,  -8.5762],
        [  1.7500,   7.4187,   4.4237],
        [  9.0004,  -7.7195,   4.6945],
        [  3.3094,  13.4969,  -3.6507],
        [ -8.9379,  -4.1655,  -2.8114],
        [  0.4996,   7.7738,  -4.4711],
        [  2.1440, -10.9811,   8.5281],
        [ -3.2736, -10.1703,   3.8712],
        [  1.9204,   7.3518,   8.4781],
        [  4.0587, -10.6112,   6.2273],
        [  7.9403,  -8.6455,  -8.7433],
        [  1.7659,   7.4127,   4.7871],
        [  6.0047,   3.1062,  -6.7914],
        [  2.0243,   7.3086,   7.5432],
        [ -3.5520,   7.4976,   9.1522],
        [  3.8182,  -3.9347,  -1.7961],
        [ -8.0608,   3.9362,  -5.9567],
        [  1.8320,   7.3871,   5.1452],
        [  1.4150,  13.9647,  -4.0915],
        [  2.1771, -10.9776,  -3.2007],
        [ 11.5198,  -4.0923,   5.5088],
        [  5.0277,   4.9589,   7.0175],
        [ 10.1798,   8.5002,  -5.8893],
        [ -8.6603,   2.8250,   5.9181],
        [  0.9809,  -4.8145,   6.8706],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.3038,   7.8092,   0.4164],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.2974,   0.8371,   2.4852],
        [ -6.4368,   5.8356,   3.3014],
        [ -6.0844,  -8.3955,   0.7705],
        [  8.1217,  -8.5024,  -0.6149],
        [ -7.8359,   4.2735,   7.2943],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.9956,  11.6590,  -7.6876],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.6681,  -4.8556,   9.2518],
        [ -5.2589,   6.7024,  -5.5809],
        [ -8.9033,   2.2329,  -8.8912],
        [  2.3373,  -4.6232,  -5.2246],
        [  4.1537,   5.9340,  -6.0423],
        [ -9.4671,  -0.5553,   0.1117],
        [  3.6154,  -4.0640,  -7.9698],
        [  7.5784,  -8.9141,   5.8410],
        [  7.3592,  11.3937,   6.2407],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.2226,  -4.2802,  -4.0428],
        [ -0.9715, -10.8630,   0.3555],
        [  4.2612, -10.5514,  -6.6268],
        [  4.2275,  -3.6320,  -0.2714],
        [ 12.3609,   3.4769,  -8.5762],
        [ -3.2486,   7.5925,  -1.3719],
        [  2.2182,  13.8007,   5.1889],
        [ -8.3539,  -5.5111,   0.3373],
        [ -3.5173,   7.5091,  -8.9585],
        [  1.5383, -11.0274,  -4.9666],
        [ 11.1469,   6.9237,   1.6160],
        [ 10.5678,   7.9261,  -0.6100],
        [  6.1579,  -0.5961,  -6.5595],
        [ -9.4430,  -0.1897,   4.8383],
        [ -7.8849,  -6.3115,  -4.7479]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.0597e-02, 1.3671e-01,  ..., 3.0747e-01, 8.2166e-01,
         8.5711e-01],
        [4.0597e-02, 0.0000e+00, 9.6138e-02,  ..., 2.8726e-01, 8.5430e-01,
         8.9652e-01],
        [1.3671e-01, 9.6138e-02, 0.0000e+00,  ..., 2.6314e-01, 9.3193e-01,
         9.9098e-01],
        ...,
        [3.0747e-01, 2.8726e-01, 2.6314e-01,  ..., 0.0000e+00, 1.1162e+00,
         1.0002e+00],
        [8.2166e-01, 8.5430e-01, 9.3193e-01,  ..., 1.1162e+00, 3.0339e-04,
         7.3098e-01],
        [8.5711e-01, 8.9652e-01, 9.9098e-01,  ..., 1.0002e+00, 7.3098e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.97949273, 12.04924159, 12.25806559, 12.1460931 , 12.1494023 ,
       12.22276278, 13.09737053,  7.60529892,  6.4020404 ,  8.83725081,
        7.92200342,  9.99145096,  7.85919687,  5.79543038, 13.3614598 ,
        8.83725081, 10.61417181, 14.02526505,  5.19986797, 12.30858677,
        8.98765644,  9.0368458 ,  9.78147542,  8.0550286 ,  4.78544875,
        5.64464807,  9.53015376,  6.59176018,  5.01114719, 11.4088043 ,
        9.19837792,  9.84314712, 11.75618513, 10.35464916,  6.38105015,
       10.6910768 ,  4.73784222,  5.26067019,  6.49538841, 10.95593729,
        9.68324553, 10.78533838,  5.77219077,  8.68832398, 12.84056604,
        7.62232307, 11.85743425, 13.89671314,  9.86089579,  7.78980299,
       11.18839169, 10.68416401,  7.5984809 , 11.36089355, 11.73848065,
        7.62011734,  6.76056625,  7.58378655,  8.29640613,  5.48276038,
        8.97052115,  7.61088125, 14.03618649, 11.1913598 , 12.22503754,
        7.06169391, 13.26209545,  9.10945678,  4.91336994,  8.47371868,
        7.81509631, 11.785566  ,  9.3349858 ,  8.68832398, 10.36844813,
       11.75807252,  8.92549752,  8.47371868, 13.59673406,  5.05989547,
        9.9353921 ,  8.51928531,  9.17905491,  5.18047638,  7.24326992,
        9.48337063,  5.43943791, 11.70016217, 13.56367857, 12.30695098,
       12.27993098,  5.35774573, 10.90638066, 11.37939695,  5.57341623,
       12.84056604,  8.25828409, 13.97779839, 10.00794468,  8.29203356,
       11.1341798 , 13.12218013, 13.20988996,  6.18668463,  9.44486609,
       10.09979747])}
{'P': tensor([[3.1526e-03, 2.2362e-01, 1.9583e-01,  ..., 1.8594e-06, 3.3364e-05,
         1.8187e-05],
        [3.1077e-05, 9.7198e-03, 1.9524e-03,  ..., 2.5243e-06, 2.0915e-04,
         2.0902e-05],
        [4.0452e-05, 2.7285e-02, 9.9844e-03,  ..., 3.0605e-06, 2.2670e-04,
         2.7804e-05],
        ...,
        [4.2795e-03, 3.5813e-03, 5.1606e-03,  ..., 3.3300e-02, 3.0976e-05,
         3.3689e-03],
        [5.4408e-08, 6.8232e-08, 6.9349e-08,  ..., 4.5336e-08, 6.2656e-03,
         4.7768e-08],
        [2.3455e-01, 1.5143e-01, 1.2265e-01,  ..., 2.8874e-06, 2.1217e-05,
         3.2119e-03]]), 'X': tensor([[  9.9756,  -6.6330,  -3.3429],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 10.8670,  -5.3428,  -2.8637],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 11.2767,  -4.6006,   0.7868],
        [  6.3569,   0.7591,   6.5011],
        [  2.3928,   7.1404,  -1.7035],
        [  5.2608,  12.6977,   4.9166],
        [  5.8902,  -1.3887,   4.0388],
        [  5.3858,   4.4260,  -9.6769],
        [  7.1441,  11.5529,   2.3232],
        [ 12.6038,   0.7665,  -3.4439],
        [ -6.8466,  -7.6488,   1.9648],
        [  4.5503,   5.5359,   7.4735],
        [ -7.3333,   4.9246,  -7.9212],
        [  5.1292,   4.8183,  -5.7997],
        [ 11.7795,  -3.4697,   3.5624],
        [ -5.1398,  -9.1367,   6.2074],
        [ -1.7009,   7.8851,  -8.2437],
        [  2.3589,  13.7668,  -3.6501],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.2247,  -0.3150,  -9.3657],
        [ -1.0407,   7.9166,  -6.6021],
        [ -6.9838,  -7.4965,   0.9796],
        [  7.7750,  11.0665,   2.8364],
        [  6.2895,   0.0405,   5.7652],
        [  5.2188,  -2.5839,   2.1833],
        [ -9.0604,   1.7707,   1.1066],
        [ -2.9497, -10.3028,  -7.3479],
        [  4.0587, -10.6112,   6.2273],
        [  6.3246,   1.6696,  -0.1544],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.8400,   2.3991,   1.3457],
        [ -7.4808,   4.7462,   2.0266],
        [ -8.3900,   3.3727,   3.1650],
        [  5.6260,  -1.9401,   3.0259],
        [ -8.1426,   3.8048,   9.3542],
        [ -9.4196,   0.0478,   9.4140],
        [  9.6816,   9.1536,  -5.8543],
        [ -8.4267,  -5.3697,   2.4172],
        [ -9.3076,   0.7852,  -6.7010],
        [  2.7558,  -4.4844,   0.0636],
        [  1.8766,   7.3694,   1.1887],
        [ 10.9394,   7.3069,  -8.9365],
        [  0.9967,  -4.8144,  -9.7370],
        [ -8.6112,   2.9319,  -7.9943],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -0.0254, -10.9932,   3.2200],
        [  2.3261,  13.7749,   5.9121],
        [ -5.9225,  -8.5355,   2.4778],
        [ 12.2692,  -1.8882,   3.9253],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.2675,  12.1379,  -8.9720],
        [ 11.4495,   6.3026,   6.8257],
        [  7.1821,  -9.1845,   9.8548],
        [ -5.2589,   6.7024,  -5.5809],
        [  0.1206,  -4.7363,   8.7103],
        [  5.5294,  -2.1106,   9.0626],
        [  5.3924, -10.1396,  -4.0863],
        [ -3.7964,  -9.9294,   7.7147],
        [  4.7544,  -3.1425,  -1.5315],
        [  3.4140,  -4.1803,  -8.8224],
        [ -1.6334, -10.7207,   7.1397],
        [ -2.2188,   7.8218,   8.6064],
        [  6.3246,   1.6696,  -0.1544],
        [  9.1820,   9.7334,  -2.7932],
        [ 11.4235,   6.3594,   5.1663],
        [ -2.1382,   7.8339,   7.3524],
        [ 11.5966,   5.9666,   6.0083],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.3367,   7.1676,  -9.5263],
        [  6.1575,  12.2046,   3.3257],
        [ -3.4488, -10.0934,   2.5195],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 11.0438,   7.1180,  -5.6991],
        [  5.6248,   3.9984,   8.4933],
        [ 12.4592,  -0.9168,  -6.7571],
        [  0.6363, -11.0348,   6.1414],
        [  2.5382,   7.0672,   4.5097],
        [  8.5093,  -8.1764,  -3.2381],
        [  4.3554,   5.7393,  -9.0766],
        [ -5.9940,   6.1985,   5.9566],
        [  0.2480, -11.0153,  -3.0183],
        [  6.2242,  -0.3171,   7.4978],
        [ -6.4464,   5.8272,   7.5038],
        [ -5.1398,  -9.1367,   6.2074],
        [ -2.1533, -10.5778,   1.0112],
        [  1.7228,  -4.7575,   8.3396],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.4867,  12.5816,   2.9549],
        [ -9.4501,  -1.7133,   2.2980],
        [  7.5930,  -8.9037,   7.2323],
        [  3.3069,   6.6089,   8.7377],
        [  6.0031,  12.2958,  -6.5992],
        [  6.0140,  -1.0661,  -2.5040],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 10.8670,  -5.3428,  -2.8637],
        [  4.7482, -10.3907,   5.9149],
        [  7.1441,  11.5529,   2.3232],
        [  9.3424,  -7.3685,  -3.6807]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0847, 0.0790,  ..., 0.9306, 0.7487, 0.0443],
        [0.0847, 0.0000, 0.0613,  ..., 0.9813, 0.7200, 0.1224],
        [0.0790, 0.0613, 0.0000,  ..., 0.9215, 0.6769, 0.1232],
        ...,
        [0.9306, 0.9813, 0.9215,  ..., 0.0000, 0.8253, 0.9348],
        [0.7487, 0.7200, 0.6769,  ..., 0.8253, 0.0000, 0.7880],
        [0.0443, 0.1224, 0.1232,  ..., 0.9348, 0.7880, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.97957425, 12.12448912, 12.10942412, 12.1460931 , 12.07818871,
       12.30858677, 12.17901033,  6.4020404 ,  7.53063885, 13.744387  ,
        6.05165248,  6.97106912, 13.58333857, 12.62711252, 10.26545792,
        7.16598273,  8.83341608,  7.03733775, 12.27991991, 10.48315436,
        8.06643487, 13.96746816, 13.18278606,  6.23262289,  7.98469166,
       10.24557047, 13.5247215 ,  6.28961988,  5.82345867,  9.23178204,
       10.7167285 , 11.36089355,  6.54127431, 12.21532282,  9.15976746,
        8.859416  ,  9.04255655,  5.95111804,  8.98765644,  9.41970031,
       13.32374056,  9.99211663,  9.34061938,  5.26342203,  7.60463349,
       13.15525803,  4.91652556,  9.09661212,  9.97964013, 10.99326577,
       13.96987806, 10.38897563, 12.41366754, 13.25017211,  5.69913817,
        9.96152788,  8.16714836, 13.66051884, 13.06957425, 11.65923396,
        8.51928531,  4.73784222,  5.91854397, 11.48434767, 10.63038565,
        5.69913817,  5.39723241, 10.84437705,  8.1303945 ,  6.54127431,
       13.38091121, 13.0743399 ,  8.12043983, 13.04156075, 13.03002338,
        7.53884104, 13.66991027, 10.66631784, 10.46739073, 13.13888841,
        6.90115901, 12.49291768, 11.05317597,  7.50917551, 11.80092578,
        7.20482224,  8.62260327, 11.018084  ,  6.2322769 ,  8.68979052,
       10.48315436, 10.794755  ,  5.05981914,  9.15866607, 13.72594022,
        9.60412735, 11.70168755,  7.39004226, 13.6829843 ,  6.10774544,
       12.3828957 , 12.53285633, 12.10942412, 11.42420508, 13.58333857,
       11.89855267])}
{'P': tensor([[4.3550e-03, 2.3653e-01, 1.9217e-01,  ..., 7.0145e-05, 1.7232e-07,
         1.5591e-06],
        [3.4997e-05, 1.1634e-02, 2.9040e-02,  ..., 1.3724e-04, 3.3547e-07,
         2.0193e-06],
        [6.1855e-05, 3.2203e-03, 2.3131e-02,  ..., 4.4249e-04, 5.7769e-07,
         3.5933e-06],
        ...,
        [2.6695e-04, 3.7639e-04, 5.6857e-04,  ..., 3.2802e-01, 1.1320e-05,
         9.7210e-05],
        [5.2596e-04, 2.5233e-04, 1.7862e-04,  ..., 6.0022e-06, 2.2147e-02,
         3.6952e-03],
        [9.6108e-02, 3.6912e-02, 3.9977e-02,  ..., 7.8133e-04, 1.3575e-05,
         4.2535e-02]]), 'X': tensor([[ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.2670e-02, 1.0151e-01,  ..., 6.0635e-01, 8.2566e-01,
         4.5714e-01],
        [9.2670e-02, 0.0000e+00, 1.2134e-01,  ..., 6.4987e-01, 8.6502e-01,
         5.4842e-01],
        [1.0151e-01, 1.2134e-01, 0.0000e+00,  ..., 5.2883e-01, 9.2594e-01,
         5.0656e-01],
        ...,
        [6.0635e-01, 6.4987e-01, 5.2883e-01,  ..., 0.0000e+00, 1.3188e+00,
         6.7014e-01],
        [8.2566e-01, 8.6502e-01, 9.2594e-01,  ..., 1.3188e+00, 3.0339e-04,
         6.9124e-01],
        [4.5714e-01, 5.4842e-01, 5.0656e-01,  ..., 6.7014e-01, 6.9124e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([11.99633819, 12.16901615, 12.19805394, 12.17451685, 12.26997165,
       12.17901033,  5.19577146, 11.50791859, 13.744387  ,  5.37343173,
        6.59176018,  7.49076899,  6.49538841,  5.0713364 ,  9.60412735,
        7.822854  , 13.79676654,  8.74470218,  9.18358562,  8.41148722,
       10.06023665, 12.05395056, 12.77402351, 10.35485331,  8.27853076,
        8.72896381,  8.92549752, 12.71843462,  5.92281142,  7.73692436,
       10.71162681,  5.14999799,  9.96152788,  6.2134961 ,  5.46244988,
        9.95552626,  5.53255882,  9.79814997, 13.41713661,  8.04664886,
        8.74470218,  9.10945678,  5.10446653, 12.76408562, 11.83570132,
        5.62005629, 13.96987806, 11.31276109, 12.0495521 , 12.67680439,
        5.16205087,  7.38844703, 12.49291768,  7.06272256, 10.00410625,
        5.10444097, 13.83139024, 11.48734978,  8.41148722,  8.62260327,
        8.859416  ,  7.92200342, 10.68031351, 11.62770332, 10.69963954,
       14.09176259, 12.05395056,  6.02779201,  5.56809335, 10.794755  ,
        7.67939546, 11.04953767, 12.30695098, 10.19036081,  9.58347456,
        8.57560275,  9.64157461,  5.5366698 ,  7.5158489 ,  6.80370254,
        8.11227668, 11.9702762 ,  5.81199422, 13.50484539,  7.6374603 ,
       14.05687815, 13.32613445,  8.62329466, 11.74901778, 10.71162681,
        4.78002665,  9.76896904, 13.83139024, 13.59170809, 10.33712258,
        8.85572292,  7.8575648 ,  7.5403822 ,  8.11227668,  8.25828409,
        6.53250937, 13.91198191,  6.99097013,  6.38105015, 10.4606265 ,
       11.28895489])}
{'P': tensor([[3.0071e-03, 1.9867e-01, 1.3785e-01,  ..., 4.2493e-07, 1.6092e-07,
         1.2132e-04],
        [1.2003e-05, 3.4111e-03, 3.5238e-01,  ..., 6.3610e-07, 1.3059e-07,
         1.0363e-04],
        [3.1684e-05, 6.0279e-05, 1.3411e-02,  ..., 2.9376e-06, 4.3803e-07,
         3.3019e-04],
        ...,
        [5.1043e-06, 3.7038e-06, 3.1696e-06,  ..., 3.4984e-02, 6.4434e-04,
         4.0391e-05],
        [2.5794e-03, 1.4275e-03, 1.0255e-03,  ..., 5.5473e-06, 4.6014e-02,
         2.2634e-03],
        [2.0186e-03, 3.6068e-03, 4.8061e-03,  ..., 4.2981e-04, 1.3133e-04,
         2.0316e-01]]), 'X': tensor([[ 10.1983,  -6.3430,  -0.3200],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.3367,  -4.4806,   0.4643],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 11.2767,  -4.6006,   0.7868],
        [ -9.1813,  -3.3550,   1.8827],
        [  0.9659,  -4.8144,  -0.9594],
        [  6.3606,   0.9204,   3.2525],
        [  2.7558,  -4.4844,   0.0636],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.3589,   0.8239,  -8.4230],
        [  4.5800,   5.5034,  -3.9432],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.4645,  -1.4880,  -2.0226],
        [  7.5309,  11.2617,   7.2656],
        [  3.6249, -10.7255,   5.6605],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.6319,  11.9066,   3.5165],
        [ -7.9925,  -6.1423,   2.8990],
        [ -8.5149,  -5.1905,   1.5239],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.9956,  11.6590,  -7.6876],
        [  2.2437,  -4.6489,   6.1430],
        [  0.9036,   7.6840,   9.9237],
        [ -8.8934,   2.2597,   3.5831],
        [  6.0310,  12.2795,  -6.4528],
        [  0.1382, -11.0073,   1.0026],
        [  1.3339,   7.5627,  -0.9034],
        [  2.6614,  -4.5192,  -9.3641],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 10.3251,   8.2928,  -3.0006],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.2437,  -4.6489,   6.1430],
        [ -5.1398,  -9.1367,   6.2074],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.4735,  -0.7274,  -2.6749],
        [  2.6614,  -4.5192,  -9.3641],
        [ 12.4996,   2.6335,  -6.3166],
        [  6.2804,   2.0019,   0.8607],
        [  5.5240,  -2.1198,   6.2055],
        [  6.9653,  -9.3225,  -3.4727],
        [  5.8902,  -1.3887,   4.0388],
        [ -6.6998,   5.5954,   0.0823],
        [  5.6928,  12.4710,   3.6001],
        [  9.9750,  -6.6338,  -1.2440],
        [ 10.1503,   8.5414,  -7.2681],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.1813,  -3.3550,   1.8827],
        [  4.0591,   6.0202,  -1.6339],
        [  0.2571,  -4.7594,   1.5618],
        [  1.6232,  -4.7717,  -2.5655],
        [  0.3715,  -4.7756,   0.2312],
        [  8.1043,  -8.5164,  -7.9290],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 11.3367,  -4.4806,   0.4643],
        [ -9.3810,  -2.3455,  -9.5194],
        [  5.1861,   4.7361,  -3.7312],
        [  1.7228,  -4.7575,   8.3396],
        [  5.1861,   4.7361,  -3.7312],
        [  4.0362,  13.2397,  -7.0030],
        [  1.1350,  14.0104,   0.6321],
        [  8.5093,  -8.1764,  -3.2381],
        [  5.2608,  -2.5256,  -9.1446],
        [  2.5128, -10.9365,   9.9645],
        [  9.6614,   9.1784,  -1.2087],
        [ -8.9636,   2.0648,   8.1946],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.7420,  -4.4896,  -3.7139],
        [  4.9452, -10.3187,  -7.8621],
        [ -4.4520,   7.1370,  -8.7867],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.3616,  -2.4750,   3.7765],
        [  3.3633,  13.4794,   0.5193],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -9.1960,  -3.2964,   1.6921],
        [  8.3791,  10.5409,   5.5590],
        [  9.8082,  -6.8396,   4.9615],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 11.5966,   5.9666,   6.0083],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -1.9600,   7.8577,   6.9121],
        [ -2.9497, -10.3028,  -7.3479],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.0534,  -6.0429,   2.4583],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.1890,   4.7318,   0.9515],
        [  6.3516,   0.6425,   0.3425],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.0140,  -1.0661,  -2.5040],
        [  1.2273,  -4.8081,   9.8129],
        [ 10.4737,  -5.9570,  -1.2108],
        [  2.4697,  -4.5836,   1.4593],
        [  1.1350,  14.0104,   0.6321],
        [ -0.6323,   7.9085,  -5.9544],
        [ -9.1907,  -3.3180,  -5.5943],
        [  6.2036,   2.4030,  -1.2500],
        [  2.5318,  13.7231,   6.3692],
        [ -8.6603,   2.8250,   5.9181],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.1292,   4.8183,  -5.7997],
        [  5.1890,   4.7318,   0.9515],
        [ -7.5720,   4.6310,   2.1717],
        [ -3.5173, -10.0623,   4.8280],
        [  5.6093,  -1.9705,   6.4807]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0599, 0.1076,  ..., 0.5414, 0.8003, 0.5096],
        [0.0599, 0.0000, 0.0477,  ..., 0.5870, 0.8377, 0.4738],
        [0.1076, 0.0477, 0.0000,  ..., 0.6256, 0.8700, 0.4500],
        ...,
        [0.5414, 0.5870, 0.6256,  ..., 0.0000, 0.2862, 0.7236],
        [0.8003, 0.8377, 0.8700,  ..., 0.2862, 0.0000, 0.8412],
        [0.5096, 0.4738, 0.4500,  ..., 0.7236, 0.8412, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.00995898, 12.1494023 , 12.22276278, 12.18998195, 12.25806559,
       12.17901033,  9.7751189 ,  4.91038884,  6.42689705,  5.26342203,
        9.70108409,  6.41202986,  7.15990676, 13.05148826,  9.58072286,
       13.54774355, 11.32149259, 12.40074901, 13.62896593, 10.08001894,
        9.97220799,  9.7712321 , 13.59673406,  5.16205087,  7.73692436,
        9.17595899, 13.68063098, 11.00812996,  7.67939546,  5.24461068,
       12.4013153 , 13.2430377 ,  8.30125201,  5.16205087, 10.48315436,
       10.99326577,  9.5014146 ,  5.24461068, 12.77402351,  6.59176018,
        5.91677188, 11.63724792,  6.05165248,  8.72896381, 13.70893492,
       11.97949273, 13.26590071,  9.13507321,  9.7751189 ,  7.26074925,
        4.7663526 ,  5.04028538,  4.79003361, 11.75618513, 10.23946247,
       12.18998195,  9.66977802,  7.02326515,  5.05981914,  7.02326515,
       13.84126197, 14.05633441, 11.80092578,  5.83560009, 11.22142195,
       13.32613445,  9.19837792, 12.05395056,  5.26067019, 11.4424816 ,
        8.41171369, 12.19805394,  9.68324553, 13.89264822, 10.52587243,
        9.76896904, 13.46553623, 11.95743558, 10.32715715, 13.04156075,
        6.49538841,  9.85330646,  8.09842704, 10.7167285 ,  9.73579738,
       10.06850677, 13.04217487,  7.02253235,  6.38399789, 13.01187484,
        6.10774544,  4.96231532, 12.04924159,  5.20659892, 14.05633441,
        7.9337647 ,  9.7712321 ,  6.652742  , 13.95472331,  9.10945678,
       12.32408516,  7.03733775,  7.02253235,  8.87587457, 10.65929968,
        5.94535529])}
{'P': tensor([[6.0882e-03, 3.1872e-01, 1.5503e-01,  ..., 3.9971e-05, 5.1917e-05,
         6.0976e-07],
        [2.3034e-05, 1.0381e-02, 8.2252e-01,  ..., 3.1046e-05, 1.8193e-04,
         1.4806e-06],
        [7.3248e-05, 1.9526e-04, 6.9312e-02,  ..., 1.2827e-04, 2.0695e-03,
         1.3976e-05],
        ...,
        [1.5850e-02, 1.3875e-02, 1.1030e-02,  ..., 8.6124e-02, 1.0924e-03,
         7.3578e-05],
        [6.9672e-06, 7.6041e-06, 8.6718e-06,  ..., 1.5251e-04, 6.6428e-02,
         9.8772e-03],
        [2.0443e-05, 1.2221e-05, 9.4717e-06,  ..., 6.1767e-04, 1.7050e-04,
         1.9076e-01]]), 'X': tensor([[ 10.4737,  -5.9570,  -1.2108],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 11.1085,  -4.9204,   0.0736],
        [ -9.1178,   1.5784,   2.1410],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1064,  13.5607,   6.7605],
        [  6.3549,   0.7096,   6.9198],
        [  7.0642,  11.6104,  -9.8316],
        [  2.7558,  -4.4844,   0.0636],
        [  1.6308,  13.9254,  -2.0984],
        [ 10.1798,   8.5002,  -5.8893],
        [ 11.4703,  -4.2006,   7.4086],
        [  0.1460,   7.8340,  -9.4614],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.3542,   0.6934,  -3.3840],
        [ -5.1046,  -9.1610,   2.3418],
        [  3.3943,   6.5483,   1.0790],
        [  4.2096,   5.8815,   1.9230],
        [  1.7500,   7.4187,   4.4237],
        [ -4.3963,   7.1629,   4.5263],
        [  6.0140,  -1.0661,  -2.5040],
        [ -8.1204,  -5.9308,   4.0761],
        [ -0.0254, -10.9932,   3.2200],
        [ -3.0644,   7.6439,  -1.4322],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -2.5056,   7.7720,   5.5878],
        [  6.1579,  -0.5961,  -6.5595],
        [ 11.0438,   7.1180,  -5.6991],
        [ 12.2311,  -2.0445,   1.3545],
        [ 11.5568,   6.0602,   9.3244],
        [  1.3403,  -4.8011,  -1.1698],
        [ -9.2508,   1.0572,   2.9978],
        [  2.2849, -10.9655,  -7.6542],
        [  1.6422,   7.4586,  -6.8056],
        [ 11.7418,   5.6070,  -1.4192],
        [  6.3606,   0.9204,   3.2525],
        [ -6.2668,   5.9808,   2.5614],
        [ -8.5149,  -5.1905,   1.5239],
        [  1.7301,   7.4262,   5.0522],
        [  2.8890,  -4.4316,   1.1600],
        [  3.9591, -10.6391,  -6.2557],
        [  5.9587,  12.3216,   9.1695],
        [  5.0277,   4.9589,   7.0175],
        [ -9.3616,  -2.4749,   0.1793],
        [ 10.2712,   8.3707,  -4.9743],
        [  2.0870,  -4.6877,   9.6823],
        [  6.8704,  -9.3809,   7.5769],
        [  0.6470, -11.0352,  -1.1931],
        [ -5.1046,  -9.1610,   2.3418],
        [ -7.8126,   4.3066,   5.1791],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.1813,  -3.3550,   1.8827],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -8.8631,  -4.3728,  -1.1746],
        [  5.9526,   3.2525,  -4.3821],
        [ -3.3116, -10.1539,   7.5919],
        [  2.1504,  -4.6726,   8.7908],
        [  7.7794,  11.0629,   8.9069],
        [ -4.4503,   7.1378,   3.3751],
        [  4.5325,  -3.3643,   6.7699],
        [  1.8721,  -4.7322,   0.5524],
        [  1.7301,   7.4262,   5.0522],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -1.1885, -10.8211,   0.0865],
        [ -5.9938,  -8.4745,   3.9735],
        [  6.2905,   1.9366,   2.1169],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -7.5401,   4.6718,   9.0033],
        [  4.2339,  -3.6268,   6.3125],
        [  8.8091,  -7.9037,   0.3631],
        [  6.2902,   0.0451,   3.7116],
        [  3.6249, -10.7255,   5.6605],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -9.1960,  -3.2964,   1.6921],
        [  1.7500,   7.4187,   4.4237],
        [ -8.6050,   2.9451,  -0.7012],
        [  2.7558,  -4.4844,   0.0636],
        [  7.1821,  -9.1845,   9.8548],
        [  2.1440, -10.9811,   8.5281],
        [ -1.6334, -10.7207,   7.1397],
        [ -8.6426,  -4.9136,   9.4941],
        [ -2.8162,   7.7059,  -9.6885],
        [  5.0233,   4.9647,  -3.1788],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -5.1377,   6.7751,  -0.4438],
        [  4.9152,  12.8653,  -1.4459],
        [  9.3753,   9.5171,   3.9131],
        [  8.2422,  -8.4041,  -1.4962],
        [ 12.5927,   0.4199,  -9.4997],
        [  3.1892, -10.8220,  -6.8409],
        [ -2.1817,   7.8274,  -4.6298],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -9.3705,   0.4208,  -3.5815],
        [  7.2340,  -9.1505,  -5.8321],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.0560,  -0.9421,  -5.0334],
        [ -6.0195,   6.1789,  -2.9496],
        [  4.7544,  -3.1425,  -1.5315],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.6451,  -4.5250,  -0.5565],
        [  1.1139,   7.6281,  -2.5680],
        [ -7.8280,   4.2848,  -7.1955]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0961, 0.1504,  ..., 0.0823, 0.0890, 0.6869],
        [0.0961, 0.0000, 0.0558,  ..., 0.0699, 0.1785, 0.7762],
        [0.1504, 0.0558, 0.0000,  ..., 0.1184, 0.2279, 0.8211],
        ...,
        [0.0823, 0.0699, 0.1184,  ..., 0.0000, 0.1700, 0.7639],
        [0.0890, 0.1785, 0.2279,  ..., 0.1700, 0.0000, 0.5988],
        [0.6869, 0.7762, 0.8211,  ..., 0.7639, 0.5988, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.04924159, 12.25806559, 12.4013153 , 12.22276278, 12.1460931 ,
       12.1494023 ,  9.25336371,  8.01009083,  5.13182506, 13.91198191,
        6.39438044, 13.59056613,  5.26342203, 14.02058621, 13.26209545,
       12.21532282,  7.83534987, 10.53368644,  6.39187285, 10.48721382,
        7.37576154,  7.23276582,  7.62232307,  8.40445872,  6.10774544,
       10.05558893, 10.99326577,  8.23525433, 12.07818871,  8.16584537,
        6.18668463, 13.13888841, 12.40074901, 13.0493368 ,  4.98461983,
        9.3109884 , 11.20100795,  7.63726134, 13.01187484,  6.42689705,
        8.66273171,  9.97220799,  7.62509582,  5.29009836, 11.35182589,
       13.68672855,  7.06169391,  9.68323686, 13.25017211,  5.13125513,
       11.62770332, 11.05413545, 10.48721382,  8.92098752, 13.09737053,
        9.7751189 ,  9.71392179,  9.88310614,  6.78325151, 10.68031351,
        5.14371005, 13.52430176,  8.41148722,  5.64464807,  5.08910491,
        7.62509582, 10.23946247, 10.88618114, 10.37998078,  6.58184474,
       12.30858677,  8.87007495,  5.5748624 , 11.83509372,  6.29034958,
       11.32149259,  9.58072286,  9.76896904,  7.62232307,  9.09502385,
        5.26342203, 11.65923396, 11.18839169, 10.84437705,  9.94173708,
        8.20436764,  7.06272256,  9.97964013,  8.50278896, 13.77222752,
       13.35927527, 11.77124373, 12.59969936, 11.28215429,  8.12580976,
       10.31613269,  9.37989631, 11.66453059, 12.41366754,  6.12886024,
        8.62631801,  5.69913817, 10.08001894,  5.24137909,  7.70897675,
        8.92395381])}
{'P': tensor([[3.7715e-03, 2.8024e-01, 2.1650e-01,  ..., 1.9857e-07, 9.9435e-08,
         1.3121e-07],
        [6.1664e-05, 2.0927e-02, 2.6251e-03,  ..., 7.6875e-07, 4.3197e-07,
         7.6987e-07],
        [6.7685e-05, 7.2693e-02, 1.5917e-02,  ..., 5.9234e-07, 3.1604e-07,
         6.0936e-07],
        ...,
        [1.0241e-02, 4.8583e-03, 4.8916e-03,  ..., 1.6169e-01, 1.1998e-03,
         3.1780e-05],
        [1.2125e-03, 6.3146e-04, 6.0176e-04,  ..., 3.3776e-01, 1.5486e-01,
         4.4865e-05],
        [2.0792e-05, 1.2009e-05, 1.3716e-05,  ..., 2.0170e-03, 6.6207e-04,
         3.7237e-02]]), 'X': tensor([[ 10.4758,  -5.9539,   1.3667],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.2767,  -4.6006,   0.7868],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.3367,  -4.4806,   0.4643],
        [ 11.5083,  -4.1177,   0.2389],
        [ -5.5525,   6.5146,  -4.4984],
        [  3.9515,  -3.8425,   6.1729],
        [ 12.2592,   3.9425,   2.0383],
        [  4.2339,  -3.6268,   6.3125],
        [  5.8889,  -1.3916,   8.8993],
        [ 11.7795,  -3.4697,   3.5624],
        [  5.4426, -10.1181,  -5.1268],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.6095,  -1.9702,  -7.8233],
        [  3.3943,   6.5483,   1.0790],
        [  6.3301,   0.3638,  -7.5685],
        [  6.1752,   2.5242,   3.5874],
        [  2.5974,   7.0363,  -7.7810],
        [ -2.2508, -10.5479,  -2.7235],
        [ 12.4690,  -0.8528,  -9.3374],
        [  2.0449,  13.8402,  -7.1642],
        [ -6.0195,   6.1789,  -2.9496],
        [ -2.0721,   7.8432,   6.6380],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.9843,  -9.8679,  -6.1449],
        [  3.6703,   6.3443,  -5.4401],
        [  6.1989,  -9.7591,   0.8289],
        [ -7.3470,  -7.0615,   5.0066],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.2675,  12.1379,  -8.9720],
        [  5.8889,  -1.3916,   8.8993],
        [ 12.6038,   0.7665,  -3.4439],
        [ -4.8665,  -9.3204,  -5.2564],
        [  0.5175,  14.0910,   4.6540],
        [  5.2199,   4.6861,  -1.9304],
        [  3.9515,  -3.8425,   6.1729],
        [  1.3767, -11.0341,   9.9415],
        [ -0.2586,   7.8825,  -9.9840],
        [  2.7295,  13.6703,  -8.9328],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.3200,   0.2690,   1.5900],
        [  5.9026,   3.3837,  -0.6573],
        [  5.8275,  -1.5334,   2.0124],
        [ -2.3317,   7.8034,  -3.6274],
        [ -9.3411,  -2.6004,  -2.7062],
        [  6.0890,  -0.8382,   2.9166],
        [  5.8617,   3.4853,  -9.8080],
        [ -8.6112,   2.9319,  -7.9943],
        [  2.1477,   7.2550,   8.9688],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.2895,   0.0405,   5.7652],
        [  6.1755,   2.5233,  -2.0042],
        [ -9.4753,  -0.8012,  -5.2050],
        [  2.1771, -10.9776,  -3.2007],
        [  5.6928,  12.4710,   3.6001],
        [ -3.2486,   7.5925,  -1.3719],
        [ -5.2589,   6.7024,  -5.5809],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.3924, -10.1396,  -4.0863],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -5.9225,  -8.5355,   2.4778],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.3860,  -2.3418,   1.0826],
        [ -4.4520,   7.1370,  -8.7867],
        [  0.9967,  -4.8144,  -9.7370],
        [  4.6231,   5.4556,   8.4973],
        [ -2.5683, -10.4433,  -2.9182],
        [ -3.2053, -10.1993,  -3.8736],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.4756,  -0.8166,  -9.6021],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.8323,  -1.5228,   5.1132],
        [  1.0000,   7.6591,  -4.5011],
        [  1.1139,   7.6281,  -2.5680],
        [  5.4429,  -2.2529,   9.7306],
        [  0.5175,  14.0910,   4.6540],
        [ -9.4159,  -2.0739,   9.6841],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.2275,  -3.6320,  -0.2714],
        [  2.2437,  -4.6489,   6.1430],
        [  9.6816,   9.1536,  -5.8543],
        [ -9.4630,  -1.5145,   0.2983],
        [  8.0381,  -8.5690,  -8.9523],
        [ 11.9936,  -2.8677,   6.0614],
        [  7.0664,  -9.2590,   8.0879],
        [ 11.1420,   6.9331,  -2.4779],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.2076,  13.8032,  -1.5099],
        [ -5.9987,   6.1949,  -6.2114],
        [  5.6314,  -1.9302,  -9.6543],
        [ -8.1666,   3.7653,   3.6075],
        [  6.2905,   1.9366,   2.1169],
        [ -0.0410,   7.8591,   9.3457],
        [ -7.7239,   4.4298,   3.1216],
        [  2.0870,  -4.6877,   9.6823],
        [  1.1274,  14.0116,   8.1753],
        [ -9.1459,   1.4780,  -4.2403],
        [  0.6757, -11.0361,  -1.3047],
        [  5.3118,   4.5452,   4.2114],
        [ -3.5173, -10.0623,   4.8280],
        [ -6.1914,  -8.2997,   8.5380],
        [ -9.4735,  -1.2620,  -3.6742]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.1494e-02, 7.7464e-02,  ..., 7.5706e-01, 9.1977e-01,
         9.1304e-01],
        [7.1494e-02, 0.0000e+00, 5.8199e-02,  ..., 8.1995e-01, 9.7626e-01,
         9.8041e-01],
        [7.7464e-02, 5.8199e-02, 0.0000e+00,  ..., 8.3175e-01, 9.9676e-01,
         9.4711e-01],
        ...,
        [7.5706e-01, 8.1995e-01, 8.3175e-01,  ..., 0.0000e+00, 2.1345e-01,
         7.1912e-01],
        [9.1977e-01, 9.7626e-01, 9.9676e-01,  ..., 2.1345e-01, 3.0339e-04,
         9.0867e-01],
        [9.1304e-01, 9.8041e-01, 9.4711e-01,  ..., 7.1912e-01, 9.0867e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.0495521 , 12.19805394, 12.17901033, 12.26997165, 12.18998195,
       12.22276278,  8.55982105,  5.51177501, 12.87752052,  5.5748624 ,
        6.05113217, 12.27991991, 11.4890861 , 10.31613269,  5.94541803,
        7.37576154,  6.34059021,  6.67121748,  7.50035355, 10.78533838,
       12.4980845 , 13.99048094,  8.62631801,  8.11227668, 12.76408562,
       12.3828957 , 10.78533838, 11.54071859,  7.32950385, 11.56146873,
       10.19036081, 10.91805683, 13.66051884,  6.05113217, 12.62711252,
       10.51437947, 14.10045797,  7.01475376,  5.51177501, 11.11969815,
        7.88677538, 13.94008948, 10.8207297 ,  6.32572323,  6.80370254,
        6.02588517,  8.14433542,  9.69629   ,  6.1463896 ,  6.81961996,
        9.09661212,  7.56617037, 10.06437598,  6.28961988,  6.67108662,
        9.50913849, 11.1913598 , 13.70893492,  8.25828409,  8.51928531,
        9.64157461, 11.48434767, 10.1979669 , 10.38897563,  8.56889738,
        5.87305005,  8.41171369,  4.91652556,  7.15100303, 10.75443106,
       10.6910768 , 11.80092578,  9.51074763,  5.57341623,  6.02779201,
        7.72415048,  7.70897675,  5.89074104, 14.10045797,  9.64157461,
       10.00802627,  5.57341623,  5.16205087, 13.32374056,  9.58347456,
       11.74901778, 12.33167382, 11.64746882, 13.12299145, 12.24467041,
       11.63724792, 13.97857591,  8.62329466,  5.95297459,  8.99276564,
        6.58184474,  7.85919687,  8.90405823,  5.13125513, 14.05687815,
        9.26455795, 11.05672136,  6.99097013, 10.65929968, 10.35464916,
        9.55720651])}
{'P': tensor([[4.8775e-03, 2.7981e-01, 2.0149e-01,  ..., 7.2688e-05, 2.5093e-05,
         4.7307e-06],
        [4.5926e-05, 5.5841e-03, 7.4914e-02,  ..., 1.2049e-04, 3.4658e-05,
         5.9353e-06],
        [5.3658e-05, 3.4390e-04, 5.4997e-03,  ..., 1.2529e-04, 3.4717e-05,
         6.2034e-06],
        ...,
        [1.4220e-07, 1.5621e-07, 1.5853e-07,  ..., 1.5296e-02, 5.9136e-03,
         5.1458e-07],
        [2.3473e-05, 2.3097e-05, 2.2942e-05,  ..., 8.2115e-04, 2.5839e-01,
         1.3105e-05],
        [1.2648e-04, 1.6816e-04, 1.8277e-04,  ..., 3.1340e-04, 1.0801e-04,
         3.4253e-01]]), 'X': tensor([[ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0346, 0.0406,  ..., 0.3242, 0.2154, 1.0848],
        [0.0346, 0.0000, 0.0074,  ..., 0.2901, 0.2439, 1.0707],
        [0.0406, 0.0074, 0.0000,  ..., 0.2854, 0.2464, 1.0642],
        ...,
        [0.3242, 0.2901, 0.2854,  ..., 0.0000, 0.5251, 1.0488],
        [0.2154, 0.2439, 0.2464,  ..., 0.5251, 0.0000, 1.0604],
        [1.0848, 1.0707, 1.0642,  ..., 1.0488, 1.0604, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.05300943, 12.11735334, 12.12641969, 12.27835996, 12.35421461,
       12.32545637,  8.99276564, 11.018084  ,  6.53290663,  9.38232121,
        5.05989547, 11.83509372,  8.0550286 , 10.48315436, 10.23946247,
        9.37989631, 12.44177641,  7.73692436,  4.7663526 ,  5.35774573,
        5.26342203,  5.13616224,  5.91854397, 10.24557047, 11.07637531,
       12.30858677, 12.24467041, 13.79762415,  8.8353674 ,  7.69752262,
        9.3349858 , 11.51811603,  7.61088125, 11.83509372, 13.48809595,
        9.15866607,  7.24326992, 11.28895489, 11.4088043 , 10.06023665,
        5.13182506, 11.4424816 , 11.77124373,  6.80559669,  8.24808832,
       13.92390689, 11.75618513,  6.01356004, 10.31613269,  5.24461068,
        4.89121805,  9.85384623,  9.41970031, 11.28895489,  6.2134961 ,
        7.07902036, 10.27716434,  9.15866607, 11.31276109, 11.75807252,
        8.10886021,  6.32572323, 14.03618649,  9.88571713,  8.62162372,
       13.12299145,  6.81084724,  9.88571713, 10.06023665, 10.06286074,
       13.93790235, 11.83075005, 10.36844813,  5.10444097,  7.32754882,
        9.37989631, 12.27993098, 13.96746816, 12.27993098, 11.4088043 ,
       10.00802627,  9.17905491,  8.53014915, 11.48434767, 13.37121923,
       14.07623266,  7.89745107, 12.889695  ,  5.57341623, 11.92133129,
        6.97106912, 12.52372629, 10.1979669 ,  7.49076899, 13.95472331,
       12.18998195,  5.13616224,  8.79284466,  8.81930525, 10.08001894,
        4.85603886, 10.69963954,  9.37296102, 13.68063098,  7.83534987,
        6.05113217])}
{'P': tensor([[5.3296e-03, 3.6617e-01, 1.6521e-01,  ..., 1.7748e-07, 2.1153e-05,
         3.0519e-07],
        [2.6517e-05, 6.9466e-03, 4.6404e-01,  ..., 2.0082e-07, 9.3363e-06,
         4.9710e-07],
        [5.7410e-05, 9.9246e-05, 4.1009e-02,  ..., 1.0460e-06, 1.8715e-05,
         4.0357e-06],
        ...,
        [1.2884e-03, 7.0151e-04, 3.1191e-04,  ..., 1.8467e-01, 3.8912e-03,
         7.4839e-05],
        [2.9218e-02, 4.2153e-02, 2.8986e-02,  ..., 5.1373e-07, 1.0895e-02,
         6.8873e-07],
        [7.3481e-06, 5.4635e-06, 3.5837e-06,  ..., 5.6053e-03, 2.9337e-05,
         5.3836e-02]]), 'X': tensor([[ 10.5057,  -5.9099,  -5.1599],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -8.9033,   2.2329,  -8.8912],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.7976,  -4.4682,  -4.2020],
        [  7.3592,  11.3937,   6.2407],
        [  8.8091,  -7.9037,   0.3631],
        [  9.4554,   9.4245,   9.8800],
        [ 11.5935,   5.9740,  -4.3952],
        [  6.3301,   0.3638,  -7.5685],
        [  6.0560,  -0.9421,  -5.0334],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.6022,  -1.9833,   8.3081],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.0073,   3.0986,   6.7937],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 12.5123,  -0.5339,  -1.0728],
        [  6.0073,   3.0986,   6.7937],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.1350,  14.0104,   0.6321],
        [ 12.5123,  -0.5339,  -1.0728],
        [  4.5767,   5.5071,   3.9069],
        [ -4.3359,  -9.6431,  -9.0616],
        [  8.9284,  10.0036,  -6.9210],
        [  5.2608,  -2.5256,  -9.1446],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.1588,   4.3364,  -8.0867],
        [ 11.5456,   6.0861,   8.1101],
        [ -9.4753,  -0.8012,  -5.2050],
        [  9.2979,   9.6048,   7.3138],
        [  0.9036,   7.6840,   9.9237],
        [  6.0955,   2.8210,  -4.3321],
        [  4.1140,  -3.7220,  -0.1273],
        [  9.3607,   9.5337,  -4.5412],
        [ -4.6393,   7.0461,   1.1624],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.6426,  -4.9136,   9.4941],
        [  9.5018,  -7.1945,  -1.1892],
        [ -3.2486,   7.5925,  -1.3719],
        [ -0.5739,   7.9056,   0.6382],
        [  6.1741,  -0.5329,   3.1913],
        [ -9.1960,  -3.2964,   1.6921],
        [  2.2076,  13.8032,  -1.5099],
        [  4.1348,   5.9514,  -0.8856],
        [  3.4401,  13.4540,   4.6576],
        [  2.1504,  -4.6726,   8.7908],
        [  7.8425,  -8.7202,   6.4530],
        [  6.1372,   2.6726,  -6.7885],
        [ 10.3251,   8.2928,  -3.0006],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.0534,  -6.0429,   2.4583],
        [ -6.9838,  -7.4965,   0.9796],
        [  6.8407,  11.7666,   3.0045],
        [  3.2159, -10.8166,  -1.0406],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.4401,  13.4540,   4.6576],
        [ -9.4378,  -1.8606,  -2.5668],
        [  0.9809,  -4.8145,   6.8706],
        [ -6.7745,  -7.7264,   8.4684],
        [  1.4489,   7.5256,   4.4529],
        [  9.0051,   9.9234,   5.4411],
        [  5.8889,  -1.3916,   8.8993],
        [  0.3493,  -4.7727,  -6.5682],
        [  5.9883,   3.1536,   6.2320],
        [ -8.5149,  -5.1905,   1.5239],
        [ -6.6895,   5.6051,   2.7899],
        [ -6.1914,  -8.2997,   8.5380],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.5199,  -0.4711,   3.0436],
        [ -5.1369,   6.7755,   8.3836],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.2182,  13.8007,   5.1889],
        [ -3.3116, -10.1539,   7.5919],
        [  6.0054,  -1.0902,   8.3079],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4671,  -0.5553,   0.1117],
        [  8.5716,  10.3596,   6.8818],
        [ 12.5154,   2.5072,  -9.3274],
        [  0.1252, -11.0062,   7.2564],
        [  6.0031,  12.2958,  -6.5992],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.3118,   4.5452,   4.2114],
        [  4.2339,  -3.6268,   6.3125],
        [  5.1519, -10.2387,   9.8191],
        [ -9.4117,   0.1160,   8.2574],
        [ -8.7409,  -4.6837,  -6.5840],
        [  7.3592,  11.3937,   6.2407],
        [ -8.3900,   3.3727,   3.1650],
        [ -7.6650,   4.5092,   0.2520],
        [  6.5428,  -9.5728,  -2.9897],
        [ 12.2914,   3.8037,   6.0978],
        [  9.3607,   9.5337,  -4.5412],
        [ -4.7389,   6.9953,   3.7638],
        [  5.1667,  -2.6543,   1.2879],
        [ -6.3489,   5.9117,   6.5161],
        [ -7.6293,  -6.6859,  -5.5161],
        [  8.0877,  -8.5297,  -1.1050],
        [ -8.7078,   2.7182,   1.8146]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0534, 0.1177,  ..., 0.9113, 0.3832, 0.8915],
        [0.0534, 0.0000, 0.0645,  ..., 0.9583, 0.3889, 0.9130],
        [0.1177, 0.0645, 0.0000,  ..., 1.0189, 0.4121, 0.9488],
        ...,
        [0.9113, 0.9583, 1.0189,  ..., 0.0000, 0.7847, 0.5985],
        [0.3832, 0.3889, 0.4121,  ..., 0.7847, 0.0000, 0.5475],
        [0.8915, 0.9130, 0.9488,  ..., 0.5985, 0.5475, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.05395056, 12.18045978, 12.35625694, 12.3828957 , 12.32545637,
       12.27835996,  9.17905491, 12.24467041,  5.27178363, 13.56367857,
       11.83509372, 13.35013374, 13.04217487,  6.34059021,  6.12886024,
        8.30125201,  5.94293185,  5.5513072 ,  6.7593917 , 11.95743558,
        9.93553424, 12.52372629,  6.7593917 ,  9.4997987 ,  5.9510489 ,
       14.05633441, 12.52372629,  7.16058587, 10.57302546, 13.40850328,
        5.83560009, 12.38881635, 12.9089559 , 13.05148826,  9.50913849,
       13.36800073,  7.73692436,  6.7166297 ,  5.54776934, 13.36092624,
        8.43626114,  7.5092834 ,  9.44486609,  6.49538841,  9.94173708,
       11.9182956 ,  8.25828409,  7.92644746,  6.19707944,  9.76896904,
       13.97857591,  7.24677664, 13.88683436,  5.14371005, 11.72802491,
        6.69389431, 13.2430377 , 10.1979669 , 10.06850677, 10.24557047,
       13.6105595 , 11.2845608 , 12.27993098, 13.88683436,  9.61942445,
        4.91336994, 10.27572673,  7.66377554, 13.40024224,  6.05113217,
        4.78544875,  6.76791483,  9.97220799,  8.7273569 , 10.35464916,
        4.91336994, 12.52875793,  8.50268322, 10.31886066, 13.97779839,
       10.68031351,  6.10360073, 13.36092624,  9.48337063, 13.44593519,
       12.76408562, 11.00694688, 13.6829843 ,  9.94173708,  6.99097013,
        5.5748624 , 11.46176259,  9.41244865,  9.91668182, 13.56367857,
        9.04255655,  8.89301516, 11.59512194, 12.86648771, 13.36092624,
        8.44940661,  5.80860165,  8.67502348, 10.14436912, 11.75438006,
        9.12220271])}
{'P': tensor([[3.9502e-03, 3.0289e-01, 1.7764e-01,  ..., 1.6029e-07, 2.6879e-04,
         6.5830e-05],
        [1.0829e-05, 3.6003e-03, 3.0184e-01,  ..., 1.1227e-07, 4.3619e-04,
         8.4140e-05],
        [1.0718e-05, 2.4401e-05, 5.4995e-03,  ..., 1.4870e-07, 1.0476e-03,
         1.7265e-04],
        ...,
        [1.3380e-03, 6.4458e-04, 4.0608e-04,  ..., 4.8816e-02, 4.7217e-06,
         3.0478e-06],
        [1.0096e-06, 1.6889e-06, 2.5182e-06,  ..., 3.6171e-07, 2.9948e-02,
         8.2017e-04],
        [2.4270e-07, 4.3077e-07, 6.8469e-07,  ..., 4.7645e-08, 6.8680e-03,
         8.1387e-03]]), 'X': tensor([[ 10.6283,  -5.7253,   7.4878],
        [ 11.4703,  -4.2006,   7.4086],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.6165,  -3.8715,   6.4594],
        [ 12.1940,  -2.1881,   7.1499],
        [  1.3670,  -4.7990,   3.2603],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -4.4520,   7.1370,  -8.7867],
        [  0.9074,  -4.8139,   8.1674],
        [  8.1217,  -8.5024,  -0.6149],
        [  9.2692,   9.6370,   7.0746],
        [ -5.2589,   6.7024,  -5.5809],
        [ -7.2986,  -7.1225,  -3.0814],
        [  4.3377,   5.7570,   1.3712],
        [  6.1752,   2.5242,   3.5874],
        [ -3.2486,   7.5925,  -1.3719],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -4.5778,  -9.5013,  -5.0667],
        [  6.3550,   0.7129,   0.4917],
        [  8.6420,  -8.0580,   6.5781],
        [  7.9403,  -8.6455,  -8.7433],
        [  2.2437,  -4.6489,   6.1430],
        [ -8.7544,   2.6096,   3.3146],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.5318,  13.7231,   6.3692],
        [  2.2437,  -4.6489,   6.1430],
        [  5.1142,  -2.7230,  -4.1966],
        [  5.4426, -10.1181,  -5.1268],
        [ -8.1666,   3.7653,   3.6075],
        [ -0.6323,   7.9085,  -5.9544],
        [  4.4915,   5.5990,   7.0405],
        [  5.0617,  12.7957,   8.3544],
        [ -1.8823, -10.6558,  -3.8952],
        [ -6.4464,   5.8272,   7.5038],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -2.7572,   7.7194,   5.9580],
        [ -8.3062,  -5.6008,   0.0287],
        [ -8.0087,   4.0173,   9.1091],
        [  5.4336,  -2.2676,   5.6933],
        [ 12.4901,  -0.7052,   1.7315],
        [  4.8674,  12.8875,   6.7622],
        [ 11.2767,  -4.6006,   0.7868],
        [  6.2770,  -0.0379,   2.4195],
        [ -9.3419,   0.5972,  -1.2841],
        [  8.5198,  10.4091,  -8.6486],
        [ -8.0608,   3.9362,  -5.9567],
        [  1.5383, -11.0274,  -4.9666],
        [ -4.6393,   7.0461,   1.1624],
        [ 10.5955,   7.8824,  -3.7006],
        [  6.0140,  -1.0661,  -2.5040],
        [ -6.6895,   5.6051,   2.7899],
        [ -0.0254, -10.9932,   3.2200],
        [  6.0305,   3.0296,   0.1375],
        [  8.4691,  10.4571,  -5.0650],
        [  0.2732,  -4.7619,  -1.1577],
        [  8.8474,  10.0868,  -8.8644],
        [  2.3367,   7.1676,  -9.5263],
        [ -2.1533, -10.5778,   1.0112],
        [  0.6470, -11.0352,  -1.1931],
        [ -7.6525,  -6.6534,   5.1721],
        [  2.7558,  -4.4844,   0.0636],
        [  4.8653,   5.1694,  -4.0823],
        [  1.5662,  13.9375,   4.5142],
        [  5.2939,  -2.4784,  -4.5840],
        [  4.5057,  -3.3894,  -0.8160],
        [ 10.0619,   8.6627,  -9.2884],
        [  1.8766,   7.3694,   1.1887],
        [  2.6614,  -4.5192,  -9.3641],
        [  7.0642,  11.6104,  -9.8316],
        [  4.9452, -10.3187,  -7.8621],
        [ -1.6275, -10.7221,   3.2376],
        [  2.1379,  -4.6757,   6.4981],
        [  8.2154,  -8.4262,   8.0171],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.4378,  -1.8606,  -2.5668],
        [  5.1519, -10.2387,   9.8191],
        [  6.3124,   0.2057,  -0.4593],
        [ -2.0229, -10.6163,   8.6930],
        [  2.1477,   7.2550,   8.9688],
        [  0.7958,  -4.8109,  -8.2248],
        [  2.6970,   6.9826,   9.7350],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.9883,   3.1536,   6.2320],
        [ -5.6169,  -8.7844,   8.9057],
        [  0.6933,   7.7336,   2.1179],
        [  1.4489,   7.5256,   4.4529],
        [ -3.4488, -10.0934,   2.5195],
        [  6.0345,   3.0175,  -1.4424],
        [  5.9526,   3.2525,  -4.3821],
        [  1.4489,   7.5256,   4.4529],
        [  9.6816,   9.1536,  -5.8543],
        [  1.0000,   7.6591,  -4.5011],
        [ -3.6381,   7.4683,   7.3544],
        [ 10.1983,  -6.3430,  -0.3200],
        [  8.1043,  -8.5164,  -7.9290],
        [  5.1292,   4.8183,  -5.7997],
        [  9.9756,  -6.6330,  -3.3429],
        [ -3.9456,  -9.8542,   0.7466],
        [  2.1625,  -4.6697,   7.0658],
        [  7.1697,  11.5343,  -0.7276],
        [ 12.3759,  -1.3943,   6.4659],
        [ -5.3328,  -8.9992,   7.3440],
        [ 10.9916,   7.2133,  -0.9179],
        [ 11.6019,   5.9540,  -5.1715]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.0004e-02, 1.1939e-01,  ..., 7.2010e-01, 8.3040e-01,
         1.1705e+00],
        [7.0004e-02, 0.0000e+00, 4.9934e-02,  ..., 7.9001e-01, 8.0500e-01,
         1.1546e+00],
        [1.1939e-01, 4.9934e-02, 0.0000e+00,  ..., 8.3853e-01, 7.8117e-01,
         1.1364e+00],
        ...,
        [7.2010e-01, 7.9001e-01, 8.3853e-01,  ..., 3.0339e-04, 1.2738e+00,
         1.4919e+00],
        [8.3040e-01, 8.0500e-01, 7.8117e-01,  ..., 1.2738e+00, 0.0000e+00,
         3.7186e-01],
        [1.1705e+00, 1.1546e+00, 1.1364e+00,  ..., 1.4919e+00, 3.7186e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.07225469, 12.21532282, 12.30695098, 12.32408516, 12.24467041,
       12.38881635,  4.98989906, 12.4013153 ,  8.41171369,  4.89869205,
       11.75807252, 13.37121923,  8.51928531, 10.1979669 ,  7.20826219,
        6.67121748,  8.25828409, 10.06407839, 10.33712258, 10.54658597,
        6.3949043 , 11.8159254 , 11.73848065,  5.16205087,  9.13507321,
        9.37989631, 13.95472331,  5.16205087,  5.79391825, 11.4890861 ,
        8.99276564,  7.9337647 ,  7.17789666, 13.76047999, 10.8207297 ,
        8.68979052,  9.69792513,  8.19703512, 10.01804223,  8.95983586,
        5.88782913, 12.50996828, 13.77604293, 12.17901033,  6.27715105,
        9.36093562, 13.45124917,  8.97052115, 11.1341798 ,  8.43626114,
       13.20597932,  6.10774544,  8.7273569 , 10.99326577,  6.74872981,
       13.45642379,  4.76969364, 13.41713661,  7.53884104, 10.794755  ,
       11.05413545, 10.14044489,  5.26342203,  7.09887557, 14.02526505,
        5.84533345,  5.63824971, 13.27717982,  7.60463349,  5.24461068,
       13.59056613, 11.4424816 , 10.84493732,  5.14124612, 11.76830981,
        6.34059021,  9.61942445, 11.46176259,  6.31576264, 10.80728788,
        7.56617037,  4.8763135 ,  7.48538905, 12.10942412,  6.76791483,
       10.42667093,  7.76457095,  7.66377554, 10.66631784,  6.74686071,
        6.78325151,  7.66377554, 13.32374056,  7.72415048,  8.30729352,
       12.00995898, 11.75618513,  7.03733775, 11.97957425, 10.61472636,
        5.14607014, 13.58101794, 12.45418236, 10.4606265 , 13.1471313 ,
       13.04051744])}
{'P': tensor([[3.9806e-03, 2.2092e-01, 1.6147e-01,  ..., 6.0759e-06, 1.0240e-05,
         9.4872e-08],
        [2.9932e-05, 5.2832e-03, 1.1674e-01,  ..., 1.2821e-05, 9.4209e-06,
         9.7010e-08],
        [2.0160e-05, 7.6451e-05, 9.3894e-03,  ..., 4.9875e-05, 1.2738e-05,
         1.6496e-07],
        ...,
        [2.1494e-06, 2.1676e-06, 2.4931e-06,  ..., 4.7036e-01, 9.7727e-05,
         6.3716e-05],
        [1.5806e-02, 1.2142e-02, 1.2304e-02,  ..., 9.6819e-05, 1.5819e-01,
         4.9320e-04],
        [2.7161e-04, 1.5126e-04, 1.1647e-04,  ..., 4.0594e-07, 4.1935e-03,
         3.8592e-02]]), 'X': tensor([[ 10.6673,  -5.6649,  -4.1646],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.4426, -10.1181,  -5.1268],
        [ -0.3432,   7.8900,   7.1805],
        [  1.9312,  13.8649,   8.3563],
        [ -7.5720,   4.6310,   2.1717],
        [ -5.6792,  -8.7352,   4.0318],
        [ 11.4743,   6.2478,  -2.1100],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.9659,  -4.8144,  -0.9594],
        [ 10.8218,  -5.4177,   6.8573],
        [  9.1409,  -7.5789,   1.5814],
        [  5.9587,  12.3216,   9.1695],
        [ 12.3357,   3.6001,   0.7903],
        [ -5.1398,  -9.1367,   6.2074],
        [  6.3549,   0.7096,   6.9198],
        [  1.9204,   7.3518,   8.4781],
        [ 12.1379,   4.4125,  -3.8833],
        [  5.7151,  -1.7700,   3.1542],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -5.9225,  -8.5355,   2.4778],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.8835,  -4.3178,   9.3080],
        [ 11.4743,   6.2478,  -2.1100],
        [ -9.0604,   1.7707,   1.1066],
        [  2.8890,  -4.4316,   1.1600],
        [  0.1041, -11.0045,   2.4311],
        [ -6.3066,  -8.1936,   9.5822],
        [  5.1787,  -2.6383,  -4.2533],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.5123,  -0.5339,  -1.0728],
        [  9.9756,  -6.6330,  -3.3429],
        [ -8.8934,   2.2597,   3.5831],
        [  6.2902,   0.0451,   3.7116],
        [  8.7131,  10.2216,   7.7915],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.1787,  -2.6383,  -4.2533],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.4366,  -0.1189,  -2.6736],
        [  0.1047, -11.0046,   0.5188],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.4691,   2.8525,  -5.5110],
        [ -1.6275, -10.7221,   3.2376],
        [ -8.9178,   2.1936,   7.5388],
        [  5.8201,   3.5841,  -8.0609],
        [  9.0051,   9.9234,   5.4411],
        [  7.0664,  -9.2590,   8.0879],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 11.5083,  -4.1177,   0.2389],
        [  9.3836,   9.5075,  -5.9278],
        [  4.5491,  -3.3485,   6.8412],
        [ 11.9695,  -2.9407,  -6.3498],
        [  3.5284, -10.7484,   5.3347],
        [  0.1206,  -4.7363,   8.7103],
        [ -4.5778,  -9.5013,  -5.0667],
        [  4.6357,   5.4415,   7.9067],
        [  7.4132,  11.3526,   0.3322],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.4242, -10.1261,   3.9595],
        [  0.9659,  -4.8144,  -0.9594],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -8.4267,  -5.3697,   2.4172],
        [ -4.4503,   7.1378,   3.3751],
        [  6.3246,   1.6696,  -0.1544],
        [  2.3200,   7.1756,   1.1938],
        [  4.6231,   5.4556,   8.4973],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.9410,   6.8428,   5.4458],
        [  4.0362,  13.2397,  -7.0030],
        [ 11.1469,   6.9237,   1.6160],
        [  2.3889,  -4.6082,  -8.7291],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.7121,   6.9743,  -8.4202],
        [  1.1274,  14.0116,   8.1753],
        [  1.9312,  13.8649,   8.3563],
        [  2.6310, -10.9195,   3.8054],
        [ -9.0604,   1.7707,   1.1066],
        [  6.3569,   0.7591,   6.5011],
        [ -8.0608,   3.9362,  -5.9567],
        [ 12.2692,  -1.8882,   3.9253],
        [ -7.0120,   5.2825,   7.7762],
        [ -5.0943,  -9.1682,   0.5073],
        [  9.0424,  -7.6779,  -7.5286],
        [ 11.9019,  -3.1379,  -3.4534],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.7813,  -3.1139,   8.2691],
        [ 11.2209,  -4.7092,   2.6117],
        [ -5.0550,   6.8231,  -9.2502],
        [  2.5318,  13.7231,   6.3692],
        [  0.4996,   7.7738,  -4.4711],
        [  5.1861,   4.7361,  -3.7312],
        [ -8.3900,   3.3727,   3.1650],
        [  7.2340,  -9.1505,  -5.8321],
        [  4.3554,   5.7393,  -9.0766],
        [  7.5784,  -8.9141,   5.8410],
        [  0.7275,  14.0667,  -0.9815],
        [  1.4749,  -4.7892,  -0.8089],
        [ -5.6792,  -8.7352,   4.0318]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0664, 0.0995,  ..., 0.3119, 0.3025, 1.0851],
        [0.0664, 0.0000, 0.1028,  ..., 0.3395, 0.3534, 1.1515],
        [0.0995, 0.1028, 0.0000,  ..., 0.2381, 0.2715, 1.1202],
        ...,
        [0.3119, 0.3395, 0.2381,  ..., 0.0000, 0.1441, 1.0358],
        [0.3025, 0.3534, 0.2715,  ..., 0.1441, 0.0000, 0.8997],
        [1.0851, 1.1515, 1.1202,  ..., 1.0358, 0.8997, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.07818871, 12.18045978, 12.30858677, 12.35625694, 12.4217523 ,
       12.12448912, 11.4890861 ,  7.89745107, 13.99877413,  8.87587457,
       10.41909211, 13.06498554,  9.36093562,  4.91038884, 12.10222204,
       11.87412439, 13.68672855, 12.85032836, 10.48315436,  6.39438044,
        7.5984809 , 12.91505111,  5.98285319,  6.83515201, 10.06437598,
       10.38897563, 12.62576026,  9.87720102, 13.06498554,  9.23178204,
        5.29009836, 11.0050379 , 10.33958639,  5.81199422,  5.16611782,
       12.52372629, 11.97957425,  9.17595899,  6.29034958, 13.43127131,
        9.78285424,  5.81199422,  7.37902766,  9.43737523, 11.00508681,
        7.83486943, 12.79126785, 10.84493732,  9.18358562,  6.83515201,
       13.40024224, 11.64746882, 10.5575601 , 12.22276278, 13.35832725,
        5.64865173, 12.32545637, 11.31276109,  4.73784222, 10.54658597,
        7.14837514, 13.55868684,  9.62958973, 11.48734978,  4.91038884,
        8.8737556 , 10.00410625,  9.99211663,  8.41148722,  6.54127431,
        7.54127425,  7.15100303, 13.01187484,  9.69629   , 13.27994304,
        7.44805901, 13.84126197, 13.12218013,  5.19065439, 10.27716434,
        7.48311002, 14.05687815, 13.99877413, 11.23201619,  9.23178204,
        6.4020404 ,  8.97052115, 12.41366754,  8.77913428, 10.48840476,
       11.86239961, 12.30858677, 11.75618513,  5.70591986, 12.16901615,
        8.49161154, 13.95472331,  7.78980299,  7.02326515,  9.04255655,
       11.66453059,  7.20482224, 11.70016217, 14.08549802,  5.01114719,
       10.41909211])}
{'P': tensor([[3.6068e-03, 2.4350e-01, 2.1789e-01,  ..., 1.3444e-06, 9.0243e-08,
         9.4671e-08],
        [2.1420e-05, 6.4841e-03, 6.8630e-04,  ..., 3.4975e-06, 1.9643e-07,
         1.8272e-07],
        [2.2145e-05, 1.8770e-02, 4.6675e-03,  ..., 2.2791e-06, 1.1710e-07,
         1.1716e-07],
        ...,
        [7.9974e-06, 6.3508e-06, 6.5299e-06,  ..., 8.9051e-02, 6.5125e-03,
         1.4595e-02],
        [3.8266e-06, 2.6800e-06, 2.4435e-06,  ..., 3.3779e-05, 2.9477e-02,
         6.8547e-02],
        [5.0088e-05, 3.1147e-05, 3.0164e-05,  ..., 5.5554e-05, 1.0070e-03,
         9.1441e-02]]), 'X': tensor([[ 10.8218,  -5.4177,   6.8573],
        [ 11.6165,  -3.8715,   6.4594],
        [ 11.4703,  -4.2006,   7.4086],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.9936,  -2.8677,   6.0614],
        [ 11.9641,  -2.9568,   7.7211],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.2311,  -2.0445,   1.3545],
        [ -9.1907,  -3.3180,  -5.5943],
        [  9.5261,  -7.1675,   1.4206],
        [  3.7916,  -3.9524,  -7.9547],
        [  1.4749,  -4.7892,  -0.8089],
        [ -3.0004,   7.6607,  -7.2640],
        [  2.6451,  -4.5250,  -0.5565],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -5.1046,  -9.1610,   2.3418],
        [  5.1203,  -2.7151,   6.2110],
        [  5.1203,  -2.7151,   6.2110],
        [  0.9659,  -4.8144,  -0.9594],
        [ -6.7643,  -7.7372,  -9.6663],
        [  5.1292,   4.8183,  -5.7997],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.3373,  13.7721,  -0.2442],
        [  6.2535,  -0.1709,  -9.4102],
        [  7.9141,  -8.6656,   9.6948],
        [  4.5800,   5.5034,  -3.9432],
        [  2.3490,  -4.6199,  -4.2489],
        [ -9.3616,  -2.4749,   0.1793],
        [ 11.4743,   6.2478,  -2.1100],
        [ 11.3367,  -4.4806,   0.4643],
        [ 11.7418,   5.6070,  -1.4192],
        [  3.1271, -10.8343,   2.6442],
        [  2.6614,  -4.5192,  -9.3641],
        [  6.9284,  11.7061,  -0.3732],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.8930,  12.3592,  -8.9201],
        [ -1.5261,   7.8988,   9.3768],
        [  1.1006,   7.6318,  -2.4877],
        [ -8.0608,   3.9362,  -5.9567],
        [ 12.1379,   4.4125,  -3.8833],
        [ -9.4748,  -0.7765,  -5.6695],
        [  4.3377,   5.7570,   1.3712],
        [  0.9392,  -4.8143,  -3.4649],
        [  9.4516,  -7.2501,  -1.8868],
        [ -7.5401,   4.6718,   9.0033],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.5900,   2.9767,   7.4811],
        [ 10.4994,  -5.9193,  -7.5153],
        [  1.9241,   7.3503,   7.9004],
        [ -2.0229, -10.6163,   8.6930],
        [  1.3403,  -4.8011,  -1.1698],
        [  4.0642,  -3.7599,  -3.4565],
        [  1.4306,  13.9620,  -0.2075],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.2728, -11.0372,   9.8121],
        [  2.3376,  -4.6231,   9.8436],
        [  8.5093,  -8.1764,  -3.2381],
        [ -7.8849,  -6.3115,  -4.7479],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -6.0195,   6.1789,  -2.9496],
        [ 12.5154,   2.5072,  -9.3274],
        [ -1.5875,   7.8944,  -9.4286],
        [  0.1041, -11.0045,   2.4311],
        [ -3.3999, -10.1152,  -1.2088],
        [ 11.0373,   7.1299,  -8.5895],
        [ 12.4013,  -1.2597,  -7.6474],
        [  3.7237,  -3.9966,  -5.0376],
        [  5.1667,  -2.6543,   1.2879],
        [  1.1994,   7.6035,   2.4295],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -7.6650,   4.5092,   0.2520],
        [ -8.6229,   2.9068,   2.4429],
        [  5.6093,  -1.9705,   6.4807],
        [  3.2159, -10.8166,  -1.0406],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.3464,   0.5708,  -1.6397],
        [  4.2612, -10.5514,  -6.6268],
        [  5.0190,   4.9705,  -7.9752],
        [ -9.4117,   0.1160,   8.2574],
        [  6.3303,   1.6145,  -8.4083],
        [  3.1480,  -4.3165,  -7.8002],
        [  4.0642,  -3.7599,  -3.4565],
        [  1.3670,  -4.7990,   3.2603],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.3200,   7.1756,   1.1938],
        [  6.6319,  11.9066,   3.5165],
        [  9.9756,  -6.6330,  -3.3429],
        [ -3.5903,   7.4847,  -8.9112],
        [ 12.5789,   0.1722,   1.6470],
        [  4.5491,  -3.3485,   6.8412],
        [ -5.7347,  -8.6908,   9.2878],
        [  4.4049,   5.6892,  -4.4394],
        [ 12.4013,  -1.2597,  -7.6474],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.9883,   3.1536,   6.2320],
        [  2.7558,  -4.4844,   0.0636],
        [  4.5430,   5.5438,  -3.8669],
        [  6.7565,  11.8237,  -4.9882],
        [ 11.2209,  -4.7092,   2.6117],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.3616,  -2.4749,   0.1793]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0742, 0.0637,  ..., 0.4116, 1.1562, 1.0013],
        [0.0742, 0.0000, 0.0446,  ..., 0.4738, 1.1969, 1.0541],
        [0.0637, 0.0446, 0.0000,  ..., 0.4753, 1.2157, 1.0642],
        ...,
        [0.4116, 0.4738, 0.4753,  ..., 0.0000, 0.7921, 0.6012],
        [1.1562, 1.1969, 1.2157,  ..., 0.7921, 0.0000, 0.2749],
        [1.0013, 1.0541, 1.0642,  ..., 0.6012, 0.2749, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.10222204, 12.24467041, 12.21532282, 12.30695098, 12.33167382,
       12.32408516, 13.04156075, 12.40074901,  9.7712321 , 11.92133129,
        5.47702766,  5.01114719,  8.22728399,  5.24137909, 12.3828957 ,
       10.37698435, 10.46739073, 10.48721382,  5.79562691,  5.79562691,
        4.91038884, 10.27716434,  7.03733775, 12.05395056, 13.96905756,
        6.25586297, 11.73567759,  7.15990676,  5.18277858,  9.68323686,
       13.06498554, 12.18998195, 13.01187484, 11.27656488,  5.24461068,
       13.60275211,  5.48276038, 13.6922425 ,  8.04483495,  7.71075226,
        8.97052115, 12.91505111,  9.50654967,  7.20826219,  4.90505566,
       11.9120321 ,  8.87007495, 11.83075005,  9.09119351, 12.05300943,
        7.59795445, 10.80728788,  4.98461983,  5.5366698 , 14.03505624,
        9.75401449, 11.11038551,  5.1805345 , 11.80092578, 10.09979747,
        6.2134961 , 12.62576026,  9.50913849,  8.62631801, 12.76408562,
        8.05242591, 11.0050379 , 10.67132524, 13.1399181 , 12.46513646,
        5.46244988,  5.80860165,  7.69752262, 10.32715715,  8.89301516,
        9.09963894,  5.94535529, 11.2845608 , 10.794755  ,  9.36378438,
       11.37939695,  7.06372172,  9.41244865,  6.53290663,  5.34250225,
        5.5366698 ,  4.98989906,  9.78286733,  7.54127425, 13.62896593,
       11.97957425,  8.30125201, 12.58006038,  5.64865173, 10.412299  ,
        7.19513806, 12.46513646,  5.01114719,  6.76791483,  5.26342203,
        7.16747878, 13.61801526, 12.16901615,  8.62260327,  9.37296102,
        9.68323686])}
{'P': tensor([[6.3213e-03, 3.1108e-01, 1.4723e-01,  ..., 1.5464e-07, 1.2087e-07,
         2.6968e-04],
        [3.0800e-05, 1.2483e-02, 7.3105e-01,  ..., 2.5508e-07, 2.4658e-07,
         3.0137e-04],
        [8.8932e-05, 2.8474e-04, 6.0902e-02,  ..., 1.1368e-06, 1.2030e-06,
         8.7516e-04],
        ...,
        [7.9609e-05, 3.1096e-05, 1.7803e-05,  ..., 3.4090e-02, 8.7633e-05,
         1.1436e-04],
        [2.6905e-05, 1.2806e-05, 8.1324e-06,  ..., 1.4886e-01, 1.7494e-01,
         2.3036e-04],
        [5.5307e-04, 9.3083e-04, 1.1095e-03,  ..., 9.6190e-05, 2.1751e-04,
         1.9660e-01]]), 'X': tensor([[ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0955, 0.1579,  ..., 0.9914, 1.1222, 0.6506],
        [0.0955, 0.0000, 0.0625,  ..., 1.0868, 1.2149, 0.6923],
        [0.1579, 0.0625, 0.0000,  ..., 1.1490, 1.2767, 0.7292],
        ...,
        [0.9914, 1.0868, 1.1490,  ..., 0.0000, 0.3345, 0.9823],
        [1.1222, 1.2149, 1.2767,  ..., 0.3345, 0.0000, 0.8991],
        [0.6506, 0.6923, 0.7292,  ..., 0.9823, 0.8991, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.10942412, 12.30858677, 12.4217523 , 12.35625694, 12.1460931 ,
       12.4013153 , 13.77222752,  5.91408504, 12.9089559 , 13.15525803,
        6.07992623,  7.33350642,  9.50913849, 13.38091121, 10.33712258,
        5.41348121,  6.5618968 ,  5.63824971,  5.08372149, 10.30972193,
        7.88677538,  6.56833864, 12.88270018, 12.35421461, 11.04629471,
       11.33837396, 10.38897563,  7.61088125,  9.37989631, 11.75618513,
        6.38399789,  9.7751189 ,  9.96152788, 13.62896593,  4.92497514,
        9.60412735,  9.66977802,  6.34059021,  5.47866221,  9.36093562,
        5.16077764,  6.383897  , 13.09426142,  9.99211663,  9.54154096,
       12.05395056,  7.50035355,  6.65991877, 11.63765273, 13.59480906,
       13.05148826,  9.93553424, 13.66991027,  9.48337063,  7.70897675,
        8.8580671 ,  9.17016857,  6.3949043 , 10.84493732,  7.88677538,
        4.78544875, 12.1460931 ,  7.5092834 ,  7.50500064, 13.84126197,
       10.6910768 , 11.4424816 , 11.83075005, 10.82106505, 10.67132524,
        7.48311002, 10.35464916,  9.15253344, 11.76830981,  7.83486943,
       10.65929968,  7.53063885,  6.74872981,  9.66977802,  9.36256096,
        9.25336371,  8.04664886,  7.39004226, 11.62158554,  8.85572292,
        4.87083348,  7.07902036, 11.81423161, 13.32613445, 13.26209545,
       12.42554808, 11.63724792,  9.09119351,  5.91854397, 11.38774149,
        5.14124612, 11.83075005,  8.40445872, 10.75443106,  8.31625171,
       10.7506423 , 11.24434374, 13.59673406, 10.04916678,  9.49511524,
        6.29821673])}
{'P': tensor([[3.6752e-03, 2.4081e-01, 1.7615e-01,  ..., 3.7868e-07, 2.7237e-07,
         7.6903e-07],
        [9.4028e-06, 3.3083e-03, 2.2722e-01,  ..., 4.4331e-07, 3.9385e-07,
         1.0939e-06],
        [2.4381e-05, 8.1796e-05, 1.3438e-02,  ..., 2.1076e-06, 1.6733e-06,
         5.0177e-06],
        ...,
        [1.1444e-05, 7.3633e-06, 6.1683e-06,  ..., 6.5539e-02, 8.2195e-06,
         1.7056e-04],
        [7.2766e-07, 7.2282e-07, 6.3577e-07,  ..., 1.2507e-04, 8.2613e-02,
         7.8544e-03],
        [2.0137e-06, 1.6964e-06, 1.4652e-06,  ..., 2.7116e-03, 5.3358e-04,
         4.5320e-02]]), 'X': tensor([[ 10.9162,  -5.2599,  -7.0812],
        [ 11.7726,  -3.4876,  -6.7402],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 12.1751,  -2.2592,  -5.7751],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.2675,  12.1379,  -8.9720],
        [ -8.9033,   2.2329,  -8.8912],
        [  0.1532,  -4.7422,  -7.0145],
        [  6.9284,  11.7061,  -0.3732],
        [ -8.6603,   2.8250,   5.9181],
        [ -8.1426,   3.8048,   9.3542],
        [ -3.5173, -10.0623,   4.8280],
        [  7.9403,  -8.6455,  -8.7433],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.0588,  -6.0341,  -2.3006],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.9003,  12.3550,   2.1917],
        [ -4.8751,  -9.3148,   4.0406],
        [ -8.4887,  -5.2448,  -0.6585],
        [  0.8940, -11.0402,  -0.9783],
        [ 11.2767,  -4.6006,   0.7868],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -8.3653,   3.4184,   9.3454],
        [ -4.7389,   6.9953,   3.7638],
        [  5.3804,  -2.3504,   7.2744],
        [  9.4925,   9.3811,   8.9842],
        [ 12.1940,  -2.1881,   7.1499],
        [ -0.4341, -10.9473,   7.4509],
        [  7.5064,  -8.9651,  -0.0994],
        [  1.6308,  13.9254,  -2.0984],
        [ -3.4488, -10.0934,   2.5195],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.4426, -10.1181,  -5.1268],
        [ -2.0445,   7.8469,  -0.5857],
        [ -9.3705,   0.4208,  -3.5815],
        [ -7.4808,   4.7462,   2.0266],
        [  5.8836,   3.4315,  -4.0327],
        [  5.4867,  12.5816,   2.9549],
        [ -1.9646,   7.8571,   3.3817],
        [  9.3836,   9.5075,  -5.9278],
        [ -4.4503,   7.1378,   3.3751],
        [ -3.4488, -10.0934,   2.5195],
        [ 11.5568,   6.0602,   9.3244],
        [  5.2188,  -2.5839,   2.1833],
        [ 10.6673,  -5.6649,  -4.1646],
        [  2.5636, -10.9293,   9.6519],
        [  3.0719, -10.8449,  -4.1673],
        [ -9.1459,   1.4780,  -4.2403],
        [  3.9591, -10.6391,  -6.2557],
        [ -0.5385,   7.9037,   6.3830],
        [  2.8890,  -4.4316,   1.1600],
        [  2.6451,  -4.5250,  -0.5565],
        [  0.3724,   7.7974,  -6.8722],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.8450,  -9.9356,   3.0767],
        [  5.4429,  -2.2529,   9.7306],
        [  0.6470, -11.0352,  -1.1931],
        [  5.5294,  -2.1106,   9.0626],
        [ -1.7290,   7.8825,  -3.3831],
        [  4.4915,   5.5990,   7.0405],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.0574,   1.7801,   8.8986],
        [ 12.5199,  -0.4711,   3.0436],
        [ -1.5408,   7.8978,   5.3470],
        [  7.3724,  -9.0577,  -8.9843],
        [ 11.4703,  -4.2006,   7.4086],
        [ -2.0445,   7.8469,  -0.5857],
        [ -6.8618,  -7.6322,   3.7027],
        [ -5.6651,   6.4380,   1.3660],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -3.5173,   7.5091,  -8.9585],
        [  9.1409,  -7.5789,   1.5814],
        [  3.2809,  13.5061,  -4.0199],
        [ -8.0608,   3.9362,  -5.9567],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.8474,   6.8979,   6.6457],
        [  6.3413,   0.4883,  -6.8416],
        [ -6.6895,   5.6051,   2.7899],
        [ -2.6069, -10.4298,  -6.1995],
        [ -0.1963,   7.8765,  -1.1371],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.1793,   4.7461,   0.3899],
        [  2.0547,   7.2957,  -1.9343],
        [  4.3523, -10.5232,   8.2138],
        [  6.3589,   0.8239,  -8.4230],
        [  0.1382, -11.0073,   1.0026],
        [  0.3493,  -4.7727,  -6.5682],
        [  9.0652,   9.8597,  -3.5949],
        [ -0.7599, -10.8995,  -6.0366],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.3360,   0.4258,  -9.2509],
        [  3.5284, -10.7484,   5.3347],
        [  4.5503,   5.5359,   7.4735],
        [  7.1697,  11.5343,  -0.7276],
        [  2.3589,  13.7668,  -3.6501],
        [  5.6314,  -1.9302,  -9.6543],
        [ -7.5603,   4.6459,   8.2002],
        [  3.0719, -10.8449,  -4.1673],
        [  1.6232,  -4.7717,  -2.5655],
        [ -8.6112,   2.9319,  -7.9943],
        [ -3.4986,   7.5152,   9.0712],
        [ -5.6651,   6.4380,   1.3660]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0566, 0.0971,  ..., 0.7812, 1.1603, 0.8068],
        [0.0566, 0.0000, 0.0422,  ..., 0.8360, 1.1695, 0.8334],
        [0.0971, 0.0422, 0.0000,  ..., 0.8779, 1.1927, 0.8669],
        ...,
        [0.7812, 0.8360, 0.8779,  ..., 0.0000, 1.1190, 0.6564],
        [1.1603, 1.1695, 1.1927,  ..., 1.1190, 0.0000, 0.4691],
        [0.8068, 0.8334, 0.8669,  ..., 0.6564, 0.4691, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.11735334, 12.27835996, 12.35421461, 12.32545637, 12.42554808,
       12.3828957 , 11.57249984, 13.66051884,  9.17905491,  4.74468406,
       13.60275211,  9.10945678,  8.98765644, 10.65929968, 11.73848065,
       12.1494023 , 10.0674887 , 12.32545637, 13.69162795, 10.51340829,
        9.97822473, 11.07637531, 12.17901033, 12.04924159,  9.0368458 ,
        8.44940661,  5.87132718, 13.34587027, 12.38881635, 10.95593729,
       11.69264243, 14.02058621, 10.66631784,  9.77777346, 11.4890861 ,
        8.10886021,  9.37989631,  8.859416  ,  6.81117802, 13.72594022,
        8.09899874, 13.35832725,  8.41148722, 10.66631784, 13.0493368 ,
        5.82345867, 12.07818871, 11.22596994, 11.27160035,  9.26455795,
       11.35182589,  7.92200342,  5.29009836,  5.24137909,  7.80625981,
       12.32408516, 11.52733581,  5.89074104, 11.05413545,  5.91854397,
        8.06990483,  7.17789666,  6.53290663,  9.23071698, 12.52875793,
        8.04664886, 11.67875855, 12.21532282,  8.10886021, 10.26327895,
        8.57560275,  8.37954853, 12.53285633,  8.29203356, 11.87412439,
       13.89886469,  8.97052115, 12.62576026,  9.78285424,  7.46248947,
        6.36004393,  8.7273569 , 10.7506423 ,  7.8788965 ,  9.37989631,
        7.02496223,  7.57946459, 11.38774149,  6.41202986, 11.00812996,
        4.78544875, 13.39372426, 10.92597196,  9.70108409,  6.35029274,
       11.31276109,  7.16598273, 13.58101794, 13.96746816,  5.95297459,
        8.8737556 , 11.27160035,  5.04028538,  9.09661212,  8.28968067,
        8.57560275])}
{'P': tensor([[5.0008e-03, 3.1611e-01, 1.6629e-01,  ..., 7.0601e-07, 3.0980e-03,
         1.0754e-05],
        [1.4635e-05, 6.1711e-03, 3.7615e-02,  ..., 1.3050e-06, 1.0574e-02,
         5.3836e-06],
        [1.1792e-05, 2.7660e-04, 5.0162e-03,  ..., 1.0258e-06, 1.3729e-02,
         3.6078e-06],
        ...,
        [4.1819e-06, 2.8994e-06, 2.3486e-06,  ..., 4.9498e-02, 1.0858e-06,
         7.1764e-06],
        [2.5091e-06, 4.7423e-06, 5.5787e-06,  ..., 5.9710e-05, 3.8981e-02,
         1.3287e-06],
        [2.4174e-01, 9.1319e-02, 6.4215e-02,  ..., 4.8767e-07, 8.9643e-04,
         5.1275e-03]]), 'X': tensor([[ 10.9599,  -5.1849,  -3.5526],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.1751,  -2.2592,  -5.7751],
        [  5.7802,  -1.6363,  -6.2782],
        [  0.3231,  -4.7691,  -1.6120],
        [ -3.9508,  -9.8515,   1.7923],
        [ -9.3739,   0.3982,  -4.2564],
        [ -7.0120,   5.2825,   7.7762],
        [  2.4697,  -4.5836,   1.4593],
        [  4.6992,   5.3690,   3.0935],
        [  4.0587, -10.6112,   6.2273],
        [  4.6231,   5.4556,   8.4973],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.0039,   7.3173,  -1.5885],
        [  7.0171,  11.6439,  -5.0178],
        [  6.0031,  12.2958,  -6.5992],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.3858,   4.4260,  -9.6769],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.5491,  -3.3485,   6.8412],
        [  7.3724,  -9.0577,  -8.9843],
        [  1.2273,  -4.8081,   9.8129],
        [ -6.3306,   5.9272,   5.6935],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -7.4887,   4.7364,   8.9074],
        [  5.8450,  -9.9356,   3.0767],
        [  6.3048,   0.1469,  -9.1691],
        [  0.7145,  -4.8071,  -2.3766],
        [ -0.5739,   7.9056,   0.6382],
        [ -8.5149,  -5.1905,   1.5239],
        [ -3.4100,   7.5436,  -7.6980],
        [  9.1620,  -7.5573,  -9.9882],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.4157,   0.0821,  -5.0025],
        [  1.3339,   7.5627,  -0.9034],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.8836,   3.4315,  -4.0327],
        [  2.5318,  13.7231,   6.3692],
        [ -1.5451,   7.8975,   7.6104],
        [  2.4932,   7.0903,  -5.4214],
        [ 12.4837,   2.7510,   0.4449],
        [  6.2242,  -0.3171,   7.4978],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.3205,   1.7068,  -3.8619],
        [  5.2608,  12.6977,   4.9166],
        [ -7.8359,   4.2735,   7.2943],
        [ -9.1658,  -3.4155,  -2.8880],
        [  1.8766,   7.3694,   1.1887],
        [  1.8447,  -4.7372,   8.1235],
        [  9.9756,  -6.6330,  -3.3429],
        [  3.5284, -10.7484,   5.3347],
        [ -1.5451,   7.8975,   7.6104],
        [  5.1203,  -2.7151,   6.2110],
        [  6.3549,   0.7096,   6.9198],
        [ 12.2921,  -1.7902,  -4.0870],
        [  1.7659,   7.4127,   4.7871],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.5506,  -5.1154,   6.3423],
        [  1.6232,  -4.7717,  -2.5655],
        [  5.6928,  12.4710,   3.6001],
        [  7.8425,  -8.7202,   6.4530],
        [  3.1064,  13.5607,   6.7605],
        [ 12.2592,   3.9425,   2.0383],
        [  2.2076,  13.8032,  -1.5099],
        [  7.0515,  11.6194,   8.4289],
        [  3.4604,  -4.1545,   1.8091],
        [ -5.1377,   6.7751,  -0.4438],
        [  0.5600, -11.0321,  -0.8660],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.0506,   4.9277,  -9.5094],
        [ -7.7842,   4.3466,  -1.7548],
        [  2.3889,  -4.6082,  -8.7291],
        [  2.0729,   7.2878,   7.0880],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.1886,  -2.6250,   6.4677],
        [  4.0642,  -3.7599,  -3.4565],
        [  4.5340,  -3.3629,   8.9652],
        [  2.4572,   7.1085,   7.7376],
        [  5.7151,  -1.7700,   3.1542],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.1579,  -0.5961,  -6.5595],
        [  6.3211,   1.7017,  -0.2164],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3305,   1.6119,   1.1538],
        [ -8.7078,   2.7182,   1.8146],
        [ 12.2435,   4.0076,   8.1257],
        [  5.3860,  -2.3418,   1.0826],
        [  1.2728, -11.0372,   9.8121],
        [ 10.9162,  -5.2599,  -7.0812],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.3542,   0.6934,  -3.3840],
        [  2.7420,  -4.4896,  -3.7139],
        [ -0.8455, -10.8853,  -6.7400],
        [ -2.7572,   7.7194,   5.9580],
        [  4.1297,  -3.7098,  -3.9838],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.2767,  -4.6006,   0.7868],
        [ -7.4807,   4.7463,   2.0616],
        [ 11.2939,   6.6325,  -5.8311],
        [  9.3424,  -7.3685,  -3.6807]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0674, 0.1120,  ..., 0.7620, 0.3519, 0.1224],
        [0.0674, 0.0000, 0.0782,  ..., 0.8074, 0.2992, 0.1894],
        [0.1120, 0.0782, 0.0000,  ..., 0.8734, 0.2417, 0.2233],
        ...,
        [0.7620, 0.8074, 0.8734,  ..., 0.0000, 1.1062, 0.6750],
        [0.3519, 0.2992, 0.2417,  ..., 1.1062, 0.0000, 0.4643],
        [0.1224, 0.1894, 0.2233,  ..., 0.6750, 0.4643, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.12448912, 12.30858677, 12.35625694, 12.4217523 , 12.49317563,
       12.3828957 ,  6.00732193,  4.78002665, 10.61417181,  9.38232121,
        8.77913428,  5.20659892,  7.13501264, 11.36089355,  7.15100303,
        9.88310614, 12.17901033,  7.58668144, 13.59480906, 13.6829843 ,
        9.64157461,  6.97106912,  5.24137909,  5.64865173, 11.67875855,
        4.96231532,  8.67227386,  9.70108409,  8.86082124, 11.52733581,
        6.30647746,  4.85995002,  7.92644746,  9.97220799,  8.27853076,
       11.87665501, 10.36844813,  9.20340031,  9.41605684,  7.67939546,
        5.95297459,  6.81117802, 13.95472331,  8.04718433,  7.5158489 ,
       12.78327037,  6.2322769 , 10.27716434,  6.54693987, 13.744387  ,
        8.92549752,  9.78147542,  7.60463349,  5.08372149, 11.97957425,
       11.31276109,  8.04718433,  5.79562691,  6.39438044, 12.4217523 ,
        7.62011734,  7.822854  ,  9.96391045,  5.04028538, 13.70893492,
       11.72802491, 13.91198191, 12.87752052, 13.97857591, 13.59170809,
        5.40689332,  8.50278896, 11.04629471, 12.07225469,  7.05627461,
        8.91551231,  5.19065439,  7.57685908,  9.79039371,  5.81481044,
        5.5366698 ,  5.64500512,  7.52117132,  5.98285319,  5.83560009,
        9.20340031,  6.18668463,  6.54615887,  5.76165934,  6.53250937,
        9.12220271, 12.88270018,  5.87305005, 11.11038551, 12.11735334,
        4.978704  ,  6.39187285,  5.26067019, 10.91805683,  8.19703512,
        5.5513072 , 14.10796466, 12.17901033,  8.85939626, 13.09737053,
       11.89855267])}
{'P': tensor([[3.6942e-03, 2.7139e-01, 1.8658e-01,  ..., 2.7890e-04, 1.8118e-03,
         9.8261e-06],
        [9.9762e-06, 3.5216e-03, 1.8962e-01,  ..., 7.6623e-05, 4.0397e-03,
         1.5359e-05],
        [1.2310e-05, 5.2479e-05, 5.0930e-03,  ..., 7.8863e-05, 7.0838e-03,
         2.8381e-05],
        ...,
        [1.7414e-03, 6.7021e-03, 1.1831e-02,  ..., 6.6995e-02, 4.9308e-03,
         7.3997e-04],
        [1.7928e-06, 2.9325e-06, 3.4776e-06,  ..., 1.5530e-06, 3.3539e-02,
         5.5663e-04],
        [3.3888e-06, 3.7626e-06, 4.1258e-06,  ..., 6.7659e-06, 1.2680e-05,
         6.6687e-02]]), 'X': tensor([[ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0519, 0.0768,  ..., 0.5471, 0.3337, 0.7245],
        [0.0519, 0.0000, 0.0300,  ..., 0.5492, 0.2849, 0.7375],
        [0.0768, 0.0300, 0.0000,  ..., 0.5315, 0.2730, 0.7248],
        ...,
        [0.5471, 0.5492, 0.5315,  ..., 0.0000, 0.7230, 0.2268],
        [0.3337, 0.2849, 0.2730,  ..., 0.7230, 0.0000, 0.9399],
        [0.7245, 0.7375, 0.7248,  ..., 0.2268, 0.9399, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.12641969, 12.27835996, 12.32545637, 12.35421461, 12.42554808,
       12.3828957 , 13.59480906, 10.7506423 , 12.67576792,  6.36004393,
       10.35093485, 12.49317563,  7.96575782,  6.21275016,  5.64464807,
       11.66453059,  7.48311002, 12.07225469, 13.59673406, 11.8159254 ,
        5.0209005 ,  7.39004226, 11.83570132,  4.74468406, 11.63765273,
        9.51074763,  5.54776934, 10.57302546,  8.68979052, 10.86693708,
       11.83075005,  8.50278896, 13.40024224,  5.05774295,  6.76791483,
        6.4020404 ,  9.58072286,  8.68832398, 12.41366754,  7.54127425,
        8.24808832, 13.50484539, 13.61801526,  6.32572323, 11.18839169,
        9.19837792,  6.89245352,  5.41348121,  6.67218687,  8.41148722,
       12.49317563, 14.02058621, 10.06407839, 14.02058621,  7.16598273,
        5.29009836,  6.55800267,  5.05963217, 11.85890664,  8.50268322,
       11.83075005,  4.83371341,  9.95552626, 11.56146873,  7.80625981,
        8.53501777,  5.94535529,  9.4997987 ,  4.75303127,  4.75303127,
       11.28093507, 11.33837396,  5.36061974,  7.89745107, 13.98691426,
       12.41012046, 13.59170809,  9.63742669, 11.46176259,  9.96391045,
       10.63038565, 13.91198191, 12.07225469,  4.85995002,  8.85572292,
        7.15100303, 10.86693708,  8.05668609,  9.79039371, 13.00380624,
        7.81509631,  5.19065439,  9.18690018, 13.1471313 ,  8.83341608,
       10.23946247, 11.21703654,  6.7166297 ,  8.1137452 ,  4.85603886,
       10.6910768 , 12.05300943, 13.01728313, 12.18998195, 13.14777433,
        7.76457095])}
{'P': tensor([[4.5307e-03, 2.6148e-01, 2.1819e-01,  ..., 8.7113e-07, 3.4935e-08,
         1.4522e-06],
        [5.4263e-05, 1.5006e-02, 6.2184e-02,  ..., 3.9588e-06, 9.8486e-08,
         2.0516e-06],
        [4.6112e-05, 1.2559e-03, 1.6877e-02,  ..., 4.8572e-06, 1.4301e-07,
         2.7646e-06],
        ...,
        [4.8625e-06, 3.5728e-06, 3.2277e-06,  ..., 4.3143e-02, 2.2358e-04,
         9.1000e-07],
        [4.1394e-06, 2.1614e-06, 2.6060e-06,  ..., 2.6309e-06, 2.0670e-02,
         1.3664e-04],
        [3.3994e-05, 3.2868e-05, 6.4970e-05,  ..., 1.1293e-07, 4.1088e-07,
         1.0316e-02]]), 'X': tensor([[ 11.0275,  -5.0666,   8.4154],
        [ 11.7796,  -3.4696,   9.0088],
        [ 11.9641,  -2.9568,   7.7211],
        [ 12.1940,  -2.1881,   7.1499],
        [ 11.8951,  -3.1570,   7.0742],
        [ 12.2589,  -1.9312,   9.8657],
        [  4.7156,  -3.1831,  -8.0224],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.4306,  13.9620,  -0.2075],
        [ -2.9497, -10.3028,  -7.3479],
        [  7.8425,  -8.7202,   6.4530],
        [  5.1292,   4.8183,  -5.7997],
        [  5.5626,  -2.0536,   8.4433],
        [  3.4009,  -4.1874,  -5.9751],
        [  1.0093,   7.6567,   2.3526],
        [ -6.1899,  -8.3011,  -7.9269],
        [  7.1441,  11.5529,   2.3232],
        [  7.3592,  11.3937,   6.2407],
        [  4.0362,  13.2397,  -7.0030],
        [ -0.0254, -10.9932,   3.2200],
        [  5.8450,  -9.9356,   3.0767],
        [ 11.5083,  -4.1177,   0.2389],
        [ -7.3554,   4.8985,  -0.6665],
        [ -8.6050,   2.9451,  -0.7012],
        [  7.1441,  11.5529,   2.3232],
        [  4.8653,   5.1694,  -4.0823],
        [  5.4336,  -2.2676,   5.6933],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 11.1085,  -4.9204,   0.0736],
        [ -2.0445,   7.8469,  -0.5857],
        [  9.4554,   9.4245,   9.8800],
        [ -9.4159,  -2.0739,   9.6841],
        [  6.8093,  -9.4178,   8.4244],
        [ 12.2592,   3.9425,   2.0383],
        [  0.6952,  -4.8060,   8.8976],
        [  3.4401,  13.4540,   4.6576],
        [ 11.8951,  -3.1570,   7.0742],
        [ -1.3940, -10.7772,   6.7835],
        [ 10.6673,  -5.6649,  -4.1646],
        [  4.4633,   5.6287,  -0.7754],
        [ -3.2486,   7.5925,  -1.3719],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.3076,   0.7852,  -6.7010],
        [  3.2990,  -4.2415,  -7.4857],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.2707,   0.9661,   4.4863],
        [ -8.5607,  -5.0938,  -5.4531],
        [  9.5261,  -7.1675,   1.4206],
        [  9.4554,   9.4245,   9.8800],
        [ 11.6165,  -3.8715,   6.4594],
        [  0.7275,  14.0667,  -0.9815],
        [  2.8602,   6.8905,   5.8779],
        [  6.3236,   0.3011,  -4.6668],
        [  2.1504,  -4.6726,   8.7908],
        [ 11.5083,  -4.1177,   0.2389],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.4773,  -1.0024,   5.2599],
        [  0.5600, -11.0321,  -0.8660],
        [  2.4932,   7.0903,  -5.4214],
        [ -6.1914,  -8.2997,   8.5380],
        [ 11.4743,   6.2478,  -2.1100],
        [ -7.0253,  -7.4493,  -1.5978],
        [  2.0039,   7.3173,  -1.5885],
        [  6.5428,  -9.5728,  -2.9897],
        [  0.7685,  -4.8098,  -2.9310],
        [ -7.5603,   4.6459,   8.2002],
        [  6.3550,   0.7129,   0.4917],
        [  2.2076,  13.8032,  -1.5099],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -9.2301,  -3.1537,  -2.7477],
        [  2.6321,  -4.5296,  -2.5159],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.8051, -10.3703,   0.5268],
        [ -5.6174,   6.4708,  -1.7339],
        [ -7.4807,   4.7463,   2.0616],
        [ -5.7347,  -8.6908,   9.2878],
        [ 11.1469,   6.9237,   1.6160],
        [ -7.0825,   5.2073,  -5.0497],
        [ -6.0844,  -8.3955,   0.7705],
        [ 11.0275,  -5.0666,   8.4154],
        [ -9.1706,   1.3863,   8.4032],
        [  3.3167,   6.6022,   6.3554],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.4117,   0.1160,   8.2574],
        [  6.2902,   0.0451,   3.7116],
        [  2.3589,  13.7668,  -3.6501],
        [ -0.5739,   7.9056,   0.6382],
        [ -3.5173, -10.0623,   4.8280],
        [  5.0506,   4.9277,  -9.5094],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.1685,  12.1979,  -6.2473],
        [  6.0560,  -0.9421,  -5.0334],
        [ -7.2515,   5.0195,   4.6434],
        [ -9.3444,   0.5821,   6.9101],
        [  2.9461,  -4.4077,  -0.4831],
        [ -9.3810,  -2.3455,  -9.5194],
        [  1.1006,   7.6318,  -2.4877],
        [  3.3094,  13.4969,  -3.6507],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.0899,  -4.6870,  -7.4260],
        [  6.2883,   0.0325,   5.2692],
        [  5.9526,   3.2525,  -4.3821],
        [ -7.5401,   4.6718,   9.0033],
        [ -8.8540,  -4.3971,  -5.2144],
        [  8.6271,  -8.0715,  -7.4813]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1123e-01, 9.4128e-02,  ..., 5.5445e-01, 1.4191e+00,
         1.2199e+00],
        [1.1123e-01, 3.0339e-04, 2.8896e-02,  ..., 6.6541e-01, 1.5018e+00,
         1.2454e+00],
        [9.4128e-02, 2.8896e-02, 0.0000e+00,  ..., 6.4486e-01, 1.4731e+00,
         1.2197e+00],
        ...,
        [5.5445e-01, 6.6541e-01, 6.4486e-01,  ..., 0.0000e+00, 1.0483e+00,
         1.2010e+00],
        [1.4191e+00, 1.5018e+00, 1.4731e+00,  ..., 1.0483e+00, 0.0000e+00,
         8.3349e-01],
        [1.2199e+00, 1.2454e+00, 1.2197e+00,  ..., 1.2010e+00, 8.3349e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.13568598, 12.27993098, 12.32408516, 12.38881635, 12.30695098,
       12.41012046,  5.68942572, 13.103709  ,  9.85384623, 14.03505624,
       10.7167285 , 11.72802491,  7.03733775,  5.92952538,  5.39450305,
        7.72291946, 10.35485331, 13.58333857, 13.56367857, 13.84126197,
       10.99326577, 11.52733581, 12.22276278,  8.83725081,  9.09502385,
       13.58333857,  7.09887557,  5.88782913, 10.32715715, 12.1494023 ,
        8.10886021, 13.35013374,  9.64157461, 11.62158554, 12.87752052,
        4.85603886, 13.88683436, 12.30695098, 10.86693708, 12.07818871,
        7.18354885,  8.25828409, 13.77222752,  9.34061938,  5.37343173,
       13.02818702,  9.32094055,  9.96152788, 11.92133129, 13.35013374,
       12.24467041, 14.08549802,  7.46053281,  6.33075682,  5.14371005,
       12.22276278,  9.79039371,  9.53015376, 11.04629471,  7.5158489 ,
       10.35464916, 13.06498554, 10.23946247,  7.58668144, 11.59512194,
        4.87083348,  8.8737556 ,  6.3949043 , 13.97857591, 12.05300943,
        9.75401449,  5.23878371, 12.05300943, 11.42947632,  8.56889738,
        8.85939626, 10.412299  , 13.12218013,  8.7907923 , 10.36844813,
       12.13568598,  9.27474386,  7.38844703, 12.4217523 ,  9.41244865,
        6.29034958, 13.96746816,  7.92644746, 10.65929968,  7.05627461,
        8.22728399, 13.66896739,  6.12886024,  8.81930525,  9.36256096,
        5.30158497,  9.66977802,  7.71075226, 13.89671314,  9.68324553,
        5.13182506,  6.28835101,  6.78325151,  8.87007495,  9.88571713,
       11.81423161])}
{'P': tensor([[7.7548e-03, 2.5028e-01, 2.2290e-01,  ..., 4.2736e-03, 2.0822e-06,
         9.1375e-05],
        [4.2328e-05, 9.3995e-03, 3.1076e-02,  ..., 1.6105e-02, 1.0202e-06,
         1.3257e-04],
        [1.7003e-05, 2.7929e-04, 8.2163e-03,  ..., 1.0569e-02, 7.5022e-07,
         2.8397e-04],
        ...,
        [1.4906e-06, 2.6609e-06, 4.0673e-06,  ..., 2.1872e-02, 1.3272e-07,
         4.9876e-04],
        [4.8224e-03, 2.0164e-03, 3.9666e-03,  ..., 4.8691e-05, 2.4962e-02,
         6.6198e-05],
        [8.2978e-06, 8.7456e-06, 1.3150e-05,  ..., 3.9693e-05, 1.1399e-05,
         7.9423e-02]]), 'X': tensor([[ 11.0891,  -4.9558,  -2.3178],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -9.4773,  -1.0024,   5.2599],
        [  0.9445,   7.6736,   0.5095],
        [  6.3551,   0.7149,   1.3344],
        [ -8.5149,  -5.1905,   1.5239],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.8994,   6.8675,  -9.5987],
        [ -2.7572,   7.7194,   5.9580],
        [ 11.1469,   6.9237,   1.6160],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.0604,   1.7707,   1.1066],
        [  7.5784,  -8.9141,   5.8410],
        [ -8.3535,  -5.5118,   9.8399],
        [  0.5175,  14.0910,   4.6540],
        [  0.6757, -11.0361,  -1.3047],
        [  6.3124,   0.2057,  -0.4593],
        [ -0.0282,   7.8575,  -1.1746],
        [  4.4290,  -3.4596,   1.4181],
        [  8.4691,  10.4571,  -5.0650],
        [ -7.8280,   4.2848,  -7.1955],
        [ -1.8367,   7.8718,  -8.6523],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.0870,  -4.6877,   9.6823],
        [ -3.5520,   7.4976,   9.1522],
        [  6.3124,   0.2057,  -0.4593],
        [ -9.2508,   1.0572,   2.9978],
        [  0.2732,  -4.7619,  -1.1577],
        [ -9.0604,   1.7707,   1.1066],
        [ -8.4267,  -5.3697,   2.4172],
        [  0.6952,  -4.8060,   8.8976],
        [ -6.7643,  -7.7372,  -9.6663],
        [  6.3497,   1.3680,  -1.9076],
        [ -2.5056,   7.7720,   5.5878],
        [  5.1142,  -2.7230,  -4.1966],
        [ 11.3118,   6.5957,  -4.5667],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -3.2053, -10.1993,  -3.8736],
        [ 11.7794,   5.5086,  -9.6929],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.1890,   4.7318,   0.9515],
        [  3.0719, -10.8449,  -4.1673],
        [ -5.9225,  -8.5355,   2.4778],
        [  4.7156,  -3.1831,  -8.0224],
        [ -4.9217,   6.8979,  -1.7667],
        [  4.6357,   5.4415,   7.9067],
        [ -9.0961,   1.6529,   3.1553],
        [  6.1989,  -9.7591,   0.8289],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.1203,  -2.7151,   6.2110],
        [ -2.2508, -10.5479,  -2.7235],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.5900,   2.9767,   7.4811],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -9.3616,  -2.4750,   3.7765],
        [ -5.1046,  -9.1610,   2.3418],
        [  4.5954,  13.0099,  -4.4439],
        [  5.6672,  -1.8631,   8.1020],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.1231,  -3.5739,   8.4708],
        [  3.1064,  13.5607,   6.7605],
        [  4.1348,   5.9514,  -0.8856],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3031,   1.8477,  -7.1706],
        [  0.5850,  -4.7982,   9.6639],
        [  5.3118,   4.5452,   4.2114],
        [  9.2692,   9.6370,   7.0746],
        [  1.1274,  14.0116,   8.1753],
        [ 12.2589,  -1.9312,   9.8657],
        [  8.5093,  -8.1764,  -3.2381],
        [ -1.5408,   7.8978,   5.3470],
        [  3.1892, -10.8220,  -6.8409],
        [  5.5424,  -2.0885,   7.2487],
        [  2.2437,  -4.6489,   6.1430],
        [  4.1637,  13.1898,  -1.9189],
        [ -3.3999, -10.1152,  -1.2088],
        [ -3.9700,   7.3453,  -4.1486],
        [ -3.0644,   7.6439,  -1.4322],
        [ 10.9931,   7.2107,   7.7933],
        [ 12.6000,   1.3839,  -3.2992],
        [  2.5128, -10.9365,   9.9645],
        [  4.0587, -10.6112,   6.2273],
        [  6.2883,   0.0325,   5.2692],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -7.9667,   4.0814,  -0.6397],
        [ 12.5717,   1.9266,   7.4903],
        [ -5.3179,  -9.0101,   9.4472],
        [  5.1519, -10.2387,   9.8191],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.3759,  -1.3943,   6.4659],
        [ -5.6174,   6.4708,  -1.7339],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -8.5900,   2.9767,   7.4811],
        [ 11.2847,  -4.5848,  -4.7503],
        [  6.0031,  12.2958,  -6.5992],
        [  2.3261,   7.1726,   6.2522],
        [ 12.3759,  -1.3943,   6.4659],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.7482, -10.3907,   5.9149],
        [  2.6613,   7.0021,   4.5783]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1217, 0.1318,  ..., 0.4148, 0.8826, 0.5888],
        [0.1217, 0.0000, 0.2019,  ..., 0.2992, 1.0039, 0.6986],
        [0.1318, 0.2019, 0.0000,  ..., 0.4354, 0.8449, 0.5072],
        ...,
        [0.4148, 0.2992, 0.4354,  ..., 0.0000, 1.2791, 0.9347],
        [0.8826, 1.0039, 0.8449,  ..., 1.2791, 0.0000, 0.4084],
        [0.5888, 0.6986, 0.5072,  ..., 0.9347, 0.4084, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.1460931 , 12.30858677, 12.4013153 , 12.4217523 , 12.52372629,
       12.25806559,  9.53015376,  7.73151309,  6.39520397,  9.97220799,
       10.30972193,  9.41605684,  7.4544866 ,  8.19703512, 13.12218013,
        9.26455795,  9.23178204, 11.70016217, 10.00802627, 14.10045797,
       11.05672136,  6.31576264,  7.8575648 ,  5.62005629, 13.45642379,
        8.92395381,  8.08320235,  9.20340031,  5.13125513,  8.29640613,
        6.31576264,  9.3109884 ,  4.76969364,  9.23178204,  9.99211663,
        4.85603886, 10.27716434,  6.49538841,  8.16584537,  5.79391825,
       13.09426142, 12.00995898, 10.6910768 , 13.00380624, 12.16901615,
        7.02253235, 11.27160035, 10.38897563,  5.68942572,  8.47371868,
        7.14837514,  9.24501829, 11.56146873,  6.27905292, 10.00794468,
        5.79562691, 10.78533838,  8.95983586,  9.09119351,  9.43737523,
        9.68324553, 10.48721382, 13.79762415,  5.96556248,  7.53063885,
        9.79814997, 13.91198191,  7.24677664,  6.27905292,  6.56833864,
        4.83371341,  6.99097013, 13.37121923, 14.05687815, 12.41012046,
       11.80092578,  8.04664886, 11.28215429,  5.92281142,  5.16205087,
       13.83139024, 10.67132524,  8.34949383,  8.23525433, 13.14690137,
       12.67576792, 11.22142195, 11.36089355,  6.28835101,  9.5014146 ,
        8.95133961, 12.71843462, 10.46238054, 11.46176259, 11.21703654,
       12.45418236,  8.56889738, 12.07818871,  9.09119351, 12.18045978,
       13.6829843 ,  7.5403822 , 12.45418236, 13.09737053, 11.42420508,
        7.49076899])}
{'P': tensor([[3.7584e-03, 3.8826e-01, 1.8196e-01,  ..., 6.0979e-05, 2.0186e-06,
         4.2803e-06],
        [3.8455e-05, 8.5559e-03, 5.9006e-01,  ..., 1.1709e-04, 3.0157e-06,
         6.3088e-06],
        [7.0110e-05, 1.0542e-04, 2.8530e-02,  ..., 2.9208e-04, 5.1180e-06,
         1.0578e-05],
        ...,
        [1.3377e-03, 1.7355e-03, 2.8240e-03,  ..., 1.2938e-01, 3.6936e-05,
         9.1721e-04],
        [3.0268e-02, 2.4677e-02, 1.4645e-02,  ..., 1.4010e-04, 5.6925e-02,
         1.0904e-04],
        [5.1173e-03, 6.2240e-03, 8.2756e-03,  ..., 1.7021e-02, 2.6950e-05,
         5.1823e-02]]), 'X': tensor([[ 11.1085,  -4.9204,   0.0736],
        [ 11.5083,  -4.1177,   0.2389],
        [ 12.0582,  -2.6640,   0.7132],
        [ 11.3367,  -4.4806,   0.4643],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 12.2328,  -2.0376,  -1.1234],
        [  5.3858,   4.4260,  -9.6769],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.6154,  -4.0640,  -7.9698],
        [  9.6614,   9.1784,  -1.2087],
        [ -6.3252,  -8.1760,  -1.0891],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3360,   0.4258,  -9.2509],
        [ -1.2454,   7.9127,  -0.5715],
        [ -5.9938,  -8.4745,   3.9735],
        [  2.2849, -10.9655,  -7.6542],
        [ -8.9624,  -4.0944,  -4.9622],
        [  5.1667,  -2.6543,   1.2879],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.3200,   0.2690,   1.5900],
        [ -9.1624,  -3.4286,   2.0647],
        [ -6.8618,  -7.6322,   3.7027],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.3357,   3.6001,   0.7903],
        [  0.9659,  -4.8144,  -0.9594],
        [  2.6613,   7.0021,   4.5783],
        [  6.3542,   0.6934,  -3.3840],
        [ 10.0995,  -6.4739,   2.0974],
        [  3.3839,  -4.1966,   8.3574],
        [ 11.2517,  -4.6495,   2.9800],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.7151,  -1.7700,   3.1542],
        [  3.4604,  -4.1545,   1.8091],
        [  6.2902,   0.0451,   3.7116],
        [  5.2199,   4.6861,  -1.9304],
        [  5.1861,   4.7361,  -3.7312],
        [  2.1504,  -4.6726,   8.7908],
        [  1.4489,   7.5256,   4.4529],
        [  5.1890,   4.7318,   0.9515],
        [  1.5246,  -4.7838,  -5.2095],
        [ -2.3317,   7.8034,  -3.6274],
        [ -5.5525,   6.5146,  -4.4984],
        [  3.6469,   6.3624,  -1.8414],
        [  8.5093,  -8.1764,  -3.2381],
        [  0.9036,   7.6840,   9.9237],
        [ -2.0840,   7.8415,  -7.8657],
        [  6.0031,  12.2958,  -6.5992],
        [  2.4639, -10.9431,  -7.3718],
        [  2.0899,  -4.6870,  -7.4260],
        [  2.3373,  13.7721,  -0.2442],
        [  7.5064,  -8.9651,  -0.0994],
        [  1.7903,  -4.7467,   0.1168],
        [  2.0729,   7.2878,   7.0880],
        [  2.6310, -10.9195,   3.8054],
        [  4.0362,  13.2397,  -7.0030],
        [ -9.4196,   0.0478,   9.4140],
        [  0.2435,   7.8191,  -6.4709],
        [ 12.2589,  -1.9312,   9.8657],
        [  8.7131,  10.2216,   7.7915],
        [ -2.4956, -10.4682,   3.5632],
        [  0.7145,  -4.8071,  -2.3766],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.2419,  -0.2313,   4.2440],
        [  2.3261,  13.7749,   5.9121],
        [ -7.4807,   4.7463,   2.0616],
        [ -1.5451,   7.8975,   7.6104],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -8.8400,   2.3991,   1.3457],
        [ 11.6634,   5.8050,   9.1590],
        [ -4.0868,   7.2981,   8.1874],
        [  6.2038,   2.4020,  -3.1692],
        [  2.7685, -10.8982,  -6.8928],
        [  2.2076,  13.8032,  -1.5099],
        [ -8.3062,  -5.6008,   0.0287],
        [ -7.6525,  -6.6534,   5.1721],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.2974,   0.8371,   2.4852],
        [ 11.9122,   5.1397,  -1.5498],
        [  2.5375,   7.0676,  -4.3490],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.3419,   0.5972,  -1.2841],
        [ -5.5525,   6.5146,  -4.4984],
        [ -9.4296,  -1.9458,   4.0833],
        [ -8.7078,   2.7182,   1.8146],
        [  5.1196,  -2.7160,   4.2705],
        [  0.6470, -11.0352,  -1.1931],
        [ 11.4703,  -4.2006,   7.4086],
        [  5.8323,  -1.5228,   5.1132],
        [  4.1537,   5.9340,  -6.0423],
        [  5.7963,  -1.6018,   1.8856],
        [ -6.3489,   5.9117,   6.5161],
        [  5.6672,  -1.8631,   8.1020],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.0965,  -5.9712,   4.3466],
        [  5.8325,  -1.5222,  -2.1341],
        [  0.9788,  -4.8145,  -3.8661],
        [  7.1697,  11.5343,  -0.7276],
        [ -6.6895,   5.6051,   2.7899],
        [  1.3705,  13.9723,   7.9266],
        [ 10.0400,   8.6922,  -6.2847],
        [  5.6022,  -1.9833,   8.3081],
        [  6.0467,  -9.8369,  -6.5816],
        [  7.9141,  -8.6656,   9.6948]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0477, 0.1074,  ..., 0.5439, 0.5369, 0.7243],
        [0.0477, 0.0000, 0.0613,  ..., 0.5207, 0.5722, 0.7213],
        [0.1074, 0.0613, 0.0000,  ..., 0.4782, 0.6299, 0.7015],
        ...,
        [0.5439, 0.5207, 0.4782,  ..., 0.0000, 1.0716, 0.3338],
        [0.5369, 0.5722, 0.6299,  ..., 1.0716, 0.0000, 1.1841],
        [0.7243, 0.7213, 0.7015,  ..., 0.3338, 1.1841, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.1494023 , 12.22276278, 12.3489324 , 12.18998195, 12.25806559,
       12.4013153 ,  6.97106912,  6.25586297,  5.43943791, 13.32613445,
       10.33712258, 11.62158554,  6.35029274,  8.01009083, 10.37998078,
       11.20100795,  9.85330646,  5.80860165, 12.05395056,  6.32572323,
        9.78285424, 10.26327895,  4.92497514, 12.87752052, 12.85032836,
        4.91038884,  7.49076899,  6.39187285, 11.99633819,  5.39098669,
       12.17451685, 12.41012046,  5.98285319,  5.40689332,  6.29034958,
        7.01475376,  7.02326515,  5.14371005,  7.66377554,  7.02253235,
        5.0209005 ,  8.14433542,  8.55982105,  7.33350642, 11.80092578,
        7.73692436,  8.1137452 , 13.6829843 , 11.21703654,  5.13182506,
       13.96905756, 11.69264243,  5.07306274,  7.57685908, 11.23201619,
       13.84126197,  9.41970031,  7.822854  , 12.41012046, 13.43127131,
       10.76154256,  4.85995002,  9.85384623,  6.24614905, 13.96987806,
        8.85939626,  8.04718433, 10.53368644,  9.15976746, 13.02818702,
        8.36445758,  6.65259775, 11.24434374, 13.97857591, 10.01804223,
       10.14044489, 10.80728788,  9.3349858 , 12.97370371,  7.5092834 ,
        6.01356004, 13.08414689,  9.36093562,  8.55982105,  9.6282763 ,
        9.12220271,  5.79543038, 11.05413545, 12.21532282,  6.02779201,
        7.24326992,  6.01356004,  8.67502348,  5.96556248,  9.95552626,
        4.9160962 , 10.06023665,  6.02789634,  4.91296205, 13.58101794,
        8.7273569 , 14.0393949 , 13.27994304,  5.94293185, 11.54672755,
       11.73567759])}
{'P': tensor([[3.8606e-03, 2.7357e-01, 2.7300e-01,  ..., 1.2844e-04, 1.1105e-06,
         9.4606e-06],
        [6.1036e-05, 2.3919e-02, 2.1511e-04,  ..., 1.6778e-03, 1.0472e-05,
         3.3914e-05],
        [3.6062e-05, 3.8511e-01, 7.7136e-03,  ..., 4.7307e-04, 3.2316e-06,
         1.5042e-05],
        ...,
        [3.0876e-07, 4.8844e-07, 4.2007e-07,  ..., 2.0126e-02, 2.2208e-03,
         4.2824e-06],
        [3.4379e-06, 3.0970e-06, 3.5040e-06,  ..., 2.3926e-05, 1.4881e-01,
         3.1107e-04],
        [1.1267e-03, 1.1202e-03, 1.3697e-03,  ..., 4.8855e-04, 1.9182e-04,
         4.8876e-02]]), 'X': tensor([[ 11.2209,  -4.7092,   2.6117],
        [ 12.0042,  -2.8350,   2.6864],
        [ 11.7349,  -3.5838,   2.1644],
        [ 11.7795,  -3.4697,   3.5624],
        [ 12.2311,  -2.0445,   1.3545],
        [ 12.2692,  -1.8882,   3.9253],
        [ -8.8400,   2.3991,   1.3457],
        [  6.0788,  -0.8710,  -6.8719],
        [  4.6992,   5.3690,   3.0935],
        [ -4.8665,  -9.3204,  -5.2564],
        [  8.5198,  10.4091,  -8.6486],
        [ -5.6169,  -8.7844,   8.9057],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 11.3367,  -4.4806,   0.4643],
        [ 11.1469,   6.9237,   1.6160],
        [  1.5662,  13.9375,   4.5142],
        [  9.0004,  -7.7195,   4.6945],
        [ -7.5720,   4.6310,   2.1717],
        [ -2.1533, -10.5778,   1.0112],
        [ 11.5198,  -4.0923,   5.5088],
        [  3.3633,  13.4794,   0.5193],
        [ 12.3759,  -1.3943,   6.4659],
        [ 11.6544,   5.8273,  -3.1375],
        [  6.1575,  12.2046,   3.3257],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.1292,   4.8183,  -5.7997],
        [  5.2199,   4.6861,  -1.9304],
        [ -7.4808,   4.7462,   2.0266],
        [ -2.1533, -10.5778,   1.0112],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.5149,  -5.1905,   1.5239],
        [  1.7218,  -4.7576,  -3.3617],
        [  5.6258,  -1.9404,  -6.0715],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.3301,   0.3638,  -7.5685],
        [  8.9284,  10.0036,  -6.9210],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.1398,  -9.1367,   6.2074],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -5.7347,  -8.6908,   9.2878],
        [  0.7145,  -4.8071,  -2.3766],
        [  5.1793,   4.7461,   0.3899],
        [  4.7813,  -3.1139,   8.2691],
        [  6.2804,   2.0019,   0.8607],
        [ -7.0253,  -7.4493,  -1.5978],
        [  1.6954,  -4.7616,   9.8198],
        [  1.0093,   7.6567,   2.3526],
        [ -5.6651,   6.4380,   1.3660],
        [ -1.3940, -10.7772,   6.7835],
        [  5.2939,  -2.4784,  -4.5840],
        [  2.1477,   7.2550,   8.9688],
        [  4.0078,  -3.8018,  -6.2168],
        [  5.1292,   4.8183,  -5.7997],
        [  3.2647, -10.8066,  -0.3768],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.1519, -10.2387,   9.8191],
        [  6.1066,  -0.7799,  -4.1746],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.5927,   0.4199,  -9.4997],
        [ -7.3333,   4.9246,  -7.9212],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.2275,  -3.6320,  -0.2714],
        [ -8.8745,   2.3097,   3.5600],
        [  2.9470,  13.6085,   2.9809],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -2.2188,   7.8218,   8.6064],
        [  5.1519, -10.2387,   9.8191],
        [ -4.2271,   7.2387,  -1.8316],
        [  4.7482, -10.3907,   5.9149],
        [ 11.2517,  -4.6495,   2.9800],
        [  6.3569,   0.7591,   6.5011],
        [ 12.4690,  -0.8528,  -9.3374],
        [  2.4639, -10.9431,  -7.3718],
        [ 11.2517,  -4.6495,   2.9800],
        [ -3.5903,   7.4847,  -8.9112],
        [  1.5653,  13.9377,   4.1773],
        [  6.3550,   0.7129,   0.4917],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.3889,  -4.6082,  -8.7291],
        [  5.6928,  12.4710,   3.6001],
        [ 12.5224,  -0.4497,   4.8448],
        [  8.1217,  -8.5024,  -0.6149],
        [  8.3718,  -8.2953,   8.4666],
        [  2.7558,  -4.4844,   0.0636],
        [  0.1460,   7.8340,  -9.4614],
        [  4.9152,  12.8653,  -1.4459],
        [  6.3535,   0.6792,  -7.5101],
        [  6.3119,   1.7797,   9.9186],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.1066,  -0.7799,  -4.1746],
        [  5.2608,  12.6977,   4.9166],
        [  3.1271, -10.8343,   2.6442],
        [  1.3670,  -4.7990,   3.2603],
        [ 11.9019,  -3.1379,  -3.4534],
        [  0.2571,  -4.7594,   1.5618],
        [ -9.1749,  -3.3803,   6.4629],
        [ -4.6393,   7.0461,   1.1624],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.6000,   1.3839,  -3.2992],
        [  9.4554,   9.4245,   9.8800],
        [  4.5340,  -3.3629,   8.9652],
        [  3.6703,   6.3443,  -5.4401],
        [  8.4691,  10.4571,  -5.0650],
        [ -3.5903,   7.4847,  -8.9112],
        [  3.4009,  -4.1874,  -5.9751]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0715, 0.0635,  ..., 0.7484, 0.8014, 0.5552],
        [0.0715, 0.0000, 0.0546,  ..., 0.7149, 0.8329, 0.5536],
        [0.0635, 0.0546, 0.0000,  ..., 0.6865, 0.7783, 0.5065],
        ...,
        [0.7484, 0.7149, 0.6865,  ..., 0.0000, 0.7074, 0.3431],
        [0.8014, 0.8329, 0.7783,  ..., 0.7074, 0.0000, 0.4130],
        [0.5552, 0.5536, 0.5065,  ..., 0.3431, 0.4130, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.16901615, 12.3344536 , 12.26997165, 12.27991991, 12.40074901,
       12.41366754,  9.15976746,  6.14086582,  7.13501264, 10.51437947,
       13.45124917, 10.42667093, 10.5575601 , 12.18998195, 13.12218013,
       14.02526505, 11.85743425,  8.87587457, 10.794755  , 12.22503754,
       13.89264822, 12.45418236, 13.03002338, 13.66991027,  4.98461983,
        7.03733775,  7.01475376,  8.859416  , 10.794755  ,  9.14973319,
        9.97220799,  5.05963217,  5.9510489 ,  4.85995002, 12.27993098,
        6.34059021, 13.40850328,  8.24808832, 10.48315436, 12.11735334,
       10.412299  ,  4.85995002,  7.02496223,  5.70591986,  6.59176018,
       10.23946247,  5.05444288,  7.72291946,  8.57560275, 10.86693708,
        5.84533345,  7.56617037,  5.52416368,  7.03733775, 11.28895489,
       12.32408516, 11.46176259,  6.1561646 , 13.79762415, 12.59969936,
        8.83341608, 12.17901033,  5.57341623,  9.17016857, 13.92390689,
        9.91668182,  8.1303945 , 11.46176259,  8.38252937, 11.42420508,
       12.17451685,  6.4020404 , 12.4980845 , 11.21703654, 12.17451685,
        8.30125201, 14.02532758,  6.3949043 , 10.24557047,  5.19065439,
       13.70893492, 12.53047489, 11.75807252, 11.785566  ,  5.26342203,
        7.83534987, 13.77222752,  6.38968357,  6.55800267,  8.83725081,
        6.1561646 , 13.744387  , 11.27656488,  4.98989906, 12.30858677,
        4.7663526 ,  9.77777346,  8.43626114,  6.14086582, 12.67576792,
       13.35013374,  5.64500512,  7.32950385, 13.45642379,  8.30125201,
        5.39450305])}
{'P': tensor([[4.2217e-03, 2.9006e-01, 2.4746e-01,  ..., 1.7598e-07, 1.8633e-04,
         6.8008e-04],
        [3.0919e-05, 1.1297e-02, 3.0613e-04,  ..., 3.2291e-07, 1.0153e-03,
         3.2669e-03],
        [3.5668e-05, 1.0038e-01, 6.8914e-03,  ..., 2.0919e-07, 3.5812e-04,
         1.1486e-03],
        ...,
        [2.7285e-03, 1.3475e-03, 1.4537e-03,  ..., 6.7487e-02, 1.9508e-05,
         6.1049e-05],
        [2.6908e-07, 4.8682e-07, 3.6093e-07,  ..., 1.3246e-07, 1.0626e-02,
         1.4834e-05],
        [6.1197e-07, 1.2911e-06, 9.2329e-07,  ..., 1.0062e-07, 3.2817e-01,
         7.6306e-03]]), 'X': tensor([[ 11.2517,  -4.6495,   2.9800],
        [ 12.0042,  -2.8350,   2.6864],
        [ 11.7795,  -3.4697,   3.5624],
        [ 11.7349,  -3.5838,   2.1644],
        [ 12.2692,  -1.8882,   3.9253],
        [ 12.5199,  -0.4711,   3.0436],
        [  4.6357,   5.4415,   7.9067],
        [  8.9284,  10.0036,  -6.9210],
        [  2.6321,  -4.5296,  -2.5159],
        [  4.6357,   5.4415,   7.9067],
        [ 11.9641,  -2.9568,   7.7211],
        [ 12.4691,   2.8525,  -5.5110],
        [  0.3715,  -4.7756,   0.2312],
        [  9.9756,  -6.6330,  -3.3429],
        [ -7.4887,   4.7364,   8.9074],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.4639, -10.9431,  -7.3718],
        [  6.3542,   0.6934,  -3.3840],
        [ -7.8849,  -6.3115,  -4.7479],
        [  5.5294,  -2.1106,   9.0626],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.1472,  -2.6801,   8.1194],
        [  3.8182,  -3.9347,  -1.7961],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.3384,  -2.6161,  -4.7630],
        [  1.0093,   7.6567,   2.3526],
        [ -2.8162,   7.7059,  -9.6885],
        [  3.2159, -10.8166,  -1.0406],
        [  1.7000,  -4.7609,   2.7935],
        [ 10.6283,  -5.7253,   7.4878],
        [  2.5375,   7.0676,  -4.3490],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.6229,   2.9068,   2.4429],
        [ -3.1671,   7.6158,   6.0765],
        [  1.7659,   7.4127,   4.7871],
        [  4.3377,   5.7570,   1.3712],
        [  0.1252, -11.0062,   7.2564],
        [  5.1519, -10.2387,   9.8191],
        [  3.4401,  13.4540,   4.6576],
        [ -7.0253,  -7.4493,  -1.5978],
        [  5.9526,   3.2525,  -4.3821],
        [ -8.1905,   3.7253,   4.1175],
        [  5.9883,   3.1536,   6.2320],
        [ -5.3382,   6.6533,   7.8054],
        [ -8.0087,   4.0173,   9.1091],
        [  1.3403,  -4.8011,  -1.1698],
        [  9.0652,   9.8597,  -3.5949],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -3.7964,  -9.9294,   7.7147],
        [ 11.2517,  -4.6495,   2.9800],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.0788,  -0.8710,  -6.8719],
        [ -5.1398,  -9.1367,   6.2074],
        [  9.9058,  -6.7203,   5.6928],
        [ -3.4100,   7.5436,  -7.6980],
        [  4.1348,   5.9514,  -0.8856],
        [ -5.0132,  -9.2234,  -8.3531],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.1206,  -4.7363,   8.7103],
        [ 12.2218,   4.0953,  -8.2072],
        [  6.1222,  -0.7261,   8.8204],
        [ -5.2589,   6.7024,  -5.5809],
        [ -4.5778,  -9.5013,  -5.0667],
        [  5.0233,   4.9647,  -3.1788],
        [  5.5158,  -2.1337,   3.5246],
        [  4.1537,   5.9340,  -6.0423],
        [  7.3724,  -9.0577,  -8.9843],
        [ 11.4495,   6.3026,   6.8257],
        [  6.2630,   2.1057,  -5.2136],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.0574,   1.7801,   8.8986],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.6614,  -4.5192,  -9.3641],
        [  1.7301,   7.4262,   5.0522],
        [  2.5382,   7.0672,   4.5097],
        [ -1.6086,   7.8928,   4.3385],
        [ 12.1379,   4.4125,  -3.8833],
        [ -8.1426,   3.8048,   9.3542],
        [  4.7482, -10.3907,   5.9149],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.9548,  -1.2273,  -2.6520],
        [ 10.4758,  -5.9539,   1.3667],
        [ 11.2939,   6.6325,  -5.8311],
        [ 12.1588,   4.3364,  -8.0867],
        [ -1.0012, -10.8576,   2.2867],
        [  9.5203,   9.3483,   5.8781],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.3200,   0.2690,   1.5900],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.5960, -11.0335,   3.1144],
        [  5.8318,   3.5567,  -4.7692],
        [  1.7122,  -4.7591,  -3.6734],
        [  9.7328,   9.0902,  -0.8917],
        [ 12.5996,   1.3971,  -1.0694],
        [ -9.4283,  -1.9585,  -0.9129],
        [  9.1409,  -7.5789,   1.5814],
        [ 11.9122,   5.1397,  -1.5498],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.4749,  -4.7892,  -0.8089],
        [ -8.4297,  -5.3637,  -9.3962],
        [  6.0467,  -9.8369,  -6.5816],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.0400,   8.6922,  -6.2847],
        [ 11.6019,   5.9540,  -5.1715]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0826, 0.0748,  ..., 0.8758, 0.8399, 0.7748],
        [0.0826, 0.0000, 0.1065,  ..., 0.9054, 0.7702, 0.7057],
        [0.0748, 0.1065, 0.0000,  ..., 0.9487, 0.8752, 0.8111],
        ...,
        [0.8758, 0.9054, 0.9487,  ..., 0.0000, 1.0752, 1.0292],
        [0.8399, 0.7702, 0.8752,  ..., 1.0752, 0.0000, 0.0659],
        [0.7748, 0.7057, 0.8111,  ..., 1.0292, 0.0659, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.17451685, 12.3344536 , 12.27991991, 12.26997165, 12.41366754,
       12.52875793,  7.14837514, 13.40850328,  5.23878371,  7.14837514,
       12.32408516, 12.79126785,  4.79003361, 11.97957425,  8.86082124,
       10.24557047, 11.21703654,  6.39187285, 10.09979747,  5.91854397,
        8.56889738,  5.80312949,  5.48276038, 13.01187484,  9.69792513,
        7.72291946,  8.20436764, 11.2845608 ,  5.05534609, 12.07225469,
        7.5092834 , 12.42554808,  9.09963894,  8.24808832,  7.62011734,
        7.20826219, 11.00694688, 11.46176259, 13.88683436, 10.23946247,
        6.78325151,  8.99791316,  6.76791483,  8.53014915,  8.95983586,
        4.98461983, 13.39372426,  9.62958973, 10.63038565, 12.17451685,
        9.95552626,  6.14086582, 10.48315436, 11.9702762 ,  8.27853076,
        7.24677664, 10.4977132 ,  5.01114719,  4.73784222, 12.889695  ,
        6.16512856,  8.51928531, 10.54658597,  7.06272256,  5.91408504,
        7.24326992, 11.67875855, 13.06957425,  6.60752373,  9.79039371,
        9.23071698, 10.65929968,  9.48337063,  5.24461068,  7.62509582,
        7.50917551,  8.0550286 , 12.91505111,  8.98765644, 11.42420508,
       10.0674887 ,  9.62238828,  6.07992623, 12.0495521 , 13.09737053,
       12.9089559 , 10.9036226 , 13.3426541 ,  5.48276038,  6.32572323,
       11.81423161, 11.04953767,  6.83083863,  5.05774295, 13.31764181,
       12.67680439,  9.62958973, 11.87412439, 12.97370371, 12.32545637,
        5.01114719,  9.99145096, 11.54672755, 10.78533838, 13.27994304,
       13.04051744])}
{'P': tensor([[4.9439e-03, 3.4357e-01, 2.3035e-01,  ..., 9.9476e-05, 9.4328e-06,
         8.3266e-06],
        [1.8719e-05, 7.5466e-03, 2.0284e-01,  ..., 2.7957e-04, 2.4302e-05,
         2.9097e-05],
        [1.1324e-05, 9.5358e-05, 5.5038e-03,  ..., 2.2778e-04, 2.3765e-05,
         3.1415e-05],
        ...,
        [7.5279e-06, 8.9630e-06, 9.3291e-06,  ..., 5.8323e-02, 2.9053e-03,
         7.9837e-04],
        [5.6756e-06, 5.6166e-06, 6.1040e-06,  ..., 5.3678e-04, 1.0869e-01,
         9.5312e-04],
        [1.2329e-06, 1.4301e-06, 1.6148e-06,  ..., 1.1135e-03, 1.3513e-01,
         3.3162e-01]]), 'X': tensor([[ 11.2767,  -4.6006,   0.7868],
        [ 12.0582,  -2.6640,   0.7132],
        [ 12.2311,  -2.0445,   1.3545],
        [ 11.5083,  -4.1177,   0.2389],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.5789,   0.1722,   1.6470],
        [ 10.5678,   7.9261,  -0.6100],
        [  9.3753,   9.5171,   3.9131],
        [ -7.0825,   5.2073,  -5.0497],
        [ -6.6895,   5.6051,   2.7899],
        [ 11.6165,  -3.8715,   6.4594],
        [  6.2555,   2.1474,   4.1685],
        [ -0.8569,   7.9156,  -6.5557],
        [  7.7750,  11.0665,   2.8364],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.9074,  -4.8139,   8.1674],
        [ -6.1914,  -8.2997,   8.5380],
        [  1.6232,  -4.7717,  -2.5655],
        [  5.8318,   3.5567,  -4.7692],
        [ -7.3446,   4.9113,  -5.9356],
        [  2.9470,  13.6085,   2.9809],
        [  6.1741,  -0.5329,   3.1913],
        [  1.6408,   7.4591,  -4.2294],
        [ 12.1751,  -2.2592,  -5.7751],
        [  3.2159, -10.8166,  -1.0406],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.2226,  -4.2802,  -4.0428],
        [ -4.2271,   7.2387,  -1.8316],
        [ -8.9636,   2.0648,   8.1946],
        [  6.9693,  -9.3200,  -8.3822],
        [ -7.4807,   4.7463,   2.0616],
        [  5.6928,  12.4710,   3.6001],
        [ 11.3691,   6.4760,   1.0787],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.2895,   0.0405,   5.7652],
        [  7.3592,  11.3937,   6.2407],
        [ -7.9925,  -6.1423,   2.8990],
        [  5.5240,  -2.1198,   6.2055],
        [  6.0310,  12.2795,  -6.4528],
        [  3.1892, -10.8220,  -6.8409],
        [ -4.0409,   7.3169,  -3.5779],
        [  8.6420,  -8.0580,   6.5781],
        [  1.0093,   7.6567,   2.3526],
        [  6.2247,  -0.3150,  -9.3657],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -1.9487,   7.8590,  -3.3330],
        [  6.2026,  12.1774,  -1.5887],
        [  8.0877,  -8.5297,  -1.1050],
        [  0.7145,  -4.8071,  -2.3766],
        [ -9.4117,   0.1160,   8.2574],
        [ -1.1885, -10.8211,   0.0865],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.0955,   2.8210,  -4.3321],
        [ -2.1817,   7.8274,  -4.6298],
        [  6.1741,  -0.5329,   3.1913],
        [  0.4996,   7.7738,  -4.4711],
        [  4.6992,   5.3690,   3.0935],
        [  4.3554,   5.7393,  -9.0766],
        [ -8.3062,  -5.6008,   0.0287],
        [  6.3569,   0.7591,   6.5011],
        [  6.9433,  -9.3362,  -0.3421],
        [  4.1999,  -3.6543,  -4.1998],
        [ -9.4773,  -1.0024,   5.2599],
        [  3.6249, -10.7255,   5.6605],
        [ -6.8618,  -7.6322,   3.7027],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.5383, -11.0274,  -4.9666],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.4267,  -5.3697,   2.4172],
        [ 10.9931,   7.2107,   7.7933],
        [  6.1372,   2.6726,  -6.7885],
        [  3.8182,  -3.9347,  -1.7961],
        [ 12.3357,   3.6001,   0.7903],
        [ -8.8363,   2.4086,  -5.6296],
        [  2.2182,  13.8007,   5.1889],
        [ -8.5607,  -5.0938,  -5.4531],
        [  4.4915,   5.5990,   7.0405],
        [  1.9505,  -4.7171,   9.9868],
        [ 11.5198,  -4.0923,   5.5088],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.5382,   7.0672,   4.5097],
        [  3.9913,  -3.8139,  -6.8311],
        [ -5.3736,   6.6310,  -1.5174],
        [  7.1821,  -9.1845,   9.8548],
        [ -4.3963,   7.1629,   4.5263],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.6035,   0.7494,   7.1536],
        [  2.5974,   7.0363,  -7.7810],
        [ 12.6038,   0.7665,  -3.4439],
        [  2.1379,  -4.6757,   6.4981],
        [ 10.4758,  -5.9539,   1.3667],
        [  6.0073,   3.0986,   6.7937],
        [ -7.5603,   4.6459,   8.2002],
        [ -2.0721,   7.8432,   6.6380],
        [  6.1575,  12.2046,   3.3257],
        [ -9.1658,  -3.4155,  -2.8880],
        [  9.1620,  -7.5573,  -9.9882],
        [ 12.6038,   0.7665,  -3.4439],
        [  2.5829,  -4.5466,  -8.4015],
        [  3.2990,  -4.2415,  -7.4857],
        [  2.0039,   7.3173,  -1.5885],
        [ -1.5451,   7.8975,   7.6104],
        [  1.1274,  14.0116,   8.1753]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.7716e-02, 9.6469e-02,  ..., 1.7664e-01, 5.2888e-01,
         7.0736e-01],
        [6.7716e-02, 0.0000e+00, 5.9509e-02,  ..., 2.0690e-01, 5.6787e-01,
         7.0047e-01],
        [9.6469e-02, 5.9509e-02, 1.0727e-04,  ..., 2.6137e-01, 5.2860e-01,
         6.4125e-01],
        ...,
        [1.7664e-01, 2.0690e-01, 2.6137e-01,  ..., 0.0000e+00, 6.5789e-01,
         8.8245e-01],
        [5.2888e-01, 5.6787e-01, 5.2860e-01,  ..., 6.5789e-01, 0.0000e+00,
         4.6209e-01],
        [7.0736e-01, 7.0047e-01, 6.4125e-01,  ..., 8.8245e-01, 4.6209e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.17901033, 12.3489324 , 12.40074901, 12.22276278, 12.50996828,
       12.58006038, 13.20988996, 13.35927527,  8.7907923 ,  8.7273569 ,
       12.24467041,  6.61386436,  7.96181288, 13.5247215 ,  8.06990483,
        5.27178363,  4.89869205, 10.35464916,  5.04028538,  6.83083863,
        8.8353674 , 13.92390689,  6.19707944,  7.6374603 , 12.3828957 ,
       11.2845608 ,  9.85384623,  9.9353921 ,  5.35774573,  8.38252937,
        9.19837792, 11.63765273,  8.85939626, 13.70893492, 13.08414689,
        9.7751189 ,  5.9510489 ,  6.28961988, 13.56367857, 10.08001894,
        5.91677188, 13.68063098, 11.28215429,  8.35856487, 11.8159254 ,
        7.72291946,  6.23262289, 12.35625694,  8.09704052, 13.6660671 ,
       11.75438006,  4.85995002,  9.41244865, 10.88618114,  9.37989631,
        6.7166297 ,  8.12580976,  6.19707944,  7.78980299,  7.13501264,
        7.20482224, 10.01804223,  6.4020404 , 11.63502679,  5.5671389 ,
        9.53015376, 11.32149259, 10.26327895, 12.04924159, 11.1341798 ,
        9.88571713,  9.99211663, 13.14690137,  6.69389431,  5.48276038,
       12.85032836,  9.15866607, 13.97779839,  9.96152788,  7.17789666,
        5.10446653, 12.22503754,  5.34250225,  7.50917551,  5.5205189 ,
        8.53501777, 11.65923396,  8.40445872,  8.50278896, 12.62576026,
        7.50035355, 12.62711252,  5.14124612, 12.0495521 ,  6.7593917 ,
        8.8737556 ,  8.11227668, 13.66991027,  9.78147542, 11.87665501,
       12.62711252,  5.22903117,  5.37343173,  7.58668144,  8.04718433,
       14.05687815])}
{'P': tensor([[4.3338e-03, 2.8950e-01, 1.7821e-01,  ..., 2.3233e-06, 5.3692e-06,
         1.5209e-03],
        [1.3554e-05, 5.6006e-03, 2.9623e-01,  ..., 5.0709e-06, 1.2224e-05,
         4.2142e-03],
        [3.2772e-05, 1.4635e-04, 1.7449e-02,  ..., 2.2013e-05, 5.2432e-05,
         2.0800e-02],
        ...,
        [3.0953e-06, 3.1519e-06, 3.5131e-06,  ..., 1.2880e-01, 2.9301e-02,
         1.4599e-05],
        [5.4742e-06, 5.4795e-06, 5.9251e-06,  ..., 2.8850e-02, 1.2796e-01,
         1.7836e-05],
        [1.9547e-06, 4.1051e-06, 5.6356e-06,  ..., 1.0237e-03, 1.9532e-03,
         2.5406e-02]]), 'X': tensor([[ 11.2847,  -4.5848,  -4.7503],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 11.9019,  -3.1379,  -3.4534],
        [  3.8111, -10.6787,  -4.6539],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 11.2209,  -4.7092,   2.6117],
        [ -3.5173, -10.0623,   4.8280],
        [ -9.3087,   0.7792,  -7.5587],
        [  5.9003,  12.3550,   2.1917],
        [  0.7685,  -4.8098,  -2.9310],
        [  3.6469,   6.3624,  -1.8414],
        [  3.9591, -10.6391,  -6.2557],
        [  3.2159, -10.8166,  -1.0406],
        [ -7.4602,   4.7718,  -4.6277],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.7685, -10.8982,  -6.8928],
        [  6.2365,  -0.2578,  -6.0212],
        [  2.0449,  13.8402,  -7.1642],
        [ 12.2328,  -2.0376,  -1.1234],
        [  9.6614,   9.1784,  -1.2087],
        [ -6.3252,  -8.1760,  -1.0891],
        [  2.0937,  13.8293,  -2.3533],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.6248,   3.9984,   8.4933],
        [ -2.0229, -10.6163,   8.6930],
        [  8.5198,  10.4091,  -8.6486],
        [  9.0051,   9.9234,   5.4411],
        [  5.1472,  -2.6801,   8.1194],
        [  3.6703,   6.3443,  -5.4401],
        [ -8.1666,   3.7653,   3.6075],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.3858,   4.4260,  -9.6769],
        [  8.6420,  -8.0580,   6.5781],
        [  2.8890,  -4.4316,   1.1600],
        [ -9.2508,   1.0572,   2.9978],
        [ 12.5224,  -0.4497,   4.8448],
        [ -7.0253,  -7.4493,  -1.5978],
        [  4.0587, -10.6112,   6.2273],
        [ -0.9715, -10.8630,   0.3555],
        [  2.0419,  -4.6978,  -5.0983],
        [  1.2273,  -4.8081,   9.8129],
        [  6.5428,  -9.5728,  -2.9897],
        [  4.8674,  12.8875,   6.7622],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.5456,   6.0861,   8.1101],
        [  4.8653,   5.1694,  -4.0823],
        [  2.7685, -10.8982,  -6.8928],
        [ -7.4733,   4.7555,   6.2490],
        [ -9.4117,   0.1160,   8.2574],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.3453,  -1.5462,   7.0212],
        [  1.3670,  -4.7990,   3.2603],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -7.8280,   4.2848,  -7.1955],
        [ -1.1885, -10.8211,   0.0865],
        [ 12.2328,  -2.0376,  -1.1234],
        [  1.8447,  -4.7372,   8.1235],
        [ -7.7239,   4.4298,   3.1216],
        [  5.5158,  -2.1337,   3.5246],
        [  3.7121,   6.3117,   0.6738],
        [ -1.6220,   7.8917,   0.0315],
        [ -9.4309,  -0.0599,  -4.0830],
        [  1.6408,   7.4591,  -4.2294],
        [  4.4039,   5.6903,   8.7366],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -7.5603,   4.6459,   8.2002],
        [  0.5600, -11.0321,  -0.8660],
        [  1.4749,  -4.7892,  -0.8089],
        [ -7.8126,   4.3066,   5.1791],
        [ -7.6525,  -6.6534,   5.1721],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.6319,  11.9066,   3.5165],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.6614,  -4.5192,  -9.3641],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.2555,   2.1474,   4.1685],
        [  6.2038,   2.4020,  -3.1692],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.9505,  -4.7171,   9.9868],
        [  3.3744,   6.5623,  -6.3636],
        [  6.3516,   0.6425,   0.3425],
        [  1.7232,  -4.7574,   3.1270],
        [  9.9058,  -6.7203,   5.6928],
        [  9.0004,  -7.7195,   4.6945],
        [  4.9957,  -2.8705,   6.5978],
        [ -9.3616,  -2.4750,   3.7765],
        [  1.6308,  13.9254,  -2.0984],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.3464,   0.5708,  -1.6397],
        [  0.9809,  -4.8145,   6.8706],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.5154,   2.5072,  -9.3274],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 12.5267,  -0.4125,  -8.7331],
        [  1.0000,   7.6591,  -4.5011],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.9600,   7.8577,   6.9121],
        [ -1.9646,   7.8571,   3.3817],
        [ 11.3691,   6.4760,   1.0787]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0645, 0.1049,  ..., 0.9592, 0.7571, 0.5785],
        [0.0645, 0.0000, 0.0454,  ..., 0.9568, 0.7631, 0.5333],
        [0.1049, 0.0454, 0.0000,  ..., 0.9312, 0.7433, 0.4892],
        ...,
        [0.9592, 0.9568, 0.9312,  ..., 0.0000, 0.2272, 0.6633],
        [0.7571, 0.7631, 0.7433,  ..., 0.2272, 0.0000, 0.5848],
        [0.5785, 0.5333, 0.4892,  ..., 0.6633, 0.5848, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.18045978, 12.35625694, 12.4217523 , 12.3828957 , 12.49317563,
       12.30858677, 11.33837396,  9.86089579, 12.16901615, 10.65929968,
        9.34127013, 13.69162795,  4.87083348,  7.33350642, 11.35182589,
       11.2845608 ,  8.85572292, 10.1371552 , 11.24434374,  6.24187044,
       13.99048094, 12.4013153 , 13.32613445, 10.33712258, 13.98691426,
        9.99211663,  6.90115901, 10.80728788, 13.45124917, 13.40024224,
        5.80312949,  7.32950385,  8.99276564, 13.27717982,  6.07992623,
        6.97106912, 11.8159254 ,  5.29009836,  9.3109884 , 12.53047489,
       10.23946247, 11.36089355, 10.90638066,  5.1224079 ,  4.96231532,
       11.59512194, 13.77604293, 10.31613269, 13.05148826,  7.09887557,
       11.24434374,  8.8580671 ,  9.41244865,  9.5642993 ,  9.97822473,
       12.44177641,  4.98989906, 12.11735334,  8.92395381, 10.88618114,
       12.4013153 ,  5.08372149,  8.90405823,  5.91408504,  7.32233733,
        8.05668609,  9.43112957,  7.6374603 ,  7.19534146,  9.61942445,
        8.8737556 , 11.04629471,  5.01114719,  8.92098752, 10.14044489,
       12.32408516, 13.62896593,  5.94909709,  8.97052115,  5.24461068,
       10.4977132 ,  6.61386436,  6.65259775, 13.41713661, 12.59840193,
        5.10446653,  7.37902766,  6.38399789,  5.05989547, 11.9702762 ,
       11.85743425,  5.76165934,  9.68324553, 14.02058621,  5.24137909,
        9.36378438,  4.91336994, 12.44177641, 12.76408562, 10.30972193,
       12.53345445,  7.72415048, 10.76154256,  8.09842704,  8.09899874,
       13.08414689])}
{'P': tensor([[4.2245e-03, 3.0151e-01, 2.9858e-01,  ..., 1.7929e-03, 7.2696e-06,
         1.1076e-07],
        [5.8792e-05, 6.1815e-03, 4.2631e-01,  ..., 3.2092e-03, 9.2726e-06,
         1.5863e-07],
        [3.4204e-05, 4.6877e-05, 1.2687e-02,  ..., 1.3748e-02, 1.4117e-05,
         2.8647e-07],
        ...,
        [2.2313e-06, 2.4560e-06, 3.6457e-06,  ..., 5.0873e-02, 7.8609e-06,
         2.0031e-06],
        [3.0697e-03, 2.7756e-03, 2.8056e-03,  ..., 2.9117e-04, 7.1339e-02,
         4.3413e-05],
        [2.5730e-04, 2.2594e-04, 1.2143e-04,  ..., 4.0058e-06, 7.8547e-04,
         6.7211e-02]]), 'X': tensor([[ 11.3367,  -4.4806,   0.4643],
        [ 11.5083,  -4.1177,   0.2389],
        [ 12.0582,  -2.6640,   0.7132],
        [ 12.2311,  -2.0445,   1.3545],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.7078,   2.7182,   1.8146],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -5.2752,  -9.0409,  -8.2568],
        [  0.9809,   7.6642,  -1.4217],
        [ 10.0567,   8.6697,   4.6329],
        [ -5.9872,   6.2037,  -9.4477],
        [  8.8091,  -7.9037,   0.3631],
        [  1.5662,  13.9375,   4.5142],
        [ -7.6525,  -6.6534,   5.1721],
        [  6.3460,   1.4252,   1.2291],
        [  3.3839,  -4.1966,   8.3574],
        [  1.7815,  -4.7481,   4.4289],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.1531,  -5.8748,   2.6934],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.3087,   0.7792,  -7.5587],
        [  5.6093,  -1.9705,   6.4807],
        [ -1.5451,   7.8975,   7.6104],
        [  5.1793,   4.7461,   0.3899],
        [  2.3200,   7.1756,   1.1938],
        [  6.2419,  -0.2313,   4.2440],
        [  3.9022,  13.2906,   5.5788],
        [ -0.8885,   7.9160,   0.0571],
        [  3.8111, -10.6787,  -4.6539],
        [ -7.8280,   4.2848,  -7.1955],
        [  6.1752,   2.5242,   3.5874],
        [  8.2422,  -8.4041,  -1.4962],
        [  4.8653,   5.1694,  -4.0823],
        [  0.5600, -11.0321,  -0.8660],
        [  7.5309,  11.2617,   7.2656],
        [ -8.9942,  -3.9990,   6.0448],
        [  6.1575,  12.2046,   3.3257],
        [ 11.7162,   5.6728,  -8.9372],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 11.9936,  -2.8677,   6.0614],
        [  6.0310,  12.2795,  -6.4528],
        [ -7.6650,   4.5092,   0.2520],
        [  6.3583,   1.1729,   8.7835],
        [  5.6095,  -1.9702,  -7.8233],
        [ -8.4297,  -5.3637,  -9.3962],
        [  0.2480, -11.0153,  -3.0183],
        [ 11.5568,   6.0602,   9.3244],
        [  4.2339,  -3.6268,   6.3125],
        [ -6.2200,  -8.2737,   5.5079],
        [ -4.4503,   7.1378,   3.3751],
        [ 10.1503,   8.5414,  -7.2681],
        [  2.5636, -10.9293,   9.6519],
        [  5.1793,   4.7461,   0.3899],
        [ 12.3025,  -1.7440,  -6.6632],
        [  0.9967,  -4.8144,  -9.7370],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 12.2589,  -1.9312,   9.8657],
        [  7.9141,  -8.6656,   9.6948],
        [  4.6357,   5.4415,   7.9067],
        [  0.3354,  -4.7708,  -4.9125],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -5.6174,   6.4708,  -1.7339],
        [ -7.0948,   5.1940,   9.8171],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.3367,   7.1676,  -9.5263],
        [  5.7784,  -1.6401,   9.5587],
        [  8.2154,  -8.4262,   8.0171],
        [  5.3804,  -2.3504,   7.2744],
        [  4.0078,  -3.8018,  -6.2168],
        [  0.1532,  -4.7422,  -7.0145],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.3038,   7.8092,   0.4164],
        [ -2.0229, -10.6163,   8.6930],
        [  6.3497,   1.3680,  -1.9076],
        [  9.0004,  -7.7195,   4.6945],
        [ -8.9033,   2.2329,  -8.8912],
        [  7.3724,  -9.0577,  -8.9843],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.1787,  -2.6383,  -4.2533],
        [  1.7232,  -4.7574,   3.1270],
        [ 11.7796,  -3.4696,   9.0088],
        [ -1.6220,   7.8917,   0.0315],
        [  6.0955,   2.8210,  -4.3321],
        [ 12.2914,   3.8037,   6.0978],
        [  6.0047,   3.1062,  -6.7914],
        [ 11.9936,  -2.8677,   6.0614],
        [ -5.1046,  -9.1610,   2.3418],
        [  6.0560,  -0.9421,  -5.0334],
        [ 12.5927,   0.4199,  -9.4997],
        [ -5.6169,  -8.7844,   8.9057],
        [ -4.8665,  -9.3204,  -5.2564],
        [  8.6271,  -8.0715,  -7.4813],
        [  6.3550,   0.7129,   0.4917],
        [ 12.5224,  -0.4497,   4.8448],
        [  1.6954,  -4.7616,   9.8198],
        [  4.9337,  12.8566,  -3.6582],
        [ -1.7009,   7.8851,  -8.2437],
        [  2.5663,   7.0526,  -6.8769],
        [  5.4336,  -2.2676,   5.6933],
        [  2.6321,  -4.5296,  -2.5159],
        [ 10.2470,   8.4054,  -0.1177],
        [  2.1625,  -4.6697,   7.0658],
        [ -7.2986,  -7.1225,  -3.0814]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0258, 0.0717,  ..., 0.3310, 0.4494, 0.9438],
        [0.0258, 0.0000, 0.0613,  ..., 0.3106, 0.4647, 0.9580],
        [0.0717, 0.0613, 0.0000,  ..., 0.2648, 0.4286, 1.0155],
        ...,
        [0.3310, 0.3106, 0.2648,  ..., 0.0000, 0.5524, 1.2587],
        [0.4494, 0.4647, 0.4286,  ..., 0.5524, 0.0000, 1.1878],
        [0.9438, 0.9580, 1.0155,  ..., 1.2587, 1.1878, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.18998195, 12.22276278, 12.3489324 , 12.40074901, 12.50996828,
       12.4013153 ,  9.12220271, 10.1979669 , 10.46739073,  7.72668467,
       13.2778378 ,  8.62162372, 11.83509372, 14.02526505, 10.14044489,
        6.50409372,  5.39098669,  5.0713364 ,  6.01356004, 10.04916678,
        4.8763135 ,  9.34127013,  5.94535529,  8.04718433,  7.02496223,
        7.54127425,  6.24614905, 13.85158417,  7.96575782, 11.33837396,
        8.92395381,  6.67121748, 11.77124373,  7.09887557, 11.04629471,
       13.54774355,  9.84314712, 13.66991027, 13.01728313,  9.69792513,
       12.33167382, 13.68063098,  8.89301516,  6.46560181,  5.94541803,
        9.99145096, 11.018084  , 13.0493368 ,  5.5748624 , 10.35093485,
        8.41148722, 13.26590071, 11.22596994,  7.02496223, 12.42554808,
        4.91652556,  9.79039371, 12.41012046, 11.73567759,  7.14837514,
        4.78258092, 10.35485331,  9.86089579,  8.56889738,  8.79284466,
        5.1224079 ,  7.53884104,  6.00663314, 11.76830981,  5.87132718,
        5.52416368,  4.74468406,  9.94285511,  7.81509631, 10.80728788,
        6.49538841, 11.85743425,  9.17905491, 11.67875855,  5.56809335,
        9.78285424,  5.81199422,  5.05989547, 12.27993098,  8.05668609,
        6.7166297 , 12.86648771,  6.76056625, 12.33167382, 10.48721382,
        6.12886024, 12.59969936, 10.42667093, 10.51437947, 11.81423161,
        6.3949043 , 12.53047489,  5.05444288, 13.77075174,  8.06643487,
        7.50500064,  5.88782913,  5.23878371, 13.25336001,  5.14607014,
       10.1979669 ])}
{'P': tensor([[4.2029e-03, 2.4826e-01, 2.2343e-01,  ..., 5.3744e-06, 5.5131e-05,
         2.5768e-07],
        [2.1730e-05, 1.0495e-02, 1.8184e-04,  ..., 4.0944e-06, 4.1512e-04,
         3.5558e-07],
        [4.4923e-05, 3.4002e-01, 1.2650e-02,  ..., 7.6242e-06, 4.2104e-04,
         4.7242e-07],
        ...,
        [4.0240e-03, 5.9460e-03, 7.9438e-03,  ..., 1.2827e-01, 2.6795e-04,
         4.2571e-06],
        [1.4051e-07, 2.0572e-07, 1.9064e-07,  ..., 9.1810e-08, 1.7532e-02,
         1.1518e-07],
        [2.0380e-03, 7.5156e-04, 8.4118e-04,  ..., 5.6185e-05, 6.7781e-07,
         1.9613e-02]]), 'X': tensor([[ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.3681e-02, 8.7182e-02,  ..., 6.7548e-01, 5.3998e-01,
         8.0086e-01],
        [7.3681e-02, 1.0727e-04, 5.9509e-02,  ..., 6.6813e-01, 4.9165e-01,
         8.6765e-01],
        [8.7182e-02, 5.9509e-02, 0.0000e+00,  ..., 6.0995e-01, 4.5301e-01,
         8.8487e-01],
        ...,
        [6.7548e-01, 6.6813e-01, 6.0995e-01,  ..., 0.0000e+00, 3.8930e-01,
         1.3103e+00],
        [5.3998e-01, 4.9165e-01, 4.5301e-01,  ..., 3.8930e-01, 0.0000e+00,
         1.3214e+00],
        [8.0086e-01, 8.6765e-01, 8.8487e-01,  ..., 1.3103e+00, 1.3214e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.19805394, 12.40074901, 12.3489324 , 12.50996828, 12.26997165,
       12.3344536 ,  9.84314712,  7.01475376,  8.36445758, 13.97430191,
        8.51928531,  7.78980299, 13.103709  ,  8.4486771 , 11.28895489,
        6.46560181,  5.79543038,  8.12043983, 11.04953767,  9.99211663,
        4.91038884, 11.1341798 , 11.18839169,  7.78980299, 10.06286074,
        7.5403822 , 12.46513646, 12.33167382, 11.57249984,  8.8353674 ,
        6.35029274,  8.83341608,  9.41244865,  8.31625171, 13.14690137,
       13.58101794,  6.06416957,  7.62011734,  6.65259775, 11.10066126,
        9.68324553, 11.22142195, 13.27717982,  7.61088125,  7.96181288,
        8.44940661, 12.30858677,  8.86082124,  5.70405525, 13.3614598 ,
        7.5158489 ,  5.10446653,  9.3109884 ,  5.92952538, 11.97957425,
        8.44940661,  9.87720102,  9.09661212, 12.26997165, 10.06286074,
        9.54154096,  7.78980299,  8.55982105, 11.81423161, 13.79762415,
       11.20100795,  5.07306274,  5.05989547,  8.8353674 ,  7.16058587,
        6.89245352, 13.97779839, 14.02532758,  9.91668182,  7.92200342,
        7.34250948,  7.72415048,  9.17595899, 12.95702647,  9.97822473,
        7.06272256, 13.77075174,  8.28968067, 14.10045797, 11.56146873,
       14.10796466, 12.38881635, 11.36089355, 13.20988996,  9.36256096,
       13.99877413, 11.8159254 , 10.75443106, 13.36800073, 13.32557133,
        7.92200342,  7.5984809 ,  9.09502385, 12.67576792, 14.02532758,
       11.50791859, 12.18998195, 13.06957425, 11.87665501, 13.77075174,
       10.98076873])}
{'P': tensor([[2.9554e-03, 2.4778e-01, 2.4661e-01,  ..., 6.3012e-08, 8.3308e-07,
         1.9494e-05],
        [1.5085e-05, 3.3999e-03, 1.2723e-02,  ..., 6.6130e-08, 8.5962e-07,
         3.7419e-05],
        [2.4351e-05, 9.7324e-04, 6.4575e-03,  ..., 1.1546e-07, 1.2802e-06,
         7.6872e-05],
        ...,
        [1.1777e-04, 8.2747e-05, 6.8123e-05,  ..., 2.6722e-02, 5.4290e-04,
         3.0454e-07],
        [2.3399e-04, 2.7040e-04, 2.0685e-04,  ..., 1.0188e-04, 1.4266e-01,
         2.2460e-05],
        [1.8772e-06, 2.0761e-06, 2.1068e-06,  ..., 3.2510e-05, 6.9164e-06,
         3.0448e-01]]), 'X': tensor([[ 11.4703,  -4.2006,   7.4086],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.9641,  -2.9568,   7.7211],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.3759,  -1.3943,   6.4659],
        [ -3.0008, -10.2827,  -8.9480],
        [  5.1292,   4.8183,  -5.7997],
        [ -7.8359,   4.2735,   7.2943],
        [  6.2365,  -0.2578,  -6.0212],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.0277,   4.9589,   7.0175],
        [ -9.4773,  -1.0024,   5.2599],
        [  9.6614,   9.1784,  -1.2087],
        [  3.3943,   6.5483,   1.0790],
        [  2.5636, -10.9293,   9.6519],
        [  2.6613,   7.0021,   4.5783],
        [  3.9663,   6.1018,  -5.4104],
        [  2.1477,   7.2550,   8.9688],
        [  2.5620,   7.0549,   8.4825],
        [ -8.8058,   2.4850,  -1.7425],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -8.9788,   2.0208,  -3.2867],
        [  8.6420,  -8.0580,   6.5781],
        [ -5.3179,  -9.0101,   9.4472],
        [  8.5198,  10.4091,  -8.6486],
        [  6.9956,  11.6590,  -7.6876],
        [  6.2770,  -0.0379,   2.4195],
        [  6.1984,  -0.4327,   1.5438],
        [  1.9241,   7.3503,   7.9004],
        [  6.0305,   3.0296,   0.1375],
        [ -9.3384,  -2.6161,  -4.7630],
        [  9.9217,  -6.7006,  -5.6728],
        [ -1.3940, -10.7772,   6.7835],
        [ 11.4495,   6.3026,   6.8257],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.9122,   5.1397,  -1.5498],
        [  5.6095,  -1.9702,  -7.8233],
        [  5.5240,  -2.1198,   6.2055],
        [  2.1440, -10.9811,   8.5281],
        [ -9.1749,  -3.3803,   6.4629],
        [  8.5716,  10.3596,   6.8818],
        [  5.6260,  -1.9401,   3.0259],
        [  0.6933,   7.7336,   2.1179],
        [ -1.6906,   7.8860,   7.7142],
        [  7.8375,  -8.7240,  -9.8913],
        [  0.1041, -11.0045,   2.4311],
        [ -2.7572,   7.7194,   5.9580],
        [ -9.2687,  -2.9794,   0.6436],
        [  1.6954,  -4.7616,   9.8198],
        [  6.6319,  11.9066,   3.5165],
        [  7.9141,  -8.6656,   9.6948],
        [  6.2804,   2.0019,   0.8607],
        [  4.5800,   5.5034,  -3.9432],
        [ 10.1798,   8.5002,  -5.8893],
        [  4.5954,  13.0099,  -4.4439],
        [  6.2036,   2.4030,  -1.2500],
        [  0.9478,  -4.8144,   0.6371],
        [  6.3505,   0.6235,  -8.5849],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -9.3444,   0.5821,   6.9101],
        [  2.9410,   6.8428,   5.4458],
        [ -3.4488, -10.0934,   2.5195],
        [  0.9478,  -4.8144,   0.6371],
        [  5.6672,  -1.8631,   8.1020],
        [ -8.5859,  -5.0393,   0.5476],
        [  3.2809,  13.5061,  -4.0199],
        [  1.0093,   7.6567,   2.3526],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 12.5996,   1.3971,  -1.0694],
        [ -7.7842,   4.3466,  -1.7548],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.1196,  -2.7160,   4.2705],
        [  0.9945,  -4.8144,  -0.1897],
        [ -3.9508,  -9.8515,   1.7923],
        [ -2.8162,   7.7059,  -9.6885],
        [ 12.5717,   1.9266,   7.4903],
        [ 11.6019,   5.9540,  -5.1715],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.2188,  -2.5839,   2.1833],
        [  0.3038,   7.8092,   0.4164],
        [ -3.9858,  -9.8334,  -3.7323],
        [ 12.5789,   0.1722,   1.6470],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.2592,   3.9425,   2.0383],
        [  9.0652,   9.8597,  -3.5949],
        [  2.7590,  13.6621,   9.4496],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.7735,   2.5640,  -6.3313],
        [ -4.4503,   7.1378,   3.3751],
        [ 11.3691,   6.4760,   1.0787],
        [ 10.3251,   8.2928,  -3.0006],
        [ -3.2053, -10.1993,  -3.8736],
        [  2.7121,   6.9743,  -8.4202],
        [  6.0073,   3.0986,   6.7937],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.1666,   3.7653,   3.6075],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3550,   0.7129,   0.4917],
        [  5.7963,  -1.6018,   1.8856],
        [ -6.0844,  -8.3955,   0.7705],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.2659,  13.7894,   6.5436]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0499, 0.0735,  ..., 1.0328, 1.0586, 0.3514],
        [0.0499, 0.0000, 0.0274,  ..., 1.0703, 1.0521, 0.3024],
        [0.0735, 0.0274, 0.0000,  ..., 1.0977, 1.0668, 0.2781],
        ...,
        [1.0328, 1.0703, 1.0977,  ..., 0.0000, 0.9926, 1.3434],
        [1.0586, 1.0521, 1.0668,  ..., 0.9926, 0.0000, 1.1223],
        [0.3514, 0.3024, 0.2781,  ..., 1.3434, 1.1223, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.21532282, 12.30695098, 12.32408516, 12.38881635, 12.44177641,
       12.45418236, 10.71162681,  7.03733775,  8.92549752,  6.24187044,
       10.61047596,  7.06169391,  9.53015376, 13.32613445,  7.37576154,
       11.22596994,  7.49076899,  7.27759233,  7.56617037,  7.50564223,
        9.14973319, 11.83075005,  9.93553424,  9.20340031, 11.8159254 ,
       10.46238054, 13.45124917, 13.59673406,  6.27715105,  6.2134961 ,
        7.59795445,  6.74872981,  9.69792513, 11.97238624, 10.86693708,
       13.06957425, 13.18278606, 12.97370371,  5.94541803,  5.91677188,
       11.18839169,  9.77777346, 13.44593519,  5.95111804,  7.76457095,
        8.06517143, 11.72749923, 11.0050379 ,  8.19703512,  9.73579738,
        5.05444288, 13.62896593, 11.73567759,  6.59176018,  7.15990676,
       13.26209545, 13.79762415,  6.652742  ,  4.90676242,  6.38105015,
        9.96152788,  9.36256096,  7.44805901, 10.66631784,  4.90676242,
        5.96556248,  9.95552626, 13.89886469,  7.72291946,  9.93553424,
        9.69792513, 12.67680439,  8.91551231,  9.50654967, 12.7125079 ,
        5.79543038,  4.9160962 , 10.61417181,  8.20436764, 12.71843462,
       13.04051744, 12.17901033,  5.82345867,  7.81509631, 10.61047596,
       12.58006038,  6.14086582, 12.87752052, 13.39372426, 13.93790235,
       13.95361098,  9.14044984,  8.41148722, 13.08414689, 13.2430377 ,
       10.6910768 ,  7.48311002,  6.7593917 ,  7.8575648 ,  8.99276564,
        5.76165934,  6.3949043 ,  6.01356004, 10.36844813,  5.34250225,
       13.97430191])}
{'P': tensor([[4.9722e-03, 3.4291e-01, 1.7273e-01,  ..., 5.7359e-06, 1.6682e-04,
         5.2248e-06],
        [3.2734e-05, 8.8589e-03, 2.3811e-01,  ..., 7.7608e-06, 3.7905e-04,
         1.2965e-05],
        [3.3138e-05, 1.5840e-04, 9.1423e-03,  ..., 7.2322e-06, 3.7357e-04,
         1.4462e-05],
        ...,
        [7.4906e-03, 5.6191e-03, 4.9832e-03,  ..., 1.2098e-01, 7.2766e-04,
         1.9816e-04],
        [9.2512e-06, 1.0969e-05, 1.1327e-05,  ..., 9.2146e-05, 5.0097e-02,
         3.1027e-02],
        [3.4072e-06, 2.9987e-06, 2.7623e-06,  ..., 2.6863e-04, 1.3343e-04,
         5.9170e-02]]), 'X': tensor([[ 11.5083,  -4.1177,   0.2389],
        [ 12.0582,  -2.6640,   0.7132],
        [ 12.2311,  -2.0445,   1.3545],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 12.4901,  -0.7052,   1.7315],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.8934,   2.2597,   3.5831],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.4430,  -0.1897,   4.8383],
        [ -2.7572,   7.7194,   5.9580],
        [  6.3551,   0.7149,   1.3344],
        [  2.5974,   7.0363,  -7.7810],
        [  5.6260,  -1.9401,   3.0259],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.2895,   0.0405,   5.7652],
        [ -2.0721,   7.8432,   6.6380],
        [ -8.9636,   2.0648,   8.1946],
        [  1.9241,   7.3503,   7.9004],
        [  6.3413,   0.4883,  -6.8416],
        [  5.1519, -10.2387,   9.8191],
        [  3.9515,  -3.8425,   6.1729],
        [ -1.9646,   7.8571,   3.3817],
        [  7.3592,  11.3937,   6.2407],
        [  2.5382,   7.0672,   4.5097],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.3569,   0.7591,   6.5011],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.7544,   2.6096,   3.3146],
        [ -4.4503,   7.1378,   3.3751],
        [ 12.0042,  -2.8350,   2.6864],
        [ 10.3251,   8.2928,  -3.0006],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -2.0721,   7.8432,   6.6380],
        [ -7.0120,   5.2825,   7.7762],
        [ -7.4602,   4.7718,  -4.6277],
        [ -7.8280,   4.2848,  -7.1955],
        [ 12.3759,  -1.3943,   6.4659],
        [ -5.6792,  -8.7352,   4.0318],
        [ -6.4620,  -8.0449,   6.2719],
        [  1.4306,  13.9620,  -0.2075],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.4932,   7.0903,  -5.4214],
        [ -0.1626, -10.9796,   6.4629],
        [ -3.5029,   7.5138,   8.1853],
        [  7.5784,  -8.9141,   5.8410],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.1440, -10.9811,   8.5281],
        [ -7.7239,   4.4298,   3.1216],
        [  5.9655,   3.2172,  -7.5953],
        [  4.2612, -10.5514,  -6.6268],
        [  7.3724,  -9.0577,  -8.9843],
        [  2.5318,  13.7231,   6.3692],
        [ -6.7992,   5.4991,   9.9186],
        [  1.7000,  -4.7609,   2.7935],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -4.7334,   6.9982,  -0.4392],
        [  5.8201,   3.5841,  -8.0609],
        [ -5.0550,   6.8231,  -9.2502],
        [  1.3103,  -4.8032,  -0.9300],
        [  4.5057,  -3.3894,  -0.8160],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.3497,   1.3680,  -1.9076],
        [  5.1667,  -2.6543,   1.2879],
        [  6.0560,  -0.9421,  -5.0334],
        [ -9.4430,  -0.1897,   4.8383],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.7845,   3.6652,   0.9175],
        [ -8.3900,   3.3727,   3.1650],
        [ -3.2486,   7.5925,  -1.3719],
        [  7.5930,  -8.9037,   7.2323],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.9912,   3.1453,  -3.9989],
        [ -9.1459,   1.4780,  -4.2403],
        [  3.6249, -10.7255,   5.6605],
        [ -5.0550,   6.8231,  -9.2502],
        [ -7.0948,   5.1940,   9.8171],
        [  3.9770, -10.6341,   8.0783],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -3.5520,   7.4976,   9.1522],
        [ 11.8951,  -3.1570,   7.0742],
        [  0.4119,  14.1019,  -4.1870],
        [ 10.0619,   8.6627,  -9.2884],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -7.4602,   4.7718,  -4.6277],
        [ -6.9838,  -7.4965,   0.9796],
        [  1.1139,   7.6281,  -2.5680],
        [  3.1480,  -4.3165,  -7.8002],
        [ 11.6544,   5.8273,  -3.1375],
        [  2.9260,  -4.4162,   8.3808],
        [ -6.3252,  -8.1760,  -1.0891],
        [  0.1287, -11.0065,   6.5757],
        [  6.1755,   2.5233,  -2.0042],
        [ 12.5996,   1.3971,  -1.0694],
        [ 10.1798,   8.5002,  -5.8893],
        [ 11.3800,  -4.3919,   1.4066],
        [  5.4426, -10.1181,  -5.1268],
        [ -8.5859,  -5.0393,   0.5476],
        [ -8.3900,   3.3727,   3.1650],
        [  6.3587,   1.1596,  -3.6629],
        [  6.3303,   1.6145,  -8.4083],
        [  0.2571,  -4.7594,   1.5618],
        [  3.6703,   6.3443,  -5.4401],
        [ -3.9700,   7.3453,  -4.1486]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.1316e-02, 1.1537e-01,  ..., 1.1132e-01, 3.3228e-01,
         4.6055e-01],
        [6.1316e-02, 0.0000e+00, 5.9509e-02,  ..., 1.3488e-01, 3.7738e-01,
         5.2187e-01],
        [1.1537e-01, 5.9509e-02, 1.0727e-04,  ..., 1.4812e-01, 4.3671e-01,
         5.7193e-01],
        ...,
        [1.1132e-01, 1.3488e-01, 1.4812e-01,  ..., 5.3633e-05, 4.0969e-01,
         4.5762e-01],
        [3.3228e-01, 3.7738e-01, 4.3671e-01,  ..., 4.0969e-01, 0.0000e+00,
         3.3136e-01],
        [4.6055e-01, 5.2187e-01, 5.7193e-01,  ..., 4.5762e-01, 3.3136e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.22276278, 12.3489324 , 12.40074901, 12.4013153 , 12.52372629,
       12.50996828,  9.38232121,  9.17595899,  7.48311002,  9.44486609,
        8.19703512,  6.39520397,  7.50035355,  5.95111804, 12.91505111,
        6.28961988,  8.11227668,  9.19837792,  7.59795445,  6.36004393,
       11.46176259,  5.51177501,  8.09899874, 13.56367857,  7.50917551,
        6.27905292,  6.4020404 ,  4.74468406,  5.83560009,  9.13507321,
        8.41148722, 12.3344536 , 13.2430377 , 12.35421461,  8.11227668,
        8.77913428,  8.85572292,  8.92395381, 12.45418236, 10.41909211,
       10.31886066, 14.03505624,  8.49161154,  9.96391045,  7.5158489 ,
       10.98076873,  8.29022197, 11.70016217,  5.27178363, 11.18839169,
        8.90405823,  6.77777606, 11.37939695, 11.67875855, 13.95472331,
        8.74470218,  5.05534609, 12.27835996,  8.4486771 ,  6.83515201,
        8.49161154,  4.978704  ,  5.63824971,  4.78002665,  6.49538841,
        5.80860165,  6.12886024,  9.44486609,  9.85384623,  6.84796202,
        9.04255655,  8.25828409, 11.70168755,  5.1224079 ,  6.76661595,
        9.26455795, 11.32149259,  8.49161154,  8.79284466, 11.35345345,
        9.96152788,  8.29640613, 12.30695098, 14.10796466, 13.27717982,
       12.11735334,  8.85572292, 10.24557047,  7.70897675,  5.34250225,
       13.03002338,  5.29755487, 10.33712258, 11.00726483,  6.67108662,
       12.67680439, 13.26209545, 12.19805394, 11.4890861 ,  9.95552626,
        9.04255655,  6.46356806,  6.53290663,  4.7663526 ,  7.32950385,
        8.34949383])}
{'P': tensor([[3.9708e-03, 2.4916e-01, 1.4709e-01,  ..., 1.0430e-07, 2.0078e-04,
         1.6156e-06],
        [2.9737e-05, 6.1862e-03, 4.7937e-01,  ..., 1.6511e-07, 5.4877e-04,
         3.2095e-06],
        [9.1381e-05, 1.4341e-04, 4.1819e-02,  ..., 1.2141e-06, 7.1647e-03,
         2.9758e-05],
        ...,
        [3.0060e-05, 1.9782e-05, 1.3093e-05,  ..., 2.7857e-01, 3.7725e-06,
         4.1294e-04],
        [1.8886e-07, 2.3447e-07, 3.1586e-07,  ..., 1.4997e-05, 1.3673e-02,
         1.0097e-03],
        [3.5105e-06, 2.8986e-06, 2.4298e-06,  ..., 1.1252e-02, 4.3086e-06,
         5.0255e-02]]), 'X': tensor([[ 11.5198,  -4.0923,   5.5088],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.5224,  -0.4497,   4.8448],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.2692,  -1.8882,   3.9253],
        [  2.9410,   6.8428,   5.4458],
        [  0.1252, -11.0062,   7.2564],
        [ 10.5678,   7.9261,  -0.6100],
        [ 10.6283,  -5.7253,   7.4878],
        [  7.7794,  11.0629,   8.9069],
        [  5.3858,   4.4260,  -9.6769],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.1041, -11.0045,   2.4311],
        [ -3.3999, -10.1152,  -1.2088],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.9193,  -1.3176,   5.1739],
        [  2.1504,  -4.6726,   8.7908],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4283,  -1.9585,  -0.9129],
        [  1.7301,   7.4262,   5.0522],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.0961,   1.6529,   3.1553],
        [  2.3589,  13.7668,  -3.6501],
        [  4.1297,  -3.7098,  -3.9838],
        [  5.9883,   3.1536,   6.2320],
        [  0.6757, -11.0361,  -1.3047],
        [ -1.9487,   7.8590,  -3.3330],
        [ -7.8280,   4.2848,  -7.1955],
        [  1.5246,  -4.7838,  -5.2095],
        [ -7.5401,   4.6718,   9.0033],
        [  6.3087,   1.8053,  -1.9481],
        [ -6.4464,   5.8272,   7.5038],
        [ -5.1369,   6.7755,   8.3836],
        [  3.8111, -10.6787,  -4.6539],
        [  6.3460,   1.4252,   1.2291],
        [ 11.9936,  -2.8677,   6.0614],
        [  7.8982,  -8.6778,   0.0239],
        [  5.5626,  -2.0536,   8.4433],
        [  2.5382,   7.0672,   4.5097],
        [ -8.9942,  -3.9990,   6.0448],
        [ -5.3736,   6.6310,  -1.5174],
        [ -8.9178,   2.1936,   7.5388],
        [ 12.4013,  -1.2597,  -7.6474],
        [  6.3587,   1.1596,  -3.6629],
        [ -7.9298,   4.1367,  -3.9773],
        [ 11.2767,  -4.6006,   0.7868],
        [ 12.5789,   0.1722,   1.6470],
        [  0.2480, -11.0153,  -3.0183],
        [ -2.6069, -10.4298,  -6.1995],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.3124,   0.2057,  -0.4593],
        [ -2.0721,   7.8432,   6.6380],
        [  1.4150,  13.9647,  -4.0915],
        [  5.1519, -10.2387,   9.8191],
        [ -8.3900,   3.3727,   3.1650],
        [  9.6662,   9.1725,   8.1056],
        [  6.9653,  -9.3225,  -3.4727],
        [ -8.9942,  -3.9990,   6.0448],
        [  4.1348,   5.9514,  -0.8856],
        [  5.0617,  12.7957,   8.3544],
        [  0.5175,  14.0910,   4.6540],
        [ -8.1426,   3.8048,   9.3542],
        [  2.2076,  13.8032,  -1.5099],
        [ 11.6019,   5.9540,  -5.1715],
        [  3.4401,  13.4540,   4.6576],
        [ 12.1588,   4.3364,  -8.0867],
        [ 11.4495,   6.3026,   6.8257],
        [  6.3516,   0.6425,   0.3425],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.2895,   0.0405,   5.7652],
        [  2.0419,  -4.6978,  -5.0983],
        [  8.4691,  10.4571,  -5.0650],
        [ -5.7347,  -8.6908,   9.2878],
        [  5.2939,  -2.4784,  -4.5840],
        [ -5.6174,   6.4708,  -1.7339],
        [ -1.5875,   7.8944,  -9.4286],
        [ 12.4013,  -1.2597,  -7.6474],
        [  2.1120,  -4.6818,   8.6144],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.3087,   0.7792,  -7.5587],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.2038,   2.4020,  -3.1692],
        [ 10.3251,   8.2928,  -3.0006],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.3118,   4.5452,   4.2114],
        [  0.9478,  -4.8144,   0.6371],
        [ 11.6019,   5.9540,  -5.1715],
        [ -9.2508,   1.0572,   2.9978],
        [  3.5939,   6.4029,  -1.8357],
        [  4.6063,  13.0051,  -3.1456],
        [  5.2188,  -2.5839,   2.1833],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.3569,   0.7591,   6.5011],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 11.5568,   6.0602,   9.3244],
        [  5.4429,  -2.2529,   9.7306],
        [  4.9337,  12.8566,  -3.6582],
        [ -6.1914,  -8.2997,   8.5380],
        [  0.3715,  -4.7756,   0.2312],
        [  7.7750,  11.0665,   2.8364],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.7792,  11.8084,   1.6251],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.6319,  11.9066,   3.5165],
        [ -6.2668,   5.9808,   2.5614]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0718, 0.1441,  ..., 1.2229, 0.3952, 0.5911],
        [0.0718, 0.0000, 0.0735,  ..., 1.2945, 0.3601, 0.6626],
        [0.1441, 0.0735, 0.0000,  ..., 1.3644, 0.3190, 0.7352],
        ...,
        [1.2229, 1.2945, 1.3644,  ..., 0.0000, 1.4592, 0.6495],
        [0.3952, 0.3601, 0.3190,  ..., 1.4592, 0.0000, 0.8969],
        [0.5911, 0.6626, 0.7352,  ..., 0.6495, 0.8969, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.22503754, 12.33167382, 12.45418236, 12.53047489, 12.44177641,
       12.41366754,  7.44805901, 11.00694688, 13.20988996, 12.07225469,
       13.52430176,  6.97106912,  9.64157461, 11.0050379 , 10.67132524,
        9.75200848,  6.06416957,  5.14371005, 13.25336001,  9.62958973,
        7.62509582,  7.83534987,  9.24501829, 13.96746816,  5.5513072 ,
        6.76791483, 11.05672136,  8.09704052,  8.92395381,  5.0209005 ,
        8.87007495,  6.5618968 ,  8.68979052,  8.50268322, 11.33837396,
        6.50409372, 12.33167382, 11.73397309,  5.92952538,  7.50917551,
        9.84314712,  8.53501777,  9.18358562, 12.46513646,  6.46356806,
        8.94395628, 12.17901033, 12.58006038, 11.018084  , 10.7506423 ,
       12.40074901,  6.31576264,  8.11227668, 14.03618649, 11.46176259,
        9.04255655, 13.32557133, 11.63724792,  9.84314712,  7.24677664,
       13.76047999, 14.10045797,  8.98765644, 13.97857591, 13.04051744,
       13.88683436, 12.9089559 , 13.06957425,  6.38399789,  4.91652556,
        6.28961988,  5.1224079 , 13.45642379, 10.412299  ,  5.84533345,
        8.56889738,  8.05242591, 12.46513646,  5.13616224,  5.83560009,
        9.34127013, 12.10942412,  6.65259775, 13.2430377 ,  8.27853076,
        6.99097013,  4.90676242, 13.04051744,  9.3109884 ,  7.34250948,
       13.79676654,  5.82345867,  8.20436764,  6.4020404 , 12.05395056,
       13.0493368 ,  5.89074104, 13.77075174, 10.35464916,  4.79003361,
       13.5247215 ,  6.27905292, 13.61600536,  9.43112957, 13.62896593,
        8.66273171])}
{'P': tensor([[3.4368e-03, 2.6484e-01, 1.9916e-01,  ..., 9.4307e-08, 6.9987e-06,
         1.8118e-06],
        [3.5901e-05, 7.2424e-03, 5.6122e-01,  ..., 1.8707e-07, 1.1832e-05,
         2.4703e-06],
        [6.5478e-05, 1.1512e-04, 3.3567e-02,  ..., 8.3463e-07, 3.5605e-05,
         6.3806e-06],
        ...,
        [4.5999e-04, 3.0646e-04, 1.7972e-04,  ..., 1.6791e-01, 4.3798e-03,
         6.8480e-03],
        [6.5936e-03, 7.0222e-03, 5.3810e-03,  ..., 1.0933e-04, 1.3407e-01,
         1.2360e-03],
        [1.8821e-01, 1.1610e-01, 7.8698e-02,  ..., 4.0461e-06, 3.6568e-04,
         6.5058e-02]]), 'X': tensor([[ 11.6165,  -3.8715,   6.4594],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.3453,  -1.5462,   7.0212],
        [ 11.8951,  -3.1570,   7.0742],
        [  3.1064,  13.5607,   6.7605],
        [  4.5325,  -3.3643,   6.7699],
        [ 10.9394,   7.3069,  -8.9365],
        [  3.6154,  -4.0640,  -7.9698],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.1120,  -4.6818,   8.6144],
        [  4.5767,   5.5071,   3.9069],
        [  3.1756, -10.8247,   6.8882],
        [ 12.2435,   4.0076,   8.1257],
        [  0.5175,  14.0910,   4.6540],
        [  3.0876,   6.7527,  -5.3540],
        [ -8.9033,   2.2329,  -8.8912],
        [ 10.5678,   7.9261,  -0.6100],
        [  6.0047,   3.1062,  -6.7914],
        [  5.8318,   3.5567,  -4.7692],
        [ 10.9931,   7.2107,   7.7933],
        [  6.2770,  -0.0379,   2.4195],
        [  5.6248,   3.9984,   8.4933],
        [  9.3560,   9.5391,   9.4943],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.6093,  -1.9705,   6.4807],
        [ -0.3432,   7.8900,   7.1805],
        [ -0.6323,   7.9085,  -5.9544],
        [  1.1644, -11.0394,   7.8511],
        [ 12.5267,  -0.4125,  -8.7331],
        [ -0.2586,   7.8825,  -9.9840],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.8450,  -9.9356,   3.0767],
        [ -4.0633,  -9.7927,  -3.0825],
        [  4.8674,  12.8875,   6.7622],
        [ -7.7842,   4.3466,  -1.7548],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.7290,   7.8825,  -3.3831],
        [ -4.4503,   7.1378,   3.3751],
        [ -1.5408,   7.8978,   5.3470],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.3444,   0.5821,   6.9101],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.0506,   4.9277,  -9.5094],
        [  4.4915,   5.5990,   7.0405],
        [  5.7784,  -1.6401,   9.5587],
        [  2.3373,  -4.6232,  -5.2246],
        [ -8.7078,   2.7182,   1.8146],
        [  1.6408,   7.4591,  -4.2294],
        [ 11.5456,   6.0861,   8.1101],
        [  4.2041,  -3.6509,  -4.9488],
        [  2.3376,  -4.6231,   9.8436],
        [  2.0729,   7.2878,   7.0880],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.1469,   6.9237,   1.6160],
        [ -1.9646,   7.8571,   3.3817],
        [  9.7328,   9.0902,  -0.8917],
        [ -2.9497, -10.3028,  -7.3479],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.6603,   2.8250,   5.9181],
        [ -6.3489,   5.9117,   6.5161],
        [ 10.9599,  -5.1849,  -3.5526],
        [  4.6231,   5.4556,   8.4973],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.1752,   2.5242,   3.5874],
        [ -9.4378,  -1.8606,  -2.5668],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.8835,  -4.3178,   9.3080],
        [  2.9260,  -4.4162,   8.3808],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.8325,  -1.5222,  -2.1341],
        [ -3.3999, -10.1152,  -1.2088],
        [  5.8617,   3.4853,  -9.8080],
        [  9.7328,   9.0902,  -0.8917],
        [  3.3474,  -4.2162,   5.2455],
        [  3.4140,  -4.1803,  -8.8224],
        [  9.0652,   9.8597,  -3.5949],
        [  2.3373,  -4.6232,  -5.2246],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -8.8154,   2.4611,   2.5198],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -0.4341, -10.9473,   7.4509],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 10.1798,   8.5002,  -5.8893],
        [  7.7794,  11.0629,   8.9069],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -4.0868,   7.2981,   8.1874],
        [  0.8940, -11.0402,  -0.9783],
        [ -2.1533, -10.5778,   1.0112],
        [  4.4290,  -3.4596,   1.4181],
        [ -5.9987,   6.1949,  -6.2114],
        [  4.2041,  -3.6509,  -4.9488],
        [  1.3767, -11.0341,   9.9415],
        [  1.3705,  13.9723,   7.9266],
        [  2.2659,  13.7894,   6.5436],
        [  9.1820,   9.7334,  -2.7932],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.4691,   2.8525,  -5.5110],
        [  1.6232,  -4.7717,  -2.5655],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.7558,  -4.4844,   0.0636],
        [  7.1821,  -9.1845,   9.8548]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0623, 0.1241,  ..., 0.8542, 0.6371, 0.2544],
        [0.0623, 0.0000, 0.0735,  ..., 0.9098, 0.6175, 0.3164],
        [0.1241, 0.0735, 0.0000,  ..., 0.9783, 0.6601, 0.3734],
        ...,
        [0.8542, 0.9098, 0.9783,  ..., 0.0000, 1.0023, 0.6249],
        [0.6371, 0.6175, 0.6601,  ..., 1.0023, 0.0000, 0.7437],
        [0.2544, 0.3164, 0.3734,  ..., 0.6249, 0.7437, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.24467041, 12.33167382, 12.45418236, 12.38881635, 12.44177641,
       12.30695098, 13.91198191,  5.64464807, 13.15525803,  5.43943791,
       13.13205357,  5.13616224,  7.16058587, 11.28093507, 12.88270018,
       14.10045797,  7.42512621,  9.17905491, 13.20988996,  6.76056625,
        6.83083863, 13.14690137,  6.27715105,  6.90115901, 13.3614598 ,
       10.0674887 ,  5.94535529,  7.89745107,  7.9337647 , 11.10066126,
       12.53345445,  7.88677538,  9.97220799, 11.52733581, 10.60226614,
       13.77604293,  8.91551231,  7.96181288,  8.06990483,  8.41148722,
        8.04664886,  7.83534987,  9.36256096, 13.14777433,  7.05627461,
        7.17789666,  6.00663314,  5.18047638,  9.12220271,  7.6374603 ,
       13.05148826,  5.56809335,  5.1805345 ,  7.57685908,  9.41244865,
       13.12218013,  8.09899874, 13.31764181, 10.7167285 , 12.46513646,
       10.24557047,  9.10945678,  8.67502348, 12.12448912,  7.15100303,
        9.55593374,  6.67121748,  9.61942445,  7.58668144,  9.87720102,
        5.29755487, 12.27991991,  6.27905292,  6.02789634, 10.67132524,
        6.81961996, 13.31764181,  5.38342077,  5.39723241, 13.39372426,
        5.18047638,  9.96152788,  9.15253344, 10.75443106,  9.51074763,
       10.95593729, 12.35421461, 13.26209545, 13.52430176,  9.66977802,
        8.36445758, 11.07637531, 10.794755  ,  5.62005629,  8.62329466,
        5.56809335, 11.11969815, 14.0393949 , 13.97430191, 13.38091121,
       10.8992524 , 12.79126785,  5.04028538, 10.1371552 ,  5.26342203,
       11.65923396])}
{'P': tensor([[5.6496e-03, 4.4766e-01, 2.4197e-01,  ..., 3.1226e-04, 1.7214e-06,
         7.3791e-04],
        [1.4729e-05, 5.2284e-03, 5.0451e-01,  ..., 4.5239e-04, 8.0534e-07,
         1.0648e-03],
        [8.6735e-06, 1.7010e-05, 5.9483e-03,  ..., 6.8684e-04, 5.4274e-07,
         1.7108e-03],
        ...,
        [4.1283e-05, 5.6243e-05, 7.5377e-05,  ..., 1.6113e-01, 2.6382e-05,
         1.3623e-02],
        [1.4727e-02, 9.8274e-03, 5.8221e-03,  ..., 2.0271e-04, 2.1302e-02,
         1.4273e-04],
        [1.3982e-05, 1.8964e-05, 2.5861e-05,  ..., 5.2364e-03, 4.8662e-06,
         3.7775e-02]]), 'X': tensor([[ 11.6801,  -3.7196,  -0.9181],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.5770,   1.8512,  -0.6578],
        [ 12.5789,   0.1722,   1.6470],
        [  3.3943,   6.5483,   1.0790],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -8.7078,   2.7182,   1.8146],
        [ 10.8218,  -5.4177,   6.8573],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.1929,   2.4501,  -6.9352],
        [  4.9152,  12.8653,  -1.4459],
        [ 11.1085,  -4.9204,   0.0736],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.3542,   0.6934,  -3.3840],
        [  4.5325,  -3.3643,   6.7699],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.1348,   5.9514,  -0.8856],
        [  6.3124,   0.2057,  -0.4593],
        [  1.0000,   7.6591,  -4.5011],
        [ -3.5029,   7.5138,   8.1853],
        [  0.6363, -11.0348,   6.1414],
        [  5.9026,   3.3837,  -0.6573],
        [  2.0899,  -4.6870,  -7.4260],
        [ 11.1420,   6.9331,  -2.4779],
        [  6.0047,   3.1062,  -6.7914],
        [  6.3301,   0.3638,  -7.5685],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.0560,  -0.9421,  -5.0334],
        [  6.0073,   3.0986,   6.7937],
        [  2.3490,  -4.6199,  -4.2489],
        [  3.8182,  -3.9347,  -1.7961],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.1787,  -2.6383,  -4.2533],
        [  7.1697,  11.5343,  -0.7276],
        [  9.3607,   9.5337,  -4.5412],
        [ 12.2435,   4.0076,   8.1257],
        [ -9.1624,  -3.4286,   2.0647],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.2728, -11.0372,   9.8121],
        [  5.3118,   4.5452,   4.2114],
        [  3.7237,  -3.9966,  -5.0376],
        [  5.5240,  -2.1198,   6.2055],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.1960,  -3.2964,   1.6921],
        [  0.9967,  -4.8144,  -9.7370],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.4572,   7.1085,   7.7376],
        [  2.8602,   6.8905,   5.8779],
        [ -9.4645,  -1.4880,  -2.0226],
        [  4.7813,  -3.1139,   8.2691],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.9204,   7.3518,   8.4781],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.9843,  -9.8679,  -6.1449],
        [  2.0243,   7.3086,   7.5432],
        [  2.0870,  -4.6877,   9.6823],
        [  7.0358,  -9.2784,  -9.8479],
        [  3.3069,   6.6089,   8.7377],
        [  9.5261,  -7.1675,   1.4206],
        [ -9.2707,   0.9661,   4.4863],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.3739,   0.3982,  -4.2564],
        [  0.8940, -11.0402,  -0.9783],
        [ -9.4501,  -1.7133,   2.2980],
        [ -8.9788,   2.0208,  -3.2867],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.7420,  -4.4896,  -3.7139],
        [  6.1730,  -0.5375,  -8.8861],
        [  1.6308,  13.9254,  -2.0984],
        [  4.6231,   5.4556,   8.4973],
        [ -8.0330,  -6.0765,   6.7945],
        [  3.6469,   6.3624,  -1.8414],
        [  2.2437,  -4.6489,   6.1430],
        [ 11.4495,   6.3026,   6.8257],
        [  9.3624,   9.5318,   0.2779],
        [ 10.0619,   8.6627,  -9.2884],
        [ -0.3432,   7.8900,   7.1805],
        [ -5.0132,  -9.2234,  -8.3531],
        [  4.5340,  -3.3629,   8.9652],
        [  6.0073,   3.0986,   6.7937],
        [  2.3261,   7.1726,   6.2522],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.3246,   1.6696,  -0.1544],
        [  5.1292,   4.8183,  -5.7997],
        [ 12.0845,  -2.5772,  -4.6073],
        [  9.9217,  -6.7006,  -5.6728],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.4996,   2.6335,  -6.3166],
        [  6.2036,   2.4030,  -1.2500],
        [  2.2182,  13.8007,   5.1889],
        [ -7.9925,  -6.1423,   2.8990],
        [  4.5057,  -3.3894,  -0.8160],
        [ -0.3432,   7.8900,   7.1805],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.2188,  -2.5839,   2.1833],
        [  4.4039,   5.6903,   8.7366],
        [ -5.0132,  -9.2234,  -8.3531],
        [  2.7976,  -4.4682,  -4.2020],
        [  4.1537,   5.9340,  -6.0423],
        [  6.0467,  -9.8369,  -6.5816],
        [  4.4633,   5.6287,  -0.7754]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0558, 0.1007,  ..., 0.2531, 0.4981, 0.0578],
        [0.0558, 0.0000, 0.0472,  ..., 0.2568, 0.5264, 0.1021],
        [0.1007, 0.0472, 0.0000,  ..., 0.2861, 0.5667, 0.1369],
        ...,
        [0.2531, 0.2568, 0.2861,  ..., 0.0000, 0.3070, 0.3020],
        [0.4981, 0.5264, 0.5667,  ..., 0.3070, 0.0000, 0.5245],
        [0.0578, 0.1021, 0.1369,  ..., 0.3020, 0.5245, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.25806559, 12.4013153 , 12.52372629, 12.67680439, 12.7125079 ,
       12.58006038,  7.37576154, 12.1460931 ,  9.12220271, 12.10222204,
        6.02789634,  6.65991877, 13.77222752, 12.1494023 , 10.00794468,
        6.39187285,  5.64464807,  5.04028538,  7.24677664,  6.31576264,
        7.72415048,  8.29022197, 11.05317597,  6.80370254,  5.13182506,
       13.12299145,  6.76056625,  6.34059021,  5.35774573,  6.12886024,
        6.7593917 ,  5.18277858,  5.48276038, 12.64235359,  5.81199422,
       13.58101794, 13.36092624, 12.88270018,  9.78285424, 12.17451685,
       11.11038551,  6.99097013,  5.46244988,  5.91677188,  9.3349858 ,
        9.43737523,  5.94535529,  9.76896904,  4.91652556,  8.53501777,
        7.52117132,  7.46053281,  9.58072286,  5.70591986, 10.1979669 ,
       10.1979669 ,  7.5984809 ,  8.27853076, 11.54071859,  7.58378655,
        5.13125513, 11.64436684,  7.39004226, 11.92133129,  9.32094055,
        4.89869205,  9.38232121, 11.07637531,  9.60412735,  9.20340031,
        8.62329466,  5.26067019,  6.19633139, 14.02058621,  7.15100303,
       10.07238463,  7.33350642,  5.16205087, 13.06957425, 13.3607379 ,
       13.27717982,  7.89745107, 10.4977132 ,  5.64500512,  6.7593917 ,
        7.5403822 , 12.07225469,  6.54127431,  7.03733775, 12.35625694,
       11.97238624, 12.62576026, 12.77402351,  6.652742  , 13.97779839,
       10.08001894,  5.63824971,  7.89745107, 11.86239961,  5.82345867,
        7.19534146, 10.4977132 ,  5.27178363,  7.24326992, 11.54672755,
        7.18354885])}
{'P': tensor([[3.5915e-03, 1.7931e-01, 1.6943e-01,  ..., 4.3064e-05, 8.3917e-05,
         5.9412e-04],
        [4.3320e-05, 4.8168e-03, 3.7964e-02,  ..., 8.3841e-05, 8.3972e-05,
         8.2898e-04],
        [1.6356e-05, 1.2014e-04, 3.5344e-03,  ..., 8.6026e-05, 3.9804e-05,
         3.3334e-04],
        ...,
        [3.4827e-07, 3.8113e-07, 4.1635e-07,  ..., 4.8112e-02, 6.3851e-06,
         1.1224e-05],
        [1.3366e-02, 1.6904e-02, 1.5146e-02,  ..., 7.8017e-04, 1.1249e-01,
         1.4453e-01],
        [9.2741e-04, 1.3079e-03, 1.4490e-03,  ..., 5.9519e-03, 3.0453e-03,
         1.4712e-01]]), 'X': tensor([[ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.4605e-02, 9.2942e-02,  ..., 3.6491e-01, 1.2538e-01,
         2.8180e-01],
        [5.4605e-02, 0.0000e+00, 1.2954e-01,  ..., 3.1370e-01, 7.8017e-02,
         2.3255e-01],
        [9.2942e-02, 1.2954e-01, 1.0727e-04,  ..., 3.9451e-01, 2.0755e-01,
         3.6086e-01],
        ...,
        [3.6491e-01, 3.1370e-01, 3.9451e-01,  ..., 0.0000e+00, 2.8376e-01,
         2.4367e-01],
        [1.2538e-01, 7.8017e-02, 2.0755e-01,  ..., 2.8376e-01, 0.0000e+00,
         1.5642e-01],
        [2.8180e-01, 2.3255e-01, 3.6086e-01,  ..., 2.4367e-01, 1.5642e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.26997165, 12.3344536 , 12.40074901, 12.50996828, 12.52875793,
       12.58006038,  8.09984231, 13.99048094,  8.09984231,  8.7273569 ,
        8.86082124, 13.68672855,  6.49538841, 13.2778378 ,  9.09502385,
        5.77219077, 13.96746816, 11.63765273, 12.97370371, 10.19036081,
        9.70108409,  7.8788965 ,  9.86089579,  9.43737523, 12.62576026,
        9.63742669,  5.70591986, 13.13205357, 11.81423161, 11.89855267,
       14.09357547,  8.29640613, 11.0050379 , 11.48434767, 10.27572673,
       11.33837396, 11.57249984,  9.77777346,  5.84450127, 13.03002338,
       12.50996828,  5.40689332, 13.103709  ,  8.06990483,  7.72668467,
        5.40689332, 10.8992524 ,  4.76969364,  7.09887557, 10.92597196,
       13.27717982,  8.44940661,  5.87132718,  4.79003361,  9.50654967,
        7.50917551,  7.02253235,  7.8788965 , 10.35093485,  5.24137909,
        5.48276038, 12.91505111,  5.53175275, 11.46176259,  8.36445758,
        9.09661212,  6.61386436, 13.91198191, 10.14044489, 11.83570132,
       10.74409834, 13.32613445, 14.03505624, 11.018084  , 12.70489563,
        9.09119351,  6.67121748, 12.67680439, 12.30695098, 12.27835996,
       13.77604293, 10.8207297 , 10.14044489, 11.22142195,  5.95297459,
       11.48434767, 11.72802491,  5.39723241, 13.01187484, 11.69264243,
       12.77402351, 10.63038565,  9.13507321, 10.06850677,  7.46053281,
        9.85330646,  7.13501264,  8.12580976,  9.64157461, 12.5769659 ,
       10.67322527, 12.4013153 ,  9.88571713, 13.92390689,  5.79543038,
        6.4020404 ])}
{'P': tensor([[2.4841e-03, 2.0098e-01, 1.7062e-01,  ..., 2.3675e-07, 1.3065e-07,
         9.5616e-04],
        [1.4826e-05, 5.5650e-03, 4.0992e-05,  ..., 4.0713e-07, 3.8455e-07,
         1.7872e-03],
        [1.7524e-05, 1.6461e-01, 2.8789e-03,  ..., 2.0608e-07, 1.5949e-07,
         1.1131e-03],
        ...,
        [2.0808e-04, 1.9279e-04, 1.6557e-04,  ..., 2.4886e-01, 1.7427e-04,
         1.1752e-04],
        [3.0733e-06, 2.2171e-06, 2.3836e-06,  ..., 3.3437e-04, 5.4819e-02,
         1.0762e-05],
        [4.3349e-04, 8.4179e-04, 6.1040e-04,  ..., 1.2153e-05, 7.7823e-05,
         1.0364e-01]]), 'X': tensor([[ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0646, 0.0422,  ..., 1.0719, 1.0622, 0.1073],
        [0.0646, 0.0000, 0.0300,  ..., 1.0653, 1.1021, 0.1148],
        [0.0422, 0.0300, 0.0000,  ..., 1.0832, 1.0985, 0.1229],
        ...,
        [1.0719, 1.0653, 1.0832,  ..., 0.0000, 0.8056, 0.9648],
        [1.0622, 1.1021, 1.0985,  ..., 0.8056, 0.0000, 0.9924],
        [0.1073, 0.1148, 0.1229,  ..., 0.9648, 0.9924, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.27835996, 12.42554808, 12.35421461, 12.49291768, 12.32545637,
       12.46513646, 12.62711252,  7.38844703, 13.38091121, 10.794755  ,
       13.35013374,  9.69792513,  7.78980299,  6.81084724, 11.75618513,
        8.41171369,  9.53015376,  6.7166297 ,  8.7273569 ,  8.53501777,
       10.06023665,  7.96575782,  7.62011734, 12.53345445,  9.79039371,
       13.00413829,  9.15976746,  9.73579738,  9.45995955,  7.62011734,
        6.16512856, 13.0493368 , 13.56367857, 12.62711252, 12.95702647,
       13.58333857, 11.95743558,  8.05668609, 10.38897563,  4.79003361,
       12.53345445,  8.09984231,  7.33350642, 13.00380624,  8.62162372,
        7.02496223,  5.87305005,  8.95983586,  9.71392179, 12.17451685,
        5.5671389 ,  9.04255655, 11.89855267,  9.13507321,  8.40445872,
       13.04217487,  4.92497514, 10.61047596,  8.38252937,  9.91668182,
       11.8159254 ,  7.89745107,  4.85603886, 13.96905756, 11.75618513,
       10.65929968,  8.0550286 , 10.4606265 ,  8.56889738, 10.91805683,
       13.79676654,  7.62011734, 13.59056613,  4.73784222, 14.02532758,
       12.62576026,  7.66377554,  9.63742669, 11.83075005,  6.54615887,
        8.83341608, 10.52587243, 11.65923396,  5.13182506, 12.0495521 ,
       11.8159254 , 11.64746882,  5.62005629,  6.83515201,  5.5513072 ,
       12.07818871, 11.32149259, 12.46513646,  4.89869205, 11.42420508,
       13.1471313 , 10.8992524 ,  6.00663314,  5.27178363,  6.46356806,
       13.41713661,  5.83560009,  7.5984809 ,  5.05444288,  9.15253344,
        6.53290663])}
{'P': tensor([[4.8482e-03, 3.6919e-01, 2.1074e-01,  ..., 3.8056e-05, 3.3715e-05,
         1.6078e-04],
        [2.9057e-05, 9.2089e-03, 3.8837e-01,  ..., 4.1793e-05, 4.4617e-05,
         5.0868e-04],
        [6.8657e-05, 2.3851e-04, 4.1614e-02,  ..., 1.1309e-04, 2.2661e-04,
         3.0384e-03],
        ...,
        [1.2606e-02, 1.3981e-02, 1.1050e-02,  ..., 8.5228e-02, 2.3652e-04,
         3.7445e-03],
        [2.1867e-04, 2.7781e-04, 5.0818e-04,  ..., 1.2943e-04, 1.8564e-01,
         4.9876e-04],
        [1.5177e-05, 1.8387e-05, 2.1419e-05,  ..., 2.2139e-04, 2.8849e-05,
         9.8912e-02]]), 'X': tensor([[ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0663, 0.1149,  ..., 0.1138, 0.7466, 0.2042],
        [0.0663, 0.0000, 0.1037,  ..., 0.0930, 0.7802, 0.2315],
        [0.1149, 0.1037, 0.0000,  ..., 0.1962, 0.6871, 0.3178],
        ...,
        [0.1138, 0.0930, 0.1962,  ..., 0.0000, 0.8597, 0.1598],
        [0.7466, 0.7802, 0.6871,  ..., 0.8597, 0.0000, 0.9041],
        [0.2042, 0.2315, 0.3178,  ..., 0.1598, 0.9041, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.27991991, 12.41366754, 12.52875793, 12.53047489, 12.50996828,
       12.58006038,  5.5205189 ,  9.7712321 ,  9.87720102,  9.24501829,
       11.9182956 , 10.14044489, 12.3344536 , 11.38774149,  5.94535529,
        5.46244988, 10.05558893,  9.78285424,  7.46053281,  9.44486609,
        9.17595899,  7.03733775,  9.38232121, 13.36092624,  9.43737523,
        9.44486609,  7.39004226, 14.09176259,  7.60463349,  7.57685908,
       13.38091121, 13.76047999, 11.83570132, 13.38091121,  6.00732193,
       11.33837396,  8.85572292,  8.98765644,  8.74470218,  7.32754882,
       11.63383221,  9.95552626,  5.14371005, 10.5575601 ,  6.38105015,
       11.92133129, 13.04156075, 10.67132524, 12.41012046, 13.04217487,
       10.52587243,  5.19577146, 13.40024224,  9.25336371,  6.46560181,
       10.5575601 ,  9.73579738, 10.14436912,  7.13501264, 13.1471313 ,
       12.889695  ,  8.92098752,  5.04028538,  6.39187285,  5.0713364 ,
        8.95983586,  4.91336994,  7.33350642,  7.78980299,  6.2134961 ,
       10.54658597, 10.06023665,  9.09661212,  5.19986797,  9.13507321,
       13.3607379 , 13.15525803, 10.99326577,  7.26074925,  5.07306274,
       13.00413829,  6.61386436,  8.31625171,  5.43544626, 11.87665501,
        9.50654967,  6.76661595, 12.10942412,  9.64157461,  7.92607342,
       11.33837396,  8.29640613, 10.78533838,  5.79543038, 13.15525803,
        8.10886021, 13.14690137,  5.94293185,  9.62238828,  9.51074763,
        9.37989631,  5.10446653, 10.09979747,  5.5748624 ,  6.38105015,
        7.5403822 ])}
{'P': tensor([[6.4300e-03, 3.3645e-01, 2.8536e-01,  ..., 7.0801e-06, 3.4661e-06,
         1.9127e-07],
        [3.4303e-05, 1.9619e-02, 1.8295e-04,  ..., 2.2981e-05, 2.2181e-05,
         3.3535e-07],
        [8.9850e-05, 5.1446e-01, 1.8516e-02,  ..., 1.4221e-05, 1.0434e-05,
         3.5068e-07],
        ...,
        [4.8197e-06, 8.4451e-06, 4.8410e-06,  ..., 6.5508e-02, 1.8121e-02,
         8.2102e-06],
        [4.7363e-07, 8.7641e-07, 5.2866e-07,  ..., 5.5362e-04, 2.5525e-01,
         2.0086e-06],
        [9.0251e-04, 3.5338e-04, 4.0487e-04,  ..., 9.5939e-05, 2.0095e-06,
         4.7239e-02]]), 'X': tensor([[ 11.7796,  -3.4696,   9.0088],
        [ 12.4877,  -0.7222,   8.8347],
        [ 12.2589,  -1.9312,   9.8657],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.3453,  -1.5462,   7.0212],
        [  2.6970,   6.9826,   9.7350],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.6757, -11.0361,  -1.3047],
        [ -1.6220,   7.8917,   0.0315],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.4426, -10.1181,  -5.1268],
        [  6.0560,  -0.9421,  -5.0334],
        [ -4.2271,   7.2387,  -1.8316],
        [  4.1140,  -3.7220,  -0.1273],
        [ -3.9700,   7.3453,  -4.1486],
        [ -8.0087,   4.0173,   9.1091],
        [  2.4639, -10.9431,  -7.3718],
        [ -9.4624,  -0.4633,  -7.3918],
        [  6.1929,   2.4501,  -6.9352],
        [  9.4925,   9.3811,   8.9842],
        [  0.2435,   7.8191,  -6.4709],
        [  8.5716,  10.3596,   6.8818],
        [ 11.7162,   5.6728,  -8.9372],
        [  3.3167,   6.6022,   6.3554],
        [ 11.2767,  -4.6006,   0.7868],
        [ -4.2040,   7.2486,   2.9875],
        [  5.0233,   4.9647,  -3.1788],
        [ -2.6735, -10.4062,  -1.5911],
        [ 11.0275,  -5.0666,   8.4154],
        [ 11.9641,  -2.9568,   7.7211],
        [ -8.0611,  -6.0302,   4.6962],
        [  1.0391,  -4.8141,  -0.0768],
        [  7.0642,  11.6104,  -9.8316],
        [  8.5198,  10.4091,  -8.6486],
        [  4.3377,   5.7570,   1.3712],
        [  3.2809,  13.5061,  -4.0199],
        [  7.0171,  11.6439,  -5.0178],
        [ 12.5224,  -0.4497,   4.8448],
        [  4.1297,  -3.7098,  -3.9838],
        [ -7.8359,   4.2735,   7.2943],
        [ -9.4117,   0.1160,   8.2574],
        [ -6.6998,   5.5954,   0.0823],
        [ -8.8540,  -4.3971,  -5.2144],
        [  1.7815,  -4.7481,   4.4289],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.2883,   0.0325,   5.2692],
        [  0.1677,  -4.7447,  -6.1529],
        [  3.1756, -10.8247,   6.8882],
        [ -8.9624,  -4.0944,  -4.9622],
        [ 11.4703,  -4.2006,   7.4086],
        [  3.3943,   6.5483,   1.0790],
        [ 12.2692,  -1.8882,   3.9253],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -5.3179,  -9.0101,   9.4472],
        [ -5.6651,   6.4380,   1.3660],
        [  3.3167,   6.6022,   6.3554],
        [ -8.0750,  -6.0072,  -5.8267],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.9883,   3.1536,   6.2320],
        [ -3.7964,  -9.9294,   7.7147],
        [ -9.4283,  -1.9585,  -0.9129],
        [  4.2275,  -3.6320,  -0.2714],
        [  6.1579,  -0.5961,  -6.5595],
        [ -7.3470,  -7.0615,   5.0066],
        [  5.3858,   4.4260,  -9.6769],
        [  6.1741,  -0.5329,   3.1913],
        [  4.5800,   5.5034,  -3.9432],
        [  9.7328,   9.0902,  -0.8917],
        [  1.6232,  -4.7717,  -2.5655],
        [ -8.9033,   2.2329,  -8.8912],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.2974,   0.8371,   2.4852],
        [  5.0190,   4.9705,  -7.9752],
        [  6.0305,   3.0296,   0.1375],
        [  6.1222,  -0.7261,   8.8204],
        [  2.3889,  -4.6082,  -8.7291],
        [  9.4925,   9.3811,   8.9842],
        [  5.9587,  12.3216,   9.1695],
        [  1.3767, -11.0341,   9.9415],
        [ 12.1940,  -2.1881,   7.1499],
        [  3.1892, -10.8220,  -6.8409],
        [  3.2366,  -4.2732,   5.1532],
        [  3.9591, -10.6391,  -6.2557],
        [  2.1824,  -4.6647,   1.1985],
        [ -9.2687,  -2.9794,   0.6436],
        [  0.5175,  14.0910,   4.6540],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -7.6293,  -6.6859,  -5.5161],
        [  4.0421,  -3.7765,  -8.3631],
        [  9.9750,  -6.6338,  -1.2440],
        [ 12.5770,   1.8512,  -0.6578],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.3705,   0.4208,  -3.5815],
        [  7.7750,  11.0665,   2.8364],
        [  2.5636, -10.9293,   9.6519],
        [ -5.9940,   6.1985,   5.9566],
        [  1.4489,   7.5256,   4.4529],
        [  3.4140,  -4.1803,  -8.8224],
        [  0.1287, -11.0065,   6.5757],
        [  6.9433,  -9.3362,  -0.3421],
        [  6.9433,  -9.3362,  -0.3421],
        [  5.6022,  -1.9833,   8.3081],
        [  5.8201,   3.5841,  -8.0609],
        [  5.8930,  12.3592,  -8.9201],
        [ -0.9715, -10.8630,   0.3555]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 1.1906e-01, 1.0292e-01,  ..., 1.1100e+00, 1.3453e+00,
         1.0158e+00],
        [1.1906e-01, 0.0000e+00, 1.6153e-02,  ..., 1.1302e+00, 1.3401e+00,
         1.1136e+00],
        [1.0292e-01, 1.6153e-02, 0.0000e+00,  ..., 1.1273e+00, 1.3408e+00,
         1.1004e+00],
        ...,
        [1.1100e+00, 1.1302e+00, 1.1273e+00,  ..., 0.0000e+00, 3.4403e-01,
         8.4536e-01],
        [1.3453e+00, 1.3401e+00, 1.3408e+00,  ..., 3.4403e-01, 0.0000e+00,
         1.1871e+00],
        [1.0158e+00, 1.1136e+00, 1.1004e+00,  ..., 8.4536e-01, 1.1871e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.27993098, 12.5085995 , 12.41012046, 12.62576026, 12.71843462,
       12.44177641,  7.48538905, 10.7167285 , 11.05672136,  8.05668609,
       10.60226614, 11.4890861 ,  6.12886024,  8.38252937,  5.54776934,
        8.34949383,  8.95983586, 11.21703654,  9.47369628,  6.65991877,
       13.34587027,  7.822854  , 13.44593519, 13.01728313,  7.38844703,
       12.17901033,  8.37954853,  7.06272256, 10.74409834, 12.13568598,
       12.32408516, 10.06703358,  4.92497514, 13.59056613, 13.45124917,
        7.20826219, 13.89886469, 13.59480906, 12.53047489,  5.5513072 ,
        8.92549752,  9.41244865,  8.72896381,  9.88571713,  5.0713364 ,
        5.22903117,  6.28835101,  4.74770899, 11.28093507,  9.85330646,
       12.21532282,  7.37576154, 12.41366754,  9.88310614, 10.46238054,
        8.57560275,  7.38844703, 10.06437598,  5.39723241,  6.76791483,
       10.63038565,  9.62958973,  5.57341623,  6.18668463, 10.19036081,
        6.97106912,  6.19707944,  7.15990676, 13.31764181,  5.04028538,
        9.17905491,  9.66977802,  9.3349858 ,  7.06372172,  6.74872981,
        6.16512856,  5.19065439, 13.34587027, 13.68672855, 11.11969815,
       12.38881635, 11.28215429,  5.36061974, 11.35182589,  5.14999799,
        9.73579738, 14.10045797,  9.43737523, 10.14436912,  5.53175275,
       11.97949273, 12.7125079 ,  8.29203356,  9.37989631, 13.5247215 ,
       11.22596994,  8.62260327,  7.66377554,  5.39723241, 11.00726483,
       11.63502679, 11.63502679,  5.94293185,  6.83515201, 13.6922425 ,
       10.90638066])}
{'P': tensor([[3.1291e-03, 3.5492e-01, 2.7344e-01,  ..., 2.3305e-04, 5.5698e-05,
         2.0251e-07],
        [1.4961e-05, 3.9918e-03, 4.9743e-01,  ..., 3.9352e-04, 4.6755e-05,
         2.8973e-07],
        [2.2598e-05, 3.6800e-05, 8.0714e-03,  ..., 9.5751e-04, 7.3024e-05,
         6.4612e-07],
        ...,
        [3.2787e-05, 3.7568e-05, 4.1496e-05,  ..., 1.3894e-01, 3.7818e-04,
         2.1194e-03],
        [3.5585e-02, 4.0973e-02, 4.3774e-02,  ..., 2.3969e-02, 2.1425e-01,
         1.8290e-04],
        [2.0381e-05, 1.6012e-05, 1.4013e-05,  ..., 5.3952e-05, 4.4114e-04,
         1.5024e-01]]), 'X': tensor([[ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.6661e-02, 8.6130e-02,  ..., 2.0723e-01, 2.1125e-01,
         9.2699e-01],
        [4.6661e-02, 3.0339e-04, 4.0196e-02,  ..., 2.3616e-01, 2.1726e-01,
         9.7147e-01],
        [8.6130e-02, 4.0196e-02, 0.0000e+00,  ..., 2.5751e-01, 2.2004e-01,
         1.0058e+00],
        ...,
        [2.0723e-01, 2.3616e-01, 2.5751e-01,  ..., 0.0000e+00, 1.1635e-01,
         7.7577e-01],
        [2.1125e-01, 2.1726e-01, 2.2004e-01,  ..., 1.1635e-01, 0.0000e+00,
         8.8051e-01],
        [9.2699e-01, 9.7147e-01, 1.0058e+00,  ..., 7.7577e-01, 8.8051e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.30695098, 12.38881635, 12.44177641, 12.45418236, 12.62576026,
       12.71843462, 10.71162681,  7.53063885, 12.27835996, 12.35421461,
       11.9120321 , 11.51557166,  9.13507321, 10.42667093, 13.40024224,
        7.32754882, 11.57249984,  9.77588801,  9.43737523,  5.63824971,
        6.38105015,  5.05444288, 11.63724792, 13.59170809, 11.20100795,
        8.53501777,  7.37902766,  8.79284466, 12.04924159, 10.14436912,
        5.24461068,  8.68979052,  7.28793663, 10.04916678,  9.3109884 ,
        9.55720651,  6.00732193,  6.38968357, 12.84056604,  5.13125513,
        9.11473391,  6.3949043 ,  8.29022197,  9.94173708,  7.81509631,
       12.87752052, 11.95743558, 13.0743399 ,  8.04483495, 12.17901033,
       13.89974693, 13.36092624,  5.39450305,  8.62631801, 14.05687815,
       10.61472636,  8.4486771 , 11.72802491, 10.24557047,  5.08910491,
        8.31625171,  6.84796202, 10.53368644,  8.09899874,  5.87305005,
       12.64235359, 11.87665501, 11.42947632, 14.02532758,  7.57946459,
       13.68063098,  6.67218687,  7.5158489 ,  5.13616224,  5.26067019,
        5.37343173,  9.34061938, 11.54071859,  6.38968357, 11.00508681,
        5.26342203, 10.88618114, 10.46238054, 12.42554808, 11.32149259,
       10.53368644,  5.0713364 ,  7.72291946, 13.01728313, 13.32645552,
        7.98469166, 11.018084  , 11.83570132,  8.99276564,  5.05981914,
        7.32233733,  7.92644746,  6.84796202,  8.87587457, 13.38091121,
        4.73784222,  6.38399789,  9.75075273,  7.38844703,  5.76165934,
        9.3349858 ])}
{'P': tensor([[4.9694e-03, 2.9103e-01, 1.6335e-01,  ..., 2.1266e-06, 1.0537e-03,
         1.8750e-04],
        [2.3359e-05, 4.9962e-03, 2.5994e-01,  ..., 3.1912e-06, 8.9395e-04,
         3.6493e-04],
        [7.2341e-06, 1.3788e-05, 6.6952e-03,  ..., 7.3551e-06, 4.2810e-04,
         1.6936e-03],
        ...,
        [5.5001e-06, 5.1347e-06, 4.1083e-06,  ..., 1.1599e-01, 9.2420e-05,
         1.2577e-05],
        [8.7792e-04, 1.6132e-03, 4.1156e-03,  ..., 1.0482e-03, 9.4814e-02,
         6.3413e-03],
        [4.3323e-08, 5.4178e-08, 8.9832e-08,  ..., 2.6573e-04, 1.3963e-06,
         3.8800e-03]]), 'X': tensor([[ 11.9019,  -3.1379,  -3.4534],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.5919,   0.4035,  -5.8460],
        [  3.9591, -10.6391,  -6.2557],
        [  6.3460,   1.4252,   1.2291],
        [ -8.8631,  -4.3728,  -1.1746],
        [  4.4049,   5.6892,  -4.4394],
        [  6.3301,   0.3638,  -7.5685],
        [  5.4336,  -2.2676,   5.6933],
        [ 12.5919,   0.4035,  -5.8460],
        [ -2.6735, -10.4062,  -1.5911],
        [ -8.0087,   4.0173,   9.1091],
        [ 12.6035,   0.7494,   7.1536],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.1960,  -3.2964,   1.6921],
        [  0.6142,  14.0802,   0.5407],
        [  6.0305,   3.0296,   0.1375],
        [ -7.8849,  -6.3115,  -4.7479],
        [  6.3534,   1.2990,  -7.2942],
        [  0.2480, -11.0153,  -3.0183],
        [  4.4915,   5.5990,   7.0405],
        [  4.5340,  -3.3629,   8.9652],
        [  4.1637,  13.1898,  -1.9189],
        [  0.2435,   7.8191,  -6.4709],
        [ -2.1817,   7.8274,  -4.6298],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -8.6112,   2.9319,  -7.9943],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.2182,  13.8007,   5.1889],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.0965,  -5.9712,   4.3466],
        [ -1.6334, -10.7207,   7.1397],
        [  5.5424,  -2.0885,   7.2487],
        [  1.3339,   7.5627,  -0.9034],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.3516,   0.6425,   0.3425],
        [  5.8889,  -1.3916,   8.8993],
        [ -7.4887,   4.7364,   8.9074],
        [ -9.4351,  -1.8891,   3.5642],
        [  3.3943,   6.5483,   1.0790],
        [ -8.1204,  -5.9308,   4.0761],
        [ -3.2486,   7.5925,  -1.3719],
        [ -9.3444,   0.5821,   6.9101],
        [  2.4639, -10.9431,  -7.3718],
        [  4.5503,   5.5359,   7.4735],
        [  6.2535,  -0.1709,  -9.4102],
        [  7.9141,  -8.6656,   9.6948],
        [  6.0140,  -1.0661,  -2.5040],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.4573,   3.2448,   5.4370],
        [ -5.7347,  -8.6908,   9.2878],
        [  3.3094,  13.4969,  -3.6507],
        [  2.5620,   7.0549,   8.4825],
        [  7.8375,  -8.7240,  -9.8913],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -6.3066,  -8.1936,   9.5822],
        [ -6.0844,  -8.3955,   0.7705],
        [ -1.9600,   7.8577,   6.9121],
        [  1.9241,   7.3503,   7.9004],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.3460,   1.4252,   1.2291],
        [  5.5294,  -2.1106,   9.0626],
        [  1.1994,   7.6035,   2.4295],
        [ -7.8359,   4.2735,   7.2943],
        [  1.8721,  -4.7322,   0.5524],
        [  5.4426, -10.1181,  -5.1268],
        [  6.3550,   0.7129,   0.4917],
        [  6.8093,  -9.4178,   8.4244],
        [ 11.1469,   6.9237,   1.6160],
        [  6.0345,   3.0175,  -1.4424],
        [  5.2188,  -2.5839,   2.1833],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -9.4157,   0.0821,  -5.0025],
        [ -9.3419,   0.5972,  -1.2841],
        [ 12.4597,  -0.9136,  -4.8763],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.9788,   2.0208,  -3.2867],
        [  1.6308,  13.9254,  -2.0984],
        [ -8.8835,  -4.3178,   9.3080],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.4524,  -1.6813,  -6.6862],
        [  7.0642,  11.6104,  -9.8316],
        [  2.4932,   7.0903,  -5.4214],
        [  9.9217,  -6.7006,  -5.6728],
        [ -4.6942,  -9.4299,  -9.1846],
        [  9.1620,  -7.5573,  -9.9882],
        [  1.4749,  -4.7892,  -0.8089],
        [ -1.9600,   7.8577,   6.9121],
        [ 10.5678,   7.9261,  -0.6100],
        [  4.1140,  -3.7220,  -0.1273],
        [  7.5064,  -8.9651,  -0.0994],
        [  5.1861,   4.7361,  -3.7312],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.1504,   4.3674,  -7.2998],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.3236,   0.3011,  -4.6668],
        [ -7.8359,   4.2735,   7.2943],
        [  4.6992,   5.3690,   3.0935],
        [  1.6232,  -4.7717,  -2.5655],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.3515,   0.6418,  -6.5823],
        [  4.9337,  12.8566,  -3.6582]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0625, 0.1147,  ..., 0.6253, 0.0485, 0.2481],
        [0.0625, 0.0000, 0.0781,  ..., 0.6659, 0.0436, 0.1972],
        [0.1147, 0.0781, 0.0000,  ..., 0.7381, 0.1199, 0.1360],
        ...,
        [0.6253, 0.6659, 0.7381,  ..., 0.0000, 0.6224, 0.8621],
        [0.0485, 0.0436, 0.1199,  ..., 0.6224, 0.0000, 0.2403],
        [0.2481, 0.1972, 0.1360,  ..., 0.8621, 0.2403, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.30858677, 12.4217523 , 12.62711252, 12.49317563, 12.67576792,
       12.59840193, 11.35182589,  6.50409372,  9.88310614,  7.19513806,
        6.34059021,  5.88782913, 12.59840193, 10.74409834,  8.95983586,
       12.62576026,  8.27853076,  9.76896904, 14.09357547,  6.74872981,
       10.09979747,  6.48486171, 11.018084  ,  7.17789666,  5.64500512,
       13.83139024,  7.822854  ,  8.12580976, 12.27835996,  9.09661212,
        8.53501777, 13.97779839, 13.89974693, 10.06023665, 10.84437705,
        5.92281142,  7.67939546,  5.48276038,  6.38399789,  6.05113217,
        8.86082124,  9.62238828,  7.37576154, 10.05558893,  8.25828409,
        9.36256096, 11.21703654,  7.16598273,  6.25586297, 11.73567759,
        6.10774544,  9.37989631,  9.05843201, 10.412299  , 13.89671314,
        7.50564223, 11.72749923,  9.58072286, 10.33958639, 10.36844813,
        8.09842704,  7.59795445, 10.05558893,  6.50409372,  5.91854397,
        7.69752262,  8.92549752,  5.08910491, 11.4890861 ,  6.3949043 ,
       11.62158554, 13.12218013,  6.74686071,  5.82345867,  9.5642993 ,
        9.41605684,  9.36093562, 12.49317563,  5.62005629,  9.20340031,
       14.02058621,  9.87720102,  9.10945678,  9.60080639, 13.59056613,
        7.5158489 , 11.97238624, 10.53368644, 11.87665501,  5.01114719,
        8.09842704, 13.20988996,  5.54776934, 11.69264243,  7.02326515,
        7.8788965 , 12.91143491,  9.50654967, 12.10222204,  6.33075682,
        8.92549752,  7.13501264,  5.04028538,  8.62162372,  6.383897  ,
       13.77075174])}
{'P': tensor([[4.4182e-03, 2.3958e-01, 1.9833e-01,  ..., 1.3843e-03, 1.1856e-07,
         4.1416e-05],
        [3.5141e-05, 7.4480e-03, 3.3957e-05,  ..., 1.7407e-03, 1.9616e-07,
         3.9849e-05],
        [3.5946e-05, 4.6049e-01, 3.6954e-03,  ..., 1.0262e-03, 9.7930e-08,
         2.5114e-05],
        ...,
        [9.4579e-04, 1.7732e-03, 1.3341e-03,  ..., 9.3295e-02, 4.8505e-05,
         1.4080e-03],
        [8.5304e-05, 5.1722e-05, 6.4874e-05,  ..., 1.6775e-04, 5.7910e-02,
         1.3969e-03],
        [1.0093e-02, 9.7508e-03, 9.8201e-03,  ..., 4.7536e-02, 3.8549e-05,
         5.4043e-02]]), 'X': tensor([[ 11.9641,  -2.9568,   7.7211],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.4877,  -0.7222,   8.8347],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.3759,  -1.3943,   6.4659],
        [  2.0243,   7.3086,   7.5432],
        [  6.2883,   0.0325,   5.2692],
        [ -8.3062,  -5.6008,   0.0287],
        [  6.9314,  -9.3435,  -1.2729],
        [  1.7301,   7.4262,   5.0522],
        [  0.8700,  -4.8132,  -7.8042],
        [  8.6271,  -8.0715,  -7.4813],
        [  3.7992,  -3.9474,  -8.9019],
        [ 10.1798,   8.5002,  -5.8893],
        [  1.7815,  -4.7481,   4.4289],
        [  6.1730,  -0.5375,  -8.8861],
        [ -7.2515,   5.0195,   4.6434],
        [  5.1472,  -2.6801,   8.1194],
        [  5.5424,  -2.0885,   7.2487],
        [  1.7232,  -4.7574,   3.1270],
        [  1.8320,   7.3871,   5.1452],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 11.5456,   6.0861,   8.1101],
        [  6.0031,  12.2958,  -6.5992],
        [  1.2728, -11.0372,   9.8121],
        [ -3.2486,   7.5925,  -1.3719],
        [  8.0877,  -8.5297,  -1.1050],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.4932,   7.0903,  -5.4214],
        [  6.3534,   1.2990,  -7.2942],
        [  1.4150,  13.9647,  -4.0915],
        [  6.2883,   0.0325,   5.2692],
        [ -6.5295,  -7.9785,  -7.4429],
        [  5.4867,  12.5816,   2.9549],
        [  3.6703,   6.3443,  -5.4401],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.2188,  -2.5839,   2.1833],
        [  9.0051,   9.9234,   5.4411],
        [  6.1579,  -0.5961,  -6.5595],
        [  5.5626,  -2.0536,   8.4433],
        [ -9.4739,  -1.2497,   0.5483],
        [ -6.1914,  -8.2997,   8.5380],
        [  6.3301,   0.3638,  -7.5685],
        [ 10.3251,   8.2928,  -3.0006],
        [ 12.5650,   2.0132,   4.6272],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.3087,   0.7792,  -7.5587],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.6801,   2.7809,  -0.8443],
        [  5.9883,   3.1536,   6.2320],
        [  2.2860,   7.1916,  -6.1011],
        [  6.1741,  -0.5329,   3.1913],
        [ -2.6735, -10.4062,  -1.5911],
        [ 12.5650,   2.0132,   4.6272],
        [  2.7121,   6.9743,  -8.4202],
        [ -5.1377,   6.7751,  -0.4438],
        [ -1.6220,   7.8917,   0.0315],
        [  5.9843,  -9.8679,  -6.1449],
        [ -0.2586,   7.8825,  -9.9840],
        [  1.3670,  -4.7990,   3.2603],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.9314,  -9.3435,  -1.2729],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.5789,   0.1722,   1.6470],
        [  8.5716,  10.3596,   6.8818],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -7.8126,   4.3066,   5.1791],
        [ -6.3066,  -8.1936,   9.5822],
        [  3.0876,   6.7527,  -5.3540],
        [  2.7558,  -4.4844,   0.0636],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -7.4887,   4.7364,   8.9074],
        [ -7.9298,   4.1367,  -3.9773],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.0457,  -3.7737,  -0.4704],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.4356,  -4.5942,  -5.0808],
        [ -1.8788, -10.6567,   3.7068],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 11.5966,   5.9666,   6.0083],
        [  4.9337,  12.8566,  -3.6582],
        [  6.8093,  -9.4178,   8.4244],
        [  6.3301,   0.3638,  -7.5685],
        [  2.5318,  13.7231,   6.3692],
        [  6.3413,   0.4883,  -6.8416],
        [  5.1861,   4.7361,  -3.7312],
        [  5.3602,  -2.3808,   7.2098],
        [ -3.2053, -10.1993,  -3.8736],
        [  2.8474,   6.8979,   6.6457],
        [  1.9241,   7.3503,   7.9004],
        [  6.1372,   2.6726,  -6.7885],
        [ -0.0410,   7.8591,   9.3457],
        [  0.6952,  -4.8060,   8.8976],
        [  6.3124,   0.2057,  -0.4593],
        [ -7.8359,   4.2735,   7.2943],
        [  9.9217,  -6.7006,  -5.6728],
        [  2.2182,  13.8007,   5.1889],
        [ -1.0012, -10.8576,   2.2867],
        [  7.9141,  -8.6656,   9.6948],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.2435,   4.0076,   8.1257],
        [  6.3569,   0.7591,   6.5011],
        [ -8.0330,  -6.0765,   6.7945],
        [  4.5340,  -3.3629,   8.9652]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.2801e-02, 2.2775e-02,  ..., 2.2367e-01, 9.2517e-01,
         1.4477e-01],
        [6.2801e-02, 0.0000e+00, 4.0196e-02,  ..., 2.1606e-01, 9.8561e-01,
         1.4399e-01],
        [2.2775e-02, 4.0196e-02, 3.0339e-04,  ..., 2.1670e-01, 9.4635e-01,
         1.3910e-01],
        ...,
        [2.2367e-01, 2.1606e-01, 2.1670e-01,  ..., 0.0000e+00, 9.4716e-01,
         7.9101e-02],
        [9.2517e-01, 9.8561e-01, 9.4635e-01,  ..., 9.4716e-01, 0.0000e+00,
         9.2888e-01],
        [1.4477e-01, 1.4399e-01, 1.3910e-01,  ..., 7.9101e-02, 9.2888e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.32408516, 12.44177641, 12.38881635, 12.5085995 , 12.62576026,
       12.45418236,  7.58378655,  6.28835101, 10.01804223, 11.63383221,
        7.62509582,  4.89121805, 11.81423161,  5.47866221, 13.26209545,
        5.0713364 ,  6.19633139,  8.81930525,  5.80312949,  5.92281142,
        5.05989547,  7.61088125,  6.21275016,  9.78286733, 13.05148826,
       13.6829843 , 11.11038551,  8.25828409, 11.75438006, 10.5575601 ,
        7.5158489 ,  6.48486171, 14.03618649,  6.28835101, 10.30972193,
       13.72594022,  7.32950385, 12.59840193,  5.82345867, 13.40024224,
        6.18668463,  5.92952538,  9.55593374, 10.35464916,  6.34059021,
       13.2430377 , 12.72524615,  6.69389431,  9.34127013,  8.41148722,
        9.11473391,  6.76791483,  7.54620849,  6.19707944, 10.74409834,
       12.72524615,  7.48311002,  8.50278896,  8.05668609, 11.54071859,
        7.88677538,  4.98989906, 10.39942497, 11.63383221, 12.30858677,
       12.58006038, 13.44593519, 10.06407839,  8.92098752, 10.33958639,
        7.42512621,  5.26342203, 12.11735334,  8.86082124,  8.94395628,
       12.70489563,  5.53255882, 12.00995898,  5.19986797, 10.82106505,
        9.62958973, 13.04156075, 13.77075174, 11.62158554,  6.34059021,
       13.95472331,  6.36004393,  7.02326515,  5.86519525, 10.6910768 ,
        7.46248947,  7.59795445,  6.69389431,  7.85919687,  4.85603886,
        6.31576264,  8.92549752, 11.97238624, 13.97779839, 10.9036226 ,
       11.73567759, 12.35625694, 12.88270018,  6.4020404 , 10.07238463,
        5.64500512])}
{'P': tensor([[3.1424e-03, 2.7900e-01, 2.0873e-01,  ..., 1.3696e-06, 1.1751e-04,
         7.6666e-07],
        [1.2439e-05, 3.2454e-03, 4.0761e-01,  ..., 1.0379e-06, 1.3672e-04,
         9.6833e-07],
        [1.2221e-05, 1.8889e-05, 4.8388e-03,  ..., 1.3012e-06, 2.2388e-04,
         1.7035e-06],
        ...,
        [9.7689e-04, 7.7754e-04, 6.8816e-04,  ..., 1.2267e-01, 8.7233e-04,
         2.2751e-04],
        [5.1226e-07, 7.3384e-07, 9.7341e-07,  ..., 3.5568e-06, 1.4582e-02,
         2.3559e-04],
        [2.8211e-06, 2.3951e-06, 2.2122e-06,  ..., 5.2824e-04, 4.3511e-06,
         8.6929e-02]]), 'X': tensor([[ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0441, 0.0751,  ..., 0.8244, 1.0629, 0.9229],
        [0.0441, 0.0000, 0.0316,  ..., 0.8408, 1.0629, 0.9580],
        [0.0751, 0.0316, 0.0000,  ..., 0.8466, 1.0566, 0.9786],
        ...,
        [0.8244, 0.8408, 0.8466,  ..., 0.0000, 0.4246, 0.4852],
        [1.0629, 1.0629, 1.0566,  ..., 0.4246, 0.0000, 0.8984],
        [0.9229, 0.9580, 0.9786,  ..., 0.4852, 0.8984, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.32545637, 12.42554808, 12.49291768, 12.59840193, 12.3828957 ,
       12.64235359,  8.24808832, 12.30858677,  9.85330646,  6.65991877,
       13.00413829,  9.95552626, 13.89264822, 13.00413829,  5.94535529,
        5.27178363,  9.76896904,  8.62162372,  8.16714836,  5.64500512,
        8.62329466, 10.31613269,  8.8580671 , 10.91805683, 11.20100795,
        7.85919687,  5.94541803,  7.15100303, 11.63765273,  6.81084724,
        8.86082124,  5.79543038, 13.15525803,  9.34127013,  9.49511524,
        6.34059021,  7.02496223, 11.27160035, 11.9120321 , 12.40074901,
       13.89264822, 13.66896739, 11.36089355, 13.12299145, 13.18278606,
       10.48315436,  6.69389431,  6.46356806,  5.18277858,  5.70405525,
        5.70591986, 13.5247215 , 12.00995898,  7.9337647 ,  9.05843201,
        7.81509631, 10.8207297 ,  5.14124612, 11.11969815,  5.19577146,
        5.18047638, 10.6910768 , 12.59969936,  8.06643487,  9.78286733,
       13.6105595 , 11.22142195, 12.7125079 ,  5.98285319, 13.89671314,
       10.0674887 , 11.64746882,  5.05444288,  8.56889738, 10.32715715,
        9.14973319, 12.85032836,  5.19065439, 12.64235359,  7.5403822 ,
       13.68063098,  9.96391045,  9.23178204,  7.26074925, 11.20100795,
       12.30858677,  5.63824971, 13.88683436,  6.42439344,  6.99097013,
       10.31613269,  5.91408504,  6.49538841,  7.83486943,  9.7751189 ,
        5.1224079 ,  9.26455795, 12.05300943,  8.68979052,  9.78147542,
        8.09984231,  7.50500064, 11.81423161,  5.0713364 , 13.2778378 ,
        8.66273171])}
{'P': tensor([[5.0602e-03, 3.9211e-01, 1.7475e-01,  ..., 8.3968e-06, 6.3349e-07,
         2.0828e-06],
        [3.1418e-05, 9.1611e-03, 1.7678e-03,  ..., 1.0923e-05, 5.8640e-07,
         4.9467e-06],
        [5.5532e-05, 2.4956e-02, 7.8216e-03,  ..., 1.0008e-05, 4.5774e-07,
         3.6622e-06],
        ...,
        [1.9686e-02, 1.7084e-02, 1.9214e-02,  ..., 2.3505e-01, 7.7228e-05,
         9.9603e-05],
        [2.2197e-03, 1.8549e-03, 1.4218e-03,  ..., 4.7432e-05, 3.0723e-01,
         3.8156e-05],
        [3.4800e-06, 2.9429e-06, 2.7487e-06,  ..., 6.8945e-05, 1.0124e-05,
         9.4840e-02]]), 'X': tensor([[ 11.9936,  -2.8677,   6.0614],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.5224,  -0.4497,   4.8448],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.5717,   1.9266,   7.4903],
        [  6.9314,  -9.3435,  -1.2729],
        [ -8.5900,   2.9767,   7.4811],
        [ -1.6275, -10.7221,   3.2376],
        [  7.3592,  11.3937,   6.2407],
        [  5.9883,   3.1536,   6.2320],
        [ -6.0844,  -8.3955,   0.7705],
        [  0.2480, -11.0153,  -3.0183],
        [  4.7482, -10.3907,   5.9149],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -3.2053, -10.1993,  -3.8736],
        [  5.8325,  -1.5222,  -2.1341],
        [  2.9410,   6.8428,   5.4458],
        [  7.3724,  -9.0577,  -8.9843],
        [  1.1644, -11.0394,   7.8511],
        [  0.6933,   7.7336,   2.1179],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.4039,   5.6903,   8.7366],
        [ -3.5520,   7.4976,   9.1522],
        [  3.9591, -10.6391,  -6.2557],
        [  3.9591, -10.6391,  -6.2557],
        [ 11.5198,  -4.0923,   5.5088],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.1752,   2.5242,   3.5874],
        [  3.2990,  -4.2415,  -7.4857],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.6078, -10.9229,   6.2869],
        [  1.6422,   7.4586,  -6.8056],
        [  6.8407,  11.7666,   3.0045],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.3535,   0.6792,  -7.5101],
        [  9.0129,  -7.7072,   9.1400],
        [  6.9284,  11.7061,  -0.3732],
        [  0.2435,   7.8191,  -6.4709],
        [ -9.4524,  -1.6813,  -6.6862],
        [  5.8979,   3.3958,  -3.8713],
        [ -5.2589,   6.7024,  -5.5809],
        [ -1.6275, -10.7221,   3.2376],
        [  5.8450,  -9.9356,   3.0767],
        [  6.0305,   3.0296,   0.1375],
        [  9.2811,  -7.4336,   5.1771],
        [  0.5850,  -4.7982,   9.6639],
        [  6.9433,  -9.3362,  -0.3421],
        [ -6.1899,  -8.3011,  -7.9269],
        [  0.9945,  -4.8144,  -0.1897],
        [ -9.1459,   1.4780,  -4.2403],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.8934,   2.2597,   3.5831],
        [ -0.5739,   7.9056,   0.6382],
        [  5.5626,  -2.0536,   8.4433],
        [  9.5261,  -7.1675,   1.4206],
        [  6.0467,  -9.8369,  -6.5816],
        [  4.0591,   6.0202,  -1.6339],
        [  6.2555,   2.1474,   4.1685],
        [ -6.2200,  -8.2737,   5.5079],
        [  3.3167,   6.6022,   6.3554],
        [  9.1409,  -7.5789,   1.5814],
        [ -8.6112,   2.9319,  -7.9943],
        [ -8.0608,   3.9362,  -5.9567],
        [ -3.4986,   7.5152,   9.0712],
        [  5.4179,  -2.2925,  -3.4641],
        [  9.4925,   9.3811,   8.9842],
        [ 11.4235,   6.3594,   5.1663],
        [ -9.4714,  -1.3301,  -3.1998],
        [  7.1441,  11.5529,   2.3232],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -8.7544,   2.6096,   3.3146],
        [ -4.2271,   7.2387,  -1.8316],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.3589,   0.8239,  -8.4230],
        [  3.0719, -10.8449,  -4.1673],
        [ -9.4196,   0.0478,   9.4140],
        [  2.6451,  -4.5250,  -0.5565],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.6000,   1.3839,  -3.2992],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.2974,   0.8371,   2.4852],
        [  1.0391,  -4.8141,  -0.0768],
        [  8.3791,  10.5409,   5.5590],
        [ -8.5506,  -5.1154,   6.3423],
        [  3.3633,  13.4794,   0.5193],
        [  8.5716,  10.3596,   6.8818],
        [  0.1382, -11.0073,   1.0026],
        [  0.1532,  -4.7422,  -7.0145],
        [  6.1741,  -0.5329,   3.1913],
        [  5.4167, -10.1293,   6.2156],
        [ -7.6525,  -6.6534,   5.1721],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -9.3444,   0.5821,   6.9101],
        [  6.8093,  -9.4178,   8.4244],
        [ -8.7409,  -4.6837,  -6.5840],
        [  1.7228,  -4.7575,   8.3396],
        [ -7.0825,   5.2073,  -5.0497],
        [  0.2571,  -4.7594,   1.5618],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.3860,  -2.3418,   1.0826],
        [  2.2644,  -4.6434,   9.7508],
        [  8.0381,  -8.5690,  -8.9523],
        [ -5.6174,   6.4708,  -1.7339]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0735, 0.0897,  ..., 0.0650, 1.1649, 0.8872],
        [0.0735, 0.0000, 0.0341,  ..., 0.1263, 1.2070, 0.9523],
        [0.0897, 0.0341, 0.0000,  ..., 0.1505, 1.2393, 0.9762],
        ...,
        [0.0650, 0.1263, 0.1505,  ..., 0.0000, 1.1005, 0.8263],
        [1.1649, 1.2070, 1.2393,  ..., 1.1005, 0.0000, 0.5595],
        [0.8872, 0.9523, 0.9762,  ..., 0.8263, 0.5595, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.33167382, 12.45418236, 12.44177641, 12.53047489, 12.62576026,
       12.71843462, 11.63383221,  9.09119351, 10.84493732, 13.56367857,
        6.76791483, 10.36844813, 11.018084  , 11.42420508, 12.0495521 ,
        9.79039371, 10.6910768 ,  6.02789634,  7.44805901, 11.67875855,
       11.10066126,  7.76457095,  8.5724042 ,  7.19534146,  8.29640613,
       11.35182589, 11.35182589, 12.22503754, 12.91505111,  6.67121748,
        5.37343173,  9.55720651, 11.22993218,  7.63726134, 13.6105595 ,
        5.70405525,  6.38968357, 11.85890664, 13.60275211,  7.822854  ,
        9.60080639,  6.80559669,  8.51928531, 10.84493732, 11.52733581,
        6.74872981, 11.89105259,  4.83371341, 11.63502679, 10.35485331,
        4.9160962 ,  9.26455795, 11.83509372,  9.17595899,  7.92644746,
        5.92952538, 11.92133129, 11.54672755,  7.26074925,  6.61386436,
       10.35093485,  7.38844703, 11.87412439,  9.09661212,  8.97052115,
        8.28968067,  5.8828986 , 13.34587027, 13.0743399 ,  9.5642993 ,
       13.58333857,  9.78147542,  9.13507321,  8.38252937,  5.18047638,
        6.41202986, 11.27160035,  9.41970031,  5.24137909, 12.84056604,
       12.67576792, 10.54658597, 12.04924159,  9.3349858 ,  4.92497514,
       13.46553623,  9.96391045, 13.89264822, 13.44593519, 11.00812996,
        4.74468406,  6.19707944, 11.48663767, 10.14044489,  9.99145096,
        9.36256096, 11.62158554,  9.91668182,  5.05981914,  8.7907923 ,
        4.7663526 , 10.53368644,  5.87305005,  5.16611782, 11.74901778,
        8.56889738])}
{'P': tensor([[4.2613e-03, 2.5807e-01, 1.8145e-01,  ..., 3.9493e-02, 1.7200e-05,
         5.0626e-05],
        [4.4256e-05, 2.0981e-02, 4.7281e-03,  ..., 2.2140e-01, 1.0878e-04,
         1.2371e-04],
        [1.8079e-05, 8.2704e-03, 5.7216e-03,  ..., 1.6676e-01, 4.7399e-05,
         2.6924e-05],
        ...,
        [5.3757e-05, 2.3320e-04, 1.2347e-04,  ..., 8.0114e-02, 1.7346e-03,
         1.0280e-04],
        [5.6616e-06, 7.3637e-06, 9.3067e-06,  ..., 2.2366e-05, 9.4667e-02,
         5.0562e-06],
        [3.6643e-03, 6.0576e-03, 3.6519e-03,  ..., 2.7296e-03, 3.8188e-05,
         1.2565e-01]]), 'X': tensor([[ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0891, 0.1101,  ..., 0.2365, 0.7095, 0.3157],
        [0.0891, 0.0000, 0.1460,  ..., 0.2400, 0.7735, 0.2772],
        [0.1101, 0.1460, 0.0000,  ..., 0.1310, 0.6288, 0.4151],
        ...,
        [0.2365, 0.2400, 0.1310,  ..., 0.0000, 0.5936, 0.5169],
        [0.7095, 0.7735, 0.6288,  ..., 0.5936, 0.0000, 1.0217],
        [0.3157, 0.2772, 0.4151,  ..., 0.5169, 1.0217, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.3344536 , 12.52875793, 12.50996828, 12.58006038, 12.72524615,
       12.41366754, 10.00410625,  9.18358562, 12.4013153 ,  8.01009083,
        6.89245352,  7.19534146,  8.8737556 ,  9.13507321,  8.99791316,
       12.41366754,  8.859416  , 13.96905756,  9.20340031, 11.21703654,
        5.84450127, 10.24557047,  4.91336994, 13.01728313,  6.7166297 ,
       11.75807252,  5.36061974,  7.62232307, 12.79126785,  6.7166297 ,
       12.5769659 ,  8.4486771 , 12.12448912,  5.60064319,  5.29009836,
       10.37698435, 11.36089355, 13.34587027, 12.17451685,  5.70591986,
       11.63383221,  8.5724042 , 11.018084  , 10.0674887 , 10.90638066,
        5.79543038,  9.6282763 , 10.6910768 ,  5.07306274,  5.38342077,
        8.30729352,  5.36061974,  5.80860165,  8.16714836, 10.84493732,
        4.75303127,  4.91038884,  7.88677538, 12.59969936,  5.53255882,
        8.7907923 , 12.30858677, 13.09426142, 12.4217523 ,  9.14044984,
       10.4606265 ,  9.95552626, 13.66991027,  9.7712321 ,  9.48969978,
       11.86239961, 13.6105595 ,  6.10360073, 14.08549802,  6.59176018,
       13.2778378 ,  9.38232121,  6.53250937,  6.54615887,  9.36093562,
        6.46560181, 12.18998195, 12.50996828,  7.32233733,  8.34949383,
       11.83570132, 11.23201619, 10.87409005, 10.31613269, 11.72749923,
        8.30125201, 12.5085995 ,  8.91551231, 13.61600536,  6.42439344,
        9.85384623, 10.0674887 ,  9.77777346, 11.95743558,  5.13182506,
       11.73567759,  4.78002665, 11.05672136, 12.78327037,  7.4544866 ,
        5.91222346])}
{'P': tensor([[3.9542e-03, 1.5770e-01, 1.5428e-01,  ..., 8.0938e-05, 3.9534e-05,
         2.0945e-05],
        [2.2423e-05, 5.9352e-03, 7.6647e-01,  ..., 5.0197e-05, 1.6601e-04,
         8.4831e-05],
        [6.4458e-05, 1.1839e-04, 3.2217e-02,  ..., 2.1747e-04, 1.3314e-03,
         6.8086e-04],
        ...,
        [4.9868e-04, 5.6252e-04, 6.8410e-04,  ..., 3.9667e-02, 7.3542e-05,
         6.3148e-05],
        [3.6997e-08, 5.0336e-08, 5.7663e-08,  ..., 2.6010e-07, 5.8224e-03,
         7.4853e-01],
        [2.9971e-06, 3.8152e-06, 4.2567e-06,  ..., 2.3814e-05, 1.0089e-03,
         5.9287e-01]]), 'X': tensor([[ 12.0582,  -2.6640,   0.7132],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.5789,   0.1722,   1.6470],
        [ 12.2311,  -2.0445,   1.3545],
        [ 12.5770,   1.8512,  -0.6578],
        [ 12.5996,   1.3971,  -1.0694],
        [  4.2096,   5.8815,   1.9230],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.1203,  -2.7151,   6.2110],
        [ 12.4901,  -0.7052,   1.7315],
        [ -9.3616,  -2.4750,   3.7765],
        [  3.5965,  -4.0754,  -3.5035],
        [ 12.4901,  -0.7052,   1.7315],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -3.5903,   7.4847,  -8.9112],
        [  9.4554,   9.4245,   9.8800],
        [ 12.4597,  -0.9136,  -4.8763],
        [  2.3373,  -4.6232,  -5.2246],
        [  1.7815,  -4.7481,   4.4289],
        [ -2.6485, -10.4151,   0.8766],
        [  3.7992,  -3.9474,  -8.9019],
        [ 11.1420,   6.9331,  -2.4779],
        [ 10.0567,   8.6697,   4.6329],
        [  4.5800,   5.5034,  -3.9432],
        [  5.8889,  -1.3916,   8.8993],
        [  6.2630,   2.1057,  -5.2136],
        [  6.3551,   0.7149,   1.3344],
        [  6.3535,   0.6792,  -7.5101],
        [  0.1460,   7.8340,  -9.4614],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.0890,  -0.8382,   2.9166],
        [ -6.4844,  -8.0230,   8.5209],
        [ -9.4283,  -1.9585,  -0.9129],
        [  3.0876,   6.7527,  -5.3540],
        [  6.3550,   0.7129,   0.4917],
        [  1.3103,  -4.8032,  -0.9300],
        [  1.7000,  -4.7609,   2.7935],
        [  5.6523,   3.9443,  -2.9449],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.5424,  -2.0885,   7.2487],
        [  3.3633,  13.4794,   0.5193],
        [ -8.9033,   2.2329,  -8.8912],
        [  7.7794,  11.0629,   8.9069],
        [  0.5850,  -4.7982,   9.6639],
        [  0.2732,  -4.7619,  -1.1577],
        [ -9.1231,  -3.5739,   8.4708],
        [ 11.2767,  -4.6006,   0.7868],
        [ -6.4822,  -8.0252,  -3.8983],
        [  2.0449,  13.8402,  -7.1642],
        [  2.0729,   7.2878,   7.0880],
        [  3.9515,  -3.8425,   6.1729],
        [ 11.9808,   4.9340,  -3.3715],
        [  7.5784,  -8.9141,   5.8410],
        [ -2.6069, -10.4298,  -6.1995],
        [  7.3724,  -9.0577,  -8.9843],
        [ -3.0008, -10.2827,  -8.9480],
        [  4.5340,  -3.3629,   8.9652],
        [ -4.5778,  -9.5013,  -5.0667],
        [  5.1886,  -2.6250,   6.4677],
        [  2.1771, -10.9776,  -3.2007],
        [ -7.9667,   4.0814,  -0.6397],
        [  6.3505,   0.6235,  -8.5849],
        [  7.3592,  11.3937,   6.2407],
        [  0.1532,  -4.7422,  -7.0145],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.3554,   4.8985,  -0.6665],
        [  2.1504,  -4.6726,   8.7908],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -9.3464,   0.5708,  -1.6397],
        [  5.6928,  12.4710,   3.6001],
        [  9.9750,  -6.6338,  -1.2440],
        [  8.3718,  -8.2953,   8.4666],
        [ 12.5267,  -0.4125,  -8.7331],
        [  5.8979,   3.3958,  -3.8713],
        [  5.4867,  12.5816,   2.9549],
        [ 11.9122,   5.1397,  -1.5498],
        [ -6.4844,  -8.0230,   8.5209],
        [  3.6817,   6.3354,   6.1233],
        [  5.8323,  -1.5228,   5.1132],
        [ -9.3107,  -2.7697,  -3.2919],
        [  6.3604,   0.9042,  -6.5901],
        [ -2.0229, -10.6163,   8.6930],
        [  0.9445,   7.6736,   0.5095],
        [  6.1741,  -0.5329,   3.1913],
        [  4.0591,   6.0202,  -1.6339],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.7349,  -3.5838,   2.1644],
        [  1.6232,  -4.7717,  -2.5655],
        [  6.3211,   1.7017,  -0.2164],
        [  6.9284,  11.7061,  -0.3732],
        [  1.5662,  13.9375,   4.5142],
        [  6.1579,  -0.5961,  -6.5595],
        [ -8.6381,  -4.9238,  -4.4077],
        [  8.6420,  -8.0580,   6.5781],
        [  7.5064,  -8.9651,  -0.0994],
        [ -6.2668,   5.9808,   2.5614],
        [  4.1297,  -3.7098,  -3.9838],
        [  2.3373,  13.7721,  -0.2442],
        [ -5.0943,  -9.1682,   0.5073],
        [  6.8093,  -9.4178,   8.4244],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.2182,  13.8007,   5.1889],
        [  0.5175,  14.0910,   4.6540]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1025, 0.1217,  ..., 0.4104, 0.6200, 0.5498],
        [0.1025, 0.0000, 0.0349,  ..., 0.4980, 0.5179, 0.4501],
        [0.1217, 0.0349, 0.0000,  ..., 0.4977, 0.5069, 0.4451],
        ...,
        [0.4104, 0.4980, 0.4977,  ..., 0.0000, 0.9962, 0.9421],
        [0.6200, 0.5179, 0.5069,  ..., 0.9962, 0.0000, 0.1055],
        [0.5498, 0.4501, 0.4451,  ..., 0.9421, 0.1055, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.3489324 , 12.50996828, 12.58006038, 12.40074901, 12.7125079 ,
       12.67680439,  7.23276582, 12.67680439,  5.79562691, 12.50996828,
        9.68324553,  5.43544626, 12.50996828,  9.48969978,  8.30125201,
       13.35013374, 12.49317563,  5.18047638,  5.0713364 , 10.74655346,
        5.47866221, 13.12299145, 13.2778378 ,  7.15990676,  6.05113217,
        6.60752373,  6.39520397,  6.38968357,  7.83534987,  9.86089579,
       10.35485331,  5.18047638,  6.1463896 , 10.31584102,  9.62958973,
        7.42512621,  6.3949043 ,  4.978704  ,  5.05534609,  6.89245352,
       10.48840476,  5.92281142, 13.89264822,  9.17905491, 13.52430176,
        4.83371341,  4.76969364,  9.79814997, 12.17901033, 10.31613269,
       13.99048094,  7.57685908,  5.51177501, 12.95702647, 11.70016217,
       10.7506423 , 11.67875855, 10.71162681,  5.64500512, 10.54658597,
        5.81481044, 11.1913598 ,  8.95133961,  6.38105015, 13.56367857,
        4.74468406, 11.9182956 ,  8.83725081,  5.14371005,  9.61942445,
        9.36378438, 13.70893492, 11.97949273, 11.785566  , 12.53345445,
        6.80559669, 13.72594022, 12.97370371, 10.31584102,  7.32754882,
        6.02779201,  9.71392179,  6.42439344, 10.80728788,  7.73151309,
        6.19707944,  7.26074925, 12.11735334, 12.19805394, 12.26997165,
        5.04028538,  6.54615887, 13.60275211, 14.02526505,  6.18668463,
        9.94285511, 11.8159254 , 11.69264243,  8.66273171,  5.5513072 ,
       13.96905756, 10.48840476, 11.62158554,  6.14086582, 13.97779839,
       14.10045797])}
{'P': tensor([[3.2405e-03, 2.4168e-01, 2.3816e-01,  ..., 5.6143e-08, 1.5307e-04,
         9.9164e-08],
        [5.3622e-05, 1.3617e-02, 6.1135e-02,  ..., 1.7064e-07, 3.9224e-04,
         4.2520e-07],
        [1.8301e-05, 5.9893e-04, 6.0095e-03,  ..., 8.0832e-08, 2.1377e-04,
         1.9610e-07],
        ...,
        [3.2662e-04, 1.8131e-04, 2.1004e-04,  ..., 5.1095e-02, 1.0675e-03,
         1.3107e-05],
        [7.5865e-04, 1.0144e-03, 1.4337e-03,  ..., 3.2513e-05, 8.8979e-02,
         4.3624e-05],
        [4.2343e-05, 2.6730e-05, 2.4991e-05,  ..., 5.4800e-03, 1.2180e-04,
         6.6456e-02]]), 'X': tensor([[ 12.0772,  -2.6014,  -7.1832],
        [ 12.4013,  -1.2597,  -7.6474],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.7903,  -4.7467,   0.1168],
        [ -9.3739,   0.3982,  -4.2564],
        [  0.1382, -11.0073,   1.0026],
        [  0.1206,  -4.7363,   8.7103],
        [  5.7845,   3.6652,   0.9175],
        [  6.8407,  11.7666,   3.0045],
        [ -2.7572,   7.7194,   5.9580],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.6970,   6.9826,   9.7350],
        [  4.6231,   5.4556,   8.4973],
        [  7.0515,  11.6194,   8.4289],
        [ -3.3812, -10.1235,   7.4384],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -8.3535,  -5.5118,   9.8399],
        [ -7.7239,   4.4298,   3.1216],
        [  3.4401,  13.4540,   4.6576],
        [  4.0078,  -3.8018,  -6.2168],
        [  3.2647, -10.8066,  -0.3768],
        [ 12.3357,   3.6001,   0.7903],
        [ -8.7735,   2.5640,  -6.3313],
        [ -1.8367,   7.8718,  -8.6523],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.1667,  -2.6543,   1.2879],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -8.0608,   3.9362,  -5.9567],
        [  1.5662,  13.9375,   4.5142],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.0876,   6.7527,  -5.3540],
        [ -6.6998,   5.5954,   0.0823],
        [ -3.5173,   7.5091,  -8.9585],
        [ -2.0721,   7.8432,   6.6380],
        [  4.7482, -10.3907,   5.9149],
        [ -9.1907,  -3.3180,  -5.5943],
        [  8.3791,  10.5409,   5.5590],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.9284,  11.7061,  -0.3732],
        [ 12.5650,   2.0132,   4.6272],
        [  5.4167, -10.1293,   6.2156],
        [  0.9809,   7.6642,  -1.4217],
        [  4.1537,   5.9340,  -6.0423],
        [  5.3924, -10.1396,  -4.0863],
        [ 10.5678,   7.9261,  -0.6100],
        [  3.2701,   6.6338,   9.9209],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 12.5919,   0.4035,  -5.8460],
        [  2.9461,  -4.4077,  -0.4831],
        [ 12.5770,   1.8512,  -0.6578],
        [ 11.5198,  -4.0923,   5.5088],
        [ 11.4703,  -4.2006,   7.4086],
        [  4.3554,   5.7393,  -9.0766],
        [  3.1064,  13.5607,   6.7605],
        [  6.1984,  -0.4327,   1.5438],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -8.8934,   2.2597,   3.5831],
        [  3.7121,   6.3117,   0.6738],
        [ -7.0253,  -7.4493,  -1.5978],
        [  5.4426, -10.1181,  -5.1268],
        [ -5.0132,  -9.2234,  -8.3531],
        [ 12.6000,   1.3839,  -3.2992],
        [ -4.4503,   7.1378,   3.3751],
        [  4.6063,  13.0051,  -3.1456],
        [ -9.4203,  -2.0340,   2.5856],
        [  7.7794,  11.0629,   8.9069],
        [  5.7963,  -1.6018,   1.8856],
        [  4.9337,  12.8566,  -3.6582],
        [ -8.6675,  -4.8569,  -9.1769],
        [  1.1006,   7.6318,  -2.4877],
        [  6.2576,   2.1359,  -3.3398],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -0.6323,   7.9085,  -5.9544],
        [  3.2159, -10.8166,  -1.0406],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.7845,   3.6652,   0.9175],
        [ -2.6069, -10.4298,  -6.1995],
        [  0.3715,  -4.7756,   0.2312],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -3.0644,   7.6439,  -1.4322],
        [  6.0073,   3.0986,   6.7937],
        [  3.2692,  13.5098,  -4.9032],
        [  0.1532,  -4.7422,  -7.0145],
        [ 11.2767,  -4.6006,   0.7868],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.7078,   2.7182,   1.8146],
        [  5.4732,  -2.2041,  -4.3830],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.2022,   2.4092,   3.0349],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.2373,  -3.1225,  -8.9763],
        [  5.1787,  -2.6383,  -4.2533],
        [ 11.7162,   5.6728,  -8.9372],
        [ -0.7599, -10.8995,  -6.0366],
        [  5.7485,  -9.9811,  -3.2355],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.6275, -10.7221,   3.2376],
        [  6.3124,   0.2057,  -0.4593],
        [ -8.8540,  -4.3971,  -5.2144]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0629, 0.0575,  ..., 1.1706, 0.5034, 1.0765],
        [0.0629, 0.0000, 0.0263,  ..., 1.2187, 0.5179, 1.1393],
        [0.0575, 0.0263, 0.0000,  ..., 1.1953, 0.4918, 1.1276],
        ...,
        [1.1706, 1.2187, 1.1953,  ..., 0.0000, 0.8419, 0.7363],
        [0.5034, 0.5179, 0.4918,  ..., 0.8419, 0.0000, 1.0814],
        [1.0765, 1.1393, 1.1276,  ..., 0.7363, 1.0814, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.35421461, 12.46513646, 12.49291768, 12.42554808, 12.64235359,
       12.59840193,  5.07306274,  9.38232121, 11.00812996,  4.73784222,
        6.84796202, 13.6105595 ,  8.19703512, 12.25806559,  7.48538905,
        7.15100303, 13.59170809, 10.67322527, 10.27716434, 10.00802627,
        8.90405823, 13.88683436,  5.52416368, 11.28895489, 12.85032836,
        9.14044984,  8.08320235,  5.84533345,  5.80860165, 10.53368644,
        8.97052115, 14.02526505, 10.8207297 ,  7.42512621,  8.72896381,
        8.29203356,  8.11227668, 11.42420508,  9.7712321 , 13.46553623,
       11.99633819, 13.60275211, 12.72524615, 11.48663767,  7.72668467,
        7.24326992, 11.48434767, 13.20988996,  7.39599986,  9.51074763,
       12.59840193,  5.30158497, 12.7125079 , 12.22503754, 12.21532282,
        7.20482224, 13.91198191,  6.2134961 , 10.53368644,  9.17595899,
        7.32233733, 10.23946247, 11.4890861 , 10.4977132 , 12.67576792,
        8.41148722, 13.79676654,  9.63742669, 13.52430176,  6.01356004,
       13.77075174,  9.93553424,  7.71075226,  6.61211224,  6.21275016,
        9.5642993 ,  7.9337647 , 11.2845608 , 12.38881635,  6.84796202,
       10.7506423 ,  4.79003361, 12.04924159,  9.55720651, 10.23946247,
        8.23525433,  6.7593917 , 13.89974693,  4.74468406, 12.17901033,
       13.06957425,  9.12220271,  5.90035324, 12.3828957 , 12.27993098,
        6.65368551,  9.15976746,  9.75075273,  5.81199422, 13.01728313,
       10.92597196, 11.51811603,  9.53015376, 10.84493732,  6.31576264,
        9.88571713])}
{'P': tensor([[4.3792e-03, 3.5954e-01, 2.3163e-01,  ..., 5.0811e-07, 1.8380e-07,
         4.7547e-05],
        [3.1032e-05, 1.0889e-02, 1.3214e-04,  ..., 1.6441e-06, 5.1828e-07,
         1.3988e-04],
        [5.5872e-05, 2.0146e-01, 6.6615e-03,  ..., 8.8482e-07, 3.0802e-07,
         1.0673e-04],
        ...,
        [7.6103e-06, 5.3898e-06, 6.5113e-06,  ..., 6.8947e-02, 2.4256e-01,
         5.7683e-07],
        [8.2453e-06, 5.1282e-06, 6.8735e-06,  ..., 2.1373e-04, 3.4880e-02,
         4.0497e-07],
        [2.7661e-06, 4.4662e-06, 3.8874e-06,  ..., 5.0098e-05, 2.1288e-05,
         1.4456e-01]]), 'X': tensor([[ 12.0845,  -2.5772,  -4.6073],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.0890,  -0.8382,   2.9166],
        [  3.1480,  -4.3165,  -7.8002],
        [  5.8318,   3.5567,  -4.7692],
        [  9.5018,  -7.1945,  -1.1892],
        [ 12.0582,  -2.6640,   0.7132],
        [  1.7659,   7.4127,   4.7871],
        [ -8.8540,  -4.3971,  -5.2144],
        [  2.2644,  -4.6434,   9.7508],
        [  6.9433,  -9.3362,  -0.3421],
        [ 11.5456,   6.0861,   8.1101],
        [ -0.0410,   7.8591,   9.3457],
        [ -9.1706,   1.3863,   8.4032],
        [  8.2422,  -8.4041,  -1.4962],
        [  6.0788,  -0.8710,  -6.8719],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.7794,   5.5086,  -9.6929],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.5636, -10.9293,   9.6519],
        [  1.1994,   7.6035,   2.4295],
        [  4.2612, -10.5514,  -6.6268],
        [ -0.5385,   7.9037,   6.3830],
        [ -8.4824,  -5.2575,  -0.0795],
        [  3.4604,  -4.1545,   1.8091],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.8275,  -1.5334,   2.0124],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.4501,  -1.7133,   2.2980],
        [  9.2811,  -7.4336,   5.1771],
        [  3.6469,   6.3624,  -1.8414],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.9003,  12.3550,   2.1917],
        [ 11.6634,   5.8050,   9.1590],
        [  1.6954,  -4.7616,   9.8198],
        [  6.9314,  -9.3435,  -1.2729],
        [ -3.3812, -10.1235,   7.4384],
        [  6.3119,   1.7797,   9.9186],
        [ -5.5525,   6.5146,  -4.4984],
        [  6.1730,  -0.5375,  -8.8861],
        [  5.8275,  -1.5334,   2.0124],
        [  6.8093,  -9.4178,   8.4244],
        [ -1.7290,   7.8825,  -3.3831],
        [ -6.3489,   5.9117,   6.5161],
        [  6.2895,   0.0405,   5.7652],
        [  4.5503,   5.5359,   7.4735],
        [  0.5960, -11.0335,   3.1144],
        [  2.5469,  13.7192,  -9.6935],
        [  5.6314,  -1.9302,  -9.6543],
        [  0.1206,  -4.7363,   8.7103],
        [ -1.5451,   7.8975,   7.6104],
        [  4.5491,  -3.3485,   6.8412],
        [  1.6954,  -4.7616,   9.8198],
        [ 11.7796,  -3.4696,   9.0088],
        [  6.2247,  -0.3150,  -9.3657],
        [  9.6587,   9.1817,  -0.8212],
        [  2.3373,  -4.6232,  -5.2246],
        [  8.4691,  10.4571,  -5.0650],
        [ 11.1469,   6.9237,   1.6160],
        [  2.5620,   7.0549,   8.4825],
        [ -6.0844,  -8.3955,   0.7705],
        [  6.6319,  11.9066,   3.5165],
        [  4.5814, -10.4485,   2.5496],
        [  1.7232,  -4.7574,   3.1270],
        [  6.1929,   2.4501,  -6.9352],
        [  7.1441,  11.5529,   2.3232],
        [  6.9956,  11.6590,  -7.6876],
        [ -0.0254, -10.9932,   3.2200],
        [  0.6396,  14.0772,  -6.3786],
        [  1.9241,   7.3503,   7.9004],
        [  6.3515,   0.6418,  -6.5823],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.0233,   4.9647,  -3.1788],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -5.3382,   6.6533,   7.8054],
        [  6.2555,   2.1474,   4.1685],
        [  6.3266,   0.3292,   9.3502],
        [  1.8320,   7.3871,   5.1452],
        [  4.9452, -10.3187,  -7.8621],
        [  5.8902,  -1.3887,   4.0388],
        [  1.7815,  -4.7481,   4.4289],
        [  9.6587,   9.1817,  -0.8212],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.1755,   2.5233,  -2.0042],
        [ 12.5789,   0.1722,   1.6470],
        [ -5.6792,  -8.7352,   4.0318],
        [ 10.9931,   7.2107,   7.7933],
        [ 11.7794,   5.5086,  -9.6929],
        [  3.2809,  13.5061,  -4.0199],
        [  6.2790,  -0.0259,  -5.4116],
        [  4.0642,  -3.7599,  -3.4565],
        [  5.5240,  -2.1198,   6.2055],
        [ -6.7992,   5.4991,   9.9186],
        [  4.0078,  -3.8018,  -6.2168],
        [ -2.5683, -10.4433,  -2.9182],
        [ -7.3554,   4.8985,  -0.6665],
        [  1.4150,  13.9647,  -4.0915],
        [  1.8447,  -4.7372,   8.1235],
        [ -3.0004,   7.6607,  -7.2640],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.4505,  -1.7081,  -0.7039],
        [  9.4925,   9.3811,   8.9842]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0621, 0.0454,  ..., 0.8657, 1.0106, 1.2263],
        [0.0621, 0.0000, 0.0631,  ..., 0.9276, 1.0725, 1.2258],
        [0.0454, 0.0631, 0.0000,  ..., 0.8863, 1.0229, 1.1833],
        ...,
        [0.8657, 0.9276, 0.8863,  ..., 0.0000, 0.2322, 1.5907],
        [1.0106, 1.0725, 1.0229,  ..., 0.2322, 0.0000, 1.5389],
        [1.2263, 1.2258, 1.1833,  ..., 1.5907, 1.5389, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.35625694, 12.49317563, 12.4217523 , 12.59840193, 12.62711252,
       12.67576792,  6.1463896 ,  5.34250225,  6.83083863, 11.9182956 ,
       12.3489324 ,  7.62011734,  9.88571713,  5.16611782, 11.63502679,
       13.05148826,  7.85919687,  9.27474386, 11.77124373,  6.14086582,
       12.86648771, 13.00380624, 12.10222204, 11.22596994,  7.69752262,
       11.37939695,  7.92200342,  9.97964013,  5.40689332,  5.16077764,
        6.02588517,  6.83515201,  9.60412735, 11.89105259,  7.33350642,
       10.00794468, 13.69162795, 13.02818702,  5.05444288, 11.63383221,
       10.67322527,  6.55800267,  8.55982105,  6.19633139,  6.02588517,
       11.62158554,  8.06990483,  8.67502348,  6.28961988,  7.16598273,
       11.04953767, 13.95361098,  5.95297459,  4.73784222,  8.04718433,
        5.64865173,  5.05444288, 12.27993098,  6.23262289, 13.32645552,
        5.18047638, 13.45642379, 13.12218013,  7.50564223, 10.36844813,
       13.62896593, 11.4088043 ,  5.05989547,  6.65991877, 13.58333857,
       13.59673406, 10.99326577, 14.09176259,  7.59795445,  6.383897  ,
        5.24137909,  7.06272256,  9.75401449,  8.53014915,  6.61386436,
        6.33517349,  7.61088125, 11.4424816 ,  6.05165248,  5.0713364 ,
       13.32645552, 12.13568598,  6.67108662, 12.58006038, 10.41909211,
       13.14690137, 13.00380624, 13.89886469,  6.27905292,  5.5366698 ,
        5.91677188,  8.74470218,  5.52416368, 10.75443106,  8.83725081,
       14.03618649,  5.08372149,  8.22728399,  9.20340031,  9.60359158,
       13.34587027])}
{'P': tensor([[3.6541e-03, 2.1423e-01, 1.5437e-01,  ..., 2.2694e-06, 1.3792e-04,
         5.5896e-06],
        [7.1932e-06, 4.0754e-03, 1.6414e-01,  ..., 4.0572e-06, 3.3758e-04,
         1.8184e-06],
        [7.4921e-06, 6.7671e-05, 5.0588e-03,  ..., 4.6282e-06, 4.9904e-04,
         1.7773e-06],
        ...,
        [1.5225e-06, 1.8568e-06, 1.7517e-06,  ..., 9.6231e-02, 7.3617e-05,
         8.7528e-07],
        [7.5086e-06, 1.0274e-05, 1.1169e-05,  ..., 7.6445e-05, 5.0406e-02,
         5.8461e-06],
        [2.2073e-01, 9.3873e-02, 8.3318e-02,  ..., 2.1500e-06, 1.5785e-04,
         1.3519e-02]]), 'X': tensor([[ 12.1751,  -2.2592,  -5.7751],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.1793,   4.7461,   0.3899],
        [  6.5428,  -9.5728,  -2.9897],
        [ 10.9875,   7.2207,  -8.8549],
        [ 11.3367,  -4.4806,   0.4643],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -9.2508,   1.0572,   2.9978],
        [  5.1793,   4.7461,   0.3899],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.6801,  -3.7196,  -0.9181],
        [  8.5716,  10.3596,   6.8818],
        [  0.1041, -11.0045,   2.4311],
        [ -8.0750,  -6.0072,  -5.8267],
        [  1.9312,  13.8649,   8.3563],
        [  1.8719,   7.3713,  -3.7465],
        [ -4.0868,   7.2981,   8.1874],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.5663,   7.0526,  -6.8769],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.6734,   4.4980,  -5.4248],
        [ 12.2592,   3.9425,   2.0383],
        [ -7.7842,   4.3466,  -1.7548],
        [ 10.1503,   8.5414,  -7.2681],
        [  3.7121,   6.3117,   0.6738],
        [ 12.6059,   0.9597,  -6.3264],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.3413,   0.4883,  -6.8416],
        [ -1.5408,   7.8978,   5.3470],
        [  5.9003,  12.3550,   2.1917],
        [ 11.0864,   7.0387,  -0.1238],
        [ -5.9872,   6.2037,  -9.4477],
        [ -5.9938,  -8.4745,   3.9735],
        [ -3.0004,   7.6607,  -7.2640],
        [  2.6321,  -4.5296,  -2.5159],
        [  2.5813,  -4.5471,   5.3597],
        [  0.9945,  -4.8144,  -0.1897],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -3.5173, -10.0623,   4.8280],
        [ -1.9487,   7.8590,  -3.3330],
        [  6.3122,  -9.6994,  -6.3282],
        [  6.3569,   0.7591,   6.5011],
        [  3.1892, -10.8220,  -6.8409],
        [ -8.6112,   2.9319,  -7.9943],
        [ -1.5408,   7.8978,   5.3470],
        [ -5.6792,  -8.7352,   4.0318],
        [  1.7228,  -4.7575,   8.3396],
        [  2.9461,  -4.4077,  -0.4831],
        [  6.3534,   1.2990,  -7.2942],
        [ 12.1588,   4.3364,  -8.0867],
        [ -1.6275, -10.7221,   3.2376],
        [  0.4996,   7.7738,  -4.4711],
        [ -1.6275, -10.7221,   3.2376],
        [  5.8836,   3.4315,  -4.0327],
        [ 11.6165,  -3.8715,   6.4594],
        [ -5.9938,  -8.4745,   3.9735],
        [  2.1477,   7.2550,   8.9688],
        [  7.1441,  11.5529,   2.3232],
        [  1.8766,   7.3694,   1.1887],
        [  0.6396,  14.0772,  -6.3786],
        [  5.1292,   4.8183,  -5.7997],
        [ 12.5224,  -0.4497,   4.8448],
        [ 12.2692,  -1.8882,   3.9253],
        [  3.2159, -10.8166,  -1.0406],
        [  6.1973,  -0.4372,   6.5720],
        [  2.2373,  -4.6506,  -1.7334],
        [ -1.2454,   7.9127,  -0.5715],
        [  0.6952,  -4.8060,   8.8976],
        [  0.4119,  14.1019,  -4.1870],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.6414, -10.0303,   7.8525],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.9883,   3.1536,   6.2320],
        [  4.2096,   5.8815,   1.9230],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.2508,   1.0572,   2.9978],
        [  3.6703,   6.3443,  -5.4401],
        [  2.8994,   6.8675,  -9.5987],
        [ -4.0868,   7.2981,   8.1874],
        [ -9.3464,   0.5708,  -1.6397],
        [  3.3943,   6.5483,   1.0790],
        [ -9.4309,  -0.0599,  -4.0830],
        [  2.9260,  -4.4162,   8.3808],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.4039,   5.6903,   8.7366],
        [  3.3069,   6.6089,   8.7377],
        [  6.2675,  12.1379,  -8.9720],
        [ -0.8569,   7.9156,  -6.5557],
        [ -1.0482, -10.8487,  -5.9066],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -7.0825,   5.2073,  -5.0497],
        [ -0.0410,   7.8591,   9.3457],
        [  1.6422,   7.4586,  -6.8056],
        [ 10.4994,  -5.9193,  -7.5153]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0736, 0.0964,  ..., 1.0821, 0.2510, 0.1568],
        [0.0736, 0.0000, 0.0309,  ..., 1.1082, 0.3228, 0.2220],
        [0.0964, 0.0309, 0.0000,  ..., 1.1378, 0.3474, 0.2330],
        ...,
        [1.0821, 1.1082, 1.1378,  ..., 0.0000, 0.9669, 1.1310],
        [0.2510, 0.3228, 0.3474,  ..., 0.9669, 0.0000, 0.1860],
        [0.1568, 0.2220, 0.2330,  ..., 1.1310, 0.1860, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.3828957 , 12.59840193, 12.64235359, 12.49317563, 12.49291768,
       12.77402351,  7.02496223, 11.59512194, 13.14777433, 12.18998195,
        5.01114719,  9.43112957,  9.79039371,  9.3109884 ,  7.02496223,
       10.61472636, 12.25806559, 13.44593519, 11.0050379 , 10.06437598,
       13.99877413,  7.60529892,  8.36445758, 11.86239961,  7.50500064,
        8.16714836,  8.89458273, 12.87752052,  8.91551231, 13.26590071,
        7.32233733, 12.64235359, 12.12448912,  6.36004393,  8.04664886,
       13.69162795, 13.13205357,  8.62162372, 10.37998078,  8.22728399,
        5.23878371,  5.2287058 ,  4.9160962 , 10.33712258, 10.65929968,
        8.09704052, 11.57249984,  6.4020404 , 11.28215429,  9.09661212,
        8.04664886, 10.41909211,  5.05981914,  5.30158497,  6.48486171,
       12.9089559 , 10.84493732,  7.78980299, 10.84493732,  6.81117802,
       12.24467041, 10.37998078,  7.56617037, 13.58333857,  7.60463349,
       14.09176259,  7.03733775, 12.53047489, 12.41366754, 11.2845608 ,
        6.21275016,  5.16077764,  8.01009083,  4.85603886, 14.10796466,
       12.79126785, 11.50791859,  9.14973319,  6.76791483,  7.23276582,
        5.19065439,  5.40689332,  9.71392179,  9.3109884 ,  7.32950385,
        7.4544866 ,  8.36445758,  9.36378438,  7.37576154,  9.43112957,
        5.29755487, 12.17901033,  7.19534146,  7.39004226, 13.66051884,
        7.96181288, 10.8992524 ,  8.7273569 ,  9.37989631, 12.4217523 ,
       10.00410625, 12.1460931 ,  8.7907923 ,  7.85919687,  7.63726134,
       12.05300943])}
{'P': tensor([[3.1790e-03, 3.9614e-01, 1.6677e-01,  ..., 5.7907e-07, 1.3116e-04,
         1.8761e-04],
        [2.3121e-05, 5.0712e-03, 3.5611e-01,  ..., 8.6579e-07, 1.5916e-04,
         3.6463e-04],
        [1.2970e-05, 1.6772e-05, 8.2517e-03,  ..., 1.0104e-06, 9.5473e-05,
         1.0725e-03],
        ...,
        [2.3572e-04, 2.2663e-04, 1.5637e-04,  ..., 7.6795e-02, 1.3601e-03,
         1.8608e-04],
        [8.9390e-03, 1.1042e-02, 1.7464e-02,  ..., 2.6808e-04, 1.0464e-01,
         1.7998e-02],
        [2.4362e-05, 2.6678e-05, 3.7937e-05,  ..., 1.2928e-04, 3.3243e-04,
         1.2204e-01]]), 'X': tensor([[ 12.1940,  -2.1881,   7.1499],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.4877,  -0.7222,   8.8347],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -3.4488, -10.0934,   2.5195],
        [  2.3490,  -4.6199,  -4.2489],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.8363,   2.4086,  -5.6296],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.9433,  -9.3362,  -0.3421],
        [  6.3205,   1.7068,  -3.8619],
        [  5.5294,  -2.1106,   9.0626],
        [ -6.2668,   5.9808,   2.5614],
        [  3.3069,   6.6089,   8.7377],
        [  2.6613,   7.0021,   4.5783],
        [ -9.3331,  -2.6465,  -8.2028],
        [  6.2576,   2.1359,  -3.3398],
        [  4.1140,  -3.7220,  -0.1273],
        [  6.1222,  -0.7261,   8.8204],
        [ 12.4691,   2.8525,  -5.5110],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.1142,  -2.7230,  -4.1966],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.3411,  -2.6004,  -2.7062],
        [  3.7916,  -3.9524,  -7.9547],
        [  3.5939,   6.4029,  -1.8357],
        [ -5.9938,  -8.4745,   3.9735],
        [  9.2811,  -7.4336,   5.1771],
        [  3.4401,  13.4540,   4.6576],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -9.2345,  -3.1345,   3.9715],
        [  2.7295,  13.6703,  -8.9328],
        [  5.3118,   4.5452,   4.2114],
        [  0.3354,  -4.7708,  -4.9125],
        [ 11.7779,   5.5127,  -2.0699],
        [ -1.8823, -10.6558,  -3.8952],
        [ 11.6165,  -3.8715,   6.4594],
        [ -6.7745,  -7.7264,   8.4684],
        [  6.1737,   2.5305,   3.9049],
        [  4.9152,  12.8653,  -1.4459],
        [  6.3534,   1.2990,  -7.2942],
        [ -0.9715, -10.8630,   0.3555],
        [ 10.6283,  -5.7253,   7.4878],
        [ -3.6381,   7.4683,   7.3544],
        [  6.3535,   0.6792,  -7.5101],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.6260,  -1.9401,   3.0259],
        [  5.6260,  -1.9401,   3.0259],
        [  9.8082,  -6.8396,   4.9615],
        [  9.1409,  -7.5789,   1.5814],
        [  4.2041,  -3.6509,  -4.9488],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -8.1531,  -5.8748,   2.6934],
        [ 11.0275,  -5.0666,   8.4154],
        [  4.7156,  -3.1831,  -8.0224],
        [  7.1441,  11.5529,   2.3232],
        [  7.0358,  -9.2784,  -9.8479],
        [  8.1043,  -8.5164,  -7.9290],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.0608,   3.9362,  -5.9567],
        [ -0.6323,   7.9085,  -5.9544],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -9.4756,  -0.8166,  -9.6021],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4541,  -0.3327,   8.8246],
        [ -3.5173,   7.5091,  -8.9585],
        [ 11.4495,   6.3026,   6.8257],
        [  6.1973,  -0.4372,   6.5720],
        [ 12.3025,  -1.7440,  -6.6632],
        [  7.0171,  11.6439,  -5.0178],
        [  4.1140,  -3.7220,  -0.1273],
        [  0.7145,  -4.8071,  -2.3766],
        [  1.1274,  14.0116,   8.1753],
        [ -8.6681,  -4.8556,   9.2518],
        [  4.6063,  13.0051,  -3.1456],
        [ -1.5451,   7.8975,   7.6104],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.1459,   1.4780,  -4.2403],
        [  2.5318,  13.7231,   6.3692],
        [  2.7590,  13.6621,   9.4496],
        [  5.7485,  -9.9811,  -3.2355],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -8.1666,   3.7653,   3.6075],
        [ -3.7964,  -9.9294,   7.7147],
        [  7.9816,  10.8938,  -0.5378],
        [  7.0171,  11.6439,  -5.0178],
        [  1.9204,   7.3518,   8.4781],
        [ 10.2470,   8.4054,  -0.1177],
        [ -7.2515,   5.0195,   4.6434],
        [  2.2659,  13.7894,   6.5436],
        [  7.4132,  11.3526,   0.3322],
        [  6.8704,  -9.3809,   7.5769],
        [  3.0876,   6.7527,  -5.3540],
        [ -1.9600,   7.8577,   6.9121],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -2.6735, -10.4062,  -1.5911],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.9026,   3.3837,  -0.6573],
        [  0.3354,  -4.7708,  -4.9125],
        [  5.8323,  -1.5228,   5.1132],
        [  2.6613,   7.0021,   4.5783]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 4.0196e-02, 1.1393e-01,  ..., 9.6328e-01, 3.0080e-01,
         3.3626e-01],
        [4.0196e-02, 0.0000e+00, 7.4275e-02,  ..., 9.6775e-01, 3.0102e-01,
         3.5387e-01],
        [1.1393e-01, 7.4275e-02, 0.0000e+00,  ..., 9.9469e-01, 3.2912e-01,
         4.0616e-01],
        ...,
        [9.6328e-01, 9.6775e-01, 9.9469e-01,  ..., 0.0000e+00, 6.6750e-01,
         6.5117e-01],
        [3.0080e-01, 3.0102e-01, 3.2912e-01,  ..., 6.6750e-01, 0.0000e+00,
         1.4985e-01],
        [3.3626e-01, 3.5387e-01, 4.0616e-01,  ..., 6.5117e-01, 1.4985e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.38881635, 12.44177641, 12.62576026, 12.45418236, 12.71843462,
       12.5085995 , 10.35485331, 10.66631784,  5.18277858, 12.62576026,
        9.15866607,  9.70108409, 11.63502679,  6.54693987,  5.91854397,
        8.66273171,  7.39004226,  7.49076899,  9.70108409,  6.61211224,
        5.54776934,  6.16512856, 12.79126785, 12.19805394,  5.27178363,
        5.79391825, 13.20597932,  9.69629   ,  5.47702766,  7.34250948,
       10.37998078, 11.89105259, 13.88683436, 12.32545637,  9.75200848,
       13.94008948,  6.99097013,  4.78258092, 13.00413829, 10.8207297 ,
       12.24467041, 10.27572673,  6.67218687, 13.77222752,  6.48486171,
       10.90638066, 12.07225469,  8.30729352,  6.38968357,  7.88677538,
        5.95111804,  5.95111804, 11.95743558, 11.87412439,  5.56809335,
       10.4977132 , 10.04916678, 12.13568598,  5.68942572, 13.58333857,
       11.64436684, 11.75618513, 13.89974693,  8.97052115,  7.9337647 ,
       12.11735334,  9.51074763,  5.57341623,  9.45995955,  8.29203356,
       13.06957425,  6.21275016, 12.42554808, 13.59480906,  5.54776934,
        4.85995002, 14.05687815,  9.9353921 , 13.79676654,  8.04718433,
        8.27853076,  9.26455795, 13.95472331, 13.93790235, 11.51811603,
       12.30858677,  8.99276564, 10.63038565, 13.50484539, 13.59480906,
        7.5984809 , 13.25336001,  8.81930525, 13.97430191, 13.55868684,
       11.62770332,  7.42512621,  8.09842704,  9.88571713, 10.74409834,
        9.3109884 ,  9.15866607,  6.80370254,  4.78258092,  6.02779201,
        7.49076899])}
{'P': tensor([[4.4563e-03, 3.7450e-01, 3.0343e-01,  ..., 8.8532e-05, 1.3317e-03,
         1.3808e-07],
        [1.7224e-05, 4.7614e-03, 6.1488e-01,  ..., 1.4566e-04, 2.5890e-03,
         1.5432e-07],
        [2.3175e-05, 3.8918e-05, 1.0591e-02,  ..., 3.9127e-04, 8.8071e-03,
         3.6827e-07],
        ...,
        [8.5911e-06, 1.0550e-05, 1.1832e-05,  ..., 7.2698e-02, 1.6931e-04,
         9.5531e-05],
        [1.3610e-06, 1.8864e-06, 2.3800e-06,  ..., 1.8021e-03, 4.2976e-02,
         1.2668e-05],
        [2.7159e-05, 1.7650e-05, 1.3966e-05,  ..., 7.3824e-06, 1.6633e-06,
         8.2216e-02]]), 'X': tensor([[ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.0727e-04, 4.7513e-02, 7.6091e-02,  ..., 4.8071e-01, 3.0961e-01,
         1.0640e+00],
        [4.7513e-02, 0.0000e+00, 3.4894e-02,  ..., 4.7176e-01, 2.9773e-01,
         1.1090e+00],
        [7.6091e-02, 3.4894e-02, 0.0000e+00,  ..., 4.9386e-01, 2.6958e-01,
         1.1401e+00],
        ...,
        [4.8071e-01, 4.7176e-01, 4.9386e-01,  ..., 0.0000e+00, 7.6037e-01,
         1.1252e+00],
        [3.0961e-01, 2.9773e-01, 2.6958e-01,  ..., 7.6037e-01, 0.0000e+00,
         1.2777e+00],
        [1.0640e+00, 1.1090e+00, 1.1401e+00,  ..., 1.1252e+00, 1.2777e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.40074901, 12.50996828, 12.58006038, 12.78327037, 12.7125079 ,
       12.85032836, 11.28215429, 10.4606265 , 13.99877413, 11.22142195,
       12.19805394,  9.55720651,  5.46244988,  4.78544875,  5.48276038,
        7.57946459, 11.11038551, 11.97957425,  4.8763135 ,  8.11227668,
       13.97857591, 12.12448912,  8.55982105, 12.59969936,  5.1224079 ,
        6.61386436,  5.9510489 , 11.11038551, 10.60226614, 13.45642379,
        5.64865173,  9.25336371, 13.744387  , 11.81423161, 12.88270018,
        5.14371005, 13.5247215 , 11.0050379 ,  9.95552626,  5.18047638,
        5.05444288,  8.72896381, 10.5575601 ,  5.82345867, 10.06703358,
        6.81084724, 10.27572673, 12.4013153 ,  7.28793663,  7.57946459,
       11.63724792,  5.92281142,  6.54615887, 12.67680439,  8.09842704,
        7.14837514, 12.49317563, 11.67875855,  9.68324553, 13.61801526,
        7.83486943,  6.383897  ,  7.32950385,  9.34127013,  9.88310614,
        5.16205087,  6.7166297 ,  8.29203356,  6.60752373,  5.05774295,
       10.4977132 ,  7.19513806,  8.37954853,  8.99791316, 10.1371552 ,
        7.44805901,  7.38844703,  9.15866607,  7.32233733,  7.48311002,
        8.12043983, 13.04156075,  7.92644746,  6.49538841, 12.16901615,
        8.91551231, 12.5769659 , 10.54658597,  5.36061974, 13.5247215 ,
       11.2845608 ,  6.27905292, 13.41713661,  6.67121748, 10.07238463,
        9.93553424,  5.27178363,  8.56889738,  8.7907923 ,  9.55593374,
       10.51340829, 11.89105259,  5.18277858,  7.59795445, 13.50484539,
        9.79039371])}
{'P': tensor([[4.5132e-03, 4.3550e-01, 1.9269e-01,  ..., 8.2040e-08, 3.9062e-05,
         1.3987e-07],
        [1.3339e-05, 4.6645e-03, 4.0154e-01,  ..., 6.7901e-08, 7.3925e-05,
         1.5984e-07],
        [1.1460e-05, 2.0450e-05, 8.9519e-03,  ..., 1.0516e-07, 3.0465e-04,
         3.6517e-07],
        ...,
        [9.1739e-04, 5.2132e-04, 2.6089e-04,  ..., 1.4699e-01, 3.7281e-06,
         1.8362e-05],
        [1.7865e-06, 2.2577e-06, 3.2744e-06,  ..., 1.1905e-05, 3.0153e-01,
         8.2686e-05],
        [5.3497e-06, 3.5016e-06, 2.1242e-06,  ..., 2.1810e-04, 9.1718e-08,
         1.9570e-02]]), 'X': tensor([[ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.7172e-02, 1.0307e-01,  ..., 9.1923e-01, 5.0648e-01,
         1.0318e+00],
        [4.7172e-02, 0.0000e+00, 5.6512e-02,  ..., 9.6467e-01, 5.0530e-01,
         1.0698e+00],
        [1.0307e-01, 5.6512e-02, 1.5170e-04,  ..., 1.0150e+00, 4.9735e-01,
         1.1221e+00],
        ...,
        [9.1923e-01, 9.6467e-01, 1.0150e+00,  ..., 0.0000e+00, 9.5385e-01,
         8.7601e-01],
        [5.0648e-01, 5.0530e-01, 4.9735e-01,  ..., 9.5385e-01, 0.0000e+00,
         1.4171e+00],
        [1.0318e+00, 1.0698e+00, 1.1221e+00,  ..., 8.7601e-01, 1.4171e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.4013153 , 12.52372629, 12.67680439, 12.7125079 , 12.78327037,
       12.67576792, 13.06957425,  6.383897  , 11.04953767,  8.92549752,
       12.64235359, 10.412299  , 10.06437598,  9.09502385, 11.66453059,
        5.5748624 ,  8.81930525, 12.3489324 ,  5.19986797,  7.33350642,
       13.00413829,  8.859416  , 11.51557166, 10.04916678, 13.38091121,
        6.00663314, 10.57302546,  5.63824971, 10.7506423 ,  5.88782913,
        5.89074104,  8.06643487, 13.77604293, 13.744387  , 12.1494023 ,
       10.35485331,  8.16714836,  9.94285511, 11.83570132,  4.96231532,
       11.64746882,  9.17905491,  7.71075226,  8.04718433,  6.29821673,
        5.91222346, 13.04051744,  8.68979052, 11.63502679,  6.54615887,
        5.5513072 ,  9.44486609, 12.16901615, 12.25806559,  5.80860165,
        5.51177501,  9.94285511,  5.64865173,  9.99145096, 13.12299145,
        5.5671389 ,  9.11473391,  8.53501777, 12.84056604,  6.65368551,
       13.89264822,  6.4020404 , 13.89264822,  8.25828409,  4.83371341,
        6.42689705, 13.45124917, 10.14436912,  8.90405823,  8.29022197,
        5.0713364 ,  6.02588517,  7.62509582, 10.412299  ,  8.37954853,
        9.88571713, 12.22503754,  5.10444097,  6.81961996, 13.39372426,
       10.92597196,  8.8353674 ,  9.99211663, 10.06023665,  5.91677188,
        4.89121805, 10.24557047, 13.04156075, 10.19036081,  6.29034958,
       10.33958639,  6.1463896 , 14.03618649, 13.35013374,  6.05113217,
       12.27991991,  5.07306274,  9.55593374, 10.71162681, 13.94008948,
        9.75200848])}
{'P': tensor([[1.3845e-02, 3.8465e-01, 1.7320e-01,  ..., 7.4005e-06, 1.8474e-04,
         4.1253e-05],
        [1.0086e-04, 1.0816e-02, 3.3771e-01,  ..., 6.7488e-06, 2.5867e-04,
         2.9859e-05],
        [1.4514e-05, 2.1783e-05, 8.2874e-03,  ..., 4.6928e-06, 5.1720e-04,
         1.0058e-05],
        ...,
        [1.6103e-04, 2.7925e-04, 6.0286e-04,  ..., 1.9592e-01, 5.9495e-03,
         1.9775e-02],
        [1.3539e-05, 1.8387e-05, 3.3281e-05,  ..., 1.6768e-04, 1.2994e-01,
         2.0410e-05],
        [7.7598e-05, 1.9204e-04, 8.5967e-04,  ..., 8.8600e-04, 1.0006e-03,
         2.9016e-02]]), 'X': tensor([[ 12.2589,  -1.9312,   9.8657],
        [ 12.4877,  -0.7222,   8.8347],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.2435,   4.0076,   8.1257],
        [ 12.6035,   0.7494,   7.1536],
        [ 11.6634,   5.8050,   9.1590],
        [  0.3715,  -4.7756,   0.2312],
        [ -9.4630,  -1.5145,   0.2983],
        [  2.3376,  -4.6231,   9.8436],
        [  5.6314,  -1.9302,  -9.6543],
        [ -5.6423,   6.4537,  -1.2091],
        [ -1.3940, -10.7772,   6.7835],
        [ -1.6220,   7.8917,   0.0315],
        [  3.3744,   6.5623,  -6.3636],
        [  2.5620,   7.0549,   8.4825],
        [  5.2911,  -2.4824,  -3.6877],
        [  3.2226,  -4.2802,  -4.0428],
        [ -2.5056,   7.7720,   5.5878],
        [ -2.5161,   7.7699,  -9.3531],
        [  2.5382,   7.0672,   4.5097],
        [  6.2036,   2.4030,  -1.2500],
        [  4.5814, -10.4485,   2.5496],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.3076,   0.7852,  -6.7010],
        [ 12.4837,   2.7510,   0.4449],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.0506,   4.9277,  -9.5094],
        [ -1.0482, -10.8487,  -5.9066],
        [ -6.3489,   5.9117,   6.5161],
        [ -2.0721,   7.8432,   6.6380],
        [ 12.5650,   2.0132,   4.6272],
        [  2.5375,   7.0676,  -4.3490],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.2914,   3.8037,   6.0978],
        [ -9.1624,  -3.4286,   2.0647],
        [  0.1252, -11.0062,   7.2564],
        [  0.9945,  -4.8144,  -0.1897],
        [  0.7145,  -4.8071,  -2.3766],
        [  7.3724,  -9.0577,  -8.9843],
        [  3.7916,  -3.9524,  -7.9547],
        [ -3.5173,   7.5091,  -8.9585],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.7735,   2.5640,  -6.3313],
        [  4.4049,   5.6892,  -4.4394],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -9.0604,   1.7707,   1.1066],
        [  5.0348,  -2.8229,   3.5436],
        [ 12.0042,  -2.8350,   2.6864],
        [ -8.9624,  -4.0944,  -4.9622],
        [  9.6816,   9.1536,  -5.8543],
        [  8.7715,  -7.9390,   5.2900],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -3.9508,  -9.8515,   1.7923],
        [ 12.4691,   2.8525,  -5.5110],
        [  1.7000,  -4.7609,   2.7935],
        [ 11.3691,   6.4760,   1.0787],
        [  5.2939,  -2.4784,  -4.5840],
        [  6.7792,  11.8084,   1.6251],
        [  6.2905,   1.9366,   2.1169],
        [ -9.1433,  -3.5003,  -3.0209],
        [  5.7218,  -9.9934,   6.5764],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.3087,   0.7792,  -7.5587],
        [  5.2608,  12.6977,   4.9166],
        [  1.2728, -11.0372,   9.8121],
        [  2.0419,  -4.6978,  -5.0983],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.4179,  -2.2925,  -3.4641],
        [  9.1820,   9.7334,  -2.7932],
        [ -2.3317,   7.8034,  -3.6274],
        [ -5.1398,  -9.1367,   6.2074],
        [ -9.3739,   0.3982,  -4.2564],
        [  6.2026,  12.1774,  -1.5887],
        [ 10.0995,  -6.4739,   2.0974],
        [  0.9788,  -4.8145,  -3.8661],
        [  8.0381,  -8.5690,  -8.9523],
        [  7.8425,  -8.7202,   6.4530],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -1.7009,   7.8851,  -8.2437],
        [  6.3305,   1.6119,   1.1538],
        [ -7.5401,   4.6718,   9.0033],
        [  7.2340,  -9.1505,  -5.8321],
        [  9.1620,  -7.5573,  -9.9882],
        [  4.4290,  -3.4596,   1.4181],
        [  5.5424,  -2.0885,   7.2487],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -9.4753,  -0.8012,  -5.2050],
        [  4.8432,   5.1968,   1.5006],
        [  9.6816,   9.1536,  -5.8543],
        [ -9.2707,   0.9661,   4.4863],
        [  2.3373,  -4.6232,  -5.2246],
        [ 12.5789,   0.1722,   1.6470],
        [ -5.6792,  -8.7352,   4.0318],
        [ 11.9695,  -2.9407,  -6.3498],
        [  9.9058,  -6.7203,   5.6928],
        [ -0.4341, -10.9473,   7.4509],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.7802,  -1.6363,  -6.2782],
        [  2.3200,   7.1756,   1.1938],
        [ 12.2921,  -1.7902,  -4.0870]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0162, 0.0648,  ..., 1.0678, 0.6750, 1.0639],
        [0.0162, 0.0000, 0.0494,  ..., 1.0705, 0.6820, 1.0658],
        [0.0648, 0.0494, 0.0000,  ..., 1.0622, 0.6886, 1.0551],
        ...,
        [1.0678, 1.0705, 1.0622,  ..., 0.0000, 0.4597, 0.0528],
        [0.6750, 0.6820, 0.6886,  ..., 0.4597, 0.0000, 0.4790],
        [1.0639, 1.0658, 1.0551,  ..., 0.0528, 0.4790, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.41012046, 12.5085995 , 12.71843462, 12.88270018, 12.62576026,
       13.02818702,  4.79003361,  9.58347456,  5.1805345 ,  5.95297459,
        8.5724042 , 10.86693708,  8.05668609,  7.37902766,  7.50564223,
        5.84450127,  5.35774573,  8.16584537,  8.16714836,  7.50917551,
        6.652742  , 11.4088043 ,  9.37296102,  9.34061938, 12.78327037,
        4.89121805,  9.58072286,  7.05627461, 10.8992524 ,  8.67502348,
        8.11227668, 12.72524615,  7.5092834 , 12.91143491, 12.86648771,
        9.78285424, 11.00694688,  4.9160962 ,  4.85995002, 11.67875855,
        5.47702766,  8.29203356, 13.77222752, 12.35625694,  9.84314712,
        9.14044984,  7.19513806, 12.3828957 ,  9.23178204,  5.77219077,
       12.3344536 ,  9.85330646, 13.32374056, 11.83075005, 10.52357479,
        9.51074763, 10.61417181, 12.79126785,  5.05534609, 13.08414689,
        5.84533345, 13.61600536,  6.58184474,  9.79039371, 11.51557166,
       12.58006038,  9.34127013, 13.744387  , 11.11038551,  5.1224079 ,
        8.89458273,  6.02789634,  5.8828986 , 13.38091121,  8.14433542,
       10.48315436,  9.38232121, 13.6660671 , 11.99633819,  4.91296205,
       11.74901778, 11.72802491, 11.62158554,  9.50913849,  8.06643487,
        6.53250937,  8.87007495, 11.66453059, 11.87665501,  5.62005629,
        5.92281142, 10.46739073,  9.50913849,  7.10378919, 13.32374056,
        9.32094055,  5.18047638, 12.58006038, 10.41909211, 12.32545637,
       11.9702762 , 10.95593729,  5.27178363,  6.00732193,  7.54127425,
       12.4217523 ])}
{'P': tensor([[5.3444e-03, 2.2539e-01, 2.1451e-01,  ..., 5.1205e-07, 5.6055e-05,
         2.3181e-05],
        [5.0059e-05, 8.7341e-03, 4.3596e-03,  ..., 1.0168e-06, 1.5851e-04,
         3.0242e-05],
        [3.0280e-05, 2.2853e-03, 5.1927e-03,  ..., 6.1827e-07, 6.8053e-05,
         1.1761e-05],
        ...,
        [8.2511e-06, 5.9851e-06, 6.0704e-06,  ..., 6.9197e-02, 2.9536e-05,
         7.7843e-05],
        [1.1781e-05, 1.3908e-05, 1.2014e-05,  ..., 7.7512e-03, 1.4670e-01,
         2.1510e-04],
        [1.9057e-03, 3.0089e-03, 1.5150e-03,  ..., 3.3220e-05, 1.7909e-03,
         7.6206e-02]]), 'X': tensor([[ 12.2692,  -1.8882,   3.9253],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.5224,  -0.4497,   4.8448],
        [ 12.5650,   2.0132,   4.6272],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.2914,   3.8037,   6.0978],
        [  0.9445,   7.6736,   0.5095],
        [  1.5383, -11.0274,  -4.9666],
        [  0.4119,  14.1019,  -4.1870],
        [ -0.8455, -10.8853,  -6.7400],
        [ -8.7078,   2.7182,   1.8146],
        [ -7.5720,   4.6310,   2.1717],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.5767,   5.5071,   3.9069],
        [ -9.3076,   0.7852,  -6.7010],
        [  9.3560,   9.5391,   9.4943],
        [ 11.5568,   6.0602,   9.3244],
        [  5.0506,   4.9277,  -9.5094],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -1.6906,   7.8860,   7.7142],
        [  6.3505,   0.6235,  -8.5849],
        [  0.9478,  -4.8144,   0.6371],
        [ -8.9178,   2.1936,   7.5388],
        [  1.0391,  -4.8141,  -0.0768],
        [ -6.1899,  -8.3011,  -7.9269],
        [  2.5128, -10.9365,   9.9645],
        [  6.3587,   1.1596,  -3.6629],
        [  6.2419,  -0.2313,   4.2440],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.5083,  -4.1177,   0.2389],
        [  5.3858,   4.4260,  -9.6769],
        [ -7.0948,   5.1940,   9.8171],
        [ 12.6035,   0.7494,   7.1536],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.3236,   0.3011,  -4.6668],
        [ 12.3025,  -1.7440,  -6.6632],
        [  3.3839,  -4.1966,   8.3574],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.1931,  -4.7491,   7.4385],
        [ -7.6525,  -6.6534,   5.1721],
        [ -5.1377,   6.7751,  -0.4438],
        [ 12.5763,   0.1333,  -9.2650],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.2508,   1.0572,   2.9978],
        [ -9.4203,  -2.0340,   2.5856],
        [  0.5600, -11.0321,  -0.8660],
        [ -8.9282,   2.1648,   8.4894],
        [  4.4915,   5.5990,   7.0405],
        [ -9.4739,  -1.2497,   0.5483],
        [  6.2419,  -0.2313,   4.2440],
        [ -9.4714,  -1.3301,  -3.1998],
        [  6.2535,  -0.1709,  -9.4102],
        [  4.4290,  -3.4596,   1.4181],
        [  6.1579,  -0.5961,  -6.5595],
        [  2.5318,  13.7231,   6.3692],
        [ 12.5650,   2.0132,   4.6272],
        [ 11.0275,  -5.0666,   8.4154],
        [  2.0729,   7.2878,   7.0880],
        [  3.1480,  -4.3165,  -7.8002],
        [ -7.5401,   4.6718,   9.0033],
        [ -9.4735,  -1.2620,  -3.6742],
        [  0.7145,  -4.8071,  -2.3766],
        [ -3.0004,   7.6607,  -7.2640],
        [ -5.1046,  -9.1610,   2.3418],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.4309,  -0.0599,  -4.0830],
        [  6.6319,  11.9066,   3.5165],
        [ 12.2311,  -2.0445,   1.3545],
        [  4.7739,  -3.1218,  -4.1188],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.3048,   0.1469,  -9.1691],
        [  7.2340,  -9.1505,  -5.8321],
        [  2.0243,   7.3086,   7.5432],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.2592,   3.9425,   2.0383],
        [  5.4336,  -2.2676,   5.6933],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.3767, -11.0341,   9.9415],
        [  0.6952,  -4.8060,   8.8976],
        [ -3.6381,   7.4683,   7.3544],
        [ 11.2571,   6.7072,   8.3693],
        [ -1.5261,   7.8988,   9.3768],
        [  3.1756, -10.8247,   6.8882],
        [  1.9204,   7.3518,   8.4781],
        [  9.6662,   9.1725,   8.1056],
        [ -5.9940,   6.1985,   5.9566],
        [  6.2535,  -0.1709,  -9.4102],
        [  3.1480,  -4.3165,  -7.8002],
        [  4.9452, -10.3187,  -7.8621],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.4721,  -1.3097,  -7.4377],
        [  6.2576,   2.1359,  -3.3398],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.2373,  -4.6506,  -1.7334],
        [ -3.9858,  -9.8334,  -3.7323],
        [  4.8432,   5.1968,   1.5006],
        [  3.9515,  -3.8425,   6.1729],
        [  5.3860,  -2.3418,   1.0826],
        [  6.0073,   3.0986,   6.7937],
        [ -9.2707,   0.9661,   4.4863],
        [ -0.1963,   7.8765,  -1.1371],
        [  5.1787,  -2.6383,  -4.2533]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1037, 0.1310,  ..., 0.8599, 0.5290, 0.6360],
        [0.1037, 0.0000, 0.2161,  ..., 0.9012, 0.4769, 0.5465],
        [0.1310, 0.2161, 0.0000,  ..., 0.9249, 0.6589, 0.7614],
        ...,
        [0.8599, 0.9012, 0.9249,  ..., 0.0000, 0.7095, 0.9997],
        [0.5290, 0.4769, 0.6589,  ..., 0.7095, 0.0000, 0.2907],
        [0.6360, 0.5465, 0.7614,  ..., 0.9997, 0.2907, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.41366754, 12.52875793, 12.53047489, 12.72524615, 12.87752052,
       12.86648771,  7.73151309, 11.1341798 , 14.10796466, 10.91805683,
        9.12220271,  8.87587457, 12.70489563,  7.16058587,  9.34061938,
       13.3614598 , 13.0493368 ,  7.05627461, 10.51437947,  9.48969978,
        8.06517143,  6.38105015,  4.90676242,  9.18358562,  4.92497514,
       10.35485331, 11.22142195,  6.46356806,  6.24614905,  5.5748624 ,
        9.43112957, 12.07225469, 12.22276278,  6.97106912,  8.79284466,
       12.62576026,  5.83560009,  6.33075682, 12.42554808,  5.39098669,
       11.89855267,  4.75303127, 10.14044489,  8.50278896, 12.5769659 ,
       10.37998078,  9.3109884 ,  9.63742669, 11.04629471,  9.18690018,
        7.17789666,  9.55593374,  6.24614905,  9.5642993 ,  6.25586297,
        5.62005629,  6.18668463, 13.95472331, 12.72524615, 12.13568598,
        7.57685908,  5.34250225,  8.87007495,  9.55720651,  4.85995002,
        8.22728399, 10.48721382,  9.14973319,  9.43112957, 13.62896593,
       12.40074901,  5.70405525, 10.06286074,  6.30647746, 11.66453059,
        7.58378655,  8.51928531, 12.87752052,  5.88782913, 10.35093485,
       11.11969815,  4.85603886,  8.30729352, 13.103709  ,  8.04483495,
       11.28093507,  7.5984809 , 13.32557133,  8.62260327,  6.25586297,
        5.34250225, 11.4424816 ,  5.19577146,  6.49538841,  9.562174  ,
        6.61211224, 12.10942412,  5.16077764, 10.61047596,  7.10378919,
        5.51177501,  5.87305005,  6.7593917 ,  9.32094055,  7.8788965 ,
        5.81199422])}
{'P': tensor([[5.3469e-03, 2.7818e-01, 2.1313e-01,  ..., 8.5339e-07, 1.2540e-07,
         3.5006e-04],
        [1.0074e-05, 4.8916e-03, 6.5525e-01,  ..., 3.5199e-07, 9.1828e-08,
         9.6687e-05],
        [2.0526e-05, 5.6816e-05, 1.2976e-02,  ..., 7.8268e-07, 2.3472e-07,
         1.9429e-04],
        ...,
        [2.8365e-02, 8.4797e-03, 6.3672e-03,  ..., 2.3460e-02, 3.9205e-06,
         6.9404e-03],
        [5.9944e-04, 2.1564e-04, 1.7033e-04,  ..., 4.7801e-02, 1.0191e-01,
         2.0080e-03],
        [6.1981e-03, 1.2528e-02, 1.3894e-02,  ..., 1.6852e-04, 4.0604e-05,
         8.5859e-02]]), 'X': tensor([[ 12.2921,  -1.7902,  -4.0870],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 12.4691,   2.8525,  -5.5110],
        [ 12.5919,   0.4035,  -5.8460],
        [  3.5939,   6.4029,  -1.8357],
        [  6.2535,  -0.1709,  -9.4102],
        [  8.5093,  -8.1764,  -3.2381],
        [  2.0449,  13.8402,  -7.1642],
        [  0.9445,   7.6736,   0.5095],
        [ -9.4735,  -1.2620,  -3.6742],
        [  6.3305,   1.6119,   1.1538],
        [  6.2555,   2.1474,   4.1685],
        [ -6.4620,  -8.0449,   6.2719],
        [  5.1519, -10.2387,   9.8191],
        [  1.3339,   7.5627,  -0.9034],
        [  5.3804,  -2.3504,   7.2744],
        [ -9.4739,  -1.2497,   0.5483],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -4.4520,   7.1370,  -8.7867],
        [ -3.3116, -10.1539,   7.5919],
        [ 10.2470,   8.4054,  -0.1177],
        [  7.3724,  -9.0577,  -8.9843],
        [ -9.4714,  -1.3301,  -3.1998],
        [  9.3624,   9.5318,   0.2779],
        [ 11.9122,   5.1397,  -1.5498],
        [ -4.7389,   6.9953,   3.7638],
        [  3.9895,   6.0816,  -6.5235],
        [ 10.8218,  -5.4177,   6.8573],
        [  3.6249, -10.7255,   5.6605],
        [  0.1382, -11.0073,   1.0026],
        [  4.8432,   5.1968,   1.5006],
        [ 11.6544,   5.8273,  -3.1375],
        [  1.5662,  13.9375,   4.5142],
        [  9.1620,  -7.5573,  -9.9882],
        [  7.5930,  -8.9037,   7.2323],
        [  9.5261,  -7.1675,   1.4206],
        [  5.1196,  -2.7160,   4.2705],
        [  7.7794,  11.0629,   8.9069],
        [ 11.0275,  -5.0666,   8.4154],
        [  1.7301,   7.4262,   5.0522],
        [  1.4489,   7.5256,   4.4529],
        [  6.0031,  12.2958,  -6.5992],
        [ -1.8788, -10.6567,   3.7068],
        [  4.9152,  12.8653,  -1.4459],
        [  0.7275,  14.0667,  -0.9815],
        [  0.9478,  -4.8144,   0.6371],
        [ -5.1974,   6.7396,   3.3484],
        [  8.0877,  -8.5297,  -1.1050],
        [ -3.9858,  -9.8334,  -3.7323],
        [  1.7500,   7.4187,   4.4237],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.0617,  12.7957,   8.3544],
        [ -9.4351,  -1.8891,   3.5642],
        [  7.8982,  -8.6778,   0.0239],
        [  2.2076,  13.8032,  -1.5099],
        [  7.9141,  -8.6656,   9.6948],
        [  6.2630,   2.1057,  -5.2136],
        [  5.8836,   3.4315,  -4.0327],
        [  5.7218,  -9.9934,   6.5764],
        [ 10.1503,   8.5414,  -7.2681],
        [ 10.1503,   8.5414,  -7.2681],
        [ -3.5903,   7.4847,  -8.9112],
        [  3.7992,  -3.9474,  -8.9019],
        [ -9.0574,   1.7801,   8.8986],
        [ -6.4368,   5.8356,   3.3014],
        [  2.3928,   7.1404,  -1.7035],
        [  2.4149,  -4.6005,  -4.3339],
        [ -7.7842,   4.3466,  -1.7548],
        [  6.1973,  -0.4372,   6.5720],
        [  5.5626,  -2.0536,   8.4433],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4671,  -0.5553,   0.1117],
        [ -9.4739,  -1.2497,   0.5483],
        [ 11.0373,   7.1299,  -8.5895],
        [  1.3403,  -4.8011,  -1.1698],
        [  8.6271,  -8.0715,  -7.4813],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.7784,  -1.6401,   9.5587],
        [ 11.4495,   6.3026,   6.8257],
        [  6.0560,  -0.9421,  -5.0334],
        [ -6.3066,  -8.1936,   9.5822],
        [ 10.8218,  -5.4177,   6.8573],
        [  7.5930,  -8.9037,   7.2323],
        [  2.0729,   7.2878,   7.0880],
        [  0.5250, -11.0306,   5.5358],
        [  3.9022,  13.2906,   5.5788],
        [ -2.2188,   7.8218,   8.6064],
        [  5.1793,   4.7461,   0.3899],
        [ -7.7239,   4.4298,   3.1216],
        [  5.7845,   3.6652,   0.9175],
        [  7.8982,  -8.6778,   0.0239],
        [  1.5662,  13.9375,   4.5142],
        [  4.7813,  -3.1139,   8.2691],
        [ -2.8162,   7.7059,  -9.6885],
        [ 12.3609,   3.4769,  -8.5762],
        [ -7.0948,   5.1940,   9.8171],
        [  2.0243,   7.3086,   7.5432],
        [ -6.3252,  -8.1760,  -1.0891],
        [  5.0277,   4.9589,   7.0175],
        [  6.3122,  -9.6994,  -6.3282],
        [ -4.8665,  -9.3204,  -5.2564],
        [  6.0560,  -0.9421,  -5.0334]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0781, 0.1045,  ..., 0.4105, 0.9477, 0.0742],
        [0.0781, 0.0000, 0.0269,  ..., 0.4855, 1.0140, 0.1166],
        [0.1045, 0.0269, 0.0000,  ..., 0.5124, 1.0396, 0.1413],
        ...,
        [0.4105, 0.4855, 0.5124,  ..., 0.0000, 0.5625, 0.3818],
        [0.9477, 1.0140, 1.0396,  ..., 0.5625, 0.0000, 0.8987],
        [0.0742, 0.1166, 0.1413,  ..., 0.3818, 0.8987, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.4217523 , 12.62711252, 12.67576792, 12.49317563, 12.79126785,
       12.59840193,  7.34250948,  6.25586297, 11.80092578, 13.99048094,
        7.73151309,  9.55720651,  6.53250937,  6.61386436, 10.31886066,
       11.46176259,  7.67939546,  5.87132718,  9.55593374,  9.48969978,
        9.69629   , 10.5575601 ,  8.41171369, 10.68031351, 13.25336001,
       11.67875855,  9.5642993 , 13.3607379 , 12.97370371,  8.44940661,
        7.27340834, 12.10222204, 11.32149259, 11.00812996,  7.10378919,
       13.03002338, 14.02526505, 11.87665501, 11.70168755, 11.92133129,
        5.79543038, 13.52430176, 12.13568598,  7.62509582,  7.66377554,
       13.6829843 , 10.82106505, 13.77222752, 14.08549802,  4.90676242,
        8.51088913, 11.75438006, 10.61047596,  7.62232307, 12.7125079 ,
       13.76047999,  9.62238828, 11.73397309, 13.97857591, 11.73567759,
        6.60752373,  6.81117802, 11.51557166, 13.26590071, 13.26590071,
        8.30125201,  5.47866221,  9.23071698,  8.68832398,  7.53063885,
        5.19577146,  8.91551231,  6.21275016,  5.92952538,  5.57341623,
        9.48337063,  9.55593374, 13.1399181 ,  4.98461983, 11.81423161,
        9.43737523,  6.00663314, 13.06957425,  6.12886024, 10.33958639,
       12.10222204, 11.70168755,  7.57685908, 11.04313641, 13.85158417,
        8.1303945 ,  7.02496223,  8.90405823,  6.84796202, 11.73397309,
       14.02526505,  5.70591986,  8.20436764, 12.84056604,  8.79284466,
        7.58378655, 10.33712258,  7.06169391, 11.57249984, 10.51437947,
        6.12886024])}
{'P': tensor([[3.2573e-03, 4.0911e-01, 1.8365e-01,  ..., 2.6106e-04, 3.4787e-05,
         3.4787e-05],
        [1.8812e-05, 4.8189e-03, 3.5900e-01,  ..., 5.4356e-04, 7.5454e-05,
         7.5454e-05],
        [1.5118e-05, 2.2629e-05, 8.5576e-03,  ..., 2.3889e-03, 3.2213e-04,
         3.2213e-04],
        ...,
        [2.2735e-07, 2.7190e-07, 4.3923e-07,  ..., 1.3054e-02, 4.4309e-02,
         4.4309e-02],
        [1.5799e-06, 1.7279e-06, 2.2205e-06,  ..., 1.2840e-04, 2.5708e-01,
         2.5708e-01],
        [1.5799e-06, 1.7279e-06, 2.2205e-06,  ..., 1.2840e-04, 2.5708e-01,
         2.5708e-01]]), 'X': tensor([[ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.1257e+01,  6.7072e+00,  8.3693e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0316, 0.0777,  ..., 0.4016, 0.0895, 0.0895],
        [0.0316, 0.0000, 0.0483,  ..., 0.3885, 0.1154, 0.1154],
        [0.0777, 0.0483, 0.0000,  ..., 0.3527, 0.1471, 0.1471],
        ...,
        [0.4016, 0.3885, 0.3527,  ..., 0.0000, 0.3908, 0.3908],
        [0.0895, 0.1154, 0.1471,  ..., 0.3908, 0.0000, 0.0000],
        [0.0895, 0.1154, 0.1471,  ..., 0.3908, 0.0000, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.42554808, 12.49291768, 12.64235359, 12.59840193, 12.70489563,
       12.77402351, 10.78533838,  5.05444288,  8.62260327,  5.2287058 ,
        9.55593374, 11.1341798 , 14.0393949 ,  5.23878371, 13.91198191,
       13.58101794,  6.28835101, 14.07623266,  6.28961988,  8.51928531,
       10.54658597, 10.4606265 , 10.14044489, 13.103709  ,  6.81961996,
       11.54071859,  5.91222346, 10.92597196, 13.98691426,  7.88677538,
       11.05317597,  9.3109884 , 10.52357479,  6.21275016, 10.09979747,
       12.49317563,  7.32950385, 12.88270018,  5.05534609, 11.00508681,
        8.4486771 ,  7.16598273,  8.87587457,  7.27759233, 13.48809595,
        7.92200342, 13.6105595 , 14.09357547,  5.43943791,  9.84314712,
       11.10066126,  7.19534146,  6.42689705,  6.2134961 ,  5.16611782,
       11.9702762 ,  9.64157461,  9.37296102, 13.744387  , 11.07637531,
       11.70016217, 13.34587027, 13.38091121, 11.28093507, 10.08001894,
       13.59673406,  9.60412735, 13.6922425 ,  8.31625171,  7.44805901,
        7.37902766, 11.63765273,  9.73579738, 13.89974693,  7.02253235,
       12.7125079 , 10.19036081,  8.50278896,  5.2287058 , 11.87665501,
       12.0495521 , 11.35345345, 10.99326577,  9.18690018,  9.23178204,
       12.67576792,  7.50917551,  6.90115901, 10.90638066,  9.17016857,
       11.1341798 ,  5.24137909,  5.35774573, 10.06437598,  7.24677664,
        7.89745107,  7.06372172,  5.98285319,  4.76969364,  9.05843201,
       14.05687815,  7.98469166,  8.04718433, 13.6660671 , 14.09176259,
       14.09176259])}
{'P': tensor([[6.1664e-03, 4.3302e-01, 2.4959e-01,  ..., 6.8164e-05, 1.5660e-05,
         5.7385e-04],
        [1.0429e-05, 5.1307e-03, 5.0479e-01,  ..., 3.3972e-05, 8.4185e-06,
         1.3405e-03],
        [1.0551e-05, 2.9725e-05, 8.1961e-03,  ..., 3.6543e-05, 9.4539e-06,
         3.3715e-03],
        ...,
        [2.0011e-04, 3.2365e-04, 3.6248e-04,  ..., 6.4230e-02, 1.2784e-02,
         2.0260e-03],
        [4.4929e-05, 6.4409e-05, 6.7736e-05,  ..., 1.1554e-02, 4.2394e-02,
         3.9577e-04],
        [1.4309e-06, 2.3665e-06, 3.1685e-06,  ..., 2.6421e-05, 1.4042e-05,
         8.5118e-02]]), 'X': tensor([[ 12.3453,  -1.5462,   7.0212],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.2435,   4.0076,   8.1257],
        [ 12.5650,   2.0132,   4.6272],
        [  6.0345,   3.0175,  -1.4424],
        [ -6.4368,   5.8356,   3.3014],
        [ -6.4844,  -8.0230,   8.5209],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.3413,   0.4883,  -6.8416],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.8602,   6.8905,   5.8779],
        [ -2.1382,   7.8339,   7.3524],
        [ -2.7572,   7.7194,   5.9580],
        [  6.9693,  -9.3200,  -8.3822],
        [ -0.5739,   7.9056,   0.6382],
        [ -8.9942,  -3.9990,   6.0448],
        [ 11.3367,  -4.4806,   0.4643],
        [  0.2732,  -4.7619,  -1.1577],
        [ -8.5900,   2.9767,   7.4811],
        [ 12.0772,  -2.6014,  -7.1832],
        [  0.9036,   7.6840,   9.9237],
        [  9.4925,   9.3811,   8.9842],
        [ -7.4807,   4.7463,   2.0616],
        [  5.1890,   4.7318,   0.9515],
        [  5.8275,  -1.5334,   2.0124],
        [  6.9693,  -9.3200,  -8.3822],
        [ -9.2687,  -2.9794,   0.6436],
        [ 11.3367,  -4.4806,   0.4643],
        [ -3.4488, -10.0934,   2.5195],
        [  6.1372,   2.6726,  -6.7885],
        [ -7.3333,   4.9246,  -7.9212],
        [ -6.4464,   5.8272,   7.5038],
        [ -8.0750,  -6.0072,  -5.8267],
        [  2.4697,  -4.5836,   1.4593],
        [  5.8323,  -1.5228,   5.1132],
        [ -5.3179,  -9.0101,   9.4472],
        [  5.9026,   3.3837,  -0.6573],
        [  0.9478,  -4.8144,   0.6371],
        [ -1.6086,   7.8928,   4.3385],
        [ -8.8745,   2.3097,   3.5600],
        [  3.9515,  -3.8425,   6.1729],
        [  6.2630,   2.1057,  -5.2136],
        [  3.8111, -10.6787,  -4.6539],
        [  1.1644, -11.0394,   7.8511],
        [  2.3928,   7.1404,  -1.7035],
        [  3.3633,  13.4794,   0.5193],
        [ 12.1588,   4.3364,  -8.0867],
        [ -0.1626, -10.9796,   6.4629],
        [ 12.4013,  -1.2597,  -7.6474],
        [  0.9809,   7.6642,  -1.4217],
        [ 10.9931,   7.2107,   7.7933],
        [ -1.6275, -10.7221,   3.2376],
        [  9.3836,   9.5075,  -5.9278],
        [  3.6154,  -4.0640,  -7.9698],
        [  3.9591, -10.6391,  -6.2557],
        [  6.2555,   2.1474,   4.1685],
        [  8.4691,  10.4571,  -5.0650],
        [ -3.3999, -10.1152,  -1.2088],
        [  5.6672,  -1.8631,   8.1020],
        [  6.3589,   0.8239,  -8.4230],
        [  6.3542,   0.6934,  -3.3840],
        [ 12.3609,   3.4769,  -8.5762],
        [ -2.2188,   7.8218,   8.6064],
        [  2.2860,   7.1916,  -6.1011],
        [ -1.8788, -10.6567,   3.7068],
        [  5.2199,   4.6861,  -1.9304],
        [  0.2435,   7.8191,  -6.4709],
        [  4.2275,  -3.6320,  -0.2714],
        [  3.7916,  -3.9524,  -7.9547],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -2.6735, -10.4062,  -1.5911],
        [  5.2188,  -2.5839,   2.1833],
        [  0.6952,  -4.8060,   8.8976],
        [ -7.4602,   4.7718,  -4.6277],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.2990,  -4.2415,  -7.4857],
        [  2.1771, -10.9776,  -3.2007],
        [  6.3266,   0.3292,   9.3502],
        [  0.3038,   7.8092,   0.4164],
        [ -4.7334,   6.9982,  -0.4392],
        [  6.8704,  -9.3809,   7.5769],
        [  5.1886,  -2.6250,   6.4677],
        [ -5.1046,  -9.1610,   2.3418],
        [  4.0587, -10.6112,   6.2273],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.2895,   0.0405,   5.7652],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.0642,  -3.7599,  -3.4565],
        [  8.2422,  -8.4041,  -1.4962],
        [  3.3943,   6.5483,   1.0790],
        [ 12.2692,  -1.8882,   3.9253],
        [  6.3604,   0.9042,  -6.5901],
        [  0.4996,   7.7738,  -4.4711],
        [ -3.2736, -10.1703,   3.8712],
        [  3.9663,   6.1018,  -5.4104],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.3928,   7.1404,  -1.7035],
        [  2.6613,   7.0021,   4.5783],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.3542,   0.6934,  -3.3840],
        [  6.3413,   0.4883,  -6.8416],
        [  5.6928,  12.4710,   3.6001]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0743, 0.1230,  ..., 0.8984, 1.0416, 0.3302],
        [0.0743, 0.0000, 0.0491,  ..., 0.9167, 1.0583, 0.2837],
        [0.1230, 0.0491, 0.0000,  ..., 0.9388, 1.0791, 0.2674],
        ...,
        [0.8984, 0.9167, 0.9388,  ..., 0.0000, 0.1441, 0.7639],
        [1.0416, 1.0583, 1.0791,  ..., 0.1441, 0.0000, 0.8939],
        [0.3302, 0.2837, 0.2674,  ..., 0.7639, 0.8939, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.44177641, 12.62576026, 12.71843462, 12.86648771, 12.88270018,
       12.72524615,  6.74686071,  8.68832398, 10.31584102,  9.11473391,
        9.36093562,  6.36004393, 10.5575601 ,  7.46053281,  8.12043983,
        8.19703512, 11.63765273,  7.92644746,  9.84314712, 12.18998195,
        4.76969364,  9.09119351, 12.35421461,  7.73692436, 13.34587027,
        8.85939626,  7.02253235,  6.02588517, 11.63765273,  9.73579738,
       12.18998195, 10.66631784,  6.69389431,  8.83341608,  8.68979052,
       10.06437598,  5.20659892,  6.02779201, 10.46238054,  6.80370254,
        4.90676242,  8.0550286 ,  9.17016857,  5.51177501,  6.60752373,
       11.33837396, 11.10066126,  7.53063885, 13.89264822, 12.9089559 ,
       10.98076873, 12.46513646,  7.72668467, 13.14690137, 10.84493732,
       13.35832725,  5.43943791, 11.35182589,  6.61386436, 13.45642379,
       10.67132524,  5.96556248,  6.41202986,  6.39187285, 12.84056604,
        8.1303945 ,  7.54620849, 10.82106505,  7.01475376,  7.822854  ,
        5.57341623,  5.47702766, 10.61047596, 10.74409834,  5.82345867,
        4.85603886,  8.85572292,  8.05242591,  5.37343173, 11.1913598 ,
        6.33517349,  7.81509631,  8.4486771 , 11.62770332,  5.81481044,
       10.48721382, 11.36089355,  9.95552626,  6.28961988,  5.1224079 ,
        5.5366698 , 11.77124373,  7.37576154, 12.41366754,  6.42439344,
        7.78980299, 10.68416401,  7.27759233, 10.24557047,  7.53063885,
        7.49076899,  9.18690018,  9.68323686,  6.39187285,  6.36004393,
       13.70893492])}
{'P': tensor([[6.5350e-03, 3.7114e-01, 2.2430e-01,  ..., 1.3669e-04, 7.6669e-07,
         8.4392e-03],
        [1.3560e-05, 5.1686e-03, 5.0852e-01,  ..., 1.7121e-04, 3.2878e-07,
         2.4238e-03],
        [1.3470e-05, 3.1359e-05, 8.6466e-03,  ..., 3.6053e-04, 4.0500e-07,
         1.7487e-03],
        ...,
        [8.0265e-06, 9.5537e-06, 1.0852e-05,  ..., 7.4213e-02, 8.1962e-06,
         1.7152e-05],
        [3.9769e-02, 1.4861e-02, 8.6478e-03,  ..., 4.6955e-05, 4.3866e-02,
         2.0359e-03],
        [1.9153e-05, 8.6100e-05, 2.1139e-04,  ..., 1.7853e-03, 3.7548e-07,
         1.4317e-02]]), 'X': tensor([[ 12.3759,  -1.3943,   6.4659],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.2914,   3.8037,   6.0978],
        [ 12.5650,   2.0132,   4.6272],
        [ 12.2435,   4.0076,   8.1257],
        [ -4.7389,   6.9953,   3.7638],
        [  5.9003,  12.3550,   2.1917],
        [  2.3589,  13.7668,  -3.6501],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.2895,   0.0405,   5.7652],
        [ -7.3554,   4.8985,  -0.6665],
        [ -0.8675, -10.8815,   6.8561],
        [ 10.0400,   8.6922,  -6.2847],
        [ -7.4887,   4.7364,   8.9074],
        [ 12.4691,   2.8525,  -5.5110],
        [  9.9058,  -6.7203,   5.6928],
        [  4.0587, -10.6112,   6.2273],
        [  4.7813,  -3.1139,   8.2691],
        [  3.1480,  -4.3165,  -7.8002],
        [  1.6954,  -4.7616,   9.8198],
        [  5.5158,  -2.1337,   3.5246],
        [  2.1504,  -4.6726,   8.7908],
        [ 12.1588,   4.3364,  -8.0867],
        [  1.5383, -11.0274,  -4.9666],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.5240,  -2.1198,   6.2055],
        [  5.1890,   4.7318,   0.9515],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.0588,  -6.0341,  -2.3006],
        [ 12.2589,  -1.9312,   9.8657],
        [  9.6662,   9.1725,   8.1056],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.5154,   2.5072,  -9.3274],
        [ -3.4986,   7.5152,   9.0712],
        [ -9.4501,  -1.7133,   2.2980],
        [ 12.1504,   4.3674,  -7.2998],
        [  9.6587,   9.1817,  -0.8212],
        [  0.7275,  14.0667,  -0.9815],
        [ -2.4956, -10.4682,   3.5632],
        [  4.0457,  -3.7737,  -0.4704],
        [  4.1297,  -3.7098,  -3.9838],
        [ -9.3604,   0.4855,  -3.0158],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.1379,   4.4125,  -3.8833],
        [  2.1477,   7.2550,   8.9688],
        [  6.2038,   2.4020,  -3.1692],
        [  5.8836,   3.4315,  -4.0327],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.1667,  -2.6543,   1.2879],
        [  2.6451,  -4.5250,  -0.5565],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4541,  -0.3327,   8.8246],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.6801,   2.7809,  -0.8443],
        [ 12.1588,   4.3364,  -8.0867],
        [ -8.6381,  -4.9238,  -4.4077],
        [ 10.5057,  -5.9099,  -5.1599],
        [  9.4554,   9.4245,   9.8800],
        [  1.7122,  -4.7591,  -3.6734],
        [  3.9913,  -3.8139,  -6.8311],
        [ -5.1046,  -9.1610,   2.3418],
        [  2.2849, -10.9655,  -7.6542],
        [  2.5813,  -4.5471,   5.3597],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.1531,  -5.8748,   2.6934],
        [ 12.1940,  -2.1881,   7.1499],
        [ -3.5029,   7.5138,   8.1853],
        [ -6.6895,   5.6051,   2.7899],
        [  9.1409,  -7.5789,   1.5814],
        [ -0.5739,   7.9056,   0.6382],
        [ 11.1469,   6.9237,   1.6160],
        [  8.8474,  10.0868,  -8.8644],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.0534,  -6.0429,   2.4583],
        [  4.6063,  13.0051,  -3.1456],
        [  6.3550,   0.7129,   0.4917],
        [  4.6992,   5.3690,   3.0935],
        [  8.2422,  -8.4041,  -1.4962],
        [ -9.1231,  -3.5739,   8.4708],
        [  5.1519, -10.2387,   9.8191],
        [  3.3839,  -4.1966,   8.3574],
        [ -6.9838,  -7.4965,   0.9796],
        [  6.3535,   0.6792,  -7.5101],
        [ -9.4196,   0.0478,   9.4140],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.1666,   3.7653,   3.6075],
        [  2.4932,   7.0903,  -5.4214],
        [ 10.0567,   8.6697,   4.6329],
        [ -6.1899,  -8.3011,  -7.9269],
        [  4.5800,   5.5034,  -3.9432],
        [  7.7750,  11.0665,   2.8364],
        [  0.1677,  -4.7447,  -6.1529],
        [  1.7903,  -4.7467,   0.1168],
        [ 10.9875,   7.2207,  -8.8549],
        [  2.0870,  -4.6877,   9.6823],
        [  3.3069,   6.6089,   8.7377],
        [ -4.7334,   6.9982,  -0.4392],
        [  2.2373,  -4.6506,  -1.7334],
        [  2.1120,  -4.6818,   8.6144],
        [  0.6933,   7.7336,   2.1179],
        [  5.7218,  -9.9934,   6.5764],
        [ 12.3357,   3.6001,   0.7903]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0815, 0.1295,  ..., 0.5597, 0.5032, 0.5854],
        [0.0815, 0.0000, 0.0491,  ..., 0.6352, 0.5767, 0.6210],
        [0.1295, 0.0491, 0.0000,  ..., 0.6753, 0.6254, 0.6343],
        ...,
        [0.5597, 0.6352, 0.6753,  ..., 0.0000, 0.4628, 0.4295],
        [0.5032, 0.5767, 0.6254,  ..., 0.4628, 0.0000, 0.8063],
        [0.5854, 0.6210, 0.6343,  ..., 0.4295, 0.8063, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.45418236, 12.62576026, 12.71843462, 12.86648771, 12.72524615,
       12.88270018,  8.44940661, 13.69162795, 13.96746816,  5.34250225,
        6.28961988,  8.83725081, 10.91601655, 13.27994304,  8.86082124,
       12.79126785, 11.9702762 , 11.36089355,  5.70591986,  5.34250225,
        5.05444288,  5.91408504,  5.14371005, 12.9089559 , 11.1341798 ,
       10.06850677,  5.91677188,  7.02253235,  9.84314712, 10.0674887 ,
       12.41012046, 13.32557133,  8.43626114, 12.76408562,  8.28968067,
        9.60412735, 12.91143491, 13.32645552, 14.08549802, 10.76154256,
        5.53255882,  5.5513072 ,  9.37296102,  9.11473391, 12.91505111,
        7.56617037,  6.65259775,  6.81117802,  4.9160962 ,  5.80860165,
        5.24137909,  7.13501264,  9.45995955,  9.0368458 ,  9.11473391,
       12.9089559 ,  9.94285511, 12.05395056, 13.35013374,  5.05774295,
        5.5205189 , 10.48721382, 11.20100795,  5.2287058 , 10.09979747,
       12.1494023 , 12.30695098, 10.04916678, 12.38881635,  8.29022197,
        8.7273569 , 11.87412439,  7.92644746, 13.12218013, 13.41713661,
        9.5014146 ,  8.12043983, 10.06850677, 13.79676654,  6.3949043 ,
        7.13501264, 11.77124373,  9.79814997, 11.46176259,  5.39098669,
       10.24557047,  6.38968357,  9.41970031, 12.42554808,  8.99276564,
        7.5158489 , 13.2778378 , 10.35485331,  7.15990676, 13.5247215 ,
        4.74770899,  5.07306274, 13.14777433,  5.13125513,  7.39004226,
        8.4486771 ,  5.16077764,  5.13616224,  7.76457095, 11.51557166,
       12.85032836])}
{'P': tensor([[5.4798e-03, 2.8999e-01, 1.6766e-01,  ..., 1.5163e-05, 6.2656e-06,
         1.2477e-04],
        [9.1763e-06, 6.1617e-03, 1.9994e-04,  ..., 7.9064e-06, 3.7048e-06,
         3.1070e-04],
        [2.2353e-05, 5.2209e-02, 6.0496e-03,  ..., 1.3527e-05, 4.7968e-06,
         3.2244e-04],
        ...,
        [5.9668e-04, 7.9146e-04, 1.2607e-03,  ..., 6.9806e-02, 1.1298e-03,
         2.2530e-03],
        [6.1309e-03, 3.8516e-03, 5.8238e-03,  ..., 2.2574e-02, 1.1251e-01,
         1.5328e-03],
        [5.9363e-06, 8.6799e-06, 8.4761e-06,  ..., 6.9551e-05, 1.2587e-04,
         5.1651e-02]]), 'X': tensor([[ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0764, 0.0581,  ..., 0.7561, 0.4135, 0.4957],
        [0.0764, 0.0000, 0.0742,  ..., 0.7849, 0.4648, 0.5571],
        [0.0581, 0.0742, 0.0000,  ..., 0.7156, 0.3906, 0.4845],
        ...,
        [0.7561, 0.7849, 0.7156,  ..., 0.0000, 0.3887, 0.4215],
        [0.4135, 0.4648, 0.3906,  ..., 0.3887, 0.0000, 0.1285],
        [0.4957, 0.5571, 0.4845,  ..., 0.4215, 0.1285, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.46513646, 12.70489563, 12.64235359, 12.77402351, 12.84056604,
       12.76408562,  7.28793663,  5.29755487, 13.69162795,  8.40445872,
        8.50278896,  9.62238828,  6.7593917 ,  7.48538905,  7.10378919,
        7.23276582, 12.22503754,  6.80370254, 11.54071859,  6.67108662,
        6.4020404 ,  8.8737556 ,  8.05668609,  6.19633139,  9.09661212,
       10.67132524, 10.05558893,  9.32094055,  8.06643487, 10.42667093,
        8.41171369, 10.24557047, 13.01728313,  9.68324553, 13.41713661,
       13.89671314, 13.69162795,  9.15976746, 10.51437947,  5.96556248,
       10.1371552 , 12.41012046,  4.89121805,  4.83371341,  6.80559669,
       13.04156075, 12.42554808,  9.17016857,  7.71075226,  6.27715105,
        8.14433542,  7.8575648 ,  8.62260327, 10.51340829, 13.83139024,
        9.17905491, 10.87409005,  7.15990676, 11.11038551, 11.00812996,
        8.51928531,  8.90405823,  6.7593917 , 11.83570132, 13.35832725,
        5.13125513, 10.54658597, 12.10942412,  6.24187044, 12.71843462,
       12.27993098,  7.71075226, 11.97949273,  9.97822473,  6.89245352,
        9.20340031, 11.52733581, 12.78327037, 11.69264243,  9.15866607,
        6.39187285,  7.92200342,  6.4020404 , 12.889695  ,  9.04255655,
       12.32545637,  8.81930525, 10.4977132 ,  9.86089579,  9.48337063,
        9.45995955,  4.89121805,  7.34250948,  7.96181288,  6.81117802,
        6.74872981,  6.61386436,  5.36061974, 11.67875855, 12.52875793,
        7.72668467,  5.14371005,  5.48276038,  5.95111804,  5.23878371,
        7.71075226])}
{'P': tensor([[3.5574e-03, 2.6502e-01, 1.6258e-01,  ..., 6.3959e-05, 7.2835e-08,
         4.3179e-05],
        [1.3961e-05, 5.2798e-03, 4.5566e-02,  ..., 6.6567e-05, 8.3215e-08,
         3.1833e-05],
        [6.6556e-05, 1.0478e-03, 3.2292e-02,  ..., 2.1472e-04, 4.0042e-07,
         1.3502e-04],
        ...,
        [4.4269e-04, 8.1067e-04, 6.0972e-04,  ..., 5.1494e-02, 2.1631e-05,
         1.2273e-03],
        [2.9680e-04, 1.6069e-04, 9.1318e-05,  ..., 8.1988e-04, 2.6333e-02,
         1.6650e-03],
        [2.3767e-02, 3.0089e-02, 2.3791e-02,  ..., 2.7371e-02, 1.2261e-04,
         1.5737e-01]]), 'X': tensor([[ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0483, 0.0926,  ..., 0.6366, 1.0065, 0.2644],
        [0.0483, 0.0000, 0.0742,  ..., 0.6223, 1.0316, 0.2674],
        [0.0926, 0.0742, 0.0000,  ..., 0.6921, 1.0985, 0.3415],
        ...,
        [0.6366, 0.6223, 0.6921,  ..., 0.0000, 0.7639, 0.3927],
        [1.0065, 1.0316, 1.0985,  ..., 0.7639, 0.0000, 0.7940],
        [0.2644, 0.2674, 0.3415,  ..., 0.3927, 0.7940, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.49291768, 12.64235359, 12.70489563, 12.77402351, 12.59840193,
       12.79126785,  9.14973319, 13.35832725, 11.95743558,  9.34061938,
       13.97430191,  8.66273171,  4.83371341,  5.43943791,  9.69792513,
        6.74872981, 10.36844813, 10.8207297 ,  8.81930525,  6.59176018,
       10.36844813,  9.13507321,  7.98469166,  4.98461983,  6.81084724,
        9.95552626, 13.744387  , 13.36092624,  6.67218687,  8.29203356,
        5.87305005,  9.76896904,  6.84796202, 11.9120321 ,  5.19577146,
       13.99048094, 11.89855267, 10.31584102, 11.62770332,  8.87007495,
        8.35856487,  6.23262289,  7.92607342,  9.54154096, 11.73567759,
        5.1805345 ,  5.79543038, 10.08001894,  8.08320235, 11.4424816 ,
        7.73151309,  9.62238828,  5.05981914, 10.95593729,  9.562174  ,
       13.99877413,  7.78980299, 10.74409834,  8.99791316,  9.75401449,
        9.94285511,  7.57946459, 12.67680439,  8.04483495, 12.41366754,
       12.70489563, 11.31276109, 13.98691426,  7.07902036,  6.48486171,
        8.50268322, 13.744387  ,  9.34061938,  9.49511524, 13.00413829,
        5.69913817,  5.29009836, 13.97857591,  7.46248947,  7.71075226,
        9.60412735,  8.09899874,  9.75075273,  5.47866221,  4.85603886,
       11.75438006, 13.99048094, 13.35832725, 10.42667093,  9.97220799,
        5.16611782,  7.66377554, 13.62896593,  7.15100303,  7.60529892,
        8.99276564, 11.00812996,  4.90676242,  7.39004226,  6.27715105,
        8.98765644,  5.2287058 ,  8.16714836,  6.32572323, 10.90638066,
        5.90035324])}
{'P': tensor([[4.5423e-03, 1.6019e-01, 1.5782e-01,  ..., 1.1251e-05, 4.4806e-02,
         4.0858e-05],
        [7.4646e-06, 6.5454e-03, 1.2123e-05,  ..., 5.5881e-06, 8.1806e-02,
         7.0727e-05],
        [2.1625e-05, 2.0111e-01, 2.9883e-03,  ..., 4.5369e-06, 7.1723e-02,
         2.1699e-05],
        ...,
        [1.3396e-02, 7.7759e-03, 9.2454e-03,  ..., 1.5017e-01, 3.7366e-03,
         5.2806e-04],
        [7.5665e-06, 1.0911e-04, 1.1688e-05,  ..., 4.7969e-06, 9.9800e-03,
         4.4995e-05],
        [1.8522e-06, 5.6333e-06, 2.2781e-06,  ..., 1.1816e-05, 6.3892e-06,
         1.1733e-01]]), 'X': tensor([[ 12.4597,  -0.9136,  -4.8763],
        [ 12.4691,   2.8525,  -5.5110],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.6000,   1.3839,  -3.2992],
        [  7.3592,  11.3937,   6.2407],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.7845,   3.6652,   0.9175],
        [  6.3550,   0.7129,   0.4917],
        [  2.5469,  13.7192,  -9.6935],
        [  2.5620,   7.0549,   8.4825],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.3589,  13.7668,  -3.6501],
        [  5.2199,   4.6861,  -1.9304],
        [ -8.6112,   2.9319,  -7.9943],
        [ 10.3251,   8.2928,  -3.0006],
        [ -8.5900,   2.9767,   7.4811],
        [ -3.0644,   7.6439,  -1.4322],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -9.4716,  -0.6673,   7.6068],
        [  7.0664,  -9.2590,   8.0879],
        [ 12.4996,   2.6335,  -6.3166],
        [  3.2990,  -4.2415,  -7.4857],
        [  4.0587, -10.6112,   6.2273],
        [  9.3424,  -7.3685,  -3.6807],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 10.6283,  -5.7253,   7.4878],
        [  5.3924, -10.1396,  -4.0863],
        [ -1.5875,   7.8944,  -9.4286],
        [ -1.5875,   7.8944,  -9.4286],
        [ -8.8934,   2.2597,   3.5831],
        [  5.2608,  -2.5256,  -9.1446],
        [ -9.1623,  -3.4288,  -4.9961],
        [  3.9591, -10.6391,  -6.2557],
        [ -3.1671,   7.6158,   6.0765],
        [  3.9913,  -3.8139,  -6.8311],
        [ -6.6895,   5.6051,   2.7899],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.3460,   1.4252,   1.2291],
        [ 12.1588,   4.3364,  -8.0867],
        [ 10.6673,  -5.6649,  -4.1646],
        [  9.1620,  -7.5573,  -9.9882],
        [  6.0345,   3.0175,  -1.4424],
        [ 11.6019,   5.9540,  -5.1715],
        [ -0.1963,   7.8765,  -1.1371],
        [  7.5309,  11.2617,   7.2656],
        [ -3.3116, -10.1539,   7.5919],
        [ -9.4157,   0.0821,  -5.0025],
        [  9.7328,   9.0902,  -0.8917],
        [ -9.3419,   0.5972,  -1.2841],
        [ 10.0400,   8.6922,  -6.2847],
        [ 10.0400,   8.6922,  -6.2847],
        [  1.2273,  -4.8081,   9.8129],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -1.0012, -10.8576,   2.2867],
        [  2.7558,  -4.4844,   0.0636],
        [ -2.0721,   7.8432,   6.6380],
        [  9.3560,   9.5391,   9.4943],
        [  0.9392,  -4.8143,  -3.4649],
        [  6.2247,  -0.3150,  -9.3657],
        [  0.1382, -11.0073,   1.0026],
        [ -5.9225,  -8.5355,   2.4778],
        [ -3.2736, -10.1703,   3.8712],
        [  6.9653,  -9.3225,  -3.4727],
        [  3.9913,  -3.8139,  -6.8311],
        [ -8.6381,  -4.9238,  -4.4077],
        [  3.0719, -10.8449,  -4.1673],
        [  6.3549,   0.7096,   6.9198],
        [  4.2339,  -3.6268,   6.3125],
        [ -9.4157,   0.0821,  -5.0025],
        [ -2.5056,   7.7720,   5.5878],
        [  8.2422,  -8.4041,  -1.4962],
        [ 11.3800,  -4.3919,   1.4066],
        [  9.6816,   9.1536,  -5.8543],
        [  0.1047, -11.0046,   0.5188],
        [  6.0788,  -0.8710,  -6.8719],
        [ -7.4887,   4.7364,   8.9074],
        [ 12.5717,   1.9266,   7.4903],
        [  6.3413,   0.4883,  -6.8416],
        [  5.3924, -10.1396,  -4.0863],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -0.8885,   7.9160,   0.0571],
        [ -0.7599, -10.8995,  -6.0366],
        [ 10.4758,  -5.9539,   1.3667],
        [ -3.9456,  -9.8542,   0.7466],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 11.6544,   5.8273,  -3.1375],
        [ -6.6895,   5.6051,   2.7899],
        [  6.0345,   3.0175,  -1.4424],
        [  3.3633,  13.4794,   0.5193],
        [ -6.4620,  -8.0449,   6.2719],
        [ -4.1296,   7.2803,  -2.7194],
        [ -3.4488, -10.0934,   2.5195],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.7735,   2.5640,  -6.3313],
        [ -4.9217,   6.8979,  -1.7667],
        [  3.9663,   6.1018,  -5.4104],
        [ -0.5739,   7.9056,   0.6382],
        [ 10.2712,   8.3707,  -4.9743],
        [ -3.4986,   7.5152,   9.0712],
        [  4.1140,  -3.7220,  -0.1273],
        [ 12.1504,   4.3674,  -7.2998],
        [  8.1528,  10.7452,   9.1493]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1031, 0.0742,  ..., 0.4127, 0.1933, 1.2612],
        [0.1031, 0.0000, 0.0602,  ..., 0.4855, 0.1045, 1.2776],
        [0.0742, 0.0602, 0.0000,  ..., 0.4848, 0.1225, 1.3120],
        ...,
        [0.4127, 0.4855, 0.4848,  ..., 0.0000, 0.5899, 0.9434],
        [0.1933, 0.1045, 0.1225,  ..., 0.5899, 0.0000, 1.3636],
        [1.2612, 1.2776, 1.3120,  ..., 0.9434, 1.3636, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.49317563, 12.79126785, 12.59840193, 12.77402351, 12.64235359,
       12.67576792, 13.56367857, 12.7125079 ,  6.84796202,  6.3949043 ,
       13.95361098,  7.50564223,  9.97220799, 13.96746816,  7.01475376,
        9.09661212, 13.2430377 ,  9.09119351,  8.23525433, 10.35485331,
        9.49511524, 11.64746882, 12.77402351,  5.37343173, 11.36089355,
       11.89855267, 12.49317563, 12.07225469, 11.48434767,  8.05242591,
        8.05242591,  9.17595899,  5.83560009,  9.78286733, 11.35182589,
        8.24808832,  5.5205189 ,  8.7273569 , 11.69264243,  6.50409372,
       12.9089559 , 12.07818871, 11.87665501,  6.74686071, 13.04051744,
        7.8788965 , 13.54774355, 10.68031351,  9.41605684, 13.31764181,
        9.36093562, 13.27994304, 13.27994304,  4.96231532, 12.46513646,
       10.9036226 ,  5.26342203,  8.11227668, 13.3614598 ,  4.90505566,
        6.23262289, 11.00812996, 10.38897563, 10.68416401, 11.63724792,
        5.5205189 ,  9.94285511, 11.27160035,  6.39438044,  5.5748624 ,
        9.41605684,  8.16584537, 11.77124373, 12.19805394, 13.32374056,
       11.00508681,  6.14086582,  8.86082124, 12.71843462,  6.36004393,
       11.48434767, 14.07623266,  9.97822473,  7.96575782, 10.92597196,
       12.0495521 , 10.61472636,  9.79039371, 13.03002338,  8.7273569 ,
        6.74686071, 13.89264822, 10.31886066,  8.36995953, 10.66631784,
        9.97822473,  9.38232121,  9.14044984,  8.47371868,  7.27759233,
        7.92644746, 13.25017211,  8.28968067,  5.54776934, 12.91143491,
       13.48809595])}
{'P': tensor([[1.8326e-03, 2.8934e-01, 2.2745e-01,  ..., 1.9485e-06, 1.9730e-06,
         1.8326e-03],
        [1.0142e-05, 2.3123e-03, 3.3228e-01,  ..., 2.2420e-06, 2.1259e-06,
         1.0142e-05],
        [1.3692e-05, 1.7018e-05, 3.9236e-03,  ..., 3.4093e-06, 3.0488e-06,
         1.3692e-05],
        ...,
        [6.0059e-03, 5.5242e-03, 4.9978e-03,  ..., 8.0931e-02, 1.8420e-04,
         6.0059e-03],
        [2.2055e-04, 2.5118e-04, 3.0729e-04,  ..., 2.3268e-05, 1.4243e-01,
         2.2055e-04],
        [1.8326e-03, 2.8934e-01, 2.2745e-01,  ..., 1.9485e-06, 1.9730e-06,
         1.8326e-03]]), 'X': tensor([[ 12.4690,  -0.8528,  -9.3374],
        [ 12.5258,  -0.4200,  -9.3493],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.3609,   3.4769,  -8.5762],
        [ -0.5385,   7.9037,   6.3830],
        [ -7.4808,   4.7462,   2.0266],
        [  2.9461,  -4.4077,  -0.4831],
        [  7.0358,  -9.2784,  -9.8479],
        [ -2.2508, -10.5479,  -2.7235],
        [ -6.1899,  -8.3011,  -7.9269],
        [  0.2732,  -4.7619,  -1.1577],
        [ -4.7334,   6.9982,  -0.4392],
        [ -5.6174,   6.4708,  -1.7339],
        [  7.9403,  -8.6455,  -8.7433],
        [ -8.9606,  -4.0994,   1.7886],
        [  0.4119,  14.1019,  -4.1870],
        [  5.7218,  -9.9934,   6.5764],
        [  2.7685, -10.8982,  -6.8928],
        [  6.3460,   1.4252,   1.2291],
        [ -2.0445,   7.8469,  -0.5857],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.2911,  -2.4824,  -3.6877],
        [ 12.4996,   2.6335,  -6.3166],
        [  3.4140,  -4.1803,  -8.8224],
        [ -0.8455, -10.8853,  -6.7400],
        [ 11.3800,  -4.3919,   1.4066],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.1142,  -2.7230,  -4.1966],
        [  9.9217,  -6.7006,  -5.6728],
        [  1.7000,  -4.7609,   2.7935],
        [ 10.9931,   7.2107,   7.7933],
        [  9.3624,   9.5318,   0.2779],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -2.8162,   7.7059,  -9.6885],
        [  6.3569,   0.7591,   6.5011],
        [ -4.2040,   7.2486,   2.9875],
        [  7.4132,  11.3526,   0.3322],
        [  3.2647, -10.8066,  -0.3768],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.4309,  -0.0599,  -4.0830],
        [  1.7903,  -4.7467,   0.1168],
        [  7.2340,  -9.1505,  -5.8321],
        [ 10.7566,   7.6212,  -2.3279],
        [  0.2435,   7.8191,  -6.4709],
        [  1.4306,  13.9620,  -0.2075],
        [  4.5814, -10.4485,   2.5496],
        [ -9.3616,  -2.4750,   3.7765],
        [  8.3791,  10.5409,   5.5590],
        [ -1.6906,   7.8860,   7.7142],
        [ 11.3367,  -4.4806,   0.4643],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 12.5267,  -0.4125,  -8.7331],
        [  2.0419,  -4.6978,  -5.0983],
        [ -2.1533, -10.5778,   1.0112],
        [  3.9085,   6.1512,  -2.2560],
        [  8.2422,  -8.4041,  -1.4962],
        [ 12.6059,   0.9597,  -6.3264],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.1795,  -3.3623,   8.5259],
        [  2.2644,  -4.6434,   9.7508],
        [  0.2480, -11.0153,  -3.0183],
        [  6.2419,  -0.2313,   4.2440],
        [  2.3589,  13.7668,  -3.6501],
        [  7.0171,  11.6439,  -5.0178],
        [  6.1575,  12.2046,   3.3257],
        [  4.4633,   5.6287,  -0.7754],
        [ -1.3940, -10.7772,   6.7835],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.0348,  -2.8229,   3.5436],
        [ -8.4267,  -5.3697,   2.4172],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.2939,   6.6325,  -5.8311],
        [ 12.4592,  -0.9168,  -6.7571],
        [  1.9312,  13.8649,   8.3563],
        [  6.2804,   2.0019,   0.8607],
        [ 11.9695,  -2.9407,  -6.3498],
        [  5.9655,   3.2172,  -7.5953],
        [ -3.3812, -10.1235,   7.4384],
        [ -3.2736, -10.1703,   3.8712],
        [ -7.2515,   5.0195,   4.6434],
        [ 12.5154,   2.5072,  -9.3274],
        [  6.3303,   1.6145,  -8.4083],
        [  0.1047, -11.0046,   0.5188],
        [  0.1931,  -4.7491,   7.4385],
        [  2.0039,   7.3173,  -1.5885],
        [  2.9461,  -4.4077,  -0.4831],
        [  4.3459,  -3.5327,   0.3667],
        [  5.9526,   3.2525,  -4.3821],
        [  6.3516,   0.6425,   0.3425],
        [  6.2038,   2.4020,  -3.1692],
        [  5.2608,  -2.5256,  -9.1446],
        [ -7.4808,   4.7462,   2.0266],
        [  6.2675,  12.1379,  -8.9720],
        [  5.6672,  -1.8631,   8.1020],
        [  5.1142,  -2.7230,  -4.1966],
        [  3.4920,  -4.1366,  -8.3955],
        [ -5.9938,  -8.4745,   3.9735],
        [ 12.2589,  -1.9312,   9.8657],
        [ -0.5739,   7.9056,   0.6382],
        [  2.0899,  -4.6870,  -7.4260],
        [ 12.3759,  -1.3943,   6.4659],
        [ 12.4690,  -0.8528,  -9.3374]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0209, 0.0397,  ..., 0.3137, 1.2405, 0.0000],
        [0.0209, 0.0000, 0.0194,  ..., 0.3318, 1.2475, 0.0209],
        [0.0397, 0.0194, 0.0000,  ..., 0.3446, 1.2472, 0.0397],
        ...,
        [0.3137, 0.3318, 0.3446,  ..., 0.0000, 1.0209, 0.3137],
        [1.2405, 1.2475, 1.2472,  ..., 1.0209, 0.0000, 1.2405],
        [0.0000, 0.0209, 0.0397,  ..., 0.3137, 1.2405, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.4980845 , 12.53285633, 12.5769659 , 12.59969936, 12.76408562,
       12.84056604,  7.92200342,  8.859416  ,  5.30158497, 11.64436684,
       10.78533838, 10.35485331,  4.76969364,  8.4486771 ,  8.56889738,
       11.73848065,  9.85384623, 14.10796466, 11.51557166, 11.24434374,
        6.50409372,  8.10886021, 11.99633819,  6.10774544,  5.84450127,
       12.77402351,  5.39723241, 10.91805683, 12.19805394,  9.43737523,
        5.79391825, 11.97238624,  5.05534609, 13.14690137, 13.3607379 ,
       10.52357479,  8.20436764,  6.4020404 ,  8.37954853, 13.55868684,
       11.28895489, 11.66453059,  9.43112957,  5.07306274, 11.66453059,
       13.18278606,  7.822854  , 14.03505624, 11.4088043 ,  9.68324553,
       13.46553623,  8.06517143, 12.18998195,  9.51074763, 12.53345445,
        5.1224079 , 10.794755  ,  7.28793663, 11.77124373, 12.64235359,
       10.794755  ,  9.77588801,  5.16611782, 11.018084  ,  6.24614905,
       13.96746816, 13.59480906, 13.66991027,  7.18354885, 10.86693708,
       12.59840193,  5.77219077,  9.99211663,  9.62238828,  6.81961996,
       13.09737053, 12.49291768, 13.99877413,  6.59176018, 12.32545637,
        6.77777606, 10.67322527, 10.68416401,  8.81930525, 12.76408562,
        6.53290663, 11.00508681,  4.75303127,  7.58668144,  5.30158497,
        5.60064319,  6.78325151,  6.38399789,  6.65259775,  5.83560009,
        8.859416  , 13.66051884,  5.96556248,  5.79391825,  5.41348121,
       10.37998078, 12.41012046,  7.92644746,  5.13182506, 12.45418236,
       12.4980845 ])}
{'P': tensor([[7.7684e-03, 2.4255e-01, 1.8283e-01,  ..., 3.4045e-05, 4.7416e-06,
         2.2545e-04],
        [1.1436e-05, 4.3510e-03, 2.8922e-01,  ..., 3.9711e-05, 2.1227e-06,
         9.0694e-05],
        [7.0160e-06, 2.1927e-05, 9.2823e-03,  ..., 1.6408e-04, 2.4384e-06,
         7.9377e-05],
        ...,
        [5.4203e-06, 5.5396e-06, 6.2640e-06,  ..., 8.8639e-02, 1.8230e-05,
         5.0013e-05],
        [6.5906e-04, 1.1524e-03, 8.1399e-04,  ..., 9.5391e-05, 1.5232e-01,
         3.5541e-02],
        [7.3695e-04, 2.3052e-03, 3.0137e-03,  ..., 8.8887e-04, 1.0892e-03,
         8.1097e-02]]), 'X': tensor([[ 12.4877,  -0.7222,   8.8347],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.2435,   4.0076,   8.1257],
        [ 12.6035,   0.7494,   7.1536],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.5568,   6.0602,   9.3244],
        [ -1.9714,   7.8563,  -1.8084],
        [ -0.9715, -10.8630,   0.3555],
        [  4.3554,   5.7393,  -9.0766],
        [  3.3633,  13.4794,   0.5193],
        [ -1.3178, -10.7939,   0.4236],
        [  6.3497,   1.3680,  -1.9076],
        [ 11.7794,   5.5086,  -9.6929],
        [  4.0642,  -3.7599,  -3.4565],
        [ -3.5520,   7.4976,   9.1522],
        [  1.1644, -11.0394,   7.8511],
        [ -0.0410,   7.8591,   9.3457],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.9808,   4.9340,  -3.3715],
        [ -1.5875,   7.8944,  -9.4286],
        [ -3.4100,   7.5436,  -7.6980],
        [ -2.0229, -10.6163,   8.6930],
        [ 11.1420,   6.9331,  -2.4779],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.1625,  -4.6697,   7.0658],
        [  4.8051, -10.3703,   0.5268],
        [  0.4119,  14.1019,  -4.1870],
        [  5.5424,  -2.0885,   7.2487],
        [  6.0054,  -1.0902,   8.3079],
        [  2.0243,   7.3086,   7.5432],
        [  1.1139,   7.6281,  -2.5680],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.1730,  -0.5375,  -8.8861],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.5663,   7.0526,  -6.8769],
        [ -2.2508, -10.5479,  -2.7235],
        [ -1.0482, -10.8487,  -5.9066],
        [  3.2692,  13.5098,  -4.9032],
        [ -1.9646,   7.8571,   3.3817],
        [  6.3583,   1.1729,   8.7835],
        [ -6.6895,   5.6051,   2.7899],
        [  7.0358,  -9.2784,  -9.8479],
        [ -9.4766,  -1.1116,  -2.1706],
        [  7.3592,  11.3937,   6.2407],
        [ -9.3331,  -2.6465,  -8.2028],
        [ 11.5966,   5.9666,   6.0083],
        [  0.1931,  -4.7491,   7.4385],
        [ -9.1813,  -3.3550,   1.8827],
        [  2.6078, -10.9229,   6.2869],
        [  4.1999,  -3.6543,  -4.1998],
        [ -6.0195,   6.1789,  -2.9496],
        [ -5.6792,  -8.7352,   4.0318],
        [  5.4167, -10.1293,   6.2156],
        [ -1.6086,   7.8928,   4.3385],
        [  0.9392,  -4.8143,  -3.4649],
        [  1.0093,   7.6567,   2.3526],
        [ 12.2914,   3.8037,   6.0978],
        [ -9.4766,  -1.1116,  -2.1706],
        [  2.8890,  -4.4316,   1.1600],
        [  4.5814, -10.4485,   2.5496],
        [  9.3424,  -7.3685,  -3.6807],
        [ -4.8665,  -9.3204,  -5.2564],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.6469,   6.3624,  -1.8414],
        [  9.6614,   9.1784,  -1.2087],
        [ -9.4505,  -1.7081,  -0.7039],
        [  5.4167, -10.1293,   6.2156],
        [ 10.9916,   7.2133,  -0.9179],
        [  0.2435,   7.8191,  -6.4709],
        [ -1.0407,   7.9166,  -6.6021],
        [ -3.0008, -10.2827,  -8.9480],
        [  2.9470,  13.6085,   2.9809],
        [  2.3373,  13.7721,  -0.2442],
        [  5.6258,  -1.9404,  -6.0715],
        [ -3.2486,   7.5925,  -1.3719],
        [ -3.5173, -10.0623,   4.8280],
        [  1.4306,  13.9620,  -0.2075],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 12.4691,   2.8525,  -5.5110],
        [  0.5960, -11.0335,   3.1144],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.2804,   2.0019,   0.8607],
        [  9.9750,  -6.6338,  -1.2440],
        [  0.1047, -11.0046,   0.5188],
        [ -8.1666,   3.7653,   3.6075],
        [  5.1292,   4.8183,  -5.7997],
        [  9.0424,  -7.6779,  -7.5286],
        [ 10.5955,   7.8824,  -3.7006],
        [  0.9392,  -4.8143,  -3.4649],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.3048,   0.1469,  -9.1691],
        [ -8.4267,  -5.3697,   2.4172],
        [ -1.0012, -10.8576,   2.2867],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -8.1426,   3.8048,   9.3542],
        [  1.0000,   7.6591,  -4.5011],
        [  1.5662,  13.9375,   4.5142],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.2435,   4.0076,   8.1257],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -5.6174,   6.4708,  -1.7339],
        [ -3.4986,   7.5152,   9.0712],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.3551,   0.7149,   1.3344]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.9448e-02, 1.1266e-01,  ..., 5.4743e-01, 9.8774e-01,
         6.1875e-01],
        [4.9448e-02, 0.0000e+00, 6.9626e-02,  ..., 5.8966e-01, 9.8153e-01,
         6.1479e-01],
        [1.1266e-01, 6.9626e-02, 0.0000e+00,  ..., 6.5864e-01, 1.0207e+00,
         6.5819e-01],
        ...,
        [5.4743e-01, 5.8966e-01, 6.5864e-01,  ..., 0.0000e+00, 9.1360e-01,
         6.2270e-01],
        [9.8774e-01, 9.8153e-01, 1.0207e+00,  ..., 9.1360e-01, 0.0000e+00,
         3.6975e-01],
        [6.1875e-01, 6.1479e-01, 6.5819e-01,  ..., 6.2270e-01, 3.6975e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.5085995 , 12.71843462, 12.88270018, 12.62576026, 13.02818702,
       13.0493368 ,  8.09984231, 10.90638066,  7.20482224, 13.89264822,
       10.87409005,  6.49538841, 13.00380624,  5.5366698 ,  8.29640613,
       11.10066126,  7.85919687,  5.1224079 , 12.95702647,  8.05242591,
        8.27853076, 10.80728788, 13.12299145, 10.46238054,  5.14607014,
       11.42947632, 14.10796466,  5.92281142,  6.10360073,  7.58378655,
        7.70897675,  4.92497514,  6.19633139,  8.01009083,  7.50500064,
       10.78533838, 10.8992524 , 13.89974693,  8.09899874,  6.46560181,
        8.7273569 , 11.64436684,  9.54154096, 13.56367857,  9.70108409,
       13.04156075,  4.75303127,  9.7751189 , 11.22993218,  5.5671389 ,
        8.62631801, 10.41909211, 11.48663767,  8.0550286 ,  4.90505566,
        7.72291946, 12.86648771,  9.54154096,  5.29009836, 11.4088043 ,
       11.89855267, 10.51437947, 10.52587243,  7.33350642, 13.32613445,
        9.60359158, 11.48663767, 13.1471313 ,  7.822854  ,  7.98469166,
       10.71162681, 13.92390689, 13.96905756,  5.9510489 ,  8.25828409,
       10.65929968, 14.03505624,  7.28793663,  9.86089579, 12.25806559,
       12.79126785, 11.04953767, 10.35485331,  6.59176018, 11.97949273,
       11.00508681,  8.99276564,  7.03733775, 11.86239961, 13.20597932,
        4.90505566, 13.15525803,  6.30647746,  9.99211663, 10.9036226 ,
       10.35485331,  8.98765644,  7.72415048, 14.02526505, 10.8992524 ,
       12.88270018,  9.4997987 ,  8.56889738,  8.28968067,  5.70405525,
        6.39520397])}
{'P': tensor([[4.2405e-03, 5.4761e-01, 1.2359e-01,  ..., 6.5754e-05, 1.0342e-04,
         3.1315e-07],
        [2.7984e-05, 7.6155e-03, 2.7826e-01,  ..., 1.6454e-04, 2.7955e-04,
         4.3177e-07],
        [1.2962e-05, 2.0982e-05, 8.4103e-03,  ..., 3.3323e-04, 1.1951e-03,
         2.5877e-07],
        ...,
        [2.9659e-06, 3.4441e-06, 5.3884e-06,  ..., 4.0016e-01, 3.8887e-03,
         6.5458e-07],
        [1.7418e-06, 2.0072e-06, 3.3926e-06,  ..., 1.0441e-03, 2.1200e-01,
         1.4308e-06],
        [2.6478e-03, 2.0390e-03, 1.1050e-03,  ..., 6.4969e-07, 3.4879e-06,
         1.8421e-02]]), 'X': tensor([[ 12.4901,  -0.7052,   1.7315],
        [ 12.5789,   0.1722,   1.6470],
        [ 12.4837,   2.7510,   0.4449],
        [ 12.3357,   3.6001,   0.7903],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.5770,   1.8512,  -0.6578],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.6050,   2.9451,  -0.7012],
        [ -8.0965,  -5.9712,   4.3466],
        [ 12.2692,  -1.8882,   3.9253],
        [  8.1043,  -8.5164,  -7.9290],
        [ -7.8849,  -6.3115,  -4.7479],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.5650,   2.0132,   4.6272],
        [ -7.4733,   4.7555,   6.2490],
        [  4.8432,   5.1968,   1.5006],
        [  1.2273,  -4.8081,   9.8129],
        [ 10.9162,  -5.2599,  -7.0812],
        [  0.8700,  -4.8132,  -7.8042],
        [  0.6396,  14.0772,  -6.3786],
        [ -0.6323,   7.9085,  -5.9544],
        [ -7.6525,  -6.6534,   5.1721],
        [ -1.5408,   7.8978,   5.3470],
        [ 11.4495,   6.3026,   6.8257],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.6928,  12.4710,   3.6001],
        [  5.6523,   3.9443,  -2.9449],
        [  9.4925,   9.3811,   8.9842],
        [  0.5850,  -4.7982,   9.6639],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -7.5720,   4.6310,   2.1717],
        [ -4.0409,   7.3169,  -3.5779],
        [  4.0362,  13.2397,  -7.0030],
        [  2.0547,   7.2957,  -1.9343],
        [ 11.7726,  -3.4876,  -6.7402],
        [  6.1066,  -0.7799,  -4.1746],
        [ -9.3604,   0.4855,  -3.0158],
        [  4.4915,   5.5990,   7.0405],
        [  6.0310,  12.2795,  -6.4528],
        [ -5.3382,   6.6533,   7.8054],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.3474,  -4.2162,   5.2455],
        [  6.0073,   3.0986,   6.7937],
        [  9.2811,  -7.4336,   5.1771],
        [  2.4149,  -4.6005,  -4.3339],
        [  4.2612, -10.5514,  -6.6268],
        [  0.1206,  -4.7363,   8.7103],
        [  3.2990,  -4.2415,  -7.4857],
        [  5.2199,   4.6861,  -1.9304],
        [ -2.3317,   7.8034,  -3.6274],
        [  4.4039,   5.6903,   8.7366],
        [ -3.5173,   7.5091,  -8.9585],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.1886,  -2.6250,   6.4677],
        [  9.9217,  -6.7006,  -5.6728],
        [  3.4604,  -4.1545,   1.8091],
        [ -2.1533, -10.5778,   1.0112],
        [ -6.4368,   5.8356,   3.3014],
        [  6.2770,  -0.0379,   2.4195],
        [ -4.3359,  -9.6431,  -9.0616],
        [  6.2883,   0.0325,   5.2692],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.9653,  -9.3225,  -3.4727],
        [  8.6420,  -8.0580,   6.5781],
        [  9.5261,  -7.1675,   1.4206],
        [  4.3554,   5.7393,  -9.0766],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.9505,  -4.7171,   9.9868],
        [  4.4290,  -3.4596,   1.4181],
        [  4.2275,  -3.6320,  -0.2714],
        [  2.6310, -10.9195,   3.8054],
        [  4.1140,  -3.7220,  -0.1273],
        [  6.2036,   2.4030,  -1.2500],
        [  6.8704,  -9.3809,   7.5769],
        [  2.9470,  13.6085,   2.9809],
        [  5.0190,   4.9705,  -7.9752],
        [  6.3246,   1.6696,  -0.1544],
        [ -5.2589,   6.7024,  -5.5809],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.2895,   0.0405,   5.7652],
        [ -9.4671,  -0.5553,   0.1117],
        [  3.6469,   6.3624,  -1.8414],
        [ -3.0008, -10.2827,  -8.9480],
        [  0.3231,  -4.7691,  -1.6120],
        [  4.1348,   5.9514,  -0.8856],
        [  8.0381,  -8.5690,  -8.9523],
        [ 11.6801,  -3.7196,  -0.9181],
        [  0.3715,  -4.7756,   0.2312],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -0.3432,   7.8900,   7.1805],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -0.5385,   7.9037,   6.3830],
        [  6.1685,  12.1979,  -6.2473],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.7784,  -1.6401,   9.5587],
        [ -8.8154,   2.4611,   2.5198],
        [  6.1752,   2.5242,   3.5874],
        [ -1.3178, -10.7939,   0.4236],
        [ 10.9599,  -5.1849,  -3.5526],
        [  1.6408,   7.4591,  -4.2294],
        [  0.6757, -11.0361,  -1.3047],
        [  5.0190,   4.9705,  -7.9752],
        [  2.7590,  13.6621,   9.4496],
        [  2.0937,  13.8293,  -2.3533],
        [  5.4426, -10.1181,  -5.1268]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0349, 0.1310,  ..., 0.6924, 0.3552, 0.6752],
        [0.0349, 0.0000, 0.1044,  ..., 0.6866, 0.3392, 0.6945],
        [0.1310, 0.1044, 0.0000,  ..., 0.7553, 0.2428, 0.6815],
        ...,
        [0.6924, 0.6866, 0.7553,  ..., 0.0000, 0.9884, 1.3442],
        [0.3552, 0.3392, 0.2428,  ..., 0.9884, 0.0000, 0.5987],
        [0.6752, 0.6945, 0.6815,  ..., 1.3442, 0.5987, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.50996828, 12.58006038, 12.78327037, 12.85032836, 12.87752052,
       12.7125079 , 10.1979669 ,  9.09502385, 10.06023665, 12.41366754,
       11.75618513, 10.09979747, 12.64235359, 12.72524615,  8.8580671 ,
        7.10378919,  4.96231532, 12.11735334,  4.89121805, 14.09176259,
        7.9337647 , 10.14044489,  8.04664886, 13.06957425,  5.90035324,
       13.70893492,  6.89245352, 13.34587027,  4.83371341, 12.1460931 ,
        8.87587457,  8.35856487, 13.84126197,  7.57946459, 12.27835996,
        6.1561646 ,  9.37296102,  7.17789666, 13.68063098,  8.53014915,
       10.07238463,  6.10774544,  5.38342077,  6.7593917 , 11.89105259,
        5.19577146, 11.37939695,  4.73784222,  5.37343173,  7.01475376,
        8.14433542,  7.19534146,  8.29203356, 12.0495521 ,  5.81481044,
       11.97238624,  5.40689332, 10.794755  ,  8.68832398,  6.27715105,
       10.57302546,  6.28835101, 12.27991991, 11.63724792, 11.8159254 ,
       11.92133129,  7.20482224, 13.04217487,  5.10446653,  5.62005629,
        5.57341623, 11.23201619,  5.54776934,  6.652742  , 11.62770332,
       13.92390689,  7.06372172,  6.54127431,  8.51928531,  5.34250225,
        6.28961988,  9.48337063,  7.33350642, 10.71162681,  4.78002665,
        7.24677664, 11.74901778, 12.25806559,  4.79003361, 12.10942412,
        7.89745107, 12.52372629,  7.92200342, 13.66896739,  5.84533345,
        6.00663314,  9.15253344,  6.67121748, 10.87409005, 12.12448912,
        7.6374603 , 11.05672136,  7.06372172, 13.93790235, 13.98691426,
       11.4890861 ])}
{'P': tensor([[4.0899e-03, 3.5208e-01, 2.6499e-01,  ..., 3.8483e-04, 7.9649e-05,
         2.5681e-04],
        [1.2915e-05, 5.6534e-03, 2.2599e-01,  ..., 4.3689e-04, 2.7185e-04,
         6.4078e-04],
        [1.3849e-05, 1.0280e-04, 6.8840e-03,  ..., 4.2124e-04, 4.0969e-04,
         7.3883e-04],
        ...,
        [2.2136e-04, 4.0756e-04, 4.3571e-04,  ..., 9.1839e-02, 3.4391e-03,
         2.1604e-02],
        [2.2718e-06, 2.9893e-06, 3.2189e-06,  ..., 8.0657e-05, 3.1772e-01,
         9.4333e-05],
        [4.5958e-07, 8.1499e-07, 9.1037e-07,  ..., 5.1497e-05, 5.4011e-03,
         1.9162e-02]]), 'X': tensor([[ 12.5123,  -0.5339,  -1.0728],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.5770,   1.8512,  -0.6578],
        [ 12.4837,   2.7510,   0.4449],
        [ 12.3357,   3.6001,   0.7903],
        [ 11.9122,   5.1397,  -1.5498],
        [  6.3303,   1.6145,  -8.4083],
        [  1.0093,   7.6567,   2.3526],
        [ -8.0750,  -6.0072,  -5.8267],
        [  4.8432,   5.1968,   1.5006],
        [ 12.5650,   2.0132,   4.6272],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.5935,   5.9740,  -4.3952],
        [ -4.7389,   6.9953,   3.7638],
        [  7.0642,  11.6104,  -9.8316],
        [  5.6523,   3.9443,  -2.9449],
        [  6.1929,   2.4501,  -6.9352],
        [ 11.2847,  -4.5848,  -4.7503],
        [  5.1667,  -2.6543,   1.2879],
        [  2.3373,  13.7721,  -0.2442],
        [  2.6970,   6.9826,   9.7350],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.5375,   7.0676,  -4.3490],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.3858,   4.4260,  -9.6769],
        [ -3.5520,   7.4976,   9.1522],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.9653,  -9.3225,  -3.4727],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.1644, -11.0394,   7.8511],
        [  9.5203,   9.3483,   5.8781],
        [ -9.4756,  -0.8166,  -9.6021],
        [  1.6232,  -4.7717,  -2.5655],
        [  0.4996,   7.7738,  -4.4711],
        [  5.2199,   4.6861,  -1.9304],
        [  4.6357,   5.4415,   7.9067],
        [  3.8111, -10.6787,  -4.6539],
        [ -8.3539,  -5.5111,   0.3373],
        [ -8.9788,   2.0208,  -3.2867],
        [ -4.0409,   7.3169,  -3.5779],
        [  9.0652,   9.8597,  -3.5949],
        [  6.3460,   1.4252,   1.2291],
        [  3.7237,  -3.9966,  -5.0376],
        [  1.7815,  -4.7481,   4.4289],
        [ 11.5568,   6.0602,   9.3244],
        [ 12.4592,  -0.9168,  -6.7571],
        [  2.5663,   7.0526,  -6.8769],
        [  5.6414, -10.0303,   7.8525],
        [  2.4639, -10.9431,  -7.3718],
        [  1.3403,  -4.8011,  -1.1698],
        [  9.3560,   9.5391,   9.4943],
        [  3.0719, -10.8449,  -4.1673],
        [  4.5430,   5.5438,  -3.8669],
        [  6.9653,  -9.3225,  -3.4727],
        [  6.1984,  -0.4327,   1.5438],
        [ -9.2974,   0.8371,   2.4852],
        [  5.9193,  -1.3176,   5.1739],
        [  6.0031,  12.2958,  -6.5992],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.4671,  -0.5553,   0.1117],
        [  8.5716,  10.3596,   6.8818],
        [ -1.9487,   7.8590,  -3.3330],
        [  4.1999,  -3.6543,  -4.1998],
        [  1.7815,  -4.7481,   4.4289],
        [  5.7485,  -9.9811,  -3.2355],
        [  8.5716,  10.3596,   6.8818],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -7.5720,   4.6310,   2.1717],
        [  8.7131,  10.2216,   7.7915],
        [  1.9241,   7.3503,   7.9004],
        [  5.8275,  -1.5334,   2.0124],
        [  3.7121,   6.3117,   0.6738],
        [  6.3505,   0.6235,  -8.5849],
        [  0.2480, -11.0153,  -3.0183],
        [ -1.6086,   7.8928,   4.3385],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.2076,  13.8032,  -1.5099],
        [  4.4915,   5.5990,   7.0405],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -8.4297,  -5.3637,  -9.3962],
        [  0.6396,  14.0772,  -6.3786],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.8540,  -4.3971,  -5.2144],
        [  3.7992,  -3.9474,  -8.9019],
        [  9.1409,  -7.5789,   1.5814],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.6019,   5.9540,  -5.1715],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.2508,   1.0572,   2.9978],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 12.2435,   4.0076,   8.1257],
        [ -7.8126,   4.3066,   5.1791],
        [ -6.8618,  -7.6322,   3.7027],
        [ 11.9122,   5.1397,  -1.5498],
        [  0.5600, -11.0321,  -0.8660],
        [  0.9392,  -4.8143,  -3.4649],
        [  3.2159, -10.8166,  -1.0406],
        [  6.2026,  12.1774,  -1.5887],
        [  6.2675,  12.1379,  -8.9720],
        [  5.3858,   4.4260,  -9.6769],
        [ -0.0282,   7.8575,  -1.1746],
        [  5.9655,   3.2172,  -7.5953],
        [  0.7275,  14.0667,  -0.9815],
        [  7.0642,  11.6104,  -9.8316]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.6512e-02, 7.5982e-02,  ..., 3.0389e-01, 3.5231e-02,
         5.5092e-01],
        [5.6512e-02, 1.5170e-04, 3.3650e-02,  ..., 3.2235e-01, 8.5805e-02,
         5.3440e-01],
        [7.5982e-02, 3.3650e-02, 0.0000e+00,  ..., 3.5599e-01, 9.5495e-02,
         5.6181e-01],
        ...,
        [3.0389e-01, 3.2235e-01, 3.5599e-01,  ..., 0.0000e+00, 3.2223e-01,
         3.4357e-01],
        [3.5231e-02, 8.5805e-02, 9.5495e-02,  ..., 3.2223e-01, 0.0000e+00,
         5.8239e-01],
        [5.5092e-01, 5.3440e-01, 5.6181e-01,  ..., 3.4357e-01, 5.8239e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.52372629, 12.67680439, 12.7125079 , 12.78327037, 12.85032836,
       12.97370371,  6.53290663,  7.72291946, 10.06437598,  7.10378919,
       12.72524615,  6.01356004, 13.04217487,  8.44940661, 13.59056613,
        6.89245352,  6.65991877, 12.18045978,  5.80860165, 13.96905756,
        7.48538905, 10.24557047,  7.5092834 ,  8.8353674 ,  6.97106912,
        8.29640613, 10.67132524, 11.63724792, 10.1979669 , 11.10066126,
       13.3426541 ,  9.51074763,  5.04028538,  7.78980299,  7.01475376,
        7.14837514, 11.33837396, 10.00794468,  9.20340031,  8.35856487,
       13.39372426,  6.50409372,  5.46244988,  5.0713364 , 13.0493368 ,
       12.49291768,  7.50500064, 11.50791859, 11.21703654,  4.98461983,
       13.3614598 , 11.27160035,  7.16747878, 11.63724792,  6.2134961 ,
        9.3349858 ,  6.06416957, 13.6829843 , 13.20597932,  9.48337063,
       13.44593519,  8.09704052,  5.5671389 ,  5.0713364 , 11.51811603,
       13.44593519,  9.86089579,  9.50654967,  8.87587457, 13.43127131,
        7.59795445,  6.02588517,  7.32233733,  6.38105015, 11.018084  ,
        8.0550286 , 10.27716434, 13.97857591,  7.17789666,  9.47369628,
       10.60226614,  9.99145096, 14.09176259,  5.5513072 ,  9.88571713,
        5.47866221, 11.87412439,  5.60064319, 13.04051744,  9.15976746,
        9.3109884 , 10.32715715, 12.88270018,  8.92098752, 10.26327895,
       12.97370371, 11.04629471,  4.90505566, 11.2845608 , 13.6660671 ,
       13.66051884,  6.97106912,  7.8575648 ,  6.77777606, 14.08549802,
       13.59056613])}
{'P': tensor([[1.1447e-02, 2.8308e-01, 2.8201e-01,  ..., 4.6278e-04, 2.7661e-06,
         2.6371e-06],
        [1.3843e-04, 1.5533e-01, 1.5792e-03,  ..., 9.0951e-04, 2.6886e-05,
         9.5287e-05],
        [8.6693e-05, 1.6367e-01, 2.5482e-02,  ..., 3.1934e-04, 3.2796e-06,
         9.3982e-06],
        ...,
        [4.1701e-03, 8.7757e-03, 6.6515e-03,  ..., 3.8105e-02, 9.6596e-05,
         8.5511e-05],
        [5.6713e-04, 3.4412e-04, 2.3190e-04,  ..., 2.0660e-03, 1.0413e-01,
         6.4547e-05],
        [1.6185e-06, 9.3175e-07, 1.1002e-06,  ..., 3.1428e-05, 1.5060e-04,
         2.5015e-02]]), 'X': tensor([[ 12.5199,  -0.4711,   3.0436],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.5650,   2.0132,   4.6272],
        [ 12.3357,   3.6001,   0.7903],
        [ 12.4837,   2.7510,   0.4449],
        [ 12.2914,   3.8037,   6.0978],
        [  4.9957,  -2.8705,   6.5978],
        [  9.2979,   9.6048,   7.3138],
        [  2.4697,  -4.5836,   1.4593],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.4645,  -1.4880,  -2.0226],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -5.0943,  -9.1682,   0.5073],
        [  5.7963,  -1.6018,   1.8856],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.6523,   3.9443,  -2.9449],
        [  4.6231,   5.4556,   8.4973],
        [ -6.8618,  -7.6322,   3.7027],
        [ 11.9641,  -2.9568,   7.7211],
        [ -7.7842,   4.3466,  -1.7548],
        [  1.1006,   7.6318,  -2.4877],
        [  9.3836,   9.5075,  -5.9278],
        [  5.8930,  12.3592,  -8.9201],
        [  7.2340,  -9.1505,  -5.8321],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.6408,   7.4591,  -4.2294],
        [ -6.2200,  -8.2737,   5.5079],
        [  6.3606,   0.9204,   3.2525],
        [ -4.4503,   7.1378,   3.3751],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.1120,  -4.6818,   8.6144],
        [  3.7237,  -3.9966,  -5.0376],
        [ 10.9394,   7.3069,  -8.9365],
        [ -6.9838,  -7.4965,   0.9796],
        [ 10.9162,  -5.2599,  -7.0812],
        [  9.1409,  -7.5789,   1.5814],
        [ -2.0445,   7.8469,  -0.5857],
        [ -4.3963,   7.1629,   4.5263],
        [  6.3606,   0.9204,   3.2525],
        [ -4.3359,  -9.6431,  -9.0616],
        [  2.1120,  -4.6818,   8.6144],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -3.7964,  -9.9294,   7.7147],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.4697,  -4.5836,   1.4593],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.3376,  -4.6231,   9.8436],
        [ -6.6895,   5.6051,   2.7899],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.2022,   2.4092,   3.0349],
        [ -8.7735,   2.5640,  -6.3313],
        [ -8.8363,   2.4086,  -5.6296],
        [ 12.4901,  -0.7052,   1.7315],
        [ 11.6801,  -3.7196,  -0.9181],
        [  5.6258,  -1.9404,  -6.0715],
        [ -9.2687,  -2.9794,   0.6436],
        [  7.1821,  -9.1845,   9.8548],
        [  5.1203,  -2.7151,   6.2110],
        [  4.3459,  -3.5327,   0.3667],
        [ -9.1706,   1.3863,   8.4032],
        [  4.1999,  -3.6543,  -4.1998],
        [  3.8111, -10.6787,  -4.6539],
        [  4.8432,   5.1968,   1.5006],
        [  1.7903,  -4.7467,   0.1168],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.0937,  13.8293,  -2.3533],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 12.3025,  -1.7440,  -6.6632],
        [  4.4039,   5.6903,   8.7366],
        [  5.8617,   3.4853,  -9.8080],
        [  2.3200,   7.1756,   1.1938],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -9.4296,  -1.9458,   4.0833],
        [  2.6970,   6.9826,   9.7350],
        [  6.2790,  -0.0259,  -5.4116],
        [  3.7237,  -3.9966,  -5.0376],
        [ -9.2707,   0.9661,   4.4863],
        [  6.2804,   2.0019,   0.8607],
        [  0.9945,  -4.8144,  -0.1897],
        [ -9.3616,  -2.4750,   3.7765],
        [  0.2480, -11.0153,  -3.0183],
        [  1.7500,   7.4187,   4.4237],
        [ -7.3554,   4.8985,  -0.6665],
        [ -2.2188,   7.8218,   8.6064],
        [  2.1504,  -4.6726,   8.7908],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.6414, -10.0303,   7.8525],
        [  6.2036,   2.4030,  -1.2500],
        [ -6.7643,  -7.7372,  -9.6663],
        [  1.3767, -11.0341,   9.9415],
        [  0.3038,   7.8092,   0.4164],
        [  5.8889,  -1.3916,   8.8993],
        [  6.2975,   0.0947,   5.1791],
        [  5.8902,  -1.3887,   4.0388],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.1737,   2.5305,   3.9049],
        [ -6.0844,  -8.3955,   0.7705],
        [  6.0890,  -0.8382,   2.9166],
        [  0.3493,  -4.7727,  -6.5682],
        [ -8.6229,   2.9068,   2.4429]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1260, 0.2286,  ..., 0.0925, 0.6371, 0.8213],
        [0.1260, 0.0000, 0.2979,  ..., 0.1673, 0.6002, 0.9143],
        [0.2286, 0.2979, 0.0000,  ..., 0.3123, 0.8656, 0.9389],
        ...,
        [0.0925, 0.1673, 0.3123,  ..., 0.0000, 0.5601, 0.7483],
        [0.6371, 0.6002, 0.8656,  ..., 0.5601, 0.0000, 0.8219],
        [0.8213, 0.9143, 0.9389,  ..., 0.7483, 0.8219, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.52875793, 12.87752052, 12.72524615, 12.85032836, 12.78327037,
       12.86648771,  5.76165934, 13.36800073,  5.20659892,  9.3349858 ,
        9.58072286, 12.49317563, 10.48840476,  6.01356004,  8.08320235,
       12.32408516,  6.89245352,  7.15100303, 10.26327895, 12.32408516,
        8.91551231,  7.71075226, 13.35832725, 13.6922425 , 11.66453059,
       13.04217487,  7.6374603 , 10.35093485,  6.42689705,  8.41148722,
        9.85384623,  5.13616224,  5.46244988, 13.15525803, 10.24557047,
       12.11735334, 11.87412439,  8.10886021,  8.40445872,  6.42689705,
       10.57302546,  5.13616224, 12.27835996, 10.63038565,  5.34250225,
        5.20659892,  5.05774295,  5.1805345 ,  8.7273569 ,  9.96152788,
       12.13568598,  6.65368551,  9.14044984,  9.15866607, 12.50996828,
       12.25806559,  5.9510489 ,  9.73579738, 11.65923396,  5.79562691,
        5.60064319,  9.27474386,  5.5671389 , 11.33837396,  7.10378919,
        5.07306274, 13.99048094,  9.93553424, 13.98691426,  9.20340031,
       13.99048094, 10.00410625, 12.42554808,  7.19534146,  6.81961996,
        7.54127425, 10.52357479,  9.97822473,  9.6282763 ,  7.48538905,
        6.27905292,  5.46244988,  9.32094055,  6.59176018,  4.9160962 ,
        9.68324553, 11.018084  ,  7.62232307,  8.83725081,  8.1303945 ,
        5.14371005, 12.42554808, 11.50791859,  6.652742  , 10.27716434,
       11.11969815,  7.81509631,  6.05113217,  6.29821673,  6.05165248,
        5.47866221,  6.67218687, 10.36844813,  6.1463896 ,  4.78544875,
        9.09963894])}
{'P': tensor([[7.7137e-03, 5.1942e-01, 1.8340e-01,  ..., 9.3897e-07, 4.8850e-06,
         9.5132e-05],
        [1.8581e-05, 1.0021e-02, 2.3964e-01,  ..., 5.6520e-07, 4.3070e-06,
         1.4733e-05],
        [6.1234e-06, 3.5844e-05, 6.1034e-03,  ..., 1.9219e-07, 2.2736e-06,
         2.9755e-06],
        ...,
        [5.0388e-03, 2.1974e-03, 7.4746e-04,  ..., 1.1605e-02, 3.8571e-05,
         3.1325e-03],
        [9.3890e-03, 5.3882e-03, 4.2166e-03,  ..., 6.1843e-03, 2.9363e-01,
         1.7111e-02],
        [1.4621e-01, 2.2628e-01, 2.5503e-01,  ..., 4.4075e-06, 2.0033e-05,
         2.0787e-02]]), 'X': tensor([[ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.2322e-02, 1.7806e-01,  ..., 6.9187e-01, 1.6222e-01,
         2.0573e-01],
        [6.2322e-02, 0.0000e+00, 1.4515e-01,  ..., 7.3455e-01, 2.2407e-01,
         2.3720e-01],
        [1.7806e-01, 1.4515e-01, 0.0000e+00,  ..., 8.6993e-01, 2.9858e-01,
         2.0459e-01],
        ...,
        [6.9187e-01, 7.3455e-01, 8.6993e-01,  ..., 0.0000e+00, 6.1838e-01,
         8.0235e-01],
        [1.6222e-01, 2.2407e-01, 2.9858e-01,  ..., 6.1838e-01, 0.0000e+00,
         1.8417e-01],
        [2.0573e-01, 2.3720e-01, 2.0459e-01,  ..., 8.0235e-01, 1.8417e-01,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.53047489, 12.72524615, 12.86648771, 13.04156075, 13.0743399 ,
       12.87752052, 10.69963954,  7.6374603 ,  7.50500064, 10.84493732,
        8.43626114,  5.10444097, 10.1979669 ,  5.5366698 , 13.99048094,
        8.94395628, 10.14436912, 10.31584102, 11.85890664,  6.81084724,
        9.43737523,  5.36061974,  5.47866221, 11.05317597, 10.7506423 ,
        7.09887557, 14.09357547,  6.3949043 ,  8.29203356, 11.73397309,
       13.18278606, 11.83075005,  7.01475376,  6.67218687,  9.17016857,
        9.53015376, 10.1979669 , 10.71162681,  8.1137452 ,  6.74686071,
        8.34949383,  5.21280156,  6.48486171, 12.35421461, 13.44593519,
       13.02818702, 10.67132524,  9.20340031,  9.91668182,  9.77588801,
       11.4088043 , 13.59673406,  7.26074925, 12.50996828,  8.8353674 ,
        9.58072286, 10.08001894, 12.49317563, 12.25806559,  6.46560181,
        7.60529892,  8.4486771 , 13.744387  ,  9.78147542, 13.00380624,
       10.31613269,  7.6374603 , 13.26209545, 10.36844813, 13.40024224,
       11.00812996, 11.59512194,  9.70108409,  7.71075226, 10.39942497,
        6.39520397, 11.97238624,  5.27178363,  7.03733775,  8.83341608,
        9.61942445, 13.04051744, 11.8159254 ,  6.58184474,  6.53250937,
        4.75303127, 11.46176259,  7.37576154, 10.09979747, 13.77604293,
        9.23178204,  8.62631801, 11.99633819, 12.49291768,  5.52416368,
       12.27835996,  5.39723241,  7.39599986, 11.00812996,  7.60463349,
        7.50500064, 13.41713661,  7.16747878, 11.63502679,  4.89869205,
       12.38881635])}
{'P': tensor([[1.8636e-03, 2.6780e-01, 2.3216e-01,  ..., 2.4354e-08, 1.0671e-04,
         1.1991e-05],
        [1.5422e-05, 3.5554e-03, 1.8125e-01,  ..., 4.8596e-08, 2.5106e-04,
         1.3246e-05],
        [2.0754e-05, 7.6274e-05, 5.3117e-03,  ..., 7.0903e-08, 3.7420e-04,
         1.6706e-05],
        ...,
        [3.1230e-06, 2.7558e-06, 2.4426e-06,  ..., 3.5952e-02, 6.6290e-07,
         1.1486e-05],
        [4.0812e-07, 4.8027e-07, 5.0993e-07,  ..., 5.6969e-06, 1.7723e-02,
         2.8127e-07],
        [5.3330e-02, 7.0223e-02, 6.1022e-02,  ..., 1.8215e-07, 5.1779e-04,
         9.5830e-03]]), 'X': tensor([[ 12.5258,  -0.4200,  -9.3493],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.2218,   4.0953,  -8.2072],
        [ -5.9987,   6.1949,  -6.2114],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.3301,   0.3638,  -7.5685],
        [ -9.2301,  -3.1537,  -2.7477],
        [  3.3474,  -4.2162,   5.2455],
        [  9.9217,  -6.7006,  -5.6728],
        [  7.0358,  -9.2784,  -9.8479],
        [  2.9410,   6.8428,   5.4458],
        [ 12.6059,   0.9597,  -6.3264],
        [  0.3038,   7.8092,   0.4164],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.9843,  -9.8679,  -6.1449],
        [ -2.1533, -10.5778,   1.0112],
        [ -5.1398,  -9.1367,   6.2074],
        [  4.3523, -10.5232,   8.2138],
        [  1.5662,  13.9375,   4.5142],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.7558,  -4.4844,   0.0636],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3569,   0.7591,   6.5011],
        [ 11.3691,   6.4760,   1.0787],
        [  2.2644,  -4.6434,   9.7508],
        [  1.1350,  14.0104,   0.6321],
        [ -5.1974,   6.7396,   3.3484],
        [ -7.4733,   4.7555,   6.2490],
        [ -0.8455, -10.8853,  -6.7400],
        [ 11.7794,   5.5086,  -9.6929],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -3.0644,   7.6439,  -1.4322],
        [ -9.4697,  -0.6157,  -6.8472],
        [  0.1532,  -4.7422,  -7.0145],
        [  3.7992,  -3.9474,  -8.9019],
        [ -9.2707,   0.9661,   4.4863],
        [  2.3261,  13.7749,   5.9121],
        [ -5.0550,   6.8231,  -9.2502],
        [ -1.9600,   7.8577,   6.9121],
        [  5.8318,   3.5567,  -4.7692],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.8363,   2.4086,  -5.6296],
        [  3.6703,   6.3443,  -5.4401],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.4366,  -0.1189,  -2.6736],
        [  2.3367,   7.1676,  -9.5263],
        [  3.1064,  13.5607,   6.7605],
        [  9.6614,   9.1784,  -1.2087],
        [  0.3724,   7.7974,  -6.8722],
        [ 10.0400,   8.6922,  -6.2847],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3589,   0.8239,  -8.4230],
        [  5.3804,  -2.3504,   7.2744],
        [ -1.2454,   7.9127,  -0.5715],
        [  2.5813,  -4.5471,   5.3597],
        [ -2.6735, -10.4062,  -1.5911],
        [ 11.8951,  -3.1570,   7.0742],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.4996,   2.6335,  -6.3166],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -7.3470,  -7.0615,   5.0066],
        [  8.1043,  -8.5164,  -7.9290],
        [ -9.4753,  -0.8012,  -5.2050],
        [  4.0078,  -3.8018,  -6.2168],
        [ -3.9508,  -9.8515,   1.7923],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.3606,   0.9204,   3.2525],
        [ -1.9600,   7.8577,   6.9121],
        [  3.1892, -10.8220,  -6.8409],
        [  6.3505,   0.6235,  -8.5849],
        [  9.5261,  -7.1675,   1.4206],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -4.7334,   6.9982,  -0.4392],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 10.0567,   8.6697,   4.6329],
        [ 11.7795,  -3.4697,   3.5624],
        [ 12.5927,   0.4199,  -9.4997],
        [ -4.1296,   7.2803,  -2.7194],
        [  2.2076,  13.8032,  -1.5099],
        [  4.9452, -10.3187,  -7.8621],
        [ -5.1369,   6.7755,   8.3836],
        [  4.0587, -10.6112,   6.2273],
        [ -8.9033,   2.2329,  -8.8912],
        [  2.3200,   7.1756,   1.1938],
        [  2.1625,  -4.6697,   7.0658],
        [  2.0449,  13.8402,  -7.1642],
        [ 11.2939,   6.6325,  -5.8311],
        [ -5.6792,  -8.7352,   4.0318],
        [  9.4554,   9.4245,   9.8800],
        [  3.7916,  -3.9524,  -7.9547],
        [  4.5325,  -3.3643,   6.7699],
        [ -8.4887,  -5.2448,  -0.6585],
        [  2.8474,   6.8979,   6.6457],
        [  8.1528,  10.7452,   9.1493],
        [ -5.3382,   6.6533,   7.8054],
        [  5.5101,  -2.1433,   9.5442],
        [  4.2096,   5.8815,   1.9230],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.9284,  11.7061,  -0.3732],
        [ 11.9695,  -2.9407,  -6.3498]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0194, 0.0311,  ..., 1.3632, 0.5448, 0.1768],
        [0.0194, 0.0000, 0.0122,  ..., 1.3780, 0.5402, 0.1917],
        [0.0311, 0.0122, 0.0000,  ..., 1.3896, 0.5422, 0.2034],
        ...,
        [1.3632, 1.3780, 1.3896,  ..., 0.0000, 1.2266, 1.1865],
        [0.5448, 0.5402, 0.5422,  ..., 1.2266, 0.0000, 0.4902],
        [0.1768, 0.1917, 0.2034,  ..., 1.1865, 0.4902, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.53285633, 12.5769659 , 12.59969936, 12.76408562, 12.84056604,
       12.889695  ,  8.62329466, 12.11735334,  6.34059021,  9.75401449,
        5.38342077, 11.97238624, 11.64436684,  7.44805901, 12.64235359,
        7.81509631, 10.51340829, 11.54071859, 10.794755  , 10.48315436,
       11.38774149, 14.02526505, 12.19805394,  5.26342203,  7.60529892,
        6.4020404 , 13.08414689,  5.16611782, 14.05633441,  8.51088913,
        8.8580671 , 10.91805683, 13.00380624,  9.5642993 ,  8.23525433,
        9.48969978,  4.74468406,  5.47866221,  9.32094055, 13.96987806,
        8.49161154,  8.09842704,  6.83083863, 13.50484539,  9.15866607,
        7.32950385, 13.32645552,  9.43737523,  7.53884104, 13.91198191,
       13.32613445,  7.80625981, 13.27994304,  7.96181288,  6.41202986,
        5.87132718,  8.01009083,  5.2287058 , 10.74409834, 12.30695098,
       10.8992524 , 12.77402351, 10.0674887 , 10.19036081, 11.75618513,
        9.50913849,  5.52416368, 10.61417181, 10.67132524,  6.42689705,
        8.09842704, 11.28215429,  6.38105015, 11.92133129, 11.70016217,
        9.7712321 ,  8.4486771 , 10.00794468, 12.4013153 , 13.2778378 ,
       12.27991991, 12.59969936,  8.36995953, 13.97857591, 11.4424816 ,
        8.50268322, 11.36089355,  9.17905491,  7.54127425,  5.14607014,
       13.99048094, 13.09737053, 10.41909211, 13.35013374,  5.47702766,
        5.64464807,  9.97822473,  7.46248947, 13.48809595,  8.53014915,
        5.91222346,  7.23276582,  9.88310614,  9.85384623, 13.60275211,
       12.32545637])}
{'P': tensor([[4.3656e-03, 2.1778e-01, 1.6933e-01,  ..., 2.4343e-08, 1.2187e-03,
         1.3935e-07],
        [1.8071e-05, 1.1059e-02, 5.8322e-04,  ..., 3.0179e-08, 5.0925e-03,
         2.0335e-07],
        [3.1531e-05, 3.7348e-02, 9.5253e-03,  ..., 4.3793e-08, 3.6453e-03,
         3.2108e-07],
        ...,
        [1.9836e-05, 8.0802e-06, 1.8511e-05,  ..., 2.2683e-02, 2.2495e-06,
         2.7134e-03],
        [3.1294e-05, 6.4445e-05, 5.1868e-05,  ..., 5.9689e-07, 4.7947e-02,
         2.7910e-06],
        [2.7614e-05, 1.6830e-05, 3.5310e-05,  ..., 3.5459e-03, 1.7003e-05,
         9.0030e-02]]), 'X': tensor([[ 12.5267,  -0.4125,  -8.7331],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.5832,   1.7543,  -7.6134],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.5927,   0.4199,  -9.4997],
        [ -0.2586,   7.8825,  -9.9840],
        [  3.3633,  13.4794,   0.5193],
        [ -3.2053, -10.1993,  -3.8736],
        [  4.3523, -10.5232,   8.2138],
        [  4.1297,  -3.7098,  -3.9838],
        [ -8.1204,  -5.9308,   4.0761],
        [ 11.0891,  -4.9558,  -2.3178],
        [  2.6310, -10.9195,   3.8054],
        [  3.9085,   6.1512,  -2.2560],
        [  3.9022,  13.2906,   5.5788],
        [  2.3367,   7.1676,  -9.5263],
        [  5.0233,   4.9647,  -3.1788],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.1579,  -0.5961,  -6.5595],
        [  2.1120,  -4.6818,   8.6144],
        [  2.9410,   6.8428,   5.4458],
        [ -8.8934,   2.2597,   3.5831],
        [  7.0358,  -9.2784,  -9.8479],
        [ 12.3609,   3.4769,  -8.5762],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.1203,  -2.7151,   6.2110],
        [  3.9022,  13.2906,   5.5788],
        [  2.1379,  -4.6757,   6.4981],
        [  6.1685,  12.1979,  -6.2473],
        [  2.3376,  -4.6231,   9.8436],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.0617,  12.7957,   8.3544],
        [  6.2576,   2.1359,  -3.3398],
        [ -4.4520,   7.1370,  -8.7867],
        [ -0.8675, -10.8815,   6.8561],
        [  7.9816,  10.8938,  -0.5378],
        [  4.5814, -10.4485,   2.5496],
        [  4.7739,  -3.1218,  -4.1188],
        [  7.9403,  -8.6455,  -8.7433],
        [  2.3261,   7.1726,   6.2522],
        [  3.7237,  -3.9966,  -5.0376],
        [ -8.9033,   2.2329,  -8.8912],
        [  6.2975,   0.0947,   5.1791],
        [ -8.7409,  -4.6837,  -6.5840],
        [  5.6314,  -1.9302,  -9.6543],
        [  7.0515,  11.6194,   8.4289],
        [ -8.8154,   2.4611,   2.5198],
        [ -7.6650,   4.5092,   0.2520],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.9036,   7.6840,   9.9237],
        [  4.4049,   5.6892,  -4.4394],
        [ -1.5451,   7.8975,   7.6104],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.3554,   5.7393,  -9.0766],
        [  5.9548,  -1.2273,  -2.6520],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 11.1420,   6.9331,  -2.4779],
        [  0.5960, -11.0335,   3.1144],
        [ -7.5603,   4.6459,   8.2002],
        [  1.5246,  -4.7838,  -5.2095],
        [ -7.4887,   4.7364,   8.9074],
        [ -1.8367,   7.8718,  -8.6523],
        [  6.0305,   3.0296,   0.1375],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.5568,   6.0602,   9.3244],
        [  6.2975,   0.0947,   5.1791],
        [  5.7784,  -1.6401,   9.5587],
        [ -2.1817,   7.8274,  -4.6298],
        [  1.8719,   7.3713,  -3.7465],
        [  3.3167,   6.6022,   6.3554],
        [  4.1637,  13.1898,  -1.9189],
        [  0.8940, -11.0402,  -0.9783],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3551,   0.7149,   1.3344],
        [  2.5813,  -4.5471,   5.3597],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -7.8359,   4.2735,   7.2943],
        [ -0.3432,   7.8900,   7.1805],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 11.5198,  -4.0923,   5.5088],
        [ -8.5859,  -5.0393,   0.5476],
        [  9.1620,  -7.5573,  -9.9882],
        [  5.9526,   3.2525,  -4.3821],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.2707,   0.9661,   4.4863],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.0140,  -1.0661,  -2.5040],
        [ -6.8618,  -7.6322,   3.7027],
        [ -9.2687,  -2.9794,   0.6436],
        [  3.2692,  13.5098,  -4.9032],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -2.0229, -10.6163,   8.6930],
        [ -7.6734,   4.4980,  -5.4248],
        [  0.9392,  -4.8143,  -3.4649],
        [ -6.2200,  -8.2737,   5.5079],
        [  9.3624,   9.5318,   0.2779],
        [  7.0171,  11.6439,  -5.0178],
        [ -8.8745,   2.3097,   3.5600],
        [ -4.6393,   7.0461,   1.1624],
        [  2.3367,   7.1676,  -9.5263],
        [  2.1440, -10.9811,   8.5281],
        [  4.3554,   5.7393,  -9.0766],
        [  1.6954,  -4.7616,   9.8198]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0774, 0.0311,  ..., 1.4177, 0.2938, 1.1357],
        [0.0774, 0.0000, 0.0711,  ..., 1.4854, 0.3712, 1.1864],
        [0.0311, 0.0711, 0.0000,  ..., 1.4145, 0.3089, 1.1218],
        ...,
        [1.4177, 1.4854, 1.4145,  ..., 0.0000, 1.1825, 0.5360],
        [0.2938, 0.3712, 0.3089,  ..., 1.1825, 0.0000, 0.9909],
        [1.1357, 1.1864, 1.1218,  ..., 0.5360, 0.9909, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.53345445, 12.76408562, 12.70489563, 12.84056604, 12.5769659 ,
       12.59969936,  7.88677538, 13.89264822, 10.6910768 , 11.38774149,
        5.5513072 , 10.05558893, 12.1460931 , 11.23201619,  7.28793663,
       13.85158417,  7.53884104,  7.06272256,  9.88571713,  6.18668463,
        5.13616224,  7.44805901,  9.17595899, 11.64436684, 12.84056604,
       10.30972193, 10.61047596,  5.79562691, 13.85158417,  5.14124612,
       13.66896739,  5.1805345 , 10.00802627,  9.62238828, 13.76047999,
        6.61211224,  8.41171369, 10.91601655, 13.50484539, 11.4088043 ,
        5.70405525, 11.73848065,  7.5403822 ,  5.46244988,  9.17905491,
        6.29821673,  9.91668182,  5.95297459, 13.59170809,  9.15253344,
        8.89301516, 10.46238054,  7.73692436,  7.19513806,  8.04718433,
       10.31584102,  7.20482224,  6.07992623,  9.62958973, 13.12299145,
       11.04953767,  8.8737556 ,  5.0209005 ,  8.86082124,  8.08320235,
        6.74872981,  8.04718433, 13.0493368 ,  6.29821673,  6.00663314,
        8.12580976,  7.60529892,  7.38844703, 13.83139024, 11.07637531,
        9.09963894,  6.39520397,  5.2287058 , 12.07818871,  8.92549752,
        7.89745107, 10.1979669 , 12.22503754,  9.95552626, 11.87665501,
        6.78325151,  9.24501829,  9.32094055,  5.84450127,  6.10774544,
       10.26327895,  9.73579738, 13.89974693,  9.4997987 , 10.80728788,
        8.89458273,  4.90505566, 10.35093485, 13.3607379 , 13.59480906,
        9.17016857,  8.43626114,  7.53884104, 11.18839169,  7.20482224,
        5.05444288])}
{'P': tensor([[3.4669e-03, 2.5275e-01, 1.7673e-01,  ..., 3.7174e-05, 7.9695e-03,
         5.6387e-06],
        [1.0566e-05, 5.6846e-03, 1.1957e-05,  ..., 4.2323e-05, 1.3699e-02,
         1.7283e-05],
        [5.9498e-05, 3.3775e-01, 4.1434e-03,  ..., 3.8506e-05, 8.7336e-03,
         6.7029e-06],
        ...,
        [4.9649e-06, 1.3136e-05, 5.2362e-06,  ..., 3.0528e-02, 5.8860e-04,
         3.5549e-02],
        [1.1588e-05, 4.6194e-05, 1.3541e-05,  ..., 3.4554e-04, 2.4063e-02,
         1.5424e-03],
        [1.0322e-06, 1.6745e-06, 1.0420e-06,  ..., 1.6072e-05, 1.1948e-05,
         3.7107e-01]]), 'X': tensor([[ 12.5763,   0.1333,  -9.2650],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.5927,   0.4199,  -9.4997],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.1588,   4.3364,  -8.0867],
        [  5.5626,  -2.0536,   8.4433],
        [  3.2692,  13.5098,  -4.9032],
        [ -8.4267,  -5.3697,   2.4172],
        [  9.9217,  -6.7006,  -5.6728],
        [ -9.4541,  -0.3327,   8.8246],
        [  6.3583,   1.1729,   8.7835],
        [ -3.5520,   7.4976,   9.1522],
        [  5.8902,  -1.3887,   4.0388],
        [  0.8700,  -4.8132,  -7.8042],
        [  3.3167,   6.6022,   6.3554],
        [ 11.1469,   6.9237,   1.6160],
        [  6.0047,   3.1062,  -6.7914],
        [  5.9003,  12.3550,   2.1917],
        [  2.6970,   6.9826,   9.7350],
        [ 12.3759,  -1.3943,   6.4659],
        [  5.7485,  -9.9811,  -3.2355],
        [ 11.1469,   6.9237,   1.6160],
        [  0.1677,  -4.7447,  -6.1529],
        [ -0.4341, -10.9473,   7.4509],
        [  2.4697,  -4.5836,   1.4593],
        [ 10.7566,   7.6212,  -2.3279],
        [ -9.4117,   0.1160,   8.2574],
        [  9.6662,   9.1725,   8.1056],
        [  1.1644, -11.0394,   7.8511],
        [  5.5240,  -2.1198,   6.2055],
        [  2.8994,   6.8675,  -9.5987],
        [ 12.1504,   4.3674,  -7.2998],
        [  2.5318,  13.7231,   6.3692],
        [ -6.4464,   5.8272,   7.5038],
        [ -1.8367,   7.8718,  -8.6523],
        [  5.3118,   4.5452,   4.2114],
        [  8.8091,  -7.9037,   0.3631],
        [ -3.3999, -10.1152,  -1.2088],
        [  4.4290,  -3.4596,   1.4181],
        [ -9.2707,   0.9661,   4.4863],
        [  5.0233,   4.9647,  -3.1788],
        [ -8.6050,   2.9451,  -0.7012],
        [ 11.2767,  -4.6006,   0.7868],
        [ 12.5258,  -0.4200,  -9.3493],
        [  0.7685,  -4.8098,  -2.9310],
        [  3.9022,  13.2906,   5.5788],
        [  9.1820,   9.7334,  -2.7932],
        [  5.4429,  -2.2529,   9.7306],
        [  5.2199,   4.6861,  -1.9304],
        [ -8.5607,  -5.0938,  -5.4531],
        [  7.1821,  -9.1845,   9.8548],
        [  2.9470,  13.6085,   2.9809],
        [ -8.8631,  -4.3728,  -1.1746],
        [  3.2701,   6.6338,   9.9209],
        [  2.8890,  -4.4316,   1.1600],
        [  4.5800,   5.5034,  -3.9432],
        [  3.7992,  -3.9474,  -8.9019],
        [  7.5930,  -8.9037,   7.2323],
        [ -2.9497, -10.3028,  -7.3479],
        [ 12.2592,   3.9425,   2.0383],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.3616,  -2.4750,   3.7765],
        [ 11.5456,   6.0861,   8.1101],
        [  4.7813,  -3.1139,   8.2691],
        [ -7.0253,  -7.4493,  -1.5978],
        [  1.3339,   7.5627,  -0.9034],
        [ -4.6393,   7.0461,   1.1624],
        [ -9.4748,  -0.7765,  -5.6695],
        [  5.6022,  -1.9833,   8.3081],
        [ -1.9646,   7.8571,   3.3817],
        [  1.7000,  -4.7609,   2.7935],
        [  3.1271, -10.8343,   2.6442],
        [  2.0729,   7.2878,   7.0880],
        [  6.3048,   0.1469,  -9.1691],
        [  0.8940, -11.0402,  -0.9783],
        [ 11.3118,   6.5957,  -4.5667],
        [  2.0870,  -4.6877,   9.6823],
        [  5.1886,  -2.6250,   6.4677],
        [ -3.3812, -10.1235,   7.4384],
        [ -2.5683, -10.4433,  -2.9182],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.3602,  -2.3808,   7.2098],
        [  3.7992,  -3.9474,  -8.9019],
        [ -7.3470,  -7.0615,   5.0066],
        [ -0.2586,   7.8825,  -9.9840],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.2571,   6.7072,   8.3693],
        [  3.9515,  -3.8425,   6.1729],
        [  6.3211,   1.7017,  -0.2164],
        [  2.0419,  -4.6978,  -5.0983],
        [ 10.4737,  -5.9570,  -1.2108],
        [  5.4426, -10.1181,  -5.1268],
        [ 12.6059,   0.9597,  -6.3264],
        [  5.6523,   3.9443,  -2.9449],
        [ -9.4196,   0.0478,   9.4140],
        [ -8.0750,  -6.0072,  -5.8267],
        [ 11.9695,  -2.9407,  -6.3498],
        [  6.8407,  11.7666,   3.0045],
        [ 12.2589,  -1.9312,   9.8657],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.3118,   6.5957,  -4.5667],
        [  2.3261,  13.7749,   5.9121]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0384, 0.0122,  ..., 1.1655, 0.2095, 1.2234],
        [0.0384, 0.0000, 0.0263,  ..., 1.1695, 0.2093, 1.2278],
        [0.0122, 0.0263, 0.0000,  ..., 1.1680, 0.2100, 1.2260],
        ...,
        [1.1655, 1.1695, 1.1680,  ..., 0.0000, 0.9603, 0.0597],
        [0.2095, 0.2093, 0.2100,  ..., 0.9603, 0.0000, 1.0188],
        [1.2234, 1.2278, 1.2260,  ..., 0.0597, 1.0188, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.5769659 , 12.76408562, 12.59969936, 12.84056604, 12.889695  ,
       12.9089559 ,  5.92952538, 13.89974693,  9.99211663, 11.97238624,
        9.45995955,  6.46560181,  8.29640613,  6.05165248,  4.89121805,
        7.38844703, 13.12218013,  6.76056625, 13.69162795,  7.48538905,
       12.45418236, 11.51811603, 13.12218013,  4.74770899, 10.95593729,
        5.20659892, 13.18278606,  9.41244865, 13.32557133, 11.10066126,
        5.91677188,  7.4544866 , 12.91143491, 13.95472331,  8.68979052,
        8.08320235,  6.99097013, 11.83509372, 10.67132524,  5.62005629,
        9.32094055,  7.06272256,  9.09502385, 12.17901033, 12.53285633,
        4.87083348, 13.85158417, 13.38091121,  5.89074104,  7.01475376,
        9.96152788, 11.65923396, 13.92390689,  9.88310614,  7.39599986,
        5.29009836,  7.15990676,  5.47866221, 11.70168755, 10.7167285 ,
       12.87752052, 10.27572673,  9.68324553, 13.05148826,  5.70591986,
       10.23946247,  7.67939546,  8.43626114,  9.50654967,  5.94293185,
        8.09899874,  5.05534609, 11.27656488,  7.57685908,  6.30647746,
       11.07637531, 13.09426142,  5.13125513,  5.81481044, 10.67322527,
       10.75443106, 12.32545637, 12.84056604,  5.86519525,  5.47866221,
       10.19036081,  7.88677538, 13.18278606, 13.103709  ,  5.51177501,
        6.54615887,  5.1224079 , 12.04924159, 11.4890861 , 12.64235359,
        6.89245352,  9.41970031, 10.06437598, 12.32545637, 13.6105595 ,
       12.41012046, 10.61472636, 12.30858677, 13.0743399 , 13.09426142,
       13.96987806])}
{'P': tensor([[6.6469e-03, 2.5590e-01, 2.5237e-01,  ..., 1.5398e-04, 6.7343e-05,
         7.5463e-04],
        [8.1813e-06, 7.7315e-03, 1.0407e-03,  ..., 7.6833e-06, 2.2310e-04,
         1.2419e-03],
        [1.0214e-05, 1.9055e-02, 7.6342e-03,  ..., 1.0213e-05, 2.3378e-04,
         7.6443e-04],
        ...,
        [1.1498e-02, 5.5848e-02, 2.4521e-02,  ..., 1.8988e-02, 5.2778e-05,
         4.5936e-03],
        [6.3935e-06, 8.3515e-06, 8.5656e-06,  ..., 3.7955e-06, 9.0179e-02,
         1.2687e-04],
        [6.1118e-05, 1.8228e-04, 1.3998e-04,  ..., 5.5085e-05, 3.8261e-03,
         9.1614e-02]]), 'X': tensor([[ 12.5789,   0.1722,   1.6470],
        [ 12.2592,   3.9425,   2.0383],
        [ 12.3357,   3.6001,   0.7903],
        [ 12.4837,   2.7510,   0.4449],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.1469,   6.9237,   1.6160],
        [ 11.5083,  -4.1177,   0.2389],
        [ 11.2209,  -4.7092,   2.6117],
        [ -7.0253,  -7.4493,  -1.5978],
        [  9.9750,  -6.6338,  -1.2440],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.1929,   2.4501,  -6.9352],
        [ -0.7599, -10.8995,  -6.0366],
        [  6.1752,   2.5242,   3.5874],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.3087,   1.8053,  -1.9481],
        [  4.4633,   5.6287,  -0.7754],
        [  0.1287, -11.0065,   6.5757],
        [ 11.7794,   5.5086,  -9.6929],
        [ -9.0961,   1.6529,   3.1553],
        [ -5.0550,   6.8231,  -9.2502],
        [ -9.4196,   0.0478,   9.4140],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.8836,   3.4315,  -4.0327],
        [  1.1274,  14.0116,   8.1753],
        [  3.5965,  -4.0754,  -3.5035],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.9653,  -9.3225,  -3.4727],
        [ -0.1963,   7.8765,  -1.1371],
        [ -4.4520,   7.1370,  -8.7867],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -6.0175,  -8.4541,  -9.8408],
        [  5.8323,  -1.5228,   5.1132],
        [  5.1203,  -2.7151,   6.2110],
        [  5.7485,  -9.9811,  -3.2355],
        [  2.5974,   7.0363,  -7.7810],
        [  2.2437,  -4.6489,   6.1430],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.0467,  -9.8369,  -6.5816],
        [ 12.5717,   1.9266,   7.4903],
        [  2.0547,   7.2957,  -1.9343],
        [  9.3753,   9.5171,   3.9131],
        [  1.3403,  -4.8011,  -1.1698],
        [  0.9809,   7.6642,  -1.4217],
        [  7.0664,  -9.2590,   8.0879],
        [ 10.7566,   7.6212,  -2.3279],
        [  3.4140,  -4.1803,  -8.8224],
        [ -0.5739,   7.9056,   0.6382],
        [  2.4149,  -4.6005,  -4.3339],
        [ -4.9217,   6.8979,  -1.7667],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -9.2707,   0.9661,   4.4863],
        [  4.5503,   5.5359,   7.4735],
        [  4.4633,   5.6287,  -0.7754],
        [  2.0937,  13.8293,  -2.3533],
        [  8.1528,  10.7452,   9.1493],
        [  2.1625,  -4.6697,   7.0658],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -8.8934,   2.2597,   3.5831],
        [ -7.7239,   4.4298,   3.1216],
        [ 12.5199,  -0.4711,   3.0436],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.9379,  -4.1655,  -2.8114],
        [  1.4150,  13.9647,  -4.0915],
        [  5.0506,   4.9277,  -9.5094],
        [ 11.7779,   5.5127,  -2.0699],
        [ -3.6381,   7.4683,   7.3544],
        [  2.6321,  -4.5296,  -2.5159],
        [  3.1271, -10.8343,   2.6442],
        [  3.5939,   6.4029,  -1.8357],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.6078, -10.9229,   6.2869],
        [  2.9260,  -4.4162,   8.3808],
        [  3.3943,   6.5483,   1.0790],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.7275,  14.0667,  -0.9815],
        [  2.2373,  -4.6506,  -1.7334],
        [  2.4639, -10.9431,  -7.3718],
        [  6.3087,   1.8053,  -1.9481],
        [  2.1477,   7.2550,   8.9688],
        [  1.0093,   7.6567,   2.3526],
        [  2.5375,   7.0676,  -4.3490],
        [  0.5250, -11.0306,   5.5358],
        [  3.1892, -10.8220,  -6.8409],
        [ -9.1623,  -3.4288,  -4.9961],
        [  1.7218,  -4.7576,  -3.3617],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.6681,  -4.8556,   9.2518],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.3236,   0.3011,  -4.6668],
        [  2.5382,   7.0672,   4.5097],
        [ 12.4597,  -0.9136,  -4.8763],
        [  3.2701,   6.6338,   9.9209],
        [ -8.5900,   2.9767,   7.4811],
        [ -6.1914,  -8.2997,   8.5380],
        [  3.6703,   6.3443,  -5.4401],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.4867,  12.5816,   2.9549],
        [  8.5716,  10.3596,   6.8818],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.3759,  -1.3943,   6.4659],
        [ -1.9714,   7.8563,  -1.8084],
        [  4.5503,   5.5359,   7.4735]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1139, 0.1047,  ..., 0.5168, 0.4540, 0.4127],
        [0.1139, 0.0000, 0.1241,  ..., 0.4659, 0.5677, 0.4023],
        [0.1047, 0.1241, 0.0000,  ..., 0.5854, 0.5063, 0.5028],
        ...,
        [0.5168, 0.4659, 0.5854,  ..., 0.0000, 0.8368, 0.1859],
        [0.4540, 0.5677, 0.5063,  ..., 0.8368, 0.0000, 0.6632],
        [0.4127, 0.4023, 0.5028,  ..., 0.1859, 0.6632, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.58006038, 12.87752052, 12.85032836, 12.78327037, 13.08414689,
       13.12218013, 12.22276278, 12.16901615, 10.23946247, 11.97949273,
       11.9120321 ,  6.65991877, 10.92597196,  6.67121748,  9.95552626,
        6.5618968 ,  7.18354885, 11.00726483, 13.00380624,  9.24501829,
        8.49161154,  9.41970031,  9.78285424,  6.81117802, 14.05687815,
        5.43544626,  8.16714836, 11.63724792,  7.8788965 ,  8.41171369,
        9.78286733, 10.37698435,  6.02779201,  5.79562691, 11.51811603,
        7.50035355,  5.16205087, 12.11735334, 11.54672755, 12.71843462,
        7.57946459, 13.35927527,  4.98461983,  7.72668467, 11.64746882,
       13.18278606,  5.39723241,  7.92644746,  5.19577146,  8.47371868,
       12.10942412,  9.32094055,  7.16598273,  7.18354885, 13.98691426,
       13.48809595,  5.14607014,  8.1137452 ,  9.51074763,  9.17595899,
        8.90405823, 12.52875793, 13.103709  ,  9.86089579, 14.03618649,
        7.05627461, 13.00413829,  8.30729352,  5.23878371, 11.27656488,
        7.34250948,  9.62958973, 10.30972193,  9.48337063, 11.22993218,
        5.29755487,  7.37576154,  4.91296205, 14.08549802,  5.16077764,
       11.21703654,  6.5618968 ,  7.56617037,  7.72291946,  7.5092834 ,
       11.04313641, 11.28215429,  9.78286733,  5.05963217, 12.46513646,
        9.9353921 , 10.42667093,  6.33075682,  7.50917551, 12.49317563,
        7.39599986,  9.09119351, 10.35464916,  7.32950385, 12.67576792,
       13.72594022, 13.44593519, 12.62711252, 12.45418236,  8.09984231,
        7.16598273])}
{'P': tensor([[4.4647e-03, 3.0560e-01, 3.0047e-01,  ..., 5.9476e-06, 4.8898e-08,
         6.7655e-06],
        [3.1576e-05, 1.4731e-02, 4.2835e-02,  ..., 1.2060e-05, 1.5118e-07,
         1.2813e-05],
        [4.4926e-05, 4.1307e-03, 2.4257e-02,  ..., 1.8089e-05, 2.8146e-07,
         2.0887e-05],
        ...,
        [9.4597e-03, 6.9329e-03, 5.8930e-03,  ..., 1.0084e-01, 2.3631e-05,
         1.1737e-02],
        [3.7106e-05, 1.7308e-05, 1.8940e-05,  ..., 4.2829e-04, 1.1333e-01,
         2.5838e-03],
        [7.7586e-03, 5.0119e-03, 5.3374e-03,  ..., 6.7542e-03, 4.6055e-05,
         6.9273e-02]]), 'X': tensor([[ 12.5919,   0.4035,  -5.8460],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.4691,   2.8525,  -5.5110],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.1379,   4.4125,  -3.8833],
        [  3.9085,   6.1512,  -2.2560],
        [ -7.0825,   5.2073,  -5.0497],
        [ -0.8569,   7.9156,  -6.5557],
        [  9.2979,   9.6048,   7.3138],
        [ -0.0282,   7.8575,  -1.1746],
        [ -9.4196,   0.0478,   9.4140],
        [ -8.9178,   2.1936,   7.5388],
        [  7.0642,  11.6104,  -9.8316],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.8325,  -1.5222,  -2.1341],
        [  5.6093,  -1.9705,   6.4807],
        [ -9.4541,  -0.3327,   8.8246],
        [ -9.3087,   0.7792,  -7.5587],
        [ -7.6719,  -6.6260,   7.1468],
        [  5.6928,  12.4710,   3.6001],
        [  1.1644, -11.0394,   7.8511],
        [ -9.4748,  -0.7765,  -5.6695],
        [  2.1379,  -4.6757,   6.4981],
        [  5.2188,  -2.5839,   2.1833],
        [  7.2340,  -9.1505,  -5.8321],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.4877,  -0.7222,   8.8347],
        [  1.3670,  -4.7990,   3.2603],
        [  9.3624,   9.5318,   0.2779],
        [  6.0140,  -1.0661,  -2.5040],
        [  1.8721,  -4.7322,   0.5524],
        [ 11.8951,  -3.1570,   7.0742],
        [  6.3505,   0.6235,  -8.5849],
        [  0.9945,  -4.8144,  -0.1897],
        [ -1.8367,   7.8718,  -8.6523],
        [  2.0419,  -4.6978,  -5.0983],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.7976,  -4.4682,  -4.2020],
        [ -7.5603,   4.6459,   8.2002],
        [  0.9036,   7.6840,   9.9237],
        [  6.3211,   1.7017,  -0.2164],
        [  1.5653,  13.9377,   4.1773],
        [  6.0890,  -0.8382,   2.9166],
        [ -0.1963,   7.8765,  -1.1371],
        [ -5.6651,   6.4380,   1.3660],
        [  6.0140,  -1.0661,  -2.5040],
        [ 11.9936,  -2.8677,   6.0614],
        [  2.6451,  -4.5250,  -0.5565],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.1379,  -4.6757,   6.4981],
        [  2.8474,   6.8979,   6.6457],
        [  1.3767, -11.0341,   9.9415],
        [  3.3744,   6.5623,  -6.3636],
        [  5.6022,  -1.9833,   8.3081],
        [  2.3589,  13.7668,  -3.6501],
        [  6.3048,   0.1469,  -9.1691],
        [ -8.1666,   3.7653,   3.6075],
        [ -8.9178,   2.1936,   7.5388],
        [  8.7715,  -7.9390,   5.2900],
        [ -6.7745,  -7.7264,   8.4684],
        [  3.0719, -10.8449,  -4.1673],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.2159, -10.8166,  -1.0406],
        [ -7.3554,   4.8985,  -0.6665],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.2707,   0.9661,   4.4863],
        [  2.9470,  13.6085,   2.9809],
        [  4.8432,   5.1968,   1.5006],
        [  5.2188,  -2.5839,   2.1833],
        [  5.8889,  -1.3916,   8.8993],
        [  4.7544,  -3.1425,  -1.5315],
        [ -8.8400,   2.3991,   1.3457],
        [ -8.7409,  -4.6837,  -6.5840],
        [  8.9284,  10.0036,  -6.9210],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.1372,   2.6726,  -6.7885],
        [  5.6523,   3.9443,  -2.9449],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.3305,   1.6119,   1.1538],
        [  0.5175,  14.0910,   4.6540],
        [ -7.8359,   4.2735,   7.2943],
        [  6.1984,  -0.4327,   1.5438],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -5.2589,   6.7024,  -5.5809],
        [ 10.9931,   7.2107,   7.7933],
        [ -5.1377,   6.7751,  -0.4438],
        [  6.3497,   1.3680,  -1.9076],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.5996,   1.3971,  -1.0694],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.1624,  -3.4286,   2.0647],
        [  5.4242, -10.1261,   3.9595],
        [  5.9548,  -1.2273,  -2.6520],
        [  4.9152,  12.8653,  -1.4459],
        [  0.5175,  14.0910,   4.6540],
        [  6.1973,  -0.4372,   6.5720],
        [  0.5850,  -4.7982,   9.6639],
        [  6.3551,   0.7149,   1.3344],
        [  5.1519, -10.2387,   9.8191],
        [ -7.0120,   5.2825,   7.7762],
        [  9.4925,   9.3811,   8.9842],
        [  4.3459,  -3.5327,   0.3667],
        [  3.4140,  -4.1803,  -8.8224],
        [ -6.8618,  -7.6322,   3.7027],
        [  3.5965,  -4.0754,  -3.5035]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.2177e-02, 6.0248e-02,  ..., 1.6567e-01, 1.3108e+00,
         2.9711e-01],
        [6.2177e-02, 0.0000e+00, 4.6603e-02,  ..., 2.2784e-01, 1.3709e+00,
         3.5168e-01],
        [6.0248e-02, 4.6603e-02, 0.0000e+00,  ..., 2.1304e-01, 1.3420e+00,
         3.1702e-01],
        ...,
        [1.6567e-01, 2.2784e-01, 2.1304e-01,  ..., 2.1453e-04, 1.1522e+00,
         1.7930e-01],
        [1.3108e+00, 1.3709e+00, 1.3420e+00,  ..., 1.1522e+00, 0.0000e+00,
         1.0289e+00],
        [2.9711e-01, 3.5168e-01, 3.1702e-01,  ..., 1.7930e-01, 1.0289e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.59840193, 12.77402351, 12.79126785, 12.64235359, 12.91143491,
       12.91505111,  7.28793663,  8.7907923 ,  7.96181288, 13.36800073,
        7.8575648 ,  9.41970031,  9.18358562, 13.59056613, 10.60226614,
        6.02789634,  5.94535529,  9.45995955,  9.34127013, 10.1371552 ,
       13.70893492, 11.10066126,  9.50654967,  5.14124612,  5.82345867,
       11.66453059,  4.92497514, 12.5085995 ,  4.98989906, 13.3607379 ,
        6.10774544,  5.08910491, 12.30695098,  6.38105015,  4.9160962 ,
        8.08320235,  5.1224079 , 10.31886066,  5.27178363,  8.8737556 ,
        7.73692436,  6.54615887, 14.02532758,  6.1463896 ,  7.8788965 ,
        8.57560275,  6.10774544, 12.33167382,  5.24137909, 10.67132524,
        5.14124612,  7.46248947, 11.11969815,  7.37902766,  5.94293185,
       13.96746816,  6.30647746,  8.99276564,  9.18358562, 11.83075005,
       10.27572673, 11.27160035, 10.52587243, 11.2845608 ,  8.83725081,
       10.67322527,  9.32094055, 13.92390689,  7.10378919,  5.82345867,
        6.05113217,  5.69913817,  9.15976746,  9.91668182, 13.40850328,
       12.35421461,  6.69389431,  6.89245352,  5.53175275,  6.53250937,
       14.10045797,  8.92549752,  6.2134961 ,  9.86089579,  8.51928531,
       13.14690137,  8.50278896,  6.49538841, 13.79762415, 12.67680439,
        4.89121805,  9.78285424, 11.48734978,  6.07992623, 13.77222752,
       14.10045797,  6.21275016,  4.83371341,  6.39520397, 11.46176259,
        8.77913428, 13.34587027,  5.60064319,  5.39723241, 10.26327895,
        5.43544626])}
{'P': tensor([[4.6193e-03, 3.7654e-01, 1.9133e-01,  ..., 1.8605e-07, 5.1707e-07,
         1.1980e-07],
        [1.1122e-05, 5.2876e-03, 2.2291e-01,  ..., 1.3338e-07, 2.8712e-07,
         1.0956e-07],
        [7.0204e-06, 3.6052e-05, 4.3005e-03,  ..., 9.2678e-08, 1.9867e-07,
         9.9050e-08],
        ...,
        [4.0041e-03, 1.7836e-03, 1.3484e-03,  ..., 4.7166e-02, 1.6935e-03,
         7.5087e-05],
        [8.8309e-05, 1.0601e-04, 1.5002e-04,  ..., 3.4042e-06, 2.3657e-02,
         9.2939e-05],
        [4.5361e-05, 4.1259e-05, 5.0501e-05,  ..., 2.0506e-04, 2.4342e-02,
         1.9054e-01]]), 'X': tensor([[ 12.5927,   0.4199,  -9.4997],
        [ 12.5154,   2.5072,  -9.3274],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.1588,   4.3364,  -8.0867],
        [ 11.7794,   5.5086,  -9.6929],
        [ 11.1469,   6.9237,   1.6160],
        [  4.0078,  -3.8018,  -6.2168],
        [ -3.3116, -10.1539,   7.5919],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.9505,  -4.7171,   9.9868],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.4837,   2.7510,   0.4449],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.0591,   6.0202,  -1.6339],
        [  2.5620,   7.0549,   8.4825],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -5.1369,   6.7755,   8.3836],
        [ -7.9298,   4.1367,  -3.9773],
        [ 11.0275,  -5.0666,   8.4154],
        [  2.4149,  -4.6005,  -4.3339],
        [  4.1297,  -3.7098,  -3.9838],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.2804,   2.0019,   0.8607],
        [  5.7963,  -1.6018,   1.8856],
        [ -5.1974,   6.7396,   3.3484],
        [  6.9693,  -9.3200,  -8.3822],
        [  5.5240,  -2.1198,   6.2055],
        [  6.3305,   1.6119,   1.1538],
        [  4.9152,  12.8653,  -1.4459],
        [ -2.0721,   7.8432,   6.6380],
        [  7.9403,  -8.6455,  -8.7433],
        [ -5.9987,   6.1949,  -6.2114],
        [ 12.1940,  -2.1881,   7.1499],
        [ -3.3812, -10.1235,   7.4384],
        [  2.1440, -10.9811,   8.5281],
        [  9.3624,   9.5318,   0.2779],
        [  4.2612, -10.5514,  -6.6268],
        [  5.2911,  -2.4824,  -3.6877],
        [  7.9141,  -8.6656,   9.6948],
        [  9.3836,   9.5075,  -5.9278],
        [ -6.3066,  -8.1936,   9.5822],
        [ -8.8934,   2.2597,   3.5831],
        [  2.3928,   7.1404,  -1.7035],
        [  1.7500,   7.4187,   4.4237],
        [  0.9074,  -4.8139,   8.1674],
        [ -7.2986,  -7.1225,  -3.0814],
        [  1.7815,  -4.7481,   4.4289],
        [  4.9152,  12.8653,  -1.4459],
        [ 10.1503,   8.5414,  -7.2681],
        [  0.5250, -11.0306,   5.5358],
        [  2.1504,  -4.6726,   8.7908],
        [ -7.0120,   5.2825,   7.7762],
        [  6.3460,   1.4252,   1.2291],
        [  7.1441,  11.5529,   2.3232],
        [ -1.7290,   7.8825,  -3.3831],
        [ -2.0445,   7.8469,  -0.5857],
        [ 11.7349,  -3.5838,   2.1644],
        [  5.4167, -10.1293,   6.2156],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -6.6998,   5.5954,   0.0823],
        [ 11.2847,  -4.5848,  -4.7503],
        [  3.3094,  13.4969,  -3.6507],
        [  6.3587,   1.1596,  -3.6629],
        [ 12.5258,  -0.4200,  -9.3493],
        [  1.4489,   7.5256,   4.4529],
        [ 12.6059,   0.9597,  -6.3264],
        [ -7.2515,   5.0195,   4.6434],
        [ 12.4013,  -1.2597,  -7.6474],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.1579,  -0.5961,  -6.5595],
        [  4.9452, -10.3187,  -7.8621],
        [  2.5974,   7.0363,  -7.7810],
        [  6.7565,  11.8237,  -4.9882],
        [  3.0719, -10.8449,  -4.1673],
        [  0.1460,   7.8340,  -9.4614],
        [ 11.2767,  -4.6006,   0.7868],
        [ -8.4573,   3.2448,   5.4370],
        [  6.9956,  11.6590,  -7.6876],
        [  6.3246,   1.6696,  -0.1544],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.2365,  -0.2578,  -6.0212],
        [  5.5158,  -2.1337,   3.5246],
        [ 10.4994,  -5.9193,  -7.5153],
        [  3.2366,  -4.2732,   5.1532],
        [  5.6258,  -1.9404,  -6.0715],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.9314,  -9.3435,  -1.2729],
        [  6.3583,   1.1729,   8.7835],
        [ -4.0868,   7.2981,   8.1874],
        [ 11.6801,  -3.7196,  -0.9181],
        [  8.9284,  10.0036,  -6.9210],
        [  5.1472,  -2.6801,   8.1194],
        [  4.8653,   5.1694,  -4.0823],
        [  2.3490,  -4.6199,  -4.2489],
        [ -8.1204,  -5.9308,   4.0761],
        [ -9.1749,  -3.3803,   6.4629],
        [ -7.0825,   5.2073,  -5.0497],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.2508,   1.0572,   2.9978],
        [  6.1730,  -0.5375,  -8.8861],
        [  6.3360,   0.4258,  -9.2509],
        [  3.8111, -10.6787,  -4.6539],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.3376,  -4.6231,   9.8436]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0263, 0.0310,  ..., 0.7369, 1.2917, 1.1699],
        [0.0263, 0.0000, 0.0139,  ..., 0.7625, 1.3014, 1.1778],
        [0.0310, 0.0139, 0.0000,  ..., 0.7609, 1.2892, 1.1652],
        ...,
        [0.7369, 0.7625, 0.7609,  ..., 0.0000, 1.0612, 1.0180],
        [1.2917, 1.3014, 1.2892,  ..., 1.0612, 0.0000, 0.1521],
        [1.1699, 1.1778, 1.1652,  ..., 1.0180, 0.1521, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.59969936, 12.76408562, 12.84056604, 12.889695  , 12.9089559 ,
       13.00380624, 13.12218013,  5.52416368, 10.68031351,  5.9510489 ,
        5.10446653,  8.44940661, 12.78327037, 12.70489563,  7.26074925,
        7.50564223, 10.46739073,  8.50268322,  8.94395628, 12.13568598,
        5.19577146,  5.5513072 ,  9.7751189 ,  6.59176018,  6.01356004,
        8.51088913, 11.63765273,  5.91677188,  6.53250937, 13.77222752,
        8.11227668, 11.73848065,  8.62329466, 12.38881635, 10.67322527,
       11.18839169, 13.3607379 , 11.37939695,  5.84450127, 11.73567759,
       13.35832725, 10.33958639,  9.17595899,  7.53063885,  7.62232307,
        4.89869205, 10.1979669 ,  5.0713364 , 13.77222752, 13.26590071,
       11.04313641,  5.14371005,  8.77913428,  6.50409372, 13.58333857,
        8.06990483,  8.10886021, 12.26997165, 11.48663767,  9.61942445,
        8.72896381, 12.18045978, 13.89671314,  6.46356806, 12.53285633,
        7.66377554, 12.64235359,  8.81930525, 12.46513646, 10.06023665,
        6.18668463, 11.4424816 ,  7.50035355, 13.61801526, 11.27160035,
        7.83534987, 12.17901033,  9.05843201, 13.59673406,  6.54127431,
       10.06407839, 10.07238463,  6.24187044,  5.91408504, 12.05300943,
        5.36061974,  5.9510489 ,  4.78002665, 11.63383221,  6.46560181,
        8.36445758, 12.25806559, 13.40850328,  5.80312949,  7.09887557,
        5.18277858, 10.05558893,  9.77777346,  8.7907923 ,  7.37902766,
        9.3109884 ,  6.19633139,  6.35029274, 11.33837396, 12.10222204,
        5.1805345 ])}
{'P': tensor([[4.5773e-03, 4.5035e-01, 1.7653e-01,  ..., 1.9204e-06, 3.7989e-08,
         9.2435e-08],
        [2.2694e-05, 6.2573e-03, 1.7370e-01,  ..., 1.8417e-06, 3.8139e-08,
         1.2308e-07],
        [1.3099e-05, 3.8568e-05, 7.7589e-03,  ..., 2.3604e-06, 4.3862e-08,
         1.6046e-07],
        ...,
        [2.3280e-04, 2.1504e-04, 4.0155e-04,  ..., 1.2908e-01, 1.9353e-04,
         1.2303e-05],
        [1.0070e-04, 6.5353e-05, 6.1936e-05,  ..., 1.0561e-02, 4.7623e-02,
         5.9280e-06],
        [2.7872e-05, 1.8257e-05, 1.0157e-05,  ..., 3.2706e-05, 3.5259e-04,
         2.5210e-02]]), 'X': tensor([[ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0491, 0.1051,  ..., 1.0853, 1.3640, 1.0524],
        [0.0491, 0.0000, 0.0867,  ..., 1.1062, 1.4003, 1.1014],
        [0.1051, 0.0867, 0.0000,  ..., 1.0354, 1.3535, 1.1217],
        ...,
        [1.0853, 1.1062, 1.0354,  ..., 0.0000, 0.5566, 1.2255],
        [1.3640, 1.4003, 1.3535,  ..., 0.5566, 0.0000, 1.0069],
        [1.0524, 1.1014, 1.1217,  ..., 1.2255, 1.0069, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.62576026, 12.71843462, 12.86648771, 12.88270018, 13.04156075,
       13.05148826,  5.20659892, 12.16901615, 12.1460931 ,  9.34061938,
        9.94285511,  9.99145096,  8.43626114,  9.26455795,  9.78147542,
        5.19065439, 12.46513646, 14.09357547,  6.97106912,  9.13507321,
       12.59840193,  8.8580671 , 11.77124373,  6.76791483,  5.68942572,
        5.05774295, 10.48315436, 14.07623266,  6.76661595,  7.16598273,
        9.69629   ,  7.32233733,  9.87720102, 11.05413545,  8.38252937,
       14.07623266,  9.96391045, 11.70016217, 10.66631784,  9.15866607,
       12.05300943, 12.45418236,  5.21280156, 11.62158554, 13.00413829,
       11.89105259,  6.14086582,  9.94285511, 12.3344536 ,  7.46053281,
        8.09842704,  9.94173708,  6.21275016, 13.6922425 ,  8.90405823,
        5.96556248, 13.04217487, 13.2778378 ,  7.71075226, 10.06023665,
        8.11227668, 10.46739073, 13.1399181 , 11.64436684, 10.91601655,
       12.24467041, 11.35345345,  7.85919687, 10.74409834, 11.00812996,
        8.89458273, 12.64235359,  5.94535529, 12.10222204,  7.67939546,
        5.53175275, 11.9702762 , 13.97430191,  9.43112957, 10.74655346,
       13.1471313 , 11.8159254 , 11.20100795,  8.0550286 , 13.59170809,
       11.72802491,  7.15990676, 13.01728313,  9.55593374,  5.81199422,
       13.88683436,  5.5366698 ,  4.91038884,  8.31625171, 11.92133129,
        7.33350642, 11.69264243,  7.56617037,  5.39723241,  6.99097013,
       12.5085995 , 13.25336001, 11.75618513,  5.94541803, 11.21703654,
       10.35093485])}
{'P': tensor([[3.4153e-03, 4.5749e-01, 1.3891e-01,  ..., 3.0179e-06, 5.2292e-05,
         1.6022e-03],
        [2.1738e-05, 4.9646e-03, 2.5058e-01,  ..., 3.8952e-06, 9.5754e-05,
         2.4232e-03],
        [7.5150e-06, 9.8898e-06, 6.8846e-03,  ..., 2.9873e-06, 2.0579e-04,
         2.9806e-03],
        ...,
        [4.5270e-03, 4.1108e-03, 2.0024e-03,  ..., 7.1732e-02, 6.2799e-05,
         4.5495e-03],
        [1.4035e-06, 1.6854e-06, 3.9508e-06,  ..., 9.1915e-06, 1.1957e-01,
         7.7706e-05],
        [9.3841e-05, 1.1320e-04, 3.4648e-04,  ..., 1.6655e-04, 3.9684e-04,
         5.2438e-02]]), 'X': tensor([[ 12.6038,   0.7665,  -3.4439],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.9808,   4.9340,  -3.3715],
        [ 11.6544,   5.8273,  -3.1375],
        [ 11.7779,   5.5127,  -2.0699],
        [  4.0591,   6.0202,  -1.6339],
        [  1.9204,   7.3518,   8.4781],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.2437,  -4.6489,   6.1430],
        [ -7.9925,  -6.1423,   2.8990],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.3200,   7.1756,   1.1938],
        [ -1.0012, -10.8576,   2.2867],
        [  5.1196,  -2.7160,   4.2705],
        [  2.0547,   7.2957,  -1.9343],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.0864,   7.0387,  -0.1238],
        [ -9.0604,   1.7707,   1.1066],
        [  1.8719,   7.3713,  -3.7465],
        [ -1.6334, -10.7207,   7.1397],
        [ 12.4837,   2.7510,   0.4449],
        [  7.7750,  11.0665,   2.8364],
        [ -1.9714,   7.8563,  -1.8084],
        [  4.3459,  -3.5327,   0.3667],
        [ -5.1398,  -9.1367,   6.2074],
        [  3.3069,   6.6089,   8.7377],
        [ -8.0330,  -6.0765,   6.7945],
        [  2.3376,  -4.6231,   9.8436],
        [ -6.4464,   5.8272,   7.5038],
        [  9.3560,   9.5391,   9.4943],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.2990,  -4.2415,  -7.4857],
        [ -5.9872,   6.2037,  -9.4477],
        [ -3.1671,   7.6158,   6.0765],
        [  7.2340,  -9.1505,  -5.8321],
        [ -2.5683, -10.4433,  -2.9182],
        [  6.2247,  -0.3150,  -9.3657],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.8280,   4.2848,  -7.1955],
        [  8.1528,  10.7452,   9.1493],
        [ -4.0633,  -9.7927,  -3.0825],
        [  8.6271,  -8.0715,  -7.4813],
        [  0.6470, -11.0352,  -1.1931],
        [  7.3592,  11.3937,   6.2407],
        [  0.1041, -11.0045,   2.4311],
        [  4.4290,  -3.4596,   1.4181],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.3459,  -3.5327,   0.3667],
        [ 10.6283,  -5.7253,   7.4878],
        [ -0.0410,   7.8591,   9.3457],
        [  1.6408,   7.4591,  -4.2294],
        [ -8.8835,  -4.3178,   9.3080],
        [  4.7813,  -3.1139,   8.2691],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.8704,  -9.3809,   7.5769],
        [ -2.0445,   7.8469,  -0.5857],
        [ -8.5900,   2.9767,   7.4811],
        [  6.2902,   0.0451,   3.7116],
        [  4.7544,  -3.1425,  -1.5315],
        [ 11.5966,   5.9666,   6.0083],
        [ -2.5683, -10.4433,  -2.9182],
        [  3.6469,   6.3624,  -1.8414],
        [  1.1006,   7.6318,  -2.4877],
        [ 12.5123,  -0.5339,  -1.0728],
        [  6.0305,   3.0296,   0.1375],
        [  1.3767, -11.0341,   9.9415],
        [ 12.5224,  -0.4497,   4.8448],
        [  7.3592,  11.3937,   6.2407],
        [  0.6952,  -4.8060,   8.8976],
        [  4.3523, -10.5232,   8.2138],
        [ -7.3446,   4.9113,  -5.9356],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.1644, -11.0394,   7.8511],
        [ -6.2668,   5.9808,   2.5614],
        [  1.7000,  -4.7609,   2.7935],
        [ -2.0840,   7.8415,  -7.8657],
        [  4.7813,  -3.1139,   8.2691],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.7779,   5.5127,  -2.0699],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.5653,  13.9377,   4.1773],
        [  4.8674,  12.8875,   6.7622],
        [ -4.3963,   7.1629,   4.5263],
        [ -0.0254, -10.9932,   3.2200],
        [ -8.6112,   2.9319,  -7.9943],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.2373,  -4.6506,  -1.7334],
        [ 11.4235,   6.3594,   5.1663],
        [ -8.0330,  -6.0765,   6.7945],
        [ -8.3900,   3.3727,   3.1650],
        [ -7.6293,  -6.6859,  -5.5161],
        [  2.0937,  13.8293,  -2.3533],
        [ 12.1504,   4.3674,  -7.2998],
        [ -2.8162,   7.7059,  -9.6885],
        [  3.2692,  13.5098,  -4.9032],
        [  5.0506,   4.9277,  -9.5094],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -0.8885,   7.9160,   0.0571],
        [ 11.2939,   6.6325,  -5.8311],
        [ -9.4505,  -1.7081,  -0.7039],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.0617,  12.7957,   8.3544],
        [  5.1861,   4.7361,  -3.7312]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0269, 0.0908,  ..., 0.3250, 1.1284, 0.1599],
        [0.0269, 0.0000, 0.0738,  ..., 0.3329, 1.1131, 0.1802],
        [0.0908, 0.0738, 0.0000,  ..., 0.4053, 1.1458, 0.2507],
        ...,
        [0.3250, 0.3329, 0.4053,  ..., 0.0000, 0.9513, 0.2053],
        [1.1284, 1.1131, 1.1458,  ..., 0.9513, 0.0000, 1.1140],
        [0.1599, 0.1802, 0.2507,  ..., 0.2053, 1.1140, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.62711252, 12.67576792, 12.91505111, 12.95702647, 13.03002338,
       13.00413829,  7.26074925,  7.5984809 ,  9.78285424,  5.16205087,
       10.08001894,  9.48337063,  7.54127425, 10.9036226 ,  5.79543038,
        7.57946459,  9.51074763, 13.13205357,  9.23178204,  7.60529892,
       10.84437705, 12.78327037, 13.5247215 ,  8.09984231,  5.60064319,
       10.48315436,  7.39004226, 10.07238463,  5.1805345 ,  8.68979052,
       13.3614598 , 10.52587243,  5.37343173,  8.62162372,  8.24808832,
       11.66453059, 10.75443106,  6.23262289, 10.74655346,  8.92549752,
        8.92395381, 13.48809595, 10.60226614, 11.81423161, 11.05413545,
       13.56367857, 11.0050379 ,  5.62005629,  5.5513072 ,  5.60064319,
       12.07225469,  7.85919687,  7.6374603 ,  9.87720102,  5.70591986,
       10.8207297 , 11.62770332,  8.10886021,  9.09119351,  6.29034958,
        5.69913817, 13.04156075, 10.75443106,  7.33350642,  7.71075226,
       12.52372629,  6.74872981, 11.11969815, 12.53047489, 13.56367857,
        4.85603886, 11.38774149,  8.8353674 ,  9.77588801, 11.10066126,
        8.66273171,  5.05534609,  8.1137452 ,  5.70591986, 11.70168755,
       13.00413829, 10.37698435,  9.77588801, 14.02532758, 13.77604293,
        8.40445872, 10.99326577,  9.09661212, 12.24467041,  5.16077764,
       13.0743399 , 10.07238463,  9.04255655, 10.14436912, 13.98691426,
       12.91143491,  8.20436764, 13.89974693,  7.05627461, 12.3828957 ,
        7.96575782, 13.09737053,  9.60359158,  5.24137909, 13.76047999,
        7.02326515])}
{'P': tensor([[3.3544e-03, 3.3784e-01, 1.8638e-01,  ..., 3.3499e-08, 1.1042e-06,
         4.7783e-05],
        [1.9415e-05, 8.0188e-03, 2.3318e-02,  ..., 8.6409e-08, 3.8927e-06,
         1.8873e-04],
        [3.9859e-05, 2.1441e-03, 1.2591e-02,  ..., 1.5759e-07, 7.4621e-06,
         2.9794e-04],
        ...,
        [3.7977e-06, 2.4463e-06, 2.7183e-06,  ..., 3.7465e-02, 1.4866e-05,
         1.7673e-06],
        [1.4298e-06, 1.2755e-06, 1.3978e-06,  ..., 1.1677e-02, 5.5728e-02,
         6.9998e-05],
        [4.8486e-06, 5.4324e-06, 5.3995e-06,  ..., 7.2974e-05, 2.5913e-03,
         6.2149e-02]]), 'X': tensor([[ 12.6059,   0.9597,  -6.3264],
        [ 12.4996,   2.6335,  -6.3166],
        [ 12.4691,   2.8525,  -5.5110],
        [ 12.1504,   4.3674,  -7.2998],
        [ 12.1588,   4.3364,  -8.0867],
        [ 12.2218,   4.0953,  -8.2072],
        [  2.6451,  -4.5250,  -0.5565],
        [ -7.4887,   4.7364,   8.9074],
        [ -7.0948,   5.1940,   9.8171],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.0890,  -0.8382,   2.9166],
        [  3.6469,   6.3624,  -1.8414],
        [ -7.0825,   5.2073,  -5.0497],
        [ -8.5859,  -5.0393,   0.5476],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.8902,  -1.3887,   4.0388],
        [ -8.8745,   2.3097,   3.5600],
        [  5.0233,   4.9647,  -3.1788],
        [ 11.6544,   5.8273,  -3.1375],
        [  1.6408,   7.4591,  -4.2294],
        [ -7.3333,   4.9246,  -7.9212],
        [  4.2275,  -3.6320,  -0.2714],
        [ -9.4697,  -0.6157,  -6.8472],
        [  9.1409,  -7.5789,   1.5814],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -2.6735, -10.4062,  -1.5911],
        [ 11.9808,   4.9340,  -3.3715],
        [  2.6321,  -4.5296,  -2.5159],
        [  1.6232,  -4.7717,  -2.5655],
        [ -3.9508,  -9.8515,   1.7923],
        [ -9.3616,  -2.4749,   0.1793],
        [  1.8447,  -4.7372,   8.1235],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.4541,  -0.3327,   8.8246],
        [ 12.2218,   4.0953,  -8.2072],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.9260,  -4.4162,   8.3808],
        [ -8.7735,   2.5640,  -6.3313],
        [ -5.3328,  -8.9992,   7.3440],
        [ -2.1382,   7.8339,   7.3524],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.5258,  -0.4200,  -9.3493],
        [ -3.5173, -10.0623,   4.8280],
        [  6.9956,  11.6590,  -7.6876],
        [  4.9337,  12.8566,  -3.6582],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -9.4159,  -2.0739,   9.6841],
        [  1.1006,   7.6318,  -2.4877],
        [ 10.9931,   7.2107,   7.7933],
        [  6.1989,  -9.7591,   0.8289],
        [ -6.3066,  -8.1936,   9.5822],
        [  7.1441,  11.5529,   2.3232],
        [  2.1824,  -4.6647,   1.1985],
        [ 11.4235,   6.3594,   5.1663],
        [ -1.1885, -10.8211,   0.0865],
        [ 12.4592,  -0.9168,  -6.7571],
        [  3.9085,   6.1512,  -2.2560],
        [  1.1644, -11.0394,   7.8511],
        [ 11.7349,  -3.5838,   2.1644],
        [  0.1252, -11.0062,   7.2564],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.5375,   7.0676,  -4.3490],
        [ -5.9225,  -8.5355,   2.4778],
        [ -0.9715, -10.8630,   0.3555],
        [  5.3804,  -2.3504,   7.2744],
        [  7.8425,  -8.7202,   6.4530],
        [  6.3031,   1.8477,  -7.1706],
        [  7.0515,  11.6194,   8.4289],
        [  8.8091,  -7.9037,   0.3631],
        [ 10.2712,   8.3707,  -4.9743],
        [  0.9478,  -4.8144,   0.6371],
        [ -5.9940,   6.1985,   5.9566],
        [  8.5198,  10.4091,  -8.6486],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.9260,  -4.4162,   8.3808],
        [  4.3459,  -3.5327,   0.3667],
        [ 12.3609,   3.4769,  -8.5762],
        [ -9.3739,   0.3982,  -4.2564],
        [  4.0078,  -3.8018,  -6.2168],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.6817,   6.3354,   6.1233],
        [  3.1271, -10.8343,   2.6442],
        [  0.4996,   7.7738,  -4.4711],
        [  4.5057,  -3.3894,  -0.8160],
        [  2.8994,   6.8675,  -9.5987],
        [  5.4336,  -2.2676,   5.6933],
        [ -0.8675, -10.8815,   6.8561],
        [  6.9314,  -9.3435,  -1.2729],
        [  6.3534,   1.2990,  -7.2942],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -8.1666,   3.7653,   3.6075],
        [  2.2182,  13.8007,   5.1889],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.3616,  -2.4749,   0.1793],
        [  5.6022,  -1.9833,   8.3081],
        [ -9.4735,  -1.2620,  -3.6742],
        [  4.3459,  -3.5327,   0.3667],
        [  3.7237,  -3.9966,  -5.0376],
        [  9.5203,   9.3483,   5.8781],
        [  4.1348,   5.9514,  -0.8856],
        [ -8.0829,  -5.9940,   5.0564],
        [ -7.4807,   4.7463,   2.0616],
        [ -2.0840,   7.8415,  -7.8657]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0393, 0.0608,  ..., 1.3964, 1.0309, 0.4971],
        [0.0393, 0.0000, 0.0466,  ..., 1.4253, 1.0595, 0.5359],
        [0.0608, 0.0466, 0.0000,  ..., 1.3942, 1.0285, 0.5308],
        ...,
        [1.3964, 1.4253, 1.3942,  ..., 0.0000, 0.3658, 1.0229],
        [1.0309, 1.0595, 1.0285,  ..., 0.3658, 0.0000, 0.6834],
        [0.4971, 0.5359, 0.5308,  ..., 1.0229, 0.6834, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.64235359, 12.77402351, 12.79126785, 12.91143491, 12.9089559 ,
       12.889695  ,  5.24137909,  8.86082124,  8.79284466,  5.54776934,
        9.87720102,  6.1463896 ,  7.33350642,  8.7907923 ,  9.95552626,
       10.31584102,  6.05165248,  9.17016857,  7.06272256, 13.03002338,
        7.6374603 ,  8.83341608,  5.57341623,  9.48969978, 11.87412439,
        9.47369628, 10.74409834, 12.95702647,  5.23878371,  5.04028538,
       10.61417181,  9.68323686,  5.08372149,  9.34127013,  9.45995955,
       12.889695  , 10.67132524,  5.29755487,  9.14044984, 10.4606265 ,
        8.12043983,  9.71392179,  8.41171369, 12.53285633, 10.65929968,
       13.59673406, 13.77075174, 10.06407839,  9.64157461,  7.71075226,
       13.14690137, 11.56146873, 10.33958639, 13.58333857,  5.14999799,
       13.0743399 , 10.88618114, 12.49291768,  7.28793663, 11.10066126,
       12.26997165, 11.00694688,  5.79391825,  7.5092834 , 10.38897563,
       10.90638066,  5.87132718, 11.72802491,  6.56833864, 13.59170809,
       11.83509372, 13.25017211,  4.90676242,  8.62260327, 13.45124917,
       12.35625694,  5.29755487,  5.60064319, 12.84056604,  9.38232121,
        5.52416368,  8.8353674 ,  7.32754882, 11.27656488,  7.78980299,
        5.63824971,  7.4544866 ,  5.88782913, 10.91601655, 11.63383221,
        6.48486171, 13.38091121,  9.47369628,  8.99276564, 13.97779839,
        8.53014915,  9.68323686,  5.94293185,  9.55720651,  5.60064319,
        5.46244988, 13.3426541 ,  7.24677664, 10.06286074,  8.85939626,
        8.1137452 ])}
{'P': tensor([[5.0409e-03, 2.5443e-01, 2.2634e-01,  ..., 1.3961e-07, 2.9498e-04,
         7.8388e-05],
        [5.9095e-06, 4.1138e-03, 1.5644e-01,  ..., 1.4018e-07, 2.7698e-04,
         5.8861e-06],
        [4.5987e-06, 6.0081e-05, 4.2846e-03,  ..., 1.6032e-07, 3.8637e-04,
         4.4087e-06],
        ...,
        [1.4924e-05, 6.1920e-06, 5.6235e-06,  ..., 6.1025e-02, 9.5708e-07,
         1.0985e-05],
        [8.4321e-07, 2.4901e-06, 3.4060e-06,  ..., 1.9288e-06, 1.9647e-02,
         1.6299e-07],
        [2.2656e-03, 1.0683e-02, 7.9895e-03,  ..., 9.6579e-08, 1.9610e-05,
         7.1445e-03]]), 'X': tensor([[ 12.6000,   1.3839,  -3.2992],
        [ 12.1379,   4.4125,  -3.8833],
        [ 11.9808,   4.9340,  -3.3715],
        [ 11.6544,   5.8273,  -3.1375],
        [ 11.7779,   5.5127,  -2.0699],
        [ 11.5935,   5.9740,  -4.3952],
        [  2.1625,  -4.6697,   7.0658],
        [  1.0391,  -4.8141,  -0.0768],
        [ -7.6719,  -6.6260,   7.1468],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4630,  -1.5145,   0.2983],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.0766,  -6.0046,  -3.1873],
        [  9.0004,  -7.7195,   4.6945],
        [  6.3048,   0.1469,  -9.1691],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 10.4758,  -5.9539,   1.3667],
        [  2.7685, -10.8982,  -6.8928],
        [  4.1140,  -3.7220,  -0.1273],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.2076,  13.8032,  -1.5099],
        [ -2.3317,   7.8034,  -3.6274],
        [  4.2339,  -3.6268,   6.3125],
        [  2.6321,  -4.5296,  -2.5159],
        [ -5.7347,  -8.6908,   9.2878],
        [ 12.5267,  -0.4125,  -8.7331],
        [ 10.6673,  -5.6649,  -4.1646],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.3444,   0.5821,   6.9101],
        [ 12.5224,  -0.4497,   4.8448],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 12.5927,   0.4199,  -9.4997],
        [ -2.0445,   7.8469,  -0.5857],
        [  6.3360,   0.4258,  -9.2509],
        [ -0.5739,   7.9056,   0.6382],
        [ -7.7842,   4.3466,  -1.7548],
        [  4.1637,  13.1898,  -1.9189],
        [  3.9591, -10.6391,  -6.2557],
        [ 12.2692,  -1.8882,   3.9253],
        [ -3.5029,   7.5138,   8.1853],
        [ 10.0619,   8.6627,  -9.2884],
        [  4.5800,   5.5034,  -3.9432],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.6614,  -4.5192,  -9.3641],
        [ -5.2589,   6.7024,  -5.5809],
        [  5.7802,  -1.6363,  -6.2782],
        [  0.9445,   7.6736,   0.5095],
        [  3.3474,  -4.2162,   5.2455],
        [ 11.6634,   5.8050,   9.1590],
        [  6.0140,  -1.0661,  -2.5040],
        [  0.1931,  -4.7491,   7.4385],
        [  1.7903,  -4.7467,   0.1168],
        [  1.7228,  -4.7575,   8.3396],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.6035,   0.7494,   7.1536],
        [ -3.0008, -10.2827,  -8.9480],
        [ -4.2040,   7.2486,   2.9875],
        [  6.3413,   0.4883,  -6.8416],
        [  5.3602,  -2.3808,   7.2098],
        [  7.1441,  11.5529,   2.3232],
        [ -1.8788, -10.6567,   3.7068],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.7685, -10.8982,  -6.8928],
        [  5.2939,  -2.4784,  -4.5840],
        [  8.6420,  -8.0580,   6.5781],
        [  5.3602,  -2.3808,   7.2098],
        [  3.9022,  13.2906,   5.5788],
        [  9.6614,   9.1784,  -1.2087],
        [  2.5813,  -4.5471,   5.3597],
        [ -2.0229, -10.6163,   8.6930],
        [  3.1756, -10.8247,   6.8882],
        [  6.9693,  -9.3200,  -8.3822],
        [  4.9452, -10.3187,  -7.8621],
        [  4.2096,   5.8815,   1.9230],
        [  0.9445,   7.6736,   0.5095],
        [ -2.4956, -10.4682,   3.5632],
        [  4.6357,   5.4415,   7.9067],
        [ -0.8885,   7.9160,   0.0571],
        [ -3.4488, -10.0934,   2.5195],
        [  1.0000,   7.6591,  -4.5011],
        [ -1.6220,   7.8917,   0.0315],
        [  5.5240,  -2.1198,   6.2055],
        [  9.6614,   9.1784,  -1.2087],
        [  1.7815,  -4.7481,   4.4289],
        [  0.9036,   7.6840,   9.9237],
        [ -9.4739,  -1.2497,   0.5483],
        [  1.7232,  -4.7574,   3.1270],
        [ -8.7544,   2.6096,   3.3146],
        [ 10.2470,   8.4054,  -0.1177],
        [  9.6614,   9.1784,  -1.2087],
        [  1.9312,  13.8649,   8.3563],
        [  7.7794,  11.0629,   8.9069],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.6231,   5.4556,   8.4973],
        [  5.9655,   3.2172,  -7.5953],
        [ -8.3539,  -5.5111,   0.3373],
        [  7.5309,  11.2617,   7.2656],
        [ 12.5258,  -0.4200,  -9.3493]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0738, 0.0668,  ..., 1.1400, 1.0626, 0.3110],
        [0.0738, 0.0000, 0.0448,  ..., 1.2099, 1.0944, 0.2671],
        [0.0668, 0.0448, 0.0000,  ..., 1.2039, 1.0496, 0.3118],
        ...,
        [1.1400, 1.2099, 1.2039,  ..., 0.0000, 1.4970, 1.2933],
        [1.0626, 1.0944, 1.0496,  ..., 1.4970, 0.0000, 1.3612],
        [0.3110, 0.2671, 0.3118,  ..., 1.2933, 1.3612, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.67576792, 12.91505111, 12.95702647, 13.03002338, 13.00413829,
       13.04217487,  5.14607014,  4.92497514, 10.1371552 , 13.25336001,
        9.58347456,  5.38342077, 10.06407839, 11.85743425,  6.30647746,
       10.38897563, 12.04924159, 12.0495521 , 11.24434374,  5.54776934,
        9.73579738, 10.0674887 , 10.06850677,  4.92497514, 13.97857591,
        8.14433542,  5.5748624 ,  5.23878371, 10.412299  , 12.53345445,
       12.07818871, 11.80092578,  9.36256096, 12.53047489, 12.64235359,
        9.71392179, 12.59969936,  8.10886021,  6.35029274,  7.92644746,
        8.91551231, 13.83139024, 11.35182589, 12.41366754,  8.29022197,
       13.27717982,  7.15990676,  4.87083348, 12.19805394,  5.27178363,
        5.24461068,  8.51928531,  6.00732193,  7.73151309,  5.38342077,
       13.02818702,  6.10774544,  4.75303127,  5.07306274,  5.05981914,
        6.00663314,  9.37989631, 12.62576026, 10.71162681,  8.37954853,
        6.36004393,  5.86519525, 13.58333857, 10.82106505, 11.63765273,
       11.24434374,  5.84533345, 11.8159254 ,  5.86519525, 13.85158417,
       13.32613445,  5.2287058 , 10.80728788, 11.28093507, 11.63765273,
       11.4424816 ,  7.23276582,  7.73151309, 10.76154256,  7.14837514,
        7.96575782, 10.66631784,  7.72415048,  8.05668609,  5.91677188,
       13.32613445,  5.0713364 ,  7.73692436,  9.55593374,  5.05989547,
        9.13507321, 13.25336001, 13.32613445, 13.99877413, 13.52430176,
       10.4606265 ,  7.15100303,  6.77777606, 10.00794468, 13.54774355,
       12.53285633])}
{'P': tensor([[5.6924e-03, 2.2755e-01, 2.2210e-01,  ..., 1.1025e-05, 8.6192e-05,
         2.6652e-07],
        [1.0550e-04, 7.0650e-03, 2.8593e-01,  ..., 1.5131e-05, 8.4549e-05,
         2.8906e-07],
        [3.2861e-06, 4.3026e-06, 3.2315e-03,  ..., 1.8966e-05, 1.1641e-05,
         6.4799e-08],
        ...,
        [3.8899e-06, 3.7771e-06, 3.8284e-06,  ..., 8.6050e-02, 8.7210e-05,
         3.3233e-05],
        [4.6845e-03, 4.6014e-03, 7.3111e-03,  ..., 2.9133e-04, 3.9665e-02,
         5.7805e-05],
        [3.5637e-03, 2.7205e-03, 7.8752e-04,  ..., 3.0416e-06, 6.1878e-03,
         2.4750e-02]]), 'X': tensor([[ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-1.5261e+00,  7.8988e+00,  9.3768e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[1.5170e-04, 3.3650e-02, 8.4292e-02,  ..., 6.8263e-01, 2.0001e-01,
         5.8054e-01],
        [3.3650e-02, 0.0000e+00, 8.7255e-02,  ..., 7.0074e-01, 2.3269e-01,
         6.0758e-01],
        [8.4292e-02, 8.7255e-02, 0.0000e+00,  ..., 7.5984e-01, 2.4360e-01,
         6.4178e-01],
        ...,
        [6.8263e-01, 7.0074e-01, 7.5984e-01,  ..., 0.0000e+00, 5.5545e-01,
         2.5249e-01],
        [2.0001e-01, 2.3269e-01, 2.4360e-01,  ..., 5.5545e-01, 0.0000e+00,
         4.0317e-01],
        [5.8054e-01, 6.0758e-01, 6.4178e-01,  ..., 2.5249e-01, 4.0317e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.67680439, 12.7125079 , 12.97370371, 13.01187484, 13.00413829,
       13.06498554, 10.1979669 ,  7.50917551, 12.50996828,  8.74470218,
       11.9182956 ,  7.24677664, 13.60275211,  8.11227668, 11.46176259,
        6.38105015,  9.50913849, 11.83570132,  5.39098669,  8.55982105,
       12.26997165,  5.37343173, 11.97957425, 12.77402351,  6.1561646 ,
       13.61600536, 13.32645552,  7.58668144,  7.822854  ,  8.53501777,
        8.1303945 ,  9.4997987 ,  7.72291946,  9.78286733,  6.33517349,
       13.34587027,  9.25336371,  7.46053281,  6.30647746,  9.71392179,
        5.05774295,  7.5403822 ,  8.72896381,  8.83725081,  8.20436764,
        6.383897  , 10.60226614,  7.46248947,  9.13507321,  5.94909709,
        8.74470218, 13.89886469,  6.21275016, 13.41713661,  8.97052115,
        5.37343173, 10.38897563,  7.57946459,  5.05774295,  6.10774544,
       11.81423161, 11.35345345, 11.54672755,  6.01356004,  6.89245352,
       10.80728788,  8.30729352, 13.6660671 , 11.24434374,  6.80559669,
       12.32408516,  5.95111804, 13.91198191,  5.48276038, 12.70489563,
       11.64746882, 13.99877413,  8.01009083,  6.69389431,  6.90115901,
        7.83486943,  6.78325151,  7.34250948,  9.87720102, 12.12641969,
        8.04483495,  6.39187285, 10.67132524,  6.99097013,  9.19837792,
       11.35345345,  6.46356806,  7.53884104,  7.42512621,  9.54154096,
       13.46553623, 12.72524615,  9.97964013, 13.27994304,  5.46244988,
       11.9120321 , 11.74901778,  8.8737556 ,  8.62631801,  6.1561646 ,
       11.4890861 ])}
{'P': tensor([[3.5334e-03, 2.3186e-01, 2.2560e-01,  ..., 1.7716e-07, 3.8665e-06,
         1.3715e-06],
        [5.3989e-05, 3.3414e-02, 4.5146e-03,  ..., 7.5024e-07, 2.0179e-05,
         8.5523e-06],
        [1.0185e-05, 1.6888e-02, 5.0747e-03,  ..., 1.0533e-07, 2.6908e-06,
         1.1602e-06],
        ...,
        [3.9602e-04, 2.8546e-04, 2.0573e-04,  ..., 2.2763e-02, 3.4160e-03,
         3.4042e-04],
        [1.2275e-03, 1.2896e-03, 9.2457e-04,  ..., 1.2426e-03, 7.5648e-02,
         2.4790e-03],
        [1.4473e-03, 1.0666e-03, 8.4440e-04,  ..., 8.6638e-03, 2.3456e-01,
         1.0835e-01]]), 'X': tensor([[ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-2.5161e+00,  7.7699e+00, -9.3531e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0445, 0.0621,  ..., 1.1676, 0.7244, 0.6680],
        [0.0445, 0.0000, 0.0528,  ..., 1.1688, 0.7207, 0.6693],
        [0.0621, 0.0528, 0.0000,  ..., 1.2202, 0.7730, 0.7205],
        ...,
        [1.1676, 1.1688, 1.2202,  ..., 0.0000, 0.4570, 0.4997],
        [0.7244, 0.7207, 0.7730,  ..., 0.4570, 0.0000, 0.0948],
        [0.6680, 0.6693, 0.7205,  ..., 0.4997, 0.0948, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.70489563, 12.91143491, 12.889695  , 12.9089559 , 12.84056604,
       13.01728313,  7.14837514,  5.05963217, 13.99048094, 10.88618114,
        9.97220799, 10.33712258,  7.78980299,  7.5158489 ,  7.32950385,
        6.54127431,  8.20436764,  9.53015376, 13.68672855, 12.62711252,
        9.53015376,  9.71392179,  5.91408504, 12.12448912, 11.2845608 ,
       10.67132524,  8.8737556 ,  6.54693987,  6.97106912, 13.32557133,
        5.13182506,  8.10886021, 11.11969815,  8.44940661, 11.72802491,
        7.27340834,  8.06990483, 14.10796466,  8.36445758,  4.96231532,
        8.36445758,  9.09502385, 12.1494023 ,  5.41348121,  6.1463896 ,
       13.59170809, 12.76408562,  9.93553424,  5.5671389 , 11.85890664,
       10.04916678, 11.92133129, 13.09426142, 13.96905756, 11.83570132,
       11.81423161,  5.26067019,  7.06169391, 12.86648771, 11.64436684,
       11.81423161,  7.02326515, 12.41366754, 13.77604293,  5.10444097,
       13.85158417, 11.36089355, 12.44177641, 10.30972193,  7.14837514,
        9.85330646,  7.24326992, 10.37698435, 11.4890861 , 11.46176259,
        7.32950385,  6.21275016,  8.95133961,  6.41202986, 12.04924159,
        8.30125201, 11.04953767,  8.16714836,  7.5092834 ,  9.7712321 ,
       11.9120321 ,  6.32572323,  5.39098669, 10.61472636, 12.50996828,
       13.00413829, 11.59512194, 10.74409834, 10.53368644,  9.37989631,
        8.8580671 , 12.24467041, 13.66051884,  5.07306274, 14.03618649,
       12.21532282,  4.87083348,  5.18047638, 11.83075005,  5.82345867,
        5.29009836])}
{'P': tensor([[5.8272e-03, 2.3583e-01, 2.0954e-01,  ..., 2.1267e-03, 8.0094e-05,
         6.1580e-08],
        [4.1084e-06, 3.0856e-03, 4.8318e-01,  ..., 3.1272e-03, 1.6282e-05,
         2.5373e-08],
        [9.1330e-06, 3.6889e-05, 8.9039e-03,  ..., 1.1165e-02, 3.8563e-05,
         7.4191e-08],
        ...,
        [1.0778e-06, 3.0405e-06, 3.6523e-06,  ..., 2.9785e-02, 4.0768e-06,
         7.9114e-07],
        [1.6429e-03, 4.0571e-03, 4.0773e-03,  ..., 4.3764e-04, 6.3209e-02,
         2.6622e-06],
        [1.0885e-04, 3.0310e-05, 2.6621e-05,  ..., 3.9925e-06, 5.4177e-05,
         1.0990e-01]]), 'X': tensor([[ 12.5770,   1.8512,  -0.6578],
        [ 11.9122,   5.1397,  -1.5498],
        [ 11.7418,   5.6070,  -1.4192],
        [ 11.7779,   5.5127,  -2.0699],
        [ 12.3357,   3.6001,   0.7903],
        [ 11.4743,   6.2478,  -2.1100],
        [ 10.9931,   7.2107,   7.7933],
        [ 11.5083,  -4.1177,   0.2389],
        [ 12.3609,   3.4769,  -8.5762],
        [ 11.7795,  -3.4697,   3.5624],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.9843,  -9.8679,  -6.1449],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.9033,   2.2329,  -8.8912],
        [  9.2979,   9.6048,   7.3138],
        [ -3.9700,   7.3453,  -4.1486],
        [ 11.2847,  -4.5848,  -4.7503],
        [  9.3624,   9.5318,   0.2779],
        [  1.1350,  14.0104,   0.6321],
        [  6.2975,   0.0947,   5.1791],
        [ 12.5258,  -0.4200,  -9.3493],
        [  7.0664,  -9.2590,   8.0879],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.9379,  -4.1655,  -2.8114],
        [  5.2608,  12.6977,   4.9166],
        [  2.5128, -10.9365,   9.9645],
        [ -8.5900,   2.9767,   7.4811],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.4336,  -2.2676,   5.6933],
        [ -8.6229,   2.9068,   2.4429],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.3534,   1.2990,  -7.2942],
        [  6.3205,   1.7068,  -3.8619],
        [  0.3354,  -4.7708,  -4.9125],
        [ 10.9162,  -5.2599,  -7.0812],
        [  7.5064,  -8.9651,  -0.0994],
        [  1.8719,   7.3713,  -3.7465],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.4351,  -1.8891,   3.5642],
        [ 10.0567,   8.6697,   4.6329],
        [ 10.9875,   7.2207,  -8.8549],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.2571,  -4.7594,   1.5618],
        [  5.9912,   3.1453,  -3.9989],
        [ -6.4464,   5.8272,   7.5038],
        [  2.6321,  -4.5296,  -2.5159],
        [ -7.4733,   4.7555,   6.2490],
        [  5.4242, -10.1261,   3.9595],
        [ -9.2345,  -3.1345,   3.9715],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.2199,   4.6861,  -1.9304],
        [  5.0506,   4.9277,  -9.5094],
        [  1.9241,   7.3503,   7.9004],
        [ 11.7726,  -3.4876,  -6.7402],
        [  9.3624,   9.5318,   0.2779],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.3739,   0.3982,  -4.2564],
        [  7.0642,  11.6104,  -9.8316],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -1.5408,   7.8978,   5.3470],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.7078,   2.7182,   1.8146],
        [  0.1532,  -4.7422,  -7.0145],
        [  2.6310, -10.9195,   3.8054],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.2242,  -0.3171,   7.4978],
        [ 10.0995,  -6.4739,   2.0974],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -7.9667,   4.0814,  -0.6397],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.8093,  -9.4178,   8.4244],
        [  2.5318,  13.7231,   6.3692],
        [ -0.0410,   7.8591,   9.3457],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.3266,   0.3292,   9.3502],
        [ -9.4157,   0.0821,  -5.0025],
        [ -5.3382,   6.6533,   7.8054],
        [ -1.6086,   7.8928,   4.3385],
        [  8.1217,  -8.5024,  -0.6149],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -0.8885,   7.9160,   0.0571],
        [  5.9912,   3.1453,  -3.9989],
        [ 12.5154,   2.5072,  -9.3274],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.0004,  -7.7195,   4.6945],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.3384,  -2.6161,  -4.7630],
        [  9.1820,   9.7334,  -2.7932],
        [ -6.6998,   5.5954,   0.0823],
        [  4.4915,   5.5990,   7.0405],
        [ 11.7162,   5.6728,  -8.9372],
        [ -2.3317,   7.8034,  -3.6274],
        [ -8.4573,   3.2448,   5.4370],
        [  7.4132,  11.3526,   0.3322],
        [  5.6928,  12.4710,   3.6001],
        [  6.3301,   0.3638,  -7.5685],
        [ -3.7964,  -9.9294,   7.7147]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0873, 0.0916,  ..., 0.5726, 0.3401, 1.1885],
        [0.0873, 0.0000, 0.0192,  ..., 0.6082, 0.3241, 1.2755],
        [0.0916, 0.0192, 0.0000,  ..., 0.5926, 0.3429, 1.2793],
        ...,
        [0.5726, 0.6082, 0.5926,  ..., 0.0000, 0.9125, 1.1702],
        [0.3401, 0.3241, 0.3429,  ..., 0.9125, 0.0000, 1.3108],
        [1.1885, 1.2755, 1.2793,  ..., 1.1702, 1.3108, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.7125079 , 12.97370371, 13.01187484, 13.00413829, 12.85032836,
       13.06498554, 13.14690137, 12.22276278, 12.84056604, 12.27991991,
        5.23878371, 11.54071859, 11.92133129, 10.0674887 ,  9.09119351,
        9.17905491, 13.36800073,  8.34949383, 12.18045978, 13.3607379 ,
       14.05633441,  6.29821673, 12.53285633, 11.64746882,  9.54154096,
        8.62631801,  6.1561646 ,  9.86089579, 13.744387  , 11.22142195,
        9.09119351,  5.84533345,  5.88782913,  9.09963894,  5.18277858,
        6.48486171,  6.54693987,  4.78258092, 12.11735334, 11.69264243,
        7.60529892, 11.59512194,  9.62238828, 13.2778378 , 13.14777433,
       12.42554808,  8.62329466,  4.7663526 ,  6.76661595,  8.68979052,
        5.23878371,  8.8580671 , 11.48734978,  9.75200848,  9.34127013,
       12.49291768,  9.14973319,  7.01475376,  7.05627461,  7.59795445,
       12.27835996, 13.3607379 ,  9.60359158,  9.38232121, 13.59056613,
        9.61942445,  8.04664886,  5.24137909,  9.12220271,  4.74468406,
       11.23201619,  6.24187044,  6.2322769 , 11.99633819,  9.48969978,
        8.95133961, 13.01728313, 11.62158554, 13.95472331,  7.85919687,
       13.26590071,  6.33517349,  9.41605684,  8.53014915,  8.0550286 ,
       11.75807252, 10.37698435,  7.96575782,  6.76661595, 12.76408562,
        5.05963217, 13.02818702,  9.68324553, 11.85743425, 12.64235359,
        9.69792513, 13.38091121,  8.72896381,  7.17789666, 13.01728313,
        8.14433542,  9.05843201, 13.55868684, 13.70893492,  6.34059021,
       10.63038565])}
{'P': tensor([[4.3339e-03, 2.8809e-01, 1.3692e-01,  ..., 3.2102e-07, 8.1864e-06,
         2.8482e-07],
        [1.0474e-05, 4.4340e-03, 3.9788e-01,  ..., 1.3259e-07, 4.7711e-06,
         3.7884e-07],
        [4.1630e-05, 8.8429e-05, 5.0203e-02,  ..., 7.4332e-07, 3.2829e-05,
         6.5551e-06],
        ...,
        [2.5844e-04, 1.7406e-04, 1.3638e-04,  ..., 3.2771e-02, 3.4846e-05,
         9.4729e-07],
        [8.6185e-03, 6.0021e-03, 3.9412e-03,  ..., 8.8225e-05, 5.3460e-02,
         2.7930e-05],
        [7.2605e-07, 4.6616e-07, 3.4436e-07,  ..., 5.0835e-06, 5.3583e-05,
         1.6568e-02]]), 'X': tensor([[ 12.5717,   1.9266,   7.4903],
        [ 12.2435,   4.0076,   8.1257],
        [ 11.5456,   6.0861,   8.1101],
        [ 11.4495,   6.3026,   6.8257],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.5966,   5.9666,   6.0083],
        [ -5.9225,  -8.5355,   2.4778],
        [  2.6078, -10.9229,   6.2869],
        [ -0.1626, -10.9796,   6.4629],
        [ -0.0410,   7.8591,   9.3457],
        [ 11.6544,   5.8273,  -3.1375],
        [  4.5325,  -3.3643,   6.7699],
        [ -5.9938,  -8.4745,   3.9735],
        [ -1.6906,   7.8860,   7.7142],
        [ -4.4520,   7.1370,  -8.7867],
        [ -6.7643,  -7.7372,  -9.6663],
        [  3.3069,   6.6089,   8.7377],
        [  1.2273,  -4.8081,   9.8129],
        [ -6.2200,  -8.2737,   5.5079],
        [ -9.4630,  -1.5145,   0.2983],
        [ -6.6895,   5.6051,   2.7899],
        [  4.3377,   5.7570,   1.3712],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.1203,  -2.7151,   6.2110],
        [  0.2571,  -4.7594,   1.5618],
        [  4.5800,   5.5034,  -3.9432],
        [  4.0457,  -3.7737,  -0.4704],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 12.3357,   3.6001,   0.7903],
        [  1.7815,  -4.7481,   4.4289],
        [  6.1579,  -0.5961,  -6.5595],
        [  2.3589,  13.7668,  -3.6501],
        [ -5.9225,  -8.5355,   2.4778],
        [  7.3592,  11.3937,   6.2407],
        [  3.2647, -10.8066,  -0.3768],
        [ -0.5709,   7.9055,   7.6089],
        [  6.6319,  11.9066,   3.5165],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.3653,   3.4184,   9.3454],
        [  5.9003,  12.3550,   2.1917],
        [ -2.4956, -10.4682,   3.5632],
        [  2.2849, -10.9655,  -7.6542],
        [  8.3718,  -8.2953,   8.4666],
        [  7.8375,  -8.7240,  -9.8913],
        [ -8.6603,   2.8250,   5.9181],
        [  5.0506,   4.9277,  -9.5094],
        [ 11.4495,   6.3026,   6.8257],
        [  1.7122,  -4.7591,  -3.6734],
        [  3.8182,  -3.9347,  -1.7961],
        [ -3.9858,  -9.8334,  -3.7323],
        [ 11.0438,   7.1180,  -5.6991],
        [  5.1203,  -2.7151,   6.2110],
        [ -8.3653,   3.4184,   9.3454],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.5240,  -2.1198,   6.2055],
        [  4.3554,   5.7393,  -9.0766],
        [  7.5064,  -8.9651,  -0.0994],
        [ 12.5650,   2.0132,   4.6272],
        [  5.3924, -10.1396,  -4.0863],
        [  1.7232,  -4.7574,   3.1270],
        [  6.2770,  -0.0379,   2.4195],
        [ -2.9497, -10.3028,  -7.3479],
        [ -3.3812, -10.1235,   7.4384],
        [  3.4604,  -4.1545,   1.8091],
        [ 11.5568,   6.0602,   9.3244],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.2038,   2.4020,  -3.1692],
        [ -9.4739,  -1.2497,   0.5483],
        [ 10.2712,   8.3707,  -4.9743],
        [  4.0421,  -3.7765,  -8.3631],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.1379,  -4.6757,   6.4981],
        [ -6.6998,   5.5954,   0.0823],
        [  2.2182,  13.8007,   5.1889],
        [  5.1787,  -2.6383,  -4.2533],
        [  8.4691,  10.4571,  -5.0650],
        [ 12.4592,  -0.9168,  -6.7571],
        [  4.0591,   6.0202,  -1.6339],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.6063,  13.0051,  -3.1456],
        [  2.6078, -10.9229,   6.2869],
        [ -5.3382,   6.6533,   7.8054],
        [ -5.8389,  -8.6056,  -3.5098],
        [ 11.1085,  -4.9204,   0.0736],
        [ 12.2692,  -1.8882,   3.9253],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -2.0445,   7.8469,  -0.5857],
        [  0.6396,  14.0772,  -6.3786],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -1.7290,   7.8825,  -3.3831],
        [ -1.6334, -10.7207,   7.1397],
        [ -6.7992,   5.4991,   9.9186],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.3549,   0.7096,   6.9198],
        [ 11.2767,  -4.6006,   0.7868],
        [  0.6952,  -4.8060,   8.8976],
        [  0.9659,  -4.8144,  -0.9594],
        [ 11.9019,  -3.1379,  -3.4534],
        [  0.7958,  -4.8109,  -8.2248],
        [  1.4749,  -4.7892,  -0.8089],
        [  3.3474,  -4.2162,   5.2455],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.1795,  -3.3623,   8.5259],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.5491,  -3.3485,   6.8412],
        [ -9.4671,  -0.5553,   0.1117]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0696, 0.1262,  ..., 1.2673, 0.2884, 1.2526],
        [0.0696, 0.0000, 0.0570,  ..., 1.3057, 0.3492, 1.3209],
        [0.1262, 0.0570, 0.0000,  ..., 1.3306, 0.3967, 1.3743],
        ...,
        [1.2673, 1.3057, 1.3306,  ..., 0.0000, 1.0147, 0.9912],
        [0.2884, 0.3492, 0.3967,  ..., 1.0147, 0.0000, 0.9883],
        [1.2526, 1.3209, 1.3743,  ..., 0.9912, 0.9883, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.71843462, 12.88270018, 13.05148826, 13.06957425, 12.86648771,
       13.04156075, 10.38897563, 11.22993218, 10.98076873,  7.85919687,
       13.03002338,  5.64464807, 10.37998078,  8.06517143,  8.41171369,
       10.27716434,  7.39004226,  4.96231532, 10.35093485,  9.58347456,
        8.7273569 ,  7.20826219,  5.23878371,  5.79562691,  4.7663526 ,
        7.15990676,  5.53255882, 12.35625694, 12.85032836,  5.0713364 ,
        6.18668463, 13.96746816, 10.38897563, 13.56367857, 11.28895489,
        7.92607342, 13.62896593, 12.62576026,  9.0368458 , 13.69162795,
       10.76154256, 11.20100795, 11.785566  , 11.72749923,  9.10945678,
        7.05627461, 13.06957425,  5.05774295,  5.48276038, 10.61047596,
       13.13888841,  5.79562691,  9.0368458 ,  9.9353921 ,  5.91677188,
        7.20482224, 11.69264243, 12.72524615, 11.48434767,  5.05989547,
        6.27715105, 10.7167285 , 10.67322527,  5.40689332, 13.0493368 ,
       13.13205357,  6.65259775,  9.55593374, 13.25017211,  5.53175275,
       12.84056604,  5.14124612,  8.72896381, 13.97779839,  5.81199422,
       13.45642379, 12.49291768,  7.26074925,  9.54154096, 13.79676654,
       11.22993218,  8.53014915, 10.39942497, 12.1494023 , 12.41366754,
       10.54658597,  8.10886021, 14.09176259, 10.52587243,  8.06990483,
       10.84437705,  8.74470218, 12.05395056,  6.39438044, 12.17901033,
        4.85603886,  4.91038884, 12.30858677,  4.8763135 ,  5.01114719,
        5.38342077, 11.72802491,  9.77588801, 12.05300943,  5.64865173,
        9.48337063])}
{'P': tensor([[8.5456e-03, 2.5245e-01, 2.1919e-01,  ..., 1.5787e-07, 6.2018e-04,
         2.2277e-02],
        [7.9984e-06, 1.0645e-02, 5.9460e-04,  ..., 1.1815e-07, 1.2097e-04,
         2.4510e-03],
        [1.3402e-05, 1.0337e-01, 1.2788e-02,  ..., 1.3858e-07, 1.2201e-04,
         2.6204e-03],
        ...,
        [6.0281e-04, 9.7931e-05, 1.0403e-04,  ..., 5.6957e-02, 6.0952e-05,
         9.0558e-05],
        [1.7737e-05, 1.7280e-04, 9.9789e-05,  ..., 1.8148e-07, 2.0559e-02,
         2.8661e-03],
        [1.0891e-04, 5.2768e-03, 2.7763e-03,  ..., 1.2952e-06, 6.3237e-03,
         1.0897e-01]]), 'X': tensor([[ 12.5650,   2.0132,   4.6272],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.4495,   6.3026,   6.8257],
        [ 10.0567,   8.6697,   4.6329],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.9843,  -9.8679,  -6.1449],
        [  4.9531,   5.0576,   5.6794],
        [  6.2902,   0.0451,   3.7116],
        [ 10.9599,  -5.1849,  -3.5526],
        [  0.8700,  -4.8132,  -7.8042],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.8745,   2.3097,   3.5600],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 10.9394,   7.3069,  -8.9365],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.6050,   2.9451,  -0.7012],
        [ -9.1907,  -3.3180,  -5.5943],
        [  3.6703,   6.3443,  -5.4401],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.3542,   0.6934,  -3.3840],
        [  4.3523, -10.5232,   8.2138],
        [ -7.4733,   4.7555,   6.2490],
        [  4.5503,   5.5359,   7.4735],
        [  4.5800,   5.5034,  -3.9432],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.9282,   2.1648,   8.4894],
        [ -8.9636,   2.0648,   8.1946],
        [  5.9026,   3.3837,  -0.6573],
        [ -9.4630,  -1.5145,   0.2983],
        [ -0.5385,   7.9037,   6.3830],
        [  6.2975,   0.0947,   5.1791],
        [  6.1372,   2.6726,  -6.7885],
        [ -7.5720,   4.6310,   2.1717],
        [  0.2480, -11.0153,  -3.0183],
        [  3.1480,  -4.3165,  -7.8002],
        [  6.1222,  -0.7261,   8.8204],
        [  6.8093,  -9.4178,   8.4244],
        [  5.0233,   4.9647,  -3.1788],
        [  1.2273,  -4.8081,   9.8129],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.6078, -10.9229,   6.2869],
        [ -1.5261,   7.8988,   9.3768],
        [  4.7156,  -3.1831,  -8.0224],
        [ -0.5739,   7.9056,   0.6382],
        [  9.4516,  -7.2501,  -1.8868],
        [ -0.8675, -10.8815,   6.8561],
        [ -7.6525,  -6.6534,   5.1721],
        [ 12.2435,   4.0076,   8.1257],
        [  8.9284,  10.0036,  -6.9210],
        [  5.3602,  -2.3808,   7.2098],
        [ -6.3252,  -8.1760,  -1.0891],
        [  7.3592,  11.3937,   6.2407],
        [  2.5620,   7.0549,   8.4825],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.4356,  -4.5942,  -5.0808],
        [ -9.4296,  -1.9458,   4.0833],
        [ -5.1398,  -9.1367,   6.2074],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.5101,  -2.1433,   9.5442],
        [ -2.2188,   7.8218,   8.6064],
        [  3.2701,   6.6338,   9.9209],
        [  4.6357,   5.4415,   7.9067],
        [ -9.3107,  -2.7697,  -3.2919],
        [  6.2036,   2.4030,  -1.2500],
        [  6.3087,   1.8053,  -1.9481],
        [ -3.9508,  -9.8515,   1.7923],
        [  1.8766,   7.3694,   1.1887],
        [ 12.6035,   0.7494,   7.1536],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -5.1046,  -9.1610,   2.3418],
        [  3.5939,   6.4029,  -1.8357],
        [  3.2226,  -4.2802,  -4.0428],
        [  4.5814, -10.4485,   2.5496],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -4.4503,   7.1378,   3.3751],
        [  3.1892, -10.8220,  -6.8409],
        [  2.8474,   6.8979,   6.6457],
        [  2.1440, -10.9811,   8.5281],
        [  3.9663,   6.1018,  -5.4104],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.4837,   2.7510,   0.4449],
        [  7.5784,  -8.9141,   5.8410],
        [  9.4925,   9.3811,   8.9842],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.2608,  -2.5256,  -9.1446],
        [  0.3354,  -4.7708,  -4.9125],
        [  4.8432,   5.1968,   1.5006],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.3583,   1.1729,   8.7835],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -4.2040,   7.2486,   2.9875],
        [ -9.4721,  -1.3097,  -7.4377],
        [  1.5653,  13.9377,   4.1773],
        [  4.4039,   5.6903,   8.7366],
        [  8.8474,  10.0868,  -8.8644],
        [ -9.3076,   0.7852,  -6.7010],
        [  0.7685,  -4.8098,  -2.9310],
        [  0.1382, -11.0073,   1.0026],
        [ 11.2939,   6.6325,  -5.8311],
        [ 11.0864,   7.0387,  -0.1238]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1329, 0.1737,  ..., 0.9101, 0.9699, 0.4938],
        [0.1329, 0.0000, 0.0582,  ..., 1.0416, 1.0424, 0.5656],
        [0.1737, 0.0582, 0.0000,  ..., 1.0705, 1.1006, 0.6238],
        ...,
        [0.9101, 1.0416, 1.0705,  ..., 0.0000, 1.0664, 0.8949],
        [0.9699, 1.0424, 1.1006,  ..., 1.0664, 0.0000, 0.4778],
        [0.4938, 0.5656, 0.6238,  ..., 0.8949, 0.4778, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.72524615, 13.0743399 , 13.04156075, 12.86648771, 13.06957425,
       13.2778378 ,  4.98461983, 11.54071859,  7.07902036,  6.29034958,
       12.12448912,  4.89121805, 10.5575601 ,  9.17016857,  9.97822473,
       13.15525803,  9.99211663,  9.09502385,  9.7712321 ,  7.32950385,
       10.91805683,  6.39187285, 11.38774149,  8.8580671 ,  7.16598273,
        7.15990676, 10.05558893,  9.18690018,  9.19837792,  6.80370254,
        9.58347456,  7.92200342,  6.29821673,  6.69389431,  8.87587457,
       11.018084  ,  5.34250225,  6.16512856, 11.62158554,  7.06272256,
        4.96231532,  8.83341608, 11.22993218,  8.04483495,  5.68942572,
        7.92644746, 11.9120321 , 10.91601655, 10.14044489, 12.88270018,
       13.40850328,  5.86519525, 10.33712258, 13.56367857,  7.50564223,
       11.97949273,  9.96391045,  5.19986797,  9.6282763 , 10.48315436,
       12.1494023 ,  5.91222346,  8.1303945 ,  7.39599986,  7.14837514,
        9.71392179,  6.652742  ,  6.5618968 , 10.61417181,  7.60463349,
       12.62576026,  9.5014146 , 10.48721382,  7.34250948,  5.35774573,
       11.4088043 ,  6.02588517,  9.71392179,  8.41148722, 11.28215429,
        7.46248947, 11.18839169,  7.27759233, 13.96905756, 12.78327037,
       11.70016217, 13.34587027,  5.83560009,  5.83560009,  4.78258092,
        7.10378919, 12.3828957 , 10.32715715,  6.46560181,  7.78980299,
        9.94285511,  8.37954853,  9.562174  , 14.02532758,  7.19534146,
       13.41713661,  9.34061938,  4.87083348, 11.00812996, 13.09737053,
       13.13205357])}
{'P': tensor([[4.4013e-03, 2.4479e-01, 2.2674e-01,  ..., 1.2166e-05, 1.7749e-08,
         5.7611e-07],
        [5.9568e-05, 4.5425e-02, 1.1291e-01,  ..., 8.5868e-05, 2.0781e-07,
         9.9910e-06],
        [5.1523e-05, 8.8405e-03, 4.2210e-02,  ..., 1.0814e-04, 2.3405e-07,
         1.2329e-05],
        ...,
        [1.3431e-05, 2.2681e-05, 3.0378e-05,  ..., 6.0307e-02, 5.7642e-05,
         4.9077e-03],
        [5.5734e-07, 2.5773e-07, 3.0029e-07,  ..., 1.4235e-05, 2.4972e-02,
         7.7011e-06],
        [3.2585e-07, 3.0743e-07, 3.6686e-07,  ..., 4.5059e-05, 6.3581e-03,
         5.2129e-02]]), 'X': tensor([[ 12.5154,   2.5072,  -9.3274],
        [ 11.7794,   5.5086,  -9.6929],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.3609,   3.4769,  -8.5762],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.2902,   0.0451,   3.7116],
        [ -4.7334,   6.9982,  -0.4392],
        [ 11.5568,   6.0602,   9.3244],
        [ -3.5173, -10.0623,   4.8280],
        [  1.7122,  -4.7591,  -3.6734],
        [ -6.9838,  -7.4965,   0.9796],
        [  0.9788,  -4.8145,  -3.8661],
        [  4.0591,   6.0202,  -1.6339],
        [  2.1625,  -4.6697,   7.0658],
        [  2.4149,  -4.6005,  -4.3339],
        [ -7.4808,   4.7462,   2.0266],
        [  6.0788,  -0.8710,  -6.8719],
        [  9.9750,  -6.6338,  -1.2440],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.3535,  -5.5118,   9.8399],
        [ -8.5900,   2.9767,   7.4811],
        [  0.1931,  -4.7491,   7.4385],
        [ -3.5520,   7.4976,   9.1522],
        [  5.6672,  -1.8631,   8.1020],
        [  3.3633,  13.4794,   0.5193],
        [  0.9445,   7.6736,   0.5095],
        [ -0.5739,   7.9056,   0.6382],
        [ -1.7290,   7.8825,  -3.3831],
        [  6.3515,   0.6418,  -6.5823],
        [  0.5600, -11.0321,  -0.8660],
        [  5.5158,  -2.1337,   3.5246],
        [ -1.5451,   7.8975,   7.6104],
        [  8.5198,  10.4091,  -8.6486],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -7.4602,   4.7718,  -4.6277],
        [ -9.3444,   0.5821,   6.9101],
        [  2.0899,  -4.6870,  -7.4260],
        [ -3.2736, -10.1703,   3.8712],
        [ -4.7334,   6.9982,  -0.4392],
        [  0.1041, -11.0045,   2.4311],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.3804,  -2.3504,   7.2744],
        [  5.0617,  12.7957,   8.3544],
        [  2.0729,   7.2878,   7.0880],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.8994,   6.8675,  -9.5987],
        [  6.6319,  11.9066,   3.5165],
        [  9.6587,   9.1817,  -0.8212],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -6.1914,  -8.2997,   8.5380],
        [  9.5018,  -7.1945,  -1.1892],
        [ -8.6603,   2.8250,   5.9181],
        [  7.5784,  -8.9141,   5.8410],
        [  2.6451,  -4.5250,  -0.5565],
        [  6.5428,  -9.5728,  -2.9897],
        [  6.8704,  -9.3809,   7.5769],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.6952,  -4.8060,   8.8976],
        [  5.8979,   3.3958,  -3.8713],
        [  5.3924, -10.1396,  -4.0863],
        [  2.0937,  13.8293,  -2.3533],
        [  2.6970,   6.9826,   9.7350],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.9410,   6.8428,   5.4458],
        [ -6.5295,  -7.9785,  -7.4429],
        [  7.0358,  -9.2784,  -9.8479],
        [  4.6992,   5.3690,   3.0935],
        [ -9.4721,  -1.3097,  -7.4377],
        [  1.3339,   7.5627,  -0.9034],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 11.5456,   6.0861,   8.1101],
        [  0.3354,  -4.7708,  -4.9125],
        [ -6.4620,  -8.0449,   6.2719],
        [ -8.6801,   2.7809,  -0.8443],
        [ -4.0409,   7.3169,  -3.5779],
        [ 12.1379,   4.4125,  -3.8833],
        [ -7.8280,   4.2848,  -7.1955],
        [ -4.1296,   7.2803,  -2.7194],
        [ 12.6059,   0.9597,  -6.3264],
        [ 10.4758,  -5.9539,   1.3667],
        [ -2.0445,   7.8469,  -0.5857],
        [  6.2905,   1.9366,   2.1169],
        [  6.0788,  -0.8710,  -6.8719],
        [ -7.9925,  -6.1423,   2.8990],
        [  2.5318,  13.7231,   6.3692],
        [ -9.3087,   0.7792,  -7.5587],
        [ -0.5709,   7.9055,   7.6089],
        [  3.2990,  -4.2415,  -7.4857],
        [ 12.5717,   1.9266,   7.4903],
        [  4.0587, -10.6112,   6.2273],
        [  6.1737,   2.5305,   3.9049],
        [  7.7794,  11.0629,   8.9069],
        [  4.4039,   5.6903,   8.7366],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.3367,   7.1676,  -9.5263],
        [ -4.5778,  -9.5013,  -5.0667],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.1379,   4.4125,  -3.8833],
        [  9.6614,   9.1784,  -1.2087],
        [  4.9531,   5.0576,   5.6794],
        [ -8.0330,  -6.0765,   6.7945],
        [ -6.3489,   5.9117,   6.5161]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0366, 0.0372,  ..., 1.0739, 1.6090, 1.3351],
        [0.0366, 0.0000, 0.0022,  ..., 1.0865, 1.6372, 1.3586],
        [0.0372, 0.0022, 0.0000,  ..., 1.0848, 1.6363, 1.3574],
        ...,
        [1.0739, 1.0865, 1.0848,  ..., 0.0000, 0.8752, 0.4919],
        [1.6090, 1.6372, 1.6363,  ..., 0.8752, 0.0000, 0.3834],
        [1.3351, 1.3586, 1.3574,  ..., 0.4919, 0.3834, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.76408562, 13.00380624, 13.01728313, 12.84056604, 12.889695  ,
       12.9089559 ,  5.1224079 ,  6.29034958,  8.4486771 , 13.0493368 ,
       10.65929968,  5.05774295, 10.24557047,  4.91296205,  7.26074925,
        5.14607014,  5.19577146,  8.859416  ,  6.14086582, 11.97949273,
       10.06286074, 10.00802627,  9.09119351,  4.75303127,  8.29640613,
        5.96556248, 13.89264822,  7.73151309,  7.92644746,  8.06990483,
        6.383897  , 11.04629471,  5.91408504,  8.04718433, 13.45124917,
       10.33712258,  8.85572292,  9.36256096,  5.13182506, 10.68416401,
        8.4486771 , 11.0050379 ,  9.69629   ,  9.7751189 ,  5.87132718,
       13.76047999,  7.57685908,  9.51074763,  7.4544866 , 13.62896593,
       13.32645552, 10.39942497, 10.35464916, 11.9182956 ,  9.10945678,
       11.70016217,  5.24137909, 11.59512194, 11.62770332,  8.62631801,
        4.85603886,  6.80559669, 11.48434767, 13.98691426,  7.48538905,
       12.10942412,  7.44805901, 10.30972193, 11.64436684,  7.13501264,
        9.562174  ,  7.67939546, 12.64235359, 12.49291768, 13.05148826,
        4.78258092, 10.31886066,  9.11473391,  8.35856487, 12.91505111,
        8.92395381,  8.36995953, 12.64235359, 12.0495521 ,  8.10886021,
        6.58184474,  6.14086582, 10.08001894, 13.95472331,  9.34127013,
        7.92607342,  5.37343173, 12.71843462, 11.36089355,  6.67218687,
       13.52430176,  7.19534146,  9.9353921 ,  7.53884104, 10.54658597,
       12.5769659 , 12.91505111, 13.32613445,  7.07902036, 10.07238463,
        8.67502348])}
{'P': tensor([[4.5526e-03, 2.1296e-01, 1.6096e-01,  ..., 7.6693e-07, 4.6723e-08,
         3.7368e-06],
        [3.2348e-05, 8.3042e-03, 7.6707e-02,  ..., 6.8952e-07, 7.7294e-08,
         4.8699e-06],
        [5.2613e-06, 4.3473e-05, 3.9287e-03,  ..., 2.0873e-07, 5.6072e-08,
         1.6324e-06],
        ...,
        [4.0216e-02, 1.6575e-02, 1.9109e-02,  ..., 7.3134e-02, 2.4634e-06,
         4.0303e-04],
        [1.3834e-05, 6.9531e-06, 6.6144e-06,  ..., 1.0176e-03, 9.0583e-02,
         5.8537e-04],
        [1.4876e-02, 1.0586e-02, 6.4306e-03,  ..., 7.0640e-03, 6.8400e-05,
         2.0195e-01]]), 'X': tensor([[ 12.4996,   2.6335,  -6.3166],
        [ 12.1504,   4.3674,  -7.2998],
        [ 11.6019,   5.9540,  -5.1715],
        [ 11.2939,   6.6325,  -5.8311],
        [ 11.0438,   7.1180,  -5.6991],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.7485,  -9.9811,  -3.2355],
        [  0.2571,  -4.7594,   1.5618],
        [ -6.6895,   5.6051,   2.7899],
        [  4.5503,   5.5359,   7.4735],
        [  9.0051,   9.9234,   5.4411],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.9531,   5.0576,   5.6794],
        [ -5.9987,   6.1949,  -6.2114],
        [ -9.4697,  -0.6157,  -6.8472],
        [  2.5318,  13.7231,   6.3692],
        [ -8.1666,   3.7653,   3.6075],
        [  5.7845,   3.6652,   0.9175],
        [  3.3839,  -4.1966,   8.3574],
        [  6.1737,   2.5305,   3.9049],
        [ -3.0008, -10.2827,  -8.9480],
        [  6.3305,   1.6119,   1.1538],
        [ -8.8745,   2.3097,   3.5600],
        [  2.1379,  -4.6757,   6.4981],
        [  0.3715,  -4.7756,   0.2312],
        [  3.2809,  13.5061,  -4.0199],
        [  1.7228,  -4.7575,   8.3396],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.4867,  12.5816,   2.9549],
        [  8.5716,  10.3596,   6.8818],
        [ -9.3411,  -2.6004,  -2.7062],
        [  3.9770, -10.6341,   8.0783],
        [ 10.7566,   7.6212,  -2.3279],
        [ -3.5173,   7.5091,  -8.9585],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.2576,   2.1359,  -3.3398],
        [ 12.6000,   1.3839,  -3.2992],
        [  0.9809,  -4.8145,   6.8706],
        [  5.9003,  12.3550,   2.1917],
        [  6.3589,   0.8239,  -8.4230],
        [ 12.4996,   2.6335,  -6.3166],
        [ 11.2939,   6.6325,  -5.8311],
        [  0.2480, -11.0153,  -3.0183],
        [  2.5974,   7.0363,  -7.7810],
        [  2.3589,  13.7668,  -3.6501],
        [  3.4009,  -4.1874,  -5.9751],
        [  5.1292,   4.8183,  -5.7997],
        [  4.5800,   5.5034,  -3.9432],
        [ -8.6050,   2.9451,  -0.7012],
        [  5.8889,  -1.3916,   8.8993],
        [  9.5261,  -7.1675,   1.4206],
        [  4.0457,  -3.7737,  -0.4704],
        [  1.3103,  -4.8032,  -0.9300],
        [ -6.0195,   6.1789,  -2.9496],
        [  2.1771, -10.9776,  -3.2007],
        [  2.2373,  -4.6506,  -1.7334],
        [ 11.2209,  -4.7092,   2.6117],
        [ -9.4501,  -1.7133,   2.2980],
        [ 10.8218,  -5.4177,   6.8573],
        [ -9.4196,   0.0478,   9.4140],
        [  2.1504,  -4.6726,   8.7908],
        [ -9.0604,   1.7707,   1.1066],
        [  6.2242,  -0.3171,   7.4978],
        [  9.1409,  -7.5789,   1.5814],
        [  6.0345,   3.0175,  -1.4424],
        [  0.1206,  -4.7363,   8.7103],
        [  5.9843,  -9.8679,  -6.1449],
        [ -0.1626, -10.9796,   6.4629],
        [  5.9026,   3.3837,  -0.6573],
        [ -1.8823, -10.6558,  -3.8952],
        [  3.8111, -10.6787,  -4.6539],
        [ 11.5198,  -4.0923,   5.5088],
        [  3.8182,  -3.9347,  -1.7961],
        [ -5.2752,  -9.0409,  -8.2568],
        [  5.4242, -10.1261,   3.9595],
        [ -3.3999, -10.1152,  -1.2088],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -4.4520,   7.1370,  -8.7867],
        [  0.1047, -11.0046,   0.5188],
        [ -9.4731,  -0.7120,  -6.0271],
        [  1.7122,  -4.7591,  -3.6734],
        [  5.8201,   3.5841,  -8.0609],
        [ -3.3116, -10.1539,   7.5919],
        [  5.1142,  -2.7230,  -4.1966],
        [ 10.5955,   7.8824,  -3.7006],
        [ -2.6069, -10.4298,  -6.1995],
        [  3.3943,   6.5483,   1.0790],
        [  8.1528,  10.7452,   9.1493],
        [ 11.9122,   5.1397,  -1.5498],
        [  0.9659,  -4.8144,  -0.9594],
        [  5.8275,  -1.5334,   2.0124],
        [ -5.3382,   6.6533,   7.8054],
        [ -9.4157,   0.0821,  -5.0025],
        [  6.2895,   0.0405,   5.7652],
        [ -1.9600,   7.8577,   6.9121],
        [  6.3549,   0.7096,   6.9198],
        [ -1.5408,   7.8978,   5.3470],
        [  3.5939,   6.4029,  -1.8357],
        [  6.3122,  -9.6994,  -6.3282],
        [  0.6470, -11.0352,  -1.1931],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -6.8618,  -7.6322,   3.7027],
        [  3.2990,  -4.2415,  -7.4857]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0638, 0.0669,  ..., 0.5705, 1.3709, 0.2508],
        [0.0638, 0.0000, 0.0936,  ..., 0.6341, 1.4333, 0.3111],
        [0.0669, 0.0936, 0.0000,  ..., 0.5667, 1.3855, 0.2812],
        ...,
        [0.5705, 0.6341, 0.5667,  ..., 0.0000, 0.8464, 0.3654],
        [1.3709, 1.4333, 1.3855,  ..., 0.8464, 0.0000, 1.1239],
        [0.2508, 0.3111, 0.2812,  ..., 0.3654, 1.1239, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.77402351, 12.91143491, 13.04051744, 13.09737053, 13.13888841,
       13.09426142,  9.75200848, 11.51811603,  4.7663526 ,  8.7273569 ,
        7.16598273, 13.40024224,  9.75401449,  7.07902036,  8.62329466,
        9.48969978, 13.95472331,  8.99276564,  6.84796202,  5.39098669,
        6.67218687, 10.71162681,  6.53250937,  9.17016857,  5.14124612,
        4.79003361, 13.89886469,  5.05981914, 10.61047596, 13.72594022,
       13.44593519,  9.69629   , 11.35345345, 13.18278606,  8.29203356,
       12.07225469,  6.61211224, 12.67576792,  4.91336994, 13.69162795,
        6.41202986, 12.77402351, 13.09737053, 11.018084  ,  7.50035355,
       13.96746816,  5.39450305,  7.03733775,  7.15990676,  9.09502385,
        6.05113217, 11.92133129,  5.53255882,  4.978704  ,  8.62631801,
       11.1913598 ,  5.16077764, 12.16901615,  9.60412735, 12.10222204,
        9.41970031,  5.14371005,  9.23178204,  6.2322769 , 11.87412439,
        6.74686071,  4.73784222, 11.54071859, 10.98076873,  6.80370254,
       10.8207297 , 11.33837396, 12.22503754,  5.48276038, 10.46739073,
       11.48734978, 10.67132524,  7.53063885,  9.88571713,  9.93553424,
        8.41171369, 11.00508681,  9.4997987 ,  5.05774295,  6.83515201,
       10.68031351,  5.79391825, 13.20597932, 10.7506423 ,  7.37576154,
       13.48809595, 12.97370371,  4.91038884,  6.02588517,  8.53014915,
        9.41605684,  6.28961988,  8.09842704,  6.39438044,  8.04664886,
        7.34250948, 11.57249984, 11.05413545, 12.00995898, 10.26327895,
        5.37343173])}
{'P': tensor([[3.6183e-03, 3.5866e-01, 1.3726e-01,  ..., 2.3966e-06, 2.1825e-04,
         5.0185e-07],
        [1.9712e-05, 4.2678e-03, 2.5628e-01,  ..., 3.3326e-06, 3.3531e-04,
         4.3821e-07],
        [9.2892e-06, 1.2428e-05, 9.6946e-03,  ..., 1.5978e-05, 1.2636e-03,
         3.9045e-07],
        ...,
        [2.9848e-06, 2.7438e-06, 2.4473e-06,  ..., 9.8605e-02, 2.0485e-05,
         5.3764e-06],
        [6.9724e-05, 8.6008e-05, 1.8222e-04,  ..., 1.6953e-03, 2.6996e-01,
         3.2458e-05],
        [5.1932e-02, 4.4144e-02, 1.5259e-02,  ..., 3.5713e-06, 9.0799e-04,
         8.4071e-02]]), 'X': tensor([[ 12.4837,   2.7510,   0.4449],
        [ 12.3357,   3.6001,   0.7903],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.0864,   7.0387,  -0.1238],
        [ 11.1469,   6.9237,   1.6160],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.2883,   0.0325,   5.2692],
        [  1.5653,  13.9377,   4.1773],
        [  6.1730,  -0.5375,  -8.8861],
        [ -3.0644,   7.6439,  -1.4322],
        [  5.3858,   4.4260,  -9.6769],
        [ -1.9646,   7.8571,   3.3817],
        [  6.9284,  11.7061,  -0.3732],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.4624,  -0.4633,  -7.3918],
        [  5.0506,   4.9277,  -9.5094],
        [ -9.4505,  -1.7081,  -0.7039],
        [  0.2480, -11.0153,  -3.0183],
        [  2.5620,   7.0549,   8.4825],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.6229,   2.9068,   2.4429],
        [  6.2242,  -0.3171,   7.4978],
        [ 10.4758,  -5.9539,   1.3667],
        [  4.8653,   5.1694,  -4.0823],
        [  3.3474,  -4.2162,   5.2455],
        [  9.0004,  -7.7195,   4.6945],
        [  2.7590,  13.6621,   9.4496],
        [  3.6249, -10.7255,   5.6605],
        [  9.6816,   9.1536,  -5.8543],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.0031,  12.2958,  -6.5992],
        [  5.1886,  -2.6250,   6.4677],
        [  6.1752,   2.5242,   3.5874],
        [  5.5101,  -2.1433,   9.5442],
        [  6.0047,   3.1062,  -6.7914],
        [ -7.9925,  -6.1423,   2.8990],
        [  0.6363, -11.0348,   6.1414],
        [ -5.9938,  -8.4745,   3.9735],
        [  4.8432,   5.1968,   1.5006],
        [ 11.1085,  -4.9204,   0.0736],
        [  2.6613,   7.0021,   4.5783],
        [  3.4604,  -4.1545,   1.8091],
        [ -1.6334, -10.7207,   7.1397],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.1519, -10.2387,   9.8191],
        [ -8.8934,   2.2597,   3.5831],
        [ -2.2508, -10.5479,  -2.7235],
        [  2.4639, -10.9431,  -7.3718],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.4505,  -1.7081,  -0.7039],
        [  1.6232,  -4.7717,  -2.5655],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.8704,  -9.3809,   7.5769],
        [ 10.9717,  -5.1646,  -6.9039],
        [  6.0345,   3.0175,  -1.4424],
        [ -6.4620,  -8.0449,   6.2719],
        [  7.3724,  -9.0577,  -8.9843],
        [ 11.0275,  -5.0666,   8.4154],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.5663,   7.0526,  -6.8769],
        [ -5.1398,  -9.1367,   6.2074],
        [  5.5101,  -2.1433,   9.5442],
        [  6.1685,  12.1979,  -6.2473],
        [ 12.5770,   1.8512,  -0.6578],
        [  0.1497,   7.8334,  -5.4412],
        [  7.5784,  -8.9141,   5.8410],
        [  7.4132,  11.3526,   0.3322],
        [  5.2608,  12.6977,   4.9166],
        [  4.5340,  -3.3629,   8.9652],
        [ 10.8670,  -5.3428,  -2.8637],
        [  8.7715,  -7.9390,   5.2900],
        [ -8.9282,   2.1648,   8.4894],
        [  2.8474,   6.8979,   6.6457],
        [ -6.6998,   5.5954,   0.0823],
        [  1.4306,  13.9620,  -0.2075],
        [  1.7122,  -4.7591,  -3.6734],
        [  7.5309,  11.2617,   7.2656],
        [ -8.7544,   2.6096,   3.3146],
        [ 11.0864,   7.0387,  -0.1238],
        [  2.1440, -10.9811,   8.5281],
        [  1.6308,  13.9254,  -2.0984],
        [  8.1217,  -8.5024,  -0.6149],
        [  8.2422,  -8.4041,  -1.4962],
        [  2.0547,   7.2957,  -1.9343],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.6681,  -4.8556,   9.2518],
        [  9.0424,  -7.6779,  -7.5286],
        [  0.9392,  -4.8143,  -3.4649],
        [ -9.3464,   0.5708,  -1.6397],
        [  9.6662,   9.1725,   8.1056],
        [ -8.0588,  -6.0341,  -2.3006],
        [  7.4132,  11.3526,   0.3322],
        [ -6.6895,   5.6051,   2.7899],
        [  2.1440, -10.9811,   8.5281],
        [  1.1274,  14.0116,   8.1753],
        [  1.2728, -11.0372,   9.8121],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -6.7745,  -7.7264,   8.4684],
        [ 10.7566,   7.6212,  -2.3279],
        [  0.7275,  14.0667,  -0.9815],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.3103,  -4.8032,  -0.9300],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.0729,   7.2878,   7.0880],
        [  9.9058,  -6.7203,   5.6928]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0358, 0.1045,  ..., 0.9423, 0.5692, 0.6244],
        [0.0358, 0.0000, 0.0688,  ..., 0.9760, 0.5635, 0.6168],
        [0.1045, 0.0688, 0.0000,  ..., 1.0427, 0.5653, 0.6144],
        ...,
        [0.9423, 0.9760, 1.0427,  ..., 0.0000, 1.0161, 1.0672],
        [0.5692, 0.5635, 0.5653,  ..., 1.0161, 0.0000, 0.0627],
        [0.6244, 0.6168, 0.6144,  ..., 1.0672, 0.0627, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.78327037, 12.85032836, 13.08414689, 13.13205357, 13.12218013,
       13.1471313 ,  6.28835101, 14.02532758,  6.19633139,  8.23525433,
        6.97106912,  8.09899874, 13.60275211,  9.60359158,  9.47369628,
        7.05627461,  9.60359158, 11.018084  ,  7.50564223,  9.73579738,
        9.99211663,  9.09963894,  6.2322769 , 12.0495521 ,  7.09887557,
        5.38342077, 11.85743425, 13.93790235, 11.32149259, 13.32374056,
        5.18277858, 13.6829843 ,  5.81481044,  6.67121748,  5.91222346,
        6.76056625, 10.08001894, 11.05317597, 10.37998078,  7.10378919,
       12.1494023 ,  7.49076899,  5.40689332, 10.84437705, 12.17901033,
       11.46176259,  9.17595899, 10.78533838, 11.21703654,  7.83534987,
        9.60359158,  5.04028538,  8.36995953, 11.62770332, 12.12641969,
        6.74686071, 10.31886066, 11.67875855, 12.13568598,  9.78147542,
        7.50500064, 10.48315436,  5.91222346, 13.66896739, 12.7125079 ,
        7.83486943, 11.70016217, 13.55868684, 13.744387  ,  5.64500512,
       12.10942412, 11.83075005,  9.18690018,  7.46248947,  8.72896381,
       14.03505624,  5.05774295, 13.54774355,  9.13507321, 13.13205357,
       11.18839169, 14.02058621, 11.75807252, 11.77124373,  7.57946459,
        9.85330646,  9.9353921 , 11.86239961,  4.90505566,  9.36378438,
       13.32557133, 10.0674887 , 13.55868684,  8.7273569 , 11.18839169,
       14.05687815, 11.11038551,  9.97964013, 10.27572673, 13.18278606,
       14.08549802, 13.12299145,  4.978704  ,  8.89458273,  7.57685908,
       11.9702762 ])}
{'P': tensor([[4.2551e-03, 2.2840e-01, 1.6822e-01,  ..., 5.1833e-06, 1.5027e-05,
         2.0803e-08],
        [4.8364e-06, 3.9591e-03, 1.5764e-01,  ..., 1.1425e-05, 3.6977e-05,
         1.2628e-08],
        [3.3506e-06, 4.0698e-05, 3.7431e-03,  ..., 1.4011e-05, 3.9873e-05,
         9.5769e-09],
        ...,
        [1.8956e-06, 1.6983e-06, 1.7425e-06,  ..., 4.9194e-02, 6.9714e-04,
         3.0610e-06],
        [4.0526e-06, 4.5482e-06, 4.5659e-06,  ..., 2.0011e-02, 9.5049e-02,
         1.7816e-05],
        [8.1004e-05, 3.6406e-05, 2.3745e-05,  ..., 5.5726e-07, 3.1781e-06,
         8.2014e-02]]), 'X': tensor([[ 12.4691,   2.8525,  -5.5110],
        [ 11.6019,   5.9540,  -5.1715],
        [ 11.2939,   6.6325,  -5.8311],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.3118,   6.5957,  -4.5667],
        [ 11.0438,   7.1180,  -5.6991],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -8.9033,   2.2329,  -8.8912],
        [ 12.0582,  -2.6640,   0.7132],
        [ -3.4488, -10.0934,   2.5195],
        [  4.0421,  -3.7765,  -8.3631],
        [  6.0054,  -1.0902,   8.3079],
        [  2.6078, -10.9229,   6.2869],
        [  5.6095,  -1.9702,  -7.8233],
        [ -9.2508,   1.0572,   2.9978],
        [  0.8700,  -4.8132,  -7.8042],
        [  5.6248,   3.9984,   8.4933],
        [  6.3583,   1.1729,   8.7835],
        [ -6.4368,   5.8356,   3.3014],
        [ -8.8058,   2.4850,  -1.7425],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -3.0644,   7.6439,  -1.4322],
        [ -1.2454,   7.9127,  -0.5715],
        [  0.7685,  -4.8098,  -2.9310],
        [  2.2849, -10.9655,  -7.6542],
        [  0.9945,  -4.8144,  -0.1897],
        [  5.4732,  -2.2041,  -4.3830],
        [ -4.2271,   7.2387,  -1.8316],
        [ -9.2508,   1.0572,   2.9978],
        [  2.1477,   7.2550,   8.9688],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.3705,   0.4208,  -3.5815],
        [  7.0358,  -9.2784,  -9.8479],
        [ -1.5261,   7.8988,   9.3768],
        [  4.9452, -10.3187,  -7.8621],
        [  3.9591, -10.6391,  -6.2557],
        [  3.2647, -10.8066,  -0.3768],
        [  4.4290,  -3.4596,   1.4181],
        [  3.4604,  -4.1545,   1.8091],
        [  5.1787,  -2.6383,  -4.2533],
        [ -9.4697,  -0.6157,  -6.8472],
        [  6.3048,   0.1469,  -9.1691],
        [  2.0547,   7.2957,  -1.9343],
        [ -9.4756,  -0.8166,  -9.6021],
        [  3.2647, -10.8066,  -0.3768],
        [  2.9410,   6.8428,   5.4458],
        [ -4.6942,  -9.4299,  -9.1846],
        [  4.4290,  -3.4596,   1.4181],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.6470, -11.0352,  -1.1931],
        [  5.9548,  -1.2273,  -2.6520],
        [ 10.1983,  -6.3430,  -0.3200],
        [  4.9957,  -2.8705,   6.5978],
        [ -8.1426,   3.8048,   9.3542],
        [  6.6319,  11.9066,   3.5165],
        [  1.3103,  -4.8032,  -0.9300],
        [ -3.5520,   7.4976,   9.1522],
        [  2.9260,  -4.4162,   8.3808],
        [ -9.4283,  -1.9585,  -0.9129],
        [  9.3607,   9.5337,  -4.5412],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.5717,   1.9266,   7.4903],
        [  7.9141,  -8.6656,   9.6948],
        [ -6.3306,   5.9272,   5.6935],
        [ 11.9808,   4.9340,  -3.3715],
        [  5.7151,  -1.7700,   3.1542],
        [  2.2644,  -4.6434,   9.7508],
        [  0.1532,  -4.7422,  -7.0145],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.0449,  13.8402,  -7.1642],
        [  4.5767,   5.5071,   3.9069],
        [  1.7659,   7.4127,   4.7871],
        [  3.9895,   6.0816,  -6.5235],
        [ 11.6634,   5.8050,   9.1590],
        [  4.2275,  -3.6320,  -0.2714],
        [ -5.5525,   6.5146,  -4.4984],
        [  6.3549,   0.7096,   6.9198],
        [  4.8653,   5.1694,  -4.0823],
        [  5.7485,  -9.9811,  -3.2355],
        [ 11.7794,   5.5086,  -9.6929],
        [  4.6063,  13.0051,  -3.1456],
        [ -8.8835,  -4.3178,   9.3080],
        [  1.8721,  -4.7322,   0.5524],
        [  6.1730,  -0.5375,  -8.8861],
        [  8.6271,  -8.0715,  -7.4813],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.3739,   0.3982,  -4.2564],
        [  3.3744,   6.5623,  -6.3636],
        [ 12.5199,  -0.4711,   3.0436],
        [ -1.8788, -10.6567,   3.7068],
        [  1.6232,  -4.7717,  -2.5655],
        [ 11.7349,  -3.5838,   2.1644],
        [ -0.0410,   7.8591,   9.3457],
        [  9.3836,   9.5075,  -5.9278],
        [  6.3515,   0.6418,  -6.5823],
        [  0.5600, -11.0321,  -0.8660],
        [  0.2571,  -4.7594,   1.5618],
        [ -8.9606,  -4.0994,   1.7886],
        [  6.0047,   3.1062,  -6.7914],
        [  5.6202,  -1.9508,   0.7471],
        [ 11.2571,   6.7072,   8.3693],
        [  5.5424,  -2.0885,   7.2487],
        [ -7.3333,   4.9246,  -7.9212],
        [ -5.3736,   6.6310,  -1.5174],
        [  2.5636, -10.9293,   9.6519]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0516, 0.0781,  ..., 0.8717, 0.7423, 1.3133],
        [0.0516, 0.0000, 0.0432,  ..., 0.9233, 0.7889, 1.3412],
        [0.0781, 0.0432, 0.0000,  ..., 0.9383, 0.8201, 1.3833],
        ...,
        [0.8717, 0.9233, 0.9383,  ..., 0.0000, 0.4070, 1.1539],
        [0.7423, 0.7889, 0.8201,  ..., 0.4070, 0.0000, 0.7888],
        [1.3133, 1.3412, 1.3833,  ..., 1.1539, 0.7888, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.79126785, 13.04051744, 13.09737053, 13.04217487, 13.09426142,
       13.13888841, 12.30858677,  9.17905491, 12.3489324 , 10.66631784,
        5.53175275,  6.10360073, 11.22993218,  5.94541803,  9.3109884 ,
        4.89121805,  6.90115901,  6.46560181,  8.68832398,  9.14973319,
       12.30858677,  8.23525433,  8.01009083,  4.87083348, 11.20100795,
        4.9160962 ,  5.90035324,  8.38252937,  9.3109884 ,  7.56617037,
       12.04924159,  9.37989631, 11.64436684,  8.04483495, 11.4424816 ,
       11.35182589, 11.28895489,  5.62005629,  5.40689332,  5.81199422,
        9.48969978,  6.30647746,  7.57946459,  9.51074763, 11.28895489,
        7.44805901, 10.53368644,  5.62005629, 10.52357479, 11.05413545,
        6.07992623, 12.00995898,  5.76165934,  8.98765644, 13.62896593,
        4.978704  ,  8.29640613,  5.29755487,  9.62958973, 13.36092624,
        9.50654967, 12.889695  , 12.71843462, 11.73567759,  8.67227386,
       12.95702647,  5.98285319,  5.16611782,  4.74468406,  5.35774573,
       13.99048094,  7.16058587,  7.62011734,  7.27340834, 13.02818702,
        5.57341623,  8.55982105,  6.39438044,  7.09887557, 11.51811603,
       13.00380624, 13.79676654,  9.87720102,  5.08910491,  6.19633139,
       11.81423161,  9.71392179,  9.38232121,  7.37902766, 12.52875793,
       10.82106505,  5.04028538, 12.26997165,  7.85919687, 13.35832725,
        6.383897  , 11.04629471,  4.7663526 ,  9.85384623,  6.76056625,
        5.94909709, 13.103709  ,  5.92281142,  8.83341608,  8.53501777,
       11.22596994])}
{'P': tensor([[2.5028e-03, 1.9602e-01, 1.9454e-01,  ..., 4.2772e-08, 8.8503e-09,
         1.9602e-01],
        [1.6536e-05, 9.0972e-03, 2.9744e-05,  ..., 1.9641e-07, 3.6446e-08,
         9.0972e-03],
        [2.1093e-05, 2.0267e-01, 3.0390e-03,  ..., 6.0554e-08, 1.2438e-08,
         2.0267e-01],
        ...,
        [1.0614e-06, 6.5245e-07, 1.0131e-06,  ..., 5.0031e-02, 1.8096e-02,
         6.5245e-07],
        [5.5291e-07, 3.2663e-07, 5.3836e-07,  ..., 1.6476e-03, 6.5315e-02,
         3.2663e-07],
        [1.6536e-05, 9.0972e-03, 2.9744e-05,  ..., 1.9641e-07, 3.6446e-08,
         9.0972e-03]]), 'X': tensor([[ 12.3609,   3.4769,  -8.5762],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.2218,   4.0953,  -8.2072],
        [ 12.1588,   4.3364,  -8.0867],
        [ 11.7794,   5.5086,  -9.6929],
        [ 11.0373,   7.1299,  -8.5895],
        [ 11.5568,   6.0602,   9.3244],
        [ 11.4703,  -4.2006,   7.4086],
        [ -9.1749,  -3.3803,   6.4629],
        [ -9.2687,  -2.9794,   0.6436],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.5506,  -5.1154,   6.3423],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.3305,   1.6119,   1.1538],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.0961,   1.6529,   3.1553],
        [  1.8320,   7.3871,   5.1452],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.8788, -10.6567,   3.7068],
        [  4.9957,  -2.8705,   6.5978],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.2911,  -2.4824,  -3.6877],
        [ -3.3999, -10.1152,  -1.2088],
        [ 11.7349,  -3.5838,   2.1644],
        [ 10.8670,  -5.3428,  -2.8637],
        [  3.6703,   6.3443,  -5.4401],
        [  5.2911,  -2.4824,  -3.6877],
        [  3.3069,   6.6089,   8.7377],
        [ -8.1426,   3.8048,   9.3542],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.4296,  -1.9458,   4.0833],
        [  3.6817,   6.3354,   6.1233],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.9003,  12.3550,   2.1917],
        [  4.3523, -10.5232,   8.2138],
        [  6.3551,   0.7149,   1.3344],
        [  3.1271, -10.8343,   2.6442],
        [ 11.9695,  -2.9407,  -6.3498],
        [  3.4920,  -4.1366,  -8.3955],
        [ -3.2736, -10.1703,   3.8712],
        [  0.1677,  -4.7447,  -6.1529],
        [  2.9461,  -4.4077,  -0.4831],
        [  2.4356,  -4.5942,  -5.0808],
        [  1.3705,  13.9723,   7.9266],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.3602,  -2.3808,   7.2098],
        [ -7.5603,   4.6459,   8.2002],
        [ -9.3616,  -2.4749,   0.1793],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.2849, -10.9655,  -7.6542],
        [  6.8407,  11.7666,   3.0045],
        [  7.1821,  -9.1845,   9.8548],
        [ -8.8363,   2.4086,  -5.6296],
        [ -8.5506,  -5.1154,   6.3423],
        [ -6.6998,   5.5954,   0.0823],
        [  6.3589,   0.8239,  -8.4230],
        [  2.3261,   7.1726,   6.2522],
        [  4.5340,  -3.3629,   8.9652],
        [  6.3124,   0.2057,  -0.4593],
        [ -2.7572,   7.7194,   5.9580],
        [ -0.1626, -10.9796,   6.4629],
        [ -7.8359,   4.2735,   7.2943],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -2.6485, -10.4151,   0.8766],
        [  5.3860,  -2.3418,   1.0826],
        [ -5.7347,  -8.6908,   9.2878],
        [  3.8111, -10.6787,  -4.6539],
        [ -9.4196,   0.0478,   9.4140],
        [  5.1793,   4.7461,   0.3899],
        [  0.6757, -11.0361,  -1.3047],
        [  3.6249, -10.7255,   5.6605],
        [ -9.3076,   0.7852,  -6.7010],
        [  2.5469,  13.7192,  -9.6935],
        [  5.6928,  12.4710,   3.6001],
        [ -9.1623,  -3.4288,  -4.9961],
        [  0.1252, -11.0062,   7.2564],
        [  7.0515,  11.6194,   8.4289],
        [  2.1504,  -4.6726,   8.7908],
        [  0.9809,   7.6642,  -1.4217],
        [  3.2647, -10.8066,  -0.3768],
        [  8.1528,  10.7452,   9.1493],
        [ -9.4203,  -2.0340,   2.5856],
        [ -8.0330,  -6.0765,   6.7945],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.6038,   0.7665,  -3.4439],
        [ -3.1671,   7.6158,   6.0765],
        [ -4.2040,   7.2486,   2.9875],
        [  6.3516,   0.6425,   0.3425],
        [  6.2804,   2.0019,   0.8607],
        [ 12.5763,   0.1333,  -9.2650],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.1989,  -9.7591,   0.8289],
        [ -4.7334,   6.9982,  -0.4392],
        [  1.3403,  -4.8011,  -1.1698],
        [  2.3376,  -4.6231,   9.8436],
        [  2.3376,  -4.6231,   9.8436],
        [  5.2939,  -2.4784,  -4.5840],
        [ -2.5683, -10.4433,  -2.9182],
        [  0.6757, -11.0361,  -1.3047],
        [ -5.0943,  -9.1682,   0.5073],
        [  2.3200,   7.1756,   1.1938],
        [  6.3549,   0.7096,   6.9198],
        [  5.1787,  -2.6383,  -4.2533],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.6681,  -4.8556,   9.2518],
        [ 11.7162,   5.6728,  -8.9372]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0371, 0.0102,  ..., 1.4711, 1.6704, 0.0371],
        [0.0371, 0.0000, 0.0365,  ..., 1.5069, 1.7045, 0.0000],
        [0.0102, 0.0365, 0.0000,  ..., 1.4704, 1.6683, 0.0365],
        ...,
        [1.4711, 1.5069, 1.4704,  ..., 0.0000, 0.2936, 1.5069],
        [1.6704, 1.7045, 1.6683,  ..., 0.2936, 0.0000, 1.7045],
        [0.0371, 0.0000, 0.0365,  ..., 1.5069, 1.7045, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.84056604, 13.01728313, 12.889695  , 12.9089559 , 13.00380624,
       13.1399181 , 13.0493368 , 12.21532282,  9.77777346,  9.73579738,
        6.49538841,  9.96391045, 12.64235359,  6.53250937, 10.61047596,
        9.24501829,  7.61088125,  4.75303127, 10.82106505,  5.76165934,
       13.09426142,  5.84450127, 10.67132524, 12.26997165, 12.10942412,
        7.32950385,  5.84450127,  7.39004226,  8.98765644,  9.26455795,
        9.6282763 ,  7.32754882,  5.0209005 , 13.69162795, 11.38774149,
        6.39520397, 11.27656488, 12.32545637,  5.41348121, 10.68416401,
        4.74770899,  5.30158497,  5.19986797, 14.0393949 , 12.17901033,
        5.86519525,  8.8737556 ,  9.68323686, 12.25806559, 11.20100795,
       13.6105595 , 11.65923396,  9.15866607,  9.96391045,  8.72896381,
        6.41202986,  7.5403822 ,  5.64500512,  6.31576264,  8.19703512,
       10.98076873,  8.92549752, 10.52357479, 10.74655346,  5.87305005,
       10.412299  , 11.33837396,  9.41970031,  7.02496223, 11.05672136,
       11.32149259,  9.34061938, 13.95361098, 13.70893492,  9.78286733,
       11.00694688, 13.59170809,  5.14371005,  7.72668467, 11.28895489,
       13.48809595,  9.63742669, 10.07238463,  9.97822473, 12.62711252,
        8.24808832,  8.37954853,  6.38399789,  6.59176018, 12.5769659 ,
       12.91505111, 11.56146873,  8.4486771 ,  4.98461983,  5.1805345 ,
        5.1805345 ,  5.84533345, 10.75443106, 11.05672136, 10.48840476,
        7.54127425,  6.39438044,  5.81199422,  9.75200848,  9.9353921 ,
       13.01728313])}
{'P': tensor([[5.1306e-03, 3.0809e-01, 2.1373e-01,  ..., 4.2338e-04, 9.4789e-08,
         3.3477e-04],
        [1.7636e-05, 1.3757e-02, 4.2663e-01,  ..., 2.6135e-04, 3.6712e-07,
         2.9337e-03],
        [5.0768e-05, 7.7670e-04, 4.4657e-02,  ..., 6.6430e-04, 1.3753e-06,
         1.3773e-02],
        ...,
        [5.4644e-03, 1.5373e-02, 1.6828e-02,  ..., 8.7625e-02, 2.2089e-05,
         1.9650e-03],
        [2.5045e-06, 1.3069e-06, 1.2743e-06,  ..., 2.0585e-05, 3.6571e-02,
         4.7004e-07],
        [6.6625e-08, 1.5820e-07, 1.9071e-07,  ..., 8.5721e-07, 3.2277e-06,
         4.4874e-03]]), 'X': tensor([[ 12.3357,   3.6001,   0.7903],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.1469,   6.9237,   1.6160],
        [ 11.0864,   7.0387,  -0.1238],
        [ 10.9916,   7.2133,  -0.9179],
        [ 10.5678,   7.9261,  -0.6100],
        [ -3.1671,   7.6158,   6.0765],
        [ -6.3252,  -8.1760,  -1.0891],
        [  8.2422,  -8.4041,  -1.4962],
        [  5.7218,  -9.9934,   6.5764],
        [  0.8940, -11.0402,  -0.9783],
        [  6.1222,  -0.7261,   8.8204],
        [ -9.0961,   1.6529,   3.1553],
        [  1.1274,  14.0116,   8.1753],
        [ -6.0844,  -8.3955,   0.7705],
        [ -5.3328,  -8.9992,   7.3440],
        [ 11.5083,  -4.1177,   0.2389],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -3.5520,   7.4976,   9.1522],
        [ -3.4100,   7.5436,  -7.6980],
        [  4.9337,  12.8566,  -3.6582],
        [  1.4306,  13.9620,  -0.2075],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.6681,  -4.8556,   9.2518],
        [  5.3118,   4.5452,   4.2114],
        [  5.7963,  -1.6018,   1.8856],
        [  0.1041, -11.0045,   2.4311],
        [  6.3497,   1.3680,  -1.9076],
        [ -6.4464,   5.8272,   7.5038],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 12.1379,   4.4125,  -3.8833],
        [  0.6363, -11.0348,   6.1414],
        [  6.1575,  12.2046,   3.3257],
        [ -6.9838,  -7.4965,   0.9796],
        [ 11.5083,  -4.1177,   0.2389],
        [  3.8182,  -3.9347,  -1.7961],
        [  0.6952,  -4.8060,   8.8976],
        [  5.6672,  -1.8631,   8.1020],
        [ 11.6544,   5.8273,  -3.1375],
        [ -8.0534,  -6.0429,   2.4583],
        [  1.7232,  -4.7574,   3.1270],
        [  4.4633,   5.6287,  -0.7754],
        [  2.3376,  -4.6231,   9.8436],
        [  0.6142,  14.0802,   0.5407],
        [  6.3360,   0.4258,  -9.2509],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.2692,  -1.8882,   3.9253],
        [  3.6249, -10.7255,   5.6605],
        [  6.2036,   2.4030,  -1.2500],
        [  6.2630,   2.1057,  -5.2136],
        [  5.1142,  -2.7230,  -4.1966],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.1504,   4.3674,  -7.2998],
        [  5.9655,   3.2172,  -7.5953],
        [  0.2732,  -4.7619,  -1.1577],
        [  1.8447,  -4.7372,   8.1235],
        [ -0.0410,   7.8591,   9.3457],
        [ -5.6169,  -8.7844,   8.9057],
        [ -0.0254, -10.9932,   3.2200],
        [ -0.1963,   7.8765,  -1.1371],
        [ -3.5173, -10.0623,   4.8280],
        [  0.1460,   7.8340,  -9.4614],
        [  6.9433,  -9.3362,  -0.3421],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -2.5161,   7.7699,  -9.3531],
        [  3.1271, -10.8343,   2.6442],
        [  2.4572,   7.1085,   7.7376],
        [  3.9515,  -3.8425,   6.1729],
        [  5.7802,  -1.6363,  -6.2782],
        [  3.2990,  -4.2415,  -7.4857],
        [ -5.6169,  -8.7844,   8.9057],
        [ -5.5525,   6.5146,  -4.4984],
        [  2.0937,  13.8293,  -2.3533],
        [  5.3804,  -2.3504,   7.2744],
        [  9.4554,   9.4245,   9.8800],
        [  4.7544,  -3.1425,  -1.5315],
        [  5.5158,  -2.1337,   3.5246],
        [  0.3231,  -4.7691,  -1.6120],
        [  3.2692,  13.5098,  -4.9032],
        [  4.6357,   5.4415,   7.9067],
        [  5.9026,   3.3837,  -0.6573],
        [  6.3305,   1.6119,   1.1538],
        [  8.8474,  10.0868,  -8.8644],
        [ 12.5717,   1.9266,   7.4903],
        [  1.1994,   7.6035,   2.4295],
        [  5.5158,  -2.1337,   3.5246],
        [  6.2895,   0.0405,   5.7652],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.3560,   9.5391,   9.4943],
        [  9.1409,  -7.5789,   1.5814],
        [ -8.0330,  -6.0765,   6.7945],
        [  7.8982,  -8.6778,   0.0239],
        [ -9.4117,   0.1160,   8.2574],
        [ -5.9987,   6.1949,  -6.2114],
        [  4.1140,  -3.7220,  -0.1273],
        [  6.3589,   0.8239,  -8.4230],
        [ -8.3653,   3.4184,   9.3454],
        [ -1.8367,   7.8718,  -8.6523],
        [ -8.8154,   2.4611,   2.5198],
        [  4.5430,   5.5438,  -3.8669],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.2804,   2.0019,   0.8607],
        [ -8.3535,  -5.5118,   9.8399],
        [  1.9312,  13.8649,   8.3563]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.8819e-02, 1.2462e-01,  ..., 1.6594e-01, 1.2268e+00,
         7.2706e-01],
        [6.8819e-02, 0.0000e+00, 6.3704e-02,  ..., 2.2010e-01, 1.2546e+00,
         6.7646e-01],
        [1.2462e-01, 6.3704e-02, 0.0000e+00,  ..., 2.4735e-01, 1.2401e+00,
         6.1300e-01],
        ...,
        [1.6594e-01, 2.2010e-01, 2.4735e-01,  ..., 1.0727e-04, 1.0757e+00,
         7.5773e-01],
        [1.2268e+00, 1.2546e+00, 1.2401e+00,  ..., 1.0757e+00, 0.0000e+00,
         1.2118e+00],
        [7.2706e-01, 6.7646e-01, 6.1300e-01,  ..., 7.5773e-01, 1.2118e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.85032836, 13.08414689, 13.12218013, 13.13205357, 13.1471313 ,
       13.20988996,  8.24808832, 10.33712258, 11.77124373, 11.51557166,
       11.07637531,  6.16512856,  9.24501829, 14.05687815, 10.36844813,
       10.4606265 , 12.22276278, 10.09979747,  8.29640613,  8.27853076,
       13.77075174, 14.03505624, 13.99048094,  9.9353921 ,  6.99097013,
        6.01356004, 11.0050379 ,  6.49538841,  8.68979052, 10.5575601 ,
       12.91505111, 11.05317597, 13.66991027, 10.24557047, 12.22276278,
        5.48276038,  4.85603886,  5.96556248, 13.03002338, 10.06850677,
        5.05989547,  7.18354885,  5.1805345 , 14.09357547,  6.35029274,
       10.92597196,  9.97822473, 12.41366754, 11.32149259,  6.652742  ,
        6.60752373,  5.79391825,  5.18277858,  5.81199422, 12.91143491,
        6.77777606,  4.76969364,  5.08372149,  7.85919687, 10.42667093,
       10.99326577,  7.8788965 , 10.65929968,  7.83534987, 11.63502679,
        9.43737523,  8.16714836, 11.27656488,  7.52117132,  5.51177501,
        6.00732193,  5.37343173, 10.42667093,  8.55982105, 13.98691426,
        5.87132718, 13.35013374,  5.69913817,  5.91408504,  4.78002665,
       13.89974693,  7.14837514,  6.80370254,  6.53250937, 13.41713661,
       12.71843462,  7.69752262,  5.91408504,  6.28961988,  5.26067019,
       13.3614598 , 11.87412439, 10.07238463, 11.73397309,  9.41244865,
        8.62329466,  5.54776934,  6.41202986,  9.0368458 ,  8.08320235,
        9.15253344,  7.16747878,  5.5366698 ,  6.59176018, 10.00802627,
       13.99877413])}
{'P': tensor([[3.9787e-03, 3.4130e-01, 2.3186e-01,  ..., 6.4598e-08, 1.5533e-07,
         1.1190e-07],
        [2.1407e-05, 1.2436e-02, 8.0266e-02,  ..., 1.1218e-07, 6.3961e-07,
         4.5225e-07],
        [2.7635e-05, 1.1729e-03, 1.6088e-02,  ..., 1.1363e-07, 9.1234e-07,
         5.6267e-07],
        ...,
        [1.7840e-03, 1.1048e-03, 7.3408e-04,  ..., 2.8382e-01, 1.3832e-06,
         8.1256e-06],
        [7.1886e-06, 4.1971e-06, 4.0590e-06,  ..., 1.6243e-05, 6.7077e-02,
         1.9065e-03],
        [1.0167e-05, 5.9339e-06, 4.9996e-06,  ..., 5.2510e-04, 1.5398e-03,
         9.9040e-02]]), 'X': tensor([[ 12.2914,   3.8037,   6.0978],
        [ 11.5966,   5.9666,   6.0083],
        [ 11.4495,   6.3026,   6.8257],
        [ 11.4235,   6.3594,   5.1663],
        [ 10.9931,   7.2107,   7.7933],
        [ 10.0567,   8.6697,   4.6329],
        [  0.1287, -11.0065,   6.5757],
        [  2.2644,  -4.6434,   9.7508],
        [  2.4932,   7.0903,  -5.4214],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.1973,  -0.4372,   6.5720],
        [  7.5064,  -8.9651,  -0.0994],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.5154,   2.5072,  -9.3274],
        [  5.6093,  -1.9705,   6.4807],
        [  4.1637,  13.1898,  -1.9189],
        [  2.6970,   6.9826,   9.7350],
        [ -2.0229, -10.6163,   8.6930],
        [  6.3200,   0.2690,   1.5900],
        [  6.3549,   0.7096,   6.9198],
        [ -8.1905,   3.7253,   4.1175],
        [  6.9693,  -9.3200,  -8.3822],
        [  6.1741,  -0.5329,   3.1913],
        [  1.3339,   7.5627,  -0.9034],
        [  7.0664,  -9.2590,   8.0879],
        [  4.9152,  12.8653,  -1.4459],
        [ 12.5123,  -0.5339,  -1.0728],
        [  4.7544,  -3.1425,  -1.5315],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.3604,   0.4855,  -3.0158],
        [  5.9526,   3.2525,  -4.3821],
        [  1.3670,  -4.7990,   3.2603],
        [  3.5965,  -4.0754,  -3.5035],
        [  0.6952,  -4.8060,   8.8976],
        [  2.1824,  -4.6647,   1.1985],
        [ -6.7643,  -7.7372,  -9.6663],
        [  7.1441,  11.5529,   2.3232],
        [  6.2036,   2.4030,  -1.2500],
        [ 11.7418,   5.6070,  -1.4192],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.3376,  -4.6231,   9.8436],
        [ -9.3616,  -2.4750,   3.7765],
        [  9.3836,   9.5075,  -5.9278],
        [  5.0277,   4.9589,   7.0175],
        [  4.2339,  -3.6268,   6.3125],
        [ -5.1377,   6.7751,  -0.4438],
        [  6.0054,  -1.0902,   8.3079],
        [  0.5175,  14.0910,   4.6540],
        [ -4.6942,  -9.4299,  -9.1846],
        [  4.5767,   5.5071,   3.9069],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.3804,  -2.3504,   7.2744],
        [ 11.5568,   6.0602,   9.3244],
        [ -8.8631,  -4.3728,  -1.1746],
        [  6.3606,   0.9204,   3.2525],
        [  5.3804,  -2.3504,   7.2744],
        [ -5.3179,  -9.0101,   9.4472],
        [  2.1440, -10.9811,   8.5281],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.3616,  -2.4750,   3.7765],
        [  1.0093,   7.6567,   2.3526],
        [  5.1142,  -2.7230,  -4.1966],
        [  9.0004,  -7.7195,   4.6945],
        [  2.7976,  -4.4682,  -4.2020],
        [  7.1441,  11.5529,   2.3232],
        [  6.3031,   1.8477,  -7.1706],
        [  1.6308,  13.9254,  -2.0984],
        [  0.2435,   7.8191,  -6.4709],
        [  4.6231,   5.4556,   8.4973],
        [  1.1644, -11.0394,   7.8511],
        [  5.1793,   4.7461,   0.3899],
        [  2.0899,  -4.6870,  -7.4260],
        [ -7.5603,   4.6459,   8.2002],
        [  6.9433,  -9.3362,  -0.3421],
        [  8.3718,  -8.2953,   8.4666],
        [  6.6319,  11.9066,   3.5165],
        [ -7.0948,   5.1940,   9.8171],
        [  0.4996,   7.7738,  -4.4711],
        [  5.0277,   4.9589,   7.0175],
        [ 12.2311,  -2.0445,   1.3545],
        [  5.0348,  -2.8229,   3.5436],
        [ 10.5678,   7.9261,  -0.6100],
        [  6.1984,  -0.4327,   1.5438],
        [  4.1537,   5.9340,  -6.0423],
        [ 11.4743,   6.2478,  -2.1100],
        [ -4.4779,  -9.5609,  -2.4505],
        [  4.4049,   5.6892,  -4.4394],
        [ 11.3118,   6.5957,  -4.5667],
        [ -6.4464,   5.8272,   7.5038],
        [ -7.6719,  -6.6260,   7.1468],
        [  4.5954,  13.0099,  -4.4439],
        [  2.5375,   7.0676,  -4.3490],
        [  4.1297,  -3.7098,  -3.9838],
        [  0.4119,  14.1019,  -4.1870],
        [ -2.0721,   7.8432,   6.6380],
        [  6.0560,  -0.9421,  -5.0334],
        [  3.3094,  13.4969,  -3.6507],
        [ -2.6485, -10.4151,   0.8766],
        [  5.5424,  -2.0885,   7.2487],
        [  0.1497,   7.8334,  -5.4412],
        [  6.2675,  12.1379,  -8.9720],
        [  8.1043,  -8.5164,  -7.9290],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.3539,  -5.5111,   0.3373]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0550, 0.0905,  ..., 1.2213, 1.1328, 1.3004],
        [0.0550, 0.0000, 0.0550,  ..., 1.2417, 1.1878, 1.3499],
        [0.0905, 0.0550, 0.0000,  ..., 1.2967, 1.2080, 1.3900],
        ...,
        [1.2213, 1.2417, 1.2967,  ..., 0.0000, 1.3494, 0.9315],
        [1.1328, 1.1878, 1.2080,  ..., 1.3494, 0.0000, 0.5833],
        [1.3004, 1.3499, 1.3900,  ..., 0.9315, 0.5833, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.86648771, 13.04156075, 13.06957425, 13.0743399 , 13.14690137,
       13.2778378 , 11.00726483,  5.16611782,  7.5158489 , 10.91805683,
        6.21275016, 11.69264243,  9.69629   , 10.8992524 , 12.76408562,
        5.94535529, 13.83139024,  7.48538905, 10.80728788,  6.32572323,
        6.39438044,  8.99791316, 11.63765273,  6.19707944,  7.67939546,
       11.64746882, 13.77222752, 12.52372629,  5.69913817,  8.62260327,
        9.37296102,  6.78325151,  4.98989906,  5.43544626,  4.85603886,
        5.14999799, 10.27716434, 13.58333857,  6.652742  , 13.01187484,
       10.54658597,  9.20340031,  5.1805345 ,  9.68324553, 13.35832725,
        7.06169391,  5.5748624 ,  8.50278896,  6.10360073, 14.10045797,
       10.53368644,  7.16058587, 12.50996828,  5.87132718, 13.0493368 ,
        9.88310614,  6.42689705,  5.87132718, 10.46238054, 11.18839169,
       10.76154256,  9.68324553,  7.72291946,  5.79391825, 11.85743425,
        5.27178363, 13.58333857,  6.56833864, 14.02058621,  7.822854  ,
        7.15100303, 11.10066126,  7.02496223,  5.13182506,  8.8737556 ,
       11.63502679, 11.785566  , 13.62896593,  8.79284466,  7.78980299,
        7.06169391, 12.40074901,  5.77219077, 13.20988996,  6.2134961 ,
        7.24326992, 13.06498554, 10.5575601 ,  7.19513806, 13.09426142,
        8.68979052, 10.1371552 , 13.79762415,  7.5092834 ,  5.5513072 ,
       14.10796466,  8.11227668,  6.12886024, 13.89671314, 10.74655346,
        5.92281142,  7.83486943, 13.66051884, 11.75618513,  9.94173708,
       10.00794468])}
{'P': tensor([[6.0002e-03, 3.3232e-01, 2.9644e-01,  ..., 3.0177e-07, 1.8309e-06,
         1.5012e-03],
        [2.4572e-05, 2.0262e-02, 3.5241e-04,  ..., 4.6666e-07, 1.3948e-05,
         1.9343e-02],
        [1.3636e-05, 2.3012e-01, 7.4203e-03,  ..., 1.9435e-07, 4.7699e-06,
         4.9847e-03],
        ...,
        [2.5220e-03, 6.9989e-04, 9.2662e-04,  ..., 1.0474e-02, 6.2541e-07,
         4.4913e-06],
        [7.5829e-07, 6.0215e-07, 6.4710e-07,  ..., 1.7174e-05, 2.4870e-02,
         5.8488e-07],
        [1.6454e-07, 3.8337e-07, 3.2937e-07,  ..., 3.1385e-08, 6.5245e-05,
         7.6484e-03]]), 'X': tensor([[ 12.2592,   3.9425,   2.0383],
        [ 11.1469,   6.9237,   1.6160],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.0864,   7.0387,  -0.1238],
        [ 10.2470,   8.4054,  -0.1177],
        [ 10.5678,   7.9261,  -0.6100],
        [  1.9241,   7.3503,   7.9004],
        [ 12.4877,  -0.7222,   8.8347],
        [  5.8450,  -9.9356,   3.0767],
        [ -2.6735, -10.4062,  -1.5911],
        [  3.1756, -10.8247,   6.8882],
        [ -2.6485, -10.4151,   0.8766],
        [  8.7131,  10.2216,   7.7915],
        [  5.6928,  12.4710,   3.6001],
        [  5.7802,  -1.6363,  -6.2782],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.8890,  -4.4316,   1.1600],
        [ -9.2508,   1.0572,   2.9978],
        [  0.2732,  -4.7619,  -1.1577],
        [  1.7122,  -4.7591,  -3.6734],
        [  3.2366,  -4.2732,   5.1532],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -7.4602,   4.7718,  -4.6277],
        [ -3.0644,   7.6439,  -1.4322],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -9.2508,   1.0572,   2.9978],
        [  0.2571,  -4.7594,   1.5618],
        [  2.0039,   7.3173,  -1.5885],
        [  2.3200,   7.1756,   1.1938],
        [  3.4009,  -4.1874,  -5.9751],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.7845,   3.6652,   0.9175],
        [  0.5850,  -4.7982,   9.6639],
        [  5.9526,   3.2525,  -4.3821],
        [ -8.8154,   2.4611,   2.5198],
        [ 12.3453,  -1.5462,   7.0212],
        [ -9.4117,   0.1160,   8.2574],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.3560,   9.5391,   9.4943],
        [  0.5175,  14.0910,   4.6540],
        [ -0.4341, -10.9473,   7.4509],
        [  4.3554,   5.7393,  -9.0766],
        [ -3.4986,   7.5152,   9.0712],
        [  4.1348,   5.9514,  -0.8856],
        [ -4.9217,   6.8979,  -1.7667],
        [ 11.6801,  -3.7196,  -0.9181],
        [  6.9314,  -9.3435,  -1.2729],
        [  2.5382,   7.0672,   4.5097],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.0876,   6.7527,  -5.3540],
        [ -8.3653,   3.4184,   9.3454],
        [  9.5261,  -7.1675,   1.4206],
        [  2.8994,   6.8675,  -9.5987],
        [  0.5600, -11.0321,  -0.8660],
        [  4.6063,  13.0051,  -3.1456],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.9653,  -9.3225,  -3.4727],
        [  0.1460,   7.8340,  -9.4614],
        [  7.3724,  -9.0577,  -8.9843],
        [  6.3236,   0.3011,  -4.6668],
        [  6.3301,   0.3638,  -7.5685],
        [  6.8093,  -9.4178,   8.4244],
        [  6.2804,   2.0019,   0.8607],
        [  6.2555,   2.1474,   4.1685],
        [  6.2895,   0.0405,   5.7652],
        [  5.8325,  -1.5222,  -2.1341],
        [  8.7715,  -7.9390,   5.2900],
        [  2.5636, -10.9293,   9.6519],
        [  9.3424,  -7.3685,  -3.6807],
        [  3.5284, -10.7484,   5.3347],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -8.4297,  -5.3637,  -9.3962],
        [  1.7228,  -4.7575,   8.3396],
        [ -9.1624,  -3.4286,   2.0647],
        [ -2.0229, -10.6163,   8.6930],
        [  3.7992,  -3.9474,  -8.9019],
        [ -8.1666,   3.7653,   3.6075],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.4430,  -0.1897,   4.8383],
        [ -0.0410,   7.8591,   9.3457],
        [  8.2154,  -8.4262,   8.0171],
        [  2.4639, -10.9431,  -7.3718],
        [  6.2555,   2.1474,   4.1685],
        [ -8.5149,  -5.1905,   1.5239],
        [  5.1793,   4.7461,   0.3899],
        [ 11.5966,   5.9666,   6.0083],
        [  4.2275,  -3.6320,  -0.2714],
        [  1.3767, -11.0341,   9.9415],
        [ -8.6801,   2.7809,  -0.8443],
        [ -2.5683, -10.4433,  -2.9182],
        [ -7.8280,   4.2848,  -7.1955],
        [ 11.0373,   7.1299,  -8.5895],
        [  8.1217,  -8.5024,  -0.6149],
        [  7.0171,  11.6439,  -5.0178],
        [  6.9956,  11.6590,  -7.6876],
        [ -0.0254, -10.9932,   3.2200],
        [ 11.6801,  -3.7196,  -0.9181],
        [  9.7328,   9.0902,  -0.8917],
        [  4.0421,  -3.7765,  -8.3631],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.4157,   0.0821,  -5.0025],
        [ 10.8218,  -5.4177,   6.8573],
        [  8.2422,  -8.4041,  -1.4962],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.2182,  13.8007,   5.1889]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0593, 0.0917,  ..., 0.5405, 1.0993, 0.4136],
        [0.0593, 0.0000, 0.0637,  ..., 0.5783, 1.1454, 0.4185],
        [0.0917, 0.0637, 0.0000,  ..., 0.5382, 1.1132, 0.4821],
        ...,
        [0.5405, 0.5783, 0.5382,  ..., 0.0000, 0.5841, 0.9032],
        [1.0993, 1.1454, 1.1132,  ..., 0.5841, 0.0000, 1.3989],
        [0.4136, 0.4185, 0.4821,  ..., 0.9032, 1.3989, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.87752052, 13.12218013, 13.08414689, 13.13205357, 13.25336001,
       13.20988996,  7.59795445, 12.5085995 , 11.52733581, 10.74409834,
       11.28093507, 10.74655346, 13.43127131, 13.70893492,  6.00732193,
        9.37989631,  5.29009836,  9.3109884 ,  4.76969364,  5.05774295,
        5.36061974, 10.37698435,  8.85572292,  8.23525433, 10.33712258,
        9.3109884 ,  4.7663526 ,  7.58668144,  7.54127425,  5.39450305,
        9.58347456,  6.84796202,  4.83371341,  6.78325151,  9.15253344,
       12.44177641,  9.41244865,  5.26067019, 13.3614598 , 14.10045797,
       10.95593729,  7.20482224,  8.28968067,  7.24677664,  8.47371868,
       12.25806559, 11.63383221,  7.50917551, 11.9120321 ,  7.42512621,
        9.0368458 , 11.92133129,  7.4544866 , 11.04629471, 13.79676654,
       12.07225469, 11.63724792,  7.83534987, 11.67875855,  6.33075682,
        6.34059021, 11.62158554,  6.59176018,  6.61386436,  6.28961988,
        6.02789634, 11.83075005, 11.22596994, 11.89855267, 11.31276109,
        9.96152788,  9.99145096,  5.05981914,  9.78285424, 10.80728788,
        5.47866221,  8.99276564,  9.50654967,  8.90405823,  9.44486609,
        7.85919687, 11.76830981, 11.21703654,  6.61386436,  9.97220799,
        7.02496223, 13.04156075,  5.57341623, 11.11969815,  9.11473391,
       10.75443106,  8.92395381, 13.1399181 , 11.75807252, 13.59480906,
       13.59673406, 10.99326577, 12.25806559, 13.31764181,  5.53175275,
        9.34127013,  9.41605684, 12.10222204, 11.77124373,  9.20340031,
       13.97779839])}
{'P': tensor([[3.2683e-03, 2.9327e-01, 2.1440e-01,  ..., 6.3158e-09, 1.6378e-07,
         1.1185e-06],
        [6.8051e-06, 3.8634e-03, 4.7908e-01,  ..., 7.9654e-09, 9.2383e-08,
         9.3689e-07],
        [2.7639e-05, 8.9416e-05, 2.0532e-02,  ..., 4.1067e-08, 3.7842e-07,
         4.3629e-06],
        ...,
        [2.9537e-06, 1.7704e-06, 1.4124e-06,  ..., 1.4653e-01, 2.0091e-03,
         1.2938e-03],
        [5.6046e-04, 4.1073e-04, 3.2271e-04,  ..., 1.1452e-06, 3.4646e-02,
         1.4896e-04],
        [2.4591e-03, 1.6323e-03, 1.3371e-03,  ..., 7.2893e-05, 7.8597e-03,
         1.5326e-01]]), 'X': tensor([[ 12.2435,   4.0076,   8.1257],
        [ 11.5456,   6.0861,   8.1101],
        [ 11.2571,   6.7072,   8.3693],
        [ 10.9931,   7.2107,   7.7933],
        [ 11.6634,   5.8050,   9.1590],
        [ 11.5568,   6.0602,   9.3244],
        [  8.0877,  -8.5297,  -1.1050],
        [  7.3724,  -9.0577,  -8.9843],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.6954,  -4.7616,   9.8198],
        [  9.6816,   9.1536,  -5.8543],
        [  4.7813,  -3.1139,   8.2691],
        [  5.8930,  12.3592,  -8.9201],
        [ -4.2271,   7.2387,  -1.8316],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4430,  -0.1897,   4.8383],
        [ 10.1503,   8.5414,  -7.2681],
        [ -2.7572,   7.7194,   5.9580],
        [  5.0617,  12.7957,   8.3544],
        [ -7.8280,   4.2848,  -7.1955],
        [ -7.4808,   4.7462,   2.0266],
        [ -0.6323,   7.9085,  -5.9544],
        [ -5.3179,  -9.0101,   9.4472],
        [ -8.4267,  -5.3697,   2.4172],
        [  0.5250, -11.0306,   5.5358],
        [ -8.6229,   2.9068,   2.4429],
        [  2.3373,  -4.6232,  -5.2246],
        [  7.5064,  -8.9651,  -0.0994],
        [  1.7232,  -4.7574,   3.1270],
        [ 12.4837,   2.7510,   0.4449],
        [  1.6308,  13.9254,  -2.0984],
        [  6.6319,  11.9066,   3.5165],
        [ -7.4807,   4.7463,   2.0616],
        [  6.3505,   0.6235,  -8.5849],
        [ -8.0087,   4.0173,   9.1091],
        [  6.2026,  12.1774,  -1.5887],
        [ 10.9916,   7.2133,  -0.9179],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -0.7599, -10.8995,  -6.0366],
        [ -1.8788, -10.6567,   3.7068],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -3.6381,   7.4683,   7.3544],
        [  2.0243,   7.3086,   7.5432],
        [  6.2905,   1.9366,   2.1169],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.2687,  -2.9794,   0.6436],
        [  7.8982,  -8.6778,   0.0239],
        [ -8.8154,   2.4611,   2.5198],
        [ -8.9624,  -4.0944,  -4.9622],
        [  3.8111, -10.6787,  -4.6539],
        [ -3.5903,   7.4847,  -8.9112],
        [ -7.9298,   4.1367,  -3.9773],
        [ -7.6719,  -6.6260,   7.1468],
        [  3.8182,  -3.9347,  -1.7961],
        [  7.3724,  -9.0577,  -8.9843],
        [  8.5093,  -8.1764,  -3.2381],
        [ -8.3653,   3.4184,   9.3454],
        [ -1.5875,   7.8944,  -9.4286],
        [ -9.1960,  -3.2964,   1.6921],
        [  1.9241,   7.3503,   7.9004],
        [ -5.2589,   6.7024,  -5.5809],
        [ 12.5832,   1.7543,  -7.6134],
        [  0.2435,   7.8191,  -6.4709],
        [  3.3167,   6.6022,   6.3554],
        [ -8.9282,   2.1648,   8.4894],
        [ -1.6334, -10.7207,   7.1397],
        [  3.2692,  13.5098,  -4.9032],
        [  9.0004,  -7.7195,   4.6945],
        [  1.9505,  -4.7171,   9.9868],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.7976,  -4.4682,  -4.2020],
        [ -8.3062,  -5.6008,   0.0287],
        [ -1.0482, -10.8487,  -5.9066],
        [  5.9843,  -9.8679,  -6.1449],
        [ -8.6229,   2.9068,   2.4429],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 10.1798,   8.5002,  -5.8893],
        [  5.5294,  -2.1106,   9.0626],
        [  6.2630,   2.1057,  -5.2136],
        [  5.5294,  -2.1106,   9.0626],
        [ 10.0567,   8.6697,   4.6329],
        [  6.3124,   0.2057,  -0.4593],
        [  5.4426, -10.1181,  -5.1268],
        [  5.5158,  -2.1337,   3.5246],
        [  1.5653,  13.9377,   4.1773],
        [  9.9750,  -6.6338,  -1.2440],
        [  2.3373,  -4.6232,  -5.2246],
        [  3.6817,   6.3354,   6.1233],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.2535,  -0.1709,  -9.4102],
        [  9.9058,  -6.7203,   5.6928],
        [ -8.8058,   2.4850,  -1.7425],
        [ -5.6423,   6.4537,  -1.2091],
        [ -2.0229, -10.6163,   8.6930],
        [  4.3377,   5.7570,   1.3712],
        [ -8.7735,   2.5640,  -6.3313],
        [  3.3744,   6.5623,  -6.3636],
        [  5.2911,  -2.4824,  -3.6877],
        [  5.0277,   4.9589,   7.0175],
        [ 12.2589,  -1.9312,   9.8657],
        [ 11.7796,  -3.4696,   9.0088],
        [  0.1206,  -4.7363,   8.7103],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 10.5057,  -5.9099,  -5.1599],
        [  2.8890,  -4.4316,   1.1600]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0570, 0.0883,  ..., 1.7621, 1.2005, 0.7236],
        [0.0570, 0.0000, 0.0329,  ..., 1.8081, 1.2266, 0.7583],
        [0.0883, 0.0329, 0.0000,  ..., 1.8401, 1.2523, 0.7874],
        ...,
        [1.7621, 1.8081, 1.8401,  ..., 0.0000, 0.8827, 1.0941],
        [1.2005, 1.2266, 1.2523,  ..., 0.8827, 0.0000, 0.5013],
        [0.7236, 0.7583, 0.7874,  ..., 1.0941, 0.5013, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.88270018, 13.05148826, 13.103709  , 13.14690137, 13.02818702,
       13.0493368 , 11.75438006, 11.67875855, 12.04924159,  5.05444288,
       13.32374056,  5.70591986, 13.6922425 ,  8.38252937,  7.5092834 ,
        9.44486609, 13.26590071,  8.19703512, 13.76047999,  8.92395381,
        8.859416  ,  7.9337647 , 10.46238054,  9.99211663, 11.04313641,
        9.09963894,  5.18047638, 11.69264243,  5.05989547, 12.78327037,
       14.02058621, 13.62896593,  8.85939626,  6.38105015,  8.95983586,
       13.6660671 , 13.1471313 , 12.05395056, 10.92597196, 10.82106505,
       10.52587243,  8.30729352,  7.58378655,  6.58184474,  6.27905292,
        6.77777606,  9.73579738, 11.73397309,  9.15253344,  9.85330646,
       11.33837396,  8.30125201,  8.94395628, 10.1371552 ,  5.48276038,
       11.67875855, 11.80092578,  9.0368458 ,  8.05242591,  9.76896904,
        7.59795445,  8.51928531, 12.70489563,  7.822854  ,  7.38844703,
        9.18690018, 10.84437705, 13.89974693, 11.85743425,  5.10446653,
        5.21280156,  5.27178363, 10.01804223, 10.8992524 , 11.54071859,
        9.09963894, 12.4217523 , 13.26209545,  5.91854397,  6.60752373,
        5.91854397, 13.2778378 ,  6.31576264, 11.4890861 ,  5.91408504,
       14.02532758, 11.97949273,  5.18047638,  7.32754882, 13.01728313,
        6.25586297, 11.9702762 ,  9.14973319,  8.5724042 , 10.80728788,
        7.20826219,  9.14044984,  7.37902766,  5.84450127,  7.06169391,
       12.41012046, 12.27993098,  4.73784222, 10.37698435, 12.05395056,
        5.29009836])}
{'P': tensor([[2.2229e-03, 2.5795e-01, 1.4825e-01,  ..., 1.5895e-06, 3.0874e-07,
         6.9765e-08],
        [2.0753e-05, 3.0398e-03, 1.5866e-01,  ..., 2.1528e-06, 4.4904e-07,
         9.0354e-08],
        [6.8239e-06, 1.0021e-05, 2.0871e-03,  ..., 8.1460e-07, 3.7163e-07,
         3.1491e-08],
        ...,
        [2.9565e-04, 3.1942e-04, 2.5775e-04,  ..., 1.7302e-01, 9.1462e-05,
         7.3789e-04],
        [4.5651e-06, 4.4501e-06, 3.3844e-06,  ..., 5.1206e-05, 1.3008e-01,
         1.3930e-05],
        [8.0468e-04, 8.4331e-04, 5.1505e-04,  ..., 8.0341e-02, 4.2350e-06,
         2.2462e-01]]), 'X': tensor([[ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [-4.2040e+00,  7.2486e+00,  2.9875e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0074, 0.0365,  ..., 1.0648, 1.2074, 1.3201],
        [0.0074, 0.0000, 0.0345,  ..., 1.0625, 1.2124, 1.3182],
        [0.0365, 0.0345, 0.0000,  ..., 1.0930, 1.2416, 1.3496],
        ...,
        [1.0648, 1.0625, 1.0930,  ..., 0.0000, 1.1553, 0.2648],
        [1.2074, 1.2124, 1.2416,  ..., 1.1553, 0.0000, 1.2163],
        [1.3201, 1.3182, 1.3496,  ..., 0.2648, 1.2163, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.889695  , 12.9089559 , 13.01728313, 13.1399181 , 13.14777433,
       13.15525803,  8.4486771 ,  5.91222346,  4.91652556,  6.05165248,
        6.652742  ,  6.56833864, 13.2778378 , 11.32149259,  8.04664886,
        6.05165248,  8.8737556 ,  8.77913428, 13.35927527,  5.46244988,
        8.04664886, 13.08414689,  6.5618968 , 13.00380624, 11.36089355,
       11.28093507,  6.48486171,  5.16077764, 11.31276109, 14.05633441,
        6.652742  , 13.09426142,  7.50564223, 11.00726483,  7.57685908,
       10.76154256, 11.24434374,  5.37343173,  7.80625981, 11.83570132,
        5.64464807, 13.1471313 , 12.67576792, 13.26209545,  7.03733775,
        6.42689705,  9.91668182,  5.34250225, 13.3614598 ,  7.48311002,
        4.75303127,  9.0368458 ,  5.26067019,  8.92395381,  6.24614905,
        5.05444288, 10.0674887 ,  7.63726134, 12.07818871, 13.97857591,
        6.3949043 ,  8.57560275,  9.38232121,  8.87007495,  8.37954853,
        6.16512856,  5.8828986 ,  9.48337063, 10.71162681,  5.40689332,
        5.38342077,  8.97052115,  5.14607014, 11.20100795,  6.25586297,
       10.99326577, 13.92390689, 12.77402351,  5.05989547, 13.1399181 ,
        7.32754882, 13.79676654, 12.62711252,  7.23276582,  5.29755487,
       13.26590071,  7.62011734,  5.70591986,  7.50500064, 12.13568598,
       10.14044489, 13.89264822,  6.54615887, 13.55868684,  5.07306274,
       10.90638066, 10.33958639,  5.21280156, 11.0050379 , 11.28215429,
        5.05981914,  9.26455795,  8.09842704,  6.4020404 ,  9.43737523,
       12.13568598])}
{'P': tensor([[3.2048e-03, 1.8736e-01, 1.6727e-01,  ..., 1.4834e-08, 3.0483e-05,
         2.5786e-08],
        [1.5255e-05, 1.1626e-02, 2.7588e-05,  ..., 4.7434e-08, 1.4321e-05,
         1.1749e-07],
        [9.6586e-06, 2.1246e-01, 2.0115e-03,  ..., 7.6875e-09, 6.1307e-06,
         1.5459e-08],
        ...,
        [1.5523e-05, 5.7901e-06, 8.3076e-06,  ..., 8.9899e-02, 1.3627e-04,
         8.1782e-05],
        [4.9917e-04, 1.1289e-03, 5.8078e-04,  ..., 1.0215e-07, 7.0723e-03,
         1.9504e-07],
        [3.3451e-07, 1.6710e-07, 2.0669e-07,  ..., 4.8076e-02, 2.2575e-06,
         1.8979e-02]]), 'X': tensor([[ 12.1588,   4.3364,  -8.0867],
        [ 11.0373,   7.1299,  -8.5895],
        [ 11.7162,   5.6728,  -8.9372],
        [ 10.9875,   7.2207,  -8.8549],
        [ 10.9394,   7.3069,  -8.9365],
        [ 10.1503,   8.5414,  -7.2681],
        [  4.5814, -10.4485,   2.5496],
        [ -3.2736, -10.1703,   3.8712],
        [ -2.8162,   7.7059,  -9.6885],
        [ -1.8823, -10.6558,  -3.8952],
        [ -7.6525,  -6.6534,   5.1721],
        [ -9.4624,  -0.4633,  -7.3918],
        [ 10.2470,   8.4054,  -0.1177],
        [ -9.4196,   0.0478,   9.4140],
        [ 11.4743,   6.2478,  -2.1100],
        [ -5.0943,  -9.1682,   0.5073],
        [  0.9809,  -4.8145,   6.8706],
        [  1.7228,  -4.7575,   8.3396],
        [  2.3589,  13.7668,  -3.6501],
        [  2.4932,   7.0903,  -5.4214],
        [  2.5829,  -4.5466,  -8.4015],
        [ -9.3739,   0.3982,  -4.2564],
        [  3.5965,  -4.0754,  -3.5035],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.7218,  -9.9934,   6.5764],
        [  3.2647, -10.8066,  -0.3768],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.2975,   0.0947,   5.1791],
        [  2.5128, -10.9365,   9.9645],
        [  6.2770,  -0.0379,   2.4195],
        [  5.0190,   4.9705,  -7.9752],
        [  2.8474,   6.8979,   6.6457],
        [ -4.0409,   7.3169,  -3.5779],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -2.5056,   7.7720,   5.5878],
        [ 11.6634,   5.8050,   9.1590],
        [  3.7916,  -3.9524,  -7.9547],
        [  5.2608,  -2.5256,  -9.1446],
        [  3.3474,  -4.2162,   5.2455],
        [  6.1685,  12.1979,  -6.2473],
        [ -7.8359,   4.2735,   7.2943],
        [  2.4356,  -4.5942,  -5.0808],
        [  5.3602,  -2.3808,   7.2098],
        [ 10.1798,   8.5002,  -5.8893],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.1685,  12.1979,  -6.2473],
        [ -7.4602,   4.7718,  -4.6277],
        [ -3.9456,  -9.8542,   0.7466],
        [  2.1120,  -4.6818,   8.6144],
        [  3.0719, -10.8449,  -4.1673],
        [ -2.6069, -10.4298,  -6.1995],
        [ -7.0253,  -7.4493,  -1.5978],
        [  6.3119,   1.7797,   9.9186],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -0.6323,   7.9085,  -5.9544],
        [ -2.1382,   7.8339,   7.3524],
        [  6.0140,  -1.0661,  -2.5040],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.1348,   5.9514,  -0.8856],
        [ -7.2515,   5.0195,   4.6434],
        [  1.6954,  -4.7616,   9.8198],
        [ -7.7842,   4.3466,  -1.7548],
        [  5.3602,  -2.3808,   7.2098],
        [ 12.5763,   0.1333,  -9.2650],
        [ -6.1914,  -8.2997,   8.5380],
        [  9.3624,   9.5318,   0.2779],
        [ -2.6735, -10.4062,  -1.5911],
        [  1.7659,   7.4127,   4.7871],
        [  6.3211,   1.7017,  -0.2164],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -8.8058,   2.4850,  -1.7425],
        [ -7.6525,  -6.6534,   5.1721],
        [  5.8617,   3.4853,  -9.8080],
        [ -8.0965,  -5.9712,   4.3466],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -5.3736,   6.6310,  -1.5174],
        [  0.9788,  -4.8145,  -3.8661],
        [ 11.5568,   6.0602,   9.3244],
        [  7.5309,  11.2617,   7.2656],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.3413,   0.4883,  -6.8416],
        [  7.8982,  -8.6778,   0.0239],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.6202,  -1.9508,   0.7471],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.8704,  -9.3809,   7.5769],
        [ 12.3357,   3.6001,   0.7903],
        [  4.6357,   5.4415,   7.9067],
        [  6.7565,  11.8237,  -4.9882],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.7420,  -4.4896,  -3.7139],
        [ -9.0961,   1.6529,   3.1553],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.3928,   7.1404,  -1.7035],
        [ -1.0482, -10.8487,  -5.9066],
        [  2.4697,  -4.5836,   1.4593],
        [  6.0890,  -0.8382,   2.9166],
        [ 11.9641,  -2.9568,   7.7211],
        [  5.1667,  -2.6543,   1.2879],
        [  5.3804,  -2.3504,   7.2744],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -3.5173, -10.0623,   4.8280],
        [ 12.4837,   2.7510,   0.4449],
        [ -8.5506,  -5.1154,   6.3423]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0329, 0.0345,  ..., 1.4610, 0.6278, 1.5814],
        [0.0329, 0.0000, 0.0126,  ..., 1.4929, 0.6484, 1.6134],
        [0.0345, 0.0126, 0.0000,  ..., 1.4952, 0.6576, 1.6156],
        ...,
        [1.4610, 1.4929, 1.4952,  ..., 0.0000, 1.0380, 0.1223],
        [0.6278, 0.6484, 0.6576,  ..., 1.0380, 0.0000, 1.1582],
        [1.5814, 1.6134, 1.6156,  ..., 0.1223, 1.1582, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.9089559 , 13.1399181 , 13.01728313, 13.14777433, 13.15525803,
       13.26590071, 11.4088043 , 10.68416401,  8.20436764, 10.8207297 ,
       10.14044489,  9.47369628, 13.25336001,  9.41970031, 13.06498554,
       10.48840476,  4.91336994,  5.05981914, 13.96746816,  7.5158489 ,
        5.22903117,  9.38232121,  5.43544626,  4.91296205, 11.51557166,
       11.28895489, 10.06286074,  6.29821673, 11.22142195,  6.27715105,
        7.06372172,  7.46248947,  8.35856487, 10.57302546,  9.61942445,
        8.16584537, 13.02818702,  5.47702766,  5.83560009,  5.38342077,
       13.66896739,  8.92549752,  5.19986797,  5.86519525, 13.26209545,
        8.49161154, 10.06437598, 13.66896739,  8.85572292, 10.61472636,
        5.13616224, 11.27160035, 10.7506423 , 10.23946247,  6.55800267,
        9.5014146 ,  7.9337647 ,  8.12043983,  6.10774544, 13.14777433,
        7.24677664,  8.81930525,  5.05444288,  8.91551231,  5.86519525,
       12.5769659 , 10.35464916, 13.3607379 , 10.74409834,  7.62011734,
        6.54615887, 12.10942412,  9.14973319, 10.14044489,  6.81961996,
       10.06023665, 10.14436912,  8.53501777,  4.91296205, 13.0493368 ,
       13.54774355, 10.60226614,  6.36004393, 11.73397309,  6.27905292,
        5.94909709,  5.27178363, 11.62770332, 12.85032836,  7.14837514,
       13.61801526, 10.8207297 ,  5.26067019,  9.24501829, 10.33958639,
        7.53063885, 10.8992524 ,  5.20659892,  6.1463896 , 12.32408516,
        5.80860165,  5.87132718, 10.39942497, 10.65929968, 12.78327037,
        9.96391045])}
{'P': tensor([[5.7655e-03, 2.1429e-01, 1.8275e-01,  ..., 4.1804e-05, 9.1363e-08,
         8.5745e-07],
        [2.1835e-05, 1.2453e-02, 3.8709e-01,  ..., 1.3790e-04, 2.0467e-07,
         3.6110e-06],
        [9.6739e-06, 8.0924e-05, 1.5062e-02,  ..., 3.6371e-04, 3.3535e-07,
         6.1602e-06],
        ...,
        [3.9088e-06, 4.4572e-06, 6.5295e-06,  ..., 8.5073e-02, 8.3255e-04,
         2.2550e-03],
        [1.5550e-05, 5.6879e-06, 4.4142e-06,  ..., 1.1751e-05, 7.2644e-02,
         3.9420e-05],
        [4.3798e-06, 2.9797e-06, 2.4048e-06,  ..., 2.9860e-05, 9.5230e-03,
         9.6220e-02]]), 'X': tensor([[ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.2303e+01, -1.7440e+00, -6.6632e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 7.8537e-02, 4.0976e-02,  ..., 8.1213e-01, 1.2274e+00,
         1.2293e+00],
        [7.8537e-02, 0.0000e+00, 5.2149e-02,  ..., 8.8879e-01, 1.2936e+00,
         1.2765e+00],
        [4.0976e-02, 5.2149e-02, 0.0000e+00,  ..., 8.4971e-01, 1.2682e+00,
         1.2684e+00],
        ...,
        [8.1213e-01, 8.8879e-01, 8.4971e-01,  ..., 0.0000e+00, 5.4440e-01,
         8.1988e-01],
        [1.2274e+00, 1.2936e+00, 1.2682e+00,  ..., 5.4440e-01, 3.0339e-04,
         4.4887e-01],
        [1.2293e+00, 1.2765e+00, 1.2684e+00,  ..., 8.1988e-01, 4.4887e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.91143491, 13.1399181 , 13.26590071, 13.09737053, 13.14777433,
       13.13888841,  6.28835101,  9.17016857, 12.53285633,  8.57560275,
        7.06272256,  8.67502348, 11.04953767,  5.92281142,  7.59795445,
        8.29640613,  6.60752373,  4.7663526 ,  5.24137909,  7.71075226,
        9.79814997,  7.62011734, 11.75807252,  8.85572292,  5.9510489 ,
       12.889695  , 13.60275211,  9.55593374,  5.29009836,  7.50500064,
        6.84796202, 10.01804223, 12.05300943,  9.50654967, 11.22142195,
        5.19065439,  9.17905491, 11.38774149,  6.05113217,  7.24326992,
        5.87132718, 11.05672136,  9.24501829, 13.66051884,  5.53255882,
        6.33517349, 13.59056613, 12.42554808,  8.09842704,  9.99145096,
        6.05165248, 11.018084  ,  5.05774295,  5.51177501,  9.51074763,
       11.05413545, 12.4013153 ,  5.91222346, 12.1494023 ,  5.39723241,
        7.53063885,  8.04664886,  6.33075682,  8.09984231,  5.46244988,
        7.50917551, 12.0495521 ,  8.29640613, 11.66453059,  7.5984809 ,
       11.87412439, 12.78327037,  7.24326992, 13.66896739,  6.48486171,
        5.92281142,  9.50913849,  9.20340031,  6.81084724,  8.36445758,
       11.51811603,  5.51177501, 13.08414689,  4.74468406, 12.46513646,
       12.91143491, 13.97779839,  6.99097013, 11.1341798 ,  7.34250948,
       11.2845608 , 13.94008948,  7.6374603 , 13.96746816,  4.75303127,
        8.49161154,  7.81509631,  6.02789634, 10.71162681,  9.05843201,
        4.75303127,  5.5366698 ,  6.65991877,  8.31625171, 10.33712258,
        9.51074763])}
{'P': tensor([[2.6037e-03, 2.2553e-01, 1.7201e-01,  ..., 1.7486e-06, 5.9690e-08,
         4.0861e-06],
        [8.0788e-06, 3.0026e-03, 4.5386e-01,  ..., 1.4314e-06, 4.1802e-08,
         5.5505e-06],
        [1.0522e-05, 1.9688e-05, 5.3802e-03,  ..., 2.2519e-06, 6.2196e-08,
         1.0677e-05],
        ...,
        [5.4456e-03, 3.6223e-03, 3.0537e-03,  ..., 8.6401e-02, 1.0054e-03,
         8.2114e-05],
        [2.1135e-03, 1.0804e-03, 8.2706e-04,  ..., 1.3883e-03, 9.7855e-02,
         3.9889e-05],
        [1.9264e-06, 2.0287e-06, 2.1102e-06,  ..., 4.4739e-05, 1.1928e-05,
         9.6895e-02]]), 'X': tensor([[ 12.1379,   4.4125,  -3.8833],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.3118,   6.5957,  -4.5667],
        [ 11.6544,   5.8273,  -3.1375],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.9808,   4.9340,  -3.3715],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.2026,  12.1774,  -1.5887],
        [ -4.8751,  -9.3148,   4.0406],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -1.6334, -10.7207,   7.1397],
        [  5.3858,   4.4260,  -9.6769],
        [  6.3551,   0.7149,   1.3344],
        [ -0.9715, -10.8630,   0.3555],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.3858,   4.4260,  -9.6769],
        [  2.7121,   6.9743,  -8.4202],
        [  0.9074,  -4.8139,   8.1674],
        [ -1.0482, -10.8487,  -5.9066],
        [  0.1287, -11.0065,   6.5757],
        [  3.3839,  -4.1966,   8.3574],
        [  6.0073,   3.0986,   6.7937],
        [ -9.3616,  -2.4750,   3.7765],
        [ 11.1085,  -4.9204,   0.0736],
        [  6.2036,   2.4030,  -1.2500],
        [  5.8323,  -1.5228,   5.1132],
        [  2.0449,  13.8402,  -7.1642],
        [  6.0305,   3.0296,   0.1375],
        [ 12.4691,   2.8525,  -5.5110],
        [ -2.6735, -10.4062,  -1.5911],
        [ -7.6650,   4.5092,   0.2520],
        [  2.7685, -10.8982,  -6.8928],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -5.9940,   6.1985,   5.9566],
        [  6.2902,   0.0451,   3.7116],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.6202,  -1.9508,   0.7471],
        [  2.2076,  13.8032,  -1.5099],
        [ -1.0407,   7.9166,  -6.6021],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.4267,  -5.3697,   2.4172],
        [  9.3624,   9.5318,   0.2779],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.2659,  13.7894,   6.5436],
        [  7.1821,  -9.1845,   9.8548],
        [ -4.7389,   6.9953,   3.7638],
        [ -1.8367,   7.8718,  -8.6523],
        [  2.2849, -10.9655,  -7.6542],
        [ -8.0965,  -5.9712,   4.3466],
        [ 12.5199,  -0.4711,   3.0436],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.3236,   0.3011,  -4.6668],
        [ 12.5199,  -0.4711,   3.0436],
        [  6.3505,   0.6235,  -8.5849],
        [ -9.4735,  -0.7274,  -2.6749],
        [  0.6933,   7.7336,   2.1179],
        [ 11.1469,   6.9237,   1.6160],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.1178,   1.5784,   2.1410],
        [  5.4429,  -2.2529,   9.7306],
        [  7.0171,  11.6439,  -5.0178],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.4426, -10.1181,  -5.1268],
        [  3.0876,   6.7527,  -5.3540],
        [ -7.0120,   5.2825,   7.7762],
        [  8.8091,  -7.9037,   0.3631],
        [ 10.9916,   7.2133,  -0.9179],
        [ -5.1974,   6.7396,   3.3484],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.2517,  -4.6495,   2.9800],
        [  9.3607,   9.5337,  -4.5412],
        [  6.3550,   0.7129,   0.4917],
        [  6.0560,  -0.9421,  -5.0334],
        [ -3.0008, -10.2827,  -8.9480],
        [ -7.4602,   4.7718,  -4.6277],
        [  2.1771, -10.9776,  -3.2007],
        [ 11.2939,   6.6325,  -5.8311],
        [  4.1999,  -3.6543,  -4.1998],
        [ -1.9600,   7.8577,   6.9121],
        [  0.9788,  -4.8145,  -3.8661],
        [  5.2608,  -2.5256,  -9.1446],
        [ -1.6334, -10.7207,   7.1397],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.5927,   0.4199,  -9.4997],
        [  9.0652,   9.8597,  -3.5949],
        [ -1.5451,   7.8975,   7.6104],
        [  4.0642,  -3.7599,  -3.4565],
        [ -5.3736,   6.6310,  -1.5174],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.3305,   1.6119,   1.1538],
        [  2.0449,  13.8402,  -7.1642],
        [  1.6308,  13.9254,  -2.0984],
        [ -9.0961,   1.6529,   3.1553],
        [  2.1379,  -4.6757,   6.4981],
        [  3.8182,  -3.9347,  -1.7961],
        [  8.7131,  10.2216,   7.7915],
        [  5.0233,   4.9647,  -3.1788],
        [  5.8318,   3.5567,  -4.7692],
        [ 10.1503,   8.5414,  -7.2681],
        [  1.7659,   7.4127,   4.7871],
        [  3.8182,  -3.9347,  -1.7961],
        [  7.8425,  -8.7202,   6.4530],
        [ -3.4986,   7.5152,   9.0712]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.5067e-02, 7.4743e-02,  ..., 3.3313e-01, 1.0607e+00,
         1.0912e+00],
        [5.5067e-02, 3.0339e-04, 1.9927e-02,  ..., 3.8621e-01, 1.1157e+00,
         1.1463e+00],
        [7.4743e-02, 1.9927e-02, 0.0000e+00,  ..., 4.0609e-01, 1.1351e+00,
         1.1659e+00],
        ...,
        [3.3313e-01, 3.8621e-01, 4.0609e-01,  ..., 0.0000e+00, 7.5668e-01,
         7.7723e-01],
        [1.0607e+00, 1.1157e+00, 1.1351e+00,  ..., 7.5668e-01, 0.0000e+00,
         7.7365e-02],
        [1.0912e+00, 1.1463e+00, 1.1659e+00,  ..., 7.7723e-01, 7.7365e-02,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.91505111, 13.04217487, 13.09426142, 13.03002338, 13.20597932,
       12.95702647,  8.83725081, 13.6660671 , 10.51340829,  9.50654967,
       10.84437705,  6.97106912,  6.39520397, 10.90638066,  5.84533345,
        6.97106912,  7.48311002,  4.89869205, 10.8992524 , 11.00726483,
        5.39098669,  6.7593917 ,  9.68324553, 12.1494023 ,  6.652742  ,
        6.02779201, 13.99048094,  6.74872981, 12.79126785, 10.74409834,
        8.89301516, 11.24434374, 10.61047596,  8.62260327,  6.29034958,
        4.90505566,  5.94909709, 13.97857591,  7.98469166,  6.83515201,
        9.99211663, 13.3607379 ,  4.76969364, 13.97430191, 11.65923396,
        8.44940661,  8.08320235, 11.20100795, 10.06023665, 12.52875793,
        9.37989631,  6.33075682, 12.52875793,  6.38105015,  9.5014146 ,
        7.76457095, 13.12218013,  9.0368458 ,  9.25336371,  5.89074104,
       13.59480906,  8.5724042 ,  9.51074763, 13.1399181 , 11.4890861 ,
        7.42512621,  8.77913428, 11.83509372, 13.1471313 ,  8.51088913,
        7.02326515, 12.17451685, 13.36092624,  6.3949043 ,  6.12886024,
       10.71162681,  8.85572292, 11.1913598 , 13.09737053,  5.5671389 ,
        8.09842704,  4.91296205,  5.83560009, 10.84437705,  9.37989631,
       12.59840193, 12.59969936, 13.39372426,  8.04718433,  5.5366698 ,
        8.53501777,  9.79039371,  6.53250937, 13.99048094, 14.02058621,
        9.24501829,  5.14124612,  5.48276038, 13.43127131,  7.06272256,
        6.83083863, 13.26590071,  7.62011734,  5.48276038, 11.72802491,
        8.28968067])}
{'P': tensor([[2.4062e-03, 3.1566e-01, 1.3851e-01,  ..., 7.9061e-04, 5.0254e-06,
         4.3605e-06],
        [9.0664e-06, 2.7115e-03, 1.8737e-01,  ..., 1.2830e-03, 7.4916e-06,
         3.8166e-06],
        [7.0723e-06, 1.3203e-05, 2.9279e-03,  ..., 1.9191e-03, 1.3047e-05,
         3.1281e-06],
        ...,
        [7.2477e-07, 9.2854e-07, 1.3115e-06,  ..., 2.7902e-02, 5.5262e-05,
         1.8635e-06],
        [1.1544e-06, 1.3234e-06, 1.7145e-06,  ..., 5.2754e-06, 5.0145e-02,
         3.2882e-05],
        [5.4137e-03, 5.4590e-03, 5.7603e-03,  ..., 1.9620e-04, 3.5400e-04,
         6.3043e-02]]), 'X': tensor([[ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 1.8447e+00, -4.7372e+00,  8.1235e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0248, 0.0868,  ..., 0.2396, 1.0587, 0.4674],
        [0.0248, 0.0000, 0.0644,  ..., 0.2614, 1.0543, 0.4597],
        [0.0868, 0.0644, 0.0000,  ..., 0.3258, 1.0149, 0.4157],
        ...,
        [0.2396, 0.2614, 0.3258,  ..., 0.0000, 1.2289, 0.6679],
        [1.0587, 1.0543, 1.0149,  ..., 1.2289, 0.0000, 0.6007],
        [0.4674, 0.4597, 0.4157,  ..., 0.6679, 0.6007, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.95702647, 13.03002338, 13.12299145, 13.20597932, 13.2430377 ,
       13.18278606,  9.77588801, 12.38881635, 13.1399181 ,  7.39004226,
       11.54071859,  7.06372172,  8.30729352,  7.27340834, 12.27991991,
        8.29640613, 10.80728788, 13.01187484, 13.62896593,  9.58072286,
       13.70893492,  5.34250225,  4.978704  , 11.83570132,  5.16077764,
       12.88270018,  8.86082124, 11.42947632, 10.06023665,  6.65368551,
        8.85939626,  6.30647746,  8.859416  , 13.1471313 , 13.44593519,
        9.96152788, 12.07818871,  5.5366698 ,  7.33350642, 11.32149259,
       14.07623266,  6.28961988,  6.39438044, 11.35182589, 13.45124917,
        5.24461068,  9.0368458 , 10.31613269, 13.83139024, 13.04156075,
       12.27991991,  8.12043983, 11.24434374, 11.20100795,  6.16512856,
        7.89745107, 14.05633441,  6.54127431, 11.11969815, 11.28895489,
       11.37939695,  6.74686071,  5.92952538,  7.54620849,  5.08372149,
        9.93553424,  5.87132718,  9.50913849,  7.44805901, 12.1494023 ,
        7.37576154,  4.91336994, 12.4217523 ,  8.11227668, 10.35464916,
        8.27853076, 13.36092624,  6.28835101, 10.33958639,  6.80559669,
        5.52416368,  5.14124612, 13.46553623, 11.87665501,  7.37576154,
        5.47866221, 13.77222752, 10.61472636,  7.42512621, 10.7167285 ,
        7.60529892, 13.32374056, 13.36092624,  7.48538905,  8.95983586,
       12.7125079 ,  7.39004226,  6.38399789,  8.09704052, 14.09357547,
        6.02588517, 10.63038565,  9.0368458 , 13.99048094,  8.28968067,
        6.01356004])}
{'P': tensor([[2.0756e-03, 3.2502e-01, 1.6621e-01,  ..., 3.1055e-07, 1.1997e-07,
         6.7257e-08],
        [1.5462e-05, 3.7320e-03, 1.1920e-01,  ..., 6.1273e-07, 2.3012e-07,
         1.2634e-07],
        [1.4293e-05, 4.7584e-05, 3.4650e-03,  ..., 6.6847e-07, 2.4779e-07,
         1.3351e-07],
        ...,
        [1.3506e-06, 1.2434e-06, 1.1402e-06,  ..., 3.0982e-02, 2.3126e-01,
         9.9933e-02],
        [1.9772e-06, 1.7654e-06, 1.5954e-06,  ..., 2.5468e-04, 3.0990e-02,
         5.5017e-01],
        [4.7825e-06, 4.1838e-06, 3.7327e-06,  ..., 1.4566e-04, 4.1244e-04,
         4.9814e-02]]), 'X': tensor([[ 11.9122,   5.1397,  -1.5498],
        [ 11.7418,   5.6070,  -1.4192],
        [ 11.4743,   6.2478,  -2.1100],
        [ 10.9916,   7.2133,  -0.9179],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.1420,   6.9331,  -2.4779],
        [  9.6587,   9.1817,  -0.8212],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.3534,   1.2990,  -7.2942],
        [  5.8889,  -1.3916,   8.8993],
        [  1.3767, -11.0341,   9.9415],
        [ 10.0619,   8.6627,  -9.2884],
        [  5.1861,   4.7361,  -3.7312],
        [  3.2366,  -4.2732,   5.1532],
        [  8.2422,  -8.4041,  -1.4962],
        [  2.0419,  -4.6978,  -5.0983],
        [ 12.3357,   3.6001,   0.7903],
        [  1.7815,  -4.7481,   4.4289],
        [  1.1139,   7.6281,  -2.5680],
        [  6.2535,  -0.1709,  -9.4102],
        [ 12.6035,   0.7494,   7.1536],
        [ -8.1531,  -5.8748,   2.6934],
        [  0.6396,  14.0772,  -6.3786],
        [  6.3301,   0.3638,  -7.5685],
        [ 12.2311,  -2.0445,   1.3545],
        [  5.0277,   4.9589,   7.0175],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -0.7599, -10.8995,  -6.0366],
        [  4.4633,   5.6287,  -0.7754],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.8058,   2.4850,  -1.7425],
        [ -0.6323,   7.9085,  -5.9544],
        [  2.2849, -10.9655,  -7.6542],
        [  6.3534,   1.2990,  -7.2942],
        [  6.2038,   2.4020,  -3.1692],
        [  5.9587,  12.3216,   9.1695],
        [  1.4489,   7.5256,   4.4529],
        [ -3.0644,   7.6439,  -1.4322],
        [  7.0171,  11.6439,  -5.0178],
        [  0.9478,  -4.8144,   0.6371],
        [ -6.4822,  -8.0252,  -3.8983],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.1222,  -0.7261,   8.8204],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -5.5525,   6.5146,  -4.4984],
        [  6.2247,  -0.3150,  -9.3657],
        [  2.5469,  13.7192,  -9.6935],
        [ -9.3616,  -2.4750,   3.7765],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.8323,  -1.5228,   5.1132],
        [ -8.0534,  -6.0429,   2.4583],
        [ 12.4690,  -0.8528,  -9.3374],
        [  0.1532,  -4.7422,  -7.0145],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.3587,   1.1596,  -3.6629],
        [  2.1440, -10.9811,   8.5281],
        [  5.7963,  -1.6018,   1.8856],
        [ -2.6485, -10.4151,   0.8766],
        [ -7.3554,   4.8985,  -0.6665],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.3087,   1.8053,  -1.9481],
        [  7.0358,  -9.2784,  -9.8479],
        [  2.0547,   7.2957,  -1.9343],
        [  4.4633,   5.6287,  -0.7754],
        [ -2.0445,   7.8469,  -0.5857],
        [  3.2647, -10.8066,  -0.3768],
        [ -1.6334, -10.7207,   7.1397],
        [  2.2437,  -4.6489,   6.1430],
        [  2.9470,  13.6085,   2.9809],
        [ 12.1504,   4.3674,  -7.2998],
        [  7.2340,  -9.1505,  -5.8321],
        [ -3.5903,   7.4847,  -8.9112],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.4366,  -0.1189,  -2.6736],
        [  1.2273,  -4.8081,   9.8129],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.2242,  -0.3171,   7.4978],
        [  3.9663,   6.1018,  -5.4104],
        [ 11.5568,   6.0602,   9.3244],
        [  5.2608,  -2.5256,  -9.1446],
        [ -1.1885, -10.8211,   0.0865],
        [  0.6952,  -4.8060,   8.8976],
        [ -2.0445,   7.8469,  -0.5857],
        [  5.4867,  12.5816,   2.9549],
        [  6.3534,   1.2990,  -7.2942],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.0140,  -1.0661,  -2.5040],
        [  5.1196,  -2.7160,   4.2705],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.9410,   6.8428,   5.4458],
        [  3.3633,  13.4794,   0.5193],
        [  1.8721,  -4.7322,   0.5524],
        [  2.8602,   6.8905,   5.8779],
        [  2.4149,  -4.6005,  -4.3339],
        [  1.9503,  -4.7172,  -6.2313],
        [ -7.9298,   4.1367,  -3.9773],
        [  5.8617,   3.4853,  -9.8080],
        [ 11.2209,  -4.7092,   2.6117],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.0750,  -6.0072,  -5.8267]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0192, 0.0594,  ..., 1.1637, 1.2417, 1.2432],
        [0.0192, 0.0000, 0.0672,  ..., 1.1789, 1.2580, 1.2597],
        [0.0594, 0.0672, 0.0000,  ..., 1.1869, 1.2598, 1.2595],
        ...,
        [1.1637, 1.1789, 1.1869,  ..., 0.0000, 0.1357, 0.1673],
        [1.2417, 1.2580, 1.2598,  ..., 0.1357, 0.0000, 0.0374],
        [1.2432, 1.2597, 1.2595,  ..., 0.1673, 0.0374, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([12.97370371, 13.01187484, 13.06498554, 13.1471313 , 13.18278606,
       13.12299145, 13.32645552,  5.07306274,  9.14973319,  6.48486171,
        6.05113217, 11.11969815, 13.27717982,  7.02326515,  5.36061974,
       11.77124373,  5.1224079 , 12.85032836,  5.0713364 ,  7.70897675,
        6.25586297, 12.62576026, 10.04916678, 14.09176259,  6.34059021,
       12.40074901,  7.06169391, 10.4977132 , 10.92597196,  7.18354885,
       10.794755  ,  9.37989631,  9.14973319,  7.9337647 , 11.20100795,
        6.48486171,  6.65259775, 13.68672855,  7.66377554,  8.23525433,
       13.59480906,  4.90676242, 10.31613269,  5.47866221,  6.16512856,
       12.11735334,  8.55982105,  6.23262289, 13.95361098,  9.68324553,
        4.91652556,  6.02779201, 10.06850677, 12.4980845 ,  4.74468406,
       13.00413829,  6.46356806, 11.18839169,  6.01356004, 10.74655346,
        8.83725081,  9.7751189 ,  6.5618968 , 11.64436684,  7.57946459,
        7.18354885,  8.10886021, 11.28895489, 10.84437705,  5.16205087,
       13.92390689, 12.91143491, 11.66453059,  8.30125201,  9.69792513,
        9.11473391,  9.26455795,  9.43737523,  4.96231532,  9.97964013,
        6.2322769 ,  7.27759233, 13.0493368 ,  5.83560009, 10.88618114,
        4.85603886,  8.10886021, 13.72594022,  6.48486171,  9.78286733,
       10.60226614,  6.10774544,  5.79543038, 12.00995898,  7.44805901,
       13.89264822,  5.08910491,  7.46053281,  5.19577146,  5.10444097,
        8.94395628,  6.81961996, 12.16901615,  9.5642993 ,  9.85330646,
       10.06437598])}
{'P': tensor([[3.7650e-03, 2.5572e-01, 2.3435e-01,  ..., 3.1271e-04, 4.3059e-08,
         1.3540e-06],
        [1.1098e-05, 4.5016e-03, 6.0681e-05,  ..., 2.3703e-04, 5.6978e-08,
         6.7828e-07],
        [1.0297e-05, 2.4242e-01, 3.6394e-03,  ..., 2.1069e-04, 4.7414e-08,
         5.6997e-07],
        ...,
        [2.7646e-05, 1.2035e-04, 1.1568e-04,  ..., 1.1456e-02, 1.1429e-06,
         1.0677e-06],
        [1.6311e-05, 1.6477e-05, 1.7180e-05,  ..., 2.0103e-04, 1.8517e-01,
         3.1678e-04],
        [5.1309e-02, 4.5382e-02, 4.8284e-02,  ..., 3.9590e-02, 5.1411e-06,
         4.3062e-02]]), 'X': tensor([[ 11.7794,   5.5086,  -9.6929],
        [ 10.9394,   7.3069,  -8.9365],
        [ 10.9875,   7.2207,  -8.8549],
        [ 10.0619,   8.6627,  -9.2884],
        [ 11.0373,   7.1299,  -8.5895],
        [  8.8474,  10.0868,  -8.8644],
        [ -0.5709,   7.9055,   7.6089],
        [ -3.4986,   7.5152,   9.0712],
        [ -4.4503,   7.1378,   3.3751],
        [  5.6414, -10.0303,   7.8525],
        [ -8.7078,   2.7182,   1.8146],
        [ -9.3419,   0.5972,  -1.2841],
        [  8.7715,  -7.9390,   5.2900],
        [  0.3715,  -4.7756,   0.2312],
        [ -7.8359,   4.2735,   7.2943],
        [ -1.1885, -10.8211,   0.0865],
        [ -6.4620,  -8.0449,   6.2719],
        [ 10.0995,  -6.4739,   2.0974],
        [  6.3542,   0.6934,  -3.3840],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.3589,   0.8239,  -8.4230],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4739,  -1.2497,   0.5483],
        [ -7.7239,   4.4298,   3.1216],
        [  5.2199,   4.6861,  -1.9304],
        [ 12.5154,   2.5072,  -9.3274],
        [  1.8719,   7.3713,  -3.7465],
        [  0.9967,  -4.8144,  -9.7370],
        [ 12.5717,   1.9266,   7.4903],
        [  6.3087,   1.8053,  -1.9481],
        [  4.4633,   5.6287,  -0.7754],
        [  5.0348,  -2.8229,   3.5436],
        [  2.0449,  13.8402,  -7.1642],
        [ -5.3328,  -8.9992,   7.3440],
        [  5.2911,  -2.4824,  -3.6877],
        [  3.9663,   6.1018,  -5.4104],
        [ -9.4203,  -2.0340,   2.5856],
        [  6.3606,   0.9204,   3.2525],
        [  4.4290,  -3.4596,   1.4181],
        [ -4.4503,   7.1378,   3.3751],
        [  9.1620,  -7.5573,  -9.9882],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.6672,  -1.8631,   8.1020],
        [  5.9843,  -9.8679,  -6.1449],
        [  3.1271, -10.8343,   2.6442],
        [ -9.4671,  -0.5553,   0.1117],
        [  0.3724,   7.7974,  -6.8722],
        [  5.8318,   3.5567,  -4.7692],
        [  2.7420,  -4.4896,  -3.7139],
        [ 11.2939,   6.6325,  -5.8311],
        [ -5.8389,  -8.6056,  -3.5098],
        [  0.9809,  -4.8145,   6.8706],
        [  7.0358,  -9.2784,  -9.8479],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.2508,   1.0572,   2.9978],
        [ -2.9497, -10.3028,  -7.3479],
        [  1.7903,  -4.7467,   0.1168],
        [  5.0233,   4.9647,  -3.1788],
        [  6.8407,  11.7666,   3.0045],
        [  2.6451,  -4.5250,  -0.5565],
        [ 12.4013,  -1.2597,  -7.6474],
        [  8.3718,  -8.2953,   8.4666],
        [  5.6314,  -1.9302,  -9.6543],
        [  7.1821,  -9.1845,   9.8548],
        [  0.5250, -11.0306,   5.5358],
        [  2.1625,  -4.6697,   7.0658],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.5010,  -4.5736,  -0.8989],
        [  4.3523, -10.5232,   8.2138],
        [  6.3087,   1.8053,  -1.9481],
        [ -1.5408,   7.8978,   5.3470],
        [  6.3460,   1.4252,   1.2291],
        [  3.9515,  -3.8425,   6.1729],
        [  6.2804,   2.0019,   0.8607],
        [ -8.5607,  -5.0938,  -5.4531],
        [  3.2990,  -4.2415,  -7.4857],
        [  2.0729,   7.2878,   7.0880],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.0042,  -2.8350,   2.6864],
        [  7.0515,  11.6194,   8.4289],
        [ -4.7389,   6.9953,   3.7638],
        [ -7.6719,  -6.6260,   7.1468],
        [  3.2692,  13.5098,  -4.9032],
        [  9.6614,   9.1784,  -1.2087],
        [  8.6271,  -8.0715,  -7.4813],
        [  9.4516,  -7.2501,  -1.8868],
        [  2.2437,  -4.6489,   6.1430],
        [  7.4132,  11.3526,   0.3322],
        [  5.6928,  12.4710,   3.6001],
        [  1.8719,   7.3713,  -3.7465],
        [  4.1297,  -3.7098,  -3.9838],
        [  3.3069,   6.6089,   8.7377],
        [ -1.5875,   7.8944,  -9.4286],
        [ -1.9600,   7.8577,   6.9121],
        [ -2.0445,   7.8469,  -0.5857],
        [ 12.3453,  -1.5462,   7.0212],
        [  9.4516,  -7.2501,  -1.8868],
        [  8.5198,  10.4091,  -8.6486],
        [ -8.6112,   2.9319,  -7.9943],
        [  9.6662,   9.1725,   8.1056],
        [  5.5424,  -2.0885,   7.2487],
        [ -8.7409,  -4.6837,  -6.5840],
        [ 10.2470,   8.4054,  -0.1177],
        [  0.5850,  -4.7982,   9.6639],
        [ 12.2921,  -1.7902,  -4.0870]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0160, 0.0128,  ..., 0.6118, 1.1993, 0.3377],
        [0.0160, 0.0000, 0.0041,  ..., 0.6148, 1.2062, 0.3498],
        [0.0128, 0.0041, 0.0000,  ..., 0.6118, 1.2026, 0.3457],
        ...,
        [0.6118, 0.6148, 0.6118,  ..., 0.0000, 0.6283, 0.4062],
        [1.1993, 1.2062, 1.2026,  ..., 0.6283, 0.0000, 0.9000],
        [0.3377, 0.3498, 0.3457,  ..., 0.4062, 0.9000, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.00380624, 13.15525803, 13.14777433, 13.27717982, 13.1399181 ,
       13.41713661,  7.92607342,  8.28968067,  8.41148722, 11.50791859,
        9.12220271,  9.36093562, 11.83075005,  4.79003361,  8.92549752,
       10.88618114, 10.31886066, 11.99633819,  6.39187285, 13.00413829,
        6.41202986,  7.5092834 ,  9.55593374,  8.90405823,  7.01475376,
       12.76408562,  7.60529892,  4.91652556, 12.71843462,  6.5618968 ,
        7.18354885,  5.77219077, 13.99048094, 10.4606265 ,  5.84450127,
        7.27759233,  9.63742669,  6.42689705,  5.62005629,  8.41148722,
       11.87665501, 10.38897563,  5.96556248, 11.54071859, 11.27656488,
        9.48337063,  7.80625981,  6.83083863,  5.26067019, 13.09737053,
       10.39942497,  4.91336994, 11.64436684, 13.02818702,  9.3109884 ,
       10.7167285 ,  5.07306274,  7.06272256, 13.6105595 ,  5.24137909,
       12.46513646, 11.785566  ,  5.95297459, 11.65923396, 11.04313641,
        5.14607014,  9.78147542,  5.21280156, 11.38774149,  6.5618968 ,
        8.04664886,  6.50409372,  5.51177501,  6.59176018,  9.96152788,
        5.37343173,  7.57685908, 10.06437598,  8.44940661, 12.3344536 ,
       13.59170809,  8.44940661, 10.1371552 , 13.89974693, 13.32613445,
       11.81423161, 11.9120321 ,  5.16205087, 13.55868684, 13.70893492,
        7.60529892,  5.5513072 ,  7.39004226,  8.05242591,  8.09842704,
        8.10886021, 12.44177641, 11.9120321 , 13.45124917,  9.09661212,
       13.32557133,  5.92281142,  9.91668182, 13.25336001,  4.83371341,
       12.4217523 ])}
{'P': tensor([[1.9669e-03, 3.3007e-01, 1.9849e-01,  ..., 2.9922e-08, 2.6016e-08,
         6.9111e-07],
        [8.0602e-06, 2.6434e-03, 2.5980e-01,  ..., 3.5132e-08, 3.5570e-08,
         8.1024e-07],
        [7.4021e-06, 1.4443e-05, 2.8919e-03,  ..., 3.4939e-08, 3.8649e-08,
         8.1666e-07],
        ...,
        [2.8069e-04, 2.0694e-04, 1.6359e-04,  ..., 1.8924e-02, 2.0570e-06,
         4.2422e-03],
        [6.8874e-06, 5.3698e-06, 4.0843e-06,  ..., 5.2578e-04, 2.5552e-02,
         1.0055e-04],
        [3.4866e-03, 3.0534e-03, 2.9016e-03,  ..., 3.2148e-03, 4.3357e-05,
         1.7926e-01]]), 'X': tensor([[ 11.7779,   5.5127,  -2.0699],
        [ 11.4743,   6.2478,  -2.1100],
        [ 11.1420,   6.9331,  -2.4779],
        [ 10.7566,   7.6212,  -2.3279],
        [ 10.3251,   8.2928,  -3.0006],
        [ 10.9916,   7.2133,  -0.9179],
        [  3.5284, -10.7484,   5.3347],
        [ 11.6019,   5.9540,  -5.1715],
        [  4.7482, -10.3907,   5.9149],
        [ -8.8835,  -4.3178,   9.3080],
        [  0.6933,   7.7336,   2.1179],
        [  0.3724,   7.7974,  -6.8722],
        [  6.2975,   0.0947,   5.1791],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.9003,  12.3550,   2.1917],
        [  5.6928,  12.4710,   3.6001],
        [  6.3200,   0.2690,   1.5900],
        [  0.1931,  -4.7491,   7.4385],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.1142,  -2.7230,  -4.1966],
        [  9.4925,   9.3811,   8.9842],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.1730,  -0.5375,  -8.8861],
        [ -9.3444,   0.5821,   6.9101],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.6523,   3.9443,  -2.9449],
        [ 11.0373,   7.1299,  -8.5895],
        [  1.9241,   7.3503,   7.9004],
        [ -9.3331,  -2.6465,  -8.2028],
        [  4.4915,   5.5990,   7.0405],
        [  5.1793,   4.7461,   0.3899],
        [  2.1440, -10.9811,   8.5281],
        [ -7.4807,   4.7463,   2.0616],
        [ -9.3087,   0.7792,  -7.5587],
        [ -6.4368,   5.8356,   3.3014],
        [ 12.0772,  -2.6014,  -7.1832],
        [  9.7328,   9.0902,  -0.8917],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -8.6112,   2.9319,  -7.9943],
        [ -6.0844,  -8.3955,   0.7705],
        [ -6.3489,   5.9117,   6.5161],
        [ -2.1533, -10.5778,   1.0112],
        [  3.4920,  -4.1366,  -8.3955],
        [ -9.2974,   0.8371,   2.4852],
        [ -6.3306,   5.9272,   5.6935],
        [ -1.6334, -10.7207,   7.1397],
        [  1.1006,   7.6318,  -2.4877],
        [  6.9653,  -9.3225,  -3.4727],
        [ -9.4283,  -1.9585,  -0.9129],
        [  9.0652,   9.8597,  -3.5949],
        [ -3.9858,  -9.8334,  -3.7323],
        [  1.3670,  -4.7990,   3.2603],
        [  6.3246,   1.6696,  -0.1544],
        [  5.3924, -10.1396,  -4.0863],
        [  4.6231,   5.4556,   8.4973],
        [  6.0467,  -9.8369,  -6.5816],
        [  1.4306,  13.9620,  -0.2075],
        [ 12.5832,   1.7543,  -7.6134],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.1531,  -5.8748,   2.6934],
        [ -5.6792,  -8.7352,   4.0318],
        [  0.1677,  -4.7447,  -6.1529],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.3246,   1.6696,  -0.1544],
        [  9.9750,  -6.6338,  -1.2440],
        [  7.9141,  -8.6656,   9.6948],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.4401,  13.4540,   4.6576],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.9314,  -9.3435,  -1.2729],
        [ -1.2454,   7.9127,  -0.5715],
        [  8.6271,  -8.0715,  -7.4813],
        [ 12.5996,   1.3971,  -1.0694],
        [  6.3246,   1.6696,  -0.1544],
        [  3.9085,   6.1512,  -2.2560],
        [ -2.5161,   7.7699,  -9.3531],
        [  6.1222,  -0.7261,   8.8204],
        [ -1.8788, -10.6567,   3.7068],
        [ -8.5900,   2.9767,   7.4811],
        [  3.7298,  13.3537,  -9.4653],
        [ 10.1503,   8.5414,  -7.2681],
        [ -9.4196,   0.0478,   9.4140],
        [  3.1271, -10.8343,   2.6442],
        [  5.1793,   4.7461,   0.3899],
        [  6.1741,  -0.5329,   3.1913],
        [  6.3360,   0.4258,  -9.2509],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.2975,   0.0947,   5.1791],
        [  3.5939,   6.4029,  -1.8357],
        [ -6.7992,   5.4991,   9.9186],
        [  4.5814, -10.4485,   2.5496],
        [  6.0890,  -0.8382,   2.9166],
        [  1.0093,   7.6567,   2.3526],
        [  0.2732,  -4.7619,  -1.1577],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.4573,   3.2448,   5.4370],
        [ 11.2571,   6.7072,   8.3693],
        [  2.5469,  13.7192,  -9.6935],
        [  4.3554,   5.7393,  -9.0766],
        [ -8.5149,  -5.1905,   1.5239],
        [  7.0171,  11.6439,  -5.0178],
        [ -7.5401,   4.6718,   9.0033],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.9452, -10.3187,  -7.8621],
        [ -4.8751,  -9.3148,   4.0406],
        [  3.7992,  -3.9474,  -8.9019]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0184, 0.0497,  ..., 0.6816, 1.2155, 0.3283],
        [0.0184, 0.0000, 0.0372,  ..., 0.6944, 1.2336, 0.3376],
        [0.0497, 0.0372, 0.0000,  ..., 0.6876, 1.2593, 0.3237],
        ...,
        [0.6816, 0.6944, 0.6876,  ..., 0.0000, 1.0119, 0.3730],
        [1.2155, 1.2336, 1.2593,  ..., 1.0119, 0.0000, 1.1507],
        [0.3283, 0.3376, 0.3237,  ..., 0.3730, 1.1507, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.00413829, 13.06498554, 13.12299145, 13.18278606, 13.2430377 ,
       13.1471313 , 11.31276109, 13.04051744, 11.42420508,  9.87720102,
        7.76457095,  7.80625981,  6.29821673, 10.61047596, 13.69162795,
       13.70893492,  6.32572323,  4.75303127, 11.86239961,  5.79391825,
       13.34587027, 10.37698435,  6.19633139,  9.36256096, 12.27993098,
        6.89245352, 13.1399181 ,  7.59795445,  9.70108409,  7.17789666,
        7.02496223, 11.18839169,  8.85939626,  9.34127013,  8.68832398,
       12.35421461, 13.31764181, 10.35485331,  9.09661212, 10.36844813,
        8.67502348, 10.794755  ,  5.41348121,  9.3349858 ,  8.67227386,
       10.84437705,  7.71075226, 11.63724792,  9.62958973, 13.39372426,
       10.61047596,  4.98989906,  6.54127431, 11.48434767,  7.15100303,
       11.54672755, 14.03505624, 12.70489563,  8.8737556 , 10.04916678,
       10.41909211,  4.74770899, 12.40074901,  6.54127431, 11.97949273,
       11.73567759,  8.05242591, 13.88683436, 10.61417181, 11.63383221,
        8.01009083, 11.81423161, 12.67680439,  6.54127431,  7.28793663,
        8.16714836,  6.16512856, 10.82106505,  9.09119351, 13.86480175,
       13.26590071,  9.41970031, 11.27656488,  7.02496223,  6.19707944,
        6.35029274, 10.06023665,  6.29821673,  7.34250948,  8.74470218,
       11.4088043 ,  6.1463896 ,  7.72291946,  4.76969364,  9.11473391,
        9.05843201, 13.103709  , 13.95361098,  7.20482224,  9.97220799,
       13.59480906,  8.87007495,  9.54154096, 11.4424816 , 10.51340829,
        5.47866221])}
{'P': tensor([[2.3207e-03, 2.0727e-01, 1.3724e-01,  ..., 3.6875e-08, 1.4805e-08,
         4.3508e-08],
        [4.4667e-06, 1.8798e-03, 2.3019e-01,  ..., 3.1851e-08, 1.2795e-08,
         2.5623e-08],
        [3.4442e-06, 7.8072e-06, 2.0124e-03,  ..., 3.5336e-08, 1.4506e-08,
         2.3939e-08],
        ...,
        [3.4962e-05, 2.0135e-05, 1.5610e-05,  ..., 2.3498e-01, 3.6103e-04,
         5.1104e-02],
        [1.1612e-05, 7.5381e-06, 6.2702e-06,  ..., 1.3043e-05, 5.1195e-02,
         6.6923e-04],
        [4.5742e-04, 2.0714e-04, 1.4277e-04,  ..., 2.3270e-06, 2.0996e-06,
         1.6342e-02]]), 'X': tensor([[ 11.7418,   5.6070,  -1.4192],
        [ 10.9916,   7.2133,  -0.9179],
        [ 10.5678,   7.9261,  -0.6100],
        [ 10.7566,   7.6212,  -2.3279],
        [  9.6614,   9.1784,  -1.2087],
        [  9.7328,   9.0902,  -0.8917],
        [  7.7750,  11.0665,   2.8364],
        [  3.0719, -10.8449,  -4.1673],
        [ -8.3900,   3.3727,   3.1650],
        [  0.3493,  -4.7727,  -6.5682],
        [ -0.4341, -10.9473,   7.4509],
        [  2.2076,  13.8032,  -1.5099],
        [  5.9843,  -9.8679,  -6.1449],
        [  9.0051,   9.9234,   5.4411],
        [ 11.2767,  -4.6006,   0.7868],
        [  6.2675,  12.1379,  -8.9720],
        [ -3.7964,  -9.9294,   7.7147],
        [  9.5018,  -7.1945,  -1.1892],
        [  1.7815,  -4.7481,   4.4289],
        [  5.1787,  -2.6383,  -4.2533],
        [  8.6420,  -8.0580,   6.5781],
        [  2.0039,   7.3173,  -1.5885],
        [ -4.9217,   6.8979,  -1.7667],
        [  2.5813,  -4.5471,   5.3597],
        [  6.2630,   2.1057,  -5.2136],
        [  5.8902,  -1.3887,   4.0388],
        [ -5.9872,   6.2037,  -9.4477],
        [ 12.4592,  -0.9168,  -6.7571],
        [  9.1620,  -7.5573,  -9.9882],
        [ -9.3810,  -2.3455,  -9.5194],
        [  4.6992,   5.3690,   3.0935],
        [ 11.1469,   6.9237,   1.6160],
        [ -0.0410,   7.8591,   9.3457],
        [  1.5662,  13.9375,   4.5142],
        [  7.7750,  11.0665,   2.8364],
        [  8.3718,  -8.2953,   8.4666],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.2437,  -4.6489,   6.1430],
        [  2.1120,  -4.6818,   8.6144],
        [  6.1730,  -0.5375,  -8.8861],
        [ -6.8618,  -7.6322,   3.7027],
        [  2.5663,   7.0526,  -6.8769],
        [  5.3858,   4.4260,  -9.6769],
        [ -8.0829,  -5.9940,   5.0564],
        [ -9.4501,  -1.7133,   2.2980],
        [ 11.2847,  -4.5848,  -4.7503],
        [  0.9659,  -4.8144,  -0.9594],
        [ -8.6801,   2.7809,  -0.8443],
        [ -9.3384,  -2.6161,  -4.7630],
        [  9.3424,  -7.3685,  -3.6807],
        [ 12.2311,  -2.0445,   1.3545],
        [  6.3551,   0.7149,   1.3344],
        [  5.6260,  -1.9401,   3.0259],
        [  7.8425,  -8.7202,   6.4530],
        [  1.7815,  -4.7481,   4.4289],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.3301,   0.3638,  -7.5685],
        [  0.9392,  -4.8143,  -3.4649],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.0277,   4.9589,   7.0175],
        [ -8.9178,   2.1936,   7.5388],
        [  6.1579,  -0.5961,  -6.5595],
        [ 11.4743,   6.2478,  -2.1100],
        [  9.6816,   9.1536,  -5.8543],
        [ -4.4779,  -9.5609,  -2.4505],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.0073,   3.0986,   6.7937],
        [  4.0457,  -3.7737,  -0.4704],
        [ -0.5709,   7.9055,   7.6089],
        [ -9.4524,  -1.6813,  -6.6862],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -8.8745,   2.3097,   3.5600],
        [  1.7122,  -4.7591,  -3.6734],
        [ -8.6681,  -4.8556,   9.2518],
        [  9.3424,  -7.3685,  -3.6807],
        [ 10.9931,   7.2107,   7.7933],
        [ -3.2486,   7.5925,  -1.3719],
        [  1.1274,  14.0116,   8.1753],
        [  2.0243,   7.3086,   7.5432],
        [  5.0277,   4.9589,   7.0175],
        [  2.4697,  -4.5836,   1.4593],
        [  4.5503,   5.5359,   7.4735],
        [  5.5626,  -2.0536,   8.4433],
        [  9.0652,   9.8597,  -3.5949],
        [  6.3301,   0.3638,  -7.5685],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.5626,  -2.0536,   8.4433],
        [ -5.1398,  -9.1367,   6.2074],
        [ -0.1626, -10.9796,   6.4629],
        [ -7.4807,   4.7463,   2.0616],
        [ -4.3963,   7.1629,   4.5263],
        [  0.1532,  -4.7422,  -7.0145],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.0047,   3.1062,  -6.7914],
        [ 10.9394,   7.3069,  -8.9365],
        [ -3.9456,  -9.8542,   0.7466],
        [  9.7328,   9.0902,  -0.8917],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.2419,  -0.2313,   4.2440],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -3.1204, -10.2345,   9.2288],
        [  5.3924, -10.1396,  -4.0863]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0504, 0.0762,  ..., 1.2482, 1.2462, 0.6449],
        [0.0504, 0.0000, 0.0259,  ..., 1.2860, 1.2393, 0.6821],
        [0.0762, 0.0259, 0.0000,  ..., 1.3061, 1.2373, 0.7023],
        ...,
        [1.2482, 1.2860, 1.3061,  ..., 0.0000, 1.1852, 0.6038],
        [1.2462, 1.2393, 1.2373,  ..., 1.1852, 0.0000, 1.0257],
        [0.6449, 0.6821, 0.7023,  ..., 0.6038, 1.0257, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.01187484, 13.1471313 , 13.20988996, 13.18278606, 13.32613445,
       13.31764181, 13.5247215 , 11.27160035,  9.04255655,  4.78544875,
       10.95593729, 13.97857591, 11.54071859, 13.40024224, 12.17901033,
       13.66051884, 10.63038565, 11.9182956 ,  5.0713364 ,  5.81199422,
       11.8159254 ,  7.58668144,  8.47371868,  5.2287058 ,  6.60752373,
        6.05165248,  8.62162372, 12.49291768, 11.87665501,  9.66977802,
        7.13501264, 13.12218013,  7.85919687, 14.02526505, 13.5247215 ,
       11.785566  ,  4.76969364,  5.16205087,  5.13616224,  6.19633139,
       10.26327895,  7.50500064,  6.97106912, 10.06286074,  9.60412735,
       12.18045978,  4.91038884,  9.11473391,  9.69792513, 11.89855267,
       12.40074901,  6.39520397,  5.95111804, 11.72802491,  5.0713364 ,
       10.74409834,  6.34059021,  4.90505566,  5.05963217, 12.1494023 ,
        7.06169391,  9.18358562,  6.18668463, 13.06498554, 13.32374056,
       10.5575601 , 13.32645552,  9.50913849,  6.7593917 ,  5.53255882,
        7.92607342,  9.60080639, 12.4980845 , 12.07818871,  9.17016857,
        5.05774295,  9.9353921 , 11.89855267, 13.14690137,  8.25828409,
       14.05687815,  7.58378655,  7.06169391,  5.20659892,  7.16598273,
        5.92952538, 13.39372426,  6.34059021, 13.01187484,  5.92952538,
       10.48315436, 10.98076873,  8.85939626,  8.40445872,  4.74468406,
       12.07818871, 13.1399181 ,  6.76056625, 13.15525803, 10.61472636,
       13.31764181,  9.88571713,  6.24614905, 10.35485331, 10.69963954,
       11.48434767])}
{'P': tensor([[1.8337e-03, 2.1172e-01, 2.0463e-01,  ..., 1.5928e-07, 2.2572e-06,
         1.5928e-07],
        [1.0344e-05, 5.0465e-03, 3.3614e-01,  ..., 3.3208e-07, 4.1302e-06,
         3.3208e-07],
        [1.4431e-05, 9.9487e-05, 7.3805e-03,  ..., 4.6701e-07, 5.7520e-06,
         4.6701e-07],
        ...,
        [8.7122e-05, 7.9153e-05, 7.5985e-05,  ..., 6.4123e-02, 4.4423e-02,
         6.4123e-02],
        [4.4363e-04, 5.2963e-04, 5.1517e-04,  ..., 1.4084e-03, 7.9966e-02,
         1.4084e-03],
        [8.7122e-05, 7.9153e-05, 7.5985e-05,  ..., 6.4123e-02, 4.4423e-02,
         6.4123e-02]]), 'X': tensor([[ 11.7162,   5.6728,  -8.9372],
        [ 10.9875,   7.2207,  -8.8549],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.0373,   7.1299,  -8.5895],
        [ 10.0619,   8.6627,  -9.2884],
        [ 10.1503,   8.5414,  -7.2681],
        [  5.0277,   4.9589,   7.0175],
        [  6.1737,   2.5305,   3.9049],
        [ -1.9714,   7.8563,  -1.8084],
        [  0.6396,  14.0772,  -6.3786],
        [ -9.3464,   0.5708,  -1.6397],
        [  6.3087,   1.8053,  -1.9481],
        [ 10.0400,   8.6922,  -6.2847],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.7275,  14.0667,  -0.9815],
        [  3.2226,  -4.2802,  -4.0428],
        [ 10.0567,   8.6697,   4.6329],
        [  0.1287, -11.0065,   6.5757],
        [  7.9403,  -8.6455,  -8.7433],
        [ 12.6000,   1.3839,  -3.2992],
        [  3.2226,  -4.2802,  -4.0428],
        [ -5.9940,   6.1985,   5.9566],
        [  5.8325,  -1.5222,  -2.1341],
        [ -6.2668,   5.9808,   2.5614],
        [  6.3200,   0.2690,   1.5900],
        [  5.9883,   3.1536,   6.2320],
        [  0.4119,  14.1019,  -4.1870],
        [ 12.2218,   4.0953,  -8.2072],
        [  3.3069,   6.6089,   8.7377],
        [  6.3497,   1.3680,  -1.9076],
        [ -5.3736,   6.6310,  -1.5174],
        [ 10.9875,   7.2207,  -8.8549],
        [ -5.2589,   6.7024,  -5.5809],
        [  1.0093,   7.6567,   2.3526],
        [ -4.7642,  -9.3860,  -5.1133],
        [  2.6970,   6.9826,   9.7350],
        [  3.4604,  -4.1545,   1.8091],
        [  2.5318,  13.7231,   6.3692],
        [  6.2975,   0.0947,   5.1791],
        [ -2.4956, -10.4682,   3.5632],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.4351,  -1.8891,   3.5642],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.2373,  -3.1225,  -8.9763],
        [  6.3087,   1.8053,  -1.9481],
        [ 11.5568,   6.0602,   9.3244],
        [  2.2373,  -4.6506,  -1.7334],
        [ -3.9700,   7.3453,  -4.1486],
        [  4.5057,  -3.3894,  -0.8160],
        [  4.0421,  -3.7765,  -8.3631],
        [ -0.2586,   7.8825,  -9.9840],
        [ -6.4002,  -8.1048,  -0.9108],
        [  4.4633,   5.6287,  -0.7754],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.3413,   0.4883,  -6.8416],
        [ 11.7796,  -3.4696,   9.0088],
        [ 12.5919,   0.4035,  -5.8460],
        [ -9.4766,  -1.1116,  -2.1706],
        [  3.9022,  13.2906,   5.5788],
        [ -6.4844,  -8.0230,   8.5209],
        [ -1.9487,   7.8590,  -3.3330],
        [ -8.9379,  -4.1655,  -2.8114],
        [  4.1637,  13.1898,  -1.9189],
        [ -8.7078,   2.7182,   1.8146],
        [  8.3791,  10.5409,   5.5590],
        [  2.3261,  13.7749,   5.9121],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.7558,  -4.4844,   0.0636],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.1787,  -2.6383,  -4.2533],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.9461,  -4.4077,  -0.4831],
        [  5.1667,  -2.6543,   1.2879],
        [ -8.6050,   2.9451,  -0.7012],
        [ -3.2736, -10.1703,   3.8712],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.4735,  -0.7274,  -2.6749],
        [  4.4290,  -3.4596,   1.4181],
        [  0.7145,  -4.8071,  -2.3766],
        [  1.0391,  -4.8141,  -0.0768],
        [  3.6154,  -4.0640,  -7.9698],
        [ -5.0550,   6.8231,  -9.2502],
        [  0.1206,  -4.7363,   8.7103],
        [ -0.7599, -10.8995,  -6.0366],
        [  0.6952,  -4.8060,   8.8976],
        [ -9.4296,  -1.9458,   4.0833],
        [  0.3724,   7.7974,  -6.8722],
        [  1.1139,   7.6281,  -2.5680],
        [ 12.2589,  -1.9312,   9.8657],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.1787,  -2.6383,  -4.2533],
        [  5.6093,  -1.9705,   6.4807],
        [  1.7228,  -4.7575,   8.3396],
        [  6.3551,   0.7149,   1.3344],
        [ 12.6038,   0.7665,  -3.4439],
        [ -9.4735,  -0.7274,  -2.6749],
        [  2.7685, -10.8982,  -6.8928],
        [  7.8425,  -8.7202,   6.4530],
        [  6.3535,   0.6792,  -7.5101],
        [ 11.8951,  -3.1570,   7.0742],
        [  2.0243,   7.3086,   7.5432],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3606,   0.9204,   3.2525],
        [  4.9957,  -2.8705,   6.5978]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0123, 0.0158,  ..., 1.0965, 0.8974, 1.0965],
        [0.0123, 0.0000, 0.0041,  ..., 1.1011, 0.9025, 1.1011],
        [0.0158, 0.0041, 0.0000,  ..., 1.1046, 0.9061, 1.1046],
        ...,
        [1.0965, 1.1011, 1.1046,  ..., 0.0000, 0.2036, 0.0000],
        [0.8974, 0.9025, 0.9061,  ..., 0.2036, 0.0000, 0.2036],
        [1.0965, 1.1011, 1.1046,  ..., 0.0000, 0.2036, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.01728313, 13.14777433, 13.15525803, 13.1399181 , 13.27717982,
       13.26590071,  7.06169391,  6.67218687,  8.09984231, 14.09176259,
        9.36378438,  6.5618968 , 13.27994304,  7.8575648 , 14.08549802,
        5.35774573, 13.2778378 , 11.00726483, 11.73848065, 12.67576792,
        5.35774573,  8.62260327,  6.02789634,  8.66273171,  6.32572323,
        6.76791483, 14.10796466, 12.889695  ,  7.39004226,  6.49538841,
        8.53501777, 13.14777433,  8.51928531,  7.72291946, 10.52587243,
        7.48538905,  5.40689332, 13.95472331,  6.29821673, 10.76154256,
       10.19036081,  9.62238828, 10.5575601 ,  8.7273569 ,  9.75075273,
        6.5618968 , 13.0493368 ,  5.16077764,  8.34949383,  5.63824971,
        5.53175275,  7.88677538, 10.32715715,  7.18354885,  7.88677538,
        6.36004393, 12.27993098, 12.59840193,  9.54154096, 13.85158417,
       10.31584102,  8.09704052,  9.86089579, 13.83139024,  9.12220271,
       13.46553623, 13.96987806, 11.75618513, 11.63765273,  5.26342203,
       10.0674887 ,  5.81199422, 12.76408562,  5.30158497,  5.80860165,
        9.09502385, 10.68416401, 11.48434767,  9.5014146 ,  5.62005629,
        4.85995002,  4.92497514,  5.43943791,  8.49161154,  4.73784222,
       10.92597196,  4.85603886,  9.6282763 ,  7.80625981,  7.70897675,
       12.41012046, 11.51811603,  5.81199422,  5.94535529,  5.05981914,
        6.39520397, 12.62711252,  9.5014146 , 11.24434374, 11.72802491,
        6.38968357, 12.30695098,  7.58378655,  5.76165934,  6.42689705,
        5.76165934])}
{'P': tensor([[3.9242e-03, 3.7441e-01, 1.6653e-01,  ..., 2.3184e-06, 4.3157e-07,
         1.0626e-08],
        [5.3554e-05, 6.7009e-03, 1.0788e-01,  ..., 4.1324e-06, 6.6025e-07,
         1.7270e-08],
        [5.9391e-05, 1.6662e-04, 7.3115e-03,  ..., 6.2431e-06, 8.7497e-07,
         2.1967e-08],
        ...,
        [2.1061e-06, 2.0383e-06, 2.1731e-06,  ..., 9.4336e-02, 1.9747e-05,
         5.2197e-04],
        [1.1314e-04, 1.0693e-04, 1.6474e-04,  ..., 1.1686e-04, 2.3564e-01,
         9.2890e-05],
        [7.0359e-06, 6.1958e-06, 6.6024e-06,  ..., 2.9346e-06, 1.3653e-03,
         5.1187e-02]]), 'X': tensor([[ 11.6634,   5.8050,   9.1590],
        [ 11.5568,   6.0602,   9.3244],
        [ 11.2571,   6.7072,   8.3693],
        [  9.4925,   9.3811,   8.9842],
        [  9.3560,   9.5391,   9.4943],
        [  9.4554,   9.4245,   9.8800],
        [  4.6992,   5.3690,   3.0935],
        [  2.6310, -10.9195,   3.8054],
        [ -9.4766,  -1.1116,  -2.1706],
        [  0.1677,  -4.7447,  -6.1529],
        [  6.3549,   0.7096,   6.9198],
        [ -3.5520,   7.4976,   9.1522],
        [ 12.6000,   1.3839,  -3.2992],
        [  6.1741,  -0.5329,   3.1913],
        [  6.0788,  -0.8710,  -6.8719],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.6202,  -1.9508,   0.7471],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.2895,   0.0405,   5.7652],
        [  6.3587,   1.1596,  -3.6629],
        [  2.5375,   7.0676,  -4.3490],
        [  4.3377,   5.7570,   1.3712],
        [ -4.8751,  -9.3148,   4.0406],
        [  0.3354,  -4.7708,  -4.9125],
        [ -0.9715, -10.8630,   0.3555],
        [  3.9913,  -3.8139,  -6.8311],
        [  7.4132,  11.3526,   0.3322],
        [ -8.6229,   2.9068,   2.4429],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.2675,  12.1379,  -8.9720],
        [ 12.6035,   0.7494,   7.1536],
        [  5.1196,  -2.7160,   4.2705],
        [ 11.7162,   5.6728,  -8.9372],
        [  5.4426, -10.1181,  -5.1268],
        [  2.5375,   7.0676,  -4.3490],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.6095,  -1.9702,  -7.8233],
        [  7.5930,  -8.9037,   7.2323],
        [ 12.6059,   0.9597,  -6.3264],
        [  6.3119,   1.7797,   9.9186],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.3211,   1.7017,  -0.2164],
        [ 11.0438,   7.1180,  -5.6991],
        [  3.2226,  -4.2802,  -4.0428],
        [  6.3549,   0.7096,   6.9198],
        [ 12.6059,   0.9597,  -6.3264],
        [  1.2728, -11.0372,   9.8121],
        [  4.2041,  -3.6509,  -4.9488],
        [ -8.3653,   3.4184,   9.3454],
        [ -0.0410,   7.8591,   9.3457],
        [  2.6614,  -4.5192,  -9.3641],
        [  9.6614,   9.1784,  -1.2087],
        [  1.6408,   7.4591,  -4.2294],
        [  8.2422,  -8.4041,  -1.4962],
        [ 12.5927,   0.4199,  -9.4997],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.0305,   3.0296,   0.1375],
        [ -8.3900,   3.3727,   3.1650],
        [ -5.6792,  -8.7352,   4.0318],
        [  3.2692,  13.5098,  -4.9032],
        [  6.2902,   0.0451,   3.7116],
        [ 12.2921,  -1.7902,  -4.0870],
        [  2.5829,  -4.5466,  -8.4015],
        [  5.2608,  -2.5256,  -9.1446],
        [ -0.3432,   7.8900,   7.1805],
        [  6.3119,   1.7797,   9.9186],
        [  2.3367,   7.1676,  -9.5263],
        [ -2.1817,   7.8274,  -4.6298],
        [  7.0358,  -9.2784,  -9.8479],
        [ 12.2914,   3.8037,   6.0978],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.3616,  -2.4749,   0.1793],
        [  9.5261,  -7.1675,   1.4206],
        [  6.1685,  12.1979,  -6.2473],
        [  3.3167,   6.6022,   6.3554],
        [ -8.0087,   4.0173,   9.1091],
        [ 11.7418,   5.6070,  -1.4192],
        [  9.4516,  -7.2501,  -1.8868],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4159,  -2.0739,   9.6841],
        [  2.7121,   6.9743,  -8.4202],
        [  3.6249, -10.7255,   5.6605],
        [  5.0348,  -2.8229,   3.5436],
        [  3.3474,  -4.2162,   5.2455],
        [  9.3753,   9.5171,   3.9131],
        [  9.7328,   9.0902,  -0.8917],
        [ -8.0087,   4.0173,   9.1091],
        [ -1.5875,   7.8944,  -9.4286],
        [  3.2809,  13.5061,  -4.0199],
        [  5.4426, -10.1181,  -5.1268],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.4573,   3.2448,   5.4370],
        [  5.3860,  -2.3418,   1.0826],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -9.4756,  -0.8166,  -9.6021],
        [  5.9912,   3.1453,  -3.9989],
        [  2.2182,  13.8007,   5.1889],
        [ -8.4297,  -5.3637,  -9.3962],
        [  5.6095,  -1.9702,  -7.8233],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3048,   0.1469,  -9.1691],
        [ -1.0482, -10.8487,  -5.9066]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0195, 0.0204,  ..., 1.1803, 1.2155, 1.5926],
        [0.0195, 0.0000, 0.0135,  ..., 1.1999, 1.2284, 1.6107],
        [0.0204, 0.0135, 0.0000,  ..., 1.1959, 1.2162, 1.6024],
        ...,
        [1.1803, 1.1999, 1.1959,  ..., 0.0000, 0.9805, 0.6514],
        [1.2155, 1.2284, 1.2162,  ..., 0.9805, 0.0000, 0.7530],
        [1.5926, 1.6107, 1.6024,  ..., 0.6514, 0.7530, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.02818702, 13.0493368 , 13.103709  , 13.34587027, 13.3614598 ,
       13.35013374,  7.13501264, 11.23201619,  9.54154096,  4.74770899,
        6.39438044,  8.29640613, 12.67576792,  6.19707944,  6.14086582,
        5.18277858,  5.94909709,  9.77588801,  6.28961988,  6.46356806,
        7.5092834 ,  7.20826219, 10.51340829,  4.78258092, 10.90638066,
        5.5205189 , 13.55868684,  9.09963894, 10.52357479, 13.66051884,
       12.62576026,  5.79543038, 13.01728313, 11.4890861 ,  7.5092834 ,
        9.54154096,  5.94541803, 11.70168755, 12.64235359,  6.55800267,
        5.22903117,  6.54615887, 13.13888841,  5.35774573,  6.39438044,
       12.64235359, 11.11038551,  5.56809335,  9.0368458 ,  7.85919687,
        5.24461068, 13.32613445,  7.6374603 , 11.77124373, 12.59969936,
        5.84450127,  6.74872981,  9.04255655, 10.41909211, 13.89974693,
        6.29034958, 12.4217523 ,  5.22903117,  5.83560009,  7.89745107,
        6.55800267,  7.53884104,  8.12580976, 11.64436684, 12.86648771,
       11.66453059,  9.68323686, 11.92133129, 13.66896739,  7.38844703,
        8.95983586, 13.01187484, 11.9120321 , 10.99326577,  9.73579738,
        6.00663314,  9.64157461,  7.48311002, 11.32149259,  5.77219077,
        5.38342077, 13.35927527, 13.31764181,  8.95983586,  8.05242591,
       13.89886469, 11.4890861 , 12.00995898,  5.19065439,  9.05843201,
        5.87305005,  8.62260327,  9.66977802,  9.51074763,  6.76661595,
       13.97779839,  9.99145096,  5.94541803,  9.09963894,  6.30647746,
       10.8992524 ])}
{'P': tensor([[2.6929e-03, 2.0337e-01, 1.9701e-01,  ..., 2.1815e-04, 8.2021e-08,
         8.7776e-08],
        [1.5455e-05, 8.6415e-03, 1.4601e-01,  ..., 2.2566e-04, 1.3483e-07,
         3.7790e-07],
        [8.2540e-06, 1.8421e-04, 6.3920e-03,  ..., 1.4442e-04, 9.5477e-08,
         2.9126e-07],
        ...,
        [4.8650e-03, 1.0296e-02, 1.3700e-02,  ..., 1.1282e-01, 4.4112e-05,
         5.0503e-05],
        [2.2469e-03, 8.7143e-04, 9.0679e-04,  ..., 2.3454e-03, 4.1678e-02,
         7.5543e-07],
        [1.7371e-05, 1.0127e-05, 8.8268e-06,  ..., 1.3572e-04, 4.2014e-04,
         1.5761e-01]]), 'X': tensor([[ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 5.6022e+00, -1.9833e+00,  8.3081e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 1.2526e+01, -4.1995e-01, -9.3493e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.7641e-02, 3.0810e-02,  ..., 3.4276e-01, 9.1926e-01,
         1.2235e+00],
        [5.7641e-02, 0.0000e+00, 6.4302e-02,  ..., 3.9793e-01, 9.7681e-01,
         1.2447e+00],
        [3.0810e-02, 6.4302e-02, 2.1453e-04,  ..., 3.5468e-01, 9.2276e-01,
         1.2529e+00],
        ...,
        [3.4276e-01, 3.9793e-01, 3.5468e-01,  ..., 0.0000e+00, 5.9443e-01,
         1.0453e+00],
        [9.1926e-01, 9.7681e-01, 9.2276e-01,  ..., 5.9443e-01, 0.0000e+00,
         1.1999e+00],
        [1.2235e+00, 1.2447e+00, 1.2529e+00,  ..., 1.0453e+00, 1.1999e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.03002338, 13.20597932, 13.2430377 , 13.12299145, 13.18278606,
       13.38091121, 13.32557133, 13.66991027,  4.89869205,  5.1224079 ,
       11.28093507, 11.97949273,  8.67227386,  6.4020404 ,  8.53014915,
        6.39520397,  4.78544875,  5.86519525,  7.72668467, 11.31276109,
        9.77777346,  4.91296205, 13.00380624, 11.37939695, 12.72524615,
       11.70168755, 11.36089355,  7.8788965 ,  5.23878371, 13.09426142,
       10.27572673,  8.95983586,  9.17016857,  5.43544626,  7.37902766,
        6.39520397, 10.31584102, 11.74901778,  9.34061938,  7.16747878,
       10.36844813,  4.92497514, 12.70489563,  8.8737556 ,  9.94173708,
        5.94541803,  5.88782913,  8.36445758,  6.46560181,  5.68942572,
        5.19577146, 12.52875793,  7.15100303, 10.67132524,  5.0209005 ,
        7.58378655, 13.96987806, 10.33712258, 12.62576026, 13.0743399 ,
       13.6829843 , 13.85158417, 10.06850677,  6.39438044,  6.58184474,
        7.5403822 , 11.21703654,  8.66273171,  5.16205087, 13.68063098,
        6.81084724,  6.61386436,  5.01114719, 13.01187484,  8.12580976,
       13.5247215 , 11.51557166,  7.72668467,  9.58347456, 13.31764181,
       11.50791859, 11.05672136, 11.24434374, 13.50484539,  9.4997987 ,
       11.83509372,  5.95111804,  8.0550286 ,  7.56617037, 11.75438006,
        5.94293185,  7.46053281, 14.10796466,  4.85603886, 12.53285633,
       11.04953767,  8.87007495, 12.21532282, 12.53285633,  8.68832398,
       12.18045978, 13.26209545,  9.34127013,  6.80370254, 11.89105259,
       10.14436912])}
{'P': tensor([[2.3413e-03, 2.0856e-01, 1.7361e-01,  ..., 1.9174e-04, 2.5011e-08,
         1.9674e-03],
        [8.0689e-06, 2.3236e-03, 1.6576e-01,  ..., 9.8434e-05, 1.6610e-08,
         2.3569e-05],
        [1.7460e-05, 4.6774e-05, 1.3268e-02,  ..., 2.8071e-04, 7.8519e-08,
         1.9114e-05],
        ...,
        [4.4998e-03, 7.2589e-03, 1.2322e-02,  ..., 7.4590e-02, 8.2401e-06,
         4.6324e-03],
        [3.9803e-04, 2.3207e-04, 1.9831e-04,  ..., 3.4919e-04, 7.3277e-02,
         5.3196e-04],
        [1.2354e-03, 4.2550e-02, 1.3155e-01,  ..., 1.6532e-04, 2.4184e-08,
         1.9953e-03]]), 'X': tensor([[ 11.6019,   5.9540,  -5.1715],
        [ 11.0438,   7.1180,  -5.6991],
        [ 10.2712,   8.3707,  -4.9743],
        [ 10.1798,   8.5002,  -5.8893],
        [  9.6816,   9.1536,  -5.8543],
        [ 11.3118,   6.5957,  -4.5667],
        [ 12.5650,   2.0132,   4.6272],
        [  9.1620,  -7.5573,  -9.9882],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -7.9925,  -6.1423,   2.8990],
        [ -3.2736, -10.1703,   3.8712],
        [  5.3924, -10.1396,  -4.0863],
        [  2.5813,  -4.5471,   5.3597],
        [  2.5375,   7.0676,  -4.3490],
        [ 11.7162,   5.6728,  -8.9372],
        [  6.3236,   0.3011,  -4.6668],
        [  7.0515,  11.6194,   8.4289],
        [  2.2644,  -4.6434,   9.7508],
        [ -1.9487,   7.8590,  -3.3330],
        [ -4.0409,   7.3169,  -3.5779],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.8400,   2.3991,   1.3457],
        [ -6.0844,  -8.3955,   0.7705],
        [ 11.9122,   5.1397,  -1.5498],
        [ -4.6942,  -9.4299,  -9.1846],
        [  1.8721,  -4.7322,   0.5524],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.0305,   3.0296,   0.1375],
        [  4.2339,  -3.6268,   6.3125],
        [  5.9193,  -1.3176,   5.1739],
        [  5.3860,  -2.3418,   1.0826],
        [  4.1537,   5.9340,  -6.0423],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.5382,   7.0672,   4.5097],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.5294,  -2.1106,   9.0626],
        [  1.7301,   7.4262,   5.0522],
        [  6.3236,   0.3011,  -4.6668],
        [  6.9956,  11.6590,  -7.6876],
        [ -0.3432,   7.8900,   7.1805],
        [  2.3261,   7.1726,   6.2522],
        [  7.8375,  -8.7240,  -9.8913],
        [ -9.4739,  -1.2497,   0.5483],
        [ -8.4267,  -5.3697,   2.4172],
        [  2.2373,  -4.6506,  -1.7334],
        [  9.4925,   9.3811,   8.9842],
        [  6.9653,  -9.3225,  -3.4727],
        [ -9.3087,   0.7792,  -7.5587],
        [  2.9410,   6.8428,   5.4458],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.3200,   7.1756,   1.1938],
        [ 10.1798,   8.5002,  -5.8893],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 10.4994,  -5.9193,  -7.5153],
        [  3.2226,  -4.2802,  -4.0428],
        [ 10.5057,  -5.9099,  -5.1599],
        [ -9.2373,  -3.1225,  -8.9763],
        [  3.1756, -10.8247,   6.8882],
        [  2.3367,   7.1676,  -9.5263],
        [  4.4633,   5.6287,  -0.7754],
        [  3.6154,  -4.0640,  -7.9698],
        [ -9.1706,   1.3863,   8.4032],
        [  2.7121,   6.9743,  -8.4202],
        [  6.3122,  -9.6994,  -6.3282],
        [  3.5939,   6.4029,  -1.8357],
        [  1.1274,  14.0116,   8.1753],
        [  5.0190,   4.9705,  -7.9752],
        [  3.2366,  -4.2732,   5.1532],
        [ -6.4620,  -8.0449,   6.2719],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.9750,  -6.6338,  -1.2440],
        [  2.7685, -10.8982,  -6.8928],
        [ 11.9936,  -2.8677,   6.0614],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.0560,  -0.9421,  -5.0334],
        [  1.7232,  -4.7574,   3.1270],
        [  4.6357,   5.4415,   7.9067],
        [  7.3724,  -9.0577,  -8.9843],
        [  4.5325,  -3.3643,   6.7699],
        [  5.0190,   4.9705,  -7.9752],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.2687,  -2.9794,   0.6436],
        [  4.5491,  -3.3485,   6.8412],
        [  5.9912,   3.1453,  -3.9989],
        [  9.2811,  -7.4336,   5.1771],
        [  5.9587,  12.3216,   9.1695],
        [  6.9314,  -9.3435,  -1.2729],
        [  3.2159, -10.8166,  -1.0406],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.1685,  12.1979,  -6.2473],
        [ -2.1382,   7.8339,   7.3524],
        [ -8.1531,  -5.8748,   2.6934],
        [ 11.9808,   4.9340,  -3.3715],
        [ -0.3432,   7.8900,   7.1805],
        [  8.5716,  10.3596,   6.8818],
        [ 11.3118,   6.5957,  -4.5667],
        [ -8.4573,   3.2448,   5.4370],
        [  9.2811,  -7.4336,   5.1771],
        [ 10.4737,  -5.9570,  -1.2108],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.8325,  -1.5222,  -2.1341],
        [  6.3303,   1.6145,  -8.4083],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.9526,   3.2525,  -4.3821],
        [  8.2154,  -8.4262,   8.0171],
        [ 11.5935,   5.9740,  -4.3952]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.4309e-02, 3.1609e-02,  ..., 2.9886e-01, 1.2106e+00,
         5.6151e-02],
        [4.4309e-02, 0.0000e+00, 5.1454e-02,  ..., 3.4031e-01, 1.2544e+00,
         9.7152e-02],
        [3.1609e-02, 5.1454e-02, 0.0000e+00,  ..., 3.1521e-01, 1.2096e+00,
         5.2452e-02],
        ...,
        [2.9886e-01, 3.4031e-01, 3.1521e-01,  ..., 0.0000e+00, 9.6707e-01,
         2.6783e-01],
        [1.2106e+00, 1.2544e+00, 1.2096e+00,  ..., 9.6707e-01, 0.0000e+00,
         1.1583e+00],
        [5.6151e-02, 9.7152e-02, 5.2452e-02,  ..., 2.6783e-01, 1.1583e+00,
         3.0339e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.04051744, 13.13888841, 13.25017211, 13.26209545, 13.32374056,
       13.09426142, 12.72524615, 11.87665501, 10.32715715, 10.08001894,
       10.68416401, 11.48434767,  5.2287058 ,  7.5092834 , 13.01728313,
        6.33075682, 13.59170809,  5.16611782,  8.09704052,  8.35856487,
       10.1371552 ,  9.15976746, 10.36844813, 12.97370371, 10.53368644,
        5.08910491,  9.68323686,  6.74872981,  5.5748624 ,  6.06416957,
        5.87305005,  7.24326992, 12.76408562,  7.50917551,  9.43737523,
        5.91854397,  7.62509582,  6.33075682, 13.59673406,  7.89745107,
        7.5403822 , 11.72749923,  9.55593374,  9.99211663,  5.16077764,
       13.34587027, 11.63724792,  9.34127013,  7.44805901, 10.00410625,
        7.54127425, 13.26209545, 10.33712258, 12.05300943,  5.35774573,
       12.05395056,  9.75075273, 11.28093507,  7.53884104,  7.18354885,
        5.43943791,  9.27474386,  7.48311002, 11.57249984,  7.34250948,
       14.05687815,  7.06372172,  5.36061974, 10.31886066,  9.66977802,
       11.97949273, 11.24434374, 12.33167382, 10.74409834,  6.12886024,
        5.05989547,  7.14837514, 11.67875855,  5.64464807,  7.06372172,
       11.1341798 ,  9.73579738,  5.64865173,  6.76661595, 11.89105259,
       13.68672855, 11.63383221, 11.2845608 ,  9.7751189 , 13.66896739,
        8.12043983, 10.04916678, 12.95702647,  7.89745107, 13.44593519,
       13.09426142,  9.05843201, 11.89105259, 12.04924159,  5.68942572,
        6.02789634,  6.53290663,  8.47371868,  6.78325151, 11.76830981,
       13.04217487])}
{'P': tensor([[4.9659e-03, 2.2143e-01, 1.7857e-01,  ..., 4.4950e-06, 1.2405e-05,
         7.6171e-08],
        [6.5832e-06, 8.2883e-03, 1.8142e-04,  ..., 2.8592e-06, 8.4313e-06,
         7.6535e-08],
        [1.3035e-05, 1.0462e-01, 8.2000e-03,  ..., 4.6431e-06, 1.5554e-05,
         8.6615e-08],
        ...,
        [1.0290e-03, 1.0902e-03, 1.8472e-03,  ..., 1.1654e-01, 3.0567e-01,
         5.5670e-05],
        [2.7917e-04, 5.3397e-04, 7.9746e-04,  ..., 2.3979e-03, 1.1304e-01,
         1.5254e-05],
        [7.6384e-04, 1.5615e-04, 2.4142e-04,  ..., 1.5905e-03, 3.5101e-04,
         5.1639e-02]]), 'X': tensor([[ 11.5966,   5.9666,   6.0083],
        [  9.5203,   9.3483,   5.8781],
        [ 10.0567,   8.6697,   4.6329],
        [  9.2692,   9.6370,   7.0746],
        [  9.0051,   9.9234,   5.4411],
        [  9.2979,   9.6048,   7.3138],
        [ -1.6906,   7.8860,   7.7142],
        [  0.3724,   7.7974,  -6.8722],
        [ -7.3446,   4.9113,  -5.9356],
        [ -7.2515,   5.0195,   4.6434],
        [  5.1472,  -2.6801,   8.1194],
        [  5.6314,  -1.9302,  -9.6543],
        [ 11.1085,  -4.9204,   0.0736],
        [  4.0457,  -3.7737,  -0.4704],
        [ -7.9667,   4.0814,  -0.6397],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.4901,  -0.7052,   1.7315],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 12.1751,  -2.2592,  -5.7751],
        [  9.6662,   9.1725,   8.1056],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -7.4602,   4.7718,  -4.6277],
        [ 12.4877,  -0.7222,   8.8347],
        [  1.7000,  -4.7609,   2.7935],
        [  6.3549,   0.7096,   6.9198],
        [  3.9663,   6.1018,  -5.4104],
        [  4.7739,  -3.1218,  -4.1188],
        [  7.1441,  11.5529,   2.3232],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.9936,  -2.8677,   6.0614],
        [ -8.6603,   2.8250,   5.9181],
        [  2.0899,  -4.6870,  -7.4260],
        [  3.1064,  13.5607,   6.7605],
        [  0.1382, -11.0073,   1.0026],
        [  2.9470,  13.6085,   2.9809],
        [ -2.2188,   7.8218,   8.6064],
        [  6.2975,   0.0947,   5.1791],
        [  5.0348,  -2.8229,   3.5436],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.6672,  -1.8631,   8.1020],
        [  6.3583,   1.1729,   8.7835],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.3303,   1.6145,  -8.4083],
        [ -1.3940, -10.7772,   6.7835],
        [ -1.5451,   7.8975,   7.6104],
        [ -4.3963,   7.1629,   4.5263],
        [ -7.8280,   4.2848,  -7.1955],
        [  2.5620,   7.0549,   8.4825],
        [ -0.5739,   7.9056,   0.6382],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.4932,   7.0903,  -5.4214],
        [  0.1206,  -4.7363,   8.7103],
        [  6.0031,  12.2958,  -6.5992],
        [  2.4149,  -4.6005,  -4.3339],
        [ -2.0445,   7.8469,  -0.5857],
        [  5.7151,  -1.7700,   3.1542],
        [  6.2242,  -0.3171,   7.4978],
        [  1.7000,  -4.7609,   2.7935],
        [  9.3836,   9.5075,  -5.9278],
        [  1.8320,   7.3871,   5.1452],
        [ -1.6906,   7.8860,   7.7142],
        [  2.0729,   7.2878,   7.0880],
        [  4.9452, -10.3187,  -7.8621],
        [ -1.9646,   7.8571,   3.3817],
        [  0.9478,  -4.8144,   0.6371],
        [  5.9843,  -9.8679,  -6.1449],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -9.3705,   0.4208,  -3.5815],
        [ 12.4592,  -0.9168,  -6.7571],
        [ 11.9122,   5.1397,  -1.5498],
        [ 12.4837,   2.7510,   0.4449],
        [  8.8091,  -7.9037,   0.3631],
        [ -8.9033,   2.2329,  -8.8912],
        [  1.7500,   7.4187,   4.4237],
        [  7.9403,  -8.6455,  -8.7433],
        [  4.0078,  -3.8018,  -6.2168],
        [  3.2647, -10.8066,  -0.3768],
        [ -6.3306,   5.9272,   5.6935],
        [ 11.0864,   7.0387,  -0.1238],
        [ -8.3653,   3.4184,   9.3454],
        [  0.8700,  -4.8132,  -7.8042],
        [  6.3200,   0.2690,   1.5900],
        [  8.2154,  -8.4262,   8.0171],
        [  6.2770,  -0.0379,   2.4195],
        [ 11.6801,  -3.7196,  -0.9181],
        [  9.3624,   9.5318,   0.2779],
        [  1.0000,   7.6591,  -4.5011],
        [  9.9750,  -6.6338,  -1.2440],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.5568,   6.0602,   9.3244],
        [ -9.1749,  -3.3803,   6.4629],
        [  0.3231,  -4.7691,  -1.6120],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.3535,   0.6792,  -7.5101],
        [  1.6422,   7.4586,  -6.8056],
        [  0.2571,  -4.7594,   1.5618],
        [  3.2990,  -4.2415,  -7.4857],
        [ 10.9394,   7.3069,  -8.9365],
        [ 11.5456,   6.0861,   8.1101],
        [ -7.4807,   4.7463,   2.0616],
        [  5.2911,  -2.4824,  -3.6877],
        [ -8.3539,  -5.5111,   0.3373],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.0047,   3.1062,  -6.7914],
        [  4.5814, -10.4485,   2.5496]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0615, 0.0985,  ..., 0.9677, 1.0323, 0.8417],
        [0.0615, 0.0000, 0.1090,  ..., 0.9978, 1.0621, 0.9008],
        [0.0985, 0.1090, 0.0000,  ..., 0.8891, 0.9533, 0.8317],
        ...,
        [0.9677, 0.9978, 0.8891,  ..., 0.0000, 0.0648, 0.7101],
        [1.0323, 1.0621, 0.9533,  ..., 0.0648, 0.0000, 0.7504],
        [0.8417, 0.9008, 0.8317,  ..., 0.7101, 0.7504, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.04156075, 13.3426541 , 13.2778378 , 13.37121923, 13.40024224,
       13.36800073,  8.06517143,  7.80625981,  8.8353674 ,  8.81930525,
        5.80312949,  5.95297459, 12.1494023 ,  5.53255882,  8.95133961,
        4.92497514, 12.50996828,  9.75401449, 12.3828957 , 13.32557133,
        9.55720651,  8.85572292, 12.5085995 ,  5.05534609,  6.39438044,
        7.27759233,  5.70405525, 13.58333857, 13.08414689, 12.33167382,
        9.10945678,  5.13182506, 13.91198191, 11.00812996, 13.92390689,
        8.1303945 ,  6.29821673,  5.77219077,  9.71392179,  9.37989631,
        5.96556248,  6.46560181, 10.26327895,  6.53290663, 10.86693708,
        8.04718433,  8.40445872,  8.92395381,  7.50564223,  7.92644746,
       12.3828957 ,  7.5158489 ,  4.73784222, 13.6829843 ,  5.19577146,
        8.10886021,  5.98285319,  6.2322769 ,  5.05534609, 13.35832725,
        7.61088125,  8.06517143,  7.57685908, 11.4424816 ,  8.09899874,
        4.90676242, 11.54071859, 10.46739073,  9.37989631, 12.49291768,
       12.97370371, 12.78327037, 11.83509372,  9.17905491,  7.62232307,
       11.73848065,  5.52416368, 11.28895489,  8.67227386, 13.13205357,
        9.0368458 ,  4.89121805,  6.32572323, 11.76830981,  6.27715105,
       12.25806559, 13.3607379 ,  7.72415048, 11.97949273,  4.87083348,
       13.0493368 ,  9.77777346,  4.78002665, 12.05395056,  6.38968357,
        7.63726134,  4.7663526 ,  5.37343173, 13.15525803, 13.05148826,
        8.85939626,  5.84450127, 10.00794468,  6.27905292,  6.76056625,
       11.4088043 ])}
{'P': tensor([[2.3796e-03, 3.5969e-01, 1.7014e-01,  ..., 1.4720e-08, 3.0179e-02,
         7.1119e-07],
        [1.2541e-05, 3.4271e-03, 1.9251e-01,  ..., 1.8856e-08, 5.4532e-02,
         9.5137e-07],
        [1.2312e-05, 2.4985e-05, 5.9084e-03,  ..., 3.1202e-08, 3.5712e-02,
         1.2783e-06],
        ...,
        [9.9279e-05, 7.5607e-05, 5.8179e-05,  ..., 6.2649e-02, 1.0633e-05,
         6.9888e-05],
        [1.2056e-05, 1.6926e-05, 7.2442e-05,  ..., 3.7706e-08, 1.5012e-02,
         3.1296e-06],
        [1.0114e-03, 8.9288e-04, 5.5885e-04,  ..., 4.4470e-05, 4.3512e-04,
         6.9877e-02]]), 'X': tensor([[ 11.5935,   5.9740,  -4.3952],
        [ 11.3118,   6.5957,  -4.5667],
        [ 10.5955,   7.8824,  -3.7006],
        [ 10.2712,   8.3707,  -4.9743],
        [  9.3607,   9.5337,  -4.5412],
        [ 10.3251,   8.2928,  -3.0006],
        [  0.9445,   7.6736,   0.5095],
        [  5.9003,  12.3550,   2.1917],
        [  2.3589,  13.7668,  -3.6501],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.4592,  -0.9168,  -6.7571],
        [  6.1989,  -9.7591,   0.8289],
        [  2.2644,  -4.6434,   9.7508],
        [ -8.5900,   2.9767,   7.4811],
        [  7.9403,  -8.6455,  -8.7433],
        [ 12.5996,   1.3971,  -1.0694],
        [ -9.3616,  -2.4750,   3.7765],
        [ -5.0550,   6.8231,  -9.2502],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.5224,  -0.4497,   4.8448],
        [  6.1730,  -0.5375,  -8.8861],
        [  0.9967,  -4.8144,  -9.7370],
        [ 10.0619,   8.6627,  -9.2884],
        [ 10.9717,  -5.1646,  -6.9039],
        [  3.0876,   6.7527,  -5.3540],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.4735,  -0.7274,  -2.6749],
        [  3.1271, -10.8343,   2.6442],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.4739,  -1.2497,   0.5483],
        [ -8.0588,  -6.0341,  -2.3006],
        [ 10.2470,   8.4054,  -0.1177],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.2480, -11.0153,  -3.0183],
        [ -0.9715, -10.8630,   0.3555],
        [  6.0047,   3.1062,  -6.7914],
        [  5.8450,  -9.9356,   3.0767],
        [  8.2422,  -8.4041,  -1.4962],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.2767,  -4.6006,   0.7868],
        [  0.9809,   7.6642,  -1.4217],
        [ -7.4887,   4.7364,   8.9074],
        [  1.0000,   7.6591,  -4.5011],
        [  6.7565,  11.8237,  -4.9882],
        [  5.9655,   3.2172,  -7.5953],
        [ 11.1085,  -4.9204,   0.0736],
        [  5.2911,  -2.4824,  -3.6877],
        [  4.1637,  13.1898,  -1.9189],
        [ -0.6323,   7.9085,  -5.9544],
        [  4.5767,   5.5071,   3.9069],
        [  2.5620,   7.0549,   8.4825],
        [ 12.5258,  -0.4200,  -9.3493],
        [  4.1297,  -3.7098,  -3.9838],
        [ -9.3810,  -2.3455,  -9.5194],
        [ 12.6000,   1.3839,  -3.2992],
        [  5.1793,   4.7461,   0.3899],
        [  0.8700,  -4.8132,  -7.8042],
        [  3.6703,   6.3443,  -5.4401],
        [  5.8325,  -1.5222,  -2.1341],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.2419,  -0.2313,   4.2440],
        [  4.5057,  -3.3894,  -0.8160],
        [  2.3589,  13.7668,  -3.6501],
        [ 11.6019,   5.9540,  -5.1715],
        [ -5.3382,   6.6533,   7.8054],
        [ -4.0409,   7.3169,  -3.5779],
        [ -8.3062,  -5.6008,   0.0287],
        [  1.6954,  -4.7616,   9.8198],
        [ 11.5456,   6.0861,   8.1101],
        [  2.7590,  13.6621,   9.4496],
        [ -9.1231,  -3.5739,   8.4708],
        [  2.2849, -10.9655,  -7.6542],
        [  2.8890,  -4.4316,   1.1600],
        [  2.1625,  -4.6697,   7.0658],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.7792,  11.8084,   1.6251],
        [ -8.9033,   2.2329,  -8.8912],
        [  4.4290,  -3.4596,   1.4181],
        [  7.7750,  11.0665,   2.8364],
        [  2.5382,   7.0672,   4.5097],
        [  3.4140,  -4.1803,  -8.8224],
        [  2.2373,  -4.6506,  -1.7334],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.7565,  11.8237,  -4.9882],
        [ -3.9858,  -9.8334,  -3.7323],
        [  0.5250, -11.0306,   5.5358],
        [  6.3460,   1.4252,   1.2291],
        [  1.3339,   7.5627,  -0.9034],
        [  2.2659,  13.7894,   6.5436],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.9693,  -9.3200,  -8.3822],
        [ -3.2736, -10.1703,   3.8712],
        [ 12.1751,  -2.2592,  -5.7751],
        [ 10.4737,  -5.9570,  -1.2108],
        [ 12.1940,  -2.1881,   7.1499],
        [  7.1697,  11.5343,  -0.7276],
        [ 10.6673,  -5.6649,  -4.1646],
        [  9.7328,   9.0902,  -0.8917],
        [ 11.6019,   5.9540,  -5.1715],
        [  8.2154,  -8.4262,   8.0171],
        [  2.2182,  13.8007,   5.1889],
        [  6.3589,   0.8239,  -8.4230],
        [  3.1756, -10.8247,   6.8882],
        [  8.9284,  10.0036,  -6.9210],
        [  2.3889,  -4.6082,  -8.7291]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 1.9927e-02, 5.9802e-02,  ..., 1.2622e+00, 1.4986e-01,
         2.8205e-01],
        [1.9927e-02, 0.0000e+00, 7.1094e-02,  ..., 1.2821e+00, 1.3242e-01,
         2.9522e-01],
        [5.9802e-02, 7.1094e-02, 0.0000e+00,  ..., 1.2318e+00, 2.0134e-01,
         3.1050e-01],
        ...,
        [1.2622e+00, 1.2821e+00, 1.2318e+00,  ..., 0.0000e+00, 1.4020e+00,
         1.1242e+00],
        [1.4986e-01, 1.3242e-01, 2.0134e-01,  ..., 1.4020e+00, 0.0000e+00,
         3.4747e-01],
        [2.8205e-01, 2.9522e-01, 3.1050e-01,  ..., 1.1242e+00, 3.4747e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.04217487, 13.09426142, 13.20597932, 13.25017211, 13.36092624,
       13.2430377 ,  7.73151309, 13.69162795, 13.96746816,  7.8788965 ,
       12.49291768, 11.56146873,  5.16611782,  9.09119351, 11.73848065,
       12.67680439,  9.68324553,  8.49161154,  5.43943791, 12.53047489,
        6.19633139,  4.91652556, 13.27717982, 12.12641969,  7.42512621,
        5.76165934,  6.33075682,  9.5014146 , 11.27656488,  9.78147542,
        9.55593374, 10.0674887 , 13.25336001,  5.35774573, 11.018084  ,
       10.90638066,  6.76056625, 11.52733581, 11.77124373, 12.19805394,
       12.17901033,  7.72668467,  8.86082124,  7.72415048, 13.61801526,
        6.77777606, 12.1494023 ,  5.84450127, 13.83139024,  7.9337647 ,
        7.16058587,  7.50564223, 12.53285633,  5.5513072 ,  9.66977802,
       12.67576792,  7.02496223,  4.89121805,  7.32950385,  6.02789634,
        4.91652556,  6.24614905,  5.63824971, 13.96746816, 13.04051744,
        8.53014915,  8.35856487, 10.01804223,  5.05444288, 13.05148826,
       13.93790235,  9.79814997, 11.20100795,  5.29009836,  5.14607014,
        5.27178363, 13.61600536,  9.17905491,  5.62005629, 13.5247215 ,
        7.50917551,  5.39723241,  5.16077764, 10.61417181, 13.61801526,
       10.61047596, 11.04313641,  6.50409372,  7.67939546, 13.97430191,
       10.61047596, 11.63765273, 10.68416401, 12.3828957 , 12.04924159,
       12.38881635, 13.58101794, 12.07818871, 13.31764181, 13.04051744,
       11.76830981, 13.97779839,  6.41202986, 11.28093507, 13.40850328,
        5.19065439])}
{'P': tensor([[4.3999e-03, 1.9730e-01, 1.8451e-01,  ..., 2.4440e-03, 2.4066e-04,
         2.2279e-07],
        [1.4997e-05, 1.8277e-02, 1.7886e-02,  ..., 1.3483e-02, 1.2594e-03,
         1.7950e-06],
        [2.9963e-05, 1.7544e-02, 3.6537e-02,  ..., 1.4892e-02, 1.7128e-03,
         3.4411e-06],
        ...,
        [1.8350e-05, 1.4926e-04, 1.4694e-04,  ..., 1.1284e-01, 2.7512e-02,
         2.7509e-05],
        [9.5139e-05, 2.9766e-04, 2.5732e-04,  ..., 4.0262e-03, 2.2988e-01,
         6.5452e-04],
        [6.2779e-06, 2.9787e-06, 2.8154e-06,  ..., 1.4762e-06, 1.9454e-05,
         9.5146e-02]]), 'X': tensor([[ 11.5568,   6.0602,   9.3244],
        [  9.4925,   9.3811,   8.9842],
        [  9.4554,   9.4245,   9.8800],
        [  9.3560,   9.5391,   9.4943],
        [  9.6662,   9.1725,   8.1056],
        [  8.7131,  10.2216,   7.7915],
        [ -9.4351,  -1.8891,   3.5642],
        [  2.7121,   6.9743,  -8.4202],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.2921,  -1.7902,  -4.0870],
        [  4.2612, -10.5514,  -6.6268],
        [ 12.0582,  -2.6640,   0.7132],
        [  1.7218,  -4.7576,  -3.3617],
        [  1.9503,  -4.7172,  -6.2313],
        [  4.6063,  13.0051,  -3.1456],
        [  0.1677,  -4.7447,  -6.1529],
        [ 10.9931,   7.2107,   7.7933],
        [  7.5064,  -8.9651,  -0.0994],
        [ 11.2939,   6.6325,  -5.8311],
        [ -1.6334, -10.7207,   7.1397],
        [  3.6817,   6.3354,   6.1233],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.8902,  -1.3887,   4.0388],
        [ 12.5770,   1.8512,  -0.6578],
        [  7.9816,  10.8938,  -0.5378],
        [ -1.8788, -10.6567,   3.7068],
        [  6.1222,  -0.7261,   8.8204],
        [  2.7976,  -4.4682,  -4.2020],
        [ -6.8466,  -7.6488,   1.9648],
        [  2.5318,  13.7231,   6.3692],
        [ -1.5875,   7.8944,  -9.4286],
        [  1.5383, -11.0274,  -4.9666],
        [  3.2692,  13.5098,  -4.9032],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -7.2986,  -7.1225,  -3.0814],
        [  2.5663,   7.0526,  -6.8769],
        [  6.1973,  -0.4372,   6.5720],
        [ -9.4196,   0.0478,   9.4140],
        [  5.0233,   4.9647,  -3.1788],
        [ -3.9508,  -9.8515,   1.7923],
        [ 12.3357,   3.6001,   0.7903],
        [  4.3377,   5.7570,   1.3712],
        [  1.8766,   7.3694,   1.1887],
        [ -7.7842,   4.3466,  -1.7548],
        [  4.4290,  -3.4596,   1.4181],
        [ 11.9019,  -3.1379,  -3.4534],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.5101,  -2.1433,   9.5442],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.4149,  -4.6005,  -4.3339],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.2535,  -0.1709,  -9.4102],
        [ -3.5029,   7.5138,   8.1853],
        [  4.3459,  -3.5327,   0.3667],
        [ -3.1671,   7.6158,   6.0765],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -9.1813,  -3.3550,   1.8827],
        [ 11.5083,  -4.1177,   0.2389],
        [  3.5939,   6.4029,  -1.8357],
        [ 12.2914,   3.8037,   6.0978],
        [ -9.4505,  -1.7081,  -0.7039],
        [  9.0051,   9.9234,   5.4411],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.2480, -11.0153,  -3.0183],
        [  0.9478,  -4.8144,   0.6371],
        [ -7.4807,   4.7463,   2.0616],
        [  4.8674,  12.8875,   6.7622],
        [ -5.9987,   6.1949,  -6.2114],
        [ -9.4773,  -1.0024,   5.2599],
        [  5.0506,   4.9277,  -9.5094],
        [  9.2692,   9.6370,   7.0746],
        [  0.3493,  -4.7727,  -6.5682],
        [  3.2809,  13.5061,  -4.0199],
        [ -3.3812, -10.1235,   7.4384],
        [ -1.0407,   7.9166,  -6.6021],
        [ -6.4002,  -8.1048,  -0.9108],
        [ 11.3800,  -4.3919,   1.4066],
        [ -5.0943,  -9.1682,   0.5073],
        [  4.5430,   5.5438,  -3.8669],
        [ 12.5789,   0.1722,   1.6470],
        [  0.9967,  -4.8144,  -9.7370],
        [  4.4039,   5.6903,   8.7366],
        [  8.5093,  -8.1764,  -3.2381],
        [  0.6396,  14.0772,  -6.3786],
        [ -6.3306,   5.9272,   5.6935],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.1907,  -3.3180,  -5.5943],
        [  0.1206,  -4.7363,   8.7103],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.5128, -10.9365,   9.9645],
        [  2.0419,  -4.6978,  -5.0983],
        [ -8.8400,   2.3991,   1.3457],
        [  5.9026,   3.3837,  -0.6573],
        [ -1.1885, -10.8211,   0.0865],
        [ -8.4267,  -5.3697,   2.4172],
        [ -0.1626, -10.9796,   6.4629],
        [  6.3087,   1.8053,  -1.9481],
        [  7.5064,  -8.9651,  -0.0994],
        [  3.3943,   6.5483,   1.0790],
        [  6.3266,   0.3292,   9.3502],
        [  9.0652,   9.8597,  -3.5949],
        [  5.9003,  12.3550,   2.1917],
        [  1.8766,   7.3694,   1.1887],
        [ -8.5506,  -5.1154,   6.3423]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0557, 0.0595,  ..., 0.4927, 0.8385, 1.2762],
        [0.0557, 0.0000, 0.0089,  ..., 0.5250, 0.8885, 1.3301],
        [0.0595, 0.0089, 0.0000,  ..., 0.5335, 0.8949, 1.3313],
        ...,
        [0.4927, 0.5250, 0.5335,  ..., 0.0000, 0.4816, 1.2153],
        [0.8385, 0.8885, 0.8949,  ..., 0.4816, 0.0000, 0.8535],
        [1.2762, 1.3301, 1.3313,  ..., 1.2153, 0.8535, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.0493368 , 13.34587027, 13.35013374, 13.3614598 , 13.32557133,
       13.43127131,  9.62238828,  7.48311002, 13.01728313, 12.4217523 ,
       11.37939695, 12.3489324 ,  5.05963217,  5.10444097, 13.79676654,
        4.74770899, 13.14690137, 11.69264243, 13.09737053, 10.84437705,
        7.32754882,  5.8828986 ,  5.18277858,  6.05165248, 12.7125079 ,
       13.50484539, 10.82106505,  6.16512856,  5.27178363, 10.26545792,
       13.95472331,  8.05242591, 11.1341798 , 13.89974693, 12.05300943,
       10.1979669 ,  7.50500064,  6.21275016,  9.41970031,  7.06272256,
       10.61417181, 12.85032836,  7.20826219,  7.60463349,  8.91551231,
        5.62005629, 12.30858677, 12.3489324 ,  5.91222346, 11.63724792,
        5.19577146,  9.75200848,  6.25586297,  8.29022197,  5.60064319,
        8.24808832, 12.00995898,  9.7751189 , 12.22276278,  7.34250948,
       12.86648771,  9.60359158, 13.40024224, 10.78533838,  4.91296205,
       11.018084  ,  4.90676242,  8.85939626, 13.77604293,  8.62329466,
        9.53015376,  7.05627461, 13.37121923,  4.78544875, 13.89886469,
       10.67322527,  7.98469166, 10.32715715, 12.19805394, 10.48840476,
        7.16747878, 12.58006038,  4.91652556,  7.19534146, 11.80092578,
       14.09176259,  8.67227386,  8.90405823,  9.7712321 ,  4.73784222,
       12.10222204, 11.22142195,  5.1224079 ,  9.15976746,  6.80370254,
       10.88618114,  9.99211663, 10.98076873,  6.5618968 , 11.69264243,
        7.37576154,  6.33517349, 13.39372426, 13.69162795,  7.60463349,
        9.96391045])}
{'P': tensor([[1.8887e-03, 2.3421e-01, 2.2434e-01,  ..., 4.4025e-08, 1.5435e-05,
         2.3009e-08],
        [1.8066e-05, 4.1483e-03, 1.4101e-01,  ..., 8.3291e-08, 3.8391e-05,
         4.5939e-08],
        [1.1361e-05, 5.9610e-05, 4.0032e-03,  ..., 8.9028e-08, 4.9336e-05,
         4.3174e-08],
        ...,
        [3.7368e-05, 3.1183e-05, 3.5891e-05,  ..., 1.2134e-01, 8.0798e-05,
         1.5653e-04],
        [1.4951e-06, 1.5390e-06, 1.7244e-06,  ..., 2.5275e-05, 3.4074e-02,
         1.8146e-05],
        [1.4657e-04, 1.0851e-04, 9.3206e-05,  ..., 9.5696e-05, 3.6993e-06,
         2.7658e-02]]), 'X': tensor([[ 11.5456,   6.0861,   8.1101],
        [ 11.2571,   6.7072,   8.3693],
        [ 10.9931,   7.2107,   7.7933],
        [  9.6662,   9.1725,   8.1056],
        [  9.4925,   9.3811,   8.9842],
        [  9.2979,   9.6048,   7.3138],
        [ -6.3306,   5.9272,   5.6935],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.2990,  -4.2415,  -7.4857],
        [  6.3360,   0.4258,  -9.2509],
        [  1.2273,  -4.8081,   9.8129],
        [  7.7794,  11.0629,   8.9069],
        [ -2.6485, -10.4151,   0.8766],
        [  3.3633,  13.4794,   0.5193],
        [  1.6232,  -4.7717,  -2.5655],
        [  4.0457,  -3.7737,  -0.4704],
        [ -9.4735,  -0.7274,  -2.6749],
        [  3.2647, -10.8066,  -0.3768],
        [ -9.4773,  -1.0024,   5.2599],
        [  6.5428,  -9.5728,  -2.9897],
        [ 12.5650,   2.0132,   4.6272],
        [ -8.7735,   2.5640,  -6.3313],
        [ -1.0407,   7.9166,  -6.6021],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3200,   0.2690,   1.5900],
        [ 10.8670,  -5.3428,  -2.8637],
        [  2.6321,  -4.5296,  -2.5159],
        [ 12.4996,   2.6335,  -6.3166],
        [ -0.0410,   7.8591,   9.3457],
        [  2.6451,  -4.5250,  -0.5565],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -6.0844,  -8.3955,   0.7705],
        [  4.6063,  13.0051,  -3.1456],
        [  7.2340,  -9.1505,  -5.8321],
        [ -6.6895,   5.6051,   2.7899],
        [  1.6422,   7.4586,  -6.8056],
        [  1.3103,  -4.8032,  -0.9300],
        [  9.3424,  -7.3685,  -3.6807],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 11.2517,  -4.6495,   2.9800],
        [ -2.6485, -10.4151,   0.8766],
        [ 12.5919,   0.4035,  -5.8460],
        [  6.3516,   0.6425,   0.3425],
        [ -2.2508, -10.5479,  -2.7235],
        [  6.3413,   0.4883,  -6.8416],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.1204,  -5.9308,   4.0761],
        [ 12.4901,  -0.7052,   1.7315],
        [ 11.7349,  -3.5838,   2.1644],
        [ -7.4808,   4.7462,   2.0266],
        [  2.3373,  13.7721,  -0.2442],
        [ -5.1377,   6.7751,  -0.4438],
        [  6.3497,   1.3680,  -1.9076],
        [ -6.0195,   6.1789,  -2.9496],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.5814, -10.4485,   2.5496],
        [ -7.3333,   4.9246,  -7.9212],
        [  1.8719,   7.3713,  -3.7465],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -8.8400,   2.3991,   1.3457],
        [  6.1755,   2.5233,  -2.0042],
        [ 10.0995,  -6.4739,   2.0974],
        [ 11.7162,   5.6728,  -8.9372],
        [ -5.1377,   6.7751,  -0.4438],
        [  5.6314,  -1.9302,  -9.6543],
        [ 10.4758,  -5.9539,   1.3667],
        [  2.7685, -10.8982,  -6.8928],
        [  6.3569,   0.7591,   6.5011],
        [  3.0876,   6.7527,  -5.3540],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -4.7389,   6.9953,   3.7638],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.3367,   7.1676,  -9.5263],
        [ -0.3432,   7.8900,   7.1805],
        [  0.1382, -11.0073,   1.0026],
        [ -8.8934,   2.2597,   3.5831],
        [  3.6154,  -4.0640,  -7.9698],
        [ -0.8675, -10.8815,   6.8561],
        [ -9.4501,  -1.7133,   2.2980],
        [  9.5203,   9.3483,   5.8781],
        [ -8.9624,  -4.0944,  -4.9622],
        [  3.2809,  13.5061,  -4.0199],
        [  5.1793,   4.7461,   0.3899],
        [ -4.4520,   7.1370,  -8.7867],
        [ -9.4624,  -0.4633,  -7.3918],
        [  6.9693,  -9.3200,  -8.3822],
        [ 12.3025,  -1.7440,  -6.6632],
        [  6.3266,   0.3292,   9.3502],
        [  0.9809,  -4.8145,   6.8706],
        [  0.9809,   7.6642,  -1.4217],
        [  6.3031,   1.8477,  -7.1706],
        [  3.5284, -10.7484,   5.3347],
        [  6.3551,   0.7149,   1.3344],
        [ -8.9379,  -4.1655,  -2.8114],
        [  6.1372,   2.6726,  -6.7885],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -3.0004,   7.6607,  -7.2640],
        [ 11.2571,   6.7072,   8.3693],
        [  3.5939,   6.4029,  -1.8357],
        [ -5.6423,   6.4537,  -1.2091],
        [ -0.0282,   7.8575,  -1.1746],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.6614,  -4.5192,  -9.3641],
        [ -4.6393,   7.0461,   1.1624],
        [  2.6310, -10.9195,   3.8054]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0329, 0.0321,  ..., 1.1907, 0.9803, 0.9472],
        [0.0329, 0.0000, 0.0146,  ..., 1.2159, 1.0129, 0.9797],
        [0.0321, 0.0146, 0.0000,  ..., 1.2039, 1.0074, 0.9779],
        ...,
        [1.1907, 1.2159, 1.2039,  ..., 0.0000, 0.6577, 0.9133],
        [0.9803, 1.0129, 1.0074,  ..., 0.6577, 0.0000, 0.2610],
        [0.9472, 0.9797, 0.9779,  ..., 0.9133, 0.2610, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.05148826, 13.103709  , 13.14690137, 13.32557133, 13.34587027,
       13.36800073,  8.67227386,  5.53255882,  5.37343173,  6.35029274,
        4.96231532, 13.52430176, 10.74655346, 13.89264822,  5.04028538,
        5.53255882,  9.5014146 , 11.28895489,  9.53015376, 11.59512194,
       12.72524615,  9.14044984,  7.98469166, 12.12448912,  9.09963894,
        6.32572323, 12.10942412,  5.23878371, 12.77402351,  7.85919687,
        5.24137909, 10.09979747, 10.36844813, 13.79676654, 11.66453059,
        8.7273569 ,  7.63726134,  4.978704  , 11.89855267, 12.11735334,
       12.17451685, 10.74655346, 12.59840193,  6.38399789, 10.78533838,
        6.36004393,  9.12220271, 10.05558893, 12.50996828, 12.26997165,
        8.859416  , 13.96905756,  8.50278896,  6.49538841,  8.62631801,
       10.74409834, 11.4088043 ,  8.83341608,  7.60529892,  9.69792513,
        9.15976746,  6.67108662, 11.99633819, 13.01728313,  8.50278896,
        5.95297459, 12.0495521 , 11.24434374,  6.4020404 ,  7.42512621,
       10.14436912,  8.44940661, 11.75807252,  7.53884104,  7.89745107,
       11.00812996,  9.17595899,  5.43943791, 10.91601655,  9.60412735,
       13.3426541 ,  9.85330646, 13.89886469,  7.02496223,  8.41171369,
        9.47369628, 11.63765273, 12.42554808,  6.33517349,  4.91336994,
        7.72668467,  6.56833864, 11.31276109,  6.39520397,  9.86089579,
        6.69389431, 12.35625694,  8.22728399, 13.103709  ,  7.34250948,
        8.5724042 ,  7.8575648 , 12.18045978,  5.24461068,  8.43626114,
       11.23201619])}
{'P': tensor([[1.8486e-03, 2.2097e-01, 1.8169e-01,  ..., 2.2806e-08, 1.8169e-01,
         6.8146e-04],
        [1.1582e-05, 5.1650e-03, 1.6351e-05,  ..., 5.9844e-08, 1.6351e-05,
         2.4464e-03],
        [1.1181e-05, 3.5994e-01, 2.2389e-03,  ..., 2.6379e-08, 2.2389e-03,
         9.7176e-04],
        ...,
        [5.2734e-05, 2.9792e-05, 3.9048e-05,  ..., 3.9526e-02, 3.9048e-05,
         1.8630e-05],
        [1.1181e-05, 3.5994e-01, 2.2389e-03,  ..., 2.6379e-08, 2.2389e-03,
         9.7176e-04],
        [4.2727e-05, 6.5445e-05, 5.2486e-05,  ..., 1.5810e-05, 5.2486e-05,
         6.0285e-02]]), 'X': tensor([[ 11.4743,   6.2478,  -2.1100],
        [ 10.7566,   7.6212,  -2.3279],
        [ 11.1420,   6.9331,  -2.4779],
        [ 10.3251,   8.2928,  -3.0006],
        [  9.6614,   9.1784,  -1.2087],
        [  9.7328,   9.0902,  -0.8917],
        [  1.8320,   7.3871,   5.1452],
        [  7.1697,  11.5343,  -0.7276],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.4721,  -1.3097,  -7.4377],
        [  6.3534,   1.2990,  -7.2942],
        [  3.5965,  -4.0754,  -3.5035],
        [ -7.3446,   4.9113,  -5.9356],
        [  5.1142,  -2.7230,  -4.1966],
        [ -5.6174,   6.4708,  -1.7339],
        [  2.2373,  -4.6506,  -1.7334],
        [ -2.5161,   7.7699,  -9.3531],
        [ 11.5935,   5.9740,  -4.3952],
        [ 12.2692,  -1.8882,   3.9253],
        [  1.8719,   7.3713,  -3.7465],
        [  0.6396,  14.0772,  -6.3786],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -6.1899,  -8.3011,  -7.9269],
        [  4.7156,  -3.1831,  -8.0224],
        [  2.6970,   6.9826,   9.7350],
        [ -2.8162,   7.7059,  -9.6885],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.0047,   3.1062,  -6.7914],
        [  0.6757, -11.0361,  -1.3047],
        [ 10.0400,   8.6922,  -6.2847],
        [  0.6757, -11.0361,  -1.3047],
        [  2.8602,   6.8905,   5.8779],
        [  3.3069,   6.6089,   8.7377],
        [  5.7963,  -1.6018,   1.8856],
        [  9.9217,  -6.7006,  -5.6728],
        [  3.6154,  -4.0640,  -7.9698],
        [  9.4925,   9.3811,   8.9842],
        [  3.6249, -10.7255,   5.6605],
        [ 11.2939,   6.6325,  -5.8311],
        [  3.6817,   6.3354,   6.1233],
        [  0.5250, -11.0306,   5.5358],
        [  1.1350,  14.0104,   0.6321],
        [  2.2437,  -4.6489,   6.1430],
        [  2.1120,  -4.6818,   8.6144],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.4283,  -1.9585,  -0.9129],
        [  0.7145,  -4.8071,  -2.3766],
        [  1.0000,   7.6591,  -4.5011],
        [ -9.1813,  -3.3550,   1.8827],
        [ -6.7992,   5.4991,   9.9186],
        [  0.8572,  14.0501,  -8.3364],
        [  2.7590,  13.6621,   9.4496],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.4150,  13.9647,  -4.0915],
        [ -4.7334,   6.9982,  -0.4392],
        [  0.1287, -11.0065,   6.5757],
        [ -3.1204, -10.2345,   9.2288],
        [  5.1861,   4.7361,  -3.7312],
        [ 10.3251,   8.2928,  -3.0006],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -6.9838,  -7.4965,   0.9796],
        [  2.7420,  -4.4896,  -3.7139],
        [ -7.4887,   4.7364,   8.9074],
        [ -1.6334, -10.7207,   7.1397],
        [ -5.3382,   6.6533,   7.8054],
        [  5.8930,  12.3592,  -8.9201],
        [ 12.6000,   1.3839,  -3.2992],
        [ -9.1706,   1.3863,   8.4032],
        [ -3.0004,   7.6607,  -7.2640],
        [  1.7500,   7.4187,   4.4237],
        [  8.6271,  -8.0715,  -7.4813],
        [ 10.0995,  -6.4739,   2.0974],
        [ -1.9646,   7.8571,   3.3817],
        [  4.2339,  -3.6268,   6.3125],
        [  2.7558,  -4.4844,   0.0636],
        [  9.3560,   9.5391,   9.4943],
        [ 11.7418,   5.6070,  -1.4192],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.2226,  -4.2802,  -4.0428],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.8162,   7.7059,  -9.6885],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.8318,   3.5567,  -4.7692],
        [  2.7558,  -4.4844,   0.0636],
        [  6.3589,   0.8239,  -8.4230],
        [  4.0642,  -3.7599,  -3.4565],
        [  0.6952,  -4.8060,   8.8976],
        [  5.1472,  -2.6801,   8.1194],
        [ -5.2589,   6.7024,  -5.5809],
        [ 10.0400,   8.6922,  -6.2847],
        [  5.6672,  -1.8631,   8.1020],
        [  9.6662,   9.1725,   8.1056],
        [ -9.3464,   0.5708,  -1.6397],
        [ -9.4196,   0.0478,   9.4140],
        [  2.0899,  -4.6870,  -7.4260],
        [ -0.8569,   7.9156,  -6.5557],
        [ -9.1749,  -3.3803,   6.4629],
        [  5.6314,  -1.9302,  -9.6543],
        [  2.7420,  -4.4896,  -3.7139],
        [ 10.4758,  -5.9539,   1.3667],
        [  5.1292,   4.8183,  -5.7997],
        [ 11.9122,   5.1397,  -1.5498],
        [ -0.9715, -10.8630,   0.3555],
        [ 11.1420,   6.9331,  -2.4779],
        [  1.8719,   7.3713,  -3.7465]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.0639e-02, 3.7167e-02,  ..., 9.9778e-01, 3.7167e-02,
         4.0273e-01],
        [3.0639e-02, 2.1453e-04, 1.4428e-02,  ..., 1.0220e+00, 1.4428e-02,
         4.1940e-01],
        [3.7167e-02, 1.4428e-02, 0.0000e+00,  ..., 1.0168e+00, 0.0000e+00,
         4.1029e-01],
        ...,
        [9.9778e-01, 1.0220e+00, 1.0168e+00,  ..., 0.0000e+00, 1.0168e+00,
         6.3704e-01],
        [3.7167e-02, 1.4428e-02, 0.0000e+00,  ..., 1.0168e+00, 0.0000e+00,
         4.1029e-01],
        [4.0273e-01, 4.1940e-01, 4.1029e-01,  ..., 6.3704e-01, 4.1029e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.06498554, 13.18278606, 13.12299145, 13.2430377 , 13.32613445,
       13.31764181,  7.61088125, 13.58101794,  9.37296102,  9.562174  ,
        6.48486171,  5.43544626,  8.8353674 ,  5.79391825,  8.56889738,
        5.16077764,  8.16714836, 13.04217487, 12.41366754,  7.60529892,
       14.09176259, 12.07818871, 10.35485331,  5.68942572,  7.48538905,
        8.20436764,  8.89458273,  6.76056625, 11.05672136, 13.27994304,
       11.05672136,  7.46053281,  7.39004226,  6.01356004, 11.97238624,
        5.43943791, 13.34587027, 11.32149259, 13.09737053,  7.32754882,
       11.04313641, 14.05633441,  5.16205087,  5.13616224, 10.75443106,
        9.62958973,  4.85995002,  7.72415048,  9.7751189 ,  8.74470218,
       14.07623266, 13.93790235,  5.84450127, 14.03618649,  8.4486771 ,
       11.00726483, 10.69963954,  7.02326515, 13.2430377 ,  4.89869205,
        9.5642993 , 10.24557047,  5.26067019,  8.86082124, 10.84437705,
        8.53014915, 13.6922425 , 12.67576792,  9.27474386,  8.22728399,
        7.62232307, 11.81423161, 11.99633819,  8.09899874,  5.5748624 ,
        5.26342203, 13.3614598 , 13.01187484, 12.27993098,  5.35774573,
        9.34127013,  8.20436764,  9.4997987 ,  6.83083863,  5.26342203,
        6.41202986,  5.5366698 ,  4.85603886,  5.80312949,  8.51928531,
       13.27994304,  5.96556248, 13.32557133,  9.36378438,  9.41970031,
        5.13182506,  7.96181288,  9.77777346,  5.95297459,  5.26067019,
       12.0495521 ,  7.03733775, 12.97370371, 10.90638066, 13.12299145,
        7.60529892])}
{'P': tensor([[3.1547e-03, 1.3995e-01, 1.3695e-01,  ..., 4.3829e-08, 2.9115e-08,
         7.6867e-07],
        [4.1543e-06, 5.2978e-03, 1.7671e-03,  ..., 4.4284e-08, 5.3673e-08,
         8.4385e-07],
        [5.8697e-06, 1.7595e-02, 7.2007e-03,  ..., 5.9519e-08, 7.0568e-08,
         1.1195e-06],
        ...,
        [5.4559e-04, 1.0944e-04, 1.0806e-04,  ..., 4.7539e-02, 2.3438e-05,
         3.5284e-03],
        [1.5013e-05, 4.4474e-06, 4.3354e-06,  ..., 6.7314e-02, 5.5822e-02,
         1.2704e-02],
        [1.1392e-03, 4.5850e-04, 4.4319e-04,  ..., 7.3663e-03, 5.4682e-04,
         1.1979e-01]]), 'X': tensor([[ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0796, 0.0907,  ..., 0.8840, 1.3453, 0.7105],
        [0.0796, 0.0000, 0.0137,  ..., 0.9634, 1.4249, 0.7845],
        [0.0907, 0.0137, 0.0000,  ..., 0.9734, 1.4354, 0.7975],
        ...,
        [0.8840, 0.9634, 0.9734,  ..., 0.0000, 0.4667, 0.4156],
        [1.3453, 1.4249, 1.4354,  ..., 0.4667, 0.0000, 0.7524],
        [0.7105, 0.7845, 0.7975,  ..., 0.4156, 0.7524, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.06957425, 13.37121923, 13.36800073, 13.3426541 , 13.32557133,
       13.14690137,  9.97220799, 13.44593519,  9.09963894,  8.99791316,
       13.27994304, 10.07238463, 13.03002338,  4.91652556,  8.4486771 ,
       13.36800073,  9.37296102, 10.04916678,  5.48276038,  8.67227386,
        5.43943791,  6.27905292, 11.33837396,  5.24461068, 10.91601655,
        9.75200848, 10.91805683,  4.91652556, 12.58006038, 12.41366754,
       11.85890664, 12.44177641,  9.86089579,  4.91336994, 12.53345445,
        7.59795445, 12.11735334, 10.37998078,  9.93553424,  7.42512621,
        7.5403822 , 13.45642379, 11.018084  , 12.07225469, 13.97857591,
       14.02532758, 12.35625694,  6.74686071,  7.96575782, 13.97857591,
        6.38399789,  8.29640613,  4.91038884, 12.12448912, 10.33712258,
       11.9702762 ,  6.54615887,  9.70108409,  6.89245352,  8.14433542,
        9.09502385, 13.13205357, 11.56146873,  8.31625171, 10.01804223,
       13.18278606,  8.38252937, 12.97370371, 13.32557133,  7.37576154,
        5.21280156, 10.51340829,  5.37343173,  9.13507321,  6.38105015,
       13.89264822,  9.18690018,  9.17595899,  7.20826219,  9.88571713,
        9.17016857, 12.13568598,  6.74686071, 13.85158417, 12.53345445,
        9.85384623,  9.69629   ,  5.89074104, 11.35182589,  7.19513806,
        5.60064319,  6.31576264,  5.89074104,  8.09899874,  7.16598273,
       13.6105595 ,  6.24614905,  5.68942572, 10.69963954,  8.38252937,
        8.09842704, 13.38091121,  5.87132718, 11.4088043 , 10.67132524,
        5.14999799])}
{'P': tensor([[4.3278e-03, 3.0083e-01, 2.0595e-01,  ..., 4.6403e-08, 2.5839e-07,
         2.2972e-03],
        [6.2077e-06, 4.3154e-03, 5.5057e-02,  ..., 2.9437e-08, 2.4472e-07,
         2.2667e-03],
        [4.0215e-06, 8.7647e-05, 4.0042e-03,  ..., 1.9792e-08, 1.6395e-07,
         1.9459e-03],
        ...,
        [4.0098e-04, 1.9607e-04, 1.0239e-04,  ..., 5.5935e-02, 4.7645e-04,
         6.0823e-05],
        [1.3289e-04, 9.9558e-05, 5.9814e-05,  ..., 2.5864e-02, 1.7698e-01,
         1.6159e-04],
        [3.7402e-04, 9.6138e-04, 1.3807e-03,  ..., 1.1139e-05, 8.9332e-05,
         2.5354e-01]]), 'X': tensor([[ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.8356e-02, 9.3621e-02,  ..., 1.1513e+00, 1.0043e+00,
         4.1971e-01],
        [4.8356e-02, 0.0000e+00, 1.0904e-01,  ..., 1.1381e+00, 9.8496e-01,
         4.4219e-01],
        [9.3621e-02, 1.0904e-01, 0.0000e+00,  ..., 1.2431e+00, 1.0928e+00,
         5.0311e-01],
        ...,
        [1.1513e+00, 1.1381e+00, 1.2431e+00,  ..., 0.0000e+00, 2.0800e-01,
         8.8415e-01],
        [1.0043e+00, 9.8496e-01, 1.0928e+00,  ..., 2.0800e-01, 0.0000e+00,
         7.9961e-01],
        [4.1971e-01, 4.4219e-01, 5.0311e-01,  ..., 8.8415e-01, 7.9961e-01,
         1.5170e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.0743399 , 13.2778378 , 13.3426541 , 13.40024224, 13.35927527,
       13.46553623,  7.46248947,  9.27474386, 12.45418236,  9.48969978,
       10.69963954, 11.73848065,  4.73784222,  5.60064319,  9.18690018,
       10.9036226 , 11.83075005,  7.26074925, 12.07818871,  7.83534987,
        6.27905292,  7.19513806, 10.54658597,  9.58072286, 13.97857591,
        9.41244865, 10.30972193, 13.06498554,  8.34949383, 13.77604293,
        7.32233733,  8.05668609,  8.31625171, 13.95472331,  4.96231532,
       13.31764181,  5.39450305, 13.06498554,  8.14433542,  6.41202986,
       13.68063098, 12.62576026,  5.05774295,  9.15253344,  9.09119351,
        9.88571713, 13.08414689,  5.77219077, 11.46176259,  6.76056625,
       12.4217523 , 13.55868684,  5.24137909,  9.05843201,  4.87083348,
        6.21275016, 13.55868684, 13.95361098, 10.794755  ,  6.28961988,
       10.61417181,  9.62238828,  4.91038884, 12.76408562, 10.82106505,
        5.46244988, 12.62576026,  9.48337063, 11.4088043 ,  4.92497514,
        5.79543038,  9.87720102, 10.84493732, 10.42667093, 13.97857591,
        9.41244865,  5.01114719,  6.05113217,  6.05113217, 10.14044489,
       14.05687815, 10.794755  ,  5.82345867, 11.27160035,  9.77777346,
       12.41012046,  8.38252937,  6.67108662, 11.0050379 , 12.22503754,
       12.78327037,  5.79391825,  6.76661595,  5.62005629,  6.54127431,
        7.71075226, 10.46739073,  6.28835101, 10.32715715, 11.51557166,
       14.08549802,  5.16205087,  7.32950385, 11.66453059,  4.89121805,
        7.44805901])}
{'P': tensor([[4.4019e-03, 1.9512e-01, 1.9398e-01,  ..., 3.3835e-05, 5.2119e-04,
         5.4375e-05],
        [7.5718e-06, 7.8234e-03, 1.9430e-05,  ..., 1.7443e-04, 1.2437e-04,
         1.9481e-04],
        [9.5250e-06, 4.0770e-01, 3.5347e-03,  ..., 4.9153e-05, 1.9858e-04,
         9.2108e-05],
        ...,
        [2.1318e-06, 2.6308e-06, 2.3110e-06,  ..., 4.8297e-02, 1.1905e-06,
         3.1348e-05],
        [5.1556e-04, 9.1585e-03, 3.9438e-03,  ..., 3.2580e-04, 9.0986e-02,
         2.4193e-02],
        [1.0107e-05, 2.5822e-05, 1.9849e-05,  ..., 1.0993e-03, 4.1738e-05,
         9.6242e-02]]), 'X': tensor([[ 11.3691,   6.4760,   1.0787],
        [  9.3624,   9.5318,   0.2779],
        [ 10.2470,   8.4054,  -0.1177],
        [ 11.1469,   6.9237,   1.6160],
        [  9.6587,   9.1817,  -0.8212],
        [  9.7328,   9.0902,  -0.8917],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.1749,  -3.3803,   6.4629],
        [  8.6420,  -8.0580,   6.5781],
        [ -4.0409,   7.3169,  -3.5779],
        [  4.5491,  -3.3485,   6.8412],
        [ -9.2508,   1.0572,   2.9978],
        [ 12.5832,   1.7543,  -7.6134],
        [  6.0073,   3.0986,   6.7937],
        [ 11.5198,  -4.0923,   5.5088],
        [  4.4039,   5.6903,   8.7366],
        [  3.2809,  13.5061,  -4.0199],
        [ -5.3736,   6.6310,  -1.5174],
        [  5.8323,  -1.5228,   5.1132],
        [ 12.2592,   3.9425,   2.0383],
        [  5.6928,  12.4710,   3.6001],
        [  9.2692,   9.6370,   7.0746],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.6000,   1.3839,  -3.2992],
        [  2.5663,   7.0526,  -6.8769],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.4901,  -0.7052,   1.7315],
        [  4.3459,  -3.5327,   0.3667],
        [ -5.6651,   6.4380,   1.3660],
        [ -2.0840,   7.8415,  -7.8657],
        [  2.1120,  -4.6818,   8.6144],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -9.1658,  -3.4155,  -2.8880],
        [  5.0277,   4.9589,   7.0175],
        [  3.3167,   6.6022,   6.3554],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.4157,   0.0821,  -5.0025],
        [  5.8201,   3.5841,  -8.0609],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.0348,  -2.8229,   3.5436],
        [  6.9653,  -9.3225,  -3.4727],
        [ -5.6651,   6.4380,   1.3660],
        [ -8.4573,   3.2448,   5.4370],
        [  3.0876,   6.7527,  -5.3540],
        [ 10.5678,   7.9261,  -0.6100],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.1960,  -3.2964,   1.6921],
        [ 11.5966,   5.9666,   6.0083],
        [ -7.8280,   4.2848,  -7.1955],
        [  4.1348,   5.9514,  -0.8856],
        [ 10.6283,  -5.7253,   7.4878],
        [  3.3839,  -4.1966,   8.3574],
        [  7.7750,  11.0665,   2.8364],
        [ -3.1671,   7.6158,   6.0765],
        [  3.1892, -10.8220,  -6.8409],
        [ -2.5056,   7.7720,   5.5878],
        [  2.4932,   7.0903,  -5.4214],
        [  0.4996,   7.7738,  -4.4711],
        [ -3.5903,   7.4847,  -8.9112],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 12.6000,   1.3839,  -3.2992],
        [ 12.2589,  -1.9312,   9.8657],
        [ -6.7643,  -7.7372,  -9.6663],
        [  7.4132,  11.3526,   0.3322],
        [ 11.5456,   6.0861,   8.1101],
        [  0.8572,  14.0501,  -8.3364],
        [ -8.7078,   2.7182,   1.8146],
        [  1.5246,  -4.7838,  -5.2095],
        [  2.3261,  13.7749,   5.9121],
        [ -8.8835,  -4.3178,   9.3080],
        [  0.2480, -11.0153,  -3.0183],
        [  0.1677,  -4.7447,  -6.1529],
        [  5.0617,  12.7957,   8.3544],
        [ -9.4430,  -0.1897,   4.8383],
        [  2.3261,  13.7749,   5.9121],
        [  2.5829,  -4.5466,  -8.4015],
        [ -9.3331,  -2.6465,  -8.2028],
        [  3.4604,  -4.1545,   1.8091],
        [ -1.0012, -10.8576,   2.2867],
        [  6.2576,   2.1359,  -3.3398],
        [ -0.8885,   7.9160,   0.0571],
        [  2.1440, -10.9811,   8.5281],
        [ 12.2914,   3.8037,   6.0978],
        [  2.7420,  -4.4896,  -3.7139],
        [ -9.1960,  -3.2964,   1.6921],
        [ -9.1178,   1.5784,   2.1410],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.2911,  -2.4824,  -3.6877],
        [  4.8653,   5.1694,  -4.0823],
        [ -3.2736, -10.1703,   3.8712],
        [  9.5203,   9.3483,   5.8781],
        [ -3.0644,   7.6439,  -1.4322],
        [ -1.6086,   7.8928,   4.3385],
        [ -2.3317,   7.8034,  -3.6274],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -2.2188,   7.8218,   8.6064],
        [ -6.7745,  -7.7264,   8.4684],
        [  3.6469,   6.3624,  -1.8414],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.1931,  -4.7491,   7.4385],
        [  1.5653,  13.9377,   4.1773],
        [ -8.5859,  -5.0393,   0.5476],
        [ -6.2668,   5.9808,   2.5614],
        [ 10.1503,   8.5414,  -7.2681],
        [ -0.2586,   7.8825,  -9.9840]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0890, 0.1115,  ..., 0.7766, 0.6554, 0.7987],
        [0.0890, 0.0000, 0.0291,  ..., 0.8319, 0.5750, 0.7723],
        [0.1115, 0.0291, 0.0000,  ..., 0.8301, 0.5474, 0.7474],
        ...,
        [0.7766, 0.8319, 0.8301,  ..., 0.0000, 1.1176, 0.7421],
        [0.6554, 0.5750, 0.5474,  ..., 1.1176, 0.0000, 0.5842],
        [0.7987, 0.7723, 0.7474,  ..., 0.7421, 0.5842, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.08414689, 13.3607379 , 13.25336001, 13.12218013, 13.32645552,
       13.31764181,  5.1224079 ,  9.77777346, 11.8159254 ,  8.35856487,
        5.64865173,  9.3109884 , 12.70489563,  6.7593917 , 12.22503754,
        7.19534146, 13.89886469,  8.53501777,  6.02779201, 12.87752052,
       13.70893492, 13.37121923, 12.38881635, 12.67576792,  7.50500064,
       13.96905756, 12.50996828,  5.60064319,  8.57560275,  8.1137452 ,
        5.13616224,  9.99145096,  9.78147542,  7.06169391,  7.38844703,
        5.24137909,  9.41605684,  6.83515201, 10.1979669 ,  5.77219077,
       11.63724792,  8.57560275,  9.05843201,  7.42512621, 13.20988996,
        4.89121805,  9.76896904, 13.04156075,  8.92395381,  7.24677664,
       12.07225469,  5.39098669, 13.5247215 ,  8.24808832, 11.28215429,
        8.16584537,  7.5158489 ,  7.78980299,  8.30125201,  9.60359158,
       12.67576792, 12.41012046, 10.27716434, 13.55868684, 13.05148826,
       14.07623266,  9.12220271,  5.0209005 , 13.96987806,  9.87720102,
       11.018084  ,  4.74770899, 13.76047999,  9.44486609, 13.96987806,
        5.22903117,  9.70108409,  5.40689332, 10.9036226 ,  6.61211224,
        7.96575782, 11.18839169, 12.86648771,  5.26067019,  9.76896904,
        9.25336371, 10.39942497,  5.84450127,  7.09887557, 10.68416401,
       13.3426541 ,  8.23525433,  8.0550286 ,  8.14433542, 12.58006038,
        9.75401449,  8.1303945 , 10.27572673,  7.33350642, 10.7167285 ,
        4.75303127, 14.02532758,  9.95552626,  8.66273171, 13.26590071,
        7.88677538])}
{'P': tensor([[2.3658e-03, 2.1752e-01, 1.3290e-01,  ..., 9.0531e-05, 5.0933e-07,
         2.4466e-06],
        [5.4954e-06, 2.9507e-03, 1.6087e-04,  ..., 1.0620e-04, 9.9967e-07,
         1.8621e-06],
        [2.2352e-05, 2.9799e-02, 5.2858e-03,  ..., 3.1747e-04, 1.5909e-06,
         3.8069e-06],
        ...,
        [2.2188e-05, 3.4693e-05, 3.6863e-05,  ..., 6.6741e-02, 4.4374e-04,
         5.2867e-05],
        [1.1772e-06, 9.0876e-07, 9.8359e-07,  ..., 6.1132e-06, 3.3882e-02,
         1.3468e-04],
        [2.5665e-02, 1.7163e-02, 1.9263e-02,  ..., 2.3013e-03, 1.6050e-04,
         1.9201e-01]]), 'X': tensor([[ 11.3118,   6.5957,  -4.5667],
        [ 10.2712,   8.3707,  -4.9743],
        [ 10.5955,   7.8824,  -3.7006],
        [  9.3607,   9.5337,  -4.5412],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.1798,   8.5002,  -5.8893],
        [  0.2732,  -4.7619,  -1.1577],
        [  5.5626,  -2.0536,   8.4433],
        [  3.1892, -10.8220,  -6.8409],
        [ 10.9916,   7.2133,  -0.9179],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.7218,  -9.9934,   6.5764],
        [  5.3924, -10.1396,  -4.0863],
        [  3.4140,  -4.1803,  -8.8224],
        [ -9.4378,  -1.8606,  -2.5668],
        [ -8.5607,  -5.0938,  -5.4531],
        [  1.4150,  13.9647,  -4.0915],
        [ -1.0407,   7.9166,  -6.6021],
        [  6.2247,  -0.3150,  -9.3657],
        [  3.0719, -10.8449,  -4.1673],
        [ -3.9700,   7.3453,  -4.1486],
        [  5.4867,  12.5816,   2.9549],
        [ -0.1963,   7.8765,  -1.1371],
        [ -7.6650,   4.5092,   0.2520],
        [ -9.4671,  -0.5553,   0.1117],
        [ -5.1369,   6.7755,   8.3836],
        [  9.1620,  -7.5573,  -9.9882],
        [  8.8474,  10.0868,  -8.8644],
        [  4.2339,  -3.6268,   6.3125],
        [ -1.9714,   7.8563,  -1.8084],
        [ -0.8675, -10.8815,   6.8561],
        [  0.4996,   7.7738,  -4.4711],
        [  4.5491,  -3.3485,   6.8412],
        [  3.7992,  -3.9474,  -8.9019],
        [  6.3515,   0.6418,  -6.5823],
        [  0.5175,  14.0910,   4.6540],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.0574,   1.7801,   8.8986],
        [ -7.2515,   5.0195,   4.6434],
        [  2.9410,   6.8428,   5.4458],
        [  2.3373,  13.7721,  -0.2442],
        [ -8.1531,  -5.8748,   2.6934],
        [  7.9403,  -8.6455,  -8.7433],
        [ -8.1204,  -5.9308,   4.0761],
        [ -9.4203,  -2.0340,   2.5856],
        [ -7.0948,   5.1940,   9.8171],
        [ -2.5056,   7.7720,   5.5878],
        [ -4.4520,   7.1370,  -8.7867],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -9.2345,  -3.1345,   3.9715],
        [ -8.1905,   3.7253,   4.1175],
        [ -7.0825,   5.2073,  -5.0497],
        [  9.3836,   9.5075,  -5.9278],
        [  2.2860,   7.1916,  -6.1011],
        [ -3.2053, -10.1993,  -3.8736],
        [ -1.8823, -10.6558,  -3.8952],
        [  5.7963,  -1.6018,   1.8856],
        [ 10.0619,   8.6627,  -9.2884],
        [ 12.2914,   3.8037,   6.0978],
        [  8.0877,  -8.5297,  -1.1050],
        [  6.0031,  12.2958,  -6.5992],
        [  4.9452, -10.3187,  -7.8621],
        [ -3.1671,   7.6158,   6.0765],
        [ -5.9940,   6.1985,   5.9566],
        [  1.9505,  -4.7171,   9.9868],
        [ -9.4721,  -1.3097,  -7.4377],
        [  3.1756, -10.8247,   6.8882],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.5240,  -2.1198,   6.2055],
        [ 12.5763,   0.1333,  -9.2650],
        [ -1.6906,   7.8860,   7.7142],
        [  2.0243,   7.3086,   7.5432],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 12.0772,  -2.6014,  -7.1832],
        [  9.3607,   9.5337,  -4.5412],
        [ 12.5832,   1.7543,  -7.6134],
        [  9.7328,   9.0902,  -0.8917],
        [  4.5814, -10.4485,   2.5496],
        [ -9.4645,  -1.4880,  -2.0226],
        [ -7.2986,  -7.1225,  -3.0814],
        [  5.8617,   3.4853,  -9.8080],
        [  4.5057,  -3.3894,  -0.8160],
        [ -1.5261,   7.8988,   9.3768],
        [  0.5250, -11.0306,   5.5358],
        [  2.5010,  -4.5736,  -0.8989],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.4149,  -4.6005,  -4.3339],
        [ -8.8363,   2.4086,  -5.6296],
        [  8.0877,  -8.5297,  -1.1050],
        [ -6.3066,  -8.1936,   9.5822],
        [  2.5318,  13.7231,   6.3692],
        [  9.6587,   9.1817,  -0.8212],
        [  2.0729,   7.2878,   7.0880],
        [ -2.1382,   7.8339,   7.3524],
        [  2.0243,   7.3086,   7.5432],
        [ -9.3384,  -2.6161,  -4.7630],
        [  1.1274,  14.0116,   8.1753],
        [  1.3705,  13.9723,   7.9266],
        [ -9.1459,   1.4780,  -4.2403],
        [ -5.9872,   6.2037,  -9.4477],
        [ 10.0400,   8.6922,  -6.2847],
        [ -4.4779,  -9.5609,  -2.4505],
        [  1.0000,   7.6591,  -4.5011],
        [  1.4489,   7.5256,   4.4529],
        [ -9.4735,  -1.2620,  -3.6742],
        [  5.4732,  -2.2041,  -4.3830]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0327, 0.0711,  ..., 0.8982, 1.2056, 0.2848],
        [0.0327, 0.0000, 0.0986,  ..., 0.9308, 1.2267, 0.3136],
        [0.0711, 0.0986, 0.0000,  ..., 0.8410, 1.2073, 0.2675],
        ...,
        [0.8982, 0.9308, 0.8410,  ..., 0.0000, 1.0035, 0.6805],
        [1.2056, 1.2267, 1.2073,  ..., 1.0035, 0.0000, 0.9448],
        [0.2848, 0.3136, 0.2675,  ..., 0.6805, 0.9448, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.09426142, 13.25017211, 13.20597932, 13.36092624, 13.32374056,
       13.26209545,  4.76969364,  5.92952538, 11.28215429, 13.1471313 ,
       12.38881635, 11.51557166, 11.48434767,  5.39723241,  9.61942445,
        9.96152788, 14.03618649,  7.98469166,  6.23262289, 11.27160035,
        8.34949383, 13.72594022,  7.8788965 ,  8.89301516,  9.48337063,
        8.50268322, 11.87665501, 13.41713661,  5.5748624 ,  8.09984231,
       10.91601655,  7.78980299,  5.64865173,  5.47866221,  6.383897  ,
       14.10045797, 11.62158554,  9.23071698,  8.81930525,  7.44805901,
       13.96905756, 10.04916678, 11.73848065, 10.05558893,  9.63742669,
        8.79284466,  8.16584537,  8.41171369, 12.07818871,  9.75200848,
        8.99791316,  8.7907923 , 13.35832725,  7.54620849, 10.6910768 ,
       10.8207297 ,  6.01356004, 13.27717982, 12.86648771, 11.75438006,
       13.6829843 , 11.4424816 ,  8.24808832,  8.62260327,  5.10446653,
        9.562174  , 11.28093507,  9.36093562,  5.91677188, 12.5769659 ,
        8.06517143,  7.58378655, 10.53368644, 12.35421461, 13.36092624,
       12.70489563, 13.31764181, 11.4088043 ,  9.58072286, 10.1979669 ,
        6.81961996,  5.63824971,  8.04483495, 11.04313641,  5.21280156,
        9.41605684,  5.19577146,  9.15866607, 11.75438006, 10.33958639,
       13.95472331, 13.32645552,  7.57685908,  8.12043983,  7.58378655,
        9.69792513, 14.05687815, 14.0393949 ,  9.26455795,  8.62162372,
       13.27994304, 10.5575601 ,  7.72415048,  7.66377554,  9.55720651,
        5.90035324])}
{'P': tensor([[1.7535e-03, 2.9087e-01, 1.7314e-01,  ..., 3.8154e-05, 1.8862e-07,
         1.7895e-07],
        [7.9432e-06, 2.1140e-03, 2.5666e-01,  ..., 3.6340e-05, 1.9120e-07,
         2.3877e-07],
        [4.4645e-06, 5.6928e-06, 2.7493e-03,  ..., 2.2621e-05, 1.5177e-07,
         4.2603e-07],
        ...,
        [1.1954e-02, 1.4124e-02, 1.8487e-02,  ..., 1.0516e-01, 2.3660e-04,
         1.0204e-04],
        [1.7025e-02, 1.3190e-02, 6.4255e-03,  ..., 2.4605e-03, 9.5423e-03,
         3.2554e-07],
        [1.9556e-06, 1.7790e-06, 1.3654e-06,  ..., 4.0385e-05, 2.6696e-05,
         4.3209e-02]]), 'X': tensor([[ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 4.0457e+00, -3.7737e+00, -4.7043e-01],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-4.9217e+00,  6.8979e+00, -1.7667e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 6.2038e+00,  2.4020e+00, -3.1692e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0105, 0.0215,  ..., 0.3683, 0.2442, 1.2619],
        [0.0105, 0.0000, 0.0127,  ..., 0.3725, 0.2545, 1.2714],
        [0.0215, 0.0127, 0.0000,  ..., 0.3844, 0.2652, 1.2834],
        ...,
        [0.3683, 0.3725, 0.3844,  ..., 0.0000, 0.3082, 1.0170],
        [0.2442, 0.2545, 0.2652,  ..., 0.3082, 0.0000, 1.0271],
        [1.2619, 1.2714, 1.2834,  ..., 1.0170, 1.0271, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.09737053, 13.13888841, 13.26209545, 13.27994304, 13.32374056,
       13.25017211,  9.6282763 ,  6.42439344,  8.19703512, 12.71843462,
        7.67939546,  5.30158497, 10.51437947,  8.05668609,  5.94909709,
       13.40850328,  5.64500512, 11.9182956 , 12.25806559,  6.00732193,
       11.38774149,  6.18668463, 11.31276109,  9.95552626, 12.33167382,
        5.64464807,  5.51177501,  6.54127431,  9.0368458 , 13.03002338,
        6.67121748, 13.62896593, 10.794755  , 10.06437598,  7.48538905,
       10.52357479,  5.53255882,  7.62232307, 13.20597932, 10.01804223,
       11.83570132,  7.13501264, 11.22142195, 11.87412439,  7.06372172,
        4.7663526 , 13.79676654,  6.46560181,  7.19513806, 12.78327037,
        6.3949043 ,  5.39450305, 13.77222752,  5.10444097, 12.67680439,
       10.80728788,  6.06416957,  8.91551231, 13.45642379, 11.32149259,
        5.39723241, 14.09176259,  8.89458273,  8.62329466, 11.22596994,
       11.18839169, 12.84056604,  5.80312949, 14.05633441,  9.58347456,
        6.55800267,  7.4544866 , 10.88618114,  9.88310614, 13.59673406,
       10.06703358, 13.86480175,  5.38342077,  7.85919687, 13.3426541 ,
       12.19805394,  8.14433542, 13.2778378 , 13.77222752, 10.61472636,
       13.32557133,  6.56833864, 12.70489563, 12.22503754,  5.9510489 ,
       10.63038565,  8.47371868, 12.10942412,  5.80860165,  5.98285319,
       13.95472331,  5.95297459, 12.35421461,  7.02496223,  8.81930525,
        9.75200848,  8.53014915, 13.48809595,  6.65259775, 12.27835996,
        9.78286733])}
{'P': tensor([[3.9510e-03, 2.7631e-01, 2.2354e-01,  ..., 1.7763e-08, 1.2206e-05,
         4.7576e-07],
        [7.0783e-06, 5.9030e-03, 2.0902e-02,  ..., 4.1691e-08, 4.0756e-05,
         5.9883e-07],
        [2.5609e-05, 2.9369e-03, 2.2545e-02,  ..., 1.3123e-07, 1.2750e-04,
         1.9216e-06],
        ...,
        [1.8206e-06, 1.2246e-06, 9.4751e-07,  ..., 3.6244e-01, 6.7214e-05,
         8.0281e-03],
        [2.1229e-06, 2.8408e-06, 2.5705e-06,  ..., 1.9778e-03, 8.2292e-02,
         5.6216e-04],
        [2.1747e-04, 1.3318e-04, 1.0158e-04,  ..., 9.0332e-05, 1.9883e-04,
         9.3902e-02]]), 'X': tensor([[ 11.2571,   6.7072,   8.3693],
        [  9.6662,   9.1725,   8.1056],
        [  9.4925,   9.3811,   8.9842],
        [  9.2979,   9.6048,   7.3138],
        [  9.3560,   9.5391,   9.4943],
        [  8.7131,  10.2216,   7.7915],
        [ -9.2687,  -2.9794,   0.6436],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.7544,   2.6096,   3.3146],
        [  3.6249, -10.7255,   5.6605],
        [  0.9392,  -4.8143,  -3.4649],
        [  3.9913,  -3.8139,  -6.8311],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.6469,   6.3624,  -1.8414],
        [ 10.2470,   8.4054,  -0.1177],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.3122,  -9.6994,  -6.3282],
        [  4.0642,  -3.7599,  -3.4565],
        [  1.7500,   7.4187,   4.4237],
        [  2.2659,  13.7894,   6.5436],
        [  2.2849, -10.9655,  -7.6542],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 11.3118,   6.5957,  -4.5667],
        [  4.2339,  -3.6268,   6.3125],
        [  5.1861,   4.7361,  -3.7312],
        [ 11.4703,  -4.2006,   7.4086],
        [ -3.0004,   7.6607,  -7.2640],
        [ 10.6283,  -5.7253,   7.4878],
        [ -9.3464,   0.5708,  -1.6397],
        [  1.0391,  -4.8141,  -0.0768],
        [  5.8930,  12.3592,  -8.9201],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.5128, -10.9365,   9.9645],
        [  3.2647, -10.8066,  -0.3768],
        [  6.0305,   3.0296,   0.1375],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.1624,  -3.4286,   2.0647],
        [ -3.5029,   7.5138,   8.1853],
        [  3.6817,   6.3354,   6.1233],
        [ 12.5927,   0.4199,  -9.4997],
        [  2.1379,  -4.6757,   6.4981],
        [  1.1350,  14.0104,   0.6321],
        [ -9.1623,  -3.4288,  -4.9961],
        [  4.1999,  -3.6543,  -4.1998],
        [  9.0424,  -7.6779,  -7.5286],
        [ 12.2921,  -1.7902,  -4.0870],
        [  4.8432,   5.1968,   1.5006],
        [ -9.4283,  -1.9585,  -0.9129],
        [  4.5340,  -3.3629,   8.9652],
        [ -8.7544,   2.6096,   3.3146],
        [ -2.5056,   7.7720,   5.5878],
        [ -8.6381,  -4.9238,  -4.4077],
        [  3.9515,  -3.8425,   6.1729],
        [  5.3804,  -2.3504,   7.2744],
        [  4.7544,  -3.1425,  -1.5315],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.3602,  -2.3808,   7.2098],
        [  5.1667,  -2.6543,   1.2879],
        [  7.1697,  11.5343,  -0.7276],
        [  9.3836,   9.5075,  -5.9278],
        [  2.8994,   6.8675,  -9.5987],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.1204,  -5.9308,   4.0761],
        [ -7.0825,   5.2073,  -5.0497],
        [ -7.3446,   4.9113,  -5.9356],
        [ 10.7566,   7.6212,  -2.3279],
        [  6.1372,   2.6726,  -6.7885],
        [  3.1756, -10.8247,   6.8882],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.6258,  -1.9404,  -6.0715],
        [  1.3103,  -4.8032,  -0.9300],
        [ -7.2986,  -7.1225,  -3.0814],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.3354,  -4.7708,  -4.9125],
        [  3.9663,   6.1018,  -5.4104],
        [  0.3493,  -4.7727,  -6.5682],
        [ -3.3999, -10.1152,  -1.2088],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -5.9938,  -8.4745,   3.9735],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.4430,  -0.1897,   4.8383],
        [ -5.5525,   6.5146,  -4.4984],
        [  3.2647, -10.8066,  -0.3768],
        [  3.6703,   6.3443,  -5.4401],
        [  9.5018,  -7.1945,  -1.1892],
        [  3.9663,   6.1018,  -5.4104],
        [  5.2608,  12.6977,   4.9166],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.8111, -10.6787,  -4.6539],
        [  4.6231,   5.4556,   8.4973],
        [  0.9478,  -4.8144,   0.6371],
        [  5.1667,  -2.6543,   1.2879],
        [  6.2895,   0.0405,   5.7652],
        [ -3.2486,   7.5925,  -1.3719],
        [ 12.4691,   2.8525,  -5.5110],
        [  0.9809,   7.6642,  -1.4217],
        [ -5.3328,  -8.9992,   7.3440],
        [  0.7145,  -4.8071,  -2.3766],
        [ -7.9298,   4.1367,  -3.9773],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -5.6423,   6.4537,  -1.2091],
        [  1.3403,  -4.8011,  -1.1698]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0389, 0.0609,  ..., 1.8832, 1.1841, 0.9415],
        [0.0389, 0.0000, 0.0287,  ..., 1.9181, 1.2189, 0.9694],
        [0.0609, 0.0287, 0.0000,  ..., 1.9439, 1.2447, 0.9978],
        ...,
        [1.8832, 1.9181, 1.9439,  ..., 0.0000, 0.6993, 1.0334],
        [1.1841, 1.2189, 1.2447,  ..., 0.6993, 0.0000, 0.4050],
        [0.9415, 0.9694, 0.9978,  ..., 1.0334, 0.4050, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.103709  , 13.32557133, 13.34587027, 13.36800073, 13.3614598 ,
       13.43127131,  9.73579738, 12.4013153 ,  9.13507321, 11.32149259,
        4.90505566,  5.5205189 ,  5.19065439,  7.33350642, 13.25336001,
       13.00413829, 11.57249984,  5.5366698 ,  7.62232307, 13.97430191,
       11.20100795, 12.4980845 , 13.09426142,  5.5748624 ,  7.02326515,
       12.21532282,  8.22728399, 12.07225469,  9.36378438,  4.92497514,
       13.6922425 ,  9.96391045, 11.63765273, 11.22142195, 11.28895489,
        6.74872981,  6.53290663,  9.78285424,  8.29022197,  7.32754882,
       12.59969936,  5.14124612, 14.05633441,  9.78286733,  5.5671389 ,
       11.86239961, 12.4217523 ,  7.10378919,  9.62958973,  5.64500512,
        9.13507321,  8.16584537,  9.94285511,  5.51177501,  5.87132718,
        5.69913817,  9.64157461,  5.86519525,  5.80860165, 13.58101794,
       13.35832725,  7.4544866 , 11.21703654, 10.05558893,  8.7907923 ,
        8.8353674 , 13.18278606,  6.69389431, 11.28093507, 11.86239961,
        5.9510489 ,  4.978704  , 10.1979669 ,  5.18277858,  4.78258092,
        7.27759233,  4.78544875, 10.67132524,  8.50268322, 12.1460931 ,
        9.5014146 , 10.37998078, 13.27717982,  9.44486609,  8.55982105,
       11.28895489,  7.32950385, 11.9182956 ,  7.27759233, 13.744387  ,
        4.76969364, 11.33837396,  7.15100303,  4.90676242,  5.80860165,
        6.28961988,  8.25828409, 12.79126785,  7.72668467, 10.4606265 ,
        4.85995002,  8.94395628,  9.38232121,  9.99145096,  8.5724042 ,
        4.98461983])}
{'P': tensor([[7.2772e-03, 2.7574e-01, 1.5106e-01,  ..., 6.9670e-08, 2.7699e-07,
         2.9091e-06],
        [9.2526e-06, 6.2922e-03, 5.9721e-02,  ..., 4.9467e-08, 1.1554e-07,
         1.1690e-06],
        [2.3530e-06, 3.0114e-05, 4.9087e-03,  ..., 5.0544e-08, 5.3801e-08,
         8.9703e-07],
        ...,
        [2.8024e-05, 7.9765e-06, 5.4335e-06,  ..., 3.1388e-02, 2.1642e-03,
         6.4480e-03],
        [1.1332e-02, 3.2705e-03, 9.7756e-04,  ..., 4.7513e-06, 5.1979e-02,
         4.4091e-04],
        [3.9135e-03, 1.5344e-03, 1.7954e-03,  ..., 2.5541e-04, 1.1996e-03,
         1.4845e-01]]), 'X': tensor([[ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4491e-01, 2.0808e-01,  ..., 1.0489e+00, 4.9556e-01,
         4.1493e-01],
        [1.4491e-01, 0.0000e+00, 3.3974e-01,  ..., 1.1418e+00, 4.5970e-01,
         5.5299e-01],
        [2.0808e-01, 3.3974e-01, 0.0000e+00,  ..., 1.0518e+00, 6.7610e-01,
         3.3236e-01],
        ...,
        [1.0489e+00, 1.1418e+00, 1.0518e+00,  ..., 0.0000e+00, 9.0483e-01,
         7.3401e-01],
        [4.9556e-01, 4.5970e-01, 6.7610e-01,  ..., 9.0483e-01, 0.0000e+00,
         6.5795e-01],
        [4.1493e-01, 5.5299e-01, 3.3236e-01,  ..., 7.3401e-01, 6.5795e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.12218013, 13.3607379 , 13.5247215 , 13.25336001, 13.55868684,
       13.58333857, 10.57302546, 10.69963954,  9.60080639, 12.27991991,
        5.34250225,  8.23525433, 13.12218013, 10.412299  ,  7.96181288,
        9.10945678,  5.19065439, 11.62158554, 11.85890664, 11.00508681,
       11.73397309,  6.38399789,  6.83083863, 12.67680439, 13.48809595,
       10.06407839,  5.70405525,  4.78258092, 13.59673406, 13.66896739,
       10.76154256,  9.58072286, 10.69963954,  7.92607342, 14.07623266,
        6.69389431,  7.53063885,  9.60080639, 13.88683436,  6.25586297,
        9.3349858 , 11.37939695, 12.59840193, 13.91198191, 11.04629471,
        6.56833864, 12.17451685,  6.4020404 ,  9.05843201,  5.91677188,
        4.87083348,  7.49076899, 11.67875855,  9.66977802,  4.76969364,
       12.38881635,  8.19703512,  7.58668144, 11.04629471, 13.62896593,
       13.45642379,  9.15866607,  9.51074763,  9.17016857,  4.91038884,
        5.5671389 , 13.2778378 , 12.62711252,  8.04718433, 11.35345345,
       12.59969936,  5.9510489 ,  7.33350642, 12.97370371,  6.39187285,
        5.43943791,  5.91677188,  6.77777606, 12.49291768, 11.32149259,
       12.85032836,  9.23178204,  9.88310614, 11.00508681,  5.88782913,
        9.58347456,  7.8788965 , 10.51437947, 11.81423161,  9.15866607,
       13.77222752,  6.38105015,  7.63726134,  7.66377554, 11.9120321 ,
        9.0368458 , 10.08001894, 10.42667093,  5.82345867,  7.5403822 ,
        6.81084724, 10.23946247, 14.10796466, 10.95593729, 11.97949273,
        5.70591986])}
{'P': tensor([[2.0528e-03, 3.3002e-01, 1.9234e-01,  ..., 4.8378e-03, 2.0066e-06,
         5.5196e-03],
        [1.0266e-05, 3.2429e-03, 1.4456e-01,  ..., 9.8437e-03, 3.8671e-06,
         3.2657e-03],
        [7.4645e-06, 2.6570e-05, 2.8508e-03,  ..., 6.5140e-03, 4.4507e-06,
         1.3678e-03],
        ...,
        [6.9324e-06, 9.8654e-06, 1.5638e-05,  ..., 2.8464e-02, 1.1646e-04,
         1.7745e-05],
        [2.0709e-06, 1.9814e-06, 1.9899e-06,  ..., 1.4793e-06, 7.5737e-02,
         2.0861e-06],
        [4.6109e-04, 1.4110e-03, 5.8720e-03,  ..., 1.2263e-02, 5.9222e-05,
         3.2319e-02]]), 'X': tensor([[ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4428e-02, 5.1252e-02,  ..., 4.8837e-01, 1.0802e+00,
         2.6999e-01],
        [1.4428e-02, 2.1453e-04, 6.1819e-02,  ..., 4.7602e-01, 1.0902e+00,
         2.8104e-01],
        [5.1252e-02, 6.1819e-02, 2.1453e-04,  ..., 5.3748e-01, 1.0856e+00,
         2.1923e-01],
        ...,
        [4.8837e-01, 4.7602e-01, 5.3748e-01,  ..., 0.0000e+00, 1.2799e+00,
         7.5557e-01],
        [1.0802e+00, 1.0902e+00, 1.0856e+00,  ..., 1.2799e+00, 3.0339e-04,
         1.1021e+00],
        [2.6999e-01, 2.8104e-01, 2.1923e-01,  ..., 7.5557e-01, 1.1021e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.12299145, 13.18278606, 13.2430377 , 13.38091121, 13.32613445,
       13.39372426,  6.33075682, 10.14044489,  8.24808832,  9.96391045,
        8.859416  , 12.26997165,  9.60080639,  8.35856487, 10.35485331,
        9.17016857,  7.24677664,  9.24501829,  5.68942572, 13.89671314,
       10.27716434,  7.72291946, 12.40074901,  6.42689705,  7.96575782,
       12.35625694,  6.46356806, 10.92597196,  8.859416  , 10.794755  ,
        5.5748624 ,  6.2322769 ,  5.91222346, 12.12641969, 10.8207297 ,
        7.6374603 ,  5.39098669, 11.51557166,  8.28968067,  8.92395381,
        5.91677188,  5.54776934, 12.3344536 , 10.14044489,  8.29022197,
       14.0393949 ,  6.54693987, 12.62576026, 13.27717982,  4.75303127,
        5.26342203,  5.21280156,  4.98461983, 12.05395056,  9.53015376,
       10.61417181,  8.36995953, 10.98076873,  7.53884104, 12.18998195,
       13.20597932,  7.13501264, 10.52587243,  6.78325151,  9.51074763,
        8.05668609,  8.91551231,  5.91677188, 13.14690137,  6.81084724,
       10.09979747,  5.04028538,  4.74468406,  6.16512856, 14.09176259,
        4.91336994,  8.85939626, 13.06957425,  7.96181288, 13.38091121,
       13.98691426,  9.77588801,  7.96181288,  9.60412735, 11.51557166,
        8.74470218,  9.79814997,  9.54154096, 10.60226614,  5.68942572,
       11.54672755,  6.54615887,  6.19633139, 13.32557133, 12.62576026,
        6.39187285, 14.02532758,  6.10774544, 13.04217487, 13.83139024,
        8.95133961, 10.61417181,  5.2287058 , 13.69162795,  9.15866607,
       13.26209545])}
{'P': tensor([[1.9934e-03, 2.5439e-01, 1.8164e-01,  ..., 6.6005e-06, 3.2820e-04,
         2.3066e-05],
        [5.9776e-06, 2.9411e-03, 3.1405e-05,  ..., 3.4697e-06, 3.9288e-04,
         1.7816e-05],
        [9.3551e-06, 1.0979e-01, 2.2190e-03,  ..., 3.4857e-06, 3.8918e-04,
         1.5771e-05],
        ...,
        [2.9568e-03, 4.6461e-03, 3.0350e-03,  ..., 1.6981e-02, 9.4731e-05,
         8.0435e-04],
        [2.4420e-04, 4.3700e-04, 3.7450e-04,  ..., 1.8875e-05, 8.8740e-02,
         2.7354e-04],
        [8.2896e-03, 1.0694e-02, 8.5295e-03,  ..., 3.6192e-03, 3.2243e-03,
         8.7774e-02]]), 'X': tensor([[ 11.0864,   7.0387,  -0.1238],
        [ 10.2470,   8.4054,  -0.1177],
        [ 10.5678,   7.9261,  -0.6100],
        [  9.6587,   9.1817,  -0.8212],
        [  9.7328,   9.0902,  -0.8917],
        [  9.3624,   9.5318,   0.2779],
        [  5.8275,  -1.5334,   2.0124],
        [ 11.1469,   6.9237,   1.6160],
        [  6.3266,   0.3292,   9.3502],
        [  5.8201,   3.5841,  -8.0609],
        [  4.3377,   5.7570,   1.3712],
        [ -2.4956, -10.4682,   3.5632],
        [  4.8674,  12.8875,   6.7622],
        [  8.4691,  10.4571,  -5.0650],
        [ -9.4756,  -0.8166,  -9.6021],
        [  0.7958,  -4.8109,  -8.2248],
        [  5.5101,  -2.1433,   9.5442],
        [  7.0171,  11.6439,  -5.0178],
        [  6.1685,  12.1979,  -6.2473],
        [ -6.2668,   5.9808,   2.5614],
        [  4.6992,   5.3690,   3.0935],
        [ -7.0253,  -7.4493,  -1.5978],
        [  4.3554,   5.7393,  -9.0766],
        [  0.9788,  -4.8145,  -3.8661],
        [  0.7275,  14.0667,  -0.9815],
        [  4.5325,  -3.3643,   6.7699],
        [  6.3542,   0.6934,  -3.3840],
        [  5.6260,  -1.9401,   3.0259],
        [  5.4179,  -2.2925,  -3.4641],
        [ -3.3812, -10.1235,   7.4384],
        [ -9.1749,  -3.3803,   6.4629],
        [  6.3587,   1.1596,  -3.6629],
        [ -7.5720,   4.6310,   2.1717],
        [  4.5057,  -3.3894,  -0.8160],
        [  4.4915,   5.5990,   7.0405],
        [ -7.0825,   5.2073,  -5.0497],
        [  0.2435,   7.8191,  -6.4709],
        [  0.3231,  -4.7691,  -1.6120],
        [ 11.3691,   6.4760,   1.0787],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.8889,  -1.3916,   8.8993],
        [  9.3624,   9.5318,   0.2779],
        [ -8.9606,  -4.0994,   1.7886],
        [  5.6022,  -1.9833,   8.3081],
        [  9.5203,   9.3483,   5.8781],
        [ -7.9925,  -6.1423,   2.8990],
        [ -5.0132,  -9.2234,  -8.3531],
        [  5.2608,  -2.5256,  -9.1446],
        [  2.3376,  -4.6231,   9.8436],
        [ -8.5149,  -5.1905,   1.5239],
        [  1.3339,   7.5627,  -0.9034],
        [  0.1497,   7.8334,  -5.4412],
        [ -8.1204,  -5.9308,   4.0761],
        [  0.8700,  -4.8132,  -7.8042],
        [ -9.3810,  -2.3455,  -9.5194],
        [  3.2990,  -4.2415,  -7.4857],
        [  5.6093,  -1.9705,   6.4807],
        [  0.1206,  -4.7363,   8.7103],
        [ 12.0042,  -2.8350,   2.6864],
        [  2.7420,  -4.4896,  -3.7139],
        [  8.5716,  10.3596,   6.8818],
        [  1.1994,   7.6035,   2.4295],
        [  8.3791,  10.5409,   5.5590],
        [  6.3119,   1.7797,   9.9186],
        [  7.0664,  -9.2590,   8.0879],
        [  5.1196,  -2.7160,   4.2705],
        [  2.5469,  13.7192,  -9.6935],
        [ -8.0750,  -6.0072,  -5.8267],
        [  3.3167,   6.6022,   6.3554],
        [ -9.4430,  -0.1897,   4.8383],
        [ 11.6165,  -3.8715,   6.4594],
        [ -7.9298,   4.1367,  -3.9773],
        [  2.6310, -10.9195,   3.8054],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.9178,   2.1936,   7.5388],
        [ 10.1503,   8.5414,  -7.2681],
        [  2.5469,  13.7192,  -9.6935],
        [  6.8093,  -9.4178,   8.4244],
        [  4.5340,  -3.3629,   8.9652],
        [  2.5128, -10.9365,   9.9645],
        [  4.0421,  -3.7765,  -8.3631],
        [ -7.6719,  -6.6260,   7.1468],
        [  7.0171,  11.6439,  -5.0178],
        [ -6.7643,  -7.7372,  -9.6663],
        [  4.1297,  -3.7098,  -3.9838],
        [  4.7813,  -3.1139,   8.2691],
        [  5.9548,  -1.2273,  -2.6520],
        [ -4.2040,   7.2486,   2.9875],
        [ -8.3900,   3.3727,   3.1650],
        [  6.3048,   0.1469,  -9.1691],
        [  6.3301,   0.3638,  -7.5685],
        [  9.6662,   9.1725,   8.1056],
        [ -8.8745,   2.3097,   3.5600],
        [ -9.4196,   0.0478,   9.4140],
        [  4.2041,  -3.6509,  -4.9488],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.4986,   7.5152,   9.0712],
        [  4.7482, -10.3907,   5.9149],
        [  9.5203,   9.3483,   5.8781],
        [  4.5340,  -3.3629,   8.9652],
        [ -0.0410,   7.8591,   9.3457],
        [  6.3587,   1.1596,  -3.6629],
        [  9.3836,   9.5075,  -5.9278],
        [ 12.2435,   4.0076,   8.1257],
        [  3.6703,   6.3443,  -5.4401],
        [  6.2555,   2.1474,   4.1685]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0228, 0.0465,  ..., 0.7256, 0.4593, 0.3972],
        [0.0228, 0.0000, 0.0476,  ..., 0.7210, 0.4785, 0.4063],
        [0.0465, 0.0476, 0.0000,  ..., 0.7684, 0.4443, 0.4430],
        ...,
        [0.7256, 0.7210, 0.7684,  ..., 0.0000, 1.1006, 0.4562],
        [0.4593, 0.4785, 0.4443,  ..., 1.1006, 0.0000, 0.6621],
        [0.3972, 0.4063, 0.4430,  ..., 0.4562, 0.6621, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.13205357, 13.25336001, 13.20988996, 13.32645552, 13.31764181,
       13.3607379 ,  6.02588517, 13.12218013,  6.33517349,  6.83515201,
        7.20826219, 10.76154256, 13.77604293, 13.45642379,  9.51074763,
        4.8763135 ,  5.91222346, 13.59480906, 13.66896739,  8.66273171,
        7.13501264, 10.23946247,  7.20482224,  4.91296205, 14.08549802,
        5.64464807,  6.39187285,  5.95111804,  5.8828986 , 10.67322527,
        9.77777346,  6.46356806,  8.87587457,  5.63824971,  7.17789666,
        8.7907923 ,  7.822854  ,  4.78002665, 13.08414689,  8.47371868,
        6.05113217, 13.3607379 ,  9.85384623,  5.94293185, 13.3426541 ,
       10.08001894, 10.4977132 ,  5.83560009,  5.1805345 ,  9.97220799,
        7.67939546,  7.83486943, 10.05558893,  4.89121805,  9.66977802,
        5.37343173,  5.94535529,  4.73784222, 12.3344536 ,  5.26067019,
       13.44593519,  7.69752262, 13.46553623,  6.55800267, 11.64746882,
        5.79543038, 13.95361098, 10.06437598,  7.38844703,  9.44486609,
       12.24467041,  8.94395628, 11.23201619,  6.49538841,  9.18358562,
       13.26590071, 13.95361098, 11.62158554,  5.64500512, 11.22142195,
        5.53175275, 10.1371552 , 13.59480906, 10.27716434,  5.5513072 ,
        5.70591986,  6.07992623,  8.37954853,  9.04255655,  6.30647746,
        6.34059021, 13.32557133,  9.17016857,  9.41970031,  5.56809335,
       10.48721382,  8.28968067, 11.42420508, 13.3426541 ,  5.64500512,
        7.85919687,  6.46356806, 13.35832725, 12.88270018,  7.32950385,
        6.61386436])}
{'P': tensor([[1.3985e-03, 1.6979e-01, 1.2376e-01,  ..., 2.7290e-05, 3.2024e-07,
         9.6072e-08],
        [3.3983e-06, 1.6412e-03, 3.0433e-01,  ..., 4.7929e-06, 5.3727e-07,
         8.3480e-08],
        [2.6795e-06, 5.0476e-06, 1.8696e-03,  ..., 2.8905e-06, 7.4792e-07,
         8.5285e-08],
        ...,
        [8.0991e-03, 2.7076e-02, 2.9771e-02,  ..., 3.2928e-03, 3.8749e-07,
         1.1678e-07],
        [5.6487e-07, 4.6537e-07, 4.3185e-07,  ..., 5.2892e-07, 1.9619e-02,
         2.3494e-05],
        [9.8628e-05, 7.6020e-05, 6.9361e-05,  ..., 3.1658e-05, 1.4595e-05,
         6.8169e-02]]), 'X': tensor([[ 11.0438,   7.1180,  -5.6991],
        [ 10.1798,   8.5002,  -5.8893],
        [  9.6816,   9.1536,  -5.8543],
        [ 10.0400,   8.6922,  -6.2847],
        [  9.3836,   9.5075,  -5.9278],
        [ 10.2712,   8.3707,  -4.9743],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.4671,  -0.5553,   0.1117],
        [ -8.8154,   2.4611,   2.5198],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -1.0012, -10.8576,   2.2867],
        [ 10.7566,   7.6212,  -2.3279],
        [  3.1892, -10.8220,  -6.8409],
        [  5.8889,  -1.3916,   8.8993],
        [ -9.4739,  -1.2497,   0.5483],
        [  3.3069,   6.6089,   8.7377],
        [ 12.4690,  -0.8528,  -9.3374],
        [  3.1271, -10.8343,   2.6442],
        [  0.5960, -11.0335,   3.1144],
        [  6.3551,   0.7149,   1.3344],
        [ 12.2218,   4.0953,  -8.2072],
        [  0.2480, -11.0153,  -3.0183],
        [  4.1637,  13.1898,  -1.9189],
        [  7.5930,  -8.9037,   7.2323],
        [ -5.2752,  -9.0409,  -8.2568],
        [ -7.3554,   4.8985,  -0.6665],
        [  9.7328,   9.0902,  -0.8917],
        [  5.6022,  -1.9833,   8.3081],
        [  9.4516,  -7.2501,  -1.8868],
        [  5.4867,  12.5816,   2.9549],
        [ 11.3118,   6.5957,  -4.5667],
        [  5.5424,  -2.0885,   7.2487],
        [  6.7565,  11.8237,  -4.9882],
        [  5.8450,  -9.9356,   3.0767],
        [  1.9241,   7.3503,   7.9004],
        [  6.0345,   3.0175,  -1.4424],
        [ -8.9282,   2.1648,   8.4894],
        [  4.1348,   5.9514,  -0.8856],
        [  8.6271,  -8.0715,  -7.4813],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -8.3653,   3.4184,   9.3454],
        [ 11.0373,   7.1299,  -8.5895],
        [ -9.4731,  -0.7120,  -6.0271],
        [  1.0093,   7.6567,   2.3526],
        [  0.3493,  -4.7727,  -6.5682],
        [  5.9193,  -1.3176,   5.1739],
        [ -0.5709,   7.9055,   7.6089],
        [  0.4996,   7.7738,  -4.4711],
        [  6.2804,   2.0019,   0.8607],
        [  6.3413,   0.4883,  -6.8416],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.0305,   3.0296,   0.1375],
        [ -7.4733,   4.7555,   6.2490],
        [  3.1271, -10.8343,   2.6442],
        [  3.9085,   6.1512,  -2.2560],
        [  7.5784,  -8.9141,   5.8410],
        [  2.7420,  -4.4896,  -3.7139],
        [ 10.2712,   8.3707,  -4.9743],
        [ 11.2939,   6.6325,  -5.8311],
        [ 12.0772,  -2.6014,  -7.1832],
        [  7.5064,  -8.9651,  -0.0994],
        [  3.1480,  -4.3165,  -7.8002],
        [ 12.5154,   2.5072,  -9.3274],
        [  6.2242,  -0.3171,   7.4978],
        [  3.3744,   6.5623,  -6.3636],
        [  6.1989,  -9.7591,   0.8289],
        [  5.8902,  -1.3887,   4.0388],
        [ 10.5678,   7.9261,  -0.6100],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.4645,  -1.4880,  -2.0226],
        [  2.1120,  -4.6818,   8.6144],
        [  3.9770, -10.6341,   8.0783],
        [  4.1140,  -3.7220,  -0.1273],
        [ -8.3535,  -5.5118,   9.8399],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.2022,   2.4092,   3.0349],
        [ 10.0400,   8.6922,  -6.2847],
        [ -1.6220,   7.8917,   0.0315],
        [  3.7992,  -3.9474,  -8.9019],
        [  8.0877,  -8.5297,  -1.1050],
        [  4.1637,  13.1898,  -1.9189],
        [ 11.7162,   5.6728,  -8.9372],
        [  5.2939,  -2.4784,  -4.5840],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.3539,  -5.5111,   0.3373],
        [ 12.4592,  -0.9168,  -6.7571],
        [  3.1271, -10.8343,   2.6442],
        [ -6.5295,  -7.9785,  -7.4429],
        [  6.1685,  12.1979,  -6.2473],
        [  2.2849, -10.9655,  -7.6542],
        [ 11.5198,  -4.0923,   5.5088],
        [ 11.7796,  -3.4696,   9.0088],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.0078,  -3.8018,  -6.2168],
        [ -7.4808,   4.7462,   2.0266],
        [ -4.9217,   6.8979,  -1.7667],
        [  9.3836,   9.5075,  -5.9278],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -9.3444,   0.5821,   6.9101],
        [  6.0031,  12.2958,  -6.5992],
        [ -0.8885,   7.9160,   0.0571],
        [ 11.7162,   5.6728,  -8.9372],
        [ -9.4766,  -1.1116,  -2.1706],
        [  4.5340,  -3.3629,   8.9652]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0127, 0.0192,  ..., 0.1315, 1.2297, 1.0424],
        [0.0127, 0.0000, 0.0069,  ..., 0.1262, 1.2422, 1.0502],
        [0.0192, 0.0069, 0.0000,  ..., 0.1266, 1.2489, 1.0518],
        ...,
        [0.1315, 0.1262, 0.1266,  ..., 0.0000, 1.2753, 1.1721],
        [1.2297, 1.2422, 1.2489,  ..., 1.2753, 0.0000, 1.1904],
        [1.0424, 1.0502, 1.0518,  ..., 1.1721, 1.1904, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.13888841, 13.26209545, 13.32374056, 13.27994304, 13.35832725,
       13.25017211, 12.35421461,  9.48337063,  9.15253344,  9.75075273,
       10.9036226 , 13.18278606, 11.28215429,  6.05113217,  9.55593374,
        7.39004226, 12.4980845 , 11.27656488, 11.04953767,  6.39520397,
       12.889695  , 11.018084  , 13.83139024, 11.70168755, 10.46739073,
        8.83725081, 13.31764181,  5.94293185, 11.9120321 , 13.72594022,
       13.09426142,  5.92281142, 13.61801526, 11.52733581,  7.59795445,
        6.74686071,  9.18690018,  7.24677664, 11.81423161,  9.51074763,
        9.0368458 , 13.1399181 ,  9.4997987 ,  7.72291946,  4.78544875,
        6.06416957,  7.92607342,  7.78980299,  6.59176018,  6.36004393,
       13.15525803,  6.74872981,  8.8580671 , 11.27656488,  7.28793663,
       11.70016217,  5.26067019, 13.25017211, 13.09737053, 12.35421461,
       11.69264243,  5.34250225, 12.76408562,  6.2322769 ,  7.37902766,
       11.56146873,  6.05165248, 13.20988996,  8.95133961,  8.92098752,
        9.58072286,  5.13616224, 11.35345345,  5.54776934, 10.00802627,
        5.56809335,  6.65368551, 13.27994304,  8.05668609,  5.47866221,
       11.75438006, 13.83139024, 13.01728313,  5.84533345, 10.23946247,
        8.40445872, 10.00794468, 12.49291768, 11.27656488, 10.30972193,
       13.66896739, 11.20100795, 12.22503754, 12.27993098,  9.97964013,
        5.52416368,  8.859416  ,  8.47371868, 13.35832725, 10.06437598,
        9.36256096, 13.6829843 ,  7.96575782, 13.01728313,  9.54154096,
        5.64500512])}
{'P': tensor([[4.1813e-03, 3.2075e-01, 2.1691e-01,  ..., 3.0028e-08, 2.1845e-04,
         2.8483e-06],
        [1.2762e-05, 5.0359e-03, 5.2146e-01,  ..., 3.3435e-08, 1.3740e-04,
         1.3830e-06],
        [6.0268e-06, 1.3394e-05, 6.9714e-03,  ..., 5.3083e-08, 1.4140e-04,
         1.1117e-06],
        ...,
        [3.8196e-06, 1.9450e-06, 1.2457e-06,  ..., 1.8797e-02, 1.4174e-06,
         6.2835e-06],
        [1.2624e-05, 2.6108e-05, 7.6645e-05,  ..., 8.0418e-07, 4.2921e-02,
         1.6676e-05],
        [1.2552e-04, 1.5352e-04, 2.8264e-04,  ..., 1.2443e-06, 2.5174e-01,
         1.3527e-01]]), 'X': tensor([[ 11.0373,   7.1299,  -8.5895],
        [ 10.0619,   8.6627,  -9.2884],
        [  8.8474,  10.0868,  -8.8644],
        [  8.5198,  10.4091,  -8.6486],
        [ 10.1503,   8.5414,  -7.2681],
        [  8.9284,  10.0036,  -6.9210],
        [  2.5010,  -4.5736,  -0.8989],
        [  0.4119,  14.1019,  -4.1870],
        [  6.2026,  12.1774,  -1.5887],
        [  4.5954,  13.0099,  -4.4439],
        [ -5.2589,   6.7024,  -5.5809],
        [  2.0243,   7.3086,   7.5432],
        [ -4.8665,  -9.3204,  -5.2564],
        [  2.7121,   6.9743,  -8.4202],
        [  2.3889,  -4.6082,  -8.7291],
        [  9.4516,  -7.2501,  -1.8868],
        [  1.1274,  14.0116,   8.1753],
        [ -9.2974,   0.8371,   2.4852],
        [ -5.3328,  -8.9992,   7.3440],
        [ -8.8934,   2.2597,   3.5831],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.3122,  -9.6994,  -6.3282],
        [  5.1519, -10.2387,   9.8191],
        [ -6.4002,  -8.1048,  -0.9108],
        [  6.0047,   3.1062,  -6.7914],
        [ -7.6734,   4.4980,  -5.4248],
        [ 11.3118,   6.5957,  -4.5667],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.8979,   3.3958,  -3.8713],
        [  6.3460,   1.4252,   1.2291],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.1737,   2.5305,   3.9049],
        [ -8.6112,   2.9319,  -7.9943],
        [  4.3459,  -3.5327,   0.3667],
        [  0.9392,  -4.8143,  -3.4649],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.7558,  -4.4844,   0.0636],
        [  2.4932,   7.0903,  -5.4214],
        [ -8.7735,   2.5640,  -6.3313],
        [  6.9284,  11.7061,  -0.3732],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.2630,   2.1057,  -5.2136],
        [  9.4516,  -7.2501,  -1.8868],
        [  6.9284,  11.7061,  -0.3732],
        [  6.8704,  -9.3809,   7.5769],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -3.4986,   7.5152,   9.0712],
        [ -1.2454,   7.9127,  -0.5715],
        [  0.9036,   7.6840,   9.9237],
        [ 10.6673,  -5.6649,  -4.1646],
        [  4.7739,  -3.1218,  -4.1188],
        [  5.8902,  -1.3887,   4.0388],
        [ 11.4235,   6.3594,   5.1663],
        [ 11.6019,   5.9540,  -5.1715],
        [  0.5600, -11.0321,  -0.8660],
        [  5.1861,   4.7361,  -3.7312],
        [ -9.1433,  -3.5003,  -3.0209],
        [  0.2435,   7.8191,  -6.4709],
        [ -7.5401,   4.6718,   9.0033],
        [ -1.3940, -10.7772,   6.7835],
        [  6.3119,   1.7797,   9.9186],
        [  6.3211,   1.7017,  -0.2164],
        [  5.4426, -10.1181,  -5.1268],
        [  5.1886,  -2.6250,   6.4677],
        [ -1.0012, -10.8576,   2.2867],
        [  3.8182,  -3.9347,  -1.7961],
        [  5.3804,  -2.3504,   7.2744],
        [ -3.5173,   7.5091,  -8.9585],
        [ 11.4495,   6.3026,   6.8257],
        [  2.2437,  -4.6489,   6.1430],
        [ 12.4592,  -0.9168,  -6.7571],
        [  5.9026,   3.3837,  -0.6573],
        [  1.7659,   7.4127,   4.7871],
        [  0.8940, -11.0402,  -0.9783],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.0937,  13.8293,  -2.3533],
        [  0.5175,  14.0910,   4.6540],
        [  1.3339,   7.5627,  -0.9034],
        [  2.1440, -10.9811,   8.5281],
        [  6.8704,  -9.3809,   7.5769],
        [  0.3724,   7.7974,  -6.8722],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.3444,   0.5821,   6.9101],
        [  2.8474,   6.8979,   6.6457],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -7.8280,   4.2848,  -7.1955],
        [  0.1041, -11.0045,   2.4311],
        [ -2.0445,   7.8469,  -0.5857],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.3604,   0.4855,  -3.0158],
        [  9.3624,   9.5318,   0.2779],
        [ -0.5709,   7.9055,   7.6089],
        [  6.2804,   2.0019,   0.8607],
        [  2.0039,   7.3173,  -1.5885],
        [ 11.7794,   5.5086,  -9.6929],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.6672,  -1.8631,   8.1020],
        [  4.2275,  -3.6320,  -0.2714],
        [ -4.3359,  -9.6431,  -9.0616],
        [  5.7784,  -1.6401,   9.5587],
        [  4.0421,  -3.7765,  -8.3631],
        [ -2.6485, -10.4151,   0.8766],
        [  6.8407,  11.7666,   3.0045],
        [ 11.2571,   6.7072,   8.3693]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0318, 0.0234,  ..., 1.2959, 0.9957, 1.3491],
        [0.0318, 0.0000, 0.0084,  ..., 1.3243, 1.0222, 1.3755],
        [0.0234, 0.0084, 0.0000,  ..., 1.3170, 1.0149, 1.3682],
        ...,
        [1.2959, 1.3243, 1.3170,  ..., 0.0000, 1.1894, 1.3399],
        [0.9957, 1.0222, 1.0149,  ..., 1.1894, 0.0000, 0.3533],
        [1.3491, 1.3755, 1.3682,  ..., 1.3399, 0.3533, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.1399181 , 13.27717982, 13.41713661, 13.45124917, 13.26590071,
       13.40850328,  5.21280156, 14.10796466, 13.6660671 , 13.79762415,
        8.51928531,  7.58378655, 10.51437947,  7.48311002,  5.19065439,
       11.9120321 , 14.05687815,  9.3349858 , 10.4606265 ,  9.17595899,
        8.41171369, 11.57249984, 11.46176259, 10.32715715,  6.76056625,
        8.89458273, 13.09426142,  9.69629   ,  6.80559669,  6.50409372,
       13.00413829,  6.23262289,  6.67218687,  9.09661212,  5.60064319,
        4.90505566, 10.412299  ,  5.26342203,  7.5158489 ,  9.14044984,
       13.60275211, 11.1341798 ,  9.62958973,  6.60752373, 11.9120321 ,
       13.60275211, 11.62770332,  9.54154096,  8.28968067,  8.01009083,
        7.73692436, 12.07818871,  5.70405525,  6.05165248, 13.0743399 ,
       13.04051744, 11.04629471,  7.02326515,  9.79039371,  7.822854  ,
        8.87007495, 10.86693708,  6.55800267,  6.54615887, 11.4890861 ,
        5.81481044, 10.9036226 ,  5.48276038,  5.87132718,  8.29203356,
       13.06957425,  5.16205087, 12.49291768,  6.80370254,  7.62011734,
       11.07637531,  8.83341608, 13.98691426, 14.10045797,  7.67939546,
       11.18839169, 11.62770332,  7.80625981,  8.09899874,  9.36256096,
        7.46248947,  9.85330646,  8.92395381, 11.0050379 ,  8.10886021,
       13.32645552,  9.37296102, 13.3607379 ,  7.92607342,  6.59176018,
        7.58668144, 13.00380624,  6.07992623,  5.96556248,  5.57341623,
       10.57302546,  6.00663314,  5.53175275, 10.74655346, 13.6105595 ,
       13.103709  ])}
{'P': tensor([[1.7906e-03, 1.5763e-01, 1.1993e-01,  ..., 1.9491e-06, 1.4798e-07,
         7.3458e-07],
        [6.3638e-06, 4.1810e-03, 7.2451e-02,  ..., 4.3776e-06, 5.2947e-07,
         3.1452e-06],
        [9.3408e-06, 1.9392e-04, 7.6283e-03,  ..., 1.1384e-05, 1.2189e-06,
         6.4579e-06],
        ...,
        [1.4148e-06, 1.9995e-06, 2.8309e-06,  ..., 4.3477e-02, 1.5322e-04,
         1.3084e-05],
        [2.8527e-06, 1.9847e-06, 2.0310e-06,  ..., 1.2743e-05, 1.1527e-01,
         6.3840e-04],
        [1.0990e-05, 9.1144e-06, 8.6221e-06,  ..., 7.3010e-06, 2.7979e-02,
         3.8095e-01]]), 'X': tensor([[ 10.9931,   7.2107,   7.7933],
        [  9.6662,   9.1725,   8.1056],
        [  9.2979,   9.6048,   7.3138],
        [  9.2692,   9.6370,   7.0746],
        [  8.7131,  10.2216,   7.7915],
        [  9.4925,   9.3811,   8.9842],
        [ -8.1204,  -5.9308,   4.0761],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.4932,   7.0903,  -5.4214],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.9655,   3.2172,  -7.5953],
        [ 11.2517,  -4.6495,   2.9800],
        [ 11.5456,   6.0861,   8.1101],
        [  4.1537,   5.9340,  -6.0423],
        [ 11.7795,  -3.4697,   3.5624],
        [  5.7845,   3.6652,   0.9175],
        [  2.7420,  -4.4896,  -3.7139],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.5974,   7.0363,  -7.7810],
        [ -3.6381,   7.4683,   7.3544],
        [ 10.1503,   8.5414,  -7.2681],
        [  1.7500,   7.4187,   4.4237],
        [ -2.4956, -10.4682,   3.5632],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.1755,   2.5233,  -2.0042],
        [  5.2199,   4.6861,  -1.9304],
        [  5.7784,  -1.6401,   9.5587],
        [  2.2076,  13.8032,  -1.5099],
        [  7.5930,  -8.9037,   7.2323],
        [  9.4554,   9.4245,   9.8800],
        [  2.7558,  -4.4844,   0.0636],
        [ -5.2589,   6.7024,  -5.5809],
        [  9.5203,   9.3483,   5.8781],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.4430,  -0.1897,   4.8383],
        [  5.1519, -10.2387,   9.8191],
        [ 10.0995,  -6.4739,   2.0974],
        [  4.6992,   5.3690,   3.0935],
        [ 11.7162,   5.6728,  -8.9372],
        [ -3.5173,   7.5091,  -8.9585],
        [ 11.9122,   5.1397,  -1.5498],
        [ 12.5996,   1.3971,  -1.0694],
        [ -3.1671,   7.6158,   6.0765],
        [  9.5261,  -7.1675,   1.4206],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.1752,   2.5242,   3.5874],
        [  6.3587,   1.1596,  -3.6629],
        [  6.9956,  11.6590,  -7.6876],
        [ 12.5650,   2.0132,   4.6272],
        [ -8.9282,   2.1648,   8.4894],
        [ -7.7239,   4.4298,   3.1216],
        [ -6.7643,  -7.7372,  -9.6663],
        [  7.0642,  11.6104,  -9.8316],
        [  9.8082,  -6.8396,   4.9615],
        [  3.9895,   6.0816,  -6.5235],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.1771, -10.9776,  -3.2007],
        [ -8.1905,   3.7253,   4.1175],
        [ 10.9931,   7.2107,   7.7933],
        [  5.3602,  -2.3808,   7.2098],
        [ -6.3489,   5.9117,   6.5161],
        [ -6.1899,  -8.3011,  -7.9269],
        [  4.9337,  12.8566,  -3.6582],
        [ -9.1178,   1.5784,   2.1410],
        [  9.6662,   9.1725,   8.1056],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.9693,  -9.3200,  -8.3822],
        [  1.9312,  13.8649,   8.3563],
        [  5.8930,  12.3592,  -8.9201],
        [  7.7794,  11.0629,   8.9069],
        [ -4.4779,  -9.5609,  -2.4505],
        [  3.7916,  -3.9524,  -7.9547],
        [  3.3167,   6.6022,   6.3554],
        [  8.9284,  10.0036,  -6.9210],
        [  4.5954,  13.0099,  -4.4439],
        [  4.9337,  12.8566,  -3.6582],
        [  6.0031,  12.2958,  -6.5992],
        [  3.2226,  -4.2802,  -4.0428],
        [ -1.9646,   7.8571,   3.3817],
        [  0.8940, -11.0402,  -0.9783],
        [  6.2038,   2.4020,  -3.1692],
        [ -9.2345,  -3.1345,   3.9715],
        [ 12.4837,   2.7510,   0.4449],
        [ -6.4368,   5.8356,   3.3014],
        [ -2.1817,   7.8274,  -4.6298],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.6801,   2.7809,  -0.8443],
        [  2.5382,   7.0672,   4.5097],
        [  7.0664,  -9.2590,   8.0879],
        [ -0.1626, -10.9796,   6.4629],
        [  6.3542,   0.6934,  -3.3840],
        [  4.3523, -10.5232,   8.2138],
        [ 11.6165,  -3.8715,   6.4594],
        [  2.0729,   7.2878,   7.0880],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3413,   0.4883,  -6.8416],
        [  4.0362,  13.2397,  -7.0030],
        [  9.6662,   9.1725,   8.1056],
        [ -5.9940,   6.1985,   5.9566],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -3.5903,   7.4847,  -8.9112],
        [ 10.6673,  -5.6649,  -4.1646],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.2687,  -2.9794,   0.6436],
        [ -9.4196,   0.0478,   9.4140]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0417, 0.0265,  ..., 1.3585, 1.4336, 1.1886],
        [0.0417, 0.0000, 0.0323,  ..., 1.3964, 1.4747, 1.2245],
        [0.0265, 0.0323, 0.0000,  ..., 1.3652, 1.4539, 1.2136],
        ...,
        [1.3585, 1.3964, 1.3652,  ..., 0.0000, 0.8497, 1.2107],
        [1.4336, 1.4747, 1.4539,  ..., 0.8497, 0.0000, 0.5348],
        [1.1886, 1.2245, 1.2136,  ..., 1.2107, 0.5348, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.14690137, 13.32557133, 13.36800073, 13.37121923, 13.43127131,
       13.34587027, 10.05558893,  7.98469166,  7.5158489 ,  8.22728399,
        6.77777606, 12.17451685, 13.05148826,  7.24326992, 12.27991991,
        6.84796202,  5.26067019, 12.35625694,  7.50035355,  8.30729352,
       13.26590071,  7.62232307, 10.76154256,  5.5205189 ,  6.67108662,
        7.01475376,  6.00663314, 13.97857591, 11.70168755, 13.35013374,
        5.26342203,  8.51928531, 13.3426541 , 11.70016217,  9.44486609,
       11.46176259, 11.99633819,  7.13501264, 13.01728313,  8.29203356,
       12.97370371, 12.67680439,  8.24808832, 11.92133129, 10.67132524,
        6.67121748,  6.46356806, 13.59673406, 12.72524615,  9.18690018,
        8.90405823, 10.27716434, 13.59056613, 11.95743558,  7.27340834,
        8.30125201, 11.1913598 ,  8.99791316, 13.14690137,  5.86519525,
        8.67502348, 10.35485331, 13.77075174,  9.25336371, 13.32557133,
        9.14973319, 11.63765273, 13.99877413, 13.6922425 , 13.52430176,
       10.5575601 ,  5.47702766,  7.38844703, 13.40850328, 13.79762415,
       13.77075174, 13.6829843 ,  5.35774573,  8.09899874, 11.07637531,
        6.65259775,  9.75200848, 12.78327037,  8.68832398,  8.12580976,
       10.1979669 ,  9.11473391,  7.50917551, 11.64746882, 10.98076873,
        6.39187285, 11.38774149, 12.24467041,  7.57685908,  7.96181288,
        6.36004393, 13.84126197, 13.32557133,  8.62260327, 10.8992524 ,
       10.06437598,  8.30125201, 12.07818871,  7.83534987,  9.73579738,
        9.41970031])}
{'P': tensor([[2.0766e-03, 2.5429e-01, 2.0237e-01,  ..., 2.7081e-05, 3.5186e-08,
         2.2539e-04],
        [9.5174e-06, 2.4532e-03, 3.0943e-01,  ..., 3.8590e-05, 4.3076e-08,
         3.3049e-04],
        [1.1474e-05, 1.7379e-05, 7.4875e-03,  ..., 1.8678e-04, 1.5725e-07,
         1.5079e-03],
        ...,
        [5.3360e-06, 5.5990e-06, 6.6114e-06,  ..., 1.0542e-01, 1.8437e-03,
         1.2159e-03],
        [1.4423e-06, 1.1182e-06, 8.1772e-07,  ..., 2.6549e-06, 1.6832e-02,
         2.0380e-06],
        [2.9531e-05, 3.3677e-05, 4.3856e-05,  ..., 2.0948e-01, 4.7639e-04,
         1.8178e-01]]), 'X': tensor([[ 10.9916,   7.2133,  -0.9179],
        [ 10.5678,   7.9261,  -0.6100],
        [  9.7328,   9.0902,  -0.8917],
        [  9.6587,   9.1817,  -0.8212],
        [  9.6614,   9.1784,  -1.2087],
        [ 10.2470,   8.4054,  -0.1177],
        [  6.1755,   2.5233,  -2.0042],
        [ 11.9019,  -3.1379,  -3.4534],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.7796,  -3.4696,   9.0088],
        [ 12.5832,   1.7543,  -7.6134],
        [  5.0277,   4.9589,   7.0175],
        [  0.1047, -11.0046,   0.5188],
        [  8.0877,  -8.5297,  -1.1050],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.8323,  -1.5228,   5.1132],
        [  2.3367,   7.1676,  -9.5263],
        [  4.0421,  -3.7765,  -8.3631],
        [  0.2435,   7.8191,  -6.4709],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -9.1960,  -3.2964,   1.6921],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.4336,  -2.2676,   5.6933],
        [  3.7916,  -3.9524,  -7.9547],
        [ -9.2373,  -3.1225,  -8.9763],
        [  5.9883,   3.1536,   6.2320],
        [ 10.3251,   8.2928,  -3.0006],
        [  2.6078, -10.9229,   6.2869],
        [ -0.4341, -10.9473,   7.4509],
        [ 12.2592,   3.9425,   2.0383],
        [  2.3889,  -4.6082,  -8.7291],
        [  5.9193,  -1.3176,   5.1739],
        [ 11.6801,  -3.7196,  -0.9181],
        [  4.7739,  -3.1218,  -4.1188],
        [ -8.9033,   2.2329,  -8.8912],
        [  2.9470,  13.6085,   2.9809],
        [ -3.3116, -10.1539,   7.5919],
        [  5.6260,  -1.9401,   3.0259],
        [ -9.2508,   1.0572,   2.9978],
        [ -8.7078,   2.7182,   1.8146],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.1813,  -3.3550,   1.8827],
        [ -9.3464,   0.5708,  -1.6397],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 10.6673,  -5.6649,  -4.1646],
        [  1.1139,   7.6281,  -2.5680],
        [  6.0073,   3.0986,   6.7937],
        [  4.7482, -10.3907,   5.9149],
        [ -2.3317,   7.8034,  -3.6274],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.5294,  -2.1106,   9.0626],
        [ 11.5568,   6.0602,   9.3244],
        [ -1.2454,   7.9127,  -0.5715],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.0547,   7.2957,  -1.9343],
        [  0.9659,  -4.8144,  -0.9594],
        [ -1.3940, -10.7772,   6.7835],
        [ -0.9715, -10.8630,   0.3555],
        [ -2.1382,   7.8339,   7.3524],
        [ -9.4430,  -0.1897,   4.8383],
        [  4.0457,  -3.7737,  -0.4704],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -9.2707,   0.9661,   4.4863],
        [ -7.8359,   4.2735,   7.2943],
        [ -8.5607,  -5.0938,  -5.4531],
        [  6.0310,  12.2795,  -6.4528],
        [ 10.9599,  -5.1849,  -3.5526],
        [  3.2226,  -4.2802,  -4.0428],
        [  1.7659,   7.4127,   4.7871],
        [ -9.2373,  -3.1225,  -8.9763],
        [ -8.7735,   2.5640,  -6.3313],
        [  6.1929,   2.4501,  -6.9352],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.1666,   3.7653,   3.6075],
        [  1.4489,   7.5256,   4.4529],
        [  4.6357,   5.4415,   7.9067],
        [  7.7750,  11.0665,   2.8364],
        [  2.1504,  -4.6726,   8.7908],
        [  6.2036,   2.4030,  -1.2500],
        [  5.6260,  -1.9401,   3.0259],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.4426, -10.1181,  -5.1268],
        [  0.3038,   7.8092,   0.4164],
        [  6.2022,   2.4092,   3.0349],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.6381,  -4.9238,  -4.4077],
        [  4.7813,  -3.1139,   8.2691],
        [  3.5284, -10.7484,   5.3347],
        [ -1.6275, -10.7221,   3.2376],
        [  0.1677,  -4.7447,  -6.1529],
        [  1.3339,   7.5627,  -0.9034],
        [  3.7916,  -3.9524,  -7.9547],
        [ 12.3357,   3.6001,   0.7903],
        [ -6.0844,  -8.3955,   0.7705],
        [ -3.2053, -10.1993,  -3.8736],
        [ -9.1459,   1.4780,  -4.2403],
        [  6.2419,  -0.2313,   4.2440],
        [  1.7218,  -4.7576,  -3.3617],
        [ -2.6485, -10.4151,   0.8766],
        [  2.7295,  13.6703,  -8.9328],
        [ -1.3940, -10.7772,   6.7835],
        [ -5.2589,   6.7024,  -5.5809],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -1.9714,   7.8563,  -1.8084]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0259, 0.0225,  ..., 0.8051, 1.2860, 0.5434],
        [0.0259, 0.0000, 0.0238,  ..., 0.8254, 1.3061, 0.5587],
        [0.0225, 0.0238, 0.0000,  ..., 0.8262, 1.3072, 0.5659],
        ...,
        [0.8051, 0.8254, 0.8262,  ..., 0.0000, 0.4810, 0.3220],
        [1.2860, 1.3061, 1.3072,  ..., 0.4810, 0.0000, 0.7775],
        [0.5434, 0.5587, 0.5659,  ..., 0.3220, 0.7775, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.1471313 , 13.20988996, 13.31764181, 13.32645552, 13.32613445,
       13.25336001,  6.67108662, 12.30858677,  6.83515201, 12.27993098,
       12.70489563,  7.06169391, 11.00508681, 11.75438006, 12.79126785,
        6.02779201,  7.53884104,  5.53175275,  7.822854  , 12.30858677,
        9.76896904, 12.79126785,  5.88782913,  5.47702766,  9.75075273,
        6.76791483, 13.2430377 , 11.22993218, 10.95593729, 12.87752052,
        5.19065439,  6.06416957, 12.25806559,  5.70405525,  9.17905491,
       13.92390689, 10.68031351,  5.95111804,  9.3109884 ,  9.12220271,
        5.16611782,  9.75401449, 10.01804223,  9.7751189 ,  9.36378438,
       10.46739073, 12.07818871,  7.70897675,  6.7593917 , 11.42420508,
        8.14433542, 10.60226614,  5.91854397, 13.0493368 ,  8.01009083,
       10.412299  ,  7.57946459,  4.91038884, 10.86693708, 10.90638066,
        8.12043983,  9.44486609,  5.53255882, 10.53368644,  9.32094055,
        8.92549752,  9.96152788, 13.68063098, 12.12448912,  5.35774573,
        7.62011734,  9.75075273,  9.14044984,  6.65991877,  9.09119351,
        8.99276564,  7.66377554,  7.14837514, 13.5247215 ,  5.14371005,
        6.652742  ,  5.95111804,  9.73579738, 11.4890861 ,  7.81509631,
        6.65368551, 10.30972193,  9.94285511,  5.70591986, 11.31276109,
       10.84493732,  4.74770899,  7.67939546,  5.47702766, 12.85032836,
       10.36844813, 10.6910768 ,  9.26455795,  6.24614905,  5.05963217,
       10.74655346, 13.94008948, 10.86693708,  8.51928531, 10.35485331,
        8.09984231])}
{'P': tensor([[2.9556e-03, 2.9945e-01, 1.9687e-01,  ..., 1.3432e-05, 1.0371e-06,
         5.6157e-02],
        [9.9249e-06, 4.4486e-03, 9.9575e-06,  ..., 1.0458e-05, 1.1290e-06,
         1.2559e-03],
        [4.8201e-05, 3.9765e-01, 3.5758e-03,  ..., 1.5494e-05, 1.2238e-06,
         5.4113e-02],
        ...,
        [1.1970e-02, 1.1257e-02, 1.1748e-02,  ..., 1.2787e-01, 1.4461e-03,
         2.0547e-02],
        [2.6862e-03, 1.7152e-03, 2.5887e-03,  ..., 1.0008e-01, 1.0330e-01,
         2.5002e-03],
        [3.7584e-05, 2.9978e-03, 4.7482e-05,  ..., 1.4312e-05, 1.4244e-06,
         4.6535e-03]]), 'X': tensor([[ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-6.3066e+00, -8.1936e+00,  9.5822e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-7.5401e+00,  4.6718e+00,  9.0033e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0185, 0.0041,  ..., 0.4630, 0.4808, 0.0652],
        [0.0185, 0.0000, 0.0145,  ..., 0.4815, 0.4993, 0.0821],
        [0.0041, 0.0145, 0.0000,  ..., 0.4670, 0.4849, 0.0686],
        ...,
        [0.4630, 0.4815, 0.4670,  ..., 0.0000, 0.0464, 0.4056],
        [0.4808, 0.4993, 0.4849,  ..., 0.0464, 0.0000, 0.4266],
        [0.0652, 0.0821, 0.0686,  ..., 0.4056, 0.4266, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.14777433, 13.27717982, 13.15525803, 13.41713661, 13.45124917,
       13.40850328, 11.97238624,  8.11227668,  6.65991877,  5.47702766,
       10.33958639,  9.70108409,  7.6374603 ,  9.85330646, 10.95593729,
       11.77124373,  9.45995955, 13.96905756, 12.07225469,  6.49538841,
       11.05317597, 13.27994304, 12.79126785, 13.3426541 ,  8.89301516,
       10.35464916,  9.51074763,  9.97822473, 10.52357479,  9.61942445,
        5.18047638, 10.27716434, 13.66051884,  9.97822473,  5.5748624 ,
        9.48337063,  8.87007495, 12.27835996,  5.13616224,  9.88571713,
       11.73567759, 10.33712258, 13.76047999,  6.27905292, 10.91805683,
        5.9510489 ,  4.83371341,  6.46560181,  8.08320235,  7.59795445,
        9.17905491, 12.3489324 ,  9.18690018,  5.04028538,  7.92200342,
       12.45418236,  9.17016857, 11.11969815,  5.95297459,  8.22728399,
       13.18278606,  9.68324553,  8.49161154, 12.4980845 ,  8.1137452 ,
        5.89074104,  9.97822473,  6.46356806, 11.04629471,  9.87720102,
       12.46513646,  7.58378655,  5.92281142,  5.05963217, 12.27991991,
        7.20482224,  9.32094055,  4.74468406, 13.35927527, 12.62711252,
       10.86693708,  5.38342077, 10.1979669 ,  7.20482224, 12.91143491,
       11.1913598 ,  8.06990483,  6.69389431, 10.26545792,  4.92497514,
       13.04051744, 11.83570132,  9.25336371, 11.70016217,  7.13501264,
        8.91551231, 12.49291768, 11.99633819, 14.05687815, 13.45642379,
        6.35029274,  4.73784222,  8.90405823,  6.46356806,  5.5513072 ,
       13.26590071])}
{'P': tensor([[3.1164e-03, 3.4656e-01, 1.8073e-01,  ..., 1.5230e-07, 2.6073e-04,
         3.2770e-07],
        [6.5277e-06, 2.9164e-03, 3.0198e-01,  ..., 7.8175e-08, 5.5790e-05,
         4.0521e-07],
        [2.3055e-06, 4.8005e-06, 2.4986e-03,  ..., 4.0290e-08, 1.0924e-05,
         5.6648e-07],
        ...,
        [1.0062e-03, 6.2530e-04, 4.6742e-04,  ..., 2.4074e-02, 1.2571e-02,
         9.5381e-07],
        [6.5122e-04, 1.9444e-03, 7.3733e-03,  ..., 1.0264e-06, 1.4280e-02,
         3.0128e-06],
        [1.0991e-06, 8.6552e-07, 7.6567e-07,  ..., 2.0608e-05, 1.7547e-06,
         5.6089e-02]]), 'X': tensor([[ 10.9394,   7.3069,  -8.9365],
        [ 10.0619,   8.6627,  -9.2884],
        [  8.8474,  10.0868,  -8.8644],
        [  8.5198,  10.4091,  -8.6486],
        [  8.9284,  10.0036,  -6.9210],
        [  7.0642,  11.6104,  -9.8316],
        [  6.2790,  -0.0259,  -5.4116],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -2.0445,   7.8469,  -0.5857],
        [ 12.5832,   1.7543,  -7.6134],
        [  4.6231,   5.4556,   8.4973],
        [  6.3303,   1.6145,  -8.4083],
        [  8.3718,  -8.2953,   8.4666],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.4541,  -0.3327,   8.8246],
        [  7.7794,  11.0629,   8.9069],
        [ -0.0410,   7.8591,   9.3457],
        [  0.7685,  -4.8098,  -2.9310],
        [  5.4426, -10.1181,  -5.1268],
        [  2.3261,   7.1726,   6.2522],
        [ -9.1433,  -3.5003,  -3.0209],
        [  2.2076,  13.8032,  -1.5099],
        [  1.6308,  13.9254,  -2.0984],
        [ 12.1504,   4.3674,  -7.2998],
        [  6.3124,   0.2057,  -0.4593],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.3900,   3.3727,   3.1650],
        [  6.0047,   3.1062,  -6.7914],
        [ -1.5261,   7.8988,   9.3768],
        [ -8.7735,   2.5640,  -6.3313],
        [ 12.5789,   0.1722,   1.6470],
        [  3.2692,  13.5098,  -4.9032],
        [  5.6928,  12.4710,   3.6001],
        [  6.0305,   3.0296,   0.1375],
        [  5.7963,  -1.6018,   1.8856],
        [ -8.7078,   2.7182,   1.8146],
        [  5.0233,   4.9647,  -3.1788],
        [ 11.5568,   6.0602,   9.3244],
        [  6.0788,  -0.8710,  -6.8719],
        [ -6.3066,  -8.1936,   9.5822],
        [  6.2535,  -0.1709,  -9.4102],
        [  6.2905,   1.9366,   2.1169],
        [ -4.6942,  -9.4299,  -9.1846],
        [  1.4306,  13.9620,  -0.2075],
        [ -1.6086,   7.8928,   4.3385],
        [  4.6357,   5.4415,   7.9067],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -1.1885, -10.8211,   0.0865],
        [ -4.0633,  -9.7927,  -3.0825],
        [  4.8432,   5.1968,   1.5006],
        [ -5.3382,   6.6533,   7.8054],
        [  9.6614,   9.1784,  -1.2087],
        [  5.5626,  -2.0536,   8.4433],
        [  3.8111, -10.6787,  -4.6539],
        [  4.2041,  -3.6509,  -4.9488],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.1984,  -0.4327,   1.5438],
        [  7.3724,  -9.0577,  -8.9843],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.4167, -10.1293,   6.2156],
        [  6.3534,   1.2990,  -7.2942],
        [  5.8930,  12.3592,  -8.9201],
        [ -8.6426,  -4.9136,   9.4941],
        [ -4.2271,   7.2387,  -1.8316],
        [  0.7685,  -4.8098,  -2.9310],
        [  1.0093,   7.6567,   2.3526],
        [  3.2692,  13.5098,  -4.9032],
        [  0.9659,  -4.8144,  -0.9594],
        [ -6.7992,   5.4991,   9.9186],
        [ -9.4117,   0.1160,   8.2574],
        [  1.7659,   7.4127,   4.7871],
        [  2.0243,   7.3086,   7.5432],
        [  4.6992,   5.3690,   3.0935],
        [  6.3587,   1.1596,  -3.6629],
        [  2.3928,   7.1404,  -1.7035],
        [  6.2365,  -0.2578,  -6.0212],
        [  6.2036,   2.4030,  -1.2500],
        [  9.9217,  -6.7006,  -5.6728],
        [ -8.0766,  -6.0046,  -3.1873],
        [  8.5198,  10.4091,  -8.6486],
        [  6.1989,  -9.7591,   0.8289],
        [ -8.1204,  -5.9308,   4.0761],
        [ -7.3554,   4.8985,  -0.6665],
        [ 11.1420,   6.9331,  -2.4779],
        [ -5.1369,   6.7755,   8.3836],
        [ -6.4844,  -8.0230,   8.5209],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 12.4690,  -0.8528,  -9.3374],
        [  4.1348,   5.9514,  -0.8856],
        [  2.2076,  13.8032,  -1.5099],
        [  2.6310, -10.9195,   3.8054],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -9.4157,   0.0821,  -5.0025],
        [ 12.5717,   1.9266,   7.4903],
        [  7.2340,  -9.1505,  -5.8321],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -2.1817,   7.8274,  -4.6298],
        [ -9.3604,   0.4855,  -3.0158],
        [ -9.4159,  -2.0739,   9.6841],
        [ 11.4703,  -4.2006,   7.4086],
        [  6.3360,   0.4258,  -9.2509],
        [ 11.7349,  -3.5838,   2.1644],
        [ 10.7566,   7.6212,  -2.3279],
        [ -9.4283,  -1.9585,  -0.9129]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.4453e-02, 6.2029e-03,  ..., 8.5264e-01, 4.1227e-01,
         1.3356e+00],
        [1.4453e-02, 0.0000e+00, 8.4106e-03,  ..., 8.6664e-01, 4.2471e-01,
         1.3481e+00],
        [6.2029e-03, 8.4106e-03, 0.0000e+00,  ..., 8.5829e-01, 4.1698e-01,
         1.3414e+00],
        ...,
        [8.5264e-01, 8.6664e-01, 8.5829e-01,  ..., 0.0000e+00, 4.7252e-01,
         9.3878e-01],
        [4.1227e-01, 4.2471e-01, 4.1698e-01,  ..., 4.7252e-01, 2.1453e-04,
         1.1911e+00],
        [1.3356e+00, 1.3481e+00, 1.3414e+00,  ..., 9.3878e-01, 1.1911e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.15525803, 13.27717982, 13.41713661, 13.45124917, 13.40850328,
       13.59056613,  6.27905292, 12.32545637,  8.10886021, 12.70489563,
        7.15100303,  6.53290663, 11.785566  , 12.97370371,  9.45995955,
       13.52430176,  7.85919687,  4.87083348, 11.4890861 ,  7.5403822 ,
        9.79039371, 13.97857591, 14.02058621, 12.91143491,  6.31576264,
        9.76896904,  9.04255655,  6.76056625,  8.04483495,  9.14044984,
       12.58006038, 13.89974693, 13.70893492,  6.74872981,  6.01356004,
        9.12220271,  7.06272256, 13.0493368 ,  6.14086582, 10.33958639,
        6.25586297,  6.58184474, 10.53368644, 14.03505624,  8.0550286 ,
        7.14837514,  9.48969978, 10.88618114, 10.60226614,  7.10378919,
        8.53014915, 13.32613445,  5.92952538, 11.33837396,  5.56809335,
       11.69264243,  6.2134961 , 11.67875855,  9.99145096,  9.94173708,
       11.48663767,  6.48486171, 13.6922425 ,  9.94173708,  8.38252937,
        4.87083348,  7.72291946, 13.89974693,  4.91038884,  8.74470218,
        9.41244865,  7.62011734,  7.58378655,  7.13501264,  6.46356806,
        7.53063885,  6.24187044,  6.652742  , 11.97238624, 10.06407839,
       13.45124917, 11.56146873, 10.05558893,  8.83725081, 13.12299145,
        8.50268322, 10.31584102,  9.99145096, 10.23946247, 12.4980845 ,
        7.24677664, 13.97857591, 11.23201619,  9.51074763,  9.41605684,
       12.71843462, 11.66453059,  9.51074763,  8.12580976,  9.37296102,
        9.64157461, 12.21532282,  6.35029274, 12.26997165, 13.18278606,
        9.62958973])}
{'P': tensor([[3.7670e-03, 2.8728e-01, 1.7734e-01,  ..., 1.1202e-06, 1.5962e-03,
         2.4791e-08],
        [2.1623e-05, 1.5660e-02, 7.1489e-04,  ..., 8.7983e-06, 8.5196e-03,
         9.0530e-08],
        [6.0602e-05, 9.2214e-02, 1.8291e-02,  ..., 8.4895e-06, 7.1888e-03,
         1.2944e-07],
        ...,
        [1.1017e-06, 8.4972e-07, 8.9916e-07,  ..., 3.6984e-02, 1.2278e-05,
         5.6634e-04],
        [6.0541e-05, 1.4159e-04, 1.0562e-04,  ..., 1.2792e-03, 6.4904e-02,
         5.3914e-06],
        [1.0010e-05, 3.9094e-06, 6.5371e-06,  ..., 1.4179e-06, 3.5248e-06,
         2.0123e-02]]), 'X': tensor([[ 10.7566,   7.6212,  -2.3279],
        [  9.1820,   9.7334,  -2.7932],
        [  9.6614,   9.1784,  -1.2087],
        [ 10.3251,   8.2928,  -3.0006],
        [  9.0652,   9.8597,  -3.5949],
        [  9.7328,   9.0902,  -0.8917],
        [  7.3724,  -9.0577,  -8.9843],
        [ 12.3609,   3.4769,  -8.5762],
        [ -8.7544,   2.6096,   3.3146],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.3373,  13.7721,  -0.2442],
        [ -2.6069, -10.4298,  -6.1995],
        [ 12.2592,   3.9425,   2.0383],
        [ -5.9987,   6.1949,  -6.2114],
        [  9.3560,   9.5391,   9.4943],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.6165,  -3.8715,   6.4594],
        [ -8.0087,   4.0173,   9.1091],
        [  5.7802,  -1.6363,  -6.2782],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.0031,  12.2958,  -6.5992],
        [ 10.0567,   8.6697,   4.6329],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.9526,   3.2525,  -4.3821],
        [  3.1756, -10.8247,   6.8882],
        [  4.5800,   5.5034,  -3.9432],
        [ 11.4495,   6.3026,   6.8257],
        [ -9.2687,  -2.9794,   0.6436],
        [  5.0277,   4.9589,   7.0175],
        [  6.3031,   1.8477,  -7.1706],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.0348,  -2.8229,   3.5436],
        [  7.3724,  -9.0577,  -8.9843],
        [ -7.6719,  -6.6260,   7.1468],
        [ -8.9033,   2.2329,  -8.8912],
        [ -8.3653,   3.4184,   9.3454],
        [  7.0515,  11.6194,   8.4289],
        [  4.5800,   5.5034,  -3.9432],
        [ 12.0845,  -2.5772,  -4.6073],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -9.4203,  -2.0340,   2.5856],
        [  9.0004,  -7.7195,   4.6945],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.5859,  -5.0393,   0.5476],
        [ -4.9217,   6.8979,  -1.7667],
        [  2.2659,  13.7894,   6.5436],
        [ -6.5295,  -7.9785,  -7.4429],
        [  3.3633,  13.4794,   0.5193],
        [ -5.1369,   6.7755,   8.3836],
        [ -1.2454,   7.9127,  -0.5715],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 10.9162,  -5.2599,  -7.0812],
        [  0.1532,  -4.7422,  -7.0145],
        [ -8.5506,  -5.1154,   6.3423],
        [  1.8447,  -4.7372,   8.1235],
        [  0.6470, -11.0352,  -1.1931],
        [ -7.9298,   4.1367,  -3.9773],
        [  3.5284, -10.7484,   5.3347],
        [ 12.4996,   2.6335,  -6.3166],
        [ -0.5739,   7.9056,   0.6382],
        [ 12.3357,   3.6001,   0.7903],
        [  4.9152,  12.8653,  -1.4459],
        [  5.6314,  -1.9302,  -9.6543],
        [ -4.4503,   7.1378,   3.3751],
        [  0.3231,  -4.7691,  -1.6120],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.9410,   6.8428,   5.4458],
        [  2.5829,  -4.5466,  -8.4015],
        [ 12.2435,   4.0076,   8.1257],
        [ 11.7726,  -3.4876,  -6.7402],
        [  6.3606,   0.9204,   3.2525],
        [  4.5954,  13.0099,  -4.4439],
        [ -2.1817,   7.8274,  -4.6298],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.2911,  -2.4824,  -3.6877],
        [ -0.0254, -10.9932,   3.2200],
        [ -9.3444,   0.5821,   6.9101],
        [ -8.6050,   2.9451,  -0.7012],
        [ 12.5763,   0.1333,  -9.2650],
        [  8.1217,  -8.5024,  -0.6149],
        [  5.5240,  -2.1198,   6.2055],
        [  4.8051, -10.3703,   0.5268],
        [ -3.9508,  -9.8515,   1.7923],
        [  2.5620,   7.0549,   8.4825],
        [  2.2849, -10.9655,  -7.6542],
        [ -0.5385,   7.9037,   6.3830],
        [  5.1886,  -2.6250,   6.4677],
        [  8.1217,  -8.5024,  -0.6149],
        [  2.1625,  -4.6697,   7.0658],
        [  1.0093,   7.6567,   2.3526],
        [  1.9204,   7.3518,   8.4781],
        [ -8.0611,  -6.0302,   4.6962],
        [  2.6614,  -4.5192,  -9.3641],
        [  3.1064,  13.5607,   6.7605],
        [ 10.1798,   8.5002,  -5.8893],
        [ -3.1204, -10.2345,   9.2288],
        [  0.1497,   7.8334,  -5.4412],
        [  9.9750,  -6.6338,  -1.2440],
        [  6.1973,  -0.4372,   6.5720],
        [  4.4915,   5.5990,   7.0405],
        [ -9.4766,  -1.1116,  -2.1706],
        [  1.6408,   7.4591,  -4.2294],
        [  0.1287, -11.0065,   6.5757]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 4.5360e-02, 1.0662e-01,  ..., 1.1849e+00, 4.2701e-01,
         1.2201e+00],
        [4.5360e-02, 0.0000e+00, 1.5128e-01,  ..., 1.1926e+00, 4.2812e-01,
         1.2539e+00],
        [1.0662e-01, 1.5128e-01, 0.0000e+00,  ..., 1.1960e+00, 4.6395e-01,
         1.1613e+00],
        ...,
        [1.1849e+00, 1.1926e+00, 1.1960e+00,  ..., 0.0000e+00, 7.6854e-01,
         8.0364e-01],
        [4.2701e-01, 4.2812e-01, 4.6395e-01,  ..., 7.6854e-01, 0.0000e+00,
         9.9551e-01],
        [1.2201e+00, 1.2539e+00, 1.1613e+00,  ..., 8.0364e-01, 9.9551e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.18278606, 13.38091121, 13.32613445, 13.2430377 , 13.39372426,
       13.31764181, 11.67875855, 12.84056604,  9.13507321,  4.76969364,
       13.96905756, 10.7506423 , 12.87752052,  8.62329466, 13.3614598 ,
        8.85572292, 13.08414689, 12.24467041,  8.95983586,  6.00732193,
       10.412299  , 13.6829843 , 13.2778378 ,  4.74468406,  6.78325151,
       11.28093507,  7.15990676, 13.06957425,  9.73579738,  7.06169391,
        6.56833864,  9.99211663,  5.77219077, 11.67875855, 10.1371552 ,
        9.17905491,  9.0368458 , 13.59170809,  7.15990676, 12.35625694,
       12.05300943,  9.63742669, 11.85743425, 10.1979669 , 10.06437598,
        8.86082124,  9.95552626,  8.47371868, 13.97430191, 10.30972193,
       13.89264822,  8.50268322,  8.01009083,  9.71392179, 12.11735334,
        4.74468406,  9.96391045,  5.08372149, 11.05413545,  8.94395628,
       11.31276109, 12.77402351,  7.92644746, 12.85032836, 13.77222752,
        5.95297459,  8.41148722,  4.78002665, 12.10222204,  7.44805901,
        5.22903117, 12.88270018, 12.27835996,  6.42689705, 13.79762415,
        8.12580976,  9.64157461,  5.84450127, 10.99326577,  9.36256096,
        9.09502385, 12.5769659 , 11.75807252,  5.91677188, 11.42947632,
       10.61417181,  7.50564223, 11.20100795,  7.92200342,  5.81481044,
       11.75807252,  5.14607014,  7.72291946,  7.5984809 , 10.06703358,
        5.24461068, 13.91198191, 13.26209545, 10.69963954,  7.83486943,
       11.97949273,  6.21275016,  7.17789666,  9.54154096,  7.6374603 ,
       11.00726483])}
{'P': tensor([[2.8601e-03, 2.5663e-01, 1.9220e-01,  ..., 4.8326e-02, 2.9577e-04,
         4.8641e-08],
        [9.7066e-06, 7.1521e-03, 4.3291e-04,  ..., 1.4162e-05, 1.4030e-03,
         1.7582e-07],
        [9.1832e-06, 2.7589e-02, 4.0181e-03,  ..., 2.9809e-05, 5.8917e-04,
         7.7586e-08],
        ...,
        [8.5323e-05, 2.5621e-01, 6.2645e-02,  ..., 2.9607e-03, 3.8615e-04,
         6.0411e-08],
        [1.4793e-05, 2.3948e-05, 2.0962e-05,  ..., 1.6522e-05, 8.7395e-02,
         3.5388e-04],
        [8.2845e-07, 5.2447e-07, 4.7894e-07,  ..., 8.4693e-07, 4.0121e-06,
         3.5433e-02]]), 'X': tensor([[ 10.5955,   7.8824,  -3.7006],
        [  9.0652,   9.8597,  -3.5949],
        [  9.3607,   9.5337,  -4.5412],
        [  9.1820,   9.7334,  -2.7932],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  1.4150,  13.9647,  -4.0915],
        [  2.0243,   7.3086,   7.5432],
        [  3.3069,   6.6089,   8.7377],
        [  6.2883,   0.0325,   5.2692],
        [  4.0642,  -3.7599,  -3.4565],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -5.3179,  -9.0101,   9.4472],
        [  0.6396,  14.0772,  -6.3786],
        [  2.0039,   7.3173,  -1.5885],
        [ -3.2053, -10.1993,  -3.8736],
        [ -2.1533, -10.5778,   1.0112],
        [  6.3303,   1.6145,  -8.4083],
        [  1.7218,  -4.7576,  -3.3617],
        [  6.2026,  12.1774,  -1.5887],
        [ -6.7745,  -7.7264,   8.4684],
        [ 12.2592,   3.9425,   2.0383],
        [  3.4140,  -4.1803,  -8.8224],
        [  2.3889,  -4.6082,  -8.7291],
        [  7.4132,  11.3526,   0.3322],
        [ 11.9936,  -2.8677,   6.0614],
        [  3.9663,   6.1018,  -5.4104],
        [  0.9809,  -4.8145,   6.8706],
        [  4.6357,   5.4415,   7.9067],
        [  4.2041,  -3.6509,  -4.9488],
        [ -4.0633,  -9.7927,  -3.0825],
        [  1.3705,  13.9723,   7.9266],
        [  1.4150,  13.9647,  -4.0915],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -8.5149,  -5.1905,   1.5239],
        [  9.2979,   9.6048,   7.3138],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.6093,  -1.9705,   6.4807],
        [  1.7000,  -4.7609,   2.7935],
        [  2.3490,  -4.6199,  -4.2489],
        [  0.8700,  -4.8132,  -7.8042],
        [  0.1041, -11.0045,   2.4311],
        [  6.2036,   2.4030,  -1.2500],
        [  9.5203,   9.3483,   5.8781],
        [  5.6523,   3.9443,  -2.9449],
        [  2.2860,   7.1916,  -6.1011],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.9587,  12.3216,   9.1695],
        [  5.2911,  -2.4824,  -3.6877],
        [  6.2026,  12.1774,  -1.5887],
        [  4.4633,   5.6287,  -0.7754],
        [  6.3505,   0.6235,  -8.5849],
        [  9.2692,   9.6370,   7.0746],
        [  1.1006,   7.6318,  -2.4877],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -0.5739,   7.9056,   0.6382],
        [  6.2576,   2.1359,  -3.3398],
        [  5.2608,  -2.5256,  -9.1446],
        [  4.7544,  -3.1425,  -1.5315],
        [  5.9026,   3.3837,  -0.6573],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 12.2311,  -2.0445,   1.3545],
        [  8.1043,  -8.5164,  -7.9290],
        [ 10.5678,   7.9261,  -0.6100],
        [ -9.3384,  -2.6161,  -4.7630],
        [ 10.9162,  -5.2599,  -7.0812],
        [  5.7218,  -9.9934,   6.5764],
        [ -4.9217,   6.8979,  -1.7667],
        [  1.8766,   7.3694,   1.1887],
        [ -1.9714,   7.8563,  -1.8084],
        [  3.1892, -10.8220,  -6.8409],
        [  7.4132,  11.3526,   0.3322],
        [  0.9478,  -4.8144,   0.6371],
        [  6.0140,  -1.0661,  -2.5040],
        [ 12.1588,   4.3364,  -8.0867],
        [ -3.5173, -10.0623,   4.8280],
        [ 12.5927,   0.4199,  -9.4997],
        [ -8.0965,  -5.9712,   4.3466],
        [ -9.4296,  -1.9458,   4.0833],
        [ -0.0282,   7.8575,  -1.1746],
        [  0.1460,   7.8340,  -9.4614],
        [ 10.9916,   7.2133,  -0.9179],
        [ -5.9225,  -8.5355,   2.4778],
        [  6.1685,  12.1979,  -6.2473],
        [ -3.2736, -10.1703,   3.8712],
        [ 12.3609,   3.4769,  -8.5762],
        [  3.7237,  -3.9966,  -5.0376],
        [ -9.3604,   0.4855,  -3.0158],
        [  4.4633,   5.6287,  -0.7754],
        [  6.8704,  -9.3809,   7.5769],
        [  9.1409,  -7.5789,   1.5814],
        [ -2.7572,   7.7194,   5.9580],
        [  2.0547,   7.2957,  -1.9343],
        [ 11.7796,  -3.4696,   9.0088],
        [ 10.9931,   7.2107,   7.7933],
        [ -0.8675, -10.8815,   6.8561],
        [  3.3633,  13.4794,   0.5193],
        [ 11.3118,   6.5957,  -4.5667],
        [ -6.6998,   5.5954,   0.0823],
        [  6.3569,   0.7591,   6.5011],
        [  6.3550,   0.7129,   0.4917],
        [ 10.3251,   8.2928,  -3.0006],
        [ -2.0445,   7.8469,  -0.5857],
        [ -7.6719,  -6.6260,   7.1468]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.3708e-02, 6.9998e-02,  ..., 6.4302e-02, 6.3113e-01,
         1.4528e+00],
        [1.3708e-02, 0.0000e+00, 8.0267e-02,  ..., 5.2811e-02, 6.3239e-01,
         1.4517e+00],
        [6.9998e-02, 8.0267e-02, 0.0000e+00,  ..., 1.3307e-01, 6.7860e-01,
         1.5070e+00],
        ...,
        [6.4302e-02, 5.2811e-02, 1.3307e-01,  ..., 2.1453e-04, 6.0689e-01,
         1.4175e+00],
        [6.3113e-01, 6.3239e-01, 6.7860e-01,  ..., 6.0689e-01, 0.0000e+00,
         8.3684e-01],
        [1.4528e+00, 1.4517e+00, 1.5070e+00,  ..., 1.4175e+00, 8.3684e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.20597932, 13.39372426, 13.36092624, 13.38091121, 13.45642379,
       13.59480906, 14.03618649,  7.58378655,  7.39004226,  6.28835101,
        5.5366698 ,  9.85330646, 10.46238054, 14.09176259,  7.58668144,
       10.6910768 , 10.794755  ,  6.53290663,  5.05963217, 13.6660671 ,
       10.27572673, 12.87752052,  5.39723241,  5.19065439, 13.55868684,
       12.33167382,  7.27759233,  4.91336994,  7.14837514,  5.56809335,
       10.60226614, 14.0393949 , 14.03618649,  9.50654967,  9.97220799,
       13.36800073, 10.31613269,  5.94535529,  5.05534609,  5.18277858,
        4.89121805, 11.0050379 ,  6.652742  , 13.3426541 ,  6.89245352,
        7.54620849, 10.06850677, 13.68672855,  5.84450127, 13.6660671 ,
        7.18354885,  6.38105015, 13.37121923,  7.71075226,  4.92497514,
        7.48311002,  9.50654967,  7.92644746,  6.61211224,  5.83560009,
        5.69913817,  6.80370254, 10.46739073, 12.40074901, 11.75618513,
       13.20988996,  9.69792513, 12.11735334, 11.51557166,  8.47371868,
        7.60463349,  8.09984231, 11.28215429, 13.55868684,  4.90676242,
        6.10774544, 12.9089559 , 10.65929968, 12.59969936, 10.06023665,
        9.6282763 ,  7.8575648 ,  7.83534987, 13.1471313 , 10.38897563,
       13.66896739, 10.68416401, 12.84056604,  5.46244988,  9.37296102,
        7.18354885, 11.62770332, 11.87412439,  8.19703512,  7.57946459,
       12.27993098, 13.14690137, 10.91601655, 13.89264822, 13.09426142,
        8.72896381,  6.4020404 ,  6.3949043 , 13.2430377 ,  8.10886021,
       10.1371552 ])}
{'P': tensor([[1.9663e-03, 2.5052e-01, 2.4802e-01,  ..., 1.3146e-02, 5.4048e-08,
         1.8560e-08],
        [3.0673e-05, 1.4575e-02, 1.1375e-04,  ..., 2.4121e-02, 4.8417e-07,
         1.4271e-07],
        [1.2413e-05, 6.1048e-01, 5.3478e-03,  ..., 1.1425e-02, 1.7288e-07,
         5.1589e-08],
        ...,
        [1.4809e-04, 1.3242e-03, 1.0499e-03,  ..., 2.5492e-02, 6.2669e-07,
         1.7606e-07],
        [3.2346e-06, 2.2917e-06, 2.3225e-06,  ..., 1.2708e-06, 8.5857e-02,
         5.5263e-01],
        [6.2141e-06, 3.8426e-06, 3.9280e-06,  ..., 1.8524e-06, 5.9305e-05,
         4.0480e-02]]), 'X': tensor([[ 10.5678,   7.9261,  -0.6100],
        [  9.6587,   9.1817,  -0.8212],
        [  9.7328,   9.0902,  -0.8917],
        [  9.6614,   9.1784,  -1.2087],
        [  9.3624,   9.5318,   0.2779],
        [ 10.2470,   8.4054,  -0.1177],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.0890,  -0.8382,   2.9166],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.0233,   4.9647,  -3.1788],
        [  1.7659,   7.4127,   4.7871],
        [ -3.5029,   7.5138,   8.1853],
        [  3.4009,  -4.1874,  -5.9751],
        [ -5.6423,   6.4537,  -1.2091],
        [  5.2939,  -2.4784,  -4.5840],
        [  5.3858,   4.4260,  -9.6769],
        [  5.7784,  -1.6401,   9.5587],
        [  5.9003,  12.3550,   2.1917],
        [  0.6363, -11.0348,   6.1414],
        [ -7.9667,   4.0814,  -0.6397],
        [  6.0345,   3.0175,  -1.4424],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.2687,  -2.9794,   0.6436],
        [ -9.3444,   0.5821,   6.9101],
        [  3.3094,  13.4969,  -3.6507],
        [  1.1350,  14.0104,   0.6321],
        [  1.5662,  13.9375,   4.5142],
        [ -7.2515,   5.0195,   4.6434],
        [  6.2022,   2.4092,   3.0349],
        [  3.6469,   6.3624,  -1.8414],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.8447,  -4.7372,   8.1235],
        [ -9.3464,   0.5708,  -1.6397],
        [  6.1929,   2.4501,  -6.9352],
        [ 11.2767,  -4.6006,   0.7868],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -7.8126,   4.3066,   5.1791],
        [ -8.9788,   2.0208,  -3.2867],
        [  6.7792,  11.8084,   1.6251],
        [ -1.2454,   7.9127,  -0.5715],
        [ 11.4743,   6.2478,  -2.1100],
        [ 12.3025,  -1.7440,  -6.6632],
        [  2.6310, -10.9195,   3.8054],
        [ -2.5056,   7.7720,   5.5878],
        [  3.3744,   6.5623,  -6.3636],
        [ -7.6525,  -6.6534,   5.1721],
        [ -7.8280,   4.2848,  -7.1955],
        [ -9.4203,  -2.0340,   2.5856],
        [ 11.9808,   4.9340,  -3.3715],
        [  4.1637,  13.1898,  -1.9189],
        [  4.5325,  -3.3643,   6.7699],
        [  2.0039,   7.3173,  -1.5885],
        [  1.2728, -11.0372,   9.8121],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.7409,  -4.6837,  -6.5840],
        [  0.1532,  -4.7422,  -7.0145],
        [ -3.5520,   7.4976,   9.1522],
        [  6.2419,  -0.2313,   4.2440],
        [  0.1532,  -4.7422,  -7.0145],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.3419,   0.5972,  -1.2841],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -9.4296,  -1.9458,   4.0833],
        [  3.0719, -10.8449,  -4.1673],
        [  5.8318,   3.5567,  -4.7692],
        [ -5.0132,  -9.2234,  -8.3531],
        [  2.4149,  -4.6005,  -4.3339],
        [ -9.3739,   0.3982,  -4.2564],
        [  9.1620,  -7.5573,  -9.9882],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.4714,  -1.3301,  -3.1998],
        [  3.3069,   6.6089,   8.7377],
        [ 10.0400,   8.6922,  -6.2847],
        [ -9.4157,   0.0821,  -5.0025],
        [  5.6672,  -1.8631,   8.1020],
        [ -7.7239,   4.4298,   3.1216],
        [  6.2576,   2.1359,  -3.3398],
        [ -8.4267,  -5.3697,   2.4172],
        [  0.1041, -11.0045,   2.4311],
        [ 11.7794,   5.5086,  -9.6929],
        [  8.8091,  -7.9037,   0.3631],
        [ -7.8126,   4.3066,   5.1791],
        [ -0.9715, -10.8630,   0.3555],
        [  7.1441,  11.5529,   2.3232],
        [  4.1297,  -3.7098,  -3.9838],
        [  0.6952,  -4.8060,   8.8976],
        [  6.2804,   2.0019,   0.8607],
        [  6.3542,   0.6934,  -3.3840],
        [ -9.4366,  -0.1189,  -2.6736],
        [  5.4429,  -2.2529,   9.7306],
        [  6.0467,  -9.8369,  -6.5816],
        [  2.2437,  -4.6489,   6.1430],
        [  8.6420,  -8.0580,   6.5781],
        [ -6.1899,  -8.3011,  -7.9269],
        [  5.4242, -10.1261,   3.9595],
        [ 11.5083,  -4.1177,   0.2389],
        [  4.2041,  -3.6509,  -4.9488],
        [ 12.0582,  -2.6640,   0.7132],
        [  2.0870,  -4.6877,   9.6823],
        [  2.7976,  -4.4682,  -4.2020],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.6675,  -4.8569,  -9.1769],
        [  9.0652,   9.8597,  -3.5949],
        [ -7.6719,  -6.6260,   7.1468],
        [ -3.3812, -10.1235,   7.4384]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0202, 0.0238,  ..., 0.2702, 1.3051, 1.2493],
        [0.0202, 0.0000, 0.0063,  ..., 0.2541, 1.3233, 1.2681],
        [0.0238, 0.0063, 0.0000,  ..., 0.2484, 1.3235, 1.2688],
        ...,
        [0.2702, 0.2541, 0.2484,  ..., 0.0000, 1.4517, 1.4111],
        [1.3051, 1.3233, 1.3235,  ..., 1.4517, 0.0000, 0.1045],
        [1.2493, 1.2681, 1.2688,  ..., 1.4111, 0.1045, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.20988996, 13.32645552, 13.31764181, 13.32613445, 13.3607379 ,
       13.25336001, 13.00413829,  6.1463896 ,  9.64157461,  7.06272256,
        7.62011734,  8.29022197,  5.39450305,  8.5724042 ,  5.84533345,
        6.97106912,  6.00663314, 13.69162795, 11.05317597,  8.95133961,
        6.74686071,  7.48311002,  9.73579738,  9.36256096, 13.89671314,
       14.05633441, 14.02526505,  8.81930525,  6.65368551,  7.33350642,
        5.84450127,  5.08372149,  9.36378438,  6.65991877, 12.17901033,
       10.09979747,  8.92098752,  9.20340031, 13.61600536,  8.01009083,
       13.06498554, 12.42554808, 11.23201619,  8.16584537,  7.37902766,
       10.14044489,  8.92395381,  9.63742669, 12.95702647, 13.83139024,
        5.64464807,  7.58668144, 11.11038551,  5.14999799,  9.91668182,
        4.74468406,  8.29640613,  6.24614905,  4.74468406,  6.01356004,
        9.36093562, 10.60226614,  9.6282763 , 11.27160035,  6.83083863,
       10.4977132 ,  5.19577146,  9.38232121, 11.87665501,  9.96391045,
        9.5642993 ,  7.39004226, 13.27994304,  9.41605684,  5.96556248,
        8.90405823,  6.61211224,  9.99211663, 11.0050379 , 13.00380624,
       11.83509372,  8.92098752, 10.90638066, 13.58333857,  5.5513072 ,
        4.85603886,  6.59176018,  6.39187285,  9.43737523,  5.89074104,
       11.54672755,  5.16205087, 11.8159254 , 10.35485331, 11.48734978,
       12.22276278,  5.56809335, 12.3489324 ,  5.13125513,  5.27178363,
       10.27716434, 10.24557047,  9.93553424, 13.39372426, 10.1371552 ,
       10.67322527])}
{'P': tensor([[3.4107e-03, 3.9715e-01, 2.9515e-01,  ..., 7.3243e-05, 9.9441e-04,
         5.2756e-05],
        [2.3896e-05, 1.3203e-02, 3.0375e-02,  ..., 1.1867e-04, 2.1282e-03,
         3.4031e-04],
        [1.7840e-05, 1.9645e-03, 9.0097e-03,  ..., 7.6948e-05, 1.5859e-03,
         1.9082e-04],
        ...,
        [8.7115e-03, 1.2572e-02, 1.4002e-02,  ..., 5.7795e-02, 1.9116e-01,
         7.9281e-04],
        [3.2653e-04, 7.4195e-04, 8.6291e-04,  ..., 5.5730e-04, 3.8833e-02,
         9.6151e-04],
        [9.8471e-06, 1.2970e-05, 1.2131e-05,  ..., 8.1460e-05, 1.2112e-04,
         1.7313e-01]]), 'X': tensor([[ 10.3251,   8.2928,  -3.0006],
        [  9.1820,   9.7334,  -2.7932],
        [  9.0652,   9.8597,  -3.5949],
        [  9.3607,   9.5337,  -4.5412],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  6.8093,  -9.4178,   8.4244],
        [  2.2182,  13.8007,   5.1889],
        [  2.5663,   7.0526,  -6.8769],
        [  2.3376,  -4.6231,   9.8436],
        [  4.6357,   5.4415,   7.9067],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.0788,  -0.8710,  -6.8719],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -7.3333,   4.9246,  -7.9212],
        [ -8.8631,  -4.3728,  -1.1746],
        [ 12.4877,  -0.7222,   8.8347],
        [ 12.5996,   1.3971,  -1.0694],
        [ 12.2692,  -1.8882,   3.9253],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.1426,   3.8048,   9.3542],
        [  2.7558,  -4.4844,   0.0636],
        [  5.7485,  -9.9811,  -3.2355],
        [  0.5960, -11.0335,   3.1144],
        [ -9.4501,  -1.7133,   2.2980],
        [  5.1519, -10.2387,   9.8191],
        [  2.7590,  13.6621,   9.4496],
        [ -4.0868,   7.2981,   8.1874],
        [  1.7659,   7.4127,   4.7871],
        [  3.3744,   6.5623,  -6.3636],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.2905,   1.9366,   2.1169],
        [ -8.8540,  -4.3971,  -5.2144],
        [  9.8082,  -6.8396,   4.9615],
        [  2.5813,  -4.5471,   5.3597],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.1624,  -3.4286,   2.0647],
        [  3.9895,   6.0816,  -6.5235],
        [  0.1382, -11.0073,   1.0026],
        [  2.4932,   7.0903,  -5.4214],
        [  4.3459,  -3.5327,   0.3667],
        [ 10.9162,  -5.2599,  -7.0812],
        [  6.0305,   3.0296,   0.1375],
        [  6.0788,  -0.8710,  -6.8719],
        [ -8.8745,   2.3097,   3.5600],
        [  7.3724,  -9.0577,  -8.9843],
        [ -8.1531,  -5.8748,   2.6934],
        [  6.3515,   0.6418,  -6.5823],
        [  4.4290,  -3.4596,   1.4181],
        [ 12.5832,   1.7543,  -7.6134],
        [  0.2480, -11.0153,  -3.0183],
        [ 11.7779,   5.5127,  -2.0699],
        [ 12.0845,  -2.5772,  -4.6073],
        [  4.5057,  -3.3894,  -0.8160],
        [  0.1677,  -4.7447,  -6.1529],
        [  2.6321,  -4.5296,  -2.5159],
        [ 11.7795,  -3.4697,   3.5624],
        [ 10.5678,   7.9261,  -0.6100],
        [  0.9659,  -4.8144,  -0.9594],
        [  0.2435,   7.8191,  -6.4709],
        [  6.3550,   0.7129,   0.4917],
        [ -4.2040,   7.2486,   2.9875],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -5.6792,  -8.7352,   4.0318],
        [  9.2692,   9.6370,   7.0746],
        [ -5.7347,  -8.6908,   9.2878],
        [  3.9913,  -3.8139,  -6.8311],
        [ 10.0400,   8.6922,  -6.2847],
        [ -8.0588,  -6.0341,  -2.3006],
        [  4.2096,   5.8815,   1.9230],
        [  0.4119,  14.1019,  -4.1870],
        [ 11.5456,   6.0861,   8.1101],
        [  6.1973,  -0.4372,   6.5720],
        [  2.0729,   7.2878,   7.0880],
        [ 11.5568,   6.0602,   9.3244],
        [  5.2608,  -2.5256,  -9.1446],
        [  4.6063,  13.0051,  -3.1456],
        [  9.9756,  -6.6330,  -3.3429],
        [  4.2275,  -3.6320,  -0.2714],
        [  5.8201,   3.5841,  -8.0609],
        [  3.2809,  13.5061,  -4.0199],
        [ -1.1885, -10.8211,   0.0865],
        [ -9.3444,   0.5821,   6.9101],
        [ 11.1420,   6.9331,  -2.4779],
        [  5.7963,  -1.6018,   1.8856],
        [  1.5662,  13.9375,   4.5142],
        [ -9.4296,  -1.9458,   4.0833],
        [  0.3715,  -4.7756,   0.2312],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.4378,  -1.8606,  -2.5668],
        [  2.7295,  13.6703,  -8.9328],
        [  2.0870,  -4.6877,   9.6823],
        [  2.1440, -10.9811,   8.5281],
        [  6.9433,  -9.3362,  -0.3421],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.3076,   0.7852,  -6.7010],
        [ -3.5903,   7.4847,  -8.9112],
        [  3.9770, -10.6341,   8.0783],
        [  5.7963,  -1.6018,   1.8856],
        [  6.0310,  12.2795,  -6.4528],
        [  5.4732,  -2.2041,  -4.3830],
        [ 12.4013,  -1.2597,  -7.6474],
        [  5.9526,   3.2525,  -4.3821],
        [  3.9663,   6.1018,  -5.4104],
        [ -4.3963,   7.1629,   4.5263]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 1.6945e-02, 5.2811e-02,  ..., 2.6653e-01, 3.3850e-01,
         9.3418e-01],
        [1.6945e-02, 0.0000e+00, 6.9419e-02,  ..., 2.7008e-01, 3.4426e-01,
         9.2464e-01],
        [5.2811e-02, 6.9419e-02, 0.0000e+00,  ..., 2.7425e-01, 3.3764e-01,
         9.7394e-01],
        ...,
        [2.6653e-01, 2.7008e-01, 2.7425e-01,  ..., 0.0000e+00, 8.3884e-02,
         7.6907e-01],
        [3.3850e-01, 3.4426e-01, 3.3764e-01,  ..., 8.3884e-02, 0.0000e+00,
         7.8003e-01],
        [9.3418e-01, 9.2464e-01, 9.7394e-01,  ..., 7.6907e-01, 7.8003e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.2430377 , 13.38091121, 13.39372426, 13.36092624, 13.45642379,
       13.59480906, 11.62158554, 13.97779839,  7.50500064,  5.1805345 ,
        7.14837514, 10.52357479,  6.14086582, 12.04924159,  8.83341608,
        9.88310614, 12.5085995 , 12.67680439, 12.41366754,  7.28793663,
        8.98765644,  5.26342203, 11.51811603, 11.04953767,  9.60412735,
       11.46176259, 13.93790235,  8.36445758,  7.62011734,  7.37902766,
       10.8207297 ,  6.58184474,  9.88571713, 11.95743558,  5.2287058 ,
        8.16714836, 10.19036081,  9.78285424,  7.27340834, 11.00812996,
        7.5158489 ,  5.60064319, 12.11735334,  6.74872981,  6.14086582,
        9.17016857, 11.67875855, 10.04916678,  6.383897  ,  5.62005629,
       12.70489563, 11.018084  , 13.00413829, 12.35625694,  5.63824971,
        4.74770899,  5.23878371, 12.27991991, 13.20988996,  4.91038884,
        7.822854  ,  6.3949043 ,  8.37954853,  9.54154096, 10.41909211,
       13.37121923, 10.412299  ,  5.5205189 , 13.27994304, 10.0674887 ,
        7.23276582, 14.10796466, 13.05148826,  6.21275016,  7.57685908,
       13.0493368 ,  5.83560009, 13.79676654, 11.97957425,  5.57341623,
        6.83515201, 13.89886469, 10.88618114,  9.36256096, 13.12299145,
        6.01356004, 14.02526505,  9.6282763 ,  4.79003361, 13.27717982,
        9.61942445, 13.94008948,  5.13125513, 11.18839169, 11.63502679,
        8.5724042 ,  9.34061938,  8.30125201, 11.35345345,  6.01356004,
       13.68063098,  5.90035324, 12.46513646,  6.78325151,  7.27759233,
        8.40445872])}
{'P': tensor([[2.2122e-03, 2.4331e-01, 1.8171e-01,  ..., 6.7611e-08, 3.7205e-04,
         1.1167e-01],
        [9.7288e-06, 3.9069e-03, 3.9614e-01,  ..., 1.6112e-07, 9.9011e-04,
         1.8938e-03],
        [4.3207e-06, 1.1103e-05, 3.6143e-03,  ..., 1.7180e-07, 1.1671e-03,
         1.6931e-05],
        ...,
        [9.1206e-07, 7.4413e-07, 5.6099e-07,  ..., 3.7844e-02, 4.7950e-06,
         6.0759e-07],
        [3.4605e-05, 4.6697e-05, 6.1739e-05,  ..., 7.8616e-04, 1.5552e-01,
         4.5037e-05],
        [2.5468e-05, 3.5363e-03, 3.1223e-01,  ..., 1.8055e-07, 1.1784e-03,
         5.5063e-03]]), 'X': tensor([[ 10.2712,   8.3707,  -4.9743],
        [  9.3607,   9.5337,  -4.5412],
        [  8.4691,  10.4571,  -5.0650],
        [  9.3836,   9.5075,  -5.9278],
        [  7.0171,  11.6439,  -5.0178],
        [  9.6816,   9.1536,  -5.8543],
        [  1.9505,  -4.7171,   9.9868],
        [  7.7794,  11.0629,   8.9069],
        [  6.1575,  12.2046,   3.3257],
        [ -0.7599, -10.8995,  -6.0366],
        [ -8.3062,  -5.6008,   0.0287],
        [  0.2571,  -4.7594,   1.5618],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -9.3419,   0.5972,  -1.2841],
        [ -8.8631,  -4.3728,  -1.1746],
        [  9.5203,   9.3483,   5.8781],
        [  7.0664,  -9.2590,   8.0879],
        [ 12.0845,  -2.5772,  -4.6073],
        [  5.1203,  -2.7151,   6.2110],
        [  6.0788,  -0.8710,  -6.8719],
        [  3.3094,  13.4969,  -3.6507],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.1222,  -0.7261,   8.8204],
        [  3.4920,  -4.1366,  -8.3955],
        [  2.6970,   6.9826,   9.7350],
        [ 11.9808,   4.9340,  -3.3715],
        [  4.5503,   5.5359,   7.4735],
        [  2.3261,   7.1726,   6.2522],
        [  1.9503,  -4.7172,  -6.2313],
        [ 12.3357,   3.6001,   0.7903],
        [  6.9314,  -9.3435,  -1.2729],
        [  3.8111, -10.6787,  -4.6539],
        [  2.3376,  -4.6231,   9.8436],
        [ -2.2508, -10.5479,  -2.7235],
        [  5.4426, -10.1181,  -5.1268],
        [ -5.9225,  -8.5355,   2.4778],
        [  2.7295,  13.6703,  -8.9328],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -9.4430,  -0.1897,   4.8383],
        [ 12.1588,   4.3364,  -8.0867],
        [  5.6248,   3.9984,   8.4933],
        [ -8.4573,   3.2448,   5.4370],
        [  4.9152,  12.8653,  -1.4459],
        [ -9.4366,  -0.1189,  -2.6736],
        [ 11.4743,   6.2478,  -2.1100],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -6.7643,  -7.7372,  -9.6663],
        [  0.4996,   7.7738,  -4.4711],
        [ -3.2736, -10.1703,   3.8712],
        [ 10.1503,   8.5414,  -7.2681],
        [ -5.0132,  -9.2234,  -8.3531],
        [  2.0729,   7.2878,   7.0880],
        [  1.1644, -11.0394,   7.8511],
        [  0.1206,  -4.7363,   8.7103],
        [  2.5829,  -4.5466,  -8.4015],
        [ -7.8359,   4.2735,   7.2943],
        [  4.7813,  -3.1139,   8.2691],
        [  2.2860,   7.1916,  -6.1011],
        [  6.3589,   0.8239,  -8.4230],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.0054,  -1.0902,   8.3079],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -7.4602,   4.7718,  -4.6277],
        [ -0.0254, -10.9932,   3.2200],
        [ 10.1798,   8.5002,  -5.8893],
        [  9.6816,   9.1536,  -5.8543],
        [ -5.6169,  -8.7844,   8.9057],
        [ -9.4117,   0.1160,   8.2574],
        [  6.2555,   2.1474,   4.1685],
        [  7.5930,  -8.9037,   7.2323],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.3211,   1.7017,  -0.2164],
        [ -8.5900,   2.9767,   7.4811],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.4877,  -0.7222,   8.8347],
        [ -9.4645,  -1.4880,  -2.0226],
        [  3.9085,   6.1512,  -2.2560],
        [  6.2790,  -0.0259,  -5.4116],
        [ -8.3653,   3.4184,   9.3454],
        [ -1.0012, -10.8576,   2.2867],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.4716,  -0.6673,   7.6068],
        [  0.9809,  -4.8145,   6.8706],
        [ -9.4541,  -0.3327,   8.8246],
        [  0.2480, -11.0153,  -3.0183],
        [  6.2365,  -0.2578,  -6.0212],
        [  5.1787,  -2.6383,  -4.2533],
        [  7.5064,  -8.9651,  -0.0994],
        [ -5.1377,   6.7751,  -0.4438],
        [  4.2041,  -3.6509,  -4.9488],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.5996,   1.3971,  -1.0694],
        [ -9.1624,  -3.4286,   2.0647],
        [  3.9515,  -3.8425,   6.1729],
        [  5.1793,   4.7461,   0.3899],
        [ -9.4735,  -1.2620,  -3.6742],
        [  5.1793,   4.7461,   0.3899],
        [ -9.4671,  -0.5553,   0.1117],
        [  0.9809,  -4.8145,   6.8706],
        [ 10.4994,  -5.9193,  -7.5153],
        [  0.1252, -11.0062,   7.2564],
        [  6.7565,  11.8237,  -4.9882],
        [ -8.1204,  -5.9308,   4.0761],
        [ -1.7290,   7.8825,  -3.3831],
        [  9.3836,   9.5075,  -5.9278]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9882e-02, 1.7434e-02,  ..., 1.4058e+00, 5.8649e-01,
         6.0580e-02],
        [2.9882e-02, 0.0000e+00, 3.9362e-02,  ..., 1.3932e+00, 5.8017e-01,
         8.8887e-02],
        [1.7434e-02, 3.9362e-02, 0.0000e+00,  ..., 1.4232e+00, 6.0382e-01,
         5.0451e-02],
        ...,
        [1.4058e+00, 1.3932e+00, 1.4232e+00,  ..., 3.0339e-04, 8.4083e-01,
         1.4517e+00],
        [5.8649e-01, 5.8017e-01, 6.0382e-01,  ..., 8.4083e-01, 0.0000e+00,
         6.2436e-01],
        [6.0580e-02, 8.8887e-02, 5.0451e-02,  ..., 1.4517e+00, 6.2436e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.25017211, 13.36092624, 13.45642379, 13.35832725, 13.59480906,
       13.32374056,  5.10446653, 13.52430176, 13.66991027, 10.92597196,
       10.01804223,  4.7663526 , 12.35625694,  9.36093562,  9.88310614,
       13.3426541 , 11.64746882, 12.35625694,  5.79562691,  6.14086582,
       13.89671314,  5.83560009,  6.16512856,  5.41348121,  7.48538905,
       12.95702647,  7.16598273,  7.5403822 ,  5.10444097, 12.85032836,
       11.63383221, 11.33837396,  5.1805345 , 10.78533838, 11.4890861 ,
       10.38897563, 13.94008948,  9.48969978,  9.44486609, 12.9089559 ,
        6.90115901,  9.05843201, 13.77222752,  9.43737523, 13.06498554,
       12.04924159, 10.27716434,  7.78980299, 10.68416401, 13.26590071,
       10.4977132 ,  7.57685908, 11.10066126,  4.73784222,  5.22903117,
        8.92549752,  5.70591986,  7.54620849,  6.41202986, 10.31886066,
        6.10360073, 10.52357479,  8.85572292, 10.99326577, 13.26209545,
       13.32374056, 10.42667093,  9.41244865,  6.61386436, 11.70168755,
        8.22728399,  6.54615887,  9.09119351, 11.52733581, 12.5085995 ,
        9.58072286,  7.28793663,  6.27905292,  9.0368458 , 10.9036226 ,
        9.62238828,  9.49511524,  4.91336994,  9.45995955, 11.018084  ,
        6.24187044,  5.81199422, 11.69264243,  8.50278896,  5.56809335,
        9.79039371,  8.41171369, 12.67680439,  9.78285424,  5.51177501,
        7.02496223,  9.55720651,  7.02496223,  9.48337063,  4.91336994,
       12.05300943, 11.00694688, 13.61801526, 10.05558893,  8.06990483,
       13.35832725])}
{'P': tensor([[2.9423e-03, 3.3938e-01, 1.7729e-01,  ..., 6.5498e-08, 8.6970e-05,
         2.0849e-05],
        [1.2578e-05, 5.0645e-03, 3.7174e-01,  ..., 8.3719e-08, 2.4080e-04,
         1.9960e-05],
        [1.5372e-05, 3.7270e-05, 1.6064e-02,  ..., 1.8265e-07, 1.1358e-03,
         3.8835e-05],
        ...,
        [4.5688e-03, 2.4927e-03, 1.0315e-03,  ..., 5.3882e-02, 2.1328e-05,
         1.2026e-03],
        [5.2630e-06, 6.4080e-06, 8.1878e-06,  ..., 7.5769e-06, 7.4453e-02,
         1.9484e-05],
        [2.5448e-03, 2.7828e-03, 4.6451e-03,  ..., 2.4648e-05, 4.4206e-04,
         8.5358e-02]]), 'X': tensor([[ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-8.9942e+00, -3.9990e+00,  6.0448e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [-8.6754e-01, -1.0881e+01,  6.8561e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-9.3331e+00, -2.6465e+00, -8.2028e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [-2.8155e-02,  7.8575e+00, -1.1746e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0291, 0.0744,  ..., 0.4520, 0.7740, 0.4794],
        [0.0291, 0.0000, 0.1006,  ..., 0.4547, 0.7716, 0.5074],
        [0.0744, 0.1006, 0.0000,  ..., 0.4902, 0.8203, 0.4316],
        ...,
        [0.4520, 0.4547, 0.4902,  ..., 0.0000, 0.3363, 0.5372],
        [0.7740, 0.7716, 0.8203,  ..., 0.3363, 0.0000, 0.8349],
        [0.4794, 0.5074, 0.4316,  ..., 0.5372, 0.8349, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.25336001, 13.3607379 , 13.50484539, 13.32645552, 13.55868684,
       13.31764181,  6.16512856,  6.19633139, 13.1471313 ,  6.81084724,
       10.51437947, 12.53047489,  8.40445872, 10.67322527,  7.37902766,
        9.09661212,  8.35856487, 12.45418236,  6.84796202, 11.89855267,
        9.24501829,  6.00663314,  6.27715105,  7.17789666,  6.39187285,
       11.11969815,  9.50913849,  9.84314712, 13.89886469, 10.91601655,
        5.47702766,  5.84533345,  9.09119351,  6.89245352,  9.70108409,
       11.785566  ,  9.36256096,  8.09899874, 13.14777433,  6.00663314,
        7.8575648 , 11.00812996,  6.38399789,  6.61211224, 10.08001894,
       12.46513646,  8.95983586, 11.69264243,  5.77219077, 12.12641969,
        7.54620849,  9.43737523,  6.32572323, 12.12448912,  8.90405823,
       14.08549802, 13.04156075, 12.4217523 ,  8.62631801,  6.27715105,
        5.05989547, 13.70893492, 12.4013153 ,  8.91551231,  7.20482224,
       10.48840476,  5.39723241,  5.5205189 , 14.03505624,  7.20826219,
        5.57341623,  9.86089579,  5.94909709, 11.85890664,  9.53015376,
        5.14607014,  7.78980299, 11.22142195,  9.87720102,  7.67939546,
        8.09842704, 11.63765273, 13.06957425, 13.12299145,  5.36061974,
        9.44486609, 13.62896593, 10.51437947, 13.91198191,  6.35029274,
       11.81423161,  7.54620849, 11.1913598 , 13.39372426, 10.7167285 ,
       11.28215429, 13.00413829,  6.42439344, 11.27160035, 11.67875855,
        7.28793663,  9.97964013, 13.91198191, 11.92133129,  8.44940661,
        6.77777606])}
{'P': tensor([[1.8004e-03, 3.3384e-01, 2.7651e-01,  ..., 2.5877e-06, 2.1899e-07,
         2.0655e-08],
        [6.4651e-06, 2.3947e-03, 4.9411e-01,  ..., 2.7018e-06, 3.4344e-07,
         2.4864e-08],
        [1.1825e-05, 1.7488e-05, 5.3518e-03,  ..., 5.3286e-06, 8.4371e-07,
         5.2669e-08],
        ...,
        [2.6619e-02, 2.4161e-02, 2.3144e-02,  ..., 1.9552e-01, 1.0091e-04,
         1.6337e-04],
        [1.1559e-06, 1.0171e-06, 9.4915e-07,  ..., 7.0386e-05, 3.3697e-02,
         1.0005e-03],
        [2.4402e-04, 1.7496e-04, 1.4467e-04,  ..., 2.7957e-03, 2.1724e-06,
         3.2357e-02]]), 'X': tensor([[ 10.1798,   8.5002,  -5.8893],
        [  9.6816,   9.1536,  -5.8543],
        [  9.3836,   9.5075,  -5.9278],
        [  8.4691,  10.4571,  -5.0650],
        [  8.9284,  10.0036,  -6.9210],
        [  7.0171,  11.6439,  -5.0178],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 12.4013,  -1.2597,  -7.6474],
        [  0.9659,  -4.8144,  -0.9594],
        [  4.8653,   5.1694,  -4.0823],
        [  3.2990,  -4.2415,  -7.4857],
        [ -1.5408,   7.8978,   5.3470],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -5.2752,  -9.0409,  -8.2568],
        [  0.3493,  -4.7727,  -6.5682],
        [  2.4639, -10.9431,  -7.3718],
        [ -4.7334,   6.9982,  -0.4392],
        [  5.9883,   3.1536,   6.2320],
        [ -7.5401,   4.6718,   9.0033],
        [ -6.5295,  -7.9785,  -7.4429],
        [  4.7813,  -3.1139,   8.2691],
        [  2.4149,  -4.6005,  -4.3339],
        [  8.1528,  10.7452,   9.1493],
        [  5.0190,   4.9705,  -7.9752],
        [  8.7715,  -7.9390,   5.2900],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.3474,  -4.2162,   5.2455],
        [  0.9945,  -4.8144,  -0.1897],
        [ -2.5056,   7.7720,   5.5878],
        [ -3.0644,   7.6439,  -1.4322],
        [  6.9433,  -9.3362,  -0.3421],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.5158,  -2.1337,   3.5246],
        [ -3.2053, -10.1993,  -3.8736],
        [ 10.5678,   7.9261,  -0.6100],
        [  5.6202,  -1.9508,   0.7471],
        [ -8.6603,   2.8250,   5.9181],
        [  9.8082,  -6.8396,   4.9615],
        [ -5.0550,   6.8231,  -9.2502],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -1.0482, -10.8487,  -5.9066],
        [ -0.5385,   7.9037,   6.3830],
        [  1.5383, -11.0274,  -4.9666],
        [ -9.1658,  -3.4155,  -2.8880],
        [  9.6587,   9.1817,  -0.8212],
        [  4.2612, -10.5514,  -6.6268],
        [ 10.9875,   7.2207,  -8.8549],
        [  9.5261,  -7.1675,   1.4206],
        [ -5.3736,   6.6310,  -1.5174],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.3860,  -2.3418,   1.0826],
        [ -1.6086,   7.8928,   4.3385],
        [  4.1537,   5.9340,  -6.0423],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.4773,  -1.0024,   5.2599],
        [  0.1382, -11.0073,   1.0026],
        [  2.3261,  13.7749,   5.9121],
        [  7.0515,  11.6194,   8.4289],
        [  0.3038,   7.8092,   0.4164],
        [ -1.0407,   7.9166,  -6.6021],
        [  1.0391,  -4.8141,  -0.0768],
        [  5.2199,   4.6861,  -1.9304],
        [  7.0171,  11.6439,  -5.0178],
        [ 12.5919,   0.4035,  -5.8460],
        [  1.1274,  14.0116,   8.1753],
        [  0.3231,  -4.7691,  -1.6120],
        [  4.1999,  -3.6543,  -4.1998],
        [  1.2273,  -4.8081,   9.8129],
        [  3.1271, -10.8343,   2.6442],
        [ 11.2209,  -4.7092,   2.6117],
        [  2.8602,   6.8905,   5.8779],
        [  4.5325,  -3.3643,   6.7699],
        [ 12.4877,  -0.7222,   8.8347],
        [  0.7145,  -4.8071,  -2.3766],
        [  0.3715,  -4.7756,   0.2312],
        [  8.3791,  10.5409,   5.5590],
        [  4.5767,   5.5071,   3.9069],
        [  0.5600, -11.0321,  -0.8660],
        [  6.2883,   0.0325,   5.2692],
        [  0.9074,  -4.8139,   8.1674],
        [ -5.5525,   6.5146,  -4.4984],
        [  0.3724,   7.7974,  -6.8722],
        [  3.9913,  -3.8139,  -6.8311],
        [ 10.0619,   8.6627,  -9.2884],
        [  3.9515,  -3.8425,   6.1729],
        [  9.0051,   9.9234,   5.4411],
        [ -7.7239,   4.4298,   3.1216],
        [  2.2076,  13.8032,  -1.5099],
        [ 11.3367,  -4.4806,   0.4643],
        [ -9.4735,  -0.7274,  -2.6749],
        [  2.4149,  -4.6005,  -4.3339],
        [ 11.0864,   7.0387,  -0.1238],
        [ -3.4100,   7.5436,  -7.6980],
        [  6.1222,  -0.7261,   8.8204],
        [  5.1793,   4.7461,   0.3899],
        [ -2.2508, -10.5479,  -2.7235],
        [  9.3607,   9.5337,  -4.5412],
        [  4.5057,  -3.3894,  -0.8160],
        [ 12.5763,   0.1333,  -9.2650],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.8890,  -4.4316,   1.1600],
        [  3.9085,   6.1512,  -2.2560],
        [ -7.5603,   4.6459,   8.2002],
        [  6.2535,  -0.1709,  -9.4102],
        [ -8.9624,  -4.0944,  -4.9622],
        [  5.3924, -10.1396,  -4.0863]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0069, 0.0086,  ..., 0.2575, 1.2890, 0.6917],
        [0.0069, 0.0000, 0.0024,  ..., 0.2643, 1.2959, 0.6984],
        [0.0086, 0.0024, 0.0000,  ..., 0.2661, 1.2975, 0.7002],
        ...,
        [0.2575, 0.2643, 0.2661,  ..., 0.0000, 1.0342, 0.4345],
        [1.2890, 1.2959, 1.2975,  ..., 1.0342, 0.0000, 0.6192],
        [0.6917, 0.6984, 0.7002,  ..., 0.4345, 0.6192, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.26209545, 13.32374056, 13.35832725, 13.45642379, 13.40850328,
       13.59480906, 12.4980845 , 12.46513646,  4.91038884,  7.09887557,
        5.37343173,  8.04664886, 10.00410625, 10.46739073,  4.78544875,
       11.21703654,  8.4486771 ,  6.76791483,  8.87007495, 10.30972193,
        5.70591986,  5.19577146, 13.48809595,  7.06372172, 11.83075005,
        5.5366698 ,  5.38342077,  4.9160962 ,  8.16584537,  8.23525433,
       11.63502679,  7.88677538,  5.91408504, 10.6910768 , 13.20988996,
        5.94909709,  9.10945678, 11.95743558,  8.49161154, 10.06437598,
       10.8992524 ,  7.92200342, 11.1341798 ,  9.78147542, 13.32645552,
       11.37939695, 13.14777433, 11.92133129,  8.53501777,  5.68942572,
        5.87305005,  8.0550286 ,  7.24326992, 11.80092578,  9.53015376,
       11.00812996, 13.96987806, 13.59170809,  7.81509631,  7.98469166,
        4.92497514,  7.01475376, 13.59480906, 12.59840193, 14.05687815,
        4.78002665,  5.5671389 ,  4.96231532, 11.27656488, 12.16901615,
        7.46053281,  5.64464807, 12.5085995 ,  4.85995002,  4.79003361,
       13.46553623,  7.16058587, 11.04629471,  6.28835101,  4.89869205,
        8.55982105,  7.80625981,  5.5205189 , 13.27717982,  5.51177501,
       13.40024224,  8.90405823, 13.97857591, 12.18998195,  9.5014146 ,
        5.19577146, 13.13205357,  8.27853076,  6.16512856,  7.02496223,
       10.78533838, 13.36092624,  5.63824971, 12.5769659 ,  8.23525433,
        5.29009836,  7.28793663,  8.8737556 ,  6.25586297,  9.85330646,
       11.48434767])}
{'P': tensor([[3.4295e-03, 3.6794e-01, 1.3614e-01,  ..., 1.9505e-08, 4.9546e-07,
         5.3120e-07],
        [9.4963e-06, 5.1984e-03, 1.4336e-02,  ..., 3.0151e-08, 1.2671e-06,
         5.9668e-07],
        [2.3186e-05, 7.8508e-04, 1.0020e-02,  ..., 4.5371e-08, 1.6274e-06,
         6.3517e-07],
        ...,
        [1.4298e-05, 8.3987e-06, 5.0196e-06,  ..., 4.8722e-02, 3.5944e-06,
         1.5628e-03],
        [3.4248e-07, 3.6649e-07, 2.5559e-07,  ..., 6.5355e-04, 6.5426e-02,
         7.6325e-05],
        [5.2294e-04, 5.5962e-04, 2.6748e-04,  ..., 9.7641e-05, 4.8590e-04,
         2.3010e-01]]), 'X': tensor([[ 10.1503,   8.5414,  -7.2681],
        [  8.9284,  10.0036,  -6.9210],
        [  8.5198,  10.4091,  -8.6486],
        [  6.9956,  11.6590,  -7.6876],
        [  8.8474,  10.0868,  -8.8644],
        [  6.1685,  12.1979,  -6.2473],
        [ -2.6485, -10.4151,   0.8766],
        [ -3.5520,   7.4976,   9.1522],
        [  2.1771, -10.9776,  -3.2007],
        [  6.9653,  -9.3225,  -3.4727],
        [ -8.3062,  -5.6008,   0.0287],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 12.2218,   4.0953,  -8.2072],
        [  9.5018,  -7.1945,  -1.1892],
        [ -8.9624,  -4.0944,  -4.9622],
        [  3.1480,  -4.3165,  -7.8002],
        [  5.6523,   3.9443,  -2.9449],
        [  4.1348,   5.9514,  -0.8856],
        [  6.0140,  -1.0661,  -2.5040],
        [  4.3459,  -3.5327,   0.3667],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.3551,   0.7149,   1.3344],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -8.0087,   4.0173,   9.1091],
        [  4.3523, -10.5232,   8.2138],
        [  4.4633,   5.6287,  -0.7754],
        [  9.0424,  -7.6779,  -7.5286],
        [  0.1497,   7.8334,  -5.4412],
        [ 12.5996,   1.3971,  -1.0694],
        [  5.1142,  -2.7230,  -4.1966],
        [ -9.4630,  -1.5145,   0.2983],
        [ -9.1706,   1.3863,   8.4032],
        [  3.9591, -10.6391,  -6.2557],
        [  3.8182,  -3.9347,  -1.7961],
        [  4.9957,  -2.8705,   6.5978],
        [  3.3069,   6.6089,   8.7377],
        [  7.0515,  11.6194,   8.4289],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 11.2517,  -4.6495,   2.9800],
        [  1.3403,  -4.8011,  -1.1698],
        [  5.5294,  -2.1106,   9.0626],
        [  0.6952,  -4.8060,   8.8976],
        [ -7.6650,   4.5092,   0.2520],
        [  5.3858,   4.4260,  -9.6769],
        [ -9.4501,  -1.7133,   2.2980],
        [ -9.2345,  -3.1345,   3.9715],
        [  2.1504,  -4.6726,   8.7908],
        [  0.9809,   7.6642,  -1.4217],
        [  2.0870,  -4.6877,   9.6823],
        [ -2.5161,   7.7699,  -9.3531],
        [ -4.3359,  -9.6431,  -9.0616],
        [ -9.4296,  -1.9458,   4.0833],
        [ 12.2692,  -1.8882,   3.9253],
        [  0.6470, -11.0352,  -1.1931],
        [  0.6470, -11.0352,  -1.1931],
        [ -3.3116, -10.1539,   7.5919],
        [  5.0348,  -2.8229,   3.5436],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -7.3333,   4.9246,  -7.9212],
        [  7.4132,  11.3526,   0.3322],
        [ 11.1420,   6.9331,  -2.4779],
        [  0.2571,  -4.7594,   1.5618],
        [  2.1477,   7.2550,   8.9688],
        [  4.7482, -10.3907,   5.9149],
        [  6.8093,  -9.4178,   8.4244],
        [  3.2647, -10.8066,  -0.3768],
        [  5.9526,   3.2525,  -4.3821],
        [  0.2571,  -4.7594,   1.5618],
        [  5.5158,  -2.1337,   3.5246],
        [ -5.6169,  -8.7844,   8.9057],
        [  4.5340,  -3.3629,   8.9652],
        [ 11.9122,   5.1397,  -1.5498],
        [  6.1755,   2.5233,  -2.0042],
        [ 12.5224,  -0.4497,   4.8448],
        [ -8.6675,  -4.8569,  -9.1769],
        [  8.8474,  10.0868,  -8.8644],
        [  5.1793,   4.7461,   0.3899],
        [ 11.0373,   7.1299,  -8.5895],
        [ 12.5258,  -0.4200,  -9.3493],
        [  5.3118,   4.5452,   4.2114],
        [  5.1787,  -2.6383,  -4.2533],
        [  6.2895,   0.0405,   5.7652],
        [  3.6249, -10.7255,   5.6605],
        [  1.8766,   7.3694,   1.1887],
        [ -1.6086,   7.8928,   4.3385],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.4697,  -0.6157,  -6.8472],
        [  4.9452, -10.3187,  -7.8621],
        [  6.3587,   1.1596,  -3.6629],
        [ -7.9298,   4.1367,  -3.9773],
        [  1.6308,  13.9254,  -2.0984],
        [  2.4356,  -4.5942,  -5.0808],
        [ -8.1905,   3.7253,   4.1175],
        [ -8.6801,   2.7809,  -0.8443],
        [  0.3231,  -4.7691,  -1.6120],
        [ -5.7347,  -8.6908,   9.2878],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.4743,   6.2478,  -2.1100],
        [ -7.6734,   4.4980,  -5.4248],
        [  9.3607,   9.5337,  -4.5412],
        [ -1.0012, -10.8576,   2.2867],
        [  0.1041, -11.0045,   2.4311],
        [ -8.9178,   2.1936,   7.5388],
        [  6.2242,  -0.3171,   7.4978]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.8675e-02, 6.5554e-02,  ..., 1.2244e+00, 1.4975e+00,
         1.0668e+00],
        [1.8675e-02, 0.0000e+00, 8.2397e-02,  ..., 1.2169e+00, 1.4881e+00,
         1.0504e+00],
        [6.5554e-02, 8.2397e-02, 0.0000e+00,  ..., 1.2771e+00, 1.5543e+00,
         1.1324e+00],
        ...,
        [1.2244e+00, 1.2169e+00, 1.2771e+00,  ..., 2.1453e-04, 3.1510e-01,
         7.5301e-01],
        [1.4975e+00, 1.4881e+00, 1.5543e+00,  ..., 3.1510e-01, 0.0000e+00,
         8.0003e-01],
        [1.0668e+00, 1.0504e+00, 1.1324e+00,  ..., 7.5301e-01, 8.0003e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.26590071, 13.40850328, 13.45124917, 13.59673406, 13.41713661,
       13.66896739, 10.74655346,  8.29640613, 11.1913598 , 11.63724792,
       10.01804223, 10.51437947, 12.889695  , 11.9182956 ,  9.85330646,
        5.34250225,  6.89245352,  7.24677664,  6.10774544,  5.60064319,
       10.53368644,  6.39520397,  6.80559669, 12.12448912,  8.95983586,
       11.38774149,  7.18354885, 11.86239961,  7.83486943, 12.67680439,
        5.79391825,  9.58347456,  9.27474386, 11.35182589,  5.48276038,
        5.76165934,  7.39004226, 13.59170809, 10.1979669 , 12.17451685,
        4.98461983,  5.91854397,  4.85603886,  8.89301516,  6.97106912,
        9.60412735,  9.75200848,  5.14371005,  7.72668467,  5.13125513,
        8.16714836, 10.57302546,  9.6282763 , 12.41366754, 11.05413545,
       11.05413545, 10.68031351,  5.77219077, 12.35625694,  8.83341608,
       13.55868684, 13.12299145,  4.7663526 ,  7.56617037, 11.42420508,
       11.62158554, 11.28895489,  6.78325151,  4.7663526 ,  5.91408504,
       10.42667093,  5.64500512, 12.97370371,  6.67108662, 12.53047489,
        9.93553424, 13.41713661,  7.02496223, 13.1399181 , 12.53285633,
        6.99097013,  5.81199422,  6.28961988, 11.32149259,  7.60463349,
        8.0550286 , 12.10942412, 13.66896739,  9.48969978, 11.4424816 ,
        6.46356806,  8.94395628, 14.02058621,  5.19986797,  8.99791316,
        9.11473391,  4.78002665, 10.412299  ,  5.1224079 , 13.06498554,
        8.89458273, 13.36092624, 10.9036226 , 11.0050379 ,  9.18358562,
        6.2322769 ])}
{'P': tensor([[2.9982e-03, 3.1046e-01, 2.3681e-01,  ..., 2.0263e-07, 3.9902e-08,
         8.6544e-07],
        [4.9980e-06, 2.6013e-03, 3.4484e-01,  ..., 9.3740e-08, 4.1269e-08,
         1.3056e-06],
        [6.7639e-06, 2.0075e-05, 4.0792e-03,  ..., 1.2937e-07, 6.8779e-08,
         2.4180e-06],
        ...,
        [1.6508e-03, 1.5914e-03, 1.6182e-03,  ..., 5.3273e-02, 8.6464e-07,
         7.6222e-06],
        [6.9288e-06, 4.3512e-06, 3.9474e-06,  ..., 1.7433e-04, 5.6814e-02,
         9.8448e-06],
        [4.3560e-07, 4.5245e-07, 4.6818e-07,  ..., 3.5838e-06, 3.0793e-03,
         4.5558e-02]]), 'X': tensor([[ 10.0619,   8.6627,  -9.2884],
        [  8.8474,  10.0868,  -8.8644],
        [  8.5198,  10.4091,  -8.6486],
        [  7.0642,  11.6104,  -9.8316],
        [  6.2675,  12.1379,  -8.9720],
        [  6.9956,  11.6590,  -7.6876],
        [ 10.1503,   8.5414,  -7.2681],
        [ 10.8670,  -5.3428,  -2.8637],
        [  6.3460,   1.4252,   1.2291],
        [  0.9392,  -4.8143,  -3.4649],
        [  5.6523,   3.9443,  -2.9449],
        [  7.0642,  11.6104,  -9.8316],
        [ -5.1398,  -9.1367,   6.2074],
        [ -9.3739,   0.3982,  -4.2564],
        [ -9.4159,  -2.0739,   9.6841],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.0547,   7.2957,  -1.9343],
        [ -8.5149,  -5.1905,   1.5239],
        [ -7.2515,   5.0195,   4.6434],
        [ -9.4766,  -1.1116,  -2.1706],
        [  1.4749,  -4.7892,  -0.8089],
        [ 10.4994,  -5.9193,  -7.5153],
        [  1.9204,   7.3518,   8.4781],
        [  1.1274,  14.0116,   8.1753],
        [ -5.9938,  -8.4745,   3.9735],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.8934,   2.2597,   3.5831],
        [  3.6469,   6.3624,  -1.8414],
        [  8.9284,  10.0036,  -6.9210],
        [  5.7963,  -1.6018,   1.8856],
        [  4.3523, -10.5232,   8.2138],
        [ 12.5770,   1.8512,  -0.6578],
        [  4.0587, -10.6112,   6.2273],
        [  2.7121,   6.9743,  -8.4202],
        [  3.6154,  -4.0640,  -7.9698],
        [  9.7328,   9.0902,  -0.8917],
        [  4.2275,  -3.6320,  -0.2714],
        [  1.9241,   7.3503,   7.9004],
        [  5.1519, -10.2387,   9.8191],
        [  3.4401,  13.4540,   4.6576],
        [ -2.6485, -10.4151,   0.8766],
        [  1.0000,   7.6591,  -4.5011],
        [  6.1755,   2.5233,  -2.0042],
        [  0.3724,   7.7974,  -6.8722],
        [  7.3592,  11.3937,   6.2407],
        [ 12.5154,   2.5072,  -9.3274],
        [  4.0642,  -3.7599,  -3.4565],
        [  6.1973,  -0.4372,   6.5720],
        [  0.6933,   7.7336,   2.1179],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.4378,  -1.8606,  -2.5668],
        [  7.3592,  11.3937,   6.2407],
        [  1.3103,  -4.8032,  -0.9300],
        [ -5.6651,   6.4380,   1.3660],
        [  6.1984,  -0.4327,   1.5438],
        [  5.2608,  -2.5256,  -9.1446],
        [  0.2480, -11.0153,  -3.0183],
        [ 11.3118,   6.5957,  -4.5667],
        [  6.3515,   0.6418,  -6.5823],
        [  4.5814, -10.4485,   2.5496],
        [ -1.0407,   7.9166,  -6.6021],
        [  4.3459,  -3.5327,   0.3667],
        [  5.7218,  -9.9934,   6.5764],
        [  6.3516,   0.6425,   0.3425],
        [  1.3403,  -4.8011,  -1.1698],
        [  6.1741,  -0.5329,   3.1913],
        [ -3.9700,   7.3453,  -4.1486],
        [  0.1287, -11.0065,   6.5757],
        [ -9.3419,   0.5972,  -1.2841],
        [ 12.2692,  -1.8882,   3.9253],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -3.3812, -10.1235,   7.4384],
        [  6.3124,   0.2057,  -0.4593],
        [  9.6662,   9.1725,   8.1056],
        [  9.0129,  -7.7072,   9.1400],
        [  2.5318,  13.7231,   6.3692],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.4748,  -0.7765,  -5.6695],
        [  5.3804,  -2.3504,   7.2744],
        [  2.5128, -10.9365,   9.9645],
        [ 11.8951,  -3.1570,   7.0742],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.5456,   6.0861,   8.1101],
        [  3.3167,   6.6022,   6.3554],
        [ 12.6000,   1.3839,  -3.2992],
        [  2.6613,   7.0021,   4.5783],
        [  0.7958,  -4.8109,  -8.2248],
        [ -4.5778,  -9.5013,  -5.0667],
        [  3.0876,   6.7527,  -5.3540],
        [ -5.5525,   6.5146,  -4.4984],
        [  9.0129,  -7.7072,   9.1400],
        [  2.7590,  13.6621,   9.4496],
        [ -5.1974,   6.7396,   3.3484],
        [ -5.0943,  -9.1682,   0.5073],
        [  1.7122,  -4.7591,  -3.6734],
        [ -7.6525,  -6.6534,   5.1721],
        [  6.3606,   0.9204,   3.2525],
        [ -9.1960,  -3.2964,   1.6921],
        [  1.2728, -11.0372,   9.8121],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -8.5859,  -5.0393,   0.5476],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.5199,  -0.4711,   3.0436],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -9.1178,   1.5784,   2.1410]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.4106e-03, 1.7130e-02,  ..., 9.1800e-01, 1.2137e+00,
         1.3624e+00],
        [8.4106e-03, 0.0000e+00, 8.7275e-03,  ..., 9.0990e-01, 1.2076e+00,
         1.3548e+00],
        [1.7130e-02, 8.7275e-03, 0.0000e+00,  ..., 9.0164e-01, 1.2010e+00,
         1.3467e+00],
        ...,
        [9.1800e-01, 9.0990e-01, 9.0164e-01,  ..., 0.0000e+00, 1.0805e+00,
         8.9585e-01],
        [1.2137e+00, 1.2076e+00, 1.2010e+00,  ..., 1.0805e+00, 0.0000e+00,
         4.3464e-01],
        [1.3624e+00, 1.3548e+00, 1.3467e+00,  ..., 8.9585e-01, 4.3464e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.27717982, 13.41713661, 13.45124917, 13.59056613, 13.66051884,
       13.59673406, 13.26590071, 12.10942412,  6.50409372,  4.90505566,
        6.89245352, 13.59056613, 10.48315436,  9.38232121,  9.64157461,
        9.53015376,  7.57946459,  9.97220799,  8.81930525,  9.54154096,
        5.01114719, 12.05300943,  7.5984809 , 14.05687815, 10.37998078,
        9.84314712,  9.17595899,  7.33350642, 13.40850328,  6.01356004,
       11.38774149, 12.7125079 , 11.36089355,  7.48311002,  5.43943791,
       13.31764181,  5.57341623,  7.59795445, 11.46176259, 13.88683436,
       10.74655346,  7.72415048,  6.67108662,  7.80625981, 13.56367857,
       12.76408562,  5.5366698 ,  6.21275016,  7.76457095,  9.17595899,
        9.61942445, 13.56367857,  4.978704  ,  8.57560275,  6.2134961 ,
        5.83560009, 11.018084  , 13.09426142,  6.383897  , 11.4088043 ,
        7.98469166,  5.60064319, 11.51557166,  6.38399789,  4.98461983,
        6.19707944,  8.34949383, 11.00726483,  9.36093562, 12.41366754,
       12.4013153 , 10.67322527,  6.31576264, 13.32557133, 11.85890664,
       13.95472331,  9.86089579,  9.50654967,  5.87132718, 11.22142195,
       12.30695098, 12.86648771, 13.05148826,  7.38844703, 12.67576792,
        7.49076899,  4.8763135 , 10.54658597,  7.42512621,  8.55982105,
       11.85890664, 13.93790235,  8.51088913, 10.48840476,  5.05774295,
       10.14044489,  6.42689705,  9.76896904, 11.11038551,  9.54154096,
        9.95552626,  9.70108409, 10.1371552 , 12.52875793, 10.60226614,
        9.25336371])}
{'P': tensor([[3.3781e-03, 2.3445e-01, 2.0661e-01,  ..., 2.0422e-04, 1.7126e-06,
         7.4897e-02],
        [9.7467e-06, 3.1413e-03, 5.7470e-01,  ..., 1.6041e-04, 1.2015e-06,
         6.9268e-02],
        [1.3336e-05, 2.0121e-05, 7.7655e-03,  ..., 3.9611e-04, 2.5272e-06,
         1.4296e-01],
        ...,
        [1.2656e-05, 2.0990e-05, 2.9539e-05,  ..., 3.4258e-01, 5.2387e-06,
         1.5619e-04],
        [6.2399e-03, 5.2705e-03, 4.4446e-03,  ..., 3.2882e-05, 1.1719e-01,
         1.5908e-03],
        [8.0330e-05, 2.2268e-04, 4.6286e-04,  ..., 2.7370e-02, 2.8817e-05,
         1.7070e-01]]), 'X': tensor([[ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0836, 0.1015,  ..., 1.0245, 0.2731, 0.0900],
        [0.0836, 0.0000, 0.0180,  ..., 1.1079, 0.3121, 0.1654],
        [0.1015, 0.0180, 0.0000,  ..., 1.1254, 0.3248, 0.1816],
        ...,
        [1.0245, 1.1079, 1.1254,  ..., 0.0000, 0.9876, 0.9482],
        [0.2731, 0.3121, 0.3248,  ..., 0.9876, 0.0000, 0.3092],
        [0.0900, 0.1654, 0.1816,  ..., 0.9482, 0.3092, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.2778378 , 13.40024224, 13.46553623, 13.35927527, 13.56367857,
       13.62896593, 11.48434767,  6.34059021,  9.34127013,  5.10444097,
       10.35093485,  9.32094055, 13.13205357, 12.25806559,  6.42689705,
        6.652742  , 11.42947632, 13.69162795, 11.2845608 ,  5.23878371,
        5.60064319, 10.76154256,  9.60412735,  7.73692436,  9.93553424,
       13.68063098, 12.5085995 , 10.57302546, 12.67680439,  8.10886021,
        7.16598273,  5.51177501,  6.54693987,  6.38399789,  6.54127431,
       11.72802491, 12.13568598, 11.52733581,  5.05444288, 11.24434374,
        6.10360073,  7.19534146,  6.74686071,  8.36995953, 10.84493732,
        5.04028538, 10.31584102, 12.17901033,  6.02779201,  8.06990483,
       12.10942412,  9.63742669, 12.77402351, 12.19805394,  5.47702766,
        9.562174  ,  5.56809335,  7.5984809 , 11.24434374, 10.51437947,
       11.83570132, 13.66051884,  9.09502385, 11.73567759,  5.10446653,
       12.67576792, 10.88618114,  7.70897675,  5.63824971, 11.00726483,
        5.10446653, 10.84493732, 13.79676654,  9.71392179,  6.61211224,
        6.19707944, 13.35013374,  7.72668467, 13.32557133,  7.32754882,
        5.47702766, 11.86239961,  8.22728399,  5.80860165,  4.98461983,
        9.85330646, 11.28215429,  6.84796202, 13.6660671 , 13.36800073,
        8.23525433,  6.29034958, 10.98076873, 11.63724792,  9.75200848,
       12.53345445,  5.29009836, 13.77075174,  7.78980299, 13.27994304,
       13.36800073, 10.57302546,  8.57560275, 14.09176259,  5.91854397,
       13.66991027])}
{'P': tensor([[1.4022e-03, 1.7365e-01, 1.2223e-01,  ..., 4.7103e-04, 5.4453e-07,
         6.0172e-04],
        [1.8272e-05, 5.7379e-03, 8.0793e-02,  ..., 1.3522e-03, 1.8816e-06,
         2.8430e-03],
        [2.0639e-05, 2.1769e-04, 8.4120e-03,  ..., 1.5361e-03, 2.3690e-06,
         4.7099e-03],
        ...,
        [2.6947e-04, 4.4552e-04, 6.2467e-04,  ..., 2.3239e-02, 6.7336e-05,
         8.7768e-02],
        [4.0136e-03, 3.2529e-03, 2.5206e-03,  ..., 1.0844e-02, 9.6574e-02,
         2.2696e-03],
        [3.0138e-05, 3.9960e-05, 4.8872e-05,  ..., 4.1720e-04, 7.5306e-05,
         4.6329e-02]]), 'X': tensor([[ 10.0400,   8.6922,  -6.2847],
        [  9.3836,   9.5075,  -5.9278],
        [  8.9284,  10.0036,  -6.9210],
        [  9.6816,   9.1536,  -5.8543],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  7.5930,  -8.9037,   7.2323],
        [ 11.1469,   6.9237,   1.6160],
        [ -4.8751,  -9.3148,   4.0406],
        [  9.2692,   9.6370,   7.0746],
        [  2.7976,  -4.4682,  -4.2020],
        [ -2.2188,   7.8218,   8.6064],
        [ -8.6801,   2.7809,  -0.8443],
        [ -5.6792,  -8.7352,   4.0318],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.8653,   5.1694,  -4.0823],
        [ -9.4351,  -1.8891,   3.5642],
        [  5.1196,  -2.7160,   4.2705],
        [ 11.4495,   6.3026,   6.8257],
        [ -2.0445,   7.8469,  -0.5857],
        [  1.7218,  -4.7576,  -3.3617],
        [  5.8275,  -1.5334,   2.0124],
        [  0.3724,   7.7974,  -6.8722],
        [  1.4749,  -4.7892,  -0.8089],
        [ 12.4877,  -0.7222,   8.8347],
        [ 10.1503,   8.5414,  -7.2681],
        [ 12.5123,  -0.5339,  -1.0728],
        [  3.9591, -10.6391,  -6.2557],
        [ -7.4807,   4.7463,   2.0616],
        [  6.3119,   1.7797,   9.9186],
        [ -7.3446,   4.9113,  -5.9356],
        [ -4.4779,  -9.5609,  -2.4505],
        [  5.5158,  -2.1337,   3.5246],
        [  5.7802,  -1.6363,  -6.2782],
        [ -8.6603,   2.8250,   5.9181],
        [  0.6952,  -4.8060,   8.8976],
        [ -9.4501,  -1.7133,   2.2980],
        [  9.6816,   9.1536,  -5.8543],
        [  7.0664,  -9.2590,   8.0879],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.5340,  -3.3629,   8.9652],
        [  6.0467,  -9.8369,  -6.5816],
        [ -8.0330,  -6.0765,   6.7945],
        [ -5.1398,  -9.1367,   6.2074],
        [ -6.6895,   5.6051,   2.7899],
        [ -5.9938,  -8.4745,   3.9735],
        [  7.0664,  -9.2590,   8.0879],
        [  0.9478,  -4.8144,   0.6371],
        [  0.8572,  14.0501,  -8.3364],
        [  5.8201,   3.5841,  -8.0609],
        [ 11.6544,   5.8273,  -3.1375],
        [ 11.7779,   5.5127,  -2.0699],
        [ -3.4488, -10.0934,   2.5195],
        [  6.9956,  11.6590,  -7.6876],
        [ -9.1231,  -3.5739,   8.4708],
        [ -4.7334,   6.9982,  -0.4392],
        [  0.6363, -11.0348,   6.1414],
        [  2.5469,  13.7192,  -9.6935],
        [  8.5093,  -8.1764,  -3.2381],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.6381,  -4.9238,  -4.4077],
        [  0.3354,  -4.7708,  -4.9125],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.0617,  12.7957,   8.3544],
        [  4.1297,  -3.7098,  -3.9838],
        [ -0.5385,   7.9037,   6.3830],
        [  6.2036,   2.4030,  -1.2500],
        [ -2.2188,   7.8218,   8.6064],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.5158,  -2.1337,   3.5246],
        [ -6.4620,  -8.0449,   6.2719],
        [  6.2535,  -0.1709,  -9.4102],
        [  0.9967,  -4.8144,  -9.7370],
        [ -3.6381,   7.4683,   7.3544],
        [  5.8318,   3.5567,  -4.7692],
        [  1.6408,   7.4591,  -4.2294],
        [  4.7739,  -3.1218,  -4.1188],
        [  9.3836,   9.5075,  -5.9278],
        [  7.0515,  11.6194,   8.4289],
        [  4.1637,  13.1898,  -1.9189],
        [ -9.0604,   1.7707,   1.1066],
        [  7.7794,  11.0629,   8.9069],
        [ 11.9808,   4.9340,  -3.3715],
        [  5.2608,  12.6977,   4.9166],
        [  2.5663,   7.0526,  -6.8769],
        [ -7.4807,   4.7463,   2.0616],
        [  5.9526,   3.2525,  -4.3821],
        [ -1.9487,   7.8590,  -3.3330],
        [  4.2275,  -3.6320,  -0.2714],
        [  2.7558,  -4.4844,   0.0636],
        [  9.3836,   9.5075,  -5.9278],
        [  0.7275,  14.0667,  -0.9815],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.4203,  -2.0340,   2.5856],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.2612, -10.5514,  -6.6268],
        [  0.2435,   7.8191,  -6.4709],
        [ -7.8126,   4.3066,   5.1791],
        [  9.6614,   9.1784,  -1.2087],
        [ -1.5451,   7.8975,   7.6104],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.3333,   4.9246,  -7.9212],
        [  3.9895,   6.0816,  -6.5235],
        [  4.7739,  -3.1218,  -4.1188],
        [  0.3724,   7.7974,  -6.8722]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0160, 0.0318,  ..., 0.3812, 0.3795, 0.5110],
        [0.0160, 0.0000, 0.0443,  ..., 0.3869, 0.3801, 0.5188],
        [0.0318, 0.0443, 0.0000,  ..., 0.3968, 0.4028, 0.5219],
        ...,
        [0.3812, 0.3869, 0.3968,  ..., 0.0000, 0.1223, 0.1460],
        [0.3795, 0.3801, 0.4028,  ..., 0.1223, 0.0000, 0.2452],
        [0.5110, 0.5188, 0.5219,  ..., 0.1460, 0.2452, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.27994304, 13.35832725, 13.40850328, 13.32374056, 13.45642379,
       13.59480906, 11.70168755, 13.12218013, 10.51340829, 13.37121923,
        5.27178363,  8.1303945 ,  9.11473391, 10.41909211, 10.4606265 ,
        7.09887557,  9.62238828,  5.79543038, 13.06957425,  8.10886021,
        5.05963217,  6.02588517,  7.80625981,  5.01114719, 12.5085995 ,
       13.26590071, 12.52372629, 11.35182589,  8.85939626,  6.55800267,
        8.8353674 , 10.5575601 ,  5.91408504,  6.00732193,  9.10945678,
        4.85603886,  9.60412735, 13.32374056, 11.64746882,  8.5724042 ,
        5.64500512, 11.54672755, 10.07238463, 10.48315436,  8.7273569 ,
       10.37998078, 11.64746882,  4.90676242, 14.07623266,  6.83515201,
       13.03002338, 13.00413829, 10.66631784, 13.59673406,  9.79814997,
        8.4486771 , 11.05317597, 13.95361098, 11.80092578, 11.8159254 ,
        9.94285511,  4.78258092,  7.98469166,  5.23878371, 13.76047999,
        5.5513072 ,  7.92200342,  6.652742  ,  8.1303945 ,  6.07992623,
        5.91408504, 10.31886066,  6.25586297,  4.91652556,  8.30729352,
        6.83083863,  7.6374603 ,  5.70405525, 13.35832725, 13.59170809,
       13.83139024,  9.23178204, 13.52430176, 12.95702647, 13.744387  ,
        7.50500064,  8.85939626,  6.78325151,  8.09704052,  5.57341623,
        5.26342203, 13.35832725, 14.08549802,  7.37902766,  9.63742669,
       12.05300943, 11.37939695,  7.822854  ,  8.92098752, 13.32613445,
        8.04718433,  8.85939626,  8.83341608,  7.27340834,  5.70405525,
        7.80625981])}
{'P': tensor([[2.5348e-03, 2.8936e-01, 2.2338e-01,  ..., 4.8516e-06, 5.4589e-08,
         5.0587e-07],
        [2.6295e-05, 3.3691e-03, 3.1931e-01,  ..., 6.1428e-06, 7.3618e-08,
         6.7042e-07],
        [3.1743e-06, 3.2797e-06, 2.3680e-03,  ..., 2.0063e-06, 6.8413e-08,
         4.1862e-07],
        ...,
        [7.7176e-03, 7.5333e-03, 6.2894e-03,  ..., 1.1368e-01, 1.5960e-05,
         1.7269e-04],
        [4.6308e-06, 4.5074e-06, 2.5145e-06,  ..., 7.1381e-05, 4.7249e-02,
         1.2023e-02],
        [9.5130e-04, 9.3483e-04, 5.6869e-04,  ..., 2.5454e-03, 8.2652e-04,
         2.0205e-01]]), 'X': tensor([[ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 6.2620e-03, 1.1246e-02,  ..., 4.0521e-01, 1.1894e+00,
         4.4175e-01],
        [6.2620e-03, 0.0000e+00, 1.2102e-02,  ..., 4.1132e-01, 1.1907e+00,
         4.4032e-01],
        [1.1246e-02, 1.2102e-02, 0.0000e+00,  ..., 4.0928e-01, 1.2006e+00,
         4.5223e-01],
        ...,
        [4.0521e-01, 4.1132e-01, 4.0928e-01,  ..., 0.0000e+00, 1.0824e+00,
         6.0470e-01],
        [1.1894e+00, 1.1907e+00, 1.2006e+00,  ..., 1.0824e+00, 0.0000e+00,
         8.1566e-01],
        [4.4175e-01, 4.4032e-01, 4.5223e-01,  ..., 6.0470e-01, 8.1566e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.31764181, 13.32645552, 13.50484539, 13.58101794, 13.60275211,
       13.55868684,  8.44940661, 13.79762415,  8.72896381,  7.39599986,
       11.75438006,  7.15990676,  6.89245352,  9.34061938,  5.76165934,
       14.02058621,  8.74470218,  7.10378919,  8.56889738, 12.0495521 ,
       11.05413545,  8.99276564,  8.11227668,  8.56889738,  8.04718433,
       10.6910768 , 10.57302546,  8.29022197, 13.2430377 , 13.54774355,
       13.97779839,  6.10360073,  5.69913817,  6.67121748,  5.56809335,
       10.36844813, 10.67132524,  7.5984809 , 13.99877413,  5.5671389 ,
        5.92952538, 12.86648771,  8.62329466, 10.26545792, 10.84493732,
       14.05633441, 13.89671314,  8.09984231,  7.59795445,  4.85603886,
       13.41713661,  9.75200848,  4.98989906, 12.1494023 ,  7.76457095,
       12.17901033,  8.40445872, 12.77402351, 13.04217487, 12.4013153 ,
        5.07306274,  8.31625171, 11.11969815,  7.06272256, 11.87665501,
        5.91222346,  9.96152788, 11.70168755,  9.09661212,  7.39004226,
       11.51557166,  5.19065439,  4.90505566,  4.78544875, 10.26545792,
        9.99211663, 11.67875855,  9.96152788, 13.45124917,  7.19513806,
       12.32545637,  7.06372172, 10.91805683,  6.42689705, 13.3426541 ,
        9.36378438, 13.45642379,  8.09899874,  6.33517349,  8.7273569 ,
       13.26590071,  6.55800267, 10.05558893, 13.89264822, 12.62711252,
        8.29640613,  5.19577146,  8.25828409,  8.90405823,  9.99211663,
       13.3607379 ,  6.74872981, 11.99633819,  6.48486171, 10.48721382,
        5.05989547])}
{'P': tensor([[2.1549e-03, 4.4462e-01, 1.5955e-01,  ..., 1.0421e-01, 5.2323e-08,
         1.6324e-07],
        [1.2129e-05, 3.7118e-03, 2.1048e-01,  ..., 2.0341e-01, 8.0944e-08,
         2.6094e-07],
        [6.0518e-06, 1.0805e-05, 2.3064e-03,  ..., 3.0054e-01, 3.7864e-08,
         1.6560e-07],
        ...,
        [3.0934e-06, 3.9076e-06, 5.5177e-06,  ..., 3.4849e-03, 3.9831e-08,
         2.0786e-07],
        [2.4048e-03, 1.9527e-03, 1.1031e-03,  ..., 4.8616e-04, 1.9259e-02,
         4.2520e-05],
        [6.3164e-04, 5.7155e-04, 5.9828e-04,  ..., 4.3061e-04, 1.9660e-03,
         3.0236e-01]]), 'X': tensor([[  9.6816,   9.1536,  -5.8543],
        [  9.3836,   9.5075,  -5.9278],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  6.7565,  11.8237,  -4.9882],
        [  6.1685,  12.1979,  -6.2473],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.7685, -10.8982,  -6.8928],
        [ 10.5955,   7.8824,  -3.7006],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.5607,  -5.0938,  -5.4531],
        [ 11.0275,  -5.0666,   8.4154],
        [ -8.7409,  -4.6837,  -6.5840],
        [  2.1120,  -4.6818,   8.6144],
        [  1.3339,   7.5627,  -0.9034],
        [  4.0591,   6.0202,  -1.6339],
        [  0.6952,  -4.8060,   8.8976],
        [ 11.4495,   6.3026,   6.8257],
        [  2.4356,  -4.5942,  -5.0808],
        [ -2.0445,   7.8469,  -0.5857],
        [ -3.5173, -10.0623,   4.8280],
        [  2.2849, -10.9655,  -7.6542],
        [  2.5829,  -4.5466,  -8.4015],
        [ 11.0864,   7.0387,  -0.1238],
        [  3.5965,  -4.0754,  -3.5035],
        [  7.7750,  11.0665,   2.8364],
        [  1.7000,  -4.7609,   2.7935],
        [  3.9022,  13.2906,   5.5788],
        [ 10.6283,  -5.7253,   7.4878],
        [  4.3554,   5.7393,  -9.0766],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.2608,  -2.5256,  -9.1446],
        [  5.9548,  -1.2273,  -2.6520],
        [  0.5850,  -4.7982,   9.6639],
        [  6.0345,   3.0175,  -1.4424],
        [  5.8979,   3.3958,  -3.8713],
        [  5.4426, -10.1181,  -5.1268],
        [  2.1771, -10.9776,  -3.2007],
        [ -5.6423,   6.4537,  -1.2091],
        [ -8.4573,   3.2448,   5.4370],
        [  9.3560,   9.5391,   9.4943],
        [ -4.0633,  -9.7927,  -3.0825],
        [  1.1994,   7.6035,   2.4295],
        [  3.9085,   6.1512,  -2.2560],
        [ -5.6792,  -8.7352,   4.0318],
        [ -8.0965,  -5.9712,   4.3466],
        [  9.4516,  -7.2501,  -1.8868],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -0.9715, -10.8630,   0.3555],
        [  1.9505,  -4.7171,   9.9868],
        [  9.3753,   9.5171,   3.9131],
        [  3.7992,  -3.9474,  -8.9019],
        [  8.0381,  -8.5690,  -8.9523],
        [  2.5974,   7.0363,  -7.7810],
        [  0.1460,   7.8340,  -9.4614],
        [  5.8325,  -1.5222,  -2.1341],
        [ -9.4378,  -1.8606,  -2.5668],
        [  1.6954,  -4.7616,   9.8198],
        [  2.6078, -10.9229,   6.2869],
        [  2.4356,  -4.5942,  -5.0808],
        [  4.4049,   5.6892,  -4.4394],
        [ -9.4203,  -2.0340,   2.5856],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.3535,   0.6792,  -7.5101],
        [ 12.3357,   3.6001,   0.7903],
        [  8.1528,  10.7452,   9.1493],
        [ 12.3025,  -1.7440,  -6.6632],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -9.4501,  -1.7133,   2.2980],
        [ -0.9715, -10.8630,   0.3555],
        [  4.6992,   5.3690,   3.0935],
        [  2.8994,   6.8675,  -9.5987],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.2373,  -4.6506,  -1.7334],
        [ -7.9667,   4.0814,  -0.6397],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.6603,   2.8250,   5.9181],
        [ -8.8934,   2.2597,   3.5831],
        [  4.7156,  -3.1831,  -8.0224],
        [  0.5850,  -4.7982,   9.6639],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.7295,  13.6703,  -8.9328],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.6019,   5.9540,  -5.1715],
        [  3.1480,  -4.3165,  -7.8002],
        [  1.7500,   7.4187,   4.4237],
        [  5.2188,  -2.5839,   2.1833],
        [ 12.5717,   1.9266,   7.4903],
        [  5.2911,  -2.4824,  -3.6877],
        [  2.2437,  -4.6489,   6.1430],
        [ 11.0275,  -5.0666,   8.4154],
        [  0.9967,  -4.8144,  -9.7370],
        [  2.4572,   7.1085,   7.7376],
        [  1.3403,  -4.8011,  -1.1698],
        [  4.8051, -10.3703,   0.5268],
        [ -1.5261,   7.8988,   9.3768],
        [  6.1372,   2.6726,  -6.7885],
        [ -9.4645,  -1.4880,  -2.0226],
        [  5.9587,  12.3216,   9.1695],
        [  2.5620,   7.0549,   8.4825],
        [ -6.4844,  -8.0230,   8.5209],
        [ -3.0644,   7.6439,  -1.4322],
        [  5.1142,  -2.7230,  -4.1966],
        [  7.0171,  11.6439,  -5.0178],
        [ 10.5057,  -5.9099,  -5.1599],
        [  3.9515,  -3.8425,   6.1729]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0024, 0.0485,  ..., 0.0618, 0.3626, 0.9647],
        [0.0024, 0.0000, 0.0505,  ..., 0.0638, 0.3643, 0.9670],
        [0.0485, 0.0505, 0.0000,  ..., 0.0133, 0.3598, 0.9186],
        ...,
        [0.0618, 0.0638, 0.0133,  ..., 0.0000, 0.3594, 0.9058],
        [0.3626, 0.3643, 0.3598,  ..., 0.3594, 0.0000, 0.8752],
        [0.9647, 0.9670, 0.9186,  ..., 0.9058, 0.8752, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.32374056, 13.35832725, 13.45642379, 13.59480906, 13.61801526,
       13.66896739,  5.46244988, 11.24434374, 13.20597932,  9.14973319,
        9.96152788, 12.13568598,  9.91668182,  5.13616224,  7.67939546,
        7.26074925,  4.85603886, 13.06957425,  5.19986797,  8.10886021,
       10.65929968, 11.20100795,  5.22903117, 13.13205357,  5.43544626,
       13.5247215 ,  5.05534609, 13.85158417, 12.07225469,  7.20482224,
        9.20340031,  5.83560009,  6.07992623,  4.83371341,  6.74686071,
        6.80559669, 11.4890861 , 11.1913598 ,  8.5724042 ,  9.05843201,
       13.3614598 , 10.60226614,  7.69752262,  7.28793663, 10.41909211,
       10.06023665, 11.9120321 , 12.30858677, 10.90638066,  5.10446653,
       13.35927527,  5.47866221, 11.74901778,  7.50035355,  7.83534987,
        6.02789634,  9.61942445,  5.05444288, 11.22993218,  5.19986797,
        7.19513806,  9.63742669,  7.96181288,  6.38968357, 12.85032836,
       13.48809595, 12.42554808, 12.49291768,  9.60412735, 10.90638066,
        7.13501264,  7.4544866 , 10.92597196,  5.16077764,  8.95133961,
        9.76896904,  9.10945678,  9.17595899,  5.68942572,  4.83371341,
        8.30125201, 13.94008948,  5.60064319, 13.04051744,  5.34250225,
        7.62232307,  5.82345867, 12.71843462,  5.84450127,  5.16205087,
       12.13568598,  4.91652556,  7.52117132,  4.98461983, 11.42947632,
        8.04483495,  6.69389431,  9.58072286, 13.68672855,  7.50564223,
       10.31584102,  8.23525433,  5.79391825, 13.59480906, 12.05395056,
        5.51177501])}
{'P': tensor([[2.1580e-03, 2.7874e-01, 1.4314e-01,  ..., 1.5557e-06, 7.6449e-08,
         2.0896e-07],
        [5.7636e-06, 2.5253e-03, 8.5118e-02,  ..., 1.3267e-06, 1.0818e-07,
         3.2600e-07],
        [7.9359e-06, 3.9674e-05, 5.4982e-03,  ..., 1.7456e-06, 2.8701e-07,
         8.8569e-07],
        ...,
        [1.2904e-03, 1.3176e-03, 7.8321e-04,  ..., 4.5960e-02, 8.4085e-06,
         1.5752e-05],
        [1.4660e-06, 1.0581e-06, 8.1596e-07,  ..., 4.1606e-05, 2.1723e-02,
         1.0581e-04],
        [9.5315e-07, 7.6525e-07, 6.1748e-07,  ..., 4.2976e-05, 1.7156e-01,
         2.8137e-02]]), 'X': tensor([[ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.1252e+01, -4.6495e+00,  2.9800e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 2.3928e+00,  7.1404e+00, -1.7035e+00],
        [ 1.9241e+00,  7.3503e+00,  7.9004e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 1.1086e+01,  7.0387e+00, -1.2384e-01],
        [-4.7642e+00, -9.3860e+00, -5.1133e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0100, 0.0641,  ..., 0.8708, 1.2795, 1.3146],
        [0.0100, 0.0000, 0.0600,  ..., 0.8742, 1.2893, 1.3241],
        [0.0641, 0.0600, 0.0000,  ..., 0.9342, 1.3210, 1.3607],
        ...,
        [0.8708, 0.8742, 0.9342,  ..., 0.0000, 1.1182, 1.0576],
        [1.2795, 1.2893, 1.3210,  ..., 1.1182, 0.0000, 0.1352],
        [1.3146, 1.3241, 1.3607,  ..., 1.0576, 0.1352, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.32557133, 13.43127131, 13.52430176, 13.54774355, 13.59170809,
       13.48809595, 11.31276109, 10.00410625,  5.39723241,  5.39450305,
        6.32572323,  5.98285319,  7.88677538,  5.91408504,  9.71392179,
        9.69792513,  5.05989547,  7.54620849,  8.09704052,  5.26342203,
        8.83725081, 13.01187484,  7.98469166,  4.87083348,  8.41148722,
       12.17451685,  7.37902766,  6.81084724, 10.57302546, 10.06023665,
        5.08910491, 13.01728313,  5.82345867, 11.8159254 ,  9.14973319,
        9.15253344,  9.09963894,  8.30729352,  7.60463349,  8.29203356,
        8.55982105, 11.62770332, 13.5247215 , 11.00812996, 11.33837396,
       13.76047999,  7.8788965 , 12.87752052, 10.65929968,  5.48276038,
       10.61047596,  5.29009836, 10.04916678, 13.84126197,  7.53063885,
        7.59795445,  9.38232121, 11.67875855, 14.09357547,  4.90505566,
        7.32754882,  5.79543038, 10.48315436, 11.63502679, 10.35093485,
       13.13205357, 10.52587243,  9.63742669,  7.60463349,  9.85384623,
        9.91668182,  7.88677538,  6.29821673, 13.97779839,  7.73151309,
        8.19703512,  7.20826219,  6.74872981,  7.32950385, 10.42667093,
       13.70893492,  5.05774295, 14.02058621, 11.83075005, 13.35927527,
        7.85919687, 10.35485331,  6.60752373,  6.81117802,  8.99276564,
        6.31576264, 11.00508681,  4.78002665,  8.57560275,  7.07902036,
       13.54774355,  6.67121748,  6.50409372,  6.83515201, 10.78533838,
        6.81084724,  5.01114719,  7.60463349,  6.31576264, 10.31584102,
        9.96391045])}
{'P': tensor([[3.5322e-03, 2.7363e-01, 2.1038e-01,  ..., 1.7835e-05, 3.2632e-06,
         2.0251e-05],
        [3.5094e-06, 2.2145e-03, 3.7883e-01,  ..., 4.9018e-06, 1.1777e-06,
         2.3409e-05],
        [7.0811e-06, 1.8749e-05, 7.7687e-03,  ..., 1.1988e-05, 3.1555e-06,
         1.1489e-04],
        ...,
        [1.0983e-02, 1.4254e-02, 1.2968e-02,  ..., 8.4176e-02, 3.1052e-03,
         2.8111e-04],
        [5.4483e-03, 5.1304e-03, 4.1615e-03,  ..., 1.6570e-01, 9.5189e-02,
         8.9298e-05],
        [1.3505e-06, 1.4860e-06, 1.6428e-06,  ..., 1.3110e-05, 1.0616e-05,
         4.2494e-02]]), 'X': tensor([[ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-9.4630e+00, -1.5145e+00,  2.9831e-01],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.9433e+00, -9.3362e+00, -3.4213e-01],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.8782e-02, 3.7343e-03,  ..., 4.2261e-01, 5.3374e-01,
         1.0002e+00],
        [2.8782e-02, 0.0000e+00, 2.5354e-02,  ..., 4.1308e-01, 5.1912e-01,
         1.0216e+00],
        [3.7343e-03, 2.5354e-02, 2.1453e-04,  ..., 4.2267e-01, 5.3310e-01,
         1.0037e+00],
        ...,
        [4.2261e-01, 4.1308e-01, 4.2267e-01,  ..., 0.0000e+00, 1.4191e-01,
         9.3505e-01],
        [5.3374e-01, 5.1912e-01, 5.3310e-01,  ..., 1.4191e-01, 0.0000e+00,
         1.0447e+00],
        [1.0002e+00, 1.0216e+00, 1.0037e+00,  ..., 9.3505e-01, 1.0447e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.32613445, 13.50484539, 13.58101794, 13.60275211, 13.6660671 ,
       13.55868684, 10.07238463,  4.75303127, 13.99877413,  6.76056625,
       12.05395056,  7.16058587,  5.84533345,  7.98469166, 11.46176259,
        9.9353921 ,  7.88677538,  5.26342203,  7.09887557, 11.65923396,
        6.55800267, 13.15525803,  9.79814997,  5.05989547,  5.16205087,
        7.32754882, 12.33167382,  6.67218687, 13.25017211, 13.96746816,
       13.3426541 ,  9.54154096,  4.91652556, 12.67680439,  6.05165248,
       11.75438006,  7.19534146,  8.30729352, 12.88270018,  9.91668182,
        5.5671389 , 11.42420508,  9.79039371, 11.27656488,  8.38252937,
       10.82106505, 12.889695  , 10.99326577, 13.12299145,  8.09899874,
       11.00812996,  4.98989906, 10.63038565,  9.48969978, 12.77402351,
        5.5513072 ,  9.58347456,  7.54127425, 10.09979747,  8.31625171,
       11.80092578,  5.51177501,  5.27178363, 13.40024224, 12.04924159,
        5.83560009, 12.07818871, 13.20988996,  6.56833864,  5.5205189 ,
        5.21280156,  7.71075226, 11.74901778, 10.60226614,  9.6282763 ,
       10.26327895,  8.14433542, 13.61600536,  4.9160962 , 10.31613269,
       13.97857591, 12.77402351, 10.61417181, 11.785566  , 10.71162681,
       11.63502679, 10.4977132 , 13.95361098,  6.27905292,  5.9510489 ,
       11.00508681, 12.41366754,  7.4544866 , 10.84437705,  6.84796202,
        7.5158489 ,  4.78002665,  9.25336371, 13.45124917,  4.83371341,
        9.15866607,  8.14433542, 12.50996828,  6.65368551,  6.28835101,
        8.89458273])}
{'P': tensor([[3.1999e-03, 3.0327e-01, 2.0395e-01,  ..., 1.2761e-02, 1.1729e-05,
         2.8638e-08],
        [3.1326e-06, 2.2618e-03, 3.8692e-01,  ..., 2.5371e-02, 3.6166e-06,
         1.3462e-08],
        [7.1520e-06, 1.9651e-05, 8.1423e-03,  ..., 1.4905e-01, 9.4504e-06,
         4.0304e-08],
        ...,
        [1.5255e-05, 3.4628e-05, 5.0118e-05,  ..., 1.6719e-01, 4.2069e-05,
         3.7954e-07],
        [1.4861e-02, 1.3121e-02, 1.1499e-02,  ..., 3.1668e-03, 5.6129e-02,
         2.6457e-05],
        [2.6261e-04, 1.2417e-04, 8.2212e-05,  ..., 7.5255e-06, 8.7714e-04,
         3.4646e-02]]), 'X': tensor([[ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.1369e+01,  6.4760e+00,  1.0787e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2102e-02, 3.1500e-02,  ..., 1.7921e-01, 2.6924e-01,
         8.9088e-01],
        [1.2102e-02, 0.0000e+00, 2.5354e-02,  ..., 1.8020e-01, 2.7928e-01,
         9.0282e-01],
        [3.1500e-02, 2.5354e-02, 2.1453e-04,  ..., 1.5779e-01, 2.7426e-01,
         9.1530e-01],
        ...,
        [1.7921e-01, 1.8020e-01, 1.5779e-01,  ..., 0.0000e+00, 2.1422e-01,
         9.2942e-01],
        [2.6924e-01, 2.7928e-01, 2.7426e-01,  ..., 2.1422e-01, 0.0000e+00,
         7.2314e-01],
        [8.9088e-01, 9.0282e-01, 9.1530e-01,  ..., 9.2942e-01, 7.2314e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.32645552, 13.50484539, 13.58101794, 13.60275211, 13.55868684,
       13.6660671 ,  7.5984809 , 10.6910768 , 13.40850328,  8.94395628,
        7.53884104,  8.27853076, 11.2845608 , 14.02532758,  7.71075226,
       11.81423161,  9.68323686, 12.19805394, 13.08414689, 11.83570132,
       13.00380624, 11.64436684,  6.2134961 ,  6.48486171,  5.07306274,
        8.95983586,  6.33075682,  6.2322769 , 10.60226614,  6.97106912,
        6.5618968 ,  9.20340031, 13.32557133, 11.42947632,  9.17016857,
        7.60529892, 11.69264243,  7.50564223, 10.46739073,  4.79003361,
        8.1303945 , 13.36092624, 14.09357547,  9.14973319,  6.652742  ,
        4.73784222, 10.53368644,  9.96152788,  4.89121805,  7.9337647 ,
       11.76830981,  8.85572292, 12.40074901,  6.81961996, 12.49291768,
       13.25017211, 11.62770332,  6.05113217,  6.74872981, 12.91505111,
       12.07225469,  9.95552626, 12.1494023 ,  9.62238828,  8.53501777,
       11.27656488,  5.13616224, 11.23201619,  6.28961988, 10.31886066,
       12.38881635, 11.64436684, 10.1371552 ,  8.66273171,  8.20436764,
        8.49161154, 11.018084  , 12.22276278,  6.01356004,  6.61211224,
       10.5575601 ,  9.87720102, 10.8207297 , 12.87752052,  9.27474386,
        6.31576264, 13.6105595 , 11.57249984,  7.06169391, 13.36800073,
       13.52430176,  9.95552626,  6.02779201,  5.10446653, 13.66896739,
        8.09704052, 11.77124373,  5.5748624 ,  5.5748624 , 12.97370371,
        6.28961988,  8.95133961,  7.13501264, 14.02058621,  6.54127431,
       11.85890664])}
{'P': tensor([[2.4642e-03, 3.0041e-01, 2.2330e-01,  ..., 4.1007e-07, 1.2262e-04,
         8.6714e-06],
        [1.4183e-05, 7.1141e-03, 1.8433e-05,  ..., 6.1582e-07, 8.9923e-05,
         1.8330e-05],
        [1.5961e-05, 5.5817e-01, 3.0510e-03,  ..., 3.9172e-07, 8.8462e-05,
         1.0334e-05],
        ...,
        [7.4077e-04, 8.4952e-04, 9.3435e-04,  ..., 1.9627e-02, 2.8762e-02,
         9.8403e-03],
        [2.4133e-03, 9.9510e-03, 5.6205e-03,  ..., 1.1780e-05, 3.3784e-02,
         1.1234e-03],
        [4.4813e-04, 6.9179e-04, 6.1616e-04,  ..., 3.4065e-04, 3.9352e-03,
         1.1163e-01]]), 'X': tensor([[ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.0939e+01,  7.3069e+00, -8.9365e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 4.0642e+00, -3.7599e+00, -3.4565e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 1.2138e+01,  4.4125e+00, -3.8833e+00],
        [ 3.1271e+00, -1.0834e+01,  2.6442e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0177, 0.0281,  ..., 1.1277, 0.6974, 1.0411],
        [0.0177, 0.0000, 0.0180,  ..., 1.1172, 0.6865, 1.0342],
        [0.0281, 0.0180, 0.0000,  ..., 1.1006, 0.6700, 1.0164],
        ...,
        [1.1277, 1.1172, 1.1006,  ..., 0.0000, 0.4314, 0.2717],
        [0.6974, 0.6865, 0.6700,  ..., 0.4314, 0.0000, 0.4146],
        [1.0411, 1.0342, 1.0164,  ..., 0.2717, 0.4146, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.3426541 , 13.46553623, 13.40024224, 13.56367857, 13.44593519,
       13.54774355,  6.32572323, 11.21703654, 12.3489324 , 12.1494023 ,
       13.31764181,  9.18690018, 13.02818702,  7.98469166,  9.38232121,
       14.02058621, 14.09357547, 11.00812996,  9.48337063, 13.59056613,
        7.60529892, 10.23946247, 11.95743558, 11.11038551, 13.15525803,
       14.07623266, 13.01187484,  6.06416957, 10.76154256, 11.42420508,
       10.51437947,  7.822854  , 12.3489324 , 11.62158554, 12.53345445,
        9.99145096,  6.58184474, 12.64235359, 12.30695098,  8.66273171,
       10.31886066,  4.8763135 ,  4.90676242,  9.5642993 ,  9.09502385,
       13.59056613,  4.92497514, 10.05558893, 12.12641969,  5.5366698 ,
       13.97779839,  9.93553424, 11.48734978, 13.99877413, 10.71162681,
        7.5403822 ,  6.84796202,  4.78544875, 10.5575601 , 11.785566  ,
       11.18839169,  9.09119351,  7.66377554,  6.28961988,  5.39723241,
       10.67132524, 11.35345345,  5.57341623,  9.61942445, 10.69963954,
        9.3109884 , 11.83570132,  5.36061974,  7.89745107, 13.89264822,
        6.84796202,  8.22728399,  8.06643487, 12.9089559 ,  9.26455795,
        5.1805345 , 12.91505111, 11.27656488, 10.86693708, 11.85890664,
       12.4980845 , 11.35182589, 11.85743425, 13.35832725, 13.34587027,
       10.4977132 , 10.48721382,  7.16058587, 14.09357547, 11.63765273,
       11.70168755, 13.66991027,  7.67939546,  8.94395628, 13.84126197,
        8.49161154, 12.41366754,  7.20482224, 12.77402351, 13.32645552,
        7.03733775])}
{'P': tensor([[2.8389e-03, 3.4024e-01, 2.8254e-01,  ..., 2.9896e-08, 5.4323e-08,
         1.0123e-07],
        [5.3862e-06, 3.2444e-03, 4.1377e-01,  ..., 2.2253e-08, 3.3848e-08,
         1.0120e-07],
        [6.1603e-06, 2.2892e-05, 4.6257e-03,  ..., 2.9802e-08, 4.3916e-08,
         1.5009e-07],
        ...,
        [8.6599e-05, 4.3684e-05, 3.9716e-05,  ..., 8.0492e-03, 4.1844e-02,
         2.7872e-05],
        [3.5966e-04, 2.2820e-04, 2.2511e-04,  ..., 5.3732e-05, 3.3126e-02,
         4.3155e-05],
        [3.8000e-05, 2.6517e-05, 2.6350e-05,  ..., 9.9959e-03, 8.9843e-03,
         1.0386e-01]]), 'X': tensor([[ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 4.2612e+00, -1.0551e+01, -6.6268e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 5.1861e+00,  4.7361e+00, -3.7312e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.2604e+01,  7.4940e-01,  7.1536e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 1.2085e+01, -2.5772e+00, -4.6073e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0398, 0.0366,  ..., 1.0521, 1.2263, 1.1813],
        [0.0398, 0.0000, 0.0034,  ..., 1.0913, 1.2626, 1.2192],
        [0.0366, 0.0034, 0.0000,  ..., 1.0880, 1.2592, 1.2158],
        ...,
        [1.0521, 1.0913, 1.0880,  ..., 0.0000, 0.3411, 0.2010],
        [1.2263, 1.2626, 1.2592,  ..., 0.3411, 0.0000, 0.1519],
        [1.1813, 1.2192, 1.2158,  ..., 0.2010, 0.1519, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.34587027, 13.48809595, 13.52430176, 13.59170809, 13.68672855,
       13.54774355,  7.92200342,  7.57685908,  9.09119351,  4.75303127,
        8.81930525,  5.69913817,  5.07306274,  6.38399789, 11.64746882,
        6.01356004,  7.83486943, 11.28093507,  7.49076899,  8.53501777,
        8.83341608, 12.9089559 , 11.9120321 , 12.05395056,  9.6282763 ,
       14.08549802,  5.87305005,  7.16598273,  9.6282763 , 11.83075005,
       13.0743399 , 14.08549802, 10.33712258, 11.37939695,  5.48276038,
        6.69389431,  9.17905491,  4.91296205,  9.32094055,  7.58378655,
        8.11227668,  7.88677538,  5.53175275, 13.1399181 , 12.35421461,
       10.35485331,  6.38968357, 13.61801526,  5.64464807, 14.09176259,
       10.04916678, 10.71162681, 11.0050379 , 10.61417181, 10.00794468,
        7.02326515,  7.88677538,  9.15253344,  5.19577146,  8.30729352,
        6.21275016,  7.20826219, 12.79126785, 11.1913598 , 13.6922425 ,
        5.87132718,  9.6282763 , 10.06703358, 11.75807252, 12.41366754,
       11.97957425,  8.49161154,  9.7712321 ,  7.27340834,  6.36004393,
       10.53368644,  6.42439344, 10.87409005, 12.87752052, 12.46513646,
        8.50278896, 10.74655346, 12.32545637,  9.23178204,  6.58184474,
        5.16077764,  8.66273171, 12.04924159,  8.89301516, 10.61472636,
       12.62576026,  7.32233733,  5.51177501, 13.05148826,  6.3949043 ,
        9.68324553,  8.91551231, 11.00508681,  6.81084724,  5.05774295,
       13.09737053,  6.24614905,  6.90115901, 11.97949273, 12.35625694,
        5.0209005 ])}
{'P': tensor([[3.4907e-03, 2.8053e-01, 2.1463e-01,  ..., 2.3864e-07, 9.0904e-07,
         3.9108e-07],
        [5.0656e-06, 2.1787e-03, 2.7786e-01,  ..., 2.1774e-07, 4.6058e-07,
         2.1942e-07],
        [4.6699e-06, 1.2020e-05, 2.4289e-03,  ..., 2.7045e-07, 4.8957e-07,
         2.4074e-07],
        ...,
        [1.9681e-06, 1.4762e-06, 1.3794e-06,  ..., 5.5227e-02, 1.8018e-03,
         7.2332e-04],
        [1.9288e-03, 1.3616e-03, 1.2396e-03,  ..., 8.1746e-05, 1.0006e-01,
         6.3037e-03],
        [3.1182e-04, 2.7174e-04, 2.6342e-04,  ..., 3.8938e-05, 5.4317e-03,
         7.7280e-02]]), 'X': tensor([[  9.4554,   9.4245,   9.8800],
        [  8.1528,  10.7452,   9.1493],
        [  7.7794,  11.0629,   8.9069],
        [  7.0515,  11.6194,   8.4289],
        [  5.9587,  12.3216,   9.1695],
        [  5.0617,  12.7957,   8.3544],
        [ -4.9217,   6.8979,  -1.7667],
        [  5.6414, -10.0303,   7.8525],
        [  2.4697,  -4.5836,   1.4593],
        [  5.6523,   3.9443,  -2.9449],
        [ -4.3963,   7.1629,   4.5263],
        [  0.2435,   7.8191,  -6.4709],
        [  6.0054,  -1.0902,   8.3079],
        [ -5.3382,   6.6533,   7.8054],
        [ -2.0840,   7.8415,  -7.8657],
        [ 12.6035,   0.7494,   7.1536],
        [  5.6022,  -1.9833,   8.3081],
        [ -3.4986,   7.5152,   9.0712],
        [  8.4691,  10.4571,  -5.0650],
        [  6.0467,  -9.8369,  -6.5816],
        [  7.2340,  -9.1505,  -5.8321],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.8320,   7.3871,   5.1452],
        [ 11.3118,   6.5957,  -4.5667],
        [ -2.7572,   7.7194,   5.9580],
        [  2.1379,  -4.6757,   6.4981],
        [  5.8617,   3.4853,  -9.8080],
        [ -8.8540,  -4.3971,  -5.2144],
        [ 11.7418,   5.6070,  -1.4192],
        [ -1.8367,   7.8718,  -8.6523],
        [  7.0515,  11.6194,   8.4289],
        [  9.0004,  -7.7195,   4.6945],
        [  1.9505,  -4.7171,   9.9868],
        [  9.0424,  -7.6779,  -7.5286],
        [  2.5382,   7.0672,   4.5097],
        [  2.3490,  -4.6199,  -4.2489],
        [ -1.0482, -10.8487,  -5.9066],
        [ -9.3604,   0.4855,  -3.0158],
        [  2.4932,   7.0903,  -5.4214],
        [ -2.9497, -10.3028,  -7.3479],
        [  5.1667,  -2.6543,   1.2879],
        [  6.3031,   1.8477,  -7.1706],
        [ -8.0534,  -6.0429,   2.4583],
        [  9.6614,   9.1784,  -1.2087],
        [ -1.0482, -10.8487,  -5.9066],
        [  2.3928,   7.1404,  -1.7035],
        [ -7.0825,   5.2073,  -5.0497],
        [  8.2422,  -8.4041,  -1.4962],
        [  9.3836,   9.5075,  -5.9278],
        [  7.9141,  -8.6656,   9.6948],
        [  5.5626,  -2.0536,   8.4433],
        [  6.3266,   0.3292,   9.3502],
        [ -8.1905,   3.7253,   4.1175],
        [ 10.8670,  -5.3428,  -2.8637],
        [  1.7122,  -4.7591,  -3.6734],
        [  6.2419,  -0.2313,   4.2440],
        [  9.4925,   9.3811,   8.9842],
        [  2.2182,  13.8007,   5.1889],
        [  2.1504,  -4.6726,   8.7908],
        [ -8.3539,  -5.5111,   0.3373],
        [ -0.8569,   7.9156,  -6.5557],
        [  2.0899,  -4.6870,  -7.4260],
        [ 12.5123,  -0.5339,  -1.0728],
        [ -5.0943,  -9.1682,   0.5073],
        [ 11.7796,  -3.4696,   9.0088],
        [ -6.7745,  -7.7264,   8.4684],
        [  5.8325,  -1.5222,  -2.1341],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.3087,   1.8053,  -1.9481],
        [  6.3301,   0.3638,  -7.5685],
        [  3.9913,  -3.8139,  -6.8311],
        [  8.0877,  -8.5297,  -1.1050],
        [  4.8674,  12.8875,   6.7622],
        [  4.1140,  -3.7220,  -0.1273],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.8154,   2.4611,   2.5198],
        [  3.1892, -10.8220,  -6.8409],
        [ 11.5198,  -4.0923,   5.5088],
        [ -2.6735, -10.4062,  -1.5911],
        [  6.0031,  12.2958,  -6.5992],
        [ -8.5149,  -5.1905,   1.5239],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.2038,   2.4020,  -3.1692],
        [  6.3124,   0.2057,  -0.4593],
        [  2.5469,  13.7192,  -9.6935],
        [  5.7963,  -1.6018,   1.8856],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.5123,  -0.5339,  -1.0728],
        [  9.9058,  -6.7203,   5.6928],
        [ 12.0582,  -2.6640,   0.7132],
        [  1.6408,   7.4591,  -4.2294],
        [  2.4572,   7.1085,   7.7376],
        [ -9.4203,  -2.0340,   2.5856],
        [ -1.0407,   7.9166,  -6.6021],
        [  5.0233,   4.9647,  -3.1788],
        [  9.6587,   9.1817,  -0.8212],
        [ 12.1504,   4.3674,  -7.2998],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.6426,  -4.9136,   9.4941],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.6928,  12.4710,   3.6001],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.9606,  -4.0994,   1.7886],
        [ -8.0330,  -6.0765,   6.7945],
        [  4.2339,  -3.6268,   6.3125],
        [  4.1140,  -3.7220,  -0.1273]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0343, 0.0315,  ..., 1.3187, 0.5160, 0.9121],
        [0.0343, 0.0000, 0.0034,  ..., 1.3496, 0.5501, 0.9434],
        [0.0315, 0.0034, 0.0000,  ..., 1.3479, 0.5472, 0.9400],
        ...,
        [1.3187, 1.3496, 1.3479,  ..., 0.0000, 0.9157, 1.0202],
        [0.5160, 0.5501, 0.5472,  ..., 0.9157, 0.0000, 0.4593],
        [0.9121, 0.9434, 0.9400,  ..., 1.0202, 0.4593, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.35013374, 13.48809595, 13.52430176, 13.59170809, 13.68672855,
       13.76047999,  8.47371868, 11.50791859,  5.20659892,  6.89245352,
        8.40445872,  7.822854  ,  6.10360073,  8.53014915,  8.1137452 ,
       12.62576026,  5.94293185,  8.28968067, 13.45642379, 11.54672755,
       11.66453059, 10.04916678,  7.61088125, 13.09426142,  8.19703512,
        5.14124612,  6.81961996,  9.88571713, 13.01187484,  8.08320235,
       13.59170809, 11.85743425,  5.10446653, 11.86239961,  7.50917551,
        5.18277858, 10.8992524 ,  9.37296102,  7.5158489 , 10.7167285 ,
        5.80860165,  6.56833864, 10.06850677, 13.32613445, 10.8992524 ,
        7.53063885,  8.7907923 , 11.77124373, 13.35832725, 11.73567759,
        5.92952538,  6.33517349,  8.99791316, 12.10942412,  5.05774295,
        6.24614905, 13.34587027, 13.97779839,  5.14371005, 10.00794468,
        7.96181288,  5.13182506, 12.52372629, 10.48840476, 12.27993098,
       10.27572673,  6.02789634, 10.42667093,  6.5618968 ,  6.34059021,
        5.5205189 , 11.75438006, 13.77604293,  5.54776934,  9.76896904,
        9.15253344, 11.28215429, 12.22503754, 10.74409834, 13.6829843 ,
        9.97220799,  5.19577146,  6.65259775,  6.31576264, 13.95361098,
        6.01356004,  5.43943791, 12.52372629, 11.9702762 , 12.3489324 ,
        7.6374603 ,  7.52117132,  9.63742669,  7.98469166,  7.06272256,
       13.32645552, 12.91143491,  9.12220271,  9.94173708, 12.59840193,
       13.70893492, 11.92133129,  9.85384623, 10.07238463,  5.5748624 ,
        5.54776934])}
{'P': tensor([[3.6700e-03, 2.0811e-01, 2.0112e-01,  ..., 1.8290e-06, 2.2008e-05,
         3.7778e-06],
        [1.3177e-05, 2.8129e-03, 3.6653e-01,  ..., 2.4782e-06, 2.5795e-05,
         2.2558e-06],
        [3.4103e-06, 4.8156e-06, 3.0414e-03,  ..., 4.0483e-06, 2.8387e-05,
         1.4954e-06],
        ...,
        [1.8002e-06, 2.3570e-06, 2.8192e-06,  ..., 2.4360e-01, 4.2441e-04,
         1.0150e-04],
        [1.7169e-05, 2.8415e-05, 4.5369e-05,  ..., 1.2189e-01, 1.5144e-01,
         1.7424e-04],
        [4.6475e-03, 5.7346e-03, 4.7228e-03,  ..., 2.4813e-04, 4.1846e-03,
         9.8004e-02]]), 'X': tensor([[  9.3836,   9.5075,  -5.9278],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  6.7565,  11.8237,  -4.9882],
        [  6.1685,  12.1979,  -6.2473],
        [  6.0310,  12.2795,  -6.4528],
        [  6.1984,  -0.4327,   1.5438],
        [ -8.5607,  -5.0938,  -5.4531],
        [  4.6992,   5.3690,   3.0935],
        [  5.9526,   3.2525,  -4.3821],
        [ 12.4013,  -1.2597,  -7.6474],
        [  3.3167,   6.6022,   6.3554],
        [  6.2576,   2.1359,  -3.3398],
        [  1.1350,  14.0104,   0.6321],
        [ -5.0132,  -9.2234,  -8.3531],
        [  6.0955,   2.8210,  -4.3321],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.2555,   2.1474,   4.1685],
        [  9.3424,  -7.3685,  -3.6807],
        [ -7.7239,   4.4298,   3.1216],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.6614,  -4.5192,  -9.3641],
        [ 11.7794,   5.5086,  -9.6929],
        [ -3.2486,   7.5925,  -1.3719],
        [  6.3535,   0.6792,  -7.5101],
        [  1.7000,  -4.7609,   2.7935],
        [  4.2096,   5.8815,   1.9230],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -9.4283,  -1.9585,  -0.9129],
        [ -7.8849,  -6.3115,  -4.7479],
        [ -5.1974,   6.7396,   3.3484],
        [ 11.7418,   5.6070,  -1.4192],
        [ -9.1907,  -3.3180,  -5.5943],
        [  7.3724,  -9.0577,  -8.9843],
        [ -2.5683, -10.4433,  -2.9182],
        [  0.2571,  -4.7594,   1.5618],
        [ -9.1231,  -3.5739,   8.4708],
        [  5.9193,  -1.3176,   5.1739],
        [  5.4867,  12.5816,   2.9549],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.7796,  -3.4696,   9.0088],
        [ -4.5778,  -9.5013,  -5.0667],
        [  9.3753,   9.5171,   3.9131],
        [ 11.0891,  -4.9558,  -2.3178],
        [  5.7963,  -1.6018,   1.8856],
        [ 11.5083,  -4.1177,   0.2389],
        [  5.4242, -10.1261,   3.9595],
        [  0.9809,   7.6642,  -1.4217],
        [  2.0729,   7.2878,   7.0880],
        [  5.1292,   4.8183,  -5.7997],
        [ 12.3453,  -1.5462,   7.0212],
        [ -6.7992,   5.4991,   9.9186],
        [  9.0004,  -7.7195,   4.6945],
        [ -9.3419,   0.5972,  -1.2841],
        [  2.3373,  -4.6232,  -5.2246],
        [ -1.9487,   7.8590,  -3.3330],
        [ -5.3179,  -9.0101,   9.4472],
        [ -9.4773,  -1.0024,   5.2599],
        [  2.1504,  -4.6726,   8.7908],
        [ 10.0567,   8.6697,   4.6329],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.4351,  -1.8891,   3.5642],
        [  1.3767, -11.0341,   9.9415],
        [ -8.8540,  -4.3971,  -5.2144],
        [  8.1528,  10.7452,   9.1493],
        [ -6.4620,  -8.0449,   6.2719],
        [ -4.2271,   7.2387,  -1.8316],
        [ -5.1377,   6.7751,  -0.4438],
        [  2.5375,   7.0676,  -4.3490],
        [ 11.7418,   5.6070,  -1.4192],
        [ -6.9838,  -7.4965,   0.9796],
        [  8.8474,  10.0868,  -8.8644],
        [  6.3542,   0.6934,  -3.3840],
        [  3.6703,   6.3443,  -5.4401],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.1755,   2.5233,  -2.0042],
        [  2.0899,  -4.6870,  -7.4260],
        [  8.7131,  10.2216,   7.7915],
        [  2.2644,  -4.6434,   9.7508],
        [ -9.3419,   0.5972,  -1.2841],
        [  8.9284,  10.0036,  -6.9210],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.7590,  13.6621,   9.4496],
        [ -2.2508, -10.5479,  -2.7235],
        [ -7.4602,   4.7718,  -4.6277],
        [ 11.1085,  -4.9204,   0.0736],
        [  1.6308,  13.9254,  -2.0984],
        [  5.5158,  -2.1337,   3.5246],
        [  3.4009,  -4.1874,  -5.9751],
        [  0.1041, -11.0045,   2.4311],
        [  5.8325,  -1.5222,  -2.1341],
        [ -5.6169,  -8.7844,   8.9057],
        [  6.0073,   3.0986,   6.7937],
        [ -1.5261,   7.8988,   9.3768],
        [ -7.0825,   5.2073,  -5.0497],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.3551,   0.7149,   1.3344],
        [ -7.6293,  -6.6859,  -5.5161],
        [  4.4049,   5.6892,  -4.4394],
        [  1.4150,  13.9647,  -4.0915],
        [  2.1379,  -4.6757,   6.4981],
        [  7.7794,  11.0629,   8.9069],
        [ -7.5401,   4.6718,   9.0033],
        [ -0.5709,   7.9055,   7.6089],
        [  6.3551,   0.7149,   1.3344]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.0451e-02, 6.3760e-02,  ..., 1.4148e+00, 1.1472e+00,
         6.4758e-01],
        [5.0451e-02, 0.0000e+00, 1.3313e-02,  ..., 1.3763e+00, 1.1020e+00,
         6.0154e-01],
        [6.3760e-02, 1.3313e-02, 0.0000e+00,  ..., 1.3660e+00, 1.0899e+00,
         5.8935e-01],
        ...,
        [1.4148e+00, 1.3763e+00, 1.3660e+00,  ..., 0.0000e+00, 4.0693e-01,
         8.1403e-01],
        [1.1472e+00, 1.1020e+00, 1.0899e+00,  ..., 4.0693e-01, 0.0000e+00,
         5.0107e-01],
        [6.4758e-01, 6.0154e-01, 5.8935e-01,  ..., 8.1403e-01, 5.0107e-01,
         1.0727e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.35832725, 13.45642379, 13.59480906, 13.61801526, 13.66896739,
       13.68063098,  6.2134961 ,  9.96152788,  7.13501264,  6.78325151,
       12.46513646,  7.38844703,  6.61211224, 14.05633441, 10.4977132 ,
        6.7166297 ,  4.78002665,  6.61386436, 11.89855267,  8.90405823,
       12.38881635,  5.22903117,  5.24461068, 13.00380624,  8.25828409,
        6.38968357,  5.05534609,  7.23276582, 12.49291768,  9.62958973,
       10.09979747,  8.51088913, 13.01187484,  9.7712321 , 11.67875855,
       10.75443106,  4.7663526 ,  9.79814997,  6.06416957, 13.72594022,
       11.27160035, 12.27993098, 10.54658597, 13.35927527, 12.1460931 ,
        6.01356004, 12.22276278, 11.48734978,  7.72668467,  7.57685908,
        7.03733775, 12.44177641,  8.74470218, 11.85743425,  9.36093562,
        5.18047638,  8.09704052, 10.46238054,  9.53015376,  5.14371005,
       13.2778378 ,  5.13125513,  9.62238828, 11.11969815,  9.88571713,
       13.48809595, 10.31886066,  8.38252937,  8.50278896,  7.5092834 ,
       13.01187484, 10.24557047, 13.41713661,  6.39187285,  7.32950385,
        9.11473391,  6.67108662,  5.13182506, 13.43127131,  5.16611782,
        9.36093562, 13.40850328, 10.8207297 , 13.93790235, 10.78533838,
        8.85572292, 12.1494023 , 14.02058621,  5.91408504,  5.39450305,
       11.0050379 ,  6.02789634, 10.42667093,  6.7593917 ,  8.04483495,
        8.7907923 , 13.15525803,  6.39520397, 10.14436912,  7.19513806,
       14.03618649,  5.14124612, 13.52430176,  8.87007495,  7.92607342,
        6.39520397])}
{'P': tensor([[2.8477e-03, 1.6132e-01, 1.5846e-01,  ..., 8.0260e-05, 3.1387e-07,
         3.4078e-07],
        [1.3532e-06, 1.6664e-03, 5.0906e-06,  ..., 1.1910e-05, 3.9258e-07,
         1.6759e-07],
        [3.8642e-06, 1.5021e-01, 1.7238e-03,  ..., 3.4211e-05, 2.6905e-07,
         2.0533e-07],
        ...,
        [1.7957e-04, 1.4136e-03, 8.7468e-04,  ..., 1.2636e-01, 2.6375e-06,
         2.0608e-05],
        [2.7803e-06, 1.7474e-06, 1.8887e-06,  ..., 2.8546e-07, 1.2189e-01,
         2.5121e-04],
        [2.8105e-04, 1.4618e-04, 2.2395e-04,  ..., 3.1331e-04, 1.8730e-05,
         1.0486e-01]]), 'X': tensor([[  9.3753,   9.5171,   3.9131],
        [  6.6319,  11.9066,   3.5165],
        [  7.7750,  11.0665,   2.8364],
        [  6.8407,  11.7666,   3.0045],
        [  6.1575,  12.2046,   3.3257],
        [  5.6928,  12.4710,   3.6001],
        [  4.6063,  13.0051,  -3.1456],
        [ 11.1469,   6.9237,   1.6160],
        [  6.2247,  -0.3150,  -9.3657],
        [  6.3535,   0.6792,  -7.5101],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -2.5161,   7.7699,  -9.3531],
        [ -7.5720,   4.6310,   2.1717],
        [ -8.6801,   2.7809,  -0.8443],
        [ -4.0409,   7.3169,  -3.5779],
        [ 10.7566,   7.6212,  -2.3279],
        [ -9.0574,   1.7801,   8.8986],
        [  5.7784,  -1.6401,   9.5587],
        [ -3.4100,   7.5436,  -7.6980],
        [  4.3459,  -3.5327,   0.3667],
        [  2.0870,  -4.6877,   9.6823],
        [  6.3549,   0.7096,   6.9198],
        [  7.8982,  -8.6778,   0.0239],
        [  4.8674,  12.8875,   6.7622],
        [  5.9548,  -1.2273,  -2.6520],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.6928,  12.4710,   3.6001],
        [ -5.3179,  -9.0101,   9.4472],
        [ 11.5966,   5.9666,   6.0083],
        [  3.1064,  13.5607,   6.7605],
        [ -0.1963,   7.8765,  -1.1371],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -2.1382,   7.8339,   7.3524],
        [  5.6523,   3.9443,  -2.9449],
        [  7.5309,  11.2617,   7.2656],
        [  7.9816,  10.8938,  -0.5378],
        [ -6.7643,  -7.7372,  -9.6663],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.8674,  12.8875,   6.7622],
        [  7.5309,  11.2617,   7.2656],
        [  4.1637,  13.1898,  -1.9189],
        [  4.7544,  -3.1425,  -1.5315],
        [  8.0381,  -8.5690,  -8.9523],
        [ 12.5789,   0.1722,   1.6470],
        [  2.8602,   6.8905,   5.8779],
        [  5.8889,  -1.3916,   8.8993],
        [  7.1821,  -9.1845,   9.8548],
        [  2.5128, -10.9365,   9.9645],
        [  0.9945,  -4.8144,  -0.1897],
        [  1.6308,  13.9254,  -2.0984],
        [  5.8889,  -1.3916,   8.8993],
        [ -8.5607,  -5.0938,  -5.4531],
        [  2.3261,  13.7749,   5.9121],
        [ -6.8466,  -7.6488,   1.9648],
        [ -1.6275, -10.7221,   3.2376],
        [  1.0093,   7.6567,   2.3526],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 12.4996,   2.6335,  -6.3166],
        [ -1.5261,   7.8988,   9.3768],
        [  8.0877,  -8.5297,  -1.1050],
        [ 12.5763,   0.1333,  -9.2650],
        [  8.7715,  -7.9390,   5.2900],
        [  4.4290,  -3.4596,   1.4181],
        [  6.9284,  11.7061,  -0.3732],
        [  1.1644, -11.0394,   7.8511],
        [  1.2728, -11.0372,   9.8121],
        [  5.1196,  -2.7160,   4.2705],
        [  0.6470, -11.0352,  -1.1931],
        [  6.3303,   1.6145,  -8.4083],
        [  6.1579,  -0.5961,  -6.5595],
        [  4.3377,   5.7570,   1.3712],
        [ 12.3357,   3.6001,   0.7903],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 12.5770,   1.8512,  -0.6578],
        [  5.9026,   3.3837,  -0.6573],
        [ -8.9178,   2.1936,   7.5388],
        [  6.7792,  11.8084,   1.6251],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.4430,  -0.1897,   4.8383],
        [  6.2419,  -0.2313,   4.2440],
        [  1.6954,  -4.7616,   9.8198],
        [  9.6662,   9.1725,   8.1056],
        [ -8.9033,   2.2329,  -8.8912],
        [  5.8450,  -9.9356,   3.0767],
        [ -1.0012, -10.8576,   2.2867],
        [ -1.6334, -10.7207,   7.1397],
        [  9.9058,  -6.7203,   5.6928],
        [  5.9193,  -1.3176,   5.1739],
        [ -0.1963,   7.8765,  -1.1371],
        [ 11.6019,   5.9540,  -5.1715],
        [  5.8275,  -1.5334,   2.0124],
        [  0.4996,   7.7738,  -4.4711],
        [  3.2366,  -4.2732,   5.1532],
        [ -3.2486,   7.5925,  -1.3719],
        [  7.9141,  -8.6656,   9.6948],
        [  9.3607,   9.5337,  -4.5412],
        [  4.5491,  -3.3485,   6.8412],
        [  6.2975,   0.0947,   5.1791],
        [  0.9788,  -4.8145,  -3.8661],
        [ -8.9942,  -3.9990,   6.0448],
        [  3.5965,  -4.0754,  -3.5035],
        [ -8.9282,   2.1648,   8.4894],
        [  6.9956,  11.6590,  -7.6876],
        [ -8.8835,  -4.3178,   9.3080],
        [  2.3490,  -4.6199,  -4.2489]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0264, 0.0766,  ..., 1.0008, 1.2340, 0.7957],
        [0.0264, 0.0000, 0.0895,  ..., 1.0140, 1.2557, 0.8176],
        [0.0766, 0.0895, 0.0000,  ..., 0.9251, 1.2548, 0.7337],
        ...,
        [1.0008, 1.0140, 0.9251,  ..., 0.0000, 1.6648, 0.4500],
        [1.2340, 1.2557, 1.2548,  ..., 1.6648, 0.0000, 1.2158],
        [0.7957, 0.8176, 0.7337,  ..., 0.4500, 1.2158, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.35927527, 13.62896593, 13.5247215 , 13.6105595 , 13.66991027,
       13.70893492, 13.79676654, 13.12218013,  6.23262289,  6.38968357,
       10.14436912,  8.16714836,  8.87587457,  9.11473391,  8.35856487,
       13.18278606,  9.23071698,  6.00663314,  8.27853076,  5.60064319,
        5.13125513,  6.39438044, 11.73397309, 13.77604293,  6.07992623,
        9.54154096, 13.70893492, 10.46238054, 13.04156075, 13.91198191,
        7.8788965 ,  9.18690018,  9.60080639,  8.12043983,  6.89245352,
       13.54774355, 13.50484539, 10.27716434,  5.1224079 , 13.77604293,
       13.54774355, 13.83139024,  5.69913817, 11.74901778, 12.58006038,
        7.46053281,  6.05113217, 11.65923396, 11.22142195,  4.9160962 ,
       14.02058621,  6.05113217,  9.96152788, 13.96987806, 10.26545792,
       10.84493732,  7.72291946,  9.97822473, 12.77402351,  8.04483495,
       11.75438006, 12.5769659 , 11.83075005,  5.62005629, 13.60275211,
       11.10066126, 11.11038551,  5.79543038, 11.05413545,  6.53290663,
        6.18668463,  7.20826219, 12.85032836,  9.4997987 , 12.7125079 ,
        6.80370254,  9.18358562, 13.61600536,  8.56889738,  9.44486609,
        6.24614905,  5.05444288, 13.32557133,  9.17905491, 11.52733581,
       10.9036226 , 10.84437705, 11.9702762 ,  6.06416957,  7.8788965 ,
       13.04051744,  6.02588517,  7.78980299,  5.36061974,  8.25828409,
       11.73567759, 13.36092624,  5.64865173,  6.29821673,  4.91296205,
        9.84314712,  5.43544626,  9.18690018, 13.59673406,  9.87720102,
        5.18277858])}
{'P': tensor([[2.7452e-03, 2.4598e-01, 2.0150e-01,  ..., 2.7515e-08, 2.6852e-08,
         7.4445e-08],
        [2.1342e-05, 1.8046e-02, 2.9106e-04,  ..., 1.8675e-07, 1.2131e-07,
         2.6754e-07],
        [6.0439e-06, 6.2108e-02, 2.6050e-03,  ..., 2.7432e-08, 1.7830e-08,
         4.4999e-08],
        ...,
        [4.7521e-05, 1.9726e-05, 2.6227e-05,  ..., 9.5312e-02, 9.2753e-03,
         2.2887e-02],
        [1.7621e-04, 6.3359e-05, 7.1594e-05,  ..., 6.2552e-06, 1.9906e-02,
         1.3476e-02],
        [1.3802e-03, 4.2045e-04, 5.7675e-04,  ..., 6.6646e-07, 1.2541e-05,
         7.8162e-03]]), 'X': tensor([[ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 9.3920e-01, -4.8143e+00, -3.4649e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 5.1890e+00,  4.7318e+00,  9.5155e-01],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 7.7750e+00,  1.1067e+01,  2.8364e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.5060e-02, 1.0061e-01,  ..., 9.8233e-01, 8.0897e-01,
         3.6709e-01],
        [2.5060e-02, 2.1453e-04, 8.2307e-02,  ..., 9.6283e-01, 8.1351e-01,
         3.5150e-01],
        [1.0061e-01, 8.2307e-02, 0.0000e+00,  ..., 9.6367e-01, 8.8580e-01,
         3.7716e-01],
        ...,
        [9.8233e-01, 9.6283e-01, 9.6367e-01,  ..., 0.0000e+00, 9.2295e-01,
         6.2700e-01],
        [8.0897e-01, 8.1351e-01, 8.8580e-01,  ..., 9.2295e-01, 0.0000e+00,
         6.3563e-01],
        [3.6709e-01, 3.5150e-01, 3.7716e-01,  ..., 6.2700e-01, 6.3563e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.3607379 , 13.55868684, 13.50484539, 13.60275211, 13.58101794,
       13.61600536,  5.22903117,  8.92098752, 12.4980845 ,  8.30125201,
       11.35345345,  5.37343173,  9.15253344,  4.90505566, 13.62896593,
        6.10774544,  8.38252937, 10.68031351,  9.24501829,  4.9160962 ,
       13.89974693,  9.61942445,  9.77777346,  8.66273171, 11.00726483,
       13.03002338,  5.04028538, 12.30858677,  8.41148722, 10.1979669 ,
        6.34059021,  6.19707944,  7.02253235, 13.744387  , 10.88618114,
        8.83341608,  5.24137909,  6.46560181,  9.14044984, 12.9089559 ,
        5.13616224, 10.36844813,  7.15990676,  4.89869205, 10.53368644,
       13.25336001, 11.72802491, 10.04916678,  9.78285424, 13.12218013,
       12.4980845 , 12.07225469, 12.22276278,  5.87132718,  8.62631801,
        5.21280156, 10.92597196,  5.0713364 , 12.07225469,  7.67939546,
        8.04664886,  9.58072286,  6.67218687, 13.34587027,  7.58378655,
       13.5247215 ,  9.15976746,  6.23262289,  6.383897  ,  7.5158489 ,
       13.3614598 , 13.62896593,  4.9160962 , 12.27993098,  9.9353921 ,
       13.68672855,  6.80559669,  7.06372172,  9.50654967,  8.89301516,
       11.81423161,  8.89458273, 14.07623266, 11.04629471,  6.81961996,
       10.69963954,  6.81084724,  5.1805345 , 12.58006038, 13.62896593,
        6.33075682,  8.30729352, 12.85032836,  5.57341623,  6.48486171,
        4.98461983,  6.60752373,  4.83371341, 11.63724792,  5.89074104,
       13.31764181, 12.5769659 , 13.89886469, 11.018084  , 11.72802491,
       12.1494023 ])}
{'P': tensor([[2.0557e-03, 2.0844e-01, 1.3856e-01,  ..., 2.8439e-05, 1.1676e-06,
         8.9679e-03],
        [6.0315e-06, 1.9634e-03, 2.5584e-01,  ..., 4.0689e-05, 1.2603e-06,
         1.6932e-02],
        [1.4525e-06, 2.2687e-06, 1.4329e-03,  ..., 5.3371e-05, 1.4056e-06,
         2.4003e-02],
        ...,
        [2.2432e-06, 2.4739e-06, 2.9710e-06,  ..., 5.0985e-02, 1.8880e-04,
         6.6285e-06],
        [1.7846e-07, 1.7184e-07, 1.9932e-07,  ..., 3.5848e-05, 2.1182e-02,
         2.4309e-07],
        [4.2532e-05, 6.5725e-05, 1.4309e-04,  ..., 5.5712e-02, 3.9633e-04,
         3.2743e-01]]), 'X': tensor([[  9.3607,   9.5337,  -4.5412],
        [  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  6.7565,  11.8237,  -4.9882],
        [  6.1685,  12.1979,  -6.2473],
        [  4.9337,  12.8566,  -3.6582],
        [  5.0233,   4.9647,  -3.1788],
        [  1.7228,  -4.7575,   8.3396],
        [ -0.1963,   7.8765,  -1.1371],
        [  1.9204,   7.3518,   8.4781],
        [  2.2860,   7.1916,  -6.1011],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.3804,  -2.3504,   7.2744],
        [  7.0515,  11.6194,   8.4289],
        [ -8.6112,   2.9319,  -7.9943],
        [ -7.6650,   4.5092,   0.2520],
        [ -9.3087,   0.7792,  -7.5587],
        [ -3.3999, -10.1152,  -1.2088],
        [ -7.5603,   4.6459,   8.2002],
        [ -8.9636,   2.0648,   8.1946],
        [  1.9241,   7.3503,   7.9004],
        [ -8.1204,  -5.9308,   4.0761],
        [  4.4633,   5.6287,  -0.7754],
        [  3.4604,  -4.1545,   1.8091],
        [  2.1504,  -4.6726,   8.7908],
        [  4.6231,   5.4556,   8.4973],
        [ -1.9487,   7.8590,  -3.3330],
        [  3.2692,  13.5098,  -4.9032],
        [ -9.3087,   0.7792,  -7.5587],
        [  9.8082,  -6.8396,   4.9615],
        [  5.6314,  -1.9302,  -9.6543],
        [  6.3246,   1.6696,  -0.1544],
        [  8.4691,  10.4571,  -5.0650],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.0345,   3.0175,  -1.4424],
        [ 11.3800,  -4.3919,   1.4066],
        [  7.9141,  -8.6656,   9.6948],
        [ -5.5525,   6.5146,  -4.4984],
        [  0.1287, -11.0065,   6.5757],
        [ -8.0087,   4.0173,   9.1091],
        [ -5.9225,  -8.5355,   2.4778],
        [  6.1575,  12.2046,   3.3257],
        [ -7.0120,   5.2825,   7.7762],
        [ -9.4117,   0.1160,   8.2574],
        [ -1.9714,   7.8563,  -1.8084],
        [  2.1440, -10.9811,   8.5281],
        [ -1.8788, -10.6567,   3.7068],
        [  2.0243,   7.3086,   7.5432],
        [  5.8450,  -9.9356,   3.0767],
        [ 12.1588,   4.3364,  -8.0867],
        [  4.9337,  12.8566,  -3.6582],
        [  5.4867,  12.5816,   2.9549],
        [  6.7565,  11.8237,  -4.9882],
        [ -0.5709,   7.9055,   7.6089],
        [  0.1931,  -4.7491,   7.4385],
        [  6.5428,  -9.5728,  -2.9897],
        [  5.4867,  12.5816,   2.9549],
        [ 12.2311,  -2.0445,   1.3545],
        [  4.0591,   6.0202,  -1.6339],
        [  6.1730,  -0.5375,  -8.8861],
        [  8.6420,  -8.0580,   6.5781],
        [ -1.9714,   7.8563,  -1.8084],
        [  6.8407,  11.7666,   3.0045],
        [ -3.1204, -10.2345,   9.2288],
        [  1.8447,  -4.7372,   8.1235],
        [  2.0870,  -4.6877,   9.6823],
        [ -9.4748,  -0.7765,  -5.6695],
        [  3.4604,  -4.1545,   1.8091],
        [  4.0078,  -3.8018,  -6.2168],
        [  4.5057,  -3.3894,  -0.8160],
        [  6.8704,  -9.3809,   7.5769],
        [ -9.0961,   1.6529,   3.1553],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.2921,  -1.7902,  -4.0870],
        [  9.0129,  -7.7072,   9.1400],
        [  6.2247,  -0.3150,  -9.3657],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.1730,  -0.5375,  -8.8861],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.2905,   1.9366,   2.1169],
        [  4.7482, -10.3907,   5.9149],
        [  5.5294,  -2.1106,   9.0626],
        [ -5.1369,   6.7755,   8.3836],
        [  3.6154,  -4.0640,  -7.9698],
        [ 12.1504,   4.3674,  -7.2998],
        [  0.7685,  -4.8098,  -2.9310],
        [  6.2905,   1.9366,   2.1169],
        [ -4.7334,   6.9982,  -0.4392],
        [ -9.1658,  -3.4155,  -2.8880],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.1756, -10.8247,   6.8882],
        [  3.3633,  13.4794,   0.5193],
        [ 12.2589,  -1.9312,   9.8657],
        [  4.7739,  -3.1218,  -4.1188],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.3653,   3.4184,   9.3454],
        [  3.4604,  -4.1545,   1.8091],
        [  2.7685, -10.8982,  -6.8928],
        [  2.3200,   7.1756,   1.1938],
        [  2.0870,  -4.6877,   9.6823],
        [ 12.4877,  -0.7222,   8.8347],
        [  9.4554,   9.4245,   9.8800],
        [ -7.0825,   5.2073,  -5.0497],
        [ -8.0087,   4.0173,   9.1091],
        [  2.0449,  13.8402,  -7.1642]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0394, 0.0267,  ..., 0.9047, 1.3742, 0.1751],
        [0.0394, 0.0000, 0.0133,  ..., 0.9197, 1.4096, 0.1589],
        [0.0267, 0.0133, 0.0000,  ..., 0.9178, 1.3995, 0.1665],
        ...,
        [0.9047, 0.9197, 0.9178,  ..., 0.0000, 0.9355, 0.8077],
        [1.3742, 1.4096, 1.3995,  ..., 0.9355, 0.0000, 1.3979],
        [0.1751, 0.1589, 0.1665,  ..., 0.8077, 1.3979, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.36092624, 13.45642379, 13.59480906, 13.61801526, 13.66896739,
       13.77075174,  7.06272256,  5.05981914,  7.8788965 ,  7.5984809 ,
        7.54620849,  9.26455795,  5.87132718, 13.59170809,  9.09661212,
        8.89301516,  9.34127013, 10.67132524,  8.8737556 ,  9.19837792,
        7.59795445, 10.05558893,  7.18354885,  5.40689332,  5.14371005,
        7.15100303,  8.09704052, 13.89974693,  9.34127013, 11.95743558,
        5.95297459,  6.54127431, 13.45642379,  5.19577146,  6.74686071,
       12.19805394, 11.73567759,  8.55982105, 11.00726483,  8.95983586,
       10.38897563, 13.66991027,  8.77913428,  9.41244865,  8.09984231,
       11.18839169, 10.82106505,  7.58378655, 11.52733581, 12.9089559 ,
       13.77075174, 13.72594022, 13.61801526,  7.92607342,  4.75303127,
       11.59512194, 13.72594022, 12.40074901,  7.26074925,  6.19633139,
       11.8159254 ,  8.09984231, 13.6105595 , 10.69963954,  5.08372149,
        5.13125513,  9.50654967,  5.40689332,  5.52416368,  5.63824971,
       11.62770332,  9.24501829,  6.2134961 , 12.4217523 , 11.85890664,
        6.23262289,  9.48969978, 10.63038565,  6.19633139,  8.34949383,
        6.58184474, 11.42420508,  5.91854397,  8.50268322,  5.43943791,
       12.91143491,  4.87083348,  6.58184474,  8.4486771 ,  9.78147542,
       11.9120321 , 11.28093507, 13.89264822, 12.41012046,  5.70405525,
        6.383897  ,  9.0368458 ,  5.40689332, 11.24434374,  7.54127425,
        5.13125513, 12.5085995 , 13.35013374,  8.7907923 ,  8.95983586,
       13.99048094])}
{'P': tensor([[3.0183e-03, 3.5861e-01, 2.6482e-01,  ..., 1.5650e-06, 1.8674e-02,
         1.7157e-08],
        [5.4405e-06, 2.7902e-03, 3.5585e-01,  ..., 1.9960e-06, 5.0191e-03,
         1.4277e-08],
        [5.9101e-06, 1.7328e-05, 3.5014e-03,  ..., 2.9000e-06, 3.4089e-03,
         1.7825e-08],
        ...,
        [2.1080e-06, 3.0162e-06, 3.4991e-06,  ..., 1.9738e-01, 7.7226e-06,
         2.0417e-05],
        [1.0908e-04, 9.9581e-04, 2.5005e-03,  ..., 6.4413e-05, 2.3990e-02,
         1.5318e-07],
        [2.2199e-05, 1.3147e-05, 1.1628e-05,  ..., 1.7570e-06, 1.4833e-05,
         1.7835e-02]]), 'X': tensor([[ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 4.2275e+00, -3.6320e+00, -2.7141e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 2.9260e+00, -4.4162e+00,  8.3808e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 8.1043e+00, -8.5164e+00, -7.9290e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 6.2630e+00,  2.1057e+00, -5.2136e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 2.7558e+00, -4.4844e+00,  6.3579e-02],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 2.1504e+00, -4.6726e+00,  8.7908e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.9503e-02, 2.6590e-02,  ..., 1.5160e+00, 1.1757e-01,
         1.1543e+00],
        [2.9503e-02, 0.0000e+00, 3.4459e-03,  ..., 1.5444e+00, 1.4194e-01,
         1.1838e+00],
        [2.6590e-02, 3.4459e-03, 0.0000e+00,  ..., 1.5410e+00, 1.3849e-01,
         1.1808e+00],
        ...,
        [1.5160e+00, 1.5444e+00, 1.5410e+00,  ..., 0.0000e+00, 1.4079e+00,
         5.0108e-01],
        [1.1757e-01, 1.4194e-01, 1.3849e-01,  ..., 1.4079e+00, 0.0000e+00,
         1.0632e+00],
        [1.1543e+00, 1.1838e+00, 1.1808e+00,  ..., 5.0108e-01, 1.0632e+00,
         7.5848e-05]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.3614598 , 13.48809595, 13.52430176, 13.59170809, 13.68672855,
       13.76047999,  5.19577146, 11.04313641,  5.1224079 ,  6.28961988,
        7.18354885,  9.44486609,  6.12886024, 11.04629471,  6.90115901,
       12.97370371,  5.70405525, 13.1399181 ,  6.77777606,  7.72415048,
       11.04313641, 11.83570132,  5.57341623,  6.38105015, 11.38774149,
        9.34127013,  7.54620849,  8.83725081,  7.03733775, 11.00508681,
       10.26545792,  5.69913817,  5.29755487,  6.36004393,  9.09502385,
        9.75401449, 13.86480175,  8.43626114,  7.83486943, 12.84056604,
       11.11969815,  9.64157461, 11.28895489,  5.20659892,  5.69913817,
        9.5014146 ,  6.67218687,  6.54127431, 11.64436684,  5.96556248,
       10.27716434,  5.13125513,  8.85572292, 11.74901778, 11.51811603,
        6.46356806, 13.20988996,  9.69629   , 10.19036081, 14.02532758,
       12.70489563, 11.75618513,  5.5748624 ,  9.97220799,  9.14044984,
       11.99633819, 12.85032836,  8.55982105, 12.46513646,  8.28968067,
       11.74901778, 10.98076873,  6.60752373, 10.37698435,  8.0550286 ,
        7.50035355, 13.45642379,  5.0713364 ,  8.25828409,  7.88677538,
       13.59056613, 12.67680439, 10.63038565,  6.38968357,  8.62631801,
       13.96746816, 11.87665501, 10.31886066,  6.00663314,  5.24137909,
        8.01009083,  5.47702766,  9.60412735,  6.28835101,  5.0713364 ,
        5.24137909,  5.26342203, 10.99326577,  5.14371005,  8.24808832,
        8.62631801, 10.04916678, 11.69264243,  8.30125201, 13.56367857,
       11.42947632])}
{'P': tensor([[2.4219e-03, 2.7141e-01, 2.4157e-01,  ..., 1.5189e-04, 1.6179e-08,
         7.1476e-05],
        [1.3391e-05, 3.8931e-03, 4.9531e-01,  ..., 1.4943e-04, 3.0792e-08,
         7.4052e-05],
        [8.5470e-05, 1.5453e-04, 6.5465e-02,  ..., 1.2206e-03, 5.5604e-07,
         6.3318e-04],
        ...,
        [6.3362e-03, 1.0073e-02, 1.5159e-02,  ..., 9.4107e-02, 1.0556e-05,
         1.5269e-02],
        [4.9560e-06, 4.2342e-06, 2.8960e-06,  ..., 5.4854e-05, 1.8289e-01,
         9.3907e-05],
        [7.9118e-03, 1.1955e-02, 1.5129e-02,  ..., 1.5396e-01, 1.3679e-05,
         9.5648e-02]]), 'X': tensor([[ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 2.1379e+00, -4.6757e+00,  6.4981e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 2.5829e+00, -4.5466e+00, -8.4015e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0135, 0.0192,  ..., 0.5025, 1.6584, 0.5493],
        [0.0135, 0.0000, 0.0242,  ..., 0.5025, 1.6570, 0.5472],
        [0.0192, 0.0242, 0.0000,  ..., 0.5218, 1.6776, 0.5684],
        ...,
        [0.5025, 0.5025, 0.5218,  ..., 0.0000, 1.1590, 0.0947],
        [1.6584, 1.6570, 1.6776,  ..., 1.1590, 0.0000, 1.1108],
        [0.5493, 0.5472, 0.5684,  ..., 0.0947, 1.1108, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.36800073, 13.44593519, 13.54774355, 13.43127131, 13.56367857,
       13.59170809,  7.58378655,  5.84533345, 10.31613269,  7.33350642,
        9.75401449, 10.80728788, 13.69162795,  5.94909709,  5.19986797,
        5.19577146,  7.16747878, 12.41012046, 13.05148826,  9.55720651,
        9.58072286, 12.44177641, 12.49317563, 12.19805394, 12.27993098,
       11.70016217, 11.80092578,  6.00732193,  5.24137909,  8.06643487,
       13.35927527, 13.1471313 , 11.4424816 , 12.12448912,  7.69752262,
        5.05444288, 12.5085995 ,  9.43737523, 12.27835996,  5.23878371,
        6.46560181, 11.42947632, 13.09737053,  9.94173708, 12.27991991,
        9.96152788, 14.07623266, 12.41012046, 10.61417181, 12.38881635,
        5.19065439,  5.94535529,  8.06517143,  8.89301516,  5.14124612,
        9.61942445, 13.01187484,  5.22903117,  7.57946459, 10.31613269,
        9.55593374,  6.12886024,  5.39723241,  6.53250937, 14.05633441,
       12.18998195, 10.69963954, 12.87752052,  7.72291946,  7.03733775,
        9.17016857,  7.88677538,  8.09984231, 11.67875855,  5.80860165,
        8.31625171,  6.65368551,  8.92549752, 12.38881635, 12.07225469,
       10.31584102, 13.1471313 ,  9.17595899, 12.59969936,  7.58378655,
        6.01356004, 11.64746882, 10.794755  ,  6.28961988, 14.03505624,
        4.7663526 ,  6.28835101,  5.70405525, 13.18278606,  9.88310614,
       13.94008948, 11.4088043 ,  9.24501829, 13.32557133,  8.83725081,
       11.51811603,  5.79543038,  9.69792513,  7.07902036, 10.54658597,
        6.99097013])}
{'P': tensor([[2.3432e-03, 3.8059e-01, 2.3364e-01,  ..., 4.1245e-05, 4.7839e-06,
         4.8833e-04],
        [1.0261e-05, 3.9406e-03, 5.0136e-01,  ..., 9.6018e-05, 6.1077e-06,
         1.0286e-03],
        [1.0945e-04, 1.9194e-04, 8.1316e-02,  ..., 2.7531e-03, 7.7069e-05,
         2.9898e-02],
        ...,
        [1.2592e-06, 1.4128e-06, 1.6264e-06,  ..., 2.5486e-02, 1.7025e-05,
         1.4620e-04],
        [3.6507e-03, 4.1539e-03, 3.6489e-03,  ..., 1.9469e-03, 1.1350e-01,
         1.5416e-03],
        [3.4311e-05, 4.2173e-05, 5.8337e-05,  ..., 4.3658e-02, 7.4712e-05,
         1.3831e-01]]), 'X': tensor([[ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 5.1142e+00, -2.7230e+00, -4.1966e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0118, 0.0314,  ..., 1.0241, 0.8720, 0.6934],
        [0.0118, 0.0000, 0.0242,  ..., 1.0353, 0.8788, 0.7052],
        [0.0314, 0.0242, 0.0000,  ..., 1.0527, 0.9027, 0.7172],
        ...,
        [1.0241, 1.0353, 1.0527,  ..., 0.0000, 0.6169, 0.4094],
        [0.8720, 0.8788, 0.9027,  ..., 0.6169, 0.0000, 0.7205],
        [0.6934, 0.7052, 0.7172,  ..., 0.4094, 0.7205, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.37121923, 13.44593519, 13.54774355, 13.56367857, 13.43127131,
       13.59170809,  5.8828986 ,  6.54615887,  9.62958973,  9.15253344,
        5.64865173,  8.98765644,  5.79562691,  7.37902766,  4.85995002,
       13.94008948,  7.06372172, 11.57249984, 10.63038565, 12.30695098,
        9.78285424,  9.88310614, 10.80728788,  5.5205189 , 11.57249984,
       12.5769659 ,  5.18047638,  6.02779201, 13.0743399 ,  9.36378438,
       13.0493368 ,  7.19513806,  5.91854397, 12.7125079 ,  5.84450127,
       13.48809595, 11.8159254 , 14.02058621,  6.02588517,  4.91038884,
        4.78002665,  7.38844703, 11.22142195, 10.63038565, 11.50791859,
        7.50035355,  5.8828986 ,  8.92098752,  7.14837514,  8.8737556 ,
       13.06957425,  9.62238828,  6.84796202,  9.43112957, 11.48734978,
       10.46739073,  8.31625171, 11.87665501,  5.1224079 , 11.18839169,
       10.67322527, 10.31886066, 13.68063098,  9.23071698,  9.18358562,
       12.67680439,  4.87083348,  4.76969364, 10.5575601 ,  7.24677664,
        8.7273569 , 13.55868684,  8.06643487, 10.00794468,  6.34059021,
        8.23525433,  4.79003361, 12.35421461,  8.5724042 ,  7.46053281,
        9.88310614,  8.92395381,  9.78147542, 14.09176259, 10.37998078,
       10.53368644,  5.79391825, 13.41713661,  5.91854397, 12.86648771,
        9.75401449,  8.08320235,  8.20436764,  4.78544875,  9.43112957,
        6.3949043 ,  6.07992623,  9.11473391, 11.0050379 , 11.05413545,
       10.84437705,  8.5724042 , 11.56146873,  8.66273171,  6.652742  ,
        8.1303945 ])}
{'P': tensor([[6.4602e-03, 2.5141e-01, 1.6922e-01,  ..., 2.7261e-07, 2.0297e-04,
         5.9259e-06],
        [3.6758e-06, 4.2735e-03, 3.9402e-02,  ..., 6.8258e-08, 4.1830e-04,
         1.2509e-06],
        [2.1533e-06, 6.5756e-05, 5.0554e-03,  ..., 5.4971e-08, 7.7710e-04,
         1.3872e-06],
        ...,
        [6.7815e-03, 1.5367e-03, 6.3453e-04,  ..., 1.9064e-02, 1.1622e-05,
         2.7823e-04],
        [5.3921e-06, 8.7774e-06, 1.1282e-05,  ..., 4.7239e-06, 8.5845e-02,
         4.8179e-05],
        [1.6522e-02, 7.1746e-03, 1.0073e-02,  ..., 8.6444e-04, 1.6324e-03,
         2.8094e-01]]), 'X': tensor([[  9.1820,   9.7334,  -2.7932],
        [  6.2026,  12.1774,  -1.5887],
        [  4.9337,  12.8566,  -3.6582],
        [  4.6063,  13.0051,  -3.1456],
        [  4.9152,  12.8653,  -1.4459],
        [  7.1697,  11.5343,  -0.7276],
        [ 11.9122,   5.1397,  -1.5498],
        [ -9.4773,  -1.0024,   5.2599],
        [ -9.3464,   0.5708,  -1.6397],
        [  9.0652,   9.8597,  -3.5949],
        [ -4.2040,   7.2486,   2.9875],
        [ -5.0550,   6.8231,  -9.2502],
        [ 12.5763,   0.1333,  -9.2650],
        [ -8.9636,   2.0648,   8.1946],
        [ -5.0943,  -9.1682,   0.5073],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.4753,  -0.8012,  -5.2050],
        [  5.5101,  -2.1433,   9.5442],
        [  6.6319,  11.9066,   3.5165],
        [  5.9548,  -1.2273,  -2.6520],
        [ 12.5832,   1.7543,  -7.6134],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -7.5401,   4.6718,   9.0033],
        [  0.3493,  -4.7727,  -6.5682],
        [ -3.7964,  -9.9294,   7.7147],
        [ 11.9936,  -2.8677,   6.0614],
        [ 12.0582,  -2.6640,   0.7132],
        [  9.5203,   9.3483,   5.8781],
        [  8.0877,  -8.5297,  -1.1050],
        [ -2.5683, -10.4433,  -2.9182],
        [ 10.9916,   7.2133,  -0.9179],
        [  1.9312,  13.8649,   8.3563],
        [ -5.1369,   6.7755,   8.3836],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.0348,  -2.8229,   3.5436],
        [  0.3231,  -4.7691,  -1.6120],
        [ 12.3357,   3.6001,   0.7903],
        [ 11.9695,  -2.9407,  -6.3498],
        [ -3.5173, -10.0623,   4.8280],
        [  0.2571,  -4.7594,   1.5618],
        [  3.3094,  13.4969,  -3.6507],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.8318,   3.5567,  -4.7692],
        [  6.3606,   0.9204,   3.2525],
        [ -6.8466,  -7.6488,   1.9648],
        [  0.6396,  14.0772,  -6.3786],
        [  0.9445,   7.6736,   0.5095],
        [ -1.7009,   7.8851,  -8.2437],
        [ -2.4956, -10.4682,   3.5632],
        [ 10.6283,  -5.7253,   7.4878],
        [  4.5057,  -3.3894,  -0.8160],
        [ -7.9667,   4.0814,  -0.6397],
        [  2.5620,   7.0549,   8.4825],
        [ -2.5161,   7.7699,  -9.3531],
        [ 12.5770,   1.8512,  -0.6578],
        [ 11.7794,   5.5086,  -9.6929],
        [  0.1287, -11.0065,   6.5757],
        [  3.2366,  -4.2732,   5.1532],
        [ -3.5903,   7.4847,  -8.9112],
        [ -8.3062,  -5.6008,   0.0287],
        [  5.9193,  -1.3176,   5.1739],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.9883,   3.1536,   6.2320],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.2707,   0.9661,   4.4863],
        [  6.8704,  -9.3809,   7.5769],
        [  6.1752,   2.5242,   3.5874],
        [ 10.0567,   8.6697,   4.6329],
        [ 12.0582,  -2.6640,   0.7132],
        [  9.9756,  -6.6330,  -3.3429],
        [ -3.6381,   7.4683,   7.3544],
        [  6.3122,  -9.6994,  -6.3282],
        [  0.5960, -11.0335,   3.1144],
        [ 12.5927,   0.4199,  -9.4997],
        [  7.5784,  -8.9141,   5.8410],
        [ -8.1531,  -5.8748,   2.6934],
        [ -4.7334,   6.9982,  -0.4392],
        [ -9.1178,   1.5784,   2.1410],
        [ 11.3367,  -4.4806,   0.4643],
        [ -8.5900,   2.9767,   7.4811],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.2849, -10.9655,  -7.6542],
        [ -0.3432,   7.8900,   7.1805],
        [  3.1892, -10.8220,  -6.8409],
        [  6.2790,  -0.0259,  -5.4116],
        [  1.1644, -11.0394,   7.8511],
        [  9.0051,   9.9234,   5.4411],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.3464,   0.5708,  -1.6397],
        [ 10.5955,   7.8824,  -3.7006],
        [  2.1440, -10.9811,   8.5281],
        [ 11.6544,   5.8273,  -3.1375],
        [  9.9058,  -6.7203,   5.6928],
        [  3.4140,  -4.1803,  -8.8224],
        [  3.9913,  -3.8139,  -6.8311],
        [ 11.7162,   5.6728,  -8.9372],
        [ -2.3317,   7.8034,  -3.6274],
        [ 12.0582,  -2.6640,   0.7132],
        [ 11.2939,   6.6325,  -5.8311],
        [  6.8407,  11.7666,   3.0045],
        [ -8.4267,  -5.3697,   2.4172],
        [ 11.9122,   5.1397,  -1.5498],
        [ 12.0582,  -2.6640,   0.7132],
        [ -6.0195,   6.1789,  -2.9496],
        [  6.3360,   0.4258,  -9.2509]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0944, 0.0873,  ..., 0.3886, 0.8107, 0.2982],
        [0.0944, 0.0000, 0.1814,  ..., 0.3229, 0.8088, 0.3614],
        [0.0873, 0.1814, 0.0000,  ..., 0.4530, 0.8132, 0.2490],
        ...,
        [0.3886, 0.3229, 0.4530,  ..., 0.0000, 0.6013, 0.4686],
        [0.8107, 0.8088, 0.8132,  ..., 0.6013, 0.0000, 0.6135],
        [0.2982, 0.3614, 0.2490,  ..., 0.4686, 0.6135, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.38091121, 13.6660671 , 13.77075174, 13.79676654, 13.77222752,
       13.58101794, 12.97370371,  9.53015376,  9.36378438, 13.39372426,
        8.37954853,  8.49161154, 12.5769659 ,  9.19837792, 10.48840476,
       11.76830981,  9.10945678,  9.50913849,  5.91222346, 13.62896593,
        6.07992623, 12.70489563,  9.97822473,  8.87007495,  4.78544875,
       10.63038565, 12.33167382, 12.3489324 , 13.3426541 , 11.75438006,
       10.75443106, 13.1471313 , 13.99877413,  8.50268322,  5.24137909,
        5.77219077,  4.78002665, 12.85032836, 12.32545637, 10.65929968,
        4.7663526 , 13.89671314,  9.58347456,  6.83083863,  6.42689705,
       10.26545792, 14.09176259,  7.73151309,  8.06643487, 10.76154256,
       12.07225469,  5.63824971,  8.95133961,  7.50564223,  8.16714836,
       12.7125079 , 13.00380624, 11.00726483,  5.36061974,  8.30125201,
       10.01804223,  6.06416957,  6.14086582,  6.76791483, 13.86480175,
        9.32094055, 11.62770332,  6.67121748, 13.2778378 , 12.3489324 ,
       11.97957425,  8.30729352, 11.57249984, 11.04953767, 12.59969936,
       11.70016217, 10.04916678,  8.4486771 ,  9.25336371, 12.18998195,
        9.09119351,  9.93553424, 11.20100795,  7.89745107, 11.28215429,
        6.27905292, 11.10066126, 13.40024224,  8.90405823,  9.36378438,
       13.20597932, 11.18839169, 13.03002338, 11.9702762 ,  5.39723241,
        5.5205189 , 13.01728313,  8.14433542, 12.3489324 , 13.09737053,
       13.6105595 ,  9.99211663, 12.97370371, 12.3489324 ,  8.62631801,
        6.35029274])}
{'P': tensor([[5.3161e-03, 2.1770e-01, 2.1489e-01,  ..., 2.7615e-04, 2.4687e-08,
         3.2666e-04],
        [8.1154e-06, 5.5023e-03, 1.6375e-05,  ..., 1.0709e-05, 1.4635e-08,
         1.2569e-04],
        [4.3675e-06, 5.9909e-01, 2.3298e-03,  ..., 5.8970e-06, 6.3458e-09,
         5.5624e-05],
        ...,
        [1.4319e-02, 3.5997e-02, 3.2418e-02,  ..., 8.6519e-03, 6.5257e-08,
         1.6820e-03],
        [6.2670e-05, 1.5917e-05, 1.7473e-05,  ..., 1.7441e-04, 4.2968e-02,
         1.1564e-04],
        [1.9289e-04, 3.6758e-04, 3.3083e-04,  ..., 2.1053e-04, 1.5452e-05,
         1.3770e-01]]), 'X': tensor([[  9.0652,   9.8597,  -3.5949],
        [  6.7565,  11.8237,  -4.9882],
        [  7.0171,  11.6439,  -5.0178],
        [  4.9337,  12.8566,  -3.6582],
        [  4.6063,  13.0051,  -3.1456],
        [  4.5954,  13.0099,  -4.4439],
        [  0.1287, -11.0065,   6.5757],
        [  0.5250, -11.0306,   5.5358],
        [ -4.7389,   6.9953,   3.7638],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.5620,   7.0549,   8.4825],
        [ -5.3736,   6.6310,  -1.5174],
        [ -8.8934,   2.2597,   3.5831],
        [ -5.9872,   6.2037,  -9.4477],
        [  4.3377,   5.7570,   1.3712],
        [ -8.9624,  -4.0944,  -4.9622],
        [  9.0424,  -7.6779,  -7.5286],
        [ 12.2311,  -2.0445,   1.3545],
        [ 11.7794,   5.5086,  -9.6929],
        [  9.2811,  -7.4336,   5.1771],
        [  1.1274,  14.0116,   8.1753],
        [  7.8982,  -8.6778,   0.0239],
        [  2.6970,   6.9826,   9.7350],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.1575,  12.2046,   3.3257],
        [ -8.8154,   2.4611,   2.5198],
        [  5.9883,   3.1536,   6.2320],
        [  6.9284,  11.7061,  -0.3732],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.1940,  -2.1881,   7.1499],
        [  1.6308,  13.9254,  -2.0984],
        [ 10.9916,   7.2133,  -0.9179],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.0870,  -4.6877,   9.6823],
        [  5.2199,   4.6861,  -1.9304],
        [  4.6357,   5.4415,   7.9067],
        [ -1.6275, -10.7221,   3.2376],
        [  3.4009,  -4.1874,  -5.9751],
        [  3.9770, -10.6341,   8.0783],
        [  3.5284, -10.7484,   5.3347],
        [  4.4039,   5.6903,   8.7366],
        [  5.6928,  12.4710,   3.6001],
        [  2.2644,  -4.6434,   9.7508],
        [  3.5965,  -4.0754,  -3.5035],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -4.1296,   7.2803,  -2.7194],
        [  6.2036,   2.4030,  -1.2500],
        [  0.9036,   7.6840,   9.9237],
        [ 11.2517,  -4.6495,   2.9800],
        [  2.3373,  13.7721,  -0.2442],
        [  3.6469,   6.3624,  -1.8414],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.9497, -10.3028,  -7.3479],
        [  8.3791,  10.5409,   5.5590],
        [  0.7145,  -4.8071,  -2.3766],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -3.9456,  -9.8542,   0.7466],
        [ -4.7389,   6.9953,   3.7638],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.5382,   7.0672,   4.5097],
        [  2.5382,   7.0672,   4.5097],
        [  6.2038,   2.4020,  -3.1692],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.5506,  -5.1154,   6.3423],
        [  0.3038,   7.8092,   0.4164],
        [ -7.8280,   4.2848,  -7.1955],
        [  0.9967,  -4.8144,  -9.7370],
        [ -7.5720,   4.6310,   2.1717],
        [  0.2571,  -4.7594,   1.5618],
        [  6.9956,  11.6590,  -7.6876],
        [ 10.9162,  -5.2599,  -7.0812],
        [  2.0243,   7.3086,   7.5432],
        [ 12.2589,  -1.9312,   9.8657],
        [  8.9284,  10.0036,  -6.9210],
        [ -9.4309,  -0.0599,  -4.0830],
        [ 11.6801,  -3.7196,  -0.9181],
        [  4.6357,   5.4415,   7.9067],
        [ -6.4368,   5.8356,   3.3014],
        [ -8.9606,  -4.0994,   1.7886],
        [  0.1206,  -4.7363,   8.7103],
        [  6.1579,  -0.5961,  -6.5595],
        [  2.5636, -10.9293,   9.6519],
        [ -3.9456,  -9.8542,   0.7466],
        [ -7.5401,   4.6718,   9.0033],
        [ -9.3411,  -2.6004,  -2.7062],
        [  5.7151,  -1.7700,   3.1542],
        [  5.6928,  12.4710,   3.6001],
        [  6.2038,   2.4020,  -3.1692],
        [ -8.3717,  -5.4769,  -9.2228],
        [ 11.7795,  -3.4697,   3.5624],
        [  0.5600, -11.0321,  -0.8660],
        [ -6.1914,  -8.2997,   8.5380],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -2.0445,   7.8469,  -0.5857],
        [ 11.3691,   6.4760,   1.0787],
        [  2.7420,  -4.4896,  -3.7139],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.0617,  12.7957,   8.3544],
        [ 12.5789,   0.1722,   1.6470],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4753,  -0.8012,  -5.2050],
        [  9.6587,   9.1817,  -0.8212],
        [  6.8704,  -9.3809,   7.5769],
        [  1.7500,   7.4187,   4.4237]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0985, 0.1059,  ..., 0.2541, 1.1468, 0.8166],
        [0.0985, 0.0000, 0.0089,  ..., 0.3517, 1.2363, 0.9071],
        [0.1059, 0.0089, 0.0000,  ..., 0.3583, 1.2450, 0.9158],
        ...,
        [0.2541, 0.3517, 0.3583,  ..., 0.0000, 0.9508, 0.6231],
        [1.1468, 1.2363, 1.2450,  ..., 0.9508, 0.0000, 0.3307],
        [0.8166, 0.9071, 0.9158,  ..., 0.6231, 0.3307, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.39372426, 13.61801526, 13.59480906, 13.77075174, 13.79676654,
       13.79762415, 11.00726483, 11.04313641,  8.44940661,  8.62329466,
        7.50564223,  8.53501777,  9.17595899,  8.62162372,  7.20826219,
        9.85330646, 11.86239961, 12.40074901, 13.00380624, 11.89105259,
       14.05687815, 11.73397309,  7.48538905,  9.60412735, 13.66991027,
        9.15253344,  6.76791483, 13.60275211, 11.05672136, 12.38881635,
       14.02058621, 13.1471313 ,  5.46244988,  5.13125513,  7.01475376,
        7.14837514, 10.84493732,  5.39450305, 11.35345345, 11.31276109,
        7.19534146, 13.70893492,  5.16611782,  5.43544626, 12.35625694,
        8.36995953,  6.652742  ,  7.73692436, 12.17451685, 13.96905756,
        7.33350642, 11.70016217,  9.34127013, 10.7167285 , 13.46553623,
        4.85995002,  9.97964013, 10.61472636,  8.44940661, 11.9120321 ,
        5.34250225,  7.50917551,  7.50917551,  6.65259775,  7.822854  ,
        9.96391045,  7.81509631,  8.92395381,  4.91652556,  8.87587457,
        4.7663526 , 13.59673406, 12.11735334,  7.58378655, 12.41012046,
       13.40850328,  9.43112957, 12.25806559,  7.14837514,  8.68832398,
        9.85384623,  4.73784222,  6.18668463, 11.22596994, 10.61472636,
        8.87007495,  9.69629   ,  5.98285319, 13.70893492,  6.65259775,
       10.00410625, 12.27991991, 11.04629471, 10.35464916, 10.23946247,
        8.10886021, 13.08414689,  5.26067019, 13.01187484, 13.76047999,
       12.58006038,  9.24501829,  9.50913849, 13.32645552, 11.62770332,
        7.62232307])}
{'P': tensor([[2.6995e-03, 4.9387e-01, 2.0145e-01,  ..., 2.6429e-05, 6.6728e-05,
         5.8488e-07],
        [1.2477e-05, 4.8155e-03, 3.9643e-01,  ..., 3.7553e-05, 1.5314e-04,
         1.2874e-06],
        [2.0932e-05, 3.8904e-05, 1.1257e-02,  ..., 4.9881e-05, 4.8373e-04,
         4.3169e-06],
        ...,
        [3.8194e-04, 4.6239e-04, 5.0376e-04,  ..., 1.5490e-01, 5.5147e-03,
         4.8856e-05],
        [3.7292e-06, 4.0716e-06, 4.6404e-06,  ..., 4.0556e-05, 6.6569e-02,
         8.7183e-03],
        [7.2828e-07, 6.6347e-07, 6.0044e-07,  ..., 1.6451e-06, 1.5816e-05,
         3.1921e-02]]), 'X': tensor([[  9.0051,   9.9234,   5.4411],
        [  8.3791,  10.5409,   5.5590],
        [  7.3592,  11.3937,   6.2407],
        [  5.2608,  12.6977,   4.9166],
        [  6.6319,  11.9066,   3.5165],
        [  5.6928,  12.4710,   3.6001],
        [ -5.1974,   6.7396,   3.3484],
        [  5.7218,  -9.9934,   6.5764],
        [ -3.9456,  -9.8542,   0.7466],
        [  4.0421,  -3.7765,  -8.3631],
        [ -5.0550,   6.8231,  -9.2502],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.0467,  -9.8369,  -6.5816],
        [ -6.4844,  -8.0230,   8.5209],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -3.9456,  -9.8542,   0.7466],
        [ 11.4235,   6.3594,   5.1663],
        [ -5.6792,  -8.7352,   4.0318],
        [ 12.2921,  -1.7902,  -4.0870],
        [  3.9895,   6.0816,  -6.5235],
        [  9.1820,   9.7334,  -2.7932],
        [  3.6154,  -4.0640,  -7.9698],
        [  5.9526,   3.2525,  -4.3821],
        [  4.4633,   5.6287,  -0.7754],
        [ -5.3382,   6.6533,   7.8054],
        [ -3.9456,  -9.8542,   0.7466],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.2076,  13.8032,  -1.5099],
        [  6.1579,  -0.5961,  -6.5595],
        [  4.4039,   5.6903,   8.7366],
        [  6.3048,   0.1469,  -9.1691],
        [  3.4401,  13.4540,   4.6576],
        [ -9.1813,  -3.3550,   1.8827],
        [ 12.0042,  -2.8350,   2.6864],
        [  0.6952,  -4.8060,   8.8976],
        [  9.1409,  -7.5789,   1.5814],
        [  0.3493,  -4.7727,  -6.5682],
        [  7.0515,  11.6194,   8.4289],
        [  5.7485,  -9.9811,  -3.2355],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -8.1905,   3.7253,   4.1175],
        [  5.4426, -10.1181,  -5.1268],
        [  3.3094,  13.4969,  -3.6507],
        [ -3.1204, -10.2345,   9.2288],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.1292,   4.8183,  -5.7997],
        [  2.1625,  -4.6697,   7.0658],
        [  5.7784,  -1.6401,   9.5587],
        [  8.3718,  -8.2953,   8.4666],
        [  2.2437,  -4.6489,   6.1430],
        [  2.6614,  -4.5192,  -9.3641],
        [ -1.8823, -10.6558,  -3.8952],
        [ -3.4986,   7.5152,   9.0712],
        [  6.2419,  -0.2313,   4.2440],
        [  1.6954,  -4.7616,   9.8198],
        [ -4.8751,  -9.3148,   4.0406],
        [  3.0719, -10.8449,  -4.1673],
        [ -5.9872,   6.2037,  -9.4477],
        [ 12.5919,   0.4035,  -5.8460],
        [  5.7151,  -1.7700,   3.1542],
        [  6.2975,   0.0947,   5.1791],
        [ -5.9225,  -8.5355,   2.4778],
        [ -9.1749,  -3.3803,   6.4629],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.3691,   6.4760,   1.0787],
        [ -6.0844,  -8.3955,   0.7705],
        [ -5.9987,   6.1949,  -6.2114],
        [ -5.3328,  -8.9992,   7.3440],
        [  4.2339,  -3.6268,   6.3125],
        [  5.8889,  -1.3916,   8.8993],
        [ -3.2053, -10.1993,  -3.8736],
        [  3.4009,  -4.1874,  -5.9751],
        [  1.7218,  -4.7576,  -3.3617],
        [ -8.8745,   2.3097,   3.5600],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.5650,   2.0132,   4.6272],
        [ -5.6423,   6.4537,  -1.2091],
        [ 11.7796,  -3.4696,   9.0088],
        [  5.5101,  -2.1433,   9.5442],
        [  0.9478,  -4.8144,   0.6371],
        [  0.5960, -11.0335,   3.1144],
        [  9.3624,   9.5318,   0.2779],
        [  6.2902,   0.0451,   3.7116],
        [  6.0310,  12.2795,  -6.4528],
        [  2.7685, -10.8982,  -6.8928],
        [ 12.5789,   0.1722,   1.6470],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.4714,  -1.3301,  -3.1998],
        [  4.7482, -10.3907,   5.9149],
        [ -1.6334, -10.7207,   7.1397],
        [  0.7685,  -4.8098,  -2.9310],
        [  9.4516,  -7.2501,  -1.8868],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -5.1046,  -9.1610,   2.3418],
        [ -3.9858,  -9.8334,  -3.7323],
        [  5.6523,   3.9443,  -2.9449],
        [  2.9260,  -4.4162,   8.3808],
        [ -8.8835,  -4.3178,   9.3080],
        [  5.9912,   3.1453,  -3.9989],
        [ 11.0864,   7.0387,  -0.1238],
        [ -3.1204, -10.2345,   9.2288],
        [  1.8719,   7.3713,  -3.7465],
        [  2.4149,  -4.6005,  -4.3339],
        [  3.6703,   6.3443,  -5.4401],
        [ -6.2668,   5.9808,   2.5614],
        [ -9.1795,  -3.3623,   8.5259]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0180, 0.0830,  ..., 1.0229, 0.9559, 1.2395],
        [0.0180, 0.0000, 0.0656,  ..., 1.0409, 0.9706, 1.2489],
        [0.0830, 0.0656, 0.0000,  ..., 1.1053, 1.0123, 1.2676],
        ...,
        [1.0229, 1.0409, 1.1053,  ..., 0.0000, 0.7101, 1.2266],
        [0.9559, 0.9706, 1.0123,  ..., 0.7101, 0.0000, 0.5177],
        [1.2395, 1.2489, 1.2676,  ..., 1.2266, 0.5177, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.40024224, 13.46553623, 13.56367857, 13.744387  , 13.62896593,
       13.70893492,  8.51088913, 11.51557166, 10.61472636,  5.53175275,
        8.49161154,  5.1224079 , 11.54672755, 10.31584102, 12.18045978,
       10.61472636, 13.0743399 , 10.41909211, 12.4217523 ,  7.27340834,
       13.38091121,  5.43943791,  6.78325151,  7.18354885,  8.53014915,
       10.61472636,  9.4997987 , 13.97857591,  6.18668463,  7.19534146,
        6.30647746, 13.88683436,  9.7751189 , 12.3344536 ,  4.85603886,
       11.87412439,  4.78544875, 13.59170809, 11.51811603,  9.48969978,
        8.99791316, 11.4890861 , 13.89671314, 10.69963954,  5.39723241,
        7.03733775,  5.14607014,  6.00663314, 11.785566  ,  5.16205087,
        5.24461068, 10.8207297 ,  8.28968067,  6.24614905,  5.05444288,
       10.51340829, 11.27160035,  8.62162372, 12.59840193,  5.98285319,
        6.29821673, 10.38897563,  9.77777346,  5.60064319, 13.08414689,
       10.36844813,  8.62329466, 10.4606265 ,  5.5748624 ,  6.05113217,
       10.6910768 ,  5.39450305,  5.05963217,  9.17016857,  8.83725081,
       12.72524615,  8.5724042 , 12.27993098,  5.91222346,  4.90676242,
       11.04953767, 13.3607379 ,  6.29034958, 13.68063098, 11.24434374,
       12.58006038,  9.0368458 ,  9.5642993 , 11.42420508, 10.84437705,
        4.87083348, 11.9120321 , 10.39942497, 10.48721382, 10.61047596,
        6.89245352,  5.29755487,  9.87720102,  6.76661595, 13.13205357,
       10.69963954,  7.60529892,  5.19577146,  7.32950385,  8.66273171,
        9.77588801])}
{'P': tensor([[3.4971e-03, 2.5309e-01, 1.9852e-01,  ..., 2.1686e-02, 5.9499e-04,
         1.8650e-07],
        [1.7339e-05, 1.1742e-02, 3.9924e-01,  ..., 2.4205e-02, 5.5487e-04,
         3.9471e-07],
        [3.2901e-05, 3.2846e-04, 4.3086e-02,  ..., 9.4199e-02, 1.2269e-03,
         1.6578e-06],
        ...,
        [3.2192e-06, 2.5930e-05, 8.7347e-05,  ..., 4.6419e-03, 5.5733e-05,
         3.0566e-07],
        [1.3636e-03, 4.0258e-03, 7.2591e-03,  ..., 9.8334e-03, 5.4596e-02,
         2.9595e-05],
        [7.7566e-05, 3.9779e-05, 4.6083e-05,  ..., 9.0387e-05, 2.9230e-04,
         5.8509e-02]]), 'X': tensor([[  8.9284,  10.0036,  -6.9210],
        [  6.9956,  11.6590,  -7.6876],
        [  6.0031,  12.2958,  -6.5992],
        [  6.0310,  12.2795,  -6.4528],
        [  6.1685,  12.1979,  -6.2473],
        [  6.7565,  11.8237,  -4.9882],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.5503,   5.5359,   7.4735],
        [ -9.4378,  -1.8606,  -2.5668],
        [  9.9750,  -6.6338,  -1.2440],
        [ -1.8367,   7.8718,  -8.6523],
        [ -9.4505,  -1.7081,  -0.7039],
        [  1.8766,   7.3694,   1.1887],
        [ 11.0275,  -5.0666,   8.4154],
        [ -5.6169,  -8.7844,   8.9057],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.1787,  -2.6383,  -4.2533],
        [  6.3301,   0.3638,  -7.5685],
        [  9.1820,   9.7334,  -2.7932],
        [  5.0348,  -2.8229,   3.5436],
        [  2.9410,   6.8428,   5.4458],
        [  4.3523, -10.5232,   8.2138],
        [  3.6154,  -4.0640,  -7.9698],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.6757, -11.0361,  -1.3047],
        [  2.0870,  -4.6877,   9.6823],
        [ -7.5603,   4.6459,   8.2002],
        [ -1.7009,   7.8851,  -8.2437],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -8.0087,   4.0173,   9.1091],
        [  3.2647, -10.8066,  -0.3768],
        [  3.9895,   6.0816,  -6.5235],
        [ -7.2515,   5.0195,   4.6434],
        [  2.1477,   7.2550,   8.9688],
        [  3.3094,  13.4969,  -3.6507],
        [  3.5939,   6.4029,  -1.8357],
        [ 11.6634,   5.8050,   9.1590],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.3534,   1.2990,  -7.2942],
        [ -1.7290,   7.8825,  -3.3831],
        [  4.2612, -10.5514,  -6.6268],
        [ -2.3317,   7.8034,  -3.6274],
        [  3.1271, -10.8343,   2.6442],
        [ -0.6323,   7.9085,  -5.9544],
        [ -8.9606,  -4.0994,   1.7886],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.1296,   7.2803,  -2.7194],
        [  3.9663,   6.1018,  -5.4104],
        [ 10.6283,  -5.7253,   7.4878],
        [  7.5784,  -8.9141,   5.8410],
        [  3.2990,  -4.2415,  -7.4857],
        [  6.3303,   1.6145,  -8.4083],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 11.4235,   6.3594,   5.1663],
        [  9.2811,  -7.4336,   5.1771],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.0729,   7.2878,   7.0880],
        [  1.2728, -11.0372,   9.8121],
        [  5.4867,  12.5816,   2.9549],
        [  4.3523, -10.5232,   8.2138],
        [  2.3261,  13.7749,   5.9121],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.2373,  -4.6506,  -1.7334],
        [ -1.1885, -10.8211,   0.0865],
        [  3.9913,  -3.8139,  -6.8311],
        [  5.6202,  -1.9508,   0.7471],
        [ -6.4822,  -8.0252,  -3.8983],
        [  5.8889,  -1.3916,   8.8993],
        [  8.3791,  10.5409,   5.5590],
        [  9.9750,  -6.6338,  -1.2440],
        [  3.3744,   6.5623,  -6.3636],
        [  4.5325,  -3.3643,   6.7699],
        [  7.7794,  11.0629,   8.9069],
        [ -8.1204,  -5.9308,   4.0761],
        [  5.1886,  -2.6250,   6.4677],
        [ 10.4737,  -5.9570,  -1.2108],
        [  1.1006,   7.6318,  -2.4877],
        [  1.6308,  13.9254,  -2.0984],
        [ 12.4592,  -0.9168,  -6.7571],
        [  2.2659,  13.7894,   6.5436],
        [ -8.9282,   2.1648,   8.4894],
        [ 12.2589,  -1.9312,   9.8657],
        [ -8.6229,   2.9068,   2.4429],
        [  0.2571,  -4.7594,   1.5618],
        [  2.7121,   6.9743,  -8.4202],
        [  0.7275,  14.0667,  -0.9815],
        [ -3.9858,  -9.8334,  -3.7323],
        [ 10.2712,   8.3707,  -4.9743],
        [  6.3305,   1.6119,   1.1538],
        [ -8.8835,  -4.3178,   9.3080],
        [ 10.4737,  -5.9570,  -1.2108],
        [  6.3587,   1.1596,  -3.6629],
        [  7.3592,  11.3937,   6.2407],
        [ -4.0409,   7.3169,  -3.5779],
        [ -1.1885, -10.8211,   0.0865],
        [  3.2990,  -4.2415,  -7.4857],
        [ -2.5056,   7.7720,   5.5878],
        [ -7.3554,   4.8985,  -0.6665],
        [ 10.5678,   7.9261,  -0.6100],
        [  4.5491,  -3.3485,   6.8412],
        [  6.3505,   0.6235,  -8.5849],
        [  5.0190,   4.9705,  -7.9752],
        [  4.6063,  13.0051,  -3.1456],
        [  3.9663,   6.1018,  -5.4104],
        [  4.2339,  -3.6268,   6.3125]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0340, 0.0424,  ..., 0.2414, 0.4078, 1.0119],
        [0.0340, 0.0000, 0.0572,  ..., 0.2690, 0.4095, 1.0360],
        [0.0424, 0.0572, 0.0000,  ..., 0.2138, 0.3657, 0.9789],
        ...,
        [0.2414, 0.2690, 0.2138,  ..., 0.0000, 0.3201, 0.7734],
        [0.4078, 0.4095, 0.3657,  ..., 0.3201, 0.0000, 0.7728],
        [1.0119, 1.0360, 0.9789,  ..., 0.7734, 0.7728, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.40850328, 13.59673406, 13.6829843 , 13.68063098, 13.66896739,
       13.61801526,  6.1561646 , 10.01804223,  7.16598273,  9.61942445,
       11.97949273,  8.08320235,  9.60359158,  7.60463349, 12.13568598,
       10.42667093,  9.95552626,  5.81199422,  6.34059021, 13.38091121,
        5.77219077,  7.44805901, 11.38774149,  5.43943791,  5.26067019,
       11.05672136,  5.13125513,  8.8737556 ,  8.06643487, 12.27835996,
        8.95983586, 11.28895489,  7.27340834,  8.81930525,  7.56617037,
       13.89671314,  7.34250948, 13.02818702, 13.15525803,  6.48486171,
        8.06990483, 11.37939695,  8.14433542, 11.27656488,  7.9337647 ,
        9.85384623, 10.60226614,  8.36995953,  7.27759233, 12.07225469,
       11.70016217,  5.37343173,  6.53290663,  9.79039371, 13.0743399 ,
       11.89105259,  6.10774544,  7.57685908, 11.11038551, 13.72594022,
       11.38774149, 13.96987806,  5.24461068,  5.16077764, 10.88618114,
        5.5205189 ,  5.94909709, 10.31613269,  6.05113217, 13.46553623,
       11.97949273,  7.37902766,  5.64464807, 13.52430176, 10.05558893,
        5.81481044, 12.04924159,  7.71075226, 14.02058621, 12.49291768,
       13.97430191,  9.18690018, 12.41012046,  9.09963894,  4.7663526 ,
        7.48311002, 14.08549802, 10.61047596, 13.25017211,  6.53250937,
        9.87720102, 12.04924159,  6.46356806, 13.56367857,  8.35856487,
       10.88618114,  5.37343173,  8.16584537,  8.83725081, 13.20988996,
        5.64865173,  6.38105015,  7.06372172, 13.79676654,  7.27759233,
        5.5748624 ])}
{'P': tensor([[2.2101e-03, 2.9298e-01, 1.5456e-01,  ..., 1.0378e-06, 2.8713e-08,
         3.8239e-08],
        [1.3872e-05, 2.8187e-03, 2.3363e-01,  ..., 1.5675e-06, 4.0712e-08,
         5.6671e-08],
        [3.4931e-06, 4.2675e-06, 3.9159e-03,  ..., 3.9244e-06, 8.2796e-08,
         1.2421e-07],
        ...,
        [7.0753e-07, 7.4090e-07, 7.8367e-07,  ..., 8.4135e-02, 3.7742e-02,
         3.2887e-02],
        [6.1482e-07, 5.8924e-07, 3.4817e-07,  ..., 2.1849e-05, 4.9900e-02,
         8.0219e-05],
        [2.5770e-07, 2.6404e-07, 2.1729e-07,  ..., 6.2588e-04, 6.5350e-02,
         1.4046e-01]]), 'X': tensor([[ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 1.2222e+01,  4.0953e+00, -8.2072e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [-9.3411e+00, -2.6004e+00, -2.7062e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 2.8602e+00,  6.8905e+00,  5.8779e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 1.2522e+01, -4.4969e-01,  4.8448e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-1.6275e+00, -1.0722e+01,  3.2376e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0087, 0.0265,  ..., 1.3777, 1.5298, 1.7047],
        [0.0087, 0.0000, 0.0247,  ..., 1.3694, 1.5217, 1.6962],
        [0.0265, 0.0247, 0.0000,  ..., 1.3613, 1.5122, 1.6903],
        ...,
        [1.3777, 1.3694, 1.3613,  ..., 0.0000, 0.1750, 0.3591],
        [1.5298, 1.5217, 1.5122,  ..., 0.1750, 0.0000, 0.3072],
        [1.7047, 1.6962, 1.6903,  ..., 0.3591, 0.3072, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.41713661, 13.45124917, 13.66051884, 13.59056613, 13.6922425 ,
       13.59673406,  8.83341608, 10.60226614,  5.5205189 , 12.889695  ,
       10.1371552 ,  7.88677538, 12.3489324 ,  7.70897675,  8.74470218,
       12.87752052,  4.89121805,  8.09842704, 10.60226614,  7.54620849,
        5.64865173,  9.27474386,  5.82345867,  6.81961996,  9.37989631,
        8.06990483, 13.04051744,  8.98765644, 12.77402351, 10.06850677,
       11.22596994,  7.37902766,  9.75075273,  9.60359158, 13.72594022,
        4.91652556, 13.3614598 ,  5.70405525,  7.23276582, 13.66051884,
       10.76154256,  9.14973319,  8.92098752,  8.31625171,  7.20482224,
       13.00413829,  9.69629   ,  9.43737523,  6.41202986,  5.38342077,
        5.13125513, 13.1471313 ,  6.24614905, 10.48840476,  4.78544875,
        7.20482224, 11.9702762 ,  6.53290663,  9.95552626,  7.46053281,
        6.02588517,  7.5158489 , 11.05413545,  5.16611782, 11.70016217,
        7.19513806, 13.14690137,  6.53290663,  5.83560009, 13.58333857,
        7.81509631,  8.06990483, 12.53047489,  8.51928531,  9.75200848,
       12.79126785, 10.84493732,  7.16747878,  7.18354885,  8.67227386,
        8.91551231, 14.03618649, 13.96905756,  7.72415048, 11.74901778,
       12.49317563,  9.14973319,  6.39520397,  7.92200342,  7.32950385,
        9.10945678, 11.63724792, 13.25017211, 13.40850328,  7.15100303,
       10.4606265 ,  6.81961996,  5.53175275,  9.47369628, 11.89105259,
        7.50500064, 11.64436684, 10.71162681,  9.17016857, 10.26327895,
        9.64157461])}
{'P': tensor([[2.9718e-03, 3.1400e-01, 2.5946e-01,  ..., 3.4441e-06, 4.5369e-06,
         1.4772e-03],
        [1.7794e-05, 7.5995e-03, 5.9173e-02,  ..., 1.2847e-05, 1.9490e-05,
         4.6292e-03],
        [8.0950e-06, 2.0269e-04, 4.7440e-03,  ..., 5.1202e-06, 1.2049e-05,
         2.7962e-03],
        ...,
        [7.9542e-07, 1.1949e-06, 9.7559e-07,  ..., 4.0512e-02, 8.4227e-04,
         2.0996e-05],
        [7.5145e-07, 7.8938e-07, 7.2355e-07,  ..., 1.7286e-05, 3.8795e-02,
         2.0708e-05],
        [3.6374e-05, 5.8454e-05, 7.0918e-05,  ..., 9.0210e-05, 3.9493e-03,
         4.8038e-02]]), 'X': tensor([[ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 3.3167e+00,  6.6022e+00,  6.3554e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.1680e+01, -3.7196e+00, -9.1810e-01],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [-9.4203e+00, -2.0340e+00,  2.5856e+00],
        [ 9.4925e+00,  9.3811e+00,  8.9842e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-9.4117e+00,  1.1605e-01,  8.2574e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 5.0233e+00,  4.9647e+00, -3.1788e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 3.9085e+00,  6.1512e+00, -2.2560e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-2.6735e+00, -1.0406e+01, -1.5911e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.8529e-02, 4.4197e-02,  ..., 1.4737e+00, 1.3174e+00,
         6.5108e-01],
        [1.8529e-02, 0.0000e+00, 6.2162e-02,  ..., 1.4570e+00, 1.3075e+00,
         6.4260e-01],
        [4.4197e-02, 6.2162e-02, 0.0000e+00,  ..., 1.5174e+00, 1.3514e+00,
         6.8294e-01],
        ...,
        [1.4737e+00, 1.4570e+00, 1.5174e+00,  ..., 0.0000e+00, 7.7545e-01,
         1.0261e+00],
        [1.3174e+00, 1.3075e+00, 1.3514e+00,  ..., 7.7545e-01, 2.1453e-04,
         6.7095e-01],
        [6.5108e-01, 6.4260e-01, 6.8294e-01,  ..., 1.0261e+00, 6.7095e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.43127131, 13.54774355, 13.59170809, 13.76047999, 13.68672855,
       13.52430176, 12.59840193,  5.39450305, 12.07225469,  8.7907923 ,
        5.68942572, 10.51437947, 13.66896739,  7.10378919, 13.35832725,
        9.85384623,  7.16747878, 11.22142195,  7.26074925,  4.98989906,
        7.50917551, 11.74901778,  7.73692436,  8.29022197,  9.09502385,
        6.29821673,  4.978704  ,  7.38844703, 12.4013153 , 10.38897563,
        9.50654967,  9.5642993 ,  6.39187285, 12.25806559,  7.5158489 ,
        5.38342077, 12.11735334,  5.05774295, 11.86239961,  5.8828986 ,
        9.63742669, 13.34587027,  7.39599986,  7.15100303,  8.41171369,
       10.54658597, 11.28215429, 13.85158417,  5.56809335,  6.2322769 ,
        7.27759233, 13.0743399 , 10.51437947, 10.09979747,  9.78147542,
       10.7506423 ,  6.89245352, 13.6922425 , 12.18045978,  6.29034958,
        7.76457095,  6.80370254, 10.42667093,  5.30158497,  9.41244865,
        5.82345867,  7.06272256,  5.23878371, 10.71162681, 11.23201619,
        5.91854397,  5.21280156,  8.31625171,  6.65368551,  5.53175275,
       13.40850328,  7.28793663,  9.5642993 ,  9.20340031,  5.43943791,
       12.05395056, 13.45124917, 11.00812996, 10.74409834, 13.00380624,
        7.20482224, 10.52357479,  7.05627461,  7.52117132, 10.67322527,
        7.24677664, 12.24467041, 12.85032836,  9.51074763,  7.37902766,
        9.76896904, 13.13888841, 12.26997165,  5.27178363,  8.09842704,
       13.744387  ,  8.89458273, 10.05558893,  8.29203356,  9.25336371,
        7.89745107])}
{'P': tensor([[3.0564e-03, 3.8886e-01, 2.8194e-01,  ..., 1.9064e-08, 1.5789e-05,
         6.2239e-08],
        [1.6520e-05, 6.9989e-03, 1.4598e-02,  ..., 3.5744e-08, 1.2691e-05,
         1.0222e-07],
        [2.4148e-05, 1.8768e-03, 9.4507e-03,  ..., 5.2660e-08, 1.0224e-05,
         1.3731e-07],
        ...,
        [2.0086e-05, 1.1261e-05, 1.2757e-05,  ..., 8.1792e-03, 4.1482e-05,
         1.7747e-02],
        [2.0931e-01, 1.8712e-01, 3.3881e-01,  ..., 3.8070e-08, 4.5648e-03,
         1.3566e-07],
        [1.5591e-03, 8.2236e-04, 8.7358e-04,  ..., 1.3011e-05, 3.2950e-03,
         1.7878e-02]]), 'X': tensor([[  8.5716,  10.3596,   6.8818],
        [  7.5309,  11.2617,   7.2656],
        [  7.3592,  11.3937,   6.2407],
        [  4.8674,  12.8875,   6.7622],
        [  5.0617,  12.7957,   8.3544],
        [  7.0515,  11.6194,   8.4289],
        [  0.5850,  -4.7982,   9.6639],
        [  8.6420,  -8.0580,   6.5781],
        [  1.7122,  -4.7591,  -3.6734],
        [  9.4516,  -7.2501,  -1.8868],
        [ -3.4488, -10.0934,   2.5195],
        [  5.1203,  -2.7151,   6.2110],
        [  5.6672,  -1.8631,   8.1020],
        [ -7.2515,   5.0195,   4.6434],
        [  6.0054,  -1.0902,   8.3079],
        [ -9.4735,  -1.2620,  -3.6742],
        [  6.2675,  12.1379,  -8.9720],
        [  5.1203,  -2.7151,   6.2110],
        [  2.0243,   7.3086,   7.5432],
        [ -2.0445,   7.8469,  -0.5857],
        [  1.6422,   7.4586,  -6.8056],
        [ -7.5720,   4.6310,   2.1717],
        [ -5.3382,   6.6533,   7.8054],
        [  5.8275,  -1.5334,   2.0124],
        [ -9.4501,  -1.7133,   2.2980],
        [ -7.4807,   4.7463,   2.0616],
        [  2.0449,  13.8402,  -7.1642],
        [  0.1252, -11.0062,   7.2564],
        [ -4.8751,  -9.3148,   4.0406],
        [  5.1861,   4.7361,  -3.7312],
        [  9.4516,  -7.2501,  -1.8868],
        [ -3.1204, -10.2345,   9.2288],
        [ -1.0407,   7.9166,  -6.6021],
        [ -5.7347,  -8.6908,   9.2878],
        [  6.1741,  -0.5329,   3.1913],
        [  9.6614,   9.1784,  -1.2087],
        [  2.0899,  -4.6870,  -7.4260],
        [  5.9526,   3.2525,  -4.3821],
        [  2.4572,   7.1085,   7.7376],
        [ 12.5763,   0.1333,  -9.2650],
        [  3.4140,  -4.1803,  -8.8224],
        [  7.8425,  -8.7202,   6.4530],
        [  3.9770, -10.6341,   8.0783],
        [  5.1886,  -2.6250,   6.4677],
        [ -8.6229,   2.9068,   2.4429],
        [  3.1892, -10.8220,  -6.8409],
        [  0.4996,   7.7738,  -4.4711],
        [  2.7685, -10.8982,  -6.8928],
        [ -8.7544,   2.6096,   3.3146],
        [ -0.3432,   7.8900,   7.1805],
        [  1.4306,  13.9620,  -0.2075],
        [ -9.3739,   0.3982,  -4.2564],
        [ -6.4620,  -8.0449,   6.2719],
        [  4.5340,  -3.3629,   8.9652],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.0560,  -0.9421,  -5.0334],
        [  7.0358,  -9.2784,  -9.8479],
        [ -7.3554,   4.8985,  -0.6665],
        [ 12.5224,  -0.4497,   4.8448],
        [ -4.0633,  -9.7927,  -3.0825],
        [  6.1579,  -0.5961,  -6.5595],
        [  9.8082,  -6.8396,   4.9615],
        [ -4.2271,   7.2387,  -1.8316],
        [ -7.3470,  -7.0615,   5.0066],
        [  0.9809,  -4.8145,   6.8706],
        [ -0.0254, -10.9932,   3.2200],
        [ -5.3179,  -9.0101,   9.4472],
        [ -8.1531,  -5.8748,   2.6934],
        [ -6.4368,   5.8356,   3.3014],
        [ -5.6423,   6.4537,  -1.2091],
        [  4.8051, -10.3703,   0.5268],
        [  5.8889,  -1.3916,   8.8993],
        [ 11.3367,  -4.4806,   0.4643],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -8.1905,   3.7253,   4.1175],
        [  8.3718,  -8.2953,   8.4666],
        [  5.1861,   4.7361,  -3.7312],
        [  5.9526,   3.2525,  -4.3821],
        [  1.7903,  -4.7467,   0.1168],
        [ -1.9714,   7.8563,  -1.8084],
        [  5.0506,   4.9277,  -9.5094],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -9.4721,  -1.3097,  -7.4377],
        [  5.4429,  -2.2529,   9.7306],
        [ -7.8280,   4.2848,  -7.1955],
        [ -6.8466,  -7.6488,   1.9648],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -0.0282,   7.8575,  -1.1746],
        [ 11.4703,  -4.2006,   7.4086],
        [  0.5250, -11.0306,   5.5358],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4157,   0.0821,  -5.0025],
        [  7.8982,  -8.6778,   0.0239],
        [  7.9816,  10.8938,  -0.5378],
        [  6.0305,   3.0296,   0.1375],
        [  4.0587, -10.6112,   6.2273],
        [  1.9505,  -4.7171,   9.9868],
        [  9.5203,   9.3483,   5.8781],
        [  7.1821,  -9.1845,   9.8548],
        [ -6.6998,   5.5954,   0.0823],
        [  4.0362,  13.2397,  -7.0030],
        [  7.1441,  11.5529,   2.3232],
        [  6.9314,  -9.3435,  -1.2729],
        [  9.5203,   9.3483,   5.8781],
        [ 11.2517,  -4.6495,   2.9800]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0242, 0.0368,  ..., 1.0847, 0.0944, 0.6162],
        [0.0242, 0.0000, 0.0609,  ..., 1.1068, 0.1184, 0.6378],
        [0.0368, 0.0609, 0.0000,  ..., 1.0530, 0.0587, 0.5857],
        ...,
        [1.0847, 1.1068, 1.0530,  ..., 0.0000, 0.9958, 0.4706],
        [0.0944, 0.1184, 0.0587,  ..., 0.9958, 0.0000, 0.5298],
        [0.6162, 0.6378, 0.5857,  ..., 0.4706, 0.5298, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.44593519, 13.54774355, 13.56367857, 13.77604293, 13.76047999,
       13.59170809,  4.83371341, 11.8159254 ,  5.05774295, 11.9120321 ,
       10.66631784,  5.79562691,  5.96556248,  8.81930525,  6.10360073,
        9.55720651, 13.66051884,  5.79562691,  7.58378655,  8.10886021,
        7.63726134,  8.87587457,  8.53014915,  6.02588517,  9.60412735,
        8.85939626, 13.99048094, 11.00694688, 10.51340829,  7.02326515,
       11.9120321 , 10.69963954,  7.98469166, 10.412299  ,  6.19707944,
       13.32613445,  5.13182506,  6.78325151,  7.52117132, 12.5769659 ,
        5.39723241, 11.72802491, 11.35345345,  5.81481044,  9.09963894,
       11.28215429,  7.78980299, 11.24434374,  9.13507321,  7.89745107,
       14.03505624,  9.38232121, 10.31886066,  5.64500512, 12.26997165,
        6.12886024, 11.64436684,  8.83725081, 12.53047489, 10.60226614,
        6.18668463, 11.95743558,  8.38252937, 10.19036081,  4.91336994,
       10.99326577, 10.46238054, 10.04916678,  8.68832398,  8.5724042 ,
       11.42947632,  6.05113217, 12.18998195, 10.06407839,  8.99791316,
       11.785566  ,  7.02326515,  6.78325151,  5.07306274,  8.09984231,
        7.05627461, 11.9120321 ,  9.60359158,  9.562174  ,  5.89074104,
        8.92395381, 10.26545792, 12.25806559,  7.8575648 , 12.21532282,
       11.04313641,  9.60080639,  9.41605684, 11.73397309, 13.50484539,
        6.74872981, 11.36089355,  5.10446653, 13.3426541 , 11.65923396,
        8.72896381, 13.84126197, 13.58333857, 11.63383221, 13.3426541 ,
       12.17451685])}
{'P': tensor([[2.3220e-03, 1.9246e-01, 1.6311e-01,  ..., 7.9879e-03, 2.9532e-06,
         1.4926e-08],
        [1.6798e-06, 1.5414e-03, 3.8198e-01,  ..., 3.0221e-03, 9.0524e-07,
         6.0896e-09],
        [5.0411e-06, 1.5649e-05, 5.6063e-03,  ..., 9.3483e-03, 2.9626e-06,
         2.1090e-08],
        ...,
        [8.6818e-05, 6.3856e-04, 9.4433e-04,  ..., 1.8581e-01, 4.4598e-05,
         8.7565e-07],
        [2.7198e-02, 1.6683e-02, 1.5307e-02,  ..., 5.2153e-03, 1.3498e-01,
         8.4647e-05],
        [4.1564e-05, 1.4326e-05, 1.2501e-05,  ..., 1.1590e-05, 2.0572e-04,
         9.8087e-03]]), 'X': tensor([[  8.5198,  10.4091,  -8.6486],
        [  6.2675,  12.1379,  -8.9720],
        [  5.8930,  12.3592,  -8.9201],
        [  6.9956,  11.6590,  -7.6876],
        [  7.0642,  11.6104,  -9.8316],
        [  3.7298,  13.3537,  -9.4653],
        [ -8.0965,  -5.9712,   4.3466],
        [  0.5250, -11.0306,   5.5358],
        [ -5.1974,   6.7396,   3.3484],
        [ -9.4505,  -1.7081,  -0.7039],
        [ -7.8280,   4.2848,  -7.1955],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -9.2508,   1.0572,   2.9978],
        [  0.6757, -11.0361,  -1.3047],
        [  6.2883,   0.0325,   5.2692],
        [ -8.3717,  -5.4769,  -9.2228],
        [  1.1006,   7.6318,  -2.4877],
        [  1.5383, -11.0274,  -4.9666],
        [ 11.6634,   5.8050,   9.1590],
        [  2.5375,   7.0676,  -4.3490],
        [ -8.6603,   2.8250,   5.9181],
        [  4.0587, -10.6112,   6.2273],
        [ 11.5935,   5.9740,  -4.3952],
        [ -9.0961,   1.6529,   3.1553],
        [  9.5018,  -7.1945,  -1.1892],
        [  0.6363, -11.0348,   6.1414],
        [ 11.6019,   5.9540,  -5.1715],
        [  6.2036,   2.4030,  -1.2500],
        [  2.1477,   7.2550,   8.9688],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.2026,  12.1774,  -1.5887],
        [ -4.7389,   6.9953,   3.7638],
        [  0.8700,  -4.8132,  -7.8042],
        [  0.3354,  -4.7708,  -4.9125],
        [  0.9074,  -4.8139,   8.1674],
        [ -9.3705,   0.4208,  -3.5815],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.0031,  12.2958,  -6.5992],
        [  2.7121,   6.9743,  -8.4202],
        [  9.0004,  -7.7195,   4.6945],
        [ -2.2188,   7.8218,   8.6064],
        [  5.7802,  -1.6363,  -6.2782],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -1.0012, -10.8576,   2.2867],
        [  9.5261,  -7.1675,   1.4206],
        [  7.0171,  11.6439,  -5.0178],
        [  9.4925,   9.3811,   8.9842],
        [ -9.3616,  -2.4750,   3.7765],
        [  5.7845,   3.6652,   0.9175],
        [  2.4149,  -4.6005,  -4.3339],
        [  9.3836,   9.5075,  -5.9278],
        [ 12.2218,   4.0953,  -8.2072],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.3331,  -2.6465,  -8.2028],
        [  5.2199,   4.6861,  -1.9304],
        [ -3.7964,  -9.9294,   7.7147],
        [ -5.1369,   6.7755,   8.3836],
        [  6.3305,   1.6119,   1.1538],
        [ 12.4877,  -0.7222,   8.8347],
        [  2.7295,  13.6703,  -8.9328],
        [ -2.9497, -10.3028,  -7.3479],
        [ -0.8675, -10.8815,   6.8561],
        [ -8.8934,   2.2597,   3.5831],
        [  4.5503,   5.5359,   7.4735],
        [  3.9591, -10.6391,  -6.2557],
        [  3.6249, -10.7255,   5.6605],
        [  6.2038,   2.4020,  -3.1692],
        [  6.9693,  -9.3200,  -8.3822],
        [ -6.0195,   6.1789,  -2.9496],
        [ 11.7162,   5.6728,  -8.9372],
        [  7.1821,  -9.1845,   9.8548],
        [  8.0381,  -8.5690,  -8.9523],
        [  9.9058,  -6.7203,   5.6928],
        [  2.2437,  -4.6489,   6.1430],
        [ -7.0825,   5.2073,  -5.0497],
        [  9.8082,  -6.8396,   4.9615],
        [  5.8930,  12.3592,  -8.9201],
        [  4.2041,  -3.6509,  -4.9488],
        [ -8.5859,  -5.0393,   0.5476],
        [  0.4996,   7.7738,  -4.4711],
        [  9.6662,   9.1725,   8.1056],
        [  5.4242, -10.1261,   3.9595],
        [ -2.1382,   7.8339,   7.3524],
        [  9.6662,   9.1725,   8.1056],
        [ -9.4766,  -1.1116,  -2.1706],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.4157,   0.0821,  -5.0025],
        [  1.4306,  13.9620,  -0.2075],
        [ -3.9508,  -9.8515,   1.7923],
        [  5.3858,   4.4260,  -9.6769],
        [  1.7815,  -4.7481,   4.4289],
        [  3.3167,   6.6022,   6.3554],
        [ -5.6423,   6.4537,  -1.2091],
        [  1.9503,  -4.7172,  -6.2313],
        [ 12.0845,  -2.5772,  -4.6073],
        [ -8.0829,  -5.9940,   5.0564],
        [ 12.3453,  -1.5462,   7.0212],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.0031,  12.2958,  -6.5992],
        [  6.1989,  -9.7591,   0.8289],
        [ -8.0750,  -6.0072,  -5.8267],
        [  8.1528,  10.7452,   9.1493],
        [ -6.1899,  -8.3011,  -7.9269],
        [  3.2809,  13.5061,  -4.0199],
        [  6.3589,   0.8239,  -8.4230],
        [  9.1409,  -7.5789,   1.5814]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0247, 0.0371,  ..., 0.2491, 0.3340, 0.8894],
        [0.0247, 0.0000, 0.0137,  ..., 0.2583, 0.3248, 0.8881],
        [0.0371, 0.0137, 0.0000,  ..., 0.2570, 0.3143, 0.8807],
        ...,
        [0.2491, 0.2583, 0.2570,  ..., 0.0000, 0.2274, 0.6670],
        [0.3340, 0.3248, 0.3143,  ..., 0.2274, 0.0000, 0.5868],
        [0.8894, 0.8881, 0.8807,  ..., 0.6670, 0.5868, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.45124917, 13.66051884, 13.6922425 , 13.59673406, 13.59056613,
       13.86480175, 10.06023665, 11.04313641,  8.51088913,  9.60359158,
        8.92395381, 10.52587243,  9.3109884 , 11.05672136,  6.28835101,
       10.00410625,  7.71075226, 11.1341798 , 13.02818702,  7.5092834 ,
        9.10945678, 11.36089355, 13.04217487,  9.24501829, 11.9182956 ,
       11.05317597, 13.04051744,  6.652742  ,  7.56617037, 10.06703358,
       13.6660671 ,  8.44940661,  4.89121805,  4.78258092,  4.89869205,
        9.37989631,  9.58072286, 13.6829843 ,  7.48311002, 11.85743425,
        8.1303945 ,  6.00732193,  9.50913849, 10.9036226 , 11.92133129,
       13.59480906, 13.34587027,  9.68324553,  6.84796202,  5.19577146,
       13.35832725, 12.889695  ,  8.22728399,  9.70108409,  7.01475376,
       10.63038565,  8.50268322,  6.53250937, 12.5085995 , 13.94008948,
       10.7167285 , 10.91601655,  9.17595899,  7.16598273, 11.35182589,
       11.32149259,  6.65259775, 11.63765273,  8.62631801, 13.01728313,
       11.65923396, 11.74901778, 11.9702762 ,  5.16205087,  8.7907923 ,
       11.95743558, 13.6922425 ,  5.56809335,  9.95552626,  7.78980299,
       13.32557133, 11.48734978,  8.12043983, 13.32557133,  9.54154096,
        5.24461068,  9.41605684, 14.03505624, 10.61417181,  6.97106912,
        5.0713364 ,  7.38844703,  8.5724042 ,  5.10444097, 12.35625694,
       10.06286074, 12.44177641,  5.94541803, 13.6829843 , 11.56146873,
       10.06437598, 13.48809595, 10.35485331, 13.89886469,  6.41202986,
       11.87412439])}
{'P': tensor([[2.1669e-03, 2.8235e-01, 2.4672e-01,  ..., 6.1691e-07, 3.3870e-05,
         7.9762e-08],
        [3.9624e-06, 2.5026e-03, 6.4350e-01,  ..., 5.0180e-07, 1.9522e-05,
         9.0299e-08],
        [1.2048e-05, 2.5824e-05, 8.6771e-03,  ..., 1.6513e-06, 6.0370e-05,
         3.1459e-07],
        ...,
        [1.9853e-03, 1.4216e-03, 1.3461e-03,  ..., 8.5311e-02, 4.8005e-02,
         5.1704e-04],
        [1.4808e-03, 2.3815e-03, 2.5652e-03,  ..., 2.8130e-04, 5.1514e-02,
         8.7371e-05],
        [2.7771e-05, 2.1000e-05, 2.0170e-05,  ..., 8.3952e-03, 1.1563e-03,
         1.2145e-01]]), 'X': tensor([[  8.4691,  10.4571,  -5.0650],
        [  7.0171,  11.6439,  -5.0178],
        [  6.7565,  11.8237,  -4.9882],
        [  6.1685,  12.1979,  -6.2473],
        [  6.0310,  12.2795,  -6.4528],
        [  6.0031,  12.2958,  -6.5992],
        [ 10.9875,   7.2207,  -8.8549],
        [ -8.7078,   2.7182,   1.8146],
        [  5.1292,   4.8183,  -5.7997],
        [  5.3860,  -2.3418,   1.0826],
        [  6.3048,   0.1469,  -9.1691],
        [ -5.3328,  -8.9992,   7.3440],
        [ -3.2736, -10.1703,   3.8712],
        [ -8.5149,  -5.1905,   1.5239],
        [  3.3633,  13.4794,   0.5193],
        [ 12.5832,   1.7543,  -7.6134],
        [ -8.3535,  -5.5118,   9.8399],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.5469,  13.7192,  -9.6935],
        [ 11.1469,   6.9237,   1.6160],
        [  2.5829,  -4.5466,  -8.4015],
        [ 10.0619,   8.6627,  -9.2884],
        [ -3.5173,   7.5091,  -8.9585],
        [ 10.6283,  -5.7253,   7.4878],
        [ -8.0611,  -6.0302,   4.6962],
        [  4.2096,   5.8815,   1.9230],
        [ -5.3736,   6.6310,  -1.5174],
        [  5.1890,   4.7318,   0.9515],
        [ -3.9858,  -9.8334,  -3.7323],
        [  6.1685,  12.1979,  -6.2473],
        [ 12.4901,  -0.7052,   1.7315],
        [  5.0506,   4.9277,  -9.5094],
        [ -7.2515,   5.0195,   4.6434],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.2975,   0.0947,   5.1791],
        [  0.4996,   7.7738,  -4.4711],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 12.6038,   0.7665,  -3.4439],
        [ -8.5859,  -5.0393,   0.5476],
        [  2.6970,   6.9826,   9.7350],
        [ 11.0891,  -4.9558,  -2.3178],
        [  6.0073,   3.0986,   6.7937],
        [  0.2480, -11.0153,  -3.0183],
        [ -4.1296,   7.2803,  -2.7194],
        [ -7.7842,   4.3466,  -1.7548],
        [  4.3459,  -3.5327,   0.3667],
        [  2.1120,  -4.6818,   8.6144],
        [  4.5491,  -3.3485,   6.8412],
        [ 12.6000,   1.3839,  -3.2992],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.8400,   2.3991,   1.3457],
        [  6.1222,  -0.7261,   8.8204],
        [  3.0719, -10.8449,  -4.1673],
        [ -9.4739,  -1.2497,   0.5483],
        [  4.3459,  -3.5327,   0.3667],
        [  6.0345,   3.0175,  -1.4424],
        [  6.0345,   3.0175,  -1.4424],
        [  4.2275,  -3.6320,  -0.2714],
        [  0.6142,  14.0802,   0.5407],
        [  2.3200,   7.1756,   1.1938],
        [  2.3490,  -4.6199,  -4.2489],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.1793,   4.7461,   0.3899],
        [ 12.2921,  -1.7902,  -4.0870],
        [ 11.6019,   5.9540,  -5.1715],
        [ -1.0407,   7.9166,  -6.6021],
        [ 11.0891,  -4.9558,  -2.3178],
        [  8.2422,  -8.4041,  -1.4962],
        [ -9.1795,  -3.3623,   8.5259],
        [  4.4915,   5.5990,   7.0405],
        [  1.4150,  13.9647,  -4.0915],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.6078, -10.9229,   6.2869],
        [  6.2242,  -0.3171,   7.4978],
        [  0.2480, -11.0153,  -3.0183],
        [ -8.3653,   3.4184,   9.3454],
        [  5.3860,  -2.3418,   1.0826],
        [  3.2701,   6.6338,   9.9209],
        [  9.4554,   9.4245,   9.8800],
        [  6.3549,   0.7096,   6.9198],
        [ -9.1813,  -3.3550,   1.8827],
        [  4.1297,  -3.7098,  -3.9838],
        [  5.7218,  -9.9934,   6.5764],
        [ -8.8058,   2.4850,  -1.7425],
        [ 11.7726,  -3.4876,  -6.7402],
        [  4.5430,   5.5438,  -3.8669],
        [  9.2979,   9.6048,   7.3138],
        [  2.5636, -10.9293,   9.6519],
        [  6.3460,   1.4252,   1.2291],
        [  4.8653,   5.1694,  -4.0823],
        [  6.1222,  -0.7261,   8.8204],
        [  7.5309,  11.2617,   7.2656],
        [ -6.4464,   5.8272,   7.5038],
        [  5.9548,  -1.2273,  -2.6520],
        [  3.2809,  13.5061,  -4.0199],
        [ -1.3178, -10.7939,   0.4236],
        [  5.1667,  -2.6543,   1.2879],
        [ -7.8359,   4.2735,   7.2943],
        [ -7.4808,   4.7462,   2.0266],
        [ -8.4267,  -5.3697,   2.4172],
        [ -7.5720,   4.6310,   2.1717],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.2535,  -0.1709,  -9.4102],
        [  5.7963,  -1.6018,   1.8856],
        [  4.8432,   5.1968,   1.5006],
        [  0.6952,  -4.8060,   8.8976]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0133, 0.0209,  ..., 0.6238, 0.6427, 1.0203],
        [0.0133, 0.0000, 0.0089,  ..., 0.6116, 0.6309, 1.0075],
        [0.0209, 0.0089, 0.0000,  ..., 0.6031, 0.6222, 0.9994],
        ...,
        [0.6238, 0.6116, 0.6031,  ..., 0.0000, 0.0442, 0.4079],
        [0.6427, 0.6309, 0.6222,  ..., 0.0442, 0.0000, 0.4056],
        [1.0203, 1.0075, 0.9994,  ..., 0.4079, 0.4056, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.45642379, 13.59480906, 13.61801526, 13.66896739, 13.68063098,
       13.6829843 , 13.14777433,  9.12220271,  7.03733775,  5.87305005,
        6.30647746, 10.4606265 , 10.68416401,  9.97220799, 13.89264822,
       12.70489563, 10.00802627,  8.23525433, 13.95361098, 13.12218013,
        5.22903117, 13.27717982,  8.29203356, 12.07225469, 10.06703358,
        7.23276582,  8.53501777,  7.02253235, 10.61047596, 13.66896739,
       12.50996828,  7.05627461,  8.81930525,  5.22903117,  6.29821673,
        7.78980299, 10.14436912, 12.62711252,  9.95552626,  7.48538905,
       12.1460931 ,  6.7593917 , 11.018084  ,  8.36995953,  8.91551231,
        5.60064319,  5.13616224,  5.64865173, 12.67576792, 10.24557047,
        9.15976746,  6.16512856, 11.27160035,  9.55593374,  5.60064319,
        6.74686071,  6.74686071,  5.57341623, 14.09357547,  7.54127425,
        5.18277858,  8.25828409,  7.02496223, 12.4217523 , 13.04051744,
        7.98469166, 12.1460931 , 11.77124373,  9.77588801,  7.17789666,
       14.03618649,  9.51074763, 11.22993218,  6.2322769 , 11.018084  ,
        9.0368458 ,  5.87305005,  7.39599986, 13.35013374,  6.39438044,
        9.7751189 ,  5.5513072 , 11.51557166,  9.14973319, 12.27835996,
        7.16747878, 13.36800073, 11.22596994,  6.50409372,  7.09887557,
        6.16512856, 13.54774355,  8.68979052,  6.07992623, 13.89886469,
       10.87409005,  5.80860165,  8.92549752,  8.859416  ,  9.99211663,
        8.87587457,  9.96391045,  6.25586297,  6.01356004,  7.10378919,
        4.85603886])}
{'P': tensor([[4.0908e-03, 3.3677e-01, 2.1353e-01,  ..., 6.4573e-07, 1.0251e-05,
         2.7352e-06],
        [1.6918e-05, 4.8954e-03, 2.2990e-01,  ..., 6.8895e-07, 7.6656e-06,
         2.4145e-06],
        [4.5202e-06, 1.2078e-05, 5.9975e-03,  ..., 9.6639e-07, 4.6271e-06,
         1.7308e-06],
        ...,
        [1.5859e-04, 1.0920e-04, 8.2569e-05,  ..., 9.3176e-02, 6.1855e-03,
         5.9921e-03],
        [2.3016e-02, 1.9258e-02, 1.5704e-02,  ..., 3.7931e-04, 9.2292e-02,
         3.0894e-03],
        [8.3647e-03, 6.5415e-03, 3.5024e-03,  ..., 3.4842e-04, 1.0956e-01,
         7.1486e-02]]), 'X': tensor([[  8.3791,  10.5409,   5.5590],
        [  7.3592,  11.3937,   6.2407],
        [  5.2608,  12.6977,   4.9166],
        [  4.8674,  12.8875,   6.7622],
        [  3.9022,  13.2906,   5.5788],
        [  5.6928,  12.4710,   3.6001],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.4296,  -1.9458,   4.0833],
        [  7.5064,  -8.9651,  -0.0994],
        [ 11.9122,   5.1397,  -1.5498],
        [  0.3724,   7.7974,  -6.8722],
        [  2.0419,  -4.6978,  -5.0983],
        [ 12.3357,   3.6001,   0.7903],
        [ 12.2435,   4.0076,   8.1257],
        [  5.3860,  -2.3418,   1.0826],
        [ -8.7409,  -4.6837,  -6.5840],
        [  8.5716,  10.3596,   6.8818],
        [  1.5662,  13.9375,   4.5142],
        [  7.1821,  -9.1845,   9.8548],
        [  2.5663,   7.0526,  -6.8769],
        [ -3.1671,   7.6158,   6.0765],
        [ 12.0042,  -2.8350,   2.6864],
        [ 11.4235,   6.3594,   5.1663],
        [  6.3301,   0.3638,  -7.5685],
        [ -3.9508,  -9.8515,   1.7923],
        [ -9.3616,  -2.4749,   0.1793],
        [  0.7275,  14.0667,  -0.9815],
        [ -7.6734,   4.4980,  -5.4248],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.9019,  -3.1379,  -3.4534],
        [  2.9461,  -4.4077,  -0.4831],
        [ -5.9938,  -8.4745,   3.9735],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.3424,  -7.3685,  -3.6807],
        [ -3.1204, -10.2345,   9.2288],
        [  9.3836,   9.5075,  -5.9278],
        [  6.2630,   2.1057,  -5.2136],
        [ 11.5456,   6.0861,   8.1101],
        [ -4.0868,   7.2981,   8.1874],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.1667,  -2.6543,   1.2879],
        [ -8.4267,  -5.3697,   2.4172],
        [ -7.2515,   5.0195,   4.6434],
        [  5.6314,  -1.9302,  -9.6543],
        [ -6.8618,  -7.6322,   3.7027],
        [ -9.4671,  -0.5553,   0.1117],
        [  2.7420,  -4.4896,  -3.7139],
        [  0.1252, -11.0062,   7.2564],
        [ -7.2515,   5.0195,   4.6434],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.9916,   7.2133,  -0.9179],
        [  5.9912,   3.1453,  -3.9989],
        [  5.5240,  -2.1198,   6.2055],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.4309,  -0.0599,  -4.0830],
        [  0.2732,  -4.7619,  -1.1577],
        [ 10.6283,  -5.7253,   7.4878],
        [ 10.0567,   8.6697,   4.6329],
        [  1.6308,  13.9254,  -2.0984],
        [  6.3534,   1.2990,  -7.2942],
        [  2.2644,  -4.6434,   9.7508],
        [  6.3549,   0.7096,   6.9198],
        [ -1.0482, -10.8487,  -5.9066],
        [ 11.1420,   6.9331,  -2.4779],
        [ -5.1377,   6.7751,  -0.4438],
        [ -3.4986,   7.5152,   9.0712],
        [ -5.6792,  -8.7352,   4.0318],
        [ -5.3736,   6.6310,  -1.5174],
        [ -6.4844,  -8.0230,   8.5209],
        [  4.5767,   5.5071,   3.9069],
        [  2.3373,  -4.6232,  -5.2246],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.5650,   2.0132,   4.6272],
        [ -9.2974,   0.8371,   2.4852],
        [  0.9809,  -4.8145,   6.8706],
        [ 11.7418,   5.6070,  -1.4192],
        [  5.6314,  -1.9302,  -9.6543],
        [  5.3858,   4.4260,  -9.6769],
        [ -8.5149,  -5.1905,   1.5239],
        [  9.0051,   9.9234,   5.4411],
        [ -5.6423,   6.4537,  -1.2091],
        [ 12.3609,   3.4769,  -8.5762],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.3087,   0.7792,  -7.5587],
        [ -2.1533, -10.5778,   1.0112],
        [  4.9531,   5.0576,   5.6794],
        [  0.1252, -11.0062,   7.2564],
        [  0.1497,   7.8334,  -5.4412],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3497,   1.3680,  -1.9076],
        [  1.5383, -11.0274,  -4.9666],
        [  4.1348,   5.9514,  -0.8856],
        [ -3.9700,   7.3453,  -4.1486],
        [  6.3604,   0.9042,  -6.5901],
        [  3.3633,  13.4794,   0.5193],
        [ -7.3446,   4.9113,  -5.9356],
        [  6.3048,   0.1469,  -9.1691],
        [  2.4149,  -4.6005,  -4.3339],
        [  4.5325,  -3.3643,   6.7699],
        [  0.9478,  -4.8144,   0.6371],
        [ -9.1658,  -3.4155,  -2.8880],
        [  6.7792,  11.8084,   1.6251],
        [  0.9945,  -4.8144,  -0.1897],
        [  6.2555,   2.1474,   4.1685],
        [  6.1973,  -0.4372,   6.5720]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0656, 0.0146,  ..., 0.7933, 0.4392, 0.3501],
        [0.0656, 0.0000, 0.0801,  ..., 0.8546, 0.4947, 0.3964],
        [0.0146, 0.0801, 0.0000,  ..., 0.7795, 0.4268, 0.3401],
        ...,
        [0.7933, 0.8546, 0.7795,  ..., 0.0000, 0.3779, 0.5076],
        [0.4392, 0.4947, 0.4268,  ..., 0.3779, 0.0000, 0.1322],
        [0.3501, 0.3964, 0.3401,  ..., 0.5076, 0.1322, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.46553623, 13.56367857, 13.744387  , 13.77604293, 13.85158417,
       13.70893492,  6.77777606,  9.6282763 , 11.69264243, 12.97370371,
        7.80625981,  5.1224079 , 12.85032836, 12.88270018,  5.87305005,
        9.91668182, 13.44593519, 14.02526505, 11.65923396,  7.50500064,
        8.24808832, 12.3344536 , 13.0743399 ,  6.34059021, 10.61417181,
        9.68323686, 14.08549802,  8.89458273,  5.1224079 , 12.30858677,
        5.30158497, 10.37998078, 11.80092578, 11.89855267, 10.69963954,
       13.35832725,  6.60752373, 13.05148826,  8.36445758, 10.00794468,
        5.80860165,  9.99211663,  8.81930525,  5.95297459, 10.26327895,
        9.48337063,  5.26067019, 11.00694688,  8.81930525, 10.78533838,
       13.1471313 ,  6.76661595,  5.91677188, 10.24557047,  9.91668182,
        9.43112957,  4.76969364, 12.07225469, 13.2778378 , 14.02058621,
        6.48486171,  5.16611782,  6.39438044, 10.8992524 , 13.12299145,
        8.50278896,  8.28968067, 10.41909211,  8.53501777, 10.31584102,
        7.16058587,  5.18047638,  8.43626114, 12.72524615,  9.3349858 ,
        4.91336994, 13.01187484,  5.95297459,  6.97106912,  9.97220799,
       13.40024224,  8.5724042 , 12.84056604,  5.70405525,  9.34127013,
       10.794755  ,  7.07902036, 11.00694688,  7.83486943,  6.02588517,
        6.49538841, 11.1341798 ,  7.24677664,  8.34949383,  6.42439344,
       13.89264822,  8.8353674 ,  6.30647746,  5.19577146,  5.64464807,
        4.90676242,  9.78147542, 13.61600536,  4.9160962 ,  6.61386436,
        6.21275016])}
{'P': tensor([[2.5640e-03, 3.2700e-01, 2.3916e-01,  ..., 4.9827e-04, 6.3898e-05,
         2.0740e-08],
        [1.5251e-05, 3.0816e-03, 3.4290e-01,  ..., 6.9920e-04, 8.7399e-05,
         2.4559e-08],
        [6.8244e-06, 8.5157e-06, 6.1809e-03,  ..., 2.0257e-03, 1.3145e-04,
         4.5798e-08],
        ...,
        [2.2820e-05, 2.5629e-05, 4.3823e-05,  ..., 8.8117e-02, 1.1486e-04,
         2.2508e-05],
        [1.8885e-05, 2.4287e-05, 6.1232e-05,  ..., 6.2222e-03, 2.7488e-01,
         1.1983e-06],
        [8.6825e-06, 7.3664e-06, 3.2893e-06,  ..., 2.0404e-06, 2.1620e-07,
         7.3064e-03]]), 'X': tensor([[ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 2.6451e+00, -4.5250e+00, -5.5650e-01],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.0358e+00, -9.2784e+00, -9.8479e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 3.3839e+00, -4.1966e+00,  8.3574e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-7.9667e+00,  4.0814e+00, -6.3974e-01],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [ 1.7815e+00, -4.7481e+00,  4.4289e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 3.1892e+00, -1.0822e+01, -6.8409e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 5.5626e+00, -2.0536e+00,  8.4433e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0034, 0.0267,  ..., 0.8172, 1.1993, 0.9621],
        [0.0034, 0.0000, 0.0248,  ..., 0.8148, 1.1959, 0.9603],
        [0.0267, 0.0248, 0.0000,  ..., 0.7907, 1.1844, 0.9355],
        ...,
        [0.8172, 0.8148, 0.7907,  ..., 0.0000, 0.9019, 0.2219],
        [1.1993, 1.1959, 1.1844,  ..., 0.9019, 0.0000, 1.0939],
        [0.9621, 0.9603, 0.9355,  ..., 0.2219, 1.0939, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.48809595, 13.52430176, 13.68672855, 13.59170809, 13.76047999,
       13.93790235,  8.09842704,  8.7907923 ,  6.38105015, 11.51557166,
        5.24137909,  7.50917551, 12.58006038, 11.83570132, 11.64436684,
       11.04953767, 14.07623266,  5.90035324,  7.81509631,  9.79039371,
        9.5014146 , 13.32645552,  7.5158489 , 10.35485331,  7.92200342,
        7.72415048,  5.39098669, 13.6829843 , 10.23946247,  9.79814997,
        7.52117132,  6.27715105,  9.09119351,  8.8353674 , 13.48809595,
       12.32545637,  9.96152788, 11.11969815, 11.8159254 ,  7.32233733,
        5.0209005 , 14.10045797,  8.95133961,  9.24501829,  8.56889738,
        7.70897675, 11.4890861 ,  8.40445872,  6.33517349, 13.6829843 ,
       11.48663767, 12.05300943,  7.46248947,  8.7907923 ,  9.77588801,
       11.28895489, 13.12299145,  5.23878371,  5.0209005 ,  9.32094055,
       11.95743558,  8.28968067, 13.61801526, 10.06023665, 10.1371552 ,
        6.65368551,  7.50500064,  4.89121805, 11.18839169,  5.96556248,
        7.89745107,  7.89745107,  5.0713364 , 11.05413545, 11.018084  ,
       10.46238054, 11.28215429,  5.39723241,  9.09119351,  5.92281142,
        6.29034958,  6.54127431, 13.04217487, 10.90638066,  5.92952538,
        7.39004226,  6.5618968 , 10.69963954,  6.33075682,  9.13507321,
        5.98285319, 10.75443106,  8.92395381, 13.27717982, 11.10066126,
        9.96152788,  8.36995953,  6.81084724, 13.04051744, 14.05687815,
       10.794755  ,  7.39004226,  5.13616224,  8.19703512, 14.10796466,
       11.36089355])}
{'P': tensor([[1.5441e-03, 2.6415e-01, 2.4646e-01,  ..., 6.1968e-08, 7.9355e-09,
         2.5990e-06],
        [9.3527e-06, 3.8753e-03, 1.2365e-01,  ..., 1.1562e-07, 1.7521e-08,
         8.8944e-06],
        [1.5238e-05, 1.5962e-04, 7.5115e-03,  ..., 2.0382e-07, 3.4484e-08,
         1.8361e-05],
        ...,
        [9.3327e-03, 6.4320e-03, 5.2749e-03,  ..., 2.8349e-02, 1.5076e-06,
         2.6457e-06],
        [1.9621e-04, 1.3075e-04, 1.2700e-04,  ..., 1.8812e-03, 1.3401e-01,
         2.4054e-06],
        [7.6468e-07, 7.6113e-07, 7.5083e-07,  ..., 3.6901e-06, 1.6197e-05,
         3.5922e-02]]), 'X': tensor([[  7.9816,  10.8938,  -0.5378],
        [  7.1697,  11.5343,  -0.7276],
        [  6.9284,  11.7061,  -0.3732],
        [  6.2026,  12.1774,  -1.5887],
        [  4.9152,  12.8653,  -1.4459],
        [  4.1637,  13.1898,  -1.9189],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -3.9456,  -9.8542,   0.7466],
        [  4.6231,   5.4556,   8.4973],
        [ -3.3999, -10.1152,  -1.2088],
        [  5.8275,  -1.5334,   2.0124],
        [  4.5503,   5.5359,   7.4735],
        [ 12.3759,  -1.3943,   6.4659],
        [ -4.7642,  -9.3860,  -5.1133],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -5.9987,   6.1949,  -6.2114],
        [ -5.6174,   6.4708,  -1.7339],
        [  1.3103,  -4.8032,  -0.9300],
        [ 11.0373,   7.1299,  -8.5895],
        [ -6.6998,   5.5954,   0.0823],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.7121,   6.9743,  -8.4202],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -2.1817,   7.8274,  -4.6298],
        [  2.3261,  13.7749,   5.9121],
        [  1.5383, -11.0274,  -4.9666],
        [  9.3560,   9.5391,   9.4943],
        [ 11.9641,  -2.9568,   7.7211],
        [ -0.1626, -10.9796,   6.4629],
        [ -8.6426,  -4.9136,   9.4941],
        [ -8.0534,  -6.0429,   2.4583],
        [ -8.0534,  -6.0429,   2.4583],
        [  3.1756, -10.8247,   6.8882],
        [ -5.3179,  -9.0101,   9.4472],
        [  1.3103,  -4.8032,  -0.9300],
        [  6.1973,  -0.4372,   6.5720],
        [  5.1890,   4.7318,   0.9515],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -1.8367,   7.8718,  -8.6523],
        [  1.2728, -11.0372,   9.8121],
        [  9.3424,  -7.3685,  -3.6807],
        [  0.3493,  -4.7727,  -6.5682],
        [  0.3724,   7.7974,  -6.8722],
        [ -9.4671,  -0.5553,   0.1117],
        [ -4.2040,   7.2486,   2.9875],
        [ 10.1798,   8.5002,  -5.8893],
        [  2.7685, -10.8982,  -6.8928],
        [  1.2273,  -4.8081,   9.8129],
        [  5.3924, -10.1396,  -4.0863],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.7295,  13.6703,  -8.9328],
        [  6.9284,  11.7061,  -0.3732],
        [ -2.1817,   7.8274,  -4.6298],
        [ -8.0829,  -5.9940,   5.0564],
        [ -8.5149,  -5.1905,   1.5239],
        [  6.1372,   2.6726,  -6.7885],
        [  2.9470,  13.6085,   2.9809],
        [  1.7232,  -4.7574,   3.1270],
        [ -9.1795,  -3.3623,   8.5259],
        [ 12.0042,  -2.8350,   2.6864],
        [ -9.2974,   0.8371,   2.4852],
        [  0.1287, -11.0065,   6.5757],
        [  4.8051, -10.3703,   0.5268],
        [ -9.1658,  -3.4155,  -2.8880],
        [ -9.4748,  -0.7765,  -5.6695],
        [  2.4572,   7.1085,   7.7376],
        [  0.1287, -11.0065,   6.5757],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.4630,  -1.5145,   0.2983],
        [  3.2159, -10.8166,  -1.0406],
        [ -9.4501,  -1.7133,   2.2980],
        [  6.3604,   0.9042,  -6.5901],
        [  2.2659,  13.7894,   6.5436],
        [ 10.3251,   8.2928,  -3.0006],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.8934,   2.2597,   3.5831],
        [  2.2659,  13.7894,   6.5436],
        [ -8.1666,   3.7653,   3.6075],
        [  5.1793,   4.7461,   0.3899],
        [  2.0899,  -4.6870,  -7.4260],
        [  1.7500,   7.4187,   4.4237],
        [ -9.3331,  -2.6465,  -8.2028],
        [ -6.8618,  -7.6322,   3.7027],
        [ -7.6719,  -6.6260,   7.1468],
        [  1.6408,   7.4591,  -4.2294],
        [  2.7295,  13.6703,  -8.9328],
        [ -6.7643,  -7.7372,  -9.6663],
        [  5.0277,   4.9589,   7.0175],
        [  5.3804,  -2.3504,   7.2744],
        [  2.5469,  13.7192,  -9.6935],
        [  3.2159, -10.8166,  -1.0406],
        [ -7.5401,   4.6718,   9.0033],
        [  6.9956,  11.6590,  -7.6876],
        [ -7.7842,   4.3466,  -1.7548],
        [ -5.6651,   6.4380,   1.3660],
        [  2.5620,   7.0549,   8.4825],
        [  5.1196,  -2.7160,   4.2705],
        [ -9.0604,   1.7707,   1.1066],
        [ -7.4733,   4.7555,   6.2490],
        [  2.2182,  13.8007,   5.1889],
        [  2.4639, -10.9431,  -7.3718],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.3444,   0.5821,   6.9101],
        [ 12.4597,  -0.9136,  -4.8763],
        [  6.8093,  -9.4178,   8.4244],
        [ -9.3604,   0.4855,  -3.0158]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.5354e-02, 2.0019e-02,  ..., 3.8004e-01, 9.7424e-01,
         1.1744e+00],
        [2.5354e-02, 2.1453e-04, 2.9231e-02,  ..., 3.5469e-01, 9.8324e-01,
         1.1593e+00],
        [2.0019e-02, 2.9231e-02, 0.0000e+00,  ..., 3.7679e-01, 9.5711e-01,
         1.1565e+00],
        ...,
        [3.8004e-01, 3.5469e-01, 3.7679e-01,  ..., 0.0000e+00, 1.1605e+00,
         9.9165e-01],
        [9.7424e-01, 9.8324e-01, 9.5711e-01,  ..., 1.1605e+00, 0.0000e+00,
         1.0440e+00],
        [1.1744e+00, 1.1593e+00, 1.1565e+00,  ..., 9.9165e-01, 1.0440e+00,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.50484539, 13.58101794, 13.60275211, 13.6660671 , 13.77222752,
       13.83139024,  9.50913849, 10.61472636,  7.15100303, 10.67132524,
        6.02588517,  7.16598273, 12.45418236, 10.52587243, 12.4980845 ,
        8.62329466,  8.56889738,  4.978704  , 13.1399181 ,  8.72896381,
        9.93553424,  7.48311002, 12.35421461,  8.12580976, 13.96987806,
       11.1341798 , 13.3614598 , 12.32408516, 10.98076873,  9.94173708,
       10.06850677, 10.06850677, 11.28093507, 10.46238054,  4.978704  ,
        6.21275016,  7.02253235,  9.66977802,  8.08320235, 11.11038551,
       11.89855267,  4.78544875,  7.80625981,  9.48337063,  8.37954853,
       13.26209545, 11.24434374,  4.96231532, 11.48434767,  9.78286733,
       13.94008948, 13.60275211,  8.12580976, 10.06286074,  9.97220799,
        6.69389431, 13.92390689,  5.05989547,  9.77588801, 12.3344536 ,
        9.3349858 , 11.00726483, 11.42947632,  9.78147542,  9.50654967,
        7.52117132, 11.00726483, 12.64235359,  9.58347456, 11.2845608 ,
        9.60412735,  6.42439344, 13.97430191, 13.2430377 , 13.41713661,
        9.17595899, 13.97430191,  8.99276564,  7.02496223,  5.13182506,
        7.62232307,  9.70108409, 10.26327895, 10.1371552 ,  7.6374603 ,
       13.94008948, 10.27716434,  7.06169391,  5.87132718, 13.95361098,
       11.2845608 ,  8.87007495, 13.59673406,  8.91551231,  8.57560275,
        7.50564223,  5.79543038,  9.23178204,  8.8580671 , 13.97779839,
       11.21703654, 10.19036081,  9.36256096, 12.49317563, 11.62158554,
        9.37296102])}
{'P': tensor([[2.9156e-03, 3.2442e-01, 2.9169e-01,  ..., 5.5750e-07, 9.9446e-04,
         1.0598e-05],
        [9.3979e-06, 6.8212e-03, 2.5358e-05,  ..., 2.2007e-06, 9.9943e-04,
         4.5062e-05],
        [1.4894e-05, 2.7175e-01, 3.5546e-03,  ..., 8.3329e-07, 9.4706e-04,
         1.8166e-05],
        ...,
        [1.2003e-06, 9.3968e-07, 1.0701e-06,  ..., 5.3150e-02, 1.4949e-05,
         5.4369e-05],
        [7.7952e-04, 2.3196e-03, 1.2390e-03,  ..., 2.7573e-04, 7.6711e-02,
         3.2307e-03],
        [1.6898e-06, 1.8489e-06, 1.8221e-06,  ..., 2.3952e-02, 3.5432e-05,
         7.4926e-02]]), 'X': tensor([[  7.7794,  11.0629,   8.9069],
        [  5.9587,  12.3216,   9.1695],
        [  7.0515,  11.6194,   8.4289],
        [  5.0617,  12.7957,   8.3544],
        [  2.7590,  13.6621,   9.4496],
        [  4.8674,  12.8875,   6.7622],
        [ 12.1940,  -2.1881,   7.1499],
        [  5.8318,   3.5567,  -4.7692],
        [ -8.4573,   3.2448,   5.4370],
        [ -8.9282,   2.1648,   8.4894],
        [ 12.3453,  -1.5462,   7.0212],
        [ -3.6381,   7.4683,   7.3544],
        [ -7.0120,   5.2825,   7.7762],
        [  6.8407,  11.7666,   3.0045],
        [ -8.8058,   2.4850,  -1.7425],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.5832,   1.7543,  -7.6134],
        [ -6.4620,  -8.0449,   6.2719],
        [  4.5340,  -3.3629,   8.9652],
        [  4.1999,  -3.6543,  -4.1998],
        [ -8.4887,  -5.2448,  -0.6585],
        [ -9.0574,   1.7801,   8.8986],
        [  4.3554,   5.7393,  -9.0766],
        [ 12.6000,   1.3839,  -3.2992],
        [ -5.1046,  -9.1610,   2.3418],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.8407,  11.7666,   3.0045],
        [  4.5954,  13.0099,  -4.4439],
        [ 12.4901,  -0.7052,   1.7315],
        [ 10.5955,   7.8824,  -3.7006],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -3.5029,   7.5138,   8.1853],
        [  3.1480,  -4.3165,  -7.8002],
        [ -3.4986,   7.5152,   9.0712],
        [  0.1677,  -4.7447,  -6.1529],
        [ -8.3653,   3.4184,   9.3454],
        [  3.9085,   6.1512,  -2.2560],
        [ -5.6169,  -8.7844,   8.9057],
        [ -8.7544,   2.6096,   3.3146],
        [  0.9478,  -4.8144,   0.6371],
        [ -8.7544,   2.6096,   3.3146],
        [  4.5503,   5.5359,   7.4735],
        [  5.4167, -10.1293,   6.2156],
        [ 10.5955,   7.8824,  -3.7006],
        [  2.2659,  13.7894,   6.5436],
        [  2.5636, -10.9293,   9.6519],
        [  6.1989,  -9.7591,   0.8289],
        [ -9.1459,   1.4780,  -4.2403],
        [  2.6321,  -4.5296,  -2.5159],
        [  5.4167, -10.1293,   6.2156],
        [ -4.9217,   6.8979,  -1.7667],
        [  4.9152,  12.8653,  -1.4459],
        [  3.7992,  -3.9474,  -8.9019],
        [  3.6249, -10.7255,   5.6605],
        [  5.1292,   4.8183,  -5.7997],
        [  6.2036,   2.4030,  -1.2500],
        [  5.6093,  -1.9705,   6.4807],
        [ -7.4807,   4.7463,   2.0616],
        [ 11.7162,   5.6728,  -8.9372],
        [ 12.4691,   2.8525,  -5.5110],
        [  4.0457,  -3.7737,  -0.4704],
        [  3.6154,  -4.0640,  -7.9698],
        [  6.2905,   1.9366,   2.1169],
        [  7.1441,  11.5529,   2.3232],
        [ 11.5198,  -4.0923,   5.5088],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -7.8126,   4.3066,   5.1791],
        [ -7.6719,  -6.6260,   7.1468],
        [  9.0004,  -7.7195,   4.6945],
        [ -6.3306,   5.9272,   5.6935],
        [  6.1741,  -0.5329,   3.1913],
        [  2.1120,  -4.6818,   8.6144],
        [ -7.8126,   4.3066,   5.1791],
        [ -9.4366,  -0.1189,  -2.6736],
        [  6.0305,   3.0296,   0.1375],
        [  1.6954,  -4.7616,   9.8198],
        [ -9.3616,  -2.4750,   3.7765],
        [  4.0457,  -3.7737,  -0.4704],
        [  4.5325,  -3.3643,   6.7699],
        [  2.1379,  -4.6757,   6.4981],
        [  2.2437,  -4.6489,   6.1430],
        [  4.6992,   5.3690,   3.0935],
        [  2.2849, -10.9655,  -7.6542],
        [  8.9284,  10.0036,  -6.9210],
        [ -2.3317,   7.8034,  -3.6274],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -2.5683, -10.4433,  -2.9182],
        [  2.1625,  -4.6697,   7.0658],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.6229,   2.9068,   2.4429],
        [  2.7420,  -4.4896,  -3.7139],
        [ 12.5717,   1.9266,   7.4903],
        [ -8.5149,  -5.1905,   1.5239],
        [ 12.5763,   0.1333,  -9.2650],
        [  1.7218,  -4.7576,  -3.3617],
        [  4.2339,  -3.6268,   6.3125],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.3087,   0.7792,  -7.5587],
        [  4.5340,  -3.3629,   8.9652],
        [  8.5198,  10.4091,  -8.6486],
        [  6.3515,   0.6418,  -6.5823],
        [  9.4554,   9.4245,   9.8800],
        [  2.4572,   7.1085,   7.7376],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.4572,   7.1085,   7.7376],
        [ -8.7544,   2.6096,   3.3146]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.4802e-02, 1.5852e-02,  ..., 1.3325e+00, 6.1985e-01,
         1.2726e+00],
        [2.4802e-02, 0.0000e+00, 1.9012e-02,  ..., 1.3078e+00, 5.9595e-01,
         1.2485e+00],
        [1.5852e-02, 1.9012e-02, 0.0000e+00,  ..., 1.3232e+00, 6.0691e-01,
         1.2600e+00],
        ...,
        [1.3325e+00, 1.3078e+00, 1.3232e+00,  ..., 0.0000e+00, 7.7278e-01,
         3.9285e-01],
        [6.1985e-01, 5.9595e-01, 6.0691e-01,  ..., 7.7278e-01, 0.0000e+00,
         6.5338e-01],
        [1.2726e+00, 1.2485e+00, 1.2600e+00,  ..., 3.9285e-01, 6.5338e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.52430176, 13.68672855, 13.59170809, 13.76047999, 13.93790235,
       13.77604293, 12.38881635,  6.83083863,  9.05843201,  9.18690018,
       12.44177641,  8.30729352,  8.77913428, 13.6105595 ,  9.14973319,
        5.05444288, 12.70489563, 10.31886066,  5.64500512,  5.5671389 ,
        9.97822473,  9.23071698,  7.20482224, 12.67576792, 10.48721382,
       10.06850677, 13.6105595 , 13.79762415, 12.50996828, 13.20597932,
        9.47369628,  8.29022197,  5.34250225,  8.28968067,  4.74770899,
        9.0368458 ,  7.28793663, 10.42667093,  9.13507321,  4.90676242,
        9.13507321,  7.16598273, 11.48663767, 13.20597932, 13.97430191,
       11.22596994, 11.56146873,  9.26455795,  5.23878371, 11.48663767,
        8.47371868, 13.77222752,  5.47866221, 11.32149259,  7.03733775,
        6.652742  ,  5.94535529,  8.85939626, 13.01728313, 12.79126785,
        5.53255882,  5.43943791,  6.58184474, 13.58333857, 12.22503754,
       12.35421461,  8.92098752, 10.1371552 , 11.85743425,  8.67227386,
        6.19707944,  5.13616224,  8.92098752,  9.43737523,  6.74872981,
        5.05444288,  9.68324553,  5.53255882,  5.64464807,  5.14124612,
        5.16205087,  7.13501264, 11.20100795, 13.40850328,  8.14433542,
       12.30858677, 10.75443106,  5.14607014,  8.04718433,  9.09963894,
        5.26067019, 12.71843462,  9.97220799, 12.5769659 ,  5.05963217,
        5.5748624 ,  7.72668467,  9.34127013,  5.64500512, 13.45124917,
        6.383897  , 13.35013374,  7.52117132,  9.9353921 ,  7.52117132,
        9.13507321])}
{'P': tensor([[1.5443e-03, 2.6174e-01, 1.6412e-01,  ..., 2.9683e-04, 5.1842e-06,
         4.7557e-08],
        [4.9119e-06, 2.2457e-03, 3.1758e-01,  ..., 5.4183e-04, 4.9747e-06,
         5.1250e-08],
        [1.0416e-05, 2.0853e-05, 6.1462e-03,  ..., 1.7990e-03, 1.0479e-05,
         1.1540e-07],
        ...,
        [8.1365e-05, 1.1101e-04, 1.4158e-04,  ..., 2.2194e-01, 3.8264e-04,
         1.4454e-05],
        [2.8217e-02, 2.9453e-02, 3.0481e-02,  ..., 3.1785e-02, 1.5482e-01,
         3.7052e-04],
        [1.0979e-02, 6.5630e-03, 4.6182e-03,  ..., 1.8546e-04, 1.3030e-02,
         3.5915e-02]]), 'X': tensor([[  7.7750,  11.0665,   2.8364],
        [  6.8407,  11.7666,   3.0045],
        [  6.1575,  12.2046,   3.3257],
        [  5.4867,  12.5816,   2.9549],
        [  5.9003,  12.3550,   2.1917],
        [  6.6319,  11.9066,   3.5165],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -4.7334,   6.9982,  -0.4392],
        [  5.8275,  -1.5334,   2.0124],
        [  9.9750,  -6.6338,  -1.2440],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -5.6792,  -8.7352,   4.0318],
        [ -1.5875,   7.8944,  -9.4286],
        [ 12.5763,   0.1333,  -9.2650],
        [ 11.6165,  -3.8715,   6.4594],
        [ -1.8367,   7.8718,  -8.6523],
        [  8.0877,  -8.5297,  -1.1050],
        [ -9.3087,   0.7792,  -7.5587],
        [  3.3094,  13.4969,  -3.6507],
        [ -6.9838,  -7.4965,   0.9796],
        [ -8.9178,   2.1936,   7.5388],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.0449,  13.8402,  -7.1642],
        [ -7.4887,   4.7364,   8.9074],
        [ 12.2592,   3.9425,   2.0383],
        [ -9.3419,   0.5972,  -1.2841],
        [ -7.0825,   5.2073,  -5.0497],
        [ -9.2974,   0.8371,   2.4852],
        [  0.5850,  -4.7982,   9.6639],
        [ -0.6323,   7.9085,  -5.9544],
        [  6.2365,  -0.2578,  -6.0212],
        [  4.4290,  -3.4596,   1.4181],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -5.9938,  -8.4745,   3.9735],
        [ -9.4117,   0.1160,   8.2574],
        [ -1.0482, -10.8487,  -5.9066],
        [  4.7813,  -3.1139,   8.2691],
        [  6.2770,  -0.0379,   2.4195],
        [  2.0449,  13.8402,  -7.1642],
        [  2.5620,   7.0549,   8.4825],
        [ -5.9225,  -8.5355,   2.4778],
        [  1.0093,   7.6567,   2.3526],
        [ -0.2586,   7.8825,  -9.9840],
        [  5.1861,   4.7361,  -3.7312],
        [ -5.1398,  -9.1367,   6.2074],
        [  5.9003,  12.3550,   2.1917],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.2188,  -2.5839,   2.1833],
        [  6.1973,  -0.4372,   6.5720],
        [ -7.0825,   5.2073,  -5.0497],
        [  6.3266,   0.3292,   9.3502],
        [ -7.9667,   4.0814,  -0.6397],
        [ -7.4807,   4.7463,   2.0616],
        [  1.0391,  -4.8141,  -0.0768],
        [  6.9956,  11.6590,  -7.6876],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -9.3739,   0.3982,  -4.2564],
        [  2.5829,  -4.5466,  -8.4015],
        [  5.7802,  -1.6363,  -6.2782],
        [ 12.5789,   0.1722,   1.6470],
        [ -8.6426,  -4.9136,   9.4941],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.1625,  -4.6697,   7.0658],
        [ -9.3810,  -2.3455,  -9.5194],
        [  2.5974,   7.0363,  -7.7810],
        [ -0.2586,   7.8825,  -9.9840],
        [  2.4697,  -4.5836,   1.4593],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 12.6000,   1.3839,  -3.2992],
        [ -7.6525,  -6.6534,   5.1721],
        [  5.8902,  -1.3887,   4.0388],
        [ 11.4743,   6.2478,  -2.1100],
        [ 11.5935,   5.9740,  -4.3952],
        [  5.6672,  -1.8631,   8.1020],
        [  9.0004,  -7.7195,   4.6945],
        [ -8.6112,   2.9319,  -7.9943],
        [  5.5424,  -2.0885,   7.2487],
        [  7.1821,  -9.1845,   9.8548],
        [  6.3246,   1.6696,  -0.1544],
        [ -9.3107,  -2.7697,  -3.2919],
        [ 11.4495,   6.3026,   6.8257],
        [ -1.8367,   7.8718,  -8.6523],
        [  1.5662,  13.9375,   4.5142],
        [ -8.8835,  -4.3178,   9.3080],
        [  7.8982,  -8.6778,   0.0239],
        [ 12.5154,   2.5072,  -9.3274],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -2.6735, -10.4062,  -1.5911],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.3889,  -4.6082,  -8.7291],
        [  4.0591,   6.0202,  -1.6339],
        [  2.5382,   7.0672,   4.5097],
        [  6.0955,   2.8210,  -4.3321],
        [ -9.4630,  -1.5145,   0.2983],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.2770,  -0.0379,   2.4195],
        [ 11.0891,  -4.9558,  -2.3178],
        [ 12.6059,   0.9597,  -6.3264],
        [  3.3943,   6.5483,   1.0790],
        [  6.3534,   1.2990,  -7.2942],
        [  1.0391,  -4.8141,  -0.0768],
        [  9.8082,  -6.8396,   4.9615],
        [ -2.1382,   7.8339,   7.3524],
        [  5.9883,   3.1536,   6.2320],
        [ 12.2692,  -1.8882,   3.9253]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0318, 0.0748,  ..., 0.6404, 0.3176, 0.2635],
        [0.0318, 0.0000, 0.0430,  ..., 0.6394, 0.3168, 0.2718],
        [0.0748, 0.0430, 0.0000,  ..., 0.6412, 0.3214, 0.2891],
        ...,
        [0.6404, 0.6394, 0.6412,  ..., 0.0000, 0.3229, 0.3925],
        [0.3176, 0.3168, 0.3214,  ..., 0.3229, 0.0000, 0.0985],
        [0.2635, 0.2718, 0.2891,  ..., 0.3925, 0.0985, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.5247215 , 13.6105595 , 13.66991027, 13.72594022, 13.69162795,
       13.62896593, 12.4013153 ,  8.4486771 ,  6.02588517, 11.97949273,
       10.60226614, 10.41909211,  8.05242591, 12.5769659 , 12.24467041,
        8.08320235, 11.75438006,  9.34127013, 13.89671314, 10.24557047,
        9.18358562,  5.24461068, 13.99048094,  8.86082124, 12.87752052,
        9.36093562,  8.7907923 ,  9.3349858 ,  4.83371341,  7.9337647 ,
        6.24187044,  5.62005629,  9.85330646, 10.37998078,  9.41244865,
       10.8992524 ,  5.70591986,  6.27715105, 13.99048094,  7.50564223,
       10.38897563,  7.72291946,  7.88677538,  7.02326515, 10.48315436,
       13.69162795,  9.95552626,  5.82345867,  6.21275016,  8.7907923 ,
        6.33517349,  8.95133961,  8.85939626,  4.92497514, 13.59673406,
       12.00995898,  9.38232121,  5.22903117,  6.00732193, 12.58006038,
        9.94173708, 11.99633819,  5.14607014,  9.66977802,  7.50035355,
        7.88677538,  5.20659892, 12.07818871, 12.67576792, 10.14044489,
        6.05165248, 13.06498554, 13.04217487,  5.96556248, 11.85743425,
        9.09661212,  5.92281142, 11.65923396,  6.54127431,  9.71392179,
       13.06957425,  8.08320235, 14.02526505,  9.87720102, 11.73397309,
       12.76408562,  9.78286733, 10.74409834,  8.83341608,  5.19065439,
        7.26074925,  7.50917551,  6.7166297 ,  9.58347456,  9.75200848,
       11.54672755,  6.27715105, 12.1460931 , 12.64235359,  7.37576154,
        6.48486171,  4.92497514, 11.95743558,  8.12043983,  6.76791483,
       12.41366754])}
{'P': tensor([[4.0557e-03, 2.9653e-01, 2.2780e-01,  ..., 1.7373e-03, 9.9370e-09,
         1.2389e-07],
        [2.4880e-06, 2.5991e-03, 6.1070e-04,  ..., 5.4346e-04, 4.6640e-09,
         1.6248e-07],
        [1.3130e-05, 1.1064e-02, 1.0046e-02,  ..., 3.1744e-03, 1.3650e-08,
         4.2439e-07],
        ...,
        [3.6084e-04, 1.3564e-03, 1.3574e-03,  ..., 1.2807e-01, 1.0968e-06,
         3.3398e-05],
        [1.5314e-04, 6.9536e-05, 4.5785e-05,  ..., 2.7509e-05, 2.3618e-01,
         9.5627e-06],
        [1.8664e-07, 1.4130e-07, 1.0871e-07,  ..., 7.3875e-07, 2.9133e-04,
         2.2895e-02]]), 'X': tensor([[ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 1.1294e+01,  6.6325e+00, -5.8311e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-8.8835e+00, -4.3178e+00,  9.3080e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 4.7544e+00, -3.1425e+00, -1.5315e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-2.7572e+00,  7.7194e+00,  5.9580e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 4.8051e+00, -1.0370e+01,  5.2675e-01],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 1.1006e+00,  7.6318e+00, -2.4877e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0184, 0.0533,  ..., 0.6400, 1.3825, 1.7535],
        [0.0184, 0.0000, 0.0676,  ..., 0.6252, 1.3648, 1.7356],
        [0.0533, 0.0676, 0.0000,  ..., 0.6502, 1.4293, 1.7826],
        ...,
        [0.6400, 0.6252, 0.6502,  ..., 0.0000, 1.1121, 1.1952],
        [1.3825, 1.3648, 1.4293,  ..., 1.1121, 0.0000, 0.8530],
        [1.7535, 1.7356, 1.7826,  ..., 1.1952, 0.8530, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.54774355, 13.77604293, 13.76047999, 13.91198191, 13.85158417,
       13.95472331, 10.09979747,  4.78258092,  7.96181288, 11.63765273,
        6.29821673,  8.0550286 ,  6.02588517, 13.52430176, 10.1979669 ,
       12.41366754,  9.38232121, 13.32645552, 11.8159254 ,  8.1137452 ,
       12.72524615,  5.07306274,  6.38105015,  4.76969364,  7.57685908,
       10.1979669 ,  8.28968067,  9.97964013,  8.8737556 , 11.56146873,
       13.09737053, 11.83509372, 13.79676654,  7.14837514,  5.43544626,
        9.77777346,  9.87720102, 12.62711252, 11.21703654, 11.24434374,
        7.32233733,  7.44805901,  5.69913817,  6.02588517, 13.32557133,
        6.02588517, 13.01187484, 10.54658597,  6.90115901,  6.41202986,
        6.59176018,  7.71075226,  7.16747878,  7.15100303,  7.5984809 ,
       13.97430191,  9.6282763 ,  5.19986797,  5.39723241, 10.86693708,
        8.19703512, 10.52357479, 10.35464916,  8.38252937, 13.66991027,
        8.4486771 ,  6.54615887,  5.56809335,  9.75200848, 12.22276278,
       11.00726483,  9.95552626, 11.57249984, 11.00812996,  9.11473391,
       11.0050379 , 11.42947632,  9.77777346,  9.09963894,  6.90115901,
        4.98461983,  8.90405823, 10.26545792,  8.67227386,  8.08320235,
        9.69792513,  8.0550286 ,  8.31625171,  9.66977802,  8.0550286 ,
       13.77075174, 13.96746816,  9.17905491,  7.71075226, 13.27994304,
       10.51437947,  9.7712321 ,  7.92644746,  5.26067019,  6.05165248,
       13.58333857,  6.41202986, 10.92597196,  7.73692436, 11.86239961,
       10.09979747])}
{'P': tensor([[4.9790e-03, 2.1118e-01, 1.5384e-01,  ..., 3.9107e-06, 9.4164e-07,
         1.8637e-04],
        [2.6889e-06, 5.3701e-03, 4.2681e-05,  ..., 1.4685e-05, 5.4029e-07,
         7.8725e-04],
        [5.4090e-06, 4.2659e-02, 2.9675e-03,  ..., 6.7207e-06, 2.8120e-07,
         2.0093e-04],
        ...,
        [4.2098e-07, 4.2056e-07, 4.7390e-07,  ..., 2.8230e-02, 9.9103e-06,
         2.8895e-05],
        [1.4875e-03, 5.8080e-04, 5.1694e-04,  ..., 1.3446e-05, 1.0348e-01,
         2.8739e-04],
        [5.1427e-06, 1.0370e-05, 7.1545e-06,  ..., 6.5958e-03, 8.9473e-05,
         8.3389e-02]]), 'X': tensor([[  7.4132,  11.3526,   0.3322],
        [  3.3633,  13.4794,   0.5193],
        [  4.9152,  12.8653,  -1.4459],
        [  6.9284,  11.7061,  -0.3732],
        [  2.3373,  13.7721,  -0.2442],
        [  4.1637,  13.1898,  -1.9189],
        [  7.0515,  11.6194,   8.4289],
        [ -8.0608,   3.9362,  -5.9567],
        [ 11.7779,   5.5127,  -2.0699],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.5636, -10.9293,   9.6519],
        [  9.9756,  -6.6330,  -3.3429],
        [  1.6308,  13.9254,  -2.0984],
        [  3.9022,  13.2906,   5.5788],
        [ -3.2486,   7.5925,  -1.3719],
        [ -7.0825,   5.2073,  -5.0497],
        [ -9.4624,  -0.4633,  -7.3918],
        [  5.2608,  -2.5256,  -9.1446],
        [ -6.2668,   5.9808,   2.5614],
        [  6.0140,  -1.0661,  -2.5040],
        [  1.9241,   7.3503,   7.9004],
        [  6.1222,  -0.7261,   8.8204],
        [  6.3505,   0.6235,  -8.5849],
        [  0.1382, -11.0073,   1.0026],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.8617,   3.4853,  -9.8080],
        [ -4.6393,   7.0461,   1.1624],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -4.7389,   6.9953,   3.7638],
        [ 12.2311,  -2.0445,   1.3545],
        [  7.4132,  11.3526,   0.3322],
        [  4.3523, -10.5232,   8.2138],
        [  5.8275,  -1.5334,   2.0124],
        [ -6.3306,   5.9272,   5.6935],
        [ 12.3357,   3.6001,   0.7903],
        [  2.3261,   7.1726,   6.2522],
        [  8.8091,  -7.9037,   0.3631],
        [  6.3048,   0.1469,  -9.1691],
        [ -8.8934,   2.2597,   3.5831],
        [  2.4697,  -4.5836,   1.4593],
        [ -8.0766,  -6.0046,  -3.1873],
        [  9.0424,  -7.6779,  -7.5286],
        [  5.1667,  -2.6543,   1.2879],
        [  8.4691,  10.4571,  -5.0650],
        [  6.2038,   2.4020,  -3.1692],
        [  9.9756,  -6.6330,  -3.3429],
        [  8.5093,  -8.1764,  -3.2381],
        [  9.6587,   9.1817,  -0.8212],
        [  9.1820,   9.7334,  -2.7932],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -8.9636,   2.0648,   8.1946],
        [  5.8930,  12.3592,  -8.9201],
        [ 11.0373,   7.1299,  -8.5895],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -2.5161,   7.7699,  -9.3531],
        [  5.8323,  -1.5228,   5.1132],
        [ -0.5709,   7.9055,   7.6089],
        [  3.3167,   6.6022,   6.3554],
        [  4.2041,  -3.6509,  -4.9488],
        [  1.7232,  -4.7574,   3.1270],
        [ 11.7726,  -3.4876,  -6.7402],
        [  3.1064,  13.5607,   6.7605],
        [  6.0054,  -1.0902,   8.3079],
        [  6.0467,  -9.8369,  -6.5816],
        [ -5.3179,  -9.0101,   9.4472],
        [ 11.7349,  -3.5838,   2.1644],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -5.3736,   6.6310,  -1.5174],
        [ -0.0282,   7.8575,  -1.1746],
        [ 12.4592,  -0.9168,  -6.7571],
        [  7.9816,  10.8938,  -0.5378],
        [  3.2366,  -4.2732,   5.1532],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.9788,   2.0208,  -3.2867],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.0937,  13.8293,  -2.3533],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -8.8058,   2.4850,  -1.7425],
        [ -9.4378,  -1.8606,  -2.5668],
        [  0.6142,  14.0802,   0.5407],
        [  2.3928,   7.1404,  -1.7035],
        [  2.7295,  13.6703,  -8.9328],
        [  6.3515,   0.6418,  -6.5823],
        [  6.0305,   3.0296,   0.1375],
        [  3.4920,  -4.1366,  -8.3955],
        [ -8.1426,   3.8048,   9.3542],
        [ 12.5650,   2.0132,   4.6272],
        [  7.1821,  -9.1845,   9.8548],
        [ -0.8455, -10.8853,  -6.7400],
        [ -4.6942,  -9.4299,  -9.1846],
        [  6.0305,   3.0296,   0.1375],
        [ 12.5927,   0.4199,  -9.4997],
        [  8.9284,  10.0036,  -6.9210],
        [  1.1994,   7.6035,   2.4295],
        [ 12.2914,   3.8037,   6.0978],
        [ 11.8951,  -3.1570,   7.0742],
        [ -8.0965,  -5.9712,   4.3466],
        [  3.1756, -10.8247,   6.8882],
        [ -9.4296,  -1.9458,   4.0833],
        [ -8.0965,  -5.9712,   4.3466],
        [ -9.4697,  -0.6157,  -6.8472],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.5424,  -2.0885,   7.2487],
        [ -6.6895,   5.6051,   2.7899]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 6.8418e-02, 1.3593e-01,  ..., 1.2736e+00, 5.2427e-01,
         8.5737e-01],
        [6.8418e-02, 0.0000e+00, 1.9705e-01,  ..., 1.2646e+00, 4.5592e-01,
         8.1191e-01],
        [1.3593e-01, 1.9705e-01, 0.0000e+00,  ..., 1.2257e+00, 6.4539e-01,
         8.9283e-01],
        ...,
        [1.2736e+00, 1.2646e+00, 1.2257e+00,  ..., 0.0000e+00, 1.3198e+00,
         7.4374e-01],
        [5.2427e-01, 4.5592e-01, 6.4539e-01,  ..., 1.3198e+00, 0.0000e+00,
         6.4848e-01],
        [8.5737e-01, 8.1191e-01, 8.9283e-01,  ..., 7.4374e-01, 6.4848e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.55868684, 13.89264822, 13.77222752, 13.60275211, 13.96905756,
       13.83139024, 13.59170809,  8.97052115, 13.00413829, 12.76408562,
       11.22596994, 11.97957425, 14.02058621, 13.85158417,  8.25828409,
        8.7907923 ,  9.47369628,  5.83560009,  8.66273171,  6.10774544,
        7.59795445,  6.16512856,  6.38105015, 11.00812996, 12.79126785,
        6.81961996,  8.43626114,  9.7712321 ,  8.44940661, 12.40074901,
       13.55868684, 11.38774149,  6.02588517,  8.67227386, 12.85032836,
        7.5403822 , 11.83509372,  6.30647746,  9.17595899,  5.20659892,
       10.06407839, 11.86239961,  5.80860165, 13.45642379,  6.65259775,
       11.97957425, 11.80092578, 13.32645552, 13.38091121, 12.4013153 ,
        9.88310614,  9.19837792, 13.6922425 , 13.1399181 , 10.27716434,
        8.16714836,  6.02779201,  7.92607342,  7.38844703,  5.56809335,
        5.05989547, 12.27835996, 13.91198191,  6.10360073, 11.54672755,
       10.46238054, 12.26997165, 13.6660671 ,  9.91668182,  8.53501777,
        7.8575648 , 12.49291768, 13.50484539,  5.36061974,  5.91677188,
        9.20340031,  9.4997987 , 13.98691426, 10.27716434,  9.14973319,
        9.61942445, 14.09357547,  7.53063885, 13.94008948,  6.383897  ,
        6.74872981,  5.41348121,  8.98765644, 12.72524615, 11.65923396,
       10.91805683, 10.53368644,  6.74872981, 12.59969936, 13.40850328,
        7.69752262, 12.86648771, 12.30695098, 10.06023665, 11.28093507,
        9.6282763 , 10.06023665,  9.48969978,  9.4997987 ,  5.92281142,
        8.7273569 ])}
{'P': tensor([[3.1883e-03, 2.5236e-01, 1.6244e-01,  ..., 6.6168e-07, 3.0602e-04,
         5.7237e-07],
        [1.9825e-06, 1.8803e-03, 4.4827e-02,  ..., 3.9588e-07, 1.9005e-05,
         3.1370e-07],
        [3.6134e-06, 5.9105e-05, 5.3218e-03,  ..., 1.0868e-06, 3.1439e-05,
         8.2737e-07],
        ...,
        [2.1064e-04, 1.2090e-04, 9.3729e-05,  ..., 1.1314e-01, 5.1736e-05,
         4.5169e-01],
        [5.2864e-04, 3.2041e-03, 1.1468e-02,  ..., 3.4284e-06, 4.0871e-02,
         2.7368e-06],
        [6.1656e-04, 3.3721e-04, 2.5109e-04,  ..., 1.6236e-02, 1.2696e-04,
         1.9216e-01]]), 'X': tensor([[  7.3592,  11.3937,   6.2407],
        [  4.8674,  12.8875,   6.7622],
        [  3.9022,  13.2906,   5.5788],
        [  5.2608,  12.6977,   4.9166],
        [  3.1064,  13.5607,   6.7605],
        [  2.5318,  13.7231,   6.3692],
        [  6.0560,  -0.9421,  -5.0334],
        [  1.7232,  -4.7574,   3.1270],
        [ 12.5770,   1.8512,  -0.6578],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.3569,   0.7591,   6.5011],
        [ -4.3963,   7.1629,   4.5263],
        [ -9.3705,   0.4208,  -3.5815],
        [  5.0233,   4.9647,  -3.1788],
        [ -7.2515,   5.0195,   4.6434],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -6.4464,   5.8272,   7.5038],
        [  6.1685,  12.1979,  -6.2473],
        [  9.5018,  -7.1945,  -1.1892],
        [ 12.6035,   0.7494,   7.1536],
        [ -1.5261,   7.8988,   9.3768],
        [  5.0277,   4.9589,   7.0175],
        [ -9.1231,  -3.5739,   8.4708],
        [ -6.0175,  -8.4541,  -9.8408],
        [  2.4572,   7.1085,   7.7376],
        [ -0.9715, -10.8630,   0.3555],
        [ 11.1420,   6.9331,  -2.4779],
        [  4.2275,  -3.6320,  -0.2714],
        [ 11.7418,   5.6070,  -1.4192],
        [ 12.3453,  -1.5462,   7.0212],
        [  9.5018,  -7.1945,  -1.1892],
        [ 10.2470,   8.4054,  -0.1177],
        [ -7.7842,   4.3466,  -1.7548],
        [  8.4691,  10.4571,  -5.0650],
        [ 11.2517,  -4.6495,   2.9800],
        [  2.3200,   7.1756,   1.1938],
        [  3.3069,   6.6089,   8.7377],
        [  3.7992,  -3.9474,  -8.9019],
        [ -8.9282,   2.1648,   8.4894],
        [ -6.2668,   5.9808,   2.5614],
        [  3.2692,  13.5098,  -4.9032],
        [  6.2419,  -0.2313,   4.2440],
        [  1.3767, -11.0341,   9.9415],
        [  4.8051, -10.3703,   0.5268],
        [ 12.3025,  -1.7440,  -6.6632],
        [  5.8889,  -1.3916,   8.8993],
        [  3.2366,  -4.2732,   5.1532],
        [ -3.0004,   7.6607,  -7.2640],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -8.3653,   3.4184,   9.3454],
        [  9.4554,   9.4245,   9.8800],
        [ -5.1369,   6.7755,   8.3836],
        [  2.5382,   7.0672,   4.5097],
        [ -6.0175,  -8.4541,  -9.8408],
        [ -9.4541,  -0.3327,   8.8246],
        [  0.8940, -11.0402,  -0.9783],
        [ -8.1905,   3.7253,   4.1175],
        [ -6.4822,  -8.0252,  -3.8983],
        [ -5.9872,   6.2037,  -9.4477],
        [ -9.4671,  -0.5553,   0.1117],
        [ -8.3062,  -5.6008,   0.0287],
        [ -9.1749,  -3.3803,   6.4629],
        [ -9.1907,  -3.3180,  -5.5943],
        [  2.5813,  -4.5471,   5.3597],
        [ -0.4341, -10.9473,   7.4509],
        [ -8.4267,  -5.3697,   2.4172],
        [ -1.8823, -10.6558,  -3.8952],
        [  6.9314,  -9.3435,  -1.2729],
        [ -3.0008, -10.2827,  -8.9480],
        [  5.8902,  -1.3887,   4.0388],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -2.2508, -10.5479,  -2.7235],
        [  2.5375,   7.0676,  -4.3490],
        [ -8.1204,  -5.9308,   4.0761],
        [ -8.8934,   2.2597,   3.5831],
        [  2.2659,  13.7894,   6.5436],
        [  5.0277,   4.9589,   7.0175],
        [  4.0587, -10.6112,   6.2273],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -2.0229, -10.6163,   8.6930],
        [ -2.9497, -10.3028,  -7.3479],
        [  0.2435,   7.8191,  -6.4709],
        [  2.3928,   7.1404,  -1.7035],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.0054,  -1.0902,   8.3079],
        [  0.5250, -11.0306,   5.5358],
        [  0.8572,  14.0501,  -8.3364],
        [  7.4132,  11.3526,   0.3322],
        [ -9.1813,  -3.3550,   1.8827],
        [ 10.9162,  -5.2599,  -7.0812],
        [  3.9591, -10.6391,  -6.2557],
        [ -8.6801,   2.7809,  -0.8443],
        [  1.9503,  -4.7172,  -6.2313],
        [ -8.6381,  -4.9238,  -4.4077],
        [ -5.6651,   6.4380,   1.3660],
        [ -5.9938,  -8.4745,   3.9735],
        [ -3.4986,   7.5152,   9.0712],
        [  7.1697,  11.5343,  -0.7276],
        [  6.6319,  11.9066,   3.5165],
        [  0.4119,  14.1019,  -4.1870],
        [  3.1064,  13.5607,   6.7605],
        [  3.9085,   6.1512,  -2.2560],
        [ -7.6650,   4.5092,   0.2520],
        [  0.9074,  -4.8139,   8.1674],
        [  6.2026,  12.1774,  -1.5887],
        [  2.1120,  -4.6818,   8.6144]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0450, 0.0476,  ..., 0.4557, 0.8426, 0.4212],
        [0.0450, 0.0000, 0.0905,  ..., 0.4769, 0.8875, 0.4427],
        [0.0476, 0.0905, 0.0000,  ..., 0.4208, 0.7980, 0.3862],
        ...,
        [0.4557, 0.4769, 0.4208,  ..., 0.0000, 0.7334, 0.0346],
        [0.8426, 0.8875, 0.7980,  ..., 0.7334, 0.0000, 0.7302],
        [0.4212, 0.4427, 0.3862,  ..., 0.0346, 0.7302, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.56367857, 13.77604293, 13.85158417, 13.744387  , 13.91198191,
       13.95472331,  6.12886024,  5.05989547, 12.7125079 ,  8.62162372,
        6.4020404 ,  8.40445872,  9.37989631,  7.06272256,  8.81930525,
        9.88571713,  8.68979052, 13.66896739, 11.9182956 , 12.62576026,
        8.04483495,  7.06169391,  9.79814997, 10.37698435,  7.52117132,
       10.90638066, 13.12299145,  5.57341623, 13.01187484, 12.44177641,
       11.9182956 , 13.25336001,  8.91551231, 13.45642379, 12.17451685,
        7.54127425,  7.39004226,  5.47866221,  9.18690018,  8.66273171,
       13.89974693,  6.24614905, 11.11969815, 11.42947632, 12.42554808,
        6.05113217,  5.36061974,  8.22728399, 12.42554808,  9.0368458 ,
       13.35013374,  8.50268322,  7.50917551, 10.37698435,  9.45995955,
       11.07637531,  8.99791316, 10.31613269,  8.62162372,  9.48337063,
       10.01804223,  9.77777346,  9.7712321 ,  5.2287058 , 10.95593729,
        9.99211663, 10.8207297 , 11.63383221, 10.71162681,  6.05165248,
       10.1979669 , 10.78533838,  7.5092834 , 10.05558893,  9.17595899,
       13.97430191,  7.06169391, 11.36089355, 10.1979669 , 10.80728788,
       10.7167285 ,  7.822854  ,  7.53063885,  4.78002665,  6.10360073,
       11.04313641, 14.07623266, 13.55868684,  9.7751189 , 12.11735334,
       11.35182589,  9.11473391,  5.10444097,  9.94285511,  8.57560275,
       10.37998078,  8.28968067, 13.58101794, 13.62896593, 14.10796466,
       13.91198191,  7.28793663,  8.89301516,  4.89869205, 13.6660671 ,
        5.13616224])}
{'P': tensor([[3.4908e-03, 2.7486e-01, 1.8662e-01,  ..., 2.1963e-03, 2.2518e-08,
         8.9376e-06],
        [7.9904e-06, 6.1766e-03, 1.0923e-05,  ..., 3.4553e-03, 4.4823e-08,
         6.6414e-06],
        [1.5874e-05, 4.7158e-01, 2.8644e-03,  ..., 1.8124e-03, 2.0209e-08,
         4.9085e-06],
        ...,
        [9.9088e-05, 2.7771e-04, 1.5148e-04,  ..., 6.3627e-02, 9.7482e-06,
         1.7664e-04],
        [1.2692e-05, 6.1407e-06, 9.8078e-06,  ..., 1.4860e-05, 4.9952e-02,
         4.6536e-04],
        [2.0429e-02, 1.6440e-02, 1.9365e-02,  ..., 3.4371e-02, 1.3914e-05,
         7.3956e-02]]), 'X': tensor([[  7.1697,  11.5343,  -0.7276],
        [  4.9152,  12.8653,  -1.4459],
        [  6.2026,  12.1774,  -1.5887],
        [  4.1637,  13.1898,  -1.9189],
        [  3.3633,  13.4794,   0.5193],
        [  6.9284,  11.7061,  -0.3732],
        [  5.9843,  -9.8679,  -6.1449],
        [  8.2422,  -8.4041,  -1.4962],
        [ -9.1749,  -3.3803,   6.4629],
        [  4.1140,  -3.7220,  -0.1273],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.8617,   3.4853,  -9.8080],
        [  0.6142,  14.0802,   0.5407],
        [  6.3589,   0.8239,  -8.4230],
        [  6.3460,   1.4252,   1.2291],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.1752,   2.5242,   3.5874],
        [ -1.8788, -10.6567,   3.7068],
        [ -8.8058,   2.4850,  -1.7425],
        [  5.4179,  -2.2925,  -3.4641],
        [ 10.9394,   7.3069,  -8.9365],
        [ -8.8154,   2.4611,   2.5198],
        [ -5.1974,   6.7396,   3.3484],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.5829,  -4.5466,  -8.4015],
        [  0.3231,  -4.7691,  -1.6120],
        [ -7.6525,  -6.6534,   5.1721],
        [  6.2790,  -0.0259,  -5.4116],
        [  5.7802,  -1.6363,  -6.2782],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 11.7779,   5.5127,  -2.0699],
        [  6.3211,   1.7017,  -0.2164],
        [  4.0642,  -3.7599,  -3.4565],
        [ -7.7842,   4.3466,  -1.7548],
        [  5.8201,   3.5841,  -8.0609],
        [  0.8572,  14.0501,  -8.3364],
        [ -1.6220,   7.8917,   0.0315],
        [  7.1697,  11.5343,  -0.7276],
        [ 11.1469,   6.9237,   1.6160],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.3087,   1.8053,  -1.9481],
        [  1.8766,   7.3694,   1.1887],
        [  6.3031,   1.8477,  -7.1706],
        [  8.9284,  10.0036,  -6.9210],
        [  6.2038,   2.4020,  -3.1692],
        [ 11.4743,   6.2478,  -2.1100],
        [  0.1460,   7.8340,  -9.4614],
        [ -4.8751,  -9.3148,   4.0406],
        [ 11.0438,   7.1180,  -5.6991],
        [ -9.4731,  -0.7120,  -6.0271],
        [  8.1528,  10.7452,   9.1493],
        [  9.9217,  -6.7006,  -5.6728],
        [  9.6587,   9.1817,  -0.8212],
        [ -0.5709,   7.9055,   7.6089],
        [ -4.8751,  -9.3148,   4.0406],
        [  6.3031,   1.8477,  -7.1706],
        [ -8.7544,   2.6096,   3.3146],
        [ 11.0438,   7.1180,  -5.6991],
        [  1.3339,   7.5627,  -0.9034],
        [ -2.2508, -10.5479,  -2.7235],
        [ 10.9162,  -5.2599,  -7.0812],
        [ 11.3691,   6.4760,   1.0787],
        [  7.1821,  -9.1845,   9.8548],
        [  4.6357,   5.4415,   7.9067],
        [ -0.2586,   7.8825,  -9.9840],
        [  1.2728, -11.0372,   9.8121],
        [  9.6587,   9.1817,  -0.8212],
        [ -4.9217,   6.8979,  -1.7667],
        [ -5.1369,   6.7755,   8.3836],
        [  9.6614,   9.1784,  -1.2087],
        [  0.3715,  -4.7756,   0.2312],
        [ 10.0995,  -6.4739,   2.0974],
        [  2.5974,   7.0363,  -7.7810],
        [  3.3943,   6.5483,   1.0790],
        [ -7.2515,   5.0195,   4.6434],
        [  0.1677,  -4.7447,  -6.1529],
        [ -7.6525,  -6.6534,   5.1721],
        [  1.7301,   7.4262,   5.0522],
        [  7.7750,  11.0665,   2.8364],
        [ -5.1974,   6.7396,   3.3484],
        [  0.8700,  -4.8132,  -7.8042],
        [  6.0140,  -1.0661,  -2.5040],
        [ -3.4100,   7.5436,  -7.6980],
        [  0.4119,  14.1019,  -4.1870],
        [  5.1861,   4.7361,  -3.7312],
        [ -3.9700,   7.3453,  -4.1486],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -1.5408,   7.8978,   5.3470],
        [  6.3587,   1.1596,  -3.6629],
        [  4.9337,  12.8566,  -3.6582],
        [  5.9587,  12.3216,   9.1695],
        [  1.7228,  -4.7575,   8.3396],
        [  2.0449,  13.8402,  -7.1642],
        [ 11.7779,   5.5127,  -2.0699],
        [  8.2154,  -8.4262,   8.0171],
        [  4.3459,  -3.5327,   0.3667],
        [  4.7544,  -3.1425,  -1.5315],
        [ 10.0567,   8.6697,   4.6329],
        [  4.5954,  13.0099,  -4.4439],
        [  6.2365,  -0.2578,  -6.0212],
        [  1.7228,  -4.7575,   8.3396],
        [ -0.1963,   7.8765,  -1.1371],
        [  2.2849, -10.9655,  -7.6542],
        [  6.0345,   3.0175,  -1.4424]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 3.9125e-02, 5.9559e-02,  ..., 5.0200e-01, 9.0171e-01,
         2.7178e-01],
        [3.9125e-02, 0.0000e+00, 2.8145e-02,  ..., 4.8448e-01, 8.6895e-01,
         2.5456e-01],
        [5.9559e-02, 2.8145e-02, 0.0000e+00,  ..., 4.9983e-01, 8.6944e-01,
         2.7132e-01],
        ...,
        [5.0200e-01, 4.8448e-01, 4.9983e-01,  ..., 0.0000e+00, 5.2222e-01,
         2.3036e-01],
        [9.0171e-01, 8.6895e-01, 8.6944e-01,  ..., 5.2222e-01, 0.0000e+00,
         6.7298e-01],
        [2.7178e-01, 2.5456e-01, 2.7132e-01,  ..., 2.3036e-01, 6.7298e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.58101794, 13.77222752, 13.6660671 , 13.83139024, 13.89264822,
       13.60275211, 11.54071859, 11.77124373,  9.77777346,  5.54776934,
        8.22728399,  6.81961996, 14.09357547,  6.41202986,  6.50409372,
       13.1399181 ,  6.67121748, 10.82106505,  9.14973319,  5.8828986 ,
       13.15525803,  9.15253344,  8.51088913,  9.93553424,  5.22903117,
        4.78002665, 10.14044489,  6.27905292,  6.00732193,  7.53063885,
        9.86089579,  9.75401449, 13.00413829,  6.54615887,  5.5366698 ,
        8.91551231,  6.83515201, 14.07623266,  8.05668609, 13.58101794,
       13.12218013, 10.37698435,  6.5618968 ,  7.60463349,  6.56833864,
       13.40850328,  6.65259775, 13.06498554,  7.83534987, 10.51340829,
       13.13888841,  9.4997987 , 13.48809595, 11.97238624, 13.32645552,
        7.92607342, 10.51340829,  6.56833864,  9.13507321, 13.13888841,
        7.67939546, 10.78533838, 12.11735334, 13.08414689, 11.65923396,
        7.14837514,  7.88677538, 11.11038551, 13.32645552,  8.47371868,
        8.50268322, 13.32613445,  4.79003361, 11.99633819,  7.50035355,
        7.37576154,  8.81930525,  4.74770899, 10.14044489,  7.62509582,
       13.5247215 ,  8.51088913,  4.89121805,  6.10774544,  8.27853076,
       14.10796466,  7.02326515,  8.34949383,  9.91668182,  8.04664886,
        6.46356806, 13.77075174, 13.68672855,  5.05981914, 13.99048094,
       13.00413829, 11.76830981,  5.60064319,  5.69913817, 13.2778378 ,
       13.79762415,  6.24187044,  5.05981914,  7.8788965 , 11.20100795,
        6.74686071])}
{'P': tensor([[2.3852e-03, 3.7805e-01, 2.3114e-01,  ..., 5.6041e-05, 1.0064e-06,
         3.4307e-07],
        [1.4798e-05, 8.1878e-03, 1.1540e-01,  ..., 2.3935e-04, 5.0762e-06,
         1.1552e-06],
        [1.2409e-05, 2.2516e-04, 6.7506e-03,  ..., 1.4952e-04, 4.7069e-06,
         1.0522e-06],
        ...,
        [4.5272e-05, 6.4823e-05, 6.3423e-05,  ..., 3.4905e-01, 3.9552e-03,
         6.1140e-05],
        [6.4289e-07, 5.7624e-07, 5.5686e-07,  ..., 5.0995e-06, 3.3518e-02,
         1.0519e-03],
        [2.0043e-04, 1.4931e-04, 1.5414e-04,  ..., 2.3433e-05, 2.5781e-04,
         2.0868e-01]]), 'X': tensor([[ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 9.6587e+00,  9.1817e+00, -8.2124e-01],
        [-7.4807e+00,  4.7463e+00,  2.0616e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 9.4775e-01, -4.8144e+00,  6.3711e-01],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [ 9.9669e-01, -4.8144e+00, -9.7370e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-6.1914e+00, -8.2997e+00,  8.5380e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [ 5.0506e+00,  4.9277e+00, -9.5094e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-3.0644e+00,  7.6439e+00, -1.4322e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 1.6422e+00,  7.4586e+00, -6.8056e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0337, 0.0924,  ..., 1.0275, 1.2280, 0.4189],
        [0.0337, 0.0000, 0.1249,  ..., 0.9954, 1.2106, 0.4238],
        [0.0924, 0.1249, 0.0000,  ..., 1.1041, 1.2536, 0.3910],
        ...,
        [1.0275, 0.9954, 1.1041,  ..., 0.0000, 0.8034, 1.0278],
        [1.2280, 1.2106, 1.2536,  ..., 0.8034, 0.0000, 0.9327],
        [0.4189, 0.4238, 0.3910,  ..., 1.0278, 0.9327, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.58333857, 13.69162795, 13.72594022, 13.66991027, 13.70893492,
       13.92390689,  7.42512621, 11.63724792,  8.68979052,  5.48276038,
        6.54693987, 12.58006038, 11.52733581, 10.06286074, 10.35093485,
       14.0393949 ,  4.89869205,  7.32950385,  6.18668463,  7.27340834,
        6.54693987,  9.77777346, 11.62770332, 13.32645552,  8.85939626,
        6.67121748,  6.25586297, 12.12641969, 11.87412439,  7.66377554,
        6.33075682, 12.27835996,  5.21280156, 13.35927527,  4.90676242,
        7.92200342, 12.49317563, 14.08549802, 10.33712258,  8.12580976,
        9.43112957, 11.83075005,  5.47702766, 12.50996828,  4.91652556,
        6.07992623, 11.38774149,  8.31625171, 13.52430176,  5.64865173,
       13.744387  , 10.46739073,  8.51088913, 12.00995898,  8.62631801,
        6.77777606, 11.0050379 , 11.72802491, 10.05558893,  8.08320235,
       13.97779839, 14.09176259, 12.59969936,  5.91222346,  6.61386436,
        9.14973319, 13.97857591, 12.35421461,  9.17016857, 10.06703358,
       13.95472331, 10.24557047, 13.20597932, 11.04953767, 12.27835996,
       10.35464916,  8.06990483,  6.56833864,  7.05627461, 11.1913598 ,
        7.09887557,  7.58668144,  6.53290663,  6.69389431, 10.1371552 ,
        7.63726134,  9.75401449,  5.84533345,  7.37902766, 12.67680439,
       13.56367857,  8.23525433,  8.56889738, 12.88270018,  8.12043983,
        9.13507321,  7.63726134, 13.04156075,  9.88310614, 12.88270018,
       12.18045978, 10.1979669 ,  8.44940661,  8.27853076,  9.78285424,
        4.85603886])}
{'P': tensor([[4.2660e-03, 2.5436e-01, 2.5250e-01,  ..., 9.3951e-07, 4.8298e-07,
         9.2816e-08],
        [2.3691e-06, 3.3993e-03, 4.4352e-06,  ..., 2.2521e-07, 4.3904e-07,
         3.0948e-08],
        [1.7681e-05, 4.1604e-01, 4.2379e-03,  ..., 8.1832e-07, 5.6197e-07,
         6.7707e-08],
        ...,
        [3.3031e-06, 8.6897e-06, 7.0819e-06,  ..., 6.5469e-03, 5.5558e-07,
         9.2709e-08],
        [5.3814e-05, 3.1134e-05, 5.0515e-05,  ..., 4.7727e-05, 9.8941e-02,
         3.1950e-03],
        [1.7310e-03, 4.1940e-04, 1.2281e-03,  ..., 1.7931e-04, 1.1956e-05,
         1.7200e-02]]), 'X': tensor([[  7.0642,  11.6104,  -9.8316],
        [  3.7298,  13.3537,  -9.4653],
        [  5.8930,  12.3592,  -8.9201],
        [  6.2675,  12.1379,  -8.9720],
        [  2.5469,  13.7192,  -9.6935],
        [  2.7295,  13.6703,  -8.9328],
        [  4.8674,  12.8875,   6.7622],
        [  1.5662,  13.9375,   4.5142],
        [ -9.1231,  -3.5739,   8.4708],
        [ -1.9487,   7.8590,  -3.3330],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -8.6112,   2.9319,  -7.9943],
        [  0.5850,  -4.7982,   9.6639],
        [ -5.9940,   6.1985,   5.9566],
        [  9.3836,   9.5075,  -5.9278],
        [ 10.4758,  -5.9539,   1.3667],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.0560,  -0.9421,  -5.0334],
        [ -7.4808,   4.7462,   2.0266],
        [  4.6992,   5.3690,   3.0935],
        [  2.2644,  -4.6434,   9.7508],
        [  6.8407,  11.7666,   3.0045],
        [ -1.9646,   7.8571,   3.3817],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.3900,   3.3727,   3.1650],
        [  3.4604,  -4.1545,   1.8091],
        [ 12.5927,   0.4199,  -9.4997],
        [ -9.4630,  -1.5145,   0.2983],
        [  1.6422,   7.4586,  -6.8056],
        [ -9.0961,   1.6529,   3.1553],
        [  5.1793,   4.7461,   0.3899],
        [ 12.5763,   0.1333,  -9.2650],
        [ -0.7599, -10.8995,  -6.0366],
        [  2.2373,  -4.6506,  -1.7334],
        [  4.5954,  13.0099,  -4.4439],
        [  2.6970,   6.9826,   9.7350],
        [ 12.1751,  -2.2592,  -5.7751],
        [  2.0870,  -4.6877,   9.6823],
        [  9.0051,   9.9234,   5.4411],
        [  4.6992,   5.3690,   3.0935],
        [ 11.5966,   5.9666,   6.0083],
        [  2.2076,  13.8032,  -1.5099],
        [  6.3535,   0.6792,  -7.5101],
        [  6.1372,   2.6726,  -6.7885],
        [ -4.7642,  -9.3860,  -5.1133],
        [  3.2701,   6.6338,   9.9209],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -6.7992,   5.4991,   9.9186],
        [  9.6587,   9.1817,  -0.8212],
        [  6.9314,  -9.3435,  -1.2729],
        [ -4.3963,   7.1629,   4.5263],
        [  6.2804,   2.0019,   0.8607],
        [  4.7156,  -3.1831,  -8.0224],
        [  5.7784,  -1.6401,   9.5587],
        [  5.8275,  -1.5334,   2.0124],
        [  5.9193,  -1.3176,   5.1739],
        [  5.5424,  -2.0885,   7.2487],
        [  9.6614,   9.1784,  -1.2087],
        [ 12.6059,   0.9597,  -6.3264],
        [  3.5965,  -4.0754,  -3.5035],
        [ 10.9916,   7.2133,  -0.9179],
        [ -7.6293,  -6.6859,  -5.5161],
        [ -7.6525,  -6.6534,   5.1721],
        [ -9.4430,  -0.1897,   4.8383],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.0955,   2.8210,  -4.3321],
        [ -6.0844,  -8.3955,   0.7705],
        [ -9.4716,  -0.6673,   7.6068],
        [  3.5939,   6.4029,  -1.8357],
        [ -7.6650,   4.5092,   0.2520],
        [ 11.5568,   6.0602,   9.3244],
        [  0.9945,  -4.8144,  -0.1897],
        [  9.7328,   9.0902,  -0.8917],
        [  0.5850,  -4.7982,   9.6639],
        [  3.2809,  13.5061,  -4.0199],
        [ -4.7389,   6.9953,   3.7638],
        [  6.0345,   3.0175,  -1.4424],
        [ 12.5763,   0.1333,  -9.2650],
        [  3.6469,   6.3624,  -1.8414],
        [  6.0467,  -9.8369,  -6.5816],
        [  6.1575,  12.2046,   3.3257],
        [ -1.0482, -10.8487,  -5.9066],
        [  6.3505,   0.6235,  -8.5849],
        [ 11.8951,  -3.1570,   7.0742],
        [ 12.5258,  -0.4200,  -9.3493],
        [  3.6154,  -4.0640,  -7.9698],
        [  5.4867,  12.5816,   2.9549],
        [ -1.9487,   7.8590,  -3.3330],
        [ 12.0042,  -2.8350,   2.6864],
        [  8.1043,  -8.5164,  -7.9290],
        [ -4.8751,  -9.3148,   4.0406],
        [  6.9284,  11.7061,  -0.3732],
        [  1.8447,  -4.7372,   8.1235],
        [  7.7794,  11.0629,   8.9069],
        [ 10.7566,   7.6212,  -2.3279],
        [ -7.6293,  -6.6859,  -5.5161],
        [  6.1066,  -0.7799,  -4.1746],
        [  2.9410,   6.8428,   5.4458],
        [ 10.0619,   8.6627,  -9.2884],
        [ 12.6035,   0.7494,   7.1536],
        [ -7.4733,   4.7555,   6.2490],
        [  3.3744,   6.5623,  -6.3636],
        [  8.1528,  10.7452,   9.1493],
        [  0.2732,  -4.7619,  -1.1577],
        [ 12.2921,  -1.7902,  -4.0870]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.5143e-02, 7.1152e-04,  ..., 1.4484e+00, 6.6196e-01,
         3.3475e-01],
        [3.5143e-02, 0.0000e+00, 3.4598e-02,  ..., 1.4576e+00, 6.4659e-01,
         3.2016e-01],
        [7.1152e-04, 3.4598e-02, 0.0000e+00,  ..., 1.4490e+00, 6.6211e-01,
         3.3490e-01],
        ...,
        [1.4484e+00, 1.4576e+00, 1.4490e+00,  ..., 0.0000e+00, 1.0520e+00,
         1.2194e+00],
        [6.6196e-01, 6.4659e-01, 6.6211e-01,  ..., 1.0520e+00, 0.0000e+00,
         3.2721e-01],
        [3.3475e-01, 3.2016e-01, 3.3490e-01,  ..., 1.2194e+00, 3.2721e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.59056613, 13.86480175, 13.6922425 , 13.66051884, 13.95361098,
       13.94008948, 13.77604293, 14.02526505,  9.79814997,  8.09704052,
       10.5575601 ,  9.51074763,  9.09661212,  4.83371341,  8.62260327,
       13.35832725, 12.0495521 ,  5.56809335,  6.12886024,  8.859416  ,
        7.13501264,  5.16611782, 13.6105595 ,  8.09899874,  6.383897  ,
        9.04255655,  5.40689332, 12.59969936,  9.58347456,  7.63726134,
        9.24501829,  7.02496223, 12.5769659 , 10.92597196,  5.16077764,
       13.79762415,  7.48538905, 12.3828957 ,  5.13125513, 13.40024224,
        7.13501264, 13.04156075, 13.97857591,  6.38968357,  6.69389431,
       10.52587243,  7.39599986, 12.49317563,  8.74470218, 13.32645552,
       11.63383221,  8.40445872,  6.59176018,  5.68942572,  6.00663314,
        6.02588517,  6.06416957,  5.92281142, 13.32613445, 12.64235359,
        5.43544626, 13.1471313 , 10.14436912, 10.14044489,  9.44486609,
       12.00995898,  6.7166297 , 10.36844813,  9.49511524,  7.34250948,
        8.89301516, 13.0493368 ,  4.9160962 , 13.31764181,  4.83371341,
       13.89886469,  8.44940661,  6.74686071, 12.5769659 ,  7.33350642,
       11.54672755, 13.66991027, 10.8992524 ,  6.38105015, 12.30695098,
       12.53285633,  5.43943791, 13.72594022,  8.09704052, 12.3344536 ,
       11.75618513, 10.51340829, 13.60275211,  5.08372149, 13.52430176,
       13.18278606, 10.14436912,  6.1561646 ,  7.44805901, 13.27717982,
       12.62576026,  8.8580671 ,  7.37902766, 13.48809595,  4.76969364,
       12.4217523 ])}
{'P': tensor([[3.4434e-03, 3.9464e-01, 2.6324e-01,  ..., 2.7508e-08, 8.4282e-07,
         4.5768e-06],
        [1.2224e-05, 9.8062e-03, 6.3905e-05,  ..., 7.5134e-08, 9.8867e-07,
         4.0847e-06],
        [2.5010e-05, 3.4537e-01, 6.7275e-03,  ..., 5.1630e-08, 1.0194e-06,
         3.8840e-06],
        ...,
        [7.0147e-06, 3.0939e-06, 4.3576e-06,  ..., 7.9376e-03, 1.1876e-04,
         2.3354e-05],
        [1.6401e-01, 6.8479e-02, 8.0359e-02,  ..., 3.9115e-07, 2.4987e-02,
         1.2364e-01],
        [3.1461e-01, 1.6032e-01, 3.0391e-01,  ..., 8.0668e-08, 1.5043e-05,
         8.3872e-03]]), 'X': tensor([[  7.0515,  11.6194,   8.4289],
        [  5.0617,  12.7957,   8.3544],
        [  5.9587,  12.3216,   9.1695],
        [  2.7590,  13.6621,   9.4496],
        [  4.8674,  12.8875,   6.7622],
        [  3.1064,  13.5607,   6.7605],
        [  5.9883,   3.1536,   6.2320],
        [  3.7298,  13.3537,  -9.4653],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.1793,   4.7461,   0.3899],
        [  2.3490,  -4.6199,  -4.2489],
        [  8.6420,  -8.0580,   6.5781],
        [  9.5261,  -7.1675,   1.4206],
        [  5.8979,   3.3958,  -3.8713],
        [  0.6396,  14.0772,  -6.3786],
        [  3.3943,   6.5483,   1.0790],
        [  6.1575,  12.2046,   3.3257],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -6.3306,   5.9272,   5.6935],
        [  4.0362,  13.2397,  -7.0030],
        [ -9.4117,   0.1160,   8.2574],
        [  5.9003,  12.3550,   2.1917],
        [  7.1821,  -9.1845,   9.8548],
        [ -9.4766,  -1.1116,  -2.1706],
        [ -3.1671,   7.6158,   6.0765],
        [  0.3715,  -4.7756,   0.2312],
        [  0.1047, -11.0046,   0.5188],
        [  0.9788,  -4.8145,  -3.8661],
        [  3.4140,  -4.1803,  -8.8224],
        [ -6.6895,   5.6051,   2.7899],
        [  5.3804,  -2.3504,   7.2744],
        [  5.0190,   4.9705,  -7.9752],
        [  3.3744,   6.5623,  -6.3636],
        [  5.3804,  -2.3504,   7.2744],
        [ 10.8218,  -5.4177,   6.8573],
        [  4.5340,  -3.3629,   8.9652],
        [  3.6703,   6.3443,  -5.4401],
        [ -8.8058,   2.4850,  -1.7425],
        [  3.3943,   6.5483,   1.0790],
        [  6.1685,  12.1979,  -6.2473],
        [  2.5128, -10.9365,   9.9645],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.2990,  -4.2415,  -7.4857],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.6095,  -1.9702,  -7.8233],
        [  1.4749,  -4.7892,  -0.8089],
        [ -6.0195,   6.1789,  -2.9496],
        [  0.5960, -11.0335,   3.1144],
        [  6.2555,   2.1474,   4.1685],
        [ -0.4341, -10.9473,   7.4509],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.9548,  -1.2273,  -2.6520],
        [ -1.9646,   7.8571,   3.3817],
        [  8.4691,  10.4571,  -5.0650],
        [ -1.5451,   7.8975,   7.6104],
        [ 12.5650,   2.0132,   4.6272],
        [ 11.9641,  -2.9568,   7.7211],
        [  4.0078,  -3.8018,  -6.2168],
        [  7.1821,  -9.1845,   9.8548],
        [  5.5101,  -2.1433,   9.5442],
        [ -5.9225,  -8.5355,   2.4778],
        [ -2.0840,   7.8415,  -7.8657],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.0961,   1.6529,   3.1553],
        [ -5.9225,  -8.5355,   2.4778],
        [  5.6414, -10.0303,   7.8525],
        [  2.9260,  -4.4162,   8.3808],
        [  3.2701,   6.6338,   9.9209],
        [ -9.2508,   1.0572,   2.9978],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.3535,   0.6792,  -7.5101],
        [ -1.5451,   7.8975,   7.6104],
        [ -9.2707,   0.9661,   4.4863],
        [  7.7750,  11.0665,   2.8364],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.3569,   0.7591,   6.5011],
        [  5.8930,  12.3592,  -8.9201],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.6000,   1.3839,  -3.2992],
        [ -7.6293,  -6.6859,  -5.5161],
        [  1.9241,   7.3503,   7.9004],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.7565,  11.8237,  -4.9882],
        [  9.6614,   9.1784,  -1.2087],
        [  8.6420,  -8.0580,   6.5781],
        [ -1.5408,   7.8978,   5.3470],
        [ -1.3940, -10.7772,   6.7835],
        [ 12.4597,  -0.9136,  -4.8763],
        [  9.0129,  -7.7072,   9.1400],
        [  7.9816,  10.8938,  -0.5378],
        [ -9.4501,  -1.7133,   2.2980],
        [ -0.6323,   7.9085,  -5.9544],
        [  5.7963,  -1.6018,   1.8856],
        [  2.1771, -10.9776,  -3.2007],
        [  2.5375,   7.0676,  -4.3490],
        [ -7.6525,  -6.6534,   5.1721],
        [ 11.3367,  -4.4806,   0.4643],
        [  0.3715,  -4.7756,   0.2312],
        [ -8.6381,  -4.9238,  -4.4077],
        [  6.3360,   0.4258,  -9.2509],
        [  6.0031,  12.2958,  -6.5992],
        [  6.3087,   1.8053,  -1.9481],
        [ -8.1531,  -5.8748,   2.6934],
        [  4.3523, -10.5232,   8.2138],
        [ 11.5966,   5.9666,   6.0083],
        [  9.4554,   9.4245,   9.8800]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0265, 0.0190,  ..., 0.8853, 0.2194, 0.0190],
        [0.0265, 0.0000, 0.0172,  ..., 0.8589, 0.2009, 0.0075],
        [0.0190, 0.0172, 0.0000,  ..., 0.8701, 0.2181, 0.0127],
        ...,
        [0.8853, 0.8589, 0.8701,  ..., 0.0000, 0.7532, 0.8664],
        [0.2194, 0.2009, 0.2181,  ..., 0.7532, 0.0000, 0.2063],
        [0.0190, 0.0075, 0.0127,  ..., 0.8664, 0.2063, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.59170809, 13.76047999, 13.68672855, 13.93790235, 13.77604293,
       13.91198191,  6.76791483, 13.86480175, 11.73848065,  7.02496223,
        5.18277858, 11.8159254 , 11.92133129,  6.80559669, 14.09176259,
        7.37576154, 13.66991027,  9.96152788,  8.67227386, 13.84126197,
        9.41244865, 13.69162795, 11.65923396,  9.54154096,  8.24808832,
        4.79003361, 11.00508681,  4.91296205,  5.39723241,  8.7273569 ,
        5.87132718,  7.06372172,  7.37902766,  5.87132718, 12.10222204,
        5.64500512,  7.32950385,  9.14973319,  7.37576154, 13.66896739,
       11.22142195,  5.5366698 ,  5.37343173, 10.06850677,  5.94541803,
        5.01114719,  8.62631801, 11.04953767,  6.61386436, 10.95593729,
        8.56889738,  6.07992623,  8.09899874, 13.45642379,  8.04718433,
       12.72524615, 12.32408516,  5.52416368, 11.65923396,  5.91222346,
       10.38897563,  8.1137452 ,  8.56889738,  9.24501829, 10.38897563,
       11.50791859,  5.29755487,  7.39599986,  9.3109884 ,  5.90035324,
        6.38968357,  8.04718433,  9.32094055, 13.5247215 ,  9.5014146 ,
        6.4020404 , 13.6922425 ,  4.8763135 , 12.67576792, 10.14436912,
        7.59795445, 12.00995898, 13.61801526, 13.32613445, 11.8159254 ,
        8.04664886, 10.86693708, 12.49317563, 11.85890664, 13.50484539,
        9.60412735,  7.9337647 ,  6.01356004, 11.1913598 ,  7.5092834 ,
       10.14044489, 12.18998195,  4.79003361,  9.94285511,  6.35029274,
       13.6829843 ,  6.5618968 , 10.04916678, 11.38774149, 13.04156075,
       13.35013374])}
{'P': tensor([[2.0176e-03, 5.1879e-01, 1.6505e-01,  ..., 7.1439e-08, 1.4604e-07,
         4.3835e-05],
        [1.1296e-05, 3.7957e-03, 3.5252e-01,  ..., 1.4069e-07, 2.6196e-07,
         8.5496e-05],
        [3.7344e-06, 4.2854e-06, 3.3296e-03,  ..., 1.8037e-07, 1.7487e-07,
         1.0367e-04],
        ...,
        [2.2715e-06, 2.1206e-06, 1.2765e-06,  ..., 5.9819e-02, 2.2210e-05,
         1.7218e-06],
        [2.4046e-04, 2.3362e-04, 1.9941e-04,  ..., 3.4504e-06, 9.6785e-02,
         2.1112e-03],
        [1.0061e-05, 1.0912e-05, 2.2989e-05,  ..., 1.7993e-05, 6.3014e-05,
         5.4526e-02]]), 'X': tensor([[  7.0171,  11.6439,  -5.0178],
        [  6.7565,  11.8237,  -4.9882],
        [  4.5954,  13.0099,  -4.4439],
        [  3.2692,  13.5098,  -4.9032],
        [  4.9337,  12.8566,  -3.6582],
        [  3.2809,  13.5061,  -4.0199],
        [ -8.0750,  -6.0072,  -5.8267],
        [ -3.3812, -10.1235,   7.4384],
        [  0.3038,   7.8092,   0.4164],
        [  6.2675,  12.1379,  -8.9720],
        [ 12.0772,  -2.6014,  -7.1832],
        [ 12.0042,  -2.8350,   2.6864],
        [ -3.1671,   7.6158,   6.0765],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.8889,  -1.3916,   8.8993],
        [  5.1292,   4.8183,  -5.7997],
        [ -7.6719,  -6.6260,   7.1468],
        [  2.6613,   7.0021,   4.5783],
        [  9.7328,   9.0902,  -0.8917],
        [  6.2576,   2.1359,  -3.3398],
        [ -9.4671,  -0.5553,   0.1117],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.6310, -10.9195,   3.8054],
        [  6.3246,   1.6696,  -0.1544],
        [  7.2340,  -9.1505,  -5.8321],
        [ -0.8675, -10.8815,   6.8561],
        [  1.7218,  -4.7576,  -3.3617],
        [  0.6142,  14.0802,   0.5407],
        [  6.0345,   3.0175,  -1.4424],
        [ 11.3800,  -4.3919,   1.4066],
        [  4.4915,   5.5990,   7.0405],
        [ -2.6735, -10.4062,  -1.5911],
        [  0.9967,  -4.8144,  -9.7370],
        [ -9.4697,  -0.6157,  -6.8472],
        [  5.7802,  -1.6363,  -6.2782],
        [  2.0419,  -4.6978,  -5.0983],
        [  8.8091,  -7.9037,   0.3631],
        [ -7.7239,   4.4298,   3.1216],
        [  2.0937,  13.8293,  -2.3533],
        [ -1.0482, -10.8487,  -5.9066],
        [  4.6357,   5.4415,   7.9067],
        [ -5.0943,  -9.1682,   0.5073],
        [ -8.9379,  -4.1655,  -2.8114],
        [  4.5325,  -3.3643,   6.7699],
        [  6.3606,   0.9204,   3.2525],
        [  5.2939,  -2.4784,  -4.5840],
        [ -8.8745,   2.3097,   3.5600],
        [ -3.3116, -10.1539,   7.5919],
        [ 10.8218,  -5.4177,   6.8573],
        [  1.7815,  -4.7481,   4.4289],
        [  5.0617,  12.7957,   8.3544],
        [  4.4290,  -3.4596,   1.4181],
        [ -0.1626, -10.9796,   6.4629],
        [  2.6970,   6.9826,   9.7350],
        [  5.2199,   4.6861,  -1.9304],
        [  7.1441,  11.5529,   2.3232],
        [  2.4149,  -4.6005,  -4.3339],
        [  2.7121,   6.9743,  -8.4202],
        [ 12.3609,   3.4769,  -8.5762],
        [  2.3200,   7.1756,   1.1938],
        [  5.1793,   4.7461,   0.3899],
        [  4.8432,   5.1968,   1.5006],
        [ -9.2301,  -3.1537,  -2.7477],
        [  8.5716,  10.3596,   6.8818],
        [  2.6970,   6.9826,   9.7350],
        [ -7.4602,   4.7718,  -4.6277],
        [ -9.1623,  -3.4288,  -4.9961],
        [  3.2990,  -4.2415,  -7.4857],
        [  1.6422,   7.4586,  -6.8056],
        [  1.6422,   7.4586,  -6.8056],
        [  1.6954,  -4.7616,   9.8198],
        [  5.8275,  -1.5334,   2.0124],
        [  0.1041, -11.0045,   2.4311],
        [  6.1973,  -0.4372,   6.5720],
        [ 11.5935,   5.9740,  -4.3952],
        [  3.9770, -10.6341,   8.0783],
        [  0.8940, -11.0402,  -0.9783],
        [ -6.2200,  -8.2737,   5.5079],
        [  1.1274,  14.0116,   8.1753],
        [ 12.2435,   4.0076,   8.1257],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.9883,   3.1536,   6.2320],
        [ 11.4495,   6.3026,   6.8257],
        [ 11.1085,  -4.9204,   0.0736],
        [  3.6154,  -4.0640,  -7.9698],
        [ 11.1420,   6.9331,  -2.4779],
        [ -2.6069, -10.4298,  -6.1995],
        [  6.0054,  -1.0902,   8.3079],
        [  6.3200,   0.2690,   1.5900],
        [ 11.1085,  -4.9204,   0.0736],
        [  2.0729,   7.2878,   7.0880],
        [  5.8836,   3.4315,  -4.0327],
        [ -6.7745,  -7.7264,   8.4684],
        [  5.5424,  -2.0885,   7.2487],
        [  7.1441,  11.5529,   2.3232],
        [ -3.5520,   7.4976,   9.1522],
        [  6.1973,  -0.4372,   6.5720],
        [  4.4915,   5.5990,   7.0405],
        [  6.2905,   1.9366,   2.1169],
        [  4.3554,   5.7393,  -9.0766],
        [ -5.1974,   6.7396,   3.3484],
        [  2.2644,  -4.6434,   9.7508],
        [  9.6816,   9.1536,  -5.8543],
        [ -5.2752,  -9.0409,  -8.2568],
        [  6.0054,  -1.0902,   8.3079],
        [ -2.5056,   7.7720,   5.5878]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.9282e-03, 4.1690e-02,  ..., 1.2246e+00, 9.7099e-01,
         1.0554e+00],
        [8.9282e-03, 0.0000e+00, 3.2777e-02,  ..., 1.2191e+00, 9.6325e-01,
         1.0465e+00],
        [4.1690e-02, 3.2777e-02, 0.0000e+00,  ..., 1.2008e+00, 9.3403e-01,
         1.0139e+00],
        ...,
        [1.2246e+00, 1.2191e+00, 1.2008e+00,  ..., 0.0000e+00, 1.4429e+00,
         1.1537e+00],
        [9.7099e-01, 9.6325e-01, 9.3403e-01,  ..., 1.4429e+00, 2.1453e-04,
         3.7891e-01],
        [1.0554e+00, 1.0465e+00, 1.0139e+00,  ..., 1.1537e+00, 3.7891e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.59480906, 13.61801526, 13.79762415, 13.89974693, 13.77075174,
       13.89886469, 10.06437598, 10.67322527,  7.81509631, 13.66051884,
       12.35421461, 12.3344536 ,  8.24808832, 12.33167382,  6.05113217,
        7.03733775, 10.1371552 ,  7.49076899, 13.31764181,  6.61211224,
        9.48337063,  4.92497514, 11.23201619,  6.54127431, 11.66453059,
       10.91601655,  5.05963217, 14.09357547,  6.74686071, 12.19805394,
        7.17789666, 10.74409834,  4.91652556,  9.48969978,  6.00732193,
        5.1224079 , 11.83509372,  8.90405823, 13.98691426, 10.8992524 ,
        7.14837514, 10.48840476,  9.86089579,  5.64464807,  6.42689705,
        5.84533345,  9.17016857, 10.68031351, 12.10222204,  5.0713364 ,
       13.76047999,  5.62005629, 10.98076873,  7.48538905,  7.01475376,
       13.58333857,  5.19577146,  7.48311002, 12.84056604,  7.54127425,
        7.02496223,  7.10378919,  9.75401449, 13.44593519,  7.48538905,
        8.85572292,  9.78286733,  5.37343173,  7.63726134,  7.63726134,
        5.05444288,  6.02588517, 11.0050379 ,  6.21275016, 13.04217487,
       11.35345345, 11.07637531, 10.35093485, 14.05687815, 12.88270018,
       12.13568598,  6.76791483, 13.06957425, 12.1494023 ,  5.43943791,
       13.12299145, 10.7506423 ,  6.10360073,  6.32572323, 12.1494023 ,
        7.57685908,  6.81117802, 10.27572673,  5.92281142, 13.58333857,
        8.29640613,  6.21275016,  7.17789666,  6.58184474,  7.20482224,
        8.51088913,  5.16611782, 13.32374056, 10.46739073,  6.10360073,
        8.16584537])}
{'P': tensor([[4.2150e-03, 2.6924e-01, 1.4331e-01,  ..., 3.9450e-07, 2.0611e-07,
         1.4169e-07],
        [3.4630e-05, 3.9447e-02, 5.4022e-05,  ..., 2.2962e-06, 3.4879e-06,
         1.1716e-06],
        [4.5412e-05, 6.7257e-01, 5.9569e-03,  ..., 5.8012e-07, 3.3839e-07,
         2.2920e-07],
        ...,
        [3.2212e-04, 3.1075e-04, 4.5537e-04,  ..., 1.4255e-01, 3.9710e-06,
         1.9354e-03],
        [8.9201e-06, 4.8278e-06, 6.7906e-06,  ..., 4.6491e-05, 2.6817e-01,
         2.2752e-04],
        [8.8483e-05, 6.1547e-05, 1.0251e-04,  ..., 1.5989e-01, 9.4627e-06,
         1.3015e-01]]), 'X': tensor([[  6.9956,  11.6590,  -7.6876],
        [  4.0362,  13.2397,  -7.0030],
        [  6.0031,  12.2958,  -6.5992],
        [  2.7295,  13.6703,  -8.9328],
        [  3.7298,  13.3537,  -9.4653],
        [  5.8930,  12.3592,  -8.9201],
        [  5.4179,  -2.2925,  -3.4641],
        [ 11.1085,  -4.9204,   0.0736],
        [ 12.5927,   0.4199,  -9.4997],
        [ 11.1420,   6.9331,  -2.4779],
        [ -5.9872,   6.2037,  -9.4477],
        [  9.0652,   9.8597,  -3.5949],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.3860,  -2.3418,   1.0826],
        [  8.4691,  10.4571,  -5.0650],
        [  7.8425,  -8.7202,   6.4530],
        [ -9.3604,   0.4855,  -3.0158],
        [  0.1287, -11.0065,   6.5757],
        [ 12.5763,   0.1333,  -9.2650],
        [ -8.6112,   2.9319,  -7.9943],
        [  3.6249, -10.7255,   5.6605],
        [ -6.8618,  -7.6322,   3.7027],
        [  2.2076,  13.8032,  -1.5099],
        [  9.5203,   9.3483,   5.8781],
        [  4.8051, -10.3703,   0.5268],
        [  3.9895,   6.0816,  -6.5235],
        [ -8.4887,  -5.2448,  -0.6585],
        [ 11.9695,  -2.9407,  -6.3498],
        [  1.7000,  -4.7609,   2.7935],
        [ 12.3759,  -1.3943,   6.4659],
        [  6.2036,   2.4030,  -1.2500],
        [  5.4732,  -2.2041,  -4.3830],
        [  2.2182,  13.8007,   5.1889],
        [  6.9433,  -9.3362,  -0.3421],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -7.8126,   4.3066,   5.1791],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.8201,   3.5841,  -8.0609],
        [ -6.8466,  -7.6488,   1.9648],
        [ -8.6050,   2.9451,  -0.7012],
        [ -6.6998,   5.5954,   0.0823],
        [ 11.1420,   6.9331,  -2.4779],
        [ -8.9379,  -4.1655,  -2.8114],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.7295,  13.6703,  -8.9328],
        [ 10.8218,  -5.4177,   6.8573],
        [  2.1824,  -4.6647,   1.1985],
        [ -1.7290,   7.8825,  -3.3831],
        [  2.5382,   7.0672,   4.5097],
        [ -2.1382,   7.8339,   7.3524],
        [ -9.2301,  -3.1537,  -2.7477],
        [  3.9591, -10.6391,  -6.2557],
        [  3.8111, -10.6787,  -4.6539],
        [ -2.6735, -10.4062,  -1.5911],
        [  4.4290,  -3.4596,   1.4181],
        [  6.2242,  -0.3171,   7.4978],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -3.2486,   7.5925,  -1.3719],
        [ 10.9717,  -5.1646,  -6.9039],
        [  4.5800,   5.5034,  -3.9432],
        [ -9.4624,  -0.4633,  -7.3918],
        [  2.3889,  -4.6082,  -8.7291],
        [  8.2154,  -8.4262,   8.0171],
        [  4.0642,  -3.7599,  -3.4565],
        [  3.2990,  -4.2415,  -7.4857],
        [  6.1575,  12.2046,   3.3257],
        [  5.3860,  -2.3418,   1.0826],
        [  6.1755,   2.5233,  -2.0042],
        [  6.2026,  12.1774,  -1.5887],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.1372,   2.6726,  -6.7885],
        [  4.3377,   5.7570,   1.3712],
        [  6.2790,  -0.0259,  -5.4116],
        [  6.1685,  12.1979,  -6.2473],
        [ -8.1426,   3.8048,   9.3542],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.9788,   2.0208,  -3.2867],
        [  2.6310, -10.9195,   3.8054],
        [  6.1730,  -0.5375,  -8.8861],
        [ -5.9987,   6.1949,  -6.2114],
        [  2.5375,   7.0676,  -4.3490],
        [  3.9515,  -3.8425,   6.1729],
        [  1.7228,  -4.7575,   8.3396],
        [  5.8275,  -1.5334,   2.0124],
        [  6.3569,   0.7591,   6.5011],
        [  2.1625,  -4.6697,   7.0658],
        [  4.2612, -10.5514,  -6.6268],
        [ 12.5927,   0.4199,  -9.4997],
        [ -7.3333,   4.9246,  -7.9212],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.1579,  -0.5961,  -6.5595],
        [ 10.9875,   7.2207,  -8.8549],
        [  5.0277,   4.9589,   7.0175],
        [ -1.8823, -10.6558,  -3.8952],
        [  2.5128, -10.9365,   9.9645],
        [ -0.5739,   7.9056,   0.6382],
        [  3.9770, -10.6341,   8.0783],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.3076,   0.7852,  -6.7010],
        [  5.6095,  -1.9702,  -7.8233],
        [  3.6703,   6.3443,  -5.4401],
        [  4.4039,   5.6903,   8.7366],
        [  6.3549,   0.7096,   6.9198],
        [ -6.0175,  -8.4541,  -9.8408],
        [  4.2339,  -3.6268,   6.3125]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0485, 0.0572,  ..., 1.0585, 1.2098, 1.0360],
        [0.0485, 0.0000, 0.0380,  ..., 1.0257, 1.1689, 1.0019],
        [0.0572, 0.0380, 0.0000,  ..., 1.0015, 1.1943, 0.9789],
        ...,
        [1.0585, 1.0257, 1.0015,  ..., 0.0000, 1.4372, 0.0462],
        [1.2098, 1.1689, 1.1943,  ..., 1.4372, 0.0000, 1.3912],
        [1.0360, 1.0019, 0.9789,  ..., 0.0462, 1.3912, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.59673406, 13.84126197, 13.6829843 , 13.94008948, 13.86480175,
       13.6922425 ,  5.8828986 , 12.1494023 , 12.59969936, 13.12299145,
        8.62162372, 13.39372426, 11.51811603,  5.87305005, 13.45642379,
       11.72802491,  9.37296102, 11.00726483, 12.5769659 ,  9.09661212,
       11.32149259, 10.26327895, 13.97857591, 13.3426541 , 11.42947632,
        7.27340834,  9.97822473, 12.32545637,  5.05534609, 12.45418236,
        6.652742  ,  5.90035324, 13.97779839, 11.63502679,  9.50654967,
        8.92098752,  6.14086582,  6.83515201, 10.26545792,  9.09502385,
        8.72896381, 13.12299145,  9.86089579,  9.96391045, 13.94008948,
       12.10222204,  5.14999799,  8.06990483,  7.50917551,  8.12043983,
        9.75401449, 11.35182589, 11.33837396, 10.74409834,  5.62005629,
        6.2322769 ,  5.68942572,  9.47369628,  8.25828409, 12.12641969,
        7.15990676,  9.47369628,  5.19065439, 11.76830981,  5.5366698 ,
        5.37343173, 13.66991027,  5.87305005,  6.67108662, 13.6660671 ,
        9.14973319,  6.69389431,  7.20826219,  6.27905292, 13.66896739,
        8.98765644, 11.76830981,  9.20340031, 11.23201619,  6.19633139,
        8.62329466,  7.5092834 ,  5.51177501,  5.05981914,  6.02588517,
        6.4020404 ,  5.14607014, 11.37939695, 12.59969936,  8.83341608,
        5.19577146,  6.18668463, 13.14777433,  7.06169391, 10.8207297 ,
       11.22142195,  7.92644746, 11.35345345,  9.91668182,  9.34061938,
        5.94541803,  7.32950385,  7.19534146,  6.39438044, 10.37698435,
        5.5748624 ])}
{'P': tensor([[3.9924e-03, 2.4540e-01, 1.8807e-01,  ..., 2.2441e-06, 3.7712e-07,
         4.9574e-07],
        [6.2830e-06, 3.3249e-03, 4.7797e-02,  ..., 3.1044e-06, 2.4818e-07,
         4.7795e-07],
        [3.7315e-06, 4.3279e-05, 5.4454e-03,  ..., 1.0055e-05, 5.5256e-07,
         6.0850e-07],
        ...,
        [4.0813e-07, 3.4286e-07, 3.7946e-07,  ..., 2.4936e-02, 7.8145e-04,
         1.2397e-04],
        [9.4566e-05, 4.8433e-05, 5.2954e-05,  ..., 9.9851e-05, 9.2170e-02,
         9.8778e-05],
        [9.2468e-05, 7.2345e-05, 3.9001e-05,  ..., 4.2837e-05, 1.5072e-04,
         1.0944e-01]]), 'X': tensor([[  6.9284,  11.7061,  -0.3732],
        [  4.9152,  12.8653,  -1.4459],
        [  3.3633,  13.4794,   0.5193],
        [  4.1637,  13.1898,  -1.9189],
        [  2.3373,  13.7721,  -0.2442],
        [  2.2076,  13.8032,  -1.5099],
        [  5.7845,   3.6652,   0.9175],
        [  6.3551,   0.7149,   1.3344],
        [  9.6587,   9.1817,  -0.8212],
        [ -9.4773,  -1.0024,   5.2599],
        [  0.9445,   7.6736,   0.5095],
        [  6.3542,   0.6934,  -3.3840],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.9942,  -3.9990,   6.0448],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.2373,  -4.6506,  -1.7334],
        [ -9.1795,  -3.3623,   8.5259],
        [  3.9591, -10.6391,  -6.2557],
        [ -7.9925,  -6.1423,   2.8990],
        [  1.8766,   7.3694,   1.1887],
        [  2.6310, -10.9195,   3.8054],
        [ 12.5770,   1.8512,  -0.6578],
        [ -4.1296,   7.2803,  -2.7194],
        [  5.4167, -10.1293,   6.2156],
        [ -5.6169,  -8.7844,   8.9057],
        [ -1.6220,   7.8917,   0.0315],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -3.3116, -10.1539,   7.5919],
        [ -8.5149,  -5.1905,   1.5239],
        [ -0.8455, -10.8853,  -6.7400],
        [  5.4426, -10.1181,  -5.1268],
        [  8.2154,  -8.4262,   8.0171],
        [  4.7482, -10.3907,   5.9149],
        [  5.3860,  -2.3418,   1.0826],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -3.2736, -10.1703,   3.8712],
        [  6.1752,   2.5242,   3.5874],
        [  3.7992,  -3.9474,  -8.9019],
        [ -6.2668,   5.9808,   2.5614],
        [  0.6142,  14.0802,   0.5407],
        [  3.0876,   6.7527,  -5.3540],
        [ -9.2974,   0.8371,   2.4852],
        [  6.3589,   0.8239,  -8.4230],
        [ -0.9715, -10.8630,   0.3555],
        [ -9.4196,   0.0478,   9.4140],
        [  9.6816,   9.1536,  -5.8543],
        [  2.9260,  -4.4162,   8.3808],
        [  5.6672,  -1.8631,   8.1020],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -5.9940,   6.1985,   5.9566],
        [  5.6202,  -1.9508,   0.7471],
        [  3.3744,   6.5623,  -6.3636],
        [  6.3589,   0.8239,  -8.4230],
        [ -2.0229, -10.6163,   8.6930],
        [ 11.2571,   6.7072,   8.3693],
        [  2.5974,   7.0363,  -7.7810],
        [ -5.1974,   6.7396,   3.3484],
        [  6.3505,   0.6235,  -8.5849],
        [  6.3124,   0.2057,  -0.4593],
        [ -8.5506,  -5.1154,   6.3423],
        [  6.3515,   0.6418,  -6.5823],
        [  3.9663,   6.1018,  -5.4104],
        [ -9.4624,  -0.4633,  -7.3918],
        [  3.4920,  -4.1366,  -8.3955],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.5375,   7.0676,  -4.3490],
        [ -7.9667,   4.0814,  -0.6397],
        [  2.5382,   7.0672,   4.5097],
        [ -1.1885, -10.8211,   0.0865],
        [  6.1929,   2.4501,  -6.9352],
        [  8.1528,  10.7452,   9.1493],
        [  4.5954,  13.0099,  -4.4439],
        [  9.3836,   9.5075,  -5.9278],
        [ -4.7389,   6.9953,   3.7638],
        [ -2.4956, -10.4682,   3.5632],
        [ -1.5875,   7.8944,  -9.4286],
        [  0.6933,   7.7336,   2.1179],
        [  6.2804,   2.0019,   0.8607],
        [  5.7784,  -1.6401,   9.5587],
        [ -1.6334, -10.7207,   7.1397],
        [  5.3804,  -2.3504,   7.2744],
        [  5.9587,  12.3216,   9.1695],
        [  2.5382,   7.0672,   4.5097],
        [  2.1379,  -4.6757,   6.4981],
        [  5.4242, -10.1261,   3.9595],
        [  6.2026,  12.1774,  -1.5887],
        [  2.0449,  13.8402,  -7.1642],
        [  3.9913,  -3.8139,  -6.8311],
        [  2.3261,   7.1726,   6.2522],
        [  5.8836,   3.4315,  -4.0327],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 12.5919,   0.4035,  -5.8460],
        [ 11.5966,   5.9666,   6.0083],
        [  3.5939,   6.4029,  -1.8357],
        [  2.9410,   6.8428,   5.4458],
        [ -8.0829,  -5.9940,   5.0564],
        [ -1.7009,   7.8851,  -8.2437],
        [  1.5246,  -4.7838,  -5.2095],
        [ -9.2301,  -3.1537,  -2.7477],
        [  4.3377,   5.7570,   1.3712],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.2182,  13.8007,   5.1889],
        [ -9.3616,  -2.4750,   3.7765],
        [  1.8447,  -4.7372,   8.1235],
        [  0.8700,  -4.8132,  -7.8042]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0637, 0.1379,  ..., 1.2178, 0.6430, 0.4645],
        [0.0637, 0.0000, 0.1971,  ..., 1.2261, 0.6936, 0.4147],
        [0.1379, 0.1971, 0.0000,  ..., 1.1502, 0.5094, 0.5426],
        ...,
        [1.2178, 1.2261, 1.1502,  ..., 0.0000, 0.8885, 1.0644],
        [0.6430, 0.6936, 0.5094,  ..., 0.8885, 0.0000, 0.8853],
        [0.4645, 0.4147, 0.5426,  ..., 1.0644, 0.8853, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.60275211, 13.77222752, 13.89264822, 13.83139024, 13.96905756,
       13.97857591,  6.84796202,  6.39520397, 13.32645552,  9.53015376,
        7.73151309,  6.39187285, 13.06957425,  9.84314712,  9.96391045,
        9.14973319,  5.16077764,  9.77588801, 11.35182589, 10.08001894,
        7.60463349, 11.23201619, 12.7125079 ,  8.36995953, 11.48663767,
       10.42667093,  8.05668609, 10.61047596, 10.68031351,  9.97220799,
       10.91805683, 11.4890861 , 11.76830981, 11.42420508,  5.87305005,
       10.61047596, 10.68416401,  6.67121748,  5.47866221,  8.66273171,
       14.09357547,  7.42512621,  9.3349858 ,  6.41202986, 10.90638066,
        9.41970031, 13.32374056,  5.29755487,  5.96556248, 12.10942412,
        8.62260327,  5.94909709,  7.37902766,  6.41202986, 10.80728788,
       13.103709  ,  7.50035355,  8.51088913,  6.38105015,  6.31576264,
        9.96391045,  6.383897  ,  7.27759233,  9.47369628,  5.41348121,
       13.27994304,  7.5092834 ,  8.95133961,  7.50917551, 10.88618114,
        6.65991877, 13.48809595, 13.79762415, 13.35832725,  8.44940661,
       10.76154256,  8.05242591,  7.76457095,  6.59176018,  6.00663314,
       10.84437705,  5.87132718, 13.68672855,  7.50917551,  5.14124612,
       11.48734978, 13.6660671 , 13.99048094,  5.5205189 ,  7.5403822 ,
        6.81117802, 10.1979669 , 12.59840193, 13.04156075,  7.34250948,
        7.44805901, 10.06286074,  8.06643487,  5.0209005 ,  9.75401449,
        7.20826219,  9.75075273, 13.97779839,  9.68324553,  5.08372149,
        4.89121805])}
{'P': tensor([[1.6848e-03, 2.6474e-01, 2.3826e-01,  ..., 2.1828e-07, 1.7957e-05,
         5.0776e-07],
        [6.8766e-06, 4.0869e-03, 1.6133e-05,  ..., 2.8019e-07, 7.0765e-05,
         6.7063e-07],
        [7.8715e-06, 2.8647e-01, 2.3200e-03,  ..., 2.1162e-07, 3.2585e-05,
         5.1915e-07],
        ...,
        [2.6626e-03, 1.8711e-03, 1.9283e-03,  ..., 4.9808e-03, 1.5162e-06,
         4.1442e-03],
        [3.4991e-06, 3.9330e-06, 3.7410e-06,  ..., 1.8926e-06, 1.0759e-01,
         2.9454e-06],
        [1.7866e-01, 9.3921e-02, 1.1908e-01,  ..., 9.8673e-05, 1.9702e-05,
         1.4219e-02]]), 'X': tensor([[  6.8407,  11.7666,   3.0045],
        [  5.4867,  12.5816,   2.9549],
        [  6.1575,  12.2046,   3.3257],
        [  5.6928,  12.4710,   3.6001],
        [  5.9003,  12.3550,   2.1917],
        [  2.9470,  13.6085,   2.9809],
        [ -5.6651,   6.4380,   1.3660],
        [  2.3367,   7.1676,  -9.5263],
        [  9.4554,   9.4245,   9.8800],
        [ -9.1623,  -3.4288,  -4.9961],
        [ 11.5198,  -4.0923,   5.5088],
        [ -8.6112,   2.9319,  -7.9943],
        [ -5.1377,   6.7751,  -0.4438],
        [ -7.4602,   4.7718,  -4.6277],
        [ 10.7566,   7.6212,  -2.3279],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 10.7566,   7.6212,  -2.3279],
        [  0.3715,  -4.7756,   0.2312],
        [ -8.9636,   2.0648,   8.1946],
        [  9.3424,  -7.3685,  -3.6807],
        [ -1.8823, -10.6558,  -3.8952],
        [ 11.2571,   6.7072,   8.3693],
        [  2.2076,  13.8032,  -1.5099],
        [ -1.2454,   7.9127,  -0.5715],
        [ -9.4756,  -0.8166,  -9.6021],
        [  6.1737,   2.5305,   3.9049],
        [  3.3167,   6.6022,   6.3554],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.9717,  -5.1646,  -6.9039],
        [  3.9913,  -3.8139,  -6.8311],
        [ -9.3087,   0.7792,  -7.5587],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.4735,  -0.7274,  -2.6749],
        [  9.0424,  -7.6779,  -7.5286],
        [  9.3607,   9.5337,  -4.5412],
        [  4.1999,  -3.6543,  -4.1998],
        [  6.1685,  12.1979,  -6.2473],
        [  5.1472,  -2.6801,   8.1194],
        [  1.7228,  -4.7575,   8.3396],
        [  6.3587,   1.1596,  -3.6629],
        [ 12.5650,   2.0132,   4.6272],
        [ 11.1469,   6.9237,   1.6160],
        [ 12.4690,  -0.8528,  -9.3374],
        [  1.5662,  13.9375,   4.5142],
        [  3.9515,  -3.8425,   6.1729],
        [  2.0449,  13.8402,  -7.1642],
        [  5.8836,   3.4315,  -4.0327],
        [  2.9470,  13.6085,   2.9809],
        [ -7.2515,   5.0195,   4.6434],
        [ -9.3087,   0.7792,  -7.5587],
        [  3.0719, -10.8449,  -4.1673],
        [ -8.9178,   2.1936,   7.5388],
        [  4.0362,  13.2397,  -7.0030],
        [  6.3569,   0.7591,   6.5011],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.6000,   1.3839,  -3.2992],
        [  8.1217,  -8.5024,  -0.6149],
        [  1.1139,   7.6281,  -2.5680],
        [  3.5939,   6.4029,  -1.8357],
        [  2.0937,  13.8293,  -2.3533],
        [  1.9503,  -4.7172,  -6.2313],
        [ 11.7779,   5.5127,  -2.0699],
        [  3.9663,   6.1018,  -5.4104],
        [  1.6408,   7.4591,  -4.2294],
        [ -3.5520,   7.4976,   9.1522],
        [  5.1196,  -2.7160,   4.2705],
        [  0.3493,  -4.7727,  -6.5682],
        [ -9.2974,   0.8371,   2.4852],
        [ -9.3107,  -2.7697,  -3.2919],
        [  5.8930,  12.3592,  -8.9201],
        [  6.2365,  -0.2578,  -6.0212],
        [ -7.0825,   5.2073,  -5.0497],
        [ -1.8823, -10.6558,  -3.8952],
        [ 11.5935,   5.9740,  -4.3952],
        [ 11.2209,  -4.7092,   2.6117],
        [ -9.1459,   1.4780,  -4.2403],
        [  9.6816,   9.1536,  -5.8543],
        [  4.8432,   5.1968,   1.5006],
        [  5.1890,   4.7318,   0.9515],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -0.0410,   7.8591,   9.3457],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.4697,  -4.5836,   1.4593],
        [  5.9883,   3.1536,   6.2320],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.0345,   3.0175,  -1.4424],
        [  5.9526,   3.2525,  -4.3821],
        [  0.5600, -11.0321,  -0.8660],
        [ 12.1940,  -2.1881,   7.1499],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.6319,  11.9066,   3.5165],
        [  9.5261,  -7.1675,   1.4206],
        [ -8.8540,  -4.3971,  -5.2144],
        [  2.5469,  13.7192,  -9.6935],
        [ -7.4887,   4.7364,   8.9074],
        [ -8.6801,   2.7809,  -0.8443],
        [  9.5203,   9.3483,   5.8781],
        [ -8.9606,  -4.0994,   1.7886],
        [  1.7659,   7.4127,   4.7871],
        [  3.7298,  13.3537,  -9.4653],
        [  7.0171,  11.6439,  -5.0178],
        [ -4.6942,  -9.4299,  -9.1846],
        [ 12.1379,   4.4125,  -3.8833],
        [ -8.4573,   3.2448,   5.4370],
        [ 11.1469,   6.9237,   1.6160]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0099, 0.0430,  ..., 0.7441, 1.0346, 0.2396],
        [0.0099, 0.0000, 0.0482,  ..., 0.7403, 1.0250, 0.2343],
        [0.0430, 0.0482, 0.0000,  ..., 0.7869, 1.0447, 0.2823],
        ...,
        [0.7441, 0.7403, 0.7869,  ..., 0.0000, 1.1827, 0.5101],
        [1.0346, 1.0250, 1.0447,  ..., 1.1827, 0.0000, 0.9844],
        [0.2396, 0.2343, 0.2823,  ..., 0.5101, 0.9844, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.6105595 , 13.72594022, 13.66991027, 13.70893492, 13.69162795,
       13.92390689,  8.57560275,  7.53884104, 13.35013374,  9.78286733,
       12.22503754,  9.09661212,  8.50278896,  8.85572292, 13.18278606,
       10.1979669 , 13.18278606,  4.79003361,  9.19837792, 11.89855267,
       10.8207297 , 13.103709  , 13.97857591,  8.01009083,  9.51074763,
        6.67218687,  7.38844703, 10.38897563, 12.12641969,  5.5205189 ,
        9.34127013,  9.58347456,  6.83515201,  9.5014146 , 11.86239961,
       13.36092624,  5.5671389 , 13.66896739,  5.80312949,  5.05981914,
        6.46356806, 12.72524615, 13.12218013, 12.4980845 , 14.02526505,
        5.51177501, 13.99048094,  6.81117802, 13.92390689,  8.81930525,
        9.34127013, 11.27160035,  9.18358562, 13.84126197,  6.4020404 ,
        8.53501777, 12.67576792, 11.75807252,  7.70897675,  7.34250948,
       13.98691426,  5.10444097, 13.00413829,  7.27759233,  7.6374603 ,
        8.29640613,  5.79543038,  4.78544875,  9.3349858 ,  9.71392179,
       13.6922425 ,  6.24187044,  8.7907923 , 10.8207297 , 13.04217487,
       12.16901615,  9.26455795, 13.32374056,  7.10378919,  7.02253235,
        9.78286733,  7.85919687,  4.78258092,  5.20659892,  6.76791483,
       12.00995898,  6.74686071,  6.78325151, 11.04629471, 12.38881635,
       12.59969936, 13.62896593, 11.92133129,  9.88571713, 13.95361098,
        8.86082124,  9.11473391, 13.3426541 ,  9.85384623,  7.62011734,
       13.86480175, 13.59480906, 10.53368644, 12.91505111,  9.05843201,
       13.12218013])}
{'P': tensor([[3.1625e-03, 2.9182e-01, 1.4453e-01,  ..., 3.2669e-06, 1.2076e-06,
         2.7653e-04],
        [1.9531e-05, 4.8639e-03, 1.7244e-01,  ..., 3.5159e-06, 1.3252e-06,
         6.0633e-04],
        [4.4300e-06, 1.0468e-05, 5.6681e-03,  ..., 2.2785e-06, 1.0813e-06,
         4.9372e-04],
        ...,
        [9.2930e-03, 7.1204e-03, 7.0993e-03,  ..., 9.1110e-02, 8.0862e-03,
         4.4838e-04],
        [2.8273e-03, 2.0574e-03, 2.2534e-03,  ..., 6.3885e-02, 1.0574e-01,
         1.1281e-04],
        [1.8647e-05, 2.5088e-05, 4.4726e-05,  ..., 2.4383e-05, 9.6151e-06,
         8.6535e-02]]), 'X': tensor([[ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 2.3373e+00,  1.3772e+01, -2.4417e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [-6.3306e+00,  5.9272e+00,  5.6935e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 1.0474e+01, -5.9570e+00, -1.2108e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.2150e+01,  4.3674e+00, -7.2998e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [-9.4541e+00, -3.3275e-01,  8.8246e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 3.5965e+00, -4.0754e+00, -3.5035e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 2.5813e+00, -4.5471e+00,  5.3597e+00],
        [-2.2188e+00,  7.8218e+00,  8.6064e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 7.8375e+00, -8.7240e+00, -9.8913e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 2.7420e+00, -4.4896e+00, -3.7139e+00],
        [-6.7643e+00, -7.7372e+00, -9.6663e+00],
        [ 3.3633e+00,  1.3479e+01,  5.1930e-01],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.2484e+01,  2.7510e+00,  4.4493e-01],
        [ 7.5784e+00, -8.9141e+00,  5.8410e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [ 9.5203e+00,  9.3483e+00,  5.8781e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0820, 0.1307,  ..., 0.4931, 0.6088, 0.7286],
        [0.0820, 0.0000, 0.2122,  ..., 0.5701, 0.6887, 0.7046],
        [0.1307, 0.2122, 0.0000,  ..., 0.3675, 0.4790, 0.7665],
        ...,
        [0.4931, 0.5701, 0.3675,  ..., 0.0000, 0.1382, 0.8668],
        [0.6088, 0.6887, 0.4790,  ..., 0.1382, 0.0000, 0.9980],
        [0.7286, 0.7046, 0.7665,  ..., 0.8668, 0.9980, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.61600536, 13.69162795, 13.89264822, 13.92390689, 13.72594022,
       13.96905756,  5.80860165, 13.77604293, 11.00508681, 13.97430191,
       10.42667093, 14.10796466,  8.8580671 , 10.06437598, 11.46176259,
       13.36800073,  8.41148722, 12.00995898,  7.01475376, 10.84437705,
       10.06407839, 12.33167382,  8.05242591, 11.97238624,  8.67227386,
        6.12886024,  7.4544866 , 12.87752052, 12.79126785, 10.99326577,
        5.18277858, 10.8207297 , 10.46739073, 12.04924159,  6.4020404 ,
       12.91143491, 10.8992524 ,  6.61386436,  5.52416368,  8.55982105,
        6.65368551,  6.7593917 ,  9.45995955, 13.98691426, 13.35832725,
        5.43544626,  9.13507321, 12.46513646, 13.27717982, 13.70893492,
        5.18047638,  5.2287058 ,  8.1303945 ,  4.76969364, 14.03505624,
       11.72749923, 12.58006038, 10.37698435, 11.04313641, 12.64235359,
        7.02496223,  8.36995953,  6.31576264,  5.26067019, 10.27716434,
       13.89264822, 12.4217523 ,  8.81930525, 12.78327037, 11.70016217,
       13.95472331,  4.98461983, 10.19036081,  6.48486171,  4.89121805,
        9.50654967, 11.75438006, 11.95743558,  9.0368458 ,  9.77777346,
       13.3426541 ,  8.56889738,  5.37343173, 12.10222204,  6.53250937,
       10.05558893,  9.71392179,  8.92395381,  9.51074763,  6.02779201,
        8.62162372, 13.6829843 , 10.6910768 ,  7.60463349,  8.68832398,
       11.83570132,  9.7712321 ,  6.83083863, 13.68063098,  5.70405525,
       11.35345345, 12.41366754,  5.94909709,  6.61211224,  6.42439344,
        8.29022197])}
{'P': tensor([[2.4853e-03, 2.3082e-01, 1.4758e-01,  ..., 5.7192e-09, 7.1478e-07,
         2.7445e-08],
        [2.0322e-06, 1.5789e-03, 2.1071e-01,  ..., 3.3776e-09, 2.9788e-07,
         1.0075e-08],
        [5.7624e-05, 1.9066e-04, 8.9127e-02,  ..., 1.6727e-07, 1.2535e-05,
         3.8999e-07],
        ...,
        [4.2851e-06, 2.2928e-06, 1.3295e-06,  ..., 1.4908e-02, 9.9158e-04,
         2.4157e-02],
        [3.6358e-03, 2.5921e-03, 1.7226e-03,  ..., 1.3618e-04, 1.2540e-01,
         3.0697e-03],
        [4.9100e-04, 2.9337e-04, 1.6147e-04,  ..., 7.0471e-06, 1.5367e-03,
         4.0853e-02]]), 'X': tensor([[  6.7565,  11.8237,  -4.9882],
        [  4.5954,  13.0099,  -4.4439],
        [  3.2692,  13.5098,  -4.9032],
        [  3.2809,  13.5061,  -4.0199],
        [  4.9337,  12.8566,  -3.6582],
        [  3.3094,  13.4969,  -3.6507],
        [  1.9204,   7.3518,   8.4781],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.1519, -10.2387,   9.8191],
        [  6.9956,  11.6590,  -7.6876],
        [  6.1984,  -0.4327,   1.5438],
        [  0.3493,  -4.7727,  -6.5682],
        [ -7.3470,  -7.0615,   5.0066],
        [ -8.5506,  -5.1154,   6.3423],
        [  2.1379,  -4.6757,   6.4981],
        [ -3.0008, -10.2827,  -8.9480],
        [  3.3069,   6.6089,   8.7377],
        [  3.2809,  13.5061,  -4.0199],
        [  6.2975,   0.0947,   5.1791],
        [  4.3459,  -3.5327,   0.3667],
        [  1.0391,  -4.8141,  -0.0768],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.8890,  -4.4316,   1.1600],
        [  5.7485,  -9.9811,  -3.2355],
        [  5.0506,   4.9277,  -9.5094],
        [  6.3549,   0.7096,   6.9198],
        [  6.8407,  11.7666,   3.0045],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.0870,  -4.6877,   9.6823],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.8617,   3.4853,  -9.8080],
        [  2.7558,  -4.4844,   0.0636],
        [  5.5294,  -2.1106,   9.0626],
        [  9.4925,   9.3811,   8.9842],
        [ 11.2571,   6.7072,   8.3693],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.6672,  -1.8631,   8.1020],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.7792,  11.8084,   1.6251],
        [  0.5175,  14.0910,   4.6540],
        [ -2.1817,   7.8274,  -4.6298],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.8653,   5.1694,  -4.0823],
        [ 11.0891,  -4.9558,  -2.3178],
        [  2.7685, -10.8982,  -6.8928],
        [ 10.3251,   8.2928,  -3.0006],
        [  0.1532,  -4.7422,  -7.0145],
        [  7.8375,  -8.7240,  -9.8913],
        [ -9.4524,  -1.6813,  -6.6862],
        [  2.0243,   7.3086,   7.5432],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.6817,   6.3354,   6.1233],
        [  6.0345,   3.0175,  -1.4424],
        [ -4.1296,   7.2803,  -2.7194],
        [  0.7685,  -4.8098,  -2.9310],
        [ -8.0534,  -6.0429,   2.4583],
        [ -3.0004,   7.6607,  -7.2640],
        [  5.6523,   3.9443,  -2.9449],
        [  3.2692,  13.5098,  -4.9032],
        [  0.1041, -11.0045,   2.4311],
        [ -9.3107,  -2.7697,  -3.2919],
        [  1.2273,  -4.8081,   9.8129],
        [  0.1677,  -4.7447,  -6.1529],
        [  6.0140,  -1.0661,  -2.5040],
        [  2.5382,   7.0672,   4.5097],
        [  6.9693,  -9.3200,  -8.3822],
        [  2.0870,  -4.6877,   9.6823],
        [  9.9756,  -6.6330,  -3.3429],
        [ -2.5161,   7.7699,  -9.3531],
        [  2.6970,   6.9826,   9.7350],
        [  1.3339,   7.5627,  -0.9034],
        [  0.3493,  -4.7727,  -6.5682],
        [  0.9445,   7.6736,   0.5095],
        [  4.7156,  -3.1831,  -8.0224],
        [  0.8572,  14.0501,  -8.3364],
        [  4.1348,   5.9514,  -0.8856],
        [  5.4336,  -2.2676,   5.6933],
        [  0.2732,  -4.7619,  -1.1577],
        [  2.4149,  -4.6005,  -4.3339],
        [ 10.0400,   8.6922,  -6.2847],
        [ 12.3453,  -1.5462,   7.0212],
        [ -4.0868,   7.2981,   8.1874],
        [  8.2154,  -8.4262,   8.0171],
        [  6.3119,   1.7797,   9.9186],
        [  9.3624,   9.5318,   0.2779],
        [ 10.0995,  -6.4739,   2.0974],
        [ -6.4620,  -8.0449,   6.2719],
        [  2.8602,   6.8905,   5.8779],
        [  1.0391,  -4.8141,  -0.0768],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -8.1204,  -5.9308,   4.0761],
        [  5.4179,  -2.2925,  -3.4641],
        [  2.6310, -10.9195,   3.8054],
        [ -6.6895,   5.6051,   2.7899],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.1066,  -0.7799,  -4.1746],
        [ -8.9033,   2.2329,  -8.8912],
        [  4.1348,   5.9514,  -0.8856],
        [  2.9410,   6.8428,   5.4458],
        [ -9.3076,   0.7852,  -6.7010],
        [ 12.2435,   4.0076,   8.1257],
        [ 11.2939,   6.6325,  -5.8311],
        [ 12.4597,  -0.9136,  -4.8763],
        [  6.8093,  -9.4178,   8.4244],
        [  6.1984,  -0.4327,   1.5438],
        [ 12.3453,  -1.5462,   7.0212]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0328, 0.0642,  ..., 1.2533, 0.5851, 1.1071],
        [0.0328, 0.0000, 0.0481,  ..., 1.2210, 0.5533, 1.0787],
        [0.0642, 0.0481, 0.0000,  ..., 1.2193, 0.5563, 1.0934],
        ...,
        [1.2533, 1.2210, 1.2193,  ..., 0.0000, 0.6704, 0.4198],
        [0.5851, 0.5533, 0.5563,  ..., 0.6704, 0.0000, 0.5656],
        [1.1071, 1.0787, 1.0934,  ..., 0.4198, 0.5656, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.61801526, 13.79762415, 13.89974693, 13.89886469, 13.77075174,
       13.89671314,  7.5984809 ,  5.1224079 , 11.46176259, 13.59673406,
        6.2134961 ,  4.78544875, 10.19036081,  9.96391045,  5.14124612,
       10.71162681,  7.39004226, 13.89886469,  6.29821673,  5.60064319,
        4.92497514, 12.17901033,  5.29009836, 11.51811603,  7.05627461,
        6.39438044, 13.6105595 ,  9.85384623,  5.13125513, 12.10942412,
        6.81961996,  5.26342203,  5.91854397, 13.34587027, 13.103709  ,
        9.15866607,  5.96556248,  4.78002665, 13.61600536, 14.10045797,
        8.12580976, 13.14777433,  7.09887557, 12.1460931 , 11.24434374,
       13.2430377 ,  4.74468406, 11.72749923,  9.60080639,  7.58378655,
       12.27993098,  7.32754882,  6.74686071,  8.36995953,  4.87083348,
       10.06850677,  8.22728399,  6.89245352, 13.89974693, 11.0050379 ,
        9.71392179,  4.96231532,  4.74770899,  6.10774544,  7.50917551,
       11.63765273,  5.13125513, 11.97957425,  8.16714836,  7.48538905,
        7.67939546,  4.78544875,  7.73151309,  5.68942572, 14.07623266,
        7.24677664,  5.88782913,  4.76969364,  5.19577146, 13.27994304,
       12.44177641,  8.36445758, 11.76830981,  6.55800267, 13.3607379 ,
       11.99633819, 10.31886066,  7.46053281,  4.92497514, 12.49291768,
       10.05558893,  5.8828986 , 11.23201619,  8.7273569 ,  9.62958973,
        6.1561646 ,  9.17905491,  7.24677664,  7.44805901,  9.34061938,
       12.88270018, 13.09737053, 12.49317563, 11.62158554,  6.2134961 ,
       12.44177641])}
{'P': tensor([[1.6094e-03, 3.0715e-01, 2.8334e-01,  ..., 7.7631e-07, 4.3201e-08,
         1.1513e-04],
        [1.0402e-05, 5.0045e-03, 2.3433e-05,  ..., 1.8975e-06, 9.9513e-08,
         2.0680e-04],
        [8.3894e-06, 3.1281e-01, 2.4303e-03,  ..., 1.0332e-06, 5.6319e-08,
         1.3623e-04],
        ...,
        [9.6071e-03, 7.5480e-03, 8.4799e-03,  ..., 1.0058e-01, 2.9756e-03,
         4.0145e-02],
        [1.8299e-02, 1.1539e-02, 1.5212e-02,  ..., 4.4125e-03, 9.8071e-02,
         1.4083e-02],
        [5.3082e-03, 8.3291e-03, 6.9720e-03,  ..., 4.5688e-04, 3.8532e-05,
         1.0822e-01]]), 'X': tensor([[ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 2.9470e+00,  1.3608e+01,  2.9809e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 9.0004e+00, -7.7195e+00,  4.6945e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 6.3205e+00,  1.7068e+00, -3.8619e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 1.1602e+01,  5.9540e+00, -5.1715e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-5.6792e+00, -8.7352e+00,  4.0318e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 1.6408e+00,  7.4591e+00, -4.2294e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 3.5284e+00, -1.0748e+01,  5.3347e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 4.8432e+00,  5.1968e+00,  1.5006e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 6.2804e+00,  2.0019e+00,  8.6070e-01],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 1.2361e+01,  3.4769e+00, -8.5762e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 6.3266e+00,  3.2921e-01,  9.3502e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [ 1.1617e+01, -3.8715e+00,  6.4594e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 8.1217e+00, -8.5024e+00, -6.1493e-01],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 3.7298e+00,  1.3354e+01, -9.4653e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 1.7000e+00, -4.7609e+00,  2.7935e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 3.7239e-01,  7.7974e+00, -6.8722e+00],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.6906e-02, 1.5135e-02,  ..., 3.5220e-01, 6.1334e-01,
         5.6355e-01],
        [1.6906e-02, 0.0000e+00, 3.1838e-02,  ..., 3.5741e-01, 6.2658e-01,
         5.7134e-01],
        [1.5135e-02, 3.1838e-02, 0.0000e+00,  ..., 3.4500e-01, 5.9978e-01,
         5.5403e-01],
        ...,
        [3.5220e-01, 3.5741e-01, 3.4500e-01,  ..., 1.5170e-04, 3.7948e-01,
         2.2531e-01],
        [6.1334e-01, 6.2658e-01, 5.9978e-01,  ..., 3.7948e-01, 0.0000e+00,
         2.5344e-01],
        [5.6355e-01, 5.7134e-01, 5.5403e-01,  ..., 2.2531e-01, 2.5344e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.62896593, 13.70893492, 13.66991027, 13.72594022, 13.92390689,
       13.88683436,  8.89458273,  7.50035355,  8.57560275, 11.85743425,
        9.68324553, 11.89105259,  7.32233733, 10.48840476,  6.54693987,
       12.3344536 ,  8.36445758, 13.04051744,  5.20659892,  9.09119351,
        5.70405525, 10.41909211,  5.84450127, 11.95743558,  8.27853076,
        4.96231532, 13.69162795,  8.83341608, 14.02532758,  7.6374603 ,
        9.04255655,  7.58378655,  8.30729352, 10.54658597,  7.70897675,
        7.92607342, 11.31276109, 12.05395056,  6.49538841, 11.80092578,
        7.58668144, 12.00995898,  7.10378919,  7.57685908,  7.62509582,
       11.83570132,  5.27178363,  7.57946459,  5.63824971,  6.4020404 ,
       13.77604293, 13.2778378 , 13.05148826,  6.59176018,  9.26455795,
       11.9702762 , 12.84056604, 11.89855267, 11.24434374,  6.33517349,
        6.02779201,  7.17789666,  8.09704052,  5.48276038,  7.50500064,
        6.65991877, 12.24467041, 10.1371552 , 11.75807252, 11.785566  ,
        7.5403822 , 12.07818871, 10.4606265 ,  7.72415048, 12.3344536 ,
       12.70489563, 10.63038565, 10.1371552 ,  7.5158489 , 11.9182956 ,
        7.23276582,  9.7751189 , 12.71843462, 13.86480175,  9.97964013,
        6.4020404 , 11.63724792,  7.32233733,  6.38105015, 10.48315436,
        8.87587457,  7.16747878,  5.84450127,  5.05534609,  6.02789634,
       10.92597196,  9.48337063, 11.20100795,  6.5618968 ,  5.76165934,
        7.80625981,  9.43737523, 13.14777433,  6.29034958, 12.52372629,
        7.37576154])}
{'P': tensor([[2.0069e-03, 4.9733e-01, 1.7290e-01,  ..., 9.1414e-05, 2.6180e-05,
         1.9772e-05],
        [9.7886e-06, 3.5069e-03, 3.4428e-01,  ..., 1.4459e-04, 5.1372e-05,
         2.9915e-05],
        [3.4367e-06, 4.2738e-06, 3.2756e-03,  ..., 5.2350e-05, 6.5843e-05,
         1.0322e-05],
        ...,
        [9.8471e-04, 1.1653e-03, 2.2095e-03,  ..., 1.3559e-01, 4.6582e-02,
         2.3939e-03],
        [3.9639e-06, 4.2563e-06, 5.9020e-06,  ..., 1.2321e-04, 9.2796e-02,
         8.3001e-05],
        [2.0883e-03, 2.3623e-03, 2.9896e-03,  ..., 5.9485e-01, 9.1360e-03,
         7.8764e-02]]), 'X': tensor([[  6.2675,  12.1379,  -8.9720],
        [  5.8930,  12.3592,  -8.9201],
        [  3.7298,  13.3537,  -9.4653],
        [  2.7295,  13.6703,  -8.9328],
        [  2.5469,  13.7192,  -9.6935],
        [  2.0449,  13.8402,  -7.1642],
        [  5.7845,   3.6652,   0.9175],
        [  2.7121,   6.9743,  -8.4202],
        [ -9.3616,  -2.4749,   0.1793],
        [  6.3048,   0.1469,  -9.1691],
        [ -6.0175,  -8.4541,  -9.8408],
        [  4.2041,  -3.6509,  -4.9488],
        [  7.5784,  -8.9141,   5.8410],
        [  3.1480,  -4.3165,  -7.8002],
        [  4.0362,  13.2397,  -7.0030],
        [  0.1047, -11.0046,   0.5188],
        [ -9.2707,   0.9661,   4.4863],
        [ 11.0275,  -5.0666,   8.4154],
        [ -0.5709,   7.9055,   7.6089],
        [  9.0004,  -7.7195,   4.6945],
        [ -1.2454,   7.9127,  -0.5715],
        [  0.3354,  -4.7708,  -4.9125],
        [ 11.7795,  -3.4697,   3.5624],
        [  3.2366,  -4.2732,   5.1532],
        [  5.9193,  -1.3176,   5.1739],
        [  5.7218,  -9.9934,   6.5764],
        [ -5.3382,   6.6533,   7.8054],
        [  5.0617,  12.7957,   8.3544],
        [ 10.6283,  -5.7253,   7.4878],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.3076,   0.7852,  -6.7010],
        [  6.8407,  11.7666,   3.0045],
        [  8.2154,  -8.4262,   8.0171],
        [  7.9816,  10.8938,  -0.5378],
        [ -1.1885, -10.8211,   0.0865],
        [  1.7500,   7.4187,   4.4237],
        [  5.1890,   4.7318,   0.9515],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.7151,  -1.7700,   3.1542],
        [ -5.7347,  -8.6908,   9.2878],
        [  3.8111, -10.6787,  -4.6539],
        [ -4.0868,   7.2981,   8.1874],
        [ -6.2668,   5.9808,   2.5614],
        [  6.3031,   1.8477,  -7.1706],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.1433,  -3.5003,  -3.0209],
        [  1.7228,  -4.7575,   8.3396],
        [ 12.6035,   0.7494,   7.1536],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.6260,  -1.9401,   3.0259],
        [ -4.4520,   7.1370,  -8.7867],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.2247,  -0.3150,  -9.3657],
        [  1.7000,  -4.7609,   2.7935],
        [  6.3569,   0.7591,   6.5011],
        [  4.2339,  -3.6268,   6.3125],
        [  4.9957,  -2.8705,   6.5978],
        [ 10.9717,  -5.1646,  -6.9039],
        [  0.6142,  14.0802,   0.5407],
        [  4.0078,  -3.8018,  -6.2168],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -8.3653,   3.4184,   9.3454],
        [  5.9843,  -9.8679,  -6.1449],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.3753,   9.5171,   3.9131],
        [  9.6816,   9.1536,  -5.8543],
        [  2.7976,  -4.4682,  -4.2020],
        [  0.1460,   7.8340,  -9.4614],
        [  2.3490,  -4.6199,  -4.2489],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.2905,   1.9366,   2.1169],
        [  7.0664,  -9.2590,   8.0879],
        [  7.8982,  -8.6778,   0.0239],
        [  5.4426, -10.1181,  -5.1268],
        [ -7.0253,  -7.4493,  -1.5978],
        [  2.5318,  13.7231,   6.3692],
        [ 12.0582,  -2.6640,   0.7132],
        [ -6.1899,  -8.3011,  -7.9269],
        [  5.4336,  -2.2676,   5.6933],
        [ -8.0766,  -6.0046,  -3.1873],
        [  5.9026,   3.3837,  -0.6573],
        [  7.3592,  11.3937,   6.2407],
        [  7.9403,  -8.6455,  -8.7433],
        [ -8.0611,  -6.0302,   4.6962],
        [ -1.8788, -10.6567,   3.7068],
        [ 12.6059,   0.9597,  -6.3264],
        [ -9.4630,  -1.5145,   0.2983],
        [ 11.2209,  -4.7092,   2.6117],
        [ -7.6525,  -6.6534,   5.1721],
        [  4.2041,  -3.6509,  -4.9488],
        [  4.3523, -10.5232,   8.2138],
        [  4.7739,  -3.1218,  -4.1188],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.4267,  -5.3697,   2.4172],
        [ -5.1046,  -9.1610,   2.3418],
        [  0.1460,   7.8340,  -9.4614],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -8.5506,  -5.1154,   6.3423],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.8058,   2.4850,  -1.7425],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.0547,   7.2957,  -1.9343],
        [ -6.6998,   5.5954,   0.0823],
        [  4.0591,   6.0202,  -1.6339]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0137, 0.0482,  ..., 0.6456, 1.0461, 0.6130],
        [0.0137, 0.0000, 0.0346,  ..., 0.6367, 1.0349, 0.6054],
        [0.0482, 0.0346, 0.0000,  ..., 0.6197, 1.0106, 0.5919],
        ...,
        [0.6456, 0.6367, 0.6197,  ..., 0.0000, 0.4480, 0.0768],
        [1.0461, 1.0349, 1.0106,  ..., 0.4480, 0.0000, 0.5175],
        [0.6130, 0.6054, 0.5919,  ..., 0.0768, 0.5175, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.66051884, 13.6922425 , 13.86480175, 13.94008948, 13.95361098,
       13.99048094,  6.84796202,  7.48311002,  9.68323686,  6.30647746,
       10.37698435,  5.56809335, 11.70016217,  5.34250225, 13.84126197,
       11.00508681,  9.32094055, 12.13568598,  7.92607342, 11.85743425,
        8.01009083,  4.78258092, 12.27991991,  5.36061974,  6.06416957,
       11.51557166,  8.53014915, 13.76047999, 12.07225469,  6.33075682,
        9.34061938, 13.6105595 , 11.76830981, 13.50484539, 10.88618114,
        7.62232307,  7.02253235, 11.73848065,  5.98285319, 10.412299  ,
       11.33837396,  8.36445758,  8.66273171,  6.56833864, 12.35421461,
        9.79039371,  5.05981914, 12.62576026, 12.49317563,  5.95111804,
        8.41171369,  6.10774544,  6.23262289,  5.05534609,  6.4020404 ,
        5.5748624 ,  5.76165934, 12.12641969, 14.09357547,  5.52416368,
        9.50913849,  9.0368458 , 11.54071859,  9.66977802, 13.35927527,
       13.32374056,  5.27178363,  7.83534987,  5.18277858, 10.91805683,
        6.58184474, 11.64746882, 11.73397309, 11.4890861 , 10.23946247,
       13.95472331, 12.3489324 , 10.35485331,  5.88782913, 10.06407839,
        6.80370254, 13.56367857, 11.73848065, 10.06703358, 10.82106505,
       12.64235359,  9.58347456, 12.16901615, 10.14044489,  5.56809335,
       11.38774149,  5.70405525, 10.54658597,  9.99211663, 10.48721382,
        7.83534987, 12.4980845 ,  9.96391045,  9.64157461,  7.822854  ,
        9.14973319,  5.19065439,  9.93553424,  7.57946459,  8.72896381,
        7.26074925])}
{'P': tensor([[2.2161e-03, 3.6484e-01, 2.4897e-01,  ..., 2.2689e-05, 1.1608e-02,
         5.8087e-04],
        [5.1665e-06, 2.9214e-03, 2.9479e-01,  ..., 4.3656e-05, 1.6077e-03,
         1.0638e-03],
        [3.5977e-06, 1.2523e-05, 2.7994e-03,  ..., 4.2559e-05, 3.3077e-04,
         1.2509e-03],
        ...,
        [1.1454e-06, 1.4466e-06, 1.5476e-06,  ..., 3.1721e-02, 9.4573e-07,
         9.9294e-05],
        [5.8624e-05, 7.5673e-04, 5.1489e-03,  ..., 2.9726e-05, 5.2782e-03,
         1.9731e-03],
        [6.0653e-05, 8.4411e-05, 1.0401e-04,  ..., 7.2546e-03, 9.1469e-05,
         2.5102e-01]]), 'X': tensor([[ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 4.9152e+00,  1.2865e+01, -1.4459e+00],
        [ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [ 3.0719e+00, -1.0845e+01, -4.1673e+00],
        [-5.3845e-01,  7.9037e+00,  6.3830e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 4.0587e+00, -1.0611e+01,  6.2273e+00],
        [ 1.2077e+01, -2.6014e+00, -7.1832e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [-2.9497e+00, -1.0303e+01, -7.3479e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 5.8836e+00,  3.4315e+00, -4.0327e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 2.5620e+00,  7.0549e+00,  8.4825e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 1.5383e+00, -1.1027e+01, -4.9666e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 7.2340e+00, -9.1505e+00, -5.8321e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 3.9895e+00,  6.0816e+00, -6.5235e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [ 1.1735e+01, -3.5838e+00,  2.1644e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-1.7290e+00,  7.8825e+00, -3.3831e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 3.9913e+00, -3.8139e+00, -6.8311e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 2.1440e+00, -1.0981e+01,  8.5281e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0281, 0.0274,  ..., 1.0888, 0.2364, 0.6692],
        [0.0281, 0.0000, 0.0325,  ..., 1.0631, 0.2618, 0.6584],
        [0.0274, 0.0325, 0.0000,  ..., 1.0706, 0.2340, 0.6419],
        ...,
        [1.0888, 1.0631, 1.0706,  ..., 0.0000, 1.2523, 0.7469],
        [0.2364, 0.2618, 0.2340,  ..., 1.2523, 0.0000, 0.6979],
        [0.6692, 0.6584, 0.6419,  ..., 0.7469, 0.6979, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.6660671 , 13.77222752, 13.83139024, 13.97857591, 13.98691426,
       14.02058621,  5.1224079 , 11.27160035,  7.92200342, 10.06023665,
        5.21280156,  7.5403822 , 11.99633819,  9.15976746, 12.7125079 ,
       11.04629471, 11.48434767, 11.36089355, 12.35421461, 13.85158417,
        7.20826219, 12.22276278, 10.7167285 ,  9.60359158, 12.4217523 ,
        7.23276582, 13.25017211,  9.41970031, 12.30695098, 13.58101794,
       11.10066126,  6.19707944, 13.79676654,  7.15100303, 13.20988996,
        5.80860165,  8.40445872,  5.91677188, 13.06498554,  9.9353921 ,
       12.21532282, 10.87409005, 10.84437705,  6.84796202, 12.27835996,
        9.93553424,  5.05963217, 11.92133129,  9.96391045,  4.98989906,
       13.76047999,  5.30158497, 13.89974693,  5.21280156,  6.81084724,
        5.92281142,  6.81117802,  6.5618968 , 13.35927527,  5.08910491,
       13.04217487,  7.50564223, 10.67132524, 13.37121923,  6.27715105,
        8.27853076, 10.61472636, 11.1341798 ,  9.48337063,  8.36995953,
       11.66453059,  9.75200848,  6.61386436,  6.67218687, 11.63724792,
        7.27340834,  7.54127425, 10.05558893,  8.8737556 ,  7.37902766,
       12.32408516,  5.88782913,  5.89074104,  9.54154096, 11.65923396,
       13.56367857, 12.26997165, 10.88618114, 11.48434767,  5.1805345 ,
        8.06990483, 12.10942412, 10.90638066,  8.95983586,  5.5205189 ,
        5.89074104, 11.4424816 , 10.05558893,  9.79814997,  6.19633139,
       11.18839169,  4.76969364, 10.69963954,  8.77913428, 13.79762415,
        8.36995953])}
{'P': tensor([[2.7403e-03, 2.3021e-01, 1.2737e-01,  ..., 2.1156e-05, 9.7902e-09,
         2.1372e-08],
        [3.8317e-06, 2.6169e-03, 8.2946e-03,  ..., 3.4644e-05, 5.5976e-09,
         1.6115e-08],
        [5.2080e-06, 2.1009e-04, 4.4281e-03,  ..., 1.1005e-04, 1.1877e-08,
         2.4491e-08],
        ...,
        [1.2549e-06, 1.4523e-06, 1.8535e-06,  ..., 4.6827e-02, 4.8634e-06,
         1.0912e-05],
        [8.4213e-05, 3.1737e-05, 4.4995e-05,  ..., 2.8550e-06, 4.7764e-02,
         3.7506e-05],
        [1.3439e-04, 5.2789e-05, 3.5424e-05,  ..., 1.1936e-06, 1.3117e-04,
         2.3633e-02]]), 'X': tensor([[ 6.1685e+00,  1.2198e+01, -6.2473e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 6.0310e+00,  1.2280e+01, -6.4528e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 8.6271e+00, -8.0715e+00, -7.4813e+00],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-8.8400e+00,  2.3991e+00,  1.3457e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.2517e-01, -1.1006e+01,  7.2564e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 1.9505e+00, -4.7171e+00,  9.9868e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [-9.4378e+00, -1.8606e+00, -2.5668e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 9.3753e+00,  9.5171e+00,  3.9131e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-9.3810e+00, -2.3455e+00, -9.5194e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.5246e+00, -4.7838e+00, -5.2095e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-9.0604e+00,  1.7707e+00,  1.1066e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 3.6154e+00, -4.0640e+00, -7.9698e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 1.4749e+00, -4.7892e+00, -8.0890e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 6.3305e+00,  1.6119e+00,  1.1538e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 1.1142e+01,  6.9331e+00, -2.4779e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-5.1046e+00, -9.1610e+00,  2.3418e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [ 5.5101e+00, -2.1433e+00,  9.5442e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 5.8902e+00, -1.3887e+00,  4.0388e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0602, 0.0798,  ..., 1.0866, 1.1849, 0.4073],
        [0.0602, 0.0000, 0.1219,  ..., 1.0792, 1.2249, 0.3730],
        [0.0798, 0.1219, 0.0000,  ..., 1.0259, 1.1064, 0.3856],
        ...,
        [1.0866, 1.0792, 1.0259,  ..., 0.0000, 0.9488, 0.7600],
        [1.1849, 1.2249, 1.1064,  ..., 0.9488, 0.0000, 1.1760],
        [0.4073, 0.3730, 0.3856,  ..., 0.7600, 1.1760, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.66896739, 13.84126197, 13.89974693, 13.99048094, 13.68063098,
       14.09176259,  6.50409372, 12.41012046, 13.00380624, 12.13568598,
        7.76457095, 13.66051884,  7.09887557, 11.57249984, 12.11735334,
        5.13616224, 10.46739073, 14.10796466,  5.94535529,  9.62958973,
       13.48809595, 10.01804223, 11.81423161,  6.76056625,  6.29821673,
       13.27717982,  6.53250937,  6.2322769 ,  7.54127425,  6.28835101,
       11.20100795,  8.97052115, 12.67680439,  6.84796202,  6.81084724,
        9.15976746,  8.89301516,  9.38232121, 11.00694688,  6.12886024,
       11.46176259,  5.10446653, 13.79762415,  8.7907923 ,  9.61942445,
       12.41366754, 11.04953767, 13.35927527, 13.89974693,  7.60463349,
        4.91038884, 13.20988996,  9.48969978, 12.9089559 ,  5.14607014,
        9.78285424,  6.39438044,  9.19837792,  9.66977802,  5.10444097,
        7.44805901,  6.07992623,  6.5618968 ,  8.68832398,  5.0209005 ,
       13.60275211,  5.29009836,  9.23178204,  5.80312949,  5.43943791,
        9.96152788,  8.04718433,  5.01114719, 11.83509372, 13.46553623,
        6.53250937,  6.54127431,  5.13616224, 11.20100795, 13.32557133,
       11.87665501,  8.06643487, 13.12299145,  6.383897  ,  4.98989906,
       10.01804223,  6.97106912, 11.24434374,  4.89121805,  5.62005629,
        9.93553424,  8.85572292, 13.6660671 ,  5.80860165, 10.5575601 ,
       10.48721382,  5.94535529,  5.91222346,  5.34250225,  6.05165248,
       13.99048094, 12.00995898, 14.10796466,  9.09502385, 12.10222204,
       11.86239961])}
{'P': tensor([[1.4903e-03, 1.9182e-01, 1.8402e-01,  ..., 1.5962e-04, 4.1507e-04,
         1.6672e-02],
        [2.1648e-05, 4.6231e-03, 2.4624e-02,  ..., 3.2316e-04, 1.4511e-03,
         4.9946e-02],
        [2.0351e-05, 5.4333e-04, 5.1553e-03,  ..., 2.2745e-04, 1.3252e-03,
         8.5849e-02],
        ...,
        [6.7833e-03, 1.3051e-02, 1.1865e-02,  ..., 9.9595e-02, 5.5429e-02,
         2.9515e-02],
        [7.4491e-05, 8.9424e-05, 9.2820e-05,  ..., 8.7476e-05, 9.1588e-02,
         4.3973e-04],
        [7.8086e-05, 1.0740e-04, 1.0638e-04,  ..., 1.1846e-04, 1.1587e-02,
         2.0389e-01]]), 'X': tensor([[  6.1575,  12.2046,   3.3257],
        [  5.6928,  12.4710,   3.6001],
        [  5.4867,  12.5816,   2.9549],
        [  2.9470,  13.6085,   2.9809],
        [  3.4401,  13.4540,   4.6576],
        [  1.5653,  13.9377,   4.1773],
        [  5.2939,  -2.4784,  -4.5840],
        [ -2.6069, -10.4298,  -6.1995],
        [ 12.2218,   4.0953,  -8.2072],
        [ -9.4630,  -1.5145,   0.2983],
        [  1.4749,  -4.7892,  -0.8089],
        [  0.1931,  -4.7491,   7.4385],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.2480, -11.0153,  -3.0183],
        [  1.4489,   7.5256,   4.4529],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.4157,   0.0821,  -5.0025],
        [ -2.6069, -10.4298,  -6.1995],
        [  4.4039,   5.6903,   8.7366],
        [  6.2975,   0.0947,   5.1791],
        [ -3.5903,   7.4847,  -8.9112],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.3453,  -1.5462,   7.0212],
        [  9.4516,  -7.2501,  -1.8868],
        [  3.7916,  -3.9524,  -7.9547],
        [  2.5010,  -4.5736,  -0.8989],
        [  6.3587,   1.1596,  -3.6629],
        [ -5.0132,  -9.2234,  -8.3531],
        [  1.7122,  -4.7591,  -3.6734],
        [ -5.3328,  -8.9992,   7.3440],
        [ -3.0004,   7.6607,  -7.2640],
        [  2.4149,  -4.6005,  -4.3339],
        [ -6.7745,  -7.7264,   8.4684],
        [ -8.8934,   2.2597,   3.5831],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -3.1204, -10.2345,   9.2288],
        [ -0.5739,   7.9056,   0.6382],
        [ 12.5832,   1.7543,  -7.6134],
        [ -6.0175,  -8.4541,  -9.8408],
        [  2.4356,  -4.5942,  -5.0808],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -7.5401,   4.6718,   9.0033],
        [  2.0899,  -4.6870,  -7.4260],
        [  0.4996,   7.7738,  -4.4711],
        [ -8.4887,  -5.2448,  -0.6585],
        [  3.3167,   6.6022,   6.3554],
        [  6.3122,  -9.6994,  -6.3282],
        [ 12.5717,   1.9266,   7.4903],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.2508,   1.0572,   2.9978],
        [ -3.9508,  -9.8515,   1.7923],
        [  3.0876,   6.7527,  -5.3540],
        [  3.1892, -10.8220,  -6.8409],
        [  0.2571,  -4.7594,   1.5618],
        [  0.9945,  -4.8144,  -0.1897],
        [  2.6970,   6.9826,   9.7350],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.1752,   2.5242,   3.5874],
        [ -4.2271,   7.2387,  -1.8316],
        [ -5.1377,   6.7751,  -0.4438],
        [ -5.1369,   6.7755,   8.3836],
        [  8.5093,  -8.1764,  -3.2381],
        [ -5.9872,   6.2037,  -9.4477],
        [  3.0876,   6.7527,  -5.3540],
        [ 10.9916,   7.2133,  -0.9179],
        [  6.2975,   0.0947,   5.1791],
        [ -9.4735,  -0.7274,  -2.6749],
        [  2.8890,  -4.4316,   1.1600],
        [ -0.3432,   7.8900,   7.1805],
        [ -4.6393,   7.0461,   1.1624],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.3554,   4.8985,  -0.6665],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.0608,   3.9362,  -5.9567],
        [ -4.3963,   7.1629,   4.5263],
        [  2.1477,   7.2550,   8.9688],
        [  4.1297,  -3.7098,  -3.9838],
        [ -3.9456,  -9.8542,   0.7466],
        [  5.1886,  -2.6250,   6.4677],
        [  5.6928,  12.4710,   3.6001],
        [ 11.3367,  -4.4806,   0.4643],
        [  3.8182,  -3.9347,  -1.7961],
        [  0.9809,   7.6642,  -1.4217],
        [ 12.4597,  -0.9136,  -4.8763],
        [ -7.0253,  -7.4493,  -1.5978],
        [ 11.0275,  -5.0666,   8.4154],
        [  7.1821,  -9.1845,   9.8548],
        [  5.3858,   4.4260,  -9.6769],
        [  2.1477,   7.2550,   8.9688],
        [  6.3211,   1.7017,  -0.2164],
        [ -8.8631,  -4.3728,  -1.1746],
        [  5.2939,  -2.4784,  -4.5840],
        [ 11.9695,  -2.9407,  -6.3498],
        [  4.0642,  -3.7599,  -3.4565],
        [  8.0877,  -8.5297,  -1.1050],
        [ 12.2218,   4.0953,  -8.2072],
        [  9.0004,  -7.7195,   4.6945],
        [ 11.2939,   6.6325,  -5.8311],
        [ -3.1204, -10.2345,   9.2288],
        [ -3.7964,  -9.9294,   7.7147],
        [ 12.5789,   0.1722,   1.6470],
        [  5.4867,  12.5816,   2.9549],
        [  5.9587,  12.3216,   9.1695],
        [ -1.6906,   7.8860,   7.7142],
        [  0.6142,  14.0802,   0.5407]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0318, 0.0482,  ..., 0.3748, 0.6221, 0.4550],
        [0.0318, 0.0000, 0.0772,  ..., 0.3443, 0.6132, 0.4781],
        [0.0482, 0.0772, 0.0000,  ..., 0.4215, 0.6125, 0.4083],
        ...,
        [0.3748, 0.3443, 0.4215,  ..., 0.0000, 0.7159, 0.8034],
        [0.6221, 0.6132, 0.6125,  ..., 0.7159, 0.0000, 0.5782],
        [0.4550, 0.4781, 0.4083,  ..., 0.8034, 0.5782, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.66991027, 13.70893492, 13.72594022, 13.92390689, 13.88683436,
       14.02532758,  5.84533345, 10.7506423 , 12.889695  ,  9.58347456,
        5.01114719,  4.75303127,  8.47371868, 11.018084  ,  7.66377554,
       12.04924159,  9.41605684, 10.7506423 ,  7.19534146,  6.29821673,
        8.30125201, 11.21703654, 12.44177641, 11.9120321 ,  5.47702766,
        5.21280156,  6.46356806, 10.4977132 ,  5.05774295, 10.4606265 ,
        8.22728399,  5.19577146, 10.27572673,  9.17595899,  9.71392179,
       10.69963954,  7.92644746, 12.70489563, 10.37698435,  5.19986797,
       12.35421461,  8.87007495,  5.13182506,  7.78980299,  9.97822473,
        7.38844703, 11.57249984, 12.71843462, 12.4217523 ,  9.3109884 ,
       10.61417181,  7.42512621, 11.28215429,  4.7663526 ,  4.9160962 ,
        7.48538905,  9.88571713,  6.67121748,  8.38252937,  8.50278896,
        8.50268322, 11.80092578,  8.62162372,  7.42512621, 13.1471313 ,
        6.29821673,  9.5014146 ,  5.29009836,  7.89745107,  8.43626114,
        9.7712321 ,  8.83725081,  8.22728399,  6.383897  ,  8.97052115,
        8.40445872,  7.56617037,  5.5513072 , 10.61472636,  5.81481044,
       13.70893492, 12.18998195,  5.48276038,  7.72668467, 12.49317563,
       10.23946247, 12.13568598, 11.65923396,  6.97106912,  7.56617037,
        6.54615887,  9.88310614,  5.84533345, 12.32545637,  5.5366698 ,
       11.75438006, 12.889695  , 11.85743425, 13.09737053, 10.69963954,
       10.63038565, 12.58006038, 13.72594022, 13.68672855,  8.06517143,
       14.09357547])}
{'P': tensor([[3.8039e-03, 3.8820e-01, 1.7559e-01,  ..., 7.9373e-08, 7.9899e-04,
         1.5509e-07],
        [5.2746e-06, 3.6692e-03, 5.0660e-01,  ..., 1.0187e-07, 3.3530e-04,
         9.2543e-08],
        [4.5541e-06, 1.0138e-05, 8.0969e-03,  ..., 3.3037e-07, 3.0656e-04,
         1.4854e-07],
        ...,
        [1.4481e-07, 1.0898e-07, 9.5853e-08,  ..., 5.5381e-02, 3.2179e-06,
         2.6769e-04],
        [1.5026e-03, 3.6800e-03, 8.4835e-03,  ..., 1.4717e-04, 2.3323e-01,
         1.1437e-04],
        [2.5924e-04, 1.5961e-04, 1.1202e-04,  ..., 1.2204e-04, 1.6134e-03,
         2.2606e-01]]), 'X': tensor([[  6.0310,  12.2795,  -6.4528],
        [  4.0362,  13.2397,  -7.0030],
        [  2.0449,  13.8402,  -7.1642],
        [  3.2692,  13.5098,  -4.9032],
        [  0.6396,  14.0772,  -6.3786],
        [  0.8572,  14.0501,  -8.3364],
        [ -9.4196,   0.0478,   9.4140],
        [  5.1196,  -2.7160,   4.2705],
        [  9.9750,  -6.6338,  -1.2440],
        [ 12.5789,   0.1722,   1.6470],
        [  3.3839,  -4.1966,   8.3574],
        [  1.5653,  13.9377,   4.1773],
        [  6.1929,   2.4501,  -6.9352],
        [  4.5325,  -3.3643,   6.7699],
        [  2.0729,   7.2878,   7.0880],
        [  5.6414, -10.0303,   7.8525],
        [ -0.8569,   7.9156,  -6.5557],
        [ -7.2986,  -7.1225,  -3.0814],
        [ 12.4597,  -0.9136,  -4.8763],
        [  3.4401,  13.4540,   4.6576],
        [  0.3493,  -4.7727,  -6.5682],
        [  2.1824,  -4.6647,   1.1985],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.4430,  -0.1897,   4.8383],
        [ -9.4671,  -0.5553,   0.1117],
        [  6.3301,   0.3638,  -7.5685],
        [  4.1637,  13.1898,  -1.9189],
        [ -9.4716,  -0.6673,   7.6068],
        [  5.5158,  -2.1337,   3.5246],
        [  3.6249, -10.7255,   5.6605],
        [  5.7218,  -9.9934,   6.5764],
        [  8.1217,  -8.5024,  -0.6149],
        [ -0.0410,   7.8591,   9.3457],
        [ 11.2767,  -4.6006,   0.7868],
        [  3.7237,  -3.9966,  -5.0376],
        [  2.6614,  -4.5192,  -9.3641],
        [  0.9809,  -4.8145,   6.8706],
        [ 10.3251,   8.2928,  -3.0006],
        [  4.5057,  -3.3894,  -0.8160],
        [  8.3718,  -8.2953,   8.4666],
        [  6.0140,  -1.0661,  -2.5040],
        [ -1.9646,   7.8571,   3.3817],
        [  5.2199,   4.6861,  -1.9304],
        [  2.5829,  -4.5466,  -8.4015],
        [  2.5010,  -4.5736,  -0.8989],
        [ -8.9788,   2.0208,  -3.2867],
        [ 12.4691,   2.8525,  -5.5110],
        [  2.8994,   6.8675,  -9.5987],
        [  6.7792,  11.8084,   1.6251],
        [ -4.3963,   7.1629,   4.5263],
        [  9.0051,   9.9234,   5.4411],
        [  2.1625,  -4.6697,   7.0658],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.0047,   3.1062,  -6.7914],
        [ -4.4503,   7.1378,   3.3751],
        [  1.6408,   7.4591,  -4.2294],
        [  0.1931,  -4.7491,   7.4385],
        [  6.1730,  -0.5375,  -8.8861],
        [ -7.3554,   4.8985,  -0.6665],
        [  0.1041, -11.0045,   2.4311],
        [  2.2076,  13.8032,  -1.5099],
        [ -7.0120,   5.2825,   7.7762],
        [ -8.0750,  -6.0072,  -5.8267],
        [  6.3550,   0.7129,   0.4917],
        [  0.6933,   7.7336,   2.1179],
        [  6.1737,   2.5305,   3.9049],
        [ 10.9931,   7.2107,   7.7933],
        [ 10.9916,   7.2133,  -0.9179],
        [ -9.1624,  -3.4286,   2.0647],
        [ -2.0229, -10.6163,   8.6930],
        [ -9.2707,   0.9661,   4.4863],
        [ -8.4267,  -5.3697,   2.4172],
        [  1.0391,  -4.8141,  -0.0768],
        [  1.5653,  13.9377,   4.1773],
        [  2.2076,  13.8032,  -1.5099],
        [  5.5101,  -2.1433,   9.5442],
        [  1.8766,   7.3694,   1.1887],
        [  5.4242, -10.1261,   3.9595],
        [  5.7802,  -1.6363,  -6.2782],
        [  9.2692,   9.6370,   7.0746],
        [  6.0310,  12.2795,  -6.4528],
        [  1.4150,  13.9647,  -4.0915],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.1824,  -4.6647,   1.1985],
        [ 11.4703,  -4.2006,   7.4086],
        [ -8.7544,   2.6096,   3.3146],
        [  2.0243,   7.3086,   7.5432],
        [  5.8902,  -1.3887,   4.0388],
        [  5.8201,   3.5841,  -8.0609],
        [  3.6249, -10.7255,   5.6605],
        [ -0.4341, -10.9473,   7.4509],
        [  4.9531,   5.0576,   5.6794],
        [ 10.9875,   7.2207,  -8.8549],
        [  4.5503,   5.5359,   7.4735],
        [  6.3246,   1.6696,  -0.1544],
        [ -9.4524,  -1.6813,  -6.6862],
        [  6.0140,  -1.0661,  -2.5040],
        [ -3.3812, -10.1235,   7.4384],
        [ -5.7347,  -8.6908,   9.2878],
        [  9.0424,  -7.6779,  -7.5286],
        [ -4.4779,  -9.5609,  -2.4505],
        [  2.0547,   7.2957,  -1.9343],
        [ -8.8058,   2.4850,  -1.7425],
        [ -8.3535,  -5.5118,   9.8399],
        [  1.1006,   7.6318,  -2.4877],
        [  6.1222,  -0.7261,   8.8204]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0465, 0.0972,  ..., 1.6126, 0.5592, 1.0490],
        [0.0465, 0.0000, 0.0661,  ..., 1.6218, 0.5641, 1.0827],
        [0.0972, 0.0661, 0.0000,  ..., 1.5721, 0.5126, 1.0681],
        ...,
        [1.6126, 1.6218, 1.5721,  ..., 0.0000, 1.0596, 0.9626],
        [0.5592, 0.5641, 0.5126,  ..., 1.0596, 0.0000, 0.7341],
        [1.0490, 1.0827, 1.0681,  ..., 0.9626, 0.7341, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.68063098, 13.84126197, 13.99048094, 13.89974693, 14.09176259,
       14.07623266,  9.41970031,  5.79543038, 11.97949273, 12.58006038,
        5.39098669, 14.02532758,  6.65991877,  5.64464807,  7.57685908,
       11.50791859,  7.96181288, 10.1979669 , 12.49317563, 13.88683436,
        4.78544875,  5.14999799, 13.02818702,  9.44486609,  9.48337063,
        6.34059021, 13.83139024,  9.49511524,  5.91408504, 11.32149259,
       11.51557166, 11.75807252,  7.85919687, 12.17901033,  5.46244988,
        5.24461068,  4.91336994, 13.2430377 ,  5.63824971, 11.785566  ,
        6.10774544,  8.09899874,  7.01475376,  5.22903117,  5.21280156,
        9.20340031, 12.79126785,  7.4544866 , 13.61600536,  8.40445872,
       13.40024224,  5.14607014, 10.26327895,  6.76056625,  8.41148722,
        7.6374603 ,  4.75303127,  6.19633139,  8.83725081, 11.0050379 ,
       13.97857591,  8.77913428, 10.06437598,  6.3949043 ,  7.76457095,
        6.67218687, 13.14690137, 13.1471313 ,  9.78285424, 10.80728788,
        9.32094055,  9.99211663,  4.92497514, 14.02532758, 13.97857591,
        5.91222346,  7.60463349, 11.48734978,  6.00732193, 13.37121923,
       13.68063098, 14.03618649, 10.5575601 ,  5.14999799, 12.21532282,
        9.13507321,  7.58378655,  6.05165248,  6.83515201, 11.32149259,
       10.95593729,  7.07902036, 13.14777433,  7.16598273,  6.54127431,
        9.60080639,  6.10774544, 10.67322527, 10.412299  , 11.86239961,
       10.5575601 ,  7.57946459,  9.14973319, 10.00802627,  7.71075226,
        6.16512856])}
{'P': tensor([[4.0269e-03, 4.5467e-01, 1.9570e-01,  ..., 5.4779e-06, 2.1719e-08,
         1.0358e-08],
        [5.2266e-06, 3.8165e-03, 5.2694e-01,  ..., 1.5221e-06, 1.6787e-08,
         8.8905e-09],
        [4.5339e-06, 1.0161e-05, 8.1151e-03,  ..., 1.1517e-06, 3.2137e-08,
         1.9060e-08],
        ...,
        [3.2824e-04, 4.6244e-04, 5.4727e-04,  ..., 1.6661e-02, 1.3877e-07,
         1.4481e-07],
        [2.8733e-05, 1.2183e-05, 5.7687e-06,  ..., 1.0620e-04, 1.8065e-02,
         8.2089e-05],
        [1.9089e-06, 9.0810e-07, 4.8832e-07,  ..., 2.9277e-05, 9.8169e-03,
         1.4770e-02]]), 'X': tensor([[ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 1.9503e+00, -4.7172e+00, -6.2313e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 5.4242e+00, -1.0126e+01,  3.9595e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [ 7.5930e+00, -8.9037e+00,  7.2323e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 5.4426e+00, -1.0118e+01, -5.1268e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-9.2373e+00, -3.1225e+00, -8.9763e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 7.0515e+00,  1.1619e+01,  8.4289e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [-9.1433e+00, -3.5003e+00, -3.0209e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 2.5663e+00,  7.0526e+00, -6.8769e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0380, 0.0929,  ..., 1.1275, 0.7320, 1.2911],
        [0.0380, 0.0000, 0.0661,  ..., 1.1592, 0.7355, 1.3022],
        [0.0929, 0.0661, 0.0000,  ..., 1.1622, 0.6868, 1.2623],
        ...,
        [1.1275, 1.1592, 1.1622,  ..., 0.0000, 0.8910, 0.8384],
        [0.7320, 0.7355, 0.6868,  ..., 0.8910, 0.0000, 0.5966],
        [1.2911, 1.3022, 1.2623,  ..., 0.8384, 0.5966, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.6829843 , 13.84126197, 13.99048094, 13.89974693, 14.09176259,
       14.07623266, 11.51557166,  7.39599986,  8.06517143, 13.6105595 ,
        5.94909709,  8.5724042 , 12.67576792, 11.73848065,  5.10444097,
       13.37121923, 13.91198191,  5.05444288,  9.7712321 ,  6.02789634,
        5.05989547, 11.05672136,  7.9337647 ,  9.44486609, 13.59480906,
       11.11038551,  7.03733775, 11.54071859, 11.05672136,  8.55982105,
       10.00794468,  7.16058587, 11.48734978,  5.5748624 , 13.95472331,
        7.9337647 ,  9.15253344,  9.3109884 ,  9.76896904,  5.87305005,
        4.74770899,  7.69752262, 12.3344536 , 11.04313641,  4.98989906,
        9.99145096, 11.35182589,  5.30158497, 11.70168755, 10.61417181,
       13.93790235,  7.19534146, 12.32545637,  9.58072286,  7.48538905,
        8.94395628, 11.73567759,  7.92607342,  6.28835101, 13.68672855,
       11.4890861 , 10.57302546, 12.00995898,  7.16058587, 10.32715715,
        9.75075273, 10.71162681, 11.99633819,  6.76791483, 12.11735334,
        8.7273569 ,  5.18277858,  7.16058587,  6.27715105,  7.58378655,
        7.60529892,  8.41148722,  9.17016857, 13.59170809,  5.88782913,
        6.27905292,  8.859416  ,  9.79039371,  8.31625171,  6.83515201,
        6.2322769 ,  6.19707944,  9.20340031, 11.04313641,  9.78285424,
        7.50500064,  7.27759233,  7.02496223,  5.64865173,  5.37343173,
        4.98461983, 13.45124917,  8.09899874, 12.44177641, 13.3614598 ,
        8.95983586, 11.10066126,  8.95983586, 13.46553623, 11.73397309,
       11.42420508])}
{'P': tensor([[3.3183e-03, 2.3654e-01, 1.7035e-01,  ..., 2.8143e-07, 8.2550e-07,
         3.9927e-05],
        [8.4133e-06, 1.1213e-02, 1.8249e-05,  ..., 9.8894e-07, 1.3189e-06,
         2.9145e-05],
        [1.9962e-05, 1.9526e-01, 3.0631e-03,  ..., 3.0825e-07, 6.9317e-07,
         2.2854e-05],
        ...,
        [2.0222e-05, 1.1274e-05, 2.0532e-05,  ..., 6.5858e-02, 9.5398e-03,
         2.6422e-04],
        [1.4044e-03, 7.4432e-04, 1.5851e-03,  ..., 5.1459e-04, 7.8438e-02,
         5.4606e-03],
        [2.5321e-02, 3.4828e-02, 3.0995e-02,  ..., 1.1780e-04, 4.9146e-04,
         1.1257e-01]]), 'X': tensor([[  5.9587,  12.3216,   9.1695],
        [  2.7590,  13.6621,   9.4496],
        [  5.0617,  12.7957,   8.3544],
        [  1.9312,  13.8649,   8.3563],
        [  1.3705,  13.9723,   7.9266],
        [  1.1274,  14.0116,   8.1753],
        [  4.0457,  -3.7737,  -0.4704],
        [ -5.3736,   6.6310,  -1.5174],
        [ -0.8675, -10.8815,   6.8561],
        [ 12.1379,   4.4125,  -3.8833],
        [  8.2154,  -8.4262,   8.0171],
        [ -8.5859,  -5.0393,   0.5476],
        [  5.7963,  -1.6018,   1.8856],
        [  9.6614,   9.1784,  -1.2087],
        [  6.1579,  -0.5961,  -6.5595],
        [  3.0876,   6.7527,  -5.3540],
        [ -2.9497, -10.3028,  -7.3479],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -5.9987,   6.1949,  -6.2114],
        [  5.9526,   3.2525,  -4.3821],
        [  5.3804,  -2.3504,   7.2744],
        [  5.0190,   4.9705,  -7.9752],
        [  2.4932,   7.0903,  -5.4214],
        [ -1.6334, -10.7207,   7.1397],
        [  8.7131,  10.2216,   7.7915],
        [ 10.6283,  -5.7253,   7.4878],
        [ 11.2767,  -4.6006,   0.7868],
        [  5.1787,  -2.6383,  -4.2533],
        [ -8.8400,   2.3991,   1.3457],
        [  5.6248,   3.9984,   8.4933],
        [ 11.7795,  -3.4697,   3.5624],
        [  2.7590,  13.6621,   9.4496],
        [  8.6420,  -8.0580,   6.5781],
        [ -1.0407,   7.9166,  -6.6021],
        [  6.3550,   0.7129,   0.4917],
        [ -2.5683, -10.4433,  -2.9182],
        [  5.2939,  -2.4784,  -4.5840],
        [  8.6271,  -8.0715,  -7.4813],
        [  6.1575,  12.2046,   3.3257],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.1459,   1.4780,  -4.2403],
        [ -9.1795,  -3.3623,   8.5259],
        [  2.7685, -10.8982,  -6.8928],
        [ -4.4779,  -9.5609,  -2.4505],
        [  0.9809,  -4.8145,   6.8706],
        [ -7.4733,   4.7555,   6.2490],
        [ -8.6675,  -4.8569,  -9.1769],
        [  3.8111, -10.6787,  -4.6539],
        [  5.0506,   4.9277,  -9.5094],
        [  0.9659,  -4.8144,  -0.9594],
        [  1.7301,   7.4262,   5.0522],
        [  2.0243,   7.3086,   7.5432],
        [ -8.6801,   2.7809,  -0.8443],
        [  4.7544,  -3.1425,  -1.5315],
        [  0.3354,  -4.7708,  -4.9125],
        [ -8.5506,  -5.1154,   6.3423],
        [  3.0876,   6.7527,  -5.3540],
        [  8.5198,  10.4091,  -8.6486],
        [ -9.0574,   1.7801,   8.8986],
        [ -5.6169,  -8.7844,   8.9057],
        [  4.3459,  -3.5327,   0.3667],
        [  2.3589,  13.7668,  -3.6501],
        [ -9.3616,  -2.4749,   0.1793],
        [  7.0358,  -9.2784,  -9.8479],
        [  1.8447,  -4.7372,   8.1235],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -9.3616,  -2.4750,   3.7765],
        [  3.2692,  13.5098,  -4.9032],
        [  2.4697,  -4.5836,   1.4593],
        [  6.1755,   2.5233,  -2.0042],
        [ -3.9456,  -9.8542,   0.7466],
        [ -8.8631,  -4.3728,  -1.1746],
        [ -3.4100,   7.5436,  -7.6980],
        [  5.0233,   4.9647,  -3.1788],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 11.6634,   5.8050,   9.1590],
        [  3.9895,   6.0816,  -6.5235],
        [  0.8572,  14.0501,  -8.3364],
        [  1.5246,  -4.7838,  -5.2095],
        [  6.3505,   0.6235,  -8.5849],
        [ -6.7745,  -7.7264,   8.4684],
        [  6.2675,  12.1379,  -8.9720],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.9314,  -9.3435,  -1.2729],
        [ 10.5955,   7.8824,  -3.7006],
        [  5.9026,   3.3837,  -0.6573],
        [  4.3459,  -3.5327,   0.3667],
        [  3.4604,  -4.1545,   1.8091],
        [  6.0140,  -1.0661,  -2.5040],
        [  6.2895,   0.0405,   5.7652],
        [ 10.9394,   7.3069,  -8.9365],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.2630,   2.1057,  -5.2136],
        [  6.8704,  -9.3809,   7.5769],
        [ -1.0482, -10.8487,  -5.9066],
        [  0.1497,   7.8334,  -5.4412],
        [ -6.1914,  -8.2997,   8.5380],
        [ -4.7334,   6.9982,  -0.4392],
        [  7.0642,  11.6104,  -9.8316],
        [  5.5240,  -2.1198,   6.2055],
        [ -8.6426,  -4.9136,   9.4941],
        [ 11.9936,  -2.8677,   6.0614],
        [  0.2732,  -4.7619,  -1.1577],
        [  6.3516,   0.6425,   0.3425],
        [  4.6231,   5.4556,   8.4973]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1050, 0.0172,  ..., 1.0318, 0.8620, 0.5203],
        [0.1050, 0.0000, 0.0933,  ..., 0.9602, 0.7977, 0.4206],
        [0.0172, 0.0933, 0.0000,  ..., 1.0146, 0.8449, 0.5050],
        ...,
        [1.0318, 0.9602, 1.0146,  ..., 0.0000, 0.1855, 0.6258],
        [0.8620, 0.7977, 0.8449,  ..., 0.1855, 0.0000, 0.5116],
        [0.5203, 0.4206, 0.5050,  ..., 0.6258, 0.5116, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.68672855, 13.93790235, 13.76047999, 13.99877413, 14.0393949 ,
       14.05687815,  5.53255882,  8.53501777, 10.91601655, 12.91505111,
       11.76830981,  9.95552626,  6.01356004, 13.32613445,  6.18668463,
        7.42512621, 10.7167285 , 10.06407839,  8.62329466,  6.78325151,
        5.87132718,  7.06372172,  7.5158489 , 10.84437705, 13.43127131,
       12.07225469, 12.17901033,  5.81199422,  9.15976746,  6.90115901,
       12.27991991, 13.93790235, 11.8159254 ,  7.98469166,  6.3949043 ,
       10.75443106,  5.84533345, 11.81423161, 13.66991027, 12.19805394,
        5.24461068,  9.26455795,  9.77588801, 11.24434374, 10.5575601 ,
        4.91336994,  8.8580671 ,  9.93553424, 11.33837396,  7.05627461,
        4.91038884,  7.62509582,  7.58378655,  9.11473391,  5.69913817,
        4.78258092,  9.96391045,  7.42512621, 13.45124917,  9.23071698,
       10.42667093,  5.60064319, 13.96746816,  9.68323686, 11.64436684,
        5.08372149, 10.00410625,  9.68324553, 13.89974693,  5.20659892,
        6.67108662, 10.61472636,  9.88310614,  8.27853076,  7.06272256,
       10.51437947, 13.02818702,  7.27340834, 14.07623266,  5.0209005 ,
        6.38105015, 10.27572673, 13.66051884,  9.37989631, 11.63383221,
       13.20597932,  6.80370254,  5.60064319,  5.40689332,  6.10774544,
        6.28961988, 13.15525803,  9.62958973,  6.60752373, 11.62770332,
       10.8992524 ,  7.83486943, 10.35464916,  8.4486771 , 13.59056613,
        5.91677188,  9.94173708, 12.33167382,  4.76969364,  6.38399789,
        7.15100303])}
{'P': tensor([[5.0246e-03, 3.3432e-01, 1.7813e-01,  ..., 3.4728e-04, 6.5456e-08,
         8.2034e-07],
        [1.2880e-05, 1.3812e-02, 1.7446e-03,  ..., 2.2425e-03, 1.0144e-07,
         1.6285e-06],
        [6.0848e-06, 3.1837e-03, 3.2946e-03,  ..., 2.1367e-04, 2.4619e-08,
         3.1906e-07],
        ...,
        [6.2943e-06, 1.3786e-05, 9.7964e-06,  ..., 4.9239e-02, 2.6113e-06,
         6.6272e-05],
        [3.1130e-03, 7.9821e-04, 8.3145e-04,  ..., 4.3066e-05, 6.2641e-02,
         1.6206e-03],
        [1.0888e-03, 5.7931e-04, 4.1717e-04,  ..., 5.0707e-04, 6.1874e-03,
         8.6751e-02]]), 'X': tensor([[  5.9003,  12.3550,   2.1917],
        [  2.9470,  13.6085,   2.9809],
        [  3.3633,  13.4794,   0.5193],
        [  1.1350,  14.0104,   0.6321],
        [  1.5653,  13.9377,   4.1773],
        [  0.6142,  14.0802,   0.5407],
        [ 12.4690,  -0.8528,  -9.3374],
        [  5.8201,   3.5841,  -8.0609],
        [ -4.2271,   7.2387,  -1.8316],
        [  9.4516,  -7.2501,  -1.8868],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -9.2707,   0.9661,   4.4863],
        [  0.1041, -11.0045,   2.4311],
        [  2.3928,   7.1404,  -1.7035],
        [ -9.2687,  -2.9794,   0.6436],
        [ -8.8058,   2.4850,  -1.7425],
        [  6.2365,  -0.2578,  -6.0212],
        [ 12.5770,   1.8512,  -0.6578],
        [  1.7815,  -4.7481,   4.4289],
        [ -9.4196,   0.0478,   9.4140],
        [  3.4604,  -4.1545,   1.8091],
        [ 11.7796,  -3.4696,   9.0088],
        [ -9.4378,  -1.8606,  -2.5668],
        [  9.3560,   9.5391,   9.4943],
        [ -3.6381,   7.4683,   7.3544],
        [ 12.2592,   3.9425,   2.0383],
        [  6.3048,   0.1469,  -9.1691],
        [  1.7232,  -4.7574,   3.1270],
        [  6.3497,   1.3680,  -1.9076],
        [ -3.7964,  -9.9294,   7.7147],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.3266,   0.3292,   9.3502],
        [ 11.6165,  -3.8715,   6.4594],
        [  3.9515,  -3.8425,   6.1729],
        [  0.1677,  -4.7447,  -6.1529],
        [  3.7237,  -3.9966,  -5.0376],
        [ -8.9942,  -3.9990,   6.0448],
        [ -9.3087,   0.7792,  -7.5587],
        [ -6.7992,   5.4991,   9.9186],
        [  2.3928,   7.1404,  -1.7035],
        [ -1.7290,   7.8825,  -3.3831],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.4541,  -0.3327,   8.8246],
        [  8.4691,  10.4571,  -5.0650],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.0087,   4.0173,   9.1091],
        [ -6.7745,  -7.7264,   8.4684],
        [ -9.3384,  -2.6161,  -4.7630],
        [  2.7976,  -4.4682,  -4.2020],
        [  1.1139,   7.6281,  -2.5680],
        [ -9.3616,  -2.4750,   3.7765],
        [ 12.2435,   4.0076,   8.1257],
        [  9.6587,   9.1817,  -0.8212],
        [  2.2076,  13.8032,  -1.5099],
        [  1.3403,  -4.8011,  -1.1698],
        [ 11.6019,   5.9540,  -5.1715],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.9653,  -9.3225,  -3.4727],
        [  9.2979,   9.6048,   7.3138],
        [  6.3413,   0.4883,  -6.8416],
        [ 12.4996,   2.6335,  -6.3166],
        [ -5.6174,   6.4708,  -1.7339],
        [ -8.9624,  -4.0944,  -4.9622],
        [  5.0617,  12.7957,   8.3544],
        [ 12.4996,   2.6335,  -6.3166],
        [  2.2076,  13.8032,  -1.5099],
        [  5.9912,   3.1453,  -3.9989],
        [  6.1973,  -0.4372,   6.5720],
        [  2.4932,   7.0903,  -5.4214],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.3246,   1.6696,  -0.1544],
        [ 12.0582,  -2.6640,   0.7132],
        [ -9.1795,  -3.3623,   8.5259],
        [  6.2770,  -0.0379,   2.4195],
        [ -4.9217,   6.8979,  -1.7667],
        [  6.8093,  -9.4178,   8.4244],
        [  3.9663,   6.1018,  -5.4104],
        [ -1.0407,   7.9166,  -6.6021],
        [  2.9410,   6.8428,   5.4458],
        [  1.1006,   7.6318,  -2.4877],
        [  8.1217,  -8.5024,  -0.6149],
        [  9.1820,   9.7334,  -2.7932],
        [ -8.4887,  -5.2448,  -0.6585],
        [  2.0870,  -4.6877,   9.6823],
        [ 11.2517,  -4.6495,   2.9800],
        [  5.3602,  -2.3808,   7.2098],
        [  0.7275,  14.0667,  -0.9815],
        [ -2.5683, -10.4433,  -2.9182],
        [ -9.3705,   0.4208,  -3.5815],
        [  4.0362,  13.2397,  -7.0030],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -5.6423,   6.4537,  -1.2091],
        [ -4.1296,   7.2803,  -2.7194],
        [ -1.0482, -10.8487,  -5.9066],
        [  4.5503,   5.5359,   7.4735],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.2790,  -0.0259,  -5.4116],
        [  3.4604,  -4.1545,   1.8091],
        [  2.1477,   7.2550,   8.9688],
        [  0.9809,  -4.8145,   6.8706],
        [ -1.6220,   7.8917,   0.0315],
        [  5.6093,  -1.9705,   6.4807],
        [ -7.4733,   4.7555,   6.2490],
        [ -5.1369,   6.7755,   8.3836],
        [ 11.7795,  -3.4697,   3.5624],
        [  5.3804,  -2.3504,   7.2744]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1282, 0.2122,  ..., 0.7943, 0.2946, 0.3298],
        [0.1282, 0.0000, 0.3358,  ..., 0.7466, 0.2874, 0.2586],
        [0.2122, 0.3358, 0.0000,  ..., 0.8453, 0.3515, 0.4590],
        ...,
        [0.7943, 0.7466, 0.8453,  ..., 0.0000, 0.5073, 0.4896],
        [0.2946, 0.2874, 0.3515,  ..., 0.5073, 0.0000, 0.1536],
        [0.3298, 0.2586, 0.4590,  ..., 0.4896, 0.1536, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.69162795, 13.92390689, 13.89264822, 14.05633441, 14.02532758,
       14.09357547, 12.4980845 ,  6.83515201,  8.38252937, 11.9120321 ,
        9.91668182,  9.32094055, 11.0050379 ,  7.53063885,  9.73579738,
        9.14973319,  6.24187044, 12.7125079 ,  5.0713364 ,  9.41970031,
        5.40689332, 12.27993098,  9.61942445, 13.3614598 ,  8.30729352,
       12.87752052,  6.30647746,  5.05989547,  6.49538841, 10.63038565,
        5.1224079 ,  6.33517349, 12.24467041,  5.51177501,  4.74770899,
        5.46244988,  9.84314712,  9.34127013,  8.74470218,  7.53063885,
        8.06990483,  8.27853076,  9.45995955, 13.45642379,  5.38342077,
        8.95983586, 10.27572673,  9.69792513,  5.27178363,  7.70897675,
        9.68324553, 12.88270018, 13.32645552, 13.97857591,  4.98461983,
       13.04051744,  9.99211663, 11.63724792, 13.36800073,  6.36004393,
       12.77402351,  8.56889738,  9.85330646, 13.76047999, 12.77402351,
       13.97857591,  6.76661595,  6.21275016,  7.5158489 ,  9.88571713,
        6.54127431, 12.3489324 ,  9.77588801,  6.27715105,  8.47371868,
       11.62158554,  7.27759233,  7.98469166,  7.44805901,  7.71075226,
       11.75807252, 13.38091121,  9.97822473,  5.13125513, 12.17451685,
        5.86519525, 14.08549802, 10.75443106,  9.37989631, 13.84126197,
        9.5014146 ,  8.5724042 ,  8.36995953, 10.8992524 ,  7.16598273,
        9.11473391,  6.27905292,  5.40689332,  7.56617037,  4.91336994,
        8.05668609,  5.94535529,  8.8580671 ,  8.50268322, 12.27991991,
        5.87132718])}
{'P': tensor([[3.3306e-03, 3.2698e-01, 2.4835e-01,  ..., 4.4460e-08, 1.5879e-07,
         5.3212e-08],
        [3.5447e-06, 2.7168e-03, 2.9963e-01,  ..., 2.3276e-08, 1.9781e-07,
         2.4735e-08],
        [5.9075e-06, 2.6472e-05, 7.8563e-03,  ..., 6.9779e-08, 7.1402e-07,
         6.0078e-08],
        ...,
        [9.3504e-05, 5.6507e-05, 5.9738e-05,  ..., 3.6934e-01, 1.0114e-05,
         8.3817e-04],
        [8.9335e-06, 5.7550e-06, 4.9211e-06,  ..., 7.7857e-05, 2.8384e-01,
         1.0728e-03],
        [1.0371e-02, 3.5416e-03, 2.5360e-03,  ..., 1.9091e-04, 2.9862e-06,
         1.4525e-01]]), 'X': tensor([[  5.8930,  12.3592,  -8.9201],
        [  3.7298,  13.3537,  -9.4653],
        [  2.7295,  13.6703,  -8.9328],
        [  2.5469,  13.7192,  -9.6935],
        [  0.8572,  14.0501,  -8.3364],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.6675,  -4.8569,  -9.1769],
        [  4.4039,   5.6903,   8.7366],
        [ 10.6673,  -5.6649,  -4.1646],
        [ -4.7847,  -9.3729,  -3.7741],
        [  0.5850,  -4.7982,   9.6639],
        [  2.7976,  -4.4682,  -4.2020],
        [ -7.4807,   4.7463,   2.0616],
        [  6.9433,  -9.3362,  -0.3421],
        [  0.5250, -11.0306,   5.5358],
        [ -4.3963,   7.1629,   4.5263],
        [  6.2038,   2.4020,  -3.1692],
        [  1.1994,   7.6035,   2.4295],
        [ 10.1983,  -6.3430,  -0.3200],
        [ -9.4766,  -1.1116,  -2.1706],
        [  5.1142,  -2.7230,  -4.1966],
        [  6.3301,   0.3638,  -7.5685],
        [  2.3373,  -4.6232,  -5.2246],
        [ -8.6229,   2.9068,   2.4429],
        [ -3.2053, -10.1993,  -3.8736],
        [ -8.3539,  -5.5111,   0.3373],
        [  2.1824,  -4.6647,   1.1985],
        [  6.2905,   1.9366,   2.1169],
        [ -6.8618,  -7.6322,   3.7027],
        [  1.1350,  14.0104,   0.6321],
        [  1.7903,  -4.7467,   0.1168],
        [  6.2365,  -0.2578,  -6.0212],
        [  9.1620,  -7.5573,  -9.9882],
        [ -9.4203,  -2.0340,   2.5856],
        [ 11.3691,   6.4760,   1.0787],
        [ 10.1798,   8.5002,  -5.8893],
        [  0.9074,  -4.8139,   8.1674],
        [  1.7000,  -4.7609,   2.7935],
        [ -3.2736, -10.1703,   3.8712],
        [  0.6142,  14.0802,   0.5407],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -7.6719,  -6.6260,   7.1468],
        [  6.3303,   1.6145,  -8.4083],
        [ -4.2271,   7.2387,  -1.8316],
        [  7.4132,  11.3526,   0.3322],
        [ -2.1817,   7.8274,  -4.6298],
        [ -6.0195,   6.1789,  -2.9496],
        [  1.3705,  13.9723,   7.9266],
        [  1.7228,  -4.7575,   8.3396],
        [ 12.4901,  -0.7052,   1.7315],
        [ -8.9282,   2.1648,   8.4894],
        [  6.2630,   2.1057,  -5.2136],
        [  6.1984,  -0.4327,   1.5438],
        [  1.9312,  13.8649,   8.3563],
        [  1.9505,  -4.7171,   9.9868],
        [  2.5128, -10.9365,   9.9645],
        [  6.9956,  11.6590,  -7.6876],
        [ -8.6229,   2.9068,   2.4429],
        [ -9.4735,  -0.7274,  -2.6749],
        [  8.7131,  10.2216,   7.7915],
        [  1.2728, -11.0372,   9.8121],
        [  5.7845,   3.6652,   0.9175],
        [ -8.4573,   3.2448,   5.4370],
        [  9.5018,  -7.1945,  -1.1892],
        [  4.4915,   5.5990,   7.0405],
        [ -1.9714,   7.8563,  -1.8084],
        [ -0.0282,   7.8575,  -1.1746],
        [  6.3305,   1.6119,   1.1538],
        [ -7.0825,   5.2073,  -5.0497],
        [  0.1206,  -4.7363,   8.7103],
        [ -6.3489,   5.9117,   6.5161],
        [ -6.4844,  -8.0230,   8.5209],
        [  1.1139,   7.6281,  -2.5680],
        [ -3.3812, -10.1235,   7.4384],
        [  2.4697,  -4.5836,   1.4593],
        [ -7.2986,  -7.1225,  -3.0814],
        [ -8.7409,  -4.6837,  -6.5840],
        [  5.4429,  -2.2529,   9.7306],
        [ -1.7009,   7.8851,  -8.2437],
        [  4.0587, -10.6112,   6.2273],
        [ 11.9695,  -2.9407,  -6.3498],
        [ 12.5789,   0.1722,   1.6470],
        [ 10.8670,  -5.3428,  -2.8637],
        [  5.1886,  -2.6250,   6.4677],
        [  5.2608,  -2.5256,  -9.1446],
        [  6.5428,  -9.5728,  -2.9897],
        [ -9.4739,  -1.2497,   0.5483],
        [ -2.0840,   7.8415,  -7.8657],
        [ 11.5198,  -4.0923,   5.5088],
        [  0.5250, -11.0306,   5.5358],
        [  1.4150,  13.9647,  -4.0915],
        [ -0.8569,   7.9156,  -6.5557],
        [ -9.4541,  -0.3327,   8.8246],
        [  2.0547,   7.2957,  -1.9343],
        [  2.8474,   6.8979,   6.6457],
        [  3.3744,   6.5623,  -6.3636],
        [  6.9314,  -9.3435,  -1.2729],
        [ -8.1905,   3.7253,   4.1175],
        [  5.8930,  12.3592,  -8.9201],
        [  0.3493,  -4.7727,  -6.5682],
        [ -8.0534,  -6.0429,   2.4583],
        [  5.2608,  -2.5256,  -9.1446],
        [  1.2728, -11.0372,   9.8121],
        [  5.7784,  -1.6401,   9.5587],
        [ -5.2752,  -9.0409,  -8.2568],
        [ 12.2921,  -1.7902,  -4.0870]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0346, 0.0907,  ..., 1.1839, 1.1917, 0.3349],
        [0.0346, 0.0000, 0.0603,  ..., 1.1829, 1.1578, 0.3202],
        [0.0907, 0.0603, 0.0000,  ..., 1.1478, 1.1029, 0.2707],
        ...,
        [1.1839, 1.1829, 1.1478,  ..., 0.0000, 1.4590, 0.8930],
        [1.1917, 1.1578, 1.1029,  ..., 1.4590, 0.0000, 1.0115],
        [0.3349, 0.3202, 0.2707,  ..., 0.8930, 1.0115, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.6922425 , 13.86480175, 13.94008948, 13.95361098, 14.07623266,
       13.99048094,  9.93553424,  7.19534146, 12.07818871, 10.52357479,
        4.83371341,  5.27178363,  8.85939626, 11.63502679, 11.04313641,
        8.40445872,  6.65259775,  7.69752262, 12.00995898,  9.54154096,
        5.79391825,  6.34059021,  5.18047638,  9.09963894, 10.6910768 ,
       10.00794468,  5.14999799,  6.58184474, 10.26327895, 14.05633441,
        5.07306274,  6.24187044, 11.87665501,  9.63742669, 13.08414689,
       13.26209545,  4.89869205,  5.05534609, 10.68416401, 14.09357547,
        9.78286733, 10.1371552 ,  6.53290663,  8.38252937, 13.55868684,
        8.12580976,  8.62631801, 14.0393949 ,  5.05981914, 12.50996828,
        9.18690018,  6.60752373,  6.2134961 , 13.99877413,  5.10446653,
       11.22142195, 13.59673406,  9.09963894,  9.5014146 , 13.43127131,
       11.11038551,  6.84796202,  9.05843201, 11.9182956 ,  7.17789666,
        8.09984231,  7.8575648 ,  6.53250937,  8.7907923 ,  4.73784222,
        8.67502348, 10.31584102,  7.70897675, 10.67322527,  5.20659892,
       10.1979669 ,  9.91668182,  5.89074104,  8.06643487, 11.36089355,
       12.32545637, 12.58006038, 12.10942412,  5.81481044,  5.83560009,
       11.59512194,  9.55593374,  8.1137452 , 12.22503754, 11.04313641,
       14.03618649,  7.96181288,  9.45995955,  7.57946459,  7.46248947,
        7.37902766, 11.63383221,  8.99791316, 13.6922425 ,  4.78544875,
       10.06850677,  5.83560009, 11.11038551,  6.00663314, 10.46739073,
       12.4217523 ])}
{'P': tensor([[3.6463e-03, 2.8682e-01, 2.3625e-01,  ..., 2.5192e-07, 3.1291e-08,
         3.1537e-06],
        [1.4197e-05, 1.4365e-02, 1.2317e-03,  ..., 2.1228e-06, 7.5523e-08,
         5.9208e-06],
        [4.0186e-06, 6.5908e-03, 2.4733e-03,  ..., 2.4016e-07, 1.5322e-08,
         8.9711e-07],
        ...,
        [1.3036e-06, 1.0443e-06, 8.6625e-07,  ..., 1.2834e-01, 5.2933e-05,
         1.5449e-04],
        [5.5028e-04, 1.5238e-04, 2.3252e-04,  ..., 2.8546e-07, 3.9770e-02,
         2.5746e-04],
        [3.3640e-03, 3.4226e-03, 1.9795e-03,  ..., 4.7481e-05, 1.7056e-05,
         8.3292e-02]]), 'X': tensor([[  5.6928,  12.4710,   3.6001],
        [  2.9470,  13.6085,   2.9809],
        [  3.4401,  13.4540,   4.6576],
        [  1.5653,  13.9377,   4.1773],
        [  1.5662,  13.9375,   4.5142],
        [  2.2182,  13.8007,   5.1889],
        [ 12.3453,  -1.5462,   7.0212],
        [  0.3038,   7.8092,   0.4164],
        [ -5.9987,   6.1949,  -6.2114],
        [ 10.0619,   8.6627,  -9.2884],
        [ -9.4756,  -0.8166,  -9.6021],
        [  2.0937,  13.8293,  -2.3533],
        [  1.8719,   7.3713,  -3.7465],
        [  2.5636, -10.9293,   9.6519],
        [ -0.4341, -10.9473,   7.4509],
        [ -3.9700,   7.3453,  -4.1486],
        [  2.0419,  -4.6978,  -5.0983],
        [  5.3858,   4.4260,  -9.6769],
        [  6.2883,   0.0325,   5.2692],
        [  2.1379,  -4.6757,   6.4981],
        [  6.3087,   1.8053,  -1.9481],
        [ 12.2218,   4.0953,  -8.2072],
        [  2.6310, -10.9195,   3.8054],
        [  0.1252, -11.0062,   7.2564],
        [ -4.1296,   7.2803,  -2.7194],
        [ -4.2271,   7.2387,  -1.8316],
        [  8.8474,  10.0868,  -8.8644],
        [  1.4150,  13.9647,  -4.0915],
        [  0.9967,  -4.8144,  -9.7370],
        [  2.2644,  -4.6434,   9.7508],
        [ 12.4996,   2.6335,  -6.3166],
        [  5.2188,  -2.5839,   2.1833],
        [ -8.8835,  -4.3178,   9.3080],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.3246,   1.6696,  -0.1544],
        [  6.2804,   2.0019,   0.8607],
        [  0.3354,  -4.7708,  -4.9125],
        [  3.7298,  13.3537,  -9.4653],
        [ 12.5267,  -0.4125,  -8.7331],
        [  1.7218,  -4.7576,  -3.3617],
        [ 12.4592,  -0.9168,  -6.7571],
        [  2.0870,  -4.6877,   9.6823],
        [ -8.0766,  -6.0046,  -3.1873],
        [  6.2630,   2.1057,  -5.2136],
        [ -1.5408,   7.8978,   5.3470],
        [  1.7659,   7.4127,   4.7871],
        [ -1.6220,   7.8917,   0.0315],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -3.0004,   7.6607,  -7.2640],
        [ 12.4901,  -0.7052,   1.7315],
        [ 12.4837,   2.7510,   0.4449],
        [  2.0419,  -4.6978,  -5.0983],
        [  9.0652,   9.8597,  -3.5949],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.0560,  -0.9421,  -5.0334],
        [ -7.3333,   4.9246,  -7.9212],
        [ -7.9667,   4.0814,  -0.6397],
        [ -1.6086,   7.8928,   4.3385],
        [ -8.4573,   3.2448,   5.4370],
        [  6.1685,  12.1979,  -6.2473],
        [  1.9241,   7.3503,   7.9004],
        [ 10.8670,  -5.3428,  -2.8637],
        [  9.7328,   9.0902,  -0.8917],
        [  1.5246,  -4.7838,  -5.2095],
        [  4.2275,  -3.6320,  -0.2714],
        [  4.7482, -10.3907,   5.9149],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.3087,   0.7792,  -7.5587],
        [ -7.5401,   4.6718,   9.0033],
        [ -3.4100,   7.5436,  -7.6980],
        [ -7.4808,   4.7462,   2.0266],
        [ -6.2200,  -8.2737,   5.5079],
        [  2.5636, -10.9293,   9.6519],
        [  5.4179,  -2.2925,  -3.4641],
        [  5.1292,   4.8183,  -5.7997],
        [  2.0870,  -4.6877,   9.6823],
        [  0.6952,  -4.8060,   8.8976],
        [ -6.1899,  -8.3011,  -7.9269],
        [  6.3124,   0.2057,  -0.4593],
        [ -6.9838,  -7.4965,   0.9796],
        [  3.5284, -10.7484,   5.3347],
        [  2.5636, -10.9293,   9.6519],
        [ -9.3616,  -2.4749,   0.1793],
        [ 12.4690,  -0.8528,  -9.3374],
        [  4.1140,  -3.7220,  -0.1273],
        [  3.3744,   6.5623,  -6.3636],
        [ -9.1960,  -3.2964,   1.6921],
        [  4.5057,  -3.3894,  -0.8160],
        [  5.7784,  -1.6401,   9.5587],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -1.2454,   7.9127,  -0.5715],
        [ -8.4297,  -5.3637,  -9.3962],
        [  1.0000,   7.6591,  -4.5011],
        [  1.8766,   7.3694,   1.1887],
        [  1.3767, -11.0341,   9.9415],
        [ -7.8359,   4.2735,   7.2943],
        [ -8.0588,  -6.0341,  -2.3006],
        [  6.3266,   0.3292,   9.3502],
        [ 11.6801,  -3.7196,  -0.9181],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.3653,   3.4184,   9.3454],
        [ -7.6293,  -6.6859,  -5.5161],
        [ 11.0275,  -5.0666,   8.4154],
        [  5.9526,   3.2525,  -4.3821]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 8.7677e-02, 8.8586e-02,  ..., 1.5736e+00, 4.7170e-01,
         8.1448e-01],
        [8.7677e-02, 0.0000e+00, 1.6866e-01,  ..., 1.4909e+00, 4.6501e-01,
         7.2949e-01],
        [8.8586e-02, 1.6866e-01, 3.0339e-04,  ..., 1.6288e+00, 4.4301e-01,
         8.9807e-01],
        ...,
        [1.5736e+00, 1.4909e+00, 1.6288e+00,  ..., 0.0000e+00, 1.4097e+00,
         9.8724e-01],
        [4.7170e-01, 4.6501e-01, 4.4301e-01,  ..., 1.4097e+00, 0.0000e+00,
         9.6741e-01],
        [8.1448e-01, 7.2949e-01, 8.9807e-01,  ..., 9.8724e-01, 9.6741e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.70893492, 13.92390689, 13.88683436, 14.02532758, 14.02526505,
       13.97779839, 12.44177641,  7.81509631,  8.62329466, 13.27717982,
        9.51074763, 13.98691426,  7.60529892, 11.22596994, 10.95593729,
        8.34949383,  5.1224079 ,  6.97106912,  6.28835101,  5.14124612,
        6.5618968 , 12.889695  , 11.23201619, 11.00694688,  8.36995953,
        8.38252937, 13.41713661, 14.03618649,  4.91652556,  5.16611782,
       12.77402351,  5.82345867,  9.87720102,  9.49511524,  6.54127431,
        6.59176018,  4.78258092, 13.86480175, 12.53345445,  5.05963217,
       12.49291768,  5.13125513, 10.06407839,  6.60752373,  8.04664886,
        7.62011734,  8.05668609, 10.33712258,  8.22728399, 12.50996828,
       12.78327037,  5.1224079 , 13.39372426,  5.48276038,  6.12886024,
        8.83341608,  8.95133961,  8.0550286 ,  9.05843201, 13.66896739,
        7.59795445, 12.10942412, 13.31764181,  5.0209005 ,  5.57341623,
       11.42420508,  6.00663314,  9.60080639,  9.34127013,  8.87007495,
        8.27853076,  8.859416  , 10.35093485, 11.22596994,  5.8828986 ,
        7.03733775,  5.13125513,  4.85603886, 10.35485331,  6.31576264,
       10.24557047, 11.31276109, 11.22596994,  9.68323686, 12.4980845 ,
        5.54776934,  7.37902766,  9.76896904,  5.63824971,  6.00663314,
        9.60080639,  8.01009083,  9.99145096,  7.72415048,  7.60463349,
       11.11969815,  8.92549752, 10.0674887 ,  6.33517349, 12.25806559,
       12.18045978,  5.19065439,  9.0368458 , 10.14436912, 12.13568598,
        6.78325151])}
{'P': tensor([[4.0848e-03, 4.1433e-01, 1.6559e-01,  ..., 4.9586e-05, 5.3857e-07,
         2.8662e-06],
        [9.6653e-06, 9.8277e-03, 3.9660e-01,  ..., 1.4575e-04, 1.3466e-06,
         1.4693e-05],
        [4.7568e-06, 3.8130e-05, 7.3409e-03,  ..., 6.5944e-05, 8.8463e-07,
         1.3577e-05],
        ...,
        [2.6897e-05, 5.0438e-05, 5.4909e-05,  ..., 2.9965e-01, 1.2693e-03,
         2.2009e-02],
        [4.4448e-05, 2.7364e-05, 1.7850e-05,  ..., 1.1012e-04, 6.1876e-02,
         1.0813e-04],
        [1.0803e-06, 1.0309e-06, 9.4378e-07,  ..., 3.0193e-05, 6.0658e-03,
         8.0399e-02]]), 'X': tensor([[  5.4867,  12.5816,   2.9549],
        [  2.9470,  13.6085,   2.9809],
        [  1.5653,  13.9377,   4.1773],
        [  1.5662,  13.9375,   4.5142],
        [  3.4401,  13.4540,   4.6576],
        [  0.5175,  14.0910,   4.6540],
        [ -1.7009,   7.8851,  -8.2437],
        [ 11.1085,  -4.9204,   0.0736],
        [ -3.5029,   7.5138,   8.1853],
        [  4.6231,   5.4556,   8.4973],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -8.7409,  -4.6837,  -6.5840],
        [  4.9531,   5.0576,   5.6794],
        [  0.1497,   7.8334,  -5.4412],
        [  0.1382, -11.0073,   1.0026],
        [ -9.4159,  -2.0739,   9.6841],
        [ 12.3609,   3.4769,  -8.5762],
        [  3.5939,   6.4029,  -1.8357],
        [ 12.2435,   4.0076,   8.1257],
        [  6.2804,   2.0019,   0.8607],
        [ -5.2752,  -9.0409,  -8.2568],
        [  0.1041, -11.0045,   2.4311],
        [  3.8182,  -3.9347,  -1.7961],
        [ -6.4844,  -8.0230,   8.5209],
        [ 11.1085,  -4.9204,   0.0736],
        [ -2.5161,   7.7699,  -9.3531],
        [  4.7739,  -3.1218,  -4.1188],
        [ -9.4541,  -0.3327,   8.8246],
        [  1.9505,  -4.7171,   9.9868],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.4592,  -0.9168,  -6.7571],
        [  6.3542,   0.6934,  -3.3840],
        [ 11.7795,  -3.4697,   3.5624],
        [ -9.4753,  -0.8012,  -5.2050],
        [  5.4732,  -2.2041,  -4.3830],
        [  3.4140,  -4.1803,  -8.8224],
        [  8.5716,  10.3596,   6.8818],
        [  8.8474,  10.0868,  -8.8644],
        [  1.7228,  -4.7575,   8.3396],
        [  4.1140,  -3.7220,  -0.1273],
        [ -5.6174,   6.4708,  -1.7339],
        [ -9.4505,  -1.7081,  -0.7039],
        [  0.9967,  -4.8144,  -9.7370],
        [  5.5424,  -2.0885,   7.2487],
        [ -2.6485, -10.4151,   0.8766],
        [ -9.3810,  -2.3455,  -9.5194],
        [ -3.5173, -10.0623,   4.8280],
        [  4.0591,   6.0202,  -1.6339],
        [ -8.6603,   2.8250,   5.9181],
        [  2.4149,  -4.6005,  -4.3339],
        [  6.3124,   0.2057,  -0.4593],
        [ -8.1426,   3.8048,   9.3542],
        [ 11.7779,   5.5127,  -2.0699],
        [  2.3490,  -4.6199,  -4.2489],
        [  5.8323,  -1.5228,   5.1132],
        [ -6.8618,  -7.6322,   3.7027],
        [  4.5767,   5.5071,   3.9069],
        [ -6.4844,  -8.0230,   8.5209],
        [  2.5010,  -4.5736,  -0.8989],
        [  3.1892, -10.8220,  -6.8409],
        [ 11.5456,   6.0861,   8.1101],
        [  1.2728, -11.0372,   9.8121],
        [  2.5829,  -4.5466,  -8.4015],
        [  6.3516,   0.6425,   0.3425],
        [  3.6154,  -4.0640,  -7.9698],
        [ -3.0644,   7.6439,  -1.4322],
        [  2.4572,   7.1085,   7.7376],
        [  2.5375,   7.0676,  -4.3490],
        [  0.1206,  -4.7363,   8.7103],
        [  3.3474,  -4.2162,   5.2455],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.7565,  11.8237,  -4.9882],
        [  3.3474,  -4.2162,   5.2455],
        [ -9.4541,  -0.3327,   8.8246],
        [ -2.5056,   7.7720,   5.5878],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.0729,   7.2878,   7.0880],
        [ -9.1795,  -3.3623,   8.5259],
        [  1.6954,  -4.7616,   9.8198],
        [ 12.4996,   2.6335,  -6.3166],
        [ -5.6792,  -8.7352,   4.0318],
        [ -9.2373,  -3.1225,  -8.9763],
        [  3.3633,  13.4794,   0.5193],
        [  2.8890,  -4.4316,   1.1600],
        [ 10.9931,   7.2107,   7.7933],
        [ -8.0534,  -6.0429,   2.4583],
        [  6.1755,   2.5233,  -2.0042],
        [ -9.2345,  -3.1345,   3.9715],
        [  9.3424,  -7.3685,  -3.6807],
        [ 12.2218,   4.0953,  -8.2072],
        [  8.5198,  10.4091,  -8.6486],
        [ 11.8951,  -3.1570,   7.0742],
        [ 11.2767,  -4.6006,   0.7868],
        [ -5.2589,   6.7024,  -5.5809],
        [  2.6970,   6.9826,   9.7350],
        [ -8.5607,  -5.0938,  -5.4531],
        [  6.3124,   0.2057,  -0.4593],
        [  0.4996,   7.7738,  -4.4711],
        [ -1.7009,   7.8851,  -8.2437],
        [  9.8082,  -6.8396,   4.9615],
        [ -8.9788,   2.0208,  -3.2867],
        [  8.6420,  -8.0580,   6.5781],
        [  9.4516,  -7.2501,  -1.8868],
        [ -4.4520,   7.1370,  -8.7867],
        [  0.9788,  -4.8145,  -3.8661],
        [ -9.3411,  -2.6004,  -2.7062]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0391, 0.1385,  ..., 1.1724, 0.7583, 1.4297],
        [0.0391, 0.0000, 0.1258,  ..., 1.1518, 0.7432, 1.3979],
        [0.1385, 0.1258, 0.0000,  ..., 1.2490, 0.8526, 1.4565],
        ...,
        [1.1724, 1.1518, 1.2490,  ..., 0.0000, 0.4361, 0.5644],
        [0.7583, 0.7432, 0.8526,  ..., 0.4361, 0.0000, 0.8687],
        [1.4297, 1.3979, 1.4565,  ..., 0.5644, 0.8687, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.72594022, 13.92390689, 14.02532758, 14.02526505, 13.88683436,
       14.10045797,  8.06643487, 12.1494023 ,  8.29022197,  7.15100303,
        9.55720651,  9.91668182,  7.07902036,  7.83486943, 11.00812996,
        9.64157461, 12.84056604,  7.34250948, 12.88270018,  6.59176018,
       10.46739073, 11.0050379 ,  5.48276038, 10.31584102, 12.1494023 ,
        8.16714836,  5.70405525,  9.45995955,  5.10446653, 13.13205357,
       12.49291768,  6.39187285, 12.27991991,  9.50913849,  5.90035324,
        5.39723241, 13.44593519, 13.41713661,  5.05981914,  5.54776934,
        8.56889738,  9.60359158,  4.91652556,  5.92281142, 10.74655346,
        9.66977802, 10.65929968,  7.26074925,  9.10945678,  5.19577146,
        6.31576264,  8.98765644, 13.00413829,  5.18277858,  6.02779201,
       10.26327895,  7.16058587, 10.31584102,  5.21280156, 11.28215429,
       13.05148826, 11.11038551,  5.22903117,  6.38399789,  5.43943791,
        8.23525433,  7.52117132,  7.5092834 ,  4.73784222,  5.38342077,
        9.99211663, 13.61801526,  5.38342077,  9.45995955,  8.16584537,
       10.32715715,  7.57685908,  9.77588801,  5.05444288, 12.77402351,
       10.41909211,  9.75075273, 13.89264822,  5.29009836, 13.14690137,
       10.06850677,  6.67108662,  9.75200848, 11.89855267, 12.889695  ,
       13.45124917, 12.30695098, 12.17901033,  8.51928531,  7.48538905,
        9.96152788,  6.31576264,  7.78980299,  8.06643487, 11.95743558,
        9.20340031, 11.8159254 , 11.9120321 ,  8.41171369,  4.91296205,
        9.69629   ])}
{'P': tensor([[2.1616e-03, 2.8921e-01, 2.1403e-01,  ..., 1.4523e-08, 7.8558e-09,
         4.7248e-06],
        [4.6336e-06, 3.3031e-03, 1.3846e-04,  ..., 2.8227e-08, 9.2415e-09,
         3.9551e-06],
        [6.9669e-06, 2.6809e-02, 2.7037e-03,  ..., 1.8794e-08, 7.0360e-09,
         3.1721e-06],
        ...,
        [4.8666e-06, 3.2195e-06, 3.0493e-06,  ..., 1.8213e-01, 3.5176e-02,
         4.4736e-04],
        [1.0528e-05, 5.9122e-06, 5.3410e-06,  ..., 4.0754e-07, 5.4996e-03,
         4.5122e-04],
        [2.3455e-03, 2.8993e-03, 1.9852e-03,  ..., 1.3868e-05, 2.2164e-05,
         6.3319e-02]]), 'X': tensor([[  5.2608,  12.6977,   4.9166],
        [  3.4401,  13.4540,   4.6576],
        [  3.9022,  13.2906,   5.5788],
        [  2.2182,  13.8007,   5.1889],
        [  2.3261,  13.7749,   5.9121],
        [  1.5662,  13.9375,   4.5142],
        [ 12.2218,   4.0953,  -8.2072],
        [ -0.8675, -10.8815,   6.8561],
        [  6.1737,   2.5305,   3.9049],
        [ 12.0772,  -2.6014,  -7.1832],
        [ -9.3444,   0.5821,   6.9101],
        [ -9.2345,  -3.1345,   3.9715],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -1.8367,   7.8718,  -8.6523],
        [ 11.0275,  -5.0666,   8.4154],
        [ -7.4602,   4.7718,  -4.6277],
        [  2.5974,   7.0363,  -7.7810],
        [ 11.9641,  -2.9568,   7.7211],
        [ -8.9788,   2.0208,  -3.2867],
        [ 11.3118,   6.5957,  -4.5667],
        [ -8.9624,  -4.0944,  -4.9622],
        [ -8.7078,   2.7182,   1.8146],
        [ -3.7964,  -9.9294,   7.7147],
        [ -8.0534,  -6.0429,   2.4583],
        [  8.7715,  -7.9390,   5.2900],
        [  2.1477,   7.2550,   8.9688],
        [ -7.4602,   4.7718,  -4.6277],
        [  6.3301,   0.3638,  -7.5685],
        [ -8.3535,  -5.5118,   9.8399],
        [  2.0729,   7.2878,   7.0880],
        [ 11.0864,   7.0387,  -0.1238],
        [  6.3587,   1.1596,  -3.6629],
        [ -1.6220,   7.8917,   0.0315],
        [ -2.4956, -10.4682,   3.5632],
        [ -9.1658,  -3.4155,  -2.8880],
        [  3.7298,  13.3537,  -9.4653],
        [  1.7000,  -4.7609,   2.7935],
        [ -4.7642,  -9.3860,  -5.1133],
        [  5.9587,  12.3216,   9.1695],
        [  1.7815,  -4.7481,   4.4289],
        [ -1.6334, -10.7207,   7.1397],
        [  4.0362,  13.2397,  -7.0030],
        [ -8.0829,  -5.9940,   5.0564],
        [  7.9403,  -8.6455,  -8.7433],
        [  0.1287, -11.0065,   6.5757],
        [  3.9515,  -3.8425,   6.1729],
        [  3.6469,   6.3624,  -1.8414],
        [ -6.3489,   5.9117,   6.5161],
        [ -9.4157,   0.0821,  -5.0025],
        [ -8.5859,  -5.0393,   0.5476],
        [ 12.3453,  -1.5462,   7.0212],
        [ -8.7544,   2.6096,   3.3146],
        [ 10.4994,  -5.9193,  -7.5153],
        [  2.5375,   7.0676,  -4.3490],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.6038,   0.7665,  -3.4439],
        [ 12.6038,   0.7665,  -3.4439],
        [  4.0591,   6.0202,  -1.6339],
        [  5.8979,   3.3958,  -3.8713],
        [ -7.4887,   4.7364,   8.9074],
        [  1.7659,   7.4127,   4.7871],
        [ -3.9700,   7.3453,  -4.1486],
        [  4.3377,   5.7570,   1.3712],
        [ 12.6059,   0.9597,  -6.3264],
        [  8.8091,  -7.9037,   0.3631],
        [  6.1737,   2.5305,   3.9049],
        [ 12.5770,   1.8512,  -0.6578],
        [  0.1382, -11.0073,   1.0026],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.6603,   2.8250,   5.9181],
        [  0.6470, -11.0352,  -1.1931],
        [  2.9470,  13.6085,   2.9809],
        [  5.9003,  12.3550,   2.1917],
        [  6.3301,   0.3638,  -7.5685],
        [ -0.2586,   7.8825,  -9.9840],
        [ -2.3317,   7.8034,  -3.6274],
        [  9.2979,   9.6048,   7.3138],
        [ 11.0373,   7.1299,  -8.5895],
        [  5.6672,  -1.8631,   8.1020],
        [ -3.7964,  -9.9294,   7.7147],
        [ 12.3025,  -1.7440,  -6.6632],
        [ -9.4505,  -1.7081,  -0.7039],
        [ 11.7726,  -3.4876,  -6.7402],
        [ -5.7347,  -8.6908,   9.2878],
        [  9.0129,  -7.7072,   9.1400],
        [ 12.5919,   0.4035,  -5.8460],
        [ -5.0550,   6.8231,  -9.2502],
        [  9.3424,  -7.3685,  -3.6807],
        [  2.6451,  -4.5250,  -0.5565],
        [ -3.0008, -10.2827,  -8.9480],
        [  5.2939,  -2.4784,  -4.5840],
        [ -8.4887,  -5.2448,  -0.6585],
        [  5.4732,  -2.2041,  -4.3830],
        [ -8.6603,   2.8250,   5.9181],
        [  9.6614,   9.1784,  -1.2087],
        [  1.8721,  -4.7322,   0.5524],
        [ -8.0588,  -6.0341,  -2.3006],
        [  4.1297,  -3.7098,  -3.9838],
        [  3.9913,  -3.8139,  -6.8311],
        [  1.2728, -11.0372,   9.8121],
        [ -9.4731,  -0.7120,  -6.0271],
        [ 11.3691,   6.4760,   1.0787],
        [  2.1120,  -4.6818,   8.6144],
        [ -1.0482, -10.8487,  -5.9066],
        [ 10.9162,  -5.2599,  -7.0812],
        [  5.0233,   4.9647,  -3.1788]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 4.8592e-02, 3.5219e-02,  ..., 1.5051e+00, 1.2011e+00,
         8.9880e-01],
        [4.8592e-02, 3.0339e-04, 8.2767e-02,  ..., 1.4630e+00, 1.1526e+00,
         8.5088e-01],
        [3.5219e-02, 8.2767e-02, 0.0000e+00,  ..., 1.5289e+00, 1.2340e+00,
         9.3015e-01],
        ...,
        [1.5051e+00, 1.4630e+00, 1.5289e+00,  ..., 0.0000e+00, 6.9167e-01,
         7.3206e-01],
        [1.2011e+00, 1.1526e+00, 1.2340e+00,  ..., 6.9167e-01, 0.0000e+00,
         3.2382e-01],
        [8.9880e-01, 8.5088e-01, 9.3015e-01,  ..., 7.3206e-01, 3.2382e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.744387  , 13.88683436, 13.85158417, 13.97779839, 13.96987806,
       14.02526505, 12.889695  , 10.91601655,  6.67218687, 12.35421461,
        9.36256096,  9.75200848, 12.18045978,  8.08320235, 12.13568598,
        8.85572292,  7.50035355, 12.32408516,  9.20340031, 13.09426142,
        9.85330646,  9.12220271, 10.63038565, 10.06850677, 11.83075005,
        7.56617037,  8.85572292,  6.34059021, 10.00802627,  7.57685908,
       13.13205357,  6.46356806,  8.05668609, 10.76154256,  9.78147542,
       13.86480175,  5.05534609, 10.52587243, 13.68672855,  5.0713364 ,
       10.84437705, 13.84126197, 10.06286074, 11.73848065, 11.00726483,
        5.51177501,  7.33350642,  8.67502348,  9.41605684,  9.95552626,
       12.44177641,  9.13507321, 12.05300943,  7.5092834 , 13.13205357,
       12.62711252, 12.62711252,  7.26074925,  6.80559669,  8.86082124,
        7.62011734,  8.34949383,  7.20826219, 12.64235359, 11.83509372,
        6.67218687, 12.7125079 , 11.00812996, 12.52875793,  9.10945678,
       11.05413545, 13.92390689, 13.69162795,  6.34059021,  7.88677538,
        8.14433542, 13.36800073, 13.1399181 ,  5.96556248, 10.63038565,
       12.42554808,  9.60359158, 12.27835996, 10.412299  , 11.85890664,
       12.59840193,  8.49161154, 11.89855267,  5.24137909, 10.71162681,
        5.84533345,  9.97822473,  5.90035324,  9.10945678, 13.32613445,
        5.08910491, 10.0674887 ,  5.5513072 ,  5.5205189 , 11.11038551,
        9.4997987 , 13.08414689,  5.13616224, 10.8992524 , 12.11735334,
        7.06272256])}
{'P': tensor([[2.6124e-03, 2.1079e-01, 1.6653e-01,  ..., 3.8358e-06, 1.7120e-08,
         4.4642e-07],
        [2.5231e-06, 3.2924e-03, 5.0108e-05,  ..., 1.2992e-05, 2.8877e-08,
         5.5725e-07],
        [1.7933e-05, 2.2553e-01, 1.1019e-02,  ..., 3.4319e-05, 7.5344e-08,
         1.8826e-06],
        ...,
        [7.1122e-07, 7.1767e-07, 7.1415e-07,  ..., 4.1467e-02, 4.2219e-04,
         1.6339e-03],
        [3.0482e-06, 1.3265e-06, 1.3594e-06,  ..., 3.4538e-06, 6.0276e-02,
         6.5233e-04],
        [1.4865e-04, 7.5518e-05, 9.1321e-05,  ..., 9.9410e-05, 2.1709e-04,
         8.0029e-02]]), 'X': tensor([[  5.0617,  12.7957,   8.3544],
        [  1.9312,  13.8649,   8.3563],
        [  2.7590,  13.6621,   9.4496],
        [  1.3705,  13.9723,   7.9266],
        [  1.1274,  14.0116,   8.1753],
        [  3.1064,  13.5607,   6.7605],
        [  2.1771, -10.9776,  -3.2007],
        [ 12.6038,   0.7665,  -3.4439],
        [  6.2242,  -0.3171,   7.4978],
        [  5.3860,  -2.3418,   1.0826],
        [  6.1752,   2.5242,   3.5874],
        [  3.2226,  -4.2802,  -4.0428],
        [ 12.4691,   2.8525,  -5.5110],
        [  3.4140,  -4.1803,  -8.8224],
        [  5.9026,   3.3837,  -0.6573],
        [ 10.9916,   7.2133,  -0.9179],
        [  1.1994,   7.6035,   2.4295],
        [ -7.6525,  -6.6534,   5.1721],
        [  0.1382, -11.0073,   1.0026],
        [  6.3606,   0.9204,   3.2525],
        [ 12.0772,  -2.6014,  -7.1832],
        [  6.1730,  -0.5375,  -8.8861],
        [  9.9217,  -6.7006,  -5.6728],
        [  5.1196,  -2.7160,   4.2705],
        [ -9.1960,  -3.2964,   1.6921],
        [  5.9843,  -9.8679,  -6.1449],
        [  1.5662,  13.9375,   4.5142],
        [  3.9663,   6.1018,  -5.4104],
        [  6.2804,   2.0019,   0.8607],
        [ -1.3940, -10.7772,   6.7835],
        [  4.5954,  13.0099,  -4.4439],
        [  8.5093,  -8.1764,  -3.2381],
        [ -9.1795,  -3.3623,   8.5259],
        [ 11.0275,  -5.0666,   8.4154],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.2770,  -0.0379,   2.4195],
        [ -6.7992,   5.4991,   9.9186],
        [  8.1217,  -8.5024,  -0.6149],
        [ -3.4100,   7.5436,  -7.6980],
        [  1.2273,  -4.8081,   9.8129],
        [ 11.2209,  -4.7092,   2.6117],
        [ -8.3539,  -5.5111,   0.3373],
        [  5.2608,  -2.5256,  -9.1446],
        [  3.9515,  -3.8425,   6.1729],
        [  5.7963,  -1.6018,   1.8856],
        [  8.1043,  -8.5164,  -7.9290],
        [ -6.2668,   5.9808,   2.5614],
        [  3.4009,  -4.1874,  -5.9751],
        [ -6.6895,   5.6051,   2.7899],
        [  2.7590,  13.6621,   9.4496],
        [ -8.1666,   3.7653,   3.6075],
        [ -1.8788, -10.6567,   3.7068],
        [ -1.0012, -10.8576,   2.2867],
        [  6.8704,  -9.3809,   7.5769],
        [ -6.2668,   5.9808,   2.5614],
        [  6.3305,   1.6119,   1.1538],
        [ -3.5173, -10.0623,   4.8280],
        [  1.0391,  -4.8141,  -0.0768],
        [  2.1625,  -4.6697,   7.0658],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.1231,  -3.5739,   8.4708],
        [  3.6154,  -4.0640,  -7.9698],
        [ -8.4824,  -5.2575,  -0.0795],
        [  6.3413,   0.4883,  -6.8416],
        [  1.3767, -11.0341,   9.9415],
        [  3.2701,   6.6338,   9.9209],
        [ 11.7418,   5.6070,  -1.4192],
        [ 11.5966,   5.9666,   6.0083],
        [  6.3236,   0.3011,  -4.6668],
        [  5.9883,   3.1536,   6.2320],
        [  0.6363, -11.0348,   6.1414],
        [  9.9217,  -6.7006,  -5.6728],
        [ 11.7796,  -3.4696,   9.0088],
        [  3.2226,  -4.2802,  -4.0428],
        [  4.3459,  -3.5327,   0.3667],
        [  8.7131,  10.2216,   7.7915],
        [  0.7275,  14.0667,  -0.9815],
        [ -4.2040,   7.2486,   2.9875],
        [  4.5767,   5.5071,   3.9069],
        [  6.1741,  -0.5329,   3.1913],
        [  2.4572,   7.1085,   7.7376],
        [  2.2644,  -4.6434,   9.7508],
        [ -8.9033,   2.2329,  -8.8912],
        [  8.7715,  -7.9390,   5.2900],
        [  8.9284,  10.0036,  -6.9210],
        [ -0.0254, -10.9932,   3.2200],
        [  6.3505,   0.6235,  -8.5849],
        [  5.1886,  -2.6250,   6.4677],
        [  0.3231,  -4.7691,  -1.6120],
        [  6.3119,   1.7797,   9.9186],
        [  8.3718,  -8.2953,   8.4666],
        [ -8.8154,   2.4611,   2.5198],
        [  0.3354,  -4.7708,  -4.9125],
        [  2.3261,  13.7749,   5.9121],
        [ -5.1046,  -9.1610,   2.3418],
        [  0.7145,  -4.8071,  -2.3766],
        [ 11.9936,  -2.8677,   6.0614],
        [  5.9526,   3.2525,  -4.3821],
        [  2.5813,  -4.5471,   5.3597],
        [  7.3592,  11.3937,   6.2407],
        [ -6.2668,   5.9808,   2.5614],
        [ 10.0619,   8.6627,  -9.2884],
        [  6.2242,  -0.3171,   7.4978],
        [ -9.4159,  -2.0739,   9.6841],
        [  0.5600, -11.0321,  -0.8660],
        [  2.0870,  -4.6877,   9.6823]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0810, 0.0933,  ..., 1.2755, 1.3559, 0.4660],
        [0.0810, 0.0000, 0.0126,  ..., 1.1945, 1.2867, 0.3925],
        [0.0933, 0.0126, 0.0000,  ..., 1.1822, 1.2781, 0.3832],
        ...,
        [1.2755, 1.1945, 1.1822,  ..., 0.0000, 0.7284, 0.8721],
        [1.3559, 1.2867, 1.2781,  ..., 0.7284, 0.0000, 0.8962],
        [0.4660, 0.3925, 0.3832,  ..., 0.8721, 0.8962, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.76047999, 13.99877413, 13.93790235, 14.0393949 , 14.05687815,
       13.91198191, 11.1913598 , 12.62711252,  6.2322769 ,  5.87305005,
        6.67121748,  5.35774573, 12.79126785,  5.39723241,  6.80370254,
       13.1471313 ,  7.69752262, 10.14044489, 11.00812996,  6.42689705,
       12.35421461,  6.19633139, 11.97238624,  5.79543038,  9.76896904,
       11.54071859, 14.02526505,  7.27759233,  6.59176018, 10.86693708,
       13.79762415, 11.80092578,  9.77588801, 12.13568598, 12.12448912,
        6.27715105,  8.74470218, 11.75807252,  8.27853076,  4.96231532,
       12.16901615, 10.00794468,  5.83560009,  5.51177501,  6.01356004,
       11.75618513,  8.66273171,  5.39450305,  8.7273569 , 13.93790235,
        8.99276564, 10.82106505, 10.9036226 , 11.62770332,  8.66273171,
        6.53250937, 10.65929968,  4.92497514,  5.14607014,  5.01114719,
        9.79814997,  5.43943791,  9.97964013,  6.36004393, 11.11969815,
        7.39599986, 13.01187484, 13.04156075,  6.33075682,  6.76791483,
       11.05317597, 11.97238624, 12.27993098,  5.35774573,  5.60064319,
       13.43127131, 14.08549802,  8.37954853,  7.16058587,  6.19707944,
        7.52117132,  5.16611782,  9.17905491, 11.83075005, 13.40850328,
       10.99326577,  6.38105015,  5.81481044,  4.78002665,  6.55800267,
       11.785566  ,  9.15253344,  4.78258092, 13.96987806, 10.48721382,
        4.85995002, 12.33167382,  6.78325151,  5.2287058 , 13.56367857,
        8.66273171, 13.27717982,  6.2322769 ,  9.64157461, 11.04629471,
        5.13125513])}
{'P': tensor([[1.5747e-03, 2.4775e-01, 2.1445e-01,  ..., 1.8636e-08, 2.2543e-07,
         2.5754e-07],
        [2.3335e-06, 1.5951e-03, 2.1880e-03,  ..., 2.1903e-08, 2.1585e-07,
         2.0778e-07],
        [3.2611e-06, 1.0891e-03, 2.0403e-03,  ..., 2.7620e-08, 2.6939e-07,
         2.7188e-07],
        ...,
        [4.7438e-06, 2.9363e-06, 2.8558e-06,  ..., 4.8060e-02, 1.4846e-02,
         1.9443e-03],
        [3.1270e-04, 2.1354e-04, 2.0523e-04,  ..., 1.0957e-03, 1.2768e-01,
         1.0819e-02],
        [1.3543e-03, 8.9044e-04, 9.2977e-04,  ..., 6.4137e-05, 1.1511e-03,
         9.5106e-02]]), 'X': tensor([[  4.9337,  12.8566,  -3.6582],
        [  3.3094,  13.4969,  -3.6507],
        [  3.2809,  13.5061,  -4.0199],
        [  2.3589,  13.7668,  -3.6501],
        [  1.4150,  13.9647,  -4.0915],
        [  2.0937,  13.8293,  -2.3533],
        [  1.6408,   7.4591,  -4.2294],
        [  1.7815,  -4.7481,   4.4289],
        [  2.3928,   7.1404,  -1.7035],
        [  1.0093,   7.6567,   2.3526],
        [  0.1497,   7.8334,  -5.4412],
        [  4.3554,   5.7393,  -9.0766],
        [  7.0664,  -9.2590,   8.0879],
        [  5.1519, -10.2387,   9.8191],
        [ -8.8363,   2.4086,  -5.6296],
        [  8.1528,  10.7452,   9.1493],
        [ -9.4748,  -0.7765,  -5.6695],
        [  3.4604,  -4.1545,   1.8091],
        [  6.3535,   0.6792,  -7.5101],
        [ -7.0120,   5.2825,   7.7762],
        [  5.1472,  -2.6801,   8.1194],
        [  1.3670,  -4.7990,   3.2603],
        [  9.9058,  -6.7203,   5.6928],
        [ -5.8389,  -8.6056,  -3.5098],
        [ -4.7847,  -9.3729,  -3.7741],
        [  4.9531,   5.0576,   5.6794],
        [ -7.9298,   4.1367,  -3.9773],
        [ 11.6544,   5.8273,  -3.1375],
        [  5.4429,  -2.2529,   9.7306],
        [ 12.5919,   0.4035,  -5.8460],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -9.1658,  -3.4155,  -2.8880],
        [ 10.0400,   8.6922,  -6.2847],
        [  2.7976,  -4.4682,  -4.2020],
        [  4.8051, -10.3703,   0.5268],
        [ 11.2939,   6.6325,  -5.8311],
        [ 10.2470,   8.4054,  -0.1177],
        [  6.0890,  -0.8382,   2.9166],
        [  5.8902,  -1.3887,   4.0388],
        [  4.0642,  -3.7599,  -3.4565],
        [ -4.0633,  -9.7927,  -3.0825],
        [  5.4167, -10.1293,   6.2156],
        [ 11.8951,  -3.1570,   7.0742],
        [  4.0078,  -3.8018,  -6.2168],
        [  7.0664,  -9.2590,   8.0879],
        [  6.3460,   1.4252,   1.2291],
        [ -3.9456,  -9.8542,   0.7466],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.4167, -10.1293,   6.2156],
        [ -0.1626, -10.9796,   6.4629],
        [ -5.0943,  -9.1682,   0.5073],
        [  8.3791,  10.5409,   5.5590],
        [  5.6672,  -1.8631,   8.1020],
        [ -8.4573,   3.2448,   5.4370],
        [ 11.6634,   5.8050,   9.1590],
        [ -6.4822,  -8.0252,  -3.8983],
        [  4.7156,  -3.1831,  -8.0224],
        [ 10.0619,   8.6627,  -9.2884],
        [  0.9945,  -4.8144,  -0.1897],
        [ -8.0330,  -6.0765,   6.7945],
        [  2.5636, -10.9293,   9.6519],
        [  2.2373,  -4.6506,  -1.7334],
        [  5.4179,  -2.2925,  -3.4641],
        [ -5.9938,  -8.4745,   3.9735],
        [  6.0305,   3.0296,   0.1375],
        [  2.5663,   7.0526,  -6.8769],
        [ -9.3739,   0.3982,  -4.2564],
        [  0.6396,  14.0772,  -6.3786],
        [ 12.6038,   0.7665,  -3.4439],
        [ 10.0619,   8.6627,  -9.2884],
        [ 12.2692,  -1.8882,   3.9253],
        [  9.6662,   9.1725,   8.1056],
        [ 12.2589,  -1.9312,   9.8657],
        [ 12.5154,   2.5072,  -9.3274],
        [ -5.9987,   6.1949,  -6.2114],
        [ -9.3087,   0.7792,  -7.5587],
        [  1.3767, -11.0341,   9.9415],
        [ 12.0582,  -2.6640,   0.7132],
        [  5.2608,  12.6977,   4.9166],
        [ 12.5650,   2.0132,   4.6272],
        [ -1.5408,   7.8978,   5.3470],
        [  3.4401,  13.4540,   4.6576],
        [ -8.1204,  -5.9308,   4.0761],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -4.4779,  -9.5609,  -2.4505],
        [ 12.1751,  -2.2592,  -5.7751],
        [  5.7151,  -1.7700,   3.1542],
        [  0.3715,  -4.7756,   0.2312],
        [  5.9912,   3.1453,  -3.9989],
        [  1.1644, -11.0394,   7.8511],
        [  4.4049,   5.6892,  -4.4394],
        [  2.3589,  13.7668,  -3.6501],
        [  5.0348,  -2.8229,   3.5436],
        [  6.2247,  -0.3150,  -9.3657],
        [ 11.6544,   5.8273,  -3.1375],
        [  3.8111, -10.6787,  -4.6539],
        [  0.9074,  -4.8139,   8.1674],
        [  7.0664,  -9.2590,   8.0879],
        [ -7.9667,   4.0814,  -0.6397],
        [ 11.5456,   6.0861,   8.1101],
        [ 11.0864,   7.0387,  -0.1238],
        [ 12.6059,   0.9597,  -6.3264],
        [  7.1697,  11.5343,  -0.7276],
        [  0.5600, -11.0321,  -0.8660],
        [  2.9461,  -4.4077,  -0.4831],
        [  5.6095,  -1.9702,  -7.8233]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0114, 0.0338,  ..., 0.9507, 0.4348, 0.2525],
        [0.0114, 0.0000, 0.0240,  ..., 0.9446, 0.4339, 0.2424],
        [0.0338, 0.0240, 0.0000,  ..., 0.9472, 0.4476, 0.2341],
        ...,
        [0.9507, 0.9446, 0.9472,  ..., 0.0000, 0.5980, 0.7408],
        [0.4348, 0.4339, 0.4476,  ..., 0.5980, 0.0000, 0.3627],
        [0.2525, 0.2424, 0.2341,  ..., 0.7408, 0.3627, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.77075174, 13.89671314, 13.89886469, 13.96746816, 14.03618649,
       13.98691426,  7.6374603 ,  5.0713364 ,  7.53063885,  7.72291946,
        7.83486943,  7.20482224, 11.64746882, 11.46176259,  9.15866607,
       13.48809595,  9.50654967,  5.40689332,  6.38968357,  8.77913428,
        5.80312949,  4.98989906, 11.9702762 , 10.39942497, 10.52357479,
        7.07902036,  8.94395628, 13.03002338,  5.89074104, 12.59840193,
       10.30972193,  9.78147542, 13.27994304,  5.27178363, 11.42947632,
       13.09737053, 13.25336001,  6.1463896 ,  6.05165248,  5.5366698 ,
       10.60226614, 11.48663767, 12.30695098,  5.52416368, 11.64746882,
        6.50409372, 10.61472636, 11.54672755, 11.48663767, 10.98076873,
       10.48840476, 13.46553623,  5.96556248,  9.05843201, 13.02818702,
       10.31613269,  5.68942572, 13.27717982,  4.9160962 , 10.07238463,
       11.22596994,  5.16077764,  5.8828986 , 10.37998078,  6.74872981,
        7.50500064,  9.38232121, 14.09176259, 12.62711252, 13.27717982,
       12.41366754, 13.32557133, 12.41012046, 12.76408562,  8.62329466,
        9.34127013, 11.11969815, 12.3489324 , 13.744387  , 12.72524615,
        8.04664886, 13.88683436, 10.05558893,  9.4997987 , 10.5575601 ,
       12.3828957 ,  5.98285319,  4.79003361,  6.76661595, 11.10066126,
        7.19513806, 13.96746816,  5.77219077,  6.23262289, 13.03002338,
       11.33837396,  4.89869205, 11.64746882,  8.95133961, 13.05148826,
       13.13205357, 12.64235359, 13.58101794, 11.04629471,  5.30158497,
        5.94541803])}
{'P': tensor([[2.1516e-03, 2.1711e-01, 2.1059e-01,  ..., 1.9664e-07, 1.9760e-08,
         1.4080e-06],
        [9.0694e-06, 2.0274e-03, 2.5302e-01,  ..., 1.5873e-07, 1.5524e-08,
         1.0129e-06],
        [9.5980e-06, 1.4457e-05, 1.0475e-02,  ..., 7.2614e-07, 5.8815e-08,
         3.2680e-06],
        ...,
        [5.6657e-04, 4.2515e-04, 3.0203e-04,  ..., 1.4851e-01, 1.6343e-02,
         8.9226e-02],
        [1.4981e-03, 9.9084e-04, 4.6475e-04,  ..., 1.2607e-03, 7.3558e-02,
         3.1910e-02],
        [8.0158e-03, 6.3054e-03, 5.1169e-03,  ..., 9.3639e-04, 2.2733e-04,
         9.9320e-02]]), 'X': tensor([[  4.9152,  12.8653,  -1.4459],
        [  4.1637,  13.1898,  -1.9189],
        [  2.2076,  13.8032,  -1.5099],
        [  2.0937,  13.8293,  -2.3533],
        [  1.6308,  13.9254,  -2.0984],
        [  2.3373,  13.7721,  -0.2442],
        [ -9.4756,  -0.8166,  -9.6021],
        [  9.9058,  -6.7203,   5.6928],
        [  2.5620,   7.0549,   8.4825],
        [ 10.1798,   8.5002,  -5.8893],
        [ -8.0330,  -6.0765,   6.7945],
        [  1.6308,  13.9254,  -2.0984],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.3333,   4.9246,  -7.9212],
        [  3.1756, -10.8247,   6.8882],
        [  5.7485,  -9.9811,  -3.2355],
        [ -6.4464,   5.8272,   7.5038],
        [  4.6231,   5.4556,   8.4973],
        [ -0.5709,   7.9055,   7.6089],
        [ -3.3999, -10.1152,  -1.2088],
        [  8.0381,  -8.5690,  -8.9523],
        [ -9.1433,  -3.5003,  -3.0209],
        [  6.3460,   1.4252,   1.2291],
        [ 10.3251,   8.2928,  -3.0006],
        [  6.3606,   0.9204,   3.2525],
        [  9.2811,  -7.4336,   5.1771],
        [ -6.3489,   5.9117,   6.5161],
        [  5.8902,  -1.3887,   4.0388],
        [ -8.6681,  -4.8556,   9.2518],
        [ -8.6801,   2.7809,  -0.8443],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.8474,   6.8979,   6.6457],
        [ -5.9940,   6.1985,   5.9566],
        [  5.6414, -10.0303,   7.8525],
        [  8.1528,  10.7452,   9.1493],
        [  8.3791,  10.5409,   5.5590],
        [  1.5246,  -4.7838,  -5.2095],
        [  9.3624,   9.5318,   0.2779],
        [ -8.6426,  -4.9136,   9.4941],
        [ -2.1533, -10.5778,   1.0112],
        [  5.0190,   4.9705,  -7.9752],
        [ 12.2592,   3.9425,   2.0383],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.9843,  -9.8679,  -6.1449],
        [ -3.2736, -10.1703,   3.8712],
        [ 11.9641,  -2.9568,   7.7211],
        [  1.2273,  -4.8081,   9.8129],
        [  7.5064,  -8.9651,  -0.0994],
        [  6.7565,  11.8237,  -4.9882],
        [  2.0547,   7.2957,  -1.9343],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.2630,   2.1057,  -5.2136],
        [  3.8182,  -3.9347,  -1.7961],
        [  2.4572,   7.1085,   7.7376],
        [  4.2096,   5.8815,   1.9230],
        [  1.5653,  13.9377,   4.1773],
        [ -9.3419,   0.5972,  -1.2841],
        [ -0.8885,   7.9160,   0.0571],
        [ -5.1377,   6.7751,  -0.4438],
        [ -2.1382,   7.8339,   7.3524],
        [ -9.3107,  -2.7697,  -3.2919],
        [  7.5309,  11.2617,   7.2656],
        [  0.5175,  14.0910,   4.6540],
        [  4.5325,  -3.3643,   6.7699],
        [ -7.6734,   4.4980,  -5.4248],
        [  6.3205,   1.7068,  -3.8619],
        [ -5.3736,   6.6310,  -1.5174],
        [  7.5784,  -8.9141,   5.8410],
        [ -9.1907,  -3.3180,  -5.5943],
        [  5.4732,  -2.2041,  -4.3830],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.7544,   2.6096,   3.3146],
        [ -6.5295,  -7.9785,  -7.4429],
        [ 12.3453,  -1.5462,   7.0212],
        [  1.7815,  -4.7481,   4.4289],
        [ -6.3489,   5.9117,   6.5161],
        [  3.9022,  13.2906,   5.5788],
        [ -8.8540,  -4.3971,  -5.2144],
        [  5.6095,  -1.9702,  -7.8233],
        [  3.3474,  -4.2162,   5.2455],
        [ -0.5739,   7.9056,   0.6382],
        [  5.1890,   4.7318,   0.9515],
        [ 12.5763,   0.1333,  -9.2650],
        [  4.7813,  -3.1139,   8.2691],
        [ 10.5678,   7.9261,  -0.6100],
        [ 12.5763,   0.1333,  -9.2650],
        [  8.8474,  10.0868,  -8.8644],
        [  4.6357,   5.4415,   7.9067],
        [  3.9770, -10.6341,   8.0783],
        [  4.7156,  -3.1831,  -8.0224],
        [ -9.3411,  -2.6004,  -2.7062],
        [ -8.6675,  -4.8569,  -9.1769],
        [ -0.8675, -10.8815,   6.8561],
        [  8.3718,  -8.2953,   8.4666],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.3119,   1.7797,   9.9186],
        [  3.4604,  -4.1545,   1.8091],
        [ -9.1749,  -3.3803,   6.4629],
        [  3.7298,  13.3537,  -9.4653],
        [ -1.0482, -10.8487,  -5.9066],
        [ -8.0611,  -6.0302,   4.6962],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.1231,  -3.5739,   8.4708],
        [  4.9957,  -2.8705,   6.5978],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.1737,   2.5305,   3.9049]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0325, 0.0741,  ..., 0.6287, 0.5249, 0.4969],
        [0.0325, 0.0000, 0.0697,  ..., 0.6507, 0.5437, 0.5162],
        [0.0741, 0.0697, 0.0000,  ..., 0.5974, 0.4856, 0.4590],
        ...,
        [0.6287, 0.6507, 0.5974,  ..., 0.0000, 0.1277, 0.1458],
        [0.5249, 0.5437, 0.4856,  ..., 0.1277, 0.0000, 0.0295],
        [0.4969, 0.5162, 0.4590,  ..., 0.1458, 0.0295, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.77222752, 13.83139024, 13.97857591, 13.98691426, 14.02058621,
       13.96905756,  9.51074763, 11.9702762 ,  7.50564223, 13.26209545,
       10.07238463, 14.02058621,  9.7712321 ,  8.83341608, 11.28093507,
       11.51811603,  8.68979052,  7.15100303,  7.92607342, 10.67132524,
       11.74901778,  9.79039371,  6.50409372, 13.2430377 ,  6.42689705,
       11.89105259,  8.67502348,  6.05165248,  9.9353921 ,  9.11473391,
        9.85384623,  7.46248947,  8.62260327, 11.50791859, 13.48809595,
       13.46553623,  5.0209005 , 13.3607379 ,  9.94173708, 10.794755  ,
        7.06372172, 12.87752052, 10.24557047, 11.54071859, 10.68416401,
       12.32408516,  4.96231532, 11.69264243, 13.61801526,  7.57946459,
        5.27178363,  6.60752373,  5.48276038,  7.52117132,  7.23276582,
       14.02532758,  9.36093562,  7.96575782,  8.50278896,  8.12043983,
        9.71392179, 13.54774355, 14.10045797,  5.64464807,  8.89458273,
        6.54693987,  8.53501777, 11.70016217,  9.7712321 ,  5.90035324,
        6.652742  ,  9.13507321, 10.30972193, 12.44177641,  5.0713364 ,
        8.67502348, 13.85158417,  9.88571713,  5.94541803,  5.38342077,
        7.92644746,  7.02253235, 12.5769659 ,  5.70591986, 13.20988996,
       12.5769659 , 13.41713661,  7.14837514, 11.35345345,  5.68942572,
        9.69629   ,  9.93553424, 10.91601655, 11.785566  ,  5.43544626,
        6.55800267,  5.40689332,  9.77777346, 13.86480175, 10.8992524 ,
       10.06703358,  6.49538841,  9.79814997,  5.76165934, 12.27991991,
        6.67218687])}
{'P': tensor([[2.0234e-03, 3.0095e-01, 2.1515e-01,  ..., 8.4791e-08, 1.2508e-05,
         5.6703e-07],
        [3.5300e-06, 2.5651e-03, 2.5232e-01,  ..., 1.1385e-07, 1.2893e-05,
         5.1473e-07],
        [7.5900e-06, 3.7234e-05, 6.6772e-03,  ..., 3.3342e-07, 3.6438e-05,
         1.2091e-06],
        ...,
        [1.0884e-05, 8.4573e-06, 8.6562e-06,  ..., 1.0330e-01, 4.4838e-04,
         2.2800e-04],
        [4.9036e-05, 7.9502e-05, 1.0490e-04,  ..., 6.3228e-04, 1.2061e-01,
         2.1506e-05],
        [1.2075e-02, 7.5929e-03, 6.4363e-03,  ..., 1.1809e-04, 4.4311e-04,
         2.4392e-01]]), 'X': tensor([[  4.8674,  12.8875,   6.7622],
        [  3.1064,  13.5607,   6.7605],
        [  2.5318,  13.7231,   6.3692],
        [  2.2659,  13.7894,   6.5436],
        [  2.3261,  13.7749,   5.9121],
        [  1.3705,  13.9723,   7.9266],
        [ -6.0175,  -8.4541,  -9.8408],
        [  6.2675,  12.1379,  -8.9720],
        [ -0.8675, -10.8815,   6.8561],
        [  0.2732,  -4.7619,  -1.1577],
        [  3.4401,  13.4540,   4.6576],
        [  5.1472,  -2.6801,   8.1194],
        [  1.3103,  -4.8032,  -0.9300],
        [  0.9478,  -4.8144,   0.6371],
        [  9.3607,   9.5337,  -4.5412],
        [  0.6363, -11.0348,   6.1414],
        [ 12.6059,   0.9597,  -6.3264],
        [ 12.0582,  -2.6640,   0.7132],
        [  6.1752,   2.5242,   3.5874],
        [  9.2979,   9.6048,   7.3138],
        [  9.0129,  -7.7072,   9.1400],
        [  6.7792,  11.8084,   1.6251],
        [ -8.7409,  -4.6837,  -6.5840],
        [  6.3124,   0.2057,  -0.4593],
        [ -4.8665,  -9.3204,  -5.2564],
        [  2.3367,   7.1676,  -9.5263],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.8719,   7.3713,  -3.7465],
        [ -2.6069, -10.4298,  -6.1995],
        [  5.0348,  -2.8229,   3.5436],
        [ -4.0868,   7.2981,   8.1874],
        [  1.8320,   7.3871,   5.1452],
        [  2.3373,  13.7721,  -0.2442],
        [ 12.4597,  -0.9136,  -4.8763],
        [ 10.0567,   8.6697,   4.6329],
        [  5.1667,  -2.6543,   1.2879],
        [  2.3261,   7.1726,   6.2522],
        [  5.9526,   3.2525,  -4.3821],
        [  5.0506,   4.9277,  -9.5094],
        [ -4.7847,  -9.3729,  -3.7741],
        [  6.9653,  -9.3225,  -3.4727],
        [  7.9141,  -8.6656,   9.6948],
        [  4.5340,  -3.3629,   8.9652],
        [ -8.9379,  -4.1655,  -2.8114],
        [ 11.3118,   6.5957,  -4.5667],
        [ -8.4267,  -5.3697,   2.4172],
        [  2.3376,  -4.6231,   9.8436],
        [ 12.5717,   1.9266,   7.4903],
        [  6.2790,  -0.0259,  -5.4116],
        [  0.9809,   7.6642,  -1.4217],
        [  0.7958,  -4.8109,  -8.2248],
        [ -9.4645,  -1.4880,  -2.0226],
        [  6.3048,   0.1469,  -9.1691],
        [  9.2979,   9.6048,   7.3138],
        [  9.6662,   9.1725,   8.1056],
        [ -9.1231,  -3.5739,   8.4708],
        [  0.9809,   7.6642,  -1.4217],
        [ 12.5927,   0.4199,  -9.4997],
        [  9.4554,   9.4245,   9.8800],
        [  2.5010,  -4.5736,  -0.8989],
        [  5.8325,  -1.5222,  -2.1341],
        [ -4.5778,  -9.5013,  -5.0667],
        [  3.3744,   6.5623,  -6.3636],
        [ 10.5955,   7.8824,  -3.7006],
        [ -3.2053, -10.1993,  -3.8736],
        [ -2.0840,   7.8415,  -7.8657],
        [  8.3718,  -8.2953,   8.4666],
        [ -7.8359,   4.2735,   7.2943],
        [ -9.2301,  -3.1537,  -2.7477],
        [  6.3505,   0.6235,  -8.5849],
        [  9.3607,   9.5337,  -4.5412],
        [ -8.8400,   2.3991,   1.3457],
        [  5.8836,   3.4315,  -4.0327],
        [  2.3261,   7.1726,   6.2522],
        [ -5.1398,  -9.1367,   6.2074],
        [  2.5829,  -4.5466,  -8.4015],
        [ -9.1813,  -3.3550,   1.8827],
        [ -8.0965,  -5.9712,   4.3466],
        [  3.0719, -10.8449,  -4.1673],
        [ -9.4756,  -0.8166,  -9.6021],
        [ 11.6544,   5.8273,  -3.1375],
        [  9.5203,   9.3483,   5.8781],
        [  1.3103,  -4.8032,  -0.9300],
        [  8.2422,  -8.4041,  -1.4962],
        [  2.0419,  -4.6978,  -5.0983],
        [  2.1625,  -4.6697,   7.0658],
        [ -9.4296,  -1.9458,   4.0833],
        [  4.7482, -10.3907,   5.9149],
        [ -8.6603,   2.8250,   5.9181],
        [ 12.0845,  -2.5772,  -4.6073],
        [  8.0877,  -8.5297,  -1.1050],
        [  2.7976,  -4.4682,  -4.2020],
        [ -5.3382,   6.6533,   7.8054],
        [  2.7420,  -4.4896,  -3.7139],
        [ -8.3653,   3.4184,   9.3454],
        [ -9.4366,  -0.1189,  -2.6736],
        [ -1.7009,   7.8851,  -8.2437],
        [  2.3490,  -4.6199,  -4.2489],
        [ -7.6525,  -6.6534,   5.1721],
        [ 11.9122,   5.1397,  -1.5498],
        [ -8.3535,  -5.5118,   9.8399],
        [  6.3360,   0.4258,  -9.2509],
        [  0.2435,   7.8191,  -6.4709],
        [  0.1532,  -4.7422,  -7.0145],
        [ -0.8569,   7.9156,  -6.5557],
        [  6.2242,  -0.3171,   7.4978]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.9867e-02, 5.5020e-02,  ..., 1.1791e+00, 1.2863e+00,
         4.0031e-01],
        [1.9867e-02, 0.0000e+00, 3.9314e-02,  ..., 1.1692e+00, 1.2748e+00,
         3.8314e-01],
        [5.5021e-02, 3.9315e-02, 0.0000e+00,  ..., 1.1313e+00, 1.2360e+00,
         3.4531e-01],
        ...,
        [1.1791e+00, 1.1692e+00, 1.1313e+00,  ..., 0.0000e+00, 1.6357e-01,
         8.5853e-01],
        [1.2863e+00, 1.2748e+00, 1.2360e+00,  ..., 1.6357e-01, 1.5170e-04,
         9.3785e-01],
        [4.0031e-01, 3.8314e-01, 3.4531e-01,  ..., 8.5853e-01, 9.3785e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.77604293, 13.91198191, 13.95472331, 13.97430191, 13.96987806,
       14.0393949 , 10.37698435, 13.66051884, 10.91601655,  4.76969364,
       13.88683436,  5.80312949,  4.978704  ,  4.90676242, 13.36092624,
       11.05317597, 12.64235359, 12.3489324 ,  6.67121748, 13.36800073,
       11.85890664, 13.61600536,  9.91668182,  6.31576264, 10.51437947,
        7.53884104,  9.38232121,  7.60529892, 10.7506423 ,  5.77219077,
        8.36445758,  7.61088125, 13.96905756, 12.49317563, 13.2778378 ,
        5.80860165,  7.5403822 ,  6.78325151,  7.05627461, 10.52357479,
       11.63724792, 11.73567759,  5.64500512,  9.86089579, 13.09426142,
        9.99211663,  5.1805345 , 12.71843462,  6.27905292,  7.72668467,
        4.8763135 ,  9.58072286,  6.30647746, 13.36800073, 13.32557133,
        9.79814997,  7.72668467, 12.59969936, 13.35013374,  5.21280156,
        6.02789634, 10.54658597,  7.37902766, 13.20597932, 10.6910768 ,
        8.1137452 , 11.785566  ,  8.92549752,  9.75401449,  6.38105015,
       13.36092624,  9.15976746,  6.81117802,  7.5403822 , 10.48315436,
        5.22903117,  9.7751189 , 10.06023665, 11.27160035,  9.51074763,
       13.03002338, 13.3426541 ,  4.978704  , 11.77124373,  5.1224079 ,
        5.14607014,  9.6282763 , 11.42420508,  9.10945678, 12.35625694,
       11.75438006,  5.27178363,  8.53014915,  5.26067019,  9.0368458 ,
        9.43737523,  8.06643487,  5.18277858, 10.14044489, 12.97370371,
       10.00802627,  6.35029274,  7.822854  ,  4.74468406,  7.96181288,
        6.2322769 ])}
{'P': tensor([[1.9015e-03, 2.3842e-01, 2.0007e-01,  ..., 3.8398e-07, 7.9396e-09,
         2.7965e-07],
        [5.7239e-06, 2.4848e-03, 5.6498e-01,  ..., 5.0051e-07, 7.9899e-09,
         3.3469e-07],
        [8.5135e-06, 1.3209e-05, 6.8970e-03,  ..., 1.4530e-06, 1.9659e-08,
         8.7891e-07],
        ...,
        [5.9356e-05, 4.4538e-05, 3.8166e-05,  ..., 9.5363e-02, 3.8949e-03,
         1.5148e-02],
        [1.1305e-04, 6.1237e-05, 4.3521e-05,  ..., 5.3409e-05, 4.6088e-02,
         1.1249e-04],
        [7.2674e-04, 5.2746e-04, 4.1751e-04,  ..., 2.2205e-03, 9.7601e-04,
         1.3497e-01]]), 'X': tensor([[ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [-8.0965e+00, -5.9712e+00,  4.3466e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 3.3069e+00,  6.6089e+00,  8.7377e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [ 1.1557e+01,  6.0602e+00,  9.3244e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 2.5974e+00,  7.0363e+00, -7.7810e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [-6.0195e+00,  6.1789e+00, -2.9496e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-7.3554e+00,  4.8985e+00, -6.6647e-01],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 2.0547e+00,  7.2957e+00, -1.9343e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [ 4.7156e+00, -3.1831e+00, -8.0224e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 1.9312e-01, -4.7491e+00,  7.4385e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [ 9.4450e-01,  7.6736e+00,  5.0946e-01],
        [ 6.2535e+00, -1.7090e-01, -9.4102e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-7.5986e-01, -1.0900e+01, -6.0366e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.2452e-02, 6.6647e-02,  ..., 5.2535e-01, 1.0026e+00,
         2.7935e-01],
        [5.2452e-02, 0.0000e+00, 1.9428e-02,  ..., 5.6491e-01, 1.0522e+00,
         2.9120e-01],
        [6.6647e-02, 1.9428e-02, 0.0000e+00,  ..., 5.6553e-01, 1.0595e+00,
         2.8146e-01],
        ...,
        [5.2535e-01, 5.6491e-01, 5.6553e-01,  ..., 5.3633e-05, 5.5997e-01,
         3.5596e-01],
        [1.0026e+00, 1.0522e+00, 1.0595e+00,  ..., 5.5997e-01, 0.0000e+00,
         9.0821e-01],
        [2.7935e-01, 2.9120e-01, 2.8146e-01,  ..., 3.5596e-01, 9.0821e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.79676654, 13.89671314, 13.96746816, 13.98691426, 13.89886469,
       14.03618649,  6.10360073,  5.29009836,  9.15866607,  8.97052115,
       10.1979669 ,  9.13507321, 11.83509372, 11.83570132, 13.54774355,
       11.2845608 ,  5.87132718, 13.03002338, 10.06023665, 13.55868684,
        7.39004226,  6.28961988,  6.80370254,  5.29009836,  6.39438044,
        6.27905292,  4.74468406,  9.54154096, 13.45642379,  8.67502348,
       13.0493368 ,  7.37902766, 13.37121923, 10.88618114,  7.50035355,
       13.2430377 ,  8.10886021,  8.62631801, 13.36800073, 13.01187484,
        5.40689332, 10.0674887 ,  8.83725081,  7.69752262,  8.51088913,
        8.50278896,  8.86082124,  7.24326992, 13.93790235,  9.9353921 ,
       13.27717982, 14.09176259,  5.23878371, 11.52733581, 11.42420508,
        7.5158489 , 11.35345345, 12.67680439, 13.89886469,  7.57946459,
       12.67576792,  5.92281142,  8.11227668,  6.07992623,  9.34127013,
        9.19837792,  5.34250225, 12.64235359, 11.95743558, 11.05317597,
        9.11473391,  5.51177501,  5.68942572, 12.46513646,  7.19534146,
        8.43626114,  4.75303127,  9.78286733,  8.81930525,  8.41148722,
        8.4486771 ,  9.60412735, 12.46513646,  6.97106912,  8.68832398,
        7.83534987, 12.41012046, 10.24557047,  9.41970031,  7.73151309,
        6.25586297,  6.35029274,  9.96391045, 11.22142195,  7.37902766,
       13.3614598 ,  9.47369628, 12.27993098, 10.92597196, 10.99326577,
        7.60463349,  7.69752262,  5.92281142,  4.7663526 , 12.10222204,
        5.5671389 ])}
{'P': tensor([[1.7578e-03, 2.4525e-01, 2.3459e-01,  ..., 8.8663e-04, 1.3523e-08,
         1.3155e-08],
        [6.0540e-06, 2.9038e-03, 2.1718e-03,  ..., 1.0888e-03, 1.8083e-08,
         2.2027e-08],
        [1.7155e-05, 5.0991e-03, 8.0193e-03,  ..., 3.6012e-03, 5.1216e-08,
         6.2586e-08],
        ...,
        [2.8676e-04, 5.1866e-04, 5.3809e-04,  ..., 9.8616e-02, 8.2723e-06,
         1.8090e-05],
        [4.7166e-04, 2.5535e-04, 2.7042e-04,  ..., 1.9312e-04, 4.5156e-02,
         1.2363e-05],
        [1.5252e-05, 9.4443e-06, 9.7767e-06,  ..., 2.7464e-05, 1.3218e-01,
         5.0643e-02]]), 'X': tensor([[  4.5954,  13.0099,  -4.4439],
        [  3.2809,  13.5061,  -4.0199],
        [  3.2692,  13.5098,  -4.9032],
        [  2.3589,  13.7668,  -3.6501],
        [  1.4150,  13.9647,  -4.0915],
        [  3.3094,  13.4969,  -3.6507],
        [  5.8836,   3.4315,  -4.0327],
        [  4.0078,  -3.8018,  -6.2168],
        [ -8.9033,   2.2329,  -8.8912],
        [ 12.2592,   3.9425,   2.0383],
        [  5.6523,   3.9443,  -2.9449],
        [  4.8653,   5.1694,  -4.0823],
        [  5.6523,   3.9443,  -2.9449],
        [ 11.2767,  -4.6006,   0.7868],
        [ 12.1379,   4.4125,  -3.8833],
        [  6.0073,   3.0986,   6.7937],
        [ -7.7239,   4.4298,   3.1216],
        [  2.2644,  -4.6434,   9.7508],
        [ -6.3489,   5.9117,   6.5161],
        [ 10.3251,   8.2928,  -3.0006],
        [  3.7121,   6.3117,   0.6738],
        [  3.2226,  -4.2802,  -4.0428],
        [ 12.2218,   4.0953,  -8.2072],
        [  9.0129,  -7.7072,   9.1400],
        [  5.6672,  -1.8631,   8.1020],
        [ -8.1905,   3.7253,   4.1175],
        [ -9.4283,  -1.9585,  -0.9129],
        [  6.1755,   2.5233,  -2.0042],
        [  1.6308,  13.9254,  -2.0984],
        [ -6.3489,   5.9117,   6.5161],
        [ -8.8934,   2.2597,   3.5831],
        [ -7.9298,   4.1367,  -3.9773],
        [  7.8982,  -8.6778,   0.0239],
        [ 12.5267,  -0.4125,  -8.7331],
        [  4.4049,   5.6892,  -4.4394],
        [  1.1006,   7.6318,  -2.4877],
        [  9.4925,   9.3811,   8.9842],
        [  3.7121,   6.3117,   0.6738],
        [  7.3592,  11.3937,   6.2407],
        [ -9.0604,   1.7707,   1.1066],
        [  4.2339,  -3.6268,   6.3125],
        [  3.1892, -10.8220,  -6.8409],
        [  7.0358,  -9.2784,  -9.8479],
        [  0.8940, -11.0402,  -0.9783],
        [  6.2895,   0.0405,   5.7652],
        [  6.3516,   0.6425,   0.3425],
        [  0.5960, -11.0335,   3.1144],
        [  8.1217,  -8.5024,  -0.6149],
        [ -5.3179,  -9.0101,   9.4472],
        [  5.6095,  -1.9702,  -7.8233],
        [  6.2036,   2.4030,  -1.2500],
        [  6.7792,  11.8084,   1.6251],
        [  8.0381,  -8.5690,  -8.9523],
        [ -8.0087,   4.0173,   9.1091],
        [  3.9591, -10.6391,  -6.2557],
        [ 10.0619,   8.6627,  -9.2884],
        [ 11.4495,   6.3026,   6.8257],
        [ -4.4503,   7.1378,   3.3751],
        [  9.6587,   9.1817,  -0.8212],
        [  1.6954,  -4.7616,   9.8198],
        [  2.7295,  13.6703,  -8.9328],
        [ -8.9636,   2.0648,   8.1946],
        [  0.5175,  14.0910,   4.6540],
        [  0.1460,   7.8340,  -9.4614],
        [ -9.2707,   0.9661,   4.4863],
        [ -1.0482, -10.8487,  -5.9066],
        [ 11.2767,  -4.6006,   0.7868],
        [  2.0870,  -4.6877,   9.6823],
        [  6.3534,   1.2990,  -7.2942],
        [  3.2226,  -4.2802,  -4.0428],
        [  2.3200,   7.1756,   1.1938],
        [ -6.9838,  -7.4965,   0.9796],
        [  5.7151,  -1.7700,   3.1542],
        [  4.3523, -10.5232,   8.2138],
        [ -3.2053, -10.1993,  -3.8736],
        [  6.1984,  -0.4327,   1.5438],
        [  0.6142,  14.0802,   0.5407],
        [  7.3592,  11.3937,   6.2407],
        [  1.7232,  -4.7574,   3.1270],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.9883,   3.1536,   6.2320],
        [ -8.0534,  -6.0429,   2.4583],
        [  2.0937,  13.8293,  -2.3533],
        [  0.1497,   7.8334,  -5.4412],
        [  6.1984,  -0.4327,   1.5438],
        [  3.1271, -10.8343,   2.6442],
        [ -5.9872,   6.2037,  -9.4477],
        [  5.0348,  -2.8229,   3.5436],
        [  5.9548,  -1.2273,  -2.6520],
        [ -8.0611,  -6.0302,   4.6962],
        [  8.5716,  10.3596,   6.8818],
        [  2.5636, -10.9293,   9.6519],
        [  3.1271, -10.8343,   2.6442],
        [ -2.0840,   7.8415,  -7.8657],
        [ -4.9217,   6.8979,  -1.7667],
        [  0.6142,  14.0802,   0.5407],
        [ 11.5198,  -4.0923,   5.5088],
        [ -7.9298,   4.1367,  -3.9773],
        [ -8.5859,  -5.0393,   0.5476],
        [  2.1504,  -4.6726,   8.7908],
        [ 12.2218,   4.0953,  -8.2072],
        [  4.7813,  -3.1139,   8.2691],
        [ -1.7290,   7.8825,  -3.3831],
        [ -0.8569,   7.9156,  -6.5557],
        [ 10.9717,  -5.1646,  -6.9039],
        [  4.2612, -10.5514,  -6.6268]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 2.7667e-02, 4.8067e-02,  ..., 5.2515e-01, 3.2343e-01,
         7.1770e-01],
        [2.7667e-02, 0.0000e+00, 5.3964e-02,  ..., 5.1174e-01, 3.2199e-01,
         7.0766e-01],
        [4.8067e-02, 5.3964e-02, 0.0000e+00,  ..., 4.8273e-01, 2.7539e-01,
         6.7238e-01],
        ...,
        [5.2515e-01, 5.1174e-01, 4.8273e-01,  ..., 1.5170e-04, 2.6880e-01,
         2.1072e-01],
        [3.2343e-01, 3.2199e-01, 2.7539e-01,  ..., 2.6880e-01, 0.0000e+00,
         4.1892e-01],
        [7.1770e-01, 7.0766e-01, 6.7238e-01,  ..., 2.1072e-01, 4.1892e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.79762415, 13.89886469, 13.89974693, 13.96746816, 14.03618649,
       13.89671314,  6.81117802,  5.52416368,  9.17905491, 12.87752052,
        6.89245352,  7.09887557,  6.89245352, 12.17901033, 12.91505111,
        6.7593917 ,  8.90405823,  5.16611782,  8.67502348, 13.2430377 ,
        7.32233733,  5.35774573, 12.889695  , 11.85890664,  5.96556248,
        8.99791316,  9.62958973,  6.67108662, 14.02058621,  8.67502348,
        9.17595899,  8.94395628, 11.73397309, 12.53345445,  7.19513806,
        7.71075226, 13.34587027,  7.32233733, 13.56367857,  9.23178204,
        5.5748624 , 11.28215429, 11.64436684, 11.07637531,  6.28961988,
        6.38399789, 11.04953767, 11.75807252, 10.46238054,  5.94541803,
        6.652742  , 13.61600536, 11.74901778,  8.95983586, 11.35182589,
       13.27717982, 13.06957425,  8.41148722, 13.32645552,  5.05444288,
       13.94008948,  9.19837792, 14.10045797,  7.83534987,  9.32094055,
       10.8992524 , 12.17901033,  5.13125513,  6.48486171,  5.35774573,
        7.54127425, 10.24557047,  5.98285319, 11.38774149, 10.6910768 ,
        6.2134961 , 14.09357547, 13.56367857,  5.05989547,  9.75200848,
        6.76791483, 10.06850677, 13.98691426,  7.83486943,  6.2134961 ,
       11.27656488,  8.62162372,  5.77219077,  6.07992623, 10.06703358,
       13.44593519, 11.22596994, 11.27656488,  8.1137452 ,  8.47371868,
       14.09357547, 12.22503754,  8.94395628,  9.95552626,  5.14371005,
       12.889695  ,  5.70591986,  8.06990483,  7.96181288, 12.12641969,
       11.37939695])}
{'P': tensor([[2.0460e-03, 2.5534e-01, 2.4258e-01,  ..., 6.4419e-07, 1.4134e-06,
         7.1430e-05],
        [6.7732e-06, 4.9075e-03, 7.6758e-03,  ..., 2.8551e-06, 5.8230e-06,
         3.1524e-04],
        [7.3883e-06, 1.7561e-03, 5.6223e-03,  ..., 2.7689e-06, 7.3497e-06,
         2.8967e-04],
        ...,
        [3.9384e-07, 4.0880e-07, 3.6108e-07,  ..., 5.6097e-02, 2.4069e-04,
         4.5787e-05],
        [5.2956e-07, 4.8478e-07, 5.0235e-07,  ..., 1.0053e-04, 3.7263e-02,
         1.2885e-05],
        [2.7811e-06, 4.1345e-06, 3.9207e-06,  ..., 1.9479e-02, 1.7062e-03,
         4.4611e-02]]), 'X': tensor([[ 4.1637e+00,  1.3190e+01, -1.9189e+00],
        [ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-8.4573e+00,  3.2448e+00,  5.4370e+00],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 4.1140e+00, -3.7220e+00, -1.2732e-01],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 6.1737e+00,  2.5305e+00,  3.9049e+00],
        [ 4.0362e+00,  1.3240e+01, -7.0030e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 5.8617e+00,  3.4853e+00, -9.8080e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [-5.9940e+00,  6.1985e+00,  5.9566e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.3403e+00, -4.8011e+00, -1.1698e+00],
        [-1.8788e+00, -1.0657e+01,  3.7068e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.1663e+01,  5.8050e+00,  9.1590e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 6.3048e+00,  1.4688e-01, -9.1691e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 6.8093e+00, -9.4178e+00,  8.4244e+00],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 1.1964e+01, -2.9568e+00,  7.7211e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 8.7131e+00,  1.0222e+01,  7.7915e+00],
        [ 8.7004e-01, -4.8132e+00, -7.8042e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 3.9022e+00,  1.3291e+01,  5.5788e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-3.2053e+00, -1.0199e+01, -3.8736e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [-8.9624e+00, -4.0944e+00, -4.9622e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0697, 0.0611,  ..., 1.3572, 1.2597, 0.9806],
        [0.0697, 0.0000, 0.0849,  ..., 1.2874, 1.2094, 0.9108],
        [0.0611, 0.0849, 0.0000,  ..., 1.3495, 1.2121, 0.9713],
        ...,
        [1.3572, 1.2874, 1.3495,  ..., 0.0000, 0.9761, 0.3788],
        [1.2597, 1.2094, 1.2121,  ..., 0.9761, 0.0000, 0.8369],
        [0.9806, 0.9108, 0.9713,  ..., 0.3788, 0.8369, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.83139024, 13.97857591, 13.98691426, 14.02058621, 14.08549802,
       14.03505624,  9.05843201, 11.83509372,  5.54776934,  5.82345867,
       13.26209545,  8.55982105, 14.03505624, 11.83075005, 11.8159254 ,
        9.43112957,  8.98765644,  8.12043983, 10.53368644,  6.67218687,
       13.84126197, 12.3489324 , 14.07623266,  8.40445872, 11.83570132,
       10.91805683,  6.81961996,  5.48276038,  7.822854  ,  8.62260327,
       13.31764181,  7.85919687,  5.83560009, 11.92133129, 13.99048094,
        9.48337063,  5.70591986, 14.02526505, 13.25017211, 12.19805394,
        6.50409372,  4.98461983, 10.82106505, 10.08001894, 10.00794468,
       12.22503754,  5.64464807, 11.4424816 , 10.01804223, 13.02818702,
        8.51088913,  4.74770899, 13.6660671 , 10.1371552 ,  9.68323686,
       10.88618114, 12.40074901,  6.30647746, 12.76408562,  8.68979052,
        5.16205087,  9.09661212, 12.77402351,  7.66377554, 10.31584102,
       10.04916678,  4.76969364, 13.59673406, 13.2430377 ,  6.89245352,
        5.94541803, 11.62158554, 13.43127131, 12.32408516,  9.09119351,
        8.01009083, 13.43127131,  4.89121805,  9.27474386, 10.54658597,
       13.40850328,  8.09842704, 10.61417181, 11.54672755, 12.33167382,
       13.32374056,  5.27178363,  7.16598273, 13.79762415,  8.36995953,
       13.20597932, 13.85158417, 10.14436912, 12.77402351, 13.32557133,
        7.24326992,  7.37576154, 10.57302546,  9.49511524, 10.6910768 ,
        4.87083348, 11.9120321 ,  8.95983586,  9.77588801,  9.85330646,
        8.81930525])}
{'P': tensor([[2.0803e-03, 2.8723e-01, 1.4279e-01,  ..., 2.4756e-06, 1.9206e-05,
         1.4486e-06],
        [7.6784e-06, 6.1326e-03, 5.3500e-01,  ..., 3.8448e-06, 2.1967e-05,
         2.4672e-06],
        [1.6678e-04, 6.6356e-04, 2.2761e-01,  ..., 1.0167e-04, 5.4650e-04,
         6.7704e-05],
        ...,
        [1.3301e-02, 1.0210e-02, 9.6625e-03,  ..., 8.9596e-02, 2.3951e-03,
         6.4842e-03],
        [5.4169e-04, 1.3492e-03, 3.5368e-03,  ..., 1.1246e-04, 2.9071e-01,
         5.2928e-05],
        [1.0852e-02, 7.2834e-03, 5.9552e-03,  ..., 2.2295e-01, 9.3107e-04,
         7.5825e-02]]), 'X': tensor([[  4.0362,  13.2397,  -7.0030],
        [  2.0449,  13.8402,  -7.1642],
        [  0.6396,  14.0772,  -6.3786],
        [  0.8572,  14.0501,  -8.3364],
        [  0.4119,  14.1019,  -4.1870],
        [  2.7295,  13.6703,  -8.9328],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.6613,   7.0021,   4.5783],
        [ -0.6323,   7.9085,  -5.9544],
        [  9.6587,   9.1817,  -0.8212],
        [ -8.6426,  -4.9136,   9.4941],
        [  5.4429,  -2.2529,   9.7306],
        [ 12.1588,   4.3364,  -8.0867],
        [  3.7298,  13.3537,  -9.4653],
        [ -9.3419,   0.5972,  -1.2841],
        [  5.0190,   4.9705,  -7.9752],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -9.2707,   0.9661,   4.4863],
        [  6.3606,   0.9204,   3.2525],
        [  0.1206,  -4.7363,   8.7103],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -0.5709,   7.9055,   7.6089],
        [ -9.4501,  -1.7133,   2.2980],
        [ -8.9178,   2.1936,   7.5388],
        [  2.3200,   7.1756,   1.1938],
        [ 11.7779,   5.5127,  -2.0699],
        [ -5.6174,   6.4708,  -1.7339],
        [  5.3602,  -2.3808,   7.2098],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.4690,  -0.8528,  -9.3374],
        [  1.8447,  -4.7372,   8.1235],
        [  3.2692,  13.5098,  -4.9032],
        [  1.9241,   7.3503,   7.9004],
        [ -7.4733,   4.7555,   6.2490],
        [  6.3542,   0.6934,  -3.3840],
        [ -7.7239,   4.4298,   3.1216],
        [ -0.5709,   7.9055,   7.6089],
        [  3.7916,  -3.9524,  -7.9547],
        [ 10.5955,   7.8824,  -3.7006],
        [ 11.3367,  -4.4806,   0.4643],
        [ -1.6220,   7.8917,   0.0315],
        [ -8.5506,  -5.1154,   6.3423],
        [  0.8572,  14.0501,  -8.3364],
        [  4.0078,  -3.8018,  -6.2168],
        [ -8.6675,  -4.8569,  -9.1769],
        [ 12.2311,  -2.0445,   1.3545],
        [ 11.7795,  -3.4697,   3.5624],
        [  6.3303,   1.6145,  -8.4083],
        [ -0.8885,   7.9160,   0.0571],
        [ -0.4341, -10.9473,   7.4509],
        [  7.9141,  -8.6656,   9.6948],
        [  0.7958,  -4.8109,  -8.2248],
        [ 12.5154,   2.5072,  -9.3274],
        [  2.0729,   7.2878,   7.0880],
        [ -9.4714,  -1.3301,  -3.1998],
        [ -3.9700,   7.3453,  -4.1486],
        [ -5.9940,   6.1985,   5.9566],
        [ -9.4117,   0.1160,   8.2574],
        [  4.5954,  13.0099,  -4.4439],
        [  3.1892, -10.8220,  -6.8409],
        [  1.8766,   7.3694,   1.1887],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.3453,  -1.5462,   7.0212],
        [ 12.2914,   3.8037,   6.0978],
        [  0.2435,   7.8191,  -6.4709],
        [ -8.0608,   3.9362,  -5.9567],
        [ -9.4753,  -0.8012,  -5.2050],
        [ -7.2515,   5.0195,   4.6434],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.3589,  13.7668,  -3.6501],
        [  2.0547,   7.2957,  -1.9343],
        [  6.2905,   1.9366,   2.1169],
        [  5.8902,  -1.3887,   4.0388],
        [  2.5318,  13.7231,   6.3692],
        [  2.4697,  -4.5836,   1.4593],
        [ -1.5875,   7.8944,  -9.4286],
        [  2.0449,  13.8402,  -7.1642],
        [  0.1931,  -4.7491,   7.4385],
        [ -1.7009,   7.8851,  -8.2437],
        [ -8.0965,  -5.9712,   4.3466],
        [  6.2804,   2.0019,   0.8607],
        [ 10.9931,   7.2107,   7.7933],
        [  0.9967,  -4.8144,  -9.7370],
        [  6.0047,   3.1062,  -6.7914],
        [ -8.9606,  -4.0994,   1.7886],
        [  3.5284, -10.7484,   5.3347],
        [ -9.3087,   0.7792,  -7.5587],
        [ -0.4341, -10.9473,   7.4509],
        [ 11.2939,   6.6325,  -5.8311],
        [  8.4691,  10.4571,  -5.0650],
        [  4.1537,   5.9340,  -6.0423],
        [  3.8182,  -3.9347,  -1.7961],
        [  6.0140,  -1.0661,  -2.5040],
        [  9.2979,   9.6048,   7.3138],
        [  3.3094,  13.4969,  -3.6507],
        [ -8.1531,  -5.8748,   2.6934],
        [  6.3303,   1.6145,  -8.4083],
        [  0.9945,  -4.8144,  -0.1897],
        [  4.2339,  -3.6268,   6.3125],
        [ -6.3306,   5.9272,   5.6935],
        [  0.9036,   7.6840,   9.9237],
        [  8.7131,  10.2216,   7.7915],
        [  5.6523,   3.9443,  -2.9449],
        [  2.3261,  13.7749,   5.9121],
        [  6.0955,   2.8210,  -4.3321]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0661, 0.2252,  ..., 0.4117, 1.1531, 0.3414],
        [0.0661, 0.0000, 0.1623,  ..., 0.3721, 1.1513, 0.2979],
        [0.2252, 0.1623, 0.0000,  ..., 0.2589, 1.1030, 0.1823],
        ...,
        [0.4117, 0.3721, 0.2589,  ..., 0.0000, 0.8628, 0.0774],
        [1.1531, 1.1513, 1.1030,  ..., 0.8628, 0.0000, 0.9294],
        [0.3414, 0.2979, 0.1823,  ..., 0.0774, 0.9294, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.84126197, 13.99048094, 14.09176259, 14.07623266, 14.10796466,
       13.94008948,  9.37989631,  7.49076899,  7.9337647 , 13.32645552,
        9.94173708,  5.89074104, 12.9089559 , 13.86480175,  9.36093562,
        7.06372172,  9.69792513,  9.32094055,  6.42689705,  4.73784222,
       10.60226614,  7.92607342,  9.60412735,  9.18358562,  7.54127425,
       13.00413829,  8.56889738,  5.86519525, 12.44177641, 12.4980845 ,
        5.08372149, 13.89974693,  7.59795445,  8.8580671 ,  6.39187285,
        8.90405823,  7.92607342,  5.47702766, 13.20597932, 12.18998195,
        8.05668609,  9.96391045, 14.07623266,  5.52416368,  9.93553424,
       12.40074901, 12.27991991,  6.53290663,  7.96575782, 10.95593729,
       11.73567759,  4.8763135 , 12.76408562,  7.57685908,  9.5642993 ,
        8.34949383,  8.62260327,  9.41244865, 13.79762415, 11.28215429,
        7.60463349,  9.5014146 ,  7.8788965 , 12.44177641, 12.86648771,
        7.822854  ,  8.97052115,  9.50913849,  8.81930525, 12.19805394,
       13.96746816,  7.57946459,  6.58184474,  6.05165248, 13.95472331,
        5.20659892,  8.05242591, 13.99048094,  4.75303127,  8.06643487,
       10.06023665,  6.59176018, 13.14690137,  4.91652556,  6.76056625,
        9.85384623, 11.31276109,  9.34127013, 10.95593729, 13.09737053,
       13.45642379,  7.24326992,  5.48276038,  6.10774544, 13.36800073,
       13.89671314, 10.04916678,  6.53290663,  4.9160962 ,  5.5748624 ,
        8.67227386,  7.73692436, 13.43127131,  6.89245352, 13.96987806,
        6.7166297 ])}
{'P': tensor([[1.7473e-03, 2.6023e-01, 2.4196e-01,  ..., 3.9775e-06, 4.5677e-07,
         1.9619e-05],
        [2.5196e-05, 1.5452e-02, 2.7636e-02,  ..., 2.0390e-05, 1.7474e-06,
         2.0581e-04],
        [9.1081e-06, 1.7358e-03, 5.7459e-03,  ..., 9.3666e-06, 7.0926e-07,
         1.0243e-04],
        ...,
        [5.1748e-04, 6.0838e-04, 8.3050e-04,  ..., 1.5300e-01, 8.5459e-04,
         3.2868e-02],
        [4.9034e-03, 4.6902e-03, 6.8859e-03,  ..., 7.3337e-02, 1.4965e-01,
         2.3828e-03],
        [1.2095e-06, 1.5467e-06, 1.7766e-06,  ..., 2.9359e-05, 1.2692e-06,
         2.8246e-02]]), 'X': tensor([[  3.9022,  13.2906,   5.5788],
        [  2.3261,  13.7749,   5.9121],
        [  2.2182,  13.8007,   5.1889],
        [  2.5318,  13.7231,   6.3692],
        [  2.2659,  13.7894,   6.5436],
        [  1.5662,  13.9375,   4.5142],
        [  2.3373,  13.7721,  -0.2442],
        [ -4.8665,  -9.3204,  -5.2564],
        [  3.7916,  -3.9524,  -7.9547],
        [ -7.3470,  -7.0615,   5.0066],
        [  1.3103,  -4.8032,  -0.9300],
        [ -1.9714,   7.8563,  -1.8084],
        [ 12.5832,   1.7543,  -7.6134],
        [ -6.6998,   5.5954,   0.0823],
        [ -5.0550,   6.8231,  -9.2502],
        [  1.7228,  -4.7575,   8.3396],
        [  2.0419,  -4.6978,  -5.0983],
        [ -9.3604,   0.4855,  -3.0158],
        [ -5.3179,  -9.0101,   9.4472],
        [ -0.1963,   7.8765,  -1.1371],
        [ 11.0891,  -4.9558,  -2.3178],
        [  4.5767,   5.5071,   3.9069],
        [ -4.9217,   6.8979,  -1.7667],
        [  1.8447,  -4.7372,   8.1235],
        [  1.6308,  13.9254,  -2.0984],
        [  4.3377,   5.7570,   1.3712],
        [ -8.9606,  -4.0994,   1.7886],
        [  2.0899,  -4.6870,  -7.4260],
        [  5.5158,  -2.1337,   3.5246],
        [ -7.3470,  -7.0615,   5.0066],
        [  7.7750,  11.0665,   2.8364],
        [ -9.4159,  -2.0739,   9.6841],
        [  5.2911,  -2.4824,  -3.6877],
        [ -9.3411,  -2.6004,  -2.7062],
        [ 11.0275,  -5.0666,   8.4154],
        [  9.9217,  -6.7006,  -5.6728],
        [  5.8979,   3.3958,  -3.8713],
        [  5.8323,  -1.5228,   5.1132],
        [  4.2275,  -3.6320,  -0.2714],
        [  3.2226,  -4.2802,  -4.0428],
        [  1.0000,   7.6591,  -4.5011],
        [  0.3354,  -4.7708,  -4.9125],
        [ -9.4748,  -0.7765,  -5.6695],
        [  6.3604,   0.9042,  -6.5901],
        [ -1.3940, -10.7772,   6.7835],
        [  8.5716,  10.3596,   6.8818],
        [ 11.7162,   5.6728,  -8.9372],
        [  2.1440, -10.9811,   8.5281],
        [ 11.5966,   5.9666,   6.0083],
        [ 12.5199,  -0.4711,   3.0436],
        [ -8.3062,  -5.6008,   0.0287],
        [  4.5767,   5.5071,   3.9069],
        [ -4.4503,   7.1378,   3.3751],
        [  2.5974,   7.0363,  -7.7810],
        [ -6.0844,  -8.3955,   0.7705],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.4877,  -0.7222,   8.8347],
        [  1.7218,  -4.7576,  -3.3617],
        [ -7.6525,  -6.6534,   5.1721],
        [ 12.0582,  -2.6640,   0.7132],
        [  3.9895,   6.0816,  -6.5235],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -7.3554,   4.8985,  -0.6665],
        [  6.0073,   3.0986,   6.7937],
        [ -8.8540,  -4.3971,  -5.2144],
        [ -0.0254, -10.9932,   3.2200],
        [  0.6757, -11.0361,  -1.3047],
        [ 12.0845,  -2.5772,  -4.6073],
        [  3.4604,  -4.1545,   1.8091],
        [  3.1480,  -4.3165,  -7.8002],
        [  5.2911,  -2.4824,  -3.6877],
        [  1.1994,   7.6035,   2.4295],
        [ -8.4824,  -5.2575,  -0.0795],
        [ -3.7964,  -9.9294,   7.7147],
        [  6.1222,  -0.7261,   8.8204],
        [ 10.9599,  -5.1849,  -3.5526],
        [  5.4732,  -2.2041,  -4.3830],
        [  5.3858,   4.4260,  -9.6769],
        [ 11.3691,   6.4760,   1.0787],
        [ 11.1420,   6.9331,  -2.4779],
        [  6.2975,   0.0947,   5.1791],
        [ 12.5832,   1.7543,  -7.6134],
        [  5.1203,  -2.7151,   6.2110],
        [ -7.4807,   4.7463,   2.0616],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.0719, -10.8449,  -4.1673],
        [ -9.1459,   1.4780,  -4.2403],
        [  5.6414, -10.0303,   7.8525],
        [  5.0233,   4.9647,  -3.1788],
        [  4.5325,  -3.3643,   6.7699],
        [ -8.9636,   2.0648,   8.1946],
        [  6.1755,   2.5233,  -2.0042],
        [ -7.4602,   4.7718,  -4.6277],
        [  8.7715,  -7.9390,   5.2900],
        [  3.1756, -10.8247,   6.8882],
        [  4.0591,   6.0202,  -1.6339],
        [ -1.9646,   7.8571,   3.3817],
        [ -9.4157,   0.0821,  -5.0025],
        [  8.1528,  10.7452,   9.1493],
        [  2.1824,  -4.6647,   1.1985],
        [ -8.0588,  -6.0341,  -2.3006],
        [ -8.5506,  -5.1154,   6.3423],
        [  0.5175,  14.0910,   4.6540],
        [  2.5663,   7.0526,  -6.8769],
        [  8.9284,  10.0036,  -6.9210],
        [ -7.0825,   5.2073,  -5.0497]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0818, 0.0713,  ..., 1.1367, 1.1708, 1.3810],
        [0.0818, 0.0000, 0.0487,  ..., 1.0853, 1.1480, 1.3109],
        [0.0713, 0.0487, 0.0000,  ..., 1.0674, 1.1136, 1.3101],
        ...,
        [1.1367, 1.0853, 1.0674,  ..., 0.0000, 0.4511, 0.4843],
        [1.1708, 1.1480, 1.1136,  ..., 0.4511, 0.0000, 0.9292],
        [1.3810, 1.3109, 1.3101,  ..., 0.4843, 0.9292, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.85158417, 13.96987806, 13.97779839, 13.95472331, 13.97430191,
       14.02526505, 13.96905756, 10.51437947,  5.47702766, 10.19036081,
        4.978704  ,  8.09984231, 12.70489563,  8.72896381,  8.49161154,
        5.05981914,  5.1224079 ,  9.37296102, 10.46238054,  7.8788965 ,
       12.1460931 ,  7.16058587,  8.47371868,  5.08372149, 14.02058621,
        7.20826219,  9.85384623,  5.13182506,  5.91408504, 10.19036081,
       13.5247215 ,  9.64157461,  5.84450127,  9.69629   , 12.13568598,
       11.97238624,  6.80559669,  6.02779201,  5.57341623,  5.35774573,
        7.72415048,  4.78258092,  9.50654967,  6.42439344, 10.86693708,
       13.44593519, 13.01728313, 11.18839169, 13.04156075, 12.52875793,
       10.01804223,  7.16058587,  8.41148722,  7.50035355, 10.36844813,
        7.8788965 , 12.5085995 ,  5.05963217, 10.14044489, 12.3489324 ,
        7.27340834, 12.1460931 ,  8.83725081,  6.7593917 ,  9.88571713,
       10.99326577, 11.05672136, 12.35625694,  5.40689332,  5.34250225,
        5.84450127,  7.69752262,  9.97964013, 10.63038565,  6.16512856,
       12.12448912,  5.90035324,  6.97106912, 13.08414689, 13.12299145,
        6.29821673, 12.70489563,  5.79562691,  8.85939626,  8.8353674 ,
       11.27160035,  9.26455795, 11.50791859,  7.06272256,  5.64464807,
        9.19837792,  6.67108662,  8.85572292, 11.83075005, 11.28093507,
        7.26074925,  8.09899874,  9.41605684, 13.48809595,  5.14999799,
       10.0674887 ,  9.96391045, 14.10045797,  7.50500064, 13.40850328,
        8.7907923 ])}
{'P': tensor([[1.6117e-03, 2.9343e-01, 1.7775e-01,  ..., 4.0148e-06, 3.1910e-07,
         2.5468e-05],
        [2.1410e-05, 1.1242e-02, 1.9706e-03,  ..., 1.7573e-05, 1.6444e-06,
         9.7545e-05],
        [1.3153e-05, 1.1516e-02, 3.9036e-03,  ..., 7.6440e-06, 6.7206e-07,
         4.0724e-05],
        ...,
        [2.1817e-03, 2.2601e-03, 3.0187e-03,  ..., 7.5578e-02, 6.9002e-04,
         2.5209e-02],
        [1.0167e-03, 7.8798e-04, 1.0437e-03,  ..., 1.2526e-01, 6.6311e-02,
         1.7570e-02],
        [5.8360e-03, 7.4505e-03, 8.8874e-03,  ..., 3.8954e-03, 3.2160e-04,
         6.5476e-02]]), 'X': tensor([[  3.7298,  13.3537,  -9.4653],
        [  2.5469,  13.7192,  -9.6935],
        [  2.7295,  13.6703,  -8.9328],
        [  0.8572,  14.0501,  -8.3364],
        [  2.0449,  13.8402,  -7.1642],
        [  0.6396,  14.0772,  -6.3786],
        [ -9.3705,   0.4208,  -3.5815],
        [ -8.6381,  -4.9238,  -4.4077],
        [  7.0642,  11.6104,  -9.8316],
        [  2.7590,  13.6621,   9.4496],
        [  5.7151,  -1.7700,   3.1542],
        [ -9.4159,  -2.0739,   9.6841],
        [ -6.2668,   5.9808,   2.5614],
        [  2.5636, -10.9293,   9.6519],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -1.7009,   7.8851,  -8.2437],
        [ -3.5029,   7.5138,   8.1853],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.0608,   3.9362,  -5.9567],
        [  2.9470,  13.6085,   2.9809],
        [  4.0591,   6.0202,  -1.6339],
        [  4.5800,   5.5034,  -3.9432],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.8201,   3.5841,  -8.0609],
        [  7.9403,  -8.6455,  -8.7433],
        [  5.4426, -10.1181,  -5.1268],
        [ -8.9379,  -4.1655,  -2.8114],
        [  0.8572,  14.0501,  -8.3364],
        [ -4.1296,   7.2803,  -2.7194],
        [  3.9515,  -3.8425,   6.1729],
        [ -6.4464,   5.8272,   7.5038],
        [  3.9591, -10.6391,  -6.2557],
        [  5.8323,  -1.5228,   5.1132],
        [  2.7558,  -4.4844,   0.0636],
        [  0.1460,   7.8340,  -9.4614],
        [  2.1771, -10.9776,  -3.2007],
        [ -6.7992,   5.4991,   9.9186],
        [  5.4732,  -2.2041,  -4.3830],
        [  2.5813,  -4.5471,   5.3597],
        [ -8.3653,   3.4184,   9.3454],
        [  4.2041,  -3.6509,  -4.9488],
        [  5.3804,  -2.3504,   7.2744],
        [  6.3048,   0.1469,  -9.1691],
        [  6.2555,   2.1474,   4.1685],
        [ -6.7643,  -7.7372,  -9.6663],
        [ -1.9487,   7.8590,  -3.3330],
        [ -2.6485, -10.4151,   0.8766],
        [  1.8447,  -4.7372,   8.1235],
        [ 11.5935,   5.9740,  -4.3952],
        [  0.6142,  14.0802,   0.5407],
        [  6.1752,   2.5242,   3.5874],
        [ 11.0864,   7.0387,  -0.1238],
        [ 11.9641,  -2.9568,   7.7211],
        [  6.9693,  -9.3200,  -8.3822],
        [ -8.6112,   2.9319,  -7.9943],
        [  4.0591,   6.0202,  -1.6339],
        [  8.3791,  10.5409,   5.5590],
        [  3.5939,   6.4029,  -1.8357],
        [ -3.2736, -10.1703,   3.8712],
        [ -2.0721,   7.8432,   6.6380],
        [ 11.7349,  -3.5838,   2.1644],
        [  9.0129,  -7.7072,   9.1400],
        [  6.5428,  -9.5728,  -2.9897],
        [ -8.7078,   2.7182,   1.8146],
        [  0.1497,   7.8334,  -5.4412],
        [ -8.1426,   3.8048,   9.3542],
        [  8.0877,  -8.5297,  -1.1050],
        [  6.3535,   0.6792,  -7.5101],
        [  0.1206,  -4.7363,   8.7103],
        [ -7.9298,   4.1367,  -3.9773],
        [  5.1890,   4.7318,   0.9515],
        [ -1.7009,   7.8851,  -8.2437],
        [  0.5600, -11.0321,  -0.8660],
        [  3.3943,   6.5483,   1.0790],
        [  4.4290,  -3.4596,   1.4181],
        [  2.6451,  -4.5250,  -0.5565],
        [  2.4572,   7.1085,   7.7376],
        [  5.6672,  -1.8631,   8.1020],
        [  9.1620,  -7.5573,  -9.9882],
        [ -1.0482, -10.8487,  -5.9066],
        [  5.1886,  -2.6250,   6.4677],
        [ -1.9646,   7.8571,   3.3817],
        [ -8.4267,  -5.3697,   2.4172],
        [  6.1685,  12.1979,  -6.2473],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -8.9788,   2.0208,  -3.2867],
        [  5.1787,  -2.6383,  -4.2533],
        [  9.0129,  -7.7072,   9.1400],
        [  1.7228,  -4.7575,   8.3396],
        [ -1.0012, -10.8576,   2.2867],
        [  6.3236,   0.3011,  -4.6668],
        [ -1.0012, -10.8576,   2.2867],
        [ 11.3691,   6.4760,   1.0787],
        [ -0.9715, -10.8630,   0.3555],
        [  2.5469,  13.7192,  -9.6935],
        [  3.9022,  13.2906,   5.5788],
        [ 11.7418,   5.6070,  -1.4192],
        [ 10.6673,  -5.6649,  -4.1646],
        [  6.2895,   0.0405,   5.7652],
        [  5.9587,  12.3216,   9.1695],
        [ -8.0829,  -5.9940,   5.0564],
        [ 12.5927,   0.4199,  -9.4997],
        [  6.0031,  12.2958,  -6.5992],
        [  4.1348,   5.9514,  -0.8856],
        [  6.3211,   1.7017,  -0.2164],
        [  3.0876,   6.7527,  -5.3540]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0894, 0.0603,  ..., 0.6402, 0.6586, 0.4454],
        [0.0894, 0.0000, 0.0317,  ..., 0.5800, 0.6093, 0.3660],
        [0.0603, 0.0317, 0.0000,  ..., 0.5916, 0.6161, 0.3873],
        ...,
        [0.6402, 0.5800, 0.5916,  ..., 0.0000, 0.0981, 0.2696],
        [0.6586, 0.6093, 0.6161,  ..., 0.0981, 0.0000, 0.3422],
        [0.4454, 0.3660, 0.3873,  ..., 0.2696, 0.3422, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.86480175, 13.95361098, 13.94008948, 14.07623266, 13.99048094,
       14.09176259,  9.37989631,  9.94285511, 13.59056613, 13.93790235,
        5.98285319,  9.64157461,  8.66273171, 11.22596994, 10.35485331,
        8.06643487,  8.29022197, 11.8159254 ,  8.97052115, 13.92390689,
        7.26074925,  7.15990676, 11.54672755,  6.83515201, 11.73848065,
       11.4890861 ,  9.86089579, 14.07623266,  8.36995953,  5.51177501,
        8.68979052, 11.35182589,  6.02779201,  5.26342203,  7.83534987,
       11.1913598 ,  8.74470218,  5.90035324,  5.2287058 ,  9.0368458 ,
        5.56809335,  5.87132718,  6.30647746,  6.61386436, 10.27716434,
        8.09704052, 10.74655346,  5.08372149, 13.04217487, 14.09357547,
        6.67121748, 13.13205357, 12.32408516, 11.63765273,  9.09661212,
        7.26074925, 13.46553623,  7.34250948, 10.68416401,  8.11227668,
       12.26997165, 11.85890664, 11.59512194,  9.12220271,  7.83486943,
        8.98765644, 11.75438006,  6.38968357,  4.73784222,  8.94395628,
        7.02253235,  8.06643487, 11.04629471,  7.37576154,  5.62005629,
        5.24137909,  7.52117132,  5.96556248, 11.87665501, 10.8992524 ,
        5.81481044,  8.09899874,  9.99211663, 13.66896739,  9.75401449,
        9.20340031,  5.81199422, 11.85890664,  5.05981914, 10.9036226 ,
        6.33075682, 10.9036226 , 13.08414689, 10.90638066, 13.95361098,
       13.85158417, 13.01187484, 12.07818871,  6.28961988, 13.68672855,
       10.06286074, 12.59969936, 13.6829843 ,  7.24677664,  6.54615887,
        7.42512621])}
{'P': tensor([[1.8974e-03, 2.8430e-01, 2.3777e-01,  ..., 1.6180e-08, 1.0159e-05,
         3.9414e-06],
        [5.7896e-06, 4.4400e-03, 3.9130e-03,  ..., 2.7678e-08, 4.7525e-06,
         2.3858e-06],
        [1.0832e-05, 6.2149e-03, 7.2720e-03,  ..., 4.5014e-08, 8.1111e-06,
         3.7694e-06],
        ...,
        [5.0246e-03, 2.1939e-03, 2.1548e-03,  ..., 2.4251e-01, 1.4309e-02,
         1.2081e-02],
        [7.1484e-02, 1.1862e-01, 9.2835e-02,  ..., 4.6314e-07, 3.2795e-02,
         2.9036e-04],
        [2.1760e-02, 3.0032e-02, 3.7202e-02,  ..., 1.4988e-07, 3.3288e-04,
         9.5827e-03]]), 'X': tensor([[ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 6.2770e+00, -3.7878e-02,  2.4195e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [-6.6998e+00,  5.5954e+00,  8.2292e-02],
        [ 6.3460e+00,  1.4252e+00,  1.2291e+00],
        [ 1.6954e+00, -4.7616e+00,  9.8198e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [ 2.8994e+00,  6.8675e+00, -9.5987e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-5.7347e+00, -8.6908e+00,  9.2878e+00],
        [-1.9714e+00,  7.8563e+00, -1.8084e+00],
        [ 1.1644e+00, -1.1039e+01,  7.8511e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 8.8091e+00, -7.9037e+00,  3.6313e-01],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 2.7685e+00, -1.0898e+01, -6.8928e+00],
        [-4.4503e+00,  7.1378e+00,  3.3751e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 2.9461e+00, -4.4077e+00, -4.8308e-01],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [-5.3382e+00,  6.6533e+00,  7.8054e+00],
        [ 3.9515e+00, -3.8425e+00,  6.1729e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-9.4731e+00, -7.1201e-01, -6.0271e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 9.0652e+00,  9.8597e+00, -3.5949e+00],
        [-9.3739e+00,  3.9822e-01, -4.2564e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 4.3377e+00,  5.7570e+00,  1.3712e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-9.2687e+00, -2.9794e+00,  6.4365e-01],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 5.9883e+00,  3.1536e+00,  6.2320e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [ 6.9284e+00,  1.1706e+01, -3.7323e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 4.2105e-02, 7.0052e-02,  ..., 3.3146e-01, 2.5625e-01,
         6.2813e-01],
        [4.2105e-02, 3.0339e-04, 3.9631e-02,  ..., 2.8958e-01, 2.6580e-01,
         6.2675e-01],
        [7.0052e-02, 3.9631e-02, 0.0000e+00,  ..., 2.7745e-01, 3.0486e-01,
         5.9204e-01],
        ...,
        [3.3146e-01, 2.8958e-01, 2.7745e-01,  ..., 0.0000e+00, 4.2952e-01,
         7.1905e-01],
        [2.5625e-01, 2.6580e-01, 3.0486e-01,  ..., 4.2952e-01, 0.0000e+00,
         8.8375e-01],
        [6.2813e-01, 6.2675e-01, 5.9204e-01,  ..., 7.1905e-01, 8.8375e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.88683436, 14.02526505, 14.02532758, 13.97779839, 14.10045797,
       13.96987806,  6.23262289,  9.15253344,  6.27715105, 12.3344536 ,
       11.05672136,  8.72896381,  6.50409372,  5.05444288, 13.55868684,
        7.4544866 , 12.00995898,  9.36378438, 12.30858677, 13.25336001,
       14.03505624, 14.03618649, 11.11969815, 13.09426142,  8.94395628,
       13.89671314, 11.28093507,  9.19837792,  8.34949383,  8.87587457,
       10.61417181,  4.73784222, 11.56146873, 10.412299  ,  8.09984231,
       11.10066126, 10.61417181,  8.53501777, 11.54071859, 14.03505624,
       11.83509372,  6.80370254,  9.17016857, 10.86693708,  7.27759233,
       10.26545792, 11.24434374,  8.41148722,  6.1463896 ,  4.78544875,
        5.30158497,  9.48969978, 12.70489563, 10.90638066,  8.53014915,
        5.51177501, 10.76154256, 13.40850328, 12.4217523 ,  5.21280156,
       12.1460931 , 10.48315436,  9.85384623,  9.4997987 ,  7.72291946,
        4.89869205,  9.97822473,  4.74770899, 10.4606265 ,  7.15990676,
       11.52733581,  7.92607342, 13.39372426,  9.38232121, 12.52372629,
        9.14044984,  6.1561646 , 11.50791859, 14.03505624,  7.20826219,
       10.67132524,  9.73579738, 11.11969815,  8.38252937,  6.29034958,
       10.84437705,  5.87305005,  7.16747878,  9.15866607,  6.33075682,
        6.29034958, 12.18045978,  9.36378438,  6.48486171, 13.06498554,
        4.7663526 ,  6.77777606, 10.8207297 ,  6.34059021, 13.01728313,
        6.81084724,  6.76791483,  7.23276582, 12.33167382, 13.68672855,
       13.60275211])}
{'P': tensor([[1.8964e-03, 2.4767e-01, 1.9990e-01,  ..., 2.8272e-07, 3.9240e-08,
         1.2625e-01],
        [4.0590e-06, 3.6511e-03, 9.8090e-01,  ..., 3.8971e-07, 4.0746e-08,
         1.6284e-05],
        [2.9197e-04, 7.8055e-04, 3.2847e-01,  ..., 3.2361e-05, 3.2463e-06,
         6.8759e-04],
        ...,
        [7.6086e-04, 4.6490e-04, 4.0418e-04,  ..., 4.9647e-02, 2.7465e-04,
         5.0588e-04],
        [3.2528e-03, 1.3137e-03, 1.0926e-03,  ..., 3.4329e-05, 2.4901e-02,
         2.6198e-03],
        [9.9050e-06, 1.2550e-01, 1.8912e-01,  ..., 2.2992e-07, 3.1861e-08,
         1.9886e-03]]), 'X': tensor([[  3.3633,  13.4794,   0.5193],
        [  1.1350,  14.0104,   0.6321],
        [  0.6142,  14.0802,   0.5407],
        [  1.4306,  13.9620,  -0.2075],
        [  2.3373,  13.7721,  -0.2442],
        [  0.7275,  14.0667,  -0.9815],
        [ -9.1658,  -3.4155,  -2.8880],
        [  2.4572,   7.1085,   7.7376],
        [  2.9260,  -4.4162,   8.3808],
        [ -4.5778,  -9.5013,  -5.0667],
        [  5.9548,  -1.2273,  -2.6520],
        [ -6.8466,  -7.6488,   1.9648],
        [ 11.7349,  -3.5838,   2.1644],
        [ -2.0721,   7.8432,   6.6380],
        [  6.2902,   0.0451,   3.7116],
        [  7.4132,  11.3526,   0.3322],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.8745,   2.3097,   3.5600],
        [  5.8979,   3.3958,  -3.8713],
        [ -1.0482, -10.8487,  -5.9066],
        [ 12.4877,  -0.7222,   8.8347],
        [  3.5965,  -4.0754,  -3.5035],
        [  6.3516,   0.6425,   0.3425],
        [  2.2076,  13.8032,  -1.5099],
        [ -9.1813,  -3.3550,   1.8827],
        [  6.3589,   0.8239,  -8.4230],
        [ -6.0844,  -8.3955,   0.7705],
        [ -3.9508,  -9.8515,   1.7923],
        [  6.3460,   1.4252,   1.2291],
        [  7.3592,  11.3937,   6.2407],
        [  6.1575,  12.2046,   3.3257],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.4159,  -2.0739,   9.6841],
        [  8.4691,  10.4571,  -5.0650],
        [ -1.6275, -10.7221,   3.2376],
        [  7.0642,  11.6104,  -9.8316],
        [  5.7845,   3.6652,   0.9175],
        [  1.4489,   7.5256,   4.4529],
        [  4.4039,   5.6903,   8.7366],
        [  6.3305,   1.6119,   1.1538],
        [ -5.1377,   6.7751,  -0.4438],
        [  7.1697,  11.5343,  -0.7276],
        [  0.8700,  -4.8132,  -7.8042],
        [ 12.2328,  -2.0376,  -1.1234],
        [  2.4639, -10.9431,  -7.3718],
        [  2.1625,  -4.6697,   7.0658],
        [  0.4996,   7.7738,  -4.4711],
        [  2.3928,   7.1404,  -1.7035],
        [ -8.3653,   3.4184,   9.3454],
        [  6.9653,  -9.3225,  -3.4727],
        [  6.3048,   0.1469,  -9.1691],
        [  5.0348,  -2.8229,   3.5436],
        [ -2.6485, -10.4151,   0.8766],
        [ -3.3999, -10.1152,  -1.2088],
        [ -9.3076,   0.7852,  -6.7010],
        [  1.7659,   7.4127,   4.7871],
        [  8.0877,  -8.5297,  -1.1050],
        [  6.2365,  -0.2578,  -6.0212],
        [ -8.9282,   2.1648,   8.4894],
        [ -1.6275, -10.7221,   3.2376],
        [  9.9217,  -6.7006,  -5.6728],
        [ -0.9715, -10.8630,   0.3555],
        [  0.5175,  14.0910,   4.6540],
        [  2.1379,  -4.6757,   6.4981],
        [ 12.6000,   1.3839,  -3.2992],
        [ -3.5029,   7.5138,   8.1853],
        [  8.9284,  10.0036,  -6.9210],
        [ -0.5739,   7.9056,   0.6382],
        [ 11.6634,   5.8050,   9.1590],
        [  0.3354,  -4.7708,  -4.9125],
        [ -1.5451,   7.8975,   7.6104],
        [ 11.6801,  -3.7196,  -0.9181],
        [  4.7813,  -3.1139,   8.2691],
        [  6.3569,   0.7591,   6.5011],
        [  9.9058,  -6.7203,   5.6928],
        [ -6.3252,  -8.1760,  -1.0891],
        [ 11.0373,   7.1299,  -8.5895],
        [  6.3583,   1.1729,   8.7835],
        [  1.1006,   7.6318,  -2.4877],
        [  8.2154,  -8.4262,   8.0171],
        [ 11.7349,  -3.5838,   2.1644],
        [  2.3261,  13.7749,   5.9121],
        [ -7.6734,   4.4980,  -5.4248],
        [  5.9883,   3.1536,   6.2320],
        [ -8.0330,  -6.0765,   6.7945],
        [  6.0788,  -0.8710,  -6.8719],
        [  5.9883,   3.1536,   6.2320],
        [ -8.8835,  -4.3178,   9.3080],
        [  6.2038,   2.4020,  -3.1692],
        [ 12.5199,  -0.4711,   3.0436],
        [ 10.1798,   8.5002,  -5.8893],
        [  2.0039,   7.3173,  -1.5885],
        [  0.9788,  -4.8145,  -3.8661],
        [ -6.7745,  -7.7264,   8.4684],
        [ 10.9599,  -5.1849,  -3.5526],
        [ 10.4758,  -5.9539,   1.3667],
        [ 12.2328,  -2.0376,  -1.1234],
        [ -8.9788,   2.0208,  -3.2867],
        [  7.0515,  11.6194,   8.4289],
        [ -3.0004,   7.6607,  -7.2640],
        [  6.3205,   1.7068,  -3.8619],
        [  3.1480,  -4.3165,  -7.8002],
        [  2.6321,  -4.5296,  -2.5159],
        [  6.1973,  -0.4372,   6.5720],
        [ 12.4691,   2.8525,  -5.5110],
        [  2.3373,  13.7721,  -0.2442]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.7978e-02, 1.7025e-01,  ..., 4.3302e-01, 5.0983e-01,
         5.9600e-02],
        [5.7978e-02, 0.0000e+00, 1.1961e-01,  ..., 3.8887e-01, 5.3452e-01,
         8.8791e-02],
        [1.7025e-01, 1.1961e-01, 0.0000e+00,  ..., 3.8318e-01, 5.3276e-01,
         1.6426e-01],
        ...,
        [4.3302e-01, 3.8887e-01, 3.8318e-01,  ..., 0.0000e+00, 9.1121e-01,
         4.7764e-01],
        [5.0983e-01, 5.3452e-01, 5.3276e-01,  ..., 9.1121e-01, 0.0000e+00,
         4.5229e-01],
        [5.9600e-02, 8.8791e-02, 1.6426e-01,  ..., 4.7764e-01, 4.5229e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.89264822, 14.05633441, 14.09357547, 14.03505624, 13.96905756,
       14.08549802,  9.78147542,  7.52117132,  5.29755487, 10.54658597,
        6.07992623, 10.26545792, 12.26997165,  8.11227668,  6.29034958,
       13.55868684,  6.49538841,  9.17016857,  6.80559669, 10.8992524 ,
       12.5085995 ,  5.43544626,  6.38399789, 13.97857591,  9.7751189 ,
        6.41202986, 10.36844813, 10.61417181,  6.50409372, 13.56367857,
       13.66991027,  8.29203356,  9.64157461, 13.45642379, 10.84493732,
       13.59056613,  6.84796202,  7.66377554,  7.19534146,  6.53250937,
        8.50278896, 13.58101794,  4.89121805, 12.4013153 , 11.21703654,
        5.14607014,  7.78980299,  7.53063885,  9.0368458 , 11.63724792,
        6.30647746,  5.77219077, 10.74655346, 10.67132524,  9.34061938,
        7.62011734, 11.75438006,  6.24187044,  9.18690018, 10.84493732,
       11.97238624, 10.90638066, 14.10045797,  5.14124612, 12.67576792,
        8.29022197, 13.40850328,  7.92644746, 13.02818702,  4.78258092,
        8.04718433, 12.25806559,  5.70591986,  6.4020404 , 11.9702762 ,
       10.33712258, 13.1399181 ,  6.46560181,  7.71075226, 11.76830981,
       12.26997165, 13.96987806,  8.89458273,  6.76791483, 10.07238463,
        6.14086582,  6.76791483,  9.87720102,  6.65259775, 12.52875793,
       13.26209545,  7.58668144,  4.91296205, 10.27572673, 12.12448912,
       12.0495521 , 12.4013153 ,  9.20340031, 13.59170809,  8.22728399,
        6.54693987,  5.34250225,  5.23878371,  6.21275016, 12.79126785,
       13.96905756])}
{'P': tensor([[1.9790e-03, 4.4998e-01, 2.5556e-01,  ..., 2.5413e-07, 1.6833e-06,
         1.5941e-05],
        [7.1644e-06, 3.7409e-03, 4.9894e-01,  ..., 4.2887e-07, 2.4020e-06,
         2.1263e-05],
        [6.1139e-06, 1.3053e-05, 4.2495e-03,  ..., 4.2508e-07, 2.1208e-06,
         1.5280e-05],
        ...,
        [1.0893e-03, 8.5917e-04, 6.3884e-04,  ..., 1.1887e-01, 2.4282e-02,
         9.9922e-03],
        [1.9447e-02, 1.6543e-02, 1.4162e-02,  ..., 6.7377e-04, 9.7290e-02,
         4.3735e-03],
        [1.5151e-03, 1.9058e-03, 1.9403e-03,  ..., 1.6774e-04, 2.6348e-04,
         7.8184e-02]]), 'X': tensor([[  3.3094,  13.4969,  -3.6507],
        [  2.3589,  13.7668,  -3.6501],
        [  1.4150,  13.9647,  -4.0915],
        [  0.4119,  14.1019,  -4.1870],
        [  1.6308,  13.9254,  -2.0984],
        [  2.0937,  13.8293,  -2.3533],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.7290,   7.8825,  -3.3831],
        [ -8.0087,   4.0173,   9.1091],
        [ -8.6112,   2.9319,  -7.9943],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 12.1504,   4.3674,  -7.2998],
        [  5.6095,  -1.9702,  -7.8233],
        [ -4.8665,  -9.3204,  -5.2564],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 12.2311,  -2.0445,   1.3545],
        [  9.9217,  -6.7006,  -5.6728],
        [  9.0129,  -7.7072,   9.1400],
        [  2.5382,   7.0672,   4.5097],
        [  2.7976,  -4.4682,  -4.2020],
        [  2.6614,  -4.5192,  -9.3641],
        [ -9.4283,  -1.9585,  -0.9129],
        [  4.7156,  -3.1831,  -8.0224],
        [  9.1820,   9.7334,  -2.7932],
        [ -9.1459,   1.4780,  -4.2403],
        [  2.7295,  13.6703,  -8.9328],
        [  3.2701,   6.6338,   9.9209],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -2.0445,   7.8469,  -0.5857],
        [ -2.2188,   7.8218,   8.6064],
        [  3.2701,   6.6338,   9.9209],
        [ 10.0619,   8.6627,  -9.2884],
        [ -7.9298,   4.1367,  -3.9773],
        [  1.5246,  -4.7838,  -5.2095],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -8.0829,  -5.9940,   5.0564],
        [  6.3534,   1.2990,  -7.2942],
        [  6.1741,  -0.5329,   3.1913],
        [  5.5158,  -2.1337,   3.5246],
        [  6.3031,   1.8477,  -7.1706],
        [ 11.4495,   6.3026,   6.8257],
        [ -8.9788,   2.0208,  -3.2867],
        [ -3.6381,   7.4683,   7.3544],
        [  2.6614,  -4.5192,  -9.3641],
        [  0.7685,  -4.8098,  -2.9310],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.7409,  -4.6837,  -6.5840],
        [ -1.5408,   7.8978,   5.3470],
        [  2.1771, -10.9776,  -3.2007],
        [ 11.3118,   6.5957,  -4.5667],
        [ 11.4495,   6.3026,   6.8257],
        [ -7.5603,   4.6459,   8.2002],
        [ -3.4986,   7.5152,   9.0712],
        [  3.4920,  -4.1366,  -8.3955],
        [ 12.5770,   1.8512,  -0.6578],
        [ -8.6603,   2.8250,   5.9181],
        [ -9.4541,  -0.3327,   8.8246],
        [  4.1140,  -3.7220,  -0.1273],
        [  0.6757, -11.0361,  -1.3047],
        [ -4.4779,  -9.5609,  -2.4505],
        [  8.3791,  10.5409,   5.5590],
        [  4.1999,  -3.6543,  -4.1998],
        [ -3.1671,   7.6158,   6.0765],
        [  4.1537,   5.9340,  -6.0423],
        [  4.9152,  12.8653,  -1.4459],
        [ -0.8569,   7.9156,  -6.5557],
        [ 10.2712,   8.3707,  -4.9743],
        [ -0.1963,   7.8765,  -1.1371],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.8994,   6.8675,  -9.5987],
        [  7.1821,  -9.1845,   9.8548],
        [  8.3718,  -8.2953,   8.4666],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.4697,  -4.5836,   1.4593],
        [ -7.9298,   4.1367,  -3.9773],
        [  4.6063,  13.0051,  -3.1456],
        [ 12.0042,  -2.8350,   2.6864],
        [  6.3542,   0.6934,  -3.3840],
        [ -1.6275, -10.7221,   3.2376],
        [  5.5158,  -2.1337,   3.5246],
        [  6.3087,   1.8053,  -1.9481],
        [ -9.1231,  -3.5739,   8.4708],
        [ -2.0229, -10.6163,   8.6930],
        [  2.4356,  -4.5942,  -5.0808],
        [ -4.2040,   7.2486,   2.9875],
        [ -9.3419,   0.5972,  -1.2841],
        [  8.7715,  -7.9390,   5.2900],
        [ 10.9875,   7.2207,  -8.8549],
        [ -6.1899,  -8.3011,  -7.9269],
        [  4.8432,   5.1968,   1.5006],
        [  6.3497,   1.3680,  -1.9076],
        [ -8.8835,  -4.3178,   9.3080],
        [ -2.2508, -10.5479,  -2.7235],
        [  1.7122,  -4.7591,  -3.6734],
        [  2.5829,  -4.5466,  -8.4015],
        [ -5.1974,   6.7396,   3.3484],
        [ -5.3328,  -8.9992,   7.3440],
        [  2.0039,   7.3173,  -1.5885],
        [  5.8836,   3.4315,  -4.0327],
        [  6.0467,  -9.8369,  -6.5816],
        [ 11.5568,   6.0602,   9.3244],
        [  4.5503,   5.5359,   7.4735],
        [  5.6202,  -1.9508,   0.7471],
        [  5.8318,   3.5567,  -4.7692],
        [  2.5382,   7.0672,   4.5097]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0194, 0.0474,  ..., 0.4617, 0.2644, 0.8081],
        [0.0194, 0.0000, 0.0280,  ..., 0.4654, 0.2534, 0.8131],
        [0.0474, 0.0280, 0.0000,  ..., 0.4695, 0.2370, 0.8186],
        ...,
        [0.4617, 0.4654, 0.4695,  ..., 0.0000, 0.3380, 0.3497],
        [0.2644, 0.2534, 0.2370,  ..., 0.3380, 0.0000, 0.6727],
        [0.8081, 0.8131, 0.8186,  ..., 0.3497, 0.6727, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.89671314, 13.96746816, 14.03618649, 14.10796466, 14.02058621,
       13.98691426,  9.53015376,  8.06990483,  8.95983586,  9.09661212,
       12.4013153 , 12.05395056, 12.91143491,  5.94541803, 10.51437947,
       12.18045978, 12.40074901, 11.97238624, 11.85890664,  7.50917551,
        5.27178363,  5.24461068,  9.62958973,  5.68942572, 13.38091121,
        9.26455795, 13.94008948,  7.39599986, 10.32715715,  8.10886021,
        8.1303945 ,  7.39599986, 13.27717982,  8.94395628,  5.0209005 ,
       12.18045978, 10.06286074,  6.48486171,  6.19707944,  5.91408504,
        6.56833864, 13.06957425,  9.20340031,  8.30729352,  5.24461068,
        4.87083348, 13.41713661,  9.91668182,  8.04664886, 11.1913598 ,
       13.09426142, 13.06957425,  8.8737556 ,  8.28968067,  5.41348121,
       12.7125079 ,  9.10945678,  9.45995955,  5.54776934, 11.05672136,
       10.5575601 , 13.46553623,  5.5671389 ,  8.24808832,  7.24326992,
       13.77222752,  7.96181288, 13.25017211,  7.8788965 , 12.18045978,
        7.4544866 , 11.65923396, 11.785566  ,  9.55720651,  5.20659892,
        8.94395628, 13.79676654, 12.3344536 ,  6.39187285, 10.84493732,
        5.91408504,  6.5618968 ,  9.79814997, 10.80728788,  5.19986797,
        8.37954853,  9.36093562, 11.83075005, 13.14777433, 10.35485331,
        7.10378919,  6.49538841,  9.87720102, 10.78533838,  5.05774295,
        5.22903117,  8.51088913, 10.4606265 ,  7.58668144,  6.81117802,
       11.54672755, 13.0493368 ,  7.16598273,  5.94909709,  6.83083863,
        7.50917551])}
{'P': tensor([[2.2646e-03, 3.4774e-01, 3.4256e-01,  ..., 1.4693e-08, 4.6131e-07,
         3.4512e-07],
        [5.4407e-06, 4.2122e-03, 1.2938e-05,  ..., 2.0944e-08, 6.3524e-07,
         6.5733e-07],
        [1.0284e-05, 4.9235e-01, 3.6914e-03,  ..., 2.1255e-08, 6.3410e-07,
         5.8538e-07],
        ...,
        [1.5019e-03, 6.8692e-04, 1.0641e-03,  ..., 1.8368e-01, 1.1735e-02,
         3.9652e-04],
        [6.0456e-03, 3.7599e-03, 4.6150e-03,  ..., 1.4940e-03, 1.9326e-01,
         1.1814e-03],
        [1.6671e-04, 1.1555e-04, 1.4404e-04,  ..., 2.3574e-02, 1.8212e-02,
         1.8420e-01]]), 'X': tensor([[ 3.2809e+00,  1.3506e+01, -4.0199e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-2.4956e+00, -1.0468e+01,  3.5632e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [ 9.5261e+00, -7.1675e+00,  1.4206e+00],
        [ 6.5428e+00, -9.5728e+00, -2.9897e+00],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 5.6093e+00, -1.9705e+00,  6.4807e+00],
        [-7.5720e+00,  4.6310e+00,  2.1717e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 2.2182e+00,  1.3801e+01,  5.1889e+00],
        [ 5.8275e+00, -1.5334e+00,  2.0124e+00],
        [ 1.0596e+01,  7.8824e+00, -3.7006e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 2.1120e+00, -4.6818e+00,  8.6144e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 6.2419e+00, -2.3128e-01,  4.2440e+00],
        [ 7.3592e+00,  1.1394e+01,  6.2407e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [ 8.0877e+00, -8.5297e+00, -1.1050e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [ 9.1820e+00,  9.7334e+00, -2.7932e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-8.0534e+00, -6.0429e+00,  2.4583e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-3.0008e+00, -1.0283e+01, -8.9480e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-8.6381e+00, -4.9238e+00, -4.4077e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 1.1139e+00,  7.6281e+00, -2.5680e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 6.3497e+00,  1.3680e+00, -1.9076e+00],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 1.0499e+01, -5.9193e+00, -7.5153e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [ 6.2576e+00,  2.1359e+00, -3.3398e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 1.1508e+01, -4.1177e+00,  2.3885e-01],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0330, 0.0145,  ..., 0.4695, 0.2313, 0.4881],
        [0.0330, 0.0000, 0.0280,  ..., 0.4555, 0.2001, 0.4716],
        [0.0145, 0.0280, 0.0000,  ..., 0.4550, 0.2203, 0.4736],
        ...,
        [0.4695, 0.4555, 0.4550,  ..., 0.0000, 0.3480, 0.0440],
        [0.2313, 0.2001, 0.2203,  ..., 0.3480, 0.0000, 0.3465],
        [0.4881, 0.4716, 0.4736,  ..., 0.0440, 0.3465, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.89886469, 14.03618649, 13.96746816, 14.10796466, 14.09176259,
       14.02058621, 13.36800073,  9.62958973, 11.72802491, 13.36092624,
       10.76154256, 10.01804223,  9.60080639, 13.37121923, 11.65923396,
        7.83486943, 12.16901615,  7.34250948, 11.92133129, 11.59512194,
       10.80728788,  5.39723241,  5.94535529,  8.87587457,  8.04718433,
       11.22993218, 13.97779839,  6.02588517, 13.20597932,  7.62011734,
        5.13616224,  5.19065439,  6.7593917 ,  6.24614905, 13.56367857,
       10.87409005, 10.0674887 ,  8.51928531, 13.04217487,  7.92607342,
       14.05687815,  9.78286733, 11.75438006,  7.16058587, 10.63038565,
        9.5014146 ,  8.36445758,  5.05981914,  5.79562691,  8.97052115,
       11.00812996,  5.83560009,  5.18047638, 13.38091121, 11.48434767,
       10.06850677,  7.50917551, 13.69162795,  5.39450305, 10.00802627,
       13.52430176, 10.71162681, 13.66991027, 10.71162681,  6.46560181,
        9.94285511, 13.6829843 ,  7.60463349,  7.37576154, 11.73567759,
        6.81084724, 13.45124917, 12.5085995 ,  6.81084724,  5.64464807,
        7.16598273, 12.30695098, 11.4088043 ,  8.40445872,  7.70897675,
        7.96575782, 12.12448912, 11.69264243,  6.49538841,  5.64464807,
        5.24461068,  5.39723241,  9.62958973,  5.52416368, 12.05300943,
       10.57302546,  7.57685908,  9.09502385,  7.78980299,  6.61211224,
        8.77913428,  5.05989547,  9.64157461,  9.43112957, 11.51557166,
        8.41171369,  9.7712321 , 10.08001894, 12.22276278,  6.24187044,
        5.07306274])}
{'P': tensor([[3.9053e-03, 3.5453e-01, 3.2184e-01,  ..., 4.6429e-07, 3.8536e-08,
         3.2107e-07],
        [8.1067e-06, 4.2158e-03, 9.7863e-01,  ..., 4.1237e-07, 4.6604e-08,
         3.0868e-07],
        [4.0221e-04, 7.6035e-04, 4.1526e-01,  ..., 3.4570e-05, 5.0980e-06,
         2.6894e-05],
        ...,
        [6.7621e-04, 5.7825e-04, 4.5506e-04,  ..., 7.0709e-02, 1.3482e-05,
         3.1051e-03],
        [9.3910e-07, 5.2582e-07, 4.0514e-07,  ..., 6.3138e-05, 8.9593e-03,
         7.8034e-05],
        [2.7908e-04, 2.2997e-04, 1.8038e-04,  ..., 3.8214e-01, 1.8254e-05,
         6.0493e-02]]), 'X': tensor([[  3.2692,  13.5098,  -4.9032],
        [  1.4150,  13.9647,  -4.0915],
        [  0.4119,  14.1019,  -4.1870],
        [  0.6396,  14.0772,  -6.3786],
        [  2.3589,  13.7668,  -3.6501],
        [  2.0449,  13.8402,  -7.1642],
        [ -9.1624,  -3.4286,   2.0647],
        [  2.1120,  -4.6818,   8.6144],
        [  7.5930,  -8.9037,   7.2323],
        [  6.3606,   0.9204,   3.2525],
        [  2.0449,  13.8402,  -7.1642],
        [  6.3266,   0.3292,   9.3502],
        [  6.3236,   0.3011,  -4.6668],
        [ 10.5678,   7.9261,  -0.6100],
        [ -6.1914,  -8.2997,   8.5380],
        [  5.7963,  -1.6018,   1.8856],
        [  4.9452, -10.3187,  -7.8621],
        [ -5.6651,   6.4380,   1.3660],
        [  5.4426, -10.1181,  -5.1268],
        [  3.3069,   6.6089,   8.7377],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.4867,  12.5816,   2.9549],
        [ -0.9715, -10.8630,   0.3555],
        [  1.9505,  -4.7171,   9.9868],
        [  3.0719, -10.8449,  -4.1673],
        [ 11.7795,  -3.4697,   3.5624],
        [ -4.3359,  -9.6431,  -9.0616],
        [  5.3804,  -2.3504,   7.2744],
        [ -8.3717,  -5.4769,  -9.2228],
        [  2.5010,  -4.5736,  -0.8989],
        [  2.3373,  -4.6232,  -5.2246],
        [ -9.4505,  -1.7081,  -0.7039],
        [  5.1203,  -2.7151,   6.2110],
        [  4.1999,  -3.6543,  -4.1998],
        [ -3.2053, -10.1993,  -3.8736],
        [  6.3589,   0.8239,  -8.4230],
        [  2.2373,  -4.6506,  -1.7334],
        [ -1.0407,   7.9166,  -6.6021],
        [ -0.5709,   7.9055,   7.6089],
        [  6.3413,   0.4883,  -6.8416],
        [ -8.3535,  -5.5118,   9.8399],
        [ -9.4756,  -0.8166,  -9.6021],
        [ -1.3178, -10.7939,   0.4236],
        [  6.3583,   1.1729,   8.7835],
        [  7.2340,  -9.1505,  -5.8321],
        [  1.1274,  14.0116,   8.1753],
        [  9.5018,  -7.1945,  -1.1892],
        [ -8.4297,  -5.3637,  -9.3962],
        [  5.1793,   4.7461,   0.3899],
        [  7.3724,  -9.0577,  -8.9843],
        [  7.8375,  -8.7240,  -9.8913],
        [  5.8979,   3.3958,  -3.8713],
        [  4.2041,  -3.6509,  -4.9488],
        [ 12.3025,  -1.7440,  -6.6632],
        [  4.1297,  -3.7098,  -3.9838],
        [ 11.5966,   5.9666,   6.0083],
        [  2.8602,   6.8905,   5.8779],
        [ 12.0845,  -2.5772,  -4.6073],
        [  2.2659,  13.7894,   6.5436],
        [ -9.2687,  -2.9794,   0.6436],
        [  3.9515,  -3.8425,   6.1729],
        [  4.9957,  -2.8705,   6.5978],
        [  4.3554,   5.7393,  -9.0766],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.4703,  -4.2006,   7.4086],
        [  0.8572,  14.0501,  -8.3364],
        [  2.4639, -10.9431,  -7.3718],
        [ -8.9178,   2.1936,   7.5388],
        [  3.1064,  13.5607,   6.7605],
        [ 12.1504,   4.3674,  -7.2998],
        [ -9.0961,   1.6529,   3.1553],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 12.5650,   2.0132,   4.6272],
        [  5.3858,   4.4260,  -9.6769],
        [ -9.3739,   0.3982,  -4.2564],
        [ 10.7566,   7.6212,  -2.3279],
        [  2.0547,   7.2957,  -1.9343],
        [  5.3602,  -2.3808,   7.2098],
        [  2.3376,  -4.6231,   9.8436],
        [  4.5340,  -3.3629,   8.9652],
        [ -5.1974,   6.7396,   3.3484],
        [ 10.1983,  -6.3430,  -0.3200],
        [  2.3261,  13.7749,   5.9121],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.3616,  -2.4750,   3.7765],
        [  5.4167, -10.1293,   6.2156],
        [ 11.0275,  -5.0666,   8.4154],
        [  6.2804,   2.0019,   0.8607],
        [ -3.2736, -10.1703,   3.8712],
        [ -5.9872,   6.2037,  -9.4477],
        [  1.9505,  -4.7171,   9.9868],
        [  1.7659,   7.4127,   4.7871],
        [  6.0467,  -9.8369,  -6.5816],
        [  5.6258,  -1.9404,  -6.0715],
        [  5.4336,  -2.2676,   5.6933],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.3118,   4.5452,   4.2114],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -6.4464,   5.8272,   7.5038],
        [ 11.2517,  -4.6495,   2.9800],
        [  0.2571,  -4.7594,   1.5618],
        [ -9.0574,   1.7801,   8.8986],
        [  6.2902,   0.0451,   3.7116],
        [  2.4639, -10.9431,  -7.3718],
        [  5.8902,  -1.3887,   4.0388]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 3.9367e-02, 1.5102e-01,  ..., 7.1081e-01, 7.5496e-01,
         7.2743e-01],
        [3.9367e-02, 0.0000e+00, 1.3254e-01,  ..., 6.7210e-01, 7.5733e-01,
         6.8874e-01],
        [1.5102e-01, 1.3254e-01, 0.0000e+00,  ..., 6.1329e-01, 6.4045e-01,
         6.2935e-01],
        ...,
        [7.1081e-01, 6.7210e-01, 6.1329e-01,  ..., 1.5170e-04, 9.5983e-01,
         1.6711e-02],
        [7.5496e-01, 7.5733e-01, 6.4045e-01,  ..., 9.5983e-01, 0.0000e+00,
         9.6896e-01],
        [7.2743e-01, 6.8874e-01, 6.2935e-01,  ..., 1.6711e-02, 9.6896e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.89974693, 14.03618649, 14.10796466, 14.09176259, 13.96746816,
       13.99048094,  9.78285424,  5.13616224, 11.70168755,  6.42689705,
       13.99048094,  6.33517349,  6.33075682, 13.20988996, 10.35464916,
        6.01356004, 11.4424816 ,  8.57560275, 11.4890861 ,  7.39004226,
       11.54672755, 13.72594022, 10.90638066,  5.10446653, 11.27160035,
       12.27991991, 10.57302546,  5.87132718, 10.00410625,  5.21280156,
        5.18047638,  9.60359158,  5.79562691,  5.5671389 , 10.6910768 ,
        6.41202986,  5.16077764,  7.98469166,  7.92607342,  6.36004393,
       10.00802627,  9.51074763, 10.87409005,  6.46560181, 11.66453059,
       14.05687815, 11.9182956 ,  9.99145096,  7.02496223, 11.67875855,
       11.72749923,  6.80559669,  5.56809335, 12.42554808,  5.5513072 ,
       13.04156075,  7.46053281, 12.35625694, 13.97430191,  9.73579738,
        5.51177501,  5.76165934,  7.20482224,  9.41244865, 12.21532282,
       14.07623266, 11.21703654,  9.18358562, 13.91198191, 12.91143491,
        9.24501829,  9.62238828,  9.55720651, 12.72524615,  6.97106912,
        9.38232121, 13.18278606,  7.57946459,  5.86519525,  5.1805345 ,
        5.64500512,  8.51088913, 12.00995898, 13.96987806, 13.03002338,
        9.68324553, 11.48663767, 12.13568598,  6.59176018, 10.68416401,
        8.62162372,  5.10446653,  7.62011734, 11.54672755,  5.9510489 ,
        5.88782913,  8.30125201,  6.99097013, 10.32715715,  8.68979052,
       12.17451685,  4.7663526 ,  9.23071698,  6.29034958, 11.21703654,
        6.05165248])}
{'P': tensor([[2.3075e-03, 4.6084e-01, 2.2698e-01,  ..., 1.6150e-08, 5.4956e-07,
         1.8587e-07],
        [8.0324e-05, 3.4138e-02, 2.1218e-04,  ..., 2.4308e-07, 6.9271e-06,
         2.8889e-06],
        [3.3777e-05, 7.4149e-01, 6.0573e-03,  ..., 4.2888e-08, 1.3285e-06,
         5.2559e-07],
        ...,
        [8.1063e-06, 6.0930e-06, 6.6698e-06,  ..., 2.8415e-02, 9.8185e-04,
         1.9668e-04],
        [1.2681e-03, 1.1534e-03, 1.2725e-03,  ..., 1.5589e-05, 4.3856e-02,
         2.5042e-04],
        [6.4337e-05, 5.8021e-05, 6.2905e-05,  ..., 2.5314e-03, 2.8111e-02,
         1.7771e-01]]), 'X': tensor([[ 3.1064e+00,  1.3561e+01,  6.7605e+00],
        [ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 2.5318e+00,  1.3723e+01,  6.3692e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-3.9508e+00, -9.8515e+00,  1.7923e+00],
        [ 8.9401e-01, -1.1040e+01, -9.7829e-01],
        [ 3.0876e+00,  6.7527e+00, -5.3540e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-8.4824e+00, -5.2575e+00, -7.9550e-02],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-8.6112e+00,  2.9319e+00, -7.9943e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 5.9598e-01, -1.1033e+01,  3.1144e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [ 9.7328e+00,  9.0902e+00, -8.9172e-01],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [ 1.2259e+01,  3.9425e+00,  2.0383e+00],
        [ 5.9655e+00,  3.2172e+00, -7.5953e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [ 4.5325e+00, -3.3643e+00,  6.7699e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 8.2422e+00, -8.4041e+00, -1.4962e+00],
        [ 4.7813e+00, -3.1139e+00,  8.2691e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-9.4283e+00, -1.9585e+00, -9.1290e-01],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 6.0955e+00,  2.8210e+00, -4.3321e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-7.4808e+00,  4.7462e+00,  2.0266e+00],
        [ 6.2242e+00, -3.1714e-01,  7.4978e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 6.3587e+00,  1.1596e+00, -3.6629e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 6.1929e+00,  2.4501e+00, -6.9352e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [-7.0253e+00, -7.4493e+00, -1.5978e+00],
        [ 6.9693e+00, -9.3200e+00, -8.3822e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [-9.1813e+00, -3.3550e+00,  1.8827e+00],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [ 6.3246e+00,  1.6696e+00, -1.5439e-01],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0764, 0.0393,  ..., 0.8114, 0.7988, 0.9993],
        [0.0764, 0.0000, 0.0437,  ..., 0.7386, 0.7395, 0.9387],
        [0.0393, 0.0437, 0.0000,  ..., 0.7823, 0.7605, 0.9608],
        ...,
        [0.8114, 0.7386, 0.7823,  ..., 0.0000, 0.7339, 0.8258],
        [0.7988, 0.7395, 0.7605,  ..., 0.7339, 0.0000, 0.2013],
        [0.9993, 0.9387, 0.9608,  ..., 0.8258, 0.2013, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.91198191, 13.97430191, 13.95472331, 14.0393949 , 14.05687815,
       13.96987806, 11.86239961,  5.18047638, 10.61417181, 11.07637531,
        7.42512621,  5.19986797, 13.36800073,  8.20436764,  7.26074925,
        6.14086582,  9.97964013,  8.34949383, 10.14436912,  7.83486943,
        9.94173708,  9.09661212,  4.78544875,  9.15866607, 11.04953767,
        6.80370254,  9.78285424, 11.33837396, 13.31764181,  8.28968067,
       13.20988996,  6.10360073,  6.4020404 ,  5.89074104,  6.83515201,
       13.03002338, 12.64235359, 13.69162795,  4.76969364, 12.87752052,
        6.77777606,  4.92497514,  5.64464807, 13.45124917,  6.23262289,
       12.5769659 ,  8.36995953,  8.0550286 ,  9.96391045, 11.77124373,
        5.70591986, 10.46238054, 13.58333857,  8.68979052,  9.9353921 ,
        6.14086582,  8.97052115,  8.99276564,  9.09502385,  9.62958973,
       11.8159254 , 11.73567759,  6.7166297 , 13.40024224,  9.53015376,
       10.48315436,  6.74872981,  5.40689332,  8.92549752, 12.13568598,
       13.14777433,  9.48337063,  8.859416  ,  6.2322769 , 13.58101794,
       13.72594022,  6.46356806,  6.28835101, 13.94008948, 11.38774149,
        7.26074925,  7.92644746,  6.65991877,  8.50268322,  4.96231532,
        9.50913849, 11.57249984, 10.4977132 ,  7.69752262,  9.69792513,
       10.9036226 , 13.20988996,  4.78002665,  5.64865173,  8.09704052,
       13.89671314, 10.23946247, 11.63765273,  9.77588801, 11.56146873,
       12.52875793, 11.83570132,  9.7751189 , 11.48663767,  6.54127431,
        5.04028538])}
{'P': tensor([[5.9499e-03, 2.4011e-01, 2.1682e-01,  ..., 2.4548e-06, 2.6163e-08,
         5.1768e-05],
        [3.6853e-05, 7.0951e-03, 9.4198e-01,  ..., 1.7401e-06, 2.7760e-08,
         3.0524e-05],
        [5.0764e-04, 6.4852e-04, 2.2875e-01,  ..., 4.0907e-05, 8.7407e-07,
         6.8147e-04],
        ...,
        [4.0890e-03, 2.2961e-03, 1.6849e-03,  ..., 9.0299e-02, 2.8548e-05,
         9.0609e-03],
        [3.4327e-06, 1.8326e-06, 1.2441e-06,  ..., 1.0424e-03, 2.4322e-02,
         1.0354e-05],
        [3.1615e-05, 3.1988e-05, 3.5554e-05,  ..., 7.4927e-05, 6.7938e-06,
         5.0037e-02]]), 'X': tensor([[  2.9470,  13.6085,   2.9809],
        [  1.5653,  13.9377,   4.1773],
        [  0.5175,  14.0910,   4.6540],
        [  1.5662,  13.9375,   4.5142],
        [  0.6142,  14.0802,   0.5407],
        [  1.1350,  14.0104,   0.6321],
        [  7.9141,  -8.6656,   9.6948],
        [  6.2036,   2.4030,  -1.2500],
        [ -8.0766,  -6.0046,  -3.1873],
        [  2.7590,  13.6621,   9.4496],
        [ -3.0004,   7.6607,  -7.2640],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -3.5520,   7.4976,   9.1522],
        [ 11.6019,   5.9540,  -5.1715],
        [ 12.2592,   3.9425,   2.0383],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.9470,  13.6085,   2.9809],
        [  5.4167, -10.1293,   6.2156],
        [ -9.4117,   0.1160,   8.2574],
        [ 11.1469,   6.9237,   1.6160],
        [  6.2576,   2.1359,  -3.3398],
        [ 12.5199,  -0.4711,   3.0436],
        [  1.9241,   7.3503,   7.9004],
        [ -9.0604,   1.7707,   1.1066],
        [ -3.1204, -10.2345,   9.2288],
        [ -7.0253,  -7.4493,  -1.5978],
        [ -0.9715, -10.8630,   0.3555],
        [  2.2437,  -4.6489,   6.1430],
        [  4.0457,  -3.7737,  -0.4704],
        [  0.1252, -11.0062,   7.2564],
        [  2.6078, -10.9229,   6.2869],
        [  3.6817,   6.3354,   6.1233],
        [ 10.9931,   7.2107,   7.7933],
        [  2.6970,   6.9826,   9.7350],
        [ -1.3940, -10.7772,   6.7835],
        [ -9.0574,   1.7801,   8.8986],
        [  4.2612, -10.5514,  -6.6268],
        [  6.0305,   3.0296,   0.1375],
        [  9.4554,   9.4245,   9.8800],
        [ -8.5900,   2.9767,   7.4811],
        [  6.0345,   3.0175,  -1.4424],
        [ -4.7389,   6.9953,   3.7638],
        [  6.3413,   0.4883,  -6.8416],
        [ 12.1940,  -2.1881,   7.1499],
        [ -8.8540,  -4.3971,  -5.2144],
        [  6.2902,   0.0451,   3.7116],
        [ 10.4758,  -5.9539,   1.3667],
        [  0.6142,  14.0802,   0.5407],
        [ -9.3384,  -2.6161,  -4.7630],
        [ -3.2486,   7.5925,  -1.3719],
        [  6.3583,   1.1729,   8.7835],
        [  4.7156,  -3.1831,  -8.0224],
        [ -7.0948,   5.1940,   9.8171],
        [  6.2630,   2.1057,  -5.2136],
        [ -8.4267,  -5.3697,   2.4172],
        [ -5.6169,  -8.7844,   8.9057],
        [ -4.6393,   7.0461,   1.1624],
        [ -1.9487,   7.8590,  -3.3330],
        [ -9.4505,  -1.7081,  -0.7039],
        [  8.5093,  -8.1764,  -3.2381],
        [ 12.4877,  -0.7222,   8.8347],
        [  1.5662,  13.9375,   4.5142],
        [ -1.5451,   7.8975,   7.6104],
        [ -8.9942,  -3.9990,   6.0448],
        [  2.6613,   7.0021,   4.5783],
        [  5.9912,   3.1453,  -3.9989],
        [  5.0190,   4.9705,  -7.9752],
        [ -9.1624,  -3.4286,   2.0647],
        [ 11.5568,   6.0602,   9.3244],
        [  1.4306,  13.9620,  -0.2075],
        [  4.3554,   5.7393,  -9.0766],
        [ 10.5057,  -5.9099,  -5.1599],
        [  3.1892, -10.8220,  -6.8409],
        [  3.5284, -10.7484,   5.3347],
        [  9.9756,  -6.6330,  -3.3429],
        [  6.2902,   0.0451,   3.7116],
        [  6.0305,   3.0296,   0.1375],
        [ -9.4753,  -0.8012,  -5.2050],
        [  6.2883,   0.0325,   5.2692],
        [  5.4167, -10.1293,   6.2156],
        [  6.2242,  -0.3171,   7.4978],
        [  5.6523,   3.9443,  -2.9449],
        [  6.3550,   0.7129,   0.4917],
        [  3.4604,  -4.1545,   1.8091],
        [ -2.8162,   7.7059,  -9.6885],
        [ -3.1671,   7.6158,   6.0765],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -7.9298,   4.1367,  -3.9773],
        [ -0.5709,   7.9055,   7.6089],
        [  1.8719,   7.3713,  -3.7465],
        [  6.3550,   0.7129,   0.4917],
        [ -8.4297,  -5.3637,  -9.3962],
        [  2.8602,   6.8905,   5.8779],
        [  5.7963,  -1.6018,   1.8856],
        [  1.1006,   7.6318,  -2.4877],
        [ -9.1813,  -3.3550,   1.8827],
        [ -6.4844,  -8.0230,   8.5209],
        [  7.7750,  11.0665,   2.8364],
        [  6.3031,   1.8477,  -7.1706],
        [  6.3535,   0.6792,  -7.5101],
        [  3.9591, -10.6391,  -6.2557],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.9912,   3.1453,  -3.9989],
        [  5.4426, -10.1181,  -5.1268],
        [ -1.5875,   7.8944,  -9.4286]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.2585e-01, 1.7093e-01,  ..., 7.1197e-01, 1.0233e+00,
         1.0752e+00],
        [1.2585e-01, 0.0000e+00, 7.6902e-02,  ..., 8.2846e-01, 1.1216e+00,
         1.1796e+00],
        [1.7093e-01, 7.6902e-02, 0.0000e+00,  ..., 8.3489e-01, 1.1057e+00,
         1.1694e+00],
        ...,
        [7.1197e-01, 8.2846e-01, 8.3489e-01,  ..., 0.0000e+00, 3.9763e-01,
         4.0420e-01],
        [1.0233e+00, 1.1216e+00, 1.1057e+00,  ..., 3.9763e-01, 0.0000e+00,
         1.0561e-01],
        [1.0752e+00, 1.1796e+00, 1.1694e+00,  ..., 4.0420e-01, 1.0561e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.92390689, 14.02532758, 14.10045797, 14.02526505, 14.09357547,
       14.05633441, 11.73567759,  6.652742  , 10.06407839, 13.93790235,
        8.22728399, 12.4217523 ,  8.29640613, 13.04051744, 12.87752052,
       12.05395056, 12.25806559, 13.92390689, 11.48663767,  9.41244865,
       13.12218013,  6.61211224, 12.52875793,  7.59795445,  9.23178204,
       10.69963954, 10.23946247, 10.90638066,  5.16205087,  5.53255882,
       11.00694688, 11.22993218,  7.32754882, 13.14690137,  7.48538905,
       10.86693708,  9.23071698, 11.37939695,  6.74872981, 13.35013374,
        9.09119351,  6.74686071,  8.44940661,  6.36004393, 12.38881635,
        9.88571713,  6.29034958, 12.0495521 , 14.09357547,  9.69792513,
        8.25828409,  6.46560181,  5.68942572,  8.79284466,  6.60752373,
        9.99211663, 10.42667093,  8.43626114,  8.09704052,  9.60359158,
       11.80092578, 12.5085995 , 14.02526505,  8.04718433,  9.84314712,
        7.49076899,  6.76661595,  7.06372172,  9.78285424, 13.0493368 ,
       14.03505624,  7.20482224, 12.05395056, 11.28215429, 11.31276109,
       11.97957425,  6.29034958,  6.74872981,  9.50913849,  6.28835101,
       11.48663767,  6.2322769 ,  6.89245352,  6.3949043 ,  5.40689332,
        8.20436764,  8.24808832, 12.3828957 ,  8.94395628,  7.92607342,
        7.60529892,  6.3949043 ,  9.99145096,  7.46053281,  6.01356004,
        7.71075226,  9.7751189 , 10.31584102, 13.5247215 ,  6.56833864,
        6.38968357, 11.35182589, 10.31584102,  6.76661595, 11.4890861 ,
        8.05242591])}
{'P': tensor([[1.0753e-02, 4.6950e-01, 2.4478e-01,  ..., 4.6722e-08, 3.4747e-08,
         3.6788e-04],
        [1.0199e-03, 2.6051e-01, 3.5539e-03,  ..., 2.0898e-06, 8.9654e-07,
         1.8962e-02],
        [1.2889e-04, 9.2521e-01, 1.6943e-02,  ..., 1.3887e-07, 6.0360e-08,
         1.2501e-03],
        ...,
        [1.1167e-07, 1.2361e-07, 1.3460e-07,  ..., 7.3310e-02, 1.4217e-03,
         3.9514e-06],
        [2.3359e-05, 1.6614e-05, 1.8813e-05,  ..., 2.2583e-06, 5.1293e-02,
         5.9808e-06],
        [7.2749e-06, 1.1213e-05, 1.0891e-05,  ..., 4.3310e-04, 2.6471e-05,
         1.1227e-01]]), 'X': tensor([[  2.7590,  13.6621,   9.4496],
        [  1.1274,  14.0116,   8.1753],
        [  1.3705,  13.9723,   7.9266],
        [  1.9312,  13.8649,   8.3563],
        [  0.5175,  14.0910,   4.6540],
        [  2.2659,  13.7894,   6.5436],
        [  4.8432,   5.1968,   1.5006],
        [  1.5653,  13.9377,   4.1773],
        [  2.7590,  13.6621,   9.4496],
        [  1.8447,  -4.7372,   8.1235],
        [ 12.6038,   0.7665,  -3.4439],
        [ -3.4100,   7.5436,  -7.6980],
        [ -5.3382,   6.6533,   7.8054],
        [  7.7750,  11.0665,   2.8364],
        [ -8.9282,   2.1648,   8.4894],
        [ -5.1369,   6.7755,   8.3836],
        [  6.8704,  -9.3809,   7.5769],
        [  0.1206,  -4.7363,   8.7103],
        [  6.1755,   2.5233,  -2.0042],
        [ -3.3116, -10.1539,   7.5919],
        [  0.3038,   7.8092,   0.4164],
        [  9.0129,  -7.7072,   9.1400],
        [  2.6310, -10.9195,   3.8054],
        [  2.3889,  -4.6082,  -8.7291],
        [ -8.1204,  -5.9308,   4.0761],
        [  6.0140,  -1.0661,  -2.5040],
        [  3.7121,   6.3117,   0.6738],
        [  5.9843,  -9.8679,  -6.1449],
        [  5.9843,  -9.8679,  -6.1449],
        [ 12.2311,  -2.0445,   1.3545],
        [  3.0719, -10.8449,  -4.1673],
        [  3.1480,  -4.3165,  -7.8002],
        [  0.2571,  -4.7594,   1.5618],
        [  9.3836,   9.5075,  -5.9278],
        [  5.6258,  -1.9404,  -6.0715],
        [ -4.0409,   7.3169,  -3.5779],
        [  9.0129,  -7.7072,   9.1400],
        [ -9.3444,   0.5821,   6.9101],
        [ -8.4297,  -5.3637,  -9.3962],
        [  5.3924, -10.1396,  -4.0863],
        [ -1.5408,   7.8978,   5.3470],
        [  2.9260,  -4.4162,   8.3808],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -5.2752,  -9.0409,  -8.2568],
        [  1.0093,   7.6567,   2.3526],
        [  0.9809,  -4.8145,   6.8706],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.5224,  -0.4497,   4.8448],
        [ -8.0766,  -6.0046,  -3.1873],
        [ -9.4157,   0.0821,  -5.0025],
        [  2.7295,  13.6703,  -8.9328],
        [  6.2022,   2.4092,   3.0349],
        [ 10.5057,  -5.9099,  -5.1599],
        [  1.7232,  -4.7574,   3.1270],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -9.4731,  -0.7120,  -6.0271],
        [  2.0729,   7.2878,   7.0880],
        [ 12.5770,   1.8512,  -0.6578],
        [ -2.1533, -10.5778,   1.0112],
        [ -9.1813,  -3.3550,   1.8827],
        [  5.9655,   3.2172,  -7.5953],
        [ -2.9497, -10.3028,  -7.3479],
        [  6.2038,   2.4020,  -3.1692],
        [  2.0547,   7.2957,  -1.9343],
        [ -3.0644,   7.6439,  -1.4322],
        [ -6.4002,  -8.1048,  -0.9108],
        [  5.9843,  -9.8679,  -6.1449],
        [  2.0937,  13.8293,  -2.3533],
        [  2.5128, -10.9365,   9.9645],
        [ -1.6086,   7.8928,   4.3385],
        [ -3.3999, -10.1152,  -1.2088],
        [  6.3122,  -9.6994,  -6.3282],
        [ -9.4624,  -0.4633,  -7.3918],
        [  2.1120,  -4.6818,   8.6144],
        [ -6.6998,   5.5954,   0.0823],
        [  4.3459,  -3.5327,   0.3667],
        [  5.8201,   3.5841,  -8.0609],
        [ -8.1666,   3.7653,   3.6075],
        [ 10.0400,   8.6922,  -6.2847],
        [  3.6817,   6.3354,   6.1233],
        [ -3.9858,  -9.8334,  -3.7323],
        [ -9.1623,  -3.4288,  -4.9961],
        [ -9.3810,  -2.3455,  -9.5194],
        [  2.3490,  -4.6199,  -4.2489],
        [ 12.5123,  -0.5339,  -1.0728],
        [ 12.6059,   0.9597,  -6.3264],
        [  9.0004,  -7.7195,   4.6945],
        [ -6.4844,  -8.0230,   8.5209],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.0788,  -0.8710,  -6.8719],
        [ -5.5525,   6.5146,  -4.4984],
        [  3.1480,  -4.3165,  -7.8002],
        [  3.3094,  13.4969,  -3.6507],
        [ -9.1907,  -3.3180,  -5.5943],
        [  2.9470,  13.6085,   2.9809],
        [ 12.5770,   1.8512,  -0.6578],
        [  1.7000,  -4.7609,   2.7935],
        [  4.4633,   5.6287,  -0.7754],
        [  2.3889,  -4.6082,  -8.7291],
        [  3.4009,  -4.1874,  -5.9751],
        [ 11.8951,  -3.1570,   7.0742],
        [ -4.6942,  -9.4299,  -9.1846],
        [  8.8091,  -7.9037,   0.3631],
        [ -7.4808,   4.7462,   2.0266]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1357, 0.0518,  ..., 1.7981, 0.9110, 1.0976],
        [0.1357, 0.0000, 0.0876,  ..., 1.6739, 0.7978, 0.9633],
        [0.0518, 0.0876, 0.0000,  ..., 1.7465, 0.8607, 1.0464],
        ...,
        [1.7981, 1.6739, 1.7465,  ..., 0.0000, 0.9051, 0.7908],
        [0.9110, 0.7978, 0.8607,  ..., 0.9051, 0.0000, 0.4413],
        [1.0976, 0.9633, 1.0464,  ..., 0.7908, 0.4413, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.93790235, 14.05687815, 14.0393949 , 13.99877413, 14.10045797,
       13.97430191,  7.10378919, 14.02532758, 13.93790235,  5.08372149,
       12.62711252,  8.27853076,  8.53014915, 13.5247215 ,  9.18690018,
        8.50268322, 11.62770332,  4.73784222,  6.67108662, 10.68031351,
        7.81509631, 11.85890664, 11.23201619,  5.19065439, 10.05558893,
        6.10774544,  7.32233733, 11.54071859, 11.54071859, 12.40074901,
       11.27160035,  5.34250225,  4.7663526 , 13.35832725,  5.9510489 ,
        8.35856487, 11.85890664,  9.36256096,  9.99145096, 11.48434767,
        8.04664886,  5.29755487, 12.05300943, 10.46739073,  7.72291946,
        4.91336994,  8.53501777, 12.53047489, 10.06407839,  9.41605684,
       13.94008948,  6.65368551, 12.05395056,  5.05989547, 10.19036081,
        9.7712321 ,  9.60080639,  9.4997987 ,  7.57685908, 12.7125079 ,
       10.794755  ,  9.7751189 ,  6.77777606, 10.7167285 ,  6.65259775,
        7.57946459,  8.23525433, 10.32715715, 11.54071859, 13.98691426,
       11.22142195,  8.0550286 , 10.67132524, 11.57249984,  9.47369628,
        5.13616224,  8.72896381,  5.60064319,  6.83515201,  8.99276564,
       13.27994304,  7.32754882, 10.61047596,  9.78286733,  9.66977802,
        5.18277858, 12.52372629, 12.64235359, 11.85743425, 10.31584102,
        5.9510489 ,  6.14086582,  8.55982105,  5.34250225, 13.89671314,
        9.7712321 , 13.92390689, 12.7125079 ,  5.05534609,  7.18354885,
        5.19065439,  5.39450305, 12.30695098, 10.53368644, 11.83509372,
        8.859416  ])}
{'P': tensor([[7.2830e-03, 8.4688e-01, 7.6828e-02,  ..., 1.3774e-07, 1.0618e-06,
         8.2205e-07],
        [4.1844e-04, 2.6209e-01, 2.2089e-01,  ..., 4.7814e-06, 3.0831e-05,
         3.0235e-05],
        [1.4210e-03, 5.5624e-02, 2.4596e-01,  ..., 6.8047e-06, 3.0517e-05,
         3.2121e-05],
        ...,
        [3.4258e-05, 3.0361e-05, 5.3911e-05,  ..., 8.0196e-02, 7.8629e-04,
         3.9414e-04],
        [1.3586e-03, 9.1650e-04, 1.0157e-03,  ..., 5.4520e-04, 9.0186e-02,
         1.5980e-03],
        [1.9342e-04, 1.3940e-04, 1.5712e-04,  ..., 2.6051e-03, 1.7817e-01,
         1.3792e-01]]), 'X': tensor([[  2.7295,  13.6703,  -8.9328],
        [  0.8572,  14.0501,  -8.3364],
        [  0.6396,  14.0772,  -6.3786],
        [  2.0449,  13.8402,  -7.1642],
        [  2.5469,  13.7192,  -9.6935],
        [  0.4119,  14.1019,  -4.1870],
        [ 12.5154,   2.5072,  -9.3274],
        [  3.8182,  -3.9347,  -1.7961],
        [ -2.2508, -10.5479,  -2.7235],
        [  0.8700,  -4.8132,  -7.8042],
        [  2.6451,  -4.5250,  -0.5565],
        [  5.8201,   3.5841,  -8.0609],
        [ -0.5709,   7.9055,   7.6089],
        [ -5.0550,   6.8231,  -9.2502],
        [ -4.3359,  -9.6431,  -9.0616],
        [ 12.4592,  -0.9168,  -6.7571],
        [ -6.0175,  -8.4541,  -9.8408],
        [ 11.6801,  -3.7196,  -0.9181],
        [ -8.0829,  -5.9940,   5.0564],
        [  5.7845,   3.6652,   0.9175],
        [ 11.1469,   6.9237,   1.6160],
        [ 10.4994,  -5.9193,  -7.5153],
        [ 12.5789,   0.1722,   1.6470],
        [  6.2630,   2.1057,  -5.2136],
        [ 10.9875,   7.2207,  -8.8549],
        [  1.6954,  -4.7616,   9.8198],
        [  4.5340,  -3.3629,   8.9652],
        [  1.7500,   7.4187,   4.4237],
        [ -8.5607,  -5.0938,  -5.4531],
        [  2.6321,  -4.5296,  -2.5159],
        [  1.7659,   7.4127,   4.7871],
        [  6.9653,  -9.3225,  -3.4727],
        [  1.3767, -11.0341,   9.9415],
        [  1.7903,  -4.7467,   0.1168],
        [  8.5716,  10.3596,   6.8818],
        [  5.6260,  -1.9401,   3.0259],
        [ -6.3066,  -8.1936,   9.5822],
        [ 11.5935,   5.9740,  -4.3952],
        [  0.3715,  -4.7756,   0.2312],
        [ -6.8618,  -7.6322,   3.7027],
        [  9.4554,   9.4245,   9.8800],
        [  9.5261,  -7.1675,   1.4206],
        [ 12.1940,  -2.1881,   7.1499],
        [ -9.4309,  -0.0599,  -4.0830],
        [ -4.0409,   7.3169,  -3.5779],
        [  6.2419,  -0.2313,   4.2440],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.7976,  -4.4682,  -4.2020],
        [  5.1861,   4.7361,  -3.7312],
        [ -7.9925,  -6.1423,   2.8990],
        [ -3.4986,   7.5152,   9.0712],
        [ -9.4748,  -0.7765,  -5.6695],
        [ -5.6423,   6.4537,  -1.2091],
        [ -9.3107,  -2.7697,  -3.2919],
        [  1.5662,  13.9375,   4.5142],
        [  0.6363, -11.0348,   6.1414],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.4378,  -1.8606,  -2.5668],
        [  0.6470, -11.0352,  -1.1931],
        [ -2.6069, -10.4298,  -6.1995],
        [  2.7976,  -4.4682,  -4.2020],
        [  4.5491,  -3.3485,   6.8412],
        [ 12.5919,   0.4035,  -5.8460],
        [ 12.2589,  -1.9312,   9.8657],
        [  0.2732,  -4.7619,  -1.1577],
        [ 10.4994,  -5.9193,  -7.5153],
        [  0.1382, -11.0073,   1.0026],
        [  9.5203,   9.3483,   5.8781],
        [  0.9036,   7.6840,   9.9237],
        [  0.9478,  -4.8144,   0.6371],
        [  6.2905,   1.9366,   2.1169],
        [ -8.7544,   2.6096,   3.3146],
        [ -5.2589,   6.7024,  -5.5809],
        [ -8.9033,   2.2329,  -8.8912],
        [ -9.0604,   1.7707,   1.1066],
        [  9.6587,   9.1817,  -0.8212],
        [ -5.9225,  -8.5355,   2.4778],
        [ 10.2470,   8.4054,  -0.1177],
        [ -6.6998,   5.5954,   0.0823],
        [  6.9284,  11.7061,  -0.3732],
        [  2.1379,  -4.6757,   6.4981],
        [  6.3124,   0.2057,  -0.4593],
        [ -8.0829,  -5.9940,   5.0564],
        [ -5.0943,  -9.1682,   0.5073],
        [ -9.4283,  -1.9585,  -0.9129],
        [  5.4336,  -2.2676,   5.6933],
        [ 10.0619,   8.6627,  -9.2884],
        [  6.2883,   0.0325,   5.2692],
        [ -9.4735,  -0.7274,  -2.6749],
        [  6.2895,   0.0405,   5.7652],
        [  2.9461,  -4.4077,  -0.4831],
        [ -6.4002,  -8.1048,  -0.9108],
        [ -8.9788,   2.0208,  -3.2867],
        [  8.0877,  -8.5297,  -1.1050],
        [ 10.5057,  -5.9099,  -5.1599],
        [  8.9284,  10.0036,  -6.9210],
        [ -3.5029,   7.5138,   8.1853],
        [  6.9314,  -9.3435,  -1.2729],
        [ 12.1940,  -2.1881,   7.1499],
        [  2.2182,  13.8007,   5.1889],
        [  4.5800,   5.5034,  -3.9432],
        [ 12.1751,  -2.2592,  -5.7751],
        [  6.1929,   2.4501,  -6.9352],
        [  5.6093,  -1.9705,   6.4807],
        [  6.0560,  -0.9421,  -5.0334],
        [  3.2226,  -4.2802,  -4.0428]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1291, 0.2127,  ..., 1.0546, 0.3408, 0.4088],
        [0.1291, 0.0000, 0.1077,  ..., 1.0034, 0.2613, 0.3179],
        [0.2127, 0.1077, 0.0000,  ..., 0.9013, 0.1563, 0.2102],
        ...,
        [1.0546, 1.0034, 0.9013,  ..., 0.0000, 0.7452, 0.7055],
        [0.3408, 0.2613, 0.1563,  ..., 0.7452, 0.0000, 0.0739],
        [0.4088, 0.3179, 0.2102,  ..., 0.7055, 0.0739, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.94008948, 14.07623266, 14.09176259, 13.99048094, 13.95361098,
       14.10796466, 12.76408562,  5.48276038, 10.78533838,  4.89121805,
        5.24137909,  6.83515201,  7.92607342,  8.49161154, 10.57302546,
       12.49291768, 10.37698435, 12.25806559, 10.06286074,  6.84796202,
       13.12218013, 12.05300943, 12.58006038,  6.60752373, 13.14777433,
        5.05444288,  5.64500512,  7.62232307,  9.96152788,  5.23878371,
        7.62011734, 11.63724792, 11.11969815,  5.07306274, 13.44593519,
        5.95111804, 10.33958639, 13.04217487,  4.79003361, 10.26327895,
       13.35013374, 11.92133129, 12.38881635,  9.43112957,  8.35856487,
        6.24614905, 10.41909211,  5.27178363,  7.02326515, 10.08001894,
        8.28968067,  9.50654967,  8.5724042 ,  9.71392179, 14.02526505,
       11.05317597, 11.83509372,  9.61942445, 11.05413545, 10.7506423 ,
        5.27178363,  5.64865173, 12.59840193, 12.41012046,  4.76969364,
       12.05300943, 11.00812996, 13.3426541 ,  7.73692436,  4.90676242,
        6.58184474,  9.13507321,  8.51928531,  9.17905491,  9.23178204,
       13.32645552, 10.38897563, 13.25336001,  8.72896381, 13.60275211,
        5.14124612,  6.31576264, 10.06286074, 10.48840476,  9.62958973,
        5.88782913, 13.27717982,  6.28835101,  9.5014146 ,  6.28961988,
        5.30158497, 10.32715715,  9.20340031, 11.75438006, 12.05395056,
       13.40850328,  8.29022197, 11.63383221, 12.38881635, 13.97779839,
        7.15990676, 12.3828957 ,  6.65991877,  5.94535529,  6.12886024,
        5.35774573])}
{'P': tensor([[2.0237e-02, 8.0864e-01, 7.8042e-02,  ..., 5.7129e-08, 4.7179e-06,
         3.2724e-07],
        [7.5760e-04, 1.9147e-01, 1.6137e-01,  ..., 5.9221e-07, 9.8575e-05,
         3.6814e-06],
        [1.9649e-03, 4.3400e-02, 1.9191e-01,  ..., 7.2248e-07, 1.7342e-04,
         5.5636e-06],
        ...,
        [8.7238e-06, 6.3873e-06, 8.1120e-06,  ..., 2.5153e-02, 9.1363e-07,
         1.9896e-03],
        [2.6108e-07, 3.4319e-07, 4.7021e-07,  ..., 1.9951e-04, 6.3974e-02,
         3.3350e-04],
        [5.8368e-05, 6.8536e-05, 1.2037e-04,  ..., 1.3001e-03, 1.1567e-04,
         1.7665e-01]]), 'X': tensor([[  2.5469,  13.7192,  -9.6935],
        [  0.8572,  14.0501,  -8.3364],
        [  0.6396,  14.0772,  -6.3786],
        [  2.0449,  13.8402,  -7.1642],
        [  0.4119,  14.1019,  -4.1870],
        [ -2.8162,   7.7059,  -9.6885],
        [  2.3200,   7.1756,   1.1938],
        [ -9.3419,   0.5972,  -1.2841],
        [  0.7685,  -4.8098,  -2.9310],
        [  3.5284, -10.7484,   5.3347],
        [  0.7685,  -4.8098,  -2.9310],
        [ 11.3800,  -4.3919,   1.4066],
        [  6.5428,  -9.5728,  -2.9897],
        [  0.5850,  -4.7982,   9.6639],
        [ -7.3470,  -7.0615,   5.0066],
        [ -1.7009,   7.8851,  -8.2437],
        [ -7.9298,   4.1367,  -3.9773],
        [ -8.0588,  -6.0341,  -2.3006],
        [  5.3602,  -2.3808,   7.2098],
        [  8.5716,  10.3596,   6.8818],
        [  9.6662,   9.1725,   8.1056],
        [  3.4604,  -4.1545,   1.8091],
        [  0.1532,  -4.7422,  -7.0145],
        [  9.8082,  -6.8396,   4.9615],
        [  6.3031,   1.8477,  -7.1706],
        [  5.9883,   3.1536,   6.2320],
        [ -8.8363,   2.4086,  -5.6296],
        [  5.3858,   4.4260,  -9.6769],
        [ 11.1420,   6.9331,  -2.4779],
        [ 12.3759,  -1.3943,   6.4659],
        [  7.0171,  11.6439,  -5.0178],
        [  6.3516,   0.6425,   0.3425],
        [ -8.6112,   2.9319,  -7.9943],
        [  2.9260,  -4.4162,   8.3808],
        [  2.8602,   6.8905,   5.8779],
        [ -8.9606,  -4.0994,   1.7886],
        [ -2.1533, -10.5778,   1.0112],
        [ 11.5935,   5.9740,  -4.3952],
        [  7.7794,  11.0629,   8.9069],
        [  5.8902,  -1.3887,   4.0388],
        [  5.1196,  -2.7160,   4.2705],
        [ -7.7239,   4.4298,   3.1216],
        [ 12.4691,   2.8525,  -5.5110],
        [  5.2608,  -2.5256,  -9.1446],
        [ -4.7642,  -9.3860,  -5.1133],
        [  1.3339,   7.5627,  -0.9034],
        [  6.3550,   0.7129,   0.4917],
        [ -9.1459,   1.4780,  -4.2403],
        [  2.1771, -10.9776,  -3.2007],
        [  2.7590,  13.6621,   9.4496],
        [  2.5620,   7.0549,   8.4825],
        [ -7.3333,   4.9246,  -7.9212],
        [ -5.0943,  -9.1682,   0.5073],
        [  4.4049,   5.6892,  -4.4394],
        [ -2.5056,   7.7720,   5.5878],
        [  7.9816,  10.8938,  -0.5378],
        [ -8.9606,  -4.0994,   1.7886],
        [  4.7544,  -3.1425,  -1.5315],
        [ 10.9916,   7.2133,  -0.9179],
        [ -2.7572,   7.7194,   5.9580],
        [ 12.2692,  -1.8882,   3.9253],
        [  0.3231,  -4.7691,  -1.6120],
        [ -9.3604,   0.4855,  -3.0158],
        [  6.7565,  11.8237,  -4.9882],
        [  0.1252, -11.0062,   7.2564],
        [  6.8704,  -9.3809,   7.5769],
        [ -9.3444,   0.5821,   6.9101],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -5.1377,   6.7751,  -0.4438],
        [ 11.7418,   5.6070,  -1.4192],
        [ -4.6942,  -9.4299,  -9.1846],
        [  5.2608,  -2.5256,  -9.1446],
        [ -8.6801,   2.7809,  -0.8443],
        [  6.1929,   2.4501,  -6.9352],
        [ -6.0195,   6.1789,  -2.9496],
        [  4.1297,  -3.7098,  -3.9838],
        [  6.3124,   0.2057,  -0.4593],
        [  1.8766,   7.3694,   1.1887],
        [  4.0362,  13.2397,  -7.0030],
        [  1.7218,  -4.7576,  -3.3617],
        [ -6.6895,   5.6051,   2.7899],
        [  5.8201,   3.5841,  -8.0609],
        [ -9.4351,  -1.8891,   3.5642],
        [ -9.4748,  -0.7765,  -5.6695],
        [ 11.2209,  -4.7092,   2.6117],
        [  5.4242, -10.1261,   3.9595],
        [  4.2275,  -3.6320,  -0.2714],
        [ 11.0373,   7.1299,  -8.5895],
        [ -3.5903,   7.4847,  -8.9112],
        [  3.7237,  -3.9966,  -5.0376],
        [ 12.4013,  -1.2597,  -7.6474],
        [  0.1677,  -4.7447,  -6.1529],
        [ 11.1469,   6.9237,   1.6160],
        [ 12.4597,  -0.9136,  -4.8763],
        [  6.3124,   0.2057,  -0.4593],
        [  0.3724,   7.7974,  -6.8722],
        [ -4.7334,   6.9982,  -0.4392],
        [  4.0421,  -3.7765,  -8.3631],
        [  7.9403,  -8.6455,  -8.7433],
        [ -2.1817,   7.8274,  -4.6298],
        [ -4.6393,   7.0461,   1.1624],
        [ 12.1588,   4.3364,  -8.0867],
        [  2.0039,   7.3173,  -1.5885],
        [  8.8091,  -7.9037,   0.3631],
        [ -9.2345,  -3.1345,   3.9715],
        [  5.5240,  -2.1198,   6.2055]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0999, 0.1910,  ..., 0.7151, 1.3934, 1.0388],
        [0.0999, 0.0000, 0.1077,  ..., 0.6356, 1.2977, 0.9904],
        [0.1910, 0.1077, 0.0000,  ..., 0.5286, 1.2046, 0.8881],
        ...,
        [0.7151, 0.6356, 0.5286,  ..., 0.0000, 0.7528, 0.4807],
        [1.3934, 1.2977, 1.2046,  ..., 0.7528, 0.0000, 0.9434],
        [1.0388, 0.9904, 0.8881,  ..., 0.4807, 0.9434, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.95361098, 14.07623266, 14.09176259, 13.99048094, 14.10796466,
        8.20436764,  7.54127425,  9.36093562,  4.87083348, 11.31276109,
        4.87083348, 12.19805394, 11.59512194,  4.83371341, 10.19036081,
        8.06643487,  8.94395628, 10.0674887 ,  5.86519525, 13.44593519,
       13.32557133,  5.40689332,  4.74468406, 11.95743558,  6.56833864,
        6.76791483,  9.15866607,  6.97106912, 13.12299145, 12.45418236,
       13.59480906,  6.38399789,  9.09661212,  5.29755487,  7.46053281,
        9.85384623, 10.794755  , 13.04217487, 13.52430176,  6.05165248,
        5.79543038,  8.90405823, 12.79126785,  5.83560009, 10.52587243,
        7.67939546,  6.3949043 ,  9.26455795, 11.1913598 , 13.93790235,
        7.50564223,  8.83341608, 10.48840476,  7.19513806,  8.16584537,
       13.50484539,  9.85384623,  5.69913817, 13.1471313 ,  8.19703512,
       12.41366754,  4.78002665,  9.37296102, 13.61801526, 11.00694688,
       11.62770332,  9.36256096,  9.7712321 ,  8.50278896, 13.01187484,
       10.53368644,  5.83560009,  9.11473391,  6.65991877,  8.62631801,
        5.5513072 ,  6.31576264,  7.60463349, 13.84126197,  5.05963217,
        8.7273569 ,  6.83515201,  9.62238828,  9.50654967, 12.16901615,
       11.48734978,  5.57341623, 13.1399181 ,  8.30125201,  5.46244988,
       12.46513646,  4.74770899, 13.12218013, 12.49317563,  6.31576264,
        7.80625981,  8.4486771 ,  5.53175275, 11.73848065,  8.12580976,
        8.43626114, 12.9089559 ,  7.58668144, 11.83509372,  9.75200848,
        5.91677188])}
{'P': tensor([[5.0807e-03, 6.2194e-01, 1.4872e-01,  ..., 2.2145e-08, 4.1214e-04,
         2.0231e-06],
        [1.2846e-04, 2.0667e-02, 3.8842e-01,  ..., 8.9191e-08, 1.7631e-03,
         8.7221e-06],
        [1.0175e-03, 1.6940e-03, 2.8185e-01,  ..., 1.8150e-06, 5.0171e-02,
         2.1627e-04],
        ...,
        [8.2783e-07, 7.4532e-07, 6.4339e-07,  ..., 2.1244e-02, 1.0497e-06,
         2.4926e-06],
        [3.9039e-06, 4.0563e-06, 5.9007e-06,  ..., 4.6342e-05, 4.9561e-02,
         1.5614e-02],
        [7.8248e-07, 7.6119e-07, 7.4871e-07,  ..., 4.3696e-03, 2.7664e-05,
         6.3890e-02]]), 'X': tensor([[  2.5318,  13.7231,   6.3692],
        [  2.2659,  13.7894,   6.5436],
        [  0.5175,  14.0910,   4.6540],
        [  1.3705,  13.9723,   7.9266],
        [  1.1274,  14.0116,   8.1753],
        [  2.3261,  13.7749,   5.9121],
        [  2.0547,   7.2957,  -1.9343],
        [ 12.4597,  -0.9136,  -4.8763],
        [  5.4336,  -2.2676,   5.6933],
        [  6.3516,   0.6425,   0.3425],
        [  6.3569,   0.7591,   6.5011],
        [ 11.8951,  -3.1570,   7.0742],
        [  2.4639, -10.9431,  -7.3718],
        [  8.6271,  -8.0715,  -7.4813],
        [  4.8432,   5.1968,   1.5006],
        [  6.3031,   1.8477,  -7.1706],
        [ -7.2515,   5.0195,   4.6434],
        [  2.3928,   7.1404,  -1.7035],
        [  6.2975,   0.0947,   5.1791],
        [  0.1047, -11.0046,   0.5188],
        [  9.0129,  -7.7072,   9.1400],
        [ 12.1751,  -2.2592,  -5.7751],
        [  4.0591,   6.0202,  -1.6339],
        [ -4.8751,  -9.3148,   4.0406],
        [ 10.9916,   7.2133,  -0.9179],
        [ -4.2271,   7.2387,  -1.8316],
        [  6.3542,   0.6934,  -3.3840],
        [  7.9403,  -8.6455,  -8.7433],
        [  6.3589,   0.8239,  -8.4230],
        [ 12.2921,  -1.7902,  -4.0870],
        [ -9.4716,  -0.6673,   7.6068],
        [  6.2905,   1.9366,   2.1169],
        [ -2.5056,   7.7720,   5.5878],
        [  5.2188,  -2.5839,   2.1833],
        [ -6.3066,  -8.1936,   9.5822],
        [  9.0004,  -7.7195,   4.6945],
        [  2.0729,   7.2878,   7.0880],
        [  1.6232,  -4.7717,  -2.5655],
        [  9.0004,  -7.7195,   4.6945],
        [ -8.9282,   2.1648,   8.4894],
        [ -9.1749,  -3.3803,   6.4629],
        [ -0.4341, -10.9473,   7.4509],
        [ -9.4731,  -0.7120,  -6.0271],
        [  5.4167, -10.1293,   6.2156],
        [  6.0047,   3.1062,  -6.7914],
        [  9.3836,   9.5075,  -5.9278],
        [  5.9655,   3.2172,  -7.5953],
        [  6.9653,  -9.3225,  -3.4727],
        [  2.7420,  -4.4896,  -3.7139],
        [  9.3624,   9.5318,   0.2779],
        [  6.2022,   2.4092,   3.0349],
        [ -2.0445,   7.8469,  -0.5857],
        [  1.1006,   7.6318,  -2.4877],
        [ -3.3116, -10.1539,   7.5919],
        [  6.2026,  12.1774,  -1.5887],
        [  4.0642,  -3.7599,  -3.4565],
        [ 11.7795,  -3.4697,   3.5624],
        [  4.0587, -10.6112,   6.2273],
        [  3.3167,   6.6022,   6.3554],
        [ 10.6673,  -5.6649,  -4.1646],
        [ 11.5568,   6.0602,   9.3244],
        [  3.3943,   6.5483,   1.0790],
        [ -4.7389,   6.9953,   3.7638],
        [ -7.4808,   4.7462,   2.0266],
        [  3.2701,   6.6338,   9.9209],
        [ -9.2301,  -3.1537,  -2.7477],
        [ -9.3616,  -2.4750,   3.7765],
        [  2.9461,  -4.4077,  -0.4831],
        [ -9.3087,   0.7792,  -7.5587],
        [  2.5813,  -4.5471,   5.3597],
        [  5.4179,  -2.2925,  -3.4641],
        [ 10.4758,  -5.9539,   1.3667],
        [ -9.3444,   0.5821,   6.9101],
        [  2.6078, -10.9229,   6.2869],
        [ -9.0961,   1.6529,   3.1553],
        [ 11.5456,   6.0861,   8.1101],
        [  1.1006,   7.6318,  -2.4877],
        [  5.1793,   4.7461,   0.3899],
        [  8.2154,  -8.4262,   8.0171],
        [  2.8890,  -4.4316,   1.1600],
        [  2.4149,  -4.6005,  -4.3339],
        [  0.5600, -11.0321,  -0.8660],
        [  1.4749,  -4.7892,  -0.8089],
        [  5.6022,  -1.9833,   8.3081],
        [  3.9770, -10.6341,   8.0783],
        [ 10.9394,   7.3069,  -8.9365],
        [ -5.3328,  -8.9992,   7.3440],
        [  0.9445,   7.6736,   0.5095],
        [  5.7151,  -1.7700,   3.1542],
        [  9.3624,   9.5318,   0.2779],
        [  6.0031,  12.2958,  -6.5992],
        [ -8.5859,  -5.0393,   0.5476],
        [  6.9653,  -9.3225,  -3.4727],
        [ -9.4716,  -0.6673,   7.6068],
        [ -1.8367,   7.8718,  -8.6523],
        [ -1.7290,   7.8825,  -3.3831],
        [ 12.2435,   4.0076,   8.1257],
        [  2.2182,  13.8007,   5.1889],
        [  9.6662,   9.1725,   8.1056],
        [  2.0039,   7.3173,  -1.5885],
        [  0.6757, -11.0361,  -1.3047],
        [  1.3767, -11.0341,   9.9415],
        [  2.5382,   7.0672,   4.5097],
        [  3.0719, -10.8449,  -4.1673],
        [ -7.7239,   4.4298,   3.1216],
        [ -7.9925,  -6.1423,   2.8990]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0437, 0.1884,  ..., 1.3304, 1.0638, 1.3367],
        [0.0437, 0.0000, 0.1588,  ..., 1.2996, 1.0226, 1.2944],
        [0.1884, 0.1588, 0.0000,  ..., 1.1422, 0.8941, 1.1738],
        ...,
        [1.3304, 1.2996, 1.1422,  ..., 0.0000, 0.5940, 0.6995],
        [1.0638, 1.0226, 0.8941,  ..., 0.5940, 0.0000, 0.2894],
        [1.3367, 1.2944, 1.1738,  ..., 0.6995, 0.2894, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.95472331, 13.97430191, 14.10045797, 14.0393949 , 14.05687815,
       13.96987806,  7.57946459, 12.49317563,  5.88782913,  6.38399789,
        6.4020404 , 12.30695098, 11.21703654, 11.81423161,  7.10378919,
        6.56833864,  8.81930525,  7.53063885,  6.29821673, 11.00508681,
       11.85890664, 12.3828957 ,  7.26074925, 10.51340829, 13.1471313 ,
        8.38252937,  6.39187285, 11.73848065,  6.41202986, 12.4217523 ,
        9.49511524,  6.58184474,  8.16584537,  5.82345867, 10.33958639,
       11.85743425,  7.57685908,  5.04028538, 11.85743425,  9.18690018,
        9.77777346, 10.95593729,  9.4997987 , 11.48663767,  6.76056625,
       13.35832725,  6.77777606, 11.63724792,  5.26067019, 13.3607379 ,
        6.65368551,  8.10886021,  7.71075226, 10.68031351, 13.6660671 ,
        5.5366698 , 12.27991991, 11.36089355,  7.38844703, 12.07818871,
       13.0493368 ,  7.37576154,  8.44940661,  8.859416  ,  7.39599986,
        9.75401449,  9.68324553,  5.30158497,  9.34127013,  5.2287058 ,
        5.8828986 , 12.0495521 ,  9.36256096, 11.22993218,  9.24501829,
       13.05148826,  7.71075226,  7.02496223, 11.76830981,  5.29009836,
        5.19577146, 11.04629471,  5.01114719,  5.94293185, 11.35345345,
       13.15525803, 10.4606265 ,  7.73151309,  5.98285319, 13.3607379 ,
       13.6829843 ,  9.95552626, 11.63724792,  9.49511524,  8.08320235,
        8.06990483, 12.88270018, 13.97779839, 13.32557133,  7.58668144,
       11.05672136, 11.11969815,  7.50917551, 11.27160035,  8.90405823,
       10.08001894])}
{'P': tensor([[3.6096e-03, 4.8143e-01, 4.4258e-01,  ..., 1.0802e-06, 1.6818e-08,
         2.1044e-08],
        [1.2869e-05, 4.1897e-03, 9.7256e-01,  ..., 1.5068e-06, 1.9623e-08,
         2.2014e-08],
        [4.1865e-04, 5.2619e-04, 2.8737e-01,  ..., 1.3290e-04, 1.3395e-06,
         1.3765e-06],
        ...,
        [6.1815e-07, 5.6044e-07, 5.3208e-07,  ..., 6.4350e-02, 1.6278e-04,
         1.3741e-04],
        [4.5914e-05, 3.4427e-05, 2.4610e-05,  ..., 1.6987e-06, 9.3643e-02,
         1.1275e-01],
        [3.3606e-04, 2.2680e-04, 1.5162e-04,  ..., 1.7526e-06, 1.7979e-04,
         7.8879e-02]]), 'X': tensor([[ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [-7.6650e+00,  4.5092e+00,  2.5203e-01],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [ 6.1222e+00, -7.2615e-01,  8.8204e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 1.1147e+01,  6.9237e+00,  1.6160e+00],
        [ 1.2583e+01,  1.7543e+00, -7.6134e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 2.6614e+00, -4.5192e+00, -9.3641e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-9.4748e+00, -7.7650e-01, -5.6695e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-8.0750e+00, -6.0072e+00, -5.8267e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [-8.1666e+00,  3.7653e+00,  3.6075e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 2.5469e+00,  1.3719e+01, -9.6935e+00],
        [ 9.8082e+00, -6.8396e+00,  4.9615e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 1.1779e+01,  5.5086e+00, -9.6929e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 6.1741e+00, -5.3295e-01,  3.1913e+00],
        [-5.0943e+00, -9.1682e+00,  5.0726e-01],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 4.6357e+00,  5.4415e+00,  7.9067e+00],
        [-9.4756e+00, -8.1663e-01, -9.6021e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 5.7151e+00, -1.7700e+00,  3.1542e+00],
        [ 7.9141e+00, -8.6656e+00,  9.6948e+00],
        [ 5.5240e+00, -2.1198e+00,  6.2055e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [ 6.3535e+00,  6.7920e-01, -7.5101e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-8.9379e+00, -4.1655e+00, -2.8114e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [-9.3616e+00, -2.4749e+00,  1.7935e-01],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [-8.5149e+00, -5.1905e+00,  1.5239e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 5.7485e+00, -9.9811e+00, -3.2355e+00],
        [ 1.1546e+01,  6.0861e+00,  8.1101e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-6.9838e+00, -7.4965e+00,  9.7957e-01],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 1.0960e+01, -5.1849e+00, -3.5526e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0280, 0.1511,  ..., 1.3057, 0.4271, 0.2927],
        [0.0280, 0.0000, 0.1325,  ..., 1.2936, 0.3996, 0.2730],
        [0.1511, 0.1325, 0.0000,  ..., 1.1645, 0.3138, 0.1416],
        ...,
        [1.3057, 1.2936, 1.1645,  ..., 0.0000, 1.1089, 1.0368],
        [0.4271, 0.3996, 0.3138,  ..., 1.1089, 0.0000, 0.2431],
        [0.2927, 0.2730, 0.1416,  ..., 1.0368, 0.2431, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.96746816, 14.03618649, 14.10796466, 14.02058621, 14.09176259,
       14.08549802, 11.35345345,  6.76661595,  8.89301516, 11.69264243,
        6.16512856,  8.28968067, 13.12218013, 12.70489563,  5.36061974,
        5.96556248,  5.24461068,  9.24501829,  6.28961988,  7.14837514,
        9.50654967,  7.88677538,  7.33350642, 13.99877413, 10.1371552 ,
       10.54658597,  9.24501829,  5.1805345 , 10.06437598,  8.77913428,
        8.99276564,  4.96231532,  8.62162372, 13.37121923, 11.83075005,
        8.91551231, 13.01728313, 13.95361098, 11.95743558, 13.70893492,
       13.00380624,  7.18354885,  6.19707944, 10.48840476,  7.16747878,
        4.91038884,  6.90115901,  7.14837514,  9.51074763, 12.79126785,
        6.41202986,  7.17789666,  6.21275016,  5.98285319, 11.73567759,
        5.91677188, 12.10942412,  9.27474386,  9.36256096,  9.19837792,
        9.19837792,  8.57560275,  5.40689332,  7.54620849,  6.38968357,
        6.28835101, 11.05413545,  6.02779201, 10.00794468, 10.98076873,
        5.13182506,  6.5618968 , 11.21703654,  9.86089579,  8.57560275,
        8.55982105,  9.68323686, 12.4980845 ,  7.24677664,  9.97220799,
       11.23201619, 11.83570132,  5.13182506, 13.54774355, 13.40024224,
       13.35832725, 10.86693708,  7.34250948, 10.8207297 , 10.4977132 ,
       12.4980845 ,  8.31625171,  6.76056625, 13.6660671 , 14.10045797,
        4.78544875,  8.7273569 , 11.51811603, 13.05148826,  7.66377554,
        6.78325151, 10.24557047, 13.37121923, 10.08001894, 11.74901778,
       12.12448912])}
{'P': tensor([[1.9860e-03, 4.7521e-01, 1.9804e-01,  ..., 3.7466e-07, 6.2594e-07,
         7.4534e-08],
        [2.0835e-05, 1.0875e-02, 3.7778e-01,  ..., 2.5787e-06, 2.8164e-06,
         4.7891e-07],
        [6.5917e-04, 3.5387e-03, 2.9762e-01,  ..., 7.8690e-05, 6.4481e-05,
         1.5141e-05],
        ...,
        [3.1343e-07, 2.8761e-07, 2.5784e-07,  ..., 2.5467e-02, 6.4339e-05,
         2.2355e-02],
        [7.9368e-03, 6.4653e-03, 4.9770e-03,  ..., 4.6217e-05, 1.0931e-01,
         2.9442e-05],
        [4.2475e-07, 3.5558e-07, 3.1684e-07,  ..., 4.2492e-05, 1.1799e-04,
         1.7315e-02]]), 'X': tensor([[  2.3373,  13.7721,  -0.2442],
        [  1.4306,  13.9620,  -0.2075],
        [  0.7275,  14.0667,  -0.9815],
        [  0.6142,  14.0802,   0.5407],
        [  1.1350,  14.0104,   0.6321],
        [  1.6308,  13.9254,  -2.0984],
        [  0.9478,  -4.8144,   0.6371],
        [  6.2036,   2.4030,  -1.2500],
        [  5.6022,  -1.9833,   8.3081],
        [  0.1931,  -4.7491,   7.4385],
        [ -9.3739,   0.3982,  -4.2564],
        [ -8.3900,   3.3727,   3.1650],
        [  6.0310,  12.2795,  -6.4528],
        [  2.1379,  -4.6757,   6.4981],
        [ 12.2914,   3.8037,   6.0978],
        [  6.3542,   0.6934,  -3.3840],
        [ -8.8363,   2.4086,  -5.6296],
        [ -2.6485, -10.4151,   0.8766],
        [  5.4242, -10.1261,   3.9595],
        [  6.3583,   1.1729,   8.7835],
        [ -9.4773,  -1.0024,   5.2599],
        [ -1.9600,   7.8577,   6.9121],
        [ -6.8466,  -7.6488,   1.9648],
        [  5.9526,   3.2525,  -4.3821],
        [ 10.8670,  -5.3428,  -2.8637],
        [ -1.1885, -10.8211,   0.0865],
        [ -9.3739,   0.3982,  -4.2564],
        [  2.3889,  -4.6082,  -8.7291],
        [ -9.4731,  -0.7120,  -6.0271],
        [  0.1677,  -4.7447,  -6.1529],
        [ 11.3691,   6.4760,   1.0787],
        [ -8.1426,   3.8048,   9.3542],
        [  4.2096,   5.8815,   1.9230],
        [ -9.2345,  -3.1345,   3.9715],
        [  0.9809,   7.6642,  -1.4217],
        [ -9.3107,  -2.7697,  -3.2919],
        [ -2.6069, -10.4298,  -6.1995],
        [  6.3534,   1.2990,  -7.2942],
        [ -3.9456,  -9.8542,   0.7466],
        [ -1.0482, -10.8487,  -5.9066],
        [ -2.5056,   7.7720,   5.5878],
        [  6.1929,   2.4501,  -6.9352],
        [  8.2154,  -8.4262,   8.0171],
        [  0.7958,  -4.8109,  -8.2248],
        [  1.7500,   7.4187,   4.4237],
        [  4.7544,  -3.1425,  -1.5315],
        [ 10.2712,   8.3707,  -4.9743],
        [  1.7500,   7.4187,   4.4237],
        [ -0.5385,   7.9037,   6.3830],
        [ -0.3432,   7.8900,   7.1805],
        [  0.6757, -11.0361,  -1.3047],
        [  5.3924, -10.1396,  -4.0863],
        [ -3.2486,   7.5925,  -1.3719],
        [  5.7802,  -1.6363,  -6.2782],
        [ -8.3539,  -5.5111,   0.3373],
        [ -1.1885, -10.8211,   0.0865],
        [ 12.5832,   1.7543,  -7.6134],
        [  2.3490,  -4.6199,  -4.2489],
        [  6.3604,   0.9042,  -6.5901],
        [  1.7122,  -4.7591,  -3.6734],
        [  6.9433,  -9.3362,  -0.3421],
        [  2.7420,  -4.4896,  -3.7139],
        [  5.8275,  -1.5334,   2.0124],
        [  4.3523, -10.5232,   8.2138],
        [ -7.9925,  -6.1423,   2.8990],
        [  5.8930,  12.3592,  -8.9201],
        [ 12.1379,   4.4125,  -3.8833],
        [  9.3607,   9.5337,  -4.5412],
        [ -3.2736, -10.1703,   3.8712],
        [  3.6817,   6.3354,   6.1233],
        [ 10.1503,   8.5414,  -7.2681],
        [  6.2804,   2.0019,   0.8607],
        [  2.0899,  -4.6870,  -7.4260],
        [  0.1677,  -4.7447,  -6.1529],
        [ -7.7239,   4.4298,   3.1216],
        [ 10.8670,  -5.3428,  -2.8637],
        [  7.0171,  11.6439,  -5.0178],
        [ 10.0619,   8.6627,  -9.2884],
        [ -0.8455, -10.8853,  -6.7400],
        [  3.3167,   6.6022,   6.3554],
        [ -9.3705,   0.4208,  -3.5815],
        [  7.2340,  -9.1505,  -5.8321],
        [ -8.8934,   2.2597,   3.5831],
        [  5.0233,   4.9647,  -3.1788],
        [  0.1047, -11.0046,   0.5188],
        [  3.1892, -10.8220,  -6.8409],
        [  2.5636, -10.9293,   9.6519],
        [  9.3607,   9.5337,  -4.5412],
        [  7.9141,  -8.6656,   9.6948],
        [  6.3515,   0.6418,  -6.5823],
        [ -8.6675,  -4.8569,  -9.1769],
        [  2.2182,  13.8007,   5.1889],
        [  2.7295,  13.6703,  -8.9328],
        [ -3.5173, -10.0623,   4.8280],
        [  5.0348,  -2.8229,   3.5436],
        [ -9.1178,   1.5784,   2.1410],
        [  1.7301,   7.4262,   5.0522],
        [ -2.6069, -10.4298,  -6.1995],
        [ -1.8823, -10.6558,  -3.8952],
        [ -8.5900,   2.9767,   7.4811],
        [ 11.3800,  -4.3919,   1.4066],
        [  3.2809,  13.5061,  -4.0199],
        [  5.0506,   4.9277,  -9.5094],
        [ -6.8618,  -7.6322,   3.7027],
        [  6.2905,   1.9366,   2.1169],
        [ -2.6735, -10.4062,  -1.5911]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[2.1453e-04, 5.2609e-02, 1.8524e-01,  ..., 1.1715e+00, 2.4883e-01,
         9.8965e-01],
        [5.2609e-02, 0.0000e+00, 1.4258e-01,  ..., 1.1202e+00, 2.0959e-01,
         9.3787e-01],
        [1.8524e-01, 1.4258e-01, 0.0000e+00,  ..., 1.0449e+00, 2.4924e-01,
         8.1915e-01],
        ...,
        [1.1715e+00, 1.1202e+00, 1.0449e+00,  ..., 0.0000e+00, 9.5136e-01,
         4.6063e-01],
        [2.4883e-01, 2.0959e-01, 2.4924e-01,  ..., 9.5136e-01, 0.0000e+00,
         8.4166e-01],
        [9.8965e-01, 9.3787e-01, 8.1915e-01,  ..., 4.6063e-01, 8.4166e-01,
         2.1453e-04]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.96905756, 14.03505624, 14.08549802, 14.09357547, 14.05633441,
       14.02058621,  4.90676242,  6.652742  ,  5.94293185,  4.75303127,
        9.38232121,  9.04255655, 13.68063098,  5.14124612, 12.86648771,
        6.39187285,  9.15866607, 10.74655346, 11.48734978,  6.46560181,
        9.53015376,  8.09842704, 10.26545792,  6.78325151, 12.10942412,
       10.88618114,  9.38232121,  5.19065439,  9.4997987 ,  4.74770899,
       13.08414689,  8.98765644,  7.23276582,  9.75200848,  7.72668467,
        9.71392179, 10.7506423 ,  6.48486171, 10.61472636, 10.8992524 ,
        8.16584537,  6.65991877, 11.76830981,  4.8763135 ,  7.62232307,
        5.69913817, 13.25017211,  7.62232307,  7.92200342,  7.89745107,
       11.05672136, 11.48434767,  8.25828409,  6.00732193, 10.00794468,
       10.88618114, 12.70489563,  5.18277858,  6.42439344,  5.05774295,
       11.63502679,  5.26067019,  6.02588517, 11.38774149, 10.08001894,
       13.6922425 , 12.91505111, 13.36092624, 10.68416401,  7.32754882,
       13.26590071,  6.59176018,  5.13182506,  4.74770899,  8.90405823,
       12.10942412, 13.59480906, 13.27717982, 10.91805683,  7.38844703,
        9.37989631, 11.66453059,  9.17595899,  7.06272256, 11.00508681,
       11.28215429, 11.22596994, 13.36092624, 11.73567759,  6.383897  ,
        9.93553424, 13.97779839, 13.94008948, 10.65929968,  5.77219077,
        9.25336371,  7.62509582, 10.7506423 , 10.8207297 ,  9.09119351,
       12.19805394, 13.89886469,  7.05627461, 10.26327895,  6.58184474,
       10.74409834])}
{'P': tensor([[1.1885e-02, 5.9396e-01, 1.0513e-01,  ..., 6.6595e-07, 6.7541e-03,
         2.6511e-06],
        [7.2707e-04, 2.3639e-01, 4.3585e-04,  ..., 1.5960e-05, 3.6406e-02,
         5.4749e-05],
        [1.2964e-04, 9.7180e-01, 4.4047e-03,  ..., 3.0707e-07, 3.1615e-03,
         9.6465e-07],
        ...,
        [3.5490e-05, 3.2920e-05, 4.1367e-05,  ..., 6.9085e-02, 6.4328e-05,
         4.1320e-04],
        [3.8307e-03, 5.1273e-02, 1.1246e-02,  ..., 2.8293e-05, 2.5170e-01,
         4.0357e-05],
        [1.0230e-04, 7.1684e-05, 8.4715e-05,  ..., 2.1140e-03, 4.8987e-05,
         1.5053e-01]]), 'X': tensor([[  2.3261,  13.7749,   5.9121],
        [  0.5175,  14.0910,   4.6540],
        [  1.5662,  13.9375,   4.5142],
        [  1.1274,  14.0116,   8.1753],
        [  1.3705,  13.9723,   7.9266],
        [  1.5653,  13.9377,   4.1773],
        [  1.3705,  13.9723,   7.9266],
        [  2.2373,  -4.6506,  -1.7334],
        [  4.6992,   5.3690,   3.0935],
        [ -6.3489,   5.9117,   6.5161],
        [ 11.2847,  -4.5848,  -4.7503],
        [ 10.4737,  -5.9570,  -1.2108],
        [ -9.0604,   1.7707,   1.1066],
        [ 12.5832,   1.7543,  -7.6134],
        [ 11.2939,   6.6325,  -5.8311],
        [ -7.6719,  -6.6260,   7.1468],
        [ 12.5154,   2.5072,  -9.3274],
        [  4.1297,  -3.7098,  -3.9838],
        [  8.3718,  -8.2953,   8.4666],
        [  6.1989,  -9.7591,   0.8289],
        [  2.3261,   7.1726,   6.2522],
        [  5.4429,  -2.2529,   9.7306],
        [ -8.0608,   3.9362,  -5.9567],
        [ 11.3800,  -4.3919,   1.4066],
        [ -8.6675,  -4.8569,  -9.1769],
        [  5.9026,   3.3837,  -0.6573],
        [  4.5430,   5.5438,  -3.8669],
        [ 12.2914,   3.8037,   6.0978],
        [ -9.3810,  -2.3455,  -9.5194],
        [  9.0051,   9.9234,   5.4411],
        [ -1.8788, -10.6567,   3.7068],
        [  9.3560,   9.5391,   9.4943],
        [  2.4572,   7.1085,   7.7376],
        [ -9.4756,  -0.8166,  -9.6021],
        [  0.6396,  14.0772,  -6.3786],
        [ -8.9282,   2.1648,   8.4894],
        [ -4.7847,  -9.3729,  -3.7741],
        [  2.5663,   7.0526,  -6.8769],
        [  2.5663,   7.0526,  -6.8769],
        [  5.1787,  -2.6383,  -4.2533],
        [  2.2849, -10.9655,  -7.6542],
        [ 11.9695,  -2.9407,  -6.3498],
        [  6.3301,   0.3638,  -7.5685],
        [ -6.3066,  -8.1936,   9.5822],
        [ -9.2373,  -3.1225,  -8.9763],
        [  0.3493,  -4.7727,  -6.5682],
        [  1.7122,  -4.7591,  -3.6734],
        [  5.2199,   4.6861,  -1.9304],
        [  6.3589,   0.8239,  -8.4230],
        [ -9.3705,   0.4208,  -3.5815],
        [  2.5469,  13.7192,  -9.6935],
        [  6.8407,  11.7666,   3.0045],
        [ -9.2974,   0.8371,   2.4852],
        [ -6.0844,  -8.3955,   0.7705],
        [ -6.3066,  -8.1936,   9.5822],
        [ -2.0840,   7.8415,  -7.8657],
        [ -9.4624,  -0.4633,  -7.3918],
        [ -6.2668,   5.9808,   2.5614],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.3459,  -3.5327,   0.3667],
        [ -3.2486,   7.5925,  -1.3719],
        [ -5.9940,   6.1985,   5.9566],
        [ 11.7726,  -3.4876,  -6.7402],
        [  5.0506,   4.9277,  -9.5094],
        [ -9.2974,   0.8371,   2.4852],
        [ 11.6544,   5.8273,  -3.1375],
        [ -9.2301,  -3.1537,  -2.7477],
        [  1.3670,  -4.7990,   3.2603],
        [  6.8093,  -9.4178,   8.4244],
        [  6.7565,  11.8237,  -4.9882],
        [ -9.3705,   0.4208,  -3.5815],
        [  6.3497,   1.3680,  -1.9076],
        [  1.6954,  -4.7616,   9.8198],
        [ -5.8389,  -8.6056,  -3.5098],
        [  5.7963,  -1.6018,   1.8856],
        [  6.0345,   3.0175,  -1.4424],
        [ 11.2847,  -4.5848,  -4.7503],
        [ -9.4366,  -0.1189,  -2.6736],
        [  6.3360,   0.4258,  -9.2509],
        [ 10.4994,  -5.9193,  -7.5153],
        [ -6.4822,  -8.0252,  -3.8983],
        [ 11.0373,   7.1299,  -8.5895],
        [ 12.2311,  -2.0445,   1.3545],
        [  3.6249, -10.7255,   5.6605],
        [ -5.7347,  -8.6908,   9.2878],
        [ -3.0004,   7.6607,  -7.2640],
        [ -9.3739,   0.3982,  -4.2564],
        [  1.9312,  13.8649,   8.3563],
        [  5.4336,  -2.2676,   5.6933],
        [ -5.5525,   6.5146,  -4.4984],
        [ -8.6381,  -4.9238,  -4.4077],
        [  2.4572,   7.1085,   7.7376],
        [ -9.4524,  -1.6813,  -6.6862],
        [  0.9478,  -4.8144,   0.6371],
        [  1.7659,   7.4127,   4.7871],
        [ -4.4503,   7.1378,   3.3751],
        [  6.2022,   2.4092,   3.0349],
        [ -5.3179,  -9.0101,   9.4472],
        [  1.8447,  -4.7372,   8.1235],
        [  1.7815,  -4.7481,   4.4289],
        [  1.4749,  -4.7892,  -0.8089],
        [ -3.6381,   7.4683,   7.3544],
        [  3.2809,  13.5061,  -4.0199],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.6142,  14.0802,   0.5407],
        [  0.9809,  -4.8145,   6.8706]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 9.3098e-02, 6.4062e-02,  ..., 9.2888e-01, 5.8247e-01,
         3.6615e-01],
        [9.3098e-02, 0.0000e+00, 9.2061e-02,  ..., 8.4596e-01, 4.9944e-01,
         2.7912e-01],
        [6.4062e-02, 9.2061e-02, 3.0339e-04,  ..., 8.8326e-01, 5.3836e-01,
         3.6832e-01],
        ...,
        [9.2888e-01, 8.4596e-01, 8.8326e-01,  ..., 0.0000e+00, 3.4663e-01,
         7.1101e-01],
        [5.8247e-01, 4.9944e-01, 5.3836e-01,  ..., 3.4663e-01, 0.0000e+00,
         4.0036e-01],
        [3.6615e-01, 2.7912e-01, 3.6832e-01,  ..., 7.1101e-01, 4.0036e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.96987806, 14.10045797, 14.02526505, 14.05687815, 14.0393949 ,
       14.02532758, 14.0393949 ,  5.16077764,  7.13501264,  8.67502348,
       12.18045978, 12.04924159,  9.23178204, 12.70489563, 13.09737053,
       10.1371552 , 12.76408562,  5.5513072 , 11.785566  , 11.56146873,
        7.5403822 ,  5.89074104,  8.97052115, 12.19805394,  9.93553424,
        6.80370254,  7.16747878, 12.86648771,  9.66977802, 13.40024224,
       10.82106505, 13.3614598 ,  7.52117132,  9.51074763, 14.09176259,
        9.18690018, 10.52357479,  7.50500064,  7.50500064,  5.81199422,
       11.20100795, 12.32545637,  6.34059021, 10.33958639,  9.75075273,
        4.78544875,  5.05774295,  7.01475376,  6.41202986,  9.37989631,
       13.95361098, 13.6105595 ,  9.3349858 , 10.36844813, 10.33958639,
        8.1137452 ,  9.47369628,  8.66273171, 12.05300943,  5.60064319,
        8.25828409,  8.62260327, 12.27835996,  7.05627461,  9.3349858 ,
       13.03002338,  9.75401449,  4.98989906, 11.62158554, 13.61801526,
        9.37989631,  6.49538841,  5.05444288, 10.39942497,  6.01356004,
        6.74686071, 12.18045978,  9.43737523,  6.35029274, 12.05300943,
       10.31613269, 13.1399181 , 12.40074901, 11.32149259, 10.412299  ,
        8.22728399,  9.38232121, 13.99877413,  5.88782913,  8.55982105,
        9.94285511,  7.52117132,  9.60080639,  4.90676242,  7.62011734,
        8.41148722,  6.65368551, 10.46238054,  5.08372149,  5.0713364 ,
        5.01114719,  8.30729352, 13.89886469,  5.35774573, 14.09357547,
        4.91336994])}
{'P': tensor([[1.5417e-02, 2.8974e-01, 1.9989e-01,  ..., 2.4451e-07, 1.6399e-07,
         2.7778e-06],
        [1.9298e-03, 3.2110e-01, 2.5835e-02,  ..., 7.0987e-06, 5.1351e-06,
         5.4582e-05],
        [3.2892e-03, 1.4116e-01, 2.2986e-01,  ..., 3.7129e-06, 2.2733e-06,
         4.4000e-05],
        ...,
        [2.2394e-06, 1.6453e-06, 1.4905e-06,  ..., 5.8593e-02, 2.7221e-02,
         3.5877e-03],
        [7.8785e-07, 6.3999e-07, 4.8147e-07,  ..., 3.3551e-03, 2.5648e-02,
         4.2574e-04],
        [1.5827e-04, 1.0113e-04, 1.3268e-04,  ..., 1.9172e-03, 8.9898e-04,
         1.8482e-01]]), 'X': tensor([[ 2.2659e+00,  1.3789e+01,  6.5436e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.5662e+00,  1.3938e+01,  4.5142e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 6.2036e+00,  2.4030e+00, -1.2500e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [-7.8280e+00,  4.2848e+00, -7.1955e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [-8.6050e+00,  2.9451e+00, -7.0117e-01],
        [-8.7735e+00,  2.5640e+00, -6.3313e+00],
        [ 5.3804e+00, -2.3504e+00,  7.2744e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 1.0916e+01, -5.2599e+00, -7.0812e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [-2.5859e-01,  7.8825e+00, -9.9840e+00],
        [-3.4319e-01,  7.8900e+00,  7.1805e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [ 4.9337e+00,  1.2857e+01, -3.6582e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 5.6202e+00, -1.9508e+00,  7.4706e-01],
        [ 5.4167e+00, -1.0129e+01,  6.2156e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [ 3.3474e+00, -4.2162e+00,  5.2455e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-9.4159e+00, -2.0739e+00,  9.6841e+00],
        [-8.5607e+00, -5.0938e+00, -5.4531e+00],
        [ 4.5800e+00,  5.5034e+00, -3.9432e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [-8.6801e+00,  2.7809e+00, -8.4433e-01],
        [-6.4464e+00,  5.8272e+00,  7.5038e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-2.1533e+00, -1.0578e+01,  1.0112e+00],
        [ 1.0198e+01, -6.3430e+00, -3.1996e-01],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.1981e+01,  4.9340e+00, -3.3715e+00],
        [ 1.2459e+01, -9.1682e-01, -6.7571e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [-9.4351e+00, -1.8891e+00,  3.5642e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [ 5.6248e+00,  3.9984e+00,  8.4933e+00],
        [ 1.1277e+01, -4.6006e+00,  7.8682e-01],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 2.3490e+00, -4.6199e+00, -4.2489e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [ 4.5430e+00,  5.5438e+00, -3.8669e+00],
        [ 8.3718e+00, -8.2953e+00,  8.4666e+00],
        [ 8.5093e+00, -8.1764e+00, -3.2381e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.0415e-01, -1.1005e+01,  2.4311e+00],
        [ 5.6003e-01, -1.1032e+01, -8.6603e-01],
        [ 1.7228e+00, -4.7575e+00,  8.3396e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.5876e-01, 1.2649e-01,  ..., 1.0485e+00, 1.2174e+00,
         3.6106e-01],
        [1.5876e-01, 0.0000e+00, 1.7380e-01,  ..., 9.1984e-01, 1.0690e+00,
         2.3649e-01],
        [1.2649e-01, 1.7380e-01, 0.0000e+00,  ..., 9.5646e-01, 1.1489e+00,
         2.8492e-01],
        ...,
        [1.0485e+00, 9.1984e-01, 9.5646e-01,  ..., 2.1453e-04, 3.1976e-01,
         6.8853e-01],
        [1.2174e+00, 1.0690e+00, 1.1489e+00,  ..., 3.1976e-01, 0.0000e+00,
         8.6443e-01],
        [3.6106e-01, 2.3649e-01, 2.8492e-01,  ..., 6.8853e-01, 8.6443e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.97430191, 14.10045797, 14.05687815, 14.0393949 , 14.02526505,
       14.02532758,  6.652742  , 13.35832725, 11.46176259,  5.19577146,
       11.05317597,  8.35856487,  7.83534987, 11.35182589, 11.00508681,
        8.92395381, 11.72802491,  8.74470218,  7.37902766, 10.5575601 ,
        9.09502385,  9.14044984,  5.87132718,  6.7593917 , 11.11038551,
       12.11735334, 13.35832725, 13.66991027,  5.36061974,  9.96391045,
        7.24326992, 10.05558893,  7.88677538,  7.89745107,  9.04255655,
       12.38881635, 12.41366754, 12.21532282,  7.92644746, 13.77075174,
       10.31584102,  7.53884104, 11.28895489,  5.94909709, 11.48663767,
        9.78285424, 13.40024224, 12.59840193, 10.51437947, 12.79126785,
        9.3109884 , 12.45418236, 13.61801526,  5.38342077,  8.09704052,
       10.19036081,  7.92607342, 10.75443106, 10.61047596, 12.32545637,
        9.64157461,  9.96152788,  7.15990676,  7.23276582, 11.83075005,
        9.11473391,  8.68979052, 13.744387  ,  5.07306274, 10.794755  ,
       12.00995898,  8.53501777, 12.95702647, 12.49291768, 13.04156075,
        6.81084724, 13.44593519,  7.32950385, 13.32613445,  9.62238828,
        6.67121748,  6.90115901, 12.17901033,  6.38105015, 14.05687815,
        8.7273569 ,  5.18277858, 11.35182589,  6.84796202,  6.76056625,
        7.16747878, 11.785566  , 11.80092578,  9.79814997, 13.32613445,
        6.28835101,  5.39723241, 10.01804223, 10.8992524 , 13.13888841,
       11.76830981, 10.87409005,  8.81930525, 11.0050379 , 11.04629471,
        5.05981914])}
{'P': tensor([[5.1477e-03, 6.6178e-01, 2.1411e-01,  ..., 1.2065e-05, 2.3012e-08,
         2.8187e-07],
        [3.4379e-04, 2.0204e-01, 3.7251e-04,  ..., 8.5632e-04, 8.2349e-07,
         9.4052e-06],
        [4.7661e-05, 9.7125e-01, 4.4022e-03,  ..., 1.3929e-05, 1.9487e-08,
         2.4883e-07],
        ...,
        [3.1846e-06, 3.5669e-06, 3.4405e-06,  ..., 2.3571e-01, 2.6366e-04,
         1.8796e-04],
        [1.7591e-04, 9.8782e-05, 1.4901e-04,  ..., 4.6170e-06, 1.1603e-01,
         7.9513e-03],
        [2.0555e-04, 1.8768e-04, 2.3457e-04,  ..., 4.6432e-05, 1.3330e-04,
         7.8097e-02]]), 'X': tensor([[  2.2182,  13.8007,   5.1889],
        [  0.5175,  14.0910,   4.6540],
        [  1.5662,  13.9375,   4.5142],
        [  1.5653,  13.9377,   4.1773],
        [  1.1274,  14.0116,   8.1753],
        [  1.3705,  13.9723,   7.9266],
        [ -2.2188,   7.8218,   8.6064],
        [  4.0591,   6.0202,  -1.6339],
        [  1.8447,  -4.7372,   8.1235],
        [ -2.0445,   7.8469,  -0.5857],
        [  4.6063,  13.0051,  -3.1456],
        [  3.2990,  -4.2415,  -7.4857],
        [  4.5340,  -3.3629,   8.9652],
        [ 11.7779,   5.5127,  -2.0699],
        [  1.2728, -11.0372,   9.8121],
        [  6.3505,   0.6235,  -8.5849],
        [ -1.0482, -10.8487,  -5.9066],
        [ -6.3306,   5.9272,   5.6935],
        [  6.1372,   2.6726,  -6.7885],
        [  2.5974,   7.0363,  -7.7810],
        [  4.3459,  -3.5327,   0.3667],
        [ 11.2209,  -4.7092,   2.6117],
        [ -5.0132,  -9.2234,  -8.3531],
        [  5.3804,  -2.3504,   7.2744],
        [ -0.0410,   7.8591,   9.3457],
        [  6.0140,  -1.0661,  -2.5040],
        [ -2.7572,   7.7194,   5.9580],
        [ -0.7599, -10.8995,  -6.0366],
        [ -2.6485, -10.4151,   0.8766],
        [ 11.7418,   5.6070,  -1.4192],
        [ -3.2736, -10.1703,   3.8712],
        [  8.8474,  10.0868,  -8.8644],
        [  7.0642,  11.6104,  -9.8316],
        [  8.1043,  -8.5164,  -7.9290],
        [ -8.6681,  -4.8556,   9.2518],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.5506,  -5.1154,   6.3423],
        [  0.9478,  -4.8144,   0.6371],
        [ 10.5678,   7.9261,  -0.6100],
        [  5.5240,  -2.1198,   6.2055],
        [ -7.6525,  -6.6534,   5.1721],
        [ -8.7078,   2.7182,   1.8146],
        [ -9.4739,  -1.2497,   0.5483],
        [  4.9452, -10.3187,  -7.8621],
        [ -9.1749,  -3.3803,   6.4629],
        [ -3.5173, -10.0623,   4.8280],
        [ -1.5408,   7.8978,   5.3470],
        [  2.6613,   7.0021,   4.5783],
        [ -5.9872,   6.2037,  -9.4477],
        [  6.1755,   2.5233,  -2.0042],
        [ -9.2345,  -3.1345,   3.9715],
        [  6.3583,   1.1729,   8.7835],
        [ -9.4714,  -1.3301,  -3.1998],
        [  8.2154,  -8.4262,   8.0171],
        [  3.3633,  13.4794,   0.5193],
        [  6.3087,   1.8053,  -1.9481],
        [ -8.6801,   2.7809,  -0.8443],
        [  4.1140,  -3.7220,  -0.1273],
        [  4.2339,  -3.6268,   6.3125],
        [ -0.8455, -10.8853,  -6.7400],
        [ 12.6035,   0.7494,   7.1536],
        [ -1.5261,   7.8988,   9.3768],
        [  5.0190,   4.9705,  -7.9752],
        [ -8.0087,   4.0173,   9.1091],
        [  6.0890,  -0.8382,   2.9166],
        [ 10.9931,   7.2107,   7.7933],
        [  0.2480, -11.0153,  -3.0183],
        [ -0.0282,   7.8575,  -1.1746],
        [ -8.3900,   3.3727,   3.1650],
        [  4.3523, -10.5232,   8.2138],
        [  4.1637,  13.1898,  -1.9189],
        [ -6.3306,   5.9272,   5.6935],
        [  5.9026,   3.3837,  -0.6573],
        [ -8.8058,   2.4850,  -1.7425],
        [ -6.0195,   6.1789,  -2.9496],
        [  1.8721,  -4.7322,   0.5524],
        [  7.0642,  11.6104,  -9.8316],
        [ -5.9987,   6.1949,  -6.2114],
        [  0.1206,  -4.7363,   8.7103],
        [  5.1472,  -2.6801,   8.1194],
        [ -3.4986,   7.5152,   9.0712],
        [  3.2226,  -4.2802,  -4.0428],
        [  0.9036,   7.6840,   9.9237],
        [  8.8474,  10.0868,  -8.8644],
        [ -8.3717,  -5.4769,  -9.2228],
        [ -4.2040,   7.2486,   2.9875],
        [  0.1382, -11.0073,   1.0026],
        [  9.9750,  -6.6338,  -1.2440],
        [ -4.4520,   7.1370,  -8.7867],
        [  2.2849, -10.9655,  -7.6542],
        [ -9.4159,  -2.0739,   9.6841],
        [  2.5813,  -4.5471,   5.3597],
        [  1.4749,  -4.7892,  -0.8089],
        [  2.1625,  -4.6697,   7.0658],
        [  5.0506,   4.9277,  -9.5094],
        [  2.2644,  -4.6434,   9.7508],
        [  0.9478,  -4.8144,   0.6371],
        [  0.9036,   7.6840,   9.9237],
        [  5.0617,  12.7957,   8.3544],
        [  7.0642,  11.6104,  -9.8316],
        [ -7.0825,   5.2073,  -5.0497],
        [  6.1066,  -0.7799,  -4.1746],
        [ -9.4731,  -0.7120,  -6.0271],
        [ -9.4283,  -1.9585,  -0.9129],
        [ 10.1983,  -6.3430,  -0.3200],
        [  6.3535,   0.6792,  -7.5101]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.0547e-01, 2.6039e-02,  ..., 1.3846e+00, 7.6095e-01,
         1.0131e+00],
        [1.0547e-01, 0.0000e+00, 9.2061e-02,  ..., 1.2812e+00, 6.7978e-01,
         9.5992e-01],
        [2.6039e-02, 9.2061e-02, 3.0339e-04,  ..., 1.3653e+00, 7.3591e-01,
         9.8715e-01],
        ...,
        [1.3846e+00, 1.2812e+00, 1.3653e+00,  ..., 0.0000e+00, 7.7762e-01,
         1.0005e+00],
        [7.6095e-01, 6.7978e-01, 7.3591e-01,  ..., 7.7762e-01, 0.0000e+00,
         3.9601e-01],
        [1.0131e+00, 9.5992e-01, 9.8715e-01,  ..., 1.0005e+00, 3.9601e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.97779839, 14.10045797, 14.02526505, 14.02532758, 14.05687815,
       14.0393949 ,  8.1303945 ,  7.26074925,  5.08372149,  8.10886021,
       13.79676654,  5.37343173,  5.64500512, 13.00413829, 11.11038551,
        6.38105015, 10.8992524 ,  8.67227386,  6.69389431,  7.50035355,
        5.60064319, 12.16901615, 10.4977132 ,  5.87132718,  7.85919687,
        6.10774544,  8.19703512, 10.92597196, 10.74655346, 13.01187484,
       10.68416401, 13.41713661, 13.59056613, 11.75618513,  9.9353921 ,
       13.99048094,  9.96391045,  4.90676242, 13.20988996,  5.91677188,
       10.14044489,  9.12220271,  9.55593374, 11.4424816 ,  9.77777346,
       10.65929968,  8.04664886,  7.49076899,  8.62162372,  6.67108662,
        9.75200848,  6.46560181,  9.5642993 , 11.76830981, 13.89264822,
        6.5618968 ,  9.11473391,  5.54776934,  5.5748624 , 10.91805683,
       12.62576026,  8.04483495,  7.06372172,  8.95983586,  6.1463896 ,
       13.14690137, 11.018084  ,  7.8575648 ,  9.04255655, 11.38774149,
       13.83139024,  8.67227386,  6.80370254,  9.14973319,  8.62631801,
        5.08910491, 13.59056613,  8.62329466,  4.73784222,  5.80312949,
        8.28968067,  5.35774573,  7.73692436, 13.41713661, 10.00410625,
        8.37954853, 11.00812996, 11.97949273,  8.41171369, 11.20100795,
        9.64157461,  5.2287058 ,  5.01114719,  5.14607014,  7.05627461,
        5.16611782,  4.90676242,  7.73692436, 13.76047999, 13.59056613,
        8.7907923 ,  6.1561646 ,  9.4997987 ,  9.62958973, 12.00995898,
        6.38968357])}
{'P': tensor([[4.3377e-03, 5.6903e-01, 1.9511e-01,  ..., 5.7499e-07, 4.7749e-04,
         2.3271e-04],
        [3.7305e-04, 1.8704e-01, 2.4368e-03,  ..., 2.1160e-05, 3.7693e-03,
         5.4062e-03],
        [2.4168e-04, 4.9927e-01, 2.1911e-02,  ..., 2.5536e-06, 1.5838e-03,
         7.8395e-04],
        ...,
        [9.9036e-04, 7.4179e-04, 7.8030e-04,  ..., 7.6637e-02, 2.5624e-04,
         8.8704e-03],
        [6.4226e-03, 1.9986e-02, 1.5844e-02,  ..., 5.2472e-06, 8.2488e-02,
         7.2666e-04],
        [1.9060e-03, 3.9362e-03, 2.1187e-03,  ..., 1.0283e-04, 3.8812e-04,
         4.9906e-02]]), 'X': tensor([[ 2.2076e+00,  1.3803e+01, -1.5099e+00],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 1.6308e+00,  1.3925e+01, -2.0984e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 5.8889e+00, -1.3916e+00,  8.8993e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-6.7992e+00,  5.4991e+00,  9.9186e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.1597e+01,  5.9666e+00,  6.0083e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 5.2199e+00,  4.6861e+00, -1.9304e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-7.6525e+00, -6.6534e+00,  5.1721e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 3.7916e+00, -3.9524e+00, -7.9547e+00],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 9.9750e+00, -6.6338e+00, -1.2440e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 6.3516e+00,  6.4250e-01,  3.4254e-01],
        [-5.9987e+00,  6.1949e+00, -6.2114e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [-9.4430e+00, -1.8972e-01,  4.8383e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-8.6603e+00,  2.8250e+00,  5.9181e+00],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.0040e+01,  8.6922e+00, -6.2847e+00],
        [ 1.0822e+01, -5.4177e+00,  6.8573e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 1.1285e+01, -4.5848e+00, -4.7503e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 2.0729e+00,  7.2878e+00,  7.0880e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.2579e+01,  1.7221e-01,  1.6470e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 3.3943e+00,  6.5483e+00,  1.0790e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-9.4524e+00, -1.6813e+00, -6.6862e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [ 6.1575e+00,  1.2205e+01,  3.3257e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 6.3031e+00,  1.8477e+00, -7.1706e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-3.7964e+00, -9.9294e+00,  7.7147e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [ 6.3303e+00,  1.6145e+00, -8.4083e+00],
        [-9.4697e+00, -6.1566e-01, -6.8472e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 1.0093e+00,  7.6567e+00,  2.3526e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1348, 0.0698,  ..., 0.3623, 0.4079, 0.4972],
        [0.1348, 0.0000, 0.1294,  ..., 0.2880, 0.4618, 0.3737],
        [0.0698, 0.1294, 0.0000,  ..., 0.3994, 0.3528, 0.5029],
        ...,
        [0.3623, 0.2880, 0.3994,  ..., 0.0000, 0.7478, 0.2385],
        [0.4079, 0.4618, 0.3528,  ..., 0.7478, 0.0000, 0.8055],
        [0.4972, 0.3737, 0.5029,  ..., 0.2385, 0.8055, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.97857591, 14.08549802, 14.02058621, 14.09357547, 14.03505624,
       14.10796466,  6.05113217,  9.78285424,  8.74470218, 10.98076873,
       13.04156075, 13.3607379 ,  6.4020404 ,  7.01475376, 14.10796466,
       10.14044489, 14.03618649, 12.4217523 ,  6.31576264, 10.09979747,
        5.47702766,  8.09704052, 13.44593519, 13.36800073,  6.383897  ,
        4.79003361, 12.67576792,  8.31625171, 11.97949273, 13.40850328,
       13.93790235,  6.38399789,  8.62329466, 13.46553623,  8.86082124,
       13.32613445, 11.97957425,  6.2134961 ,  5.16611782, 10.00802627,
       11.11038551,  6.31576264,  9.44486609,  5.70405525,  9.10945678,
        8.86082124, 10.86693708, 10.88618114,  6.61386436,  6.3949043 ,
        9.3109884 , 13.41713661, 11.89105259, 10.67322527, 13.27994304,
       12.10222204, 12.49317563, 14.03505624,  8.57560275, 12.18045978,
        6.3949043 , 13.99877413, 13.32557133, 12.97370371,  7.57685908,
       11.8159254 , 10.53368644,  6.80559669,  5.19986797, 12.19805394,
       11.11969815,  7.83486943,  7.62011734, 12.58006038,  5.5513072 ,
        5.04028538,  9.36378438,  7.37576154, 10.87409005,  9.60080639,
        5.13125513, 13.66991027,  9.26455795,  8.29640613,  9.09963894,
        9.5642993 ,  8.36445758,  7.18354885,  6.56833864, 10.87409005,
        9.94173708,  9.26455795,  5.83560009,  9.18690018, 10.63038565,
       13.58101794,  7.76457095, 10.57302546,  8.44940661, 10.26545792,
        6.53290663,  9.48969978, 10.91805683,  6.1463896 , 13.99048094,
        7.72291946])}
{'P': tensor([[2.9263e-03, 4.1134e-01, 8.4372e-02,  ..., 6.5428e-07, 1.6387e-05,
         6.1140e-08],
        [1.1741e-04, 2.5931e-02, 5.9088e-01,  ..., 5.9484e-06, 1.6641e-04,
         4.7839e-07],
        [1.3126e-03, 2.5445e-03, 1.9531e-01,  ..., 4.7371e-05, 1.5499e-03,
         3.0798e-06],
        ...,
        [5.0182e-05, 4.6643e-05, 4.0677e-05,  ..., 7.8318e-02, 1.1632e-04,
         1.4347e-04],
        [6.5722e-07, 6.6751e-07, 6.7820e-07,  ..., 1.6181e-03, 3.3924e-02,
         3.2823e-07],
        [9.2074e-04, 9.3072e-04, 1.1808e-03,  ..., 3.1916e-06, 1.1460e-06,
         2.9282e-02]]), 'X': tensor([[  2.0937,  13.8293,  -2.3533],
        [  1.6308,  13.9254,  -2.0984],
        [  0.7275,  14.0667,  -0.9815],
        [  0.4119,  14.1019,  -4.1870],
        [  1.4150,  13.9647,  -4.0915],
        [  0.6142,  14.0802,   0.5407],
        [ 12.6038,   0.7665,  -3.4439],
        [ -1.6906,   7.8860,   7.7142],
        [  2.5813,  -4.5471,   5.3597],
        [  1.5246,  -4.7838,  -5.2095],
        [ -6.4368,   5.8356,   3.3014],
        [  5.8617,   3.4853,  -9.8080],
        [ -0.1963,   7.8765,  -1.1371],
        [ 12.4837,   2.7510,   0.4449],
        [ 11.6634,   5.8050,   9.1590],
        [ -9.4735,  -1.2620,  -3.6742],
        [  2.7420,  -4.4896,  -3.7139],
        [ -9.2687,  -2.9794,   0.6436],
        [ 11.7418,   5.6070,  -1.4192],
        [  1.8447,  -4.7372,   8.1235],
        [ -9.4524,  -1.6813,  -6.6862],
        [ -0.0282,   7.8575,  -1.1746],
        [  9.3607,   9.5337,  -4.5412],
        [  1.1350,  14.0104,   0.6321],
        [  5.3860,  -2.3418,   1.0826],
        [  2.0870,  -4.6877,   9.6823],
        [ -6.9838,  -7.4965,   0.9796],
        [ -5.3736,   6.6310,  -1.5174],
        [ 11.3800,  -4.3919,   1.4066],
        [  2.0419,  -4.6978,  -5.0983],
        [ 11.0891,  -4.9558,  -2.3178],
        [ -7.5401,   4.6718,   9.0033],
        [ -5.2589,   6.7024,  -5.5809],
        [ 11.7794,   5.5086,  -9.6929],
        [  2.1625,  -4.6697,   7.0658],
        [ 10.0995,  -6.4739,   2.0974],
        [  7.8425,  -8.7202,   6.4530],
        [ -8.5607,  -5.0938,  -5.4531],
        [ -5.9225,  -8.5355,   2.4778],
        [ 12.0582,  -2.6640,   0.7132],
        [ 10.9599,  -5.1849,  -3.5526],
        [ -9.2373,  -3.1225,  -8.9763],
        [  2.5663,   7.0526,  -6.8769],
        [  6.0140,  -1.0661,  -2.5040],
        [ -6.6998,   5.5954,   0.0823],
        [  5.7151,  -1.7700,   3.1542],
        [ -9.4748,  -0.7765,  -5.6695],
        [  4.0587, -10.6112,   6.2273],
        [  5.6202,  -1.9508,   0.7471],
        [  1.3670,  -4.7990,   3.2603],
        [ 12.6038,   0.7665,  -3.4439],
        [  9.6587,   9.1817,  -0.8212],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.3515,   0.6418,  -6.5823],
        [  5.8275,  -1.5334,   2.0124],
        [  2.1625,  -4.6697,   7.0658],
        [  5.1667,  -2.6543,   1.2879],
        [ 12.6000,   1.3839,  -3.2992],
        [ -6.6998,   5.5954,   0.0823],
        [  3.2366,  -4.2732,   5.1532],
        [  9.0652,   9.8597,  -3.5949],
        [  5.8275,  -1.5334,   2.0124],
        [ 10.9931,   7.2107,   7.7933],
        [ -0.2586,   7.8825,  -9.9840],
        [  6.3048,   0.1469,  -9.1691],
        [  2.2373,  -4.6506,  -1.7334],
        [ -9.2974,   0.8371,   2.4852],
        [  5.3602,  -2.3808,   7.2098],
        [  1.6308,  13.9254,  -2.0984],
        [ 10.9599,  -5.1849,  -3.5526],
        [  2.2659,  13.7894,   6.5436],
        [  5.0277,   4.9589,   7.0175],
        [  5.7485,  -9.9811,  -3.2355],
        [ -4.0633,  -9.7927,  -3.0825],
        [  4.9957,  -2.8705,   6.5978],
        [  6.3303,   1.6145,  -8.4083],
        [ -8.6381,  -4.9238,  -4.4077],
        [  5.1787,  -2.6383,  -4.2533],
        [ -8.6681,  -4.8556,   9.2518],
        [  3.3839,  -4.1966,   8.3574],
        [  4.1348,   5.9514,  -0.8856],
        [  2.5636, -10.9293,   9.6519],
        [ -7.4808,   4.7462,   2.0266],
        [  7.7794,  11.0629,   8.9069],
        [  5.5240,  -2.1198,   6.2055],
        [ -9.4735,  -0.7274,  -2.6749],
        [ -7.4887,   4.7364,   8.9074],
        [  6.2555,   2.1474,   4.1685],
        [  2.7976,  -4.4682,  -4.2020],
        [  3.9515,  -3.8425,   6.1729],
        [ -6.5295,  -7.9785,  -7.4429],
        [ -8.7078,   2.7182,   1.8146],
        [  6.8704,  -9.3809,   7.5769],
        [  2.3490,  -4.6199,  -4.2489],
        [  1.9505,  -4.7171,   9.9868],
        [ -4.3963,   7.1629,   4.5263],
        [ -8.5607,  -5.0938,  -5.4531],
        [  2.3261,   7.1726,   6.2522],
        [ -9.4753,  -0.8012,  -5.2050],
        [  2.2659,  13.7894,   6.5436],
        [  5.7151,  -1.7700,   3.1542],
        [  4.9452, -10.3187,  -7.8621],
        [ 12.5832,   1.7543,  -7.6134],
        [  0.9659,  -4.8144,  -0.9594],
        [ -9.4735,  -1.2620,  -3.6742],
        [ 11.5568,   6.0602,   9.3244]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0662, 0.1918,  ..., 0.3319, 1.1445, 0.9753],
        [0.0662, 0.0000, 0.1294,  ..., 0.2657, 1.0869, 0.9589],
        [0.1918, 0.1294, 0.0000,  ..., 0.1598, 1.0166, 0.8886],
        ...,
        [0.3319, 0.2657, 0.1598,  ..., 0.0000, 0.8619, 0.9483],
        [1.1445, 1.0869, 1.0166,  ..., 0.8619, 0.0000, 1.6559],
        [0.9753, 0.9589, 0.8886,  ..., 0.9483, 1.6559, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.98691426, 14.02058621, 14.08549802, 14.10796466, 14.03618649,
       14.09357547, 12.62711252,  8.06517143,  5.2287058 ,  5.0209005 ,
        8.68832398,  6.81961996,  7.8788965 , 12.78327037, 13.02818702,
        9.55720651,  5.26067019,  9.73579738, 13.01187484,  5.08372149,
        9.60080639,  7.8575648 , 13.36092624, 14.05633441,  5.87305005,
        5.13125513, 10.24557047,  8.53501777, 12.19805394,  5.1224079 ,
       12.1460931 ,  8.87007495,  8.51928531, 13.00380624,  5.14607014,
       11.99633819, 11.72802491,  9.96152788, 10.38897563, 12.3489324 ,
       12.12448912,  9.75075273,  7.50500064,  6.10774544,  8.72896381,
        5.98285319,  9.50654967, 11.36089355,  5.94909709,  4.98989906,
       12.62711252, 13.32645552,  5.18047638,  6.383897  ,  6.02588517,
        5.14607014,  5.80860165, 12.67576792,  8.72896381,  5.36061974,
       13.39372426,  6.02588517, 13.14690137,  7.88677538,  6.30647746,
        5.16077764,  9.3349858 ,  5.86519525, 14.02058621, 12.12448912,
       13.97430191,  7.06169391, 11.51811603, 10.60226614,  5.76165934,
        6.53290663,  9.94285511,  5.81199422,  9.9353921 ,  5.39098669,
        7.24677664, 11.22596994,  8.859416  , 13.52430176,  5.91677188,
        9.5014146 ,  8.86082124,  6.61386436,  5.27178363,  5.51177501,
       10.30972193,  9.12220271, 11.62770332,  5.18277858,  5.10446653,
        8.40445872,  9.96152788,  7.5403822 ,  9.50913849, 13.97430191,
        5.98285319, 11.4424816 , 12.70489563,  4.91038884,  9.55720651,
       13.0493368 ])}
{'P': tensor([[7.7647e-03, 6.7738e-01, 2.6001e-01,  ..., 2.7553e-06, 3.5512e-06,
         9.3693e-07],
        [7.4827e-04, 2.5667e-01, 5.8046e-02,  ..., 6.9670e-05, 1.1252e-04,
         2.7684e-05],
        [1.8780e-03, 2.1820e-01, 2.5891e-01,  ..., 6.0597e-05, 5.9353e-05,
         2.4352e-05],
        ...,
        [3.7598e-03, 3.5621e-03, 2.2104e-03,  ..., 6.7619e-02, 1.6986e-03,
         9.4055e-04],
        [2.1234e-04, 3.5562e-04, 1.5296e-04,  ..., 4.0611e-04, 2.2623e-01,
         1.1826e-04],
        [8.4475e-04, 6.6542e-04, 5.2042e-04,  ..., 1.4932e-01, 5.9454e-04,
         8.5208e-02]]), 'X': tensor([[ 2.0449e+00,  1.3840e+01, -7.1642e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [ 1.4150e+00,  1.3965e+01, -4.0915e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-7.6719e+00, -6.6260e+00,  7.1468e+00],
        [ 3.7237e+00, -3.9966e+00, -5.0376e+00],
        [-1.9646e+00,  7.8571e+00,  3.3817e+00],
        [ 1.3821e-01, -1.1007e+01,  1.0026e+00],
        [-9.4766e+00, -1.1116e+00, -2.1706e+00],
        [ 1.2600e+01,  1.3839e+00, -3.2992e+00],
        [ 6.2026e+00,  1.2177e+01, -1.5887e+00],
        [ 9.5018e+00, -7.1945e+00, -1.1892e+00],
        [-4.4779e+00, -9.5609e+00, -2.4505e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [ 6.3505e+00,  6.2348e-01, -8.5849e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [ 5.8323e+00, -1.5228e+00,  5.1132e+00],
        [ 1.2469e+01,  2.8525e+00, -5.5110e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 1.1027e+01, -5.0666e+00,  8.4154e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [ 2.2437e+00, -4.6489e+00,  6.1430e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [-2.6069e+00, -1.0430e+01, -6.1995e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [ 1.0057e+01,  8.6697e+00,  4.6329e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 2.7976e+00, -4.4682e+00, -4.2020e+00],
        [ 4.0591e+00,  6.0202e+00, -1.6339e+00],
        [ 8.6420e+00, -8.0580e+00,  6.5781e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00],
        [ 3.6249e+00, -1.0725e+01,  5.6605e+00],
        [ 4.5503e+00,  5.5359e+00,  7.4735e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 9.3836e+00,  9.5075e+00, -5.9278e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 4.2339e+00, -3.6268e+00,  6.3125e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 7.5064e+00, -8.9651e+00, -9.9366e-02],
        [-7.6293e+00, -6.6859e+00, -5.5161e+00],
        [ 5.6260e+00, -1.9401e+00,  3.0259e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [ 5.1787e+00, -2.6383e+00, -4.2533e+00],
        [-7.4733e+00,  4.7555e+00,  6.2490e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [-9.0961e+00,  1.6529e+00,  3.1553e+00],
        [ 5.7784e+00, -1.6401e+00,  9.5587e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [-7.4887e+00,  4.7364e+00,  8.9074e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [-9.2508e+00,  1.0572e+00,  2.9978e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 5.8201e+00,  3.5841e+00, -8.0609e+00],
        [-9.1624e+00, -3.4286e+00,  2.0647e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 7.4132e+00,  1.1353e+01,  3.3218e-01],
        [-9.2301e+00, -3.1537e+00, -2.7477e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [-5.6174e+00,  6.4708e+00, -1.7339e+00],
        [ 6.2247e+00, -3.1500e-01, -9.3657e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-9.4645e+00, -1.4880e+00, -2.0226e+00],
        [ 4.1999e+00, -3.6543e+00, -4.1998e+00],
        [ 5.6928e+00,  1.2471e+01,  3.6001e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [-4.3963e+00,  7.1629e+00,  4.5263e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 2.3200e+00,  7.1756e+00,  1.1938e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 2.3889e+00, -4.6082e+00, -8.7291e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 1.1337e+01, -4.4806e+00,  4.6434e-01],
        [ 1.2159e+01,  4.3364e+00, -8.0867e+00],
        [ 1.1423e+01,  6.3594e+00,  5.1663e+00],
        [ 6.0345e+00,  3.0175e+00, -1.4424e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 5.9548e+00, -1.2273e+00, -2.6520e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.6229e-01, 1.1489e-01,  ..., 4.6273e-01, 1.1356e+00,
         3.7800e-01],
        [1.6229e-01, 0.0000e+00, 1.0770e-01,  ..., 3.6173e-01, 1.0298e+00,
         2.7203e-01],
        [1.1489e-01, 1.0770e-01, 0.0000e+00,  ..., 4.6619e-01, 1.1366e+00,
         3.7664e-01],
        ...,
        [4.6273e-01, 3.6173e-01, 4.6619e-01,  ..., 0.0000e+00, 6.7392e-01,
         8.9711e-02],
        [1.1356e+00, 1.0298e+00, 1.1366e+00,  ..., 6.7392e-01, 2.1453e-04,
         7.6185e-01],
        [3.7800e-01, 2.7203e-01, 3.7664e-01,  ..., 8.9711e-02, 7.6185e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.99048094, 14.09176259, 14.07623266, 14.10796466, 14.03618649,
        8.22728399, 10.1371552 ,  5.46244988,  8.09899874, 11.00812996,
        9.54154096, 12.67576792, 13.6660671 , 11.9182956 , 10.5575601 ,
       12.1494023 , 13.52430176,  6.38105015,  4.83371341,  6.67108662,
        6.02779201, 12.79126785, 10.86693708,  7.15100303,  8.1137452 ,
       12.13568598,  7.83534987,  6.81084724,  5.16205087,  8.01009083,
        7.24326992, 13.88683436, 10.7506423 ,  4.91336994, 13.2778378 ,
       13.48809595, 11.05317597, 13.2778378 , 13.54774355,  5.27178363,
        7.26074925, 11.8159254 ,  5.52416368, 11.32149259,  7.16598273,
        8.51088913, 13.35832725,  7.27759233, 13.6829843 ,  5.5748624 ,
        4.74468406,  4.85603886, 13.2430377 , 11.69264243, 10.14436912,
        5.95111804,  5.64500512,  8.8580671 ,  5.81199422,  8.8580671 ,
       10.60226614,  7.39599986,  9.24501829,  6.00663314,  4.978704  ,
        8.86082124,  5.19986797,  9.3109884 ,  8.53501777,  9.99145096,
        6.83515201,  9.78285424, 10.4977132 ,  8.95983586,  7.58668144,
       10.27572673, 13.55868684,  9.75401449, 11.62770332,  7.07902036,
       10.06407839,  8.56889738,  6.23262289,  9.9353921 ,  9.58072286,
        5.5671389 , 13.70893492,  4.87083348,  8.40445872,  7.48538905,
       10.1979669 ,  7.02496223,  8.16584537,  6.97106912,  5.41348121,
        7.54127425,  6.33075682,  5.19065439, 10.88618114,  4.7663526 ,
       12.18998195, 12.9089559 , 13.0743399 ,  6.74686071,  7.56617037,
        6.07992623])}
{'P': tensor([[1.9035e-03, 4.1834e-01, 1.5484e-01,  ..., 2.3853e-07, 3.0109e-07,
         2.2884e-07],
        [4.9054e-04, 2.1798e-01, 2.9738e-03,  ..., 2.0440e-05, 3.0849e-05,
         2.5198e-05],
        [5.9512e-05, 4.8221e-01, 8.8307e-03,  ..., 9.8399e-07, 1.3016e-06,
         1.0395e-06],
        ...,
        [1.0188e-04, 1.2430e-04, 1.3685e-04,  ..., 2.0242e-02, 1.4179e-06,
         6.3046e-07],
        [6.2653e-03, 5.1505e-03, 5.5610e-03,  ..., 6.3712e-04, 3.0794e-01,
         7.1434e-03],
        [2.9182e-04, 2.4239e-04, 2.5911e-04,  ..., 5.6646e-05, 2.1849e-01,
         8.5361e-02]]), 'X': tensor([[ 1.9312e+00,  1.3865e+01,  8.3563e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [-7.7842e+00,  4.3466e+00, -1.7548e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 5.1292e+00,  4.8183e+00, -5.7997e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [ 1.4598e-01,  7.8340e+00, -9.4614e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [-6.7745e+00, -7.7264e+00,  8.4684e+00],
        [ 6.3360e+00,  4.2583e-01, -9.2509e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 1.1380e+01, -4.3919e+00,  1.4066e+00],
        [ 3.2701e+00,  6.6338e+00,  9.9209e+00],
        [ 5.4336e+00, -2.2676e+00,  5.6933e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [-4.0633e+00, -9.7927e+00, -3.0825e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-6.4620e+00, -8.0449e+00,  6.2719e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 1.2273e+00, -4.8081e+00,  9.8129e+00],
        [ 1.2515e+01,  2.5072e+00, -9.3274e+00],
        [ 2.0870e+00, -4.6877e+00,  9.6823e+00],
        [-8.0766e+00, -6.0046e+00, -3.1873e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [ 5.6314e+00, -1.9302e+00, -9.6543e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-3.9858e+00, -9.8334e+00, -3.7323e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 4.6063e+00,  1.3005e+01, -3.1456e+00],
        [ 5.6672e+00, -1.8631e+00,  8.1020e+00],
        [ 9.9453e-01, -4.8144e+00, -1.8966e-01],
        [-9.1960e+00, -3.2964e+00,  1.6921e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-8.1531e+00, -5.8748e+00,  2.6934e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-5.8389e+00, -8.6056e+00, -3.5098e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 4.1348e+00,  5.9514e+00, -8.8556e-01],
        [ 2.0039e+00,  7.3173e+00, -1.5885e+00],
        [-8.9636e+00,  2.0648e+00,  8.1946e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 2.7295e+00,  1.3670e+01, -8.9328e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 6.1989e+00, -9.7591e+00,  8.2889e-01],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [ 8.1528e+00,  1.0745e+01,  9.1493e+00],
        [ 3.9770e+00, -1.0634e+01,  8.0783e+00],
        [ 6.9314e+00, -9.3435e+00, -1.2729e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [ 1.0150e+01,  8.5414e+00, -7.2681e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-9.3107e+00, -2.7697e+00, -3.2919e+00],
        [ 2.2849e+00, -1.0965e+01, -7.6542e+00],
        [ 2.1824e+00, -4.6647e+00,  1.1985e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 6.2675e+00,  1.2138e+01, -8.9720e+00],
        [ 6.1372e+00,  2.6726e+00, -6.7885e+00],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [-8.7409e+00, -4.6837e+00, -6.5840e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 9.0129e+00, -7.7072e+00,  9.1400e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [ 5.1793e+00,  4.7461e+00,  3.8990e-01],
        [-7.3333e+00,  4.9246e+00, -7.9212e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 5.2188e+00, -2.5839e+00,  2.1833e+00],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 6.9956e+00,  1.1659e+01, -7.6876e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1468, 0.0611,  ..., 1.3396, 0.4114, 0.4220],
        [0.1468, 0.0000, 0.0876,  ..., 1.2859, 0.2911, 0.2953],
        [0.0611, 0.0876, 0.0000,  ..., 1.3051, 0.3536, 0.3630],
        ...,
        [1.3396, 1.2859, 1.3051,  ..., 0.0000, 1.0407, 1.0566],
        [0.4114, 0.2911, 0.3536,  ..., 1.0407, 0.0000, 0.0297],
        [0.4220, 0.2953, 0.3630,  ..., 1.0566, 0.0297, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([13.99877413, 14.05687815, 14.0393949 , 14.10045797,  8.29022197,
        8.28968067,  8.91551231,  8.85572292,  7.03733775,  6.54615887,
        6.12886024,  7.83534987, 10.31613269, 10.27572673,  6.35029274,
        9.43112957,  6.14086582, 12.19805394,  7.39599986,  5.88782913,
       13.88683436,  5.1224079 ,  7.8788965 ,  7.98469166, 10.60226614,
        8.8353674 ,  6.81084724, 10.31886066,  5.94541803,  4.96231532,
       12.76408562,  5.13125513, 10.06407839,  6.14086582,  4.78544875,
       12.30858677,  5.95297459,  5.91854397, 10.42667093, 10.61047596,
       11.04313641,  9.78147542, 11.11969815, 13.79676654,  5.96556248,
        4.9160962 ,  9.76896904, 14.05687815, 13.37121923, 11.23201619,
       10.04916678, 11.46176259, 10.91805683, 10.87409005, 10.39942497,
        8.36995953,  7.24677664,  7.58668144,  9.19837792,  7.37902766,
       13.94008948, 13.69162795, 11.86239961,  7.81509631, 11.56146873,
        9.09963894, 13.48809595, 11.35345345, 11.63383221,  4.78544875,
        6.27905292, 13.2430377 , 11.46176259, 11.83570132, 11.64746882,
        5.47866221, 13.26590071,  7.24326992,  9.71392179, 11.20100795,
        5.14999799,  4.74770899, 10.9036226 ,  7.60463349, 11.65923396,
        8.66273171, 13.66051884,  6.69389431, 10.37998078,  9.91668182,
        7.46248947, 11.85890664, 10.51437947, 11.21703654,  7.02496223,
        8.83341608,  9.9353921 ,  9.96391045, 11.97957425, 13.3614598 ,
        5.82345867,  8.8737556 , 10.31613269, 13.59673406,  6.21275016,
        5.64865173])}
{'P': tensor([[2.6028e-02, 5.9310e-01, 1.8756e-01,  ..., 2.7197e-06, 1.8160e-05,
         1.0928e-07],
        [2.9871e-03, 2.2928e-01, 1.0594e-01,  ..., 3.4490e-05, 1.0282e-04,
         8.5719e-07],
        [5.2475e-03, 4.1871e-02, 2.9082e-01,  ..., 2.8578e-05, 1.8055e-04,
         1.4494e-06],
        ...,
        [4.5497e-07, 4.6754e-07, 3.1055e-07,  ..., 5.2325e-02, 4.1106e-06,
         1.4585e-04],
        [1.5841e-02, 9.9487e-03, 2.8800e-02,  ..., 1.5859e-05, 3.0365e-01,
         6.8152e-05],
        [1.4773e-05, 9.3128e-06, 1.2610e-05,  ..., 1.2473e-06, 1.8710e-03,
         6.4114e-02]]), 'X': tensor([[  1.6308,  13.9254,  -2.0984],
        [  0.7275,  14.0667,  -0.9815],
        [  0.4119,  14.1019,  -4.1870],
        [  0.6142,  14.0802,   0.5407],
        [  1.1350,  14.0104,   0.6321],
        [  1.4306,  13.9620,  -0.2075],
        [  2.4639, -10.9431,  -7.3718],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.4592,  -0.9168,  -6.7571],
        [  4.7739,  -3.1218,  -4.1188],
        [ 12.1751,  -2.2592,  -5.7751],
        [ -4.6942,  -9.4299,  -9.1846],
        [ -1.5261,   7.8988,   9.3768],
        [ -9.3419,   0.5972,  -1.2841],
        [  6.9284,  11.7061,  -0.3732],
        [ 12.2435,   4.0076,   8.1257],
        [  1.0391,  -4.8141,  -0.0768],
        [  0.9036,   7.6840,   9.9237],
        [ -0.5739,   7.9056,   0.6382],
        [  6.3236,   0.3011,  -4.6668],
        [ -9.1795,  -3.3623,   8.5259],
        [ -9.4735,  -1.2620,  -3.6742],
        [ -8.4267,  -5.3697,   2.4172],
        [  5.2199,   4.6861,  -1.9304],
        [  4.2339,  -3.6268,   6.3125],
        [  5.8889,  -1.3916,   8.8993],
        [  2.4932,   7.0903,  -5.4214],
        [ -9.1623,  -3.4288,  -4.9961],
        [  0.3038,   7.8092,   0.4164],
        [  2.7558,  -4.4844,   0.0636],
        [  8.6420,  -8.0580,   6.5781],
        [ -8.0965,  -5.9712,   4.3466],
        [ -1.3940, -10.7772,   6.7835],
        [  6.2022,   2.4092,   3.0349],
        [ -3.5520,   7.4976,   9.1522],
        [  1.1350,  14.0104,   0.6321],
        [ 12.3609,   3.4769,  -8.5762],
        [  5.3804,  -2.3504,   7.2744],
        [  3.9913,  -3.8139,  -6.8311],
        [  8.3718,  -8.2953,   8.4666],
        [  9.4925,   9.3811,   8.9842],
        [  4.6063,  13.0051,  -3.1456],
        [  6.3246,   1.6696,  -0.1544],
        [ 12.3759,  -1.3943,   6.4659],
        [ -1.7290,   7.8825,  -3.3831],
        [  5.8979,   3.3958,  -3.8713],
        [ 10.5057,  -5.9099,  -5.1599],
        [ 10.6673,  -5.6649,  -4.1646],
        [  6.3236,   0.3011,  -4.6668],
        [ 11.1420,   6.9331,  -2.4779],
        [ 11.2939,   6.6325,  -5.8311],
        [ -0.4341, -10.9473,   7.4509],
        [  4.7813,  -3.1139,   8.2691],
        [ -2.3317,   7.8034,  -3.6274],
        [ -4.7847,  -9.3729,  -3.7741],
        [ -9.3739,   0.3982,  -4.2564],
        [  2.8890,  -4.4316,   1.1600],
        [ -0.5385,   7.9037,   6.3830],
        [ -0.8455, -10.8853,  -6.7400],
        [  6.3534,   1.2990,  -7.2942],
        [ 11.6544,   5.8273,  -3.1375],
        [  6.3205,   1.7068,  -3.8619],
        [ -9.4748,  -0.7765,  -5.6695],
        [  3.2990,  -4.2415,  -7.4857],
        [ 11.0438,   7.1180,  -5.6991],
        [ -9.1433,  -3.5003,  -3.0209],
        [ 11.5935,   5.9740,  -4.3952],
        [  1.5662,  13.9375,   4.5142],
        [ -1.8788, -10.6567,   3.7068],
        [ -7.6719,  -6.6260,   7.1468],
        [  5.6414, -10.0303,   7.8525],
        [ -8.5506,  -5.1154,   6.3423],
        [ -5.5525,   6.5146,  -4.4984],
        [  3.2159, -10.8166,  -1.0406],
        [  7.7794,  11.0629,   8.9069],
        [  4.7739,  -3.1218,  -4.1188],
        [  3.5939,   6.4029,  -1.8357],
        [  5.8323,  -1.5228,   5.1132],
        [ 12.4901,  -0.7052,   1.7315],
        [  6.3587,   1.1596,  -3.6629],
        [ -1.5261,   7.8988,   9.3768],
        [  4.0587, -10.6112,   6.2273],
        [  1.7232,  -4.7574,   3.1270],
        [  6.9284,  11.7061,  -0.3732],
        [  6.0031,  12.2958,  -6.5992],
        [  6.2365,  -0.2578,  -6.0212],
        [  0.3231,  -4.7691,  -1.6120],
        [ -8.6681,  -4.8556,   9.2518],
        [ -7.0825,   5.2073,  -5.0497],
        [ -3.5029,   7.5138,   8.1853],
        [  0.7145,  -4.8071,  -2.3766],
        [  6.3497,   1.3680,  -1.9076],
        [ -9.1960,  -3.2964,   1.6921],
        [ -5.8389,  -8.6056,  -3.5098],
        [  6.2790,  -0.0259,  -5.4116],
        [ -7.2515,   5.0195,   4.6434],
        [  3.6469,   6.3624,  -1.8414],
        [ -9.2707,   0.9661,   4.4863],
        [  3.2809,  13.5061,  -4.0199],
        [ 10.9394,   7.3069,  -8.9365],
        [  6.3516,   0.6425,   0.3425],
        [ 10.5057,  -5.9099,  -5.1599],
        [  6.1929,   2.4501,  -6.9352],
        [ -6.2200,  -8.2737,   5.5079],
        [  5.0506,   4.9277,  -9.5094],
        [  7.3724,  -9.0577,  -8.9843]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1294, 0.1687,  ..., 1.1893, 0.3322, 0.5081],
        [0.1294, 0.0000, 0.2606,  ..., 1.0663, 0.3808, 0.5377],
        [0.1687, 0.2606, 0.0000,  ..., 1.2417, 0.1818, 0.3626],
        ...,
        [1.1893, 1.0663, 1.2417,  ..., 0.0000, 1.2064, 1.2018],
        [0.3322, 0.3808, 0.1818,  ..., 1.2064, 0.0000, 0.1809],
        [0.5081, 0.5377, 0.3626,  ..., 1.2018, 0.1809, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.02058621, 14.08549802, 14.10796466, 14.09357547, 14.05633441,
       14.03505624, 11.21703654, 12.4013153 , 12.49291768,  5.70405525,
       12.3828957 , 10.53368644,  8.04483495,  9.36093562, 13.60275211,
       12.88270018,  4.92497514,  7.73692436,  7.92644746,  6.33075682,
        9.77588801,  9.55720651,  9.99211663,  7.01475376,  5.5748624 ,
        6.05113217,  7.5158489 ,  9.78286733,  7.81509631,  5.26342203,
       11.8159254 , 10.06023665, 10.86693708,  6.65368551,  8.29640613,
       14.05633441, 12.84056604,  5.87132718,  5.5205189 , 11.785566  ,
       13.34587027, 13.79676654,  6.54127431, 12.45418236,  8.06990483,
        6.80559669, 12.05395056, 12.07818871,  6.33075682, 13.12299145,
       13.09737053, 10.95593729,  5.70591986,  8.14433542, 10.52357479,
        9.38232121,  5.29009836,  7.92200342, 10.91805683,  6.48486171,
       13.03002338,  6.54693987,  9.50654967,  5.37343173, 13.13888841,
        9.79039371, 13.04217487, 14.02526505, 10.82106505, 10.1371552 ,
       11.50791859,  9.96391045,  8.55982105, 11.2845608 , 13.52430176,
        5.70405525,  7.34250948,  6.02779201, 12.50996828,  6.46356806,
        8.04483495, 11.36089355,  5.05989547, 13.60275211, 13.6829843 ,
        6.24187044,  4.78002665,  9.9353921 ,  8.7907923 ,  8.29022197,
        4.85995002,  6.49538841,  9.76896904, 10.39942497,  6.27905292,
        8.81930525,  7.33350642,  9.32094055, 13.89886469, 13.15525803,
        6.38399789, 12.05395056,  6.65991877, 10.35093485,  7.05627461,
       11.67875855])}
{'P': tensor([[4.4026e-03, 9.7133e-01, 3.8800e-03,  ..., 3.3252e-06, 2.2161e-04,
         2.9057e-05],
        [4.9573e-04, 2.6886e-01, 7.6224e-04,  ..., 2.6104e-04, 3.7361e-03,
         1.1707e-03],
        [6.1344e-03, 9.5295e-01, 7.1777e-03,  ..., 5.8348e-06, 3.8654e-04,
         5.2092e-05],
        ...,
        [2.0832e-07, 2.0838e-07, 2.1784e-07,  ..., 2.6640e-02, 3.3878e-07,
         7.4909e-06],
        [1.0085e-03, 2.5834e-03, 1.2462e-03,  ..., 4.0392e-06, 2.9374e-03,
         8.2897e-05],
        [1.3695e-03, 1.6326e-03, 1.5518e-03,  ..., 5.0622e-04, 4.9150e-03,
         6.5102e-02]]), 'X': tensor([[  1.5662,  13.9375,   4.5142],
        [  0.5175,  14.0910,   4.6540],
        [  1.5653,  13.9377,   4.1773],
        [  0.6142,  14.0802,   0.5407],
        [ -4.3963,   7.1629,   4.5263],
        [ -3.1671,   7.6158,   6.0765],
        [ -8.8400,   2.3991,   1.3457],
        [ -9.3087,   0.7792,  -7.5587],
        [ 12.2592,   3.9425,   2.0383],
        [  4.9957,  -2.8705,   6.5978],
        [  1.9241,   7.3503,   7.9004],
        [ -9.4203,  -2.0340,   2.5856],
        [ -9.1960,  -3.2964,   1.6921],
        [ -8.9606,  -4.0994,   1.7886],
        [  4.5800,   5.5034,  -3.9432],
        [  9.4516,  -7.2501,  -1.8868],
        [ -9.1459,   1.4780,  -4.2403],
        [ -2.0721,   7.8432,   6.6380],
        [  5.6414, -10.0303,   7.8525],
        [  3.9913,  -3.8139,  -6.8311],
        [  6.1066,  -0.7799,  -4.1746],
        [ 11.9019,  -3.1379,  -3.4534],
        [  9.0129,  -7.7072,   9.1400],
        [  6.3122,  -9.6994,  -6.3282],
        [ 10.4994,  -5.9193,  -7.5153],
        [  4.4290,  -3.4596,   1.4181],
        [ 11.2767,  -4.6006,   0.7868],
        [  4.1140,  -3.7220,  -0.1273],
        [ 10.9599,  -5.1849,  -3.5526],
        [  6.1989,  -9.7591,   0.8289],
        [  6.9693,  -9.3200,  -8.3822],
        [  3.7237,  -3.9966,  -5.0376],
        [  6.1737,   2.5305,   3.9049],
        [  5.8201,   3.5841,  -8.0609],
        [ 12.5224,  -0.4497,   4.8448],
        [  9.1409,  -7.5789,   1.5814],
        [ 11.9936,  -2.8677,   6.0614],
        [ -2.3317,   7.8034,  -3.6274],
        [ -9.0604,   1.7707,   1.1066],
        [ -6.4620,  -8.0449,   6.2719],
        [  1.7000,  -4.7609,   2.7935],
        [ -2.2508, -10.5479,  -2.7235],
        [  8.6271,  -8.0715,  -7.4813],
        [  3.4604,  -4.1545,   1.8091],
        [  6.3534,   1.2990,  -7.2942],
        [  1.4306,  13.9620,  -0.2075],
        [ -2.3317,   7.8034,  -3.6274],
        [  4.2041,  -3.6509,  -4.9488],
        [  6.3589,   0.8239,  -8.4230],
        [  6.1973,  -0.4372,   6.5720],
        [  1.1644, -11.0394,   7.8511],
        [ 12.5770,   1.8512,  -0.6578],
        [ -3.0004,   7.6607,  -7.2640],
        [ -5.1046,  -9.1610,   2.3418],
        [  0.9809,   7.6642,  -1.4217],
        [ -1.6086,   7.8928,   4.3385],
        [  6.8093,  -9.4178,   8.4244],
        [  2.2644,  -4.6434,   9.7508],
        [  1.5383, -11.0274,  -4.9666],
        [  0.9809,   7.6642,  -1.4217],
        [  3.9770, -10.6341,   8.0783],
        [  2.7976,  -4.4682,  -4.2020],
        [  6.3535,   0.6792,  -7.5101],
        [  1.4150,  13.9647,  -4.0915],
        [  5.3858,   4.4260,  -9.6769],
        [ -7.6293,  -6.6859,  -5.5161],
        [  8.2154,  -8.4262,   8.0171],
        [  8.1043,  -8.5164,  -7.9290],
        [  4.8432,   5.1968,   1.5006],
        [  7.0358,  -9.2784,  -9.8479],
        [  4.0362,  13.2397,  -7.0030],
        [ -9.1623,  -3.4288,  -4.9961],
        [  5.6022,  -1.9833,   8.3081],
        [  2.0419,  -4.6978,  -5.0983],
        [  4.5340,  -3.3629,   8.9652],
        [  3.3744,   6.5623,  -6.3636],
        [  1.1644, -11.0394,   7.8511],
        [  9.5018,  -7.1945,  -1.1892],
        [ -7.6719,  -6.6260,   7.1468],
        [ -4.0633,  -9.7927,  -3.0825],
        [ -4.5778,  -9.5013,  -5.0667],
        [ -8.5506,  -5.1154,   6.3423],
        [ -8.5149,  -5.1905,   1.5239],
        [ -7.3554,   4.8985,  -0.6665],
        [ -7.3446,   4.9113,  -5.9356],
        [  3.4920,  -4.1366,  -8.3955],
        [  0.9478,  -4.8144,   0.6371],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -2.8162,   7.7059,  -9.6885],
        [  0.3724,   7.7974,  -6.8722],
        [  5.1861,   4.7361,  -3.7312],
        [ -9.1907,  -3.3180,  -5.5943],
        [ -7.8849,  -6.3115,  -4.7479],
        [  2.6451,  -4.5250,  -0.5565],
        [ -8.0608,   3.9362,  -5.9567],
        [  1.5662,  13.9375,   4.5142],
        [ 11.2847,  -4.5848,  -4.7503],
        [  2.1477,   7.2550,   8.9688],
        [  6.3534,   1.2990,  -7.2942],
        [  1.6954,  -4.7616,   9.8198],
        [  3.0719, -10.8449,  -4.1673],
        [  5.1886,  -2.6250,   6.4677],
        [  5.9883,   3.1536,   6.2320],
        [ -9.1623,  -3.4288,  -4.9961],
        [  2.3373,  13.7721,  -0.2442],
        [  1.8719,   7.3713,  -3.7465]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[3.0339e-04, 9.2061e-02, 3.9631e-02,  ..., 1.5975e+00, 5.5514e-01,
         9.1911e-01],
        [9.2061e-02, 0.0000e+00, 7.6902e-02,  ..., 1.5204e+00, 5.4295e-01,
         8.7205e-01],
        [3.9631e-02, 7.6902e-02, 0.0000e+00,  ..., 1.5612e+00, 5.1865e-01,
         8.7948e-01],
        ...,
        [1.5975e+00, 1.5204e+00, 1.5612e+00,  ..., 0.0000e+00, 1.2711e+00,
         8.5189e-01],
        [5.5514e-01, 5.4295e-01, 5.1865e-01,  ..., 1.2711e+00, 2.1453e-04,
         4.4670e-01],
        [9.1911e-01, 8.7205e-01, 8.7948e-01,  ..., 8.5189e-01, 4.4670e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.02526505, 14.10045797, 14.02532758, 14.09357547,  8.40445872,
        8.24808832,  9.15976746,  9.34127013, 12.87752052,  5.76165934,
        7.59795445,  9.63742669,  9.76896904,  9.85384623,  7.15990676,
       11.9120321 ,  9.26455795,  8.11227668, 11.50791859,  5.5205189 ,
        6.1561646 , 12.30858677, 11.85890664, 11.57249984, 12.05300943,
        5.62005629, 12.17901033,  5.54776934, 12.12448912, 11.56146873,
       11.63765273,  5.46244988,  6.67218687,  6.83515201, 12.53047489,
       11.87412439, 12.33167382,  8.14433542,  9.23178204, 10.31886066,
        5.05534609, 10.78533838, 11.81423161,  5.40689332,  6.48486171,
       14.03505624,  8.14433542,  5.56809335,  6.41202986,  6.21275016,
       11.10066126, 12.7125079 ,  8.22728399, 10.48721382,  7.72668467,
        8.0550286 , 11.62158554,  5.16611782, 11.1341798 ,  7.72668467,
       11.35345345,  5.27178363,  6.38968357, 14.03618649,  6.97106912,
       10.14436912, 11.76830981, 11.75618513,  7.10378919, 11.64436684,
       13.84126197,  9.78286733,  5.94293185,  5.1224079 ,  5.64500512,
        7.37902766, 11.10066126, 11.9182956 , 10.1371552 , 10.60226614,
       10.54658597,  9.96391045,  9.97220799,  8.83725081,  8.8353674 ,
        5.41348121,  4.90676242, 10.33712258,  8.20436764,  7.80625981,
        7.02326515,  9.7712321 , 10.09979747,  5.24137909,  8.97052115,
       14.02526505, 12.18045978,  7.56617037,  6.48486171,  5.05444288,
       11.27160035,  5.81481044,  6.76791483,  9.78286733, 13.96905756,
        7.60529892])}
{'P': tensor([[7.0894e-03, 9.4122e-01, 6.5776e-03,  ..., 2.4367e-03, 3.9496e-04,
         1.0593e-08],
        [4.3371e-04, 1.5298e-01, 2.3560e-02,  ..., 5.6766e-02, 1.1297e-02,
         2.0621e-07],
        [8.2696e-03, 3.7022e-02, 1.8174e-01,  ..., 9.7082e-02, 1.3412e-02,
         4.8029e-07],
        ...,
        [7.0213e-06, 9.2368e-06, 9.3524e-06,  ..., 2.1152e-02, 9.3139e-02,
         1.1907e-06],
        [4.1740e-06, 4.8100e-06, 4.7763e-06,  ..., 3.2923e-04, 7.7594e-02,
         9.5104e-06],
        [8.7075e-07, 5.6870e-07, 1.4754e-06,  ..., 7.8244e-07, 2.5355e-07,
         1.3387e-02]]), 'X': tensor([[  1.5653,  13.9377,   4.1773],
        [  0.5175,  14.0910,   4.6540],
        [  0.6142,  14.0802,   0.5407],
        [  1.5662,  13.9375,   4.5142],
        [ -4.3963,   7.1629,   4.5263],
        [ -4.2040,   7.2486,   2.9875],
        [ 12.5199,  -0.4711,   3.0436],
        [ 12.2311,  -2.0445,   1.3545],
        [  1.2273,  -4.8081,   9.8129],
        [  0.1382, -11.0073,   1.0026],
        [ -8.4573,   3.2448,   5.4370],
        [  1.2728, -11.0372,   9.8121],
        [  0.6952,  -4.8060,   8.8976],
        [  2.0729,   7.2878,   7.0880],
        [ 11.4743,   6.2478,  -2.1100],
        [  8.2154,  -8.4262,   8.0171],
        [ -0.5709,   7.9055,   7.6089],
        [  2.2659,  13.7894,   6.5436],
        [ -8.3900,   3.3727,   3.1650],
        [  4.9957,  -2.8705,   6.5978],
        [ -4.4503,   7.1378,   3.3751],
        [ 10.9717,  -5.1646,  -6.9039],
        [  3.9591, -10.6391,  -6.2557],
        [  2.7420,  -4.4896,  -3.7139],
        [ -9.4117,   0.1160,   8.2574],
        [ -3.9700,   7.3453,  -4.1486],
        [  2.0449,  13.8402,  -7.1642],
        [ -8.6603,   2.8250,   5.9181],
        [ -1.3940, -10.7772,   6.7835],
        [  8.3718,  -8.2953,   8.4666],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.7239,   4.4298,   3.1216],
        [  6.3535,   0.6792,  -7.5101],
        [ 10.0619,   8.6627,  -9.2884],
        [  6.2555,   2.1474,   4.1685],
        [ 11.5456,   6.0861,   8.1101],
        [ -8.0588,  -6.0341,  -2.3006],
        [  0.6952,  -4.8060,   8.8976],
        [ -0.0410,   7.8591,   9.3457],
        [  9.2692,   9.6370,   7.0746],
        [  1.1994,   7.6035,   2.4295],
        [  5.4167, -10.1293,   6.2156],
        [ 11.3800,  -4.3919,   1.4066],
        [ -7.2515,   5.0195,   4.6434],
        [  1.1274,  14.0116,   8.1753],
        [ -6.3252,  -8.1760,  -1.0891],
        [ -3.4100,   7.5436,  -7.6980],
        [ -9.2687,  -2.9794,   0.6436],
        [  1.7228,  -4.7575,   8.3396],
        [ -4.4779,  -9.5609,  -2.4505],
        [ -8.1204,  -5.9308,   4.0761],
        [  2.9260,  -4.4162,   8.3808],
        [  1.0000,   7.6591,  -4.5011],
        [ 10.3251,   8.2928,  -3.0006],
        [  9.3624,   9.5318,   0.2779],
        [  2.0937,  13.8293,  -2.3533],
        [ -7.0948,   5.1940,   9.8171],
        [ -2.6735, -10.4062,  -1.5911],
        [ 12.4837,   2.7510,   0.4449],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -1.6906,   7.8860,   7.7142],
        [  3.1480,  -4.3165,  -7.8002],
        [  1.5246,  -4.7838,  -5.2095],
        [  5.2608,  12.6977,   4.9166],
        [ -3.5520,   7.4976,   9.1522],
        [  5.4429,  -2.2529,   9.7306],
        [ 11.9122,   5.1397,  -1.5498],
        [ -7.0825,   5.2073,  -5.0497],
        [ -7.9298,   4.1367,  -3.9773],
        [  9.0004,  -7.7195,   4.6945],
        [ -2.2188,   7.8218,   8.6064],
        [ -1.6086,   7.8928,   4.3385],
        [  2.6321,  -4.5296,  -2.5159],
        [ 12.6035,   0.7494,   7.1536],
        [  0.6757, -11.0361,  -1.3047],
        [ -0.6323,   7.9085,  -5.9544],
        [  3.3094,  13.4969,  -3.6507],
        [ -8.4267,  -5.3697,   2.4172],
        [ -8.6229,   2.9068,   2.4429],
        [  6.3211,   1.7017,  -0.2164],
        [ 12.4877,  -0.7222,   8.8347],
        [ 10.9875,   7.2207,  -8.8549],
        [  3.3839,  -4.1966,   8.3574],
        [  9.2692,   9.6370,   7.0746],
        [ -7.4808,   4.7462,   2.0266],
        [  4.1999,  -3.6543,  -4.1998],
        [ -9.4716,  -0.6673,   7.6068],
        [ 11.7418,   5.6070,  -1.4192],
        [  9.1620,  -7.5573,  -9.9882],
        [  2.0419,  -4.6978,  -5.0983],
        [  6.2576,   2.1359,  -3.3398],
        [ -7.4807,   4.7463,   2.0616],
        [  6.3606,   0.9204,   3.2525],
        [ -9.4671,  -0.5553,   0.1117],
        [  5.7963,  -1.6018,   1.8856],
        [  7.0664,  -9.2590,   8.0879],
        [  2.3889,  -4.6082,  -8.7291],
        [ -0.7599, -10.8995,  -6.0366],
        [  6.2535,  -0.1709,  -9.4102],
        [ -2.5683, -10.4433,  -2.9182],
        [  5.8836,   3.4315,  -4.0327],
        [ 12.2328,  -2.0376,  -1.1234],
        [ 12.1588,   4.3364,  -8.0867],
        [ -5.6651,   6.4380,   1.3660],
        [ -8.6229,   2.9068,   2.4429],
        [  7.8375,  -8.7240,  -9.8913]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0769, 0.4989,  ..., 0.8823, 1.0498, 1.1574],
        [0.0769, 0.0000, 0.4994,  ..., 0.8335, 0.9915, 1.1625],
        [0.4989, 0.4994, 0.0000,  ..., 0.5722, 0.7946, 0.6632],
        ...,
        [0.8823, 0.8335, 0.5722,  ..., 0.0000, 0.2286, 0.7947],
        [1.0498, 0.9915, 0.7946,  ..., 0.2286, 0.0000, 0.9822],
        [1.1574, 1.1625, 0.6632,  ..., 0.7947, 0.9822, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.02532758, 14.10045797, 14.09357547, 14.02526505,  8.40445872,
        8.37954853, 12.52875793, 12.40074901,  4.96231532, 11.00812996,
        9.05843201, 11.11038551,  4.85603886,  7.57685908, 13.06498554,
       11.76830981,  7.92607342, 13.97430191,  9.04255655,  5.76165934,
        8.41148722, 12.12641969, 11.35182589,  5.26067019,  9.41244865,
        8.34949383, 13.99048094,  9.10945678, 10.86693708, 11.785566  ,
       12.85032836,  8.90405823,  6.38968357, 13.27717982,  6.61386436,
       13.05148826, 10.0674887 ,  4.85603886,  7.85919687, 13.37121923,
        7.69752262, 11.48663767, 12.19805394,  8.81930525, 14.05687815,
       10.33712258,  8.27853076,  9.73579738,  5.05981914, 10.5575601 ,
       10.05558893,  5.29755487,  7.72415048, 13.2430377 , 13.3607379 ,
       13.98691426,  8.79284466, 10.74409834, 12.78327037, 12.30858677,
        8.06517143,  5.34250225,  5.0209005 , 13.744387  ,  8.29640613,
        5.89074104, 12.97370371,  8.7907923 ,  8.94395628, 11.85743425,
        8.1303945 ,  8.0550286 ,  5.23878371, 12.62576026, 11.05672136,
        7.9337647 , 13.89671314,  9.99211663,  9.09963894,  6.54615887,
       12.5085995 , 13.14777433,  5.39098669, 13.37121923,  8.859416  ,
        5.5671389 ,  9.49511524, 13.01187484, 11.87665501,  5.1224079 ,
        6.61211224,  8.85939626,  6.42689705,  9.48337063,  6.01356004,
       11.64746882,  5.19065439, 10.92597196,  6.25586297, 10.75443106,
        6.81117802, 12.4013153 , 12.9089559 ,  8.57560275,  9.09963894,
       11.72749923])}
{'P': tensor([[6.7131e-03, 3.3208e-01, 2.3319e-01,  ..., 6.3030e-04, 1.4847e-05,
         2.8259e-06],
        [1.3842e-03, 1.7478e-01, 5.9112e-02,  ..., 2.4073e-02, 4.6557e-04,
         5.7102e-05],
        [1.8194e-03, 1.2525e-01, 1.5302e-01,  ..., 1.2139e-02, 4.3975e-04,
         5.6233e-05],
        ...,
        [9.5840e-06, 1.2531e-05, 1.0550e-05,  ..., 5.3197e-02, 1.8859e-03,
         1.3242e-05],
        [8.4349e-07, 8.2476e-07, 8.7251e-07,  ..., 1.8816e-05, 5.9179e-02,
         4.2198e-05],
        [1.4418e-02, 1.0597e-02, 1.4411e-02,  ..., 7.0473e-04, 1.9007e-04,
         1.5259e-01]]), 'X': tensor([[ 1.4306e+00,  1.3962e+01, -2.0748e-01],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-9.3604e+00,  4.8546e-01, -3.0158e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 6.0073e+00,  3.0986e+00,  6.7937e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-2.8162e+00,  7.7059e+00, -9.6885e+00],
        [ 1.1912e+01,  5.1397e+00, -1.5498e+00],
        [ 6.0054e+00, -1.0902e+00,  8.3079e+00],
        [ 1.1716e+01,  5.6728e+00, -8.9372e+00],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 6.1755e+00,  2.5233e+00, -2.0042e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-2.0840e+00,  7.8415e+00, -7.8657e+00],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 1.2868e-01, -1.1007e+01,  6.5757e+00],
        [ 1.2592e+01,  4.0347e-01, -5.8460e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 2.4149e+00, -4.6005e+00, -4.3339e+00],
        [ 5.6095e+00, -1.9702e+00, -7.8233e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 1.2058e+01, -2.6640e+00,  7.1316e-01],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 3.2159e+00, -1.0817e+01, -1.0406e+00],
        [ 1.6765e-01, -4.7447e+00, -6.1529e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-1.0482e+00, -1.0849e+01, -5.9066e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [-5.2752e+00, -9.0409e+00, -8.2568e+00],
        [-1.5875e+00,  7.8944e+00, -9.4286e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [-1.0407e+00,  7.9166e+00, -6.6021e+00],
        [ 9.3624e+00,  9.5318e+00,  2.7792e-01],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [-3.3999e+00, -1.0115e+01, -1.2088e+00],
        [-9.3419e+00,  5.9722e-01, -1.2841e+00],
        [-5.7389e-01,  7.9056e+00,  6.3817e-01],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [-9.3705e+00,  4.2084e-01, -3.5815e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 5.7963e+00, -1.6018e+00,  1.8856e+00],
        [ 4.3554e+00,  5.7393e+00, -9.0766e+00],
        [ 6.1752e+00,  2.5242e+00,  3.5874e+00],
        [-7.8359e+00,  4.2735e+00,  7.2943e+00],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [ 2.2373e+00, -4.6506e+00, -1.7334e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [ 1.0628e+01, -5.7253e+00,  7.4878e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [-9.2345e+00, -3.1345e+00,  3.9715e+00],
        [ 5.2939e+00, -2.4784e+00, -4.5840e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [ 9.0360e-01,  7.6840e+00,  9.9237e+00],
        [ 1.2512e+01, -5.3390e-01, -1.0728e+00],
        [ 6.1730e+00, -5.3750e-01, -8.8861e+00],
        [ 5.1472e+00, -2.6801e+00,  8.1194e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-1.3940e+00, -1.0777e+01,  6.7835e+00],
        [-8.4887e+00, -5.2448e+00, -6.5852e-01],
        [-7.5603e+00,  4.6459e+00,  8.2002e+00],
        [-9.4624e+00, -4.6325e-01, -7.3918e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 6.3211e+00,  1.7017e+00, -2.1637e-01],
        [-1.9487e+00,  7.8590e+00, -3.3330e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-5.0132e+00, -9.2234e+00, -8.3531e+00],
        [-5.6423e+00,  6.4537e+00, -1.2091e+00],
        [ 6.2895e+00,  4.0471e-02,  5.7652e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [ 6.3542e+00,  6.9335e-01, -3.3840e+00],
        [ 1.2606e+01,  9.5968e-01, -6.3264e+00],
        [-4.0987e-02,  7.8591e+00,  9.3457e+00],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 6.3551e+00,  7.1488e-01,  1.3344e+00],
        [ 1.0568e+01,  7.9261e+00, -6.1004e-01],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 8.8474e+00,  1.0087e+01, -8.8644e+00],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 4.9957e+00, -2.8705e+00,  6.5978e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 5.8979e+00,  3.3958e+00, -3.8713e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1167, 0.1426,  ..., 0.8306, 1.1687, 0.3087],
        [0.1167, 0.0000, 0.1540,  ..., 0.7140, 1.0891, 0.3308],
        [0.1426, 0.1540, 0.0000,  ..., 0.8045, 1.0440, 0.1790],
        ...,
        [0.8306, 0.7140, 0.8045,  ..., 0.0000, 0.8603, 0.9007],
        [1.1687, 1.0891, 1.0440,  ..., 0.8603, 0.0000, 0.9779],
        [0.3087, 0.3308, 0.1790,  ..., 0.9007, 0.9779, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.03505624, 14.09357547, 14.08549802, 14.05633441, 14.10796466,
        8.31625171, 12.72524615,  9.37296102, 11.46176259,  6.7593917 ,
        5.94541803, 13.96987806, 10.69963954,  8.20436764, 12.97370371,
        6.10360073, 13.01728313, 14.09357547,  6.67108662,  9.55593374,
        8.1137452 , 13.76047999,  6.83083863, 10.54658597, 11.00726483,
       12.59840193,  7.81509631, 12.1460931 ,  5.19577146,  5.94541803,
        5.76165934, 12.3489324 ,  7.53884104,  8.06517143, 11.2845608 ,
        4.74770899, 12.46513646, 10.8992524 ,  7.66377554, 10.46739073,
        8.05242591, 13.04217487,  7.98469166, 13.3607379 ,  6.4020404 ,
        7.16058587, 10.67132524,  9.36093562,  7.92644746,  8.4486771 ,
       14.05633441,  9.37989631, 13.32613445,  6.01356004,  7.20482224,
        6.67121748,  8.92549752, 10.38897563, 10.37698435,  5.16077764,
       11.11969815, 12.07225469,  5.60064319, 13.744387  , 10.33712258,
        9.75200848,  5.84533345, 12.7125079 , 13.40850328,  7.73692436,
       12.52372629,  6.19633139,  5.80312949, 13.13888841, 10.86693708,
        9.97822473,  8.8737556 ,  9.47369628,  6.18668463, 10.35485331,
        6.54615887,  8.09704052,  8.12043983, 10.68416401, 12.38881635,
       10.4977132 ,  8.5724042 ,  6.28961988, 13.744387  ,  5.04028538,
        6.39187285, 12.64235359,  7.85919687,  5.90035324, 10.75443106,
        6.39520397, 13.20988996, 13.98691426, 13.41713661, 10.68416401,
        5.40689332,  5.76165934, 14.10045797,  8.67502348,  9.78147542,
        6.80559669])}
{'P': tensor([[4.1395e-03, 9.6090e-01, 1.0129e-02,  ..., 1.8905e-08, 1.6835e-03,
         4.8551e-08],
        [2.9996e-04, 1.6382e-01, 3.5054e-02,  ..., 6.5414e-07, 7.6076e-02,
         1.5235e-06],
        [8.5348e-03, 1.1686e-01, 1.5966e-01,  ..., 5.3416e-07, 3.9956e-02,
         1.7079e-06],
        ...,
        [1.0210e-04, 6.8051e-05, 4.9044e-05,  ..., 2.8235e-02, 1.8908e-05,
         1.2935e-03],
        [3.5561e-05, 4.7453e-05, 4.0751e-05,  ..., 5.0228e-06, 8.5315e-02,
         2.5275e-06],
        [9.8002e-04, 6.6234e-04, 1.0301e-03,  ..., 5.9376e-06, 2.7167e-05,
         1.5689e-02]]), 'X': tensor([[  1.4150,  13.9647,  -4.0915],
        [  0.4119,  14.1019,  -4.1870],
        [  0.6396,  14.0772,  -6.3786],
        [  0.7275,  14.0667,  -0.9815],
        [ -3.9700,   7.3453,  -4.1486],
        [ -4.0409,   7.3169,  -3.5779],
        [ -5.5525,   6.5146,  -4.4984],
        [ -3.0008, -10.2827,  -8.9480],
        [  0.6142,  14.0802,   0.5407],
        [  2.3373,  -4.6232,  -5.2246],
        [  6.3589,   0.8239,  -8.4230],
        [  1.1644, -11.0394,   7.8511],
        [ -9.3331,  -2.6465,  -8.2028],
        [  4.0457,  -3.7737,  -0.4704],
        [  5.1886,  -2.6250,   6.4677],
        [ 11.7794,   5.5086,  -9.6929],
        [ -2.3317,   7.8034,  -3.6274],
        [ -6.4822,  -8.0252,  -3.8983],
        [  8.1043,  -8.5164,  -7.9290],
        [  6.3305,   1.6119,   1.1538],
        [ 12.4690,  -0.8528,  -9.3374],
        [ -9.2974,   0.8371,   2.4852],
        [  4.4633,   5.6287,  -0.7754],
        [ -4.8751,  -9.3148,   4.0406],
        [  3.9591, -10.6391,  -6.2557],
        [ -6.4002,  -8.1048,  -0.9108],
        [  0.3724,   7.7974,  -6.8722],
        [ -9.1813,  -3.3550,   1.8827],
        [  3.3069,   6.6089,   8.7377],
        [ -9.0961,   1.6529,   3.1553],
        [  5.1793,   4.7461,   0.3899],
        [  8.5093,  -8.1764,  -3.2381],
        [  8.0381,  -8.5690,  -8.9523],
        [ 11.9641,  -2.9568,   7.7211],
        [ -6.3306,   5.9272,   5.6935],
        [ -8.8400,   2.3991,   1.3457],
        [ -2.0840,   7.8415,  -7.8657],
        [  5.0506,   4.9277,  -9.5094],
        [  2.6310, -10.9195,   3.8054],
        [ -7.8280,   4.2848,  -7.1955],
        [  8.0877,  -8.5297,  -1.1050],
        [  9.6662,   9.1725,   8.1056],
        [ -7.3446,   4.9113,  -5.9356],
        [  0.5600, -11.0321,  -0.8660],
        [ -8.6050,   2.9451,  -0.7012],
        [  5.5158,  -2.1337,   3.5246],
        [  4.5800,   5.5034,  -3.9432],
        [  1.5662,  13.9375,   4.5142],
        [ -8.1531,  -5.8748,   2.6934],
        [  1.5662,  13.9375,   4.5142],
        [ -7.5401,   4.6718,   9.0033],
        [  9.2811,  -7.4336,   5.1771],
        [  5.0348,  -2.8229,   3.5436],
        [  3.3633,  13.4794,   0.5193],
        [ -8.1531,  -5.8748,   2.6934],
        [ -9.1706,   1.3863,   8.4032],
        [  6.3301,   0.3638,  -7.5685],
        [  8.6420,  -8.0580,   6.5781],
        [  4.0457,  -3.7737,  -0.4704],
        [  2.3373,  -4.6232,  -5.2246],
        [  9.6816,   9.1536,  -5.8543],
        [  3.2159, -10.8166,  -1.0406],
        [  1.1350,  14.0104,   0.6321],
        [  2.2849, -10.9655,  -7.6542],
        [ -1.0482, -10.8487,  -5.9066],
        [ -4.7642,  -9.3860,  -5.1133],
        [ -4.2040,   7.2486,   2.9875],
        [  6.2770,  -0.0379,   2.4195],
        [  2.8602,   6.8905,   5.8779],
        [  5.0233,   4.9647,  -3.1788],
        [  3.3943,   6.5483,   1.0790],
        [  5.6414, -10.0303,   7.8525],
        [  3.3069,   6.6089,   8.7377],
        [ -6.1899,  -8.3011,  -7.9269],
        [ -3.2053, -10.1993,  -3.8736],
        [ -9.2974,   0.8371,   2.4852],
        [  6.9314,  -9.3435,  -1.2729],
        [ -8.1531,  -5.8748,   2.6934],
        [  5.3602,  -2.3808,   7.2098],
        [  2.6614,  -4.5192,  -9.3641],
        [  2.7420,  -4.4896,  -3.7139],
        [  8.2422,  -8.4041,  -1.4962],
        [ -8.4824,  -5.2575,  -0.0795],
        [  4.0457,  -3.7737,  -0.4704],
        [  2.8890,  -4.4316,   1.1600],
        [  6.2242,  -0.3171,   7.4978],
        [ 10.6283,  -5.7253,   7.4878],
        [ -2.5683, -10.4433,  -2.9182],
        [  1.4749,  -4.7892,  -0.8089],
        [ -9.1813,  -3.3550,   1.8827],
        [ 12.5154,   2.5072,  -9.3274],
        [ -3.3812, -10.1235,   7.4384],
        [ -5.0132,  -9.2234,  -8.3531],
        [ -1.8367,   7.8718,  -8.6523],
        [  2.3490,  -4.6199,  -4.2489],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.1203,  -2.7151,   6.2110],
        [  0.2732,  -4.7619,  -1.1577],
        [ 12.5789,   0.1722,   1.6470],
        [  6.0031,  12.2958,  -6.5992],
        [ -7.8280,   4.2848,  -7.1955],
        [ -6.7643,  -7.7372,  -9.6663],
        [  8.5716,  10.3596,   6.8818],
        [ 11.7349,  -3.5838,   2.1644],
        [ -5.3736,   6.6310,  -1.5174],
        [ 12.5154,   2.5072,  -9.3274]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1325, 0.2160,  ..., 0.5991, 0.7335, 0.2376],
        [0.1325, 0.0000, 0.1382,  ..., 0.5275, 0.6017, 0.3126],
        [0.2160, 0.1382, 0.0000,  ..., 0.6361, 0.5959, 0.2739],
        ...,
        [0.5991, 0.5275, 0.6361,  ..., 0.0000, 0.5376, 0.8300],
        [0.7335, 0.6017, 0.5959,  ..., 0.5376, 0.0000, 0.8697],
        [0.2376, 0.3126, 0.2739,  ..., 0.8300, 0.8697, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.03618649, 14.10796466, 14.09176259, 14.08549802,  8.34949383,
        8.35856487,  8.55982105, 10.71162681, 14.09357547,  5.18047638,
        6.41202986, 11.10066126,  9.70108409,  5.53255882,  5.81481044,
       13.00380624,  8.14433542, 10.31613269, 11.75618513,  6.53250937,
       12.4980845 ,  9.3349858 ,  7.18354885, 10.51340829, 11.35182589,
       10.32715715,  7.80625981,  9.7751189 ,  7.39004226,  9.24501829,
        7.02496223, 11.80092578, 11.74901778, 12.32408516,  8.67227386,
        9.15976746,  8.1137452 ,  7.05627461, 11.23201619,  8.92395381,
       11.75438006, 13.32557133,  8.8353674 , 11.04629471,  9.09502385,
        5.91408504,  7.15990676, 14.02526505, 10.04916678, 14.02526505,
        8.87007495, 11.89105259,  5.77219077, 13.89264822, 10.04916678,
        9.27474386,  6.34059021, 11.8159254 ,  5.53255882,  5.18047638,
       13.32374056, 11.2845608 , 14.05633441, 11.20100795, 10.8992524 ,
       10.52587243,  8.37954853,  6.27715105,  7.46053281,  7.06272256,
        7.37576154, 11.50791859,  7.39004226, 10.35485331, 10.6910768 ,
        9.3349858 , 11.63383221, 10.04916678,  5.86519525,  5.24461068,
        5.26067019, 11.77124373,  9.97964013,  5.53255882,  5.29009836,
        6.2322769 , 12.07225469, 10.75443106,  5.01114719,  9.7751189 ,
       12.76408562, 10.67322527, 10.4977132 ,  8.08320235,  5.18277858,
        8.30125201,  5.79562691,  4.76969364, 12.58006038, 13.6829843 ,
        8.92395381, 10.27716434, 13.44593519, 12.26997165,  8.53501777,
       12.76408562])}
{'P': tensor([[1.6521e-02, 9.0215e-01, 1.7865e-02,  ..., 3.7044e-06, 3.4823e-08,
         3.3866e-05],
        [2.5010e-03, 1.8333e-01, 1.1259e-01,  ..., 3.5636e-05, 3.6738e-07,
         3.7988e-04],
        [1.3741e-02, 2.0609e-02, 2.5615e-01,  ..., 1.4802e-04, 7.9484e-07,
         1.2633e-03],
        ...,
        [1.1365e-04, 1.0537e-04, 4.9445e-04,  ..., 1.8206e-01, 1.6924e-05,
         6.2628e-04],
        [2.2759e-06, 1.9845e-06, 3.3329e-06,  ..., 6.0815e-05, 1.0555e-02,
         4.0632e-07],
        [5.6855e-07, 5.5811e-07, 8.2193e-07,  ..., 7.7746e-06, 7.4120e-05,
         5.1171e-02]]), 'X': tensor([[ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 1.1274e+00,  1.4012e+01,  8.1753e+00],
        [ 5.1750e-01,  1.4091e+01,  4.6540e+00],
        [-3.5029e+00,  7.5138e+00,  8.1853e+00],
        [-3.6381e+00,  7.4683e+00,  7.3544e+00],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [ 5.9587e+00,  1.2322e+01,  9.1695e+00],
        [-8.0611e+00, -6.0302e+00,  4.6962e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 5.7845e+00,  3.6652e+00,  9.1755e-01],
        [ 2.8890e+00, -4.4316e+00,  1.1600e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 1.2490e+01, -7.0522e-01,  1.7315e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-8.8934e+00,  2.2597e+00,  3.5831e+00],
        [ 5.0190e+00,  4.9705e+00, -7.9752e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 4.4290e+00, -3.4596e+00,  1.4181e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 5.2504e-01, -1.1031e+01,  5.5358e+00],
        [ 1.1089e+01, -4.9558e+00, -2.3178e+00],
        [ 1.1780e+01, -3.4696e+00,  9.0088e+00],
        [-2.2508e+00, -1.0548e+01, -2.7235e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [ 4.1537e+00,  5.9340e+00, -6.0423e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [-7.3470e+00, -7.0615e+00,  5.0066e+00],
        [ 4.6992e+00,  5.3690e+00,  3.0935e+00],
        [-3.5173e+00, -1.0062e+01,  4.8280e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 5.4179e+00, -2.2925e+00, -3.4641e+00],
        [ 2.5708e-01, -4.7594e+00,  1.5618e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 7.9403e+00, -8.6455e+00, -8.7433e+00],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 6.7792e+00,  1.1808e+01,  1.6251e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-8.0588e+00, -6.0341e+00, -2.3006e+00],
        [ 6.3606e+00,  9.2044e-01,  3.2525e+00],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [ 7.3724e+00, -9.0577e+00, -8.9843e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 8.9284e+00,  1.0004e+01, -6.9210e+00],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-1.6906e+00,  7.8860e+00,  7.7142e+00],
        [ 6.2905e+00,  1.9366e+00,  2.1169e+00],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 2.0243e+00,  7.3086e+00,  7.5432e+00],
        [ 5.3860e+00, -2.3418e+00,  1.0826e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [-9.4714e+00, -1.3301e+00, -3.1998e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-3.4986e+00,  7.5152e+00,  9.0712e+00],
        [ 7.1454e-01, -4.8071e+00, -2.3766e+00],
        [ 3.2990e+00, -4.2415e+00, -7.4857e+00],
        [ 1.0476e+01, -5.9539e+00,  1.3667e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [ 5.6523e+00,  3.9443e+00, -2.9449e+00],
        [-8.0829e+00, -5.9940e+00,  5.0564e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 5.8501e-01, -4.7982e+00,  9.6639e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [-9.1231e+00, -3.5739e+00,  8.4708e+00],
        [ 1.2728e+00, -1.1037e+01,  9.8121e+00],
        [ 6.0890e+00, -8.3818e-01,  2.9166e+00],
        [ 2.5128e+00, -1.0936e+01,  9.9645e+00],
        [ 3.7155e-01, -4.7756e+00,  2.3116e-01],
        [ 5.5158e+00, -2.1337e+00,  3.5246e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [ 5.8325e+00, -1.5222e+00, -2.1341e+00],
        [-7.3446e+00,  4.9113e+00, -5.9356e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 5.0348e+00, -2.8229e+00,  3.5436e+00],
        [ 1.1773e+01, -3.4876e+00, -6.7402e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-3.3116e+00, -1.0154e+01,  7.5919e+00],
        [ 1.5320e-01, -4.7422e+00, -7.0145e+00],
        [ 1.3103e+00, -4.8032e+00, -9.2997e-01],
        [ 5.4732e+00, -2.2041e+00, -4.3830e+00],
        [ 1.0000e+00,  7.6591e+00, -4.5011e+00],
        [ 2.3376e+00, -4.6231e+00,  9.8436e+00],
        [-7.0120e+00,  5.2825e+00,  7.7762e+00],
        [ 2.3367e+00,  7.1676e+00, -9.5263e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.0876, 0.1976,  ..., 1.2905, 1.1510, 1.4004],
        [0.0876, 0.0000, 0.1738,  ..., 1.2413, 1.1005, 1.3177],
        [0.1976, 0.1738, 0.0000,  ..., 1.0948, 0.9560, 1.2624],
        ...,
        [1.2905, 1.2413, 1.0948,  ..., 0.0000, 0.1419, 0.8261],
        [1.1510, 1.1005, 0.9560,  ..., 0.1419, 0.0000, 0.7879],
        [1.4004, 1.3177, 1.2624,  ..., 0.8261, 0.7879, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.0393949 , 14.05687815, 14.10045797,  8.29022197,  8.30729352,
        8.36445758, 13.68672855, 10.06703358, 13.59480906,  6.84796202,
        5.29009836,  9.17016857,  9.99211663,  7.66377554, 12.50996828,
        9.50913849,  6.18668463,  9.17595899,  7.06372172, 12.07818871,
        7.62509582,  5.62005629, 13.77604293, 11.04313641, 12.1460931 ,
       12.27993098, 10.78533838, 11.83570132,  5.91408504,  7.24326992,
        8.53501777, 11.018084  ,  9.78147542,  9.15253344,  6.41202986,
       10.19036081,  7.13501264, 10.65929968,  9.60359158,  5.8828986 ,
        4.7663526 ,  7.44805901, 10.09979747,  6.83083863, 11.73848065,
       12.4013153 ,  9.50913849, 13.61600536, 13.06498554, 13.44593519,
       12.27991991, 10.0674887 ,  6.42689705, 13.2430377 , 11.67875855,
       11.05413545, 10.31584102, 13.40850328,  8.25828409,  8.06517143,
        6.58184474, 11.83570132,  7.58378655,  5.87305005,  4.73784222,
        9.5642993 , 14.10796466,  8.28968067,  4.85995002,  5.37343173,
       12.0495521 , 10.00802627,  6.89245352, 10.06286074, 12.72524615,
       13.98691426,  4.83371341, 10.42667093,  9.79814997, 11.11038551,
        6.1463896 , 11.22142195,  4.79003361,  5.91408504,  7.8788965 ,
       13.58333857, 13.79762415, 10.31584102,  6.02789634,  8.8353674 ,
       12.45418236,  5.77219077, 12.27835996,  8.08320235,  4.78002665,
       11.9120321 , 10.68031351,  4.74468406,  4.978704  ,  5.90035324,
        7.72415048,  5.1805345 ,  8.77913428,  7.53884104, 11.89855267,
        9.60359158])}
{'P': tensor([[3.6276e-03, 9.7458e-01, 7.7414e-03,  ..., 2.8803e-07, 2.3267e-08,
         8.7472e-09],
        [4.4980e-04, 1.8928e-01, 6.4019e-02,  ..., 1.6646e-05, 1.1214e-06,
         4.5233e-07],
        [2.1431e-02, 1.5197e-01, 1.8566e-01,  ..., 1.2326e-05, 1.0871e-06,
         5.4192e-07],
        ...,
        [2.5285e-07, 2.3627e-07, 1.9498e-07,  ..., 3.1671e-02, 6.4702e-05,
         1.1156e-05],
        [3.0859e-04, 2.4667e-04, 2.4057e-04,  ..., 4.2169e-07, 2.7079e-02,
         1.0564e-05],
        [2.0837e-05, 1.7906e-05, 2.5103e-05,  ..., 4.3553e-07, 4.2504e-03,
         9.1692e-02]]), 'X': tensor([[ 1.1350e+00,  1.4010e+01,  6.3209e-01],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [-9.4501e+00, -1.7133e+00,  2.2980e+00],
        [ 6.3515e+00,  6.4185e-01, -6.5823e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-4.8751e+00, -9.3148e+00,  4.0406e+00],
        [ 3.8182e+00, -3.9347e+00, -1.7961e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 6.3119e+00,  1.7797e+00,  9.9186e+00],
        [ 8.7715e+00, -7.9390e+00,  5.2900e+00],
        [ 2.1625e+00, -4.6697e+00,  7.0658e+00],
        [-4.5778e+00, -9.5013e+00, -5.0667e+00],
        [ 4.5340e+00, -3.3629e+00,  8.9652e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 5.7802e+00, -1.6363e+00, -6.2782e+00],
        [ 2.5382e+00,  7.0672e+00,  4.5097e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [-6.6895e+00,  5.6051e+00,  2.7899e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 9.0051e+00,  9.9234e+00,  5.4411e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-6.4822e+00, -8.0252e+00, -3.8983e+00],
        [ 1.2576e+01,  1.3325e-01, -9.2650e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-7.8849e+00, -6.3115e+00, -4.7479e+00],
        [ 6.9517e-01, -4.8060e+00,  8.8976e+00],
        [ 4.9960e-01,  7.7738e+00, -4.4711e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 2.6970e+00,  6.9826e+00,  9.7350e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [ 6.0467e+00, -9.8369e+00, -6.5816e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [ 9.9058e+00, -6.7203e+00,  5.6928e+00],
        [ 7.0664e+00, -9.2590e+00,  8.0879e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [ 1.2604e+01,  7.6652e-01, -3.4439e+00],
        [-6.4368e+00,  5.8356e+00,  3.3014e+00],
        [-5.3179e+00, -9.0101e+00,  9.4472e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 5.3602e+00, -2.3808e+00,  7.2098e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 1.0993e+01,  7.2107e+00,  7.7933e+00],
        [ 6.0788e+00, -8.7102e-01, -6.8719e+00],
        [ 1.7500e+00,  7.4187e+00,  4.4237e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 2.7590e+00,  1.3662e+01,  9.4496e+00],
        [ 1.2600e+01,  1.3971e+00, -1.0694e+00],
        [ 6.3604e+00,  9.0416e-01, -6.5901e+00],
        [ 6.6319e+00,  1.1907e+01,  3.5165e+00],
        [-9.1623e+00, -3.4288e+00, -4.9961e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-6.0844e+00, -8.3955e+00,  7.7048e-01],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [ 1.2345e+01, -1.5462e+00,  7.0212e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 5.4867e+00,  1.2582e+01,  2.9549e+00],
        [ 3.2647e+00, -1.0807e+01, -3.7677e-01],
        [ 1.0271e+01,  8.3707e+00, -4.9743e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [-8.5859e+00, -5.0393e+00,  5.4759e-01],
        [-7.7239e+00,  4.4298e+00,  3.1216e+00],
        [ 1.3767e+00, -1.1034e+01,  9.9415e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [-8.0330e+00, -6.0765e+00,  6.7945e+00],
        [ 5.4429e+00, -2.2529e+00,  9.7306e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [-9.2974e+00,  8.3708e-01,  2.4852e+00],
        [ 5.2608e+00,  1.2698e+01,  4.9166e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [-4.3359e+00, -9.6431e+00, -9.0616e+00],
        [-9.3076e+00,  7.8517e-01, -6.7010e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [-1.8823e+00, -1.0656e+01, -3.8952e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-8.3062e+00, -5.6008e+00,  2.8744e-02],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 7.0642e+00,  1.1610e+01, -9.8316e+00],
        [ 2.4639e+00, -1.0943e+01, -7.3718e+00],
        [-5.6169e+00, -8.7844e+00,  8.9057e+00],
        [ 1.2231e+01, -2.0445e+00,  1.3545e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 1.1961e-01, 2.1505e-01,  ..., 1.1571e+00, 1.5416e-01,
         6.9966e-01],
        [1.1961e-01, 0.0000e+00, 1.5405e-01,  ..., 1.0631e+00, 4.2076e-02,
         6.5170e-01],
        [2.1505e-01, 1.5405e-01, 0.0000e+00,  ..., 1.1419e+00, 1.7735e-01,
         4.9877e-01],
        ...,
        [1.1571e+00, 1.0631e+00, 1.1419e+00,  ..., 0.0000e+00, 1.0210e+00,
         1.4341e+00],
        [1.5416e-01, 4.2076e-02, 1.7735e-01,  ..., 1.0210e+00, 1.0727e-04,
         6.6813e-01],
        [6.9966e-01, 6.5170e-01, 4.9877e-01,  ..., 1.4341e+00, 6.6813e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.05633441, 14.09357547, 14.08549802,  8.31625171,  8.43626114,
        8.4486771 , 13.25336001,  9.60412735,  6.383897  , 13.96987806,
       10.51340829,  5.48276038, 12.7125079 ,  6.55800267, 11.83075005,
        5.14607014, 10.54658597,  5.64500512, 10.33712258,  6.00732193,
        7.50917551,  5.07306274,  8.7273569 ,  4.98989906, 12.71843462,
       13.40024224, 13.1399181 , 13.32557133, 14.02532758, 11.35182589,
        7.62509582,  9.09119351, 10.31613269, 12.5769659 , 12.46513646,
       10.09979747,  4.85603886,  7.78980299, 10.48315436,  7.48538905,
        6.81084724, 10.46238054,  7.07902036, 11.54672755,  8.49161154,
       10.4606265 , 11.46176259, 11.9702762 , 11.64746882,  8.16584537,
       13.45124917, 10.8207297 , 12.62711252,  8.68832398, 10.46238054,
       11.52733581,  5.86519525, 11.99633819, 13.14690137,  6.14086582,
        7.62232307,  7.62011734, 13.32613445, 13.93790235, 12.67680439,
        6.42439344, 13.62896593,  9.78286733, 10.48315436,  9.96391045,
       10.36844813,  4.98989906,  8.57560275,  7.46248947, 12.44177641,
       13.45642379,  9.78147542, 13.72594022, 11.28895489, 13.25017211,
       10.98076873,  5.05774295,  5.16611782,  9.95552626,  8.90405823,
       11.11969815,  9.36256096,  7.23276582, 10.07238463,  5.89074104,
        4.8763135 ,  9.3349858 , 13.744387  ,  4.89869205, 10.57302546,
        9.34061938, 11.48434767, 10.8207297 ,  7.8788965 , 10.01804223,
       13.27717982, 13.59056613, 11.21703654, 10.42667093, 12.40074901,
       11.87665501])}
{'P': tensor([[1.5259e-01, 1.1058e-01, 1.0916e-01,  ..., 8.4380e-05, 4.9981e-03,
         4.6243e-02],
        [3.9753e-05, 2.3711e-02, 5.8980e-01,  ..., 1.6252e-04, 2.3287e-04,
         9.3305e-03],
        [5.9041e-05, 1.2840e-03, 5.3924e-02,  ..., 4.4326e-04, 3.9002e-04,
         1.4369e-02],
        ...,
        [3.4673e-07, 6.5783e-06, 7.2397e-06,  ..., 5.6784e-02, 1.3861e-05,
         1.6699e-05],
        [8.1972e-04, 1.1849e-02, 1.3107e-02,  ..., 1.8122e-03, 1.0653e-01,
         1.6507e-01],
        [3.1504e-05, 2.5524e-03, 4.2352e-03,  ..., 2.3974e-03, 3.7005e-04,
         4.8993e-02]]), 'X': tensor([[  1.1274,  14.0116,   8.1753],
        [ -3.5029,   7.5138,   8.1853],
        [ -4.0868,   7.2981,   8.1874],
        [ -3.6381,   7.4683,   7.3544],
        [ -3.4986,   7.5152,   9.0712],
        [ -3.5520,   7.4976,   9.1522],
        [ -7.0120,   5.2825,   7.7762],
        [  3.2809,  13.5061,  -4.0199],
        [  9.7328,   9.0902,  -0.8917],
        [ -0.1626, -10.9796,   6.4629],
        [ -5.9938,  -8.4745,   3.9735],
        [  0.2571,  -4.7594,   1.5618],
        [  6.1685,  12.1979,  -6.2473],
        [ -7.4808,   4.7462,   2.0266],
        [  5.0277,   4.9589,   7.0175],
        [  5.0190,   4.9705,  -7.9752],
        [  1.4150,  13.9647,  -4.0915],
        [  4.1637,  13.1898,  -1.9189],
        [ 11.9019,  -3.1379,  -3.4534],
        [ -5.6169,  -8.7844,   8.9057],
        [ -8.7078,   2.7182,   1.8146],
        [  1.8721,  -4.7322,   0.5524],
        [ -8.0750,  -6.0072,  -5.8267],
        [  1.3403,  -4.8011,  -1.1698],
        [  6.7565,  11.8237,  -4.9882],
        [ 12.5224,  -0.4497,   4.8448],
        [ 11.1085,  -4.9204,   0.0736],
        [ 11.2939,   6.6325,  -5.8311],
        [  5.4426, -10.1181,  -5.1268],
        [  3.9022,  13.2906,   5.5788],
        [  5.7963,  -1.6018,   1.8856],
        [ -9.4351,  -1.8891,   3.5642],
        [ -1.6334, -10.7207,   7.1397],
        [  5.9655,   3.2172,  -7.5953],
        [ -9.4721,  -1.3097,  -7.4377],
        [ -3.4488, -10.0934,   2.5195],
        [ -4.7389,   6.9953,   3.7638],
        [ -7.6734,   4.4980,  -5.4248],
        [  4.0457,  -3.7737,  -0.4704],
        [  9.8082,  -6.8396,   4.9615],
        [ -4.5778,  -9.5013,  -5.0667],
        [  3.0876,   6.7527,  -5.3540],
        [  3.9515,  -3.8425,   6.1729],
        [ -6.4844,  -8.0230,   8.5209],
        [ 11.8951,  -3.1570,   7.0742],
        [ -2.0721,   7.8432,   6.6380],
        [  1.9204,   7.3518,   8.4781],
        [  1.4749,  -4.7892,  -0.8089],
        [ 11.3800,  -4.3919,   1.4066],
        [ 11.7794,   5.5086,  -9.6929],
        [ -7.3470,  -7.0615,   5.0066],
        [ -9.3810,  -2.3455,  -9.5194],
        [  3.3943,   6.5483,   1.0790],
        [  5.9193,  -1.3176,   5.1739],
        [ 11.1085,  -4.9204,   0.0736],
        [ -7.0120,   5.2825,   7.7762],
        [ 11.7418,   5.6070,  -1.4192],
        [ -8.8934,   2.2597,   3.5831],
        [ -6.0844,  -8.3955,   0.7705],
        [ -8.8540,  -4.3971,  -5.2144],
        [  3.3069,   6.6089,   8.7377],
        [  2.2182,  13.8007,   5.1889],
        [  5.3858,   4.4260,  -9.6769],
        [  6.3542,   0.6934,  -3.3840],
        [ 12.4597,  -0.9136,  -4.8763],
        [  6.9284,  11.7061,  -0.3732],
        [  5.8275,  -1.5334,   2.0124],
        [  1.1350,  14.0104,   0.6321],
        [ -9.4283,  -1.9585,  -0.9129],
        [  9.3560,   9.5391,   9.4943],
        [  6.1222,  -0.7261,   8.8204],
        [ 10.8218,  -5.4177,   6.8573],
        [ 10.9162,  -5.2599,  -7.0812],
        [ -6.4002,  -8.1048,  -0.9108],
        [  2.0547,   7.2957,  -1.9343],
        [ 12.3357,   3.6001,   0.7903],
        [ -7.6719,  -6.6260,   7.1468],
        [ -3.5173,   7.5091,  -8.9585],
        [ -8.6801,   2.7809,  -0.8443],
        [ -7.3446,   4.9113,  -5.9356],
        [ 11.4235,   6.3594,   5.1663],
        [ -3.5173, -10.0623,   4.8280],
        [  6.3549,   0.7096,   6.9198],
        [  2.1504,  -4.6726,   8.7908],
        [ -1.6275, -10.7221,   3.2376],
        [  8.6420,  -8.0580,   6.5781],
        [ -2.1817,   7.8274,  -4.6298],
        [ -2.5683, -10.4433,  -2.9182],
        [  1.7903,  -4.7467,   0.1168],
        [ -8.3900,   3.3727,   3.1650],
        [  1.8719,   7.3713,  -3.7465],
        [  6.2895,   0.0405,   5.7652],
        [  2.0039,   7.3173,  -1.5885],
        [ -8.7078,   2.7182,   1.8146],
        [ -8.9624,  -4.0944,  -4.9622],
        [  6.3505,   0.6235,  -8.5849],
        [ -3.5903,   7.4847,  -8.9112],
        [  5.6258,  -1.9404,  -6.0715],
        [  6.2804,   2.0019,   0.8607],
        [  2.5375,   7.0676,  -4.3490],
        [ -6.0195,   6.1789,  -2.9496],
        [  3.9085,   6.1512,  -2.2560],
        [ -8.4297,  -5.3637,  -9.3962],
        [ -9.1433,  -3.5003,  -3.0209],
        [ -0.5739,   7.9056,   0.6382],
        [ -4.7389,   6.9953,   3.7638]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000e+00, 5.4370e-01, 5.7530e-01,  ..., 1.5015e+00, 7.9746e-01,
         7.2939e-01],
        [5.4370e-01, 0.0000e+00, 3.3658e-02,  ..., 1.0231e+00, 5.2185e-01,
         2.6175e-01],
        [5.7530e-01, 3.3658e-02, 0.0000e+00,  ..., 1.0045e+00, 5.3089e-01,
         2.5486e-01],
        ...,
        [1.5015e+00, 1.0231e+00, 1.0045e+00,  ..., 0.0000e+00, 7.7912e-01,
         7.8014e-01],
        [7.9746e-01, 5.2185e-01, 5.3089e-01,  ..., 7.7912e-01, 4.7405e-06,
         3.2017e-01],
        [7.2939e-01, 2.6175e-01, 2.5486e-01,  ..., 7.8014e-01, 3.2017e-01,
         0.0000e+00]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.05687815,  8.29022197,  8.36445758,  8.30729352,  8.28968067,
        8.29640613,  8.77913428, 13.89886469, 13.31764181, 10.98076873,
       10.37998078,  4.7663526 , 13.66896739,  8.859416  ,  7.06169391,
        7.06372172, 14.03618649, 13.83139024, 12.30858677, 10.42667093,
        9.12220271,  5.08910491, 10.06437598,  4.98461983, 13.61801526,
       12.53047489, 12.1494023 , 13.09737053, 11.4890861 , 13.85158417,
        6.01356004,  9.62238828, 10.84437705,  6.77777606,  9.562174  ,
       10.66631784,  8.44940661,  8.89458273,  5.53255882, 11.95743558,
       10.54658597,  7.42512621,  5.51177501, 10.31584102, 12.30695098,
        8.11227668,  7.5984809 ,  5.01114719, 12.19805394, 13.00380624,
       10.19036081,  9.66977802,  7.37576154,  6.06416957, 12.1494023 ,
        8.77913428, 13.01187484,  9.17595899, 10.36844813,  9.88571713,
        7.39004226, 13.97779839,  6.97106912,  6.39187285, 12.49317563,
       13.60275211,  6.02588517, 14.05633441,  9.62958973, 13.3614598 ,
        6.16512856, 12.10222204, 12.11735334, 10.32715715,  7.57946459,
       12.85032836, 10.1371552 ,  8.29203356,  9.11473391,  8.8353674 ,
       13.0743399 , 10.65929968,  6.39438044,  5.14371005, 10.84493732,
       11.8159254 ,  8.12580976, 10.75443106,  5.07306274,  9.04255655,
        7.60529892,  6.28961988,  7.58668144,  9.12220271,  9.85330646,
        6.38105015,  8.30125201,  5.9510489 ,  6.59176018,  7.5092834 ,
        8.62631801,  7.28793663,  9.99145096,  9.79039371,  7.92644746,
        8.44940661])}
{'P': tensor([[1.6538e-01, 1.3938e-01, 1.1815e-01,  ..., 5.3369e-07, 8.3756e-07,
         1.5342e-07],
        [4.2388e-02, 1.8743e-01, 9.9259e-02,  ..., 7.2929e-07, 9.3840e-07,
         2.5312e-07],
        [4.2069e-05, 4.4792e-05, 2.4164e-02,  ..., 1.2778e-06, 2.5015e-06,
         2.5298e-07],
        ...,
        [6.5079e-06, 8.1525e-06, 5.4522e-06,  ..., 2.3939e-02, 1.2468e-03,
         4.9255e-04],
        [1.0267e-05, 9.4108e-06, 9.1442e-06,  ..., 5.4936e-04, 1.6041e-02,
         1.5995e-05],
        [1.7879e-06, 2.8157e-06, 1.5283e-06,  ..., 2.9131e-03, 1.0778e-04,
         5.3248e-02]]), 'X': tensor([[ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [ 6.0031e+00,  1.2296e+01, -6.5992e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 9.2811e+00, -7.4336e+00,  5.1771e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 4.2041e+00, -3.6509e+00, -4.9488e+00],
        [ 1.0987e+01,  7.2207e+00, -8.8549e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [-7.9925e+00, -6.1423e+00,  2.8990e+00],
        [ 7.9577e-01, -4.8109e+00, -8.2248e+00],
        [ 1.9204e+00,  7.3518e+00,  8.4781e+00],
        [-8.8745e+00,  2.3097e+00,  3.5600e+00],
        [ 6.9653e+00, -9.3225e+00, -3.4727e+00],
        [ 2.3373e+00, -4.6232e+00, -5.2246e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [ 1.0247e+01,  8.4054e+00, -1.1769e-01],
        [ 1.2233e+01, -2.0376e+00, -1.1234e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-8.6681e+00, -4.8556e+00,  9.2518e+00],
        [ 2.9410e+00,  6.8428e+00,  5.4458e+00],
        [-9.4739e+00, -1.2497e+00,  5.4829e-01],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-1.0012e+00, -1.0858e+01,  2.2867e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [ 3.2366e+00, -4.2732e+00,  5.1532e+00],
        [ 3.2692e+00,  1.3510e+01, -4.9032e+00],
        [ 1.0667e+01, -5.6649e+00, -4.1646e+00],
        [ 5.3858e+00,  4.4260e+00, -9.6769e+00],
        [ 9.9217e+00, -6.7006e+00, -5.6728e+00],
        [ 1.1044e+01,  7.1180e+00, -5.6991e+00],
        [-9.1459e+00,  1.4780e+00, -4.2403e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.2058e-01, -4.7363e+00,  8.7103e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [-2.0445e+00,  7.8469e+00, -5.8569e-01],
        [ 1.2194e+01, -2.1881e+00,  7.1499e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [ 6.2902e+00,  4.5065e-02,  3.7116e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 2.4697e+00, -4.5836e+00,  1.4593e+00],
        [ 9.8093e-01,  7.6642e+00, -1.4217e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [-9.7149e-01, -1.0863e+01,  3.5546e-01],
        [ 1.2336e+01,  3.6001e+00,  7.9029e-01],
        [ 6.0140e+00, -1.0661e+00, -2.5040e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 5.7218e+00, -9.9934e+00,  6.5764e+00],
        [ 1.7218e+00, -4.7576e+00, -3.3617e+00],
        [-8.0087e+00,  4.0173e+00,  9.1091e+00],
        [ 6.2790e+00, -2.5947e-02, -5.4116e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [-5.3736e+00,  6.6310e+00, -1.5174e+00],
        [ 1.2175e+01, -2.2592e+00, -5.7751e+00],
        [ 2.4932e+00,  7.0903e+00, -5.4214e+00],
        [ 9.4554e+00,  9.4245e+00,  9.8800e+00],
        [ 5.9912e+00,  3.1453e+00, -3.9989e+00],
        [ 1.2244e+01,  4.0076e+00,  8.1257e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [-8.8631e+00, -4.3728e+00, -1.1746e+00],
        [ 5.5424e+00, -2.0885e+00,  7.2487e+00],
        [ 1.0100e+01, -6.4739e+00,  2.0974e+00],
        [ 3.9591e+00, -1.0639e+01, -6.2557e+00],
        [ 1.3705e+00,  1.3972e+01,  7.9266e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.2401e+01, -1.2597e+00, -7.6474e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-5.9938e+00, -8.4745e+00,  3.9735e+00],
        [ 1.1994e+01, -2.8677e+00,  6.0614e+00],
        [ 4.8674e+00,  1.2887e+01,  6.7622e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.1778e+01,  5.5127e+00, -2.0699e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 6.3583e+00,  1.1729e+00,  8.7835e+00],
        [-8.9606e+00, -4.0994e+00,  1.7886e+00],
        [-2.0721e+00,  7.8432e+00,  6.6380e+00],
        [-6.0175e+00, -8.4541e+00, -9.8408e+00],
        [-6.8466e+00, -7.6488e+00,  1.9648e+00],
        [-3.5520e+00,  7.4976e+00,  9.1522e+00],
        [ 4.3459e+00, -3.5327e+00,  3.6674e-01],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-9.0574e+00,  1.7801e+00,  8.8986e+00],
        [ 1.1221e+01, -4.7092e+00,  2.6117e+00],
        [-9.4196e+00,  4.7830e-02,  9.4140e+00],
        [-7.0948e+00,  5.1940e+00,  9.8171e+00],
        [ 6.3413e+00,  4.8834e-01, -6.8416e+00],
        [-6.3252e+00, -8.1760e+00, -1.0891e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [ 8.8144e+00, -7.8988e+00, -1.2816e-02],
        [ 9.0424e+00, -7.6779e+00, -7.5286e+00],
        [ 8.2154e+00, -8.4262e+00,  8.0171e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1077, 0.4707,  ..., 0.6079, 0.1974, 1.2534],
        [0.1077, 0.0000, 0.4324,  ..., 0.5015, 0.1810, 1.1460],
        [0.4707, 0.4324, 0.0000,  ..., 0.5127, 0.2736, 1.1534],
        ...,
        [0.6079, 0.5015, 0.5127,  ..., 0.0000, 0.5282, 0.6705],
        [0.1974, 0.1810, 0.2736,  ..., 0.5282, 0.0000, 1.1979],
        [1.2534, 1.1460, 1.1534,  ..., 0.6705, 1.1979, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.07623266, 14.09176259,  8.30125201,  8.29203356,  8.27853076,
        8.41171369, 13.6829843 ,  7.49076899, 11.89105259, 11.05413545,
       11.99633819,  5.56809335, 13.14777433,  8.4486771 , 10.88618114,
       10.08001894,  4.8763135 ,  7.5984809 ,  9.17016857, 11.63724792,
        5.18047638, 10.48315436, 13.25336001, 12.4013153 ,  9.6282763 ,
        9.9353921 ,  7.44805901,  9.55593374,  9.15866607, 10.9036226 ,
        8.49161154, 10.98076873, 11.22596994,  8.05668609,  8.09842704,
        5.36061974, 13.89974693, 12.07818871,  6.97106912, 11.97238624,
       13.13888841,  9.26455795,  8.04664886,  4.73784222,  6.24187044,
       11.65923396,  8.10886021, 12.38881635,  8.38252937,  6.29034958,
       12.12641969,  5.20659892,  7.72668467,  8.94395628,  9.17905491,
       10.90638066, 12.85032836,  6.10774544, 13.3614598 , 11.51557166,
        5.05963217,  8.95983586,  6.27905292,  5.21280156,  6.06416957,
        9.41605684,  8.53501777, 12.3828957 ,  7.5158489 , 13.35013374,
        6.76661595, 12.88270018, 13.32557133, 10.98076873,  9.88310614,
        5.92281142, 11.99633819, 11.35182589, 14.0393949 ,  6.4020404 ,
       12.46513646,  8.01009083, 10.37998078, 12.33167382, 13.77604293,
       13.1471313 , 13.00413829,  5.83560009,  6.46560181,  9.85384623,
        8.11227668, 10.37698435, 10.26545792,  8.29640613,  5.60064319,
        6.76056625,  9.23071698, 12.16901615,  9.41970031,  8.79284466,
        6.36004393, 10.33712258,  7.67939546, 11.83570132, 11.86239961,
       11.76830981])}
{'P': tensor([[1.5113e-01, 1.2371e-01, 1.0592e-01,  ..., 4.8931e-04, 1.1455e-04,
         1.0543e-03],
        [5.9221e-02, 1.7510e-01, 9.8517e-02,  ..., 3.8570e-04, 1.6044e-04,
         8.8353e-04],
        [1.0257e-04, 1.0514e-04, 9.0123e-02,  ..., 1.8559e-04, 6.9901e-05,
         2.6953e-04],
        ...,
        [9.7021e-03, 5.3761e-03, 3.2398e-02,  ..., 2.0991e-01, 1.9006e-04,
         1.4593e-01],
        [4.4031e-03, 7.6976e-03, 3.3710e-03,  ..., 2.1692e-04, 1.0045e-01,
         6.5399e-04],
        [1.3341e-02, 7.9413e-03, 6.7736e-02,  ..., 2.7859e-02, 2.7223e-04,
         1.4201e-01]]), 'X': tensor([[ 7.2746e-01,  1.4067e+01, -9.8147e-01],
        [ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-4.0868e+00,  7.2981e+00,  8.1874e+00],
        [-6.4002e+00, -8.1048e+00, -9.1076e-01],
        [-3.2736e+00, -1.0170e+01,  3.8712e+00],
        [-3.9456e+00, -9.8542e+00,  7.4660e-01],
        [ 3.7121e+00,  6.3117e+00,  6.7381e-01],
        [ 1.2500e+01,  2.6335e+00, -6.3166e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 5.3118e+00,  4.5452e+00,  4.2114e+00],
        [ 1.2527e+01, -4.1248e-01, -8.7331e+00],
        [-8.3535e+00, -5.5118e+00,  9.8399e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 2.2644e+00, -4.6434e+00,  9.7508e+00],
        [ 7.5309e+00,  1.1262e+01,  7.2656e+00],
        [ 1.7122e+00, -4.7591e+00, -3.6734e+00],
        [ 6.7566e-01, -1.1036e+01, -1.3047e+00],
        [ 4.5954e+00,  1.3010e+01, -4.4439e+00],
        [-1.6334e+00, -1.0721e+01,  7.1397e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 1.0062e+01,  8.6627e+00, -9.2884e+00],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [-8.9788e+00,  2.0208e+00, -3.2867e+00],
        [ 4.9531e+00,  5.0576e+00,  5.6794e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-5.7091e-01,  7.9055e+00,  7.6089e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 6.7565e+00,  1.1824e+01, -4.9882e+00],
        [-8.4297e+00, -5.3637e+00, -9.3962e+00],
        [ 8.5716e+00,  1.0360e+01,  6.8818e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00],
        [ 5.9193e+00, -1.3176e+00,  5.1739e+00],
        [-8.4267e+00, -5.3697e+00,  2.4172e+00],
        [ 2.3589e+00,  1.3767e+01, -3.6501e+00],
        [ 6.9331e-01,  7.7336e+00,  2.1179e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 4.5767e+00,  5.5071e+00,  3.9069e+00],
        [ 3.8111e+00, -1.0679e+01, -4.6539e+00],
        [-6.4844e+00, -8.0230e+00,  8.5209e+00],
        [-2.5378e-02, -1.0993e+01,  3.2200e+00],
        [ 1.1994e+00,  7.6035e+00,  2.4295e+00],
        [-8.3539e+00, -5.5111e+00,  3.3733e-01],
        [-9.4366e+00, -1.1888e-01, -2.6736e+00],
        [ 5.8930e+00,  1.2359e+01, -8.9201e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 1.2593e+01,  4.1985e-01, -9.4997e+00],
        [-9.4716e+00, -6.6731e-01,  7.6068e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [ 5.5294e+00, -2.1106e+00,  9.0626e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 9.8086e-01, -4.8145e+00,  6.8706e+00],
        [ 2.3261e+00,  1.3775e+01,  5.9121e+00],
        [-4.7847e+00, -9.3729e+00, -3.7741e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [ 3.2226e+00, -4.2802e+00, -4.0428e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 1.1037e+01,  7.1299e+00, -8.5895e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [-6.1899e+00, -8.3011e+00, -7.9269e+00],
        [ 9.0737e-01, -4.8139e+00,  8.1674e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 9.6816e+00,  9.1536e+00, -5.8543e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [-8.1204e+00, -5.9308e+00,  4.0761e+00],
        [ 1.8719e+00,  7.3713e+00, -3.7465e+00],
        [-6.3489e+00,  5.9117e+00,  6.5161e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [-8.6229e+00,  2.9068e+00,  2.4429e+00],
        [-8.6426e+00, -4.9136e+00,  9.4941e+00],
        [ 6.3122e+00, -9.6994e+00, -6.3282e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 3.9663e+00,  6.1018e+00, -5.4104e+00],
        [ 9.3424e+00, -7.3685e+00, -3.6807e+00],
        [ 6.3087e+00,  1.8053e+00, -1.9481e+00],
        [ 3.3542e-01, -4.7708e+00, -4.9125e+00],
        [-9.4773e+00, -1.0024e+00,  5.2599e+00],
        [ 3.2306e-01, -4.7691e+00, -1.6120e+00],
        [ 6.0305e+00,  3.0296e+00,  1.3754e-01],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [ 1.5653e+00,  1.3938e+01,  4.1773e+00],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-4.4520e+00,  7.1370e+00, -8.7867e+00],
        [-3.1204e+00, -1.0235e+01,  9.2288e+00],
        [-8.6675e+00, -4.8569e+00, -9.1769e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 7.0171e+00,  1.1644e+01, -5.0178e+00],
        [ 1.0972e+01, -5.1646e+00, -6.9039e+00],
        [ 1.2376e+01, -1.3943e+00,  6.4659e+00],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 2.7121e+00,  6.9743e+00, -8.4202e+00],
        [ 4.4915e+00,  5.5990e+00,  7.0405e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1540, 0.4354,  ..., 0.4135, 0.5788, 0.3425],
        [0.1540, 0.0000, 0.4852,  ..., 0.5543, 0.4313, 0.4810],
        [0.4354, 0.4852, 0.0000,  ..., 0.3771, 0.6939, 0.3363],
        ...,
        [0.4135, 0.5543, 0.3771,  ..., 0.0000, 0.9275, 0.0740],
        [0.5788, 0.4313, 0.6939,  ..., 0.9275, 0.0000, 0.8547],
        [0.3425, 0.4810, 0.3363,  ..., 0.0740, 0.8547, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.08549802, 14.09357547,  8.38252937,  8.31625171,  8.25828409,
        8.4486771 ,  8.36445758, 10.32715715, 10.68416401, 10.61472636,
        7.32233733, 12.77402351,  6.24187044,  6.99097013, 12.53345445,
       10.00802627, 10.4606265 ,  5.16611782, 13.54774355,  5.05774295,
       11.05672136, 13.79762415, 10.84437705,  9.34127013,  7.62011734,
        6.48486171, 13.27717982, 13.36092624,  9.09119351,  9.18690018,
        8.16584537,  9.20340031,  7.07902036,  8.12043983,  7.92607342,
       12.5085995 , 13.61801526,  9.99145096, 13.44593519,  5.53175275,
        6.06416957,  9.99211663, 13.96746816,  7.76457095,  6.78325151,
       11.4424816 ,  9.60359158,  7.16058587, 11.33837396, 10.31584102,
       10.99326577,  7.69752262, 10.00794468,  9.43737523, 13.6922425 ,
        5.34250225, 12.59969936,  9.49511524,  8.34949383,  9.15866607,
        5.91854397,  5.40689332,  4.91336994, 13.96987806, 10.52357479,
       13.06498554,  5.35774573, 12.45418236, 13.1399181 , 12.32545637,
       10.35485331,  4.89869205, 12.3344536 , 13.32374056,  5.13182506,
       10.05558893,  7.60529892,  8.67502348,  8.85572292,  5.08910491,
        9.09963894,  9.94173708, 11.57249984, 11.50791859,  7.27759233,
       11.89855267,  6.5618968 ,  4.78258092,  9.53015376,  4.78002665,
        6.74872981,  9.43112957,  8.36995953, 14.02532758,  9.7712321 ,
        8.41171369, 10.69963954,  9.93553424, 13.58333857, 13.59480906,
       12.12641969, 12.45418236,  5.07306274,  7.48311002,  7.17789666,
        7.54620849])}
{'P': tensor([[1.7427e-01, 1.2755e-01, 1.1319e-01,  ..., 7.4218e-02, 3.2309e-04,
         2.1192e-05],
        [4.2419e-02, 1.9825e-01, 7.7194e-02,  ..., 9.2656e-02, 6.6265e-04,
         2.3054e-05],
        [2.3379e-04, 2.4262e-04, 1.0747e-01,  ..., 7.2711e-03, 9.6157e-05,
         2.0353e-04],
        ...,
        [2.2473e-04, 2.5759e-04, 3.3447e-02,  ..., 3.6875e-02, 1.2833e-04,
         6.6792e-05],
        [2.6250e-04, 6.5868e-04, 3.5882e-04,  ..., 1.4775e-03, 3.7641e-02,
         1.6600e-05],
        [1.9227e-04, 1.7287e-04, 5.7168e-04,  ..., 8.1379e-04, 3.1030e-04,
         1.0400e-01]]), 'X': tensor([[ 6.3961e-01,  1.4077e+01, -6.3786e+00],
        [ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-3.4100e+00,  7.5436e+00, -7.6980e+00],
        [-3.0004e+00,  7.6607e+00, -7.2640e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [ 2.4356e+00, -4.5942e+00, -5.0808e+00],
        [ 4.3523e+00, -1.0523e+01,  8.2138e+00],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 9.2692e+00,  9.6370e+00,  7.0746e+00],
        [-1.8367e+00,  7.8718e+00, -8.6523e+00],
        [ 6.1579e+00, -5.9609e-01, -6.5595e+00],
        [-4.7389e+00,  6.9953e+00,  3.7638e+00],
        [-8.7078e+00,  2.7182e+00,  1.8146e+00],
        [ 3.4401e+00,  1.3454e+01,  4.6576e+00],
        [ 1.8721e+00, -4.7322e+00,  5.5237e-01],
        [ 1.2269e+01, -1.8882e+00,  3.9253e+00],
        [ 1.3339e+00,  7.5627e+00, -9.0343e-01],
        [-9.1907e+00, -3.3180e+00, -5.5943e+00],
        [-1.5408e+00,  7.8978e+00,  5.3470e+00],
        [ 1.2488e+01, -7.2223e-01,  8.8347e+00],
        [ 3.7992e+00, -3.9474e+00, -8.9019e+00],
        [-9.3384e+00, -2.6161e+00, -4.7630e+00],
        [ 9.6592e-01, -4.8144e+00, -9.5942e-01],
        [ 6.3589e+00,  8.2387e-01, -8.4230e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [-5.9225e+00, -8.5355e+00,  2.4778e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-7.9298e+00,  4.1367e+00, -3.9773e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [-9.4296e+00, -1.9458e+00,  4.0833e+00],
        [-2.5056e+00,  7.7720e+00,  5.5878e+00],
        [ 5.9003e+00,  1.2355e+01,  2.1917e+00],
        [ 1.1970e+01, -2.9407e+00, -6.3498e+00],
        [ 1.0867e+01, -5.3428e+00, -2.8637e+00],
        [-7.8126e+00,  4.3066e+00,  5.1791e+00],
        [ 3.4920e+00, -4.1366e+00, -8.3955e+00],
        [ 1.7301e+00,  7.4262e+00,  5.0522e+00],
        [ 7.9816e+00,  1.0894e+01, -5.3776e-01],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-9.4721e+00, -1.3097e+00, -7.4377e+00],
        [ 4.8653e+00,  5.1694e+00, -4.0823e+00],
        [ 1.1449e+01,  6.3026e+00,  6.8257e+00],
        [ 5.9026e+00,  3.3837e+00, -6.5727e-01],
        [-8.5900e+00,  2.9767e+00,  7.4811e+00],
        [ 1.2460e+01, -9.1362e-01, -4.8763e+00],
        [ 6.3550e+00,  7.1295e-01,  4.9169e-01],
        [ 3.3744e+00,  6.5623e+00, -6.3636e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 6.0047e+00,  3.1062e+00, -6.7914e+00],
        [-6.8618e+00, -7.6322e+00,  3.7027e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 2.5375e+00,  7.0676e+00, -4.3490e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [ 2.6321e+00, -4.5296e+00, -2.5159e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-9.4735e+00, -1.2620e+00, -3.6742e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [ 5.3924e+00, -1.0140e+01, -4.0863e+00],
        [ 2.6310e+00, -1.0920e+01,  3.8054e+00],
        [-1.6220e+00,  7.8917e+00,  3.1501e-02],
        [ 3.1756e+00, -1.0825e+01,  6.8882e+00],
        [ 5.1519e+00, -1.0239e+01,  9.8191e+00],
        [-2.5683e+00, -1.0443e+01, -2.9182e+00],
        [ 2.6078e+00, -1.0923e+01,  6.2869e+00],
        [ 9.2979e+00,  9.6048e+00,  7.3138e+00],
        [ 6.2975e+00,  9.4668e-02,  5.1791e+00],
        [-4.8665e+00, -9.3204e+00, -5.2564e+00],
        [-9.3616e+00, -2.4750e+00,  3.7765e+00],
        [ 1.0180e+01,  8.5002e+00, -5.8893e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [ 6.1984e+00, -4.3266e-01,  1.5438e+00],
        [ 6.3534e+00,  1.2990e+00, -7.2942e+00],
        [ 5.8318e+00,  3.5567e+00, -4.7692e+00],
        [ 1.2577e+01,  1.8512e+00, -6.5785e-01],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [ 5.8845e+00,  3.4294e+00,  9.7870e-03],
        [-9.1658e+00, -3.4155e+00, -2.8880e+00],
        [ 4.5491e+00, -3.3485e+00,  6.8412e+00],
        [-7.0825e+00,  5.2073e+00, -5.0497e+00],
        [ 6.8704e+00, -9.3809e+00,  7.5769e+00],
        [ 3.1480e+00, -4.3165e+00, -7.8002e+00],
        [ 3.6469e+00,  6.3624e+00, -1.8414e+00],
        [ 9.1620e+00, -7.5573e+00, -9.9882e+00],
        [ 1.1108e+01, -4.9204e+00,  7.3577e-02],
        [-2.0229e+00, -1.0616e+01,  8.6930e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [-8.7544e+00,  2.6096e+00,  3.3146e+00],
        [ 2.1771e+00, -1.0978e+01, -3.2007e+00],
        [ 1.2291e+01,  3.8037e+00,  6.0978e+00],
        [ 2.0899e+00, -4.6870e+00, -7.4260e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 2.8474e+00,  6.8979e+00,  6.6457e+00],
        [-9.1706e+00,  1.3863e+00,  8.4032e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [-9.3464e+00,  5.7078e-01, -1.6397e+00],
        [-2.1817e+00,  7.8274e+00, -4.6298e+00],
        [ 1.8320e+00,  7.3871e+00,  5.1452e+00],
        [ 4.0078e+00, -3.8018e+00, -6.2168e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.1382, 0.4286,  ..., 0.3793, 0.8906, 0.1238],
        [0.1382, 0.0000, 0.5101,  ..., 0.4168, 0.7959, 0.1075],
        [0.4286, 0.5101, 0.0000,  ..., 0.1686, 0.8868, 0.4053],
        ...,
        [0.3793, 0.4168, 0.1686,  ..., 0.0000, 0.7250, 0.3101],
        [0.8906, 0.7959, 0.8868,  ..., 0.7250, 0.0000, 0.7673],
        [0.1238, 0.1075, 0.4053,  ..., 0.3101, 0.7673, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.09176259, 14.10796466,  8.27853076,  8.22728399,  8.51928531,
        8.34949383,  5.19986797, 11.38774149,  7.37902766, 13.37121923,
        8.08320235,  6.18668463,  8.44940661,  9.12220271, 13.88683436,
        5.08910491, 12.41366754,  7.67939546,  9.7712321 ,  8.04664886,
       12.5085995 ,  5.47866221,  9.69792513,  4.91038884,  6.41202986,
        5.21280156, 10.38897563,  6.65368551,  8.94395628,  6.65368551,
        9.6282763 ,  8.16584537, 13.69162795, 12.32545637, 12.10942412,
        8.92098752,  5.41348121,  7.62509582, 13.50484539,  5.79543038,
        9.562174  ,  7.09887557, 13.06957425,  6.80370254,  9.09119351,
       12.49317563,  6.3949043 ,  7.37902766,  7.81509631,  6.76056625,
       10.26327895, 11.54071859,  7.5092834 ,  9.50913849,  5.23878371,
        6.12886024,  9.55720651,  8.12043983, 11.48434767, 11.23201619,
        8.05668609, 11.28093507, 11.46176259, 10.75443106, 11.22993218,
       13.36800073,  6.29821673, 10.51437947,  9.68324553, 13.26209545,
        6.39438044, 10.67322527,  7.822854  , 12.1494023 ,  7.19534146,
        5.79562691,  9.36378438,  6.2134961 ,  6.48486171,  6.83083863,
       12.7125079 , 12.27991991,  6.81084724,  9.78147542,  5.64865173,
        8.7907923 , 11.62770332,  5.34250225,  7.33350642, 11.87665501,
       12.1494023 , 10.80728788, 13.58333857,  9.13507321, 11.1913598 ,
       12.86648771,  5.13182506,  6.4020404 ,  7.46248947,  9.27474386,
        9.18690018, 13.1471313 ,  9.36378438,  8.12580976,  7.61088125,
        5.52416368])}
{'P': tensor([[1.1323e-01, 8.1363e-02, 7.7935e-02,  ..., 5.1685e-02, 1.0848e-05,
         1.4735e-05],
        [2.8976e-05, 1.6576e-02, 8.5630e-02,  ..., 1.6125e-02, 1.1916e-05,
         4.1941e-05],
        [2.5132e-05, 1.5042e-03, 2.8448e-02,  ..., 2.9354e-02, 1.5907e-05,
         6.6721e-05],
        ...,
        [3.5948e-05, 2.7207e-03, 8.7325e-03,  ..., 6.1191e-02, 2.2083e-05,
         9.8552e-05],
        [2.1368e-03, 2.9936e-03, 1.6641e-03,  ..., 7.8472e-04, 2.6805e-01,
         2.8097e-03],
        [5.2480e-05, 2.3128e-04, 1.6070e-04,  ..., 9.5618e-05, 5.4940e-02,
         1.0413e-01]]), 'X': tensor([[ 6.1417e-01,  1.4080e+01,  5.4069e-01],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-4.7334e+00,  6.9982e+00, -4.3919e-01],
        [-5.1377e+00,  6.7751e+00, -4.4381e-01],
        [-3.2486e+00,  7.5925e+00, -1.3719e+00],
        [-8.8853e-01,  7.9160e+00,  5.7145e-02],
        [ 1.1654e+01,  5.8273e+00, -3.1375e+00],
        [ 1.0992e+01,  7.2133e+00, -9.1793e-01],
        [ 1.0468e-01, -1.1005e+01,  5.1884e-01],
        [ 1.2259e+01, -1.9312e+00,  9.8657e+00],
        [-8.9178e+00,  2.1936e+00,  7.5388e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 5.0277e+00,  4.9589e+00,  7.0175e+00],
        [ 2.3261e+00,  7.1726e+00,  6.2522e+00],
        [ 4.1297e+00, -3.7098e+00, -3.9838e+00],
        [-8.5506e+00, -5.1154e+00,  6.3423e+00],
        [-8.3900e+00,  3.3727e+00,  3.1650e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [-9.1178e+00,  1.5784e+00,  2.1410e+00],
        [ 8.5719e-01,  1.4050e+01, -8.3364e+00],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-8.8363e+00,  2.4086e+00, -5.6296e+00],
        [-3.4488e+00, -1.0093e+01,  2.5195e+00],
        [-4.6942e+00, -9.4299e+00, -9.1846e+00],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [-9.4753e+00, -8.0124e-01, -5.2050e+00],
        [-9.4671e+00, -5.5534e-01,  1.1172e-01],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [ 4.6231e+00,  5.4556e+00,  8.4973e+00],
        [ 5.2608e+00, -2.5256e+00, -9.1446e+00],
        [ 2.7318e-01, -4.7619e+00, -1.1577e+00],
        [-9.4735e+00, -7.2744e-01, -2.6749e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 3.6703e+00,  6.3443e+00, -5.4401e+00],
        [ 2.4347e-01,  7.8191e+00, -6.4709e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 3.0382e-01,  7.8092e+00,  4.1639e-01],
        [ 7.1821e+00, -9.1845e+00,  9.8548e+00],
        [ 8.0381e+00, -8.5690e+00, -8.9523e+00],
        [ 1.1520e+01, -4.0923e+00,  5.5088e+00],
        [ 8.4691e+00,  1.0457e+01, -5.0650e+00],
        [-5.1398e+00, -9.1367e+00,  6.2074e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-6.2200e+00, -8.2737e+00,  5.5079e+00],
        [-2.1382e+00,  7.8339e+00,  7.3524e+00],
        [-8.8154e+00,  2.4611e+00,  2.5198e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [ 3.4140e+00, -4.1803e+00, -8.8224e+00],
        [ 1.4489e+00,  7.5256e+00,  4.4529e+00],
        [ 7.8425e+00, -8.7202e+00,  6.4530e+00],
        [ 6.4697e-01, -1.1035e+01, -1.1931e+00],
        [ 4.7739e+00, -3.1218e+00, -4.1188e+00],
        [-5.3328e+00, -8.9992e+00,  7.3440e+00],
        [ 2.4799e-01, -1.1015e+01, -3.0183e+00],
        [-2.3317e+00,  7.8034e+00, -3.6274e+00],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [-9.1795e+00, -3.3623e+00,  8.5259e+00],
        [ 6.3549e+00,  7.0956e-01,  6.9198e+00],
        [ 4.5814e+00, -1.0449e+01,  2.5496e+00],
        [-8.9033e+00,  2.2329e+00, -8.8912e+00],
        [ 6.0560e+00, -9.4209e-01, -5.0334e+00],
        [-7.4602e+00,  4.7718e+00, -4.6277e+00],
        [-5.6651e+00,  6.4380e+00,  1.3660e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 8.3791e+00,  1.0541e+01,  5.5590e+00],
        [-9.3087e+00,  7.7916e-01, -7.5587e+00],
        [-8.3653e+00,  3.4184e+00,  9.3454e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [-7.2986e+00, -7.1225e+00, -3.0814e+00],
        [ 3.5939e+00,  6.4029e+00, -1.8357e+00],
        [-5.9872e+00,  6.2037e+00, -9.4477e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 9.7881e-01, -4.8145e+00, -3.8661e+00],
        [ 5.1196e+00, -2.7160e+00,  4.2705e+00],
        [-1.6086e+00,  7.8928e+00,  4.3385e+00],
        [-4.6393e+00,  7.0461e+00,  1.1624e+00],
        [-3.5903e+00,  7.4847e+00, -8.9112e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 2.1477e+00,  7.2550e+00,  8.9688e+00],
        [ 4.4039e+00,  5.6903e+00,  8.7366e+00],
        [-9.3444e+00,  5.8213e-01,  6.9101e+00],
        [-9.4505e+00, -1.7081e+00, -7.0391e-01],
        [ 3.3094e+00,  1.3497e+01, -3.6507e+00],
        [ 5.9843e+00, -9.8679e+00, -6.1449e+00],
        [ 6.2883e+00,  3.2484e-02,  5.2692e+00],
        [ 1.7232e+00, -4.7574e+00,  3.1270e+00],
        [-1.5451e+00,  7.8975e+00,  7.6104e+00],
        [ 7.8982e+00, -8.6778e+00,  2.3928e-02],
        [-1.9600e+00,  7.8577e+00,  6.9121e+00],
        [-6.5295e+00, -7.9785e+00, -7.4429e+00],
        [ 4.5057e+00, -3.3894e+00, -8.1598e-01],
        [ 9.4516e+00, -7.2501e+00, -1.8868e+00],
        [-9.1749e+00, -3.3803e+00,  6.4629e+00],
        [-3.5173e+00,  7.5091e+00, -8.9585e+00],
        [ 2.2860e+00,  7.1916e+00, -6.1011e+00],
        [-1.9628e-01,  7.8765e+00, -1.1371e+00],
        [-1.3178e+00, -1.0794e+01,  4.2360e-01],
        [ 1.7903e+00, -4.7467e+00,  1.1676e-01],
        [ 5.0617e+00,  1.2796e+01,  8.3544e+00],
        [ 9.6662e+00,  9.1725e+00,  8.1056e+00],
        [ 7.7794e+00,  1.1063e+01,  8.9069e+00],
        [-5.1974e+00,  6.7396e+00,  3.3484e+00],
        [ 6.2365e+00, -2.5781e-01, -6.0212e+00],
        [ 2.0419e+00, -4.6978e+00, -5.0983e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.4359, 0.5029,  ..., 0.5672, 0.4167, 0.3914],
        [0.4359, 0.0000, 0.1236,  ..., 0.2974, 0.5162, 0.4196],
        [0.5029, 0.1236, 0.0000,  ..., 0.1824, 0.6374, 0.5420],
        ...,
        [0.5672, 0.2974, 0.1824,  ..., 0.0000, 0.7890, 0.6992],
        [0.4167, 0.5162, 0.6374,  ..., 0.7890, 0.0000, 0.1003],
        [0.3914, 0.4196, 0.5420,  ..., 0.6992, 0.1003, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.09357547,  8.31625171,  8.43626114,  8.4486771 ,  8.50278896,
        8.25828409,  7.96575782, 13.03002338, 13.1471313 , 11.00508681,
       12.41012046,  9.18358562,  8.51928531,  7.66377554,  7.06169391,
        7.5403822 ,  5.5513072 ,  9.96391045,  9.04255655, 10.87409005,
        9.25336371, 14.07623266, 11.05317597,  9.15866607, 10.66631784,
       10.53368644, 11.52733581,  9.50913849,  9.48337063,  8.43626114,
        7.15100303,  5.83560009,  4.76969364,  9.5014146 ,  6.1561646 ,
        7.32950385,  7.822854  , 11.87412439,  7.81509631, 11.65923396,
       11.74901778, 12.22503754, 13.45642379, 10.48315436,  8.01009083,
       10.35093485,  8.12043983,  9.15253344, 11.018084  ,  5.39723241,
        7.66377554, 11.72802491, 11.05413545,  5.70405525, 10.4606265 ,
       11.018084  ,  8.14433542,  8.50268322,  9.77588801,  6.39438044,
       11.4088043 ,  9.17905491,  6.12886024,  8.85572292,  8.57560275,
        6.61386436, 13.46553623,  9.34127013,  9.0368458 , 13.45124917,
       10.1979669 ,  7.34250948,  8.62162372,  6.28835101,  4.91296205,
        5.79543038,  8.0550286 ,  8.43626114,  8.30125201, 10.00410625,
        7.56617037,  7.19534146,  9.36256096,  9.60359158, 13.89671314,
       11.54071859,  6.28835101,  5.05989547,  8.04718433, 11.73397309,
        8.09842704, 10.30972193,  5.63824971, 11.9120321 ,  9.77777346,
        8.29203356,  7.54620849,  7.8788965 , 10.87409005,  5.07306274,
       13.76047999, 13.32557133, 13.52430176,  8.51088913,  6.24187044,
        5.1224079 ])}
{'P': tensor([[1.7125e-01, 1.2409e-01, 1.1514e-01,  ..., 1.7274e-05, 5.2979e-04,
         7.6467e-07],
        [5.5437e-05, 5.1035e-02, 1.7213e-01,  ..., 4.6593e-05, 1.1402e-05,
         3.9322e-06],
        [3.6110e-05, 6.1481e-03, 4.2488e-02,  ..., 3.5174e-05, 9.2147e-06,
         3.4889e-06],
        ...,
        [1.0973e-03, 1.3386e-03, 1.0495e-03,  ..., 1.6978e-01, 6.4690e-05,
         1.0937e-03],
        [1.7464e-03, 1.9289e-03, 2.6772e-03,  ..., 1.9817e-06, 1.2243e-01,
         4.1568e-07],
        [3.9013e-05, 1.9196e-05, 1.7381e-05,  ..., 1.1751e-03, 3.3070e-05,
         3.2158e-02]]), 'X': tensor([[  0.5175,  14.0910,   4.6540],
        [ -4.3963,   7.1629,   4.5263],
        [ -4.7389,   6.9953,   3.7638],
        [ -4.4503,   7.1378,   3.3751],
        [ -4.2040,   7.2486,   2.9875],
        [ -3.1671,   7.6158,   6.0765],
        [ -3.5173,   7.5091,  -8.9585],
        [ -9.2974,   0.8371,   2.4852],
        [  1.7228,  -4.7575,   8.3396],
        [ -5.2589,   6.7024,  -5.5809],
        [  6.3200,   0.2690,   1.5900],
        [  6.6319,  11.9066,   3.5165],
        [ 12.6035,   0.7494,   7.1536],
        [  4.9957,  -2.8705,   6.5978],
        [  8.0877,  -8.5297,  -1.1050],
        [  0.2435,   7.8191,  -6.4709],
        [  3.9022,  13.2906,   5.5788],
        [  0.1497,   7.8334,  -5.4412],
        [ 11.7795,  -3.4697,   3.5624],
        [  0.6933,   7.7336,   2.1179],
        [ -8.9942,  -3.9990,   6.0448],
        [ -6.2668,   5.9808,   2.5614],
        [ -7.6293,  -6.6859,  -5.5161],
        [  1.7218,  -4.7576,  -3.3617],
        [ 11.5456,   6.0861,   8.1101],
        [  5.2199,   4.6861,  -1.9304],
        [  0.9659,  -4.8144,  -0.9594],
        [  1.1994,   7.6035,   2.4295],
        [  4.7544,  -3.1425,  -1.5315],
        [  3.3069,   6.6089,   8.7377],
        [  7.9816,  10.8938,  -0.5378],
        [  4.1297,  -3.7098,  -3.9838],
        [ -4.4520,   7.1370,  -8.7867],
        [ 12.0772,  -2.6014,  -7.1832],
        [  3.9515,  -3.8425,   6.1729],
        [ 10.8218,  -5.4177,   6.8573],
        [  6.3200,   0.2690,   1.5900],
        [  4.8051, -10.3703,   0.5268],
        [  4.5767,   5.5071,   3.9069],
        [ -2.5683, -10.4433,  -2.9182],
        [  0.4119,  14.1019,  -4.1870],
        [  6.3589,   0.8239,  -8.4230],
        [ -9.4630,  -1.5145,   0.2983],
        [  5.9548,  -1.2273,  -2.6520],
        [  5.7963,  -1.6018,   1.8856],
        [ -3.2736, -10.1703,   3.8712],
        [  4.1140,  -3.7220,  -0.1273],
        [  9.4925,   9.3811,   8.9842],
        [  6.0054,  -1.0902,   8.3079],
        [ -7.7239,   4.4298,   3.1216],
        [ -9.0604,   1.7707,   1.1066],
        [  1.6408,   7.4591,  -4.2294],
        [  6.3124,   0.2057,  -0.4593],
        [ 11.6801,  -3.7196,  -0.9181],
        [  2.3261,  13.7749,   5.9121],
        [  0.6142,  14.0802,   0.5407],
        [ -9.4716,  -0.6673,   7.6068],
        [ 11.5198,  -4.0923,   5.5088],
        [  2.0937,  13.8293,  -2.3533],
        [  6.9693,  -9.3200,  -8.3822],
        [  0.9788,  -4.8145,  -3.8661],
        [  6.3205,   1.7068,  -3.8619],
        [  5.9587,  12.3216,   9.1695],
        [ 12.4996,   2.6335,  -6.3166],
        [ -3.2736, -10.1703,   3.8712],
        [  3.2990,  -4.2415,  -7.4857],
        [  7.7750,  11.0665,   2.8364],
        [ -9.0604,   1.7707,   1.1066],
        [ -8.8154,   2.4611,   2.5198],
        [ 11.2847,  -4.5848,  -4.7503],
        [  1.1274,  14.0116,   8.1753],
        [  6.9433,  -9.3362,  -0.3421],
        [ -5.5525,   6.5146,  -4.4984],
        [  2.9410,   6.8428,   5.4458],
        [ -5.6792,  -8.7352,   4.0318],
        [  2.8994,   6.8675,  -9.5987],
        [  2.0547,   7.2957,  -1.9343],
        [  1.9312,  13.8649,   8.3563],
        [  8.4691,  10.4571,  -5.0650],
        [  7.5064,  -8.9651,  -0.0994],
        [  2.6451,  -4.5250,  -0.5565],
        [ -9.2301,  -3.1537,  -2.7477],
        [ 12.4690,  -0.8528,  -9.3374],
        [ 11.0864,   7.0387,  -0.1238],
        [  3.3943,   6.5483,   1.0790],
        [ -4.9217,   6.8979,  -1.7667],
        [ 12.4597,  -0.9136,  -4.8763],
        [  4.4633,   5.6287,  -0.7754],
        [  8.7131,  10.2216,   7.7915],
        [  5.6523,   3.9443,  -2.9449],
        [ -9.4296,  -1.9458,   4.0833],
        [ 10.3251,   8.2928,  -3.0006],
        [  1.4749,  -4.7892,  -0.8089],
        [  6.3301,   0.3638,  -7.5685],
        [  5.1203,  -2.7151,   6.2110],
        [  5.9526,   3.2525,  -4.3821],
        [  1.9204,   7.3518,   8.4781],
        [  4.5340,  -3.3629,   8.9652],
        [ -7.6650,   4.5092,   0.2520],
        [  5.7151,  -1.7700,   3.1542],
        [  3.6703,   6.3443,  -5.4401],
        [ -5.6792,  -8.7352,   4.0318],
        [ -1.1885, -10.8211,   0.0865],
        [  5.8889,  -1.3916,   8.8993],
        [  2.0449,  13.8402,  -7.1642],
        [ 10.4758,  -5.9539,   1.3667]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.6470, 0.6865,  ..., 0.1538, 1.0819, 0.5542],
        [0.6470, 0.0000, 0.0636,  ..., 0.4957, 1.0698, 0.4033],
        [0.6865, 0.0636, 0.0000,  ..., 0.5379, 1.0323, 0.3838],
        ...,
        [0.1539, 0.4957, 0.5379,  ..., 0.0000, 1.0674, 0.4639],
        [1.0819, 1.0698, 1.0323,  ..., 1.0674, 0.0000, 0.6888],
        [0.5542, 0.4033, 0.3838,  ..., 0.4639, 0.6888, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.10045797,  8.40445872,  8.44940661,  8.41148722,  8.37954853,
        8.24808832,  8.29203356,  9.3349858 ,  5.05981914,  8.51928531,
        6.32572323, 13.62896593, 12.62576026,  5.76165934, 11.75438006,
        7.822854  , 13.85158417,  7.83486943, 12.27991991,  7.76457095,
        9.84314712,  8.66273171, 10.14436912,  5.05963217, 13.05148826,
        7.01475376,  4.91038884,  7.69752262,  5.69913817,  7.39004226,
       13.50484539,  5.5513072 ,  8.41171369, 12.35421461,  5.51177501,
       12.10222204,  6.32572323, 11.42947632,  7.16058587, 10.75443106,
       14.10796466,  6.41202986,  9.58347456,  6.07992623,  6.01356004,
       10.68416401,  5.54776934, 13.34587027,  6.10360073,  8.90405823,
        9.23178204,  7.6374603 ,  6.31576264, 12.25806559, 13.96987806,
       14.09357547,  9.49511524, 12.22503754, 13.98691426, 11.63765273,
        4.91296205,  6.54693987, 13.68672855, 12.77402351, 10.68416401,
        5.37343173, 13.5247215 ,  9.23178204,  9.15253344, 12.18045978,
       14.05687815, 11.63502679,  8.55982105,  7.44805901, 10.41909211,
        7.4544866 ,  7.57946459, 13.99877413, 13.45642379, 11.69264243,
        5.24137909,  9.75401449, 12.4980845 , 13.13205357,  7.37576154,
        8.47371868, 12.49317563,  7.18354885, 13.43127131,  6.89245352,
        9.6282763 , 13.2430377 ,  5.01114719,  6.34059021,  5.79562691,
        6.78325151,  7.5984809 ,  5.64500512,  8.89301516,  5.98285319,
        7.32950385, 10.41909211, 10.88618114,  6.05113217, 13.99048094,
       12.0495521 ])}
{'P': tensor([[1.4647e-01, 1.0793e-01, 1.0610e-01,  ..., 1.2004e-03, 2.4085e-05,
         1.4957e-05],
        [4.7301e-05, 3.0546e-02, 3.8074e-02,  ..., 1.1099e-04, 1.9310e-05,
         4.7610e-05],
        [6.0276e-05, 2.0318e-02, 4.1110e-02,  ..., 1.5119e-04, 2.7962e-05,
         6.0019e-05],
        ...,
        [6.4641e-03, 2.9128e-02, 3.7600e-02,  ..., 8.2664e-02, 3.6953e-04,
         5.6547e-05],
        [1.3579e-03, 1.2664e-03, 1.4558e-03,  ..., 4.4682e-02, 6.2828e-02,
         8.6173e-05],
        [2.1449e-04, 5.5753e-04, 5.0492e-04,  ..., 8.3604e-04, 1.7294e-03,
         1.5102e-01]]), 'X': tensor([[ 4.1193e-01,  1.4102e+01, -4.1870e+00],
        [-3.9700e+00,  7.3453e+00, -4.1486e+00],
        [-4.0409e+00,  7.3169e+00, -3.5779e+00],
        [-4.1296e+00,  7.2803e+00, -2.7194e+00],
        [-5.2589e+00,  6.7024e+00, -5.5809e+00],
        [-5.5525e+00,  6.5146e+00, -4.4984e+00],
        [ 2.5010e+00, -4.5736e+00, -8.9890e-01],
        [ 6.3633e-01, -1.1035e+01,  6.1414e+00],
        [-3.3812e+00, -1.0123e+01,  7.4384e+00],
        [ 1.0757e+01,  7.6212e+00, -2.3279e+00],
        [-8.0608e+00,  3.9362e+00, -5.9567e+00],
        [-9.4309e+00, -5.9902e-02, -4.0830e+00],
        [ 6.2555e+00,  2.1474e+00,  4.1685e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-4.2271e+00,  7.2387e+00, -1.8316e+00],
        [-1.1885e+00, -1.0821e+01,  8.6494e-02],
        [ 1.0325e+01,  8.2928e+00, -3.0006e+00],
        [-3.1671e+00,  7.6158e+00,  6.0765e+00],
        [-9.2707e+00,  9.6612e-01,  4.4863e+00],
        [ 1.3670e+00, -4.7990e+00,  3.2603e+00],
        [ 7.1697e+00,  1.1534e+01, -7.2756e-01],
        [ 4.4633e+00,  5.6287e+00, -7.7540e-01],
        [ 5.8450e+00, -9.9356e+00,  3.0767e+00],
        [ 1.1902e+01, -3.1379e+00, -3.4534e+00],
        [-9.4157e+00,  8.2118e-02, -5.0025e+00],
        [ 6.3236e+00,  3.0105e-01, -4.6668e+00],
        [ 5.6414e+00, -1.0030e+01,  7.8525e+00],
        [ 1.1474e+01,  6.2478e+00, -2.1100e+00],
        [-8.9282e+00,  2.1648e+00,  8.4894e+00],
        [-5.0550e+00,  6.8231e+00, -9.2502e+00],
        [ 5.9526e+00,  3.2525e+00, -4.3821e+00],
        [-1.7009e+00,  7.8851e+00, -8.2437e+00],
        [ 6.1973e+00, -4.3723e-01,  6.5720e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 1.1594e+01,  5.9740e+00, -4.3952e+00],
        [ 9.6614e+00,  9.1784e+00, -1.2087e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 1.1742e+01,  5.6070e+00, -1.4192e+00],
        [ 6.3301e+00,  3.6378e-01, -7.5685e+00],
        [-8.3717e+00, -5.4769e+00, -9.2228e+00],
        [ 2.0937e+00,  1.3829e+01, -2.3533e+00],
        [ 1.2520e+01, -4.7113e-01,  3.0436e+00],
        [ 1.1470e+01, -4.2006e+00,  7.4086e+00],
        [ 1.2004e+01, -2.8350e+00,  2.6864e+00],
        [ 9.3560e+00,  9.5391e+00,  9.4943e+00],
        [ 2.4572e+00,  7.1085e+00,  7.7376e+00],
        [ 9.1409e+00, -7.5789e+00,  1.5814e+00],
        [ 3.6817e+00,  6.3354e+00,  6.1233e+00],
        [ 6.1066e+00, -7.7986e-01, -4.1746e+00],
        [ 5.1667e+00, -2.6543e+00,  1.2879e+00],
        [-8.1426e+00,  3.8048e+00,  9.3542e+00],
        [ 5.6258e+00, -1.9404e+00, -6.0715e+00],
        [ 3.4604e+00, -4.1545e+00,  1.8091e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [ 2.6613e+00,  7.0021e+00,  4.5783e+00],
        [ 1.1312e+01,  6.5957e+00, -4.5667e+00],
        [ 1.2572e+01,  1.9266e+00,  7.4903e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 9.9756e+00, -6.6330e+00, -3.3429e+00],
        [ 1.7659e+00,  7.4127e+00,  4.7871e+00],
        [ 1.1780e+01, -3.4697e+00,  3.5624e+00],
        [-1.6257e-01, -1.0980e+01,  6.4629e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 1.0506e+01, -5.9099e+00, -5.1599e+00],
        [-1.2454e+00,  7.9127e+00, -5.7153e-01],
        [-6.3231e-01,  7.9085e+00, -5.9544e+00],
        [ 3.4931e-01, -4.7727e+00, -6.5682e+00],
        [-8.5687e-01,  7.9156e+00, -6.5557e+00],
        [-8.8058e+00,  2.4850e+00, -1.7425e+00],
        [-7.2515e+00,  5.0195e+00,  4.6434e+00],
        [ 1.2565e+01,  2.0132e+00,  4.6272e+00],
        [-4.3415e-01, -1.0947e+01,  7.4509e+00],
        [ 5.2911e+00, -2.4824e+00, -3.6877e+00],
        [ 4.4049e+00,  5.6892e+00, -4.4394e+00],
        [ 6.3569e+00,  7.5912e-01,  6.5011e+00],
        [ 1.0391e+00, -4.8141e+00, -7.6827e-02],
        [-2.6485e+00, -1.0415e+01,  8.7660e-01],
        [-5.1369e+00,  6.7755e+00,  8.3836e+00],
        [ 4.7482e+00, -1.0391e+01,  5.9149e+00],
        [ 1.2292e+01, -1.7902e+00, -4.0870e+00],
        [ 2.5636e+00, -1.0929e+01,  9.6519e+00],
        [ 6.8407e+00,  1.1767e+01,  3.0045e+00],
        [-6.2668e+00,  5.9808e+00,  2.5614e+00],
        [ 1.6232e+00, -4.7717e+00, -2.5655e+00],
        [-3.7089e+00,  7.4434e+00, -1.7881e-03],
        [-8.8540e+00, -4.3971e+00, -5.2144e+00],
        [ 1.4973e-01,  7.8334e+00, -5.4412e+00],
        [ 4.9452e+00, -1.0319e+01, -7.8621e+00],
        [ 3.4009e+00, -4.1874e+00, -5.9751e+00],
        [ 7.1441e+00,  1.1553e+01,  2.3232e+00],
        [ 5.1203e+00, -2.7151e+00,  6.2110e+00],
        [-8.1905e+00,  3.7253e+00,  4.1175e+00],
        [ 1.2469e+01, -8.5278e-01, -9.3374e+00],
        [-7.6734e+00,  4.4980e+00, -5.4248e+00],
        [ 8.5198e+00,  1.0409e+01, -8.6486e+00],
        [ 7.6853e-01, -4.8098e+00, -2.9310e+00],
        [ 4.2096e+00,  5.8815e+00,  1.9230e+00],
        [ 6.3200e+00,  2.6900e-01,  1.5900e+00],
        [-8.4549e-01, -1.0885e+01, -6.7400e+00],
        [ 1.1895e+01, -3.1570e+00,  7.0742e+00],
        [ 6.3124e+00,  2.0571e-01, -4.5926e-01],
        [ 9.3607e+00,  9.5337e+00, -4.5412e+00],
        [ 1.8766e+00,  7.3694e+00,  1.1887e+00],
        [ 6.2022e+00,  2.4092e+00,  3.0349e+00],
        [ 4.0421e+00, -3.7765e+00, -8.3631e+00]]), 'num flow neighbors': 5, 'precomputed_distances': tensor([[0.0000, 0.4997, 0.5066,  ..., 0.4861, 0.5842, 0.1135],
        [0.4997, 0.0000, 0.0361,  ..., 0.4397, 0.6723, 0.4241],
        [0.5066, 0.0361, 0.0000,  ..., 0.4114, 0.6456, 0.4375],
        ...,
        [0.4861, 0.4397, 0.4114,  ..., 0.0000, 0.2362, 0.5108],
        [0.5842, 0.6723, 0.6456,  ..., 0.2362, 0.0000, 0.6475],
        [0.1135, 0.4241, 0.4375,  ..., 0.5108, 0.6475, 0.0000]]), 'neighbors': tensor([[0., 0., 0., 0., 0.],
        [1., 2., 3., 4., 5.]]), 'labels': array([14.10796466,  8.34949383,  8.35856487,  8.36995953,  8.51928531,
        8.55982105,  5.21280156, 11.05317597, 10.67322527, 13.18278606,
        8.97052115,  9.43112957,  6.61386436, 12.4980845 ,  8.38252937,
       10.88618114, 13.2430377 ,  8.24808832,  9.32094055,  4.98989906,
       13.58101794,  7.18354885, 11.52733581, 12.30858677,  9.41605684,
        6.33075682, 11.50791859, 13.06498554,  9.18690018,  8.49161154,
        6.78325151,  8.06643487,  6.21275016, 13.45124917, 13.04217487,
       13.32613445, 11.42420508,  7.23276582, 13.01187484,  6.34059021,
       10.00410625, 13.98691426, 12.52875793, 12.21532282, 12.3344536 ,
       13.3614598 ,  7.52117132, 11.87412439,  7.32754882,  6.1561646 ,
        5.80860165,  8.98765644,  5.9510489 ,  5.40689332,  4.78544875,
        7.49076899, 13.09426142, 12.71843462,  7.23276582, 11.97957425,
        7.62011734, 12.27991991, 10.98076873, 13.58333857, 12.05395056,
        8.01009083,  7.9337647 ,  4.78544875,  7.96181288,  9.14973319,
        8.81930525, 12.72524615, 10.95593729,  5.84450127,  7.19513806,
        6.4020404 ,  4.92497514, 10.74655346,  8.50268322, 11.42420508,
       12.4217523 , 11.22596994, 13.6105595 ,  8.66273171,  5.04028538,
        8.31625171,  9.88571713,  7.83486943, 11.4424816 ,  5.39450305,
       13.58333857,  5.79562691,  8.99791316, 12.4980845 ,  8.89458273,
       13.45124917,  4.87083348,  7.23276582,  6.32572323, 10.91805683,
       12.30695098,  6.31576264, 13.36092624,  7.60463349,  6.65368551,
        5.53175275])}
{% endraw %}

Dataloaders

Because we're handling batching in a, sahll we say, delicate way, the standard way of batching data doesn't work (without some downstream effort). In fact, we're treating each output element of the dataset as a batch by itself, so we can get nice behavior just by disabling batching entirely.

{% raw %}
from torch.utils.data import DataLoader
dataloader = DataLoader(ds, batch_size=None)
{% endraw %} {% raw %}
for d in dataloader:
    print(d["X"].type())
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
torch.FloatTensor
{% endraw %}

Convenience functions

{% raw %}

dataloader_from_ndarray[source]

dataloader_from_ndarray(X, flow, labels)

{% endraw %} {% raw %}
{% endraw %} {% raw %}
dataloader.dataset.X
tensor([[ 0.1206, -4.7363,  8.7103],
        [ 0.1532, -4.7422, -7.0145],
        [ 0.1677, -4.7447, -6.1529],
        ...,
        [ 0.6142, 14.0802,  0.5407],
        [ 0.5175, 14.0910,  4.6540],
        [ 0.4119, 14.1019, -4.1870]])
{% endraw %}